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
39bool 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
53OSCollectionIterator *
54OSCollectionIterator::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
67void 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
83void 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
101bool 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
117OSObject *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