1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */
#ifndef UNPACK_PACKET_H
#define UNPACK_PACKET_H
#include "RawPacket.h"
#include <string>
#include <vector>
#include <cstring>
#include <stdexcept>
#include <boost/shared_ptr.hpp>
namespace netcode
{
class UnpackPacketException : public std::runtime_error {
public:
UnpackPacketException(const std::string& what) : std::runtime_error(what) {}
};
class UnpackPacket
{
public:
UnpackPacket(boost::shared_ptr<const RawPacket>, size_t skipBytes = 0);
template <typename T>
void operator>>(T& t)
{
if ((pos + sizeof(T)) > pckt->length) {
throw UnpackPacketException("Unpack failure (type)");
}
t = *(T*)(pckt->data + pos);
pos += sizeof(T);
}
template <typename element>
void operator>>(std::vector<element>& vec)
{
if ((pckt->length - pos) < (vec.size() * sizeof(element))) {
throw UnpackPacketException("Unpack failure (vector)");
}
const size_t toCopy = vec.size() * sizeof(element);
std::memcpy((void*)(&vec[0]), pckt->data + pos, toCopy);
pos += toCopy;
}
void operator>>(std::string& text)
{
int i = pos;
for (; i < pckt->length; ++i) {
if (pckt->data[i] == '\0') {
break;
}
}
if (i >= pckt->length) {
throw UnpackPacketException("Unpack failure (string)");
}
text = std::string((char*)(pckt->data + pos));
pos += text.size() + 1;
}
private:
boost::shared_ptr<const RawPacket> pckt;
size_t pos;
};
} // namespace netcode
#endif // UNPACK_PACKET_H
|
