UNIT_TEST_CONTEXT Context\r
)\r
{\r
- UINT8 Prime[64];\r
+ UINT8 Prime[512];\r
UINT8 PublicKey1[64];\r
UINTN PublicKey1Length;\r
UINT8 PublicKey2[64];\r
Key1Length = sizeof (Key1);\r
Key2Length = sizeof (Key2);\r
\r
- Status = DhGenerateParameter (mDh1, 2, 64, Prime);\r
+ Status = DhGenerateParameter (mDh1, 2, sizeof (Prime), Prime);\r
UT_ASSERT_TRUE (Status);\r
\r
- Status = DhSetParameter (mDh2, 2, 64, Prime);\r
+ Status = DhSetParameter (mDh2, 2, sizeof (Prime), Prime);\r
UT_ASSERT_TRUE (Status);\r
\r
Status = DhGenerateKey (mDh1, PublicKey1, &PublicKey1Length);\r