3 Copyright (C) 2015 nerzhul, Loic Blot <loic.blot@unix-experience.fr>
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.
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.
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.
20 #include "networkpacket.h"
22 #include "networkexceptions.h"
23 #include "util/serialize.h"
24 #include "networkprotocol.h"
26 NetworkPacket::NetworkPacket(u16 command, u32 datasize, session_t peer_id):
27 m_datasize(datasize), m_command(command), m_peer_id(peer_id)
29 m_data.resize(m_datasize);
32 NetworkPacket::NetworkPacket(u16 command, u32 datasize):
33 m_datasize(datasize), m_command(command)
35 m_data.resize(m_datasize);
38 NetworkPacket::~NetworkPacket()
43 void NetworkPacket::checkReadOffset(u32 from_offset, u32 field_size)
45 if (from_offset + field_size > m_datasize) {
47 ss << "Reading outside packet (offset: " <<
48 from_offset << ", packet size: " << getSize() << ")";
49 throw PacketError(ss.str());
53 void NetworkPacket::putRawPacket(u8 *data, u32 datasize, session_t peer_id)
55 // If a m_command is already set, we are rewriting on same packet
56 // This is not permitted
57 assert(m_command == 0);
59 m_datasize = datasize - 2;
62 m_data.resize(m_datasize);
64 // split command and datas
65 m_command = readU16(&data[0]);
66 memcpy(m_data.data(), &data[2], m_datasize);
69 const char* NetworkPacket::getString(u32 from_offset)
71 checkReadOffset(from_offset, 0);
73 return (char*)&m_data[from_offset];
76 void NetworkPacket::putRawString(const char* src, u32 len)
78 if (m_read_offset + len > m_datasize) {
79 m_datasize = m_read_offset + len;
80 m_data.resize(m_datasize);
86 memcpy(&m_data[m_read_offset], src, len);
90 NetworkPacket& NetworkPacket::operator>>(std::string& dst)
92 checkReadOffset(m_read_offset, 2);
93 u16 strLen = readU16(&m_data[m_read_offset]);
102 checkReadOffset(m_read_offset, strLen);
105 dst.append((char*)&m_data[m_read_offset], strLen);
107 m_read_offset += strLen;
111 NetworkPacket& NetworkPacket::operator<<(const std::string &src)
113 u16 msgsize = src.size();
114 if (msgsize > STRING_MAX_LEN) {
115 throw PacketError("String too long");
120 putRawString(src.c_str(), (u32)msgsize);
125 void NetworkPacket::putLongString(const std::string &src)
127 u32 msgsize = src.size();
128 if (msgsize > LONG_STRING_MAX_LEN) {
129 throw PacketError("String too long");
134 putRawString(src.c_str(), msgsize);
137 NetworkPacket& NetworkPacket::operator>>(std::wstring& dst)
139 checkReadOffset(m_read_offset, 2);
140 u16 strLen = readU16(&m_data[m_read_offset]);
149 checkReadOffset(m_read_offset, strLen * 2);
152 for(u16 i=0; i<strLen; i++) {
153 wchar_t c16 = readU16(&m_data[m_read_offset]);
155 m_read_offset += sizeof(u16);
161 NetworkPacket& NetworkPacket::operator<<(const std::wstring &src)
163 u16 msgsize = src.size();
164 if (msgsize > WIDE_STRING_MAX_LEN) {
165 throw PacketError("String too long");
171 for (u16 i=0; i<msgsize; i++) {
172 *this << (u16) src[i];
178 std::string NetworkPacket::readLongString()
180 checkReadOffset(m_read_offset, 4);
181 u32 strLen = readU32(&m_data[m_read_offset]);
188 if (strLen > LONG_STRING_MAX_LEN) {
189 throw PacketError("String too long");
192 checkReadOffset(m_read_offset, strLen);
197 dst.append((char*)&m_data[m_read_offset], strLen);
199 m_read_offset += strLen;
204 NetworkPacket& NetworkPacket::operator>>(char& dst)
206 checkReadOffset(m_read_offset, 1);
208 dst = readU8(&m_data[m_read_offset]);
214 char NetworkPacket::getChar(u32 offset)
216 checkReadOffset(offset, 1);
218 return readU8(&m_data[offset]);
221 NetworkPacket& NetworkPacket::operator<<(char src)
225 writeU8(&m_data[m_read_offset], src);
231 NetworkPacket& NetworkPacket::operator<<(u8 src)
235 writeU8(&m_data[m_read_offset], src);
241 NetworkPacket& NetworkPacket::operator<<(bool src)
245 writeU8(&m_data[m_read_offset], src);
251 NetworkPacket& NetworkPacket::operator<<(u16 src)
255 writeU16(&m_data[m_read_offset], src);
261 NetworkPacket& NetworkPacket::operator<<(u32 src)
265 writeU32(&m_data[m_read_offset], src);
271 NetworkPacket& NetworkPacket::operator<<(u64 src)
275 writeU64(&m_data[m_read_offset], src);
281 NetworkPacket& NetworkPacket::operator<<(std::time_t src)
287 NetworkPacket& NetworkPacket::operator<<(float src)
291 writeF32(&m_data[m_read_offset], src);
297 NetworkPacket& NetworkPacket::operator>>(bool& dst)
299 checkReadOffset(m_read_offset, 1);
301 dst = readU8(&m_data[m_read_offset]);
307 NetworkPacket& NetworkPacket::operator>>(u8& dst)
309 checkReadOffset(m_read_offset, 1);
311 dst = readU8(&m_data[m_read_offset]);
317 u8 NetworkPacket::getU8(u32 offset)
319 checkReadOffset(offset, 1);
321 return readU8(&m_data[offset]);
324 u8* NetworkPacket::getU8Ptr(u32 from_offset)
326 if (m_datasize == 0) {
330 checkReadOffset(from_offset, 1);
332 return (u8*)&m_data[from_offset];
335 NetworkPacket& NetworkPacket::operator>>(u16& dst)
337 checkReadOffset(m_read_offset, 2);
339 dst = readU16(&m_data[m_read_offset]);
345 u16 NetworkPacket::getU16(u32 from_offset)
347 checkReadOffset(from_offset, 2);
349 return readU16(&m_data[from_offset]);
352 NetworkPacket& NetworkPacket::operator>>(u32& dst)
354 checkReadOffset(m_read_offset, 4);
356 dst = readU32(&m_data[m_read_offset]);
362 NetworkPacket& NetworkPacket::operator>>(u64& dst)
364 checkReadOffset(m_read_offset, 8);
366 dst = readU64(&m_data[m_read_offset]);
372 NetworkPacket& NetworkPacket::operator>>(std::time_t& dst)
374 checkReadOffset(m_read_offset, 8);
376 dst = readU64(&m_data[m_read_offset]);
382 NetworkPacket& NetworkPacket::operator>>(float& dst)
384 checkReadOffset(m_read_offset, 4);
386 dst = readF32(&m_data[m_read_offset]);
392 NetworkPacket& NetworkPacket::operator>>(v2f& dst)
394 checkReadOffset(m_read_offset, 8);
396 dst = readV2F32(&m_data[m_read_offset]);
402 NetworkPacket& NetworkPacket::operator>>(v3f& dst)
404 checkReadOffset(m_read_offset, 12);
406 dst = readV3F32(&m_data[m_read_offset]);
412 NetworkPacket& NetworkPacket::operator>>(s16& dst)
414 checkReadOffset(m_read_offset, 2);
416 dst = readS16(&m_data[m_read_offset]);
422 NetworkPacket& NetworkPacket::operator<<(s16 src)
428 NetworkPacket& NetworkPacket::operator>>(s32& dst)
430 checkReadOffset(m_read_offset, 4);
432 dst = readS32(&m_data[m_read_offset]);
438 NetworkPacket& NetworkPacket::operator<<(s32 src)
444 NetworkPacket& NetworkPacket::operator>>(v3s16& dst)
446 checkReadOffset(m_read_offset, 6);
448 dst = readV3S16(&m_data[m_read_offset]);
454 NetworkPacket& NetworkPacket::operator>>(v2s32& dst)
456 checkReadOffset(m_read_offset, 8);
458 dst = readV2S32(&m_data[m_read_offset]);
464 NetworkPacket& NetworkPacket::operator>>(v3s32& dst)
466 checkReadOffset(m_read_offset, 12);
468 dst = readV3S32(&m_data[m_read_offset]);
474 NetworkPacket& NetworkPacket::operator<<(v2f src)
476 *this << (float) src.X;
477 *this << (float) src.Y;
481 NetworkPacket& NetworkPacket::operator<<(v3f src)
483 *this << (float) src.X;
484 *this << (float) src.Y;
485 *this << (float) src.Z;
489 NetworkPacket& NetworkPacket::operator<<(v3s16 src)
491 *this << (s16) src.X;
492 *this << (s16) src.Y;
493 *this << (s16) src.Z;
497 NetworkPacket& NetworkPacket::operator<<(v2s32 src)
499 *this << (s32) src.X;
500 *this << (s32) src.Y;
504 NetworkPacket& NetworkPacket::operator<<(v3s32 src)
506 *this << (s32) src.X;
507 *this << (s32) src.Y;
508 *this << (s32) src.Z;
512 NetworkPacket& NetworkPacket::operator>>(video::SColor& dst)
514 checkReadOffset(m_read_offset, 4);
516 dst = readARGB8(&m_data[m_read_offset]);
522 NetworkPacket& NetworkPacket::operator<<(video::SColor src)
526 writeU32(&m_data[m_read_offset], src.color);
532 SharedBuffer<u8> NetworkPacket::oldForgePacket()
534 SharedBuffer<u8> sb(m_datasize + 2);
535 writeU16(&sb[0], m_command);
537 u8* datas = getU8Ptr(0);
540 memcpy(&sb[2], datas, m_datasize);