1 files changed, 79 insertions, 0 deletions
diff --git a/src/tests/test_modes.cpp b/src/tests/test_modes.cpp
new file mode 100644
index 000000000..1b546cea3
--- /dev/null
+++ b/src/tests/test_modes.cpp
@@ -0,0 +1,79 @@
+#include "tests.h"
+
+#include <botan/hex.h>
+#include <botan/lookup.h>
+#include <botan/cipher_mode.h>
+#include <botan/filters.h>
+#include <iostream>
+#include <fstream>
+#include <memory>
+
+using namespace Botan;
+
+namespace {
+
+secure_vector<byte> run_mode(const std::string& algo,
+                             Cipher_Dir dir,
+                             const secure_vector<byte>& pt,
+                             const secure_vector<byte>& nonce,
+                             const secure_vector<byte>& key)
+   {
+#if 0
+   std::unique_ptr<Cipher_Mode> cipher(get_cipher(algo, dir));
+
+   cipher->set_key(key);
+   cipher->start_vec(nonce);
+
+   secure_vector<byte> ct = pt;
+   cipher->finish(ct);
+#endif
+
+   Pipe pipe(get_cipher(algo, SymmetricKey(key), InitializationVector(nonce), dir));
+
+   pipe.process_msg(pt);
+
+   return pipe.read_all();
+   }
+
+bool mode_test(const std::string& algo,
+               const std::string& pt,
+               const std::string& ct,
+               const std::string& key_hex,
+               const std::string& nonce_hex)
+   {
+   auto nonce = hex_decode_locked(nonce_hex);
+   auto key = hex_decode_locked(key_hex);
+
+   const std::string ct2 = hex_encode(run_mode(algo,
+                                               ENCRYPTION,
+                                               hex_decode_locked(pt),
+                                               nonce,
+                                               key));
+
+   if(ct != ct2)
+      std::cout << algo << " got ct " << ct2 << " expected " << ct << "\n";
+
+   const std::string pt2 = hex_encode(run_mode(algo,
+                                               DECRYPTION,
+                                               hex_decode_locked(ct),
+                                               nonce,
+                                               key));
+
+   if(pt != pt2)
+      std::cout << algo << " got pt " << pt2 << " expected " << pt << "\n";
+
+   return (ct == ct2) && (pt == pt2);
+   }
+
+}
+
+size_t test_modes()
+   {
+   std::ifstream vec(CHECKS_DIR "/modes.vec");
+
+   return run_tests_bb(vec, "Mode", "Out", true,
+             [](std::map<std::string, std::string> m)
+             {
+             return mode_test(m["Mode"], m["In"], m["Out"], m["Key"], m["Nonce"]);
+             });
+   }