Remove some unnecessary usages of PK_Signing_Key

2 files changed, 30 insertions, 53 deletions

diff --git a/src/ssl/cert_ver.cpp b/src/ssl/cert_ver.cpp
index 5ac28dd2e..22dae8350 100644
--- a/src/ssl/cert_ver.cpp
+++ b/src/ssl/cert_ver.cpp
@@ -22,32 +22,22 @@ Certificate_Verify::Certificate_Verify(RandomNumberGenerator& rng,
                                        HandshakeHash& hash,
                                        const Private_Key* priv_key)
    {
-   const PK_Signing_Key* sign_key =
-      dynamic_cast<const PK_Signing_Key*>(priv_key);
+   std::auto_ptr<PK_Signer> signer;
 
-   if(sign_key)
+   if(const RSA_PrivateKey* rsa = dynamic_cast<const RSA_PrivateKey*>(priv_key))
       {
-      PK_Signer* signer = 0;
-      try
-         {
-         if(dynamic_cast<const RSA_PrivateKey*>(sign_key))
-            signer = get_pk_signer(*sign_key, "EMSA3(TLS.Digest.0)");
-         else if(dynamic_cast<const DSA_PrivateKey*>(sign_key))
-            signer = get_pk_signer(*sign_key, "EMSA1(SHA-1)");
-         else
-            throw Invalid_Argument("Unknown PK algo for TLS signature");
-
-         signature = signer->sign_message(hash.final(), rng);
-         delete signer;
-         }
-      catch(...)
-         {
-         delete signer;
-         throw;
-         }
-
-      send(writer, hash);
+      signer.reset(get_pk_signer(*rsa, "EMSA3(TLS.Digest.0)"));
       }
+   else if(const DSA_PrivateKey* dsa =
+           dynamic_cast<const DSA_PrivateKey*>(priv_key))
+      {
+      signer.reset(get_pk_signer(*dsa, "EMSA1(SHA-1)"));
+      }
+   else
+      throw Invalid_Argument("Unknown PK algo for TLS signature");
+
+   signature = signer->sign_message(hash.final(), rng);
+   send(writer, hash);
    }
 
 /**
diff --git a/src/ssl/s_kex.cpp b/src/ssl/s_kex.cpp
index fd49fcb8c..d568ef14f 100644
--- a/src/ssl/s_kex.cpp
+++ b/src/ssl/s_kex.cpp
@@ -43,38 +43,25 @@ Server_Key_Exchange::Server_Key_Exchange(RandomNumberGenerator& rng,
    else
       throw Invalid_Argument("Bad key for TLS key exchange: not DH or RSA");
 
-   // FIXME: dup of stuff in cert_ver.cpp
-   // FIXME: it's OK for the server to be anonymous....
-   const PK_Signing_Key* sign_key =
-      dynamic_cast<const PK_Signing_Key*>(priv_key);
-
-   if(!sign_key)
-      throw Invalid_Argument("Server Kex: Private key not for signing");
-
-   PK_Signer* signer = 0;
-   try {
-      if(dynamic_cast<const RSA_PrivateKey*>(sign_key))
-         signer = get_pk_signer(*sign_key, "EMSA3(TLS.Digest.0)");
-      else if(dynamic_cast<const DSA_PrivateKey*>(sign_key))
-         {
-         signer = get_pk_signer(*sign_key, "EMSA1(SHA-1)");
-         signer->set_output_format(DER_SEQUENCE);
-         }
-      else
-         throw Invalid_Argument("Bad key for TLS signature: not RSA or DSA");
-
-      signer->update(c_random);
-      signer->update(s_random);
-      signer->update(serialize_params());
-      signature = signer->signature(rng);
-
-      delete signer;
-   }
-   catch(...)
+   std::auto_ptr<PK_Signer> signer;
+
+   if(const RSA_PrivateKey* rsa = dynamic_cast<const RSA_PrivateKey*>(priv_key))
+      {
+      signer.reset(get_pk_signer(*rsa, "EMSA3(TLS.Digest.0)"));
+      }
+   else if(const DSA_PrivateKey* dsa =
+           dynamic_cast<const DSA_PrivateKey*>(priv_key))
       {
-      delete signer;
-      throw;
+      signer.reset(get_pk_signer(*dsa, "EMSA1(SHA-1)"));
+      signer->set_output_format(DER_SEQUENCE);
       }
+   else
+      throw Invalid_Argument("Bad key for TLS signature: not RSA or DSA");
+
+   signer->update(c_random);
+   signer->update(s_random);
+   signer->update(serialize_params());
+   signature = signer->signature(rng);
 
    send(writer, hash);
    }