3 #include <msp/core/maputils.h>
4 #include <msp/strings/lexicalcast.h>
11 using Msp::Net::buffer_error;
17 static char *assemble(const T &v, char *, char *);
18 static const char *disassemble(T &, const char *, const char *);
22 class Assembler<vector<T> >
25 static char *assemble(const vector<T> &v, char *, char *);
26 static const char *disassemble(vector<T> &, const char *, const char *);
30 char *Assembler<T>::assemble(const T &v, char *data, char *end)
32 // XXX Assumes little-endian
33 const char *ptr = reinterpret_cast<const char *>(&v)+sizeof(T);
34 for(unsigned i=0; i<sizeof(T); ++i)
37 throw buffer_error("overflow");
44 char *Assembler<string>::assemble(const string &v, char *data, char *end)
46 data = Assembler<unsigned short>::assemble(v.size(), data, end);
47 if(end-data<static_cast<int>(v.size()))
48 throw buffer_error("overflow");
49 memcpy(data, v.data(), v.size());
54 char *Assembler<vector<T> >::assemble(const vector<T> &v, char *data, char *end)
56 data = Assembler<unsigned short>::assemble(v.size(), data, end);
57 for(typename vector<T>::const_iterator i=v.begin(); i!=v.end(); ++i)
58 data = Assembler<T>::assemble(*i, data, end);
63 const char *Assembler<T>::disassemble(T &v, const char *data, const char *end)
65 char *ptr = reinterpret_cast<char *>(&v)+sizeof(T);
66 for(unsigned i=0; i<sizeof(T); ++i)
69 throw buffer_error("underflow");
76 const char *Assembler<string>::disassemble(string &v, const char *data, const char *end)
79 data = Assembler<unsigned short>::disassemble(size, data, end);
81 throw buffer_error("underflow");
82 v.assign(data, data+size);
87 const char *Assembler<vector<T> >::disassemble(vector<T> &v, const char *data, const char *end)
89 /* We assume that the vector is in pristine state - this holds because the
90 only code path leading here is from PacketDef<P>::disassemble, which creates
93 data = Assembler<unsigned short>::disassemble(size, data, end);
94 for(unsigned i=0; i<size; ++i)
97 data = Assembler<T>::disassemble(u, data, end);
108 Protocol::Protocol(unsigned npi):
112 Protocol::~Protocol()
114 for(map<unsigned, PacketDefBase *>::iterator i=packet_class_defs.begin(); i!=packet_class_defs.end(); ++i)
118 void Protocol::add_packet(PacketDefBase &pdef)
120 PacketDefBase *&ptr = packet_class_defs[pdef.get_class_id()];
124 packet_id_defs[pdef.get_id()] = &pdef;
127 const Protocol::PacketDefBase &Protocol::get_packet_by_class(unsigned id) const
129 return *get_item(packet_class_defs, id);
132 const Protocol::PacketDefBase &Protocol::get_packet_by_id(unsigned id) const
134 return *get_item(packet_id_defs, id);
137 unsigned Protocol::disassemble(ReceiverBase &rcv, const char *data, unsigned size) const
139 const unsigned char *udata = reinterpret_cast<const unsigned char *>(data);
140 unsigned id = (udata[0]<<8)+udata[1];
141 unsigned psz = (udata[2]<<8)+udata[3];
143 throw bad_packet("truncated");
144 const PacketDefBase &pdef = get_packet_by_id(id);
145 const char *ptr = pdef.disassemble(rcv, data+4, data+psz);
149 unsigned Protocol::get_packet_size(const char *data, unsigned size) const
153 const unsigned char *udata = reinterpret_cast<const unsigned char *>(data);
154 return (udata[2]<<8)+udata[3];
157 unsigned Protocol::get_hash() const
163 void Protocol::assemble_header(char *buf, unsigned id, unsigned size)
165 buf[0] = (id>>8)&0xFF;
167 buf[2] = (size>>8)&0xFF;
172 char *Protocol::assemble_field(const T &v, char *d, char *e)
173 { return Assembler<T>::assemble(v, d, e); }
175 template char *Protocol::assemble_field<>(const char &v, char *d, char *e);
176 template char *Protocol::assemble_field<>(const signed char &v, char *d, char *e);
177 template char *Protocol::assemble_field<>(const unsigned char &v, char *d, char *e);
178 template char *Protocol::assemble_field<>(const short &v, char *d, char *e);
179 template char *Protocol::assemble_field<>(const unsigned short &v, char *d, char *e);
180 template char *Protocol::assemble_field<>(const int &v, char *d, char *e);
181 template char *Protocol::assemble_field<>(const unsigned &v, char *d, char *e);
182 template char *Protocol::assemble_field<>(const long &v, char *d, char *e);
183 template char *Protocol::assemble_field<>(const unsigned long &v, char *d, char *e);
184 template char *Protocol::assemble_field<>(const float &v, char *d, char *e);
185 template char *Protocol::assemble_field<>(const double &v, char *d, char *e);
186 template char *Protocol::assemble_field<>(const string &v, char *d, char *e);
187 template char *Protocol::assemble_field<>(const vector<char> &v, char *d, char *e);
188 template char *Protocol::assemble_field<>(const vector<signed char> &v, char *d, char *e);
189 template char *Protocol::assemble_field<>(const vector<unsigned char> &v, char *d, char *e);
190 template char *Protocol::assemble_field<>(const vector<short> &v, char *d, char *e);
191 template char *Protocol::assemble_field<>(const vector<unsigned short> &v, char *d, char *e);
192 template char *Protocol::assemble_field<>(const vector<int> &v, char *d, char *e);
193 template char *Protocol::assemble_field<>(const vector<unsigned> &v, char *d, char *e);
194 template char *Protocol::assemble_field<>(const vector<long> &v, char *d, char *e);
195 template char *Protocol::assemble_field<>(const vector<unsigned long> &v, char *d, char *e);
196 template char *Protocol::assemble_field<>(const vector<float> &v, char *d, char *e);
197 template char *Protocol::assemble_field<>(const vector<double> &v, char *d, char *e);
198 template char *Protocol::assemble_field<>(const vector<string> &v, char *d, char *e);
201 const char *Protocol::disassemble_field(T &v, const char *d, const char *e)
202 { return Assembler<T>::disassemble(v, d, e); }
204 template const char *Protocol::disassemble_field<>(char &v, const char *d, const char *e);
205 template const char *Protocol::disassemble_field<>(signed char &v, const char *d, const char *e);
206 template const char *Protocol::disassemble_field<>(unsigned char &v, const char *d, const char *e);
207 template const char *Protocol::disassemble_field<>(short &v, const char *d, const char *e);
208 template const char *Protocol::disassemble_field<>(unsigned short &v, const char *d, const char *e);
209 template const char *Protocol::disassemble_field<>(int &v, const char *d, const char *e);
210 template const char *Protocol::disassemble_field<>(unsigned &v, const char *d, const char *e);
211 template const char *Protocol::disassemble_field<>(long &v, const char *d, const char *e);
212 template const char *Protocol::disassemble_field<>(unsigned long &v, const char *d, const char *e);
213 template const char *Protocol::disassemble_field<>(float &v, const char *d, const char *e);
214 template const char *Protocol::disassemble_field<>(double &v, const char *d, const char *e);
215 template const char *Protocol::disassemble_field<>(string &v, const char *d, const char *e);
216 template const char *Protocol::disassemble_field<>(vector<char> &v, const char *d, const char *e);
217 template const char *Protocol::disassemble_field<>(vector<signed char> &v, const char *d, const char *e);
218 template const char *Protocol::disassemble_field<>(vector<unsigned char> &v, const char *d, const char *e);
219 template const char *Protocol::disassemble_field<>(vector<short> &v, const char *d, const char *e);
220 template const char *Protocol::disassemble_field<>(vector<unsigned short> &v, const char *d, const char *e);
221 template const char *Protocol::disassemble_field<>(vector<int> &v, const char *d, const char *e);
222 template const char *Protocol::disassemble_field<>(vector<unsigned> &v, const char *d, const char *e);
223 template const char *Protocol::disassemble_field<>(vector<long> &v, const char *d, const char *e);
224 template const char *Protocol::disassemble_field<>(vector<unsigned long> &v, const char *d, const char *e);
225 template const char *Protocol::disassemble_field<>(vector<float> &v, const char *d, const char *e);
226 template const char *Protocol::disassemble_field<>(vector<double> &v, const char *d, const char *e);
227 template const char *Protocol::disassemble_field<>(vector<string> &v, const char *d, const char *e);
229 unsigned Protocol::PacketDefBase::next_class_id = 1;