CryptoPkg/Test/UnitTest/Library/BaseCryptLib/BaseCryptLibUnitTests.c

   1 /** @file
   2   This is defines the tests that will run on BaseCryptLib
   3
   4   Copyright (c) Microsoft Corporation.<BR>
   5   SPDX-License-Identifier: BSD-2-Clause-Patent
   6
   7 **/
   8 #include "TestBaseCryptLib.h"
   9
  10 SUITE_DESC  mSuiteDesc[] = {
  11     //
  12     // Title--------------------------Package-------------------Sup--Tdn----TestNum------------TestDesc
  13     //
  14     {"EKU verify tests",            "CryptoPkg.BaseCryptLib", NULL, NULL, &mPkcs7EkuTestNum,       mPkcs7EkuTest},
  15     {"HASH verify tests",           "CryptoPkg.BaseCryptLib", NULL, NULL, &mHashTestNum,           mHashTest},
  16     {"HMAC verify tests",           "CryptoPkg.BaseCryptLib", NULL, NULL, &mHmacTestNum,           mHmacTest},
  17     {"BlockCipher verify tests",    "CryptoPkg.BaseCryptLib", NULL, NULL, &mBlockCipherTestNum,    mBlockCipherTest},
  18     {"RSA verify tests",            "CryptoPkg.BaseCryptLib", NULL, NULL, &mRsaTestNum,            mRsaTest},
  19     {"RSACert verify tests",        "CryptoPkg.BaseCryptLib", NULL, NULL, &mRsaCertTestNum,        mRsaCertTest},
  20     {"PKCS7 verify tests",          "CryptoPkg.BaseCryptLib", NULL, NULL, &mPkcs7TestNum,          mPkcs7Test},
  21     {"PKCS5 verify tests",          "CryptoPkg.BaseCryptLib", NULL, NULL, &mPkcs5TestNum,          mPkcs5Test},
  22     {"Authenticode verify tests",   "CryptoPkg.BaseCryptLib", NULL, NULL, &mAuthenticodeTestNum,   mAuthenticodeTest},
  23     {"ImageTimestamp verify tests", "CryptoPkg.BaseCryptLib", NULL, NULL, &mImageTimestampTestNum, mImageTimestampTest},
  24     {"DH verify tests",             "CryptoPkg.BaseCryptLib", NULL, NULL, &mDhTestNum,             mDhTest},
  25     {"PRNG verify tests",           "CryptoPkg.BaseCryptLib", NULL, NULL, &mPrngTestNum,           mPrngTest},
  26     {"OAEP encrypt verify tests",   "CryptoPkg.BaseCryptLib", NULL, NULL, &mOaepTestNum,           mOaepTest},
  27 };
  28
  29 EFI_STATUS
  30 EFIAPI
  31 CreateUnitTest (
  32     IN     CHAR8*                     UnitTestName,
  33     IN     CHAR8*                     UnitTestVersion,
  34     IN OUT UNIT_TEST_FRAMEWORK_HANDLE* Framework
  35 ) {
  36     EFI_STATUS                  Status;
  37     UINTN                       SuiteIndex;
  38     UINTN                       TestIndex;
  39
  40     if ( Framework == NULL || UnitTestVersion == NULL || UnitTestName == NULL) {
  41         return EFI_INVALID_PARAMETER;
  42     }
  43     Status = EFI_SUCCESS;
  44     //
  45     // Start setting up the test framework for running the tests.
  46     //
  47     Status = InitUnitTestFramework (Framework, UnitTestName, gEfiCallerBaseName, UnitTestVersion);
  48     if (EFI_ERROR (Status)) {
  49         DEBUG((DEBUG_ERROR, "Failed in InitUnitTestFramework. Status = %r\n", Status));
  50         goto EXIT;
  51     }
  52
  53     for (SuiteIndex = 0; SuiteIndex < ARRAY_SIZE(mSuiteDesc); SuiteIndex++) {
  54         UNIT_TEST_SUITE_HANDLE Suite = NULL;
  55         Status = CreateUnitTestSuite (&Suite, *Framework, mSuiteDesc[SuiteIndex].Title, mSuiteDesc[SuiteIndex].Package, mSuiteDesc[SuiteIndex].Sup, mSuiteDesc[SuiteIndex].Tdn);
  56         if (EFI_ERROR (Status)) {
  57             Status = EFI_OUT_OF_RESOURCES;
  58             goto EXIT;
  59         }
  60         for (TestIndex = 0; TestIndex < *mSuiteDesc[SuiteIndex].TestNum; TestIndex++) {
  61             AddTestCase (Suite, (mSuiteDesc[SuiteIndex].TestDesc + TestIndex)->Description, (mSuiteDesc[SuiteIndex].TestDesc + TestIndex)->ClassName, (mSuiteDesc[SuiteIndex].TestDesc + TestIndex)->Func, (mSuiteDesc[SuiteIndex].TestDesc + TestIndex)->PreReq, (mSuiteDesc[SuiteIndex].TestDesc + TestIndex)->CleanUp, (mSuiteDesc[SuiteIndex].TestDesc + TestIndex)->Context);
  62         }
  63     }
  64     EXIT:
  65     return Status;
  66 }