//---------------------------------------------------------------------------------------
//
// Initialize the static instance and lock.
//
HRESULT
LOADEDMODULES::InitializeStatics()
{
HRESULT hr = S_OK;
if (VolatileLoad(&s_pLoadedModules) == NULL)
{
// Initialize global read-write lock
{
NewHolder<UTSemReadWrite> pSemReadWrite = new (nothrow) UTSemReadWrite();
IfNullGo(pSemReadWrite);
IfFailGo(pSemReadWrite->Init());
if (InterlockedCompareExchangeT<UTSemReadWrite *>(&m_pSemReadWrite, pSemReadWrite, NULL) == NULL)
{ // We won the initialization race
pSemReadWrite.SuppressRelease();
}
}
// Initialize the global instance
{
NewHolder<LOADEDMODULES> pLoadedModules = new (nothrow) LOADEDMODULES();
IfNullGo(pLoadedModules);
{
LOCKWRITE();
if (VolatileLoad(&s_pLoadedModules) == NULL)
{
VolatileStore(&s_pLoadedModules, pLoadedModules.Extract());
}
}
}
}
ErrExit:
return hr;
} // LOADEDMODULES::InitializeStatics
// Given a PID attempt to find or create a DbgTransportSession instance to manage a connection to a runtime in
// that process. Returns E_UNEXPECTED if the process can't be found. Also returns a handle that can be waited
// on for process termination.
HRESULT DbgTransportTarget::GetTransportForProcess(DWORD dwPID,
DbgTransportSession **ppTransport,
HANDLE *phProcessHandle)
{
RSLockHolder lock(&m_sLock);
HRESULT hr = S_OK;
ProcessEntry *entry = LocateProcessByPID(dwPID);
if (entry == NULL)
{
NewHolder<ProcessEntry> newEntry = new(nothrow) ProcessEntry();
if (newEntry == NULL)
return E_OUTOFMEMORY;
NewHolder<DbgTransportSession> transport = new(nothrow) DbgTransportSession();
if (transport == NULL)
{
return E_OUTOFMEMORY;
}
HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwPID);
if (hProcess == NULL)
{
transport->Shutdown();
return HRESULT_FROM_GetLastError();
}
// Initialize it (this immediately starts the remote connection process).
hr = transport->Init(dwPID, hProcess);
if (FAILED(hr))
{
transport->Shutdown();
CloseHandle(hProcess);
return hr;
}
entry = newEntry;
newEntry.SuppressRelease();
entry->m_dwPID = dwPID;
entry->m_hProcess = hProcess;
entry->m_transport = transport;
transport.SuppressRelease();
entry->m_cProcessRef = 0;
// Adding new entry to the list.
entry->m_pNext = m_pProcessList;
m_pProcessList = entry;
}
entry->m_cProcessRef++;
_ASSERTE(entry->m_cProcessRef > 0);
_ASSERTE(entry->m_transport != NULL);
_ASSERTE(entry->m_hProcess > 0);
*ppTransport = entry->m_transport;
if (!DuplicateHandle(GetCurrentProcess(),
entry->m_hProcess,
GetCurrentProcess(),
phProcessHandle,
0, // ignored since we are going to pass DUPLICATE_SAME_ACCESS
FALSE,
DUPLICATE_SAME_ACCESS))
{
return HRESULT_FROM_GetLastError();
}
return hr;
}