]> git.cryptolib.org Git - arm-crypto-lib.git/blob - host/bigint_test.rb
adding mgf1 source
[arm-crypto-lib.git] / host / bigint_test.rb
1 #!/usr/bin/ruby
2 # bigint_test.rb
3 =begin
4     This file is part of the AVR-Crypto-Lib.
5     Copyright (C) 2008, 2009  Daniel Otte (daniel.otte@rub.de)
6
7     This program is free software: you can redistribute it and/or modify
8     it under the terms of the GNU General Public License as published by
9     the Free Software Foundation, either version 3 of the License, or
10     (at your option) any later version.
11
12     This program is distributed in the hope that it will be useful,
13     but WITHOUT ANY WARRANTY; without even the implied warranty of
14     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15     GNU General Public License for more details.
16
17     You should have received a copy of the GNU General Public License
18     along with this program.  If not, see <http://www.gnu.org/licenses/>.
19 =end
20
21 $debug = true
22 $debug = false
23 require 'rubygems'
24 require 'serialport'
25 require 'getopt/std'
26 require 'fileutils'
27 require 'date'
28 $buffer_size = 0
29 $conffile_check = Hash.new
30 $conffile_check.default = 0
31
32 ################################################################################
33 # readconfigfile                                                               #
34 ################################################################################
35
36 def readconfigfile(fname, conf)
37   return conf if $conffile_check[fname]==1
38   $conffile_check[fname]=1
39   section = "default"
40   if ! File.exists?(fname)
41     return conf
42   end
43   file = File.open(fname, "r")
44   until file.eof
45     line = file.gets()
46         next if /[\s]*#/.match(line)
47         if m=/\[[\s]*([^\s]*)[\s]*\]/.match(line)
48           section=m[1]
49           conf[m[1]] = Hash.new
50           next
51         end
52         next if ! /=/.match(line)
53         m=/[\s]*([^\s]*)[\s]*=[\s]*([^\s]*)/.match(line)
54         if m[1]=="include"
55           Dir.glob(m[2]){ |fn| conf = readconfigfile(fn, conf) }
56         else
57           conf[section][m[1]] = m[2]
58         end
59   end
60   file.close()
61   return conf
62 end
63
64 ################################################################################
65 # expmod                                                                       #
66 ################################################################################
67
68 def expmod(base, power, mod)
69   result = 1
70   while power > 0
71     result = (result * base) % mod if power & 1 == 1
72     base = (base * base) % mod
73     power >>= 1;
74   end
75   return result
76 end
77
78 ################################################################################
79 # gcdext                                                                 #
80 ################################################################################
81
82 def gcdext(x,y)
83   g=1
84   while(x&1==0 && y&1==0) do
85     x>>=1
86     y>>=1
87     g<<=1
88   end
89   u=x; v=y; a=1; b=0; c=0; d=1
90   begin
91     while(u&1==0) do
92       if(a%2==1 || b%2==1)
93         a += y
94         b -= x
95       end
96       u>>=1; a>>=1; b>>=1
97     end
98     while(v&1==0) do
99       if(c%2==1 || d%2==1)
100         c += y
101         d -= x
102       end
103       v>>=1; c>>=1; d>>=1;
104     end
105     if(u>=v)
106       u -= v; a-=c; b-=d
107     else
108       v -= u; c-=a; d-=b
109     end
110   end while(u!=0)
111   return[g*v, c, d]
112 end
113
114 ################################################################################
115 # reset_system                                                                 #
116 ################################################################################
117
118 def reset_system
119   $sp.print("exit\r")
120   sleep 0.1
121   $sp.print("exit\r")
122   sleep 0.1
123 end
124
125 ################################################################################
126 # init_system                                                                  #
127 ################################################################################
128
129 def init_system(test_prog)
130   begin
131     line = $sp.gets()
132   end while line!=nil
133   $sp.print("echo off \r")
134   print("DBG i: " + "echo off \r"+"\n") if $debug
135   sleep 0.1
136   $sp.print("#{test_prog}\r")
137   print("DBG i: " + "#{test_prog} \r"+"\n") if $debug
138   sleep 1
139 end
140
141 ################################################################################
142 # wait_for_prompt                                                                  #
143 ################################################################################
144
145 def wait_for_prompt(prompt)
146   prompt = /[\s]*#{prompt}[\s]*/ if(prompt.class == String)
147   start_time = Time.now.to_i
148   acc = ''
149   begin
150     line = $sp.gets()
151     puts("DBG got (#{__LINE__}): "+line) if $debug && line
152     line = "" if line==nil
153     if /^(Error:|Crypto-VS).*/.match(line)
154       puts line
155       return false
156     end
157     if (Time.now.to_i- start_time) > $max_timeout
158       return false
159     end
160   acc += line
161   end while ! m=prompt.match(acc)
162   return m
163 end
164
165 ################################################################################
166 # screen_progress                                                              #
167 ################################################################################
168 def screen_progress(v)
169   if $linepos==0
170     printf("\n%5d [%04d]: ", $testno, $size)
171   end
172   putc((v)?('*'):('!'))
173   $testno += 1
174   $linepos = ($linepos+1)%$linewidth
175 end
176
177 ################################################################################
178 # add_test                                                                     #
179 ################################################################################
180
181 def add_test(a,b)
182   begin
183     line = $sp.gets()
184     line = "" if line==nil
185     puts("DBG got: "+line) if $debug
186     if /^Error:.*/.match(line)
187       puts line
188       return false
189     end
190   end while ! /[\s]*enter a:[\s]*/.match(line)
191   $sp.print(a.to_s(16)+" ")
192   begin
193     line = $sp.gets()
194     line = "" if line==nil
195     puts("DBG got: "+line) if $debug
196     if /^Error:.*/.match(line)
197       puts line
198       return false
199     end
200   end while ! /[\s]*enter b:[\s]*/.match(line)
201   $sp.print(b.to_s(16)+" ")
202   begin
203     line = $sp.gets()
204     line = "" if line==nil
205     puts("DBG got: "+line) if $debug
206     if /^Error:.*/.match(line)
207       puts line
208       return false
209     end
210   end while ! m=/[\s]*([-]?[0-9a-fA-F]*)[\s]+\+[\s]+([+-]?[0-9a-fA-F]*)[\s]*=[\s]*([+-]?[0-9a-fA-F]*)/.match(line)
211   a_ = m[1].to_i(16)
212   b_ = m[2].to_i(16)
213   c_ = m[3].to_i(16)
214   line.chomp!
215   if(a_== a && b_ == b && c_ == (a+b))
216     $logfile.printf("[pass]: %s\n", line)
217     return true
218   else
219     $logfile.printf("[fail (%s%s%s)]: %s", (a==a_)?"":"a", (b==b_)?"":"b", (c_==a+b)?"":"c",line)
220     $logfile.printf(" ; should %s + %s = %s\n", a.to_s(16), b.to_s(16), (a+b).to_s(16))
221     return false
222   end
223   return false
224 end
225
226 ################################################################################
227 # mul_test                                                                     #
228 ################################################################################
229
230 def mul_test(a,b)
231   begin
232     line = $sp.gets()
233     line = "" if line==nil
234     puts("DBG got: "+line) if $debug
235     if /^Error:.*/.match(line)
236       puts line
237       return false
238     end
239   end while ! /[\s]*enter a:[\s]*/.match(line)
240   $sp.print(a.to_s(16)+" ")
241   begin
242     line = $sp.gets()
243     line = "" if line==nil
244     puts("DBG got: "+line) if $debug
245     if /^Error:.*/.match(line)
246       puts line
247       return false
248     end
249   end while ! /[\s]*enter b:[\s]*/.match(line)
250   $sp.print(b.to_s(16)+" ")
251   begin
252     line = $sp.gets()
253     line = "" if line==nil
254     puts("DBG got: "+line) if $debug
255     if /^Error:.*/.match(line)
256       puts line
257       return false
258     end
259   end while ! m=/[\s]*([+-]?[0-9a-fA-F]*)[\s]+\*[\s]+([+-]?[0-9a-fA-F]*)[\s]*=[\s]*([+-]?[0-9a-fA-F]*)/.match(line)
260   a_ = m[1].to_i(16)
261   b_ = m[2].to_i(16)
262   c_ = m[3].to_i(16)
263   line.chomp!
264   if(a_== a && b_ == b && c_ == (a*b))
265     $logfile.printf("[pass]: %s\n", line)
266     return true
267   else
268     $logfile.printf("[fail (%s%s%s)]: %s", (a==a_)?"":"a", (b==b_)?"":"b", (c_==a*b)?"":"c",line)
269     $logfile.printf(" ; should %s * %s = %s\n", a.to_s(16), b.to_s(16), (a*b).to_s(16))
270     return false
271   end
272   return false
273 end
274
275 ################################################################################
276 # add_scale_test                                                               #
277 ################################################################################
278 def add_scale_test_dummy(a, b, scale)
279   should = a + (b<<(8*scale))
280   printf("[dummy] %s + %s <<8*%04x = %s\n",a.to_s(16), b.to_s(16), scale, should.to_s(16))
281 end
282
283 def add_scale_test(a, b, scale)
284   m = wait_for_prompt("enter a:")
285   return false if !m 
286   puts("DBG put (#{__LINE__}): "+a.to_s(16)+" ") if $debug
287   $sp.print(a.to_s(16)+" ")
288   m = wait_for_prompt("enter b:")
289   return false if !m 
290   puts("DBG put (#{__LINE__}): "+b.to_s(16)+" ") if $debug
291   $sp.print(b.to_s(16)+" ")
292   m = wait_for_prompt("enter scale:")
293   return false if !m 
294   puts("DBG put (#{__LINE__}): "+scale.to_s(10)+" ") if $debug
295   $sp.print(scale.to_s(10)+"\r")
296   should = a + (b<<(8*scale))
297   m = wait_for_prompt(/[\s]*([-]?[0-9a-fA-F]+)[\s]+\+[\s]+([+-]?[0-9a-fA-F]+)[\s]*<<8\*[\s]*([+-]?[0-9a-fA-F]+)[\s]*=[\s]*([+-]?[0-9a-fA-F]+)/)
298   if !m 
299     $logfile.printf("[fail (CRASH)]:")
300     $logfile.printf(" ; should %s + %s << 8*%s = %s\n", a.to_s(16), b.to_s(16), scale.to_s(16), should.to_s(16))
301     return false
302   end 
303   line = m[0]
304   a_ = m[1].to_i(16)
305   b_ = m[2].to_i(16)
306   s_ = m[3].to_i(16)
307   c_ = m[4].to_i(16)
308   line.chomp!
309   if(a_== a && b_ == b && s_ == scale && c_ == should )
310     $logfile.printf("[pass]: %s\n", line)
311     return true
312   else
313     $logfile.printf("[fail (%s%s%s%s)]: %s", (a==a_)?"":"a", (b==b_)?"":"b", (scale==s_)?"":"s",(c_==should)?"":"c",line)
314     $logfile.printf(" ; should %s + %s << 8*%s = %s\n", a.to_s(16), b.to_s(16), scale.to_s(16), should.to_s(16))
315     return false
316   end
317   return false
318 end
319
320 ################################################################################
321 # square_test                                                                  #
322 ################################################################################
323
324 def square_test(a)
325   begin
326     line = $sp.gets()
327     line = "" if line==nil
328     puts("DBG got: "+line) if $debug
329     if /^Error:.*/.match(line)
330       puts line
331       return false
332     end
333   end while ! /[\s]*enter a:[\s]*/.match(line)
334   $sp.print(a.to_s(16)+" ")
335   begin
336     line = $sp.gets()
337     line = "" if line==nil
338     puts("DBG got: "+line) if $debug
339     if /^Error:.*/.match(line)
340       puts line
341       return false
342     end
343   end while ! m=/[\s]*([+-]?[0-9a-fA-F]*)[\s]*\*\*2[\s]*=[\s]*([+-]?[0-9a-fA-F]*)/.match(line)
344   a_ = m[1].to_i(16)
345   c_ = m[2].to_i(16)
346   line.chomp!
347   if(a_== a && c_ == (a**2))
348     $logfile.printf("[pass]: %s\n", line)
349     return true
350   else
351     $logfile.printf("[fail (%s%s)]: %s", (a==a_)?"":"a", (c_==a**2)?"":"c",line)
352     $logfile.printf(" ; should %s **2 = %s\n", a.to_s(16), (a**2).to_s(16))
353     return false
354   end
355   return false
356 end
357
358 ################################################################################
359 # reduce_test                                                                  #
360 ################################################################################
361
362 def reduce_test(a,b)
363   begin
364     line = $sp.gets()
365     line = "" if line==nil
366     puts("DBG got: "+line) if $debug
367     if /^Error:.*/.match(line)
368       puts line
369       return false
370     end
371   end while ! /[\s]*enter a:[\s]*/.match(line)
372   $sp.print(a.to_s(16)+" ")
373   begin
374     line = $sp.gets()
375     line = "" if line==nil
376     puts("DBG got: "+line) if $debug
377     if /^Error:.*/.match(line)
378       puts line
379       return false
380     end
381   end while ! /[\s]*enter b:[\s]*/.match(line)
382   $sp.print(b.to_s(16)+" ")
383   line=''
384   begin
385     line_tmp = $sp.gets()
386     line_tmp = '' if line_tmp==nil
387     line += line_tmp
388     puts("DBG got: "+line) if $debug
389     if /^Error:.*/.match(line)
390       puts line
391       return false
392     end
393   end while ! m=/[\s]*([+-]?[0-9a-fA-F]*)[\s]+%[\s]+([+-]?[0-9a-fA-F]*)[\s]*=[\s]*([+-]?[0-9a-fA-F]+)/.match(line)
394   a_ = m[1].to_i(16)
395   b_ = m[2].to_i(16)
396   c_ = m[3].to_i(16)
397   line.chomp!
398   if(a_== a && b_ == b && c_ == (a%b))
399     $logfile.printf("[pass]: %s\n", line)
400     return true
401   else
402     $logfile.printf("[fail (%s%s%s)]: %s", (a==a_)?"":"a", (b==b_)?"":"b", (c_==a%b)?"":"c",line)
403     $logfile.printf(" ; should %s %% %s = %s\n", a.to_s(16), b.to_s(16), (a%b).to_s(16))
404     return false
405   end
406   return false
407 end
408
409 ################################################################################
410 # expmod_test                                                                  #
411 ################################################################################
412
413 def expmod_test(a,b,c)
414   begin
415     line = $sp.gets()
416     line = "" if line==nil
417     puts("DBG got: "+line) if $debug
418     if /^Error:.*/.match(line)
419       puts line
420       return false
421     end
422   end while ! /[\s]*enter a:[\s]*/.match(line)
423   $sp.print(a.to_s(16)+" ")
424   begin
425     line = $sp.gets()
426     line = "" if line==nil
427     puts("DBG got: "+line) if $debug
428     if /^Error:.*/.match(line)
429       puts line
430       return false
431     end
432   end while ! /[\s]*enter b:[\s]*/.match(line)
433   $sp.print(b.to_s(16)+" ")
434   begin
435     line = $sp.gets()
436     line = "" if line==nil
437     puts("DBG got: "+line) if $debug
438     if /^Error:.*/.match(line)
439       puts line
440       return false
441     end
442   end while ! /[\s]*enter c:[\s]*/.match(line)
443   $sp.print(c.to_s(16)+" ")
444   line=''
445   begin
446     line_tmp = $sp.gets()
447     line_tmp = '' if line_tmp==nil
448     line += line_tmp
449     puts("DBG got: "+line) if $debug
450     if /^Error:.*/.match(line)
451       puts line
452       return false
453     end
454   end while ! m=/[\s]*([+-]?[0-9a-fA-F]*)\*\*([+-]?[0-9a-fA-F]*)[\s]+%[\s]+([+-]?[0-9a-fA-F]*)[\s]*=[\s]*([+-]?[0-9a-fA-F]+)/.match(line)
455   a_ = m[1].to_i(16)
456   b_ = m[2].to_i(16)
457   c_ = m[3].to_i(16)
458   d_ = m[4].to_i(16)
459   line.chomp!
460   if(a_== a && b_ == b && c_ == c && d_ ==expmod(a,b,c) )
461     $logfile.printf("[pass]: %s\n", line)
462     return true
463   else
464     $logfile.printf("[fail (%s%s%s%s)]: %s", (a==a_)?'':'a', (b==b_)?'':'b', (c_==c)?'':'c', (d_==expmod(a,b,c))?'':'d',line)
465     $logfile.printf(" ; should %s**%s %% %s = %s\n", a.to_s(16), b.to_s(16), c.to_s(16), expmod(a,b,c).to_s(16))
466     return false
467   end
468   return false
469 end
470
471 ################################################################################
472 # gcdext_test                                                                  #
473 ################################################################################
474
475 def gcdext_test(a,b)
476   begin
477     line = $sp.gets()
478     line = "" if line==nil
479     puts("DBG got: "+line) if $debug
480     if /^Error:.*/.match(line)
481       puts line
482       return false
483     end
484   end while ! /[\s]*enter a:[\s]*/.match(line)
485   $sp.print(a.to_s(16)+" ")
486   begin
487     line = $sp.gets()
488     line = "" if line==nil
489     puts("DBG got: "+line) if $debug
490     if /^Error:.*/.match(line)
491       puts line
492       return false
493     end
494   end while ! /[\s]*enter b:[\s]*/.match(line)
495   $sp.print(b.to_s(16)+" ")
496   line=''
497   begin
498     line_tmp = $sp.gets()
499     line_tmp = '' if line_tmp==nil
500     line = ''  if line.end_with?('\n')
501     line += line_tmp
502     puts("DBG got: "+line) if $debug
503     if /^Error:.*/.match(line)
504       puts line
505       return false
506     end
507   end while ! m=/gcdext\([\s]*([+-]?[0-9a-fA-F]*)[\s]*,[\s]*([+-]?[0-9a-fA-F]*)[\s]*\)[\s]*=> a = ([+-]?[0-9a-fA-F]+); b = ([+-]?[0-9a-fA-F]+); gcd = ([+-]?[0-9a-fA-F]+)/.match(line)
508   a_ = m[1].to_i(16)
509   b_ = m[2].to_i(16)
510   c_ = m[3].to_i(16)
511   d_ = m[4].to_i(16)
512   e_ = m[5].to_i(16)
513   line.chomp!
514   line.gsub!("\r",'')
515   line.gsub!("\n",'')
516   ref = gcdext(a,b)
517   if(a_== a && b_ == b && c_ == ref[1] && d_ == ref[2] && e_== ref[0])
518     $logfile.printf("[pass]: %s\n", line)
519     return true
520   else
521     $logfile.printf("[fail (%s%s%s%s%s)]: %s", (a==a_)?'':'a', (b==b_)?'':'b', 
522        (c_==ref[1])?'':'c', (d_==ref[2])?'':'d', (e_==ref[0])?'':'e', line)
523     $logfile.printf(" ; should gcdext( %s, %s) => a = %s; b = %s; gcd = %s\n",
524        a.to_s(16), b.to_s(16), ref[1].to_s(16), ref[2].to_s(16), ref[0].to_s(16))
525     return false
526   end
527   return false
528 end
529
530 ################################################################################
531 # run_test_add                                                                 #
532 ################################################################################
533
534 def run_test_add(skip=0)
535   length_a_B = skip+1
536   length_b_B = skip+1
537   begin
538     $size = length_a_B
539     (0..16).each do |i|
540       s_a = rand(2)*2-1
541       s_b = rand(2)*2-1
542       a = rand(256**length_a_B) * s_a
543       b = rand(256**length_a_B) * s_b
544       v = add_test(a, b)
545       screen_progress(v)
546       v = add_test(b, a)
547       screen_progress(v)
548     end
549     (0..16).each do |i|
550       s_a = rand(2)-1
551       s_b = rand(2)-1
552       b_size = rand(length_b_B+1)
553       a = rand(256**length_a_B) * s_a
554       b = rand(256**b_size) * s_b
555       v = add_test(a, b)
556       screen_progress(v)      
557       v = add_test(b, a)
558       screen_progress(v)
559
560     end
561     length_a_B += 1
562     length_b_B += 1
563   end while length_a_B<4096/8
564 end
565
566 ################################################################################
567 # run_test_add_scale                                                           #
568 ################################################################################
569
570 def run_test_add_scale(skip=0)
571   length_a_B = skip+1
572   length_b_B = skip+1
573   begin
574     $size = length_a_B
575     (0..4).each do |i|
576       scales = [0, 300]
577       16.times { scales << rand(301) }
578       scales.sort!
579       scales.each do |scale|
580         a = rand(256**length_a_B)
581         b = rand(256**length_a_B)
582         v = add_scale_test(a, b, scale)
583         screen_progress(v)
584         v = add_scale_test(b, a, scale)
585         screen_progress(v)
586       end
587     end
588     (0..4).each do |i|
589       scales = [0, 300]
590       16.times { scales << rand(301) }
591       scales.sort!
592       scales.each do |scale|
593         b_size = rand(length_b_B+1)+1
594         a = rand(256**length_a_B)
595         b = rand(256**b_size)
596         v = add_scale_test(a, b, scale)
597         screen_progress(v)      
598         v = add_scale_test(b, a, scale)
599         screen_progress(v)
600       end
601     end
602     length_a_B += 10
603     length_b_B += 10
604   end while length_a_B<4096/8
605 end
606
607 def run_test_add_scale_dummy(skip=0)
608   length_a_B = skip+1
609   length_b_B = skip+1
610   begin
611     $size = length_a_B
612     (0..4).each do |i|
613       scales = [0, 300]
614       16.times { scales << rand(301) }
615       scales.sort!
616       scales.each do |scale|
617         a = rand(256**length_a_B)
618         b = rand(256**length_a_B)
619         v = add_scale_test_dummy(a, b, scale)
620         v = add_scale_test_dummy(b, a, scale)
621       end
622     end
623     (0..4).each do |i|
624       scales = [0, 300]
625       16.times { scales << rand(301) }
626       scales.sort!
627       scales.each do |scale|
628         b_size = rand(length_b_B+1)
629         a = rand(256**length_a_B)
630         b = rand(256**b_size)
631         v = add_scale_test_dummy(a, b, scale)
632         v = add_scale_test_dummy(b, a, scale)
633       end
634     end
635     length_a_B += 10
636     length_b_B += 10
637   end while length_a_B<4096/8
638 end
639
640 ################################################################################
641 # run_test_mul                                                                 #
642 ################################################################################
643
644 def run_test_mul(skip=0)
645   length_a_B = skip+1
646   length_b_B = skip+1
647   begin
648     $size = length_a_B
649     (0..16).each do |i|
650       s_a = rand(2)*2-1
651       s_b = rand(2)*2-1
652       a = rand(256**length_a_B) * s_a
653       b = rand(256**length_a_B) * s_b
654       v = mul_test(a, b)
655       screen_progress(v)
656       v = mul_test(b, a)
657       screen_progress(v)
658     end
659     (0..16).each do |i|
660       s_a = rand(2)-1
661       s_b = rand(2)-1
662       b_size = rand(length_b_B+1)
663       a = rand(256**length_a_B) * s_a
664       b = rand(256**b_size) * s_b
665       v = mul_test(a, b)
666       screen_progress(v)      
667       v = mul_test(b, a)
668       screen_progress(v)
669     end
670     length_a_B += 1
671     length_b_B += 1
672   end while length_a_B<4096/8
673 end
674
675 ################################################################################
676 # run_test_square                                                              #
677 ################################################################################
678
679 def run_test_square(skip=0)
680   length_a_B = skip+1
681   begin
682     $size = length_a_B
683     (0..16).each do |i|
684       a = rand(256**length_a_B)
685       v = square_test(a)
686       screen_progress(v)
687     end
688     length_a_B += 1
689   end while length_a_B<4096/8
690 end
691
692 ################################################################################
693 # run_test_reduce                                                              #
694 ################################################################################
695
696 def run_test_reduce(skip=0)
697   length_a_B = skip+1
698   length_b_B = skip+1
699   begin
700     $size = length_a_B
701     (0..16).each do |i|
702       a = rand(256**length_a_B)
703       b = rand(256**length_a_B)+1
704       v = reduce_test(a, b)
705       screen_progress(v)
706       end
707     (0..16).each do |i|
708       b_size = rand(length_b_B+1)
709       a = rand(256**length_a_B)
710       b = rand(256**b_size)+1 
711       v = reduce_test(a, b)
712       screen_progress(v)      
713       end
714     length_a_B += 1
715     length_b_B += 1
716   end while length_a_B<4096/8
717 end
718
719 ################################################################################
720 # run_test_expmod                                                              #
721 ################################################################################
722
723 def run_test_expmod(skip=0)
724   length_a_B = skip+1
725   length_b_B = skip+1
726   length_c_B = skip+1
727   begin
728     $size = length_a_B
729     (0..16).each do |i|
730       a = rand(256**length_a_B)
731       b = rand(256**length_b_B)+1
732       c = rand(256**length_c_B)+1
733       v = expmod_test(a, b, c)
734       screen_progress(v)
735       end
736     (0..16).each do |i|
737       b_size = rand(length_b_B+1)
738       a = rand(256**length_a_B)
739       b = rand(256**b_size)+1 
740       c = rand(256**b_size)+1
741       v = expmod_test(a, b, c)
742       screen_progress(v)      
743       end
744     length_a_B += 1
745     length_b_B += 1
746   end while length_a_B<4096/8
747 end
748
749 ################################################################################
750 # run_test_gcdext                                                              #
751 ################################################################################
752
753 def run_test_gcdext(skip=0)
754   length_a_B = skip+1
755   length_b_B = skip+1
756   begin
757     $size = length_a_B
758     (0..16).each do |i|
759       a = rand(256**length_a_B)
760       b = rand(256**length_a_B)+1
761       v = gcdext_test(a, b)
762       $logfile.flush()
763       screen_progress(v)
764       end
765     (0..16).each do |i|
766       b_size = rand(length_b_B+1)
767       a = rand(256**length_a_B)
768       b = rand(256**b_size)+1 
769       v = gcdext_test(a, b)
770       $logfile.flush()
771       screen_progress(v)      
772       end
773     length_a_B += 1
774     length_b_B += 1
775   end while length_a_B<4096/8
776 end
777
778 def init_serialport(conf)
779   puts("serial port interface version: " + SerialPort::VERSION);
780   $linewidth = 64
781   $linepos = 0
782   $testno = 0
783   params = { "baud"       => conf["PORT"]["baud"].to_i,
784               "data_bits" => conf["PORT"]["databits"].to_i,
785               "stop_bits" => conf["PORT"]["stopbits"].to_i,
786               "parity"    => SerialPort::NONE }
787   params["paraty"] = SerialPort::ODD   if conf["PORT"]["paraty"].downcase == "odd"
788   params["paraty"] = SerialPort::EVEN  if conf["PORT"]["paraty"].downcase == "even"
789   params["paraty"] = SerialPort::MARK  if conf["PORT"]["paraty"].downcase == "mark"
790   params["paraty"] = SerialPort::SPACE if conf["PORT"]["paraty"].downcase == "space"
791   
792   puts("\nPort: "+conf["PORT"]["port"]+"@"    +
793                   params["baud"].to_s      +
794                   " "                      +
795                   params["data_bits"].to_s +
796                   conf["PORT"]["paraty"][0,1].upcase +
797                   params["stop_bits"].to_s +
798                   "\n")
799   
800   $sp = SerialPort.new(conf["PORT"]["port"], params)
801   
802   $sp.read_timeout=1000; # 5 minutes
803   $sp.flow_control = SerialPort::SOFT
804   
805   reset_system()
806 end
807
808 ################################################################################
809 # MAIN                                                                         #
810 ################################################################################
811
812 opts = Getopt::Std.getopts("s:f:i:a:hd")
813
814 conf = Hash.new
815 conf = readconfigfile("/etc/testport.conf", conf)
816 conf = readconfigfile("~/.testport.conf", conf)
817 conf = readconfigfile("testport.conf", conf)
818 conf = readconfigfile(opts["f"], conf) if opts["f"]
819
820 #puts conf.inspect
821 init_serialport(conf)
822
823   $max_timeout = 5 * 60
824
825 if opts['d']
826   $debug = true
827 end
828
829 logfilename = conf['PORT']['testlogbase']+'bigint.txt'
830 if File.exists?(logfilename)
831   i=1
832   begin
833     logfilename = sprintf('%s%04d%s', conf['PORT']['testlogbase']+'bigint_',i,'.txt')
834     i+=1
835   end while(File.exists?(logfilename))
836   while(i>2) do
837     n1 = sprintf('%s%04d%s', conf['PORT']['testlogbase']+'bigint_',i-2,'.txt')
838     n2 = sprintf('%s%04d%s', conf['PORT']['testlogbase']+'bigint_',i-1,'.txt')
839     File.rename(n1, n2)
840     printf("%s -> %s\n", n1, n2) 
841     i-=1
842   end
843   n1 = sprintf('%s%s', conf['PORT']['testlogbase'],'bigint.txt')
844   n2 = sprintf('%s%04d%s', conf['PORT']['testlogbase']+'bigint_',1,'.txt')
845   File.rename(n1, n2) 
846   printf("%s -> %s\n", n1, n2)  
847   logfilename = conf['PORT']['testlogbase']+'bigint.txt' 
848 end
849 $logfile = File.open(logfilename, 'w')
850 printf("logfile: %s\n", logfilename)
851
852 $logfile.printf("bigint test from: %s\n", Time.now.to_s)
853 $logfile.printf("skip = %s\n", opts['s']) if opts['s']
854 $logfile.printf("seed = 0x%X\n", 0xdeadbeef)
855
856 tests = Hash.new
857 tests['a'] = proc {|x| run_test_add(x) }
858 tests['m'] = proc {|x| run_test_mul(x) }
859 tests['x'] = proc {|x| run_test_add_scale(x) }
860 tests['s'] = proc {|x| run_test_square(x) }
861 tests['r'] = proc {|x| run_test_reduce(x) }
862 tests['e'] = proc {|x| run_test_expmod(x) }
863 tests['g'] = proc {|x| run_test_gcdext(x) }
864 init_str = Hash.new
865 init_str['a'] = 'add-test'
866 init_str['x'] = 'add-scale-test'
867 init_str['m'] = 'mul-test'
868 init_str['s'] = 'square-test'
869 init_str['r'] = 'reduce-test'
870 init_str['e'] = 'expmod-test'
871 init_str['g'] = 'gcdext-test'
872
873 srand(0xdeadbeef)
874
875 if opts['a']
876   opts['a'].each_char do |x|
877     if tests[x]
878       puts init_str[x]
879       init_system(init_str[x])
880       tests[x].call(opts['s']?opts['s'].to_i():0) 
881     else
882       puts "no test defiened for '#{x}'"
883     end  
884   end
885 else
886   'amsre'.each_char do |x|
887     if tests[x]
888       puts init_str[x]
889       init_system(init_str[x])
890       tests[x].call(opts['s']?opts['s'].to_i():0) 
891     else
892       puts "no test defiened for '#{x}'"
893     end  
894   end
895 end
896
897
898 $logile.close()