]> git.proxmox.com Git - mirror_edk2.git/blob - IntelFrameworkPkg/Library/FrameworkIfrSupportLib/IfrCommon.c
Fixed two K8 report issue
[mirror_edk2.git] / IntelFrameworkPkg / Library / FrameworkIfrSupportLib / IfrCommon.c
1 /** @file
2 Common Library Routines to assist in IFR creation on-the-fly
3
4 Copyright (c) 2006, Intel Corporation
5 All rights reserved. This program and the accompanying materials
6 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution. The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
9
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
12
13 **/
14
15 #include "IfrSupportLibInternal.h"
16
17 /**
18 Determine what is the current language setting
19 The setting is stored in language variable in flash. This routine
20 will get setting by accesssing that variable. If failed to access
21 language variable, then use default setting that 'eng' as current
22 language setting.
23
24 @param Lang Pointer of system language
25
26 @return whether sucess to get setting from variable
27 **/
28 EFI_STATUS
29 EFIAPI
30 GetCurrentLanguage (
31 OUT CHAR16 *Lang
32 )
33 {
34 EFI_STATUS Status;
35 UINTN Size;
36 UINTN Index;
37 CHAR8 Language[4];
38
39 //
40 // Getting the system language and placing it into our Global Data
41 //
42 Size = sizeof (Language);
43 Status = gRT->GetVariable (
44 (CHAR16 *) L"Lang",
45 &gEfiGlobalVariableGuid,
46 NULL,
47 &Size,
48 Language
49 );
50 if (EFI_ERROR (Status)) {
51 AsciiStrCpy (Language, "eng");
52 }
53
54 for (Index = 0; Index < 3; Index++) {
55 //
56 // Bitwise AND ascii value with 0xDF yields an uppercase value.
57 // Sign extend into a unicode value
58 //
59 Lang[Index] = (CHAR16) (Language[Index] & 0xDF);
60 }
61
62 //
63 // Null-terminate the value
64 //
65 Lang[3] = (CHAR16) 0;
66
67 return Status;
68 }
69
70 /**
71 Add a string to the incoming buffer and return the token and offset data.
72
73 @param StringBuffer The incoming buffer
74 @param Language Currrent language
75 @param String The string to be added
76 @param StringToken The index where the string placed
77
78 @retval EFI_OUT_OF_RESOURCES No enough buffer to allocate
79 @retval EFI_SUCCESS String successfully added to the incoming buffer
80 **/
81 EFI_STATUS
82 EFIAPI
83 AddString (
84 IN VOID *StringBuffer,
85 IN CHAR16 *Language,
86 IN CHAR16 *String,
87 IN OUT STRING_REF *StringToken
88 )
89 {
90 EFI_HII_STRING_PACK *StringPack;
91 EFI_HII_STRING_PACK *StringPackBuffer;
92 VOID *NewBuffer;
93 RELOFST *PackSource;
94 RELOFST *PackDestination;
95 UINT8 *Source;
96 UINT8 *Destination;
97 UINTN Index;
98 BOOLEAN Finished;
99 UINTN SizeofLanguage;
100 UINTN SizeofString;
101
102 StringPack = (EFI_HII_STRING_PACK *) StringBuffer;
103 Finished = FALSE;
104
105 //
106 // Pre-allocate a buffer sufficient for us to work on.
107 // We will use it as a destination scratch pad to build data on
108 // and when complete shift the data back to the original buffer
109 //
110 NewBuffer = AllocateZeroPool (DEFAULT_STRING_BUFFER_SIZE);
111 if (NewBuffer == NULL) {
112 return EFI_OUT_OF_RESOURCES;
113 }
114
115 StringPackBuffer = (EFI_HII_STRING_PACK *) NewBuffer;
116
117 //
118 // StringPack is terminated with a length 0 entry
119 //
120 for (; StringPack->Header.Length != 0;) {
121 //
122 // If this stringpack's language is same as CurrentLanguage, use it
123 //
124 if (CompareMem ((VOID *) ((CHAR8 *) (StringPack) + StringPack->LanguageNameString), Language, 3) == 0) {
125 //
126 // We have some data in this string pack, copy the string package up to the string data
127 //
128 CopyMem (&StringPackBuffer->Header, &StringPack->Header, sizeof (StringPack));
129
130 //
131 // These are references in the structure to tokens, need to increase them by the space occupied by an additional StringPointer
132 //
133 StringPackBuffer->LanguageNameString = (UINT16) (StringPackBuffer->LanguageNameString + (UINT16) sizeof (RELOFST));
134 StringPackBuffer->PrintableLanguageName = (UINT16) (StringPackBuffer->PrintableLanguageName + (UINT16) sizeof (RELOFST));
135
136 PackSource = (RELOFST *) (StringPack + 1);
137 PackDestination = (RELOFST *) (StringPackBuffer + 1);
138 for (Index = 0; PackSource[Index] != 0x0000; Index++) {
139 //
140 // Copy the stringpointers from old to new buffer
141 // remember that we are adding a string, so the string offsets will all go up by sizeof (RELOFST)
142 //
143 PackDestination[Index] = (UINT16) (PackDestination[Index] + sizeof (RELOFST));
144 }
145
146 //
147 // Add a new stringpointer in the new buffer since we are adding a string. Null terminate it
148 //
149 PackDestination[Index] = (UINT16)(PackDestination[Index - 1] +
150 StrSize((CHAR16 *)((CHAR8 *)(StringPack) + PackSource[Index - 1])));
151 PackDestination[Index + 1] = (UINT16) 0;
152
153 //
154 // Index is the token value for the new string
155 //
156 *StringToken = (UINT16) Index;
157
158 //
159 // Source now points to the beginning of the old buffer strings
160 // Destination now points to the beginning of the new buffer strings
161 //
162 Source = (UINT8 *) &PackSource[Index + 1];
163 Destination = (UINT8 *) &PackDestination[Index + 2];
164
165 //
166 // This should copy all the strings from the old buffer to the new buffer
167 //
168 for (; Index != 0; Index--) {
169 //
170 // Copy Source string to destination buffer
171 //
172 StrCpy ((CHAR16 *) Destination, (CHAR16 *) Source);
173
174 //
175 // Adjust the source/destination to the next string location
176 //
177 Destination = Destination + StrSize ((CHAR16 *) Source);
178 Source = Source + StrSize ((CHAR16 *) Source);
179 }
180
181 //
182 // This copies the new string to the destination buffer
183 //
184 StrCpy ((CHAR16 *) Destination, (CHAR16 *) String);
185
186 //
187 // Adjust the size of the changed string pack by adding the size of the new string
188 // along with the size of the additional offset entry for the new string
189 //
190 StringPackBuffer->Header.Length = (UINT32) ((UINTN) StringPackBuffer->Header.Length + StrSize (String) + sizeof (RELOFST));
191
192 //
193 // Advance the buffers to point to the next spots.
194 //
195 StringPackBuffer = (EFI_HII_STRING_PACK *) ((CHAR8 *) (StringPackBuffer) + StringPackBuffer->Header.Length);
196 StringPack = (EFI_HII_STRING_PACK *) ((CHAR8 *) (StringPack) + StringPack->Header.Length);
197 Finished = TRUE;
198 continue;
199 }
200 //
201 // This isn't the language of the stringpack we were asked to add a string to
202 // so we need to copy it to the new buffer.
203 //
204 CopyMem (&StringPackBuffer->Header, &StringPack->Header, StringPack->Header.Length);
205
206 //
207 // Advance the buffers to point to the next spots.
208 //
209 StringPackBuffer = (EFI_HII_STRING_PACK *) ((CHAR8 *) (StringPackBuffer) + StringPack->Header.Length);
210 StringPack = (EFI_HII_STRING_PACK *) ((CHAR8 *) (StringPack) + StringPack->Header.Length);
211 }
212
213 //
214 // If we didn't copy the new data to a stringpack yet
215 //
216 if (!Finished) {
217 PackDestination = (RELOFST *) (StringPackBuffer + 1);
218 //
219 // Pointing to a new string pack location
220 //
221 SizeofLanguage = StrSize (Language);
222 SizeofString = StrSize (String);
223 StringPackBuffer->Header.Length = (UINT32)
224 (
225 sizeof (EFI_HII_STRING_PACK) -
226 sizeof (EFI_STRING) +
227 sizeof (RELOFST) +
228 sizeof (RELOFST) +
229 SizeofLanguage +
230 SizeofString
231 );
232 StringPackBuffer->Header.Type = EFI_HII_STRING;
233 StringPackBuffer->LanguageNameString = (UINT16) ((UINTN) &PackDestination[3] - (UINTN) StringPackBuffer);
234 StringPackBuffer->PrintableLanguageName = (UINT16) ((UINTN) &PackDestination[3] - (UINTN) StringPackBuffer);
235 StringPackBuffer->Attributes = 0;
236 PackDestination[0] = (UINT16) ((UINTN) &PackDestination[3] - (UINTN) StringPackBuffer);
237 PackDestination[1] = (UINT16) (PackDestination[0] + StrSize (Language));
238 PackDestination[2] = (UINT16) 0;
239
240 //
241 // The first string location will be set to destination. The minimum number of strings
242 // associated with a stringpack will always be token 0 stored as the languagename (e.g. ENG, SPA, etc)
243 // and token 1 as the new string being added and and null entry for the stringpointers
244 //
245 Destination = (UINT8 *) &PackDestination[3];
246
247 //
248 // Copy the language name string to the new buffer
249 //
250 StrCpy ((CHAR16 *) Destination, Language);
251
252 //
253 // Advance the destination to the new empty spot
254 //
255 Destination = Destination + StrSize (Language);
256
257 //
258 // Copy the string to the new buffer
259 //
260 StrCpy ((CHAR16 *) Destination, String);
261
262 //
263 // Since we are starting with a new string pack - we know the new string is token 1
264 //
265 *StringToken = (UINT16) 1;
266 }
267
268 //
269 // Zero out the original buffer and copy the updated data in the new buffer to the old buffer
270 //
271 ZeroMem (StringBuffer, DEFAULT_STRING_BUFFER_SIZE);
272 CopyMem (StringBuffer, NewBuffer, DEFAULT_STRING_BUFFER_SIZE);
273
274 //
275 // Free the newly created buffer since we don't need it anymore
276 //
277 FreePool (NewBuffer);
278 return EFI_SUCCESS;
279 }
280
281 /**
282 Add op-code data to the FormBuffer.
283
284 @param FormBuffer Form buffer to be inserted to
285 @param OpCodeData Op-code data to be inserted
286
287 @retval EFI_OUT_OF_RESOURCES No enough buffer to allocate
288 @retval EFI_SUCCESS Op-code data successfully inserted
289 **/
290 EFI_STATUS
291 EFIAPI
292 AddOpCode (
293 IN VOID *FormBuffer,
294 IN OUT VOID *OpCodeData
295 )
296 {
297 EFI_HII_PACK_HEADER *NewBuffer;
298 UINT8 *Source;
299 UINT8 *Destination;
300
301 //
302 // Pre-allocate a buffer sufficient for us to work on.
303 // We will use it as a destination scratch pad to build data on
304 // and when complete shift the data back to the original buffer
305 //
306 NewBuffer = AllocateZeroPool (DEFAULT_FORM_BUFFER_SIZE);
307 if (NewBuffer == NULL) {
308 return EFI_OUT_OF_RESOURCES;
309 }
310
311 //
312 // Copy the IFR Package header to the new buffer
313 //
314 Source = (UINT8 *) FormBuffer;
315 Destination = (UINT8 *) NewBuffer;
316 CopyMem (Destination, Source, sizeof (EFI_HII_PACK_HEADER));
317
318 //
319 // Advance Source and Destination to next op-code
320 //
321 Source = Source + sizeof (EFI_HII_PACK_HEADER);
322 Destination = Destination + sizeof (EFI_HII_PACK_HEADER);
323
324 //
325 // Copy data to the new buffer until we run into the end_form
326 //
327 for (; ((FRAMEWORK_EFI_IFR_OP_HEADER *) Source)->OpCode != FRAMEWORK_EFI_IFR_END_FORM_OP;) {
328 //
329 // If the opcode is an end_form_set we better be creating and endform
330 // Nonetheless, we will add data before the end_form_set. This also provides
331 // for interesting behavior in the code we will run, but has no bad side-effects
332 // since we will possibly do a 0 byte copy in this particular end-case.
333 //
334 if (((FRAMEWORK_EFI_IFR_OP_HEADER *) Source)->OpCode == FRAMEWORK_EFI_IFR_END_FORM_SET_OP) {
335 break;
336 }
337
338 //
339 // Copy data to new buffer
340 //
341 CopyMem (Destination, Source, ((FRAMEWORK_EFI_IFR_OP_HEADER *) Source)->Length);
342
343 //
344 // Adjust Source/Destination to next op-code location
345 //
346 Destination = Destination + (UINTN) ((FRAMEWORK_EFI_IFR_OP_HEADER *) Source)->Length;
347 Source = Source + (UINTN) ((FRAMEWORK_EFI_IFR_OP_HEADER *) Source)->Length;
348 }
349
350 //
351 // Prior to the end_form is where we insert the new op-code data
352 //
353 CopyMem (Destination, OpCodeData, ((FRAMEWORK_EFI_IFR_OP_HEADER *) OpCodeData)->Length);
354
355 Destination = Destination + (UINTN) ((FRAMEWORK_EFI_IFR_OP_HEADER *) OpCodeData)->Length;
356 NewBuffer->Length = (UINT32) (NewBuffer->Length + (UINT32) (((FRAMEWORK_EFI_IFR_OP_HEADER *) OpCodeData)->Length));
357
358 //
359 // Copy end-form data to new buffer
360 //
361 CopyMem (Destination, Source, ((FRAMEWORK_EFI_IFR_OP_HEADER *) Source)->Length);
362
363 //
364 // Copy end-formset data to new buffer
365 //
366 Destination = Destination + (UINTN) ((FRAMEWORK_EFI_IFR_OP_HEADER *) Source)->Length;
367 Source = Source + (UINTN) ((FRAMEWORK_EFI_IFR_OP_HEADER *) Source)->Length;
368 CopyMem (Destination, Source, ((FRAMEWORK_EFI_IFR_OP_HEADER *) Source)->Length);
369
370 //
371 // Zero out the original buffer and copy the updated data in the new buffer to the old buffer
372 //
373 ZeroMem (FormBuffer, DEFAULT_FORM_BUFFER_SIZE);
374 CopyMem (FormBuffer, NewBuffer, DEFAULT_FORM_BUFFER_SIZE);
375
376 //
377 // Free the newly created buffer since we don't need it anymore
378 //
379 FreePool (NewBuffer);
380 return EFI_SUCCESS;
381 }
382
383 /**
384 Get the HII protocol interface.
385
386 @param Hii HII protocol interface
387
388 @return the statue of locating HII protocol
389 **/
390 EFI_STATUS
391 EFIAPI
392 GetHiiInterface (
393 OUT EFI_HII_PROTOCOL **Hii
394 )
395 {
396 EFI_STATUS Status;
397
398 //
399 // There should only be one HII protocol
400 //
401 Status = gBS->LocateProtocol (
402 &gEfiHiiProtocolGuid,
403 NULL,
404 (VOID **) Hii
405 );
406
407 return Status;;
408 }
409
410 /**
411 Extract information pertaining to the HiiHandle.
412
413 @param HiiHandle Hii handle
414 @param ImageLength For input, length of DefaultImage;
415 For output, length of actually required
416 @param DefaultImage Image buffer prepared by caller
417 @param Guid Guid information about the form
418
419 @retval EFI_OUT_OF_RESOURCES No enough buffer to allocate
420 @retval EFI_BUFFER_TOO_SMALL DefualtImage has no enough ImageLength
421 @retval EFI_SUCCESS Successfully extract data from Hii database.
422 **/
423 EFI_STATUS
424 EFIAPI
425 ExtractDataFromHiiHandle (
426 IN FRAMEWORK_EFI_HII_HANDLE HiiHandle,
427 IN OUT UINT16 *ImageLength,
428 OUT UINT8 *DefaultImage,
429 OUT EFI_GUID *Guid
430 )
431 {
432 EFI_STATUS Status;
433 EFI_HII_PROTOCOL *Hii;
434 UINTN DataLength;
435 UINT8 *RawData;
436 UINT8 *OldData;
437 UINTN Index;
438 UINTN Temp;
439 UINTN SizeOfNvStore;
440 UINTN CachedStart;
441
442 DataLength = DEFAULT_FORM_BUFFER_SIZE;
443 SizeOfNvStore = 0;
444 CachedStart = 0;
445
446 if (DefaultImage == NULL || Guid == NULL) {
447 return EFI_INVALID_PARAMETER;
448 }
449
450 Status = GetHiiInterface (&Hii);
451 if (EFI_ERROR (Status)) {
452 return Status;
453 }
454
455 //
456 // Allocate space for retrieval of IFR data
457 //
458 RawData = AllocateZeroPool (DataLength);
459 if (RawData == NULL) {
460 return EFI_OUT_OF_RESOURCES;
461 }
462
463 //
464 // Get all the forms associated with this HiiHandle
465 //
466 Status = Hii->GetForms (Hii, HiiHandle, 0, &DataLength, RawData);
467 if (EFI_ERROR (Status)) {
468 FreePool (RawData);
469
470 //
471 // Allocate space for retrieval of IFR data
472 //
473 RawData = AllocateZeroPool (DataLength);
474 if (RawData == NULL) {
475 return EFI_OUT_OF_RESOURCES;
476 }
477
478 //
479 // Get all the forms associated with this HiiHandle
480 //
481 Status = Hii->GetForms (Hii, HiiHandle, 0, &DataLength, RawData);
482 }
483
484 OldData = RawData;
485
486 //
487 // Point RawData to the beginning of the form data
488 //
489 RawData = (UINT8 *) ((UINTN) RawData + sizeof (EFI_HII_PACK_HEADER));
490
491 for (Index = 0; RawData[Index] != FRAMEWORK_EFI_IFR_END_FORM_SET_OP;) {
492 switch (RawData[Index]) {
493
494 case FRAMEWORK_EFI_IFR_FORM_SET_OP:
495 //
496 // Copy the GUID information from this handle
497 //
498 CopyGuid (Guid, (GUID *)(VOID *)&((FRAMEWORK_EFI_IFR_FORM_SET *) &RawData[Index])->Guid);
499 break;
500
501 case FRAMEWORK_EFI_IFR_ONE_OF_OP:
502 case FRAMEWORK_EFI_IFR_CHECKBOX_OP:
503 case FRAMEWORK_EFI_IFR_NUMERIC_OP:
504 case FRAMEWORK_EFI_IFR_DATE_OP:
505 case FRAMEWORK_EFI_IFR_TIME_OP:
506 case FRAMEWORK_EFI_IFR_PASSWORD_OP:
507 case FRAMEWORK_EFI_IFR_STRING_OP:
508 //
509 // Remember, multiple op-codes may reference the same item, so let's keep a running
510 // marker of what the highest QuestionId that wasn't zero length. This will accurately
511 // maintain the Size of the NvStore
512 //
513 if (((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->Width != 0) {
514 Temp = ((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->QuestionId + ((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->Width;
515 if (SizeOfNvStore < Temp) {
516 SizeOfNvStore = ((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->QuestionId + ((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->Width;
517 }
518 }
519 }
520
521 Index = RawData[Index + 1] + Index;
522 }
523
524 //
525 // Return an error if buffer is too small
526 //
527 if (SizeOfNvStore > *ImageLength) {
528 FreePool (OldData);
529 *ImageLength = (UINT16) SizeOfNvStore;
530 return EFI_BUFFER_TOO_SMALL;
531 }
532
533 if (DefaultImage != NULL) {
534 ZeroMem (DefaultImage, SizeOfNvStore);
535 }
536
537 //
538 // Copy the default image information to the user's buffer
539 //
540 for (Index = 0; RawData[Index] != FRAMEWORK_EFI_IFR_END_FORM_SET_OP;) {
541 switch (RawData[Index]) {
542
543 case FRAMEWORK_EFI_IFR_ONE_OF_OP:
544 CachedStart = ((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->QuestionId;
545 break;
546
547 case FRAMEWORK_EFI_IFR_ONE_OF_OPTION_OP:
548 if (((FRAMEWORK_EFI_IFR_ONE_OF_OPTION *) &RawData[Index])->Flags & FRAMEWORK_EFI_IFR_FLAG_DEFAULT) {
549 CopyMem (&DefaultImage[CachedStart], &((FRAMEWORK_EFI_IFR_ONE_OF_OPTION *) &RawData[Index])->Value, 2);
550 }
551 break;
552
553 case FRAMEWORK_EFI_IFR_CHECKBOX_OP:
554 DefaultImage[((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->QuestionId] = ((FRAMEWORK_EFI_IFR_CHECKBOX *) &RawData[Index])->Flags;
555 break;
556
557 case FRAMEWORK_EFI_IFR_NUMERIC_OP:
558 CopyMem (
559 &DefaultImage[((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->QuestionId],
560 &((FRAMEWORK_EFI_IFR_NUMERIC *) &RawData[Index])->Default,
561 2
562 );
563 break;
564
565 }
566
567 Index = RawData[Index + 1] + Index;
568 }
569
570 *ImageLength = (UINT16) SizeOfNvStore;
571
572 //
573 // Free our temporary repository of form data
574 //
575 FreePool (OldData);
576
577 return EFI_SUCCESS;
578 }
579
580 /**
581 Finds HII handle for given pack GUID previously registered with the HII.
582
583 @param HiiProtocol pointer to pointer to HII protocol interface.
584 If NULL, the interface will be found but not returned.
585 If it points to NULL, the interface will be found and
586 written back to the pointer that is pointed to.
587 @param Guid The GUID of the pack that registered with the HII.
588
589 @return Handle to the HII pack previously registered by the memory driver.
590 **/
591 FRAMEWORK_EFI_HII_HANDLE
592 EFIAPI
593 FindHiiHandle (
594 IN OUT EFI_HII_PROTOCOL **HiiProtocol, OPTIONAL
595 IN EFI_GUID *Guid
596 )
597 {
598 EFI_STATUS Status;
599 FRAMEWORK_EFI_HII_HANDLE *HiiHandleBuffer;
600 FRAMEWORK_EFI_HII_HANDLE HiiHandle;
601 UINT16 HiiHandleBufferLength;
602 UINT32 NumberOfHiiHandles;
603 EFI_GUID HiiGuid;
604 EFI_HII_PROTOCOL *HiiProt;
605 UINT32 Index;
606 UINT16 Length;
607
608 HiiHandle = 0;
609 if ((HiiProtocol != NULL) && (*HiiProtocol != NULL)) {
610 //
611 // The protocol has been passed in
612 //
613 HiiProt = *HiiProtocol;
614 } else {
615 gBS->LocateProtocol (
616 &gEfiHiiProtocolGuid,
617 NULL,
618 (VOID **) &HiiProt
619 );
620 if (HiiProt == NULL) {
621 return HiiHandle;
622 }
623
624 if (HiiProtocol != NULL) {
625 //
626 // Return back the HII protocol for the caller as promissed
627 //
628 *HiiProtocol = HiiProt;
629 }
630 }
631 //
632 // Allocate buffer
633 //
634 HiiHandleBufferLength = 10;
635 HiiHandleBuffer = AllocatePool (HiiHandleBufferLength);
636 ASSERT (HiiHandleBuffer != NULL);
637
638 //
639 // Get the Handles of the packages that were registered with Hii
640 //
641 Status = HiiProt->FindHandles (
642 HiiProt,
643 &HiiHandleBufferLength,
644 HiiHandleBuffer
645 );
646
647 //
648 // Get a bigger bugffer if this one is to small, and try again
649 //
650 if (Status == EFI_BUFFER_TOO_SMALL) {
651
652 FreePool (HiiHandleBuffer);
653
654 HiiHandleBuffer = AllocatePool (HiiHandleBufferLength);
655 ASSERT (HiiHandleBuffer != NULL);
656
657 Status = HiiProt->FindHandles (
658 HiiProt,
659 &HiiHandleBufferLength,
660 HiiHandleBuffer
661 );
662 }
663
664 if (EFI_ERROR (Status)) {
665 goto lbl_exit;
666 }
667
668 NumberOfHiiHandles = HiiHandleBufferLength / sizeof (FRAMEWORK_EFI_HII_HANDLE );
669
670 //
671 // Iterate Hii handles and look for the one that matches our Guid
672 //
673 for (Index = 0; Index < NumberOfHiiHandles; Index++) {
674
675 Length = 0;
676 ExtractDataFromHiiHandle (HiiHandleBuffer[Index], &Length, NULL, &HiiGuid);
677
678 if (CompareGuid (&HiiGuid, Guid)) {
679 HiiHandle = HiiHandleBuffer[Index];
680 break;
681 }
682 }
683
684 lbl_exit:
685 FreePool (HiiHandleBuffer);
686 return HiiHandle;
687 }
688
689 /**
690 Validate that the data associated with the HiiHandle in NVRAM is within
691 the reasonable parameters for that FormSet. Values for strings and passwords
692 are not verified due to their not having the equivalent of valid range settings.
693
694 @param HiiHandle Handle of the HII database entry to query
695 @param Results If return Status is EFI_SUCCESS, Results provides valid data
696 TRUE = NVRAM Data is within parameters
697 FALSE = NVRAM Data is NOT within parameters
698
699 @retval EFI_OUT_OF_RESOURCES No enough buffer to allocate
700 @retval EFI_SUCCESS Data successfully validated
701 **/
702 EFI_STATUS
703 EFIAPI
704 ValidateDataFromHiiHandle (
705 IN FRAMEWORK_EFI_HII_HANDLE HiiHandle,
706 OUT BOOLEAN *Results
707 )
708 {
709 EFI_STATUS Status;
710 EFI_HII_PROTOCOL *Hii;
711 EFI_GUID Guid;
712 UINT8 *RawData;
713 UINT8 *OldData;
714 UINTN RawDataLength;
715 UINT8 *VariableData;
716 UINTN Index;
717 UINTN Temp;
718 UINTN SizeOfNvStore;
719 UINTN CachedStart;
720 BOOLEAN GotMatch;
721
722 RawDataLength = DEFAULT_FORM_BUFFER_SIZE;
723 SizeOfNvStore = 0;
724 CachedStart = 0;
725 GotMatch = FALSE;
726 *Results = TRUE;
727
728 Status = GetHiiInterface (&Hii);
729 if (EFI_ERROR (Status)) {
730 return Status;
731 }
732
733 //
734 // Allocate space for retrieval of IFR data
735 //
736 RawData = AllocateZeroPool (RawDataLength);
737 if (RawData == NULL) {
738 return EFI_OUT_OF_RESOURCES;
739 }
740
741 //
742 // Get all the forms associated with this HiiHandle
743 //
744 Status = Hii->GetForms (Hii, HiiHandle, 0, &RawDataLength, RawData);
745 if (EFI_ERROR (Status)) {
746 FreePool (RawData);
747
748 //
749 // Allocate space for retrieval of IFR data
750 //
751 RawData = AllocateZeroPool (RawDataLength);
752 if (RawData == NULL) {
753 return EFI_OUT_OF_RESOURCES;
754 }
755
756 //
757 // Get all the forms associated with this HiiHandle
758 //
759 Status = Hii->GetForms (Hii, HiiHandle, 0, &RawDataLength, RawData);
760 }
761
762 OldData = RawData;
763
764 //
765 // Point RawData to the beginning of the form data
766 //
767 RawData = (UINT8 *) ((UINTN) RawData + sizeof (EFI_HII_PACK_HEADER));
768
769 for (Index = 0; RawData[Index] != FRAMEWORK_EFI_IFR_END_FORM_SET_OP;) {
770 if (RawData[Index] == FRAMEWORK_EFI_IFR_FORM_SET_OP) {
771 CopyGuid (&Guid, (GUID *)(VOID *)&((FRAMEWORK_EFI_IFR_FORM_SET *) &RawData[Index])->Guid);
772 break;
773 }
774 Index = RawData[Index + 1] + Index;
775 }
776
777 for (Index = 0; RawData[Index] != FRAMEWORK_EFI_IFR_END_FORM_SET_OP;) {
778 switch (RawData[Index]) {
779
780 case FRAMEWORK_EFI_IFR_FORM_SET_OP:
781 break;
782
783 case FRAMEWORK_EFI_IFR_ONE_OF_OP:
784 case FRAMEWORK_EFI_IFR_CHECKBOX_OP:
785 case FRAMEWORK_EFI_IFR_NUMERIC_OP:
786 case FRAMEWORK_EFI_IFR_DATE_OP:
787 case FRAMEWORK_EFI_IFR_TIME_OP:
788 case FRAMEWORK_EFI_IFR_PASSWORD_OP:
789 case FRAMEWORK_EFI_IFR_STRING_OP:
790 //
791 // Remember, multiple op-codes may reference the same item, so let's keep a running
792 // marker of what the highest QuestionId that wasn't zero length. This will accurately
793 // maintain the Size of the NvStore
794 //
795 if (((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->Width != 0) {
796 Temp = ((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->QuestionId + ((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->Width;
797 if (SizeOfNvStore < Temp) {
798 SizeOfNvStore = ((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->QuestionId + ((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->Width;
799 }
800 }
801 }
802
803 Index = RawData[Index + 1] + Index;
804 }
805
806 //
807 // Allocate memory for our File Form Tags
808 //
809 VariableData = AllocateZeroPool (SizeOfNvStore);
810 if (VariableData == NULL) {
811 FreePool (OldData);
812 return EFI_OUT_OF_RESOURCES;
813 }
814
815 Status = gRT->GetVariable (
816 (CHAR16 *) L"Setup",
817 &Guid,
818 NULL,
819 &SizeOfNvStore,
820 (VOID *) VariableData
821 );
822
823 if (EFI_ERROR (Status)) {
824
825 //
826 // If there is a variable that exists already and it is larger than what we calculated the
827 // storage needs to be, we must assume the variable size from GetVariable is correct and not
828 // allow the truncation of the variable. It is very possible that the user who created the IFR
829 // we are cracking is not referring to a variable that was in a previous map, however we cannot
830 // allow it's truncation.
831 //
832 if (Status == EFI_BUFFER_TOO_SMALL) {
833 //
834 // Free the buffer that was allocated that was too small
835 //
836 FreePool (VariableData);
837
838 VariableData = AllocatePool (SizeOfNvStore);
839 if (VariableData == NULL) {
840 FreePool (OldData);
841 return EFI_OUT_OF_RESOURCES;
842 }
843
844 Status = gRT->GetVariable (
845 (CHAR16 *) L"Setup",
846 &Guid,
847 NULL,
848 &SizeOfNvStore,
849 (VOID *) VariableData
850 );
851 }
852 }
853
854 //
855 // Walk through the form and see that the variable data it refers to is ok.
856 // This allows for the possibility of stale (obsoleted) data in the variable
857 // can be overlooked without causing an error
858 //
859 for (Index = 0; RawData[Index] != FRAMEWORK_EFI_IFR_END_FORM_SET_OP;) {
860 switch (RawData[Index]) {
861
862 case FRAMEWORK_EFI_IFR_ONE_OF_OP:
863 //
864 // A one_of has no data, its the option that does - cache the storage Id
865 //
866 CachedStart = ((FRAMEWORK_EFI_IFR_ONE_OF *) &RawData[Index])->QuestionId;
867 break;
868
869 case FRAMEWORK_EFI_IFR_ONE_OF_OPTION_OP:
870 //
871 // A one_of_option can be any value
872 //
873 if (VariableData[CachedStart] == ((FRAMEWORK_EFI_IFR_ONE_OF_OPTION *) &RawData[Index])->Value) {
874 GotMatch = TRUE;
875 }
876 break;
877
878 case FRAMEWORK_EFI_IFR_END_ONE_OF_OP:
879 //
880 // At this point lets make sure that the data value in the NVRAM matches one of the options
881 //
882 if (!GotMatch) {
883 *Results = FALSE;
884 goto EXIT;
885 }
886 break;
887
888 case FRAMEWORK_EFI_IFR_CHECKBOX_OP:
889 //
890 // A checkbox is a boolean, so 0 and 1 are valid
891 // Remember, QuestionId corresponds to the offset location of the data in the variable
892 //
893 if (VariableData[((FRAMEWORK_EFI_IFR_CHECKBOX *) &RawData[Index])->QuestionId] > 1) {
894 *Results = FALSE;
895 goto EXIT;
896 }
897 break;
898
899 case FRAMEWORK_EFI_IFR_NUMERIC_OP:
900 if ((VariableData[((FRAMEWORK_EFI_IFR_NUMERIC *)&RawData[Index])->QuestionId] < ((FRAMEWORK_EFI_IFR_NUMERIC *)&RawData[Index])->Minimum) ||
901 (VariableData[((FRAMEWORK_EFI_IFR_NUMERIC *)&RawData[Index])->QuestionId] > ((FRAMEWORK_EFI_IFR_NUMERIC *)&RawData[Index])->Maximum)) {
902 *Results = FALSE;
903 goto EXIT;
904 }
905 break;
906
907 }
908
909 Index = RawData[Index + 1] + Index;
910 }
911
912 EXIT:
913 //
914 // Free our temporary repository of form data
915 //
916 FreePool (OldData);
917 FreePool (VariableData);
918
919 return EFI_SUCCESS;
920 }
921
922