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