From 8ba146d31e7e462d5ffbdef71408b189a311b35a Mon Sep 17 00:00:00 2001 From: Filip Strajnar Date: Sun, 27 Oct 2024 10:18:40 +0100 Subject: [PATCH] Check if iommu_group devices path exists, before listing it. --- src/pci_passthrough_assist/pci_device.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/pci_passthrough_assist/pci_device.py b/src/pci_passthrough_assist/pci_device.py index 234c0d7..1268c47 100644 --- a/src/pci_passthrough_assist/pci_device.py +++ b/src/pci_passthrough_assist/pci_device.py @@ -55,8 +55,11 @@ class PciDevice: driver.write(self.device_id) def devices_in_iommu_group(self) -> list['PciDevice']: - device_ids: list[str] = listdir( - f"/sys/bus/pci/devices/{self.device_id}/iommu_group/devices") + iommu_group_device_path = f"/sys/bus/pci/devices/{self.device_id}/iommu_group/devices" + if not exists(iommu_group_device_path): + print("Device does not have iommu_group devices.") + return [] + device_ids: list[str] = listdir(iommu_group_device_path) return [PciDevice(device_id) for device_id in device_ids] def __str__(self) -> str: