IOPMWorkArbiter.cpp [plain text]
#include "IOPMWorkArbiter.h"
#include "IOKit/IOLocks.h"
#undef super
#define super IOEventSource
OSDefineMetaClassAndStructors(IOPMWorkArbiter, IOEventSource);
static
void * _dequeue(void ** inList, SInt32 inOffset)
{
void * oldListHead;
void * newListHead;
do {
oldListHead = *inList;
if (oldListHead == NULL) {
break;
}
newListHead = *(void **) (((char *) oldListHead) + inOffset);
} while (! OSCompareAndSwap((UInt32)oldListHead,
(UInt32)newListHead, (UInt32 *)inList));
return oldListHead;
}
static
void _enqueue(void ** inList, void * inNewLink, SInt32 inOffset)
{
void * oldListHead;
void * newListHead = inNewLink;
void ** newLinkNextPtr = (void **) (((char *) inNewLink) + inOffset);
do {
oldListHead = *inList;
*newLinkNextPtr = oldListHead;
} while (! OSCompareAndSwap((UInt32)oldListHead, (UInt32)newListHead,
(UInt32 *)inList));
}
IOPMWorkArbiter *IOPMWorkArbiter::pmWorkArbiter(
IOPMrootDomain *inOwner)
{
IOPMWorkArbiter *me = new IOPMWorkArbiter;
if(me && !me->init((OSObject *)inOwner, 0) )
{
me->release();
return NULL;
}
return me;
}
bool IOPMWorkArbiter::init(OSObject *owner, Action action)
{
if(!(super::init(owner, (IOEventSource::Action) action)))
return false;
events = NULL;
fRootDomain = (IOPMrootDomain *)owner;
if (!(tmpLock = IOLockAlloc())) {
panic("IOPMWorkArbiter::init can't alloc lock");
}
return true;
}
bool IOPMWorkArbiter::driverAckedOccurred(IOService *inTarget)
{
PMEventEntry *new_one = NULL;
new_one = (PMEventEntry *)IOMalloc(sizeof(PMEventEntry));
if(!new_one) return false;
new_one->actionType = IOPMWorkArbiter::kDriverAcked;
new_one->target = inTarget;
IOLockLock(tmpLock);
_enqueue((void **)&events, (void *)new_one, 0);
IOLockUnlock(tmpLock);
signalWorkAvailable();
return true;
}
bool IOPMWorkArbiter::allAckedOccurred(IOService *inTarget)
{
PMEventEntry *new_one = NULL;
new_one = (PMEventEntry *)IOMalloc(sizeof(PMEventEntry));
if(!new_one) return false;
new_one->actionType = IOPMWorkArbiter::kAllAcked;
new_one->target = inTarget;
IOLockLock(tmpLock);
_enqueue((void **)&events, (void *)new_one, 0);
IOLockUnlock(tmpLock);
signalWorkAvailable();
return true;
}
bool IOPMWorkArbiter::clamshellStateChangeOccurred(uint32_t messageValue)
{
PMEventEntry *new_one = NULL;
new_one = (PMEventEntry *)IOMalloc(sizeof(PMEventEntry));
if(!new_one) return false;
new_one->actionType = IOPMWorkArbiter::kRootDomainClamshellChanged;
new_one->target = (IOService *)fRootDomain;
new_one->intArgument = messageValue;
IOLockLock(tmpLock);
_enqueue((void **)&events, (void *)new_one, 0);
IOLockUnlock(tmpLock);
signalWorkAvailable();
return true;
}
void IOPMWorkArbiter::checkForWorkThreadFunc(void *refcon)
{
PMEventEntry *theNode = (PMEventEntry *)refcon;
IOService *theTarget;
UInt16 theAction;
if(!theNode) return;
theTarget = theNode->target;
theAction = theNode->actionType;
IOFree(theNode, sizeof(PMEventEntry));
theNode = NULL;
switch (theAction)
{
case kAllAcked:
theTarget->all_acked_threaded();
break;
case kDriverAcked:
theTarget->driver_acked_threaded();
break;
}
}
bool IOPMWorkArbiter::checkForWork()
{
PMEventEntry *theNode;
IOService *theTarget;
UInt16 theAction;
IOLockLock(tmpLock);
if((theNode = (PMEventEntry *)_dequeue((void **)&events, 0)))
{
IOLockUnlock(tmpLock);
theTarget = theNode->target;
theAction = theNode->actionType;
IOFree((void *)theNode, sizeof(PMEventEntry));
switch (theAction)
{
case kAllAcked:
theTarget->all_acked_threaded();
break;
case kDriverAcked:
theTarget->driver_acked_threaded();
break;
case kRootDomainClamshellChanged:
theTarget->messageClients(
kIOPMMessageClamshellStateChange,
(void *)theNode->intArgument);
break;
}
}
else {
IOLockUnlock(tmpLock);
}
if(events) return true;
else return false;
}