#include <IOKit/system.h>
#include <IOKit/IOService.h>
#include <libkern/OSDebug.h>
#include <libkern/c++/OSContainers.h>
#include <libkern/c++/OSKext.h>
#include <libkern/c++/OSUnserialize.h>
#include <IOKit/IOCatalogue.h>
#include <IOKit/IOCommand.h>
#include <IOKit/IODeviceTreeSupport.h>
#include <IOKit/IODeviceMemory.h>
#include <IOKit/IOInterrupts.h>
#include <IOKit/IOInterruptController.h>
#include <IOKit/IOPlatformExpert.h>
#include <IOKit/IOMessage.h>
#include <IOKit/IOLib.h>
#include <IOKit/IOKitKeysPrivate.h>
#include <IOKit/IOBSD.h>
#include <IOKit/IOUserClient.h>
#include <IOKit/IOWorkLoop.h>
#include <IOKit/IOTimeStamp.h>
#include <IOKit/IOHibernatePrivate.h>
#include <IOKit/IOInterruptAccountingPrivate.h>
#include <IOKit/IOKernelReporters.h>
#include <IOKit/AppleKeyStoreInterface.h>
#include <IOKit/IOCPU.h>
#include <mach/sync_policy.h>
#include <IOKit/assert.h>
#include <sys/errno.h>
#include <sys/kdebug.h>
#include <string.h>
#include <machine/pal_routines.h>
#define LOG kprintf
#define MATCH_DEBUG 0
#define OBFUSCATE(x) ((void *)(VM_KERNEL_ADDRPERM(x)))
#include "IOServicePrivate.h"
#include "IOKitKernelInternal.h"
#define super IORegistryEntry
OSDefineMetaClassAndStructors(IOService, IORegistryEntry)
OSDefineMetaClassAndStructors(_IOServiceNotifier, IONotifier)
OSDefineMetaClassAndStructors(_IOServiceInterestNotifier, IONotifier)
OSDefineMetaClassAndStructors(_IOConfigThread, OSObject)
OSDefineMetaClassAndStructors(_IOServiceJob, OSObject)
OSDefineMetaClassAndStructors(IOResources, IOService)
OSDefineMetaClassAndStructors(_IOOpenServiceIterator, OSIterator)
OSDefineMetaClassAndAbstractStructors(IONotifier, OSObject)
static IOPlatformExpert * gIOPlatform;
static class IOPMrootDomain * gIOPMRootDomain;
const IORegistryPlane * gIOServicePlane;
const IORegistryPlane * gIOPowerPlane;
const OSSymbol * gIODeviceMemoryKey;
const OSSymbol * gIOInterruptControllersKey;
const OSSymbol * gIOInterruptSpecifiersKey;
const OSSymbol * gIOResourcesKey;
const OSSymbol * gIOResourceMatchKey;
const OSSymbol * gIOProviderClassKey;
const OSSymbol * gIONameMatchKey;
const OSSymbol * gIONameMatchedKey;
const OSSymbol * gIOPropertyMatchKey;
const OSSymbol * gIOLocationMatchKey;
const OSSymbol * gIOParentMatchKey;
const OSSymbol * gIOPathMatchKey;
const OSSymbol * gIOMatchCategoryKey;
const OSSymbol * gIODefaultMatchCategoryKey;
const OSSymbol * gIOMatchedServiceCountKey;
const OSSymbol * gIOMapperIDKey;
const OSSymbol * gIOUserClientClassKey;
const OSSymbol * gIOKitDebugKey;
const OSSymbol * gIOCommandPoolSizeKey;
const OSSymbol * gIOConsoleLockedKey;
const OSSymbol * gIOConsoleUsersKey;
const OSSymbol * gIOConsoleSessionUIDKey;
const OSSymbol * gIOConsoleSessionAuditIDKey;
const OSSymbol * gIOConsoleUsersSeedKey;
const OSSymbol * gIOConsoleSessionOnConsoleKey;
const OSSymbol * gIOConsoleSessionLoginDoneKey;
const OSSymbol * gIOConsoleSessionSecureInputPIDKey;
const OSSymbol * gIOConsoleSessionScreenLockedTimeKey;
clock_sec_t gIOConsoleLockTime;
static bool gIOConsoleLoggedIn;
#if HIBERNATION
static uint32_t gIOScreenLockState;
#endif
static IORegistryEntry * gIOChosenEntry;
static int gIOResourceGenerationCount;
const OSSymbol * gIOServiceKey;
const OSSymbol * gIOPublishNotification;
const OSSymbol * gIOFirstPublishNotification;
const OSSymbol * gIOMatchedNotification;
const OSSymbol * gIOFirstMatchNotification;
const OSSymbol * gIOTerminatedNotification;
const OSSymbol * gIOGeneralInterest;
const OSSymbol * gIOBusyInterest;
const OSSymbol * gIOAppPowerStateInterest;
const OSSymbol * gIOPriorityPowerStateInterest;
const OSSymbol * gIOConsoleSecurityInterest;
const OSSymbol * gAKSGetKey;
#if defined(__i386__) || defined(__x86_64__)
const OSSymbol * gIOCreateEFIDevicePathSymbol;
#endif
static OSDictionary * gNotifications;
static IORecursiveLock * gNotificationLock;
static IOService * gIOResources;
static IOService * gIOServiceRoot;
static OSOrderedSet * gJobs;
static semaphore_port_t gJobsSemaphore;
static IOLock * gJobsLock;
static int gOutstandingJobs;
static int gNumConfigThreads;
static int gNumWaitingThreads;
static IOLock * gIOServiceBusyLock;
static bool gCPUsRunning;
static thread_t gIOTerminateThread;
static UInt32 gIOTerminateWork;
static OSArray * gIOTerminatePhase2List;
static OSArray * gIOStopList;
static OSArray * gIOStopProviderList;
static OSArray * gIOFinalizeList;
static SInt32 gIOConsoleUsersSeed;
static OSData * gIOConsoleUsersSeedValue;
extern const OSSymbol * gIODTPHandleKey;
const OSSymbol * gIOPlatformFunctionHandlerSet;
static IOLock * gIOConsoleUsersLock;
static thread_call_t gIOConsoleLockCallout;
#define LOCKREADNOTIFY() \
IORecursiveLockLock( gNotificationLock )
#define LOCKWRITENOTIFY() \
IORecursiveLockLock( gNotificationLock )
#define LOCKWRITE2READNOTIFY()
#define UNLOCKNOTIFY() \
IORecursiveLockUnlock( gNotificationLock )
#define SLEEPNOTIFY(event) \
IORecursiveLockSleep( gNotificationLock, (void *)(event), THREAD_UNINT )
#define SLEEPNOTIFYTO(event, deadline) \
IORecursiveLockSleepDeadline( gNotificationLock, (void *)(event), deadline, THREAD_UNINT )
#define WAKEUPNOTIFY(event) \
IORecursiveLockWakeup( gNotificationLock, (void *)(event), false )
#define randomDelay() \
int del = read_processor_clock(); \
del = (((int)IOThreadSelf()) ^ del ^ (del >> 10)) & 0x3ff; \
IOSleep( del );
#define queue_element(entry, element, type, field) do { \
vm_address_t __ele = (vm_address_t) (entry); \
__ele -= -4 + ((size_t)(&((type) 4)->field)); \
(element) = (type) __ele; \
} while(0)
#define iterqueue(que, elt) \
for (queue_entry_t elt = queue_first(que); \
!queue_end(que, elt); \
elt = queue_next(elt))
struct IOInterruptAccountingReporter {
IOSimpleReporter * reporter;
IOInterruptAccountingData * statistics;
};
struct ArbitrationLockQueueElement {
queue_chain_t link;
IOThread thread;
IOService * service;
unsigned count;
bool required;
bool aborted;
};
static queue_head_t gArbitrationLockQueueActive;
static queue_head_t gArbitrationLockQueueWaiting;
static queue_head_t gArbitrationLockQueueFree;
static IOLock * gArbitrationLockQueueLock;
bool IOService::isInactive( void ) const
{ return( 0 != (kIOServiceInactiveState & getState())); }
#if defined(__i386__) || defined(__x86_64__)
struct CpuDelayEntry
{
IOService * fService;
UInt32 fMaxDelay;
UInt32 fDelayType;
};
enum {
kCpuDelayBusStall, kCpuDelayInterrupt,
kCpuNumDelayTypes
};
static OSData *sCpuDelayData = OSData::withCapacity(8 * sizeof(CpuDelayEntry));
static IORecursiveLock *sCpuDelayLock = IORecursiveLockAlloc();
static OSArray *sCpuLatencyHandlers[kCpuNumDelayTypes];
const OSSymbol *sCPULatencyFunctionName[kCpuNumDelayTypes];
static OSNumber * sCPULatencyHolder[kCpuNumDelayTypes];
static char sCPULatencyHolderName[kCpuNumDelayTypes][128];
static OSNumber * sCPULatencySet[kCpuNumDelayTypes];
static void
requireMaxCpuDelay(IOService * service, UInt32 ns, UInt32 delayType);
static IOReturn
setLatencyHandler(UInt32 delayType, IOService * target, bool enable);
#endif
void IOService::initialize( void )
{
kern_return_t err;
gIOServicePlane = IORegistryEntry::makePlane( kIOServicePlane );
gIOPowerPlane = IORegistryEntry::makePlane( kIOPowerPlane );
gIOProviderClassKey = OSSymbol::withCStringNoCopy( kIOProviderClassKey );
gIONameMatchKey = OSSymbol::withCStringNoCopy( kIONameMatchKey );
gIONameMatchedKey = OSSymbol::withCStringNoCopy( kIONameMatchedKey );
gIOPropertyMatchKey = OSSymbol::withCStringNoCopy( kIOPropertyMatchKey );
gIOPathMatchKey = OSSymbol::withCStringNoCopy( kIOPathMatchKey );
gIOLocationMatchKey = OSSymbol::withCStringNoCopy( kIOLocationMatchKey );
gIOParentMatchKey = OSSymbol::withCStringNoCopy( kIOParentMatchKey );
gIOMatchCategoryKey = OSSymbol::withCStringNoCopy( kIOMatchCategoryKey );
gIODefaultMatchCategoryKey = OSSymbol::withCStringNoCopy(
kIODefaultMatchCategoryKey );
gIOMatchedServiceCountKey = OSSymbol::withCStringNoCopy(
kIOMatchedServiceCountKey );
gIOUserClientClassKey = OSSymbol::withCStringNoCopy( kIOUserClientClassKey );
gIOResourcesKey = OSSymbol::withCStringNoCopy( kIOResourcesClass );
gIOResourceMatchKey = OSSymbol::withCStringNoCopy( kIOResourceMatchKey );
gIODeviceMemoryKey = OSSymbol::withCStringNoCopy( "IODeviceMemory" );
gIOInterruptControllersKey
= OSSymbol::withCStringNoCopy("IOInterruptControllers");
gIOInterruptSpecifiersKey
= OSSymbol::withCStringNoCopy("IOInterruptSpecifiers");
gIOMapperIDKey = OSSymbol::withCStringNoCopy(kIOMapperIDKey);
gIOKitDebugKey = OSSymbol::withCStringNoCopy( kIOKitDebugKey );
gIOCommandPoolSizeKey = OSSymbol::withCStringNoCopy( kIOCommandPoolSizeKey );
gIOGeneralInterest = OSSymbol::withCStringNoCopy( kIOGeneralInterest );
gIOBusyInterest = OSSymbol::withCStringNoCopy( kIOBusyInterest );
gIOAppPowerStateInterest = OSSymbol::withCStringNoCopy( kIOAppPowerStateInterest );
gIOPriorityPowerStateInterest = OSSymbol::withCStringNoCopy( kIOPriorityPowerStateInterest );
gIOConsoleSecurityInterest = OSSymbol::withCStringNoCopy( kIOConsoleSecurityInterest );
gNotifications = OSDictionary::withCapacity( 1 );
gIOPublishNotification = OSSymbol::withCStringNoCopy(
kIOPublishNotification );
gIOFirstPublishNotification = OSSymbol::withCStringNoCopy(
kIOFirstPublishNotification );
gIOMatchedNotification = OSSymbol::withCStringNoCopy(
kIOMatchedNotification );
gIOFirstMatchNotification = OSSymbol::withCStringNoCopy(
kIOFirstMatchNotification );
gIOTerminatedNotification = OSSymbol::withCStringNoCopy(
kIOTerminatedNotification );
gIOServiceKey = OSSymbol::withCStringNoCopy( kIOServiceClass);
gIOConsoleLockedKey = OSSymbol::withCStringNoCopy( kIOConsoleLockedKey);
gIOConsoleUsersKey = OSSymbol::withCStringNoCopy( kIOConsoleUsersKey);
gIOConsoleSessionUIDKey = OSSymbol::withCStringNoCopy( kIOConsoleSessionUIDKey);
gIOConsoleSessionAuditIDKey = OSSymbol::withCStringNoCopy( kIOConsoleSessionAuditIDKey);
gIOConsoleUsersSeedKey = OSSymbol::withCStringNoCopy(kIOConsoleUsersSeedKey);
gIOConsoleSessionOnConsoleKey = OSSymbol::withCStringNoCopy(kIOConsoleSessionOnConsoleKey);
gIOConsoleSessionLoginDoneKey = OSSymbol::withCStringNoCopy(kIOConsoleSessionLoginDoneKey);
gIOConsoleSessionSecureInputPIDKey = OSSymbol::withCStringNoCopy(kIOConsoleSessionSecureInputPIDKey);
gIOConsoleSessionScreenLockedTimeKey = OSSymbol::withCStringNoCopy(kIOConsoleSessionScreenLockedTimeKey);
gIOConsoleUsersSeedValue = OSData::withBytesNoCopy(&gIOConsoleUsersSeed, sizeof(gIOConsoleUsersSeed));
gIOPlatformFunctionHandlerSet = OSSymbol::withCStringNoCopy(kIOPlatformFunctionHandlerSet);
#if defined(__i386__) || defined(__x86_64__)
sCPULatencyFunctionName[kCpuDelayBusStall] = OSSymbol::withCStringNoCopy(kIOPlatformFunctionHandlerMaxBusDelay);
sCPULatencyFunctionName[kCpuDelayInterrupt] = OSSymbol::withCStringNoCopy(kIOPlatformFunctionHandlerMaxInterruptDelay);
uint32_t idx;
for (idx = 0; idx < kCpuNumDelayTypes; idx++)
{
sCPULatencySet[idx] = OSNumber::withNumber(-1U, 32);
sCPULatencyHolder[idx] = OSNumber::withNumber(0ULL, 64);
assert(sCPULatencySet[idx] && sCPULatencyHolder[idx]);
}
gIOCreateEFIDevicePathSymbol = OSSymbol::withCString("CreateEFIDevicePath");
#endif
gNotificationLock = IORecursiveLockAlloc();
gAKSGetKey = OSSymbol::withCStringNoCopy(AKS_PLATFORM_FUNCTION_GETKEY);
assert( gIOServicePlane && gIODeviceMemoryKey
&& gIOInterruptControllersKey && gIOInterruptSpecifiersKey
&& gIOResourcesKey && gNotifications && gNotificationLock
&& gIOProviderClassKey && gIONameMatchKey && gIONameMatchedKey
&& gIOMatchCategoryKey && gIODefaultMatchCategoryKey
&& gIOPublishNotification && gIOMatchedNotification
&& gIOTerminatedNotification && gIOServiceKey
&& gIOConsoleUsersKey && gIOConsoleSessionUIDKey
&& gIOConsoleSessionOnConsoleKey && gIOConsoleSessionSecureInputPIDKey
&& gIOConsoleUsersSeedKey && gIOConsoleUsersSeedValue);
gJobsLock = IOLockAlloc();
gJobs = OSOrderedSet::withCapacity( 10 );
gIOServiceBusyLock = IOLockAlloc();
gIOConsoleUsersLock = IOLockAlloc();
err = semaphore_create(kernel_task, &gJobsSemaphore, SYNC_POLICY_FIFO, 0);
gIOConsoleLockCallout = thread_call_allocate(&IOService::consoleLockTimer, NULL);
IORegistryEntry::getRegistryRoot()->setProperty(gIOConsoleLockedKey, kOSBooleanTrue);
assert( gIOServiceBusyLock && gJobs && gJobsLock && gIOConsoleUsersLock
&& gIOConsoleLockCallout && (err == KERN_SUCCESS) );
gIOResources = IOResources::resources();
assert( gIOResources );
gArbitrationLockQueueLock = IOLockAlloc();
queue_init(&gArbitrationLockQueueActive);
queue_init(&gArbitrationLockQueueWaiting);
queue_init(&gArbitrationLockQueueFree);
assert( gArbitrationLockQueueLock );
gIOTerminatePhase2List = OSArray::withCapacity( 2 );
gIOStopList = OSArray::withCapacity( 16 );
gIOStopProviderList = OSArray::withCapacity( 16 );
gIOFinalizeList = OSArray::withCapacity( 16 );
assert( gIOTerminatePhase2List && gIOStopList && gIOStopProviderList && gIOFinalizeList );
}
#if defined(__i386__) || defined(__x86_64__)
extern "C" {
const char *getCpuDelayBusStallHolderName(void) {
return sCPULatencyHolderName[kCpuDelayBusStall];
}
}
#endif
#if IOMATCHDEBUG
static UInt64 getDebugFlags( OSDictionary * props )
{
OSNumber * debugProp;
UInt64 debugFlags;
debugProp = OSDynamicCast( OSNumber,
props->getObject( gIOKitDebugKey ));
if( debugProp)
debugFlags = debugProp->unsigned64BitValue();
else
debugFlags = gIOKitDebug;
return( debugFlags );
}
#endif
IOService * IOService::probe( IOService * provider,
SInt32 * score )
{
return( this );
}
bool IOService::start( IOService * provider )
{
return( true );
}
void IOService::stop( IOService * provider )
{
}
bool IOService::init( OSDictionary * dictionary )
{
bool ret;
ret = super::init(dictionary);
if (!ret) return (false);
if (reserved) return (true);
reserved = IONew(ExpansionData, 1);
if (!reserved) return (false);
bzero(reserved, sizeof(*reserved));
reserved->interruptStatisticsLock = IOLockAlloc();
if (!reserved->interruptStatisticsLock) return (false);
return (true);
}
bool IOService::init( IORegistryEntry * from,
const IORegistryPlane * inPlane )
{
bool ret;
ret = super::init(from, inPlane);
if (!ret) return (false);
if (reserved) return (true);
reserved = IONew(ExpansionData, 1);
if (!reserved) return (false);
bzero(reserved, sizeof(*reserved));
reserved->interruptStatisticsLock = IOLockAlloc();
if (!reserved->interruptStatisticsLock) return (false);
return (true);
}
void IOService::free( void )
{
int i = 0;
requireMaxBusStall(0);
requireMaxInterruptDelay(0);
if( getPropertyTable())
unregisterAllInterest();
PMfree();
if (reserved) {
if (reserved->interruptStatisticsArray) {
for (i = 0; i < reserved->interruptStatisticsArrayCount; i++) {
if (reserved->interruptStatisticsArray[i].reporter)
reserved->interruptStatisticsArray[i].reporter->release();
}
IODelete(reserved->interruptStatisticsArray, IOInterruptAccountingReporter, reserved->interruptStatisticsArrayCount);
}
if (reserved->interruptStatisticsLock)
IOLockFree(reserved->interruptStatisticsLock);
IODelete(reserved, ExpansionData, 1);
}
if (_numInterruptSources && _interruptSources)
{
IOFree(_interruptSources, _numInterruptSources * sizeof(IOInterruptSource));
_interruptSources = 0;
}
super::free();
}
bool IOService::attach( IOService * provider )
{
bool ok;
if( provider) {
if( gIOKitDebug & kIOLogAttach)
LOG( "%s::attach(%s)\n", getName(),
provider->getName());
provider->lockForArbitration();
if( provider->__state[0] & kIOServiceInactiveState)
ok = false;
else
ok = attachToParent( provider, gIOServicePlane);
provider->unlockForArbitration();
} else {
gIOServiceRoot = this;
ok = attachToParent( getRegistryRoot(), gIOServicePlane);
}
if (ok && !__provider) (void) getProvider();
return( ok );
}
IOService * IOService::getServiceRoot( void )
{
return( gIOServiceRoot );
}
void IOService::detach( IOService * provider )
{
IOService * newProvider = 0;
SInt32 busy;
bool adjParent;
if( gIOKitDebug & kIOLogAttach)
LOG("%s::detach(%s)\n", getName(), provider->getName());
lockForArbitration();
uint64_t regID1 = provider->getRegistryEntryID();
uint64_t regID2 = getRegistryEntryID();
IOServiceTrace(
IOSERVICE_DETACH,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32));
adjParent = ((busy = (__state[1] & kIOServiceBusyStateMask))
&& (provider == getProvider()));
detachFromParent( provider, gIOServicePlane );
if( busy) {
newProvider = getProvider();
if( busy && (__state[1] & kIOServiceTermPhase3State) && (0 == newProvider))
_adjustBusy( -busy );
}
if (kIOServiceInactiveState & __state[0]) {
getMetaClass()->removeInstance(this);
IORemoveServicePlatformActions(this);
}
unlockForArbitration();
if( newProvider) {
newProvider->lockForArbitration();
newProvider->_adjustBusy(1);
newProvider->unlockForArbitration();
}
if( provider->lockForArbitration( true ))
{
if (kIOServiceStartState & __state[1])
{
provider->scheduleTerminatePhase2();
}
if( adjParent) provider->_adjustBusy( -1 );
if( (provider->__state[1] & kIOServiceTermPhase3State)
&& (0 == provider->getClient())) {
provider->scheduleFinalize();
}
provider->unlockForArbitration();
}
}
void IOService::registerService( IOOptionBits options )
{
char * pathBuf;
const char * path;
char * skip;
int len;
enum { kMaxPathLen = 256 };
enum { kMaxChars = 63 };
IORegistryEntry * parent = this;
IORegistryEntry * root = getRegistryRoot();
while( parent && (parent != root))
parent = parent->getParentEntry( gIOServicePlane);
if( parent != root) {
IOLog("%s: not registry member at registerService()\n", getName());
return;
}
if( gIOPlatform && (!gIOPlatform->platformAdjustService(this)))
return;
IOInstallServicePlatformActions(this);
if( (this != gIOResources)
&& (kIOLogRegister & gIOKitDebug)) {
pathBuf = (char *) IOMalloc( kMaxPathLen );
IOLog( "Registering: " );
len = kMaxPathLen;
if( pathBuf && getPath( pathBuf, &len, gIOServicePlane)) {
path = pathBuf;
if( len > kMaxChars) {
IOLog("..");
len -= kMaxChars;
path += len;
if( (skip = strchr( path, '/')))
path = skip;
}
} else
path = getName();
IOLog( "%s\n", path );
if( pathBuf)
IOFree( pathBuf, kMaxPathLen );
}
startMatching( options );
}
void IOService::startMatching( IOOptionBits options )
{
IOService * provider;
UInt32 prevBusy = 0;
bool needConfig;
bool needWake = false;
bool ok;
bool sync;
bool waitAgain;
lockForArbitration();
sync = (options & kIOServiceSynchronous)
|| ((provider = getProvider())
&& (provider->__state[1] & kIOServiceSynchronousState));
if ( options & kIOServiceAsynchronous )
sync = false;
needConfig = (0 == (__state[1] & (kIOServiceNeedConfigState | kIOServiceConfigState)))
&& (0 == (__state[0] & kIOServiceInactiveState));
__state[1] |= kIOServiceNeedConfigState;
if( needConfig) {
needWake = (0 != (kIOServiceSyncPubState & __state[1]));
}
if( sync)
__state[1] |= kIOServiceSynchronousState;
else
__state[1] &= ~kIOServiceSynchronousState;
if( needConfig) prevBusy = _adjustBusy( 1 );
unlockForArbitration();
if( needConfig) {
if( needWake) {
IOLockLock( gIOServiceBusyLock );
thread_wakeup( (event_t) this );
IOLockUnlock( gIOServiceBusyLock );
} else if( !sync || (kIOServiceAsynchronous & options)) {
ok = (0 != _IOServiceJob::startJob( this, kMatchNubJob, options ));
} else do {
if( (__state[1] & kIOServiceNeedConfigState))
doServiceMatch( options );
lockForArbitration();
IOLockLock( gIOServiceBusyLock );
waitAgain = ((prevBusy < (__state[1] & kIOServiceBusyStateMask))
&& (0 == (__state[0] & kIOServiceInactiveState)));
if( waitAgain)
__state[1] |= kIOServiceSyncPubState | kIOServiceBusyWaiterState;
else
__state[1] &= ~kIOServiceSyncPubState;
unlockForArbitration();
if( waitAgain)
assert_wait( (event_t) this, THREAD_UNINT);
IOLockUnlock( gIOServiceBusyLock );
if( waitAgain)
thread_block(THREAD_CONTINUE_NULL);
} while( waitAgain );
}
}
IOReturn IOService::catalogNewDrivers( OSOrderedSet * newTables )
{
OSDictionary * table;
OSSet * set;
OSSet * allSet = 0;
IOService * service;
#if IOMATCHDEBUG
SInt32 count = 0;
#endif
newTables->retain();
while( (table = (OSDictionary *) newTables->getFirstObject())) {
LOCKWRITENOTIFY();
set = (OSSet *) copyExistingServices( table,
kIOServiceRegisteredState,
kIOServiceExistingSet);
UNLOCKNOTIFY();
if( set) {
#if IOMATCHDEBUG
count += set->getCount();
#endif
if (allSet) {
allSet->merge((const OSSet *) set);
set->release();
}
else
allSet = set;
}
#if IOMATCHDEBUG
if( getDebugFlags( table ) & kIOLogMatch)
LOG("Matching service count = %ld\n", (long)count);
#endif
newTables->removeObject(table);
}
if (allSet) {
while( (service = (IOService *) allSet->getAnyObject())) {
service->startMatching(kIOServiceAsynchronous);
allSet->removeObject(service);
}
allSet->release();
}
newTables->release();
return( kIOReturnSuccess );
}
_IOServiceJob * _IOServiceJob::startJob( IOService * nub, int type,
IOOptionBits options )
{
_IOServiceJob * job;
job = new _IOServiceJob;
if( job && !job->init()) {
job->release();
job = 0;
}
if( job) {
job->type = type;
job->nub = nub;
job->options = options;
nub->retain(); pingConfig( job );
}
return( job );
}
bool IOService::matchPropertyTable( OSDictionary * table, SInt32 * score )
{
return( matchPropertyTable(table) );
}
bool IOService::matchPropertyTable( OSDictionary * table )
{
return( true );
}
IOReturn IOService::getResources( void )
{
return( kIOReturnSuccess);
}
IOService * IOService::getProvider( void ) const
{
IOService * self = (IOService *) this;
IOService * parent;
SInt32 generation;
generation = getGenerationCount();
if( __providerGeneration == generation)
return( __provider );
parent = (IOService *) getParentEntry( gIOServicePlane);
if( parent == IORegistryEntry::getRegistryRoot())
parent = 0;
self->__provider = parent;
OSMemoryBarrier();
self->__providerGeneration = generation;
return( parent );
}
IOWorkLoop * IOService::getWorkLoop() const
{
IOService *provider = getProvider();
if (provider)
return provider->getWorkLoop();
else
return 0;
}
OSIterator * IOService::getProviderIterator( void ) const
{
return( getParentIterator( gIOServicePlane));
}
IOService * IOService::getClient( void ) const
{
return( (IOService *) getChildEntry( gIOServicePlane));
}
OSIterator * IOService::getClientIterator( void ) const
{
return( getChildIterator( gIOServicePlane));
}
OSIterator * _IOOpenServiceIterator::iterator( OSIterator * _iter,
const IOService * client,
const IOService * provider )
{
_IOOpenServiceIterator * inst;
if( !_iter)
return( 0 );
inst = new _IOOpenServiceIterator;
if( inst && !inst->init()) {
inst->release();
inst = 0;
}
if( inst) {
inst->iter = _iter;
inst->client = client;
inst->provider = provider;
}
return( inst );
}
void _IOOpenServiceIterator::free()
{
iter->release();
if( last)
last->unlockForArbitration();
OSIterator::free();
}
OSObject * _IOOpenServiceIterator::getNextObject()
{
IOService * next;
if( last)
last->unlockForArbitration();
while( (next = (IOService *) iter->getNextObject())) {
next->lockForArbitration();
if( (client && (next->isOpen( client )))
|| (provider && (provider->isOpen( next ))) )
break;
next->unlockForArbitration();
}
last = next;
return( next );
}
bool _IOOpenServiceIterator::isValid()
{
return( iter->isValid() );
}
void _IOOpenServiceIterator::reset()
{
if( last) {
last->unlockForArbitration();
last = 0;
}
iter->reset();
}
OSIterator * IOService::getOpenProviderIterator( void ) const
{
return( _IOOpenServiceIterator::iterator( getProviderIterator(), this, 0 ));
}
OSIterator * IOService::getOpenClientIterator( void ) const
{
return( _IOOpenServiceIterator::iterator( getClientIterator(), 0, this ));
}
IOReturn IOService::callPlatformFunction( const OSSymbol * functionName,
bool waitForFunction,
void *param1, void *param2,
void *param3, void *param4 )
{
IOReturn result = kIOReturnUnsupported;
IOService *provider;
if (gIOPlatformFunctionHandlerSet == functionName)
{
#if defined(__i386__) || defined(__x86_64__)
const OSSymbol * functionHandlerName = (const OSSymbol *) param1;
IOService * target = (IOService *) param2;
bool enable = (param3 != 0);
if (sCPULatencyFunctionName[kCpuDelayBusStall] == functionHandlerName)
result = setLatencyHandler(kCpuDelayBusStall, target, enable);
else if (sCPULatencyFunctionName[kCpuDelayInterrupt] == param1)
result = setLatencyHandler(kCpuDelayInterrupt, target, enable);
#endif
}
if ((kIOReturnUnsupported == result) && (provider = getProvider())) {
result = provider->callPlatformFunction(functionName, waitForFunction,
param1, param2, param3, param4);
}
return result;
}
IOReturn IOService::callPlatformFunction( const char * functionName,
bool waitForFunction,
void *param1, void *param2,
void *param3, void *param4 )
{
IOReturn result = kIOReturnNoMemory;
const OSSymbol *functionSymbol = OSSymbol::withCString(functionName);
if (functionSymbol != 0) {
result = callPlatformFunction(functionSymbol, waitForFunction,
param1, param2, param3, param4);
functionSymbol->release();
}
return result;
}
IOPlatformExpert * IOService::getPlatform( void )
{
return( gIOPlatform);
}
class IOPMrootDomain * IOService::getPMRootDomain( void )
{
return( gIOPMRootDomain);
}
IOService * IOService::getResourceService( void )
{
return( gIOResources );
}
void IOService::setPlatform( IOPlatformExpert * platform)
{
gIOPlatform = platform;
gIOResources->attachToParent( gIOServiceRoot, gIOServicePlane );
#if defined(__i386__) || defined(__x86_64__)
static const char * keys[kCpuNumDelayTypes] = {
kIOPlatformMaxBusDelay, kIOPlatformMaxInterruptDelay };
const OSObject * objs[2];
OSArray * array;
uint32_t idx;
for (idx = 0; idx < kCpuNumDelayTypes; idx++)
{
objs[0] = sCPULatencySet[idx];
objs[1] = sCPULatencyHolder[idx];
array = OSArray::withObjects(objs, 2);
if (!array) break;
platform->setProperty(keys[idx], array);
array->release();
}
#endif
}
void IOService::setPMRootDomain( class IOPMrootDomain * rootDomain)
{
gIOPMRootDomain = rootDomain;
publishResource("IOKit");
}
bool IOService::lockForArbitration( bool isSuccessRequired )
{
bool found;
bool success;
ArbitrationLockQueueElement * element;
ArbitrationLockQueueElement * active;
ArbitrationLockQueueElement * waiting;
enum { kPutOnFreeQueue, kPutOnActiveQueue, kPutOnWaitingQueue } action;
IOTakeLock( gArbitrationLockQueueLock );
if( !queue_empty( &gArbitrationLockQueueFree )) {
queue_remove_first( &gArbitrationLockQueueFree,
element,
ArbitrationLockQueueElement *,
link );
} else {
element = IONew( ArbitrationLockQueueElement, 1 );
assert( element );
}
element->thread = IOThreadSelf();
element->service = this;
element->count = 1;
element->required = isSuccessRequired;
element->aborted = false;
found = false;
queue_iterate( &gArbitrationLockQueueActive,
active,
ArbitrationLockQueueElement *,
link )
{
if( active->service == element->service ) {
found = true;
break;
}
}
if( found ) {
if( active->thread != element->thread ) {
ArbitrationLockQueueElement * victim = 0;
while( 1 ) {
found = false;
queue_iterate( &gArbitrationLockQueueWaiting,
waiting,
ArbitrationLockQueueElement *,
link )
{
if( waiting->thread == active->thread ) {
assert( false == waiting->aborted );
found = true;
break;
}
}
if( found ) {
if( false == waiting->required )
victim = waiting;
found = false;
queue_iterate( &gArbitrationLockQueueActive,
active, ArbitrationLockQueueElement *,
link )
{
if( active->service == waiting->service ) {
found = true;
break;
}
}
assert( found );
if( active->thread == element->thread ) {
if( false == element->required ) {
success = false; break;
} else {
if( victim ) {
victim->aborted = true;
queue_remove( &gArbitrationLockQueueWaiting,
victim,
ArbitrationLockQueueElement *,
link );
IOLockWakeup( gArbitrationLockQueueLock,
victim,
true );
success = true; break; } else {
panic("I/O Kit: Unrecoverable deadlock.");
}
}
} else {
}
} else {
success = true; break; }
}
if( success ) { kern_return_t wait_result;
queue_enter( &gArbitrationLockQueueWaiting,
element,
ArbitrationLockQueueElement *,
link );
restart_sleep: wait_result = assert_wait( element,
element->required ? THREAD_UNINT
: THREAD_INTERRUPTIBLE );
IOUnlock( gArbitrationLockQueueLock );
if (wait_result == THREAD_WAITING)
wait_result = thread_block(THREAD_CONTINUE_NULL);
if( THREAD_INTERRUPTED == wait_result ) {
IOTakeLock( gArbitrationLockQueueLock );
found = false;
queue_iterate( &gArbitrationLockQueueWaiting,
waiting, ArbitrationLockQueueElement *,
link )
{
if( waiting == element ) {
found = true;
break;
}
}
if( found ) {
if( false == element->required ) {
element->aborted = true;
queue_remove( &gArbitrationLockQueueWaiting,
element,
ArbitrationLockQueueElement *,
link );
} else {
goto restart_sleep;
}
}
IOUnlock( gArbitrationLockQueueLock );
wait_result = THREAD_AWAKENED;
}
assert( THREAD_AWAKENED == wait_result );
if( element->aborted ) {
assert( false == element->required );
IOTakeLock( gArbitrationLockQueueLock );
action = kPutOnFreeQueue;
success = false;
} else {
return true;
}
} else {
action = kPutOnFreeQueue;
}
} else {
active->count++;
action = kPutOnFreeQueue;
success = true;
}
} else { action = kPutOnActiveQueue;
success = true;
}
if( kPutOnActiveQueue == action ) {
queue_enter( &gArbitrationLockQueueActive,
element,
ArbitrationLockQueueElement *,
link );
} else if( kPutOnFreeQueue == action ) {
queue_enter( &gArbitrationLockQueueFree,
element,
ArbitrationLockQueueElement *,
link );
} else {
assert( 0 ); }
IOUnlock( gArbitrationLockQueueLock );
return( success );
}
void IOService::unlockForArbitration( void )
{
bool found;
ArbitrationLockQueueElement * element;
IOTakeLock( gArbitrationLockQueueLock );
found = false;
queue_iterate( &gArbitrationLockQueueActive,
element,
ArbitrationLockQueueElement *,
link )
{
if( element->service == this ) {
found = true;
break;
}
}
assert( found );
if( element->count > 1 ) {
element->count--;
} else {
queue_remove( &gArbitrationLockQueueActive,
element,
ArbitrationLockQueueElement *,
link );
queue_enter( &gArbitrationLockQueueFree,
element,
ArbitrationLockQueueElement *,
link );
found = false;
queue_iterate( &gArbitrationLockQueueWaiting,
element,
ArbitrationLockQueueElement *,
link )
{
if( element->service == this ) {
found = true;
break;
}
}
if ( found ) {
queue_remove( &gArbitrationLockQueueWaiting,
element,
ArbitrationLockQueueElement *,
link );
queue_enter( &gArbitrationLockQueueActive,
element,
ArbitrationLockQueueElement *,
link );
IOLockWakeup( gArbitrationLockQueueLock,
element,
true );
}
}
IOUnlock( gArbitrationLockQueueLock );
}
void IOService::applyToProviders( IOServiceApplierFunction applier,
void * context )
{
applyToParents( (IORegistryEntryApplierFunction) applier,
context, gIOServicePlane );
}
void IOService::applyToClients( IOServiceApplierFunction applier,
void * context )
{
applyToChildren( (IORegistryEntryApplierFunction) applier,
context, gIOServicePlane );
}
IOReturn IOService::messageClient( UInt32 type, OSObject * client,
void * argument, vm_size_t argSize )
{
IOReturn ret;
IOService * service;
_IOServiceInterestNotifier * notify;
if( (service = OSDynamicCast( IOService, client)))
ret = service->message( type, this, argument );
else if( (notify = OSDynamicCast( _IOServiceInterestNotifier, client))) {
_IOServiceNotifierInvocation invocation;
bool willNotify;
invocation.thread = current_thread();
LOCKWRITENOTIFY();
willNotify = (0 != (kIOServiceNotifyEnable & notify->state));
if( willNotify) {
queue_enter( ¬ify->handlerInvocations, &invocation,
_IOServiceNotifierInvocation *, link );
}
UNLOCKNOTIFY();
if( willNotify) {
ret = (*notify->handler)( notify->target, notify->ref,
type, this, argument, argSize );
LOCKWRITENOTIFY();
queue_remove( ¬ify->handlerInvocations, &invocation,
_IOServiceNotifierInvocation *, link );
if( kIOServiceNotifyWaiter & notify->state) {
notify->state &= ~kIOServiceNotifyWaiter;
WAKEUPNOTIFY( notify );
}
UNLOCKNOTIFY();
} else
ret = kIOReturnSuccess;
} else
ret = kIOReturnBadArgument;
return( ret );
}
static void
applyToInterestNotifiers(const IORegistryEntry *target,
const OSSymbol * typeOfInterest,
OSObjectApplierFunction applier,
void * context )
{
OSArray * copyArray = 0;
LOCKREADNOTIFY();
IOCommand *notifyList =
OSDynamicCast( IOCommand, target->getProperty( typeOfInterest ));
if( notifyList) {
copyArray = OSArray::withCapacity(1);
iterqueue(¬ifyList->fCommandChain, entry) {
_IOServiceInterestNotifier * notify;
queue_element(entry, notify, _IOServiceInterestNotifier *, chain);
copyArray->setObject(notify);
}
}
UNLOCKNOTIFY();
if( copyArray) {
unsigned int index;
OSObject * next;
for( index = 0; (next = copyArray->getObject( index )); index++)
(*applier)(next, context);
copyArray->release();
}
}
void IOService::applyToInterested( const OSSymbol * typeOfInterest,
OSObjectApplierFunction applier,
void * context )
{
if (gIOGeneralInterest == typeOfInterest)
applyToClients( (IOServiceApplierFunction) applier, context );
applyToInterestNotifiers(this, typeOfInterest, applier, context);
}
struct MessageClientsContext {
IOService * service;
UInt32 type;
void * argument;
vm_size_t argSize;
IOReturn ret;
};
static void messageClientsApplier( OSObject * object, void * ctx )
{
IOReturn ret;
MessageClientsContext * context = (MessageClientsContext *) ctx;
ret = context->service->messageClient( context->type,
object, context->argument, context->argSize );
if( kIOReturnSuccess != ret)
context->ret = ret;
}
IOReturn IOService::messageClients( UInt32 type,
void * argument, vm_size_t argSize )
{
MessageClientsContext context;
context.service = this;
context.type = type;
context.argument = argument;
context.argSize = argSize;
context.ret = kIOReturnSuccess;
applyToInterested( gIOGeneralInterest,
&messageClientsApplier, &context );
return( context.ret );
}
IOReturn IOService::acknowledgeNotification( IONotificationRef notification,
IOOptionBits response )
{
return( kIOReturnUnsupported );
}
IONotifier * IOService::registerInterest( const OSSymbol * typeOfInterest,
IOServiceInterestHandler handler, void * target, void * ref )
{
_IOServiceInterestNotifier * notify = 0;
IOReturn rc = kIOReturnError;
notify = new _IOServiceInterestNotifier;
if (!notify) return NULL;
if(notify->init()) {
rc = registerInterestForNotifer(notify, typeOfInterest,
handler, target, ref);
}
if (rc != kIOReturnSuccess) {
notify->release();
notify = 0;
}
return( notify );
}
IOReturn IOService::registerInterestForNotifer( IONotifier *svcNotify, const OSSymbol * typeOfInterest,
IOServiceInterestHandler handler, void * target, void * ref )
{
IOReturn rc = kIOReturnSuccess;
_IOServiceInterestNotifier *notify = 0;
if( (typeOfInterest != gIOGeneralInterest)
&& (typeOfInterest != gIOBusyInterest)
&& (typeOfInterest != gIOAppPowerStateInterest)
&& (typeOfInterest != gIOConsoleSecurityInterest)
&& (typeOfInterest != gIOPriorityPowerStateInterest))
return( kIOReturnBadArgument );
if (!svcNotify || !(notify = OSDynamicCast(_IOServiceInterestNotifier, svcNotify)))
return( kIOReturnBadArgument );
lockForArbitration();
if( 0 == (__state[0] & kIOServiceInactiveState)) {
notify->handler = handler;
notify->target = target;
notify->ref = ref;
notify->state = kIOServiceNotifyEnable;
LOCKWRITENOTIFY();
IOCommand * notifyList;
OSObject * obj = copyProperty( typeOfInterest );
if (!(notifyList = OSDynamicCast(IOCommand, obj))) {
notifyList = OSTypeAlloc(IOCommand);
if (notifyList) {
notifyList->init();
bool ok = setProperty( typeOfInterest, notifyList);
notifyList->release();
if (!ok) notifyList = 0;
}
}
if (obj) obj->release();
if (notifyList) {
enqueue(¬ifyList->fCommandChain, ¬ify->chain);
notify->retain(); }
UNLOCKNOTIFY();
}
else {
rc = kIOReturnNotReady;
}
unlockForArbitration();
return rc;
}
static void cleanInterestList( OSObject * head )
{
IOCommand *notifyHead = OSDynamicCast(IOCommand, head);
if (!notifyHead)
return;
LOCKWRITENOTIFY();
while ( queue_entry_t entry = dequeue(¬ifyHead->fCommandChain) ) {
queue_next(entry) = queue_prev(entry) = 0;
_IOServiceInterestNotifier * notify;
queue_element(entry, notify, _IOServiceInterestNotifier *, chain);
notify->release();
}
UNLOCKNOTIFY();
}
void IOService::unregisterAllInterest( void )
{
cleanInterestList( getProperty( gIOGeneralInterest ));
cleanInterestList( getProperty( gIOBusyInterest ));
cleanInterestList( getProperty( gIOAppPowerStateInterest ));
cleanInterestList( getProperty( gIOPriorityPowerStateInterest ));
cleanInterestList( getProperty( gIOConsoleSecurityInterest ));
}
void _IOServiceInterestNotifier::wait()
{
_IOServiceNotifierInvocation * next;
bool doWait;
do {
doWait = false;
queue_iterate( &handlerInvocations, next,
_IOServiceNotifierInvocation *, link) {
if( next->thread != current_thread() ) {
doWait = true;
break;
}
}
if( doWait) {
state |= kIOServiceNotifyWaiter;
SLEEPNOTIFY(this);
}
} while( doWait );
}
void _IOServiceInterestNotifier::free()
{
assert( queue_empty( &handlerInvocations ));
OSObject::free();
}
void _IOServiceInterestNotifier::remove()
{
LOCKWRITENOTIFY();
if( queue_next( &chain )) {
remqueue(&chain);
queue_next( &chain) = queue_prev( &chain) = 0;
release();
}
state &= ~kIOServiceNotifyEnable;
wait();
UNLOCKNOTIFY();
release();
}
bool _IOServiceInterestNotifier::disable()
{
bool ret;
LOCKWRITENOTIFY();
ret = (0 != (kIOServiceNotifyEnable & state));
state &= ~kIOServiceNotifyEnable;
if( ret)
wait();
UNLOCKNOTIFY();
return( ret );
}
void _IOServiceInterestNotifier::enable( bool was )
{
LOCKWRITENOTIFY();
if( was)
state |= kIOServiceNotifyEnable;
else
state &= ~kIOServiceNotifyEnable;
UNLOCKNOTIFY();
}
bool _IOServiceInterestNotifier::init()
{
queue_init( &handlerInvocations );
return (OSObject::init());
}
#define tailQ(o) setObject(o)
#define headQ(o) setObject(0, o)
#define TLOG(fmt, args...) { if(kIOLogYield & gIOKitDebug) { IOLog("[%llx] ", thread_tid(current_thread())); IOLog(fmt, ## args); }}
static void _workLoopAction( IOWorkLoop::Action action,
IOService * service,
void * p0 = 0, void * p1 = 0,
void * p2 = 0, void * p3 = 0 )
{
IOWorkLoop * wl;
if( (wl = service->getWorkLoop())) {
wl->retain();
wl->runAction( action, service, p0, p1, p2, p3 );
wl->release();
} else
(*action)( service, p0, p1, p2, p3 );
}
bool IOService::requestTerminate( IOService * provider, IOOptionBits options )
{
bool ok;
ok = isParent( provider, gIOServicePlane, true);
if( ok) {
provider->terminateClient( this, options | kIOServiceRecursing );
ok = (0 != (kIOServiceInactiveState & __state[0]));
}
return( ok );
}
bool IOService::terminatePhase1( IOOptionBits options )
{
IOService * victim;
IOService * client;
OSIterator * iter;
OSArray * makeInactive;
OSArray * waitingInactive;
int waitResult = THREAD_AWAKENED;
bool wait;
bool ok;
bool didInactive;
bool startPhase2 = false;
TLOG("%s[0x%qx]::terminatePhase1(%08llx)\n", getName(), getRegistryEntryID(), (long long)options);
uint64_t regID = getRegistryEntryID();
IOServiceTrace(
IOSERVICE_TERMINATE_PHASE1,
(uintptr_t) regID,
(uintptr_t) (regID >> 32),
(uintptr_t) this,
(uintptr_t) options);
if( options & kIOServiceRecursing) {
lockForArbitration();
if (0 == (kIOServiceInactiveState & __state[0]))
{
__state[0] |= kIOServiceInactiveState;
__state[1] |= kIOServiceRecursing | kIOServiceTermPhase1State;
}
unlockForArbitration();
return( true );
}
makeInactive = OSArray::withCapacity( 16 );
waitingInactive = OSArray::withCapacity( 16 );
if(!makeInactive || !waitingInactive) return( false );
victim = this;
victim->retain();
while( victim )
{
didInactive = victim->lockForArbitration( true );
if( didInactive)
{
uint64_t regID1 = victim->getRegistryEntryID();
IOServiceTrace(IOSERVICE_TERM_SET_INACTIVE,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) victim->__state[1],
(uintptr_t) 0);
enum { kRP1 = kIOServiceRecursing | kIOServiceTermPhase1State };
didInactive = (kRP1 == (victim->__state[1] & kRP1))
|| (0 == (victim->__state[0] & kIOServiceInactiveState));
if (!didInactive)
{
if (-1U == waitingInactive->getNextIndexOfObject(victim, 0)) do
{
IOLockLock(gIOServiceBusyLock);
wait = (victim->__state[1] & kIOServiceTermPhase1State);
if( wait) {
TLOG("%s[0x%qx]::waitPhase1(%s[0x%qx])\n",
getName(), getRegistryEntryID(), victim->getName(), victim->getRegistryEntryID());
victim->__state[1] |= kIOServiceTerm1WaiterState;
victim->unlockForArbitration();
assert_wait((event_t)&victim->__state[1], THREAD_UNINT);
}
IOLockUnlock(gIOServiceBusyLock);
if( wait) {
waitResult = thread_block(THREAD_CONTINUE_NULL);
TLOG("%s[0x%qx]::did waitPhase1(%s[0x%qx])\n",
getName(), getRegistryEntryID(), victim->getName(), victim->getRegistryEntryID());
victim->lockForArbitration();
}
}
while (wait && (waitResult != THREAD_TIMED_OUT));
}
else
{
victim->__state[0] |= kIOServiceInactiveState;
victim->__state[0] &= ~(kIOServiceRegisteredState | kIOServiceMatchedState
| kIOServiceFirstPublishState | kIOServiceFirstMatchState);
victim->__state[1] &= ~kIOServiceRecursing;
victim->__state[1] |= kIOServiceTermPhase1State;
waitingInactive->headQ(victim);
if (victim == this)
{
if (kIOServiceTerminateNeedWillTerminate & options)
{
victim->__state[1] |= kIOServiceNeedWillTerminate;
}
}
victim->_adjustBusy( 1 );
}
victim->unlockForArbitration();
}
if( victim == this) startPhase2 = didInactive;
if (didInactive)
{
victim->deliverNotification( gIOTerminatedNotification, 0, 0xffffffff );
IOUserClient::destroyUserReferences( victim );
iter = victim->getClientIterator();
if( iter) {
while( (client = (IOService *) iter->getNextObject())) {
TLOG("%s[0x%qx]::requestTerminate(%s[0x%qx], %08llx)\n",
client->getName(), client->getRegistryEntryID(),
victim->getName(), victim->getRegistryEntryID(), (long long)options);
ok = client->requestTerminate( victim, options );
TLOG("%s[0x%qx]::requestTerminate(%s[0x%qx], ok = %d)\n",
client->getName(), client->getRegistryEntryID(),
victim->getName(), victim->getRegistryEntryID(), ok);
uint64_t regID1 = client->getRegistryEntryID();
uint64_t regID2 = victim->getRegistryEntryID();
IOServiceTrace(
(ok ? IOSERVICE_TERMINATE_REQUEST_OK
: IOSERVICE_TERMINATE_REQUEST_FAIL),
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32));
if( ok)
makeInactive->setObject( client );
}
iter->release();
}
}
victim->release();
victim = (IOService *) makeInactive->getObject(0);
if( victim) {
victim->retain();
makeInactive->removeObject(0);
}
}
makeInactive->release();
while ((victim = (IOService *) waitingInactive->getObject(0)))
{
victim->retain();
waitingInactive->removeObject(0);
victim->lockForArbitration();
victim->__state[1] &= ~kIOServiceTermPhase1State;
if (kIOServiceTerm1WaiterState & victim->__state[1])
{
victim->__state[1] &= ~kIOServiceTerm1WaiterState;
TLOG("%s[0x%qx]::wakePhase1\n", victim->getName(), victim->getRegistryEntryID());
IOLockLock( gIOServiceBusyLock );
thread_wakeup( (event_t) &victim->__state[1]);
IOLockUnlock( gIOServiceBusyLock );
}
victim->unlockForArbitration();
victim->release();
}
waitingInactive->release();
if( startPhase2)
{
retain();
lockForArbitration();
scheduleTerminatePhase2(options);
unlockForArbitration();
release();
}
return( true );
}
void IOService::setTerminateDefer(IOService * provider, bool defer)
{
lockForArbitration();
if (defer) __state[1] |= kIOServiceStartState;
else __state[1] &= ~kIOServiceStartState;
unlockForArbitration();
if (provider && !defer)
{
provider->lockForArbitration();
provider->scheduleTerminatePhase2();
provider->unlockForArbitration();
}
}
void IOService::scheduleTerminatePhase2( IOOptionBits options )
{
AbsoluteTime deadline;
uint64_t regID1;
int waitResult = THREAD_AWAKENED;
bool wait, haveDeadline = false;
if (!(__state[0] & kIOServiceInactiveState)) return;
regID1 = getRegistryEntryID();
IOServiceTrace(
IOSERVICE_TERM_SCHED_PHASE2,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) __state[1],
(uintptr_t) options);
if (__state[1] & kIOServiceTermPhase1State) return;
retain();
unlockForArbitration();
options |= kIOServiceRequired;
IOLockLock( gJobsLock );
if( (options & kIOServiceSynchronous)
&& (current_thread() != gIOTerminateThread)) {
do {
wait = (gIOTerminateThread != 0);
if( wait) {
IOLockSleep( gJobsLock, &gIOTerminateThread, THREAD_UNINT);
}
} while( wait );
gIOTerminateThread = current_thread();
gIOTerminatePhase2List->setObject( this );
gIOTerminateWork++;
do {
while( gIOTerminateWork )
terminateWorker( options );
wait = (0 != (__state[1] & kIOServiceBusyStateMask));
if( wait) {
if( !haveDeadline) {
clock_interval_to_deadline( 15, kSecondScale, &deadline );
haveDeadline = true;
}
waitResult = IOLockSleepDeadline( gJobsLock, &gIOTerminateWork,
deadline, THREAD_UNINT );
if( waitResult == THREAD_TIMED_OUT) {
IOLog("%s[0x%qx]::terminate(kIOServiceSynchronous) timeout\n", getName(), getRegistryEntryID());
}
}
} while(gIOTerminateWork || (wait && (waitResult != THREAD_TIMED_OUT)));
gIOTerminateThread = 0;
IOLockWakeup( gJobsLock, (event_t) &gIOTerminateThread, false);
} else {
gIOTerminatePhase2List->setObject( this );
if( 0 == gIOTerminateWork++) {
if( !gIOTerminateThread)
kernel_thread_start(&terminateThread, (void *)(uintptr_t) options, &gIOTerminateThread);
else
IOLockWakeup(gJobsLock, (event_t) &gIOTerminateWork, false );
}
}
IOLockUnlock( gJobsLock );
lockForArbitration();
release();
}
void IOService::terminateThread( void * arg, wait_result_t waitResult )
{
IOLockLock( gJobsLock );
while (gIOTerminateWork)
terminateWorker( (uintptr_t) arg );
thread_deallocate(gIOTerminateThread);
gIOTerminateThread = 0;
IOLockWakeup( gJobsLock, (event_t) &gIOTerminateThread, false);
IOLockUnlock( gJobsLock );
}
void IOService::scheduleStop( IOService * provider )
{
uint64_t regID1 = getRegistryEntryID();
uint64_t regID2 = provider->getRegistryEntryID();
TLOG("%s[0x%qx]::scheduleStop(%s[0x%qx])\n", getName(), regID1, provider->getName(), regID2);
IOServiceTrace(
IOSERVICE_TERMINATE_SCHEDULE_STOP,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32));
IOLockLock( gJobsLock );
gIOStopList->tailQ( this );
gIOStopProviderList->tailQ( provider );
if( 0 == gIOTerminateWork++) {
if( !gIOTerminateThread)
kernel_thread_start(&terminateThread, (void *) 0, &gIOTerminateThread);
else
IOLockWakeup(gJobsLock, (event_t) &gIOTerminateWork, false );
}
IOLockUnlock( gJobsLock );
}
void IOService::scheduleFinalize( void )
{
uint64_t regID1 = getRegistryEntryID();
TLOG("%s[0x%qx]::scheduleFinalize\n", getName(), regID1);
IOServiceTrace(
IOSERVICE_TERMINATE_SCHEDULE_FINALIZE,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
0, 0);
IOLockLock( gJobsLock );
gIOFinalizeList->tailQ( this );
if( 0 == gIOTerminateWork++) {
if( !gIOTerminateThread)
kernel_thread_start(&terminateThread, (void *) 0, &gIOTerminateThread);
else
IOLockWakeup(gJobsLock, (event_t) &gIOTerminateWork, false );
}
IOLockUnlock( gJobsLock );
}
bool IOService::willTerminate( IOService * provider, IOOptionBits options )
{
return( true );
}
bool IOService::didTerminate( IOService * provider, IOOptionBits options, bool * defer )
{
if( false == *defer) {
if( lockForArbitration( true )) {
if( false == provider->handleIsOpen( this ))
scheduleStop( provider );
else {
message( kIOMessageServiceIsRequestingClose, provider, (void *)(uintptr_t) options );
if( false == provider->handleIsOpen( this ))
scheduleStop( provider );
}
unlockForArbitration();
}
}
return( true );
}
void IOService::actionWillTerminate( IOService * victim, IOOptionBits options,
OSArray * doPhase2List,
void *unused2 __unused,
void *unused3 __unused )
{
OSIterator * iter;
IOService * client;
bool ok;
uint64_t regID1, regID2 = victim->getRegistryEntryID();
iter = victim->getClientIterator();
if( iter) {
while( (client = (IOService *) iter->getNextObject())) {
regID1 = client->getRegistryEntryID();
TLOG("%s[0x%qx]::willTerminate(%s[0x%qx], %08llx)\n",
client->getName(), regID1,
victim->getName(), regID2, (long long)options);
IOServiceTrace(
IOSERVICE_TERMINATE_WILL,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32));
ok = client->willTerminate( victim, options );
doPhase2List->tailQ( client );
}
iter->release();
}
}
void IOService::actionDidTerminate( IOService * victim, IOOptionBits options,
void *unused1 __unused, void *unused2 __unused,
void *unused3 __unused )
{
OSIterator * iter;
IOService * client;
bool defer;
uint64_t regID1, regID2 = victim->getRegistryEntryID();
victim->messageClients( kIOMessageServiceIsTerminated, (void *)(uintptr_t) options );
iter = victim->getClientIterator();
if( iter) {
while( (client = (IOService *) iter->getNextObject())) {
regID1 = client->getRegistryEntryID();
TLOG("%s[0x%qx]::didTerminate(%s[0x%qx], %08llx)\n",
client->getName(), regID1,
victim->getName(), regID2, (long long)options);
defer = false;
client->didTerminate( victim, options, &defer );
IOServiceTrace(
(defer ? IOSERVICE_TERMINATE_DID_DEFER
: IOSERVICE_TERMINATE_DID),
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32));
TLOG("%s[0x%qx]::didTerminate(%s[0x%qx], defer %d)\n",
client->getName(), regID1,
victim->getName(), regID2, defer);
}
iter->release();
}
}
void IOService::actionWillStop( IOService * victim, IOOptionBits options,
void *unused1 __unused, void *unused2 __unused,
void *unused3 __unused )
{
OSIterator * iter;
IOService * provider;
bool ok;
uint64_t regID1, regID2 = victim->getRegistryEntryID();
iter = victim->getProviderIterator();
if( iter) {
while( (provider = (IOService *) iter->getNextObject())) {
regID1 = provider->getRegistryEntryID();
TLOG("%s[0x%qx]::willTerminate(%s[0x%qx], %08llx)\n",
victim->getName(), regID2,
provider->getName(), regID1, (long long)options);
IOServiceTrace(
IOSERVICE_TERMINATE_WILL,
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32),
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32));
ok = victim->willTerminate( provider, options );
}
iter->release();
}
}
void IOService::actionDidStop( IOService * victim, IOOptionBits options,
void *unused1 __unused, void *unused2 __unused,
void *unused3 __unused )
{
OSIterator * iter;
IOService * provider;
bool defer = false;
uint64_t regID1, regID2 = victim->getRegistryEntryID();
iter = victim->getProviderIterator();
if( iter) {
while( (provider = (IOService *) iter->getNextObject())) {
regID1 = provider->getRegistryEntryID();
TLOG("%s[0x%qx]::didTerminate(%s[0x%qx], %08llx)\n",
victim->getName(), regID2,
provider->getName(), regID1, (long long)options);
victim->didTerminate( provider, options, &defer );
IOServiceTrace(
(defer ? IOSERVICE_TERMINATE_DID_DEFER
: IOSERVICE_TERMINATE_DID),
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32),
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32));
TLOG("%s[0x%qx]::didTerminate(%s[0x%qx], defer %d)\n",
victim->getName(), regID2,
provider->getName(), regID1, defer);
}
iter->release();
}
}
void IOService::actionFinalize( IOService * victim, IOOptionBits options,
void *unused1 __unused, void *unused2 __unused,
void *unused3 __unused )
{
uint64_t regID1 = victim->getRegistryEntryID();
TLOG("%s[0x%qx]::finalize(%08llx)\n", victim->getName(), regID1, (long long)options);
IOServiceTrace(
IOSERVICE_TERMINATE_FINALIZE,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
0, 0);
victim->finalize( options );
}
void IOService::actionStop( IOService * provider, IOService * client,
void *unused1 __unused, void *unused2 __unused,
void *unused3 __unused )
{
uint64_t regID1 = provider->getRegistryEntryID();
uint64_t regID2 = client->getRegistryEntryID();
TLOG("%s[0x%qx]::stop(%s[0x%qx])\n", client->getName(), regID2, provider->getName(), regID1);
IOServiceTrace(
IOSERVICE_TERMINATE_STOP,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32));
client->stop( provider );
if( provider->isOpen( client ))
provider->close( client );
TLOG("%s[0x%qx]::detach(%s[0x%qx])\n", client->getName(), regID2, provider->getName(), regID1);
client->detach( provider );
}
void IOService::terminateWorker( IOOptionBits options )
{
OSArray * doPhase2List;
OSArray * didPhase2List;
OSSet * freeList;
OSIterator * iter;
UInt32 workDone;
IOService * victim;
IOService * client;
IOService * provider;
unsigned int idx;
bool moreToDo;
bool doPhase2;
bool doPhase3;
options |= kIOServiceRequired;
doPhase2List = OSArray::withCapacity( 16 );
didPhase2List = OSArray::withCapacity( 16 );
freeList = OSSet::withCapacity( 16 );
if( (0 == doPhase2List) || (0 == didPhase2List) || (0 == freeList))
return;
do {
workDone = gIOTerminateWork;
while( (victim = (IOService *) gIOTerminatePhase2List->getObject(0) )) {
victim->retain();
gIOTerminatePhase2List->removeObject(0);
IOLockUnlock( gJobsLock );
uint64_t regID1 = victim->getRegistryEntryID();
IOServiceTrace(
IOSERVICE_TERM_START_PHASE2,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) 0,
(uintptr_t) 0);
while( victim ) {
doPhase2 = victim->lockForArbitration( true );
if( doPhase2) {
doPhase2 = (0 != (kIOServiceInactiveState & victim->__state[0]));
if( doPhase2) {
uint64_t regID1 = victim->getRegistryEntryID();
IOServiceTrace(
IOSERVICE_TERM_TRY_PHASE2,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) victim->__state[1],
(uintptr_t) 0);
doPhase2 = (0 == (victim->__state[1] & kIOServiceTermPhase2State))
&& (0 == (victim->__state[1] & kIOServiceConfigState));
if (doPhase2 && (iter = victim->getClientIterator())) {
while (doPhase2 && (client = (IOService *) iter->getNextObject())) {
doPhase2 = (0 == (client->__state[1] & kIOServiceStartState));
if (!doPhase2)
{
uint64_t regID1 = client->getRegistryEntryID();
IOServiceTrace(
IOSERVICE_TERM_UC_DEFER,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) client->__state[1],
(uintptr_t) 0);
TLOG("%s[0x%qx]::defer phase2(%s[0x%qx])\n",
victim->getName(), victim->getRegistryEntryID(),
client->getName(), client->getRegistryEntryID());
}
}
iter->release();
}
if( doPhase2)
victim->__state[1] |= kIOServiceTermPhase2State;
}
victim->unlockForArbitration();
}
if( doPhase2) {
if (kIOServiceNeedWillTerminate & victim->__state[1]) {
_workLoopAction( (IOWorkLoop::Action) &actionWillStop,
victim, (void *)(uintptr_t) options, NULL );
}
if( 0 == victim->getClient()) {
IOLockLock( gJobsLock );
gIOFinalizeList->tailQ( victim );
IOLockUnlock( gJobsLock );
} else {
_workLoopAction( (IOWorkLoop::Action) &actionWillTerminate,
victim, (void *)(uintptr_t) options, (void *)(uintptr_t) doPhase2List );
}
didPhase2List->headQ( victim );
}
victim->release();
victim = (IOService *) doPhase2List->getObject(0);
if( victim) {
victim->retain();
doPhase2List->removeObject(0);
}
}
while( (victim = (IOService *) didPhase2List->getObject(0)) ) {
if( victim->lockForArbitration( true )) {
victim->__state[1] |= kIOServiceTermPhase3State;
victim->unlockForArbitration();
}
_workLoopAction( (IOWorkLoop::Action) &actionDidTerminate,
victim, (void *)(uintptr_t) options );
if (kIOServiceNeedWillTerminate & victim->__state[1]) {
_workLoopAction( (IOWorkLoop::Action) &actionDidStop,
victim, (void *)(uintptr_t) options, NULL );
}
didPhase2List->removeObject(0);
}
IOLockLock( gJobsLock );
}
do {
doPhase3 = false;
while( (victim = (IOService *) gIOFinalizeList->getObject(0))) {
IOLockUnlock( gJobsLock );
_workLoopAction( (IOWorkLoop::Action) &actionFinalize,
victim, (void *)(uintptr_t) options );
IOLockLock( gJobsLock );
freeList->setObject( victim );
gIOFinalizeList->removeObject(0);
}
for( idx = 0;
(!doPhase3) && (client = (IOService *) gIOStopList->getObject(idx)); ) {
provider = (IOService *) gIOStopProviderList->getObject(idx);
assert( provider );
uint64_t regID1 = provider->getRegistryEntryID();
uint64_t regID2 = client->getRegistryEntryID();
if( !provider->isChild( client, gIOServicePlane )) {
TLOG("%s[0x%qx]::nop stop(%s[0x%qx])\n", client->getName(), regID2, provider->getName(), regID1);
IOServiceTrace(
IOSERVICE_TERMINATE_STOP_NOP,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32));
} else {
if( (kIOServiceInactiveState & client->__state[0]) && client->getClient()) {
TLOG("%s[0x%qx]::defer stop(%s[0x%qx])\n",
client->getName(), regID2,
client->getClient()->getName(), client->getClient()->getRegistryEntryID());
IOServiceTrace(
IOSERVICE_TERMINATE_STOP_DEFER,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32));
idx++;
continue;
}
IOLockUnlock( gJobsLock );
_workLoopAction( (IOWorkLoop::Action) &actionStop,
provider, (void *) client );
IOLockLock( gJobsLock );
doPhase3 = true;
}
freeList->setObject( client );
freeList->setObject( provider );
gIOStopList->removeObject( idx );
gIOStopProviderList->removeObject( idx );
idx = 0;
}
} while( doPhase3 );
gIOTerminateWork -= workDone;
moreToDo = (gIOTerminateWork != 0);
if( !moreToDo) {
TLOG("iokit terminate done, %d stops remain\n", gIOStopList->getCount());
IOServiceTrace(
IOSERVICE_TERMINATE_DONE,
(uintptr_t) gIOStopList->getCount(), 0, 0, 0);
}
} while( moreToDo );
IOLockUnlock( gJobsLock );
freeList->release();
doPhase2List->release();
didPhase2List->release();
IOLockLock( gJobsLock );
}
bool IOService::finalize( IOOptionBits options )
{
OSIterator * iter;
IOService * provider;
uint64_t regID1, regID2 = getRegistryEntryID();
iter = getProviderIterator();
assert( iter );
if( iter) {
while( (provider = (IOService *) iter->getNextObject())) {
if( 0 == (__state[1] & kIOServiceTermPhase3State)) {
regID1 = provider->getRegistryEntryID();
TLOG("%s[0x%qx]::stop1(%s[0x%qx])\n", getName(), regID2, provider->getName(), regID1);
IOServiceTrace(
IOSERVICE_TERMINATE_STOP,
(uintptr_t) regID1,
(uintptr_t) (regID1 >> 32),
(uintptr_t) regID2,
(uintptr_t) (regID2 >> 32));
stop( provider );
if( provider->isOpen( this ))
provider->close( this );
detach( provider );
} else {
if( provider->lockForArbitration( true )) {
if( 0 == (provider->__state[1] & kIOServiceTermPhase3State))
scheduleStop( provider );
provider->unlockForArbitration();
}
}
}
iter->release();
}
return( true );
}
#undef tailQ
#undef headQ
void IOService::doServiceTerminate( IOOptionBits options )
{
}
bool IOService::terminateClient( IOService * client, IOOptionBits options )
{
bool ok;
if( client->isParent( this, gIOServicePlane, true))
ok = client->terminate( options );
else
ok = true;
return( ok );
}
bool IOService::terminate( IOOptionBits options )
{
options |= kIOServiceTerminate;
return( terminatePhase1( options ));
}
struct ServiceOpenMessageContext
{
IOService * service;
UInt32 type;
IOService * excludeClient;
IOOptionBits options;
};
static void serviceOpenMessageApplier( OSObject * object, void * ctx )
{
ServiceOpenMessageContext * context = (ServiceOpenMessageContext *) ctx;
if( object != context->excludeClient)
context->service->messageClient( context->type, object, (void *)(uintptr_t) context->options );
}
bool IOService::open( IOService * forClient,
IOOptionBits options,
void * arg )
{
bool ok;
ServiceOpenMessageContext context;
context.service = this;
context.type = kIOMessageServiceIsAttemptingOpen;
context.excludeClient = forClient;
context.options = options;
applyToInterested( gIOGeneralInterest,
&serviceOpenMessageApplier, &context );
if( false == lockForArbitration(false) )
return false;
ok = (0 == (__state[0] & kIOServiceInactiveState));
if( ok)
ok = handleOpen( forClient, options, arg );
unlockForArbitration();
return( ok );
}
void IOService::close( IOService * forClient,
IOOptionBits options )
{
bool wasClosed;
bool last = false;
lockForArbitration();
wasClosed = handleIsOpen( forClient );
if( wasClosed) {
handleClose( forClient, options );
last = (__state[1] & kIOServiceTermPhase3State);
}
unlockForArbitration();
if( last)
forClient->scheduleStop( this );
else if( wasClosed) {
ServiceOpenMessageContext context;
context.service = this;
context.type = kIOMessageServiceWasClosed;
context.excludeClient = forClient;
context.options = options;
applyToInterested( gIOGeneralInterest,
&serviceOpenMessageApplier, &context );
}
}
bool IOService::isOpen( const IOService * forClient ) const
{
IOService * self = (IOService *) this;
bool ok;
self->lockForArbitration();
ok = handleIsOpen( forClient );
self->unlockForArbitration();
return( ok );
}
bool IOService::handleOpen( IOService * forClient,
IOOptionBits options,
void * arg )
{
bool ok;
ok = (0 == __owner);
if( ok )
__owner = forClient;
else if( options & kIOServiceSeize ) {
ok = (kIOReturnSuccess == messageClient( kIOMessageServiceIsRequestingClose,
__owner, (void *)(uintptr_t) options ));
if( ok && (0 == __owner ))
__owner = forClient;
else
ok = false;
}
return( ok );
}
void IOService::handleClose( IOService * forClient,
IOOptionBits options )
{
if( __owner == forClient)
__owner = 0;
}
bool IOService::handleIsOpen( const IOService * forClient ) const
{
if( forClient)
return( __owner == forClient );
else
return( __owner != forClient );
}
static SInt32 IONotifyOrdering( const OSMetaClassBase * inObj1, const OSMetaClassBase * inObj2, void * ref )
{
const _IOServiceNotifier * obj1 = (const _IOServiceNotifier *) inObj1;
const _IOServiceNotifier * obj2 = (const _IOServiceNotifier *) inObj2;
SInt32 val1;
SInt32 val2;
val1 = 0;
val2 = 0;
if ( obj1 )
val1 = obj1->priority;
if ( obj2 )
val2 = obj2->priority;
return ( val1 - val2 );
}
static SInt32 IOServiceObjectOrder( const OSObject * entry, void * ref)
{
OSDictionary * dict;
IOService * service;
_IOServiceNotifier * notify;
OSSymbol * key = (OSSymbol *) ref;
OSNumber * offset;
if( (dict = OSDynamicCast( OSDictionary, entry)))
offset = OSDynamicCast(OSNumber, dict->getObject( key ));
else if( (notify = OSDynamicCast( _IOServiceNotifier, entry)))
return( notify->priority );
else if( (service = OSDynamicCast( IOService, entry)))
offset = OSDynamicCast(OSNumber, service->getProperty( key ));
else {
assert( false );
offset = 0;
}
if( offset)
return( (SInt32) offset->unsigned32BitValue());
else
return( kIODefaultProbeScore );
}
SInt32 IOServiceOrdering( const OSMetaClassBase * inObj1, const OSMetaClassBase * inObj2, void * ref )
{
const OSObject * obj1 = (const OSObject *) inObj1;
const OSObject * obj2 = (const OSObject *) inObj2;
SInt32 val1;
SInt32 val2;
val1 = 0;
val2 = 0;
if ( obj1 )
val1 = IOServiceObjectOrder( obj1, ref );
if ( obj2 )
val2 = IOServiceObjectOrder( obj2, ref );
return ( val1 - val2 );
}
IOService * IOService::copyClientWithCategory( const OSSymbol * category )
{
IOService * service = 0;
OSIterator * iter;
const OSSymbol * nextCat;
iter = getClientIterator();
if( iter) {
while( (service = (IOService *) iter->getNextObject())) {
if( kIOServiceInactiveState & service->__state[0])
continue;
nextCat = (const OSSymbol *) OSDynamicCast( OSSymbol,
service->getProperty( gIOMatchCategoryKey ));
if( category == nextCat)
{
service->retain();
break;
}
}
iter->release();
}
return( service );
}
IOService * IOService::getClientWithCategory( const OSSymbol * category )
{
IOService *
service = copyClientWithCategory(category);
if (service)
service->release();
return (service);
}
bool IOService::invokeNotifer( _IOServiceNotifier * notify )
{
_IOServiceNotifierInvocation invocation;
bool willNotify;
bool ret = true;
invocation.thread = current_thread();
LOCKWRITENOTIFY();
willNotify = (0 != (kIOServiceNotifyEnable & notify->state));
if( willNotify) {
queue_enter( ¬ify->handlerInvocations, &invocation,
_IOServiceNotifierInvocation *, link );
}
UNLOCKNOTIFY();
if( willNotify) {
ret = (*notify->handler)(notify->target, notify->ref, this, notify);
LOCKWRITENOTIFY();
queue_remove( ¬ify->handlerInvocations, &invocation,
_IOServiceNotifierInvocation *, link );
if( kIOServiceNotifyWaiter & notify->state) {
notify->state &= ~kIOServiceNotifyWaiter;
WAKEUPNOTIFY( notify );
}
UNLOCKNOTIFY();
}
return( ret );
}
void IOService::probeCandidates( OSOrderedSet * matches )
{
OSDictionary * match = 0;
OSSymbol * symbol;
IOService * inst;
IOService * newInst;
OSDictionary * props;
SInt32 score;
OSNumber * newPri;
OSOrderedSet * familyMatches = 0;
OSOrderedSet * startList;
OSDictionary * startDict = 0;
const OSSymbol * category;
OSIterator * iter;
_IOServiceNotifier * notify;
OSObject * nextMatch = 0;
bool started;
bool needReloc = false;
#if IOMATCHDEBUG
SInt64 debugFlags;
#endif
IOService * client = NULL;
assert( matches );
while( !needReloc && (nextMatch = matches->getFirstObject())) {
nextMatch->retain();
matches->removeObject(nextMatch);
if( (notify = OSDynamicCast( _IOServiceNotifier, nextMatch ))) {
lockForArbitration();
if( 0 == (__state[0] & kIOServiceInactiveState))
invokeNotifer( notify );
unlockForArbitration();
nextMatch->release();
nextMatch = 0;
continue;
} else if( !(match = OSDynamicCast( OSDictionary, nextMatch ))) {
nextMatch->release();
nextMatch = 0;
continue;
}
props = 0;
#if IOMATCHDEBUG
debugFlags = getDebugFlags( match );
#endif
do {
category = OSDynamicCast( OSSymbol,
match->getObject( gIOMatchCategoryKey ));
if( 0 == category)
category = gIODefaultMatchCategoryKey;
if( (client = copyClientWithCategory(category)) ) {
#if IOMATCHDEBUG
if( (debugFlags & kIOLogMatch) && (this != gIOResources))
LOG("%s: match category %s exists\n", getName(),
category->getCStringNoCopy());
#endif
nextMatch->release();
nextMatch = 0;
client->release();
client = NULL;
continue;
}
props = OSDictionary::withDictionary( match, match->getCount());
if( 0 == props)
continue;
props->setCapacityIncrement(1);
if( false == matchPassive(props, kIOServiceChangesOK | kIOServiceClassDone))
continue;
needReloc = (false == gIOCatalogue->isModuleLoaded( match ));
if( needReloc) {
#if IOMATCHDEBUG
if( debugFlags & kIOLogCatalogue)
LOG("%s: stalling for module\n", getName());
#endif
continue;
}
if( 0 == familyMatches)
familyMatches = OSOrderedSet::withCapacity( 1,
IOServiceOrdering, (void *) gIOProbeScoreKey );
if( familyMatches)
familyMatches->setObject( props );
} while( false );
if (nextMatch) {
nextMatch->release();
nextMatch = 0;
}
if( props)
props->release();
}
matches->release();
matches = 0;
if( familyMatches) {
while( !needReloc
&& (props = (OSDictionary *) familyMatches->getFirstObject())) {
props->retain();
familyMatches->removeObject( props );
inst = 0;
newInst = 0;
#if IOMATCHDEBUG
debugFlags = getDebugFlags( props );
#endif
do {
symbol = OSDynamicCast( OSSymbol,
props->getObject( gIOClassKey));
if( !symbol)
continue;
inst = (IOService *) OSMetaClass::allocClassWithName( symbol);
if( !inst) {
IOLog("Couldn't alloc class \"%s\"\n",
symbol->getCStringNoCopy());
continue;
}
if( !(inst->init( props ))) {
#if IOMATCHDEBUG
if( debugFlags & kIOLogStart)
IOLog("%s::init fails\n", symbol->getCStringNoCopy());
#endif
continue;
}
if( __state[1] & kIOServiceSynchronousState)
inst->__state[1] |= kIOServiceSynchronousState;
category = OSDynamicCast( OSSymbol,
props->getObject( gIOMatchCategoryKey ));
if( 0 == category)
category = gIODefaultMatchCategoryKey;
inst->setProperty( gIOMatchCategoryKey, (OSObject *) category );
if( !(inst->attach( this )))
continue;
score = familyMatches->orderObject( props );
#if IOMATCHDEBUG
if( debugFlags & kIOLogProbe)
LOG("%s::probe(%s)\n",
inst->getMetaClass()->getClassName(), getName());
#endif
newInst = inst->probe( this, &score );
inst->detach( this );
if( 0 == newInst) {
#if IOMATCHDEBUG
if( debugFlags & kIOLogProbe)
IOLog("%s::probe fails\n", symbol->getCStringNoCopy());
#endif
continue;
}
newPri = OSNumber::withNumber( score, 32 );
if( newPri) {
newInst->setProperty( gIOProbeScoreKey, newPri );
newPri->release();
}
if( 0 == startDict)
startDict = OSDictionary::withCapacity( 1 );
assert( startDict );
startList = (OSOrderedSet *)
startDict->getObject( category );
if( 0 == startList) {
startList = OSOrderedSet::withCapacity( 1,
IOServiceOrdering, (void *) gIOProbeScoreKey );
if( startDict && startList) {
startDict->setObject( category, startList );
startList->release();
}
}
assert( startList );
if( startList)
startList->setObject( newInst );
} while( false );
props->release();
if( inst)
inst->release();
}
familyMatches->release();
familyMatches = 0;
}
iter = OSCollectionIterator::withCollection( startDict );
if( iter) {
while( (category = (const OSSymbol *) iter->getNextObject())) {
startList = (OSOrderedSet *) startDict->getObject( category );
assert( startList );
if( !startList)
continue;
started = false;
while( true && (inst = (IOService *)startList->getFirstObject())) {
inst->retain();
startList->removeObject(inst);
#if IOMATCHDEBUG
debugFlags = getDebugFlags( inst->getPropertyTable() );
if( debugFlags & kIOLogStart) {
if( started)
LOG( "match category exists, skipping " );
LOG( "%s::start(%s) <%d>\n", inst->getName(),
getName(), inst->getRetainCount());
}
#endif
if( false == started)
started = startCandidate( inst );
#if IOMATCHDEBUG
if( (debugFlags & kIOLogStart) && (false == started))
LOG( "%s::start(%s) <%d> failed\n", inst->getName(), getName(),
inst->getRetainCount());
#endif
inst->release();
}
}
iter->release();
}
lockForArbitration();
SInt32 adjBusy = 0;
uint64_t regID = getRegistryEntryID();
if( needReloc) {
adjBusy = (__state[1] & kIOServiceModuleStallState) ? 0 : 1;
if( adjBusy) {
IOServiceTrace(
IOSERVICE_MODULESTALL,
(uintptr_t) regID,
(uintptr_t) (regID >> 32),
(uintptr_t) this,
0);
__state[1] |= kIOServiceModuleStallState;
}
} else if( __state[1] & kIOServiceModuleStallState) {
IOServiceTrace(
IOSERVICE_MODULEUNSTALL,
(uintptr_t) regID,
(uintptr_t) (regID >> 32),
(uintptr_t) this,
0);
__state[1] &= ~kIOServiceModuleStallState;
adjBusy = -1;
}
if( adjBusy)
_adjustBusy( adjBusy );
unlockForArbitration();
if( startDict)
startDict->release();
}
bool IOService::startCandidate( IOService * service )
{
bool ok;
ok = service->attach( this );
if( ok)
{
if (this != gIOResources)
{
checkResources();
service->checkResources();
}
AbsoluteTime startTime;
AbsoluteTime endTime;
UInt64 nano;
if (kIOLogStart & gIOKitDebug)
clock_get_uptime(&startTime);
ok = service->start(this);
if (kIOLogStart & gIOKitDebug)
{
clock_get_uptime(&endTime);
if (CMP_ABSOLUTETIME(&endTime, &startTime) > 0)
{
SUB_ABSOLUTETIME(&endTime, &startTime);
absolutetime_to_nanoseconds(endTime, &nano);
if (nano > 500000000ULL)
IOLog("%s::start took %ld ms\n", service->getName(), (long)(UInt32)(nano / 1000000ULL));
}
}
if( !ok)
service->detach( this );
}
return( ok );
}
void IOService::publishResource( const char * key, OSObject * value )
{
const OSSymbol * sym;
if( (sym = OSSymbol::withCString( key))) {
publishResource( sym, value);
sym->release();
}
}
void IOService::publishResource( const OSSymbol * key, OSObject * value )
{
if( 0 == value)
value = (OSObject *) gIOServiceKey;
gIOResources->setProperty( key, value);
if( IORecursiveLockHaveLock( gNotificationLock))
return;
gIOResourceGenerationCount++;
gIOResources->registerService();
}
bool IOService::addNeededResource( const char * key )
{
OSObject * resourcesProp;
OSSet * set;
OSString * newKey;
bool ret;
resourcesProp = getProperty( gIOResourceMatchKey );
newKey = OSString::withCString( key );
if( (0 == resourcesProp) || (0 == newKey))
return( false);
set = OSDynamicCast( OSSet, resourcesProp );
if( !set) {
set = OSSet::withCapacity( 1 );
if( set)
set->setObject( resourcesProp );
}
else
set->retain();
set->setObject( newKey );
newKey->release();
ret = setProperty( gIOResourceMatchKey, set );
set->release();
return( ret );
}
bool IOService::checkResource( OSObject * matching )
{
OSString * str;
OSDictionary * table;
if( (str = OSDynamicCast( OSString, matching ))) {
if( gIOResources->getProperty( str ))
return( true );
}
if( str)
table = resourceMatching( str );
else if( (table = OSDynamicCast( OSDictionary, matching )))
table->retain();
else {
IOLog("%s: Can't match using: %s\n", getName(),
matching->getMetaClass()->getClassName());
return( true );
}
if( gIOKitDebug & kIOLogConfig)
LOG("config(%p): stalling %s\n", OBFUSCATE(IOThreadSelf()), getName());
waitForService( table );
if( gIOKitDebug & kIOLogConfig)
LOG("config(%p): waking\n", OBFUSCATE(IOThreadSelf()) );
return( true );
}
bool IOService::checkResources( void )
{
OSObject * resourcesProp;
OSSet * set;
OSIterator * iter;
bool ok;
resourcesProp = getProperty( gIOResourceMatchKey );
if( 0 == resourcesProp)
return( true );
if( (set = OSDynamicCast( OSSet, resourcesProp ))) {
iter = OSCollectionIterator::withCollection( set );
ok = (0 != iter);
while( ok && (resourcesProp = iter->getNextObject()) )
ok = checkResource( resourcesProp );
if( iter)
iter->release();
} else
ok = checkResource( resourcesProp );
return( ok );
}
void _IOConfigThread::configThread( void )
{
_IOConfigThread * inst;
do {
if( !(inst = new _IOConfigThread))
continue;
if( !inst->init())
continue;
thread_t unused;
if (KERN_SUCCESS != kernel_thread_start(&_IOConfigThread::main, inst, &unused))
continue;
return;
} while( false);
if( inst)
inst->release();
return;
}
void _IOConfigThread::free( void )
{
thread_deallocate(current_thread());
OSObject::free();
}
void IOService::doServiceMatch( IOOptionBits options )
{
_IOServiceNotifier * notify;
OSIterator * iter;
OSOrderedSet * matches;
SInt32 catalogGeneration;
bool keepGuessing = true;
bool reRegistered = true;
bool didRegister;
while( keepGuessing ) {
matches = gIOCatalogue->findDrivers( this, &catalogGeneration );
if( matches) {
lockForArbitration();
if( 0 == (__state[0] & kIOServiceFirstPublishState)) {
getMetaClass()->addInstance(this);
deliverNotification( gIOFirstPublishNotification,
kIOServiceFirstPublishState, 0xffffffff );
}
LOCKREADNOTIFY();
__state[1] &= ~kIOServiceNeedConfigState;
__state[1] |= kIOServiceConfigState;
didRegister = (0 == (kIOServiceRegisteredState & __state[0]));
__state[0] |= kIOServiceRegisteredState;
keepGuessing &= (0 == (__state[0] & kIOServiceInactiveState));
if (reRegistered && keepGuessing) {
iter = OSCollectionIterator::withCollection( (OSOrderedSet *)
gNotifications->getObject( gIOPublishNotification ) );
if( iter) {
while((notify = (_IOServiceNotifier *)
iter->getNextObject())) {
if( matchPassive(notify->matching, 0)
&& (kIOServiceNotifyEnable & notify->state))
matches->setObject( notify );
}
iter->release();
}
}
UNLOCKNOTIFY();
unlockForArbitration();
if (keepGuessing && matches->getCount() && (kIOReturnSuccess == getResources()))
probeCandidates( matches );
else
matches->release();
}
lockForArbitration();
reRegistered = (0 != (__state[1] & kIOServiceNeedConfigState));
keepGuessing =
(reRegistered || (catalogGeneration !=
gIOCatalogue->getGenerationCount()))
&& (0 == (__state[0] & kIOServiceInactiveState));
if( keepGuessing)
unlockForArbitration();
}
if( (0 == (__state[0] & kIOServiceInactiveState))
&& (0 == (__state[1] & kIOServiceModuleStallState)) ) {
deliverNotification( gIOMatchedNotification,
kIOServiceMatchedState, 0xffffffff );
if( 0 == (__state[0] & kIOServiceFirstMatchState))
deliverNotification( gIOFirstMatchNotification,
kIOServiceFirstMatchState, 0xffffffff );
}
__state[1] &= ~kIOServiceConfigState;
scheduleTerminatePhase2();
_adjustBusy( -1 );
unlockForArbitration();
}
UInt32 IOService::_adjustBusy( SInt32 delta )
{
IOService * next;
UInt32 count;
UInt32 result;
bool wasQuiet, nowQuiet, needWake;
next = this;
result = __state[1] & kIOServiceBusyStateMask;
if( delta) do {
if( next != this)
next->lockForArbitration();
count = next->__state[1] & kIOServiceBusyStateMask;
wasQuiet = (0 == count);
if (((delta < 0) && wasQuiet) || ((delta > 0) && (kIOServiceBusyMax == count)))
OSReportWithBacktrace("%s: bad busy count (%d,%d)\n", next->getName(), count, delta);
else
count += delta;
next->__state[1] = (next->__state[1] & ~kIOServiceBusyStateMask) | count;
nowQuiet = (0 == count);
needWake = (0 != (kIOServiceBusyWaiterState & next->__state[1]));
if( needWake) {
next->__state[1] &= ~kIOServiceBusyWaiterState;
IOLockLock( gIOServiceBusyLock );
thread_wakeup( (event_t) next);
IOLockUnlock( gIOServiceBusyLock );
}
if( next != this)
next->unlockForArbitration();
if( (wasQuiet || nowQuiet) ) {
uint64_t regID = next->getRegistryEntryID();
IOServiceTrace(
((wasQuiet) ? IOSERVICE_BUSY : IOSERVICE_NONBUSY),
(uintptr_t) regID,
(uintptr_t) (regID >> 32),
(uintptr_t) next,
0);
if (wasQuiet)
{
next->__timeBusy = mach_absolute_time();
}
else
{
next->__accumBusy += mach_absolute_time() - next->__timeBusy;
next->__timeBusy = 0;
}
MessageClientsContext context;
context.service = next;
context.type = kIOMessageServiceBusyStateChange;
context.argument = (void *) wasQuiet;
context.argSize = 0;
applyToInterestNotifiers( next, gIOBusyInterest,
&messageClientsApplier, &context );
#if !NO_KEXTD
if( nowQuiet && (next == gIOServiceRoot)) {
OSKext::considerUnloads();
IOServiceTrace(IOSERVICE_REGISTRY_QUIET, 0, 0, 0, 0);
}
#endif
}
delta = nowQuiet ? -1 : +1;
} while( (wasQuiet || nowQuiet) && (next = next->getProvider()));
return( result );
}
void IOService::adjustBusy( SInt32 delta )
{
lockForArbitration();
_adjustBusy( delta );
unlockForArbitration();
}
uint64_t IOService::getAccumulatedBusyTime( void )
{
uint64_t accumBusy = __accumBusy;
uint64_t timeBusy = __timeBusy;
uint64_t nano;
do
{
accumBusy = __accumBusy;
timeBusy = __timeBusy;
if (timeBusy)
accumBusy += mach_absolute_time() - timeBusy;
}
while (timeBusy != __timeBusy);
absolutetime_to_nanoseconds(*(AbsoluteTime *)&accumBusy, &nano);
return (nano);
}
UInt32 IOService::getBusyState( void )
{
return( __state[1] & kIOServiceBusyStateMask );
}
IOReturn IOService::waitForState( UInt32 mask, UInt32 value,
mach_timespec_t * timeout )
{
panic("waitForState");
return (kIOReturnUnsupported);
}
IOReturn IOService::waitForState( UInt32 mask, UInt32 value,
uint64_t timeout )
{
bool wait;
int waitResult = THREAD_AWAKENED;
bool computeDeadline = true;
AbsoluteTime abstime;
do {
lockForArbitration();
IOLockLock( gIOServiceBusyLock );
wait = (value != (__state[1] & mask));
if( wait) {
__state[1] |= kIOServiceBusyWaiterState;
unlockForArbitration();
if( timeout != UINT64_MAX ) {
if( computeDeadline ) {
AbsoluteTime nsinterval;
nanoseconds_to_absolutetime(timeout, &nsinterval );
clock_absolutetime_interval_to_deadline(nsinterval, &abstime);
computeDeadline = false;
}
assert_wait_deadline((event_t)this, THREAD_UNINT, __OSAbsoluteTime(abstime));
}
else
assert_wait((event_t)this, THREAD_UNINT );
} else
unlockForArbitration();
IOLockUnlock( gIOServiceBusyLock );
if( wait)
waitResult = thread_block(THREAD_CONTINUE_NULL);
} while( wait && (waitResult != THREAD_TIMED_OUT));
if( waitResult == THREAD_TIMED_OUT)
return( kIOReturnTimeout );
else
return( kIOReturnSuccess );
}
IOReturn IOService::waitQuiet( uint64_t timeout )
{
IOReturn ret;
ret = waitForState( kIOServiceBusyStateMask, 0, timeout );
if ((kIOReturnTimeout == ret) && (timeout >= 41000000000) && (kIOWaitQuietPanics & gIOKitDebug))
{
IORegistryIterator * iter;
OSOrderedSet * set;
OSOrderedSet * leaves;
IOService * next;
IOService * nextParent;
char * string;
char * s;
size_t len, l;
len = 256;
string = IONew(char, len);
set = NULL;
iter = IORegistryIterator::iterateOver(this, gIOServicePlane, kIORegistryIterateRecursively);
leaves = OSOrderedSet::withCapacity(4);
if (iter) set = iter->iterateAll();
if (string && leaves && set)
{
while ((next = (IOService *) set->getLastObject()))
{
if (next->getBusyState())
{
leaves->setObject(next);
nextParent = next;
while ((nextParent = nextParent->getProvider()))
{
set->removeObject(nextParent);
leaves->removeObject(nextParent);
}
}
set->removeObject(next);
}
s = string;
while ((next = (IOService *) leaves->getLastObject()))
{
l = snprintf(s, len, "%s'%s'", ((s == string) ? "" : ", "), next->getName());
if (l >= len) break;
s += l;
len -= l;
leaves->removeObject(next);
}
}
panic("busy timeout(%llds): %s", timeout / 1000000000ULL, string ? string : "");
}
return (ret);
}
IOReturn IOService::waitQuiet( mach_timespec_t * timeout )
{
uint64_t timeoutNS;
if (timeout)
{
timeoutNS = timeout->tv_sec;
timeoutNS *= kSecondScale;
timeoutNS += timeout->tv_nsec;
}
else
timeoutNS = UINT64_MAX;
return (waitQuiet(timeoutNS));
}
bool IOService::serializeProperties( OSSerialize * s ) const
{
#if 0
((IOService *)this)->setProperty( ((IOService *)this)->__state,
sizeof( __state), "__state");
#endif
return( super::serializeProperties(s) );
}
void _IOConfigThread::main(void * arg, wait_result_t result)
{
_IOConfigThread * self = (_IOConfigThread *) arg;
_IOServiceJob * job;
IOService * nub;
bool alive = true;
kern_return_t kr;
thread_precedence_policy_data_t precedence = { -1 };
kr = thread_policy_set(current_thread(),
THREAD_PRECEDENCE_POLICY,
(thread_policy_t) &precedence,
THREAD_PRECEDENCE_POLICY_COUNT);
if (KERN_SUCCESS != kr)
IOLog("thread_policy_set(%d)\n", kr);
do {
semaphore_wait( gJobsSemaphore );
IOTakeLock( gJobsLock );
job = (_IOServiceJob *) gJobs->getFirstObject();
job->retain();
gJobs->removeObject(job);
if( job) {
gOutstandingJobs--;
gNumWaitingThreads--; }
IOUnlock( gJobsLock );
if( job) {
nub = job->nub;
if( gIOKitDebug & kIOLogConfig)
LOG("config(%p): starting on %s, %d\n",
OBFUSCATE(IOThreadSelf()), job->nub->getName(), job->type);
switch( job->type) {
case kMatchNubJob:
nub->doServiceMatch( job->options );
break;
default:
LOG("config(%p): strange type (%d)\n",
OBFUSCATE(IOThreadSelf()), job->type );
break;
}
nub->release();
job->release();
IOTakeLock( gJobsLock );
alive = (gOutstandingJobs > gNumWaitingThreads);
if( alive)
gNumWaitingThreads++; else {
if( 0 == --gNumConfigThreads) {
IOLockWakeup( gJobsLock, (event_t) &gNumConfigThreads, false );
}
}
IOUnlock( gJobsLock );
}
} while( alive );
if( gIOKitDebug & kIOLogConfig)
LOG("config(%p): terminating\n", OBFUSCATE(IOThreadSelf()) );
self->release();
}
IOReturn IOService::waitMatchIdle( UInt32 msToWait )
{
bool wait;
int waitResult = THREAD_AWAKENED;
bool computeDeadline = true;
AbsoluteTime deadline;
IOLockLock( gJobsLock );
do {
wait = (0 != gNumConfigThreads);
if( wait) {
if( msToWait) {
if( computeDeadline ) {
clock_interval_to_deadline(
msToWait, kMillisecondScale, &deadline );
computeDeadline = false;
}
waitResult = IOLockSleepDeadline( gJobsLock, &gNumConfigThreads,
deadline, THREAD_UNINT );
} else {
waitResult = IOLockSleep( gJobsLock, &gNumConfigThreads,
THREAD_UNINT );
}
}
} while( wait && (waitResult != THREAD_TIMED_OUT));
IOLockUnlock( gJobsLock );
if( waitResult == THREAD_TIMED_OUT)
return( kIOReturnTimeout );
else
return( kIOReturnSuccess );
}
void IOService::cpusRunning(void)
{
gCPUsRunning = true;
}
void _IOServiceJob::pingConfig( _IOServiceJob * job )
{
int count;
bool create;
assert( job );
IOTakeLock( gJobsLock );
gOutstandingJobs++;
gJobs->setLastObject( job );
count = gNumWaitingThreads;
create = ( (gOutstandingJobs > count)
&& ((gNumConfigThreads < kMaxConfigThreads)
|| (job->nub == gIOResources)
|| !gCPUsRunning));
if( create) {
gNumConfigThreads++;
gNumWaitingThreads++;
}
IOUnlock( gJobsLock );
job->release();
if( create) {
if( gIOKitDebug & kIOLogConfig)
LOG("config(%d): creating\n", gNumConfigThreads - 1);
_IOConfigThread::configThread();
}
semaphore_signal( gJobsSemaphore );
}
struct IOServiceMatchContext
{
OSDictionary * table;
OSObject * result;
uint32_t options;
uint32_t state;
uint32_t count;
uint32_t done;
};
bool IOService::instanceMatch(const OSObject * entry, void * context)
{
IOServiceMatchContext * ctx = (typeof(ctx)) context;
IOService * service = (typeof(service)) entry;
OSDictionary * table = ctx->table;
uint32_t options = ctx->options;
uint32_t state = ctx->state;
uint32_t done;
bool match;
done = 0;
do
{
match = ((state == (state & service->__state[0]))
&& (0 == (service->__state[0] & kIOServiceInactiveState)));
if (!match) break;
ctx->count += table->getCount();
match = service->matchInternal(table, options, &done);
ctx->done += done;
}
while (false);
if (!match)
return (false);
if ((kIONotifyOnce & options) && (ctx->done == ctx->count))
{
service->retain();
ctx->result = service;
return (true);
}
else if (!ctx->result)
{
ctx->result = OSSet::withObjects((const OSObject **) &service, 1, 1);
}
else
{
((OSSet *)ctx->result)->setObject(service);
}
return (false);
}
OSObject * IOService::copyExistingServices( OSDictionary * matching,
IOOptionBits inState, IOOptionBits options )
{
OSObject * current = 0;
OSIterator * iter;
IOService * service;
OSObject * obj;
OSString * str;
if( !matching)
return( 0 );
#if MATCH_DEBUG
OSSerialize * s = OSSerialize::withCapacity(128);
matching->serialize(s);
#endif
if((obj = matching->getObject(gIOProviderClassKey))
&& gIOResourcesKey
&& gIOResourcesKey->isEqualTo(obj)
&& (service = gIOResources))
{
if( (inState == (service->__state[0] & inState))
&& (0 == (service->__state[0] & kIOServiceInactiveState))
&& service->matchPassive(matching, options))
{
if( options & kIONotifyOnce)
{
service->retain();
current = service;
}
else
current = OSSet::withObjects((const OSObject **) &service, 1, 1 );
}
}
else
{
IOServiceMatchContext ctx;
ctx.table = matching;
ctx.state = inState;
ctx.count = 0;
ctx.done = 0;
ctx.options = options;
ctx.result = 0;
if ((str = OSDynamicCast(OSString, obj)))
{
const OSSymbol * sym = OSSymbol::withString(str);
OSMetaClass::applyToInstancesOfClassName(sym, instanceMatch, &ctx);
sym->release();
}
else
{
IOService::gMetaClass.applyToInstances(instanceMatch, &ctx);
}
current = ctx.result;
options |= kIOServiceInternalDone | kIOServiceClassDone;
if (current && (ctx.done != ctx.count))
{
OSSet *
source = OSDynamicCast(OSSet, current);
current = 0;
while ((service = (IOService *) source->getAnyObject()))
{
if (service->matchPassive(matching, options))
{
if( options & kIONotifyOnce)
{
service->retain();
current = service;
break;
}
if( current)
{
((OSSet *)current)->setObject( service );
}
else
{
current = OSSet::withObjects(
(const OSObject **) &service, 1, 1 );
}
}
source->removeObject(service);
}
source->release();
}
}
#if MATCH_DEBUG
{
OSObject * _current = 0;
iter = IORegistryIterator::iterateOver( gIOServicePlane,
kIORegistryIterateRecursively );
if( iter) {
do {
iter->reset();
while( (service = (IOService *) iter->getNextObject())) {
if( (inState == (service->__state[0] & inState))
&& (0 == (service->__state[0] & kIOServiceInactiveState))
&& service->matchPassive(matching, 0)) {
if( options & kIONotifyOnce) {
service->retain();
_current = service;
break;
}
if( _current)
((OSSet *)_current)->setObject( service );
else
_current = OSSet::withObjects(
(const OSObject **) &service, 1, 1 );
}
}
} while( !service && !iter->isValid());
iter->release();
}
if ( ((current != 0) != (_current != 0))
|| (current && _current && !current->isEqualTo(_current)))
{
OSSerialize * s1 = OSSerialize::withCapacity(128);
OSSerialize * s2 = OSSerialize::withCapacity(128);
current->serialize(s1);
_current->serialize(s2);
kprintf("**mismatch** %p %p\n%s\n%s\n%s\n", OBFUSCATE(current),
OBFUSCATE(_current), s->text(), s1->text(), s2->text());
s1->release();
s2->release();
}
if (_current) _current->release();
}
s->release();
#endif
if( current && (0 == (options & (kIONotifyOnce | kIOServiceExistingSet)))) {
iter = OSCollectionIterator::withCollection( (OSSet *)current );
current->release();
current = iter;
}
return( current );
}
OSIterator * IOService::getMatchingServices( OSDictionary * matching )
{
OSIterator * iter;
LOCKWRITENOTIFY();
iter = (OSIterator *) copyExistingServices( matching,
kIOServiceMatchedState );
UNLOCKNOTIFY();
return( iter );
}
IOService * IOService::copyMatchingService( OSDictionary * matching )
{
IOService * service;
LOCKWRITENOTIFY();
service = (IOService *) copyExistingServices( matching,
kIOServiceMatchedState, kIONotifyOnce );
UNLOCKNOTIFY();
return( service );
}
struct _IOServiceMatchingNotificationHandlerRef
{
IOServiceNotificationHandler handler;
void * ref;
};
static bool _IOServiceMatchingNotificationHandler( void * target, void * refCon,
IOService * newService,
IONotifier * notifier )
{
return ((*((_IOServiceNotifier *) notifier)->compatHandler)(target, refCon, newService));
}
IONotifier * IOService::setNotification(
const OSSymbol * type, OSDictionary * matching,
IOServiceMatchingNotificationHandler handler, void * target, void * ref,
SInt32 priority )
{
_IOServiceNotifier * notify = 0;
OSOrderedSet * set;
if( !matching)
return( 0 );
notify = new _IOServiceNotifier;
if( notify && !notify->init()) {
notify->release();
notify = 0;
}
if( notify) {
notify->handler = handler;
notify->target = target;
notify->matching = matching;
matching->retain();
if (handler == &_IOServiceMatchingNotificationHandler)
{
notify->compatHandler = ((_IOServiceMatchingNotificationHandlerRef *)ref)->handler;
notify->ref = ((_IOServiceMatchingNotificationHandlerRef *)ref)->ref;
}
else
notify->ref = ref;
notify->priority = priority;
notify->state = kIOServiceNotifyEnable;
queue_init( ¬ify->handlerInvocations );
if( 0 == (set = (OSOrderedSet *) gNotifications->getObject( type ))) {
set = OSOrderedSet::withCapacity( 1,
IONotifyOrdering, 0 );
if( set) {
gNotifications->setObject( type, set );
set->release();
}
}
notify->whence = set;
if( set)
set->setObject( notify );
}
return( notify );
}
IONotifier * IOService::doInstallNotification(
const OSSymbol * type, OSDictionary * matching,
IOServiceMatchingNotificationHandler handler,
void * target, void * ref,
SInt32 priority, OSIterator ** existing )
{
OSIterator * exist;
IONotifier * notify;
IOOptionBits inState;
if( !matching)
return( 0 );
if( type == gIOPublishNotification)
inState = kIOServiceRegisteredState;
else if( type == gIOFirstPublishNotification)
inState = kIOServiceFirstPublishState;
else if( (type == gIOMatchedNotification)
|| (type == gIOFirstMatchNotification))
inState = kIOServiceMatchedState;
else if( type == gIOTerminatedNotification)
inState = 0;
else
return( 0 );
notify = setNotification( type, matching, handler, target, ref, priority );
if( inState)
exist = (OSIterator *) copyExistingServices( matching, inState );
else
exist = 0;
*existing = exist;
return( notify );
}
#if !defined(__LP64__)
IONotifier * IOService::installNotification(const OSSymbol * type, OSDictionary * matching,
IOServiceNotificationHandler handler,
void * target, void * refCon,
SInt32 priority, OSIterator ** existing )
{
IONotifier * result;
_IOServiceMatchingNotificationHandlerRef ref;
ref.handler = handler;
ref.ref = refCon;
result = (_IOServiceNotifier *) installNotification( type, matching,
&_IOServiceMatchingNotificationHandler,
target, &ref, priority, existing );
if (result)
matching->release();
return (result);
}
#endif
IONotifier * IOService::installNotification(
const OSSymbol * type, OSDictionary * matching,
IOServiceMatchingNotificationHandler handler,
void * target, void * ref,
SInt32 priority, OSIterator ** existing )
{
IONotifier * notify;
LOCKWRITENOTIFY();
notify = doInstallNotification( type, matching, handler, target, ref,
priority, existing );
UNLOCKNOTIFY();
return( notify );
}
IONotifier * IOService::addNotification(
const OSSymbol * type, OSDictionary * matching,
IOServiceNotificationHandler handler,
void * target, void * refCon,
SInt32 priority )
{
IONotifier * result;
_IOServiceMatchingNotificationHandlerRef ref;
ref.handler = handler;
ref.ref = refCon;
result = addMatchingNotification(type, matching, &_IOServiceMatchingNotificationHandler,
target, &ref, priority);
if (result)
matching->release();
return (result);
}
IONotifier * IOService::addMatchingNotification(
const OSSymbol * type, OSDictionary * matching,
IOServiceMatchingNotificationHandler handler,
void * target, void * ref,
SInt32 priority )
{
OSIterator * existing = NULL;
_IOServiceNotifier * notify;
IOService * next;
notify = (_IOServiceNotifier *) installNotification( type, matching,
handler, target, ref, priority, &existing );
if( existing) {
notify->retain(); while( (next = (IOService *) existing->getNextObject())) {
next->lockForArbitration();
if( 0 == (next->__state[0] & kIOServiceInactiveState))
next->invokeNotifer( notify );
next->unlockForArbitration();
}
notify->release();
existing->release();
}
return( notify );
}
bool IOService::syncNotificationHandler(
void * , void * ref,
IOService * newService,
IONotifier * notifier )
{
LOCKWRITENOTIFY();
if (!*((IOService **) ref))
{
newService->retain();
(*(IOService **) ref) = newService;
WAKEUPNOTIFY(ref);
}
UNLOCKNOTIFY();
return( false );
}
IOService * IOService::waitForMatchingService( OSDictionary * matching,
uint64_t timeout)
{
IONotifier * notify = 0;
SInt32 priority = 0;
IOService * result;
if (!matching)
return( 0 );
result = NULL;
LOCKWRITENOTIFY();
do
{
result = (IOService *) copyExistingServices( matching,
kIOServiceMatchedState, kIONotifyOnce );
if (result)
break;
notify = IOService::setNotification( gIOMatchedNotification, matching,
&IOService::syncNotificationHandler, (void *) 0,
&result, priority );
if (!notify)
break;
if (UINT64_MAX != timeout)
{
AbsoluteTime deadline;
nanoseconds_to_absolutetime(timeout, &deadline);
clock_absolutetime_interval_to_deadline(deadline, &deadline);
SLEEPNOTIFYTO(&result, deadline);
}
else
{
SLEEPNOTIFY(&result);
}
}
while( false );
UNLOCKNOTIFY();
if (notify)
notify->remove();
return( result );
}
IOService * IOService::waitForService( OSDictionary * matching,
mach_timespec_t * timeout )
{
IOService * result;
uint64_t timeoutNS;
if (timeout)
{
timeoutNS = timeout->tv_sec;
timeoutNS *= kSecondScale;
timeoutNS += timeout->tv_nsec;
}
else
timeoutNS = UINT64_MAX;
result = waitForMatchingService(matching, timeoutNS);
matching->release();
if (result)
result->release();
return (result);
}
void IOService::deliverNotification( const OSSymbol * type,
IOOptionBits orNewState, IOOptionBits andNewState )
{
_IOServiceNotifier * notify;
OSIterator * iter;
OSArray * willSend = 0;
lockForArbitration();
if( (0 == (__state[0] & kIOServiceInactiveState))
|| (type == gIOTerminatedNotification)) {
LOCKREADNOTIFY();
iter = OSCollectionIterator::withCollection( (OSOrderedSet *)
gNotifications->getObject( type ) );
if( iter) {
while( (notify = (_IOServiceNotifier *) iter->getNextObject())) {
if( matchPassive(notify->matching, 0)
&& (kIOServiceNotifyEnable & notify->state)) {
if( 0 == willSend)
willSend = OSArray::withCapacity(8);
if( willSend)
willSend->setObject( notify );
}
}
iter->release();
}
__state[0] = (__state[0] | orNewState) & andNewState;
UNLOCKNOTIFY();
}
if( willSend) {
for( unsigned int idx = 0;
(notify = (_IOServiceNotifier *) willSend->getObject(idx));
idx++) {
invokeNotifer( notify );
}
willSend->release();
}
unlockForArbitration();
}
IOOptionBits IOService::getState( void ) const
{
return( __state[0] );
}
OSDictionary * IOService::serviceMatching( const OSString * name,
OSDictionary * table )
{
const OSString * str;
str = OSSymbol::withString(name);
if( !str)
return( 0 );
if( !table)
table = OSDictionary::withCapacity( 2 );
if( table)
table->setObject(gIOProviderClassKey, (OSObject *)str );
str->release();
return( table );
}
OSDictionary * IOService::serviceMatching( const char * name,
OSDictionary * table )
{
const OSString * str;
str = OSSymbol::withCString( name );
if( !str)
return( 0 );
table = serviceMatching( str, table );
str->release();
return( table );
}
OSDictionary * IOService::nameMatching( const OSString * name,
OSDictionary * table )
{
if( !table)
table = OSDictionary::withCapacity( 2 );
if( table)
table->setObject( gIONameMatchKey, (OSObject *)name );
return( table );
}
OSDictionary * IOService::nameMatching( const char * name,
OSDictionary * table )
{
const OSString * str;
str = OSSymbol::withCString( name );
if( !str)
return( 0 );
table = nameMatching( str, table );
str->release();
return( table );
}
OSDictionary * IOService::resourceMatching( const OSString * str,
OSDictionary * table )
{
table = serviceMatching( gIOResourcesKey, table );
if( table)
table->setObject( gIOResourceMatchKey, (OSObject *) str );
return( table );
}
OSDictionary * IOService::resourceMatching( const char * name,
OSDictionary * table )
{
const OSSymbol * str;
str = OSSymbol::withCString( name );
if( !str)
return( 0 );
table = resourceMatching( str, table );
str->release();
return( table );
}
OSDictionary * IOService::propertyMatching( const OSSymbol * key, const OSObject * value,
OSDictionary * table )
{
OSDictionary * properties;
properties = OSDictionary::withCapacity( 2 );
if( !properties)
return( 0 );
properties->setObject( key, value );
if( !table)
table = OSDictionary::withCapacity( 2 );
if( table)
table->setObject( gIOPropertyMatchKey, properties );
properties->release();
return( table );
}
OSDictionary * IOService::registryEntryIDMatching( uint64_t entryID,
OSDictionary * table )
{
OSNumber * num;
num = OSNumber::withNumber( entryID, 64 );
if( !num)
return( 0 );
if( !table)
table = OSDictionary::withCapacity( 2 );
if( table)
table->setObject( gIORegistryEntryIDKey, num );
if (num)
num->release();
return( table );
}
void _IOServiceNotifier::wait()
{
_IOServiceNotifierInvocation * next;
bool doWait;
do {
doWait = false;
queue_iterate( &handlerInvocations, next,
_IOServiceNotifierInvocation *, link) {
if( next->thread != current_thread() ) {
doWait = true;
break;
}
}
if( doWait) {
state |= kIOServiceNotifyWaiter;
SLEEPNOTIFY(this);
}
} while( doWait );
}
void _IOServiceNotifier::free()
{
assert( queue_empty( &handlerInvocations ));
OSObject::free();
}
void _IOServiceNotifier::remove()
{
LOCKWRITENOTIFY();
if( whence) {
whence->removeObject( (OSObject *) this );
whence = 0;
}
if( matching) {
matching->release();
matching = 0;
}
state &= ~kIOServiceNotifyEnable;
wait();
UNLOCKNOTIFY();
release();
}
bool _IOServiceNotifier::disable()
{
bool ret;
LOCKWRITENOTIFY();
ret = (0 != (kIOServiceNotifyEnable & state));
state &= ~kIOServiceNotifyEnable;
if( ret)
wait();
UNLOCKNOTIFY();
return( ret );
}
void _IOServiceNotifier::enable( bool was )
{
LOCKWRITENOTIFY();
if( was)
state |= kIOServiceNotifyEnable;
else
state &= ~kIOServiceNotifyEnable;
UNLOCKNOTIFY();
}
IOService * IOResources::resources( void )
{
IOResources * inst;
inst = new IOResources;
if( inst && !inst->init()) {
inst->release();
inst = 0;
}
return( inst );
}
bool IOResources::init( OSDictionary * dictionary )
{
if ( !super::init() )
return false;
const char *property_name;
int property_value;
pal_get_resource_property( &property_name, &property_value );
if( property_name ) {
OSNumber *num;
const OSSymbol * sym;
if( (num = OSNumber::withNumber(property_value, 32)) != 0 ) {
if( (sym = OSSymbol::withCString( property_name)) != 0 ) {
this->setProperty( sym, num );
sym->release();
}
num->release();
}
}
return true;
}
IOReturn IOResources::newUserClient(task_t owningTask, void * securityID,
UInt32 type, OSDictionary * properties,
IOUserClient ** handler)
{
return( kIOReturnUnsupported );
}
IOWorkLoop * IOResources::getWorkLoop() const
{
if (this == (IOResources *) gIOResources)
return getPlatform()->getWorkLoop();
else
return IOService::getWorkLoop();
}
bool IOResources::matchPropertyTable( OSDictionary * table )
{
OSObject * prop;
OSString * str;
OSSet * set;
OSIterator * iter;
bool ok = true;
prop = table->getObject( gIOResourceMatchKey );
str = OSDynamicCast( OSString, prop );
if( str)
ok = (0 != getProperty( str ));
else if( (set = OSDynamicCast( OSSet, prop))) {
iter = OSCollectionIterator::withCollection( set );
ok = (iter != 0);
while( ok && (str = OSDynamicCast( OSString, iter->getNextObject()) ))
ok = (0 != getProperty( str ));
if( iter)
iter->release();
}
return( ok );
}
void IOService::consoleLockTimer(thread_call_param_t p0, thread_call_param_t p1)
{
IOService::updateConsoleUsers(NULL, 0);
}
void IOService::updateConsoleUsers(OSArray * consoleUsers, IOMessage systemMessage)
{
IORegistryEntry * regEntry;
OSObject * locked = kOSBooleanFalse;
uint32_t idx;
bool publish;
OSDictionary * user;
static IOMessage sSystemPower;
regEntry = IORegistryEntry::getRegistryRoot();
if (!gIOChosenEntry)
gIOChosenEntry = IORegistryEntry::fromPath("/chosen", gIODTPlane);
IOLockLock(gIOConsoleUsersLock);
if (systemMessage)
{
sSystemPower = systemMessage;
#if HIBERNATION
if ((kIOMessageSystemHasPoweredOn == systemMessage) && IOHibernateWasScreenLocked())
{
locked = kOSBooleanTrue;
}
#endif
}
if (consoleUsers)
{
OSNumber * num = 0;
gIOConsoleLoggedIn = false;
for (idx = 0;
(user = OSDynamicCast(OSDictionary, consoleUsers->getObject(idx)));
idx++)
{
gIOConsoleLoggedIn |= ((kOSBooleanTrue == user->getObject(gIOConsoleSessionOnConsoleKey))
&& (kOSBooleanTrue == user->getObject(gIOConsoleSessionLoginDoneKey)));
if (!num)
{
num = OSDynamicCast(OSNumber, user->getObject(gIOConsoleSessionScreenLockedTimeKey));
}
}
gIOConsoleLockTime = num ? num->unsigned32BitValue() : 0;
}
if (!gIOConsoleLoggedIn
|| (kIOMessageSystemWillSleep == sSystemPower)
|| (kIOMessageSystemPagingOff == sSystemPower))
{
locked = kOSBooleanTrue;
}
else if (gIOConsoleLockTime)
{
clock_sec_t now;
clock_usec_t microsecs;
clock_get_calendar_microtime(&now, µsecs);
if (gIOConsoleLockTime > now)
{
AbsoluteTime deadline;
clock_interval_to_deadline(gIOConsoleLockTime - now, kSecondScale, &deadline);
thread_call_enter_delayed(gIOConsoleLockCallout, deadline);
}
else
{
locked = kOSBooleanTrue;
}
}
publish = (consoleUsers || (locked != regEntry->getProperty(gIOConsoleLockedKey)));
if (publish)
{
regEntry->setProperty(gIOConsoleLockedKey, locked);
if (consoleUsers)
{
regEntry->setProperty(gIOConsoleUsersKey, consoleUsers);
}
OSIncrementAtomic( &gIOConsoleUsersSeed );
}
#if HIBERNATION
if (gIOChosenEntry)
{
if (locked == kOSBooleanTrue) gIOScreenLockState = kIOScreenLockLocked;
else if (gIOConsoleLockTime) gIOScreenLockState = kIOScreenLockUnlocked;
else gIOScreenLockState = kIOScreenLockNoLock;
gIOChosenEntry->setProperty(kIOScreenLockStateKey, &gIOScreenLockState, sizeof(gIOScreenLockState));
}
#endif
IOLockUnlock(gIOConsoleUsersLock);
if (publish)
{
publishResource( gIOConsoleUsersSeedKey, gIOConsoleUsersSeedValue );
MessageClientsContext context;
context.service = getServiceRoot();
context.type = kIOMessageConsoleSecurityChange;
context.argument = (void *) regEntry;
context.argSize = 0;
applyToInterestNotifiers(getServiceRoot(), gIOConsoleSecurityInterest,
&messageClientsApplier, &context );
}
}
IOReturn IOResources::setProperties( OSObject * properties )
{
IOReturn err;
const OSSymbol * key;
OSDictionary * dict;
OSCollectionIterator * iter;
err = IOUserClient::clientHasPrivilege(current_task(), kIOClientPrivilegeAdministrator);
if ( kIOReturnSuccess != err)
return( err );
dict = OSDynamicCast(OSDictionary, properties);
if( 0 == dict)
return( kIOReturnBadArgument);
iter = OSCollectionIterator::withCollection( dict);
if( 0 == iter)
return( kIOReturnBadArgument);
while( (key = OSDynamicCast(OSSymbol, iter->getNextObject())))
{
if (gIOConsoleUsersKey == key) do
{
OSArray * consoleUsers;
consoleUsers = OSDynamicCast(OSArray, dict->getObject(key));
if (!consoleUsers)
continue;
IOService::updateConsoleUsers(consoleUsers, 0);
}
while (false);
publishResource( key, dict->getObject(key) );
}
iter->release();
return( kIOReturnSuccess );
}
bool IOService::compareProperty( OSDictionary * matching,
const char * key )
{
OSObject * value;
bool ok;
value = matching->getObject( key );
if( value)
ok = value->isEqualTo( getProperty( key ));
else
ok = true;
return( ok );
}
bool IOService::compareProperty( OSDictionary * matching,
const OSString * key )
{
OSObject * value;
bool ok;
value = matching->getObject( key );
if( value)
ok = value->isEqualTo( getProperty( key ));
else
ok = true;
return( ok );
}
bool IOService::compareProperties( OSDictionary * matching,
OSCollection * keys )
{
OSCollectionIterator * iter;
const OSString * key;
bool ok = true;
if( !matching || !keys)
return( false );
iter = OSCollectionIterator::withCollection( keys );
if( iter) {
while( ok && (key = OSDynamicCast( OSString, iter->getNextObject())))
ok = compareProperty( matching, key );
iter->release();
}
keys->release();
return( ok );
}
OSDictionary * IOService::addLocation( OSDictionary * table )
{
OSDictionary * dict;
if( !table)
return( 0 );
dict = OSDictionary::withCapacity( 1 );
if( dict) {
table->setObject( gIOLocationMatchKey, dict );
dict->release();
}
return( dict );
}
IOService * IOService::matchLocation( IOService * )
{
IOService * parent;
parent = getProvider();
if( parent)
parent = parent->matchLocation( this );
return( parent );
}
bool IOService::matchInternal(OSDictionary * table, uint32_t options, uint32_t * did)
{
OSString * matched;
OSObject * obj;
OSString * str;
IORegistryEntry * entry;
OSNumber * num;
bool match = true;
bool changesOK = (0 != (kIOServiceChangesOK & options));
uint32_t count;
uint32_t done;
do
{
count = table->getCount();
done = 0;
str = OSDynamicCast(OSString, table->getObject(gIOProviderClassKey));
if (str) {
done++;
match = ((kIOServiceClassDone & options) || (0 != metaCast(str)));
#if MATCH_DEBUG
match = (0 != metaCast( str ));
if ((kIOServiceClassDone & options) && !match) panic("classDone");
#endif
if ((!match) || (done == count)) break;
}
obj = table->getObject( gIONameMatchKey );
if( obj) {
done++;
match = compareNames( obj, changesOK ? &matched : 0 );
if (!match) break;
if( changesOK && matched) {
table->setObject( gIONameMatchedKey, matched );
matched->release();
}
if (done == count) break;
}
str = OSDynamicCast( OSString, table->getObject( gIOLocationMatchKey ));
if (str)
{
const OSSymbol * sym;
done++;
match = false;
sym = copyLocation();
if (sym) {
match = sym->isEqualTo( str );
sym->release();
}
if ((!match) || (done == count)) break;
}
obj = table->getObject( gIOPropertyMatchKey );
if( obj)
{
OSDictionary * dict;
OSDictionary * nextDict;
OSIterator * iter;
done++;
match = false;
dict = dictionaryWithProperties();
if( dict) {
nextDict = OSDynamicCast( OSDictionary, obj);
if( nextDict)
iter = 0;
else
iter = OSCollectionIterator::withCollection(
OSDynamicCast(OSCollection, obj));
while( nextDict
|| (iter && (0 != (nextDict = OSDynamicCast(OSDictionary,
iter->getNextObject()))))) {
match = dict->isEqualTo( nextDict, nextDict);
if( match)
break;
nextDict = 0;
}
dict->release();
if( iter)
iter->release();
}
if ((!match) || (done == count)) break;
}
str = OSDynamicCast( OSString, table->getObject( gIOPathMatchKey ));
if( str) {
done++;
entry = IORegistryEntry::fromPath( str->getCStringNoCopy() );
match = (this == entry);
if( entry)
entry->release();
if ((!match) || (done == count)) break;
}
num = OSDynamicCast( OSNumber, table->getObject( gIORegistryEntryIDKey ));
if (num) {
done++;
match = (getRegistryEntryID() == num->unsigned64BitValue());
if ((!match) || (done == count)) break;
}
num = OSDynamicCast( OSNumber, table->getObject( gIOMatchedServiceCountKey ));
if( num)
{
OSIterator * iter;
IOService * service = 0;
UInt32 serviceCount = 0;
done++;
iter = getClientIterator();
if( iter) {
while( (service = (IOService *) iter->getNextObject())) {
if( kIOServiceInactiveState & service->__state[0])
continue;
if( 0 == service->getProperty( gIOMatchCategoryKey ))
continue;
++serviceCount;
}
iter->release();
}
match = (serviceCount == num->unsigned32BitValue());
if ((!match) || (done == count)) break;
}
#define propMatch(key) \
obj = table->getObject(key); \
if (obj) \
{ \
OSObject * prop; \
done++; \
prop = copyProperty(key); \
match = obj->isEqualTo(prop); \
if (prop) prop->release(); \
if ((!match) || (done == count)) break; \
}
propMatch(kIOBSDNameKey)
propMatch(kIOBSDMajorKey)
propMatch(kIOBSDMinorKey)
propMatch(kIOBSDUnitKey)
#undef propMatch
}
while (false);
if (did) *did = done;
return (match);
}
bool IOService::passiveMatch( OSDictionary * table, bool changesOK )
{
return (matchPassive(table, changesOK ? kIOServiceChangesOK : 0));
}
bool IOService::matchPassive(OSDictionary * table, uint32_t options)
{
IOService * where;
OSDictionary * nextTable;
SInt32 score;
OSNumber * newPri;
bool match = true;
bool matchParent = false;
uint32_t count;
uint32_t done;
assert( table );
#if MATCH_DEBUG
OSDictionary * root = table;
#endif
where = this;
do
{
do
{
count = table->getCount();
if (!(kIOServiceInternalDone & options))
{
match = where->matchInternal(table, options, &done);
if ((!match) || (done == count)) break;
}
score = IOServiceObjectOrder( table, (void *) gIOProbeScoreKey);
match = where->matchPropertyTable( table, &score );
if( !match) {
#if IOMATCHDEBUG
if( kIOLogMatch & getDebugFlags( table ))
LOG("%s: family specific matching fails\n", where->getName());
#endif
break;
}
if (kIOServiceChangesOK & options) {
newPri = OSNumber::withNumber( score, 32 );
if( newPri) {
table->setObject( gIOProbeScoreKey, newPri );
newPri->release();
}
}
options = 0;
matchParent = false;
nextTable = OSDynamicCast(OSDictionary,
table->getObject( gIOParentMatchKey ));
if( nextTable) {
match = false;
matchParent = true;
table = nextTable;
break;
}
table = OSDynamicCast(OSDictionary,
table->getObject( gIOLocationMatchKey ));
if (table) {
match = false;
where = where->getProvider();
if (where && (where = where->matchLocation(where))) continue;
}
break;
}
while (true);
}
while( matchParent && (!match) && (where = where->getProvider()) );
#if MATCH_DEBUG
if (where != this)
{
OSSerialize * s = OSSerialize::withCapacity(128);
root->serialize(s);
kprintf("parent match 0x%llx, %d,\n%s\n", getRegistryEntryID(), match, s->text());
s->release();
}
#endif
return( match );
}
IOReturn IOService::newUserClient( task_t owningTask, void * securityID,
UInt32 type, OSDictionary * properties,
IOUserClient ** handler )
{
const OSSymbol *userClientClass = 0;
IOUserClient *client;
OSObject *temp;
if (kIOReturnSuccess == newUserClient( owningTask, securityID, type, handler ))
return kIOReturnSuccess;
temp = getProperty(gIOUserClientClassKey);
if (temp) {
if (OSDynamicCast(OSSymbol, temp))
userClientClass = (const OSSymbol *) temp;
else if (OSDynamicCast(OSString, temp)) {
userClientClass = OSSymbol::withString((OSString *) temp);
if (userClientClass)
setProperty(kIOUserClientClassKey,
(OSObject *) userClientClass);
}
}
if (!userClientClass)
return kIOReturnUnsupported;
temp = OSMetaClass::allocClassWithName(userClientClass);
if (!temp)
return kIOReturnNoMemory;
if (OSDynamicCast(IOUserClient, temp))
client = (IOUserClient *) temp;
else {
temp->release();
return kIOReturnUnsupported;
}
if ( !client->initWithTask(owningTask, securityID, type, properties) ) {
client->release();
return kIOReturnBadArgument;
}
if ( !client->attach(this) ) {
client->release();
return kIOReturnUnsupported;
}
if ( !client->start(this) ) {
client->detach(this);
client->release();
return kIOReturnUnsupported;
}
*handler = client;
return kIOReturnSuccess;
}
IOReturn IOService::newUserClient( task_t owningTask, void * securityID,
UInt32 type, IOUserClient ** handler )
{
return( kIOReturnUnsupported );
}
IOReturn IOService::requestProbe( IOOptionBits options )
{
return( kIOReturnUnsupported);
}
const char * IOService::stringFromReturn( IOReturn rtn )
{
static const IONamedValue IOReturn_values[] = {
{kIOReturnSuccess, "success" },
{kIOReturnError, "general error" },
{kIOReturnNoMemory, "memory allocation error" },
{kIOReturnNoResources, "resource shortage" },
{kIOReturnIPCError, "Mach IPC failure" },
{kIOReturnNoDevice, "no such device" },
{kIOReturnNotPrivileged, "privilege violation" },
{kIOReturnBadArgument, "invalid argument" },
{kIOReturnLockedRead, "device is read locked" },
{kIOReturnLockedWrite, "device is write locked" },
{kIOReturnExclusiveAccess, "device is exclusive access" },
{kIOReturnBadMessageID, "bad IPC message ID" },
{kIOReturnUnsupported, "unsupported function" },
{kIOReturnVMError, "virtual memory error" },
{kIOReturnInternalError, "internal driver error" },
{kIOReturnIOError, "I/O error" },
{kIOReturnCannotLock, "cannot acquire lock" },
{kIOReturnNotOpen, "device is not open" },
{kIOReturnNotReadable, "device is not readable" },
{kIOReturnNotWritable, "device is not writeable" },
{kIOReturnNotAligned, "alignment error" },
{kIOReturnBadMedia, "media error" },
{kIOReturnStillOpen, "device is still open" },
{kIOReturnRLDError, "rld failure" },
{kIOReturnDMAError, "DMA failure" },
{kIOReturnBusy, "device is busy" },
{kIOReturnTimeout, "I/O timeout" },
{kIOReturnOffline, "device is offline" },
{kIOReturnNotReady, "device is not ready" },
{kIOReturnNotAttached, "device/channel is not attached" },
{kIOReturnNoChannels, "no DMA channels available" },
{kIOReturnNoSpace, "no space for data" },
{kIOReturnPortExists, "device port already exists" },
{kIOReturnCannotWire, "cannot wire physical memory" },
{kIOReturnNoInterrupt, "no interrupt attached" },
{kIOReturnNoFrames, "no DMA frames enqueued" },
{kIOReturnMessageTooLarge, "message is too large" },
{kIOReturnNotPermitted, "operation is not permitted" },
{kIOReturnNoPower, "device is without power" },
{kIOReturnNoMedia, "media is not present" },
{kIOReturnUnformattedMedia, "media is not formatted" },
{kIOReturnUnsupportedMode, "unsupported mode" },
{kIOReturnUnderrun, "data underrun" },
{kIOReturnOverrun, "data overrun" },
{kIOReturnDeviceError, "device error" },
{kIOReturnNoCompletion, "no completion routine" },
{kIOReturnAborted, "operation was aborted" },
{kIOReturnNoBandwidth, "bus bandwidth would be exceeded" },
{kIOReturnNotResponding, "device is not responding" },
{kIOReturnInvalid, "unanticipated driver error" },
{0, NULL }
};
return IOFindNameForValue(rtn, IOReturn_values);
}
int IOService::errnoFromReturn( IOReturn rtn )
{
if (unix_err(err_get_code(rtn)) == rtn)
return err_get_code(rtn);
switch(rtn) {
case kIOReturnSuccess:
return(0);
case kIOReturnNoMemory:
return(ENOMEM);
case kIOReturnNoDevice:
return(ENXIO);
case kIOReturnVMError:
return(EFAULT);
case kIOReturnNotPermitted:
return(EPERM);
case kIOReturnNotPrivileged:
return(EACCES);
case kIOReturnIOError:
return(EIO);
case kIOReturnNotWritable:
return(EROFS);
case kIOReturnBadArgument:
return(EINVAL);
case kIOReturnUnsupported:
return(ENOTSUP);
case kIOReturnBusy:
return(EBUSY);
case kIOReturnNoPower:
return(EPWROFF);
case kIOReturnDeviceError:
return(EDEVERR);
case kIOReturnTimeout:
return(ETIMEDOUT);
case kIOReturnMessageTooLarge:
return(EMSGSIZE);
case kIOReturnNoSpace:
return(ENOSPC);
case kIOReturnCannotLock:
return(ENOLCK);
case kIOReturnBadMessageID:
case kIOReturnNoCompletion:
case kIOReturnNotAligned:
return(EINVAL);
case kIOReturnNotReady:
return(EBUSY);
case kIOReturnRLDError:
return(EBADMACHO);
case kIOReturnPortExists:
case kIOReturnStillOpen:
return(EEXIST);
case kIOReturnExclusiveAccess:
case kIOReturnLockedRead:
case kIOReturnLockedWrite:
case kIOReturnNotOpen:
case kIOReturnNotReadable:
return(EACCES);
case kIOReturnCannotWire:
case kIOReturnNoResources:
return(ENOMEM);
case kIOReturnAborted:
case kIOReturnOffline:
case kIOReturnNotResponding:
return(EBUSY);
case kIOReturnBadMedia:
case kIOReturnNoMedia:
case kIOReturnNotAttached:
case kIOReturnUnformattedMedia:
return(ENXIO); case kIOReturnDMAError:
case kIOReturnOverrun:
case kIOReturnUnderrun:
return(EIO); case kIOReturnNoBandwidth:
case kIOReturnNoChannels:
case kIOReturnNoFrames:
case kIOReturnNoInterrupt:
return(EIO); case kIOReturnError:
case kIOReturnInternalError:
case kIOReturnInvalid:
return(EIO); case kIOReturnIPCError:
return(EIO); default:
return(EIO); }
}
IOReturn IOService::message( UInt32 type, IOService * provider,
void * argument )
{
return kIOReturnUnsupported;
}
IOItemCount IOService::getDeviceMemoryCount( void )
{
OSArray * array;
IOItemCount count;
array = OSDynamicCast( OSArray, getProperty( gIODeviceMemoryKey));
if( array)
count = array->getCount();
else
count = 0;
return( count);
}
IODeviceMemory * IOService::getDeviceMemoryWithIndex( unsigned int index )
{
OSArray * array;
IODeviceMemory * range;
array = OSDynamicCast( OSArray, getProperty( gIODeviceMemoryKey));
if( array)
range = (IODeviceMemory *) array->getObject( index );
else
range = 0;
return( range);
}
IOMemoryMap * IOService::mapDeviceMemoryWithIndex( unsigned int index,
IOOptionBits options )
{
IODeviceMemory * range;
IOMemoryMap * map;
range = getDeviceMemoryWithIndex( index );
if( range)
map = range->map( options );
else
map = 0;
return( map );
}
OSArray * IOService::getDeviceMemory( void )
{
return( OSDynamicCast( OSArray, getProperty( gIODeviceMemoryKey)));
}
void IOService::setDeviceMemory( OSArray * array )
{
setProperty( gIODeviceMemoryKey, array);
}
void IOService::
setCPUSnoopDelay(UInt32 __unused ns)
{
#if defined(__i386__) || defined(__x86_64__)
ml_set_maxsnoop(ns);
#endif
}
UInt32 IOService::
getCPUSnoopDelay()
{
#if defined(__i386__) || defined(__x86_64__)
return ml_get_maxsnoop();
#else
return 0;
#endif
}
#if defined(__i386__) || defined(__x86_64__)
static void
requireMaxCpuDelay(IOService * service, UInt32 ns, UInt32 delayType)
{
static const UInt kNoReplace = -1U; UInt replace = kNoReplace;
bool setCpuDelay = false;
IORecursiveLockLock(sCpuDelayLock);
UInt count = sCpuDelayData->getLength() / sizeof(CpuDelayEntry);
CpuDelayEntry *entries = (CpuDelayEntry *) sCpuDelayData->getBytesNoCopy();
IOService * holder = NULL;
if (ns) {
const CpuDelayEntry ne = {service, ns, delayType};
holder = service;
for (UInt i = 0; i < count; i++) {
IOService *thisService = entries[i].fService;
bool sameType = (delayType == entries[i].fDelayType);
if ((service == thisService) && sameType)
replace = i;
else if (!thisService) {
if (kNoReplace == replace)
replace = i;
}
else if (sameType) {
const UInt32 thisMax = entries[i].fMaxDelay;
if (thisMax < ns)
{
ns = thisMax;
holder = thisService;
}
}
}
setCpuDelay = true;
if (kNoReplace == replace)
sCpuDelayData->appendBytes(&ne, sizeof(ne));
else
entries[replace] = ne;
}
else {
ns = -1U;
for (UInt i = 0; i < count; i++) {
IOService *thisService = entries[i].fService;
if (thisService && (delayType == entries[i].fDelayType)) {
UInt32 thisMax = entries[i].fMaxDelay;
if (service == thisService)
replace = i;
else if (thisMax < ns) {
ns = thisMax;
holder = thisService;
}
}
}
if (kNoReplace != replace) {
entries[replace].fService = 0; setCpuDelay = true;
}
}
if (setCpuDelay)
{
if (holder && debug_boot_arg) {
strlcpy(sCPULatencyHolderName[delayType], holder->getName(), sizeof(sCPULatencyHolderName[delayType]));
}
if (delayType == kCpuDelayBusStall)
{
ml_set_maxbusdelay(ns);
}
else if (delayType == kCpuDelayInterrupt)
{
ml_set_maxintdelay(ns);
}
sCPULatencyHolder[delayType]->setValue(holder ? holder->getRegistryEntryID() : 0);
sCPULatencySet [delayType]->setValue(ns);
OSArray * handlers = sCpuLatencyHandlers[delayType];
IOService * target;
if (handlers) for (unsigned int idx = 0;
(target = (IOService *) handlers->getObject(idx));
idx++)
{
target->callPlatformFunction(sCPULatencyFunctionName[delayType], false,
(void *) (uintptr_t) ns, holder,
NULL, NULL);
}
}
IORecursiveLockUnlock(sCpuDelayLock);
}
static IOReturn
setLatencyHandler(UInt32 delayType, IOService * target, bool enable)
{
IOReturn result = kIOReturnNotFound;
OSArray * array;
unsigned int idx;
IORecursiveLockLock(sCpuDelayLock);
do
{
if (enable && !sCpuLatencyHandlers[delayType])
sCpuLatencyHandlers[delayType] = OSArray::withCapacity(4);
array = sCpuLatencyHandlers[delayType];
if (!array)
break;
idx = array->getNextIndexOfObject(target, 0);
if (!enable)
{
if (-1U != idx)
{
array->removeObject(idx);
result = kIOReturnSuccess;
}
}
else
{
if (-1U != idx) {
result = kIOReturnExclusiveAccess;
break;
}
array->setObject(target);
UInt count = sCpuDelayData->getLength() / sizeof(CpuDelayEntry);
CpuDelayEntry *entries = (CpuDelayEntry *) sCpuDelayData->getBytesNoCopy();
UInt32 ns = -1U; IOService * holder = NULL;
for (UInt i = 0; i < count; i++) {
if (entries[i].fService
&& (delayType == entries[i].fDelayType)
&& (entries[i].fMaxDelay < ns)) {
ns = entries[i].fMaxDelay;
holder = entries[i].fService;
}
}
target->callPlatformFunction(sCPULatencyFunctionName[delayType], false,
(void *) (uintptr_t) ns, holder,
NULL, NULL);
result = kIOReturnSuccess;
}
}
while (false);
IORecursiveLockUnlock(sCpuDelayLock);
return (result);
}
#endif
void IOService::
requireMaxBusStall(UInt32 __unused ns)
{
#if defined(__i386__) || defined(__x86_64__)
requireMaxCpuDelay(this, ns, kCpuDelayBusStall);
#endif
}
void IOService::
requireMaxInterruptDelay(uint32_t __unused ns)
{
#if defined(__i386__) || defined(__x86_64__)
requireMaxCpuDelay(this, ns, kCpuDelayInterrupt);
#endif
}
IOReturn IOService::resolveInterrupt(IOService *nub, int source)
{
IOInterruptController *interruptController;
OSArray *array;
OSData *data;
OSSymbol *interruptControllerName;
long numSources;
IOInterruptSource *interruptSources;
array = OSDynamicCast(OSArray, nub->getProperty(gIOInterruptControllersKey));
if (array == 0) return kIOReturnNoResources;
if (nub->_interruptSources == 0) {
numSources = array->getCount();
interruptSources = (IOInterruptSource *)IOMalloc(numSources * sizeof(IOInterruptSource));
if (interruptSources == 0) return kIOReturnNoMemory;
bzero(interruptSources, numSources * sizeof(IOInterruptSource));
nub->_numInterruptSources = numSources;
nub->_interruptSources = interruptSources;
return kIOReturnSuccess;
}
interruptControllerName = OSDynamicCast(OSSymbol,array->getObject(source));
if (interruptControllerName == 0) return kIOReturnNoResources;
interruptController = getPlatform()->lookUpInterruptController(interruptControllerName);
if (interruptController == 0) return kIOReturnNoResources;
array = OSDynamicCast(OSArray, nub->getProperty(gIOInterruptSpecifiersKey));
if (array == 0) return kIOReturnNoResources;
data = OSDynamicCast(OSData, array->getObject(source));
if (data == 0) return kIOReturnNoResources;
interruptSources = nub->_interruptSources;
interruptSources[source].interruptController = interruptController;
interruptSources[source].vectorData = data;
return kIOReturnSuccess;
}
IOReturn IOService::lookupInterrupt(int source, bool resolve, IOInterruptController **interruptController)
{
IOReturn ret;
if (_interruptSources == 0) {
ret = resolveInterrupt(this, source);
if (ret != kIOReturnSuccess) return ret;
}
if ((source < 0) || (source >= _numInterruptSources))
return kIOReturnNoInterrupt;
*interruptController = _interruptSources[source].interruptController;
if (*interruptController == NULL) {
if (!resolve) return kIOReturnNoInterrupt;
ret = resolveInterrupt(this, source);
if (ret != kIOReturnSuccess) return ret;
*interruptController = _interruptSources[source].interruptController;
}
return kIOReturnSuccess;
}
IOReturn IOService::registerInterrupt(int source, OSObject *target,
IOInterruptAction handler,
void *refCon)
{
IOInterruptController *interruptController;
IOReturn ret;
ret = lookupInterrupt(source, true, &interruptController);
if (ret != kIOReturnSuccess) return ret;
return interruptController->registerInterrupt(this, source, target,
(IOInterruptHandler)handler,
refCon);
}
IOReturn IOService::unregisterInterrupt(int source)
{
IOInterruptController *interruptController;
IOReturn ret;
ret = lookupInterrupt(source, false, &interruptController);
if (ret != kIOReturnSuccess) return ret;
return interruptController->unregisterInterrupt(this, source);
}
IOReturn IOService::addInterruptStatistics(IOInterruptAccountingData * statistics, int source)
{
IOReportLegend * legend = NULL;
IOInterruptAccountingData * oldValue = NULL;
IOInterruptAccountingReporter * newArray = NULL;
char subgroupName[64];
int newArraySize = 0;
int i = 0;
if (source < 0) {
return kIOReturnBadArgument;
}
if (source > IA_INDEX_MAX) {
panic("addInterruptStatistics called for an excessively large index (%d)", source);
}
IOLockLock(reserved->interruptStatisticsLock);
if (!reserved->interruptStatisticsArray) {
reserved->interruptStatisticsArray = IONew(IOInterruptAccountingReporter, 1);
assert(reserved->interruptStatisticsArray);
reserved->interruptStatisticsArrayCount = 1;
bzero(reserved->interruptStatisticsArray, sizeof(*reserved->interruptStatisticsArray));
}
if (source >= reserved->interruptStatisticsArrayCount) {
newArraySize = (reserved->interruptStatisticsArrayCount << 1);
while (newArraySize <= source)
newArraySize = (newArraySize << 1);
newArray = IONew(IOInterruptAccountingReporter, newArraySize);
assert(newArray);
bzero(newArray, newArraySize * sizeof(*newArray));
memcpy(newArray, reserved->interruptStatisticsArray, reserved->interruptStatisticsArrayCount * sizeof(*newArray));
IODelete(reserved->interruptStatisticsArray, IOInterruptAccountingReporter, reserved->interruptStatisticsArrayCount);
reserved->interruptStatisticsArray = newArray;
reserved->interruptStatisticsArrayCount = newArraySize;
}
if (!reserved->interruptStatisticsArray[source].reporter) {
reserved->interruptStatisticsArray[source].reporter = IOSimpleReporter::with(this, kIOReportCategoryPower, kIOReportUnitNone);
for (i = 0; i < IA_NUM_INTERRUPT_ACCOUNTING_STATISTICS; i++) {
if (IA_GET_STATISTIC_ENABLED(i))
reserved->interruptStatisticsArray[source].reporter->addChannel(IA_GET_CHANNEL_ID(source, i), kInterruptAccountingStatisticNameArray[i]);
}
legend = IOReportLegend::with(OSDynamicCast(OSArray, getProperty(kIOReportLegendKey)));
snprintf(subgroupName, sizeof(subgroupName), "%s %d", getName(), source);
subgroupName[sizeof(subgroupName) - 1] = 0;
legend->addReporterLegend(reserved->interruptStatisticsArray[source].reporter, kInterruptAccountingGroupName, subgroupName);
setProperty(kIOReportLegendKey, legend->getLegend());
legend->release();
setProperty(kIOReportLegendPublicKey, true);
}
oldValue = reserved->interruptStatisticsArray[source].statistics;
if (oldValue) {
panic("addInterruptStatistics call for index %d would have clobbered existing statistics", source);
}
reserved->interruptStatisticsArray[source].statistics = statistics;
interruptAccountingDataInheritChannels(reserved->interruptStatisticsArray[source].statistics, reserved->interruptStatisticsArray[source].reporter);
IOLockUnlock(reserved->interruptStatisticsLock);
return kIOReturnSuccess;
}
IOReturn IOService::removeInterruptStatistics(int source)
{
IOInterruptAccountingData * value = NULL;
if (source < 0) {
return kIOReturnBadArgument;
}
IOLockLock(reserved->interruptStatisticsLock);
if (source >= reserved->interruptStatisticsArrayCount) {
panic("removeInterruptStatistics called for index %d, which was never registered", source);
}
assert(reserved->interruptStatisticsArray);
value = reserved->interruptStatisticsArray[source].statistics;
if (!value) {
panic("removeInterruptStatistics called for empty index %d", source);
}
interruptAccountingDataUpdateChannels(reserved->interruptStatisticsArray[source].statistics, reserved->interruptStatisticsArray[source].reporter);
reserved->interruptStatisticsArray[source].statistics = NULL;
IOLockUnlock(reserved->interruptStatisticsLock);
return kIOReturnSuccess;
}
IOReturn IOService::getInterruptType(int source, int *interruptType)
{
IOInterruptController *interruptController;
IOReturn ret;
ret = lookupInterrupt(source, true, &interruptController);
if (ret != kIOReturnSuccess) return ret;
return interruptController->getInterruptType(this, source, interruptType);
}
IOReturn IOService::enableInterrupt(int source)
{
IOInterruptController *interruptController;
IOReturn ret;
ret = lookupInterrupt(source, false, &interruptController);
if (ret != kIOReturnSuccess) return ret;
return interruptController->enableInterrupt(this, source);
}
IOReturn IOService::disableInterrupt(int source)
{
IOInterruptController *interruptController;
IOReturn ret;
ret = lookupInterrupt(source, false, &interruptController);
if (ret != kIOReturnSuccess) return ret;
return interruptController->disableInterrupt(this, source);
}
IOReturn IOService::causeInterrupt(int source)
{
IOInterruptController *interruptController;
IOReturn ret;
ret = lookupInterrupt(source, false, &interruptController);
if (ret != kIOReturnSuccess) return ret;
return interruptController->causeInterrupt(this, source);
}
IOReturn IOService::configureReport(IOReportChannelList *channelList,
IOReportConfigureAction action,
void *result,
void *destination)
{
unsigned cnt;
for (cnt = 0; cnt < channelList->nchannels; cnt++) {
if ( channelList->channels[cnt].channel_id == kPMPowerStatesChID ) {
if (pwrMgt) configurePowerStatesReport(action, result);
else return kIOReturnUnsupported;
}
else if ( channelList->channels[cnt].channel_id == kPMCurrStateChID ) {
if (pwrMgt) configureSimplePowerReport(action, result);
else return kIOReturnUnsupported;
}
}
IOLockLock(reserved->interruptStatisticsLock);
for (cnt = 0; cnt < (unsigned) reserved->interruptStatisticsArrayCount; cnt++) {
if (reserved->interruptStatisticsArray[cnt].reporter) {
if (reserved->interruptStatisticsArray[cnt].statistics)
interruptAccountingDataUpdateChannels(reserved->interruptStatisticsArray[cnt].statistics, reserved->interruptStatisticsArray[cnt].reporter);
reserved->interruptStatisticsArray[cnt].reporter->configureReport(channelList, action, result, destination);
}
}
IOLockUnlock(reserved->interruptStatisticsLock);
return kIOReturnSuccess;
}
IOReturn IOService::updateReport(IOReportChannelList *channelList,
IOReportUpdateAction action,
void *result,
void *destination)
{
unsigned cnt;
for (cnt = 0; cnt < channelList->nchannels; cnt++) {
if ( channelList->channels[cnt].channel_id == kPMPowerStatesChID ) {
if (pwrMgt) updatePowerStatesReport(action, result, destination);
else return kIOReturnUnsupported;
}
else if ( channelList->channels[cnt].channel_id == kPMCurrStateChID ) {
if (pwrMgt) updateSimplePowerReport(action, result, destination);
else return kIOReturnUnsupported;
}
}
IOLockLock(reserved->interruptStatisticsLock);
for (cnt = 0; cnt < (unsigned) reserved->interruptStatisticsArrayCount; cnt++) {
if (reserved->interruptStatisticsArray[cnt].reporter) {
if (reserved->interruptStatisticsArray[cnt].statistics)
interruptAccountingDataUpdateChannels(reserved->interruptStatisticsArray[cnt].statistics, reserved->interruptStatisticsArray[cnt].reporter);
reserved->interruptStatisticsArray[cnt].reporter->updateReport(channelList, action, result, destination);
}
}
IOLockUnlock(reserved->interruptStatisticsLock);
return kIOReturnSuccess;
}
uint64_t IOService::getAuthorizationID( void )
{
return reserved->authorizationID;
}
IOReturn IOService::setAuthorizationID( uint64_t authorizationID )
{
OSObject * entitlement;
IOReturn status;
entitlement = IOUserClient::copyClientEntitlement( current_task( ), "com.apple.private.iokit.IOServiceSetAuthorizationID" );
if ( entitlement )
{
if ( entitlement == kOSBooleanTrue )
{
reserved->authorizationID = authorizationID;
status = kIOReturnSuccess;
}
else
{
status = kIOReturnNotPrivileged;
}
entitlement->release( );
}
else
{
status = kIOReturnNotPrivileged;
}
return status;
}
#if __LP64__
OSMetaClassDefineReservedUsed(IOService, 0);
OSMetaClassDefineReservedUsed(IOService, 1);
OSMetaClassDefineReservedUnused(IOService, 2);
OSMetaClassDefineReservedUnused(IOService, 3);
OSMetaClassDefineReservedUnused(IOService, 4);
OSMetaClassDefineReservedUnused(IOService, 5);
OSMetaClassDefineReservedUnused(IOService, 6);
OSMetaClassDefineReservedUnused(IOService, 7);
#else
OSMetaClassDefineReservedUsed(IOService, 0);
OSMetaClassDefineReservedUsed(IOService, 1);
OSMetaClassDefineReservedUsed(IOService, 2);
OSMetaClassDefineReservedUsed(IOService, 3);
OSMetaClassDefineReservedUsed(IOService, 4);
OSMetaClassDefineReservedUsed(IOService, 5);
OSMetaClassDefineReservedUsed(IOService, 6);
OSMetaClassDefineReservedUsed(IOService, 7);
#endif
OSMetaClassDefineReservedUnused(IOService, 8);
OSMetaClassDefineReservedUnused(IOService, 9);
OSMetaClassDefineReservedUnused(IOService, 10);
OSMetaClassDefineReservedUnused(IOService, 11);
OSMetaClassDefineReservedUnused(IOService, 12);
OSMetaClassDefineReservedUnused(IOService, 13);
OSMetaClassDefineReservedUnused(IOService, 14);
OSMetaClassDefineReservedUnused(IOService, 15);
OSMetaClassDefineReservedUnused(IOService, 16);
OSMetaClassDefineReservedUnused(IOService, 17);
OSMetaClassDefineReservedUnused(IOService, 18);
OSMetaClassDefineReservedUnused(IOService, 19);
OSMetaClassDefineReservedUnused(IOService, 20);
OSMetaClassDefineReservedUnused(IOService, 21);
OSMetaClassDefineReservedUnused(IOService, 22);
OSMetaClassDefineReservedUnused(IOService, 23);
OSMetaClassDefineReservedUnused(IOService, 24);
OSMetaClassDefineReservedUnused(IOService, 25);
OSMetaClassDefineReservedUnused(IOService, 26);
OSMetaClassDefineReservedUnused(IOService, 27);
OSMetaClassDefineReservedUnused(IOService, 28);
OSMetaClassDefineReservedUnused(IOService, 29);
OSMetaClassDefineReservedUnused(IOService, 30);
OSMetaClassDefineReservedUnused(IOService, 31);
OSMetaClassDefineReservedUnused(IOService, 32);
OSMetaClassDefineReservedUnused(IOService, 33);
OSMetaClassDefineReservedUnused(IOService, 34);
OSMetaClassDefineReservedUnused(IOService, 35);
OSMetaClassDefineReservedUnused(IOService, 36);
OSMetaClassDefineReservedUnused(IOService, 37);
OSMetaClassDefineReservedUnused(IOService, 38);
OSMetaClassDefineReservedUnused(IOService, 39);
OSMetaClassDefineReservedUnused(IOService, 40);
OSMetaClassDefineReservedUnused(IOService, 41);
OSMetaClassDefineReservedUnused(IOService, 42);
OSMetaClassDefineReservedUnused(IOService, 43);
OSMetaClassDefineReservedUnused(IOService, 44);
OSMetaClassDefineReservedUnused(IOService, 45);
OSMetaClassDefineReservedUnused(IOService, 46);
OSMetaClassDefineReservedUnused(IOService, 47);