]> git.proxmox.com Git - mirror_edk2.git/blame - ShellPkg/Library/UefiShellDebug1CommandsLib/Comp.c
ShellPkg: fix Comp command output
[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / Comp.c
CommitLineData
5d73d92f 1/** @file\r
2 Main file for Comp shell Debug1 function.\r
3\r
33eb033f 4 Copyright (c) 2010 - 2014, 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
3737ac2b 17/**\r
18 Function for 'comp' command.\r
19\r
20 @param[in] ImageHandle Handle to the Image (NULL if Internal).\r
21 @param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
22**/\r
5d73d92f 23SHELL_STATUS\r
24EFIAPI\r
25ShellCommandRunComp (\r
26 IN EFI_HANDLE ImageHandle,\r
27 IN EFI_SYSTEM_TABLE *SystemTable\r
28 )\r
29{\r
30 EFI_STATUS Status;\r
31 LIST_ENTRY *Package;\r
32 CHAR16 *ProblemParam;\r
33 SHELL_STATUS ShellStatus;\r
34 UINTN LoopVar;\r
35 SHELL_FILE_HANDLE FileHandle1;\r
36 SHELL_FILE_HANDLE FileHandle2;\r
37 UINT8 ErrorCount;\r
38 UINT64 Size1;\r
39 UINT64 Size2;\r
40 UINT8 DataFromFile1;\r
41 UINT8 DataFromFile2;\r
42 UINT8 ADF_File11;\r
43 UINT8 ADF_File12;\r
44 UINT8 ADF_File13;\r
45 UINT8 ADF_File21;\r
46 UINT8 ADF_File22;\r
47 UINT8 ADF_File23;\r
48 UINTN DataSizeFromFile1;\r
49 UINTN DataSizeFromFile2;\r
50 CHAR16 *FileName1;\r
51 CHAR16 *FileName2;\r
ecae5117 52 CONST CHAR16 *TempParam;\r
33eb033f 53 UINTN ErrorAddress;\r
5d73d92f 54\r
55 ErrorCount = 0;\r
56 ShellStatus = SHELL_SUCCESS;\r
57 Status = EFI_SUCCESS;\r
58 FileName1 = NULL;\r
59 FileName2 = NULL;\r
60 FileHandle1 = NULL;\r
61 FileHandle2 = NULL;\r
62 Size1 = 0;\r
63\r
64 //\r
65 // initialize the shell lib (we must be in non-auto-init...)\r
66 //\r
67 Status = ShellInitialize();\r
68 ASSERT_EFI_ERROR(Status);\r
69\r
70 Status = CommandInit();\r
71 ASSERT_EFI_ERROR(Status);\r
72\r
73 //\r
74 // parse the command line\r
75 //\r
76 Status = ShellCommandLineParse (EmptyParamList, &Package, &ProblemParam, TRUE);\r
77 if (EFI_ERROR(Status)) {\r
78 if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
79 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
80 FreePool(ProblemParam);\r
81 ShellStatus = SHELL_INVALID_PARAMETER;\r
82 } else {\r
83 ASSERT(FALSE);\r
84 }\r
85 } else {\r
86 if (ShellCommandLineGetCount(Package) > 3) {\r
87 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
88 ShellStatus = SHELL_INVALID_PARAMETER;\r
89 } else if (ShellCommandLineGetCount(Package) < 3) {\r
90 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
91 ShellStatus = SHELL_INVALID_PARAMETER;\r
92 } else {\r
ecae5117 93 TempParam = ShellCommandLineGetRawValue(Package, 1);\r
94 ASSERT(TempParam != NULL);\r
95 FileName1 = ShellFindFilePath(TempParam);\r
3737ac2b 96 if (FileName1 == NULL) {\r
ecae5117 97 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_FIND_FAIL), gShellDebug1HiiHandle, TempParam);\r
5d73d92f 98 ShellStatus = SHELL_NOT_FOUND;\r
3737ac2b 99 } else {\r
100 Status = ShellOpenFileByName(FileName1, &FileHandle1, EFI_FILE_MODE_READ, 0);\r
101 if (EFI_ERROR(Status)) {\r
ecae5117 102 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, TempParam, Status);\r
3737ac2b 103 ShellStatus = SHELL_NOT_FOUND;\r
104 }\r
5d73d92f 105 }\r
ecae5117 106 TempParam = ShellCommandLineGetRawValue(Package, 2);\r
107 ASSERT(TempParam != NULL);\r
108 FileName2 = ShellFindFilePath(TempParam);\r
3737ac2b 109 if (FileName2 == NULL) {\r
ecae5117 110 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_FIND_FAIL), gShellDebug1HiiHandle, TempParam);\r
5d73d92f 111 ShellStatus = SHELL_NOT_FOUND;\r
3737ac2b 112 } else {\r
113 Status = ShellOpenFileByName(FileName2, &FileHandle2, EFI_FILE_MODE_READ, 0);\r
114 if (EFI_ERROR(Status)) {\r
ecae5117 115 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, TempParam, Status);\r
3737ac2b 116 ShellStatus = SHELL_NOT_FOUND;\r
117 }\r
5d73d92f 118 }\r
119 if (ShellStatus == SHELL_SUCCESS) {\r
120 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_HEADER), gShellDebug1HiiHandle, FileName1, FileName2);\r
121 Status = gEfiShellProtocol->GetFileSize(FileHandle1, &Size1);\r
122 ASSERT_EFI_ERROR(Status);\r
123 Status = gEfiShellProtocol->GetFileSize(FileHandle2, &Size2);\r
124 ASSERT_EFI_ERROR(Status);\r
125 if (Size1 != Size2) {\r
126 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_SIZE_FAIL), gShellDebug1HiiHandle);\r
127 ErrorCount++;\r
128 ShellStatus = SHELL_NOT_EQUAL;\r
129 }\r
130 }\r
131 if (ShellStatus == SHELL_SUCCESS) {\r
132 for (LoopVar = 0 ; LoopVar < Size1 && ErrorCount <= 10 ; LoopVar++) {\r
133 DataSizeFromFile1 = 1;\r
134 DataSizeFromFile2 = 1;\r
135 Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &DataFromFile1);\r
136 ASSERT_EFI_ERROR(Status);\r
137 Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &DataFromFile2);\r
138 ASSERT_EFI_ERROR(Status);\r
139 if (DataFromFile1 != DataFromFile2) {\r
33eb033f 140 ErrorAddress = LoopVar;\r
5d73d92f 141 ADF_File11 = 0;\r
142 ADF_File12 = 0;\r
143 ADF_File13 = 0;\r
144 ADF_File21 = 0;\r
145 ADF_File22 = 0;\r
146 ADF_File23 = 0;\r
33eb033f
JC
147\r
148 //\r
149 // Now check the next 3 bytes if possible. This will make output\r
150 // cleaner when there are a sequence of differences.\r
151 //\r
5d73d92f 152 if (LoopVar + 1 < Size1) {\r
153 LoopVar++;\r
154 DataSizeFromFile1 = 1;\r
155 DataSizeFromFile2 = 1;\r
156 Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &ADF_File11);\r
157 ASSERT_EFI_ERROR(Status);\r
158 Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &ADF_File21);\r
159 ASSERT_EFI_ERROR(Status);\r
160 if (LoopVar + 1 < Size1) {\r
161 LoopVar++;\r
162 DataSizeFromFile1 = 1;\r
163 DataSizeFromFile2 = 1;\r
164 Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &ADF_File12);\r
165 ASSERT_EFI_ERROR(Status);\r
166 Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &ADF_File22);\r
167 ASSERT_EFI_ERROR(Status);\r
168 if (LoopVar + 1 < Size1) {\r
169 LoopVar++;\r
170 DataSizeFromFile1 = 1;\r
171 DataSizeFromFile2 = 1;\r
172 Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &ADF_File13);\r
173 ASSERT_EFI_ERROR(Status);\r
174 Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &ADF_File23);\r
175 ASSERT_EFI_ERROR(Status);\r
176 }\r
177 }\r
178 }\r
33eb033f
JC
179\r
180 //\r
181 // Print out based on highest of the 4 bytes that are different.\r
182 //\r
5d73d92f 183 if (ADF_File13 != ADF_File23) {\r
184 ShellPrintHiiEx(\r
185 -1,\r
186 -1,\r
187 NULL,\r
188 STRING_TOKEN (STR_COMP_SPOT_FAIL4),\r
189 gShellDebug1HiiHandle,\r
190 ++ErrorCount,\r
191 FileName1,\r
33eb033f 192 ErrorAddress,\r
5d73d92f 193 DataFromFile1, ADF_File11, ADF_File12, ADF_File13,\r
194 DataFromFile1, ADF_File11, ADF_File12, ADF_File13,\r
195 FileName2,\r
33eb033f 196 ErrorAddress,\r
5d73d92f 197 DataFromFile2, ADF_File21, ADF_File22, ADF_File23,\r
198 DataFromFile2, ADF_File21, ADF_File22, ADF_File23\r
199 );\r
200 } else if (ADF_File12 != ADF_File22) {\r
201 ShellPrintHiiEx(\r
202 -1,\r
203 -1,\r
204 NULL,\r
205 STRING_TOKEN (STR_COMP_SPOT_FAIL3),\r
206 gShellDebug1HiiHandle,\r
207 ++ErrorCount,\r
208 FileName1,\r
33eb033f 209 ErrorAddress,\r
5d73d92f 210 DataFromFile1, ADF_File11, ADF_File12,\r
211 DataFromFile1, ADF_File11, ADF_File12,\r
212 FileName2,\r
33eb033f 213 ErrorAddress,\r
5d73d92f 214 DataFromFile2, ADF_File21, ADF_File22,\r
215 DataFromFile2, ADF_File21, ADF_File22\r
216 );\r
217 } else if (ADF_File11 != ADF_File21) {\r
218 ShellPrintHiiEx(\r
219 -1,\r
220 -1,\r
221 NULL,\r
222 STRING_TOKEN (STR_COMP_SPOT_FAIL2),\r
223 gShellDebug1HiiHandle,\r
224 ++ErrorCount,\r
225 FileName1,\r
33eb033f 226 ErrorAddress,\r
5d73d92f 227 DataFromFile1, ADF_File11,\r
228 DataFromFile1, ADF_File11,\r
229 FileName2,\r
33eb033f 230 ErrorAddress,\r
5d73d92f 231 DataFromFile2, ADF_File21,\r
232 DataFromFile2, ADF_File21\r
233 );\r
234 } else {\r
235 ShellPrintHiiEx(\r
236 -1,\r
237 -1,\r
238 NULL,\r
239 STRING_TOKEN (STR_COMP_SPOT_FAIL1),\r
240 gShellDebug1HiiHandle,\r
241 ++ErrorCount,\r
242 FileName1,\r
33eb033f 243 ErrorAddress,\r
5d73d92f 244 DataFromFile1,\r
245 DataFromFile1,\r
246 FileName2,\r
33eb033f 247 ErrorAddress,\r
5d73d92f 248 DataFromFile2,\r
249 DataFromFile2\r
250 );\r
251 }\r
252 ShellStatus = SHELL_NOT_EQUAL;\r
253 }\r
254 }\r
3737ac2b 255 if (ErrorCount == 0) {\r
256 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_FOOTER_PASS), gShellDebug1HiiHandle);\r
257 } else {\r
258 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_FOOTER_FAIL), gShellDebug1HiiHandle);\r
259 }\r
5d73d92f 260 }\r
261 }\r
262\r
263 ShellCommandLineFreeVarList (Package);\r
264 }\r
3737ac2b 265 SHELL_FREE_NON_NULL(FileName1);\r
266 SHELL_FREE_NON_NULL(FileName2);\r
267\r
5d73d92f 268 if (FileHandle1 != NULL) {\r
269 gEfiShellProtocol->CloseFile(FileHandle1);\r
270 }\r
271 if (FileHandle2 != NULL) {\r
272 gEfiShellProtocol->CloseFile(FileHandle2);\r
273 }\r
274\r
275 return (ShellStatus);\r
276}\r