1 files changed, 31 insertions, 19 deletions

diff --git a/ssh-ecdsa.c b/ssh-ecdsa.c
index 766338941..10ad9da60 100644
--- a/ssh-ecdsa.c
+++ b/ssh-ecdsa.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: ssh-ecdsa.c,v 1.6 2013/05/17 00:13:14 djm Exp $ */
+/* $OpenBSD: ssh-ecdsa.c,v 1.8 2014/01/09 23:20:00 djm Exp $ */
 /*
  * Copyright (c) 2000 Markus Friedl.  All rights reserved.
  * Copyright (c) 2010 Damien Miller.  All rights reserved.
@@ -42,27 +42,34 @@
 #include "compat.h"
 #include "log.h"
 #include "key.h"
+#include "digest.h"
 
 int
 ssh_ecdsa_sign(const Key *key, u_char **sigp, u_int *lenp,
     const u_char *data, u_int datalen)
 {
         ECDSA_SIG *sig;
-        const EVP_MD *evp_md;
-        EVP_MD_CTX md;
-        u_char digest[EVP_MAX_MD_SIZE];
+        int hash_alg;
+        u_char digest[SSH_DIGEST_MAX_LENGTH];
         u_int len, dlen;
         Buffer b, bb;
 
-        if (key == NULL || key->ecdsa == NULL ||
-            (key->type != KEY_ECDSA && key->type != KEY_ECDSA_CERT)) {
+        if (key == NULL || key_type_plain(key->type) != KEY_ECDSA ||
+            key->ecdsa == NULL) {
                 error("%s: no ECDSA key", __func__);
                 return -1;
         }
-        evp_md = key_ec_nid_to_evpmd(key->ecdsa_nid);
-        EVP_DigestInit(&md, evp_md);
-        EVP_DigestUpdate(&md, data, datalen);
-        EVP_DigestFinal(&md, digest, &dlen);
+
+        hash_alg = key_ec_nid_to_hash_alg(key->ecdsa_nid);
+        if ((dlen = ssh_digest_bytes(hash_alg)) == 0) {
+                error("%s: bad hash algorithm %d", __func__, hash_alg);
+                return -1;
+        }
+        if (ssh_digest_memory(hash_alg, data, datalen,
+            digest, sizeof(digest)) != 0) {
+                error("%s: digest_memory failed", __func__);
+                return -1;
+        }
 
         sig = ECDSA_do_sign(digest, dlen, key->ecdsa);
         memset(digest, 'd', sizeof(digest));
@@ -97,20 +104,18 @@ ssh_ecdsa_verify(const Key *key, const u_char *signature, u_int signaturelen,
     const u_char *data, u_int datalen)
 {
         ECDSA_SIG *sig;
-        const EVP_MD *evp_md;
-        EVP_MD_CTX md;
-        u_char digest[EVP_MAX_MD_SIZE], *sigblob;
+        int hash_alg;
+        u_char digest[SSH_DIGEST_MAX_LENGTH], *sigblob;
         u_int len, dlen;
         int rlen, ret;
         Buffer b, bb;
         char *ktype;
 
-        if (key == NULL || key->ecdsa == NULL ||
-            (key->type != KEY_ECDSA && key->type != KEY_ECDSA_CERT)) {
+        if (key == NULL || key_type_plain(key->type) != KEY_ECDSA ||
+            key->ecdsa == NULL) {
                 error("%s: no ECDSA key", __func__);
                 return -1;
         }
-        evp_md = key_ec_nid_to_evpmd(key->ecdsa_nid);
 
         /* fetch signature */
         buffer_init(&b);
@@ -152,9 +157,16 @@ ssh_ecdsa_verify(const Key *key, const u_char *signature, u_int signaturelen,
         free(sigblob);
 
         /* hash the data */
-        EVP_DigestInit(&md, evp_md);
-        EVP_DigestUpdate(&md, data, datalen);
-        EVP_DigestFinal(&md, digest, &dlen);
+        hash_alg = key_ec_nid_to_hash_alg(key->ecdsa_nid);
+        if ((dlen = ssh_digest_bytes(hash_alg)) == 0) {
+                error("%s: bad hash algorithm %d", __func__, hash_alg);
+                return -1;
+        }
+        if (ssh_digest_memory(hash_alg, data, datalen,
+            digest, sizeof(digest)) != 0) {
+                error("%s: digest_memory failed", __func__);
+                return -1;
+        }
 
         ret = ECDSA_do_verify(digest, dlen, sig, key->ecdsa);
         memset(digest, 'd', sizeof(digest));