[mirror_edk2.git] / EdkNt32Pkg / Dxe / PlatformBds / Generic / MemoryTest.c

/*++\r
\r
Copyright (c) 2006, Intel Corporation                                                         \r
All rights reserved. This program and the accompanying materials                          \r
are licensed and made available under the terms and conditions of the BSD License         \r
which accompanies this distribution.  The full text of the license may be found at        \r
http://opensource.org/licenses/bsd-license.php                                            \r
                                                                                          \r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     \r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             \r
\r
Module Name:\r
\r
  MemoryTest.c\r
\r
Abstract:\r
\r
  Perform the platform memory test\r
\r
--*/\r
\r
#include "bds.h"\r
#include "BdsPlatform.h"\r
#include "String.h"\r
\r
//\r
// BDS Platform Functions\r
//\r
EFI_STATUS\r
PlatformBdsShowProgress (\r
  IN EFI_GRAPHICS_OUTPUT_BLT_PIXEL TitleForeground,\r
  IN EFI_GRAPHICS_OUTPUT_BLT_PIXEL TitleBackground,\r
  IN CHAR16        *Title,\r
  IN EFI_GRAPHICS_OUTPUT_BLT_PIXEL ProgressColor,\r
  IN UINTN         Progress,\r
  IN UINTN         PreviousValue\r
  )\r
/*++\r
\r
Routine Description:\r
  \r
  Show progress bar with title above it. It only works in UGA mode.\r
\r
Arguments:\r
  \r
  TitleForeground - Foreground color for Title.\r
  TitleBackground - Background color for Title.\r
  Title           - Title above progress bar.\r
  ProgressColor   - Progress bar color.\r
  Progress        - Progress (0-100)\r
\r
Returns: \r
  \r
  EFI_STATUS      - Success update the progress bar\r
\r
--*/\r
{\r
  EFI_STATUS            Status;\r
  EFI_GRAPHICS_OUTPUT_PROTOCOL   *GraphicsOutput;\r
  EFI_UGA_DRAW_PROTOCOL *UgaDraw;\r
  UINT32                SizeOfX;\r
  UINT32                SizeOfY;\r
  UINT32                ColorDepth;\r
  UINT32                RefreshRate;\r
  EFI_GRAPHICS_OUTPUT_BLT_PIXEL  Color;\r
  UINTN                 BlockHeight;\r
  UINTN                 BlockWidth;\r
  UINTN                 BlockNum;\r
  UINTN                 PosX;\r
  UINTN                 PosY;\r
  UINTN                 Index;\r
\r
  if (Progress > 100) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  UgaDraw = NULL;\r
  Status = gBS->HandleProtocol (\r
                  gST->ConsoleOutHandle,\r
                  &gEfiGraphicsOutputProtocolGuid,\r
                  &GraphicsOutput\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    GraphicsOutput = NULL;\r
\r
    Status = gBS->HandleProtocol (\r
                    gST->ConsoleOutHandle,\r
                    &gEfiUgaDrawProtocolGuid,\r
                    &UgaDraw\r
                    );\r
    if (EFI_ERROR (Status)) {\r
      return EFI_UNSUPPORTED;\r
    }\r
  }\r
\r
  if (GraphicsOutput != NULL) {\r
    SizeOfX = GraphicsOutput->Mode->Info->HorizontalResolution;\r
    SizeOfY = GraphicsOutput->Mode->Info->VerticalResolution;\r
  } else {\r
    Status = UgaDraw->GetMode (\r
                        UgaDraw,\r
                        &SizeOfX,\r
                        &SizeOfY,\r
                        &ColorDepth,\r
                        &RefreshRate\r
                        );\r
    if (EFI_ERROR (Status)) {\r
      return EFI_UNSUPPORTED;\r
    }\r
  }\r
\r
  BlockWidth  = SizeOfX / 100;\r
  BlockHeight = SizeOfY / 50;\r
\r
  BlockNum    = Progress;\r
\r
  PosX        = 0;\r
  PosY        = SizeOfY * 48 / 50;\r
\r
  if (BlockNum == 0) {\r
    //\r
    // Clear progress area\r
    //\r
    SetMem (&Color, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0x0);\r
\r
    if (GraphicsOutput != NULL) {\r
      Status = GraphicsOutput->Blt (\r
                          GraphicsOutput,\r
                          &Color,\r
                          EfiBltVideoFill,\r
                          0,\r
                          0,\r
                          0,\r
                          PosY - GLYPH_HEIGHT - 1,\r
                          SizeOfX,\r
                          SizeOfY - (PosY - GLYPH_HEIGHT - 1),\r
                          SizeOfX * sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL)\r
                          );\r
    } else {\r
      Status = UgaDraw->Blt (\r
                          UgaDraw,\r
                          (EFI_UGA_PIXEL *) &Color,\r
                          EfiUgaVideoFill,\r
                          0,\r
                          0,\r
                          0,\r
                          PosY - GLYPH_HEIGHT - 1,\r
                          SizeOfX,\r
                          SizeOfY - (PosY - GLYPH_HEIGHT - 1),\r
                          SizeOfX * sizeof (EFI_UGA_PIXEL)\r
                          );\r
    }\r
  }\r
  //\r
  // Show progress by drawing blocks\r
  //\r
  for (Index = PreviousValue; Index < BlockNum; Index++) {\r
    PosX = Index * BlockWidth;\r
    if (GraphicsOutput != NULL) {\r
      Status = GraphicsOutput->Blt (\r
                          GraphicsOutput,\r
                          &ProgressColor,\r
                          EfiBltVideoFill,\r
                          0,\r
                          0,\r
                          PosX,\r
                          PosY,\r
                          BlockWidth - 1,\r
                          BlockHeight,\r
                          (BlockWidth) * sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL)\r
                          );\r
    } else {\r
      Status = UgaDraw->Blt (\r
                          UgaDraw,\r
                          (EFI_UGA_PIXEL *) &ProgressColor,\r
                          EfiUgaVideoFill,\r
                          0,\r
                          0,\r
                          PosX,\r
                          PosY,\r
                          BlockWidth - 1,\r
                          BlockHeight,\r
                          (BlockWidth) * sizeof (EFI_UGA_PIXEL)\r
                          );\r
    }\r
  }\r
\r
  PrintXY (\r
    (SizeOfX - StrLen (Title) * GLYPH_WIDTH) / 2,\r
    PosY - GLYPH_HEIGHT - 1,\r
    &TitleForeground,\r
    &TitleBackground,\r
    Title\r
    );\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFI_STATUS\r
BdsMemoryTest (\r
  IN EXTENDMEM_COVERAGE_LEVEL Level\r
  )\r
/*++\r
\r
Routine Description:\r
  \r
  Perform the memory test base on the memory test intensive level, \r
  and update the memory resource.\r
\r
Arguments:\r
  \r
  Level  - The memory test intensive level.\r
\r
Returns: \r
  \r
  EFI_STATUS      - Success test all the system memory and update\r
                    the memory resource\r
                    \r
--*/\r
{\r
  EFI_STATUS                        Status;\r
  EFI_STATUS                        InitStatus;\r
  EFI_STATUS                        KeyStatus;\r
  EFI_STATUS                        ReturnStatus;\r
  BOOLEAN                           RequireSoftECCInit;\r
  EFI_GENERIC_MEMORY_TEST_PROTOCOL  *GenMemoryTest;\r
  UINT64                            TestedMemorySize;\r
  UINT64                            TotalMemorySize;\r
  UINTN                             TestPercent;\r
  UINT64                            PreviousValue;\r
  BOOLEAN                           ErrorOut;\r
  BOOLEAN                           TestAbort;\r
  EFI_INPUT_KEY                     Key;\r
  CHAR16                            StrPercent[16];\r
  CHAR16                            *StrTotalMemory;\r
  CHAR16                            *Pos;\r
  CHAR16                            *TmpStr;\r
  EFI_GRAPHICS_OUTPUT_BLT_PIXEL     Foreground;\r
  EFI_GRAPHICS_OUTPUT_BLT_PIXEL     Background;\r
  EFI_GRAPHICS_OUTPUT_BLT_PIXEL     Color;\r
  UINT8                             Value;\r
  UINTN                             DataSize;\r
  UINT32                            Attributes;\r
\r
  ReturnStatus = EFI_SUCCESS;\r
  ZeroMem (&Key, sizeof (EFI_INPUT_KEY));\r
\r
  Pos = AllocatePool (128);\r
\r
  if (Pos == NULL) {\r
    return ReturnStatus;\r
  }\r
\r
  StrTotalMemory    = Pos;\r
\r
  TestedMemorySize  = 0;\r
  TotalMemorySize   = 0;\r
  PreviousValue     = 0;\r
  ErrorOut          = FALSE;\r
  TestAbort         = FALSE;\r
\r
  SetMem (&Foreground, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0xff);\r
  SetMem (&Background, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0x0);\r
  SetMem (&Color, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0xff);\r
\r
  RequireSoftECCInit = FALSE;\r
\r
  gST->ConOut->ClearScreen (gST->ConOut);\r
  gST->ConOut->SetAttribute (gST->ConOut, EFI_YELLOW | EFI_BRIGHT);\r
  gST->ConOut->EnableCursor (gST->ConOut, FALSE);\r
\r
  Status = gBS->LocateProtocol (\r
                  &gEfiGenericMemTestProtocolGuid,\r
                  NULL,\r
                  &GenMemoryTest\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    gBS->FreePool (Pos);\r
    return EFI_SUCCESS;\r
  }\r
\r
  InitStatus = GenMemoryTest->MemoryTestInit (\r
                                GenMemoryTest,\r
                                Level,\r
                                &RequireSoftECCInit\r
                                );\r
  if (InitStatus == EFI_NO_MEDIA) {\r
    //\r
    // The PEI codes also have the relevant memory test code to check the memory,\r
    // it can select to test some range of the memory or all of them. If PEI code\r
    // checks all the memory, this BDS memory test will has no not-test memory to\r
    // do the test, and then the status of EFI_NO_MEDIA will be returned by\r
    // "MemoryTestInit". So it does not need to test memory again, just return.\r
    //\r
    gBS->FreePool (Pos);\r
    return EFI_SUCCESS;\r
  }\r
\r
  gST->ConOut->SetCursorPosition (gST->ConOut, 0, 2);\r
  TmpStr = GetStringById (STRING_TOKEN (STR_ESC_TO_SKIP_MEM_TEST));\r
\r
  if (TmpStr != NULL) {\r
    gST->ConOut->OutputString (gST->ConOut, TmpStr);\r
    gBS->FreePool (TmpStr);\r
  }\r
\r
  do {\r
    Status = GenMemoryTest->PerformMemoryTest (\r
                              GenMemoryTest,\r
                              &TestedMemorySize,\r
                              &TotalMemorySize,\r
                              &ErrorOut,\r
                              TestAbort\r
                              );\r
    if (ErrorOut && (Status == EFI_DEVICE_ERROR)) {\r
      TmpStr = GetStringById (STRING_TOKEN (STR_SYSTEM_MEM_ERROR));\r
      if (TmpStr != NULL) {\r
        PrintXY (10, 10, NULL, NULL, TmpStr);\r
        gST->ConOut->SetCursorPosition (gST->ConOut, 0, 4);\r
        gST->ConOut->OutputString (gST->ConOut, TmpStr);\r
        gBS->FreePool (TmpStr);\r
      }\r
\r
      ASSERT (0);\r
    }\r
\r
    TestPercent = (UINTN) DivU64x32 (\r
                            DivU64x32 (MultU64x32 (TestedMemorySize, 100), 16),\r
                            (UINTN)DivU64x32 (TotalMemorySize, 16)\r
                            );\r
    if (TestPercent != PreviousValue) {\r
      UnicodeValueToString (StrPercent, 0, TestPercent, 0);\r
      gST->ConOut->SetCursorPosition (gST->ConOut, 0, 0);\r
      TmpStr = GetStringById (STRING_TOKEN (STR_MEMORY_TEST_PERCENT));\r
      if (TmpStr != NULL) {\r
        BdsLibOutputStrings (gST->ConOut, StrPercent, TmpStr, NULL);\r
        gBS->FreePool (TmpStr);\r
      }\r
\r
      TmpStr = GetStringById (STRING_TOKEN (STR_PERFORM_MEM_TEST));\r
      if (TmpStr != NULL) {\r
        PlatformBdsShowProgress (\r
          Foreground,\r
          Background,\r
          TmpStr,\r
          Color,\r
          TestPercent,\r
          (UINTN) PreviousValue\r
          );\r
        gBS->FreePool (TmpStr);\r
      }\r
    }\r
\r
    PreviousValue = TestPercent;\r
\r
    KeyStatus     = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);\r
    if (Key.ScanCode == SCAN_ESC) {\r
      if (!RequireSoftECCInit) {\r
        TmpStr = GetStringById (STRING_TOKEN (STR_PERFORM_MEM_TEST));\r
        if (TmpStr != NULL) {\r
          PlatformBdsShowProgress (\r
            Foreground,\r
            Background,\r
            TmpStr,\r
            Color,\r
            100,\r
            (UINTN) PreviousValue\r
            );\r
          gBS->FreePool (TmpStr);\r
        }\r
\r
        gST->ConOut->SetCursorPosition (gST->ConOut, 0, 0);\r
        gST->ConOut->OutputString (gST->ConOut, L"100");\r
        Status = GenMemoryTest->Finished (GenMemoryTest);\r
        goto Done;\r
      }\r
\r
      TestAbort = TRUE;\r
    }\r
  } while (Status != EFI_NOT_FOUND);\r
\r
  Status = GenMemoryTest->Finished (GenMemoryTest);\r
\r
Done:\r
  UnicodeValueToString (StrTotalMemory, COMMA_TYPE, (UINTN) TotalMemorySize, 0);\r
  if (StrTotalMemory[0] == L',') {\r
    StrTotalMemory++;\r
  }\r
\r
  TmpStr = GetStringById (STRING_TOKEN (STR_MEM_TEST_COMPLETED));\r
  if (TmpStr != NULL) {\r
    StrCat (StrTotalMemory, TmpStr);\r
    gBS->FreePool (TmpStr);\r
  }\r
\r
  gST->ConOut->ClearScreen (gST->ConOut);\r
  gST->ConOut->SetAttribute (gST->ConOut, EFI_YELLOW | EFI_BRIGHT);\r
  gST->ConOut->EnableCursor (gST->ConOut, FALSE);\r
  gST->ConOut->OutputString (gST->ConOut, StrTotalMemory);\r
  PlatformBdsShowProgress (\r
    Foreground,\r
    Background,\r
    StrTotalMemory,\r
    Color,\r
    100,\r
    (UINTN) PreviousValue\r
    );\r
\r
  gBS->FreePool (Pos);\r
\r
  DataSize = sizeof (Value);\r
  Status = gRT->GetVariable (\r
                  L"BootState",\r
                  &gEfiBootStateGuid,\r
                  &Attributes,\r
                  &DataSize,\r
                  &Value\r
                  );\r
\r
  if (EFI_ERROR (Status)) {\r
    Value = 1;\r
    gRT->SetVariable (\r
          L"BootState",\r
          &gEfiBootStateGuid,\r
          EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
          sizeof (Value),\r
          &Value\r
          );\r
  }\r
\r
  return ReturnStatus;\r
}\r
Commit	Line	Data
878ddf1f	1	/*++\r
	2	\r
	3	Copyright (c) 2006, Intel Corporation \r
	4	All rights reserved. This program and the accompanying materials \r
	5	are licensed and made available under the terms and conditions of the BSD License \r
	6	which accompanies this distribution. The full text of the license may be found at \r
	7	http://opensource.org/licenses/bsd-license.php \r
	8	\r
	9	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, \r
	10	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. \r
	11	\r
	12	Module Name:\r
	13	\r
	14	MemoryTest.c\r
	15	\r
	16	Abstract:\r
	17	\r
	18	Perform the platform memory test\r
	19	\r
	20	--*/\r
	21	\r
	22	#include "bds.h"\r
	23	#include "BdsPlatform.h"\r
	24	#include "String.h"\r
	25	\r
	26	//\r
	27	// BDS Platform Functions\r
	28	//\r
	29	EFI_STATUS\r
	30	PlatformBdsShowProgress (\r
406adcd1	31	IN EFI_GRAPHICS_OUTPUT_BLT_PIXEL TitleForeground,\r
406adcd1	32	IN EFI_GRAPHICS_OUTPUT_BLT_PIXEL TitleBackground,\r
878ddf1f	33	IN CHAR16 *Title,\r
406adcd1	34	IN EFI_GRAPHICS_OUTPUT_BLT_PIXEL ProgressColor,\r
878ddf1f	35	IN UINTN Progress,\r
	36	IN UINTN PreviousValue\r
	37	)\r
	38	/*++\r
	39	\r
	40	Routine Description:\r
	41	\r
	42	Show progress bar with title above it. It only works in UGA mode.\r
	43	\r
	44	Arguments:\r
	45	\r
	46	TitleForeground - Foreground color for Title.\r
	47	TitleBackground - Background color for Title.\r
	48	Title - Title above progress bar.\r
	49	ProgressColor - Progress bar color.\r
	50	Progress - Progress (0-100)\r
	51	\r
	52	Returns: \r
	53	\r
	54	EFI_STATUS - Success update the progress bar\r
	55	\r
	56	--*/\r
	57	{\r
	58	EFI_STATUS Status;\r
406adcd1	59	EFI_GRAPHICS_OUTPUT_PROTOCOL *GraphicsOutput;\r
878ddf1f	60	EFI_UGA_DRAW_PROTOCOL *UgaDraw;\r
	61	UINT32 SizeOfX;\r
	62	UINT32 SizeOfY;\r
	63	UINT32 ColorDepth;\r
	64	UINT32 RefreshRate;\r
406adcd1	65	EFI_GRAPHICS_OUTPUT_BLT_PIXEL Color;\r
878ddf1f	66	UINTN BlockHeight;\r
	67	UINTN BlockWidth;\r
	68	UINTN BlockNum;\r
	69	UINTN PosX;\r
	70	UINTN PosY;\r
	71	UINTN Index;\r
	72	\r
	73	if (Progress > 100) {\r
	74	return EFI_INVALID_PARAMETER;\r
	75	}\r
	76	\r
406adcd1	77	UgaDraw = NULL;\r
878ddf1f	78	Status = gBS->HandleProtocol (\r
878ddf1f	79	gST->ConsoleOutHandle,\r
406adcd1	80	&gEfiGraphicsOutputProtocolGuid,\r
406adcd1	81	&GraphicsOutput\r
878ddf1f	82	);\r
878ddf1f	83	if (EFI_ERROR (Status)) {\r
406adcd1	84	GraphicsOutput = NULL;\r
	85	\r
	86	Status = gBS->HandleProtocol (\r
	87	gST->ConsoleOutHandle,\r
	88	&gEfiUgaDrawProtocolGuid,\r
	89	&UgaDraw\r
	90	);\r
	91	if (EFI_ERROR (Status)) {\r
	92	return EFI_UNSUPPORTED;\r
	93	}\r
878ddf1f	94	}\r
878ddf1f	95	\r
406adcd1	96	if (GraphicsOutput != NULL) {\r
	97	SizeOfX = GraphicsOutput->Mode->Info->HorizontalResolution;\r
	98	SizeOfY = GraphicsOutput->Mode->Info->VerticalResolution;\r
	99	} else {\r
	100	Status = UgaDraw->GetMode (\r
	101	UgaDraw,\r
	102	&SizeOfX,\r
	103	&SizeOfY,\r
	104	&ColorDepth,\r
	105	&RefreshRate\r
	106	);\r
	107	if (EFI_ERROR (Status)) {\r
	108	return EFI_UNSUPPORTED;\r
	109	}\r
878ddf1f	110	}\r
	111	\r
	112	BlockWidth = SizeOfX / 100;\r
	113	BlockHeight = SizeOfY / 50;\r
	114	\r
	115	BlockNum = Progress;\r
	116	\r
	117	PosX = 0;\r
	118	PosY = SizeOfY * 48 / 50;\r
	119	\r
	120	if (BlockNum == 0) {\r
	121	//\r
	122	// Clear progress area\r
	123	//\r
406adcd1	124	SetMem (&Color, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0x0);\r
	125	\r
	126	if (GraphicsOutput != NULL) {\r
	127	Status = GraphicsOutput->Blt (\r
	128	GraphicsOutput,\r
	129	&Color,\r
	130	EfiBltVideoFill,\r
	131	0,\r
	132	0,\r
	133	0,\r
	134	PosY - GLYPH_HEIGHT - 1,\r
	135	SizeOfX,\r
	136	SizeOfY - (PosY - GLYPH_HEIGHT - 1),\r
	137	SizeOfX * sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL)\r
	138	);\r
	139	} else {\r
	140	Status = UgaDraw->Blt (\r
	141	UgaDraw,\r
	142	(EFI_UGA_PIXEL *) &Color,\r
	143	EfiUgaVideoFill,\r
	144	0,\r
	145	0,\r
	146	0,\r
	147	PosY - GLYPH_HEIGHT - 1,\r
	148	SizeOfX,\r
	149	SizeOfY - (PosY - GLYPH_HEIGHT - 1),\r
	150	SizeOfX * sizeof (EFI_UGA_PIXEL)\r
	151	);\r
	152	}\r
878ddf1f	153	}\r
	154	//\r
	155	// Show progress by drawing blocks\r
	156	//\r
	157	for (Index = PreviousValue; Index < BlockNum; Index++) {\r
	158	PosX = Index * BlockWidth;\r
406adcd1	159	if (GraphicsOutput != NULL) {\r
	160	Status = GraphicsOutput->Blt (\r
	161	GraphicsOutput,\r
	162	&ProgressColor,\r
	163	EfiBltVideoFill,\r
	164	0,\r
	165	0,\r
	166	PosX,\r
	167	PosY,\r
	168	BlockWidth - 1,\r
	169	BlockHeight,\r
	170	(BlockWidth) * sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL)\r
	171	);\r
	172	} else {\r
	173	Status = UgaDraw->Blt (\r
	174	UgaDraw,\r
	175	(EFI_UGA_PIXEL *) &ProgressColor,\r
	176	EfiUgaVideoFill,\r
	177	0,\r
	178	0,\r
	179	PosX,\r
	180	PosY,\r
	181	BlockWidth - 1,\r
	182	BlockHeight,\r
	183	(BlockWidth) * sizeof (EFI_UGA_PIXEL)\r
	184	);\r
	185	}\r
878ddf1f	186	}\r
	187	\r
	188	PrintXY (\r
	189	(SizeOfX - StrLen (Title) * GLYPH_WIDTH) / 2,\r
	190	PosY - GLYPH_HEIGHT - 1,\r
	191	&TitleForeground,\r
	192	&TitleBackground,\r
	193	Title\r
	194	);\r
	195	\r
	196	return EFI_SUCCESS;\r
	197	}\r
	198	\r
	199	EFI_STATUS\r
	200	BdsMemoryTest (\r
	201	IN EXTENDMEM_COVERAGE_LEVEL Level\r
	202	)\r
	203	/*++\r
	204	\r
	205	Routine Description:\r
	206	\r
	207	Perform the memory test base on the memory test intensive level, \r
	208	and update the memory resource.\r
	209	\r
	210	Arguments:\r
	211	\r
	212	Level - The memory test intensive level.\r
	213	\r
	214	Returns: \r
	215	\r
	216	EFI_STATUS - Success test all the system memory and update\r
	217	the memory resource\r
	218	\r
	219	--*/\r
	220	{\r
	221	EFI_STATUS Status;\r
	222	EFI_STATUS InitStatus;\r
	223	EFI_STATUS KeyStatus;\r
	224	EFI_STATUS ReturnStatus;\r
	225	BOOLEAN RequireSoftECCInit;\r
	226	EFI_GENERIC_MEMORY_TEST_PROTOCOL *GenMemoryTest;\r
	227	UINT64 TestedMemorySize;\r
	228	UINT64 TotalMemorySize;\r
	229	UINTN TestPercent;\r
	230	UINT64 PreviousValue;\r
	231	BOOLEAN ErrorOut;\r
	232	BOOLEAN TestAbort;\r
	233	EFI_INPUT_KEY Key;\r
	234	CHAR16 StrPercent[16];\r
	235	CHAR16 *StrTotalMemory;\r
	236	CHAR16 *Pos;\r
	237	CHAR16 *TmpStr;\r
406adcd1	238	EFI_GRAPHICS_OUTPUT_BLT_PIXEL Foreground;\r
	239	EFI_GRAPHICS_OUTPUT_BLT_PIXEL Background;\r
	240	EFI_GRAPHICS_OUTPUT_BLT_PIXEL Color;\r
878ddf1f	241	UINT8 Value;\r
	242	UINTN DataSize;\r
	243	UINT32 Attributes;\r
	244	\r
	245	ReturnStatus = EFI_SUCCESS;\r
	246	ZeroMem (&Key, sizeof (EFI_INPUT_KEY));\r
	247	\r
	248	Pos = AllocatePool (128);\r
	249	\r
	250	if (Pos == NULL) {\r
	251	return ReturnStatus;\r
	252	}\r
	253	\r
	254	StrTotalMemory = Pos;\r
	255	\r
	256	TestedMemorySize = 0;\r
	257	TotalMemorySize = 0;\r
	258	PreviousValue = 0;\r
	259	ErrorOut = FALSE;\r
	260	TestAbort = FALSE;\r
	261	\r
406adcd1	262	SetMem (&Foreground, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0xff);\r
	263	SetMem (&Background, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0x0);\r
	264	SetMem (&Color, sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL), 0xff);\r
878ddf1f	265	\r
	266	RequireSoftECCInit = FALSE;\r
	267	\r
	268	gST->ConOut->ClearScreen (gST->ConOut);\r
	269	gST->ConOut->SetAttribute (gST->ConOut, EFI_YELLOW \| EFI_BRIGHT);\r
	270	gST->ConOut->EnableCursor (gST->ConOut, FALSE);\r
	271	\r
	272	Status = gBS->LocateProtocol (\r
	273	&gEfiGenericMemTestProtocolGuid,\r
	274	NULL,\r
	275	&GenMemoryTest\r
	276	);\r
	277	if (EFI_ERROR (Status)) {\r
	278	gBS->FreePool (Pos);\r
	279	return EFI_SUCCESS;\r
	280	}\r
	281	\r
	282	InitStatus = GenMemoryTest->MemoryTestInit (\r
	283	GenMemoryTest,\r
	284	Level,\r
	285	&RequireSoftECCInit\r
	286	);\r
	287	if (InitStatus == EFI_NO_MEDIA) {\r
	288	//\r
	289	// The PEI codes also have the relevant memory test code to check the memory,\r
	290	// it can select to test some range of the memory or all of them. If PEI code\r
	291	// checks all the memory, this BDS memory test will has no not-test memory to\r
	292	// do the test, and then the status of EFI_NO_MEDIA will be returned by\r
	293	// "MemoryTestInit". So it does not need to test memory again, just return.\r
	294	//\r
	295	gBS->FreePool (Pos);\r
	296	return EFI_SUCCESS;\r
	297	}\r
	298	\r
	299	gST->ConOut->SetCursorPosition (gST->ConOut, 0, 2);\r
	300	TmpStr = GetStringById (STRING_TOKEN (STR_ESC_TO_SKIP_MEM_TEST));\r
	301	\r
	302	if (TmpStr != NULL) {\r
	303	gST->ConOut->OutputString (gST->ConOut, TmpStr);\r
	304	gBS->FreePool (TmpStr);\r
	305	}\r
	306	\r
	307	do {\r
	308	Status = GenMemoryTest->PerformMemoryTest (\r
	309	GenMemoryTest,\r
	310	&TestedMemorySize,\r
	311	&TotalMemorySize,\r
	312	&ErrorOut,\r
	313	TestAbort\r
	314	);\r
	315	if (ErrorOut && (Status == EFI_DEVICE_ERROR)) {\r
	316	TmpStr = GetStringById (STRING_TOKEN (STR_SYSTEM_MEM_ERROR));\r
	317	if (TmpStr != NULL) {\r
	318	PrintXY (10, 10, NULL, NULL, TmpStr);\r
	319	gST->ConOut->SetCursorPosition (gST->ConOut, 0, 4);\r
	320	gST->ConOut->OutputString (gST->ConOut, TmpStr);\r
	321	gBS->FreePool (TmpStr);\r
	322	}\r
	323	\r
	324	ASSERT (0);\r
	325	}\r
	326	\r
	327	TestPercent = (UINTN) DivU64x32 (\r
	328	DivU64x32 (MultU64x32 (TestedMemorySize, 100), 16),\r
329	(UINTN)DivU64x32 (TotalMemorySize, 16)\r
330	);\r
331	if (TestPercent != PreviousValue) {\r
332	UnicodeValueToString (StrPercent, 0, TestPercent, 0);\r
333	gST->ConOut->SetCursorPosition (gST->ConOut, 0, 0);\r
334	TmpStr = GetStringById (STRING_TOKEN (STR_MEMORY_TEST_PERCENT));\r
335	if (TmpStr != NULL) {\r
336	BdsLibOutputStrings (gST->ConOut, StrPercent, TmpStr, NULL);\r
337	gBS->FreePool (TmpStr);\r
338	}\r
339	\r
340	TmpStr = GetStringById (STRING_TOKEN (STR_PERFORM_MEM_TEST));\r
341	if (TmpStr != NULL) {\r
342	PlatformBdsShowProgress (\r
343	Foreground,\r
344	Background,\r
345	TmpStr,\r
346	Color,\r
347	TestPercent,\r
348	(UINTN) PreviousValue\r
349	);\r
350	gBS->FreePool (TmpStr);\r
351	}\r
352	}\r
353	\r
354	PreviousValue = TestPercent;\r
355	\r
356	KeyStatus = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);\r
357	if (Key.ScanCode == SCAN_ESC) {\r
358	if (!RequireSoftECCInit) {\r
359	TmpStr = GetStringById (STRING_TOKEN (STR_PERFORM_MEM_TEST));\r
360	if (TmpStr != NULL) {\r
361	PlatformBdsShowProgress (\r
362	Foreground,\r
363	Background,\r
364	TmpStr,\r
365	Color,\r
366	100,\r
367	(UINTN) PreviousValue\r
368	);\r
369	gBS->FreePool (TmpStr);\r
370	}\r
371	\r
372	gST->ConOut->SetCursorPosition (gST->ConOut, 0, 0);\r
373	gST->ConOut->OutputString (gST->ConOut, L"100");\r
374	Status = GenMemoryTest->Finished (GenMemoryTest);\r
375	goto Done;\r
376	}\r
377	\r
378	TestAbort = TRUE;\r
379	}\r
380	} while (Status != EFI_NOT_FOUND);\r
381	\r
382	Status = GenMemoryTest->Finished (GenMemoryTest);\r
383	\r
384	Done:\r
385	UnicodeValueToString (StrTotalMemory, COMMA_TYPE, (UINTN) TotalMemorySize, 0);\r
386	if (StrTotalMemory[0] == L',') {\r
387	StrTotalMemory++;\r
388	}\r
389	\r
390	TmpStr = GetStringById (STRING_TOKEN (STR_MEM_TEST_COMPLETED));\r
391	if (TmpStr != NULL) {\r
392	StrCat (StrTotalMemory, TmpStr);\r
393	gBS->FreePool (TmpStr);\r
394	}\r
395	\r
396	gST->ConOut->ClearScreen (gST->ConOut);\r
397	gST->ConOut->SetAttribute (gST->ConOut, EFI_YELLOW \| EFI_BRIGHT);\r
398	gST->ConOut->EnableCursor (gST->ConOut, FALSE);\r
399	gST->ConOut->OutputString (gST->ConOut, StrTotalMemory);\r
400	PlatformBdsShowProgress (\r
401	Foreground,\r
402	Background,\r
403	StrTotalMemory,\r
404	Color,\r
405	100,\r
406	(UINTN) PreviousValue\r
407	);\r
408	\r
409	gBS->FreePool (Pos);\r
410	\r
411	DataSize = sizeof (Value);\r
412	Status = gRT->GetVariable (\r
413	L"BootState",\r
414	&gEfiBootStateGuid,\r
415	&Attributes,\r
416	&DataSize,\r
417	&Value\r
418	);\r
419	\r
420	if (EFI_ERROR (Status)) {\r
421	Value = 1;\r
422	gRT->SetVariable (\r
423	L"BootState",\r
424	&gEfiBootStateGuid,\r
425	EFI_VARIABLE_NON_VOLATILE \| EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
426	sizeof (Value),\r
427	&Value\r
428	);\r
429	}\r
430	\r
431	return ReturnStatus;\r
432	}\r