gems-kernel/source/THIRDPARTY/xnu/libkern/c++/OSCollection.cpp
2024-06-03 11:29:39 -05:00

143 lines
4 KiB
C++

/*
* Copyright (c) 2000 Apple Computer, Inc. All rights reserved.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_START@
*
* This file contains Original Code and/or Modifications of Original Code
* as defined in and that are subject to the Apple Public Source License
* Version 2.0 (the 'License'). You may not use this file except in
* compliance with the License. The rights granted to you under the License
* may not be used to create, or enable the creation or redistribution of,
* unlawful or unlicensed copies of an Apple operating system, or to
* circumvent, violate, or enable the circumvention or violation of, any
* terms of an Apple operating system software license agreement.
*
* Please obtain a copy of the License at
* http://www.opensource.apple.com/apsl/ and read it before using this file.
*
* The Original Code and all software distributed under the License are
* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
* INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
* Please see the License for the specific language governing rights and
* limitations under the License.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_END@
*/
/* IOArray.h created by rsulack on Thu 11-Sep-1997 */
#define IOKIT_ENABLE_SHARED_PTR
#include <libkern/OSDebug.h>
#include <libkern/c++/OSCollection.h>
#include <libkern/c++/OSDictionary.h>
#include <IOKit/IOKitDebug.h>
#define super OSObject
OSDefineMetaClassAndAbstractStructors(OSCollection, OSObject)
OSMetaClassDefineReservedUsedX86(OSCollection, 0);
OSMetaClassDefineReservedUsedX86(OSCollection, 1);
OSMetaClassDefineReservedUnused(OSCollection, 2);
OSMetaClassDefineReservedUnused(OSCollection, 3);
OSMetaClassDefineReservedUnused(OSCollection, 4);
OSMetaClassDefineReservedUnused(OSCollection, 5);
OSMetaClassDefineReservedUnused(OSCollection, 6);
OSMetaClassDefineReservedUnused(OSCollection, 7);
bool
OSCollection::init()
{
if (!super::init()) {
return false;
}
updateStamp = 0;
return true;
}
void
OSCollection::haveUpdated()
{
if (fOptions & kImmutable) {
if (!(gIOKitDebug & kOSRegistryModsMode)) {
panic("Trying to change a collection in the registry");
} else {
OSReportWithBacktrace("Trying to change a collection in the registry");
}
}
updateStamp++;
}
unsigned
OSCollection::setOptions(unsigned options, unsigned mask, void *)
{
unsigned old = fOptions;
if (mask) {
fOptions = (old & ~mask) | (options & mask);
}
return old;
}
OSSharedPtr<OSCollection>
OSCollection::copyCollection(OSDictionary *cycleDict)
{
if (cycleDict) {
OSObject *obj = cycleDict->getObject((const OSSymbol *) this);
return OSSharedPtr<OSCollection>(reinterpret_cast<OSCollection *>(obj), OSRetain);
} else {
// If we are here it means that there is a collection subclass that
// hasn't overridden the copyCollection method. In which case just
// return a reference to ourselves.
// Hopefully this collection will not be inserted into the registry
return OSSharedPtr<OSCollection>(this, OSRetain);
}
}
bool
OSCollection::iterateObjects(void * refcon, bool (*callback)(void * refcon, OSObject * object))
{
uint64_t iteratorStore[2];
unsigned int initialUpdateStamp;
bool done;
assert(iteratorSize() < sizeof(iteratorStore));
if (!initIterator(&iteratorStore[0])) {
return false;
}
initialUpdateStamp = updateStamp;
done = false;
do{
OSObject * object;
if (!getNextObjectForIterator(&iteratorStore[0], &object)) {
break;
}
done = callback(refcon, object);
}while (!done && (initialUpdateStamp == updateStamp));
return initialUpdateStamp == updateStamp;
}
static bool
OSCollectionIterateObjectsBlock(void * refcon, OSObject * object)
{
bool (^block)(OSObject * object) = (typeof(block))refcon;
return block(object);
}
bool
OSCollection::iterateObjects(bool (^block)(OSObject * object))
{
return iterateObjects((void *) block, OSCollectionIterateObjectsBlock);
}