codon/stdlib/internal/dlopen.codon

45 lines
1.0 KiB
Python

@pure
@C
def seq_is_macos() -> bool: pass
# <dlfcn.h>
from C import dlerror() -> cobj as c_dlerror
from C import dlopen(cobj, int) -> cobj as c_dlopen
from C import dlsym(cobj, cobj) -> cobj as c_dlsym
from C import dlclose(cobj) -> i32 as c_dlclose
RTLD_NOW = 2
RTLD_GLOBAL = 8 if seq_is_macos() else 256
RTLD_LOCAL = 0 if seq_is_macos() else 256
def dlext():
if seq_is_macos():
return 'dylib'
else:
return 'so'
@pure
def dlerror():
return str.from_ptr(c_dlerror())
def dlopen(name: str, flag: int = RTLD_NOW | RTLD_GLOBAL) -> cobj:
h = c_dlopen(cobj(0) if name == "" else name.c_str(), flag)
if h == cobj():
raise CError(dlerror())
return h
def dlsym[Fn](lib, name: str) -> Fn:
h = cobj()
if isinstance(lib, str):
h = dlopen(lib)
else:
h = lib
fn = c_dlsym(h, name.c_str())
if fn == cobj():
raise CError(dlerror())
return Fn(fn)
def dlclose(handle: cobj):
if c_dlclose(handle) != i32(0):
raise CError(dlerror())