获取进程主线程的句柄

Question

我在一些小型测试应用程序中创建了一个附加线程，并希望从该附加线程中挂起主线程。附加线程是通过外部进程的

CreateRemoteThread

创建的。

由于

SuspendThread

需要一个

HANDLE

到应该挂起的线程，我想知道如何从我的附加线程中运行的代码中获取这个

HANDLE

。

Answer 1

我认为一旦进程启动，主线程与其他线程就没有任何区别。但是，您可以枚举进程中的所有线程，并使用GetThreadTimes查找创建时间最早的线程。调用 OpenThread 从线程 ID 中获取

HANDLE

。

Answer 2

使用此函数获取线程ID：

/* 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;
}

简单打开线程获取句柄：

/*
 * 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);
}

Answer 3

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 4

为什么不创建一个程序范围的全局（如果必须的话，使用 extern）

HANDLE mainThread ;
DWORD mainThreadId ;

在 main 的第一行，（在创建任何线程之前）执行

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

您可以使用任何形式的 IPC 与远程进程共享 id 或 HANDLE（尚未验证共享 HANDLE 是否有效，但应该可以！）

Answer 5

许多此类有用的 API 函数都位于（当然！）工具帮助套件中。

CreateToolhelp32Snapshot()

API 将为指定进程的正在运行的线程拍摄快照。

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

完整示例代码在这里。

返回的结构体并不区分主线程和其他线程。我不知道有什么机制可以做到这一点；虽然某些版本的 C 运行时将所有 ExitProcess() 都放在主线程末尾，但在所有最新版本中，该进程将继续运行，直到最后一个线程退出。

Interjay 建议使用 GetThreadTimes 可能是最好的选择。如果您可以

CreateProcess()

目标进程，则

PROCESS_INFORMATION

块的 hThread 成员包含主线程的 tid。欢迎其他人提出任何想法。