diff -r e7a09ceb4505 -r 3874575d924d sources/network/rconsession.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/sources/network/rconsession.cpp Fri Dec 12 00:55:51 2014 +0200 @@ -0,0 +1,219 @@ +#include "rconsession.h" + +// ------------------------------------------------------------------------------------------------- +// +RCONSession::RCONSession() : + m_state (RCON_DISCONNECTED), + m_lastPing (0) +{ + if (not m_socket.set_blocking (false)) + { + // TODO: find a better way to deal with errors + fprintf (stderr, "unable to set socket as non-blocking: %s\n", + m_socket.error_string().chars()); + } +} + +// ------------------------------------------------------------------------------------------------- +// +RCONSession::~RCONSession() {} + +// ------------------------------------------------------------------------------------------------- +// +METHOD +RCONSession::connect (IPAddress address) -> void +{ + m_address = address; + m_state = RCON_CONNECTING; +} + +// ------------------------------------------------------------------------------------------------- +// +METHOD +RCONSession::disconnect() -> void +{ + if (m_state == RCON_CONNECTED) + { + // Say goodbye to remote + Bytestream packet; + packet.write_byte (CLRC_DISCONNECT); + this->send (packet); + } + + m_state = RCON_DISCONNECTED; +} + +// ------------------------------------------------------------------------------------------------- +// +METHOD +RCONSession::send (const Bytestream& packet) -> void +{ + m_socket.send (m_address, packet); +} + +// ------------------------------------------------------------------------------------------------- +// +METHOD +RCONSession::tick() -> void +{ + time_t now; + time (&now); + + if (m_lastPing < now) + { + if (m_state == RCON_CONNECTING) + { + send_hello(); + } + else if (m_state == RCON_AUTHENTICATING) + { + send_password(); + } + else if (m_state == RCON_CONNECTED and m_lastPing + 5 < now) + { + Bytestream packet; + packet.write_byte (CLRC_PONG); + send (packet); + bump_last_ping(); + } + } + + for (Datagram datagram; m_socket.read (datagram);) + handle_packet (datagram.data, datagram.from); +} + +// ------------------------------------------------------------------------------------------------- +// +METHOD +RCONSession::handle_packet (Bytestream& packet, const IPAddress& from) -> void +{ + print ("Handling packet of %1 bytes\n", packet.written_length()); + bool ok = true; + + while (packet.bytes_left() > 0) + { + print ("%1/%2 bytes left\n", packet.bytes_left(), packet.written_length()); + int header = packet.read_byte (&ok); + print ("recieved HEADER with %1\n", header); + + switch (ServerResponse (header)) + { + case SVRC_OLDPROTOCOL: + fprintf (stderr, "wrong version\n"); + m_state = RCON_DISCONNECTED; + break; + + case SVRC_BANNED: + fprintf (stderr, "you're banned\n"); + m_state = RCON_DISCONNECTED; + break; + + case SVRC_SALT: + { + String salt = packet.read_string(); + m_salt = salt; + m_state = RCON_AUTHENTICATING; + send_password(); + } + break; + + case SVRC_LOGGEDIN: + fprintf (stderr, "login successful\n"); + m_state = RCON_CONNECTED; + break; + + case SVRC_INVALIDPASSWORD: + fprintf (stderr, "bad password\n"); + m_state = RCON_DISCONNECTED; + break; + + case SVRC_MESSAGE: + { + String message = packet.read_string(); + print_to (stderr, "message: %1\n", message); + } + break; + + case SVRC_UPDATE: + switch (RCONUpdateType (packet.read_byte (&ok))) + { + case SVRCU_PLAYERDATA: + { + int numplayers = packet.read_byte (&ok); + Vector players; + + while (numplayers--) + players << packet.read_string (&ok); + + print_to (stderr, "players: %1\n", players); + } + break; + + case SVRCU_ADMINCOUNT: + print_to (stderr, "num admins: %1\n", packet.read_byte (&ok)); + break; + + case SVRCU_MAP: + print_to (stderr, "new map: %1\n", packet.read_string (&ok)); + break; + } + + break; + } + + if (not ok) + print_to (stderr, "error while reading packet\n"); + } +} + +// ------------------------------------------------------------------------------------------------- +// +METHOD +RCONSession::socket() -> UDPSocket* +{ + return &m_socket; +} + +// ------------------------------------------------------------------------------------------------- +// +METHOD +RCONSession::send_hello() -> void +{ + print ("connecting to %1...\n", m_address.to_string (IP_WITH_PORT)); + Bytestream packet; + packet.write_byte (CLRC_BEGINCONNECTION); + packet.write_byte (RCON_PROTOCOL_VERSION); + send (packet); + bump_last_ping(); +} + +// ------------------------------------------------------------------------------------------------- +// +METHOD +RCONSession::send_password() -> void +{ + print ("sending password...\n"); + Bytestream packet; + packet.write_byte (CLRC_PASSWORD); + packet.write_string ((m_salt + m_password).md5()); + send (packet); + bump_last_ping(); +} + +// ------------------------------------------------------------------------------------------------- +// +METHOD +RCONSession::set_password (const String& password) -> void +{ + m_password = password; +} + +// ------------------------------------------------------------------------------------------------- +// +METHOD +RCONSession::bump_last_ping() -> void +{ + time_t now; + time (&now); + m_lastPing = now; +}