test_suite_ecdsa.function 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524
  1. /* BEGIN_HEADER */
  2. #include "mbedtls/ecdsa.h"
  3. #include "hash_info.h"
  4. #include "mbedtls/legacy_or_psa.h"
  5. #if (defined(MBEDTLS_ECDSA_DETERMINISTIC) && defined(MBEDTLS_SHA256_C)) || \
  6. (!defined(MBEDTLS_ECDSA_DETERMINISTIC) && defined(MBEDTLS_HAS_ALG_SHA_256_VIA_LOWLEVEL_OR_PSA))
  7. #define MBEDTLS_HAS_ALG_SHA_256_VIA_MD_IF_DETERMINISTIC
  8. #endif
  9. /* END_HEADER */
  10. /* BEGIN_DEPENDENCIES
  11. * depends_on:MBEDTLS_ECDSA_C
  12. * END_DEPENDENCIES
  13. */
  14. /* BEGIN_CASE */
  15. void ecdsa_prim_zero(int id)
  16. {
  17. mbedtls_ecp_group grp;
  18. mbedtls_ecp_point Q;
  19. mbedtls_mpi d, r, s;
  20. mbedtls_test_rnd_pseudo_info rnd_info;
  21. unsigned char buf[MBEDTLS_HASH_MAX_SIZE];
  22. mbedtls_ecp_group_init(&grp);
  23. mbedtls_ecp_point_init(&Q);
  24. mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
  25. memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
  26. memset(buf, 0, sizeof(buf));
  27. TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
  28. TEST_ASSERT(mbedtls_ecp_gen_keypair(&grp, &d, &Q,
  29. &mbedtls_test_rnd_pseudo_rand,
  30. &rnd_info) == 0);
  31. TEST_ASSERT(mbedtls_ecdsa_sign(&grp, &r, &s, &d, buf, sizeof(buf),
  32. &mbedtls_test_rnd_pseudo_rand,
  33. &rnd_info) == 0);
  34. TEST_ASSERT(mbedtls_ecdsa_verify(&grp, buf, sizeof(buf), &Q, &r, &s) == 0);
  35. exit:
  36. mbedtls_ecp_group_free(&grp);
  37. mbedtls_ecp_point_free(&Q);
  38. mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
  39. }
  40. /* END_CASE */
  41. /* BEGIN_CASE */
  42. void ecdsa_prim_random(int id)
  43. {
  44. mbedtls_ecp_group grp;
  45. mbedtls_ecp_point Q;
  46. mbedtls_mpi d, r, s;
  47. mbedtls_test_rnd_pseudo_info rnd_info;
  48. unsigned char buf[MBEDTLS_HASH_MAX_SIZE];
  49. mbedtls_ecp_group_init(&grp);
  50. mbedtls_ecp_point_init(&Q);
  51. mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
  52. memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
  53. memset(buf, 0, sizeof(buf));
  54. /* prepare material for signature */
  55. TEST_ASSERT(mbedtls_test_rnd_pseudo_rand(&rnd_info,
  56. buf, sizeof(buf)) == 0);
  57. TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
  58. TEST_ASSERT(mbedtls_ecp_gen_keypair(&grp, &d, &Q,
  59. &mbedtls_test_rnd_pseudo_rand,
  60. &rnd_info) == 0);
  61. TEST_ASSERT(mbedtls_ecdsa_sign(&grp, &r, &s, &d, buf, sizeof(buf),
  62. &mbedtls_test_rnd_pseudo_rand,
  63. &rnd_info) == 0);
  64. TEST_ASSERT(mbedtls_ecdsa_verify(&grp, buf, sizeof(buf), &Q, &r, &s) == 0);
  65. exit:
  66. mbedtls_ecp_group_free(&grp);
  67. mbedtls_ecp_point_free(&Q);
  68. mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
  69. }
  70. /* END_CASE */
  71. /* BEGIN_CASE */
  72. void ecdsa_prim_test_vectors(int id, char *d_str, char *xQ_str,
  73. char *yQ_str, data_t *rnd_buf,
  74. data_t *hash, char *r_str, char *s_str,
  75. int result)
  76. {
  77. mbedtls_ecp_group grp;
  78. mbedtls_ecp_point Q;
  79. mbedtls_mpi d, r, s, r_check, s_check, zero;
  80. mbedtls_test_rnd_buf_info rnd_info;
  81. mbedtls_ecp_group_init(&grp);
  82. mbedtls_ecp_point_init(&Q);
  83. mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
  84. mbedtls_mpi_init(&r_check); mbedtls_mpi_init(&s_check);
  85. mbedtls_mpi_init(&zero);
  86. TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
  87. TEST_ASSERT(mbedtls_ecp_point_read_string(&Q, 16, xQ_str, yQ_str) == 0);
  88. TEST_ASSERT(mbedtls_test_read_mpi(&d, d_str) == 0);
  89. TEST_ASSERT(mbedtls_test_read_mpi(&r_check, r_str) == 0);
  90. TEST_ASSERT(mbedtls_test_read_mpi(&s_check, s_str) == 0);
  91. rnd_info.fallback_f_rng = mbedtls_test_rnd_std_rand;
  92. rnd_info.fallback_p_rng = NULL;
  93. rnd_info.buf = rnd_buf->x;
  94. rnd_info.length = rnd_buf->len;
  95. /* Fix rnd_buf->x by shifting it left if necessary */
  96. if (grp.nbits % 8 != 0) {
  97. unsigned char shift = 8 - (grp.nbits % 8);
  98. size_t i;
  99. for (i = 0; i < rnd_info.length - 1; i++) {
  100. rnd_buf->x[i] = rnd_buf->x[i] << shift | rnd_buf->x[i+1] >> (8 - shift);
  101. }
  102. rnd_buf->x[rnd_info.length-1] <<= shift;
  103. }
  104. TEST_ASSERT(mbedtls_ecdsa_sign(&grp, &r, &s, &d, hash->x, hash->len,
  105. mbedtls_test_rnd_buffer_rand, &rnd_info) == result);
  106. if (result == 0) {
  107. /* Check we generated the expected values */
  108. TEST_EQUAL(mbedtls_mpi_cmp_mpi(&r, &r_check), 0);
  109. TEST_EQUAL(mbedtls_mpi_cmp_mpi(&s, &s_check), 0);
  110. /* Valid signature */
  111. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len,
  112. &Q, &r_check, &s_check), 0);
  113. /* Invalid signature: wrong public key (G instead of Q) */
  114. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len,
  115. &grp.G, &r_check, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  116. /* Invalid signatures: r or s or both one off */
  117. TEST_EQUAL(mbedtls_mpi_sub_int(&r, &r_check, 1), 0);
  118. TEST_EQUAL(mbedtls_mpi_add_int(&s, &s_check, 1), 0);
  119. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  120. &r, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  121. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  122. &r_check, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  123. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  124. &r, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  125. /* Invalid signatures: r, s or both (CVE-2022-21449) are zero */
  126. TEST_EQUAL(mbedtls_mpi_lset(&zero, 0), 0);
  127. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  128. &zero, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  129. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  130. &r_check, &zero), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  131. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  132. &zero, &zero), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  133. /* Invalid signatures: r, s or both are == N */
  134. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  135. &grp.N, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  136. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  137. &r_check, &grp.N), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  138. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  139. &grp.N, &grp.N), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  140. /* Invalid signatures: r, s or both are negative */
  141. TEST_EQUAL(mbedtls_mpi_sub_mpi(&r, &r_check, &grp.N), 0);
  142. TEST_EQUAL(mbedtls_mpi_sub_mpi(&s, &s_check, &grp.N), 0);
  143. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  144. &r, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  145. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  146. &r_check, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  147. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  148. &r, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  149. /* Invalid signatures: r or s or both are > N */
  150. TEST_EQUAL(mbedtls_mpi_add_mpi(&r, &r_check, &grp.N), 0);
  151. TEST_EQUAL(mbedtls_mpi_add_mpi(&s, &s_check, &grp.N), 0);
  152. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  153. &r, &s_check), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  154. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  155. &r_check, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  156. TEST_EQUAL(mbedtls_ecdsa_verify(&grp, hash->x, hash->len, &Q,
  157. &r, &s), MBEDTLS_ERR_ECP_VERIFY_FAILED);
  158. }
  159. exit:
  160. mbedtls_ecp_group_free(&grp);
  161. mbedtls_ecp_point_free(&Q);
  162. mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
  163. mbedtls_mpi_free(&r_check); mbedtls_mpi_free(&s_check);
  164. mbedtls_mpi_free(&zero);
  165. }
  166. /* END_CASE */
  167. /* BEGIN_CASE depends_on:MBEDTLS_ECDSA_DETERMINISTIC */
  168. void ecdsa_det_test_vectors(int id, char *d_str, int md_alg, data_t *hash,
  169. char *r_str, char *s_str)
  170. {
  171. mbedtls_ecp_group grp;
  172. mbedtls_mpi d, r, s, r_check, s_check;
  173. mbedtls_ecp_group_init(&grp);
  174. mbedtls_mpi_init(&d); mbedtls_mpi_init(&r); mbedtls_mpi_init(&s);
  175. mbedtls_mpi_init(&r_check); mbedtls_mpi_init(&s_check);
  176. TEST_ASSERT(mbedtls_ecp_group_load(&grp, id) == 0);
  177. TEST_ASSERT(mbedtls_test_read_mpi(&d, d_str) == 0);
  178. TEST_ASSERT(mbedtls_test_read_mpi(&r_check, r_str) == 0);
  179. TEST_ASSERT(mbedtls_test_read_mpi(&s_check, s_str) == 0);
  180. TEST_ASSERT(
  181. mbedtls_ecdsa_sign_det_ext(&grp, &r, &s, &d,
  182. hash->x, hash->len, md_alg,
  183. mbedtls_test_rnd_std_rand,
  184. NULL)
  185. == 0);
  186. TEST_ASSERT(mbedtls_mpi_cmp_mpi(&r, &r_check) == 0);
  187. TEST_ASSERT(mbedtls_mpi_cmp_mpi(&s, &s_check) == 0);
  188. exit:
  189. mbedtls_ecp_group_free(&grp);
  190. mbedtls_mpi_free(&d); mbedtls_mpi_free(&r); mbedtls_mpi_free(&s);
  191. mbedtls_mpi_free(&r_check); mbedtls_mpi_free(&s_check);
  192. }
  193. /* END_CASE */
  194. /* BEGIN_CASE depends_on:MBEDTLS_HAS_ALG_SHA_256_VIA_MD_IF_DETERMINISTIC */
  195. void ecdsa_write_read_zero(int id)
  196. {
  197. mbedtls_ecdsa_context ctx;
  198. mbedtls_test_rnd_pseudo_info rnd_info;
  199. unsigned char hash[32];
  200. unsigned char sig[200];
  201. size_t sig_len, i;
  202. mbedtls_ecdsa_init(&ctx);
  203. memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
  204. memset(hash, 0, sizeof(hash));
  205. memset(sig, 0x2a, sizeof(sig));
  206. /* generate signing key */
  207. TEST_ASSERT(mbedtls_ecdsa_genkey(&ctx, id,
  208. &mbedtls_test_rnd_pseudo_rand,
  209. &rnd_info) == 0);
  210. /* generate and write signature, then read and verify it */
  211. TEST_ASSERT(mbedtls_ecdsa_write_signature(&ctx, MBEDTLS_MD_SHA256,
  212. hash, sizeof(hash),
  213. sig, sizeof(sig), &sig_len,
  214. &mbedtls_test_rnd_pseudo_rand,
  215. &rnd_info) == 0);
  216. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  217. sig, sig_len) == 0);
  218. /* check we didn't write past the announced length */
  219. for (i = sig_len; i < sizeof(sig); i++) {
  220. TEST_ASSERT(sig[i] == 0x2a);
  221. }
  222. /* try verification with invalid length */
  223. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  224. sig, sig_len - 1) != 0);
  225. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  226. sig, sig_len + 1) != 0);
  227. /* try invalid sequence tag */
  228. sig[0]++;
  229. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  230. sig, sig_len) != 0);
  231. sig[0]--;
  232. /* try modifying r */
  233. sig[10]++;
  234. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  235. sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
  236. sig[10]--;
  237. /* try modifying s */
  238. sig[sig_len - 1]++;
  239. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  240. sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
  241. sig[sig_len - 1]--;
  242. exit:
  243. mbedtls_ecdsa_free(&ctx);
  244. }
  245. /* END_CASE */
  246. /* BEGIN_CASE depends_on:MBEDTLS_HAS_ALG_SHA_256_VIA_MD_IF_DETERMINISTIC */
  247. void ecdsa_write_read_random(int id)
  248. {
  249. mbedtls_ecdsa_context ctx;
  250. mbedtls_test_rnd_pseudo_info rnd_info;
  251. unsigned char hash[32];
  252. unsigned char sig[200];
  253. size_t sig_len, i;
  254. mbedtls_ecdsa_init(&ctx);
  255. memset(&rnd_info, 0x00, sizeof(mbedtls_test_rnd_pseudo_info));
  256. memset(hash, 0, sizeof(hash));
  257. memset(sig, 0x2a, sizeof(sig));
  258. /* prepare material for signature */
  259. TEST_ASSERT(mbedtls_test_rnd_pseudo_rand(&rnd_info,
  260. hash, sizeof(hash)) == 0);
  261. /* generate signing key */
  262. TEST_ASSERT(mbedtls_ecdsa_genkey(&ctx, id,
  263. &mbedtls_test_rnd_pseudo_rand,
  264. &rnd_info) == 0);
  265. /* generate and write signature, then read and verify it */
  266. TEST_ASSERT(mbedtls_ecdsa_write_signature(&ctx, MBEDTLS_MD_SHA256,
  267. hash, sizeof(hash),
  268. sig, sizeof(sig), &sig_len,
  269. &mbedtls_test_rnd_pseudo_rand,
  270. &rnd_info) == 0);
  271. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  272. sig, sig_len) == 0);
  273. /* check we didn't write past the announced length */
  274. for (i = sig_len; i < sizeof(sig); i++) {
  275. TEST_ASSERT(sig[i] == 0x2a);
  276. }
  277. /* try verification with invalid length */
  278. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  279. sig, sig_len - 1) != 0);
  280. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  281. sig, sig_len + 1) != 0);
  282. /* try invalid sequence tag */
  283. sig[0]++;
  284. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  285. sig, sig_len) != 0);
  286. sig[0]--;
  287. /* try modifying r */
  288. sig[10]++;
  289. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  290. sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
  291. sig[10]--;
  292. /* try modifying s */
  293. sig[sig_len - 1]++;
  294. TEST_ASSERT(mbedtls_ecdsa_read_signature(&ctx, hash, sizeof(hash),
  295. sig, sig_len) == MBEDTLS_ERR_ECP_VERIFY_FAILED);
  296. sig[sig_len - 1]--;
  297. exit:
  298. mbedtls_ecdsa_free(&ctx);
  299. }
  300. /* END_CASE */
  301. /* BEGIN_CASE depends_on:MBEDTLS_ECP_RESTARTABLE */
  302. void ecdsa_read_restart(int id, data_t *pk, data_t *hash, data_t *sig,
  303. int max_ops, int min_restart, int max_restart)
  304. {
  305. mbedtls_ecdsa_context ctx;
  306. mbedtls_ecdsa_restart_ctx rs_ctx;
  307. int ret, cnt_restart;
  308. mbedtls_ecdsa_init(&ctx);
  309. mbedtls_ecdsa_restart_init(&rs_ctx);
  310. TEST_ASSERT(mbedtls_ecp_group_load(&ctx.grp, id) == 0);
  311. TEST_ASSERT(mbedtls_ecp_point_read_binary(&ctx.grp, &ctx.Q,
  312. pk->x, pk->len) == 0);
  313. mbedtls_ecp_set_max_ops(max_ops);
  314. cnt_restart = 0;
  315. do {
  316. ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
  317. hash->x, hash->len, sig->x, sig->len,
  318. &rs_ctx);
  319. } while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS && ++cnt_restart);
  320. TEST_ASSERT(ret == 0);
  321. TEST_ASSERT(cnt_restart >= min_restart);
  322. TEST_ASSERT(cnt_restart <= max_restart);
  323. /* try modifying r */
  324. TEST_ASSERT(sig->len > 10);
  325. sig->x[10]++;
  326. do {
  327. ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
  328. hash->x, hash->len, sig->x, sig->len,
  329. &rs_ctx);
  330. } while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
  331. TEST_ASSERT(ret == MBEDTLS_ERR_ECP_VERIFY_FAILED);
  332. sig->x[10]--;
  333. /* try modifying s */
  334. sig->x[sig->len - 1]++;
  335. do {
  336. ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
  337. hash->x, hash->len, sig->x, sig->len,
  338. &rs_ctx);
  339. } while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
  340. TEST_ASSERT(ret == MBEDTLS_ERR_ECP_VERIFY_FAILED);
  341. sig->x[sig->len - 1]--;
  342. /* Do we leak memory when aborting an operation?
  343. * This test only makes sense when we actually restart */
  344. if (min_restart > 0) {
  345. ret = mbedtls_ecdsa_read_signature_restartable(&ctx,
  346. hash->x, hash->len, sig->x, sig->len,
  347. &rs_ctx);
  348. TEST_ASSERT(ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
  349. }
  350. exit:
  351. mbedtls_ecdsa_free(&ctx);
  352. mbedtls_ecdsa_restart_free(&rs_ctx);
  353. }
  354. /* END_CASE */
  355. /* BEGIN_CASE depends_on:MBEDTLS_ECP_RESTARTABLE:MBEDTLS_ECDSA_DETERMINISTIC */
  356. void ecdsa_write_restart(int id, char *d_str, int md_alg,
  357. data_t *hash, data_t *sig_check,
  358. int max_ops, int min_restart, int max_restart)
  359. {
  360. int ret, cnt_restart;
  361. mbedtls_ecdsa_restart_ctx rs_ctx;
  362. mbedtls_ecdsa_context ctx;
  363. unsigned char sig[MBEDTLS_ECDSA_MAX_LEN];
  364. size_t slen;
  365. mbedtls_ecdsa_restart_init(&rs_ctx);
  366. mbedtls_ecdsa_init(&ctx);
  367. memset(sig, 0, sizeof(sig));
  368. TEST_ASSERT(mbedtls_ecp_group_load(&ctx.grp, id) == 0);
  369. TEST_ASSERT(mbedtls_test_read_mpi(&ctx.d, d_str) == 0);
  370. mbedtls_ecp_set_max_ops(max_ops);
  371. slen = sizeof(sig);
  372. cnt_restart = 0;
  373. do {
  374. ret = mbedtls_ecdsa_write_signature_restartable(&ctx,
  375. md_alg,
  376. hash->x,
  377. hash->len,
  378. sig,
  379. sizeof(sig),
  380. &slen,
  381. mbedtls_test_rnd_std_rand,
  382. NULL,
  383. &rs_ctx);
  384. } while (ret == MBEDTLS_ERR_ECP_IN_PROGRESS && ++cnt_restart);
  385. TEST_ASSERT(ret == 0);
  386. TEST_ASSERT(slen == sig_check->len);
  387. TEST_ASSERT(memcmp(sig, sig_check->x, slen) == 0);
  388. TEST_ASSERT(cnt_restart >= min_restart);
  389. TEST_ASSERT(cnt_restart <= max_restart);
  390. /* Do we leak memory when aborting an operation?
  391. * This test only makes sense when we actually restart */
  392. if (min_restart > 0) {
  393. ret = mbedtls_ecdsa_write_signature_restartable(&ctx,
  394. md_alg,
  395. hash->x,
  396. hash->len,
  397. sig,
  398. sizeof(sig),
  399. &slen,
  400. mbedtls_test_rnd_std_rand,
  401. NULL,
  402. &rs_ctx);
  403. TEST_ASSERT(ret == MBEDTLS_ERR_ECP_IN_PROGRESS);
  404. }
  405. exit:
  406. mbedtls_ecdsa_restart_free(&rs_ctx);
  407. mbedtls_ecdsa_free(&ctx);
  408. }
  409. /* END_CASE */
  410. /* BEGIN_CASE */
  411. void ecdsa_verify(int grp_id, char *x, char *y, char *r, char *s, data_t *content, int expected)
  412. {
  413. mbedtls_ecdsa_context ctx;
  414. mbedtls_mpi sig_r, sig_s;
  415. mbedtls_ecdsa_init(&ctx);
  416. mbedtls_mpi_init(&sig_r);
  417. mbedtls_mpi_init(&sig_s);
  418. /* Prepare ECP group context */
  419. TEST_EQUAL(mbedtls_ecp_group_load(&ctx.grp, grp_id), 0);
  420. /* Prepare public key */
  421. TEST_EQUAL(mbedtls_test_read_mpi(&ctx.Q.X, x), 0);
  422. TEST_EQUAL(mbedtls_test_read_mpi(&ctx.Q.Y, y), 0);
  423. TEST_EQUAL(mbedtls_mpi_lset(&ctx.Q.Z, 1), 0);
  424. /* Prepare signature R & S */
  425. TEST_EQUAL(mbedtls_test_read_mpi(&sig_r, r), 0);
  426. TEST_EQUAL(mbedtls_test_read_mpi(&sig_s, s), 0);
  427. /* Test whether public key has expected validity */
  428. TEST_EQUAL(mbedtls_ecp_check_pubkey(&ctx.grp, &ctx.Q),
  429. expected == MBEDTLS_ERR_ECP_INVALID_KEY ? MBEDTLS_ERR_ECP_INVALID_KEY : 0);
  430. /* Verification */
  431. int result = mbedtls_ecdsa_verify(&ctx.grp, content->x, content->len, &ctx.Q, &sig_r, &sig_s);
  432. TEST_EQUAL(result, expected);
  433. exit:
  434. mbedtls_ecdsa_free(&ctx);
  435. mbedtls_mpi_free(&sig_r);
  436. mbedtls_mpi_free(&sig_s);
  437. }
  438. /* END_CASE */