network_turn.cpp

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 */
 
#include "../stdafx.h"
#include "../debug.h"
#include "../error.h"
#include "../strings_func.h"
#include "network_coordinator.h"
#include "network_turn.h"
 
#include "table/strings.h"
 
#include "../safeguards.h"
 
class NetworkTurnConnecter : public TCPConnecter {
private:
  ClientNetworkTurnSocketHandler *handler;
 
public:
  NetworkTurnConnecter(ClientNetworkTurnSocketHandler *handler, const std::string &connection_string) : TCPConnecter(connection_string, NETWORK_TURN_SERVER_PORT), handler(handler) {}
 
  void OnFailure() override
  {
    Debug(net, 9, "Turn::OnFailure()");
 
    this->handler->connecter = nullptr;
 
    this->handler->ConnectFailure();
  }
 
  void OnConnect(SOCKET s) override
  {
    Debug(net, 9, "Turn::OnConnect()");
 
    this->handler->connecter = nullptr;
 
    this->handler->sock = s;
  }
};
 
bool ClientNetworkTurnSocketHandler::Receive_TURN_ERROR(Packet &)
{
  Debug(net, 9, "Receive_TURN_ERROR()");
 
  this->ConnectFailure();
 
  return false;
}
 
bool ClientNetworkTurnSocketHandler::Receive_TURN_CONNECTED(Packet &p)
{
  Debug(net, 9, "Receive_TURN_CONNECTED()");
 
  std::string hostname = p.Recv_string(NETWORK_HOSTNAME_LENGTH);
 
  /* Act like we no longer have a socket, as we are handing it over to the
   * game handler. */
  SOCKET game_sock = this->sock;
  this->sock = INVALID_SOCKET;
 
  NetworkAddress address = NetworkAddress(hostname, NETWORK_DEFAULT_PORT);
  _network_coordinator_client.ConnectSuccess(this->token, game_sock, address);
 
  return false;
}
 
void ClientNetworkTurnSocketHandler::Connect()
{
  Debug(net, 9, "Turn::Connect()");
 
  this->connect_started = true;
  this->connecter = TCPConnecter::Create<NetworkTurnConnecter>(this, this->connection_string);
}
 
/* static */ std::unique_ptr<ClientNetworkTurnSocketHandler> ClientNetworkTurnSocketHandler::Turn(const std::string &token, uint8_t tracking_number, const std::string &ticket, const std::string &connection_string)
{
  auto turn_handler = std::make_unique<ClientNetworkTurnSocketHandler>(token, tracking_number, connection_string);
 
  auto p = std::make_unique<Packet>(turn_handler.get(), PACKET_TURN_SERCLI_CONNECT);
  p->Send_uint8(NETWORK_COORDINATOR_VERSION);
  p->Send_string(ticket);
 
  turn_handler->SendPacket(std::move(p));
 
  return turn_handler;
}
 
void ClientNetworkTurnSocketHandler::ConnectFailure()
{
  _network_coordinator_client.ConnectFailure(this->token, this->tracking_number);
}
 
NetworkRecvStatus ClientNetworkTurnSocketHandler::CloseConnection(bool error)
{
  NetworkTurnSocketHandler::CloseConnection(error);
 
  /* Also make sure any pending connecter is killed ASAP. */
  if (this->connecter != nullptr) {
    this->connecter->Kill();
    this->connecter = nullptr;
  }
 
  return NETWORK_RECV_STATUS_OKAY;
}
 
ClientNetworkTurnSocketHandler::~ClientNetworkTurnSocketHandler()
{
  if (this->connecter != nullptr) {
    this->connecter->Kill();
    this->connecter = nullptr;
  }
}
 
void ClientNetworkTurnSocketHandler::SendReceive()
{
  if (this->sock == INVALID_SOCKET) return;
 
  if (this->CanSendReceive()) {
    this->ReceivePackets();
  }
 
  this->SendPackets();
}