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()