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 <https://www.gnu.org/licenses/old-licenses/gpl-2.0>.
 */

 
#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 *>, std::less<>> connecter; 
  std::map<std::string, TCPServerConnecter *, std::less<>> connecter_pre; 
  std::map<std::string, std::map<int, std::unique_ptr<ClientNetworkStunSocketHandler>>, std::less<>> stun_handlers; 
  std::map<std::string, std::unique_ptr<ClientNetworkTurnSocketHandler>, std::less<>> turn_handlers; 
  std::shared_ptr<TCPConnecter> game_connecter{}; 
 
  uint32_t newgrf_lookup_table_cursor = 0; 
  GameInfoNewGRFLookupTable newgrf_lookup_table; 
 
protected:
  bool ReceiveGameCoordinatorError(Packet &p) override;
  bool ReceiveGameCoordinatorRegisterAck(Packet &p) override;
  bool ReceiveGameCoordinatorListing(Packet &p) override;
  bool ReceiveGameCoordinatorConnecting(Packet &p) override;
  bool ReceiveGameCoordinatorConnectFailed(Packet &p) override;
  bool ReceiveGameCoordinatorDirectConnect(Packet &p) override;
  bool ReceiveGameCoordinatorStunRequest(Packet &p) override;
  bool ReceiveGameCoordinatorStunConnect(Packet &p) override;
  bool ReceiveGameCoordinatorNewGRFLookup(Packet &p) override;
  bool ReceiveGameCoordinatorTurnConnect(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(std::string_view token, uint8_t tracking_number);
  void ConnectSuccess(std::string_view token, SOCKET sock, NetworkAddress &address);
  void StunResult(std::string_view token, uint8_t family, bool result);
 
  void Connect();
  void CloseToken(std::string_view token);
  void CloseAllConnections();
  void CloseStunHandler(std::string_view token, uint8_t family = AF_UNSPEC);
  void CloseTurnHandler(std::string_view token);
 
  void Register();
  void SendServerUpdate();
  void GetListing();
 
  void ConnectToServer(std::string_view invite_code, TCPServerConnecter *connecter);
  void StartTurnConnection(std::string_view token);
};
 
extern ClientNetworkCoordinatorSocketHandler _network_coordinator_client;
 
#endif /* NETWORK_COORDINATOR_H */