system/qemu
system/qemu
1
0
Fork 0
mirror of https://gitlab.com/qemu-project/qemu synced 2024-10-15 15:32:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

python/qmp/protocol: add open_with_socket()

Browse source 
Instead of listening for incoming connections with a SocketAddr, add a
new method open_with_socket() that accepts an existing socket.

Signed-off-by: Marc-André Lureau <marcandre.lureau@redhat.com>
Reviewed-by: Daniel P. Berrangé <berrange@redhat.com>
Message-id: 20230111080101.969151-2-marcandre.lureau@redhat.com
Signed-off-by: John Snow <jsnow@redhat.com>
This commit is contained in:
Marc-André Lureau 2023-01-11 12:00:59 +04:00 committed by John Snow
parent 166464c6ce
commit a3cfea92e2
1 changed files with 20 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
python/qemu/qmp/protocol.py
View file

@ -18,6 +18,7 @@
from enum import Enum from enum import Enum
from functools import wraps from functools import wraps
import logging import logging
import socket
from ssl import SSLContext from ssl import SSLContext
from typing import ( from typing import (
Any, Any,
@ -296,6 +297,19 @@ async def start_server_and_accept(
await self.accept() await self.accept()
assert self.runstate == Runstate.RUNNING assert self.runstate == Runstate.RUNNING
@upper_half
@require(Runstate.IDLE)
async def open_with_socket(self, sock: socket.socket) -> None:
"""
Start connection with given socket.
:param sock: A socket.
:raise StateError: When the `Runstate` is not `IDLE`.
"""
self._reader, self._writer = await asyncio.open_connection(sock=sock)
self._set_state(Runstate.CONNECTING)
@upper_half @upper_half
@require(Runstate.IDLE) @require(Runstate.IDLE)
async def start_server(self, address: SocketAddrT, async def start_server(self, address: SocketAddrT,
@ -343,11 +357,12 @@ async def accept(self) -> None:
protocol-level failure occurs while establishing a new protocol-level failure occurs while establishing a new
session, the wrapped error may also be an `QMPError`. session, the wrapped error may also be an `QMPError`.
""" """
if self._accepted is None: if not self._reader:
raise QMPError("Cannot call accept() before start_server().") if self._accepted is None:
await self._session_guard( raise QMPError("Cannot call accept() before start_server().")
self._do_accept(), await self._session_guard(
'Failed to establish connection') self._do_accept(),
'Failed to establish connection')
await self._session_guard( await self._session_guard(
self._establish_session(), self._establish_session(),
'Failed to establish session') 'Failed to establish session')