SecurityPkg/HashLibBaseCryptoRouter: recognize the SM3 digest algorithm

[mirror_edk2.git] / SecurityPkg / Library / HashLibBaseCryptoRouter / HashLibBaseCryptoRouterCommon.c

/** @file\r
  This is BaseCrypto router support function.\r
\r
Copyright (c) 2013 - 2016, Intel Corporation. All rights reserved. <BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\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/MemoryAllocationLib.h>\r
#include <Library/HashLib.h>\r
#include <Protocol/Tcg2Protocol.h>\r
\r
typedef struct {\r
  EFI_GUID  Guid;\r
  UINT32    Mask;\r
} TPM2_HASH_MASK;\r
\r
TPM2_HASH_MASK mTpm2HashMask[] = {\r
  {HASH_ALGORITHM_SHA1_GUID,         HASH_ALG_SHA1},\r
  {HASH_ALGORITHM_SHA256_GUID,       HASH_ALG_SHA256},\r
  {HASH_ALGORITHM_SHA384_GUID,       HASH_ALG_SHA384},\r
  {HASH_ALGORITHM_SHA512_GUID,       HASH_ALG_SHA512},\r
  {HASH_ALGORITHM_SM3_256_GUID,      HASH_ALG_SM3_256},\r
};\r
\r
/**\r
  The function get hash mask info from algorithm.\r
\r
  @param HashGuid Hash Guid\r
\r
  @return HashMask\r
**/\r
UINT32\r
EFIAPI\r
Tpm2GetHashMaskFromAlgo (\r
  IN EFI_GUID  *HashGuid\r
  )\r
{\r
  UINTN  Index;\r
  for (Index = 0; Index < sizeof(mTpm2HashMask)/sizeof(mTpm2HashMask[0]); Index++) {\r
    if (CompareGuid (HashGuid, &mTpm2HashMask[Index].Guid)) {\r
      return mTpm2HashMask[Index].Mask;\r
    }\r
  }\r
  return 0;\r
}\r
\r
/**\r
  The function set digest to digest list.\r
\r
  @param DigestList digest list\r
  @param Digest     digest data\r
**/\r
VOID\r
EFIAPI\r
Tpm2SetHashToDigestList (\r
  IN OUT TPML_DIGEST_VALUES *DigestList,\r
  IN TPML_DIGEST_VALUES     *Digest\r
  )\r
{\r
  CopyMem (\r
    &DigestList->digests[DigestList->count],\r
    &Digest->digests[0],\r
    sizeof(Digest->digests[0])\r
    );\r
  DigestList->count ++;\r
}\r