reactos/dll/win32/msports/classinst.c

765 lines
22 KiB
C
Raw Normal View History

/*
* PROJECT: ReactOS system libraries
* LICENSE: GPL - See COPYING in the top level directory
* FILE: dlls\win32\msports\classinst.c
* PURPOSE: Ports class installer
* PROGRAMMERS: Copyright 2011 Eric Kohl
*/
#include "precomp.h"
#include <wchar.h>
#define NTOS_MODE_USER
#include <ndk/cmtypes.h>
typedef enum _PORT_TYPE
{
UnknownPort,
ParallelPort,
SerialPort
} PORT_TYPE;
LPWSTR pszCom = L"COM";
LPWSTR pszLpt = L"LPT";
BOOL
GetBootResourceList(HDEVINFO DeviceInfoSet,
PSP_DEVINFO_DATA DeviceInfoData,
PCM_RESOURCE_LIST *ppResourceList)
{
HKEY hDeviceKey = NULL;
HKEY hConfigKey = NULL;
LPBYTE lpBuffer = NULL;
DWORD dwDataSize;
LONG lError;
BOOL ret = FALSE;
*ppResourceList = NULL;
hDeviceKey = SetupDiCreateDevRegKeyW(DeviceInfoSet,
DeviceInfoData,
DICS_FLAG_GLOBAL,
0,
DIREG_DEV,
NULL,
NULL);
if (!hDeviceKey)
return FALSE;
lError = RegOpenKeyExW(hDeviceKey,
L"LogConf",
0,
KEY_QUERY_VALUE,
&hConfigKey);
if (lError != ERROR_SUCCESS)
goto done;
/* Get the configuration data size */
lError = RegQueryValueExW(hConfigKey,
L"BootConfig",
NULL,
NULL,
NULL,
&dwDataSize);
if (lError != ERROR_SUCCESS)
goto done;
/* Allocate the buffer */
lpBuffer = HeapAlloc(GetProcessHeap(), 0, dwDataSize);
if (lpBuffer == NULL)
goto done;
/* Retrieve the configuration data */
lError = RegQueryValueExW(hConfigKey,
L"BootConfig",
NULL,
NULL,
(LPBYTE)lpBuffer,
&dwDataSize);
if (lError == ERROR_SUCCESS)
ret = TRUE;
done:
if (ret == FALSE && lpBuffer != NULL)
HeapFree(GetProcessHeap(), 0, lpBuffer);
if (hConfigKey)
RegCloseKey(hConfigKey);
if (hDeviceKey)
RegCloseKey(hDeviceKey);
if (ret == TRUE)
*ppResourceList = (PCM_RESOURCE_LIST)lpBuffer;
return ret;
}
DWORD
GetSerialPortNumber(IN HDEVINFO DeviceInfoSet,
IN PSP_DEVINFO_DATA DeviceInfoData)
{
PCM_RESOURCE_LIST lpResourceList = NULL;
PCM_PARTIAL_RESOURCE_DESCRIPTOR lpResDes;
ULONG i;
DWORD dwBaseAddress = 0;
DWORD dwPortNumber = 0;
TRACE("GetSerialPortNumber(%p, %p)\n",
DeviceInfoSet, DeviceInfoData);
if (GetBootResourceList(DeviceInfoSet,
DeviceInfoData,
&lpResourceList))
{
TRACE("Full resource descriptors: %ul\n", lpResourceList->Count);
if (lpResourceList->Count > 0)
{
TRACE("Partial resource descriptors: %ul\n", lpResourceList->List[0].PartialResourceList.Count);
for (i = 0; i < lpResourceList->List[0].PartialResourceList.Count; i++)
{
lpResDes = &lpResourceList->List[0].PartialResourceList.PartialDescriptors[i];
TRACE("Type: %u\n", lpResDes->Type);
switch (lpResDes->Type)
{
case CmResourceTypePort:
TRACE("Port: Start: %I64x Length: %lu\n",
lpResDes->u.Port.Start.QuadPart,
lpResDes->u.Port.Length);
if ((lpResDes->u.Port.Start.HighPart == 0) && (dwBaseAddress == 0))
dwBaseAddress = (DWORD)lpResDes->u.Port.Start.LowPart;
break;
case CmResourceTypeInterrupt:
TRACE("Interrupt: Level: %lu Vector: %lu\n",
lpResDes->u.Interrupt.Level,
lpResDes->u.Interrupt.Vector);
break;
}
}
}
HeapFree(GetProcessHeap(), 0, lpResourceList);
}
switch (dwBaseAddress)
{
case 0x3f8:
dwPortNumber = 1;
break;
case 0x2f8:
dwPortNumber = 2;
break;
case 0x3e8:
dwPortNumber = 3;
break;
case 0x2e8:
dwPortNumber = 4;
break;
}
return dwPortNumber;
}
DWORD
GetParallelPortNumber(IN HDEVINFO DeviceInfoSet,
IN PSP_DEVINFO_DATA DeviceInfoData)
{
PCM_RESOURCE_LIST lpResourceList = NULL;
PCM_PARTIAL_RESOURCE_DESCRIPTOR lpResDes;
ULONG i;
DWORD dwBaseAddress = 0;
DWORD dwPortNumber = 0;
TRACE("GetParallelPortNumber(%p, %p)\n",
DeviceInfoSet, DeviceInfoData);
if (GetBootResourceList(DeviceInfoSet,
DeviceInfoData,
&lpResourceList))
{
TRACE("Full resource descriptors: %ul\n", lpResourceList->Count);
if (lpResourceList->Count > 0)
{
TRACE("Partial resource descriptors: %ul\n", lpResourceList->List[0].PartialResourceList.Count);
for (i = 0; i < lpResourceList->List[0].PartialResourceList.Count; i++)
{
lpResDes = &lpResourceList->List[0].PartialResourceList.PartialDescriptors[i];
TRACE("Type: %u\n", lpResDes->Type);
switch (lpResDes->Type)
{
case CmResourceTypePort:
TRACE("Port: Start: %I64x Length: %lu\n",
lpResDes->u.Port.Start.QuadPart,
lpResDes->u.Port.Length);
if ((lpResDes->u.Port.Start.HighPart == 0) && (dwBaseAddress == 0))
dwBaseAddress = (DWORD)lpResDes->u.Port.Start.LowPart;
break;
case CmResourceTypeInterrupt:
TRACE("Interrupt: Level: %lu Vector: %lu\n",
lpResDes->u.Interrupt.Level,
lpResDes->u.Interrupt.Vector);
break;
}
}
}
HeapFree(GetProcessHeap(), 0, lpResourceList);
}
switch (dwBaseAddress)
{
case 0x378:
dwPortNumber = 1;
break;
case 0x278:
dwPortNumber = 2;
break;
}
return dwPortNumber;
}
static DWORD
InstallSerialPort(IN HDEVINFO DeviceInfoSet,
IN PSP_DEVINFO_DATA DeviceInfoData)
{
WCHAR szDeviceDescription[256];
WCHAR szFriendlyName[256];
WCHAR szPortName[8];
DWORD dwPortNumber = 0;
DWORD dwSize;
HCOMDB hComDB = HCOMDB_INVALID_HANDLE_VALUE;
HKEY hKey;
LONG lError;
TRACE("InstallSerialPort(%p, %p)\n",
DeviceInfoSet, DeviceInfoData);
/* Open the com port database */
ComDBOpen(&hComDB);
/* Try to read the 'PortName' value and determine the port number */
hKey = SetupDiCreateDevRegKeyW(DeviceInfoSet,
DeviceInfoData,
DICS_FLAG_GLOBAL,
0,
DIREG_DEV,
NULL,
NULL);
if (hKey != INVALID_HANDLE_VALUE)
{
dwSize = sizeof(szPortName);
lError = RegQueryValueEx(hKey,
L"PortName",
NULL,
NULL,
(PBYTE)szPortName,
&dwSize);
if (lError == ERROR_SUCCESS)
{
if (_wcsnicmp(szPortName, pszCom, wcslen(pszCom)) == 0)
{
dwPortNumber = _wtoi(szPortName + wcslen(pszCom));
TRACE("COM port number found: %lu\n", dwPortNumber);
}
}
RegCloseKey(hKey);
}
/* Determine the port number from its resources ... */
if (dwPortNumber == 0)
dwPortNumber = GetSerialPortNumber(DeviceInfoSet,
DeviceInfoData);
if (dwPortNumber != 0)
{
/* ... and claim the port number in the database */
ComDBClaimPort(hComDB,
dwPortNumber,
FALSE,
NULL);
}
else
{
/* ... or claim the next free port number */
ComDBClaimNextFreePort(hComDB,
&dwPortNumber);
}
/* Build the name of the port device */
swprintf(szPortName, L"%s%u", pszCom, dwPortNumber);
/* Close the com port database */
if (hComDB != HCOMDB_INVALID_HANDLE_VALUE)
ComDBClose(hComDB);
/* Set the 'PortName' value */
hKey = SetupDiCreateDevRegKeyW(DeviceInfoSet,
DeviceInfoData,
DICS_FLAG_GLOBAL,
0,
DIREG_DEV,
NULL,
NULL);
if (hKey != INVALID_HANDLE_VALUE)
{
RegSetValueExW(hKey,
L"PortName",
0,
REG_SZ,
(LPBYTE)szPortName,
(wcslen(szPortName) + 1) * sizeof(WCHAR));
RegCloseKey(hKey);
}
/* Install the device */
if (!SetupDiInstallDevice(DeviceInfoSet,
DeviceInfoData))
{
return GetLastError();
}
/* Get the device description... */
if (SetupDiGetDeviceRegistryPropertyW(DeviceInfoSet,
DeviceInfoData,
SPDRP_DEVICEDESC,
NULL,
(LPBYTE)szDeviceDescription,
256 * sizeof(WCHAR),
NULL))
{
/* ... and use it to build a new friendly name */
swprintf(szFriendlyName,
L"%s (%s)",
szDeviceDescription,
szPortName);
}
else
{
/* ... or build a generic friendly name */
swprintf(szFriendlyName,
L"Serial Port (%s)",
szPortName);
}
/* Set the friendly name for the device */
SetupDiSetDeviceRegistryPropertyW(DeviceInfoSet,
DeviceInfoData,
SPDRP_FRIENDLYNAME,
(LPBYTE)szFriendlyName,
(wcslen(szFriendlyName) + 1) * sizeof(WCHAR));
return ERROR_SUCCESS;
}
static DWORD
InstallParallelPort(IN HDEVINFO DeviceInfoSet,
IN PSP_DEVINFO_DATA DeviceInfoData)
{
WCHAR szDeviceDescription[256];
WCHAR szFriendlyName[256];
WCHAR szPortName[8];
DWORD dwPortNumber = 0;
DWORD dwSize;
LONG lError;
HKEY hKey;
TRACE("InstallParallelPort(%p, %p)\n",
DeviceInfoSet, DeviceInfoData);
/* Try to read the 'PortName' value and determine the port number */
hKey = SetupDiCreateDevRegKeyW(DeviceInfoSet,
DeviceInfoData,
DICS_FLAG_GLOBAL,
0,
DIREG_DEV,
NULL,
NULL);
if (hKey != INVALID_HANDLE_VALUE)
{
dwSize = sizeof(szPortName);
lError = RegQueryValueEx(hKey,
L"PortName",
NULL,
NULL,
(PBYTE)szPortName,
&dwSize);
if (lError == ERROR_SUCCESS)
{
if (_wcsnicmp(szPortName, pszLpt, wcslen(pszLpt)) == 0)
{
dwPortNumber = _wtoi(szPortName + wcslen(pszLpt));
TRACE("LPT port number found: %lu\n", dwPortNumber);
}
}
RegCloseKey(hKey);
}
/* ... try to determine the port number from its resources */
if (dwPortNumber == 0)
{
dwPortNumber = GetParallelPortNumber(DeviceInfoSet,
DeviceInfoData);
TRACE("GetParallelPortNumber() returned port number: %lu\n", dwPortNumber);
}
if (dwPortNumber == 0)
{
/* FIXME */
FIXME("Got no valid port numer!\n");
}
if (dwPortNumber != 0)
{
swprintf(szPortName, L"%s%u", pszLpt, dwPortNumber);
}
else
{
wcscpy(szPortName, L"LPTx");
}
if (dwPortNumber != 0)
{
/* Set the 'PortName' value */
hKey = SetupDiCreateDevRegKeyW(DeviceInfoSet,
DeviceInfoData,
DICS_FLAG_GLOBAL,
0,
DIREG_DEV,
NULL,
NULL);
if (hKey != INVALID_HANDLE_VALUE)
{
RegSetValueExW(hKey,
L"PortName",
0,
REG_SZ,
(LPBYTE)szPortName,
(wcslen(szPortName) + 1) * sizeof(WCHAR));
RegCloseKey(hKey);
}
}
/* Install the device */
if (!SetupDiInstallDevice(DeviceInfoSet,
DeviceInfoData))
{
return GetLastError();
}
/* Get the device description... */
if (SetupDiGetDeviceRegistryPropertyW(DeviceInfoSet,
DeviceInfoData,
SPDRP_DEVICEDESC,
NULL,
(LPBYTE)szDeviceDescription,
256 * sizeof(WCHAR),
NULL))
{
/* ... and use it to build a new friendly name */
swprintf(szFriendlyName,
L"%s (%s)",
szDeviceDescription,
szPortName);
}
else
{
/* ... or build a generic friendly name */
swprintf(szFriendlyName,
L"Parallel Port (%s)",
szPortName);
}
TRACE("Friendly name: %S\n", szFriendlyName);
/* Set the friendly name for the device */
SetupDiSetDeviceRegistryPropertyW(DeviceInfoSet,
DeviceInfoData,
SPDRP_FRIENDLYNAME,
(LPBYTE)szFriendlyName,
(wcslen(szFriendlyName) + 1) * sizeof(WCHAR));
return ERROR_SUCCESS;
}
VOID
InstallDeviceData(IN HDEVINFO DeviceInfoSet,
IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL)
{
HKEY hKey = NULL;
HINF hInf = INVALID_HANDLE_VALUE;
SP_DRVINFO_DATA DriverInfoData;
PSP_DRVINFO_DETAIL_DATA DriverInfoDetailData;
WCHAR InfSectionWithExt[256];
BYTE buffer[2048];
DWORD dwRequired;
TRACE("InstallDeviceData()\n");
hKey = SetupDiCreateDevRegKeyW(DeviceInfoSet,
DeviceInfoData,
DICS_FLAG_GLOBAL,
0,
DIREG_DRV,
NULL,
NULL);
if (hKey == NULL)
goto done;
DriverInfoData.cbSize = sizeof(SP_DRVINFO_DATA);
if (!SetupDiGetSelectedDriverW(DeviceInfoSet,
DeviceInfoData,
&DriverInfoData))
{
goto done;
}
DriverInfoDetailData = (PSP_DRVINFO_DETAIL_DATA)buffer;
DriverInfoDetailData->cbSize = sizeof(SP_DRVINFO_DETAIL_DATA);
if (!SetupDiGetDriverInfoDetailW(DeviceInfoSet,
DeviceInfoData,
&DriverInfoData,
DriverInfoDetailData,
2048,
&dwRequired))
{
if (GetLastError() != ERROR_INSUFFICIENT_BUFFER)
goto done;
}
TRACE("Inf file name: %S\n", DriverInfoDetailData->InfFileName);
hInf = SetupOpenInfFileW(DriverInfoDetailData->InfFileName,
NULL,
INF_STYLE_WIN4,
NULL);
if (hInf == INVALID_HANDLE_VALUE)
goto done;
TRACE("Section name: %S\n", DriverInfoDetailData->SectionName);
SetupDiGetActualSectionToInstallW(hInf,
DriverInfoDetailData->SectionName,
InfSectionWithExt,
256,
NULL,
NULL);
TRACE("InfSectionWithExt: %S\n", InfSectionWithExt);
SetupInstallFromInfSectionW(NULL,
hInf,
InfSectionWithExt,
SPINST_REGISTRY,
hKey,
NULL,
0,
NULL,
NULL,
NULL,
NULL);
TRACE("Done\n");
done:;
if (hKey != NULL)
RegCloseKey(hKey);
if (hInf != INVALID_HANDLE_VALUE)
SetupCloseInfFile(hInf);
}
PORT_TYPE
GetPortType(IN HDEVINFO DeviceInfoSet,
IN PSP_DEVINFO_DATA DeviceInfoData)
{
HKEY hKey = NULL;
DWORD dwSize;
DWORD dwType = 0;
BYTE bData = 0;
PORT_TYPE PortType = UnknownPort;
LONG lError;
TRACE("GetPortType()\n");
hKey = SetupDiCreateDevRegKeyW(DeviceInfoSet,
DeviceInfoData,
DICS_FLAG_GLOBAL,
0,
DIREG_DRV,
NULL,
NULL);
if (hKey == NULL)
{
goto done;
}
dwSize = sizeof(BYTE);
lError = RegQueryValueExW(hKey,
L"PortSubClass",
NULL,
&dwType,
&bData,
&dwSize);
TRACE("lError: %ld\n", lError);
TRACE("dwSize: %lu\n", dwSize);
TRACE("dwType: %lu\n", dwType);
if (lError == ERROR_SUCCESS &&
dwSize == sizeof(BYTE) &&
dwType == REG_BINARY)
{
if (bData == 0)
PortType = ParallelPort;
else
PortType = SerialPort;
}
done:;
if (hKey != NULL)
RegCloseKey(hKey);
TRACE("GetPortType() returns %u \n", PortType);
return PortType;
}
static DWORD
InstallPort(IN HDEVINFO DeviceInfoSet,
IN PSP_DEVINFO_DATA DeviceInfoData)
{
PORT_TYPE PortType;
InstallDeviceData(DeviceInfoSet, DeviceInfoData);
PortType = GetPortType(DeviceInfoSet, DeviceInfoData);
switch (PortType)
{
case ParallelPort:
return InstallParallelPort(DeviceInfoSet, DeviceInfoData);
case SerialPort:
return InstallSerialPort(DeviceInfoSet, DeviceInfoData);
default:
return ERROR_DI_DO_DEFAULT;
}
}
static DWORD
RemovePort(IN HDEVINFO DeviceInfoSet,
IN PSP_DEVINFO_DATA DeviceInfoData)
{
PORT_TYPE PortType;
HCOMDB hComDB = HCOMDB_INVALID_HANDLE_VALUE;
HKEY hKey;
LONG lError;
DWORD dwPortNumber;
DWORD dwPortNameSize;
WCHAR szPortName[8];
/* If we are removing a serial port ... */
PortType = GetPortType(DeviceInfoSet, DeviceInfoData);
if (PortType == SerialPort)
{
/* Open the port database */
if (ComDBOpen(&hComDB) == ERROR_SUCCESS)
{
/* Open the device key */
hKey = SetupDiOpenDevRegKey(DeviceInfoSet,
DeviceInfoData,
DICS_FLAG_GLOBAL,
0,
DIREG_DEV,
KEY_READ);
if (hKey != INVALID_HANDLE_VALUE)
{
/* Query the port name */
dwPortNameSize = sizeof(szPortName);
lError = RegQueryValueEx(hKey,
L"PortName",
NULL,
NULL,
(PBYTE)szPortName,
&dwPortNameSize);
/* Close the device key */
RegCloseKey(hKey);
/* If we got a valid port name ...*/
if (lError == ERROR_SUCCESS)
{
/* Get the port number */
dwPortNumber = _wtoi(szPortName + wcslen(pszCom));
/* Release the port */
ComDBReleasePort(hComDB, dwPortNumber);
}
}
/* Close the port database */
ComDBClose(hComDB);
}
}
/* Remove the device */
if (!SetupDiRemoveDevice(DeviceInfoSet, DeviceInfoData))
return GetLastError();
return ERROR_SUCCESS;
}
DWORD
WINAPI
PortsClassInstaller(IN DI_FUNCTION InstallFunction,
IN HDEVINFO DeviceInfoSet,
IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL)
{
TRACE("PortsClassInstaller(%lu, %p, %p)\n",
InstallFunction, DeviceInfoSet, DeviceInfoData);
switch (InstallFunction)
{
case DIF_INSTALLDEVICE:
return InstallPort(DeviceInfoSet, DeviceInfoData);
case DIF_REMOVE:
return RemovePort(DeviceInfoSet, DeviceInfoData);
default:
TRACE("Install function %u ignored\n", InstallFunction);
return ERROR_DI_DO_DEFAULT;
}
}
/* EOF */