xref: /xnu-11215.41.3/iokit/Kernel/IONVRAMV3Handler.cpp (revision 33de042d024d46de5ff4e89f2471de6608e37fa4)
1*33de042dSApple OSS Distributions /*
2*33de042dSApple OSS Distributions  * Copyright (c) 2021-2022 Apple Inc. All rights reserved.
3*33de042dSApple OSS Distributions  *
4*33de042dSApple OSS Distributions  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5*33de042dSApple OSS Distributions  *
6*33de042dSApple OSS Distributions  * This file contains Original Code and/or Modifications of Original Code
7*33de042dSApple OSS Distributions  * as defined in and that are subject to the Apple Public Source License
8*33de042dSApple OSS Distributions  * Version 2.0 (the 'License'). You may not use this file except in
9*33de042dSApple OSS Distributions  * compliance with the License. The rights granted to you under the License
10*33de042dSApple OSS Distributions  * may not be used to create, or enable the creation or redistribution of,
11*33de042dSApple OSS Distributions  * unlawful or unlicensed copies of an Apple operating system, or to
12*33de042dSApple OSS Distributions  * circumvent, violate, or enable the circumvention or violation of, any
13*33de042dSApple OSS Distributions  * terms of an Apple operating system software license agreement.
14*33de042dSApple OSS Distributions  *
15*33de042dSApple OSS Distributions  * Please obtain a copy of the License at
16*33de042dSApple OSS Distributions  * http://www.opensource.apple.com/apsl/ and read it before using this file.
17*33de042dSApple OSS Distributions  *
18*33de042dSApple OSS Distributions  * The Original Code and all software distributed under the License are
19*33de042dSApple OSS Distributions  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20*33de042dSApple OSS Distributions  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21*33de042dSApple OSS Distributions  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22*33de042dSApple OSS Distributions  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23*33de042dSApple OSS Distributions  * Please see the License for the specific language governing rights and
24*33de042dSApple OSS Distributions  * limitations under the License.
25*33de042dSApple OSS Distributions  *
26*33de042dSApple OSS Distributions  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27*33de042dSApple OSS Distributions  */
28*33de042dSApple OSS Distributions 
29*33de042dSApple OSS Distributions #include <libkern/libkern.h>
30*33de042dSApple OSS Distributions 
31*33de042dSApple OSS Distributions #define VARIABLE_STORE_SIGNATURE         'NVV3'
32*33de042dSApple OSS Distributions 
33*33de042dSApple OSS Distributions // Variable Store Version
34*33de042dSApple OSS Distributions #define VARIABLE_STORE_VERSION           0x1
35*33de042dSApple OSS Distributions 
36*33de042dSApple OSS Distributions #define VARIABLE_DATA                    0x55AA
37*33de042dSApple OSS Distributions #define INVALIDATED_VARIABLE_DATA        0x0000
38*33de042dSApple OSS Distributions 
39*33de042dSApple OSS Distributions // Variable State flags
40*33de042dSApple OSS Distributions #define VAR_IN_DELETED_TRANSITION     0xFE  // Variable is in obsolete transistion
41*33de042dSApple OSS Distributions #define VAR_DELETED                   0xFD  // Variable is obsolete
42*33de042dSApple OSS Distributions #define VAR_INACTIVE                  0xFB  // Variable is inactive due to failing CRC
43*33de042dSApple OSS Distributions #define VAR_ADDED                     0x7F  // Variable has been completely added
44*33de042dSApple OSS Distributions 
45*33de042dSApple OSS Distributions // No changes needed on save
46*33de042dSApple OSS Distributions #define VAR_NEW_STATE_NONE            0x01
47*33de042dSApple OSS Distributions // Remove existing entry on save
48*33de042dSApple OSS Distributions #define VAR_NEW_STATE_REMOVE          0x02
49*33de042dSApple OSS Distributions // Add new value on save, mark previous as inactive
50*33de042dSApple OSS Distributions #define VAR_NEW_STATE_APPEND          0x03
51*33de042dSApple OSS Distributions 
52*33de042dSApple OSS Distributions #pragma pack(1)
53*33de042dSApple OSS Distributions struct v3_store_header {
54*33de042dSApple OSS Distributions 	uint32_t     name;
55*33de042dSApple OSS Distributions 	uint32_t     size;
56*33de042dSApple OSS Distributions 	uint32_t     generation;
57*33de042dSApple OSS Distributions 	uint8_t      state;
58*33de042dSApple OSS Distributions 	uint8_t      flags;
59*33de042dSApple OSS Distributions 	uint8_t      version;
60*33de042dSApple OSS Distributions 	uint8_t      reserved1;
61*33de042dSApple OSS Distributions 	uint32_t     system_size;
62*33de042dSApple OSS Distributions 	uint32_t     common_size;
63*33de042dSApple OSS Distributions };
64*33de042dSApple OSS Distributions 
65*33de042dSApple OSS Distributions struct v3_var_header {
66*33de042dSApple OSS Distributions 	uint16_t     startId;
67*33de042dSApple OSS Distributions 	uint8_t      state;
68*33de042dSApple OSS Distributions 	uint8_t      reserved;
69*33de042dSApple OSS Distributions 	uint32_t     attributes;
70*33de042dSApple OSS Distributions 	uint32_t     nameSize;
71*33de042dSApple OSS Distributions 	uint32_t     dataSize;
72*33de042dSApple OSS Distributions 	uuid_t       guid;
73*33de042dSApple OSS Distributions 	uint32_t     crc;
74*33de042dSApple OSS Distributions 	uint8_t      name_data_buf[];
75*33de042dSApple OSS Distributions };
76*33de042dSApple OSS Distributions #pragma pack()
77*33de042dSApple OSS Distributions 
78*33de042dSApple OSS Distributions struct nvram_v3_var_entry {
79*33de042dSApple OSS Distributions 	uint8_t                new_state;
80*33de042dSApple OSS Distributions 	size_t                 existing_offset;
81*33de042dSApple OSS Distributions 	struct v3_var_header   header;
82*33de042dSApple OSS Distributions };
83*33de042dSApple OSS Distributions 
84*33de042dSApple OSS Distributions static size_t
nvram_v3_var_container_size(const struct v3_var_header * header)85*33de042dSApple OSS Distributions nvram_v3_var_container_size(const struct v3_var_header *header)
86*33de042dSApple OSS Distributions {
87*33de042dSApple OSS Distributions 	return sizeof(struct nvram_v3_var_entry) + header->nameSize + header->dataSize;
88*33de042dSApple OSS Distributions }
89*33de042dSApple OSS Distributions 
90*33de042dSApple OSS Distributions static size_t
variable_length(const struct v3_var_header * header)91*33de042dSApple OSS Distributions variable_length(const struct v3_var_header *header)
92*33de042dSApple OSS Distributions {
93*33de042dSApple OSS Distributions 	return sizeof(struct v3_var_header) + header->nameSize + header->dataSize;
94*33de042dSApple OSS Distributions }
95*33de042dSApple OSS Distributions 
96*33de042dSApple OSS Distributions static bool
valid_store_header(const struct v3_store_header * header)97*33de042dSApple OSS Distributions valid_store_header(const struct v3_store_header *header)
98*33de042dSApple OSS Distributions {
99*33de042dSApple OSS Distributions 	return (header->name == VARIABLE_STORE_SIGNATURE) && (header->version == VARIABLE_STORE_VERSION);
100*33de042dSApple OSS Distributions }
101*33de042dSApple OSS Distributions 
102*33de042dSApple OSS Distributions static bool
valid_variable_header(const struct v3_var_header * header,size_t buf_len)103*33de042dSApple OSS Distributions valid_variable_header(const struct v3_var_header *header, size_t buf_len)
104*33de042dSApple OSS Distributions {
105*33de042dSApple OSS Distributions 	return (buf_len > sizeof(struct v3_var_header)) &&
106*33de042dSApple OSS Distributions 	       (header->startId == VARIABLE_DATA) &&
107*33de042dSApple OSS Distributions 	       (variable_length(header) <= buf_len);
108*33de042dSApple OSS Distributions }
109*33de042dSApple OSS Distributions 
110*33de042dSApple 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)111*33de042dSApple OSS Distributions find_active_var_in_image(const struct v3_var_header *var, const uint8_t *image, uint32_t offset, uint32_t len)
112*33de042dSApple OSS Distributions {
113*33de042dSApple OSS Distributions 	const struct v3_var_header *store_var;
114*33de042dSApple OSS Distributions 	uint32_t var_offset = 0;
115*33de042dSApple OSS Distributions 
116*33de042dSApple OSS Distributions 	while ((offset + sizeof(struct v3_var_header) < len)) {
117*33de042dSApple OSS Distributions 		store_var = (const struct v3_var_header *)(image + offset);
118*33de042dSApple OSS Distributions 
119*33de042dSApple OSS Distributions 		if (valid_variable_header(store_var, len - offset)) {
120*33de042dSApple OSS Distributions 			if ((store_var->state == VAR_ADDED) &&
121*33de042dSApple OSS Distributions 			    (uuid_compare(var->guid, store_var->guid) == 0) &&
122*33de042dSApple OSS Distributions 			    (var->nameSize == store_var->nameSize) &&
123*33de042dSApple OSS Distributions 			    (memcmp(var->name_data_buf, store_var->name_data_buf, var->nameSize) == 0)) {
124*33de042dSApple OSS Distributions 				var_offset = offset;
125*33de042dSApple OSS Distributions 				break;
126*33de042dSApple OSS Distributions 			}
127*33de042dSApple OSS Distributions 		} else {
128*33de042dSApple OSS Distributions 			break;
129*33de042dSApple OSS Distributions 		}
130*33de042dSApple OSS Distributions 
131*33de042dSApple OSS Distributions 		offset += variable_length(store_var);
132*33de042dSApple OSS Distributions 	}
133*33de042dSApple OSS Distributions 
134*33de042dSApple OSS Distributions 	return var_offset;
135*33de042dSApple OSS Distributions }
136*33de042dSApple OSS Distributions 
137*33de042dSApple OSS Distributions static IOReturn
find_current_offset_in_image(const uint8_t * image,uint32_t len,uint32_t * newOffset)138*33de042dSApple OSS Distributions find_current_offset_in_image(const uint8_t *image, uint32_t len, uint32_t *newOffset)
139*33de042dSApple OSS Distributions {
140*33de042dSApple OSS Distributions 	uint32_t offset = 0;
141*33de042dSApple OSS Distributions 	uint32_t inner_offset = 0;
142*33de042dSApple OSS Distributions 
143*33de042dSApple OSS Distributions 	if (valid_store_header((const struct v3_store_header *)(image + offset))) {
144*33de042dSApple OSS Distributions 		DEBUG_INFO("valid store header @ %#x\n", offset);
145*33de042dSApple OSS Distributions 		offset += sizeof(struct v3_store_header);
146*33de042dSApple OSS Distributions 	}
147*33de042dSApple OSS Distributions 
148*33de042dSApple OSS Distributions 	while (offset < len) {
149*33de042dSApple OSS Distributions 		const struct v3_var_header *store_var = (const struct v3_var_header *)(image + offset);
150*33de042dSApple OSS Distributions 		uuid_string_t uuidString;
151*33de042dSApple OSS Distributions 
152*33de042dSApple OSS Distributions 		if (valid_variable_header(store_var, len - offset)) {
153*33de042dSApple OSS Distributions 			uuid_unparse(store_var->guid, uuidString);
154*33de042dSApple OSS Distributions 			DEBUG_INFO("Valid var @ %#08x, state=%#02x, length=%#08zx, %s:%s\n", offset, store_var->state,
155*33de042dSApple OSS Distributions 			    variable_length(store_var), uuidString, store_var->name_data_buf);
156*33de042dSApple OSS Distributions 			offset += variable_length(store_var);
157*33de042dSApple OSS Distributions 		} else {
158*33de042dSApple OSS Distributions 			break;
159*33de042dSApple OSS Distributions 		}
160*33de042dSApple OSS Distributions 	}
161*33de042dSApple OSS Distributions 
162*33de042dSApple OSS Distributions 	while (offset < len) {
163*33de042dSApple OSS Distributions 		if (image[offset] == 0xFF) {
164*33de042dSApple OSS Distributions 			DEBUG_INFO("scanning for clear memory @ %#x\n", offset);
165*33de042dSApple OSS Distributions 
166*33de042dSApple OSS Distributions 			inner_offset = offset;
167*33de042dSApple OSS Distributions 
168*33de042dSApple OSS Distributions 			while ((inner_offset < len) && (image[inner_offset] == 0xFF)) {
169*33de042dSApple OSS Distributions 				inner_offset++;
170*33de042dSApple OSS Distributions 			}
171*33de042dSApple OSS Distributions 
172*33de042dSApple OSS Distributions 			if (inner_offset == len) {
173*33de042dSApple OSS Distributions 				DEBUG_INFO("found start of clear mem @ %#x\n", offset);
174*33de042dSApple OSS Distributions 				break;
175*33de042dSApple OSS Distributions 			} else {
176*33de042dSApple OSS Distributions 				DEBUG_ERROR("ERROR!!!!! found non-clear byte @ %#x\n", offset);
177*33de042dSApple OSS Distributions 				return kIOReturnInvalid;
178*33de042dSApple OSS Distributions 			}
179*33de042dSApple OSS Distributions 		}
180*33de042dSApple OSS Distributions 		offset++;
181*33de042dSApple OSS Distributions 	}
182*33de042dSApple OSS Distributions 
183*33de042dSApple OSS Distributions 	*newOffset = offset;
184*33de042dSApple OSS Distributions 
185*33de042dSApple OSS Distributions 	return kIOReturnSuccess;
186*33de042dSApple OSS Distributions }
187*33de042dSApple OSS Distributions 
188*33de042dSApple OSS Distributions class IONVRAMV3Handler : public IODTNVRAMFormatHandler, IOTypedOperatorsMixin<IONVRAMV3Handler>
189*33de042dSApple OSS Distributions {
190*33de042dSApple OSS Distributions private:
191*33de042dSApple OSS Distributions 	IONVRAMController            *_nvramController;
192*33de042dSApple OSS Distributions 	IODTNVRAM                    *_provider;
193*33de042dSApple OSS Distributions 
194*33de042dSApple OSS Distributions 	bool                         _newData;
195*33de042dSApple OSS Distributions 	bool                         _resetData;
196*33de042dSApple OSS Distributions 	bool                         _reload;
197*33de042dSApple OSS Distributions 
198*33de042dSApple OSS Distributions 	bool                         _rawController;
199*33de042dSApple OSS Distributions 
200*33de042dSApple OSS Distributions 	uint32_t                     _generation;
201*33de042dSApple OSS Distributions 
202*33de042dSApple OSS Distributions 	uint8_t                      *_nvramImage;
203*33de042dSApple OSS Distributions 
204*33de042dSApple OSS Distributions 	OSSharedPtr<OSDictionary>    &_varDict;
205*33de042dSApple OSS Distributions 
206*33de042dSApple OSS Distributions 	uint32_t                     _commonSize;
207*33de042dSApple OSS Distributions 	uint32_t                     _systemSize;
208*33de042dSApple OSS Distributions 
209*33de042dSApple OSS Distributions 	uint32_t                     _commonUsed;
210*33de042dSApple OSS Distributions 	uint32_t                     _systemUsed;
211*33de042dSApple OSS Distributions 
212*33de042dSApple OSS Distributions 	uint32_t                     _currentOffset;
213*33de042dSApple OSS Distributions 
214*33de042dSApple OSS Distributions 	OSSharedPtr<OSArray>         _varEntries;
215*33de042dSApple OSS Distributions 
216*33de042dSApple OSS Distributions 	IOReturn unserializeImage(const uint8_t *image, IOByteCount length);
217*33de042dSApple OSS Distributions 	IOReturn reclaim(void);
218*33de042dSApple OSS Distributions 	uint32_t findCurrentBank(void);
219*33de042dSApple OSS Distributions 	size_t   getAppendSize(void);
220*33de042dSApple OSS Distributions 
221*33de042dSApple OSS Distributions 	static bool convertObjectToProp(uint8_t *buffer, uint32_t *length, const char *propSymbol, OSObject *propObject);
222*33de042dSApple OSS Distributions 	static bool convertPropToObject(const uint8_t *propName, uint32_t propNameLength, const uint8_t *propData, uint32_t propDataLength,
223*33de042dSApple OSS Distributions 	    OSSharedPtr<const OSSymbol>& propSymbol, OSSharedPtr<OSObject>& propObject);
224*33de042dSApple OSS Distributions 
225*33de042dSApple OSS Distributions 	IOReturn reloadInternal(void);
226*33de042dSApple OSS Distributions 	IOReturn setVariableInternal(const uuid_t varGuid, const char *variableName, OSObject *object);
227*33de042dSApple OSS Distributions 
228*33de042dSApple OSS Distributions 	void setEntryForRemove(struct nvram_v3_var_entry *v3Entry, bool system);
229*33de042dSApple OSS Distributions 	void findExistingEntry(const uuid_t varGuid, const char *varName, struct nvram_v3_var_entry **existing, unsigned int *existingIndex);
230*33de042dSApple OSS Distributions 	IOReturn syncRaw(void);
231*33de042dSApple OSS Distributions 	IOReturn syncBlock(void);
232*33de042dSApple OSS Distributions 	IOReturn handleEphDM(void);
233*33de042dSApple OSS Distributions 
234*33de042dSApple OSS Distributions public:
235*33de042dSApple OSS Distributions 	virtual
236*33de042dSApple OSS Distributions 	~IONVRAMV3Handler() APPLE_KEXT_OVERRIDE;
237*33de042dSApple OSS Distributions 	IONVRAMV3Handler(OSSharedPtr<OSDictionary> &varDict);
238*33de042dSApple OSS Distributions 
239*33de042dSApple OSS Distributions 	static bool isValidImage(const uint8_t *image, IOByteCount length);
240*33de042dSApple OSS Distributions 
241*33de042dSApple OSS Distributions 	static  IONVRAMV3Handler *init(IODTNVRAM *provider, const uint8_t *image, IOByteCount length,
242*33de042dSApple OSS Distributions 	    OSSharedPtr<OSDictionary> &varDict);
243*33de042dSApple OSS Distributions 
244*33de042dSApple OSS Distributions 	virtual bool     getNVRAMProperties(void) APPLE_KEXT_OVERRIDE;
245*33de042dSApple OSS Distributions 	virtual IOReturn unserializeVariables(void) APPLE_KEXT_OVERRIDE;
246*33de042dSApple OSS Distributions 	virtual IOReturn setVariable(const uuid_t varGuid, const char *variableName, OSObject *object) APPLE_KEXT_OVERRIDE;
247*33de042dSApple OSS Distributions 	virtual bool     setController(IONVRAMController *controller) APPLE_KEXT_OVERRIDE;
248*33de042dSApple OSS Distributions 	virtual IOReturn sync(void) APPLE_KEXT_OVERRIDE;
249*33de042dSApple OSS Distributions 	virtual IOReturn flush(const uuid_t guid, IONVRAMOperation op) APPLE_KEXT_OVERRIDE;
250*33de042dSApple OSS Distributions 	virtual void     reload(void) APPLE_KEXT_OVERRIDE;
251*33de042dSApple OSS Distributions 	virtual uint32_t getGeneration(void) const APPLE_KEXT_OVERRIDE;
252*33de042dSApple OSS Distributions 	virtual uint32_t getVersion(void) const APPLE_KEXT_OVERRIDE;
253*33de042dSApple OSS Distributions 	virtual uint32_t getSystemUsed(void) const APPLE_KEXT_OVERRIDE;
254*33de042dSApple OSS Distributions 	virtual uint32_t getCommonUsed(void) const APPLE_KEXT_OVERRIDE;
255*33de042dSApple OSS Distributions 	virtual bool     getSystemPartitionActive(void) const APPLE_KEXT_OVERRIDE;
256*33de042dSApple OSS Distributions };
257*33de042dSApple OSS Distributions 
~IONVRAMV3Handler()258*33de042dSApple OSS Distributions IONVRAMV3Handler::~IONVRAMV3Handler()
259*33de042dSApple OSS Distributions {
260*33de042dSApple OSS Distributions }
261*33de042dSApple OSS Distributions 
IONVRAMV3Handler(OSSharedPtr<OSDictionary> & varDict)262*33de042dSApple OSS Distributions IONVRAMV3Handler::IONVRAMV3Handler(OSSharedPtr<OSDictionary> &varDict) :
263*33de042dSApple OSS Distributions 	_varDict(varDict)
264*33de042dSApple OSS Distributions {
265*33de042dSApple OSS Distributions }
266*33de042dSApple OSS Distributions 
267*33de042dSApple OSS Distributions bool
isValidImage(const uint8_t * image,IOByteCount length)268*33de042dSApple OSS Distributions IONVRAMV3Handler::isValidImage(const uint8_t *image, IOByteCount length)
269*33de042dSApple OSS Distributions {
270*33de042dSApple OSS Distributions 	const struct v3_store_header *header = (const struct v3_store_header *)image;
271*33de042dSApple OSS Distributions 
272*33de042dSApple OSS Distributions 	if ((header == nullptr) || (length < sizeof(*header))) {
273*33de042dSApple OSS Distributions 		return false;
274*33de042dSApple OSS Distributions 	}
275*33de042dSApple OSS Distributions 
276*33de042dSApple OSS Distributions 	return valid_store_header(header);
277*33de042dSApple OSS Distributions }
278*33de042dSApple OSS Distributions 
279*33de042dSApple OSS Distributions IONVRAMV3Handler*
init(IODTNVRAM * provider,const uint8_t * image,IOByteCount length,OSSharedPtr<OSDictionary> & varDict)280*33de042dSApple OSS Distributions IONVRAMV3Handler::init(IODTNVRAM *provider, const uint8_t *image, IOByteCount length,
281*33de042dSApple OSS Distributions     OSSharedPtr<OSDictionary> &varDict)
282*33de042dSApple OSS Distributions {
283*33de042dSApple OSS Distributions 	OSSharedPtr<IORegistryEntry> entry;
284*33de042dSApple OSS Distributions 	OSSharedPtr<OSObject>        prop;
285*33de042dSApple OSS Distributions 	bool                         propertiesOk;
286*33de042dSApple OSS Distributions 
287*33de042dSApple OSS Distributions 	IONVRAMV3Handler *handler = new IONVRAMV3Handler(varDict);
288*33de042dSApple OSS Distributions 
289*33de042dSApple OSS Distributions 	handler->_provider = provider;
290*33de042dSApple OSS Distributions 
291*33de042dSApple OSS Distributions 	propertiesOk = handler->getNVRAMProperties();
292*33de042dSApple OSS Distributions 	require_action(propertiesOk, exit, DEBUG_ERROR("Unable to get NVRAM properties\n"));
293*33de042dSApple OSS Distributions 
294*33de042dSApple OSS Distributions 	require_action(length == handler->_bankSize, exit, DEBUG_ERROR("length %#llx != _bankSize %#x\n", length, handler->_bankSize));
295*33de042dSApple OSS Distributions 
296*33de042dSApple OSS Distributions 	if ((image != nullptr) && (length != 0)) {
297*33de042dSApple OSS Distributions 		if (handler->unserializeImage(image, length) != kIOReturnSuccess) {
298*33de042dSApple OSS Distributions 			DEBUG_ERROR("Unable to unserialize image, len=%#x\n", (unsigned int)length);
299*33de042dSApple OSS Distributions 		}
300*33de042dSApple OSS Distributions 	}
301*33de042dSApple OSS Distributions 
302*33de042dSApple OSS Distributions 	return handler;
303*33de042dSApple OSS Distributions 
304*33de042dSApple OSS Distributions exit:
305*33de042dSApple OSS Distributions 	delete handler;
306*33de042dSApple OSS Distributions 
307*33de042dSApple OSS Distributions 	return nullptr;
308*33de042dSApple OSS Distributions }
309*33de042dSApple OSS Distributions 
310*33de042dSApple OSS Distributions bool
getNVRAMProperties()311*33de042dSApple OSS Distributions IONVRAMV3Handler::getNVRAMProperties()
312*33de042dSApple OSS Distributions {
313*33de042dSApple OSS Distributions 	bool                         ok    = false;
314*33de042dSApple OSS Distributions 	const char                   *rawControllerKey = "nvram-raw";
315*33de042dSApple OSS Distributions 	OSSharedPtr<IORegistryEntry> entry;
316*33de042dSApple OSS Distributions 	OSSharedPtr<OSObject>        prop;
317*33de042dSApple OSS Distributions 	OSData *                     data;
318*33de042dSApple OSS Distributions 
319*33de042dSApple OSS Distributions 	require_action(IODTNVRAMFormatHandler::getNVRAMProperties(), exit, DEBUG_ERROR("parent getNVRAMProperties failed\n"));
320*33de042dSApple OSS Distributions 
321*33de042dSApple OSS Distributions 	entry = IORegistryEntry::fromPath("/chosen", gIODTPlane);
322*33de042dSApple OSS Distributions 	require_action(entry, exit, DEBUG_ERROR("Unable to find chosen node\n"));
323*33de042dSApple OSS Distributions 
324*33de042dSApple OSS Distributions 	prop = entry->copyProperty(rawControllerKey);
325*33de042dSApple OSS Distributions 	require_action(prop != nullptr, exit, DEBUG_ERROR("No %s entry\n", rawControllerKey));
326*33de042dSApple OSS Distributions 
327*33de042dSApple OSS Distributions 	data = OSDynamicCast(OSData, prop.get());
328*33de042dSApple OSS Distributions 	require(data != nullptr, exit);
329*33de042dSApple OSS Distributions 
330*33de042dSApple OSS Distributions 	_rawController = *((uint32_t*)data->getBytesNoCopy());
331*33de042dSApple OSS Distributions 	DEBUG_INFO("_rawController = %d\n", _rawController);
332*33de042dSApple OSS Distributions 
333*33de042dSApple OSS Distributions 	ok = true;
334*33de042dSApple OSS Distributions 
335*33de042dSApple OSS Distributions exit:
336*33de042dSApple OSS Distributions 	return ok;
337*33de042dSApple OSS Distributions }
338*33de042dSApple OSS Distributions 
339*33de042dSApple OSS Distributions IOReturn
flush(const uuid_t guid,IONVRAMOperation op)340*33de042dSApple OSS Distributions IONVRAMV3Handler::flush(const uuid_t guid, IONVRAMOperation op)
341*33de042dSApple OSS Distributions {
342*33de042dSApple OSS Distributions 	IOReturn ret = kIOReturnSuccess;
343*33de042dSApple OSS Distributions 	bool     flushSystem;
344*33de042dSApple OSS Distributions 	bool     flushCommon;
345*33de042dSApple OSS Distributions 
346*33de042dSApple OSS Distributions 	flushSystem = getSystemPartitionActive() && (uuid_compare(guid, gAppleSystemVariableGuid) == 0);
347*33de042dSApple OSS Distributions 	flushCommon = uuid_compare(guid, gAppleNVRAMGuid) == 0;
348*33de042dSApple OSS Distributions 
349*33de042dSApple OSS Distributions 	DEBUG_INFO("flushSystem=%d, flushCommon=%d\n", flushSystem, flushCommon);
350*33de042dSApple OSS Distributions 
351*33de042dSApple OSS Distributions 	if (flushSystem || flushCommon) {
352*33de042dSApple OSS Distributions 		const OSSymbol                    *canonicalKey;
353*33de042dSApple OSS Distributions 		OSSharedPtr<OSDictionary>         dictCopy;
354*33de042dSApple OSS Distributions 		OSSharedPtr<OSCollectionIterator> iter;
355*33de042dSApple OSS Distributions 		uuid_string_t                     uuidString;
356*33de042dSApple OSS Distributions 
357*33de042dSApple OSS Distributions 		dictCopy = OSDictionary::withDictionary(_varDict.get());
358*33de042dSApple OSS Distributions 		iter = OSCollectionIterator::withCollection(dictCopy.get());
359*33de042dSApple OSS Distributions 		require_action(dictCopy && iter, exit, ret = kIOReturnNoMemory);
360*33de042dSApple OSS Distributions 
361*33de042dSApple OSS Distributions 		while ((canonicalKey = OSDynamicCast(OSSymbol, iter->getNextObject()))) {
362*33de042dSApple OSS Distributions 			const char *varName;
363*33de042dSApple OSS Distributions 			uuid_t     varGuid;
364*33de042dSApple OSS Distributions 			bool       clear;
365*33de042dSApple OSS Distributions 
366*33de042dSApple OSS Distributions 			parseVariableName(canonicalKey->getCStringNoCopy(), &varGuid, &varName);
367*33de042dSApple OSS Distributions 
368*33de042dSApple OSS Distributions 			uuid_unparse(varGuid, uuidString);
369*33de042dSApple OSS Distributions 
370*33de042dSApple OSS Distributions 			clear = ((flushSystem && (uuid_compare(varGuid, gAppleSystemVariableGuid) == 0)) ||
371*33de042dSApple OSS Distributions 			    (flushCommon && (uuid_compare(varGuid, gAppleSystemVariableGuid) != 0))) &&
372*33de042dSApple OSS Distributions 			    verifyPermission(op, varGuid, varName, getSystemPartitionActive());
373*33de042dSApple OSS Distributions 
374*33de042dSApple OSS Distributions 			if (clear) {
375*33de042dSApple OSS Distributions 				DEBUG_INFO("Clearing entry for %s:%s\n", uuidString, varName);
376*33de042dSApple OSS Distributions 				setVariableInternal(varGuid, varName, nullptr);
377*33de042dSApple OSS Distributions 			} else {
378*33de042dSApple OSS Distributions 				DEBUG_INFO("Keeping entry for %s:%s\n", uuidString, varName);
379*33de042dSApple OSS Distributions 			}
380*33de042dSApple OSS Distributions 		}
381*33de042dSApple OSS Distributions 
382*33de042dSApple OSS Distributions 		_newData = true;
383*33de042dSApple OSS Distributions 	}
384*33de042dSApple OSS Distributions 
385*33de042dSApple OSS Distributions 	DEBUG_INFO("_commonUsed %#x, _systemUsed %#x\n", _commonUsed, _systemUsed);
386*33de042dSApple OSS Distributions 
387*33de042dSApple OSS Distributions exit:
388*33de042dSApple OSS Distributions 	return ret;
389*33de042dSApple OSS Distributions }
390*33de042dSApple OSS Distributions 
391*33de042dSApple OSS Distributions IOReturn
reloadInternal(void)392*33de042dSApple OSS Distributions IONVRAMV3Handler::reloadInternal(void)
393*33de042dSApple OSS Distributions {
394*33de042dSApple OSS Distributions 	IOReturn                     ret;
395*33de042dSApple OSS Distributions 	uint32_t                     controllerBank;
396*33de042dSApple OSS Distributions 	uint8_t                      *controllerImage;
397*33de042dSApple OSS Distributions 	struct nvram_v3_var_entry    *v3Entry;
398*33de042dSApple OSS Distributions 	const struct v3_store_header *storeHeader;
399*33de042dSApple OSS Distributions 	const struct v3_var_header   *storeVar;
400*33de042dSApple OSS Distributions 	OSData                       *entryContainer;
401*33de042dSApple OSS Distributions 
402*33de042dSApple OSS Distributions 	controllerBank = findCurrentBank();
403*33de042dSApple OSS Distributions 
404*33de042dSApple OSS Distributions 	if (_currentBank != controllerBank) {
405*33de042dSApple OSS Distributions 		DEBUG_ERROR("_currentBank %#x != controllerBank %#x", _currentBank, controllerBank);
406*33de042dSApple OSS Distributions 	}
407*33de042dSApple OSS Distributions 
408*33de042dSApple OSS Distributions 	_currentBank = controllerBank;
409*33de042dSApple OSS Distributions 
410*33de042dSApple OSS Distributions 	controllerImage = (uint8_t *)IOMallocData(_bankSize);
411*33de042dSApple OSS Distributions 
412*33de042dSApple OSS Distributions 	_nvramController->select(_currentBank);
413*33de042dSApple OSS Distributions 	_nvramController->read(0, controllerImage, _bankSize);
414*33de042dSApple OSS Distributions 
415*33de042dSApple OSS Distributions 	require_action(isValidImage(controllerImage, _bankSize), exit,
416*33de042dSApple OSS Distributions 	    (ret = kIOReturnInvalid, DEBUG_ERROR("Invalid image at bank %d\n", _currentBank)));
417*33de042dSApple OSS Distributions 
418*33de042dSApple OSS Distributions 	DEBUG_INFO("valid image found\n");
419*33de042dSApple OSS Distributions 
420*33de042dSApple OSS Distributions 	storeHeader = (const struct v3_store_header *)controllerImage;
421*33de042dSApple OSS Distributions 
422*33de042dSApple OSS Distributions 	_generation = storeHeader->generation;
423*33de042dSApple OSS Distributions 
424*33de042dSApple OSS Distributions 	// We must sync any existing variables offset on the controller image with our internal representation
425*33de042dSApple OSS Distributions 	// If we find an existing entry and the data is still the same we record the existing offset and mark it
426*33de042dSApple OSS Distributions 	// as VAR_NEW_STATE_NONE meaning no action needed
427*33de042dSApple 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
428*33de042dSApple OSS Distributions 	// which will have us invalidate the existing entry if there is one and append it on the next save
429*33de042dSApple OSS Distributions 	for (unsigned int i = 0; i < _varEntries->getCount(); i++) {
430*33de042dSApple OSS Distributions 		uint32_t offset = sizeof(struct v3_store_header);
431*33de042dSApple OSS Distributions 		uint32_t latestOffset;
432*33de042dSApple OSS Distributions 		uint32_t prevOffset = 0;
433*33de042dSApple OSS Distributions 
434*33de042dSApple OSS Distributions 		entryContainer = (OSDynamicCast(OSData, _varEntries->getObject(i)));
435*33de042dSApple OSS Distributions 		v3Entry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
436*33de042dSApple OSS Distributions 
437*33de042dSApple OSS Distributions 		DEBUG_INFO("Looking for %s\n", v3Entry->header.name_data_buf);
438*33de042dSApple OSS Distributions 		while ((latestOffset = find_active_var_in_image(&v3Entry->header, controllerImage, offset, _bankSize))) {
439*33de042dSApple OSS Distributions 			DEBUG_INFO("Found offset for %s @ %#08x\n", v3Entry->header.name_data_buf, latestOffset);
440*33de042dSApple OSS Distributions 			if (prevOffset) {
441*33de042dSApple OSS Distributions 				DEBUG_INFO("Marking prev offset for %s at %#08x invalid\n", v3Entry->header.name_data_buf, offset);
442*33de042dSApple OSS Distributions 				// Invalidate any previous duplicate entries in the store
443*33de042dSApple OSS Distributions 				struct v3_var_header *prevVarHeader = (struct v3_var_header *)(controllerImage + prevOffset);
444*33de042dSApple OSS Distributions 				uint8_t state = prevVarHeader->state & VAR_DELETED & VAR_IN_DELETED_TRANSITION;
445*33de042dSApple OSS Distributions 
446*33de042dSApple OSS Distributions 				ret = _nvramController->write(prevOffset + offsetof(struct v3_var_header, state), &state, sizeof(state));
447*33de042dSApple OSS Distributions 				require_noerr_action(ret, exit, DEBUG_ERROR("existing state w fail, ret=%#x\n", ret));
448*33de042dSApple OSS Distributions 			}
449*33de042dSApple OSS Distributions 
450*33de042dSApple OSS Distributions 			prevOffset = latestOffset;
451*33de042dSApple OSS Distributions 			offset += latestOffset;
452*33de042dSApple OSS Distributions 		}
453*33de042dSApple OSS Distributions 
454*33de042dSApple OSS Distributions 		v3Entry->existing_offset = latestOffset ? latestOffset : prevOffset;
455*33de042dSApple OSS Distributions 		DEBUG_INFO("Existing offset for %s at %#08zx\n", v3Entry->header.name_data_buf, v3Entry->existing_offset);
456*33de042dSApple OSS Distributions 
457*33de042dSApple OSS Distributions 		if (v3Entry->existing_offset == 0) {
458*33de042dSApple OSS Distributions 			DEBUG_ERROR("%s is not in the NOR image\n", v3Entry->header.name_data_buf);
459*33de042dSApple OSS Distributions 			if (v3Entry->new_state != VAR_NEW_STATE_REMOVE) {
460*33de042dSApple OSS Distributions 				DEBUG_INFO("%s marked for append\n", v3Entry->header.name_data_buf);
461*33de042dSApple OSS Distributions 				// Doesn't exist in the store, just append it on next sync
462*33de042dSApple OSS Distributions 				v3Entry->new_state = VAR_NEW_STATE_APPEND;
463*33de042dSApple OSS Distributions 			}
464*33de042dSApple OSS Distributions 		} else {
465*33de042dSApple OSS Distributions 			DEBUG_INFO("Found offset for %s @ %#zx\n", v3Entry->header.name_data_buf, v3Entry->existing_offset);
466*33de042dSApple OSS Distributions 			storeVar = (const struct v3_var_header *)&controllerImage[v3Entry->existing_offset];
467*33de042dSApple OSS Distributions 
468*33de042dSApple OSS Distributions 			if (v3Entry->new_state != VAR_NEW_STATE_REMOVE) {
469*33de042dSApple OSS Distributions 				// Verify that the existing data matches the store data
470*33de042dSApple OSS Distributions 				if ((variable_length(&v3Entry->header) == variable_length(storeVar)) &&
471*33de042dSApple OSS Distributions 				    (memcmp(v3Entry->header.name_data_buf, storeVar->name_data_buf, storeVar->nameSize + storeVar->dataSize) == 0)) {
472*33de042dSApple OSS Distributions 					DEBUG_INFO("Store var data for %s matches, marking new state none\n", v3Entry->header.name_data_buf);
473*33de042dSApple OSS Distributions 					v3Entry->new_state = VAR_NEW_STATE_NONE;
474*33de042dSApple OSS Distributions 				} else {
475*33de042dSApple OSS Distributions 					DEBUG_INFO("Store var data for %s differs, marking new state append\n", v3Entry->header.name_data_buf);
476*33de042dSApple OSS Distributions 					v3Entry->new_state = VAR_NEW_STATE_APPEND;
477*33de042dSApple OSS Distributions 				}
478*33de042dSApple OSS Distributions 			} else {
479*33de042dSApple OSS Distributions 				// Store has entry but it has been removed from our collection, keep it marked for delete but with updated
480*33de042dSApple OSS Distributions 				// existing_offset for coherence
481*33de042dSApple OSS Distributions 				DEBUG_INFO("Removing entry at %#08zx with next sync\n", v3Entry->existing_offset);
482*33de042dSApple OSS Distributions 			}
483*33de042dSApple OSS Distributions 		}
484*33de042dSApple OSS Distributions 	}
485*33de042dSApple OSS Distributions 
486*33de042dSApple OSS Distributions 	ret = find_current_offset_in_image(controllerImage, _bankSize, &_currentOffset);
487*33de042dSApple OSS Distributions 	if (ret != kIOReturnSuccess) {
488*33de042dSApple OSS Distributions 		DEBUG_ERROR("Unidentified bytes in image, reclaiming\n");
489*33de042dSApple OSS Distributions 		ret = reclaim();
490*33de042dSApple OSS Distributions 		require_noerr_action(ret, exit, DEBUG_ERROR("Reclaim byte recovery failed, invalid controller state!!! ret=%#x\n", ret));
491*33de042dSApple OSS Distributions 	}
492*33de042dSApple OSS Distributions 	DEBUG_INFO("New _currentOffset=%#x\n", _currentOffset);
493*33de042dSApple OSS Distributions 
494*33de042dSApple OSS Distributions exit:
495*33de042dSApple OSS Distributions 	IOFreeData(controllerImage, _bankSize);
496*33de042dSApple OSS Distributions 	return ret;
497*33de042dSApple OSS Distributions }
498*33de042dSApple OSS Distributions 
499*33de042dSApple OSS Distributions void
reload(void)500*33de042dSApple OSS Distributions IONVRAMV3Handler::reload(void)
501*33de042dSApple OSS Distributions {
502*33de042dSApple OSS Distributions 	_reload = true;
503*33de042dSApple OSS Distributions 
504*33de042dSApple OSS Distributions 	DEBUG_INFO("reload marked\n");
505*33de042dSApple OSS Distributions }
506*33de042dSApple OSS Distributions 
507*33de042dSApple OSS Distributions void
setEntryForRemove(struct nvram_v3_var_entry * v3Entry,bool system)508*33de042dSApple OSS Distributions IONVRAMV3Handler::setEntryForRemove(struct nvram_v3_var_entry *v3Entry, bool system)
509*33de042dSApple OSS Distributions {
510*33de042dSApple OSS Distributions 	OSSharedPtr<const OSSymbol> canonicalKey;
511*33de042dSApple OSS Distributions 	const char                  *variableName;
512*33de042dSApple OSS Distributions 	uint32_t                    variableSize;
513*33de042dSApple OSS Distributions 
514*33de042dSApple OSS Distributions 	require_action(v3Entry != nullptr, exit, DEBUG_INFO("remove with no entry\n"));
515*33de042dSApple OSS Distributions 
516*33de042dSApple OSS Distributions 	variableName = (const char *)v3Entry->header.name_data_buf;
517*33de042dSApple OSS Distributions 	variableSize = (uint32_t)variable_length(&v3Entry->header);
518*33de042dSApple OSS Distributions 	canonicalKey = keyWithGuidAndCString(v3Entry->header.guid, variableName);
519*33de042dSApple OSS Distributions 
520*33de042dSApple OSS Distributions 	if (v3Entry->new_state == VAR_NEW_STATE_REMOVE) {
521*33de042dSApple OSS Distributions 		DEBUG_INFO("entry %s already marked for remove\n", variableName);
522*33de042dSApple OSS Distributions 	} else {
523*33de042dSApple OSS Distributions 		DEBUG_INFO("marking entry %s for remove\n", variableName);
524*33de042dSApple OSS Distributions 
525*33de042dSApple OSS Distributions 		v3Entry->new_state = VAR_NEW_STATE_REMOVE;
526*33de042dSApple OSS Distributions 
527*33de042dSApple OSS Distributions 		_provider->_varDict->removeObject(canonicalKey.get());
528*33de042dSApple OSS Distributions 
529*33de042dSApple OSS Distributions 		if (system) {
530*33de042dSApple OSS Distributions 			if (_systemUsed < variableSize) {
531*33de042dSApple OSS Distributions 				panic("Invalid _systemUsed size\n");
532*33de042dSApple OSS Distributions 			}
533*33de042dSApple OSS Distributions 			_systemUsed -= variableSize;
534*33de042dSApple OSS Distributions 		} else {
535*33de042dSApple OSS Distributions 			if (_commonUsed < variableSize) {
536*33de042dSApple OSS Distributions 				panic("Invalid _commonUsed size\n");
537*33de042dSApple OSS Distributions 			}
538*33de042dSApple OSS Distributions 			_commonUsed -= variableSize;
539*33de042dSApple OSS Distributions 		}
540*33de042dSApple OSS Distributions 
541*33de042dSApple OSS Distributions 		if (_provider->_diags) {
542*33de042dSApple OSS Distributions 			_provider->_diags->logVariable(getPartitionTypeForGUID(v3Entry->header.guid),
543*33de042dSApple OSS Distributions 			    kIONVRAMOperationDelete,
544*33de042dSApple OSS Distributions 			    variableName,
545*33de042dSApple OSS Distributions 			    nullptr);
546*33de042dSApple OSS Distributions 		}
547*33de042dSApple OSS Distributions 	}
548*33de042dSApple OSS Distributions 
549*33de042dSApple OSS Distributions exit:
550*33de042dSApple OSS Distributions 	return;
551*33de042dSApple OSS Distributions }
552*33de042dSApple OSS Distributions 
553*33de042dSApple OSS Distributions void
findExistingEntry(const uuid_t varGuid,const char * varName,struct nvram_v3_var_entry ** existing,unsigned int * existingIndex)554*33de042dSApple OSS Distributions IONVRAMV3Handler::findExistingEntry(const uuid_t varGuid, const char *varName, struct nvram_v3_var_entry **existing, unsigned int *existingIndex)
555*33de042dSApple OSS Distributions {
556*33de042dSApple OSS Distributions 	struct nvram_v3_var_entry *v3Entry = nullptr;
557*33de042dSApple OSS Distributions 	OSData                    *entryContainer = nullptr;
558*33de042dSApple OSS Distributions 	unsigned int              index = 0;
559*33de042dSApple OSS Distributions 	uint32_t                  nameLen = (uint32_t)strlen(varName) + 1;
560*33de042dSApple OSS Distributions 
561*33de042dSApple OSS Distributions 	for (index = 0; index < _varEntries->getCount(); index++) {
562*33de042dSApple OSS Distributions 		entryContainer = (OSDynamicCast(OSData, _varEntries->getObject(index)));
563*33de042dSApple OSS Distributions 		v3Entry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
564*33de042dSApple OSS Distributions 
565*33de042dSApple OSS Distributions 		if ((v3Entry->header.nameSize == nameLen) &&
566*33de042dSApple OSS Distributions 		    (memcmp(v3Entry->header.name_data_buf, varName, nameLen) == 0)) {
567*33de042dSApple OSS Distributions 			if (varGuid) {
568*33de042dSApple OSS Distributions 				if (uuid_compare(varGuid, v3Entry->header.guid) == 0) {
569*33de042dSApple OSS Distributions 					uuid_string_t uuidString;
570*33de042dSApple OSS Distributions 					uuid_unparse(varGuid, uuidString);
571*33de042dSApple OSS Distributions 					DEBUG_INFO("found existing entry for %s:%s, e_off=%#lx, len=%#lx, new_state=%#x\n", uuidString, varName,
572*33de042dSApple OSS Distributions 					    v3Entry->existing_offset, variable_length(&v3Entry->header), v3Entry->new_state);
573*33de042dSApple OSS Distributions 					break;
574*33de042dSApple OSS Distributions 				}
575*33de042dSApple OSS Distributions 			} else {
576*33de042dSApple OSS Distributions 				DEBUG_INFO("found existing entry for %s, e_off=%#lx, len=%#lx\n", varName, v3Entry->existing_offset, variable_length(&v3Entry->header));
577*33de042dSApple OSS Distributions 				break;
578*33de042dSApple OSS Distributions 			}
579*33de042dSApple OSS Distributions 		}
580*33de042dSApple OSS Distributions 
581*33de042dSApple OSS Distributions 		v3Entry = nullptr;
582*33de042dSApple OSS Distributions 	}
583*33de042dSApple OSS Distributions 
584*33de042dSApple OSS Distributions 	if (v3Entry != nullptr) {
585*33de042dSApple OSS Distributions 		if (existing) {
586*33de042dSApple OSS Distributions 			*existing = v3Entry;
587*33de042dSApple OSS Distributions 		}
588*33de042dSApple OSS Distributions 
589*33de042dSApple OSS Distributions 		if (existingIndex) {
590*33de042dSApple OSS Distributions 			*existingIndex = index;
591*33de042dSApple OSS Distributions 		}
592*33de042dSApple OSS Distributions 	}
593*33de042dSApple OSS Distributions }
594*33de042dSApple OSS Distributions 
595*33de042dSApple OSS Distributions IOReturn
unserializeImage(const uint8_t * image,IOByteCount length)596*33de042dSApple OSS Distributions IONVRAMV3Handler::unserializeImage(const uint8_t *image, IOByteCount length)
597*33de042dSApple OSS Distributions {
598*33de042dSApple OSS Distributions 	IOReturn                     ret = kIOReturnInvalid;
599*33de042dSApple OSS Distributions 	const struct v3_store_header *storeHeader;
600*33de042dSApple OSS Distributions 
601*33de042dSApple OSS Distributions 	require(isValidImage(image, length), exit);
602*33de042dSApple OSS Distributions 
603*33de042dSApple OSS Distributions 	storeHeader = (const struct v3_store_header *)image;
604*33de042dSApple OSS Distributions 	require_action(storeHeader->size == (uint32_t)length, exit,
605*33de042dSApple OSS Distributions 	    DEBUG_ERROR("Image size %#x != header size %#x\n", (unsigned int)length, storeHeader->size));
606*33de042dSApple OSS Distributions 
607*33de042dSApple OSS Distributions 	_generation = storeHeader->generation;
608*33de042dSApple OSS Distributions 	_systemSize = storeHeader->system_size;
609*33de042dSApple OSS Distributions 	_commonSize = storeHeader->common_size - sizeof(struct v3_store_header);
610*33de042dSApple OSS Distributions 
611*33de042dSApple OSS Distributions 	_systemUsed = 0;
612*33de042dSApple OSS Distributions 	_commonUsed = 0;
613*33de042dSApple OSS Distributions 
614*33de042dSApple OSS Distributions 	if (_nvramImage) {
615*33de042dSApple OSS Distributions 		IOFreeData(_nvramImage, _bankSize);
616*33de042dSApple OSS Distributions 	}
617*33de042dSApple OSS Distributions 
618*33de042dSApple OSS Distributions 	_varEntries.reset();
619*33de042dSApple OSS Distributions 	_varEntries = OSArray::withCapacity(40);
620*33de042dSApple OSS Distributions 
621*33de042dSApple OSS Distributions 	_nvramImage = IONewData(uint8_t, length);
622*33de042dSApple OSS Distributions 	_bankSize = (uint32_t)length;
623*33de042dSApple OSS Distributions 	bcopy(image, _nvramImage, _bankSize);
624*33de042dSApple OSS Distributions 
625*33de042dSApple OSS Distributions 	ret = kIOReturnSuccess;
626*33de042dSApple OSS Distributions 
627*33de042dSApple OSS Distributions exit:
628*33de042dSApple OSS Distributions 	return ret;
629*33de042dSApple OSS Distributions }
630*33de042dSApple OSS Distributions 
631*33de042dSApple OSS Distributions typedef struct {
632*33de042dSApple OSS Distributions 	const char            *name;
633*33de042dSApple OSS Distributions 	OSSharedPtr<OSObject> value;
634*33de042dSApple OSS Distributions } ephDMAllowListEntry;
635*33de042dSApple OSS Distributions 
636*33de042dSApple OSS Distributions static
637*33de042dSApple OSS Distributions ephDMAllowListEntry ephDMEntries[] = {
638*33de042dSApple OSS Distributions 	// Mobile Obliteration clears the following variables after it runs
639*33de042dSApple OSS Distributions 	{ .name = "oblit-begins" },
640*33de042dSApple OSS Distributions 	{ .name = "orig-oblit" },
641*33de042dSApple OSS Distributions 	{ .name = "oblit-failure" },
642*33de042dSApple OSS Distributions 	{ .name = "oblit-inprogress" },
643*33de042dSApple OSS Distributions 	{ .name = "obliteration" },
644*33de042dSApple OSS Distributions 	// darwin-init is used for configuring internal builds
645*33de042dSApple OSS Distributions 	{ .name = "darwin-init" }
646*33de042dSApple OSS Distributions };
647*33de042dSApple OSS Distributions 
648*33de042dSApple OSS Distributions IOReturn
handleEphDM(void)649*33de042dSApple OSS Distributions IONVRAMV3Handler::handleEphDM(void)
650*33de042dSApple OSS Distributions {
651*33de042dSApple OSS Distributions 	OSSharedPtr<IORegistryEntry> entry;
652*33de042dSApple OSS Distributions 	OSData*                      data;
653*33de042dSApple OSS Distributions 	OSSharedPtr<OSObject>        prop;
654*33de042dSApple OSS Distributions 	uint32_t                     ephDM = 0;
655*33de042dSApple OSS Distributions 	IOReturn                     ret = kIOReturnSuccess;
656*33de042dSApple OSS Distributions 	OSSharedPtr<const OSSymbol>  canonicalKey;
657*33de042dSApple OSS Distributions 	uint32_t                     skip = 0;
658*33de042dSApple OSS Distributions 
659*33de042dSApple OSS Distributions 	// For ephemeral data mode, NVRAM needs to be cleared on every boot
660*33de042dSApple OSS Distributions 	// For system region supported targets, iBoot clears the system region
661*33de042dSApple OSS Distributions 	// For other targets, iBoot clears all the persistent variables
662*33de042dSApple OSS Distributions 	// So xnu only needs to clear the common region
663*33de042dSApple OSS Distributions 	entry = IORegistryEntry::fromPath("/product", gIODTPlane);
664*33de042dSApple OSS Distributions 	if (entry) {
665*33de042dSApple OSS Distributions 		prop = entry->copyProperty("ephemeral-data-mode");
666*33de042dSApple OSS Distributions 		if (prop) {
667*33de042dSApple OSS Distributions 			data = OSDynamicCast(OSData, prop.get());
668*33de042dSApple OSS Distributions 			if (data) {
669*33de042dSApple OSS Distributions 				ephDM = *((uint32_t *)data->getBytesNoCopy());
670*33de042dSApple OSS Distributions 			}
671*33de042dSApple OSS Distributions 		}
672*33de042dSApple OSS Distributions 	}
673*33de042dSApple OSS Distributions 
674*33de042dSApple OSS Distributions 	require_action(ephDM != 0, exit, DEBUG_ALWAYS("ephemeral-data-mode not supported\n"));
675*33de042dSApple OSS Distributions 	require_action(_systemSize != 0, exit, DEBUG_ALWAYS("No system region, no need to clear\n"));
676*33de042dSApple OSS Distributions 
677*33de042dSApple OSS Distributions 	if (PE_parse_boot_argn("epdm-skip-nvram", &skip, sizeof(skip))) {
678*33de042dSApple OSS Distributions 		require_action(!(gInternalBuild && (skip == 1)), exit, DEBUG_ALWAYS("Internal build + epdm-skip-nvram set to true, skip nvram clearing\n"));
679*33de042dSApple OSS Distributions 	}
680*33de042dSApple OSS Distributions 
681*33de042dSApple OSS Distributions 	// Go through the allowlist and stash the values
682*33de042dSApple OSS Distributions 	for (uint32_t entry = 0; entry < ARRAY_SIZE(ephDMEntries); entry++) {
683*33de042dSApple OSS Distributions 		canonicalKey = keyWithGuidAndCString(gAppleNVRAMGuid, ephDMEntries[entry].name);
684*33de042dSApple OSS Distributions 		ephDMEntries[entry].value.reset(OSDynamicCast(OSData, _varDict->getObject(canonicalKey.get())), OSRetain);
685*33de042dSApple OSS Distributions 	}
686*33de042dSApple OSS Distributions 
687*33de042dSApple OSS Distributions 	DEBUG_ALWAYS("Obliterating common region\n");
688*33de042dSApple OSS Distributions 	ret = flush(gAppleNVRAMGuid, kIONVRAMOperationObliterate);
689*33de042dSApple OSS Distributions 	require_noerr_action(ret, exit, DEBUG_ERROR("Flushing common region failed, ret=%#08x\n", ret));
690*33de042dSApple OSS Distributions 
691*33de042dSApple OSS Distributions 	// Now write the allowlist variables back
692*33de042dSApple OSS Distributions 	for (uint32_t entry = 0; entry < ARRAY_SIZE(ephDMEntries); entry++) {
693*33de042dSApple OSS Distributions 		if (ephDMEntries[entry].value.get() == nullptr) {
694*33de042dSApple OSS Distributions 			continue;
695*33de042dSApple OSS Distributions 		}
696*33de042dSApple OSS Distributions 		ret = setVariableInternal(gAppleNVRAMGuid, ephDMEntries[entry].name, ephDMEntries[entry].value.get());
697*33de042dSApple OSS Distributions 		require_noerr_action(ret, exit, DEBUG_ERROR("Setting allowlist variable %s failed, ret=%#08x\n", ephDMEntries[entry].name, ret));
698*33de042dSApple OSS Distributions 	}
699*33de042dSApple OSS Distributions 
700*33de042dSApple OSS Distributions exit:
701*33de042dSApple OSS Distributions 	return ret;
702*33de042dSApple OSS Distributions }
703*33de042dSApple OSS Distributions 
704*33de042dSApple OSS Distributions IOReturn
unserializeVariables(void)705*33de042dSApple OSS Distributions IONVRAMV3Handler::unserializeVariables(void)
706*33de042dSApple OSS Distributions {
707*33de042dSApple OSS Distributions 	IOReturn                     ret = kIOReturnSuccess;
708*33de042dSApple OSS Distributions 	OSSharedPtr<const OSSymbol>  propSymbol;
709*33de042dSApple OSS Distributions 	OSSharedPtr<OSObject>        propObject;
710*33de042dSApple OSS Distributions 	OSSharedPtr<OSData>          entryContainer;
711*33de042dSApple OSS Distributions 	struct nvram_v3_var_entry    *v3Entry;
712*33de042dSApple OSS Distributions 	const struct v3_var_header   *header;
713*33de042dSApple OSS Distributions 	size_t                       offset = sizeof(struct v3_store_header);
714*33de042dSApple OSS Distributions 	uint32_t                     crc;
715*33de042dSApple OSS Distributions 	unsigned int                 i;
716*33de042dSApple OSS Distributions 	bool                         system;
717*33de042dSApple OSS Distributions 	uuid_string_t                uuidString;
718*33de042dSApple OSS Distributions 	size_t                       existingSize;
719*33de042dSApple OSS Distributions 
720*33de042dSApple OSS Distributions 	if (_systemSize || _commonSize) {
721*33de042dSApple OSS Distributions 		_varDict = OSDictionary::withCapacity(1);
722*33de042dSApple OSS Distributions 	}
723*33de042dSApple OSS Distributions 
724*33de042dSApple OSS Distributions 	while ((offset + sizeof(struct v3_var_header)) < _bankSize) {
725*33de042dSApple OSS Distributions 		struct nvram_v3_var_entry *existingEntry = nullptr;
726*33de042dSApple OSS Distributions 		unsigned int              existingIndex = 0;
727*33de042dSApple OSS Distributions 
728*33de042dSApple OSS Distributions 		header = (const struct v3_var_header *)(_nvramImage + offset);
729*33de042dSApple OSS Distributions 
730*33de042dSApple OSS Distributions 		for (i = 0; i < sizeof(struct v3_var_header); i++) {
731*33de042dSApple OSS Distributions 			if ((_nvramImage[offset + i] != 0) && (_nvramImage[offset + i] != 0xFF)) {
732*33de042dSApple OSS Distributions 				break;
733*33de042dSApple OSS Distributions 			}
734*33de042dSApple OSS Distributions 		}
735*33de042dSApple OSS Distributions 
736*33de042dSApple OSS Distributions 		if (i == sizeof(struct v3_var_header)) {
737*33de042dSApple OSS Distributions 			DEBUG_INFO("No more variables after offset %#lx\n", offset);
738*33de042dSApple OSS Distributions 			break;
739*33de042dSApple OSS Distributions 		}
740*33de042dSApple OSS Distributions 
741*33de042dSApple OSS Distributions 		if (!valid_variable_header(header, _bankSize - offset)) {
742*33de042dSApple OSS Distributions 			DEBUG_ERROR("invalid header @ %#lx\n", offset);
743*33de042dSApple OSS Distributions 			offset += sizeof(struct v3_var_header);
744*33de042dSApple OSS Distributions 			continue;
745*33de042dSApple OSS Distributions 		}
746*33de042dSApple OSS Distributions 
747*33de042dSApple OSS Distributions 		uuid_unparse(header->guid, uuidString);
748*33de042dSApple OSS Distributions 		DEBUG_INFO("Valid var @ %#08zx, state=%#02x, length=%#08zx, %s:%s\n", offset, header->state,
749*33de042dSApple OSS Distributions 		    variable_length(header), uuidString, header->name_data_buf);
750*33de042dSApple OSS Distributions 
751*33de042dSApple OSS Distributions 		if (header->state != VAR_ADDED) {
752*33de042dSApple OSS Distributions 			goto skip;
753*33de042dSApple OSS Distributions 		}
754*33de042dSApple OSS Distributions 
755*33de042dSApple OSS Distributions 		crc = crc32(0, header->name_data_buf + header->nameSize, header->dataSize);
756*33de042dSApple OSS Distributions 
757*33de042dSApple OSS Distributions 		if (crc != header->crc) {
758*33de042dSApple OSS Distributions 			DEBUG_ERROR("invalid crc @ %#lx, calculated=%#x, read=%#x\n", offset, crc, header->crc);
759*33de042dSApple OSS Distributions 			goto skip;
760*33de042dSApple OSS Distributions 		}
761*33de042dSApple OSS Distributions 
762*33de042dSApple OSS Distributions 		v3Entry = (struct nvram_v3_var_entry *)IOMallocZeroData(nvram_v3_var_container_size(header));
763*33de042dSApple OSS Distributions 		__nochk_memcpy(&v3Entry->header, _nvramImage + offset, variable_length(header));
764*33de042dSApple OSS Distributions 
765*33de042dSApple 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
766*33de042dSApple OSS Distributions 		// read from the controller, which for various reasons due to the setting of states and saves from iBoot, can be
767*33de042dSApple OSS Distributions 		// different. We will have an initial existing_offset of 0 and once the controller is set we will read
768*33de042dSApple OSS Distributions 		// out the image there and update the existing offset with what is present on the NOR image
769*33de042dSApple OSS Distributions 		v3Entry->existing_offset = 0;
770*33de042dSApple OSS Distributions 		v3Entry->new_state = VAR_NEW_STATE_NONE;
771*33de042dSApple OSS Distributions 
772*33de042dSApple OSS Distributions 		// safe guard for any strange duplicate entries in the store
773*33de042dSApple OSS Distributions 		findExistingEntry(v3Entry->header.guid, (const char *)v3Entry->header.name_data_buf, &existingEntry, &existingIndex);
774*33de042dSApple OSS Distributions 
775*33de042dSApple OSS Distributions 		if (existingEntry != nullptr) {
776*33de042dSApple OSS Distributions 			existingSize = variable_length(&existingEntry->header);
777*33de042dSApple OSS Distributions 
778*33de042dSApple OSS Distributions 			entryContainer = OSData::withBytes(v3Entry, (uint32_t)nvram_v3_var_container_size(header));
779*33de042dSApple OSS Distributions 			_varEntries->replaceObject(existingIndex, entryContainer.get());
780*33de042dSApple OSS Distributions 
781*33de042dSApple OSS Distributions 			DEBUG_INFO("Found existing for %s, resetting when controller available\n", v3Entry->header.name_data_buf);
782*33de042dSApple OSS Distributions 			_resetData = true;
783*33de042dSApple OSS Distributions 		} else {
784*33de042dSApple OSS Distributions 			entryContainer = OSData::withBytes(v3Entry, (uint32_t)nvram_v3_var_container_size(header));
785*33de042dSApple OSS Distributions 			_varEntries->setObject(entryContainer.get());
786*33de042dSApple OSS Distributions 			existingSize = 0;
787*33de042dSApple OSS Distributions 		}
788*33de042dSApple OSS Distributions 
789*33de042dSApple OSS Distributions 		system = (_systemSize != 0) && (uuid_compare(v3Entry->header.guid, gAppleSystemVariableGuid) == 0);
790*33de042dSApple OSS Distributions 		if (system) {
791*33de042dSApple OSS Distributions 			_systemUsed = _systemUsed + (uint32_t)variable_length(header) - (uint32_t)existingSize;
792*33de042dSApple OSS Distributions 		} else {
793*33de042dSApple OSS Distributions 			_commonUsed = _commonUsed + (uint32_t)variable_length(header) - (uint32_t)existingSize;
794*33de042dSApple OSS Distributions 		}
795*33de042dSApple OSS Distributions 
796*33de042dSApple OSS Distributions 		if (convertPropToObject(v3Entry->header.name_data_buf, v3Entry->header.nameSize,
797*33de042dSApple OSS Distributions 		    v3Entry->header.name_data_buf + v3Entry->header.nameSize, v3Entry->header.dataSize,
798*33de042dSApple OSS Distributions 		    propSymbol, propObject)) {
799*33de042dSApple OSS Distributions 			OSSharedPtr<const OSSymbol> canonicalKey = keyWithGuidAndCString(v3Entry->header.guid, (const char *)v3Entry->header.name_data_buf);
800*33de042dSApple OSS Distributions 
801*33de042dSApple OSS Distributions 			DEBUG_INFO("adding %s, dataLength=%u, system=%d\n",
802*33de042dSApple OSS Distributions 			    canonicalKey->getCStringNoCopy(), v3Entry->header.dataSize, system);
803*33de042dSApple OSS Distributions 
804*33de042dSApple OSS Distributions 			_varDict->setObject(canonicalKey.get(), propObject.get());
805*33de042dSApple OSS Distributions 
806*33de042dSApple OSS Distributions 			if (_provider->_diags) {
807*33de042dSApple OSS Distributions 				_provider->_diags->logVariable(getPartitionTypeForGUID(v3Entry->header.guid),
808*33de042dSApple OSS Distributions 				    kIONVRAMOperationInit, propSymbol.get()->getCStringNoCopy(),
809*33de042dSApple OSS Distributions 				    (void *)(uintptr_t)(header->name_data_buf + header->nameSize));
810*33de042dSApple OSS Distributions 			}
811*33de042dSApple OSS Distributions 		}
812*33de042dSApple OSS Distributions 		IOFreeData(v3Entry, nvram_v3_var_container_size(header));
813*33de042dSApple OSS Distributions skip:
814*33de042dSApple OSS Distributions 		offset += variable_length(header);
815*33de042dSApple OSS Distributions 	}
816*33de042dSApple OSS Distributions 
817*33de042dSApple OSS Distributions 	_currentOffset = (uint32_t)offset;
818*33de042dSApple OSS Distributions 
819*33de042dSApple OSS Distributions 	DEBUG_ALWAYS("_commonSize %#x, _systemSize %#x, _currentOffset %#x\n", _commonSize, _systemSize, _currentOffset);
820*33de042dSApple OSS Distributions 
821*33de042dSApple OSS Distributions 	ret = handleEphDM();
822*33de042dSApple OSS Distributions 	verify_noerr_action(ret, panic("handleEphDM failed with ret=%08x", ret));
823*33de042dSApple OSS Distributions 
824*33de042dSApple OSS Distributions 	DEBUG_INFO("_commonUsed %#x, _systemUsed %#x\n", _commonUsed, _systemUsed);
825*33de042dSApple OSS Distributions 
826*33de042dSApple OSS Distributions 	_newData = true;
827*33de042dSApple OSS Distributions 
828*33de042dSApple OSS Distributions 	if (_provider->_diags) {
829*33de042dSApple OSS Distributions 		OSSharedPtr<OSNumber> val = OSNumber::withNumber(getSystemUsed(), 32);
830*33de042dSApple OSS Distributions 		_provider->_diags->setProperty(kNVRAMSystemUsedKey, val.get());
831*33de042dSApple OSS Distributions 		DEBUG_INFO("%s=%u\n", kNVRAMSystemUsedKey, getSystemUsed());
832*33de042dSApple OSS Distributions 
833*33de042dSApple OSS Distributions 		val = OSNumber::withNumber(getCommonUsed(), 32);
834*33de042dSApple OSS Distributions 		_provider->_diags->setProperty(kNVRAMCommonUsedKey, val.get());
835*33de042dSApple OSS Distributions 		DEBUG_INFO("%s=%u\n", kNVRAMCommonUsedKey, getCommonUsed());
836*33de042dSApple OSS Distributions 	}
837*33de042dSApple OSS Distributions 
838*33de042dSApple OSS Distributions 	return ret;
839*33de042dSApple OSS Distributions }
840*33de042dSApple OSS Distributions 
841*33de042dSApple OSS Distributions IOReturn
setVariableInternal(const uuid_t varGuid,const char * variableName,OSObject * object)842*33de042dSApple OSS Distributions IONVRAMV3Handler::setVariableInternal(const uuid_t varGuid, const char *variableName, OSObject *object)
843*33de042dSApple OSS Distributions {
844*33de042dSApple OSS Distributions 	struct nvram_v3_var_entry   *v3Entry = nullptr;
845*33de042dSApple OSS Distributions 	struct nvram_v3_var_entry   *newV3Entry;
846*33de042dSApple OSS Distributions 	OSSharedPtr<OSData>         newContainer;
847*33de042dSApple OSS Distributions 	OSSharedPtr<const OSSymbol> canonicalKey;
848*33de042dSApple OSS Distributions 	bool                        unset = (object == nullptr);
849*33de042dSApple OSS Distributions 	bool                        system = false;
850*33de042dSApple OSS Distributions 	IOReturn                    ret = kIOReturnSuccess;
851*33de042dSApple OSS Distributions 	size_t                      entryNameLen = strlen(variableName) + 1;
852*33de042dSApple OSS Distributions 	unsigned int                existingEntryIndex;
853*33de042dSApple OSS Distributions 	uint32_t                    dataSize = 0;
854*33de042dSApple OSS Distributions 	size_t                      existingVariableSize = 0;
855*33de042dSApple OSS Distributions 	size_t                      newVariableSize = 0;
856*33de042dSApple OSS Distributions 	size_t                      newEntrySize;
857*33de042dSApple OSS Distributions 	uuid_string_t               uuidString;
858*33de042dSApple OSS Distributions 
859*33de042dSApple OSS Distributions 	system = (uuid_compare(varGuid, gAppleSystemVariableGuid) == 0);
860*33de042dSApple OSS Distributions 	canonicalKey = keyWithGuidAndCString(varGuid, variableName);
861*33de042dSApple OSS Distributions 
862*33de042dSApple OSS Distributions 	uuid_unparse(varGuid, uuidString);
863*33de042dSApple OSS Distributions 	DEBUG_INFO("setting %s:%s, system=%d, current var count=%u\n", uuidString, variableName, system, _varEntries->getCount());
864*33de042dSApple OSS Distributions 
865*33de042dSApple OSS Distributions 	findExistingEntry(varGuid, variableName, &v3Entry, &existingEntryIndex);
866*33de042dSApple OSS Distributions 
867*33de042dSApple OSS Distributions 	if (unset == true) {
868*33de042dSApple OSS Distributions 		setEntryForRemove(v3Entry, system);
869*33de042dSApple OSS Distributions 	} else {
870*33de042dSApple OSS Distributions 		if ((v3Entry != nullptr) && (v3Entry->new_state != VAR_NEW_STATE_REMOVE)) {
871*33de042dSApple OSS Distributions 			// Sizing was subtracted in setEntryForRemove
872*33de042dSApple OSS Distributions 			existingVariableSize = variable_length(&v3Entry->header);
873*33de042dSApple OSS Distributions 		}
874*33de042dSApple OSS Distributions 
875*33de042dSApple OSS Distributions 		convertObjectToProp(nullptr, &dataSize, variableName, object);
876*33de042dSApple OSS Distributions 
877*33de042dSApple OSS Distributions 		newVariableSize = sizeof(struct v3_var_header) + entryNameLen + dataSize;
878*33de042dSApple OSS Distributions 		newEntrySize = sizeof(struct nvram_v3_var_entry) + entryNameLen + dataSize;
879*33de042dSApple OSS Distributions 
880*33de042dSApple OSS Distributions 		if (system) {
881*33de042dSApple OSS Distributions 			if (_systemUsed - existingVariableSize + newVariableSize > _systemSize) {
882*33de042dSApple OSS Distributions 				DEBUG_ERROR("system region full\n");
883*33de042dSApple OSS Distributions 				ret = kIOReturnNoSpace;
884*33de042dSApple OSS Distributions 				goto exit;
885*33de042dSApple OSS Distributions 			}
886*33de042dSApple OSS Distributions 		} else if (_commonUsed - existingVariableSize + newVariableSize > _commonSize) {
887*33de042dSApple OSS Distributions 			DEBUG_ERROR("common region full\n");
888*33de042dSApple OSS Distributions 			ret = kIOReturnNoSpace;
889*33de042dSApple OSS Distributions 			goto exit;
890*33de042dSApple OSS Distributions 		}
891*33de042dSApple OSS Distributions 
892*33de042dSApple OSS Distributions 		DEBUG_INFO("creating new entry for %s, existingVariableSize=%#zx, newVariableSize=%#zx\n", variableName, existingVariableSize, newVariableSize);
893*33de042dSApple OSS Distributions 		newV3Entry = (struct nvram_v3_var_entry *)IOMallocZeroData(newEntrySize);
894*33de042dSApple OSS Distributions 
895*33de042dSApple OSS Distributions 		memcpy(newV3Entry->header.name_data_buf, variableName, entryNameLen);
896*33de042dSApple OSS Distributions 		convertObjectToProp(newV3Entry->header.name_data_buf + entryNameLen, &dataSize, variableName, object);
897*33de042dSApple OSS Distributions 
898*33de042dSApple OSS Distributions 		newV3Entry->header.startId = VARIABLE_DATA;
899*33de042dSApple OSS Distributions 		newV3Entry->header.nameSize = (uint32_t)entryNameLen;
900*33de042dSApple OSS Distributions 		newV3Entry->header.dataSize = dataSize;
901*33de042dSApple OSS Distributions 		newV3Entry->header.crc = crc32(0, newV3Entry->header.name_data_buf + entryNameLen, dataSize);
902*33de042dSApple OSS Distributions 		memcpy(newV3Entry->header.guid, varGuid, sizeof(gAppleNVRAMGuid));
903*33de042dSApple OSS Distributions 		newV3Entry->new_state = VAR_NEW_STATE_APPEND;
904*33de042dSApple OSS Distributions 
905*33de042dSApple OSS Distributions 		if (v3Entry) {
906*33de042dSApple OSS Distributions 			newV3Entry->existing_offset = v3Entry->existing_offset;
907*33de042dSApple OSS Distributions 			newV3Entry->header.state = v3Entry->header.state;
908*33de042dSApple OSS Distributions 			newV3Entry->header.attributes = v3Entry->header.attributes;
909*33de042dSApple OSS Distributions 
910*33de042dSApple OSS Distributions 			newContainer = OSData::withBytes(newV3Entry, (uint32_t)newEntrySize);
911*33de042dSApple OSS Distributions 			_varEntries->replaceObject(existingEntryIndex, newContainer.get());
912*33de042dSApple OSS Distributions 		} else {
913*33de042dSApple OSS Distributions 			newContainer = OSData::withBytes(newV3Entry, (uint32_t)newEntrySize);
914*33de042dSApple OSS Distributions 			_varEntries->setObject(newContainer.get());
915*33de042dSApple OSS Distributions 		}
916*33de042dSApple OSS Distributions 
917*33de042dSApple OSS Distributions 		if (system) {
918*33de042dSApple OSS Distributions 			_systemUsed = _systemUsed + (uint32_t)newVariableSize - (uint32_t)existingVariableSize;
919*33de042dSApple OSS Distributions 		} else {
920*33de042dSApple OSS Distributions 			_commonUsed = _commonUsed + (uint32_t)newVariableSize - (uint32_t)existingVariableSize;
921*33de042dSApple OSS Distributions 		}
922*33de042dSApple OSS Distributions 
923*33de042dSApple OSS Distributions 		_varDict->setObject(canonicalKey.get(), object);
924*33de042dSApple OSS Distributions 
925*33de042dSApple OSS Distributions 		if (_provider->_diags) {
926*33de042dSApple OSS Distributions 			_provider->_diags->logVariable(getPartitionTypeForGUID(varGuid),
927*33de042dSApple OSS Distributions 			    kIONVRAMOperationWrite, variableName,
928*33de042dSApple OSS Distributions 			    (void *)(uintptr_t)dataSize);
929*33de042dSApple OSS Distributions 		}
930*33de042dSApple OSS Distributions 
931*33de042dSApple OSS Distributions 		IOFreeData(newV3Entry, newEntrySize);
932*33de042dSApple OSS Distributions 	}
933*33de042dSApple OSS Distributions 
934*33de042dSApple OSS Distributions exit:
935*33de042dSApple OSS Distributions 	_newData = true;
936*33de042dSApple OSS Distributions 
937*33de042dSApple OSS Distributions 	if (_provider->_diags) {
938*33de042dSApple OSS Distributions 		OSSharedPtr<OSNumber> val = OSNumber::withNumber(getSystemUsed(), 32);
939*33de042dSApple OSS Distributions 		_provider->_diags->setProperty(kNVRAMSystemUsedKey, val.get());
940*33de042dSApple OSS Distributions 
941*33de042dSApple OSS Distributions 		val = OSNumber::withNumber(getCommonUsed(), 32);
942*33de042dSApple OSS Distributions 		_provider->_diags->setProperty(kNVRAMCommonUsedKey, val.get());
943*33de042dSApple OSS Distributions 	}
944*33de042dSApple OSS Distributions 
945*33de042dSApple OSS Distributions 	DEBUG_INFO("_commonUsed %#x, _systemUsed %#x\n", _commonUsed, _systemUsed);
946*33de042dSApple OSS Distributions 
947*33de042dSApple OSS Distributions 	return ret;
948*33de042dSApple OSS Distributions }
949*33de042dSApple OSS Distributions 
950*33de042dSApple OSS Distributions IOReturn
setVariable(const uuid_t varGuid,const char * variableName,OSObject * object)951*33de042dSApple OSS Distributions IONVRAMV3Handler::setVariable(const uuid_t varGuid, const char *variableName, OSObject *object)
952*33de042dSApple OSS Distributions {
953*33de042dSApple OSS Distributions 	uuid_t destGuid;
954*33de042dSApple OSS Distributions 
955*33de042dSApple OSS Distributions 	if (strcmp(variableName, "reclaim-int") == 0) {
956*33de042dSApple OSS Distributions 		return reclaim();
957*33de042dSApple OSS Distributions 	}
958*33de042dSApple OSS Distributions 
959*33de042dSApple OSS Distributions 	if (getSystemPartitionActive()) {
960*33de042dSApple OSS Distributions 		// System region case, if they're using the GUID directly or it's on the system allow list
961*33de042dSApple OSS Distributions 		// force it to use the System GUID
962*33de042dSApple OSS Distributions 		if ((uuid_compare(varGuid, gAppleSystemVariableGuid) == 0) || variableInAllowList(variableName)) {
963*33de042dSApple OSS Distributions 			uuid_copy(destGuid, gAppleSystemVariableGuid);
964*33de042dSApple OSS Distributions 		} else {
965*33de042dSApple OSS Distributions 			uuid_copy(destGuid, varGuid);
966*33de042dSApple OSS Distributions 		}
967*33de042dSApple OSS Distributions 	} else {
968*33de042dSApple OSS Distributions 		// No system region, store System GUID as Common GUID
969*33de042dSApple OSS Distributions 		if ((uuid_compare(varGuid, gAppleSystemVariableGuid) == 0) || variableInAllowList(variableName)) {
970*33de042dSApple OSS Distributions 			uuid_copy(destGuid, gAppleNVRAMGuid);
971*33de042dSApple OSS Distributions 		} else {
972*33de042dSApple OSS Distributions 			uuid_copy(destGuid, varGuid);
973*33de042dSApple OSS Distributions 		}
974*33de042dSApple OSS Distributions 	}
975*33de042dSApple OSS Distributions 
976*33de042dSApple OSS Distributions 	return setVariableInternal(destGuid, variableName, object);
977*33de042dSApple OSS Distributions }
978*33de042dSApple OSS Distributions 
979*33de042dSApple OSS Distributions uint32_t
findCurrentBank(void)980*33de042dSApple OSS Distributions IONVRAMV3Handler::findCurrentBank(void)
981*33de042dSApple OSS Distributions {
982*33de042dSApple OSS Distributions 	struct v3_store_header storeHeader;
983*33de042dSApple OSS Distributions 	uint32_t               maxGen = 0;
984*33de042dSApple OSS Distributions 	uint32_t               currentBank = 0;
985*33de042dSApple OSS Distributions 
986*33de042dSApple OSS Distributions 	for (unsigned int i = 0; i < _bankCount; i++) {
987*33de042dSApple OSS Distributions 		_nvramController->select(i);
988*33de042dSApple OSS Distributions 		_nvramController->read(0, (uint8_t *)&storeHeader, sizeof(storeHeader));
989*33de042dSApple OSS Distributions 
990*33de042dSApple OSS Distributions 		if (valid_store_header(&storeHeader) && (storeHeader.generation >= maxGen)) {
991*33de042dSApple OSS Distributions 			currentBank = i;
992*33de042dSApple OSS Distributions 			maxGen = storeHeader.generation;
993*33de042dSApple OSS Distributions 		}
994*33de042dSApple OSS Distributions 	}
995*33de042dSApple OSS Distributions 
996*33de042dSApple OSS Distributions 	DEBUG_ALWAYS("currentBank=%#x, gen=%#x", currentBank, maxGen);
997*33de042dSApple OSS Distributions 
998*33de042dSApple OSS Distributions 	return currentBank;
999*33de042dSApple OSS Distributions }
1000*33de042dSApple OSS Distributions 
1001*33de042dSApple OSS Distributions bool
setController(IONVRAMController * controller)1002*33de042dSApple OSS Distributions IONVRAMV3Handler::setController(IONVRAMController *controller)
1003*33de042dSApple OSS Distributions {
1004*33de042dSApple OSS Distributions 	IOReturn ret = kIOReturnSuccess;
1005*33de042dSApple OSS Distributions 
1006*33de042dSApple OSS Distributions 	if (_nvramController == NULL) {
1007*33de042dSApple OSS Distributions 		_nvramController = controller;
1008*33de042dSApple OSS Distributions 	}
1009*33de042dSApple OSS Distributions 
1010*33de042dSApple OSS Distributions 	DEBUG_INFO("Controller name: %s\n", _nvramController->getName());
1011*33de042dSApple OSS Distributions 
1012*33de042dSApple OSS Distributions 	require(_bankSize != 0, exit);
1013*33de042dSApple OSS Distributions 
1014*33de042dSApple OSS Distributions 	if (_resetData) {
1015*33de042dSApple OSS Distributions 		_resetData = false;
1016*33de042dSApple OSS Distributions 		DEBUG_ERROR("_resetData set, issuing reclaim recovery\n");
1017*33de042dSApple OSS Distributions 		ret = reclaim();
1018*33de042dSApple OSS Distributions 		require_noerr_action(ret, exit, DEBUG_ERROR("Reclaim recovery failed, invalid controller state!!! ret=%#x\n", ret));
1019*33de042dSApple OSS Distributions 		goto exit;
1020*33de042dSApple OSS Distributions 	}
1021*33de042dSApple OSS Distributions 
1022*33de042dSApple OSS Distributions 	ret = reloadInternal();
1023*33de042dSApple OSS Distributions 	if (ret != kIOReturnSuccess) {
1024*33de042dSApple OSS Distributions 		DEBUG_ERROR("Invalid image found, issuing reclaim recovery\n");
1025*33de042dSApple OSS Distributions 		ret = reclaim();
1026*33de042dSApple OSS Distributions 		require_noerr_action(ret, exit, DEBUG_ERROR("Reclaim recovery failed, invalid controller state!!! ret=%#x\n", ret));
1027*33de042dSApple OSS Distributions 	}
1028*33de042dSApple OSS Distributions 
1029*33de042dSApple OSS Distributions exit:
1030*33de042dSApple OSS Distributions 	return ret == kIOReturnSuccess;
1031*33de042dSApple OSS Distributions }
1032*33de042dSApple OSS Distributions 
1033*33de042dSApple OSS Distributions IOReturn
reclaim(void)1034*33de042dSApple OSS Distributions IONVRAMV3Handler::reclaim(void)
1035*33de042dSApple OSS Distributions {
1036*33de042dSApple OSS Distributions 	IOReturn             ret;
1037*33de042dSApple OSS Distributions 	struct               v3_store_header newStoreHeader;
1038*33de042dSApple OSS Distributions 	struct               v3_var_header *varHeader;
1039*33de042dSApple OSS Distributions 	struct               nvram_v3_var_entry *varEntry;
1040*33de042dSApple OSS Distributions 	OSData               *entryContainer;
1041*33de042dSApple OSS Distributions 	size_t               new_bank_offset = sizeof(struct v3_store_header);
1042*33de042dSApple OSS Distributions 	uint32_t             next_bank = (_currentBank + 1) % _bankCount;
1043*33de042dSApple OSS Distributions 	uint8_t              *bankData;
1044*33de042dSApple OSS Distributions 	OSSharedPtr<OSArray> remainingEntries;
1045*33de042dSApple OSS Distributions 
1046*33de042dSApple OSS Distributions 	DEBUG_INFO("called\n");
1047*33de042dSApple OSS Distributions 
1048*33de042dSApple OSS Distributions 	bankData = (uint8_t *)IOMallocData(_bankSize);
1049*33de042dSApple OSS Distributions 	require_action(bankData != nullptr, exit, ret = kIOReturnNoMemory);
1050*33de042dSApple OSS Distributions 
1051*33de042dSApple OSS Distributions 	ret = _nvramController->select(next_bank);
1052*33de042dSApple OSS Distributions 	verify_noerr_action(ret, DEBUG_INFO("select of bank %#08x failed\n", next_bank));
1053*33de042dSApple OSS Distributions 
1054*33de042dSApple OSS Distributions 	ret = _nvramController->eraseBank();
1055*33de042dSApple OSS Distributions 	verify_noerr_action(ret, DEBUG_INFO("eraseBank failed, ret=%#08x\n", ret));
1056*33de042dSApple OSS Distributions 
1057*33de042dSApple OSS Distributions 	_currentBank = next_bank;
1058*33de042dSApple OSS Distributions 
1059*33de042dSApple OSS Distributions 	remainingEntries = OSArray::withCapacity(_varEntries->getCapacity());
1060*33de042dSApple OSS Distributions 
1061*33de042dSApple OSS Distributions 	for (unsigned int i = 0; i < _varEntries->getCount(); i++) {
1062*33de042dSApple OSS Distributions 		entryContainer = OSDynamicCast(OSData, _varEntries->getObject(i));
1063*33de042dSApple OSS Distributions 		varEntry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
1064*33de042dSApple OSS Distributions 		varHeader = &varEntry->header;
1065*33de042dSApple OSS Distributions 
1066*33de042dSApple OSS Distributions 		DEBUG_INFO("entry %u %s, new_state=%#x, e_offset=%#lx, state=%#x\n",
1067*33de042dSApple OSS Distributions 		    i, varEntry->header.name_data_buf, varEntry->new_state, varEntry->existing_offset, varHeader->state);
1068*33de042dSApple OSS Distributions 
1069*33de042dSApple OSS Distributions 		if ((varEntry->new_state == VAR_NEW_STATE_NONE) ||
1070*33de042dSApple OSS Distributions 		    (varEntry->new_state == VAR_NEW_STATE_APPEND)) {
1071*33de042dSApple OSS Distributions 			varHeader->state = VAR_ADDED;
1072*33de042dSApple OSS Distributions 
1073*33de042dSApple OSS Distributions 			memcpy(bankData + new_bank_offset, (uint8_t *)varHeader, variable_length(varHeader));
1074*33de042dSApple OSS Distributions 
1075*33de042dSApple OSS Distributions 			varEntry->new_state = VAR_NEW_STATE_NONE;
1076*33de042dSApple OSS Distributions 			varEntry->existing_offset = new_bank_offset;
1077*33de042dSApple OSS Distributions 			new_bank_offset += variable_length(varHeader);
1078*33de042dSApple OSS Distributions 
1079*33de042dSApple OSS Distributions 			remainingEntries->setObject(entryContainer);
1080*33de042dSApple OSS Distributions 		} else {
1081*33de042dSApple OSS Distributions 			// entryContainer not added to remainingEntries, entry dropped
1082*33de042dSApple OSS Distributions 		}
1083*33de042dSApple OSS Distributions 	}
1084*33de042dSApple OSS Distributions 
1085*33de042dSApple OSS Distributions 	memcpy(&newStoreHeader, _nvramImage, sizeof(newStoreHeader));
1086*33de042dSApple OSS Distributions 
1087*33de042dSApple OSS Distributions 	_generation += 1;
1088*33de042dSApple OSS Distributions 
1089*33de042dSApple OSS Distributions 	newStoreHeader.generation = _generation;
1090*33de042dSApple OSS Distributions 
1091*33de042dSApple OSS Distributions 	memcpy(bankData, (uint8_t *)&newStoreHeader, sizeof(newStoreHeader));
1092*33de042dSApple OSS Distributions 
1093*33de042dSApple OSS Distributions 	ret = _nvramController->write(0, bankData, new_bank_offset);
1094*33de042dSApple OSS Distributions 	require_noerr_action(ret, exit, DEBUG_ERROR("reclaim bank write failed, ret=%08x\n", ret));
1095*33de042dSApple OSS Distributions 
1096*33de042dSApple OSS Distributions 	_currentOffset = (uint32_t)new_bank_offset;
1097*33de042dSApple OSS Distributions 
1098*33de042dSApple OSS Distributions 	DEBUG_INFO("Reclaim complete, _currentBank=%u _generation=%u, _currentOffset=%#x\n", _currentBank, _generation, _currentOffset);
1099*33de042dSApple OSS Distributions 
1100*33de042dSApple OSS Distributions 	_newData = false;
1101*33de042dSApple OSS Distributions 
1102*33de042dSApple OSS Distributions 	_varEntries.reset(remainingEntries.get(), OSRetain);
1103*33de042dSApple OSS Distributions 
1104*33de042dSApple OSS Distributions exit:
1105*33de042dSApple OSS Distributions 	IOFreeData(bankData, _bankSize);
1106*33de042dSApple OSS Distributions 
1107*33de042dSApple OSS Distributions 	return ret;
1108*33de042dSApple OSS Distributions }
1109*33de042dSApple OSS Distributions 
1110*33de042dSApple OSS Distributions size_t
getAppendSize(void)1111*33de042dSApple OSS Distributions IONVRAMV3Handler::getAppendSize(void)
1112*33de042dSApple OSS Distributions {
1113*33de042dSApple OSS Distributions 	struct nvram_v3_var_entry *varEntry;
1114*33de042dSApple OSS Distributions 	struct v3_var_header      *varHeader;
1115*33de042dSApple OSS Distributions 	OSData                    *entryContainer;
1116*33de042dSApple OSS Distributions 	size_t                    appendSize = 0;
1117*33de042dSApple OSS Distributions 
1118*33de042dSApple OSS Distributions 	for (unsigned int i = 0; i < _varEntries->getCount(); i++) {
1119*33de042dSApple OSS Distributions 		entryContainer = OSDynamicCast(OSData, _varEntries->getObject(i));
1120*33de042dSApple OSS Distributions 		varEntry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
1121*33de042dSApple OSS Distributions 		varHeader = &varEntry->header;
1122*33de042dSApple OSS Distributions 
1123*33de042dSApple OSS Distributions 		if (varEntry->new_state == VAR_NEW_STATE_APPEND) {
1124*33de042dSApple OSS Distributions 			appendSize += variable_length(varHeader);
1125*33de042dSApple OSS Distributions 		}
1126*33de042dSApple OSS Distributions 	}
1127*33de042dSApple OSS Distributions 
1128*33de042dSApple OSS Distributions 	return appendSize;
1129*33de042dSApple OSS Distributions }
1130*33de042dSApple OSS Distributions 
1131*33de042dSApple OSS Distributions IOReturn
syncRaw(void)1132*33de042dSApple OSS Distributions IONVRAMV3Handler::syncRaw(void)
1133*33de042dSApple OSS Distributions {
1134*33de042dSApple OSS Distributions 	IOReturn                  ret = kIOReturnSuccess;
1135*33de042dSApple OSS Distributions 	struct nvram_v3_var_entry *varEntry;
1136*33de042dSApple OSS Distributions 	struct v3_var_header      *varHeader;
1137*33de042dSApple OSS Distributions 	OSData                    *entryContainer;
1138*33de042dSApple OSS Distributions 	OSSharedPtr<OSArray>      remainingEntries;
1139*33de042dSApple OSS Distributions 	uint8_t                   *appendBuffer = nullptr;
1140*33de042dSApple OSS Distributions 	size_t                    appendBufferOffset = 0;
1141*33de042dSApple OSS Distributions 	size_t                    *invalidateOffsets = nullptr;
1142*33de042dSApple OSS Distributions 	size_t                    invalidateOffsetsCount = 0;
1143*33de042dSApple OSS Distributions 	size_t                    invalidateOffsetIndex = 0;
1144*33de042dSApple OSS Distributions 	size_t                    invalidatedSize = 0;
1145*33de042dSApple OSS Distributions 
1146*33de042dSApple OSS Distributions 	require_action(_nvramController != nullptr, exit, DEBUG_INFO("No _nvramController\n"));
1147*33de042dSApple OSS Distributions 	require_action(_newData == true, exit, DEBUG_INFO("No _newData to sync\n"));
1148*33de042dSApple OSS Distributions 	require_action(_bankSize != 0, exit, DEBUG_INFO("No nvram size info\n"));
1149*33de042dSApple OSS Distributions 
1150*33de042dSApple OSS Distributions 	DEBUG_INFO("_varEntries->getCount()=%#x\n", _varEntries->getCount());
1151*33de042dSApple OSS Distributions 
1152*33de042dSApple OSS Distributions 	if (getAppendSize() + _currentOffset < _bankSize) {
1153*33de042dSApple OSS Distributions 		// No reclaim, build append and invalidate list
1154*33de042dSApple OSS Distributions 
1155*33de042dSApple OSS Distributions 		remainingEntries = OSArray::withCapacity(_varEntries->getCapacity());
1156*33de042dSApple OSS Distributions 
1157*33de042dSApple OSS Distributions 		appendBuffer = (uint8_t *)IOMallocData(_bankSize);
1158*33de042dSApple OSS Distributions 		require_action(appendBuffer, exit, ret = kIOReturnNoMemory);
1159*33de042dSApple OSS Distributions 
1160*33de042dSApple OSS Distributions 		invalidateOffsetsCount = _varEntries->getCount();
1161*33de042dSApple OSS Distributions 		invalidateOffsets = (size_t *)IOMallocData(invalidateOffsetsCount * sizeof(size_t));
1162*33de042dSApple OSS Distributions 		require_action(invalidateOffsets, exit, ret = kIOReturnNoMemory);
1163*33de042dSApple OSS Distributions 
1164*33de042dSApple OSS Distributions 		for (unsigned int i = 0; i < _varEntries->getCount(); i++) {
1165*33de042dSApple OSS Distributions 			entryContainer = OSDynamicCast(OSData, _varEntries->getObject(i));
1166*33de042dSApple OSS Distributions 			varEntry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
1167*33de042dSApple OSS Distributions 			varHeader = &varEntry->header;
1168*33de042dSApple OSS Distributions 
1169*33de042dSApple OSS Distributions 			DEBUG_INFO("entry %s, new_state=%#02x state=%#02x, existing_offset=%#zx\n",
1170*33de042dSApple OSS Distributions 			    varEntry->header.name_data_buf, varEntry->new_state, varEntry->header.state, varEntry->existing_offset);
1171*33de042dSApple OSS Distributions 
1172*33de042dSApple OSS Distributions 			if (varEntry->new_state == VAR_NEW_STATE_APPEND) {
1173*33de042dSApple OSS Distributions 				size_t varSize = variable_length(varHeader);
1174*33de042dSApple OSS Distributions 				size_t prevOffset = varEntry->existing_offset;
1175*33de042dSApple OSS Distributions 
1176*33de042dSApple OSS Distributions 				varHeader->state = VAR_ADDED;
1177*33de042dSApple OSS Distributions 				varEntry->existing_offset = _currentOffset + appendBufferOffset;
1178*33de042dSApple OSS Distributions 				varEntry->new_state = VAR_NEW_STATE_NONE;
1179*33de042dSApple OSS Distributions 
1180*33de042dSApple OSS Distributions 				DEBUG_INFO("Appending %s in append buffer offset %#zx, actual offset %#zx, prevOffset %#zx, varsize=%#zx\n",
1181*33de042dSApple OSS Distributions 				    varEntry->header.name_data_buf, appendBufferOffset, varEntry->existing_offset, prevOffset, varSize);
1182*33de042dSApple OSS Distributions 
1183*33de042dSApple OSS Distributions 				// Write to append buffer
1184*33de042dSApple OSS Distributions 				memcpy(appendBuffer + appendBufferOffset, (uint8_t *)varHeader, varSize);
1185*33de042dSApple OSS Distributions 				appendBufferOffset += varSize;
1186*33de042dSApple OSS Distributions 
1187*33de042dSApple OSS Distributions 				if (prevOffset) {
1188*33de042dSApple OSS Distributions 					invalidateOffsets[invalidateOffsetIndex++] = prevOffset;
1189*33de042dSApple OSS Distributions 					invalidatedSize += variable_length((struct v3_var_header *)prevOffset);
1190*33de042dSApple OSS Distributions 				}
1191*33de042dSApple OSS Distributions 
1192*33de042dSApple OSS Distributions 				remainingEntries->setObject(entryContainer);
1193*33de042dSApple OSS Distributions 			} else if (varEntry->new_state == VAR_NEW_STATE_REMOVE) {
1194*33de042dSApple OSS Distributions 				if (varEntry->existing_offset) {
1195*33de042dSApple OSS Distributions 					DEBUG_INFO("marking entry at offset %#lx deleted\n", varEntry->existing_offset);
1196*33de042dSApple OSS Distributions 
1197*33de042dSApple OSS Distributions 					invalidateOffsets[invalidateOffsetIndex++] = varEntry->existing_offset;
1198*33de042dSApple OSS Distributions 					invalidatedSize += variable_length((struct v3_var_header *)varEntry->existing_offset);
1199*33de042dSApple OSS Distributions 				} else {
1200*33de042dSApple OSS Distributions 					DEBUG_INFO("No existing_offset , removing\n");
1201*33de042dSApple OSS Distributions 				}
1202*33de042dSApple OSS Distributions 
1203*33de042dSApple OSS Distributions 				// not re-added to remainingEntries
1204*33de042dSApple OSS Distributions 			} else {
1205*33de042dSApple OSS Distributions 				DEBUG_INFO("skipping\n");
1206*33de042dSApple OSS Distributions 				remainingEntries->setObject(entryContainer);
1207*33de042dSApple OSS Distributions 			}
1208*33de042dSApple OSS Distributions 		}
1209*33de042dSApple OSS Distributions 
1210*33de042dSApple OSS Distributions 		if (appendBufferOffset > 0) {
1211*33de042dSApple OSS Distributions 			// Write appendBuffer
1212*33de042dSApple OSS Distributions 			DEBUG_INFO("Appending append buffer size=%#zx at offset=%#x\n", appendBufferOffset, _currentOffset);
1213*33de042dSApple OSS Distributions 			ret = _nvramController->write(_currentOffset, appendBuffer, appendBufferOffset);
1214*33de042dSApple OSS Distributions 			require_noerr_action(ret, exit, DEBUG_ERROR("could not re-append, ret=%#x\n", ret));
1215*33de042dSApple OSS Distributions 
1216*33de042dSApple OSS Distributions 			_currentOffset += appendBufferOffset;
1217*33de042dSApple OSS Distributions 		} else {
1218*33de042dSApple OSS Distributions 			DEBUG_INFO("No entries to append\n");
1219*33de042dSApple OSS Distributions 		}
1220*33de042dSApple OSS Distributions 
1221*33de042dSApple OSS Distributions 		if (invalidateOffsetIndex > 0) {
1222*33de042dSApple OSS Distributions 			// Invalidate Entries
1223*33de042dSApple OSS Distributions 			for (unsigned int i = 0; i < invalidateOffsetIndex; i++) {
1224*33de042dSApple OSS Distributions 				uint8_t state = VAR_ADDED & VAR_DELETED & VAR_IN_DELETED_TRANSITION;
1225*33de042dSApple OSS Distributions 
1226*33de042dSApple OSS Distributions 				ret = _nvramController->write(invalidateOffsets[i] + offsetof(struct v3_var_header, state), &state, sizeof(state));
1227*33de042dSApple OSS Distributions 				require_noerr_action(ret, exit, DEBUG_ERROR("unable to invalidate at offset %#zx, ret=%#x\n", invalidateOffsets[i], ret));
1228*33de042dSApple OSS Distributions 				DEBUG_INFO("Invalidated entry at offset=%#zx\n", invalidateOffsets[i]);
1229*33de042dSApple OSS Distributions 			}
1230*33de042dSApple OSS Distributions 		} else {
1231*33de042dSApple OSS Distributions 			DEBUG_INFO("No entries to invalidate\n");
1232*33de042dSApple OSS Distributions 		}
1233*33de042dSApple OSS Distributions 
1234*33de042dSApple OSS Distributions 		_newData = false;
1235*33de042dSApple OSS Distributions 
1236*33de042dSApple OSS Distributions 		_varEntries.reset(remainingEntries.get(), OSRetain);
1237*33de042dSApple OSS Distributions 	} else {
1238*33de042dSApple OSS Distributions 		// Will need to reclaim, rebuild store and write everything at once
1239*33de042dSApple OSS Distributions 		ret = reclaim();
1240*33de042dSApple OSS Distributions 	}
1241*33de042dSApple OSS Distributions 
1242*33de042dSApple OSS Distributions exit:
1243*33de042dSApple OSS Distributions 	IOFreeData(appendBuffer, _bankSize);
1244*33de042dSApple OSS Distributions 	IOFreeData(invalidateOffsets, invalidateOffsetsCount * sizeof(size_t));
1245*33de042dSApple OSS Distributions 
1246*33de042dSApple OSS Distributions 	return ret;
1247*33de042dSApple OSS Distributions }
1248*33de042dSApple OSS Distributions 
1249*33de042dSApple OSS Distributions IOReturn
syncBlock(void)1250*33de042dSApple OSS Distributions IONVRAMV3Handler::syncBlock(void)
1251*33de042dSApple OSS Distributions {
1252*33de042dSApple OSS Distributions 	IOReturn             ret = kIOReturnSuccess;
1253*33de042dSApple OSS Distributions 	struct               v3_store_header newStoreHeader;
1254*33de042dSApple OSS Distributions 	struct               v3_var_header *varHeader;
1255*33de042dSApple OSS Distributions 	struct               nvram_v3_var_entry *varEntry;
1256*33de042dSApple OSS Distributions 	OSData               *entryContainer;
1257*33de042dSApple OSS Distributions 	size_t               new_bank_offset = sizeof(struct v3_store_header);
1258*33de042dSApple OSS Distributions 	uint8_t              *block;
1259*33de042dSApple OSS Distributions 	OSSharedPtr<OSArray> remainingEntries;
1260*33de042dSApple OSS Distributions 	uint32_t             next_bank = (_currentBank + 1) % _bankCount;
1261*33de042dSApple OSS Distributions 
1262*33de042dSApple OSS Distributions 	DEBUG_INFO("called\n");
1263*33de042dSApple OSS Distributions 
1264*33de042dSApple OSS Distributions 	require_action(_nvramController != nullptr, exit, DEBUG_INFO("No _nvramController\n"));
1265*33de042dSApple OSS Distributions 	require_action(_newData == true, exit, DEBUG_INFO("No _newData to sync\n"));
1266*33de042dSApple OSS Distributions 	require_action(_bankSize != 0, exit, DEBUG_INFO("No nvram size info\n"));
1267*33de042dSApple OSS Distributions 
1268*33de042dSApple OSS Distributions 	block = (uint8_t *)IOMallocData(_bankSize);
1269*33de042dSApple OSS Distributions 
1270*33de042dSApple OSS Distributions 	remainingEntries = OSArray::withCapacity(_varEntries->getCapacity());
1271*33de042dSApple OSS Distributions 
1272*33de042dSApple OSS Distributions 	ret = _nvramController->select(next_bank);
1273*33de042dSApple OSS Distributions 	verify_noerr_action(ret, DEBUG_INFO("select of bank %#x failed\n", next_bank));
1274*33de042dSApple OSS Distributions 
1275*33de042dSApple OSS Distributions 	ret = _nvramController->eraseBank();
1276*33de042dSApple OSS Distributions 	verify_noerr_action(ret, DEBUG_INFO("eraseBank failed, ret=%#08x\n", ret));
1277*33de042dSApple OSS Distributions 
1278*33de042dSApple OSS Distributions 	_currentBank = next_bank;
1279*33de042dSApple OSS Distributions 
1280*33de042dSApple OSS Distributions 	memcpy(&newStoreHeader, _nvramImage, sizeof(newStoreHeader));
1281*33de042dSApple OSS Distributions 
1282*33de042dSApple OSS Distributions 	_generation += 1;
1283*33de042dSApple OSS Distributions 
1284*33de042dSApple OSS Distributions 	newStoreHeader.generation = _generation;
1285*33de042dSApple OSS Distributions 
1286*33de042dSApple OSS Distributions 	memcpy(block, (uint8_t *)&newStoreHeader, sizeof(newStoreHeader));
1287*33de042dSApple OSS Distributions 
1288*33de042dSApple OSS Distributions 	for (unsigned int i = 0; i < _varEntries->getCount(); i++) {
1289*33de042dSApple OSS Distributions 		entryContainer = OSDynamicCast(OSData, _varEntries->getObject(i));
1290*33de042dSApple OSS Distributions 		varEntry = (struct nvram_v3_var_entry *)entryContainer->getBytesNoCopy();
1291*33de042dSApple OSS Distributions 		varHeader = &varEntry->header;
1292*33de042dSApple OSS Distributions 
1293*33de042dSApple OSS Distributions 		DEBUG_INFO("entry %u %s, new_state=%#x, e_offset=%#lx, state=%#x\n",
1294*33de042dSApple OSS Distributions 		    i, varEntry->header.name_data_buf, varEntry->new_state, varEntry->existing_offset, varHeader->state);
1295*33de042dSApple OSS Distributions 
1296*33de042dSApple OSS Distributions 		if (varEntry->new_state != VAR_NEW_STATE_REMOVE) {
1297*33de042dSApple OSS Distributions 			varHeader->state = VAR_ADDED;
1298*33de042dSApple OSS Distributions 
1299*33de042dSApple OSS Distributions 			memcpy(block + new_bank_offset, (uint8_t *)varHeader, variable_length(varHeader));
1300*33de042dSApple OSS Distributions 
1301*33de042dSApple OSS Distributions 			varEntry->existing_offset = new_bank_offset;
1302*33de042dSApple OSS Distributions 			new_bank_offset += variable_length(varHeader);
1303*33de042dSApple OSS Distributions 			varEntry->new_state = VAR_NEW_STATE_NONE;
1304*33de042dSApple OSS Distributions 
1305*33de042dSApple OSS Distributions 			remainingEntries->setObject(entryContainer);
1306*33de042dSApple OSS Distributions 		} else {
1307*33de042dSApple OSS Distributions 			DEBUG_INFO("Dropping %s\n", varEntry->header.name_data_buf);
1308*33de042dSApple OSS Distributions 		}
1309*33de042dSApple OSS Distributions 	}
1310*33de042dSApple OSS Distributions 
1311*33de042dSApple OSS Distributions 	ret = _nvramController->write(0, block, _bankSize);
1312*33de042dSApple OSS Distributions 	verify_noerr_action(ret, DEBUG_ERROR("w fail, ret=%#x\n", ret));
1313*33de042dSApple OSS Distributions 
1314*33de042dSApple OSS Distributions 	_nvramController->sync();
1315*33de042dSApple OSS Distributions 
1316*33de042dSApple OSS Distributions 	_varEntries.reset(remainingEntries.get(), OSRetain);
1317*33de042dSApple OSS Distributions 
1318*33de042dSApple OSS Distributions 	_newData = false;
1319*33de042dSApple OSS Distributions 
1320*33de042dSApple OSS Distributions 	DEBUG_INFO("Save complete, _generation=%u\n", _generation);
1321*33de042dSApple OSS Distributions 
1322*33de042dSApple OSS Distributions 	IOFreeData(block, _bankSize);
1323*33de042dSApple OSS Distributions 
1324*33de042dSApple OSS Distributions exit:
1325*33de042dSApple OSS Distributions 	return ret;
1326*33de042dSApple OSS Distributions }
1327*33de042dSApple OSS Distributions 
1328*33de042dSApple OSS Distributions IOReturn
sync(void)1329*33de042dSApple OSS Distributions IONVRAMV3Handler::sync(void)
1330*33de042dSApple OSS Distributions {
1331*33de042dSApple OSS Distributions 	IOReturn ret;
1332*33de042dSApple OSS Distributions 
1333*33de042dSApple OSS Distributions 	if (_reload) {
1334*33de042dSApple OSS Distributions 		ret = reloadInternal();
1335*33de042dSApple OSS Distributions 		require_noerr_action(ret, exit, DEBUG_ERROR("Reload failed, ret=%#x", ret));
1336*33de042dSApple OSS Distributions 
1337*33de042dSApple OSS Distributions 		_reload = false;
1338*33de042dSApple OSS Distributions 	}
1339*33de042dSApple OSS Distributions 
1340*33de042dSApple OSS Distributions 	if (_rawController == true) {
1341*33de042dSApple OSS Distributions 		ret = syncRaw();
1342*33de042dSApple OSS Distributions 
1343*33de042dSApple OSS Distributions 		if (ret != kIOReturnSuccess) {
1344*33de042dSApple OSS Distributions 			ret = reclaim();
1345*33de042dSApple OSS Distributions 			require_noerr_action(ret, exit, DEBUG_ERROR("Reclaim recovery failed, ret=%#x", ret));
1346*33de042dSApple OSS Distributions 		}
1347*33de042dSApple OSS Distributions 	} else {
1348*33de042dSApple OSS Distributions 		ret = syncBlock();
1349*33de042dSApple OSS Distributions 	}
1350*33de042dSApple OSS Distributions 
1351*33de042dSApple OSS Distributions exit:
1352*33de042dSApple OSS Distributions 	return ret;
1353*33de042dSApple OSS Distributions }
1354*33de042dSApple OSS Distributions 
1355*33de042dSApple OSS Distributions uint32_t
getGeneration(void) const1356*33de042dSApple OSS Distributions IONVRAMV3Handler::getGeneration(void) const
1357*33de042dSApple OSS Distributions {
1358*33de042dSApple OSS Distributions 	return _generation;
1359*33de042dSApple OSS Distributions }
1360*33de042dSApple OSS Distributions 
1361*33de042dSApple OSS Distributions uint32_t
getVersion(void) const1362*33de042dSApple OSS Distributions IONVRAMV3Handler::getVersion(void) const
1363*33de042dSApple OSS Distributions {
1364*33de042dSApple OSS Distributions 	return kNVRAMVersion3;
1365*33de042dSApple OSS Distributions }
1366*33de042dSApple OSS Distributions 
1367*33de042dSApple OSS Distributions uint32_t
getSystemUsed(void) const1368*33de042dSApple OSS Distributions IONVRAMV3Handler::getSystemUsed(void) const
1369*33de042dSApple OSS Distributions {
1370*33de042dSApple OSS Distributions 	return _systemUsed;
1371*33de042dSApple OSS Distributions }
1372*33de042dSApple OSS Distributions 
1373*33de042dSApple OSS Distributions uint32_t
getCommonUsed(void) const1374*33de042dSApple OSS Distributions IONVRAMV3Handler::getCommonUsed(void) const
1375*33de042dSApple OSS Distributions {
1376*33de042dSApple OSS Distributions 	return _commonUsed;
1377*33de042dSApple OSS Distributions }
1378*33de042dSApple OSS Distributions 
1379*33de042dSApple OSS Distributions bool
getSystemPartitionActive(void) const1380*33de042dSApple OSS Distributions IONVRAMV3Handler::getSystemPartitionActive(void) const
1381*33de042dSApple OSS Distributions {
1382*33de042dSApple OSS Distributions 	return _systemSize != 0;
1383*33de042dSApple OSS Distributions }
1384*33de042dSApple OSS Distributions 
1385*33de042dSApple OSS Distributions bool
convertObjectToProp(uint8_t * buffer,uint32_t * length,const char * propName,OSObject * propObject)1386*33de042dSApple OSS Distributions IONVRAMV3Handler::convertObjectToProp(uint8_t *buffer, uint32_t *length,
1387*33de042dSApple OSS Distributions     const char *propName, OSObject *propObject)
1388*33de042dSApple OSS Distributions {
1389*33de042dSApple OSS Distributions 	uint32_t             offset;
1390*33de042dSApple OSS Distributions 	IONVRAMVariableType  propType;
1391*33de042dSApple OSS Distributions 	OSBoolean            *tmpBoolean = nullptr;
1392*33de042dSApple OSS Distributions 	OSNumber             *tmpNumber = nullptr;
1393*33de042dSApple OSS Distributions 	OSString             *tmpString = nullptr;
1394*33de042dSApple OSS Distributions 	OSData               *tmpData = nullptr;
1395*33de042dSApple OSS Distributions 
1396*33de042dSApple OSS Distributions 	propType = getVariableType(propName);
1397*33de042dSApple OSS Distributions 
1398*33de042dSApple OSS Distributions 	// Get the size of the data.
1399*33de042dSApple OSS Distributions 	offset = 0;
1400*33de042dSApple OSS Distributions 	switch (propType) {
1401*33de042dSApple OSS Distributions 	case kOFVariableTypeBoolean:
1402*33de042dSApple OSS Distributions 		tmpBoolean = OSDynamicCast(OSBoolean, propObject);
1403*33de042dSApple OSS Distributions 		if (tmpBoolean != nullptr) {
1404*33de042dSApple OSS Distributions 			const char *bool_buf;
1405*33de042dSApple OSS Distributions 			if (tmpBoolean->getValue()) {
1406*33de042dSApple OSS Distributions 				bool_buf = "true";
1407*33de042dSApple OSS Distributions 			} else {
1408*33de042dSApple OSS Distributions 				bool_buf = "false";
1409*33de042dSApple OSS Distributions 			}
1410*33de042dSApple OSS Distributions 
1411*33de042dSApple OSS Distributions 			offset = (uint32_t)strlen(bool_buf);
1412*33de042dSApple OSS Distributions 
1413*33de042dSApple OSS Distributions 			if (buffer) {
1414*33de042dSApple OSS Distributions 				if (*length < offset) {
1415*33de042dSApple OSS Distributions 					return false;
1416*33de042dSApple OSS Distributions 				} else {
1417*33de042dSApple OSS Distributions 					memcpy(buffer, bool_buf, offset);
1418*33de042dSApple OSS Distributions 				}
1419*33de042dSApple OSS Distributions 			}
1420*33de042dSApple OSS Distributions 		}
1421*33de042dSApple OSS Distributions 		break;
1422*33de042dSApple OSS Distributions 
1423*33de042dSApple OSS Distributions 	case kOFVariableTypeNumber:
1424*33de042dSApple OSS Distributions 		tmpNumber = OSDynamicCast(OSNumber, propObject);
1425*33de042dSApple OSS Distributions 		if (tmpNumber != nullptr) {
1426*33de042dSApple OSS Distributions 			char num_buf[12];
1427*33de042dSApple OSS Distributions 			char *end_buf = num_buf;
1428*33de042dSApple OSS Distributions 			uint32_t tmpValue = tmpNumber->unsigned32BitValue();
1429*33de042dSApple OSS Distributions 			if (tmpValue == 0xFFFFFFFF) {
1430*33de042dSApple OSS Distributions 				end_buf += snprintf(end_buf, sizeof(num_buf), "-1");
1431*33de042dSApple OSS Distributions 			} else if (tmpValue < 1000) {
1432*33de042dSApple OSS Distributions 				end_buf += snprintf(end_buf, sizeof(num_buf), "%d", (uint32_t)tmpValue);
1433*33de042dSApple OSS Distributions 			} else {
1434*33de042dSApple OSS Distributions 				end_buf += snprintf(end_buf, sizeof(num_buf), "%#x", (uint32_t)tmpValue);
1435*33de042dSApple OSS Distributions 			}
1436*33de042dSApple OSS Distributions 
1437*33de042dSApple OSS Distributions 			offset = (uint32_t)(end_buf - num_buf);
1438*33de042dSApple OSS Distributions 			if (buffer) {
1439*33de042dSApple OSS Distributions 				if (*length < offset) {
1440*33de042dSApple OSS Distributions 					return false;
1441*33de042dSApple OSS Distributions 				} else {
1442*33de042dSApple OSS Distributions 					memcpy(buffer, num_buf, offset);
1443*33de042dSApple OSS Distributions 				}
1444*33de042dSApple OSS Distributions 			}
1445*33de042dSApple OSS Distributions 		}
1446*33de042dSApple OSS Distributions 		break;
1447*33de042dSApple OSS Distributions 
1448*33de042dSApple OSS Distributions 	case kOFVariableTypeString:
1449*33de042dSApple OSS Distributions 		tmpString = OSDynamicCast(OSString, propObject);
1450*33de042dSApple OSS Distributions 		if (tmpString != nullptr) {
1451*33de042dSApple OSS Distributions 			offset = tmpString->getLength();
1452*33de042dSApple OSS Distributions 
1453*33de042dSApple OSS Distributions 			if (buffer) {
1454*33de042dSApple OSS Distributions 				if (*length < offset) {
1455*33de042dSApple OSS Distributions 					return false;
1456*33de042dSApple OSS Distributions 				} else {
1457*33de042dSApple OSS Distributions 					bcopy(tmpString->getCStringNoCopy(), buffer, offset);
1458*33de042dSApple OSS Distributions 				}
1459*33de042dSApple OSS Distributions 			}
1460*33de042dSApple OSS Distributions 		}
1461*33de042dSApple OSS Distributions 		break;
1462*33de042dSApple OSS Distributions 
1463*33de042dSApple OSS Distributions 	case kOFVariableTypeData:
1464*33de042dSApple OSS Distributions 		tmpData = OSDynamicCast(OSData, propObject);
1465*33de042dSApple OSS Distributions 		if (tmpData != nullptr) {
1466*33de042dSApple OSS Distributions 			offset = tmpData->getLength();
1467*33de042dSApple OSS Distributions 
1468*33de042dSApple OSS Distributions 			if (buffer) {
1469*33de042dSApple OSS Distributions 				if (*length < offset) {
1470*33de042dSApple OSS Distributions 					return false;
1471*33de042dSApple OSS Distributions 				} else {
1472*33de042dSApple OSS Distributions 					bcopy(tmpData->getBytesNoCopy(), buffer, offset);
1473*33de042dSApple OSS Distributions 				}
1474*33de042dSApple OSS Distributions 			}
1475*33de042dSApple OSS Distributions 		}
1476*33de042dSApple OSS Distributions 		break;
1477*33de042dSApple OSS Distributions 
1478*33de042dSApple OSS Distributions 	default:
1479*33de042dSApple OSS Distributions 		return false;
1480*33de042dSApple OSS Distributions 	}
1481*33de042dSApple OSS Distributions 
1482*33de042dSApple OSS Distributions 	*length = offset;
1483*33de042dSApple OSS Distributions 
1484*33de042dSApple OSS Distributions 	return offset != 0;
1485*33de042dSApple OSS Distributions }
1486*33de042dSApple OSS Distributions 
1487*33de042dSApple OSS Distributions 
1488*33de042dSApple 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)1489*33de042dSApple OSS Distributions IONVRAMV3Handler::convertPropToObject(const uint8_t *propName, uint32_t propNameLength,
1490*33de042dSApple OSS Distributions     const uint8_t *propData, uint32_t propDataLength,
1491*33de042dSApple OSS Distributions     OSSharedPtr<const OSSymbol>& propSymbol,
1492*33de042dSApple OSS Distributions     OSSharedPtr<OSObject>& propObject)
1493*33de042dSApple OSS Distributions {
1494*33de042dSApple OSS Distributions 	OSSharedPtr<const OSSymbol> tmpSymbol;
1495*33de042dSApple OSS Distributions 	OSSharedPtr<OSNumber>       tmpNumber;
1496*33de042dSApple OSS Distributions 	OSSharedPtr<OSString>       tmpString;
1497*33de042dSApple OSS Distributions 	OSSharedPtr<OSObject>       tmpObject = nullptr;
1498*33de042dSApple OSS Distributions 
1499*33de042dSApple OSS Distributions 	tmpSymbol = OSSymbol::withCString((const char *)propName);
1500*33de042dSApple OSS Distributions 
1501*33de042dSApple OSS Distributions 	if (tmpSymbol == nullptr) {
1502*33de042dSApple OSS Distributions 		return false;
1503*33de042dSApple OSS Distributions 	}
1504*33de042dSApple OSS Distributions 
1505*33de042dSApple OSS Distributions 	switch (getVariableType(tmpSymbol.get())) {
1506*33de042dSApple OSS Distributions 	case kOFVariableTypeBoolean:
1507*33de042dSApple OSS Distributions 		if (!strncmp("true", (const char *)propData, propDataLength)) {
1508*33de042dSApple OSS Distributions 			tmpObject.reset(kOSBooleanTrue, OSRetain);
1509*33de042dSApple OSS Distributions 		} else if (!strncmp("false", (const char *)propData, propDataLength)) {
1510*33de042dSApple OSS Distributions 			tmpObject.reset(kOSBooleanFalse, OSRetain);
1511*33de042dSApple OSS Distributions 		}
1512*33de042dSApple OSS Distributions 		break;
1513*33de042dSApple OSS Distributions 
1514*33de042dSApple OSS Distributions 	case kOFVariableTypeNumber:
1515*33de042dSApple OSS Distributions 		tmpNumber = OSNumber::withNumber(strtol((const char *)propData, nullptr, 0), 32);
1516*33de042dSApple OSS Distributions 		if (tmpNumber != nullptr) {
1517*33de042dSApple OSS Distributions 			tmpObject = tmpNumber;
1518*33de042dSApple OSS Distributions 		}
1519*33de042dSApple OSS Distributions 		break;
1520*33de042dSApple OSS Distributions 
1521*33de042dSApple OSS Distributions 	case kOFVariableTypeString:
1522*33de042dSApple OSS Distributions 		tmpString = OSString::withCString((const char *)propData, propDataLength);
1523*33de042dSApple OSS Distributions 		if (tmpString != nullptr) {
1524*33de042dSApple OSS Distributions 			tmpObject = tmpString;
1525*33de042dSApple OSS Distributions 		}
1526*33de042dSApple OSS Distributions 		break;
1527*33de042dSApple OSS Distributions 
1528*33de042dSApple OSS Distributions 	case kOFVariableTypeData:
1529*33de042dSApple OSS Distributions 		tmpObject = OSData::withBytes(propData, propDataLength);
1530*33de042dSApple OSS Distributions 		break;
1531*33de042dSApple OSS Distributions 
1532*33de042dSApple OSS Distributions 	default:
1533*33de042dSApple OSS Distributions 		break;
1534*33de042dSApple OSS Distributions 	}
1535*33de042dSApple OSS Distributions 
1536*33de042dSApple OSS Distributions 	if (tmpObject == nullptr) {
1537*33de042dSApple OSS Distributions 		tmpSymbol.reset();
1538*33de042dSApple OSS Distributions 		return false;
1539*33de042dSApple OSS Distributions 	}
1540*33de042dSApple OSS Distributions 
1541*33de042dSApple OSS Distributions 	propSymbol = tmpSymbol;
1542*33de042dSApple OSS Distributions 	propObject = tmpObject;
1543*33de042dSApple OSS Distributions 
1544*33de042dSApple OSS Distributions 	return true;
1545*33de042dSApple OSS Distributions }
1546