Follow up EDKT236.

[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
EFI_WIN_NT_THUNK_PROTOCOL   *mWinNt;\r
\r
//\r
// Cache of standard output handle . \r
// \r
HANDLE                      mStdOut;\r
\r
/**\r
\r
  Initialize OEM status code device .\r
\r
**/\r
VOID\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;\r
}\r
\r
/**\r
  Report status code to OEM device.\r
 \r