gw-tests/lib.sh

   1 set -e
   2
   3 COUNT=10000
   4
   5 OPT_TRAFFIC=oneatatime
   6 OPT_LOAD=none
   7
   8 error() {
   9     echo $1 >&2
  10     exit 1
  11 }
  12
  13 while [ $# -gt 0 ]; do
  14     case "$1" in
  15         -P|--no-plot) OPT_PLOT_DISABLE=1; shift;;
  16         -p|--plot) OPT_PLOT_ONLY=1; shift;;
  17         -X|--no-x11-plot) OPT_NO_X11=1; shift;;
  18         -t) case "$2" in
  19                 all|flood|50|oneatatime) OPT_TRAFFIC=$2;;
  20                 *) error "Unknown traffic specification: $2";;
  21             esac;
  22             shift 2;;
  23         -l) case "$2" in
  24                 all|none|cpu|eth|can) OPT_LOAD=$2;;
  25                 *) error "Unknown load specification: $2";;
  26             esac;
  27             shift 2;;
  28     esac
  29 done
  30
  31 PATH=$PWD/../_compiled/bin/:$PATH
  32
  33 sshgw() {
  34     local socket="$HOME/.ssh/cangw-connection"
  35
  36     if [[ ! -S $socket ]] || ! ssh -x -a -S $socket root@192.168.2.3 true; then
  37         # Create master connection to speed up subsequenct command.
  38         # The ssh is put into background and the connection is closed
  39         # after 10 minutes)
  40         ssh -f -M -S $socket root@192.168.2.3 sleep 600 > /dev/null 2>&1
  41     fi
  42     ssh -x -a -S $socket root@192.168.2.3 "$@"
  43 }
  44
  45 cleanupgw() {
  46     sshgw 'cangw -F'
  47 }
  48
  49 _plot() {
  50     local testname=`basename $0 .sh`
  51
  52     plot_cmds | sed -e "/set title/ s/[\"']\(.*\)[\"']/\"\1\\\\n(GW kernel $kvers, traffic $traffic)\"/" > plot.gp
  53     if [[ ! -s plot.gp ]]; then return; fi
  54     if [ -z "$OPT_NO_X11" ]; then
  55         echo "set terminal x11 enhanced; $(< plot.gp)" | gnuplot -persist
  56     fi
  57     I=''
  58     echo 'set terminal postscript color eps enhanced size 6cm,4cm lw 1 "Times-Roman" 10;' \
  59         'set lmargin 8;' \
  60         "$(< plot.gp)" | gnuplot > ${testname}$I.eps
  61     echo "set terminal postscript color eps enhanced;" \
  62         "$(< plot.gp)" | gnuplot | epstopdf --filter > ${testname}$I.pdf
  63     mkdir -p thumb
  64     convert -density 30  -gamma 0.5 -quality 90 -type Palette -depth 8 ${testname}$I.pdf thumb/${testname}$I.png
  65     convert -density 150 -gamma 0.7 -quality 90 -type Palette -depth 8 ${testname}$I.pdf ${testname}$I.png
  66
  67 }
  68
  69 echo_plot() {
  70     plot=$1
  71     [[ "$_plot_separator" ]] && echo ", \\"
  72     _plot_separator=t
  73     echo -n "    " $plot
  74 }
  75
  76 traffic_and_length() {
  77     local opts
  78     case $traffic in
  79         flood) opts='';;
  80         50)        opts="-p $((2*(44+$1*8)))";;
  81         oneatatime)        opts="-o";;
  82         *) error "Unknown traffic specification: $traffic"
  83     esac
  84     echo $opts -l $1
  85 }
  86
  87 start_load() {
  88     case $load in
  89         none) ;;
  90         cpu) sshgw 'hackbench -g 3 -l 10000' & loadpid=$!;;
  91         eth) ping -f -s 60000 -q 192.168.2.3 & loadpid=$!;; # TODO: Generate eth load from another computer
  92         can) latester -q -d can1 -i 0x7ff & loadpid=$!;;
  93         *) error "Unknown load specification: $load"
  94     esac
  95 }
  96
  97 kill_load() {
  98     case $load in
  99         none) ;;
 100         cpu) kill $loadpid; sshgw "killall -q hackbench || :";;
 101         eth) kill $loadpid;;
 102         can) kill $loadpid;;
 103         *) error "Unknown load specification: $load"
 104     esac
 105 }
 106
 107 _measure() {
 108     # Remove data from the last measurement
 109     rm -rf *
 110     touch .results
 111     cat > plot.sh <<-EOF
 112         #!/bin/bash
 113         export kvers=$kvers
 114         export hostkvers=$hostkvers
 115         export traffic=$traffic
 116         export load=$load
 117         cd \$(dirname \$0)/$(dirname $script)
 118         exec ./$(basename $script) --plot "\$@"
 119         EOF
 120     chmod +x plot.sh
 121     # Kill load generators left possibly from the past runs
 122     killall -q ping || :
 123     killall -q latester || :
 124     sshgw 'killall -q hackbench || :'
 125     # Set can interfaces up
 126     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'
 127     # Delete all vcan interfaces
 128     sshgw 'for dev in $(ip l|grep -o vcan[^:]\\+); do ip link del dev $dev; done'
 129     # Reset priorities
 130     sshgw 'if pid=`pidof irq/145-can0`; then chrt -p -f 50 $pid > /dev/null; fi'
 131     sshgw 'if pid=`pidof irq/146-can1`; then chrt -p -f 50 $pid > /dev/null; fi'
 132     sshgw 'if pid=`pidof sirq-net-rx/0`; then chrt -p -f 49 $pid > /dev/null; fi'
 133     sshgw 'if pid=`pidof sirq-net-tx/0`; then chrt -p -f 49 $pid > /dev/null; fi'
 134     # Set the length of qdisc queue to avoid ENOBUFS errors
 135     ifconfig can0 txqueuelen 200
 136     ifconfig can1 txqueuelen 200
 137     cleanupgw
 138
 139     type prepare >/dev/null 2>&1 && prepare || :
 140     start_load
 141     main
 142     kill_load
 143 }
 144
 145
 146 _run() {
 147     if [[ $OPT_TRAFFIC = all ]]
 148     then traffics="flood 50 oneatatime"
 149     else traffics=${traffic:-$OPT_TRAFFIC}
 150     fi
 151     if [[ $OPT_LOAD = all ]]
 152     then loads="none cpu eth" # TODO: can
 153     else loads=${load:-$OPT_LOAD}
 154     fi
 155     hostkvers=${hostkvers:-host-$(uname -r)}
 156     kvers=${kvers:-$(sshgw uname -r)}
 157     test=$(basename $0 .sh)
 158     for load in $loads; do
 159     for traffic in $traffics; do
 160         dir="results/$hostkvers/$kvers/$traffic/$load/$test"
 161         mkdir -p $dir
 162         script=$(echo $dir | sed -e 's/[^/]*/../g')/${test}.sh
 163         cd $dir
 164         if [[ ! "$OPT_PLOT_ONLY" ]]; then
 165             echo "Working directory: $dir"
 166             _measure
 167         fi
 168         if [[ ! "$OPT_PLOT_DISABLE" ]]; then
 169             _plot
 170         fi
 171         cd - > /dev/null
 172     done
 173     done
 174 }
 175
 176 test_end() {
 177     test_end_called=t
 178     exit_ok=
 179     _run
 180     exit_ok=t
 181 }
 182
 183 _myexit() {
 184     code=$?
 185     cmd=$BASH_COMMAND
 186     if [[ ! "$test_end_called" ]]; then
 187         test_end_called=t
 188         error "bug in the test script: No test_end called"
 189     fi
 190     if ! test -n "$exit_ok"; then
 191         error "FATAL: Command '$cmd' exit with code $code"
 192     fi
 193 }
 194
 195 trap '_myexit' EXIT