ecae51177e
This adds lots of pointer verification with ASSERTs only used when the condition should be impossible and never for memory allocation. signed-off-by: jcarsey reviewed-by: geekboy15a git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12523 6f19259b-4bc3-4df7-8a09-765794883524
266 lines
9.8 KiB
C
266 lines
9.8 KiB
C
/** @file
|
|
Main file for Comp shell Debug1 function.
|
|
|
|
Copyright (c) 2010 - 2011, Intel Corporation. All rights reserved.<BR>
|
|
This program and the accompanying materials
|
|
are licensed and made available under the terms and conditions of the BSD License
|
|
which accompanies this distribution. The full text of the license may be found at
|
|
http://opensource.org/licenses/bsd-license.php
|
|
|
|
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
|
|
|
|
**/
|
|
|
|
#include "UefiShellDebug1CommandsLib.h"
|
|
|
|
/**
|
|
Function for 'comp' command.
|
|
|
|
@param[in] ImageHandle Handle to the Image (NULL if Internal).
|
|
@param[in] SystemTable Pointer to the System Table (NULL if Internal).
|
|
**/
|
|
SHELL_STATUS
|
|
EFIAPI
|
|
ShellCommandRunComp (
|
|
IN EFI_HANDLE ImageHandle,
|
|
IN EFI_SYSTEM_TABLE *SystemTable
|
|
)
|
|
{
|
|
EFI_STATUS Status;
|
|
LIST_ENTRY *Package;
|
|
CHAR16 *ProblemParam;
|
|
SHELL_STATUS ShellStatus;
|
|
UINTN LoopVar;
|
|
SHELL_FILE_HANDLE FileHandle1;
|
|
SHELL_FILE_HANDLE FileHandle2;
|
|
UINT8 ErrorCount;
|
|
UINT64 Size1;
|
|
UINT64 Size2;
|
|
UINT8 DataFromFile1;
|
|
UINT8 DataFromFile2;
|
|
UINT8 ADF_File11;
|
|
UINT8 ADF_File12;
|
|
UINT8 ADF_File13;
|
|
UINT8 ADF_File21;
|
|
UINT8 ADF_File22;
|
|
UINT8 ADF_File23;
|
|
UINTN DataSizeFromFile1;
|
|
UINTN DataSizeFromFile2;
|
|
CHAR16 *FileName1;
|
|
CHAR16 *FileName2;
|
|
CONST CHAR16 *TempParam;
|
|
|
|
ErrorCount = 0;
|
|
ShellStatus = SHELL_SUCCESS;
|
|
Status = EFI_SUCCESS;
|
|
FileName1 = NULL;
|
|
FileName2 = NULL;
|
|
FileHandle1 = NULL;
|
|
FileHandle2 = NULL;
|
|
Size1 = 0;
|
|
|
|
//
|
|
// initialize the shell lib (we must be in non-auto-init...)
|
|
//
|
|
Status = ShellInitialize();
|
|
ASSERT_EFI_ERROR(Status);
|
|
|
|
Status = CommandInit();
|
|
ASSERT_EFI_ERROR(Status);
|
|
|
|
//
|
|
// parse the command line
|
|
//
|
|
Status = ShellCommandLineParse (EmptyParamList, &Package, &ProblemParam, TRUE);
|
|
if (EFI_ERROR(Status)) {
|
|
if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);
|
|
FreePool(ProblemParam);
|
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
|
} else {
|
|
ASSERT(FALSE);
|
|
}
|
|
} else {
|
|
if (ShellCommandLineGetCount(Package) > 3) {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);
|
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
|
} else if (ShellCommandLineGetCount(Package) < 3) {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);
|
|
ShellStatus = SHELL_INVALID_PARAMETER;
|
|
} else {
|
|
TempParam = ShellCommandLineGetRawValue(Package, 1);
|
|
ASSERT(TempParam != NULL);
|
|
FileName1 = ShellFindFilePath(TempParam);
|
|
if (FileName1 == NULL) {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_FIND_FAIL), gShellDebug1HiiHandle, TempParam);
|
|
ShellStatus = SHELL_NOT_FOUND;
|
|
} else {
|
|
Status = ShellOpenFileByName(FileName1, &FileHandle1, EFI_FILE_MODE_READ, 0);
|
|
if (EFI_ERROR(Status)) {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, TempParam, Status);
|
|
ShellStatus = SHELL_NOT_FOUND;
|
|
}
|
|
}
|
|
TempParam = ShellCommandLineGetRawValue(Package, 2);
|
|
ASSERT(TempParam != NULL);
|
|
FileName2 = ShellFindFilePath(TempParam);
|
|
if (FileName2 == NULL) {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_FIND_FAIL), gShellDebug1HiiHandle, TempParam);
|
|
ShellStatus = SHELL_NOT_FOUND;
|
|
} else {
|
|
Status = ShellOpenFileByName(FileName2, &FileHandle2, EFI_FILE_MODE_READ, 0);
|
|
if (EFI_ERROR(Status)) {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_FILE_OPEN_FAIL), gShellDebug1HiiHandle, TempParam, Status);
|
|
ShellStatus = SHELL_NOT_FOUND;
|
|
}
|
|
}
|
|
if (ShellStatus == SHELL_SUCCESS) {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_HEADER), gShellDebug1HiiHandle, FileName1, FileName2);
|
|
Status = gEfiShellProtocol->GetFileSize(FileHandle1, &Size1);
|
|
ASSERT_EFI_ERROR(Status);
|
|
Status = gEfiShellProtocol->GetFileSize(FileHandle2, &Size2);
|
|
ASSERT_EFI_ERROR(Status);
|
|
if (Size1 != Size2) {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_SIZE_FAIL), gShellDebug1HiiHandle);
|
|
ErrorCount++;
|
|
ShellStatus = SHELL_NOT_EQUAL;
|
|
}
|
|
}
|
|
if (ShellStatus == SHELL_SUCCESS) {
|
|
for (LoopVar = 0 ; LoopVar < Size1 && ErrorCount <= 10 ; LoopVar++) {
|
|
DataSizeFromFile1 = 1;
|
|
DataSizeFromFile2 = 1;
|
|
Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &DataFromFile1);
|
|
ASSERT_EFI_ERROR(Status);
|
|
Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &DataFromFile2);
|
|
ASSERT_EFI_ERROR(Status);
|
|
if (DataFromFile1 != DataFromFile2) {
|
|
ADF_File11 = 0;
|
|
ADF_File12 = 0;
|
|
ADF_File13 = 0;
|
|
ADF_File21 = 0;
|
|
ADF_File22 = 0;
|
|
ADF_File23 = 0;
|
|
if (LoopVar + 1 < Size1) {
|
|
LoopVar++;
|
|
DataSizeFromFile1 = 1;
|
|
DataSizeFromFile2 = 1;
|
|
Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &ADF_File11);
|
|
ASSERT_EFI_ERROR(Status);
|
|
Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &ADF_File21);
|
|
ASSERT_EFI_ERROR(Status);
|
|
if (LoopVar + 1 < Size1) {
|
|
LoopVar++;
|
|
DataSizeFromFile1 = 1;
|
|
DataSizeFromFile2 = 1;
|
|
Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &ADF_File12);
|
|
ASSERT_EFI_ERROR(Status);
|
|
Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &ADF_File22);
|
|
ASSERT_EFI_ERROR(Status);
|
|
if (LoopVar + 1 < Size1) {
|
|
LoopVar++;
|
|
DataSizeFromFile1 = 1;
|
|
DataSizeFromFile2 = 1;
|
|
Status = gEfiShellProtocol->ReadFile(FileHandle1, &DataSizeFromFile1, &ADF_File13);
|
|
ASSERT_EFI_ERROR(Status);
|
|
Status = gEfiShellProtocol->ReadFile(FileHandle2, &DataSizeFromFile2, &ADF_File23);
|
|
ASSERT_EFI_ERROR(Status);
|
|
}
|
|
}
|
|
}
|
|
if (ADF_File13 != ADF_File23) {
|
|
ShellPrintHiiEx(
|
|
-1,
|
|
-1,
|
|
NULL,
|
|
STRING_TOKEN (STR_COMP_SPOT_FAIL4),
|
|
gShellDebug1HiiHandle,
|
|
++ErrorCount,
|
|
FileName1,
|
|
LoopVar,
|
|
DataFromFile1, ADF_File11, ADF_File12, ADF_File13,
|
|
DataFromFile1, ADF_File11, ADF_File12, ADF_File13,
|
|
FileName2,
|
|
LoopVar,
|
|
DataFromFile2, ADF_File21, ADF_File22, ADF_File23,
|
|
DataFromFile2, ADF_File21, ADF_File22, ADF_File23
|
|
);
|
|
} else if (ADF_File12 != ADF_File22) {
|
|
ShellPrintHiiEx(
|
|
-1,
|
|
-1,
|
|
NULL,
|
|
STRING_TOKEN (STR_COMP_SPOT_FAIL3),
|
|
gShellDebug1HiiHandle,
|
|
++ErrorCount,
|
|
FileName1,
|
|
LoopVar,
|
|
DataFromFile1, ADF_File11, ADF_File12,
|
|
DataFromFile1, ADF_File11, ADF_File12,
|
|
FileName2,
|
|
LoopVar,
|
|
DataFromFile2, ADF_File21, ADF_File22,
|
|
DataFromFile2, ADF_File21, ADF_File22
|
|
);
|
|
} else if (ADF_File11 != ADF_File21) {
|
|
ShellPrintHiiEx(
|
|
-1,
|
|
-1,
|
|
NULL,
|
|
STRING_TOKEN (STR_COMP_SPOT_FAIL2),
|
|
gShellDebug1HiiHandle,
|
|
++ErrorCount,
|
|
FileName1,
|
|
LoopVar,
|
|
DataFromFile1, ADF_File11,
|
|
DataFromFile1, ADF_File11,
|
|
FileName2,
|
|
LoopVar,
|
|
DataFromFile2, ADF_File21,
|
|
DataFromFile2, ADF_File21
|
|
);
|
|
} else {
|
|
ShellPrintHiiEx(
|
|
-1,
|
|
-1,
|
|
NULL,
|
|
STRING_TOKEN (STR_COMP_SPOT_FAIL1),
|
|
gShellDebug1HiiHandle,
|
|
++ErrorCount,
|
|
FileName1,
|
|
LoopVar,
|
|
DataFromFile1,
|
|
DataFromFile1,
|
|
FileName2,
|
|
LoopVar,
|
|
DataFromFile2,
|
|
DataFromFile2
|
|
);
|
|
}
|
|
ShellStatus = SHELL_NOT_EQUAL;
|
|
}
|
|
}
|
|
if (ErrorCount == 0) {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_FOOTER_PASS), gShellDebug1HiiHandle);
|
|
} else {
|
|
ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_COMP_FOOTER_FAIL), gShellDebug1HiiHandle);
|
|
}
|
|
}
|
|
}
|
|
|
|
ShellCommandLineFreeVarList (Package);
|
|
}
|
|
SHELL_FREE_NON_NULL(FileName1);
|
|
SHELL_FREE_NON_NULL(FileName2);
|
|
|
|
if (FileHandle1 != NULL) {
|
|
gEfiShellProtocol->CloseFile(FileHandle1);
|
|
}
|
|
if (FileHandle2 != NULL) {
|
|
gEfiShellProtocol->CloseFile(FileHandle2);
|
|
}
|
|
|
|
return (ShellStatus);
|
|
}
|