__init__.py 955 B

123456789101112131415161718192021222324252627282930313233343536
  1. import ctypes
  2. import ctypes.util
  3. import sys
  4. # dlfcn.h
  5. _RTLD_DI_LINKMAP = 2
  6. class _LinkMap(ctypes.Structure):
  7. # link.h
  8. _fields_ = [
  9. ('l_addr', ctypes.c_void_p),
  10. ('l_name', ctypes.c_char_p),
  11. ('l_ld', ctypes.c_void_p),
  12. ('l_next', ctypes.c_void_p),
  13. ('l_previous', ctypes.c_void_p),
  14. ]
  15. _LIBDL = ctypes.cdll.LoadLibrary(ctypes.util.find_library('dl'))
  16. _DLINFO = _LIBDL.dlinfo
  17. _DLINFO.argtypes = ctypes.c_void_p, ctypes.c_int, ctypes.c_void_p
  18. _DLINFO.restype = ctypes.c_int
  19. class DLInfo:
  20. def __init__(self, cdll: ctypes.CDLL):
  21. self._linkmap = ctypes.c_void_p()
  22. if _DLINFO(cdll._handle, _RTLD_DI_LINKMAP, ctypes.byref(self._linkmap)) != 0:
  23. raise Exception('dlinfo on {} failed'.format(cdll._name))
  24. @property
  25. def path(self) -> str:
  26. return ctypes.cast(self._linkmap, ctypes.POINTER(_LinkMap)).contents.l_name \
  27. .decode(sys.getdefaultencoding())