Commit 1a12486d4b8b55079b963d7830a25493235c0912
2019-05-08T21:08:16
add mp_rand test
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
diff --git a/demo/test.c b/demo/test.c
index 01bc959..332348d 100644
--- a/demo/test.c
+++ b/demo/test.c
@@ -68,6 +68,45 @@ LBL_ERR:
return EXIT_FAILURE;
}
+static int very_random_source(void *out, size_t size)
+{
+ memset(out, 0xff, size);
+ return MP_OKAY;
+}
+
+static int test_mp_rand(void)
+{
+ mp_int a, b;
+ int err, n;
+ if (mp_init_multi(&a, &b, NULL)!= MP_OKAY) {
+ return EXIT_FAILURE;
+ }
+ mp_rand_source(very_random_source);
+ for (n = 1; n < 1024; ++n) {
+ if ((err = mp_rand(&a, n)) != MP_OKAY) {
+ printf("Failed mp_rand() %s.\n", mp_error_to_string(err));
+ break;
+ }
+ if ((err = mp_incr(&a)) != MP_OKAY) {
+ printf("Failed mp_incr() %s.\n", mp_error_to_string(err));
+ break;
+ }
+ if ((err = mp_div_2d(&a, n * MP_DIGIT_BIT, &b, NULL)) != MP_OKAY) {
+ printf("Failed mp_div_2d() %s.\n", mp_error_to_string(err));
+ break;
+ }
+ if (mp_cmp_d(&b, 1) != MP_EQ) {
+ ndraw(&a, "mp_rand() a");
+ ndraw(&b, "mp_rand() b");
+ err = MP_ERR;
+ break;
+ }
+ }
+ mp_rand_source(NULL);
+ mp_clear_multi(&a, &b, NULL);
+ return err == MP_OKAY ? EXIT_SUCCESS : EXIT_FAILURE;
+}
+
static int test_mp_jacobi(void)
{
struct mp_jacobi_st {
@@ -1828,6 +1867,7 @@ int unit_tests(int argc, char **argv)
T(mp_montgomery_reduce),
T(mp_prime_is_prime),
T(mp_prime_random_ex),
+ T(mp_rand),
T(mp_read_radix),
T(mp_reduce_2k),
T(mp_reduce_2k_l),