amd-node-scraper 0.0.1__py3-none-any.whl

nodescraper/plugins/inband/cmdline/cmdline_analyzer.py

@@ -0,0 +1,113 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from typing import Optional
+
+from nodescraper.enums import EventCategory, EventPriority, ExecutionStatus
+from nodescraper.interfaces import DataAnalyzer
+from nodescraper.models import TaskResult
+
+from .analyzer_args import CmdlineAnalyzerArgs
+from .cmdlinedata import CmdlineDataModel
+
+
+class CmdlineAnalyzer(DataAnalyzer[CmdlineDataModel, CmdlineAnalyzerArgs]):
+    """Check cmdline matches expected kernel cmdline"""
+
+    DATA_MODEL = CmdlineDataModel
+
+    def _compare_cmdline(self, cmdline: str, required_cmdline: list, banned_cmdline: list) -> bool:
+        """Compare the kernel cmdline against required and banned cmdline arguments.
+
+        Args:
+            cmdline (str): Kernel command line arguments as a string.
+            required_cmdline (list): required kernel cmdline arguments that must be present.
+            banned_cmdline (list): banned kernel cmdline arguments that must not be present.
+
+        Returns:
+            bool: True if the cmdline matches the required arguments and does not contain banned arguments,
+            False otherwise.
+        """
+        # Check for missing required arguments
+        missing_required = [arg for arg in required_cmdline if arg not in cmdline]
+        found_banned = [arg for arg in banned_cmdline if arg in cmdline]
+
+        if len(missing_required) >= 1:
+            self._log_event(
+                category=EventCategory.OS,
+                description=f"Missing {len(missing_required)} required kernel cmdline arguments",
+                priority=EventPriority.ERROR,
+                data={"missing_required": missing_required},
+                console_log=True,
+            )
+
+        if len(found_banned) >= 1:
+            self._log_event(
+                category=EventCategory.OS,
+                description=f"Found {len(found_banned)} banned kernel cmdline arguments",
+                priority=EventPriority.ERROR,
+                data={"found_banned": found_banned},
+                console_log=True,
+            )
+
+        return not (missing_required or found_banned), missing_required, found_banned
+
+    def analyze_data(
+        self, data: CmdlineDataModel, args: Optional[CmdlineAnalyzerArgs] = None
+    ) -> TaskResult:
+        """Analyze the cmdline data against the provided arguments.
+
+        Args:
+            data (CmdlineDataModel): Cmdline data model containing the kernel command line.
+            args (Optional[CmdlineAnalyzerArgs], optional): Cmdline analysis arguments. Defaults to None.
+
+        Returns:
+            TaskResult: Result of the cmdline analysis containing status and message.
+        """
+
+        if not args:
+            self.result.message = "Cmdline analysis args not provided"
+            self.result.status = ExecutionStatus.NOT_RAN
+            return self.result
+
+        # check if any of the cmdline defined in the list match the actual kernel cmdline
+        check, missing_required, found_banned = self._compare_cmdline(
+            data.cmdline, args.required_cmdline, args.banned_cmdline
+        )
+
+        if check:
+            self.result.message = "Kernel cmdline matches expected"
+            self.result.status = ExecutionStatus.OK
+            return self.result
+
+        self.result.message = f"Illegal kernel cmdline, found_banned: {found_banned}, missing required: {missing_required}"
+        self.result.status = ExecutionStatus.ERROR
+        self._log_event(
+            category=EventCategory.OS,
+            description=self.result.message,
+            priority=EventPriority.CRITICAL,
+            console_log=True,
+        )
+        return self.result

nodescraper/plugins/inband/cmdline/cmdline_collector.py

@@ -0,0 +1,77 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from typing import Optional
+
+from nodescraper.base import InBandDataCollector
+from nodescraper.enums import EventCategory, EventPriority, ExecutionStatus, OSFamily
+from nodescraper.models import TaskResult
+
+from .cmdlinedata import CmdlineDataModel
+
+
+class CmdlineCollector(InBandDataCollector[CmdlineDataModel, None]):
+    """Read linux cmdline data"""
+
+    SUPPORTED_OS_FAMILY = {OSFamily.LINUX}
+
+    DATA_MODEL = CmdlineDataModel
+
+    CMD = "cat /proc/cmdline"
+
+    def collect_data(
+        self,
+        args=None,
+    ) -> tuple[TaskResult, Optional[CmdlineDataModel]]:
+        """
+        Collects the cmdline data from the system.
+
+        Returns:
+            tuple[TaskResult, Optional[CmdlineDataModel]]: tuple containing the task result and the cmdline data model if successful, otherwise None.
+        """
+        res = self._run_sut_cmd(self.CMD)
+        cmdline_data = None
+        if res.exit_code == 0:
+            cmdline_data = CmdlineDataModel(cmdline=res.stdout)
+            self._log_event(
+                category="CMDLINE_READ",
+                description="cmdline read",
+                data=cmdline_data.model_dump(),
+                priority=EventPriority.INFO,
+            )
+            self.result.message = f"cmdline: {res.stdout}"
+            self.result.status = ExecutionStatus.OK
+        else:
+            self._log_event(
+                category=EventCategory.OS,
+                description="Error checking cmdline",
+                data={"command": res.command, "exit_code": res.exit_code},
+                priority=EventPriority.ERROR,
+                console_log=True,
+            )
+            self.result.message = "cmdline not found"
+            self.result.status = ExecutionStatus.ERROR
+
+        return self.result, cmdline_data

nodescraper/plugins/inband/cmdline/cmdline_plugin.py

@@ -0,0 +1,43 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from nodescraper.base import InBandDataPlugin
+
+from .analyzer_args import CmdlineAnalyzerArgs
+from .cmdline_analyzer import CmdlineAnalyzer
+from .cmdline_collector import CmdlineCollector
+from .cmdlinedata import CmdlineDataModel
+
+
+class CmdlinePlugin(InBandDataPlugin[CmdlineDataModel, None, CmdlineAnalyzerArgs]):
+    """Plugin for collection and analysis of linux cmdline data"""
+
+    DATA_MODEL = CmdlineDataModel
+
+    COLLECTOR = CmdlineCollector
+
+    ANALYZER = CmdlineAnalyzer
+
+    ANALYZER_ARGS = CmdlineAnalyzerArgs

nodescraper/plugins/inband/cmdline/cmdlinedata.py

@@ -0,0 +1,30 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from nodescraper.models import DataModel
+
+
+class CmdlineDataModel(DataModel):
+    cmdline: str

nodescraper/plugins/inband/device_enumeration/__init__.py

@@ -0,0 +1,29 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from .analyzer_args import DeviceEnumerationAnalyzerArgs
+from .device_enumeration_plugin import DeviceEnumerationPlugin
+
+__all__ = ["DeviceEnumerationPlugin", "DeviceEnumerationAnalyzerArgs"]

nodescraper/plugins/inband/device_enumeration/analyzer_args.py

@@ -0,0 +1,73 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from typing import Any, Optional
+
+from pydantic import field_validator
+
+from nodescraper.models import AnalyzerArgs
+
+from .deviceenumdata import DeviceEnumerationDataModel
+
+
+class DeviceEnumerationAnalyzerArgs(AnalyzerArgs):
+    cpu_count: Optional[list[int]] = None
+    gpu_count: Optional[list[int]] = None
+    vf_count: Optional[list[int]] = None
+
+    @field_validator("cpu_count", "gpu_count", "vf_count", mode="before")
+    @classmethod
+    def normalize_to_list(cls, v: Any) -> Optional[list[int]]:
+        """Convert single integer values to lists for consistent handling.
+
+        Args:
+            v: The input value (can be int, list[int], or None).
+
+        Returns:
+            Optional[list[int]]: The normalized list value or None.
+        """
+        if v is None:
+            return None
+        if isinstance(v, int):
+            return [v]
+        return v
+
+    @classmethod
+    def build_from_model(
+        cls, datamodel: DeviceEnumerationDataModel
+    ) -> "DeviceEnumerationAnalyzerArgs":
+        """build analyzer args from data model
+
+        Args:
+            datamodel (DeviceEnumerationDataModel): data model for plugin
+
+        Returns:
+            DeviceEnumerationAnalyzerArgs: instance of analyzer args class
+        """
+        return cls(
+            cpu_count=[datamodel.cpu_count] if datamodel.cpu_count is not None else None,
+            gpu_count=[datamodel.gpu_count] if datamodel.gpu_count is not None else None,
+            vf_count=[datamodel.vf_count] if datamodel.vf_count is not None else None,
+        )

nodescraper/plugins/inband/device_enumeration/device_enumeration_analyzer.py

@@ -0,0 +1,81 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from typing import Optional
+
+from nodescraper.enums import EventCategory, EventPriority, ExecutionStatus
+from nodescraper.interfaces import DataAnalyzer
+from nodescraper.models import TaskResult
+
+from .analyzer_args import DeviceEnumerationAnalyzerArgs
+from .deviceenumdata import DeviceEnumerationDataModel
+
+
+class DeviceEnumerationAnalyzer(
+    DataAnalyzer[DeviceEnumerationDataModel, DeviceEnumerationAnalyzerArgs]
+):
+    """Check Device Enumeration matches expected cpu and gpu count
+    supported by all OSs, SKUs, and platforms."""
+
+    DATA_MODEL = DeviceEnumerationDataModel
+
+    def analyze_data(
+        self, data: DeviceEnumerationDataModel, args: Optional[DeviceEnumerationAnalyzerArgs] = None
+    ) -> TaskResult:
+
+        if args is None:
+            self.result.status = ExecutionStatus.NOT_RAN
+            self.result.message = (
+                "Expected Device Enumeration data not provided, skipping analysis."
+            )
+            return self.result
+
+        checks = {}
+        if args.cpu_count is not None and args.cpu_count != []:
+            checks["cpu_count"] = args.cpu_count
+        if args.gpu_count is not None and args.gpu_count != []:
+            checks["gpu_count"] = args.gpu_count
+        if args.vf_count is not None and args.vf_count != []:
+            checks["vf_count"] = args.vf_count
+
+        self.result.message = ""
+        for check, accepted_counts in checks.items():
+            actual_count = getattr(data, check)
+            if actual_count not in accepted_counts:
+                message = f"Expected {check} in {accepted_counts}, but got {actual_count}. "
+                self.result.message += message
+                self.result.status = ExecutionStatus.ERROR
+                self._log_event(
+                    category=EventCategory.PLATFORM,
+                    description=message,
+                    data={check: actual_count},
+                    priority=EventPriority.CRITICAL,
+                    console_log=True,
+                )
+        if self.result.message == "":
+            self.result.status = ExecutionStatus.OK
+            self.result.message = f"Device Enumeration validated on {checks.keys()}."
+
+        return self.result

nodescraper/plugins/inband/device_enumeration/device_enumeration_collector.py

@@ -0,0 +1,176 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from typing import Optional
+
+from nodescraper.base import InBandDataCollector
+from nodescraper.connection.inband.inband import CommandArtifact, TextFileArtifact
+from nodescraper.enums import EventCategory, EventPriority, ExecutionStatus, OSFamily
+from nodescraper.models import TaskResult
+
+from .deviceenumdata import DeviceEnumerationDataModel
+
+
+class DeviceEnumerationCollector(InBandDataCollector[DeviceEnumerationDataModel, None]):
+    """Collect CPU and GPU count"""
+
+    DATA_MODEL = DeviceEnumerationDataModel
+
+    CMD_GPU_COUNT_LINUX = "lspci -d {vendorid_ep}: | grep -i 'VGA\\|Display\\|3D' | wc -l"
+    CMD_VF_COUNT_LINUX = "lspci -d {vendorid_ep}: | grep -i 'Virtual Function' | wc -l"
+    CMD_LSCPU_LINUX = "lscpu"
+    CMD_LSHW_LINUX = "lshw"
+
+    CMD_CPU_COUNT_WINDOWS = (
+        'powershell -Command "(Get-WmiObject -Class Win32_Processor | Measure-Object).Count"'
+    )
+    CMD_GPU_COUNT_WINDOWS = 'powershell -Command "(wmic path win32_VideoController get name | findstr AMD | Measure-Object).Count"'
+    CMD_VF_COUNT_WINDOWS = (
+        'powershell -Command "(Get-VMHostPartitionableGpu | Measure-Object).Count"'
+    )
+
+    def _warning(
+        self,
+        description: str,
+        command: CommandArtifact,
+        category: EventCategory = EventCategory.PLATFORM,
+    ):
+        self._log_event(
+            category=category,
+            description=description,
+            data={
+                "command": command.command,
+                "exit_code": command.exit_code,
+                "stderr": command.stderr,
+            },
+            priority=EventPriority.WARNING,
+        )
+
+    def collect_data(self, args=None) -> tuple[TaskResult, Optional[DeviceEnumerationDataModel]]:
+        """
+        Read CPU and GPU count
+        On Linux, use lscpu and lspci
+        On Windows, use WMI and hyper-v cmdlets
+        """
+        if self.system_info.os_family == OSFamily.LINUX:
+            lscpu_res = self._run_sut_cmd(self.CMD_LSCPU_LINUX, log_artifact=False)
+
+            # Count all AMD GPUs
+            vendor_id = format(self.system_info.vendorid_ep, "x")
+            gpu_count_res = self._run_sut_cmd(
+                self.CMD_GPU_COUNT_LINUX.format(vendorid_ep=vendor_id)
+            )
+
+            # Count AMD Virtual Functions
+            vf_count_res = self._run_sut_cmd(self.CMD_VF_COUNT_LINUX.format(vendorid_ep=vendor_id))
+
+            # Collect lshw output
+            lshw_res = self._run_sut_cmd(self.CMD_LSHW_LINUX, sudo=True, log_artifact=False)
+        else:
+            cpu_count_res = self._run_sut_cmd(self.CMD_CPU_COUNT_WINDOWS)
+            gpu_count_res = self._run_sut_cmd(self.CMD_GPU_COUNT_WINDOWS)
+            vf_count_res = self._run_sut_cmd(self.CMD_VF_COUNT_WINDOWS)
+
+        device_enum = DeviceEnumerationDataModel()
+
+        if self.system_info.os_family == OSFamily.LINUX:
+            if lscpu_res.exit_code == 0 and lscpu_res.stdout:
+                # Extract socket count from lscpu output
+                for line in lscpu_res.stdout.splitlines():
+                    if line.startswith("Socket(s):"):
+                        try:
+                            device_enum.cpu_count = int(line.split(":")[1].strip())
+                            break
+                        except (ValueError, IndexError):
+                            self._warning(
+                                description="Cannot parse CPU count from lscpu output",
+                                command=lscpu_res,
+                            )
+                device_enum.lscpu_output = lscpu_res.stdout
+                self._log_event(
+                    category=EventCategory.PLATFORM,
+                    description="Collected lscpu output",
+                    priority=EventPriority.INFO,
+                )
+            else:
+                self._warning(description="Cannot collect lscpu output", command=lscpu_res)
+        else:
+            if cpu_count_res.exit_code == 0:
+                device_enum.cpu_count = int(cpu_count_res.stdout)
+            else:
+                self._warning(description="Cannot determine CPU count", command=cpu_count_res)
+
+        if gpu_count_res.exit_code == 0:
+            device_enum.gpu_count = int(gpu_count_res.stdout)
+        else:
+            self._warning(description="Cannot determine GPU count", command=gpu_count_res)
+
+        if vf_count_res.exit_code == 0:
+            device_enum.vf_count = int(vf_count_res.stdout)
+        else:
+            self._warning(
+                description="Cannot determine VF count",
+                command=vf_count_res,
+                category=EventCategory.SW_DRIVER,
+            )
+
+        # Collect lshw output on Linux
+        if self.system_info.os_family == OSFamily.LINUX:
+            if lshw_res.exit_code == 0 and lshw_res.stdout:
+                device_enum.lshw_output = lshw_res.stdout
+                self.result.artifacts.append(
+                    TextFileArtifact(filename="lshw.txt", contents=lshw_res.stdout)
+                )
+                self._log_event(
+                    category=EventCategory.PLATFORM,
+                    description="Collected lshw output",
+                    priority=EventPriority.INFO,
+                )
+            else:
+                self._warning(description="Cannot collect lshw output", command=lshw_res)
+
+        if device_enum.cpu_count or device_enum.gpu_count or device_enum.vf_count:
+            log_data = device_enum.model_dump(
+                exclude_none=True,
+                exclude={"lscpu_output", "lshw_output", "task_name", "task_type", "parent"},
+            )
+            self._log_event(
+                category=EventCategory.PLATFORM,
+                description=f"Counted {device_enum.cpu_count} CPUs, {device_enum.gpu_count} GPUs, {device_enum.vf_count} VFs",
+                data=log_data,
+                priority=EventPriority.INFO,
+            )
+            self.result.message = f"Device Enumeration: {log_data}"
+            self.result.status = ExecutionStatus.OK
+            return self.result, device_enum
+        else:
+            self.result.message = "Device Enumeration info not found"
+            self.result.status = ExecutionStatus.EXECUTION_FAILURE
+            self._log_event(
+                category=EventCategory.SW_DRIVER,
+                description=self.result.message,
+                priority=EventPriority.CRITICAL,
+            )
+            return self.result, None

nodescraper/plugins/inband/device_enumeration/device_enumeration_plugin.py

@@ -0,0 +1,45 @@
+###############################################################################
+#
+# MIT License
+#
+# Copyright (c) 2025 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+#
+###############################################################################
+from nodescraper.base import InBandDataPlugin
+
+from .analyzer_args import DeviceEnumerationAnalyzerArgs
+from .device_enumeration_analyzer import DeviceEnumerationAnalyzer
+from .device_enumeration_collector import DeviceEnumerationCollector
+from .deviceenumdata import DeviceEnumerationDataModel
+
+
+class DeviceEnumerationPlugin(
+    InBandDataPlugin[DeviceEnumerationDataModel, None, DeviceEnumerationAnalyzerArgs]
+):
+    """Plugin for collection and analysis of BIOS data"""
+
+    DATA_MODEL = DeviceEnumerationDataModel
+
+    COLLECTOR = DeviceEnumerationCollector
+
+    ANALYZER = DeviceEnumerationAnalyzer
+
+    ANALYZER_ARGS = DeviceEnumerationAnalyzerArgs