Loading...
--- xnu/xnu-344.26/libkern/c++/OSMetaClass.cpp
+++ xnu/xnu-792.25.20/libkern/c++/OSMetaClass.cpp
@@ -22,7 +22,6 @@
/* OSMetaClass.cpp created by gvdl on Fri 1998-11-17 */
#include <string.h>
-#include <sys/systm.h>
#include <libkern/OSReturn.h>
@@ -39,10 +38,13 @@
#include <libkern/c++/OSLib.h>
#include <libkern/OSAtomic.h>
+#include <IOKit/pwr_mgt/RootDomain.h>
+#include <IOKit/IOMessage.h>
+
__BEGIN_DECLS
+#include <sys/systm.h>
#include <mach/mach_types.h>
-#include <mach/etap_events.h>
#include <kern/lock.h>
#include <kern/clock.h>
#include <kern/thread_call.h>
@@ -69,9 +71,9 @@
static const int kClassCapacityIncrement = 40;
static const int kKModCapacityIncrement = 10;
-static OSDictionary *sAllClassesDict, *sKModClassesDict;
-
-static mutex_t *loadLock;
+static OSDictionary *sAllClassesDict, *sKModClassesDict, *sSortedByClassesDict;
+
+static mutex_t *loadLock = 0;
static struct StalledData {
const char *kmodName;
OSReturn result;
@@ -81,6 +83,8 @@
} *sStalled;
static unsigned int sConsiderUnloadDelay = 60; /* secs */
+static bool unloadsEnabled = true; // set to false when system going to sleep
+static thread_call_t unloadCallout = 0;
static const char OSMetaClassBasePanicMsg[] =
"OSMetaClassBase::_RESERVEDOSMetaClassBase%d called\n";
@@ -273,7 +277,7 @@
sStalled->capacity += kKModCapacityIncrement;
memmove(sStalled->classes, oldStalled, oldSize);
- kfree((vm_offset_t)oldStalled, oldSize);
+ kfree(oldStalled, oldSize);
ACCUMSIZE(newSize - oldSize);
}
@@ -286,8 +290,10 @@
do {
OSCollectionIterator *iter;
- if (sAllClassesDict)
+ if (sAllClassesDict) {
sAllClassesDict->removeObject(className);
+ className->release();
+ }
iter = OSCollectionIterator::withCollection(sKModClassesDict);
if (!iter)
@@ -319,7 +325,6 @@
memmove(&sStalled->classes[i], &sStalled->classes[i+1],
(sStalled->count - i) * sizeof(OSMetaClass *));
}
- return;
}
}
@@ -345,7 +350,7 @@
void *OSMetaClass::preModLoad(const char *kmodName)
{
if (!loadLock) {
- loadLock = mutex_alloc(ETAP_IO_AHA);
+ loadLock = mutex_alloc(0);
mutex_lock(loadLock);
}
else
@@ -356,7 +361,7 @@
sStalled->classes = (OSMetaClass **)
kalloc(kKModCapacityIncrement * sizeof(OSMetaClass *));
if (!sStalled->classes) {
- kfree((vm_offset_t) sStalled, sizeof(*sStalled));
+ kfree(sStalled, sizeof(*sStalled));
return 0;
}
ACCUMSIZE((kKModCapacityIncrement * sizeof(OSMetaClass *)) + sizeof(*sStalled));
@@ -381,6 +386,7 @@
{
OSReturn result = kOSReturnSuccess;
OSSet *kmodSet = 0;
+ OSSymbol *myname = 0;
if (!sStalled || loadHandle != sStalled) {
logError(kOSMetaClassInternal);
@@ -397,7 +403,8 @@
case kMakingDictionaries:
sKModClassesDict = OSDictionary::withCapacity(kKModCapacityIncrement);
sAllClassesDict = OSDictionary::withCapacity(kClassCapacityIncrement);
- if (!sAllClassesDict || !sKModClassesDict) {
+ sSortedByClassesDict = OSDictionary::withCapacity(kClassCapacityIncrement);
+ if (!sAllClassesDict || !sKModClassesDict || !sSortedByClassesDict) {
result = kOSMetaClassNoDicts;
break;
}
@@ -406,6 +413,7 @@
case kCompletedBootstrap:
{
unsigned int i;
+ myname = (OSSymbol *)OSSymbol::withCStringNoCopy(sStalled->kmodName);
if (!sStalled->count)
break; // Nothing to do so just get out
@@ -429,19 +437,20 @@
break;
}
- if (!sKModClassesDict->setObject(sStalled->kmodName, kmodSet)) {
+ if (!sKModClassesDict->setObject(myname, kmodSet)) {
result = kOSMetaClassNoInsKModSet;
break;
}
// Second pass symbolling strings and inserting classes in dictionary
- for (unsigned int i = 0; i < sStalled->count; i++) {
+ for (i = 0; i < sStalled->count; i++) {
OSMetaClass *me = sStalled->classes[i];
me->className =
OSSymbol::withCStringNoCopy((const char *) me->className);
sAllClassesDict->setObject(me->className, me);
kmodSet->setObject(me);
+ sSortedByClassesDict->setObject((const OSSymbol *)me, myname);
}
sBootstrapState = kCompletedBootstrap;
break;
@@ -455,12 +464,15 @@
if (kmodSet)
kmodSet->release();
+ if (myname)
+ myname->release();
+
if (sStalled) {
ACCUMSIZE(-(sStalled->capacity * sizeof(OSMetaClass *)
+ sizeof(*sStalled)));
- kfree((vm_offset_t) sStalled->classes,
+ kfree(sStalled->classes,
sStalled->capacity * sizeof(OSMetaClass *));
- kfree((vm_offset_t) sStalled, sizeof(*sStalled));
+ kfree(sStalled, sizeof(*sStalled));
sStalled = 0;
}
@@ -491,7 +503,7 @@
bool result = false;
if (!loadLock) {
- loadLock = mutex_alloc(ETAP_IO_AHA);
+ loadLock = mutex_alloc(0);
mutex_lock(loadLock);
}
else
@@ -550,6 +562,36 @@
iter->release();
}
+
+
+extern "C" {
+
+IOReturn OSMetaClassSystemSleepOrWake(UInt32 messageType)
+{
+ mutex_lock(loadLock);
+
+ /* If the system is going to sleep, cancel the reaper thread timer
+ * and mark unloads disabled in case it just fired but hasn't
+ * taken the lock yet. If we are coming back from sleep, just
+ * set unloads enabled; IOService's normal operation will cause
+ * unloads to be considered soon enough.
+ */
+ if (messageType == kIOMessageSystemWillSleep) {
+ if (unloadCallout) {
+ thread_call_cancel(unloadCallout);
+ }
+ unloadsEnabled = false;
+ } else if (messageType == kIOMessageSystemHasPoweredOn) {
+ unloadsEnabled = true;
+ }
+ mutex_unlock(loadLock);
+
+ return kIOReturnSuccess;
+}
+
+};
+
+extern "C" kern_return_t kmod_unload_cache(void);
static void _OSMetaClassConsiderUnloads(thread_call_param_t p0,
thread_call_param_t p1)
@@ -564,6 +606,11 @@
bool didUnload;
mutex_lock(loadLock);
+
+ if (!unloadsEnabled) {
+ mutex_unlock(loadLock);
+ return;
+ }
do {
@@ -611,11 +658,12 @@
} while (didUnload);
mutex_unlock(loadLock);
+
+ kmod_unload_cache();
}
void OSMetaClass::considerUnloads()
{
- static thread_call_t unloadCallout;
AbsoluteTime when;
mutex_lock(loadLock);
@@ -767,6 +815,11 @@
return superClassLink;
}
+const OSSymbol *OSMetaClass::getKmodName() const
+{
+ return (const OSSymbol *)sSortedByClassesDict->getObject((OSSymbol *)this);
+}
+
unsigned int OSMetaClass::getInstanceCount() const
{
return instanceCount;
@@ -798,31 +851,50 @@
OSDictionary * OSMetaClass::getClassDictionary()
{
- return sAllClassesDict;
+ panic("OSMetaClass::getClassDictionary(): Obsoleted\n");
+ return 0;
}
bool OSMetaClass::serialize(OSSerialize *s) const
{
- OSDictionary * dict;
- OSNumber * off;
- bool ok = false;
-
- if (s->previouslySerialized(this)) return true;
-
- dict = 0;// IODictionary::withCapacity(2);
- off = OSNumber::withNumber(getInstanceCount(), 32);
-
- if (dict) {
- dict->setObject("InstanceCount", off );
- ok = dict->serialize(s);
- } else if( off)
- ok = off->serialize(s);
-
- if (dict)
- dict->release();
- if (off)
- off->release();
-
- return ok;
-}
-
+ panic("OSMetaClass::serialize(): Obsoleted\n");
+ return false;
+}
+
+void OSMetaClass::serializeClassDictionary(OSDictionary *serializeDictionary)
+{
+ OSDictionary *classDict;
+
+ classDict = OSDictionary::withCapacity(sAllClassesDict->getCount());
+ if (!classDict)
+ return;
+
+ mutex_lock(loadLock);
+ do {
+ OSCollectionIterator *classes;
+ const OSSymbol *className;
+
+ classes = OSCollectionIterator::withCollection(sAllClassesDict);
+ if (!classes)
+ break;
+
+ while ((className = (const OSSymbol *) classes->getNextObject())) {
+ const OSMetaClass *meta;
+ OSNumber *count;
+
+ meta = (OSMetaClass *) sAllClassesDict->getObject(className);
+ count = OSNumber::withNumber(meta->getInstanceCount(), 32);
+ if (count) {
+ classDict->setObject(className, count);
+ count->release();
+ }
+ }
+ classes->release();
+
+ serializeDictionary->setObject("Classes", classDict);
+ } while (0);
+
+ mutex_unlock(loadLock);
+
+ classDict->release();
+}