]> git.saurik.com Git - apple/xnu.git/blobdiff - libkern/c++/OSArray.cpp
xnu-4903.270.47.tar.gz
[apple/xnu.git] / libkern / c++ / OSArray.cpp
index 5089da70f349a8a8da9a04ec570c09b88ec76f64..92558ac4ccb828833deab6b36c4c31784688ade9 100644 (file)
@@ -2,7 +2,7 @@
  * Copyright (c) 2000 Apple Computer, Inc. All rights reserved.
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
- * 
+ *
  * 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
  * 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.
- * 
+ *
  * 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,
@@ -22,7 +22,7 @@
  * 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_OSREFERENCE_LICENSE_HEADER_END@
  */
 /* IOArray.m created by rsulack on Fri 12-Sep-1997 */
@@ -51,437 +51,512 @@ OSMetaClassDefineReservedUnused(OSArray, 7);
 #define EXT_CAST(obj) \
     reinterpret_cast<OSObject *>(const_cast<OSMetaClassBase *>(obj))
 
-bool OSArray::initWithCapacity(unsigned int inCapacity)
+bool
+OSArray::initWithCapacity(unsigned int inCapacity)
 {
-    unsigned int size;
+       unsigned int size;
 
-    if (!super::init())
-        return false;
+       if (!super::init()) {
+               return false;
+       }
 
-    // integer overflow check
-    if (inCapacity > (UINT_MAX / sizeof(const OSMetaClassBase*)))
-        return false;
+       // integer overflow check
+       if (inCapacity > (UINT_MAX / sizeof(const OSMetaClassBase*))) {
+               return false;
+       }
 
-    size = sizeof(const OSMetaClassBase *) * inCapacity;
-    array = (const OSMetaClassBase **) kalloc_container(size);
-    if (!array)
-        return false;
+       size = sizeof(const OSMetaClassBase *) * inCapacity;
+       array = (const OSMetaClassBase **) kalloc_container(size);
+       if (!array) {
+               return false;
+       }
 
-    count = 0;
-    capacity = inCapacity;
-    capacityIncrement = (inCapacity)? inCapacity : 16;
+       count = 0;
+       capacity = inCapacity;
+       capacityIncrement = (inCapacity)? inCapacity : 16;
 
-    bzero(array, size);
-    OSCONTAINER_ACCUMSIZE(size);
+       bzero(array, size);
+       OSCONTAINER_ACCUMSIZE(size);
 
-    return true;
+       return true;
 }
 
-bool OSArray::initWithObjects(const OSObject *objects[],
-                              unsigned int theCount,
-                              unsigned int theCapacity)
+bool
+OSArray::initWithObjects(const OSObject *objects[],
+    unsigned int theCount,
+    unsigned int theCapacity)
 {
-    unsigned int initCapacity;
-
-    if (!theCapacity)
-        initCapacity = theCount;
-    else if (theCount > theCapacity)
-        return false;
-    else
-        initCapacity = theCapacity;
+       unsigned int initCapacity;
+
+       if (!theCapacity) {
+               initCapacity = theCount;
+       } else if (theCount > theCapacity) {
+               return false;
+       } else {
+               initCapacity = theCapacity;
+       }
 
-    if (!objects || !initWithCapacity(initCapacity))
-        return false;
+       if (!objects || !initWithCapacity(initCapacity)) {
+               return false;
+       }
 
-    for ( unsigned int i = 0; i < theCount; i++ ) {
-        const OSMetaClassBase *newObject = *objects++;
+       for (unsigned int i = 0; i < theCount; i++) {
+               const OSMetaClassBase *newObject = *objects++;
 
-        if (!newObject)
-            return false;
+               if (!newObject) {
+                       return false;
+               }
 
-        array[count++] = newObject;
-        newObject->taggedRetain(OSTypeID(OSCollection));
-    }
+               array[count++] = newObject;
+               newObject->taggedRetain(OSTypeID(OSCollection));
+       }
 
-    return true;       
+       return true;
 }
 
-bool OSArray::initWithArray(const OSArray *anArray,
-                            unsigned int theCapacity)
+bool
+OSArray::initWithArray(const OSArray *anArray,
+    unsigned int theCapacity)
 {
-    if ( !anArray )
-        return false;
+       if (!anArray) {
+               return false;
+       }
 
-    return initWithObjects((const OSObject **) anArray->array,
-                           anArray->count, theCapacity);
+       return initWithObjects((const OSObject **) anArray->array,
+                  anArray->count, theCapacity);
 }
 
-OSArray *OSArray::withCapacity(unsigned int capacity)
+OSArray *
+OSArray::withCapacity(unsigned int capacity)
 {
-    OSArray *me = new OSArray;
+       OSArray *me = new OSArray;
 
-    if (me && !me->initWithCapacity(capacity)) {
-        me->release();
-        return 0;
-    }
+       if (me && !me->initWithCapacity(capacity)) {
+               me->release();
+               return 0;
+       }
 
-    return me;
+       return me;
 }
 
-OSArray *OSArray::withObjects(const OSObject *objects[],
-                              unsigned int count,
-                              unsigned int capacity)
+OSArray *
+OSArray::withObjects(const OSObject *objects[],
+    unsigned int count,
+    unsigned int capacity)
 {
-    OSArray *me = new OSArray;
+       OSArray *me = new OSArray;
 
-    if (me && !me->initWithObjects(objects, count, capacity)) {
-        me->release();
-        return 0;
-    }
+       if (me && !me->initWithObjects(objects, count, capacity)) {
+               me->release();
+               return 0;
+       }
 
-    return me;
+       return me;
 }
 
-OSArray *OSArray::withArray(const OSArray *array,
-                            unsigned int capacity)
+OSArray *
+OSArray::withArray(const OSArray *array,
+    unsigned int capacity)
 {
-    OSArray *me = new OSArray;
+       OSArray *me = new OSArray;
 
-    if (me && !me->initWithArray(array, capacity)) {
-        me->release();
-        return 0;
-    }
+       if (me && !me->initWithArray(array, capacity)) {
+               me->release();
+               return 0;
+       }
 
-    return me;
+       return me;
 }
 
-void OSArray::free()
+void
+OSArray::free()
 {
-    // Clear immutability - assumes the container is doing the right thing
-    (void) super::setOptions(0, kImmutable);
+       // Clear immutability - assumes the container is doing the right thing
+       (void) super::setOptions(0, kImmutable);
 
-    flushCollection();
+       flushCollection();
 
-    if (array) {
-        kfree(array, sizeof(const OSMetaClassBase *) * capacity);
-        OSCONTAINER_ACCUMSIZE( -(sizeof(const OSMetaClassBase *) * capacity) );
-    }
+       if (array) {
+               kfree(array, sizeof(const OSMetaClassBase *) * capacity);
+               OSCONTAINER_ACCUMSIZE( -(sizeof(const OSMetaClassBase *) * capacity));
+       }
 
-    super::free();
+       super::free();
 }
 
 
-unsigned int OSArray::getCount() const { return count; }
-unsigned int OSArray::getCapacity() const { return capacity; }
-unsigned int OSArray::getCapacityIncrement() const { return capacityIncrement; }
-unsigned int OSArray::setCapacityIncrement(unsigned int increment)
+unsigned int
+OSArray::getCount() const
 {
-    capacityIncrement = (increment)? increment : 16;
+       return count;
+}
+unsigned int
+OSArray::getCapacity() const
+{
+       return capacity;
+}
+unsigned int
+OSArray::getCapacityIncrement() const
+{
+       return capacityIncrement;
+}
+unsigned int
+OSArray::setCapacityIncrement(unsigned int increment)
+{
+       capacityIncrement = (increment)? increment : 16;
 
-    return capacityIncrement;
+       return capacityIncrement;
 }
 
-unsigned int OSArray::ensureCapacity(unsigned int newCapacity)
+unsigned int
+OSArray::ensureCapacity(unsigned int newCapacity)
 {
-    const OSMetaClassBase **newArray;
-    unsigned int finalCapacity;
-    vm_size_t    oldSize, newSize;
+       const OSMetaClassBase **newArray;
+       unsigned int finalCapacity;
+       vm_size_t    oldSize, newSize;
 
-    if (newCapacity <= capacity)
-        return capacity;
+       if (newCapacity <= capacity) {
+               return capacity;
+       }
 
-    // round up
-    finalCapacity = (((newCapacity - 1) / capacityIncrement) + 1)
-                * capacityIncrement;
+       // round up
+       finalCapacity = (((newCapacity - 1) / capacityIncrement) + 1)
+           * capacityIncrement;
 
-    // integer overflow check
-    if ((finalCapacity < newCapacity) || (finalCapacity > (UINT_MAX / sizeof(const OSMetaClassBase*))))
-        return capacity;
+       // integer overflow check
+       if ((finalCapacity < newCapacity) || (finalCapacity > (UINT_MAX / sizeof(const OSMetaClassBase*)))) {
+               return capacity;
+       }
 
-    newSize = sizeof(const OSMetaClassBase *) * finalCapacity;
+       newSize = sizeof(const OSMetaClassBase *) * finalCapacity;
 
-    newArray = (const OSMetaClassBase **) kallocp_container(&newSize);
-    if (newArray) {
-        // use all of the actual allocation size
-        finalCapacity = newSize / sizeof(const OSMetaClassBase *);
+       newArray = (const OSMetaClassBase **) kallocp_container(&newSize);
+       if (newArray) {
+               // use all of the actual allocation size
+               finalCapacity = newSize / sizeof(const OSMetaClassBase *);
 
-        oldSize = sizeof(const OSMetaClassBase *) * capacity;
+               oldSize = sizeof(const OSMetaClassBase *) * capacity;
 
-        OSCONTAINER_ACCUMSIZE(((size_t)newSize) - ((size_t)oldSize));
+               OSCONTAINER_ACCUMSIZE(((size_t)newSize) - ((size_t)oldSize));
 
-        bcopy(array, newArray, oldSize);
-        bzero(&newArray[capacity], newSize - oldSize);
-        kfree(array, oldSize);
-        array = newArray;
-        capacity = finalCapacity;
-    }
+               bcopy(array, newArray, oldSize);
+               bzero(&newArray[capacity], newSize - oldSize);
+               kfree(array, oldSize);
+               array = newArray;
+               capacity = finalCapacity;
+       }
 
-    return capacity;
+       return capacity;
 }
 
-void OSArray::flushCollection()
+void
+OSArray::flushCollection()
 {
-    unsigned int i;
+       unsigned int i;
 
-    haveUpdated();
-    for (i = 0; i < count; i++) {
-        array[i]->taggedRelease(OSTypeID(OSCollection));
-    }
-    count = 0;
+       haveUpdated();
+       for (i = 0; i < count; i++) {
+               array[i]->taggedRelease(OSTypeID(OSCollection));
+       }
+       count = 0;
 }
 
-bool OSArray::setObject(const OSMetaClassBase *anObject)
+bool
+OSArray::setObject(const OSMetaClassBase *anObject)
 {
-    return setObject(count, anObject);
+       return setObject(count, anObject);
 }
 
-bool OSArray::setObject(unsigned int index, const OSMetaClassBase *anObject)
+bool
+OSArray::setObject(unsigned int index, const OSMetaClassBase *anObject)
 {
-    unsigned int i;
-    unsigned int newCount = count + 1;
-
-    if ((index > count) || !anObject)
-        return false;
-
-    // do we need more space?
-    if (newCount > capacity && newCount > ensureCapacity(newCount))
-        return false;
-
-    haveUpdated();
-    if (index != count) {
-        for (i = count; i > index; i--)
-            array[i] = array[i-1];
-    }
-    array[index] = anObject;
-    anObject->taggedRetain(OSTypeID(OSCollection));
-    count++;
-
-    return true;
+       unsigned int i;
+       unsigned int newCount = count + 1;
+
+       if ((index > count) || !anObject) {
+               return false;
+       }
+
+       // do we need more space?
+       if (newCount > capacity && newCount > ensureCapacity(newCount)) {
+               return false;
+       }
+
+       haveUpdated();
+       if (index != count) {
+               for (i = count; i > index; i--) {
+                       array[i] = array[i - 1];
+               }
+       }
+       array[index] = anObject;
+       anObject->taggedRetain(OSTypeID(OSCollection));
+       count++;
+
+       return true;
 }
 
-bool OSArray::merge(const OSArray * otherArray)
+bool
+OSArray::merge(const OSArray * otherArray)
 {
-    unsigned int otherCount = otherArray->getCount();
-    unsigned int newCount = count + otherCount;
+       unsigned int otherCount = otherArray->getCount();
+       unsigned int newCount = count + otherCount;
 
-    if (!otherCount)
-        return true;
+       if (!otherCount) {
+               return true;
+       }
 
-    if (newCount < count)
-        return false;
+       if (newCount < count) {
+               return false;
+       }
 
-    // do we need more space?
-    if (newCount > capacity && newCount > ensureCapacity(newCount))
-        return false;
+       // do we need more space?
+       if (newCount > capacity && newCount > ensureCapacity(newCount)) {
+               return false;
+       }
 
-    haveUpdated();
-    for (unsigned int i = 0; i < otherCount; i++) {
-        const OSMetaClassBase *newObject = otherArray->getObject(i);
+       haveUpdated();
+       for (unsigned int i = 0; i < otherCount; i++) {
+               const OSMetaClassBase *newObject = otherArray->getObject(i);
 
-        array[count++] = newObject;
-        newObject->taggedRetain(OSTypeID(OSCollection));
-    }
+               array[count++] = newObject;
+               newObject->taggedRetain(OSTypeID(OSCollection));
+       }
 
-    return true;
+       return true;
 }
 
-void OSArray::
+void
+OSArray::
 replaceObject(unsigned int index, const OSMetaClassBase *anObject)
 {
-    const OSMetaClassBase *oldObject;
+       const OSMetaClassBase *oldObject;
 
-    if ((index >= count) || !anObject)
-        return;
+       if ((index >= count) || !anObject) {
+               return;
+       }
 
-    haveUpdated();
-    oldObject = array[index];
-    array[index] = anObject;
-    anObject->taggedRetain(OSTypeID(OSCollection));
+       haveUpdated();
+       oldObject = array[index];
+       array[index] = anObject;
+       anObject->taggedRetain(OSTypeID(OSCollection));
 
-    oldObject->taggedRelease(OSTypeID(OSCollection));
+       oldObject->taggedRelease(OSTypeID(OSCollection));
 }
 
-void OSArray::removeObject(unsigned int index)
+void
+OSArray::removeObject(unsigned int index)
 {
-    unsigned int i;
-    const OSMetaClassBase *oldObject;
+       unsigned int i;
+       const OSMetaClassBase *oldObject;
 
-    if (index >= count)
-        return;
+       if (index >= count) {
+               return;
+       }
 
-    haveUpdated();
-    oldObject = array[index];
+       haveUpdated();
+       oldObject = array[index];
 
-    count--;
-    for (i = index; i < count; i++)
-        array[i] = array[i+1];
+       count--;
+       for (i = index; i < count; i++) {
+               array[i] = array[i + 1];
+       }
 
-    oldObject->taggedRelease(OSTypeID(OSCollection));
+       oldObject->taggedRelease(OSTypeID(OSCollection));
 }
 
-bool OSArray::isEqualTo(const OSArray *anArray) const
+bool
+OSArray::isEqualTo(const OSArray *anArray) const
 {
-    unsigned int i;
-    
-    if ( this == anArray )
-        return true;
-    
-    if ( count != anArray->getCount() )
-        return false;
-
-    for ( i = 0; i < count; i++ ) {
-        if ( !array[i]->isEqualTo(anArray->getObject(i)) )
-            return false;
-    }
-
-    return true;
+       unsigned int i;
+
+       if (this == anArray) {
+               return true;
+       }
+
+       if (count != anArray->getCount()) {
+               return false;
+       }
+
+       for (i = 0; i < count; i++) {
+               if (!array[i]->isEqualTo(anArray->getObject(i))) {
+                       return false;
+               }
+       }
+
+       return true;
 }
 
-bool OSArray::isEqualTo(const OSMetaClassBase *anObject) const
+bool
+OSArray::isEqualTo(const OSMetaClassBase *anObject) const
 {
-    OSArray *otherArray;
+       OSArray *otherArray;
 
-    otherArray = OSDynamicCast(OSArray, anObject);
-    if ( otherArray )
-        return isEqualTo(otherArray);
-    else
-        return false;
+       otherArray = OSDynamicCast(OSArray, anObject);
+       if (otherArray) {
+               return isEqualTo(otherArray);
+       } else {
+               return false;
+       }
 }
 
-OSObject *OSArray::getObject(unsigned int index) const
+OSObject *
+OSArray::getObject(unsigned int index) const
 {
-    if (index >= count)
-        return 0;
-    else
-        return (OSObject *) (const_cast<OSMetaClassBase *>(array[index]));
+       if (index >= count) {
+               return 0;
+       } else {
+               return (OSObject *) (const_cast<OSMetaClassBase *>(array[index]));
+       }
 }
 
-OSObject *OSArray::getLastObject() const
+OSObject *
+OSArray::getLastObject() const
 {
-    if (count == 0)
-        return 0;
-    else
-        return ( OSObject *) (const_cast<OSMetaClassBase *>(array[count - 1]));
+       if (count == 0) {
+               return 0;
+       } else {
+               return (OSObject *) (const_cast<OSMetaClassBase *>(array[count - 1]));
+       }
 }
 
-unsigned int OSArray::getNextIndexOfObject(const OSMetaClassBase * anObject,
-                                            unsigned int index) const
+unsigned int
+OSArray::getNextIndexOfObject(const OSMetaClassBase * anObject,
+    unsigned int index) const
 {
-    while ((index < count) && (array[index] != anObject))
-        index++;
-    if (index >= count)
-        index = (unsigned int)-1;
-    return index;
+       while ((index < count) && (array[index] != anObject)) {
+               index++;
+       }
+       if (index >= count) {
+               index = (unsigned int)-1;
+       }
+       return index;
 }
 
-unsigned int OSArray::iteratorSize() const
+unsigned int
+OSArray::iteratorSize() const
 {
-    return sizeof(unsigned int);
+       return sizeof(unsigned int);
 }
 
-bool OSArray::initIterator(void *inIterator) const
+bool
+OSArray::initIterator(void *inIterator) const
 {
-    unsigned int *iteratorP = (unsigned int *) inIterator;
+       unsigned int *iteratorP = (unsigned int *) inIterator;
 
-    *iteratorP = 0;
-    return true;
+       *iteratorP = 0;
+       return true;
 }
 
-bool OSArray::getNextObjectForIterator(void *inIterator, OSObject **ret) const
+bool
+OSArray::getNextObjectForIterator(void *inIterator, OSObject **ret) const
 {
-    unsigned int *iteratorP = (unsigned int *) inIterator;
-    unsigned int index = (*iteratorP)++;
-
-    if (index < count) {
-        *ret = (OSObject *)(const_cast<OSMetaClassBase *> (array[index]));
-        return true;
-    }
-    else {
-        *ret = 0;
-        return false;
-    }
+       unsigned int *iteratorP = (unsigned int *) inIterator;
+       unsigned int index = (*iteratorP)++;
+
+       if (index < count) {
+               *ret = (OSObject *)(const_cast<OSMetaClassBase *> (array[index]));
+               return true;
+       } else {
+               *ret = 0;
+               return false;
+       }
 }
 
-bool OSArray::serialize(OSSerialize *s) const
+bool
+OSArray::serialize(OSSerialize *s) const
 {
-    if (s->previouslySerialized(this)) return true;
-    
-    if (!s->addXMLStartTag(this, "array")) return false;
+       if (s->previouslySerialized(this)) {
+               return true;
+       }
 
-    for (unsigned i = 0; i < count; i++) {
-        if (array[i] == NULL || !array[i]->serialize(s)) return false;
-    }
+       if (!s->addXMLStartTag(this, "array")) {
+               return false;
+       }
 
-    return s->addXMLEndTag("array");
+       for (unsigned i = 0; i < count; i++) {
+               if (array[i] == NULL || !array[i]->serialize(s)) {
+                       return false;
+               }
+       }
+
+       return s->addXMLEndTag("array");
 }
 
-unsigned OSArray::setOptions(unsigned options, unsigned mask, void *)
+unsigned
+OSArray::setOptions(unsigned options, unsigned mask, void *)
 {
-    unsigned old = super::setOptions(options, mask);
-    if ((old ^ options) & mask) {
-
-       // Value changed need to recurse over all of the child collections
-       for ( unsigned i = 0; i < count; i++ ) {
-           OSCollection *coll = OSDynamicCast(OSCollection, array[i]);
-           if (coll)
-               coll->setOptions(options, mask);
+       unsigned old = super::setOptions(options, mask);
+       if ((old ^ options) & mask) {
+               // Value changed need to recurse over all of the child collections
+               for (unsigned i = 0; i < count; i++) {
+                       OSCollection *coll = OSDynamicCast(OSCollection, array[i]);
+                       if (coll) {
+                               coll->setOptions(options, mask);
+                       }
+               }
        }
-    }
 
-    return old;
+       return old;
 }
 
-OSCollection * OSArray::copyCollection(OSDictionary *cycleDict)
+OSCollection *
+OSArray::copyCollection(OSDictionary *cycleDict)
 {
-    bool allocDict = !cycleDict;
-    OSCollection *ret = 0;
-    OSArray *newArray = 0;
-
-    if (allocDict) {
-       cycleDict = OSDictionary::withCapacity(16);
-       if (!cycleDict)
-           return 0;
-    }
-
-    do {
-       // Check for a cycle
-       ret = super::copyCollection(cycleDict);
-       if (ret)
-           continue;
-       
-       newArray = OSArray::withArray(this);
-       if (!newArray)
-           continue;
-
-       // Insert object into cycle Dictionary
-       cycleDict->setObject((const OSSymbol *) this, newArray);
-
-       for (unsigned int i = 0; i < count; i++) {
-           OSCollection *coll =
-               OSDynamicCast(OSCollection, EXT_CAST(newArray->array[i]));
-
-           if (coll) {
-               OSCollection *newColl = coll->copyCollection(cycleDict);
-               if (!newColl)
-                   goto abortCopy;
-
-               newArray->replaceObject(i, newColl);
-               newColl->release();
-           };
-       };
-
-       ret = newArray;
-       newArray = 0;
-
-    } while (false);
+       bool allocDict = !cycleDict;
+       OSCollection *ret = 0;
+       OSArray *newArray = 0;
+
+       if (allocDict) {
+               cycleDict = OSDictionary::withCapacity(16);
+               if (!cycleDict) {
+                       return 0;
+               }
+       }
+
+       do {
+               // Check for a cycle
+               ret = super::copyCollection(cycleDict);
+               if (ret) {
+                       continue;
+               }
+
+               newArray = OSArray::withArray(this);
+               if (!newArray) {
+                       continue;
+               }
+
+               // Insert object into cycle Dictionary
+               cycleDict->setObject((const OSSymbol *) this, newArray);
+
+               for (unsigned int i = 0; i < count; i++) {
+                       OSCollection *coll =
+                           OSDynamicCast(OSCollection, EXT_CAST(newArray->array[i]));
+
+                       if (coll) {
+                               OSCollection *newColl = coll->copyCollection(cycleDict);
+                               if (!newColl) {
+                                       goto abortCopy;
+                               }
+
+                               newArray->replaceObject(i, newColl);
+                               newColl->release();
+                       }
+                       ;
+               }
+               ;
+
+               ret = newArray;
+               newArray = 0;
+       } while (false);
 
 abortCopy:
-    if (newArray)
-       newArray->release();
+       if (newArray) {
+               newArray->release();
+       }
 
-    if (allocDict)
-       cycleDict->release();
+       if (allocDict) {
+               cycleDict->release();
+       }
 
-    return ret;
+       return ret;
 }
-