From d5ffa9a8cf2f3a6e2e871cc42742006c608113f8 Mon Sep 17 00:00:00 2001 From: Mykola Grymalyuk Date: Mon, 13 Mar 2023 11:42:36 -0600 Subject: [PATCH] device_probe.py: Fix string assumption for IOName --- resources/device_probe.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/resources/device_probe.py b/resources/device_probe.py index d66164f61..7657e2553 100644 --- a/resources/device_probe.py +++ b/resources/device_probe.py @@ -47,8 +47,12 @@ class PCIDevice: vendor_id_unspoofed = None device_id_unspoofed = None - if "IOName" in properties and properties["IOName"].startswith("pci"): - vendor_id_unspoofed, device_id_unspoofed = (int(i, 16) for i in properties["IOName"][3:].split(",")) + if "IOName" in properties: + ioname = properties["IOName"] + 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