dlls/setupapi/tests/query.c - wine - Git at Google


            
            /*

            
             * 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"


            /* 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);

            
            static BOOL (WINAPI *pSetupGetSourceFileLocationA)(HINF,PINFCONTEXT,PCSTR,PUINT,PSTR,DWORD,PDWORD);

            
            static BOOL (WINAPI *pSetupGetSourceInfoA)(HINF,UINT,UINT,PSTR,DWORD,PDWORD);

            
            static BOOL (WINAPI *pSetupGetTargetPathA)(HINF,PINFCONTEXT,PCSTR,PSTR,DWORD,PDWORD);


            CHAR CURR_DIR[MAX_PATH];

            
            CHAR WIN_DIR[MAX_PATH];


            static void init_function_pointers(void)

            
            {

            
                hSetupAPI = GetModuleHandleA("setupapi.dll");


                pSetupCloseInfFile = (void *)GetProcAddress(hSetupAPI, "SetupCloseInfFile");

            
                pSetupGetInfInformationA = (void *)GetProcAddress(hSetupAPI, "SetupGetInfInformationA");

            
                pSetupOpenInfFileA = (void *)GetProcAddress(hSetupAPI, "SetupOpenInfFileA");

            
                pSetupQueryInfFileInformationA = (void *)GetProcAddress(hSetupAPI, "SetupQueryInfFileInformationA");

            
                pSetupGetSourceFileLocationA = (void *)GetProcAddress(hSetupAPI, "SetupGetSourceFileLocationA");

            
                pSetupGetSourceInfoA = (void *)GetProcAddress(hSetupAPI, "SetupGetSourceInfoA");

            
                pSetupGetTargetPathA = (void *)GetProcAddress(hSetupAPI, "SetupGetTargetPathA");

            
            }


            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 (!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(LPCSTR 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);

            
                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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA("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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, NULL);

            
                ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");

            
                ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size - 1, NULL);

            
                ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");

            
                ret = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(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 = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);

            
                info = alloc_inf_info(hinf, INFINFO_INF_SPEC_IS_HINF, &size);

            
                ret = pSetupGetInfInformationA(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");

            
                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");

            
                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");

            
                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 = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);

            
                ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");


                required = 0;

            
                source_id = 0;


                ret = pSetupGetSourceFileLocationA(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);


                pSetupCloseInfFile(hinf);

            
                DeleteFileA(inf_filename);


                create_inf_file2(inf_filename);


                SetLastError(0xdeadbeef);

            
                hinf = pSetupOpenInfFileA(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 = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_OLDNT, NULL);

            
                ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");


                ret = pSetupGetSourceFileLocationA(hinf, NULL, "", &source_id, buffer, sizeof(buffer), &required);

            
                ok(!ret, "SetupGetSourceFileLocation succeeded\n");


                pSetupCloseInfFile(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 = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);

            
                ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");


                required = 0;


                ret = pSetupGetSourceInfoA(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 = pSetupGetSourceInfoA(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 = pSetupGetSourceInfoA(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);


                pSetupCloseInfFile(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 = pSetupOpenInfFileA(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 = pSetupGetTargetPathA(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);


                pSetupCloseInfFile(hinf);

            
                DeleteFileA(inf_filename);

            
            }


            START_TEST(query)

            
            {

            
                init_function_pointers();

            
                get_directories();


                test_SetupGetInfInformation();

            
                test_SetupGetSourceFileLocation();

            
                test_SetupGetSourceInfo();

            
                test_SetupGetTargetPath();

            
            }
	/*
	* 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"

	/* 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);
	static BOOL (WINAPI *pSetupGetSourceFileLocationA)(HINF,PINFCONTEXT,PCSTR,PUINT,PSTR,DWORD,PDWORD);
	static BOOL (WINAPI *pSetupGetSourceInfoA)(HINF,UINT,UINT,PSTR,DWORD,PDWORD);
	static BOOL (WINAPI *pSetupGetTargetPathA)(HINF,PINFCONTEXT,PCSTR,PSTR,DWORD,PDWORD);

	CHAR CURR_DIR[MAX_PATH];
	CHAR WIN_DIR[MAX_PATH];

	static void init_function_pointers(void)
	{
	hSetupAPI = GetModuleHandleA("setupapi.dll");

	pSetupCloseInfFile = (void *)GetProcAddress(hSetupAPI, "SetupCloseInfFile");
	pSetupGetInfInformationA = (void *)GetProcAddress(hSetupAPI, "SetupGetInfInformationA");
	pSetupOpenInfFileA = (void *)GetProcAddress(hSetupAPI, "SetupOpenInfFileA");
	pSetupQueryInfFileInformationA = (void *)GetProcAddress(hSetupAPI, "SetupQueryInfFileInformationA");
	pSetupGetSourceFileLocationA = (void *)GetProcAddress(hSetupAPI, "SetupGetSourceFileLocationA");
	pSetupGetSourceInfoA = (void *)GetProcAddress(hSetupAPI, "SetupGetSourceInfoA");
	pSetupGetTargetPathA = (void *)GetProcAddress(hSetupAPI, "SetupGetTargetPathA");
	}

	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 (!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(LPCSTR 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);
	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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA("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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, NULL);
	ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
	ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size - 1, NULL);
	ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
	ret = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(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 = pSetupGetInfInformationA(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 = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
	info = alloc_inf_info(hinf, INFINFO_INF_SPEC_IS_HINF, &size);
	ret = pSetupGetInfInformationA(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");
	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");
	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");
	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 = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
	ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");

	required = 0;
	source_id = 0;

	ret = pSetupGetSourceFileLocationA(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);

	pSetupCloseInfFile(hinf);
	DeleteFileA(inf_filename);

	create_inf_file2(inf_filename);

	SetLastError(0xdeadbeef);
	hinf = pSetupOpenInfFileA(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 = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_OLDNT, NULL);
	ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");

	ret = pSetupGetSourceFileLocationA(hinf, NULL, "", &source_id, buffer, sizeof(buffer), &required);
	ok(!ret, "SetupGetSourceFileLocation succeeded\n");

	pSetupCloseInfFile(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 = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
	ok(hinf != INVALID_HANDLE_VALUE, "could not open inf file\n");

	required = 0;

	ret = pSetupGetSourceInfoA(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 = pSetupGetSourceInfoA(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 = pSetupGetSourceInfoA(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);

	pSetupCloseInfFile(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 = pSetupOpenInfFileA(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 = pSetupGetTargetPathA(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);

	pSetupCloseInfFile(hinf);
	DeleteFileA(inf_filename);
	}

	START_TEST(query)
	{
	init_function_pointers();
	get_directories();

	test_SetupGetInfInformation();
	test_SetupGetSourceFileLocation();
	test_SetupGetSourceInfo();
	test_SetupGetTargetPath();
	}