#!/usr/bin/ruby
# cmacvs_test.rb
=begin
    This file is part of the AVR-Crypto-Lib.
    Copyright (C) 2006-2015 Daniel Otte (bg@nerilex.org)

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
=end

$debug = true
$debug = false
require 'rubygems'
require 'serialport'
require 'getopt/std'

$buffer_size = 0
$conffile_check = Hash.new
$conffile_check.default = 0

################################################################################
# readconfigfile                                                               #
################################################################################

def readconfigfile(fname, conf)
  return conf if $conffile_check[fname]==1
  $conffile_check[fname]=1
  section = "default"
  if not File.exists?(fname)
    return conf
  end
  file = File.open(fname, "r")
  until file.eof
    line = file.gets()
	next if /[\s]*#/.match(line)
	if m=/\[[\s]*([^\s]*)[\s]*\]/.match(line)
	  section=m[1]
	  conf[m[1]] = Hash.new
	  next
	end
	next if not /=/.match(line)
	m=/[\s]*([^\s]*)[\s]*=[\s]*([^\s]*)/.match(line)
	if m[1]=="include"
	  Dir.glob(m[2]){ |fn| conf = readconfigfile(fn, conf) }
	else
  	  conf[section][m[1]] = m[2]
	end
  end
  file.close()
  return conf
end

################################################################################
# reset_system                                                                 #
################################################################################

def reset_system
  $sp.print("exit\r")
  sleep 0.1
  $sp.print("exit\r")
  sleep 0.1
end

################################################################################
# scan_system                                                                  #
################################################################################

def scan_system
  algos = Hash.new
  $sp.print("cmacvs_list\r")
  while true
    line=$sp.gets()
    return algos if /^>$/.match(line)
    if m = /[\*\ ]([a-z]):[\s]*([a-zA-Z0-9+_-]+)/.match(line)
      algos[m[2]]=m[1]
    end
  end
end

################################################################################
# init_system                                                                  #
################################################################################

def init_system(algo_select, algo_id)
  $sp.print("echo off \r")
  print("DBG i: " + "echo off \r"+"\n") if $debug
 sleep 1
  $sp.print("cmacvs_set #{algo_select}\r")
  print("DBG i: " + "cmacvs_set #{$algo_select} \r"+"\n") if $debug
  sleep 1
  $sp.print("cmacvs_test#{algo_id} \r")
  print("DBG i: " + "cmacvs_test#{algo_id} \r"+"\n") if $debug
  begin
    line=$sp.gets()
  end while not m=/buffer_size[\s]*=[\s]*0x([0-9A-Fa-f]*)/.match(line)
  $buffer_size = m[1].to_i(16)
end

################################################################################
# get_md                                                                       #
################################################################################

def get_mac
  begin
    line = $sp.gets()
	line = "" if line==nil
	puts("DBG got: "+line) if $debug
  end while not /[\s]*Mac[\s]*=.*/.match(line)
  return line
end

################################################################################
# get_result                                                                   #
################################################################################

def get_result
  begin
    line = $sp.gets()
    line = "" if line==nil
    puts("DBG got: "+line) if $debug
  end while not /[\s]*Result[\s]*=.*/.match(line)
  puts "DBG i: got result: "+line if $debug
  return line
end

################################################################################
# send_md                                                                      #
################################################################################

def send_test(klen, mlen, tlen, key, msg, mac=nil)
  $sp.printf("Klen = %s\n\r", klen)
  $sp.printf("Mlen = %s\n\r", mlen)
  $sp.printf("Tlen = %s\n\r", tlen)
  $sp.printf("Key = %s\n\r",  key)
  $sp.print("Msg = ")
  for i in 0..msg.length-1
    $sp.print(msg[i].chr)
#   print("DBG s: "+ md_string[i].chr) if $debug
#   sleep(0.001)
    if((i%($buffer_size*2)==0)&&(i!=0))
      begin
      line=$sp.gets()
      end while not /\./.match(line)
    end
  end
  $sp.printf("Mac = %s\n\r",  mac) if mac
end

################################################################################
# get_next_kv_pair                                                             #
################################################################################

def get_next_kv_pair(file)
  loop do
    return nil if file.eof
    lb = file.gets()
    m=lb.match(/[\s]*([\w\d_-]*)[\s]*=[\s]*([\w\d_-]*)/)
    puts "DBG i: found #{m[1]} with value #{m[2]}" if m && $debug
    return [m[1],m[2]] if m
  end
end

################################################################################
# run_test_gen                                                                 #
################################################################################

def run_test_gen(filename, skip=0)
  nerrors = 0
  line=1
  if not File.exist?(filename)
  	puts("ERROR file "+filename+" does not exist!")
  	return nerrors
  end
  pos = 0
  file = File.new(filename, "r");
  until file.eof
    params = Hash.new
    begin
      m = get_next_kv_pair(file)
      return nerrors if m==nil
      params[m[0]] = m[1]
    end until m[0]=='Mac'
    if(skip>0)
  	  skip -= 1
  	  redo
  	end
    puts("DBG sending: ") if $debug
	  send_test(params['Klen'], params['Mlen'], params['Tlen'], params['Key'], params['Msg'])
	  avr_md = get_mac()
    a = params['Mac'];
	  b = (/[\s]*Mac[\s]*=[\s]*([0-9a-fA-F]*).*/.match(avr_md))[1];
  	a.upcase!
  	b.upcase!
	  printf("\n%4d (%4d) [%5d]: ", line, (line-1)*$linewidth, params['Count']) if (pos%$linewidth==0 and $linewidth!=0)
	  line += 1               if (pos%$linewidth==0 and $linewidth!=0)
	  #sleep(1)
	  #putc((a==b)?'*':'!')
  	if(a==b)
	    putc('*')
  	else
	    putc('!')
	 #  printf("<%d>",len)
	    printf("\nError @%05d: %s [should]\n           != %s [is]- ",  params['Count'].to_i , a, b)
	    nerrors += 1
	  end
	  pos += 1
  end
  file.close()
  return nerrors
end

################################################################################
# run_test_ver                                                                 #
################################################################################

def run_test_ver(filename, skip=0)
  nerrors = 0
  line=1
  if not File.exist?(filename)
    puts("ERROR file "+filename+" does not exist!")
    return nerrors
  end
  pos = 0
  file = File.new(filename, "r");
  until file.eof
    params = Hash.new
    begin
      m = get_next_kv_pair(file)
      return nerrors if m==nil
      params[m[0]] = m[1]
    end until m[0]=='Result'
    if(skip>0)
      skip -= 1
      redo
    end
    puts("DBG sending: ") if $debug
    send_test(params['Klen'], params['Mlen'], params['Tlen'], params['Key'], params['Msg'], params['Mac'])
    avr_res = get_result()
    a = params['Result'].match(/[\s]*([PF])/)[1];
    b = /[\s]*Result[\s]*=[\s]*([PF])/.match(avr_res)[1];
    a.upcase!
    b.upcase!
    printf("\n%4d (%4d) [%5d]: ", line, (line-1)*$linewidth, params['Count']) if (pos%$linewidth==0 and $linewidth!=0)
    line += 1               if (pos%$linewidth==0 and $linewidth!=0)
    #sleep(1)
    #putc((a==b)?'*':'!')
    if(a==b)
      putc('*')
    else
      putc('!')
   #  printf("<%d>",len)
      printf("\nError @%05d: %s [should]\n           != %s [is]- ",  params['Count'].to_i , a, b)
      nerrors += 1
    end
    pos += 1
  end
  file.close()
  return nerrors
end

################################################################################
# MAIN                                                                         #
################################################################################

opts = Getopt::Std.getopts("s:f:i:j:hdca")

conf = Hash.new
conf = readconfigfile("/etc/testport.conf", conf)
conf = readconfigfile("~/.testport.conf", conf)
conf = readconfigfile("testport.conf", conf)
conf = readconfigfile(opts["f"], conf) if opts["f"]

#puts conf.inspect

puts("serial port interface version: " + SerialPort::VERSION);
$linewidth = 64
params = { "baud"       => conf["PORT"]["baud"].to_i,
            "data_bits" => conf["PORT"]["databits"].to_i,
            "stop_bits" => conf["PORT"]["stopbits"].to_i,
            "parity"    => SerialPort::NONE }
params["paraty"] = SerialPort::ODD   if conf["PORT"]["paraty"].downcase == "odd"
params["paraty"] = SerialPort::EVEN  if conf["PORT"]["paraty"].downcase == "even"
params["paraty"] = SerialPort::MARK  if conf["PORT"]["paraty"].downcase == "mark"
params["paraty"] = SerialPort::SPACE if conf["PORT"]["paraty"].downcase == "space"

puts("\nPort: "+conf["PORT"]["port"]+"@"    +
                params["baud"].to_s      +
                " "                      +
                params["data_bits"].to_s +
                conf["PORT"]["paraty"][0,1].upcase +
                params["stop_bits"].to_s +
                "\n")

$sp = SerialPort.new(conf["PORT"]["port"], params)

$sp.read_timeout=1000; # 5 minutes
$sp.flow_control = SerialPort::SOFT

reset_system()
algos=scan_system()
#puts algos.inspect

if opts["d"]
  $debug = true
end

if opts["s"]
  algos_rev = algos.invert
  algo_tasks = Array.new
  opts["s"].each_byte{ |x|
    if algos_rev[x.chr]
      algo_tasks << [algos_rev[x.chr],x.chr]
    end
  }
else
  algo_tasks=algos.sort
end

algo_tasks.each do |algoa|
  algo = algoa[0]
  if conf[algo]==nil
    puts("No test-set defined for #{algo} \r\n")
    next
  else
	i=0
	i = opts["j"] if opts["j"]
	logfile=File.open(conf["PORT"]["testlogbase"]+algo+".txt", "a")
	while conf[algo]["file_#{i}"] != nil
	  puts("Testing #{algo} with #{conf[algo]["file_#{i}"]}")
	  reset_system()
	  init_system(algoa[1], (conf[algo]["file_#{i}_test"]=='gen')?'1':'2')
	  skip=0
	  skip=opts["i"].to_i if opts["i"]
    nerrors=run_test_gen(conf[algo]["file_#{i}"], skip) if conf[algo]["file_#{i}_test"]=='gen'
    nerrors=run_test_ver(conf[algo]["file_#{i}"], skip) if conf[algo]["file_#{i}_test"]=='ver'
	    if nerrors == 0
        puts("\n[ok]")
        logfile.puts("[ok] "+conf[algo]["file_#{i}"]+ " ("+Time.now.to_s()+")")
      else
        puts("\n[errors: "+ nerrors.to_s() +"]")
        logfile.puts("[error] "+nerrors.to_s+" "+conf[algo]["file_#{i}"]+ " ("+Time.now.to_s()+")")
      end
      i += 1
    end
    logfile.close()
  end
end