1e7776783SApple OSS Distributions /*
2e7776783SApple OSS Distributions  * Copyright (c) 2021-2022 Apple Inc. All rights reserved.
3e7776783SApple OSS Distributions  *
4e7776783SApple OSS Distributions  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5e7776783SApple OSS Distributions  *
6e7776783SApple OSS Distributions  * This file contains Original Code and/or Modifications of Original Code
7e7776783SApple OSS Distributions  * as defined in and that are subject to the Apple Public Source License
8e7776783SApple OSS Distributions  * Version 2.0 (the 'License'). You may not use this file except in
9e7776783SApple OSS Distributions  * compliance with the License. The rights granted to you under the License
10e7776783SApple OSS Distributions  * may not be used to create, or enable the creation or redistribution of,
11e7776783SApple OSS Distributions  * unlawful or unlicensed copies of an Apple operating system, or to
12e7776783SApple OSS Distributions  * circumvent, violate, or enable the circumvention or violation of, any
13e7776783SApple OSS Distributions  * terms of an Apple operating system software license agreement.
14e7776783SApple OSS Distributions  *
15e7776783SApple OSS Distributions  * Please obtain a copy of the License at
16e7776783SApple OSS Distributions  * http://www.opensource.apple.com/apsl/ and read it before using this file.
17e7776783SApple OSS Distributions  *
18e7776783SApple OSS Distributions  * The Original Code and all software distributed under the License are
19e7776783SApple OSS Distributions  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20e7776783SApple OSS Distributions  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21e7776783SApple OSS Distributions  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22e7776783SApple OSS Distributions  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23e7776783SApple OSS Distributions  * Please see the License for the specific language governing rights and
24e7776783SApple OSS Distributions  * limitations under the License.
25e7776783SApple OSS Distributions  *
26e7776783SApple OSS Distributions  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27e7776783SApple OSS Distributions  */
28e7776783SApple OSS Distributions 
29e7776783SApple OSS Distributions #include <libkern/libkern.h>
30e7776783SApple OSS Distributions 
31e7776783SApple OSS Distributions #define VARIABLE_STORE_SIGNATURE         'NVV3'
32e7776783SApple OSS Distributions 
33e7776783SApple OSS Distributions // Variable Store Version
34e7776783SApple OSS Distributions #define VARIABLE_STORE_VERSION           0x1
35e7776783SApple OSS Distributions 
36e7776783SApple OSS Distributions #define VARIABLE_DATA                    0x55AA
37e7776783SApple OSS Distributions #define INVALIDATED_VARIABLE_DATA        0x0000
38e7776783SApple OSS Distributions 
39e7776783SApple OSS Distributions // Variable State flags
40e7776783SApple OSS Distributions #define VAR_IN_DELETED_TRANSITION     0xFE  // Variable is in obsolete transistion
41e7776783SApple OSS Distributions #define VAR_DELETED                   0xFD  // Variable is obsolete
42e7776783SApple OSS Distributions #define VAR_INACTIVE                  0xFB  // Variable is inactive due to failing CRC
43e7776783SApple OSS Distributions #define VAR_ADDED                     0x7F  // Variable has been completely added
44e7776783SApple OSS Distributions 
45e7776783SApple OSS Distributions // No changes needed on save
46e7776783SApple OSS Distributions #define VAR_NEW_STATE_NONE            0x01
47e7776783SApple OSS Distributions // Remove existing entry on save
48e7776783SApple OSS Distributions #define VAR_NEW_STATE_REMOVE          0x02
49e7776783SApple OSS Distributions // Add new value on save, mark previous as inactive
50e7776783SApple OSS Distributions #define VAR_NEW_STATE_APPEND          0x03
51e7776783SApple OSS Distributions 
52e7776783SApple OSS Distributions #pragma pack(1)
53e7776783SApple OSS Distributions struct v3_store_header {
54e7776783SApple OSS Distributions 	uint32_t     name;
55e7776783SApple OSS Distributions 	uint32_t     size;
56e7776783SApple OSS Distributions 	uint32_t     generation;
57e7776783SApple OSS Distributions 	uint8_t      state;
58e7776783SApple OSS Distributions 	uint8_t      flags;
59e7776783SApple OSS Distributions 	uint8_t      version;
60e7776783SApple OSS Distributions 	uint8_t      reserved1;
61e7776783SApple OSS Distributions 	uint32_t     system_size;
62e7776783SApple OSS Distributions 	uint32_t     common_size;
63e7776783SApple OSS Distributions };
64e7776783SApple OSS Distributions 
65e7776783SApple OSS Distributions struct v3_var_header {
66e7776783SApple OSS Distributions 	uint16_t     startId;
67e7776783SApple OSS Distributions 	uint8_t      state;
68e7776783SApple OSS Distributions 	uint8_t      reserved;
69e7776783SApple OSS Distributions 	uint32_t     attributes;
70e7776783SApple OSS Distributions 	uint32_t     nameSize;
71e7776783SApple OSS Distributions 	uint32_t     dataSize;
72e7776783SApple OSS Distributions 	uuid_t       guid;
73e7776783SApple OSS Distributions 	uint32_t     crc;
74e7776783SApple OSS Distributions 	uint8_t      name_data_buf[];
75e7776783SApple OSS Distributions };
76e7776783SApple OSS Distributions #pragma pack()
77e7776783SApple OSS Distributions 
78e7776783SApple OSS Distributions struct nvram_v3_var_entry {
79e7776783SApple OSS Distributions 	uint8_t                new_state;
80e7776783SApple OSS Distributions 	size_t                 existing_offset;
81e7776783SApple OSS Distributions 	struct v3_var_header   header;
82e7776783SApple OSS Distributions };
83e7776783SApple OSS Distributions 
84e7776783SApple OSS Distributions static size_t
nvram_v3_var_container_size(const struct v3_var_header * header)855c2921b0SApple OSS Distributions nvram_v3_var_container_size(const struct v3_var_header *header)
86e7776783SApple OSS Distributions {
87e7776783SApple OSS Distributions 	return sizeof(struct nvram_v3_var_entry) + header->nameSize + header->dataSize;
88e7776783SApple OSS Distributions }
89e7776783SApple OSS Distributions 
90e7776783SApple OSS Distributions static size_t
variable_length(const struct v3_var_header * header)91e7776783SApple OSS Distributions variable_length(const struct v3_var_header *header)
92e7776783SApple OSS Distributions {
93e7776783SApple OSS Distributions 	return sizeof(struct v3_var_header) + header->nameSize + header->dataSize;
94e7776783SApple OSS Distributions }
95e7776783SApple OSS Distributions 
96e7776783SApple OSS Distributions static bool
valid_store_header(const struct v3_store_header * header)97e7776783SApple OSS Distributions valid_store_header(const struct v3_store_header *header)
98e7776783SApple OSS Distributions {
99e7776783SApple OSS Distributions 	return (header->name == VARIABLE_STORE_SIGNATURE) && (header->version == VARIABLE_STORE_VERSION);
100e7776783SApple OSS Distributions }
101e7776783SApple OSS Distributions 
102e7776783SApple OSS Distributions static bool
valid_variable_header(const struct v3_var_header * header,size_t buf_len)103e7776783SApple OSS Distributions valid_variable_header(const struct v3_var_header *header, size_t buf_len)
104e7776783SApple OSS Distributions {
105e7776783SApple OSS Distributions 	return (buf_len > sizeof(struct v3_var_header)) &&
106e7776783SApple OSS Distributions 	       (header->startId == VARIABLE_DATA) &&
107e7776783SApple OSS Distributions 	       (variable_length(header) <= buf_len);
108e7776783SApple OSS Distributions }
109e7776783SApple OSS Distributions 
110e7776783SApple OSS Distributions static uint32_t
find_active_var_in_image(const struct v3_var_header * var,const uint8_t * image,uint32_t offset,uint32_t len)1115c2921b0SApple OSS Distributions find_active_var_in_image(const struct v3_var_header *var, const uint8_t *image, uint32_t offset, uint32_t len)
112e7776783SApple OSS Distributions {
113e7776783SApple OSS Distributions 	const struct v3_var_header *store_var;
114e7776783SApple OSS Distributions 	uint32_t var_offset = 0;
115e7776783SApple OSS Distributions 
116e7776783SApple OSS Distributions 	while ((offset + sizeof(struct v3_var_header) < len)) {
117e7776783SApple OSS Distributions 		store_var = (const struct v3_var_header *)(image + offset);
118e7776783SApple OSS Distributions 
119e7776783SApple OSS Distributions 		if (valid_variable_header(store_var, len - offset)) {
120e7776783SApple OSS Distributions 			if ((store_var->state == VAR_ADDED) &&
1215c2921b0SApple OSS Distributions 			    (uuid_compare(var->guid, store_var->guid) == 0) &&
122e7776783SApple OSS Distributions 			    (var->nameSize == store_var->nameSize) &&
123e7776783SApple OSS Distributions 			    (memcmp(var->name_data_buf, store_var->name_data_buf, var->nameSize) == 0)) {
124e7776783SApple OSS Distributions 				var_offset = offset;
125e7776783SApple OSS Distributions 				break;
126e7776783SApple OSS Distributions 			}
127e7776783SApple OSS Distributions 		} else {
128e7776783SApple OSS Distributions 			break;
129e7776783SApple OSS Distributions 		}
130e7776783SApple OSS Distributions 
131e7776783SApple OSS Distributions 		offset += variable_length(store_var);
132e7776783SApple OSS Distributions 	}
133e7776783SApple OSS Distributions 
134e7776783SApple OSS Distributions 	return var_offset;
135e7776783SApple OSS Distributions }
136e7776783SApple OSS Distributions 
1375c2921b0SApple OSS Distributions static IOReturn
find_current_offset_in_image(const uint8_t * image,uint32_t len,uint32_t * newOffset)1385c2921b0SApple OSS Distributions find_current_offset_in_image(const uint8_t *image, uint32_t len, uint32_t *newOffset)
139e7776783SApple OSS Distributions {
140e7776783SApple OSS Distributions 	uint32_t offset = 0;
141e7776783SApple OSS Distributions 	uint32_t inner_offset = 0;
142e7776783SApple OSS Distributions 
143e7776783SApple OSS Distributions 	if (valid_store_header((const struct v3_store_header *)(image + offset))) {
144e7776783SApple OSS Distributions 		DEBUG_INFO("valid store header @ %#x\n", offset);
145e7776783SApple OSS Distributions 		offset += sizeof(struct v3_store_header);
146e7776783SApple OSS Distributions 	}
147e7776783SApple OSS Distributions 
148e7776783SApple OSS Distributions 	while (offset < len) {
1495c2921b0SApple OSS Distributions 		const struct v3_var_header *store_var = (const struct v3_var_header *)(image + offset);
1505c2921b0SApple OSS Distributions 		uuid_string_t uuidString;
1515c2921b0SApple OSS Distributions 
1525c2921b0SApple OSS Distributions 		if (valid_variable_header(store_var, len - offset)) {
1535c2921b0SApple OSS Distributions 			uuid_unparse(store_var->guid, uuidString);
1545c2921b0SApple OSS Distributions 			DEBUG_INFO("Valid var @ %#08x, state=%#02x, length=%#08zx, %s:%s\n", offset, store_var->state,
1555c2921b0SApple OSS Distributions 			    variable_length(store_var), uuidString, store_var->name_data_buf);
1565c2921b0SApple OSS Distributions 			offset += variable_length(store_var);
157e7776783SApple OSS Distributions 		} else {
158e7776783SApple OSS Distributions 			break;
159e7776783SApple OSS Distributions 		}
160e7776783SApple OSS Distributions 	}
161e7776783SApple OSS Distributions 
162e7776783SApple OSS Distributions 	while (offset < len) {
163e7776783SApple OSS Distributions 		if (image[offset] == 0xFF) {
164e7776783SApple OSS Distributions 			DEBUG_INFO("scanning for clear memory @ %#x\n", offset);
165e7776783SApple OSS Distributions 
166e7776783SApple OSS Distributions 			inner_offset = offset;
167e7776783SApple OSS Distributions 
168e7776783SApple OSS Distributions 			while ((inner_offset < len) && (image[inner_offset] == 0xFF)) {
169e7776783SApple OSS Distributions 				inner_offset++;
170e7776783SApple OSS Distributions 			}
171e7776783SApple OSS Distributions 
172e7776783SApple OSS Distributions 			if (inner_offset == len) {
173e7776783SApple OSS Distributions 				DEBUG_INFO("found start of clear mem @ %#x\n", offset);
1745c2921b0SApple OSS Distributions 				break;
175e7776783SApple OSS Distributions 			} else {
176e7776783SApple OSS Distributions 				DEBUG_ERROR("ERROR!!!!! found non-clear byte @ %#x\n", offset);
1775c2921b0SApple OSS Distributions 				return kIOReturnInvalid;
178e7776783SApple OSS Distributions 			}
179e7776783SApple OSS Distributions 		}
180e7776783SApple OSS Distributions 		offset++;
181e7776783SApple OSS Distributions 	}
182e7776783SApple OSS Distributions 
1835c2921b0SApple OSS Distributions 	*newOffset = offset;
1845c2921b0SApple OSS Distributions 
1855c2921b0SApple OSS Distributions 	return kIOReturnSuccess;
186e7776783SApple OSS Distributions }
187e7776783SApple OSS Distributions 
1885c2921b0SApple OSS Distributions class IONVRAMV3Handler : public IODTNVRAMFormatHandler, IOTypedOperatorsMixin<IONVRAMV3Handler>
189e7776783SApple OSS Distributions {
190e7776783SApple OSS Distributions private:
191e7776783SApple OSS Distributions 	IONVRAMController            *_nvramController;
192e7776783SApple OSS Distributions 	IODTNVRAM                    *_provider;
193e7776783SApple OSS Distributions 
194e7776783SApple OSS Distributions 	bool                         _newData;
1955c2921b0SApple OSS Distributions 	bool                         _resetData;
1965c2921b0SApple OSS Distributions 	bool                         _reload;
1975c2921b0SApple OSS Distributions 
1985c2921b0SApple OSS Distributions 	bool                         _rawController;
199e7776783SApple OSS Distributions 
200e7776783SApple OSS Distributions 	uint32_t                     _generation;
201e7776783SApple OSS Distributions 
202e7776783SApple OSS Distributions 	uint8_t                      *_nvramImage;
203e7776783SApple OSS Distributions 
204aca3beaaSApple OSS Distributions 	OSSharedPtr<OSDictionary>    &_varDict;
205e7776783SApple OSS Distributions 
206e7776783SApple OSS Distributions 	uint32_t                     _commonSize;
207e7776783SApple OSS Distributions 	uint32_t                     _systemSize;
208e7776783SApple OSS Distributions 
209e7776783SApple OSS Distributions 	uint32_t                     _commonUsed;
210e7776783SApple OSS Distributions 	uint32_t                     _systemUsed;
211e7776783SApple OSS Distributions 
212e7776783SApple OSS Distributions 	uint32_t                     _currentOffset;
213e7776783SApple OSS Distributions 
214e7776783SApple OSS Distributions 	OSSharedPtr<OSArray>         _varEntries;
215e7776783SApple OSS Distributions 
216e7776783SApple OSS Distributions 	IOReturn unserializeImage(const uint8_t *image, IOByteCount length);
217e7776783SApple OSS Distributions 	IOReturn reclaim(void);
2185c2921b0SApple OSS Distributions 	uint32_t findCurrentBank(void);
21994d3b452SApple OSS Distributions 	size_t   getAppendSize(void);
220e7776783SApple OSS Distributions 
221e7776783SApple OSS Distributions 	static bool convertObjectToProp(uint8_t *buffer, uint32_t *length, const char *propSymbol, OSObject *propObject);
222e7776783SApple OSS Distributions 	static bool convertPropToObject(const uint8_t *propName, uint32_t propNameLength, const uint8_t *propData, uint32_t propDataLength,
223e7776783SApple OSS Distributions 	    OSSharedPtr<const OSSymbol>& propSymbol, OSSharedPtr<OSObject>& propObject);
224e7776783SApple OSS Distributions 
2255c2921b0SApple OSS Distributions 	IOReturn reloadInternal(void);
226aca3beaaSApple OSS Distributions 	IOReturn setVariableInternal(const uuid_t varGuid, const char *variableName, OSObject *object);
2275c2921b0SApple OSS Distributions 
2285c2921b0SApple OSS Distributions 	void setEntryForRemove(struct nvram_v3_var_entry *v3Entry, bool system);
229aca3beaaSApple OSS Distributions 	void findExistingEntry(const uuid_t varGuid, const char *varName, struct nvram_v3_var_entry **existing, unsigned int *existingIndex);
2305c2921b0SApple OSS Distributions 	IOReturn syncRaw(void);
2315c2921b0SApple OSS Distributions 	IOReturn syncBlock(void);
232*8d741a5dSApple OSS Distributions 	IOReturn handleEphDM(void);
233e7776783SApple OSS Distributions 
234e7776783SApple OSS Distributions public:
235e7776783SApple OSS Distributions 	virtual
236e7776783SApple OSS Distributions 	~IONVRAMV3Handler() APPLE_KEXT_OVERRIDE;
237aca3beaaSApple OSS Distributions 	IONVRAMV3Handler(OSSharedPtr<OSDictionary> &varDict);
238e7776783SApple OSS Distributions 
239e7776783SApple OSS Distributions 	static bool isValidImage(const uint8_t *image, IOByteCount length);
240e7776783SApple OSS Distributions 
241e7776783SApple OSS Distributions 	static  IONVRAMV3Handler *init(IODTNVRAM *provider, const uint8_t *image, IOByteCount length,
242aca3beaaSApple OSS Distributions 	    OSSharedPtr<OSDictionary> &varDict);
243e7776783SApple OSS Distributions 
2445c2921b0SApple OSS Distributions 	virtual bool     getNVRAMProperties(void) APPLE_KEXT_OVERRIDE;
245aca3beaaSApple OSS Distributions 	virtual IOReturn unserializeVariables(void) APPLE_KEXT_OVERRIDE;
2465c2921b0SApple OSS Distributions 	virtual IOReturn setVariable(const uuid_t varGuid, const char *variableName, OSObject *object) APPLE_KEXT_OVERRIDE;
247e7776783SApple OSS Distributions 	virtual bool     setController(IONVRAMController *controller) APPLE_KEXT_OVERRIDE;
24894d3b452SApple OSS Distributions 	virtual IOReturn sync(void) APPLE_KEXT_OVERRIDE;
2495c2921b0SApple OSS Distributions 	virtual IOReturn flush(const uuid_t guid, IONVRAMOperation op) APPLE_KEXT_OVERRIDE;
2505c2921b0SApple OSS Distributions 	virtual void     reload(void) APPLE_KEXT_OVERRIDE;
251e7776783SApple OSS Distributions 	virtual uint32_t getGeneration(void) const APPLE_KEXT_OVERRIDE;
252e7776783SApple OSS Distributions 	virtual uint32_t getVersion(void) const APPLE_KEXT_OVERRIDE;
253e7776783SApple OSS Distributions 	virtual uint32_t getSystemUsed(void) const APPLE_KEXT_OVERRIDE;
254e7776783SApple OSS Distributions 	virtual uint32_t getCommonUsed(void) const APPLE_KEXT_OVERRIDE;
255aca3beaaSApple OSS Distributions 	virtual bool     getSystemPartitionActive(void) const APPLE_KEXT_OVERRIDE;
256e7776783SApple OSS Distributions };
257e7776783SApple OSS Distributions 
~IONVRAMV3Handler()258e7776783SApple OSS Distributions IONVRAMV3Handler::~IONVRAMV3Handler()
259e7776783SApple OSS Distributions {
260e7776783SApple OSS Distributions }
261e7776783SApple OSS Distributions 
IONVRAMV3Handler(OSSharedPtr<OSDictionary> & varDict)262aca3beaaSApple OSS Distributions IONVRAMV3Handler::IONVRAMV3Handler(OSSharedPtr<OSDictionary> &varDict) :
263aca3beaaSApple OSS Distributions 	_varDict(varDict)
264e7776783SApple OSS Distributions {
265e7776783SApple OSS Distributions }
266e7776783SApple OSS Distributions 
267e7776783SApple OSS Distributions bool
isValidImage(const uint8_t * image,IOByteCount length)268e7776783SApple OSS Distributions IONVRAMV3Handler::isValidImage(const uint8_t *image, IOByteCount length)
269e7776783SApple OSS Distributions {
270e7776783SApple OSS Distributions 	const struct v3_store_header *header = (const struct v3_store_header *)image;
271e7776783SApple OSS Distributions 
272e7776783SApple OSS Distributions 	if ((header == nullptr) || (length < sizeof(*header))) {
273e7776783SApple OSS Distributions 		return false;
274e7776783SApple OSS Distributions 	}
275e7776783SApple OSS Distributions 
276e7776783SApple OSS Distributions 	return valid_store_header(header);
277e7776783SApple OSS Distributions }
278e7776783SApple OSS Distributions 
279e7776783SApple OSS Distributions IONVRAMV3Handler*
init(IODTNVRAM * provider,const uint8_t * image,IOByteCount length,OSSharedPtr<OSDictionary> & varDict)280e7776783SApple OSS Distributions IONVRAMV3Handler::init(IODTNVRAM *provider, const uint8_t *image, IOByteCount length,
281aca3beaaSApple OSS Distributions     OSSharedPtr<OSDictionary> &varDict)
282e7776783SApple OSS Distributions {
2835c2921b0SApple OSS Distributions 	OSSharedPtr<IORegistryEntry> entry;
2845c2921b0SApple OSS Distributions 	OSSharedPtr<OSObject>        prop;
2855c2921b0SApple OSS Distributions 	bool                         propertiesOk;
2865c2921b0SApple OSS Distributions 
287aca3beaaSApple OSS Distributions 	IONVRAMV3Handler *handler = new IONVRAMV3Handler(varDict);
288e7776783SApple OSS Distributions 
289e7776783SApple OSS Distributions 	handler->_provider = provider;
290e7776783SApple OSS Distributions 
2915c2921b0SApple OSS Distributions 	propertiesOk = handler->getNVRAMProperties();
2925c2921b0SApple OSS Distributions 	require_action(propertiesOk, exit, DEBUG_ERROR("Unable to get NVRAM properties\n"));
2935c2921b0SApple OSS Distributions 
2945c2921b0SApple OSS Distributions 	require_action(length == handler->_bankSize, exit, DEBUG_ERROR("length %#llx != _bankSize %#x\n", length, handler->_bankSize));
2955c2921b0SApple OSS Distributions 
296e7776783SApple OSS Distributions 	if ((image != nullptr) && (length != 0)) {
297e7776783SApple OSS Distributions 		if (handler->unserializeImage(image, length) != kIOReturnSuccess) {
298e7776783SApple OSS Distributions 			DEBUG_ERROR("Unable to unserialize image, len=%#x\n", (unsigned int)length);
299e7776783SApple OSS Distributions 		}
300e7776783SApple OSS Distributions 	}
301e7776783SApple OSS Distributions 
302e7776783SApple OSS Distributions 	return handler;
3035c2921b0SApple OSS Distributions 
3045c2921b0SApple OSS Distributions exit:
3055c2921b0SApple OSS Distributions 	delete handler;
3065c2921b0SApple OSS Distributions 
3075c2921b0SApple OSS Distributions 	return nullptr;
3085c2921b0SApple OSS Distributions }
3095c2921b0SApple OSS Distributions 
3105c2921b0SApple OSS Distributions bool
getNVRAMProperties()3115c2921b0SApple OSS Distributions IONVRAMV3Handler::getNVRAMProperties()
3125c2921b0SApple OSS Distributions {
3135c2921b0SApple OSS Distributions 	bool                         ok    = false;
3145c2921b0SApple OSS Distributions 	const char                   *rawControllerKey = "nvram-raw";
3155c2921b0SApple OSS Distributions 	OSSharedPtr<IORegistryEntry> entry;
3165c2921b0SApple OSS Distributions 	OSSharedPtr<OSObject>        prop;
3175c2921b0SApple OSS Distributions 	OSData *                     data;
3185c2921b0SApple OSS Distributions 
3195c2921b0SApple OSS Distributions 	require_action(IODTNVRAMFormatHandler::getNVRAMProperties(), exit, DEBUG_ERROR("parent getNVRAMProperties failed\n"));
3205c2921b0SApple OSS Distributions 
3215c2921b0SApple OSS Distributions 	entry = IORegistryEntry::fromPath("/chosen", gIODTPlane);
3225c2921b0SApple OSS Distributions 	require_action(entry, exit, DEBUG_ERROR("Unable to find chosen node\n"));
3235c2921b0SApple OSS Distributions 
3245c2921b0SApple OSS Distributions 	prop = entry->copyProperty(rawControllerKey);
3255c2921b0SApple OSS Distributions 	require_action(prop != nullptr, exit, DEBUG_ERROR("No %s entry\n", rawControllerKey));
3265c2921b0SApple OSS Distributions 
3275c2921b0SApple OSS Distributions 	data = OSDynamicCast(OSData, prop.get());
3285c2921b0SApple OSS Distributions 	require(data != nullptr, exit);
3295c2921b0SApple OSS Distributions 
3305c2921b0SApple OSS Distributions 	_rawController = *((uint32_t*)data->getBytesNoCopy());
3315c2921b0SApple OSS Distributions 	DEBUG_INFO("_rawController = %d\n", _rawController);
3325c2921b0SApple OSS Distributions 
3335c2921b0SApple OSS Distributions 	ok = true;
3345c2921b0SApple OSS Distributions 
3355c2921b0SApple OSS Distributions exit:
3365c2921b0SApple OSS Distributions 	return ok;
3375c2921b0SApple OSS Distributions }
3385c2921b0SApple OSS Distributions 
3395c2921b0SApple OSS Distributions IOReturn
flush(const uuid_t guid,IONVRAMOperation op)3405c2921b0SApple OSS Distributions IONVRAMV3Handler::flush(const uuid_t guid, IONVRAMOperation op)
3415c2921b0SApple OSS Distributions {
3425c2921b0SApple OSS Distributions 	IOReturn ret = kIOReturnSuccess;
343aca3beaaSApple OSS Distributions 	bool     flushSystem;
344aca3beaaSApple OSS Distributions 	bool     flushCommon;
3455c2921b0SApple OSS Distributions 
346aca3beaaSApple OSS Distributions 	flushSystem = getSystemPartitionActive() && (uuid_compare(guid, gAppleSystemVariableGuid) == 0);
347aca3beaaSApple OSS Distributions 	flushCommon = uuid_compare(guid, gAppleNVRAMGuid) == 0;
348aca3beaaSApple OSS Distributions 
349aca3beaaSApple OSS Distributions 	DEBUG_INFO("flushSystem=%d, flushCommon=%d\n", flushSystem, flushCommon);
350aca3beaaSApple OSS Distributions 
351aca3beaaSApple OSS Distributions 	if (flushSystem || flushCommon) {
352aca3beaaSApple OSS Distributions 		const OSSymbol                    *canonicalKey;
353aca3beaaSApple OSS Distributions 		OSSharedPtr<OSDictionary>         dictCopy;
3545c2921b0SApple OSS Distributions 		OSSharedPtr<OSCollectionIterator> iter;
3555c2921b0SApple OSS Distributions 		uuid_string_t                     uuidString;
3565c2921b0SApple OSS Distributions 
357aca3beaaSApple OSS Distributions 		dictCopy = OSDictionary::withDictionary(_varDict.get());
358aca3beaaSApple OSS Distributions 		iter = OSCollectionIterator::withCollection(dictCopy.get());
359aca3beaaSApple OSS Distributions 		require_action(dictCopy && iter, exit, ret = kIOReturnNoMemory);
3605c2921b0SApple OSS Distributions 
361aca3beaaSApple OSS Distributions 		while ((canonicalKey = OSDynamicCast(OSSymbol, iter->getNextObject()))) {
362aca3beaaSApple OSS Distributions 			const char *varName;
363aca3beaaSApple OSS Distributions 			uuid_t     varGuid;
364aca3beaaSApple OSS Distributions 			bool       clear;
3655c2921b0SApple OSS Distributions 
366aca3beaaSApple OSS Distributions 			parseVariableName(canonicalKey->getCStringNoCopy(), &varGuid, &varName);
367aca3beaaSApple OSS Distributions 
368aca3beaaSApple OSS Distributions 			uuid_unparse(varGuid, uuidString);
369aca3beaaSApple OSS Distributions 
370aca3beaaSApple OSS Distributions 			clear = ((flushSystem && (uuid_compare(varGuid, gAppleSystemVariableGuid) == 0)) ||
371aca3beaaSApple OSS Distributions 			    (flushCommon && (uuid_compare(varGuid, gAppleSystemVariableGuid) != 0))) &&
372aca3beaaSApple OSS Distributions 			    verifyPermission(op, varGuid, varName, getSystemPartitionActive());
373aca3beaaSApple OSS Distributions 
374aca3beaaSApple OSS Distributions 			if (clear) {
375aca3beaaSApple OSS Distributions 				DEBUG_INFO("Clearing entry for %s:%s\n", uuidString, varName);
376aca3beaaSApple OSS Distributions 				setVariableInternal(varGuid, varName, nullptr);
3775c2921b0SApple OSS Distributions 			} else {
378aca3beaaSApple OSS Distributions 				DEBUG_INFO("Keeping entry for %s:%s\n", uuidString, varName);
3795c2921b0SApple OSS Distributions 			}
3805c2921b0SApple OSS Distributions 		}
3815c2921b0SApple OSS Distributions 
3825c2921b0SApple OSS Distributions 		_newData = true;
3835c2921b0SApple OSS Distributions 	}
3845c2921b0SApple OSS Distributions 
3855c2921b0SApple OSS Distributions 	DEBUG_INFO("_commonUsed %#x, _systemUsed %#x\n", _commonUsed, _systemUsed);
3865c2921b0SApple OSS Distributions 
3875c2921b0SApple OSS Distributions exit:
3885c2921b0SApple OSS Distributions 	return ret;
3895c2921b0SApple OSS Distributions }
3905c2921b0SApple OSS Distributions 
3915c2921b0SApple OSS Distributions IOReturn
reloadInternal(void)3925c2921b0SApple OSS Distributions IONVRAMV3Handler::reloadInternal(void)
3935c2921b0SApple OSS Distributions {
3945c2921b0SApple OSS Distributions 	IOReturn                     ret;
3955c2921b0SApple OSS Distributions 	uint32_t                     controllerBank;
3965c2921b0SApple OSS Distributions 	uint8_t                      *controllerImage;
3975c2921b0SApple OSS Distributions 	struct nvram_v3_var_entry    *v3Entry;
3985c2921b0SApple OSS Distributions 	const struct v3_store_header *storeHeader;
3995c2921b0SApple OSS Distributions 	const struct v3_var_header   *storeVar;
4005c2921b0SApple OSS Distributions 	OSData                       *entryContainer;
4015c2921b0SApple OSS Distributions 
4025c2921b0SApple OSS Distributions 	controllerBank = findCurrentBank();
4035c2921b0SApple OSS Distributions 
4045c2921b0SApple OSS Distributions 	if (_currentBank != controllerBank) {
4055c2921b0SApple OSS Distributions 		DEBUG_ERROR("_currentBank %#x != controllerBank %#x", _currentBank, controllerBank);
4065c2921b0SApple OSS Distributions 	}
4075c2921b0SApple OSS Distributions 
4085c2921b0SApple OSS Distributions 	_currentBank = controllerBank;
4095c2921b0SApple OSS Distributions 
4105c2921b0SApple OSS Distributions 	controllerImage = (uint8_t *)IOMallocData(_bankSize);
4115c2921b0SApple OSS Distributions 
4125c2921b0SApple OSS Distributions 	_nvramController->select(_currentBank);
4135c2921b0SApple OSS Distributions 	_nvramController->read(0, controllerImage, _bankSize);
4145c2921b0SApple OSS Distributions 
4155c2921b0SApple OSS Distributions 	require_action(isValidImage(controllerImage, _bankSize), exit,
4165c2921b0SApple OSS Distributions 	    (ret = kIOReturnInvalid, DEBUG_ERROR("Invalid image at bank %d\n", _currentBank)));
4175c2921b0SApple OSS Distributions 
4185c2921b0SApple OSS Distributions 	DEBUG_INFO("valid image found\n");
4195c2921b0SApple OSS Distributions 
4205c2921b0SApple OSS Distributions 	storeHeader = (const struct v3_store_header *)controllerImage;
4215c2921b0SApple OSS Distributions 
4225c2921b0SApple OSS Distributions 	_generation = storeHeader->generation;
4235c2921b0SApple OSS Distributions 
4245c2921b0SApple OSS Distributions 	// We must sync any existing variables offset on the controller image with our internal representation
4255c2921b0SApple OSS Distributions 	// If we find an existing entry and the data is still the same we record the existing offset and mark it
4265c2921b0SApple OSS Distributions 	// as VAR_NEW_STATE_NONE meaning no action needed
4275c2921b0SApple OSS Distributions 	// Otherwise if the data is different or it is not found on the controller image we mark it as VAR_NEW_STATE_APPEND
4285c2921b0SApple OSS Distributions 	// which will have us invalidate the existing entry if there is one and append it on the next save
4295c2921b0SApple OSS Distributions 	for (unsigned int i = 0; i < _varEntries->getCount(); i++) {
4305c2921b0SApple OSS Distributions 		uint32_t offset = sizeof(struct v3_store_header);
4315c2921b0SApple OSS Distributions 		uint32_t latestOffset;
4325c2921b0SApple OSS Distributions 		uint32_t prevOffset = 0;
4335c2921b0SApple OSS Distributions 
4345c2921b0SApple OSS Distributions 		entryContainer = (OSDynamicCast(OSData, _varEntries->getObject(i)));
4355c2921b0SApple OSS Distributions 		v3Entry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
4365c2921b0SApple OSS Distributions 
4375c2921b0SApple OSS Distributions 		DEBUG_INFO("Looking for %s\n", v3Entry->header.name_data_buf);
4385c2921b0SApple OSS Distributions 		while ((latestOffset = find_active_var_in_image(&v3Entry->header, controllerImage, offset, _bankSize))) {
4395c2921b0SApple OSS Distributions 			DEBUG_INFO("Found offset for %s @ %#08x\n", v3Entry->header.name_data_buf, latestOffset);
4405c2921b0SApple OSS Distributions 			if (prevOffset) {
4415c2921b0SApple OSS Distributions 				DEBUG_INFO("Marking prev offset for %s at %#08x invalid\n", v3Entry->header.name_data_buf, offset);
4425c2921b0SApple OSS Distributions 				// Invalidate any previous duplicate entries in the store
4435c2921b0SApple OSS Distributions 				struct v3_var_header *prevVarHeader = (struct v3_var_header *)(controllerImage + prevOffset);
4445c2921b0SApple OSS Distributions 				uint8_t state = prevVarHeader->state & VAR_DELETED & VAR_IN_DELETED_TRANSITION;
4455c2921b0SApple OSS Distributions 
4465c2921b0SApple OSS Distributions 				ret = _nvramController->write(prevOffset + offsetof(struct v3_var_header, state), &state, sizeof(state));
4475c2921b0SApple OSS Distributions 				require_noerr_action(ret, exit, DEBUG_ERROR("existing state w fail, ret=%#x\n", ret));
4485c2921b0SApple OSS Distributions 			}
4495c2921b0SApple OSS Distributions 
4505c2921b0SApple OSS Distributions 			prevOffset = latestOffset;
4515c2921b0SApple OSS Distributions 			offset += latestOffset;
4525c2921b0SApple OSS Distributions 		}
4535c2921b0SApple OSS Distributions 
4545c2921b0SApple OSS Distributions 		v3Entry->existing_offset = latestOffset ? latestOffset : prevOffset;
4555c2921b0SApple OSS Distributions 		DEBUG_INFO("Existing offset for %s at %#08zx\n", v3Entry->header.name_data_buf, v3Entry->existing_offset);
4565c2921b0SApple OSS Distributions 
4575c2921b0SApple OSS Distributions 		if (v3Entry->existing_offset == 0) {
4585c2921b0SApple OSS Distributions 			DEBUG_ERROR("%s is not in the NOR image\n", v3Entry->header.name_data_buf);
4595c2921b0SApple OSS Distributions 			if (v3Entry->new_state != VAR_NEW_STATE_REMOVE) {
4605c2921b0SApple OSS Distributions 				DEBUG_INFO("%s marked for append\n", v3Entry->header.name_data_buf);
4615c2921b0SApple OSS Distributions 				// Doesn't exist in the store, just append it on next sync
4625c2921b0SApple OSS Distributions 				v3Entry->new_state = VAR_NEW_STATE_APPEND;
4635c2921b0SApple OSS Distributions 			}
4645c2921b0SApple OSS Distributions 		} else {
4655c2921b0SApple OSS Distributions 			DEBUG_INFO("Found offset for %s @ %#zx\n", v3Entry->header.name_data_buf, v3Entry->existing_offset);
4665c2921b0SApple OSS Distributions 			storeVar = (const struct v3_var_header *)&controllerImage[v3Entry->existing_offset];
4675c2921b0SApple OSS Distributions 
4685c2921b0SApple OSS Distributions 			if (v3Entry->new_state != VAR_NEW_STATE_REMOVE) {
4695c2921b0SApple OSS Distributions 				// Verify that the existing data matches the store data
4705c2921b0SApple OSS Distributions 				if ((variable_length(&v3Entry->header) == variable_length(storeVar)) &&
4715c2921b0SApple OSS Distributions 				    (memcmp(v3Entry->header.name_data_buf, storeVar->name_data_buf, storeVar->nameSize + storeVar->dataSize) == 0)) {
4725c2921b0SApple OSS Distributions 					DEBUG_INFO("Store var data for %s matches, marking new state none\n", v3Entry->header.name_data_buf);
4735c2921b0SApple OSS Distributions 					v3Entry->new_state = VAR_NEW_STATE_NONE;
4745c2921b0SApple OSS Distributions 				} else {
4755c2921b0SApple OSS Distributions 					DEBUG_INFO("Store var data for %s differs, marking new state append\n", v3Entry->header.name_data_buf);
4765c2921b0SApple OSS Distributions 					v3Entry->new_state = VAR_NEW_STATE_APPEND;
4775c2921b0SApple OSS Distributions 				}
4785c2921b0SApple OSS Distributions 			} else {
4795c2921b0SApple OSS Distributions 				// Store has entry but it has been removed from our collection, keep it marked for delete but with updated
4805c2921b0SApple OSS Distributions 				// existing_offset for coherence
4815c2921b0SApple OSS Distributions 				DEBUG_INFO("Removing entry at %#08zx with next sync\n", v3Entry->existing_offset);
4825c2921b0SApple OSS Distributions 			}
4835c2921b0SApple OSS Distributions 		}
4845c2921b0SApple OSS Distributions 	}
4855c2921b0SApple OSS Distributions 
4865c2921b0SApple OSS Distributions 	ret = find_current_offset_in_image(controllerImage, _bankSize, &_currentOffset);
4875c2921b0SApple OSS Distributions 	if (ret != kIOReturnSuccess) {
4885c2921b0SApple OSS Distributions 		DEBUG_ERROR("Unidentified bytes in image, reclaiming\n");
4895c2921b0SApple OSS Distributions 		ret = reclaim();
4905c2921b0SApple OSS Distributions 		require_noerr_action(ret, exit, DEBUG_ERROR("Reclaim byte recovery failed, invalid controller state!!! ret=%#x\n", ret));
4915c2921b0SApple OSS Distributions 	}
4925c2921b0SApple OSS Distributions 	DEBUG_INFO("New _currentOffset=%#x\n", _currentOffset);
4935c2921b0SApple OSS Distributions 
4945c2921b0SApple OSS Distributions exit:
4955c2921b0SApple OSS Distributions 	IOFreeData(controllerImage, _bankSize);
4965c2921b0SApple OSS Distributions 	return ret;
4975c2921b0SApple OSS Distributions }
4985c2921b0SApple OSS Distributions 
4995c2921b0SApple OSS Distributions void
reload(void)5005c2921b0SApple OSS Distributions IONVRAMV3Handler::reload(void)
5015c2921b0SApple OSS Distributions {
5025c2921b0SApple OSS Distributions 	_reload = true;
5035c2921b0SApple OSS Distributions 
5045c2921b0SApple OSS Distributions 	DEBUG_INFO("reload marked\n");
5055c2921b0SApple OSS Distributions }
5065c2921b0SApple OSS Distributions 
5075c2921b0SApple OSS Distributions void
setEntryForRemove(struct nvram_v3_var_entry * v3Entry,bool system)5085c2921b0SApple OSS Distributions IONVRAMV3Handler::setEntryForRemove(struct nvram_v3_var_entry *v3Entry, bool system)
5095c2921b0SApple OSS Distributions {
510aca3beaaSApple OSS Distributions 	OSSharedPtr<const OSSymbol> canonicalKey;
5115c2921b0SApple OSS Distributions 	const char                  *variableName;
5125c2921b0SApple OSS Distributions 	uint32_t                    variableSize;
5135c2921b0SApple OSS Distributions 
5145c2921b0SApple OSS Distributions 	require_action(v3Entry != nullptr, exit, DEBUG_INFO("remove with no entry\n"));
5155c2921b0SApple OSS Distributions 
5165c2921b0SApple OSS Distributions 	variableName = (const char *)v3Entry->header.name_data_buf;
5175c2921b0SApple OSS Distributions 	variableSize = (uint32_t)variable_length(&v3Entry->header);
518aca3beaaSApple OSS Distributions 	canonicalKey = keyWithGuidAndCString(v3Entry->header.guid, variableName);
5195c2921b0SApple OSS Distributions 
5205c2921b0SApple OSS Distributions 	if (v3Entry->new_state == VAR_NEW_STATE_REMOVE) {
5215c2921b0SApple OSS Distributions 		DEBUG_INFO("entry %s already marked for remove\n", variableName);
5225c2921b0SApple OSS Distributions 	} else {
5235c2921b0SApple OSS Distributions 		DEBUG_INFO("marking entry %s for remove\n", variableName);
5245c2921b0SApple OSS Distributions 
5255c2921b0SApple OSS Distributions 		v3Entry->new_state = VAR_NEW_STATE_REMOVE;
5265c2921b0SApple OSS Distributions 
527aca3beaaSApple OSS Distributions 		_provider->_varDict->removeObject(canonicalKey.get());
5285c2921b0SApple OSS Distributions 
529aca3beaaSApple OSS Distributions 		if (system) {
5305c2921b0SApple OSS Distributions 			if (_systemUsed < variableSize) {
5315c2921b0SApple OSS Distributions 				panic("Invalid _systemUsed size\n");
5325c2921b0SApple OSS Distributions 			}
5335c2921b0SApple OSS Distributions 			_systemUsed -= variableSize;
5345c2921b0SApple OSS Distributions 		} else {
5355c2921b0SApple OSS Distributions 			if (_commonUsed < variableSize) {
5365c2921b0SApple OSS Distributions 				panic("Invalid _commonUsed size\n");
5375c2921b0SApple OSS Distributions 			}
5385c2921b0SApple OSS Distributions 			_commonUsed -= variableSize;
5395c2921b0SApple OSS Distributions 		}
5405c2921b0SApple OSS Distributions 
5415c2921b0SApple OSS Distributions 		if (_provider->_diags) {
5425c2921b0SApple OSS Distributions 			_provider->_diags->logVariable(getPartitionTypeForGUID(v3Entry->header.guid),
5435c2921b0SApple OSS Distributions 			    kIONVRAMOperationDelete,
5445c2921b0SApple OSS Distributions 			    variableName,
5455c2921b0SApple OSS Distributions 			    nullptr);
5465c2921b0SApple OSS Distributions 		}
5475c2921b0SApple OSS Distributions 	}
5485c2921b0SApple OSS Distributions 
5495c2921b0SApple OSS Distributions exit:
5505c2921b0SApple OSS Distributions 	return;
5515c2921b0SApple OSS Distributions }
5525c2921b0SApple OSS Distributions 
5535c2921b0SApple OSS Distributions void
findExistingEntry(const uuid_t varGuid,const char * varName,struct nvram_v3_var_entry ** existing,unsigned int * existingIndex)554aca3beaaSApple OSS Distributions IONVRAMV3Handler::findExistingEntry(const uuid_t varGuid, const char *varName, struct nvram_v3_var_entry **existing, unsigned int *existingIndex)
5555c2921b0SApple OSS Distributions {
5565c2921b0SApple OSS Distributions 	struct nvram_v3_var_entry *v3Entry = nullptr;
5575c2921b0SApple OSS Distributions 	OSData                    *entryContainer = nullptr;
5585c2921b0SApple OSS Distributions 	unsigned int              index = 0;
5595c2921b0SApple OSS Distributions 	uint32_t                  nameLen = (uint32_t)strlen(varName) + 1;
5605c2921b0SApple OSS Distributions 
5615c2921b0SApple OSS Distributions 	for (index = 0; index < _varEntries->getCount(); index++) {
5625c2921b0SApple OSS Distributions 		entryContainer = (OSDynamicCast(OSData, _varEntries->getObject(index)));
5635c2921b0SApple OSS Distributions 		v3Entry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
5645c2921b0SApple OSS Distributions 
5655c2921b0SApple OSS Distributions 		if ((v3Entry->header.nameSize == nameLen) &&
5665c2921b0SApple OSS Distributions 		    (memcmp(v3Entry->header.name_data_buf, varName, nameLen) == 0)) {
5675c2921b0SApple OSS Distributions 			if (varGuid) {
568aca3beaaSApple OSS Distributions 				if (uuid_compare(varGuid, v3Entry->header.guid) == 0) {
5695c2921b0SApple OSS Distributions 					uuid_string_t uuidString;
570aca3beaaSApple OSS Distributions 					uuid_unparse(varGuid, uuidString);
5715c2921b0SApple OSS Distributions 					DEBUG_INFO("found existing entry for %s:%s, e_off=%#lx, len=%#lx, new_state=%#x\n", uuidString, varName,
5725c2921b0SApple OSS Distributions 					    v3Entry->existing_offset, variable_length(&v3Entry->header), v3Entry->new_state);
5735c2921b0SApple OSS Distributions 					break;
5745c2921b0SApple OSS Distributions 				}
5755c2921b0SApple OSS Distributions 			} else {
5765c2921b0SApple OSS Distributions 				DEBUG_INFO("found existing entry for %s, e_off=%#lx, len=%#lx\n", varName, v3Entry->existing_offset, variable_length(&v3Entry->header));
5775c2921b0SApple OSS Distributions 				break;
5785c2921b0SApple OSS Distributions 			}
5795c2921b0SApple OSS Distributions 		}
5805c2921b0SApple OSS Distributions 
5815c2921b0SApple OSS Distributions 		v3Entry = nullptr;
5825c2921b0SApple OSS Distributions 	}
5835c2921b0SApple OSS Distributions 
5845c2921b0SApple OSS Distributions 	if (v3Entry != nullptr) {
5855c2921b0SApple OSS Distributions 		if (existing) {
5865c2921b0SApple OSS Distributions 			*existing = v3Entry;
5875c2921b0SApple OSS Distributions 		}
5885c2921b0SApple OSS Distributions 
5895c2921b0SApple OSS Distributions 		if (existingIndex) {
5905c2921b0SApple OSS Distributions 			*existingIndex = index;
5915c2921b0SApple OSS Distributions 		}
5925c2921b0SApple OSS Distributions 	}
593e7776783SApple OSS Distributions }
594e7776783SApple OSS Distributions 
595e7776783SApple OSS Distributions IOReturn
unserializeImage(const uint8_t * image,IOByteCount length)596e7776783SApple OSS Distributions IONVRAMV3Handler::unserializeImage(const uint8_t *image, IOByteCount length)
597e7776783SApple OSS Distributions {
598aca3beaaSApple OSS Distributions 	IOReturn                     ret = kIOReturnInvalid;
599e7776783SApple OSS Distributions 	const struct v3_store_header *storeHeader;
600e7776783SApple OSS Distributions 
601e7776783SApple OSS Distributions 	require(isValidImage(image, length), exit);
602e7776783SApple OSS Distributions 
603e7776783SApple OSS Distributions 	storeHeader = (const struct v3_store_header *)image;
604e7776783SApple OSS Distributions 	require_action(storeHeader->size == (uint32_t)length, exit,
605e7776783SApple OSS Distributions 	    DEBUG_ERROR("Image size %#x != header size %#x\n", (unsigned int)length, storeHeader->size));
606e7776783SApple OSS Distributions 
607e7776783SApple OSS Distributions 	_generation = storeHeader->generation;
608e7776783SApple OSS Distributions 	_systemSize = storeHeader->system_size;
609e7776783SApple OSS Distributions 	_commonSize = storeHeader->common_size - sizeof(struct v3_store_header);
610e7776783SApple OSS Distributions 
611e7776783SApple OSS Distributions 	_systemUsed = 0;
612e7776783SApple OSS Distributions 	_commonUsed = 0;
613e7776783SApple OSS Distributions 
614e7776783SApple OSS Distributions 	if (_nvramImage) {
6155c2921b0SApple OSS Distributions 		IOFreeData(_nvramImage, _bankSize);
616e7776783SApple OSS Distributions 	}
617e7776783SApple OSS Distributions 
618e7776783SApple OSS Distributions 	_varEntries.reset();
619e7776783SApple OSS Distributions 	_varEntries = OSArray::withCapacity(40);
620e7776783SApple OSS Distributions 
621e7776783SApple OSS Distributions 	_nvramImage = IONewData(uint8_t, length);
6225c2921b0SApple OSS Distributions 	_bankSize = (uint32_t)length;
6235c2921b0SApple OSS Distributions 	bcopy(image, _nvramImage, _bankSize);
624e7776783SApple OSS Distributions 
625aca3beaaSApple OSS Distributions 	ret = kIOReturnSuccess;
626aca3beaaSApple OSS Distributions 
627aca3beaaSApple OSS Distributions exit:
628aca3beaaSApple OSS Distributions 	return ret;
629e7776783SApple OSS Distributions }
630e7776783SApple OSS Distributions 
631*8d741a5dSApple OSS Distributions typedef struct {
632*8d741a5dSApple OSS Distributions 	const char            *name;
633*8d741a5dSApple OSS Distributions 	OSSharedPtr<OSObject> value;
634*8d741a5dSApple OSS Distributions } ephDMAllowListEntry;
635*8d741a5dSApple OSS Distributions 
636*8d741a5dSApple OSS Distributions static
637*8d741a5dSApple OSS Distributions ephDMAllowListEntry ephDMEntries[] = {
638*8d741a5dSApple OSS Distributions 	// Mobile Obliteration clears the following variables after it runs
639*8d741a5dSApple OSS Distributions 	{ .name = "oblit-begins" },
640*8d741a5dSApple OSS Distributions 	{ .name = "orig-oblit" },
641*8d741a5dSApple OSS Distributions 	{ .name = "oblit-failure" },
642*8d741a5dSApple OSS Distributions 	{ .name = "oblit-inprogress" },
643*8d741a5dSApple OSS Distributions 	{ .name = "obliteration" },
644*8d741a5dSApple OSS Distributions 	// darwin-init is used for configuring internal builds
645*8d741a5dSApple OSS Distributions 	{ .name = "darwin-init" }
646*8d741a5dSApple OSS Distributions };
647*8d741a5dSApple OSS Distributions 
648*8d741a5dSApple OSS Distributions IOReturn
handleEphDM(void)649*8d741a5dSApple OSS Distributions IONVRAMV3Handler::handleEphDM(void)
650*8d741a5dSApple OSS Distributions {
651*8d741a5dSApple OSS Distributions 	OSSharedPtr<IORegistryEntry> entry;
652*8d741a5dSApple OSS Distributions 	OSData*                      data;
653*8d741a5dSApple OSS Distributions 	OSSharedPtr<OSObject>        prop;
654*8d741a5dSApple OSS Distributions 	uint32_t                     ephDM = 0;
655*8d741a5dSApple OSS Distributions 	IOReturn                     ret = kIOReturnSuccess;
656*8d741a5dSApple OSS Distributions 	OSSharedPtr<const OSSymbol>  canonicalKey;
657*8d741a5dSApple OSS Distributions 	uint32_t                     skip = 0;
658*8d741a5dSApple OSS Distributions 
659*8d741a5dSApple OSS Distributions 	// For ephemeral data mode, NVRAM needs to be cleared on every boot
660*8d741a5dSApple OSS Distributions 	// For system region supported targets, iBoot clears the system region
661*8d741a5dSApple OSS Distributions 	// For other targets, iBoot clears all the persistent variables
662*8d741a5dSApple OSS Distributions 	// So xnu only needs to clear the common region
663*8d741a5dSApple OSS Distributions 	entry = IORegistryEntry::fromPath("/product", gIODTPlane);
664*8d741a5dSApple OSS Distributions 	if (entry) {
665*8d741a5dSApple OSS Distributions 		prop = entry->copyProperty("ephemeral-data-mode");
666*8d741a5dSApple OSS Distributions 		if (prop) {
667*8d741a5dSApple OSS Distributions 			data = OSDynamicCast(OSData, prop.get());
668*8d741a5dSApple OSS Distributions 			if (data) {
669*8d741a5dSApple OSS Distributions 				ephDM = *((uint32_t *)data->getBytesNoCopy());
670*8d741a5dSApple OSS Distributions 			}
671*8d741a5dSApple OSS Distributions 		}
672*8d741a5dSApple OSS Distributions 	}
673*8d741a5dSApple OSS Distributions 
674*8d741a5dSApple OSS Distributions 	require_action(ephDM != 0, exit, DEBUG_ALWAYS("ephemeral-data-mode not supported\n"));
675*8d741a5dSApple OSS Distributions 	require_action(_systemSize != 0, exit, DEBUG_ALWAYS("No system region, no need to clear\n"));
676*8d741a5dSApple OSS Distributions 
677*8d741a5dSApple OSS Distributions 	if (PE_parse_boot_argn("epdm-skip-nvram", &skip, sizeof(skip))) {
678*8d741a5dSApple OSS Distributions 		require_action(!(gInternalBuild && (skip == 1)), exit, DEBUG_ALWAYS("Internal build + epdm-skip-nvram set to true, skip nvram clearing\n"));
679*8d741a5dSApple OSS Distributions 	}
680*8d741a5dSApple OSS Distributions 
681*8d741a5dSApple OSS Distributions 	// Go through the allowlist and stash the values
682*8d741a5dSApple OSS Distributions 	for (uint32_t entry = 0; entry < ARRAY_SIZE(ephDMEntries); entry++) {
683*8d741a5dSApple OSS Distributions 		canonicalKey = keyWithGuidAndCString(gAppleNVRAMGuid, ephDMEntries[entry].name);
684*8d741a5dSApple OSS Distributions 		ephDMEntries[entry].value.reset(OSDynamicCast(OSData, _varDict->getObject(canonicalKey.get())), OSRetain);
685*8d741a5dSApple OSS Distributions 	}
686*8d741a5dSApple OSS Distributions 
687*8d741a5dSApple OSS Distributions 	DEBUG_ALWAYS("Obliterating common region\n");
688*8d741a5dSApple OSS Distributions 	ret = flush(gAppleNVRAMGuid, kIONVRAMOperationObliterate);
689*8d741a5dSApple OSS Distributions 	require_noerr_action(ret, exit, DEBUG_ERROR("Flushing common region failed, ret=%#08x\n", ret));
690*8d741a5dSApple OSS Distributions 
691*8d741a5dSApple OSS Distributions 	// Now write the allowlist variables back
692*8d741a5dSApple OSS Distributions 	for (uint32_t entry = 0; entry < ARRAY_SIZE(ephDMEntries); entry++) {
693*8d741a5dSApple OSS Distributions 		if (ephDMEntries[entry].value.get() == nullptr) {
694*8d741a5dSApple OSS Distributions 			continue;
695*8d741a5dSApple OSS Distributions 		}
696*8d741a5dSApple OSS Distributions 		ret = setVariableInternal(gAppleNVRAMGuid, ephDMEntries[entry].name, ephDMEntries[entry].value.get());
697*8d741a5dSApple OSS Distributions 		require_noerr_action(ret, exit, DEBUG_ERROR("Setting allowlist variable %s failed, ret=%#08x\n", ephDMEntries[entry].name, ret));
698*8d741a5dSApple OSS Distributions 	}
699*8d741a5dSApple OSS Distributions 
700*8d741a5dSApple OSS Distributions exit:
701*8d741a5dSApple OSS Distributions 	return ret;
702*8d741a5dSApple OSS Distributions }
703*8d741a5dSApple OSS Distributions 
704aca3beaaSApple OSS Distributions IOReturn
unserializeVariables(void)705aca3beaaSApple OSS Distributions IONVRAMV3Handler::unserializeVariables(void)
706aca3beaaSApple OSS Distributions {
707aca3beaaSApple OSS Distributions 	IOReturn                     ret = kIOReturnSuccess;
708aca3beaaSApple OSS Distributions 	OSSharedPtr<const OSSymbol>  propSymbol;
709aca3beaaSApple OSS Distributions 	OSSharedPtr<OSObject>        propObject;
710aca3beaaSApple OSS Distributions 	OSSharedPtr<OSData>          entryContainer;
711aca3beaaSApple OSS Distributions 	struct nvram_v3_var_entry    *v3Entry;
712aca3beaaSApple OSS Distributions 	const struct v3_var_header   *header;
713aca3beaaSApple OSS Distributions 	size_t                       offset = sizeof(struct v3_store_header);
714aca3beaaSApple OSS Distributions 	uint32_t                     crc;
715aca3beaaSApple OSS Distributions 	unsigned int                 i;
716aca3beaaSApple OSS Distributions 	bool                         system;
717aca3beaaSApple OSS Distributions 	uuid_string_t                uuidString;
718aca3beaaSApple OSS Distributions 	size_t                       existingSize;
719aca3beaaSApple OSS Distributions 
720aca3beaaSApple OSS Distributions 	if (_systemSize || _commonSize) {
721aca3beaaSApple OSS Distributions 		_varDict = OSDictionary::withCapacity(1);
722e7776783SApple OSS Distributions 	}
723e7776783SApple OSS Distributions 
724aca3beaaSApple OSS Distributions 	while ((offset + sizeof(struct v3_var_header)) < _bankSize) {
7255c2921b0SApple OSS Distributions 		struct nvram_v3_var_entry *existingEntry = nullptr;
7265c2921b0SApple OSS Distributions 		unsigned int              existingIndex = 0;
7275c2921b0SApple OSS Distributions 
728aca3beaaSApple OSS Distributions 		header = (const struct v3_var_header *)(_nvramImage + offset);
729e7776783SApple OSS Distributions 
730e7776783SApple OSS Distributions 		for (i = 0; i < sizeof(struct v3_var_header); i++) {
731aca3beaaSApple OSS Distributions 			if ((_nvramImage[offset + i] != 0) && (_nvramImage[offset + i] != 0xFF)) {
732e7776783SApple OSS Distributions 				break;
733e7776783SApple OSS Distributions 			}
734e7776783SApple OSS Distributions 		}
735e7776783SApple OSS Distributions 
736e7776783SApple OSS Distributions 		if (i == sizeof(struct v3_var_header)) {
737e7776783SApple OSS Distributions 			DEBUG_INFO("No more variables after offset %#lx\n", offset);
738e7776783SApple OSS Distributions 			break;
739e7776783SApple OSS Distributions 		}
740e7776783SApple OSS Distributions 
741aca3beaaSApple OSS Distributions 		if (!valid_variable_header(header, _bankSize - offset)) {
742e7776783SApple OSS Distributions 			DEBUG_ERROR("invalid header @ %#lx\n", offset);
743e7776783SApple OSS Distributions 			offset += sizeof(struct v3_var_header);
744e7776783SApple OSS Distributions 			continue;
745e7776783SApple OSS Distributions 		}
746e7776783SApple OSS Distributions 
7475c2921b0SApple OSS Distributions 		uuid_unparse(header->guid, uuidString);
7485c2921b0SApple OSS Distributions 		DEBUG_INFO("Valid var @ %#08zx, state=%#02x, length=%#08zx, %s:%s\n", offset, header->state,
7495c2921b0SApple OSS Distributions 		    variable_length(header), uuidString, header->name_data_buf);
7505c2921b0SApple OSS Distributions 
751e7776783SApple OSS Distributions 		if (header->state != VAR_ADDED) {
752e7776783SApple OSS Distributions 			goto skip;
753e7776783SApple OSS Distributions 		}
754e7776783SApple OSS Distributions 
755e7776783SApple OSS Distributions 		crc = crc32(0, header->name_data_buf + header->nameSize, header->dataSize);
756e7776783SApple OSS Distributions 
757e7776783SApple OSS Distributions 		if (crc != header->crc) {
758e7776783SApple OSS Distributions 			DEBUG_ERROR("invalid crc @ %#lx, calculated=%#x, read=%#x\n", offset, crc, header->crc);
759e7776783SApple OSS Distributions 			goto skip;
760e7776783SApple OSS Distributions 		}
761e7776783SApple OSS Distributions 
7625c2921b0SApple OSS Distributions 		v3Entry = (struct nvram_v3_var_entry *)IOMallocZeroData(nvram_v3_var_container_size(header));
763e7776783SApple OSS Distributions 		__nochk_memcpy(&v3Entry->header, _nvramImage + offset, variable_length(header));
764e7776783SApple OSS Distributions 
765e7776783SApple OSS Distributions 		// It is assumed that the initial image being unserialized here is going to be the proxy data from EDT and not the image
766e7776783SApple OSS Distributions 		// read from the controller, which for various reasons due to the setting of states and saves from iBoot, can be
7675c2921b0SApple OSS Distributions 		// different. We will have an initial existing_offset of 0 and once the controller is set we will read
7685c2921b0SApple OSS Distributions 		// out the image there and update the existing offset with what is present on the NOR image
769e7776783SApple OSS Distributions 		v3Entry->existing_offset = 0;
7705c2921b0SApple OSS Distributions 		v3Entry->new_state = VAR_NEW_STATE_NONE;
771e7776783SApple OSS Distributions 
7725c2921b0SApple OSS Distributions 		// safe guard for any strange duplicate entries in the store
773aca3beaaSApple OSS Distributions 		findExistingEntry(v3Entry->header.guid, (const char *)v3Entry->header.name_data_buf, &existingEntry, &existingIndex);
7745c2921b0SApple OSS Distributions 
7755c2921b0SApple OSS Distributions 		if (existingEntry != nullptr) {
7765c2921b0SApple OSS Distributions 			existingSize = variable_length(&existingEntry->header);
7775c2921b0SApple OSS Distributions 
7785c2921b0SApple OSS Distributions 			entryContainer = OSData::withBytes(v3Entry, (uint32_t)nvram_v3_var_container_size(header));
7795c2921b0SApple OSS Distributions 			_varEntries->replaceObject(existingIndex, entryContainer.get());
7805c2921b0SApple OSS Distributions 
7815c2921b0SApple OSS Distributions 			DEBUG_INFO("Found existing for %s, resetting when controller available\n", v3Entry->header.name_data_buf);
7825c2921b0SApple OSS Distributions 			_resetData = true;
7835c2921b0SApple OSS Distributions 		} else {
7845c2921b0SApple OSS Distributions 			entryContainer = OSData::withBytes(v3Entry, (uint32_t)nvram_v3_var_container_size(header));
785e7776783SApple OSS Distributions 			_varEntries->setObject(entryContainer.get());
7865c2921b0SApple OSS Distributions 			existingSize = 0;
7875c2921b0SApple OSS Distributions 		}
788e7776783SApple OSS Distributions 
789e7776783SApple OSS Distributions 		system = (_systemSize != 0) && (uuid_compare(v3Entry->header.guid, gAppleSystemVariableGuid) == 0);
790e7776783SApple OSS Distributions 		if (system) {
7915c2921b0SApple OSS Distributions 			_systemUsed = _systemUsed + (uint32_t)variable_length(header) - (uint32_t)existingSize;
792e7776783SApple OSS Distributions 		} else {
7935c2921b0SApple OSS Distributions 			_commonUsed = _commonUsed + (uint32_t)variable_length(header) - (uint32_t)existingSize;
794e7776783SApple OSS Distributions 		}
795e7776783SApple OSS Distributions 
796e7776783SApple OSS Distributions 		if (convertPropToObject(v3Entry->header.name_data_buf, v3Entry->header.nameSize,
797e7776783SApple OSS Distributions 		    v3Entry->header.name_data_buf + v3Entry->header.nameSize, v3Entry->header.dataSize,
798e7776783SApple OSS Distributions 		    propSymbol, propObject)) {
799aca3beaaSApple OSS Distributions 			OSSharedPtr<const OSSymbol> canonicalKey = keyWithGuidAndCString(v3Entry->header.guid, (const char *)v3Entry->header.name_data_buf);
800e7776783SApple OSS Distributions 
801aca3beaaSApple OSS Distributions 			DEBUG_INFO("adding %s, dataLength=%u, system=%d\n",
802aca3beaaSApple OSS Distributions 			    canonicalKey->getCStringNoCopy(), v3Entry->header.dataSize, system);
803aca3beaaSApple OSS Distributions 
804aca3beaaSApple OSS Distributions 			_varDict->setObject(canonicalKey.get(), propObject.get());
805e7776783SApple OSS Distributions 
806e7776783SApple OSS Distributions 			if (_provider->_diags) {
807aca3beaaSApple OSS Distributions 				_provider->_diags->logVariable(getPartitionTypeForGUID(v3Entry->header.guid),
808e7776783SApple OSS Distributions 				    kIONVRAMOperationInit, propSymbol.get()->getCStringNoCopy(),
809e7776783SApple OSS Distributions 				    (void *)(uintptr_t)(header->name_data_buf + header->nameSize));
810e7776783SApple OSS Distributions 			}
811e7776783SApple OSS Distributions 		}
8125c2921b0SApple OSS Distributions 		IOFreeData(v3Entry, nvram_v3_var_container_size(header));
813e7776783SApple OSS Distributions skip:
814e7776783SApple OSS Distributions 		offset += variable_length(header);
815e7776783SApple OSS Distributions 	}
816e7776783SApple OSS Distributions 
817e7776783SApple OSS Distributions 	_currentOffset = (uint32_t)offset;
818e7776783SApple OSS Distributions 
819e7776783SApple OSS Distributions 	DEBUG_ALWAYS("_commonSize %#x, _systemSize %#x, _currentOffset %#x\n", _commonSize, _systemSize, _currentOffset);
820*8d741a5dSApple OSS Distributions 
821*8d741a5dSApple OSS Distributions 	ret = handleEphDM();
822*8d741a5dSApple OSS Distributions 	verify_noerr_action(ret, panic("handleEphDM failed with ret=%08x", ret));
823*8d741a5dSApple OSS Distributions 
8245c2921b0SApple OSS Distributions 	DEBUG_INFO("_commonUsed %#x, _systemUsed %#x\n", _commonUsed, _systemUsed);
8255c2921b0SApple OSS Distributions 
826e7776783SApple OSS Distributions 	_newData = true;
827e7776783SApple OSS Distributions 
828e7776783SApple OSS Distributions 	if (_provider->_diags) {
829e7776783SApple OSS Distributions 		OSSharedPtr<OSNumber> val = OSNumber::withNumber(getSystemUsed(), 32);
830e7776783SApple OSS Distributions 		_provider->_diags->setProperty(kNVRAMSystemUsedKey, val.get());
831e7776783SApple OSS Distributions 		DEBUG_INFO("%s=%u\n", kNVRAMSystemUsedKey, getSystemUsed());
832e7776783SApple OSS Distributions 
833e7776783SApple OSS Distributions 		val = OSNumber::withNumber(getCommonUsed(), 32);
834e7776783SApple OSS Distributions 		_provider->_diags->setProperty(kNVRAMCommonUsedKey, val.get());
835e7776783SApple OSS Distributions 		DEBUG_INFO("%s=%u\n", kNVRAMCommonUsedKey, getCommonUsed());
836e7776783SApple OSS Distributions 	}
837e7776783SApple OSS Distributions 
838e7776783SApple OSS Distributions 	return ret;
839e7776783SApple OSS Distributions }
840e7776783SApple OSS Distributions 
841e7776783SApple OSS Distributions IOReturn
setVariableInternal(const uuid_t varGuid,const char * variableName,OSObject * object)842aca3beaaSApple OSS Distributions IONVRAMV3Handler::setVariableInternal(const uuid_t varGuid, const char *variableName, OSObject *object)
843e7776783SApple OSS Distributions {
844e7776783SApple OSS Distributions 	struct nvram_v3_var_entry   *v3Entry = nullptr;
845e7776783SApple OSS Distributions 	struct nvram_v3_var_entry   *newV3Entry;
846e7776783SApple OSS Distributions 	OSSharedPtr<OSData>         newContainer;
847aca3beaaSApple OSS Distributions 	OSSharedPtr<const OSSymbol> canonicalKey;
848e7776783SApple OSS Distributions 	bool                        unset = (object == nullptr);
849e7776783SApple OSS Distributions 	bool                        system = false;
850e7776783SApple OSS Distributions 	IOReturn                    ret = kIOReturnSuccess;
851e7776783SApple OSS Distributions 	size_t                      entryNameLen = strlen(variableName) + 1;
852e7776783SApple OSS Distributions 	unsigned int                existingEntryIndex;
853e7776783SApple OSS Distributions 	uint32_t                    dataSize = 0;
8545c2921b0SApple OSS Distributions 	size_t                      existingVariableSize = 0;
8555c2921b0SApple OSS Distributions 	size_t                      newVariableSize = 0;
856e7776783SApple OSS Distributions 	size_t                      newEntrySize;
8575c2921b0SApple OSS Distributions 	uuid_string_t               uuidString;
858e7776783SApple OSS Distributions 
859aca3beaaSApple OSS Distributions 	system = (uuid_compare(varGuid, gAppleSystemVariableGuid) == 0);
860aca3beaaSApple OSS Distributions 	canonicalKey = keyWithGuidAndCString(varGuid, variableName);
861e7776783SApple OSS Distributions 
8625c2921b0SApple OSS Distributions 	uuid_unparse(varGuid, uuidString);
8635c2921b0SApple OSS Distributions 	DEBUG_INFO("setting %s:%s, system=%d, current var count=%u\n", uuidString, variableName, system, _varEntries->getCount());
864e7776783SApple OSS Distributions 
865aca3beaaSApple OSS Distributions 	findExistingEntry(varGuid, variableName, &v3Entry, &existingEntryIndex);
866e7776783SApple OSS Distributions 
867e7776783SApple OSS Distributions 	if (unset == true) {
8685c2921b0SApple OSS Distributions 		setEntryForRemove(v3Entry, system);
869e7776783SApple OSS Distributions 	} else {
8705c2921b0SApple OSS Distributions 		if ((v3Entry != nullptr) && (v3Entry->new_state != VAR_NEW_STATE_REMOVE)) {
8715c2921b0SApple OSS Distributions 			// Sizing was subtracted in setEntryForRemove
8725c2921b0SApple OSS Distributions 			existingVariableSize = variable_length(&v3Entry->header);
873e7776783SApple OSS Distributions 		}
874e7776783SApple OSS Distributions 
875e7776783SApple OSS Distributions 		convertObjectToProp(nullptr, &dataSize, variableName, object);
876e7776783SApple OSS Distributions 
8775c2921b0SApple OSS Distributions 		newVariableSize = sizeof(struct v3_var_header) + entryNameLen + dataSize;
878e7776783SApple OSS Distributions 		newEntrySize = sizeof(struct nvram_v3_var_entry) + entryNameLen + dataSize;
879e7776783SApple OSS Distributions 
8805c2921b0SApple OSS Distributions 		if (system) {
8815c2921b0SApple OSS Distributions 			if (_systemUsed - existingVariableSize + newVariableSize > _systemSize) {
882e7776783SApple OSS Distributions 				DEBUG_ERROR("system region full\n");
883e7776783SApple OSS Distributions 				ret = kIOReturnNoSpace;
884e7776783SApple OSS Distributions 				goto exit;
8855c2921b0SApple OSS Distributions 			}
8865c2921b0SApple OSS Distributions 		} else if (_commonUsed - existingVariableSize + newVariableSize > _commonSize) {
887e7776783SApple OSS Distributions 			DEBUG_ERROR("common region full\n");
888e7776783SApple OSS Distributions 			ret = kIOReturnNoSpace;
889e7776783SApple OSS Distributions 			goto exit;
890e7776783SApple OSS Distributions 		}
891e7776783SApple OSS Distributions 
8925c2921b0SApple OSS Distributions 		DEBUG_INFO("creating new entry for %s, existingVariableSize=%#zx, newVariableSize=%#zx\n", variableName, existingVariableSize, newVariableSize);
893e7776783SApple OSS Distributions 		newV3Entry = (struct nvram_v3_var_entry *)IOMallocZeroData(newEntrySize);
894e7776783SApple OSS Distributions 
895e7776783SApple OSS Distributions 		memcpy(newV3Entry->header.name_data_buf, variableName, entryNameLen);
896e7776783SApple OSS Distributions 		convertObjectToProp(newV3Entry->header.name_data_buf + entryNameLen, &dataSize, variableName, object);
897e7776783SApple OSS Distributions 
898e7776783SApple OSS Distributions 		newV3Entry->header.startId = VARIABLE_DATA;
899e7776783SApple OSS Distributions 		newV3Entry->header.nameSize = (uint32_t)entryNameLen;
900e7776783SApple OSS Distributions 		newV3Entry->header.dataSize = dataSize;
901e7776783SApple OSS Distributions 		newV3Entry->header.crc = crc32(0, newV3Entry->header.name_data_buf + entryNameLen, dataSize);
902aca3beaaSApple OSS Distributions 		memcpy(newV3Entry->header.guid, varGuid, sizeof(gAppleNVRAMGuid));
903e7776783SApple OSS Distributions 		newV3Entry->new_state = VAR_NEW_STATE_APPEND;
904e7776783SApple OSS Distributions 
905e7776783SApple OSS Distributions 		if (v3Entry) {
906e7776783SApple OSS Distributions 			newV3Entry->existing_offset = v3Entry->existing_offset;
907e7776783SApple OSS Distributions 			newV3Entry->header.state = v3Entry->header.state;
908e7776783SApple OSS Distributions 			newV3Entry->header.attributes = v3Entry->header.attributes;
909e7776783SApple OSS Distributions 
910e7776783SApple OSS Distributions 			newContainer = OSData::withBytes(newV3Entry, (uint32_t)newEntrySize);
911e7776783SApple OSS Distributions 			_varEntries->replaceObject(existingEntryIndex, newContainer.get());
912e7776783SApple OSS Distributions 		} else {
913e7776783SApple OSS Distributions 			newContainer = OSData::withBytes(newV3Entry, (uint32_t)newEntrySize);
914e7776783SApple OSS Distributions 			_varEntries->setObject(newContainer.get());
915e7776783SApple OSS Distributions 		}
916e7776783SApple OSS Distributions 
917e7776783SApple OSS Distributions 		if (system) {
9185c2921b0SApple OSS Distributions 			_systemUsed = _systemUsed + (uint32_t)newVariableSize - (uint32_t)existingVariableSize;
919e7776783SApple OSS Distributions 		} else {
9205c2921b0SApple OSS Distributions 			_commonUsed = _commonUsed + (uint32_t)newVariableSize - (uint32_t)existingVariableSize;
921e7776783SApple OSS Distributions 		}
922e7776783SApple OSS Distributions 
923aca3beaaSApple OSS Distributions 		_varDict->setObject(canonicalKey.get(), object);
924aca3beaaSApple OSS Distributions 
925e7776783SApple OSS Distributions 		if (_provider->_diags) {
926aca3beaaSApple OSS Distributions 			_provider->_diags->logVariable(getPartitionTypeForGUID(varGuid),
927aca3beaaSApple OSS Distributions 			    kIONVRAMOperationWrite, variableName,
928aca3beaaSApple OSS Distributions 			    (void *)(uintptr_t)dataSize);
929e7776783SApple OSS Distributions 		}
930e7776783SApple OSS Distributions 
931e7776783SApple OSS Distributions 		IOFreeData(newV3Entry, newEntrySize);
932e7776783SApple OSS Distributions 	}
933e7776783SApple OSS Distributions 
934e7776783SApple OSS Distributions exit:
935e7776783SApple OSS Distributions 	_newData = true;
936e7776783SApple OSS Distributions 
937e7776783SApple OSS Distributions 	if (_provider->_diags) {
938e7776783SApple OSS Distributions 		OSSharedPtr<OSNumber> val = OSNumber::withNumber(getSystemUsed(), 32);
939e7776783SApple OSS Distributions 		_provider->_diags->setProperty(kNVRAMSystemUsedKey, val.get());
940e7776783SApple OSS Distributions 
941e7776783SApple OSS Distributions 		val = OSNumber::withNumber(getCommonUsed(), 32);
942e7776783SApple OSS Distributions 		_provider->_diags->setProperty(kNVRAMCommonUsedKey, val.get());
943e7776783SApple OSS Distributions 	}
944e7776783SApple OSS Distributions 
9455c2921b0SApple OSS Distributions 	DEBUG_INFO("_commonUsed %#x, _systemUsed %#x\n", _commonUsed, _systemUsed);
9465c2921b0SApple OSS Distributions 
947e7776783SApple OSS Distributions 	return ret;
948e7776783SApple OSS Distributions }
949e7776783SApple OSS Distributions 
950aca3beaaSApple OSS Distributions IOReturn
setVariable(const uuid_t varGuid,const char * variableName,OSObject * object)951aca3beaaSApple OSS Distributions IONVRAMV3Handler::setVariable(const uuid_t varGuid, const char *variableName, OSObject *object)
952aca3beaaSApple OSS Distributions {
953aca3beaaSApple OSS Distributions 	uuid_t destGuid;
954aca3beaaSApple OSS Distributions 
95594d3b452SApple OSS Distributions 	if (strcmp(variableName, "reclaim-int") == 0) {
95694d3b452SApple OSS Distributions 		return reclaim();
95794d3b452SApple OSS Distributions 	}
95894d3b452SApple OSS Distributions 
959aca3beaaSApple OSS Distributions 	if (getSystemPartitionActive()) {
960aca3beaaSApple OSS Distributions 		// System region case, if they're using the GUID directly or it's on the system allow list
961aca3beaaSApple OSS Distributions 		// force it to use the System GUID
962aca3beaaSApple OSS Distributions 		if ((uuid_compare(varGuid, gAppleSystemVariableGuid) == 0) || variableInAllowList(variableName)) {
963aca3beaaSApple OSS Distributions 			uuid_copy(destGuid, gAppleSystemVariableGuid);
964aca3beaaSApple OSS Distributions 		} else {
965aca3beaaSApple OSS Distributions 			uuid_copy(destGuid, varGuid);
966aca3beaaSApple OSS Distributions 		}
967aca3beaaSApple OSS Distributions 	} else {
968aca3beaaSApple OSS Distributions 		// No system region, store System GUID as Common GUID
969aca3beaaSApple OSS Distributions 		if ((uuid_compare(varGuid, gAppleSystemVariableGuid) == 0) || variableInAllowList(variableName)) {
970aca3beaaSApple OSS Distributions 			uuid_copy(destGuid, gAppleNVRAMGuid);
971aca3beaaSApple OSS Distributions 		} else {
972aca3beaaSApple OSS Distributions 			uuid_copy(destGuid, varGuid);
973aca3beaaSApple OSS Distributions 		}
974aca3beaaSApple OSS Distributions 	}
975aca3beaaSApple OSS Distributions 
976aca3beaaSApple OSS Distributions 	return setVariableInternal(destGuid, variableName, object);
977aca3beaaSApple OSS Distributions }
978aca3beaaSApple OSS Distributions 
9795c2921b0SApple OSS Distributions uint32_t
findCurrentBank(void)9805c2921b0SApple OSS Distributions IONVRAMV3Handler::findCurrentBank(void)
9815c2921b0SApple OSS Distributions {
9825c2921b0SApple OSS Distributions 	struct v3_store_header storeHeader;
9835c2921b0SApple OSS Distributions 	uint32_t               maxGen = 0;
9845c2921b0SApple OSS Distributions 	uint32_t               currentBank = 0;
9855c2921b0SApple OSS Distributions 
9865c2921b0SApple OSS Distributions 	for (unsigned int i = 0; i < _bankCount; i++) {
9875c2921b0SApple OSS Distributions 		_nvramController->select(i);
9885c2921b0SApple OSS Distributions 		_nvramController->read(0, (uint8_t *)&storeHeader, sizeof(storeHeader));
9895c2921b0SApple OSS Distributions 
9905c2921b0SApple OSS Distributions 		if (valid_store_header(&storeHeader) && (storeHeader.generation >= maxGen)) {
9915c2921b0SApple OSS Distributions 			currentBank = i;
9925c2921b0SApple OSS Distributions 			maxGen = storeHeader.generation;
9935c2921b0SApple OSS Distributions 		}
9945c2921b0SApple OSS Distributions 	}
9955c2921b0SApple OSS Distributions 
9965c2921b0SApple OSS Distributions 	DEBUG_ALWAYS("currentBank=%#x, gen=%#x", currentBank, maxGen);
9975c2921b0SApple OSS Distributions 
9985c2921b0SApple OSS Distributions 	return currentBank;
9995c2921b0SApple OSS Distributions }
10005c2921b0SApple OSS Distributions 
1001e7776783SApple OSS Distributions bool
setController(IONVRAMController * controller)1002e7776783SApple OSS Distributions IONVRAMV3Handler::setController(IONVRAMController *controller)
1003e7776783SApple OSS Distributions {
1004e7776783SApple OSS Distributions 	IOReturn ret = kIOReturnSuccess;
1005e7776783SApple OSS Distributions 
1006e7776783SApple OSS Distributions 	if (_nvramController == NULL) {
1007e7776783SApple OSS Distributions 		_nvramController = controller;
1008e7776783SApple OSS Distributions 	}
1009e7776783SApple OSS Distributions 
10105c2921b0SApple OSS Distributions 	DEBUG_INFO("Controller name: %s\n", _nvramController->getName());
1011e7776783SApple OSS Distributions 
10125c2921b0SApple OSS Distributions 	require(_bankSize != 0, exit);
1013e7776783SApple OSS Distributions 
10145c2921b0SApple OSS Distributions 	if (_resetData) {
10155c2921b0SApple OSS Distributions 		_resetData = false;
10165c2921b0SApple OSS Distributions 		DEBUG_ERROR("_resetData set, issuing reclaim recovery\n");
10175c2921b0SApple OSS Distributions 		ret = reclaim();
10185c2921b0SApple OSS Distributions 		require_noerr_action(ret, exit, DEBUG_ERROR("Reclaim recovery failed, invalid controller state!!! ret=%#x\n", ret));
10195c2921b0SApple OSS Distributions 		goto exit;
1020e7776783SApple OSS Distributions 	}
1021e7776783SApple OSS Distributions 
10225c2921b0SApple OSS Distributions 	ret = reloadInternal();
10235c2921b0SApple OSS Distributions 	if (ret != kIOReturnSuccess) {
1024e7776783SApple OSS Distributions 		DEBUG_ERROR("Invalid image found, issuing reclaim recovery\n");
1025e7776783SApple OSS Distributions 		ret = reclaim();
1026e7776783SApple OSS Distributions 		require_noerr_action(ret, exit, DEBUG_ERROR("Reclaim recovery failed, invalid controller state!!! ret=%#x\n", ret));
1027e7776783SApple OSS Distributions 	}
1028e7776783SApple OSS Distributions 
1029e7776783SApple OSS Distributions exit:
1030e7776783SApple OSS Distributions 	return ret == kIOReturnSuccess;
1031e7776783SApple OSS Distributions }
1032e7776783SApple OSS Distributions 
1033e7776783SApple OSS Distributions IOReturn
reclaim(void)1034e7776783SApple OSS Distributions IONVRAMV3Handler::reclaim(void)
1035e7776783SApple OSS Distributions {
1036e7776783SApple OSS Distributions 	IOReturn             ret;
1037e7776783SApple OSS Distributions 	struct               v3_store_header newStoreHeader;
1038e7776783SApple OSS Distributions 	struct               v3_var_header *varHeader;
1039e7776783SApple OSS Distributions 	struct               nvram_v3_var_entry *varEntry;
1040e7776783SApple OSS Distributions 	OSData               *entryContainer;
1041e7776783SApple OSS Distributions 	size_t               new_bank_offset = sizeof(struct v3_store_header);
10425c2921b0SApple OSS Distributions 	uint32_t             next_bank = (_currentBank + 1) % _bankCount;
104394d3b452SApple OSS Distributions 	uint8_t              *bankData;
104494d3b452SApple OSS Distributions 	OSSharedPtr<OSArray> remainingEntries;
1045e7776783SApple OSS Distributions 
1046e7776783SApple OSS Distributions 	DEBUG_INFO("called\n");
1047e7776783SApple OSS Distributions 
104894d3b452SApple OSS Distributions 	bankData = (uint8_t *)IOMallocData(_bankSize);
104994d3b452SApple OSS Distributions 	require_action(bankData != nullptr, exit, ret = kIOReturnNoMemory);
105094d3b452SApple OSS Distributions 
10515c2921b0SApple OSS Distributions 	ret = _nvramController->select(next_bank);
10525c2921b0SApple OSS Distributions 	verify_noerr_action(ret, DEBUG_INFO("select of bank %#08x failed\n", next_bank));
10535c2921b0SApple OSS Distributions 
10545c2921b0SApple OSS Distributions 	ret = _nvramController->eraseBank();
10555c2921b0SApple OSS Distributions 	verify_noerr_action(ret, DEBUG_INFO("eraseBank failed, ret=%#08x\n", ret));
10565c2921b0SApple OSS Distributions 
10575c2921b0SApple OSS Distributions 	_currentBank = next_bank;
1058e7776783SApple OSS Distributions 
105994d3b452SApple OSS Distributions 	remainingEntries = OSArray::withCapacity(_varEntries->getCapacity());
106094d3b452SApple OSS Distributions 
1061e7776783SApple OSS Distributions 	for (unsigned int i = 0; i < _varEntries->getCount(); i++) {
1062e7776783SApple OSS Distributions 		entryContainer = OSDynamicCast(OSData, _varEntries->getObject(i));
1063e7776783SApple OSS Distributions 		varEntry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
1064e7776783SApple OSS Distributions 		varHeader = &varEntry->header;
1065e7776783SApple OSS Distributions 
1066e7776783SApple OSS Distributions 		DEBUG_INFO("entry %u %s, new_state=%#x, e_offset=%#lx, state=%#x\n",
1067e7776783SApple OSS Distributions 		    i, varEntry->header.name_data_buf, varEntry->new_state, varEntry->existing_offset, varHeader->state);
1068e7776783SApple OSS Distributions 
106994d3b452SApple OSS Distributions 		if ((varEntry->new_state == VAR_NEW_STATE_NONE) ||
107094d3b452SApple OSS Distributions 		    (varEntry->new_state == VAR_NEW_STATE_APPEND)) {
107194d3b452SApple OSS Distributions 			varHeader->state = VAR_ADDED;
1072e7776783SApple OSS Distributions 
107394d3b452SApple OSS Distributions 			memcpy(bankData + new_bank_offset, (uint8_t *)varHeader, variable_length(varHeader));
107494d3b452SApple OSS Distributions 
107594d3b452SApple OSS Distributions 			varEntry->new_state = VAR_NEW_STATE_NONE;
1076e7776783SApple OSS Distributions 			varEntry->existing_offset = new_bank_offset;
1077e7776783SApple OSS Distributions 			new_bank_offset += variable_length(varHeader);
107894d3b452SApple OSS Distributions 
107994d3b452SApple OSS Distributions 			remainingEntries->setObject(entryContainer);
1080e7776783SApple OSS Distributions 		} else {
108194d3b452SApple OSS Distributions 			// entryContainer not added to remainingEntries, entry dropped
1082e7776783SApple OSS Distributions 		}
1083e7776783SApple OSS Distributions 	}
1084e7776783SApple OSS Distributions 
1085e7776783SApple OSS Distributions 	memcpy(&newStoreHeader, _nvramImage, sizeof(newStoreHeader));
1086e7776783SApple OSS Distributions 
1087e7776783SApple OSS Distributions 	_generation += 1;
1088e7776783SApple OSS Distributions 
1089e7776783SApple OSS Distributions 	newStoreHeader.generation = _generation;
1090e7776783SApple OSS Distributions 
109194d3b452SApple OSS Distributions 	memcpy(bankData, (uint8_t *)&newStoreHeader, sizeof(newStoreHeader));
109294d3b452SApple OSS Distributions 
109394d3b452SApple OSS Distributions 	ret = _nvramController->write(0, bankData, new_bank_offset);
109494d3b452SApple OSS Distributions 	require_noerr_action(ret, exit, DEBUG_ERROR("reclaim bank write failed, ret=%08x\n", ret));
1095e7776783SApple OSS Distributions 
1096e7776783SApple OSS Distributions 	_currentOffset = (uint32_t)new_bank_offset;
1097e7776783SApple OSS Distributions 
109894d3b452SApple OSS Distributions 	DEBUG_INFO("Reclaim complete, _currentBank=%u _generation=%u, _currentOffset=%#x\n", _currentBank, _generation, _currentOffset);
109994d3b452SApple OSS Distributions 
110094d3b452SApple OSS Distributions 	_newData = false;
110194d3b452SApple OSS Distributions 
110294d3b452SApple OSS Distributions 	_varEntries.reset(remainingEntries.get(), OSRetain);
1103e7776783SApple OSS Distributions 
1104e7776783SApple OSS Distributions exit:
110594d3b452SApple OSS Distributions 	IOFreeData(bankData, _bankSize);
110694d3b452SApple OSS Distributions 
1107e7776783SApple OSS Distributions 	return ret;
1108e7776783SApple OSS Distributions }
1109e7776783SApple OSS Distributions 
111094d3b452SApple OSS Distributions size_t
getAppendSize(void)111194d3b452SApple OSS Distributions IONVRAMV3Handler::getAppendSize(void)
111294d3b452SApple OSS Distributions {
111394d3b452SApple OSS Distributions 	struct nvram_v3_var_entry *varEntry;
111494d3b452SApple OSS Distributions 	struct v3_var_header      *varHeader;
111594d3b452SApple OSS Distributions 	OSData                    *entryContainer;
111694d3b452SApple OSS Distributions 	size_t                    appendSize = 0;
111794d3b452SApple OSS Distributions 
111894d3b452SApple OSS Distributions 	for (unsigned int i = 0; i < _varEntries->getCount(); i++) {
111994d3b452SApple OSS Distributions 		entryContainer = OSDynamicCast(OSData, _varEntries->getObject(i));
112094d3b452SApple OSS Distributions 		varEntry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
112194d3b452SApple OSS Distributions 		varHeader = &varEntry->header;
112294d3b452SApple OSS Distributions 
112394d3b452SApple OSS Distributions 		if (varEntry->new_state == VAR_NEW_STATE_APPEND) {
112494d3b452SApple OSS Distributions 			appendSize += variable_length(varHeader);
112594d3b452SApple OSS Distributions 		}
112694d3b452SApple OSS Distributions 	}
112794d3b452SApple OSS Distributions 
112894d3b452SApple OSS Distributions 	return appendSize;
112994d3b452SApple OSS Distributions }
113094d3b452SApple OSS Distributions 
1131e7776783SApple OSS Distributions IOReturn
syncRaw(void)11325c2921b0SApple OSS Distributions IONVRAMV3Handler::syncRaw(void)
1133e7776783SApple OSS Distributions {
1134e7776783SApple OSS Distributions 	IOReturn                  ret = kIOReturnSuccess;
1135e7776783SApple OSS Distributions 	struct nvram_v3_var_entry *varEntry;
1136e7776783SApple OSS Distributions 	struct v3_var_header      *varHeader;
1137e7776783SApple OSS Distributions 	OSData                    *entryContainer;
1138e7776783SApple OSS Distributions 	OSSharedPtr<OSArray>      remainingEntries;
113994d3b452SApple OSS Distributions 	uint8_t                   *appendBuffer = nullptr;
114094d3b452SApple OSS Distributions 	size_t                    appendBufferOffset = 0;
114194d3b452SApple OSS Distributions 	size_t                    *invalidateOffsets = nullptr;
114294d3b452SApple OSS Distributions 	size_t                    invalidateOffsetsCount = 0;
114394d3b452SApple OSS Distributions 	size_t                    invalidateOffsetIndex = 0;
114494d3b452SApple OSS Distributions 	size_t                    invalidatedSize = 0;
1145e7776783SApple OSS Distributions 
1146e7776783SApple OSS Distributions 	require_action(_nvramController != nullptr, exit, DEBUG_INFO("No _nvramController\n"));
1147e7776783SApple OSS Distributions 	require_action(_newData == true, exit, DEBUG_INFO("No _newData to sync\n"));
11485c2921b0SApple OSS Distributions 	require_action(_bankSize != 0, exit, DEBUG_INFO("No nvram size info\n"));
1149e7776783SApple OSS Distributions 
1150e7776783SApple OSS Distributions 	DEBUG_INFO("_varEntries->getCount()=%#x\n", _varEntries->getCount());
1151e7776783SApple OSS Distributions 
115294d3b452SApple OSS Distributions 	if (getAppendSize() + _currentOffset < _bankSize) {
115394d3b452SApple OSS Distributions 		// No reclaim, build append and invalidate list
115494d3b452SApple OSS Distributions 
1155e7776783SApple OSS Distributions 		remainingEntries = OSArray::withCapacity(_varEntries->getCapacity());
1156e7776783SApple OSS Distributions 
115794d3b452SApple OSS Distributions 		appendBuffer = (uint8_t *)IOMallocData(_bankSize);
115894d3b452SApple OSS Distributions 		require_action(appendBuffer, exit, ret = kIOReturnNoMemory);
1159e7776783SApple OSS Distributions 
116094d3b452SApple OSS Distributions 		invalidateOffsetsCount = _varEntries->getCount();
116194d3b452SApple OSS Distributions 		invalidateOffsets = (size_t *)IOMallocData(invalidateOffsetsCount * sizeof(size_t));
116294d3b452SApple OSS Distributions 		require_action(invalidateOffsets, exit, ret = kIOReturnNoMemory);
116394d3b452SApple OSS Distributions 
116494d3b452SApple OSS Distributions 		for (unsigned int i = 0; i < _varEntries->getCount(); i++) {
1165e7776783SApple OSS Distributions 			entryContainer = OSDynamicCast(OSData, _varEntries->getObject(i));
1166e7776783SApple OSS Distributions 			varEntry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
1167e7776783SApple OSS Distributions 			varHeader = &varEntry->header;
1168e7776783SApple OSS Distributions 
116994d3b452SApple OSS Distributions 			DEBUG_INFO("entry %s, new_state=%#02x state=%#02x, existing_offset=%#zx\n",
117094d3b452SApple OSS Distributions 			    varEntry->header.name_data_buf, varEntry->new_state, varEntry->header.state, varEntry->existing_offset);
1171e7776783SApple OSS Distributions 
1172e7776783SApple OSS Distributions 			if (varEntry->new_state == VAR_NEW_STATE_APPEND) {
117394d3b452SApple OSS Distributions 				size_t varSize = variable_length(varHeader);
117494d3b452SApple OSS Distributions 				size_t prevOffset = varEntry->existing_offset;
1175e7776783SApple OSS Distributions 
1176e7776783SApple OSS Distributions 				varHeader->state = VAR_ADDED;
117794d3b452SApple OSS Distributions 				varEntry->existing_offset = _currentOffset + appendBufferOffset;
1178e7776783SApple OSS Distributions 				varEntry->new_state = VAR_NEW_STATE_NONE;
1179e7776783SApple OSS Distributions 
118094d3b452SApple OSS Distributions 				DEBUG_INFO("Appending %s in append buffer offset %#zx, actual offset %#zx, prevOffset %#zx, varsize=%#zx\n",
118194d3b452SApple OSS Distributions 				    varEntry->header.name_data_buf, appendBufferOffset, varEntry->existing_offset, prevOffset, varSize);
118294d3b452SApple OSS Distributions 
118394d3b452SApple OSS Distributions 				// Write to append buffer
118494d3b452SApple OSS Distributions 				memcpy(appendBuffer + appendBufferOffset, (uint8_t *)varHeader, varSize);
118594d3b452SApple OSS Distributions 				appendBufferOffset += varSize;
118694d3b452SApple OSS Distributions 
118794d3b452SApple OSS Distributions 				if (prevOffset) {
118894d3b452SApple OSS Distributions 					invalidateOffsets[invalidateOffsetIndex++] = prevOffset;
118994d3b452SApple OSS Distributions 					invalidatedSize += variable_length((struct v3_var_header *)prevOffset);
119094d3b452SApple OSS Distributions 				}
1191e7776783SApple OSS Distributions 
1192e7776783SApple OSS Distributions 				remainingEntries->setObject(entryContainer);
1193e7776783SApple OSS Distributions 			} else if (varEntry->new_state == VAR_NEW_STATE_REMOVE) {
1194e7776783SApple OSS Distributions 				if (varEntry->existing_offset) {
1195e7776783SApple OSS Distributions 					DEBUG_INFO("marking entry at offset %#lx deleted\n", varEntry->existing_offset);
1196e7776783SApple OSS Distributions 
119794d3b452SApple OSS Distributions 					invalidateOffsets[invalidateOffsetIndex++] = varEntry->existing_offset;
119894d3b452SApple OSS Distributions 					invalidatedSize += variable_length((struct v3_var_header *)varEntry->existing_offset);
1199e7776783SApple OSS Distributions 				} else {
120094d3b452SApple OSS Distributions 					DEBUG_INFO("No existing_offset , removing\n");
1201e7776783SApple OSS Distributions 				}
1202e7776783SApple OSS Distributions 
1203e7776783SApple OSS Distributions 				// not re-added to remainingEntries
1204e7776783SApple OSS Distributions 			} else {
1205e7776783SApple OSS Distributions 				DEBUG_INFO("skipping\n");
1206e7776783SApple OSS Distributions 				remainingEntries->setObject(entryContainer);
1207e7776783SApple OSS Distributions 			}
1208e7776783SApple OSS Distributions 		}
1209e7776783SApple OSS Distributions 
121094d3b452SApple OSS Distributions 		if (appendBufferOffset > 0) {
121194d3b452SApple OSS Distributions 			// Write appendBuffer
121294d3b452SApple OSS Distributions 			DEBUG_INFO("Appending append buffer size=%#zx at offset=%#x\n", appendBufferOffset, _currentOffset);
121394d3b452SApple OSS Distributions 			ret = _nvramController->write(_currentOffset, appendBuffer, appendBufferOffset);
121494d3b452SApple OSS Distributions 			require_noerr_action(ret, exit, DEBUG_ERROR("could not re-append, ret=%#x\n", ret));
121594d3b452SApple OSS Distributions 
121694d3b452SApple OSS Distributions 			_currentOffset += appendBufferOffset;
121794d3b452SApple OSS Distributions 		} else {
121894d3b452SApple OSS Distributions 			DEBUG_INFO("No entries to append\n");
121994d3b452SApple OSS Distributions 		}
122094d3b452SApple OSS Distributions 
122194d3b452SApple OSS Distributions 		if (invalidateOffsetIndex > 0) {
122294d3b452SApple OSS Distributions 			// Invalidate Entries
122394d3b452SApple OSS Distributions 			for (unsigned int i = 0; i < invalidateOffsetIndex; i++) {
122494d3b452SApple OSS Distributions 				uint8_t state = VAR_ADDED & VAR_DELETED & VAR_IN_DELETED_TRANSITION;
122594d3b452SApple OSS Distributions 
122694d3b452SApple OSS Distributions 				ret = _nvramController->write(invalidateOffsets[i] + offsetof(struct v3_var_header, state), &state, sizeof(state));
122794d3b452SApple OSS Distributions 				require_noerr_action(ret, exit, DEBUG_ERROR("unable to invalidate at offset %#zx, ret=%#x\n", invalidateOffsets[i], ret));
122894d3b452SApple OSS Distributions 				DEBUG_INFO("Invalidated entry at offset=%#zx\n", invalidateOffsets[i]);
122994d3b452SApple OSS Distributions 			}
123094d3b452SApple OSS Distributions 		} else {
123194d3b452SApple OSS Distributions 			DEBUG_INFO("No entries to invalidate\n");
123294d3b452SApple OSS Distributions 		}
1233e7776783SApple OSS Distributions 
1234e7776783SApple OSS Distributions 		_newData = false;
1235e7776783SApple OSS Distributions 
123694d3b452SApple OSS Distributions 		_varEntries.reset(remainingEntries.get(), OSRetain);
123794d3b452SApple OSS Distributions 	} else {
123894d3b452SApple OSS Distributions 		// Will need to reclaim, rebuild store and write everything at once
123994d3b452SApple OSS Distributions 		ret = reclaim();
124094d3b452SApple OSS Distributions 	}
124194d3b452SApple OSS Distributions 
1242e7776783SApple OSS Distributions exit:
124394d3b452SApple OSS Distributions 	IOFreeData(appendBuffer, _bankSize);
124494d3b452SApple OSS Distributions 	IOFreeData(invalidateOffsets, invalidateOffsetsCount * sizeof(size_t));
124594d3b452SApple OSS Distributions 
1246e7776783SApple OSS Distributions 	return ret;
1247e7776783SApple OSS Distributions }
1248e7776783SApple OSS Distributions 
12495c2921b0SApple OSS Distributions IOReturn
syncBlock(void)12505c2921b0SApple OSS Distributions IONVRAMV3Handler::syncBlock(void)
12515c2921b0SApple OSS Distributions {
12525c2921b0SApple OSS Distributions 	IOReturn             ret = kIOReturnSuccess;
12535c2921b0SApple OSS Distributions 	struct               v3_store_header newStoreHeader;
12545c2921b0SApple OSS Distributions 	struct               v3_var_header *varHeader;
12555c2921b0SApple OSS Distributions 	struct               nvram_v3_var_entry *varEntry;
12565c2921b0SApple OSS Distributions 	OSData               *entryContainer;
12575c2921b0SApple OSS Distributions 	size_t               new_bank_offset = sizeof(struct v3_store_header);
12585c2921b0SApple OSS Distributions 	uint8_t              *block;
12595c2921b0SApple OSS Distributions 	OSSharedPtr<OSArray> remainingEntries;
12605c2921b0SApple OSS Distributions 	uint32_t             next_bank = (_currentBank + 1) % _bankCount;
12615c2921b0SApple OSS Distributions 
12625c2921b0SApple OSS Distributions 	DEBUG_INFO("called\n");
12635c2921b0SApple OSS Distributions 
12645c2921b0SApple OSS Distributions 	require_action(_nvramController != nullptr, exit, DEBUG_INFO("No _nvramController\n"));
12655c2921b0SApple OSS Distributions 	require_action(_newData == true, exit, DEBUG_INFO("No _newData to sync\n"));
12665c2921b0SApple OSS Distributions 	require_action(_bankSize != 0, exit, DEBUG_INFO("No nvram size info\n"));
12675c2921b0SApple OSS Distributions 
12685c2921b0SApple OSS Distributions 	block = (uint8_t *)IOMallocData(_bankSize);
12695c2921b0SApple OSS Distributions 
12705c2921b0SApple OSS Distributions 	remainingEntries = OSArray::withCapacity(_varEntries->getCapacity());
12715c2921b0SApple OSS Distributions 
12725c2921b0SApple OSS Distributions 	ret = _nvramController->select(next_bank);
12735c2921b0SApple OSS Distributions 	verify_noerr_action(ret, DEBUG_INFO("select of bank %#x failed\n", next_bank));
12745c2921b0SApple OSS Distributions 
12755c2921b0SApple OSS Distributions 	ret = _nvramController->eraseBank();
12765c2921b0SApple OSS Distributions 	verify_noerr_action(ret, DEBUG_INFO("eraseBank failed, ret=%#08x\n", ret));
12775c2921b0SApple OSS Distributions 
12785c2921b0SApple OSS Distributions 	_currentBank = next_bank;
12795c2921b0SApple OSS Distributions 
12805c2921b0SApple OSS Distributions 	memcpy(&newStoreHeader, _nvramImage, sizeof(newStoreHeader));
12815c2921b0SApple OSS Distributions 
12825c2921b0SApple OSS Distributions 	_generation += 1;
12835c2921b0SApple OSS Distributions 
12845c2921b0SApple OSS Distributions 	newStoreHeader.generation = _generation;
12855c2921b0SApple OSS Distributions 
12865c2921b0SApple OSS Distributions 	memcpy(block, (uint8_t *)&newStoreHeader, sizeof(newStoreHeader));
12875c2921b0SApple OSS Distributions 
12885c2921b0SApple OSS Distributions 	for (unsigned int i = 0; i < _varEntries->getCount(); i++) {
12895c2921b0SApple OSS Distributions 		entryContainer = OSDynamicCast(OSData, _varEntries->getObject(i));
12905c2921b0SApple OSS Distributions 		varEntry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
12915c2921b0SApple OSS Distributions 		varHeader = &varEntry->header;
12925c2921b0SApple OSS Distributions 
12935c2921b0SApple OSS Distributions 		DEBUG_INFO("entry %u %s, new_state=%#x, e_offset=%#lx, state=%#x\n",
12945c2921b0SApple OSS Distributions 		    i, varEntry->header.name_data_buf, varEntry->new_state, varEntry->existing_offset, varHeader->state);
12955c2921b0SApple OSS Distributions 
12965c2921b0SApple OSS Distributions 		if (varEntry->new_state != VAR_NEW_STATE_REMOVE) {
129794d3b452SApple OSS Distributions 			varHeader->state = VAR_ADDED;
129894d3b452SApple OSS Distributions 
12995c2921b0SApple OSS Distributions 			memcpy(block + new_bank_offset, (uint8_t *)varHeader, variable_length(varHeader));
13005c2921b0SApple OSS Distributions 
13015c2921b0SApple OSS Distributions 			varEntry->existing_offset = new_bank_offset;
13025c2921b0SApple OSS Distributions 			new_bank_offset += variable_length(varHeader);
13035c2921b0SApple OSS Distributions 			varEntry->new_state = VAR_NEW_STATE_NONE;
13045c2921b0SApple OSS Distributions 
13055c2921b0SApple OSS Distributions 			remainingEntries->setObject(entryContainer);
13065c2921b0SApple OSS Distributions 		} else {
13075c2921b0SApple OSS Distributions 			DEBUG_INFO("Dropping %s\n", varEntry->header.name_data_buf);
13085c2921b0SApple OSS Distributions 		}
13095c2921b0SApple OSS Distributions 	}
13105c2921b0SApple OSS Distributions 
13115c2921b0SApple OSS Distributions 	ret = _nvramController->write(0, block, _bankSize);
13125c2921b0SApple OSS Distributions 	verify_noerr_action(ret, DEBUG_ERROR("w fail, ret=%#x\n", ret));
13135c2921b0SApple OSS Distributions 
13145c2921b0SApple OSS Distributions 	_nvramController->sync();
13155c2921b0SApple OSS Distributions 
13165c2921b0SApple OSS Distributions 	_varEntries.reset(remainingEntries.get(), OSRetain);
13175c2921b0SApple OSS Distributions 
13185c2921b0SApple OSS Distributions 	_newData = false;
13195c2921b0SApple OSS Distributions 
13205c2921b0SApple OSS Distributions 	DEBUG_INFO("Save complete, _generation=%u\n", _generation);
13215c2921b0SApple OSS Distributions 
13225c2921b0SApple OSS Distributions 	IOFreeData(block, _bankSize);
13235c2921b0SApple OSS Distributions 
13245c2921b0SApple OSS Distributions exit:
13255c2921b0SApple OSS Distributions 	return ret;
13265c2921b0SApple OSS Distributions }
13275c2921b0SApple OSS Distributions 
132894d3b452SApple OSS Distributions IOReturn
sync(void)1329e7776783SApple OSS Distributions IONVRAMV3Handler::sync(void)
1330e7776783SApple OSS Distributions {
1331e7776783SApple OSS Distributions 	IOReturn ret;
1332e7776783SApple OSS Distributions 
13335c2921b0SApple OSS Distributions 	if (_reload) {
13345c2921b0SApple OSS Distributions 		ret = reloadInternal();
13355c2921b0SApple OSS Distributions 		require_noerr_action(ret, exit, DEBUG_ERROR("Reload failed, ret=%#x", ret));
13365c2921b0SApple OSS Distributions 
13375c2921b0SApple OSS Distributions 		_reload = false;
13385c2921b0SApple OSS Distributions 	}
13395c2921b0SApple OSS Distributions 
13405c2921b0SApple OSS Distributions 	if (_rawController == true) {
13415c2921b0SApple OSS Distributions 		ret = syncRaw();
1342e7776783SApple OSS Distributions 
1343e7776783SApple OSS Distributions 		if (ret != kIOReturnSuccess) {
1344e7776783SApple OSS Distributions 			ret = reclaim();
1345e7776783SApple OSS Distributions 			require_noerr_action(ret, exit, DEBUG_ERROR("Reclaim recovery failed, ret=%#x", ret));
13465c2921b0SApple OSS Distributions 		}
13475c2921b0SApple OSS Distributions 	} else {
13485c2921b0SApple OSS Distributions 		ret = syncBlock();
1349e7776783SApple OSS Distributions 	}
1350e7776783SApple OSS Distributions 
1351e7776783SApple OSS Distributions exit:
135294d3b452SApple OSS Distributions 	return ret;
1353e7776783SApple OSS Distributions }
1354e7776783SApple OSS Distributions 
1355e7776783SApple OSS Distributions uint32_t
getGeneration(void) const1356e7776783SApple OSS Distributions IONVRAMV3Handler::getGeneration(void) const
1357e7776783SApple OSS Distributions {
1358e7776783SApple OSS Distributions 	return _generation;
1359e7776783SApple OSS Distributions }
1360e7776783SApple OSS Distributions 
1361e7776783SApple OSS Distributions uint32_t
getVersion(void) const1362e7776783SApple OSS Distributions IONVRAMV3Handler::getVersion(void) const
1363e7776783SApple OSS Distributions {
1364e7776783SApple OSS Distributions 	return kNVRAMVersion3;
1365e7776783SApple OSS Distributions }
1366e7776783SApple OSS Distributions 
1367e7776783SApple OSS Distributions uint32_t
getSystemUsed(void) const1368e7776783SApple OSS Distributions IONVRAMV3Handler::getSystemUsed(void) const
1369e7776783SApple OSS Distributions {
1370e7776783SApple OSS Distributions 	return _systemUsed;
1371e7776783SApple OSS Distributions }
1372e7776783SApple OSS Distributions 
1373e7776783SApple OSS Distributions uint32_t
getCommonUsed(void) const1374e7776783SApple OSS Distributions IONVRAMV3Handler::getCommonUsed(void) const
1375e7776783SApple OSS Distributions {
1376e7776783SApple OSS Distributions 	return _commonUsed;
1377e7776783SApple OSS Distributions }
1378e7776783SApple OSS Distributions 
1379e7776783SApple OSS Distributions bool
getSystemPartitionActive(void) const1380aca3beaaSApple OSS Distributions IONVRAMV3Handler::getSystemPartitionActive(void) const
1381aca3beaaSApple OSS Distributions {
1382aca3beaaSApple OSS Distributions 	return _systemSize != 0;
1383aca3beaaSApple OSS Distributions }
1384aca3beaaSApple OSS Distributions 
1385aca3beaaSApple OSS Distributions bool
convertObjectToProp(uint8_t * buffer,uint32_t * length,const char * propName,OSObject * propObject)1386e7776783SApple OSS Distributions IONVRAMV3Handler::convertObjectToProp(uint8_t *buffer, uint32_t *length,
1387e7776783SApple OSS Distributions     const char *propName, OSObject *propObject)
1388e7776783SApple OSS Distributions {
1389e7776783SApple OSS Distributions 	uint32_t             offset;
1390e7776783SApple OSS Distributions 	IONVRAMVariableType  propType;
1391e7776783SApple OSS Distributions 	OSBoolean            *tmpBoolean = nullptr;
1392e7776783SApple OSS Distributions 	OSNumber             *tmpNumber = nullptr;
1393e7776783SApple OSS Distributions 	OSString             *tmpString = nullptr;
1394e7776783SApple OSS Distributions 	OSData               *tmpData = nullptr;
1395e7776783SApple OSS Distributions 
1396e7776783SApple OSS Distributions 	propType = getVariableType(propName);
1397e7776783SApple OSS Distributions 
1398e7776783SApple OSS Distributions 	// Get the size of the data.
1399e7776783SApple OSS Distributions 	offset = 0;
1400e7776783SApple OSS Distributions 	switch (propType) {
1401e7776783SApple OSS Distributions 	case kOFVariableTypeBoolean:
1402e7776783SApple OSS Distributions 		tmpBoolean = OSDynamicCast(OSBoolean, propObject);
1403e7776783SApple OSS Distributions 		if (tmpBoolean != nullptr) {
1404e7776783SApple OSS Distributions 			const char *bool_buf;
1405e7776783SApple OSS Distributions 			if (tmpBoolean->getValue()) {
1406e7776783SApple OSS Distributions 				bool_buf = "true";
1407e7776783SApple OSS Distributions 			} else {
1408e7776783SApple OSS Distributions 				bool_buf = "false";
1409e7776783SApple OSS Distributions 			}
1410e7776783SApple OSS Distributions 
1411e7776783SApple OSS Distributions 			offset = (uint32_t)strlen(bool_buf);
1412e7776783SApple OSS Distributions 
1413e7776783SApple OSS Distributions 			if (buffer) {
1414e7776783SApple OSS Distributions 				if (*length < offset) {
1415e7776783SApple OSS Distributions 					return false;
1416e7776783SApple OSS Distributions 				} else {
1417e7776783SApple OSS Distributions 					memcpy(buffer, bool_buf, offset);
1418e7776783SApple OSS Distributions 				}
1419e7776783SApple OSS Distributions 			}
1420e7776783SApple OSS Distributions 		}
1421e7776783SApple OSS Distributions 		break;
1422e7776783SApple OSS Distributions 
1423e7776783SApple OSS Distributions 	case kOFVariableTypeNumber:
1424e7776783SApple OSS Distributions 		tmpNumber = OSDynamicCast(OSNumber, propObject);
1425e7776783SApple OSS Distributions 		if (tmpNumber != nullptr) {
1426e7776783SApple OSS Distributions 			char num_buf[12];
1427e7776783SApple OSS Distributions 			char *end_buf = num_buf;
1428e7776783SApple OSS Distributions 			uint32_t tmpValue = tmpNumber->unsigned32BitValue();
1429e7776783SApple OSS Distributions 			if (tmpValue == 0xFFFFFFFF) {
1430e7776783SApple OSS Distributions 				end_buf += snprintf(end_buf, sizeof(num_buf), "-1");
1431e7776783SApple OSS Distributions 			} else if (tmpValue < 1000) {
1432e7776783SApple OSS Distributions 				end_buf += snprintf(end_buf, sizeof(num_buf), "%d", (uint32_t)tmpValue);
1433e7776783SApple OSS Distributions 			} else {
1434e7776783SApple OSS Distributions 				end_buf += snprintf(end_buf, sizeof(num_buf), "%#x", (uint32_t)tmpValue);
1435e7776783SApple OSS Distributions 			}
1436e7776783SApple OSS Distributions 
1437e7776783SApple OSS Distributions 			offset = (uint32_t)(end_buf - num_buf);
1438e7776783SApple OSS Distributions 			if (buffer) {
1439e7776783SApple OSS Distributions 				if (*length < offset) {
1440e7776783SApple OSS Distributions 					return false;
1441e7776783SApple OSS Distributions 				} else {
1442e7776783SApple OSS Distributions 					memcpy(buffer, num_buf, offset);
1443e7776783SApple OSS Distributions 				}
1444e7776783SApple OSS Distributions 			}
1445e7776783SApple OSS Distributions 		}
1446e7776783SApple OSS Distributions 		break;
1447e7776783SApple OSS Distributions 
1448e7776783SApple OSS Distributions 	case kOFVariableTypeString:
1449e7776783SApple OSS Distributions 		tmpString = OSDynamicCast(OSString, propObject);
1450e7776783SApple OSS Distributions 		if (tmpString != nullptr) {
1451e7776783SApple OSS Distributions 			offset = tmpString->getLength();
1452e7776783SApple OSS Distributions 
1453e7776783SApple OSS Distributions 			if (buffer) {
1454e7776783SApple OSS Distributions 				if (*length < offset) {
1455e7776783SApple OSS Distributions 					return false;
1456e7776783SApple OSS Distributions 				} else {
1457e7776783SApple OSS Distributions 					bcopy(tmpString->getCStringNoCopy(), buffer, offset);
1458e7776783SApple OSS Distributions 				}
1459e7776783SApple OSS Distributions 			}
1460e7776783SApple OSS Distributions 		}
1461e7776783SApple OSS Distributions 		break;
1462e7776783SApple OSS Distributions 
1463e7776783SApple OSS Distributions 	case kOFVariableTypeData:
1464e7776783SApple OSS Distributions 		tmpData = OSDynamicCast(OSData, propObject);
1465e7776783SApple OSS Distributions 		if (tmpData != nullptr) {
1466e7776783SApple OSS Distributions 			offset = tmpData->getLength();
1467e7776783SApple OSS Distributions 
1468e7776783SApple OSS Distributions 			if (buffer) {
1469e7776783SApple OSS Distributions 				if (*length < offset) {
1470e7776783SApple OSS Distributions 					return false;
1471e7776783SApple OSS Distributions 				} else {
1472e7776783SApple OSS Distributions 					bcopy(tmpData->getBytesNoCopy(), buffer, offset);
1473e7776783SApple OSS Distributions 				}
1474e7776783SApple OSS Distributions 			}
1475e7776783SApple OSS Distributions 		}
1476e7776783SApple OSS Distributions 		break;
1477e7776783SApple OSS Distributions 
1478e7776783SApple OSS Distributions 	default:
1479e7776783SApple OSS Distributions 		return false;
1480e7776783SApple OSS Distributions 	}
1481e7776783SApple OSS Distributions 
1482e7776783SApple OSS Distributions 	*length = offset;
1483e7776783SApple OSS Distributions 
1484e7776783SApple OSS Distributions 	return offset != 0;
1485e7776783SApple OSS Distributions }
1486e7776783SApple OSS Distributions 
1487e7776783SApple OSS Distributions 
1488e7776783SApple OSS Distributions bool
convertPropToObject(const uint8_t * propName,uint32_t propNameLength,const uint8_t * propData,uint32_t propDataLength,OSSharedPtr<const OSSymbol> & propSymbol,OSSharedPtr<OSObject> & propObject)1489e7776783SApple OSS Distributions IONVRAMV3Handler::convertPropToObject(const uint8_t *propName, uint32_t propNameLength,
1490e7776783SApple OSS Distributions     const uint8_t *propData, uint32_t propDataLength,
1491e7776783SApple OSS Distributions     OSSharedPtr<const OSSymbol>& propSymbol,
1492e7776783SApple OSS Distributions     OSSharedPtr<OSObject>& propObject)
1493e7776783SApple OSS Distributions {
1494e7776783SApple OSS Distributions 	OSSharedPtr<const OSSymbol> tmpSymbol;
1495e7776783SApple OSS Distributions 	OSSharedPtr<OSNumber>       tmpNumber;
1496e7776783SApple OSS Distributions 	OSSharedPtr<OSString>       tmpString;
1497e7776783SApple OSS Distributions 	OSSharedPtr<OSObject>       tmpObject = nullptr;
1498e7776783SApple OSS Distributions 
1499e7776783SApple OSS Distributions 	tmpSymbol = OSSymbol::withCString((const char *)propName);
1500e7776783SApple OSS Distributions 
1501e7776783SApple OSS Distributions 	if (tmpSymbol == nullptr) {
1502e7776783SApple OSS Distributions 		return false;
1503e7776783SApple OSS Distributions 	}
1504e7776783SApple OSS Distributions 
1505e7776783SApple OSS Distributions 	switch (getVariableType(tmpSymbol.get())) {
1506e7776783SApple OSS Distributions 	case kOFVariableTypeBoolean:
1507e7776783SApple OSS Distributions 		if (!strncmp("true", (const char *)propData, propDataLength)) {
1508e7776783SApple OSS Distributions 			tmpObject.reset(kOSBooleanTrue, OSRetain);
1509e7776783SApple OSS Distributions 		} else if (!strncmp("false", (const char *)propData, propDataLength)) {
1510e7776783SApple OSS Distributions 			tmpObject.reset(kOSBooleanFalse, OSRetain);
1511e7776783SApple OSS Distributions 		}
1512e7776783SApple OSS Distributions 		break;
1513e7776783SApple OSS Distributions 
1514e7776783SApple OSS Distributions 	case kOFVariableTypeNumber:
1515e7776783SApple OSS Distributions 		tmpNumber = OSNumber::withNumber(strtol((const char *)propData, nullptr, 0), 32);
1516e7776783SApple OSS Distributions 		if (tmpNumber != nullptr) {
1517e7776783SApple OSS Distributions 			tmpObject = tmpNumber;
1518e7776783SApple OSS Distributions 		}
1519e7776783SApple OSS Distributions 		break;
1520e7776783SApple OSS Distributions 
1521e7776783SApple OSS Distributions 	case kOFVariableTypeString:
1522e7776783SApple OSS Distributions 		tmpString = OSString::withCString((const char *)propData, propDataLength);
1523e7776783SApple OSS Distributions 		if (tmpString != nullptr) {
1524e7776783SApple OSS Distributions 			tmpObject = tmpString;
1525e7776783SApple OSS Distributions 		}
1526e7776783SApple OSS Distributions 		break;
1527e7776783SApple OSS Distributions 
1528e7776783SApple OSS Distributions 	case kOFVariableTypeData:
1529e7776783SApple OSS Distributions 		tmpObject = OSData::withBytes(propData, propDataLength);
1530e7776783SApple OSS Distributions 		break;
1531e7776783SApple OSS Distributions 
1532e7776783SApple OSS Distributions 	default:
1533e7776783SApple OSS Distributions 		break;
1534e7776783SApple OSS Distributions 	}
1535e7776783SApple OSS Distributions 
1536e7776783SApple OSS Distributions 	if (tmpObject == nullptr) {
1537e7776783SApple OSS Distributions 		tmpSymbol.reset();
1538e7776783SApple OSS Distributions 		return false;
1539e7776783SApple OSS Distributions 	}
1540e7776783SApple OSS Distributions 
1541e7776783SApple OSS Distributions 	propSymbol = tmpSymbol;
1542e7776783SApple OSS Distributions 	propObject = tmpObject;
1543e7776783SApple OSS Distributions 
1544e7776783SApple OSS Distributions 	return true;
1545e7776783SApple OSS Distributions }
1546