1 /*
2 * Copyright 2020-2021 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 <string.h>
11 #include <limits.h>
12 #include <openssl/store.h>
13 #include <openssl/ui.h>
14 #include "testutil.h"
15
16 #ifndef PATH_MAX
17 # if defined(_WIN32) && defined(_MAX_PATH)
18 # define PATH_MAX _MAX_PATH
19 # else
20 # define PATH_MAX 4096
21 # endif
22 #endif
23
24 typedef enum OPTION_choice {
25 OPT_ERR = -1,
26 OPT_EOF = 0,
27 OPT_INPUTDIR,
28 OPT_INFILE,
29 OPT_SM2FILE,
30 OPT_DATADIR,
31 OPT_TEST_ENUM
32 } OPTION_CHOICE;
33
34 static const char *inputdir = NULL;
35 static const char *infile = NULL;
36 static const char *sm2file = NULL;
37 static const char *datadir = NULL;
38
test_store_open(void)39 static int test_store_open(void)
40 {
41 int ret = 0;
42 OSSL_STORE_CTX *sctx = NULL;
43 OSSL_STORE_SEARCH *search = NULL;
44 UI_METHOD *ui_method = NULL;
45 char *input = test_mk_file_path(inputdir, infile);
46
47 ret = TEST_ptr(input)
48 && TEST_ptr(search = OSSL_STORE_SEARCH_by_alias("nothing"))
49 && TEST_ptr(ui_method= UI_create_method("DummyUI"))
50 && TEST_ptr(sctx = OSSL_STORE_open_ex(input, NULL, NULL, ui_method,
51 NULL, NULL, NULL, NULL))
52 && TEST_false(OSSL_STORE_find(sctx, NULL))
53 && TEST_true(OSSL_STORE_find(sctx, search));
54 UI_destroy_method(ui_method);
55 OSSL_STORE_SEARCH_free(search);
56 OSSL_STORE_close(sctx);
57 OPENSSL_free(input);
58 return ret;
59 }
60
test_store_search_by_key_fingerprint_fail(void)61 static int test_store_search_by_key_fingerprint_fail(void)
62 {
63 int ret;
64 OSSL_STORE_SEARCH *search = NULL;
65
66 ret = TEST_ptr_null(search = OSSL_STORE_SEARCH_by_key_fingerprint(
67 EVP_sha256(), NULL, 0));
68 OSSL_STORE_SEARCH_free(search);
69 return ret;
70 }
71
get_params(const char * uri,const char * type)72 static int get_params(const char *uri, const char *type)
73 {
74 EVP_PKEY *pkey = NULL;
75 OSSL_STORE_CTX *ctx = NULL;
76 OSSL_STORE_INFO *info;
77 int ret = 0;
78
79 ctx = OSSL_STORE_open_ex(uri, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
80 if (!TEST_ptr(ctx))
81 goto err;
82
83 while (!OSSL_STORE_eof(ctx)
84 && (info = OSSL_STORE_load(ctx)) != NULL
85 && pkey == NULL) {
86 if (OSSL_STORE_INFO_get_type(info) == OSSL_STORE_INFO_PARAMS) {
87 pkey = OSSL_STORE_INFO_get1_PARAMS(info);
88 }
89 OSSL_STORE_INFO_free(info);
90 info = NULL;
91 }
92
93 if (pkey != NULL)
94 ret = EVP_PKEY_is_a(pkey, type);
95 EVP_PKEY_free(pkey);
96
97 err:
98 OSSL_STORE_close(ctx);
99 return ret;
100 }
101
test_store_get_params(int idx)102 static int test_store_get_params(int idx)
103 {
104 const char *type;
105 const char *urifmt;
106 char uri[PATH_MAX];
107
108 switch(idx) {
109 #ifndef OPENSSL_NO_DH
110 case 0:
111 type = "DH";
112 break;
113 case 1:
114 type = "DHX";
115 break;
116 #else
117 case 0:
118 case 1:
119 return 1;
120 #endif
121 case 2:
122 #ifndef OPENSSL_NO_DSA
123 type = "DSA";
124 break;
125 #else
126 return 1;
127 #endif
128 default:
129 TEST_error("Invalid test index");
130 return 0;
131 }
132
133 urifmt = "%s/%s-params.pem";
134 #ifdef __VMS
135 {
136 char datadir_end = datadir[strlen(datadir) - 1];
137
138 if (datadir_end == ':' || datadir_end == ']' || datadir_end == '>')
139 urifmt = "%s%s-params.pem";
140 }
141 #endif
142 if (!TEST_true(BIO_snprintf(uri, sizeof(uri), urifmt, datadir, type)))
143 return 0;
144
145 TEST_info("Testing uri: %s", uri);
146 if (!TEST_true(get_params(uri, type)))
147 return 0;
148
149 return 1;
150 }
151
152 /*
153 * This test verifies that calling OSSL_STORE_ATTACH does not set an
154 * "unregistered scheme" error when called.
155 */
test_store_attach_unregistered_scheme(void)156 static int test_store_attach_unregistered_scheme(void)
157 {
158 int ret;
159 OSSL_STORE_CTX *store_ctx = NULL;
160 OSSL_PROVIDER *provider = NULL;
161 OSSL_LIB_CTX *libctx = NULL;
162 BIO *bio = NULL;
163 char *input = test_mk_file_path(inputdir, sm2file);
164
165 ret = TEST_ptr(input)
166 && TEST_ptr(libctx = OSSL_LIB_CTX_new())
167 && TEST_ptr(provider = OSSL_PROVIDER_load(libctx, "default"))
168 && TEST_ptr(bio = BIO_new_file(input, "r"))
169 && TEST_ptr(store_ctx = OSSL_STORE_attach(bio, "file", libctx, NULL,
170 NULL, NULL, NULL, NULL, NULL))
171 && TEST_int_ne(ERR_GET_LIB(ERR_peek_error()), ERR_LIB_OSSL_STORE)
172 && TEST_int_ne(ERR_GET_REASON(ERR_peek_error()),
173 OSSL_STORE_R_UNREGISTERED_SCHEME);
174
175 BIO_free(bio);
176 OSSL_STORE_close(store_ctx);
177 OSSL_PROVIDER_unload(provider);
178 OSSL_LIB_CTX_free(libctx);
179 OPENSSL_free(input);
180 return ret;
181 }
182
test_get_options(void)183 const OPTIONS *test_get_options(void)
184 {
185 static const OPTIONS test_options[] = {
186 OPT_TEST_OPTIONS_DEFAULT_USAGE,
187 { "dir", OPT_INPUTDIR, '/' },
188 { "in", OPT_INFILE, '<' },
189 { "sm2", OPT_SM2FILE, '<' },
190 { "data", OPT_DATADIR, 's' },
191 { NULL }
192 };
193 return test_options;
194 }
195
setup_tests(void)196 int setup_tests(void)
197 {
198 OPTION_CHOICE o;
199
200 while ((o = opt_next()) != OPT_EOF) {
201 switch (o) {
202 case OPT_INPUTDIR:
203 inputdir = opt_arg();
204 break;
205 case OPT_INFILE:
206 infile = opt_arg();
207 break;
208 case OPT_SM2FILE:
209 sm2file = opt_arg();
210 break;
211 case OPT_DATADIR:
212 datadir = opt_arg();
213 break;
214 case OPT_TEST_CASES:
215 break;
216 default:
217 case OPT_ERR:
218 return 0;
219 }
220 }
221
222 if (datadir == NULL) {
223 TEST_error("No data directory specified");
224 return 0;
225 }
226 if (inputdir == NULL) {
227 TEST_error("No input directory specified");
228 return 0;
229 }
230
231 if (infile != NULL)
232 ADD_TEST(test_store_open);
233 ADD_TEST(test_store_search_by_key_fingerprint_fail);
234 ADD_ALL_TESTS(test_store_get_params, 3);
235 if (sm2file != NULL)
236 ADD_TEST(test_store_attach_unregistered_scheme);
237 return 1;
238 }
239