HashInstanceLibSm3.c

Go to the documentation of this file.

 
#include <PiPei.h>
#include <Library/BaseLib.h>
#include <Library/BaseMemoryLib.h>
#include <Library/Tpm2CommandLib.h>
#include <Library/DebugLib.h>
#include <Library/BaseCryptLib.h>
#include <Library/MemoryAllocationLib.h>
#include <Library/HashLib.h>
 
VOID
Tpm2SetSm3ToDigestList (
  IN TPML_DIGEST_VALUES  *DigestList,
  IN UINT8               *Sm3Digest
  )
{
  DigestList->count              = 1;
  DigestList->digests[0].hashAlg = TPM_ALG_SM3_256;
  CopyMem (
    DigestList->digests[0].digest.sm3_256,
    Sm3Digest,
    SM3_256_DIGEST_SIZE
    );
}
 
EFI_STATUS
EFIAPI
Sm3HashInit (
  OUT HASH_HANDLE  *HashHandle
  )
{
  VOID   *Sm3Ctx;
  UINTN  CtxSize;
 
  CtxSize = Sm3GetContextSize ();
  Sm3Ctx  = AllocatePool (CtxSize);
  if (Sm3Ctx == NULL) {
    return EFI_OUT_OF_RESOURCES;
  }
 
  Sm3Init (Sm3Ctx);
 
  *HashHandle = (HASH_HANDLE)Sm3Ctx;
 
  return EFI_SUCCESS;
}
 
EFI_STATUS
EFIAPI
Sm3HashUpdate (
  IN HASH_HANDLE  HashHandle,
  IN VOID         *DataToHash,
  IN UINTN        DataToHashLen
  )
{
  VOID  *Sm3Ctx;
 
  Sm3Ctx = (VOID *)HashHandle;
  Sm3Update (Sm3Ctx, DataToHash, DataToHashLen);
 
  return EFI_SUCCESS;
}
 
EFI_STATUS
EFIAPI
Sm3HashFinal (
  IN HASH_HANDLE          HashHandle,
  OUT TPML_DIGEST_VALUES  *DigestList
  )
{
  UINT8  Digest[SM3_256_DIGEST_SIZE];
  VOID   *Sm3Ctx;
 
  Sm3Ctx = (VOID *)HashHandle;
  Sm3Final (Sm3Ctx, Digest);
 
  FreePool (Sm3Ctx);
 
  Tpm2SetSm3ToDigestList (DigestList, Digest);
 
  return EFI_SUCCESS;
}
 
HASH_INTERFACE  mSm3InternalHashInstance = {
  HASH_ALGORITHM_SM3_256_GUID,
  Sm3HashInit,
  Sm3HashUpdate,
  Sm3HashFinal,
};
 
EFI_STATUS
EFIAPI
HashInstanceLibSm3Constructor (
  VOID
  )
{
  EFI_STATUS  Status;
 
  Status = RegisterHashInterfaceLib (&mSm3InternalHashInstance);
  if ((Status == EFI_SUCCESS) || (Status == EFI_UNSUPPORTED)) {
    //
    // Unsupported means platform policy does not need this instance enabled.
    //
    return EFI_SUCCESS;
  }
 
  return Status;
}