FreeRDP/winpr/libwinpr/library/test/TestLibraryGetProcAddress.c

110 lines
2.3 KiB
C
Raw Normal View History

#include <stdio.h>
#include <winpr/crt.h>
2012-10-05 07:19:05 +04:00
#include <winpr/path.h>
#include <winpr/tchar.h>
#include <winpr/windows.h>
2012-10-05 22:45:54 +04:00
#include <winpr/library.h>
typedef int (*TEST_AB_FN)(int a, int b);
int TestLibraryGetProcAddress(int argc, char* argv[])
{
2012-10-05 07:19:05 +04:00
char* str;
int length;
int a, b, c;
2012-10-05 07:19:05 +04:00
LPTSTR BasePath;
2012-10-18 23:37:00 +04:00
HINSTANCE library;
TEST_AB_FN pFunctionA;
TEST_AB_FN pFunctionB;
2012-10-18 23:37:00 +04:00
LPCTSTR SharedLibraryExtension;
TCHAR LibraryPath[PATHCCH_MAX_CCH];
2012-10-05 07:19:05 +04:00
str = argv[1];
#ifdef UNICODE
length = MultiByteToWideChar(CP_UTF8, 0, str, strlen(str), NULL, 0);
BasePath = (WCHAR*) malloc((length + 1) * sizeof(WCHAR));
if (!BasePath)
{
_tprintf(_T("Memory allocation failed\n"));
return -1;
}
2012-10-05 07:19:05 +04:00
MultiByteToWideChar(CP_UTF8, 0, str, length, (LPWSTR) BasePath, length * sizeof(WCHAR));
BasePath[length] = 0;
#else
2012-10-05 22:45:54 +04:00
BasePath = _strdup(str);
if (!BasePath)
{
printf("Memory allocation failed");
return -1;
}
2012-10-18 23:37:00 +04:00
length = strlen(BasePath);
2012-10-05 07:19:05 +04:00
#endif
2012-10-18 23:37:00 +04:00
CopyMemory(LibraryPath, BasePath, length * sizeof(TCHAR));
LibraryPath[length] = 0;
2012-10-18 23:37:00 +04:00
NativePathCchAppend(LibraryPath, PATHCCH_MAX_CCH, _T("TestLibraryA")); /* subdirectory */
NativePathCchAppend(LibraryPath, PATHCCH_MAX_CCH, _T("TestLibraryA")); /* file name without extension */
2012-10-05 07:19:05 +04:00
2012-10-18 23:37:00 +04:00
SharedLibraryExtension = PathGetSharedLibraryExtension(PATH_SHARED_LIB_EXT_WITH_DOT);
NativePathCchAddExtension(LibraryPath, PATHCCH_MAX_CCH, SharedLibraryExtension); /* add shared library extension */
2012-10-05 07:19:05 +04:00
2012-10-18 23:37:00 +04:00
_tprintf(_T("Loading Library: %s\n"), LibraryPath);
2012-10-05 07:19:05 +04:00
library = LoadLibrary(LibraryPath);
if (!library)
{
2012-10-18 23:37:00 +04:00
_tprintf(_T("LoadLibrary failure\n"));
2012-10-05 07:19:05 +04:00
return -1;
}
pFunctionA = (TEST_AB_FN) GetProcAddress(library, "FunctionA");
if (!pFunctionA)
{
_tprintf(_T("GetProcAddress failure (FunctionA)\n"));
return -1;
}
pFunctionB = (TEST_AB_FN) GetProcAddress(library, "FunctionB");
if (!pFunctionB)
{
_tprintf(_T("GetProcAddress failure (FunctionB)\n"));
return -1;
}
a = 2;
b = 3;
c = pFunctionA(a, b); /* LibraryA / FunctionA multiplies a and b */
if (c != (a * b))
{
_tprintf(_T("pFunctionA call failed\n"));
return -1;
}
a = 10;
b = 5;
c = pFunctionB(a, b); /* LibraryA / FunctionB divides a by b */
if (c != (a / b))
{
_tprintf(_T("pFunctionB call failed\n"));
return -1;
}
2012-10-05 07:19:05 +04:00
if (!FreeLibrary(library))
{
2012-10-18 23:37:00 +04:00
_tprintf(_T("FreeLibrary failure\n"));
2012-10-05 07:19:05 +04:00
return -1;
2012-10-18 23:37:00 +04:00
}
2012-10-05 07:19:05 +04:00
return 0;
}