udp.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_CORE_UDP_H
#define NETWORK_CORE_UDP_H
 
#include "address.h"
#include "packet.h"

enum class PacketUDPType : uint8_t {
  ClientFindServer, 
  ServerResponse, 
};

template <> struct IsEnumPacketType<PacketUDPType> {
  static constexpr bool value = true; 
};

class NetworkUDPSocketHandler : public NetworkSocketHandler {
protected:
  NetworkAddressList bind;
  SocketList sockets;
 
  void ReceiveInvalidPacket(PacketUDPType, NetworkAddress &client_addr);

  virtual void ReceiveClientFindServer(Packet &p, NetworkAddress &client_addr);

  virtual void ReceiveServerResponse(Packet &p, NetworkAddress &client_addr);
 
  void HandleUDPPacket(Packet &p, NetworkAddress &client_addr);
public:
  NetworkUDPSocketHandler(NetworkAddressList *bind = nullptr);

  ~NetworkUDPSocketHandler() override { this->CloseSocket(); }
 
  bool Listen();
  void CloseSocket();
 
  void SendPacket(Packet &p, NetworkAddress &recv, bool all = false, bool broadcast = false);
  void ReceivePackets();
};
 
#endif /* NETWORK_CORE_UDP_H */