src/net/Packet.hpp

   1 #ifndef BLANK_NET_PACKET_HPP_
   2 #define BLANK_NET_PACKET_HPP_
   3
   4 #include <cstdint>
   5 #include <ostream>
   6 #include <string>
   7 #include <SDL_net.h>
   8
   9
  10 namespace blank {
  11
  12 class Entity;
  13
  14 struct Packet {
  15
  16         static constexpr std::uint32_t TAG = 0xFB1AB1AF;
  17
  18         static const char *Type2String(std::uint8_t) noexcept;
  19
  20         struct TControl {
  21                 std::uint16_t seq;
  22                 std::uint16_t ack;
  23                 std::uint32_t hist;
  24
  25                 // true if this contains an ack for given (remote) seq
  26                 bool Acks(std::uint16_t) const noexcept;
  27                 std::uint16_t AckBegin() const noexcept { return ack; }
  28                 std::uint16_t AckEnd() const noexcept { return ack + std::uint16_t(33); }
  29         };
  30
  31         struct Header {
  32                 std::uint32_t tag;
  33                 TControl ctrl;
  34                 std::uint8_t type;
  35                 std::uint8_t reserved1;
  36                 std::uint8_t reserved2;
  37                 std::uint8_t reserved3;
  38         } header;
  39
  40         static constexpr std::size_t MAX_PAYLOAD_LEN = 500 - sizeof(Header);
  41
  42         std::uint8_t payload[MAX_PAYLOAD_LEN];
  43
  44
  45         void Tag() noexcept { header.tag = TAG; }
  46
  47         void Type(std::uint8_t t) noexcept { header.type = t; }
  48         std::uint8_t Type() const noexcept { return header.type; }
  49         const char *TypeString() const noexcept { return Type2String(Type()); }
  50
  51
  52         struct Payload {
  53                 std::size_t length;
  54                 std::uint8_t *data;
  55
  56                 template<class T>
  57                 void Write(const T &, size_t off) noexcept;
  58                 template<class T>
  59                 void Read(T &, size_t off) const noexcept;
  60
  61                 void WriteString(const std::string &src, std::size_t off, std::size_t maxlen) noexcept;
  62                 void ReadString(std::string &dst, std::size_t off, std::size_t maxlen) const noexcept;
  63         };
  64
  65         struct Ping : public Payload {
  66                 static constexpr std::uint8_t TYPE = 0;
  67                 static constexpr std::size_t MAX_LEN = 0;
  68         };
  69
  70         struct Login : public Payload {
  71                 static constexpr std::uint8_t TYPE = 1;
  72                 static constexpr std::size_t MAX_LEN = 32;
  73
  74                 void WritePlayerName(const std::string &) noexcept;
  75                 void ReadPlayerName(std::string &) const noexcept;
  76         };
  77
  78         struct Join : public Payload {
  79                 static constexpr std::uint8_t TYPE = 2;
  80                 static constexpr std::size_t MAX_LEN = 100;
  81
  82                 void WritePlayer(const Entity &) noexcept;
  83                 void ReadPlayerID(std::uint32_t &) const noexcept;
  84                 void ReadPlayer(Entity &) const noexcept;
  85                 void WriteWorldName(const std::string &) noexcept;
  86                 void ReadWorldName(std::string &) const noexcept;
  87         };
  88
  89         struct Part : public Payload {
  90                 static constexpr std::uint8_t TYPE = 3;
  91                 static constexpr std::size_t MAX_LEN = 0;
  92         };
  93
  94         struct PlayerUpdate : public Payload {
  95                 static constexpr std::uint8_t TYPE = 4;
  96                 static constexpr std::size_t MAX_LEN = 64;
  97
  98                 void WritePlayer(const Entity &) noexcept;
  99                 void ReadPlayer(Entity &) const noexcept;
 100         };
 101
 102
 103         template<class PayloadType>
 104         PayloadType As() {
 105                 PayloadType result;
 106                 result.length = PayloadType::MAX_LEN;
 107                 result.data = &payload[0];
 108                 return result;
 109         }
 110
 111         template<class PayloadType>
 112         static PayloadType As(const UDPpacket &pack) {
 113                 PayloadType result;
 114                 result.length = std::min(pack.len - sizeof(Header), PayloadType::MAX_LEN);
 115                 result.data = pack.data + sizeof(Header);
 116                 return result;
 117         }
 118
 119         template<class PayloadType>
 120         static PayloadType Make(UDPpacket &udp_pack) {
 121                 Packet &pack = *reinterpret_cast<Packet *>(udp_pack.data);
 122                 pack.Tag();
 123                 pack.Type(PayloadType::TYPE);
 124
 125                 udp_pack.len = sizeof(Header) + PayloadType::MAX_LEN;
 126
 127                 PayloadType result;
 128                 result.length = PayloadType::MAX_LEN;
 129                 result.data = pack.payload;
 130                 return result;
 131         }
 132
 133 };
 134
 135 }
 136
 137 #endif