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