b24cef1f5e3f03ef067711271480c2eb1ab378cb
[libs/net.git] / source / net / protocol.cpp
1 #include <cstring>
2 #include <string>
3 #include <msp/core/hash.h>
4 #include <msp/core/maputils.h>
5 #include <msp/strings/format.h>
6 #include <msp/strings/lexicalcast.h>
7 #include "protocol.h"
8
9 using namespace std;
10
11 namespace Msp {
12 namespace Net {
13
14 Protocol::Protocol(unsigned npi):
15         header_def(0),
16         next_packet_id(npi)
17 {
18         PacketDefBuilder<PacketHeader, NullSerializer<PacketHeader> >(*this, header_def, NullSerializer<PacketHeader>())
19                 (&PacketHeader::type)(&PacketHeader::length);
20 }
21
22 Protocol::~Protocol()
23 {
24         for(map<unsigned, PacketDefBase *>::iterator i=packet_class_defs.begin(); i!=packet_class_defs.end(); ++i)
25                 delete i->second;
26 }
27
28 void Protocol::add_packet(PacketDefBase *pdef)
29 {
30         PacketDefBase *&ptr = packet_class_defs[pdef->get_class_id()];
31         if(ptr)
32         {
33                 packet_id_defs.erase(ptr->get_id());
34                 delete ptr;
35         }
36         ptr = pdef;
37         if(unsigned id = pdef->get_id())
38                 packet_id_defs[id] = pdef;
39 }
40
41 const Protocol::PacketDefBase &Protocol::get_packet_by_class_id(unsigned id) const
42 {
43         return *get_item(packet_class_defs, id);
44 }
45
46 const Protocol::PacketDefBase &Protocol::get_packet_by_id(unsigned id) const
47 {
48         return *get_item(packet_id_defs, id);
49 }
50
51 unsigned Protocol::dispatch(ReceiverBase &rcv, const char *buf, unsigned size) const
52 {
53         PacketHeader header;
54         buf = header_def.deserialize(header, buf, buf+size);
55         if(header.length>size)
56                 throw bad_packet("truncated");
57         const PacketDefBase &pdef = get_packet_by_id(header.type);
58         const char *ptr = pdef.dispatch(rcv, buf, buf+header.length);
59         return ptr-buf;
60 }
61
62 unsigned Protocol::get_packet_size(const char *buf, unsigned size) const
63 {
64         if(size<4)
65                 return 0;
66         PacketHeader header;
67         header_def.deserialize(header, buf, buf+size);
68         return header.length;
69 }
70
71 UInt64 Protocol::get_hash() const
72 {
73         string description;
74         for(PacketMap::const_iterator i=packet_id_defs.begin(); i!=packet_id_defs.end(); ++i)
75                 description += format("%d:%s\n", i->first, i->second->describe());
76         return hash64(description);
77 }
78
79
80 /* TODO These assumes the machine is little-endian; are there any relevant
81 big-endian platforms these days? */
82 template<typename T>
83 char *Protocol::BasicSerializer<T>::serialize(const T &value, char *buf, char *end) const
84 {
85         if(end-buf<static_cast<int>(sizeof(T)))
86                 throw buffer_error("overflow");
87
88         const char *ptr = reinterpret_cast<const char *>(&value)+sizeof(T);
89         for(unsigned i=0; i<sizeof(T); ++i)
90                 *buf++ = *--ptr;
91
92         return buf;
93 }
94
95 template<typename T>
96 const char *Protocol::BasicSerializer<T>::deserialize(T &value, const char *buf, const char *end) const
97 {
98         if(end-buf<static_cast<int>(sizeof(T)))
99                 throw buffer_error("underflow");
100
101         char *ptr = reinterpret_cast<char *>(&value)+sizeof(T);
102         for(unsigned i=0; i<sizeof(T); ++i)
103                 *--ptr = *buf++;
104
105         return buf;
106 }
107
108 template char *Protocol::BasicSerializer<Int8>::serialize(const Int8 &, char *, char *) const;
109 template char *Protocol::BasicSerializer<Int16>::serialize(const Int16 &, char *, char *) const;
110 template char *Protocol::BasicSerializer<Int32>::serialize(const Int32 &, char *, char *) const;
111 template char *Protocol::BasicSerializer<Int64>::serialize(const Int64 &, char *, char *) const;
112 template char *Protocol::BasicSerializer<UInt8>::serialize(const UInt8 &, char *, char *) const;
113 template char *Protocol::BasicSerializer<UInt16>::serialize(const UInt16 &, char *, char *) const;
114 template char *Protocol::BasicSerializer<UInt32>::serialize(const UInt32 &, char *, char *) const;
115 template char *Protocol::BasicSerializer<UInt64>::serialize(const UInt64 &, char *, char *) const;
116 template char *Protocol::BasicSerializer<float>::serialize(const float &, char *, char *) const;
117 template char *Protocol::BasicSerializer<double>::serialize(const double &, char *, char *) const;
118 template const char *Protocol::BasicSerializer<Int8>::deserialize(Int8 &, const char *, const char *) const;
119 template const char *Protocol::BasicSerializer<Int16>::deserialize(Int16 &, const char *, const char *) const;
120 template const char *Protocol::BasicSerializer<Int32>::deserialize(Int32 &, const char *, const char *) const;
121 template const char *Protocol::BasicSerializer<Int64>::deserialize(Int64 &, const char *, const char *) const;
122 template const char *Protocol::BasicSerializer<UInt8>::deserialize(UInt8 &, const char *, const char *) const;
123 template const char *Protocol::BasicSerializer<UInt16>::deserialize(UInt16 &, const char *, const char *) const;
124 template const char *Protocol::BasicSerializer<UInt32>::deserialize(UInt32 &, const char *, const char *) const;
125 template const char *Protocol::BasicSerializer<UInt64>::deserialize(UInt64 &, const char *, const char *) const;
126 template const char *Protocol::BasicSerializer<float>::deserialize(float &, const char *, const char *) const;
127 template const char *Protocol::BasicSerializer<double>::deserialize(double &, const char *, const char *) const;
128
129
130 Protocol::StringSerializer::StringSerializer(const Protocol &p):
131         length_serializer(p)
132 { }
133
134 char *Protocol::StringSerializer::serialize(const string &str, char *buf, char *end) const
135 {
136         buf = length_serializer.serialize(str.size(), buf, end);
137         if(end-buf<static_cast<int>(str.size()))
138                 throw buffer_error("overflow");
139         copy(str.begin(), str.end(), buf);
140         return buf+str.size();
141 }
142
143 const char *Protocol::StringSerializer::deserialize(string &str, const char *buf, const char *end) const
144 {
145         UInt16 length;
146         buf = length_serializer.deserialize(length, buf, end);
147         if(end-buf<static_cast<int>(length))
148                 throw buffer_error("underflow");
149         str.assign(buf, buf+length);
150         return buf+length;
151 }
152
153
154 unsigned Protocol::PacketDefBase::next_class_id = 1;
155
156 Protocol::PacketDefBase::PacketDefBase(unsigned i):
157         id(i)
158 { }
159
160
161 Protocol::PacketHeader::PacketHeader():
162         type(0),
163         length(0)
164 { }
165
166 Protocol::PacketHeader::PacketHeader(UInt16 t, UInt16 l):
167         type(t),
168         length(l)
169 { }
170
171 } // namespace Net
172 } // namespace Msp