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 "exceptions.h"
23 #include "util/serialize.h"
25 NetworkPacket::NetworkPacket(u8 *data, u32 datasize, u16 peer_id):
26 m_read_offset(0), m_peer_id(peer_id)
29 m_datasize = datasize - 2;
31 // split command and datas
32 m_command = readU16(&data[0]);
33 m_data = std::vector<u8>(&data[2], &data[2 + m_datasize]);
36 NetworkPacket::NetworkPacket(u16 command, u32 datasize, u16 peer_id):
37 m_datasize(datasize), m_read_offset(0), m_command(command), m_peer_id(peer_id)
39 m_data.resize(m_datasize);
42 NetworkPacket::NetworkPacket(u16 command, u32 datasize):
43 m_datasize(datasize), m_read_offset(0), m_command(command), m_peer_id(0)
45 m_data.resize(m_datasize);
48 NetworkPacket::~NetworkPacket()
53 char* NetworkPacket::getString(u32 from_offset)
55 if (from_offset >= m_datasize)
56 throw SerializationError("Malformed packet read");
58 return (char*)&m_data[from_offset];
61 void NetworkPacket::putRawString(const char* src, u32 len)
63 if (m_read_offset + len * sizeof(char) >= m_datasize) {
64 m_datasize += len * sizeof(char);
65 m_data.resize(m_datasize);
68 memcpy(&m_data[m_read_offset], src, len);
72 NetworkPacket& NetworkPacket::operator>>(std::string& dst)
74 u16 strLen = readU16(&m_data[m_read_offset]);
75 m_read_offset += sizeof(u16);
84 dst.append((char*)&m_data[m_read_offset], strLen);
86 m_read_offset += strLen * sizeof(char);
90 NetworkPacket& NetworkPacket::operator<<(std::string src)
92 u16 msgsize = src.size();
93 if (msgsize > 0xFFFF) {
99 if (m_read_offset + msgsize * sizeof(char) >= m_datasize) {
100 m_datasize += msgsize * sizeof(char);
101 m_data.resize(m_datasize);
104 memcpy(&m_data[m_read_offset], src.c_str(), msgsize);
105 m_read_offset += msgsize;
110 void NetworkPacket::putLongString(std::string src)
112 u32 msgsize = src.size();
113 if (msgsize > 0xFFFFFFFF) {
114 msgsize = 0xFFFFFFFF;
119 if (m_read_offset + msgsize * sizeof(char) >= m_datasize) {
120 m_datasize += msgsize * sizeof(char);
121 m_data.resize(m_datasize);
124 memcpy(&m_data[m_read_offset], src.c_str(), msgsize);
125 m_read_offset += msgsize;
128 NetworkPacket& NetworkPacket::operator>>(std::wstring& dst)
130 u16 strLen = readU16(&m_data[m_read_offset]);
131 m_read_offset += sizeof(u16);
140 for(u16 i=0; i<strLen; i++) {
141 wchar_t c16 = readU16(&m_data[m_read_offset]);
143 m_read_offset += sizeof(u16);
149 NetworkPacket& NetworkPacket::operator<<(std::wstring src)
151 u16 msgsize = src.size();
152 if (msgsize > 0xFFFF) {
159 for (u16 i=0; i<msgsize; i++) {
160 *this << (u16) src[i];
166 std::string NetworkPacket::readLongString()
168 u32 strLen = readU32(&m_data[m_read_offset]);
169 m_read_offset += sizeof(u32);
178 dst.append((char*)&m_data[m_read_offset], strLen);
180 m_read_offset += strLen*sizeof(char);
185 NetworkPacket& NetworkPacket::operator>>(char& dst)
187 if (m_read_offset >= m_datasize)
188 throw SerializationError("Malformed packet read");
190 dst = readU8(&m_data[m_read_offset]);
196 char NetworkPacket::getChar(u32 offset)
198 if (offset >= m_datasize)
199 throw SerializationError("Malformed packet read");
201 return readU8(&m_data[offset]);
204 NetworkPacket& NetworkPacket::operator<<(char src)
208 writeU8(&m_data[m_read_offset], src);
214 NetworkPacket& NetworkPacket::operator<<(u8 src)
218 writeU8(&m_data[m_read_offset], src);
224 NetworkPacket& NetworkPacket::operator<<(bool src)
228 writeU8(&m_data[m_read_offset], src);
234 NetworkPacket& NetworkPacket::operator<<(u16 src)
236 checkDataSize<u16>();
238 writeU16(&m_data[m_read_offset], src);
244 NetworkPacket& NetworkPacket::operator<<(u32 src)
246 checkDataSize<u32>();
248 writeU32(&m_data[m_read_offset], src);
254 NetworkPacket& NetworkPacket::operator<<(u64 src)
256 checkDataSize<u64>();
258 writeU64(&m_data[m_read_offset], src);
264 NetworkPacket& NetworkPacket::operator<<(float src)
266 checkDataSize<float>();
268 writeF1000(&m_data[m_read_offset], src);
274 NetworkPacket& NetworkPacket::operator>>(bool& dst)
276 if (m_read_offset >= m_datasize)
277 throw SerializationError("Malformed packet read");
279 dst = readU8(&m_data[m_read_offset]);
285 NetworkPacket& NetworkPacket::operator>>(u8& dst)
287 if (m_read_offset >= m_datasize)
288 throw SerializationError("Malformed packet read");
290 dst = readU8(&m_data[m_read_offset]);
296 u8 NetworkPacket::getU8(u32 offset)
298 if (offset >= m_datasize)
299 throw SerializationError("Malformed packet read");
301 return readU8(&m_data[offset]);
304 u8* NetworkPacket::getU8Ptr(u32 from_offset)
306 if (m_datasize == 0) {
310 if (from_offset >= m_datasize)
311 throw SerializationError("Malformed packet read");
313 return (u8*)&m_data[from_offset];
316 NetworkPacket& NetworkPacket::operator>>(u16& dst)
318 if (m_read_offset >= m_datasize)
319 throw SerializationError("Malformed packet read");
321 dst = readU16(&m_data[m_read_offset]);
327 u16 NetworkPacket::getU16(u32 from_offset)
329 if (from_offset >= m_datasize)
330 throw SerializationError("Malformed packet read");
332 return readU16(&m_data[from_offset]);
335 NetworkPacket& NetworkPacket::operator>>(u32& dst)
337 if (m_read_offset >= m_datasize)
338 throw SerializationError("Malformed packet read");
340 dst = readU32(&m_data[m_read_offset]);
346 NetworkPacket& NetworkPacket::operator>>(u64& dst)
348 if (m_read_offset >= m_datasize)
349 throw SerializationError("Malformed packet read");
351 dst = readU64(&m_data[m_read_offset]);
357 NetworkPacket& NetworkPacket::operator>>(float& dst)
359 if (m_read_offset >= m_datasize)
360 throw SerializationError("Malformed packet read");
362 dst = readF1000(&m_data[m_read_offset]);
368 NetworkPacket& NetworkPacket::operator>>(v2f& dst)
370 if (m_read_offset >= m_datasize)
371 throw SerializationError("Malformed packet read");
373 dst = readV2F1000(&m_data[m_read_offset]);
379 NetworkPacket& NetworkPacket::operator>>(v3f& dst)
381 if (m_read_offset >= m_datasize)
382 throw SerializationError("Malformed packet read");
384 dst = readV3F1000(&m_data[m_read_offset]);
390 NetworkPacket& NetworkPacket::operator>>(s16& dst)
392 if (m_read_offset >= m_datasize)
393 throw SerializationError("Malformed packet read");
395 dst = readS16(&m_data[m_read_offset]);
401 NetworkPacket& NetworkPacket::operator<<(s16 src)
407 NetworkPacket& NetworkPacket::operator>>(s32& dst)
409 if (m_read_offset >= m_datasize)
410 throw SerializationError("Malformed packet read");
412 dst = readS32(&m_data[m_read_offset]);
418 NetworkPacket& NetworkPacket::operator<<(s32 src)
424 NetworkPacket& NetworkPacket::operator>>(v3s16& dst)
426 if (m_read_offset >= m_datasize)
427 throw SerializationError("Malformed packet read");
429 dst = readV3S16(&m_data[m_read_offset]);
435 NetworkPacket& NetworkPacket::operator>>(v2s32& dst)
437 dst = readV2S32(&m_data[m_read_offset]);
443 NetworkPacket& NetworkPacket::operator>>(v3s32& dst)
445 if (m_read_offset >= m_datasize)
446 throw SerializationError("Malformed packet read");
448 dst = readV3S32(&m_data[m_read_offset]);
454 NetworkPacket& NetworkPacket::operator<<(v2f src)
456 *this << (float) src.X;
457 *this << (float) src.Y;
461 NetworkPacket& NetworkPacket::operator<<(v3f src)
463 *this << (float) src.X;
464 *this << (float) src.Y;
465 *this << (float) src.Z;
469 NetworkPacket& NetworkPacket::operator<<(v3s16 src)
471 *this << (s16) src.X;
472 *this << (s16) src.Y;
473 *this << (s16) src.Z;
477 NetworkPacket& NetworkPacket::operator<<(v2s32 src)
479 *this << (s32) src.X;
480 *this << (s32) src.Y;
484 NetworkPacket& NetworkPacket::operator<<(v3s32 src)
486 *this << (s32) src.X;
487 *this << (s32) src.Y;
488 *this << (s32) src.Z;
492 NetworkPacket& NetworkPacket::operator>>(video::SColor& dst)
494 if (m_read_offset >= m_datasize)
495 throw SerializationError("Malformed packet read");
497 dst = readARGB8(&m_data[m_read_offset]);
503 NetworkPacket& NetworkPacket::operator<<(video::SColor src)
505 checkDataSize<u32>();
507 writeU32(&m_data[m_read_offset], src.color);
513 Buffer<u8> NetworkPacket::oldForgePacket()
515 Buffer<u8> sb(m_datasize + 2);
516 writeU16(&sb[0], m_command);
518 u8* datas = getU8Ptr(0);
521 memcpy(&sb[2], datas, m_datasize);