src/network/networkpacket.cpp

   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 #include "networkpacket.h"
  21 #include "debug.h"
  22 #include "exceptions.h"
  23 #include "util/serialize.h"
  24
  25 NetworkPacket::NetworkPacket(u16 command, u32 datasize, u16 peer_id):
  26 m_datasize(datasize), m_read_offset(0), m_command(command), m_peer_id(peer_id)
  27 {
  28         m_data.resize(m_datasize);
  29 }
  30
  31 NetworkPacket::NetworkPacket(u16 command, u32 datasize):
  32 m_datasize(datasize), m_read_offset(0), m_command(command), m_peer_id(0)
  33 {
  34         m_data.resize(m_datasize);
  35 }
  36
  37 NetworkPacket::~NetworkPacket()
  38 {
  39         m_data.clear();
  40 }
  41
  42 void NetworkPacket::checkReadOffset(u32 from_offset)
  43 {
  44         if (from_offset >= m_datasize) {
  45                 std::stringstream ss;
  46                 ss << "Reading outside packet (offset: " <<
  47                                 from_offset << ", packet size: " << getSize() << ")";
  48                 throw PacketError(ss.str());
  49         }
  50 }
  51
  52 void NetworkPacket::putRawPacket(u8 *data, u32 datasize, u16 peer_id)
  53 {
  54         // If a m_command is already set, we are rewriting on same packet
  55         // This is not permitted
  56         assert(m_command == 0);
  57
  58         m_datasize = datasize - 2;
  59         m_peer_id = peer_id;
  60
  61         // split command and datas
  62         m_command = readU16(&data[0]);
  63         m_data = std::vector<u8>(&data[2], &data[2 + m_datasize]);
  64 }
  65
  66 char* NetworkPacket::getString(u32 from_offset)
  67 {
  68         checkReadOffset(from_offset);
  69
  70         return (char*)&m_data[from_offset];
  71 }
  72
  73 void NetworkPacket::putRawString(const char* src, u32 len)
  74 {
  75         if (m_read_offset + len * sizeof(char) >= m_datasize) {
  76                 m_datasize += len * sizeof(char);
  77                 m_data.resize(m_datasize);
  78         }
  79
  80         memcpy(&m_data[m_read_offset], src, len);
  81         m_read_offset += len;
  82 }
  83
  84 NetworkPacket& NetworkPacket::operator>>(std::string& dst)
  85 {
  86         u16 strLen = readU16(&m_data[m_read_offset]);
  87         m_read_offset += sizeof(u16);
  88
  89         dst.clear();
  90
  91         if (strLen == 0) {
  92                 return *this;
  93         }
  94
  95         dst.reserve(strLen);
  96         dst.append((char*)&m_data[m_read_offset], strLen);
  97
  98         m_read_offset += strLen * sizeof(char);
  99         return *this;
 100 }
 101
 102 NetworkPacket& NetworkPacket::operator<<(std::string src)
 103 {
 104         u16 msgsize = src.size();
 105         if (msgsize > 0xFFFF) {
 106                 msgsize = 0xFFFF;
 107         }
 108
 109         *this << msgsize;
 110
 111         if (m_read_offset + msgsize * sizeof(char) >= m_datasize) {
 112                 m_datasize += msgsize * sizeof(char);
 113                 m_data.resize(m_datasize);
 114         }
 115
 116         memcpy(&m_data[m_read_offset], src.c_str(), msgsize);
 117         m_read_offset += msgsize;
 118
 119         return *this;
 120 }
 121
 122 void NetworkPacket::putLongString(std::string src)
 123 {
 124         u32 msgsize = src.size();
 125         if (msgsize > 0xFFFFFFFF) {
 126                 msgsize = 0xFFFFFFFF;
 127         }
 128
 129         *this << msgsize;
 130
 131         if (m_read_offset + msgsize * sizeof(char) >= m_datasize) {
 132                 m_datasize += msgsize * sizeof(char);
 133                 m_data.resize(m_datasize);
 134         }
 135
 136         memcpy(&m_data[m_read_offset], src.c_str(), msgsize);
 137         m_read_offset += msgsize;
 138 }
 139
 140 NetworkPacket& NetworkPacket::operator>>(std::wstring& dst)
 141 {
 142         u16 strLen = readU16(&m_data[m_read_offset]);
 143         m_read_offset += sizeof(u16);
 144
 145         dst.clear();
 146
 147         if (strLen == 0) {
 148                 return *this;
 149         }
 150
 151         dst.reserve(strLen);
 152         for(u16 i=0; i<strLen; i++) {
 153                 wchar_t c16 = readU16(&m_data[m_read_offset]);
 154                 dst.append(&c16, 1);
 155                 m_read_offset += sizeof(u16);
 156         }
 157
 158         return *this;
 159 }
 160
 161 NetworkPacket& NetworkPacket::operator<<(std::wstring src)
 162 {
 163         u16 msgsize = src.size();
 164         if (msgsize > 0xFFFF) {
 165                 msgsize = 0xFFFF;
 166         }
 167
 168         *this << msgsize;
 169
 170         // Write string
 171         for (u16 i=0; i<msgsize; i++) {
 172                 *this << (u16) src[i];
 173         }
 174
 175         return *this;
 176 }
 177
 178 std::string NetworkPacket::readLongString()
 179 {
 180         u32 strLen = readU32(&m_data[m_read_offset]);
 181         m_read_offset += sizeof(u32);
 182
 183         if (strLen == 0) {
 184                 return "";
 185         }
 186
 187         std::string dst;
 188
 189         dst.reserve(strLen);
 190         dst.append((char*)&m_data[m_read_offset], strLen);
 191
 192         m_read_offset += strLen*sizeof(char);
 193
 194         return dst;
 195 }
 196
 197 NetworkPacket& NetworkPacket::operator>>(char& dst)
 198 {
 199         checkReadOffset(m_read_offset);
 200
 201         dst = readU8(&m_data[m_read_offset]);
 202
 203         incrOffset<char>();
 204         return *this;
 205 }
 206
 207 char NetworkPacket::getChar(u32 offset)
 208 {
 209         checkReadOffset(offset);
 210
 211         return readU8(&m_data[offset]);
 212 }
 213
 214 NetworkPacket& NetworkPacket::operator<<(char src)
 215 {
 216         checkDataSize<u8>();
 217
 218         writeU8(&m_data[m_read_offset], src);
 219
 220         incrOffset<char>();
 221         return *this;
 222 }
 223
 224 NetworkPacket& NetworkPacket::operator<<(u8 src)
 225 {
 226         checkDataSize<u8>();
 227
 228         writeU8(&m_data[m_read_offset], src);
 229
 230         incrOffset<u8>();
 231         return *this;
 232 }
 233
 234 NetworkPacket& NetworkPacket::operator<<(bool src)
 235 {
 236         checkDataSize<u8>();
 237
 238         writeU8(&m_data[m_read_offset], src);
 239
 240         incrOffset<u8>();
 241         return *this;
 242 }
 243
 244 NetworkPacket& NetworkPacket::operator<<(u16 src)
 245 {
 246         checkDataSize<u16>();
 247
 248         writeU16(&m_data[m_read_offset], src);
 249
 250         incrOffset<u16>();
 251         return *this;
 252 }
 253
 254 NetworkPacket& NetworkPacket::operator<<(u32 src)
 255 {
 256         checkDataSize<u32>();
 257
 258         writeU32(&m_data[m_read_offset], src);
 259
 260         incrOffset<u32>();
 261         return *this;
 262 }
 263
 264 NetworkPacket& NetworkPacket::operator<<(u64 src)
 265 {
 266         checkDataSize<u64>();
 267
 268         writeU64(&m_data[m_read_offset], src);
 269
 270         incrOffset<u64>();
 271         return *this;
 272 }
 273
 274 NetworkPacket& NetworkPacket::operator<<(float src)
 275 {
 276         checkDataSize<float>();
 277
 278         writeF1000(&m_data[m_read_offset], src);
 279
 280         incrOffset<float>();
 281         return *this;
 282 }
 283
 284 NetworkPacket& NetworkPacket::operator>>(bool& dst)
 285 {
 286         checkReadOffset(m_read_offset);
 287
 288         dst = readU8(&m_data[m_read_offset]);
 289
 290         incrOffset<u8>();
 291         return *this;
 292 }
 293
 294 NetworkPacket& NetworkPacket::operator>>(u8& dst)
 295 {
 296         checkReadOffset(m_read_offset);
 297
 298         dst = readU8(&m_data[m_read_offset]);
 299
 300         incrOffset<u8>();
 301         return *this;
 302 }
 303
 304 u8 NetworkPacket::getU8(u32 offset)
 305 {
 306         checkReadOffset(offset);
 307
 308         return readU8(&m_data[offset]);
 309 }
 310
 311 u8* NetworkPacket::getU8Ptr(u32 from_offset)
 312 {
 313         if (m_datasize == 0) {
 314                 return NULL;
 315         }
 316
 317         checkReadOffset(from_offset);
 318
 319         return (u8*)&m_data[from_offset];
 320 }
 321
 322 NetworkPacket& NetworkPacket::operator>>(u16& dst)
 323 {
 324         checkReadOffset(m_read_offset);
 325
 326         dst = readU16(&m_data[m_read_offset]);
 327
 328         incrOffset<u16>();
 329         return *this;
 330 }
 331
 332 u16 NetworkPacket::getU16(u32 from_offset)
 333 {
 334         checkReadOffset(from_offset);
 335
 336         return readU16(&m_data[from_offset]);
 337 }
 338
 339 NetworkPacket& NetworkPacket::operator>>(u32& dst)
 340 {
 341         checkReadOffset(m_read_offset);
 342
 343         dst = readU32(&m_data[m_read_offset]);
 344
 345         incrOffset<u32>();
 346         return *this;
 347 }
 348
 349 NetworkPacket& NetworkPacket::operator>>(u64& dst)
 350 {
 351         checkReadOffset(m_read_offset);
 352
 353         dst = readU64(&m_data[m_read_offset]);
 354
 355         incrOffset<u64>();
 356         return *this;
 357 }
 358
 359 NetworkPacket& NetworkPacket::operator>>(float& dst)
 360 {
 361         checkReadOffset(m_read_offset);
 362
 363         dst = readF1000(&m_data[m_read_offset]);
 364
 365         incrOffset<float>();
 366         return *this;
 367 }
 368
 369 NetworkPacket& NetworkPacket::operator>>(v2f& dst)
 370 {
 371         checkReadOffset(m_read_offset);
 372
 373         dst = readV2F1000(&m_data[m_read_offset]);
 374
 375         incrOffset<v2f>();
 376         return *this;
 377 }
 378
 379 NetworkPacket& NetworkPacket::operator>>(v3f& dst)
 380 {
 381         checkReadOffset(m_read_offset);
 382
 383         dst = readV3F1000(&m_data[m_read_offset]);
 384
 385         incrOffset<v3f>();
 386         return *this;
 387 }
 388
 389 NetworkPacket& NetworkPacket::operator>>(s16& dst)
 390 {
 391         checkReadOffset(m_read_offset);
 392
 393         dst = readS16(&m_data[m_read_offset]);
 394
 395         incrOffset<s16>();
 396         return *this;
 397 }
 398
 399 NetworkPacket& NetworkPacket::operator<<(s16 src)
 400 {
 401         *this << (u16) src;
 402         return *this;
 403 }
 404
 405 NetworkPacket& NetworkPacket::operator>>(s32& dst)
 406 {
 407         checkReadOffset(m_read_offset);
 408
 409         dst = readS32(&m_data[m_read_offset]);
 410
 411         incrOffset<s32>();
 412         return *this;
 413 }
 414
 415 NetworkPacket& NetworkPacket::operator<<(s32 src)
 416 {
 417         *this << (u32) src;
 418         return *this;
 419 }
 420
 421 NetworkPacket& NetworkPacket::operator>>(v3s16& dst)
 422 {
 423         checkReadOffset(m_read_offset);
 424
 425         dst = readV3S16(&m_data[m_read_offset]);
 426
 427         incrOffset<v3s16>();
 428         return *this;
 429 }
 430
 431 NetworkPacket& NetworkPacket::operator>>(v2s32& dst)
 432 {
 433         dst = readV2S32(&m_data[m_read_offset]);
 434
 435         incrOffset<v2s32>();
 436         return *this;
 437 }
 438
 439 NetworkPacket& NetworkPacket::operator>>(v3s32& dst)
 440 {
 441         checkReadOffset(m_read_offset);
 442
 443         dst = readV3S32(&m_data[m_read_offset]);
 444
 445         incrOffset<v3s32>();
 446         return *this;
 447 }
 448
 449 NetworkPacket& NetworkPacket::operator<<(v2f src)
 450 {
 451         *this << (float) src.X;
 452         *this << (float) src.Y;
 453         return *this;
 454 }
 455
 456 NetworkPacket& NetworkPacket::operator<<(v3f src)
 457 {
 458         *this << (float) src.X;
 459         *this << (float) src.Y;
 460         *this << (float) src.Z;
 461         return *this;
 462 }
 463
 464 NetworkPacket& NetworkPacket::operator<<(v3s16 src)
 465 {
 466         *this << (s16) src.X;
 467         *this << (s16) src.Y;
 468         *this << (s16) src.Z;
 469         return *this;
 470 }
 471
 472 NetworkPacket& NetworkPacket::operator<<(v2s32 src)
 473 {
 474         *this << (s32) src.X;
 475         *this << (s32) src.Y;
 476         return *this;
 477 }
 478
 479 NetworkPacket& NetworkPacket::operator<<(v3s32 src)
 480 {
 481         *this << (s32) src.X;
 482         *this << (s32) src.Y;
 483         *this << (s32) src.Z;
 484         return *this;
 485 }
 486
 487 NetworkPacket& NetworkPacket::operator>>(video::SColor& dst)
 488 {
 489         checkReadOffset(m_read_offset);
 490
 491         dst = readARGB8(&m_data[m_read_offset]);
 492
 493         incrOffset<u32>();
 494         return *this;
 495 }
 496
 497 NetworkPacket& NetworkPacket::operator<<(video::SColor src)
 498 {
 499         checkDataSize<u32>();
 500
 501         writeU32(&m_data[m_read_offset], src.color);
 502
 503         incrOffset<u32>();
 504         return *this;
 505 }
 506
 507 Buffer<u8> NetworkPacket::oldForgePacket()
 508 {
 509         Buffer<u8> sb(m_datasize + 2);
 510         writeU16(&sb[0], m_command);
 511
 512         u8* datas = getU8Ptr(0);
 513
 514         if (datas != NULL)
 515                 memcpy(&sb[2], datas, m_datasize);
 516         return sb;
 517 }