135 lines
5.0 KiB
C
135 lines
5.0 KiB
C
/*
|
|
* Unit tests for module/DLL/library API
|
|
*
|
|
* Copyright (c) 2004 Eric Pouech
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include "wine/test.h"
|
|
#include <windows.h>
|
|
|
|
static BOOL cmpStrAW(const char* a, const WCHAR* b, DWORD lenA, DWORD lenB)
|
|
{
|
|
WCHAR aw[1024];
|
|
|
|
DWORD len = MultiByteToWideChar( AreFileApisANSI() ? CP_ACP : CP_OEMCP, 0,
|
|
a, lenA, aw, sizeof(aw) / sizeof(aw[0]) );
|
|
if (len != lenB) return FALSE;
|
|
return memcmp(aw, b, len * sizeof(WCHAR)) == 0;
|
|
}
|
|
|
|
static void testGetModuleFileName(const char* name)
|
|
{
|
|
HMODULE hMod;
|
|
char bufA[MAX_PATH];
|
|
WCHAR bufW[MAX_PATH];
|
|
DWORD len1A, len1W, len2A, len2W;
|
|
|
|
hMod = (name) ? GetModuleHandle(name) : NULL;
|
|
|
|
/* first test, with enough space in buffer */
|
|
memset(bufA, '-', sizeof(bufA));
|
|
len1A = GetModuleFileNameA(hMod, bufA, sizeof(bufA));
|
|
ok(len1A > 0, "Getting module filename for handle %p\n", hMod);
|
|
memset(bufW, '-', sizeof(bufW));
|
|
len1W = GetModuleFileNameW(hMod, bufW, sizeof(bufW) / sizeof(WCHAR));
|
|
ok(len1W > 0, "Getting module filename for handle %p\n", hMod);
|
|
ok(len1A == strlen(bufA), "Unexpected length of GetModuleFilenameA (%ld/%d)\n", len1A, strlen(bufA));
|
|
ok(len1W == lstrlenW(bufW), "Unexpected length of GetModuleFilenameW (%ld/%d)\n", len1W, lstrlenW(bufW));
|
|
ok(cmpStrAW(bufA, bufW, len1A, len1W), "Comparing GetModuleFilenameAW results\n");
|
|
|
|
/* second test with a buffer too small */
|
|
memset(bufA, '-', sizeof(bufA));
|
|
len2A = GetModuleFileNameA(hMod, bufA, len1A / 2);
|
|
ok(len2A > 0, "Getting module filename for handle %p\n", hMod);
|
|
memset(bufW, '-', sizeof(bufW));
|
|
len2W = GetModuleFileNameW(hMod, bufW, len1W / 2);
|
|
ok(len2W > 0, "Getting module filename for handle %p\n", hMod);
|
|
ok(cmpStrAW(bufA, bufW, len2A, len2W), "Comparing GetModuleFilenameAW results with buffer too small\n" );
|
|
ok(len1A / 2 == len2A, "Correct length in GetModuleFilenameA with buffer too small (%ld/%ld)\n", len1A / 2, len2A);
|
|
ok(len1W / 2 == len2W, "Correct length in GetModuleFilenameW with buffer too small (%ld/%ld)\n", len1W / 2, len2W);
|
|
}
|
|
|
|
static void testGetModuleFileName_Wrong(void)
|
|
{
|
|
char bufA[MAX_PATH];
|
|
WCHAR bufW[MAX_PATH];
|
|
|
|
/* test wrong handle */
|
|
bufW[0] = '*';
|
|
ok(GetModuleFileNameW((void*)0xffffffff, bufW, sizeof(bufW) / sizeof(WCHAR)) == 0, "Unexpected success in module handle\n");
|
|
ok(bufW[0] == '*', "When failing, buffer shouldn't be written to\n");
|
|
|
|
bufA[0] = '*';
|
|
ok(GetModuleFileNameA((void*)0xffffffff, bufA, sizeof(bufA)) == 0, "Unexpected success in module handle\n");
|
|
ok(bufA[0] == '*', "When failing, buffer shouldn't be written to\n");
|
|
}
|
|
|
|
static void testLoadLibraryA(void)
|
|
{
|
|
HMODULE hModule;
|
|
FARPROC fp;
|
|
|
|
SetLastError(0xdeadbeef);
|
|
hModule = LoadLibraryA("ntdll.dll");
|
|
ok( hModule != NULL, "ntdll.dll should be loadable\n");
|
|
ok( GetLastError() == 0xdeadbeef, "GetLastError should be 0xdeadbeef but is %08lx\n", GetLastError());
|
|
|
|
fp = GetProcAddress(hModule, "LdrLoadDll");
|
|
ok( fp != NULL, "Call should be there\n");
|
|
ok( GetLastError() == 0xdeadbeef, "GetLastError should be 0xdeadbeef but is %08lx\n", GetLastError());
|
|
|
|
FreeLibrary(hModule);
|
|
}
|
|
|
|
static void testLoadLibraryA_Wrong(void)
|
|
{
|
|
HMODULE hModule;
|
|
|
|
/* Try to load a non-existing dll */
|
|
SetLastError(0xdeadbeef);
|
|
hModule = LoadLibraryA("non_ex_pv.dll");
|
|
ok( !hModule, "non_ex_pv.dll should be not loadable\n");
|
|
ok( GetLastError() == ERROR_MOD_NOT_FOUND, "Expected ERROR_MOD_NOT_FOUND, got %08lx\n", GetLastError());
|
|
|
|
/* Just in case */
|
|
FreeLibrary(hModule);
|
|
}
|
|
|
|
static void testGetProcAddress_Wrong(void)
|
|
{
|
|
FARPROC fp;
|
|
|
|
SetLastError(0xdeadbeef);
|
|
fp = GetProcAddress(NULL, "non_ex_call");
|
|
ok( !fp, "non_ex_call should not be found\n");
|
|
ok( GetLastError() == ERROR_PROC_NOT_FOUND, "Expected ERROR_PROC_NOT_FOUND, got %08lx\n", GetLastError());
|
|
fp = GetProcAddress((HMODULE)0xdeadbeef, "non_ex_call");
|
|
ok( !fp, "non_ex_call should not be found\n");
|
|
ok( GetLastError() == ERROR_MOD_NOT_FOUND, "Expected ERROR_MOD_NOT_FOUND, got %08lx\n", GetLastError());
|
|
}
|
|
|
|
START_TEST(module)
|
|
{
|
|
testGetModuleFileName(NULL);
|
|
testGetModuleFileName("kernel32.dll");
|
|
testGetModuleFileName_Wrong();
|
|
|
|
testLoadLibraryA();
|
|
testLoadLibraryA_Wrong();
|
|
testGetProcAddress_Wrong();
|
|
}
|