]> git.proxmox.com Git - mirror_edk2.git/blame - ShellPkg/Library/UefiShellDebug1CommandsLib/HexEdit/HexEdit.c
Set EXECUTE flags in so RELEASE builds can tun correctly.
[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
46 CONST CHAR16 *Cwd;\r
980d554e 47 CHAR16 *NewName;\r
632820d1 48 CHAR16 *Spot;\r
49 CONST CHAR16 *Name;\r
50 UINTN Offset;\r
51 UINTN Size;\r
52 UINT64 LastOffset;\r
53 EDIT_FILE_TYPE WhatToDo;\r
54\r
55 Buffer = NULL;\r
56 ShellStatus = SHELL_SUCCESS;\r
980d554e 57 NewName = NULL;\r
632820d1 58 Cwd = NULL;\r
59 Buffer = NULL;\r
60 Name = NULL;\r
61 Spot = NULL;\r
62 Offset = 0;\r
63 Size = 0;\r
64 LastOffset = 0;\r
65 WhatToDo = FileTypeNone;\r
66\r
67 //\r
68 // initialize the shell lib (we must be in non-auto-init...)\r
69 //\r
70 Status = ShellInitialize();\r
71 ASSERT_EFI_ERROR(Status);\r
72\r
73 Status = CommandInit();\r
74 ASSERT_EFI_ERROR(Status);\r
75\r
76 //\r
77 // parse the command line\r
78 //\r
79 Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
80 if (EFI_ERROR(Status)) {\r
81 if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
82 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
83 FreePool(ProblemParam);\r
84 ShellStatus = SHELL_INVALID_PARAMETER;\r
85 } else {\r
86 ASSERT(FALSE);\r
87 }\r
88 } else {\r
89 //\r
90 // Check for -d\r
91 //\r
92 if (ShellCommandLineGetFlag(Package, L"-d")){\r
392fa5a6 93 if (ShellCommandLineGetCount(Package) < 4) {\r
94 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
95 ShellStatus = SHELL_INVALID_PARAMETER;\r
96 } else if (ShellCommandLineGetCount(Package) > 4) {\r
632820d1 97 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
98 ShellStatus = SHELL_INVALID_PARAMETER;\r
99 } else {\r
100 WhatToDo = FileTypeDiskBuffer;\r
101 Name = ShellCommandLineGetRawValue(Package, 1);\r
102 Offset = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 2));\r
103 Size = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 3));\r
104 }\r
392fa5a6 105 if (Offset == (UINTN)-1 || Size == (UINTN)-1) {\r
106 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM_VAL), gShellDebug1HiiHandle, L"-d");\r
107 ShellStatus = SHELL_INVALID_PARAMETER;\r
108 }\r
632820d1 109 }\r
110\r
111 //\r
112 // check for -f\r
113 //\r
114 if (ShellCommandLineGetFlag(Package, L"-f") && (WhatToDo == FileTypeNone)){\r
392fa5a6 115 if (ShellCommandLineGetCount(Package) < 2) {\r
116 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
117 ShellStatus = SHELL_INVALID_PARAMETER;\r
118 } else if (ShellCommandLineGetCount(Package) > 2) {\r
632820d1 119 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
120 ShellStatus = SHELL_INVALID_PARAMETER;\r
121 } else {\r
122 Name = ShellCommandLineGetRawValue(Package, 1);\r
33c031ee 123 if (Name == NULL || !IsValidFileName(Name)) {\r
632820d1 124 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Name);\r
125 ShellStatus = SHELL_INVALID_PARAMETER;\r
126 } else {\r
127 WhatToDo = FileTypeFileBuffer;\r
128 }\r
129 }\r
130 }\r
131\r
132 //\r
133 // check for -m\r
134 //\r
135 if (ShellCommandLineGetFlag(Package, L"-m") && (WhatToDo == FileTypeNone)){\r
136 if (ShellCommandLineGetCount(Package) < 3) {\r
137 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
138 ShellStatus = SHELL_INVALID_PARAMETER;\r
139 } else if (ShellCommandLineGetCount(Package) > 3) {\r
140 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
141 ShellStatus = SHELL_INVALID_PARAMETER;\r
142 } else {\r
143 WhatToDo = FileTypeMemBuffer;\r
144 Offset = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 1));\r
145 Size = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 2));\r
146 }\r
147 }\r
ae724571 148 Name = ShellCommandLineGetRawValue(Package, 1);\r
149 if (WhatToDo == FileTypeNone && Name != NULL) {\r
980d554e 150 if (!IsValidFileName(Name)) {\r
151 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Name);\r
152 ShellStatus = SHELL_INVALID_PARAMETER;\r
153 } else {\r
154 WhatToDo = FileTypeFileBuffer;\r
155 }\r
156 } else if (WhatToDo == FileTypeNone) {\r
157 if (gEfiShellProtocol->GetCurDir(NULL) == NULL) {\r
158 ShellStatus = SHELL_NOT_FOUND;\r
159 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellDebug1HiiHandle);\r
160 } else {\r
161 NewName = EditGetDefaultFileName(L"bin");\r
162 Name = NewName;\r
163 WhatToDo = FileTypeFileBuffer;\r
164 }\r
165 }\r
632820d1 166\r
167 if (ShellStatus == SHELL_SUCCESS && WhatToDo == FileTypeNone) {\r
168 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
169 ShellStatus = SHELL_INVALID_PARAMETER;\r
392fa5a6 170 } else if (WhatToDo == FileTypeFileBuffer && ShellGetCurrentDir(NULL) == NULL) {\r
171 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellDebug1HiiHandle);\r
172 ShellStatus = SHELL_INVALID_PARAMETER;\r
e0c2cc6f 173 }\r
632820d1 174\r
175 if (ShellStatus == SHELL_SUCCESS) {\r
176 //\r
177 // Do the editor\r
178 //\r
179 Status = HMainEditorInit ();\r
180 if (EFI_ERROR (Status)) {\r
181 gST->ConOut->ClearScreen (gST->ConOut);\r
182 gST->ConOut->EnableCursor (gST->ConOut, TRUE);\r
183 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_INIT_FAILED), gShellDebug1HiiHandle);\r
184 } else {\r
185 HMainEditorBackup ();\r
186 switch (WhatToDo) {\r
187 case FileTypeFileBuffer:\r
188 Status = HBufferImageRead (\r
189 Name,\r
190 NULL,\r
191 0,\r
192 0,\r
193 0,\r
194 0,\r
195 FileTypeFileBuffer,\r
196 FALSE\r
197 );\r
198 break;\r
199\r
200 case FileTypeDiskBuffer:\r
201 Status = HBufferImageRead (\r
202 NULL,\r
203 Name,\r
204 Offset,\r
205 Size,\r
206 0,\r
207 0,\r
208 FileTypeDiskBuffer,\r
209 FALSE\r
210 );\r
211 break;\r
212\r
213 case FileTypeMemBuffer:\r
214 Status = HBufferImageRead (\r
215 NULL,\r
216 NULL,\r
217 0,\r
218 0,\r
219 (UINT32) Offset,\r
220 Size,\r
221 FileTypeMemBuffer,\r
222 FALSE\r
223 );\r
224 break;\r
225\r
e0c2cc6f 226 default:\r
227 Status = EFI_NOT_FOUND;\r
228 break;\r
632820d1 229 }\r
230 if (!EFI_ERROR (Status)) {\r
231 HMainEditorRefresh ();\r
232 Status = HMainEditorKeyInput ();\r
233 }\r
234 if (Status != EFI_OUT_OF_RESOURCES) {\r
235 //\r
236 // back up the status string\r
237 //\r
238 Buffer = CatSPrint (NULL, L"%s", StatusBarGetString());\r
239 }\r
240 }\r
241\r
242 //\r
243 // cleanup\r
244 //\r
245 HMainEditorCleanup ();\r
246\r
247 if (!EFI_ERROR (Status)) {\r
248 if (ShellStatus == SHELL_SUCCESS) {\r
249 ShellStatus = SHELL_UNSUPPORTED;\r
250 }\r
251 }\r
252\r
253 //\r
254 // print editor exit code on screen\r
255 //\r
256 if (Status == EFI_OUT_OF_RESOURCES) {\r
257 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle);\r
258 } else if (EFI_ERROR(Status)){\r
259 if (Buffer != NULL) {\r
260 if (StrCmp (Buffer, L"") != 0) {\r
261 //\r
262 // print out the status string\r
263 //\r
264 ShellPrintEx(-1, -1, L"%s", gShellDebug1HiiHandle, Buffer);\r
265 } else {\r
266 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_UNKNOWN_EDITOR), gShellDebug1HiiHandle);\r
267 }\r
268 } else {\r
269 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_UNKNOWN_EDITOR), gShellDebug1HiiHandle);\r
270 }\r
271 }\r
272 }\r
980d554e 273 ShellCommandLineFreeVarList (Package);\r
632820d1 274 }\r
275\r
276 SHELL_FREE_NON_NULL (Buffer);\r
980d554e 277 SHELL_FREE_NON_NULL (NewName);\r
632820d1 278 return ShellStatus;\r
279}\r