]> git.proxmox.com Git - mirror_edk2.git/blame - ShellPkg/Library/UefiShellDebug1CommandsLib/Comp.c
ShellPkg: Verify memory allocations without ASSERT.
[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / Comp.c
CommitLineData
5d73d92f 1/** @file\r
2 Main file for Comp shell Debug1 function.\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
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
52\r
53 ErrorCount = 0;\r
54 ShellStatus = SHELL_SUCCESS;\r
55 Status = EFI_SUCCESS;\r
56 FileName1 = NULL;\r
57 FileName2 = NULL;\r
58 FileHandle1 = NULL;\r
59 FileHandle2 = NULL;\r
60 Size1 = 0;\r
61\r
62 //\r
63 // initialize the shell lib (we must be in non-auto-init...)\r
64 //\r
65 Status = ShellInitialize();\r
66 ASSERT_EFI_ERROR(Status);\r
67\r
68 Status = CommandInit();\r
69 ASSERT_EFI_ERROR(Status);\r
70\r
71 //\r
72 // parse the command line\r
73 //\r
74 Status = ShellCommandLineParse (EmptyParamList, &Package, &ProblemParam, TRUE);\r
75 if (EFI_ERROR(Status)) {\r
76 if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
77 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
78 FreePool(ProblemParam);\r
79 ShellStatus = SHELL_INVALID_PARAMETER;\r
80 } else {\r
81 ASSERT(FALSE);\r
82 }\r
83 } else {\r
84 if (ShellCommandLineGetCount(Package) > 3) {\r
85 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
86 ShellStatus = SHELL_INVALID_PARAMETER;\r
87 } else if (ShellCommandLineGetCount(Package) < 3) {\r
88 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
89 ShellStatus = SHELL_INVALID_PARAMETER;\r
90 } else {\r
91 FileName1 = ShellFindFilePath(ShellCommandLineGetRawValue(Package, 1));\r
3737ac2b 92 if (FileName1 == NULL) {\r
93 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_FIND_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue(Package, 1));\r
5d73d92f 94 ShellStatus = SHELL_NOT_FOUND;\r
3737ac2b 95 } else {\r
96 Status = ShellOpenFileByName(FileName1, &FileHandle1, EFI_FILE_MODE_READ, 0);\r
97 if (EFI_ERROR(Status)) {\r
98 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue(Package, 1), Status);\r
99 ShellStatus = SHELL_NOT_FOUND;\r
100 }\r
5d73d92f 101 }\r
3737ac2b 102 FileName2 = ShellFindFilePath(ShellCommandLineGetRawValue(Package, 2));\r
103 if (FileName2 == NULL) {\r
104 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_FIND_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue(Package, 2));\r
5d73d92f 105 ShellStatus = SHELL_NOT_FOUND;\r
3737ac2b 106 } else {\r
107 Status = ShellOpenFileByName(FileName2, &FileHandle2, EFI_FILE_MODE_READ, 0);\r
108 if (EFI_ERROR(Status)) {\r
109 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, ShellCommandLineGetRawValue(Package, 2), Status);\r
110 ShellStatus = SHELL_NOT_FOUND;\r
111 }\r
5d73d92f 112 }\r
113 if (ShellStatus == SHELL_SUCCESS) {\r
114 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_HEADER), gShellDebug1HiiHandle, FileName1, FileName2);\r
115 Status = gEfiShellProtocol->GetFileSize(FileHandle1, &Size1);\r
116 ASSERT_EFI_ERROR(Status);\r
117 Status = gEfiShellProtocol->GetFileSize(FileHandle2, &Size2);\r
118 ASSERT_EFI_ERROR(Status);\r
119 if (Size1 != Size2) {\r
120 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_SIZE_FAIL), gShellDebug1HiiHandle);\r
121 ErrorCount++;\r
122 ShellStatus = SHELL_NOT_EQUAL;\r
123 }\r
124 }\r
125 if (ShellStatus == SHELL_SUCCESS) {\r
126 for (LoopVar = 0 ; LoopVar < Size1 && ErrorCount <= 10 ; LoopVar++) {\r
127 DataSizeFromFile1 = 1;\r
128 DataSizeFromFile2 = 1;\r
129 Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &DataFromFile1);\r
130 ASSERT_EFI_ERROR(Status);\r
131 Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &DataFromFile2);\r
132 ASSERT_EFI_ERROR(Status);\r
133 if (DataFromFile1 != DataFromFile2) {\r
134 ADF_File11 = 0;\r
135 ADF_File12 = 0;\r
136 ADF_File13 = 0;\r
137 ADF_File21 = 0;\r
138 ADF_File22 = 0;\r
139 ADF_File23 = 0;\r
140 if (LoopVar + 1 < Size1) {\r
141 LoopVar++;\r
142 DataSizeFromFile1 = 1;\r
143 DataSizeFromFile2 = 1;\r
144 Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &ADF_File11);\r
145 ASSERT_EFI_ERROR(Status);\r
146 Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &ADF_File21);\r
147 ASSERT_EFI_ERROR(Status);\r
148 if (LoopVar + 1 < Size1) {\r
149 LoopVar++;\r
150 DataSizeFromFile1 = 1;\r
151 DataSizeFromFile2 = 1;\r
152 Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &ADF_File12);\r
153 ASSERT_EFI_ERROR(Status);\r
154 Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &ADF_File22);\r
155 ASSERT_EFI_ERROR(Status);\r
156 if (LoopVar + 1 < Size1) {\r
157 LoopVar++;\r
158 DataSizeFromFile1 = 1;\r
159 DataSizeFromFile2 = 1;\r
160 Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &ADF_File13);\r
161 ASSERT_EFI_ERROR(Status);\r
162 Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &ADF_File23);\r
163 ASSERT_EFI_ERROR(Status);\r
164 }\r
165 }\r
166 }\r
167 if (ADF_File13 != ADF_File23) {\r
168 ShellPrintHiiEx(\r
169 -1,\r
170 -1,\r
171 NULL,\r
172 STRING_TOKEN (STR_COMP_SPOT_FAIL4),\r
173 gShellDebug1HiiHandle,\r
174 ++ErrorCount,\r
175 FileName1,\r
176 LoopVar,\r
177 DataFromFile1, ADF_File11, ADF_File12, ADF_File13,\r
178 DataFromFile1, ADF_File11, ADF_File12, ADF_File13,\r
179 FileName2,\r
180 LoopVar,\r
181 DataFromFile2, ADF_File21, ADF_File22, ADF_File23,\r
182 DataFromFile2, ADF_File21, ADF_File22, ADF_File23\r
183 );\r
184 } else if (ADF_File12 != ADF_File22) {\r
185 ShellPrintHiiEx(\r
186 -1,\r
187 -1,\r
188 NULL,\r
189 STRING_TOKEN (STR_COMP_SPOT_FAIL3),\r
190 gShellDebug1HiiHandle,\r
191 ++ErrorCount,\r
192 FileName1,\r
193 LoopVar,\r
194 DataFromFile1, ADF_File11, ADF_File12,\r
195 DataFromFile1, ADF_File11, ADF_File12,\r
196 FileName2,\r
197 LoopVar,\r
198 DataFromFile2, ADF_File21, ADF_File22,\r
199 DataFromFile2, ADF_File21, ADF_File22\r
200 );\r
201 } else if (ADF_File11 != ADF_File21) {\r
202 ShellPrintHiiEx(\r
203 -1,\r
204 -1,\r
205 NULL,\r
206 STRING_TOKEN (STR_COMP_SPOT_FAIL2),\r
207 gShellDebug1HiiHandle,\r
208 ++ErrorCount,\r
209 FileName1,\r
210 LoopVar,\r
211 DataFromFile1, ADF_File11,\r
212 DataFromFile1, ADF_File11,\r
213 FileName2,\r
214 LoopVar,\r
215 DataFromFile2, ADF_File21,\r
216 DataFromFile2, ADF_File21\r
217 );\r
218 } else {\r
219 ShellPrintHiiEx(\r
220 -1,\r
221 -1,\r
222 NULL,\r
223 STRING_TOKEN (STR_COMP_SPOT_FAIL1),\r
224 gShellDebug1HiiHandle,\r
225 ++ErrorCount,\r
226 FileName1,\r
227 LoopVar,\r
228 DataFromFile1,\r
229 DataFromFile1,\r
230 FileName2,\r
231 LoopVar,\r
232 DataFromFile2,\r
233 DataFromFile2\r
234 );\r
235 }\r
236 ShellStatus = SHELL_NOT_EQUAL;\r
237 }\r
238 }\r
3737ac2b 239 if (ErrorCount == 0) {\r
240 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_FOOTER_PASS), gShellDebug1HiiHandle);\r
241 } else {\r
242 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_FOOTER_FAIL), gShellDebug1HiiHandle);\r
243 }\r
5d73d92f 244 }\r
245 }\r
246\r
247 ShellCommandLineFreeVarList (Package);\r
248 }\r
3737ac2b 249 SHELL_FREE_NON_NULL(FileName1);\r
250 SHELL_FREE_NON_NULL(FileName2);\r
251\r
5d73d92f 252 if (FileHandle1 != NULL) {\r
253 gEfiShellProtocol->CloseFile(FileHandle1);\r
254 }\r
255 if (FileHandle2 != NULL) {\r
256 gEfiShellProtocol->CloseFile(FileHandle2);\r
257 }\r
258\r
259 return (ShellStatus);\r
260}\r