network_server.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_SERVER_H
#define NETWORK_SERVER_H
 
#include "network_internal.h"
#include "core/tcp_listen.h"
 
class ServerNetworkGameSocketHandler;
typedef ServerNetworkGameSocketHandler NetworkClientSocket;
typedef Pool<NetworkClientSocket, ClientIndex, 8, MAX_CLIENT_SLOTS, PT_NCLIENT> NetworkClientSocketPool;
extern NetworkClientSocketPool _networkclientsocket_pool;
 
class ServerNetworkGameSocketHandler : public NetworkClientSocketPool::PoolItem<&_networkclientsocket_pool>, public NetworkGameSocketHandler, public TCPListenHandler<ServerNetworkGameSocketHandler, PACKET_SERVER_FULL, PACKET_SERVER_BANNED> {
protected:
  std::unique_ptr<class NetworkAuthenticationServerHandler> authentication_handler; 
  std::string peer_public_key; 
 
  NetworkRecvStatus Receive_CLIENT_JOIN(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_IDENTIFY(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_GAME_INFO(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_AUTH_RESPONSE(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_GETMAP(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_MAP_OK(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_ACK(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_COMMAND(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_CHAT(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_SET_NAME(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_QUIT(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_ERROR(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_RCON(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_NEWGRFS_CHECKED(Packet &p) override;
  NetworkRecvStatus Receive_CLIENT_MOVE(Packet &p) override;
 
  NetworkRecvStatus SendGameInfo();
  NetworkRecvStatus SendNewGRFCheck();
  NetworkRecvStatus SendWelcome();
  NetworkRecvStatus SendAuthRequest();
  NetworkRecvStatus SendEnableEncryption();
 
public:
  enum ClientStatus {
    STATUS_INACTIVE,      
    STATUS_AUTH_GAME,     
    STATUS_IDENTIFY,      
    STATUS_NEWGRFS_CHECK, 
    STATUS_AUTHORIZED,    
    STATUS_MAP_WAIT,      
    STATUS_MAP,           
    STATUS_DONE_MAP,      
    STATUS_PRE_ACTIVE,    
    STATUS_ACTIVE,        
    STATUS_END,           
  };
 
  uint8_t lag_test;               
  uint8_t last_token;             
  uint32_t last_token_frame;     
  ClientStatus status;         
  CommandQueue outgoing_queue; 
  size_t receive_limit;        
 
  std::shared_ptr<struct PacketWriter> savegame; 
  NetworkAddress client_address; 
 
  ServerNetworkGameSocketHandler(SOCKET s);
  ~ServerNetworkGameSocketHandler();
 
  std::unique_ptr<Packet> ReceivePacket() override;
  NetworkRecvStatus CloseConnection(NetworkRecvStatus status) override;
  std::string GetClientName() const;
 
  void CheckNextClientToSendMap(NetworkClientSocket *ignore_cs = nullptr);
 
  NetworkRecvStatus SendWait();
  NetworkRecvStatus SendMap();
  NetworkRecvStatus SendErrorQuit(ClientID client_id, NetworkErrorCode errorno);
  NetworkRecvStatus SendQuit(ClientID client_id);
  NetworkRecvStatus SendShutdown();
  NetworkRecvStatus SendNewGame();
  NetworkRecvStatus SendRConResult(uint16_t colour, const std::string &command);
  NetworkRecvStatus SendMove(ClientID client_id, CompanyID company_id);
 
  NetworkRecvStatus SendClientInfo(NetworkClientInfo *ci);
  NetworkRecvStatus SendError(NetworkErrorCode error, const std::string &reason = {});
  NetworkRecvStatus SendChat(NetworkAction action, ClientID client_id, bool self_send, const std::string &msg, int64_t data);
  NetworkRecvStatus SendExternalChat(const std::string &source, TextColour colour, const std::string &user, const std::string &msg);
  NetworkRecvStatus SendJoin(ClientID client_id);
  NetworkRecvStatus SendFrame();
  NetworkRecvStatus SendSync();
  NetworkRecvStatus SendCommand(const CommandPacket &cp);
  NetworkRecvStatus SendConfigUpdate();
 
  static void Send();
  static void AcceptConnection(SOCKET s, const NetworkAddress &address);
  static bool AllowConnection();
 
  static const char *GetName()
  {
    return "server";
  }
 
  const std::string &GetClientIP();
  std::string_view GetPeerPublicKey() const { return this->peer_public_key; }
 
  static ServerNetworkGameSocketHandler *GetByClientID(ClientID client_id);
};
 
void NetworkServer_Tick(bool send_frame);
void ChangeNetworkRestartTime(bool reset);
 
#endif /* NETWORK_SERVER_H */