src/network/networkpacket.h

   1 /*
   2 Minetest
   3 Copyright (C) 2015 nerzhul, Loic Blot <loic.blot@unix-experience.fr>
   4
   5 This program is free software; you can redistribute it and/or modify
   6 it under the terms of the GNU Lesser General Public License as published by
   7 the Free Software Foundation; either version 2.1 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU Lesser General Public License for more details.
  14
  15 You should have received a copy of the GNU Lesser General Public License along
  16 with this program; if not, write to the Free Software Foundation, Inc.,
  17 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 */
  19
  20 #ifndef NETWORKPACKET_HEADER
  21 #define NETWORKPACKET_HEADER
  22
  23 #include "util/pointer.h"
  24 #include "util/numeric.h"
  25 #include "networkprotocol.h"
  26
  27 class NetworkPacket
  28 {
  29
  30 public:
  31                 NetworkPacket(u16 command, u32 datasize, u16 peer_id);
  32                 NetworkPacket(u16 command, u32 datasize);
  33                 NetworkPacket(): m_datasize(0), m_read_offset(0), m_command(0),
  34                                 m_peer_id(0) {}
  35                 ~NetworkPacket();
  36
  37                 void putRawPacket(u8 *data, u32 datasize, u16 peer_id);
  38
  39                 // Getters
  40                 u32 getSize() { return m_datasize; }
  41                 u16 getPeerId() { return m_peer_id; }
  42                 u16 getCommand() { return m_command; }
  43                 const u32 getRemainingBytes() const { return m_datasize - m_read_offset; }
  44                 const char* getRemainingString() { return getString(m_read_offset); }
  45
  46                 // Returns a c-string without copying.
  47                 // A better name for this would be getRawString()
  48                 const char* getString(u32 from_offset);
  49                 // major difference to putCString(): doesn't write len into the buffer
  50                 void putRawString(const char* src, u32 len);
  51                 void putRawString(const std::string &src)
  52                         { putRawString(src.c_str(), src.size()); }
  53
  54                 NetworkPacket& operator>>(std::string& dst);
  55                 NetworkPacket& operator<<(std::string src);
  56
  57                 void putLongString(std::string src);
  58
  59                 NetworkPacket& operator>>(std::wstring& dst);
  60                 NetworkPacket& operator<<(std::wstring src);
  61
  62                 std::string readLongString();
  63
  64                 char getChar(u32 offset);
  65                 NetworkPacket& operator>>(char& dst);
  66                 NetworkPacket& operator<<(char src);
  67
  68                 NetworkPacket& operator>>(bool& dst);
  69                 NetworkPacket& operator<<(bool src);
  70
  71                 u8 getU8(u32 offset);
  72
  73                 NetworkPacket& operator>>(u8& dst);
  74                 NetworkPacket& operator<<(u8 src);
  75
  76                 u8* getU8Ptr(u32 offset);
  77
  78                 u16 getU16(u32 from_offset);
  79                 NetworkPacket& operator>>(u16& dst);
  80                 NetworkPacket& operator<<(u16 src);
  81
  82                 NetworkPacket& operator>>(u32& dst);
  83                 NetworkPacket& operator<<(u32 src);
  84
  85                 NetworkPacket& operator>>(u64& dst);
  86                 NetworkPacket& operator<<(u64 src);
  87
  88                 NetworkPacket& operator>>(float& dst);
  89                 NetworkPacket& operator<<(float src);
  90
  91                 NetworkPacket& operator>>(v2f& dst);
  92                 NetworkPacket& operator<<(v2f src);
  93
  94                 NetworkPacket& operator>>(v3f& dst);
  95                 NetworkPacket& operator<<(v3f src);
  96
  97                 NetworkPacket& operator>>(s16& dst);
  98                 NetworkPacket& operator<<(s16 src);
  99
 100                 NetworkPacket& operator>>(s32& dst);
 101                 NetworkPacket& operator<<(s32 src);
 102
 103                 NetworkPacket& operator>>(v2s32& dst);
 104                 NetworkPacket& operator<<(v2s32 src);
 105
 106                 NetworkPacket& operator>>(v3s16& dst);
 107                 NetworkPacket& operator<<(v3s16 src);
 108
 109                 NetworkPacket& operator>>(v3s32& dst);
 110                 NetworkPacket& operator<<(v3s32 src);
 111
 112                 NetworkPacket& operator>>(video::SColor& dst);
 113                 NetworkPacket& operator<<(video::SColor src);
 114
 115                 // Temp, we remove SharedBuffer when migration finished
 116                 Buffer<u8> oldForgePacket();
 117 private:
 118                 void checkReadOffset(u32 from_offset, u32 field_size);
 119
 120                 inline void checkDataSize(u32 field_size)
 121                 {
 122                         if (m_read_offset + field_size > m_datasize) {
 123                                 m_datasize = m_read_offset + field_size;
 124                                 m_data.resize(m_datasize);
 125                         }
 126                 }
 127
 128                 std::vector<u8> m_data;
 129                 u32 m_datasize;
 130                 u32 m_read_offset;
 131                 u16 m_command;
 132                 u16 m_peer_id;
 133 };
 134
 135 #endif