[tor-commits] [chutney/master] Use a separate asyncore socketmap for every TrafficTester

[teor at torproject.org]

commit 494f3fc1c3c0f53bade89fe58a9dcd6395080a65
Author: Nick Mathewson <nickm at torproject.org>
Date:   Fri May 10 15:56:13 2019 -0400

    Use a separate asyncore socketmap for every TrafficTester
    
    This should prevent multiple in-process runs from interfering
---
 lib/chutney/Traffic.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/lib/chutney/Traffic.py b/lib/chutney/Traffic.py
index f2448a8..ed4ad69 100755
--- a/lib/chutney/Traffic.py
+++ b/lib/chutney/Traffic.py
@@ -118,7 +118,7 @@ class Listener(asyncore.dispatcher):
     "A TCP listener, binding, listening and accepting new connections."
 
     def __init__(self, tt, endpoint):
-        asyncore.dispatcher.__init__(self)
+        asyncore.dispatcher.__init__(self, map=tt.socket_map)
         self.create_socket(addr_to_family(endpoint[0]), socket.SOCK_STREAM)
         self.set_reuse_addr()
         self.bind(endpoint)
@@ -199,7 +199,7 @@ class DataChecker(object):
 class Sink(asynchat.async_chat):
     "A data sink, reading from its peer and verifying the data."
     def __init__(self, sock, tt):
-        asynchat.async_chat.__init__(self, sock)
+        asynchat.async_chat.__init__(self, sock, map=tt.socket_map)
         self.set_terminator(None)
         self.tt = tt
         self.data_checker = DataChecker(tt.data_source.copy())
@@ -243,7 +243,7 @@ class Source(asynchat.async_chat):
     CONNECTED = 5
 
     def __init__(self, tt, server, proxy=None):
-        asynchat.async_chat.__init__(self)
+        asynchat.async_chat.__init__(self, map=tt.socket_map)
         self.data_source = tt.data_source.copy()
         self.inbuf = b''
         self.proxy = proxy
@@ -299,7 +299,7 @@ class Source(asynchat.async_chat):
 
 class EchoServer(asynchat.async_chat):
     def __init__(self, sock, tt):
-        asynchat.async_chat.__init__(self, sock)
+        asynchat.async_chat.__init__(self, sock, map=tt.socket_map)
         self.set_terminator(None)
         self.tt = tt
 
@@ -363,6 +363,8 @@ class TrafficTester(object):
             self.client_class = Source
             self.responder_class = Sink
 
+        self.socket_map = {}
+
         self.listener = Listener(self, endpoint)
         self.pending_close = []
         self.timeout = timeout
@@ -405,7 +407,7 @@ class TrafficTester(object):
         while now < end and not self.tests.all_done():
             # run only one iteration at a time, with a nice short timeout, so we
             # can actually detect completion and timeouts.
-            asyncore.loop(0.2, False, None, 1)
+            asyncore.loop(0.2, False, self.socket_map, 1)
             now = time.time()
             if now > dump_at:
                 debug("Test status: %s"%self.tests.status())