/** * WinPR: Windows Portable Runtime * Library Loader * * Copyright 2012 Marc-Andre Moreau * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include #include #include #include #include "../log.h" #define TAG WINPR_TAG("library") /** * api-ms-win-core-libraryloader-l1-1-1.dll: * * AddDllDirectory * RemoveDllDirectory * SetDefaultDllDirectories * DisableThreadLibraryCalls * EnumResourceLanguagesExA * EnumResourceLanguagesExW * EnumResourceNamesExA * EnumResourceNamesExW * EnumResourceTypesExA * EnumResourceTypesExW * FindResourceExW * FindStringOrdinal * FreeLibrary * FreeLibraryAndExitThread * FreeResource * GetModuleFileNameA * GetModuleFileNameW * GetModuleHandleA * GetModuleHandleExA * GetModuleHandleExW * GetModuleHandleW * GetProcAddress * LoadLibraryExA * LoadLibraryExW * LoadResource * LoadStringA * LoadStringW * LockResource * QueryOptionalDelayLoadedAPI * SizeofResource */ #if !defined(_WIN32) || defined(_UWP) #ifndef _WIN32 #include #include #include #include #include #include #ifdef __MACOSX__ #include #endif #endif DLL_DIRECTORY_COOKIE AddDllDirectory(PCWSTR NewDirectory) { /* TODO: Implement */ WLog_ERR(TAG, "not implemented"); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); return NULL; } BOOL RemoveDllDirectory(DLL_DIRECTORY_COOKIE Cookie) { /* TODO: Implement */ WLog_ERR(TAG, "not implemented"); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); return FALSE; } BOOL SetDefaultDllDirectories(DWORD DirectoryFlags) { /* TODO: Implement */ WLog_ERR(TAG, "not implemented"); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); return FALSE; } HMODULE LoadLibraryA(LPCSTR lpLibFileName) { #if defined(_UWP) int status; HMODULE hModule = NULL; WCHAR* filenameW = NULL; if (!lpLibFileName) return NULL; filenameW = ConvertUtf8ToWCharAlloc(lpLibFileName, NULL); if (filenameW) return NULL; hModule = LoadLibraryW(filenameW); free(filenameW); return hModule; #else HMODULE library; library = dlopen(lpLibFileName, RTLD_LOCAL | RTLD_LAZY); if (!library) { const char* err = dlerror(); WLog_ERR(TAG, "failed with %s", err); return NULL; } return library; #endif } HMODULE LoadLibraryW(LPCWSTR lpLibFileName) { if (!lpLibFileName) return NULL; #if defined(_UWP) return LoadPackagedLibrary(lpLibFileName, 0); #else HMODULE module; char* name = ConvertWCharToUtf8Alloc(lpLibFileName, NULL); if (!name) return NULL; module = LoadLibraryA(name); free(name); return module; #endif } HMODULE LoadLibraryExA(LPCSTR lpLibFileName, HANDLE hFile, DWORD dwFlags) { if (dwFlags != 0) WLog_WARN(TAG, "does not support dwFlags 0x%08" PRIx32, dwFlags); if (hFile) WLog_WARN(TAG, "does not support hFile != NULL"); return LoadLibraryA(lpLibFileName); } HMODULE LoadLibraryExW(LPCWSTR lpLibFileName, HANDLE hFile, DWORD dwFlags) { if (dwFlags != 0) WLog_WARN(TAG, "does not support dwFlags 0x%08" PRIx32, dwFlags); if (hFile) WLog_WARN(TAG, "does not support hFile != NULL"); return LoadLibraryW(lpLibFileName); } #endif #if !defined(_WIN32) && !defined(__CYGWIN__) FARPROC GetProcAddress(HMODULE hModule, LPCSTR lpProcName) { FARPROC proc; proc = dlsym(hModule, lpProcName); if (proc == NULL) { WLog_ERR(TAG, "GetProcAddress: could not find procedure %s: %s", lpProcName, dlerror()); return (FARPROC)NULL; } return proc; } BOOL FreeLibrary(HMODULE hLibModule) { int status; status = dlclose(hLibModule); if (status != 0) return FALSE; return TRUE; } HMODULE GetModuleHandleA(LPCSTR lpModuleName) { /* TODO: Implement */ WLog_ERR(TAG, "not implemented"); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); return NULL; } HMODULE GetModuleHandleW(LPCWSTR lpModuleName) { /* TODO: Implement */ WLog_ERR(TAG, "not implemented"); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); return NULL; } /** * GetModuleFileName: * http://msdn.microsoft.com/en-us/library/windows/desktop/ms683197/ * * Finding current executable's path without /proc/self/exe: * http://stackoverflow.com/questions/1023306/finding-current-executables-path-without-proc-self-exe */ DWORD GetModuleFileNameW(HMODULE hModule, LPWSTR lpFilename, DWORD nSize) { DWORD status; if (!lpFilename) { SetLastError(ERROR_INTERNAL_ERROR); return 0; } char* name = calloc(nSize, sizeof(char)); if (!name) { SetLastError(ERROR_INTERNAL_ERROR); return 0; } status = GetModuleFileNameA(hModule, name, nSize); if ((status > INT_MAX) || (nSize > INT_MAX)) { SetLastError(ERROR_INTERNAL_ERROR); status = 0; } if (status > 0) { if (ConvertUtf8NToWChar(name, status, lpFilename, nSize) < 0) { free(name); SetLastError(ERROR_INTERNAL_ERROR); return 0; } } free(name); return status; } DWORD GetModuleFileNameA(HMODULE hModule, LPSTR lpFilename, DWORD nSize) { #if defined(__linux__) SSIZE_T status; size_t length; char path[64]; if (!hModule) { char buffer[4096]; sprintf_s(path, ARRAYSIZE(path), "/proc/%d/exe", getpid()); status = readlink(path, buffer, sizeof(buffer)); if (status < 0) { SetLastError(ERROR_INTERNAL_ERROR); return 0; } buffer[status] = '\0'; length = strnlen(buffer, sizeof(buffer)); if (length < nSize) { CopyMemory(lpFilename, buffer, length); lpFilename[length] = '\0'; return (DWORD)length; } CopyMemory(lpFilename, buffer, nSize - 1); lpFilename[nSize - 1] = '\0'; SetLastError(ERROR_INSUFFICIENT_BUFFER); return nSize; } #elif defined(__MACOSX__) int status; size_t length; if (!hModule) { char path[4096]; char buffer[4096]; uint32_t size = sizeof(path); status = _NSGetExecutablePath(path, &size); if (status != 0) { /* path too small */ SetLastError(ERROR_INTERNAL_ERROR); return 0; } /* * _NSGetExecutablePath may not return the canonical path, * so use realpath to find the absolute, canonical path. */ realpath(path, buffer); length = strnlen(buffer, sizeof(buffer)); if (length < nSize) { CopyMemory(lpFilename, buffer, length); lpFilename[length] = '\0'; return (DWORD)length; } CopyMemory(lpFilename, buffer, nSize - 1); lpFilename[nSize - 1] = '\0'; SetLastError(ERROR_INSUFFICIENT_BUFFER); return nSize; } #endif WLog_ERR(TAG, "is not implemented"); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); return 0; } #endif HMODULE LoadLibraryX(LPCSTR lpLibFileName) { if (!lpLibFileName) return NULL; #if defined(_WIN32) HMODULE hm = NULL; WCHAR* wstr = ConvertUtf8ToWCharAlloc(lpLibFileName, NULL); if (wstr) hm = LoadLibraryW(wstr); free(wstr); return hm; #else return LoadLibraryA(lpLibFileName); #endif } HMODULE LoadLibraryExX(LPCSTR lpLibFileName, HANDLE hFile, DWORD dwFlags) { if (!lpLibFileName) return NULL; #if defined(_WIN32) HMODULE hm = NULL; WCHAR* wstr = ConvertUtf8ToWCharAlloc(lpLibFileName, NULL); if (wstr) hm = LoadLibraryExW(wstr, hFile, dwFlags); free(wstr); return hm; #else return LoadLibraryExA(lpLibFileName, hFile, dwFlags); #endif }