Add load tester script

Author: mhaas

contrib/load-tester/fire_requests.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python2
+# -*- coding: utf -*-
+
+
+import socket
+import fcntl
+import struct
+
+import uuid
+import sys
+
+from httplib import HTTPConnection
+
+PORT = "2060"
+
+def main(targetIF, prefix, maxI):
+    target = get_ip_address(targetIF)
+    for i in xrange(int(maxI)):
+        source = get_ip_address(prefix + str(i))
+        # source_address requires python 2.7
+        # urllib2 does not nicely expose source_address, so use
+        # lower-level API
+        conn = HTTPConnection(target, PORT, timeout=10, source_address=(source, 0))
+        conn.connect()
+        conn.request("GET", "/")
+        try:
+            resp = conn.getresponse()
+            resp.read()
+        except:
+            pass
+        conn = HTTPConnection(target, PORT, timeout=10, source_address=(source, 0))
+        conn.connect()
+        conn.request("GET", "/wifidog/auth?token=" + str(uuid.uuid4()))
+        try:
+            resp = conn.getresponse()
+            # this causes wifidog to ask our mock auth server if the token is
+            # correct
+            resp.read()
+        except:
+            pass
+
+
+# http://code.activestate.com/recipes/439094-get-the-ip-address-associated-with-a-network-inter/
+def get_ip_address(ifname):
+    print "ifname: %s" % ifname
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    return socket.inet_ntoa(fcntl.ioctl(
+        s.fileno(),
+        0x8915,  # SIOCGIFADDR
+        struct.pack('256s', ifname[:15])
+        )[20:24])
+
+if __name__ == "__main__":
+    while True:
+        main(sys.argv[1], sys.argv[2], sys.argv[3])

contrib/load-tester/generate_interfaces.sh

@@ -0,0 +1,56 @@
+#/bin/bash
+
+IF="eth0"
+# for whatever reason, if you use eth0.x as your virtual interface name,
+# dhclient will lose the carrier
+PREFIX="mac"
+# don't touch resolv.conf
+DHCP="dhcpcd --waitip -C resolv.conf"
+#DHCP="dhclient -v"
+
+NET="10.0.10."
+
+
+function start() {
+    echo "IP address       HW type     Flags       HW address            Mask     Device" > /tmp/arp
+    # Add internal address for GW
+    sudo ip link add internal0 link $IF type macvlan mode bridge || exit 1
+    sudo ip addr add ${NET}254 dev internal0
+    sudo ip link set internal0 up || exit 2
+    for i in `seq 0 9`; do
+        echo "Add link $i"
+        # sudo ip link add virtual0 link eth0 type macvlan mode bridge
+        sudo ip link add ${PREFIX}${i} link $IF type macvlan mode bridge || exit 1
+        echo "Assigning temporary IP address"
+        # use link-local address. Ideally, we would check if the
+        # address is already aissgned
+        sudo ip addr add ${NET}$(($i + 1)) dev ${PREFIX}${i}
+        MAC=`ip link show ${PREFIX}${i} | grep ether | sed -e 's,.*link/ether ,,' -e 's, brd.*,,'`
+        echo "Marking link $i up"
+        sudo ip link set ${PREFIX}${i} up || exit 2
+        #echo "Acquiring IP for link $i"
+        # when using DHCP, the interface would immediately
+        # go down?
+        #sudo $DHCP ${PREFIX}${i} || exit 3
+        echo " ${NET}$(($i + 1))     0x1         0x2         $MAC     *        ${PREFIX}${i}" >> /tmp/arp
+    done
+}
+
+function stop() {
+    sudo ip link del internal0 2>/dev/null || true
+    for i in `seq 0 9`; do
+        echo "Deleting link $i"
+        sudo ip link del ${PREFIX}${i} 2>/dev/null || true
+    done
+}
+
+
+
+if [[ x$1 == x"start" ]]; then
+    stop
+    start
+elif [[ x$1 == x"stop" ]]; then
+    stop
+else
+    echo "Unknown command"
+fi

contrib/load-tester/mock_auth.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+# -*- coding: utf -*-
+
+from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
+
+import random
+
+def main():
+    server = HTTPServer(('', 8080), AuthHandler)
+    server.serve_forever()
+
+
+class AuthHandler(BaseHTTPRequestHandler):
+    def do_GET(self):
+        self.send_response(200)
+        self.send_header('Content-type', 'text/html')
+        self.end_headers()
+        if "ping" in self.path:
+            self.wfile.write("Pong\n")
+        else:
+            self.wfile.write("Auth: %s\n" % random.choice([0,1]))
+        return
+
+if __name__ == "__main__":
+    main()

contrib/load-tester/run.sh

@@ -0,0 +1,23 @@
+#!/bin/bash
+
+
+echo "Make sure to configure GatewayInterface in wifidog_mock.conf"
+
+#./generate_interfaces.sh start || exit 1
+
+./mock_auth.py &
+
+# 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 &
+
+IF=`grep GatewayInterface wifidog-mock.conf | cut -f 2 -d ' '`
+
+echo "Waiting for wifidog to come up"
+
+sleep 10
+./fire_requests.py $IF mac 9
+
+#./generate_interfaces.sh stop
+
+

contrib/load-tester/wifidog-mock.conf

@@ -0,0 +1,32 @@
+ExternalInterface eth0
+GatewayInterface internal0
+
+AuthServer {
+    Hostname localhost
+    HTTPPort 8080
+    Path /
+}
+
+ClientTimeout 5
+
+FirewallRuleSet global {
+}
+
+FirewallRuleSet validating-users {
+    FirewallRule allow to 0.0.0.0/0
+}
+
+FirewallRuleSet known-users {
+    FirewallRule allow to 0.0.0.0/0
+}
+
+FirewallRuleSet unknown-users {
+    FirewallRule allow udp port 53
+    FirewallRule allow tcp port 53
+    FirewallRule allow udp port 67
+    FirewallRule allow tcp port 67
+}
+
+FirewallRuleSet locked-users {
+    FirewallRule block to 0.0.0.0/0
+}