move export-libs to gardenlinux.build and use pyelftools to retrieve interpreter

Author: Leon-hk

.gitignore

@@ -12,6 +12,7 @@ __pycache__/
 # Distribution / packaging
 .Python
 build/
+!/src/gardenlinux/build
 develop-eggs/
 dist/
 downloads/

poetry.lock

@@ -16,6 +16,7 @@ cryptography = "^46.0.1"
 jsonschema = "^4.25.1"
 networkx = "^3.5"
 oras = "^0.2.38"
+pyelftools = "^0.32"
 pygit2 = "^1.19.0"
 pygments = "^2.19.2"
 PyYAML = "^6.0.2"
@@ -42,7 +43,7 @@ gl-flavors-parse = "gardenlinux.flavors.__main__:main"
 gl-oci = "gardenlinux.oci.__main__:main"
 gl-s3 = "gardenlinux.s3.__main__:main"
 gl-gh = "gardenlinux.github.__main__:main"
-gl-python-exportlibs = "gardenlinux.export_libs.__main__:main"
+gl-build = "gardenlinux.build.__main__:main"
 
 [tool.pytest.ini_options]
 pythonpath = ["src"]

pyproject.toml

@@ -2,15 +2,15 @@
 # -*- coding: utf-8 -*-
 
 """
-gl-python-exportlibs main entrypoint
+gl-build main entrypoint
 """
 
 from argparse import ArgumentParser
 
 from .exporter import _get_default_package_dir, export
 
 _ARGS_TYPE_ALLOWED = [
-    "copy",
+    "export-python-libs",
 ]
 
 
@@ -43,14 +43,18 @@ def parse_args():
 
 def main():
     """
-    gl-python-exportlibs main()
+    gl-gl-build main()
 
     :since: TODO
     """
 
     args = parse_args()
 
-    export(output_dir=args.output_dir, package_dir=args.package_dir)
+    match args.type:
+        case "export-python-libs":
+            export(output_dir=args.output_dir, package_dir=args.package_dir)
+        case _:
+            raise NotImplementedError
 
 
 if __name__ == "__main__":

src/gardenlinux/export_libs/__main__.py src/gardenlinux/build/__main__.pysrc/gardenlinux/export_libs/__main__.py renamed to src/gardenlinux/build/__main__.py

@@ -1,11 +1,14 @@
 # -*- coding: utf-8 -*-
 
 import os
+import pathlib
 import re
 import shutil
 import subprocess
 from os import PathLike
 
+from elftools.elf.elffile import ELFFile
+
 from ..logger import LoggerSetup
 
 # Parses dependencies from ld output
@@ -41,32 +44,27 @@ def _getInterpreter(path: str, logger) -> str:
     """
 
     with open(path, "rb") as f:
-        head = f.read(19)
+        elf = ELFFile(f)
+        interp = elf.get_section_by_name(".interp")
 
-        if head[5] == 1:
-            arch = head[17:]
-        elif head[5] == 2:
-            arch = head[17:][::-1]
-        else:
-            logger.error(
-                f"Error: Unknown endianess value for {path}: expected 1 or 2, but was {head[5]}"
-            )
-            exit(1)
-
-        if arch == b"\x00\xb7":  # 00b7: aarch64
-            return "/lib/ld-linux-aarch64.so.1"
-        elif arch == b"\x00\x3e":  # 003e: x86_64
-            return "/lib64/ld-linux-x86-64.so.2"
-        elif arch == b"\x00\x03":  # 0003: i686
-            return "/lib/ld-linux.so.2"
+        if interp:
+            return interp.data().split(b"\x00")[0].decode()
         else:
-            logger.error(
-                f"Error: Unsupported architecture for {path}: only support x86_64 (003e), aarch64 (00b7) and i686 (0003), but was {arch}"
-            )
-            exit(1)
-
-
-def _get_default_package_dir() -> str:
+            match elf.header["e_machine"]:
+                case "EM_AARCH64":
+                    return "/lib/ld-linux-aarch64.so.1"
+                case "EM_386":
+                    return "/lib/ld-linux.so.2"
+                case "EM_X86_64":
+                    return "/lib64/ld-linux-x86-64.so.2"
+                case arch:
+                    logger.error(
+                        f"Error: Unsupported architecture for {path}: only support x86_64 (003e), aarch64 (00b7) and i686 (0003), but was {arch}"
+                    )
+                    exit(1)
+
+
+def _get_default_package_dir() -> PathLike[str]:
     """
     Finds the default site-packages or dist-packages directory of the default python3 environment
 
@@ -79,12 +77,12 @@ def _get_default_package_dir() -> str:
         ["/bin/sh", "-c", 'python3 -c "import site; print(site.getsitepackages()[0])"'],
         stdout=subprocess.PIPE,
     )
-    return out.stdout.decode().strip()
+    return pathlib.Path(out.stdout.decode().strip())
 
 
 def export(
     output_dir: str | PathLike[str] = "/required_libs",
-    package_dir: str | PathLike[str] = None,
+    package_dir: str | PathLike[str] | None = None,
     logger=None,
 ) -> None:
     """

src/gardenlinux/export_libs/exporter.py src/gardenlinux/build/exporter.pysrc/gardenlinux/export_libs/exporter.py renamed to src/gardenlinux/build/exporter.py

@@ -36,4 +36,6 @@ def main() -> None:
     if args.action == "download-artifacts-from-bucket":
         S3Artifacts(args.bucket).download_to_directory(args.cname, args.path)
     elif args.action == "upload-artifacts-to-bucket":
-        S3Artifacts(args.bucket).upload_from_directory(args.cname, args.path, dry_run=args.dry_run)
+        S3Artifacts(args.bucket).upload_from_directory(
+            args.cname, args.path, dry_run=args.dry_run
+        )