commit 3b9a5032032cceb1c88fdbcd9edd8c8d3d47a390
author Zhiyi Zhang <zhiyi@cs.ucla.edu> Thu Feb 18 11:15:09 2021 -0800
committer Zhiyi Zhang <zhiyi@cs.ucla.edu> Thu Feb 18 11:15:09 2021 -0800
tree fcb09e86557a5bf0e2bcf4376b097dfca6709877
parent de58e605fe26391833c883f1927f7f4cdc226e5b

update

Change-Id: I1bc32282305af59579b17037c6f878a37c85f556

tests/unit-tests/crypto-helpers.t.cpp

diff --git a/tests/unit-tests/crypto-helpers.t.cpp b/tests/unit-tests/crypto-helpers.t.cpp
index 4e493db..feb0c17 100644
--- a/tests/unit-tests/crypto-helpers.t.cpp
+++ b/tests/unit-tests/crypto-helpers.t.cpp
@@ -309,26 +309,35 @@
   // long string encryption
   auto block = encodeBlockWithAesGcm128(ndn::tlv::Content, key, (uint8_t*)plaintext.c_str(), plaintext.size(),
                                         (uint8_t*)associatedData.c_str(), associatedData.size(), encryptionIv);
+  BOOST_CHECK_EQUAL(encryptionIv.size(), 12);
+  // the decryption's random component cannot be the same as encryption IV
+  BOOST_CHECK_THROW(decodeBlockWithAesGcm128(block, key,
+                                            (uint8_t*)associatedData.c_str(),
+                                            associatedData.size(), decryptionIv, encryptionIv),
+                    std::runtime_error);
   auto decoded = decodeBlockWithAesGcm128(block, key, (uint8_t*)associatedData.c_str(), associatedData.size(),
-                                          decryptionIv);
+                                          decryptionIv, std::vector<uint8_t>());
+  BOOST_CHECK_EQUAL(decryptionIv.size(), 12);
   BOOST_CHECK_EQUAL(plaintext, std::string(decoded.get<char>(), decoded.size()));
 
   // short string encryption
   block = encodeBlockWithAesGcm128(ndn::tlv::Content, key, (uint8_t*)plaintext2.c_str(), plaintext2.size(),
                                    (uint8_t*)associatedData.c_str(), associatedData.size(), encryptionIv);
   decoded = decodeBlockWithAesGcm128(block, key, (uint8_t*)associatedData.c_str(), associatedData.size(),
-                                     decryptionIv);
+                                     decryptionIv, std::vector<uint8_t>());
   BOOST_CHECK_EQUAL(plaintext2, std::string(decoded.get<char>(), decoded.size()));
 
   // use wrong associated data
   BOOST_CHECK_THROW(decodeBlockWithAesGcm128(block, key,
                                              (uint8_t*)wrongAssociatedData.c_str(),
-                                             wrongAssociatedData.size(), decryptionIv), std::runtime_error);
+                                             wrongAssociatedData.size(), decryptionIv, std::vector<uint8_t>()),
+                    std::runtime_error);
   // use wrong last observed IV
   decryptionIv[0] += 1;
   BOOST_CHECK_THROW(decodeBlockWithAesGcm128(block, key,
                                              (uint8_t*)associatedData.c_str(),
-                                             associatedData.size(), decryptionIv), std::runtime_error);
+                                             associatedData.size(), decryptionIv, std::vector<uint8_t>()),
+                    std::runtime_error);
 }
 
 BOOST_AUTO_TEST_SUITE_END()