2005-10-31 00:14:45 +00:00
|
|
|
|
/*
|
|
|
|
|
* COPYRIGHT: See COPYING in the top level directory
|
|
|
|
|
* PROJECT: ReactOS Mouse class driver
|
|
|
|
|
* FILE: drivers/mouclass/mouclass.c
|
|
|
|
|
* PURPOSE: Mouse class driver
|
|
|
|
|
*
|
|
|
|
|
* PROGRAMMERS: Herv<EFBFBD> Poussineau (hpoussin@reactos.org)
|
|
|
|
|
*/
|
2000-10-18 09:11:01 +00:00
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
#define NDEBUG
|
|
|
|
|
#include <debug.h>
|
2000-10-18 09:11:01 +00:00
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
#define INITGUID
|
|
|
|
|
#include "mouclass.h"
|
2000-10-18 09:11:01 +00:00
|
|
|
|
|
2005-11-09 11:16:43 +00:00
|
|
|
|
static NTSTATUS
|
|
|
|
|
SearchForLegacyDrivers(
|
|
|
|
|
IN PDRIVER_OBJECT DriverObject,
|
|
|
|
|
IN PCLASS_DRIVER_EXTENSION DriverExtension);
|
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
static VOID NTAPI
|
|
|
|
|
DriverUnload(IN PDRIVER_OBJECT DriverObject)
|
|
|
|
|
{
|
|
|
|
|
// nothing to do here yet
|
|
|
|
|
}
|
2000-10-18 09:11:01 +00:00
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
static NTSTATUS NTAPI
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ClassCreate(
|
2005-10-31 00:14:45 +00:00
|
|
|
|
IN PDEVICE_OBJECT DeviceObject,
|
|
|
|
|
IN PIRP Irp)
|
|
|
|
|
{
|
|
|
|
|
DPRINT("IRP_MJ_CREATE\n");
|
2000-10-18 09:11:01 +00:00
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
if (!((PCOMMON_DEVICE_EXTENSION)DeviceObject->DeviceExtension)->IsClassDO)
|
|
|
|
|
return ForwardIrpAndForget(DeviceObject, Irp);
|
2002-12-16 23:23:21 +00:00
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
/* FIXME: open all associated Port devices */
|
|
|
|
|
return STATUS_SUCCESS;
|
|
|
|
|
}
|
2005-10-06 21:39:18 +00:00
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
static NTSTATUS NTAPI
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ClassClose(
|
2005-10-31 00:14:45 +00:00
|
|
|
|
IN PDEVICE_OBJECT DeviceObject,
|
|
|
|
|
IN PIRP Irp)
|
|
|
|
|
{
|
|
|
|
|
DPRINT("IRP_MJ_CLOSE\n");
|
|
|
|
|
|
|
|
|
|
if (!((PCOMMON_DEVICE_EXTENSION)DeviceObject->DeviceExtension)->IsClassDO)
|
|
|
|
|
return ForwardIrpAndForget(DeviceObject, Irp);
|
2005-06-03 17:09:47 +00:00
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
/* FIXME: close all associated Port devices */
|
|
|
|
|
return STATUS_SUCCESS;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static NTSTATUS NTAPI
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ClassCleanup(
|
|
|
|
|
IN PDEVICE_OBJECT DeviceObject,
|
|
|
|
|
IN PIRP Irp)
|
|
|
|
|
{
|
|
|
|
|
DPRINT("IRP_MJ_CLEANUP\n");
|
|
|
|
|
|
|
|
|
|
if (!((PCOMMON_DEVICE_EXTENSION)DeviceObject->DeviceExtension)->IsClassDO)
|
|
|
|
|
return ForwardIrpAndForget(DeviceObject, Irp);
|
|
|
|
|
|
|
|
|
|
/* FIXME: cleanup all associated Port devices */
|
|
|
|
|
return STATUS_SUCCESS;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static NTSTATUS NTAPI
|
|
|
|
|
ClassRead(
|
2005-10-31 00:14:45 +00:00
|
|
|
|
IN PDEVICE_OBJECT DeviceObject,
|
|
|
|
|
IN PIRP Irp)
|
2000-10-18 09:11:01 +00:00
|
|
|
|
{
|
2005-10-31 00:14:45 +00:00
|
|
|
|
DPRINT("IRP_MJ_READ\n");
|
|
|
|
|
|
|
|
|
|
if (!((PCOMMON_DEVICE_EXTENSION)DeviceObject->DeviceExtension)->IsClassDO)
|
|
|
|
|
return ForwardIrpAndForget(DeviceObject, Irp);
|
|
|
|
|
|
|
|
|
|
if (IoGetCurrentIrpStackLocation(Irp)->Parameters.Read.Length < sizeof(MOUSE_INPUT_DATA))
|
|
|
|
|
{
|
|
|
|
|
Irp->IoStatus.Status = STATUS_BUFFER_TOO_SMALL;
|
|
|
|
|
Irp->IoStatus.Information = 0;
|
|
|
|
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
|
|
|
|
|
|
|
|
|
return STATUS_BUFFER_TOO_SMALL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
IoMarkIrpPending(Irp);
|
|
|
|
|
IoStartPacket(DeviceObject, Irp, NULL, NULL);
|
|
|
|
|
return STATUS_PENDING;
|
2000-10-18 09:11:01 +00:00
|
|
|
|
}
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
static NTSTATUS NTAPI
|
|
|
|
|
ClassDeviceControl(
|
|
|
|
|
IN PDEVICE_OBJECT DeviceObject,
|
|
|
|
|
IN PIRP Irp)
|
|
|
|
|
{
|
|
|
|
|
PCLASS_DEVICE_EXTENSION DeviceExtension;
|
|
|
|
|
NTSTATUS Status;
|
|
|
|
|
|
|
|
|
|
DPRINT("IRP_MJ_DEVICE_CONTROL\n");
|
|
|
|
|
|
|
|
|
|
if (!((PCOMMON_DEVICE_EXTENSION)DeviceObject->DeviceExtension)->IsClassDO)
|
|
|
|
|
return ForwardIrpAndForget(DeviceObject, Irp);
|
|
|
|
|
|
|
|
|
|
DeviceExtension = (PCLASS_DEVICE_EXTENSION)DeviceObject->DeviceExtension;
|
|
|
|
|
|
|
|
|
|
switch (IoGetCurrentIrpStackLocation(Irp)->Parameters.DeviceIoControl.IoControlCode)
|
|
|
|
|
{
|
|
|
|
|
case IOCTL_MOUSE_QUERY_ATTRIBUTES:
|
|
|
|
|
/* FIXME */
|
|
|
|
|
Status = STATUS_NOT_SUPPORTED;
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
DPRINT1("IRP_MJ_DEVICE_CONTROL / unknown I/O control code 0x%lx\n",
|
|
|
|
|
IoGetCurrentIrpStackLocation(Irp)->Parameters.DeviceIoControl.IoControlCode);
|
|
|
|
|
Status = STATUS_NOT_SUPPORTED;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Irp->IoStatus.Status = Status;
|
|
|
|
|
Irp->IoStatus.Information = 0;
|
|
|
|
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
|
|
|
|
|
|
|
|
|
return Status;
|
|
|
|
|
}
|
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
static NTSTATUS NTAPI
|
|
|
|
|
IrpStub(
|
|
|
|
|
IN PDEVICE_OBJECT DeviceObject,
|
|
|
|
|
IN PIRP Irp)
|
2000-10-18 09:11:01 +00:00
|
|
|
|
{
|
2005-10-31 00:14:45 +00:00
|
|
|
|
NTSTATUS Status = STATUS_NOT_SUPPORTED;
|
|
|
|
|
|
|
|
|
|
if (!((PCOMMON_DEVICE_EXTENSION)DeviceObject->DeviceExtension)->IsClassDO)
|
|
|
|
|
{
|
|
|
|
|
/* Forward some IRPs to lower device */
|
|
|
|
|
switch (IoGetCurrentIrpStackLocation(Irp)->MajorFunction)
|
|
|
|
|
{
|
2005-12-18 20:50:26 +00:00
|
|
|
|
case IRP_MJ_PNP:
|
2005-10-31 00:14:45 +00:00
|
|
|
|
case IRP_MJ_INTERNAL_DEVICE_CONTROL:
|
|
|
|
|
return ForwardIrpAndForget(DeviceObject, Irp);
|
|
|
|
|
default:
|
|
|
|
|
{
|
|
|
|
|
DPRINT1("Port DO stub for major function 0x%lx\n",
|
|
|
|
|
IoGetCurrentIrpStackLocation(Irp)->MajorFunction);
|
|
|
|
|
ASSERT(FALSE);
|
|
|
|
|
Status = Irp->IoStatus.Status;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
DPRINT1("Class DO stub for major function 0x%lx\n",
|
|
|
|
|
IoGetCurrentIrpStackLocation(Irp)->MajorFunction);
|
|
|
|
|
ASSERT(FALSE);
|
|
|
|
|
Status = Irp->IoStatus.Status;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Irp->IoStatus.Status = Status;
|
|
|
|
|
IoCompleteRequest(Irp, IO_NO_INCREMENT);
|
|
|
|
|
return Status;
|
2000-10-18 09:11:01 +00:00
|
|
|
|
}
|
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
static NTSTATUS
|
|
|
|
|
ReadRegistryEntries(
|
|
|
|
|
IN PUNICODE_STRING RegistryPath,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
IN PCLASS_DRIVER_EXTENSION DriverExtension)
|
2000-10-18 09:11:01 +00:00
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
UNICODE_STRING ParametersRegistryKey;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
RTL_QUERY_REGISTRY_TABLE Parameters[4];
|
|
|
|
|
NTSTATUS Status;
|
|
|
|
|
|
|
|
|
|
ULONG DefaultConnectMultiplePorts = 1;
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ULONG DefaultDataQueueSize = 0x64;
|
|
|
|
|
UNICODE_STRING DefaultDeviceBaseName = RTL_CONSTANT_STRING(L"PointerClass");
|
|
|
|
|
|
|
|
|
|
ParametersRegistryKey.Length = 0;
|
|
|
|
|
ParametersRegistryKey.MaximumLength = RegistryPath->Length + sizeof(L"\\Parameters") + sizeof(UNICODE_NULL);
|
|
|
|
|
ParametersRegistryKey.Buffer = ExAllocatePool(PagedPool, ParametersRegistryKey.MaximumLength);
|
|
|
|
|
if (!ParametersRegistryKey.Buffer)
|
|
|
|
|
{
|
|
|
|
|
DPRINT("ExAllocatePool() failed\n");
|
|
|
|
|
return STATUS_INSUFFICIENT_RESOURCES;
|
|
|
|
|
}
|
|
|
|
|
RtlCopyUnicodeString(&ParametersRegistryKey, RegistryPath);
|
|
|
|
|
RtlAppendUnicodeToString(&ParametersRegistryKey, L"\\Parameters");
|
|
|
|
|
ParametersRegistryKey.Buffer[ParametersRegistryKey.Length / sizeof(WCHAR)] = UNICODE_NULL;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
|
|
|
|
RtlZeroMemory(Parameters, sizeof(Parameters));
|
|
|
|
|
|
|
|
|
|
Parameters[0].Flags = RTL_QUERY_REGISTRY_DIRECT | RTL_REGISTRY_OPTIONAL;
|
|
|
|
|
Parameters[0].Name = L"ConnectMultiplePorts";
|
|
|
|
|
Parameters[0].EntryContext = &DriverExtension->ConnectMultiplePorts;
|
|
|
|
|
Parameters[0].DefaultType = REG_DWORD;
|
|
|
|
|
Parameters[0].DefaultData = &DefaultConnectMultiplePorts;
|
|
|
|
|
Parameters[0].DefaultLength = sizeof(ULONG);
|
2005-11-05 08:37:01 +00:00
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
Parameters[1].Flags = RTL_QUERY_REGISTRY_DIRECT | RTL_REGISTRY_OPTIONAL;
|
|
|
|
|
Parameters[1].Name = L"MouseDataQueueSize";
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Parameters[1].EntryContext = &DriverExtension->DataQueueSize;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
Parameters[1].DefaultType = REG_DWORD;
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Parameters[1].DefaultData = &DefaultDataQueueSize;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
Parameters[1].DefaultLength = sizeof(ULONG);
|
2005-11-05 08:37:01 +00:00
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
Parameters[2].Flags = RTL_QUERY_REGISTRY_DIRECT | RTL_REGISTRY_OPTIONAL;
|
|
|
|
|
Parameters[2].Name = L"PointerDeviceBaseName";
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Parameters[2].EntryContext = &DriverExtension->DeviceBaseName;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
Parameters[2].DefaultType = REG_SZ;
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Parameters[2].DefaultData = &DefaultDeviceBaseName;
|
2005-11-09 11:16:43 +00:00
|
|
|
|
Parameters[2].DefaultLength = 0;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
|
|
|
|
Status = RtlQueryRegistryValues(
|
|
|
|
|
RTL_REGISTRY_ABSOLUTE,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ParametersRegistryKey.Buffer,
|
2005-10-31 00:14:45 +00:00
|
|
|
|
Parameters,
|
|
|
|
|
NULL,
|
|
|
|
|
NULL);
|
|
|
|
|
|
|
|
|
|
if (NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
/* Check values */
|
|
|
|
|
if (DriverExtension->ConnectMultiplePorts != 0
|
|
|
|
|
&& DriverExtension->ConnectMultiplePorts != 1)
|
|
|
|
|
{
|
|
|
|
|
DriverExtension->ConnectMultiplePorts = DefaultConnectMultiplePorts;
|
|
|
|
|
}
|
2005-11-05 08:37:01 +00:00
|
|
|
|
if (DriverExtension->DataQueueSize == 0)
|
2005-10-31 00:14:45 +00:00
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DriverExtension->DataQueueSize = DefaultDataQueueSize;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
2005-11-05 08:37:01 +00:00
|
|
|
|
else if (Status == STATUS_OBJECT_NAME_NOT_FOUND)
|
|
|
|
|
{
|
|
|
|
|
/* Registry path doesn't exist. Set defaults */
|
|
|
|
|
DriverExtension->ConnectMultiplePorts = DefaultConnectMultiplePorts;
|
|
|
|
|
DriverExtension->DataQueueSize = DefaultDataQueueSize;
|
|
|
|
|
Status = RtlDuplicateUnicodeString(
|
|
|
|
|
RTL_DUPLICATE_UNICODE_STRING_NULL_TERMINATE,
|
|
|
|
|
&DefaultDeviceBaseName,
|
|
|
|
|
&DriverExtension->DeviceBaseName);
|
|
|
|
|
}
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
|
|
|
|
return Status;
|
2000-10-18 09:11:01 +00:00
|
|
|
|
}
|
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
static NTSTATUS
|
2005-11-05 08:37:01 +00:00
|
|
|
|
CreateClassDeviceObject(
|
2005-10-31 00:14:45 +00:00
|
|
|
|
IN PDRIVER_OBJECT DriverObject,
|
|
|
|
|
OUT PDEVICE_OBJECT *ClassDO OPTIONAL)
|
2000-10-18 09:11:01 +00:00
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
PCLASS_DRIVER_EXTENSION DriverExtension;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
ULONG DeviceId = 0;
|
|
|
|
|
ULONG PrefixLength;
|
|
|
|
|
UNICODE_STRING DeviceNameU;
|
|
|
|
|
PWSTR DeviceIdW = NULL; /* Pointer into DeviceNameU.Buffer */
|
|
|
|
|
PDEVICE_OBJECT Fdo;
|
2005-11-05 08:37:01 +00:00
|
|
|
|
PCLASS_DEVICE_EXTENSION DeviceExtension;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
NTSTATUS Status;
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DPRINT("CreateClassDeviceObject(0x%p)\n", DriverObject);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
|
|
|
|
/* Create new device object */
|
|
|
|
|
DriverExtension = IoGetDriverObjectExtension(DriverObject, DriverObject);
|
|
|
|
|
DeviceNameU.Length = 0;
|
|
|
|
|
DeviceNameU.MaximumLength =
|
2005-11-05 08:37:01 +00:00
|
|
|
|
wcslen(L"\\Device\\") * sizeof(WCHAR) /* "\Device\" */
|
|
|
|
|
+ DriverExtension->DeviceBaseName.Length /* "PointerClass" */
|
|
|
|
|
+ 4 * sizeof(WCHAR) /* Id between 0 and 9999 */
|
|
|
|
|
+ sizeof(UNICODE_NULL); /* Final NULL char */
|
2005-10-31 00:14:45 +00:00
|
|
|
|
DeviceNameU.Buffer = ExAllocatePool(PagedPool, DeviceNameU.MaximumLength);
|
|
|
|
|
if (!DeviceNameU.Buffer)
|
|
|
|
|
{
|
|
|
|
|
DPRINT("ExAllocatePool() failed\n");
|
|
|
|
|
return STATUS_INSUFFICIENT_RESOURCES;
|
|
|
|
|
}
|
|
|
|
|
Status = RtlAppendUnicodeToString(&DeviceNameU, L"\\Device\\");
|
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
DPRINT("RtlAppendUnicodeToString() failed with status 0x%08lx\n", Status);
|
|
|
|
|
goto cleanup;
|
|
|
|
|
}
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Status = RtlAppendUnicodeStringToString(&DeviceNameU, &DriverExtension->DeviceBaseName);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
DPRINT("RtlAppendUnicodeStringToString() failed with status 0x%08lx\n", Status);
|
|
|
|
|
goto cleanup;
|
|
|
|
|
}
|
|
|
|
|
PrefixLength = DeviceNameU.MaximumLength - 4 * sizeof(WCHAR) - sizeof(UNICODE_NULL);
|
|
|
|
|
DeviceIdW = &DeviceNameU.Buffer[PrefixLength / sizeof(WCHAR)];
|
|
|
|
|
while (DeviceId < 9999)
|
|
|
|
|
{
|
2005-10-31 16:46:46 +00:00
|
|
|
|
DeviceNameU.Length = PrefixLength + swprintf(DeviceIdW, L"%lu", DeviceId) * sizeof(WCHAR);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
Status = IoCreateDevice(
|
|
|
|
|
DriverObject,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
sizeof(CLASS_DEVICE_EXTENSION),
|
2005-10-31 00:14:45 +00:00
|
|
|
|
&DeviceNameU,
|
|
|
|
|
FILE_DEVICE_MOUSE,
|
|
|
|
|
FILE_DEVICE_SECURE_OPEN,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
TRUE,
|
2005-10-31 00:14:45 +00:00
|
|
|
|
&Fdo);
|
|
|
|
|
if (NT_SUCCESS(Status))
|
|
|
|
|
goto cleanup;
|
|
|
|
|
else if (Status != STATUS_OBJECT_NAME_COLLISION)
|
|
|
|
|
{
|
|
|
|
|
DPRINT("IoCreateDevice() failed with status 0x%08lx\n", Status);
|
|
|
|
|
goto cleanup;
|
|
|
|
|
}
|
|
|
|
|
DeviceId++;
|
|
|
|
|
}
|
2006-01-02 11:32:03 +00:00
|
|
|
|
DPRINT("Too many devices starting with '\\Device\\%wZ'\n", &DriverExtension->DeviceBaseName);
|
|
|
|
|
Status = STATUS_TOO_MANY_NAMES;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
cleanup:
|
|
|
|
|
if (!NT_SUCCESS(Status))
|
2005-11-05 08:37:01 +00:00
|
|
|
|
{
|
|
|
|
|
ExFreePool(DeviceNameU.Buffer);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
return Status;
|
2005-11-05 08:37:01 +00:00
|
|
|
|
}
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DeviceExtension = (PCLASS_DEVICE_EXTENSION)Fdo->DeviceExtension;
|
|
|
|
|
RtlZeroMemory(DeviceExtension, sizeof(CLASS_DEVICE_EXTENSION));
|
2005-10-31 00:14:45 +00:00
|
|
|
|
DeviceExtension->Common.IsClassDO = TRUE;
|
|
|
|
|
DeviceExtension->DriverExtension = DriverExtension;
|
|
|
|
|
KeInitializeSpinLock(&(DeviceExtension->SpinLock));
|
|
|
|
|
DeviceExtension->ReadIsPending = FALSE;
|
|
|
|
|
DeviceExtension->InputCount = 0;
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DeviceExtension->PortData = ExAllocatePool(NonPagedPool, DeviceExtension->DriverExtension->DataQueueSize * sizeof(MOUSE_INPUT_DATA));
|
2005-10-31 00:14:45 +00:00
|
|
|
|
Fdo->Flags |= DO_POWER_PAGABLE;
|
|
|
|
|
Fdo->Flags &= ~DO_DEVICE_INITIALIZING;
|
|
|
|
|
|
2005-11-09 11:16:43 +00:00
|
|
|
|
/* Add entry entry to HKEY_LOCAL_MACHINE\HARDWARE\DEVICEMAP\[DeviceBaseName] */
|
|
|
|
|
RtlWriteRegistryValue(
|
|
|
|
|
RTL_REGISTRY_DEVICEMAP,
|
|
|
|
|
DriverExtension->DeviceBaseName.Buffer,
|
|
|
|
|
DeviceNameU.Buffer,
|
|
|
|
|
REG_SZ,
|
|
|
|
|
DriverExtension->RegistryPath.Buffer,
|
|
|
|
|
DriverExtension->RegistryPath.MaximumLength);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ExFreePool(DeviceNameU.Buffer);
|
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
if (ClassDO)
|
|
|
|
|
*ClassDO = Fdo;
|
|
|
|
|
|
|
|
|
|
return STATUS_SUCCESS;
|
2001-05-26 08:19:03 +00:00
|
|
|
|
}
|
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
static BOOLEAN
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ClassCallback(
|
2005-10-31 00:14:45 +00:00
|
|
|
|
IN PDEVICE_OBJECT ClassDeviceObject,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
IN OUT PMOUSE_INPUT_DATA DataStart,
|
|
|
|
|
IN PMOUSE_INPUT_DATA DataEnd,
|
2005-10-31 00:14:45 +00:00
|
|
|
|
IN OUT PULONG ConsumedCount)
|
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
PCLASS_DEVICE_EXTENSION ClassDeviceExtension = ClassDeviceObject->DeviceExtension;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
PIRP Irp = NULL;
|
|
|
|
|
KIRQL OldIrql;
|
|
|
|
|
PIO_STACK_LOCATION Stack;
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ULONG InputCount = DataEnd - DataStart;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
ULONG ReadSize;
|
|
|
|
|
|
|
|
|
|
ASSERT(ClassDeviceExtension->Common.IsClassDO);
|
|
|
|
|
|
2005-11-18 17:28:19 +00:00
|
|
|
|
KeAcquireSpinLock(&ClassDeviceExtension->SpinLock, &OldIrql);
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DPRINT("ClassCallback()\n");
|
2005-10-31 00:14:45 +00:00
|
|
|
|
/* A filter driver might have consumed all the data already; I'm
|
|
|
|
|
* not sure if they are supposed to move the packets when they
|
|
|
|
|
* consume them though.
|
|
|
|
|
*/
|
|
|
|
|
if (ClassDeviceExtension->ReadIsPending == TRUE && InputCount)
|
|
|
|
|
{
|
|
|
|
|
Irp = ClassDeviceObject->CurrentIrp;
|
|
|
|
|
ClassDeviceObject->CurrentIrp = NULL;
|
|
|
|
|
Stack = IoGetCurrentIrpStackLocation(Irp);
|
|
|
|
|
|
|
|
|
|
/* A read request is waiting for input, so go straight to it */
|
2005-10-31 16:46:46 +00:00
|
|
|
|
/* FIXME: use SEH */
|
|
|
|
|
RtlCopyMemory(
|
|
|
|
|
Irp->MdlAddress ? MmGetSystemAddressForMdlSafe(Irp->MdlAddress, NormalPagePriority) : Irp->UserBuffer,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DataStart,
|
2005-10-31 00:14:45 +00:00
|
|
|
|
sizeof(MOUSE_INPUT_DATA));
|
|
|
|
|
|
|
|
|
|
/* Go to next packet and complete this request with STATUS_SUCCESS */
|
|
|
|
|
Irp->IoStatus.Status = STATUS_SUCCESS;
|
|
|
|
|
Irp->IoStatus.Information = sizeof(MOUSE_INPUT_DATA);
|
|
|
|
|
Stack->Parameters.Read.Length = sizeof(MOUSE_INPUT_DATA);
|
|
|
|
|
|
|
|
|
|
ClassDeviceExtension->ReadIsPending = FALSE;
|
|
|
|
|
|
|
|
|
|
/* Skip the packet we just sent away */
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DataStart++;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
(*ConsumedCount)++;
|
|
|
|
|
InputCount--;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* If we have data from the port driver and a higher service to send the data to */
|
|
|
|
|
if (InputCount != 0)
|
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
if (ClassDeviceExtension->InputCount + InputCount > ClassDeviceExtension->DriverExtension->DataQueueSize)
|
|
|
|
|
ReadSize = ClassDeviceExtension->DriverExtension->DataQueueSize - ClassDeviceExtension->InputCount;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
else
|
|
|
|
|
ReadSize = InputCount;
|
|
|
|
|
|
|
|
|
|
/*
|
2005-11-05 08:37:01 +00:00
|
|
|
|
* FIXME: If we exceed the buffer, data gets thrown away.. better
|
2005-10-31 00:14:45 +00:00
|
|
|
|
* solution?
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
/*
|
2005-11-05 08:37:01 +00:00
|
|
|
|
* Move the input data from the port data queue to our class data
|
2005-10-31 00:14:45 +00:00
|
|
|
|
* queue.
|
|
|
|
|
*/
|
|
|
|
|
RtlMoveMemory(
|
|
|
|
|
ClassDeviceExtension->PortData,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
(PCHAR)DataStart,
|
2005-10-31 00:14:45 +00:00
|
|
|
|
sizeof(MOUSE_INPUT_DATA) * ReadSize);
|
|
|
|
|
|
|
|
|
|
/* Move the pointer and counter up */
|
|
|
|
|
ClassDeviceExtension->PortData += ReadSize;
|
|
|
|
|
ClassDeviceExtension->InputCount += ReadSize;
|
|
|
|
|
|
|
|
|
|
(*ConsumedCount) += ReadSize;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DPRINT("ClassCallBack() entered, InputCount = %lu - DOING NOTHING\n", InputCount);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
2005-11-18 17:28:19 +00:00
|
|
|
|
KeReleaseSpinLock(&ClassDeviceExtension->SpinLock, OldIrql);
|
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
if (Irp != NULL)
|
|
|
|
|
{
|
|
|
|
|
IoStartNextPacket(ClassDeviceObject, FALSE);
|
|
|
|
|
IoCompleteRequest(Irp, IO_MOUSE_INCREMENT);
|
|
|
|
|
}
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DPRINT("Leaving ClassCallback()\n");
|
2005-10-31 00:14:45 +00:00
|
|
|
|
return TRUE;
|
|
|
|
|
}
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
/* Send IOCTL_INTERNAL_*_CONNECT to port */
|
2005-10-31 00:14:45 +00:00
|
|
|
|
static NTSTATUS
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ConnectPortDriver(
|
|
|
|
|
IN PDEVICE_OBJECT PortDO,
|
|
|
|
|
IN PDEVICE_OBJECT ClassDO)
|
2005-10-31 00:14:45 +00:00
|
|
|
|
{
|
|
|
|
|
KEVENT Event;
|
|
|
|
|
PIRP Irp;
|
|
|
|
|
IO_STATUS_BLOCK IoStatus;
|
|
|
|
|
CONNECT_DATA ConnectData;
|
|
|
|
|
NTSTATUS Status;
|
|
|
|
|
|
|
|
|
|
KeInitializeEvent(&Event, NotificationEvent, FALSE);
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ConnectData.ClassDeviceObject = ClassDO;
|
|
|
|
|
ConnectData.ClassService = ClassCallback;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
|
|
|
|
Irp = IoBuildDeviceIoControlRequest(IOCTL_INTERNAL_MOUSE_CONNECT,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
PortDO,
|
2005-10-31 00:14:45 +00:00
|
|
|
|
&ConnectData, sizeof(CONNECT_DATA),
|
|
|
|
|
NULL, 0,
|
|
|
|
|
TRUE, &Event, &IoStatus);
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Status = IoCallDriver(PortDO, Irp);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
|
|
|
|
if (Status == STATUS_PENDING)
|
|
|
|
|
KeWaitForSingleObject(&Event, Suspended, KernelMode, FALSE, NULL);
|
|
|
|
|
else
|
|
|
|
|
IoStatus.Status = Status;
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
if (NT_SUCCESS(IoStatus.Status))
|
|
|
|
|
ObReferenceObject(PortDO);
|
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
return IoStatus.Status;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static NTSTATUS NTAPI
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ClassAddDevice(
|
2005-10-31 00:14:45 +00:00
|
|
|
|
IN PDRIVER_OBJECT DriverObject,
|
|
|
|
|
IN PDEVICE_OBJECT Pdo)
|
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
PCLASS_DRIVER_EXTENSION DriverExtension;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
PDEVICE_OBJECT Fdo;
|
2005-11-05 08:37:01 +00:00
|
|
|
|
PPORT_DEVICE_EXTENSION DeviceExtension;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
NTSTATUS Status;
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DPRINT("ClassAddDevice called. Pdo = 0x%p\n", Pdo);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
|
|
|
|
DriverExtension = IoGetDriverObjectExtension(DriverObject, DriverObject);
|
|
|
|
|
|
2005-11-09 11:16:43 +00:00
|
|
|
|
if (Pdo == NULL)
|
|
|
|
|
/* We're getting a NULL Pdo at the first call as we're a legacy driver.
|
|
|
|
|
* Use it to search for legacy port drivers. */
|
|
|
|
|
return SearchForLegacyDrivers(DriverObject, DriverExtension);
|
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
/* Create new device object */
|
|
|
|
|
Status = IoCreateDevice(
|
|
|
|
|
DriverObject,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
sizeof(PORT_DEVICE_EXTENSION),
|
2005-10-31 00:14:45 +00:00
|
|
|
|
NULL,
|
|
|
|
|
Pdo->DeviceType,
|
|
|
|
|
FILE_DEVICE_SECURE_OPEN,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
TRUE,
|
2005-10-31 00:14:45 +00:00
|
|
|
|
&Fdo);
|
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
DPRINT("IoCreateDevice() failed with status 0x%08lx\n", Status);
|
|
|
|
|
return Status;
|
|
|
|
|
}
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DeviceExtension = (PPORT_DEVICE_EXTENSION)Fdo->DeviceExtension;
|
|
|
|
|
RtlZeroMemory(DeviceExtension, sizeof(CLASS_DEVICE_EXTENSION));
|
2005-10-31 00:14:45 +00:00
|
|
|
|
DeviceExtension->Common.IsClassDO = FALSE;
|
|
|
|
|
DeviceExtension->PnpState = dsStopped;
|
|
|
|
|
Status = IoAttachDeviceToDeviceStackSafe(Fdo, Pdo, &DeviceExtension->LowerDevice);
|
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
DPRINT("IoAttachDeviceToDeviceStackSafe() failed with status 0x%08lx\n", Status);
|
|
|
|
|
IoDeleteDevice(Fdo);
|
|
|
|
|
return Status;
|
|
|
|
|
}
|
2005-11-18 17:28:19 +00:00
|
|
|
|
if (DeviceExtension->LowerDevice->Flags & DO_POWER_PAGABLE)
|
|
|
|
|
Fdo->Flags |= DO_POWER_PAGABLE;
|
|
|
|
|
if (DeviceExtension->LowerDevice->Flags & DO_BUFFERED_IO)
|
|
|
|
|
Fdo->Flags |= DO_BUFFERED_IO;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
|
|
|
|
if (DriverExtension->ConnectMultiplePorts)
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Status = ConnectPortDriver(Fdo, DriverExtension->MainClassDeviceObject);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
else
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Status = ConnectPortDriver(Fdo, Fdo);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DPRINT("ConnectPortDriver() failed with status 0x%08lx\n", Status);
|
|
|
|
|
IoDetachDevice(DeviceExtension->LowerDevice);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
/* FIXME: why can't I cleanup without error? */
|
|
|
|
|
//IoDeleteDevice(Fdo);
|
|
|
|
|
return Status;
|
|
|
|
|
}
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Fdo->Flags &= ~DO_DEVICE_INITIALIZING;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
/* Register interface */
|
2005-10-31 00:14:45 +00:00
|
|
|
|
Status = IoRegisterDeviceInterface(
|
|
|
|
|
Pdo,
|
|
|
|
|
&GUID_DEVINTERFACE_MOUSE,
|
|
|
|
|
NULL,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
&DeviceExtension->InterfaceName);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
DPRINT("IoRegisterDeviceInterface() failed with status 0x%08lx\n", Status);
|
|
|
|
|
return Status;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return STATUS_SUCCESS;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static VOID NTAPI
|
2005-11-05 08:37:01 +00:00
|
|
|
|
ClassStartIo(
|
2005-10-31 00:14:45 +00:00
|
|
|
|
IN PDEVICE_OBJECT DeviceObject,
|
|
|
|
|
IN PIRP Irp)
|
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
PCLASS_DEVICE_EXTENSION DeviceExtension = DeviceObject->DeviceExtension;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
PIO_STACK_LOCATION Stack = IoGetCurrentIrpStackLocation(Irp);
|
|
|
|
|
|
|
|
|
|
ASSERT(DeviceExtension->Common.IsClassDO);
|
|
|
|
|
|
|
|
|
|
if (DeviceExtension->InputCount > 0)
|
|
|
|
|
{
|
|
|
|
|
KIRQL oldIrql;
|
|
|
|
|
|
|
|
|
|
KeAcquireSpinLock(&DeviceExtension->SpinLock, &oldIrql);
|
|
|
|
|
|
2005-11-18 17:28:19 +00:00
|
|
|
|
DPRINT("Mdl: %p, UserBuffer: %p, InputCount: %lu\n",
|
|
|
|
|
Irp->MdlAddress,
|
|
|
|
|
Irp->UserBuffer,
|
|
|
|
|
DeviceExtension->InputCount);
|
|
|
|
|
|
2005-10-31 16:46:46 +00:00
|
|
|
|
/* FIXME: use SEH */
|
|
|
|
|
RtlCopyMemory(
|
|
|
|
|
Irp->MdlAddress ? MmGetSystemAddressForMdlSafe(Irp->MdlAddress, NormalPagePriority) : Irp->UserBuffer,
|
2005-10-31 00:14:45 +00:00
|
|
|
|
DeviceExtension->PortData - DeviceExtension->InputCount,
|
|
|
|
|
sizeof(MOUSE_INPUT_DATA));
|
|
|
|
|
|
|
|
|
|
if (DeviceExtension->InputCount > 1)
|
|
|
|
|
{
|
|
|
|
|
RtlMoveMemory(
|
|
|
|
|
DeviceExtension->PortData - DeviceExtension->InputCount,
|
|
|
|
|
DeviceExtension->PortData - DeviceExtension->InputCount + 1,
|
|
|
|
|
(DeviceExtension->InputCount - 1) * sizeof(MOUSE_INPUT_DATA));
|
|
|
|
|
}
|
|
|
|
|
DeviceExtension->PortData--;
|
|
|
|
|
DeviceExtension->InputCount--;
|
|
|
|
|
DeviceExtension->ReadIsPending = FALSE;
|
|
|
|
|
|
|
|
|
|
/* Go to next packet and complete this request with STATUS_SUCCESS */
|
|
|
|
|
Irp->IoStatus.Status = STATUS_SUCCESS;
|
|
|
|
|
Irp->IoStatus.Information = sizeof(MOUSE_INPUT_DATA);
|
|
|
|
|
Stack->Parameters.Read.Length = sizeof(MOUSE_INPUT_DATA);
|
|
|
|
|
IoCompleteRequest(Irp, IO_MOUSE_INCREMENT);
|
|
|
|
|
|
|
|
|
|
IoStartNextPacket(DeviceObject, FALSE);
|
|
|
|
|
KeReleaseSpinLock(&DeviceExtension->SpinLock, oldIrql);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
DeviceExtension->ReadIsPending = TRUE;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static NTSTATUS
|
|
|
|
|
SearchForLegacyDrivers(
|
2005-10-31 16:46:46 +00:00
|
|
|
|
IN PDRIVER_OBJECT DriverObject,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
IN PCLASS_DRIVER_EXTENSION DriverExtension)
|
2005-10-31 00:14:45 +00:00
|
|
|
|
{
|
2005-10-31 16:46:46 +00:00
|
|
|
|
UNICODE_STRING DeviceMapKeyU = RTL_CONSTANT_STRING(L"\\REGISTRY\\MACHINE\\HARDWARE\\DEVICEMAP");
|
|
|
|
|
UNICODE_STRING PortBaseName = {0, };
|
|
|
|
|
PKEY_VALUE_BASIC_INFORMATION KeyValueInformation = NULL;
|
|
|
|
|
OBJECT_ATTRIBUTES ObjectAttributes;
|
|
|
|
|
HANDLE hDeviceMapKey = (HANDLE)-1;
|
|
|
|
|
HANDLE hPortKey = (HANDLE)-1;
|
|
|
|
|
ULONG Index = 0;
|
|
|
|
|
ULONG Size, ResultLength;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
NTSTATUS Status;
|
|
|
|
|
|
2005-10-31 16:46:46 +00:00
|
|
|
|
/* Create port base name, by replacing Class by Port at the end of the class base name */
|
|
|
|
|
Status = RtlDuplicateUnicodeString(
|
|
|
|
|
RTL_DUPLICATE_UNICODE_STRING_NULL_TERMINATE,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
&DriverExtension->DeviceBaseName,
|
2005-10-31 16:46:46 +00:00
|
|
|
|
&PortBaseName);
|
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
DPRINT("RtlDuplicateUnicodeString() failed with status 0x%08lx\n", Status);
|
|
|
|
|
goto cleanup;
|
|
|
|
|
}
|
|
|
|
|
PortBaseName.Length -= (sizeof(L"Class") - sizeof(UNICODE_NULL));
|
|
|
|
|
RtlAppendUnicodeToString(&PortBaseName, L"Port");
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
2005-10-31 16:46:46 +00:00
|
|
|
|
/* Allocate memory */
|
|
|
|
|
Size = sizeof(KEY_VALUE_BASIC_INFORMATION) + MAX_PATH;
|
|
|
|
|
KeyValueInformation = ExAllocatePool(PagedPool, Size);
|
|
|
|
|
if (!KeyValueInformation)
|
2005-10-31 00:14:45 +00:00
|
|
|
|
{
|
2005-10-31 16:46:46 +00:00
|
|
|
|
DPRINT("ExAllocatePool() failed\n");
|
|
|
|
|
Status = STATUS_INSUFFICIENT_RESOURCES;
|
|
|
|
|
goto cleanup;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
2005-10-31 16:46:46 +00:00
|
|
|
|
/* Open HKEY_LOCAL_MACHINE\HARDWARE\DEVICEMAP */
|
|
|
|
|
InitializeObjectAttributes(&ObjectAttributes, &DeviceMapKeyU, OBJ_KERNEL_HANDLE | OBJ_CASE_INSENSITIVE, NULL, NULL);
|
|
|
|
|
Status = ZwOpenKey(&hDeviceMapKey, 0, &ObjectAttributes);
|
2005-11-05 08:37:01 +00:00
|
|
|
|
if (Status == STATUS_OBJECT_NAME_NOT_FOUND)
|
|
|
|
|
{
|
|
|
|
|
DPRINT("HKLM\\HARDWARE\\DEVICEMAP is non-existent\n");
|
|
|
|
|
Status = STATUS_SUCCESS;
|
|
|
|
|
goto cleanup;
|
|
|
|
|
}
|
|
|
|
|
else if (!NT_SUCCESS(Status))
|
2005-10-31 00:14:45 +00:00
|
|
|
|
{
|
2005-10-31 16:46:46 +00:00
|
|
|
|
DPRINT("ZwOpenKey() failed with status 0x%08lx\n", Status);
|
|
|
|
|
goto cleanup;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
}
|
2005-10-31 16:46:46 +00:00
|
|
|
|
|
|
|
|
|
/* Open sub key */
|
|
|
|
|
InitializeObjectAttributes(&ObjectAttributes, &PortBaseName, OBJ_KERNEL_HANDLE | OBJ_CASE_INSENSITIVE, hDeviceMapKey, NULL);
|
|
|
|
|
Status = ZwOpenKey(&hPortKey, KEY_QUERY_VALUE, &ObjectAttributes);
|
2005-11-05 08:37:01 +00:00
|
|
|
|
if (Status == STATUS_OBJECT_NAME_NOT_FOUND)
|
|
|
|
|
{
|
|
|
|
|
DPRINT("HKLM\\HARDWARE\\DEVICEMAP\\%wZ is non-existent\n", &PortBaseName);
|
|
|
|
|
Status = STATUS_SUCCESS;
|
|
|
|
|
goto cleanup;
|
|
|
|
|
}
|
|
|
|
|
else if (!NT_SUCCESS(Status))
|
2005-10-31 00:14:45 +00:00
|
|
|
|
{
|
2005-10-31 16:46:46 +00:00
|
|
|
|
DPRINT("ZwOpenKey() failed with status 0x%08lx\n", Status);
|
|
|
|
|
goto cleanup;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
}
|
2005-10-31 16:46:46 +00:00
|
|
|
|
|
|
|
|
|
/* Read each value name */
|
|
|
|
|
while (ZwEnumerateValueKey(hPortKey, Index++, KeyValueBasicInformation, KeyValueInformation, Size, &ResultLength) == STATUS_SUCCESS)
|
|
|
|
|
{
|
|
|
|
|
UNICODE_STRING PortName;
|
|
|
|
|
PDEVICE_OBJECT PortDeviceObject = NULL;
|
|
|
|
|
PFILE_OBJECT FileObject = NULL;
|
|
|
|
|
|
|
|
|
|
PortName.Length = PortName.MaximumLength = KeyValueInformation->NameLength;
|
|
|
|
|
PortName.Buffer = KeyValueInformation->Name;
|
|
|
|
|
|
|
|
|
|
/* Open the device object pointer */
|
|
|
|
|
Status = IoGetDeviceObjectPointer(&PortName, FILE_READ_ATTRIBUTES, &FileObject, &PortDeviceObject);
|
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
DPRINT("IoGetDeviceObjectPointer(%wZ) failed with status 0x%08lx\n", Status);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Connect the port device object */
|
|
|
|
|
if (DriverExtension->ConnectMultiplePorts)
|
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Status = ConnectPortDriver(PortDeviceObject, DriverExtension->MainClassDeviceObject);
|
2005-10-31 16:46:46 +00:00
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
/* FIXME: Log the error */
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DPRINT("ConnectPortDriver() failed with status 0x%08lx\n", Status);
|
2005-10-31 16:46:46 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
PDEVICE_OBJECT ClassDO;
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Status = CreateClassDeviceObject(DriverObject, &ClassDO);
|
2005-10-31 16:46:46 +00:00
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
/* FIXME: Log the error */
|
|
|
|
|
DPRINT("CreatePointerClassDeviceObject() failed with status 0x%08lx\n", Status);
|
|
|
|
|
continue;
|
|
|
|
|
}
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Status = ConnectPortDriver(PortDeviceObject, ClassDO);
|
2005-10-31 16:46:46 +00:00
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
/* FIXME: Log the error */
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DPRINT("ConnectPortDriver() failed with status 0x%08lx\n", Status);
|
|
|
|
|
IoDeleteDevice(ClassDO);
|
2005-10-31 16:46:46 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if (Status == STATUS_NO_MORE_ENTRIES)
|
|
|
|
|
Status = STATUS_SUCCESS;
|
|
|
|
|
|
|
|
|
|
cleanup:
|
|
|
|
|
if (KeyValueInformation != NULL)
|
|
|
|
|
ExFreePool(KeyValueInformation);
|
|
|
|
|
if (hDeviceMapKey != (HANDLE)-1)
|
|
|
|
|
ZwClose(hDeviceMapKey);
|
|
|
|
|
if (hPortKey != (HANDLE)-1)
|
|
|
|
|
ZwClose(hPortKey);
|
|
|
|
|
return Status;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* Standard DriverEntry method.
|
|
|
|
|
*/
|
|
|
|
|
NTSTATUS NTAPI
|
|
|
|
|
DriverEntry(
|
|
|
|
|
IN PDRIVER_OBJECT DriverObject,
|
|
|
|
|
IN PUNICODE_STRING RegistryPath)
|
2000-10-18 09:11:01 +00:00
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
PCLASS_DRIVER_EXTENSION DriverExtension;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
ULONG i;
|
|
|
|
|
NTSTATUS Status;
|
|
|
|
|
|
|
|
|
|
Status = IoAllocateDriverObjectExtension(
|
|
|
|
|
DriverObject,
|
|
|
|
|
DriverObject,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
sizeof(CLASS_DRIVER_EXTENSION),
|
2005-10-31 00:14:45 +00:00
|
|
|
|
(PVOID*)&DriverExtension);
|
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
DPRINT("IoAllocateDriverObjectExtension() failed with status 0x%08lx\n", Status);
|
|
|
|
|
return Status;
|
|
|
|
|
}
|
2005-11-05 08:37:01 +00:00
|
|
|
|
RtlZeroMemory(DriverExtension, sizeof(CLASS_DRIVER_EXTENSION));
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
2005-11-09 11:16:43 +00:00
|
|
|
|
Status = RtlDuplicateUnicodeString(
|
|
|
|
|
RTL_DUPLICATE_UNICODE_STRING_NULL_TERMINATE,
|
|
|
|
|
RegistryPath,
|
|
|
|
|
&DriverExtension->RegistryPath);
|
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
DPRINT("RtlDuplicateUnicodeString() failed with status 0x%08lx\n", Status);
|
|
|
|
|
return Status;
|
|
|
|
|
}
|
|
|
|
|
|
2005-10-31 00:14:45 +00:00
|
|
|
|
Status = ReadRegistryEntries(RegistryPath, DriverExtension);
|
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
|
|
|
|
DPRINT("ReadRegistryEntries() failed with status 0x%08lx\n", Status);
|
|
|
|
|
return Status;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (DriverExtension->ConnectMultiplePorts == 1)
|
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
Status = CreateClassDeviceObject(
|
2005-10-31 00:14:45 +00:00
|
|
|
|
DriverObject,
|
2005-11-05 08:37:01 +00:00
|
|
|
|
&DriverExtension->MainClassDeviceObject);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
|
|
|
{
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DPRINT("CreateClassDeviceObject() failed with status 0x%08lx\n", Status);
|
2005-10-31 00:14:45 +00:00
|
|
|
|
return Status;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DriverObject->DriverExtension->AddDevice = ClassAddDevice;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
DriverObject->DriverUnload = DriverUnload;
|
|
|
|
|
|
2005-12-18 20:50:26 +00:00
|
|
|
|
for (i = 0; i <= IRP_MJ_MAXIMUM_FUNCTION; i++)
|
2005-10-31 00:14:45 +00:00
|
|
|
|
DriverObject->MajorFunction[i] = IrpStub;
|
|
|
|
|
|
2005-11-05 08:37:01 +00:00
|
|
|
|
DriverObject->MajorFunction[IRP_MJ_CREATE] = ClassCreate;
|
|
|
|
|
DriverObject->MajorFunction[IRP_MJ_CLOSE] = ClassClose;
|
|
|
|
|
DriverObject->MajorFunction[IRP_MJ_CLEANUP] = ClassCleanup;
|
|
|
|
|
DriverObject->MajorFunction[IRP_MJ_READ] = ClassRead;
|
|
|
|
|
DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = ClassDeviceControl;
|
|
|
|
|
DriverObject->DriverStartIo = ClassStartIo;
|
2005-10-31 00:14:45 +00:00
|
|
|
|
|
2005-11-09 11:16:43 +00:00
|
|
|
|
return STATUS_SUCCESS;
|
2000-10-18 09:11:01 +00:00
|
|
|
|
}
|