dlls/kernel/tests/module.c

   1 /*
   2  * Unit tests for module/DLL/library API
   3  *
   4  * Copyright (c) 2004 Eric Pouech
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "wine/test.h"
  22 #include <windows.h>
  23
  24 static BOOL is_unicode_enabled = TRUE;
  25
  26 static BOOL cmpStrAW(const char* a, const WCHAR* b, DWORD lenA, DWORD lenB)
  27 {
  28     WCHAR       aw[1024];
  29
  30     DWORD len = MultiByteToWideChar( AreFileApisANSI() ? CP_ACP : CP_OEMCP, 0,
  31                                      a, lenA, aw, sizeof(aw) / sizeof(aw[0]) );
  32     if (len != lenB) return FALSE;
  33     return memcmp(aw, b, len * sizeof(WCHAR)) == 0;
  34 }
  35
  36 static void testGetModuleFileName(const char* name)
  37 {
  38     HMODULE     hMod;
  39     char        bufA[MAX_PATH];
  40     WCHAR       bufW[MAX_PATH];
  41     DWORD       len1A, len1W = 0, len2A, len2W = 0;
  42
  43     hMod = (name) ? GetModuleHandle(name) : NULL;
  44
  45     /* first test, with enough space in buffer */
  46     memset(bufA, '-', sizeof(bufA));
  47     len1A = GetModuleFileNameA(hMod, bufA, sizeof(bufA));
  48     ok(len1A > 0, "Getting module filename for handle %p\n", hMod);
  49
  50     if (is_unicode_enabled)
  51     {
  52         memset(bufW, '-', sizeof(bufW));
  53         len1W = GetModuleFileNameW(hMod, bufW, sizeof(bufW) / sizeof(WCHAR));
  54         ok(len1W > 0, "Getting module filename for handle %p\n", hMod);
  55     }
  56
  57     ok(len1A == strlen(bufA), "Unexpected length of GetModuleFilenameA (%ld/%d)\n", len1A, lstrlenA(bufA));
  58
  59     if (is_unicode_enabled)
  60     {
  61         ok(len1W == lstrlenW(bufW), "Unexpected length of GetModuleFilenameW (%ld/%d)\n", len1W, lstrlenW(bufW));
  62         ok(cmpStrAW(bufA, bufW, len1A, len1W), "Comparing GetModuleFilenameAW results\n");
  63     }
  64
  65     /* second test with a buffer too small */
  66     memset(bufA, '-', sizeof(bufA));
  67     len2A = GetModuleFileNameA(hMod, bufA, len1A / 2);
  68     ok(len2A > 0, "Getting module filename for handle %p\n", hMod);
  69
  70     if (is_unicode_enabled)
  71     {
  72         memset(bufW, '-', sizeof(bufW));
  73         len2W = GetModuleFileNameW(hMod, bufW, len1W / 2);
  74         ok(len2W > 0, "Getting module filename for handle %p\n", hMod);
  75         ok(cmpStrAW(bufA, bufW, len2A, len2W), "Comparing GetModuleFilenameAW results with buffer too small\n" );
  76         ok(len1W / 2 == len2W, "Correct length in GetModuleFilenameW with buffer too small (%ld/%ld)\n", len1W / 2, len2W);
  77     }
  78
  79     ok(len1A / 2 == len2A ||
  80        len1A / 2 == len2A + 1, /* Win9x */
  81        "Correct length in GetModuleFilenameA with buffer too small (%ld/%ld)\n", len1A / 2, len2A);
  82 }
  83
  84 static void testGetModuleFileName_Wrong(void)
  85 {
  86     char        bufA[MAX_PATH];
  87     WCHAR       bufW[MAX_PATH];
  88
  89     /* test wrong handle */
  90     if (is_unicode_enabled)
  91     {
  92         bufW[0] = '*';
  93         ok(GetModuleFileNameW((void*)0xffffffff, bufW, sizeof(bufW) / sizeof(WCHAR)) == 0, "Unexpected success in module handle\n");
  94         ok(bufW[0] == '*', "When failing, buffer shouldn't be written to\n");
  95     }
  96
  97     bufA[0] = '*';
  98     ok(GetModuleFileNameA((void*)0xffffffff, bufA, sizeof(bufA)) == 0, "Unexpected success in module handle\n");
  99     ok(bufA[0] == '*' ||
 100        bufA[0] == 0 /* Win9x */,
 101        "When failing, buffer shouldn't be written to\n");
 102 }
 103
 104 static void testLoadLibraryA(void)
 105 {
 106     HMODULE hModule, hModule1;
 107     FARPROC fp;
 108
 109     SetLastError(0xdeadbeef);
 110     hModule = LoadLibraryA("kernel32.dll");
 111     ok( hModule != NULL, "kernel32.dll should be loadable\n");
 112     ok( GetLastError() == 0xdeadbeef, "GetLastError should be 0xdeadbeef but is %08lx\n", GetLastError());
 113
 114     fp = GetProcAddress(hModule, "CreateFileA");
 115     ok( fp != NULL, "CreateFileA should be there\n");
 116     ok( GetLastError() == 0xdeadbeef, "GetLastError should be 0xdeadbeef but is %08lx\n", GetLastError());
 117
 118     SetLastError(0xdeadbeef);
 119     hModule1 = LoadLibraryA("kernel32   ");
 120     /* Only winNT does this */
 121     if (GetLastError() != ERROR_DLL_NOT_FOUND)
 122     {
 123         ok( hModule1 != NULL, "\"kernel32   \" should be loadable\n");
 124         ok( GetLastError() == 0xdeadbeef, "GetLastError should be 0xdeadbeef but is %08lx\n", GetLastError());
 125         ok( hModule == hModule1, "Loaded wrong module\n");
 126         FreeLibrary(hModule1);
 127     }
 128     FreeLibrary(hModule);
 129 }
 130
 131 static void testLoadLibraryA_Wrong(void)
 132 {
 133     HMODULE hModule;
 134
 135     /* Try to load a nonexistent dll */
 136     SetLastError(0xdeadbeef);
 137     hModule = LoadLibraryA("non_ex_pv.dll");
 138     ok( !hModule, "non_ex_pv.dll should be not loadable\n");
 139     ok( GetLastError() == ERROR_MOD_NOT_FOUND || GetLastError() == ERROR_DLL_NOT_FOUND,
 140         "Expected ERROR_MOD_NOT_FOUND or ERROR_DLL_NOT_FOUND (win9x), got %08lx\n", GetLastError());
 141
 142     /* Just in case */
 143     FreeLibrary(hModule);
 144 }
 145
 146 static void testGetProcAddress_Wrong(void)
 147 {
 148     FARPROC fp;
 149
 150     SetLastError(0xdeadbeef);
 151     fp = GetProcAddress(NULL, "non_ex_call");
 152     ok( !fp, "non_ex_call should not be found\n");
 153     ok( GetLastError() == ERROR_PROC_NOT_FOUND || GetLastError() == ERROR_INVALID_HANDLE,
 154         "Expected ERROR_PROC_NOT_FOUND or ERROR_INVALID_HANDLE(win9x), got %08lx\n", GetLastError());
 155
 156     SetLastError(0xdeadbeef);
 157     fp = GetProcAddress((HMODULE)0xdeadbeef, "non_ex_call");
 158     ok( !fp, "non_ex_call should not be found\n");
 159     ok( GetLastError() == ERROR_MOD_NOT_FOUND || GetLastError() == ERROR_INVALID_HANDLE,
 160         "Expected ERROR_MOD_NOT_FOUND or ERROR_INVALID_HANDLE(win9x), got %08lx\n", GetLastError());
 161 }
 162
 163 START_TEST(module)
 164 {
 165     WCHAR filenameW[MAX_PATH];
 166
 167     /* Test if we can use GetModuleFileNameW */
 168
 169     SetLastError(0xdeadbeef);
 170     GetModuleFileNameW(NULL, filenameW, MAX_PATH);
 171     if (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
 172     {
 173         trace("GetModuleFileNameW not existing on this platform, skipping W-calls\n");
 174         is_unicode_enabled = FALSE;
 175     }
 176
 177     testGetModuleFileName(NULL);
 178     testGetModuleFileName("kernel32.dll");
 179     testGetModuleFileName_Wrong();
 180
 181     testLoadLibraryA();
 182     testLoadLibraryA_Wrong();
 183     testGetProcAddress_Wrong();
 184 }