#define DEBUGGING_LEVEL 0 // 1 = low; 2 = high; 3 = extreme
#define DEBUGLOG IOLog
#import <IOKit/firewire/IOFWWorkLoop.h>
#import <IOKit/IOWorkLoop.h>
#import <IOKit/IOLocksPrivate.h>
SInt32 IOFWWorkLoop::sLockGroupCount = 0;
OSDefineMetaClassAndStructors( IOFWWorkLoop, IOWorkLoop )
IOFWWorkLoop * IOFWWorkLoop::workLoop()
{
IOFWWorkLoop *loop;
loop = OSTypeAlloc( IOFWWorkLoop );
if( !loop )
return loop;
if( !loop->init() )
{
loop->release();
loop = NULL;
}
return loop;
}
bool IOFWWorkLoop::init( void )
{
bool success = true;
if( success )
{
SInt32 count = OSIncrementAtomic( &sLockGroupCount );
char name[64];
snprintf( name, sizeof(name), "FireWire %d", (int)count );
fLockGroup = lck_grp_alloc_init( name, LCK_GRP_ATTR_NULL );
if( !fLockGroup )
{
success = false;
}
}
if( success )
{
gateLock = IORecursiveLockAllocWithLockGroup( fLockGroup );
}
if( success )
{
fRemoveSourceDeferredSet = OSSet::withCapacity( 1 );
if( fRemoveSourceDeferredSet == NULL )
{
success = false;
}
}
if( success )
{
success = IOWorkLoop::init();
}
return success;
}
void IOFWWorkLoop::free( void )
{
if( fLockGroup )
{
lck_grp_free( fLockGroup );
fLockGroup = NULL;
}
if( fRemoveSourceDeferredSet )
{
fRemoveSourceDeferredSet->release();
fRemoveSourceDeferredSet = NULL;
}
IOWorkLoop::free();
}
IOReturn IOFWWorkLoop::removeEventSource(IOEventSource *toRemove)
{
IOReturn status = kIOReturnSuccess;
IOWorkLoop::closeGate();
if( fRemoveSourceThread != NULL )
{
IOLog( "IOFWWorkLoop::removeEventSource - fRemoveSourceThread = (%p) != NULL\n", fRemoveSourceThread );
}
fRemoveSourceThread = IOThreadSelf();
if( fSleepToken )
{
fRemoveSourceDeferredSet->setObject( toRemove );
}
status = IOWorkLoop::removeEventSource( toRemove );
fRemoveSourceThread = NULL;
IOWorkLoop::openGate();
return status;
}
void IOFWWorkLoop::closeGate()
{
IOWorkLoop::closeGate();
if( fSleepToken &&
(fRemoveSourceThread != IOThreadSelf()) )
{
IOReturn res;
do
{
res = sleepGate( fSleepToken, THREAD_ABORTSAFE );
if( res == kIOReturnSuccess )
break;
IOLog("sleepGate returned 0x%x\n", res);
}
while( true );
}
}
bool IOFWWorkLoop::tryCloseGate()
{
bool ret;
ret = IOWorkLoop::tryCloseGate();
if( ret &&
fSleepToken &&
(fRemoveSourceThread != IOThreadSelf()) )
{
openGate();
ret = false;
}
return ret;
}
IOReturn IOFWWorkLoop::sleep(void *token)
{
if( fSleepToken )
{
DEBUGLOG( "IOFWWorkLoop::sleep: Already asleep: %p\n", token );
return kIOReturnError;
}
fSleepToken = token;
openGate();
return kIOReturnSuccess;
}
IOReturn IOFWWorkLoop::wake(void *token)
{
if( fSleepToken != token )
{
DEBUGLOG( "IOFWWorkLoop::wake: wrong token: %p<->%p\n", token, fSleepToken );
return kIOReturnError;
}
IORecursiveLockLock( gateLock );
fSleepToken = NULL;
fRemoveSourceDeferredSet->flushCollection();
wakeupGate( token, false );
return kIOReturnSuccess;
}