dlls/setupapi/tests/query.c

   1 /*
   2  * Unit tests for setupapi.dll query functions
   3  *
   4  * Copyright (C) 2006 James Hawkins
   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 <stdio.h>
  22 #include <windows.h>
  23 #include <setupapi.h>
  24 #include "wine/test.h"
  25
  26 CHAR CURR_DIR[MAX_PATH];
  27 CHAR WIN_DIR[MAX_PATH];
  28
  29 static void get_directories(void)
  30 {
  31     int len;
  32
  33     GetCurrentDirectoryA(MAX_PATH, CURR_DIR);
  34     len = lstrlenA(CURR_DIR);
  35
  36     if(len && (CURR_DIR[len-1] == '\\'))
  37         CURR_DIR[len-1] = 0;
  38
  39     GetWindowsDirectoryA(WIN_DIR, MAX_PATH);
  40     len = lstrlenA(WIN_DIR);
  41
  42     if (len && (WIN_DIR[len-1] == '\\'))
  43         WIN_DIR[len-1] = 0;
  44 }
  45
  46 static void append_str(char **str, const char *data)
  47 {
  48     sprintf(*str, data);
  49     *str += strlen(*str);
  50 }
  51
  52 static void create_inf_file(LPSTR filename)
  53 {
  54     char data[1024];
  55     char *ptr = data;
  56     DWORD dwNumberOfBytesWritten;
  57     HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
  58                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
  59
  60     append_str(&ptr, "[Version]\n");
  61     append_str(&ptr, "Signature=\"$Chicago$\"\n");
  62     append_str(&ptr, "AdvancedINF=2.5\n");
  63     append_str(&ptr, "[SourceDisksNames]\n");
  64     append_str(&ptr, "2 = %%SrcDiskName%%, LANCOM\\LANtools\\lanconf.cab\n");
  65     append_str(&ptr, "[SourceDisksFiles]\n");
  66     append_str(&ptr, "lanconf.exe = 2\n");
  67     append_str(&ptr, "[DestinationDirs]\n");
  68     append_str(&ptr, "DefaultDestDir = 24, %%DefaultDest%%\n");
  69     append_str(&ptr, "[Strings]\n");
  70     append_str(&ptr, "LangDir = english\n");
  71     append_str(&ptr, "DefaultDest = LANCOM\n");
  72     append_str(&ptr, "SrcDiskName = \"LANCOM Software CD\"\n");
  73
  74     WriteFile(hf, data, ptr - data, &dwNumberOfBytesWritten, NULL);
  75     CloseHandle(hf);
  76 }
  77
  78 static void create_inf_file2(LPSTR filename)
  79 {
  80     char data[1024];
  81     char *ptr = data;
  82     DWORD dwNumberOfBytesWritten;
  83     HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
  84                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
  85
  86     append_str(&ptr, "[SourceFileInfo]\n");
  87     append_str(&ptr, "sp1qfe\\bitsinst.exe=250B3702C7CCD7C2F9E4DAA1555C933E,000600060A28062C,27136,SP1QFE\n");
  88     append_str(&ptr, "sp1qfe\\bitsprx2.dll=4EBEA67F4BB4EB402E725CA7CA2857AE,000600060A280621,7680,SP1QFE\n");
  89     append_str(&ptr, "sp1qfe\\bitsprx3.dll=C788A1D9330DA011EF25E95D3BC7BDE5,000600060A280621,7168,SP1QFE\n");
  90     append_str(&ptr, "sp1qfe\\qmgr.dll=696AC82FB290A03F205901442E0E9589,000600060A280621,361984,SP1QFE\n");
  91     append_str(&ptr, "sp1qfe\\qmgrprxy.dll=8B5848144829E1BC985EA4C3D8CA7E3F,000600060A280621,17408,SP1QFE\n");
  92     append_str(&ptr, "sp1qfe\\winhttp.dll=3EC6F518114606CA59D4160322077437,000500010A280615,331776,SP1QFE\n");
  93     append_str(&ptr, "sp1qfe\\xpob2res.dll=DB83156B9F496F20D1EA70E4ABEC0166,000500010A280622,158720,SP1QFE\n");
  94
  95     WriteFile(hf, data, ptr - data, &dwNumberOfBytesWritten, NULL);
  96     CloseHandle(hf);
  97 }
  98
  99 static BOOL check_info_filename(PSP_INF_INFORMATION info, LPSTR test)
 100 {
 101     LPSTR filename;
 102     DWORD size;
 103     BOOL ret = FALSE;
 104
 105     if (!SetupQueryInfFileInformationA(info, 0, NULL, 0, &size))
 106         return FALSE;
 107
 108     filename = HeapAlloc(GetProcessHeap(), 0, size);
 109     if (!filename)
 110         return FALSE;
 111
 112     SetupQueryInfFileInformationA(info, 0, filename, size, &size);
 113
 114     if (!lstrcmpiA(test, filename))
 115         ret = TRUE;
 116
 117     HeapFree(GetProcessHeap(), 0, filename);
 118     return ret;
 119 }
 120
 121 static PSP_INF_INFORMATION alloc_inf_info(LPCSTR filename, DWORD search, PDWORD size)
 122 {
 123     PSP_INF_INFORMATION info;
 124     BOOL ret;
 125
 126     ret = SetupGetInfInformationA(filename, search, NULL, 0, size);
 127     if (!ret)
 128         return NULL;
 129
 130     info = HeapAlloc(GetProcessHeap(), 0, *size);
 131     return info;
 132 }
 133
 134 static void test_SetupGetInfInformation(void)
 135 {
 136     PSP_INF_INFORMATION info;
 137     CHAR inf_filename[MAX_PATH];
 138     CHAR inf_one[MAX_PATH], inf_two[MAX_PATH];
 139     DWORD size;
 140     HINF hinf;
 141     BOOL ret;
 142
 143     lstrcpyA(inf_filename, CURR_DIR);
 144     lstrcatA(inf_filename, "\\");
 145     lstrcatA(inf_filename, "test.inf");
 146
 147     /* try an invalid inf handle */
 148     size = 0xdeadbeef;
 149     SetLastError(0xbeefcafe);
 150     ret = SetupGetInfInformationA(NULL, INFINFO_INF_SPEC_IS_HINF, NULL, 0, &size);
 151     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
 152     ok(GetLastError() == ERROR_INVALID_HANDLE,
 153        "Expected ERROR_INVALID_HANDLE, got %d\n", GetLastError());
 154     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");
 155
 156     /* try an invalid inf filename */
 157     size = 0xdeadbeef;
 158     SetLastError(0xbeefcafe);
 159     ret = SetupGetInfInformationA(NULL, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size);
 160     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
 161     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 162        "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
 163     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");
 164
 165     create_inf_file(inf_filename);
 166
 167     /* try an invalid search flag */
 168     size = 0xdeadbeef;
 169     SetLastError(0xbeefcafe);
 170     ret = SetupGetInfInformationA(inf_filename, -1, NULL, 0, &size);
 171     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
 172     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 173        "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
 174     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");
 175
 176     /* try a nonexistent inf file */
 177     size = 0xdeadbeef;
 178     SetLastError(0xbeefcafe);
 179     ret = SetupGetInfInformationA("idontexist", INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size);
 180     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
 181     ok(GetLastError() == ERROR_FILE_NOT_FOUND,
 182        "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
 183     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");
 184
 185     /* successfully open the inf file */
 186     size = 0xdeadbeef;
 187     ret = SetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size);
 188     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
 189     ok(size != 0xdeadbeef, "Expected a valid size on return\n");
 190
 191     /* set ReturnBuffer to NULL and ReturnBufferSize to non-zero value 'size' */
 192     SetLastError(0xbeefcafe);
 193     ret = SetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size, &size);
 194     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
 195     ok(GetLastError() == ERROR_INVALID_PARAMETER,
 196        "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
 197
 198     /* set ReturnBuffer to NULL and ReturnBufferSize to non-zero value 'size-1' */
 199     ret = SetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size-1, &size);
 200     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
 201
 202     /* some tests for behaviour with a NULL RequiredSize pointer */
 203     ret = SetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, NULL);
 204     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
 205     ret = SetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size - 1, NULL);
 206     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
 207     ret = SetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size, NULL);
 208     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
 209
 210     info = HeapAlloc(GetProcessHeap(), 0, size);
 211
 212     /* try valid ReturnBuffer but too small size */
 213     SetLastError(0xbeefcafe);
 214     ret = SetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, info, size - 1, &size);
 215     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
 216     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
 217        "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
 218
 219     /* successfully get the inf information */
 220     ret = SetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, info, size, &size);
 221     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
 222     ok(check_info_filename(info, inf_filename), "Expected returned filename to be equal\n");
 223
 224     HeapFree(GetProcessHeap(), 0, info);
 225
 226     /* try the INFINFO_INF_SPEC_IS_HINF search flag */
 227     hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
 228     info = alloc_inf_info(hinf, INFINFO_INF_SPEC_IS_HINF, &size);
 229     ret = SetupGetInfInformationA(hinf, INFINFO_INF_SPEC_IS_HINF, info, size, &size);
 230     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
 231     ok(check_info_filename(info, inf_filename), "Expected returned filename to be equal\n");
 232     SetupCloseInfFile(hinf);
 233
 234     lstrcpyA(inf_one, WIN_DIR);
 235     lstrcatA(inf_one, "\\inf\\");
 236     lstrcatA(inf_one, "test.inf");
 237     create_inf_file(inf_one);
 238
 239     lstrcpyA(inf_two, WIN_DIR);
 240     lstrcatA(inf_two, "\\system32\\");
 241     lstrcatA(inf_two, "test.inf");
 242     create_inf_file(inf_two);
 243
 244     HeapFree(GetProcessHeap(), 0, info);
 245     info = alloc_inf_info("test.inf", INFINFO_DEFAULT_SEARCH, &size);
 246
 247     /* test the INFINFO_DEFAULT_SEARCH search flag */
 248     ret = SetupGetInfInformationA("test.inf", INFINFO_DEFAULT_SEARCH, info, size, &size);
 249     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
 250     ok(check_info_filename(info, inf_one), "Expected returned filename to be equal\n");
 251
 252     HeapFree(GetProcessHeap(), 0, info);
 253     info = alloc_inf_info("test.inf", INFINFO_REVERSE_DEFAULT_SEARCH, &size);
 254
 255     /* test the INFINFO_REVERSE_DEFAULT_SEARCH search flag */
 256     ret = SetupGetInfInformationA("test.inf", INFINFO_REVERSE_DEFAULT_SEARCH, info, size, &size);
 257     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
 258     ok(check_info_filename(info, inf_two), "Expected returned filename to be equal\n");
 259
 260     DeleteFileA(inf_filename);
 261     DeleteFileA(inf_one);
 262     DeleteFileA(inf_two);
 263 }
 264
 265 static void test_SetupGetSourceFileLocation(void)
 266 {
 267     char buffer[MAX_PATH] = "not empty", inf_filename[MAX_PATH];
 268     UINT source_id;
 269     DWORD required, error;
 270     HINF hinf;
 271     BOOL ret;
 272
 273     lstrcpyA(inf_filename, CURR_DIR);
 274     lstrcatA(inf_filename, "\\");
 275     lstrcatA(inf_filename, "test.inf");
 276
 277     create_inf_file(inf_filename);
 278
 279     hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
 280     ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
 281
 282     required = 0;
 283     source_id = 0;
 284
 285     ret = SetupGetSourceFileLocationA(hinf, NULL, "lanconf.exe", &source_id, buffer, sizeof(buffer), &required);
 286     ok(ret, "SetupGetSourceFileLocation failed\n");
 287
 288     ok(required == 1, "unexpected required size: %d\n", required);
 289     ok(source_id == 2, "unexpected source id: %d\n", source_id);
 290     ok(!lstrcmpA("", buffer), "unexpected result string: %s\n", buffer);
 291
 292     SetupCloseInfFile(hinf);
 293     DeleteFileA(inf_filename);
 294
 295     create_inf_file2(inf_filename);
 296
 297     SetLastError(0xdeadbeef);
 298     hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
 299     error = GetLastError();
 300     ok(hinf == INVALID_HANDLE_VALUE, "could open inf file\n");
 301     ok(error == ERROR_WRONG_INF_STYLE, "got wrong error: %d\n", error);
 302
 303     hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_OLDNT, NULL);
 304     ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
 305
 306     ret = SetupGetSourceFileLocationA(hinf, NULL, "", &source_id, buffer, sizeof(buffer), &required);
 307     ok(!ret, "SetupGetSourceFileLocation succeeded\n");
 308
 309     SetupCloseInfFile(hinf);
 310     DeleteFileA(inf_filename);
 311 }
 312
 313 static void test_SetupGetSourceInfo(void)
 314 {
 315     char buffer[MAX_PATH], inf_filename[MAX_PATH];
 316     DWORD required;
 317     HINF hinf;
 318     BOOL ret;
 319
 320     lstrcpyA(inf_filename, CURR_DIR);
 321     lstrcatA(inf_filename, "\\");
 322     lstrcatA(inf_filename, "test.inf");
 323
 324     create_inf_file(inf_filename);
 325
 326     hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
 327     ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
 328
 329     required = 0;
 330
 331     ret = SetupGetSourceInfoA(hinf, 2, SRCINFO_PATH, buffer, sizeof(buffer), &required);
 332     ok(ret, "SetupGetSourceInfoA failed\n");
 333
 334     ok(required == 1, "unexpected required size: %d\n", required);
 335     ok(!lstrcmpA("", buffer), "unexpected result string: %s\n", buffer);
 336
 337     required = 0;
 338     buffer[0] = 0;
 339
 340     ret = SetupGetSourceInfoA(hinf, 2, SRCINFO_TAGFILE, buffer, sizeof(buffer), &required);
 341     ok(ret, "SetupGetSourceInfoA failed\n");
 342
 343     ok(required == 28, "unexpected required size: %d\n", required);
 344     ok(!lstrcmpA("LANCOM\\LANtools\\lanconf.cab", buffer), "unexpected result string: %s\n", buffer);
 345
 346     required = 0;
 347     buffer[0] = 0;
 348
 349     ret = SetupGetSourceInfoA(hinf, 2, SRCINFO_DESCRIPTION, buffer, sizeof(buffer), &required);
 350     ok(ret, "SetupGetSourceInfoA failed\n");
 351
 352     ok(required == 19, "unexpected required size: %d\n", required);
 353     ok(!lstrcmpA("LANCOM Software CD", buffer), "unexpected result string: %s\n", buffer);
 354
 355     SetupCloseInfFile(hinf);
 356     DeleteFileA(inf_filename);
 357 }
 358
 359 static void test_SetupGetTargetPath(void)
 360 {
 361     char buffer[MAX_PATH], inf_filename[MAX_PATH];
 362     DWORD required;
 363     HINF hinf;
 364     INFCONTEXT ctx;
 365     BOOL ret;
 366
 367     lstrcpyA(inf_filename, CURR_DIR);
 368     lstrcatA(inf_filename, "\\");
 369     lstrcatA(inf_filename, "test.inf");
 370
 371     create_inf_file(inf_filename);
 372
 373     hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
 374     ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
 375
 376     ctx.Inf = hinf;
 377     ctx.CurrentInf = hinf;
 378     ctx.Section = 7;
 379     ctx.Line = 0;
 380
 381     required = 0;
 382
 383     ret = SetupGetTargetPathA(hinf, &ctx, NULL, buffer, sizeof(buffer), &required);
 384     ok(ret, "SetupGetTargetPathA failed\n");
 385
 386     ok(required == 10, "unexpected required size: %d\n", required);
 387     ok(!lstrcmpiA("C:\\LANCOM", buffer), "unexpected result string: %s\n", buffer);
 388
 389     SetupCloseInfFile(hinf);
 390     DeleteFileA(inf_filename);
 391 }
 392
 393 START_TEST(query)
 394 {
 395     get_directories();
 396
 397     test_SetupGetInfInformation();
 398     test_SetupGetSourceFileLocation();
 399     test_SetupGetSourceInfo();
 400     test_SetupGetTargetPath();
 401 }