async def _send_cmd(transport: Transport, serializer, **req) -> dict:
"""
Raises:
Backend
BackendNotAvailable
BackendProtocolError
BackendCmdsInvalidRequest
BackendCmdsInvalidResponse
BackendNotAvailable
BackendCmdsBadResponse
"""
transport.logger.info("Request", cmd=req["cmd"])
try:
raw_req = serializer.req_dumps(req)
except ProtocolError as exc:
transport.logger.exception("Invalid request data",
cmd=req["cmd"],
error=exc)
raise BackendProtocolError("Invalid request data") from exc
try:
await transport.send(raw_req)
raw_rep = await transport.recv()
except TransportError as exc:
transport.logger.debug("Request failed (backend not available)",
cmd=req["cmd"])
raise BackendNotAvailable(exc) from exc
try:
rep = serializer.rep_loads(raw_rep)
except ProtocolError as exc:
transport.logger.exception("Invalid response data",
cmd=req["cmd"],
error=exc)
raise BackendProtocolError("Invalid response data") from exc
if rep["status"] == "invalid_msg_format":
transport.logger.error("Invalid request data according to backend",
cmd=req["cmd"],
rep=rep)
raise BackendProtocolError("Invalid request data according to backend")
if rep["status"] == "bad_timestamp":
raise BackendOutOfBallparkError(rep)
# Backward compatibility with older backends (<= v2.3)
if rep["status"] == "invalid_certification" and "timestamp" in rep[
"reason"]:
raise BackendOutOfBallparkError(rep)
return rep
async def _send_cmd(transport: Transport, serializer, **req) -> dict:
"""
Raises:
Backend
BackendNotAvailable
BackendProtocolError
BackendCmdsInvalidRequest
BackendCmdsInvalidResponse
BackendNotAvailable
BackendCmdsBadResponse
"""
transport.logger.info("Request", cmd=req["cmd"])
try:
raw_req = serializer.req_dumps(req)
except ProtocolError as exc:
transport.logger.exception("Invalid request data",
cmd=req["cmd"],
error=exc)
raise BackendProtocolError("Invalid request data") from exc
try:
await transport.send(raw_req)
raw_rep = await transport.recv()
except TransportError as exc:
transport.logger.debug("Request failed (backend not available)",
cmd=req["cmd"])
raise BackendNotAvailable(exc) from exc
try:
rep = serializer.rep_loads(raw_rep)
except ProtocolError as exc:
transport.logger.exception("Invalid response data",
cmd=req["cmd"],
error=exc)
raise BackendProtocolError("Invalid response data") from exc
if rep["status"] == "invalid_msg_format":
transport.logger.error("Invalid request data according to backend",
cmd=req["cmd"],
rep=rep)
raise BackendProtocolError("Invalid request data according to backend")
return rep
async def _do_handshake(transport: Transport, handshake):
try:
challenge_req = await transport.recv()
answer_req = handshake.process_challenge_req(challenge_req)
await transport.send(answer_req)
result_req = await transport.recv()
handshake.process_result_req(result_req)
except TransportError as exc:
raise BackendNotAvailable(exc) from exc
except HandshakeOutOfBallparkError as exc:
raise BackendOutOfBallparkError(exc) from exc
except HandshakeError as exc:
if str(exc) == "Invalid handshake: Invitation not found":
raise BackendInvitationNotFound(str(exc)) from exc
elif str(exc) == "Invalid handshake: Invitation already deleted":
raise BackendInvitationAlreadyUsed(str(exc)) from exc
else:
raise BackendConnectionRefused(str(exc)) from exc
except ProtocolError as exc:
transport.logger.exception("Protocol error during handshake")
raise BackendProtocolError(exc) from exc
async def _anonymous_cmd(serializer, addr: BackendAddr,
organization_id: OrganizationID, **req) -> dict:
"""
Raises:
BackendNotAvailable
BackendProtocolError
"""
logger.info("Request", cmd=req["cmd"])
try:
raw_req = serializer.req_dumps(req)
except ProtocolError as exc:
logger.exception("Invalid request data", cmd=req["cmd"], error=exc)
raise BackendProtocolError("Invalid request data") from exc
url = addr.to_http_domain_url(f"/anonymous/{organization_id}")
try:
raw_rep = await http_request(url=url, method="POST", data=raw_req)
except OSError as exc:
logger.debug("Request failed (backend not available)",
cmd=req["cmd"],
exc_info=exc)
raise BackendNotAvailable(exc) from exc
try:
rep = serializer.rep_loads(raw_rep)
except ProtocolError as exc:
logger.exception("Invalid response data", cmd=req["cmd"], error=exc)
raise BackendProtocolError("Invalid response data") from exc
if rep["status"] == "invalid_msg_format":
logger.error("Invalid request data according to backend",
cmd=req["cmd"],
rep=rep)
raise BackendProtocolError("Invalid request data according to backend")
if rep["status"] == "bad_timestamp":
raise BackendOutOfBallparkError(rep)
return rep
async def _do_handshake(transport: Transport, handshake):
try:
challenge_req = await transport.recv()
answer_req = handshake.process_challenge_req(challenge_req)
await transport.send(answer_req)
result_req = await transport.recv()
handshake.process_result_req(result_req)
except TransportError as exc:
raise BackendNotAvailable(exc) from exc
except HandshakeError as exc:
raise BackendConnectionRefused(str(exc)) from exc
except ProtocolError as exc:
transport.logger.exception("Protocol error during handshake")
raise BackendProtocolError(exc) from exc