sm2_internal_test.c 12 KB

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