mirror of https://github.com/python/cpython
bpo-38019: correctly handle pause/resume reading of closed asyncio unix pipe (GH-16472)
This commit is contained in:
parent
9a7d951950
commit
58498bc717
|
@ -445,6 +445,7 @@ class _UnixReadPipeTransport(transports.ReadTransport):
|
|||
self._fileno = pipe.fileno()
|
||||
self._protocol = protocol
|
||||
self._closing = False
|
||||
self._paused = False
|
||||
|
||||
mode = os.fstat(self._fileno).st_mode
|
||||
if not (stat.S_ISFIFO(mode) or
|
||||
|
@ -506,10 +507,20 @@ class _UnixReadPipeTransport(transports.ReadTransport):
|
|||
self._loop.call_soon(self._call_connection_lost, None)
|
||||
|
||||
def pause_reading(self):
|
||||
if self._closing or self._paused:
|
||||
return
|
||||
self._paused = True
|
||||
self._loop._remove_reader(self._fileno)
|
||||
if self._loop.get_debug():
|
||||
logger.debug("%r pauses reading", self)
|
||||
|
||||
def resume_reading(self):
|
||||
if self._closing or not self._paused:
|
||||
return
|
||||
self._paused = False
|
||||
self._loop._add_reader(self._fileno, self._read_ready)
|
||||
if self._loop.get_debug():
|
||||
logger.debug("%r resumes reading", self)
|
||||
|
||||
def set_protocol(self, protocol):
|
||||
self._protocol = protocol
|
||||
|
|
|
@ -736,6 +736,7 @@ class UnixReadPipeTransportTests(test_utils.TestCase):
|
|||
@mock.patch('os.read')
|
||||
def test_resume_reading(self, m_read):
|
||||
tr = self.read_pipe_transport()
|
||||
tr.pause_reading()
|
||||
tr.resume_reading()
|
||||
self.loop.assert_reader(5, tr._read_ready)
|
||||
|
||||
|
@ -790,6 +791,32 @@ class UnixReadPipeTransportTests(test_utils.TestCase):
|
|||
self.assertIsNone(tr._protocol)
|
||||
self.assertIsNone(tr._loop)
|
||||
|
||||
def test_pause_reading_on_closed_pipe(self):
|
||||
tr = self.read_pipe_transport()
|
||||
tr.close()
|
||||
test_utils.run_briefly(self.loop)
|
||||
self.assertIsNone(tr._loop)
|
||||
tr.pause_reading()
|
||||
|
||||
def test_pause_reading_on_paused_pipe(self):
|
||||
tr = self.read_pipe_transport()
|
||||
tr.pause_reading()
|
||||
# the second call should do nothing
|
||||
tr.pause_reading()
|
||||
|
||||
def test_resume_reading_on_closed_pipe(self):
|
||||
tr = self.read_pipe_transport()
|
||||
tr.close()
|
||||
test_utils.run_briefly(self.loop)
|
||||
self.assertIsNone(tr._loop)
|
||||
tr.resume_reading()
|
||||
|
||||
def test_resume_reading_on_paused_pipe(self):
|
||||
tr = self.read_pipe_transport()
|
||||
# the pipe is not paused
|
||||
# resuming should do nothing
|
||||
tr.resume_reading()
|
||||
|
||||
|
||||
class UnixWritePipeTransportTests(test_utils.TestCase):
|
||||
|
||||
|
|
|
@ -0,0 +1 @@
|
|||
Correctly handle pause/resume reading of closed asyncio unix pipe.
|
Loading…
Reference in New Issue