1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
"""Asynchronous RCON."""
from asyncio import StreamReader, StreamWriter, open_connection, wait_for
from rcon.exceptions import SessionTimeout, WrongPassword
from rcon.source.proto import Packet, Type
__all__ = ["rcon"]
async def close(writer: StreamWriter) -> None:
"""Close socket asynchronously."""
writer.close()
await writer.wait_closed()
async def communicate(
reader: StreamReader,
writer: StreamWriter,
packet: Packet,
*,
frag_threshold: int = 4096,
frag_detect_cmd: str = "",
raise_unexpected_terminator: bool = False,
) -> Packet:
"""Make an asynchronous request."""
writer.write(bytes(packet))
await writer.drain()
response = await Packet.aread(reader, raise_unexpected_terminator)
if len(response.payload) < frag_threshold:
return response
writer.write(bytes(Packet.make_command(frag_detect_cmd)))
await writer.drain()
while (
successor := await Packet.aread(reader, raise_unexpected_terminator)
).id == response.id:
response += successor
return response
async def rcon(
command: str,
*arguments: str,
host: str,
port: int,
passwd: str,
encoding: str = "utf-8",
frag_threshold: int = 4096,
frag_detect_cmd: str = "",
timeout: int | None = None,
enforce_id: bool = True,
raise_unexpected_terminator: bool = False,
) -> str:
"""Run a command asynchronously."""
reader, writer = await wait_for(open_connection(host, port), timeout=timeout)
response = await communicate(
reader,
writer,
Packet.make_login(passwd, encoding=encoding),
frag_threshold=frag_threshold,
frag_detect_cmd=frag_detect_cmd,
)
# Wait for SERVERDATA_AUTH_RESPONSE according to:
# https://developer.valvesoftware.com/wiki/Source_RCON_Protocol
while response.type != Type.SERVERDATA_AUTH_RESPONSE:
response = await Packet.aread(reader, raise_unexpected_terminator)
if response.id == -1:
await close(writer)
raise WrongPassword()
request = Packet.make_command(command, *arguments, encoding=encoding)
response = await communicate(
reader, writer, request, raise_unexpected_terminator=raise_unexpected_terminator
)
await close(writer)
if enforce_id and response.id != request.id:
raise SessionTimeout()
return response.payload.decode(encoding)
|
