1 #ifndef MSP_NET_PROTOCOL_H_
2 #define MSP_NET_PROTOCOL_H_
13 class bad_packet: public std::runtime_error
16 bad_packet(const std::string &w): std::runtime_error(w) { }
20 class buffer_error: public std::runtime_error
23 buffer_error(const std::string &w): std::runtime_error(w) { }
30 template<typename T, std::uint8_t K>
37 struct CompoundTypeDef
39 virtual ~CompoundTypeDef() { }
41 virtual std::string describe() const = 0;
42 virtual char *serialize(const C &, char *, char *) const = 0;
43 virtual const char *deserialize(C &, const char *, const char *) const = 0;
46 template<typename C, typename S>
47 struct CompoundDef: public CompoundTypeDef<C>
51 CompoundDef(const S &);
53 virtual std::string describe() const;
54 virtual char *serialize(const C &, char *, char *) const;
55 virtual const char *deserialize(C &, const char *, const char *) const;
64 BasicSerializer(const Protocol &) { }
66 std::string describe() const { return get_type_signature<T>(); }
67 char *serialize(const T &, char *, char *) const;
68 const char *deserialize(T &, const char *, const char *) const;
71 class StringSerializer
74 typedef std::string ValueType;
77 BasicSerializer<std::uint16_t> length_serializer;
80 StringSerializer(const Protocol &);
82 std::string describe() const { return get_type_signature<std::string>(); }
83 char *serialize(const std::string &, char *, char *) const;
84 const char *deserialize(std::string &, const char *, const char *) const;
94 BasicSerializer<std::uint16_t> length_serializer;
95 typename Traits<typename A::value_type>::Serializer element_serializer;
98 ArraySerializer(const Protocol &);
100 std::string describe() const;
101 char *serialize(const A &, char *, char *) const;
102 const char *deserialize(A &, const char *, const char *) const;
106 class CompoundSerializer
112 const CompoundTypeDef<C> &def;
115 CompoundSerializer(const Protocol &);
117 std::string describe() const { return def.describe(); }
118 char *serialize(const C &, char *, char *) const;
119 const char *deserialize(C &, const char *, const char *) const;
122 template<typename P, typename Head, typename S>
123 class Serializer: public Head
129 typedef Serializer<P, Serializer<P, Head, S>, typename Traits<N>::Serializer> Type;
133 typedef typename S::ValueType P::*Pointer;
139 Serializer(const Head &, Pointer, const Protocol &);
141 std::string describe() const;
142 char *serialize(const P &, char *, char *) const;
143 const char *deserialize(P &, const char *, const char *) const;
153 typedef Serializer<P, NullSerializer, typename Traits<N>::Serializer> Type;
156 std::string describe() const { return std::string(); }
157 char *serialize(const P &, char *b, char *) const { return b; }
158 const char *deserialize(P &, const char *b, const char *) const { return b; }
166 PacketDefBase(unsigned);
168 virtual ~PacketDefBase() { }
169 virtual unsigned get_class_id() const = 0;
170 unsigned get_id() const { return id; }
171 virtual std::string describe() const = 0;
172 virtual const char *dispatch(ReceiverBase &, const char *, const char *) const = 0;
176 class PacketTypeDef: public PacketDefBase
179 CompoundTypeDef<P> *compound;
182 PacketTypeDef(unsigned);
185 virtual unsigned get_class_id() const { return get_packet_class_id<P>(); }
188 void set_serializer(const S &);
190 const CompoundTypeDef<P> &get_compound() const { return *compound; }
192 virtual std::string describe() const { return compound->describe(); }
193 char *serialize(const P &, char *, char *) const;
194 const char *deserialize(P &, const char *, const char *) const;
195 virtual const char *dispatch(ReceiverBase &, const char *, const char *) const;
198 template<typename P, typename S>
199 class PacketDefBuilder
202 const Protocol &protocol;
203 PacketTypeDef<P> &pktdef;
207 PacketDefBuilder(const Protocol &, PacketTypeDef<P> &, const S &);
210 PacketDefBuilder<P, typename S::template Next<T>::Type> operator()(T P::*);
216 std::uint16_t length;
219 PacketHeader(std::uint16_t, std::uint16_t);
222 typedef std::map<unsigned, PacketDefBase *> PacketMap;
224 PacketTypeDef<PacketHeader> header_def;
225 unsigned next_packet_id;
226 PacketMap packet_class_defs;
227 PacketMap packet_id_defs;
230 Protocol(unsigned = 1);
235 static unsigned get_next_packet_class_id();
238 static unsigned get_packet_class_id();
240 void add_packet(PacketDefBase *);
244 PacketDefBuilder<P, NullSerializer<P> > add(unsigned);
247 PacketDefBuilder<P, NullSerializer<P> > add();
249 const PacketDefBase &get_packet_by_class_id(unsigned) const;
250 const PacketDefBase &get_packet_by_id(unsigned) const;
253 const PacketTypeDef<P> &get_packet_by_class() const;
257 std::size_t serialize(const P &, char *, std::size_t) const;
259 std::size_t get_packet_size(const char *, std::size_t) const;
260 std::size_t dispatch(ReceiverBase &, const char *, std::size_t) const;
262 std::uint64_t get_hash() const;
266 static std::string get_type_signature();
271 unsigned Protocol::get_packet_class_id()
273 static unsigned id = get_next_packet_class_id();
278 Protocol::PacketDefBuilder<P, Protocol::NullSerializer<P> > Protocol::add(unsigned id)
280 PacketTypeDef<P> *pdef = new PacketTypeDef<P>(id);
282 return PacketDefBuilder<P, NullSerializer<P> >(*this, *pdef, NullSerializer<P>());
286 Protocol::PacketDefBuilder<P, Protocol::NullSerializer<P> > Protocol::add()
288 return add<P>(next_packet_id++);
292 const Protocol::PacketTypeDef<P> &Protocol::get_packet_by_class() const
294 const PacketDefBase &pdef = get_packet_by_class_id(get_packet_class_id<P>());
295 return static_cast<const PacketTypeDef<P> &>(pdef);
299 std::size_t Protocol::serialize(const P &pkt, char *buf, std::size_t size) const
301 const PacketTypeDef<P> &pdef = get_packet_by_class<P>();
303 throw std::invalid_argument("no packet id");
304 char *ptr = pdef.serialize(pkt, buf+4, buf+size);
306 header_def.serialize(PacketHeader(pdef.get_id(), size), buf, buf+4);
311 std::string Protocol::get_type_signature()
313 const std::uint16_t sig = Traits<T>::signature;
317 result += '0'+(sig>>8);
322 template<typename T, std::uint8_t K>
323 struct Protocol::BasicTraits
325 static const std::uint16_t signature = K | (sizeof(T)<<8);
326 typedef BasicSerializer<T> Serializer;
330 struct Protocol::Traits
332 static const std::uint16_t signature = 'C';
333 typedef CompoundSerializer<T> Serializer;
336 template<> struct Protocol::Traits<bool>: BasicTraits<bool, 'B'> { };
337 template<> struct Protocol::Traits<std::int8_t>: BasicTraits<std::int8_t, 'I'> { };
338 template<> struct Protocol::Traits<std::uint8_t>: BasicTraits<std::uint8_t, 'U'> { };
339 template<> struct Protocol::Traits<std::int16_t>: BasicTraits<std::int16_t, 'I'> { };
340 template<> struct Protocol::Traits<std::uint16_t>: BasicTraits<std::uint16_t, 'U'> { };
341 template<> struct Protocol::Traits<std::int32_t>: BasicTraits<std::int32_t, 'I'> { };
342 template<> struct Protocol::Traits<std::uint32_t>: BasicTraits<std::uint32_t, 'U'> { };
343 template<> struct Protocol::Traits<std::int64_t>: BasicTraits<std::int64_t, 'I'> { };
344 template<> struct Protocol::Traits<std::uint64_t>: BasicTraits<std::uint64_t, 'U'> { };
345 template<> struct Protocol::Traits<float>: BasicTraits<float, 'F'> { };
346 template<> struct Protocol::Traits<double>: BasicTraits<double, 'F'> { };
348 template<> struct Protocol::Traits<std::string>
350 static const std::uint16_t signature = 'S';
351 typedef StringSerializer Serializer;
355 struct Protocol::Traits<std::vector<T> >
357 static const std::uint16_t signature = 'A';
358 typedef ArraySerializer<std::vector<T> > Serializer;
363 template<typename C, typename S>
364 Protocol::CompoundDef<C, S>::CompoundDef(const S &s):
368 template<typename C, typename S>
369 std::string Protocol::CompoundDef<C, S>::describe() const
371 return "{"+serializer.describe()+"}";
374 template<typename C, typename S>
375 char *Protocol::CompoundDef<C, S>::serialize(const C &com, char *buf, char *end) const
377 return serializer.serialize(com, buf, end);
380 template<typename C, typename S>
381 const char *Protocol::CompoundDef<C, S>::deserialize(C &com, const char *buf, const char *end) const
383 return serializer.deserialize(com, buf, end);
388 Protocol::ArraySerializer<A>::ArraySerializer(const Protocol &proto):
389 length_serializer(proto),
390 element_serializer(proto)
394 std::string Protocol::ArraySerializer<A>::describe() const
396 return "["+element_serializer.describe()+"]";
400 char *Protocol::ArraySerializer<A>::serialize(const A &array, char *buf, char *end) const
402 buf = length_serializer.serialize(array.size(), buf, end);
403 for(typename A::const_iterator i=array.begin(); i!=array.end(); ++i)
404 buf = element_serializer.serialize(*i, buf, end);
409 const char *Protocol::ArraySerializer<A>::deserialize(A &array, const char *buf, const char *end) const
411 std::uint16_t length;
412 buf = length_serializer.deserialize(length, buf, end);
413 array.resize(length);
414 for(unsigned i=0; i<length; ++i)
415 buf = element_serializer.deserialize(array[i], buf, end);
421 Protocol::CompoundSerializer<C>::CompoundSerializer(const Protocol &proto):
422 def(proto.get_packet_by_class<C>().get_compound())
426 char *Protocol::CompoundSerializer<C>::serialize(const C &com, char *buf, char *end) const
428 return def.serialize(com, buf, end);
432 const char *Protocol::CompoundSerializer<C>::deserialize(C &com, const char *buf, const char *end) const
434 return def.deserialize(com, buf, end);
438 template<typename P, typename Head, typename S>
439 Protocol::Serializer<P, Head, S>::Serializer(const Head &h, Pointer p, const Protocol &proto):
445 template<typename P, typename Head, typename S>
446 std::string Protocol::Serializer<P, Head, S>::describe() const
448 return Head::describe()+ser.describe();
451 template<typename P, typename Head, typename S>
452 char *Protocol::Serializer<P, Head, S>::serialize(const P &pkt, char *buf, char *end) const
454 buf = Head::serialize(pkt, buf, end);
455 return ser.serialize(pkt.*ptr, buf, end);
458 template<typename P, typename Head, typename S>
459 const char *Protocol::Serializer<P, Head, S>::deserialize(P &pkt, const char *buf, const char *end) const
461 buf = Head::deserialize(pkt, buf, end);
462 return ser.deserialize(pkt.*ptr, buf, end);
467 Protocol::PacketTypeDef<P>::PacketTypeDef(unsigned i):
469 compound(new CompoundDef<P, NullSerializer<P> >(NullSerializer<P>()))
473 Protocol::PacketTypeDef<P>::~PacketTypeDef()
480 void Protocol::PacketTypeDef<P>::set_serializer(const S &ser)
483 compound = new CompoundDef<P, S>(ser);
487 char *Protocol::PacketTypeDef<P>::serialize(const P &pkt, char *buf, char *end) const
489 return compound->serialize(pkt, buf, end);
493 const char *Protocol::PacketTypeDef<P>::deserialize(P &pkt, const char *buf, const char *end) const
495 return compound->deserialize(pkt, buf, end);
499 const char *Protocol::PacketTypeDef<P>::dispatch(ReceiverBase &rcv, const char *buf, const char *end) const
501 PacketReceiver<P> *prcv = dynamic_cast<PacketReceiver<P> *>(&rcv);
503 throw bad_packet("unsupported");
505 buf = deserialize(pkt, buf, end);
511 template<typename P, typename S>
512 Protocol::PacketDefBuilder<P, S>::PacketDefBuilder(const Protocol &p, PacketTypeDef<P> &k, const S &s):
518 template<typename P, typename S>
520 Protocol::PacketDefBuilder<P, typename S::template Next<T>::Type> Protocol::PacketDefBuilder<P, S>::operator()(T P::*ptr)
522 typename S::template Next<T>::Type next_ser(serializer, ptr, protocol);
523 pktdef.set_serializer(next_ser);
524 return PacketDefBuilder<P, typename S::template Next<T>::Type>(protocol, pktdef, next_ser);