mirror of https://github.com/conqp/rcon
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.6 KiB
57 lines
1.6 KiB
"""BattlEye RCon client."""
|
|
|
|
from ipaddress import IPv4Address
|
|
from socket import SOCK_DGRAM
|
|
from typing import Union
|
|
|
|
from rcon.battleye.proto import Command
|
|
from rcon.battleye.proto import CommandResponse
|
|
from rcon.battleye.proto import Header
|
|
from rcon.battleye.proto import LoginRequest
|
|
from rcon.battleye.proto import LoginResponse
|
|
from rcon.client import BaseClient
|
|
from rcon.exceptions import WrongPassword
|
|
|
|
|
|
__all__ = ['Client']
|
|
|
|
|
|
Host = Union[str, IPv4Address]
|
|
|
|
|
|
class Client(BaseClient, socket_type=SOCK_DGRAM):
|
|
"""BattlEye RCon client."""
|
|
|
|
def send(self, data: bytes) -> None:
|
|
"""Sends bytes."""
|
|
with self._socket.makefile('wb') as file:
|
|
file.write(data)
|
|
|
|
def _login(self, login_request: LoginRequest) -> LoginResponse:
|
|
"""Logs the user in."""
|
|
self.send(bytes(login_request))
|
|
|
|
with self._socket.makefile('rb') as file:
|
|
return LoginResponse.read(file)
|
|
|
|
def login(self, passwd: str) -> bool:
|
|
"""Logs the user in."""
|
|
if not self._login(LoginRequest.from_passwd(passwd)).success:
|
|
raise WrongPassword()
|
|
|
|
return True
|
|
|
|
def _run(self, command: Command) -> CommandResponse:
|
|
"""Executes a command."""
|
|
self.send(bytes(command))
|
|
|
|
with self._socket.makefile('rb') as file:
|
|
header = Header.read(file)
|
|
|
|
# TODO: Can we determine the packet size?
|
|
remainder = self._socket.recv(4096)
|
|
return CommandResponse.from_bytes(header, remainder)
|
|
|
|
def run(self, command: str, *args: str) -> str:
|
|
"""Executes a command."""
|
|
return self._run(Command.from_command(command, *args)).message
|
|
|