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"

 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 const char inf_data1[] =
     "[Version]\n"
     "Signature=\"$Chicago$\"\n"
     "AdvancedINF=2.5\n"
     "[SourceDisksNames]\n"
     "2 = %SrcDiskName%, LANCOM\\LANtools\\lanconf.cab\n"
     "[SourceDisksFiles]\n"
     "lanconf.exe = 2\n"
     "[DestinationDirs]\n"
     "DefaultDestDir = 24, %DefaultDest%\n"
     "[Strings]\n"
     "LangDir = english\n"
     "DefaultDest = LANCOM\n"
     "SrcDiskName = \"LANCOM Software CD\"\n";

 static const char inf_data2[] =
     "[SourceFileInfo]\n"
     "sp1qfe\\bitsinst.exe=250B3702C7CCD7C2F9E4DAA1555C933E,000600060A28062C,27136,SP1QFE\n"
     "sp1qfe\\bitsprx2.dll=4EBEA67F4BB4EB402E725CA7CA2857AE,000600060A280621,7680,SP1QFE\n"
     "sp1qfe\\bitsprx3.dll=C788A1D9330DA011EF25E95D3BC7BDE5,000600060A280621,7168,SP1QFE\n"
     "sp1qfe\\qmgr.dll=696AC82FB290A03F205901442E0E9589,000600060A280621,361984,SP1QFE\n"
     "sp1qfe\\qmgrprxy.dll=8B5848144829E1BC985EA4C3D8CA7E3F,000600060A280621,17408,SP1QFE\n"
     "sp1qfe\\winhttp.dll=3EC6F518114606CA59D4160322077437,000500010A280615,331776,SP1QFE\n"
     "sp1qfe\\xpob2res.dll=DB83156B9F496F20D1EA70E4ABEC0166,000500010A280622,158720,SP1QFE\n";

 static const char inf_data3[] =
     "[Version]\n"
     "Signature = \"$Windows NT$\"\n"
     "[DestinationDirs]\n"
     "CopyAlways.Windir.files = 10\n"
     "[CopyAlways.Windir.Files]\n"
     "WindowsCodecs.dll\n";

 static const char inf_data4[] =
     "[Version]\n"
     "Signature = \"$Windows NT$\"\n"
     "[CopyAlways.System32.Files]\n"
     "WindowsCodecs.dll\n";

 static const char inf_data5[] =
     "[Version]\n"
     "Signature = \"$Windows NT$\"\n"
     "[DestinationDirs]\n"
     "DefaultDestDir = 11\n"
     "CopyAlways.Windir.files = 10\n"
     "[CopyAlways.Windir.Files]\n"
     "WindowsCodecs.dll\n";

 static const char inf_data6[] =
     "[Version]\n"
     "Signature = \"$Windows NT$\"\n"
     "[DestinationDirs]\n"
     "DefaultDestDir = 10\n"
     "[CopyAlways.Windir.Files]\n"
     "WindowsCodecs.dll\n";

 static void create_inf_file(LPSTR filename, const char *data, DWORD size)
 {
     DWORD dwNumberOfBytesWritten;
     HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
     WriteFile(hf, data, size, &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];
     LPSTR revfile;
     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 ||
        broken(GetLastError() == ERROR_BAD_PATHNAME) || /* win95 */
        broken(GetLastError() == ERROR_FILE_NOT_FOUND) || /* win98 */
        broken(GetLastError() == ERROR_PATH_NOT_FOUND) || /* NT4 */
        broken(GetLastError() == ERROR_INVALID_NAME) || /* win2k */
        broken(GetLastError() == ERROR_GENERAL_SYNTAX), /* another win2k / winMe */
        "Expected ERROR_INVALID_HANDLE, got %d\n", GetLastError());
     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");

     /* try an invalid inf filename */
     /* do not use NULL as absolute inf filename on win9x/winMe (crash) */
     if ((GetLastError() != ERROR_BAD_PATHNAME) &&   /* win95 */
         (GetLastError() != ERROR_FILE_NOT_FOUND) &&  /* win98 */
         (GetLastError() != ERROR_GENERAL_SYNTAX))  /* winMe */
     {
         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, inf_data1, sizeof(inf_data1) - 1);

     /* 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_two, WIN_DIR);
     lstrcatA(inf_two, "\\system32\\");
     lstrcatA(inf_two, "test.inf");
     create_inf_file(inf_two, inf_data1, sizeof(inf_data1) - 1);

     HeapFree(GetProcessHeap(), 0, info);
     info = alloc_inf_info("test.inf", INFINFO_DEFAULT_SEARCH, &size);

     /* check if system32 is searched for inf */
     ret = SetupGetInfInformationA("test.inf", INFINFO_DEFAULT_SEARCH, info, size, &size);
     if (!ret && GetLastError() == ERROR_FILE_NOT_FOUND)
         revfile = inf_one; /* Vista */
     else
         revfile = inf_two;

     lstrcpyA(inf_one, WIN_DIR);
     lstrcatA(inf_one, "\\inf\\");
     lstrcatA(inf_one, "test.inf");
     create_inf_file(inf_one, inf_data1, sizeof(inf_data1) - 1);

     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: %d\n", GetLastError());
     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, revfile), "Expected returned filename to be equal\n");

     HeapFree(GetProcessHeap(), 0, info);

     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, inf_data1, sizeof(inf_data1) - 1);

     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_file(inf_filename, inf_data2, sizeof(inf_data2) - 1);

     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, inf_data1, sizeof(inf_data1) - 1);

     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];
     char destfile[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, inf_data1, sizeof(inf_data1) - 1);

     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);
     /* Retrieve the system drive from the windows directory.
      * (%SystemDrive% is not available on Win9x)
      */
     lstrcpyA(destfile, WIN_DIR);
     destfile[3] = '\0';
     lstrcatA(destfile, "LANCOM");
     ok(!lstrcmpiA(destfile, buffer), "unexpected result string: %s\n", buffer);

     SetupCloseInfFile(hinf);
     DeleteFileA(inf_filename);

     create_inf_file(inf_filename, inf_data3, sizeof(inf_data3) - 1);

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

     required = 0;

     ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.Windir.Files", buffer, sizeof(buffer), &required);
     ok(ret, "SetupGetTargetPathA failed\n");

     lstrcpyA(destfile, WIN_DIR);

     ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
     ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);

     SetupCloseInfFile(hinf);
     DeleteFileA(inf_filename);

     create_inf_file(inf_filename, inf_data4, sizeof(inf_data4) - 1);

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

     required = 0;

     ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.System32.Files", buffer, sizeof(buffer), &required);
     ok(ret, "SetupGetTargetPathA failed\n");

     GetSystemDirectoryA(destfile, MAX_PATH);

     ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
     ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);

     SetupCloseInfFile(hinf);
     DeleteFileA(inf_filename);

     create_inf_file(inf_filename, inf_data5, sizeof(inf_data5) - 1);

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

     required = 0;

     ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.Windir.Files", buffer, sizeof(buffer), &required);
     ok(ret, "SetupGetTargetPathA failed\n");

     lstrcpyA(destfile, WIN_DIR);

     ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
     ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);

     SetupCloseInfFile(hinf);
     DeleteFileA(inf_filename);

     create_inf_file(inf_filename, inf_data6, sizeof(inf_data6) - 1);

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

     required = 0;

     ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.Windir.Files", buffer, sizeof(buffer), &required);
     ok(ret, "SetupGetTargetPathA failed\n");

     lstrcpyA(destfile, WIN_DIR);

     ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
     ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);

     SetupCloseInfFile(hinf);
     DeleteFileA(inf_filename);
 }

 START_TEST(query)
 {
     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"

	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 const char inf_data1[] =
	"[Version]\n"
	"Signature=\"$Chicago$\"\n"
	"AdvancedINF=2.5\n"
	"[SourceDisksNames]\n"
	"2 = %SrcDiskName%, LANCOM\\LANtools\\lanconf.cab\n"
	"[SourceDisksFiles]\n"
	"lanconf.exe = 2\n"
	"[DestinationDirs]\n"
	"DefaultDestDir = 24, %DefaultDest%\n"
	"[Strings]\n"
	"LangDir = english\n"
	"DefaultDest = LANCOM\n"
	"SrcDiskName = \"LANCOM Software CD\"\n";

	static const char inf_data2[] =
	"[SourceFileInfo]\n"
	"sp1qfe\\bitsinst.exe=250B3702C7CCD7C2F9E4DAA1555C933E,000600060A28062C,27136,SP1QFE\n"
	"sp1qfe\\bitsprx2.dll=4EBEA67F4BB4EB402E725CA7CA2857AE,000600060A280621,7680,SP1QFE\n"
	"sp1qfe\\bitsprx3.dll=C788A1D9330DA011EF25E95D3BC7BDE5,000600060A280621,7168,SP1QFE\n"
	"sp1qfe\\qmgr.dll=696AC82FB290A03F205901442E0E9589,000600060A280621,361984,SP1QFE\n"
	"sp1qfe\\qmgrprxy.dll=8B5848144829E1BC985EA4C3D8CA7E3F,000600060A280621,17408,SP1QFE\n"
	"sp1qfe\\winhttp.dll=3EC6F518114606CA59D4160322077437,000500010A280615,331776,SP1QFE\n"
	"sp1qfe\\xpob2res.dll=DB83156B9F496F20D1EA70E4ABEC0166,000500010A280622,158720,SP1QFE\n";

	static const char inf_data3[] =
	"[Version]\n"
	"Signature = \"$Windows NT$\"\n"
	"[DestinationDirs]\n"
	"CopyAlways.Windir.files = 10\n"
	"[CopyAlways.Windir.Files]\n"
	"WindowsCodecs.dll\n";

	static const char inf_data4[] =
	"[Version]\n"
	"Signature = \"$Windows NT$\"\n"
	"[CopyAlways.System32.Files]\n"
	"WindowsCodecs.dll\n";

	static const char inf_data5[] =
	"[Version]\n"
	"Signature = \"$Windows NT$\"\n"
	"[DestinationDirs]\n"
	"DefaultDestDir = 11\n"
	"CopyAlways.Windir.files = 10\n"
	"[CopyAlways.Windir.Files]\n"
	"WindowsCodecs.dll\n";

	static const char inf_data6[] =
	"[Version]\n"
	"Signature = \"$Windows NT$\"\n"
	"[DestinationDirs]\n"
	"DefaultDestDir = 10\n"
	"[CopyAlways.Windir.Files]\n"
	"WindowsCodecs.dll\n";

	static void create_inf_file(LPSTR filename, const char *data, DWORD size)
	{
	DWORD dwNumberOfBytesWritten;
	HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
	CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
	WriteFile(hf, data, size, &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];
	LPSTR revfile;
	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 \|\|
	broken(GetLastError() == ERROR_BAD_PATHNAME) \|\| /* win95 */
	broken(GetLastError() == ERROR_FILE_NOT_FOUND) \|\| /* win98 */
	broken(GetLastError() == ERROR_PATH_NOT_FOUND) \|\| /* NT4 */
	broken(GetLastError() == ERROR_INVALID_NAME) \|\| /* win2k */
	broken(GetLastError() == ERROR_GENERAL_SYNTAX), /* another win2k / winMe */
	"Expected ERROR_INVALID_HANDLE, got %d\n", GetLastError());
	ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");

	/* try an invalid inf filename */
	/* do not use NULL as absolute inf filename on win9x/winMe (crash) */
	if ((GetLastError() != ERROR_BAD_PATHNAME) && /* win95 */
	(GetLastError() != ERROR_FILE_NOT_FOUND) && /* win98 */
	(GetLastError() != ERROR_GENERAL_SYNTAX)) /* winMe */
	{
	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, inf_data1, sizeof(inf_data1) - 1);

	/* 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_two, WIN_DIR);
	lstrcatA(inf_two, "\\system32\\");
	lstrcatA(inf_two, "test.inf");
	create_inf_file(inf_two, inf_data1, sizeof(inf_data1) - 1);

	HeapFree(GetProcessHeap(), 0, info);
	info = alloc_inf_info("test.inf", INFINFO_DEFAULT_SEARCH, &size);

	/* check if system32 is searched for inf */
	ret = SetupGetInfInformationA("test.inf", INFINFO_DEFAULT_SEARCH, info, size, &size);
	if (!ret && GetLastError() == ERROR_FILE_NOT_FOUND)
	revfile = inf_one; /* Vista */
	else
	revfile = inf_two;

	lstrcpyA(inf_one, WIN_DIR);
	lstrcatA(inf_one, "\\inf\\");
	lstrcatA(inf_one, "test.inf");
	create_inf_file(inf_one, inf_data1, sizeof(inf_data1) - 1);

	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: %d\n", GetLastError());
	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, revfile), "Expected returned filename to be equal\n");

	HeapFree(GetProcessHeap(), 0, info);

	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, inf_data1, sizeof(inf_data1) - 1);

	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_file(inf_filename, inf_data2, sizeof(inf_data2) - 1);

	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, inf_data1, sizeof(inf_data1) - 1);

	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];
	char destfile[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, inf_data1, sizeof(inf_data1) - 1);

	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);
	/* Retrieve the system drive from the windows directory.
	* (%SystemDrive% is not available on Win9x)
	*/
	lstrcpyA(destfile, WIN_DIR);
	destfile[3] = '\0';
	lstrcatA(destfile, "LANCOM");
	ok(!lstrcmpiA(destfile, buffer), "unexpected result string: %s\n", buffer);

	SetupCloseInfFile(hinf);
	DeleteFileA(inf_filename);

	create_inf_file(inf_filename, inf_data3, sizeof(inf_data3) - 1);

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

	required = 0;

	ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.Windir.Files", buffer, sizeof(buffer), &required);
	ok(ret, "SetupGetTargetPathA failed\n");

	lstrcpyA(destfile, WIN_DIR);

	ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
	ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);

	SetupCloseInfFile(hinf);
	DeleteFileA(inf_filename);

	create_inf_file(inf_filename, inf_data4, sizeof(inf_data4) - 1);

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

	required = 0;

	ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.System32.Files", buffer, sizeof(buffer), &required);
	ok(ret, "SetupGetTargetPathA failed\n");

	GetSystemDirectoryA(destfile, MAX_PATH);

	ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
	ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);

	SetupCloseInfFile(hinf);
	DeleteFileA(inf_filename);

	create_inf_file(inf_filename, inf_data5, sizeof(inf_data5) - 1);

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

	required = 0;

	ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.Windir.Files", buffer, sizeof(buffer), &required);
	ok(ret, "SetupGetTargetPathA failed\n");

	lstrcpyA(destfile, WIN_DIR);

	ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
	ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);

	SetupCloseInfFile(hinf);
	DeleteFileA(inf_filename);

	create_inf_file(inf_filename, inf_data6, sizeof(inf_data6) - 1);

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

	required = 0;

	ret = SetupGetTargetPathA(hinf, NULL, "CopyAlways.Windir.Files", buffer, sizeof(buffer), &required);
	ok(ret, "SetupGetTargetPathA failed\n");

	lstrcpyA(destfile, WIN_DIR);

	ok(required == lstrlenA(destfile) + 1, "unexpected required size: %d\n", required);
	ok(!lstrcmpiA(buffer, destfile), "unexpected path: %s\n", buffer);

	SetupCloseInfFile(hinf);
	DeleteFileA(inf_filename);
	}

	START_TEST(query)
	{
	get_directories();

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