]> git.proxmox.com Git - mirror_edk2.git/blame - ShellPkg/Library/UefiShellDebug1CommandsLib/HexEdit/HexEdit.c
ShellPkg: Fixed build error 'variable set but not used'
[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / HexEdit / HexEdit.c
CommitLineData
632820d1 1/** @file\r
2 Main entry point of editor\r
3 \r
4 Copyright (c) 2005 - 2011, Intel Corporation. All rights reserved. <BR>\r
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#include "HexEditor.h"\r
17\r
18//\r
19// Global Variables\r
20//\r
21STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
22 {L"-f", TypeFlag},\r
23 {L"-d", TypeFlag},\r
24 {L"-m", TypeFlag},\r
25 {NULL, TypeMax}\r
26 };\r
27\r
28/**\r
29 Function for 'hexedit' command.\r
30\r
31 @param[in] ImageHandle Handle to the Image (NULL if Internal).\r
32 @param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
33**/\r
34SHELL_STATUS\r
35EFIAPI\r
36ShellCommandRunHexEdit (\r
37 IN EFI_HANDLE ImageHandle,\r
38 IN EFI_SYSTEM_TABLE *SystemTable\r
39 )\r
40{\r
41 EFI_STATUS Status;\r
42 CHAR16 *Buffer;\r
43 CHAR16 *ProblemParam;\r
44 SHELL_STATUS ShellStatus;\r
45 LIST_ENTRY *Package;\r
980d554e 46 CHAR16 *NewName;\r
632820d1 47 CONST CHAR16 *Name;\r
48 UINTN Offset;\r
49 UINTN Size;\r
632820d1 50 EDIT_FILE_TYPE WhatToDo;\r
51\r
52 Buffer = NULL;\r
53 ShellStatus = SHELL_SUCCESS;\r
980d554e 54 NewName = NULL;\r
632820d1 55 Buffer = NULL;\r
56 Name = NULL;\r
632820d1 57 Offset = 0;\r
58 Size = 0;\r
632820d1 59 WhatToDo = FileTypeNone;\r
60\r
61 //\r
62 // initialize the shell lib (we must be in non-auto-init...)\r
63 //\r
64 Status = ShellInitialize();\r
65 ASSERT_EFI_ERROR(Status);\r
66\r
67 Status = CommandInit();\r
68 ASSERT_EFI_ERROR(Status);\r
69\r
70 //\r
71 // parse the command line\r
72 //\r
73 Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
74 if (EFI_ERROR(Status)) {\r
75 if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
76 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
77 FreePool(ProblemParam);\r
78 ShellStatus = SHELL_INVALID_PARAMETER;\r
79 } else {\r
80 ASSERT(FALSE);\r
81 }\r
82 } else {\r
83 //\r
84 // Check for -d\r
85 //\r
86 if (ShellCommandLineGetFlag(Package, L"-d")){\r
392fa5a6 87 if (ShellCommandLineGetCount(Package) < 4) {\r
88 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
89 ShellStatus = SHELL_INVALID_PARAMETER;\r
90 } else if (ShellCommandLineGetCount(Package) > 4) {\r
632820d1 91 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
92 ShellStatus = SHELL_INVALID_PARAMETER;\r
93 } else {\r
94 WhatToDo = FileTypeDiskBuffer;\r
95 Name = ShellCommandLineGetRawValue(Package, 1);\r
96 Offset = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 2));\r
97 Size = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 3));\r
98 }\r
392fa5a6 99 if (Offset == (UINTN)-1 || Size == (UINTN)-1) {\r
100 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM_VAL), gShellDebug1HiiHandle, L"-d");\r
101 ShellStatus = SHELL_INVALID_PARAMETER;\r
102 }\r
632820d1 103 }\r
104\r
105 //\r
106 // check for -f\r
107 //\r
108 if (ShellCommandLineGetFlag(Package, L"-f") && (WhatToDo == FileTypeNone)){\r
392fa5a6 109 if (ShellCommandLineGetCount(Package) < 2) {\r
110 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
111 ShellStatus = SHELL_INVALID_PARAMETER;\r
112 } else if (ShellCommandLineGetCount(Package) > 2) {\r
632820d1 113 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
114 ShellStatus = SHELL_INVALID_PARAMETER;\r
115 } else {\r
116 Name = ShellCommandLineGetRawValue(Package, 1);\r
33c031ee 117 if (Name == NULL || !IsValidFileName(Name)) {\r
632820d1 118 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Name);\r
119 ShellStatus = SHELL_INVALID_PARAMETER;\r
120 } else {\r
121 WhatToDo = FileTypeFileBuffer;\r
122 }\r
123 }\r
124 }\r
125\r
126 //\r
127 // check for -m\r
128 //\r
129 if (ShellCommandLineGetFlag(Package, L"-m") && (WhatToDo == FileTypeNone)){\r
130 if (ShellCommandLineGetCount(Package) < 3) {\r
131 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
132 ShellStatus = SHELL_INVALID_PARAMETER;\r
133 } else if (ShellCommandLineGetCount(Package) > 3) {\r
134 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
135 ShellStatus = SHELL_INVALID_PARAMETER;\r
136 } else {\r
137 WhatToDo = FileTypeMemBuffer;\r
138 Offset = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 1));\r
139 Size = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 2));\r
140 }\r
141 }\r
ae724571 142 Name = ShellCommandLineGetRawValue(Package, 1);\r
143 if (WhatToDo == FileTypeNone && Name != NULL) {\r
411a3c39 144 if (ShellCommandLineGetCount(Package) > 2) {\r
145 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
146 ShellStatus = SHELL_INVALID_PARAMETER;\r
147 } else if (!IsValidFileName(Name)) {\r
148 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Name);\r
149 ShellStatus = SHELL_INVALID_PARAMETER;\r
150 } else {\r
151 WhatToDo = FileTypeFileBuffer;\r
152 }\r
980d554e 153 } else if (WhatToDo == FileTypeNone) {\r
154 if (gEfiShellProtocol->GetCurDir(NULL) == NULL) {\r
155 ShellStatus = SHELL_NOT_FOUND;\r
156 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellDebug1HiiHandle);\r
157 } else {\r
158 NewName = EditGetDefaultFileName(L"bin");\r
159 Name = NewName;\r
160 WhatToDo = FileTypeFileBuffer;\r
161 }\r
162 }\r
632820d1 163\r
164 if (ShellStatus == SHELL_SUCCESS && WhatToDo == FileTypeNone) {\r
165 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
166 ShellStatus = SHELL_INVALID_PARAMETER;\r
392fa5a6 167 } else if (WhatToDo == FileTypeFileBuffer && ShellGetCurrentDir(NULL) == NULL) {\r
168 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellDebug1HiiHandle);\r
169 ShellStatus = SHELL_INVALID_PARAMETER;\r
e0c2cc6f 170 }\r
632820d1 171\r
172 if (ShellStatus == SHELL_SUCCESS) {\r
173 //\r
174 // Do the editor\r
175 //\r
176 Status = HMainEditorInit ();\r
177 if (EFI_ERROR (Status)) {\r
178 gST->ConOut->ClearScreen (gST->ConOut);\r
179 gST->ConOut->EnableCursor (gST->ConOut, TRUE);\r
180 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_INIT_FAILED), gShellDebug1HiiHandle);\r
181 } else {\r
182 HMainEditorBackup ();\r
183 switch (WhatToDo) {\r
184 case FileTypeFileBuffer:\r
185 Status = HBufferImageRead (\r
4d10ab79 186 Name==NULL?L"":Name,\r
632820d1 187 NULL,\r
188 0,\r
189 0,\r
190 0,\r
191 0,\r
192 FileTypeFileBuffer,\r
193 FALSE\r
194 );\r
195 break;\r
196\r
197 case FileTypeDiskBuffer:\r
198 Status = HBufferImageRead (\r
199 NULL,\r
4d10ab79 200 Name==NULL?L"":Name,\r
632820d1 201 Offset,\r
202 Size,\r
203 0,\r
204 0,\r
205 FileTypeDiskBuffer,\r
206 FALSE\r
207 );\r
208 break;\r
209\r
210 case FileTypeMemBuffer:\r
211 Status = HBufferImageRead (\r
212 NULL,\r
213 NULL,\r
214 0,\r
215 0,\r
216 (UINT32) Offset,\r
217 Size,\r
218 FileTypeMemBuffer,\r
219 FALSE\r
220 );\r
221 break;\r
222\r
e0c2cc6f 223 default:\r
224 Status = EFI_NOT_FOUND;\r
225 break;\r
632820d1 226 }\r
227 if (!EFI_ERROR (Status)) {\r
228 HMainEditorRefresh ();\r
229 Status = HMainEditorKeyInput ();\r
230 }\r
231 if (Status != EFI_OUT_OF_RESOURCES) {\r
232 //\r
233 // back up the status string\r
234 //\r
411a3c39 235 Buffer = CatSPrint (NULL, L"%s\r\n", StatusBarGetString());\r
632820d1 236 }\r
237 }\r
238\r
239 //\r
240 // cleanup\r
241 //\r
242 HMainEditorCleanup ();\r
243\r
244 if (!EFI_ERROR (Status)) {\r
245 if (ShellStatus == SHELL_SUCCESS) {\r
246 ShellStatus = SHELL_UNSUPPORTED;\r
247 }\r
248 }\r
249\r
250 //\r
251 // print editor exit code on screen\r
252 //\r
253 if (Status == EFI_OUT_OF_RESOURCES) {\r
254 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle);\r
255 } else if (EFI_ERROR(Status)){\r
256 if (Buffer != NULL) {\r
257 if (StrCmp (Buffer, L"") != 0) {\r
258 //\r
259 // print out the status string\r
260 //\r
411a3c39 261 ShellPrintEx(-1, -1, L"%s", Buffer);\r
632820d1 262 } else {\r
263 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_UNKNOWN_EDITOR), gShellDebug1HiiHandle);\r
264 }\r
265 } else {\r
266 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_UNKNOWN_EDITOR), gShellDebug1HiiHandle);\r
267 }\r
268 }\r
269 }\r
980d554e 270 ShellCommandLineFreeVarList (Package);\r
632820d1 271 }\r
272\r
273 SHELL_FREE_NON_NULL (Buffer);\r
980d554e 274 SHELL_FREE_NON_NULL (NewName);\r
632820d1 275 return ShellStatus;\r
276}\r