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