dlls/hidclass.sys/main.c

   1 /*
   2  * WINE Hid device services
   3  *
   4  * Copyright 2015 Aric Stewart
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #define NONAMELESSUNION
  22 #include <unistd.h>
  23 #include <stdarg.h>
  24 #include "hid.h"
  25 #include "wine/debug.h"
  26 #include "wine/unicode.h"
  27 #include "wine/list.h"
  28
  29 WINE_DEFAULT_DEBUG_CHANNEL(hid);
  30
  31 static struct list minidriver_list = LIST_INIT(minidriver_list);
  32
  33 minidriver* find_minidriver(DRIVER_OBJECT *driver)
  34 {
  35     minidriver *md;
  36     LIST_FOR_EACH_ENTRY(md, &minidriver_list, minidriver, entry)
  37     {
  38         if (md->minidriver.DriverObject == driver)
  39             return md;
  40     }
  41     return NULL;
  42 }
  43
  44 static VOID WINAPI UnloadDriver(DRIVER_OBJECT *driver)
  45 {
  46     minidriver *md;
  47
  48     TRACE("Driver Unload\n");
  49     md = find_minidriver(driver);
  50     if (md)
  51     {
  52         if (md->DriverUnload)
  53             md->DriverUnload(md->minidriver.DriverObject);
  54         list_remove(&md->entry);
  55         HeapFree( GetProcessHeap(), 0, md );
  56     }
  57 }
  58
  59 NTSTATUS WINAPI HidRegisterMinidriver(HID_MINIDRIVER_REGISTRATION *registration)
  60 {
  61     minidriver *driver;
  62     driver = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*driver));
  63
  64     if (!driver)
  65         return STATUS_NO_MEMORY;
  66
  67     driver->DriverUnload = registration->DriverObject->DriverUnload;
  68     registration->DriverObject->DriverUnload = UnloadDriver;
  69
  70     registration->DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = HID_Device_ioctl;
  71     registration->DriverObject->MajorFunction[IRP_MJ_READ] = HID_Device_read;
  72     registration->DriverObject->MajorFunction[IRP_MJ_WRITE] = HID_Device_write;
  73     registration->DriverObject->MajorFunction[IRP_MJ_CREATE] = HID_Device_create;
  74     registration->DriverObject->MajorFunction[IRP_MJ_CLOSE] = HID_Device_close;
  75
  76     driver->PNPDispatch = registration->DriverObject->MajorFunction[IRP_MJ_PNP];
  77     registration->DriverObject->MajorFunction[IRP_MJ_PNP] = HID_PNP_Dispatch;
  78
  79     driver->AddDevice = registration->DriverObject->DriverExtension->AddDevice;
  80     registration->DriverObject->DriverExtension->AddDevice = PNP_AddDevice;
  81
  82     driver->minidriver = *registration;
  83     list_add_tail(&minidriver_list, &driver->entry);
  84
  85     return STATUS_SUCCESS;
  86 }
  87
  88 static NTSTATUS WINAPI internalComplete(DEVICE_OBJECT *deviceObject, IRP *irp,
  89     void *context)
  90 {
  91     HANDLE event = context;
  92     SetEvent(event);
  93     return STATUS_MORE_PROCESSING_REQUIRED;
  94 }
  95
  96 NTSTATUS call_minidriver(ULONG code, DEVICE_OBJECT *device, void *in_buff, ULONG in_size, void *out_buff, ULONG out_size)
  97 {
  98     IRP *irp;
  99     IO_STATUS_BLOCK irp_status;
 100     NTSTATUS status;
 101     void *buffer = NULL;
 102
 103     HANDLE event = CreateEventA(NULL, FALSE, FALSE, NULL);
 104
 105     if (out_size)
 106     {
 107         buffer = HeapAlloc(GetProcessHeap(), 0, out_size);
 108         memcpy(buffer, out_buff, out_size);
 109     }
 110
 111     irp = IoBuildDeviceIoControlRequest(code, device, in_buff, in_size,
 112         buffer, out_size, TRUE, NULL, &irp_status);
 113
 114     IoSetCompletionRoutine(irp, internalComplete, event, TRUE, TRUE, TRUE);
 115     status = IoCallDriver(device, irp);
 116
 117     if (status == STATUS_PENDING)
 118         WaitForSingleObject(event, INFINITE);
 119
 120     memcpy(out_buff, buffer, out_size);
 121     status = irp->IoStatus.u.Status;
 122
 123     IoCompleteRequest(irp, IO_NO_INCREMENT );
 124     CloseHandle(event);
 125
 126     return status;
 127 }