Support compound types in network packets

[libs/net.git] / source / net / protocol.h

diff --git a/source/net/protocol.h b/source/net/protocol.h
index f722a76c82a7cc73b83d4880b7ac8043f45bd889..0a8110c9518182f91ada3cfe48f0f09ecd579b98 100644 (file)
--- a/source/net/protocol.h
+++ b/source/net/protocol.h
@@ -105,6 +105,23 @@ private:
                const char *deserialize(A &, const char *, const char *) const;
        };
 
+       template<typename C>
+       class CompoundSerializer
+       {
+       public:
+               typedef C ValueType;
+
+       private:
+               const CompoundTypeDef<C> &def;
+
+       public:
+               CompoundSerializer(const Protocol &);
+
+               std::string describe() const { return def.describe(); }
+               char *serialize(const C &, char *, char *) const;
+               const char *deserialize(C &, const char *, const char *) const;
+       };
+
        template<typename P, typename Head, typename S>
        class Serializer: public Head
        {
@@ -226,6 +243,9 @@ private:
        void add_packet(PacketDefBase *);
 
 protected:
+       template<typename P>
+       PacketDefBuilder<P, NullSerializer<P> > add(unsigned);
+
        template<typename P>
        PacketDefBuilder<P, NullSerializer<P> > add();
 
@@ -251,13 +271,19 @@ private:
 
 
 template<typename P>
-Protocol::PacketDefBuilder<P, Protocol::NullSerializer<P> > Protocol::add()
+Protocol::PacketDefBuilder<P, Protocol::NullSerializer<P> > Protocol::add(unsigned id)
 {
-       PacketTypeDef<P> *pdef = new PacketTypeDef<P>(next_packet_id++);
+       PacketTypeDef<P> *pdef = new PacketTypeDef<P>(id);
        add_packet(pdef);
        return PacketDefBuilder<P, NullSerializer<P> >(*this, *pdef, NullSerializer<P>());
 }
 
+template<typename P>
+Protocol::PacketDefBuilder<P, Protocol::NullSerializer<P> > Protocol::add()
+{
+       return add<P>(next_packet_id++);
+}
+
 template<typename P>
 const Protocol::PacketTypeDef<P> &Protocol::get_packet_by_class() const
 {
@@ -294,6 +320,13 @@ struct Protocol::BasicTraits
        typedef BasicSerializer<T> Serializer;
 };
 
+template<typename T>
+struct Protocol::Traits
+{
+       static const UInt16 signature = 'C';
+       typedef CompoundSerializer<T> Serializer;
+};
+
 template<> struct Protocol::Traits<Int8>: BasicTraits<Int8, 'I'> { };
 template<> struct Protocol::Traits<UInt8>: BasicTraits<UInt8, 'U'> { };
 template<> struct Protocol::Traits<Int16>: BasicTraits<Int16, 'I'> { };
@@ -383,6 +416,24 @@ const char *Protocol::ArraySerializer<A>::deserialize(A &array, const char *buf,
 }
 
 
+template<typename C>
+Protocol::CompoundSerializer<C>::CompoundSerializer(const Protocol &proto):
+       def(proto.get_packet_by_class<C>().get_compound())
+{ }
+
+template<typename C>
+char *Protocol::CompoundSerializer<C>::serialize(const C &com, char *buf, char *end) const
+{
+       return def.serialize(com, buf, end);
+}
+
+template<typename C>
+const char *Protocol::CompoundSerializer<C>::deserialize(C &com, const char *buf, const char *end) const
+{
+       return def.deserialize(com, buf, end);
+}
+
+
 template<typename P, typename Head, typename S>
 Protocol::Serializer<P, Head, S>::Serializer(const Head &h, Pointer p, const Protocol &proto):
        Head(h),