1 /*
2 * Copyright 2019-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 <openssl/evp.h>
11 #include "internal/namemap.h"
12 #include "testutil.h"
13
14 #define NAME1 "name1"
15 #define NAME2 "name2"
16 #define ALIAS1 "alias1"
17 #define ALIAS1_UC "ALIAS1"
18
test_namemap_empty(void)19 static int test_namemap_empty(void)
20 {
21 OSSL_NAMEMAP *nm = NULL;
22 int ok;
23
24 ok = TEST_int_eq(ossl_namemap_empty(NULL), 1)
25 && TEST_ptr(nm = ossl_namemap_new())
26 && TEST_int_eq(ossl_namemap_empty(nm), 1)
27 && TEST_int_ne(ossl_namemap_add_name(nm, 0, NAME1), 0)
28 && TEST_int_eq(ossl_namemap_empty(nm), 0);
29 ossl_namemap_free(nm);
30 return ok;
31 }
32
test_namemap(OSSL_NAMEMAP * nm)33 static int test_namemap(OSSL_NAMEMAP *nm)
34 {
35 int num1 = ossl_namemap_add_name(nm, 0, NAME1);
36 int num2 = ossl_namemap_add_name(nm, 0, NAME2);
37 int num3 = ossl_namemap_add_name(nm, num1, ALIAS1);
38 int num4 = ossl_namemap_add_name(nm, 0, ALIAS1_UC);
39 int check1 = ossl_namemap_name2num(nm, NAME1);
40 int check2 = ossl_namemap_name2num(nm, NAME2);
41 int check3 = ossl_namemap_name2num(nm, ALIAS1);
42 int check4 = ossl_namemap_name2num(nm, ALIAS1_UC);
43 int false1 = ossl_namemap_name2num(nm, "cookie");
44
45 return TEST_int_ne(num1, 0)
46 && TEST_int_ne(num2, 0)
47 && TEST_int_eq(num1, num3)
48 && TEST_int_eq(num3, num4)
49 && TEST_int_eq(num1, check1)
50 && TEST_int_eq(num2, check2)
51 && TEST_int_eq(num3, check3)
52 && TEST_int_eq(num4, check4)
53 && TEST_int_eq(false1, 0);
54 }
55
test_namemap_independent(void)56 static int test_namemap_independent(void)
57 {
58 OSSL_NAMEMAP *nm = ossl_namemap_new();
59 int ok = TEST_ptr(nm) && test_namemap(nm);
60
61 ossl_namemap_free(nm);
62 return ok;
63 }
64
test_namemap_stored(void)65 static int test_namemap_stored(void)
66 {
67 OSSL_NAMEMAP *nm = ossl_namemap_stored(NULL);
68
69 return TEST_ptr(nm)
70 && test_namemap(nm);
71 }
72
73 /*
74 * Test that EVP_get_digestbyname() will use the namemap when it can't find
75 * entries in the legacy method database.
76 */
test_digestbyname(void)77 static int test_digestbyname(void)
78 {
79 int id;
80 OSSL_NAMEMAP *nm = ossl_namemap_stored(NULL);
81 const EVP_MD *sha256, *foo;
82
83 if (!TEST_ptr(nm))
84 return 0;
85 id = ossl_namemap_add_name(nm, 0, "SHA256");
86 if (!TEST_int_ne(id, 0))
87 return 0;
88 if (!TEST_int_eq(ossl_namemap_add_name(nm, id, "foo"), id))
89 return 0;
90
91 sha256 = EVP_get_digestbyname("SHA256");
92 if (!TEST_ptr(sha256))
93 return 0;
94 foo = EVP_get_digestbyname("foo");
95 if (!TEST_ptr_eq(sha256, foo))
96 return 0;
97
98 return 1;
99 }
100
101 /*
102 * Test that EVP_get_cipherbyname() will use the namemap when it can't find
103 * entries in the legacy method database.
104 */
test_cipherbyname(void)105 static int test_cipherbyname(void)
106 {
107 int id;
108 OSSL_NAMEMAP *nm = ossl_namemap_stored(NULL);
109 const EVP_CIPHER *aes128, *bar;
110
111 if (!TEST_ptr(nm))
112 return 0;
113 id = ossl_namemap_add_name(nm, 0, "AES-128-CBC");
114 if (!TEST_int_ne(id, 0))
115 return 0;
116 if (!TEST_int_eq(ossl_namemap_add_name(nm, id, "bar"), id))
117 return 0;
118
119 aes128 = EVP_get_cipherbyname("AES-128-CBC");
120 if (!TEST_ptr(aes128))
121 return 0;
122 bar = EVP_get_cipherbyname("bar");
123 if (!TEST_ptr_eq(aes128, bar))
124 return 0;
125
126 return 1;
127 }
128
129 /*
130 * Test that EVP_CIPHER_is_a() responds appropriately, even for ciphers that
131 * are entirely legacy.
132 */
test_cipher_is_a(void)133 static int test_cipher_is_a(void)
134 {
135 EVP_CIPHER *fetched = EVP_CIPHER_fetch(NULL, "AES-256-CCM", NULL);
136 int rv = 1;
137
138 if (!TEST_ptr(fetched))
139 return 0;
140 if (!TEST_true(EVP_CIPHER_is_a(fetched, "id-aes256-CCM"))
141 || !TEST_false(EVP_CIPHER_is_a(fetched, "AES-128-GCM")))
142 rv = 0;
143 if (!TEST_true(EVP_CIPHER_is_a(EVP_aes_256_gcm(), "AES-256-GCM"))
144 || !TEST_false(EVP_CIPHER_is_a(EVP_aes_256_gcm(), "AES-128-CCM")))
145 rv = 0;
146
147 EVP_CIPHER_free(fetched);
148 return rv;
149 }
150
151 /*
152 * Test that EVP_MD_is_a() responds appropriately, even for MDs that are
153 * entirely legacy.
154 */
test_digest_is_a(void)155 static int test_digest_is_a(void)
156 {
157 EVP_MD *fetched = EVP_MD_fetch(NULL, "SHA2-512", NULL);
158 int rv = 1;
159
160 if (!TEST_ptr(fetched))
161 return 0;
162 if (!TEST_true(EVP_MD_is_a(fetched, "SHA512"))
163 || !TEST_false(EVP_MD_is_a(fetched, "SHA1")))
164 rv = 0;
165 if (!TEST_true(EVP_MD_is_a(EVP_sha256(), "SHA2-256"))
166 || !TEST_false(EVP_MD_is_a(EVP_sha256(), "SHA3-256")))
167 rv = 0;
168
169 EVP_MD_free(fetched);
170 return rv;
171 }
172
setup_tests(void)173 int setup_tests(void)
174 {
175 ADD_TEST(test_namemap_empty);
176 ADD_TEST(test_namemap_independent);
177 ADD_TEST(test_namemap_stored);
178 ADD_TEST(test_digestbyname);
179 ADD_TEST(test_cipherbyname);
180 ADD_TEST(test_digest_is_a);
181 ADD_TEST(test_cipher_is_a);
182 return 1;
183 }
184