network_coordinator.h

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/>.
 */
 
#ifndef NETWORK_COORDINATOR_H
#define NETWORK_COORDINATOR_H
 
#include "core/tcp_coordinator.h"
#include "network_stun.h"
#include "network_turn.h"
 
class ClientNetworkCoordinatorSocketHandler : public NetworkCoordinatorSocketHandler {
private:
  std::chrono::steady_clock::time_point next_update; 
  std::map<std::string, std::pair<std::string, TCPServerConnecter *>> connecter; 
  std::map<std::string, TCPServerConnecter *> connecter_pre; 
  std::map<std::string, std::map<int, std::unique_ptr<ClientNetworkStunSocketHandler>>> stun_handlers; 
  std::map<std::string, std::unique_ptr<ClientNetworkTurnSocketHandler>> turn_handlers; 
  std::shared_ptr<TCPConnecter> game_connecter{}; 
 
  uint32_t newgrf_lookup_table_cursor = 0; 
  GameInfoNewGRFLookupTable newgrf_lookup_table; 
 
protected:
  bool Receive_GC_ERROR(Packet &p) override;
  bool Receive_GC_REGISTER_ACK(Packet &p) override;
  bool Receive_GC_LISTING(Packet &p) override;
  bool Receive_GC_CONNECTING(Packet &p) override;
  bool Receive_GC_CONNECT_FAILED(Packet &p) override;
  bool Receive_GC_DIRECT_CONNECT(Packet &p) override;
  bool Receive_GC_STUN_REQUEST(Packet &p) override;
  bool Receive_GC_STUN_CONNECT(Packet &p) override;
  bool Receive_GC_NEWGRF_LOOKUP(Packet &p) override;
  bool Receive_GC_TURN_CONNECT(Packet &p) override;
 
public:
  static constexpr std::chrono::seconds IDLE_TIMEOUT = std::chrono::seconds(60);
 
  std::chrono::steady_clock::time_point last_activity;  
  bool connecting; 
 
  ClientNetworkCoordinatorSocketHandler() : connecting(false) {}
 
  NetworkRecvStatus CloseConnection(bool error = true) override;
  void SendReceive();
 
  void ConnectFailure(const std::string &token, uint8_t tracking_number);
  void ConnectSuccess(const std::string &token, SOCKET sock, NetworkAddress &address);
  void StunResult(const std::string &token, uint8_t family, bool result);
 
  void Connect();
  void CloseToken(const std::string &token);
  void CloseAllConnections();
  void CloseStunHandler(const std::string &token, uint8_t family = AF_UNSPEC);
  void CloseTurnHandler(const std::string &token);
 
  void Register();
  void SendServerUpdate();
  void GetListing();
 
  void ConnectToServer(const std::string &invite_code, TCPServerConnecter *connecter);
  void StartTurnConnection(std::string &token);
};
 
extern ClientNetworkCoordinatorSocketHandler _network_coordinator_client;
 
#endif /* NETWORK_COORDINATOR_H */