/**

 * WinPR: Windows Portable Runtime

 * Library Loader

 *

 * Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>

 *

 * 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.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <winpr/crt.h>

#include <winpr/platform.h>

#include <winpr/library.h>

#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 <dlfcn.h>

#include <stdio.h>

#include <stdlib.h>

#include <unistd.h>

#include <sys/types.h>

#include <sys/stat.h>

#ifdef __MACOSX__

#include <mach-o/dyld.h>

#endif

#endif

DLL_DIRECTORY_COOKIE AddDllDirectory(PCWSTR NewDirectory)

{

	/* TODO: Implement */

	WLog_ERR(TAG, "%s not implemented", __FUNCTION__);

	SetLastError(ERROR_CALL_NOT_IMPLEMENTED);

	return NULL;

}

BOOL RemoveDllDirectory(DLL_DIRECTORY_COOKIE Cookie)

{

	/* TODO: Implement */

	WLog_ERR(TAG, "%s not implemented", __FUNCTION__);

	SetLastError(ERROR_CALL_NOT_IMPLEMENTED);

	return FALSE;

}

BOOL SetDefaultDllDirectories(DWORD DirectoryFlags)

{

	/* TODO: Implement */

	WLog_ERR(TAG, "%s not implemented", __FUNCTION__);

	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;

	status = ConvertToUnicode(CP_UTF8, 0, lpLibFileName, -1, &filenameW, 0);

	if (status < 1)

		return NULL;

	hModule = LoadLibraryW(filenameW);

	free(filenameW);

	return hModule;

#else

	HMODULE library;

	library = dlopen(lpLibFileName, RTLD_LOCAL | RTLD_LAZY);

	if (!library)

	{

		WLog_ERR(TAG, "%s failed with %s", __FUNCTION__, dlerror());

		return NULL;

	}

	return library;

#endif

}

HMODULE LoadLibraryW(LPCWSTR lpLibFileName)

{

#if defined(_UWP)

	return LoadPackagedLibrary(lpLibFileName, 0);

#else

	char* name = NULL;

	HMODULE module;

	int rc = ConvertFromUnicode(CP_UTF8, 0, lpLibFileName, -1, &name, 0, NULL, NULL);

	if (rc < 0)

		return NULL;

	module = LoadLibraryA(name);

	free(name);

	return module;

#endif

}

HMODULE LoadLibraryExA(LPCSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)

{

	if (dwFlags != 0)

		WLog_WARN(TAG, "%s does not support dwFlags 0x%08" PRIx32, __FUNCTION__, dwFlags);

	if (hFile)

		WLog_WARN(TAG, "%s does not support hFile != NULL", __FUNCTION__);

	return LoadLibraryA(lpLibFileName);

}

HMODULE LoadLibraryExW(LPCWSTR lpLibFileName, HANDLE hFile, DWORD dwFlags)

{

	if (dwFlags != 0)

		WLog_WARN(TAG, "%s does not support dwFlags 0x%08" PRIx32, __FUNCTION__, dwFlags);

	if (hFile)

		WLog_WARN(TAG, "%s does not support hFile != NULL", __FUNCTION__);

	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, "%s not implemented", __FUNCTION__);

	SetLastError(ERROR_CALL_NOT_IMPLEMENTED);

	return NULL;

}

HMODULE GetModuleHandleW(LPCWSTR lpModuleName)

{

	/* TODO: Implement */

	WLog_ERR(TAG, "%s not implemented", __FUNCTION__);

	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;

	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)

	{

		int rc = ConvertToUnicode(CP_UTF8, 0, name, (int)status, &lpFilename, (int)nSize);

		if (rc < 0)

		{

			free(name);

			SetLastError(ERROR_INTERNAL_ERROR);

			return 0;

		}

	}

	free(name);

	return status;

}

DWORD GetModuleFileNameA(HMODULE hModule, LPSTR lpFilename, DWORD nSize)

{

#if defined(__linux__)

	int 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 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, "%s is not implemented", __FUNCTION__);

	SetLastError(ERROR_CALL_NOT_IMPLEMENTED);

	return 0;

}

#endif