为何SetThreadAffinityMask设置线程亲和性不生效？

你的代码无法设置线程亲和性主要有三个核心问题，下面逐一说明并给出修复后的代码：

1. 线程句柄权限不足 #

调用OpenThread时仅传入QUERY_INFORMATION权限，但设置线程亲和性需要SET_INFORMATION权限。必须在打开线程时请求足够的权限，否则后续的亲和性设置调用会失败。

2. 忽略处理器组信息 #

PROCESSOR_NUMBER结构体中的Group字段被完全忽略。在现代多处理器组的Windows系统中，处理器会被划分为不同的组，亲和性掩码是组内局部的。直接使用1 << processorNumber.Number无法关联到正确的处理器组，应该使用SetThreadGroupAffinityAPI来同时指定组和组内掩码。

3. 未检查API调用返回值 #

代码中没有对任何Win32 API的返回结果做校验，无法定位调用失败的具体环节，比如线程句柄是否成功打开、亲和性设置是否成功。

修复后的代码 #

using System;
using System.Diagnostics;
using System.Runtime.InteropServices;

namespace ThreadAffinity
{
    class Program
    {
        [DllImport("kernel32.dll")]
        static extern IntPtr OpenThread(ThreadAccess dwDesiredAccess, bool bInheritHandle, uint dwThreadId);

        [DllImport("kernel32.dll")]
        static extern bool SetThreadGroupAffinity(IntPtr hThread, ref GROUP_AFFINITY GroupAffinity, IntPtr PreviousGroupAffinity);

        [DllImport("kernel32.dll")]
        static extern bool GetThreadIdealProcessorEx(IntPtr hThread, ref PROCESSOR_NUMBER lpIdealProcessor);

        [DllImport("kernel32.dll")]
        static extern bool CloseHandle(IntPtr hObject);

        [StructLayout(LayoutKind.Sequential)]
        public struct PROCESSOR_NUMBER
        {
            public ushort Group;
            public byte Number;
            public byte Reserved;
        }

        [StructLayout(LayoutKind.Sequential)]
        public struct GROUP_AFFINITY
        {
            public IntPtr Mask;
            public ushort Group;
            public ushort Reserved[3];
        }

        [Flags]
        public enum ThreadAccess : int
        {
            QUERY_INFORMATION = 0x0040,
            SET_INFORMATION = 0x0020,
            THREAD_ALL_ACCESS = 0x1F03FF
        }

        static void Main(string[] args)
        {
            Console.WriteLine("Enter process id:");
            if (!int.TryParse(Console.ReadLine(), out int processId))
            {
                Console.WriteLine("Invalid process ID");
                Console.ReadKey();
                return;
            }

            Process process;
            try
            {
                process = Process.GetProcessById(processId);
            }
            catch (ArgumentException)
            {
                Console.WriteLine("Process not found");
                Console.ReadKey();
                return;
            }

            foreach (ProcessThread thread in process.Threads)
            {
                // 打开线程时请求必要的权限：查询+设置信息
                IntPtr hThread = OpenThread(ThreadAccess.QUERY_INFORMATION | ThreadAccess.SET_INFORMATION, false, (uint)thread.Id);
                if (hThread == IntPtr.Zero)
                {
                    Console.WriteLine($"Failed to open thread {thread.Id}, error: {Marshal.GetLastWin32Error()}");
                    continue;
                }

                PROCESSOR_NUMBER processorNumber = new PROCESSOR_NUMBER();
                if (!GetThreadIdealProcessorEx(hThread, ref processorNumber))
                {
                    Console.WriteLine($"Failed to get ideal processor for thread {thread.Id}, error: {Marshal.GetLastWin32Error()}");
                    CloseHandle(hThread);
                    continue;
                }

                Console.WriteLine($"Thread {thread.Id} ideal processor: Group {processorNumber.Group}, Core {processorNumber.Number}");

                // 构建组亲和性结构体
                GROUP_AFFINITY groupAffinity = new GROUP_AFFINITY
                {
                    Group = processorNumber.Group,
                    Mask = (IntPtr)(1L << processorNumber.Number) // 使用long避免32位溢出
                };

                // 设置组亲和性
                if (!SetThreadGroupAffinity(hThread, ref groupAffinity, IntPtr.Zero))
                {
                    Console.WriteLine($"Failed to set affinity for thread {thread.Id}, error: {Marshal.GetLastWin32Error()}");
                }
                else
                {
                    Console.WriteLine($"Successfully set affinity for thread {thread.Id} to Group {processorNumber.Group}, Core {processorNumber.Number}");
                }

                CloseHandle(hThread);
            }

            Console.WriteLine("Press any key to exit...");
            Console.ReadKey();
        }
    }
}

关键修改说明 #

• 权限修正：OpenThread时传入QUERY_INFORMATION | SET_INFORMATION，确保有设置线程信息的权限。
• 处理器组支持：改用SetThreadGroupAffinityAPI，通过GROUP_AFFINITY结构体同时指定处理器组和组内核心掩码，适配多处理器组系统。
• 返回值校验：添加了对每个Win32 API调用的结果检查，并通过Marshal.GetLastWin32Error()获取具体错误码，方便调试。
• 类型安全：使用1L << processorNumber.Number避免32位系统下的整数溢出问题。

内容的提问来源于stack exchange，提问作者Aspector