drivers/usb/usbport/usbport.c

   1 /*
   2  * PROJECT:     ReactOS USB Port Driver
   3  * LICENSE:     GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
   4  * PURPOSE:     USBPort main driver functions
   5  * COPYRIGHT:   Copyright 2017 Vadim Galyant <vgal@rambler.ru>
   6  */
   7
   8 #include "usbport.h"
   9
  10 #define NDEBUG
  11 #include <debug.h>
  12
  13 #define NDEBUG_USBPORT_CORE
  14 #define NDEBUG_USBPORT_INTERRUPT
  15 #define NDEBUG_USBPORT_TIMER
  16 #include "usbdebug.h"
  17
  18 LIST_ENTRY USBPORT_MiniPortDrivers = {NULL, NULL};
  19 LIST_ENTRY USBPORT_USB1FdoList = {NULL, NULL};
  20 LIST_ENTRY USBPORT_USB2FdoList = {NULL, NULL};
  21
  22 KSPIN_LOCK USBPORT_SpinLock;
  23 BOOLEAN USBPORT_Initialized = FALSE;
  24
  25 PDEVICE_OBJECT
  26 NTAPI
  27 USBPORT_FindUSB2Controller(IN PDEVICE_OBJECT FdoDevice)
  28 {
  29     PUSBPORT_DEVICE_EXTENSION FdoExtension;
  30     PUSBPORT_DEVICE_EXTENSION USB2FdoExtension;
  31     KIRQL OldIrql;
  32     PLIST_ENTRY USB2FdoEntry;
  33     PDEVICE_OBJECT USB2FdoDevice = NULL;
  34
  35     DPRINT("USBPORT_FindUSB2Controller: FdoDevice - %p\n", FdoDevice);
  36
  37     FdoExtension = FdoDevice->DeviceExtension;
  38
  39     KeAcquireSpinLock(&USBPORT_SpinLock, &OldIrql);
  40
  41     USB2FdoEntry = USBPORT_USB2FdoList.Flink;
  42
  43     while (USB2FdoEntry && USB2FdoEntry != &USBPORT_USB2FdoList)
  44     {
  45         USB2FdoExtension = CONTAINING_RECORD(USB2FdoEntry,
  46                                              USBPORT_DEVICE_EXTENSION,
  47                                              ControllerLink);
  48
  49         if (USB2FdoExtension->BusNumber == FdoExtension->BusNumber &&
  50             USB2FdoExtension->PciDeviceNumber == FdoExtension->PciDeviceNumber)
  51         {
  52             USB2FdoDevice = USB2FdoExtension->CommonExtension.SelfDevice;
  53             break;
  54         }
  55
  56         USB2FdoEntry = USB2FdoEntry->Flink;
  57     }
  58
  59     KeReleaseSpinLock(&USBPORT_SpinLock, OldIrql);
  60
  61     return USB2FdoDevice;
  62 }
  63
  64 VOID
  65 NTAPI
  66 USBPORT_AddUSB1Fdo(IN PDEVICE_OBJECT FdoDevice)
  67 {
  68     PUSBPORT_DEVICE_EXTENSION FdoExtension;
  69
  70     DPRINT("USBPORT_AddUSB1Fdo: FdoDevice - %p\n", FdoDevice);
  71
  72     FdoExtension = FdoDevice->DeviceExtension;
  73     FdoExtension->Flags |= USBPORT_FLAG_REGISTERED_FDO;
  74
  75     ExInterlockedInsertTailList(&USBPORT_USB1FdoList,
  76                                 &FdoExtension->ControllerLink,
  77                                 &USBPORT_SpinLock);
  78 }
  79
  80 VOID
  81 NTAPI
  82 USBPORT_AddUSB2Fdo(IN PDEVICE_OBJECT FdoDevice)
  83 {
  84     PUSBPORT_DEVICE_EXTENSION FdoExtension;
  85
  86     DPRINT("USBPORT_AddUSB2Fdo: FdoDevice - %p\n", FdoDevice);
  87
  88     FdoExtension = FdoDevice->DeviceExtension;
  89     FdoExtension->Flags |= USBPORT_FLAG_REGISTERED_FDO;
  90
  91     ExInterlockedInsertTailList(&USBPORT_USB2FdoList,
  92                                 &FdoExtension->ControllerLink,
  93                                 &USBPORT_SpinLock);
  94 }
  95
  96 VOID
  97 NTAPI
  98 USBPORT_RemoveUSBxFdo(IN PDEVICE_OBJECT FdoDevice)
  99 {
 100     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 101     KIRQL OldIrql;
 102
 103     DPRINT("USBPORT_RemoveUSBxFdo: FdoDevice - %p\n", FdoDevice);
 104
 105     FdoExtension = FdoDevice->DeviceExtension;
 106
 107     KeAcquireSpinLock(&USBPORT_SpinLock, &OldIrql);
 108     RemoveEntryList(&FdoExtension->ControllerLink);
 109     KeReleaseSpinLock(&USBPORT_SpinLock, OldIrql);
 110
 111     FdoExtension->Flags &= ~USBPORT_FLAG_REGISTERED_FDO;
 112
 113     FdoExtension->ControllerLink.Flink = NULL;
 114     FdoExtension->ControllerLink.Blink = NULL;
 115 }
 116
 117 BOOLEAN
 118 NTAPI
 119 USBPORT_IsCompanionFdoExtension(IN PDEVICE_OBJECT USB2FdoDevice,
 120                                 IN PUSBPORT_DEVICE_EXTENSION USB1FdoExtension)
 121 {
 122     PUSBPORT_DEVICE_EXTENSION USB2FdoExtension;
 123
 124     DPRINT("USBPORT_IsCompanionFdoExtension: USB2Fdo - %p, USB1FdoExtension - %p\n",
 125            USB2FdoDevice,
 126            USB1FdoExtension);
 127
 128     USB2FdoExtension = USB2FdoDevice->DeviceExtension;
 129
 130     return USB2FdoExtension->BusNumber == USB1FdoExtension->BusNumber &&
 131            USB2FdoExtension->PciDeviceNumber == USB1FdoExtension->PciDeviceNumber;
 132 }
 133
 134 PDEVICE_RELATIONS
 135 NTAPI
 136 USBPORT_FindCompanionControllers(IN PDEVICE_OBJECT USB2FdoDevice,
 137                                  IN BOOLEAN IsObRefer,
 138                                  IN BOOLEAN IsFDOsReturned)
 139 {
 140     PLIST_ENTRY USB1FdoList;
 141     PUSBPORT_DEVICE_EXTENSION USB1FdoExtension;
 142     ULONG NumControllers = 0;
 143     PDEVICE_OBJECT * Entry;
 144     PDEVICE_RELATIONS ControllersList = NULL;
 145     KIRQL OldIrql;
 146
 147     DPRINT("USBPORT_FindCompanionControllers: USB2Fdo - %p, IsObRefer - %x, IsFDOs - %x\n",
 148            USB2FdoDevice,
 149            IsObRefer,
 150            IsFDOsReturned);
 151
 152     KeAcquireSpinLock(&USBPORT_SpinLock, &OldIrql);
 153
 154     USB1FdoList = USBPORT_USB1FdoList.Flink;
 155
 156     while (USB1FdoList && USB1FdoList != &USBPORT_USB1FdoList)
 157     {
 158         USB1FdoExtension = CONTAINING_RECORD(USB1FdoList,
 159                                              USBPORT_DEVICE_EXTENSION,
 160                                              ControllerLink);
 161
 162         if (USB1FdoExtension->Flags & USBPORT_FLAG_COMPANION_HC &&
 163             USBPORT_IsCompanionFdoExtension(USB2FdoDevice, USB1FdoExtension))
 164         {
 165             ++NumControllers;
 166         }
 167
 168         USB1FdoList = USB1FdoExtension->ControllerLink.Flink;
 169     }
 170
 171     DPRINT("USBPORT_FindCompanionControllers: NumControllers - %x\n",
 172            NumControllers);
 173
 174     if (!NumControllers)
 175     {
 176         goto Exit;
 177     }
 178
 179     ControllersList = ExAllocatePoolWithTag(NonPagedPool,
 180                                             NumControllers * sizeof(DEVICE_RELATIONS),
 181                                             USB_PORT_TAG);
 182
 183     if (!ControllersList)
 184     {
 185         goto Exit;
 186     }
 187
 188     RtlZeroMemory(ControllersList, NumControllers * sizeof(DEVICE_RELATIONS));
 189
 190     ControllersList->Count = NumControllers;
 191
 192     USB1FdoList = USBPORT_USB1FdoList.Flink;
 193
 194     Entry = &ControllersList->Objects[0];
 195
 196     while (USB1FdoList && USB1FdoList != &USBPORT_USB1FdoList)
 197     {
 198         USB1FdoExtension = CONTAINING_RECORD(USB1FdoList,
 199                                              USBPORT_DEVICE_EXTENSION,
 200                                              ControllerLink);
 201
 202         if (USB1FdoExtension->Flags & USBPORT_FLAG_COMPANION_HC &&
 203             USBPORT_IsCompanionFdoExtension(USB2FdoDevice, USB1FdoExtension))
 204         {
 205             *Entry = USB1FdoExtension->CommonExtension.LowerPdoDevice;
 206
 207             if (IsObRefer)
 208             {
 209                 ObReferenceObject(USB1FdoExtension->CommonExtension.LowerPdoDevice);
 210             }
 211
 212             if (IsFDOsReturned)
 213             {
 214                 *Entry = USB1FdoExtension->CommonExtension.SelfDevice;
 215             }
 216
 217             ++Entry;
 218         }
 219
 220         USB1FdoList = USB1FdoExtension->ControllerLink.Flink;
 221     }
 222
 223 Exit:
 224
 225     KeReleaseSpinLock(&USBPORT_SpinLock, OldIrql);
 226
 227     return ControllersList;
 228 }
 229
 230 MPSTATUS
 231 NTAPI
 232 USBPORT_NtStatusToMpStatus(NTSTATUS NtStatus)
 233 {
 234     DPRINT("USBPORT_NtStatusToMpStatus: NtStatus - %x\n", NtStatus);
 235
 236     if (NtStatus == STATUS_SUCCESS)
 237     {
 238         return MP_STATUS_SUCCESS;
 239     }
 240     else
 241     {
 242         return MP_STATUS_UNSUCCESSFUL;
 243     }
 244 }
 245
 246 NTSTATUS
 247 NTAPI
 248 USBPORT_SetRegistryKeyValue(IN PDEVICE_OBJECT DeviceObject,
 249                             IN BOOL UseDriverKey,
 250                             IN ULONG Type,
 251                             IN PCWSTR ValueNameString,
 252                             IN PVOID Data,
 253                             IN ULONG DataSize)
 254 {
 255     UNICODE_STRING ValueName;
 256     HANDLE KeyHandle;
 257     NTSTATUS Status;
 258
 259     DPRINT("USBPORT_SetRegistryKeyValue: ValueNameString - %S \n",
 260            ValueNameString);
 261
 262     if (UseDriverKey)
 263     {
 264         Status = IoOpenDeviceRegistryKey(DeviceObject,
 265                                          PLUGPLAY_REGKEY_DRIVER,
 266                                          STANDARD_RIGHTS_ALL,
 267                                          &KeyHandle);
 268     }
 269     else
 270     {
 271         Status = IoOpenDeviceRegistryKey(DeviceObject,
 272                                          PLUGPLAY_REGKEY_DEVICE,
 273                                          STANDARD_RIGHTS_ALL,
 274                                          &KeyHandle);
 275     }
 276
 277     if (NT_SUCCESS(Status))
 278     {
 279         RtlInitUnicodeString(&ValueName, ValueNameString);
 280
 281         Status = ZwSetValueKey(KeyHandle,
 282                                &ValueName,
 283                                0,
 284                                Type,
 285                                Data,
 286                                DataSize);
 287
 288         ZwClose(KeyHandle);
 289     }
 290
 291     return Status;
 292 }
 293
 294 NTSTATUS
 295 NTAPI
 296 USBPORT_GetRegistryKeyValueFullInfo(IN PDEVICE_OBJECT FdoDevice,
 297                                     IN PDEVICE_OBJECT PdoDevice,
 298                                     IN BOOL UseDriverKey,
 299                                     IN PCWSTR SourceString,
 300                                     IN ULONG LengthStr,
 301                                     IN PVOID Buffer,
 302                                     IN ULONG BufferLength)
 303 {
 304     NTSTATUS Status;
 305     PKEY_VALUE_FULL_INFORMATION KeyValue;
 306     UNICODE_STRING ValueName;
 307     HANDLE KeyHandle;
 308     ULONG LengthKey;
 309
 310     DPRINT("USBPORT_GetRegistryKeyValue: UseDriverKey - %x, SourceString - %S, LengthStr - %x, Buffer - %p, BufferLength - %x\n",
 311            UseDriverKey,
 312            SourceString,
 313            LengthStr,
 314            Buffer,
 315            BufferLength);
 316
 317     if (UseDriverKey)
 318     {
 319         Status = IoOpenDeviceRegistryKey(PdoDevice,
 320                                          PLUGPLAY_REGKEY_DRIVER,
 321                                          STANDARD_RIGHTS_ALL,
 322                                          &KeyHandle);
 323     }
 324     else
 325     {
 326         Status = IoOpenDeviceRegistryKey(PdoDevice,
 327                                          PLUGPLAY_REGKEY_DEVICE,
 328                                          STANDARD_RIGHTS_ALL,
 329                                          &KeyHandle);
 330     }
 331
 332     if (NT_SUCCESS(Status))
 333     {
 334         RtlInitUnicodeString(&ValueName, SourceString);
 335
 336         LengthKey = sizeof(KEY_VALUE_FULL_INFORMATION) +
 337                     LengthStr +
 338                     BufferLength;
 339
 340         KeyValue = ExAllocatePoolWithTag(PagedPool,
 341                                          LengthKey,
 342                                          USB_PORT_TAG);
 343
 344         if (KeyValue)
 345         {
 346             RtlZeroMemory(KeyValue, LengthKey);
 347
 348             Status = ZwQueryValueKey(KeyHandle,
 349                                      &ValueName,
 350                                      KeyValueFullInformation,
 351                                      KeyValue,
 352                                      LengthKey,
 353                                      &LengthKey);
 354
 355             if (NT_SUCCESS(Status))
 356             {
 357                 RtlCopyMemory(Buffer,
 358                               (PUCHAR)KeyValue + KeyValue->DataOffset,
 359                               BufferLength);
 360             }
 361
 362             ExFreePoolWithTag(KeyValue, USB_PORT_TAG);
 363         }
 364
 365         ZwClose(KeyHandle);
 366     }
 367
 368     return Status;
 369 }
 370
 371 MPSTATUS
 372 NTAPI
 373 USBPORT_GetMiniportRegistryKeyValue(IN PVOID MiniPortExtension,
 374                                     IN BOOL UseDriverKey,
 375                                     IN PCWSTR SourceString,
 376                                     IN SIZE_T LengthStr,
 377                                     IN PVOID Buffer,
 378                                     IN SIZE_T BufferLength)
 379 {
 380     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 381     PDEVICE_OBJECT FdoDevice;
 382     NTSTATUS Status;
 383
 384     DPRINT("USBPORT_GetMiniportRegistryKeyValue: MiniPortExtension - %p, UseDriverKey - %x, SourceString - %S, LengthStr - %x, Buffer - %p, BufferLength - %x\n",
 385            MiniPortExtension,
 386            UseDriverKey,
 387            SourceString,
 388            LengthStr,
 389            Buffer,
 390            BufferLength);
 391
 392     FdoExtension = (PUSBPORT_DEVICE_EXTENSION)((ULONG_PTR)MiniPortExtension -
 393                                                sizeof(USBPORT_DEVICE_EXTENSION));
 394
 395     FdoDevice = FdoExtension->CommonExtension.SelfDevice;
 396
 397     Status = USBPORT_GetRegistryKeyValueFullInfo(FdoDevice,
 398                                                  FdoExtension->CommonExtension.LowerPdoDevice,
 399                                                  UseDriverKey,
 400                                                  SourceString,
 401                                                  LengthStr,
 402                                                  Buffer,
 403                                                  BufferLength);
 404
 405     return USBPORT_NtStatusToMpStatus(Status);
 406 }
 407
 408 NTSTATUS
 409 NTAPI
 410 USBPORT_GetSetConfigSpaceData(IN PDEVICE_OBJECT FdoDevice,
 411                               IN BOOLEAN IsReadData,
 412                               IN PVOID Buffer,
 413                               IN ULONG Offset,
 414                               IN ULONG Length)
 415 {
 416     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 417     ULONG BytesReadWrite;
 418
 419     DPRINT("USBPORT_GetSetConfigSpaceData ... \n");
 420
 421     FdoExtension = FdoDevice->DeviceExtension;
 422
 423     BytesReadWrite = Length;
 424
 425     if (IsReadData)
 426     {
 427         RtlZeroMemory(Buffer, Length);
 428
 429         BytesReadWrite = (*FdoExtension->BusInterface.GetBusData)
 430                           (FdoExtension->BusInterface.Context,
 431                            PCI_WHICHSPACE_CONFIG,
 432                            Buffer,
 433                            Offset,
 434                            Length);
 435     }
 436     else
 437     {
 438         BytesReadWrite = (*FdoExtension->BusInterface.SetBusData)
 439                           (FdoExtension->BusInterface.Context,
 440                            PCI_WHICHSPACE_CONFIG,
 441                            Buffer,
 442                            Offset,
 443                            Length);
 444     }
 445
 446     if (BytesReadWrite == Length)
 447     {
 448         return STATUS_SUCCESS;
 449     }
 450
 451     return STATUS_UNSUCCESSFUL;
 452 }
 453
 454 MPSTATUS
 455 NTAPI
 456 USBPORT_ReadWriteConfigSpace(IN PVOID MiniPortExtension,
 457                              IN BOOLEAN IsReadData,
 458                              IN PVOID Buffer,
 459                              IN ULONG Offset,
 460                              IN ULONG Length)
 461 {
 462     NTSTATUS Status;
 463     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 464     PDEVICE_OBJECT FdoDevice;
 465
 466     DPRINT("USBPORT_ReadWriteConfigSpace: ... \n");
 467
 468     //FdoExtension->MiniPortExt = (PVOID)((ULONG_PTR)FdoExtension + sizeof(USBPORT_DEVICE_EXTENSION));
 469     FdoExtension = (PUSBPORT_DEVICE_EXTENSION)((ULONG_PTR)MiniPortExtension -
 470                                                sizeof(USBPORT_DEVICE_EXTENSION));
 471
 472     FdoDevice = FdoExtension->CommonExtension.SelfDevice;
 473
 474     Status = USBPORT_GetSetConfigSpaceData(FdoDevice,
 475                                            IsReadData,
 476                                            Buffer,
 477                                            Offset,
 478                                            Length);
 479
 480     return USBPORT_NtStatusToMpStatus(Status);
 481 }
 482
 483 NTSTATUS
 484 NTAPI
 485 USBPORT_USBDStatusToNtStatus(IN PURB Urb,
 486                              IN USBD_STATUS USBDStatus)
 487 {
 488     NTSTATUS Status;
 489
 490     if (USBD_ERROR(USBDStatus))
 491     {
 492         DPRINT1("USBPORT_USBDStatusToNtStatus: Urb - %p, USBDStatus - %x\n",
 493                 Urb,
 494                 USBDStatus);
 495     }
 496
 497     if (Urb)
 498         Urb->UrbHeader.Status = USBDStatus;
 499
 500     switch (USBDStatus)
 501     {
 502         case USBD_STATUS_SUCCESS:
 503             Status = STATUS_SUCCESS;
 504             break;
 505
 506         case USBD_STATUS_INSUFFICIENT_RESOURCES:
 507             Status = STATUS_INSUFFICIENT_RESOURCES;
 508             break;
 509
 510         case USBD_STATUS_DEVICE_GONE:
 511             Status = STATUS_DEVICE_NOT_CONNECTED;
 512             break;
 513
 514         case USBD_STATUS_CANCELED:
 515             Status = STATUS_CANCELLED;
 516             break;
 517
 518         case USBD_STATUS_NOT_SUPPORTED:
 519             Status = STATUS_NOT_SUPPORTED;
 520             break;
 521
 522         case USBD_STATUS_INVALID_URB_FUNCTION:
 523         case USBD_STATUS_INVALID_PARAMETER:
 524         case USBD_STATUS_INVALID_PIPE_HANDLE:
 525         case USBD_STATUS_BAD_START_FRAME:
 526             Status = STATUS_INVALID_PARAMETER;
 527             break;
 528
 529         default:
 530             if (USBD_ERROR(USBDStatus))
 531                 Status = STATUS_UNSUCCESSFUL;
 532             else
 533                 Status = STATUS_SUCCESS;
 534
 535             break;
 536     }
 537
 538     return Status;
 539 }
 540
 541 NTSTATUS
 542 NTAPI
 543 USBPORT_Wait(IN PVOID MiniPortExtension,
 544              IN ULONG Milliseconds)
 545 {
 546     LARGE_INTEGER Interval = {{0, 0}};
 547
 548     DPRINT("USBPORT_Wait: Milliseconds - %x\n", Milliseconds);
 549     Interval.QuadPart -= 10000 * Milliseconds + (KeQueryTimeIncrement() - 1);
 550     return KeDelayExecutionThread(KernelMode, FALSE, &Interval);
 551 }
 552
 553 VOID
 554 NTAPI
 555 USBPORT_MiniportInterrupts(IN PDEVICE_OBJECT FdoDevice,
 556                            IN BOOLEAN IsEnable)
 557 {
 558     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 559     PUSBPORT_REGISTRATION_PACKET Packet;
 560     BOOLEAN IsLock;
 561     KIRQL OldIrql;
 562
 563     DPRINT_INT("USBPORT_MiniportInterrupts: IsEnable - %p\n", IsEnable);
 564
 565     FdoExtension = FdoDevice->DeviceExtension;
 566     Packet = &FdoExtension->MiniPortInterface->Packet;
 567
 568     IsLock = (Packet->MiniPortFlags & USB_MINIPORT_FLAGS_NOT_LOCK_INT) == 0;
 569
 570     if (IsLock)
 571         KeAcquireSpinLock(&FdoExtension->MiniportSpinLock, &OldIrql);
 572
 573     if (IsEnable)
 574     {
 575         FdoExtension->Flags |= USBPORT_FLAG_INTERRUPT_ENABLED;
 576         Packet->EnableInterrupts(FdoExtension->MiniPortExt);
 577     }
 578     else
 579     {
 580         Packet->DisableInterrupts(FdoExtension->MiniPortExt);
 581         FdoExtension->Flags &= ~USBPORT_FLAG_INTERRUPT_ENABLED;
 582     }
 583
 584     if (IsLock)
 585         KeReleaseSpinLock(&FdoExtension->MiniportSpinLock, OldIrql);
 586 }
 587
 588 VOID
 589 NTAPI
 590 USBPORT_SoftInterruptDpc(IN PRKDPC Dpc,
 591                          IN PVOID DeferredContext,
 592                          IN PVOID SystemArgument1,
 593                          IN PVOID SystemArgument2)
 594 {
 595     PDEVICE_OBJECT FdoDevice;
 596     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 597
 598     DPRINT_INT("USBPORT_SoftInterruptDpc: ... \n");
 599
 600     FdoDevice = DeferredContext;
 601     FdoExtension = FdoDevice->DeviceExtension;
 602
 603     if (!KeInsertQueueDpc(&FdoExtension->IsrDpc, NULL, (PVOID)1))
 604     {
 605         InterlockedDecrement(&FdoExtension->IsrDpcCounter);
 606     }
 607 }
 608
 609 VOID
 610 NTAPI
 611 USBPORT_SoftInterrupt(IN PDEVICE_OBJECT FdoDevice)
 612 {
 613     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 614     LARGE_INTEGER DueTime = {{0, 0}};
 615
 616     DPRINT_INT("USBPORT_SoftInterrupt: ... \n");
 617
 618     FdoExtension = FdoDevice->DeviceExtension;
 619
 620     KeInitializeTimer(&FdoExtension->TimerSoftInterrupt);
 621
 622     KeInitializeDpc(&FdoExtension->SoftInterruptDpc,
 623                     USBPORT_SoftInterruptDpc,
 624                     FdoDevice);
 625
 626     DueTime.QuadPart -= 10000 + (KeQueryTimeIncrement() - 1);
 627
 628     KeSetTimer(&FdoExtension->TimerSoftInterrupt,
 629                DueTime,
 630                &FdoExtension->SoftInterruptDpc);
 631 }
 632
 633 VOID
 634 NTAPI
 635 USBPORT_InvalidateControllerHandler(IN PDEVICE_OBJECT FdoDevice,
 636                                     IN ULONG Type)
 637 {
 638     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 639
 640     DPRINT("USBPORT_InvalidateControllerHandler: Invalidate Type - %x\n",
 641            Type);
 642
 643     FdoExtension = FdoDevice->DeviceExtension;
 644
 645     switch (Type)
 646     {
 647         case USBPORT_INVALIDATE_CONTROLLER_RESET:
 648             DPRINT1("USBPORT_InvalidateControllerHandler: INVALIDATE_CONTROLLER_RESET UNIMPLEMENTED. FIXME. \n");
 649             break;
 650
 651         case USBPORT_INVALIDATE_CONTROLLER_SURPRISE_REMOVE:
 652             DPRINT1("USBPORT_InvalidateControllerHandler: INVALIDATE_CONTROLLER_SURPRISE_REMOVE UNIMPLEMENTED. FIXME. \n");
 653             break;
 654
 655         case USBPORT_INVALIDATE_CONTROLLER_SOFT_INTERRUPT:
 656             if (InterlockedIncrement(&FdoExtension->IsrDpcCounter))
 657             {
 658                 InterlockedDecrement(&FdoExtension->IsrDpcCounter);
 659             }
 660             else
 661             {
 662                 USBPORT_SoftInterrupt(FdoDevice);
 663             }
 664             break;
 665     }
 666 }
 667
 668 ULONG
 669 NTAPI
 670 USBPORT_InvalidateController(IN PVOID MiniPortExtension,
 671                              IN ULONG Type)
 672 {
 673     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 674     PDEVICE_OBJECT FdoDevice;
 675
 676     DPRINT("USBPORT_InvalidateController: Invalidate Type - %x\n", Type);
 677
 678     //FdoExtension->MiniPortExt = (PVOID)((ULONG_PTR)FdoExtension + sizeof(USBPORT_DEVICE_EXTENSION));
 679     FdoExtension = (PUSBPORT_DEVICE_EXTENSION)((ULONG_PTR)MiniPortExtension -
 680                                                sizeof(USBPORT_DEVICE_EXTENSION));
 681     FdoDevice = FdoExtension->CommonExtension.SelfDevice;
 682
 683     USBPORT_InvalidateControllerHandler(FdoDevice, Type);
 684
 685     return 0;
 686 }
 687
 688 ULONG
 689 NTAPI
 690 USBPORT_NotifyDoubleBuffer(IN PVOID MiniPortExtension,
 691                            IN PVOID MiniPortTransfer,
 692                            IN PVOID Buffer,
 693                            IN SIZE_T Length)
 694 {
 695     DPRINT1("USBPORT_NotifyDoubleBuffer: UNIMPLEMENTED. FIXME. \n");
 696     return 0;
 697 }
 698
 699 VOID
 700 NTAPI
 701 USBPORT_WorkerRequestDpc(IN PRKDPC Dpc,
 702                          IN PVOID DeferredContext,
 703                          IN PVOID SystemArgument1,
 704                          IN PVOID SystemArgument2)
 705 {
 706     PDEVICE_OBJECT FdoDevice;
 707     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 708
 709     DPRINT("USBPORT_WorkerRequestDpc: ... \n");
 710
 711     FdoDevice = DeferredContext;
 712     FdoExtension = FdoDevice->DeviceExtension;
 713
 714     if (!InterlockedIncrement(&FdoExtension->IsrDpcHandlerCounter))
 715     {
 716         USBPORT_DpcHandler(FdoDevice);
 717     }
 718
 719     InterlockedDecrement(&FdoExtension->IsrDpcHandlerCounter);
 720 }
 721
 722 VOID
 723 NTAPI
 724 USBPORT_DoneTransfer(IN PUSBPORT_TRANSFER Transfer)
 725 {
 726     PUSBPORT_ENDPOINT          Endpoint;
 727     PDEVICE_OBJECT             FdoDevice;
 728     PUSBPORT_DEVICE_EXTENSION  FdoExtension;
 729     PURB                       Urb;
 730     PIRP                       Irp;
 731     KIRQL                      CancelIrql;
 732     KIRQL                      OldIrql;
 733
 734     DPRINT_CORE("USBPORT_DoneTransfer: Transfer - %p\n", Transfer);
 735
 736     Endpoint = Transfer->Endpoint;
 737     FdoDevice = Endpoint->FdoDevice;
 738     FdoExtension = FdoDevice->DeviceExtension;
 739
 740     Urb = Transfer->Urb;
 741     Irp = Transfer->Irp;
 742
 743     KeAcquireSpinLock(&FdoExtension->FlushTransferSpinLock, &OldIrql);
 744
 745     if (Irp)
 746     {
 747         IoAcquireCancelSpinLock(&CancelIrql);
 748         IoSetCancelRoutine(Irp, NULL);
 749         IoReleaseCancelSpinLock(CancelIrql);
 750
 751         USBPORT_RemoveActiveTransferIrp(FdoDevice, Irp);
 752     }
 753
 754     KeReleaseSpinLock(&FdoExtension->FlushTransferSpinLock, OldIrql);
 755
 756     USBPORT_USBDStatusToNtStatus(Transfer->Urb, Transfer->USBDStatus);
 757     USBPORT_CompleteTransfer(Urb, Urb->UrbHeader.Status);
 758
 759     DPRINT_CORE("USBPORT_DoneTransfer: exit\n");
 760 }
 761
 762 VOID
 763 NTAPI
 764 USBPORT_FlushDoneTransfers(IN PDEVICE_OBJECT FdoDevice)
 765 {
 766     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 767     PLIST_ENTRY DoneTransferList;
 768     PUSBPORT_TRANSFER Transfer;
 769     PUSBPORT_ENDPOINT Endpoint;
 770     ULONG TransferCount;
 771     KIRQL OldIrql;
 772     BOOLEAN IsHasTransfers;
 773
 774     DPRINT_CORE("USBPORT_FlushDoneTransfers: ... \n");
 775
 776     FdoExtension = FdoDevice->DeviceExtension;
 777     DoneTransferList = &FdoExtension->DoneTransferList;
 778
 779     while (TRUE)
 780     {
 781         KeAcquireSpinLock(&FdoExtension->DoneTransferSpinLock, &OldIrql);
 782
 783         if (IsListEmpty(DoneTransferList))
 784             break;
 785
 786         Transfer = CONTAINING_RECORD(DoneTransferList->Flink,
 787                                      USBPORT_TRANSFER,
 788                                      TransferLink);
 789
 790         RemoveHeadList(DoneTransferList);
 791         KeReleaseSpinLock(&FdoExtension->DoneTransferSpinLock, OldIrql);
 792
 793         if (Transfer)
 794         {
 795             Endpoint = Transfer->Endpoint;
 796
 797             if ((Transfer->Flags & TRANSFER_FLAG_SPLITED))
 798             {
 799                 USBPORT_DoneSplitTransfer(Transfer);
 800             }
 801             else
 802             {
 803                 USBPORT_DoneTransfer(Transfer);
 804             }
 805
 806             IsHasTransfers = USBPORT_EndpointHasQueuedTransfers(FdoDevice,
 807                                                                 Endpoint,
 808                                                                 &TransferCount);
 809
 810             if (IsHasTransfers && !TransferCount)
 811             {
 812                 USBPORT_InvalidateEndpointHandler(FdoDevice,
 813                                                   Endpoint,
 814                                                   INVALIDATE_ENDPOINT_WORKER_DPC);
 815             }
 816         }
 817     }
 818
 819     KeReleaseSpinLock(&FdoExtension->DoneTransferSpinLock, OldIrql);
 820 }
 821
 822
 823 VOID
 824 NTAPI
 825 USBPORT_TransferFlushDpc(IN PRKDPC Dpc,
 826                          IN PVOID DeferredContext,
 827                          IN PVOID SystemArgument1,
 828                          IN PVOID SystemArgument2)
 829 {
 830     PDEVICE_OBJECT FdoDevice;
 831
 832     DPRINT_CORE("USBPORT_TransferFlushDpc: ... \n");
 833     FdoDevice = DeferredContext;
 834     USBPORT_FlushDoneTransfers(FdoDevice);
 835 }
 836
 837 BOOLEAN
 838 NTAPI
 839 USBPORT_QueueDoneTransfer(IN PUSBPORT_TRANSFER Transfer,
 840                           IN USBD_STATUS USBDStatus)
 841 {
 842     PDEVICE_OBJECT FdoDevice;
 843     PUSBPORT_DEVICE_EXTENSION  FdoExtension;
 844
 845     DPRINT_CORE("USBPORT_QueueDoneTransfer: Transfer - %p, USBDStatus - %p\n",
 846                 Transfer,
 847                 USBDStatus);
 848
 849     FdoDevice = Transfer->Endpoint->FdoDevice;
 850     FdoExtension = FdoDevice->DeviceExtension;
 851
 852     RemoveEntryList(&Transfer->TransferLink);
 853     Transfer->USBDStatus = USBDStatus;
 854
 855     ExInterlockedInsertTailList(&FdoExtension->DoneTransferList,
 856                                 &Transfer->TransferLink,
 857                                 &FdoExtension->DoneTransferSpinLock);
 858
 859     return KeInsertQueueDpc(&FdoExtension->TransferFlushDpc, NULL, NULL);
 860 }
 861
 862 VOID
 863 NTAPI
 864 USBPORT_DpcHandler(IN PDEVICE_OBJECT FdoDevice)
 865 {
 866     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 867     PUSBPORT_ENDPOINT Endpoint;
 868     PLIST_ENTRY Entry;
 869     LIST_ENTRY List;
 870     LONG LockCounter;
 871
 872     DPRINT("USBPORT_DpcHandler: ... \n");
 873
 874     FdoExtension = FdoDevice->DeviceExtension;
 875
 876     InitializeListHead(&List);
 877
 878     KeAcquireSpinLockAtDpcLevel(&FdoExtension->EndpointListSpinLock);
 879     Entry = FdoExtension->EndpointList.Flink;
 880
 881     while (Entry && Entry != &FdoExtension->EndpointList)
 882     {
 883         Endpoint = CONTAINING_RECORD(Entry,
 884                                      USBPORT_ENDPOINT,
 885                                      EndpointLink);
 886
 887         LockCounter = InterlockedIncrement(&Endpoint->LockCounter);
 888
 889         if (USBPORT_GetEndpointState(Endpoint) != USBPORT_ENDPOINT_ACTIVE ||
 890             LockCounter ||
 891             Endpoint->Flags & ENDPOINT_FLAG_ROOTHUB_EP0)
 892         {
 893             InterlockedDecrement(&Endpoint->LockCounter);
 894         }
 895         else
 896         {
 897             InsertTailList(&List, &Endpoint->DispatchLink);
 898
 899             if (Endpoint->WorkerLink.Flink && Endpoint->WorkerLink.Blink)
 900             {
 901                 RemoveEntryList(&Endpoint->WorkerLink);
 902
 903                 Endpoint->WorkerLink.Flink = NULL;
 904                 Endpoint->WorkerLink.Blink = NULL;
 905             }
 906         }
 907
 908         Entry = Endpoint->EndpointLink.Flink;
 909     }
 910
 911     KeReleaseSpinLockFromDpcLevel(&FdoExtension->EndpointListSpinLock);
 912
 913     while (!IsListEmpty(&List))
 914     {
 915         Endpoint = CONTAINING_RECORD(List.Flink,
 916                                      USBPORT_ENDPOINT,
 917                                      DispatchLink);
 918
 919         RemoveEntryList(List.Flink);
 920         Endpoint->DispatchLink.Flink = NULL;
 921         Endpoint->DispatchLink.Blink = NULL;
 922
 923         USBPORT_EndpointWorker(Endpoint, TRUE);
 924         USBPORT_FlushPendingTransfers(Endpoint);
 925     }
 926
 927     KeAcquireSpinLockAtDpcLevel(&FdoExtension->EndpointListSpinLock);
 928
 929     if (!IsListEmpty(&FdoExtension->WorkerList))
 930     {
 931         USBPORT_SignalWorkerThread(FdoDevice);
 932     }
 933
 934     KeReleaseSpinLockFromDpcLevel(&FdoExtension->EndpointListSpinLock);
 935
 936     USBPORT_FlushDoneTransfers(FdoDevice);
 937 }
 938
 939 VOID
 940 NTAPI
 941 USBPORT_IsrDpcHandler(IN PDEVICE_OBJECT FdoDevice,
 942                       IN BOOLEAN IsDpcHandler)
 943 {
 944     PUSBPORT_DEVICE_EXTENSION FdoExtension;
 945     PUSBPORT_REGISTRATION_PACKET Packet;
 946     PUSBPORT_ENDPOINT Endpoint;
 947     PLIST_ENTRY List;
 948     ULONG FrameNumber;
 949
 950     ASSERT(KeGetCurrentIrql() == DISPATCH_LEVEL);
 951
 952     DPRINT_CORE("USBPORT_IsrDpcHandler: IsDpcHandler - %x\n", IsDpcHandler);
 953
 954     FdoExtension = FdoDevice->DeviceExtension;
 955     Packet = &FdoExtension->MiniPortInterface->Packet;
 956
 957     if (InterlockedIncrement(&FdoExtension->IsrDpcHandlerCounter))
 958     {
 959         KeInsertQueueDpc(&FdoExtension->IsrDpc, NULL, NULL);
 960         InterlockedDecrement(&FdoExtension->IsrDpcHandlerCounter);
 961         return;
 962     }
 963
 964     for (List = ExInterlockedRemoveHeadList(&FdoExtension->EpStateChangeList,
 965                                             &FdoExtension->EpStateChangeSpinLock);
 966          List != NULL;
 967          List = ExInterlockedRemoveHeadList(&FdoExtension->EpStateChangeList,
 968                                             &FdoExtension->EpStateChangeSpinLock))
 969     {
 970         Endpoint = CONTAINING_RECORD(List,
 971                                      USBPORT_ENDPOINT,
 972                                      StateChangeLink);
 973
 974         DPRINT_CORE("USBPORT_IsrDpcHandler: Endpoint - %p\n", Endpoint);
 975
 976         KeAcquireSpinLockAtDpcLevel(&Endpoint->EndpointSpinLock);
 977
 978         KeAcquireSpinLockAtDpcLevel(&FdoExtension->MiniportSpinLock);
 979         FrameNumber = Packet->Get32BitFrameNumber(FdoExtension->MiniPortExt);
 980         KeReleaseSpinLockFromDpcLevel(&FdoExtension->MiniportSpinLock);
 981
 982         if (FrameNumber <= Endpoint->FrameNumber &&
 983             !(Endpoint->Flags & ENDPOINT_FLAG_NUKE))
 984         {
 985             KeReleaseSpinLockFromDpcLevel(&Endpoint->EndpointSpinLock);
 986
 987             ExInterlockedInsertHeadList(&FdoExtension->EpStateChangeList,
 988                                         &Endpoint->StateChangeLink,
 989                                         &FdoExtension->EpStateChangeSpinLock);
 990
 991             KeAcquireSpinLockAtDpcLevel(&FdoExtension->MiniportSpinLock);
 992             Packet->InterruptNextSOF(FdoExtension->MiniPortExt);
 993             KeReleaseSpinLockFromDpcLevel(&FdoExtension->MiniportSpinLock);
 994
 995             break;
 996         }
 997
 998         KeReleaseSpinLockFromDpcLevel(&Endpoint->EndpointSpinLock);
 999
1000         KeAcquireSpinLockAtDpcLevel(&Endpoint->StateChangeSpinLock);
1001         Endpoint->StateLast = Endpoint->StateNext;
1002         KeReleaseSpinLockFromDpcLevel(&Endpoint->StateChangeSpinLock);
1003
1004         DPRINT_CORE("USBPORT_IsrDpcHandler: Endpoint->StateLast - %x\n",
1005                     Endpoint->StateLast);
1006
1007         if (IsDpcHandler)
1008         {
1009             USBPORT_InvalidateEndpointHandler(FdoDevice,
1010                                               Endpoint,
1011                                               INVALIDATE_ENDPOINT_ONLY);
1012         }
1013         else
1014         {
1015             USBPORT_InvalidateEndpointHandler(FdoDevice,
1016                                               Endpoint,
1017                                               INVALIDATE_ENDPOINT_WORKER_THREAD);
1018         }
1019     }
1020
1021     if (IsDpcHandler)
1022     {
1023         USBPORT_DpcHandler(FdoDevice);
1024     }
1025
1026     InterlockedDecrement(&FdoExtension->IsrDpcHandlerCounter);
1027 }
1028
1029 VOID
1030 NTAPI
1031 USBPORT_IsrDpc(IN PRKDPC Dpc,
1032                IN PVOID DeferredContext,
1033                IN PVOID SystemArgument1,
1034                IN PVOID SystemArgument2)
1035 {
1036     PDEVICE_OBJECT FdoDevice;
1037     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1038     PUSBPORT_REGISTRATION_PACKET Packet;
1039     BOOLEAN InterruptEnable;
1040
1041     DPRINT_INT("USBPORT_IsrDpc: DeferredContext - %p, SystemArgument2 - %p\n",
1042                DeferredContext,
1043                SystemArgument2);
1044
1045     FdoDevice = DeferredContext;
1046     FdoExtension = FdoDevice->DeviceExtension;
1047     Packet = &FdoExtension->MiniPortInterface->Packet;
1048
1049     if (SystemArgument2)
1050     {
1051         InterlockedDecrement(&FdoExtension->IsrDpcCounter);
1052     }
1053
1054     KeAcquireSpinLockAtDpcLevel(&FdoExtension->MiniportInterruptsSpinLock);
1055     InterruptEnable = (FdoExtension->Flags & USBPORT_FLAG_INTERRUPT_ENABLED) ==
1056                        USBPORT_FLAG_INTERRUPT_ENABLED;
1057
1058     Packet->InterruptDpc(FdoExtension->MiniPortExt, InterruptEnable);
1059
1060     KeReleaseSpinLockFromDpcLevel(&FdoExtension->MiniportInterruptsSpinLock);
1061
1062     if (FdoExtension->Flags & USBPORT_FLAG_HC_SUSPEND &&
1063         FdoExtension->TimerFlags & USBPORT_TMFLAG_WAKE)
1064     {
1065         USBPORT_CompletePdoWaitWake(FdoDevice);
1066     }
1067     else
1068     {
1069         USBPORT_IsrDpcHandler(FdoDevice, TRUE);
1070     }
1071
1072     DPRINT_INT("USBPORT_IsrDpc: exit\n");
1073 }
1074
1075 BOOLEAN
1076 NTAPI
1077 USBPORT_InterruptService(IN PKINTERRUPT Interrupt,
1078                          IN PVOID ServiceContext)
1079 {
1080     PDEVICE_OBJECT FdoDevice;
1081     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1082     PUSBPORT_REGISTRATION_PACKET Packet;
1083     BOOLEAN Result = FALSE;
1084
1085     FdoDevice = ServiceContext;
1086     FdoExtension = FdoDevice->DeviceExtension;
1087     Packet = &FdoExtension->MiniPortInterface->Packet;
1088
1089     DPRINT_INT("USBPORT_InterruptService: FdoExtension[%p]->Flags - %08X\n",
1090            FdoExtension,
1091            FdoExtension->Flags);
1092
1093     if (FdoExtension->Flags & USBPORT_FLAG_INTERRUPT_ENABLED &&
1094         FdoExtension->MiniPortFlags & USBPORT_MPFLAG_INTERRUPTS_ENABLED)
1095     {
1096         Result = Packet->InterruptService(FdoExtension->MiniPortExt);
1097
1098         if (Result)
1099         {
1100             KeInsertQueueDpc(&FdoExtension->IsrDpc, NULL, NULL);
1101         }
1102     }
1103
1104     DPRINT_INT("USBPORT_InterruptService: return - %x\n", Result);
1105
1106     return Result;
1107 }
1108
1109 VOID
1110 NTAPI
1111 USBPORT_SignalWorkerThread(IN PDEVICE_OBJECT FdoDevice)
1112 {
1113     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1114     KIRQL OldIrql;
1115
1116     DPRINT_CORE("USBPORT_SignalWorkerThread ... \n");
1117
1118     FdoExtension = FdoDevice->DeviceExtension;
1119
1120     KeAcquireSpinLock(&FdoExtension->WorkerThreadEventSpinLock, &OldIrql);
1121     KeSetEvent(&FdoExtension->WorkerThreadEvent, EVENT_INCREMENT, FALSE);
1122     KeReleaseSpinLock(&FdoExtension->WorkerThreadEventSpinLock, OldIrql);
1123 }
1124
1125 VOID
1126 NTAPI
1127 USBPORT_WorkerThreadHandler(IN PDEVICE_OBJECT FdoDevice)
1128 {
1129     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1130     PUSBPORT_REGISTRATION_PACKET Packet;
1131     PLIST_ENTRY workerList;
1132     KIRQL OldIrql;
1133     PUSBPORT_ENDPOINT Endpoint;
1134     LIST_ENTRY list;
1135     BOOLEAN Result;
1136
1137     DPRINT_CORE("USBPORT_WorkerThreadHandler: ... \n");
1138
1139     FdoExtension = FdoDevice->DeviceExtension;
1140     Packet = &FdoExtension->MiniPortInterface->Packet;
1141
1142     KeAcquireSpinLock(&FdoExtension->MiniportSpinLock, &OldIrql);
1143
1144     if (!(FdoExtension->Flags & USBPORT_FLAG_HC_SUSPEND))
1145     {
1146         Packet->CheckController(FdoExtension->MiniPortExt);
1147     }
1148
1149     KeReleaseSpinLock(&FdoExtension->MiniportSpinLock, OldIrql);
1150
1151     InitializeListHead(&list);
1152
1153     USBPORT_FlushAllEndpoints(FdoDevice);
1154
1155     while (TRUE)
1156     {
1157         KeRaiseIrql(DISPATCH_LEVEL, &OldIrql);
1158         KeAcquireSpinLockAtDpcLevel(&FdoExtension->EndpointListSpinLock);
1159
1160         workerList = &FdoExtension->WorkerList;
1161
1162         if (IsListEmpty(workerList))
1163             break;
1164
1165         Endpoint = CONTAINING_RECORD(workerList->Flink,
1166                                      USBPORT_ENDPOINT,
1167                                      WorkerLink);
1168
1169         DPRINT_CORE("USBPORT_WorkerThreadHandler: Endpoint - %p\n", Endpoint);
1170
1171         RemoveHeadList(workerList);
1172         Endpoint->WorkerLink.Blink = NULL;
1173         Endpoint->WorkerLink.Flink = NULL;
1174
1175         KeReleaseSpinLockFromDpcLevel(&FdoExtension->EndpointListSpinLock);
1176
1177         Result = USBPORT_EndpointWorker(Endpoint, FALSE);
1178         KeAcquireSpinLockAtDpcLevel(&FdoExtension->EndpointListSpinLock);
1179
1180         if (Result)
1181         {
1182             if (Endpoint->FlushAbortLink.Flink == NULL ||
1183                 Endpoint->FlushAbortLink.Blink == NULL)
1184             {
1185                 InsertTailList(&list, &Endpoint->FlushAbortLink);
1186             }
1187         }
1188
1189         while (!IsListEmpty(&list))
1190         {
1191             Endpoint = CONTAINING_RECORD(list.Flink,
1192                                          USBPORT_ENDPOINT,
1193                                          FlushAbortLink);
1194
1195             RemoveHeadList(&list);
1196
1197             Endpoint->FlushAbortLink.Flink = NULL;
1198             Endpoint->FlushAbortLink.Blink = NULL;
1199
1200             if (Endpoint->WorkerLink.Flink == NULL ||
1201                 Endpoint->WorkerLink.Blink == NULL)
1202             {
1203                 InsertTailList(&FdoExtension->WorkerList,
1204                                &Endpoint->WorkerLink);
1205
1206                 USBPORT_SignalWorkerThread(FdoDevice);
1207             }
1208         }
1209
1210         KeReleaseSpinLockFromDpcLevel(&FdoExtension->EndpointListSpinLock);
1211         KeLowerIrql(OldIrql);
1212     }
1213
1214     KeReleaseSpinLockFromDpcLevel(&FdoExtension->EndpointListSpinLock);
1215     KeLowerIrql(OldIrql);
1216
1217     USBPORT_FlushClosedEndpointList(FdoDevice);
1218 }
1219
1220 VOID
1221 NTAPI
1222 USBPORT_DoRootHubCallback(IN PDEVICE_OBJECT FdoDevice)
1223 {
1224     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1225     PDEVICE_OBJECT PdoDevice;
1226     PUSBPORT_RHDEVICE_EXTENSION PdoExtension;
1227     PRH_INIT_CALLBACK RootHubInitCallback;
1228     PVOID RootHubInitContext;
1229
1230     FdoExtension = FdoDevice->DeviceExtension;
1231
1232     DPRINT("USBPORT_DoRootHubCallback: FdoDevice - %p\n", FdoDevice);
1233
1234     PdoDevice = FdoExtension->RootHubPdo;
1235
1236     if (PdoDevice)
1237     {
1238         PdoExtension = PdoDevice->DeviceExtension;
1239
1240         RootHubInitContext = PdoExtension->RootHubInitContext;
1241         RootHubInitCallback = PdoExtension->RootHubInitCallback;
1242
1243         PdoExtension->RootHubInitCallback = NULL;
1244         PdoExtension->RootHubInitContext = NULL;
1245
1246         if (RootHubInitCallback)
1247         {
1248             RootHubInitCallback(RootHubInitContext);
1249         }
1250     }
1251
1252     DPRINT("USBPORT_DoRootHubCallback: exit\n");
1253 }
1254
1255 VOID
1256 NTAPI
1257 USBPORT_SynchronizeRootHubCallback(IN PDEVICE_OBJECT FdoDevice,
1258                                    IN PDEVICE_OBJECT Usb2FdoDevice)
1259 {
1260     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1261     PUSBPORT_REGISTRATION_PACKET Packet;
1262     PUSBPORT_DEVICE_EXTENSION Usb2FdoExtension;
1263     PDEVICE_RELATIONS CompanionControllersList;
1264     PUSBPORT_DEVICE_EXTENSION CompanionFdoExtension;
1265     PDEVICE_OBJECT * Entry;
1266     ULONG ix;
1267
1268     DPRINT("USBPORT_SynchronizeRootHubCallback: FdoDevice - %p, Usb2FdoDevice - %p\n",
1269            FdoDevice,
1270            Usb2FdoDevice);
1271
1272     FdoExtension = FdoDevice->DeviceExtension;
1273     Packet = &FdoExtension->MiniPortInterface->Packet;
1274
1275     if (Usb2FdoDevice == NULL &&
1276         !(Packet->MiniPortFlags & USB_MINIPORT_FLAGS_USB2))
1277     {
1278         /* Not Companion USB11 Controller */
1279         USBPORT_DoRootHubCallback(FdoDevice);
1280
1281         FdoExtension->Flags &= ~USBPORT_FLAG_RH_INIT_CALLBACK;
1282         InterlockedCompareExchange(&FdoExtension->RHInitCallBackLock, 0, 1);
1283
1284         DPRINT("USBPORT_SynchronizeRootHubCallback: exit \n");
1285         return;
1286     }
1287
1288     /* USB2 or Companion USB11 */
1289
1290     DPRINT("USBPORT_SynchronizeRootHubCallback: FdoExtension->Flags - %p\n",
1291            FdoExtension->Flags);
1292
1293     if (!(FdoExtension->Flags & USBPORT_FLAG_COMPANION_HC))
1294     {
1295         KeWaitForSingleObject(&FdoExtension->ControllerSemaphore,
1296                               Executive,
1297                               KernelMode,
1298                               FALSE,
1299                               NULL);
1300
1301         FdoExtension->Flags |= USBPORT_FLAG_PWR_AND_CHIRP_LOCK;
1302
1303         if (!(FdoExtension->Flags & (USBPORT_FLAG_HC_SUSPEND |
1304                                      USBPORT_FLAG_POWER_AND_CHIRP_OK)))
1305         {
1306             USBPORT_RootHubPowerAndChirpAllCcPorts(FdoDevice);
1307             FdoExtension->Flags |= USBPORT_FLAG_POWER_AND_CHIRP_OK;
1308         }
1309
1310         FdoExtension->Flags &= ~USBPORT_FLAG_PWR_AND_CHIRP_LOCK;
1311
1312         KeReleaseSemaphore(&FdoExtension->ControllerSemaphore,
1313                            LOW_REALTIME_PRIORITY,
1314                            1,
1315                            FALSE);
1316
1317         CompanionControllersList = USBPORT_FindCompanionControllers(FdoDevice,
1318                                                                     FALSE,
1319                                                                     TRUE);
1320
1321         if (CompanionControllersList)
1322         {
1323             Entry = &CompanionControllersList->Objects[0];
1324
1325             for (ix = 0; ix < CompanionControllersList->Count; ++ix)
1326             {
1327                 CompanionFdoExtension = ((*Entry)->DeviceExtension);
1328
1329                 InterlockedCompareExchange(&CompanionFdoExtension->RHInitCallBackLock,
1330                                            0,
1331                                            1);
1332
1333                 ++Entry;
1334             }
1335
1336             ExFreePoolWithTag(CompanionControllersList, USB_PORT_TAG);
1337         }
1338
1339         USBPORT_DoRootHubCallback(FdoDevice);
1340
1341         FdoExtension->Flags &= ~USBPORT_FLAG_RH_INIT_CALLBACK;
1342         InterlockedCompareExchange(&FdoExtension->RHInitCallBackLock, 0, 1);
1343     }
1344     else
1345     {
1346         Usb2FdoExtension = Usb2FdoDevice->DeviceExtension;
1347
1348         USBPORT_Wait(FdoDevice, 50);
1349
1350         while (FdoExtension->RHInitCallBackLock)
1351         {
1352             USBPORT_Wait(FdoDevice, 10);
1353
1354             Usb2FdoExtension->Flags |= USBPORT_FLAG_RH_INIT_CALLBACK;
1355             USBPORT_SignalWorkerThread(Usb2FdoDevice);
1356         }
1357
1358         USBPORT_DoRootHubCallback(FdoDevice);
1359
1360         FdoExtension->Flags &= ~USBPORT_FLAG_RH_INIT_CALLBACK;
1361     }
1362
1363     DPRINT("USBPORT_SynchronizeRootHubCallback: exit \n");
1364 }
1365
1366 VOID
1367 NTAPI
1368 USBPORT_WorkerThread(IN PVOID StartContext)
1369 {
1370     PDEVICE_OBJECT FdoDevice;
1371     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1372     LARGE_INTEGER OldTime;
1373     LARGE_INTEGER NewTime;
1374     KIRQL OldIrql;
1375
1376     DPRINT_CORE("USBPORT_WorkerThread ... \n");
1377
1378     FdoDevice = StartContext;
1379     FdoExtension = FdoDevice->DeviceExtension;
1380
1381     FdoExtension->WorkerThread = KeGetCurrentThread();
1382
1383     do
1384     {
1385         KeQuerySystemTime(&OldTime);
1386
1387         KeWaitForSingleObject(&FdoExtension->WorkerThreadEvent,
1388                               Suspended,
1389                               KernelMode,
1390                               FALSE,
1391                               NULL);
1392
1393         KeQuerySystemTime(&NewTime);
1394
1395         KeAcquireSpinLock(&FdoExtension->WorkerThreadEventSpinLock, &OldIrql);
1396         KeResetEvent(&FdoExtension->WorkerThreadEvent);
1397         KeReleaseSpinLock(&FdoExtension->WorkerThreadEventSpinLock, OldIrql);
1398         DPRINT_CORE("USBPORT_WorkerThread: run \n");
1399
1400         if (FdoExtension->MiniPortFlags & USBPORT_MPFLAG_INTERRUPTS_ENABLED)
1401         {
1402             USBPORT_DoSetPowerD0(FdoDevice);
1403
1404             if (FdoExtension->Flags & USBPORT_FLAG_RH_INIT_CALLBACK)
1405             {
1406                 PDEVICE_OBJECT USB2FdoDevice = NULL;
1407
1408                 USB2FdoDevice = USBPORT_FindUSB2Controller(FdoDevice);
1409                 USBPORT_SynchronizeRootHubCallback(FdoDevice, USB2FdoDevice);
1410             }
1411         }
1412
1413         USBPORT_WorkerThreadHandler(FdoDevice);
1414     }
1415     while (!(FdoExtension->Flags & USBPORT_FLAG_WORKER_THREAD_ON));
1416
1417     PsTerminateSystemThread(0);
1418 }
1419
1420 NTSTATUS
1421 NTAPI
1422 USBPORT_CreateWorkerThread(IN PDEVICE_OBJECT FdoDevice)
1423 {
1424     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1425     NTSTATUS Status;
1426
1427     DPRINT("USBPORT_CreateWorkerThread ... \n");
1428
1429     FdoExtension = FdoDevice->DeviceExtension;
1430
1431     FdoExtension->Flags &= ~USBPORT_FLAG_WORKER_THREAD_ON;
1432
1433     KeInitializeEvent(&FdoExtension->WorkerThreadEvent,
1434                       NotificationEvent,
1435                       FALSE);
1436
1437     Status = PsCreateSystemThread(&FdoExtension->WorkerThreadHandle,
1438                                   THREAD_ALL_ACCESS,
1439                                   NULL,
1440                                   NULL,
1441                                   NULL,
1442                                   USBPORT_WorkerThread,
1443                                   (PVOID)FdoDevice);
1444
1445     return Status;
1446 }
1447
1448 VOID
1449 NTAPI
1450 USBPORT_SynchronizeControllersStart(IN PDEVICE_OBJECT FdoDevice)
1451 {
1452     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1453     PDEVICE_OBJECT PdoDevice;
1454     PUSBPORT_RHDEVICE_EXTENSION PdoExtension;
1455     PDEVICE_OBJECT USB2FdoDevice = NULL;
1456     PUSBPORT_DEVICE_EXTENSION USB2FdoExtension;
1457     BOOLEAN IsOn;
1458
1459     DPRINT_TIMER("USBPORT_SynchronizeControllersStart: FdoDevice - %p\n",
1460                  FdoDevice);
1461
1462     FdoExtension = FdoDevice->DeviceExtension;
1463
1464     PdoDevice = FdoExtension->RootHubPdo;
1465
1466     if (!PdoDevice)
1467     {
1468         return;
1469     }
1470
1471     PdoExtension = PdoDevice->DeviceExtension;
1472
1473     if (PdoExtension->RootHubInitCallback == NULL ||
1474         FdoExtension->Flags & USBPORT_FLAG_RH_INIT_CALLBACK)
1475     {
1476         return;
1477     }
1478
1479     DPRINT_TIMER("USBPORT_SynchronizeControllersStart: Flags - %p\n",
1480                  FdoExtension->Flags);
1481
1482     if (FdoExtension->Flags & USBPORT_FLAG_COMPANION_HC)
1483     {
1484         IsOn = FALSE;
1485
1486         USB2FdoDevice = USBPORT_FindUSB2Controller(FdoDevice);
1487
1488         DPRINT_TIMER("USBPORT_SynchronizeControllersStart: USB2FdoDevice - %p\n",
1489                      USB2FdoDevice);
1490
1491         if (USB2FdoDevice)
1492         {
1493             USB2FdoExtension = USB2FdoDevice->DeviceExtension;
1494
1495             if (USB2FdoExtension->CommonExtension.PnpStateFlags &
1496                 USBPORT_PNP_STATE_STARTED)
1497             {
1498                 IsOn = TRUE;
1499             }
1500         }
1501
1502         if (!(FdoExtension->Flags & USBPORT_FLAG_NO_HACTION))
1503         {
1504             goto Start;
1505         }
1506
1507         USB2FdoDevice = NULL;
1508     }
1509
1510     IsOn = TRUE;
1511
1512   Start:
1513
1514     if (IsOn &&
1515         !InterlockedCompareExchange(&FdoExtension->RHInitCallBackLock, 1, 0))
1516     {
1517         FdoExtension->Flags |= USBPORT_FLAG_RH_INIT_CALLBACK;
1518         USBPORT_SignalWorkerThread(FdoDevice);
1519
1520         if (USB2FdoDevice)
1521         {
1522             USB2FdoExtension = USB2FdoDevice->DeviceExtension;
1523
1524             USB2FdoExtension->Flags |= USBPORT_FLAG_RH_INIT_CALLBACK;
1525             USBPORT_SignalWorkerThread(USB2FdoDevice);
1526         }
1527     }
1528
1529     DPRINT_TIMER("USBPORT_SynchronizeControllersStart: exit\n");
1530 }
1531
1532 VOID
1533 NTAPI
1534 USBPORT_TimerDpc(IN PRKDPC Dpc,
1535                  IN PVOID DeferredContext,
1536                  IN PVOID SystemArgument1,
1537                  IN PVOID SystemArgument2)
1538 {
1539     PDEVICE_OBJECT FdoDevice;
1540     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1541     PUSBPORT_REGISTRATION_PACKET Packet;
1542     LARGE_INTEGER DueTime = {{0, 0}};
1543     ULONG TimerFlags;
1544     PTIMER_WORK_QUEUE_ITEM IdleQueueItem;
1545     KIRQL OldIrql;
1546     KIRQL TimerOldIrql;
1547
1548     DPRINT_TIMER("USBPORT_TimerDpc: Dpc - %p, DeferredContext - %p\n",
1549            Dpc,
1550            DeferredContext);
1551
1552     FdoDevice = DeferredContext;
1553     FdoExtension = FdoDevice->DeviceExtension;
1554     Packet = &FdoExtension->MiniPortInterface->Packet;
1555
1556     KeAcquireSpinLock(&FdoExtension->TimerFlagsSpinLock, &TimerOldIrql);
1557
1558     TimerFlags = FdoExtension->TimerFlags;
1559
1560     DPRINT_TIMER("USBPORT_TimerDpc: Flags - %p, TimerFlags - %p\n",
1561                  FdoExtension->Flags,
1562                  TimerFlags);
1563
1564     if (FdoExtension->Flags & USBPORT_FLAG_HC_SUSPEND &&
1565         FdoExtension->Flags & USBPORT_FLAG_HC_WAKE_SUPPORT &&
1566         !(TimerFlags & USBPORT_TMFLAG_HC_RESUME))
1567     {
1568         KeAcquireSpinLock(&FdoExtension->MiniportSpinLock, &OldIrql);
1569         Packet->PollController(FdoExtension->MiniPortExt);
1570         KeReleaseSpinLock(&FdoExtension->MiniportSpinLock, OldIrql);
1571     }
1572
1573     USBPORT_SynchronizeControllersStart(FdoDevice);
1574
1575     if (TimerFlags & USBPORT_TMFLAG_HC_SUSPENDED)
1576     {
1577         USBPORT_BadRequestFlush(FdoDevice);
1578         goto Exit;
1579     }
1580
1581     KeAcquireSpinLock(&FdoExtension->MiniportSpinLock, &OldIrql);
1582
1583     if (!(FdoExtension->Flags & USBPORT_FLAG_HC_SUSPEND))
1584     {
1585         Packet->CheckController(FdoExtension->MiniPortExt);
1586     }
1587
1588     KeReleaseSpinLock(&FdoExtension->MiniportSpinLock, OldIrql);
1589
1590     if (FdoExtension->Flags & USBPORT_FLAG_HC_POLLING)
1591     {
1592         KeAcquireSpinLock(&FdoExtension->MiniportSpinLock, &OldIrql);
1593         Packet->PollController(FdoExtension->MiniPortExt);
1594         KeReleaseSpinLock(&FdoExtension->MiniportSpinLock, OldIrql);
1595     }
1596
1597     USBPORT_IsrDpcHandler(FdoDevice, FALSE);
1598
1599     DPRINT_TIMER("USBPORT_TimerDpc: USBPORT_TimeoutAllEndpoints UNIMPLEMENTED.\n");
1600     //USBPORT_TimeoutAllEndpoints(FdoDevice);
1601     DPRINT_TIMER("USBPORT_TimerDpc: USBPORT_CheckIdleEndpoints UNIMPLEMENTED.\n");
1602     //USBPORT_CheckIdleEndpoints(FdoDevice);
1603
1604     USBPORT_BadRequestFlush(FdoDevice);
1605
1606     if (FdoExtension->IdleLockCounter > -1 &&
1607         !(TimerFlags & USBPORT_TMFLAG_IDLE_QUEUEITEM_ON))
1608     {
1609         IdleQueueItem = ExAllocatePoolWithTag(NonPagedPool,
1610                                               sizeof(TIMER_WORK_QUEUE_ITEM),
1611                                               USB_PORT_TAG);
1612
1613         DPRINT("USBPORT_TimerDpc: IdleLockCounter - %x, IdleQueueItem - %p\n",
1614                FdoExtension->IdleLockCounter,
1615                IdleQueueItem);
1616
1617         if (IdleQueueItem)
1618         {
1619             RtlZeroMemory(IdleQueueItem, sizeof(TIMER_WORK_QUEUE_ITEM));
1620
1621             IdleQueueItem->WqItem.List.Flink = NULL;
1622             IdleQueueItem->WqItem.WorkerRoutine = USBPORT_DoIdleNotificationCallback;
1623             IdleQueueItem->WqItem.Parameter = IdleQueueItem;
1624
1625             IdleQueueItem->FdoDevice = FdoDevice;
1626             IdleQueueItem->Context = 0;
1627
1628             FdoExtension->TimerFlags |= USBPORT_TMFLAG_IDLE_QUEUEITEM_ON;
1629
1630             ExQueueWorkItem(&IdleQueueItem->WqItem, CriticalWorkQueue);
1631         }
1632     }
1633
1634 Exit:
1635
1636     KeReleaseSpinLock(&FdoExtension->TimerFlagsSpinLock, TimerOldIrql);
1637
1638     if (TimerFlags & USBPORT_TMFLAG_TIMER_QUEUED)
1639     {
1640         DueTime.QuadPart -= FdoExtension->TimerValue * 10000 +
1641                             (KeQueryTimeIncrement() - 1);
1642
1643         KeSetTimer(&FdoExtension->TimerObject,
1644                    DueTime,
1645                    &FdoExtension->TimerDpc);
1646     }
1647
1648     DPRINT_TIMER("USBPORT_TimerDpc: exit\n");
1649 }
1650
1651 BOOLEAN
1652 NTAPI
1653 USBPORT_StartTimer(IN PDEVICE_OBJECT FdoDevice,
1654                    IN ULONG Time)
1655 {
1656     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1657     LARGE_INTEGER DueTime = {{0, 0}};
1658     ULONG TimeIncrement;
1659     BOOLEAN Result;
1660
1661     DPRINT_TIMER("USBPORT_StartTimer: FdoDevice - %p, Time - %x\n",
1662            FdoDevice,
1663            Time);
1664
1665     FdoExtension = FdoDevice->DeviceExtension;
1666
1667     TimeIncrement = KeQueryTimeIncrement();
1668
1669     FdoExtension->TimerFlags |= USBPORT_TMFLAG_TIMER_QUEUED;
1670     FdoExtension->TimerValue = Time;
1671
1672     KeInitializeTimer(&FdoExtension->TimerObject);
1673     KeInitializeDpc(&FdoExtension->TimerDpc, USBPORT_TimerDpc, FdoDevice);
1674
1675     DueTime.QuadPart -= 10000 * Time + (TimeIncrement - 1);
1676
1677     Result = KeSetTimer(&FdoExtension->TimerObject,
1678                         DueTime,
1679                         &FdoExtension->TimerDpc);
1680
1681     return Result;
1682 }
1683
1684 PUSBPORT_COMMON_BUFFER_HEADER
1685 NTAPI
1686 USBPORT_AllocateCommonBuffer(IN PDEVICE_OBJECT FdoDevice,
1687                              IN SIZE_T BufferLength)
1688 {
1689     PUSBPORT_COMMON_BUFFER_HEADER HeaderBuffer = NULL;
1690     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1691     PDMA_ADAPTER DmaAdapter;
1692     PDMA_OPERATIONS DmaOperations;
1693     SIZE_T HeaderSize;
1694     ULONG Length = 0;
1695     ULONG LengthPadded;
1696     PHYSICAL_ADDRESS LogicalAddress;
1697     ULONG_PTR BaseVA;
1698     ULONG_PTR StartBufferVA;
1699     ULONG_PTR StartBufferPA;
1700
1701     DPRINT("USBPORT_AllocateCommonBuffer: FdoDevice - %p, BufferLength - %p\n",
1702            FdoDevice,
1703            BufferLength);
1704
1705     if (BufferLength == 0)
1706         goto Exit;
1707
1708     FdoExtension = FdoDevice->DeviceExtension;
1709
1710     DmaAdapter = FdoExtension->DmaAdapter;
1711     DmaOperations = DmaAdapter->DmaOperations;
1712
1713     HeaderSize = sizeof(USBPORT_COMMON_BUFFER_HEADER);
1714     Length = ROUND_TO_PAGES(BufferLength + HeaderSize);
1715     LengthPadded = Length - (BufferLength + HeaderSize);
1716
1717     BaseVA = (ULONG_PTR)DmaOperations->AllocateCommonBuffer(DmaAdapter,
1718                                                             Length,
1719                                                             &LogicalAddress,
1720                                                             TRUE);
1721
1722     if (!BaseVA)
1723         goto Exit;
1724
1725     StartBufferVA = BaseVA & ~(PAGE_SIZE - 1);
1726     StartBufferPA = LogicalAddress.LowPart & ~(PAGE_SIZE - 1);
1727
1728     HeaderBuffer = (PUSBPORT_COMMON_BUFFER_HEADER)(StartBufferVA +
1729                                                    BufferLength +
1730                                                    LengthPadded);
1731
1732     HeaderBuffer->Length = Length;
1733     HeaderBuffer->BaseVA = BaseVA;
1734     HeaderBuffer->LogicalAddress = LogicalAddress;
1735
1736     HeaderBuffer->BufferLength = BufferLength + LengthPadded;
1737     HeaderBuffer->VirtualAddress = StartBufferVA;
1738     HeaderBuffer->PhysicalAddress = StartBufferPA;
1739
1740     RtlZeroMemory((PVOID)StartBufferVA, BufferLength + LengthPadded);
1741
1742 Exit:
1743     return HeaderBuffer;
1744 }
1745
1746 VOID
1747 NTAPI
1748 USBPORT_FreeCommonBuffer(IN PDEVICE_OBJECT FdoDevice,
1749                          IN PUSBPORT_COMMON_BUFFER_HEADER HeaderBuffer)
1750 {
1751     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1752     PDMA_ADAPTER DmaAdapter;
1753     PDMA_OPERATIONS DmaOperations;
1754
1755     DPRINT("USBPORT_FreeCommonBuffer: ... \n");
1756
1757     FdoExtension = FdoDevice->DeviceExtension;
1758
1759     DmaAdapter = FdoExtension->DmaAdapter;
1760     DmaOperations = DmaAdapter->DmaOperations;
1761
1762     DmaOperations->FreeCommonBuffer(FdoExtension->DmaAdapter,
1763                                     HeaderBuffer->Length,
1764                                     HeaderBuffer->LogicalAddress,
1765                                     (PVOID)HeaderBuffer->VirtualAddress,
1766                                     TRUE);
1767 }
1768
1769 PUSBPORT_MINIPORT_INTERFACE
1770 NTAPI
1771 USBPORT_FindMiniPort(IN PDRIVER_OBJECT DriverObject)
1772 {
1773     KIRQL OldIrql;
1774     PLIST_ENTRY List;
1775     PUSBPORT_MINIPORT_INTERFACE MiniPortInterface;
1776     BOOLEAN IsFound = FALSE;
1777
1778     DPRINT("USBPORT_FindMiniPort: ... \n");
1779
1780     KeAcquireSpinLock(&USBPORT_SpinLock, &OldIrql);
1781
1782     for (List = USBPORT_MiniPortDrivers.Flink;
1783          List != &USBPORT_MiniPortDrivers;
1784          List = List->Flink)
1785     {
1786         MiniPortInterface = CONTAINING_RECORD(List,
1787                                               USBPORT_MINIPORT_INTERFACE,
1788                                               DriverLink);
1789
1790         if (MiniPortInterface->DriverObject == DriverObject)
1791         {
1792             DPRINT("USBPORT_FindMiniPort: find MiniPortInterface - %p\n",
1793                    MiniPortInterface);
1794
1795             IsFound = TRUE;
1796             break;
1797         }
1798     }
1799
1800     KeReleaseSpinLock(&USBPORT_SpinLock, OldIrql);
1801
1802     if (IsFound)
1803         return MiniPortInterface;
1804     else
1805         return NULL;
1806
1807 }
1808
1809 NTSTATUS
1810 NTAPI
1811 USBPORT_AddDevice(IN PDRIVER_OBJECT DriverObject,
1812                   IN PDEVICE_OBJECT PhysicalDeviceObject)
1813 {
1814     NTSTATUS Status;
1815     PUSBPORT_MINIPORT_INTERFACE MiniPortInterface;
1816     ULONG DeviceNumber = 0;
1817     WCHAR CharDeviceName[64];
1818     UNICODE_STRING DeviceName;
1819     PDEVICE_OBJECT DeviceObject;
1820     PUSBPORT_DEVICE_EXTENSION FdoExtension;
1821     PUSBPORT_COMMON_DEVICE_EXTENSION FdoCommonExtension;
1822     PDEVICE_OBJECT LowerDevice;
1823     ULONG Length;
1824
1825     DPRINT("USBPORT_AddDevice: DriverObject - %p, PhysicalDeviceObject - %p\n",
1826            DriverObject,
1827            PhysicalDeviceObject);
1828
1829     MiniPortInterface = USBPORT_FindMiniPort(DriverObject);
1830
1831     if (!MiniPortInterface)
1832     {
1833         DPRINT("USBPORT_AddDevice: USBPORT_FindMiniPort not found MiniPortInterface\n");
1834         return STATUS_UNSUCCESSFUL;
1835     }
1836
1837     while (TRUE)
1838     {
1839         /* Construct device name */
1840         RtlStringCbPrintfW(CharDeviceName,
1841                            sizeof(CharDeviceName),
1842                            L"\\Device\\USBFDO-%d",
1843                            DeviceNumber);
1844
1845         RtlInitUnicodeString(&DeviceName, CharDeviceName);
1846
1847         Length = sizeof(USBPORT_DEVICE_EXTENSION) +
1848                  MiniPortInterface->Packet.MiniPortExtensionSize;
1849
1850         /* Create device */
1851         Status = IoCreateDevice(DriverObject,
1852                                 Length,
1853                                 &DeviceName,
1854                                 FILE_DEVICE_CONTROLLER,
1855                                 0,
1856                                 FALSE,
1857                                 &DeviceObject);
1858
1859         /* Check for success */
1860         if (NT_SUCCESS(Status)) break;
1861
1862         /* Is there a device object with that same name */
1863         if ((Status == STATUS_OBJECT_NAME_EXISTS) ||
1864             (Status == STATUS_OBJECT_NAME_COLLISION))
1865         {
1866             /* Try the next name */
1867             DeviceNumber++;
1868             continue;
1869         }
1870
1871         /* Bail out on other errors */
1872         if (!NT_SUCCESS(Status))
1873         {
1874             DPRINT1("USBPORT_AddDevice: failed to create %wZ, Status %x\n",
1875                     &DeviceName,
1876                     Status);
1877
1878             return Status;
1879         }
1880     }
1881
1882     DPRINT("USBPORT_AddDevice: created device %p <%wZ>, Status %x\n",
1883            DeviceObject,
1884            &DeviceName,
1885            Status);
1886
1887     FdoExtension = DeviceObject->DeviceExtension;
1888     FdoCommonExtension = &FdoExtension->CommonExtension;
1889
1890     RtlZeroMemory(FdoExtension, sizeof(USBPORT_DEVICE_EXTENSION));
1891
1892     FdoCommonExtension->SelfDevice = DeviceObject;
1893     FdoCommonExtension->LowerPdoDevice = PhysicalDeviceObject;
1894     FdoCommonExtension->IsPDO = FALSE;
1895
1896     LowerDevice = IoAttachDeviceToDeviceStack(DeviceObject,
1897                                               PhysicalDeviceObject);
1898
1899     FdoCommonExtension->LowerDevice = LowerDevice;
1900
1901     FdoCommonExtension->DevicePowerState = PowerDeviceD3;
1902
1903     FdoExtension->MiniPortExt = (PVOID)((ULONG_PTR)FdoExtension +
1904                                         sizeof(USBPORT_DEVICE_EXTENSION));
1905
1906     FdoExtension->MiniPortInterface = MiniPortInterface;
1907     FdoExtension->FdoNameNumber = DeviceNumber;
1908
1909     KeInitializeSemaphore(&FdoExtension->DeviceSemaphore, 1, 1);
1910     KeInitializeSemaphore(&FdoExtension->ControllerSemaphore, 1, 1);
1911
1912     InitializeListHead(&FdoExtension->EndpointList);
1913     InitializeListHead(&FdoExtension->DoneTransferList);
1914     InitializeListHead(&FdoExtension->WorkerList);
1915     InitializeListHead(&FdoExtension->EpStateChangeList);
1916     InitializeListHead(&FdoExtension->MapTransferList);
1917     InitializeListHead(&FdoExtension->DeviceHandleList);
1918     InitializeListHead(&FdoExtension->IdleIrpList);
1919     InitializeListHead(&FdoExtension->BadRequestList);
1920     InitializeListHead(&FdoExtension->EndpointClosedList);
1921
1922     DeviceObject->Flags &= ~DO_DEVICE_INITIALIZING;
1923
1924     return Status;
1925 }
1926
1927 VOID
1928 NTAPI
1929 USBPORT_Unload(IN PDRIVER_OBJECT DriverObject)
1930 {
1931     PUSBPORT_MINIPORT_INTERFACE MiniPortInterface;
1932
1933     DPRINT1("USBPORT_Unload: FIXME!\n");
1934
1935     MiniPortInterface = USBPORT_FindMiniPort(DriverObject);
1936
1937     if (!MiniPortInterface)
1938     {
1939         DPRINT("USBPORT_Unload: CRITICAL ERROR!!! USBPORT_FindMiniPort not found MiniPortInterface\n");
1940         KeBugCheckEx(BUGCODE_USB_DRIVER, 1, 0, 0, 0);
1941     }
1942
1943     DPRINT1("USBPORT_Unload: UNIMPLEMENTED. FIXME. \n");
1944     //MiniPortInterface->DriverUnload(DriverObject); // Call MiniPort _HCI_Unload
1945 }
1946
1947 VOID
1948 NTAPI
1949 USBPORT_MiniportCompleteTransfer(IN PVOID MiniPortExtension,
1950                                  IN PVOID MiniPortEndpoint,
1951                                  IN PVOID TransferParameters,
1952                                  IN USBD_STATUS USBDStatus,
1953                                  IN ULONG TransferLength)
1954 {
1955     PUSBPORT_TRANSFER Transfer;
1956     PUSBPORT_TRANSFER ParentTransfer;
1957     PUSBPORT_TRANSFER SplitTransfer;
1958     PLIST_ENTRY SplitHead;
1959     PLIST_ENTRY Entry;
1960     KIRQL OldIrql;
1961
1962     DPRINT_CORE("USBPORT_MiniportCompleteTransfer: USBDStatus - %x, TransferLength - %x\n",
1963                 USBDStatus,
1964                 TransferLength);
1965
1966     Transfer = CONTAINING_RECORD(TransferParameters,
1967                                  USBPORT_TRANSFER,
1968                                  TransferParameters);
1969
1970     Transfer->Flags |= TRANSFER_FLAG_COMPLETED;
1971     Transfer->CompletedTransferLen = TransferLength;
1972
1973     if (((Transfer->Flags & TRANSFER_FLAG_SPLITED) == 0) ||
1974         TransferLength >= Transfer->TransferParameters.TransferBufferLength)
1975     {
1976         goto Exit;
1977     }
1978
1979     ParentTransfer = Transfer->ParentTransfer;
1980
1981     KeAcquireSpinLock(&ParentTransfer->TransferSpinLock, &OldIrql);
1982
1983     if (IsListEmpty(&ParentTransfer->SplitTransfersList))
1984     {
1985         goto Exit;
1986     }
1987
1988     SplitHead = &ParentTransfer->SplitTransfersList;
1989     Entry = SplitHead->Flink;
1990
1991     while (Entry && !IsListEmpty(SplitHead))
1992     {
1993         SplitTransfer = CONTAINING_RECORD(Entry,
1994                                           USBPORT_TRANSFER,
1995                                           SplitLink);
1996
1997         if (!(SplitTransfer->Flags & TRANSFER_FLAG_SUBMITED))
1998         {
1999             DPRINT1("USBPORT_MiniportCompleteTransfer: SplitTransfer->Flags - %X\n",
2000                     SplitTransfer->Flags);
2001             //Add TRANSFER_FLAG_xxx
2002         }
2003
2004         Entry = Entry->Flink;
2005     }
2006
2007     KeReleaseSpinLock(&ParentTransfer->TransferSpinLock, OldIrql);
2008
2009 Exit:
2010     USBPORT_QueueDoneTransfer(Transfer, USBDStatus);
2011 }
2012
2013 VOID
2014 NTAPI
2015 USBPORT_AsyncTimerDpc(IN PRKDPC Dpc,
2016                       IN PVOID DeferredContext,
2017                       IN PVOID SystemArgument1,
2018                       IN PVOID SystemArgument2)
2019 {
2020     PDEVICE_OBJECT FdoDevice;
2021     PUSBPORT_DEVICE_EXTENSION FdoExtension;
2022     PUSBPORT_ASYNC_CALLBACK_DATA AsyncCallbackData;
2023
2024     DPRINT("USBPORT_AsyncTimerDpc: ... \n");
2025
2026     AsyncCallbackData = DeferredContext;
2027     FdoDevice = AsyncCallbackData->FdoDevice;
2028     FdoExtension = FdoDevice->DeviceExtension;
2029
2030     (*AsyncCallbackData->CallbackFunction)(FdoExtension->MiniPortExt,
2031                                            &AsyncCallbackData->CallbackContext);
2032
2033     ExFreePoolWithTag(AsyncCallbackData, USB_PORT_TAG);
2034 }
2035
2036 ULONG
2037 NTAPI
2038 USBPORT_RequestAsyncCallback(IN PVOID MiniPortExtension,
2039                              IN ULONG TimerValue,
2040                              IN PVOID Buffer,
2041                              IN SIZE_T Length,
2042                              IN ASYNC_TIMER_CALLBACK * Callback)
2043 {
2044     PUSBPORT_DEVICE_EXTENSION FdoExtension;
2045     PDEVICE_OBJECT FdoDevice;
2046     PUSBPORT_ASYNC_CALLBACK_DATA AsyncCallbackData;
2047     LARGE_INTEGER DueTime = {{0, 0}};
2048
2049     DPRINT("USBPORT_RequestAsyncCallback: ... \n");
2050
2051     FdoExtension = (PUSBPORT_DEVICE_EXTENSION)((ULONG_PTR)MiniPortExtension -
2052                                                sizeof(USBPORT_DEVICE_EXTENSION));
2053
2054     FdoDevice = FdoExtension->CommonExtension.SelfDevice;
2055
2056     AsyncCallbackData = ExAllocatePoolWithTag(NonPagedPool,
2057                                               sizeof(USBPORT_ASYNC_CALLBACK_DATA) + Length,
2058                                               USB_PORT_TAG);
2059
2060     if (!AsyncCallbackData)
2061     {
2062         DPRINT1("USBPORT_RequestAsyncCallback: Not allocated AsyncCallbackData!\n");
2063         return 0;
2064     }
2065
2066     RtlZeroMemory(AsyncCallbackData,
2067                   sizeof(USBPORT_ASYNC_CALLBACK_DATA) + Length);
2068
2069     if (Length)
2070     {
2071         RtlCopyMemory(&AsyncCallbackData->CallbackContext, Buffer, Length);
2072     }
2073
2074     AsyncCallbackData->FdoDevice = FdoDevice;
2075     AsyncCallbackData->CallbackFunction = Callback;
2076
2077     KeInitializeTimer(&AsyncCallbackData->AsyncTimer);
2078
2079     KeInitializeDpc(&AsyncCallbackData->AsyncTimerDpc,
2080                     USBPORT_AsyncTimerDpc,
2081                     AsyncCallbackData);
2082
2083     DueTime.QuadPart -= (KeQueryTimeIncrement() - 1) + 10000 * TimerValue;
2084
2085     KeSetTimer(&AsyncCallbackData->AsyncTimer,
2086                DueTime,
2087                &AsyncCallbackData->AsyncTimerDpc);
2088
2089     return 0;
2090 }
2091
2092 PVOID
2093 NTAPI
2094 USBPORT_GetMappedVirtualAddress(IN PVOID PhysicalAddress,
2095                                 IN PVOID MiniPortExtension,
2096                                 IN PVOID MiniPortEndpoint)
2097 {
2098     PUSBPORT_COMMON_BUFFER_HEADER HeaderBuffer;
2099     PUSBPORT_ENDPOINT Endpoint;
2100     ULONG Offset;
2101     ULONG_PTR VirtualAddress;
2102
2103     DPRINT_CORE("USBPORT_GetMappedVirtualAddress ... \n");
2104
2105     Endpoint = (PUSBPORT_ENDPOINT)((ULONG_PTR)MiniPortEndpoint -
2106                                    sizeof(USBPORT_ENDPOINT));
2107
2108     if (!Endpoint)
2109     {
2110         ASSERT(FALSE);
2111     }
2112
2113     HeaderBuffer = Endpoint->HeaderBuffer;
2114
2115     Offset = (ULONG_PTR)PhysicalAddress - HeaderBuffer->PhysicalAddress;
2116     VirtualAddress = HeaderBuffer->VirtualAddress + Offset;
2117
2118     return (PVOID)VirtualAddress;
2119 }
2120
2121 ULONG
2122 NTAPI
2123 USBPORT_InvalidateEndpoint(IN PVOID MiniPortExtension,
2124                            IN PVOID MiniPortEndpoint)
2125 {
2126     PUSBPORT_DEVICE_EXTENSION FdoExtension;
2127     PDEVICE_OBJECT FdoDevice;
2128     PUSBPORT_ENDPOINT Endpoint;
2129
2130     DPRINT_CORE("USBPORT_InvalidateEndpoint: ... \n");
2131
2132     FdoExtension = (PUSBPORT_DEVICE_EXTENSION)((ULONG_PTR)MiniPortExtension -
2133                                                sizeof(USBPORT_DEVICE_EXTENSION));
2134
2135     FdoDevice = FdoExtension->CommonExtension.SelfDevice;
2136
2137     if (!MiniPortEndpoint)
2138     {
2139         USBPORT_InvalidateEndpointHandler(FdoDevice,
2140                                           NULL,
2141                                           INVALIDATE_ENDPOINT_ONLY);
2142         return 0;
2143     }
2144
2145     Endpoint = (PUSBPORT_ENDPOINT)((ULONG_PTR)MiniPortEndpoint -
2146                                    sizeof(USBPORT_ENDPOINT));
2147
2148     USBPORT_InvalidateEndpointHandler(FdoDevice,
2149                                       Endpoint,
2150                                       INVALIDATE_ENDPOINT_ONLY);
2151
2152     return 0;
2153 }
2154
2155 VOID
2156 NTAPI
2157 USBPORT_CompleteTransfer(IN PURB Urb,
2158                          IN USBD_STATUS TransferStatus)
2159 {
2160     struct _URB_CONTROL_TRANSFER *UrbTransfer;
2161     PUSBPORT_TRANSFER Transfer;
2162     NTSTATUS Status;
2163     PIRP Irp;
2164     KIRQL OldIrql;
2165     PRKEVENT Event;
2166     BOOLEAN WriteToDevice;
2167     BOOLEAN IsFlushSuccess;
2168     PMDL Mdl;
2169     ULONG_PTR CurrentVa;
2170     SIZE_T TransferLength;
2171     PUSBPORT_ENDPOINT Endpoint;
2172     PDEVICE_OBJECT FdoDevice;
2173     PUSBPORT_DEVICE_EXTENSION FdoExtension;
2174     PDMA_OPERATIONS DmaOperations;
2175
2176     DPRINT("USBPORT_CompleteTransfer: Urb - %p, TransferStatus - %X\n",
2177            Urb,
2178            TransferStatus);
2179
2180     UrbTransfer = &Urb->UrbControlTransfer;
2181     Transfer = UrbTransfer->hca.Reserved8[0];
2182
2183     Transfer->USBDStatus = TransferStatus;
2184     Status = USBPORT_USBDStatusToNtStatus(Urb, TransferStatus);
2185
2186     UrbTransfer->TransferBufferLength = Transfer->CompletedTransferLen;
2187
2188     if (Transfer->Flags & TRANSFER_FLAG_DMA_MAPPED)
2189     {
2190         Endpoint = Transfer->Endpoint;
2191         FdoDevice = Endpoint->FdoDevice;
2192         FdoExtension = FdoDevice->DeviceExtension;
2193         DmaOperations = FdoExtension->DmaAdapter->DmaOperations;
2194
2195         WriteToDevice = Transfer->Direction == USBPORT_DMA_DIRECTION_TO_DEVICE;
2196         Mdl = UrbTransfer->TransferBufferMDL;
2197         CurrentVa = (ULONG_PTR)MmGetMdlVirtualAddress(Mdl);
2198         TransferLength = UrbTransfer->TransferBufferLength;
2199
2200         IsFlushSuccess = DmaOperations->FlushAdapterBuffers(FdoExtension->DmaAdapter,
2201                                                             Mdl,
2202                                                             Transfer->MapRegisterBase,
2203                                                             (PVOID)CurrentVa,
2204                                                             TransferLength,
2205                                                             WriteToDevice);
2206
2207         if (!IsFlushSuccess)
2208         {
2209             DPRINT("USBPORT_CompleteTransfer: no FlushAdapterBuffers !!!\n");
2210             ASSERT(FALSE);
2211         }
2212
2213         KeRaiseIrql(DISPATCH_LEVEL, &OldIrql);
2214
2215         DmaOperations->FreeMapRegisters(FdoExtension->DmaAdapter,
2216                                         Transfer->MapRegisterBase,
2217                                         Transfer->NumberOfMapRegisters);
2218
2219         KeLowerIrql(OldIrql);
2220     }
2221
2222     if (Urb->UrbHeader.UsbdFlags & USBD_FLAG_ALLOCATED_MDL)
2223     {
2224         IoFreeMdl(Transfer->TransferBufferMDL);
2225         Urb->UrbHeader.UsbdFlags |= ~USBD_FLAG_ALLOCATED_MDL;
2226     }
2227
2228     Urb->UrbControlTransfer.hca.Reserved8[0] = NULL;
2229     Urb->UrbHeader.UsbdFlags |= ~USBD_FLAG_ALLOCATED_TRANSFER;
2230
2231     Irp = Transfer->Irp;
2232
2233     if (Irp)
2234     {
2235         if (!NT_SUCCESS(Status))
2236         {
2237             //DbgBreakPoint();
2238             DPRINT1("USBPORT_CompleteTransfer: Irp - %p complete with Status - %lx\n",
2239                     Irp,
2240                     Status);
2241
2242             USBPORT_DumpingURB(Urb);
2243         }
2244
2245         Irp->IoStatus.Status = Status;
2246         Irp->IoStatus.Information = 0;
2247
2248         KeRaiseIrql(DISPATCH_LEVEL, &OldIrql);
2249         IoCompleteRequest(Irp, IO_NO_INCREMENT);
2250         KeLowerIrql(OldIrql);
2251     }
2252
2253     Event = Transfer->Event;
2254
2255     if (Event)
2256     {
2257         KeSetEvent(Event, EVENT_INCREMENT, FALSE);
2258     }
2259
2260     ExFreePoolWithTag(Transfer, USB_PORT_TAG);
2261
2262     DPRINT_CORE("USBPORT_CompleteTransfer: exit\n");
2263 }
2264
2265 IO_ALLOCATION_ACTION
2266 NTAPI
2267 USBPORT_MapTransfer(IN PDEVICE_OBJECT FdoDevice,
2268                     IN PIRP Irp,
2269                     IN PVOID MapRegisterBase,
2270                     IN PVOID Context)
2271 {
2272     PUSBPORT_DEVICE_EXTENSION FdoExtension;
2273     PDMA_ADAPTER DmaAdapter;
2274     PUSBPORT_TRANSFER Transfer;
2275     PURB Urb;
2276     PUSBPORT_ENDPOINT Endpoint;
2277     PMDL Mdl;
2278     ULONG_PTR CurrentVa;
2279     PUSBPORT_SCATTER_GATHER_LIST sgList;
2280     SIZE_T CurrentLength;
2281     ULONG ix;
2282     BOOLEAN WriteToDevice;
2283     PHYSICAL_ADDRESS PhAddr = {{0, 0}};
2284     PHYSICAL_ADDRESS PhAddress = {{0, 0}};
2285     SIZE_T TransferLength;
2286     SIZE_T SgCurrentLength;
2287     SIZE_T ElementLength;
2288     PUSBPORT_DEVICE_HANDLE DeviceHandle;
2289     PDMA_OPERATIONS DmaOperations;
2290     USBD_STATUS USBDStatus;
2291     LIST_ENTRY List;
2292     PUSBPORT_TRANSFER transfer;
2293
2294     DPRINT_CORE("USBPORT_MapTransfer: ... \n");
2295
2296     FdoExtension = FdoDevice->DeviceExtension;
2297     DmaAdapter = FdoExtension->DmaAdapter;
2298     DmaOperations = DmaAdapter->DmaOperations;
2299
2300     Transfer = Context;
2301
2302     Urb = Transfer->Urb;
2303     Endpoint = Transfer->Endpoint;
2304     TransferLength = Transfer->TransferParameters.TransferBufferLength;
2305
2306     Mdl = Urb->UrbControlTransfer.TransferBufferMDL;
2307     CurrentVa = (ULONG_PTR)MmGetMdlVirtualAddress(Mdl);
2308
2309     sgList = &Transfer->SgList;
2310
2311     sgList->Flags = 0;
2312     sgList->CurrentVa = CurrentVa;
2313     sgList->MappedSystemVa = MmGetSystemAddressForMdlSafe(Mdl,
2314                                                           NormalPagePriority);
2315     Transfer->MapRegisterBase = MapRegisterBase;
2316
2317     ix = 0;
2318     CurrentLength = 0;
2319
2320     do
2321     {
2322         WriteToDevice = Transfer->Direction == USBPORT_DMA_DIRECTION_TO_DEVICE;
2323         ASSERT(Transfer->Direction != 0);
2324
2325         PhAddress = DmaOperations->MapTransfer(DmaAdapter,
2326                                                Mdl,
2327                                                MapRegisterBase,
2328                                                (PVOID)CurrentVa,
2329                                                &TransferLength,
2330                                                WriteToDevice);
2331
2332         DPRINT_CORE("USBPORT_MapTransfer: PhAddress.LowPart - %p, PhAddress.HighPart - %x, TransferLength - %x\n",
2333                PhAddress.LowPart,
2334                PhAddress.HighPart,
2335                TransferLength);
2336
2337         PhAddress.HighPart = 0;
2338         SgCurrentLength = TransferLength;
2339
2340         do
2341         {
2342             ElementLength = PAGE_SIZE - (PhAddress.LowPart & (PAGE_SIZE - 1));
2343
2344             if (ElementLength > SgCurrentLength)
2345                 ElementLength = SgCurrentLength;
2346
2347             DPRINT_CORE("USBPORT_MapTransfer: PhAddress.LowPart - %p, HighPart - %x, ElementLength - %x\n",
2348                    PhAddress.LowPart,
2349                    PhAddress.HighPart,
2350                    ElementLength);
2351
2352             sgList->SgElement[ix].SgPhysicalAddress = PhAddress;
2353             sgList->SgElement[ix].SgTransferLength = ElementLength;
2354             sgList->SgElement[ix].SgOffset = CurrentLength +
2355                                              (TransferLength - SgCurrentLength);
2356
2357             PhAddress.LowPart += ElementLength;
2358             SgCurrentLength -= ElementLength;
2359
2360             ++ix;
2361         }
2362         while (SgCurrentLength);
2363
2364         if (PhAddr.QuadPart == PhAddress.QuadPart)
2365         {
2366             DPRINT1("USBPORT_MapTransfer: PhAddr == PhAddress\n");
2367             ASSERT(FALSE);
2368         }
2369
2370         PhAddr = PhAddress;
2371
2372         CurrentLength += TransferLength;
2373         CurrentVa += TransferLength;
2374
2375         TransferLength = Transfer->TransferParameters.TransferBufferLength -
2376                          CurrentLength;
2377     }
2378     while (CurrentLength != Transfer->TransferParameters.TransferBufferLength);
2379
2380     sgList->SgElementCount = ix;
2381
2382     if (Endpoint->EndpointProperties.DeviceSpeed == UsbHighSpeed)
2383     {
2384         Transfer->Flags |= TRANSFER_FLAG_HIGH_SPEED;
2385     }
2386
2387     Transfer->Flags |= TRANSFER_FLAG_DMA_MAPPED;
2388
2389     if ((Transfer->Flags & TRANSFER_FLAG_ISO) == 0)
2390     {
2391         KeAcquireSpinLock(&Endpoint->EndpointSpinLock,
2392                           &Endpoint->EndpointOldIrql);
2393
2394         USBPORT_SplitTransfer(FdoDevice, Endpoint, Transfer, &List);
2395
2396         while (!IsListEmpty(&List))
2397         {
2398             transfer = CONTAINING_RECORD(List.Flink,
2399                                          USBPORT_TRANSFER,
2400                                          TransferLink);
2401
2402             RemoveHeadList(&List);
2403             InsertTailList(&Endpoint->TransferList, &transfer->TransferLink);
2404         }
2405
2406         KeReleaseSpinLock(&Endpoint->EndpointSpinLock,
2407                           Endpoint->EndpointOldIrql);
2408     }
2409     else
2410     {
2411         USBDStatus = USBPORT_InitializeIsoTransfer(FdoDevice,
2412                                                    &Urb->UrbIsochronousTransfer,
2413                                                    Transfer);
2414
2415         if (USBDStatus != USBD_STATUS_SUCCESS)
2416         {
2417             KeAcquireSpinLock(&Endpoint->EndpointSpinLock,
2418                               &Endpoint->EndpointOldIrql);
2419
2420             USBPORT_QueueDoneTransfer(Transfer, USBDStatus);
2421
2422             KeReleaseSpinLock(&Endpoint->EndpointSpinLock,
2423                               Endpoint->EndpointOldIrql);
2424         }
2425     }
2426
2427     DeviceHandle = Urb->UrbHeader.UsbdDeviceHandle;
2428     InterlockedDecrement(&DeviceHandle->DeviceHandleLock);
2429
2430     if (USBPORT_EndpointWorker(Endpoint, 0))
2431     {
2432         USBPORT_InvalidateEndpointHandler(FdoDevice,
2433                                           Endpoint,
2434                                           INVALIDATE_ENDPOINT_WORKER_THREAD);
2435     }
2436
2437     return DeallocateObjectKeepRegisters;
2438 }
2439
2440 VOID
2441 NTAPI
2442 USBPORT_FlushMapTransfers(IN PDEVICE_OBJECT FdoDevice)
2443 {
2444     PUSBPORT_DEVICE_EXTENSION FdoExtension;
2445     PLIST_ENTRY MapTransferList;
2446     PUSBPORT_TRANSFER Transfer;
2447     ULONG NumMapRegisters;
2448     PMDL Mdl;
2449     SIZE_T TransferBufferLength;
2450     ULONG_PTR VirtualAddr;
2451     KIRQL OldIrql;
2452     NTSTATUS Status;
2453     PDMA_OPERATIONS DmaOperations;
2454
2455     DPRINT_CORE("USBPORT_FlushMapTransfers: ... \n");
2456
2457     FdoExtension = FdoDevice->DeviceExtension;
2458     DmaOperations = FdoExtension->DmaAdapter->DmaOperations;
2459
2460     KeRaiseIrql(DISPATCH_LEVEL, &OldIrql);
2461
2462     while (TRUE)
2463     {
2464         MapTransferList = &FdoExtension->MapTransferList;
2465
2466         if (IsListEmpty(&FdoExtension->MapTransferList))
2467         {
2468             KeLowerIrql(OldIrql);
2469             return;
2470         }
2471
2472         Transfer = CONTAINING_RECORD(MapTransferList->Flink,
2473                                      USBPORT_TRANSFER,
2474                                      TransferLink);
2475
2476         RemoveHeadList(MapTransferList);
2477
2478         Mdl = Transfer->Urb->UrbControlTransfer.TransferBufferMDL;
2479         TransferBufferLength = Transfer->TransferParameters.TransferBufferLength;
2480         VirtualAddr = (ULONG_PTR)MmGetMdlVirtualAddress(Mdl);
2481
2482         NumMapRegisters = ADDRESS_AND_SIZE_TO_SPAN_PAGES(VirtualAddr,
2483                                                          TransferBufferLength);
2484
2485         Transfer->NumberOfMapRegisters = NumMapRegisters;
2486
2487         Status = DmaOperations->AllocateAdapterChannel(FdoExtension->DmaAdapter,
2488                                                        FdoDevice,
2489                                                        NumMapRegisters,
2490                                                        USBPORT_MapTransfer,
2491                                                        Transfer);
2492
2493         if (!NT_SUCCESS(Status))
2494             ASSERT(FALSE);
2495     }
2496
2497     KeLowerIrql(OldIrql);
2498 }
2499
2500 USBD_STATUS
2501 NTAPI
2502 USBPORT_AllocateTransfer(IN PDEVICE_OBJECT FdoDevice,
2503                          IN PURB Urb,
2504                          IN PUSBPORT_DEVICE_HANDLE DeviceHandle,
2505                          IN PIRP Irp,
2506                          IN PRKEVENT Event)
2507 {
2508     PUSBPORT_DEVICE_EXTENSION FdoExtension;
2509     SIZE_T TransferLength;
2510     PMDL Mdl;
2511     ULONG_PTR VirtualAddr;
2512     ULONG PagesNeed = 0;
2513     SIZE_T PortTransferLength;
2514     SIZE_T FullTransferLength;
2515     PUSBPORT_TRANSFER Transfer;
2516     PUSBPORT_PIPE_HANDLE PipeHandle;
2517     USBD_STATUS USBDStatus;
2518     SIZE_T IsoBlockLen = 0;
2519
2520     DPRINT_CORE("USBPORT_AllocateTransfer: FdoDevice - %p, Urb - %p, DeviceHandle - %p, Irp - %p, Event - %p\n",
2521            FdoDevice,
2522            Urb,
2523            DeviceHandle,
2524            Irp,
2525            Event);
2526
2527     FdoExtension = FdoDevice->DeviceExtension;
2528
2529     TransferLength = Urb->UrbControlTransfer.TransferBufferLength;
2530     PipeHandle = Urb->UrbControlTransfer.PipeHandle;
2531
2532     if (TransferLength)
2533     {
2534         Mdl = Urb->UrbControlTransfer.TransferBufferMDL;
2535         VirtualAddr = (ULONG_PTR)MmGetMdlVirtualAddress(Mdl);
2536
2537         PagesNeed = ADDRESS_AND_SIZE_TO_SPAN_PAGES(VirtualAddr,
2538                                                    TransferLength);
2539         if (PagesNeed > 0)
2540         {
2541             PagesNeed--;
2542         }
2543     }
2544
2545     if (Urb->UrbHeader.Function == URB_FUNCTION_ISOCH_TRANSFER)
2546     {
2547         DPRINT1("USBPORT_AllocateTransfer: ISOCH_TRANSFER UNIMPLEMENTED. FIXME\n");
2548
2549         //IsoBlockLen = sizeof(USBPORT_ISO_BLOCK) +
2550         //              Urb->UrbIsochronousTransfer.NumberOfPackets *
2551         //              sizeof(USBPORT_ISO_BLOCK_PACKET);
2552     }
2553
2554     PortTransferLength = sizeof(USBPORT_TRANSFER) +
2555                          PagesNeed * sizeof(USBPORT_SCATTER_GATHER_ELEMENT) +
2556                          IsoBlockLen;
2557
2558     FullTransferLength = PortTransferLength +
2559                          FdoExtension->MiniPortInterface->Packet.MiniPortTransferSize;
2560
2561     Transfer = ExAllocatePoolWithTag(NonPagedPool,
2562                                      FullTransferLength,
2563                                      USB_PORT_TAG);
2564
2565     if (!Transfer)
2566     {
2567         DPRINT1("USBPORT_AllocateTransfer: Transfer not allocated!\n");
2568         return USBD_STATUS_INSUFFICIENT_RESOURCES;
2569     }
2570
2571     RtlZeroMemory(Transfer, FullTransferLength);
2572
2573     Transfer->Irp = Irp;
2574     Transfer->Urb = Urb;
2575     Transfer->Endpoint = PipeHandle->Endpoint;
2576     Transfer->Event = Event;
2577     Transfer->PortTransferLength = PortTransferLength;
2578     Transfer->FullTransferLength = FullTransferLength;
2579     Transfer->IsoBlockPtr = NULL;
2580     Transfer->Period = 0;
2581     Transfer->ParentTransfer = Transfer;
2582
2583     if (IsoBlockLen)
2584     {
2585         Transfer->IsoBlockPtr = (PVOID)((ULONG_PTR)Transfer +
2586                                  PortTransferLength - IsoBlockLen);
2587
2588         Transfer->Period = PipeHandle->Endpoint->EndpointProperties.Period;
2589         Transfer->Flags |= TRANSFER_FLAG_ISO;
2590     }
2591
2592     Transfer->MiniportTransfer = (PVOID)((ULONG_PTR)Transfer +
2593                                                     PortTransferLength);
2594
2595     KeInitializeSpinLock(&Transfer->TransferSpinLock);
2596
2597     Urb->UrbControlTransfer.hca.Reserved8[0] = Transfer;
2598     Urb->UrbHeader.UsbdFlags |= USBD_FLAG_ALLOCATED_TRANSFER;
2599
2600     USBDStatus = USBD_STATUS_SUCCESS;
2601
2602     DPRINT_CORE("USBPORT_AllocateTransfer: return USBDStatus - %x\n",
2603                 USBDStatus);
2604
2605     return USBDStatus;
2606 }
2607
2608 NTSTATUS
2609 NTAPI
2610 USBPORT_Dispatch(IN PDEVICE_OBJECT DeviceObject,
2611                  IN PIRP Irp)
2612 {
2613     PUSBPORT_COMMON_DEVICE_EXTENSION DeviceExtension;
2614     PIO_STACK_LOCATION IoStack;
2615     NTSTATUS Status = STATUS_SUCCESS;
2616
2617     DeviceExtension = DeviceObject->DeviceExtension;
2618     IoStack = IoGetCurrentIrpStackLocation(Irp);
2619
2620     if (DeviceExtension->PnpStateFlags & USBPORT_PNP_STATE_FAILED)
2621     {
2622         DPRINT1("USBPORT_Dispatch: USBPORT_PNP_STATE_FAILED\n");
2623         DbgBreakPoint();
2624     }
2625
2626     switch (IoStack->MajorFunction)
2627     {
2628         case IRP_MJ_DEVICE_CONTROL:
2629             if (DeviceExtension->IsPDO)
2630             {
2631                 DPRINT("USBPORT_Dispatch: PDO IRP_MJ_DEVICE_CONTROL. Major - %d, Minor - %d\n",
2632                        IoStack->MajorFunction,
2633                        IoStack->MinorFunction);
2634
2635                 Status = USBPORT_PdoDeviceControl(DeviceObject, Irp);
2636             }
2637             else
2638             {
2639                 DPRINT("USBPORT_Dispatch: FDO IRP_MJ_DEVICE_CONTROL. Major - %d, Minor - %d\n",
2640                        IoStack->MajorFunction,
2641                        IoStack->MinorFunction);
2642
2643                 Status = USBPORT_FdoDeviceControl(DeviceObject, Irp);
2644             }
2645
2646             break;
2647
2648         case IRP_MJ_INTERNAL_DEVICE_CONTROL:
2649             if (DeviceExtension->IsPDO)
2650             {
2651                 DPRINT("USBPORT_Dispatch: PDO IRP_MJ_INTERNAL_DEVICE_CONTROL. Major - %d, Minor - %d\n",
2652                        IoStack->MajorFunction,
2653                        IoStack->MinorFunction);
2654
2655                 Status = USBPORT_PdoInternalDeviceControl(DeviceObject, Irp);
2656             }
2657             else
2658             {
2659                 DPRINT("USBPORT_Dispatch: FDO IRP_MJ_INTERNAL_DEVICE_CONTROL. Major - %d, Minor - %d\n",
2660                        IoStack->MajorFunction,
2661                        IoStack->MinorFunction);
2662
2663                 Status = USBPORT_FdoInternalDeviceControl(DeviceObject, Irp);
2664             }
2665
2666             break;
2667
2668         case IRP_MJ_POWER:
2669             if (DeviceExtension->IsPDO)
2670             {
2671                 DPRINT("USBPORT_Dispatch: PDO IRP_MJ_POWER. Major - %d, Minor - %d\n",
2672                        IoStack->MajorFunction,
2673                        IoStack->MinorFunction);
2674
2675                 Status = USBPORT_PdoPower(DeviceObject, Irp);
2676             }
2677             else
2678             {
2679                 DPRINT("USBPORT_Dispatch: FDO IRP_MJ_POWER. Major - %d, Minor - %d\n",
2680                        IoStack->MajorFunction,
2681                        IoStack->MinorFunction);
2682
2683                 Status = USBPORT_FdoPower(DeviceObject, Irp);
2684             }
2685
2686             break;
2687
2688         case IRP_MJ_SYSTEM_CONTROL:
2689             if (DeviceExtension->IsPDO)
2690             {
2691                 DPRINT("USBPORT_Dispatch: PDO IRP_MJ_SYSTEM_CONTROL. Major - %d, Minor - %d\n",
2692                        IoStack->MajorFunction,
2693                        IoStack->MinorFunction);
2694
2695                 Irp->IoStatus.Status = Status;
2696                 IoCompleteRequest(Irp, IO_NO_INCREMENT);
2697             }
2698             else
2699             {
2700                 DPRINT("USBPORT_Dispatch: FDO IRP_MJ_SYSTEM_CONTROL. Major - %d, Minor - %d\n",
2701                        IoStack->MajorFunction,
2702                        IoStack->MinorFunction);
2703
2704                 IoSkipCurrentIrpStackLocation(Irp);
2705                 Status = IoCallDriver(DeviceExtension->LowerDevice, Irp);
2706             }
2707
2708             break;
2709
2710         case IRP_MJ_PNP:
2711             if (DeviceExtension->IsPDO)
2712             {
2713                 DPRINT("USBPORT_Dispatch: PDO IRP_MJ_PNP. Major - %d, Minor - %d\n",
2714                        IoStack->MajorFunction,
2715                        IoStack->MinorFunction);
2716
2717                 Status = USBPORT_PdoPnP(DeviceObject, Irp);
2718             }
2719             else
2720             {
2721                 DPRINT("USBPORT_Dispatch: FDO IRP_MJ_PNP. Major - %d, Minor - %d\n",
2722                        IoStack->MajorFunction,
2723                        IoStack->MinorFunction);
2724
2725                 Status = USBPORT_FdoPnP(DeviceObject, Irp);
2726             }
2727
2728             break;
2729
2730         case IRP_MJ_CREATE:
2731         case IRP_MJ_CLOSE:
2732             DPRINT("USBPORT_Dispatch: IRP_MJ_CREATE | IRP_MJ_CLOSE\n");
2733             Irp->IoStatus.Status = Status;
2734             IoCompleteRequest(Irp, IO_NO_INCREMENT);
2735             break;
2736
2737         default:
2738             if (DeviceExtension->IsPDO)
2739             {
2740                 DPRINT("USBPORT_Dispatch: PDO unhandled IRP_MJ_???. Major - %d, Minor - %d\n",
2741                        IoStack->MajorFunction,
2742                        IoStack->MinorFunction);
2743             }
2744             else
2745             {
2746                 DPRINT("USBPORT_Dispatch: FDO unhandled IRP_MJ_???. Major - %d, Minor - %d\n",
2747                        IoStack->MajorFunction,
2748                        IoStack->MinorFunction);
2749             }
2750
2751             Status = STATUS_INVALID_DEVICE_REQUEST;
2752             Irp->IoStatus.Status = Status;
2753             IoCompleteRequest(Irp, IO_NO_INCREMENT);
2754             break;
2755     }
2756
2757     DPRINT("USBPORT_Dispatch: Status - %x\n", Status);
2758     return Status;
2759 }
2760
2761 ULONG
2762 NTAPI
2763 USBPORT_GetHciMn(VOID)
2764 {
2765     return USBPORT_HCI_MN;
2766 }
2767
2768 NTSTATUS
2769 NTAPI
2770 USBPORT_RegisterUSBPortDriver(IN PDRIVER_OBJECT DriverObject,
2771                               IN ULONG Version,
2772                               IN PUSBPORT_REGISTRATION_PACKET RegPacket)
2773 {
2774     PUSBPORT_MINIPORT_INTERFACE MiniPortInterface;
2775
2776     DPRINT("USBPORT_RegisterUSBPortDriver: DriverObject - %p, Version - %p, RegPacket - %p\n",
2777            DriverObject,
2778            Version,
2779            RegPacket);
2780
2781     DPRINT("USBPORT_RegisterUSBPortDriver: sizeof(USBPORT_MINIPORT_INTERFACE) - %x\n",
2782            sizeof(USBPORT_MINIPORT_INTERFACE));
2783
2784     DPRINT("USBPORT_RegisterUSBPortDriver: sizeof(USBPORT_DEVICE_EXTENSION)   - %x\n",
2785            sizeof(USBPORT_DEVICE_EXTENSION));
2786
2787     if (Version < USB10_MINIPORT_INTERFACE_VERSION)
2788     {
2789         return STATUS_UNSUCCESSFUL;
2790     }
2791
2792     if (!USBPORT_Initialized)
2793     {
2794         InitializeListHead(&USBPORT_MiniPortDrivers);
2795         InitializeListHead(&USBPORT_USB1FdoList);
2796         InitializeListHead(&USBPORT_USB2FdoList);
2797
2798         KeInitializeSpinLock(&USBPORT_SpinLock);
2799         USBPORT_Initialized = TRUE;
2800     }
2801
2802     MiniPortInterface = ExAllocatePoolWithTag(NonPagedPool,
2803                                               sizeof(USBPORT_MINIPORT_INTERFACE),
2804                                               USB_PORT_TAG);
2805     if (!MiniPortInterface)
2806     {
2807         return STATUS_INSUFFICIENT_RESOURCES;
2808     }
2809
2810     RtlZeroMemory(MiniPortInterface, sizeof(USBPORT_MINIPORT_INTERFACE));
2811
2812     MiniPortInterface->DriverObject = DriverObject;
2813     MiniPortInterface->DriverUnload = DriverObject->DriverUnload;
2814     MiniPortInterface->Version = Version;
2815
2816     ExInterlockedInsertTailList(&USBPORT_MiniPortDrivers,
2817                                 &MiniPortInterface->DriverLink,
2818                                 &USBPORT_SpinLock);
2819
2820     DriverObject->DriverExtension->AddDevice = USBPORT_AddDevice;
2821     DriverObject->DriverUnload = USBPORT_Unload;
2822
2823     DriverObject->MajorFunction[IRP_MJ_CREATE] = USBPORT_Dispatch;
2824     DriverObject->MajorFunction[IRP_MJ_CLOSE] = USBPORT_Dispatch;
2825     DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = USBPORT_Dispatch;
2826     DriverObject->MajorFunction[IRP_MJ_INTERNAL_DEVICE_CONTROL] = USBPORT_Dispatch;
2827     DriverObject->MajorFunction[IRP_MJ_PNP] = USBPORT_Dispatch;
2828     DriverObject->MajorFunction[IRP_MJ_POWER] = USBPORT_Dispatch;
2829     DriverObject->MajorFunction[IRP_MJ_SYSTEM_CONTROL] = USBPORT_Dispatch;
2830
2831     RegPacket->UsbPortDbgPrint = USBPORT_DbgPrint;
2832     RegPacket->UsbPortTestDebugBreak = USBPORT_TestDebugBreak;
2833     RegPacket->UsbPortAssertFailure = USBPORT_AssertFailure;
2834     RegPacket->UsbPortGetMiniportRegistryKeyValue = USBPORT_GetMiniportRegistryKeyValue;
2835     RegPacket->UsbPortInvalidateRootHub = USBPORT_InvalidateRootHub;
2836     RegPacket->UsbPortInvalidateEndpoint = USBPORT_InvalidateEndpoint;
2837     RegPacket->UsbPortCompleteTransfer = USBPORT_MiniportCompleteTransfer;
2838     RegPacket->UsbPortCompleteIsoTransfer = USBPORT_CompleteIsoTransfer;
2839     RegPacket->UsbPortLogEntry = USBPORT_LogEntry;
2840     RegPacket->UsbPortGetMappedVirtualAddress = USBPORT_GetMappedVirtualAddress;
2841     RegPacket->UsbPortRequestAsyncCallback = USBPORT_RequestAsyncCallback;
2842     RegPacket->UsbPortReadWriteConfigSpace = USBPORT_ReadWriteConfigSpace;
2843     RegPacket->UsbPortWait = USBPORT_Wait;
2844     RegPacket->UsbPortInvalidateController = USBPORT_InvalidateController;
2845     RegPacket->UsbPortBugCheck = USBPORT_BugCheck;
2846     RegPacket->UsbPortNotifyDoubleBuffer = USBPORT_NotifyDoubleBuffer;
2847
2848     RtlCopyMemory(&MiniPortInterface->Packet,
2849                   RegPacket,
2850                   sizeof(USBPORT_REGISTRATION_PACKET));
2851
2852     return STATUS_SUCCESS;
2853 }
2854
2855 NTSTATUS
2856 NTAPI
2857 DriverEntry(IN PDRIVER_OBJECT DriverObject,
2858             IN PUNICODE_STRING RegistryPath)
2859 {
2860     return STATUS_SUCCESS;
2861 }