#include <IOKit/IOLib.h>
#include <IOKit/IONVRAM.h>
#include <IOKit/IOPlatformExpert.h>
#include <IOKit/IOUserClient.h>
#include <IOKit/IOKitKeys.h>
#include <IOKit/IOKitKeysPrivate.h>
#include <kern/debug.h>
#include <pexpert/pexpert.h>
#if CONFIG_MACF
extern "C" {
#include <security/mac.h>
#include <security/mac_framework.h>
};
#endif
#define super IOService
#define kIONVRAMPrivilege kIOClientPrivilegeAdministrator
OSDefineMetaClassAndStructors(IODTNVRAM, IOService);
bool IODTNVRAM::init(IORegistryEntry *old, const IORegistryPlane *plane)
{
OSDictionary *dict;
if (!super::init(old, plane)) return false;
dict = OSDictionary::withCapacity(1);
if (dict == 0) return false;
setPropertyTable(dict);
_nvramImage = IONew(UInt8, kIODTNVRAMImageSize);
if (_nvramImage == 0) return false;
_nvramPartitionOffsets = OSDictionary::withCapacity(1);
if (_nvramPartitionOffsets == 0) return false;
_nvramPartitionLengths = OSDictionary::withCapacity(1);
if (_nvramPartitionLengths == 0) return false;
_registryPropertiesKey = OSSymbol::withCStringNoCopy("aapl,pci");
if (_registryPropertiesKey == 0) return false;
initProxyData();
return true;
}
void IODTNVRAM::initProxyData(void)
{
IORegistryEntry *entry;
const char *key = "nvram-proxy-data";
OSObject *prop;
OSData *data;
const void *bytes;
entry = IORegistryEntry::fromPath("/chosen", gIODTPlane);
if (entry != 0) {
prop = entry->getProperty(key);
if (prop != 0) {
data = OSDynamicCast(OSData, prop);
if (data != 0) {
bytes = data->getBytesNoCopy();
if ((bytes != 0) && (data->getLength() <= kIODTNVRAMImageSize)) {
bcopy(bytes, _nvramImage, data->getLength());
initNVRAMImage();
_isProxied = true;
}
}
}
entry->removeProperty(key);
entry->release();
}
}
void IODTNVRAM::registerNVRAMController(IONVRAMController *nvram)
{
if (_nvramController != 0) return;
_nvramController = nvram;
if (!_isProxied) {
_nvramController->read(0, _nvramImage, kIODTNVRAMImageSize);
initNVRAMImage();
} else {
syncOFVariables();
}
}
void IODTNVRAM::initNVRAMImage(void)
{
char partitionID[18];
UInt32 partitionOffset, partitionLength;
UInt32 freePartitionOffset, freePartitionSize;
UInt32 currentLength, currentOffset = 0;
OSNumber *partitionOffsetNumber, *partitionLengthNumber;
_ofPartitionOffset = 0xFFFFFFFF;
_piPartitionOffset = 0xFFFFFFFF;
freePartitionOffset = 0xFFFFFFFF;
freePartitionSize = 0;
while (currentOffset < kIODTNVRAMImageSize) {
currentLength = ((UInt16 *)(_nvramImage + currentOffset))[1] * 16;
if (currentLength < 16) break;
partitionOffset = currentOffset + 16;
partitionLength = currentLength - 16;
if ((partitionOffset + partitionLength) > kIODTNVRAMImageSize) break;
if (strncmp((const char *)_nvramImage + currentOffset + 4,
kIODTNVRAMOFPartitionName, 12) == 0) {
_ofPartitionOffset = partitionOffset;
_ofPartitionSize = partitionLength;
} else if (strncmp((const char *)_nvramImage + currentOffset + 4,
kIODTNVRAMXPRAMPartitionName, 12) == 0) {
} else if (strncmp((const char *)_nvramImage + currentOffset + 4,
kIODTNVRAMPanicInfoPartitonName, 12) == 0) {
_piPartitionOffset = partitionOffset;
_piPartitionSize = partitionLength;
} else if (strncmp((const char *)_nvramImage + currentOffset + 4,
kIODTNVRAMFreePartitionName, 12) == 0) {
freePartitionOffset = currentOffset;
freePartitionSize = currentLength;
} else {
snprintf(partitionID, sizeof(partitionID), "0x%02x,",
*(UInt8 *)(_nvramImage + currentOffset));
strncpy(partitionID + 5,
(const char *)(_nvramImage + currentOffset + 4), 12);
partitionID[17] = '\0';
partitionOffsetNumber = OSNumber::withNumber(partitionOffset, 32);
partitionLengthNumber = OSNumber::withNumber(partitionLength, 32);
_nvramPartitionOffsets->setObject(partitionID, partitionOffsetNumber);
_nvramPartitionLengths->setObject(partitionID, partitionLengthNumber);
partitionOffsetNumber->release();
partitionLengthNumber->release();
}
currentOffset += currentLength;
}
if (_ofPartitionOffset != 0xFFFFFFFF)
_ofImage = _nvramImage + _ofPartitionOffset;
if (_piPartitionOffset == 0xFFFFFFFF) {
if (freePartitionSize > 0x20) {
_nvramImage[freePartitionOffset] = 0xa1;
_nvramImage[freePartitionOffset + 1] = 0;
strncpy((char *)(_nvramImage + freePartitionOffset + 4),
kIODTNVRAMPanicInfoPartitonName, 12);
_piPartitionOffset = freePartitionOffset + 0x10;
_piPartitionSize = 0x800;
if (_piPartitionSize + 0x20 > freePartitionSize)
_piPartitionSize = freePartitionSize - 0x20;
_piImage = _nvramImage + _piPartitionOffset;
bzero(_piImage, _piPartitionSize);
*(UInt16 *)(_nvramImage + freePartitionOffset + 2) =
(_piPartitionSize / 0x10) + 1;
_nvramImage[freePartitionOffset + 1] =
calculatePartitionChecksum(_nvramImage + freePartitionOffset);
freePartitionOffset += _piPartitionSize + 0x10;
freePartitionSize -= _piPartitionSize + 0x10;
_nvramImage[freePartitionOffset] = 0x7f;
_nvramImage[freePartitionOffset + 1] = 0;
strncpy((char *)(_nvramImage + freePartitionOffset + 4),
kIODTNVRAMFreePartitionName, 12);
*(UInt16 *)(_nvramImage + freePartitionOffset + 2) =
freePartitionSize / 0x10;
_nvramImage[freePartitionOffset + 1] =
calculatePartitionChecksum(_nvramImage + freePartitionOffset);
if (_nvramController != 0) {
_nvramController->write(0, _nvramImage, kIODTNVRAMImageSize);
}
}
} else {
_piImage = _nvramImage + _piPartitionOffset;
}
_lastDeviceSync = 0;
_freshInterval = TRUE;
initOFVariables();
}
void IODTNVRAM::syncInternal(bool rateLimit)
{
if (_nvramController == 0) return;
if (rateLimit && !safeToSync()) return;
_nvramController->sync();
}
void IODTNVRAM::sync(void)
{
syncInternal(false);
}
bool IODTNVRAM::serializeProperties(OSSerialize *s) const
{
bool result, hasPrivilege;
UInt32 variablePerm;
const OSSymbol *key;
OSDictionary *dict;
OSCollectionIterator *iter = 0;
hasPrivilege = (kIOReturnSuccess == IOUserClient::clientHasPrivilege(current_task(), kIONVRAMPrivilege));
if (_ofDict == 0) {
dict = OSDictionary::withCapacity(1);
if (dict == 0) return false;
} else {
IOLockLock(_ofLock);
dict = OSDictionary::withDictionary(_ofDict);
IOLockUnlock(_ofLock);
if (dict == 0) return false;
iter = OSCollectionIterator::withCollection(dict);
if (iter == 0) {
dict->release();
return false;
}
while (1) {
key = OSDynamicCast(OSSymbol, iter->getNextObject());
if (key == 0) break;
variablePerm = getOFVariablePerm(key);
if ((hasPrivilege || (variablePerm != kOFVariablePermRootOnly)) &&
( ! (variablePerm == kOFVariablePermKernelOnly && current_task() != kernel_task) )
#if CONFIG_MACF
&& (current_task() == kernel_task || mac_iokit_check_nvram_get(kauth_cred_get(), key->getCStringNoCopy()) == 0)
#endif
) { }
else {
dict->removeObject(key);
iter->reset();
}
}
}
result = dict->serialize(s);
dict->release();
if (iter != 0) iter->release();
return result;
}
OSObject *IODTNVRAM::copyProperty(const OSSymbol *aKey) const
{
IOReturn result;
UInt32 variablePerm;
OSObject *theObject;
if (_ofDict == 0) return 0;
variablePerm = getOFVariablePerm(aKey);
result = IOUserClient::clientHasPrivilege(current_task(), kIONVRAMPrivilege);
if (result != kIOReturnSuccess) {
if (variablePerm == kOFVariablePermRootOnly) return 0;
}
if (variablePerm == kOFVariablePermKernelOnly && current_task() != kernel_task) return 0;
#if CONFIG_MACF
if (current_task() != kernel_task &&
mac_iokit_check_nvram_get(kauth_cred_get(), aKey->getCStringNoCopy()) != 0)
return 0;
#endif
IOLockLock(_ofLock);
theObject = _ofDict->getObject(aKey);
if (theObject) theObject->retain();
IOLockUnlock(_ofLock);
return theObject;
}
OSObject *IODTNVRAM::copyProperty(const char *aKey) const
{
const OSSymbol *keySymbol;
OSObject *theObject = 0;
keySymbol = OSSymbol::withCString(aKey);
if (keySymbol != 0) {
theObject = copyProperty(keySymbol);
keySymbol->release();
}
return theObject;
}
OSObject *IODTNVRAM::getProperty(const OSSymbol *aKey) const
{
OSObject *theObject;
theObject = copyProperty(aKey);
if (theObject) theObject->release();
return theObject;
}
OSObject *IODTNVRAM::getProperty(const char *aKey) const
{
OSObject *theObject;
theObject = copyProperty(aKey);
if (theObject) theObject->release();
return theObject;
}
bool IODTNVRAM::setProperty(const OSSymbol *aKey, OSObject *anObject)
{
bool result;
UInt32 propType, propPerm;
OSString *tmpString;
OSObject *propObject = 0;
if (_ofDict == 0) return false;
propPerm = getOFVariablePerm(aKey);
result = IOUserClient::clientHasPrivilege(current_task(), kIONVRAMPrivilege);
if (result != kIOReturnSuccess) {
if (propPerm != kOFVariablePermUserWrite) return false;
}
if (propPerm == kOFVariablePermKernelOnly && current_task() != kernel_task) return 0;
if (aKey->isEqualTo(kIODTNVRAMPanicInfoKey)) return false;
#if CONFIG_MACF
if (current_task() != kernel_task &&
mac_iokit_check_nvram_set(kauth_cred_get(), aKey->getCStringNoCopy(), anObject) != 0)
return false;
#endif
propType = getOFVariableType(aKey);
switch (propType) {
case kOFVariableTypeBoolean :
propObject = OSDynamicCast(OSBoolean, anObject);
break;
case kOFVariableTypeNumber :
propObject = OSDynamicCast(OSNumber, anObject);
break;
case kOFVariableTypeString :
propObject = OSDynamicCast(OSString, anObject);
break;
case kOFVariableTypeData :
propObject = OSDynamicCast(OSData, anObject);
if (propObject == 0) {
tmpString = OSDynamicCast(OSString, anObject);
if (tmpString != 0) {
propObject = OSData::withBytes(tmpString->getCStringNoCopy(),
tmpString->getLength());
}
}
break;
}
if (propObject == 0) return false;
IOLockLock(_ofLock);
result = _ofDict->setObject(aKey, propObject);
IOLockUnlock(_ofLock);
if (result) {
syncOFVariables();
}
return result;
}
void IODTNVRAM::removeProperty(const OSSymbol *aKey)
{
bool result;
UInt32 propPerm;
if (_ofDict == 0) return;
propPerm = getOFVariablePerm(aKey);
result = IOUserClient::clientHasPrivilege(current_task(), kIOClientPrivilegeAdministrator);
if (result != kIOReturnSuccess) {
if (propPerm != kOFVariablePermUserWrite) return;
}
if (propPerm == kOFVariablePermKernelOnly && current_task() != kernel_task) return;
if (aKey->isEqualTo(kIODTNVRAMPanicInfoKey)) return;
#if CONFIG_MACF
if (current_task() != kernel_task &&
mac_iokit_check_nvram_delete(kauth_cred_get(), aKey->getCStringNoCopy()) != 0)
return;
#endif
IOLockLock(_ofLock);
result = _ofDict->getObject(aKey) != 0;
if (result) {
_ofDict->removeObject(aKey);
}
IOLockUnlock(_ofLock);
if (result) {
syncOFVariables();
}
}
IOReturn IODTNVRAM::setProperties(OSObject *properties)
{
bool result = true;
OSObject *object;
const OSSymbol *key;
const OSString *tmpStr;
OSDictionary *dict;
OSCollectionIterator *iter;
dict = OSDynamicCast(OSDictionary, properties);
if (dict == 0) return kIOReturnBadArgument;
iter = OSCollectionIterator::withCollection(dict);
if (iter == 0) return kIOReturnBadArgument;
while (result) {
key = OSDynamicCast(OSSymbol, iter->getNextObject());
if (key == 0) break;
object = dict->getObject(key);
if (object == 0) continue;
if (key->isEqualTo(kIONVRAMDeletePropertyKey)) {
tmpStr = OSDynamicCast(OSString, object);
if (tmpStr != 0) {
key = OSSymbol::withString(tmpStr);
removeProperty(key);
key->release();
result = true;
} else {
result = false;
}
} else if(key->isEqualTo(kIONVRAMSyncNowPropertyKey) || key->isEqualTo(kIONVRAMForceSyncNowPropertyKey)) {
tmpStr = OSDynamicCast(OSString, object);
if (tmpStr != 0) {
result = true;
syncInternal(key->isEqualTo(kIONVRAMSyncNowPropertyKey));
} else {
result = false;
}
}
else {
result = setProperty(key, object);
}
}
iter->release();
if (result) return kIOReturnSuccess;
else return kIOReturnError;
}
IOReturn IODTNVRAM::readXPRAM(IOByteCount offset, UInt8 *buffer,
IOByteCount length)
{
return kIOReturnUnsupported;
}
IOReturn IODTNVRAM::writeXPRAM(IOByteCount offset, UInt8 *buffer,
IOByteCount length)
{
return kIOReturnUnsupported;
}
IOReturn IODTNVRAM::readNVRAMProperty(IORegistryEntry *entry,
const OSSymbol **name,
OSData **value)
{
IOReturn err;
err = readNVRAMPropertyType1(entry, name, value);
return err;
}
IOReturn IODTNVRAM::writeNVRAMProperty(IORegistryEntry *entry,
const OSSymbol *name,
OSData *value)
{
IOReturn err;
err = writeNVRAMPropertyType1(entry, name, value);
return err;
}
OSDictionary *IODTNVRAM::getNVRAMPartitions(void)
{
return _nvramPartitionLengths;
}
IOReturn IODTNVRAM::readNVRAMPartition(const OSSymbol *partitionID,
IOByteCount offset, UInt8 *buffer,
IOByteCount length)
{
OSNumber *partitionOffsetNumber, *partitionLengthNumber;
UInt32 partitionOffset, partitionLength, end;
partitionOffsetNumber =
(OSNumber *)_nvramPartitionOffsets->getObject(partitionID);
partitionLengthNumber =
(OSNumber *)_nvramPartitionLengths->getObject(partitionID);
if ((partitionOffsetNumber == 0) || (partitionLengthNumber == 0))
return kIOReturnNotFound;
partitionOffset = partitionOffsetNumber->unsigned32BitValue();
partitionLength = partitionLengthNumber->unsigned32BitValue();
if (os_add_overflow(offset, length, &end)) return kIOReturnBadArgument;
if ((buffer == 0) || (length == 0) || (end > partitionLength))
return kIOReturnBadArgument;
bcopy(_nvramImage + partitionOffset + offset, buffer, length);
return kIOReturnSuccess;
}
IOReturn IODTNVRAM::writeNVRAMPartition(const OSSymbol *partitionID,
IOByteCount offset, UInt8 *buffer,
IOByteCount length)
{
OSNumber *partitionOffsetNumber, *partitionLengthNumber;
UInt32 partitionOffset, partitionLength, end;
partitionOffsetNumber =
(OSNumber *)_nvramPartitionOffsets->getObject(partitionID);
partitionLengthNumber =
(OSNumber *)_nvramPartitionLengths->getObject(partitionID);
if ((partitionOffsetNumber == 0) || (partitionLengthNumber == 0))
return kIOReturnNotFound;
partitionOffset = partitionOffsetNumber->unsigned32BitValue();
partitionLength = partitionLengthNumber->unsigned32BitValue();
if (os_add_overflow(offset, length, &end)) return kIOReturnBadArgument;
if ((buffer == 0) || (length == 0) || (end > partitionLength))
return kIOReturnBadArgument;
bcopy(buffer, _nvramImage + partitionOffset + offset, length);
if (_nvramController != 0) {
_nvramController->write(0, _nvramImage, kIODTNVRAMImageSize);
}
return kIOReturnSuccess;
}
IOByteCount IODTNVRAM::savePanicInfo(UInt8 *buffer, IOByteCount length)
{
if ((_piImage == 0) || (length <= 0)) return 0;
if (length > (_piPartitionSize - 4))
length = _piPartitionSize - 4;
bcopy(buffer, _piImage + 4, length);
*(UInt32 *)_piImage = length;
if (_nvramController != 0) {
_nvramController->write(0, _nvramImage, kIODTNVRAMImageSize);
}
_systemPaniced = true;
sync();
return length;
}
UInt8 IODTNVRAM::calculatePartitionChecksum(UInt8 *partitionHeader)
{
UInt8 cnt, isum, csum = 0;
for (cnt = 0; cnt < 0x10; cnt++) {
isum = csum + partitionHeader[cnt];
if (isum < csum) isum++;
csum = isum;
}
return csum;
}
IOReturn IODTNVRAM::initOFVariables(void)
{
UInt32 cnt;
UInt8 *propName, *propData;
UInt32 propNameLength, propDataLength;
const OSSymbol *propSymbol;
OSObject *propObject;
if (_ofImage == 0) return kIOReturnNotReady;
_ofDict = OSDictionary::withCapacity(1);
_ofLock = IOLockAlloc();
if (!_ofDict || !_ofLock) return kIOReturnNoMemory;
cnt = 0;
while (cnt < _ofPartitionSize) {
if (_ofImage[cnt] == '\0') break;
propName = _ofImage + cnt;
for (propNameLength = 0; (cnt + propNameLength) < _ofPartitionSize;
propNameLength++) {
if (_ofImage[cnt + propNameLength] == '=') break;
}
if ((cnt + propNameLength) >= _ofPartitionSize) break;
cnt += propNameLength + 1;
propData = _ofImage + cnt;
for (propDataLength = 0; (cnt + propDataLength) < _ofPartitionSize;
propDataLength++) {
if (_ofImage[cnt + propDataLength] == '\0') break;
}
if ((cnt + propDataLength) >= _ofPartitionSize) break;
cnt += propDataLength + 1;
if (convertPropToObject(propName, propNameLength,
propData, propDataLength,
&propSymbol, &propObject)) {
_ofDict->setObject(propSymbol, propObject);
propSymbol->release();
propObject->release();
}
}
if (_ofDict->getObject("boot-args") == 0) {
propObject = OSString::withCStringNoCopy("");
if (propObject != 0) {
_ofDict->setObject("boot-args", propObject);
propObject->release();
}
}
if (_piImage != 0) {
propDataLength = *(UInt32 *)_piImage;
if ((propDataLength != 0) && (propDataLength <= (_piPartitionSize - 4))) {
propObject = OSData::withBytes(_piImage + 4, propDataLength);
_ofDict->setObject(kIODTNVRAMPanicInfoKey, propObject);
propObject->release();
*(UInt32 *)_piImage = 0;
if (_nvramController != 0) {
_nvramController->write(0, _nvramImage, kIODTNVRAMImageSize);
}
}
}
return kIOReturnSuccess;
}
IOReturn IODTNVRAM::syncOFVariables(void)
{
bool ok;
UInt32 length, maxLength;
UInt8 *buffer, *tmpBuffer;
const OSSymbol *tmpSymbol;
OSObject *tmpObject;
OSCollectionIterator *iter;
if ((_ofImage == 0) || (_ofDict == 0) || _systemPaniced) return kIOReturnNotReady;
buffer = tmpBuffer = IONew(UInt8, _ofPartitionSize);
if (buffer == 0) return kIOReturnNoMemory;
bzero(buffer, _ofPartitionSize);
ok = true;
maxLength = _ofPartitionSize;
IOLockLock(_ofLock);
iter = OSCollectionIterator::withCollection(_ofDict);
if (iter == 0) ok = false;
while (ok) {
tmpSymbol = OSDynamicCast(OSSymbol, iter->getNextObject());
if (tmpSymbol == 0) break;
if (tmpSymbol->isEqualTo(kIODTNVRAMPanicInfoKey)) continue;
tmpObject = _ofDict->getObject(tmpSymbol);
length = maxLength;
ok = convertObjectToProp(tmpBuffer, &length, tmpSymbol, tmpObject);
if (ok) {
tmpBuffer += length;
maxLength -= length;
}
}
iter->release();
IOLockUnlock(_ofLock);
if (ok) {
bcopy(buffer, _ofImage, _ofPartitionSize);
}
IODelete(buffer, UInt8, _ofPartitionSize);
if (!ok) return kIOReturnBadArgument;
if (_nvramController != 0) {
_nvramController->write(0, _nvramImage, kIODTNVRAMImageSize);
}
return kIOReturnSuccess;
}
struct OFVariable {
const char *variableName;
UInt32 variableType;
UInt32 variablePerm;
SInt32 variableOffset;
};
typedef struct OFVariable OFVariable;
enum {
kOWVariableOffsetNumber = 8,
kOWVariableOffsetString = 17
};
static const
OFVariable gOFVariables[] = {
{"little-endian?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 0},
{"real-mode?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 1},
{"auto-boot?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 2},
{"diag-switch?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 3},
{"fcode-debug?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 4},
{"oem-banner?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 5},
{"oem-logo?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 6},
{"use-nvramrc?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 7},
{"use-generic?", kOFVariableTypeBoolean, kOFVariablePermUserRead, -1},
{"default-mac-address?", kOFVariableTypeBoolean, kOFVariablePermUserRead,-1},
{"real-base", kOFVariableTypeNumber, kOFVariablePermUserRead, 8},
{"real-size", kOFVariableTypeNumber, kOFVariablePermUserRead, 9},
{"virt-base", kOFVariableTypeNumber, kOFVariablePermUserRead, 10},
{"virt-size", kOFVariableTypeNumber, kOFVariablePermUserRead, 11},
{"load-base", kOFVariableTypeNumber, kOFVariablePermUserRead, 12},
{"pci-probe-list", kOFVariableTypeNumber, kOFVariablePermUserRead, 13},
{"pci-probe-mask", kOFVariableTypeNumber, kOFVariablePermUserRead, -1},
{"screen-#columns", kOFVariableTypeNumber, kOFVariablePermUserRead, 14},
{"screen-#rows", kOFVariableTypeNumber, kOFVariablePermUserRead, 15},
{"selftest-#megs", kOFVariableTypeNumber, kOFVariablePermUserRead, 16},
{"boot-device", kOFVariableTypeString, kOFVariablePermUserRead, 17},
{"boot-file", kOFVariableTypeString, kOFVariablePermUserRead, 18},
{"boot-screen", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"console-screen", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"diag-device", kOFVariableTypeString, kOFVariablePermUserRead, 19},
{"diag-file", kOFVariableTypeString, kOFVariablePermUserRead, 20},
{"input-device", kOFVariableTypeString, kOFVariablePermUserRead, 21},
{"output-device", kOFVariableTypeString, kOFVariablePermUserRead, 22},
{"input-device-1", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"output-device-1", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"mouse-device", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"oem-banner", kOFVariableTypeString, kOFVariablePermUserRead, 23},
{"oem-logo", kOFVariableTypeString, kOFVariablePermUserRead, 24},
{"nvramrc", kOFVariableTypeString, kOFVariablePermUserRead, 25},
{"boot-command", kOFVariableTypeString, kOFVariablePermUserRead, 26},
{"default-client-ip", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"default-server-ip", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"default-gateway-ip", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"default-subnet-mask", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"default-router-ip", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"boot-script", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"boot-args", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"aapl,pci", kOFVariableTypeData, kOFVariablePermRootOnly, -1},
{"security-mode", kOFVariableTypeString, kOFVariablePermUserRead, -1},
{"security-password", kOFVariableTypeData, kOFVariablePermRootOnly, -1},
{"boot-image", kOFVariableTypeData, kOFVariablePermUserWrite, -1},
{"com.apple.System.fp-state", kOFVariableTypeData, kOFVariablePermKernelOnly, -1},
#if CONFIG_EMBEDDED
{"backlight-level", kOFVariableTypeData, kOFVariablePermUserWrite, -1},
{"com.apple.System.sep.art", kOFVariableTypeData, kOFVariablePermKernelOnly, -1},
{"com.apple.System.boot-nonce", kOFVariableTypeString, kOFVariablePermKernelOnly, -1},
{"darkboot", kOFVariableTypeBoolean, kOFVariablePermUserWrite, -1},
{"acc-mb-ld-lifetime", kOFVariableTypeNumber, kOFVariablePermKernelOnly, -1},
{"acc-cm-override-charger-count", kOFVariableTypeNumber, kOFVariablePermKernelOnly, -1},
{"acc-cm-override-count", kOFVariableTypeNumber, kOFVariablePermKernelOnly, -1},
{"enter-tdm-mode", kOFVariableTypeBoolean, kOFVariablePermUserWrite, -1},
#endif
{0, kOFVariableTypeData, kOFVariablePermUserRead, -1}
};
UInt32 IODTNVRAM::getOFVariableType(const OSSymbol *propSymbol) const
{
const OFVariable *ofVar;
ofVar = gOFVariables;
while (1) {
if ((ofVar->variableName == 0) ||
propSymbol->isEqualTo(ofVar->variableName)) break;
ofVar++;
}
return ofVar->variableType;
}
UInt32 IODTNVRAM::getOFVariablePerm(const OSSymbol *propSymbol) const
{
const OFVariable *ofVar;
ofVar = gOFVariables;
while (1) {
if ((ofVar->variableName == 0) ||
propSymbol->isEqualTo(ofVar->variableName)) break;
ofVar++;
}
return ofVar->variablePerm;
}
bool IODTNVRAM::getOWVariableInfo(UInt32 variableNumber, const OSSymbol **propSymbol,
UInt32 *propType, UInt32 *propOffset)
{
const OFVariable *ofVar;
ofVar = gOFVariables;
while (1) {
if (ofVar->variableName == 0) return false;
if (ofVar->variableOffset == (SInt32) variableNumber) break;
ofVar++;
}
*propSymbol = OSSymbol::withCStringNoCopy(ofVar->variableName);
*propType = ofVar->variableType;
switch (*propType) {
case kOFVariableTypeBoolean :
*propOffset = 1 << (31 - variableNumber);
break;
case kOFVariableTypeNumber :
*propOffset = variableNumber - kOWVariableOffsetNumber;
break;
case kOFVariableTypeString :
*propOffset = variableNumber - kOWVariableOffsetString;
break;
}
return true;
}
bool IODTNVRAM::convertPropToObject(UInt8 *propName, UInt32 propNameLength,
UInt8 *propData, UInt32 propDataLength,
const OSSymbol **propSymbol,
OSObject **propObject)
{
UInt32 propType;
const OSSymbol *tmpSymbol;
OSObject *tmpObject;
OSNumber *tmpNumber;
OSString *tmpString;
propName[propNameLength] = '\0';
tmpSymbol = OSSymbol::withCString((const char *)propName);
propName[propNameLength] = '=';
if (tmpSymbol == 0) {
return false;
}
propType = getOFVariableType(tmpSymbol);
tmpObject = 0;
switch (propType) {
case kOFVariableTypeBoolean :
if (!strncmp("true", (const char *)propData, propDataLength)) {
tmpObject = kOSBooleanTrue;
} else if (!strncmp("false", (const char *)propData, propDataLength)) {
tmpObject = kOSBooleanFalse;
}
break;
case kOFVariableTypeNumber :
tmpNumber = OSNumber::withNumber(strtol((const char *)propData, 0, 0), 32);
if (tmpNumber != 0) tmpObject = tmpNumber;
break;
case kOFVariableTypeString :
tmpString = OSString::withCString((const char *)propData);
if (tmpString != 0) tmpObject = tmpString;
break;
case kOFVariableTypeData :
tmpObject = unescapeBytesToData(propData, propDataLength);
break;
}
if (tmpObject == 0) {
tmpSymbol->release();
return false;
}
*propSymbol = tmpSymbol;
*propObject = tmpObject;
return true;
}
bool IODTNVRAM::convertObjectToProp(UInt8 *buffer, UInt32 *length,
const OSSymbol *propSymbol, OSObject *propObject)
{
const UInt8 *propName;
UInt32 propNameLength, propDataLength, remaining;
UInt32 propType, tmpValue;
OSBoolean *tmpBoolean = 0;
OSNumber *tmpNumber = 0;
OSString *tmpString = 0;
OSData *tmpData = 0;
propName = (const UInt8 *)propSymbol->getCStringNoCopy();
propNameLength = propSymbol->getLength();
propType = getOFVariableType(propSymbol);
propDataLength = 0xFFFFFFFF;
switch (propType) {
case kOFVariableTypeBoolean :
tmpBoolean = OSDynamicCast(OSBoolean, propObject);
if (tmpBoolean != 0) propDataLength = 5;
break;
case kOFVariableTypeNumber :
tmpNumber = OSDynamicCast(OSNumber, propObject);
if (tmpNumber != 0) propDataLength = 10;
break;
case kOFVariableTypeString :
tmpString = OSDynamicCast(OSString, propObject);
if (tmpString != 0) propDataLength = tmpString->getLength();
break;
case kOFVariableTypeData :
tmpData = OSDynamicCast(OSData, propObject);
if (tmpData != 0) {
tmpData = escapeDataToData(tmpData);
propDataLength = tmpData->getLength();
}
break;
}
if (propDataLength == 0xFFFFFFFF) return false;
if ((propNameLength + propDataLength + 2) > *length) return false;
buffer += snprintf((char *)buffer, *length, "%s=", propName);
remaining = *length - propNameLength - 1;
switch (propType) {
case kOFVariableTypeBoolean :
if (tmpBoolean->getValue()) {
strlcpy((char *)buffer, "true", remaining);
} else {
strlcpy((char *)buffer, "false", remaining);
}
break;
case kOFVariableTypeNumber :
tmpValue = tmpNumber->unsigned32BitValue();
if (tmpValue == 0xFFFFFFFF) {
strlcpy((char *)buffer, "-1", remaining);
} else if (tmpValue < 1000) {
snprintf((char *)buffer, remaining, "%d", (uint32_t)tmpValue);
} else {
snprintf((char *)buffer, remaining, "0x%x", (uint32_t)tmpValue);
}
break;
case kOFVariableTypeString :
strlcpy((char *)buffer, tmpString->getCStringNoCopy(), remaining);
break;
case kOFVariableTypeData :
bcopy(tmpData->getBytesNoCopy(), buffer, propDataLength);
tmpData->release();
break;
}
propDataLength = strlen((const char *)buffer);
*length = propNameLength + propDataLength + 2;
return true;
}
UInt16 IODTNVRAM::generateOWChecksum(UInt8 *buffer)
{
UInt32 cnt, checksum = 0;
UInt16 *tmpBuffer = (UInt16 *)buffer;
for (cnt = 0; cnt < _ofPartitionSize / 2; cnt++)
checksum += tmpBuffer[cnt];
return checksum % 0x0000FFFF;
}
bool IODTNVRAM::validateOWChecksum(UInt8 *buffer)
{
UInt32 cnt, checksum, sum = 0;
UInt16 *tmpBuffer = (UInt16 *)buffer;
for (cnt = 0; cnt < _ofPartitionSize / 2; cnt++)
sum += tmpBuffer[cnt];
checksum = (sum >> 16) + (sum & 0x0000FFFF);
if (checksum == 0x10000) checksum--;
checksum = (checksum ^ 0x0000FFFF) & 0x0000FFFF;
return checksum == 0;
}
void IODTNVRAM::updateOWBootArgs(const OSSymbol *key, OSObject *value)
{
bool wasBootArgs, bootr = false;
UInt32 cnt;
OSString *tmpString, *bootCommand, *bootArgs = 0;
const UInt8 *bootCommandData, *bootArgsData;
UInt8 *tmpData;
UInt32 bootCommandDataLength, bootArgsDataLength, tmpDataLength;
tmpString = OSDynamicCast(OSString, value);
if (tmpString == 0) return;
if (key->isEqualTo("boot-command")) {
wasBootArgs = false;
bootCommand = tmpString;
} else if (key->isEqualTo("boot-args")) {
wasBootArgs = true;
bootArgs = tmpString;
bootCommand = OSDynamicCast(OSString, _ofDict->getObject("boot-command"));
if (bootCommand == 0) return;
} else return;
bootCommandData = (const UInt8 *)bootCommand->getCStringNoCopy();
bootCommandDataLength = bootCommand->getLength();
if (bootCommandData == 0) return;
for (cnt = 0; cnt < bootCommandDataLength; cnt++) {
if ((bootCommandData[cnt] == 'b') &&
!strncmp("bootr", (const char *)bootCommandData + cnt, 5)) {
cnt += 5;
while (bootCommandData[cnt] == ' ') cnt++;
bootr = true;
break;
}
}
if (!bootr) {
_ofDict->removeObject("boot-args");
return;
}
if (wasBootArgs) {
bootArgsData = (const UInt8 *)bootArgs->getCStringNoCopy();
bootArgsDataLength = bootArgs->getLength();
if (bootArgsData == 0) return;
tmpDataLength = cnt + bootArgsDataLength;
tmpData = IONew(UInt8, tmpDataLength + 1);
if (tmpData == 0) return;
cnt -= strlcpy((char *)tmpData, (const char *)bootCommandData, cnt);
strlcat((char *)tmpData, (const char *)bootArgsData, cnt);
bootCommand = OSString::withCString((const char *)tmpData);
if (bootCommand != 0) {
_ofDict->setObject("boot-command", bootCommand);
bootCommand->release();
}
IODelete(tmpData, UInt8, tmpDataLength + 1);
} else {
bootArgs = OSString::withCString((const char *)(bootCommandData + cnt));
if (bootArgs != 0) {
_ofDict->setObject("boot-args", bootArgs);
bootArgs->release();
}
}
}
bool IODTNVRAM::searchNVRAMProperty(IONVRAMDescriptor *hdr, UInt32 *where)
{
return false;
}
IOReturn IODTNVRAM::readNVRAMPropertyType0(IORegistryEntry *entry,
const OSSymbol **name,
OSData **value)
{
return kIOReturnUnsupported;
}
IOReturn IODTNVRAM::writeNVRAMPropertyType0(IORegistryEntry *entry,
const OSSymbol *name,
OSData *value)
{
return kIOReturnUnsupported;
}
OSData *IODTNVRAM::unescapeBytesToData(const UInt8 *bytes, UInt32 length)
{
OSData *data = 0;
UInt32 totalLength = 0;
UInt32 cnt, cnt2;
UInt8 byte;
bool ok;
ok = true;
totalLength = 0;
for (cnt = 0; cnt < length;) {
byte = bytes[cnt++];
if (byte == 0xFF) {
byte = bytes[cnt++];
if (byte == 0x00) {
ok = false;
break;
}
cnt2 = byte & 0x7F;
} else
cnt2 = 1;
totalLength += cnt2;
}
if (ok) {
data = OSData::withCapacity(totalLength);
if (data != 0) {
for (cnt = 0; cnt < length;) {
byte = bytes[cnt++];
if (byte == 0xFF) {
byte = bytes[cnt++];
cnt2 = byte & 0x7F;
byte = (byte & 0x80) ? 0xFF : 0x00;
} else
cnt2 = 1;
data->appendByte(byte, cnt2);
}
}
}
return data;
}
OSData * IODTNVRAM::escapeDataToData(OSData * value)
{
OSData * result;
const UInt8 * startPtr;
const UInt8 * endPtr;
const UInt8 * wherePtr;
UInt8 byte;
bool ok = true;
wherePtr = (const UInt8 *) value->getBytesNoCopy();
endPtr = wherePtr + value->getLength();
result = OSData::withCapacity(endPtr - wherePtr);
if (!result)
return result;
while (wherePtr < endPtr) {
startPtr = wherePtr;
byte = *wherePtr++;
if ((byte == 0x00) || (byte == 0xFF)) {
for (;
((wherePtr - startPtr) < 0x80) && (wherePtr < endPtr) && (byte == *wherePtr);
wherePtr++) {}
ok &= result->appendByte(0xff, 1);
byte = (byte & 0x80) | (wherePtr - startPtr);
}
ok &= result->appendByte(byte, 1);
}
ok &= result->appendByte(0, 1);
if (!ok) {
result->release();
result = 0;
}
return result;
}
static bool IsApplePropertyName(const char * propName)
{
char c;
while ((c = *propName++)) {
if ((c >= 'A') && (c <= 'Z'))
break;
}
return (c == 0);
}
IOReturn IODTNVRAM::readNVRAMPropertyType1(IORegistryEntry *entry,
const OSSymbol **name,
OSData **value)
{
IOReturn err = kIOReturnNoResources;
OSData *data;
const UInt8 *startPtr;
const UInt8 *endPtr;
const UInt8 *wherePtr;
const UInt8 *nvPath = 0;
const char *nvName = 0;
const char *resultName = 0;
const UInt8 *resultValue = 0;
UInt32 resultValueLen = 0;
UInt8 byte;
if (_ofDict == 0) return err;
IOLockLock(_ofLock);
data = OSDynamicCast(OSData, _ofDict->getObject(_registryPropertiesKey));
IOLockUnlock(_ofLock);
if (data == 0) return err;
startPtr = (const UInt8 *) data->getBytesNoCopy();
endPtr = startPtr + data->getLength();
wherePtr = startPtr;
while (wherePtr < endPtr) {
byte = *(wherePtr++);
if (byte)
continue;
if (nvPath == 0)
nvPath = startPtr;
else if (nvName == 0)
nvName = (const char *) startPtr;
else {
IORegistryEntry * compareEntry = IORegistryEntry::fromPath((const char *) nvPath, gIODTPlane);
if (compareEntry)
compareEntry->release();
if (entry == compareEntry) {
bool appleProp = IsApplePropertyName(nvName);
if (!appleProp || !resultName) {
resultName = nvName;
resultValue = startPtr;
resultValueLen = wherePtr - startPtr - 1;
}
if (!appleProp)
break;
}
nvPath = 0;
nvName = 0;
}
startPtr = wherePtr;
}
if (resultName) {
*name = OSSymbol::withCString(resultName);
*value = unescapeBytesToData(resultValue, resultValueLen);
if ((*name != 0) && (*value != 0))
err = kIOReturnSuccess;
else
err = kIOReturnNoMemory;
}
return err;
}
IOReturn IODTNVRAM::writeNVRAMPropertyType1(IORegistryEntry *entry,
const OSSymbol *propName,
OSData *value)
{
OSData *oldData;
OSData *data = 0;
const UInt8 *startPtr;
const UInt8 *propStart;
const UInt8 *endPtr;
const UInt8 *wherePtr;
const UInt8 *nvPath = 0;
const char *nvName = 0;
const char * comp;
const char * name;
UInt8 byte;
bool ok = true;
bool settingAppleProp;
if (_ofDict == 0) return kIOReturnNoResources;
settingAppleProp = IsApplePropertyName(propName->getCStringNoCopy());
IOLockLock(_ofLock);
oldData = OSDynamicCast(OSData, _ofDict->getObject(_registryPropertiesKey));
if (oldData) {
startPtr = (const UInt8 *) oldData->getBytesNoCopy();
endPtr = startPtr + oldData->getLength();
propStart = startPtr;
wherePtr = startPtr;
while (wherePtr < endPtr) {
byte = *(wherePtr++);
if (byte)
continue;
if (nvPath == 0)
nvPath = startPtr;
else if (nvName == 0)
nvName = (const char *) startPtr;
else {
IORegistryEntry * compareEntry = IORegistryEntry::fromPath((const char *) nvPath, gIODTPlane);
if (compareEntry)
compareEntry->release();
if (entry == compareEntry) {
if ((settingAppleProp && propName->isEqualTo(nvName))
|| (!settingAppleProp && !IsApplePropertyName(nvName))) {
data = OSData::withBytes(propStart, nvPath - propStart);
if (data)
ok &= data->appendBytes(wherePtr, endPtr - wherePtr);
break;
}
}
nvPath = 0;
nvName = 0;
}
startPtr = wherePtr;
}
}
if (!data) {
if (oldData)
data = OSData::withData(oldData);
else
data = OSData::withCapacity(16);
if (!data) ok = false;
}
if (ok && value && value->getLength()) do {
OSArray *array = OSArray::withCapacity(5);
if (!array) {
ok = false;
break;
}
do
array->setObject(entry);
while ((entry = entry->getParentEntry(gIODTPlane)));
for (int i = array->getCount() - 3;
(entry = (IORegistryEntry *) array->getObject(i));
i--) {
name = entry->getName(gIODTPlane);
comp = entry->getLocation(gIODTPlane);
if (comp) ok &= data->appendBytes("/@", 2);
else {
if (!name) continue;
ok &= data->appendByte('/', 1);
comp = name;
}
ok &= data->appendBytes(comp, strlen(comp));
}
ok &= data->appendByte(0, 1);
array->release();
ok &= data->appendBytes(propName->getCStringNoCopy(), propName->getLength() + 1);
oldData = escapeDataToData(value);
ok &= (oldData != 0);
if (ok) ok &= data->appendBytes(oldData);
} while (false);
if (ok) {
ok = _ofDict->setObject(_registryPropertiesKey, data);
}
IOLockUnlock(_ofLock);
if (data) data->release();
if (ok) syncOFVariables();
return ok ? kIOReturnSuccess : kIOReturnNoMemory;
}
bool IODTNVRAM::safeToSync(void)
{
AbsoluteTime delta;
UInt64 delta_ns;
SInt32 delta_secs;
clock_get_uptime(&delta);
absolutetime_to_nanoseconds(delta, &delta_ns);
delta_secs = (SInt32)(delta_ns / NSEC_PER_SEC);
if ((delta_secs > (_lastDeviceSync + MIN_SYNC_NOW_INTERVAL)) || _freshInterval)
{
_lastDeviceSync = delta_secs;
_freshInterval = FALSE;
return TRUE;
}
return FALSE;
}