Manejando el hilo principal del proceso

Question

He creado un hilo adicional en una pequeña aplicación de prueba y quiero suspender el hilo principal de este hilo adicional. El hilo adicional se crea a través de CreateRemoteThread desde un proceso externo.

Dado que SuspendThread necesita un HANDLE al hilo que se debe suspender, quiero saber cómo obtener este HANDLE del código que se ejecuta en el hilo adicional.

Answer 1

DWORD GetMainThreadId() { 
    const std::tr1::shared_ptr<void> hThreadSnapshot(
     CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD, 0), CloseHandle); 
    if (hThreadSnapshot.get() == INVALID_HANDLE_VALUE) { 
     throw std::runtime_error("GetMainThreadId failed"); 
    } 
    THREADENTRY32 tEntry; 
    tEntry.dwSize = sizeof(THREADENTRY32); 
    DWORD result = 0; 
    DWORD currentPID = GetCurrentProcessId(); 
    for (BOOL success = Thread32First(hThreadSnapshot.get(), &tEntry); 
     !result && success && GetLastError() != ERROR_NO_MORE_FILES; 
     success = Thread32Next(hThreadSnapshot.get(), &tEntry)) 
    { 
     if (tEntry.th32OwnerProcessID == currentPID) { 
      result = tEntry.th32ThreadID; 
     } 
    } 
    return result; 
} 

Answer 2

No creo que haya algo que diferencie el hilo principal de otros hilos una vez que el proceso haya comenzado. Sin embargo, puede enumerate all threads in the process, y usar GetThreadTimes para encontrar el hilo con la hora de creación más temprana. Llame al OpenThread para obtener un HANDLE desde un ID de hilo.

Answer 3

Una serie de útiles funciones de API de este tipo se encuentran bajo (¡por supuesto!) Tool Help suite. La API CreateToolhelp32Snapshot() tomará una instantánea de los subprocesos en ejecución para un proceso específico.

// Take a snapshot of all running threads 
hThreadSnap = CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD, 0); 
if(hThreadSnap == INVALID_HANDLE_VALUE) 
    return(FALSE); 

ejemplo completo de código here.

La estructura devolver no diferenciar el hilo principal de las otras. No conozco un mecanismo para hacerlo; mientras que algunas versiones del tiempo de ejecución de C tendrán ExitProcess() al final del subproceso principal, en todas las versiones recientes, el proceso continúa ejecutándose hasta que finalice el último.

La recomendación de Interjay de utilizar GetThreadTimes puede ser la mejor opción. Si puede CreateProcess() el proceso de destino, el miembro hThread del bloque PROCESS_INFORMATION contiene el tid para el hilo primario. Bienvenido cualquier idea de los demás.

Answer 4

¿Por qué no basta con crear un amplio programa mundial (uso extern si tiene que)

HANDLE mainThread ; 
DWORD mainThreadId ; 

En la primera línea del principal, (antes de crear hilos) hacer

mainThread = GetCurrentThread() ; 
mainThreadId = GetCurrentThreadId() ; 

puede utilizar any form of IPC compartir el ID o el mango con el proceso remoto (no han verificado compartir el MANGO funcionará pero debería!)

Answer 5

obtener el ID del hilo con esta función:

/* CAUTION: ONLY x86 TESTED 
* get the thread id of the main thread of a target process 
* 
* params: 
*  DWORD dwPid process id of the target process 
* 
* return: 
*  Success  thread id 
*  Error  NULL 
*/ 
DWORD GetMainThreadId(DWORD dwPid) 
{ 
    LPVOID lpTid; 

    _asm 
    { 
     mov eax, fs:[18h] 
     add eax, 36 
     mov [lpTid], eax 
    } 

    HANDLE hProcess = OpenProcess(PROCESS_VM_READ, FALSE, dwPid); 
    if(hProcess == NULL) 
     return NULL; 

    DWORD dwTid; 
    if(ReadProcessMemory(hProcess, lpTid, &dwTid, sizeof(dwTid), NULL) == FALSE) 
    { 
     CloseHandle(hProcess); 
     return NULL; 
    } 

    CloseHandle(hProcess); 

    return dwTid; 
} 

simple abrir el hilo para obtener el identificador:

/* 
* get a handle to the main thread of a target process 
* if successfull, the returned handle must be closed with CloseHandle() 
* 
* params: 
*  DWORD dwPid    process id of the target process 
*  DWORD dwDesiredAccess desired access rights to the thread 
* 
* return: 
*  Success  thread handle with desired access rights 
*  Error  NULL 
*/ 
HANDLE GetMainThreadHandle(DWORD dwPid, DWORD dwDesiredAccess) 
{ 
    DWORD dwTid = GetMainThreadId(dwPid); 
    if(dwTid == FALSE) 
     return NULL; 

    return OpenThread(dwDesiredAccess, FALSE, dwTid); 
}