Mercurial > dropbear
comparison libtommath/mtest/mtest.c @ 302:973fccb59ea4 ucc-axis-hack
propagate from branch 'au.asn.ucc.matt.dropbear' (head 11034278bd1917bebcbdc69cf53b1891ce9db121)
to branch 'au.asn.ucc.matt.dropbear.ucc-axis-hack' (head 10a1f614fec73d0820c3f61160d9db409b9beb46)
author | Matt Johnston <matt@ucc.asn.au> |
---|---|
date | Sat, 25 Mar 2006 12:59:58 +0000 |
parents | eed26cff980b |
children | 5ff8218bcee9 |
comparison
equal
deleted
inserted
replaced
299:740e782679be | 302:973fccb59ea4 |
---|---|
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 } |