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(u8 *data, u32 datasize, u16 peer_id);
  32                 NetworkPacket(u16 command, u32 datasize, u16 peer_id);
  33                 NetworkPacket(u16 command, u32 datasize);
  34                 ~NetworkPacket();
  35
  36                 // Getters
  37                 u32 getSize() { return m_datasize; }
  38                 u16 getPeerId() { return m_peer_id; }
  39                 u16 getCommand() { return m_command; }
  40
  41                 // Data extractors
  42                 char* getString(u32 from_offset);
  43                 void putRawString(const char* src, u32 len);
  44
  45                 NetworkPacket& operator>>(std::string& dst);
  46                 NetworkPacket& operator<<(std::string src);
  47
  48                 void putLongString(std::string src);
  49
  50                 NetworkPacket& operator>>(std::wstring& dst);
  51                 NetworkPacket& operator<<(std::wstring src);
  52
  53                 std::string readLongString();
  54
  55                 char getChar(u32 offset);
  56                 NetworkPacket& operator>>(char& dst);
  57                 NetworkPacket& operator<<(char src);
  58
  59                 NetworkPacket& operator>>(bool& dst);
  60                 NetworkPacket& operator<<(bool src);
  61
  62                 u8 getU8(u32 offset);
  63
  64                 NetworkPacket& operator>>(u8& dst);
  65                 NetworkPacket& operator<<(u8 src);
  66
  67                 u8* getU8Ptr(u32 offset);
  68
  69                 u16 getU16(u32 from_offset);
  70                 NetworkPacket& operator>>(u16& dst);
  71                 NetworkPacket& operator<<(u16 src);
  72
  73                 NetworkPacket& operator>>(u32& dst);
  74                 NetworkPacket& operator<<(u32 src);
  75
  76                 NetworkPacket& operator>>(u64& dst);
  77                 NetworkPacket& operator<<(u64 src);
  78
  79                 NetworkPacket& operator>>(float& dst);
  80                 NetworkPacket& operator<<(float src);
  81
  82                 NetworkPacket& operator>>(v2f& dst);
  83                 NetworkPacket& operator<<(v2f src);
  84
  85                 NetworkPacket& operator>>(v3f& dst);
  86                 NetworkPacket& operator<<(v3f src);
  87
  88                 NetworkPacket& operator>>(s16& dst);
  89                 NetworkPacket& operator<<(s16 src);
  90
  91                 NetworkPacket& operator>>(s32& dst);
  92                 NetworkPacket& operator<<(s32 src);
  93
  94                 NetworkPacket& operator>>(v2s32& dst);
  95                 NetworkPacket& operator<<(v2s32 src);
  96
  97                 NetworkPacket& operator>>(v3s16& dst);
  98                 NetworkPacket& operator<<(v3s16 src);
  99
 100                 NetworkPacket& operator>>(v3s32& dst);
 101                 NetworkPacket& operator<<(v3s32 src);
 102
 103                 NetworkPacket& operator>>(video::SColor& dst);
 104                 NetworkPacket& operator<<(video::SColor src);
 105
 106                 // Temp, we remove SharedBuffer when migration finished
 107                 SharedBuffer<u8> oldForgePacket();
 108 private:
 109                 template<typename T> void checkDataSize()
 110                 {
 111                         if (m_read_offset + sizeof(T) > m_datasize) {
 112                                 m_datasize += sizeof(T);
 113                                 m_data.resize(m_datasize);
 114                         }
 115                 }
 116
 117                 template<typename T> void incrOffset()
 118                 {
 119                         m_read_offset += sizeof(T);
 120                 }
 121
 122                 std::vector<u8> m_data;
 123                 u32 m_datasize;
 124                 u32 m_read_offset;
 125                 u16 m_command;
 126                 u16 m_peer_id;
 127 };
 128
 129 #endif