1/*
2 * Copyright 2020-2022 The OpenSSL Project Authors. All Rights Reserved.
3 *
4 * Licensed under the Apache License 2.0 (the "License").  You may not use
5 * this file except in compliance with the License.  You can obtain a copy
6 * in the file LICENSE in the source distribution or at
7 * https://www.openssl.org/source/license.html
8 */
9
10#include <openssl/core.h>
11#include <openssl/buffer.h>
12#include "internal/asn1.h"
13#include "prov/bio.h"
14#include "endecoder_local.h"
15
16OSSL_FUNC_keymgmt_new_fn *
17ossl_prov_get_keymgmt_new(const OSSL_DISPATCH *fns)
18{
19    /* Pilfer the keymgmt dispatch table */
20    for (; fns->function_id != 0; fns++)
21        if (fns->function_id == OSSL_FUNC_KEYMGMT_NEW)
22            return OSSL_FUNC_keymgmt_new(fns);
23
24    return NULL;
25}
26
27OSSL_FUNC_keymgmt_free_fn *
28ossl_prov_get_keymgmt_free(const OSSL_DISPATCH *fns)
29{
30    /* Pilfer the keymgmt dispatch table */
31    for (; fns->function_id != 0; fns++)
32        if (fns->function_id == OSSL_FUNC_KEYMGMT_FREE)
33            return OSSL_FUNC_keymgmt_free(fns);
34
35    return NULL;
36}
37
38OSSL_FUNC_keymgmt_import_fn *
39ossl_prov_get_keymgmt_import(const OSSL_DISPATCH *fns)
40{
41    /* Pilfer the keymgmt dispatch table */
42    for (; fns->function_id != 0; fns++)
43        if (fns->function_id == OSSL_FUNC_KEYMGMT_IMPORT)
44            return OSSL_FUNC_keymgmt_import(fns);
45
46    return NULL;
47}
48
49OSSL_FUNC_keymgmt_export_fn *
50ossl_prov_get_keymgmt_export(const OSSL_DISPATCH *fns)
51{
52    /* Pilfer the keymgmt dispatch table */
53    for (; fns->function_id != 0; fns++)
54        if (fns->function_id == OSSL_FUNC_KEYMGMT_EXPORT)
55            return OSSL_FUNC_keymgmt_export(fns);
56
57    return NULL;
58}
59
60void *ossl_prov_import_key(const OSSL_DISPATCH *fns, void *provctx,
61                           int selection, const OSSL_PARAM params[])
62{
63    OSSL_FUNC_keymgmt_new_fn *kmgmt_new = ossl_prov_get_keymgmt_new(fns);
64    OSSL_FUNC_keymgmt_free_fn *kmgmt_free = ossl_prov_get_keymgmt_free(fns);
65    OSSL_FUNC_keymgmt_import_fn *kmgmt_import =
66        ossl_prov_get_keymgmt_import(fns);
67    void *key = NULL;
68
69    if (kmgmt_new != NULL && kmgmt_import != NULL && kmgmt_free != NULL) {
70        if ((key = kmgmt_new(provctx)) == NULL
71            || !kmgmt_import(key, selection, params)) {
72            kmgmt_free(key);
73            key = NULL;
74        }
75    }
76    return key;
77}
78
79void ossl_prov_free_key(const OSSL_DISPATCH *fns, void *key)
80{
81    OSSL_FUNC_keymgmt_free_fn *kmgmt_free = ossl_prov_get_keymgmt_free(fns);
82
83    if (kmgmt_free != NULL)
84        kmgmt_free(key);
85}
86
87int ossl_read_der(PROV_CTX *provctx, OSSL_CORE_BIO *cin,  unsigned char **data,
88                  long *len)
89{
90    BUF_MEM *mem = NULL;
91    BIO *in = ossl_bio_new_from_core_bio(provctx, cin);
92    int ok;
93
94    if (in == NULL)
95        return 0;
96    ok = (asn1_d2i_read_bio(in, &mem) >= 0);
97    if (ok) {
98        *data = (unsigned char *)mem->data;
99        *len = (long)mem->length;
100        OPENSSL_free(mem);
101    }
102    BIO_free(in);
103    return ok;
104}
105