1 files changed, 71 insertions, 0 deletions
diff --git a/src/pubkey/keypair/keypair.cpp b/src/pubkey/keypair/keypair.cpp
new file mode 100644
index 000000000..940f0c028
--- /dev/null
+++ b/src/pubkey/keypair/keypair.cpp
@@ -0,0 +1,71 @@
+/*************************************************
+* Keypair Checks Source File                     *
+* (C) 1999-2007 Jack Lloyd                       *
+*************************************************/
+
+#include <botan/keypair.h>
+#include <botan/look_pk.h>
+#include <memory>
+
+namespace Botan {
+
+namespace KeyPair {
+
+/*************************************************
+* Check an encryption key pair for consistency   *
+*************************************************/
+void check_key(RandomNumberGenerator& rng,
+               PK_Encryptor* encryptor, PK_Decryptor* decryptor)
+   {
+   if(encryptor->maximum_input_size() == 0)
+      return;
+
+   std::auto_ptr<PK_Encryptor> enc(encryptor);
+   std::auto_ptr<PK_Decryptor> dec(decryptor);
+
+   SecureVector<byte> message(enc->maximum_input_size() - 1);
+   rng.randomize(message, message.size());
+
+   SecureVector<byte> ciphertext = enc->encrypt(message, rng);
+   if(ciphertext == message)
+      throw Self_Test_Failure("Encryption key pair consistency failure");
+
+   SecureVector<byte> message2 = dec->decrypt(ciphertext);
+   if(message != message2)
+      throw Self_Test_Failure("Encryption key pair consistency failure");
+   }
+
+/*************************************************
+* Check a signature key pair for consistency     *
+*************************************************/
+void check_key(RandomNumberGenerator& rng,
+               PK_Signer* signer, PK_Verifier* verifier)
+   {
+   std::auto_ptr<PK_Signer> sig(signer);
+   std::auto_ptr<PK_Verifier> ver(verifier);
+
+   SecureVector<byte> message(16);
+   rng.randomize(message, message.size());
+
+   SecureVector<byte> signature;
+
+   try
+      {
+      signature = sig->sign_message(message, rng);
+      }
+   catch(Encoding_Error)
+      {
+      return;
+      }
+
+   if(!ver->verify_message(message, signature))
+      throw Self_Test_Failure("Signature key pair consistency failure");
+
+   ++message[0];
+   if(ver->verify_message(message, signature))
+      throw Self_Test_Failure("Signature key pair consistency failure");
+   }
+
+}
+
+}