主頁 > 資料庫 > 在具有兩個以上處理器組的雙套接字系統上通過ctypes在Python中使用GetLogicalProcessorInformationEx()

在具有兩個以上處理器組的雙套接字系統上通過ctypes在Python中使用GetLogicalProcessorInformationEx()

2021-10-25 11:03:12 資料庫

所以我試圖通過套接字獲取處理器分組。我在 C 中有一個 POC 可以做到這一點，輸出如下。

[[0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,],[0,1,2,3,4,5,6,7,8,9,10,11,],]
[[12,13,14,15,16,17,18,19,20,21,22,23,],[0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,],]

雖然我可以做以下類似的事情，但我更愿意保持代碼連續性并保持專案純 python。另外，為了體驗，這似乎是一個很好的謎題。

socket_groups = eval(run_cmd_popen("get_logical_processor_information.exe", return_output=True).strip("\n"))

到目前為止我所做的作業，但僅在一個套接字系統上，對于兩個套接字系統，第二個套接字似乎沒有任何資料，并列印以下內容。

[[0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,][0,1,2,3,4,5,6,7,8,9,10,11,]]
[]

這是我到目前為止所擁有的：

import os
import math
import platform
from ctypes.wintypes import BYTE, INT, WORD, DWORD, BOOL, PDWORD, UINT
from ctypes import Structure, Union, WinDLL, c_uint64, POINTER, byref, get_last_error, WinError

from wmi import WMI


if platform.machine().endswith('64'):
    KAFFINITY = c_uint64
else:
    KAFFINITY = UINT

wmi = WMI()
NUM_SOCKETS = 0
for wmi_obj in wmi.query(f'SELECT * FROM Win32_processor '):
    if wmi_obj is not None:
        NUM_SOCKETS  = 1

NUM_CPUS = os.cpu_count()
NUM_CPU_GROUPS = math.ceil(NUM_CPUS / 64)
ANYSIZE_ARRAY = NUM_CPU_GROUPS

ERROR_INSUFFICIENT_BUFFER = 122

RELATION_CACHE = 2
RELATION_NUMA_NODE = 1
RELATION_PROCESSOR_CORE = 0
RELATION_PROCESSOR_PACKAGE = 3
RELATION_GROUP = 4
RELATION_ALL = 0xffff


class _GROUP_AFFINITY(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-group_affinity
        typedef struct _GROUP_AFFINITY {
            KAFFINITY Mask;
            WORD      Group;
            WORD      Reserved[3];
        } GROUP_AFFINITY, *PGROUP_AFFINITY;
    '''
    _fields_ = (
        ('Mask', KAFFINITY),
        ('Group', WORD),
        ('Reserved', WORD * 3)
    )


class _PROCESSOR_GROUP_INFO(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-processor_group_info
        typedef struct _PROCESSOR_GROUP_INFO {
            BYTE      MaximumProcessorCount;
            BYTE      ActiveProcessorCount;
            BYTE      Reserved[38];
            KAFFINITY ActiveProcessorMask;
        } PROCESSOR_GROUP_INFO, *PPROCESSOR_GROUP_INFO;
    '''
    _fields_ = (
        ('MaximumProcessorCount', BYTE),
        ('ActiveProcessorCount', BYTE),
        ('Reserved', BYTE * 38),
        ('ActiveProcessorMask', KAFFINITY),
    )


class _GROUP_RELATIONSHIP(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-group_relationship
        typedef struct _GROUP_RELATIONSHIP {
            WORD                 MaximumGroupCount;
            WORD                 ActiveGroupCount;
            BYTE                 Reserved[20];
            PROCESSOR_GROUP_INFO GroupInfo[ANYSIZE_ARRAY];
        } GROUP_RELATIONSHIP, *PGROUP_RELATIONSHIP;
    '''
    _fields_ = (
        ('MaximumGroupCount', WORD),
        ('ActiveGroupCount', WORD),
        ('Reserved', BYTE * 20),
        ('GroupInfo', _PROCESSOR_GROUP_INFO * ANYSIZE_ARRAY)
    )


class _DUMMYUNIONNAME_CACHE_RELATIONSHIP(Union):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-cache_relationship
        union {
            GROUP_AFFINITY GroupMask;
            GROUP_AFFINITY GroupMasks[ANYSIZE_ARRAY];
        } DUMMYUNIONNAME;
    '''
    _fields_ = (
        ('GroupMask', _GROUP_AFFINITY),
        ('GroupMasks', _GROUP_AFFINITY * ANYSIZE_ARRAY)  # * number of groups
    )


class _CACHE_RELATIONSHIP(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-cache_relationship
        typedef struct _CACHE_RELATIONSHIP {
            BYTE                 Level;
            BYTE                 Associativity;
            WORD                 LineSize;
            DWORD                CacheSize;
            PROCESSOR_CACHE_TYPE Type;
            BYTE                 Reserved[18];
            WORD                 GroupCount;
            union {
                GROUP_AFFINITY GroupMask;
                GROUP_AFFINITY GroupMasks[ANYSIZE_ARRAY];
            } DUMMYUNIONNAME;
        } CACHE_RELATIONSHIP, *PCACHE_RELATIONSHIP;
    '''
    _fields_ = (
        ('Level', BYTE),
        ('Associativity', BYTE),
        ('LineSize', WORD),
        ('CacheSize', DWORD),
        ('Type', INT),  # PROCESSOR_CACHE_TYPE is an Enum type, which, At least for GCC, is just a simple numeric type. (https://stackoverflow.com/questions/1546355/using-enums-in-ctypes-structure/1546467#1546467)
        ('Reserved', BYTE * 18),
        ('GroupCount', WORD),
        ('DUMMYUNIONNAME ', _DUMMYUNIONNAME_CACHE_RELATIONSHIP)
    )


class _DUMMYUNIONNAME_NUMA_NODE_RELATIONSHIP(Union):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-numa_node_relationship
        union {
            GROUP_AFFINITY GroupMask;
            GROUP_AFFINITY GroupMasks[ANYSIZE_ARRAY];
        } DUMMYUNIONNAME;
    '''
    _fields_ = (
        ('GroupMask', _GROUP_AFFINITY),
        ('GroupMasks', _GROUP_AFFINITY * ANYSIZE_ARRAY)  # * number of groups
    )


class _NUMA_NODE_RELATIONSHIP(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-numa_node_relationship
        typedef struct _NUMA_NODE_RELATIONSHIP {
            DWORD NodeNumber;
            BYTE  Reserved[18];
            WORD  GroupCount;
            union {
                GROUP_AFFINITY GroupMask;
                GROUP_AFFINITY GroupMasks[ANYSIZE_ARRAY];
            } DUMMYUNIONNAME;
        } NUMA_NODE_RELATIONSHIP, *PNUMA_NODE_RELATIONSHIP;
    '''
    _fields_ = (
        ('NodeNumber', DWORD),
        ('Reserved', BYTE * 18),
        ('GroupCount', WORD),
        ('DUMMYUNIONNAME', _DUMMYUNIONNAME_NUMA_NODE_RELATIONSHIP)
    )


class _PROCESSOR_RELATIONSHIP(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-processor_relationship
        typedef struct _PROCESSOR_RELATIONSHIP {
            BYTE           Flags;
            BYTE           EfficiencyClass;
            BYTE           Reserved[20];
            WORD           GroupCount;
            GROUP_AFFINITY GroupMask[ANYSIZE_ARRAY];
        } PROCESSOR_RELATIONSHIP, *PPROCESSOR_RELATIONSHIP;
    '''
    _fields_ = (
        ('Flags', BYTE),
        ('EfficiencyClass', BYTE),
        ('Reserved', BYTE * 20),
        ('GroupCount', WORD),
        ('GroupMask', _GROUP_AFFINITY * ANYSIZE_ARRAY)  # * number of groups
    )


class _DUMMYUNIONNAME_SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX(Union):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-system_logical_processor_information_ex
            union {
                PROCESSOR_RELATIONSHIP Processor;
                NUMA_NODE_RELATIONSHIP NumaNode;
                CACHE_RELATIONSHIP     Cache;
                GROUP_RELATIONSHIP     Group;
            } DUMMYUNIONNAME;
    '''
    _fields_ = (
        ('Processor', _PROCESSOR_RELATIONSHIP),
        ('NumaNode', _NUMA_NODE_RELATIONSHIP),
        ('Cache', _CACHE_RELATIONSHIP),
        ('Group', _GROUP_RELATIONSHIP)
    )


class _SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-system_logical_processor_information_ex
        typedef struct _SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX {
            LOGICAL_PROCESSOR_RELATIONSHIP Relationship;
            DWORD                          Size;
            union {
                PROCESSOR_RELATIONSHIP Processor;
                NUMA_NODE_RELATIONSHIP NumaNode;
                CACHE_RELATIONSHIP     Cache;
                GROUP_RELATIONSHIP     Group;
            } DUMMYUNIONNAME;
        } SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX, *PSYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX;
    '''
    _anonymous_ = 'DUMMYUNIONNAME',
    _fields_ = (
        ('Relationship', INT),  # _LOGICAL_PROCESSOR_RELATIONSHIP is an Enum type, which, At least for GCC, is just a simple numeric type. (https://stackoverflow.com/questions/1546355/using-enums-in-ctypes-structure/1546467#1546467)
        ('Size', DWORD),
        ('DUMMYUNIONNAME', _DUMMYUNIONNAME_SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX)
    )


def GetLogicalProcessorInformationEx(relation_type):
    '''https://docs.microsoft.com/en-us/windows/win32/api/sysinfoapi/nf-sysinfoapi-getlogicalprocessorinformationex
        BOOL GetLogicalProcessorInformationEx(
            [in]            LOGICAL_PROCESSOR_RELATIONSHIP           RelationshipType,
            [out, optional] PSYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX Buffer,
            [in, out]       PDWORD                                   ReturnedLength
        );
    '''
    dll = WinDLL('kernel32', use_last_error=True)
    dll.GetLogicalProcessorInformationEx.argtypes = INT, POINTER(_SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX), PDWORD
    dll.GetLogicalProcessorInformationEx.restype = BOOL

    byte_len = DWORD()

    # Call with null buffer to get required buffer size
    result = dll.GetLogicalProcessorInformationEx(relation_type, None, byref(byte_len))
    if (err := get_last_error()) != ERROR_INSUFFICIENT_BUFFER:
        raise WinError(err)

    # Allocate buffer
    allocated_structure = (_SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX * NUM_SOCKETS)()

    global byte_len_for_advance
    byte_len_for_advance = byte_len

    # Now do the call with the allocated buffer to fill it up
    result = dll.GetLogicalProcessorInformationEx(relation_type, allocated_structure, byref(byte_len))
    if not result:
        raise WinError(get_last_error())

    return allocated_structure


def _print_mask(mask):
    # print(f"{mask=}")
    digits = [int(x) for x in mask]

    group_core_list = []
    print("[", end='')
    core = 0
    for i in digits:
        if i == 1:
            print(f"{core},", end='')
            group_core_list.append(core)
        core  = 1
    print("]", end='')

    # print(group_core_list)


if __name__ == "__main__":
    enum_info = GetLogicalProcessorInformationEx(RELATION_PROCESSOR_PACKAGE)
    for p_info in enum_info:
        print("[", end='')
        for group_index in range(0, p_info.Processor.GroupCount):
            _print_mask(f"{p_info.Processor.GroupMask[group_index].Mask:8b}")
        print("]\n", end='')
    breakpoint

我想我需要做一些將指標轉移到下一個結構的事情嗎？但我不知道該怎么做。此外，當我獲得第二個套接字資訊時，我對 _print_mask() 是否會完全按照我的要求執行操作也不是很確定，但那是后來的問題。

任何思想互聯網？

編輯

感謝 Mark Tolonen 的回答/幫助！

這是我的作業解決方案：

from typing import List, Dict
from ctypes.wintypes import BYTE, INT, WORD, DWORD, BOOL, PDWORD, WPARAM
from ctypes import Structure, Union, WinDLL, POINTER, byref, cast, get_last_error, WinError, c_char_p, create_string_buffer

KAFFINITY = WPARAM   # WPARAM is 32-bit or 64-bit unsigned depending on platform

ANYSIZE_ARRAY = 1

ERROR_INSUFFICIENT_BUFFER = 122

RELATION_CACHE = 2
RELATION_NUMA_NODE = 1
RELATION_PROCESSOR_CORE = 0
RELATION_PROCESSOR_PACKAGE = 3
RELATION_GROUP = 4
RELATION_ALL = 0xffff


class _GROUP_AFFINITY(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-group_affinity
        typedef struct _GROUP_AFFINITY {
            KAFFINITY Mask;
            WORD      Group;
            WORD      Reserved[3];
        } GROUP_AFFINITY, *PGROUP_AFFINITY;
    '''
    _fields_ = (
        ('Mask', KAFFINITY),
        ('Group', WORD),
        ('Reserved', WORD * 3)
    )


class _PROCESSOR_GROUP_INFO(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-processor_group_info
        typedef struct _PROCESSOR_GROUP_INFO {
            BYTE      MaximumProcessorCount;
            BYTE      ActiveProcessorCount;
            BYTE      Reserved[38];
            KAFFINITY ActiveProcessorMask;
        } PROCESSOR_GROUP_INFO, *PPROCESSOR_GROUP_INFO;
    '''
    _fields_ = (
        ('MaximumProcessorCount', BYTE),
        ('ActiveProcessorCount', BYTE),
        ('Reserved', BYTE * 38),
        ('ActiveProcessorMask', KAFFINITY),
    )


class _GROUP_RELATIONSHIP(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-group_relationship
        typedef struct _GROUP_RELATIONSHIP {
            WORD                 MaximumGroupCount;
            WORD                 ActiveGroupCount;
            BYTE                 Reserved[20];
            PROCESSOR_GROUP_INFO GroupInfo[ANYSIZE_ARRAY];
        } GROUP_RELATIONSHIP, *PGROUP_RELATIONSHIP;
    '''
    _fields_ = (
        ('MaximumGroupCount', WORD),
        ('ActiveGroupCount', WORD),
        ('Reserved', BYTE * 20),
        ('GroupInfo', _PROCESSOR_GROUP_INFO * ANYSIZE_ARRAY)
    )


class _DUMMYUNIONNAME_CACHE_RELATIONSHIP(Union):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-cache_relationship
        union {
            GROUP_AFFINITY GroupMask;
            GROUP_AFFINITY GroupMasks[ANYSIZE_ARRAY];
        } DUMMYUNIONNAME;
    '''
    _fields_ = (
        ('GroupMask', _GROUP_AFFINITY),
        ('GroupMasks', _GROUP_AFFINITY * ANYSIZE_ARRAY)  # * number of groups
    )


class _CACHE_RELATIONSHIP(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-cache_relationship
        typedef struct _CACHE_RELATIONSHIP {
            BYTE                 Level;
            BYTE                 Associativity;
            WORD                 LineSize;
            DWORD                CacheSize;
            PROCESSOR_CACHE_TYPE Type;
            BYTE                 Reserved[18];
            WORD                 GroupCount;
            union {
                GROUP_AFFINITY GroupMask;
                GROUP_AFFINITY GroupMasks[ANYSIZE_ARRAY];
            } DUMMYUNIONNAME;
        } CACHE_RELATIONSHIP, *PCACHE_RELATIONSHIP;
    '''
    _fields_ = (
        ('Level', BYTE),
        ('Associativity', BYTE),
        ('LineSize', WORD),
        ('CacheSize', DWORD),
        ('Type', INT),  # PROCESSOR_CACHE_TYPE is an Enum type, which, At least for GCC, is just a simple numeric type. (https://stackoverflow.com/questions/1546355/using-enums-in-ctypes-structure/1546467#1546467)
        ('Reserved', BYTE * 18),
        ('GroupCount', WORD),
        ('DUMMYUNIONNAME ', _DUMMYUNIONNAME_CACHE_RELATIONSHIP)
    )


class _DUMMYUNIONNAME_NUMA_NODE_RELATIONSHIP(Union):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-numa_node_relationship
        union {
            GROUP_AFFINITY GroupMask;
            GROUP_AFFINITY GroupMasks[ANYSIZE_ARRAY];
        } DUMMYUNIONNAME;
    '''
    _fields_ = (
        ('GroupMask', _GROUP_AFFINITY),
        ('GroupMasks', _GROUP_AFFINITY * ANYSIZE_ARRAY)  # * number of groups
    )


class _NUMA_NODE_RELATIONSHIP(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-numa_node_relationship
        typedef struct _NUMA_NODE_RELATIONSHIP {
            DWORD NodeNumber;
            BYTE  Reserved[18];
            WORD  GroupCount;
            union {
                GROUP_AFFINITY GroupMask;
                GROUP_AFFINITY GroupMasks[ANYSIZE_ARRAY];
            } DUMMYUNIONNAME;
        } NUMA_NODE_RELATIONSHIP, *PNUMA_NODE_RELATIONSHIP;
    '''
    _fields_ = (
        ('NodeNumber', DWORD),
        ('Reserved', BYTE * 18),
        ('GroupCount', WORD),
        ('DUMMYUNIONNAME', _DUMMYUNIONNAME_NUMA_NODE_RELATIONSHIP)
    )


class _PROCESSOR_RELATIONSHIP(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-processor_relationship
        typedef struct _PROCESSOR_RELATIONSHIP {
            BYTE           Flags;
            BYTE           EfficiencyClass;
            BYTE           Reserved[20];
            WORD           GroupCount;
            GROUP_AFFINITY GroupMask[ANYSIZE_ARRAY];
        } PROCESSOR_RELATIONSHIP, *PPROCESSOR_RELATIONSHIP;
    '''
    _fields_ = (
        ('Flags', BYTE),
        ('EfficiencyClass', BYTE),
        ('Reserved', BYTE * 20),
        ('GroupCount', WORD),
        ('GroupMask', _GROUP_AFFINITY * ANYSIZE_ARRAY)  # * number of groups
    )


class _DUMMYUNIONNAME_SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX(Union):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-system_logical_processor_information_ex
            union {
                PROCESSOR_RELATIONSHIP Processor;
                NUMA_NODE_RELATIONSHIP NumaNode;
                CACHE_RELATIONSHIP     Cache;
                GROUP_RELATIONSHIP     Group;
            } DUMMYUNIONNAME;
    '''
    _fields_ = (
        ('Processor', _PROCESSOR_RELATIONSHIP),
        ('NumaNode', _NUMA_NODE_RELATIONSHIP),
        ('Cache', _CACHE_RELATIONSHIP),
        ('Group', _GROUP_RELATIONSHIP)
    )


class _SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX(Structure):
    '''https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-system_logical_processor_information_ex
        typedef struct _SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX {
            LOGICAL_PROCESSOR_RELATIONSHIP Relationship;
            DWORD                          Size;
            union {
                PROCESSOR_RELATIONSHIP Processor;
                NUMA_NODE_RELATIONSHIP NumaNode;
                CACHE_RELATIONSHIP     Cache;
                GROUP_RELATIONSHIP     Group;
            } DUMMYUNIONNAME;
        } SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX, *PSYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX;
    '''
    _anonymous_ = 'DUMMYUNIONNAME',
    _fields_ = (
        ('Relationship', INT),  # _LOGICAL_PROCESSOR_RELATIONSHIP is an Enum type, which, At least for GCC, is just a simple numeric type. (https://stackoverflow.com/questions/1546355/using-enums-in-ctypes-structure/1546467#1546467)
        ('Size', DWORD),
        ('DUMMYUNIONNAME', _DUMMYUNIONNAME_SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX)
    )


def _GetLogicalProcessorInformationEx(relation_type):
    '''https://docs.microsoft.com/en-us/windows/win32/api/sysinfoapi/nf-sysinfoapi-getlogicalprocessorinformationex
        BOOL GetLogicalProcessorInformationEx(
            [in]            LOGICAL_PROCESSOR_RELATIONSHIP           RelationshipType,
            [out, optional] PSYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX Buffer,
            [in, out]       PDWORD                                   ReturnedLength
        );
    '''
    dll = WinDLL('kernel32', use_last_error=True)
    # manage the allocated buffer as a c_char_p for easier pointer manipulation
    dll.GetLogicalProcessorInformationEx.argtypes = INT, c_char_p, PDWORD
    dll.GetLogicalProcessorInformationEx.restype = BOOL

    byte_len = DWORD()

    # Call with null buffer to get required buffer size
    result = dll.GetLogicalProcessorInformationEx(relation_type, None, byref(byte_len))
    if (err := get_last_error()) != ERROR_INSUFFICIENT_BUFFER:
        raise WinError(err)

    # Allocate byte buffer
    buffer = create_string_buffer(byte_len.value)

    # Now do the call with the buffer to fill it up
    result = dll.GetLogicalProcessorInformationEx(relation_type, buffer, byref(byte_len))
    if not result:
        raise WinError(get_last_error())

    return buffer, byte_len.value   # return buffer and length


def get_group_affinty_mask_info() -> List[Dict[int, list]]:
    '''
    '''
    # The returned structure array can vary in element size, so walk the buffer by the size of its element.
    buffer, len = _GetLogicalProcessorInformationEx(RELATION_PROCESSOR_PACKAGE)
    offset = 0
    socket_group_infos = []
    while offset < len:
        # cast the current offset to the structure and extract contents
        processor_info = cast(byref(buffer, offset), POINTER(_SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX)).contents

        # print(f"{processor_info.Size=} {processor_info.Processor.GroupCount=}")

        # cast the group mask array to the correct size
        group_affinitys = cast(byref(processor_info.Processor.GroupMask), POINTER(_GROUP_AFFINITY * processor_info.Processor.GroupCount)).contents

        groups = {}
        for group_affinity in group_affinitys:
            # print(f"{group_affinity.Mask=:016X} {group_affinity.Group=}")

            group_cores = []
            core_index = 0
            # Convert the mask to the binary representation of an 8-digit, zero-padded on the left string.
            # Then turn that string of 1's and 0's into a list, and reverse it.
            for i in [int(x) for x in f"{group_affinity.Mask:8b}"][::-1]:
                if i == 1:
                    group_cores.append(core_index)
                core_index  = 1
            groups[group_affinity.Group] = group_cores

        socket_group_infos.append(groups)

        # advance by the size consumed
        offset  = processor_info.Size

    return socket_group_infos

uj5u.com熱心網友回復：

我只有一個帶有 8 個邏輯處理器的內核，所以我無法驗證我的一切都適合你的情況，但試試這個：

from ctypes.wintypes import BYTE, INT, WORD, DWORD, BOOL, PDWORD, UINT, WPARAM
from ctypes import (Structure, Union, WinDLL, c_uint64, POINTER, byref, cast,
                    get_last_error, WinError, c_char_p, create_string_buffer)

KAFFINITY = WPARAM   # WPARAM is 32-bit or 64-bit unsigned depending on platform
ANYSIZE_ARRAY = 1    # actual definition

#
# unchanged portion elided
#

def GetLogicalProcessorInformationEx(relation_type):
    '''https://docs.microsoft.com/en-us/windows/win32/api/sysinfoapi/nf-sysinfoapi-getlogicalprocessorinformationex
        BOOL GetLogicalProcessorInformationEx(
            [in]            LOGICAL_PROCESSOR_RELATIONSHIP           RelationshipType,
            [out, optional] PSYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX Buffer,
            [in, out]       PDWORD                                   ReturnedLength
        );
    '''
    dll = WinDLL('kernel32', use_last_error=True)
    # manage the allocated buffer as a c_char_p for easier pointer manipulation
    dll.GetLogicalProcessorInformationEx.argtypes = INT, c_char_p, PDWORD
    dll.GetLogicalProcessorInformationEx.restype = BOOL

    byte_len = DWORD()

    # Call with null buffer to get required buffer size
    result = dll.GetLogicalProcessorInformationEx(relation_type, None, byref(byte_len))
    if (err := get_last_error()) != ERROR_INSUFFICIENT_BUFFER:
        raise WinError(err)

    # Allocate byte buffer
    buffer = create_string_buffer(byte_len.value)

    # Now do the call with the buffer to fill it up
    result = dll.GetLogicalProcessorInformationEx(relation_type, buffer, byref(byte_len))
    if not result:
        raise WinError(get_last_error())

    return buffer,byte_len.value   # return buffer and length


if __name__ == "__main__":
    # The returned structure array can vary in element size,
    # so walk the buffer by the size of its element.
    buffer,len = GetLogicalProcessorInformationEx(RELATION_PROCESSOR_PACKAGE)
    offset = 0
    while offset < len:
        # cast the current offset to the structure and extract contents
        info = cast(byref(buffer,offset), POINTER(_SYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX)).contents
        print(f'{info.Size=} {info.Processor.GroupCount=}')

        # cast the group mask array to the correct size
        gaffs = cast(byref(info.Processor.GroupMask),POINTER(_GROUP_AFFINITY * info.Processor.GroupCount)).contents
        for gaff in gaffs:
            print(f'{gaff.Mask=:016X} {gaff.Group=}')

        # advance by the size consumed
        offset  = info.Size

在我的一個核心 8 個邏輯 proc 系統上的輸出：

info.Size=48 info.Processor.GroupCount=1
gaff.Mask=00000000000000FF gaff.Group=0

如果這對您的復雜系統不起作用，請發布您的系統回傳buffer,len的我的解決方案回傳的結果，我會修復它。

轉載請註明出處，本文鏈接：https://www.uj5u.com/shujuku/335566.html

標籤：Python 视窗 登录 类型 处理器

上一篇：QueueUserAPC功能不作業，隨機報錯31

下一篇：“系統”的PerlWindows替代品（打開多個行程）

標籤雲
其他(157675) Python(38076) JavaScript(25376) Java(17977) C(15215) 區塊鏈(8255) C＃(7972) AI(7469) 爪哇(7425) MySQL(7132) html(6777) 基礎類(6313) sql(6102) 熊猫(6058) PHP(5869) 数组(5741) R(5409) Linux(5327) 反应(5209) 腳本語言(PerlPython)(5129) 非技術區(4971) Android(4554) 数据框(4311) css(4259) 节点.js(4032) C語言(3288) json(3245) 列表(3129) 扑(3119) C++語言(3117) 安卓(2998) 打字稿(2995) VBA(2789) Java相關(2746) 疑難問題(2699) 细绳(2522) 單片機工控(2479) iOS(2429) ASP.NET(2402) MongoDB(2323) 麻木的(2285) 正则表达式(2254) 字典(2211) 循环(2198) 迅速(2185) 擅长(2169) 镖(2155) 功能(1967) .NET技术(1958) Web開發(1951) python-3.x(1918) HtmlCss(1915) 弹簧靴(1913) C++(1909) xml(1889) PostgreSQL(1872) .NETCore(1853) 谷歌表格(1846) Unity3D(1843) for循环(1842)

熱門瀏覽
  • GPU虛擬機創建時間深度優化

    **?桔妹導讀：**GPU虛擬機實體創建速度慢是公有云面臨的普遍問題，由于通常情況下創建虛擬機屬于低頻操作而未引起業界的重視，實際生產中還是存在對GPU實體創建時間有苛刻要求的業務場景。本文將介紹滴滴云在解決該問題時的思路、方法、并展示最終的優化成果。 從公有云服務商那里購買過虛擬主機的資深用戶，一 ......

    uj5u.com 2020-09-10 06:09:13 more
  • 可編程網卡芯片在滴滴云網路的應用實踐

    **?桔妹導讀：**隨著云規模不斷擴大以及業務層面對延遲、帶寬的要求越來越高，采用DPDK 加速網路報文處理的方式在橫向縱向擴展都出現了局限性。可編程芯片成為業界熱點。本文主要講述了可編程網卡芯片在滴滴云網路中的應用實踐，遇到的問題、帶來的收益以及開源社區貢獻。 #1. 資料中心面臨的問題 隨著滴滴 ......

    uj5u.com 2020-09-10 06:10:21 more
  • 滴滴資料通道服務演進之路

    **?桔妹導讀：**滴滴資料通道引擎承載著全公司的資料同步，為下游實時和離線場景提供了必不可少的源資料。隨著任務量的不斷增加，資料通道的整體架構也隨之發生改變。本文介紹了滴滴資料通道的發展歷程，遇到的問題以及今后的規劃。 #1. 背景 資料，對于任何一家互聯網公司來說都是非常重要的資產，公司的大資料 ......

    uj5u.com 2020-09-10 06:11:05 more
  • 滴滴AI Labs斬獲國際機器翻譯大賽中譯英方向世界第三

    **桔妹導讀：**深耕人工智能領域，致力于探索AI讓出行更美好的滴滴AI Labs再次斬獲國際大獎，這次獲獎的專案是什么呢？一起來看看詳細報道吧！ 近日，由國際計算語言學協會ACL（The Association for Computational Linguistics）舉辦的世界最具影響力的機器 ......

    uj5u.com 2020-09-10 06:11:29 more
  • MPP (Massively Parallel Processing)大規模并行處理

    1、什么是mpp？ MPP (Massively Parallel Processing)，即大規模并行處理，在資料庫非共享集群中，每個節點都有獨立的磁盤存盤系統和記憶體系統，業務資料根據資料庫模型和應用特點劃分到各個節點上，每臺資料節點通過專用網路或者商業通用網路互相連接，彼此協同計算，作為整體提供 ......

    uj5u.com 2020-09-10 06:11:41 more
  • 滴滴資料倉庫指標體系建設實踐

    **桔妹導讀：**指標體系是什么？如何使用OSM模型和AARRR模型搭建指標體系？如何統一流程、規范化、工具化管理指標體系？本文會對建設的方法論結合滴滴資料指標體系建設實踐進行解答分析。 #1. 什么是指標體系 ##1.1 指標體系定義 指標體系是將零散單點的具有相互聯系的指標，系統化的組織起來，通 ......

    uj5u.com 2020-09-10 06:12:52 more
  • 單表千萬行資料庫 LIKE 搜索優化手記

    我們經常在資料庫中使用 LIKE 運算子來完成對資料的模糊搜索，LIKE 運算子用于在 WHERE 子句中搜索列中的指定模式。 如果需要查找客戶表中所有姓氏是“張”的資料，可以使用下面的 SQL 陳述句： SELECT * FROM Customer WHERE Name LIKE '張%' 如果需要 ......

    uj5u.com 2020-09-10 06:13:25 more
  • 滴滴Ceph分布式存盤系統優化之鎖優化

    **桔妹導讀：**Ceph是國際知名的開源分布式存盤系統，在工業界和學術界都有著重要的影響。Ceph的架構和演算法設計發表在國際系統領域頂級會議OSDI、SOSP、SC等上。Ceph社區得到Red Hat、SUSE、Intel等大公司的大力支持。Ceph是國際云計算領域應用最廣泛的開源分布式存盤系統， ......

    uj5u.com 2020-09-10 06:14:51 more
  • es~通過ElasticsearchTemplate進行聚合~嵌套聚合

    之前寫過《es~通過ElasticsearchTemplate進行聚合操作》的文章，這一次主要寫一個嵌套的聚合，例如先對sex集合，再對desc聚合，最后再對age求和，共三層嵌套。 Aggregations的部分特性類似于SQL語言中的group by，avg，sum等函式,Aggregation ......

    uj5u.com 2020-09-10 06:14:59 more
  • 爬蟲日志監控 -- Elastc Stack（ELK）部署

    傻瓜式部署，只需替換IP與用戶 導讀： 現ELK四大組件分別為：Elasticsearch（核心）、logstash（處理）、filebeat（采集）、kibana（可視化） 下載均在https://www.elastic.co/cn/downloads/下tar包，各組件版本最好一致，配合fdm會 ......

    uj5u.com 2020-09-10 06:15:05 more
最新发布
  • day02-2-商鋪查詢快取

    功能02-商鋪查詢快取 3.商鋪詳情快取查詢 3.1什么是快取？ 快取就是資料交換的緩沖區（稱作Cache），是存盤資料的臨時地方，一般讀寫性能較高。 快取的作用： 降低后端負載 提高讀寫效率，降低回應時間 快取的成本： 資料一致性成本 代碼維護成本 運維成本 3.2需求說明 如下，當我們點擊商店詳 ......

    uj5u.com 2023-04-20 08:33:24 more
  • MySQL中binlog備份腳本分享

    關于MySQL的二進制日志（binlog），我們都知道二進制日志（binlog）非常重要，尤其當你需要point to point災難恢復的時侯，所以我們要對其進行備份。關于二進制日志（binlog)的備份,可以基于flush logs方式先切換binlog，然后拷貝&壓縮到到遠程服務器或本地服務器 ......

    uj5u.com 2023-04-20 08:28:06 more
  • day02-短信登錄

    功能實作02 2.功能01-短信登錄 2.1基于Session實作登錄 2.1.1思路分析 2.1.2代碼實作 2.1.2.1發送短信驗證碼 發送短信驗證碼： 發送驗證碼的介面為：http://127.0.0.1:8080/api/user/code?phone=xxxxx<手機號> 請求方式：PO ......

    uj5u.com 2023-04-20 08:27:27 more
  • 快取與資料庫雙寫一致性幾種策略分析

    本文將對幾種快取與資料庫保證資料一致性的使用方式進行分析。為保證高并發性能，以下分析場景不考慮執行的原子性及加鎖等強一致性要求的場景，僅追求最終一致性。 ......

    uj5u.com 2023-04-20 08:26:48 more
  • sql陳述句優化

    問題查找及措施 問題查找 需要找到具體的代碼，對其進行一對一優化，而非一直把關注點放在服務器和sql平臺 降低簡化每個事務中處理的問題，盡量不要讓一個事務拖太長的時間 例如檔案上傳時，應將檔案上傳這一步放在事務外面 微軟建議 4.啟動sql定時執行計劃 怎么啟動sqlserver代理服務-百度經驗 ......

    uj5u.com 2023-04-20 08:26:35 more
  • 云時代，MySQL到ClickHouse資料同步產品對比推薦

    ClickHouse 在執行分析查詢時的速度優勢很好的彌補了MySQL的不足，但是對于很多開發者和DBA來說，如何將MySQL穩定、高效、簡單的同步到 ClickHouse 卻很困難。本文對比了 NineData、MaterializeMySQL（ClickHouse自帶）、Bifrost 三款產品... ......

    uj5u.com 2023-04-20 08:26:29 more
  • sql陳述句優化

    問題查找及措施 問題查找 需要找到具體的代碼，對其進行一對一優化，而非一直把關注點放在服務器和sql平臺 降低簡化每個事務中處理的問題，盡量不要讓一個事務拖太長的時間 例如檔案上傳時，應將檔案上傳這一步放在事務外面 微軟建議 4.啟動sql定時執行計劃 怎么啟動sqlserver代理服務-百度經驗 ......

    uj5u.com 2023-04-20 08:25:13 more
  • Redis 報”OutOfDirectMemoryError“（堆外記憶體溢位)

    Redis 報錯“OutOfDirectMemoryError（堆外記憶體溢位) ”問題如下： 一、報錯資訊： 使用 Redis 的業務介面 ，產生 OutOfDirectMemoryError（堆外記憶體溢位），如圖： 格式化后的報錯資訊： { "timestamp": "2023-04-17 22: ......

    uj5u.com 2023-04-20 08:24:54 more
  • day02-2-商鋪查詢快取

    功能02-商鋪查詢快取 3.商鋪詳情快取查詢 3.1什么是快取？ 快取就是資料交換的緩沖區（稱作Cache），是存盤資料的臨時地方，一般讀寫性能較高。 快取的作用： 降低后端負載 提高讀寫效率，降低回應時間 快取的成本： 資料一致性成本 代碼維護成本 運維成本 3.2需求說明 如下，當我們點擊商店詳 ......

    uj5u.com 2023-04-20 08:24:03 more
  • day02-短信登錄

    功能實作02 2.功能01-短信登錄 2.1基于Session實作登錄 2.1.1思路分析 2.1.2代碼實作 2.1.2.1發送短信驗證碼 發送短信驗證碼： 發送驗證碼的介面為：http://127.0.0.1:8080/api/user/code?phone=xxxxx<手機號> 請求方式：PO ......

    uj5u.com 2023-04-20 08:23:11 more

Copyright © 2010-2020 有解無憂