[or-cvs] r17937: {} Add python version of speedracer from Karsten. (torflow/trunk/NetworkScanners)

[mikeperry at seul.org]

Author: mikeperry
Date: 2009-01-05 18:55:24 -0500 (Mon, 05 Jan 2009)
New Revision: 17937

Added:
   torflow/trunk/NetworkScanners/speedracer.py
Log:

Add python version of speedracer from Karsten.



Added: torflow/trunk/NetworkScanners/speedracer.py
===================================================================
--- torflow/trunk/NetworkScanners/speedracer.py	                        (rev 0)
+++ torflow/trunk/NetworkScanners/speedracer.py	2009-01-05 23:55:24 UTC (rev 17937)
@@ -0,0 +1,171 @@
+#!/usr/bin/python
+#
+# 2009 Mike Perry, Karsten Loesing
+
+"""
+Speedracer
+
+Speedracer continuously requests the Tor design paper over the Tor network
+and measures how long circuit building and downloading takes.
+"""
+
+import socket
+from time import time
+import sys
+import urllib2
+import re
+
+sys.path.append("../")
+from TorCtl.TorUtil import plog
+from TorCtl.TorUtil import meta_port, meta_host, control_port, control_host, tor_port, tor_host
+
+sys.path.append("./libs")
+from SocksiPy import socks
+
+# Some constants for measurements
+user_agent = "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; .NET CLR 1.0.3705; .NET CLR 1.1.4322)"
+url = "https://svn.torproject.org/svn/tor/trunk/doc/design-paper/tor-design.pdf"
+count = 2
+start_pct = 0
+stop_pct = 20
+pct_step = 5
+
+class MetatrollerException(Exception):
+    "Metatroller does not accept this command."
+    pass
+
+# Connector to the metatroller
+class MetatrollerConnector:
+
+    def __init__(self, host, port):
+        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.sock.connect((host, port))
+        self.buffer = self.sock.makefile('rb')
+
+    def send_command_and_check(self, line):
+        self.sock.send(line + '\r\n')
+        reply = self.readline()
+        if reply[:3] != '250':
+            plog('ERROR', reply)
+            raise MetatrollerException(reply)
+        return reply
+
+    def readline(self):
+        response = self.buffer.readline()
+        if not response:
+            raise EOFError
+        elif response[-2:] == '\r\n':
+            response = response[:-2]
+        elif response[-1:] in '\r\n':
+            response = response[:-1]
+        return response 
+
+def get_exit_node(meta):
+    ''' ask metatroller for the last exit used '''
+    reply = meta.send_command_and_check("GETLASTEXIT")
+    p = re.compile('250 LASTEXIT=[\S]+')
+    m = p.match(reply)
+    exit_node = m.group()[13:]
+    plog('DEBUG', 'Current node: ' + exit_node)
+    return exit_node
+
+
+def http_request(address):
+    ''' perform an http GET-request and return 1 for success or 0 for failure '''
+
+    request = urllib2.Request(address)
+    request.add_header('User-Agent', user_agent)
+
+    try:
+        reply = urllib2.urlopen(request)
+        reply.read()
+        return 1
+    except (ValueError, urllib2.URLError):
+        plog('ERROR', 'The http-request address ' + address + ' is malformed')
+        return 0
+    except (IndexError, TypeError):
+        plog('ERROR', 'An error occured while negotiating socks5 with Tor')
+        return 0
+
+def speedrace(meta, skip, pct):
+
+    meta.send_command_and_check('PERCENTSKIP ' + str(skip))
+    meta.send_command_and_check('PERCENTFAST ' + str(pct))
+
+    attempt = 0
+    successful = 0
+    while successful < count:
+
+        meta.send_command_and_check('NEWNYM')
+        
+        attempt += 1
+        
+        t0 = time()
+        ret = http_request(url)
+        if ret == 1:
+            successful += 1
+        delta_build = time() - t0
+        if delta_build >= 550.0:
+            plog('NOTICE', 'Timer exceeded limit: ' + delta_build + '\n')
+
+        build_exit = get_exit_node(meta)
+        fetch_exit = build_exit
+
+        plog('DEBUG', 'circuit build+fetch took ' + str(delta_build) + ' for ' + str(fetch_exit))
+
+    plog('INFO', str(skip) + '-' + str(pct) + '% ' + str(count) + ' fetches took ' + str(attempt) + ' tries.')
+
+def main(argv):
+    # establish a metatroller connection
+    plog('INFO', 'Connecting to metatroller...')
+    try:
+        meta = MetatrollerConnector(meta_host, meta_port)
+    except socket.error:
+        plog('ERROR', 'Couldn\'t connect to metatroller. Is it on?')
+        exit()
+
+    # skip two lines of metatroller introduction
+    meta.readline()
+    meta.readline()
+        
+    # configure metatroller
+    commands = [
+        'PATHLEN 2',
+        'UNIFORM 1',
+        'ORDEREXITS 1',
+        'GUARDNODES 0']
+    plog('INFO', 'Executing preliminary configuration commands')
+    for c in commands:
+        meta.send_command_and_check(c)
+
+    # set SOCKS proxy
+    socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, tor_host, tor_port)
+    socket.socket = socks.socksocket
+
+    pct = start_pct
+    plog('INFO', 'Beginning time loop')
+    
+    while pct < stop_pct:
+        meta.send_command_and_check('RESETSTATS')
+        meta.send_command_and_check('COMMIT')
+        plog('DEBUG', 'Reset stats')
+
+        speedrace(meta, pct, pct + pct_step)
+
+        plog('DEBUG', 'speedroced')
+        meta.send_command_and_check('CLOSEALLCIRCS')
+        meta.send_command_and_check('SAVESTATS ./data/speedraces/stats-' + str(pct) + ':' + str(pct + pct_step))
+        plog('DEBUG', 'Wrote stats')
+        pct += pct_step
+        meta.send_command_and_check('COMMIT')
+
+# initiate the program
+if __name__ == '__main__':
+    try:
+        main(sys.argv)
+    except KeyboardInterrupt:
+        plog('INFO', "Ctrl + C was pressed. Exiting ... ")
+    except Exception, e:
+        plog('ERROR', "An unexpected error occured.")
+        plog('ERROR', e)
+


Property changes on: torflow/trunk/NetworkScanners/speedracer.py
___________________________________________________________________
Name: svn:executable
   + *