summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
authorBerker Peksag <berker.peksag@gmail.com>2016-09-17 20:22:06 (GMT)
committerBerker Peksag <berker.peksag@gmail.com>2016-09-17 20:22:06 (GMT)
commitbcfb35f80d9d1f87d9fa6993c1d3bc35dd5db865 (patch)
tree8210a1031ab648ac60c6fa391f89e8964cec1bb2 /Lib
parent9a1c91a10d16b603d57fbc97bc54699acee3f052 (diff)
downloadcpython-bcfb35f80d9d1f87d9fa6993c1d3bc35dd5db865.zip
cpython-bcfb35f80d9d1f87d9fa6993c1d3bc35dd5db865.tar.gz
cpython-bcfb35f80d9d1f87d9fa6993c1d3bc35dd5db865.tar.bz2
Issue #26384: Fix UnboundLocalError in socket._sendfile_use_sendfile
Diffstat (limited to 'Lib')
-rw-r--r--Lib/socket.py2
-rw-r--r--Lib/test/test_socket.py19
2 files changed, 20 insertions, 1 deletions
diff --git a/Lib/socket.py b/Lib/socket.py
index 6dddfe1..a9cab32 100644
--- a/Lib/socket.py
+++ b/Lib/socket.py
@@ -258,7 +258,7 @@ class socket(_socket.socket):
raise _GiveupOnSendfile(err) # not a regular file
try:
fsize = os.fstat(fileno).st_size
- except OSError:
+ except OSError as err:
raise _GiveupOnSendfile(err) # not a regular file
if not fsize:
return 0 # empty file
diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py
index 4b047ee..c5975c8 100644
--- a/Lib/test/test_socket.py
+++ b/Lib/test/test_socket.py
@@ -1447,6 +1447,25 @@ class GeneralModuleTests(unittest.TestCase):
self.assertEqual(s.family, 42424)
self.assertEqual(s.type, 13331)
+ @unittest.skipUnless(hasattr(os, 'sendfile'), 'test needs os.sendfile()')
+ def test__sendfile_use_sendfile(self):
+ class File:
+ def __init__(self, fd):
+ self.fd = fd
+
+ def fileno(self):
+ return self.fd
+ with socket.socket() as sock:
+ fd = os.open(os.curdir, os.O_RDONLY)
+ os.close(fd)
+ with self.assertRaises(socket._GiveupOnSendfile):
+ sock._sendfile_use_sendfile(File(fd))
+ with self.assertRaises(OverflowError):
+ sock._sendfile_use_sendfile(File(2**1000))
+ with self.assertRaises(TypeError):
+ sock._sendfile_use_sendfile(File(None))
+
+
@unittest.skipUnless(HAVE_SOCKET_CAN, 'SocketCan required for this test.')
class BasicCANTest(unittest.TestCase):