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