The comp command was potentially displaying the incorrect offset into the file due to comparing 3 extra bytes for difference groups. Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Jaben Carsey <Jaben.carsey@intel.com> Reviewed-by: Tapan Shah <tapandshah@hp.com> Reviewed-by: Erik Bjorge <erik.c.bjorge@intel.com> git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@15558 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			277 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			277 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
  Main file for Comp shell Debug1 function.
 | 
						|
 | 
						|
  Copyright (c) 2010 - 2014, 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;
 | 
						|
  UINTN               ErrorAddress;
 | 
						|
 | 
						|
  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) {
 | 
						|
            ErrorAddress = LoopVar;
 | 
						|
            ADF_File11 = 0;
 | 
						|
            ADF_File12 = 0;
 | 
						|
            ADF_File13 = 0;
 | 
						|
            ADF_File21 = 0;
 | 
						|
            ADF_File22 = 0;
 | 
						|
            ADF_File23 = 0;
 | 
						|
 | 
						|
            //
 | 
						|
            // Now check the next 3 bytes if possible.  This will make output
 | 
						|
            // cleaner when there are a sequence of differences.
 | 
						|
            //
 | 
						|
            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);
 | 
						|
                }
 | 
						|
              }
 | 
						|
            }
 | 
						|
 | 
						|
            //
 | 
						|
            // Print out based on highest of the 4 bytes that are different.
 | 
						|
            //
 | 
						|
            if (ADF_File13 != ADF_File23) {
 | 
						|
              ShellPrintHiiEx(
 | 
						|
                -1,
 | 
						|
                -1,
 | 
						|
                NULL,
 | 
						|
                STRING_TOKEN (STR_COMP_SPOT_FAIL4),
 | 
						|
                gShellDebug1HiiHandle,
 | 
						|
                ++ErrorCount,
 | 
						|
                FileName1,
 | 
						|
                ErrorAddress,
 | 
						|
                DataFromFile1, ADF_File11, ADF_File12, ADF_File13,
 | 
						|
                DataFromFile1, ADF_File11, ADF_File12, ADF_File13,
 | 
						|
                FileName2,
 | 
						|
                ErrorAddress,
 | 
						|
                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,
 | 
						|
                ErrorAddress,
 | 
						|
                DataFromFile1, ADF_File11, ADF_File12,
 | 
						|
                DataFromFile1, ADF_File11, ADF_File12,
 | 
						|
                FileName2,
 | 
						|
                ErrorAddress,
 | 
						|
                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,
 | 
						|
                ErrorAddress,
 | 
						|
                DataFromFile1, ADF_File11,
 | 
						|
                DataFromFile1, ADF_File11,
 | 
						|
                FileName2,
 | 
						|
                ErrorAddress,
 | 
						|
                DataFromFile2, ADF_File21,
 | 
						|
                DataFromFile2, ADF_File21
 | 
						|
               );
 | 
						|
            } else {
 | 
						|
              ShellPrintHiiEx(
 | 
						|
                -1,
 | 
						|
                -1,
 | 
						|
                NULL,
 | 
						|
                STRING_TOKEN (STR_COMP_SPOT_FAIL1),
 | 
						|
                gShellDebug1HiiHandle,
 | 
						|
                ++ErrorCount,
 | 
						|
                FileName1,
 | 
						|
                ErrorAddress,
 | 
						|
                DataFromFile1,
 | 
						|
                DataFromFile1,
 | 
						|
                FileName2,
 | 
						|
                ErrorAddress,
 | 
						|
                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);
 | 
						|
}
 |