获取进程主线程的句柄

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()< /a> 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 块包含主线程的 tid。欢迎其他人提出任何想法。