python+ctypes枚举windows设备为XML树

三里屯摇滚

python+ctypes枚举windows设备为XML树

1. from ctypes import *
   
2. cfg = windll.cfgmgr32
   
3. 
   
4. 
   
5. RERVALS = {
   
6.          0x00000000:"CR_SUCCESS",
   
7.          0x00000001:"CR_DEFAULT",
   
8.          0x00000002:"CR_OUT_OF_MEMORY",
   
9.          0x00000003:"CR_INVALID_POINTER",
   
10.          0x00000004:"CR_INVALID_FLAG",
    
11.          0x00000005:"CR_INVALID_DEVNODE",
    
12.          0x00000006:"CR_INVALID_RES_DES",
    
13.          0x00000007:"CR_INVALID_LOG_CONF",
    
14.          0x00000008:"CR_INVALID_ARBITRATOR",
    
15.          0x00000009:"CR_INVALID_NODELIST",
    
16.          0x0000000A:"CR_DEVNODE_HAS_REQS",
    
17.          0x0000000B:"CR_INVALID_RESOURCEID",
    
18.          0x0000000C:"CR_DLVXD_NOT_FOUND",   
    
19.          0x0000000D:"CR_NO_SUCH_DEVNODE",
    
20.          0x0000000E:"CR_NO_MORE_LOG_CONF",
    
21.          0x0000000F:"CR_NO_MORE_RES_DES",
    
22.          0x00000010:"CR_ALREADY_SUCH_DEVNODE",
    
23.          0x00000011:"CR_INVALID_RANGE_LIST",
    
24.          0x00000012:"CR_INVALID_RANGE",
    
25.          0x00000013:"CR_FAILURE",
    
26.          0x00000014:"CR_NO_SUCH_LOGICAL_DEV",
    
27.          0x00000015:"CR_CREATE_BLOCKED",
    
28.          0x00000016:"CR_NOT_SYSTEM_VM",   
    
29.          0x00000017:"CR_REMOVE_VETOED",
    
30.          0x00000018:"CR_APM_VETOED",
    
31.          0x00000019:"CR_INVALID_LOAD_TYPE",
    
32.          0x0000001A:"CR_BUFFER_SMALL",
    
33.          0x0000001B:"CR_NO_ARBITRATOR",
    
34.          0x0000001C:"CR_NO_REGISTRY_HANDLE",
    
35.          0x0000001D:"CR_REGISTRY_ERROR",
    
36.          0x0000001E:"CR_INVALID_DEVICE_ID",
    
37.          0x0000001F:"CR_INVALID_DATA",
    
38.          0x00000020:"CR_INVALID_API",
    
39.          0x00000021:"CR_DEVLOADER_NOT_READY",
    
40.          0x00000022:"CR_NEED_RESTART",
    
41.          0x00000023:"CR_NO_MORE_HW_PROFILES",
    
42.          0x00000024:"CR_DEVICE_NOT_THERE",
    
43.          0x00000025:"CR_NO_SUCH_VALUE",
    
44.          0x00000026:"CR_WRONG_TYPE",
    
45.          0x00000027:"CR_INVALID_PRIORITY",
    
46.          0x00000028:"CR_NOT_DISABLEABLE",
    
47.          0x00000029:"CR_FREE_RESOURCES",
    
48.          0x0000002A:"CR_QUERY_VETOED",
    
49.          0x0000002B:"CR_CANT_SHARE_IRQ",
    
50.          0x0000002C:"CR_NO_DEPENDENT",
    
51.          0x0000002D:"CR_SAME_RESOURCES",
    
52.          0x0000002E:"CR_NO_SUCH_REGISTRY_KEY",
    
53.          0x0000002F:"CR_INVALID_MACHINENAME",   
    
54.          0x00000030:"CR_REMOTE_COMM_FAILURE",   
    
55.          0x00000031:"CR_MACHINE_UNAVAILABLE",   
    
56.          0x00000032:"CR_NO_CM_SERVICES",   
    
57.          0x00000033:"CR_ACCESS_DENIED",   
    
58.          0x00000034:"CR_CALL_NOT_IMPLEMENTED",
    
59.          0x00000035:"CR_INVALID_PROPERTY",
    
60.          0x00000036:"CR_DEVICE_INTERFACE_ACTIVE",
    
61.          0x00000037:"CR_NO_SUCH_DEVICE_INTERFACE",
    
62.          0x00000038:"CR_INVALID_REFERENCE_STRING",
    
63.          0x00000039:"CR_INVALID_CONFLICT_LIST",
    
64.          0x0000003A:"CR_INVALID_INDEX",
    
65.          0x0000003B:"CR_INVALID_STRUCTURE_SIZE",
    
66.          0x0000003C:"NUM_CR_RESULTS"
    
67.      }
    
68. 
    
69. 
    
70. CM_DRP_DEVICEDESC = 1
    
71. CM_DRP_DRIVER = 0x0000A
    
72. NULL = 0
    
73. 
    
74. 
    
75. def get_dev_desc(devInst):
    
76.      buf = (c_wchar*1024)()
    
77.      blen = c_int(1024)
    
78.      cr = cfg.CM_Get_DevNode_Registry_PropertyW(devInst, CM_DRP_DEVICEDESC, NULL, buf, byref(blen), 0)
    
79.      if cr == 0:
    
80.          return buf.value
    
81.      else:
    
82.          return "ERR(%d):%s"%(devInst, RERVALS[cr])
    
83. 
    
84. 
    
85. def get_dev_id(devInst):
    
86.      buf = (c_wchar*1024)()
    
87.      blen = c_int(1024)
    
88.      cr = cfg.CM_Get_Device_IDW(devInst, buf, byref(blen), 0)
    
89.      if cr == 0:
    
90.          return buf.value
    
91.      else:
    
92.          return "ERR(%d):%s"%(devInst, RERVALS[cr])
    
93. 
    
94. 
    
95. drivers = []
    
96. def get_dev_driver(devInst):
    
97.      global drivers
    
98.      buf = (c_wchar*1024)()
    
99.      blen = c_int(1024)
    
100.      cr = cfg.CM_Get_DevNode_Registry_PropertyW(devInst, CM_DRP_DRIVER, NULL, buf, byref(blen), 0);
     
101.      if cr == 0:
     
102.          drivers.append(buf.value)
     
103.          return buf.value
     
104.      else:
     
105.          return "ERR(%d):%s"%(devInst, RERVALS[cr])
     
106. 
     
107. 
     
108. from xml.dom.minidom import *
     
109. def dev_xml():
     
110.      def dev_child(devInst, tree, lev, dom):
     
111.          devParent = c_int(devInst)
     
112.          devChild = c_int(0)
     
113.          devNextChild = c_int(0)
     
114.          if cfg.CM_Get_Child(byref(devChild), devParent, 0) == 0:
     
115.              desc = get_dev_desc(devChild.value)
     
116.              devId = get_dev_id(devChild.value)
     
117.              driver = get_dev_driver(devChild.value)
     
118.              node = dom.createElement("Device")
     
119.              node.setAttribute("DevInst", str(devChild.value))
     
120.              node.setAttribute("Desc", desc)
     
121.              node.setAttribute("Lev", str(lev))
     
122.              node.setAttribute("DevId", devId)
     
123.              node.setAttribute("Driver", driver)
     
124.              tree.appendChild(node)
     
125.              dev_child(devChild.value, node, lev + 1, dom)
     
126.              while cfg.CM_Get_Sibling(byref(devNextChild), devChild, 0) == 0:
     
127.                  devChild.value = devNextChild.value
     
128.                  desc = get_dev_desc(devChild.value)
     
129.                  devId = get_dev_id(devChild.value)
     
130.                  driver = get_dev_driver(devChild.value)
     
131.                  node = dom.createElement("Device")
     
132.                  node.setAttribute("DevInst", str(devChild.value))
     
133.                  node.setAttribute("Desc", desc)
     
134.                  node.setAttribute("Lev", str(lev))
     
135.                  node.setAttribute("DevId", devId)
     
136.                  node.setAttribute("Driver", driver)
     
137.                  tree.appendChild(node)
     
138.                  dev_child(devChild.value, node, lev + 1, dom)
     
139. 
     
140. 
     
141.     dom = Document()
     
142.      dom.appendChild(dom.createElement("DeviceTree"))
     
143.      devInst = c_int(0)
     
144.      devInstNext = c_int(0)
     
145.      lev = 0
     
146.      if 0 == cfg.CM_Locate_DevNodeW(byref(devInst), 0, 0):
     
147.          desc = get_dev_desc(devInst.value)
     
148.          devId = get_dev_id(devInst.value)
     
149.          driver = get_dev_driver(devInst.value)
     
150.          node = dom.createElement("Device")
     
151.          node.setAttribute("DevInst", str(devInst.value))
     
152.          node.setAttribute("Desc", desc)
     
153.          node.setAttribute("Lev", str(lev))
     
154.          node.setAttribute("DevId", devId)
     
155.          node.setAttribute("Driver", driver)
     
156.          dom.documentElement.appendChild(node)
     
157.          while 0 == cfg.CM_Get_Sibling(byref(devInstNext), devInst, 0):
     
158.              devInst.value = devInstNext.value
     
159.              desc = get_dev_desc(devInst.value)
     
160.              devId = get_dev_id(devInst.value)
     
161.              driver = get_dev_driver(devInst.value)
     
162.              node = dom.createElement("Device")
     
163.              node.setAttribute("DevInst", str(devInst.value))
     
164.              node.setAttribute("Desc", desc)
     
165.              node.setAttribute("Lev", str(lev))
     
166.              node.setAttribute("DevId", devId)
     
167.              node.setAttribute("Driver", driver)
     
168.              dom.documentElement.appendChild(node)
     
169.      for child in dom.documentElement.childNodes:
     
170.          k = int(child.getAttribute("DevInst"))
     
171.          dev_child(k, child, lev + 1, dom)
     
172.      return dom.toprettyxml()
     
173. import time
     
174. st = time.time()
     
175. xml = dev_xml()
     
176. et = time.time()
     
177. print("use time:", et - st)
     
178. open("D:\\DeviceTree.xml", "wb").write(xml.encode("utf8"))
     
179. #print(xml)

复制代码

寂寞冲咖啡

谢谢分享  希望于楼主多多交流