Issue #11179: Make ccbench work under Python 3.1 and 2.7 again.

pitrou · pitrou · commit 375ff587c2c4 · 2011-03-11T20:57:11.000+01:00
diff --git a/Misc/NEWS b/Misc/NEWS
@@ -81,6 +81,11 @@ Build
 - Issue #11268: Prevent Mac OS X Installer failure if Documentation
   package had previously been installed.
 
+Tools/Demos
+-----------
+
+- Issue #11179: Make ccbench work under Python 3.1 and 2.7 again.
+
 Tests
 -----
 
diff --git a/Tools/ccbench/ccbench.py b/Tools/ccbench/ccbench.py
@@ -276,7 +276,8 @@ def _recv(sock, n):
     return sock.recv(n).decode('ascii')
 
 def latency_client(addr, nb_pings, interval):
-    with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as sock:
+    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    try:
         _time = time.time
         _sleep = time.sleep
         def _ping():
@@ -289,6 +290,8 @@ def _ping():
             _sleep(interval)
             _ping()
         _sendto(sock, LAT_END + "\n", addr)
+    finally:
+        sock.close()
 
 def run_latency_client(**kwargs):
     cmd_line = [sys.executable, '-E', os.path.abspath(__file__)]
