dlls/shlwapi/tests/ordinal.c

   1 /* Unit test suite for SHLWAPI ordinal functions
   2  *
   3  * Copyright 2004 Jon Griffiths
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18  */
  19
  20 #include <stdlib.h>
  21 #include <stdio.h>
  22
  23 #include "wine/test.h"
  24 #include "winbase.h"
  25 #include "winerror.h"
  26 #include "winuser.h"
  27 #define NO_SHLWAPI_REG
  28 #define NO_SHLWAPI_PATH
  29 #define NO_SHLWAPI_GDI
  30 #define NO_SHLWAPI_STREAM
  31 #include "shlwapi.h"
  32
  33 /* Function ptrs for ordinal calls */
  34 static HMODULE hShlwapi;
  35 static int (WINAPI *pSHSearchMapInt)(const int*,const int*,int,int);
  36 static HRESULT (WINAPI *pGetAcceptLanguagesA)(LPSTR,LPDWORD);
  37
  38 static HANDLE (WINAPI *pSHAllocShared)(LPCVOID,DWORD,DWORD);
  39 static LPVOID (WINAPI *pSHLockShared)(HANDLE,DWORD);
  40 static BOOL   (WINAPI *pSHUnlockShared)(LPVOID);
  41 static BOOL   (WINAPI *pSHFreeShared)(HANDLE,DWORD);
  42
  43 static void test_GetAcceptLanguagesA(void)
  44 {   HRESULT retval;
  45     DWORD buffersize, buffersize2, exactsize;
  46     char buffer[100];
  47
  48     if (!pGetAcceptLanguagesA)
  49         return;
  50
  51     buffersize = sizeof(buffer);
  52     memset(buffer, 0, sizeof(buffer));
  53     SetLastError(ERROR_SUCCESS);
  54     retval = pGetAcceptLanguagesA( buffer, &buffersize);
  55     trace("GetAcceptLanguagesA: retval %08lx, size %08lx, buffer (%s),"
  56         " last error %ld\n", retval, buffersize, buffer, GetLastError());
  57     if(retval != S_OK) {
  58         trace("GetAcceptLanguagesA: skipping tests\n");
  59         return;
  60     }
  61     ok( (ERROR_NO_IMPERSONATION_TOKEN == GetLastError()) ||
  62         (ERROR_CLASS_DOES_NOT_EXIST == GetLastError()) ||
  63         (ERROR_PROC_NOT_FOUND == GetLastError()) ||
  64         (ERROR_CALL_NOT_IMPLEMENTED == GetLastError()) ||
  65         (ERROR_SUCCESS == GetLastError()), "last error set to %ld\n", GetLastError());
  66     exactsize = strlen(buffer);
  67
  68     SetLastError(ERROR_SUCCESS);
  69     retval = pGetAcceptLanguagesA( NULL, NULL);
  70     ok(retval == E_FAIL,
  71        "function result wrong: got %08lx; expected E_FAIL\n", retval);
  72     ok(ERROR_SUCCESS == GetLastError(), "last error set to %ld\n", GetLastError());
  73
  74     buffersize = sizeof(buffer);
  75     SetLastError(ERROR_SUCCESS);
  76     retval = pGetAcceptLanguagesA( NULL, &buffersize);
  77     ok(retval == E_FAIL,
  78        "function result wrong: got %08lx; expected E_FAIL\n", retval);
  79     ok(buffersize == sizeof(buffer),
  80        "buffersize was changed (2nd parameter; not on Win2k)\n");
  81     ok(ERROR_SUCCESS == GetLastError(), "last error set to %ld\n", GetLastError());
  82
  83     SetLastError(ERROR_SUCCESS);
  84     retval = pGetAcceptLanguagesA( buffer, NULL);
  85     ok(retval == E_FAIL,
  86        "function result wrong: got %08lx; expected E_FAIL\n", retval);
  87     ok(ERROR_SUCCESS == GetLastError(), "last error set to %ld\n", GetLastError());
  88
  89     buffersize = 0;
  90     memset(buffer, 0, sizeof(buffer));
  91     SetLastError(ERROR_SUCCESS);
  92     retval = pGetAcceptLanguagesA( buffer, &buffersize);
  93     ok(retval == E_FAIL,
  94        "function result wrong: got %08lx; expected E_FAIL\n", retval);
  95     ok(buffersize == 0,
  96        "buffersize wrong(changed) got %08lx; expected 0 (2nd parameter; not on Win2k)\n", buffersize);
  97     ok(ERROR_SUCCESS == GetLastError(), "last error set to %ld\n", GetLastError());
  98
  99     buffersize = buffersize2 = 1;
 100     memset(buffer, 0, sizeof(buffer));
 101     SetLastError(ERROR_SUCCESS);
 102     retval = pGetAcceptLanguagesA( buffer, &buffersize);
 103     switch(retval) {
 104         case 0L:
 105             if(buffersize == exactsize) {
 106             ok( (ERROR_SUCCESS == GetLastError()) || (ERROR_CALL_NOT_IMPLEMENTED == GetLastError()) ||
 107                 (ERROR_PROC_NOT_FOUND == GetLastError()) || (ERROR_NO_IMPERSONATION_TOKEN == GetLastError()),
 108                 "last error wrong: got %08lx; expected ERROR_SUCCESS(NT4)/ERROR_CALL_NOT_IMPLEMENTED(98/ME)/"
 109                 "ERROR_PROC_NOT_FOUND(NT4)/ERROR_NO_IMPERSONATION_TOKEN(XP)\n", GetLastError());
 110             ok(exactsize == strlen(buffer),
 111                  "buffer content (length) wrong: got %08x, expected %08lx \n", strlen(buffer), exactsize);
 112             } else if((buffersize +1) == buffersize2) {
 113                 ok(ERROR_SUCCESS == GetLastError(),
 114                     "last error wrong: got %08lx; expected ERROR_SUCCESS\n", GetLastError());
 115                 ok(buffersize == strlen(buffer),
 116                     "buffer content (length) wrong: got %08x, expected %08lx \n", strlen(buffer), buffersize);
 117             } else
 118                 ok( 0, "retval %08lx, size %08lx, buffer (%s), last error %ld\n",
 119                     retval, buffersize, buffer, GetLastError());
 120             break;
 121         case E_INVALIDARG:
 122             ok(buffersize == 0,
 123                "buffersize wrong: got %08lx, expected 0 (2nd parameter;Win2k)\n", buffersize);
 124             ok(ERROR_INSUFFICIENT_BUFFER == GetLastError(),
 125                "last error wrong: got %08lx; expected ERROR_INSUFFICIENT_BUFFER\n", GetLastError());
 126             ok(buffersize2 == strlen(buffer),
 127                "buffer content (length) wrong: got %08x, expected %08lx \n", strlen(buffer), buffersize2);
 128             break;
 129         default:
 130             ok( 0, "retval %08lx, size %08lx, buffer (%s), last error %ld\n",
 131                 retval, buffersize, buffer, GetLastError());
 132             break;
 133     }
 134
 135     buffersize = buffersize2 = exactsize;
 136     memset(buffer, 0, sizeof(buffer));
 137     SetLastError(ERROR_SUCCESS);
 138     retval = pGetAcceptLanguagesA( buffer, &buffersize);
 139     switch(retval) {
 140         case 0L:
 141             ok(ERROR_SUCCESS == GetLastError(),
 142                  "last error wrong: got %08lx; expected ERROR_SUCCESS\n", GetLastError());
 143             if((buffersize == exactsize) /* XP */ ||
 144                ((buffersize +1)== exactsize) /* 98 */)
 145                 ok(buffersize == strlen(buffer),
 146                     "buffer content (length) wrong: got %08x, expected %08lx \n", strlen(buffer), buffersize);
 147             else
 148                 ok( 0, "retval %08lx, size %08lx, buffer (%s), last error %ld\n",
 149                     retval, buffersize, buffer, GetLastError());
 150             break;
 151         case E_INVALIDARG:
 152             ok(buffersize == 0,
 153                "buffersize wrong: got %08lx, expected 0 (2nd parameter;Win2k)\n", buffersize);
 154             ok(ERROR_INSUFFICIENT_BUFFER == GetLastError(),
 155                "last error wrong: got %08lx; expected ERROR_INSUFFICIENT_BUFFER\n", GetLastError());
 156             ok(buffersize2 == strlen(buffer),
 157                "buffer content (length) wrong: got %08x, expected %08lx \n", strlen(buffer), buffersize2);
 158             break;
 159         default:
 160             ok( 0, "retval %08lx, size %08lx, buffer (%s), last error %ld\n",
 161                 retval, buffersize, buffer, GetLastError());
 162             break;
 163     }
 164 }
 165
 166 static void test_SHSearchMapInt(void)
 167 {
 168   int keys[8], values[8];
 169   int i = 0;
 170
 171   if (!pSHSearchMapInt)
 172     return;
 173
 174   memset(keys, 0, sizeof(keys));
 175   memset(values, 0, sizeof(values));
 176   keys[0] = 99; values[0] = 101;
 177
 178   /* NULL key/value lists crash native, so skip testing them */
 179
 180   /* 1 element */
 181   i = pSHSearchMapInt(keys, values, 1, keys[0]);
 182   ok(i == values[0], "Len 1, expected %d, got %d\n", values[0], i);
 183
 184   /* Key doesn't exist */
 185   i = pSHSearchMapInt(keys, values, 1, 100);
 186   ok(i == -1, "Len 1 - bad key, expected -1, got %d\n", i);
 187
 188   /* Len = 0 => not found */
 189   i = pSHSearchMapInt(keys, values, 0, keys[0]);
 190   ok(i == -1, "Len 1 - passed len 0, expected -1, got %d\n", i);
 191
 192   /* 2 elements, len = 1 */
 193   keys[1] = 98; values[1] = 102;
 194   i = pSHSearchMapInt(keys, values, 1, keys[1]);
 195   ok(i == -1, "Len 1 - array len 2, expected -1, got %d\n", i);
 196
 197   /* 2 elements, len = 2 */
 198   i = pSHSearchMapInt(keys, values, 2, keys[1]);
 199   ok(i == values[1], "Len 2, expected %d, got %d\n", values[1], i);
 200
 201   /* Searches forward */
 202   keys[2] = 99; values[2] = 103;
 203   i = pSHSearchMapInt(keys, values, 3, keys[0]);
 204   ok(i == values[0], "Len 3, expected %d, got %d\n", values[0], i);
 205 }
 206
 207 static void test_alloc_shared()
 208 {
 209     DWORD procid;
 210     HANDLE hmem;
 211     int val;
 212     int* p;
 213
 214     procid=GetCurrentProcessId();
 215     hmem=pSHAllocShared(NULL,10,procid);
 216     ok(hmem!=NULL,"SHAllocShared(NULL...) failed: %ld\n", GetLastError());
 217     ok(pSHFreeShared(hmem, procid),
 218        "SHFreeShared failed: %ld\n", GetLastError());
 219
 220     val=0x12345678;
 221     hmem=pSHAllocShared(&val,4,procid);
 222     ok(hmem!=NULL,"SHAllocShared(NULL...) failed: %ld\n", GetLastError());
 223
 224     p=(int*)pSHLockShared(hmem,procid);
 225     ok(p!=NULL,"SHLockShared failed: %ld\n", GetLastError());
 226     if (p!=NULL)
 227         ok(*p==val,"Wrong value in shared memory: %d instead of %d\n",*p,val);
 228     ok(pSHUnlockShared(p),"SHUnlockShared failed: %ld\n", GetLastError());
 229
 230     ok(pSHFreeShared(hmem, procid),
 231        "SHFreeShared failed: %ld\n", GetLastError());
 232 }
 233
 234 START_TEST(ordinal)
 235 {
 236   hShlwapi = LoadLibraryA("shlwapi.dll");
 237   ok(hShlwapi != 0, "LoadLibraryA failed\n");
 238   if (!hShlwapi)
 239     return;
 240
 241   pGetAcceptLanguagesA = (void*)GetProcAddress(hShlwapi, (LPSTR)14);
 242   pSHSearchMapInt = (void*)GetProcAddress(hShlwapi, (LPSTR)198);
 243   pSHAllocShared=(void*)GetProcAddress(hShlwapi,(char*)7);
 244   pSHLockShared=(void*)GetProcAddress(hShlwapi,(char*)8);
 245   pSHUnlockShared=(void*)GetProcAddress(hShlwapi,(char*)9);
 246   pSHFreeShared=(void*)GetProcAddress(hShlwapi,(char*)10);
 247
 248   test_GetAcceptLanguagesA();
 249   test_SHSearchMapInt();
 250   test_alloc_shared();
 251   FreeLibrary(hShlwapi);
 252 }