Mercurial > dropbear
comparison mtest/mtest.c @ 2:86e0b50a9b58 libtommath-orig ltm-0.30-orig
ltm 0.30 orig import
author | Matt Johnston <matt@ucc.asn.au> |
---|---|
date | Mon, 31 May 2004 18:25:22 +0000 |
parents | |
children | d29b64170cf0 |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 2:86e0b50a9b58 |
---|---|
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)) % 1031; | |
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 = 1 + ((fgetc(rng)<<8) + fgetc(rng)) % 97; | |
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 (clock() - t1 > CLOCKS_PER_SEC) { | |
113 sleep(2); | |
114 t1 = clock(); | |
115 } | |
116 | |
117 n = fgetc(rng) % 15; | |
118 | |
119 if (n == 0) { | |
120 /* add tests */ | |
121 rand_num(&a); | |
122 rand_num(&b); | |
123 mp_add(&a, &b, &c); | |
124 printf("add\n"); | |
125 mp_to64(&a, buf); | |
126 printf("%s\n", buf); | |
127 mp_to64(&b, buf); | |
128 printf("%s\n", buf); | |
129 mp_to64(&c, buf); | |
130 printf("%s\n", buf); | |
131 } else if (n == 1) { | |
132 /* sub tests */ | |
133 rand_num(&a); | |
134 rand_num(&b); | |
135 mp_sub(&a, &b, &c); | |
136 printf("sub\n"); | |
137 mp_to64(&a, buf); | |
138 printf("%s\n", buf); | |
139 mp_to64(&b, buf); | |
140 printf("%s\n", buf); | |
141 mp_to64(&c, buf); | |
142 printf("%s\n", buf); | |
143 } else if (n == 2) { | |
144 /* mul tests */ | |
145 rand_num(&a); | |
146 rand_num(&b); | |
147 mp_mul(&a, &b, &c); | |
148 printf("mul\n"); | |
149 mp_to64(&a, buf); | |
150 printf("%s\n", buf); | |
151 mp_to64(&b, buf); | |
152 printf("%s\n", buf); | |
153 mp_to64(&c, buf); | |
154 printf("%s\n", buf); | |
155 } else if (n == 3) { | |
156 /* div tests */ | |
157 rand_num(&a); | |
158 rand_num(&b); | |
159 mp_div(&a, &b, &c, &d); | |
160 printf("div\n"); | |
161 mp_to64(&a, buf); | |
162 printf("%s\n", buf); | |
163 mp_to64(&b, buf); | |
164 printf("%s\n", buf); | |
165 mp_to64(&c, buf); | |
166 printf("%s\n", buf); | |
167 mp_to64(&d, buf); | |
168 printf("%s\n", buf); | |
169 } else if (n == 4) { | |
170 /* sqr tests */ | |
171 rand_num(&a); | |
172 mp_sqr(&a, &b); | |
173 printf("sqr\n"); | |
174 mp_to64(&a, buf); | |
175 printf("%s\n", buf); | |
176 mp_to64(&b, buf); | |
177 printf("%s\n", buf); | |
178 } else if (n == 5) { | |
179 /* mul_2d test */ | |
180 rand_num(&a); | |
181 mp_copy(&a, &b); | |
182 n = fgetc(rng) & 63; | |
183 mp_mul_2d(&b, n, &b); | |
184 mp_to64(&a, buf); | |
185 printf("mul2d\n"); | |
186 printf("%s\n", buf); | |
187 printf("%d\n", n); | |
188 mp_to64(&b, buf); | |
189 printf("%s\n", buf); | |
190 } else if (n == 6) { | |
191 /* div_2d test */ | |
192 rand_num(&a); | |
193 mp_copy(&a, &b); | |
194 n = fgetc(rng) & 63; | |
195 mp_div_2d(&b, n, &b, NULL); | |
196 mp_to64(&a, buf); | |
197 printf("div2d\n"); | |
198 printf("%s\n", buf); | |
199 printf("%d\n", n); | |
200 mp_to64(&b, buf); | |
201 printf("%s\n", buf); | |
202 } else if (n == 7) { | |
203 /* gcd test */ | |
204 rand_num(&a); | |
205 rand_num(&b); | |
206 a.sign = MP_ZPOS; | |
207 b.sign = MP_ZPOS; | |
208 mp_gcd(&a, &b, &c); | |
209 printf("gcd\n"); | |
210 mp_to64(&a, buf); | |
211 printf("%s\n", buf); | |
212 mp_to64(&b, buf); | |
213 printf("%s\n", buf); | |
214 mp_to64(&c, buf); | |
215 printf("%s\n", buf); | |
216 } else if (n == 8) { | |
217 /* lcm test */ | |
218 rand_num(&a); | |
219 rand_num(&b); | |
220 a.sign = MP_ZPOS; | |
221 b.sign = MP_ZPOS; | |
222 mp_lcm(&a, &b, &c); | |
223 printf("lcm\n"); | |
224 mp_to64(&a, buf); | |
225 printf("%s\n", buf); | |
226 mp_to64(&b, buf); | |
227 printf("%s\n", buf); | |
228 mp_to64(&c, buf); | |
229 printf("%s\n", buf); | |
230 } else if (n == 9) { | |
231 /* exptmod test */ | |
232 rand_num2(&a); | |
233 rand_num2(&b); | |
234 rand_num2(&c); | |
235 // if (c.dp[0]&1) mp_add_d(&c, 1, &c); | |
236 a.sign = b.sign = c.sign = 0; | |
237 mp_exptmod(&a, &b, &c, &d); | |
238 printf("expt\n"); | |
239 mp_to64(&a, buf); | |
240 printf("%s\n", buf); | |
241 mp_to64(&b, buf); | |
242 printf("%s\n", buf); | |
243 mp_to64(&c, buf); | |
244 printf("%s\n", buf); | |
245 mp_to64(&d, buf); | |
246 printf("%s\n", buf); | |
247 } else if (n == 10) { | |
248 /* invmod test */ | |
249 rand_num2(&a); | |
250 rand_num2(&b); | |
251 b.sign = MP_ZPOS; | |
252 a.sign = MP_ZPOS; | |
253 mp_gcd(&a, &b, &c); | |
254 if (mp_cmp_d(&c, 1) != 0) continue; | |
255 if (mp_cmp_d(&b, 1) == 0) continue; | |
256 mp_invmod(&a, &b, &c); | |
257 printf("invmod\n"); | |
258 mp_to64(&a, buf); | |
259 printf("%s\n", buf); | |
260 mp_to64(&b, buf); | |
261 printf("%s\n", buf); | |
262 mp_to64(&c, buf); | |
263 printf("%s\n", buf); | |
264 } else if (n == 11) { | |
265 rand_num(&a); | |
266 mp_mul_2(&a, &a); | |
267 mp_div_2(&a, &b); | |
268 printf("div2\n"); | |
269 mp_to64(&a, buf); | |
270 printf("%s\n", buf); | |
271 mp_to64(&b, buf); | |
272 printf("%s\n", buf); | |
273 } else if (n == 12) { | |
274 rand_num2(&a); | |
275 mp_mul_2(&a, &b); | |
276 printf("mul2\n"); | |
277 mp_to64(&a, buf); | |
278 printf("%s\n", buf); | |
279 mp_to64(&b, buf); | |
280 printf("%s\n", buf); | |
281 } else if (n == 13) { | |
282 rand_num2(&a); | |
283 tmp = abs(rand()) & THE_MASK; | |
284 mp_add_d(&a, tmp, &b); | |
285 printf("add_d\n"); | |
286 mp_to64(&a, buf); | |
287 printf("%s\n%d\n", buf, tmp); | |
288 mp_to64(&b, buf); | |
289 printf("%s\n", buf); | |
290 } else if (n == 14) { | |
291 rand_num2(&a); | |
292 tmp = abs(rand()) & THE_MASK; | |
293 mp_sub_d(&a, tmp, &b); | |
294 printf("sub_d\n"); | |
295 mp_to64(&a, buf); | |
296 printf("%s\n%d\n", buf, tmp); | |
297 mp_to64(&b, buf); | |
298 printf("%s\n", buf); | |
299 } | |
300 } | |
301 fclose(rng); | |
302 return 0; | |
303 } |