+ Implemented option to cleanup directories on PlanetLab slivers.

[nepi.git] / tunbench.py

diff --git a/tunbench.py b/tunbench.py
index b8ac97c..63d842f 100644 (file)
--- a/tunbench.py
+++ b/tunbench.py
@@ -21,15 +21,35 @@ def rread(remote, maxlen, remote_fd = remote.fileno(), os_read=os.read):
     bytes += len(rv)
     return rv
 
-def test(cipher, passphrase):
+def test(cipher, passphrase, plr=None, queuemodule=None):
+   if plr:
+        import random
+        def accept(packet, direction, rng=random.random):
+            return rng() > 0.5
+   else:
+        accept = None
+   if queuemodule:
+        import os, os.path
+        sys.path.append(os.path.join(
+            os.path.dirname(__file__), 
+            'src','nepi','testbeds','planetlab','scripts'))
+        queuemodule = __import__(queuemodule)
+        queueclass = queuemodule.queueclass
+   else:
+        queueclass = None
    TERMINATE = []
+   SUSPEND = []
+
    def stopme():
        time.sleep(100)
        TERMINATE.append(None)
+
    t = threading.Thread(target=stopme)
    t.start()
-   tunchannel.tun_fwd(tun, remote, True, True, passphrase, True, TERMINATE, None, tunkqueue=500,
-        rwrite = rwrite, rread = rread, cipher=cipher)
+   tunchannel.tun_fwd(tun, remote, True, True, passphrase, True, TERMINATE,
+            SUSPEND, None, tunkqueue=500, rwrite = rwrite, rread = rread, 
+            cipher=cipher, queueclass=queueclass, accept_local = accept,
+            accept_remote = accept)
 
 # Swallow exceptions on decryption
 def decrypt(packet, crypter, super=tunchannel.decrypt):
@@ -52,4 +72,19 @@ for cipher in (None, 'AES', 'Blowfish', 'DES', 'DES3'):
     
     print "Bandwidth (%s): %.4fMb/s" % ( cipher, bytes / 200.0 * 8 / 2**20, )
 
+bytes = 0
+cProfile.runctx('test(None,None,0.5)',globals(),locals(),'tunchannel.plr.profile')
+
+print "Profile (50% PLR):"
+pstats.Stats('tunchannel.plr.profile').strip_dirs().sort_stats('time').print_stats()
+
+print "Bandwidth (50%% PLR): %.4fMb/s" % ( bytes / 200.0 * 8 / 2**20, )
+
+bytes = 0
+cProfile.runctx('test(None,None,None,"tosqueue")',globals(),locals(),'tunchannel.tos.profile')
+
+print "Profile (TOS):"
+pstats.Stats('tunchannel.tos.profile').strip_dirs().sort_stats('time').print_stats()
+
+print "Bandwidth (TOS): %.4fMb/s" % ( bytes / 200.0 * 8 / 2**20, )