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