feat: add linker module

CMakeLists.txt

@@ -230,6 +230,14 @@ nanobind_add_stub(
   DEPENDS llvmpym_ext
 )
 
+nanobind_add_stub(
+  llvmpym_ext_stub_linker
+  MODULE llvmpym_ext.linker
+  OUTPUT linker.pyi
+  PYTHON_PATH $<TARGET_FILE_DIR:llvmpym_ext>
+  DEPENDS llvmpym_ext
+)
+
 
 # Install directive for scikit-build-core
 install(TARGETS llvmpym_ext LIBRARY DESTINATION ${SKBUILD_PROJECT_NAME})
@@ -246,5 +254,6 @@ install(FILES
   ${CMAKE_BINARY_DIR}/target_machine.pyi
   ${CMAKE_BINARY_DIR}/bit_reader.pyi
   ${CMAKE_BINARY_DIR}/disassembler.pyi
-
+  ${CMAKE_BINARY_DIR}/linker.pyi
+
   DESTINATION ${SKBUILD_PROJECT_NAME}/llvmpym_ext)

docs/source/index.rst

@@ -84,7 +84,7 @@ Status
    "``ErrorHandling.h``", "``error_handling``"
    "``ExecutionEngine``", ""
    "``IRReader.h``", "``core``"
-   "``Linker.h``", ""
+   "``Linker.h``", "``linker``"
    "``LLJIT.h``", ""
    "``LLJITUtils.h``", ""
    "``lto.h``", ""

src/llvm/Disassembler.cpp

@@ -15,6 +15,17 @@ void populateDisassembler(nb::module_ &m) {
   // DisasmContextClass
   //     .def("__init__",
   //          [](PymDisasmContext *dc, const char *tripleName, nb::any DisInfo,
-  //          int tagType, LLVMO));
+  //             int tagType,
+  //             std::function<int(void *, uint64_t, uint64_t,
+  //                               uint64_t, uint64_t,
+  //                               int, void *)> getOpInfo,
+  //             std::function<const char*>(void *,
+  //                                        uint64_t,
+  //                                        uint64_t *,
+  //                                        uint64_t,
+  //                                        const char **)
+  //             ) {
+
+  //          });
 }
 

src/llvm/ErrorHandling.cpp

@@ -9,9 +9,6 @@
 namespace nb = nanobind;
 using namespace nb::literals;
 
-namespace nb = nanobind;
-using namespace nb::literals;
-
 void populateErrorHandling(nb::module_ &m) {
   m.def("install_fatal_error_handler",
         [](std::function<void(const char*)> handler) {

src/llvm/Linker.cpp

@@ -0,0 +1,29 @@
+#include "Linker.h"
+
+#include <nanobind/nanobind.h>
+#include <llvm-c/Linker.h>
+#include "types_priv.h"
+
+namespace nb = nanobind;
+using namespace nb::literals;
+
+void populateLinker(nanobind::module_ &m) {
+  nb::enum_<LLVMLinkerMode>(m, "LinkerMode" "LinkerMode")
+      // LLVMLinkerPreserveSource_Removed is deprecated
+      .value("DestroySource", LLVMLinkerMode::LLVMLinkerDestroySource,
+             "This is the default behavior.");
+
+  // TODO customize (if error, throw an runtime error containing the reason. Just
+  // like what llvmlite does)
+  m.def("link_module",
+        [](PymModule dest, PymModule src) {
+          return LLVMLinkModules2(dest.get(), src.get()) != 0;
+        },
+        "dest"_a, "src"_a,
+        "Links the source module into the destination module. The source module is"
+        "destroyed.\n"
+        "The return value is true if an error occurred, false otherwise.\n"
+        "Use the diagnostic handler to get any diagnostic message.");
+}
+
+

src/llvm/Linker.h

@@ -0,0 +1,9 @@
+#ifndef LINKER_H
+#define LINKER_H
+
+#include <nanobind/nanobind.h>
+
+void populateLinker(nanobind::module_ &m);
+
+
+#endif

src/llvmpym/linker.py

@@ -0,0 +1 @@
+from .llvmpym_ext.linker import *

src/llvmpym_ext.cpp

@@ -9,6 +9,7 @@
 #include "llvm/TargetMachine.h"
 #include "llvm/Disassembler.h"
 #include "llvm/BitReader.h"
+#include "llvm/Linker.h"
 
 namespace nb = nanobind;
 using namespace nb::literals;
@@ -43,7 +44,11 @@ NB_MODULE(llvmpym_ext, m) {
   auto bitReaderModule = m.def_submodule("bit_reader", "bit_reader");
   populateBitReader(bitReaderModule);
 
+  // NOTE currently not implemented
   auto disassemblerModule = m.def_submodule("disassembler", "disassembler");
   populateDisassembler(disassemblerModule);
 
+  auto linkerModule = m.def_submodule("linker", "linker");
+  populateLinker(linkerModule);
+
 }