1 files changed, 52 insertions, 0 deletions
diff --git a/scripts/package-build/linux-kernel/build.py b/scripts/package-build/linux-kernel/build.py
index 1bcab686..1433eaa8 100755
--- a/scripts/package-build/linux-kernel/build.py
+++ b/scripts/package-build/linux-kernel/build.py
@@ -63,6 +63,40 @@ def clone_or_update_repo(repo_dir: Path, scm_url: str, commit_id: str) -> None:
         run(['git', 'checkout', commit_id], cwd=repo_dir, check=True)
 
 
+def create_tarball(package_name, source_dir=None):
+    """Creates a .tar.gz archive of the specified directory.
+
+    Args:
+        package_name (str): The name of the package. This will also be the name of the output tarball.
+        source_dir (str, optional): The directory to be archived. If not provided, defaults to `package_name`.
+
+    Raises:
+        FileNotFoundError: If the specified `source_dir` does not exist.
+        Exception: If an error occurs during tarball creation.
+
+    Example:
+        >>> create_tarball("linux-6.6.56")
+        I: Tarball created: linux-6.6.56.tar.gz
+
+        >>> create_tarball("my-package", "/path/to/source")
+        I: Tarball created: my-package.tar.gz
+    """
+    # Use package_name as the source directory if source_dir is not provided
+    source_dir = source_dir or package_name
+    output_tarball = f"{package_name}.tar.gz"
+
+    # Check if the source directory exists
+    if not os.path.isdir(source_dir):
+        raise FileNotFoundError(f"Directory '{source_dir}' does not exist.")
+
+    # Create the tarball
+    try:
+        shutil.make_archive(base_name=output_tarball.replace('.tar.gz', ''), format='gztar', root_dir=source_dir)
+        print(f"I: Tarball created: {output_tarball}")
+    except Exception as e:
+        print(f"I: Failed to create tarball for {package_name}: {e}")
+
+
 def build_package(package: dict, dependencies: list) -> None:
     """Build a package from the repository
 
@@ -88,20 +122,28 @@ def build_package(package: dict, dependencies: list) -> None:
         # Execute the build command
         if package['build_cmd'] == 'build_kernel':
             build_kernel(package['kernel_version'])
+            create_tarball(f'{package["name"]}-{package["kernel_version"]}', f'linux-{package["kernel_version"]}')
         elif package['build_cmd'] == 'build_linux_firmware':
             build_linux_firmware(package['commit_id'], package['scm_url'])
+            create_tarball(f'{package["name"]}-{package["commit_id"]}', f'{package["name"]}')
         elif package['build_cmd'] == 'build_accel_ppp':
             build_accel_ppp(package['commit_id'], package['scm_url'])
+            create_tarball(f'{package["name"]}-{package["commit_id"]}', f'{package["name"]}')
         elif package['build_cmd'] == 'build_intel_qat':
             build_intel_qat()
         elif package['build_cmd'] == 'build_intel_ixgbe':
             build_intel_ixgbe()
         elif package['build_cmd'] == 'build_intel_ixgbevf':
             build_intel_ixgbevf()
+        elif package['build_cmd'] == 'build_mellanox_ofed':
+            build_mellanox_ofed()
+        elif package['build_cmd'] == 'build_realtek_r8152':
+            build_realtek_r8152()
         elif package['build_cmd'] == 'build_jool':
             build_jool()
         elif package['build_cmd'] == 'build_openvpn_dco':
             build_openvpn_dco(package['commit_id'], package['scm_url'])
+            create_tarball(f'{package["name"]}-{package["commit_id"]}', f'{package["name"]}')
         elif package['build_cmd'] == 'build_nat_rtsp':
             build_nat_rtsp(package['commit_id'], package['scm_url'])
         else:
@@ -183,6 +225,16 @@ def build_intel_ixgbevf():
     run(['./build-intel-ixgbevf.sh'], check=True)
 
 
+def build_mellanox_ofed():
+    """Build Mellanox OFED"""
+    run(['sudo', './build-mellanox-ofed.sh'], check=True)
+
+
+def build_realtek_r8152():
+    """Build Realtek r8152"""
+    run(['sudo', './build-realtek-r8152.py'], check=True)
+
+
 def build_jool():
     """Build Jool"""
     run(['echo y | ./build-jool.py'], check=True, shell=True)