#define DEBUGGING_LEVEL 0 // 1 = low; 2 = high; 3 = extreme
#define DEBUGLOG IOLog
#include <IOKit/IOWorkLoop.h>
#include <IOKit/usb/IOUSBWorkLoop.h>
OSDefineMetaClassAndStructors( IOUSBWorkLoop, IOWorkLoop )
IOUSBWorkLoop * IOUSBWorkLoop::workLoop()
{
IOUSBWorkLoop *loop;
loop = new IOUSBWorkLoop;
if(!loop)
return loop;
if(!loop->init())
{
loop->release();
loop = NULL;
}
return loop;
}
void IOUSBWorkLoop::closeGate()
{
IOWorkLoop::closeGate();
if(fSleepToken)
{
IOReturn res;
do
{
res = sleepGate(fSleepToken, THREAD_ABORTSAFE);
if(res == kIOReturnSuccess)
break;
IOLog("sleepGate returned 0x%x\n", res);
} while (true);
}
}
bool IOUSBWorkLoop::tryCloseGate()
{
bool ret;
ret = IOWorkLoop::tryCloseGate();
if(ret && fSleepToken)
{
openGate();
ret = false;
}
return ret;
}
IOReturn IOUSBWorkLoop::sleep(void *token)
{
if(fSleepToken)
{
DEBUGLOG("IOUSBWorkLoop::sleep: Already asleep: %p\n", token);
return kIOReturnError;
}
fSleepToken = token;
openGate();
return kIOReturnSuccess;
}
IOReturn
IOUSBWorkLoop::wake(void *token)
{
if(fSleepToken != token)
{
DEBUGLOG("IOUSBWorkLoop::wake: wrong token: %p<->%p\n", token, fSleepToken);
return kIOReturnError;
}
IORecursiveLockLock(gateLock);
fSleepToken = NULL;
wakeupGate(token, false);
return kIOReturnSuccess;
}
void
IOUSBWorkLoop::CloseGate(void)
{
closeGate();
}
void
IOUSBWorkLoop::OpenGate(void)
{
openGate();
}