python ctype_Python + ctype 枚举windows设备树(加强版)

from ctypes import *

cfg = windll.cfgmgr32

adv = windll.Advapi32

CRVALS = {

0x00000000:"CR_SUCCESS",

0x00000001:"CR_DEFAULT",

0x00000002:"CR_OUT_OF_MEMORY",

0x00000003:"CR_INVALID_POINTER",

0x00000004:"CR_INVALID_FLAG",

0x00000005:"CR_INVALID_DEVNODE",

0x00000006:"CR_INVALID_RES_DES",

0x00000007:"CR_INVALID_LOG_CONF",

0x00000008:"CR_INVALID_ARBITRATOR",

0x00000009:"CR_INVALID_NODELIST",

0x0000000A:"CR_DEVNODE_HAS_REQS",

0x0000000B:"CR_INVALID_RESOURCEID",

0x0000000C:"CR_DLVXD_NOT_FOUND",

0x0000000D:"CR_NO_SUCH_DEVNODE",

0x0000000E:"CR_NO_MORE_LOG_CONF",

0x0000000F:"CR_NO_MORE_RES_DES",

0x00000010:"CR_ALREADY_SUCH_DEVNODE",

0x00000011:"CR_INVALID_RANGE_LIST",

0x00000012:"CR_INVALID_RANGE",

0x00000013:"CR_FAILURE",

0x00000014:"CR_NO_SUCH_LOGICAL_DEV",

0x00000015:"CR_CREATE_BLOCKED",

0x00000016:"CR_NOT_SYSTEM_VM",

0x00000017:"CR_REMOVE_VETOED",

0x00000018:"CR_APM_VETOED",

0x00000019:"CR_INVALID_LOAD_TYPE",

0x0000001A:"CR_BUFFER_SMALL",

0x0000001B:"CR_NO_ARBITRATOR",

0x0000001C:"CR_NO_REGISTRY_HANDLE",

0x0000001D:"CR_REGISTRY_ERROR",

0x0000001E:"CR_INVALID_DEVICE_ID",

0x0000001F:"CR_INVALID_DATA",

0x00000020:"CR_INVALID_API",

0x00000021:"CR_DEVLOADER_NOT_READY",

0x00000022:"CR_NEED_RESTART",

0x00000023:"CR_NO_MORE_HW_PROFILES",

0x00000024:"CR_DEVICE_NOT_THERE",

0x00000025:"CR_NO_SUCH_VALUE",

0x00000026:"CR_WRONG_TYPE",

0x00000027:"CR_INVALID_PRIORITY",

0x00000028:"CR_NOT_DISABLEABLE",

0x00000029:"CR_FREE_RESOURCES",

0x0000002A:"CR_QUERY_VETOED",

0x0000002B:"CR_CANT_SHARE_IRQ",

0x0000002C:"CR_NO_DEPENDENT",

0x0000002D:"CR_SAME_RESOURCES",

0x0000002E:"CR_NO_SUCH_REGISTRY_KEY",

0x0000002F:"CR_INVALID_MACHINENAME",

0x00000030:"CR_REMOTE_COMM_FAILURE",

0x00000031:"CR_MACHINE_UNAVAILABLE",

0x00000032:"CR_NO_CM_SERVICES",

0x00000033:"CR_ACCESS_DENIED",

0x00000034:"CR_CALL_NOT_IMPLEMENTED",

0x00000035:"CR_INVALID_PROPERTY",

0x00000036:"CR_DEVICE_INTERFACE_ACTIVE",

0x00000037:"CR_NO_SUCH_DEVICE_INTERFACE",

0x00000038:"CR_INVALID_REFERENCE_STRING",

0x00000039:"CR_INVALID_CONFLICT_LIST",

0x0000003A:"CR_INVALID_INDEX",

0x0000003B:"CR_INVALID_STRUCTURE_SIZE",

0x0000003C:"NUM_CR_RESULTS"

}

CM_DRP_DEVICEDESC     =   0x00000001

CM_DRP_HARDWAREID     =   0x00000002

CM_DRP_COMPATIBLEIDS  =   0x00000003

CM_DRP_UNUSED0        =   0x00000004

CM_DRP_SERVICE        =   0x00000005

CM_DRP_CLASS          =   0x00000008

CM_DRP_CLASSGUID      =   0x00000009

CM_DRP_DRIVER         =   0x0000000A

CM_DRP_CONFIGFLAGS    =   0x0000000B

CM_DRP_MFG            =   0x0000000C

CM_DRP_FRIENDLYNAME   =   0x0000000D

NULL = 0

KEY_QUERY_VALUE = 1

RegDisposition_OpenExisting = 1

CM_REGISTRY_HARDWARE = 0

INVALID_HANDLE_VALUE = -1

def get_dev_id(devInst):

buf = (c_wchar*1024)()

blen = c_int(1024)

cr = cfg.CM_Get_Device_IDW(devInst, buf, byref(blen), 0)

if cr == 0:

return buf.value

else:

return ""#"ERR(%d):%s"%(devInst, CRVALS[cr])

def get_dev_prop(devInst, prop, mz = False):

buf = (c_wchar*1024)()

blen = c_int(1024)

cr = cfg.CM_Get_DevNode_Registry_PropertyW(devInst, prop, NULL, buf, byref(blen), 0)

if cr == 0:

if not mz:

return buf.value

else:

s = buf[:]

s = s[:s.find('\x00\x00')]

s = s.replace("\x00", "\n")

return s

else:

return "" #"ERR(%d):%s"%(devInst, CRVALS[cr])

def get_dev_desc(devInst):

return get_dev_prop(devInst, CM_DRP_DEVICEDESC)

def get_dev_driver(devInst):

return get_dev_prop(devInst, CM_DRP_DRIVER)

def get_dev_hwId(devInst):

return get_dev_prop(devInst, CM_DRP_HARDWAREID)

def get_dev_compat_ids(devInst):

return get_dev_prop(devInst, CM_DRP_COMPATIBLEIDS, True)

def get_dev_class(devInst):

return get_dev_prop(devInst, CM_DRP_CLASS)

def get_dev_mfg(devInst):

return get_dev_prop(devInst, CM_DRP_MFG)

def get_dev_name(devInst):

return get_dev_prop(devInst, CM_DRP_FRIENDLYNAME)

def get_dev_com_port(devInst):

hkDevice = c_int()

portBuf = (c_wchar*256)()

portLen = c_int(256)

if (0 == cfg.CM_Open_DevNode_Key(devInst, KEY_QUERY_VALUE, 0, RegDisposition_OpenExisting, byref(hkDevice), CM_REGISTRY_HARDWARE)):

if (hkDevice != INVALID_HANDLE_VALUE):

adv.RegQueryValueExW(hkDevice, "PortName", NULL, NULL, portBuf, byref(portLen))

adv.RegCloseKey(hkDevice);

return portBuf.value

def set_node_text(node, name, value, dom):

textNode = dom.createElement(name)

textNode.appendChild(dom.createTextNode(str(value)))

node.appendChild(textNode)

def get_dev_attr(node, devInst, lev, dom):

desc = get_dev_desc(devInst)

devId = get_dev_id(devInst)

driver = get_dev_driver(devInst)

cls = get_dev_class(devInst)

mfg = get_dev_mfg(devInst)

name = get_dev_name(devInst)

port = get_dev_com_port(devInst)

cids = get_dev_compat_ids(devInst)

node.setAttribute("DevInst", str(devInst))

node.setAttribute("Desc", desc)

node.setAttribute("Lev", str(lev))

infoNode = dom.createElement("DevInfo")

node.appendChild(infoNode)

set_node_text(infoNode, "DevId", devId, dom)

set_node_text(infoNode, "Driver", driver, dom)

set_node_text(infoNode, "Class", cls, dom)

set_node_text(infoNode, "Mfg", mfg, dom)

set_node_text(infoNode, "FriendlyName", name, dom)

set_node_text(infoNode, "ComPort", port, dom)

set_node_text(infoNode, "CompatIds", cids, dom)

from xml.dom.minidom import *

def dev_xml():

def dev_child(devInst, tree, lev, dom):

devParent = c_int(devInst)

devChild = c_int(0)

devNextChild = c_int(0)

if cfg.CM_Get_Child(byref(devChild), devParent, 0) == 0:

node = dom.createElement("Device")

get_dev_attr(node, devChild.value, lev, dom)

tree.appendChild(node)

dev_child(devChild.value, node, lev + 1, dom)

while cfg.CM_Get_Sibling(byref(devNextChild), devChild, 0) == 0:

devChild.value = devNextChild.value

node = dom.createElement("Device")

get_dev_attr(node, devChild.value, lev, dom)

tree.appendChild(node)

dev_child(devChild.value, node, lev + 1, dom)

dom = Document()

dom.appendChild(dom.createElement("DeviceTree"))

devInst = c_int(0)

devInstNext = c_int(0)

lev = 0

if 0 == cfg.CM_Locate_DevNodeW(byref(devInst), 0, 0):

node = dom.createElement("Device")

get_dev_attr(node, devInst.value, lev, dom)

dom.documentElement.appendChild(node)

while 0 == cfg.CM_Get_Sibling(byref(devInstNext), devInst, 0):

devInst.value = devInstNext.value

node = dom.createElement("Device")

get_dev_attr(node, devInst.value, lev, dom)

dom.documentElement.appendChild(node)

for child in dom.documentElement.childNodes:

k = int(child.getAttribute("DevInst"))

dev_child(k, child, lev + 1, dom)

return dom.toprettyxml()

import time

st = time.time()

xml = dev_xml()

et = time.time()

print("use time:", et - st)

open("D:\\DeviceTree.xml", "wb").write(xml.encode("utf8"))

#print(xml)