2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS Keyboard class driver
4 * FILE: drivers/kbdclass/kbdclass.c
5 * PURPOSE: Keyboard class driver
7 * PROGRAMMERS: Hervé Poussineau (hpoussin@reactos.org)
17 DriverUnload(IN PDRIVER_OBJECT DriverObject
)
19 // nothing to do here yet
24 IN PDEVICE_OBJECT DeviceObject
,
27 DPRINT("IRP_MJ_CREATE\n");
29 if (!((PCOMMON_DEVICE_EXTENSION
)DeviceObject
->DeviceExtension
)->IsClassDO
)
30 return ForwardIrpAndForget(DeviceObject
, Irp
);
32 /* FIXME: open all associated Port devices */
33 return STATUS_SUCCESS
;
38 IN PDEVICE_OBJECT DeviceObject
,
41 DPRINT("IRP_MJ_CLOSE\n");
43 if (!((PCOMMON_DEVICE_EXTENSION
)DeviceObject
->DeviceExtension
)->IsClassDO
)
44 return ForwardIrpAndForget(DeviceObject
, Irp
);
46 /* FIXME: close all associated Port devices */
47 return STATUS_SUCCESS
;
52 IN PDEVICE_OBJECT DeviceObject
,
55 DPRINT("IRP_MJ_CLEANUP\n");
57 if (!((PCOMMON_DEVICE_EXTENSION
)DeviceObject
->DeviceExtension
)->IsClassDO
)
58 return ForwardIrpAndForget(DeviceObject
, Irp
);
60 /* FIXME: cleanup all associated Port devices */
61 return STATUS_SUCCESS
;
66 IN PDEVICE_OBJECT DeviceObject
,
69 DPRINT("IRP_MJ_READ\n");
71 if (!((PCOMMON_DEVICE_EXTENSION
)DeviceObject
->DeviceExtension
)->IsClassDO
)
72 return ForwardIrpAndForget(DeviceObject
, Irp
);
74 if (IoGetCurrentIrpStackLocation(Irp
)->Parameters
.Read
.Length
< sizeof(KEYBOARD_INPUT_DATA
))
76 Irp
->IoStatus
.Status
= STATUS_BUFFER_TOO_SMALL
;
77 Irp
->IoStatus
.Information
= 0;
78 IoCompleteRequest(Irp
, IO_NO_INCREMENT
);
80 return STATUS_BUFFER_TOO_SMALL
;
83 IoMarkIrpPending(Irp
);
84 IoStartPacket(DeviceObject
, Irp
, NULL
, NULL
);
85 return STATUS_PENDING
;
90 IN PDEVICE_OBJECT DeviceObject
,
93 PCLASS_DEVICE_EXTENSION DeviceExtension
;
96 DPRINT("IRP_MJ_DEVICE_CONTROL\n");
98 if (!((PCOMMON_DEVICE_EXTENSION
)DeviceObject
->DeviceExtension
)->IsClassDO
)
99 return ForwardIrpAndForget(DeviceObject
, Irp
);
101 DeviceExtension
= (PCLASS_DEVICE_EXTENSION
)DeviceObject
->DeviceExtension
;
103 switch (IoGetCurrentIrpStackLocation(Irp
)->Parameters
.DeviceIoControl
.IoControlCode
)
105 case IOCTL_KEYBOARD_QUERY_ATTRIBUTES
:
106 case IOCTL_KEYBOARD_QUERY_INDICATOR_TRANSLATION
:
107 case IOCTL_KEYBOARD_QUERY_INDICATORS
:
108 case IOCTL_KEYBOARD_QUERY_TYPEMATIC
:
109 case IOCTL_KEYBOARD_SET_INDICATORS
:
110 case IOCTL_KEYBOARD_SET_TYPEMATIC
: /* not in MSDN, would seem logical */
111 /* FIXME: send it to all associated Port devices */
112 Status
= STATUS_NOT_SUPPORTED
;
115 DPRINT1("IRP_MJ_DEVICE_CONTROL / unknown I/O control code 0x%lx\n",
116 IoGetCurrentIrpStackLocation(Irp
)->Parameters
.DeviceIoControl
.IoControlCode
);
117 Status
= STATUS_NOT_SUPPORTED
;
120 Irp
->IoStatus
.Status
= Status
;
121 Irp
->IoStatus
.Information
= 0;
122 IoCompleteRequest(Irp
, IO_NO_INCREMENT
);
127 static NTSTATUS NTAPI
129 IN PDEVICE_OBJECT DeviceObject
,
132 NTSTATUS Status
= STATUS_NOT_SUPPORTED
;
134 if (!((PCOMMON_DEVICE_EXTENSION
)DeviceObject
->DeviceExtension
)->IsClassDO
)
136 /* Forward some IRPs to lower device */
137 switch (IoGetCurrentIrpStackLocation(Irp
)->MajorFunction
)
139 case IRP_MJ_INTERNAL_DEVICE_CONTROL
:
140 return ForwardIrpAndForget(DeviceObject
, Irp
);
143 DPRINT1("Port DO stub for major function 0x%lx\n",
144 IoGetCurrentIrpStackLocation(Irp
)->MajorFunction
);
146 Status
= Irp
->IoStatus
.Status
;
152 DPRINT1("Class DO stub for major function 0x%lx\n",
153 IoGetCurrentIrpStackLocation(Irp
)->MajorFunction
);
155 Status
= Irp
->IoStatus
.Status
;
158 Irp
->IoStatus
.Status
= Status
;
159 IoCompleteRequest(Irp
, IO_NO_INCREMENT
);
165 IN PUNICODE_STRING RegistryPath
,
166 IN PCLASS_DRIVER_EXTENSION DriverExtension
)
168 UNICODE_STRING ParametersRegistryKey
;
169 RTL_QUERY_REGISTRY_TABLE Parameters
[4];
172 ULONG DefaultConnectMultiplePorts
= 0;
173 ULONG DefaultDataQueueSize
= 0x64;
174 UNICODE_STRING DefaultDeviceBaseName
= RTL_CONSTANT_STRING(L
"KeyboardClass");
176 ParametersRegistryKey
.Length
= 0;
177 ParametersRegistryKey
.MaximumLength
= RegistryPath
->Length
+ sizeof(L
"\\Parameters") + sizeof(UNICODE_NULL
);
178 ParametersRegistryKey
.Buffer
= ExAllocatePool(PagedPool
, ParametersRegistryKey
.MaximumLength
);
179 if (!ParametersRegistryKey
.Buffer
)
181 DPRINT("ExAllocatePool() failed\n");
182 return STATUS_INSUFFICIENT_RESOURCES
;
184 RtlCopyUnicodeString(&ParametersRegistryKey
, RegistryPath
);
185 RtlAppendUnicodeToString(&ParametersRegistryKey
, L
"\\Parameters");
186 ParametersRegistryKey
.Buffer
[ParametersRegistryKey
.Length
/ sizeof(WCHAR
)] = UNICODE_NULL
;
188 RtlZeroMemory(Parameters
, sizeof(Parameters
));
190 Parameters
[0].Flags
= RTL_QUERY_REGISTRY_DIRECT
| RTL_REGISTRY_OPTIONAL
;
191 Parameters
[0].Name
= L
"ConnectMultiplePorts";
192 Parameters
[0].EntryContext
= &DriverExtension
->ConnectMultiplePorts
;
193 Parameters
[0].DefaultType
= REG_DWORD
;
194 Parameters
[0].DefaultData
= &DefaultConnectMultiplePorts
;
195 Parameters
[0].DefaultLength
= sizeof(ULONG
);
197 Parameters
[1].Flags
= RTL_QUERY_REGISTRY_DIRECT
| RTL_REGISTRY_OPTIONAL
;
198 Parameters
[1].Name
= L
"KeyboardDataQueueSize";
199 Parameters
[1].EntryContext
= &DriverExtension
->DataQueueSize
;
200 Parameters
[1].DefaultType
= REG_DWORD
;
201 Parameters
[1].DefaultData
= &DefaultDataQueueSize
;
202 Parameters
[1].DefaultLength
= sizeof(ULONG
);
204 Parameters
[2].Flags
= RTL_QUERY_REGISTRY_DIRECT
| RTL_REGISTRY_OPTIONAL
;
205 Parameters
[2].Name
= L
"KeyboardDeviceBaseName";
206 Parameters
[2].EntryContext
= &DriverExtension
->DeviceBaseName
;
207 Parameters
[2].DefaultType
= REG_SZ
;
208 Parameters
[2].DefaultData
= &DefaultDeviceBaseName
;
209 Parameters
[2].DefaultLength
= sizeof(ULONG
);
211 Status
= RtlQueryRegistryValues(
212 RTL_REGISTRY_ABSOLUTE
,
213 ParametersRegistryKey
.Buffer
,
218 if (NT_SUCCESS(Status
))
221 if (DriverExtension
->ConnectMultiplePorts
!= 0
222 && DriverExtension
->ConnectMultiplePorts
!= 1)
224 DriverExtension
->ConnectMultiplePorts
= DefaultConnectMultiplePorts
;
226 if (DriverExtension
->DataQueueSize
== 0)
228 DriverExtension
->DataQueueSize
= DefaultDataQueueSize
;
231 else if (Status
== STATUS_OBJECT_NAME_NOT_FOUND
)
233 /* Registry path doesn't exist. Set defaults */
234 DriverExtension
->ConnectMultiplePorts
= DefaultConnectMultiplePorts
;
235 DriverExtension
->DataQueueSize
= DefaultDataQueueSize
;
236 Status
= RtlDuplicateUnicodeString(
237 RTL_DUPLICATE_UNICODE_STRING_NULL_TERMINATE
,
238 &DefaultDeviceBaseName
,
239 &DriverExtension
->DeviceBaseName
);
246 CreateClassDeviceObject(
247 IN PDRIVER_OBJECT DriverObject
,
248 OUT PDEVICE_OBJECT
*ClassDO OPTIONAL
)
250 PCLASS_DRIVER_EXTENSION DriverExtension
;
251 UNICODE_STRING SymbolicLinkName
= RTL_CONSTANT_STRING(L
"\\??\\Keyboard");
254 UNICODE_STRING DeviceNameU
;
255 PWSTR DeviceIdW
= NULL
; /* Pointer into DeviceNameU.Buffer */
257 PCLASS_DEVICE_EXTENSION DeviceExtension
;
260 DPRINT("CreateClassDeviceObject(0x%p)\n", DriverObject
);
262 /* Create new device object */
263 DriverExtension
= IoGetDriverObjectExtension(DriverObject
, DriverObject
);
264 DeviceNameU
.Length
= 0;
265 DeviceNameU
.MaximumLength
=
266 wcslen(L
"\\Device\\") * sizeof(WCHAR
) /* "\Device\" */
267 + DriverExtension
->DeviceBaseName
.Length
/* "KeyboardClass" */
268 + 4 * sizeof(WCHAR
) /* Id between 0 and 9999 */
269 + sizeof(UNICODE_NULL
); /* Final NULL char */
270 DeviceNameU
.Buffer
= ExAllocatePool(PagedPool
, DeviceNameU
.MaximumLength
);
271 if (!DeviceNameU
.Buffer
)
273 DPRINT("ExAllocatePool() failed\n");
274 return STATUS_INSUFFICIENT_RESOURCES
;
276 Status
= RtlAppendUnicodeToString(&DeviceNameU
, L
"\\Device\\");
277 if (!NT_SUCCESS(Status
))
279 DPRINT("RtlAppendUnicodeToString() failed with status 0x%08lx\n", Status
);
282 Status
= RtlAppendUnicodeStringToString(&DeviceNameU
, &DriverExtension
->DeviceBaseName
);
283 if (!NT_SUCCESS(Status
))
285 DPRINT("RtlAppendUnicodeStringToString() failed with status 0x%08lx\n", Status
);
288 PrefixLength
= DeviceNameU
.MaximumLength
- 4 * sizeof(WCHAR
) - sizeof(UNICODE_NULL
);
289 DeviceIdW
= &DeviceNameU
.Buffer
[PrefixLength
/ sizeof(WCHAR
)];
290 while (DeviceId
< 9999)
292 DeviceNameU
.Length
= PrefixLength
+ swprintf(DeviceIdW
, L
"%lu", DeviceId
) * sizeof(WCHAR
);
293 Status
= IoCreateDevice(
295 sizeof(CLASS_DEVICE_EXTENSION
),
297 FILE_DEVICE_KEYBOARD
,
298 FILE_DEVICE_SECURE_OPEN
,
301 if (NT_SUCCESS(Status
))
303 else if (Status
!= STATUS_OBJECT_NAME_COLLISION
)
305 DPRINT("IoCreateDevice() failed with status 0x%08lx\n", Status
);
310 DPRINT("Too much devices starting with '\\Device\\%wZ'\n", &DriverExtension
->DeviceBaseName
);
311 Status
= STATUS_UNSUCCESSFUL
;
313 if (!NT_SUCCESS(Status
))
315 ExFreePool(DeviceNameU
.Buffer
);
319 DeviceExtension
= (PCLASS_DEVICE_EXTENSION
)Fdo
->DeviceExtension
;
320 RtlZeroMemory(DeviceExtension
, sizeof(CLASS_DEVICE_EXTENSION
));
321 DeviceExtension
->Common
.IsClassDO
= TRUE
;
322 DeviceExtension
->DriverExtension
= DriverExtension
;
323 KeInitializeSpinLock(&(DeviceExtension
->SpinLock
));
324 DeviceExtension
->ReadIsPending
= FALSE
;
325 DeviceExtension
->InputCount
= 0;
326 DeviceExtension
->PortData
= ExAllocatePool(NonPagedPool
, DeviceExtension
->DriverExtension
->DataQueueSize
* sizeof(KEYBOARD_INPUT_DATA
));
327 Fdo
->Flags
|= DO_POWER_PAGABLE
| DO_BUFFERED_IO
;
328 Fdo
->Flags
&= ~DO_DEVICE_INITIALIZING
;
330 /* FIXME: create registry entry in HKEY_LOCAL_MACHINE\HARDWARE\DEVICEMAP */
332 /* HACK: 1st stage setup needs a keyboard to open it in user-mode
333 * Create a link to user space... */
334 IoCreateSymbolicLink(&SymbolicLinkName
, &DeviceNameU
);
336 ExFreePool(DeviceNameU
.Buffer
);
341 return STATUS_SUCCESS
;
346 IN PDEVICE_OBJECT ClassDeviceObject
,
347 IN OUT PKEYBOARD_INPUT_DATA DataStart
,
348 IN PKEYBOARD_INPUT_DATA DataEnd
,
349 IN OUT PULONG ConsumedCount
)
351 PCLASS_DEVICE_EXTENSION ClassDeviceExtension
= ClassDeviceObject
->DeviceExtension
;
354 PIO_STACK_LOCATION Stack
;
355 ULONG InputCount
= DataEnd
- DataStart
;
358 ASSERT(ClassDeviceExtension
->Common
.IsClassDO
);
360 DPRINT("ClassCallback()\n");
361 /* A filter driver might have consumed all the data already; I'm
362 * not sure if they are supposed to move the packets when they
363 * consume them though.
365 if (ClassDeviceExtension
->ReadIsPending
== TRUE
&& InputCount
)
367 Irp
= ClassDeviceObject
->CurrentIrp
;
368 ClassDeviceObject
->CurrentIrp
= NULL
;
369 Stack
= IoGetCurrentIrpStackLocation(Irp
);
371 /* A read request is waiting for input, so go straight to it */
374 Irp
->MdlAddress
? MmGetSystemAddressForMdlSafe(Irp
->MdlAddress
, NormalPagePriority
) : Irp
->AssociatedIrp
.SystemBuffer
,
376 sizeof(KEYBOARD_INPUT_DATA
));
378 /* Go to next packet and complete this request with STATUS_SUCCESS */
379 Irp
->IoStatus
.Status
= STATUS_SUCCESS
;
380 Irp
->IoStatus
.Information
= sizeof(KEYBOARD_INPUT_DATA
);
381 Stack
->Parameters
.Read
.Length
= sizeof(KEYBOARD_INPUT_DATA
);
383 ClassDeviceExtension
->ReadIsPending
= FALSE
;
385 /* Skip the packet we just sent away */
391 /* If we have data from the port driver and a higher service to send the data to */
394 KeAcquireSpinLock(&ClassDeviceExtension
->SpinLock
, &OldIrql
);
396 if (ClassDeviceExtension
->InputCount
+ InputCount
> ClassDeviceExtension
->DriverExtension
->DataQueueSize
)
397 ReadSize
= ClassDeviceExtension
->DriverExtension
->DataQueueSize
- ClassDeviceExtension
->InputCount
;
399 ReadSize
= InputCount
;
402 * FIXME: If we exceed the buffer, data gets thrown away.. better
407 * Move the input data from the port data queue to our class data
411 ClassDeviceExtension
->PortData
,
413 sizeof(KEYBOARD_INPUT_DATA
) * ReadSize
);
415 /* Move the pointer and counter up */
416 ClassDeviceExtension
->PortData
+= ReadSize
;
417 ClassDeviceExtension
->InputCount
+= ReadSize
;
419 KeReleaseSpinLock(&ClassDeviceExtension
->SpinLock
, OldIrql
);
420 (*ConsumedCount
) += ReadSize
;
424 DPRINT("ClassCallBack() entered, InputCount = %lu - DOING NOTHING\n", InputCount
);
429 IoStartNextPacket(ClassDeviceObject
, FALSE
);
430 IoCompleteRequest(Irp
, IO_KEYBOARD_INCREMENT
);
433 DPRINT("Leaving ClassCallback()\n");
437 /* Send IOCTL_INTERNAL_*_CONNECT to port */
440 IN PDEVICE_OBJECT PortDO
,
441 IN PDEVICE_OBJECT ClassDO
)
445 IO_STATUS_BLOCK IoStatus
;
446 CONNECT_DATA ConnectData
;
449 KeInitializeEvent(&Event
, NotificationEvent
, FALSE
);
451 ConnectData
.ClassDeviceObject
= ClassDO
;
452 ConnectData
.ClassService
= ClassCallback
;
454 Irp
= IoBuildDeviceIoControlRequest(IOCTL_INTERNAL_KEYBOARD_CONNECT
,
456 &ConnectData
, sizeof(CONNECT_DATA
),
458 TRUE
, &Event
, &IoStatus
);
460 Status
= IoCallDriver(PortDO
, Irp
);
462 if (Status
== STATUS_PENDING
)
463 KeWaitForSingleObject(&Event
, Suspended
, KernelMode
, FALSE
, NULL
);
465 IoStatus
.Status
= Status
;
467 if (NT_SUCCESS(Status
))
468 ObReferenceObject(PortDO
);
470 return IoStatus
.Status
;
473 static NTSTATUS NTAPI
475 IN PDRIVER_OBJECT DriverObject
,
476 IN PDEVICE_OBJECT Pdo
)
478 PCLASS_DRIVER_EXTENSION DriverExtension
;
480 PPORT_DEVICE_EXTENSION DeviceExtension
;
483 DPRINT("ClassAddDevice called. Pdo = 0x%p\n", Pdo
);
486 return STATUS_SUCCESS
;
488 DriverExtension
= IoGetDriverObjectExtension(DriverObject
, DriverObject
);
490 /* Create new device object */
491 Status
= IoCreateDevice(
493 sizeof(PORT_DEVICE_EXTENSION
),
496 FILE_DEVICE_SECURE_OPEN
,
499 if (!NT_SUCCESS(Status
))
501 DPRINT("IoCreateDevice() failed with status 0x%08lx\n", Status
);
505 DeviceExtension
= (PPORT_DEVICE_EXTENSION
)Fdo
->DeviceExtension
;
506 RtlZeroMemory(DeviceExtension
, sizeof(CLASS_DEVICE_EXTENSION
));
507 DeviceExtension
->Common
.IsClassDO
= FALSE
;
508 DeviceExtension
->PnpState
= dsStopped
;
509 Fdo
->Flags
|= DO_POWER_PAGABLE
;
510 Status
= IoAttachDeviceToDeviceStackSafe(Fdo
, Pdo
, &DeviceExtension
->LowerDevice
);
511 if (!NT_SUCCESS(Status
))
513 DPRINT("IoAttachDeviceToDeviceStackSafe() failed with status 0x%08lx\n", Status
);
517 Fdo
->Flags
|= DO_BUFFERED_IO
;
519 if (DriverExtension
->ConnectMultiplePorts
)
520 Status
= ConnectPortDriver(Fdo
, DriverExtension
->MainClassDeviceObject
);
522 Status
= ConnectPortDriver(Fdo
, Fdo
);
523 if (!NT_SUCCESS(Status
))
525 DPRINT("ConnectPortDriver() failed with status 0x%08lx\n", Status
);
526 IoDetachDevice(DeviceExtension
->LowerDevice
);
527 /* FIXME: why can't I cleanup without error? */
528 //IoDeleteDevice(Fdo);
531 Fdo
->Flags
&= ~DO_DEVICE_INITIALIZING
;
533 /* Register interface */
534 Status
= IoRegisterDeviceInterface(
536 &GUID_DEVINTERFACE_KEYBOARD
,
538 &DeviceExtension
->InterfaceName
);
539 if (!NT_SUCCESS(Status
))
541 DPRINT("IoRegisterDeviceInterface() failed with status 0x%08lx\n", Status
);
545 return STATUS_SUCCESS
;
550 IN PDEVICE_OBJECT DeviceObject
,
553 PCLASS_DEVICE_EXTENSION DeviceExtension
= DeviceObject
->DeviceExtension
;
554 PIO_STACK_LOCATION Stack
= IoGetCurrentIrpStackLocation(Irp
);
556 ASSERT(DeviceExtension
->Common
.IsClassDO
);
558 if (DeviceExtension
->InputCount
> 0)
562 KeAcquireSpinLock(&DeviceExtension
->SpinLock
, &oldIrql
);
566 Irp
->MdlAddress
? MmGetSystemAddressForMdlSafe(Irp
->MdlAddress
, NormalPagePriority
) : Irp
->UserBuffer
,
567 DeviceExtension
->PortData
- DeviceExtension
->InputCount
,
568 sizeof(KEYBOARD_INPUT_DATA
));
570 if (DeviceExtension
->InputCount
> 1)
573 DeviceExtension
->PortData
- DeviceExtension
->InputCount
,
574 DeviceExtension
->PortData
- DeviceExtension
->InputCount
+ 1,
575 (DeviceExtension
->InputCount
- 1) * sizeof(KEYBOARD_INPUT_DATA
));
577 DeviceExtension
->PortData
--;
578 DeviceExtension
->InputCount
--;
579 DeviceExtension
->ReadIsPending
= FALSE
;
581 /* Go to next packet and complete this request with STATUS_SUCCESS */
582 Irp
->IoStatus
.Status
= STATUS_SUCCESS
;
583 Irp
->IoStatus
.Information
= sizeof(KEYBOARD_INPUT_DATA
);
584 Stack
->Parameters
.Read
.Length
= sizeof(KEYBOARD_INPUT_DATA
);
585 IoCompleteRequest(Irp
, IO_KEYBOARD_INCREMENT
);
587 IoStartNextPacket(DeviceObject
, FALSE
);
588 KeReleaseSpinLock(&DeviceExtension
->SpinLock
, oldIrql
);
592 DeviceExtension
->ReadIsPending
= TRUE
;
597 SearchForLegacyDrivers(
598 IN PDRIVER_OBJECT DriverObject
,
599 IN PCLASS_DRIVER_EXTENSION DriverExtension
)
601 UNICODE_STRING DeviceMapKeyU
= RTL_CONSTANT_STRING(L
"\\REGISTRY\\MACHINE\\HARDWARE\\DEVICEMAP");
602 UNICODE_STRING PortBaseName
= {0, };
603 PKEY_VALUE_BASIC_INFORMATION KeyValueInformation
= NULL
;
604 OBJECT_ATTRIBUTES ObjectAttributes
;
605 HANDLE hDeviceMapKey
= (HANDLE
)-1;
606 HANDLE hPortKey
= (HANDLE
)-1;
608 ULONG Size
, ResultLength
;
611 /* Create port base name, by replacing Class by Port at the end of the class base name */
612 Status
= RtlDuplicateUnicodeString(
613 RTL_DUPLICATE_UNICODE_STRING_NULL_TERMINATE
,
614 &DriverExtension
->DeviceBaseName
,
616 if (!NT_SUCCESS(Status
))
618 DPRINT("RtlDuplicateUnicodeString() failed with status 0x%08lx\n", Status
);
621 PortBaseName
.Length
-= (sizeof(L
"Class") - sizeof(UNICODE_NULL
));
622 RtlAppendUnicodeToString(&PortBaseName
, L
"Port");
624 /* Allocate memory */
625 Size
= sizeof(KEY_VALUE_BASIC_INFORMATION
) + MAX_PATH
;
626 KeyValueInformation
= ExAllocatePool(PagedPool
, Size
);
627 if (!KeyValueInformation
)
629 DPRINT("ExAllocatePool() failed\n");
630 Status
= STATUS_INSUFFICIENT_RESOURCES
;
634 /* Open HKEY_LOCAL_MACHINE\HARDWARE\DEVICEMAP */
635 InitializeObjectAttributes(&ObjectAttributes
, &DeviceMapKeyU
, OBJ_KERNEL_HANDLE
| OBJ_CASE_INSENSITIVE
, NULL
, NULL
);
636 Status
= ZwOpenKey(&hDeviceMapKey
, 0, &ObjectAttributes
);
637 if (Status
== STATUS_OBJECT_NAME_NOT_FOUND
)
639 DPRINT("HKLM\\HARDWARE\\DEVICEMAP is non-existent\n");
640 Status
= STATUS_SUCCESS
;
643 else if (!NT_SUCCESS(Status
))
645 DPRINT("ZwOpenKey() failed with status 0x%08lx\n", Status
);
650 InitializeObjectAttributes(&ObjectAttributes
, &PortBaseName
, OBJ_KERNEL_HANDLE
| OBJ_CASE_INSENSITIVE
, hDeviceMapKey
, NULL
);
651 Status
= ZwOpenKey(&hPortKey
, KEY_QUERY_VALUE
, &ObjectAttributes
);
652 if (Status
== STATUS_OBJECT_NAME_NOT_FOUND
)
654 DPRINT("HKLM\\HARDWARE\\DEVICEMAP\\%wZ is non-existent\n", &PortBaseName
);
655 Status
= STATUS_SUCCESS
;
658 else if (!NT_SUCCESS(Status
))
660 DPRINT("ZwOpenKey() failed with status 0x%08lx\n", Status
);
664 /* Read each value name */
665 while (ZwEnumerateValueKey(hPortKey
, Index
++, KeyValueBasicInformation
, KeyValueInformation
, Size
, &ResultLength
) == STATUS_SUCCESS
)
667 UNICODE_STRING PortName
;
668 PDEVICE_OBJECT PortDeviceObject
= NULL
;
669 PFILE_OBJECT FileObject
= NULL
;
671 PortName
.Length
= PortName
.MaximumLength
= KeyValueInformation
->NameLength
;
672 PortName
.Buffer
= KeyValueInformation
->Name
;
674 /* Open the device object pointer */
675 Status
= IoGetDeviceObjectPointer(&PortName
, FILE_READ_ATTRIBUTES
, &FileObject
, &PortDeviceObject
);
676 if (!NT_SUCCESS(Status
))
678 DPRINT("IoGetDeviceObjectPointer(%wZ) failed with status 0x%08lx\n", Status
);
681 /* Connect the port device object */
682 if (DriverExtension
->ConnectMultiplePorts
)
684 Status
= ConnectPortDriver(PortDeviceObject
, DriverExtension
->MainClassDeviceObject
);
685 if (!NT_SUCCESS(Status
))
687 /* FIXME: Log the error */
688 DPRINT("ConnectPortDriver() failed with status 0x%08lx\n", Status
);
689 ObDereferenceObject(PortDeviceObject
);
694 PDEVICE_OBJECT ClassDO
;
695 Status
= CreateClassDeviceObject(DriverObject
, &ClassDO
);
696 if (!NT_SUCCESS(Status
))
698 /* FIXME: Log the error */
699 DPRINT("CreatePointerClassDeviceObject() failed with status 0x%08lx\n", Status
);
700 ObDereferenceObject(PortDeviceObject
);
703 Status
= ConnectPortDriver(PortDeviceObject
, ClassDO
);
704 if (!NT_SUCCESS(Status
))
706 /* FIXME: Log the error */
707 DPRINT("ConnectPortDriver() failed with status 0x%08lx\n", Status
);
708 ObDereferenceObject(PortDeviceObject
);
709 IoDeleteDevice(ClassDO
);
713 if (Status
== STATUS_NO_MORE_ENTRIES
)
714 Status
= STATUS_SUCCESS
;
717 if (KeyValueInformation
!= NULL
)
718 ExFreePool(KeyValueInformation
);
719 if (hDeviceMapKey
!= (HANDLE
)-1)
720 ZwClose(hDeviceMapKey
);
721 if (hPortKey
!= (HANDLE
)-1)
727 * Standard DriverEntry method.
731 IN PDRIVER_OBJECT DriverObject
,
732 IN PUNICODE_STRING RegistryPath
)
734 PCLASS_DRIVER_EXTENSION DriverExtension
;
738 Status
= IoAllocateDriverObjectExtension(
741 sizeof(CLASS_DRIVER_EXTENSION
),
742 (PVOID
*)&DriverExtension
);
743 if (!NT_SUCCESS(Status
))
745 DPRINT("IoAllocateDriverObjectExtension() failed with status 0x%08lx\n", Status
);
748 RtlZeroMemory(DriverExtension
, sizeof(CLASS_DRIVER_EXTENSION
));
750 Status
= ReadRegistryEntries(RegistryPath
, DriverExtension
);
751 if (!NT_SUCCESS(Status
))
753 DPRINT("ReadRegistryEntries() failed with status 0x%08lx\n", Status
);
757 if (DriverExtension
->ConnectMultiplePorts
== 1)
759 Status
= CreateClassDeviceObject(
761 &DriverExtension
->MainClassDeviceObject
);
762 if (!NT_SUCCESS(Status
))
764 DPRINT("CreateClassDeviceObject() failed with status 0x%08lx\n", Status
);
769 DriverObject
->DriverExtension
->AddDevice
= ClassAddDevice
;
770 DriverObject
->DriverUnload
= DriverUnload
;
772 for (i
= 0; i
< IRP_MJ_MAXIMUM_FUNCTION
; i
++)
773 DriverObject
->MajorFunction
[i
] = IrpStub
;
775 DriverObject
->MajorFunction
[IRP_MJ_CREATE
] = ClassCreate
;
776 DriverObject
->MajorFunction
[IRP_MJ_CLOSE
] = ClassClose
;
777 DriverObject
->MajorFunction
[IRP_MJ_CLEANUP
] = ClassCleanup
;
778 DriverObject
->MajorFunction
[IRP_MJ_READ
] = ClassRead
;
779 DriverObject
->MajorFunction
[IRP_MJ_DEVICE_CONTROL
] = ClassDeviceControl
;
780 DriverObject
->DriverStartIo
= ClassStartIo
;
782 Status
= SearchForLegacyDrivers(DriverObject
, DriverExtension
);