[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]
[PATCH 06/10] python/aqmp: refactor _do_accept() into two distinct steps
From: |
John Snow |
Subject: |
[PATCH 06/10] python/aqmp: refactor _do_accept() into two distinct steps |
Date: |
Fri, 25 Feb 2022 15:59:44 -0500 |
Refactor _do_accept() into _do_start_server() and _do_accept(). As of
this commit, the former calls the latter, but in subsequent commits
they'll be split apart.
(So please forgive the misnomer for _do_start_server(); it will live up
to its name shortly, and the docstring will be updated then too. I'm
just cutting down on some churn.)
Signed-off-by: John Snow <jsnow@redhat.com>
---
python/qemu/aqmp/protocol.py | 29 ++++++++++++++++++++++++-----
python/tests/protocol.py | 4 ++--
2 files changed, 26 insertions(+), 7 deletions(-)
diff --git a/python/qemu/aqmp/protocol.py b/python/qemu/aqmp/protocol.py
index 631bcdaa55..e2bdad542d 100644
--- a/python/qemu/aqmp/protocol.py
+++ b/python/qemu/aqmp/protocol.py
@@ -295,7 +295,7 @@ async def start_server_and_accept(
session, the wrapped error may also be an `QMPError`.
"""
await self._session_guard(
- self._do_accept(address, ssl),
+ self._do_start_server(address, ssl),
'Failed to establish connection')
await self._session_guard(
self._establish_session(),
@@ -509,8 +509,8 @@ def _bind_hack(self, address: Union[str, Tuple[str, int]])
-> None:
self._sock = sock
@upper_half
- async def _do_accept(self, address: SocketAddrT,
- ssl: Optional[SSLContext] = None) -> None:
+ async def _do_start_server(self, address: SocketAddrT,
+ ssl: Optional[SSLContext] = None) -> None:
"""
Acting as the transport server, accept a single connection.
@@ -551,9 +551,28 @@ async def _do_accept(self, address: SocketAddrT,
# otherwise yield.
await asyncio.sleep(0)
- self._server = await coro # Starts listening
- await self._accepted.wait() # Waits for the callback to finish
+ # This will start the server (bind(2), listen(2)). It will also
+ # call accept(2) if we yield, but we don't block on that here.
+ self._server = await coro
+
+ # Just for this one commit, wait for a peer.
+ # This gets split out in the next patch.
+ await self._do_accept()
+
+ @upper_half
+ async def _do_accept(self) -> None:
+ """
+ Wait for and accept an incoming connection.
+
+ Requires that we have not yet accepted an incoming connection
+ from the upper_half, but it's OK if the server is no longer
+ running because the bottom_half has already accepted the
+ connection.
+ """
+ assert self._accepted is not None
+ await self._accepted.wait()
assert self._server is None
+ self._accepted = None
self._sock = None
self.logger.debug("Connection accepted.")
diff --git a/python/tests/protocol.py b/python/tests/protocol.py
index 8dd26c4ed1..5e442e1efb 100644
--- a/python/tests/protocol.py
+++ b/python/tests/protocol.py
@@ -41,12 +41,12 @@ async def _establish_session(self):
self.trigger_input = asyncio.Event()
await super()._establish_session()
- async def _do_accept(self, address, ssl=None):
+ async def _do_start_server(self, address, ssl=None):
if self.fake_session:
self._set_state(Runstate.CONNECTING)
await asyncio.sleep(0)
else:
- await super()._do_accept(address, ssl)
+ await super()._do_start_server(address, ssl)
async def _do_connect(self, address, ssl=None):
if self.fake_session:
--
2.34.1
- [PATCH 00/10] Python: Fix qmp race condition on accept(), John Snow, 2022/02/25
- [PATCH 02/10] python/aqmp: rename 'accept()' to 'start_server_and_accept()', John Snow, 2022/02/25
- [PATCH 01/10] python/aqmp: add _session_guard(), John Snow, 2022/02/25
- [PATCH 03/10] python/aqmp: remove _new_session and _establish_connection, John Snow, 2022/02/25
- [PATCH 06/10] python/aqmp: refactor _do_accept() into two distinct steps,
John Snow <=
- [PATCH 05/10] python/aqmp: squelch pylint warning for too many lines, John Snow, 2022/02/25
- [PATCH 04/10] python/aqmp: split _client_connected_cb() out as _incoming(), John Snow, 2022/02/25
- [PATCH 09/10] python/aqmp: fix race condition in legacy.py, John Snow, 2022/02/25
- [PATCH 07/10] python/aqmp: stop the server during disconnect(), John Snow, 2022/02/25
- [PATCH 10/10] python/aqmp: drop _bind_hack(), John Snow, 2022/02/25
- [PATCH 08/10] python/aqmp: add start_server() and accept() methods, John Snow, 2022/02/25