try using pefile py module to inspect dll

Author: rickeylev

docs/pyproject.toml

@@ -12,5 +12,6 @@ dependencies = [
     "readthedocs-sphinx-ext",
     "absl-py",
     "typing-extensions",
-    "sphinx-reredirects"
+    "sphinx-reredirects",
+    "pefile"
 ]

docs/requirements.txt

@@ -111,9 +111,9 @@ colorama==0.4.6 ; sys_platform == 'win32' \
     --hash=sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44 \
     --hash=sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6
     # via sphinx
-docutils==0.22 \
-    --hash=sha256:4ed966a0e96a0477d852f7af31bdcb3adc049fbb35ccba358c2ea8a03287615e \
-    --hash=sha256:ba9d57750e92331ebe7c08a1bbf7a7f8143b86c476acd51528b042216a6aad0f
+docutils==0.21.2 \
+    --hash=sha256:3a6b18732edf182daa3cd12775bbb338cf5691468f91eeeb109deff6ebfa986f \
+    --hash=sha256:dafca5b9e384f0e419294eb4d2ff9fa826435bf15f15b7bd45723e8ad76811b2
     # via
     #   myst-parser
     #   sphinx
@@ -232,6 +232,10 @@ packaging==25.0 \
     # via
     #   readthedocs-sphinx-ext
     #   sphinx
+pefile==2024.8.26 \
+    --hash=sha256:3ff6c5d8b43e8c37bb6e6dd5085658d658a7a0bdcd20b6a07b1fcfc1c4e9d632 \
+    --hash=sha256:76f8b485dcd3b1bb8166f1128d395fa3d87af26360c2358fb75b80019b957c6f
+    # via rules-python-docs (docs/pyproject.toml)
 pygments==2.19.2 \
     --hash=sha256:636cb2477cec7f8952536970bc533bc43743542f70392ae026374600add5b887 \
     --hash=sha256:86540386c03d588bb81d44bc3928634ff26449851e99741617ecb9037ee5ec0b

tests/cc/current_py_cc_headers/BUILD.bazel

@@ -14,6 +14,7 @@
 
 load("@rules_cc//cc:cc_binary.bzl", "cc_binary")
 load("@rules_shell//shell:sh_test.bzl", "sh_test")
+load("//python:py_test.bzl", "py_test")
 load(":current_py_cc_headers_tests.bzl", "current_py_cc_headers_test_suite")
 
 current_py_cc_headers_test_suite(name = "current_py_cc_headers_tests")
@@ -28,9 +29,20 @@ cc_binary(
     ],
 )
 
-sh_test(
-    name = "abi3_headers_linkage_test",
-    srcs = ["abi3_headers_linkage_test.sh"],
+#sh_test(
+#    name = "abi3_headers_linkage_test",
+#    srcs = ["abi3_headers_linkage_test.sh"],
+#    data = [":bin_abi3"],
+#    deps = ["@bazel_tools//tools/bash/runfiles"],
+#)
+
+py_test(
+    name = "abi3_headers_linkage_test_py",
+    srcs = ["abi3_headers_linkage_test.py"],
     data = [":bin_abi3"],
-    deps = ["@bazel_tools//tools/bash/runfiles"],
+    main = "abi3_headers_linkage_test.py",
+    deps = [
+        "//python/runfiles",
+        "@dev_pip//pefile",
+    ],
 )

tests/cc/current_py_cc_headers/abi3_headers_linkage_test.py

@@ -0,0 +1,41 @@
+
+import os.path
+import sys
+import pefile
+import unittest
+
+class CheckLinkageTest(unittest.TestCase):
+    def test_linkage(self):
+
+        file_path = rf.Rlocation("_main/tests/cc/current_py_cc_headers/libbin_abi3.dll")
+        if not file_path:
+            self.fail("dll not found")
+
+        print(f"[*] Analyzing dependencies for: {os.path.basename(file_path)}\n")
+
+        try:
+            # Parse the PE file
+            pe = pefile.PE(file_path)
+
+            if not hasattr(pe, 'DIRECTORY_ENTRY_IMPORT'):
+                print("[!] No import directory found. The file may not have dependencies or is packed.")
+                raise Exception("no deps?")
+
+            print("Imported DLLs:")
+
+            # Iterate over the import directory entries
+            # Each 'entry' corresponds to one imported DLL
+            for entry in pe.DIRECTORY_ENTRY_IMPORT:
+                # entry.dll is a bytes string, so we decode it to utf-8
+                dll_name = entry.dll.decode('utf-8')
+                print(f"  - {dll_name}")
+
+        except pefile.PEFormatError as e:
+            print(f"Error: Not a valid PE file (DLL/EXE). \nDetails: {e}")
+        except Exception as e:
+            print(f"An unexpected error occurred: {e}")
+
+        raise Exception("done")
+
+if __name__ == "__main__":
+    unittest.main()

tests/cc/current_py_cc_headers/abi3_headers_linkage_test.sh

@@ -20,4 +20,6 @@ lib=$(rlocation _main/tests/cc/current_py_cc_headers/libbin_abi3.so)
 dumpbin /nologo /DEPENDENTS "$lib"
 dumpbin.exe /nologo /DEPENDENTS "$lib"
 
+ls 
+
 exit 1