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 #pragma once
  21
  22 #include <ctime>
  23 #include "util/pointer.h"
  24 #include "util/numeric.h"
  25 #include "networkprotocol.h"
  26 #include <SColor.h>
  27
  28 class NetworkPacket
  29 {
  30
  31 public:
  32         NetworkPacket(u16 command, u32 datasize, session_t peer_id);
  33         NetworkPacket(u16 command, u32 datasize);
  34         NetworkPacket() = default;
  35
  36         ~NetworkPacket();
  37
  38         void putRawPacket(u8 *data, u32 datasize, session_t peer_id);
  39
  40         // Getters
  41         u32 getSize() const { return m_datasize; }
  42         session_t getPeerId() const { return m_peer_id; }
  43         u16 getCommand() { return m_command; }
  44         const u32 getRemainingBytes() const { return m_datasize - m_read_offset; }
  45         const char *getRemainingString() { return getString(m_read_offset); }
  46
  47         // Returns a c-string without copying.
  48         // A better name for this would be getRawString()
  49         const char *getString(u32 from_offset);
  50         // major difference to putCString(): doesn't write len into the buffer
  51         void putRawString(const char *src, u32 len);
  52         void putRawString(const std::string &src)
  53         {
  54                 putRawString(src.c_str(), src.size());
  55         }
  56
  57         NetworkPacket &operator>>(std::string &dst);
  58         NetworkPacket &operator<<(const std::string &src);
  59
  60         void putLongString(const std::string &src);
  61
  62         NetworkPacket &operator>>(std::wstring &dst);
  63         NetworkPacket &operator<<(const std::wstring &src);
  64
  65         std::string readLongString();
  66
  67         char getChar(u32 offset);
  68         NetworkPacket &operator>>(char &dst);
  69         NetworkPacket &operator<<(char src);
  70
  71         NetworkPacket &operator>>(bool &dst);
  72         NetworkPacket &operator<<(bool src);
  73
  74         u8 getU8(u32 offset);
  75
  76         NetworkPacket &operator>>(u8 &dst);
  77         NetworkPacket &operator<<(u8 src);
  78
  79         u8 *getU8Ptr(u32 offset);
  80
  81         u16 getU16(u32 from_offset);
  82         NetworkPacket &operator>>(u16 &dst);
  83         NetworkPacket &operator<<(u16 src);
  84
  85         NetworkPacket &operator>>(u32 &dst);
  86         NetworkPacket &operator<<(u32 src);
  87
  88         NetworkPacket &operator>>(u64 &dst);
  89         NetworkPacket &operator<<(u64 src);
  90
  91         NetworkPacket &operator>>(std::time_t &dst);
  92         NetworkPacket &operator<<(std::time_t src);
  93
  94         NetworkPacket &operator>>(float &dst);
  95         NetworkPacket &operator<<(float src);
  96
  97         NetworkPacket &operator>>(v2f &dst);
  98         NetworkPacket &operator<<(v2f src);
  99
 100         NetworkPacket &operator>>(v3f &dst);
 101         NetworkPacket &operator<<(v3f src);
 102
 103         NetworkPacket &operator>>(s16 &dst);
 104         NetworkPacket &operator<<(s16 src);
 105
 106         NetworkPacket &operator>>(s32 &dst);
 107         NetworkPacket &operator<<(s32 src);
 108
 109         NetworkPacket &operator>>(v2s32 &dst);
 110         NetworkPacket &operator<<(v2s32 src);
 111
 112         NetworkPacket &operator>>(v3s16 &dst);
 113         NetworkPacket &operator<<(v3s16 src);
 114
 115         NetworkPacket &operator>>(v3s32 &dst);
 116         NetworkPacket &operator<<(v3s32 src);
 117
 118         NetworkPacket &operator>>(video::SColor &dst);
 119         NetworkPacket &operator<<(video::SColor src);
 120
 121         // Temp, we remove SharedBuffer when migration finished
 122         SharedBuffer<u8> oldForgePacket();
 123
 124 private:
 125         void checkReadOffset(u32 from_offset, u32 field_size);
 126
 127         inline void checkDataSize(u32 field_size)
 128         {
 129                 if (m_read_offset + field_size > m_datasize) {
 130                         m_datasize = m_read_offset + field_size;
 131                         m_data.resize(m_datasize);
 132                 }
 133         }
 134
 135         std::vector<u8> m_data;
 136         u32 m_datasize = 0;
 137         u32 m_read_offset = 0;
 138         u16 m_command = 0;
 139         session_t m_peer_id = 0;
 140 };