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 [[ "$NO_GW_NEEDED" ]]; then return; fi
  37
  38     if [[ ! -S $socket ]] || ! ssh -x -a -S $socket root@192.168.2.3 true; then
  39         # Create master connection to speed up subsequenct command.
  40         ssh -N -f -M -S $socket root@192.168.2.3 >/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     rm -rf *.pdf *.png
  52     for i in "" $ADDITIONAL_PLOTS; do
  53         plot_cmds $i | sed -e "/set title/ s/[\"']\(.*\)[\"']/\"\1\\\\n{\/*0.75 (GW kernel $kvers, Traffic $traffic, Load $load)}\"/" > plot.gp
  54         if [[ ! -s plot.gp ]]; then return; fi
  55         if [ -z "$OPT_NO_X11" ]; then
  56             echo "set terminal x11 enhanced; $(< plot.gp)" | gnuplot -persist
  57         fi
  58         echo "set terminal pdfcairo solid color enhanced; " \
  59              "set output \"graph$i.pdf\";" \
  60              "set pointsize 0.5;" \
  61             "$(< plot.gp)" | gnuplot
  62         echo 'set terminal pngcairo color enhanced size 750,525 font ",10" ;' \
  63             "$(< plot.gp)" | gnuplot > graph$i.png
  64         mkdir -p thumb
  65         convert -resize 150x105 -gamma 0.5 -quality 90 -type Palette -depth 8 graph$i.png tgraph$i.png
  66 #     echo 'set terminal pngcairo color enhanced size 150,105 font ",1";' \
  67 #       "$(< plot.gp)" | gnuplot > thumb/${testname}$i.png
  68     done
  69
  70 }
  71
  72 echo_plot() {
  73     plot=$1
  74     [[ "$_plot_separator" ]] && echo ", \\"
  75     _plot_separator=t
  76     echo -n "    " $plot
  77 }
  78
  79 traffic_and_length() {
  80     local opts
  81     case $traffic in
  82         flood) opts='';;
  83         50)        opts="-p $((2*(44+$1*8)))";;
  84         oneatatime)        opts="-o";;
  85         *) error "Unknown traffic specification: $traffic"
  86     esac
  87     echo $opts -l $1
  88 }
  89
  90 start_load() {
  91     case $load in
  92         none) ;;
  93         cpu) sshgw 'hackbench -g 3 -l 10000' & loadpid=$!;;
  94         eth) ping -f -s 60000 -q 192.168.2.3 & loadpid=$!;; # TODO: Generate eth load from another computer
  95         can) latester -q -d can1 -i 0x7ff & loadpid=$!;;
  96         *) error "Unknown load specification: $load"
  97     esac
  98 }
  99
 100 kill_load() {
 101     case $load in
 102         none) ;;
 103         cpu) kill $loadpid; sshgw "killall -q hackbench || :";;
 104         eth) kill $loadpid;;
 105         can) kill $loadpid;;
 106         *) error "Unknown load specification: $load"
 107     esac
 108 }
 109
 110 _measure() {
 111     # Remove data from the last measurement
 112     rm -rf *
 113     # Kill load generators left possibly from the past runs
 114     killall -q ping || :
 115     killall -q latester || :
 116     if [[ ! "$NO_GW_NEEDED" ]]; then
 117         sshgw 'killall -q hackbench || :'
 118         # Set can interfaces up
 119         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'
 120         # Delete all vcan interfaces
 121         sshgw 'for dev in $(ip l|grep -o vcan[^:]\\+); do ip link del dev $dev; done'
 122         # Reset priorities
 123         sshgw 'if pid=`pidof irq/145-can0`; then chrt -p -f 50 $pid > /dev/null; fi'
 124         sshgw 'if pid=`pidof irq/146-can1`; then chrt -p -f 50 $pid > /dev/null; fi'
 125         sshgw 'if pid=`pidof sirq-net-rx/0`; then chrt -p -f 49 $pid > /dev/null; fi'
 126         sshgw 'if pid=`pidof sirq-net-tx/0`; then chrt -p -f 49 $pid > /dev/null; fi'
 127         # Set the length of qdisc queue to avoid ENOBUFS errors
 128         ifconfig can0 txqueuelen 200
 129         ifconfig can1 txqueuelen 200
 130         cleanupgw
 131
 132         type prepare >/dev/null 2>&1 && prepare || :
 133         start_load
 134     fi
 135
 136     main
 137
 138     if [[ ! "$NO_GW_NEEDED" ]]; then
 139         kill_load
 140     fi
 141
 142     # Create a plot script at the end to signalize that the
 143     # measurement was successfully finished
 144     cat > plot.sh <<-EOF
 145         #!/bin/bash
 146         export kvers=$kvers
 147         export hostkvers=$hostkvers
 148         export traffic=$traffic
 149         export load=$load
 150         cd \$(dirname \$0)/$(dirname $script)
 151         exec ./$(basename $script) --plot "\$@"
 152         EOF
 153     chmod +x plot.sh
 154 }
 155
 156
 157 _run() {
 158     if [[ $OPT_TRAFFIC = all ]]
 159     then traffics="flood 50 oneatatime"
 160     else traffics=${traffic:-$OPT_TRAFFIC}
 161     fi
 162     if [[ $OPT_LOAD = all ]]
 163     then loads="none cpu eth" # TODO: can
 164     else loads=${load:-$OPT_LOAD}
 165     fi
 166     hostkvers=${hostkvers:-host-$(uname -r)}
 167     kvers=${kvers:-$(sshgw uname -r)}
 168     test=$(basename $0 .sh)
 169     for load in $loads; do
 170     for traffic in $traffics; do
 171         dir="results/$hostkvers/$kvers/$traffic/$load/$test"
 172         mkdir -p $dir
 173         script=$(echo $dir | sed -e 's/[^/]*/../g')/${test}.sh
 174         cd $dir
 175         if [[ ! "$OPT_PLOT_ONLY" ]]; then
 176             echo "Working directory: $dir"
 177             _measure
 178         fi
 179         if [[ ! "$OPT_PLOT_DISABLE" ]]; then
 180             _plot
 181         fi
 182         cd - > /dev/null
 183     done
 184     done
 185 }
 186
 187 test_end() {
 188     test_end_called=t
 189     exit_ok=
 190     _run
 191     exit_ok=t
 192 }
 193
 194 _myexit() {
 195     code=$?
 196     cmd=$BASH_COMMAND
 197     if [[ ! "$test_end_called" ]]; then
 198         test_end_called=t
 199         error "bug in the test script: No test_end called"
 200     fi
 201     if ! test -n "$exit_ok"; then
 202         error "FATAL: Command '$cmd' exit with code $code"
 203     fi
 204 }
 205
 206 trap '_myexit' EXIT