Fix dynamic public key method lookup.
[openssl.git] / crypto / asn1 / ameth_lib.c
index ddb6333913346dd9c6dbc69fc986d2fbaa191320..6b4690e5b7680240db24e3868081bd3b8ac54e04 100644 (file)
 
 extern const EVP_PKEY_ASN1_METHOD rsa_asn1_meths[];
 extern const EVP_PKEY_ASN1_METHOD dsa_asn1_meths[];
+extern const EVP_PKEY_ASN1_METHOD dh_asn1_meth;
 extern const EVP_PKEY_ASN1_METHOD eckey_asn1_meth;
 
 /* Keep this sorted in type order !! */
 const EVP_PKEY_ASN1_METHOD *standard_methods[] = 
        {
+#ifndef OPENSSL_NO_RSA
        &rsa_asn1_meths[0],
        &rsa_asn1_meths[1],
+#endif
+#ifndef OPENSSL_NO_DH
+       &dh_asn1_meth,
+#endif
+#ifndef OPENSSL_NO_DSA
        &dsa_asn1_meths[0],
        &dsa_asn1_meths[1],
        &dsa_asn1_meths[2],
        &dsa_asn1_meths[3],
        &dsa_asn1_meths[4],
+#endif
+#ifndef OPENSSL_NO_EC
        &eckey_asn1_meth
+#endif
        };
 
+typedef int sk_cmp_fn_type(const char * const *a, const char * const *b);
+static STACK *app_methods = NULL;
+
+
+
 #ifdef TEST
 void main()
        {
@@ -86,8 +101,9 @@ void main()
        for (i = 0;
                i < sizeof(standard_methods)/sizeof(EVP_PKEY_ASN1_METHOD *);
                i++)
-               fprintf(stderr, "Number %d id=%d\n", i,
-                       standard_methods[i]->pkey_id);
+               fprintf(stderr, "Number %d id=%d (%s)\n", i,
+                       standard_methods[i]->pkey_id,
+                       OBJ_nid2sn(standard_methods[i]->pkey_id));
        }
 #endif
 
@@ -97,17 +113,222 @@ static int ameth_cmp(const EVP_PKEY_ASN1_METHOD * const *a,
         return ((*a)->pkey_id - (*b)->pkey_id);
        }
 
-const EVP_PKEY_ASN1_METHOD *EVP_PKEY_ASN1_find(int type)
+int EVP_PKEY_asn1_get_count(void)
+       {
+       int num = sizeof(standard_methods)/sizeof(EVP_PKEY_ASN1_METHOD *);
+       if (app_methods)
+               num += sk_num(app_methods);
+       return num;
+       }
+
+const EVP_PKEY_ASN1_METHOD *EVP_PKEY_asn1_get0(int idx)
+       {
+       int num = sizeof(standard_methods)/sizeof(EVP_PKEY_ASN1_METHOD *);
+       if (idx < 0)
+               return NULL; 
+       if (idx < num)
+               return standard_methods[idx];
+       idx -= num;
+       return (const EVP_PKEY_ASN1_METHOD *)sk_value(app_methods, idx);
+       }
+
+const EVP_PKEY_ASN1_METHOD *EVP_PKEY_asn1_find(int type)
        {
        EVP_PKEY_ASN1_METHOD tmp, *t = &tmp, **ret;
        tmp.pkey_id = type;
+       if (app_methods)
+               {
+               int idx;
+               idx = sk_find(app_methods, (char *)&tmp);
+               if (idx >= 0)
+                       return (EVP_PKEY_ASN1_METHOD *)
+                               sk_value(app_methods, idx);
+               }
        ret = (EVP_PKEY_ASN1_METHOD **) OBJ_bsearch((char *)&t,
                        (char *)standard_methods,
                        sizeof(standard_methods)/sizeof(EVP_PKEY_ASN1_METHOD *),
                        sizeof(EVP_PKEY_ASN1_METHOD *),
                        (int (*)(const void *, const void *))ameth_cmp);
+       if (!ret || !*ret)
+               return NULL;
        if ((*ret)->pkey_flags & ASN1_PKEY_ALIAS)
-               return EVP_PKEY_ASN1_find((*ret)->pkey_base_id);
+               return EVP_PKEY_asn1_find((*ret)->pkey_base_id);
        return *ret;
        }
 
+const EVP_PKEY_ASN1_METHOD *EVP_PKEY_asn1_find_str(const char *str, int len)
+       {
+       int i;
+       const EVP_PKEY_ASN1_METHOD *ameth;
+       for (i = 0; i < EVP_PKEY_asn1_get_count(); i++)
+               {
+               ameth = EVP_PKEY_asn1_get0(i);
+               if (ameth->pkey_flags & ASN1_PKEY_ALIAS)
+                       continue;
+               if ((strlen(ameth->pem_str) == len) && 
+                       !strncasecmp(ameth->pem_str, str, len))
+                       return ameth;
+               }
+       return NULL;
+       }
+
+int EVP_PKEY_asn1_add(const EVP_PKEY_ASN1_METHOD *ameth)
+       {
+       if (app_methods == NULL)
+               {
+               app_methods = sk_new((sk_cmp_fn_type *)ameth_cmp);
+               if (!app_methods)
+                       return 0;
+               }
+       if (!sk_push(app_methods, (char *)ameth))
+               return 0;
+       sk_sort(app_methods);
+       return 1;
+       }
+
+int EVP_PKEY_asn1_get0_info(int *ppkey_id, int *ppkey_base_id, int *ppkey_flags,
+                               const char **pinfo, const char **ppem_str,
+                                       const EVP_PKEY_ASN1_METHOD *ameth)
+       {
+       if (!ameth)
+               return 0;
+       if (ppkey_id)
+               *ppkey_id = ameth->pkey_id;
+       if (ppkey_base_id)
+               *ppkey_base_id = ameth->pkey_base_id;
+       if (ppkey_flags)
+               *ppkey_flags = ameth->pkey_flags;
+       if (pinfo)
+               *pinfo = ameth->info;
+       if (ppem_str)
+               *ppem_str = ameth->pem_str;
+       return 1;
+       }
+
+EVP_PKEY_ASN1_METHOD* EVP_PKEY_asn1_new(int id,
+                                       const char *pem_str, const char *info)
+       {
+       EVP_PKEY_ASN1_METHOD *ameth;
+       ameth = OPENSSL_malloc(sizeof(EVP_PKEY_ASN1_METHOD));
+       if (!ameth)
+               return NULL;
+
+       ameth->pkey_id = id;
+       ameth->pkey_base_id = id;
+       ameth->pkey_flags = ASN1_PKEY_DYNAMIC;
+
+       if (info)
+               {
+               ameth->info = BUF_strdup(info);
+               if (!ameth->info)
+                       goto err;
+               }
+
+       if (pem_str)
+               {
+               ameth->pem_str = BUF_strdup(pem_str);
+               if (!ameth->pem_str)
+                       goto err;
+               }
+       
+       ameth->pub_decode = 0;
+       ameth->pub_encode = 0;
+       ameth->pub_cmp = 0;
+       ameth->pub_print = 0;
+
+       ameth->priv_decode = 0;
+       ameth->priv_encode = 0;
+       ameth->priv_print = 0;
+
+       ameth->pkey_size = 0;
+       ameth->pkey_bits = 0;
+
+       ameth->param_decode = 0;
+       ameth->param_encode = 0;
+       ameth->param_missing = 0;
+       ameth->param_copy = 0;
+       ameth->param_cmp = 0;
+       ameth->param_print = 0;
+
+       ameth->pkey_free = 0;
+       ameth->pkey_ctrl = 0;
+
+       return ameth;
+
+       err:
+
+       EVP_PKEY_asn1_free(ameth);
+       return NULL;
+
+       }
+
+void EVP_PKEY_asn1_free(EVP_PKEY_ASN1_METHOD *ameth)
+       {
+       if (ameth && (ameth->pkey_flags & ASN1_PKEY_DYNAMIC))
+               {
+               if (ameth->pem_str)
+                       OPENSSL_free(ameth->pem_str);
+               if (ameth->info)
+                       OPENSSL_free(ameth->info);
+               OPENSSL_free(ameth);
+               }
+       }
+
+void EVP_PKEY_asn1_set_public(EVP_PKEY_ASN1_METHOD *ameth,
+               int (*pub_decode)(EVP_PKEY *pk, X509_PUBKEY *pub),
+               int (*pub_encode)(X509_PUBKEY *pub, const EVP_PKEY *pk),
+               int (*pub_cmp)(const EVP_PKEY *a, const EVP_PKEY *b),
+               int (*pub_print)(BIO *out, const EVP_PKEY *pkey, int indent,
+                                                       ASN1_PCTX *pctx),
+               int (*pkey_size)(const EVP_PKEY *pk),
+               int (*pkey_bits)(const EVP_PKEY *pk))
+       {
+       ameth->pub_decode = pub_decode;
+       ameth->pub_encode = pub_encode;
+       ameth->pub_cmp = pub_cmp;
+       ameth->pub_print = pub_print;
+       ameth->pkey_size = pkey_size;
+       ameth->pkey_bits = pkey_bits;
+       }
+
+void EVP_PKEY_asn1_set_private(EVP_PKEY_ASN1_METHOD *ameth,
+               int (*priv_decode)(EVP_PKEY *pk, PKCS8_PRIV_KEY_INFO *p8inf),
+               int (*priv_encode)(PKCS8_PRIV_KEY_INFO *p8, const EVP_PKEY *pk),
+               int (*priv_print)(BIO *out, const EVP_PKEY *pkey, int indent,
+                                                       ASN1_PCTX *pctx))
+       {
+       ameth->priv_decode = priv_decode;
+       ameth->priv_encode = priv_encode;
+       ameth->priv_print = priv_print;
+       }
+
+void EVP_PKEY_asn1_set_param(EVP_PKEY_ASN1_METHOD *ameth,
+               int (*param_decode)(EVP_PKEY *pkey,
+                               const unsigned char **pder, int derlen),
+               int (*param_encode)(const EVP_PKEY *pkey, unsigned char **pder),
+               int (*param_missing)(const EVP_PKEY *pk),
+               int (*param_copy)(EVP_PKEY *to, const EVP_PKEY *from),
+               int (*param_cmp)(const EVP_PKEY *a, const EVP_PKEY *b),
+               int (*param_print)(BIO *out, const EVP_PKEY *pkey, int indent,
+                                                       ASN1_PCTX *pctx))
+       {
+       ameth->param_decode = param_decode;
+       ameth->param_encode = param_encode;
+       ameth->param_missing = param_missing;
+       ameth->param_copy = param_copy;
+       ameth->param_cmp = param_cmp;
+       ameth->param_print = param_print;
+       }
+
+void EVP_PKEY_asn1_set_free(EVP_PKEY_ASN1_METHOD *ameth,
+               void (*pkey_free)(EVP_PKEY *pkey))
+       {
+       ameth->pkey_free = pkey_free;
+       }
+
+void EVP_PKEY_asn1_set_ctrl(EVP_PKEY_ASN1_METHOD *ameth,
+               void (*pkey_ctrl)(EVP_PKEY *pkey, int op,
+                                                       long arg1, void *arg2))
+       {
+       ameth->pkey_ctrl = pkey_ctrl;
+       }