#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));
}