#include <libkern/OSDebug.h>
#include <IOKit/IOCommandGate.h>
#include <IOKit/IOWorkLoop.h>
#include <IOKit/IOReturn.h>
#include <IOKit/IOTimeStamp.h>
#define super IOEventSource
OSDefineMetaClassAndStructors(IOCommandGate, IOEventSource)
OSMetaClassDefineReservedUnused(IOCommandGate, 0);
OSMetaClassDefineReservedUnused(IOCommandGate, 1);
OSMetaClassDefineReservedUnused(IOCommandGate, 2);
OSMetaClassDefineReservedUnused(IOCommandGate, 3);
OSMetaClassDefineReservedUnused(IOCommandGate, 4);
OSMetaClassDefineReservedUnused(IOCommandGate, 5);
OSMetaClassDefineReservedUnused(IOCommandGate, 6);
OSMetaClassDefineReservedUnused(IOCommandGate, 7);
bool IOCommandGate::checkForWork() { return false; }
bool IOCommandGate::init(OSObject *inOwner, Action inAction)
{
return super::init(inOwner, (IOEventSource::Action) inAction);
}
IOCommandGate *
IOCommandGate::commandGate(OSObject *inOwner, Action inAction)
{
IOCommandGate *me = new IOCommandGate;
if (me && !me->init(inOwner, inAction)) {
me->release();
return 0;
}
return me;
}
void IOCommandGate::disable()
{
if (workLoop && !workLoop->inGate())
OSReportWithBacktrace("IOCommandGate::disable() called when not gated");
super::disable();
}
void IOCommandGate::enable()
{
if (workLoop) {
closeGate();
super::enable();
wakeupGate(&enabled, false); openGate();
}
}
void IOCommandGate::free()
{
setWorkLoop(0);
super::free();
}
void IOCommandGate::setWorkLoop(IOWorkLoop *inWorkLoop)
{
uintptr_t *sleepersP = (uintptr_t *) &reserved;
if (!inWorkLoop && workLoop) { closeGate();
*sleepersP |= 1;
while (*sleepersP >> 1) {
thread_wakeup_with_result(&enabled, THREAD_INTERRUPTED);
sleepGate(sleepersP, THREAD_UNINT);
}
*sleepersP = 0;
openGate();
}
else
super::setWorkLoop(inWorkLoop);
}
IOReturn IOCommandGate::runCommand(void *arg0, void *arg1,
void *arg2, void *arg3)
{
return runAction((Action) action, arg0, arg1, arg2, arg3);
}
IOReturn IOCommandGate::attemptCommand(void *arg0, void *arg1,
void *arg2, void *arg3)
{
return attemptAction((Action) action, arg0, arg1, arg2, arg3);
}
IOReturn IOCommandGate::runAction(Action inAction,
void *arg0, void *arg1,
void *arg2, void *arg3)
{
if (!inAction)
return kIOReturnBadArgument;
IOTimeStampConstant(IODBG_CMDQ(IOCMDQ_ACTION),
(unsigned int) inAction, (unsigned int) owner);
closeGate();
IOReturn res;
if (!workLoop->onThread()) {
while (!enabled) {
uintptr_t *sleepersP = (uintptr_t *) &reserved;
*sleepersP += 2;
IOReturn res = sleepGate(&enabled, THREAD_ABORTSAFE);
*sleepersP -= 2;
bool wakeupTearDown = (*sleepersP & 1);
if (res || wakeupTearDown) {
openGate();
if (wakeupTearDown)
commandWakeup(sleepersP);
return kIOReturnAborted;
}
}
}
res = (*inAction)(owner, arg0, arg1, arg2, arg3);
openGate();
return res;
}
IOReturn IOCommandGate::attemptAction(Action inAction,
void *arg0, void *arg1,
void *arg2, void *arg3)
{
IOReturn res;
if (!inAction)
return kIOReturnBadArgument;
if (!tryCloseGate())
return kIOReturnCannotLock;
if (!workLoop->onThread() && !enabled)
res = kIOReturnNotPermitted;
else {
IOTimeStampConstant(IODBG_CMDQ(IOCMDQ_ACTION),
(unsigned int) inAction, (unsigned int) owner);
res = (*inAction)(owner, arg0, arg1, arg2, arg3);
}
openGate();
return res;
}
IOReturn IOCommandGate::commandSleep(void *event, UInt32 interruptible)
{
if (!workLoop->inGate())
return kIOReturnNotPermitted;
return sleepGate(event, interruptible);
}
void IOCommandGate::commandWakeup(void *event, bool oneThread)
{
wakeupGate(event, oneThread);
}