userapps/opensource/sshd/libtommath/mtest/mtest.c

   1 /* makes a bignum test harness with NUM tests per operation
   2  *
   3  * the output is made in the following format [one parameter per line]
   4
   5 operation
   6 operand1
   7 operand2
   8 [... operandN]
   9 result1
  10 result2
  11 [... resultN]
  12
  13 So for example "a * b mod n" would be
  14
  15 mulmod
  16 a
  17 b
  18 n
  19 a*b mod n
  20
  21 e.g. if a=3, b=4 n=11 then
  22
  23 mulmod
  24 3
  25 4
  26 11
  27 1
  28
  29  */
  30
  31 #ifdef MP_8BIT
  32 #define THE_MASK 127
  33 #else
  34 #define THE_MASK 32767
  35 #endif
  36
  37 #include <stdio.h>
  38 #include <stdlib.h>
  39 #include <time.h>
  40 #include "mpi.c"
  41
  42 FILE *rng;
  43
  44 void rand_num(mp_int *a)
  45 {
  46    int n, size;
  47    unsigned char buf[2048];
  48
  49    size = 1 + ((fgetc(rng)<<8) + fgetc(rng)) % 101;
  50    buf[0] = (fgetc(rng)&1)?1:0;
  51    fread(buf+1, 1, size, rng);
  52    while (buf[1] == 0) buf[1] = fgetc(rng);
  53    mp_read_raw(a, buf, 1+size);
  54 }
  55
  56 void rand_num2(mp_int *a)
  57 {
  58    int n, size;
  59    unsigned char buf[2048];
  60
  61    size = 10 + ((fgetc(rng)<<8) + fgetc(rng)) % 101;
  62    buf[0] = (fgetc(rng)&1)?1:0;
  63    fread(buf+1, 1, size, rng);
  64    while (buf[1] == 0) buf[1] = fgetc(rng);
  65    mp_read_raw(a, buf, 1+size);
  66 }
  67
  68 #define mp_to64(a, b) mp_toradix(a, b, 64)
  69
  70 int main(void)
  71 {
  72    int n, tmp;
  73    mp_int a, b, c, d, e;
  74    clock_t t1;
  75    char buf[4096];
  76
  77    mp_init(&a);
  78    mp_init(&b);
  79    mp_init(&c);
  80    mp_init(&d);
  81    mp_init(&e);
  82
  83
  84    /* initial (2^n - 1)^2 testing, makes sure the comba multiplier works [it has the new carry code] */
  85 /*
  86    mp_set(&a, 1);
  87    for (n = 1; n < 8192; n++) {
  88        mp_mul(&a, &a, &c);
  89        printf("mul\n");
  90        mp_to64(&a, buf);
  91        printf("%s\n%s\n", buf, buf);
  92        mp_to64(&c, buf);
  93        printf("%s\n", buf);
  94
  95        mp_add_d(&a, 1, &a);
  96        mp_mul_2(&a, &a);
  97        mp_sub_d(&a, 1, &a);
  98    }
  99 */
 100
 101    rng = fopen("/dev/urandom", "rb");
 102    if (rng == NULL) {
 103       rng = fopen("/dev/random", "rb");
 104       if (rng == NULL) {
 105          fprintf(stderr, "\nWarning:  stdin used as random source\n\n");
 106          rng = stdin;
 107       }
 108    }
 109
 110    t1 = clock();
 111    for (;;) {
 112 #if 0
 113       if (clock() - t1 > CLOCKS_PER_SEC) {
 114          sleep(2);
 115          t1 = clock();
 116       }
 117 #endif
 118        n = fgetc(rng) % 15;
 119
 120    if (n == 0) {
 121        /* add tests */
 122        rand_num(&a);
 123        rand_num(&b);
 124        mp_add(&a, &b, &c);
 125        printf("add\n");
 126        mp_to64(&a, buf);
 127        printf("%s\n", buf);
 128        mp_to64(&b, buf);
 129        printf("%s\n", buf);
 130        mp_to64(&c, buf);
 131        printf("%s\n", buf);
 132    } else if (n == 1) {
 133       /* sub tests */
 134        rand_num(&a);
 135        rand_num(&b);
 136        mp_sub(&a, &b, &c);
 137        printf("sub\n");
 138        mp_to64(&a, buf);
 139        printf("%s\n", buf);
 140        mp_to64(&b, buf);
 141        printf("%s\n", buf);
 142        mp_to64(&c, buf);
 143        printf("%s\n", buf);
 144    } else if (n == 2) {
 145        /* mul tests */
 146        rand_num(&a);
 147        rand_num(&b);
 148        mp_mul(&a, &b, &c);
 149        printf("mul\n");
 150        mp_to64(&a, buf);
 151        printf("%s\n", buf);
 152        mp_to64(&b, buf);
 153        printf("%s\n", buf);
 154        mp_to64(&c, buf);
 155        printf("%s\n", buf);
 156    } else if (n == 3) {
 157       /* div tests */
 158        rand_num(&a);
 159        rand_num(&b);
 160        mp_div(&a, &b, &c, &d);
 161        printf("div\n");
 162        mp_to64(&a, buf);
 163        printf("%s\n", buf);
 164        mp_to64(&b, buf);
 165        printf("%s\n", buf);
 166        mp_to64(&c, buf);
 167        printf("%s\n", buf);
 168        mp_to64(&d, buf);
 169        printf("%s\n", buf);
 170    } else if (n == 4) {
 171       /* sqr tests */
 172        rand_num(&a);
 173        mp_sqr(&a, &b);
 174        printf("sqr\n");
 175        mp_to64(&a, buf);
 176        printf("%s\n", buf);
 177        mp_to64(&b, buf);
 178        printf("%s\n", buf);
 179    } else if (n == 5) {
 180       /* mul_2d test */
 181       rand_num(&a);
 182       mp_copy(&a, &b);
 183       n = fgetc(rng) & 63;
 184       mp_mul_2d(&b, n, &b);
 185       mp_to64(&a, buf);
 186       printf("mul2d\n");
 187       printf("%s\n", buf);
 188       printf("%d\n", n);
 189       mp_to64(&b, buf);
 190       printf("%s\n", buf);
 191    } else if (n == 6) {
 192       /* div_2d test */
 193       rand_num(&a);
 194       mp_copy(&a, &b);
 195       n = fgetc(rng) & 63;
 196       mp_div_2d(&b, n, &b, NULL);
 197       mp_to64(&a, buf);
 198       printf("div2d\n");
 199       printf("%s\n", buf);
 200       printf("%d\n", n);
 201       mp_to64(&b, buf);
 202       printf("%s\n", buf);
 203    } else if (n == 7) {
 204       /* gcd test */
 205       rand_num(&a);
 206       rand_num(&b);
 207       a.sign = MP_ZPOS;
 208       b.sign = MP_ZPOS;
 209       mp_gcd(&a, &b, &c);
 210       printf("gcd\n");
 211       mp_to64(&a, buf);
 212       printf("%s\n", buf);
 213       mp_to64(&b, buf);
 214       printf("%s\n", buf);
 215       mp_to64(&c, buf);
 216       printf("%s\n", buf);
 217    } else if (n == 8) {
 218       /* lcm test */
 219       rand_num(&a);
 220       rand_num(&b);
 221       a.sign = MP_ZPOS;
 222       b.sign = MP_ZPOS;
 223       mp_lcm(&a, &b, &c);
 224       printf("lcm\n");
 225       mp_to64(&a, buf);
 226       printf("%s\n", buf);
 227       mp_to64(&b, buf);
 228       printf("%s\n", buf);
 229       mp_to64(&c, buf);
 230       printf("%s\n", buf);
 231    } else if (n == 9) {
 232       /* exptmod test */
 233       rand_num2(&a);
 234       rand_num2(&b);
 235       rand_num2(&c);
 236 //      if (c.dp[0]&1) mp_add_d(&c, 1, &c);
 237       a.sign = b.sign = c.sign = 0;
 238       mp_exptmod(&a, &b, &c, &d);
 239       printf("expt\n");
 240       mp_to64(&a, buf);
 241       printf("%s\n", buf);
 242       mp_to64(&b, buf);
 243       printf("%s\n", buf);
 244       mp_to64(&c, buf);
 245       printf("%s\n", buf);
 246       mp_to64(&d, buf);
 247       printf("%s\n", buf);
 248    } else if (n == 10) {
 249       /* invmod test */
 250       rand_num2(&a);
 251       rand_num2(&b);
 252       b.sign = MP_ZPOS;
 253       a.sign = MP_ZPOS;
 254       mp_gcd(&a, &b, &c);
 255       if (mp_cmp_d(&c, 1) != 0) continue;
 256       if (mp_cmp_d(&b, 1) == 0) continue;
 257       mp_invmod(&a, &b, &c);
 258       printf("invmod\n");
 259       mp_to64(&a, buf);
 260       printf("%s\n", buf);
 261       mp_to64(&b, buf);
 262       printf("%s\n", buf);
 263       mp_to64(&c, buf);
 264       printf("%s\n", buf);
 265    } else if (n == 11) {
 266       rand_num(&a);
 267       mp_mul_2(&a, &a);
 268       mp_div_2(&a, &b);
 269       printf("div2\n");
 270       mp_to64(&a, buf);
 271       printf("%s\n", buf);
 272       mp_to64(&b, buf);
 273       printf("%s\n", buf);
 274    } else if (n == 12) {
 275       rand_num2(&a);
 276       mp_mul_2(&a, &b);
 277       printf("mul2\n");
 278       mp_to64(&a, buf);
 279       printf("%s\n", buf);
 280       mp_to64(&b, buf);
 281       printf("%s\n", buf);
 282    } else if (n == 13) {
 283       rand_num2(&a);
 284       tmp = abs(rand()) & THE_MASK;
 285       mp_add_d(&a, tmp, &b);
 286       printf("add_d\n");
 287       mp_to64(&a, buf);
 288       printf("%s\n%d\n", buf, tmp);
 289       mp_to64(&b, buf);
 290       printf("%s\n", buf);
 291    } else if (n == 14) {
 292       rand_num2(&a);
 293       tmp = abs(rand()) & THE_MASK;
 294       mp_sub_d(&a, tmp, &b);
 295       printf("sub_d\n");
 296       mp_to64(&a, buf);
 297       printf("%s\n%d\n", buf, tmp);
 298       mp_to64(&b, buf);
 299       printf("%s\n", buf);
 300    }
 301    }
 302    fclose(rng);
 303    return 0;
 304 }