host/shavs_test.rb

   1 #!/usr/bin/ruby
   2 # shavs_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
  26 def init_system
  27 #  sleep 1
  28   $sp.print("exit\r")
  29   sleep 0.5
  30   $sp.print("exit\r")
  31   sleep 0.5
  32   $sp.print("echo off \r")
  33   print("DBG i: " + "echo off \r"+"\n") if $debug
  34 #  line = $sp.readlines()
  35 #  print("DBG 0.0: ")
  36 #  print(line)
  37   sleep 1
  38   $sp.print("shavs_set #{$algo_select}\r")
  39   print("DBG i: " + "shavs_set #{$algo_select} \r"+"\n") # if $debug
  40 #  line = $sp.readlines()
  41 #  print("DBG 0.1: ")
  42 #  print(line)
  43   sleep 1
  44   $sp.print("shavs_test1 \r")
  45   print("DBG i: " + "shavs_test1 \r"+"\n") if $debug
  46 #  line = $sp.readlines()
  47 #  print("DBG 0.2: ")
  48 #  print(line)
  49 end
  50
  51 def get_md
  52   begin
  53     line = $sp.gets()
  54         line = "" if line==nil
  55         puts("DBG got: "+line) if $debug && line!=""
  56   end while not /[\s]*MD[\s]*=.*/.match(line)
  57   return line
  58 end
  59
  60 def send_md(md_string)
  61   sleep(0.15)
  62   for i in 0..md_string.length-1
  63     $sp.print(md_string[i].chr)
  64 #       print("DBG s: "+ md_string[i].chr) if $debug
  65         if(i%5==4)
  66                 sleep(0.15)
  67         end
  68   end
  69 end
  70
  71 def run_test(filename)
  72   nerrors = 0
  73   line=1
  74   if not File.exist?(filename)
  75         puts("ERROR file "+filename+" does not exist!")
  76   end
  77   pos = 0
  78   file = File.new(filename, "r");
  79   until file.eof
  80     sleep(0.5)
  81     begin
  82       lb=file.gets()
  83     end while not (file.eof or (/[\s]*Len[\s]*=.*/.match(lb)))
  84     puts("DBG sending: "+lb) if $debug
  85         return if file.eof
  86         $sp.print(lb.strip)
  87         $sp.print("\r")
  88     begin
  89           lb=file.gets()
  90     end while not (file.eof or (/[\s]*Msg[\s]*=.*/.match(lb)))
  91     return if file.eof
  92     puts("DBG sending: "+lb) if $debug
  93         send_md(lb.strip)
  94         avr_md = get_md()
  95     begin
  96           lb=file.gets()
  97     end while not /[\s]*MD[\s]*=.*/.match(lb)
  98         a = (/[\s]*MD[\s]*=[\s]*([0-9a-fA-F]*).*/.match(lb))[1];
  99         b = (/[\s]*MD[\s]*=[\s]*([0-9a-fA-F]*).*/.match(avr_md))[1];
 100         a.upcase!
 101         b.upcase!
 102         printf("\n%4d (%4d): ", line, (line-1)*$linewidth) if (pos%$linewidth==0 and $linewidth!=0)
 103         line += 1               if (pos%$linewidth==0 and $linewidth!=0)
 104         sleep(1)
 105         #putc((a==b)?'*':'!')
 106         if(a==b)
 107           putc('*')
 108         else
 109           putc('!')
 110 #         printf("\nshould: %s\ngot:   %s\n",lb,avr_md)
 111           nerrors += 1
 112         end
 113         pos += 1
 114   end
 115   return nerrors.to_i
 116 end
 117
 118 if ARGV.size < 6
 119   STDERR.print <<EOF
 120   Usage: ruby #{$0} port bps nbits stopb algo_select testfile ...
 121 EOF
 122   exit(1)
 123 end
 124
 125 puts("\nPort: "+ARGV[0]+ "@"+ARGV[1]+" "+ARGV[2]+"N"+ARGV[3]+"\n");
 126 puts("serial port interface version: " + SerialPort::VERSION);
 127 $linewidth = 64
 128 $params = { "baud"      => ARGV[1].to_i,
 129             "data_bits" => ARGV[2].to_i,
 130             "stop_bits" => ARGV[3].to_i,
 131             "parity"    => SerialPort::NONE }
 132 $sp = SerialPort.new(ARGV[0], $params)
 133 #$sp = SerialPort.new(ARGV[0], ARGV[1].to_i, ARGV[2].to_i, ARGV[3].to_i, SerialPort::NONE);
 134
 135 $sp.read_timeout=1000; # 5 minutes
 136 $sp.flow_control = SerialPort::SOFT
 137 $algo_select = ARGV[4]
 138 #irb
 139
 140 init_system()
 141
 142 nerrors = 0
 143 for i in (5..(ARGV.size-1))
 144   nerrors = run_test(ARGV[i])
 145   if nerrors == 0
 146     puts("\n[ok]")
 147   else
 148     puts("\n[errors: "+ nerrors.to_s() +"]")
 149   end
 150 end
 151  $sp.print("EXIT\r");
 152
 153 #exit(0);
 154
 155