feat: add support for tool(s) that generated the SBOM

Signed-off-by: Paul Horton <[email protected]>

Author: madpah

README.md

@@ -161,7 +161,7 @@ _Note: We refer throughout using XPath, but the same is true for both XML and JS
          <td><code>/bom/metadata</code></td>
          <td>Y</td><td>Y</td><td>N/A</td><td>N/A</td>
          <td>
-            Only <code>timestamp</code> is currently supported 
+            <code>timestamp</code> and <code>tools</code> are currently supported 
          </td>
       </tr>
       <tr>

cyclonedx/model/__init__.py

@@ -15,9 +15,55 @@
 # SPDX-License-Identifier: Apache-2.0
 #
 
+from enum import Enum
+
 """
 Uniform set of models to represent objects within a CycloneDX software bill-of-materials.
 
 You can either create a `cyclonedx.model.bom.Bom` yourself programmatically, or generate a `cyclonedx.model.bom.Bom`
 from a `cyclonedx.parser.BaseParser` implementation.
 """
+
+
+class HashAlgorithm(Enum):
+    """
+    This is out internal representation of the hashAlg simple type within the CycloneDX standard.
+
+    .. note::
+        See the CycloneDX Schema: https://cyclonedx.org/docs/1.3/#type_hashAlg
+    """
+
+    BLAKE2B_256 = 'BLAKE2b-256'
+    BLAKE2B_384 = 'BLAKE2b-384'
+    BLAKE2B_512 = 'BLAKE2b-512'
+    BLAKE3 = 'BLAKE3'
+    MD5 = 'MD5'
+    SHA_1 = 'SHA-1'
+    SHA_256 = 'SHA-256'
+    SHA_384 = 'SHA-384'
+    SHA_512 = 'SHA-512'
+    SHA3_256 = 'SHA3-256'
+    SHA3_384 = 'SHA3-384'
+    SHA3_512 = 'SHA3-512'
+
+
+class HashType:
+    """
+    This is out internal representation of the hashType complex type within the CycloneDX standard.
+
+    .. note::
+        See the CycloneDX Schema for hashType: https://cyclonedx.org/docs/1.3/#type_hashType
+    """
+
+    _algorithm: HashAlgorithm
+    _value: str
+
+    def __init__(self, algorithm: HashAlgorithm, hash_value: str):
+        self._algorithm = algorithm
+        self._value = hash_value
+
+    def get_algorithm(self) -> HashAlgorithm:
+        return self._algorithm
+
+    def get_hash_value(self) -> str:
+        return self._value

cyclonedx/model/bom.py

@@ -18,13 +18,82 @@
 # Copyright (c) OWASP Foundation. All Rights Reserved.
 
 import datetime
+from importlib.metadata import version
 from typing import List
 from uuid import uuid4
 
+from . import HashType
 from .component import Component
 from ..parser import BaseParser
 
 
+class Tool:
+    """
+    This is out internal representation of the toolType complex type within the CycloneDX standard.
+
+    Tool(s) are the things used in the creation of the BOM.
+
+    .. note::
+        See the CycloneDX Schema for toolType: https://cyclonedx.org/docs/1.3/#type_toolType
+    """
+
+    _vendor: str = None
+    _name: str = None
+    _version: str = None
+    _hashes: List[HashType] = []
+
+    def __init__(self, vendor: str, name: str, version: str, hashes: List[HashType] = []):
+        self._vendor = vendor
+        self._name = name
+        self._version = version
+        self._hashes = hashes
+
+    def get_hashes(self) -> List[HashType]:
+        """
+        List of cryptographic hashes that identify this version of this Tool.
+
+        Returns:
+            `List` of `HashType` objects where there are any hashes, else an empty `List`.
+        """
+        return self._hashes
+
+    def get_name(self) -> str:
+        """
+        The name of this Tool.
+
+        Returns:
+            `str` representing the name of the Tool
+        """
+        return self._name
+
+    def get_vendor(self) -> str:
+        """
+        The vendor of this Tool.
+
+        Returns:
+            `str` representing the vendor of the Tool
+        """
+        return self._vendor
+
+    def get_version(self) -> str:
+        """
+        The version of this Tool.
+
+        Returns:
+            `str` representing the version of the Tool
+        """
+        return self._version
+
+    def __repr__(self):
+        return '<Tool {}:{}:{}>'.format(self._vendor, self._name, self._version)
+
+
+try:
+    ThisTool = Tool(vendor='CycloneDX', name='cyclonedx-python-lib', version=version('cyclonedx-python-lib'))
+except Exception:
+    ThisTool = Tool(vendor='CycloneDX', name='cyclonedx-python-lib', version='UNKNOWN')
+
+
 class BomMetaData:
     """
     This is our internal representation of the metadata complex type within the CycloneDX standard.
@@ -34,9 +103,13 @@ class BomMetaData:
     """
 
     _timestamp: datetime.datetime
+    _tools: List[Tool] = []
 
-    def __init__(self):
+    def __init__(self, tools: List[Tool] = []):
         self._timestamp = datetime.datetime.now(tz=datetime.timezone.utc)
+        if len(tools) == 0:
+            tools.append(ThisTool)
+        self._tools = tools
 
     def get_timestamp(self) -> datetime.datetime:
         """
@@ -47,6 +120,15 @@ def get_timestamp(self) -> datetime.datetime:
         """
         return self._timestamp
 
+    def get_tools(self) -> List[Tool]:
+        """
+        Tools used to create this BOM.
+
+        Returns:
+            `List` of `Tool` objects where there are any, else an empty `List`.
+        """
+        return self._tools
+
 
 class Bom:
     """

cyclonedx/output/json.py

@@ -59,11 +59,22 @@ def _get_component_as_dict(self, component: Component) -> dict:
         return c
 
     def _get_metadata_as_dict(self) -> dict:
-        metadata = self.get_bom().get_metadata()
-        return {
-            "timestamp": metadata.get_timestamp().isoformat()
+        bom_metadata = self.get_bom().get_metadata()
+        metadata = {
+            "timestamp": bom_metadata.get_timestamp().isoformat()
         }
 
+        if self.bom_metadata_supports_tools() and len(bom_metadata.get_tools()) > 0:
+            metadata['tools'] = []
+            for tool in bom_metadata.get_tools():
+                metadata['tools'].append({
+                    "vendor": tool.get_vendor(),
+                    "name": tool.get_name(),
+                    "version": tool.get_version()
+                })
+
+        return metadata
+
 
 class JsonV1Dot0(Json, SchemaVersion1Dot0):
     pass

cyclonedx/output/schema.py

@@ -22,6 +22,9 @@
 
 class BaseSchemaVersion(ABC):
 
+    def bom_metadata_supports_tools(self) -> bool:
+        return True
+
     def bom_supports_metadata(self) -> bool:
         return True
 
@@ -49,6 +52,9 @@ def get_schema_version(self) -> str:
 
 class SchemaVersion1Dot1(BaseSchemaVersion):
 
+    def bom_metadata_supports_tools(self) -> bool:
+        return False
+
     def bom_supports_metadata(self) -> bool:
         return False
 
@@ -61,6 +67,9 @@ def get_schema_version(self) -> str:
 
 class SchemaVersion1Dot0(BaseSchemaVersion):
 
+    def bom_metadata_supports_tools(self) -> bool:
+        return False
+
     def bom_supports_metadata(self) -> bool:
         return False
 

cyclonedx/output/xml.py

@@ -187,8 +187,24 @@ def _get_vulnerability_as_xml_element(bom_ref: str, vulnerability: Vulnerability
         return vulnerability_element
 
     def _add_metadata(self, bom: ElementTree.Element) -> ElementTree.Element:
+        bom_metadata = self.get_bom().get_metadata()
+
         metadata_e = ElementTree.SubElement(bom, 'metadata')
-        ElementTree.SubElement(metadata_e, 'timestamp').text = self.get_bom().get_metadata().get_timestamp().isoformat()
+        ElementTree.SubElement(metadata_e, 'timestamp').text = bom_metadata.get_timestamp().isoformat()
+
+        if self.bom_metadata_supports_tools() and len(bom_metadata.get_tools()) > 0:
+            tools_e = ElementTree.SubElement(metadata_e, 'tools')
+            for tool in bom_metadata.get_tools():
+                tool_e = ElementTree.SubElement(tools_e, 'tool')
+                ElementTree.SubElement(tool_e, 'vendor').text = tool.get_vendor()
+                ElementTree.SubElement(tool_e, 'name').text = tool.get_name()
+                ElementTree.SubElement(tool_e, 'version').text = tool.get_version()
+                if len(tool.get_hashes()) > 0:
+                    hashes_e = ElementTree.SubElement(tool_e, 'hashes')
+                    for hash in tool.get_hashes():
+                        ElementTree.SubElement(hashes_e, 'hash',
+                                               {'alg': hash.get_algorithm().value}).text = hash.get_hash_value()
+
         return bom
 
 

tests/base.py

@@ -20,10 +20,13 @@
 import json
 import xml.etree.ElementTree
 from datetime import datetime, timezone
+from importlib.metadata import version
 from unittest import TestCase
 from uuid import uuid4
 from xml.dom import minidom
 
+cyclonedx_lib_name: str = 'cyclonedx-python-lib'
+cyclonedx_lib_version: str = version(cyclonedx_lib_name)
 single_uuid: str = 'urn:uuid:{}'.format(uuid4())
 
 
@@ -50,6 +53,17 @@ def assertEqualJsonBom(self, a: str, b: str):
         ab['metadata']['timestamp'] = now.isoformat()
         bb['metadata']['timestamp'] = now.isoformat()
 
+        # Align 'this' Tool Version
+        if 'tools' in ab['metadata'].keys():
+            for i, tool in enumerate(ab['metadata']['tools']):
+                if tool['name'] == cyclonedx_lib_name:
+                    ab['metadata']['tools'][i]['version'] = cyclonedx_lib_version
+
+        if 'tools' in bb['metadata'].keys():
+            for i, tool in enumerate(bb['metadata']['tools']):
+                if tool['name'] == cyclonedx_lib_name:
+                    bb['metadata']['tools'][i]['version'] = cyclonedx_lib_version
+
         self.assertEqualJson(json.dumps(ab), json.dumps(bb))
 
 
@@ -80,6 +94,14 @@ def assertEqualXmlBom(self, a: str, b: str, namespace: str):
         if metadata_ts_b is not None:
             metadata_ts_b.text = now.isoformat()
 
+        # Align 'this' Tool Version
+        this_tool = ba.find('.//*/{{{}}}tool[{{{}}}version="VERSION"]'.format(namespace, namespace))
+        if this_tool:
+            this_tool.find('./{{{}}}version'.format(namespace)).text = cyclonedx_lib_version
+        this_tool = bb.find('.//*/{{{}}}tool[{{{}}}version="VERSION"]'.format(namespace, namespace))
+        if this_tool:
+            this_tool.find('./{{{}}}version'.format(namespace)).text = cyclonedx_lib_version
+
         self.assertEqualXml(
             xml.etree.ElementTree.tostring(ba, 'unicode'),
             xml.etree.ElementTree.tostring(bb, 'unicode')

tests/fixtures/bom_v1.2_setuptools.json

@@ -4,7 +4,14 @@
   "serialNumber": "urn:uuid:3e671687-395b-41f5-a30f-a58921a69b79",
   "version": 1,
   "metadata": {
-    "timestamp": "2021-09-01T10:50:42.051979+00:00"
+    "timestamp": "2021-09-01T10:50:42.051979+00:00",
+    "tools": [
+      {
+        "vendor": "CycloneDX",
+        "name": "cyclonedx-python-lib",
+        "version": "VERSION"
+      }
+    ]
   },
   "components": [
     {

tests/fixtures/bom_v1.2_setuptools.xml

@@ -2,6 +2,13 @@
 <bom xmlns="http://cyclonedx.org/schema/bom/1.2" version="1">
     <metadata>
         <timestamp>2021-09-01T10:50:42.051979+00:00</timestamp>
+        <tools>
+            <tool>
+                <vendor>CycloneDX</vendor>
+                <name>cyclonedx-python-lib</name>
+                <version>VERSION</version>
+            </tool>
+        </tools>
     </metadata>
     <components>
         <component type="library" bom-ref="pkg:pypi/[email protected]?extension=tar.gz">

tests/fixtures/bom_v1.3_setuptools.json

@@ -4,7 +4,14 @@
   "serialNumber": "urn:uuid:3e671687-395b-41f5-a30f-a58921a69b79",
   "version": 1,
   "metadata": {
-    "timestamp": "2021-09-01T10:50:42.051979+00:00"
+    "timestamp": "2021-09-01T10:50:42.051979+00:00",
+    "tools": [
+      {
+        "vendor": "CycloneDX",
+        "name": "cyclonedx-python-lib",
+        "version": "VERSION"
+      }
+    ]
   },
   "components": [
     {