[PATCH 1/2] bcrypt: Use privkey if pubkey is absent in key_asymmetric_encrypt().

[Paul Gofman]

From: Paul Gofman

---
 dlls/bcrypt/gnutls.c | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/dlls/bcrypt/gnutls.c b/dlls/bcrypt/gnutls.c
index bb79e885652..ed4e5176e80 100644
--- a/dlls/bcrypt/gnutls.c
+++ b/dlls/bcrypt/gnutls.c
@@ -2203,16 +2203,38 @@ static NTSTATUS key_asymmetric_encrypt( void *args )
     const struct key_asymmetric_encrypt_params *params = args;
     gnutls_datum_t d, e = { 0 };
     NTSTATUS status = STATUS_SUCCESS;
+    gnutls_pubkey_t pubkey;
     int ret;

d.data = params->input;
     d.size = params->input_len;
-    if ((ret = pgnutls_pubkey_encrypt_data(key_data(params->key)->a.pubkey, 0, &d, &e)))
+
+    if (!(pubkey = key_data(params->key)->a.pubkey))
+    {
+        if (!key_data(params->key)->a.privkey) return STATUS_INVALID_HANDLE;
+
+        if ((ret = pgnutls_pubkey_init( &pubkey )))
+        {
+            pgnutls_perror( ret );
+            return STATUS_INTERNAL_ERROR;
+        }
+        if ((ret = pgnutls_pubkey_import_privkey( pubkey, key_data(params->key)->a.privkey, 0, 0 )))
+        {
+            pgnutls_perror( ret );
+            pgnutls_pubkey_deinit( pubkey );
+            return STATUS_INTERNAL_ERROR;
+        }
+    }
+
+    if ((ret = pgnutls_pubkey_encrypt_data(pubkey, 0, &d, &e)))
     {
         pgnutls_perror( ret );
         return STATUS_INTERNAL_ERROR;
     }

+    if (pubkey != key_data(params->key)->a.pubkey)
+        pgnutls_pubkey_deinit( pubkey );
+
     *params->ret_len = e.size;
     if (params->output_len >= e.size) memcpy( params->output, e.data, *params->ret_len );
     else if (params->output_len == 0) status = STATUS_SUCCESS;