UefiCpuPkg: Move AsmRelocateApLoopStart from Mpfuncs.nasm to AmdSev.nasm

[mirror_edk2.git] / EmbeddedPkg / Universal / MmcDxe / Diagnostics.c

/** @file\r
  Diagnostics Protocol implementation for the MMC DXE driver\r
\r
  Copyright (c) 2011-2020, ARM Limited. All rights reserved.\r
\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include <Uefi.h>\r
#include <Library/DebugLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/BaseLib.h>\r
\r
#include "Mmc.h"\r
\r
#define DIAGNOSTIC_LOGBUFFER_MAXCHAR  1024\r
\r
CHAR16  *mLogBuffer    = NULL;\r
UINTN   mLogRemainChar = 0;\r
\r
CHAR16 *\r
DiagnosticInitLog (\r
  UINTN  MaxBufferChar\r
  )\r
{\r
  mLogRemainChar = MaxBufferChar;\r
  mLogBuffer     = AllocatePool ((UINTN)MaxBufferChar * sizeof (CHAR16));\r
  return mLogBuffer;\r
}\r
\r
UINTN\r
DiagnosticLog (\r
  CONST CHAR16  *Str\r
  )\r
{\r
  UINTN  len = StrLen (Str);\r
\r
  if (len < mLogRemainChar) {\r
    StrCpyS (mLogBuffer, mLogRemainChar, Str);\r
    mLogRemainChar -= len;\r
    mLogBuffer     += len;\r
    return len;\r
  } else {\r
    return 0;\r
  }\r
}\r
\r
VOID\r
GenerateRandomBuffer (\r
  VOID   *Buffer,\r
  UINTN  BufferSize\r
  )\r
{\r
  UINT64  i;\r
  UINT64  *Buffer64 = (UINT64 *)Buffer;\r
\r
  for (i = 0; i < (BufferSize >> 3); i++) {\r
    *Buffer64 = i | LShiftU64 (~i, 32);\r
    Buffer64++;\r
  }\r
}\r
\r
BOOLEAN\r
CompareBuffer (\r
  VOID   *BufferA,\r
  VOID   *BufferB,\r
  UINTN  BufferSize\r
  )\r
{\r
  UINTN   i;\r
  UINT64  *BufferA64 = (UINT64 *)BufferA;\r
  UINT64  *BufferB64 = (UINT64 *)BufferB;\r
\r
  for (i = 0; i < (BufferSize >> 3); i++) {\r
    if (*BufferA64 != *BufferB64) {\r
      DEBUG ((DEBUG_ERROR, "CompareBuffer: Error at %i", i));\r
      DEBUG ((DEBUG_ERROR, "(0x%lX) != (0x%lX)\n", *BufferA64, *BufferB64));\r
      return FALSE;\r
    }\r
\r
    BufferA64++;\r
    BufferB64++;\r
  }\r
\r
  return TRUE;\r
}\r
\r
EFI_STATUS\r
MmcReadWriteDataTest (\r
  MMC_HOST_INSTANCE  *MmcHostInstance,\r
  EFI_LBA            Lba,\r
  UINTN              BufferSize\r
  )\r
{\r
  VOID        *BackBuffer;\r
  VOID        *WriteBuffer;\r
  VOID        *ReadBuffer;\r
  EFI_STATUS  Status;\r
\r
  // Check if a Media is Present\r
  if (!MmcHostInstance->BlockIo.Media->MediaPresent) {\r
    DiagnosticLog (L"ERROR: No Media Present\n");\r
    return EFI_NO_MEDIA;\r
  }\r
\r
  if (MmcHostInstance->State != MmcTransferState) {\r
    DiagnosticLog (L"ERROR: Not ready for Transfer state\n");\r
    return EFI_NOT_READY;\r
  }\r
\r
  BackBuffer  = AllocatePool (BufferSize);\r
  WriteBuffer = AllocatePool (BufferSize);\r
  ReadBuffer  = AllocatePool (BufferSize);\r
\r
  // Read (and save) buffer at a specific location\r
  Status = MmcReadBlocks (&(MmcHostInstance->BlockIo), MmcHostInstance->BlockIo.Media->MediaId, Lba, BufferSize, BackBuffer);\r
  if (Status != EFI_SUCCESS) {\r
    DiagnosticLog (L"ERROR: Fail to Read Block (1)\n");\r
    return Status;\r
  }\r
\r
  // Write buffer at the same location\r
  GenerateRandomBuffer (WriteBuffer, BufferSize);\r
  Status = MmcWriteBlocks (&(MmcHostInstance->BlockIo), MmcHostInstance->BlockIo.Media->MediaId, Lba, BufferSize, WriteBuffer);\r
  if (Status != EFI_SUCCESS) {\r
    DiagnosticLog (L"ERROR: Fail to Write Block (1)\n");\r
    return Status;\r
  }\r
\r
  // Read the buffer at the same location\r
  Status = MmcReadBlocks (&(MmcHostInstance->BlockIo), MmcHostInstance->BlockIo.Media->MediaId, Lba, BufferSize, ReadBuffer);\r
  if (Status != EFI_SUCCESS) {\r
    DiagnosticLog (L"ERROR: Fail to Read Block (2)\n");\r
    return Status;\r
  }\r
\r
  // Check that is conform\r
  if (!CompareBuffer (ReadBuffer, WriteBuffer, BufferSize)) {\r
    DiagnosticLog (L"ERROR: Fail to Read/Write Block (1)\n");\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  // Restore content at the original location\r
  Status = MmcWriteBlocks (&(MmcHostInstance->BlockIo), MmcHostInstance->BlockIo.Media->MediaId, Lba, BufferSize, BackBuffer);\r
  if (Status != EFI_SUCCESS) {\r
    DiagnosticLog (L"ERROR: Fail to Write Block (2)\n");\r
    return Status;\r
  }\r
\r
  // Read the restored content\r
  Status = MmcReadBlocks (&(MmcHostInstance->BlockIo), MmcHostInstance->BlockIo.Media->MediaId, Lba, BufferSize, ReadBuffer);\r
  if (Status != EFI_SUCCESS) {\r
    DiagnosticLog (L"ERROR: Fail to Read Block (3)\n");\r
    return Status;\r
  }\r
\r
  // Check the content is correct\r
  if (!CompareBuffer (ReadBuffer, BackBuffer, BufferSize)) {\r
    DiagnosticLog (L"ERROR: Fail to Read/Write Block (2)\n");\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
EFIAPI\r
MmcDriverDiagnosticsRunDiagnostics (\r
  IN  EFI_DRIVER_DIAGNOSTICS_PROTOCOL  *This,\r
  IN  EFI_HANDLE                       ControllerHandle,\r
  IN  EFI_HANDLE                       ChildHandle  OPTIONAL,\r
  IN  EFI_DRIVER_DIAGNOSTIC_TYPE       DiagnosticType,\r
  IN  CHAR8                            *Language,\r
  OUT EFI_GUID                         **ErrorType,\r
  OUT UINTN                            *BufferSize,\r
  OUT CHAR16                           **Buffer\r
  )\r
{\r
  LIST_ENTRY         *CurrentLink;\r
  MMC_HOST_INSTANCE  *MmcHostInstance;\r
  EFI_STATUS         Status;\r
\r
  if ((Language         == NULL) ||\r
      (ErrorType        == NULL) ||\r
      (Buffer           == NULL) ||\r
      (ControllerHandle == NULL) ||\r
      (BufferSize       == NULL))\r
  {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  // Check Language is supported (i.e. is "en-*" - only English is supported)\r
  if (AsciiStrnCmp (Language, "en", 2) != 0) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  Status      = EFI_SUCCESS;\r
  *ErrorType  = NULL;\r
  *BufferSize = DIAGNOSTIC_LOGBUFFER_MAXCHAR;\r
  *Buffer     = DiagnosticInitLog (DIAGNOSTIC_LOGBUFFER_MAXCHAR);\r
\r
  DiagnosticLog (L"MMC Driver Diagnostics\n");\r
\r
  // Find the MMC Host instance on which we have been asked to run diagnostics\r
  MmcHostInstance = NULL;\r
  CurrentLink     = mMmcHostPool.ForwardLink;\r
  while (CurrentLink != NULL && CurrentLink != &mMmcHostPool && (Status == EFI_SUCCESS)) {\r
    MmcHostInstance = MMC_HOST_INSTANCE_FROM_LINK (CurrentLink);\r
    ASSERT (MmcHostInstance != NULL);\r
    if (MmcHostInstance->MmcHandle == ControllerHandle) {\r
      break;\r
    }\r
\r
    CurrentLink = CurrentLink->ForwardLink;\r
  }\r
\r
  // If we didn't find the controller, return EFI_UNSUPPORTED\r
  if (  (MmcHostInstance == NULL)\r
     || (MmcHostInstance->MmcHandle != ControllerHandle))\r
  {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  // LBA=1 Size=BlockSize\r
  DiagnosticLog (L"MMC Driver Diagnostics - Test: First Block\n");\r
  Status = MmcReadWriteDataTest (MmcHostInstance, 1, MmcHostInstance->BlockIo.Media->BlockSize);\r
\r
  // LBA=2 Size=BlockSize\r
  DiagnosticLog (L"MMC Driver Diagnostics - Test: Second Block\n");\r
  Status = MmcReadWriteDataTest (MmcHostInstance, 2, MmcHostInstance->BlockIo.Media->BlockSize);\r
\r
  // LBA=10 Size=BlockSize\r
  DiagnosticLog (L"MMC Driver Diagnostics - Test: Any Block\n");\r
  Status = MmcReadWriteDataTest (\r
             MmcHostInstance,\r
             RShiftU64 (MmcHostInstance->BlockIo.Media->LastBlock, 1),\r
             MmcHostInstance->BlockIo.Media->BlockSize\r
             );\r
\r
  // LBA=LastBlock Size=BlockSize\r
  DiagnosticLog (L"MMC Driver Diagnostics - Test: Last Block\n");\r
  Status = MmcReadWriteDataTest (MmcHostInstance, MmcHostInstance->BlockIo.Media->LastBlock, MmcHostInstance->BlockIo.Media->BlockSize);\r
\r
  // LBA=1 Size=2*BlockSize\r
  DiagnosticLog (L"MMC Driver Diagnostics - Test: First Block / 2 BlockSSize\n");\r
  Status = MmcReadWriteDataTest (MmcHostInstance, 1, 2 * MmcHostInstance->BlockIo.Media->BlockSize);\r
\r
  return Status;\r
}\r
\r
//\r
// EFI Driver Diagnostics 2 Protocol\r
//\r
GLOBAL_REMOVE_IF_UNREFERENCED EFI_DRIVER_DIAGNOSTICS2_PROTOCOL  gMmcDriverDiagnostics2 = {\r
  (EFI_DRIVER_DIAGNOSTICS2_RUN_DIAGNOSTICS)MmcDriverDiagnosticsRunDiagnostics,\r
  "en"\r
};\r