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