Make fire_requests.py multithreaded.

This is much more evil

Author: mhaas

contrib/load-tester/fire_requests.py

@@ -10,6 +10,12 @@
 import sys
 import random
 
+import argparse
+import functools
+
+
+from multiprocessing import Pool
+
 from httplib import HTTPConnection
 
 PORT = "2060"
@@ -18,6 +24,9 @@
 def main(targetIF, prefix, maxI):
     target = get_ip_address(targetIF)
     for i in xrange(int(maxI)):
+        main_single(target, prefix, i)
+
+def main_single(target, prefix, i):
         source = get_ip_address(prefix + str(i))
         # source_address requires python 2.7
         # urllib2 does not nicely expose source_address, so use
@@ -64,5 +73,22 @@ def get_ip_address(ifname):
         )[20:24])
 
 if __name__ == "__main__":
+
+    parser = argparse.ArgumentParser(description='Hammer a wifidog instance with requests')
+    parser.add_argument('--target-interface', required=True,
+                   help='Interface where Wifidog is listening')
+    parser.add_argument('--source-interface-prefix', required=True,
+                   help='Prefix of the virtual interfaces from which Wifidog is exercised.')
+    parser.add_argument('--source-interface-count', required=True,
+                   help='Number of virtual interfaces, where interface is prefix+index')
+    parser.add_argument('--process-count', required=True,
+                   help='How many processes to run')
+
+    args = parser.parse_args()
+
+    target = get_ip_address(args.target_interface)
+    p = Pool(int(args.process_count))
+    partial = functools.partial(main_single, target, args.source_interface_prefix)
     while True:
-        main(sys.argv[1], sys.argv[2], sys.argv[3])
+        p.map(partial, list(xrange(int(args.source_interface_count))))
+

contrib/load-tester/run.sh

@@ -4,8 +4,8 @@
 # echo "core.%e.%p" > /proc/sys/kernel/core_pattern
 # http://stackoverflow.com/a/18368068
 
+echo "core.%e.%p" > /proc/sys/kernel/core_pattern
 ulimit -c unlimited
-ulimit -a
 
 COUNT=40
 echo "Make sure to configure GatewayInterface in wifidog_mock.conf"
@@ -27,7 +27,18 @@ IF=`grep GatewayInterface wifidog-mock.conf | cut -f 2 -d ' '`
 echo "Waiting for wifidog to come up"
 
 sleep 10
-./fire_requests.py $IF mac $COUNT
+
+usage: fire_requests.py [-h] --target-interface TARGET_INTERFACE
+                        --source-interface-prefix SOURCE_INTERFACE_PREFIX
+                        --source-interface-count SOURCE_INTERFACE_COUNT
+                        --process-count PROCESS_COUNT
+
+
+./fire_requests.py \
+    --target-interface $IF \
+    --source-interface-prefix mac \
+    --source-interface-count $COUNT \
+    --process-count 3
 
 #./generate_interfaces.sh stop