mirror of
https://github.com/reactos/reactos.git
synced 2024-08-29 22:58:34 +00:00
402 lines
14 KiB
C
402 lines
14 KiB
C
/*
|
|
* 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"
|
|
|
|
CHAR CURR_DIR[MAX_PATH];
|
|
CHAR WIN_DIR[MAX_PATH];
|
|
|
|
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 (!SetupQueryInfFileInformationA(info, 0, NULL, 0, &size))
|
|
return FALSE;
|
|
|
|
filename = HeapAlloc(GetProcessHeap(), 0, size);
|
|
if (!filename)
|
|
return FALSE;
|
|
|
|
SetupQueryInfFileInformationA(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 = SetupGetInfInformationA(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 = SetupGetInfInformationA(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 = SetupGetInfInformationA(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 = SetupGetInfInformationA(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 = SetupGetInfInformationA("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 = SetupGetInfInformationA(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 = SetupGetInfInformationA(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 = SetupGetInfInformationA(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 = SetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, NULL);
|
|
ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
|
|
ret = SetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size - 1, NULL);
|
|
ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
|
|
ret = SetupGetInfInformationA(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 = SetupGetInfInformationA(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 = SetupGetInfInformationA(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 = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
|
|
info = alloc_inf_info(hinf, INFINFO_INF_SPEC_IS_HINF, &size);
|
|
ret = SetupGetInfInformationA(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");
|
|
SetupCloseInfFile(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 = SetupGetInfInformationA("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 = SetupGetInfInformationA("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 = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
|
|
ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
|
|
|
|
required = 0;
|
|
source_id = 0;
|
|
|
|
ret = SetupGetSourceFileLocationA(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);
|
|
|
|
SetupCloseInfFile(hinf);
|
|
DeleteFileA(inf_filename);
|
|
|
|
create_inf_file2(inf_filename);
|
|
|
|
SetLastError(0xdeadbeef);
|
|
hinf = SetupOpenInfFileA(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 = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_OLDNT, NULL);
|
|
ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
|
|
|
|
ret = SetupGetSourceFileLocationA(hinf, NULL, "", &source_id, buffer, sizeof(buffer), &required);
|
|
ok(!ret, "SetupGetSourceFileLocation succeeded\n");
|
|
|
|
SetupCloseInfFile(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 = SetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
|
|
ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");
|
|
|
|
required = 0;
|
|
|
|
ret = SetupGetSourceInfoA(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 = SetupGetSourceInfoA(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 = SetupGetSourceInfoA(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);
|
|
|
|
SetupCloseInfFile(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 = SetupOpenInfFileA(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 = SetupGetTargetPathA(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);
|
|
|
|
SetupCloseInfFile(hinf);
|
|
DeleteFileA(inf_filename);
|
|
}
|
|
|
|
START_TEST(query)
|
|
{
|
|
get_directories();
|
|
|
|
test_SetupGetInfInformation();
|
|
test_SetupGetSourceFileLocation();
|
|
test_SetupGetSourceInfo();
|
|
test_SetupGetTargetPath();
|
|
}
|