gw-tests/lib.sh

   1 set -e
   2
   3 while [ $# -gt 0 ]; do
   4     case "$1" in
   5         -P) OPT_PLOT_DISABLE=1; shift;;
   6         -p) OPT_PLOT_ONLY=1; shift;;
   7         -X|--no-x11-plot) OPT_NO_X11=1; shift;;
   8     esac
   9 done
  10
  11 PATH=$PWD/../_compiled/bin/:$PATH
  12
  13 COUNT=10000
  14
  15 error() {
  16     echo $1 >&2
  17     exit 1
  18 }
  19
  20 sshgw() {
  21     local socket="$HOME/.ssh/cangw-connection"
  22
  23     if [[ ! -S $socket ]]; then
  24         # Create master connection to speed up subsequenct command.
  25         # The ssh is put into background and the connection is closed
  26         # after 10 minutes)
  27         ssh -f -M -S $socket root@192.168.2.3 sleep 600 > /dev/null 2>&1
  28     fi
  29     ssh -x -a -S $socket root@192.168.2.3 "$@"
  30 }
  31
  32 cleanupgw() {
  33     # Set can interfaces up
  34     sshgw 'for i in 0 1; do ip link show dev can$i|grep -q UP || ip link set can$i up type can bitrate 1000000; done'
  35     sshgw 'cangw -F'
  36 }
  37
  38 _plot() {
  39     local testname=`basename $0 .sh`
  40
  41     cmd=$(plot_cmds)                    # Get plot commands
  42     cmd=$(echo "$cmd" | sed -e "/set title/ s/[\"']\(.*\)[\"']/\"\1\\\\n($kvers)\"/")
  43     if [ -z "$OPT_NO_X11" ]; then
  44         echo "set terminal x11 enhanced; ${cmd}" | gnuplot -persist
  45     fi
  46     I=''
  47     echo 'set terminal postscript color eps enhanced size 6cm,4cm lw 1 "Times-Roman" 10;' \
  48         'set lmargin 8;' \
  49         "${cmd}" | gnuplot > ${testname}$I.eps
  50     echo "set terminal postscript color eps enhanced;" \
  51         "${cmd}" | gnuplot | epstopdf --filter > ${testname}$I.pdf
  52     mkdir -p thumb
  53     convert -density 30  -gamma 0.5 -quality 90 -type Palette -depth 8 ${testname}$I.pdf thumb/${testname}$I.png
  54     convert -density 150 -gamma 0.7 -quality 90 -type Palette -depth 8 ${testname}$I.pdf ${testname}$I.png
  55
  56 }
  57
  58 create_dirs_and_links() {
  59     local test=$1
  60     local kver=$2
  61
  62     local  d=results/by-kern/$kver/$test
  63     mkdir -p $d
  64     mkdir -p results/by-test/$test
  65     ln -sfT ../../${d#results/} results/by-test/$test/$kver
  66     echo $d
  67 }
  68
  69 _run() {
  70     if [[ ! "$OPT_PLOT_ONLY" ]]; then
  71         kernel_versions=$(sshgw uname -r)
  72     else
  73         kernel_versions=$(ls results/by-kern)
  74     fi
  75     for kvers in $kernel_versions; do
  76         dir=$(create_dirs_and_links $(basename $0 .sh) $kvers)
  77         script=$PWD/$0
  78         cd $dir
  79         echo "Working directory: $dir"
  80         if [[ ! "$OPT_PLOT_ONLY" ]]; then
  81             rm -rf *
  82             cleanupgw
  83             main
  84             cp $script .
  85         fi
  86         _plot
  87     done
  88 }
  89
  90 test_end() {
  91     test_end_called=t
  92     exit_ok=
  93     _run
  94     exit_ok=t
  95 }
  96
  97 _myexit() {
  98     code=$?
  99     cmd=$BASH_COMMAND
 100     if [[ ! "$test_end_called" ]]; then
 101         test_end_called=t
 102         error "bug in the test script: No test_end called"
 103     fi
 104     if ! test -n "$exit_ok"; then
 105         error "FATAL: Command '$cmd' exited with code $code"
 106     fi
 107 }
 108
 109 trap '_myexit' EXIT