2 * Copyright 2017-2018 The OpenSSL Project Authors. All Rights Reserved.
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
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>
22 #ifndef OPENSSL_NO_SM2
24 # include "internal/sm2.h"
26 static RAND_METHOD fake_rand;
27 static const RAND_METHOD *saved_rand;
29 static uint8_t *fake_rand_bytes = NULL;
30 static size_t fake_rand_bytes_offset = 0;
32 static int get_faked_bytes(unsigned char *buf, int num)
36 if (fake_rand_bytes == NULL)
37 return saved_rand->bytes(buf, num);
39 for (i = 0; i != num; ++i)
40 buf[i] = fake_rand_bytes[fake_rand_bytes_offset + i];
41 fake_rand_bytes_offset += num;
45 static int start_fake_rand(const char *hex_bytes)
47 /* save old rand method */
48 if (!TEST_ptr(saved_rand = RAND_get_rand_method()))
51 fake_rand = *saved_rand;
52 /* use own random function */
53 fake_rand.bytes = get_faked_bytes;
55 fake_rand_bytes = OPENSSL_hexstr2buf(hex_bytes, NULL);
56 fake_rand_bytes_offset = 0;
58 /* set new RAND_METHOD */
59 if (!TEST_true(RAND_set_rand_method(&fake_rand)))
64 static int restore_rand(void)
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)))
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,
86 EC_POINT *generator = NULL;
87 EC_GROUP *group = NULL;
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)))
95 group = EC_GROUP_new_curve_GFp(p, a, b, NULL);
99 generator = EC_POINT_new(group);
100 if (!TEST_ptr(generator))
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,
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)))
121 EC_POINT_free(generator);
125 EC_GROUP_free(group);
132 static int test_sm2_crypt(const EC_GROUP *group,
133 const EVP_MD *digest,
134 const char *privkey_hex,
136 const char *k_hex, const char *ctext_hex)
138 const size_t msg_len = strlen(message);
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;
150 if (!TEST_ptr(expected)
151 || !TEST_true(BN_hex2bn(&priv, privkey_hex)))
156 || !TEST_true(EC_KEY_set_group(key, group))
157 || !TEST_true(EC_KEY_set_private_key(key, priv)))
160 pt = EC_POINT_new(group);
162 || !TEST_true(EC_POINT_mul(group, pt, priv, NULL, NULL, NULL))
163 || !TEST_true(EC_KEY_set_public_key(key, pt))
164 || !TEST_true(SM2_ciphertext_size(key, digest, msg_len, &ctext_len)))
167 ctext = OPENSSL_zalloc(ctext_len);
168 if (!TEST_ptr(ctext))
171 start_fake_rand(k_hex);
172 if (!TEST_true(SM2_encrypt(key, digest, (const uint8_t *)message, msg_len,
173 ctext, &ctext_len))) {
179 if (!TEST_mem_eq(ctext, ctext_len, expected, ctext_len))
182 if (!TEST_true(SM2_plaintext_size(key, digest, ctext_len, &ptext_len))
183 || !TEST_int_eq(ptext_len, msg_len))
186 recovered = OPENSSL_zalloc(ptext_len);
187 if (!TEST_ptr(recovered)
188 || !TEST_true(SM2_decrypt(key, digest, ctext, ctext_len, recovered, &recovered_len))
189 || !TEST_int_eq(recovered_len, msg_len)
190 || !TEST_mem_eq(recovered, recovered_len, message, msg_len))
198 OPENSSL_free(recovered);
199 OPENSSL_free(expected);
204 static int sm2_crypt_test(void)
207 EC_GROUP *test_group =
209 ("8542D69E4C044F18E8B92435BF6FF7DE457283915C45517D722EDB8B08F1DFC3",
210 "787968B4FA32C3FD2417842E73BBFEFF2F3C848B6831D7E0EC65228B3937E498",
211 "63E4C6D3B23B0C849CF84241484BFE48F61D59A5B16BA06E6E12D1DA27C5249A",
212 "421DEBD61B62EAB6746434EBC3CC315E32220B3BADD50BDC4C4E6C147FEDD43D",
213 "0680512BCBB42C07D47349D2153B70C4E5D7FDFCBFA36EA1A85841B9E46E09A2",
214 "8542D69E4C044F18E8B92435BF6FF7DD297720630485628D5AE74EE7C32E79B7",
217 if (!TEST_ptr(test_group))
223 "1649AB77A00637BD5E2EFE283FBF353534AA7F7CB89463F208DDBC2920BB0DA0",
224 "encryption standard",
225 "004C62EEFD6ECFC2B95B92FD6C3D9575148AFA17425546D49018E5388D49DD7B4F",
226 "307B0220245C26FB68B1DDDDB12C4B6BF9F2B6D5FE60A383B0D18D1C4144ABF1"
227 "7F6252E7022076CB9264C2A7E88E52B19903FDC47378F605E36811F5C07423A2"
228 "4B84400F01B804209C3D7360C30156FAB7C80A0276712DA9D8094A634B766D3A"
229 "285E07480653426D0413650053A89B41C418B0C3AAD00D886C00286467"))
232 /* Same test as above except using SHA-256 instead of SM3 */
236 "1649AB77A00637BD5E2EFE283FBF353534AA7F7CB89463F208DDBC2920BB0DA0",
237 "encryption standard",
238 "004C62EEFD6ECFC2B95B92FD6C3D9575148AFA17425546D49018E5388D49DD7B4F",
239 "307B0220245C26FB68B1DDDDB12C4B6BF9F2B6D5FE60A383B0D18D1C4144ABF17F"
240 "6252E7022076CB9264C2A7E88E52B19903FDC47378F605E36811F5C07423A24B84"
241 "400F01B80420BE89139D07853100EFA763F60CBE30099EA3DF7F8F364F9D10A5E9"
242 "88E3C5AAFC0413229E6C9AEE2BB92CAD649FE2C035689785DA33"))
247 EC_GROUP_free(test_group);
252 static int test_sm2_sign(const EC_GROUP *group,
254 const char *privkey_hex,
260 const size_t msg_len = strlen(message);
265 ECDSA_SIG *sig = NULL;
266 const BIGNUM *sig_r = NULL;
267 const BIGNUM *sig_s = NULL;
271 if (!TEST_true(BN_hex2bn(&priv, privkey_hex)))
276 || !TEST_true(EC_KEY_set_group(key, group))
277 || !TEST_true(EC_KEY_set_private_key(key, priv)))
280 pt = EC_POINT_new(group);
282 || !TEST_true(EC_POINT_mul(group, pt, priv, NULL, NULL, NULL))
283 || !TEST_true(EC_KEY_set_public_key(key, pt)))
286 start_fake_rand(k_hex);
287 sig = SM2_do_sign(key, EVP_sm3(), userid, (const uint8_t *)message, msg_len);
293 ECDSA_SIG_get0(sig, &sig_r, &sig_s);
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))
301 ok = SM2_do_verify(key, EVP_sm3(), sig, userid, (const uint8_t *)message,
304 /* We goto done whether this passes or fails */
318 static int sm2_sig_test(void)
321 /* From draft-shen-sm2-ecdsa-02 */
322 EC_GROUP *test_group =
324 ("8542D69E4C044F18E8B92435BF6FF7DE457283915C45517D722EDB8B08F1DFC3",
325 "787968B4FA32C3FD2417842E73BBFEFF2F3C848B6831D7E0EC65228B3937E498",
326 "63E4C6D3B23B0C849CF84241484BFE48F61D59A5B16BA06E6E12D1DA27C5249A",
327 "421DEBD61B62EAB6746434EBC3CC315E32220B3BADD50BDC4C4E6C147FEDD43D",
328 "0680512BCBB42C07D47349D2153B70C4E5D7FDFCBFA36EA1A85841B9E46E09A2",
329 "8542D69E4C044F18E8B92435BF6FF7DD297720630485628D5AE74EE7C32E79B7",
332 if (!TEST_ptr(test_group))
335 if (!TEST_true(test_sm2_sign(
337 "ALICE123@YAHOO.COM",
338 "128B2FA8BD433C6C068C8D803DFF79792A519A55171B1B650C23661D15897263",
340 "006CB28D99385C175C94F94E934817663FC176D925DD72B727260DBAAE1FB2F96F",
341 "40F1EC59F793D9F49E09DCEF49130D4194F79FB1EED2CAA55BACDB49C4E755D1",
342 "6FC6DAC32C5D5CF10C77DFB20F7C2EB667A457872FB09EC56327A67EC7DEEBE7")))
348 EC_GROUP_free(test_group);
355 int setup_tests(void)
357 #ifdef OPENSSL_NO_SM2
358 TEST_note("SM2 is disabled.");
360 ADD_TEST(sm2_crypt_test);
361 ADD_TEST(sm2_sig_test);