#include <libkern/OSDebug.h>
#include <libkern/c++/OSCollection.h>
#include <libkern/c++/OSDictionary.h>
#include <IOKit/IOKitDebug.h>
#define super OSObject
OSDefineMetaClassAndAbstractStructors(OSCollection, OSObject)
OSMetaClassDefineReservedUsed(OSCollection, 0);
OSMetaClassDefineReservedUsed(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;
}
OSCollection *
OSCollection::copyCollection(OSDictionary *cycleDict)
{
if (cycleDict) {
OSObject *obj = cycleDict->getObject((const OSSymbol *) this);
if (obj) {
obj->retain();
}
return reinterpret_cast<OSCollection *>(obj);
} else {
retain();
return this;
}
}
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);
}