Loading...
libkern/c++/OSObject.cpp xnu-344 xnu-4570.51.1
--- xnu/xnu-344/libkern/c++/OSObject.cpp
+++ xnu/xnu-4570.51.1/libkern/c++/OSObject.cpp
@@ -1,43 +1,50 @@
 /*
- * Copyright (c) 2000 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 2000 Apple Inc. All rights reserved.
  *
- * @APPLE_LICENSE_HEADER_START@
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 
- * The contents of this file constitute Original Code as defined in and
- * are subject to the Apple Public Source License Version 1.1 (the
- * "License").  You may not use this file except in compliance with the
- * License.  Please obtain a copy of the License at
- * http://www.apple.com/publicsource and read it before using this file.
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. The rights granted to you under the License
+ * may not be used to create, or enable the creation or redistribution of,
+ * unlawful or unlicensed copies of an Apple operating system, or to
+ * circumvent, violate, or enable the circumvention or violation of, any
+ * terms of an Apple operating system software license agreement.
  * 
- * This Original Code and all software distributed under the License are
- * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
- * License for the specific language governing rights and limitations
- * under the License.
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
  * 
- * @APPLE_LICENSE_HEADER_END@
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  */
 /* OSObject.cpp created by gvdl on Fri 1998-11-17 */
 
 #include <libkern/c++/OSObject.h>
+#include <libkern/c++/OSString.h>
+#include <libkern/c++/OSArray.h>
 #include <libkern/c++/OSSerialize.h>
 #include <libkern/c++/OSLib.h>
+#include <libkern/OSDebug.h>
 #include <libkern/c++/OSCPPDebug.h>
+#include <IOKit/IOKitDebug.h>
 #include <libkern/OSAtomic.h>
 
 #include <libkern/c++/OSCollection.h>
+
+#include <kern/queue.h>
 
 __BEGIN_DECLS
 int debug_ivars_size;
 __END_DECLS
 
-#if OSALLOCDEBUG
-#define ACCUMSIZE(s) do { debug_ivars_size += (s); } while(0)
-#else
-#define ACCUMSIZE(s)
-#endif
 
 // OSDefineMetaClassAndAbstractStructors(OSObject, 0);
 /* Class global data */
@@ -46,8 +53,6 @@
 const OSMetaClass * const OSObject::superClass = 0;
 
 /* Class member functions - Can't use defaults */
-OSObject::OSObject()			{ retainCount = 1; }
-OSObject::OSObject(const OSMetaClass *)	{ retainCount = 1; }
 OSObject::~OSObject()			{ }
 const OSMetaClass * OSObject::getMetaClass() const
     { return &gMetaClass; }
@@ -75,22 +80,6 @@
 OSMetaClassDefineReservedUnused(OSObject, 13);
 OSMetaClassDefineReservedUnused(OSObject, 14);
 OSMetaClassDefineReservedUnused(OSObject, 15);
-OSMetaClassDefineReservedUnused(OSObject, 16);
-OSMetaClassDefineReservedUnused(OSObject, 17);
-OSMetaClassDefineReservedUnused(OSObject, 18);
-OSMetaClassDefineReservedUnused(OSObject, 19);
-OSMetaClassDefineReservedUnused(OSObject, 20);
-OSMetaClassDefineReservedUnused(OSObject, 21);
-OSMetaClassDefineReservedUnused(OSObject, 22);
-OSMetaClassDefineReservedUnused(OSObject, 23);
-OSMetaClassDefineReservedUnused(OSObject, 24);
-OSMetaClassDefineReservedUnused(OSObject, 25);
-OSMetaClassDefineReservedUnused(OSObject, 26);
-OSMetaClassDefineReservedUnused(OSObject, 27);
-OSMetaClassDefineReservedUnused(OSObject, 28);
-OSMetaClassDefineReservedUnused(OSObject, 29);
-OSMetaClassDefineReservedUnused(OSObject, 30);
-OSMetaClassDefineReservedUnused(OSObject, 31);
 
 static const char *getClassName(const OSObject *obj)
 {
@@ -98,18 +87,6 @@
     return (meta) ? meta->getClassName() : "unknown class?";
 }
 
-bool OSObject::init()
-    { return true; }
-
-void OSObject::free()
-{
-    const OSMetaClass *meta = getMetaClass();
-
-    if (meta)
-	meta->instanceDestructed();
-    delete this;
-}
-
 int OSObject::getRetainCount() const
 {
     return (int) ((UInt16) retainCount);
@@ -117,42 +94,44 @@
 
 void OSObject::taggedRetain(const void *tag) const
 {
-#if !DEBUG
     volatile UInt32 *countP = (volatile UInt32 *) &retainCount;
     UInt32 inc = 1;
     UInt32 origCount;
     UInt32 newCount;
 
-    // Increment the collecion bucket.
+    // Increment the collection bucket.
     if ((const void *) OSTypeID(OSCollection) == tag)
 	inc |= (1UL<<16);
 
     do {
 	origCount = *countP;
-	if (-1UL == origCount)
-	    // @@@ Pinot: panic("Attempting to retain a freed object");
-	    return;
+        if ( ((UInt16) origCount | 0x1) == 0xffff ) {
+            const char *msg;
+            if (origCount & 0x1) {
+                // If count == 0xffff that means we are freeing now so we can
+                // just return obviously somebody is cleaning up dangling
+                // references.
+                msg = "Attempting to retain a freed object";
+            }
+            else {
+                // If count == 0xfffe then we have wrapped our reference count.
+                // We should stop counting now as this reference must be
+                // leaked rather than accidently wrapping around the clock and
+                // freeing a very active object later.
+
+#if !DEBUG
+		break;	// Break out of update loop which pegs the reference
+#else /* DEBUG */
+                // @@@ gvdl: eventually need to make this panic optional
+                // based on a boot argument i.e. debug= boot flag
+                msg = "About to wrap the reference count, reference leak?";
+#endif /* !DEBUG */
+            }
+            panic("OSObject::refcount: %s", msg);
+        }
 
 	newCount = origCount + inc;
-    } while (!OSCompareAndSwap(origCount, newCount, (UInt32 *) countP));
-#else
-    volatile UInt32 *countP = (volatile UInt32 *) &retainCount;
-    UInt32 inc = 1;
-    UInt32 origCount;
-    UInt32 newCount;
-
-    // Increment the collecion bucket.
-    if ((const void *) OSTypeID(OSCollection) == tag)
-	inc |= (1UL<<16);
-
-    do {
-	origCount = *countP;
-	if (-1UL == origCount)
-	    return;	// We are freeing so leave now.
-
-	newCount = origCount + inc;
-    } while (!OSCompareAndSwap(origCount, newCount, (UInt32 *) countP));
-#endif
+    } while (!OSCompareAndSwap(origCount, newCount, const_cast<UInt32 *>(countP)));
 }
 
 void OSObject::taggedRelease(const void *tag) const
@@ -162,86 +141,69 @@
 
 void OSObject::taggedRelease(const void *tag, const int when) const
 {
-#if !DEBUG
     volatile UInt32 *countP = (volatile UInt32 *) &retainCount;
     UInt32 dec = 1;
     UInt32 origCount;
     UInt32 newCount;
     UInt32 actualCount;
 
-    // Increment the collecion bucket.
+    // Increment the collection bucket.
     if ((const void *) OSTypeID(OSCollection) == tag)
 	dec |= (1UL<<16);
 
     do {
 	origCount = *countP;
-	if (-1UL == origCount)
-	    return;	// We are freeing already leave now.
-
+        
+        if ( ((UInt16) origCount | 0x1) == 0xffff ) {
+            if (origCount & 0x1) {
+                // If count == 0xffff that means we are freeing now so we can
+                // just return obviously somebody is cleaning up some dangling
+                // references.  So we blow out immediately.
+                return;
+            }
+            else {
+                // If count == 0xfffe then we have wrapped our reference
+                // count.  We should stop counting now as this reference must be
+                // leaked rather than accidently freeing an active object later.
+
+#if !DEBUG
+		return;	// return out of function which pegs the reference
+#else /* DEBUG */
+                // @@@ gvdl: eventually need to make this panic optional
+                // based on a boot argument i.e. debug= boot flag
+                panic("OSObject::refcount: %s",
+                      "About to unreference a pegged object, reference leak?");
+#endif /* !DEBUG */
+            }
+        }
 	actualCount = origCount - dec;
-        if ((SInt16) actualCount < when)
-            newCount = (UInt32) -1;
+        if ((UInt16) actualCount < when)
+            newCount = 0xffff;
         else
             newCount = actualCount;
 
-    } while (!OSCompareAndSwap(origCount, newCount, (UInt32 *) countP));
+    } while (!OSCompareAndSwap(origCount, newCount, const_cast<UInt32 *>(countP)));
 
     //
     // This panic means that we have just attempted to release an object
-    // who's retain count has gone to less than the number of collections
+    // whose retain count has gone to less than the number of collections
     // it is a member off.  Take a panic immediately.
-    // In Fact the panic MAY not be a registry corruption but it is 
+    // In fact the panic MAY not be a registry corruption but it is 
     // ALWAYS the wrong thing to do.  I call it a registry corruption 'cause
     // the registry is the biggest single use of a network of collections.
     //
-    if ((UInt16) actualCount < (actualCount >> 16))
-	panic("A driver releasing a(n) %s has corrupted the registry\n",
-	    getClassName(this));
+// xxx - this error message is overly-specific;
+// xxx - any code in the kernel could trip this,
+// xxx - and it applies as noted to all collections, not just the registry
+    if ((UInt16) actualCount < (actualCount >> 16)) {
+        panic("A kext releasing a(n) %s has corrupted the registry.",
+            getClassName(this));
+    }
 
     // Check for a 'free' condition and that if we are first through
-    if ((UInt32) -1 == newCount)
-	((OSObject *) this)->free();
-#else
-    // @@@ Pinot:  Need to update the debug build release code.
-    volatile UInt32 *countP = (volatile UInt32 *) &retainCount;
-    UInt32 dec = 1;
-    UInt32 origCount;
-    UInt32 newCount;
-
-    // Increment the collecion bucket.
-    if ((const void *) OSTypeID(OSCollection) == tag)
-	dec |= (1UL<<16);
-
-    do {
-	origCount = *countP;
-	if (-1UL == origCount)
-	    return;	// We are freeing already leave now.
-
-	newCount = origCount - dec;
-    } while (!OSCompareAndSwap(origCount, newCount, (UInt32 *) countP));
-
-    //
-    // This panic means that we have just attempted to release an object
-    // who's retain count has gone to less than the number of collections
-    // it is a member off.  Take a panic immediately.
-    // In Fact the panic MAY not be a registry corruption but it is 
-    // ALWAYS the wrong thing to do.  I call it a registry corruption 'cause
-    // the registry is the biggest single use of a network of collections.
-    //
-    if ((UInt16) newCount < (newCount >> 16))
-	panic("A driver releasing a(n) %s has corrupted the registry\n",
-	    getClassName(this));
-
-    // Check for a release too many
-    if ((SInt16) newCount < 0)
-	panic("An object has had a release too many\n",
-	    getClassName(this));
-
-    // Check for a 'free' condition and that if we are first through
-    if ((SInt16) newCount < when
-    && OSCompareAndSwap(newCount, -1UL, (UInt32 *) countP))
-	((OSObject *) this)->free();
-#endif
+    if (newCount == 0xffff) {
+        (const_cast<OSObject *>(this))->free();
+    }
 }
 
 void OSObject::release() const
@@ -254,6 +216,18 @@
     taggedRetain(0);
 }
 
+extern "C" void
+osobject_retain(void * object)
+{
+    ((OSObject *)object)->retain();
+}
+
+extern "C" void
+osobject_release(void * object)
+{
+    ((OSObject *)object)->release();
+}
+
 void OSObject::release(int when) const
 {
     taggedRelease(0, when);
@@ -261,30 +235,87 @@
 
 bool OSObject::serialize(OSSerialize *s) const
 {
-    if (s->previouslySerialized(this)) return true;
-
-    if (!s->addXMLStartTag(this, "string")) return false;
-
-    if (!s->addString(getClassName(this))) return false;
-    if (!s->addString(" is not serializable")) return false;
-    
-    return s->addXMLEndTag("string");
+    char cstr[128];
+    bool ok;
+
+    snprintf(cstr, sizeof(cstr), "%s is not serializable", getClassName(this));
+
+    OSString * str;
+    str = OSString::withCStringNoCopy(cstr);
+    if (!str) return false;
+
+    ok = str->serialize(s);
+    str->release();
+
+    return (ok);
 }
 
 void *OSObject::operator new(size_t size)
 {
-    void *mem = (void *) kalloc(size);
+#if IOTRACKING
+    if (kIOTracking & gIOKitDebug) return (OSMetaClass::trackedNew(size));
+#endif
+
+    void * mem = kalloc_tag_bt(size, VM_KERN_MEMORY_LIBKERN);
     assert(mem);
     bzero(mem, size);
-
-    ACCUMSIZE(size);
-
-    return mem;
-}
-
-void OSObject::operator delete(void *mem, size_t size)
-{
-    kfree((vm_offset_t) mem, size);
-
-    ACCUMSIZE(-size);
-}
+    OSIVAR_ACCUMSIZE(size);
+
+    return (void *) mem;
+}
+
+void OSObject::operator delete(void * mem, size_t size)
+{
+    if (!mem) return;
+
+#if IOTRACKING
+    if (kIOTracking & gIOKitDebug) return (OSMetaClass::trackedDelete(mem, size));
+#endif
+
+    kfree(mem, size);
+    OSIVAR_ACCUMSIZE(-size);
+}
+
+bool OSObject::init()
+{
+#if IOTRACKING
+    if (kIOTracking & gIOKitDebug) getMetaClass()->trackedInstance(this);
+#endif
+    return true;
+}
+
+void OSObject::free()
+{
+    const OSMetaClass *meta = getMetaClass();
+
+    if (meta)
+    {
+	meta->instanceDestructed();
+#if IOTRACKING
+	if (kIOTracking & gIOKitDebug) getMetaClass()->trackedFree(this);
+#endif
+    }
+    delete this;
+}
+
+#if IOTRACKING
+void OSObject::trackingAccumSize(size_t size)
+{
+    if (kIOTracking & gIOKitDebug) getMetaClass()->trackedAccumSize(this, size);
+}
+#endif
+
+/* Class member functions - Can't use defaults */
+/* During constructor vtable is always OSObject's - can't call any subclass */
+
+OSObject::OSObject()
+{
+    retainCount = 1;
+//    if (kIOTracking & gIOKitDebug) getMetaClass()->trackedInstance(this);
+}
+
+OSObject::OSObject(const OSMetaClass *)
+{
+    retainCount = 1;
+//    if (kIOTracking & gIOKitDebug) getMetaClass()->trackedInstance(this);
+}