Function DriverEntry

kmclass.cpp:39–87

extern "C"
#endif
NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject, IN PUNICODE_STRING  RegistryPath)
{
    UNICODE_STRING     DeviceName, Win32Device;
    PDEVICE_OBJECT     DeviceObject = NULL;
    NTSTATUS           status;


    RtlInitUnicodeString(&DeviceName, KEYMOUSE_DEVICE_NAME);
    RtlInitUnicodeString(&Win32Device, KEYMOUSE_DOS_DEVICE_NAME);

    for (ULONG i = 0; i <= IRP_MJ_MAXIMUM_FUNCTION; i++)
        DriverObject->MajorFunction[i] = kmclassDefaultHandler;

    DriverObject->MajorFunction[IRP_MJ_CREATE] = kmclassCreateClose;
    DriverObject->MajorFunction[IRP_MJ_CLOSE] = kmclassCreateClose;
    DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = kmclassDispatchDeviceControl;

    DriverObject->DriverUnload = kmclassUnload;
    status = IoCreateDevice(DriverObject, sizeof(DEVICE_EXTENSION), &DeviceName, FILE_DEVICE_KEYMOUSE, 0, TRUE, &DeviceObject);

    if (!NT_SUCCESS(status))
        return status;
    if (!DeviceObject)
        return STATUS_UNEXPECTED_IO_ERROR;

    DeviceObject->Flags |= DO_DIRECT_IO;
    DeviceObject->AlignmentRequirement = FILE_WORD_ALIGNMENT;
    status = IoCreateSymbolicLink(&Win32Device, &DeviceName);

    //    status = GetKmclassInfo(DeviceObject, KEYBOARD_DEVICE);
    status = SearchKdbServiceCallBack(DriverObject);
    if (!NT_SUCCESS(status))
    {
        KdPrint(("KEYBOARD_DEVICE ERROR, error = 0x%08lx\n", status));
        return status;
    }

    //    status = GetKmclassInfo(DeviceObject, MOUSE_DEVICE);
    status = SearchMouServiceCallBack(DriverObject);
    if (!NT_SUCCESS(status))
    {
        KdPrint(("MOUSE_DEVICE ERROR, error = 0x%08lx\n", status));
        return status;
    }

    DeviceObject->Flags &= ~DO_DEVICE_INITIALIZING;

    return STATUS_SUCCESS;
}

void kmclassUnload(IN PDRIVER_OBJECT DriverObject)
{