github-personal/Hardware-Sniffer
1
0
Fork 0
mirror of https://github.com/outbackdingo/Hardware-Sniffer.git synced 2026-01-27 10:19:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix GPU identification error with unknown device class

Browse Source
This commit is contained in:
Hoang Hong Quan
2024-11-09 13:04:32 +07:00
parent ffd04dfa63
commit ecad758ee3
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
Scripts/platforms/windows.py
View File

@@ -58,14 +58,14 @@ class WindowsHardwareInfo:
return device_info
def unknown_class_device(self, device_id):
def unknown_class_device(self, device_name, device_id):
if device_id in pci_data.BluetoothIDs:
return "Bluetooth"
elif device_id in pci_data.NetworkIDs:
return "Net"
elif device_id in pci_data.RealtekCardReaderIDs:
return "SDHost"
elif self.classify_gpu(device_id).get("Codename") != "Unknown":
elif self.utils.contains_any(("Video Controller", "Video Adapter", "Graphics Controller"), device_name) and self.classify_gpu(device_id).get("Codename") != "Unknown":
return "Display"
return "Unknown"
@@ -91,6 +91,7 @@ class WindowsHardwareInfo:
}
for device in c.Win32_PnPEntity():
device_name = device.Name or "Unknown"
device_class = device.PNPClass or "Unknown"
pnp_device_id = device.PNPDeviceID or None
@@ -99,7 +100,7 @@ class WindowsHardwareInfo:
if device_info.get("Device ID"):
if device_class in "Unknown":
device_class = self.unknown_class_device(device_info.get("Device ID"))
device_class = self.unknown_class_device(device_name, device_info.get("Device ID"))
elif device_class in "System" and chipset_data.chipset_controllers.get(device_info.get("Device ID")):
self.chipset_model = chipset_data.chipset_controllers.get(device_info.get("Device ID"))