1*b0d17251Schristos /*
2*b0d17251Schristos * Copyright 2021-2022 The OpenSSL Project Authors. All Rights Reserved.
3*b0d17251Schristos *
4*b0d17251Schristos * Licensed under the Apache License 2.0 (the "License"). You may not use
5*b0d17251Schristos * this file except in compliance with the License. You can obtain a copy
6*b0d17251Schristos * in the file LICENSE in the source distribution or at
7*b0d17251Schristos * https://www.openssl.org/source/license.html
8*b0d17251Schristos */
9*b0d17251Schristos
10*b0d17251Schristos #include <stddef.h>
11*b0d17251Schristos #include <string.h>
12*b0d17251Schristos #include <openssl/provider.h>
13*b0d17251Schristos #include <openssl/params.h>
14*b0d17251Schristos #include <openssl/core_names.h>
15*b0d17251Schristos #include <openssl/evp.h>
16*b0d17251Schristos #include <openssl/store.h>
17*b0d17251Schristos #include "testutil.h"
18*b0d17251Schristos #include "fake_rsaprov.h"
19*b0d17251Schristos
20*b0d17251Schristos static OSSL_LIB_CTX *libctx = NULL;
21*b0d17251Schristos
22*b0d17251Schristos /* Fetch SIGNATURE method using a libctx and propq */
fetch_sig(OSSL_LIB_CTX * ctx,const char * alg,const char * propq,OSSL_PROVIDER * expected_prov)23*b0d17251Schristos static int fetch_sig(OSSL_LIB_CTX *ctx, const char *alg, const char *propq,
24*b0d17251Schristos OSSL_PROVIDER *expected_prov)
25*b0d17251Schristos {
26*b0d17251Schristos OSSL_PROVIDER *prov;
27*b0d17251Schristos EVP_SIGNATURE *sig = EVP_SIGNATURE_fetch(ctx, "RSA", propq);
28*b0d17251Schristos int ret = 0;
29*b0d17251Schristos
30*b0d17251Schristos if (!TEST_ptr(sig))
31*b0d17251Schristos return 0;
32*b0d17251Schristos
33*b0d17251Schristos if (!TEST_ptr(prov = EVP_SIGNATURE_get0_provider(sig)))
34*b0d17251Schristos goto end;
35*b0d17251Schristos
36*b0d17251Schristos if (!TEST_ptr_eq(prov, expected_prov)) {
37*b0d17251Schristos TEST_info("Fetched provider: %s, Expected provider: %s",
38*b0d17251Schristos OSSL_PROVIDER_get0_name(prov),
39*b0d17251Schristos OSSL_PROVIDER_get0_name(expected_prov));
40*b0d17251Schristos goto end;
41*b0d17251Schristos }
42*b0d17251Schristos
43*b0d17251Schristos ret = 1;
44*b0d17251Schristos end:
45*b0d17251Schristos EVP_SIGNATURE_free(sig);
46*b0d17251Schristos return ret;
47*b0d17251Schristos }
48*b0d17251Schristos
49*b0d17251Schristos
test_pkey_sig(void)50*b0d17251Schristos static int test_pkey_sig(void)
51*b0d17251Schristos {
52*b0d17251Schristos OSSL_PROVIDER *deflt = NULL;
53*b0d17251Schristos OSSL_PROVIDER *fake_rsa = NULL;
54*b0d17251Schristos int i, ret = 0;
55*b0d17251Schristos EVP_PKEY *pkey = NULL;
56*b0d17251Schristos EVP_PKEY_CTX *ctx = NULL;
57*b0d17251Schristos
58*b0d17251Schristos if (!TEST_ptr(fake_rsa = fake_rsa_start(libctx)))
59*b0d17251Schristos return 0;
60*b0d17251Schristos
61*b0d17251Schristos if (!TEST_ptr(deflt = OSSL_PROVIDER_load(libctx, "default")))
62*b0d17251Schristos goto end;
63*b0d17251Schristos
64*b0d17251Schristos /* Do a direct fetch to see it works */
65*b0d17251Schristos if (!TEST_true(fetch_sig(libctx, "RSA", "provider=fake-rsa", fake_rsa))
66*b0d17251Schristos || !TEST_true(fetch_sig(libctx, "RSA", "?provider=fake-rsa", fake_rsa)))
67*b0d17251Schristos goto end;
68*b0d17251Schristos
69*b0d17251Schristos /* Construct a pkey using precise propq to use our provider */
70*b0d17251Schristos if (!TEST_ptr(ctx = EVP_PKEY_CTX_new_from_name(libctx, "RSA",
71*b0d17251Schristos "provider=fake-rsa"))
72*b0d17251Schristos || !TEST_true(EVP_PKEY_fromdata_init(ctx))
73*b0d17251Schristos || !TEST_true(EVP_PKEY_fromdata(ctx, &pkey, EVP_PKEY_KEYPAIR, NULL))
74*b0d17251Schristos || !TEST_ptr(pkey))
75*b0d17251Schristos goto end;
76*b0d17251Schristos
77*b0d17251Schristos EVP_PKEY_CTX_free(ctx);
78*b0d17251Schristos ctx = NULL;
79*b0d17251Schristos
80*b0d17251Schristos /* try exercising signature_init ops a few times */
81*b0d17251Schristos for (i = 0; i < 3; i++) {
82*b0d17251Schristos size_t siglen;
83*b0d17251Schristos
84*b0d17251Schristos /*
85*b0d17251Schristos * Create a signing context for our pkey with optional propq.
86*b0d17251Schristos * The sign init should pick both keymgmt and signature from
87*b0d17251Schristos * fake-rsa as the key is not exportable.
88*b0d17251Schristos */
89*b0d17251Schristos if (!TEST_ptr(ctx = EVP_PKEY_CTX_new_from_pkey(libctx, pkey,
90*b0d17251Schristos "?provider=default")))
91*b0d17251Schristos goto end;
92*b0d17251Schristos
93*b0d17251Schristos /*
94*b0d17251Schristos * If this picks the wrong signature without realizing it
95*b0d17251Schristos * we can get a segfault or some internal error. At least watch
96*b0d17251Schristos * whether fake-rsa sign_init is is exercised by calling sign.
97*b0d17251Schristos */
98*b0d17251Schristos if (!TEST_int_eq(EVP_PKEY_sign_init(ctx), 1))
99*b0d17251Schristos goto end;
100*b0d17251Schristos
101*b0d17251Schristos if (!TEST_int_eq(EVP_PKEY_sign(ctx, NULL, &siglen, NULL, 0), 1)
102*b0d17251Schristos || !TEST_size_t_eq(siglen, 256))
103*b0d17251Schristos goto end;
104*b0d17251Schristos
105*b0d17251Schristos EVP_PKEY_CTX_free(ctx);
106*b0d17251Schristos ctx = NULL;
107*b0d17251Schristos }
108*b0d17251Schristos
109*b0d17251Schristos ret = 1;
110*b0d17251Schristos
111*b0d17251Schristos end:
112*b0d17251Schristos fake_rsa_finish(fake_rsa);
113*b0d17251Schristos OSSL_PROVIDER_unload(deflt);
114*b0d17251Schristos EVP_PKEY_CTX_free(ctx);
115*b0d17251Schristos EVP_PKEY_free(pkey);
116*b0d17251Schristos return ret;
117*b0d17251Schristos }
118*b0d17251Schristos
test_alternative_keygen_init(void)119*b0d17251Schristos static int test_alternative_keygen_init(void)
120*b0d17251Schristos {
121*b0d17251Schristos EVP_PKEY_CTX *ctx = NULL;
122*b0d17251Schristos OSSL_PROVIDER *deflt = NULL;
123*b0d17251Schristos OSSL_PROVIDER *fake_rsa = NULL;
124*b0d17251Schristos const OSSL_PROVIDER *provider;
125*b0d17251Schristos const char *provname;
126*b0d17251Schristos int ret = 0;
127*b0d17251Schristos
128*b0d17251Schristos if (!TEST_ptr(deflt = OSSL_PROVIDER_load(libctx, "default")))
129*b0d17251Schristos goto end;
130*b0d17251Schristos
131*b0d17251Schristos /* first try without the fake RSA provider loaded */
132*b0d17251Schristos if (!TEST_ptr(ctx = EVP_PKEY_CTX_new_from_name(libctx, "RSA", NULL)))
133*b0d17251Schristos goto end;
134*b0d17251Schristos
135*b0d17251Schristos if (!TEST_int_gt(EVP_PKEY_keygen_init(ctx), 0))
136*b0d17251Schristos goto end;
137*b0d17251Schristos
138*b0d17251Schristos if (!TEST_ptr(provider = EVP_PKEY_CTX_get0_provider(ctx)))
139*b0d17251Schristos goto end;
140*b0d17251Schristos
141*b0d17251Schristos if (!TEST_ptr(provname = OSSL_PROVIDER_get0_name(provider)))
142*b0d17251Schristos goto end;
143*b0d17251Schristos
144*b0d17251Schristos if (!TEST_str_eq(provname, "default"))
145*b0d17251Schristos goto end;
146*b0d17251Schristos
147*b0d17251Schristos EVP_PKEY_CTX_free(ctx);
148*b0d17251Schristos ctx = NULL;
149*b0d17251Schristos
150*b0d17251Schristos /* now load fake RSA and try again */
151*b0d17251Schristos if (!TEST_ptr(fake_rsa = fake_rsa_start(libctx)))
152*b0d17251Schristos return 0;
153*b0d17251Schristos
154*b0d17251Schristos if (!TEST_ptr(ctx = EVP_PKEY_CTX_new_from_name(libctx, "RSA",
155*b0d17251Schristos "?provider=fake-rsa")))
156*b0d17251Schristos goto end;
157*b0d17251Schristos
158*b0d17251Schristos if (!TEST_int_gt(EVP_PKEY_keygen_init(ctx), 0))
159*b0d17251Schristos goto end;
160*b0d17251Schristos
161*b0d17251Schristos if (!TEST_ptr(provider = EVP_PKEY_CTX_get0_provider(ctx)))
162*b0d17251Schristos goto end;
163*b0d17251Schristos
164*b0d17251Schristos if (!TEST_ptr(provname = OSSL_PROVIDER_get0_name(provider)))
165*b0d17251Schristos goto end;
166*b0d17251Schristos
167*b0d17251Schristos if (!TEST_str_eq(provname, "fake-rsa"))
168*b0d17251Schristos goto end;
169*b0d17251Schristos
170*b0d17251Schristos ret = 1;
171*b0d17251Schristos
172*b0d17251Schristos end:
173*b0d17251Schristos fake_rsa_finish(fake_rsa);
174*b0d17251Schristos OSSL_PROVIDER_unload(deflt);
175*b0d17251Schristos EVP_PKEY_CTX_free(ctx);
176*b0d17251Schristos return ret;
177*b0d17251Schristos }
178*b0d17251Schristos
test_pkey_eq(void)179*b0d17251Schristos static int test_pkey_eq(void)
180*b0d17251Schristos {
181*b0d17251Schristos OSSL_PROVIDER *deflt = NULL;
182*b0d17251Schristos OSSL_PROVIDER *fake_rsa = NULL;
183*b0d17251Schristos EVP_PKEY *pkey_fake = NULL;
184*b0d17251Schristos EVP_PKEY *pkey_dflt = NULL;
185*b0d17251Schristos EVP_PKEY_CTX *ctx = NULL;
186*b0d17251Schristos OSSL_PARAM *params = NULL;
187*b0d17251Schristos int ret = 0;
188*b0d17251Schristos
189*b0d17251Schristos if (!TEST_ptr(fake_rsa = fake_rsa_start(libctx)))
190*b0d17251Schristos return 0;
191*b0d17251Schristos
192*b0d17251Schristos if (!TEST_ptr(deflt = OSSL_PROVIDER_load(libctx, "default")))
193*b0d17251Schristos goto end;
194*b0d17251Schristos
195*b0d17251Schristos /* Construct a public key for fake-rsa */
196*b0d17251Schristos if (!TEST_ptr(params = fake_rsa_key_params(0))
197*b0d17251Schristos || !TEST_ptr(ctx = EVP_PKEY_CTX_new_from_name(libctx, "RSA",
198*b0d17251Schristos "provider=fake-rsa"))
199*b0d17251Schristos || !TEST_true(EVP_PKEY_fromdata_init(ctx))
200*b0d17251Schristos || !TEST_true(EVP_PKEY_fromdata(ctx, &pkey_fake, EVP_PKEY_PUBLIC_KEY,
201*b0d17251Schristos params))
202*b0d17251Schristos || !TEST_ptr(pkey_fake))
203*b0d17251Schristos goto end;
204*b0d17251Schristos
205*b0d17251Schristos EVP_PKEY_CTX_free(ctx);
206*b0d17251Schristos ctx = NULL;
207*b0d17251Schristos OSSL_PARAM_free(params);
208*b0d17251Schristos params = NULL;
209*b0d17251Schristos
210*b0d17251Schristos /* Construct a public key for default */
211*b0d17251Schristos if (!TEST_ptr(params = fake_rsa_key_params(0))
212*b0d17251Schristos || !TEST_ptr(ctx = EVP_PKEY_CTX_new_from_name(libctx, "RSA",
213*b0d17251Schristos "provider=default"))
214*b0d17251Schristos || !TEST_true(EVP_PKEY_fromdata_init(ctx))
215*b0d17251Schristos || !TEST_true(EVP_PKEY_fromdata(ctx, &pkey_dflt, EVP_PKEY_PUBLIC_KEY,
216*b0d17251Schristos params))
217*b0d17251Schristos || !TEST_ptr(pkey_dflt))
218*b0d17251Schristos goto end;
219*b0d17251Schristos
220*b0d17251Schristos EVP_PKEY_CTX_free(ctx);
221*b0d17251Schristos ctx = NULL;
222*b0d17251Schristos OSSL_PARAM_free(params);
223*b0d17251Schristos params = NULL;
224*b0d17251Schristos
225*b0d17251Schristos /* now test for equality */
226*b0d17251Schristos if (!TEST_int_eq(EVP_PKEY_eq(pkey_fake, pkey_dflt), 1))
227*b0d17251Schristos goto end;
228*b0d17251Schristos
229*b0d17251Schristos ret = 1;
230*b0d17251Schristos end:
231*b0d17251Schristos fake_rsa_finish(fake_rsa);
232*b0d17251Schristos OSSL_PROVIDER_unload(deflt);
233*b0d17251Schristos EVP_PKEY_CTX_free(ctx);
234*b0d17251Schristos EVP_PKEY_free(pkey_fake);
235*b0d17251Schristos EVP_PKEY_free(pkey_dflt);
236*b0d17251Schristos OSSL_PARAM_free(params);
237*b0d17251Schristos return ret;
238*b0d17251Schristos }
239*b0d17251Schristos
test_pkey_store(int idx)240*b0d17251Schristos static int test_pkey_store(int idx)
241*b0d17251Schristos {
242*b0d17251Schristos OSSL_PROVIDER *deflt = NULL;
243*b0d17251Schristos OSSL_PROVIDER *fake_rsa = NULL;
244*b0d17251Schristos int ret = 0;
245*b0d17251Schristos EVP_PKEY *pkey = NULL;
246*b0d17251Schristos OSSL_STORE_LOADER *loader = NULL;
247*b0d17251Schristos OSSL_STORE_CTX *ctx = NULL;
248*b0d17251Schristos OSSL_STORE_INFO *info;
249*b0d17251Schristos const char *propq = idx == 0 ? "?provider=fake-rsa"
250*b0d17251Schristos : "?provider=default";
251*b0d17251Schristos
252*b0d17251Schristos /* It's important to load the default provider first for this test */
253*b0d17251Schristos if (!TEST_ptr(deflt = OSSL_PROVIDER_load(libctx, "default")))
254*b0d17251Schristos goto end;
255*b0d17251Schristos
256*b0d17251Schristos if (!TEST_ptr(fake_rsa = fake_rsa_start(libctx)))
257*b0d17251Schristos goto end;
258*b0d17251Schristos
259*b0d17251Schristos if (!TEST_ptr(loader = OSSL_STORE_LOADER_fetch(libctx, "fake_rsa",
260*b0d17251Schristos propq)))
261*b0d17251Schristos goto end;
262*b0d17251Schristos
263*b0d17251Schristos OSSL_STORE_LOADER_free(loader);
264*b0d17251Schristos
265*b0d17251Schristos if (!TEST_ptr(ctx = OSSL_STORE_open_ex("fake_rsa:test", libctx, propq,
266*b0d17251Schristos NULL, NULL, NULL, NULL, NULL)))
267*b0d17251Schristos goto end;
268*b0d17251Schristos
269*b0d17251Schristos while (!OSSL_STORE_eof(ctx)
270*b0d17251Schristos && (info = OSSL_STORE_load(ctx)) != NULL
271*b0d17251Schristos && pkey == NULL) {
272*b0d17251Schristos if (OSSL_STORE_INFO_get_type(info) == OSSL_STORE_INFO_PKEY)
273*b0d17251Schristos pkey = OSSL_STORE_INFO_get1_PKEY(info);
274*b0d17251Schristos OSSL_STORE_INFO_free(info);
275*b0d17251Schristos info = NULL;
276*b0d17251Schristos }
277*b0d17251Schristos
278*b0d17251Schristos if (!TEST_ptr(pkey) || !TEST_int_eq(EVP_PKEY_is_a(pkey, "RSA"), 1))
279*b0d17251Schristos goto end;
280*b0d17251Schristos
281*b0d17251Schristos ret = 1;
282*b0d17251Schristos
283*b0d17251Schristos end:
284*b0d17251Schristos fake_rsa_finish(fake_rsa);
285*b0d17251Schristos OSSL_PROVIDER_unload(deflt);
286*b0d17251Schristos OSSL_STORE_close(ctx);
287*b0d17251Schristos EVP_PKEY_free(pkey);
288*b0d17251Schristos return ret;
289*b0d17251Schristos }
290*b0d17251Schristos
setup_tests(void)291*b0d17251Schristos int setup_tests(void)
292*b0d17251Schristos {
293*b0d17251Schristos libctx = OSSL_LIB_CTX_new();
294*b0d17251Schristos if (libctx == NULL)
295*b0d17251Schristos return 0;
296*b0d17251Schristos
297*b0d17251Schristos ADD_TEST(test_pkey_sig);
298*b0d17251Schristos ADD_TEST(test_alternative_keygen_init);
299*b0d17251Schristos ADD_TEST(test_pkey_eq);
300*b0d17251Schristos ADD_ALL_TESTS(test_pkey_store, 2);
301*b0d17251Schristos
302*b0d17251Schristos return 1;
303*b0d17251Schristos }
304*b0d17251Schristos
cleanup_tests(void)305*b0d17251Schristos void cleanup_tests(void)
306*b0d17251Schristos {
307*b0d17251Schristos OSSL_LIB_CTX_free(libctx);
308*b0d17251Schristos }
309