Add memory plot to load-tester; drop privileges

Author: mhaas

contrib/load-tester/memory.gnuplot

@@ -0,0 +1,7 @@
+set terminal png size 700,550 enhanced font "Helvetica,20"
+set output "memory.png"
+set xlabel "Time in seconds"
+set ylabel "Memory usage in kB"
+plot "vmdata.log" using 1:2 with lines title "VmData", \
+     "vmrss.log" using 1:2 with lines title "VmRSS"
+

contrib/load-tester/plot_memory.sh

@@ -0,0 +1,30 @@
+#!/bin/bash
+
+# empty
+> vmdata.log
+> vmrss.log
+
+WD_PID=$1
+
+function getdatum {
+    grep $1 /proc/$WD_PID/status | cut -f 3 -d " "
+}
+
+function plot {
+    gnuplot memory.plot 
+}
+
+SLEEP=5
+COUNT=0
+while true; do
+    vmdata=`getdatum VmData`
+    vmrss=`getdatum VmRSS`
+    echo "$(($COUNT * $SLEEP)) $vmdata" >> vmdata.log
+    echo "$(($COUNT * $SLEEP)) $vmrss" >> vmrss.log
+    COUNT=$(($COUNT + 1))
+    sleep $SLEEP
+done
+
+
+# on exit, do plot
+trap plot EXIT

contrib/load-tester/run.sh

@@ -4,6 +4,12 @@
 # echo "core.%e.%p" > /proc/sys/kernel/core_pattern
 # http://stackoverflow.com/a/18368068
 
+
+if [[ -z "$SUDO_USER" ]]; then
+    echo "Hey, you should run this with sudo"
+    exit 1
+fi
+
 echo "core.%e.%p" > /proc/sys/kernel/core_pattern
 ulimit -c unlimited
 
@@ -12,38 +18,47 @@ echo "Make sure to configure GatewayInterface in wifidog_mock.conf"
 
 ./generate_interfaces.sh start $COUNT || exit 1
 
-./mock_auth.py &
+sudo -u "$SUDO_USER" ./mock_auth.py &
 MA_PID="$!"
 
+# work around libtool stuff - do not execute wrapper!
+#EXEC="../../src/.libs/wifidog"
+#export LD_LIBRARY_PATH="../../libhttpd/.libs/"
 # trace-children is necessary because of the libtool wrapper -.-
-#sudo valgrind --leak-check=full --trace-children=yes --trace-children-skip=/bin/sh \
-#    --log-file=valgrind.log ../../src/wifidog -d 7 -f -c wifidog-mock.conf 2> wifidog.log &
+#valgrind --leak-check=full --trace-children=yes --trace-children-skip=/bin/sh \
+#    --log-file=valgrind.log $EXEC -d 7 -f -c wifidog-mock.conf -a /tmp/arp 2> wifidog.log &
 
 # for -fsanitize=address
 export ASAN_OPTIONS=check_initialization_order=1
 export ASAN_SYMBOLIZER_PATH=/usr/bin/llvm-symbolizer-3.5
 
-../../src/wifidog -d 7 -f -c wifidog-mock.conf -a /tmp/arp 2> wifidog.log &
+
+../../src/wifidog -d 7 -f -c wifidog-mock.conf -a /tmp/arp &> wifidog.log &
 WD_PID="$!"
 
-IF=`grep GatewayInterface wifidog-mock.conf | cut -f 2 -d ' '`
+
+sudo -u "$SUDO_USER" ./plot_memory.sh $WD_PID &
+M_PID="$!"
+
+IF=`sudo -u "$SUDO_USER" grep GatewayInterface wifidog-mock.conf | cut -f 2 -d ' '`
 
 echo "Waiting for wifidog to come up"
 
 sleep 10
 
-./fire_requests.py \
+sudo -u "$SUDO_USER" ./fire_requests.py \
     --target-interface $IF \
     --source-interface-prefix mac \
     --source-interface-count $COUNT \
-    --process-count 2
+    --process-count 3
 
 #./generate_interfaces.sh stop
 
 function cleanup() {
 
     kill $MA_PID
     kill $WD_PID
+    kill $M_PID
     ./generate_interfaces.sh stop $COUNT
 
 }