| /* |
| * Unit tests for setupapi.dll query functions |
| * |
| * Copyright (C) 2006 James Hawkins |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| |
| #include <stdio.h> |
| #include <windows.h> |
| #include <setupapi.h> |
| #include "wine/test.h" |
| |
| /* function pointers */ |
| static HMODULE hSetupAPI; |
| static void (WINAPI *pSetupCloseInfFile)(HINF); |
| static BOOL (WINAPI *pSetupGetInfInformationA)(LPCVOID,DWORD,PSP_INF_INFORMATION,DWORD,PDWORD); |
| static HINF (WINAPI *pSetupOpenInfFileA)(PCSTR,PCSTR,DWORD,PUINT); |
| static BOOL (WINAPI *pSetupQueryInfFileInformationA)(PSP_INF_INFORMATION,UINT,PSTR,DWORD,PDWORD); |
| static BOOL (WINAPI *pSetupGetSourceFileLocationA)(HINF,PINFCONTEXT,PCSTR,PUINT,PSTR,DWORD,PDWORD); |
| static BOOL (WINAPI *pSetupGetSourceInfoA)(HINF,UINT,UINT,PSTR,DWORD,PDWORD); |
| static BOOL (WINAPI *pSetupGetTargetPathA)(HINF,PINFCONTEXT,PCSTR,PSTR,DWORD,PDWORD); |
| |
| CHAR CURR_DIR[MAX_PATH]; |
| CHAR WIN_DIR[MAX_PATH]; |
| |
| static void init_function_pointers(void) |
| { |
| hSetupAPI = GetModuleHandleA("setupapi.dll"); |
| |
| pSetupCloseInfFile = (void *)GetProcAddress(hSetupAPI, "SetupCloseInfFile"); |
| pSetupGetInfInformationA = (void *)GetProcAddress(hSetupAPI, "SetupGetInfInformationA"); |
| pSetupOpenInfFileA = (void *)GetProcAddress(hSetupAPI, "SetupOpenInfFileA"); |
| pSetupQueryInfFileInformationA = (void *)GetProcAddress(hSetupAPI, "SetupQueryInfFileInformationA"); |
| pSetupGetSourceFileLocationA = (void *)GetProcAddress(hSetupAPI, "SetupGetSourceFileLocationA"); |
| pSetupGetSourceInfoA = (void *)GetProcAddress(hSetupAPI, "SetupGetSourceInfoA"); |
| pSetupGetTargetPathA = (void *)GetProcAddress(hSetupAPI, "SetupGetTargetPathA"); |
| } |
| |
| static void get_directories(void) |
| { |
| int len; |
| |
| GetCurrentDirectoryA(MAX_PATH, CURR_DIR); |
| len = lstrlenA(CURR_DIR); |
| |
| if(len && (CURR_DIR[len-1] == '\\')) |
| CURR_DIR[len-1] = 0; |
| |
| GetWindowsDirectoryA(WIN_DIR, MAX_PATH); |
| len = lstrlenA(WIN_DIR); |
| |
| if (len && (WIN_DIR[len-1] == '\\')) |
| WIN_DIR[len-1] = 0; |
| } |
| |
| static void append_str(char **str, const char *data) |
| { |
| 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); |
| CloseHandle(hf); |
| } |
| |
| static BOOL check_info_filename(PSP_INF_INFORMATION info, LPSTR test) |
| { |
| LPSTR filename; |
| DWORD size; |
| BOOL ret = FALSE; |
| |
| if (!pSetupQueryInfFileInformationA(info, 0, NULL, 0, &size)) |
| return FALSE; |
| |
| filename = HeapAlloc(GetProcessHeap(), 0, size); |
| if (!filename) |
| return FALSE; |
| |
| pSetupQueryInfFileInformationA(info, 0, filename, size, &size); |
| |
| if (!lstrcmpiA(test, filename)) |
| ret = TRUE; |
| |
| HeapFree(GetProcessHeap(), 0, filename); |
| return ret; |
| } |
| |
| static PSP_INF_INFORMATION alloc_inf_info(LPCSTR filename, DWORD search, PDWORD size) |
| { |
| PSP_INF_INFORMATION info; |
| BOOL ret; |
| |
| ret = pSetupGetInfInformationA(filename, search, NULL, 0, size); |
| if (!ret) |
| return NULL; |
| |
| info = HeapAlloc(GetProcessHeap(), 0, *size); |
| return info; |
| } |
| |
| static void test_SetupGetInfInformation(void) |
| { |
| PSP_INF_INFORMATION info; |
| CHAR inf_filename[MAX_PATH]; |
| CHAR inf_one[MAX_PATH], inf_two[MAX_PATH]; |
| DWORD size; |
| HINF hinf; |
| BOOL ret; |
| |
| lstrcpyA(inf_filename, CURR_DIR); |
| lstrcatA(inf_filename, "\\"); |
| lstrcatA(inf_filename, "test.inf"); |
| |
| /* try an invalid inf handle */ |
| size = 0xdeadbeef; |
| SetLastError(0xbeefcafe); |
| ret = pSetupGetInfInformationA(NULL, INFINFO_INF_SPEC_IS_HINF, NULL, 0, &size); |
| ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); |
| ok(GetLastError() == ERROR_INVALID_HANDLE, |
| "Expected ERROR_INVALID_HANDLE, got %d\n", GetLastError()); |
| ok(size == 0xdeadbeef, "Expected size to remain unchanged\n"); |
| |
| /* try an invalid inf filename */ |
| size = 0xdeadbeef; |
| SetLastError(0xbeefcafe); |
| ret = pSetupGetInfInformationA(NULL, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size); |
| ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); |
| 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); |
| |
| /* try an invalid search flag */ |
| size = 0xdeadbeef; |
| SetLastError(0xbeefcafe); |
| ret = pSetupGetInfInformationA(inf_filename, -1, NULL, 0, &size); |
| ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); |
| ok(GetLastError() == ERROR_INVALID_PARAMETER, |
| "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError()); |
| ok(size == 0xdeadbeef, "Expected size to remain unchanged\n"); |
| |
| /* try a nonexistent inf file */ |
| size = 0xdeadbeef; |
| SetLastError(0xbeefcafe); |
| ret = pSetupGetInfInformationA("idontexist", INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size); |
| ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); |
| ok(GetLastError() == ERROR_FILE_NOT_FOUND, |
| "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError()); |
| ok(size == 0xdeadbeef, "Expected size to remain unchanged\n"); |
| |
| /* successfully open the inf file */ |
| size = 0xdeadbeef; |
| ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size); |
| ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n"); |
| ok(size != 0xdeadbeef, "Expected a valid size on return\n"); |
| |
| /* set ReturnBuffer to NULL and ReturnBufferSize to non-zero value 'size' */ |
| SetLastError(0xbeefcafe); |
| ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size, &size); |
| ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); |
| ok(GetLastError() == ERROR_INVALID_PARAMETER, |
| "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError()); |
| |
| /* set ReturnBuffer to NULL and ReturnBufferSize to non-zero value 'size-1' */ |
| ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size-1, &size); |
| ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n"); |
| |
| /* some tests for behaviour with a NULL RequiredSize pointer */ |
| ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, NULL); |
| ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n"); |
| ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size - 1, NULL); |
| ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n"); |
| ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size, NULL); |
| ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); |
| |
| info = HeapAlloc(GetProcessHeap(), 0, size); |
| |
| /* try valid ReturnBuffer but too small size */ |
| SetLastError(0xbeefcafe); |
| ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, info, size - 1, &size); |
| ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); |
| ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, |
| "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError()); |
| |
| /* successfully get the inf information */ |
| ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, info, size, &size); |
| ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n"); |
| ok(check_info_filename(info, inf_filename), "Expected returned filename to be equal\n"); |
| |
| HeapFree(GetProcessHeap(), 0, info); |
| |
| /* try the INFINFO_INF_SPEC_IS_HINF search flag */ |
| hinf = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL); |
| info = alloc_inf_info(hinf, INFINFO_INF_SPEC_IS_HINF, &size); |
| ret = pSetupGetInfInformationA(hinf, INFINFO_INF_SPEC_IS_HINF, info, size, &size); |
| ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n"); |
| ok(check_info_filename(info, inf_filename), "Expected returned filename to be equal\n"); |
| pSetupCloseInfFile(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); |
| |
| HeapFree(GetProcessHeap(), 0, info); |
| info = alloc_inf_info("test.inf", INFINFO_DEFAULT_SEARCH, &size); |
| |
| /* test the INFINFO_DEFAULT_SEARCH search flag */ |
| ret = pSetupGetInfInformationA("test.inf", INFINFO_DEFAULT_SEARCH, info, size, &size); |
| ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n"); |
| ok(check_info_filename(info, inf_one), "Expected returned filename to be equal\n"); |
| |
| HeapFree(GetProcessHeap(), 0, info); |
| info = alloc_inf_info("test.inf", INFINFO_REVERSE_DEFAULT_SEARCH, &size); |
| |
| /* test the INFINFO_REVERSE_DEFAULT_SEARCH search flag */ |
| ret = pSetupGetInfInformationA("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"); |
| |
| DeleteFileA(inf_filename); |
| DeleteFileA(inf_one); |
| DeleteFileA(inf_two); |
| } |
| |
| static void test_SetupGetSourceFileLocation(void) |
| { |
| char buffer[MAX_PATH] = "not empty", inf_filename[MAX_PATH]; |
| UINT source_id; |
| DWORD required, error; |
| HINF hinf; |
| BOOL ret; |
| |
| lstrcpyA(inf_filename, CURR_DIR); |
| lstrcatA(inf_filename, "\\"); |
| lstrcatA(inf_filename, "test.inf"); |
| |
| create_inf_file(inf_filename); |
| |
| hinf = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL); |
| ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n"); |
| |
| required = 0; |
| source_id = 0; |
| |
| ret = pSetupGetSourceFileLocationA(hinf, NULL, "lanconf.exe", &source_id, buffer, sizeof(buffer), &required); |
| ok(ret, "SetupGetSourceFileLocation failed\n"); |
| |
| ok(required == 1, "unexpected required size: %d\n", required); |
| ok(source_id == 2, "unexpected source id: %d\n", source_id); |
| ok(!lstrcmpA("", buffer), "unexpected result string: %s\n", buffer); |
| |
| pSetupCloseInfFile(hinf); |
| DeleteFileA(inf_filename); |
| |
| create_inf_file2(inf_filename); |
| |
| SetLastError(0xdeadbeef); |
| hinf = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL); |
| error = GetLastError(); |
| ok(hinf == INVALID_HANDLE_VALUE, "could open inf file\n"); |
| ok(error == ERROR_WRONG_INF_STYLE, "got wrong error: %d\n", error); |
| |
| hinf = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_OLDNT, NULL); |
| ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n"); |
| |
| ret = pSetupGetSourceFileLocationA(hinf, NULL, "", &source_id, buffer, sizeof(buffer), &required); |
| ok(!ret, "SetupGetSourceFileLocation succeeded\n"); |
| |
| pSetupCloseInfFile(hinf); |
| DeleteFileA(inf_filename); |
| } |
| |
| static void test_SetupGetSourceInfo(void) |
| { |
| char buffer[MAX_PATH], inf_filename[MAX_PATH]; |
| DWORD required; |
| HINF hinf; |
| BOOL ret; |
| |
| lstrcpyA(inf_filename, CURR_DIR); |
| lstrcatA(inf_filename, "\\"); |
| lstrcatA(inf_filename, "test.inf"); |
| |
| create_inf_file(inf_filename); |
| |
| hinf = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL); |
| ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n"); |
| |
| required = 0; |
| |
| ret = pSetupGetSourceInfoA(hinf, 2, SRCINFO_PATH, buffer, sizeof(buffer), &required); |
| ok(ret, "SetupGetSourceInfoA failed\n"); |
| |
| ok(required == 1, "unexpected required size: %d\n", required); |
| ok(!lstrcmpA("", buffer), "unexpected result string: %s\n", buffer); |
| |
| required = 0; |
| buffer[0] = 0; |
| |
| ret = pSetupGetSourceInfoA(hinf, 2, SRCINFO_TAGFILE, buffer, sizeof(buffer), &required); |
| ok(ret, "SetupGetSourceInfoA failed\n"); |
| |
| ok(required == 28, "unexpected required size: %d\n", required); |
| ok(!lstrcmpA("LANCOM\\LANtools\\lanconf.cab", buffer), "unexpected result string: %s\n", buffer); |
| |
| required = 0; |
| buffer[0] = 0; |
| |
| ret = pSetupGetSourceInfoA(hinf, 2, SRCINFO_DESCRIPTION, buffer, sizeof(buffer), &required); |
| ok(ret, "SetupGetSourceInfoA failed\n"); |
| |
| ok(required == 19, "unexpected required size: %d\n", required); |
| ok(!lstrcmpA("LANCOM Software CD", buffer), "unexpected result string: %s\n", buffer); |
| |
| pSetupCloseInfFile(hinf); |
| DeleteFileA(inf_filename); |
| } |
| |
| static void test_SetupGetTargetPath(void) |
| { |
| char buffer[MAX_PATH], inf_filename[MAX_PATH]; |
| DWORD required; |
| HINF hinf; |
| INFCONTEXT ctx; |
| BOOL ret; |
| |
| lstrcpyA(inf_filename, CURR_DIR); |
| lstrcatA(inf_filename, "\\"); |
| lstrcatA(inf_filename, "test.inf"); |
| |
| create_inf_file(inf_filename); |
| |
| hinf = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL); |
| ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n"); |
| |
| ctx.Inf = hinf; |
| ctx.CurrentInf = hinf; |
| ctx.Section = 7; |
| ctx.Line = 0; |
| |
| required = 0; |
| |
| ret = pSetupGetTargetPathA(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); |
| |
| pSetupCloseInfFile(hinf); |
| DeleteFileA(inf_filename); |
| } |
| |
| START_TEST(query) |
| { |
| init_function_pointers(); |
| get_directories(); |
| |
| test_SetupGetInfInformation(); |
| test_SetupGetSourceFileLocation(); |
| test_SetupGetSourceInfo(); |
| test_SetupGetTargetPath(); |
| } |