1 /*
2 * Copyright 2017-2021 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 "crypto/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 static size_t fake_rand_size = 0;
32
get_faked_bytes(unsigned char * buf,int num)33 static int get_faked_bytes(unsigned char *buf, int num)
34 {
35 if (fake_rand_bytes == NULL)
36 return saved_rand->bytes(buf, num);
37
38 if (!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
start_fake_rand(const char * hex_bytes)50 static int start_fake_rand(const char *hex_bytes)
51 {
52 /* save old rand method */
53 if (!TEST_ptr(saved_rand = RAND_get_rand_method()))
54 return 0;
55
56 fake_rand = *saved_rand;
57 /* use own random function */
58 fake_rand.bytes = get_faked_bytes;
59
60 fake_rand_bytes = OPENSSL_hexstr2buf(hex_bytes, NULL);
61 fake_rand_bytes_offset = 0;
62 fake_rand_size = strlen(hex_bytes) / 2;
63
64 /* set new RAND_METHOD */
65 if (!TEST_true(RAND_set_rand_method(&fake_rand)))
66 return 0;
67 return 1;
68 }
69
restore_rand(void)70 static int restore_rand(void)
71 {
72 OPENSSL_free(fake_rand_bytes);
73 fake_rand_bytes = NULL;
74 fake_rand_bytes_offset = 0;
75 if (!TEST_true(RAND_set_rand_method(saved_rand)))
76 return 0;
77 return 1;
78 }
79
create_EC_group(const char * p_hex,const char * a_hex,const char * b_hex,const char * x_hex,const char * y_hex,const char * order_hex,const char * cof_hex)80 static EC_GROUP *create_EC_group(const char *p_hex, const char *a_hex,
81 const char *b_hex, const char *x_hex,
82 const char *y_hex, const char *order_hex,
83 const char *cof_hex)
84 {
85 BIGNUM *p = NULL;
86 BIGNUM *a = NULL;
87 BIGNUM *b = NULL;
88 BIGNUM *g_x = NULL;
89 BIGNUM *g_y = NULL;
90 BIGNUM *order = NULL;
91 BIGNUM *cof = NULL;
92 EC_POINT *generator = NULL;
93 EC_GROUP *group = NULL;
94 int ok = 0;
95
96 if (!TEST_true(BN_hex2bn(&p, p_hex))
97 || !TEST_true(BN_hex2bn(&a, a_hex))
98 || !TEST_true(BN_hex2bn(&b, b_hex)))
99 goto done;
100
101 group = EC_GROUP_new_curve_GFp(p, a, b, NULL);
102 if (!TEST_ptr(group))
103 goto done;
104
105 generator = EC_POINT_new(group);
106 if (!TEST_ptr(generator))
107 goto done;
108
109 if (!TEST_true(BN_hex2bn(&g_x, x_hex))
110 || !TEST_true(BN_hex2bn(&g_y, y_hex))
111 || !TEST_true(EC_POINT_set_affine_coordinates(group, generator, g_x,
112 g_y, NULL)))
113 goto done;
114
115 if (!TEST_true(BN_hex2bn(&order, order_hex))
116 || !TEST_true(BN_hex2bn(&cof, cof_hex))
117 || !TEST_true(EC_GROUP_set_generator(group, generator, order, cof)))
118 goto done;
119
120 ok = 1;
121 done:
122 BN_free(p);
123 BN_free(a);
124 BN_free(b);
125 BN_free(g_x);
126 BN_free(g_y);
127 EC_POINT_free(generator);
128 BN_free(order);
129 BN_free(cof);
130 if (!ok) {
131 EC_GROUP_free(group);
132 group = NULL;
133 }
134
135 return group;
136 }
137
test_sm2_crypt(const EC_GROUP * group,const EVP_MD * digest,const char * privkey_hex,const char * message,const char * k_hex,const char * ctext_hex)138 static int test_sm2_crypt(const EC_GROUP *group,
139 const EVP_MD *digest,
140 const char *privkey_hex,
141 const char *message,
142 const char *k_hex, const char *ctext_hex)
143 {
144 const size_t msg_len = strlen(message);
145 BIGNUM *priv = NULL;
146 EC_KEY *key = NULL;
147 EC_POINT *pt = NULL;
148 unsigned char *expected = OPENSSL_hexstr2buf(ctext_hex, NULL);
149 size_t ctext_len = 0;
150 size_t ptext_len = 0;
151 uint8_t *ctext = NULL;
152 uint8_t *recovered = NULL;
153 size_t recovered_len = msg_len;
154 int rc = 0;
155
156 if (!TEST_ptr(expected)
157 || !TEST_true(BN_hex2bn(&priv, privkey_hex)))
158 goto done;
159
160 key = EC_KEY_new();
161 if (!TEST_ptr(key)
162 || !TEST_true(EC_KEY_set_group(key, group))
163 || !TEST_true(EC_KEY_set_private_key(key, priv)))
164 goto done;
165
166 pt = EC_POINT_new(group);
167 if (!TEST_ptr(pt)
168 || !TEST_true(EC_POINT_mul(group, pt, priv, NULL, NULL, NULL))
169 || !TEST_true(EC_KEY_set_public_key(key, pt))
170 || !TEST_true(sm2_ciphertext_size(key, digest, msg_len, &ctext_len)))
171 goto done;
172
173 ctext = OPENSSL_zalloc(ctext_len);
174 if (!TEST_ptr(ctext))
175 goto done;
176
177 start_fake_rand(k_hex);
178 if (!TEST_true(sm2_encrypt(key, digest, (const uint8_t *)message, msg_len,
179 ctext, &ctext_len))) {
180 restore_rand();
181 goto done;
182 }
183 restore_rand();
184
185 if (!TEST_mem_eq(ctext, ctext_len, expected, ctext_len))
186 goto done;
187
188 if (!TEST_true(sm2_plaintext_size(ctext, ctext_len, &ptext_len))
189 || !TEST_int_eq(ptext_len, msg_len))
190 goto done;
191
192 recovered = OPENSSL_zalloc(ptext_len);
193 if (!TEST_ptr(recovered)
194 || !TEST_true(sm2_decrypt(key, digest, ctext, ctext_len, recovered, &recovered_len))
195 || !TEST_int_eq(recovered_len, msg_len)
196 || !TEST_mem_eq(recovered, recovered_len, message, msg_len))
197 goto done;
198
199 rc = 1;
200 done:
201 BN_free(priv);
202 EC_POINT_free(pt);
203 OPENSSL_free(ctext);
204 OPENSSL_free(recovered);
205 OPENSSL_free(expected);
206 EC_KEY_free(key);
207 return rc;
208 }
209
sm2_crypt_test(void)210 static int sm2_crypt_test(void)
211 {
212 int testresult = 0;
213 EC_GROUP *gm_group = NULL;
214 EC_GROUP *test_group =
215 create_EC_group
216 ("8542D69E4C044F18E8B92435BF6FF7DE457283915C45517D722EDB8B08F1DFC3",
217 "787968B4FA32C3FD2417842E73BBFEFF2F3C848B6831D7E0EC65228B3937E498",
218 "63E4C6D3B23B0C849CF84241484BFE48F61D59A5B16BA06E6E12D1DA27C5249A",
219 "421DEBD61B62EAB6746434EBC3CC315E32220B3BADD50BDC4C4E6C147FEDD43D",
220 "0680512BCBB42C07D47349D2153B70C4E5D7FDFCBFA36EA1A85841B9E46E09A2",
221 "8542D69E4C044F18E8B92435BF6FF7DD297720630485628D5AE74EE7C32E79B7",
222 "1");
223
224 if (!TEST_ptr(test_group))
225 goto done;
226
227 if (!test_sm2_crypt(
228 test_group,
229 EVP_sm3(),
230 "1649AB77A00637BD5E2EFE283FBF353534AA7F7CB89463F208DDBC2920BB0DA0",
231 "encryption standard",
232 "004C62EEFD6ECFC2B95B92FD6C3D9575148AFA17425546D49018E5388D49DD7B4F"
233 "0092e8ff62146873c258557548500ab2df2a365e0609ab67640a1f6d57d7b17820"
234 "008349312695a3e1d2f46905f39a766487f2432e95d6be0cb009fe8c69fd8825a7",
235 "307B0220245C26FB68B1DDDDB12C4B6BF9F2B6D5FE60A383B0D18D1C4144ABF1"
236 "7F6252E7022076CB9264C2A7E88E52B19903FDC47378F605E36811F5C07423A2"
237 "4B84400F01B804209C3D7360C30156FAB7C80A0276712DA9D8094A634B766D3A"
238 "285E07480653426D0413650053A89B41C418B0C3AAD00D886C00286467"))
239 goto done;
240
241 /* Same test as above except using SHA-256 instead of SM3 */
242 if (!test_sm2_crypt(
243 test_group,
244 EVP_sha256(),
245 "1649AB77A00637BD5E2EFE283FBF353534AA7F7CB89463F208DDBC2920BB0DA0",
246 "encryption standard",
247 "004C62EEFD6ECFC2B95B92FD6C3D9575148AFA17425546D49018E5388D49DD7B4F"
248 "003da18008784352192d70f22c26c243174a447ba272fec64163dd4742bae8bc98"
249 "00df17605cf304e9dd1dfeb90c015e93b393a6f046792f790a6fa4228af67d9588",
250 "307B0220245C26FB68B1DDDDB12C4B6BF9F2B6D5FE60A383B0D18D1C4144ABF17F"
251 "6252E7022076CB9264C2A7E88E52B19903FDC47378F605E36811F5C07423A24B84"
252 "400F01B80420BE89139D07853100EFA763F60CBE30099EA3DF7F8F364F9D10A5E9"
253 "88E3C5AAFC0413229E6C9AEE2BB92CAD649FE2C035689785DA33"))
254 goto done;
255
256 /* From Annex C in both GM/T0003.5-2012 and GB/T 32918.5-2016.*/
257 gm_group = create_EC_group(
258 "fffffffeffffffffffffffffffffffffffffffff00000000ffffffffffffffff",
259 "fffffffeffffffffffffffffffffffffffffffff00000000fffffffffffffffc",
260 "28e9fa9e9d9f5e344d5a9e4bcf6509a7f39789f515ab8f92ddbcbd414d940e93",
261 "32c4ae2c1f1981195f9904466a39c9948fe30bbff2660be1715a4589334c74c7",
262 "bc3736a2f4f6779c59bdcee36b692153d0a9877cc62a474002df32e52139f0a0",
263 "fffffffeffffffffffffffffffffffff7203df6b21c6052b53bbf40939d54123",
264 "1");
265
266 if (!TEST_ptr(gm_group))
267 goto done;
268
269 if (!test_sm2_crypt(
270 gm_group,
271 EVP_sm3(),
272 /* privkey (from which the encrypting public key is derived) */
273 "3945208F7B2144B13F36E38AC6D39F95889393692860B51A42FB81EF4DF7C5B8",
274 /* plaintext message */
275 "encryption standard",
276 /* ephemeral nonce k */
277 "59276E27D506861A16680F3AD9C02DCCEF3CC1FA3CDBE4CE6D54B80DEAC1BC21",
278 /*
279 * expected ciphertext, the field values are from GM/T 0003.5-2012
280 * (Annex C), but serialized following the ASN.1 format specified
281 * in GM/T 0009-2012 (Sec. 7.2).
282 */
283 "307C" /* SEQUENCE, 0x7c bytes */
284 "0220" /* INTEGER, 0x20 bytes */
285 "04EBFC718E8D1798620432268E77FEB6415E2EDE0E073C0F4F640ECD2E149A73"
286 "0221" /* INTEGER, 0x21 bytes */
287 "00" /* leading 00 due to DER for pos. int with topmost bit set */
288 "E858F9D81E5430A57B36DAAB8F950A3C64E6EE6A63094D99283AFF767E124DF0"
289 "0420" /* OCTET STRING, 0x20 bytes */
290 "59983C18F809E262923C53AEC295D30383B54E39D609D160AFCB1908D0BD8766"
291 "0413" /* OCTET STRING, 0x13 bytes */
292 "21886CA989CA9C7D58087307CA93092D651EFA"))
293 goto done;
294
295 testresult = 1;
296 done:
297 EC_GROUP_free(test_group);
298 EC_GROUP_free(gm_group);
299
300 return testresult;
301 }
302
test_sm2_sign(const EC_GROUP * group,const char * userid,const char * privkey_hex,const char * message,const char * k_hex,const char * r_hex,const char * s_hex)303 static int test_sm2_sign(const EC_GROUP *group,
304 const char *userid,
305 const char *privkey_hex,
306 const char *message,
307 const char *k_hex,
308 const char *r_hex,
309 const char *s_hex)
310 {
311 const size_t msg_len = strlen(message);
312 int ok = 0;
313 BIGNUM *priv = NULL;
314 EC_POINT *pt = NULL;
315 EC_KEY *key = NULL;
316 ECDSA_SIG *sig = NULL;
317 const BIGNUM *sig_r = NULL;
318 const BIGNUM *sig_s = NULL;
319 BIGNUM *r = NULL;
320 BIGNUM *s = NULL;
321
322 if (!TEST_true(BN_hex2bn(&priv, privkey_hex)))
323 goto done;
324
325 key = EC_KEY_new();
326 if (!TEST_ptr(key)
327 || !TEST_true(EC_KEY_set_group(key, group))
328 || !TEST_true(EC_KEY_set_private_key(key, priv)))
329 goto done;
330
331 pt = EC_POINT_new(group);
332 if (!TEST_ptr(pt)
333 || !TEST_true(EC_POINT_mul(group, pt, priv, NULL, NULL, NULL))
334 || !TEST_true(EC_KEY_set_public_key(key, pt)))
335 goto done;
336
337 start_fake_rand(k_hex);
338 sig = sm2_do_sign(key, EVP_sm3(), (const uint8_t *)userid, strlen(userid),
339 (const uint8_t *)message, msg_len);
340 if (!TEST_ptr(sig)) {
341 restore_rand();
342 goto done;
343 }
344 restore_rand();
345
346 ECDSA_SIG_get0(sig, &sig_r, &sig_s);
347
348 if (!TEST_true(BN_hex2bn(&r, r_hex))
349 || !TEST_true(BN_hex2bn(&s, s_hex))
350 || !TEST_BN_eq(r, sig_r)
351 || !TEST_BN_eq(s, sig_s))
352 goto done;
353
354 ok = sm2_do_verify(key, EVP_sm3(), sig, (const uint8_t *)userid,
355 strlen(userid), (const uint8_t *)message, msg_len);
356
357 /* We goto done whether this passes or fails */
358 TEST_true(ok);
359
360 done:
361 ECDSA_SIG_free(sig);
362 EC_POINT_free(pt);
363 EC_KEY_free(key);
364 BN_free(priv);
365 BN_free(r);
366 BN_free(s);
367
368 return ok;
369 }
370
sm2_sig_test(void)371 static int sm2_sig_test(void)
372 {
373 int testresult = 0;
374 /* From draft-shen-sm2-ecdsa-02 */
375 EC_GROUP *test_group =
376 create_EC_group
377 ("8542D69E4C044F18E8B92435BF6FF7DE457283915C45517D722EDB8B08F1DFC3",
378 "787968B4FA32C3FD2417842E73BBFEFF2F3C848B6831D7E0EC65228B3937E498",
379 "63E4C6D3B23B0C849CF84241484BFE48F61D59A5B16BA06E6E12D1DA27C5249A",
380 "421DEBD61B62EAB6746434EBC3CC315E32220B3BADD50BDC4C4E6C147FEDD43D",
381 "0680512BCBB42C07D47349D2153B70C4E5D7FDFCBFA36EA1A85841B9E46E09A2",
382 "8542D69E4C044F18E8B92435BF6FF7DD297720630485628D5AE74EE7C32E79B7",
383 "1");
384
385 if (!TEST_ptr(test_group))
386 goto done;
387
388 if (!TEST_true(test_sm2_sign(
389 test_group,
390 "ALICE123@YAHOO.COM",
391 "128B2FA8BD433C6C068C8D803DFF79792A519A55171B1B650C23661D15897263",
392 "message digest",
393 "006CB28D99385C175C94F94E934817663FC176D925DD72B727260DBAAE1FB2F96F"
394 "007c47811054c6f99613a578eb8453706ccb96384fe7df5c171671e760bfa8be3a",
395 "40F1EC59F793D9F49E09DCEF49130D4194F79FB1EED2CAA55BACDB49C4E755D1",
396 "6FC6DAC32C5D5CF10C77DFB20F7C2EB667A457872FB09EC56327A67EC7DEEBE7")))
397 goto done;
398
399 testresult = 1;
400
401 done:
402 EC_GROUP_free(test_group);
403
404 return testresult;
405 }
406
407 #endif
408
setup_tests(void)409 int setup_tests(void)
410 {
411 #ifdef OPENSSL_NO_SM2
412 TEST_note("SM2 is disabled.");
413 #else
414 ADD_TEST(sm2_crypt_test);
415 ADD_TEST(sm2_sig_test);
416 #endif
417 return 1;
418 }
419