1 #ifndef MSP_NET_PROTOCOL_H_
2 #define MSP_NET_PROTOCOL_H_
6 #include <msp/core/except.h>
20 PacketDefBase(unsigned i): id(i) { }
22 virtual ~PacketDefBase() { }
23 virtual unsigned get_class_id() const = 0;
24 unsigned get_id() const { return id; }
25 virtual const char *disassemble(ReceiverBase &, const char *, const char *) const = 0;
27 static unsigned next_class_id;
36 virtual ~FieldBase() { }
37 virtual char *assemble(const P &, char *, char *) const = 0;
38 virtual const char *disassemble(P &, const char *, const char *) const = 0;
41 template<typename P, typename T>
42 class Field: public FieldBase<P>
48 Field(T P::*p): ptr(p) { }
50 virtual char *assemble(const P &p, char *d, char *e) const
51 { return assemble_field(p.*ptr, d, e); }
53 virtual const char *disassemble(P &p, const char *d, const char *e) const
54 { return disassemble_field(p.*ptr, d, e); }
59 class PacketDef: public PacketDefBase
62 std::vector<FieldBase<P> *> fields;
65 PacketDef(unsigned i): PacketDefBase(i)
66 { if(!class_id) class_id = next_class_id++; }
70 for(typename std::vector<FieldBase<P> *>::const_iterator i=fields.begin(); i!=fields.end(); ++i)
74 virtual unsigned get_class_id() const { return class_id; }
77 PacketDef &operator()(T P::*p)
78 { fields.push_back(new Field<P, T>(p)); return *this; }
80 char *assemble(const P &p, char *d, char *e) const
82 for(typename std::vector<FieldBase<P> *>::const_iterator i=fields.begin(); i!=fields.end(); ++i)
83 d = (*i)->assemble(p, d, e);
87 const char *disassemble(ReceiverBase &r, const char *d, const char *e) const
89 PacketReceiver<P> *prcv = dynamic_cast<PacketReceiver<P> *>(&r);
91 throw Exception("Packet type not supported by receiver");
93 for(typename std::vector<FieldBase<P> *>::const_iterator i=fields.begin(); i!=fields.end(); ++i)
94 d = (*i)->disassemble(pkt, d, e);
99 static unsigned class_id;
102 typedef std::map<unsigned, PacketDefBase *> PacketMap;
104 unsigned next_packet_id;
105 PacketMap packet_class_defs;
106 PacketMap packet_id_defs;
108 Protocol(unsigned = 1);
113 void add_packet(PacketDefBase &);
119 PacketDef<P> *pdef = new PacketDef<P>(next_packet_id++);
124 const PacketDefBase &get_packet_by_class(unsigned) const;
125 const PacketDefBase &get_packet_by_id(unsigned) const;
129 unsigned assemble(const P &pkt, char *buf, unsigned size) const
131 unsigned id = PacketDef<P>::class_id;
132 const PacketDef<P> &pdef = static_cast<const PacketDef<P> &>(get_packet_by_class(id));
133 char *ptr = pdef.assemble(pkt, buf+4, buf+size);
134 assemble_header(buf, pdef.get_id(), (size = ptr-buf));
138 unsigned disassemble(ReceiverBase &, const char *, unsigned) const;
140 unsigned get_packet_size(const char *, unsigned) const;
142 unsigned get_hash() const;
145 static void assemble_header(char *, unsigned, unsigned);
148 static char *assemble_field(const T &, char *, char *);
151 static const char *disassemble_field(T &, const char *, const char *);
155 unsigned Protocol::PacketDef<P>::class_id = 0;