]>
Commit | Line | Data |
---|---|---|
878ddf1f | 1 | /*++\r |
2 | \r | |
3 | Copyright (c) 2006, Intel Corporation \r | |
4 | All rights reserved. This program and the accompanying materials \r | |
5 | are licensed and made available under the terms and conditions of the BSD License \r | |
6 | which accompanies this distribution. The full text of the license may be found at \r | |
7 | http://opensource.org/licenses/bsd-license.php \r | |
8 | \r | |
9 | THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, \r | |
10 | WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. \r | |
11 | \r | |
12 | Module Name:\r | |
13 | \r | |
14 | Forms.c\r | |
15 | \r | |
16 | Abstract:\r | |
17 | \r | |
18 | This file contains the form processing code to the HII database.\r | |
19 | \r | |
20 | --*/\r | |
21 | \r | |
22 | \r | |
23 | #include "HiiDatabase.h"\r | |
24 | \r | |
1cc8ee78 | 25 | STATIC\r |
878ddf1f | 26 | CHAR16*\r |
27 | Ascii2Unicode (\r | |
28 | OUT CHAR16 *UnicodeStr,\r | |
29 | IN CHAR8 *AsciiStr\r | |
30 | )\r | |
31 | /*++\r | |
32 | \r | |
33 | Routine Description:\r | |
34 | \r | |
35 | This function converts ASCII string to Unicode string.\r | |
36 | \r | |
37 | Arguments:\r | |
38 | \r | |
39 | UnicodeStr - NULL terminated Unicode output string.\r | |
40 | AsciieStr - NULL terminated ASCII input string.\r | |
41 | \r | |
42 | Returns: \r | |
43 | \r | |
44 | Start of the Unicode ouput string.\r | |
45 | \r | |
46 | --*/\r | |
47 | \r | |
48 | {\r | |
49 | CHAR16 *Str = UnicodeStr; \r | |
50 | while (TRUE) {\r | |
51 | *(UnicodeStr++) = (CHAR16) *AsciiStr;\r | |
52 | if (*(AsciiStr++) == '\0') {\r | |
53 | return Str;\r | |
54 | }\r | |
55 | }\r | |
56 | }\r | |
57 | \r | |
1cc8ee78 | 58 | STATIC\r |
878ddf1f | 59 | CHAR8*\r |
60 | Unicode2Ascii (\r | |
61 | OUT CHAR8 *AsciiStr,\r | |
62 | IN CHAR16 *UnicodeStr\r | |
63 | )\r | |
64 | /*++\r | |
65 | \r | |
66 | Routine Description:\r | |
67 | \r | |
68 | This function converts Unicode string to ASCII string.\r | |
69 | \r | |
70 | Arguments:\r | |
71 | \r | |
72 | AsciieStr - NULL terminated ASCII output string.\r | |
73 | UnicodeStr - NULL terminated Unicode input string.\r | |
74 | \r | |
75 | Returns: \r | |
76 | \r | |
77 | Start of the ASCII ouput string.\r | |
78 | \r | |
79 | --*/\r | |
80 | \r | |
81 | {\r | |
82 | CHAR8 *Str = AsciiStr; \r | |
83 | while (TRUE) {\r | |
84 | *(AsciiStr++) = (CHAR8) *UnicodeStr;\r | |
85 | if (*(UnicodeStr++) == '\0') {\r | |
86 | return Str;\r | |
87 | }\r | |
88 | }\r | |
89 | }\r | |
90 | \r | |
1cc8ee78 | 91 | STATIC\r |
878ddf1f | 92 | VOID\r |
93 | ExtractDevicePathData (\r | |
94 | IN EFI_HII_DATA_TABLE *DataTable,\r | |
95 | IN UINT8 *IfrData,\r | |
96 | IN OUT UINT8 **ExportBufferPtr\r | |
97 | )\r | |
98 | /*++\r | |
99 | \r | |
100 | Routine Description:\r | |
101 | \r | |
102 | Arguments:\r | |
103 | \r | |
104 | Returns: \r | |
105 | \r | |
106 | --*/\r | |
107 | {\r | |
108 | UINT8 *ExportBuffer;\r | |
109 | \r | |
110 | ExportBuffer = *ExportBufferPtr;\r | |
111 | \r | |
112 | //\r | |
113 | // BUGBUG - don't have devicepath data yet, setting dummy value\r | |
114 | //\r | |
115 | DataTable++;\r | |
116 | ExportBuffer = (UINT8 *) DataTable;\r | |
117 | ((EFI_HII_DEVICE_PATH_PACK *) ExportBuffer)->Header.Type = EFI_HII_DEVICE_PATH;\r | |
118 | ((EFI_HII_DEVICE_PATH_PACK *) ExportBuffer)->Header.Length = (UINT32) (sizeof (EFI_HII_DEVICE_PATH_PACK) + sizeof (EFI_DEVICE_PATH_PROTOCOL));\r | |
119 | \r | |
120 | //\r | |
121 | // BUGBUG - part of hack - skip the Device Path Pack.....place some data\r | |
122 | //\r | |
123 | ExportBuffer = ExportBuffer + sizeof (EFI_HII_DEVICE_PATH_PACK);\r | |
124 | \r | |
125 | ((EFI_DEVICE_PATH_PROTOCOL *) ExportBuffer)->Type = EFI_END_ENTIRE_DEVICE_PATH;\r | |
126 | ((EFI_DEVICE_PATH_PROTOCOL *) ExportBuffer)->SubType = EFI_END_ENTIRE_DEVICE_PATH_SUBTYPE;\r | |
127 | \r | |
128 | //\r | |
129 | // BUGBUG - still part of hack....\r | |
130 | //\r | |
131 | ExportBuffer = ExportBuffer + sizeof (EFI_DEVICE_PATH_PROTOCOL);\r | |
132 | *ExportBufferPtr = ExportBuffer;\r | |
133 | }\r | |
134 | \r | |
1cc8ee78 | 135 | STATIC\r |
878ddf1f | 136 | VOID\r |
137 | ExtractVariableData (\r | |
138 | IN OUT EFI_HII_DATA_TABLE *DataTable,\r | |
139 | IN UINT8 *IfrData,\r | |
140 | IN OUT UINT8 **ExportBufferPtr\r | |
141 | )\r | |
142 | /*++\r | |
143 | \r | |
144 | Routine Description:\r | |
145 | \r | |
146 | This function extract the EFI_HII_VARIABLE_PACK portion from the \r | |
147 | each of the EFI_HII_PACKAGE_INSTANCE in HII handle database.\r | |
148 | \r | |
149 | Arguments:\r | |
150 | \r | |
511710d6 | 151 | DataTable - On input, this parameter point to the EFI_HII_DATA_TABLE structure\r |
878ddf1f | 152 | of the final data buffer for the EFI_HII_EXPORT interface. This function\r |
153 | update the NumberOfVariableData attribute.\r | |
154 | IfrData - It points to a staring address of a EFI_HII_IFR_PACK structure.\r | |
511710d6 | 155 | ExportBufferPtr - On input, it points the starting address of the data buffer to \r |
878ddf1f | 156 | host the variable pack. On output, it is the starting address\r |
157 | of data buffer for the next extraction operation.\r | |
158 | Returns: \r | |
159 | \r | |
160 | VOID\r | |
161 | \r | |
162 | --*/\r | |
163 | {\r | |
164 | EFI_HII_VARIABLE_PACK *VariableContents;\r | |
165 | UINT8 *ExportBuffer;\r | |
166 | UINTN Index;\r | |
167 | UINTN Index2;\r | |
168 | UINTN TempValue;\r | |
169 | UINTN TempValue2;\r | |
170 | EFI_FORM_CALLBACK_PROTOCOL *FormCallback;\r | |
171 | EFI_PHYSICAL_ADDRESS CallbackHandle;\r | |
172 | EFI_STATUS Status;\r | |
173 | CHAR16 *String;\r | |
174 | \r | |
175 | FormCallback = NULL;\r | |
176 | CallbackHandle = 0;\r | |
177 | ExportBuffer = *ExportBufferPtr;\r | |
178 | \r | |
179 | for (Index = 0; IfrData[Index] != EFI_IFR_END_FORM_SET_OP;) {\r | |
180 | VariableContents = (EFI_HII_VARIABLE_PACK *) ExportBuffer;\r | |
181 | \r | |
182 | switch (IfrData[Index]) {\r | |
183 | case EFI_IFR_FORM_SET_OP:\r | |
184 | TempValue = EFI_HII_VARIABLE;\r | |
185 | CopyMem (&VariableContents->Header.Type, &TempValue, sizeof (UINT16));\r | |
186 | CopyMem (&TempValue, &((EFI_IFR_FORM_SET *) &IfrData[Index])->NvDataSize, sizeof (UINT16));\r | |
187 | \r | |
188 | //\r | |
189 | // If the variable has 0 size, do not process it\r | |
190 | //\r | |
191 | if (TempValue == 0) {\r | |
192 | break;\r | |
193 | }\r | |
194 | //\r | |
195 | // Add the size of the variable pack overhead. Later, will also add the size of the\r | |
196 | // name of the variable.\r | |
197 | //\r | |
198 | TempValue = TempValue + sizeof (EFI_HII_VARIABLE_PACK);\r | |
199 | \r | |
200 | CopyMem (&VariableContents->Header.Length, &TempValue, sizeof (UINT32));\r | |
201 | CopyMem (\r | |
202 | &CallbackHandle,\r | |
203 | &((EFI_IFR_FORM_SET *) &IfrData[Index])->CallbackHandle,\r | |
204 | sizeof (EFI_PHYSICAL_ADDRESS)\r | |
205 | );\r | |
206 | if (CallbackHandle != 0) {\r | |
207 | Status = gBS->HandleProtocol (\r | |
208 | (EFI_HANDLE) (UINTN) CallbackHandle,\r | |
209 | &gEfiFormCallbackProtocolGuid,\r | |
210 | (VOID *) &FormCallback\r | |
211 | );\r | |
1cc8ee78 | 212 | ASSERT_EFI_ERROR (Status);\r |
878ddf1f | 213 | }\r |
214 | //\r | |
215 | // Since we have a "Setup" variable that wasn't specified by a variable op-code\r | |
216 | // it will have a VariableId of 0. All other variable op-codes will have a designation\r | |
217 | // of VariableId 1+\r | |
218 | //\r | |
219 | TempValue = 0;\r | |
220 | CopyMem (&VariableContents->VariableId, &TempValue, sizeof (UINT16));\r | |
221 | CopyMem (&VariableContents->VariableGuid, &((EFI_IFR_FORM_SET *) &IfrData[Index])->Guid, sizeof (EFI_GUID));\r | |
222 | TempValue = sizeof (SETUP_MAP_NAME);\r | |
223 | CopyMem (&VariableContents->VariableNameLength, &TempValue, sizeof (UINT32));\r | |
224 | \r | |
225 | //\r | |
226 | // Add the size of the name to the Header Length\r | |
227 | //\r | |
228 | TempValue2 = 0;\r | |
229 | CopyMem (&TempValue2, &VariableContents->Header.Length, sizeof (UINT32));\r | |
230 | TempValue2 = TempValue + TempValue2;\r | |
231 | CopyMem (&VariableContents->Header.Length, &TempValue2, sizeof (UINT32));\r | |
232 | \r | |
233 | ExportBuffer = ExportBuffer + sizeof (EFI_HII_VARIABLE_PACK);\r | |
234 | CopyMem (ExportBuffer, SETUP_MAP_NAME, sizeof (SETUP_MAP_NAME));\r | |
235 | ExportBuffer = ExportBuffer + sizeof (SETUP_MAP_NAME);\r | |
236 | \r | |
237 | CopyMem (&TempValue, &((EFI_IFR_FORM_SET *) &IfrData[Index])->NvDataSize, sizeof (UINT16));\r | |
238 | \r | |
239 | if ((FormCallback != NULL) && (FormCallback->NvRead != NULL)) {\r | |
240 | Status = FormCallback->NvRead (\r | |
241 | FormCallback,\r | |
242 | (CHAR16 *) SETUP_MAP_NAME,\r | |
243 | (EFI_GUID *)(UINTN)&VariableContents->VariableGuid,\r | |
244 | NULL,\r | |
245 | &TempValue,\r | |
246 | ExportBuffer\r | |
247 | );\r | |
1cc8ee78 | 248 | ASSERT_EFI_ERROR (Status);\r |
878ddf1f | 249 | } else {\r |
250 | Status = gRT->GetVariable (\r | |
251 | (CHAR16 *) SETUP_MAP_NAME,\r | |
252 | (EFI_GUID *)(UINTN)&VariableContents->VariableGuid,\r | |
253 | NULL,\r | |
254 | &TempValue,\r | |
255 | ExportBuffer\r | |
256 | );\r | |
1cc8ee78 | 257 | ASSERT_EFI_ERROR (Status);\r |
878ddf1f | 258 | }\r |
259 | \r | |
260 | ExportBuffer = (UINT8 *) (UINTN) (((UINTN) ExportBuffer) + TempValue);\r | |
261 | DataTable->NumberOfVariableData++;\r | |
262 | break;\r | |
263 | \r | |
264 | case EFI_IFR_VARSTORE_OP:\r | |
265 | TempValue = EFI_HII_VARIABLE;\r | |
266 | CopyMem (&VariableContents->Header.Type, &TempValue, sizeof (UINT16));\r | |
267 | CopyMem (&TempValue, &((EFI_IFR_VARSTORE *) &IfrData[Index])->Size, sizeof (UINT16));\r | |
268 | \r | |
269 | //\r | |
270 | // If the variable has 0 size, do not process it\r | |
271 | //\r | |
272 | if (TempValue == 0) {\r | |
273 | break;\r | |
274 | }\r | |
275 | //\r | |
276 | // Add the size of the variable pack overhead. Later, will also add the size of the\r | |
277 | // name of the variable.\r | |
278 | //\r | |
279 | TempValue = TempValue + sizeof (EFI_HII_VARIABLE_PACK);\r | |
280 | \r | |
281 | CopyMem (&VariableContents->Header.Length, &TempValue, sizeof (UINT32));\r | |
282 | CopyMem (&VariableContents->VariableId, &((EFI_IFR_VARSTORE *) &IfrData[Index])->VarId, sizeof (UINT16));\r | |
283 | CopyMem (&VariableContents->VariableGuid, &((EFI_IFR_VARSTORE *) &IfrData[Index])->Guid, sizeof (EFI_GUID));\r | |
284 | TempValue = (UINTN) ((EFI_IFR_VARSTORE *) &IfrData[Index])->Header.Length - sizeof (EFI_IFR_VARSTORE);\r | |
285 | TempValue = TempValue * 2;\r | |
286 | CopyMem (&VariableContents->VariableNameLength, &TempValue, sizeof (UINT32));\r | |
287 | \r | |
288 | //\r | |
289 | // Add the size of the name to the Header Length\r | |
290 | //\r | |
291 | TempValue2 = 0;\r | |
292 | CopyMem (&TempValue2, &VariableContents->Header.Length, sizeof (UINT32));\r | |
293 | TempValue2 = TempValue + TempValue2;\r | |
294 | CopyMem (&VariableContents->Header.Length, &TempValue2, sizeof (UINT32));\r | |
295 | \r | |
296 | ExportBuffer = ExportBuffer + sizeof (EFI_HII_VARIABLE_PACK);\r | |
297 | String = (CHAR16 *) ExportBuffer;\r | |
298 | for (Index2 = 0; Index2 < TempValue / 2; Index2++) {\r | |
299 | ExportBuffer[Index2 * 2] = IfrData[Index + sizeof (EFI_IFR_VARSTORE) + Index2];\r | |
300 | ExportBuffer[Index2 * 2 + 1] = 0;\r | |
301 | }\r | |
302 | \r | |
303 | ExportBuffer = ExportBuffer + TempValue;\r | |
304 | \r | |
305 | CopyMem (&TempValue, &((EFI_IFR_VARSTORE *) &IfrData[Index])->Size, sizeof (UINT16));\r | |
306 | \r | |
307 | if ((FormCallback != NULL) && (FormCallback->NvRead != NULL)) {\r | |
308 | Status = FormCallback->NvRead (\r | |
309 | FormCallback,\r | |
310 | String,\r | |
311 | (EFI_GUID *)(UINTN)&VariableContents->VariableGuid,\r | |
312 | NULL,\r | |
313 | &TempValue,\r | |
314 | ExportBuffer\r | |
315 | );\r | |
1cc8ee78 | 316 | ASSERT_EFI_ERROR (Status);\r |
878ddf1f | 317 | } else {\r |
318 | Status = gRT->GetVariable (\r | |
319 | String,\r | |
320 | (EFI_GUID *)(UINTN)&VariableContents->VariableGuid,\r | |
321 | NULL,\r | |
322 | &TempValue,\r | |
323 | ExportBuffer\r | |
324 | );\r | |
1cc8ee78 | 325 | ASSERT_EFI_ERROR (Status);\r |
878ddf1f | 326 | }\r |
327 | \r | |
328 | ExportBuffer = (UINT8 *) (UINTN) (((UINTN) ExportBuffer) + TempValue);\r | |
329 | DataTable->NumberOfVariableData++;\r | |
330 | break;\r | |
331 | }\r | |
332 | \r | |
333 | Index = IfrData[Index + 1] + Index;\r | |
334 | }\r | |
335 | //\r | |
336 | // If we have added a variable pack, add a dummy empty one to signify the end\r | |
337 | //\r | |
338 | if (ExportBuffer != *ExportBufferPtr) {\r | |
339 | VariableContents = (EFI_HII_VARIABLE_PACK *) ExportBuffer;\r | |
340 | TempValue = EFI_HII_VARIABLE;\r | |
341 | CopyMem (&VariableContents->Header.Type, &TempValue, sizeof (UINT16));\r | |
342 | TempValue = sizeof (EFI_HII_VARIABLE_PACK);\r | |
343 | CopyMem (&VariableContents->Header.Length, &TempValue, sizeof (UINT32));\r | |
344 | ExportBuffer = ExportBuffer + sizeof (EFI_HII_VARIABLE_PACK);\r | |
345 | }\r | |
346 | \r | |
347 | *ExportBufferPtr = ExportBuffer;\r | |
348 | }\r | |
349 | \r | |
350 | EFI_STATUS\r | |
351 | EFIAPI\r | |
352 | HiiExportDatabase (\r | |
353 | IN EFI_HII_PROTOCOL *This,\r | |
354 | IN EFI_HII_HANDLE Handle,\r | |
355 | IN OUT UINTN *BufferSize,\r | |
356 | OUT VOID *Buffer\r | |
357 | )\r | |
358 | /*++\r | |
359 | \r | |
360 | Routine Description:\r | |
361 | \r | |
362 | This function allows a program to extract a form or form package that has \r | |
363 | previously been registered with the EFI HII database.\r | |
364 | \r | |
365 | Arguments:\r | |
366 | \r | |
367 | Returns: \r | |
368 | \r | |
369 | --*/\r | |
370 | {\r | |
371 | EFI_HII_PACKAGE_INSTANCE *PackageInstance;\r | |
372 | EFI_HII_DATA *HiiData;\r | |
373 | EFI_HII_HANDLE_DATABASE *HandleDatabase;\r | |
374 | EFI_HII_IFR_PACK *FormPack;\r | |
375 | UINT8 *RawData;\r | |
376 | UINT8 *ExportBuffer;\r | |
377 | EFI_HII_EXPORT_TABLE *ExportTable;\r | |
378 | EFI_HII_DATA_TABLE *DataTable;\r | |
878ddf1f | 379 | BOOLEAN VariableExist;\r |
380 | UINT16 NumberOfHiiDataTables;\r | |
381 | UINTN SizeNeeded;\r | |
382 | UINTN Index;\r | |
383 | UINTN VariableSize;\r | |
384 | UINTN TempValue;\r | |
385 | \r | |
386 | if (This == NULL) {\r | |
387 | return EFI_INVALID_PARAMETER;\r | |
388 | }\r | |
389 | \r | |
390 | HiiData = EFI_HII_DATA_FROM_THIS (This);\r | |
391 | \r | |
392 | HandleDatabase = HiiData->DatabaseHead;\r | |
393 | \r | |
394 | FormPack = NULL;\r | |
395 | RawData = NULL;\r | |
396 | PackageInstance = NULL;\r | |
878ddf1f | 397 | NumberOfHiiDataTables = 0;\r |
398 | VariableSize = 0;\r | |
399 | TempValue = 0;\r | |
400 | SizeNeeded = sizeof (EFI_HII_EXPORT_TABLE);\r | |
401 | \r | |
402 | //\r | |
403 | // How many total tables are there?\r | |
404 | //\r | |
405 | for (; HandleDatabase != NULL; HandleDatabase = HandleDatabase->NextHandleDatabase) {\r | |
406 | if ((Handle != 0) && (Handle != HandleDatabase->Handle)) {\r | |
407 | continue;\r | |
408 | }\r | |
409 | \r | |
410 | VariableExist = FALSE;\r | |
411 | NumberOfHiiDataTables++;\r | |
412 | PackageInstance = HandleDatabase->Buffer;\r | |
413 | if (PackageInstance == NULL) {\r | |
414 | continue;\r | |
415 | }\r | |
416 | //\r | |
417 | // Extract Size of Export Package\r | |
418 | //\r | |
419 | SizeNeeded = SizeNeeded + PackageInstance->IfrSize \r | |
420 | + PackageInstance->StringSize\r | |
421 | + sizeof (EFI_HII_DATA_TABLE)\r | |
422 | + sizeof (EFI_HII_DEVICE_PATH_PACK);\r | |
423 | \r | |
424 | //\r | |
425 | // BUGBUG We aren't inserting Device path data yet\r | |
426 | //\r | |
427 | SizeNeeded = SizeNeeded + sizeof (EFI_DEVICE_PATH_PROTOCOL);\r | |
428 | \r | |
429 | //\r | |
430 | // Extract Size of Variable Data\r | |
431 | //\r | |
432 | if (PackageInstance->IfrSize > 0) {\r | |
433 | FormPack = (EFI_HII_IFR_PACK *) ((CHAR8 *) (&PackageInstance->IfrData) + sizeof (EFI_HII_PACK_HEADER));\r | |
434 | } else {\r | |
435 | //\r | |
436 | // No IFR? No variable information\r | |
437 | //\r | |
438 | continue;\r | |
439 | }\r | |
440 | \r | |
441 | RawData = (UINT8 *) FormPack;\r | |
442 | \r | |
443 | for (Index = 0; RawData[Index] != EFI_IFR_END_FORM_SET_OP;) {\r | |
444 | switch (RawData[Index]) {\r | |
445 | case EFI_IFR_FORM_SET_OP:\r | |
446 | CopyMem (&VariableSize, &((EFI_IFR_FORM_SET *) &RawData[Index])->NvDataSize, sizeof (UINT16));\r | |
447 | SizeNeeded = SizeNeeded + VariableSize + sizeof (SETUP_MAP_NAME) + sizeof (EFI_HII_VARIABLE_PACK);\r | |
448 | VariableExist = TRUE;\r | |
449 | break;\r | |
450 | \r | |
451 | case EFI_IFR_VARSTORE_OP:\r | |
452 | CopyMem (&VariableSize, &((EFI_IFR_VARSTORE *) &RawData[Index])->Size, sizeof (UINT16));\r | |
453 | SizeNeeded = SizeNeeded + VariableSize + sizeof (EFI_HII_VARIABLE_PACK);\r | |
454 | //\r | |
455 | // We will be expanding the stored ASCII name to a Unicode string. This will cause some memory overhead\r | |
456 | // Since the VARSTORE size already takes in consideration the ASCII size, we need to size it and add another\r | |
457 | // instance of it. Essentially, 2 ASCII strings == 1 Unicode string in size.\r | |
458 | //\r | |
459 | TempValue = (UINTN) ((EFI_IFR_VARSTORE *) &RawData[Index])->Header.Length - sizeof (EFI_IFR_VARSTORE);\r | |
460 | SizeNeeded = SizeNeeded + TempValue * 2;\r | |
461 | VariableExist = TRUE;\r | |
462 | break;\r | |
463 | }\r | |
464 | \r | |
465 | Index = RawData[Index + 1] + Index;\r | |
466 | }\r | |
467 | //\r | |
468 | // If a variable exists for this handle, add an additional variable pack overhead to\r | |
469 | // indicate that we will have an extra null Variable Pack to signify the end of the Variable Packs\r | |
470 | //\r | |
471 | if (VariableExist) {\r | |
472 | SizeNeeded = SizeNeeded + sizeof (EFI_HII_VARIABLE_PACK);\r | |
473 | }\r | |
474 | }\r | |
475 | \r | |
476 | if (SizeNeeded > *BufferSize) {\r | |
477 | *BufferSize = SizeNeeded;\r | |
478 | return EFI_BUFFER_TOO_SMALL;\r | |
479 | }\r | |
480 | //\r | |
481 | // Zero out the incoming buffer\r | |
482 | //\r | |
483 | ZeroMem (Buffer, *BufferSize);\r | |
484 | \r | |
485 | //\r | |
486 | // Cast the Buffer to EFI_HII_EXPORT_TABLE\r | |
487 | //\r | |
488 | ExportTable = (EFI_HII_EXPORT_TABLE *) Buffer;\r | |
489 | \r | |
490 | //\r | |
491 | // Set the Revision for the Export Table\r | |
492 | //\r | |
493 | CopyMem (&ExportTable->Revision, &gEfiHiiProtocolGuid, sizeof (EFI_GUID));\r | |
494 | \r | |
495 | ExportBuffer = (UINT8 *) (UINTN) (((UINT8 *) ExportTable) + sizeof (EFI_HII_EXPORT_TABLE));\r | |
496 | HandleDatabase = HiiData->DatabaseHead;\r | |
497 | \r | |
498 | //\r | |
499 | // Check numeric value against the head of the database\r | |
500 | //\r | |
501 | for (; HandleDatabase != NULL; HandleDatabase = HandleDatabase->NextHandleDatabase) {\r | |
502 | DataTable = (EFI_HII_DATA_TABLE *) ExportBuffer;\r | |
503 | PackageInstance = HandleDatabase->Buffer;\r | |
504 | //\r | |
505 | // If not asking for a specific handle, export the entire database\r | |
506 | //\r | |
507 | if (Handle == 0) {\r | |
508 | ExportTable->NumberOfHiiDataTables = NumberOfHiiDataTables;\r | |
509 | CopyMem (&DataTable->PackageGuid, &PackageInstance->Guid, sizeof (EFI_GUID));\r | |
510 | DataTable->HiiHandle = PackageInstance->Handle;\r | |
511 | DataTable->DevicePathOffset = (UINT32) (sizeof (EFI_HII_DATA_TABLE));\r | |
512 | \r | |
513 | //\r | |
514 | // Start Dumping DevicePath\r | |
515 | //\r | |
516 | ExtractDevicePathData (DataTable, RawData, &ExportBuffer);\r | |
517 | \r | |
518 | if (((UINTN) ExportBuffer) == ((UINTN) DataTable)) {\r | |
519 | //\r | |
520 | // If there is no DevicePath information - set offset to 0 to signify the absence of data to parse\r | |
521 | //\r | |
522 | DataTable->DevicePathOffset = 0;\r | |
523 | }\r | |
524 | \r | |
525 | DataTable->VariableDataOffset = (UINT32) (((UINTN) ExportBuffer) - ((UINTN) DataTable));\r | |
526 | \r | |
527 | if (PackageInstance->IfrSize > 0) {\r | |
528 | FormPack = (EFI_HII_IFR_PACK *) ((CHAR8 *) (&PackageInstance->IfrData) + sizeof (EFI_HII_PACK_HEADER));\r | |
529 | \r | |
530 | RawData = (UINT8 *) FormPack;\r | |
531 | TempValue = 0;\r | |
532 | \r | |
533 | //\r | |
534 | // Start dumping the Variable Data\r | |
535 | //\r | |
536 | ExtractVariableData (DataTable, RawData, &ExportBuffer);\r | |
537 | DataTable->IfrDataOffset = (UINT32) (((UINTN) ExportBuffer) - ((UINTN) DataTable));\r | |
538 | \r | |
539 | if (DataTable->VariableDataOffset == DataTable->IfrDataOffset) {\r | |
540 | DataTable->VariableDataOffset = 0;\r | |
541 | }\r | |
542 | //\r | |
543 | // Start dumping the IFR data (Note: It is in an IFR PACK)\r | |
544 | //\r | |
545 | CopyMem (ExportBuffer, &PackageInstance->IfrData, PackageInstance->IfrSize);\r | |
546 | ExportBuffer = (UINT8 *) (UINTN) (((UINTN) ExportBuffer) + PackageInstance->IfrSize);\r | |
547 | DataTable->StringDataOffset = (UINT32) (((UINTN) ExportBuffer) - ((UINTN) DataTable));\r | |
548 | \r | |
549 | //\r | |
550 | // Start dumping the String data (Note: It is in a String PACK)\r | |
551 | //\r | |
552 | if (PackageInstance->StringSize > 0) {\r | |
553 | RawData = (UINT8 *) (((UINTN) &PackageInstance->IfrData) + PackageInstance->IfrSize);\r | |
554 | CopyMem (ExportBuffer, RawData, PackageInstance->StringSize);\r | |
555 | DataTable->DataTableSize = (UINT32) (DataTable->StringDataOffset + PackageInstance->StringSize);\r | |
556 | \r | |
557 | CopyMem (&TempValue, &((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length, sizeof (UINT32));\r | |
558 | for (; TempValue != 0;) {\r | |
559 | DataTable->NumberOfLanguages++;\r | |
560 | ExportBuffer = ExportBuffer + ((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length;\r | |
561 | CopyMem (&TempValue, &((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length, sizeof (UINT32));\r | |
562 | }\r | |
563 | \r | |
564 | ExportBuffer = ExportBuffer + sizeof (EFI_HII_STRING_PACK);\r | |
565 | } else {\r | |
566 | DataTable->StringDataOffset = 0;\r | |
567 | }\r | |
568 | } else {\r | |
569 | //\r | |
570 | // No IFR? No variable information. If Offset is 0, means there is none. (Hmm - this might be prunable - no strings to export if no IFR - we always have a stub)\r | |
571 | //\r | |
572 | DataTable->VariableDataOffset = 0;\r | |
573 | DataTable->IfrDataOffset = 0;\r | |
574 | DataTable->StringDataOffset = (UINT32) (((UINTN) ExportBuffer) - ((UINTN) DataTable));\r | |
575 | \r | |
576 | //\r | |
577 | // Start dumping the String data - NOTE: It is in String Pack form\r | |
578 | //\r | |
579 | if (PackageInstance->StringSize > 0) {\r | |
580 | RawData = (UINT8 *) (((UINTN) &PackageInstance->IfrData) + PackageInstance->IfrSize);\r | |
581 | CopyMem (ExportBuffer, RawData, PackageInstance->StringSize);\r | |
582 | DataTable->DataTableSize = (UINT32) (DataTable->StringDataOffset + PackageInstance->StringSize);\r | |
583 | \r | |
584 | CopyMem (&TempValue, &((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length, sizeof (UINT32));\r | |
585 | for (; TempValue != 0;) {\r | |
586 | DataTable->NumberOfLanguages++;\r | |
587 | ExportBuffer = ExportBuffer + ((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length;\r | |
588 | CopyMem (&TempValue, &((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length, sizeof (UINT32));\r | |
589 | }\r | |
590 | \r | |
591 | ExportBuffer = ExportBuffer + sizeof (EFI_HII_STRING_PACK);\r | |
592 | } else {\r | |
593 | DataTable->StringDataOffset = 0;\r | |
594 | }\r | |
595 | }\r | |
596 | } else {\r | |
597 | //\r | |
598 | // Match the numeric value with the database entry - if matched, extract PackageInstance\r | |
599 | //\r | |
600 | if (Handle == HandleDatabase->Handle) {\r | |
601 | PackageInstance = HandleDatabase->Buffer;\r | |
602 | ExportTable->NumberOfHiiDataTables = NumberOfHiiDataTables;\r | |
603 | DataTable->HiiHandle = PackageInstance->Handle;\r | |
604 | CopyMem (&DataTable->PackageGuid, &PackageInstance->Guid, sizeof (EFI_GUID));\r | |
605 | \r | |
606 | //\r | |
607 | // Start Dumping DevicePath\r | |
608 | //\r | |
609 | ExtractDevicePathData (DataTable, RawData, &ExportBuffer);\r | |
610 | DataTable->VariableDataOffset = (UINT32) (((UINTN) ExportBuffer) - ((UINTN) DataTable));\r | |
611 | \r | |
612 | if (PackageInstance->IfrSize > 0) {\r | |
613 | FormPack = (EFI_HII_IFR_PACK *) ((CHAR8 *) (&PackageInstance->IfrData) + sizeof (EFI_HII_PACK_HEADER));\r | |
614 | \r | |
615 | RawData = (UINT8 *) FormPack;\r | |
616 | TempValue = 0;\r | |
617 | \r | |
618 | //\r | |
619 | // Start dumping the Variable Data\r | |
620 | //\r | |
621 | ExtractVariableData (DataTable, RawData, &ExportBuffer);\r | |
622 | DataTable->IfrDataOffset = (UINT32) (((UINTN) ExportBuffer) - ((UINTN) DataTable));\r | |
623 | \r | |
624 | if (DataTable->VariableDataOffset == DataTable->IfrDataOffset) {\r | |
625 | DataTable->VariableDataOffset = 0;\r | |
626 | }\r | |
627 | //\r | |
628 | // Start dumping the IFR data\r | |
629 | //\r | |
630 | CopyMem (ExportBuffer, &PackageInstance->IfrData, PackageInstance->IfrSize);\r | |
631 | ExportBuffer = (UINT8 *) (UINTN) (((UINTN) ExportBuffer) + PackageInstance->IfrSize);\r | |
632 | DataTable->StringDataOffset = (UINT32) (((UINTN) ExportBuffer) - ((UINTN) DataTable));\r | |
633 | \r | |
634 | //\r | |
635 | // Start dumping the String data - NOTE: It is in String Pack form\r | |
636 | //\r | |
637 | if (PackageInstance->StringSize > 0) {\r | |
638 | RawData = (UINT8 *) (((UINTN) &PackageInstance->IfrData) + PackageInstance->IfrSize);\r | |
639 | CopyMem (ExportBuffer, RawData, PackageInstance->StringSize);\r | |
640 | DataTable->DataTableSize = (UINT32) (DataTable->StringDataOffset + PackageInstance->StringSize);\r | |
641 | \r | |
642 | CopyMem (&TempValue, &((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length, sizeof (UINT32));\r | |
643 | for (; TempValue != 0;) {\r | |
644 | DataTable->NumberOfLanguages++;\r | |
645 | ExportBuffer = ExportBuffer + ((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length;\r | |
646 | CopyMem (&TempValue, &((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length, sizeof (UINT32));\r | |
647 | }\r | |
648 | \r | |
649 | ExportBuffer = ExportBuffer + sizeof (EFI_HII_STRING_PACK);\r | |
650 | } else {\r | |
651 | DataTable->StringDataOffset = 0;\r | |
652 | }\r | |
653 | } else {\r | |
654 | //\r | |
655 | // No IFR? No variable information. If Offset is 0, means there is none.\r | |
656 | //\r | |
657 | DataTable->VariableDataOffset = 0;\r | |
658 | DataTable->IfrDataOffset = 0;\r | |
659 | DataTable->StringDataOffset = (UINT32) (((UINTN) ExportBuffer) - ((UINTN) DataTable));\r | |
660 | \r | |
661 | //\r | |
662 | // Start dumping the String data - Note: It is in String Pack form\r | |
663 | //\r | |
664 | if (PackageInstance->StringSize > 0) {\r | |
665 | RawData = (UINT8 *) (((UINTN) &PackageInstance->IfrData) + PackageInstance->IfrSize);\r | |
666 | CopyMem (ExportBuffer, RawData, PackageInstance->StringSize);\r | |
667 | DataTable->DataTableSize = (UINT32) (DataTable->StringDataOffset + PackageInstance->StringSize);\r | |
668 | \r | |
669 | CopyMem (&TempValue, &((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length, sizeof (UINT32));\r | |
670 | for (; TempValue != 0;) {\r | |
671 | DataTable->NumberOfLanguages++;\r | |
672 | ExportBuffer = ExportBuffer + ((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length;\r | |
673 | CopyMem (&TempValue, &((EFI_HII_STRING_PACK *) ExportBuffer)->Header.Length, sizeof (UINT32));\r | |
674 | }\r | |
675 | \r | |
676 | ExportBuffer = ExportBuffer + sizeof (EFI_HII_STRING_PACK);\r | |
677 | } else {\r | |
678 | DataTable->StringDataOffset = 0;\r | |
679 | }\r | |
680 | }\r | |
681 | break;\r | |
682 | }\r | |
683 | }\r | |
684 | }\r | |
685 | \r | |
686 | return EFI_SUCCESS;\r | |
687 | }\r | |
688 | \r | |
689 | EFI_STATUS\r | |
690 | EFIAPI\r | |
691 | HiiGetForms (\r | |
692 | IN EFI_HII_PROTOCOL *This,\r | |
693 | IN EFI_HII_HANDLE Handle,\r | |
694 | IN EFI_FORM_ID FormId,\r | |
695 | IN OUT UINTN *BufferLengthTemp,\r | |
696 | OUT UINT8 *Buffer\r | |
697 | )\r | |
698 | /*++\r | |
699 | \r | |
700 | Routine Description:\r | |
701 | \r | |
702 | This function allows a program to extract a form or form package that has \r | |
703 | previously been registered with the EFI HII database.\r | |
704 | \r | |
705 | Arguments:\r | |
706 | This - A pointer to the EFI_HII_PROTOCOL instance.\r | |
707 | \r | |
708 | Handle - Handle on which the form resides. Type EFI_HII_HANDLE is defined in \r | |
709 | EFI_HII_PROTOCOL.NewPack() in the Packages section.\r | |
710 | \r | |
711 | FormId - The ID of the form to return. If the ID is zero, the entire form package is returned.\r | |
712 | Type EFI_FORM_ID is defined in "Related Definitions" below.\r | |
713 | \r | |
714 | BufferLength - On input, the length of the Buffer. On output, the length of the returned buffer, if\r | |
715 | the length was sufficient and, if it was not, the length that is required to fit the\r | |
716 | requested form(s).\r | |
717 | \r | |
718 | Buffer - The buffer designed to receive the form(s).\r | |
719 | \r | |
720 | Returns: \r | |
721 | \r | |
722 | EFI_SUCCESS - Buffer filled with the requested forms. BufferLength\r | |
723 | was updated.\r | |
724 | \r | |
725 | EFI_INVALID_PARAMETER - The handle is unknown.\r | |
726 | \r | |
727 | EFI_NOT_FOUND - A form on the requested handle cannot be found with the\r | |
728 | requested FormId.\r | |
729 | \r | |
730 | EFI_BUFFER_TOO_SMALL - The buffer provided was not large enough to allow the form to be stored.\r | |
731 | \r | |
732 | --*/\r | |
733 | {\r | |
734 | EFI_HII_PACKAGE_INSTANCE *PackageInstance;\r | |
735 | EFI_HII_DATA *HiiData;\r | |
736 | EFI_HII_HANDLE_DATABASE *HandleDatabase;\r | |
737 | EFI_HII_IFR_PACK *FormPack;\r | |
738 | EFI_IFR_FORM *Form;\r | |
739 | EFI_IFR_OP_HEADER *Location;\r | |
740 | UINT16 *BufferLength = (UINT16 *) BufferLengthTemp;\r | |
741 | \r | |
742 | if (This == NULL) {\r | |
743 | return EFI_INVALID_PARAMETER;\r | |
744 | }\r | |
745 | \r | |
746 | HiiData = EFI_HII_DATA_FROM_THIS (This);\r | |
747 | \r | |
748 | HandleDatabase = HiiData->DatabaseHead;\r | |
749 | \r | |
750 | PackageInstance = NULL;\r | |
751 | \r | |
752 | //\r | |
753 | // Check numeric value against the head of the database\r | |
754 | //\r | |
755 | for (; HandleDatabase != NULL; HandleDatabase = HandleDatabase->NextHandleDatabase) {\r | |
756 | //\r | |
757 | // Match the numeric value with the database entry - if matched, extract PackageInstance\r | |
758 | //\r | |
759 | if (Handle == HandleDatabase->Handle) {\r | |
760 | PackageInstance = HandleDatabase->Buffer;\r | |
761 | break;\r | |
762 | }\r | |
763 | }\r | |
764 | //\r | |
765 | // No handle was found - error condition\r | |
766 | //\r | |
767 | if (PackageInstance == NULL) {\r | |
768 | return EFI_NOT_FOUND;\r | |
769 | }\r | |
770 | //\r | |
771 | // Based on if there is IFR data in this package instance, determine\r | |
772 | // what the location is of the beginning of the string data.\r | |
773 | //\r | |
774 | if (PackageInstance->IfrSize > 0) {\r | |
775 | FormPack = (EFI_HII_IFR_PACK *) (&PackageInstance->IfrData);\r | |
776 | } else {\r | |
777 | //\r | |
778 | // If there is no IFR data return an error\r | |
779 | //\r | |
780 | return EFI_NOT_FOUND;\r | |
781 | }\r | |
782 | //\r | |
783 | // If requesting the entire Form Package\r | |
784 | //\r | |
785 | if (FormId == 0) {\r | |
786 | //\r | |
787 | // Return an error if buffer is too small\r | |
788 | //\r | |
789 | if (PackageInstance->IfrSize > *BufferLength || Buffer == NULL) {\r | |
790 | *BufferLength = (UINT16) PackageInstance->IfrSize;\r | |
791 | return EFI_BUFFER_TOO_SMALL;\r | |
792 | }\r | |
793 | \r | |
794 | CopyMem (Buffer, FormPack, PackageInstance->IfrSize);\r | |
795 | return EFI_SUCCESS;\r | |
796 | } else {\r | |
797 | FormPack = (EFI_HII_IFR_PACK *) ((CHAR8 *) (&PackageInstance->IfrData) + sizeof (EFI_HII_PACK_HEADER));\r | |
798 | Location = (EFI_IFR_OP_HEADER *) FormPack;\r | |
799 | \r | |
800 | //\r | |
801 | // Look for the FormId requested\r | |
802 | //\r | |
803 | for (; Location->OpCode != EFI_IFR_END_FORM_SET_OP;) {\r | |
804 | switch (Location->OpCode) {\r | |
805 | case EFI_IFR_FORM_OP:\r | |
806 | Form = (EFI_IFR_FORM *) Location;\r | |
807 | \r | |
808 | //\r | |
809 | // If we found a Form Op-code and it is of the correct Id, copy it and return\r | |
810 | //\r | |
811 | if (Form->FormId == FormId) {\r | |
812 | if (Location->Length > *BufferLength || Buffer == NULL) {\r | |
813 | *BufferLength = Location->Length;\r | |
814 | return EFI_BUFFER_TOO_SMALL;\r | |
815 | } else {\r | |
816 | for (; Location->OpCode != EFI_IFR_END_FORM_OP;) {\r | |
817 | CopyMem (Buffer, Location, Location->Length);\r | |
818 | Buffer = Buffer + Location->Length;\r | |
819 | Location = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (Location) + Location->Length);\r | |
820 | }\r | |
821 | \r | |
822 | CopyMem (Buffer, Location, Location->Length);\r | |
823 | return EFI_SUCCESS;\r | |
824 | }\r | |
825 | }\r | |
826 | \r | |
827 | default:\r | |
828 | break;\r | |
829 | }\r | |
830 | //\r | |
831 | // Go to the next Op-Code\r | |
832 | //\r | |
833 | Location = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (Location) + Location->Length);\r | |
834 | }\r | |
835 | }\r | |
836 | \r | |
837 | return EFI_SUCCESS;\r | |
838 | }\r | |
839 | \r | |
840 | //\r | |
841 | // Helper functions to HiiGetDefaultImage()\r | |
842 | //\r | |
843 | \r | |
844 | STATIC\r | |
845 | UINT8*\r | |
846 | HiiGetDefaultImageInitPack (\r | |
847 | IN OUT EFI_HII_VARIABLE_PACK_LIST *VariablePackItem,\r | |
848 | IN EFI_IFR_VARSTORE *VarStore\r | |
849 | )\r | |
850 | /*++\r | |
851 | \r | |
852 | Routine Description:\r | |
853 | \r | |
854 | Initialize the EFI_HII_VARIABLE_PACK_LIST structure and\r | |
855 | prepare it ready to be used by HiiGetDefaultImagePopulateMap ().\r | |
856 | \r | |
857 | Arguments:\r | |
858 | \r | |
859 | VariablePackItem - Variable Package List.\r | |
860 | VarStore - IFR variable storage.\r | |
861 | \r | |
862 | Returns: \r | |
863 | \r | |
864 | Return the pointer to the Map space.\r | |
865 | \r | |
866 | --*/\r | |
867 | {\r | |
868 | CHAR16 *Name16;\r | |
869 | CHAR8 *Name8;\r | |
870 | CHAR8 *Map;\r | |
871 | EFI_HII_VARIABLE_PACK *VariablePack;\r | |
872 | \r | |
873 | //\r | |
874 | // Set pointer the pack right after the node\r | |
875 | //\r | |
876 | VariablePackItem->VariablePack = (EFI_HII_VARIABLE_PACK *) (VariablePackItem + 1);\r | |
877 | VariablePack = VariablePackItem->VariablePack;\r | |
878 | \r | |
879 | //\r | |
880 | // Copy the var name to VariablePackItem from VarStore\r | |
881 | // Needs ASCII->Unicode conversion.\r | |
882 | //\r | |
883 | ASSERT (VarStore->Header.Length > sizeof (*VarStore));\r | |
884 | Name8 = (CHAR8 *) (VarStore + 1);\r | |
885 | Name16 = (CHAR16 *) (VariablePack + 1);\r | |
886 | Ascii2Unicode (Name16, Name8);\r | |
887 | \r | |
888 | //\r | |
889 | // Compute the other fields of the VariablePackItem\r | |
890 | //\r | |
891 | VariablePack->VariableId = VarStore->VarId;\r | |
892 | CopyMem (&VariablePack->VariableGuid, &VarStore->Guid, sizeof (EFI_GUID));\r | |
893 | VariablePack->VariableNameLength = (UINT32) ((StrLen (Name16) + 1) * 2);\r | |
894 | VariablePack->Header.Length = sizeof (*VariablePack) \r | |
895 | + VariablePack->VariableNameLength\r | |
896 | + VarStore->Size;\r | |
897 | //\r | |
898 | // Return the pointer to the Map space.\r | |
899 | //\r | |
900 | Map = (CHAR8 *) Name16 + VariablePack->VariableNameLength;\r | |
901 | \r | |
902 | return (UINT8 *)Map;\r | |
903 | }\r | |
904 | \r | |
905 | STATIC\r | |
906 | VOID\r | |
907 | HiiGetDefaultImagePopulateMap (\r | |
908 | IN OUT UINT8 *Map, \r | |
909 | IN EFI_IFR_OP_HEADER *FormSet,\r | |
910 | IN EFI_IFR_VARSTORE *VarStore,\r | |
911 | IN UINTN DefaultMask\r | |
912 | )\r | |
913 | /*++\r | |
914 | \r | |
915 | Routine Description:\r | |
916 | \r | |
917 | Fill the Map with all the default values either from NV or Hii database.\r | |
918 | \r | |
919 | Arguments:\r | |
920 | \r | |
921 | Map - Memory pointer to hold the default values.\r | |
922 | FormSet - The starting EFI_IFR_OP_HEADER to begin retriving default values.\r | |
923 | VarStore - IFR variable storage.\r | |
924 | DefaultMask - The mask used to get the default variable.\r | |
925 | \r | |
926 | Returns: \r | |
927 | \r | |
928 | VOID\r | |
929 | \r | |
930 | --*/\r | |
931 | {\r | |
932 | EFI_STATUS Status;\r | |
933 | EFI_IFR_OP_HEADER *IfrItem;\r | |
934 | UINT16 VarId;\r | |
935 | EFI_IFR_VARSTORE_SELECT *VarSelect;\r | |
936 | EFI_IFR_ONE_OF_OPTION *OneOfOpt;\r | |
937 | EFI_IFR_CHECKBOX *CheckBox;\r | |
938 | EFI_IFR_NUMERIC *Numeric;\r | |
939 | UINTN Size;\r | |
940 | UINTN SizeTmp;\r | |
941 | EFI_IFR_NV_DATA *IfrNvData;\r | |
942 | EFI_GUID Guid;\r | |
943 | CHAR16 *Name16;\r | |
944 | CHAR8 *Name8; \r | |
945 | EFI_HANDLE CallbackHandle;\r | |
946 | EFI_FORM_CALLBACK_PROTOCOL *FormCallbackProt;\r | |
947 | \r | |
948 | //\r | |
949 | // Get the Map's Name/Guid/Szie from the Varstore.\r | |
950 | // VARSTORE contains the Name in ASCII format (@#$^&!), must convert it to Unicode.\r | |
951 | //\r | |
952 | ASSERT (VarStore->Header.Length >= sizeof (*VarStore));\r | |
953 | Name8 = (CHAR8 *) (VarStore + 1);\r | |
954 | Name16 = AllocateZeroPool ((VarStore->Header.Length - sizeof (*VarStore)) * sizeof (CHAR16));\r | |
955 | Ascii2Unicode (Name16, Name8);\r | |
956 | CopyMem (&Guid, &VarStore->Guid, sizeof(EFI_GUID));\r | |
957 | Size = VarStore->Size;\r | |
958 | \r | |
959 | //\r | |
960 | // First, check if the map exists in the NV. If so, get it from NV and exit.\r | |
961 | //\r | |
962 | if (DefaultMask == EFI_IFR_FLAG_MANUFACTURING) {\r | |
963 | //\r | |
964 | // Check if Manufaturing Defaults exist in the NV.\r | |
965 | //\r | |
966 | Status = EfiLibHiiVariableOverrideBySuffix (\r | |
967 | HII_VARIABLE_SUFFIX_MANUFACTURING_OVERRIDE,\r | |
968 | Name16,\r | |
969 | &Guid,\r | |
970 | Size,\r | |
971 | Map\r | |
972 | );\r | |
973 | } else {\r | |
974 | //\r | |
975 | // All other cases default to Defaults. Check if Defaults exist in the NV.\r | |
976 | //\r | |
977 | Status = EfiLibHiiVariableOverrideBySuffix (\r | |
978 | HII_VARIABLE_SUFFIX_DEFAULT_OVERRIDE,\r | |
979 | Name16,\r | |
980 | &Guid,\r | |
981 | Size,\r | |
982 | Map\r | |
983 | );\r | |
984 | }\r | |
985 | if (!EFI_ERROR (Status)) {\r | |
986 | //\r | |
987 | // Either Defaults/Manufacturing variable exists and appears to be valid. \r | |
988 | // The map is read, exit w/ success now.\r | |
989 | //\r | |
990 | gBS->FreePool (Name16);\r | |
991 | return;\r | |
992 | }\r | |
993 | \r | |
994 | //\r | |
995 | // First, prime the map with what already is in the NV.\r | |
996 | // This is needed to cover a situation where the IFR does not contain all the \r | |
997 | // defaults; either deliberately not having appropriate IFR, or in case of IFR_STRING, there is no default.\r | |
998 | // Ignore status. Either it gets read or not. \r | |
999 | // \r | |
1000 | FormCallbackProt = NULL;\r | |
1001 | CopyMem (&CallbackHandle, &((EFI_IFR_FORM_SET*) FormSet)->CallbackHandle, sizeof (CallbackHandle));\r | |
1002 | if (CallbackHandle != NULL) {\r | |
1003 | Status = gBS->HandleProtocol (\r | |
1004 | (EFI_HANDLE) (UINTN) CallbackHandle,\r | |
1005 | &gEfiFormCallbackProtocolGuid,\r | |
1006 | (VOID *) &FormCallbackProt\r | |
1007 | );\r | |
1008 | }\r | |
1009 | if ((NULL != FormCallbackProt) && (NULL != FormCallbackProt->NvRead)) {\r | |
1010 | //\r | |
1011 | // Attempt to read using NvRead() callback. Probe first for existence and correct variable size.\r | |
1012 | //\r | |
1013 | SizeTmp = 0;\r | |
1014 | Status = FormCallbackProt->NvRead (\r | |
1015 | FormCallbackProt,\r | |
1016 | Name16,\r | |
1017 | &Guid,\r | |
1018 | 0,\r | |
1019 | &SizeTmp,\r | |
1020 | NULL\r | |
1021 | );\r | |
1022 | if ((EFI_BUFFER_TOO_SMALL == Status) && (SizeTmp == Size)) {\r | |
1023 | Status = FormCallbackProt->NvRead (\r | |
1024 | FormCallbackProt,\r | |
1025 | Name16,\r | |
1026 | &Guid,\r | |
1027 | 0,\r | |
1028 | &SizeTmp,\r | |
1029 | Map\r | |
1030 | );\r | |
1031 | ASSERT_EFI_ERROR (Status);\r | |
1032 | ASSERT (SizeTmp == Size);\r | |
1033 | }\r | |
1034 | } else {\r | |
1035 | //\r | |
1036 | // No callback available for this formset, read straight from NV. Deliberately ignore the Status. \r | |
1037 | // The buffer will only be written if variable exists nd has correct size.\r | |
1038 | //\r | |
1039 | Status = EfiLibHiiVariableRetrieveFromNv (\r | |
1040 | Name16,\r | |
1041 | &Guid,\r | |
1042 | Size,\r | |
1043 | (VOID **) &Map\r | |
1044 | );\r | |
1045 | }\r | |
1046 | \r | |
1047 | //\r | |
1048 | // Iterate all IFR statements and for applicable, retrieve the default into the Map.\r | |
1049 | //\r | |
1050 | for (IfrItem = FormSet, VarId = 0; \r | |
1051 | IfrItem->OpCode != EFI_IFR_END_FORM_SET_OP; \r | |
1052 | IfrItem = (EFI_IFR_OP_HEADER *) ((UINT8*) IfrItem + IfrItem->Length)\r | |
1053 | ) {\r | |
1054 | \r | |
1055 | //\r | |
1056 | // Observe VarStore switch.\r | |
1057 | //\r | |
1058 | if (EFI_IFR_VARSTORE_SELECT_OP == IfrItem->OpCode) {\r | |
1059 | VarSelect = (EFI_IFR_VARSTORE_SELECT *) IfrItem;\r | |
1060 | VarId = VarSelect->VarId;\r | |
1061 | continue;\r | |
1062 | }\r | |
1063 | \r | |
1064 | \r | |
1065 | //\r | |
1066 | // Skip opcodes that reference other VarStore than that specific to current map.\r | |
1067 | // \r | |
1068 | if (VarId != VarStore->VarId) {\r | |
1069 | continue;\r | |
1070 | }\r | |
1071 | \r | |
1072 | //\r | |
1073 | // Extract the default value from this opcode if applicable, and apply it to the map.\r | |
1074 | //\r | |
1075 | IfrNvData = (EFI_IFR_NV_DATA *) IfrItem;\r | |
1076 | switch (IfrItem->OpCode) {\r | |
1077 | \r | |
1078 | case EFI_IFR_ONE_OF_OP:\r | |
1079 | ASSERT (IfrNvData->QuestionId + IfrNvData->StorageWidth <= VarStore->Size);\r | |
1080 | //\r | |
1081 | // Get to the first EFI_IFR_ONE_OF_OPTION_OP\r | |
1082 | //\r | |
1083 | IfrItem = (EFI_IFR_OP_HEADER *) ((UINT8*) IfrItem + IfrItem->Length); \r | |
1084 | ASSERT (EFI_IFR_ONE_OF_OPTION_OP == IfrItem->OpCode);\r | |
1085 | \r | |
1086 | OneOfOpt = (EFI_IFR_ONE_OF_OPTION *)IfrItem;\r | |
1087 | //\r | |
1088 | // In the worst case, the first will be the default.\r | |
1089 | //\r | |
1090 | CopyMem (Map + IfrNvData->QuestionId, &OneOfOpt->Value, IfrNvData->StorageWidth);\r | |
1091 | \r | |
1092 | while (EFI_IFR_ONE_OF_OPTION_OP == IfrItem->OpCode) {\r | |
1093 | \r | |
1094 | OneOfOpt = (EFI_IFR_ONE_OF_OPTION *)IfrItem;\r | |
1095 | if (DefaultMask == EFI_IFR_FLAG_MANUFACTURING) {\r | |
1096 | if (0 != (OneOfOpt->Flags & EFI_IFR_FLAG_MANUFACTURING)) {\r | |
1097 | //\r | |
1098 | // In the worst case, the first will be the default.\r | |
1099 | //\r | |
1100 | CopyMem (Map + IfrNvData->QuestionId, &OneOfOpt->Value, IfrNvData->StorageWidth);\r | |
1101 | break;\r | |
1102 | }\r | |
1103 | } else {\r | |
1104 | if (OneOfOpt->Flags & EFI_IFR_FLAG_DEFAULT) {\r | |
1105 | //\r | |
1106 | // In the worst case, the first will be the default.\r | |
1107 | //\r | |
1108 | CopyMem (Map + IfrNvData->QuestionId, &OneOfOpt->Value, IfrNvData->StorageWidth);\r | |
1109 | break;\r | |
1110 | }\r | |
1111 | }\r | |
1112 | \r | |
1113 | IfrItem = (EFI_IFR_OP_HEADER *)((UINT8*)IfrItem + IfrItem->Length);\r | |
1114 | }\r | |
1115 | continue;\r | |
1116 | break;\r | |
1117 | \r | |
1118 | case EFI_IFR_CHECKBOX_OP:\r | |
1119 | ASSERT (IfrNvData->QuestionId + IfrNvData->StorageWidth <= VarStore->Size);\r | |
1120 | CheckBox = (EFI_IFR_CHECK_BOX *)IfrItem; \r | |
1121 | if (DefaultMask == EFI_IFR_FLAG_MANUFACTURING) {\r | |
1122 | if (0 != (CheckBox->Flags & EFI_IFR_FLAG_MANUFACTURING)) {\r | |
1123 | *(UINT8 *) (Map + IfrNvData->QuestionId) = TRUE;\r | |
1124 | }\r | |
1125 | } else {\r | |
1126 | if (CheckBox->Flags & EFI_IFR_FLAG_DEFAULT) {\r | |
1127 | *(UINT8 *) (Map + IfrNvData->QuestionId) = TRUE;\r | |
1128 | }\r | |
1129 | }\r | |
1130 | break;\r | |
1131 | \r | |
1132 | case EFI_IFR_NUMERIC_OP:\r | |
1133 | ASSERT (IfrNvData->QuestionId + IfrNvData->StorageWidth <= VarStore->Size);\r | |
1134 | Numeric = (EFI_IFR_NUMERIC *) IfrItem;\r | |
1135 | CopyMem (Map + IfrNvData->QuestionId, &Numeric->Default, IfrNvData->StorageWidth);\r | |
1136 | break;\r | |
1137 | \r | |
1138 | case EFI_IFR_ORDERED_LIST_OP:\r | |
1139 | case EFI_IFR_PASSWORD_OP:\r | |
1140 | case EFI_IFR_STRING_OP:\r | |
1141 | //\r | |
1142 | // No support for default value for these opcodes.\r | |
1143 | //\r | |
1144 | break;\r | |
1145 | }\r | |
1146 | }\r | |
1147 | \r | |
1148 | gBS->FreePool (Name16);\r | |
1149 | \r | |
1150 | }\r | |
1151 | \r | |
1152 | \r | |
1153 | EFI_STATUS\r | |
1154 | EFIAPI\r | |
1155 | HiiGetDefaultImage (\r | |
1156 | IN EFI_HII_PROTOCOL *This,\r | |
1157 | IN EFI_HII_HANDLE Handle,\r | |
1158 | IN UINTN DefaultMask,\r | |
1159 | OUT EFI_HII_VARIABLE_PACK_LIST **VariablePackList\r | |
1160 | )\r | |
1161 | /*++\r | |
1162 | \r | |
1163 | Routine Description:\r | |
1164 | \r | |
1165 | This function allows a program to extract the NV Image \r | |
1166 | that represents the default storage image\r | |
1167 | \r | |
1168 | Arguments:\r | |
1169 | This - A pointer to the EFI_HII_PROTOCOL instance.\r | |
1170 | Handle - The HII handle from which will have default data retrieved.\r | |
1171 | UINTN - Mask used to retrieve the default image.\r | |
1172 | VariablePackList - Callee allocated, tightly-packed, link list data \r | |
1173 | structure that contain all default varaible packs\r | |
1174 | from the Hii Database.\r | |
1175 | \r | |
1176 | Returns: \r | |
1177 | EFI_NOT_FOUND - If Hii database does not contain any default images.\r | |
1178 | EFI_INVALID_PARAMETER - Invalid input parameter.\r | |
1179 | EFI_SUCCESS - Operation successful.\r | |
1180 | \r | |
1181 | --*/\r | |
1182 | {\r | |
1183 | EFI_HII_HANDLE_DATABASE *HandleDatabase;\r | |
1184 | EFI_HII_PACKAGE_INSTANCE *PackageInstance;\r | |
1185 | EFI_IFR_OP_HEADER *FormSet;\r | |
1186 | EFI_IFR_OP_HEADER *IfrItem;\r | |
1187 | EFI_IFR_VARSTORE *VarStore;\r | |
1188 | EFI_IFR_VARSTORE *VarStoreDefault;\r | |
1189 | UINTN SetupMapNameSize;\r | |
1190 | UINTN SizeOfMaps;\r | |
1191 | EFI_HII_VARIABLE_PACK_LIST *PackList;\r | |
1192 | EFI_HII_VARIABLE_PACK_LIST *PackListNext; \r | |
1193 | EFI_HII_VARIABLE_PACK_LIST *PackListLast;\r | |
1194 | UINT8 *Map;\r | |
1195 | \r | |
1196 | \r | |
1197 | //\r | |
1198 | // Find the IFR pack from the handle. Then get the formset from the pack.\r | |
1199 | //\r | |
1200 | PackageInstance = NULL;\r | |
1201 | HandleDatabase = (EFI_HII_DATA_FROM_THIS (This))->DatabaseHead;\r | |
1202 | for ( ; HandleDatabase != NULL; HandleDatabase = HandleDatabase->NextHandleDatabase) {\r | |
1203 | if (Handle == HandleDatabase->Handle) {\r | |
1204 | PackageInstance = HandleDatabase->Buffer;\r | |
1205 | break;\r | |
1206 | }\r | |
1207 | }\r | |
1208 | if (PackageInstance == NULL) {\r | |
1209 | return EFI_INVALID_PARAMETER;\r | |
1210 | }\r | |
1211 | FormSet = (EFI_IFR_OP_HEADER *) ((UINT8 *) &PackageInstance->IfrData + sizeof (EFI_HII_IFR_PACK));\r | |
1212 | \r | |
1213 | //\r | |
1214 | // Get the sizes of all the VARSTOREs in this VFR.\r | |
1215 | // Then allocate enough space for all of them plus all maps\r | |
1216 | //\r | |
1217 | SizeOfMaps = 0;\r | |
1218 | IfrItem = FormSet;\r | |
1219 | while (EFI_IFR_END_FORM_SET_OP != IfrItem->OpCode) {\r | |
1220 | \r | |
1221 | if (EFI_IFR_VARSTORE_OP == IfrItem->OpCode) {\r | |
1222 | VarStore = (EFI_IFR_VARSTORE *) IfrItem;\r | |
1223 | //\r | |
1224 | // Size of the map\r | |
1225 | //\r | |
1226 | SizeOfMaps += VarStore->Size; \r | |
1227 | //\r | |
1228 | // add the size of the string, in Unicode\r | |
1229 | //\r | |
1230 | SizeOfMaps += (VarStore->Header.Length - sizeof (*VarStore)) * 2; \r | |
1231 | //\r | |
1232 | // Space for node\r | |
1233 | //\r | |
1234 | SizeOfMaps += sizeof (EFI_HII_VARIABLE_PACK); \r | |
1235 | //\r | |
1236 | // Space for linked list node \r | |
1237 | //\r | |
1238 | SizeOfMaps += sizeof (EFI_HII_VARIABLE_PACK_LIST); \r | |
1239 | }\r | |
1240 | \r | |
1241 | IfrItem = (EFI_IFR_OP_HEADER *) ((UINT8 *) IfrItem + IfrItem->Length);\r | |
1242 | }\r | |
1243 | \r | |
1244 | //\r | |
1245 | // If the FormSet OpCode has a non-zero NvDataSize. There is a default \r | |
1246 | // NvMap with ID=0, GUID that of the formset itself and "Setup" as name.\r | |
1247 | //\r | |
1248 | SetupMapNameSize = StrLen (SETUP_MAP_NAME) + 1;\r | |
1249 | VarStoreDefault = AllocateZeroPool (sizeof (*VarStoreDefault) + SetupMapNameSize);\r | |
1250 | \r | |
1251 | if (0 != ((EFI_IFR_FORM_SET*)FormSet)->NvDataSize) {\r | |
1252 | \r | |
1253 | VarStoreDefault->Header.OpCode = EFI_IFR_VARSTORE_OP;\r | |
1254 | VarStoreDefault->Header.Length = (UINT8) (sizeof (*VarStoreDefault) + SetupMapNameSize);\r | |
1255 | Unicode2Ascii ((CHAR8 *) (VarStoreDefault + 1), SETUP_MAP_NAME);\r | |
1256 | CopyMem (&VarStoreDefault->Guid, &((EFI_IFR_FORM_SET*) FormSet)->Guid, sizeof (EFI_GUID));\r | |
1257 | VarStoreDefault->VarId = 0;\r | |
1258 | VarStoreDefault->Size = ((EFI_IFR_FORM_SET*) FormSet)->NvDataSize;\r | |
1259 | \r | |
1260 | //\r | |
1261 | // Size of the map\r | |
1262 | //\r | |
1263 | SizeOfMaps += VarStoreDefault->Size; \r | |
1264 | //\r | |
1265 | // add the size of the string\r | |
1266 | //\r | |
1267 | SizeOfMaps += sizeof (SETUP_MAP_NAME); \r | |
1268 | //\r | |
1269 | // Space for node\r | |
1270 | //\r | |
1271 | SizeOfMaps += sizeof (EFI_HII_VARIABLE_PACK); \r | |
1272 | //\r | |
1273 | // Space for linked list node \r | |
1274 | //\r | |
1275 | SizeOfMaps += sizeof (EFI_HII_VARIABLE_PACK_LIST); \r | |
1276 | }\r | |
1277 | \r | |
1278 | if (0 == SizeOfMaps) {\r | |
1279 | //\r | |
1280 | // The IFR does not have any explicit or default map(s).\r | |
1281 | //\r | |
1282 | return EFI_NOT_FOUND; \r | |
1283 | }\r | |
1284 | \r | |
1285 | //\r | |
1286 | // Allocate the return buffer\r | |
1287 | //\r | |
1288 | PackList = AllocateZeroPool (SizeOfMaps);\r | |
1289 | ASSERT (NULL != PackList); \r | |
1290 | \r | |
1291 | PackListNext = PackList;\r | |
1292 | PackListLast = PackList;\r | |
1293 | \r | |
1294 | //\r | |
1295 | // Handle the default map first, if any.\r | |
1296 | //\r | |
1297 | if (0 != VarStoreDefault->Size) {\r | |
1298 | \r | |
1299 | Map = HiiGetDefaultImageInitPack (PackListNext, VarStoreDefault);\r | |
1300 | \r | |
1301 | HiiGetDefaultImagePopulateMap (Map, FormSet, VarStoreDefault, DefaultMask);\r | |
1302 | \r | |
1303 | PackListNext->NextVariablePack = (EFI_HII_VARIABLE_PACK_LIST *) ((UINT8 *) PackListNext->VariablePack + PackListNext->VariablePack->Header.Length);\r | |
1304 | PackListLast = PackListNext;\r | |
1305 | PackListNext = PackListNext->NextVariablePack;\r | |
1306 | }\r | |
1307 | \r | |
1308 | \r | |
1309 | //\r | |
1310 | // Handle the explicit varstore(s)\r | |
1311 | //\r | |
1312 | IfrItem = FormSet;\r | |
1313 | while (EFI_IFR_END_FORM_SET_OP != IfrItem->OpCode) {\r | |
1314 | \r | |
1315 | if (EFI_IFR_VARSTORE_OP == IfrItem->OpCode) {\r | |
1316 | \r | |
1317 | Map = HiiGetDefaultImageInitPack (PackListNext, (EFI_IFR_VARSTORE *) IfrItem);\r | |
1318 | \r | |
1319 | HiiGetDefaultImagePopulateMap (Map, FormSet, (EFI_IFR_VARSTORE *) IfrItem, DefaultMask);\r | |
1320 | \r | |
1321 | PackListNext->NextVariablePack = (EFI_HII_VARIABLE_PACK_LIST *) ((UINT8 *) PackListNext->VariablePack + PackListNext->VariablePack->Header.Length);\r | |
1322 | PackListLast = PackListNext;\r | |
1323 | PackListNext = PackListNext->NextVariablePack;\r | |
1324 | }\r | |
1325 | \r | |
1326 | IfrItem = (EFI_IFR_OP_HEADER *) ((UINT8 *) IfrItem + IfrItem->Length);\r | |
1327 | }\r | |
1328 | \r | |
1329 | PackListLast->NextVariablePack = NULL;\r | |
1330 | *VariablePackList = PackList;\r | |
1331 | \r | |
1332 | return EFI_SUCCESS;\r | |
1333 | }\r | |
1334 | \r | |
1335 | \r | |
1336 | EFI_STATUS\r | |
1337 | EFIAPI\r | |
1338 | HiiUpdateForm (\r | |
1339 | IN EFI_HII_PROTOCOL *This,\r | |
1340 | IN EFI_HII_HANDLE Handle,\r | |
1341 | IN EFI_FORM_LABEL Label,\r | |
1342 | IN BOOLEAN AddData,\r | |
1343 | IN EFI_HII_UPDATE_DATA *Data\r | |
1344 | )\r | |
1345 | /*++\r | |
1346 | \r | |
1347 | Routine Description:\r | |
1348 | This function allows the caller to update a form that has \r | |
1349 | previously been registered with the EFI HII database.\r | |
1350 | \r | |
1351 | Arguments:\r | |
1352 | Handle - Hii Handle associated with the Formset to modify\r | |
1353 | Label - Update information starting immediately after this label in the IFR\r | |
1354 | AddData - If TRUE, add data. If FALSE, remove data\r | |
1355 | Data - If adding data, this is the pointer to the data to add\r | |
1356 | \r | |
1357 | Returns: \r | |
1358 | EFI_SUCCESS - Update success.\r | |
1359 | Other - Update fail.\r | |
1360 | \r | |
1361 | --*/\r | |
1362 | {\r | |
1363 | EFI_HII_PACKAGE_INSTANCE *PackageInstance;\r | |
1364 | EFI_HII_DATA *HiiData;\r | |
1365 | EFI_HII_HANDLE_DATABASE *HandleDatabase;\r | |
1366 | EFI_HII_IFR_PACK *FormPack;\r | |
1367 | EFI_IFR_OP_HEADER *Location;\r | |
1368 | EFI_IFR_OP_HEADER *DataLocation;\r | |
1369 | UINT8 *OtherBuffer;\r | |
1370 | UINT8 *TempBuffer;\r | |
1371 | UINT8 *OrigTempBuffer;\r | |
1372 | UINTN TempBufferSize;\r | |
1373 | UINTN Index;\r | |
1374 | \r | |
1375 | OtherBuffer = NULL;\r | |
1376 | \r | |
1377 | if (This == NULL) {\r | |
1378 | return EFI_INVALID_PARAMETER;\r | |
1379 | }\r | |
1380 | \r | |
1381 | HiiData = EFI_HII_DATA_FROM_THIS (This);\r | |
1382 | \r | |
1383 | HandleDatabase = HiiData->DatabaseHead;\r | |
1384 | \r | |
1385 | PackageInstance = NULL;\r | |
1386 | \r | |
1387 | //\r | |
1388 | // Check numeric value against the head of the database\r | |
1389 | //\r | |
1390 | for (; HandleDatabase != NULL; HandleDatabase = HandleDatabase->NextHandleDatabase) {\r | |
1391 | //\r | |
1392 | // Match the numeric value with the database entry - if matched, extract PackageInstance\r | |
1393 | //\r | |
1394 | if (Handle == HandleDatabase->Handle) {\r | |
1395 | PackageInstance = HandleDatabase->Buffer;\r | |
1396 | break;\r | |
1397 | }\r | |
1398 | }\r | |
1399 | //\r | |
1400 | // No handle was found - error condition\r | |
1401 | //\r | |
1402 | if (PackageInstance == NULL) {\r | |
1403 | return EFI_INVALID_PARAMETER;\r | |
1404 | }\r | |
1405 | //\r | |
1406 | // Calculate and allocate space for retrieval of IFR data\r | |
1407 | //\r | |
1408 | DataLocation = (EFI_IFR_OP_HEADER *) &Data->Data;\r | |
1409 | TempBufferSize = (CHAR8 *) (&PackageInstance->IfrData) - (CHAR8 *) (PackageInstance);\r | |
1410 | \r | |
1411 | for (Index = 0; Index < Data->DataCount; Index++) {\r | |
1412 | TempBufferSize += DataLocation->Length;\r | |
1413 | DataLocation = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (DataLocation) + DataLocation->Length);\r | |
1414 | }\r | |
1415 | \r | |
1416 | TempBufferSize += PackageInstance->IfrSize + PackageInstance->StringSize;\r | |
1417 | \r | |
1418 | TempBuffer = AllocateZeroPool (TempBufferSize);\r | |
41a907e4 | 1419 | ASSERT (TempBuffer != NULL);\r |
1420 | \r | |
878ddf1f | 1421 | OrigTempBuffer = TempBuffer;\r |
1422 | \r | |
1423 | //\r | |
1424 | // We update only packages with IFR information in it\r | |
1425 | //\r | |
1426 | if (PackageInstance->IfrSize == 0) {\r | |
1427 | return EFI_INVALID_PARAMETER;\r | |
1428 | }\r | |
1429 | \r | |
1430 | CopyMem (\r | |
1431 | TempBuffer,\r | |
1432 | PackageInstance,\r | |
1433 | ((CHAR8 *) (&PackageInstance->IfrData) + sizeof (EFI_HII_PACK_HEADER) - (CHAR8 *) (PackageInstance))\r | |
1434 | );\r | |
1435 | \r | |
1436 | TempBuffer = TempBuffer + ((CHAR8 *) (&PackageInstance->IfrData) + sizeof (EFI_HII_PACK_HEADER) - (CHAR8 *) (PackageInstance));\r | |
1437 | \r | |
1438 | //\r | |
1439 | // Based on if there is IFR data in this package instance, determine\r | |
1440 | // what the location is of the beginning of the string data.\r | |
1441 | //\r | |
1442 | FormPack = (EFI_HII_IFR_PACK *) ((CHAR8 *) (&PackageInstance->IfrData) + sizeof (EFI_HII_PACK_HEADER));\r | |
1443 | Location = (EFI_IFR_OP_HEADER *) FormPack;\r | |
1444 | \r | |
1445 | //\r | |
1446 | // Look for the FormId requested\r | |
1447 | //\r | |
1448 | for (; Location->OpCode != EFI_IFR_END_FORM_SET_OP;) {\r | |
1449 | switch (Location->OpCode) {\r | |
1450 | case EFI_IFR_FORM_SET_OP:\r | |
1451 | //\r | |
1452 | // If the FormSet has an update pending, pay attention.\r | |
1453 | //\r | |
1454 | if (Data->FormSetUpdate) {\r | |
1455 | ((EFI_IFR_FORM_SET *) Location)->CallbackHandle = Data->FormCallbackHandle;\r | |
1456 | }\r | |
1457 | \r | |
1458 | CopyMem (TempBuffer, Location, Location->Length);\r | |
1459 | TempBuffer = TempBuffer + Location->Length;\r | |
1460 | break;\r | |
1461 | \r | |
1462 | case EFI_IFR_FORM_OP:\r | |
1463 | //\r | |
1464 | // If the Form has an update pending, pay attention.\r | |
1465 | //\r | |
1466 | if (Data->FormUpdate) {\r | |
1467 | ((EFI_IFR_FORM *) Location)->FormTitle = Data->FormTitle;\r | |
1468 | }\r | |
1469 | \r | |
1470 | CopyMem (TempBuffer, Location, Location->Length);\r | |
1471 | TempBuffer = TempBuffer + Location->Length;\r | |
1472 | break;\r | |
1473 | \r | |
1474 | case EFI_IFR_LABEL_OP:\r | |
1475 | //\r | |
1476 | // If the label does not match the requested update point, ignore it\r | |
1477 | //\r | |
1478 | if (((EFI_IFR_LABEL *) Location)->LabelId != Label) {\r | |
1479 | //\r | |
1480 | // Copy the label\r | |
1481 | //\r | |
1482 | CopyMem (TempBuffer, Location, Location->Length);\r | |
1483 | TempBuffer = TempBuffer + Location->Length;\r | |
1484 | \r | |
1485 | //\r | |
1486 | // Go to the next Op-Code\r | |
1487 | //\r | |
1488 | Location = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (Location) + Location->Length);\r | |
1489 | continue;\r | |
1490 | }\r | |
1491 | \r | |
1492 | if (AddData) {\r | |
1493 | //\r | |
1494 | // Copy the label\r | |
1495 | //\r | |
1496 | CopyMem (TempBuffer, Location, Location->Length);\r | |
1497 | TempBuffer = TempBuffer + Location->Length;\r | |
1498 | \r | |
1499 | //\r | |
1500 | // Add the DataCount amount of opcodes to TempBuffer\r | |
1501 | //\r | |
1502 | DataLocation = (EFI_IFR_OP_HEADER *) &Data->Data;\r | |
1503 | for (Index = 0; Index < Data->DataCount; Index++) {\r | |
1504 | CopyMem (TempBuffer, DataLocation, DataLocation->Length);\r | |
1505 | ((EFI_HII_PACKAGE_INSTANCE *) OrigTempBuffer)->IfrSize += DataLocation->Length;\r | |
1506 | OtherBuffer = ((UINT8 *) &((EFI_HII_PACKAGE_INSTANCE *) OrigTempBuffer)->StringSize + sizeof (UINTN));\r | |
1507 | CopyMem (OtherBuffer, &((EFI_HII_PACKAGE_INSTANCE *) OrigTempBuffer)->IfrSize, 2);\r | |
1508 | TempBuffer = TempBuffer + DataLocation->Length;\r | |
1509 | DataLocation = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (DataLocation) + DataLocation->Length);\r | |
1510 | }\r | |
1511 | //\r | |
1512 | // Go to the next Op-Code\r | |
1513 | //\r | |
1514 | Location = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (Location) + Location->Length);\r | |
1515 | continue;\r | |
1516 | } else {\r | |
1517 | //\r | |
1518 | // Copy the label\r | |
1519 | //\r | |
1520 | CopyMem (TempBuffer, Location, Location->Length);\r | |
1521 | TempBuffer = TempBuffer + Location->Length;\r | |
1522 | Location = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (Location) + Location->Length);\r | |
1523 | \r | |
1524 | //\r | |
1525 | // Remove the DataCount amount of opcodes unless we run into an end of form or a label\r | |
1526 | //\r | |
1527 | for (Index = 0; Index < Data->DataCount; Index++) {\r | |
1528 | //\r | |
1529 | // If we are about to skip an end form - bail out, since that is illegal\r | |
1530 | //\r | |
1531 | if ((Location->OpCode == EFI_IFR_END_FORM_OP) || (Location->OpCode == EFI_IFR_LABEL_OP)) {\r | |
1532 | break;\r | |
1533 | }\r | |
1534 | //\r | |
1535 | // By skipping Location entries, we are in effect not copying what was previously there\r | |
1536 | //\r | |
1537 | ((EFI_HII_PACKAGE_INSTANCE *) OrigTempBuffer)->IfrSize -= Location->Length;\r | |
1538 | OtherBuffer = ((UINT8 *) &((EFI_HII_PACKAGE_INSTANCE *) OrigTempBuffer)->StringSize + sizeof (UINTN));\r | |
1539 | CopyMem (OtherBuffer, &((EFI_HII_PACKAGE_INSTANCE *) OrigTempBuffer)->IfrSize, 2);\r | |
1540 | Location = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (Location) + Location->Length);\r | |
1541 | }\r | |
1542 | }\r | |
1543 | \r | |
1544 | default:\r | |
1545 | CopyMem (TempBuffer, Location, Location->Length);\r | |
1546 | TempBuffer = TempBuffer + Location->Length;\r | |
1547 | break;\r | |
1548 | }\r | |
1549 | //\r | |
1550 | // Go to the next Op-Code\r | |
1551 | //\r | |
1552 | Location = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (Location) + Location->Length);\r | |
1553 | }\r | |
1554 | //\r | |
1555 | // Copy the last op-code left behind from the for loop\r | |
1556 | //\r | |
1557 | CopyMem (TempBuffer, Location, Location->Length);\r | |
1558 | \r | |
1559 | //\r | |
1560 | // Advance to beginning of strings and copy them\r | |
1561 | //\r | |
1562 | TempBuffer = TempBuffer + Location->Length;\r | |
1563 | Location = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (Location) + Location->Length);\r | |
1564 | CopyMem (TempBuffer, Location, PackageInstance->StringSize);\r | |
1565 | \r | |
1566 | //\r | |
1567 | // Free the old buffer, and assign into our database the latest buffer\r | |
1568 | //\r | |
1569 | gBS->FreePool (HandleDatabase->Buffer);\r | |
1570 | HandleDatabase->Buffer = OrigTempBuffer;\r | |
1571 | \r | |
1572 | return EFI_SUCCESS;\r | |
1573 | }\r |