summaryrefslogtreecommitdiff
path: root/tests/sign-verify-ext4.c
diff options
context:
space:
mode:
Diffstat (limited to 'tests/sign-verify-ext4.c')
-rw-r--r--tests/sign-verify-ext4.c5
1 files changed, 5 insertions, 0 deletions
diff --git a/tests/sign-verify-ext4.c b/tests/sign-verify-ext4.c
index 94daf000e7..81aa345bf0 100644
--- a/tests/sign-verify-ext4.c
+++ b/tests/sign-verify-ext4.c
@@ -71,6 +71,7 @@ struct key_cb_data {
gnutls_privkey_t rkey; /* the real thing */
unsigned pk;
unsigned sig;
+ unsigned bits;
};
static int key_cb_info_func(gnutls_privkey_t key, unsigned int flags, void *userdata)
@@ -79,6 +80,8 @@ static int key_cb_info_func(gnutls_privkey_t key, unsigned int flags, void *user
if (flags & GNUTLS_PRIVKEY_INFO_PK_ALGO)
return p->pk;
+ else if (flags & GNUTLS_PRIVKEY_INFO_PK_ALGO_BITS)
+ return p->bits;
else if (flags & GNUTLS_PRIVKEY_INFO_HAVE_SIGN_ALGO) {
unsigned sig = GNUTLS_FLAGS_TO_SIGN_ALGO(flags);
@@ -153,6 +156,8 @@ static gnutls_privkey_t load_virt_privkey(const gnutls_datum_t *txtkey, gnutls_p
if (ret < 0)
fail("gnutls_privkey_import\n");
+ gnutls_privkey_get_pk_algorithm(userdata->rkey, &userdata->bits);
+
userdata->pk = pk;
userdata->sig = sig;