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
  44                 // Returns a c-string without copying.
  45                 // A better name for this would be getRawString()
  46                 char* getString(u32 from_offset);
  47                 // major difference to putCString(): doesn't write len into the buffer
  48                 void putRawString(const char* src, u32 len);
  49
  50                 NetworkPacket& operator>>(std::string& dst);
  51                 NetworkPacket& operator<<(std::string src);
  52
  53                 void putLongString(std::string src);
  54
  55                 NetworkPacket& operator>>(std::wstring& dst);
  56                 NetworkPacket& operator<<(std::wstring src);
  57
  58                 std::string readLongString();
  59
  60                 char getChar(u32 offset);
  61                 NetworkPacket& operator>>(char& dst);
  62                 NetworkPacket& operator<<(char src);
  63
  64                 NetworkPacket& operator>>(bool& dst);
  65                 NetworkPacket& operator<<(bool src);
  66
  67                 u8 getU8(u32 offset);
  68
  69                 NetworkPacket& operator>>(u8& dst);
  70                 NetworkPacket& operator<<(u8 src);
  71
  72                 u8* getU8Ptr(u32 offset);
  73
  74                 u16 getU16(u32 from_offset);
  75                 NetworkPacket& operator>>(u16& dst);
  76                 NetworkPacket& operator<<(u16 src);
  77
  78                 NetworkPacket& operator>>(u32& dst);
  79                 NetworkPacket& operator<<(u32 src);
  80
  81                 NetworkPacket& operator>>(u64& dst);
  82                 NetworkPacket& operator<<(u64 src);
  83
  84                 NetworkPacket& operator>>(float& dst);
  85                 NetworkPacket& operator<<(float src);
  86
  87                 NetworkPacket& operator>>(v2f& dst);
  88                 NetworkPacket& operator<<(v2f src);
  89
  90                 NetworkPacket& operator>>(v3f& dst);
  91                 NetworkPacket& operator<<(v3f src);
  92
  93                 NetworkPacket& operator>>(s16& dst);
  94                 NetworkPacket& operator<<(s16 src);
  95
  96                 NetworkPacket& operator>>(s32& dst);
  97                 NetworkPacket& operator<<(s32 src);
  98
  99                 NetworkPacket& operator>>(v2s32& dst);
 100                 NetworkPacket& operator<<(v2s32 src);
 101
 102                 NetworkPacket& operator>>(v3s16& dst);
 103                 NetworkPacket& operator<<(v3s16 src);
 104
 105                 NetworkPacket& operator>>(v3s32& dst);
 106                 NetworkPacket& operator<<(v3s32 src);
 107
 108                 NetworkPacket& operator>>(video::SColor& dst);
 109                 NetworkPacket& operator<<(video::SColor src);
 110
 111                 // Temp, we remove SharedBuffer when migration finished
 112                 Buffer<u8> oldForgePacket();
 113 private:
 114                 void checkReadOffset(u32 from_offset);
 115
 116                 template<typename T> void checkDataSize()
 117                 {
 118                         if (m_read_offset + sizeof(T) > m_datasize) {
 119                                 m_datasize = m_read_offset + sizeof(T);
 120                                 m_data.resize(m_datasize);
 121                         }
 122                 }
 123
 124                 template<typename T> void incrOffset()
 125                 {
 126                         m_read_offset += sizeof(T);
 127                 }
 128
 129                 std::vector<u8> m_data;
 130                 u32 m_datasize;
 131                 u32 m_read_offset;
 132                 u16 m_command;
 133                 u16 m_peer_id;
 134 };
 135
 136 #endif