mirror of
https://github.com/reactos/reactos.git
synced 2024-08-29 22:58:34 +00:00
397 lines
14 KiB
C
397 lines
14 KiB
C
/*
|
|
* Unit test for setupapi.dll install functions
|
|
*
|
|
* Copyright 2007 Misha Koshelev
|
|
*
|
|
* 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 <stdarg.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "winnls.h"
|
|
#include "winuser.h"
|
|
#include "winreg.h"
|
|
#include "winsvc.h"
|
|
#include "setupapi.h"
|
|
|
|
#include "wine/test.h"
|
|
|
|
static const char inffile[] = "test.inf";
|
|
static char CURR_DIR[MAX_PATH];
|
|
|
|
/* Notes on InstallHinfSectionA/W:
|
|
* - InstallHinfSectionW on Win98 and InstallHinfSectionA on WinXP seem to be stubs - they do not do anything
|
|
* and simply return without displaying any error message or setting last error. We test whether
|
|
* InstallHinfSectionA sets last error, and if it doesn't we set it to NULL and use the W version if available.
|
|
* - These functions do not return a value and do not always set last error to ERROR_SUCCESS when installation still
|
|
* occurs (e.g., unquoted inf file with spaces, registry keys are written but last error is 6). Also, on Win98 last error
|
|
* is set to ERROR_SUCCESS even if install fails (e.g., quoted inf file with spaces, no registry keys set, MessageBox with
|
|
* "Installation Error" displayed). Thus, we must use functional tests (e.g., is registry key created) to determine whether
|
|
* or not installation occurred.
|
|
* - On installation problems, a MessageBox() is displayed and a Beep() is issued. The MessageBox() is disabled with a
|
|
* CBT hook.
|
|
*/
|
|
|
|
static void (WINAPI *pInstallHinfSectionA)(HWND, HINSTANCE, LPCSTR, INT);
|
|
static void (WINAPI *pInstallHinfSectionW)(HWND, HINSTANCE, LPCWSTR, INT);
|
|
|
|
/*
|
|
* Helpers
|
|
*/
|
|
|
|
static void create_inf_file(LPCSTR filename, const char *data)
|
|
{
|
|
DWORD res;
|
|
HANDLE handle = CreateFile(filename, GENERIC_WRITE, 0, NULL,
|
|
CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
assert(handle != INVALID_HANDLE_VALUE);
|
|
assert(WriteFile(handle, data, strlen(data), &res, NULL));
|
|
CloseHandle(handle);
|
|
}
|
|
|
|
/* CBT hook to ensure a window (e.g., MessageBox) cannot be created */
|
|
static HHOOK hhook;
|
|
static LRESULT CALLBACK cbt_hook_proc(int nCode, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
return nCode == HCBT_CREATEWND ? 1: CallNextHookEx(hhook, nCode, wParam, lParam);
|
|
}
|
|
|
|
/*
|
|
* Tests
|
|
*/
|
|
|
|
static const char *cmdline_inf = "[Version]\n"
|
|
"Signature=\"$Chicago$\"\n"
|
|
"[DefaultInstall]\n"
|
|
"AddReg=Add.Settings\n"
|
|
"[Add.Settings]\n"
|
|
"HKCU,Software\\Wine\\setupapitest,,\n";
|
|
|
|
static void run_cmdline(LPCSTR section, int mode, LPCSTR path)
|
|
{
|
|
CHAR cmdline[MAX_PATH * 2];
|
|
|
|
sprintf(cmdline, "%s %d %s", section, mode, path);
|
|
if (pInstallHinfSectionA) pInstallHinfSectionA(NULL, NULL, cmdline, 0);
|
|
else
|
|
{
|
|
WCHAR cmdlinew[MAX_PATH * 2];
|
|
MultiByteToWideChar(CP_ACP, 0, cmdline, -1, cmdlinew, MAX_PATH*2);
|
|
pInstallHinfSectionW(NULL, NULL, cmdlinew, 0);
|
|
}
|
|
}
|
|
|
|
static void ok_registry(BOOL expectsuccess)
|
|
{
|
|
LONG ret;
|
|
|
|
/* Functional tests for success of install and clean up */
|
|
ret = RegDeleteKey(HKEY_CURRENT_USER, "Software\\Wine\\setupapitest");
|
|
ok((expectsuccess && ret == ERROR_SUCCESS) ||
|
|
(!expectsuccess && ret == ERROR_FILE_NOT_FOUND),
|
|
"Expected registry key Software\\Wine\\setupapitest to %s, RegDeleteKey returned %d\n",
|
|
expectsuccess ? "exist" : "not exist",
|
|
ret);
|
|
}
|
|
|
|
/* Test command line processing */
|
|
static void test_cmdline(void)
|
|
{
|
|
static const char infwithspaces[] = "test file.inf";
|
|
char path[MAX_PATH];
|
|
|
|
create_inf_file(inffile, cmdline_inf);
|
|
sprintf(path, "%s\\%s", CURR_DIR, inffile);
|
|
run_cmdline("DefaultInstall", 128, path);
|
|
ok_registry(TRUE);
|
|
ok(DeleteFile(inffile), "Expected source inf to exist, last error was %d\n", GetLastError());
|
|
|
|
/* Test handling of spaces in path, unquoted and quoted */
|
|
create_inf_file(infwithspaces, cmdline_inf);
|
|
|
|
sprintf(path, "%s\\%s", CURR_DIR, infwithspaces);
|
|
run_cmdline("DefaultInstall", 128, path);
|
|
ok_registry(TRUE);
|
|
|
|
sprintf(path, "\"%s\\%s\"", CURR_DIR, infwithspaces);
|
|
run_cmdline("DefaultInstall", 128, path);
|
|
ok_registry(FALSE);
|
|
|
|
ok(DeleteFile(infwithspaces), "Expected source inf to exist, last error was %d\n", GetLastError());
|
|
}
|
|
|
|
static void test_install_svc_from(void)
|
|
{
|
|
char inf[2048];
|
|
char path[MAX_PATH];
|
|
HINF infhandle;
|
|
BOOL ret;
|
|
SC_HANDLE scm_handle, svc_handle;
|
|
|
|
/* Bail out if we are on win98 */
|
|
SetLastError(0xdeadbeef);
|
|
scm_handle = OpenSCManagerA(NULL, NULL, GENERIC_ALL);
|
|
|
|
if (!scm_handle && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
|
|
{
|
|
skip("OpenSCManagerA is not implemented, we are most likely on win9x\n");
|
|
return;
|
|
}
|
|
CloseServiceHandle(scm_handle);
|
|
|
|
/* Basic inf file to satisfy SetupOpenInfFileA */
|
|
strcpy(inf, "[Version]\nSignature=\"$Chicago$\"\n");
|
|
create_inf_file(inffile, inf);
|
|
sprintf(path, "%s\\%s", CURR_DIR, inffile);
|
|
infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
|
|
|
|
/* Nothing but the Version section */
|
|
SetLastError(0xdeadbeef);
|
|
ret = SetupInstallServicesFromInfSectionA(infhandle, "Winetest.Services", 0);
|
|
ok(!ret, "Expected failure\n");
|
|
ok(GetLastError() == ERROR_SECTION_NOT_FOUND,
|
|
"Expected ERROR_SECTION_NOT_FOUND, got %08x\n", GetLastError());
|
|
SetupCloseInfFile(infhandle);
|
|
DeleteFile(inffile);
|
|
|
|
/* Add the section */
|
|
strcat(inf, "[Winetest.Services]\n");
|
|
create_inf_file(inffile, inf);
|
|
infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
|
|
SetLastError(0xdeadbeef);
|
|
ret = SetupInstallServicesFromInfSectionA(infhandle, "Winetest.Services", 0);
|
|
ok(!ret, "Expected failure\n");
|
|
ok(GetLastError() == ERROR_SECTION_NOT_FOUND,
|
|
"Expected ERROR_SECTION_NOT_FOUND, got %08x\n", GetLastError());
|
|
SetupCloseInfFile(infhandle);
|
|
DeleteFile(inffile);
|
|
|
|
/* Add a reference */
|
|
strcat(inf, "AddService=Winetest,,Winetest.Service\n");
|
|
create_inf_file(inffile, inf);
|
|
infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
|
|
SetLastError(0xdeadbeef);
|
|
ret = SetupInstallServicesFromInfSectionA(infhandle, "Winetest.Services", 0);
|
|
ok(!ret, "Expected failure\n");
|
|
ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
|
|
"Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", GetLastError());
|
|
SetupCloseInfFile(infhandle);
|
|
DeleteFile(inffile);
|
|
|
|
/* Add the section */
|
|
strcat(inf, "[Winetest.Service]\n");
|
|
create_inf_file(inffile, inf);
|
|
infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
|
|
SetLastError(0xdeadbeef);
|
|
ret = SetupInstallServicesFromInfSectionA(infhandle, "Winetest.Services", 0);
|
|
ok(!ret, "Expected failure\n");
|
|
ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
|
|
"Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", GetLastError());
|
|
SetupCloseInfFile(infhandle);
|
|
DeleteFile(inffile);
|
|
|
|
/* Just the ServiceBinary */
|
|
strcat(inf, "ServiceBinary=%12%\\winetest.sys\n");
|
|
create_inf_file(inffile, inf);
|
|
infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
|
|
SetLastError(0xdeadbeef);
|
|
ret = SetupInstallServicesFromInfSectionA(infhandle, "Winetest.Services", 0);
|
|
ok(!ret, "Expected failure\n");
|
|
ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
|
|
"Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", GetLastError());
|
|
SetupCloseInfFile(infhandle);
|
|
DeleteFile(inffile);
|
|
|
|
/* Add the ServiceType */
|
|
strcat(inf, "ServiceType=1\n");
|
|
create_inf_file(inffile, inf);
|
|
infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
|
|
SetLastError(0xdeadbeef);
|
|
ret = SetupInstallServicesFromInfSectionA(infhandle, "Winetest.Services", 0);
|
|
ok(!ret, "Expected failure\n");
|
|
ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
|
|
"Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", GetLastError());
|
|
SetupCloseInfFile(infhandle);
|
|
DeleteFile(inffile);
|
|
|
|
/* Add the StartType */
|
|
strcat(inf, "StartType=4\n");
|
|
create_inf_file(inffile, inf);
|
|
infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
|
|
SetLastError(0xdeadbeef);
|
|
ret = SetupInstallServicesFromInfSectionA(infhandle, "Winetest.Services", 0);
|
|
ok(!ret, "Expected failure\n");
|
|
ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
|
|
"Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", GetLastError());
|
|
SetupCloseInfFile(infhandle);
|
|
DeleteFile(inffile);
|
|
|
|
/* This should be it, the minimal entries to install a service */
|
|
strcat(inf, "ErrorControl=1");
|
|
create_inf_file(inffile, inf);
|
|
infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
|
|
SetLastError(0xdeadbeef);
|
|
ret = SetupInstallServicesFromInfSectionA(infhandle, "Winetest.Services", 0);
|
|
ok(ret, "Expected success\n");
|
|
ok(GetLastError() == ERROR_SUCCESS,
|
|
"Expected ERROR_SUCCESS, got %08x\n", GetLastError());
|
|
SetupCloseInfFile(infhandle);
|
|
DeleteFile(inffile);
|
|
|
|
scm_handle = OpenSCManagerA(NULL, NULL, GENERIC_ALL);
|
|
|
|
/* Open the service to see if it's really there */
|
|
svc_handle = OpenServiceA(scm_handle, "Winetest", DELETE);
|
|
ok(svc_handle != NULL, "Service was not created\n");
|
|
|
|
SetLastError(0xdeadbeef);
|
|
ret = DeleteService(svc_handle);
|
|
ok(ret, "Service could not be deleted : %d\n", GetLastError());
|
|
|
|
CloseServiceHandle(svc_handle);
|
|
CloseServiceHandle(scm_handle);
|
|
|
|
/* TODO: Test the Flags */
|
|
}
|
|
|
|
static void test_driver_install(void)
|
|
{
|
|
HANDLE handle;
|
|
SC_HANDLE scm_handle, svc_handle;
|
|
BOOL ret;
|
|
char path[MAX_PATH], windir[MAX_PATH], driver[MAX_PATH];
|
|
DWORD attrs;
|
|
/* Minimal stuff needed */
|
|
static const char *inf =
|
|
"[Version]\n"
|
|
"Signature=\"$Chicago$\"\n"
|
|
"[DestinationDirs]\n"
|
|
"Winetest.DriverFiles=12\n"
|
|
"[DefaultInstall]\n"
|
|
"CopyFiles=Winetest.DriverFiles\n"
|
|
"[DefaultInstall.Services]\n"
|
|
"AddService=Winetest,,Winetest.Service\n"
|
|
"[Winetest.Service]\n"
|
|
"ServiceBinary=%12%\\winetest.sys\n"
|
|
"ServiceType=1\n"
|
|
"StartType=4\n"
|
|
"ErrorControl=1\n"
|
|
"[Winetest.DriverFiles]\n"
|
|
"winetest.sys";
|
|
|
|
/* Bail out if we are on win98 */
|
|
SetLastError(0xdeadbeef);
|
|
scm_handle = OpenSCManagerA(NULL, NULL, GENERIC_ALL);
|
|
|
|
if (!scm_handle && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
|
|
{
|
|
skip("OpenSCManagerA is not implemented, we are most likely on win9x\n");
|
|
return;
|
|
}
|
|
CloseServiceHandle(scm_handle);
|
|
|
|
/* Place where we expect the driver to be installed */
|
|
GetWindowsDirectoryA(windir, MAX_PATH);
|
|
lstrcpyA(driver, windir);
|
|
lstrcatA(driver, "\\system32\\drivers\\winetest.sys");
|
|
|
|
/* Create a dummy driver file */
|
|
handle = CreateFileA("winetest.sys", GENERIC_WRITE, 0, NULL,
|
|
CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
CloseHandle(handle);
|
|
|
|
create_inf_file(inffile, inf);
|
|
sprintf(path, "%s\\%s", CURR_DIR, inffile);
|
|
run_cmdline("DefaultInstall", 128, path);
|
|
|
|
/* Driver should have been installed */
|
|
attrs = GetFileAttributes(driver);
|
|
ok(attrs != INVALID_FILE_ATTRIBUTES, "Expected driver to exist\n");
|
|
|
|
scm_handle = OpenSCManagerA(NULL, NULL, GENERIC_ALL);
|
|
|
|
/* Open the service to see if it's really there */
|
|
svc_handle = OpenServiceA(scm_handle, "Winetest", DELETE);
|
|
ok(svc_handle != NULL, "Service was not created\n");
|
|
|
|
SetLastError(0xdeadbeef);
|
|
ret = DeleteService(svc_handle);
|
|
ok(ret, "Service could not be deleted : %d\n", GetLastError());
|
|
|
|
CloseServiceHandle(svc_handle);
|
|
CloseServiceHandle(scm_handle);
|
|
|
|
/* File cleanup */
|
|
DeleteFile(inffile);
|
|
DeleteFile("winetest.sys");
|
|
DeleteFile(driver);
|
|
}
|
|
|
|
START_TEST(install)
|
|
{
|
|
HMODULE hsetupapi = GetModuleHandle("setupapi.dll");
|
|
char temp_path[MAX_PATH], prev_path[MAX_PATH];
|
|
DWORD len;
|
|
|
|
GetCurrentDirectory(MAX_PATH, prev_path);
|
|
GetTempPath(MAX_PATH, temp_path);
|
|
SetCurrentDirectory(temp_path);
|
|
|
|
strcpy(CURR_DIR, temp_path);
|
|
len = strlen(CURR_DIR);
|
|
if(len && (CURR_DIR[len - 1] == '\\'))
|
|
CURR_DIR[len - 1] = 0;
|
|
|
|
pInstallHinfSectionA = (void *)GetProcAddress(hsetupapi, "InstallHinfSectionA");
|
|
pInstallHinfSectionW = (void *)GetProcAddress(hsetupapi, "InstallHinfSectionW");
|
|
if (pInstallHinfSectionA)
|
|
{
|
|
/* Check if pInstallHinfSectionA sets last error or is a stub (as on WinXP) */
|
|
static const char *minimal_inf = "[Version]\nSignature=\"$Chicago$\"\n";
|
|
char cmdline[MAX_PATH*2];
|
|
create_inf_file(inffile, minimal_inf);
|
|
sprintf(cmdline, "DefaultInstall 128 %s\\%s", CURR_DIR, inffile);
|
|
SetLastError(0xdeadbeef);
|
|
pInstallHinfSectionA(NULL, NULL, cmdline, 0);
|
|
if (GetLastError() == 0xdeadbeef)
|
|
{
|
|
skip("InstallHinfSectionA is broken (stub)\n");
|
|
pInstallHinfSectionA = NULL;
|
|
}
|
|
ok(DeleteFile(inffile), "Expected source inf to exist, last error was %d\n", GetLastError());
|
|
}
|
|
if (!pInstallHinfSectionW && !pInstallHinfSectionA)
|
|
skip("InstallHinfSectionA and InstallHinfSectionW are not available\n");
|
|
else
|
|
{
|
|
/* Set CBT hook to disallow MessageBox creation in current thread */
|
|
hhook = SetWindowsHookExA(WH_CBT, cbt_hook_proc, 0, GetCurrentThreadId());
|
|
assert(hhook != 0);
|
|
|
|
test_cmdline();
|
|
test_install_svc_from();
|
|
test_driver_install();
|
|
|
|
UnhookWindowsHookEx(hhook);
|
|
}
|
|
|
|
SetCurrentDirectory(prev_path);
|
|
}
|