KernelLand/Modules/USB/EHCI/ehci.c

   1 /*
   2  * Acess2 EHCI Driver
   3  * - By John Hodge (thePowersGang)
   4  *
   5  * ehci.c
   6  * - Driver Core
   7  */
   8 #define DEBUG   1
   9 #define VERSION VER2(0,1)
  10 #include <acess.h>
  11 #include <modules.h>
  12 #include <usb_host.h>
  13 #include "ehci.h"
  14 #include <drv_pci.h>
  15 #include <limits.h>
  16 #include <events.h>
  17 #include <timers.h>
  18
  19 // === CONSTANTS ===
  20 #define EHCI_MAX_CONTROLLERS    4
  21 #define EHCI_THREADEVENT_IOC    THREAD_EVENT_USER1
  22 #define EHCI_THREADEVENT_PORTSC THREAD_EVENT_USER2
  23
  24 // === PROTOTYPES ===
  25  int    EHCI_Initialise(char **Arguments);
  26  int    EHCI_Cleanup(void);
  27  int    EHCI_InitController(tPAddr BaseAddress, Uint8 InterruptNum);
  28 void    EHCI_InterruptHandler(int IRQ, void *Ptr);
  29 // -- API ---
  30 void    *EHCI_InitInterrupt(void *Ptr, int Endpoint, int bInput, int Period, tUSBHostCb Cb, void *CbData, void *Buf, size_t Length);
  31 void    *EHCI_InitIsoch  (void *Ptr, int Endpoint, size_t MaxPacketSize);
  32 void    *EHCI_InitControl(void *Ptr, int Endpoint, size_t MaxPacketSize);
  33 void    *EHCI_InitBulk   (void *Ptr, int Endpoint, size_t MaxPacketSize);
  34 void    EHCI_RemEndpoint(void *Ptr, void *Handle);
  35 void    *EHCI_SendControl(void *Ptr, void *Dest, tUSBHostCb Cb, void *CbData,
  36         int isOutbound,
  37         const void *SetupData, size_t SetupLength,
  38         const void *OutData, size_t OutLength,
  39         void *InData, size_t InLength
  40         );
  41 void    *EHCI_SendBulk(void *Ptr, void *Dest, tUSBHostCb Cb, void *CbData, int Dir, void *Data, size_t Length);
  42 void    EHCI_FreeOp(void *Ptr, void *Handle);
  43 Uint32  EHCI_int_RootHub_FeatToMask(int Feat);
  44 void    EHCI_RootHub_SetPortFeature(void *Ptr, int Port, int Feat);
  45 void    EHCI_RootHub_ClearPortFeature(void *Ptr, int Port, int Feat);
  46  int    EHCI_RootHub_GetPortStatus(void *Ptr, int Port, int Flag);
  47 // --- Internals ---
  48 tEHCI_qTD       *EHCI_int_AllocateTD(tEHCI_Controller *Cont, int PID, void *Data, size_t Length, tUSBHostCb Cb, void *CbData);
  49 void    EHCI_int_DeallocateTD(tEHCI_Controller *Cont, tEHCI_qTD *TD);
  50 void    EHCI_int_AppendTD(tEHCI_Controller *Cont, tEHCI_QH *QH, tEHCI_qTD *TD);
  51 tEHCI_QH        *EHCI_int_AllocateQH(tEHCI_Controller *Cont, int Endpoint, size_t MaxPacketSize);
  52 void    EHCI_int_DeallocateQH(tEHCI_Controller *Cont, tEHCI_QH *QH);
  53 void    EHCI_int_InterruptThread(void *ControllerPtr);
  54
  55 // === GLOBALS ===
  56 MODULE_DEFINE(0, VERSION, USB_EHCI, EHCI_Initialise, NULL, "USB_Core", NULL);
  57 tEHCI_Controller        gaEHCI_Controllers[EHCI_MAX_CONTROLLERS];
  58 tUSBHostDef     gEHCI_HostDef = {
  59         .InitInterrupt = EHCI_InitInterrupt,
  60         .InitIsoch     = EHCI_InitIsoch,
  61         .InitControl   = EHCI_InitControl,
  62         .InitBulk      = EHCI_InitBulk,
  63         .RemEndpoint   = EHCI_RemEndpoint,
  64         .SendIsoch   = NULL,
  65         .SendControl = EHCI_SendControl,
  66         .SendBulk    = EHCI_SendBulk,
  67         .FreeOp      = EHCI_FreeOp,
  68
  69         .CheckPorts = NULL,     // No need
  70         .SetPortFeature   = EHCI_RootHub_SetPortFeature,
  71         .ClearPortFeature = EHCI_RootHub_ClearPortFeature,
  72         .GetPortStatus    = EHCI_RootHub_GetPortStatus,
  73         };
  74
  75 // === CODE ===
  76 int EHCI_Initialise(char **Arguments)
  77 {
  78         for( int id = -1; (id = PCI_GetDeviceByClass(0x0C0320, 0xFFFFFF, id)) >= 0;  )
  79         {
  80                 Uint32  addr = PCI_GetBAR(id, 0);
  81                 if( addr == 0 ) {
  82                         // Oops, PCI BIOS emulation time
  83                 }
  84                 Uint8   irq = PCI_GetIRQ(id);
  85                 if( irq == 0 ) {
  86                         // TODO: The same
  87                 }
  88
  89                 Log_Log("ECHI", "Controller at PCI %i 0x%x IRQ 0x%x",
  90                         id, addr, irq);
  91
  92                 if( EHCI_InitController(addr, irq) )
  93                 {
  94                         // TODO: Detect other forms of failure than "out of slots"
  95                         break ;
  96                 }
  97         }
  98
  99         for( int i = 0; Arguments[i]; i ++ )
 100         {
 101                 char *pos = Arguments[i], *next;
 102                 LOG("pos = '%s'", pos);
 103                 tPAddr base = strtoull(pos, &next, 16);
 104                 if( base == 0 )
 105                         continue;
 106                 pos = next;
 107                 LOG("pos = '%s'", pos);
 108                 if( *pos++ != '-' )
 109                         continue;
 110                 LOG("pos = '%s'", pos);
 111                 int irq = strtol(pos, &next, 16);
 112                 if( irq == 0 )
 113                         continue ;
 114                 if( *next != 0 )
 115                         continue;
 116                 LOG("base=%x, irq=%i", base, irq);
 117                 if( EHCI_InitController(base, irq) )
 118                 {
 119                         continue ;
 120                 }
 121         }
 122
 123         return 0;
 124 }
 125
 126 int EHCI_Cleanup(void)
 127 {
 128         return 0;
 129 }
 130
 131 // --- Driver Init ---
 132 int EHCI_InitController(tPAddr BaseAddress, Uint8 InterruptNum)
 133 {
 134         tEHCI_Controller        *cont = NULL;
 135
 136         for( int i = 0; i < EHCI_MAX_CONTROLLERS; i ++ )
 137         {
 138                 if( gaEHCI_Controllers[i].PhysBase == 0 ) {
 139                         cont = &gaEHCI_Controllers[i];
 140                         cont->PhysBase = BaseAddress;
 141                         break;
 142                 }
 143         }
 144         if(!cont) {
 145                 Log_Notice("EHCI", "Too many controllers (EHCI_MAX_CONTROLLERS=%i)",
 146                         EHCI_MAX_CONTROLLERS);
 147                 return 1;
 148         }
 149
 150         // - Nuke a couple of fields so error handling code doesn't derp
 151         cont->CapRegs = NULL;
 152         cont->PeriodicQueue = NULL;
 153         cont->TDPool = NULL;
 154
 155         // -- Build up structure --
 156         cont->CapRegs = (void*)MM_MapHWPages(BaseAddress, 1);
 157         if( !cont->CapRegs ) {
 158                 Log_Warning("EHCI", "Can't map 1 page at %P into kernel space", BaseAddress);
 159                 goto _error;
 160         }
 161         // TODO: Error check
 162         if( (cont->CapRegs->CapLength & 3) ) {
 163                 Log_Warning("EHCI", "Controller at %P non-aligned op regs (%x)",
 164                         BaseAddress, cont->CapRegs->CapLength);
 165                 goto _error;
 166         }
 167         cont->OpRegs = (void*)( (Uint32*)cont->CapRegs + cont->CapRegs->CapLength / 4 );
 168         // - Allocate periodic queue
 169         tPAddr  unused;
 170         cont->PeriodicQueue = (void*)MM_AllocDMA(1, 32, &unused);
 171         if( !cont->PeriodicQueue ) {
 172                 Log_Warning("ECHI", "Can't allocate 1 32-bit page for periodic queue");
 173                 goto _error;
 174         }
 175         for( int i = 0; i < 1024; i ++ )
 176                 cont->PeriodicQueue[i] = 1;
 177         // TODO: Error check
 178         //  > Populate queue
 179
 180         // - Allocate TD pool
 181         cont->TDPool = (void*)MM_AllocDMA(1, 32, &unused);
 182         if( !cont->TDPool ) {
 183                 Log_Warning("ECHI", "Can't allocate 1 32-bit page for qTD pool");
 184                 goto _error;
 185         }
 186         for( int i = 0; i < TD_POOL_SIZE; i ++ ) {
 187                 cont->TDPool[i].Token = 3 << 8;
 188         }
 189
 190         // Get port count
 191         cont->nPorts = cont->CapRegs->HCSParams & 0xF;
 192
 193         // -- Bind IRQ --
 194         IRQ_AddHandler(InterruptNum, EHCI_InterruptHandler, cont);
 195         cont->InterruptThread = Proc_SpawnWorker(EHCI_int_InterruptThread, cont);
 196         if( !cont->InterruptThread ) {
 197                 Log_Warning("EHCI", "Can't spawn interrupt worker thread");
 198                 goto _error;
 199         }
 200         LOG("cont->InterruptThread = %p", cont->InterruptThread);
 201
 202         // -- Initialisation procedure (from ehci-r10) --
 203         // - Reset controller
 204         cont->OpRegs->USBCmd = USBCMD_HCReset;
 205         // - Set CTRLDSSEGMENT (TODO: 64-bit support)
 206         // - Set USBINTR
 207         cont->OpRegs->USBIntr = USBINTR_IOC|USBINTR_PortChange|USBINTR_FrameRollover;
 208         // - Set PERIODICLIST BASE
 209         cont->OpRegs->PeridocListBase = MM_GetPhysAddr( cont->PeriodicQueue );
 210         // - Enable controller
 211         cont->OpRegs->USBCmd = (0x40 << 16) | USBCMD_PeriodicEnable | USBCMD_Run;
 212         // - Route all ports
 213         cont->OpRegs->ConfigFlag = 1;
 214
 215         cont->DeadTD = EHCI_int_AllocateTD(cont, 0, NULL, 0, NULL, NULL);
 216         cont->DeadTD->Link = 1;
 217         cont->DeadTD->Link2 = 1;
 218         cont->DeadTD->Token = 0;
 219
 220         // -- Register with USB Core --
 221         cont->RootHub = USB_RegisterHost(&gEHCI_HostDef, cont, cont->nPorts);
 222
 223         return 0;
 224 _error:
 225         cont->PhysBase = 0;
 226         if( cont->CapRegs )
 227                 MM_Deallocate( (tVAddr)cont->CapRegs );
 228         if( cont->PeriodicQueue )
 229                 MM_Deallocate( (tVAddr)cont->PeriodicQueue );
 230         if( cont->TDPool )
 231                 MM_Deallocate( (tVAddr)cont->TDPool );
 232         return 2;
 233 }
 234
 235 void EHCI_InterruptHandler(int IRQ, void *Ptr)
 236 {
 237         tEHCI_Controller *Cont = Ptr;
 238         Uint32  sts = Cont->OpRegs->USBSts;
 239
 240         // Clear interrupts
 241         Cont->OpRegs->USBSts = sts;
 242
 243         if( sts & 0xFFFF0FC0 ) {
 244                 LOG("Oops, reserved bits set (%08x), funny hardware?", sts);
 245                 sts &= ~0xFFFF0FFC0;
 246         }
 247
 248         // Unmask read-only bits
 249         sts &= ~(0xF000);
 250
 251         if( sts & USBINTR_IOC ) {
 252                 // IOC
 253                 Threads_PostEvent(Cont->InterruptThread, EHCI_THREADEVENT_IOC);
 254                 sts &= ~USBINTR_IOC;
 255         }
 256
 257         if( sts & USBINTR_PortChange ) {
 258                 // Port change, determine what port and poke helper thread
 259                 LOG("Port status change");
 260                 Threads_PostEvent(Cont->InterruptThread, EHCI_THREADEVENT_PORTSC);
 261                 sts &= ~USBINTR_PortChange;
 262         }
 263
 264         if( sts & USBINTR_FrameRollover ) {
 265                 // Frame rollover, used to aid timing (trigger per-second operations)
 266                 LOG("Frame rollover");
 267                 sts &= ~USBINTR_FrameRollover;
 268         }
 269
 270         if( sts ) {
 271                 // Unhandled interupt bits
 272                 // TODO: Warn
 273                 LOG("WARN - Bitmask %x unhandled", sts);
 274         }
 275
 276
 277 }
 278
 279 // --------------------------------------------------------------------
 280 // USB API
 281 // --------------------------------------------------------------------
 282 void *EHCI_InitInterrupt(void *Ptr, int Endpoint, int bOutbound, int Period,
 283         tUSBHostCb Cb, void *CbData, void *Buf, size_t Length)
 284 {
 285         tEHCI_Controller        *Cont = Ptr;
 286          int    pow2period, period_pow;
 287
 288         if( Endpoint >= 256*16 )
 289                 return NULL;
 290         if( Period <= 0 )
 291                 return NULL;
 292         if( Period > 256 )
 293                 Period = 256;
 294
 295         LOG("Endpoint=%x, bOutbound=%i, Period=%i, Length=%i", Endpoint, bOutbound, Period, Length);
 296
 297         // Round the period to the closest power of two
 298         pow2period = 1;
 299         period_pow = 0;
 300         // - Find the first power above the period
 301         while( pow2period < Period )
 302         {
 303                 pow2period *= 2;
 304                 period_pow ++;
 305         }
 306         // - Check which is closest
 307         if( Period - pow2period / 2 > pow2period - Period )
 308                 Period = pow2period;
 309         else {
 310                 Period = pow2period/2;
 311                 period_pow --;
 312         }
 313
 314         // Allocate a QH
 315         tEHCI_QH *qh = EHCI_int_AllocateQH(Cont, Endpoint, Length);
 316         qh->Impl.IntPeriodPow = period_pow;
 317
 318         Mutex_Acquire(&Cont->PeriodicListLock);
 319
 320         // Choose an interrupt slot to use
 321         int minslot = 0, minslot_load = INT_MAX;
 322         for( int slot = 0; slot < Period; slot ++ )
 323         {
 324                  int    load = 0;
 325                 for( int i = 0; i < PERIODIC_SIZE; i += Period )
 326                         load += Cont->InterruptLoad[i+slot];
 327                 if( load == 0 ) break;
 328                 if( load < minslot_load ) {
 329                         minslot = slot;
 330                         minslot_load = load;
 331                 }
 332         }
 333         // Increase loading on the selected slot
 334         for( int i = minslot; i < PERIODIC_SIZE; i += Period )
 335                 Cont->InterruptLoad[i] += Length;
 336         qh->Impl.IntOfs = minslot;
 337
 338         // Allocate TD for the data
 339         tEHCI_qTD *td = EHCI_int_AllocateTD(Cont, (bOutbound ? PID_OUT : PID_IN), Buf, Length, Cb, CbData);
 340         EHCI_int_AppendTD(Cont, qh, td);
 341
 342         // Insert into the periodic list
 343         for( int i = 0; i < PERIODIC_SIZE; i += Period )
 344         {
 345                 // Walk list until
 346                 // - the end is reached
 347                 // - this QH is found
 348                 // - A QH with a lower period is encountered
 349                 tEHCI_QH        *pqh = NULL;
 350                 tEHCI_QH        *nqh;
 351                 for( nqh = Cont->PeriodicQueueV[i]; nqh; pqh = nqh, nqh = nqh->Impl.Next )
 352                 {
 353                         if( nqh == qh )
 354                                 break;
 355                         if( nqh->Impl.IntPeriodPow < period_pow )
 356                                 break;
 357                 }
 358
 359                 // Somehow, we've already been added to this queue.
 360                 if( nqh && nqh == qh )
 361                         continue ;
 362
 363                 if( qh->Impl.Next && qh->Impl.Next != nqh ) {
 364                         Log_Warning("EHCI", "Suspected bookkeeping error on %p - int list %i+%i overlap",
 365                                 Cont, period_pow, minslot);
 366                         break;
 367                 }
 368
 369                 if( nqh ) {
 370                         qh->Impl.Next = nqh;
 371                         qh->HLink = MM_GetPhysAddr(nqh) | 2;
 372                 }
 373                 else {
 374                         qh->Impl.Next = NULL;
 375                         qh->HLink = 2|1;        // QH, Terminate
 376                 }
 377
 378                 if( pqh ) {
 379                         pqh->Impl.Next = qh;
 380                         pqh->HLink = MM_GetPhysAddr(qh) | 2;
 381                 }
 382                 else {
 383                         Cont->PeriodicQueueV[i] = qh;
 384                         Cont->PeriodicQueue[i] = MM_GetPhysAddr(qh) | 2;
 385                 }
 386         }
 387         Mutex_Release(&Cont->PeriodicListLock);
 388
 389         return qh;
 390 }
 391
 392 void *EHCI_InitIsoch(void *Ptr, int Endpoint, size_t MaxPacketSize)
 393 {
 394         return (void*)(tVAddr)(Endpoint + 1);
 395 }
 396 void *EHCI_InitControl(void *Ptr, int Endpoint, size_t MaxPacketSize)
 397 {
 398         tEHCI_Controller *Cont = Ptr;
 399
 400         // Allocate a QH
 401         tEHCI_QH *qh = EHCI_int_AllocateQH(Cont, Endpoint, MaxPacketSize);
 402         qh->CurrentTD = MM_GetPhysAddr(Cont->DeadTD);
 403
 404         // Append to async list
 405         if( Cont->LastAsyncHead ) {
 406                 Cont->LastAsyncHead->HLink = MM_GetPhysAddr(qh)|2;
 407                 Cont->LastAsyncHead->Impl.Next = qh;
 408                 LOG("- Placed after %p", Cont->LastAsyncHead);
 409         }
 410         else {
 411                 Cont->OpRegs->AsyncListAddr = MM_GetPhysAddr(qh)|2;
 412         }
 413         qh->HLink = Cont->OpRegs->AsyncListAddr;
 414         Cont->OpRegs->USBCmd |= USBCMD_AsyncEnable;
 415         Cont->LastAsyncHead = qh;
 416
 417         LOG("Created %p for %p Ep 0x%x - %i bytes MPS", qh, Ptr, Endpoint, MaxPacketSize);
 418
 419         return qh;
 420 }
 421 void *EHCI_InitBulk(void *Ptr, int Endpoint, size_t MaxPacketSize)
 422 {
 423         return EHCI_InitControl(Ptr, Endpoint, MaxPacketSize);
 424 }
 425 void EHCI_RemEndpoint(void *Ptr, void *Handle)
 426 {
 427         if( Handle == NULL )
 428                 return ;
 429         else if( (tVAddr)Handle <= 256*16 )
 430                 return ;        // Isoch
 431         else {
 432                 tEHCI_QH        *qh = Ptr;
 433
 434                 // Remove QH from list
 435                 // - If it's a polling endpoint, need to remove from all periodic lists
 436                 if( qh->Impl.IntPeriodPow != 0xFF) {
 437                         // Poll
 438                 }
 439                 else {
 440                         // GP
 441                 }
 442
 443                 // Deallocate QH
 444                 EHCI_int_DeallocateQH(Ptr, Handle);
 445         }
 446 }
 447
 448 void *EHCI_SendControl(void *Ptr, void *Dest, tUSBHostCb Cb, void *CbData,
 449         int isOutbound,
 450         const void *SetupData, size_t SetupLength,
 451         const void *OutData, size_t OutLength,
 452         void *InData, size_t InLength
 453         )
 454 {
 455         tEHCI_Controller *Cont = Ptr;
 456         tEHCI_qTD       *td_setup, *td_data, *td_status;
 457
 458         // Sanity checks
 459         if( (tVAddr)Dest <= 256*16 )
 460                 return NULL;
 461
 462         LOG("Dest=%p, isOutbound=%i, Lengths(Setup:%i,Out:%i,In:%i)", Dest, isOutbound, SetupLength, OutLength, InLength);
 463
 464         // Check size of SETUP and status
 465
 466         // Allocate TDs
 467         td_setup = EHCI_int_AllocateTD(Cont, PID_SETUP, (void*)SetupData, SetupLength, NULL, NULL);
 468         if( isOutbound )
 469         {
 470                 td_data = OutData ? EHCI_int_AllocateTD(Cont, PID_OUT, (void*)OutData, OutLength, NULL, NULL) : NULL;
 471                 td_status = EHCI_int_AllocateTD(Cont, PID_IN, InData, InLength, Cb, CbData);
 472         }
 473         else
 474         {
 475                 td_data = InData ? EHCI_int_AllocateTD(Cont, PID_IN, InData, InLength, NULL, NULL) : NULL;
 476                 td_status = EHCI_int_AllocateTD(Cont, PID_OUT, (void*)OutData, OutLength, Cb, CbData);
 477                 td_status->Token |= (1 << 15);  // IOC
 478         }
 479
 480         // Append TDs
 481         if( td_data ) {
 482                 td_setup->Link = MM_GetPhysAddr(td_data);
 483                 td_data->Link = MM_GetPhysAddr(td_status) | 1;
 484                 td_data->Token |= (1 << 8);     // Active
 485         }
 486         else {
 487                 td_setup->Link = MM_GetPhysAddr(td_status) | 1;
 488         }
 489         td_setup->Token |= (1 << 8);    // Active
 490         td_status->Token |= (1 << 8);
 491         EHCI_int_AppendTD(Cont, Dest, td_setup);
 492
 493         return td_status;
 494 }
 495
 496 void *EHCI_SendBulk(void *Ptr, void *Dest, tUSBHostCb Cb, void *CbData, int Dir, void *Data, size_t Length)
 497 {
 498         tEHCI_Controller        *Cont = Ptr;
 499
 500         // Sanity check the pointer
 501         // - Can't be NULL or an isoch
 502         if( (tVAddr)Dest <= 256*16 )
 503                 return NULL;
 504
 505         // Allocate single TD
 506         tEHCI_qTD       *td = EHCI_int_AllocateTD(Cont, (Dir ? PID_OUT : PID_IN), Data, Length, Cb, CbData);
 507         EHCI_int_AppendTD(Cont, Dest, td);
 508
 509         return td;
 510 }
 511
 512 void EHCI_FreeOp(void *Ptr, void *Handle)
 513 {
 514         tEHCI_Controller        *Cont = Ptr;
 515
 516         EHCI_int_DeallocateTD(Cont, Handle);
 517 }
 518
 519 Uint32 EHCI_int_RootHub_FeatToMask(int Feat)
 520 {
 521         switch(Feat)
 522         {
 523         case PORT_RESET:        return PORTSC_PortReset;
 524         case PORT_ENABLE:       return PORTSC_PortEnabled;
 525         default:
 526                 Log_Warning("EHCI", "Unknown root hub port feature %i", Feat);
 527                 return 0;
 528         }
 529 }
 530
 531 void EHCI_RootHub_SetPortFeature(void *Ptr, int Port, int Feat)
 532 {
 533         tEHCI_Controller        *Cont = Ptr;
 534         if(Port >= Cont->nPorts)        return;
 535
 536         Cont->OpRegs->PortSC[Port] |= EHCI_int_RootHub_FeatToMask(Feat);
 537 }
 538
 539 void EHCI_RootHub_ClearPortFeature(void *Ptr, int Port, int Feat)
 540 {
 541         tEHCI_Controller        *Cont = Ptr;
 542         if(Port >= Cont->nPorts)        return;
 543
 544         Cont->OpRegs->PortSC[Port] &= ~EHCI_int_RootHub_FeatToMask(Feat);
 545 }
 546
 547 int EHCI_RootHub_GetPortStatus(void *Ptr, int Port, int Flag)
 548 {
 549         tEHCI_Controller        *Cont = Ptr;
 550         if(Port >= Cont->nPorts)        return 0;
 551
 552         return !!(Cont->OpRegs->PortSC[Port] & EHCI_int_RootHub_FeatToMask(Flag));
 553 }
 554
 555 // --------------------------------------------------------------------
 556 // Internals
 557 // --------------------------------------------------------------------
 558 tEHCI_qTD *EHCI_int_GetTDFromPhys(tEHCI_Controller *Cont, Uint32 Addr)
 559 {
 560         if( Addr == 0 ) return NULL;
 561         LOG("%p + (%x - %x)", Cont->TDPool, Addr, MM_GetPhysAddr(Cont->TDPool));
 562         return Cont->TDPool + (Addr - MM_GetPhysAddr(Cont->TDPool))/sizeof(tEHCI_qTD);
 563 }
 564
 565 tEHCI_qTD *EHCI_int_AllocateTD(tEHCI_Controller *Cont, int PID, void *Data, size_t Length, tUSBHostCb Cb, void *CbData)
 566 {
 567 //      Semaphore_Wait(&Cont->TDSemaphore, 1);
 568         Mutex_Acquire(&Cont->TDPoolMutex);
 569         for( int i = 0; i < TD_POOL_SIZE; i ++ )
 570         {
 571                 if( ((Cont->TDPool[i].Token >> 8) & 3) != 3 )
 572                         continue ;
 573                 Cont->TDPool[i].Token = (PID << 8) | (Length << 16);
 574                 // NOTE: Assumes that `Length` is <= PAGE_SIZE
 575                 Cont->TDPool[i].Pages[0] = MM_GetPhysAddr(Data);
 576                 if( (Cont->TDPool[i].Pages[0] & (PAGE_SIZE-1)) + Length - 1 > PAGE_SIZE )
 577                         Cont->TDPool[i].Pages[1] = MM_GetPhysAddr((char*)Data + Length - 1) & ~(PAGE_SIZE-1);
 578                 Mutex_Release(&Cont->TDPoolMutex);
 579                 LOG("Allocated %p for PID %i on %p", &Cont->TDPool[i], PID, Cont);
 580                 return &Cont->TDPool[i];
 581         }
 582
 583         Mutex_Release(&Cont->TDPoolMutex);
 584         return NULL;
 585 }
 586
 587 void EHCI_int_DeallocateTD(tEHCI_Controller *Cont, tEHCI_qTD *TD)
 588 {
 589         UNIMPLEMENTED();
 590 }
 591
 592 void EHCI_int_AppendTD(tEHCI_Controller *Cont, tEHCI_QH *QH, tEHCI_qTD *TD)
 593 {
 594         tEHCI_qTD       *ptd = NULL;
 595         Uint32  link = QH->CurrentTD;
 596
 597         // TODO: Need locking and validation here
 598         while( link && !(link & 1) )
 599         {
 600                 ptd = EHCI_int_GetTDFromPhys(Cont, link);
 601                 link = ptd->Link;
 602         }
 603         // TODO: Figure out how to follow this properly
 604         if( !ptd ) {
 605                 QH->CurrentTD = MM_GetPhysAddr(TD);
 606                 LOG("Appended %p to beginning of %p", TD, QH);
 607         }
 608         else {
 609                 ptd->Link = MM_GetPhysAddr(TD);
 610                 LOG("Appended %p to end of %p", TD, QH);
 611         }
 612 }
 613
 614 tEHCI_QH *EHCI_int_AllocateQH(tEHCI_Controller *Cont, int Endpoint, size_t MaxPacketSize)
 615 {
 616         tEHCI_QH        *ret;
 617         Mutex_Acquire(&Cont->QHPoolMutex);
 618         for( int i = 0; i < QH_POOL_SIZE; i ++ )
 619         {
 620                 if( !MM_GetPhysAddr( Cont->QHPools[i/QH_POOL_NPERPAGE] ) ) {
 621                         tPAddr  tmp;
 622                         Cont->QHPools[i/QH_POOL_NPERPAGE] = (void*)MM_AllocDMA(1, 32, &tmp);
 623                         memset(Cont->QHPools[i/QH_POOL_NPERPAGE], 0, PAGE_SIZE);
 624                 }
 625
 626                 ret = &Cont->QHPools[i/QH_POOL_NPERPAGE][i%QH_POOL_NPERPAGE];
 627                 if( ret->HLink == 0 ) {
 628                         ret->HLink = 1;
 629                         ret->Overlay.Link = 1;
 630                         ret->Endpoint = (Endpoint >> 4) | 0x80 | ((Endpoint & 0xF) << 8)
 631                                 | (MaxPacketSize << 16);
 632                         // TODO: Endpoint speed (13:12) 0:Full, 1:Low, 2:High
 633                         // TODO: Control Endpoint Flag (27) 0:*, 1:Full/Low Control
 634                         Mutex_Release(&Cont->QHPoolMutex);
 635                         return ret;
 636                 }
 637         }
 638         Mutex_Release(&Cont->QHPoolMutex);
 639         return NULL;
 640 }
 641
 642 void EHCI_int_DeallocateQH(tEHCI_Controller *Cont, tEHCI_QH *QH)
 643 {
 644         // TODO: Ensure it's unused (somehow)
 645         QH->HLink = 0;
 646 }
 647
 648 void EHCI_int_HandlePortConnectChange(tEHCI_Controller *Cont, int Port)
 649 {
 650         // Connect Event
 651         if( Cont->OpRegs->PortSC[Port] & PORTSC_CurrentConnectStatus )
 652         {
 653                 // Is the device low-speed?
 654                 if( (Cont->OpRegs->PortSC[Port] & PORTSC_LineStatus_MASK) == PORTSC_LineStatus_Kstate )
 655                 {
 656                         LOG("Low speed device on %p Port %i, giving to companion", Cont, Port);
 657                         Cont->OpRegs->PortSC[Port] |= PORTSC_PortOwner;
 658                 }
 659                 // not a low-speed device, EHCI reset
 660                 else
 661                 {
 662                         LOG("Device connected on %p #%i", Cont, Port);
 663                         // Reset procedure.
 664                         USB_PortCtl_BeginReset(Cont->RootHub, Port);
 665                 }
 666         }
 667         // Disconnect Event
 668         else
 669         {
 670                 if( Cont->OpRegs->PortSC[Port] & PORTSC_PortOwner ) {
 671                         LOG("Companion port %i disconnected, taking it back", Port);
 672                         Cont->OpRegs->PortSC[Port] &= ~PORTSC_PortOwner;
 673                 }
 674                 else {
 675                         LOG("Port %i disconnected", Port);
 676                         USB_DeviceDisconnected(Cont->RootHub, Port);
 677                 }
 678         }
 679 }
 680
 681 void EHCI_int_InterruptThread(void *ControllerPtr)
 682 {
 683         tEHCI_Controller        *Cont = ControllerPtr;
 684         while(Cont->OpRegs)
 685         {
 686                 Uint32  events;
 687
 688                 LOG("sleeping for events");
 689                 events = Threads_WaitEvents(EHCI_THREADEVENT_IOC|EHCI_THREADEVENT_PORTSC);
 690                 if( !events ) {
 691                         // TODO: Should this cause a termination?
 692                 }
 693                 LOG("events = 0x%x", events);
 694
 695                 if( events & EHCI_THREADEVENT_IOC )
 696                 {
 697                         // IOC, Do whatever it is you do
 698                 }
 699
 700                 // Port status change interrupt
 701                 if( events & EHCI_THREADEVENT_PORTSC )
 702                 {
 703                         // Check for port status changes
 704                         for(int i = 0; i < Cont->nPorts; i ++ )
 705                         {
 706                                 Uint32  sts = Cont->OpRegs->PortSC[i];
 707                                 LOG("Port %i: sts = %x", i, sts);
 708                                 Cont->OpRegs->PortSC[i] = sts;
 709                                 if( sts & PORTSC_ConnectStatusChange )
 710                                         EHCI_int_HandlePortConnectChange(Cont, i);
 711
 712                                 if( sts & PORTSC_PortEnableChange )
 713                                 {
 714                                         // Handle enable/disable
 715                                 }
 716
 717                                 if( sts & PORTSC_OvercurrentChange )
 718                                 {
 719                                         // Handle over-current change
 720                                 }
 721                         }
 722                 }
 723
 724                 LOG("Going back to sleep");
 725         }
 726 }