]> git.saurik.com Git - apple/xnu.git/blame - iokit/Kernel/IONVRAM.cpp
xnu-1699.32.7.tar.gz
[apple/xnu.git] / iokit / Kernel / IONVRAM.cpp
CommitLineData
1c79356b 1/*
2d21ac55 2 * Copyright (c) 1998-2006 Apple Computer, Inc. All rights reserved.
1c79356b 3 *
2d21ac55 4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
1c79356b 5 *
2d21ac55
A
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. The rights granted to you under the License
10 * may not be used to create, or enable the creation or redistribution of,
11 * unlawful or unlicensed copies of an Apple operating system, or to
12 * circumvent, violate, or enable the circumvention or violation of, any
13 * terms of an Apple operating system software license agreement.
8f6c56a5 14 *
2d21ac55
A
15 * Please obtain a copy of the License at
16 * http://www.opensource.apple.com/apsl/ and read it before using this file.
17 *
18 * The Original Code and all software distributed under the License are
19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
8f6c56a5
A
20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
2d21ac55
A
22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23 * Please see the License for the specific language governing rights and
24 * limitations under the License.
8f6c56a5 25 *
2d21ac55 26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
1c79356b
A
27 */
28
29#include <IOKit/IOLib.h>
30#include <IOKit/IONVRAM.h>
31#include <IOKit/IOPlatformExpert.h>
32#include <IOKit/IOUserClient.h>
33#include <IOKit/IOKitKeys.h>
593a1d5f
A
34#include <kern/debug.h>
35#include <pexpert/pexpert.h>
1c79356b
A
36
37#define super IOService
38
91447636
A
39#define kIONVRAMPrivilege kIOClientPrivilegeAdministrator
40//#define kIONVRAMPrivilege kIOClientPrivilegeLocalUser
41
1c79356b
A
42OSDefineMetaClassAndStructors(IODTNVRAM, IOService);
43
44bool IODTNVRAM::init(IORegistryEntry *old, const IORegistryPlane *plane)
45{
46 OSDictionary *dict;
47
48 if (!super::init(old, plane)) return false;
49
50 dict = OSDictionary::withCapacity(1);
51 if (dict == 0) return false;
52 setPropertyTable(dict);
53
54 _nvramImage = IONew(UInt8, kIODTNVRAMImageSize);
55 if (_nvramImage == 0) return false;
d52fe63f
A
56
57 _nvramPartitionOffsets = OSDictionary::withCapacity(1);
58 if (_nvramPartitionOffsets == 0) return false;
59
60 _nvramPartitionLengths = OSDictionary::withCapacity(1);
61 if (_nvramPartitionLengths == 0) return false;
62
1c79356b
A
63 _registryPropertiesKey = OSSymbol::withCStringNoCopy("aapl,pci");
64 if (_registryPropertiesKey == 0) return false;
65
66 return true;
67}
68
69void IODTNVRAM::registerNVRAMController(IONVRAMController *nvram)
70{
d52fe63f
A
71 char partitionID[18];
72 UInt32 partitionOffset, partitionLength;
73 UInt32 freePartitionOffset, freePartitionSize;
74 UInt32 currentLength, currentOffset = 0;
75 OSNumber *partitionOffsetNumber, *partitionLengthNumber;
1c79356b
A
76
77 if (_nvramController != 0) return;
78
79 _nvramController = nvram;
80
81 _nvramController->read(0, _nvramImage, kIODTNVRAMImageSize);
82
9bccf70c 83 // Find the offsets for the OF, XPRAM, NameRegistry and PanicInfo partitions.
1c79356b
A
84 _ofPartitionOffset = 0xFFFFFFFF;
85 _xpramPartitionOffset = 0xFFFFFFFF;
86 _nrPartitionOffset = 0xFFFFFFFF;
9bccf70c 87 _piPartitionOffset = 0xFFFFFFFF;
d52fe63f
A
88 freePartitionOffset = 0xFFFFFFFF;
89 freePartitionSize = 0;
1c79356b
A
90 if (getPlatform()->getBootROMType()) {
91 // Look through the partitions to find the OF, MacOS partitions.
92 while (currentOffset < kIODTNVRAMImageSize) {
d52fe63f
A
93 currentLength = ((UInt16 *)(_nvramImage + currentOffset))[1] * 16;
94
95 partitionOffset = currentOffset + 16;
96 partitionLength = currentLength - 16;
97
98 if (strncmp((const char *)_nvramImage + currentOffset + 4,
99 kIODTNVRAMOFPartitionName, 12) == 0) {
100 _ofPartitionOffset = partitionOffset;
101 _ofPartitionSize = partitionLength;
102 } else if (strncmp((const char *)_nvramImage + currentOffset + 4,
103 kIODTNVRAMXPRAMPartitionName, 12) == 0) {
104 _xpramPartitionOffset = partitionOffset;
1c79356b
A
105 _xpramPartitionSize = kIODTNVRAMXPRAMSize;
106 _nrPartitionOffset = _xpramPartitionOffset + _xpramPartitionSize;
d52fe63f 107 _nrPartitionSize = partitionLength - _xpramPartitionSize;
9bccf70c
A
108 } else if (strncmp((const char *)_nvramImage + currentOffset + 4,
109 kIODTNVRAMPanicInfoPartitonName, 12) == 0) {
110 _piPartitionOffset = partitionOffset;
111 _piPartitionSize = partitionLength;
d52fe63f
A
112 } else if (strncmp((const char *)_nvramImage + currentOffset + 4,
113 kIODTNVRAMFreePartitionName, 12) == 0) {
114 freePartitionOffset = currentOffset;
115 freePartitionSize = currentLength;
116 } else {
117 // Construct the partition ID from the signature and name.
2d21ac55 118 snprintf(partitionID, sizeof(partitionID), "0x%02x,",
d52fe63f
A
119 *(UInt8 *)(_nvramImage + currentOffset));
120 strncpy(partitionID + 5,
121 (const char *)(_nvramImage + currentOffset + 4), 12);
122 partitionID[17] = '\0';
123
124 partitionOffsetNumber = OSNumber::withNumber(partitionOffset, 32);
125 partitionLengthNumber = OSNumber::withNumber(partitionLength, 32);
126
127 // Save the partition offset and length
128 _nvramPartitionOffsets->setObject(partitionID, partitionOffsetNumber);
129 _nvramPartitionLengths->setObject(partitionID, partitionLengthNumber);
130
131 partitionOffsetNumber->release();
132 partitionLengthNumber->release();
1c79356b 133 }
d52fe63f 134 currentOffset += currentLength;
1c79356b
A
135 }
136 } else {
137 // Use the fixed address for old world machines.
138 _ofPartitionOffset = 0x1800;
139 _ofPartitionSize = 0x0800;
140 _xpramPartitionOffset = 0x1300;
141 _xpramPartitionSize = 0x0100;
142 _nrPartitionOffset = 0x1400;
143 _nrPartitionSize = 0x0400;
144 }
145
146 if (_ofPartitionOffset != 0xFFFFFFFF)
147 _ofImage = _nvramImage + _ofPartitionOffset;
148 if (_xpramPartitionOffset != 0xFFFFFFFF)
149 _xpramImage = _nvramImage + _xpramPartitionOffset;
150 if (_nrPartitionOffset != 0xFFFFFFFF)
151 _nrImage = _nvramImage + _nrPartitionOffset;
152
9bccf70c
A
153 if (_piPartitionOffset == 0xFFFFFFFF) {
154 if (freePartitionSize > 0x20) {
155 // Set the signature to 0xa1.
156 _nvramImage[freePartitionOffset] = 0xa1;
157 // Set the checksum to 0.
158 _nvramImage[freePartitionOffset + 1] = 0;
159 // Set the name for the Panic Info partition.
160 strncpy((char *)(_nvramImage + freePartitionOffset + 4),
161 kIODTNVRAMPanicInfoPartitonName, 12);
162
163 // Calculate the partition offset and size.
164 _piPartitionOffset = freePartitionOffset + 0x10;
165 _piPartitionSize = 0x800;
166 if (_piPartitionSize + 0x20 > freePartitionSize)
167 _piPartitionSize = freePartitionSize - 0x20;
168
169 _piImage = _nvramImage + _piPartitionOffset;
170
171 // Zero the new partition.
172 bzero(_piImage, _piPartitionSize);
173
174 // Set the partition size.
175 *(UInt16 *)(_nvramImage + freePartitionOffset + 2) =
176 (_piPartitionSize / 0x10) + 1;
177
178 // Set the partition checksum.
179 _nvramImage[freePartitionOffset + 1] =
180 calculatePartitionChecksum(_nvramImage + freePartitionOffset);
181
182 // Calculate the free partition offset and size.
183 freePartitionOffset += _piPartitionSize + 0x10;
184 freePartitionSize -= _piPartitionSize + 0x10;
185
186 // Set the signature to 0x7f.
187 _nvramImage[freePartitionOffset] = 0x7f;
188 // Set the checksum to 0.
189 _nvramImage[freePartitionOffset + 1] = 0;
190 // Set the name for the free partition.
191 strncpy((char *)(_nvramImage + freePartitionOffset + 4),
192 kIODTNVRAMFreePartitionName, 12);
193 // Set the partition size.
194 *(UInt16 *)(_nvramImage + freePartitionOffset + 2) =
195 freePartitionSize / 0x10;
196 // Set the partition checksum.
197 _nvramImage[freePartitionOffset + 1] =
198 calculatePartitionChecksum(_nvramImage + freePartitionOffset);
199
200 // Set the nvram image as dirty.
201 _nvramImageDirty = true;
202 }
203 } else {
204 _piImage = _nvramImage + _piPartitionOffset;
205 }
206
6d2010ae
A
207 _lastDeviceSync = 0;
208 _freshInterval = TRUE; // we will allow sync() even before the first 15 minutes have passed.
209
1c79356b
A
210 initOFVariables();
211}
212
213void IODTNVRAM::sync(void)
214{
215 if (!_nvramImageDirty && !_ofImageDirty) return;
216
9bccf70c
A
217 // Don't try to sync OF Variables if the system has already paniced.
218 if (!_systemPaniced) syncOFVariables();
1c79356b
A
219
220 _nvramController->write(0, _nvramImage, kIODTNVRAMImageSize);
221 _nvramController->sync();
222
223 _nvramImageDirty = false;
224}
225
91447636 226bool IODTNVRAM::serializeProperties(OSSerialize *s) const
1c79356b 227{
cf7d32b8 228 bool result, hasPrivilege;
1c79356b
A
229 UInt32 variablePerm;
230 const OSSymbol *key;
cf7d32b8 231 OSDictionary *dict = 0, *tmpDict = 0;
1c79356b
A
232 OSCollectionIterator *iter = 0;
233
1c79356b 234 // Verify permissions.
cf7d32b8
A
235 hasPrivilege = (kIOReturnSuccess == IOUserClient::clientHasPrivilege(current_task(), kIONVRAMPrivilege));
236
237 tmpDict = OSDictionary::withCapacity(1);
238 if (tmpDict == 0) return false;
6d2010ae
A
239
240 if (_ofDict == 0) {
241 /* No nvram. Return an empty dictionary. */
242 dict = tmpDict;
243 } else {
244 /* Copy properties with client privilege. */
245 iter = OSCollectionIterator::withCollection(_ofDict);
246 if (iter == 0) return false;
1c79356b 247
6d2010ae
A
248 while (1) {
249 key = OSDynamicCast(OSSymbol, iter->getNextObject());
250 if (key == 0) break;
1c79356b 251
6d2010ae
A
252 variablePerm = getOFVariablePerm(key);
253 if ((hasPrivilege || (variablePerm != kOFVariablePermRootOnly)) &&
254 ( ! (variablePerm == kOFVariablePermKernelOnly && current_task() != kernel_task) )) {
255 tmpDict->setObject(key, _ofDict->getObject(key));
256 }
257 dict = tmpDict;
1c79356b 258 }
1c79356b 259 }
cf7d32b8 260
91447636 261 result = dict->serialize(s);
1c79356b
A
262
263 if (tmpDict != 0) tmpDict->release();
264 if (iter != 0) iter->release();
265
266 return result;
267}
268
269OSObject *IODTNVRAM::getProperty(const OSSymbol *aKey) const
270{
271 IOReturn result;
272 UInt32 variablePerm;
273
274 if (_ofDict == 0) return 0;
275
276 // Verify permissions.
cf7d32b8 277 variablePerm = getOFVariablePerm(aKey);
91447636 278 result = IOUserClient::clientHasPrivilege(current_task(), kIONVRAMPrivilege);
1c79356b 279 if (result != kIOReturnSuccess) {
1c79356b
A
280 if (variablePerm == kOFVariablePermRootOnly) return 0;
281 }
cf7d32b8 282 if (variablePerm == kOFVariablePermKernelOnly && current_task() != kernel_task) return 0;
1c79356b
A
283
284 return _ofDict->getObject(aKey);
285}
286
287OSObject *IODTNVRAM::getProperty(const char *aKey) const
288{
289 const OSSymbol *keySymbol;
290 OSObject *theObject = 0;
291
292 keySymbol = OSSymbol::withCStringNoCopy(aKey);
293 if (keySymbol != 0) {
294 theObject = getProperty(keySymbol);
295 keySymbol->release();
296 }
297
298 return theObject;
299}
300
301bool IODTNVRAM::setProperty(const OSSymbol *aKey, OSObject *anObject)
302{
303 bool result;
304 UInt32 propType, propPerm;
305 OSString *tmpString;
306 OSObject *propObject = 0;
307
308 if (_ofDict == 0) return false;
309
310 // Verify permissions.
cf7d32b8 311 propPerm = getOFVariablePerm(aKey);
91447636 312 result = IOUserClient::clientHasPrivilege(current_task(), kIONVRAMPrivilege);
1c79356b 313 if (result != kIOReturnSuccess) {
1c79356b
A
314 if (propPerm != kOFVariablePermUserWrite) return false;
315 }
cf7d32b8
A
316 if (propPerm == kOFVariablePermKernelOnly && current_task() != kernel_task) return 0;
317
1c79356b
A
318 // Don't allow creation of new properties on old world machines.
319 if (getPlatform()->getBootROMType() == 0) {
320 if (_ofDict->getObject(aKey) == 0) return false;
321 }
322
9bccf70c
A
323 // Don't allow change of 'aapl,panic-info'.
324 if (aKey->isEqualTo(kIODTNVRAMPanicInfoKey)) return false;
325
1c79356b
A
326 // Make sure the object is of the correct type.
327 propType = getOFVariableType(aKey);
328 switch (propType) {
329 case kOFVariableTypeBoolean :
330 propObject = OSDynamicCast(OSBoolean, anObject);
331 break;
332
333 case kOFVariableTypeNumber :
334 propObject = OSDynamicCast(OSNumber, anObject);
335 break;
336
337 case kOFVariableTypeString :
338 propObject = OSDynamicCast(OSString, anObject);
339 break;
340
341 case kOFVariableTypeData :
342 propObject = OSDynamicCast(OSData, anObject);
343 if (propObject == 0) {
344 tmpString = OSDynamicCast(OSString, anObject);
345 if (tmpString != 0) {
346 propObject = OSData::withBytes(tmpString->getCStringNoCopy(),
347 tmpString->getLength());
348 }
349 }
350 break;
351 }
352
353 if (propObject == 0) return false;
354
355 result = _ofDict->setObject(aKey, propObject);
356
357 if (result) {
358 if (getPlatform()->getBootROMType() == 0) {
359 updateOWBootArgs(aKey, propObject);
360 }
361
362 _ofImageDirty = true;
363 }
364
365 return result;
366}
367
91447636
A
368void IODTNVRAM::removeProperty(const OSSymbol *aKey)
369{
370 bool result;
371 UInt32 propPerm;
372
373 if (_ofDict == 0) return;
374
375 // Verify permissions.
cf7d32b8 376 propPerm = getOFVariablePerm(aKey);
91447636
A
377 result = IOUserClient::clientHasPrivilege(current_task(), kIOClientPrivilegeAdministrator);
378 if (result != kIOReturnSuccess) {
91447636
A
379 if (propPerm != kOFVariablePermUserWrite) return;
380 }
cf7d32b8 381 if (propPerm == kOFVariablePermKernelOnly && current_task() != kernel_task) return;
91447636
A
382
383 // Don't allow removal of properties on old world machines.
384 if (getPlatform()->getBootROMType() == 0) return;
385
386 // Don't allow change of 'aapl,panic-info'.
387 if (aKey->isEqualTo(kIODTNVRAMPanicInfoKey)) return;
388
389 // If the object exists, remove it from the dictionary.
390 result = _ofDict->getObject(aKey) != 0;
391 if (result) {
392 _ofDict->removeObject(aKey);
393
394 _ofImageDirty = true;
395 }
396}
397
1c79356b
A
398IOReturn IODTNVRAM::setProperties(OSObject *properties)
399{
400 bool result = true;
401 OSObject *object;
402 const OSSymbol *key;
91447636 403 const OSString *tmpStr;
1c79356b
A
404 OSDictionary *dict;
405 OSCollectionIterator *iter;
406
407 dict = OSDynamicCast(OSDictionary, properties);
408 if (dict == 0) return kIOReturnBadArgument;
409
410 iter = OSCollectionIterator::withCollection(dict);
411 if (iter == 0) return kIOReturnBadArgument;
412
413 while (result) {
414 key = OSDynamicCast(OSSymbol, iter->getNextObject());
415 if (key == 0) break;
416
417 object = dict->getObject(key);
418 if (object == 0) continue;
419
91447636 420 if (key->isEqualTo(kIONVRAMDeletePropertyKey)) {
6d2010ae
A
421 tmpStr = OSDynamicCast(OSString, object);
422 if (tmpStr != 0) {
423 key = OSSymbol::withString(tmpStr);
424 removeProperty(key);
425 key->release();
426 result = true;
427 } else {
428 result = false;
429 }
430 } else if(key->isEqualTo(kIONVRAMSyncNowPropertyKey)) {
431 tmpStr = OSDynamicCast(OSString, object);
432 if (tmpStr != 0) {
433
434 result = true; // We are not going to gaurantee sync, this is best effort
435
436 if(safeToSync())
437 sync();
438
439 } else {
440 result = false;
441 }
442 }
443 else {
444 result = setProperty(key, object);
91447636 445 }
6d2010ae 446
1c79356b
A
447 }
448
449 iter->release();
450
451 if (result) return kIOReturnSuccess;
452 else return kIOReturnError;
453}
454
455IOReturn IODTNVRAM::readXPRAM(IOByteCount offset, UInt8 *buffer,
456 IOByteCount length)
457{
fa4905b1 458 if (_xpramImage == 0) return kIOReturnUnsupported;
1c79356b 459
91447636 460 if ((buffer == 0) || (length == 0) ||
1c79356b
A
461 (offset + length > kIODTNVRAMXPRAMSize))
462 return kIOReturnBadArgument;
463
464 bcopy(_nvramImage + _xpramPartitionOffset + offset, buffer, length);
465
466 return kIOReturnSuccess;
467}
468
469IOReturn IODTNVRAM::writeXPRAM(IOByteCount offset, UInt8 *buffer,
470 IOByteCount length)
471{
fa4905b1 472 if (_xpramImage == 0) return kIOReturnUnsupported;
1c79356b 473
91447636 474 if ((buffer == 0) || (length == 0) ||
1c79356b
A
475 (offset + length > kIODTNVRAMXPRAMSize))
476 return kIOReturnBadArgument;
477
478 bcopy(buffer, _nvramImage + _xpramPartitionOffset + offset, length);
479
480 _nvramImageDirty = true;
481
482 return kIOReturnSuccess;
483}
484
485IOReturn IODTNVRAM::readNVRAMProperty(IORegistryEntry *entry,
486 const OSSymbol **name,
487 OSData **value)
488{
489 IOReturn err;
490
491 if (getPlatform()->getBootROMType())
492 err = readNVRAMPropertyType1(entry, name, value);
493 else
494 err = readNVRAMPropertyType0(entry, name, value);
495
496 return err;
497}
498
499IOReturn IODTNVRAM::writeNVRAMProperty(IORegistryEntry *entry,
500 const OSSymbol *name,
501 OSData *value)
502{
503 IOReturn err;
504
505 if (getPlatform()->getBootROMType())
506 err = writeNVRAMPropertyType1(entry, name, value);
507 else
508 err = writeNVRAMPropertyType0(entry, name, value);
509
510 return err;
511}
512
d52fe63f
A
513OSDictionary *IODTNVRAM::getNVRAMPartitions(void)
514{
515 return _nvramPartitionLengths;
516}
1c79356b 517
d52fe63f
A
518IOReturn IODTNVRAM::readNVRAMPartition(const OSSymbol *partitionID,
519 IOByteCount offset, UInt8 *buffer,
520 IOByteCount length)
521{
522 OSNumber *partitionOffsetNumber, *partitionLengthNumber;
523 UInt32 partitionOffset, partitionLength;
524
525 partitionOffsetNumber =
526 (OSNumber *)_nvramPartitionOffsets->getObject(partitionID);
527 partitionLengthNumber =
528 (OSNumber *)_nvramPartitionLengths->getObject(partitionID);
529
530 if ((partitionOffsetNumber == 0) || (partitionLengthNumber == 0))
531 return kIOReturnNotFound;
532
533 partitionOffset = partitionOffsetNumber->unsigned32BitValue();
534 partitionLength = partitionLengthNumber->unsigned32BitValue();
535
91447636 536 if ((buffer == 0) || (length == 0) ||
d52fe63f
A
537 (offset + length > partitionLength))
538 return kIOReturnBadArgument;
539
540 bcopy(_nvramImage + partitionOffset + offset, buffer, length);
541
542 return kIOReturnSuccess;
543}
544
545IOReturn IODTNVRAM::writeNVRAMPartition(const OSSymbol *partitionID,
546 IOByteCount offset, UInt8 *buffer,
547 IOByteCount length)
548{
549 OSNumber *partitionOffsetNumber, *partitionLengthNumber;
550 UInt32 partitionOffset, partitionLength;
551
552 partitionOffsetNumber =
553 (OSNumber *)_nvramPartitionOffsets->getObject(partitionID);
554 partitionLengthNumber =
555 (OSNumber *)_nvramPartitionLengths->getObject(partitionID);
556
557 if ((partitionOffsetNumber == 0) || (partitionLengthNumber == 0))
558 return kIOReturnNotFound;
559
560 partitionOffset = partitionOffsetNumber->unsigned32BitValue();
561 partitionLength = partitionLengthNumber->unsigned32BitValue();
562
91447636 563 if ((buffer == 0) || (length == 0) ||
d52fe63f
A
564 (offset + length > partitionLength))
565 return kIOReturnBadArgument;
566
567 bcopy(buffer, _nvramImage + partitionOffset + offset, length);
568
569 _nvramImageDirty = true;
570
571 return kIOReturnSuccess;
572}
1c79356b 573
b0d623f7 574IOByteCount IODTNVRAM::savePanicInfo(UInt8 *buffer, IOByteCount length)
9bccf70c
A
575{
576 if ((_piImage == 0) || (length <= 0)) return 0;
577
578 if (length > (_piPartitionSize - 4))
579 length = _piPartitionSize - 4;
580
581 // Save the Panic Info.
582 bcopy(buffer, _piImage + 4, length);
583
584 // Save the Panic Info length.
585 *(UInt32 *)_piImage = length;
586
587 _nvramImageDirty = true;
2d21ac55
A
588 /*
589 * This prevents OF variables from being committed if the system has panicked
590 */
9bccf70c 591 _systemPaniced = true;
2d21ac55
A
592 /* The call to sync() forces the NVRAM controller to write the panic info
593 * partition to NVRAM.
594 */
595 sync();
596
9bccf70c
A
597 return length;
598}
599
600// Private methods
601
602UInt8 IODTNVRAM::calculatePartitionChecksum(UInt8 *partitionHeader)
603{
604 UInt8 cnt, isum, csum = 0;
605
606 for (cnt = 0; cnt < 0x10; cnt++) {
607 isum = csum + partitionHeader[cnt];
608 if (isum < csum) isum++;
609 csum = isum;
610 }
611
612 return csum;
613}
1c79356b
A
614
615struct OWVariablesHeader {
616 UInt16 owMagic;
617 UInt8 owVersion;
618 UInt8 owPages;
619 UInt16 owChecksum;
620 UInt16 owHere;
621 UInt16 owTop;
622 UInt16 owNext;
623 UInt32 owFlags;
624 UInt32 owNumbers[9];
625 struct {
626 UInt16 offset;
627 UInt16 length;
628 } owStrings[10];
629};
630typedef struct OWVariablesHeader OWVariablesHeader;
631
632IOReturn IODTNVRAM::initOFVariables(void)
633{
634 UInt32 cnt, propOffset, propType;
635 UInt8 *propName, *propData;
636 UInt32 propNameLength, propDataLength;
637 const OSSymbol *propSymbol;
638 OSObject *propObject;
639 OWVariablesHeader *owHeader;
640
641 if (_ofImage == 0) return kIOReturnNotReady;
642
643 _ofDict = OSDictionary::withCapacity(1);
644 if (_ofDict == 0) return kIOReturnNoMemory;
645
646 if (getPlatform()->getBootROMType()) {
647 cnt = 0;
648 while (cnt < _ofPartitionSize) {
649 // Break if there is no name.
650 if (_ofImage[cnt] == '\0') break;
651
652 // Find the length of the name.
653 propName = _ofImage + cnt;
654 for (propNameLength = 0; (cnt + propNameLength) < _ofPartitionSize;
655 propNameLength++) {
656 if (_ofImage[cnt + propNameLength] == '=') break;
657 }
658
659 // Break if the name goes past the end of the partition.
660 if ((cnt + propNameLength) >= _ofPartitionSize) break;
661 cnt += propNameLength + 1;
662
663 propData = _ofImage + cnt;
664 for (propDataLength = 0; (cnt + propDataLength) < _ofPartitionSize;
665 propDataLength++) {
666 if (_ofImage[cnt + propDataLength] == '\0') break;
667 }
668
669 // Break if the data goes past the end of the partition.
670 if ((cnt + propDataLength) >= _ofPartitionSize) break;
671 cnt += propDataLength + 1;
672
673 if (convertPropToObject(propName, propNameLength,
674 propData, propDataLength,
675 &propSymbol, &propObject)) {
676 _ofDict->setObject(propSymbol, propObject);
677 propSymbol->release();
678 propObject->release();
679 }
680 }
681
682 // Create the boot-args property if it is not in the dictionary.
683 if (_ofDict->getObject("boot-args") == 0) {
684 propObject = OSString::withCStringNoCopy("");
685 if (propObject != 0) {
686 _ofDict->setObject("boot-args", propObject);
687 propObject->release();
688 }
689 }
9bccf70c
A
690
691 // Create the 'aapl,panic-info' property if needed.
692 if (_piImage != 0) {
693 propDataLength = *(UInt32 *)_piImage;
2d21ac55 694 if ((propDataLength != 0) && (propDataLength <= (_piPartitionSize - 4))) {
9bccf70c
A
695 propObject = OSData::withBytes(_piImage + 4, propDataLength);
696 _ofDict->setObject(kIODTNVRAMPanicInfoKey, propObject);
697 propObject->release();
698
699 // Clear the length from _piImage and mark dirty.
700 *(UInt32 *)_piImage = 0;
701 _nvramImageDirty = true;
702 }
703 }
1c79356b
A
704 } else {
705 owHeader = (OWVariablesHeader *)_ofImage;
706 if (!validateOWChecksum(_ofImage)) {
707 _ofDict->release();
708 _ofDict = 0;
709 return kIOReturnBadMedia;
710 }
711
712 cnt = 0;
713 while (1) {
714 if (!getOWVariableInfo(cnt++, &propSymbol, &propType, &propOffset))
715 break;
716
717 switch (propType) {
718 case kOFVariableTypeBoolean :
719 propObject = OSBoolean::withBoolean(owHeader->owFlags & propOffset);
720 break;
721
722 case kOFVariableTypeNumber :
723 propObject = OSNumber::withNumber(owHeader->owNumbers[propOffset], 32);
724 break;
725
726 case kOFVariableTypeString :
727 propData = _ofImage + owHeader->owStrings[propOffset].offset -
728 _ofPartitionOffset;
729 propDataLength = owHeader->owStrings[propOffset].length;
730 propName = IONew(UInt8, propDataLength + 1);
731 if (propName != 0) {
732 strncpy((char *)propName, (const char *)propData, propDataLength);
733 propName[propDataLength] = '\0';
734 propObject = OSString::withCString((const char *)propName);
735 IODelete(propName, UInt8, propDataLength + 1);
736 }
737 break;
738 }
739
740 if (propObject == 0) break;
741
742 _ofDict->setObject(propSymbol, propObject);
743 propSymbol->release();
744 propObject->release();
745 }
746
747 // Create the boot-args property.
748 propSymbol = OSSymbol::withCString("boot-command");
749 if (propSymbol != 0) {
750 propObject = _ofDict->getObject(propSymbol);
751 if (propObject != 0) {
752 updateOWBootArgs(propSymbol, propObject);
753 }
754 propSymbol->release();
755 }
756 }
757
758 return kIOReturnSuccess;
759}
760
761IOReturn IODTNVRAM::syncOFVariables(void)
762{
763 bool ok;
764 UInt32 cnt, length, maxLength;
765 UInt32 curOffset, tmpOffset, tmpType, tmpDataLength;
91447636
A
766 UInt8 *buffer, *tmpBuffer;
767 const UInt8 *tmpData;
1c79356b
A
768 const OSSymbol *tmpSymbol;
769 OSObject *tmpObject;
770 OSBoolean *tmpBoolean;
771 OSNumber *tmpNumber;
772 OSString *tmpString;
773 OSCollectionIterator *iter;
774 OWVariablesHeader *owHeader, *owHeaderOld;
775
776 if ((_ofImage == 0) || (_ofDict == 0)) return kIOReturnNotReady;
777
778 if (!_ofImageDirty) return kIOReturnSuccess;
779
780 if (getPlatform()->getBootROMType()) {
781 buffer = tmpBuffer = IONew(UInt8, _ofPartitionSize);
782 if (buffer == 0) return kIOReturnNoMemory;
783 bzero(buffer, _ofPartitionSize);
784
785 ok = true;
786 maxLength = _ofPartitionSize;
787
788 iter = OSCollectionIterator::withCollection(_ofDict);
789 if (iter == 0) ok = false;
790
791 while (ok) {
792 tmpSymbol = OSDynamicCast(OSSymbol, iter->getNextObject());
793 if (tmpSymbol == 0) break;
794
9bccf70c
A
795 // Don't save 'aapl,panic-info'.
796 if (tmpSymbol->isEqualTo(kIODTNVRAMPanicInfoKey)) continue;
797
1c79356b
A
798 tmpObject = _ofDict->getObject(tmpSymbol);
799
800 length = maxLength;
801 ok = convertObjectToProp(tmpBuffer, &length, tmpSymbol, tmpObject);
802 if (ok) {
803 tmpBuffer += length;
804 maxLength -= length;
805 }
806 }
807 iter->release();
808
809 if (ok) {
810 bcopy(buffer, _ofImage, _ofPartitionSize);
811 }
812
813 IODelete(buffer, UInt8, _ofPartitionSize);
814
815 if (!ok) return kIOReturnBadArgument;
816 } else {
817 buffer = IONew(UInt8, _ofPartitionSize);
818 if (buffer == 0) return kIOReturnNoMemory;
819 bzero(buffer, _ofPartitionSize);
820
821 owHeader = (OWVariablesHeader *)buffer;
822 owHeaderOld = (OWVariablesHeader *)_ofImage;
823
824 owHeader->owMagic = owHeaderOld->owMagic;
825 owHeader->owVersion = owHeaderOld->owVersion;
826 owHeader->owPages = owHeaderOld->owPages;
827
828 curOffset = _ofPartitionSize;
829
830 ok = true;
831 cnt = 0;
832 while (ok) {
833 if (!getOWVariableInfo(cnt++, &tmpSymbol, &tmpType, &tmpOffset))
834 break;
835
836 tmpObject = _ofDict->getObject(tmpSymbol);
837
838 switch (tmpType) {
839 case kOFVariableTypeBoolean :
840 tmpBoolean = OSDynamicCast(OSBoolean, tmpObject);
841 if (tmpBoolean->getValue()) owHeader->owFlags |= tmpOffset;
842 break;
843
844 case kOFVariableTypeNumber :
845 tmpNumber = OSDynamicCast(OSNumber, tmpObject);
846 owHeader->owNumbers[tmpOffset] = tmpNumber->unsigned32BitValue();
847 break;
848
849 case kOFVariableTypeString :
850 tmpString = OSDynamicCast(OSString, tmpObject);
91447636 851 tmpData = (const UInt8 *)tmpString->getCStringNoCopy();
1c79356b
A
852 tmpDataLength = tmpString->getLength();
853
854 if ((curOffset - tmpDataLength) < sizeof(OWVariablesHeader)) {
855 ok = false;
856 break;
857 }
858
859 owHeader->owStrings[tmpOffset].length = tmpDataLength;
860 curOffset -= tmpDataLength;
861 owHeader->owStrings[tmpOffset].offset = curOffset + _ofPartitionOffset;
862 if (tmpDataLength != 0)
863 bcopy(tmpData, buffer + curOffset, tmpDataLength);
864 break;
865 }
866 }
867
868 if (ok) {
869 owHeader->owHere = _ofPartitionOffset + sizeof(OWVariablesHeader);
870 owHeader->owTop = _ofPartitionOffset + curOffset;
871 owHeader->owNext = 0;
872
873 owHeader->owChecksum = 0;
874 owHeader->owChecksum = ~generateOWChecksum(buffer);
875
876 bcopy(buffer, _ofImage, _ofPartitionSize);
877 }
878
879 IODelete(buffer, UInt8, _ofPartitionSize);
880
881 if (!ok) return kIOReturnBadArgument;
882 }
883
884 _ofImageDirty = false;
885 _nvramImageDirty = true;
886
887 return kIOReturnSuccess;
888}
889
890struct OFVariable {
91447636
A
891 const char *variableName;
892 UInt32 variableType;
893 UInt32 variablePerm;
894 SInt32 variableOffset;
1c79356b
A
895};
896typedef struct OFVariable OFVariable;
897
898enum {
899 kOWVariableOffsetNumber = 8,
900 kOWVariableOffsetString = 17
901};
902
903OFVariable gOFVariables[] = {
904 {"little-endian?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 0},
905 {"real-mode?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 1},
906 {"auto-boot?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 2},
907 {"diag-switch?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 3},
908 {"fcode-debug?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 4},
909 {"oem-banner?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 5},
910 {"oem-logo?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 6},
911 {"use-nvramrc?", kOFVariableTypeBoolean, kOFVariablePermUserRead, 7},
912 {"use-generic?", kOFVariableTypeBoolean, kOFVariablePermUserRead, -1},
913 {"default-mac-address?", kOFVariableTypeBoolean, kOFVariablePermUserRead,-1},
914 {"real-base", kOFVariableTypeNumber, kOFVariablePermUserRead, 8},
915 {"real-size", kOFVariableTypeNumber, kOFVariablePermUserRead, 9},
916 {"virt-base", kOFVariableTypeNumber, kOFVariablePermUserRead, 10},
917 {"virt-size", kOFVariableTypeNumber, kOFVariablePermUserRead, 11},
918 {"load-base", kOFVariableTypeNumber, kOFVariablePermUserRead, 12},
919 {"pci-probe-list", kOFVariableTypeNumber, kOFVariablePermUserRead, 13},
920 {"pci-probe-mask", kOFVariableTypeNumber, kOFVariablePermUserRead, -1},
921 {"screen-#columns", kOFVariableTypeNumber, kOFVariablePermUserRead, 14},
922 {"screen-#rows", kOFVariableTypeNumber, kOFVariablePermUserRead, 15},
923 {"selftest-#megs", kOFVariableTypeNumber, kOFVariablePermUserRead, 16},
924 {"boot-device", kOFVariableTypeString, kOFVariablePermUserRead, 17},
925 {"boot-file", kOFVariableTypeString, kOFVariablePermUserRead, 18},
926 {"boot-screen", kOFVariableTypeString, kOFVariablePermUserRead, -1},
927 {"console-screen", kOFVariableTypeString, kOFVariablePermUserRead, -1},
928 {"diag-device", kOFVariableTypeString, kOFVariablePermUserRead, 19},
929 {"diag-file", kOFVariableTypeString, kOFVariablePermUserRead, 20},
930 {"input-device", kOFVariableTypeString, kOFVariablePermUserRead, 21},
931 {"output-device", kOFVariableTypeString, kOFVariablePermUserRead, 22},
932 {"input-device-1", kOFVariableTypeString, kOFVariablePermUserRead, -1},
933 {"output-device-1", kOFVariableTypeString, kOFVariablePermUserRead, -1},
934 {"mouse-device", kOFVariableTypeString, kOFVariablePermUserRead, -1},
935 {"oem-banner", kOFVariableTypeString, kOFVariablePermUserRead, 23},
936 {"oem-logo", kOFVariableTypeString, kOFVariablePermUserRead, 24},
937 {"nvramrc", kOFVariableTypeString, kOFVariablePermUserRead, 25},
938 {"boot-command", kOFVariableTypeString, kOFVariablePermUserRead, 26},
939 {"default-client-ip", kOFVariableTypeString, kOFVariablePermUserRead, -1},
940 {"default-server-ip", kOFVariableTypeString, kOFVariablePermUserRead, -1},
941 {"default-gateway-ip", kOFVariableTypeString, kOFVariablePermUserRead, -1},
942 {"default-subnet-mask", kOFVariableTypeString, kOFVariablePermUserRead, -1},
943 {"default-router-ip", kOFVariableTypeString, kOFVariablePermUserRead, -1},
944 {"boot-script", kOFVariableTypeString, kOFVariablePermUserRead, -1},
945 {"boot-args", kOFVariableTypeString, kOFVariablePermUserRead, -1},
946 {"aapl,pci", kOFVariableTypeData, kOFVariablePermRootOnly, -1},
947 {"security-mode", kOFVariableTypeString, kOFVariablePermUserRead, -1},
948 {"security-password", kOFVariableTypeData, kOFVariablePermRootOnly, -1},
91447636 949 {"boot-image", kOFVariableTypeData, kOFVariablePermUserWrite, -1},
cf7d32b8 950 {"com.apple.System.fp-state", kOFVariableTypeData, kOFVariablePermKernelOnly, -1},
593a1d5f
A
951#if CONFIG_EMBEDDED
952 {"backlight-level", kOFVariableTypeData, kOFVariablePermUserWrite, -1},
953#endif
1c79356b
A
954 {0, kOFVariableTypeData, kOFVariablePermUserRead, -1}
955};
956
957UInt32 IODTNVRAM::getOFVariableType(const OSSymbol *propSymbol) const
958{
959 OFVariable *ofVar;
960
961 ofVar = gOFVariables;
962 while (1) {
963 if ((ofVar->variableName == 0) ||
964 propSymbol->isEqualTo(ofVar->variableName)) break;
965 ofVar++;
966 }
967
968 return ofVar->variableType;
969}
970
971UInt32 IODTNVRAM::getOFVariablePerm(const OSSymbol *propSymbol) const
972{
973 OFVariable *ofVar;
974
975 ofVar = gOFVariables;
976 while (1) {
977 if ((ofVar->variableName == 0) ||
978 propSymbol->isEqualTo(ofVar->variableName)) break;
979 ofVar++;
980 }
981
982 return ofVar->variablePerm;
983}
984
985bool IODTNVRAM::getOWVariableInfo(UInt32 variableNumber, const OSSymbol **propSymbol,
986 UInt32 *propType, UInt32 *propOffset)
987{
988 OFVariable *ofVar;
989
990 ofVar = gOFVariables;
991 while (1) {
992 if (ofVar->variableName == 0) return false;
993
994 if (ofVar->variableOffset == (SInt32) variableNumber) break;
995
996 ofVar++;
997 }
998
999 *propSymbol = OSSymbol::withCStringNoCopy(ofVar->variableName);
1000 *propType = ofVar->variableType;
1001
1002 switch (*propType) {
1003 case kOFVariableTypeBoolean :
1004 *propOffset = 1 << (31 - variableNumber);
1005 break;
1006
1007 case kOFVariableTypeNumber :
1008 *propOffset = variableNumber - kOWVariableOffsetNumber;
1009 break;
1010
1011 case kOFVariableTypeString :
1012 *propOffset = variableNumber - kOWVariableOffsetString;
1013 break;
1014 }
1015
1016 return true;
1017}
1018
1019bool IODTNVRAM::convertPropToObject(UInt8 *propName, UInt32 propNameLength,
1020 UInt8 *propData, UInt32 propDataLength,
1021 const OSSymbol **propSymbol,
1022 OSObject **propObject)
1023{
1024 UInt32 propType;
1025 const OSSymbol *tmpSymbol;
1026 OSObject *tmpObject;
1027 OSNumber *tmpNumber;
1028 OSString *tmpString;
1029
1030 // Create the symbol.
1031 propName[propNameLength] = '\0';
1032 tmpSymbol = OSSymbol::withCString((const char *)propName);
1033 propName[propNameLength] = '=';
1034 if (tmpSymbol == 0) {
1035 return false;
1036 }
1037
1038 propType = getOFVariableType(tmpSymbol);
1039
1040 // Create the object.
1041 tmpObject = 0;
1042 switch (propType) {
1043 case kOFVariableTypeBoolean :
1044 if (!strncmp("true", (const char *)propData, propDataLength)) {
1045 tmpObject = kOSBooleanTrue;
1046 } else if (!strncmp("false", (const char *)propData, propDataLength)) {
1047 tmpObject = kOSBooleanFalse;
1048 }
1049 break;
1050
1051 case kOFVariableTypeNumber :
1052 tmpNumber = OSNumber::withNumber(strtol((const char *)propData, 0, 0), 32);
1053 if (tmpNumber != 0) tmpObject = tmpNumber;
1054 break;
1055
1056 case kOFVariableTypeString :
1057 tmpString = OSString::withCString((const char *)propData);
1058 if (tmpString != 0) tmpObject = tmpString;
1059 break;
1060
1061 case kOFVariableTypeData :
1062 tmpObject = unescapeBytesToData(propData, propDataLength);
1063 break;
1064 }
1065
1066 if (tmpObject == 0) {
1067 tmpSymbol->release();
1068 return false;
1069 }
1070
1071 *propSymbol = tmpSymbol;
1072 *propObject = tmpObject;
1073
1074 return true;
1075}
1076
1077bool IODTNVRAM::convertObjectToProp(UInt8 *buffer, UInt32 *length,
1078 const OSSymbol *propSymbol, OSObject *propObject)
1079{
91447636 1080 const UInt8 *propName;
1c79356b
A
1081 UInt32 propNameLength, propDataLength;
1082 UInt32 propType, tmpValue;
1083 OSBoolean *tmpBoolean = 0;
1084 OSNumber *tmpNumber = 0;
1085 OSString *tmpString = 0;
1086 OSData *tmpData = 0;
1087
91447636 1088 propName = (const UInt8 *)propSymbol->getCStringNoCopy();
1c79356b
A
1089 propNameLength = propSymbol->getLength();
1090 propType = getOFVariableType(propSymbol);
1091
1092 // Get the size of the data.
1093 propDataLength = 0xFFFFFFFF;
1094 switch (propType) {
1095 case kOFVariableTypeBoolean :
1096 tmpBoolean = OSDynamicCast(OSBoolean, propObject);
1097 if (tmpBoolean != 0) propDataLength = 5;
1098 break;
1099
1100 case kOFVariableTypeNumber :
1101 tmpNumber = OSDynamicCast(OSNumber, propObject);
1102 if (tmpNumber != 0) propDataLength = 10;
1103 break;
1104
1105 case kOFVariableTypeString :
1106 tmpString = OSDynamicCast(OSString, propObject);
1107 if (tmpString != 0) propDataLength = tmpString->getLength();
1108 break;
1109
1110 case kOFVariableTypeData :
1111 tmpData = OSDynamicCast(OSData, propObject);
1112 if (tmpData != 0) {
1113 tmpData = escapeDataToData(tmpData);
1114 propDataLength = tmpData->getLength();
1115 }
1116 break;
1117 }
1118
1119 // Make sure the propertySize is known and will fit.
1120 if (propDataLength == 0xFFFFFFFF) return false;
1121 if ((propNameLength + propDataLength + 2) > *length) return false;
1122
1123 // Copy the property name equal sign.
2d21ac55 1124 buffer += snprintf((char *)buffer, *length, "%s=", propName);
1c79356b
A
1125
1126 switch (propType) {
1127 case kOFVariableTypeBoolean :
1128 if (tmpBoolean->getValue()) {
2d21ac55 1129 strlcpy((char *)buffer, "true", *length - propNameLength);
1c79356b 1130 } else {
2d21ac55 1131 strlcpy((char *)buffer, "false", *length - propNameLength);
1c79356b
A
1132 }
1133 break;
1134
1135 case kOFVariableTypeNumber :
1136 tmpValue = tmpNumber->unsigned32BitValue();
1137 if (tmpValue == 0xFFFFFFFF) {
2d21ac55 1138 strlcpy((char *)buffer, "-1", *length - propNameLength);
1c79356b 1139 } else if (tmpValue < 1000) {
b0d623f7 1140 snprintf((char *)buffer, *length - propNameLength, "%d", (uint32_t)tmpValue);
1c79356b 1141 } else {
b0d623f7 1142 snprintf((char *)buffer, *length - propNameLength, "0x%x", (uint32_t)tmpValue);
1c79356b
A
1143 }
1144 break;
1145
1146 case kOFVariableTypeString :
2d21ac55 1147 strlcpy((char *)buffer, tmpString->getCStringNoCopy(), *length - propNameLength);
1c79356b
A
1148 break;
1149
1150 case kOFVariableTypeData :
1151 bcopy(tmpData->getBytesNoCopy(), buffer, propDataLength);
1152 tmpData->release();
1153 break;
1154 }
1155
1156 propDataLength = strlen((const char *)buffer);
1157
1158 *length = propNameLength + propDataLength + 2;
1159
1160 return true;
1161}
1162
1163
1164UInt16 IODTNVRAM::generateOWChecksum(UInt8 *buffer)
1165{
1166 UInt32 cnt, checksum = 0;
1167 UInt16 *tmpBuffer = (UInt16 *)buffer;
1168
1169 for (cnt = 0; cnt < _ofPartitionSize / 2; cnt++)
1170 checksum += tmpBuffer[cnt];
1171
1172 return checksum % 0x0000FFFF;
1173}
1174
1175bool IODTNVRAM::validateOWChecksum(UInt8 *buffer)
1176{
1177 UInt32 cnt, checksum, sum = 0;
1178 UInt16 *tmpBuffer = (UInt16 *)buffer;
1179
1180 for (cnt = 0; cnt < _ofPartitionSize / 2; cnt++)
1181 sum += tmpBuffer[cnt];
1182
1183 checksum = (sum >> 16) + (sum & 0x0000FFFF);
1184 if (checksum == 0x10000) checksum--;
1185 checksum = (checksum ^ 0x0000FFFF) & 0x0000FFFF;
1186
1187 return checksum == 0;
1188}
1189
1190void IODTNVRAM::updateOWBootArgs(const OSSymbol *key, OSObject *value)
1191{
91447636
A
1192 bool wasBootArgs, bootr = false;
1193 UInt32 cnt;
1194 OSString *tmpString, *bootCommand, *bootArgs = 0;
1195 const UInt8 *bootCommandData, *bootArgsData;
1196 UInt8 *tmpData;
1197 UInt32 bootCommandDataLength, bootArgsDataLength, tmpDataLength;
1c79356b
A
1198
1199 tmpString = OSDynamicCast(OSString, value);
1200 if (tmpString == 0) return;
1201
1202 if (key->isEqualTo("boot-command")) {
1203 wasBootArgs = false;
1204 bootCommand = tmpString;
1205 } else if (key->isEqualTo("boot-args")) {
1206 wasBootArgs = true;
1207 bootArgs = tmpString;
1208 bootCommand = OSDynamicCast(OSString, _ofDict->getObject("boot-command"));
1209 if (bootCommand == 0) return;
1210 } else return;
1211
91447636 1212 bootCommandData = (const UInt8 *)bootCommand->getCStringNoCopy();
1c79356b
A
1213 bootCommandDataLength = bootCommand->getLength();
1214
1215 if (bootCommandData == 0) return;
1216
1217 for (cnt = 0; cnt < bootCommandDataLength; cnt++) {
1218 if ((bootCommandData[cnt] == 'b') &&
1219 !strncmp("bootr", (const char *)bootCommandData + cnt, 5)) {
1220 cnt += 5;
1221 while (bootCommandData[cnt] == ' ') cnt++;
1222 bootr = true;
1223 break;
1224 }
1225 }
1226 if (!bootr) {
1227 _ofDict->removeObject("boot-args");
1228 return;
1229 }
1230
1231 if (wasBootArgs) {
91447636 1232 bootArgsData = (const UInt8 *)bootArgs->getCStringNoCopy();
1c79356b
A
1233 bootArgsDataLength = bootArgs->getLength();
1234 if (bootArgsData == 0) return;
1235
1236 tmpDataLength = cnt + bootArgsDataLength;
1237 tmpData = IONew(UInt8, tmpDataLength + 1);
1238 if (tmpData == 0) return;
1239
2d21ac55
A
1240 cnt -= strlcpy((char *)tmpData, (const char *)bootCommandData, cnt);
1241 strlcat((char *)tmpData, (const char *)bootArgsData, cnt);
1c79356b
A
1242
1243 bootCommand = OSString::withCString((const char *)tmpData);
1244 if (bootCommand != 0) {
1245 _ofDict->setObject("boot-command", bootCommand);
1246 bootCommand->release();
1247 }
1248
1249 IODelete(tmpData, UInt8, tmpDataLength + 1);
1250 } else {
1251 bootArgs = OSString::withCString((const char *)(bootCommandData + cnt));
1252 if (bootArgs != 0) {
1253 _ofDict->setObject("boot-args", bootArgs);
1254 bootArgs->release();
1255 }
1256 }
1257}
1258
1259
1260// Private methods for Name Registry access.
1261
1262enum {
1263 kMaxNVNameLength = 4,
1264 kMaxNVDataLength = 8
1265};
1266
1c79356b
A
1267struct NVRAMProperty
1268{
1269 IONVRAMDescriptor header;
1270 UInt8 nameLength;
1271 UInt8 name[ kMaxNVNameLength ];
1272 UInt8 dataLength;
1273 UInt8 data[ kMaxNVDataLength ];
1274};
1c79356b
A
1275
1276bool IODTNVRAM::searchNVRAMProperty(IONVRAMDescriptor *hdr, UInt32 *where)
1277{
1278 UInt32 offset;
1279 SInt32 nvEnd;
1280
1281 nvEnd = *((UInt16 *)_nrImage);
1282 if(getPlatform()->getBootROMType()) {
1283 // on NewWorld, offset to partition start
1284 nvEnd -= 0x100;
1285 } else {
1286 // on old world, absolute
1287 nvEnd -= _nrPartitionOffset;
1288 }
1289 if((nvEnd < 0) || (nvEnd >= kIODTNVRAMNameRegistrySize))
1290 nvEnd = 2;
1291
1292 offset = 2;
1293 while ((offset + sizeof(NVRAMProperty)) <= (UInt32)nvEnd) {
1294 if (bcmp(_nrImage + offset, hdr, sizeof(*hdr)) == 0) {
1295 *where = offset;
1296 return true;
1297 }
1298 offset += sizeof(NVRAMProperty);
1299 }
1300
1301 if ((nvEnd + sizeof(NVRAMProperty)) <= kIODTNVRAMNameRegistrySize)
1302 *where = nvEnd;
1303 else
1304 *where = 0;
1305
1306 return false;
1307}
1308
1309IOReturn IODTNVRAM::readNVRAMPropertyType0(IORegistryEntry *entry,
1310 const OSSymbol **name,
1311 OSData **value)
1312{
1313 IONVRAMDescriptor hdr;
1314 NVRAMProperty *prop;
1315 IOByteCount length;
1316 UInt32 offset;
1317 IOReturn err;
1318 char nameBuf[kMaxNVNameLength + 1];
1319
1320 if (_nrImage == 0) return kIOReturnUnsupported;
1321 if ((entry == 0) || (name == 0) || (value == 0)) return kIOReturnBadArgument;
1322
1323 err = IODTMakeNVDescriptor(entry, &hdr);
1324 if (err != kIOReturnSuccess) return err;
1325
1326 if (searchNVRAMProperty(&hdr, &offset)) {
1327 prop = (NVRAMProperty *)(_nrImage + offset);
1328
1329 length = prop->nameLength;
1330 if (length > kMaxNVNameLength) length = kMaxNVNameLength;
1331 strncpy(nameBuf, (const char *)prop->name, length);
1332 nameBuf[length] = 0;
1333 *name = OSSymbol::withCString(nameBuf);
1334
1335 length = prop->dataLength;
1336 if (length > kMaxNVDataLength) length = kMaxNVDataLength;
1337 *value = OSData::withBytes(prop->data, length);
1338
1339 if ((*name != 0) && (*value != 0)) return kIOReturnSuccess;
1340 else return kIOReturnNoMemory;
1341 }
1342
1343 return kIOReturnNoResources;
1344}
1345
1346IOReturn IODTNVRAM::writeNVRAMPropertyType0(IORegistryEntry *entry,
1347 const OSSymbol *name,
1348 OSData *value)
1349{
1350 IONVRAMDescriptor hdr;
1351 NVRAMProperty *prop;
1352 IOByteCount nameLength;
1353 IOByteCount dataLength;
1354 UInt32 offset;
1355 IOReturn err;
1356 UInt16 nvLength;
1357 bool exists;
1358
1359 if (_nrImage == 0) return kIOReturnUnsupported;
1360 if ((entry == 0) || (name == 0) || (value == 0)) return kIOReturnBadArgument;
1361
1362 nameLength = name->getLength();
1363 dataLength = value->getLength();
1364 if (nameLength > kMaxNVNameLength) return kIOReturnNoSpace;
1365 if (dataLength > kMaxNVDataLength) return kIOReturnNoSpace;
1366
1367 err = IODTMakeNVDescriptor(entry, &hdr);
1368 if (err != kIOReturnSuccess) return err;
1369
1370 exists = searchNVRAMProperty(&hdr, &offset);
1371 if (offset == 0) return kIOReturnNoMemory;
1372
1373 prop = (NVRAMProperty *)(_nrImage + offset);
1374 if (!exists) bcopy(&hdr, &prop->header, sizeof(hdr));
1375
1376 prop->nameLength = nameLength;
1377 bcopy(name->getCStringNoCopy(), prop->name, nameLength);
1378 prop->dataLength = dataLength;
1379 bcopy(value->getBytesNoCopy(), prop->data, dataLength);
1380
1381 if (!exists) {
1382 nvLength = offset + sizeof(NVRAMProperty);
1383 if (getPlatform()->getBootROMType())
1384 nvLength += 0x100;
1385 else
1386 nvLength += _nrPartitionOffset;
1387 *((UInt16 *)_nrImage) = nvLength;
1388 }
1389
1390 _nvramImageDirty = true;
1391
1392 return err;
1393}
1394
91447636 1395OSData *IODTNVRAM::unescapeBytesToData(const UInt8 *bytes, UInt32 length)
1c79356b
A
1396{
1397 OSData *data = 0;
1398 UInt32 totalLength = 0;
1399 UInt32 cnt, cnt2;
1400 UInt8 byte;
1401 bool ok;
1402
1403 // Calculate the actual length of the data.
1404 ok = true;
1405 totalLength = 0;
1406 for (cnt = 0; cnt < length;) {
1407 byte = bytes[cnt++];
1408 if (byte == 0xFF) {
1409 byte = bytes[cnt++];
1410 if (byte == 0x00) {
1411 ok = false;
1412 break;
1413 }
1414 cnt2 = byte & 0x7F;
1415 } else
1416 cnt2 = 1;
1417 totalLength += cnt2;
1418 }
1419
1420 if (ok) {
1421 // Create an empty OSData of the correct size.
1422 data = OSData::withCapacity(totalLength);
1423 if (data != 0) {
1424 for (cnt = 0; cnt < length;) {
1425 byte = bytes[cnt++];
1426 if (byte == 0xFF) {
1427 byte = bytes[cnt++];
1428 cnt2 = byte & 0x7F;
1429 byte = (byte & 0x80) ? 0xFF : 0x00;
1430 } else
1431 cnt2 = 1;
1432 data->appendByte(byte, cnt2);
1433 }
1434 }
1435 }
1436
1437 return data;
1438}
1439
1440OSData * IODTNVRAM::escapeDataToData(OSData * value)
1441{
91447636
A
1442 OSData * result;
1443 const UInt8 * startPtr;
1444 const UInt8 * endPtr;
1445 const UInt8 * wherePtr;
1446 UInt8 byte;
1447 bool ok = true;
1c79356b 1448
91447636
A
1449 wherePtr = (const UInt8 *) value->getBytesNoCopy();
1450 endPtr = wherePtr + value->getLength();
1c79356b 1451
91447636 1452 result = OSData::withCapacity(endPtr - wherePtr);
1c79356b
A
1453 if (!result)
1454 return result;
1455
91447636
A
1456 while (wherePtr < endPtr) {
1457 startPtr = wherePtr;
1458 byte = *wherePtr++;
1c79356b
A
1459 if ((byte == 0x00) || (byte == 0xFF)) {
1460 for (;
91447636
A
1461 ((wherePtr - startPtr) < 0x80) && (wherePtr < endPtr) && (byte == *wherePtr);
1462 wherePtr++) {}
1c79356b 1463 ok &= result->appendByte(0xff, 1);
91447636 1464 byte = (byte & 0x80) | (wherePtr - startPtr);
1c79356b
A
1465 }
1466 ok &= result->appendByte(byte, 1);
1467 }
1468 ok &= result->appendByte(0, 1);
1469
1470 if (!ok) {
1471 result->release();
1472 result = 0;
1473 }
1474
1475 return result;
1476}
1477
91447636
A
1478static bool IsApplePropertyName(const char * propName)
1479{
1480 char c;
1481 while ((c = *propName++)) {
1482 if ((c >= 'A') && (c <= 'Z'))
1483 break;
1484 }
1485
1486 return (c == 0);
1487}
1488
1c79356b
A
1489IOReturn IODTNVRAM::readNVRAMPropertyType1(IORegistryEntry *entry,
1490 const OSSymbol **name,
1491 OSData **value)
1492{
91447636
A
1493 IOReturn err = kIOReturnNoResources;
1494 OSData *data;
1495 const UInt8 *startPtr;
1496 const UInt8 *endPtr;
1497 const UInt8 *wherePtr;
1498 const UInt8 *nvPath = 0;
1499 const char *nvName = 0;
1500 const char *resultName = 0;
1501 const UInt8 *resultValue = 0;
1502 UInt32 resultValueLen = 0;
1503 UInt8 byte;
1c79356b
A
1504
1505 if (_ofDict == 0) return err;
1506 data = OSDynamicCast(OSData, _ofDict->getObject(_registryPropertiesKey));
1507 if (data == 0) return err;
1508
91447636
A
1509 startPtr = (const UInt8 *) data->getBytesNoCopy();
1510 endPtr = startPtr + data->getLength();
1c79356b 1511
91447636
A
1512 wherePtr = startPtr;
1513 while (wherePtr < endPtr) {
1514 byte = *(wherePtr++);
1c79356b
A
1515 if (byte)
1516 continue;
1517
1518 if (nvPath == 0)
91447636 1519 nvPath = startPtr;
1c79356b 1520 else if (nvName == 0)
91447636 1521 nvName = (const char *) startPtr;
ac5ea4a9
A
1522 else {
1523 IORegistryEntry * compareEntry = IORegistryEntry::fromPath((const char *) nvPath, gIODTPlane);
ac5ea4a9
A
1524 if (compareEntry)
1525 compareEntry->release();
91447636
A
1526 if (entry == compareEntry) {
1527 bool appleProp = IsApplePropertyName(nvName);
1528 if (!appleProp || !resultName) {
1529 resultName = nvName;
1530 resultValue = startPtr;
1531 resultValueLen = wherePtr - startPtr - 1;
1532 }
1533 if (!appleProp)
1534 break;
1535 }
1536 nvPath = 0;
1537 nvName = 0;
ac5ea4a9 1538 }
91447636
A
1539 startPtr = wherePtr;
1540 }
1541 if (resultName) {
1542 *name = OSSymbol::withCString(resultName);
1543 *value = unescapeBytesToData(resultValue, resultValueLen);
1544 if ((*name != 0) && (*value != 0))
1545 err = kIOReturnSuccess;
1546 else
1547 err = kIOReturnNoMemory;
1c79356b 1548 }
1c79356b
A
1549 return err;
1550}
1551
1552IOReturn IODTNVRAM::writeNVRAMPropertyType1(IORegistryEntry *entry,
1553 const OSSymbol *propName,
1554 OSData *value)
1555{
91447636
A
1556 OSData *oldData;
1557 OSData *data = 0;
1558 const UInt8 *startPtr;
1559 const UInt8 *propStart;
1560 const UInt8 *endPtr;
1561 const UInt8 *wherePtr;
1562 const UInt8 *nvPath = 0;
1563 const char *nvName = 0;
1c79356b
A
1564 const char * comp;
1565 const char * name;
91447636
A
1566 UInt8 byte;
1567 bool ok = true;
1568 bool settingAppleProp;
1c79356b
A
1569
1570 if (_ofDict == 0) return kIOReturnNoResources;
1571
91447636
A
1572 settingAppleProp = IsApplePropertyName(propName->getCStringNoCopy());
1573
1c79356b
A
1574 // copy over existing properties for other entries
1575
1576 oldData = OSDynamicCast(OSData, _ofDict->getObject(_registryPropertiesKey));
1577 if (oldData) {
91447636
A
1578 startPtr = (const UInt8 *) oldData->getBytesNoCopy();
1579 endPtr = startPtr + oldData->getLength();
1c79356b 1580
91447636
A
1581 propStart = startPtr;
1582 wherePtr = startPtr;
1583 while (wherePtr < endPtr) {
1584 byte = *(wherePtr++);
1c79356b
A
1585 if (byte)
1586 continue;
1587 if (nvPath == 0)
91447636 1588 nvPath = startPtr;
1c79356b 1589 else if (nvName == 0)
91447636 1590 nvName = (const char *) startPtr;
ac5ea4a9
A
1591 else {
1592 IORegistryEntry * compareEntry = IORegistryEntry::fromPath((const char *) nvPath, gIODTPlane);
ac5ea4a9
A
1593 if (compareEntry)
1594 compareEntry->release();
91447636
A
1595 if (entry == compareEntry) {
1596 if ((settingAppleProp && propName->isEqualTo(nvName))
1597 || (!settingAppleProp && !IsApplePropertyName(nvName))) {
1598 // delete old property (nvPath -> wherePtr)
1599 data = OSData::withBytes(propStart, nvPath - propStart);
1600 if (data)
1601 ok &= data->appendBytes(wherePtr, endPtr - wherePtr);
1602 break;
1603 }
1604 }
1605 nvPath = 0;
1606 nvName = 0;
ac5ea4a9 1607 }
1c79356b 1608
91447636 1609 startPtr = wherePtr;
1c79356b
A
1610 }
1611 }
1612
1613 // make the new property
1614
1615 if (!data) {
1616 if (oldData)
1617 data = OSData::withData(oldData);
1618 else
1619 data = OSData::withCapacity(16);
1620 if (!data)
1621 return kIOReturnNoMemory;
1622 }
1623
91447636
A
1624 if (value && value->getLength()) {
1625 // get entries in path
1626 OSArray *array = OSArray::withCapacity(5);
1627 if (!array) {
1628 data->release();
1629 return kIOReturnNoMemory;
1630 }
1631 do
1632 array->setObject(entry);
1633 while ((entry = entry->getParentEntry(gIODTPlane)));
1634
1635 // append path
1636 for (int i = array->getCount() - 3;
1637 (entry = (IORegistryEntry *) array->getObject(i));
1638 i--) {
1639
1640 name = entry->getName(gIODTPlane);
1641 comp = entry->getLocation(gIODTPlane);
2d21ac55
A
1642 if( comp && (0 == strncmp("pci", name, sizeof("pci")))
1643 && (0 == strncmp("80000000", comp, sizeof("80000000")))) {
91447636
A
1644 // yosemite hack
1645 comp = "/pci@80000000";
1646 } else {
1647 if (comp)
1648 ok &= data->appendBytes("/@", 2);
1649 else {
1650 if (!name)
1651 continue;
1652 ok &= data->appendByte('/', 1);
1653 comp = name;
1654 }
1655 }
1656 ok &= data->appendBytes(comp, strlen(comp));
1657 }
1658 ok &= data->appendByte(0, 1);
1659 array->release();
1660
1661 // append prop name
1662 ok &= data->appendBytes(propName->getCStringNoCopy(), propName->getLength() + 1);
1663
1664 // append escaped data
1665 oldData = escapeDataToData(value);
1666 ok &= (oldData != 0);
1667 if (ok)
1668 ok &= data->appendBytes(oldData);
1669 }
1c79356b
A
1670 if (ok) {
1671 ok = _ofDict->setObject(_registryPropertiesKey, data);
1672 if (ok)
1673 _ofImageDirty = true;
1674 }
1675 data->release();
1676
1677 return ok ? kIOReturnSuccess : kIOReturnNoMemory;
1678}
6d2010ae
A
1679
1680bool IODTNVRAM::safeToSync(void)
1681{
1682 AbsoluteTime delta;
1683 UInt64 delta_ns;
1684 SInt32 delta_secs;
1685
1686 // delta interval went by
1687 clock_get_uptime(&delta);
1688
1689 // Figure it in seconds.
1690 absolutetime_to_nanoseconds(delta, &delta_ns);
1691 delta_secs = (SInt32)(delta_ns / NSEC_PER_SEC);
1692
1693 if ((delta_secs > (_lastDeviceSync + MIN_SYNC_NOW_INTERVAL)) || _freshInterval)
1694 {
1695 _lastDeviceSync = delta_secs;
1696 _freshInterval = FALSE;
1697 return TRUE;
1698 }
1699
1700 return FALSE;
1701}