]> git.proxmox.com Git - mirror_edk2.git/blame - ShellPkg/Library/UefiShellDebug1CommandsLib/UefiShellDebug1CommandsLib.c
Clean up the private GUID definition in module Level.
[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / UefiShellDebug1CommandsLib.c
CommitLineData
5d73d92f 1/** @file\r
2 Main file for NULL named library for debug1 profile shell command functions.\r
3\r
3737ac2b 4 Copyright (c) 2010 - 2011, Intel Corporation. All rights reserved.<BR>\r
5d73d92f 5 This program and the accompanying materials\r
6 are licensed and made available under the terms and conditions of the BSD License\r
7 which accompanies this distribution. The full text of the license may be found at\r
8 http://opensource.org/licenses/bsd-license.php\r
9\r
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
12\r
13**/\r
14\r
15#include "UefiShellDebug1CommandsLib.h"\r
16\r
17STATIC CONST CHAR16 mFileName[] = L"Debug1Commands";\r
18EFI_HANDLE gShellDebug1HiiHandle = NULL;\r
5d73d92f 19\r
3737ac2b 20/**\r
21 Gets the debug file name. This will be used if HII is not working.\r
22\r
23 @retval NULL No file is available.\r
24 @return The NULL-terminated filename to get help from.\r
25**/\r
5d73d92f 26CONST CHAR16*\r
27EFIAPI\r
28ShellCommandGetManFileNameDebug1 (\r
29 VOID\r
30 )\r
31{\r
32 return (mFileName);\r
33}\r
34\r
35/**\r
36 Constructor for the Shell Debug1 Commands library.\r
37\r
38 @param ImageHandle the image handle of the process\r
39 @param SystemTable the EFI System Table pointer\r
40\r
41 @retval EFI_SUCCESS the shell command handlers were installed sucessfully\r
42 @retval EFI_UNSUPPORTED the shell level required was not found.\r
43**/\r
44EFI_STATUS\r
45EFIAPI\r
46UefiShellDebug1CommandsLibConstructor (\r
47 IN EFI_HANDLE ImageHandle,\r
48 IN EFI_SYSTEM_TABLE *SystemTable\r
49 )\r
50{\r
51 //\r
52 // check our bit of the profiles mask\r
53 //\r
54 if ((PcdGet8(PcdShellProfileMask) & BIT1) == 0) {\r
82571fb5 55 return (EFI_SUCCESS);\r
5d73d92f 56 }\r
57\r
58 //\r
59 // install the HII stuff.\r
60 //\r
61 gShellDebug1HiiHandle = HiiAddPackages (&gShellDebug1HiiGuid, gImageHandle, UefiShellDebug1CommandsLibStrings, NULL);\r
62 if (gShellDebug1HiiHandle == NULL) {\r
63 return (EFI_DEVICE_ERROR);\r
64 }\r
65\r
66 //\r
67 // install our shell command handlers that are always installed\r
68 //\r
3737ac2b 69 ShellCommandRegisterCommandName(L"setsize", ShellCommandRunSetSize , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_SETSIZE) );\r
5d73d92f 70 ShellCommandRegisterCommandName(L"comp", ShellCommandRunComp , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_COMP) );\r
71 ShellCommandRegisterCommandName(L"mode", ShellCommandRunMode , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_MODE) );\r
72 ShellCommandRegisterCommandName(L"memmap", ShellCommandRunMemMap , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_MEMMAP) );\r
73 ShellCommandRegisterCommandName(L"eficompress", ShellCommandRunEfiCompress , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_EFICOMPRESS) );\r
74 ShellCommandRegisterCommandName(L"efidecompress", ShellCommandRunEfiDecompress , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_EFIDCOMPRESS) );\r
75 ShellCommandRegisterCommandName(L"dmem", ShellCommandRunDmem , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_DMEM) );\r
3737ac2b 76 ShellCommandRegisterCommandName(L"loadpcirom", ShellCommandRunLoadPciRom , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_LOAD_PCI_ROM) );\r
5d73d92f 77 ShellCommandRegisterCommandName(L"mm", ShellCommandRunMm , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_MM) );\r
3737ac2b 78 ShellCommandRegisterCommandName(L"setvar", ShellCommandRunSetVar , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_SETVAR) );\r
79 ShellCommandRegisterCommandName(L"sermode", ShellCommandRunSerMode , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_SERMODE) );\r
80 ShellCommandRegisterCommandName(L"pci", ShellCommandRunPci , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_PCI) );\r
5d73d92f 81 ShellCommandRegisterCommandName(L"smbiosview", ShellCommandRunSmbiosView , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_SMBIOSVIEW) );\r
82 ShellCommandRegisterCommandName(L"dmpstore", ShellCommandRunDmpStore , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_DMPSTORE) );\r
83 ShellCommandRegisterCommandName(L"dblk", ShellCommandRunDblk , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_DBLK) );\r
3737ac2b 84 ShellCommandRegisterCommandName(L"edit", ShellCommandRunEdit , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_EDIT) );\r
85 ShellCommandRegisterCommandName(L"hexedit", ShellCommandRunHexEdit , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_HEXEDIT) );\r
5d73d92f 86\r
87 //\r
88 // check install profile bit of the profiles mask is set\r
89 //\r
90 if ((PcdGet8(PcdShellProfileMask) & BIT2) == 0) {\r
91 ShellCommandRegisterCommandName(L"bcfg", ShellCommandRunBcfg , ShellCommandGetManFileNameDebug1, 0, L"Debug1", TRUE, gShellDebug1HiiHandle, STRING_TOKEN(STR_GET_HELP_BCFG) );\r
92 }\r
93\r
3737ac2b 94\r
95\r
5d73d92f 96\r
97 ShellCommandRegisterAlias(L"dmem", L"mem");\r
98\r
99 return (EFI_SUCCESS);\r
100}\r
101\r
102/**\r
103 Destructor for the library. free any resources.\r
3737ac2b 104\r
105 @param ImageHandle The image handle of the process.\r
106 @param SystemTable The EFI System Table pointer.\r
5d73d92f 107**/\r
108EFI_STATUS\r
109EFIAPI\r
110UefiShellDebug1CommandsLibDestructor (\r
111 IN EFI_HANDLE ImageHandle,\r
112 IN EFI_SYSTEM_TABLE *SystemTable\r
113 )\r
114{\r
115 if (gShellDebug1HiiHandle != NULL) {\r
116 HiiRemovePackages(gShellDebug1HiiHandle);\r
117 }\r
118 return (EFI_SUCCESS);\r
119}\r
120\r
121STATIC CONST CHAR8 Hex[] = {\r
122 '0',\r
123 '1',\r
124 '2',\r
125 '3',\r
126 '4',\r
127 '5',\r
128 '6',\r
129 '7',\r
130 '8',\r
131 '9',\r
132 'A',\r
133 'B',\r
134 'C',\r
135 'D',\r
136 'E',\r
137 'F'\r
138};\r
139\r
3737ac2b 140/**\r
141 Dump some hexadecimal data to the screen.\r
142\r
143 @param[in] Indent How many spaces to indent the output.\r
144 @param[in] Offset The offset of the printing.\r
145 @param[in] DataSize The size in bytes of UserData.\r
146 @param[in] UserData The data to print out.\r
147**/\r
5d73d92f 148VOID\r
5d73d92f 149DumpHex (\r
150 IN UINTN Indent,\r
151 IN UINTN Offset,\r
152 IN UINTN DataSize,\r
153 IN VOID *UserData\r
154 )\r
155{\r
156 UINT8 *Data;\r
157\r
158 CHAR8 Val[50];\r
159\r
160 CHAR8 Str[20];\r
161\r
3737ac2b 162 UINT8 TempByte;\r
5d73d92f 163 UINTN Size;\r
164 UINTN Index;\r
165\r
5d73d92f 166 Data = UserData;\r
167 while (DataSize != 0) {\r
168 Size = 16;\r
169 if (Size > DataSize) {\r
170 Size = DataSize;\r
171 }\r
172\r
173 for (Index = 0; Index < Size; Index += 1) {\r
3737ac2b 174 TempByte = Data[Index];\r
175 Val[Index * 3 + 0] = Hex[TempByte >> 4];\r
176 Val[Index * 3 + 1] = Hex[TempByte & 0xF];\r
5d73d92f 177 Val[Index * 3 + 2] = (CHAR8) ((Index == 7) ? '-' : ' ');\r
3737ac2b 178 Str[Index] = (CHAR8) ((TempByte < ' ' || TempByte > 'z') ? '.' : TempByte);\r
5d73d92f 179 }\r
180\r
181 Val[Index * 3] = 0;\r
182 Str[Index] = 0;\r
183 ShellPrintEx(-1, -1, L"%*a%02X: %-.48a *%a*\r\n", Indent, "", Offset, Val, Str);\r
184\r
185 Data += Size;\r
186 Offset += Size;\r
187 DataSize -= Size;\r
188 }\r
189}\r
190\r
191/**\r
192 Convert a Unicode character to upper case only if\r
193 it maps to a valid small-case ASCII character.\r
194\r
195 This internal function only deal with Unicode character\r
196 which maps to a valid small-case ASCII character, i.e.\r
197 L'a' to L'z'. For other Unicode character, the input character\r
198 is returned directly.\r
199\r
200 @param Char The character to convert.\r
201\r
202 @retval LowerCharacter If the Char is with range L'a' to L'z'.\r
203 @retval Unchanged Otherwise.\r
204\r
205\r
206 //Stolen from MdePkg Baselib\r
207**/\r
208CHAR16\r
209EFIAPI\r
210CharToUpper (\r
211 IN CHAR16 Char\r
212 )\r
213{\r
214 if (Char >= L'a' && Char <= L'z') {\r
215 return (CHAR16) (Char - (L'a' - L'A'));\r
216 }\r
217\r
218 return Char;\r
219}\r
220\r
221/**\r
222 Function returns a system configuration table that is stored in the\r
223 EFI System Table based on the provided GUID.\r
224\r
4ff7e37b
ED
225 @param[in] TableGuid A pointer to the table's GUID type.\r
226 @param[in, out] Table On exit, a pointer to a system configuration table.\r
5d73d92f 227\r
228 @retval EFI_SUCCESS A configuration table matching TableGuid was found.\r
229 @retval EFI_NOT_FOUND A configuration table matching TableGuid was not found.\r
230**/\r
231EFI_STATUS\r
232EFIAPI\r
233GetSystemConfigurationTable (\r
234 IN EFI_GUID *TableGuid,\r
235 IN OUT VOID **Table\r
236 )\r
237{\r
238 UINTN Index;\r
239 ASSERT (Table != NULL);\r
240\r
241 for (Index = 0; Index < gST->NumberOfTableEntries; Index++) {\r
3737ac2b 242 if (CompareGuid (TableGuid, &(gST->ConfigurationTable[Index].VendorGuid))) {\r
5d73d92f 243 *Table = gST->ConfigurationTable[Index].VendorTable;\r
244 return EFI_SUCCESS;\r
245 }\r
246 }\r
247\r
248 return EFI_NOT_FOUND;\r
249}\r
250\r
251/**\r
252 Convert a Unicode character to numerical value.\r
253\r
254 This internal function only deal with Unicode character\r
255 which maps to a valid hexadecimal ASII character, i.e.\r
256 L'0' to L'9', L'a' to L'f' or L'A' to L'F'. For other\r
257 Unicode character, the value returned does not make sense.\r
258\r
259 @param Char The character to convert.\r
260\r
261 @return The numerical value converted.\r
262\r
263**/\r
264UINTN\r
265EFIAPI\r
266HexCharToUintn (\r
267 IN CHAR16 Char\r
268 )\r
269{\r
270 if (Char >= L'0' && Char <= L'9') {\r
271 return Char - L'0';\r
272 }\r
273\r
274 return (UINTN) (10 + CharToUpper (Char) - L'A');\r
275}\r
276\r
3737ac2b 277/**\r
278 Convert a string representation of a guid to a Guid value.\r
279\r
4ff7e37b
ED
280 @param[in] StringGuid The pointer to the string of a guid.\r
281 @param[in, out] Guid The pointer to the GUID structure to populate.\r
3737ac2b 282\r
283 @retval EFI_INVALID_PARAMETER A parameter was invalid.\r
284 @retval EFI_SUCCESS The conversion was successful.\r
285**/\r
5d73d92f 286EFI_STATUS\r
287EFIAPI\r
288ConvertStringToGuid (\r
289 IN CONST CHAR16 *StringGuid,\r
290 IN OUT EFI_GUID *Guid\r
291 )\r
292{\r
3737ac2b 293 CHAR16 *TempCopy;\r
294 CHAR16 *TempSpot;\r
295 CHAR16 *Walker;\r
296 UINT64 TempVal;\r
297 EFI_STATUS Status;\r
298\r
299 if (StringGuid == NULL) {\r
300 return (EFI_INVALID_PARAMETER);\r
301 } else if (StrLen(StringGuid) != 36) {\r
5d73d92f 302 return (EFI_INVALID_PARAMETER);\r
3737ac2b 303 } \r
304 TempCopy = NULL;\r
305 TempCopy = StrnCatGrow(&TempCopy, NULL, StringGuid, 0);\r
306 Walker = TempCopy;\r
307 TempSpot = StrStr(Walker, L"-");\r
33c031ee 308 if (TempSpot != NULL) {\r
309 *TempSpot = CHAR_NULL;\r
310 }\r
3737ac2b 311 Status = ShellConvertStringToUint64(Walker, &TempVal, TRUE, FALSE);\r
312 if (EFI_ERROR(Status)) {\r
313 FreePool(TempCopy);\r
314 return (Status);\r
315 }\r
316 Guid->Data1 = (UINT32)TempVal;\r
317 Walker += 9;\r
318 TempSpot = StrStr(Walker, L"-");\r
33c031ee 319 if (TempSpot != NULL) {\r
320 *TempSpot = CHAR_NULL;\r
321 }\r
3737ac2b 322 Status = ShellConvertStringToUint64(Walker, &TempVal, TRUE, FALSE);\r
323 if (EFI_ERROR(Status)) {\r
324 FreePool(TempCopy);\r
325 return (Status);\r
326 }\r
327 Guid->Data2 = (UINT16)TempVal;\r
328 Walker += 5;\r
329 TempSpot = StrStr(Walker, L"-");\r
33c031ee 330 if (TempSpot != NULL) {\r
331 *TempSpot = CHAR_NULL;\r
332 }\r
3737ac2b 333 Status = ShellConvertStringToUint64(Walker, &TempVal, TRUE, FALSE);\r
334 if (EFI_ERROR(Status)) {\r
335 FreePool(TempCopy);\r
336 return (Status);\r
337 }\r
338 Guid->Data3 = (UINT16)TempVal;\r
339 Walker += 5;\r
340 Guid->Data4[0] = (UINT8)(HexCharToUintn(Walker[0]) * 16);\r
341 Guid->Data4[0] = (UINT8)(Guid->Data4[0]+ (UINT8)HexCharToUintn(Walker[1]));\r
342 Walker += 2;\r
343 Guid->Data4[1] = (UINT8)(HexCharToUintn(Walker[0]) * 16);\r
344 Guid->Data4[1] = (UINT8)(Guid->Data4[1] + (UINT8)HexCharToUintn(Walker[1]));\r
345 Walker += 3;\r
346 Guid->Data4[2] = (UINT8)(HexCharToUintn(Walker[0]) * 16);\r
347 Guid->Data4[2] = (UINT8)(Guid->Data4[2] + (UINT8)HexCharToUintn(Walker[1]));\r
348 Walker += 2;\r
349 Guid->Data4[3] = (UINT8)(HexCharToUintn(Walker[0]) * 16);\r
350 Guid->Data4[3] = (UINT8)(Guid->Data4[3] + (UINT8)HexCharToUintn(Walker[1]));\r
351 Walker += 2;\r
352 Guid->Data4[4] = (UINT8)(HexCharToUintn(Walker[0]) * 16);\r
353 Guid->Data4[4] = (UINT8)(Guid->Data4[4] + (UINT8)HexCharToUintn(Walker[1]));\r
354 Walker += 2;\r
355 Guid->Data4[5] = (UINT8)(HexCharToUintn(Walker[0]) * 16);\r
356 Guid->Data4[5] = (UINT8)(Guid->Data4[5] + (UINT8)HexCharToUintn(Walker[1]));\r
357 Walker += 2;\r
358 Guid->Data4[6] = (UINT8)(HexCharToUintn(Walker[0]) * 16);\r
359 Guid->Data4[6] = (UINT8)(Guid->Data4[6] + (UINT8)HexCharToUintn(Walker[1]));\r
360 Walker += 2;\r
361 Guid->Data4[7] = (UINT8)(HexCharToUintn(Walker[0]) * 16);\r
362 Guid->Data4[7] = (UINT8)(Guid->Data4[7] + (UINT8)HexCharToUintn(Walker[1]));\r
363 FreePool(TempCopy);\r
364 return (EFI_SUCCESS);\r
365}\r
366\r
3737ac2b 367/**\r
368 Clear the line at the specified Row.\r
369 \r
370 @param[in] Row The row number to be cleared ( start from 1 )\r
371 @param[in] LastCol The last printable column.\r
372 @param[in] LastRow The last printable row.\r
373**/\r
374VOID\r
375EFIAPI\r
376EditorClearLine (\r
377 IN UINTN Row,\r
378 IN UINTN LastCol,\r
379 IN UINTN LastRow\r
380 )\r
381{\r
382 CHAR16 Line[200];\r
383\r
384 if (Row == 0) {\r
385 Row = 1;\r
386 }\r
387\r
388 //\r
389 // prepare a blank line\r
390 //\r
391 SetMem16(Line, LastCol*sizeof(CHAR16), L' ');\r
392\r
393 if (Row == LastRow) {\r
394 //\r
395 // if CHAR_NULL is still at position 80, it will cause first line error\r
396 //\r
397 Line[LastCol - 1] = CHAR_NULL;\r
5d73d92f 398 } else {\r
3737ac2b 399 Line[LastCol] = CHAR_NULL;\r
400 }\r
401\r
402 //\r
403 // print out the blank line\r
404 //\r
405 ShellPrintEx (0, ((INT32)Row) - 1, Line);\r
406}\r
407\r
408/**\r
409 Determine if the character is valid for a filename.\r
410\r
411 @param[in] Ch The character to test.\r
412\r
413 @retval TRUE The character is valid.\r
414 @retval FALSE The character is not valid.\r
415**/\r
416BOOLEAN\r
417EFIAPI\r
418IsValidFileNameChar (\r
419 IN CONST CHAR16 Ch\r
420 )\r
421{\r
422 //\r
423 // See if there are any illegal characters within the name\r
424 //\r
425 if (Ch < 0x20 || Ch == L'\"' || Ch == L'*' || Ch == L'/' || Ch == L'<' || Ch == L'>' || Ch == L'?' || Ch == L'|' || Ch == L' ') {\r
426 return FALSE;\r
427 }\r
428\r
429 return TRUE;\r
430}\r
431\r
432/**\r
433 Check if file name has illegal characters.\r
434 \r
435 @param Name The filename to check.\r
436\r
437 @retval TRUE The filename is ok.\r
438 @retval FALSE The filename is not ok.\r
439**/\r
440BOOLEAN\r
441EFIAPI\r
442IsValidFileName (\r
443 IN CONST CHAR16 *Name\r
444 )\r
445{\r
446\r
447 UINTN Index;\r
448 UINTN Len;\r
449\r
450 //\r
451 // check the length of Name\r
452 //\r
453 for (Len = 0, Index = StrLen (Name) - 1; Index + 1 != 0; Index--, Len++) {\r
454 if (Name[Index] == '\\' || Name[Index] == ':') {\r
455 break;\r
456 }\r
457 }\r
458\r
459 if (Len == 0 || Len > 255) {\r
460 return FALSE;\r
461 }\r
462 //\r
463 // check whether any char in Name not appears in valid file name char\r
464 //\r
465 for (Index = 0; Index < StrLen (Name); Index++) {\r
466 if (!IsValidFileNameChar (Name[Index])) {\r
467 return FALSE;\r
468 }\r
5d73d92f 469 }\r
3737ac2b 470\r
471 return TRUE;\r
472}\r
473\r
474/**\r
475 Find a filename that is valid (not taken) with the given extension.\r
476\r
477 @param[in] Extension The file extension.\r
478\r
479 @retval NULL Something went wrong.\r
480 @return the valid filename.\r
481**/\r
482CHAR16 *\r
483EFIAPI\r
484EditGetDefaultFileName (\r
485 IN CONST CHAR16 *Extension\r
486 )\r
487{\r
488 EFI_STATUS Status;\r
489 UINTN Suffix;\r
490 BOOLEAN FoundNewFile;\r
491 CHAR16 *FileNameTmp;\r
492\r
493 Suffix = 0;\r
494 FoundNewFile = FALSE;\r
495\r
496 do {\r
497 FileNameTmp = CatSPrint (NULL, L"NewFile%d.%s", Suffix, Extension);\r
498\r
499 //\r
500 // after that filename changed to path\r
501 //\r
502 Status = ShellFileExists (FileNameTmp);\r
503\r
504 if (Status == EFI_NOT_FOUND) {\r
505 return FileNameTmp;\r
506 }\r
507\r
508 FreePool (FileNameTmp);\r
509 FileNameTmp = NULL;\r
510 Suffix++;\r
511 } while (Suffix != 0);\r
512\r
513 FreePool (FileNameTmp);\r
514 return NULL;\r
5d73d92f 515}\r
516\r
3737ac2b 517/**\r
518 Read a file into an allocated buffer. The buffer is the responsibility \r
519 of the caller to free.\r
520\r
521 @param[in] FileName The filename of the file to open.\r
522 @param[out] Buffer Upon successful return, the pointer to the \r
523 address of the allocated buffer. \r
524 @param[out] BufferSize If not NULL, then the pointer to the size\r
525 of the allocated buffer.\r
526 @param[out] ReadOnly Upon successful return TRUE if the file is\r
527 read only. FALSE otherwise.\r
528\r
529 @retval EFI_NOT_FOUND The filename did not represent a file in the \r
530 file system.\r
531 @retval EFI_SUCCESS The file was read into the buffer.\r
532 @retval EFI_OUT_OF_RESOURCES A memory allocation failed.\r
533 @retval EFI_LOAD_ERROR The file read operation failed.\r
534 @retval EFI_INVALID_PARAMETER A parameter was invalid.\r
535 @retval EFI_INVALID_PARAMETER FileName was NULL.\r
536 @retval EFI_INVALID_PARAMETER FileName was a directory.\r
537**/\r
538EFI_STATUS\r
539EFIAPI\r
540ReadFileIntoBuffer (\r
541 IN CONST CHAR16 *FileName,\r
542 OUT VOID **Buffer,\r
543 OUT UINTN *BufferSize OPTIONAL,\r
544 OUT BOOLEAN *ReadOnly\r
545 )\r
546{\r
547 VOID *InternalBuffer;\r
548 UINTN FileSize;\r
549 SHELL_FILE_HANDLE FileHandle;\r
550 BOOLEAN CreateFile;\r
551 EFI_STATUS Status;\r
552 EFI_FILE_INFO *Info;\r
553\r
554 InternalBuffer = NULL;\r
555 FileSize = 0;\r
556 FileHandle = NULL;\r
557 CreateFile = FALSE;\r
558 Status = EFI_SUCCESS;\r
559 Info = NULL;\r
560\r
561 if (FileName == NULL || Buffer == NULL || ReadOnly == NULL) {\r
562 return (EFI_INVALID_PARAMETER);\r
563 }\r
564\r
565 //\r
566 // try to open the file\r
567 //\r
568 Status = ShellOpenFileByName (FileName, &FileHandle, EFI_FILE_MODE_READ, 0);\r
569\r
570 if (!EFI_ERROR(Status)) {\r
571 ASSERT(CreateFile == FALSE);\r
572 if (FileHandle == NULL) {\r
573 return EFI_LOAD_ERROR;\r
574 }\r
575\r
576 Info = ShellGetFileInfo(FileHandle);\r
577 \r
578 if (Info->Attribute & EFI_FILE_DIRECTORY) {\r
579 FreePool (Info);\r
580 return EFI_INVALID_PARAMETER;\r
581 }\r
582\r
583 if (Info->Attribute & EFI_FILE_READ_ONLY) {\r
584 *ReadOnly = TRUE;\r
585 } else {\r
586 *ReadOnly = FALSE;\r
587 }\r
588 //\r
589 // get file size\r
590 //\r
591 FileSize = (UINTN) Info->FileSize;\r
592\r
593 FreePool (Info);\r
594 } else if (Status == EFI_NOT_FOUND) {\r
595 //\r
596 // file not exists. add create and try again\r
597 //\r
598 Status = ShellOpenFileByName (FileName, &FileHandle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE|EFI_FILE_MODE_CREATE, 0);\r
599 if (EFI_ERROR (Status)) {\r
600 return Status;\r
601 } else {\r
602 //\r
603 // it worked. now delete it and move on with the name (now validated)\r
604 //\r
605 Status = ShellDeleteFile (&FileHandle);\r
606 if (Status == EFI_WARN_DELETE_FAILURE) {\r
607 Status = EFI_ACCESS_DENIED;\r
608 }\r
609 if (EFI_ERROR (Status)) {\r
610 return Status;\r
611 }\r
612 }\r
613 //\r
614 // file doesn't exist, so set CreateFile to TRUE and can't be read-only\r
615 //\r
616 CreateFile = TRUE;\r
617 *ReadOnly = FALSE;\r
618 }\r
619\r
620 //\r
621 // the file exists\r
622 //\r
623 if (!CreateFile) {\r
624 //\r
625 // allocate buffer to read file\r
626 //\r
627 InternalBuffer = AllocateZeroPool (FileSize);\r
628 if (InternalBuffer == NULL) {\r
629 return EFI_OUT_OF_RESOURCES;\r
630 }\r
631 //\r
632 // read file into InternalBuffer\r
633 //\r
634 Status = ShellReadFile (FileHandle, &FileSize, InternalBuffer);\r
635 ShellCloseFile(&FileHandle);\r
636 FileHandle = NULL;\r
637 if (EFI_ERROR (Status)) {\r
638 SHELL_FREE_NON_NULL (InternalBuffer);\r
639 return EFI_LOAD_ERROR;\r
640 }\r
641 }\r
642 *Buffer = InternalBuffer;\r
643 if (BufferSize != NULL) {\r
644 *BufferSize = FileSize;\r
645 }\r
646 return (EFI_SUCCESS);\r
647\r
648}\r