87 lines
3.5 KiB
C
87 lines
3.5 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");
|
|
}
|
|
|
|
START_TEST(module)
|
|
{
|
|
testGetModuleFileName(NULL);
|
|
testGetModuleFileName("kernel32.dll");
|
|
testGetModuleFileName_Wrong();
|
|
}
|