From afd1b5d2a75ae9ded982bb149d7b9d619f5bd328 Mon Sep 17 00:00:00 2001 From: Mykola Grymalyuk Date: Mon, 13 Mar 2023 11:44:05 -0600 Subject: [PATCH] device_probe.py: Fix missing pci check --- resources/device_probe.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/resources/device_probe.py b/resources/device_probe.py index 7657e2553..5506289af 100644 --- a/resources/device_probe.py +++ b/resources/device_probe.py @@ -52,10 +52,11 @@ class PCIDevice: if type(ioname) is bytes: ioname = ioname.strip(b"\0").decode() - vendor_id_unspoofed, device_id_unspoofed = (int(i, 16) for i in ioname[3:].split(",")) - if anti_spoof: - vendor_id = vendor_id_unspoofed - device_id = device_id_unspoofed + if ioname.startswith("pci"): + vendor_id_unspoofed, device_id_unspoofed = (int(i, 16) for i in ioname[3:].split(",")) + if anti_spoof: + vendor_id = vendor_id_unspoofed + device_id = device_id_unspoofed if vendor_id is None and device_id is None: vendor_id, device_id = [int.from_bytes(properties[i][:4], byteorder="little") for i in ["vendor-id", "device-id"]]