Add the following Pcds declaration in EdkModulePkg.spd file. EdkModulePkg PeiVariable...

[mirror_edk2.git] / EdkNt32Pkg / Library / Nt32OemHookStatusCodeLib / Nt32OemHookStatusCodeLib.c

/** @file\r
  OEM hook status code library functions with no library constructor/destructor\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:  Nt32OemHookStatusCodeLib.c\r
\r
**/\r
\r
#include <stdio.h>\r
\r
//\r
// Cache of WinNtThunk protocol \r
//\r
STATIC\r
EFI_WIN_NT_THUNK_PROTOCOL   *mWinNt;\r
\r
//\r
// Cache of standard output handle . \r
//\r
STATIC\r
HANDLE                      mStdOut;\r
\r
/**\r
\r
  Initialize OEM status code device .\r
\r
  @return    Always return EFI_SUCCESS.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
OemHookStatusCodeInitialize (\r
  VOID\r
  )\r
{\r
  PEI_NT_THUNK_PPI  *NtThunkPpi;\r
  EFI_STATUS        Status;\r
\r
  if (FeaturePcdGet (PcdNt32StatusCodeLibUseForPei)) {\r
    //\r
    // Locate NtThunkPpi for retrieving standard output handle\r
    //\r
    Status = PeiServicesLocatePpi (\r
               &gPeiNtThunkPpiGuid,\r
               0,\r
               NULL,\r
               (VOID **) &NtThunkPpi\r
               );\r
\r
    ASSERT_EFI_ERROR (Status);\r
\r
    mWinNt  = (EFI_WIN_NT_THUNK_PROTOCOL *) NtThunkPpi->NtThunk ();\r
  } else {\r
    EFI_HOB_GUID_TYPE        *GuidHob;\r
\r
    //\r
    // Retrieve WinNtThunkProtocol from GUID'ed HOB\r
    //\r
    GuidHob = GetFirstGuidHob (&gEfiWinNtThunkProtocolGuid);\r
    ASSERT (GuidHob != NULL);\r
    mWinNt = (EFI_WIN_NT_THUNK_PROTOCOL *)(*(UINTN *)(GET_GUID_HOB_DATA (GuidHob)));\r
    ASSERT (mWinNt != NULL);\r
  }\r
\r
  //\r
  // Cache standard output handle.\r
  //\r
  mStdOut = mWinNt->GetStdHandle (STD_OUTPUT_HANDLE);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Report status code to OEM device.\r
 \r