]>
Commit | Line | Data |
---|---|---|
c7f33ca4 | 1 | /*++\r |
2 | \r | |
3e99020d | 3 | Copyright (c) 2007 - 2010, Intel Corporation. All rights reserved.<BR>\r |
4ea9375a | 4 | This program and the accompanying materials\r |
c7f33ca4 | 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 | UefiIfrForm.c\r | |
15 | \r | |
16 | Abstract:\r | |
17 | \r | |
18 | Common Library Routines to assist handle HII elements.\r | |
19 | \r | |
20 | --*/\r | |
21 | \r | |
22 | #include "UefiIfrLibrary.h"\r | |
23 | \r | |
24 | //\r | |
25 | // Fake <ConfigHdr>\r | |
26 | //\r | |
27 | UINT16 mFakeConfigHdr[] = L"GUID=00000000000000000000000000000000&NAME=0000&PATH=0";\r | |
28 | \r | |
c7f33ca4 | 29 | EFI_STATUS\r |
30 | GetPackageDataFromPackageList (\r | |
31 | IN EFI_HII_PACKAGE_LIST_HEADER *HiiPackageList,\r | |
32 | IN UINT32 PackageIndex,\r | |
33 | OUT UINT32 *BufferLen,\r | |
34 | OUT EFI_HII_PACKAGE_HEADER **Buffer\r | |
35 | )\r | |
36 | {\r | |
37 | UINT32 Index;\r | |
38 | EFI_HII_PACKAGE_HEADER *Package;\r | |
39 | UINT32 Offset;\r | |
40 | UINT32 PackageListLength;\r | |
41 | EFI_HII_PACKAGE_HEADER PackageHeader = {0, 0};\r | |
42 | \r | |
43 | ASSERT(HiiPackageList != NULL);\r | |
44 | \r | |
45 | if ((BufferLen == NULL) || (Buffer == NULL)) {\r | |
46 | return EFI_INVALID_PARAMETER;\r | |
47 | }\r | |
48 | \r | |
49 | Package = NULL;\r | |
50 | Index = 0;\r | |
51 | Offset = sizeof (EFI_HII_PACKAGE_LIST_HEADER);\r | |
52 | EfiCopyMem (&PackageListLength, &HiiPackageList->PackageLength, sizeof (UINT32));\r | |
53 | while (Offset < PackageListLength) {\r | |
54 | Package = (EFI_HII_PACKAGE_HEADER *) (((UINT8 *) HiiPackageList) + Offset);\r | |
55 | EfiCopyMem (&PackageHeader, Package, sizeof (EFI_HII_PACKAGE_HEADER));\r | |
56 | if (Index == PackageIndex) {\r | |
57 | break;\r | |
58 | }\r | |
59 | Offset += PackageHeader.Length;\r | |
60 | Index++;\r | |
61 | }\r | |
62 | if (Offset >= PackageListLength) {\r | |
63 | //\r | |
64 | // no package found in this Package List\r | |
65 | //\r | |
66 | return EFI_NOT_FOUND;\r | |
67 | }\r | |
68 | \r | |
69 | *BufferLen = PackageHeader.Length;\r | |
70 | *Buffer = Package;\r | |
71 | return EFI_SUCCESS;\r | |
72 | }\r | |
73 | \r | |
c7f33ca4 | 74 | EFI_STATUS\r |
75 | UpdateFormPackageData (\r | |
76 | IN EFI_GUID *FormSetGuid,\r | |
77 | IN EFI_FORM_ID FormId,\r | |
78 | IN EFI_HII_PACKAGE_HEADER *Package,\r | |
79 | IN UINT32 PackageLength,\r | |
80 | IN UINT16 Label,\r | |
81 | IN BOOLEAN Insert,\r | |
82 | IN EFI_HII_UPDATE_DATA *Data,\r | |
83 | OUT UINT8 **TempBuffer,\r | |
84 | OUT UINT32 *TempBufferSize\r | |
85 | )\r | |
86 | {\r | |
c7f33ca4 | 87 | UINT8 *BufferPos;\r |
88 | EFI_HII_PACKAGE_HEADER PackageHeader;\r | |
3e99020d | 89 | UINT32 Offset;\r |
c7f33ca4 | 90 | EFI_IFR_OP_HEADER *IfrOpHdr;\r |
91 | BOOLEAN GetFormSet;\r | |
92 | BOOLEAN GetForm;\r | |
93 | UINT8 ExtendOpCode;\r | |
94 | UINT16 LabelNumber;\r | |
95 | BOOLEAN Updated;\r | |
c7f33ca4 | 96 | \r |
97 | if ((TempBuffer == NULL) || (TempBufferSize == NULL)) {\r | |
98 | return EFI_INVALID_PARAMETER;\r | |
99 | }\r | |
100 | \r | |
101 | *TempBufferSize = PackageLength;\r | |
102 | if (Data != NULL) {\r | |
103 | *TempBufferSize += Data->Offset;\r | |
104 | }\r | |
105 | *TempBuffer = EfiLibAllocateZeroPool (*TempBufferSize);\r | |
106 | if (*TempBuffer == NULL) {\r | |
107 | return EFI_OUT_OF_RESOURCES;\r | |
108 | }\r | |
109 | \r | |
110 | EfiCopyMem (*TempBuffer, Package, sizeof (EFI_HII_PACKAGE_HEADER));\r | |
111 | *TempBufferSize = sizeof (EFI_HII_PACKAGE_HEADER);\r | |
112 | BufferPos = *TempBuffer + sizeof (EFI_HII_PACKAGE_HEADER);\r | |
113 | \r | |
114 | EfiCopyMem (&PackageHeader, Package, sizeof (EFI_HII_PACKAGE_HEADER));\r | |
115 | IfrOpHdr = (EFI_IFR_OP_HEADER *)((UINT8 *) Package + sizeof (EFI_HII_PACKAGE_HEADER));\r | |
116 | Offset = sizeof (EFI_HII_PACKAGE_HEADER);\r | |
4cb43192 | 117 | GetFormSet = (BOOLEAN)((FormSetGuid == NULL) ? TRUE : FALSE);\r |
c7f33ca4 | 118 | GetForm = FALSE;\r |
119 | Updated = FALSE;\r | |
120 | \r | |
3e99020d | 121 | while (!Updated && Offset < PackageHeader.Length) {\r |
c7f33ca4 | 122 | EfiCopyMem (BufferPos, IfrOpHdr, IfrOpHdr->Length);\r |
123 | BufferPos += IfrOpHdr->Length;\r | |
124 | *TempBufferSize += IfrOpHdr->Length;\r | |
125 | \r | |
126 | switch (IfrOpHdr->OpCode) {\r | |
127 | case EFI_IFR_FORM_SET_OP :\r | |
128 | if (FormSetGuid != NULL) {\r | |
129 | if (EfiCompareMem (&((EFI_IFR_FORM_SET *) IfrOpHdr)->Guid, FormSetGuid, sizeof (EFI_GUID)) == 0) {\r | |
130 | GetFormSet = TRUE;\r | |
3e99020d LG |
131 | } else {\r |
132 | GetFormSet = FALSE;\r | |
c7f33ca4 | 133 | }\r |
134 | }\r | |
135 | break;\r | |
136 | \r | |
137 | case EFI_IFR_FORM_OP:\r | |
138 | if (EfiCompareMem (&((EFI_IFR_FORM *) IfrOpHdr)->FormId, &FormId, sizeof (EFI_FORM_ID)) == 0) {\r | |
139 | GetForm = TRUE;\r | |
3e99020d LG |
140 | } else {\r |
141 | GetForm = FALSE;\r | |
c7f33ca4 | 142 | }\r |
143 | break;\r | |
144 | \r | |
145 | case EFI_IFR_GUID_OP :\r | |
3e99020d | 146 | if (!GetFormSet || !GetForm) {\r |
c7f33ca4 | 147 | //\r |
148 | // Go to the next Op-Code\r | |
149 | //\r | |
3e99020d LG |
150 | break;\r |
151 | }\r | |
152 | \r | |
153 | if (!EfiCompareGuid (&((EFI_IFR_GUID *) IfrOpHdr)->Guid, &mIfrVendorGuid)) {\r | |
154 | //\r | |
155 | // GUID mismatch, skip this op-code\r | |
156 | //\r | |
157 | break;\r | |
c7f33ca4 | 158 | }\r |
159 | \r | |
160 | ExtendOpCode = ((EFI_IFR_GUID_LABEL *) IfrOpHdr)->ExtendOpCode;\r | |
161 | EfiCopyMem (&LabelNumber, &((EFI_IFR_GUID_LABEL *)IfrOpHdr)->Number, sizeof (UINT16));\r | |
162 | if ((ExtendOpCode != EFI_IFR_EXTEND_OP_LABEL) || (LabelNumber != Label)) {\r | |
163 | //\r | |
164 | // Go to the next Op-Code\r | |
165 | //\r | |
3e99020d | 166 | break;\r |
c7f33ca4 | 167 | }\r |
168 | \r | |
3e99020d | 169 | if (Insert) {\r |
c7f33ca4 | 170 | //\r |
3e99020d | 171 | // Insert data after current Label, skip myself\r |
c7f33ca4 | 172 | //\r |
3e99020d LG |
173 | Offset += IfrOpHdr->Length;\r |
174 | IfrOpHdr = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (IfrOpHdr) + IfrOpHdr->Length);\r | |
c7f33ca4 | 175 | } else {\r |
176 | //\r | |
3e99020d | 177 | // Replace data between two paired Label, try to find the next Label.\r |
c7f33ca4 | 178 | //\r |
179 | while (TRUE) {\r | |
180 | Offset += IfrOpHdr->Length;\r | |
181 | //\r | |
182 | // Search the next label and Fail if not label found.\r | |
183 | //\r | |
184 | if (Offset >= PackageHeader.Length) {\r | |
185 | goto Fail;\r | |
186 | }\r | |
187 | IfrOpHdr = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (IfrOpHdr) + IfrOpHdr->Length);\r | |
188 | if (IfrOpHdr->OpCode == EFI_IFR_GUID_OP) {\r | |
189 | ExtendOpCode = ((EFI_IFR_GUID_LABEL *) IfrOpHdr)->ExtendOpCode;\r | |
3e99020d | 190 | if (EfiCompareGuid (&((EFI_IFR_GUID *) IfrOpHdr)->Guid, &mIfrVendorGuid) && ExtendOpCode == EFI_IFR_EXTEND_OP_LABEL) {\r |
c7f33ca4 | 191 | break;\r |
192 | }\r | |
193 | }\r | |
194 | }\r | |
3e99020d | 195 | }\r |
c7f33ca4 | 196 | \r |
3e99020d LG |
197 | //\r |
198 | // Fill in the update data\r | |
199 | //\r | |
200 | if (Data != NULL) {\r | |
201 | EfiCopyMem (BufferPos, Data->Data, Data->Offset);\r | |
202 | BufferPos += Data->Offset;\r | |
203 | *TempBufferSize += Data->Offset;\r | |
c7f33ca4 | 204 | }\r |
205 | \r | |
3e99020d LG |
206 | //\r |
207 | // Copy the reset data\r | |
208 | //\r | |
209 | EfiCopyMem (BufferPos, IfrOpHdr, PackageHeader.Length - Offset);\r | |
210 | *TempBufferSize += PackageHeader.Length - Offset;\r | |
211 | \r | |
c7f33ca4 | 212 | Updated = TRUE;\r |
213 | break;\r | |
214 | default :\r | |
215 | break;\r | |
216 | }\r | |
217 | \r | |
218 | //\r | |
219 | // Go to the next Op-Code\r | |
220 | //\r | |
221 | Offset += IfrOpHdr->Length;\r | |
222 | IfrOpHdr = (EFI_IFR_OP_HEADER *) ((CHAR8 *) (IfrOpHdr) + IfrOpHdr->Length);\r | |
223 | }\r | |
224 | \r | |
225 | //\r | |
226 | // Update the package length.\r | |
227 | //\r | |
228 | PackageHeader.Length = *TempBufferSize;\r | |
229 | EfiCopyMem (*TempBuffer, &PackageHeader, sizeof (EFI_HII_PACKAGE_HEADER));\r | |
230 | \r | |
231 | Fail:\r | |
232 | if (!Updated) {\r | |
233 | gBS->FreePool (*TempBuffer);\r | |
234 | *TempBufferSize = 0;\r | |
235 | return EFI_NOT_FOUND;\r | |
236 | }\r | |
237 | \r | |
238 | return EFI_SUCCESS;\r | |
239 | }\r | |
240 | \r | |
3e99020d LG |
241 | EFI_STATUS\r |
242 | IfrLibInitUpdateData (\r | |
243 | IN OUT EFI_HII_UPDATE_DATA *UpdateData,\r | |
244 | IN UINT32 BufferSize\r | |
245 | )\r | |
246 | /*++\r | |
247 | \r | |
248 | Routine Description:\r | |
249 | This function initialize the data structure for dynamic opcode.\r | |
250 | \r | |
251 | Arguments:\r | |
252 | UpdateData - The adding data;\r | |
253 | BufferSize - Length of the buffer to fill dynamic opcodes.\r | |
254 | \r | |
255 | Returns:\r | |
256 | EFI_SUCCESS - Update data is initialized.\r | |
257 | EFI_INVALID_PARAMETER - UpdateData is NULL.\r | |
258 | EFI_OUT_OF_RESOURCES - No enough memory to allocate.\r | |
259 | \r | |
260 | --*/\r | |
261 | {\r | |
262 | if (UpdateData == NULL) {\r | |
263 | return EFI_INVALID_PARAMETER;\r | |
264 | }\r | |
265 | \r | |
266 | UpdateData->BufferSize = BufferSize;\r | |
267 | UpdateData->Offset = 0;\r | |
268 | UpdateData->Data = EfiLibAllocatePool (BufferSize);\r | |
269 | \r | |
270 | return (UpdateData->Data != NULL) ? EFI_SUCCESS : EFI_OUT_OF_RESOURCES;\r | |
271 | }\r | |
272 | \r | |
273 | EFI_STATUS\r | |
274 | IfrLibFreeUpdateData (\r | |
275 | IN EFI_HII_UPDATE_DATA *UpdateData\r | |
276 | )\r | |
277 | /*++\r | |
278 | \r | |
279 | Routine Description:\r | |
280 | This function free the resource of update data.\r | |
281 | \r | |
282 | Arguments:\r | |
283 | UpdateData - The adding data;\r | |
284 | \r | |
285 | Returns:\r | |
286 | EFI_SUCCESS - Resource in UpdateData is released.\r | |
287 | EFI_INVALID_PARAMETER - UpdateData is NULL.\r | |
288 | \r | |
289 | --*/\r | |
290 | {\r | |
291 | EFI_STATUS Status;\r | |
292 | \r | |
293 | if (UpdateData == NULL) {\r | |
294 | return EFI_INVALID_PARAMETER;\r | |
295 | }\r | |
296 | \r | |
297 | Status = gBS->FreePool (UpdateData->Data);\r | |
298 | UpdateData->Data = NULL;\r | |
299 | \r | |
300 | return Status;\r | |
301 | }\r | |
302 | \r | |
c7f33ca4 | 303 | EFI_STATUS\r |
304 | IfrLibUpdateForm (\r | |
305 | IN EFI_HII_HANDLE Handle,\r | |
306 | IN EFI_GUID *FormSetGuid, OPTIONAL\r | |
307 | IN EFI_FORM_ID FormId,\r | |
308 | IN UINT16 Label,\r | |
309 | IN BOOLEAN Insert,\r | |
310 | IN EFI_HII_UPDATE_DATA *Data\r | |
311 | )\r | |
312 | /*++\r | |
313 | \r | |
314 | Routine Description:\r | |
315 | This function allows the caller to update a form that has\r | |
316 | previously been registered with the EFI HII database.\r | |
317 | \r | |
318 | Arguments:\r | |
319 | Handle - Hii Handle\r | |
320 | FormSetGuid - The formset should be updated.\r | |
321 | FormId - The form should be updated.\r | |
322 | Label - Update information starting immediately after this label in the IFR\r | |
323 | Insert - If TRUE and Data is not NULL, insert data after Label.\r | |
324 | If FALSE, replace opcodes between two labels with Data\r | |
325 | Data - The adding data; If NULL, remove opcodes between two Label.\r | |
326 | \r | |
327 | Returns:\r | |
328 | EFI_SUCCESS - Update success.\r | |
329 | Other - Update fail.\r | |
330 | \r | |
331 | --*/\r | |
332 | {\r | |
333 | EFI_STATUS Status;\r | |
334 | EFI_HII_DATABASE_PROTOCOL *HiiDatabase;\r | |
335 | EFI_HII_PACKAGE_LIST_HEADER *HiiPackageList;\r | |
336 | UINT32 Index;\r | |
337 | EFI_HII_PACKAGE_LIST_HEADER *UpdateBuffer;\r | |
338 | UINTN BufferSize;\r | |
339 | UINT8 *UpdateBufferPos;\r | |
340 | EFI_HII_PACKAGE_HEADER PackageHeader;\r | |
341 | EFI_HII_PACKAGE_HEADER *Package;\r | |
342 | UINT32 PackageLength;\r | |
343 | EFI_HII_PACKAGE_HEADER *TempBuffer;\r | |
344 | UINT32 TempBufferSize;\r | |
345 | BOOLEAN Updated;\r | |
346 | \r | |
347 | if (Data == NULL) {\r | |
348 | return EFI_INVALID_PARAMETER;\r | |
349 | }\r | |
350 | \r | |
351 | LocateHiiProtocols ();\r | |
352 | HiiDatabase = gIfrLibHiiDatabase;\r | |
353 | \r | |
354 | //\r | |
355 | // Get the orginal package list\r | |
356 | //\r | |
357 | BufferSize = 0;\r | |
358 | HiiPackageList = NULL;\r | |
359 | Status = HiiDatabase->ExportPackageLists (HiiDatabase, Handle, &BufferSize, HiiPackageList);\r | |
360 | if (Status == EFI_BUFFER_TOO_SMALL) {\r | |
361 | HiiPackageList = EfiLibAllocatePool (BufferSize);\r | |
362 | ASSERT (HiiPackageList != NULL);\r | |
363 | \r | |
364 | Status = HiiDatabase->ExportPackageLists (HiiDatabase, Handle, &BufferSize, HiiPackageList);\r | |
365 | if (EFI_ERROR (Status)) {\r | |
366 | gBS->FreePool (HiiPackageList);\r | |
367 | return Status;\r | |
368 | }\r | |
369 | }\r | |
370 | \r | |
371 | //\r | |
372 | // Calculate and allocate space for retrieval of IFR data\r | |
373 | //\r | |
374 | BufferSize += Data->Offset;\r | |
375 | UpdateBuffer = EfiLibAllocateZeroPool (BufferSize);\r | |
376 | if (UpdateBuffer == NULL) {\r | |
377 | return EFI_OUT_OF_RESOURCES;\r | |
378 | }\r | |
379 | \r | |
380 | UpdateBufferPos = (UINT8 *) UpdateBuffer;\r | |
381 | \r | |
382 | //\r | |
383 | // copy the package list header\r | |
384 | //\r | |
385 | EfiCopyMem (UpdateBufferPos, HiiPackageList, sizeof (EFI_HII_PACKAGE_LIST_HEADER));\r | |
386 | UpdateBufferPos += sizeof (EFI_HII_PACKAGE_LIST_HEADER);\r | |
387 | \r | |
388 | Updated = FALSE;\r | |
389 | for (Index = 0; ; Index++) {\r | |
390 | Status = GetPackageDataFromPackageList (HiiPackageList, Index, &PackageLength, &Package);\r | |
391 | if (Status == EFI_SUCCESS) {\r | |
392 | EfiCopyMem (&PackageHeader, Package, sizeof (EFI_HII_PACKAGE_HEADER));\r | |
3e99020d | 393 | if ((PackageHeader.Type == EFI_HII_PACKAGE_FORMS) && !Updated) {\r |
c7f33ca4 | 394 | Status = UpdateFormPackageData (FormSetGuid, FormId, Package, PackageLength, Label, Insert, Data, (UINT8 **)&TempBuffer, &TempBufferSize);\r |
395 | if (!EFI_ERROR(Status)) {\r | |
396 | if (FormSetGuid == NULL) {\r | |
397 | Updated = TRUE;\r | |
398 | }\r | |
399 | EfiCopyMem (UpdateBufferPos, TempBuffer, TempBufferSize);\r | |
400 | UpdateBufferPos += TempBufferSize;\r | |
401 | gBS->FreePool (TempBuffer);\r | |
402 | continue;\r | |
403 | }\r | |
404 | }\r | |
405 | \r | |
406 | EfiCopyMem (UpdateBufferPos, Package, PackageLength);\r | |
407 | UpdateBufferPos += PackageLength;\r | |
408 | } else if (Status == EFI_NOT_FOUND) {\r | |
409 | break;\r | |
410 | } else {\r | |
411 | gBS->FreePool (HiiPackageList);\r | |
412 | return Status;\r | |
413 | }\r | |
414 | }\r | |
415 | \r | |
416 | //\r | |
417 | // Update package list length\r | |
418 | //\r | |
419 | BufferSize = UpdateBufferPos - (UINT8 *) UpdateBuffer;\r | |
420 | EfiCopyMem (&UpdateBuffer->PackageLength, &BufferSize, sizeof (UINT32));\r | |
421 | \r | |
422 | gBS->FreePool (HiiPackageList);\r | |
423 | \r | |
424 | return HiiDatabase->UpdatePackageList (HiiDatabase, Handle, UpdateBuffer);\r | |
425 | }\r | |
426 | \r | |
427 | EFI_STATUS\r | |
428 | IfrLibCreatePopUp (\r | |
429 | IN UINTN NumberOfLines,\r | |
430 | OUT EFI_INPUT_KEY *KeyValue,\r | |
431 | IN CHAR16 *String,\r | |
432 | ...\r | |
433 | )\r | |
434 | /*++\r | |
435 | \r | |
436 | Routine Description:\r | |
437 | Draw a dialog and return the selected key.\r | |
438 | \r | |
439 | Arguments:\r | |
440 | NumberOfLines - The number of lines for the dialog box\r | |
441 | KeyValue - The EFI_KEY value returned if HotKey is TRUE..\r | |
442 | String - Pointer to the first string in the list\r | |
443 | ... - A series of (quantity == NumberOfLines) text strings which\r | |
444 | will be used to construct the dialog box\r | |
445 | \r | |
446 | Returns:\r | |
447 | EFI_SUCCESS - Displayed dialog and received user interaction\r | |
448 | EFI_INVALID_PARAMETER - One of the parameters was invalid.\r | |
449 | \r | |
450 | --*/\r | |
451 | {\r | |
452 | UINTN Index;\r | |
453 | UINTN Count;\r | |
454 | UINTN Start;\r | |
455 | UINTN End;\r | |
456 | UINTN Top;\r | |
457 | UINTN Bottom;\r | |
458 | CHAR16 *StringPtr;\r | |
459 | UINTN LeftColumn;\r | |
460 | UINTN RightColumn;\r | |
461 | UINTN TopRow;\r | |
462 | UINTN BottomRow;\r | |
463 | UINTN DimensionsWidth;\r | |
464 | UINTN DimensionsHeight;\r | |
465 | VA_LIST Marker;\r | |
466 | EFI_INPUT_KEY Key;\r | |
467 | UINTN LargestString;\r | |
468 | CHAR16 *StackString;\r | |
469 | EFI_STATUS Status;\r | |
470 | UINTN StringLen;\r | |
471 | CHAR16 *LineBuffer;\r | |
472 | CHAR16 **StringArray;\r | |
473 | EFI_EVENT TimerEvent;\r | |
474 | EFI_EVENT WaitList[2];\r | |
475 | UINTN CurrentAttribute;\r | |
476 | EFI_SIMPLE_TEXT_OUT_PROTOCOL *ConOut;\r | |
477 | \r | |
478 | if ((KeyValue == NULL) || (String == NULL)) {\r | |
479 | return EFI_INVALID_PARAMETER;\r | |
480 | }\r | |
481 | \r | |
482 | TopRow = 0;\r | |
483 | BottomRow = 0;\r | |
484 | LeftColumn = 0;\r | |
485 | RightColumn = 0;\r | |
486 | \r | |
487 | ConOut = gST->ConOut;\r | |
488 | ConOut->QueryMode (ConOut, ConOut->Mode->Mode, &RightColumn, &BottomRow);\r | |
489 | \r | |
490 | DimensionsWidth = RightColumn - LeftColumn;\r | |
491 | DimensionsHeight = BottomRow - TopRow;\r | |
492 | \r | |
493 | CurrentAttribute = ConOut->Mode->Attribute;\r | |
494 | \r | |
495 | LineBuffer = EfiLibAllocateZeroPool (DimensionsWidth * sizeof (CHAR16));\r | |
496 | ASSERT (LineBuffer != NULL);\r | |
497 | \r | |
498 | //\r | |
499 | // Determine the largest string in the dialog box\r | |
500 | // Notice we are starting with 1 since String is the first string\r | |
501 | //\r | |
502 | StringArray = EfiLibAllocateZeroPool (NumberOfLines * sizeof (CHAR16 *));\r | |
503 | LargestString = EfiStrLen (String);\r | |
504 | StringArray[0] = String;\r | |
505 | \r | |
506 | VA_START (Marker, String);\r | |
507 | for (Index = 1; Index < NumberOfLines; Index++) {\r | |
508 | StackString = VA_ARG (Marker, CHAR16 *);\r | |
509 | \r | |
510 | if (StackString == NULL) {\r | |
511 | return EFI_INVALID_PARAMETER;\r | |
512 | }\r | |
513 | \r | |
514 | StringArray[Index] = StackString;\r | |
515 | StringLen = EfiStrLen (StackString);\r | |
516 | if (StringLen > LargestString) {\r | |
517 | LargestString = StringLen;\r | |
518 | }\r | |
519 | }\r | |
520 | \r | |
521 | if ((LargestString + 2) > DimensionsWidth) {\r | |
522 | LargestString = DimensionsWidth - 2;\r | |
523 | }\r | |
524 | \r | |
525 | //\r | |
526 | // Subtract the PopUp width from total Columns, allow for one space extra on\r | |
527 | // each end plus a border.\r | |
528 | //\r | |
529 | Start = (DimensionsWidth - LargestString - 2) / 2 + LeftColumn + 1;\r | |
530 | End = Start + LargestString + 1;\r | |
531 | \r | |
532 | Top = ((DimensionsHeight - NumberOfLines - 2) / 2) + TopRow - 1;\r | |
533 | Bottom = Top + NumberOfLines + 2;\r | |
534 | \r | |
535 | //\r | |
536 | // Disable cursor\r | |
537 | //\r | |
538 | ConOut->EnableCursor (ConOut, FALSE);\r | |
539 | ConOut->SetAttribute (ConOut, EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE);\r | |
540 | \r | |
541 | StringPtr = &LineBuffer[0];\r | |
542 | *StringPtr++ = BOXDRAW_DOWN_RIGHT;\r | |
543 | for (Index = 0; Index < LargestString; Index++) {\r | |
544 | *StringPtr++ = BOXDRAW_HORIZONTAL;\r | |
545 | }\r | |
546 | *StringPtr++ = BOXDRAW_DOWN_LEFT;\r | |
547 | *StringPtr = L'\0';\r | |
548 | \r | |
549 | ConOut->SetCursorPosition (ConOut, Start, Top);\r | |
550 | ConOut->OutputString (ConOut, LineBuffer);\r | |
551 | \r | |
552 | for (Index = 0; Index < NumberOfLines; Index++) {\r | |
553 | StringPtr = &LineBuffer[0];\r | |
554 | *StringPtr++ = BOXDRAW_VERTICAL;\r | |
555 | \r | |
556 | for (Count = 0; Count < LargestString; Count++) {\r | |
557 | StringPtr[Count] = L' ';\r | |
558 | }\r | |
559 | \r | |
560 | StringLen = EfiStrLen (StringArray[Index]);\r | |
561 | if (StringLen > LargestString) {\r | |
562 | StringLen = LargestString;\r | |
563 | }\r | |
564 | EfiCopyMem (\r | |
565 | StringPtr + ((LargestString - StringLen) / 2),\r | |
566 | StringArray[Index],\r | |
567 | StringLen * sizeof (CHAR16)\r | |
568 | );\r | |
569 | StringPtr += LargestString;\r | |
570 | \r | |
571 | *StringPtr++ = BOXDRAW_VERTICAL;\r | |
572 | *StringPtr = L'\0';\r | |
573 | \r | |
574 | ConOut->SetCursorPosition (ConOut, Start, Top + 1 + Index);\r | |
575 | ConOut->OutputString (ConOut, LineBuffer);\r | |
576 | }\r | |
577 | \r | |
578 | StringPtr = &LineBuffer[0];\r | |
579 | *StringPtr++ = BOXDRAW_UP_RIGHT;\r | |
580 | for (Index = 0; Index < LargestString; Index++) {\r | |
581 | *StringPtr++ = BOXDRAW_HORIZONTAL;\r | |
582 | }\r | |
583 | *StringPtr++ = BOXDRAW_UP_LEFT;\r | |
584 | *StringPtr = L'\0';\r | |
585 | \r | |
586 | ConOut->SetCursorPosition (ConOut, Start, Top + NumberOfLines + 1);\r | |
587 | ConOut->OutputString (ConOut, LineBuffer);\r | |
588 | \r | |
589 | do {\r | |
590 | Status = gBS->CreateEvent (EFI_EVENT_TIMER, 0, NULL, NULL, &TimerEvent);\r | |
591 | \r | |
592 | //\r | |
593 | // Set a timer event of 1 second expiration\r | |
594 | //\r | |
595 | gBS->SetTimer (\r | |
596 | TimerEvent,\r | |
597 | TimerRelative,\r | |
598 | 10000000\r | |
599 | );\r | |
600 | \r | |
601 | //\r | |
602 | // Wait for the keystroke event or the timer\r | |
603 | //\r | |
604 | WaitList[0] = gST->ConIn->WaitForKey;\r | |
605 | WaitList[1] = TimerEvent;\r | |
606 | Status = gBS->WaitForEvent (2, WaitList, &Index);\r | |
607 | \r | |
608 | //\r | |
609 | // Check for the timer expiration\r | |
610 | //\r | |
611 | if (!EFI_ERROR (Status) && Index == 1) {\r | |
612 | Status = EFI_TIMEOUT;\r | |
613 | }\r | |
614 | \r | |
615 | gBS->CloseEvent (TimerEvent);\r | |
616 | } while (Status == EFI_TIMEOUT);\r | |
617 | \r | |
618 | Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);\r | |
619 | EfiCopyMem (KeyValue, &Key, sizeof (EFI_INPUT_KEY));\r | |
620 | \r | |
621 | ConOut->SetAttribute (ConOut, CurrentAttribute);\r | |
622 | ConOut->EnableCursor (ConOut, TRUE);\r | |
623 | \r | |
624 | return Status;\r | |
625 | }\r | |
626 | \r | |
627 | EFI_STATUS\r | |
628 | ExtractDefault(\r | |
629 | IN VOID *Buffer,\r | |
630 | IN UINTN *BufferSize,\r | |
631 | UINTN Number,\r | |
632 | ...\r | |
633 | )\r | |
634 | /*++\r | |
635 | \r | |
636 | Routine Description:\r | |
637 | \r | |
638 | Configure the buffer accrording to ConfigBody strings.\r | |
639 | \r | |
640 | Arguments:\r | |
641 | DefaultId - the ID of default.\r | |
642 | Buffer - the start address of buffer.\r | |
643 | BufferSize - the size of buffer.\r | |
644 | Number - the number of the strings.\r | |
645 | \r | |
646 | Returns:\r | |
647 | EFI_BUFFER_TOO_SMALL - the BufferSize is too small to operate.\r | |
648 | EFI_INVALID_PARAMETER - Buffer is NULL or BufferSize is 0.\r | |
649 | EFI_SUCCESS - Operation successful.\r | |
650 | \r | |
651 | --*/\r | |
652 | {\r | |
653 | VA_LIST Args;\r | |
654 | UINTN Index;\r | |
655 | UINT32 TotalLen;\r | |
656 | UINT8 *BufCfgArray;\r | |
657 | UINT8 *BufferPos;\r | |
658 | UINT16 Offset;\r | |
659 | UINT16 Width;\r | |
660 | UINT8 *Value;\r | |
661 | \r | |
662 | if ((Buffer == NULL) || (BufferSize == NULL)) {\r | |
663 | return EFI_INVALID_PARAMETER;\r | |
664 | }\r | |
665 | \r | |
666 | Offset = 0;\r | |
667 | Width = 0;\r | |
668 | Value = NULL;\r | |
669 | \r | |
670 | VA_START (Args, Number);\r | |
671 | for (Index = 0; Index < Number; Index++) {\r | |
672 | BufCfgArray = (UINT8 *) VA_ARG (Args, VOID *);\r | |
673 | EfiCopyMem (&TotalLen, BufCfgArray, sizeof (UINT32));\r | |
674 | BufferPos = BufCfgArray + sizeof (UINT32);\r | |
675 | \r | |
676 | while ((UINT32)(BufferPos - BufCfgArray) < TotalLen) {\r | |
677 | EfiCopyMem (&Offset, BufferPos, sizeof (UINT16));\r | |
678 | BufferPos += sizeof (UINT16);\r | |
679 | EfiCopyMem (&Width, BufferPos, sizeof (UINT16));\r | |
680 | BufferPos += sizeof (UINT16);\r | |
681 | Value = BufferPos;\r | |
682 | BufferPos += Width;\r | |
683 | \r | |
684 | if ((UINTN)(Offset + Width) > *BufferSize) {\r | |
685 | return EFI_BUFFER_TOO_SMALL;\r | |
686 | }\r | |
687 | \r | |
688 | EfiCopyMem ((UINT8 *)Buffer + Offset, Value, Width);\r | |
689 | }\r | |
690 | }\r | |
691 | VA_END (Args);\r | |
692 | \r | |
693 | *BufferSize = (UINTN)Offset;\r | |
694 | \r | |
695 | return EFI_SUCCESS;\r | |
696 | }\r | |
697 | \r | |
3e99020d LG |
698 | EFI_STATUS\r |
699 | ExtractBlockName (\r | |
700 | IN UINT8 *Buffer,\r | |
701 | OUT CHAR16 **BlockName\r | |
702 | )\r | |
703 | /*++\r | |
704 | \r | |
705 | Routine Description:\r | |
706 | \r | |
707 | Extract block name from the array generated by VFR compiler. The name of\r | |
708 | this array is "Vfr + <StorageName> + BlockName", e.g. "VfrMyIfrNVDataBlockName".\r | |
709 | Format of this array is:\r | |
710 | Array length | 4-bytes\r | |
711 | Offset | 2-bytes\r | |
712 | Width | 2-bytes\r | |
713 | Offset | 2-bytes\r | |
714 | Width | 2-bytes\r | |
715 | ... ...\r | |
716 | \r | |
717 | Arguments:\r | |
718 | Buffer - Array generated by VFR compiler.\r | |
719 | BlockName - The returned <BlockName>\r | |
720 | \r | |
721 | Returns:\r | |
722 | EFI_OUT_OF_RESOURCES - Run out of memory resource.\r | |
723 | EFI_INVALID_PARAMETER - Buffer is NULL or BlockName is NULL.\r | |
724 | EFI_SUCCESS - Operation successful.\r | |
725 | \r | |
726 | --*/\r | |
727 | {\r | |
728 | UINTN Index;\r | |
729 | UINT32 Length;\r | |
730 | UINT32 BlockNameNumber;\r | |
731 | UINTN HexStringBufferLen;\r | |
732 | CHAR16 *StringPtr;\r | |
733 | \r | |
734 | if ((Buffer == NULL) || (BlockName == NULL)) {\r | |
735 | return EFI_INVALID_PARAMETER;\r | |
736 | }\r | |
737 | \r | |
738 | //\r | |
739 | // Calculate number of Offset/Width pair\r | |
740 | //\r | |
741 | EfiCopyMem (&Length, Buffer, sizeof (UINT32));\r | |
742 | BlockNameNumber = (Length - sizeof (UINT32)) / (sizeof (UINT16) * 2);\r | |
743 | \r | |
744 | //\r | |
745 | // <BlockName> ::= &OFFSET=1234&WIDTH=1234\r | |
746 | // | 8 | 4 | 7 | 4 |\r | |
747 | //\r | |
748 | StringPtr = EfiLibAllocateZeroPool ((BlockNameNumber * (8 + 4 + 7 + 4) + 1) * sizeof (CHAR16));\r | |
749 | *BlockName = StringPtr;\r | |
750 | if (StringPtr == NULL) {\r | |
751 | return EFI_OUT_OF_RESOURCES;\r | |
752 | }\r | |
753 | \r | |
754 | Buffer += sizeof (UINT32);\r | |
755 | for (Index = 0; Index < BlockNameNumber; Index++) {\r | |
756 | EfiStrCpy (StringPtr, L"&OFFSET=");\r | |
757 | StringPtr += 8;\r | |
758 | \r | |
759 | HexStringBufferLen = 5;\r | |
760 | BufToHexString (StringPtr, &HexStringBufferLen, Buffer, sizeof (UINT16));\r | |
761 | Buffer += sizeof (UINT16);\r | |
762 | StringPtr += 4;\r | |
763 | \r | |
764 | EfiStrCpy (StringPtr, L"&WIDTH=");\r | |
765 | StringPtr += 7;\r | |
766 | \r | |
767 | HexStringBufferLen = 5;\r | |
768 | BufToHexString (StringPtr, &HexStringBufferLen, Buffer, sizeof (UINT16));\r | |
769 | Buffer += sizeof (UINT16);\r | |
770 | StringPtr += 4;\r | |
771 | }\r | |
772 | \r | |
773 | return EFI_SUCCESS;\r | |
774 | }\r | |
775 | \r | |
776 | EFI_STATUS\r | |
777 | ExtractBlockConfig (\r | |
778 | IN UINT8 *Buffer,\r | |
779 | OUT CHAR16 **BlockConfig\r | |
780 | )\r | |
781 | /*++\r | |
782 | \r | |
783 | Routine Description:\r | |
784 | \r | |
785 | Extract block config from the array generated by VFR compiler. The name of\r | |
786 | this array is "Vfr + <StorageName> + Default<HexCh>4", e.g. "VfrMyIfrNVDataDefault0000".\r | |
787 | \r | |
788 | Arguments:\r | |
789 | Buffer - Array generated by VFR compiler.\r | |
790 | BlockConfig - The returned <BlockConfig>\r | |
791 | \r | |
792 | Returns:\r | |
793 | EFI_OUT_OF_RESOURCES - Run out of memory resource.\r | |
794 | EFI_INVALID_PARAMETER - Buffer is NULL or BlockConfig is NULL.\r | |
795 | EFI_SUCCESS - Operation successful.\r | |
796 | \r | |
797 | --*/\r | |
798 | {\r | |
799 | UINT32 Length;\r | |
800 | UINT16 Width;\r | |
801 | UINTN HexStringBufferLen;\r | |
802 | CHAR16 *StringPtr;\r | |
803 | UINT8 *BufferEnd;\r | |
804 | CHAR16 *StringEnd;\r | |
805 | EFI_STATUS Status;\r | |
806 | \r | |
807 | if ((Buffer == NULL) || (BlockConfig == NULL)) {\r | |
808 | return EFI_INVALID_PARAMETER;\r | |
809 | }\r | |
810 | \r | |
811 | //\r | |
812 | // Calculate length of AltResp string\r | |
813 | // Format of Default value array is:\r | |
814 | // Array length | 4-bytes\r | |
815 | // Offset | 2-bytes\r | |
816 | // Width | 2-bytes\r | |
817 | // Value | Variable length\r | |
818 | // Offset | 2-bytes\r | |
819 | // Width | 2-bytes\r | |
820 | // Value | Variable length\r | |
821 | // ... ...\r | |
822 | // When value is 1 byte in length, overhead of AltResp string will be maximum,\r | |
823 | // BlockConfig ::= <&OFFSET=1234&WIDTH=1234&VALUE=12>+\r | |
824 | // | 8 | 4 | 7 | 4 | 7 |2|\r | |
825 | // so the maximum length of BlockConfig could be calculated as:\r | |
826 | // (ArrayLength / 5) * (8 + 4 + 7 + 4 + 7 + 2) = ArrayLength * 6.4 < ArrayLength * 7\r | |
827 | //\r | |
828 | EfiCopyMem (&Length, Buffer, sizeof (UINT32));\r | |
829 | BufferEnd = Buffer + Length;\r | |
830 | StringPtr = EfiLibAllocatePool (Length * 7 * sizeof (CHAR16));\r | |
831 | *BlockConfig = StringPtr;\r | |
832 | if (StringPtr == NULL) {\r | |
833 | return EFI_OUT_OF_RESOURCES;\r | |
834 | }\r | |
835 | StringEnd = StringPtr + (Length * 7);\r | |
836 | \r | |
837 | Buffer += sizeof (UINT32);\r | |
838 | while (Buffer < BufferEnd) {\r | |
839 | EfiStrCpy (StringPtr, L"&OFFSET=");\r | |
840 | StringPtr += 8;\r | |
841 | \r | |
842 | HexStringBufferLen = 5;\r | |
843 | BufToHexString (StringPtr, &HexStringBufferLen, Buffer, sizeof (UINT16));\r | |
844 | Buffer += sizeof (UINT16);\r | |
845 | StringPtr += 4;\r | |
846 | \r | |
847 | EfiStrCpy (StringPtr, L"&WIDTH=");\r | |
848 | StringPtr += 7;\r | |
849 | \r | |
850 | HexStringBufferLen = 5;\r | |
851 | BufToHexString (StringPtr, &HexStringBufferLen, Buffer, sizeof (UINT16));\r | |
852 | EfiCopyMem (&Width, Buffer, sizeof (UINT16));\r | |
853 | Buffer += sizeof (UINT16);\r | |
854 | StringPtr += 4;\r | |
855 | \r | |
856 | EfiStrCpy (StringPtr, L"&VALUE=");\r | |
857 | StringPtr += 7;\r | |
858 | \r | |
859 | HexStringBufferLen = StringEnd - StringPtr;\r | |
860 | Status = BufToHexString (StringPtr, &HexStringBufferLen, Buffer, Width);\r | |
861 | if (EFI_ERROR (Status)) {\r | |
862 | return Status;\r | |
863 | }\r | |
864 | Buffer += Width;\r | |
865 | StringPtr += (Width * 2);\r | |
866 | }\r | |
867 | \r | |
868 | return EFI_SUCCESS;\r | |
869 | }\r | |
870 | \r | |
871 | EFI_STATUS\r | |
872 | ConstructConfigAltResp (\r | |
873 | IN EFI_STRING ConfigRequest, OPTIONAL\r | |
874 | OUT EFI_STRING *Progress,\r | |
875 | OUT EFI_STRING *ConfigAltResp,\r | |
876 | IN EFI_GUID *Guid,\r | |
877 | IN CHAR16 *Name,\r | |
878 | IN EFI_HANDLE *DriverHandle,\r | |
879 | IN VOID *BufferStorage,\r | |
880 | IN UINTN BufferStorageSize,\r | |
881 | IN VOID *BlockNameArray, OPTIONAL\r | |
882 | IN UINTN NumberAltCfg,\r | |
883 | ...\r | |
884 | //IN UINT16 AltCfgId,\r | |
885 | //IN VOID *DefaultValueArray,\r | |
886 | )\r | |
887 | /*++\r | |
888 | \r | |
889 | Routine Description:\r | |
890 | \r | |
891 | Construct <ConfigAltResp> for a buffer storage.\r | |
892 | \r | |
893 | Arguments:\r | |
894 | ConfigRequest - The Config request string. If set to NULL, all the\r | |
895 | configurable elements will be extracted from BlockNameArray.\r | |
896 | ConfigAltResp - The returned <ConfigAltResp>.\r | |
897 | Progress - On return, points to a character in the Request.\r | |
898 | Guid - GUID of the buffer storage.\r | |
899 | Name - Name of the buffer storage.\r | |
900 | DriverHandle - The DriverHandle which is used to invoke HiiDatabase\r | |
901 | protocol interface NewPackageList().\r | |
902 | BufferStorage - Content of the buffer storage.\r | |
903 | BufferStorageSize - Length in bytes of the buffer storage.\r | |
904 | BlockNameArray - Array generated by VFR compiler.\r | |
905 | NumberAltCfg - Number of Default value array generated by VFR compiler.\r | |
906 | The sequential input parameters will be number of\r | |
907 | AltCfgId and DefaultValueArray pairs. When set to 0,\r | |
908 | there will be no <AltResp>.\r | |
909 | \r | |
910 | Returns:\r | |
911 | EFI_OUT_OF_RESOURCES - Run out of memory resource.\r | |
912 | EFI_INVALID_PARAMETER - ConfigAltResp is NULL.\r | |
913 | EFI_SUCCESS - Operation successful.\r | |
914 | \r | |
915 | --*/\r | |
916 | {\r | |
917 | EFI_STATUS Status;\r | |
918 | CHAR16 *ConfigHdr;\r | |
919 | CHAR16 *BlockName;\r | |
920 | CHAR16 *DescHdr;\r | |
921 | CHAR16 *StringPtr;\r | |
922 | CHAR16 **AltCfg;\r | |
923 | UINT16 AltCfgId;\r | |
924 | VOID *DefaultValueArray;\r | |
925 | UINTN StrBufferLen;\r | |
926 | EFI_STRING ConfigResp;\r | |
927 | EFI_STRING TempStr;\r | |
928 | VA_LIST Args;\r | |
929 | UINTN AltRespLen;\r | |
930 | UINTN Index;\r | |
931 | BOOLEAN NeedFreeConfigRequest;\r | |
932 | EFI_HII_CONFIG_ROUTING_PROTOCOL *HiiConfigRouting;\r | |
933 | \r | |
934 | if (ConfigAltResp == NULL) {\r | |
935 | return EFI_INVALID_PARAMETER;\r | |
936 | }\r | |
937 | \r | |
938 | //\r | |
939 | // Construct <ConfigHdr> : "GUID=...&NAME=...&PATH=..."\r | |
940 | //\r | |
941 | ConfigHdr = NULL;\r | |
942 | StrBufferLen = 0;\r | |
943 | Status = ConstructConfigHdr (\r | |
944 | ConfigHdr,\r | |
945 | &StrBufferLen,\r | |
946 | Guid,\r | |
947 | Name,\r | |
948 | DriverHandle\r | |
949 | );\r | |
950 | if (Status == EFI_BUFFER_TOO_SMALL) {\r | |
951 | ConfigHdr = EfiLibAllocateZeroPool (StrBufferLen);\r | |
952 | Status = ConstructConfigHdr (\r | |
953 | ConfigHdr,\r | |
954 | &StrBufferLen,\r | |
955 | Guid,\r | |
956 | Name,\r | |
957 | DriverHandle\r | |
958 | );\r | |
959 | }\r | |
960 | \r | |
54790583 | 961 | if (EFI_ERROR (Status) || (ConfigHdr == NULL)) {\r |
3e99020d LG |
962 | return Status;\r |
963 | }\r | |
964 | \r | |
965 | //\r | |
966 | // Construct <ConfigResp>\r | |
967 | //\r | |
968 | NeedFreeConfigRequest = FALSE;\r | |
969 | if (ConfigRequest == NULL) {\r | |
970 | //\r | |
971 | // If ConfigRequest is set to NULL, export all configurable elements in BlockNameArray\r | |
972 | //\r | |
973 | Status = ExtractBlockName (BlockNameArray, &BlockName);\r | |
974 | if (EFI_ERROR (Status)) {\r | |
975 | return Status;\r | |
976 | }\r | |
977 | \r | |
978 | StrBufferLen = EfiStrSize (ConfigHdr);\r | |
979 | StrBufferLen = StrBufferLen + EfiStrSize (BlockName) - sizeof (CHAR16);\r | |
980 | ConfigRequest = EfiLibAllocateZeroPool (StrBufferLen);\r | |
981 | EfiStrCpy (ConfigRequest, ConfigHdr);\r | |
982 | EfiStrCat (ConfigRequest, BlockName);\r | |
983 | NeedFreeConfigRequest = TRUE;\r | |
984 | }\r | |
985 | \r | |
986 | Status = gBS->LocateProtocol (&gEfiHiiConfigRoutingProtocolGuid, NULL, (VOID **) &HiiConfigRouting);\r | |
987 | if (EFI_ERROR (Status)) {\r | |
988 | return Status;\r | |
989 | }\r | |
990 | \r | |
991 | Status = HiiConfigRouting->BlockToConfig (\r | |
992 | HiiConfigRouting,\r | |
993 | ConfigRequest,\r | |
994 | BufferStorage,\r | |
995 | BufferStorageSize,\r | |
996 | &ConfigResp,\r | |
997 | (Progress == NULL) ? &TempStr : Progress\r | |
998 | );\r | |
999 | if (EFI_ERROR (Status)) {\r | |
1000 | return Status;\r | |
1001 | }\r | |
1002 | \r | |
1003 | //\r | |
1004 | // Construct <AltResp>\r | |
1005 | //\r | |
1006 | DescHdr = EfiLibAllocateZeroPool (NumberAltCfg * 16 * sizeof (CHAR16));\r | |
1007 | StringPtr = DescHdr;\r | |
1008 | AltCfg = EfiLibAllocateZeroPool (NumberAltCfg * sizeof (CHAR16 *));\r | |
1009 | AltRespLen = 0;\r | |
1010 | VA_START (Args, NumberAltCfg);\r | |
1011 | for (Index = 0; Index < NumberAltCfg; Index++) {\r | |
1012 | AltCfgId = (UINT16) VA_ARG (Args, UINT16);\r | |
1013 | DefaultValueArray = (UINT8 *) VA_ARG (Args, VOID *);\r | |
1014 | \r | |
1015 | //\r | |
1016 | // '&' <ConfigHdr>\r | |
1017 | //\r | |
1018 | AltRespLen += (EfiStrLen (ConfigHdr) + 1);\r | |
1019 | \r | |
1020 | StringPtr = DescHdr + Index * 16;\r | |
1021 | EfiStrCpy (StringPtr, L"&ALTCFG=");\r | |
1022 | AltRespLen += (8 + sizeof (UINT16) * 2);\r | |
1023 | \r | |
1024 | StrBufferLen = 5;\r | |
1025 | BufToHexString (StringPtr + 8, &StrBufferLen, (UINT8 *) &AltCfgId, sizeof (UINT16));\r | |
1026 | Status = ExtractBlockConfig (DefaultValueArray, &AltCfg[Index]);\r | |
1027 | if (EFI_ERROR (Status)) {\r | |
1028 | return Status;\r | |
1029 | }\r | |
1030 | AltRespLen += EfiStrLen (AltCfg[Index]);\r | |
1031 | }\r | |
1032 | VA_END (Args);\r | |
1033 | \r | |
1034 | //\r | |
1035 | // Generate the final <ConfigAltResp>\r | |
1036 | //\r | |
1037 | StrBufferLen = (EfiStrLen ((CHAR16 *) ConfigResp) + AltRespLen + 1) * sizeof (CHAR16);\r | |
1038 | TempStr = EfiLibAllocateZeroPool (StrBufferLen);\r | |
1039 | *ConfigAltResp = TempStr;\r | |
1040 | if (TempStr == NULL) {\r | |
1041 | return EFI_OUT_OF_RESOURCES;\r | |
1042 | }\r | |
1043 | \r | |
1044 | //\r | |
1045 | // <ConfigAltResp> ::= <ConfigResp> ['&' <AltResp>]*\r | |
1046 | //\r | |
1047 | EfiStrCpy (TempStr, ConfigResp);\r | |
1048 | for (Index = 0; Index < NumberAltCfg; Index++) {\r | |
1049 | EfiStrCat (TempStr, L"&");\r | |
1050 | EfiStrCat (TempStr, ConfigHdr);\r | |
1051 | EfiStrCat (TempStr, DescHdr + Index * 16);\r | |
1052 | EfiStrCat (TempStr, AltCfg[Index]);\r | |
1053 | \r | |
1054 | gBS->FreePool (AltCfg[Index]);\r | |
1055 | }\r | |
1056 | \r | |
1057 | if (NeedFreeConfigRequest) {\r | |
1058 | gBS->FreePool (ConfigRequest);\r | |
1059 | }\r | |
1060 | gBS->FreePool (ConfigHdr);\r | |
1061 | gBS->FreePool (ConfigResp);\r | |
1062 | gBS->FreePool (DescHdr);\r | |
1063 | gBS->FreePool (AltCfg);\r | |
1064 | \r | |
1065 | return EFI_SUCCESS;\r | |
1066 | }\r | |
1067 | \r | |
c7f33ca4 | 1068 | VOID\r |
1069 | SwapBuffer (\r | |
1070 | IN OUT UINT8 *Buffer,\r | |
1071 | IN UINTN BufferSize\r | |
1072 | )\r | |
1073 | /*++\r | |
1074 | \r | |
1075 | Routine Description:\r | |
1076 | Swap bytes in the buffer.\r | |
1077 | \r | |
1078 | Arguments:\r | |
1079 | Buffer - Binary buffer.\r | |
1080 | BufferSize - Size of the buffer in bytes.\r | |
1081 | \r | |
1082 | Returns:\r | |
1083 | None.\r | |
1084 | \r | |
1085 | --*/\r | |
1086 | {\r | |
1087 | UINTN Index;\r | |
1088 | UINT8 Temp;\r | |
1089 | UINTN SwapCount;\r | |
1090 | \r | |
3e99020d | 1091 | SwapCount = BufferSize / 2;\r |
c7f33ca4 | 1092 | for (Index = 0; Index < SwapCount; Index++) {\r |
1093 | Temp = Buffer[Index];\r | |
1094 | Buffer[Index] = Buffer[BufferSize - 1 - Index];\r | |
1095 | Buffer[BufferSize - 1 - Index] = Temp;\r | |
1096 | }\r | |
1097 | }\r | |
1098 | \r | |
3e99020d LG |
1099 | VOID\r |
1100 | ToLower (\r | |
1101 | IN OUT CHAR16 *Str\r | |
1102 | )\r | |
1103 | /*++\r | |
1104 | \r | |
1105 | Routine Description:\r | |
1106 | Converts the unicode character of the string from uppercase to lowercase.\r | |
1107 | \r | |
1108 | Arguments:\r | |
1109 | Str - String to be converted\r | |
1110 | \r | |
1111 | Returns:\r | |
1112 | \r | |
1113 | --*/\r | |
1114 | {\r | |
1115 | CHAR16 *Ptr;\r | |
1116 | \r | |
1117 | for (Ptr = Str; *Ptr != L'\0'; Ptr++) {\r | |
1118 | if (*Ptr >= L'A' && *Ptr <= L'Z') {\r | |
1119 | *Ptr = (CHAR16) (*Ptr - L'A' + L'a');\r | |
1120 | }\r | |
1121 | }\r | |
1122 | }\r | |
1123 | \r | |
c7f33ca4 | 1124 | EFI_STATUS\r |
1125 | BufferToHexString (\r | |
1126 | IN OUT CHAR16 *Str,\r | |
1127 | IN UINT8 *Buffer,\r | |
1128 | IN UINTN BufferSize\r | |
1129 | )\r | |
1130 | /*++\r | |
1131 | \r | |
1132 | Routine Description:\r | |
1133 | Converts binary buffer to Unicode string in reversed byte order from BufToHexString().\r | |
1134 | \r | |
1135 | Arguments:\r | |
1136 | Str - String for output\r | |
1137 | Buffer - Binary buffer.\r | |
1138 | BufferSize - Size of the buffer in bytes.\r | |
1139 | \r | |
1140 | Returns:\r | |
1141 | EFI_SUCCESS - The function completed successfully.\r | |
1142 | \r | |
1143 | --*/\r | |
1144 | {\r | |
1145 | EFI_STATUS Status;\r | |
1146 | UINT8 *NewBuffer;\r | |
1147 | UINTN StrBufferLen;\r | |
1148 | \r | |
1149 | NewBuffer = EfiLibAllocateCopyPool (BufferSize, Buffer);\r | |
1150 | SwapBuffer (NewBuffer, BufferSize);\r | |
1151 | \r | |
3e99020d | 1152 | StrBufferLen = BufferSize * 2 + 1;\r |
c7f33ca4 | 1153 | Status = BufToHexString (Str, &StrBufferLen, NewBuffer, BufferSize);\r |
1154 | \r | |
1155 | gBS->FreePool (NewBuffer);\r | |
3e99020d LG |
1156 | //\r |
1157 | // Convert the uppercase to lowercase since <HexAf> is defined in lowercase format.\r | |
1158 | //\r | |
1159 | ToLower (Str);\r | |
c7f33ca4 | 1160 | \r |
1161 | return Status;\r | |
1162 | }\r | |
1163 | \r | |
1164 | EFI_STATUS\r | |
1165 | HexStringToBuffer (\r | |
1166 | IN OUT UINT8 *Buffer,\r | |
1167 | IN OUT UINTN *BufferSize,\r | |
1168 | IN CHAR16 *Str\r | |
1169 | )\r | |
1170 | /*++\r | |
1171 | \r | |
1172 | Routine Description:\r | |
1173 | Converts Hex String to binary buffer in reversed byte order from HexStringToBuf().\r | |
1174 | \r | |
1175 | Arguments:\r | |
1176 | Buffer - Pointer to buffer that receives the data.\r | |
1177 | BufferSize - Length in bytes of the buffer to hold converted data.\r | |
1178 | If routine return with EFI_SUCCESS, containing length of converted data.\r | |
1179 | If routine return with EFI_BUFFER_TOO_SMALL, containg length of buffer desired.\r | |
1180 | Str - String to be converted from.\r | |
1181 | \r | |
1182 | Returns:\r | |
1183 | EFI_SUCCESS - The function completed successfully.\r | |
1184 | \r | |
1185 | --*/\r | |
1186 | {\r | |
1187 | EFI_STATUS Status;\r | |
1188 | UINTN ConvertedStrLen;\r | |
1189 | \r | |
1190 | ConvertedStrLen = 0;\r | |
1191 | Status = HexStringToBuf (Buffer, BufferSize, Str, &ConvertedStrLen);\r | |
1192 | if (!EFI_ERROR (Status)) {\r | |
3e99020d | 1193 | SwapBuffer (Buffer, (ConvertedStrLen + 1) / 2);\r |
c7f33ca4 | 1194 | }\r |
1195 | \r | |
1196 | return Status;\r | |
1197 | }\r | |
1198 | \r | |
3e99020d LG |
1199 | EFI_STATUS\r |
1200 | ConfigStringToUnicode (\r | |
1201 | IN OUT CHAR16 *UnicodeString,\r | |
1202 | IN OUT UINTN *StrBufferLen,\r | |
1203 | IN CHAR16 *ConfigString\r | |
1204 | )\r | |
1205 | /*++\r | |
1206 | \r | |
1207 | Routine Description:\r | |
1208 | Convert binary representation Config string (e.g. "0041004200430044") to the\r | |
1209 | original string (e.g. "ABCD"). Config string appears in <ConfigHdr> (i.e.\r | |
1210 | "&NAME=<string>"), or Name/Value pair in <ConfigBody> (i.e. "label=<string>").\r | |
1211 | \r | |
1212 | Arguments:\r | |
1213 | UnicodeString - Original Unicode string.\r | |
1214 | StrBufferLen - On input: Length in bytes of buffer to hold the Unicode string.\r | |
1215 | Includes tailing '\0' character.\r | |
1216 | On output:\r | |
1217 | If return EFI_SUCCESS, containing length of Unicode string buffer.\r | |
1218 | If return EFI_BUFFER_TOO_SMALL, containg length of string buffer desired.\r | |
1219 | ConfigString - Binary representation of Unicode String, <string> := (<HexCh>4)+\r | |
1220 | \r | |
1221 | Returns:\r | |
1222 | EFI_SUCCESS - Routine success.\r | |
1223 | EFI_BUFFER_TOO_SMALL - The string buffer is too small.\r | |
1224 | \r | |
1225 | --*/\r | |
1226 | {\r | |
1227 | UINTN Index;\r | |
1228 | UINTN Len;\r | |
1229 | UINTN BufferSize;\r | |
1230 | CHAR16 BackupChar;\r | |
1231 | \r | |
1232 | Len = EfiStrLen (ConfigString) / 4;\r | |
1233 | BufferSize = (Len + 1) * sizeof (CHAR16);\r | |
1234 | \r | |
1235 | if (*StrBufferLen < BufferSize) {\r | |
1236 | *StrBufferLen = BufferSize;\r | |
1237 | return EFI_BUFFER_TOO_SMALL;\r | |
1238 | }\r | |
1239 | \r | |
1240 | *StrBufferLen = BufferSize;\r | |
1241 | \r | |
1242 | for (Index = 0; Index < Len; Index++) {\r | |
1243 | BackupChar = ConfigString[4];\r | |
1244 | ConfigString[4] = L'\0';\r | |
1245 | \r | |
1246 | HexStringToBuf ((UINT8 *) UnicodeString, &BufferSize, ConfigString, NULL);\r | |
1247 | \r | |
1248 | ConfigString[4] = BackupChar;\r | |
1249 | \r | |
1250 | ConfigString += 4;\r | |
1251 | UnicodeString += 1;\r | |
1252 | }\r | |
1253 | \r | |
1254 | //\r | |
1255 | // Add tailing '\0' character\r | |
1256 | //\r | |
1257 | *UnicodeString = L'\0';\r | |
1258 | \r | |
1259 | return EFI_SUCCESS;\r | |
1260 | }\r | |
1261 | \r | |
1262 | EFI_STATUS\r | |
1263 | UnicodeToConfigString (\r | |
1264 | IN OUT CHAR16 *ConfigString,\r | |
1265 | IN OUT UINTN *StrBufferLen,\r | |
1266 | IN CHAR16 *UnicodeString\r | |
1267 | )\r | |
1268 | /*++\r | |
1269 | \r | |
1270 | Routine Description:\r | |
1271 | Convert Unicode string to binary representation Config string, e.g.\r | |
1272 | "ABCD" => "0041004200430044". Config string appears in <ConfigHdr> (i.e.\r | |
1273 | "&NAME=<string>"), or Name/Value pair in <ConfigBody> (i.e. "label=<string>").\r | |
1274 | \r | |
1275 | Arguments:\r | |
1276 | ConfigString - Binary representation of Unicode String, <string> := (<HexCh>4)+\r | |
1277 | StrBufferLen - On input: Length in bytes of buffer to hold the Unicode string.\r | |
1278 | Includes tailing '\0' character.\r | |
1279 | On output:\r | |
1280 | If return EFI_SUCCESS, containing length of Unicode string buffer.\r | |
1281 | If return EFI_BUFFER_TOO_SMALL, containg length of string buffer desired.\r | |
1282 | UnicodeString - Original Unicode string.\r | |
1283 | \r | |
1284 | Returns:\r | |
1285 | EFI_SUCCESS - Routine success.\r | |
1286 | EFI_BUFFER_TOO_SMALL - The string buffer is too small.\r | |
1287 | \r | |
1288 | --*/\r | |
1289 | {\r | |
1290 | UINTN Index;\r | |
1291 | UINTN Len;\r | |
1292 | UINTN BufferSize;\r | |
1293 | CHAR16 *String;\r | |
1294 | \r | |
1295 | Len = EfiStrLen (UnicodeString);\r | |
1296 | BufferSize = (Len * 4 + 1) * sizeof (CHAR16);\r | |
1297 | \r | |
1298 | if (*StrBufferLen < BufferSize) {\r | |
1299 | *StrBufferLen = BufferSize;\r | |
1300 | return EFI_BUFFER_TOO_SMALL;\r | |
1301 | }\r | |
1302 | \r | |
1303 | *StrBufferLen = BufferSize;\r | |
1304 | String = ConfigString;\r | |
1305 | \r | |
1306 | for (Index = 0; Index < Len; Index++) {\r | |
1307 | BufToHexString (ConfigString, &BufferSize, (UINT8 *) UnicodeString, 2);\r | |
1308 | \r | |
1309 | ConfigString += 4;\r | |
1310 | UnicodeString += 1;\r | |
1311 | }\r | |
1312 | \r | |
1313 | //\r | |
1314 | // Add tailing '\0' character\r | |
1315 | //\r | |
1316 | *ConfigString = L'\0';\r | |
1317 | \r | |
1318 | //\r | |
1319 | // Convert the uppercase to lowercase since <HexAf> is defined in lowercase format.\r | |
1320 | //\r | |
1321 | ToLower (String);\r | |
1322 | return EFI_SUCCESS;\r | |
1323 | }\r | |
1324 | \r | |
c7f33ca4 | 1325 | EFI_STATUS\r |
1326 | ConstructConfigHdr (\r | |
1327 | IN OUT CHAR16 *ConfigHdr,\r | |
1328 | IN OUT UINTN *StrBufferLen,\r | |
1329 | IN EFI_GUID *Guid,\r | |
1330 | IN CHAR16 *Name, OPTIONAL\r | |
1331 | IN EFI_HANDLE *DriverHandle\r | |
1332 | )\r | |
1333 | /*++\r | |
1334 | \r | |
1335 | Routine Description:\r | |
1336 | Construct <ConfigHdr> using routing information GUID/NAME/PATH.\r | |
1337 | \r | |
1338 | Arguments:\r | |
1339 | ConfigHdr - Pointer to the ConfigHdr string.\r | |
1340 | StrBufferLen - On input: Length in bytes of buffer to hold the ConfigHdr string.\r | |
1341 | Includes tailing '\0' character.\r | |
1342 | On output:\r | |
1343 | If return EFI_SUCCESS, containing length of ConfigHdr string buffer.\r | |
1344 | If return EFI_BUFFER_TOO_SMALL, containg length of string buffer desired.\r | |
1345 | Guid - Routing information: GUID.\r | |
1346 | Name - Routing information: NAME.\r | |
1347 | DriverHandle - Driver handle which contains the routing information: PATH.\r | |
1348 | \r | |
1349 | Returns:\r | |
1350 | EFI_SUCCESS - Routine success.\r | |
1351 | EFI_BUFFER_TOO_SMALL - The ConfigHdr string buffer is too small.\r | |
1352 | \r | |
1353 | --*/\r | |
1354 | {\r | |
1355 | EFI_STATUS Status;\r | |
1356 | UINTN NameStrLen;\r | |
1357 | UINTN DevicePathSize;\r | |
1358 | UINTN BufferSize;\r | |
1359 | CHAR16 *StrPtr;\r | |
1360 | EFI_DEVICE_PATH_PROTOCOL *DevicePath;\r | |
1361 | \r | |
1362 | if (Name == NULL) {\r | |
1363 | //\r | |
1364 | // There will be no "NAME" in <ConfigHdr> for Name/Value storage\r | |
1365 | //\r | |
1366 | NameStrLen = 0;\r | |
1367 | } else {\r | |
1368 | //\r | |
1369 | // For buffer storage\r | |
1370 | //\r | |
1371 | NameStrLen = EfiStrLen (Name);\r | |
1372 | }\r | |
1373 | \r | |
1374 | //\r | |
1375 | // Retrieve DevicePath Protocol associated with this HiiPackageList\r | |
1376 | //\r | |
1377 | Status = gBS->HandleProtocol (\r | |
1378 | DriverHandle,\r | |
1379 | &gEfiDevicePathProtocolGuid,\r | |
57d40fe2 | 1380 | (VOID **) &DevicePath\r |
c7f33ca4 | 1381 | );\r |
1382 | if (EFI_ERROR (Status)) {\r | |
1383 | return Status;\r | |
1384 | }\r | |
1385 | \r | |
1386 | DevicePathSize = EfiDevicePathSize (DevicePath);\r | |
1387 | \r | |
1388 | //\r | |
3e99020d LG |
1389 | // GUID=<HexCh>32&NAME=<Char>NameStrLen&PATH=<HexChar>DevicePathStrLen <NULL>\r |
1390 | // | 5 | 32 | 6 | NameStrLen*4 | 6 | DevicePathStrLen | 1 |\r | |
c7f33ca4 | 1391 | //\r |
3e99020d | 1392 | BufferSize = (5 + 32 + 6 + NameStrLen * 4 + 6 + DevicePathSize * 2 + 1) * sizeof (CHAR16);\r |
c7f33ca4 | 1393 | if (*StrBufferLen < BufferSize) {\r |
1394 | *StrBufferLen = BufferSize;\r | |
1395 | return EFI_BUFFER_TOO_SMALL;\r | |
1396 | }\r | |
1397 | \r | |
1398 | *StrBufferLen = BufferSize;\r | |
1399 | \r | |
1400 | StrPtr = ConfigHdr;\r | |
1401 | \r | |
1402 | EfiStrCpy (StrPtr, L"GUID=");\r | |
1403 | StrPtr += 5;\r | |
1404 | BufferToHexString (StrPtr, (UINT8 *) Guid, sizeof (EFI_GUID));\r | |
1405 | StrPtr += 32;\r | |
1406 | \r | |
3e99020d LG |
1407 | //\r |
1408 | // Convert name string, e.g. name "ABCD" => "&NAME=0041004200430044"\r | |
1409 | //\r | |
c7f33ca4 | 1410 | EfiStrCpy (StrPtr, L"&NAME=");\r |
1411 | StrPtr += 6;\r | |
1412 | if (Name != NULL) {\r | |
3e99020d LG |
1413 | BufferSize = (NameStrLen * 4 + 1) * sizeof (CHAR16);\r |
1414 | UnicodeToConfigString (StrPtr, &BufferSize, Name);\r | |
1415 | StrPtr += (NameStrLen * 4);\r | |
c7f33ca4 | 1416 | }\r |
1417 | \r | |
1418 | EfiStrCpy (StrPtr, L"&PATH=");\r | |
1419 | StrPtr += 6;\r | |
1420 | BufferToHexString (StrPtr, (UINT8 *) DevicePath, DevicePathSize);\r | |
1421 | \r | |
1422 | return EFI_SUCCESS;\r | |
1423 | }\r | |
1424 | \r | |
3e99020d LG |
1425 | BOOLEAN\r |
1426 | IsConfigHdrMatch (\r | |
1427 | IN EFI_STRING ConfigString,\r | |
1428 | IN EFI_GUID *StorageGuid, OPTIONAL\r | |
1429 | IN CHAR16 *StorageName OPTIONAL\r | |
1430 | )\r | |
1431 | /*++\r | |
1432 | \r | |
1433 | Routine Description:\r | |
1434 | Determines if the Routing data (Guid and Name) is correct in <ConfigHdr>.\r | |
1435 | \r | |
1436 | Arguments:\r | |
1437 | ConfigString - Either <ConfigRequest> or <ConfigResp>.\r | |
1438 | StorageGuid - GUID of the storage.\r | |
1439 | StorageName - Name of the stoarge.\r | |
1440 | \r | |
1441 | Returns:\r | |
1442 | TRUE - Routing information is correct in ConfigString.\r | |
1443 | FALSE - Routing information is incorrect in ConfigString.\r | |
1444 | \r | |
1445 | --*/\r | |
1446 | {\r | |
1447 | EFI_STATUS Status;\r | |
1448 | BOOLEAN Match;\r | |
1449 | EFI_GUID Guid;\r | |
1450 | CHAR16 *Name;\r | |
1451 | CHAR16 *StrPtr;\r | |
1452 | UINTN BufferSize;\r | |
1453 | \r | |
1454 | //\r | |
1455 | // <ConfigHdr> ::=\r | |
1456 | // GUID=<HexCh>32&NAME=<Char>NameStrLen&PATH=<HexChar>DevicePathStrLen <NULL>\r | |
1457 | // | 5 | 32 | 6 | NameStrLen*4 | 6 | DevicePathStrLen | 1 |\r | |
1458 | //\r | |
1459 | if (EfiStrLen (ConfigString) <= (5 + 32 + 6)) {\r | |
1460 | return FALSE;\r | |
1461 | }\r | |
1462 | \r | |
1463 | //\r | |
1464 | // Compare GUID\r | |
1465 | //\r | |
1466 | if (StorageGuid != NULL) {\r | |
1467 | \r | |
1468 | StrPtr = ConfigString + 5 + 32;\r | |
1469 | if (*StrPtr != L'&') {\r | |
1470 | return FALSE;\r | |
1471 | }\r | |
1472 | *StrPtr = L'\0';\r | |
1473 | \r | |
1474 | BufferSize = sizeof (EFI_GUID);\r | |
1475 | Status = HexStringToBuffer (\r | |
1476 | (UINT8 *) &Guid,\r | |
1477 | &BufferSize,\r | |
1478 | ConfigString + 5\r | |
1479 | );\r | |
1480 | *StrPtr = L'&';\r | |
1481 | \r | |
1482 | if (EFI_ERROR (Status)) {\r | |
1483 | return FALSE;\r | |
1484 | }\r | |
1485 | \r | |
1486 | if (!EfiCompareGuid (&Guid, StorageGuid)) {\r | |
1487 | return FALSE;\r | |
1488 | }\r | |
1489 | }\r | |
1490 | \r | |
1491 | //\r | |
1492 | // Compare Name\r | |
1493 | //\r | |
1494 | Match = TRUE;\r | |
1495 | if (StorageName != NULL) {\r | |
1496 | StrPtr = ConfigString + 5 + 32 + 6;\r | |
1497 | while (*StrPtr != L'\0' && *StrPtr != L'&') {\r | |
1498 | StrPtr++;\r | |
1499 | }\r | |
1500 | if (*StrPtr != L'&') {\r | |
1501 | return FALSE;\r | |
1502 | }\r | |
1503 | \r | |
1504 | *StrPtr = L'\0';\r | |
1505 | BufferSize = (EfiStrLen (ConfigString + 5 + 32 + 6) + 1) * sizeof (CHAR16);\r | |
1506 | Name = EfiLibAllocatePool (BufferSize);\r | |
1507 | ASSERT (Name != NULL);\r | |
1508 | Status = ConfigStringToUnicode (\r | |
1509 | Name,\r | |
1510 | &BufferSize,\r | |
1511 | ConfigString + 5 + 32 + 6\r | |
1512 | );\r | |
1513 | *StrPtr = L'&';\r | |
1514 | \r | |
1515 | if (EFI_ERROR (Status) || (EfiStrCmp (Name, StorageName) != 0)) {\r | |
1516 | Match = FALSE;\r | |
1517 | }\r | |
1518 | gBS->FreePool (Name);\r | |
1519 | }\r | |
1520 | \r | |
1521 | return Match;\r | |
1522 | }\r | |
1523 | \r | |
c7f33ca4 | 1524 | BOOLEAN\r |
1525 | FindBlockName (\r | |
1526 | IN OUT CHAR16 *String,\r | |
1527 | UINTN Offset,\r | |
1528 | UINTN Width\r | |
1529 | )\r | |
1530 | /*++\r | |
1531 | \r | |
1532 | Routine Description:\r | |
1533 | Search BlockName "&OFFSET=Offset&WIDTH=Width" in a string.\r | |
1534 | \r | |
1535 | Arguments:\r | |
1536 | String - The string to be searched in.\r | |
1537 | Offset - Offset in BlockName.\r | |
1538 | Width - Width in BlockName.\r | |
1539 | \r | |
1540 | Returns:\r | |
1541 | TRUE - Block name found.\r | |
1542 | FALSE - Block name not found.\r | |
1543 | \r | |
1544 | --*/\r | |
1545 | {\r | |
1546 | EFI_STATUS Status;\r | |
1547 | UINTN Data;\r | |
1548 | UINTN BufferSize;\r | |
1549 | UINTN ConvertedStrLen;\r | |
1550 | \r | |
1551 | while ((String = EfiStrStr (String, L"&OFFSET=")) != NULL) {\r | |
1552 | //\r | |
1553 | // Skip '&OFFSET='\r | |
1554 | //\r | |
1555 | String = String + 8;\r | |
1556 | \r | |
1557 | Data = 0;\r | |
1558 | BufferSize = sizeof (UINTN);\r | |
1559 | Status = HexStringToBuf ((UINT8 *) &Data, &BufferSize, String, &ConvertedStrLen);\r | |
1560 | if (EFI_ERROR (Status)) {\r | |
1561 | return FALSE;\r | |
1562 | }\r | |
1563 | String = String + ConvertedStrLen;\r | |
1564 | \r | |
1565 | if (Data != Offset) {\r | |
1566 | continue;\r | |
1567 | }\r | |
1568 | \r | |
1569 | if (EfiStrnCmp (String, L"&WIDTH=", 7) != 0) {\r | |
1570 | return FALSE;\r | |
1571 | }\r | |
1572 | String = String + 7;\r | |
1573 | \r | |
1574 | Data = 0;\r | |
1575 | BufferSize = sizeof (UINTN);\r | |
1576 | Status = HexStringToBuf ((UINT8 *) &Data, &BufferSize, String, &ConvertedStrLen);\r | |
1577 | if (EFI_ERROR (Status)) {\r | |
1578 | return FALSE;\r | |
1579 | }\r | |
1580 | if (Data == Width) {\r | |
1581 | return TRUE;\r | |
1582 | }\r | |
1583 | \r | |
1584 | String = String + ConvertedStrLen;\r | |
1585 | }\r | |
1586 | \r | |
1587 | return FALSE;\r | |
1588 | }\r | |
1589 | \r | |
1590 | EFI_STATUS\r | |
1591 | GetBrowserData (\r | |
1592 | EFI_GUID *VariableGuid, OPTIONAL\r | |
1593 | CHAR16 *VariableName, OPTIONAL\r | |
1594 | UINTN *BufferSize,\r | |
1595 | UINT8 *Buffer\r | |
1596 | )\r | |
1597 | /*++\r | |
1598 | \r | |
1599 | Routine Description:\r | |
1600 | This routine is invoked by ConfigAccess.Callback() to retrived uncommitted data from Form Browser.\r | |
1601 | \r | |
1602 | Arguments:\r | |
1603 | VariableGuid - An optional field to indicate the target variable GUID name to use.\r | |
1604 | VariableName - An optional field to indicate the target human-readable variable name.\r | |
1605 | BufferSize - On input: Length in bytes of buffer to hold retrived data.\r | |
1606 | On output:\r | |
1607 | If return EFI_BUFFER_TOO_SMALL, containg length of buffer desired.\r | |
1608 | Buffer - Buffer to hold retrived data.\r | |
1609 | \r | |
1610 | Returns:\r | |
1611 | EFI_SUCCESS - Routine success.\r | |
1612 | EFI_BUFFER_TOO_SMALL - The intput buffer is too small.\r | |
1613 | \r | |
1614 | --*/\r | |
1615 | {\r | |
1616 | EFI_STATUS Status;\r | |
1617 | CHAR16 *ConfigHdr;\r | |
1618 | CHAR16 *ConfigResp;\r | |
1619 | CHAR16 *StringPtr;\r | |
1620 | UINTN HeaderLen;\r | |
1621 | UINTN BufferLen;\r | |
1622 | CHAR16 *Progress;\r | |
1623 | EFI_FORM_BROWSER2_PROTOCOL *FormBrowser2;\r | |
1624 | EFI_HII_CONFIG_ROUTING_PROTOCOL *HiiConfigRouting;\r | |
1625 | \r | |
1626 | //\r | |
1627 | // Locate protocols for use\r | |
1628 | //\r | |
57d40fe2 | 1629 | Status = gBS->LocateProtocol (&gEfiFormBrowser2ProtocolGuid, NULL, (VOID **) &FormBrowser2);\r |
c7f33ca4 | 1630 | if (EFI_ERROR (Status)) {\r |
1631 | return Status;\r | |
1632 | }\r | |
1633 | \r | |
57d40fe2 | 1634 | Status = gBS->LocateProtocol (&gEfiHiiConfigRoutingProtocolGuid, NULL, (VOID **) &HiiConfigRouting);\r |
c7f33ca4 | 1635 | if (EFI_ERROR (Status)) {\r |
1636 | return Status;\r | |
1637 | }\r | |
1638 | \r | |
1639 | //\r | |
1640 | // Retrive formset storage data from Form Browser\r | |
1641 | //\r | |
1642 | ConfigHdr = mFakeConfigHdr;\r | |
1643 | HeaderLen = EfiStrLen (ConfigHdr);\r | |
1644 | \r | |
1645 | BufferLen = 0x4000;\r | |
3e99020d | 1646 | ConfigResp = EfiLibAllocateZeroPool (BufferLen + (HeaderLen + 1) * sizeof (CHAR16));\r |
c7f33ca4 | 1647 | \r |
1648 | StringPtr = ConfigResp + HeaderLen;\r | |
1649 | *StringPtr = L'&';\r | |
1650 | StringPtr++;\r | |
1651 | \r | |
1652 | Status = FormBrowser2->BrowserCallback (\r | |
1653 | FormBrowser2,\r | |
1654 | &BufferLen,\r | |
1655 | StringPtr,\r | |
1656 | TRUE,\r | |
1657 | VariableGuid,\r | |
1658 | VariableName\r | |
1659 | );\r | |
1660 | if (Status == EFI_BUFFER_TOO_SMALL) {\r | |
1661 | gBS->FreePool (ConfigResp);\r | |
3e99020d | 1662 | ConfigResp = EfiLibAllocateZeroPool (BufferLen + (HeaderLen + 1) * sizeof (CHAR16));\r |
c7f33ca4 | 1663 | \r |
1664 | StringPtr = ConfigResp + HeaderLen;\r | |
1665 | *StringPtr = L'&';\r | |
1666 | StringPtr++;\r | |
1667 | \r | |
1668 | Status = FormBrowser2->BrowserCallback (\r | |
1669 | FormBrowser2,\r | |
1670 | &BufferLen,\r | |
1671 | StringPtr,\r | |
1672 | TRUE,\r | |
1673 | VariableGuid,\r | |
1674 | VariableName\r | |
1675 | );\r | |
1676 | }\r | |
1677 | if (EFI_ERROR (Status)) {\r | |
1678 | gBS->FreePool (ConfigResp);\r | |
1679 | return Status;\r | |
1680 | }\r | |
1681 | EfiCopyMem (ConfigResp, ConfigHdr, HeaderLen * sizeof (UINT16));\r | |
1682 | \r | |
1683 | //\r | |
1684 | // Convert <ConfigResp> to buffer data\r | |
1685 | //\r | |
1686 | Status = HiiConfigRouting->ConfigToBlock (\r | |
1687 | HiiConfigRouting,\r | |
1688 | ConfigResp,\r | |
1689 | Buffer,\r | |
1690 | BufferSize,\r | |
1691 | &Progress\r | |
1692 | );\r | |
1693 | gBS->FreePool (ConfigResp);\r | |
1694 | \r | |
1695 | return Status;\r | |
1696 | }\r | |
1697 | \r | |
1698 | EFI_STATUS\r | |
1699 | SetBrowserData (\r | |
1700 | EFI_GUID *VariableGuid, OPTIONAL\r | |
1701 | CHAR16 *VariableName, OPTIONAL\r | |
1702 | UINTN BufferSize,\r | |
1703 | UINT8 *Buffer,\r | |
1704 | CHAR16 *RequestElement OPTIONAL\r | |
1705 | )\r | |
1706 | /*++\r | |
1707 | \r | |
1708 | Routine Description:\r | |
1709 | This routine is invoked by ConfigAccess.Callback() to update uncommitted data of Form Browser.\r | |
1710 | \r | |
1711 | Arguments:\r | |
1712 | VariableGuid - An optional field to indicate the target variable GUID name to use.\r | |
1713 | VariableName - An optional field to indicate the target human-readable variable name.\r | |
1714 | BufferSize - Length in bytes of buffer to hold retrived data.\r | |
1715 | Buffer - Buffer to hold retrived data.\r | |
1716 | RequestElement - An optional field to specify which part of the buffer data\r | |
1717 | will be send back to Browser. If NULL, the whole buffer of\r | |
1718 | data will be committed to Browser.\r | |
1719 | <RequestElement> ::= &OFFSET=<Number>&WIDTH=<Number>*\r | |
1720 | \r | |
1721 | Returns:\r | |
1722 | EFI_SUCCESS - Routine success.\r | |
1723 | Other - Updating Browser uncommitted data failed.\r | |
1724 | \r | |
1725 | --*/\r | |
1726 | {\r | |
1727 | EFI_STATUS Status;\r | |
1728 | CHAR16 *ConfigHdr;\r | |
1729 | CHAR16 *ConfigResp;\r | |
1730 | CHAR16 *StringPtr;\r | |
1731 | UINTN HeaderLen;\r | |
1732 | UINTN BufferLen;\r | |
1733 | CHAR16 *Progress;\r | |
1734 | EFI_FORM_BROWSER2_PROTOCOL *FormBrowser2;\r | |
1735 | EFI_HII_CONFIG_ROUTING_PROTOCOL *HiiConfigRouting;\r | |
1736 | CHAR16 BlockName[33];\r | |
1737 | CHAR16 *ConfigRequest;\r | |
1738 | CHAR16 *Request;\r | |
1739 | \r | |
1740 | //\r | |
1741 | // Locate protocols for use\r | |
1742 | //\r | |
57d40fe2 | 1743 | Status = gBS->LocateProtocol (&gEfiFormBrowser2ProtocolGuid, NULL, (VOID **) &FormBrowser2);\r |
c7f33ca4 | 1744 | if (EFI_ERROR (Status)) {\r |
1745 | return Status;\r | |
1746 | }\r | |
1747 | \r | |
57d40fe2 | 1748 | Status = gBS->LocateProtocol (&gEfiHiiConfigRoutingProtocolGuid, NULL, (VOID **) &HiiConfigRouting);\r |
c7f33ca4 | 1749 | if (EFI_ERROR (Status)) {\r |
1750 | return Status;\r | |
1751 | }\r | |
1752 | \r | |
1753 | //\r | |
1754 | // Prepare <ConfigRequest>\r | |
1755 | //\r | |
1756 | ConfigHdr = mFakeConfigHdr;\r | |
1757 | HeaderLen = EfiStrLen (ConfigHdr);\r | |
1758 | \r | |
1759 | if (RequestElement == NULL) {\r | |
1760 | //\r | |
1761 | // RequestElement not specified, use "&OFFSET=0&WIDTH=<BufferSize>" as <BlockName>\r | |
1762 | //\r | |
1763 | BlockName[0] = L'\0';\r | |
1764 | EfiStrCpy (BlockName, L"&OFFSET=0&WIDTH=");\r | |
1765 | \r | |
1766 | //\r | |
1767 | // String lenghth of L"&OFFSET=0&WIDTH=" is 16\r | |
1768 | //\r | |
1769 | StringPtr = BlockName + 16;\r | |
1770 | BufferLen = sizeof (BlockName) - (16 * sizeof (CHAR16));\r | |
1771 | BufToHexString (StringPtr, &BufferLen, (UINT8 *) &BufferSize, sizeof (UINTN));\r | |
1772 | \r | |
1773 | Request = BlockName;\r | |
1774 | } else {\r | |
1775 | Request = RequestElement;\r | |
1776 | }\r | |
1777 | \r | |
1778 | BufferLen = HeaderLen * sizeof (CHAR16) + EfiStrSize (Request);\r | |
1779 | ConfigRequest = EfiLibAllocateZeroPool (BufferLen);\r | |
1780 | \r | |
1781 | EfiCopyMem (ConfigRequest, ConfigHdr, HeaderLen * sizeof (CHAR16));\r | |
1782 | StringPtr = ConfigRequest + HeaderLen;\r | |
1783 | EfiStrCpy (StringPtr, Request);\r | |
1784 | \r | |
1785 | //\r | |
1786 | // Convert buffer to <ConfigResp>\r | |
1787 | //\r | |
1788 | Status = HiiConfigRouting->BlockToConfig (\r | |
1789 | HiiConfigRouting,\r | |
1790 | ConfigRequest,\r | |
1791 | Buffer,\r | |
1792 | BufferSize,\r | |
1793 | &ConfigResp,\r | |
1794 | &Progress\r | |
1795 | );\r | |
1796 | if (EFI_ERROR (Status)) {\r | |
3e99020d | 1797 | gBS->FreePool (ConfigRequest);\r |
c7f33ca4 | 1798 | return Status;\r |
1799 | }\r | |
1800 | \r | |
1801 | //\r | |
1802 | // Skip <ConfigHdr> and '&'\r | |
1803 | //\r | |
1804 | StringPtr = ConfigResp + HeaderLen + 1;\r | |
1805 | \r | |
1806 | //\r | |
1807 | // Change uncommitted data in Browser\r | |
1808 | //\r | |
1809 | Status = FormBrowser2->BrowserCallback (\r | |
1810 | FormBrowser2,\r | |
1811 | &BufferSize,\r | |
1812 | StringPtr,\r | |
1813 | FALSE,\r | |
3e99020d LG |
1814 | VariableGuid,\r |
1815 | VariableName\r | |
c7f33ca4 | 1816 | );\r |
1817 | gBS->FreePool (ConfigResp);\r | |
3e99020d | 1818 | gBS->FreePool (ConfigRequest);\r |
c7f33ca4 | 1819 | return Status;\r |
1820 | }\r |