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
);
485 DriverExtension
= IoGetDriverObjectExtension(DriverObject
, DriverObject
);
487 /* Create new device object */
488 Status
= IoCreateDevice(
490 sizeof(PORT_DEVICE_EXTENSION
),
493 FILE_DEVICE_SECURE_OPEN
,
496 if (!NT_SUCCESS(Status
))
498 DPRINT("IoCreateDevice() failed with status 0x%08lx\n", Status
);
502 DeviceExtension
= (PPORT_DEVICE_EXTENSION
)Fdo
->DeviceExtension
;
503 RtlZeroMemory(DeviceExtension
, sizeof(CLASS_DEVICE_EXTENSION
));
504 DeviceExtension
->Common
.IsClassDO
= FALSE
;
505 DeviceExtension
->PnpState
= dsStopped
;
506 Fdo
->Flags
|= DO_POWER_PAGABLE
;
507 Status
= IoAttachDeviceToDeviceStackSafe(Fdo
, Pdo
, &DeviceExtension
->LowerDevice
);
508 if (!NT_SUCCESS(Status
))
510 DPRINT("IoAttachDeviceToDeviceStackSafe() failed with status 0x%08lx\n", Status
);
514 Fdo
->Flags
|= DO_BUFFERED_IO
;
516 if (DriverExtension
->ConnectMultiplePorts
)
517 Status
= ConnectPortDriver(Fdo
, DriverExtension
->MainClassDeviceObject
);
519 Status
= ConnectPortDriver(Fdo
, Fdo
);
520 if (!NT_SUCCESS(Status
))
522 DPRINT("ConnectPortDriver() failed with status 0x%08lx\n", Status
);
523 IoDetachDevice(DeviceExtension
->LowerDevice
);
524 /* FIXME: why can't I cleanup without error? */
525 //IoDeleteDevice(Fdo);
528 Fdo
->Flags
&= ~DO_DEVICE_INITIALIZING
;
530 /* Register interface */
531 Status
= IoRegisterDeviceInterface(
533 &GUID_DEVINTERFACE_KEYBOARD
,
535 &DeviceExtension
->InterfaceName
);
536 if (!NT_SUCCESS(Status
))
538 DPRINT("IoRegisterDeviceInterface() failed with status 0x%08lx\n", Status
);
542 return STATUS_SUCCESS
;
547 IN PDEVICE_OBJECT DeviceObject
,
550 PCLASS_DEVICE_EXTENSION DeviceExtension
= DeviceObject
->DeviceExtension
;
551 PIO_STACK_LOCATION Stack
= IoGetCurrentIrpStackLocation(Irp
);
553 ASSERT(DeviceExtension
->Common
.IsClassDO
);
555 if (DeviceExtension
->InputCount
> 0)
559 KeAcquireSpinLock(&DeviceExtension
->SpinLock
, &oldIrql
);
563 Irp
->MdlAddress
? MmGetSystemAddressForMdlSafe(Irp
->MdlAddress
, NormalPagePriority
) : Irp
->UserBuffer
,
564 DeviceExtension
->PortData
- DeviceExtension
->InputCount
,
565 sizeof(KEYBOARD_INPUT_DATA
));
567 if (DeviceExtension
->InputCount
> 1)
570 DeviceExtension
->PortData
- DeviceExtension
->InputCount
,
571 DeviceExtension
->PortData
- DeviceExtension
->InputCount
+ 1,
572 (DeviceExtension
->InputCount
- 1) * sizeof(KEYBOARD_INPUT_DATA
));
574 DeviceExtension
->PortData
--;
575 DeviceExtension
->InputCount
--;
576 DeviceExtension
->ReadIsPending
= FALSE
;
578 /* Go to next packet and complete this request with STATUS_SUCCESS */
579 Irp
->IoStatus
.Status
= STATUS_SUCCESS
;
580 Irp
->IoStatus
.Information
= sizeof(KEYBOARD_INPUT_DATA
);
581 Stack
->Parameters
.Read
.Length
= sizeof(KEYBOARD_INPUT_DATA
);
582 IoCompleteRequest(Irp
, IO_KEYBOARD_INCREMENT
);
584 IoStartNextPacket(DeviceObject
, FALSE
);
585 KeReleaseSpinLock(&DeviceExtension
->SpinLock
, oldIrql
);
589 DeviceExtension
->ReadIsPending
= TRUE
;
594 SearchForLegacyDrivers(
595 IN PDRIVER_OBJECT DriverObject
,
596 IN PCLASS_DRIVER_EXTENSION DriverExtension
)
598 UNICODE_STRING DeviceMapKeyU
= RTL_CONSTANT_STRING(L
"\\REGISTRY\\MACHINE\\HARDWARE\\DEVICEMAP");
599 UNICODE_STRING PortBaseName
= {0, };
600 PKEY_VALUE_BASIC_INFORMATION KeyValueInformation
= NULL
;
601 OBJECT_ATTRIBUTES ObjectAttributes
;
602 HANDLE hDeviceMapKey
= (HANDLE
)-1;
603 HANDLE hPortKey
= (HANDLE
)-1;
605 ULONG Size
, ResultLength
;
608 /* Create port base name, by replacing Class by Port at the end of the class base name */
609 Status
= RtlDuplicateUnicodeString(
610 RTL_DUPLICATE_UNICODE_STRING_NULL_TERMINATE
,
611 &DriverExtension
->DeviceBaseName
,
613 if (!NT_SUCCESS(Status
))
615 DPRINT("RtlDuplicateUnicodeString() failed with status 0x%08lx\n", Status
);
618 PortBaseName
.Length
-= (sizeof(L
"Class") - sizeof(UNICODE_NULL
));
619 RtlAppendUnicodeToString(&PortBaseName
, L
"Port");
621 /* Allocate memory */
622 Size
= sizeof(KEY_VALUE_BASIC_INFORMATION
) + MAX_PATH
;
623 KeyValueInformation
= ExAllocatePool(PagedPool
, Size
);
624 if (!KeyValueInformation
)
626 DPRINT("ExAllocatePool() failed\n");
627 Status
= STATUS_INSUFFICIENT_RESOURCES
;
631 /* Open HKEY_LOCAL_MACHINE\HARDWARE\DEVICEMAP */
632 InitializeObjectAttributes(&ObjectAttributes
, &DeviceMapKeyU
, OBJ_KERNEL_HANDLE
| OBJ_CASE_INSENSITIVE
, NULL
, NULL
);
633 Status
= ZwOpenKey(&hDeviceMapKey
, 0, &ObjectAttributes
);
634 if (Status
== STATUS_OBJECT_NAME_NOT_FOUND
)
636 DPRINT("HKLM\\HARDWARE\\DEVICEMAP is non-existent\n");
637 Status
= STATUS_SUCCESS
;
640 else if (!NT_SUCCESS(Status
))
642 DPRINT("ZwOpenKey() failed with status 0x%08lx\n", Status
);
647 InitializeObjectAttributes(&ObjectAttributes
, &PortBaseName
, OBJ_KERNEL_HANDLE
| OBJ_CASE_INSENSITIVE
, hDeviceMapKey
, NULL
);
648 Status
= ZwOpenKey(&hPortKey
, KEY_QUERY_VALUE
, &ObjectAttributes
);
649 if (Status
== STATUS_OBJECT_NAME_NOT_FOUND
)
651 DPRINT("HKLM\\HARDWARE\\DEVICEMAP\\%wZ is non-existent\n", &PortBaseName
);
652 Status
= STATUS_SUCCESS
;
655 else if (!NT_SUCCESS(Status
))
657 DPRINT("ZwOpenKey() failed with status 0x%08lx\n", Status
);
661 /* Read each value name */
662 while (ZwEnumerateValueKey(hPortKey
, Index
++, KeyValueBasicInformation
, KeyValueInformation
, Size
, &ResultLength
) == STATUS_SUCCESS
)
664 UNICODE_STRING PortName
;
665 PDEVICE_OBJECT PortDeviceObject
= NULL
;
666 PFILE_OBJECT FileObject
= NULL
;
668 PortName
.Length
= PortName
.MaximumLength
= KeyValueInformation
->NameLength
;
669 PortName
.Buffer
= KeyValueInformation
->Name
;
671 /* Open the device object pointer */
672 Status
= IoGetDeviceObjectPointer(&PortName
, FILE_READ_ATTRIBUTES
, &FileObject
, &PortDeviceObject
);
673 if (!NT_SUCCESS(Status
))
675 DPRINT("IoGetDeviceObjectPointer(%wZ) failed with status 0x%08lx\n", Status
);
678 /* Connect the port device object */
679 if (DriverExtension
->ConnectMultiplePorts
)
681 Status
= ConnectPortDriver(PortDeviceObject
, DriverExtension
->MainClassDeviceObject
);
682 if (!NT_SUCCESS(Status
))
684 /* FIXME: Log the error */
685 DPRINT("ConnectPortDriver() failed with status 0x%08lx\n", Status
);
686 ObDereferenceObject(PortDeviceObject
);
691 PDEVICE_OBJECT ClassDO
;
692 Status
= CreateClassDeviceObject(DriverObject
, &ClassDO
);
693 if (!NT_SUCCESS(Status
))
695 /* FIXME: Log the error */
696 DPRINT("CreatePointerClassDeviceObject() failed with status 0x%08lx\n", Status
);
697 ObDereferenceObject(PortDeviceObject
);
700 Status
= ConnectPortDriver(PortDeviceObject
, ClassDO
);
701 if (!NT_SUCCESS(Status
))
703 /* FIXME: Log the error */
704 DPRINT("ConnectPortDriver() failed with status 0x%08lx\n", Status
);
705 ObDereferenceObject(PortDeviceObject
);
706 IoDeleteDevice(ClassDO
);
710 if (Status
== STATUS_NO_MORE_ENTRIES
)
711 Status
= STATUS_SUCCESS
;
714 if (KeyValueInformation
!= NULL
)
715 ExFreePool(KeyValueInformation
);
716 if (hDeviceMapKey
!= (HANDLE
)-1)
717 ZwClose(hDeviceMapKey
);
718 if (hPortKey
!= (HANDLE
)-1)
724 * Standard DriverEntry method.
728 IN PDRIVER_OBJECT DriverObject
,
729 IN PUNICODE_STRING RegistryPath
)
731 PCLASS_DRIVER_EXTENSION DriverExtension
;
735 Status
= IoAllocateDriverObjectExtension(
738 sizeof(CLASS_DRIVER_EXTENSION
),
739 (PVOID
*)&DriverExtension
);
740 if (!NT_SUCCESS(Status
))
742 DPRINT("IoAllocateDriverObjectExtension() failed with status 0x%08lx\n", Status
);
745 RtlZeroMemory(DriverExtension
, sizeof(CLASS_DRIVER_EXTENSION
));
747 Status
= ReadRegistryEntries(RegistryPath
, DriverExtension
);
748 if (!NT_SUCCESS(Status
))
750 DPRINT("ReadRegistryEntries() failed with status 0x%08lx\n", Status
);
754 if (DriverExtension
->ConnectMultiplePorts
== 1)
756 Status
= CreateClassDeviceObject(
758 &DriverExtension
->MainClassDeviceObject
);
759 if (!NT_SUCCESS(Status
))
761 DPRINT("CreateClassDeviceObject() failed with status 0x%08lx\n", Status
);
766 DriverObject
->DriverExtension
->AddDevice
= ClassAddDevice
;
767 DriverObject
->DriverUnload
= DriverUnload
;
769 for (i
= 0; i
< IRP_MJ_MAXIMUM_FUNCTION
; i
++)
770 DriverObject
->MajorFunction
[i
] = IrpStub
;
772 DriverObject
->MajorFunction
[IRP_MJ_CREATE
] = ClassCreate
;
773 DriverObject
->MajorFunction
[IRP_MJ_CLOSE
] = ClassClose
;
774 DriverObject
->MajorFunction
[IRP_MJ_CLEANUP
] = ClassCleanup
;
775 DriverObject
->MajorFunction
[IRP_MJ_READ
] = ClassRead
;
776 DriverObject
->MajorFunction
[IRP_MJ_DEVICE_CONTROL
] = ClassDeviceControl
;
777 DriverObject
->DriverStartIo
= ClassStartIo
;
779 Status
= SearchForLegacyDrivers(DriverObject
, DriverExtension
);