Exemplo n.º 1
0
 def connect(self):
     try:
         self.address = socket.gethostbyname(self.address)
         self.header = MSG_PREFIX + encode_bytes(
             *map(int, self.address.split('.'))) + encode_bytes(
                 self.port & 0xFF, self.port >> 8 & 0xFF)
         self.socket = self.socket_cls(socket.AF_INET, socket.SOCK_DGRAM)
         self.socket.settimeout(self.timeout)
         return self
     except socket.error as e:
         raise ConnectionError(e)
Exemplo n.º 2
0
    def send_request(self, opcode, extras=b'', return_response=True):
        ip_address = encode_bytes(*[(int(n)) for n in self.address.split('.')])
        port_number = encode_bytes(self.port & 0xFF, self.port >> 8 & 0xFF)
        body = MSG_PREFIX \
               + ip_address \
               + port_number \
               + opcode \
               + extras
        self.socket.sendto(body, (self.address, self.port))

        if return_response:
            return self.receive()
Exemplo n.º 3
0
 def send_rcon_command(self, command, args=tuple(), fetch_response=True):
     """
     Send any command to the server
     leading whitespace is stripped from the response
     :param command: the comand to send
     :param args: tuple or list of arguments to be appended to the command. Can be also a string or an int if only one argument is expected.
     :param fetch_response: Whether to receive response from server. Set this to False if you're not expecting a response; WARNING: If there is a response and you don't fetch it, it may be output as a response of your next command.
     :return list of lines responded from the server or None if fetch_response == False
     """
     command = build_rcon_command(command, args)
     command_length = encode_bytes(
         len(command) & 0xFF,
         len(command) >> 8 & 0xFF)
     payload = self.rcon_password_bytes + command_length + command
     self.send_request(OPCODE_RCON, extras=payload, return_response=False)
     if fetch_response:
         result = []
         while True:
             response = self.receive()
             if response is None:
                 break
             line = decode_string(response, 0, 2)
             if line:
                 result.append(line.lstrip())
             else:
                 break
     if len(result) == 1 and result[0] == 'Invalid RCON password.':
         raise InvalidRconPassword
     return result
Exemplo n.º 4
0
 def rcon_password_bytes(self):
     """
     password prefixed with its encoded length
     """
     if not self.rcon_password:
         raise RconError('Rcon password was not provided')
     pass_len = len(self.rcon_password)
     return encode_bytes(pass_len & 0xFF, pass_len >> 8 & 0xFF) + bytes(
         self.rcon_password, ENCODING)
Exemplo n.º 5
0
 def test_encode(self):
     expected = bytes(chr(127) + chr(0) + chr(0) + chr(1), 'utf-8')
     self.assertEqual(expected, utils.encode_bytes(127, 0, 0, 1))