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