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 
37 OSDefineMetaClassAndStructors(OSCollectionIterator, OSIterator)
38 
39 bool OSCollectionIterator::initWithCollection(const OSCollection *inColl)
40 {
41     if ( !super::init() || !inColl)
42         return false;
43 
44     inColl->retain();
45     collection = inColl;
46     collIterator = 0;
47     initialUpdateStamp = 0;
48     valid = false;
49 
50     return true;
51 }
52 
53 OSCollectionIterator *
54 OSCollectionIterator::withCollection(const OSCollection *inColl)
55 {
56 
57     OSCollectionIterator *me = new OSCollectionIterator;
58 
59     if (me && !me->initWithCollection(inColl)) {
60         me->release();
61         return 0;
62     }
63 
64     return me;
65 }
66 
67 void OSCollectionIterator::free()
68 {
69     if (collIterator) {
70         kfree(collIterator, collection->iteratorSize());
71 	OSCONTAINER_ACCUMSIZE(-((size_t) collection->iteratorSize()));
72         collIterator = 0;
73     }
74 
75     if (collection) {
76         collection->release();
77         collection = 0;
78     }
79 
80     super::free();
81 }
82 
83 void OSCollectionIterator::reset()
84 {
85     valid = false;
86 
87     if (!collIterator) {
88         collIterator = (void *)kalloc_container(collection->iteratorSize());
89 	OSCONTAINER_ACCUMSIZE(collection->iteratorSize());
90         if (!collIterator)
91             return;
92     }
93 
94     if (!collection->initIterator(collIterator))
95         return;
96 
97     initialUpdateStamp = collection->updateStamp;
98     valid = true;
99 }
100 
101 bool OSCollectionIterator::isValid()
102 {
103     if (!collIterator) {
104         collIterator = (void *)kalloc_container(collection->iteratorSize());
105 	OSCONTAINER_ACCUMSIZE(collection->iteratorSize());
106         if (!collection->initIterator(collIterator))
107             return false;
108         initialUpdateStamp = collection->updateStamp;
109         valid = true;
110     }
111     else if (!valid || collection->updateStamp != initialUpdateStamp)
112         return false;
113 
114     return true;
115 }
116 
117 OSObject *OSCollectionIterator::getNextObject()
118 {
119     OSObject *retObj;
120     bool retVal;
121 
122     if (!isValid())
123         return 0;
124 
125     retVal = collection->getNextObjectForIterator(collIterator, &retObj);
126     return (retVal)? retObj : 0;
127 }
128 
129