#ifndef qDEBUG
#define qDEBUG 0
#endif
#if qDEBUG
#warning qDEBUG is 1!
#endif
#include <sys/types.h>
#include <IOKit/IOService.h>
#include <IOKit/IOLib.h>
#define kIOHDIXControllerClassName "IOHDIXController"
#define kDIRootImageKey "di-root-image"
#define kDIRootImageResultKey "di-root-image-result"
#define kDIRootImageDevNameKey "di-root-image-devname"
#define kDIRootImageDevTKey "di-root-image-devt"
#define kDIRootRamFileKey "di-root-ram-file"
static IOService *
di_load_controller( void )
{
OSIterator * controllerIterator = 0;
OSDictionary * matchDictionary = 0;
IOService * controller = 0;
do {
IOService::getResourceService()->publishResource("com.apple.AppleDiskImageController.load", kOSBooleanTrue);
IOService::getResourceService()->waitQuiet();
matchDictionary = IOService::serviceMatching(kIOHDIXControllerClassName);
if (!matchDictionary)
break;
controllerIterator = IOService::getMatchingServices(matchDictionary);
if (!controllerIterator)
break;
controller = OSDynamicCast(IOService, controllerIterator->getNextObject());
if (!controller)
break;
controller->retain();
} while (false);
if (matchDictionary) matchDictionary->release();
if (controllerIterator) controllerIterator->release();
return controller;
}
extern "C" {
int di_root_image(const char *path, char devname[], dev_t *dev_p)
{
IOReturn res = 0;
IOService * controller = 0;
OSString * pathString = 0;
OSNumber * myResult = 0;
OSString * myDevName = 0;
OSNumber * myDevT = 0;
if (devname) *devname = 0;
if (dev_p) *dev_p = 0;
if (!path) return kIOReturnBadArgument;
if (!devname) return kIOReturnBadArgument;
if (!dev_p) return kIOReturnBadArgument;
controller = di_load_controller();
if (!controller) {
res = kIOReturnNotFound;
goto NoIOHDIXController;
}
pathString = OSString::withCString(path);
if (!pathString) {
res = kIOReturnNoMemory;
goto CannotCreatePathOSString;
}
if (!controller->setProperty(kDIRootImageKey, pathString))
IOLog("IOHDIXController::setProperty(%s, %s) failed.\n", kDIRootImageKey, pathString->getCStringNoCopy());
myResult = OSDynamicCast(OSNumber, controller->getProperty(kDIRootImageResultKey));
res = kIOReturnError;
if (myResult)
res = myResult->unsigned32BitValue();
if (res) {
IOLog("%s is 0x%08X/%d\n", kDIRootImageResultKey, res, res);
goto di_root_image_FAILED;
}
myDevT = OSDynamicCast(OSNumber, controller->getProperty(kDIRootImageDevTKey));
if (myDevT)
*dev_p = myDevT->unsigned32BitValue();
else {
IOLog("could not get %s\n", kDIRootImageDevTKey);
res = kIOReturnError;
goto di_root_image_FAILED;
}
myDevName = OSDynamicCast(OSString, controller->getProperty(kDIRootImageDevNameKey));
if (myDevName) {
strlcpy(devname, myDevName->getCStringNoCopy(), 16);
} else {
IOLog("could not get %s\n", kDIRootImageDevNameKey);
res = kIOReturnError;
goto di_root_image_FAILED;
}
di_root_image_FAILED:
CannotCreatePathOSString:
NoIOHDIXController:
if (pathString) pathString->release();
if (controller) controller->release();
return res;
}
void di_root_ramfile( IORegistryEntry * entry )
{
OSData * data;
IOMemoryDescriptor * mem;
uint64_t dmgSize;
uint64_t remain, length;
OSData * extentData = 0;
IOAddressRange * extentList;
uint64_t extentSize;
uint32_t extentCount;
do {
data = OSDynamicCast(OSData, entry->getProperty("boot-ramdmg-size"));
if (!data || (data->getLength() != sizeof(uint64_t)))
break;
dmgSize = *(uint64_t *) data->getBytesNoCopy();
if (!dmgSize)
break;
data = OSDynamicCast(OSData, entry->getProperty("boot-ramdmg-extents"));
if (!data || (data->getLength() == 0) ||
((data->getLength() & (sizeof(IOAddressRange)-1)) != 0))
break;
extentData = OSData::withData(data);
assert(extentData);
extentList = (IOAddressRange *) extentData->getBytesNoCopy();
extentCount = extentData->getLength() / sizeof(IOAddressRange);
extentSize = 0;
remain = dmgSize;
for (uint32_t i = 0; i < extentCount; i++)
{
length = extentList[i].length;
if (!length) break;
extentSize += length;
if (length >= remain)
{
extentList[i].length = remain;
extentCount = i + 1;
break;
}
remain -= length;
}
if (extentSize < dmgSize)
break;
mem = IOMemoryDescriptor::withAddressRanges(
extentList, extentCount,
kIODirectionOut | kIOMemoryMapperNone, NULL);
if (mem)
{
IOService * controller = di_load_controller();
if (controller)
{
controller->setProperty(kDIRootRamFileKey, mem);
controller->release();
}
mem->release();
}
} while (false);
if (extentData)
extentData->release();
}
};