/*
 * Copyright (c) 2016 Daniel Boland <dboland@xs4all.nl>.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the copyright holders nor the names of its 
 *    contributors may be used to endorse or promote products derived 
 *    from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS 
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF 
 * THE POSSIBILITY OF SUCH DAMAGE.
 *
 */

#include <ddk/ntifs.h>		/* Installable File System */

/************************************************************/

VOID 
PDOLookupClass(LPCWSTR ClassID, LPWSTR Result)
{
	WIN_VNODE vNode;
	DWORD dwResult;
	WIN_FLAGS wFlags = {GENERIC_READ, 0, 0, REG_SZ, FALSE};
	WIN_NAMEIDATA wPath;

	if (reg_open(reg_namei(&wPath, REG_CLASS, ClassID), &wFlags, &vNode)){
		if (!reg_read(&vNode, L"Class", Result, MAX_NAME, &dwResult)){
			win_wcscpy(Result, L"UNKNOWN");
		}
		reg_close(&vNode);
		win_wcslcase(Result);
	}
}
DWORD 
PDOLookupBus(LPCWSTR BusName, DWORD Class)
{
	DWORD dwResult = Class;

	if (!wcsncmp(BusName, L"PCI", 3)){
		dwResult |= DEV_BUS_PCI;

	}else if (!wcsncmp(BusName, L"USB", 3)){
		dwResult |= DEV_BUS_USB;

	}else if (!wcscmp(BusName, L"HID")){
		dwResult |= DEV_BUS_HID;

	}else if (!wcscmp(BusName, L"ACPI")){
		dwResult |= DEV_BUS_ACPI;

	}else if (!wcscmp(BusName, L"IDE")){
		dwResult |= DEV_BUS_IDE;

	}else if (!wcscmp(BusName, L"SCSI")){
		dwResult |= DEV_BUS_SCSI;

	}else if (!wcscmp(BusName, L"FDC")){
		dwResult |= DEV_BUS_FDC;

	}
	return(dwResult);
}
DWORD 
PDOLookupService(LPCWSTR Service, DWORD Bus)
{
	DWORD dwResult = Bus;

	if (!wcscmp(Service, L"Serial")){
		dwResult |= DEV_CLASS_SERIAL;

	}else if (!wcscmp(Service, L"Parport")){
		dwResult |= DEV_CLASS_PRINTER;

	}else if (!wcsncmp(Service, L"BTH", 3)){
		dwResult |= DEV_CLASS_USB;

	}
	return(dwResult);
}
DWORD 
PDOLookup(LPCWSTR Bus, LPCWSTR Class, LPCWSTR Service)
{
	DWORD dwResult = DEV_CLASS_DULL;

	if (!wcscmp(Class, L"diskdrive")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_DISK);

	}else if (!wcscmp(Class, L"cdrom")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_DISK);

	}else if (!wcscmp(Class, L"floppydisk")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_DISK);

	}else if (!wcscmp(Class, L"net")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_IFNET);

	}else if (!wcscmp(Class, L"display")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_DISPLAY);

	}else if (!wcscmp(Class, L"keyboard")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_KEYBOARD);

	}else if (!wcscmp(Class, L"mouse")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_MOUSE);

	}else if (!wcscmp(Class, L"media")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_MEDIA);

	}else if (!wcscmp(Class, L"image")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_MEDIA);

	}else if (!wcscmp(Class, L"volume")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_STORAGE);

	}else if (!wcscmp(Class, L"ports")){
		dwResult = PDOLookupService(Service, DEV_BUS_ISA);

	}else if (!wcscmp(Service, L"usbprint")){	/* no "Printer" class for USB */
		dwResult = DEV_TYPE_USBPRINT;

	}else if (!wcscmp(Class, L"usb")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_USB);

	}else if (!wcscmp(Class, L"hidclass")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_HID);

	}else if (!wcscmp(Class, L"system")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_SYSTEM);

		/* Vista */

	}else if (!wcscmp(Class, L"hdc")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_DISK);

	}else if (!wcscmp(Class, L"printqueue")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_PRINTER);

	}else if (!wcscmp(Class, L"monitor")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_DISPLAY);

	}else if (!wcscmp(Class, L"camera")){
		dwResult = PDOLookupBus(Bus, DEV_CLASS_MEDIA);

	}else if (!wcscmp(Class, L"processor")){
		dwResult = DEV_TYPE_PROCESSOR;

	}else{
		dwResult = PDOLookupBus(Bus, DEV_CLASS_DULL);

	}
	return(dwResult);
}

/************************************************************/

BOOL 
pdo_lookup(WIN_CFDATA *Config, DWORD Flags, WIN_CFDRIVER *Driver)
{
	BOOL bResult = FALSE;
	WIN_VNODE vNode = {0};
	DWORD dwSize;
	WIN_FLAGS wFlags = {GENERIC_READ, 0, 0, REG_SZ, FALSE};
	WIN_NAMEIDATA wPath;
	WCHAR szClass[MAX_GUID];
	WCHAR szText[MAX_TEXT];

	ZeroMemory(Driver, sizeof(WIN_CFDRIVER));
	if (reg_open(reg_namei(&wPath, REG_DRIVER, Config->DosPath), &wFlags, &vNode)){

		/* GUID of attached device
		 */
		win_wcscpy(Driver->ClassId, wPath.R);

		/* Device Class ("Components" in msinfo32.exe)
		 */
		if (reg_read(&vNode, L"ClassGUID", szClass, MAX_GUID, &dwSize)){
			PDOLookupClass(szClass, Driver->NtClass);
		}
		/* Device Driver (*.sys)
		 */
		if (!reg_read(&vNode, L"Service", Driver->Service, MAX_NAME, &dwSize)){
			win_wcscpy(Driver->Service, Driver->NtClass);
		}
		if (!reg_read(&vNode, L"LocationInformation", Driver->Location, MAX_COMMENT, &dwSize)){
			win_wcscpy(win_wcpcpy(Driver->Location, Driver->Service), L".sys");
		}
		if (reg_read(&vNode, L"FriendlyName", szText, MAX_TEXT, &dwSize)){
			win_wcscpy(Driver->Comment, szText);
		}else if (reg_read(&vNode, L"DeviceDesc", szText, MAX_TEXT, &dwSize)){
			win_wcscpy(Driver->Comment, szText);
		}
		bResult = reg_close(&vNode);

	}else{
		WIN_ERR("reg_open(%ls): %s\n", wPath.Resolved, win_strerror(GetLastError()));
	}
	Config->DeviceType = PDOLookup(Config->BusName, Driver->NtClass, Driver->Service);
	return(bResult);
}