File size: 1,445 Bytes
76f9669 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | # SPDX-FileCopyrightText: Copyright (c) 2021-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
from libc.stddef cimport wchar_t
from libc.stdint cimport uintptr_t
from cpython cimport PyUnicode_AsWideCharString, PyMem_Free
cdef extern from "windows.h" nogil:
ctypedef void* HMODULE
ctypedef void* HANDLE
ctypedef void* FARPROC
ctypedef unsigned long DWORD
ctypedef const wchar_t *LPCWSTR
ctypedef const char *LPCSTR
ctypedef int BOOL
cdef DWORD LOAD_LIBRARY_SEARCH_SYSTEM32 = 0x00000800
HMODULE _LoadLibraryExW "LoadLibraryExW"(
LPCWSTR lpLibFileName,
HANDLE hFile,
DWORD dwFlags
)
FARPROC _GetProcAddress "GetProcAddress"(HMODULE hModule, LPCSTR lpProcName)
BOOL _FreeLibrary "FreeLibrary"(HMODULE hLibModule)
cdef inline uintptr_t LoadLibraryExW(str path, HANDLE hFile, DWORD dwFlags):
cdef uintptr_t result
cdef wchar_t* wpath = PyUnicode_AsWideCharString(path, NULL)
with nogil:
result = <uintptr_t>_LoadLibraryExW(
wpath,
hFile,
dwFlags
)
PyMem_Free(wpath)
return result
cdef inline FARPROC GetProcAddress(uintptr_t hModule, const char* lpProcName) nogil:
return _GetProcAddress(<HMODULE>hModule, lpProcName)
cdef inline BOOL FreeLibrary(uintptr_t hLibModule) nogil:
return _FreeLibrary(<HMODULE>hLibModule)
|