[Python-checkins] r88623 - in python/branches/release32-maint: Lib/socket.py Lib/test/test_socket.py Misc/NEWS

[antoine.pitrou]

Author: antoine.pitrou
Date: Sat Feb 26 00:14:08 2011
New Revision: 88623

Log:
Merged revisions 88622 via svnmerge from 
svn+ssh://pythondev@svn.python.org/python/branches/py3k

........
  r88622 | antoine.pitrou | 2011-02-26 00:07:44 +0100 (sam., 26 févr. 2011) | 5 lines
  
  Issue #7322: Trying to read from a socket's file-like object after a timeout
  occurred now raises an error instead of silently losing data.
  Patch by Ross Lagerwall.
........


Modified:
   python/branches/release32-maint/   (props changed)
   python/branches/release32-maint/Lib/socket.py
   python/branches/release32-maint/Lib/test/test_socket.py
   python/branches/release32-maint/Misc/NEWS

Modified: python/branches/release32-maint/Lib/socket.py
==============================================================================
--- python/branches/release32-maint/Lib/socket.py	(original)
+++ python/branches/release32-maint/Lib/socket.py	Sat Feb 26 00:14:08 2011
@@ -257,6 +257,7 @@
         self._mode = mode
         self._reading = "r" in mode
         self._writing = "w" in mode
+        self._timeout_occurred = False
 
     def readinto(self, b):
         """Read up to len(b) bytes into the writable buffer *b* and return
@@ -268,9 +269,14 @@
         """
         self._checkClosed()
         self._checkReadable()
+        if self._timeout_occurred:
+            raise IOError("cannot read from timed out object")
         while True:
             try:
                 return self._sock.recv_into(b)
+            except timeout:
+                self._timeout_occurred = True
+                raise
             except error as e:
                 n = e.args[0]
                 if n == EINTR:

Modified: python/branches/release32-maint/Lib/test/test_socket.py
==============================================================================
--- python/branches/release32-maint/Lib/test/test_socket.py	(original)
+++ python/branches/release32-maint/Lib/test/test_socket.py	Sat Feb 26 00:14:08 2011
@@ -1109,6 +1109,23 @@
         self.write_file = None
         SocketConnectedTest.clientTearDown(self)
 
+    def testReadAfterTimeout(self):
+        # Issue #7322: A file object must disallow further reads
+        # after a timeout has occurred.
+        self.cli_conn.settimeout(1)
+        self.read_file.read(3)
+        # First read raises a timeout
+        self.assertRaises(socket.timeout, self.read_file.read, 1)
+        # Second read is disallowed
+        with self.assertRaises(IOError) as ctx:
+            self.read_file.read(1)
+        self.assertIn("cannot read from timed out object", str(ctx.exception))
+
+    def _testReadAfterTimeout(self):
+        self.write_file.write(self.write_msg[0:3])
+        self.write_file.flush()
+        self.serv_finished.wait()
+
     def testSmallRead(self):
         # Performing small file read test
         first_seg = self.read_file.read(len(self.read_msg)-3)

Modified: python/branches/release32-maint/Misc/NEWS
==============================================================================
--- python/branches/release32-maint/Misc/NEWS	(original)
+++ python/branches/release32-maint/Misc/NEWS	Sat Feb 26 00:14:08 2011
@@ -24,6 +24,9 @@
 Library
 -------
 
+- Issue #7322: Trying to read from a socket's file-like object after a timeout
+  occurred now raises an error instead of silently losing data.
+
 - Issue #10956: Buffered I/O classes retry reading or writing after a signal
   has arrived and the handler returned successfully.