| /* |
| * Unit tests for advpack.dll install 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 <advpub.h> |
| #include "wine/test.h" |
| |
| static HMODULE hAdvPack; |
| /* function pointers */ |
| static HRESULT (WINAPI *pRunSetupCommand)(HWND, LPCSTR, LPCSTR, LPCSTR, LPCSTR, HANDLE*, DWORD, LPVOID); |
| static HRESULT (WINAPI *pLaunchINFSection)(HWND, HINSTANCE, LPSTR, INT); |
| static HRESULT (WINAPI *pLaunchINFSectionEx)(HWND, HINSTANCE, LPSTR, INT); |
| |
| static char CURR_DIR[MAX_PATH]; |
| |
| static BOOL init_function_pointers(void) |
| { |
| hAdvPack = LoadLibraryA("advpack.dll"); |
| if (!hAdvPack) |
| return FALSE; |
| |
| pRunSetupCommand = (void *)GetProcAddress(hAdvPack, "RunSetupCommand"); |
| pLaunchINFSection = (void *)GetProcAddress(hAdvPack, "LaunchINFSection"); |
| pLaunchINFSectionEx = (void *)GetProcAddress(hAdvPack, "LaunchINFSectionEx"); |
| |
| if (!pRunSetupCommand || !pLaunchINFSection || !pLaunchINFSectionEx) |
| return FALSE; |
| |
| return TRUE; |
| } |
| |
| static BOOL is_spapi_err(DWORD err) |
| { |
| const DWORD SPAPI_PREFIX = 0x800F0000L; |
| const DWORD SPAPI_MASK = 0xFFFF0000L; |
| |
| return (((err & SPAPI_MASK) ^ SPAPI_PREFIX) == 0); |
| } |
| |
| static void create_inf_file(LPCSTR filename) |
| { |
| DWORD dwNumberOfBytesWritten; |
| HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL, |
| CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); |
| |
| static const char data[] = |
| "[Version]\n" |
| "Signature=\"$Chicago$\"\n" |
| "AdvancedINF=2.5\n" |
| "[DefaultInstall]\n" |
| "CheckAdminRights=1\n"; |
| |
| WriteFile(hf, data, sizeof(data) - 1, &dwNumberOfBytesWritten, NULL); |
| CloseHandle(hf); |
| } |
| |
| static void test_RunSetupCommand(void) |
| { |
| HRESULT hr; |
| HANDLE hexe; |
| char path[MAX_PATH]; |
| char dir[MAX_PATH]; |
| char systemdir[MAX_PATH]; |
| |
| GetSystemDirectoryA(systemdir, sizeof(systemdir)); |
| |
| /* try an invalid cmd name */ |
| hr = pRunSetupCommand(NULL, NULL, "Install", "Dir", "Title", NULL, 0, NULL); |
| ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %d\n", hr); |
| |
| /* try an invalid directory */ |
| hr = pRunSetupCommand(NULL, "winver.exe", "Install", NULL, "Title", NULL, 0, NULL); |
| ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %d\n", hr); |
| |
| /* try to run a nonexistent exe */ |
| hexe = (HANDLE)0xdeadbeef; |
| hr = pRunSetupCommand(NULL, "idontexist.exe", "Install", systemdir, "Title", &hexe, 0, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), |
| "Expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got %d\n", hr); |
| ok(hexe == NULL, "Expcted hexe to be NULL\n"); |
| ok(!TerminateProcess(hexe, 0), "Expected TerminateProcess to fail\n"); |
| |
| /* try a bad directory */ |
| hexe = (HANDLE)0xdeadbeef; |
| hr = pRunSetupCommand(NULL, "winver.exe", "Install", "non\\existent\\directory", "Title", &hexe, 0, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_DIRECTORY), |
| "Expected HRESULT_FROM_WIN32(ERROR_DIRECTORY), got %d\n", hr); |
| ok(hexe == NULL, "Expcted hexe to be NULL\n"); |
| ok(!TerminateProcess(hexe, 0), "Expected TerminateProcess to fail\n"); |
| |
| /* try to run an exe with the RSC_FLAG_INF flag */ |
| hexe = (HANDLE)0xdeadbeef; |
| hr = pRunSetupCommand(NULL, "winver.exe", "Install", systemdir, "Title", &hexe, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(is_spapi_err(hr) || |
| hr == E_FAIL, /* win9x */ |
| "Expected a setupapi error or E_FAIL, got %d\n", hr); |
| ok(hexe == (HANDLE)0xdeadbeef, "Expected hexe to be 0xdeadbeef\n"); |
| ok(!TerminateProcess(hexe, 0), "Expected TerminateProcess to fail\n"); |
| |
| /* run winver.exe */ |
| hexe = (HANDLE)0xdeadbeef; |
| hr = pRunSetupCommand(NULL, "winver.exe", "Install", systemdir, "Title", &hexe, 0, NULL); |
| ok(hr == S_ASYNCHRONOUS, "Expected S_ASYNCHRONOUS, got %d\n", hr); |
| ok(hexe != NULL, "Expected hexe to be non-NULL\n"); |
| ok(TerminateProcess(hexe, 0), "Expected TerminateProcess to succeed\n"); |
| |
| CreateDirectoryA("one", NULL); |
| create_inf_file("one\\test.inf"); |
| |
| /* try a full path to the INF, with working dir provided */ |
| lstrcpy(path, CURR_DIR); |
| lstrcat(path, "\\one\\test.inf"); |
| lstrcpy(dir, CURR_DIR); |
| lstrcat(dir, "\\one"); |
| hr = pRunSetupCommand(NULL, path, "DefaultInstall", dir, "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", hr); |
| |
| /* try a full path to the INF, NULL working dir */ |
| hr = pRunSetupCommand(NULL, path, "DefaultInstall", NULL, "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), |
| "Expected HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got %d\n", hr); |
| |
| /* try a full path to the INF, empty working dir */ |
| hr = pRunSetupCommand(NULL, path, "DefaultInstall", "", "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", hr); |
| |
| /* try a relative path to the INF, with working dir provided */ |
| hr = pRunSetupCommand(NULL, "one\\test.inf", "DefaultInstall", dir, "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == ERROR_SUCCESS || |
| hr == E_FAIL, /* win9x */ |
| "Expected ERROR_SUCCESS, got %d\n", hr); |
| |
| /* try a relative path to the INF, NULL working dir */ |
| hr = pRunSetupCommand(NULL, "one\\test.inf", "DefaultInstall", NULL, "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), |
| "Expected HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got %d\n", hr); |
| |
| /* try a relative path to the INF, empty working dir */ |
| hr = pRunSetupCommand(NULL, "one\\test.inf", "DefaultInstall", "", "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == ERROR_SUCCESS || |
| hr == E_FAIL, /* win9x */ |
| "Expected ERROR_SUCCESS or E_FAIL, got %d\n", hr); |
| |
| /* try only the INF filename, with working dir provided */ |
| hr = pRunSetupCommand(NULL, "test.inf", "DefaultInstall", dir, "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) || |
| hr == E_FAIL, /* win9x */ |
| "Expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) or E_FAIL, got %d\n", hr); |
| |
| /* try only the INF filename, NULL working dir */ |
| hr = pRunSetupCommand(NULL, "test.inf", "DefaultInstall", NULL, "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), |
| "Expected HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got %d\n", hr); |
| |
| /* try only the INF filename, empty working dir */ |
| hr = pRunSetupCommand(NULL, "test.inf", "DefaultInstall", "", "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) || |
| hr == E_FAIL, /* win9x */ |
| "Expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) or E_FAIL, got %d\n", hr); |
| |
| DeleteFileA("one\\test.inf"); |
| RemoveDirectoryA("one"); |
| |
| create_inf_file("test.inf"); |
| |
| /* try INF file in the current directory, working directory provided */ |
| hr = pRunSetupCommand(NULL, "test.inf", "DefaultInstall", CURR_DIR, "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) || |
| hr == E_FAIL, /* win9x */ |
| "Expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) or E_FAIL, got %d\n", hr); |
| |
| /* try INF file in the current directory, NULL working directory */ |
| hr = pRunSetupCommand(NULL, "test.inf", "DefaultInstall", NULL, "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), |
| "Expected HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER), got %d\n", hr); |
| |
| /* try INF file in the current directory, empty working directory */ |
| hr = pRunSetupCommand(NULL, "test.inf", "DefaultInstall", CURR_DIR, "Title", NULL, RSC_FLAG_INF | RSC_FLAG_QUIET, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) || |
| hr == E_FAIL, /* win9x */ |
| "Expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) or E_FAIL, got %d\n", hr); |
| } |
| |
| static void test_LaunchINFSection(void) |
| { |
| HRESULT hr; |
| char cmdline[MAX_PATH]; |
| static char file[] = "test.inf,DefaultInstall,4,0"; |
| |
| /* The 'No UI' flag seems to have no effect whatsoever on Windows. |
| * So only do this test in interactive mode. |
| */ |
| if (winetest_interactive) |
| { |
| /* try an invalid cmdline */ |
| hr = pLaunchINFSection(NULL, NULL, NULL, 0); |
| ok(hr == 1, "Expected 1, got %d\n", hr); |
| } |
| |
| CreateDirectoryA("one", NULL); |
| create_inf_file("one\\test.inf"); |
| |
| /* try a full path to the INF */ |
| lstrcpy(cmdline, CURR_DIR); |
| lstrcat(cmdline, "\\"); |
| lstrcat(cmdline, "one\\test.inf,DefaultInstall,,4"); |
| hr = pLaunchINFSection(NULL, NULL, cmdline, 0); |
| ok(hr == 0, "Expected 0, got %d\n", hr); |
| |
| DeleteFileA("one\\test.inf"); |
| RemoveDirectoryA("one"); |
| |
| create_inf_file("test.inf"); |
| |
| /* try just the INF filename */ |
| hr = pLaunchINFSection(NULL, NULL, file, 0); |
| ok(hr == 0, "Expected 0, got %d\n", hr); |
| |
| DeleteFileA("test.inf"); |
| } |
| |
| static void test_LaunchINFSectionEx(void) |
| { |
| HRESULT hr; |
| char cmdline[MAX_PATH]; |
| |
| create_inf_file("test.inf"); |
| |
| /* try an invalid CAB filename with an absolute INF name */ |
| lstrcpy(cmdline, CURR_DIR); |
| lstrcat(cmdline, "\\"); |
| lstrcat(cmdline, "test.inf,DefaultInstall,c:imacab.cab,4"); |
| hr = pLaunchINFSectionEx(NULL, NULL, cmdline, 0); |
| ok(hr == 0, "Expected 0, got %d\n", hr); |
| |
| /* The 'No UI' flag seems to have no effect whatsoever on Windows. |
| * So only do this test in interactive mode. |
| */ |
| if (winetest_interactive) |
| { |
| /* try an invalid CAB filename with a relative INF name */ |
| lstrcpy(cmdline, "test.inf,DefaultInstall,c:imacab.cab,4"); |
| hr = pLaunchINFSectionEx(NULL, NULL, cmdline, 0); |
| ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %d\n", hr); |
| } |
| |
| DeleteFileA("test.inf"); |
| } |
| |
| START_TEST(install) |
| { |
| DWORD len; |
| char temp_path[MAX_PATH], prev_path[MAX_PATH]; |
| |
| if (!init_function_pointers()) |
| return; |
| |
| GetCurrentDirectoryA(MAX_PATH, prev_path); |
| GetTempPath(MAX_PATH, temp_path); |
| SetCurrentDirectoryA(temp_path); |
| |
| lstrcpyA(CURR_DIR, temp_path); |
| len = lstrlenA(CURR_DIR); |
| |
| if(len && (CURR_DIR[len - 1] == '\\')) |
| CURR_DIR[len - 1] = 0; |
| |
| test_RunSetupCommand(); |
| test_LaunchINFSection(); |
| test_LaunchINFSectionEx(); |
| |
| FreeLibrary(hAdvPack); |
| SetCurrentDirectoryA(prev_path); |
| } |