// Setup.cpp -- Helper classes for setup programs
// Copyright (C) 1999 by Walter Oney
// All rights reserved

#include "stdafx.h"
#include "Setup.h"

#include <initguid.h>
#include <devguid.h>
#include <newdev.h>

DWORD GetField(PINFCONTEXT ic, DWORD index, CString& value);

///////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////

CInf::CInf()
{
	m_hinf = INVALID_HANDLE_VALUE;
}

CInf::~CInf()
{
	if (m_hinf)	SetupCloseInfFile(m_hinf);
}

///////////////////////////////////////////////////////////////////////////////

DWORD CInf::EnumDevices(const CString& modelname, PENUMDEVCALLBACK callback, PVOID context)
{
	INFCONTEXT mod;
	DWORD code;
	
	if (!SetupFindFirstLine(m_hinf, modelname, NULL, &mod))
		return GetLastError();
	
	do
	{
		CString devname, instname, id;
		CStringArray idlist;
		
		if ((code = GetField(&mod, 0, devname)) != 0
			|| (code = GetField(&mod, 1, instname)) != 0)
			return code;
		//	devname	:"WDM Book: Wdm1 Example, free build"
		//	instname:"Wdm1.Install"
		int nfields = SetupGetFieldCount(&mod);
		for (int i = 2; i <= nfields; ++i)
		{
			if (GetField(&mod, i, id) == 0)
				idlist.Add(id);
			else	break;
		}
		if (idlist.GetSize() == 0)
			idlist.Add(_T(""));
		if (!(*callback)(this, devname, instname, idlist, context, code))
			return code;
	}while (SetupFindNextLine(&mod, &mod));

	return ERROR_NO_MORE_ITEMS;
}

///////////////////////////////////////////////////////////////////////////////
DWORD CInf::EnumManufacturers(PENUMMFGCALLBACK callback, PVOID context)
{
	INFCONTEXT mfg;
	DWORD code;
	
	if (!SetupFindFirstLine(m_hinf, _T("Manufacturer"), NULL, &mfg))
		return GetLastError();
	
	// Loop throught the manufacturer section, calling the specified callback
	// function for each one
	
	do
	{
		CString mfgname, modelname;
		if ((code = GetField(&mfg, 0, mfgname)) != 0
			|| (code = GetField(&mfg, 1, modelname)) != 0)
			return code;
		
		// Invoke callback function. It returns TRUE if we should continue
		// the enumeration or FALSE if we should stop.
		
		if (!(*callback)(this, mfgname, modelname, context, code))
			return code;
	}while (SetupFindNextLine(&mfg, &mfg));
	
	return ERROR_NO_MORE_ITEMS;
}							// CInf::EnumManufacturers

///////////////////////////////////////////////////////////////////////////////
//	打開(kāi)INF文件,獲取CLASS名和ClassGuid
DWORD CInf::Open(LPCTSTR name, BOOL defsearch /* = TRUE */)
{							// CInf::Open
	CString infname(name);
	if (infname[0] == '"')
		infname = infname.Mid(2, infname.GetLength() - 2); // remove quotes from name
	
	//	若INF文件名不帶目錄,則將其變?yōu)?quot;.\xxx.INF"的形式,對(duì)".\xxx.INF"形式,其默認(rèn)INF
	//	文件所在路徑為HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\DevicePath值,
	//	該值一般為"%SystemRoot%\inf"
	if (!defsearch && _tcschr(infname, _T('\\')) == NULL)
		m_name = _T(".\\");
	m_name += infname;
	m_hinf = SetupOpenInfFile(m_name, NULL, INF_STYLE_WIN4, NULL);
	if (m_hinf == INVALID_HANDLE_VALUE)
		return GetLastError();
	//	獲取CLASS和ClassGuid
	TCHAR classname[64];
	if (!SetupDiGetINFClass(m_name, &m_guid, classname, arraysize(classname), NULL))
		return GetLastError();
	m_classname = classname;
	DWORD junk;
	//	若INF文件中未指明ClassGuid,則由CLASS名查找ClassGuid.
	if (m_guid == GUID_NULL) SetupDiClassGuidsFromName(classname, &m_guid, 1, &junk);
	return 0;
}							// CInf::Open

///////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////

CDevInfoSet::CDevInfoSet()
{							// CDevInfoSet::CDevInfoSet
	m_hinfo = INVALID_HANDLE_VALUE;
	memset(&m_devinfo, 0, sizeof(m_devinfo));
	m_devinfo.cbSize = sizeof(m_devinfo);
}							// CDevInfoSet::CDevInfoSet

CDevInfoSet::~CDevInfoSet()
{							// CDevInfoSet::~CDevInfoSet
	UnregisterDevice();
	if (m_hinfo != INVALID_HANDLE_VALUE)
		SetupDiDestroyDeviceInfoList(m_hinfo);
}							// CDevInfoSet::~CDevInfoSet

///////////////////////////////////////////////////////////////////////////////

DWORD CDevInfoSet::Create(CInf* inf, HWND hwnd)
{							// CDevInfoSet::Create
	m_inf = inf;
	m_hwnd = hwnd;
	m_hinfo = SetupDiCreateDeviceInfoList(&m_inf->m_guid, hwnd);
	if (m_hinfo == INVALID_HANDLE_VALUE)
		return GetLastError();
	
	return 0;
}							// CDevInfoSet::Create

///////////////////////////////////////////////////////////////////////////////

DWORD CDevInfoSet::CreateDeviceInfo(const CString& devid)
{							// CDevInfoSet::CreateDeviceInfo
	if (!SetupDiCreateDeviceInfo(m_hinfo, m_inf->m_classname, &m_inf->m_guid,
		NULL, m_hwnd, DICD_GENERATE_ID, &m_devinfo))
		return GetLastError();
	
	m_devid = devid;
	DWORD size = (devid.GetLength() + 2) * sizeof(TCHAR);
	PBYTE hwid = new BYTE[size];
	memset(hwid, 0, size);
	memcpy(hwid, (LPCTSTR) devid, size - 2 * sizeof(TCHAR));
	
	BOOL okay = SetupDiSetDeviceRegistryProperty(m_hinfo, &m_devinfo, SPDRP_HARDWAREID,
		hwid, size);
	
	delete[] hwid;
	
	if (okay)
		return 0;
	
	return GetLastError();
}

///////////////////////////////////////////////////////////////////////////////

DWORD CDevInfoSet::RegisterDevice()
{
	if (!SetupDiCallClassInstaller(DIF_REGISTERDEVICE, m_hinfo, &m_devinfo))
		return GetLastError();
	m_registered = TRUE;
	return 0;
}

///////////////////////////////////////////////////////////////////////////////

DWORD CDevInfoSet::UnregisterDevice()
{
	if (m_registered)
		if (!SetupDiCallClassInstaller(DIF_REMOVE, m_hinfo, &m_devinfo))
			return GetLastError();
		return 0;
}

///////////////////////////////////////////////////////////////////////////////

DWORD CDevInfoSet::UpdateDriver()
{							// CDevInfoSet::UpdateDriver
	BOOL reboot = FALSE;
	if (!UpdateDriverForPlugAndPlayDevices(m_hwnd, m_devid, m_inf->m_name,
		INSTALLFLAG_FORCE, &reboot))
		return GetLastError();
	
	m_registered = FALSE;		// prevent unregistration when done
	return 0;
}							// CDevInfoSet::UpdateDriver

///////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////

DWORD GetField(PINFCONTEXT ic, DWORD index, CString& value)
{							// GetField
	DWORD need;
	DWORD code;
	
	if (!SetupGetStringField(ic, index, NULL, 0, &need))
		return GetLastError();
	LPTSTR buffer = value.GetBuffer(need);
	if (SetupGetStringField(ic, index, buffer, need, NULL))
		code = 0;
	else
	{
		code = GetLastError();
		buffer[0] = 0;
	}
	
	value.ReleaseBuffer();
	return code;
}							// GetField