Mercurial > dropbear

comparison libtommath/mtest/mtest.c @ 330:5488db2e9e4e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
merge of 332f709a4cb39cde4cedab7c3be89e05f3023067 and ca4ca78b82c5d430c69ce01bf794e8886ce81431
author Matt Johnston <matt@ucc.asn.au>
date Sat, 10 Jun 2006 16:39:40 +0000
parents eed26cff980b
children 5ff8218bcee9
comparison
equal deleted inserted replaced
329:8ed0dce45126 330:5488db2e9e4e
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 }