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_CORE("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_CORE("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 KeClearEvent(&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 sizeof(USB2_HC_EXTENSION); 1850 1851 /* Create device */ 1852 Status = IoCreateDevice(DriverObject, 1853 Length, 1854 &DeviceName, 1855 FILE_DEVICE_CONTROLLER, 1856 0, 1857 FALSE, 1858 &DeviceObject); 1859 1860 /* Check for success */ 1861 if (NT_SUCCESS(Status)) break; 1862 1863 /* Is there a device object with that same name */ 1864 if ((Status == STATUS_OBJECT_NAME_EXISTS) || 1865 (Status == STATUS_OBJECT_NAME_COLLISION)) 1866 { 1867 /* Try the next name */ 1868 DeviceNumber++; 1869 continue; 1870 } 1871 1872 /* Bail out on other errors */ 1873 if (!NT_SUCCESS(Status)) 1874 { 1875 DPRINT1("USBPORT_AddDevice: failed to create %wZ, Status %x\n", 1876 &DeviceName, 1877 Status); 1878 1879 return Status; 1880 } 1881 } 1882 1883 DPRINT("USBPORT_AddDevice: created device %p <%wZ>, Status %x\n", 1884 DeviceObject, 1885 &DeviceName, 1886 Status); 1887 1888 FdoExtension = DeviceObject->DeviceExtension; 1889 FdoCommonExtension = &FdoExtension->CommonExtension; 1890 1891 RtlZeroMemory(FdoExtension, sizeof(USBPORT_DEVICE_EXTENSION)); 1892 1893 FdoCommonExtension->SelfDevice = DeviceObject; 1894 FdoCommonExtension->LowerPdoDevice = PhysicalDeviceObject; 1895 FdoCommonExtension->IsPDO = FALSE; 1896 1897 LowerDevice = IoAttachDeviceToDeviceStack(DeviceObject, 1898 PhysicalDeviceObject); 1899 1900 FdoCommonExtension->LowerDevice = LowerDevice; 1901 1902 FdoCommonExtension->DevicePowerState = PowerDeviceD3; 1903 1904 FdoExtension->MiniPortExt = (PVOID)((ULONG_PTR)FdoExtension + 1905 sizeof(USBPORT_DEVICE_EXTENSION)); 1906 1907 if (MiniPortInterface->Packet.MiniPortFlags & USB_MINIPORT_FLAGS_USB2) 1908 { 1909 FdoExtension->Usb2Extension = 1910 (PUSB2_HC_EXTENSION)((ULONG_PTR)FdoExtension->MiniPortExt + 1911 MiniPortInterface->Packet.MiniPortExtensionSize); 1912 1913 DPRINT("USBPORT_AddDevice: Usb2Extension - %p\n", 1914 FdoExtension->Usb2Extension); 1915 1916 USB2_InitController(FdoExtension->Usb2Extension); 1917 } 1918 else 1919 { 1920 FdoExtension->Usb2Extension = NULL; 1921 } 1922 1923 FdoExtension->MiniPortInterface = MiniPortInterface; 1924 FdoExtension->FdoNameNumber = DeviceNumber; 1925 1926 KeInitializeSemaphore(&FdoExtension->DeviceSemaphore, 1, 1); 1927 KeInitializeSemaphore(&FdoExtension->ControllerSemaphore, 1, 1); 1928 1929 InitializeListHead(&FdoExtension->EndpointList); 1930 InitializeListHead(&FdoExtension->DoneTransferList); 1931 InitializeListHead(&FdoExtension->WorkerList); 1932 InitializeListHead(&FdoExtension->EpStateChangeList); 1933 InitializeListHead(&FdoExtension->MapTransferList); 1934 InitializeListHead(&FdoExtension->DeviceHandleList); 1935 InitializeListHead(&FdoExtension->IdleIrpList); 1936 InitializeListHead(&FdoExtension->BadRequestList); 1937 InitializeListHead(&FdoExtension->EndpointClosedList); 1938 1939 DeviceObject->Flags &= ~DO_DEVICE_INITIALIZING; 1940 1941 return Status; 1942 } 1943 1944 VOID 1945 NTAPI 1946 USBPORT_Unload(IN PDRIVER_OBJECT DriverObject) 1947 { 1948 PUSBPORT_MINIPORT_INTERFACE MiniPortInterface; 1949 1950 DPRINT1("USBPORT_Unload: FIXME!\n"); 1951 1952 MiniPortInterface = USBPORT_FindMiniPort(DriverObject); 1953 1954 if (!MiniPortInterface) 1955 { 1956 DPRINT("USBPORT_Unload: CRITICAL ERROR!!! Not found MiniPortInterface\n"); 1957 KeBugCheckEx(BUGCODE_USB_DRIVER, 1, 0, 0, 0); 1958 } 1959 1960 DPRINT1("USBPORT_Unload: UNIMPLEMENTED. FIXME. \n"); 1961 //MiniPortInterface->DriverUnload(DriverObject); // Call MiniPort _HCI_Unload 1962 } 1963 1964 VOID 1965 NTAPI 1966 USBPORT_MiniportCompleteTransfer(IN PVOID MiniPortExtension, 1967 IN PVOID MiniPortEndpoint, 1968 IN PVOID TransferParameters, 1969 IN USBD_STATUS USBDStatus, 1970 IN ULONG TransferLength) 1971 { 1972 PUSBPORT_TRANSFER Transfer; 1973 PUSBPORT_TRANSFER ParentTransfer; 1974 PUSBPORT_TRANSFER SplitTransfer; 1975 PLIST_ENTRY SplitHead; 1976 PLIST_ENTRY Entry; 1977 KIRQL OldIrql; 1978 1979 DPRINT_CORE("USBPORT_MiniportCompleteTransfer: USBDStatus - %x, TransferLength - %x\n", 1980 USBDStatus, 1981 TransferLength); 1982 1983 Transfer = CONTAINING_RECORD(TransferParameters, 1984 USBPORT_TRANSFER, 1985 TransferParameters); 1986 1987 Transfer->Flags |= TRANSFER_FLAG_COMPLETED; 1988 Transfer->CompletedTransferLen = TransferLength; 1989 1990 if (((Transfer->Flags & TRANSFER_FLAG_SPLITED) == 0) || 1991 TransferLength >= Transfer->TransferParameters.TransferBufferLength) 1992 { 1993 goto Exit; 1994 } 1995 1996 ParentTransfer = Transfer->ParentTransfer; 1997 1998 KeAcquireSpinLock(&ParentTransfer->TransferSpinLock, &OldIrql); 1999 2000 if (IsListEmpty(&ParentTransfer->SplitTransfersList)) 2001 { 2002 goto Exit; 2003 } 2004 2005 SplitHead = &ParentTransfer->SplitTransfersList; 2006 Entry = SplitHead->Flink; 2007 2008 while (Entry && !IsListEmpty(SplitHead)) 2009 { 2010 SplitTransfer = CONTAINING_RECORD(Entry, 2011 USBPORT_TRANSFER, 2012 SplitLink); 2013 2014 if (!(SplitTransfer->Flags & TRANSFER_FLAG_SUBMITED)) 2015 { 2016 DPRINT1("USBPORT_MiniportCompleteTransfer: SplitTransfer->Flags - %X\n", 2017 SplitTransfer->Flags); 2018 //Add TRANSFER_FLAG_xxx 2019 } 2020 2021 Entry = Entry->Flink; 2022 } 2023 2024 KeReleaseSpinLock(&ParentTransfer->TransferSpinLock, OldIrql); 2025 2026 Exit: 2027 USBPORT_QueueDoneTransfer(Transfer, USBDStatus); 2028 } 2029 2030 VOID 2031 NTAPI 2032 USBPORT_AsyncTimerDpc(IN PRKDPC Dpc, 2033 IN PVOID DeferredContext, 2034 IN PVOID SystemArgument1, 2035 IN PVOID SystemArgument2) 2036 { 2037 PDEVICE_OBJECT FdoDevice; 2038 PUSBPORT_DEVICE_EXTENSION FdoExtension; 2039 PUSBPORT_ASYNC_CALLBACK_DATA AsyncCallbackData; 2040 2041 DPRINT("USBPORT_AsyncTimerDpc: ... \n"); 2042 2043 AsyncCallbackData = DeferredContext; 2044 FdoDevice = AsyncCallbackData->FdoDevice; 2045 FdoExtension = FdoDevice->DeviceExtension; 2046 2047 (*AsyncCallbackData->CallbackFunction)(FdoExtension->MiniPortExt, 2048 &AsyncCallbackData->CallbackContext); 2049 2050 ExFreePoolWithTag(AsyncCallbackData, USB_PORT_TAG); 2051 } 2052 2053 ULONG 2054 NTAPI 2055 USBPORT_RequestAsyncCallback(IN PVOID MiniPortExtension, 2056 IN ULONG TimerValue, 2057 IN PVOID Buffer, 2058 IN SIZE_T Length, 2059 IN ASYNC_TIMER_CALLBACK * Callback) 2060 { 2061 PUSBPORT_DEVICE_EXTENSION FdoExtension; 2062 PDEVICE_OBJECT FdoDevice; 2063 PUSBPORT_ASYNC_CALLBACK_DATA AsyncCallbackData; 2064 LARGE_INTEGER DueTime = {{0, 0}}; 2065 2066 DPRINT("USBPORT_RequestAsyncCallback: ... \n"); 2067 2068 FdoExtension = (PUSBPORT_DEVICE_EXTENSION)((ULONG_PTR)MiniPortExtension - 2069 sizeof(USBPORT_DEVICE_EXTENSION)); 2070 2071 FdoDevice = FdoExtension->CommonExtension.SelfDevice; 2072 2073 AsyncCallbackData = ExAllocatePoolWithTag(NonPagedPool, 2074 sizeof(USBPORT_ASYNC_CALLBACK_DATA) + Length, 2075 USB_PORT_TAG); 2076 2077 if (!AsyncCallbackData) 2078 { 2079 DPRINT1("USBPORT_RequestAsyncCallback: Not allocated AsyncCallbackData!\n"); 2080 return 0; 2081 } 2082 2083 RtlZeroMemory(AsyncCallbackData, 2084 sizeof(USBPORT_ASYNC_CALLBACK_DATA) + Length); 2085 2086 if (Length) 2087 { 2088 RtlCopyMemory(&AsyncCallbackData->CallbackContext, Buffer, Length); 2089 } 2090 2091 AsyncCallbackData->FdoDevice = FdoDevice; 2092 AsyncCallbackData->CallbackFunction = Callback; 2093 2094 KeInitializeTimer(&AsyncCallbackData->AsyncTimer); 2095 2096 KeInitializeDpc(&AsyncCallbackData->AsyncTimerDpc, 2097 USBPORT_AsyncTimerDpc, 2098 AsyncCallbackData); 2099 2100 DueTime.QuadPart -= (KeQueryTimeIncrement() - 1) + 10000 * TimerValue; 2101 2102 KeSetTimer(&AsyncCallbackData->AsyncTimer, 2103 DueTime, 2104 &AsyncCallbackData->AsyncTimerDpc); 2105 2106 return 0; 2107 } 2108 2109 PVOID 2110 NTAPI 2111 USBPORT_GetMappedVirtualAddress(IN PVOID PhysicalAddress, 2112 IN PVOID MiniPortExtension, 2113 IN PVOID MiniPortEndpoint) 2114 { 2115 PUSBPORT_COMMON_BUFFER_HEADER HeaderBuffer; 2116 PUSBPORT_ENDPOINT Endpoint; 2117 ULONG Offset; 2118 ULONG_PTR VirtualAddress; 2119 2120 DPRINT_CORE("USBPORT_GetMappedVirtualAddress ... \n"); 2121 2122 Endpoint = (PUSBPORT_ENDPOINT)((ULONG_PTR)MiniPortEndpoint - 2123 sizeof(USBPORT_ENDPOINT)); 2124 2125 if (!Endpoint) 2126 { 2127 ASSERT(FALSE); 2128 } 2129 2130 HeaderBuffer = Endpoint->HeaderBuffer; 2131 2132 Offset = (ULONG_PTR)PhysicalAddress - HeaderBuffer->PhysicalAddress; 2133 VirtualAddress = HeaderBuffer->VirtualAddress + Offset; 2134 2135 return (PVOID)VirtualAddress; 2136 } 2137 2138 ULONG 2139 NTAPI 2140 USBPORT_InvalidateEndpoint(IN PVOID MiniPortExtension, 2141 IN PVOID MiniPortEndpoint) 2142 { 2143 PUSBPORT_DEVICE_EXTENSION FdoExtension; 2144 PDEVICE_OBJECT FdoDevice; 2145 PUSBPORT_ENDPOINT Endpoint; 2146 2147 DPRINT_CORE("USBPORT_InvalidateEndpoint: ... \n"); 2148 2149 FdoExtension = (PUSBPORT_DEVICE_EXTENSION)((ULONG_PTR)MiniPortExtension - 2150 sizeof(USBPORT_DEVICE_EXTENSION)); 2151 2152 FdoDevice = FdoExtension->CommonExtension.SelfDevice; 2153 2154 if (!MiniPortEndpoint) 2155 { 2156 USBPORT_InvalidateEndpointHandler(FdoDevice, 2157 NULL, 2158 INVALIDATE_ENDPOINT_ONLY); 2159 return 0; 2160 } 2161 2162 Endpoint = (PUSBPORT_ENDPOINT)((ULONG_PTR)MiniPortEndpoint - 2163 sizeof(USBPORT_ENDPOINT)); 2164 2165 USBPORT_InvalidateEndpointHandler(FdoDevice, 2166 Endpoint, 2167 INVALIDATE_ENDPOINT_ONLY); 2168 2169 return 0; 2170 } 2171 2172 VOID 2173 NTAPI 2174 USBPORT_CompleteTransfer(IN PURB Urb, 2175 IN USBD_STATUS TransferStatus) 2176 { 2177 struct _URB_CONTROL_TRANSFER *UrbTransfer; 2178 PUSBPORT_TRANSFER Transfer; 2179 NTSTATUS Status; 2180 PIRP Irp; 2181 KIRQL OldIrql; 2182 PRKEVENT Event; 2183 BOOLEAN WriteToDevice; 2184 BOOLEAN IsFlushSuccess; 2185 PMDL Mdl; 2186 ULONG_PTR CurrentVa; 2187 SIZE_T TransferLength; 2188 PUSBPORT_ENDPOINT Endpoint; 2189 PDEVICE_OBJECT FdoDevice; 2190 PUSBPORT_DEVICE_EXTENSION FdoExtension; 2191 PDMA_OPERATIONS DmaOperations; 2192 2193 DPRINT("USBPORT_CompleteTransfer: Urb - %p, TransferStatus - %X\n", 2194 Urb, 2195 TransferStatus); 2196 2197 UrbTransfer = &Urb->UrbControlTransfer; 2198 Transfer = UrbTransfer->hca.Reserved8[0]; 2199 2200 Transfer->USBDStatus = TransferStatus; 2201 Status = USBPORT_USBDStatusToNtStatus(Urb, TransferStatus); 2202 2203 UrbTransfer->TransferBufferLength = Transfer->CompletedTransferLen; 2204 2205 if (Transfer->Flags & TRANSFER_FLAG_DMA_MAPPED) 2206 { 2207 Endpoint = Transfer->Endpoint; 2208 FdoDevice = Endpoint->FdoDevice; 2209 FdoExtension = FdoDevice->DeviceExtension; 2210 DmaOperations = FdoExtension->DmaAdapter->DmaOperations; 2211 2212 WriteToDevice = Transfer->Direction == USBPORT_DMA_DIRECTION_TO_DEVICE; 2213 Mdl = UrbTransfer->TransferBufferMDL; 2214 CurrentVa = (ULONG_PTR)MmGetMdlVirtualAddress(Mdl); 2215 TransferLength = UrbTransfer->TransferBufferLength; 2216 2217 IsFlushSuccess = DmaOperations->FlushAdapterBuffers(FdoExtension->DmaAdapter, 2218 Mdl, 2219 Transfer->MapRegisterBase, 2220 (PVOID)CurrentVa, 2221 TransferLength, 2222 WriteToDevice); 2223 2224 if (!IsFlushSuccess) 2225 { 2226 DPRINT("USBPORT_CompleteTransfer: no FlushAdapterBuffers !!!\n"); 2227 ASSERT(FALSE); 2228 } 2229 2230 KeRaiseIrql(DISPATCH_LEVEL, &OldIrql); 2231 2232 DmaOperations->FreeMapRegisters(FdoExtension->DmaAdapter, 2233 Transfer->MapRegisterBase, 2234 Transfer->NumberOfMapRegisters); 2235 2236 KeLowerIrql(OldIrql); 2237 } 2238 2239 if (Urb->UrbHeader.UsbdFlags & USBD_FLAG_ALLOCATED_MDL) 2240 { 2241 IoFreeMdl(Transfer->TransferBufferMDL); 2242 Urb->UrbHeader.UsbdFlags |= ~USBD_FLAG_ALLOCATED_MDL; 2243 } 2244 2245 Urb->UrbControlTransfer.hca.Reserved8[0] = NULL; 2246 Urb->UrbHeader.UsbdFlags |= ~USBD_FLAG_ALLOCATED_TRANSFER; 2247 2248 Irp = Transfer->Irp; 2249 2250 if (Irp) 2251 { 2252 if (!NT_SUCCESS(Status)) 2253 { 2254 //DbgBreakPoint(); 2255 DPRINT1("USBPORT_CompleteTransfer: Irp - %p complete with Status - %lx\n", 2256 Irp, 2257 Status); 2258 2259 USBPORT_DumpingURB(Urb); 2260 } 2261 2262 Irp->IoStatus.Status = Status; 2263 Irp->IoStatus.Information = 0; 2264 2265 KeRaiseIrql(DISPATCH_LEVEL, &OldIrql); 2266 IoCompleteRequest(Irp, IO_NO_INCREMENT); 2267 KeLowerIrql(OldIrql); 2268 } 2269 2270 Event = Transfer->Event; 2271 2272 if (Event) 2273 { 2274 KeSetEvent(Event, EVENT_INCREMENT, FALSE); 2275 } 2276 2277 ExFreePoolWithTag(Transfer, USB_PORT_TAG); 2278 2279 DPRINT_CORE("USBPORT_CompleteTransfer: exit\n"); 2280 } 2281 2282 IO_ALLOCATION_ACTION 2283 NTAPI 2284 USBPORT_MapTransfer(IN PDEVICE_OBJECT FdoDevice, 2285 IN PIRP Irp, 2286 IN PVOID MapRegisterBase, 2287 IN PVOID Context) 2288 { 2289 PUSBPORT_DEVICE_EXTENSION FdoExtension; 2290 PDMA_ADAPTER DmaAdapter; 2291 PUSBPORT_TRANSFER Transfer; 2292 PURB Urb; 2293 PUSBPORT_ENDPOINT Endpoint; 2294 PMDL Mdl; 2295 ULONG_PTR CurrentVa; 2296 PUSBPORT_SCATTER_GATHER_LIST sgList; 2297 SIZE_T CurrentLength; 2298 ULONG ix; 2299 BOOLEAN WriteToDevice; 2300 PHYSICAL_ADDRESS PhAddr = {{0, 0}}; 2301 PHYSICAL_ADDRESS PhAddress = {{0, 0}}; 2302 SIZE_T TransferLength; 2303 SIZE_T SgCurrentLength; 2304 SIZE_T ElementLength; 2305 PUSBPORT_DEVICE_HANDLE DeviceHandle; 2306 PDMA_OPERATIONS DmaOperations; 2307 USBD_STATUS USBDStatus; 2308 LIST_ENTRY List; 2309 PUSBPORT_TRANSFER transfer; 2310 2311 DPRINT_CORE("USBPORT_MapTransfer: ... \n"); 2312 2313 FdoExtension = FdoDevice->DeviceExtension; 2314 DmaAdapter = FdoExtension->DmaAdapter; 2315 DmaOperations = DmaAdapter->DmaOperations; 2316 2317 Transfer = Context; 2318 2319 Urb = Transfer->Urb; 2320 Endpoint = Transfer->Endpoint; 2321 TransferLength = Transfer->TransferParameters.TransferBufferLength; 2322 2323 Mdl = Urb->UrbControlTransfer.TransferBufferMDL; 2324 CurrentVa = (ULONG_PTR)MmGetMdlVirtualAddress(Mdl); 2325 2326 sgList = &Transfer->SgList; 2327 2328 sgList->Flags = 0; 2329 sgList->CurrentVa = CurrentVa; 2330 sgList->MappedSystemVa = MmGetSystemAddressForMdlSafe(Mdl, 2331 NormalPagePriority); 2332 Transfer->MapRegisterBase = MapRegisterBase; 2333 2334 ix = 0; 2335 CurrentLength = 0; 2336 2337 do 2338 { 2339 WriteToDevice = Transfer->Direction == USBPORT_DMA_DIRECTION_TO_DEVICE; 2340 ASSERT(Transfer->Direction != 0); 2341 2342 PhAddress = DmaOperations->MapTransfer(DmaAdapter, 2343 Mdl, 2344 MapRegisterBase, 2345 (PVOID)CurrentVa, 2346 &TransferLength, 2347 WriteToDevice); 2348 2349 DPRINT_CORE("USBPORT_MapTransfer: PhAddress.LowPart - %p, PhAddress.HighPart - %x, TransferLength - %x\n", 2350 PhAddress.LowPart, 2351 PhAddress.HighPart, 2352 TransferLength); 2353 2354 PhAddress.HighPart = 0; 2355 SgCurrentLength = TransferLength; 2356 2357 do 2358 { 2359 ElementLength = PAGE_SIZE - (PhAddress.LowPart & (PAGE_SIZE - 1)); 2360 2361 if (ElementLength > SgCurrentLength) 2362 ElementLength = SgCurrentLength; 2363 2364 DPRINT_CORE("USBPORT_MapTransfer: PhAddress.LowPart - %p, HighPart - %x, ElementLength - %x\n", 2365 PhAddress.LowPart, 2366 PhAddress.HighPart, 2367 ElementLength); 2368 2369 sgList->SgElement[ix].SgPhysicalAddress = PhAddress; 2370 sgList->SgElement[ix].SgTransferLength = ElementLength; 2371 sgList->SgElement[ix].SgOffset = CurrentLength + 2372 (TransferLength - SgCurrentLength); 2373 2374 PhAddress.LowPart += ElementLength; 2375 SgCurrentLength -= ElementLength; 2376 2377 ++ix; 2378 } 2379 while (SgCurrentLength); 2380 2381 if (PhAddr.QuadPart == PhAddress.QuadPart) 2382 { 2383 DPRINT1("USBPORT_MapTransfer: PhAddr == PhAddress\n"); 2384 ASSERT(FALSE); 2385 } 2386 2387 PhAddr = PhAddress; 2388 2389 CurrentLength += TransferLength; 2390 CurrentVa += TransferLength; 2391 2392 TransferLength = Transfer->TransferParameters.TransferBufferLength - 2393 CurrentLength; 2394 } 2395 while (CurrentLength != Transfer->TransferParameters.TransferBufferLength); 2396 2397 sgList->SgElementCount = ix; 2398 2399 if (Endpoint->EndpointProperties.DeviceSpeed == UsbHighSpeed) 2400 { 2401 Transfer->Flags |= TRANSFER_FLAG_HIGH_SPEED; 2402 } 2403 2404 Transfer->Flags |= TRANSFER_FLAG_DMA_MAPPED; 2405 2406 if ((Transfer->Flags & TRANSFER_FLAG_ISO) == 0) 2407 { 2408 KeAcquireSpinLock(&Endpoint->EndpointSpinLock, 2409 &Endpoint->EndpointOldIrql); 2410 2411 USBPORT_SplitTransfer(FdoDevice, Endpoint, Transfer, &List); 2412 2413 while (!IsListEmpty(&List)) 2414 { 2415 transfer = CONTAINING_RECORD(List.Flink, 2416 USBPORT_TRANSFER, 2417 TransferLink); 2418 2419 RemoveHeadList(&List); 2420 InsertTailList(&Endpoint->TransferList, &transfer->TransferLink); 2421 } 2422 2423 KeReleaseSpinLock(&Endpoint->EndpointSpinLock, 2424 Endpoint->EndpointOldIrql); 2425 } 2426 else 2427 { 2428 USBDStatus = USBPORT_InitializeIsoTransfer(FdoDevice, 2429 &Urb->UrbIsochronousTransfer, 2430 Transfer); 2431 2432 if (USBDStatus != USBD_STATUS_SUCCESS) 2433 { 2434 KeAcquireSpinLock(&Endpoint->EndpointSpinLock, 2435 &Endpoint->EndpointOldIrql); 2436 2437 USBPORT_QueueDoneTransfer(Transfer, USBDStatus); 2438 2439 KeReleaseSpinLock(&Endpoint->EndpointSpinLock, 2440 Endpoint->EndpointOldIrql); 2441 } 2442 } 2443 2444 DeviceHandle = Urb->UrbHeader.UsbdDeviceHandle; 2445 InterlockedDecrement(&DeviceHandle->DeviceHandleLock); 2446 2447 if (USBPORT_EndpointWorker(Endpoint, 0)) 2448 { 2449 USBPORT_InvalidateEndpointHandler(FdoDevice, 2450 Endpoint, 2451 INVALIDATE_ENDPOINT_WORKER_THREAD); 2452 } 2453 2454 return DeallocateObjectKeepRegisters; 2455 } 2456 2457 VOID 2458 NTAPI 2459 USBPORT_FlushMapTransfers(IN PDEVICE_OBJECT FdoDevice) 2460 { 2461 PUSBPORT_DEVICE_EXTENSION FdoExtension; 2462 PLIST_ENTRY MapTransferList; 2463 PUSBPORT_TRANSFER Transfer; 2464 ULONG NumMapRegisters; 2465 PMDL Mdl; 2466 SIZE_T TransferBufferLength; 2467 ULONG_PTR VirtualAddr; 2468 KIRQL OldIrql; 2469 NTSTATUS Status; 2470 PDMA_OPERATIONS DmaOperations; 2471 2472 DPRINT_CORE("USBPORT_FlushMapTransfers: ... \n"); 2473 2474 FdoExtension = FdoDevice->DeviceExtension; 2475 DmaOperations = FdoExtension->DmaAdapter->DmaOperations; 2476 2477 KeRaiseIrql(DISPATCH_LEVEL, &OldIrql); 2478 2479 while (TRUE) 2480 { 2481 MapTransferList = &FdoExtension->MapTransferList; 2482 2483 if (IsListEmpty(&FdoExtension->MapTransferList)) 2484 { 2485 KeLowerIrql(OldIrql); 2486 return; 2487 } 2488 2489 Transfer = CONTAINING_RECORD(MapTransferList->Flink, 2490 USBPORT_TRANSFER, 2491 TransferLink); 2492 2493 RemoveHeadList(MapTransferList); 2494 2495 Mdl = Transfer->Urb->UrbControlTransfer.TransferBufferMDL; 2496 TransferBufferLength = Transfer->TransferParameters.TransferBufferLength; 2497 VirtualAddr = (ULONG_PTR)MmGetMdlVirtualAddress(Mdl); 2498 2499 NumMapRegisters = ADDRESS_AND_SIZE_TO_SPAN_PAGES(VirtualAddr, 2500 TransferBufferLength); 2501 2502 Transfer->NumberOfMapRegisters = NumMapRegisters; 2503 2504 Status = DmaOperations->AllocateAdapterChannel(FdoExtension->DmaAdapter, 2505 FdoDevice, 2506 NumMapRegisters, 2507 USBPORT_MapTransfer, 2508 Transfer); 2509 2510 if (!NT_SUCCESS(Status)) 2511 ASSERT(FALSE); 2512 } 2513 2514 KeLowerIrql(OldIrql); 2515 } 2516 2517 USBD_STATUS 2518 NTAPI 2519 USBPORT_AllocateTransfer(IN PDEVICE_OBJECT FdoDevice, 2520 IN PURB Urb, 2521 IN PUSBPORT_DEVICE_HANDLE DeviceHandle, 2522 IN PIRP Irp, 2523 IN PRKEVENT Event) 2524 { 2525 PUSBPORT_DEVICE_EXTENSION FdoExtension; 2526 SIZE_T TransferLength; 2527 PMDL Mdl; 2528 ULONG_PTR VirtualAddr; 2529 ULONG PagesNeed = 0; 2530 SIZE_T PortTransferLength; 2531 SIZE_T FullTransferLength; 2532 PUSBPORT_TRANSFER Transfer; 2533 PUSBPORT_PIPE_HANDLE PipeHandle; 2534 USBD_STATUS USBDStatus; 2535 SIZE_T IsoBlockLen = 0; 2536 2537 DPRINT_CORE("USBPORT_AllocateTransfer: FdoDevice - %p, Urb - %p, DeviceHandle - %p, Irp - %p, Event - %p\n", 2538 FdoDevice, 2539 Urb, 2540 DeviceHandle, 2541 Irp, 2542 Event); 2543 2544 FdoExtension = FdoDevice->DeviceExtension; 2545 2546 TransferLength = Urb->UrbControlTransfer.TransferBufferLength; 2547 PipeHandle = Urb->UrbControlTransfer.PipeHandle; 2548 2549 if (TransferLength) 2550 { 2551 Mdl = Urb->UrbControlTransfer.TransferBufferMDL; 2552 VirtualAddr = (ULONG_PTR)MmGetMdlVirtualAddress(Mdl); 2553 2554 PagesNeed = ADDRESS_AND_SIZE_TO_SPAN_PAGES(VirtualAddr, 2555 TransferLength); 2556 if (PagesNeed > 0) 2557 { 2558 PagesNeed--; 2559 } 2560 } 2561 2562 if (Urb->UrbHeader.Function == URB_FUNCTION_ISOCH_TRANSFER) 2563 { 2564 DPRINT1("USBPORT_AllocateTransfer: ISOCH_TRANSFER UNIMPLEMENTED. FIXME\n"); 2565 2566 //IsoBlockLen = sizeof(USBPORT_ISO_BLOCK) + 2567 // Urb->UrbIsochronousTransfer.NumberOfPackets * 2568 // sizeof(USBPORT_ISO_BLOCK_PACKET); 2569 } 2570 2571 PortTransferLength = sizeof(USBPORT_TRANSFER) + 2572 PagesNeed * sizeof(USBPORT_SCATTER_GATHER_ELEMENT) + 2573 IsoBlockLen; 2574 2575 FullTransferLength = PortTransferLength + 2576 FdoExtension->MiniPortInterface->Packet.MiniPortTransferSize; 2577 2578 Transfer = ExAllocatePoolWithTag(NonPagedPool, 2579 FullTransferLength, 2580 USB_PORT_TAG); 2581 2582 if (!Transfer) 2583 { 2584 DPRINT1("USBPORT_AllocateTransfer: Transfer not allocated!\n"); 2585 return USBD_STATUS_INSUFFICIENT_RESOURCES; 2586 } 2587 2588 RtlZeroMemory(Transfer, FullTransferLength); 2589 2590 Transfer->Irp = Irp; 2591 Transfer->Urb = Urb; 2592 Transfer->Endpoint = PipeHandle->Endpoint; 2593 Transfer->Event = Event; 2594 Transfer->PortTransferLength = PortTransferLength; 2595 Transfer->FullTransferLength = FullTransferLength; 2596 Transfer->IsoBlockPtr = NULL; 2597 Transfer->Period = 0; 2598 Transfer->ParentTransfer = Transfer; 2599 2600 if (IsoBlockLen) 2601 { 2602 Transfer->IsoBlockPtr = (PVOID)((ULONG_PTR)Transfer + 2603 PortTransferLength - IsoBlockLen); 2604 2605 Transfer->Period = PipeHandle->Endpoint->EndpointProperties.Period; 2606 Transfer->Flags |= TRANSFER_FLAG_ISO; 2607 } 2608 2609 Transfer->MiniportTransfer = (PVOID)((ULONG_PTR)Transfer + 2610 PortTransferLength); 2611 2612 KeInitializeSpinLock(&Transfer->TransferSpinLock); 2613 2614 Urb->UrbControlTransfer.hca.Reserved8[0] = Transfer; 2615 Urb->UrbHeader.UsbdFlags |= USBD_FLAG_ALLOCATED_TRANSFER; 2616 2617 USBDStatus = USBD_STATUS_SUCCESS; 2618 2619 DPRINT_CORE("USBPORT_AllocateTransfer: return USBDStatus - %x\n", 2620 USBDStatus); 2621 2622 return USBDStatus; 2623 } 2624 2625 NTSTATUS 2626 NTAPI 2627 USBPORT_Dispatch(IN PDEVICE_OBJECT DeviceObject, 2628 IN PIRP Irp) 2629 { 2630 PUSBPORT_COMMON_DEVICE_EXTENSION DeviceExtension; 2631 PIO_STACK_LOCATION IoStack; 2632 NTSTATUS Status = STATUS_SUCCESS; 2633 2634 DeviceExtension = DeviceObject->DeviceExtension; 2635 IoStack = IoGetCurrentIrpStackLocation(Irp); 2636 2637 if (DeviceExtension->PnpStateFlags & USBPORT_PNP_STATE_FAILED) 2638 { 2639 DPRINT1("USBPORT_Dispatch: USBPORT_PNP_STATE_FAILED\n"); 2640 DbgBreakPoint(); 2641 } 2642 2643 switch (IoStack->MajorFunction) 2644 { 2645 case IRP_MJ_DEVICE_CONTROL: 2646 if (DeviceExtension->IsPDO) 2647 { 2648 DPRINT("USBPORT_Dispatch: PDO IRP_MJ_DEVICE_CONTROL. Major - %d, Minor - %d\n", 2649 IoStack->MajorFunction, 2650 IoStack->MinorFunction); 2651 2652 Status = USBPORT_PdoDeviceControl(DeviceObject, Irp); 2653 } 2654 else 2655 { 2656 DPRINT("USBPORT_Dispatch: FDO IRP_MJ_DEVICE_CONTROL. Major - %d, Minor - %d\n", 2657 IoStack->MajorFunction, 2658 IoStack->MinorFunction); 2659 2660 Status = USBPORT_FdoDeviceControl(DeviceObject, Irp); 2661 } 2662 2663 break; 2664 2665 case IRP_MJ_INTERNAL_DEVICE_CONTROL: 2666 if (DeviceExtension->IsPDO) 2667 { 2668 DPRINT("USBPORT_Dispatch: PDO IRP_MJ_INTERNAL_DEVICE_CONTROL. Major - %d, Minor - %d\n", 2669 IoStack->MajorFunction, 2670 IoStack->MinorFunction); 2671 2672 Status = USBPORT_PdoInternalDeviceControl(DeviceObject, Irp); 2673 } 2674 else 2675 { 2676 DPRINT("USBPORT_Dispatch: FDO IRP_MJ_INTERNAL_DEVICE_CONTROL. Major - %d, Minor - %d\n", 2677 IoStack->MajorFunction, 2678 IoStack->MinorFunction); 2679 2680 Status = USBPORT_FdoInternalDeviceControl(DeviceObject, Irp); 2681 } 2682 2683 break; 2684 2685 case IRP_MJ_POWER: 2686 if (DeviceExtension->IsPDO) 2687 { 2688 DPRINT("USBPORT_Dispatch: PDO IRP_MJ_POWER. Major - %d, Minor - %d\n", 2689 IoStack->MajorFunction, 2690 IoStack->MinorFunction); 2691 2692 Status = USBPORT_PdoPower(DeviceObject, Irp); 2693 } 2694 else 2695 { 2696 DPRINT("USBPORT_Dispatch: FDO IRP_MJ_POWER. Major - %d, Minor - %d\n", 2697 IoStack->MajorFunction, 2698 IoStack->MinorFunction); 2699 2700 Status = USBPORT_FdoPower(DeviceObject, Irp); 2701 } 2702 2703 break; 2704 2705 case IRP_MJ_SYSTEM_CONTROL: 2706 if (DeviceExtension->IsPDO) 2707 { 2708 DPRINT("USBPORT_Dispatch: PDO IRP_MJ_SYSTEM_CONTROL. Major - %d, Minor - %d\n", 2709 IoStack->MajorFunction, 2710 IoStack->MinorFunction); 2711 2712 Irp->IoStatus.Status = Status; 2713 IoCompleteRequest(Irp, IO_NO_INCREMENT); 2714 } 2715 else 2716 { 2717 DPRINT("USBPORT_Dispatch: FDO IRP_MJ_SYSTEM_CONTROL. Major - %d, Minor - %d\n", 2718 IoStack->MajorFunction, 2719 IoStack->MinorFunction); 2720 2721 IoSkipCurrentIrpStackLocation(Irp); 2722 Status = IoCallDriver(DeviceExtension->LowerDevice, Irp); 2723 } 2724 2725 break; 2726 2727 case IRP_MJ_PNP: 2728 if (DeviceExtension->IsPDO) 2729 { 2730 DPRINT("USBPORT_Dispatch: PDO IRP_MJ_PNP. Major - %d, Minor - %d\n", 2731 IoStack->MajorFunction, 2732 IoStack->MinorFunction); 2733 2734 Status = USBPORT_PdoPnP(DeviceObject, Irp); 2735 } 2736 else 2737 { 2738 DPRINT("USBPORT_Dispatch: FDO IRP_MJ_PNP. Major - %d, Minor - %d\n", 2739 IoStack->MajorFunction, 2740 IoStack->MinorFunction); 2741 2742 Status = USBPORT_FdoPnP(DeviceObject, Irp); 2743 } 2744 2745 break; 2746 2747 case IRP_MJ_CREATE: 2748 case IRP_MJ_CLOSE: 2749 DPRINT("USBPORT_Dispatch: IRP_MJ_CREATE | IRP_MJ_CLOSE\n"); 2750 Irp->IoStatus.Status = Status; 2751 IoCompleteRequest(Irp, IO_NO_INCREMENT); 2752 break; 2753 2754 default: 2755 if (DeviceExtension->IsPDO) 2756 { 2757 DPRINT("USBPORT_Dispatch: PDO unhandled IRP_MJ_???. Major - %d, Minor - %d\n", 2758 IoStack->MajorFunction, 2759 IoStack->MinorFunction); 2760 } 2761 else 2762 { 2763 DPRINT("USBPORT_Dispatch: FDO unhandled IRP_MJ_???. Major - %d, Minor - %d\n", 2764 IoStack->MajorFunction, 2765 IoStack->MinorFunction); 2766 } 2767 2768 Status = STATUS_INVALID_DEVICE_REQUEST; 2769 Irp->IoStatus.Status = Status; 2770 IoCompleteRequest(Irp, IO_NO_INCREMENT); 2771 break; 2772 } 2773 2774 DPRINT("USBPORT_Dispatch: Status - %x\n", Status); 2775 return Status; 2776 } 2777 2778 ULONG 2779 NTAPI 2780 USBPORT_GetHciMn(VOID) 2781 { 2782 return USBPORT_HCI_MN; 2783 } 2784 2785 NTSTATUS 2786 NTAPI 2787 USBPORT_RegisterUSBPortDriver(IN PDRIVER_OBJECT DriverObject, 2788 IN ULONG Version, 2789 IN PUSBPORT_REGISTRATION_PACKET RegPacket) 2790 { 2791 PUSBPORT_MINIPORT_INTERFACE MiniPortInterface; 2792 2793 DPRINT("USBPORT_RegisterUSBPortDriver: DriverObject - %p, Version - %p, RegPacket - %p\n", 2794 DriverObject, 2795 Version, 2796 RegPacket); 2797 2798 DPRINT("USBPORT_RegisterUSBPortDriver: sizeof(USBPORT_MINIPORT_INTERFACE) - %x\n", 2799 sizeof(USBPORT_MINIPORT_INTERFACE)); 2800 2801 DPRINT("USBPORT_RegisterUSBPortDriver: sizeof(USBPORT_DEVICE_EXTENSION) - %x\n", 2802 sizeof(USBPORT_DEVICE_EXTENSION)); 2803 2804 if (Version < USB10_MINIPORT_INTERFACE_VERSION) 2805 { 2806 return STATUS_UNSUCCESSFUL; 2807 } 2808 2809 if (!USBPORT_Initialized) 2810 { 2811 InitializeListHead(&USBPORT_MiniPortDrivers); 2812 InitializeListHead(&USBPORT_USB1FdoList); 2813 InitializeListHead(&USBPORT_USB2FdoList); 2814 2815 KeInitializeSpinLock(&USBPORT_SpinLock); 2816 USBPORT_Initialized = TRUE; 2817 } 2818 2819 MiniPortInterface = ExAllocatePoolWithTag(NonPagedPool, 2820 sizeof(USBPORT_MINIPORT_INTERFACE), 2821 USB_PORT_TAG); 2822 if (!MiniPortInterface) 2823 { 2824 return STATUS_INSUFFICIENT_RESOURCES; 2825 } 2826 2827 RtlZeroMemory(MiniPortInterface, sizeof(USBPORT_MINIPORT_INTERFACE)); 2828 2829 MiniPortInterface->DriverObject = DriverObject; 2830 MiniPortInterface->DriverUnload = DriverObject->DriverUnload; 2831 MiniPortInterface->Version = Version; 2832 2833 ExInterlockedInsertTailList(&USBPORT_MiniPortDrivers, 2834 &MiniPortInterface->DriverLink, 2835 &USBPORT_SpinLock); 2836 2837 DriverObject->DriverExtension->AddDevice = USBPORT_AddDevice; 2838 DriverObject->DriverUnload = USBPORT_Unload; 2839 2840 DriverObject->MajorFunction[IRP_MJ_CREATE] = USBPORT_Dispatch; 2841 DriverObject->MajorFunction[IRP_MJ_CLOSE] = USBPORT_Dispatch; 2842 DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = USBPORT_Dispatch; 2843 DriverObject->MajorFunction[IRP_MJ_INTERNAL_DEVICE_CONTROL] = USBPORT_Dispatch; 2844 DriverObject->MajorFunction[IRP_MJ_PNP] = USBPORT_Dispatch; 2845 DriverObject->MajorFunction[IRP_MJ_POWER] = USBPORT_Dispatch; 2846 DriverObject->MajorFunction[IRP_MJ_SYSTEM_CONTROL] = USBPORT_Dispatch; 2847 2848 RegPacket->UsbPortDbgPrint = USBPORT_DbgPrint; 2849 RegPacket->UsbPortTestDebugBreak = USBPORT_TestDebugBreak; 2850 RegPacket->UsbPortAssertFailure = USBPORT_AssertFailure; 2851 RegPacket->UsbPortGetMiniportRegistryKeyValue = USBPORT_GetMiniportRegistryKeyValue; 2852 RegPacket->UsbPortInvalidateRootHub = USBPORT_InvalidateRootHub; 2853 RegPacket->UsbPortInvalidateEndpoint = USBPORT_InvalidateEndpoint; 2854 RegPacket->UsbPortCompleteTransfer = USBPORT_MiniportCompleteTransfer; 2855 RegPacket->UsbPortCompleteIsoTransfer = USBPORT_CompleteIsoTransfer; 2856 RegPacket->UsbPortLogEntry = USBPORT_LogEntry; 2857 RegPacket->UsbPortGetMappedVirtualAddress = USBPORT_GetMappedVirtualAddress; 2858 RegPacket->UsbPortRequestAsyncCallback = USBPORT_RequestAsyncCallback; 2859 RegPacket->UsbPortReadWriteConfigSpace = USBPORT_ReadWriteConfigSpace; 2860 RegPacket->UsbPortWait = USBPORT_Wait; 2861 RegPacket->UsbPortInvalidateController = USBPORT_InvalidateController; 2862 RegPacket->UsbPortBugCheck = USBPORT_BugCheck; 2863 RegPacket->UsbPortNotifyDoubleBuffer = USBPORT_NotifyDoubleBuffer; 2864 2865 RtlCopyMemory(&MiniPortInterface->Packet, 2866 RegPacket, 2867 sizeof(USBPORT_REGISTRATION_PACKET)); 2868 2869 return STATUS_SUCCESS; 2870 } 2871 2872 NTSTATUS 2873 NTAPI 2874 DriverEntry(IN PDRIVER_OBJECT DriverObject, 2875 IN PUNICODE_STRING RegistryPath) 2876 { 2877 return STATUS_SUCCESS; 2878 } 2879