SecurityPkg/AuthVariableLib: Check SHA-256 OID with ContentInfo present

[mirror_edk2.git] / SecurityPkg / Library / HashInstanceLibSm3 / HashInstanceLibSm3.c

/** @file\r
  BaseCrypto SM3 hash instance library.\r
  It can be registered to BaseCrypto router, to serve as hash engine.\r
\r
  Copyright (c) 2013 - 2019, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
**/\r
\r
#include <PiPei.h>\r
#include <Library/BaseLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/Tpm2CommandLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/BaseCryptLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/HashLib.h>\r
\r
/**\r
  The function set SM3 to digest list.\r
\r
  @param DigestList   digest list\r
  @param Sm3Digest    SM3 digest\r
**/\r
VOID\r
Tpm2SetSm3ToDigestList (\r
  IN TPML_DIGEST_VALUES  *DigestList,\r
  IN UINT8               *Sm3Digest\r
  )\r
{\r
  DigestList->count              = 1;\r
  DigestList->digests[0].hashAlg = TPM_ALG_SM3_256;\r
  CopyMem (\r
    DigestList->digests[0].digest.sm3_256,\r
    Sm3Digest,\r
    SM3_256_DIGEST_SIZE\r
    );\r
}\r
\r
/**\r
  Start hash sequence.\r
\r
  @param HashHandle Hash handle.\r
\r
  @retval EFI_SUCCESS          Hash sequence start and HandleHandle returned.\r
  @retval EFI_OUT_OF_RESOURCES No enough resource to start hash.\r
**/\r
EFI_STATUS\r
EFIAPI\r
Sm3HashInit (\r
  OUT HASH_HANDLE  *HashHandle\r
  )\r
{\r
  VOID   *Sm3Ctx;\r
  UINTN  CtxSize;\r
\r
  CtxSize = Sm3GetContextSize ();\r
  Sm3Ctx  = AllocatePool (CtxSize);\r
  if (Sm3Ctx == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  Sm3Init (Sm3Ctx);\r
\r
  *HashHandle = (HASH_HANDLE)Sm3Ctx;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Update hash sequence data.\r
\r
  @param HashHandle    Hash handle.\r
  @param DataToHash    Data to be hashed.\r
  @param DataToHashLen Data size.\r
\r
  @retval EFI_SUCCESS     Hash sequence updated.\r
**/\r
EFI_STATUS\r
EFIAPI\r
Sm3HashUpdate (\r
  IN HASH_HANDLE  HashHandle,\r
  IN VOID         *DataToHash,\r
  IN UINTN        DataToHashLen\r
  )\r
{\r
  VOID  *Sm3Ctx;\r
\r
  Sm3Ctx = (VOID *)HashHandle;\r
  Sm3Update (Sm3Ctx, DataToHash, DataToHashLen);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Complete hash sequence complete.\r
\r
  @param HashHandle    Hash handle.\r
  @param DigestList    Digest list.\r
\r
  @retval EFI_SUCCESS     Hash sequence complete and DigestList is returned.\r
**/\r
EFI_STATUS\r
EFIAPI\r
Sm3HashFinal (\r
  IN HASH_HANDLE          HashHandle,\r
  OUT TPML_DIGEST_VALUES  *DigestList\r
  )\r
{\r
  UINT8  Digest[SM3_256_DIGEST_SIZE];\r
  VOID   *Sm3Ctx;\r
\r
  Sm3Ctx = (VOID *)HashHandle;\r
  Sm3Final (Sm3Ctx, Digest);\r
\r
  FreePool (Sm3Ctx);\r
\r
  Tpm2SetSm3ToDigestList (DigestList, Digest);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
HASH_INTERFACE  mSm3InternalHashInstance = {\r
  HASH_ALGORITHM_SM3_256_GUID,\r
  Sm3HashInit,\r
  Sm3HashUpdate,\r
  Sm3HashFinal,\r
};\r
\r
/**\r
  The function register SM3 instance.\r
\r
  @retval EFI_SUCCESS   SM3 instance is registered, or system dose not support register SM3 instance\r
**/\r
EFI_STATUS\r
EFIAPI\r
HashInstanceLibSm3Constructor (\r
  VOID\r
  )\r
{\r
  EFI_STATUS  Status;\r
\r
  Status = RegisterHashInterfaceLib (&mSm3InternalHashInstance);\r
  if ((Status == EFI_SUCCESS) || (Status == EFI_UNSUPPORTED)) {\r
    //\r
    // Unsupported means platform policy does not need this instance enabled.\r
    //\r
    return EFI_SUCCESS;\r
  }\r
\r
  return Status;\r
}\r