1 files changed, 67 insertions, 0 deletions
diff --git a/src/tests/test_elg.cpp b/src/tests/test_elg.cpp
new file mode 100644
index 000000000..332e5ae0e
--- /dev/null
+++ b/src/tests/test_elg.cpp
@@ -0,0 +1,67 @@
+#include "tests.h"
+#include "test_pubkey.h"
+
+#include <botan/auto_rng.h>
+#include <botan/pubkey.h>
+#include <botan/dl_group.h>
+#include <botan/hex.h>
+#include <iostream>
+#include <fstream>
+
+#if defined(BOTAN_HAS_ELGAMAL)
+  #include <botan/elgamal.h>
+#endif
+
+using namespace Botan;
+
+namespace {
+
+size_t elgamal_kat(const std::string& p,
+                   const std::string& g,
+                   const std::string& x,
+                   const std::string& msg,
+                   std::string padding,
+                   const std::string& nonce,
+                   const std::string& ciphertext)
+   {
+   AutoSeeded_RNG rng;
+
+   const BigInt p_bn = BigInt(p);
+   const BigInt g_bn = BigInt(g);
+   const BigInt x_bn = BigInt(x);
+
+   DL_Group group(p_bn, g_bn);
+#if defined(BOTAN_HAS_ELGAMAL)
+   ElGamal_PrivateKey privkey(rng, group, x_bn);
+
+   ElGamal_PublicKey pubkey = privkey;
+
+   if(padding == "")
+      padding = "Raw";
+
+   PK_Encryptor_EME enc(pubkey, padding);
+   PK_Decryptor_EME dec(privkey, padding);
+
+   return validate_encryption(enc, dec, "ElGamal/" + padding, msg, nonce, ciphertext);
+#else
+   return 1;
+#endif
+   }
+
+}
+
+size_t test_elgamal()
+   {
+   std::ifstream elgamal_enc(PK_TEST_DATA_DIR "/elgamal.vec");
+
+   size_t fails = 0;
+
+   fails += run_tests_bb(elgamal_enc, "ElGamal Encryption", "Ciphertext", true,
+             [](std::map<std::string, std::string> m) -> size_t
+             {
+             return elgamal_kat(m["P"], m["G"], m["X"], m["Msg"],
+                              m["Padding"], m["Nonce"], m["Ciphertext"]);
+             });
+
+   return fails;
+   }