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 | |