1/* 2 * Copyright (c) 2000 Apple Computer, Inc. All rights reserved. 3 * 4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@ 5 * 6 * This file contains Original Code and/or Modifications of Original Code 7 * as defined in and that are subject to the Apple Public Source License 8 * Version 2.0 (the 'License'). You may not use this file except in 9 * compliance with the License. The rights granted to you under the License 10 * may not be used to create, or enable the creation or redistribution of, 11 * unlawful or unlicensed copies of an Apple operating system, or to 12 * circumvent, violate, or enable the circumvention or violation of, any 13 * terms of an Apple operating system software license agreement. 14 * 15 * Please obtain a copy of the License at 16 * http://www.opensource.apple.com/apsl/ and read it before using this file. 17 * 18 * The Original Code and all software distributed under the License are 19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 23 * Please see the License for the specific language governing rights and 24 * limitations under the License. 25 * 26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ 27 */ 28/* IOArray.h created by rsulack on Thu 11-Sep-1997 */ 29 30#include <libkern/c++/OSCollectionIterator.h> 31#include <libkern/c++/OSCollection.h> 32#include <libkern/c++/OSArray.h> 33#include <libkern/c++/OSLib.h> 34 35#define super OSIterator 36 37OSDefineMetaClassAndStructors(OSCollectionIterator, OSIterator) 38 39#if OSALLOCDEBUG 40extern "C" { 41 extern int debug_container_malloc_size; 42}; 43#define ACCUMSIZE(s) do { debug_container_malloc_size += (s); } while(0) 44#else 45#define ACCUMSIZE(s) 46#endif 47 48bool OSCollectionIterator::initWithCollection(const OSCollection *inColl) 49{ 50 if ( !super::init() || !inColl) 51 return false; 52 53 inColl->retain(); 54 collection = inColl; 55 collIterator = 0; 56 initialUpdateStamp = 0; 57 valid = false; 58 59 return this; 60} 61 62OSCollectionIterator * 63OSCollectionIterator::withCollection(const OSCollection *inColl) 64{ 65 66 OSCollectionIterator *me = new OSCollectionIterator; 67 68 if (me && !me->initWithCollection(inColl)) { 69 me->release(); 70 return 0; 71 } 72 73 return me; 74} 75 76void OSCollectionIterator::free() 77{ 78 if (collIterator) { 79 kfree(collIterator, collection->iteratorSize()); 80 ACCUMSIZE(-(collection->iteratorSize())); 81 collIterator = 0; 82 } 83 84 if (collection) { 85 collection->release(); 86 collection = 0; 87 } 88 89 super::free(); 90} 91 92void OSCollectionIterator::reset() 93{ 94 valid = false; 95 96 if (!collIterator) { 97 collIterator = (void *)kalloc(collection->iteratorSize()); 98 ACCUMSIZE(collection->iteratorSize()); 99 if (!collIterator) 100 return; 101 } 102 103 if (!collection->initIterator(collIterator)) 104 return; 105 106 initialUpdateStamp = collection->updateStamp; 107 valid = true; 108} 109 110bool OSCollectionIterator::isValid() 111{ 112 if (!collIterator) { 113 collIterator = (void *)kalloc(collection->iteratorSize()); 114 ACCUMSIZE(collection->iteratorSize()); 115 if (!collection->initIterator(collIterator)) 116 return false; 117 initialUpdateStamp = collection->updateStamp; 118 valid = true; 119 } 120 else if (!valid || collection->updateStamp != initialUpdateStamp) 121 return false; 122 123 return true; 124} 125 126OSObject *OSCollectionIterator::getNextObject() 127{ 128 OSObject *retObj; 129 bool retVal; 130 131 if (!isValid()) 132 return 0; 133 134 retVal = collection->getNextObjectForIterator(collIterator, &retObj); 135 return (retVal)? retObj : 0; 136} 137 138