]>
Commit | Line | Data |
---|---|---|
e14d6cf6 MC |
1 | /* |
2 | * Copyright 2017-2018 The OpenSSL Project Authors. All Rights Reserved. | |
3 | * | |
4 | * Licensed under the OpenSSL license (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 <stdio.h> | |
11 | #include <stdlib.h> | |
12 | #include <string.h> | |
13 | ||
14 | #include <openssl/bio.h> | |
15 | #include <openssl/evp.h> | |
16 | #include <openssl/bn.h> | |
17 | #include <openssl/crypto.h> | |
18 | #include <openssl/err.h> | |
19 | #include <openssl/rand.h> | |
20 | #include "testutil.h" | |
21 | ||
22 | #ifndef OPENSSL_NO_SM2 | |
23 | ||
24 | # include "internal/sm2.h" | |
25 | ||
26 | static RAND_METHOD fake_rand; | |
27 | static const RAND_METHOD *saved_rand; | |
28 | ||
29 | static uint8_t *fake_rand_bytes = NULL; | |
30 | static size_t fake_rand_bytes_offset = 0; | |
31 | ||
32 | static int get_faked_bytes(unsigned char *buf, int num) | |
33 | { | |
34 | int i; | |
35 | ||
36 | if (fake_rand_bytes == NULL) | |
37 | return saved_rand->bytes(buf, num); | |
38 | ||
39 | for (i = 0; i != num; ++i) | |
40 | buf[i] = fake_rand_bytes[fake_rand_bytes_offset + i]; | |
41 | fake_rand_bytes_offset += num; | |
42 | return 1; | |
43 | } | |
44 | ||
45 | static int start_fake_rand(const char *hex_bytes) | |
46 | { | |
47 | /* save old rand method */ | |
48 | if (!TEST_ptr(saved_rand = RAND_get_rand_method())) | |
49 | return 0; | |
50 | ||
51 | fake_rand = *saved_rand; | |
52 | /* use own random function */ | |
53 | fake_rand.bytes = get_faked_bytes; | |
54 | ||
55 | fake_rand_bytes = OPENSSL_hexstr2buf(hex_bytes, NULL); | |
56 | fake_rand_bytes_offset = 0; | |
57 | ||
58 | /* set new RAND_METHOD */ | |
59 | if (!TEST_true(RAND_set_rand_method(&fake_rand))) | |
60 | return 0; | |
61 | return 1; | |
62 | } | |
63 | ||
64 | static int restore_rand(void) | |
65 | { | |
66 | OPENSSL_free(fake_rand_bytes); | |
67 | fake_rand_bytes = NULL; | |
68 | fake_rand_bytes_offset = 0; | |
69 | if (!TEST_true(RAND_set_rand_method(saved_rand))) | |
70 | return 0; | |
71 | return 1; | |
72 | } | |
73 | ||
74 | static EC_GROUP *create_EC_group(const char *p_hex, const char *a_hex, | |
75 | const char *b_hex, const char *x_hex, | |
76 | const char *y_hex, const char *order_hex, | |
77 | const char *cof_hex) | |
78 | { | |
79 | BIGNUM *p = NULL; | |
80 | BIGNUM *a = NULL; | |
81 | BIGNUM *b = NULL; | |
82 | BIGNUM *g_x = NULL; | |
83 | BIGNUM *g_y = NULL; | |
84 | BIGNUM *order = NULL; | |
85 | BIGNUM *cof = NULL; | |
86 | EC_POINT *generator = NULL; | |
87 | EC_GROUP *group = NULL; | |
88 | int ok = 0; | |
89 | ||
90 | if (!TEST_true(BN_hex2bn(&p, p_hex)) | |
91 | || !TEST_true(BN_hex2bn(&a, a_hex)) | |
92 | || !TEST_true(BN_hex2bn(&b, b_hex))) | |
93 | goto done; | |
94 | ||
95 | group = EC_GROUP_new_curve_GFp(p, a, b, NULL); | |
96 | if (!TEST_ptr(group)) | |
97 | goto done; | |
98 | ||
99 | generator = EC_POINT_new(group); | |
100 | if (!TEST_ptr(generator)) | |
101 | goto done; | |
102 | ||
103 | if (!TEST_true(BN_hex2bn(&g_x, x_hex)) | |
104 | || !TEST_true(BN_hex2bn(&g_y, y_hex)) | |
105 | || !TEST_true(EC_POINT_set_affine_coordinates_GFp(group, generator, | |
106 | g_x, g_y, NULL))) | |
107 | goto done; | |
108 | ||
109 | if (!TEST_true(BN_hex2bn(&order, order_hex)) | |
110 | || !TEST_true(BN_hex2bn(&cof, cof_hex)) | |
111 | || !TEST_true(EC_GROUP_set_generator(group, generator, order, cof))) | |
112 | goto done; | |
113 | ||
114 | ok = 1; | |
115 | done: | |
116 | BN_free(p); | |
117 | BN_free(a); | |
118 | BN_free(b); | |
119 | BN_free(g_x); | |
120 | BN_free(g_y); | |
121 | EC_POINT_free(generator); | |
122 | BN_free(order); | |
123 | BN_free(cof); | |
124 | if (!ok) { | |
125 | EC_GROUP_free(group); | |
126 | group = NULL; | |
127 | } | |
128 | ||
129 | return group; | |
130 | } | |
131 | ||
132 | static int test_sm2_crypt(const EC_GROUP *group, | |
133 | const EVP_MD *digest, | |
134 | const char *privkey_hex, | |
135 | const char *message, | |
136 | const char *k_hex, const char *ctext_hex) | |
137 | { | |
138 | const size_t msg_len = strlen(message); | |
139 | BIGNUM *priv = NULL; | |
140 | EC_KEY *key = NULL; | |
141 | EC_POINT *pt = NULL; | |
142 | unsigned char *expected = OPENSSL_hexstr2buf(ctext_hex, NULL); | |
143 | size_t ctext_len = 0; | |
144 | size_t ptext_len = 0; | |
145 | uint8_t *ctext = NULL; | |
146 | uint8_t *recovered = NULL; | |
147 | size_t recovered_len = msg_len; | |
148 | int rc = 0; | |
149 | ||
150 | if (!TEST_ptr(expected) | |
151 | || !TEST_true(BN_hex2bn(&priv, privkey_hex))) | |
152 | goto done; | |
153 | ||
154 | key = EC_KEY_new(); | |
155 | if (!TEST_ptr(key) | |
156 | || !TEST_true(EC_KEY_set_group(key, group)) | |
157 | || !TEST_true(EC_KEY_set_private_key(key, priv))) | |
158 | goto done; | |
159 | ||
160 | pt = EC_POINT_new(group); | |
161 | if (!TEST_ptr(pt) | |
162 | || !TEST_true(EC_POINT_mul(group, pt, priv, NULL, NULL, NULL)) | |
163 | || !TEST_true(EC_KEY_set_public_key(key, pt)) | |
2167239a | 164 | || !TEST_true(sm2_ciphertext_size(key, digest, msg_len, &ctext_len))) |
e14d6cf6 MC |
165 | goto done; |
166 | ||
167 | ctext = OPENSSL_zalloc(ctext_len); | |
168 | if (!TEST_ptr(ctext)) | |
169 | goto done; | |
170 | ||
171 | start_fake_rand(k_hex); | |
2167239a | 172 | if (!TEST_true(sm2_encrypt(key, digest, (const uint8_t *)message, msg_len, |
e14d6cf6 MC |
173 | ctext, &ctext_len))) { |
174 | restore_rand(); | |
175 | goto done; | |
176 | } | |
177 | restore_rand(); | |
178 | ||
179 | if (!TEST_mem_eq(ctext, ctext_len, expected, ctext_len)) | |
180 | goto done; | |
181 | ||
2167239a | 182 | if (!TEST_true(sm2_plaintext_size(key, digest, ctext_len, &ptext_len)) |
e14d6cf6 MC |
183 | || !TEST_int_eq(ptext_len, msg_len)) |
184 | goto done; | |
185 | ||
186 | recovered = OPENSSL_zalloc(ptext_len); | |
187 | if (!TEST_ptr(recovered) | |
2167239a | 188 | || !TEST_true(sm2_decrypt(key, digest, ctext, ctext_len, recovered, &recovered_len)) |
e14d6cf6 MC |
189 | || !TEST_int_eq(recovered_len, msg_len) |
190 | || !TEST_mem_eq(recovered, recovered_len, message, msg_len)) | |
191 | goto done; | |
192 | ||
193 | rc = 1; | |
194 | done: | |
195 | BN_free(priv); | |
196 | EC_POINT_free(pt); | |
197 | OPENSSL_free(ctext); | |
198 | OPENSSL_free(recovered); | |
199 | OPENSSL_free(expected); | |
200 | EC_KEY_free(key); | |
201 | return rc; | |
202 | } | |
203 | ||
204 | static int sm2_crypt_test(void) | |
205 | { | |
206 | int testresult = 1; | |
207 | EC_GROUP *test_group = | |
208 | create_EC_group | |
209 | ("8542D69E4C044F18E8B92435BF6FF7DE457283915C45517D722EDB8B08F1DFC3", | |
210 | "787968B4FA32C3FD2417842E73BBFEFF2F3C848B6831D7E0EC65228B3937E498", | |
211 | "63E4C6D3B23B0C849CF84241484BFE48F61D59A5B16BA06E6E12D1DA27C5249A", | |
212 | "421DEBD61B62EAB6746434EBC3CC315E32220B3BADD50BDC4C4E6C147FEDD43D", | |
213 | "0680512BCBB42C07D47349D2153B70C4E5D7FDFCBFA36EA1A85841B9E46E09A2", | |
214 | "8542D69E4C044F18E8B92435BF6FF7DD297720630485628D5AE74EE7C32E79B7", | |
215 | "1"); | |
216 | ||
217 | if (!TEST_ptr(test_group)) | |
218 | goto done; | |
219 | ||
220 | if (!test_sm2_crypt( | |
221 | test_group, | |
222 | EVP_sm3(), | |
223 | "1649AB77A00637BD5E2EFE283FBF353534AA7F7CB89463F208DDBC2920BB0DA0", | |
224 | "encryption standard", | |
225 | "004C62EEFD6ECFC2B95B92FD6C3D9575148AFA17425546D49018E5388D49DD7B4F", | |
226 | "307B0220245C26FB68B1DDDDB12C4B6BF9F2B6D5FE60A383B0D18D1C4144ABF1" | |
227 | "7F6252E7022076CB9264C2A7E88E52B19903FDC47378F605E36811F5C07423A2" | |
228 | "4B84400F01B804209C3D7360C30156FAB7C80A0276712DA9D8094A634B766D3A" | |
229 | "285E07480653426D0413650053A89B41C418B0C3AAD00D886C00286467")) | |
230 | goto done; | |
231 | ||
232 | /* Same test as above except using SHA-256 instead of SM3 */ | |
233 | if (!test_sm2_crypt( | |
234 | test_group, | |
235 | EVP_sha256(), | |
236 | "1649AB77A00637BD5E2EFE283FBF353534AA7F7CB89463F208DDBC2920BB0DA0", | |
237 | "encryption standard", | |
238 | "004C62EEFD6ECFC2B95B92FD6C3D9575148AFA17425546D49018E5388D49DD7B4F", | |
239 | "307B0220245C26FB68B1DDDDB12C4B6BF9F2B6D5FE60A383B0D18D1C4144ABF17F" | |
240 | "6252E7022076CB9264C2A7E88E52B19903FDC47378F605E36811F5C07423A24B84" | |
241 | "400F01B80420BE89139D07853100EFA763F60CBE30099EA3DF7F8F364F9D10A5E9" | |
242 | "88E3C5AAFC0413229E6C9AEE2BB92CAD649FE2C035689785DA33")) | |
243 | goto done; | |
244 | ||
245 | testresult = 1; | |
246 | done: | |
247 | EC_GROUP_free(test_group); | |
248 | ||
249 | return testresult; | |
250 | } | |
251 | ||
252 | static int test_sm2_sign(const EC_GROUP *group, | |
253 | const char *userid, | |
254 | const char *privkey_hex, | |
255 | const char *message, | |
256 | const char *k_hex, | |
257 | const char *r_hex, | |
258 | const char *s_hex) | |
259 | { | |
260 | const size_t msg_len = strlen(message); | |
261 | int ok = 0; | |
262 | BIGNUM *priv = NULL; | |
263 | EC_POINT *pt = NULL; | |
264 | EC_KEY *key = NULL; | |
265 | ECDSA_SIG *sig = NULL; | |
266 | const BIGNUM *sig_r = NULL; | |
267 | const BIGNUM *sig_s = NULL; | |
268 | BIGNUM *r = NULL; | |
269 | BIGNUM *s = NULL; | |
270 | ||
271 | if (!TEST_true(BN_hex2bn(&priv, privkey_hex))) | |
272 | goto done; | |
273 | ||
274 | key = EC_KEY_new(); | |
275 | if (!TEST_ptr(key) | |
276 | || !TEST_true(EC_KEY_set_group(key, group)) | |
277 | || !TEST_true(EC_KEY_set_private_key(key, priv))) | |
278 | goto done; | |
279 | ||
280 | pt = EC_POINT_new(group); | |
281 | if (!TEST_ptr(pt) | |
282 | || !TEST_true(EC_POINT_mul(group, pt, priv, NULL, NULL, NULL)) | |
283 | || !TEST_true(EC_KEY_set_public_key(key, pt))) | |
284 | goto done; | |
285 | ||
286 | start_fake_rand(k_hex); | |
2167239a | 287 | sig = sm2_do_sign(key, EVP_sm3(), userid, (const uint8_t *)message, msg_len); |
e14d6cf6 MC |
288 | restore_rand(); |
289 | ||
290 | if (!TEST_ptr(sig)) | |
291 | goto done; | |
292 | ||
293 | ECDSA_SIG_get0(sig, &sig_r, &sig_s); | |
294 | ||
295 | if (!TEST_true(BN_hex2bn(&r, r_hex)) | |
296 | || !TEST_true(BN_hex2bn(&s, s_hex)) | |
297 | || !TEST_BN_eq(r, sig_r) | |
298 | || !TEST_BN_eq(s, sig_s)) | |
299 | goto done; | |
300 | ||
2167239a | 301 | ok = sm2_do_verify(key, EVP_sm3(), sig, userid, (const uint8_t *)message, |
e14d6cf6 MC |
302 | msg_len); |
303 | ||
304 | /* We goto done whether this passes or fails */ | |
305 | TEST_true(ok); | |
306 | ||
307 | done: | |
308 | ECDSA_SIG_free(sig); | |
309 | EC_POINT_free(pt); | |
310 | EC_KEY_free(key); | |
311 | BN_free(priv); | |
312 | BN_free(r); | |
313 | BN_free(s); | |
314 | ||
315 | return ok; | |
316 | } | |
317 | ||
318 | static int sm2_sig_test(void) | |
319 | { | |
320 | int testresult = 0; | |
321 | /* From draft-shen-sm2-ecdsa-02 */ | |
322 | EC_GROUP *test_group = | |
323 | create_EC_group | |
324 | ("8542D69E4C044F18E8B92435BF6FF7DE457283915C45517D722EDB8B08F1DFC3", | |
325 | "787968B4FA32C3FD2417842E73BBFEFF2F3C848B6831D7E0EC65228B3937E498", | |
326 | "63E4C6D3B23B0C849CF84241484BFE48F61D59A5B16BA06E6E12D1DA27C5249A", | |
327 | "421DEBD61B62EAB6746434EBC3CC315E32220B3BADD50BDC4C4E6C147FEDD43D", | |
328 | "0680512BCBB42C07D47349D2153B70C4E5D7FDFCBFA36EA1A85841B9E46E09A2", | |
329 | "8542D69E4C044F18E8B92435BF6FF7DD297720630485628D5AE74EE7C32E79B7", | |
330 | "1"); | |
331 | ||
332 | if (!TEST_ptr(test_group)) | |
333 | goto done; | |
334 | ||
335 | if (!TEST_true(test_sm2_sign( | |
336 | test_group, | |
337 | "ALICE123@YAHOO.COM", | |
338 | "128B2FA8BD433C6C068C8D803DFF79792A519A55171B1B650C23661D15897263", | |
339 | "message digest", | |
340 | "006CB28D99385C175C94F94E934817663FC176D925DD72B727260DBAAE1FB2F96F", | |
341 | "40F1EC59F793D9F49E09DCEF49130D4194F79FB1EED2CAA55BACDB49C4E755D1", | |
342 | "6FC6DAC32C5D5CF10C77DFB20F7C2EB667A457872FB09EC56327A67EC7DEEBE7"))) | |
343 | goto done; | |
344 | ||
345 | testresult = 1; | |
346 | ||
347 | done: | |
348 | EC_GROUP_free(test_group); | |
349 | ||
350 | return testresult; | |
351 | } | |
352 | ||
353 | #endif | |
354 | ||
355 | int setup_tests(void) | |
356 | { | |
357 | #ifdef OPENSSL_NO_SM2 | |
358 | TEST_note("SM2 is disabled."); | |
359 | #else | |
360 | ADD_TEST(sm2_crypt_test); | |
361 | ADD_TEST(sm2_sig_test); | |
362 | #endif | |
363 | return 1; | |
364 | } |