dlls/setupapi/tests/setupcab.c

   1 /*
   2  * Unit tests for SetupIterateCabinet
   3  *
   4  * Copyright 2007 Hans Leidekker
   5  * Copyright 2010 Andrew Nguyen
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include <stdarg.h>
  23
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "wingdi.h"
  27 #include "winuser.h"
  28 #include "winreg.h"
  29 #include "setupapi.h"
  30 #include "wine/test.h"
  31
  32 static void create_source_fileA(LPSTR filename, const BYTE *data, DWORD size)
  33 {
  34     HANDLE handle;
  35     DWORD written;
  36
  37     handle = CreateFileA(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
  38                          FILE_ATTRIBUTE_NORMAL, NULL);
  39     WriteFile(handle, data, size, &written, NULL);
  40     CloseHandle(handle);
  41 }
  42
  43 static void create_source_fileW(LPWSTR filename, const BYTE *data, DWORD size)
  44 {
  45     HANDLE handle;
  46     DWORD written;
  47
  48     handle = CreateFileW(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
  49                          FILE_ATTRIBUTE_NORMAL, NULL);
  50     WriteFile(handle, data, size, &written, NULL);
  51     CloseHandle(handle);
  52 }
  53
  54 static UINT CALLBACK dummy_callbackA(PVOID Context, UINT Notification,
  55                                      UINT_PTR Param1, UINT_PTR Param2)
  56 {
  57     ok(0, "Received unexpected notification (%p, %u, %lu, %lu)\n", Context,
  58        Notification, Param1, Param2);
  59     return 0;
  60 }
  61
  62 static UINT CALLBACK dummy_callbackW(PVOID Context, UINT Notification,
  63                                      UINT_PTR Param1, UINT_PTR Param2)
  64 {
  65     ok(0, "Received unexpected notification (%p, %u, %lu, %lu)\n", Context,
  66        Notification, Param1, Param2);
  67     return 0;
  68 }
  69
  70 static void test_invalid_parametersA(void)
  71 {
  72     BOOL ret;
  73     char source[MAX_PATH], temp[MAX_PATH];
  74     int i;
  75
  76     const struct
  77     {
  78         PCSTR CabinetFile;
  79         PSP_FILE_CALLBACK MsgHandler;
  80         DWORD expected_lasterror;
  81         int todo_lasterror;
  82     } invalid_parameters[] =
  83     {
  84         {NULL,                  NULL,            ERROR_INVALID_PARAMETER},
  85         {NULL,                  dummy_callbackA, ERROR_INVALID_PARAMETER},
  86         {"c:\\nonexistent.cab", NULL,            ERROR_FILE_NOT_FOUND},
  87         {"c:\\nonexistent.cab", dummy_callbackA, ERROR_FILE_NOT_FOUND},
  88         {source,                NULL,            ERROR_INVALID_DATA, 1},
  89         {source,                dummy_callbackA, ERROR_INVALID_DATA, 1},
  90     };
  91
  92     GetTempPathA(sizeof(temp), temp);
  93     GetTempFileNameA(temp, "doc", 0, source);
  94
  95     create_source_fileA(source, NULL, 0);
  96
  97     for (i = 0; i < sizeof(invalid_parameters)/sizeof(invalid_parameters[0]); i++)
  98     {
  99         SetLastError(0xdeadbeef);
 100         ret = SetupIterateCabinetA(invalid_parameters[i].CabinetFile, 0,
 101                                    invalid_parameters[i].MsgHandler, NULL);
 102         ok(!ret, "[%d] Expected SetupIterateCabinetA to return 0, got %d\n", i, ret);
 103         if (invalid_parameters[i].todo_lasterror)
 104         {
 105             todo_wine
 106             ok(GetLastError() == invalid_parameters[i].expected_lasterror,
 107                "[%d] Expected GetLastError() to return %u, got %u\n",
 108                i, invalid_parameters[i].expected_lasterror, GetLastError());
 109         }
 110         else
 111         {
 112             ok(GetLastError() == invalid_parameters[i].expected_lasterror,
 113                "[%d] Expected GetLastError() to return %u, got %u\n",
 114                i, invalid_parameters[i].expected_lasterror, GetLastError());
 115         }
 116     }
 117
 118     SetLastError(0xdeadbeef);
 119     ret = SetupIterateCabinetA("", 0, NULL, NULL);
 120     ok(!ret, "Expected SetupIterateCabinetA to return 0, got %d\n", ret);
 121     ok(GetLastError() == ERROR_NOT_ENOUGH_MEMORY ||
 122        GetLastError() == ERROR_FILE_NOT_FOUND, /* Win9x/NT4/Win2k */
 123        "Expected GetLastError() to return ERROR_NOT_ENOUGH_MEMORY, got %u\n",
 124        GetLastError());
 125
 126     SetLastError(0xdeadbeef);
 127     ret = SetupIterateCabinetA("", 0, dummy_callbackA, NULL);
 128     ok(!ret, "Expected SetupIterateCabinetA to return 0, got %d\n", ret);
 129     ok(GetLastError() == ERROR_NOT_ENOUGH_MEMORY ||
 130        GetLastError() == ERROR_FILE_NOT_FOUND, /* Win9x/NT4/Win2k */
 131        "Expected GetLastError() to return ERROR_NOT_ENOUGH_MEMORY, got %u\n",
 132        GetLastError());
 133
 134     DeleteFileA(source);
 135 }
 136
 137 static void test_invalid_parametersW(void)
 138 {
 139     static const WCHAR nonexistentW[] = {'c',':','\\','n','o','n','e','x','i','s','t','e','n','t','.','c','a','b',0};
 140     static const WCHAR docW[] = {'d','o','c',0};
 141     static const WCHAR emptyW[] = {0};
 142
 143     BOOL ret;
 144     WCHAR source[MAX_PATH], temp[MAX_PATH];
 145     int i;
 146
 147     const struct
 148     {
 149         PCWSTR CabinetFile;
 150         PSP_FILE_CALLBACK MsgHandler;
 151         DWORD expected_lasterror;
 152         int todo_lasterror;
 153     } invalid_parameters[] =
 154     {
 155         {nonexistentW, NULL,            ERROR_FILE_NOT_FOUND},
 156         {nonexistentW, dummy_callbackW, ERROR_FILE_NOT_FOUND},
 157         {source,       NULL,            ERROR_INVALID_DATA, 1},
 158         {source,       dummy_callbackW, ERROR_INVALID_DATA, 1},
 159     };
 160
 161     ret = SetupIterateCabinetW(NULL, 0, NULL, NULL);
 162     if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
 163     {
 164         win_skip("SetupIterateCabinetW is not available\n");
 165         return;
 166     }
 167
 168     GetTempPathW(sizeof(temp)/sizeof(WCHAR), temp);
 169     GetTempFileNameW(temp, docW, 0, source);
 170
 171     create_source_fileW(source, NULL, 0);
 172
 173     for (i = 0; i < sizeof(invalid_parameters)/sizeof(invalid_parameters[0]); i++)
 174     {
 175         SetLastError(0xdeadbeef);
 176         ret = SetupIterateCabinetW(invalid_parameters[i].CabinetFile, 0,
 177                                    invalid_parameters[i].MsgHandler, NULL);
 178         ok(!ret, "[%d] Expected SetupIterateCabinetW to return 0, got %d\n", i, ret);
 179         if (invalid_parameters[i].todo_lasterror)
 180         {
 181             todo_wine
 182             ok(GetLastError() == invalid_parameters[i].expected_lasterror,
 183                "[%d] Expected GetLastError() to return %u, got %u\n",
 184                i, invalid_parameters[i].expected_lasterror, GetLastError());
 185         }
 186         else
 187         {
 188             ok(GetLastError() == invalid_parameters[i].expected_lasterror,
 189                "[%d] Expected GetLastError() to return %u, got %u\n",
 190                i, invalid_parameters[i].expected_lasterror, GetLastError());
 191         }
 192     }
 193
 194     SetLastError(0xdeadbeef);
 195     ret = SetupIterateCabinetW(NULL, 0, NULL, NULL);
 196     ok(!ret, "Expected SetupIterateCabinetW to return 0, got %d\n", ret);
 197     ok(GetLastError() == ERROR_INVALID_PARAMETER ||
 198        GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Vista/Win2k8 */
 199        "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
 200        GetLastError());
 201
 202     SetLastError(0xdeadbeef);
 203     ret = SetupIterateCabinetW(NULL, 0, dummy_callbackW, NULL);
 204     ok(!ret, "Expected SetupIterateCabinetW to return 0, got %d\n", ret);
 205     ok(GetLastError() == ERROR_INVALID_PARAMETER ||
 206        GetLastError() == ERROR_NOT_ENOUGH_MEMORY, /* Vista/Win2k8 */
 207        "Expected GetLastError() to return ERROR_INVALID_PARAMETER, got %u\n",
 208        GetLastError());
 209
 210     SetLastError(0xdeadbeef);
 211     ret = SetupIterateCabinetW(emptyW, 0, NULL, NULL);
 212     ok(!ret, "Expected SetupIterateCabinetW to return 0, got %d\n", ret);
 213     ok(GetLastError() == ERROR_NOT_ENOUGH_MEMORY ||
 214        GetLastError() == ERROR_FILE_NOT_FOUND, /* NT4/Win2k */
 215        "Expected GetLastError() to return ERROR_NOT_ENOUGH_MEMORY, got %u\n",
 216        GetLastError());
 217
 218     SetLastError(0xdeadbeef);
 219     ret = SetupIterateCabinetW(emptyW, 0, dummy_callbackW, NULL);
 220     ok(!ret, "Expected SetupIterateCabinetW to return 0, got %d\n", ret);
 221     ok(GetLastError() == ERROR_NOT_ENOUGH_MEMORY ||
 222        GetLastError() == ERROR_FILE_NOT_FOUND, /* NT4/Win2k */
 223        "Expected GetLastError() to return ERROR_NOT_ENOUGH_MEMORY, got %u\n",
 224        GetLastError());
 225
 226     DeleteFileW(source);
 227 }
 228
 229 START_TEST(setupcab)
 230 {
 231     test_invalid_parametersA();
 232     test_invalid_parametersW();
 233 }