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

nodescraper/plugins/inband/device_enumeration/deviceenumdata.py

@@ -0,0 +1,36 @@
+###############################################################################
+#
+# 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.models import DataModel
+
+
+class DeviceEnumerationDataModel(DataModel):
+    cpu_count: Optional[int] = None
+    gpu_count: Optional[int] = None
+    vf_count: Optional[int] = None
+    lscpu_output: Optional[str] = None
+    lshw_output: Optional[str] = None

nodescraper/plugins/inband/dimm/__init__.py

@@ -0,0 +1,25 @@
+###############################################################################
+#
+# 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.
+#
+###############################################################################

nodescraper/plugins/inband/dimm/collector_args.py

@@ -0,0 +1,31 @@
+###############################################################################
+#
+# 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 CollectorArgs
+
+
+class DimmCollectorArgs(CollectorArgs):
+    skip_sudo: bool = False

nodescraper/plugins/inband/dimm/dimm_collector.py

@@ -0,0 +1,151 @@
+###############################################################################
+#
+# 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 import TextFileArtifact
+from nodescraper.enums import EventCategory, EventPriority, ExecutionStatus, OSFamily
+from nodescraper.models import TaskResult
+
+from .collector_args import DimmCollectorArgs
+from .dimmdata import DimmDataModel
+
+
+class DimmCollector(InBandDataCollector[DimmDataModel, DimmCollectorArgs]):
+    """Collect data on installed DIMMs"""
+
+    DATA_MODEL = DimmDataModel
+
+    CMD_WINDOWS = "wmic memorychip get Capacity"
+    CMD = """sh -c 'dmidecode -t 17 | tr -s " " | grep -v "Volatile\\|None\\|Module" | grep Size' 2>/dev/null"""
+    CMD_DMIDECODE_FULL = "dmidecode"
+
+    def collect_data(
+        self,
+        args: Optional[DimmCollectorArgs] = None,
+    ) -> tuple[TaskResult, Optional[DimmDataModel]]:
+        """Collect data on installed DIMMs"""
+        if args is None:
+            args = DimmCollectorArgs()
+
+        dimm_str = None
+        if self.system_info.os_family == OSFamily.WINDOWS:
+            res = self._run_sut_cmd(self.CMD_WINDOWS)
+            if res.exit_code == 0:
+                capacities = {}
+                total = 0
+                for line in res.stdout.splitlines():
+                    value = line.strip()
+                    if value.isdigit():
+                        value = int(value)
+                        total += value
+                        if value not in capacities:
+                            capacities[value] = 1
+                        else:
+                            capacities[value] += 1
+                dimm_str = f"{total / 1024 / 1024:.2f}GB @ "
+                for capacity, count in capacities.items():
+                    dimm_str += f"{count} x {capacity / 1024 / 1024:.2f}GB "
+        else:
+            if args.skip_sudo:
+                self.result.message = "Skipping sudo plugin"
+                self.result.status = ExecutionStatus.NOT_RAN
+                return self.result, None
+
+            # Collect full dmidecode output as artifact
+            dmidecode_full_res = self._run_sut_cmd(self.CMD_DMIDECODE_FULL, sudo=True)
+            if dmidecode_full_res.exit_code == 0 and dmidecode_full_res.stdout:
+                self.result.artifacts.append(
+                    TextFileArtifact(filename="dmidecode.txt", contents=dmidecode_full_res.stdout)
+                )
+            else:
+                self._log_event(
+                    category=EventCategory.OS,
+                    description="Could not collect full dmidecode output",
+                    data={
+                        "command": dmidecode_full_res.command,
+                        "exit_code": dmidecode_full_res.exit_code,
+                        "stderr": dmidecode_full_res.stderr,
+                    },
+                    priority=EventPriority.WARNING,
+                )
+
+            res = self._run_sut_cmd(self.CMD, sudo=True)
+            if res.exit_code == 0:
+                total = 0
+                topology = {}
+                size = None
+                for d in res.stdout.splitlines():
+                    split = d.split()
+                    size = split[2]
+                    key = split[1] + split[2]
+                    if not topology.get(key, None):
+                        topology[key] = 1
+                    else:
+                        topology[key] += 1
+                    num_gb = int(split[1])
+                    total += num_gb
+                topology["total"] = total
+                topology["size"] = size
+                total_gb = topology.pop("total")
+                size = topology.pop("size")
+                dimm_str = str(total_gb) + size + " @"
+                for size, count in topology.items():
+                    dimm_str += f" {count} x {size}"
+
+        if res.exit_code != 0:
+            self._log_event(
+                category=EventCategory.OS,
+                description="Error checking dimms",
+                data={
+                    "command": res.command,
+                    "exit_code": res.exit_code,
+                    "stderr": res.stderr,
+                },
+                priority=EventPriority.ERROR,
+                console_log=True,
+            )
+
+        if dimm_str:
+            dimm_data = DimmDataModel(dimms=dimm_str)
+            self._log_event(
+                category=EventCategory.IO,
+                description="Installed DIMM check",
+                data=dimm_data.model_dump(),
+                priority=EventPriority.INFO,
+            )
+            self.result.message = f"DIMM: {dimm_str}"
+        else:
+            dimm_data = None
+            self._log_event(
+                category=EventCategory.IO,
+                description="DIMM info not found",
+                priority=EventPriority.CRITICAL,
+            )
+            self.result.message = "DIMM info not found"
+            self.result.status = ExecutionStatus.ERROR
+
+        return self.result, dimm_data

nodescraper/plugins/inband/dimm/dimm_plugin.py

@@ -0,0 +1,40 @@
+###############################################################################
+#
+# 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 .collector_args import DimmCollectorArgs
+from .dimm_collector import DimmCollector
+from .dimmdata import DimmDataModel
+
+
+class DimmPlugin(InBandDataPlugin[DimmDataModel, DimmCollectorArgs, None]):
+    """Plugin for collection and analysis of DIMM data"""
+
+    DATA_MODEL = DimmDataModel
+
+    COLLECTOR = DimmCollector
+
+    COLLECTOR_ARGS = DimmCollectorArgs

nodescraper/plugins/inband/dimm/dimmdata.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 DimmDataModel(DataModel):
+    dimms: str

nodescraper/plugins/inband/dkms/__init__.py

@@ -0,0 +1,25 @@
+###############################################################################
+#
+# 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.
+#
+###############################################################################

nodescraper/plugins/inband/dkms/analyzer_args.py

@@ -0,0 +1,85 @@
+###############################################################################
+#
+# 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, Union
+
+from pydantic import Field, field_validator
+
+from nodescraper.models import AnalyzerArgs
+from nodescraper.plugins.inband.dkms.dkmsdata import DkmsDataModel
+
+
+class DkmsAnalyzerArgs(AnalyzerArgs):
+    dkms_status: Union[str, list] = Field(default_factory=list)
+    dkms_version: Union[str, list] = Field(default_factory=list)
+    regex_match: bool = False
+
+    def model_post_init(self, __context: Any) -> None:
+        if not self.dkms_status and not self.dkms_version:
+            raise ValueError("At least one of dkms_status or dkms_version must be provided")
+
+    @field_validator("dkms_status", mode="before")
+    @classmethod
+    def validate_dkms_status(cls, dkms_status: Union[str, list]) -> list:
+        """support str or list input for dkms_status
+
+        Args:
+            dkms_status (Union[str, list]): dkms status to check
+
+        Returns:
+            list: list of dkms status
+        """
+        if isinstance(dkms_status, str):
+            dkms_status = [dkms_status]
+
+        return dkms_status
+
+    @field_validator("dkms_version", mode="before")
+    @classmethod
+    def validate_dkms_version(cls, dkms_version: Union[str, list]) -> list:
+        """support str or list input for dkms_version
+
+        Args:
+            dkms_version (Union[str, list]): dkms version to check
+
+        Returns:
+            list: list of dkms version
+        """
+        if isinstance(dkms_version, str):
+            dkms_version = [dkms_version]
+
+        return dkms_version
+
+    @classmethod
+    def build_from_model(cls, datamodel: DkmsDataModel) -> "DkmsAnalyzerArgs":
+        """build analyzer args from data model
+
+        Args:
+            datamodel (DkmsDataModel): data model for plugin
+
+        Returns:
+            DkmsAnalyzerArgs: instance of analyzer args class
+        """
+        return cls(dkms_status=datamodel.status, dkms_version=datamodel.version)

nodescraper/plugins/inband/dkms/dkms_analyzer.py

@@ -0,0 +1,106 @@
+###############################################################################
+#
+# 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.
+#
+###############################################################################
+import re
+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 DkmsAnalyzerArgs
+from .dkmsdata import DkmsDataModel
+
+
+class DkmsAnalyzer(DataAnalyzer[DkmsDataModel, DkmsAnalyzerArgs]):
+    """Check dkms matches expected status and version"""
+
+    DATA_MODEL = DkmsDataModel
+
+    def analyze_data(
+        self, data: DkmsDataModel, args: Optional[DkmsAnalyzerArgs] = None
+    ) -> TaskResult:
+        """
+        Analyze the provided DKMS data against the expected status and version.
+
+        Args:
+            data (DkmsDataModel): Dkms data to analyze.
+            args (Optional[DkmsAnalyzerArgs], optional): Dkms analysis arguments. Defaults to None.
+
+        Returns:
+            TaskResult: The result of the analysis containing status and message.
+        """
+        # Check if the required data is provided
+        if not args:
+            self.result.message = "DKMS analysis args not provided"
+            self.result.status = ExecutionStatus.NOT_RAN
+            return self.result
+
+        # Convert the status and version to lists if they are not already
+        check_map = {
+            "status": args.dkms_status,
+            "version": args.dkms_version,
+        }
+
+        error_state = False
+
+        actual_values = []
+        expected_values = []
+        for check, accepted_values in check_map.items():
+            actual_value = getattr(data, check)
+            for accepted_value in accepted_values:
+                if args.regex_match:
+                    try:
+                        regex_data = re.compile(accepted_value)
+                    except re.error:
+                        self._log_event(
+                            category=EventCategory.RUNTIME,
+                            description=f"DKMS {check} regex is invalid",
+                            data={"regex": accepted_value},
+                            priority=EventPriority.ERROR,
+                        )
+                    if regex_data.match(actual_value):
+                        break
+                elif actual_value == accepted_value:
+                    break
+            else:
+                expected_values.append(accepted_values)
+                actual_values.append(actual_value)
+                self._log_event(
+                    category=EventCategory.SW_DRIVER,
+                    description=f"DKMS {check} has an unexpected value",
+                    data={"expected": accepted_values, "actual": actual_value},
+                    priority=EventPriority.CRITICAL,
+                    console_log=True,
+                )
+                error_state = True
+
+        if error_state:
+            self.result.status = ExecutionStatus.ERROR
+            self.result.message = (
+                f"DKMS data mismatch. \nActual: {actual_values}.\nExpected: {expected_values}"
+            )
+
+        return self.result

nodescraper/plugins/inband/dkms/dkms_collector.py

@@ -0,0 +1,76 @@
+###############################################################################
+#
+# 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 EventPriority, ExecutionStatus, OSFamily
+from nodescraper.models import TaskResult
+
+from .dkmsdata import DkmsDataModel
+
+
+class DkmsCollector(InBandDataCollector[DkmsDataModel, None]):
+    """Collect DKMS status and version data"""
+
+    SUPPORTED_OS_FAMILY = {OSFamily.LINUX}
+
+    DATA_MODEL = DkmsDataModel
+
+    CMD_STATUS = "dkms status"
+    CMD_VERSION = "dkms --version"
+
+    def collect_data(
+        self,
+        args=None,
+    ) -> tuple[TaskResult, Optional[DkmsDataModel]]:
+        """
+        Collect DKMS status and version information.
+
+        Returns:
+            tuple[TaskResult, Optional[DkmsDataModel]]: tuple containing the task result and DKMS data model if available.
+        """
+
+        dkms_data = DkmsDataModel()
+        dkms_status = self._run_sut_cmd(self.CMD_STATUS)
+        dkms_version = self._run_sut_cmd(self.CMD_VERSION)
+
+        if dkms_status.exit_code == 0:
+            dkms_data.status = dkms_status.stdout
+
+        if dkms_version.exit_code == 0:
+            dkms_data.version = dkms_version.stdout
+
+        dkms = bool(dkms_data.model_fields_set)
+        if dkms:
+            self._log_event(
+                category="DKMS_READ",
+                description="DKMS data read",
+                data=dkms_data.model_dump(),
+                priority=EventPriority.INFO,
+            )
+        self.result.message = f"DKMS: {dkms_data.model_dump()}" if dkms else "DKMS info not found"
+        self.result.status = ExecutionStatus.OK if dkms else ExecutionStatus.ERROR
+        return self.result, dkms_data if dkms else None

nodescraper/plugins/inband/dkms/dkms_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 DkmsAnalyzerArgs
+from .dkms_analyzer import DkmsAnalyzer
+from .dkms_collector import DkmsCollector
+from .dkmsdata import DkmsDataModel
+
+
+class DkmsPlugin(InBandDataPlugin[DkmsDataModel, None, DkmsAnalyzerArgs]):
+    """Plugin for collection and analysis of DKMS data"""
+
+    DATA_MODEL = DkmsDataModel
+
+    COLLECTOR = DkmsCollector
+
+    ANALYZER = DkmsAnalyzer
+
+    ANALYZER_ARGS = DkmsAnalyzerArgs