CryptoPkg: Apply uncrustify changes

[mirror_edk2.git] / CryptoPkg / Test / UnitTest / Library / BaseCryptLib / DhTests.c

/** @file\r
  Application for Diffie-Hellman Primitives Validation.\r
\r
Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "TestBaseCryptLib.h"\r
\r
VOID  *mDh1;\r
VOID  *mDh2;\r
\r
UNIT_TEST_STATUS\r
EFIAPI\r
TestVerifyDhPreReq (\r
  UNIT_TEST_CONTEXT  Context\r
  )\r
{\r
  mDh1 = DhNew ();\r
  if (mDh1 == NULL) {\r
    return UNIT_TEST_ERROR_TEST_FAILED;\r
  }\r
\r
  mDh2 = DhNew ();\r
  if (mDh2 == NULL) {\r
    return UNIT_TEST_ERROR_TEST_FAILED;\r
  }\r
\r
  return UNIT_TEST_PASSED;\r
}\r
\r
VOID\r
EFIAPI\r
TestVerifyDhCleanUp (\r
  UNIT_TEST_CONTEXT  Context\r
  )\r
{\r
  if (mDh1 != NULL) {\r
    DhFree (mDh1);\r
    mDh1 = NULL;\r
  }\r
\r
  if (mDh2 != NULL) {\r
    DhFree (mDh2);\r
    mDh2 = NULL;\r
  }\r
}\r
\r
UNIT_TEST_STATUS\r
EFIAPI\r
TestVerifyDhGenerateKey (\r
  UNIT_TEST_CONTEXT  Context\r
  )\r
{\r
  UINT8    Prime[64];\r
  UINT8    PublicKey1[64];\r
  UINTN    PublicKey1Length;\r
  UINT8    PublicKey2[64];\r
  UINTN    PublicKey2Length;\r
  UINT8    Key1[64];\r
  UINTN    Key1Length;\r
  UINT8    Key2[64];\r
  UINTN    Key2Length;\r
  BOOLEAN  Status;\r
\r
  //\r
  // Initialize Key Length\r
  //\r
  PublicKey1Length = sizeof (PublicKey1);\r
  PublicKey2Length = sizeof (PublicKey2);\r
  Key1Length       = sizeof (Key1);\r
  Key2Length       = sizeof (Key2);\r
\r
  Status = DhGenerateParameter (mDh1, 2, 64, Prime);\r
  UT_ASSERT_TRUE (Status);\r
\r
  Status = DhSetParameter (mDh2, 2, 64, Prime);\r
  UT_ASSERT_TRUE (Status);\r
\r
  Status = DhGenerateKey (mDh1, PublicKey1, &PublicKey1Length);\r
  UT_ASSERT_TRUE (Status);\r
\r
  Status = DhGenerateKey (mDh2, PublicKey2, &PublicKey2Length);\r
  UT_ASSERT_TRUE (Status);\r
\r
  Status = DhComputeKey (mDh1, PublicKey2, PublicKey2Length, Key1, &Key1Length);\r
  UT_ASSERT_TRUE (Status);\r
\r
  Status = DhComputeKey (mDh2, PublicKey1, PublicKey1Length, Key2, &Key2Length);\r
  UT_ASSERT_TRUE (Status);\r
\r
  UT_ASSERT_EQUAL (Key1Length, Key2Length);\r
\r
  UT_ASSERT_MEM_EQUAL (Key1, Key2, Key1Length);\r
\r
  return UNIT_TEST_PASSED;\r
}\r
\r
TEST_DESC  mDhTest[] = {\r
  //\r
  // -----Description--------------------------------Class---------------------Function----------------Pre-----------------Post------------Context\r
  //\r
  { "TestVerifyDhGenerateKey()", "CryptoPkg.BaseCryptLib.Dh", TestVerifyDhGenerateKey, TestVerifyDhPreReq, TestVerifyDhCleanUp, NULL },\r
};\r
\r
UINTN  mDhTestNum = ARRAY_SIZE (mDhTest);\r