summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--CryptoPkg/Test/UnitTest/Library/BaseCryptLib/DhTests.c6
1 files changed, 3 insertions, 3 deletions
diff --git a/CryptoPkg/Test/UnitTest/Library/BaseCryptLib/DhTests.c b/CryptoPkg/Test/UnitTest/Library/BaseCryptLib/DhTests.c
index 5cfe8d7053..29e892a151 100644
--- a/CryptoPkg/Test/UnitTest/Library/BaseCryptLib/DhTests.c
+++ b/CryptoPkg/Test/UnitTest/Library/BaseCryptLib/DhTests.c
@@ -53,7 +53,7 @@ TestVerifyDhGenerateKey (
UNIT_TEST_CONTEXT Context
)
{
- UINT8 Prime[64];
+ UINT8 Prime[512];
UINT8 PublicKey1[64];
UINTN PublicKey1Length;
UINT8 PublicKey2[64];
@@ -72,10 +72,10 @@ TestVerifyDhGenerateKey (
Key1Length = sizeof (Key1);
Key2Length = sizeof (Key2);
- Status = DhGenerateParameter (mDh1, 2, 64, Prime);
+ Status = DhGenerateParameter (mDh1, 2, sizeof (Prime), Prime);
UT_ASSERT_TRUE (Status);
- Status = DhSetParameter (mDh2, 2, 64, Prime);
+ Status = DhSetParameter (mDh2, 2, sizeof (Prime), Prime);
UT_ASSERT_TRUE (Status);
Status = DhGenerateKey (mDh1, PublicKey1, &PublicKey1Length);