mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import os
|
|
import traceback
|
|
from pathlib import Path
|
|
from typing import List
|
|
|
|
from cffi import FFI
|
|
from cffi.backend_ctypes import CTypesBackend
|
|
|
|
# We assume this file is in <root>/lib/python/revng/api
|
|
ROOT = (Path(__file__) / "../../../../..").resolve()
|
|
HEADERS_DIR = ROOT / "include" / "revng" / "PipelineC"
|
|
|
|
header_paths = [
|
|
HEADERS_DIR / "ForwardDeclarationsC.h",
|
|
HEADERS_DIR / "Prototypes.h",
|
|
]
|
|
|
|
ffi = FFI()
|
|
for header_path in header_paths:
|
|
with open(header_path, encoding="utf-8") as header_file:
|
|
lines = []
|
|
for line in header_file:
|
|
if not line.startswith("#"):
|
|
lines.append(line)
|
|
ffi.cdef("\n".join(lines))
|
|
|
|
|
|
@ffi.callback("void ()")
|
|
def pytraceback():
|
|
traceback.print_stack()
|
|
|
|
|
|
LIBRARY_PATH = ROOT / "lib" / "librevngPipelineC.so"
|
|
assert LIBRARY_PATH.is_file()
|
|
|
|
ARGV: List[str] = []
|
|
|
|
REVNG_ANALYSIS_LIBRARIES = os.getenv("REVNG_ANALYSIS_LIBRARIES", "")
|
|
ANALYSIS_LIBRARIES = [Path(f) for f in REVNG_ANALYSIS_LIBRARIES.split(":")]
|
|
|
|
assert all(lib.is_file() for lib in ANALYSIS_LIBRARIES), "Analysis libraries must all be present"
|
|
|
|
|
|
_ANALYSIS_LIBRARIES = [
|
|
ffi.new("char[]", str(s.resolve()).encode("utf-8")) for s in ANALYSIS_LIBRARIES
|
|
]
|
|
|
|
ctypes_backend = CTypesBackend()
|
|
|
|
_api = ffi.dlopen(str(LIBRARY_PATH), flags=ctypes_backend.RTLD_NOW | ctypes_backend.RTLD_GLOBAL)
|
|
_api.rp_set_custom_abort_hook(pytraceback)
|
|
|
|
|
|
def initialize():
|
|
"""Initialize library, must be called exactly once"""
|
|
success = _api.rp_initialize(len(ARGV), ARGV, len(_ANALYSIS_LIBRARIES), _ANALYSIS_LIBRARIES)
|
|
assert success, "Failed revng C API initialization"
|