From ac6c9410587c7b0c3ec5614ba1c8289dd525a407 Mon Sep 17 00:00:00 2001 From: James Hawkins Date: Tue, 31 Jan 2006 12:22:48 +0100 Subject: [PATCH] setupapi: Add tests for SetupGetInfInformation. --- dlls/setupapi/tests/Makefile.in | 1 + dlls/setupapi/tests/query.c | 278 ++++++++++++++++++++++++++++++++++++++++ 2 files changed, 279 insertions(+) create mode 100644 dlls/setupapi/tests/query.c diff --git a/dlls/setupapi/tests/Makefile.in b/dlls/setupapi/tests/Makefile.in index 517d44f1645..7730f688b28 100644 --- a/dlls/setupapi/tests/Makefile.in +++ b/dlls/setupapi/tests/Makefile.in @@ -7,6 +7,7 @@ IMPORTS = setupapi kernel32 CTESTS = \ parser.c \ + query.c \ stringtable.c @MAKE_TEST_RULES@ diff --git a/dlls/setupapi/tests/query.c b/dlls/setupapi/tests/query.c new file mode 100644 index 00000000000..71dc50efd20 --- /dev/null +++ b/dlls/setupapi/tests/query.c @@ -0,0 +1,278 @@ +/* + * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +#include +#include +#include +#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); + +CHAR CURR_DIR[MAX_PATH]; +CHAR WIN_DIR[MAX_PATH]; + +static void init_function_pointers(void) +{ + hSetupAPI = LoadLibraryA("setupapi.dll"); + + if (hSetupAPI) + { + pSetupCloseInfFile = (void *)GetProcAddress(hSetupAPI, "SetupCloseInfFile"); + pSetupGetInfInformationA = (void *)GetProcAddress(hSetupAPI, "SetupGetInfInformationA"); + pSetupOpenInfFileA = (void *)GetProcAddress(hSetupAPI, "SetupOpenInfFileA"); + pSetupQueryInfFileInformationA = (void *)GetProcAddress(hSetupAPI, "SetupQueryInfFileInformationA"); + } +} + +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"); + + 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(LPSTR 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); + todo_wine + { + ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); + ok(GetLastError() == ERROR_INVALID_HANDLE, + "Expected ERROR_INVALID_HANDLE, got %ld\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); + todo_wine + { + ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); + ok(GetLastError() == ERROR_INVALID_PARAMETER, + "Expected ERROR_INVALID_PARAMETER, got %ld\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); + todo_wine + { + ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); + ok(GetLastError() == ERROR_INVALID_PARAMETER, + "Expected ERROR_INVALID_PARAMETER, got %ld\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); + todo_wine + { + ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); + ok(GetLastError() == ERROR_FILE_NOT_FOUND, + "Expected ERROR_FILE_NOT_FOUND, got %ld\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"); + todo_wine + { + ok(size != 0xdeadbeef, "Expected a valid size on return\n"); + } + + /* set ReturnBuffer to NULL and ReturnBufferSize to non-zero */ + SetLastError(0xbeefcafe); + ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size, &size); + todo_wine + { + ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); + ok(GetLastError() == ERROR_INVALID_PARAMETER, + "Expected ERROR_INVALID_PARAMETER, got %ld\n", GetLastError()); + } + + 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); + todo_wine + { + ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n"); + ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, + "Expected ERROR_INSUFFICIENT_BUFFER, got %ld\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"); + todo_wine + { + ok(check_info_filename(info, inf_filename), "Expected returned filename to be equal\n"); + } + + /* try the INFINFO_INF_SPEC_IS_HINF search flag */ + hinf = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL); + ret = pSetupGetInfInformationA(hinf, INFINFO_INF_SPEC_IS_HINF, info, size, &size); + ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n"); + todo_wine + { + 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"); + todo_wine + { + 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"); + todo_wine + { + ok(check_info_filename(info, inf_two), "Expected returned filename to be equal\n"); + } + + DeleteFileA(inf_filename); + DeleteFileA(inf_one); + DeleteFileA(inf_two); +} + +START_TEST(query) +{ + init_function_pointers(); + get_directories(); + + test_SetupGetInfInformation(); +} -- 2.11.4.GIT