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