dlls/winebus.sys/main.c

   1 /*
   2  * WINE Platform native bus driver
   3  *
   4  * Copyright 2016 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 #include "config.h"
  21 #include <stdarg.h>
  22
  23 #define NONAMELESSUNION
  24 #define NONAMELESSSTRUCT
  25
  26 #include "ntstatus.h"
  27 #define WIN32_NO_STATUS
  28 #include "windef.h"
  29 #include "winbase.h"
  30 #include "winuser.h"
  31 #include "winternl.h"
  32 #include "winreg.h"
  33 #include "setupapi.h"
  34 #include "winioctl.h"
  35 #include "ddk/wdm.h"
  36 #include "ddk/hidport.h"
  37 #include "wine/debug.h"
  38 #include "wine/unicode.h"
  39 #include "wine/list.h"
  40
  41 #include "bus.h"
  42
  43 WINE_DEFAULT_DEBUG_CHANNEL(plugplay);
  44 WINE_DECLARE_DEBUG_CHANNEL(hid_report);
  45
  46 #define VID_MICROSOFT 0x045e
  47
  48 static const WORD PID_XBOX_CONTROLLERS[] =  {
  49     0x0202, /* Xbox Controller */
  50     0x0285, /* Xbox Controller S */
  51     0x0289, /* Xbox Controller S */
  52     0x028e, /* Xbox360 Controller */
  53     0x028f, /* Xbox360 Wireless Controller */
  54     0x02d1, /* Xbox One Controller */
  55     0x02dd, /* Xbox One Controller (Covert Forces/Firmware 2015) */
  56     0x02e0, /* Xbox One X Controller */
  57     0x02e3, /* Xbox One Elite Controller */
  58     0x02e6, /* Wireless XBox Controller Dongle */
  59     0x02ea, /* Xbox One S Controller */
  60     0x02fd, /* Xbox One S Controller (Firmware 2017) */
  61     0x0719, /* Xbox 360 Wireless Adapter */
  62 };
  63
  64 struct pnp_device
  65 {
  66     struct list entry;
  67     DEVICE_OBJECT *device;
  68 };
  69
  70 struct device_extension
  71 {
  72     struct pnp_device *pnp_device;
  73
  74     WORD vid, pid, input;
  75     DWORD uid, version, index;
  76     BOOL is_gamepad;
  77     WCHAR *serial;
  78     const WCHAR *busid;  /* Expected to be a static constant */
  79
  80     const platform_vtbl *vtbl;
  81
  82     BYTE *last_report;
  83     DWORD last_report_size;
  84     BOOL last_report_read;
  85     DWORD buffer_size;
  86     LIST_ENTRY irp_queue;
  87     CRITICAL_SECTION report_cs;
  88
  89     BYTE platform_private[1];
  90 };
  91
  92 static CRITICAL_SECTION device_list_cs;
  93 static CRITICAL_SECTION_DEBUG critsect_debug =
  94 {
  95     0, 0, &device_list_cs,
  96     { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
  97       0, 0, { (DWORD_PTR)(__FILE__ ": device_list_cs") }
  98 };
  99 static CRITICAL_SECTION device_list_cs = { &critsect_debug, -1, 0, 0, 0, 0 };
 100
 101 static struct list pnp_devset = LIST_INIT(pnp_devset);
 102
 103 static const WCHAR zero_serialW[]= {'0','0','0','0',0};
 104 static const WCHAR miW[] = {'M','I',0};
 105 static const WCHAR igW[] = {'I','G',0};
 106
 107 static inline WCHAR *strdupW(const WCHAR *src)
 108 {
 109     WCHAR *dst;
 110     if (!src) return NULL;
 111     dst = HeapAlloc(GetProcessHeap(), 0, (strlenW(src) + 1)*sizeof(WCHAR));
 112     if (dst) strcpyW(dst, src);
 113     return dst;
 114 }
 115
 116 void *get_platform_private(DEVICE_OBJECT *device)
 117 {
 118     struct device_extension *ext = (struct device_extension *)device->DeviceExtension;
 119     return ext->platform_private;
 120 }
 121
 122 static DWORD get_device_index(WORD vid, WORD pid, WORD input)
 123 {
 124     struct pnp_device *ptr;
 125     DWORD index = 0;
 126
 127     LIST_FOR_EACH_ENTRY(ptr, &pnp_devset, struct pnp_device, entry)
 128     {
 129         struct device_extension *ext = (struct device_extension *)ptr->device->DeviceExtension;
 130         if (ext->vid == vid && ext->pid == pid && ext->input == input)
 131             index = max(ext->index + 1, index);
 132     }
 133
 134     return index;
 135 }
 136
 137 static WCHAR *get_instance_id(DEVICE_OBJECT *device)
 138 {
 139     static const WCHAR formatW[] =  {'%','s','\\','v','i','d','_','%','0','4','x','&','p','i','d','_','%','0','4','x',
 140                                      '\\','%','i','&','%','s','&','%','x','&','%','i',0};
 141     static const WCHAR format_inputW[] = {'%','s','\\','v','i','d','_','%','0','4','x','&','p','i','d','_','%','0','4','x','&',
 142                                      '%','s','_','%','i','\\','%','i','&','%','s','&','%','x','&','%','i',0};
 143     struct device_extension *ext = (struct device_extension *)device->DeviceExtension;
 144     const WCHAR *serial = ext->serial ? ext->serial : zero_serialW;
 145     DWORD len = strlenW(ext->busid) + strlenW(serial) + 64;
 146     WCHAR *dst;
 147
 148     if ((dst = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR))))
 149     {
 150         if (ext->input == (WORD)-1)
 151         {
 152             sprintfW(dst, formatW, ext->busid, ext->vid, ext->pid,
 153                     ext->version, serial, ext->uid, ext->index);
 154         }
 155         else
 156         {
 157             sprintfW(dst, format_inputW, ext->busid, ext->vid, ext->pid, ext->is_gamepad ? igW : miW,
 158                     ext->input, ext->version, serial, ext->uid, ext->index);
 159         }
 160     }
 161
 162     return dst;
 163 }
 164
 165 static WCHAR *get_device_id(DEVICE_OBJECT *device)
 166 {
 167     static const WCHAR formatW[] =  {'%','s','\\','V','i','d','_','%','0','4','x','&','P','i','d','_','%','0','4','x',0};
 168     struct device_extension *ext = (struct device_extension *)device->DeviceExtension;
 169     DWORD len = strlenW(ext->busid) + 19;
 170     WCHAR *dst;
 171
 172     if ((dst = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR))))
 173         sprintfW(dst, formatW, ext->busid, ext->vid, ext->pid);
 174
 175     return dst;
 176 }
 177
 178 static WCHAR *get_compatible_ids(DEVICE_OBJECT *device)
 179 {
 180     struct device_extension *ext = (struct device_extension *)device->DeviceExtension;
 181     WCHAR *iid, *did, *dst, *ptr;
 182     DWORD len;
 183
 184     if (!(iid = get_instance_id(device)))
 185         return NULL;
 186
 187     if (!(did = get_device_id(device)))
 188     {
 189         HeapFree(GetProcessHeap(), 0, iid);
 190         return NULL;
 191     }
 192
 193     len = strlenW(iid) + strlenW(did) + strlenW(ext->busid) + 4;
 194     if ((dst = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR))))
 195     {
 196         ptr = dst;
 197         strcpyW(ptr, iid);
 198         ptr += strlenW(iid) + 1;
 199         strcpyW(ptr, did);
 200         ptr += strlenW(did) + 1;
 201         strcpyW(ptr, ext->busid);
 202         ptr += strlenW(ext->busid) + 1;
 203         *ptr = 0;
 204     }
 205
 206     HeapFree(GetProcessHeap(), 0, iid);
 207     HeapFree(GetProcessHeap(), 0, did);
 208     return dst;
 209 }
 210
 211 DEVICE_OBJECT *bus_create_hid_device(DRIVER_OBJECT *driver, const WCHAR *busidW, WORD vid, WORD pid,
 212                                      WORD input, DWORD version, DWORD uid, const WCHAR *serialW, BOOL is_gamepad,
 213                                      const GUID *class, const platform_vtbl *vtbl, DWORD platform_data_size)
 214 {
 215     static const WCHAR device_name_fmtW[] = {'\\','D','e','v','i','c','e','\\','%','s','#','%','p',0};
 216     struct device_extension *ext;
 217     struct pnp_device *pnp_dev;
 218     DEVICE_OBJECT *device;
 219     UNICODE_STRING nameW;
 220     WCHAR dev_name[256];
 221     HDEVINFO devinfo;
 222     SP_DEVINFO_DATA data = {sizeof(data)};
 223     WCHAR *instance = NULL;
 224     NTSTATUS status;
 225     DWORD length;
 226
 227     TRACE("(%p, %s, %04x, %04x, %04x, %u, %u, %s, %u, %s, %p, %u)\n", driver,
 228             debugstr_w(busidW), vid, pid, input, version, uid, debugstr_w(serialW),
 229             is_gamepad, debugstr_guid(class), vtbl, platform_data_size);
 230
 231     if (!(pnp_dev = HeapAlloc(GetProcessHeap(), 0, sizeof(*pnp_dev))))
 232         return NULL;
 233
 234     sprintfW(dev_name, device_name_fmtW, busidW, pnp_dev);
 235     RtlInitUnicodeString(&nameW, dev_name);
 236     length = FIELD_OFFSET(struct device_extension, platform_private[platform_data_size]);
 237     status = IoCreateDevice(driver, length, &nameW, 0, 0, FALSE, &device);
 238     if (status)
 239     {
 240         FIXME("failed to create device error %x\n", status);
 241         HeapFree(GetProcessHeap(), 0, pnp_dev);
 242         return NULL;
 243     }
 244
 245     EnterCriticalSection(&device_list_cs);
 246
 247     /* fill out device_extension struct */
 248     ext = (struct device_extension *)device->DeviceExtension;
 249     ext->pnp_device         = pnp_dev;
 250     ext->vid                = vid;
 251     ext->pid                = pid;
 252     ext->input              = input;
 253     ext->uid                = uid;
 254     ext->version            = version;
 255     ext->index              = get_device_index(vid, pid, input);
 256     ext->is_gamepad         = is_gamepad;
 257     ext->serial             = strdupW(serialW);
 258     ext->busid              = busidW;
 259     ext->vtbl               = vtbl;
 260     ext->last_report        = NULL;
 261     ext->last_report_size   = 0;
 262     ext->last_report_read   = TRUE;
 263     ext->buffer_size        = 0;
 264
 265     memset(ext->platform_private, 0, platform_data_size);
 266
 267     InitializeListHead(&ext->irp_queue);
 268     InitializeCriticalSection(&ext->report_cs);
 269     ext->report_cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": report_cs");
 270
 271     /* add to list of pnp devices */
 272     pnp_dev->device = device;
 273     list_add_tail(&pnp_devset, &pnp_dev->entry);
 274
 275     LeaveCriticalSection(&device_list_cs);
 276
 277     devinfo = SetupDiCreateDeviceInfoList(class, NULL);
 278     if (devinfo == INVALID_HANDLE_VALUE)
 279     {
 280         ERR("failed to create device info list, error %#x\n", GetLastError());
 281         goto error;
 282     }
 283
 284     instance = get_instance_id(device);
 285     if (!instance)
 286     {
 287         ERR("failed to generate instance id\n");
 288         goto error;
 289     }
 290
 291     if (SetupDiCreateDeviceInfoW(devinfo, instance, class, NULL, NULL, DICD_INHERIT_CLASSDRVS, &data))
 292     {
 293         if (!SetupDiRegisterDeviceInfo(devinfo, &data, 0, NULL, NULL, NULL))
 294             ERR("failed to register device info, error %#x\n", GetLastError());
 295     }
 296     else if (GetLastError() != ERROR_DEVINST_ALREADY_EXISTS)
 297         ERR("failed to create device info, error %#x\n", GetLastError());
 298
 299 error:
 300     HeapFree(GetProcessHeap(), 0, instance);
 301     SetupDiDestroyDeviceInfoList(devinfo);
 302     return device;
 303 }
 304
 305 DEVICE_OBJECT *bus_find_hid_device(const platform_vtbl *vtbl, void *platform_dev)
 306 {
 307     struct pnp_device *dev;
 308     DEVICE_OBJECT *ret = NULL;
 309
 310     TRACE("(%p, %p)\n", vtbl, platform_dev);
 311
 312     EnterCriticalSection(&device_list_cs);
 313     LIST_FOR_EACH_ENTRY(dev, &pnp_devset, struct pnp_device, entry)
 314     {
 315         struct device_extension *ext = (struct device_extension *)dev->device->DeviceExtension;
 316         if (ext->vtbl != vtbl) continue;
 317         if (ext->vtbl->compare_platform_device(dev->device, platform_dev) == 0)
 318         {
 319             ret = dev->device;
 320             break;
 321         }
 322     }
 323     LeaveCriticalSection(&device_list_cs);
 324
 325     TRACE("returning %p\n", ret);
 326     return ret;
 327 }
 328
 329 DEVICE_OBJECT* bus_enumerate_hid_devices(const platform_vtbl *vtbl, enum_func function, void* context)
 330 {
 331     struct pnp_device *dev;
 332     DEVICE_OBJECT *ret = NULL;
 333
 334     TRACE("(%p)\n", vtbl);
 335
 336     EnterCriticalSection(&device_list_cs);
 337     LIST_FOR_EACH_ENTRY(dev, &pnp_devset, struct pnp_device, entry)
 338     {
 339         struct device_extension *ext = (struct device_extension *)dev->device->DeviceExtension;
 340         if (ext->vtbl != vtbl) continue;
 341         if (function(dev->device, context) == 0)
 342         {
 343             ret = dev->device;
 344             break;
 345         }
 346     }
 347     LeaveCriticalSection(&device_list_cs);
 348     return ret;
 349 }
 350
 351 void bus_remove_hid_device(DEVICE_OBJECT *device)
 352 {
 353     struct device_extension *ext = (struct device_extension *)device->DeviceExtension;
 354     struct pnp_device *pnp_device = ext->pnp_device;
 355     LIST_ENTRY *entry;
 356     IRP *irp;
 357
 358     TRACE("(%p)\n", device);
 359
 360     EnterCriticalSection(&device_list_cs);
 361     list_remove(&pnp_device->entry);
 362     LeaveCriticalSection(&device_list_cs);
 363
 364     /* Cancel pending IRPs */
 365     EnterCriticalSection(&ext->report_cs);
 366     while ((entry = RemoveHeadList(&ext->irp_queue)) != &ext->irp_queue)
 367     {
 368         irp = CONTAINING_RECORD(entry, IRP, Tail.Overlay.s.ListEntry);
 369         irp->IoStatus.u.Status = STATUS_CANCELLED;
 370         irp->IoStatus.Information = 0;
 371         IoCompleteRequest(irp, IO_NO_INCREMENT);
 372     }
 373     LeaveCriticalSection(&ext->report_cs);
 374
 375     ext->report_cs.DebugInfo->Spare[0] = 0;
 376     DeleteCriticalSection(&ext->report_cs);
 377
 378     HeapFree(GetProcessHeap(), 0, ext->serial);
 379     HeapFree(GetProcessHeap(), 0, ext->last_report);
 380     IoDeleteDevice(device);
 381
 382     /* pnp_device must be released after the device is gone */
 383     HeapFree(GetProcessHeap(), 0, pnp_device);
 384 }
 385
 386 static NTSTATUS build_device_relations(DEVICE_RELATIONS **devices)
 387 {
 388     int i;
 389     struct pnp_device *ptr;
 390
 391     EnterCriticalSection(&device_list_cs);
 392     *devices = HeapAlloc(GetProcessHeap(), 0, sizeof(DEVICE_RELATIONS) +
 393         list_count(&pnp_devset) * sizeof (void *));
 394
 395     if (!*devices)
 396     {
 397         LeaveCriticalSection(&device_list_cs);
 398         return STATUS_INSUFFICIENT_RESOURCES;
 399     }
 400
 401     i = 0;
 402     LIST_FOR_EACH_ENTRY(ptr, &pnp_devset, struct pnp_device, entry)
 403     {
 404         (*devices)->Objects[i] = ptr->device;
 405         i++;
 406     }
 407     LeaveCriticalSection(&device_list_cs);
 408     (*devices)->Count = i;
 409     return STATUS_SUCCESS;
 410 }
 411
 412 static NTSTATUS handle_IRP_MN_QUERY_DEVICE_RELATIONS(IRP *irp)
 413 {
 414     NTSTATUS status = irp->IoStatus.u.Status;
 415     IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation( irp );
 416
 417     TRACE("IRP_MN_QUERY_DEVICE_RELATIONS\n");
 418     switch (irpsp->Parameters.QueryDeviceRelations.Type)
 419     {
 420         case EjectionRelations:
 421         case RemovalRelations:
 422         case TargetDeviceRelation:
 423         case PowerRelations:
 424             FIXME("Unhandled Device Relation %x\n",irpsp->Parameters.QueryDeviceRelations.Type);
 425             break;
 426         case BusRelations:
 427             status = build_device_relations((DEVICE_RELATIONS**)&irp->IoStatus.Information);
 428             break;
 429         default:
 430             FIXME("Unknown Device Relation %x\n",irpsp->Parameters.QueryDeviceRelations.Type);
 431             break;
 432     }
 433
 434     return status;
 435 }
 436
 437 static NTSTATUS handle_IRP_MN_QUERY_ID(DEVICE_OBJECT *device, IRP *irp)
 438 {
 439     NTSTATUS status = irp->IoStatus.u.Status;
 440     IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation( irp );
 441     BUS_QUERY_ID_TYPE type = irpsp->Parameters.QueryId.IdType;
 442
 443     TRACE("(%p, %p)\n", device, irp);
 444
 445     switch (type)
 446     {
 447         case BusQueryHardwareIDs:
 448             TRACE("BusQueryHardwareIDs\n");
 449             irp->IoStatus.Information = (ULONG_PTR)get_compatible_ids(device);
 450             break;
 451         case BusQueryCompatibleIDs:
 452             TRACE("BusQueryCompatibleIDs\n");
 453             irp->IoStatus.Information = (ULONG_PTR)get_compatible_ids(device);
 454             break;
 455         case BusQueryDeviceID:
 456             TRACE("BusQueryDeviceID\n");
 457             irp->IoStatus.Information = (ULONG_PTR)get_device_id(device);
 458             break;
 459         case BusQueryInstanceID:
 460             TRACE("BusQueryInstanceID\n");
 461             irp->IoStatus.Information = (ULONG_PTR)get_instance_id(device);
 462             break;
 463         default:
 464             FIXME("Unhandled type %08x\n", type);
 465             return status;
 466     }
 467
 468     status = irp->IoStatus.Information ? STATUS_SUCCESS : STATUS_NO_MEMORY;
 469     return status;
 470 }
 471
 472 NTSTATUS WINAPI common_pnp_dispatch(DEVICE_OBJECT *device, IRP *irp)
 473 {
 474     NTSTATUS status = irp->IoStatus.u.Status;
 475     IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation(irp);
 476
 477     switch (irpsp->MinorFunction)
 478     {
 479         case IRP_MN_QUERY_DEVICE_RELATIONS:
 480             TRACE("IRP_MN_QUERY_DEVICE_RELATIONS\n");
 481             status = handle_IRP_MN_QUERY_DEVICE_RELATIONS(irp);
 482             irp->IoStatus.u.Status = status;
 483             break;
 484         case IRP_MN_QUERY_ID:
 485             TRACE("IRP_MN_QUERY_ID\n");
 486             status = handle_IRP_MN_QUERY_ID(device, irp);
 487             irp->IoStatus.u.Status = status;
 488             break;
 489         case IRP_MN_QUERY_CAPABILITIES:
 490             TRACE("IRP_MN_QUERY_CAPABILITIES\n");
 491             break;
 492         default:
 493             FIXME("Unhandled function %08x\n", irpsp->MinorFunction);
 494             break;
 495     }
 496
 497     IoCompleteRequest(irp, IO_NO_INCREMENT);
 498     return status;
 499 }
 500
 501 static NTSTATUS deliver_last_report(struct device_extension *ext, DWORD buffer_length, BYTE* buffer, ULONG_PTR *out_length)
 502 {
 503     if (buffer_length < ext->last_report_size)
 504     {
 505         *out_length = 0;
 506         return STATUS_BUFFER_TOO_SMALL;
 507     }
 508     else
 509     {
 510         if (ext->last_report)
 511             memcpy(buffer, ext->last_report, ext->last_report_size);
 512         *out_length = ext->last_report_size;
 513         return STATUS_SUCCESS;
 514     }
 515 }
 516
 517 NTSTATUS WINAPI hid_internal_dispatch(DEVICE_OBJECT *device, IRP *irp)
 518 {
 519     NTSTATUS status = irp->IoStatus.u.Status;
 520     IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation(irp);
 521     struct device_extension *ext = (struct device_extension *)device->DeviceExtension;
 522
 523     TRACE("(%p, %p)\n", device, irp);
 524
 525     switch (irpsp->Parameters.DeviceIoControl.IoControlCode)
 526     {
 527         case IOCTL_HID_GET_DEVICE_ATTRIBUTES:
 528         {
 529             HID_DEVICE_ATTRIBUTES *attr = (HID_DEVICE_ATTRIBUTES *)irp->UserBuffer;
 530             TRACE("IOCTL_HID_GET_DEVICE_ATTRIBUTES\n");
 531
 532             if (irpsp->Parameters.DeviceIoControl.OutputBufferLength < sizeof(*attr))
 533             {
 534                 irp->IoStatus.u.Status = status = STATUS_BUFFER_TOO_SMALL;
 535                 break;
 536             }
 537
 538             memset(attr, 0, sizeof(*attr));
 539             attr->Size = sizeof(*attr);
 540             attr->VendorID = ext->vid;
 541             attr->ProductID = ext->pid;
 542             attr->VersionNumber = ext->version;
 543
 544             irp->IoStatus.u.Status = status = STATUS_SUCCESS;
 545             irp->IoStatus.Information = sizeof(*attr);
 546             break;
 547         }
 548         case IOCTL_HID_GET_DEVICE_DESCRIPTOR:
 549         {
 550             HID_DESCRIPTOR *descriptor = (HID_DESCRIPTOR *)irp->UserBuffer;
 551             DWORD length;
 552             TRACE("IOCTL_HID_GET_DEVICE_DESCRIPTOR\n");
 553
 554             if (irpsp->Parameters.DeviceIoControl.OutputBufferLength < sizeof(*descriptor))
 555             {
 556                 irp->IoStatus.u.Status = status = STATUS_BUFFER_TOO_SMALL;
 557                 break;
 558             }
 559
 560             status = ext->vtbl->get_reportdescriptor(device, NULL, 0, &length);
 561             if (status != STATUS_SUCCESS && status != STATUS_BUFFER_TOO_SMALL)
 562             {
 563                 WARN("Failed to get platform report descriptor length\n");
 564                 irp->IoStatus.u.Status = status;
 565                 break;
 566             }
 567
 568             memset(descriptor, 0, sizeof(*descriptor));
 569             descriptor->bLength = sizeof(*descriptor);
 570             descriptor->bDescriptorType = HID_HID_DESCRIPTOR_TYPE;
 571             descriptor->bcdHID = HID_REVISION;
 572             descriptor->bCountry = 0;
 573             descriptor->bNumDescriptors = 1;
 574             descriptor->DescriptorList[0].bReportType = HID_REPORT_DESCRIPTOR_TYPE;
 575             descriptor->DescriptorList[0].wReportLength = length;
 576
 577             irp->IoStatus.u.Status = status = STATUS_SUCCESS;
 578             irp->IoStatus.Information = sizeof(*descriptor);
 579             break;
 580         }
 581         case IOCTL_HID_GET_REPORT_DESCRIPTOR:
 582         {
 583             DWORD length = irpsp->Parameters.DeviceIoControl.OutputBufferLength;
 584             TRACE("IOCTL_HID_GET_REPORT_DESCRIPTOR\n");
 585
 586             irp->IoStatus.u.Status = status = ext->vtbl->get_reportdescriptor(device, irp->UserBuffer, length, &length);
 587             irp->IoStatus.Information = length;
 588             break;
 589         }
 590         case IOCTL_HID_GET_STRING:
 591         {
 592             DWORD length = irpsp->Parameters.DeviceIoControl.OutputBufferLength / sizeof(WCHAR);
 593             DWORD index = (ULONG_PTR)irpsp->Parameters.DeviceIoControl.Type3InputBuffer;
 594             TRACE("IOCTL_HID_GET_STRING[%08x]\n", index);
 595
 596             irp->IoStatus.u.Status = status = ext->vtbl->get_string(device, index, (WCHAR *)irp->UserBuffer, length);
 597             if (status == STATUS_SUCCESS)
 598                 irp->IoStatus.Information = (strlenW((WCHAR *)irp->UserBuffer) + 1) * sizeof(WCHAR);
 599             break;
 600         }
 601         case IOCTL_HID_GET_INPUT_REPORT:
 602         {
 603             HID_XFER_PACKET *packet = (HID_XFER_PACKET*)(irp->UserBuffer);
 604             TRACE_(hid_report)("IOCTL_HID_GET_INPUT_REPORT\n");
 605             EnterCriticalSection(&ext->report_cs);
 606             status = ext->vtbl->begin_report_processing(device);
 607             if (status != STATUS_SUCCESS)
 608             {
 609                 irp->IoStatus.u.Status = status;
 610                 LeaveCriticalSection(&ext->report_cs);
 611                 break;
 612             }
 613
 614             irp->IoStatus.u.Status = status = deliver_last_report(ext,
 615                 packet->reportBufferLen, packet->reportBuffer,
 616                 &irp->IoStatus.Information);
 617
 618             if (status == STATUS_SUCCESS)
 619                 packet->reportBufferLen = irp->IoStatus.Information;
 620             LeaveCriticalSection(&ext->report_cs);
 621             break;
 622         }
 623         case IOCTL_HID_READ_REPORT:
 624         {
 625             TRACE_(hid_report)("IOCTL_HID_READ_REPORT\n");
 626             EnterCriticalSection(&ext->report_cs);
 627             status = ext->vtbl->begin_report_processing(device);
 628             if (status != STATUS_SUCCESS)
 629             {
 630                 irp->IoStatus.u.Status = status;
 631                 LeaveCriticalSection(&ext->report_cs);
 632                 break;
 633             }
 634             if (!ext->last_report_read)
 635             {
 636                 irp->IoStatus.u.Status = status = deliver_last_report(ext,
 637                     irpsp->Parameters.DeviceIoControl.OutputBufferLength,
 638                     irp->UserBuffer, &irp->IoStatus.Information);
 639                 ext->last_report_read = TRUE;
 640             }
 641             else
 642             {
 643                 InsertTailList(&ext->irp_queue, &irp->Tail.Overlay.s.ListEntry);
 644                 status = STATUS_PENDING;
 645             }
 646             LeaveCriticalSection(&ext->report_cs);
 647             break;
 648         }
 649         case IOCTL_HID_SET_OUTPUT_REPORT:
 650         case IOCTL_HID_WRITE_REPORT:
 651         {
 652             HID_XFER_PACKET *packet = (HID_XFER_PACKET*)(irp->UserBuffer);
 653             TRACE_(hid_report)("IOCTL_HID_WRITE_REPORT / IOCTL_HID_SET_OUTPUT_REPORT\n");
 654             irp->IoStatus.u.Status = status = ext->vtbl->set_output_report(
 655                 device, packet->reportId, packet->reportBuffer,
 656                 packet->reportBufferLen, &irp->IoStatus.Information);
 657             break;
 658         }
 659         case IOCTL_HID_GET_FEATURE:
 660         {
 661             HID_XFER_PACKET *packet = (HID_XFER_PACKET*)(irp->UserBuffer);
 662             TRACE_(hid_report)("IOCTL_HID_GET_FEATURE\n");
 663             irp->IoStatus.u.Status = status = ext->vtbl->get_feature_report(
 664                 device, packet->reportId, packet->reportBuffer,
 665                 packet->reportBufferLen, &irp->IoStatus.Information);
 666             packet->reportBufferLen = irp->IoStatus.Information;
 667             break;
 668         }
 669         case IOCTL_HID_SET_FEATURE:
 670         {
 671             HID_XFER_PACKET *packet = (HID_XFER_PACKET*)(irp->UserBuffer);
 672             TRACE_(hid_report)("IOCTL_HID_SET_FEATURE\n");
 673             irp->IoStatus.u.Status = status = ext->vtbl->set_feature_report(
 674                 device, packet->reportId, packet->reportBuffer,
 675                 packet->reportBufferLen, &irp->IoStatus.Information);
 676             break;
 677         }
 678         default:
 679         {
 680             ULONG code = irpsp->Parameters.DeviceIoControl.IoControlCode;
 681             FIXME("Unsupported ioctl %x (device=%x access=%x func=%x method=%x)\n",
 682                   code, code >> 16, (code >> 14) & 3, (code >> 2) & 0xfff, code & 3);
 683             break;
 684         }
 685     }
 686
 687     if (status != STATUS_PENDING)
 688         IoCompleteRequest(irp, IO_NO_INCREMENT);
 689
 690     return status;
 691 }
 692
 693 void process_hid_report(DEVICE_OBJECT *device, BYTE *report, DWORD length)
 694 {
 695     struct device_extension *ext = (struct device_extension*)device->DeviceExtension;
 696     IRP *irp;
 697     LIST_ENTRY *entry;
 698
 699     if (!length || !report)
 700         return;
 701
 702     EnterCriticalSection(&ext->report_cs);
 703     if (length > ext->buffer_size)
 704     {
 705         HeapFree(GetProcessHeap(), 0, ext->last_report);
 706         ext->last_report = HeapAlloc(GetProcessHeap(), 0, length);
 707         if (!ext->last_report)
 708         {
 709             ERR_(hid_report)("Failed to alloc last report\n");
 710             ext->buffer_size = 0;
 711             ext->last_report_size = 0;
 712             ext->last_report_read = TRUE;
 713             LeaveCriticalSection(&ext->report_cs);
 714             return;
 715         }
 716         else
 717             ext->buffer_size = length;
 718     }
 719
 720     memcpy(ext->last_report, report, length);
 721     ext->last_report_size = length;
 722     ext->last_report_read = FALSE;
 723
 724     while ((entry = RemoveHeadList(&ext->irp_queue)) != &ext->irp_queue)
 725     {
 726         IO_STACK_LOCATION *irpsp;
 727         TRACE_(hid_report)("Processing Request\n");
 728         irp = CONTAINING_RECORD(entry, IRP, Tail.Overlay.s.ListEntry);
 729         irpsp = IoGetCurrentIrpStackLocation(irp);
 730         irp->IoStatus.u.Status = deliver_last_report(ext,
 731             irpsp->Parameters.DeviceIoControl.OutputBufferLength,
 732             irp->UserBuffer, &irp->IoStatus.Information);
 733         ext->last_report_read = TRUE;
 734         IoCompleteRequest(irp, IO_NO_INCREMENT);
 735     }
 736     LeaveCriticalSection(&ext->report_cs);
 737 }
 738
 739 DWORD check_bus_option(UNICODE_STRING *registry_path, const UNICODE_STRING *option, DWORD default_value)
 740 {
 741     OBJECT_ATTRIBUTES attr;
 742     HANDLE key;
 743     DWORD output = default_value;
 744
 745     InitializeObjectAttributes(&attr, registry_path, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, NULL, NULL);
 746     if (NtOpenKey(&key, KEY_ALL_ACCESS, &attr) == STATUS_SUCCESS)
 747     {
 748         DWORD size;
 749         char buffer[FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION, Data[sizeof(DWORD)])];
 750
 751         KEY_VALUE_PARTIAL_INFORMATION *info = (KEY_VALUE_PARTIAL_INFORMATION*)buffer;
 752
 753         if (NtQueryValueKey(key, option, KeyValuePartialInformation, info, sizeof(buffer), &size) == STATUS_SUCCESS)
 754         {
 755             if (info->Type == REG_DWORD)
 756                 output = *(DWORD*)info->Data;
 757         }
 758
 759         NtClose(key);
 760     }
 761
 762     return output;
 763 }
 764
 765 BOOL is_xbox_gamepad(WORD vid, WORD pid)
 766 {
 767     int i;
 768
 769     if (vid != VID_MICROSOFT)
 770         return FALSE;
 771
 772     for (i = 0; i < ARRAY_SIZE(PID_XBOX_CONTROLLERS); i++)
 773         if (pid == PID_XBOX_CONTROLLERS[i]) return TRUE;
 774
 775     return FALSE;
 776 }
 777
 778 static void WINAPI driver_unload(DRIVER_OBJECT *driver)
 779 {
 780     udev_driver_unload();
 781     iohid_driver_unload();
 782     sdl_driver_unload();
 783 }
 784
 785 NTSTATUS WINAPI DriverEntry( DRIVER_OBJECT *driver, UNICODE_STRING *path )
 786 {
 787     static const WCHAR udevW[] = {'\\','D','r','i','v','e','r','\\','U','D','E','V',0};
 788     static UNICODE_STRING udev = {sizeof(udevW) - sizeof(WCHAR), sizeof(udevW), (WCHAR *)udevW};
 789     static const WCHAR iohidW[] = {'\\','D','r','i','v','e','r','\\','I','O','H','I','D',0};
 790     static UNICODE_STRING iohid = {sizeof(iohidW) - sizeof(WCHAR), sizeof(iohidW), (WCHAR *)iohidW};
 791     static const WCHAR sdlW[] = {'\\','D','r','i','v','e','r','\\','S','D','L','J','O','Y',0};
 792     static UNICODE_STRING sdl = {sizeof(sdlW) - sizeof(WCHAR), sizeof(sdlW), (WCHAR *)sdlW};
 793     static const WCHAR SDL_enabledW[] = {'E','n','a','b','l','e',' ','S','D','L',0};
 794     static const UNICODE_STRING SDL_enabled = {sizeof(SDL_enabledW) - sizeof(WCHAR), sizeof(SDL_enabledW), (WCHAR*)SDL_enabledW};
 795
 796     TRACE( "(%p, %s)\n", driver, debugstr_w(path->Buffer) );
 797
 798     if (check_bus_option(path, &SDL_enabled, 1))
 799     {
 800         if (IoCreateDriver(&sdl, sdl_driver_init) == STATUS_SUCCESS)
 801             return STATUS_SUCCESS;
 802     }
 803     IoCreateDriver(&udev, udev_driver_init);
 804     IoCreateDriver(&iohid, iohid_driver_init);
 805
 806     driver->DriverUnload = driver_unload;
 807
 808     return STATUS_SUCCESS;
 809 }