X-Git-Url: https://git.cryptolib.org/?p=avr-crypto-lib.git;a=blobdiff_plain;f=host%2Fget_test.rb;fp=host%2Fget_test.rb;h=987a5b1f7ea1a8bb6773afd8c720eec4baa2d5a2;hp=0000000000000000000000000000000000000000;hb=0076b72ccd398bf7571c5144ba36eda8a24f6c70;hpb=cca97d0e0c937aea8b7e0ae5b931da5f80d9ff52

diff --git a/host/get_test.rb b/host/get_test.rb
new file mode 100644
index 0000000..987a5b1
--- /dev/null
+++ b/host/get_test.rb
@@ -0,0 +1,111 @@
+#!/usr/bin/ruby 
+
+require 'serialport'
+
+if ARGV.size < 5
+  STDERR.print <<EOF
+  Usage: ruby #{$0} port bps nbits stopb command [target_dir] [additional specifier]
+EOF
+  exit(1)
+end
+
+command=ARGV[4]+" ";
+$dir=(ARGV.size>=6)?ARGV[5]:"";
+param=(ARGV.size>=7)?ARGV[6]:"";
+
+puts("\nPort: "+ARGV[0]+ "@"+ARGV[1]+" "+ARGV[2]+"N"+ARGV[3]+"\n");
+$linewidth = 16
+$sp = SerialPort.new(ARGV[0], ARGV[1].to_i, ARGV[2].to_i, ARGV[3].to_i, SerialPort::NONE);
+$sp.read_timeout=1*60*1000; # 5 minutes
+$extended_wait=10;
+$sp.write(command);
+
+def readTestVector(param)
+  fname=$dir;
+  lb="";
+  buffer="";
+  set=0;
+  vector=0;
+  begin
+    ctr=$extended_wait;
+    while((lb=$sp.gets())==nil && ctr>=0)do
+      ctr -= 1;
+    end
+    if (m=/unknown command/.match(lb) || m=/[Ee][Rr]{2}[Oo][Rr]/.match(lb))
+      puts("ERROR: "+lb);
+      exit(2);
+    end
+    if(lb==nil)
+      return false;
+    end
+  end while(m=/\*+/.match(lb));
+  
+  buffer += lb;
+  begin
+    ctr=$extended_wait;
+    while((lb=$sp.gets())==nil && ctr>=0)do
+      ctr -= 1;
+    end
+    if(lb==nil)
+      return false;
+    end
+    buffer+=lb;
+  end while(m=/\*.*/.match(lb));
+
+  while(!(m=/Test vectors/.match(lb))) 
+    m=/[^:]*:[\s]([A-Za-z0-9_-]*)/.match(lb);
+    if(m) 
+      fname+=m[1]+".";
+    end
+    buffer+=lb;
+    ctr=$extended_wait;
+    while((lb=$sp.gets())==nil && ctr>=0)do
+      ctr -= 1;
+    end
+  end
+  if(param!="")
+    fname+=param+".";
+  end
+  puts("-> "+fname+"txt");
+  file=File.new(fname+"txt", "w");
+    buffer+=lb;
+    file.write(buffer);
+    begin
+      if (m=/Test\ vectors\ \-\-\ set[\s]+([0-9]+)/.match(lb))
+	set=m[1].to_i;
+	print("\nSet "+m[1]+":");
+      end
+      if (m=/Set [0-9]*, vector#[\s]*([0-9]+):/.match(lb))
+        vector=m[1].to_i;
+	#print(" "+m[1]);
+	if(vector!=0 && vector % $linewidth==0)
+	  print("\n      ")
+	end
+        printf(" %4u", vector);
+      end
+      ctr=$extended_wait;
+      while((lb=$sp.gets())==nil && ctr>=0)do
+        ctr -= 1;
+      end
+      if(lb==nil)
+        file.close();
+        return false;
+      end
+      file.write(lb);
+    end while(!m=/End of test vectors/.match(lb));
+    puts("\n");
+  file.close();
+  return true
+end
+
+if(readTestVector(param)==false)
+  puts("ERROR: test seems not to be implemented");
+  exit(3);
+end
+
+while(readTestVector(param))
+end
+
+exit(0);
+
+