| /* |
| * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| */ |
| |
| #include <stdio.h> |
| #include <windows.h> |
| #include <advpub.h> |
| #include "wine/test.h" |
| |
| /* function pointers */ |
| static HRESULT (WINAPI *pRunSetupCommand)(HWND, LPCSTR, LPCSTR, LPCSTR, LPCSTR, HANDLE*, DWORD, LPVOID); |
| |
| static BOOL init_function_pointers() |
| { |
| HMODULE hAdvPack = LoadLibraryA("advpack.dll"); |
| if (!hAdvPack) |
| return FALSE; |
| |
| pRunSetupCommand = (void *)GetProcAddress(hAdvPack, "RunSetupCommand"); |
| |
| if (!pRunSetupCommand) |
| 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 test_RunSetupCommand() |
| { |
| HRESULT hr; |
| HANDLE hexe; |
| |
| /* try an invalid cmd name */ |
| hr = pRunSetupCommand(NULL, NULL, "Install", "Dir", "Title", NULL, 0, NULL); |
| ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %ld\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 %ld\n", hr); |
| |
| /* try to run a non-existent exe */ |
| hexe = (HANDLE)0xdeadbeef; |
| hr = pRunSetupCommand(NULL, "idontexist.exe", "Install", "c:\\windows\\system32", "Title", &hexe, 0, NULL); |
| ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), |
| "Expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), got %ld\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 %ld\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", "c:\\windows\\system32", "Title", &hexe, RSC_FLAG_INF, NULL); |
| ok(is_spapi_err(hr), "Expected a setupapi error, got %ld\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", "c:\\windows\\system32", "Title", &hexe, 0, NULL); |
| ok(hr == S_ASYNCHRONOUS, "Expected S_ASYNCHRONOUS, got %ld\n", hr); |
| ok(hexe != NULL, "Expected hexe to be non-NULL\n"); |
| ok(TerminateProcess(hexe, 0), "Expected TerminateProcess to succeed\n"); |
| } |
| |
| START_TEST(install) |
| { |
| if (!init_function_pointers()) |
| return; |
| |
| test_RunSetupCommand(); |
| } |