host/get_test.rb

   1 #!/usr/bin/ruby
   2 # get_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 require 'serialport'
  22
  23 def read_line(error_msg=true)
  24   s = $sp.gets()
  25   if s==nil
  26     puts("ERROR: read timeout!\n") if error_msg
  27         return nil
  28   end
  29   s.gsub(/\006/, '');
  30 end
  31
  32 def readTestVector(param)
  33   fname=$dir;
  34   lb="";
  35   buffer="";
  36   set=0;
  37   vector=0;
  38   begin
  39     lb = read_line(false)
  40         if (m=/unknown command/.match(lb) || m=/[Ee][Rr]{2}[Oo][Rr]/.match(lb))
  41       puts("ERROR: "+lb);
  42       exit(2);
  43     end
  44     if(lb==nil)
  45       return false;
  46     end
  47   end while(m=/\*+/.match(lb));
  48
  49   buffer += lb;
  50   begin
  51     lb = read_line()
  52     if(lb==nil)
  53       return false;
  54     end
  55     buffer+=lb;
  56   end while(m=/\*.*/.match(lb));
  57
  58   while(!(m=/Test vectors/.match(lb)))
  59     m=/[^:]*:[\s]([A-Za-z0-9_-]*)/.match(lb);
  60     if(m)
  61       fname+=m[1]+".";
  62     end
  63     return false if lb==nil
  64     buffer+=lb;
  65     lb = read_line();
  66   end
  67   if(param!="")
  68     fname+=param+".";
  69   end
  70   puts("-> "+fname+"txt");
  71   file=File.new(fname+"txt", "w");
  72     buffer+=lb;
  73     file.write(buffer);
  74     begin
  75       if (m=/Test\ vectors\ \-\-\ set[\s]+([0-9]+)/.match(lb))
  76         set=m[1].to_i;
  77         print("\nSet "+m[1]+":");
  78       end
  79       if (m=/Set [0-9]*, vector#[\s]*([0-9]+):/.match(lb))
  80         vector=m[1].to_i;
  81         #print(" "+m[1]);
  82         if(vector!=0 && vector % $linewidth==0)
  83           print("\n      ")
  84         end
  85         printf(" %4u", vector);
  86       end
  87       lb=read_line();
  88       if(lb==nil)
  89         file.close();
  90         return false;
  91       end
  92       file.write(lb);
  93     end while(!m=/End of test vectors/.match(lb));
  94     puts("\n");
  95   file.close();
  96   return true
  97 end
  98
  99
 100 if ARGV.size < 5
 101   STDERR.print <<EOF
 102   Usage: ruby #{$0} port bps nbits stopb command [target_dir] [additional specifier]
 103 EOF
 104   exit(1)
 105 end
 106
 107 command=ARGV[4]+"\r";
 108 $dir=(ARGV.size>=6)?ARGV[5]:"";
 109 param=(ARGV.size>=7)?ARGV[6]:"";
 110
 111 puts("\nPort: "+ARGV[0]+ "@"+ARGV[1]+" "+ARGV[2]+"N"+ARGV[3]+"\n");
 112 $linewidth = 16
 113 $sp = SerialPort.new(ARGV[0], ARGV[1].to_i, ARGV[2].to_i, ARGV[3].to_i, SerialPort::NONE);
 114 $sp.read_timeout=1000; # 1 secound
 115 $extended_wait=100;
 116 $sp.write(command);
 117
 118 if(readTestVector(param)==false)
 119   puts("ERROR: test seems not to be implemented");
 120   exit(3);
 121 end
 122
 123 while(readTestVector(param))
 124 end
 125
 126 exit(0);
 127
 128