sync rostests to r44455
[reactos.git] / rostests / winetests / setupapi / query.c
index 87f1673..bbc2c65 100644 (file)
@@ -43,56 +43,68 @@ static void get_directories(void)
         WIN_DIR[len-1] = 0;
 }
 
-static void append_str(char **str, const char *data)
+static const char inf_data1[] =
+    "[Version]\n"
+    "Signature=\"$Chicago$\"\n"
+    "AdvancedINF=2.5\n"
+    "[SourceDisksNames]\n"
+    "2 = %SrcDiskName%, LANCOM\\LANtools\\lanconf.cab\n"
+    "[SourceDisksFiles]\n"
+    "lanconf.exe = 2\n"
+    "[DestinationDirs]\n"
+    "DefaultDestDir = 24, %DefaultDest%\n"
+    "[Strings]\n"
+    "LangDir = english\n"
+    "DefaultDest = LANCOM\n"
+    "SrcDiskName = \"LANCOM Software CD\"\n";
+
+static const char inf_data2[] =
+    "[SourceFileInfo]\n"
+    "sp1qfe\\bitsinst.exe=250B3702C7CCD7C2F9E4DAA1555C933E,000600060A28062C,27136,SP1QFE\n"
+    "sp1qfe\\bitsprx2.dll=4EBEA67F4BB4EB402E725CA7CA2857AE,000600060A280621,7680,SP1QFE\n"
+    "sp1qfe\\bitsprx3.dll=C788A1D9330DA011EF25E95D3BC7BDE5,000600060A280621,7168,SP1QFE\n"
+    "sp1qfe\\qmgr.dll=696AC82FB290A03F205901442E0E9589,000600060A280621,361984,SP1QFE\n"
+    "sp1qfe\\qmgrprxy.dll=8B5848144829E1BC985EA4C3D8CA7E3F,000600060A280621,17408,SP1QFE\n"
+    "sp1qfe\\winhttp.dll=3EC6F518114606CA59D4160322077437,000500010A280615,331776,SP1QFE\n"
+    "sp1qfe\\xpob2res.dll=DB83156B9F496F20D1EA70E4ABEC0166,000500010A280622,158720,SP1QFE\n";
+
+static const char inf_data3[] =
+    "[Version]\n"
+    "Signature = \"$Windows NT$\"\n"
+    "[DestinationDirs]\n"
+    "CopyAlways.Windir.files = 10\n"
+    "[CopyAlways.Windir.Files]\n"
+    "WindowsCodecs.dll\n";
+
+static const char inf_data4[] =
+    "[Version]\n"
+    "Signature = \"$Windows NT$\"\n"
+    "[CopyAlways.System32.Files]\n"
+    "WindowsCodecs.dll\n";
+
+static const char inf_data5[] =
+    "[Version]\n"
+    "Signature = \"$Windows NT$\"\n"
+    "[DestinationDirs]\n"
+    "DefaultDestDir = 11\n"
+    "CopyAlways.Windir.files = 10\n"
+    "[CopyAlways.Windir.Files]\n"
+    "WindowsCodecs.dll\n";
+
+static const char inf_data6[] =
+    "[Version]\n"
+    "Signature = \"$Windows NT$\"\n"
+    "[DestinationDirs]\n"
+    "DefaultDestDir = 10\n"
+    "[CopyAlways.Windir.Files]\n"
+    "WindowsCodecs.dll\n";
+
+static void create_inf_file(LPSTR filename, const char *data, DWORD size)
 {
-    sprintf(*str, data);
-    *str += strlen(*str);
-}
-
-static void create_inf_file(LPSTR filename)
-{
-    char data[1024];
-    char *ptr = data;
     DWORD dwNumberOfBytesWritten;
     HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
-
-    append_str(&ptr, "[Version]\n");
-    append_str(&ptr, "Signature=\"$Chicago$\"\n");
-    append_str(&ptr, "AdvancedINF=2.5\n");
-    append_str(&ptr, "[SourceDisksNames]\n");
-    append_str(&ptr, "2 = %%SrcDiskName%%, LANCOM\\LANtools\\lanconf.cab\n");
-    append_str(&ptr, "[SourceDisksFiles]\n");
-    append_str(&ptr, "lanconf.exe = 2\n");
-    append_str(&ptr, "[DestinationDirs]\n");
-    append_str(&ptr, "DefaultDestDir = 24, %%DefaultDest%%\n");
-    append_str(&ptr, "[Strings]\n");
-    append_str(&ptr, "LangDir = english\n");
-    append_str(&ptr, "DefaultDest = LANCOM\n");
-    append_str(&ptr, "SrcDiskName = \"LANCOM Software CD\"\n");
-
-    WriteFile(hf, data, ptr - data, &dwNumberOfBytesWritten, NULL);
-    CloseHandle(hf);
-}
-
-static void create_inf_file2(LPSTR filename)
-{
-    char data[1024];
-    char *ptr = data;
-    DWORD dwNumberOfBytesWritten;
-    HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
-                           CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
-
-    append_str(&ptr, "[SourceFileInfo]\n");
-    append_str(&ptr, "sp1qfe\\bitsinst.exe=250B3702C7CCD7C2F9E4DAA1555C933E,000600060A28062C,27136,SP1QFE\n");
-    append_str(&ptr, "sp1qfe\\bitsprx2.dll=4EBEA67F4BB4EB402E725CA7CA2857AE,000600060A280621,7680,SP1QFE\n");
-    append_str(&ptr, "sp1qfe\\bitsprx3.dll=C788A1D9330DA011EF25E95D3BC7BDE5,000600060A280621,7168,SP1QFE\n");
-    append_str(&ptr, "sp1qfe\\qmgr.dll=696AC82FB290A03F205901442E0E9589,000600060A280621,361984,SP1QFE\n");
-    append_str(&ptr, "sp1qfe\\qmgrprxy.dll=8B5848144829E1BC985EA4C3D8CA7E3F,000600060A280621,17408,SP1QFE\n");
-    append_str(&ptr, "sp1qfe\\winhttp.dll=3EC6F518114606CA59D4160322077437,000500010A280615,331776,SP1QFE\n");
-    append_str(&ptr, "sp1qfe\\xpob2res.dll=DB83156B9F496F20D1EA70E4ABEC0166,000500010A280622,158720,SP1QFE\n");
-
-    WriteFile(hf, data, ptr - data, &dwNumberOfBytesWritten, NULL);
+    WriteFile(hf, data, size, &dwNumberOfBytesWritten, NULL);
     CloseHandle(hf);
 }
 
@@ -136,6 +148,7 @@ static void test_SetupGetInfInformation(void)
     PSP_INF_INFORMATION info;
     CHAR inf_filename[MAX_PATH];
     CHAR inf_one[MAX_PATH], inf_two[MAX_PATH];
+    LPSTR revfile;
     DWORD size;
     HINF hinf;
     BOOL ret;
@@ -149,11 +162,20 @@ static void test_SetupGetInfInformation(void)
     SetLastError(0xbeefcafe);
     ret = SetupGetInfInformationA(NULL, INFINFO_INF_SPEC_IS_HINF, NULL, 0, &size);
     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
-    ok(GetLastError() == ERROR_INVALID_HANDLE,
+    ok(GetLastError() == ERROR_INVALID_HANDLE ||
+       broken(GetLastError() == ERROR_BAD_PATHNAME) || /* win95 */
+       broken(GetLastError() == ERROR_FILE_NOT_FOUND) || /* win98 */
+       broken(GetLastError() == ERROR_PATH_NOT_FOUND) || /* NT4 */
+       broken(GetLastError() == ERROR_INVALID_NAME) || /* win2k */
+       broken(GetLastError() == ERROR_GENERAL_SYNTAX), /* another win2k */
        "Expected ERROR_INVALID_HANDLE, got %d\n", GetLastError());
     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");
 
     /* try an invalid inf filename */
+    /* do not use NULL as absolute inf filename on win9x (crash) */
+    if ((GetLastError() != ERROR_BAD_PATHNAME) &&   /* win95 */
+        (GetLastError() != ERROR_FILE_NOT_FOUND))  /* win98 */
+    {
     size = 0xdeadbeef;
     SetLastError(0xbeefcafe);
     ret = SetupGetInfInformationA(NULL, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size);
@@ -161,8 +183,9 @@ static void test_SetupGetInfInformation(void)
     ok(GetLastError() == ERROR_INVALID_PARAMETER,
        "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");
+    }
 
-    create_inf_file(inf_filename);
+    create_inf_file(inf_filename, inf_data1, sizeof(inf_data1) - 1);
 
     /* try an invalid search flag */
     size = 0xdeadbeef;
@@ -231,22 +254,32 @@ static void test_SetupGetInfInformation(void)
     ok(check_info_filename(info, inf_filename), "Expected returned filename to be equal\n");
     SetupCloseInfFile(hinf);
 
-    lstrcpyA(inf_one, WIN_DIR);
-    lstrcatA(inf_one, "\\inf\\");
-    lstrcatA(inf_one, "test.inf");
-    create_inf_file(inf_one);
-
     lstrcpyA(inf_two, WIN_DIR);
     lstrcatA(inf_two, "\\system32\\");
     lstrcatA(inf_two, "test.inf");
-    create_inf_file(inf_two);
+    create_inf_file(inf_two, inf_data1, sizeof(inf_data1) - 1);
+
+    HeapFree(GetProcessHeap(), 0, info);
+    info = alloc_inf_info("test.inf", INFINFO_DEFAULT_SEARCH, &size);
+
+    /* check if system32 is searched for inf */
+    ret = SetupGetInfInformationA("test.inf", INFINFO_DEFAULT_SEARCH, info, size, &size);
+    if (!ret && GetLastError() == ERROR_FILE_NOT_FOUND)
+        revfile = inf_one; /* Vista */
+    else
+        revfile = inf_two;
+
+    lstrcpyA(inf_one, WIN_DIR);
+    lstrcatA(inf_one, "\\inf\\");
+    lstrcatA(inf_one, "test.inf");
+    create_inf_file(inf_one, inf_data1, sizeof(inf_data1) - 1);
 
     HeapFree(GetProcessHeap(), 0, info);
     info = alloc_inf_info("test.inf", INFINFO_DEFAULT_SEARCH, &size);
 
     /* test the INFINFO_DEFAULT_SEARCH search flag */
     ret = SetupGetInfInformationA("test.inf", INFINFO_DEFAULT_SEARCH, info, size, &size);
-    ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
+    ok(ret == TRUE, "Expected SetupGetInfInformation to succeed: %d\n", GetLastError());
     ok(check_info_filename(info, inf_one), "Expected returned filename to be equal\n");
 
     HeapFree(GetProcessHeap(), 0, info);
@@ -255,7 +288,9 @@ static void test_SetupGetInfInformation(void)
     /* test the INFINFO_REVERSE_DEFAULT_SEARCH search flag */
     ret = SetupGetInfInformationA("test.inf", INFINFO_REVERSE_DEFAULT_SEARCH, info, size, &size);
     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
-    ok(check_info_filename(info, inf_two), "Expected returned filename to be equal\n");
+    ok(check_info_filename(info, revfile), "Expected returned filename to be equal\n");
+
+    HeapFree(GetProcessHeap(), 0, info);
 
     DeleteFileA(inf_filename);
     DeleteFileA(inf_one);
@@ -274,7 +309,7 @@ static void test_SetupGetSourceFileLocation(void)
     lstrcatA(inf_filename, "\\");
     lstrcatA(inf_filename, "test.inf");
 
-    create_inf_file(inf_filename);
+    create_inf_file(inf_filename, inf_data1, sizeof(inf_data1) - 1);
 
     hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
     ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
@@ -292,7 +327,7 @@ static void test_SetupGetSourceFileLocation(void)
     SetupCloseInfFile(hinf);
     DeleteFileA(inf_filename);
 
-    create_inf_file2(inf_filename);
+    create_inf_file(inf_filename, inf_data2, sizeof(inf_data2) - 1);
 
     SetLastError(0xdeadbeef);
     hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
@@ -321,7 +356,7 @@ static void test_SetupGetSourceInfo(void)
     lstrcatA(inf_filename, "\\");
     lstrcatA(inf_filename, "test.inf");
 
-    create_inf_file(inf_filename);
+    create_inf_file(inf_filename, inf_data1, sizeof(inf_data1) - 1);
 
     hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
     ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
@@ -359,6 +394,7 @@ static void test_SetupGetSourceInfo(void)
 static void test_SetupGetTargetPath(void)
 {
     char buffer[MAX_PATH], inf_filename[MAX_PATH];
+    char destfile[MAX_PATH];
     DWORD required;
     HINF hinf;
     INFCONTEXT ctx;
@@ -368,7 +404,7 @@ static void test_SetupGetTargetPath(void)
     lstrcatA(inf_filename, "\\");
     lstrcatA(inf_filename, "test.inf");
 
-    create_inf_file(inf_filename);
+    create_inf_file(inf_filename, inf_data1, sizeof(inf_data1) - 1);
 
     hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
     ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
@@ -382,9 +418,86 @@ static void test_SetupGetTargetPath(void)
 
     ret = SetupGetTargetPathA(hinf, &ctx, NULL, buffer, sizeof(buffer), &required);
     ok(ret, "SetupGetTargetPathA failed\n");
-
     ok(required == 10, "unexpected required size: %d\n", required);
-    ok(!lstrcmpiA("C:\\LANCOM", buffer), "unexpected result string: %s\n", buffer);
+    /* Retrieve the system drive from the windows directory.
+     * (%SystemDrive% is not available on Win9x)
+     */
+    lstrcpyA(destfile, WIN_DIR);
+    destfile[3] = '\0';
+    lstrcatA(destfile, "LANCOM");
+    ok(!lstrcmpiA(destfile, buffer), "unexpected result string: %s\n", buffer);
+
+    SetupCloseInfFile(hinf);
+    DeleteFileA(inf_filename);
+
+    create_inf_file(inf_filename, inf_data3, sizeof(inf_data3) - 1);
+
+    hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
+    ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
+
+    required = 0;
+
+    ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.Windir.Files", buffer, sizeof(buffer), &required);
+    ok(ret, "SetupGetTargetPathA failed\n");
+
+    lstrcpyA(destfile, WIN_DIR);
+
+    ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
+    ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);
+
+    SetupCloseInfFile(hinf);
+    DeleteFileA(inf_filename);
+
+    create_inf_file(inf_filename, inf_data4, sizeof(inf_data4) - 1);
+
+    hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
+    ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
+
+    required = 0;
+
+    ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.System32.Files", buffer, sizeof(buffer), &required);
+    ok(ret, "SetupGetTargetPathA failed\n");
+
+    GetSystemDirectoryA(destfile, MAX_PATH);
+
+    ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
+    ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);
+
+    SetupCloseInfFile(hinf);
+    DeleteFileA(inf_filename);
+
+    create_inf_file(inf_filename, inf_data5, sizeof(inf_data5) - 1);
+
+    hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
+    ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
+
+    required = 0;
+
+    ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.Windir.Files", buffer, sizeof(buffer), &required);
+    ok(ret, "SetupGetTargetPathA failed\n");
+
+    lstrcpyA(destfile, WIN_DIR);
+
+    ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
+    ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);
+
+    SetupCloseInfFile(hinf);
+    DeleteFileA(inf_filename);
+
+    create_inf_file(inf_filename, inf_data6, sizeof(inf_data6) - 1);
+
+    hinf = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
+    ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
+
+    required = 0;
+
+    ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.Windir.Files", buffer, sizeof(buffer), &required);
+    ok(ret, "SetupGetTargetPathA failed\n");
+
+    lstrcpyA(destfile, WIN_DIR);
+
+    ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
+    ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);
 
     SetupCloseInfFile(hinf);
     DeleteFileA(inf_filename);