Acutally implement Protocol::get_hash
authorMikko Rasa <tdb@tdb.fi>
Sun, 9 Oct 2016 15:03:24 +0000 (18:03 +0300)
committerMikko Rasa <tdb@tdb.fi>
Sun, 9 Oct 2016 15:08:31 +0000 (18:08 +0300)
Also trigger an error in Communicator if the hashes don't match

source/net/communicator.cpp
source/net/communicator.h
source/net/protocol.cpp
source/net/protocol.h

index 92279c24e642bc7c6f9c2b80eba60f69d1b9c8a2..07c3be4f6b1d7c25623a957217a98f5bccea21a6 100644 (file)
@@ -10,7 +10,7 @@ using namespace Msp::Net;
 
 struct Handshake
 {
-       unsigned hash;
+       Msp::UInt64 hash;
 };
 
 
@@ -30,11 +30,11 @@ HandshakeProtocol::HandshakeProtocol():
 class HandshakeReceiver: public PacketReceiver<Handshake>
 {
 private:
-       unsigned hash;
+       Msp::UInt64 hash;
 
 public:
        HandshakeReceiver();
-       unsigned get_hash() const { return hash; }
+       Msp::UInt64 get_hash() const { return hash; }
        virtual void receive(const Handshake &);
 };
 
@@ -125,15 +125,16 @@ void Communicator::data_available()
                                HandshakeReceiver hsrecv;
                                if((more = receive_packet(hsproto, hsrecv)))
                                {
+                                       if(handshake_status==0)
+                                               send_handshake();
+
                                        if(hsrecv.get_hash()==protocol.get_hash())
                                        {
-                                               if(handshake_status==0)
-                                                       send_handshake();
                                                handshake_status = 2;
                                                signal_handshake_done.emit();
                                        }
                                        else
-                                               good = false;
+                                               throw incompatible_protocol("hash mismatch");
                                }
                        }
                }
index af5ff58791ce1ad72a1d97c07591264927bfae64..63114fd9ba18cd09caa2124cca6b2119b6e1a7e8 100644 (file)
@@ -16,6 +16,13 @@ public:
        virtual ~sequence_error() throw() { }
 };
 
+class incompatible_protocol: public std::runtime_error
+{
+public:
+       incompatible_protocol(const std::string &w): std::runtime_error(w) { }
+       virtual ~incompatible_protocol() throw() { }
+};
+
 
 class Communicator
 {
index d862ed82f72a37d7416160307dac7b7069d749bf..4d3f9628f9d1eddbc4a9bc81f9547bba6dbf153b 100644 (file)
@@ -1,12 +1,13 @@
 #include <cstring>
 #include <string>
+#include <msp/core/hash.h>
 #include <msp/core/maputils.h>
+#include <msp/strings/format.h>
 #include <msp/strings/lexicalcast.h>
 #include "protocol.h"
 
 using namespace std;
 
-
 namespace Msp {
 namespace Net {
 
@@ -63,10 +64,12 @@ unsigned Protocol::get_packet_size(const char *buf, unsigned size) const
        return header.length;
 }
 
-unsigned Protocol::get_hash() const
+UInt64 Protocol::get_hash() const
 {
-       // TODO
-       return 123;
+       string description;
+       for(PacketMap::const_iterator i=packet_id_defs.begin(); i!=packet_id_defs.end(); ++i)
+               description += format("%d:%s\n", i->first, i->second->describe());
+       return hash64(description);
 }
 
 
index f6c382c9f1ca003e2dbb982e0d7f59867c3a5ae3..f722a76c82a7cc73b83d4880b7ac8043f45bd889 100644 (file)
@@ -40,6 +40,7 @@ private:
        {
                virtual ~CompoundTypeDef() { }
 
+               virtual std::string describe() const = 0;
                virtual char *serialize(const C &, char *, char *) const = 0;
                virtual const char *deserialize(C &, const char *, const char *) const = 0;
        };
@@ -52,6 +53,7 @@ private:
                CompoundDef(const S &);
                virtual ~CompoundDef();
 
+               virtual std::string describe() const;
                virtual char *serialize(const C &, char *, char *) const;
                virtual const char *deserialize(C &, const char *, const char *) const;
        };
@@ -64,6 +66,7 @@ private:
 
                BasicSerializer(const Protocol &) { }
 
+               std::string describe() const { return get_type_signature<T>(); }
                char *serialize(const T &, char *, char *) const;
                const char *deserialize(T &, const char *, const char *) const;
        };
@@ -79,6 +82,7 @@ private:
        public:
                StringSerializer(const Protocol &);
 
+               std::string describe() const { return get_type_signature<std::string>(); }
                char *serialize(const std::string &, char *, char *) const;
                const char *deserialize(std::string &, const char *, const char *) const;
        };
@@ -96,6 +100,7 @@ private:
        public:
                ArraySerializer(const Protocol &);
 
+               std::string describe() const;
                char *serialize(const A &, char *, char *) const;
                const char *deserialize(A &, const char *, const char *) const;
        };
@@ -119,6 +124,7 @@ private:
        public:
                Serializer(const Head &, Pointer, const Protocol &);
 
+               std::string describe() const;
                char *serialize(const P &, char *, char *) const;
                const char *deserialize(P &, const char *, const char *) const;
        };
@@ -133,6 +139,7 @@ private:
                        typedef Serializer<P, NullSerializer, typename Traits<N>::Serializer> Type;
                };
 
+               std::string describe() const { return std::string(); }
                char *serialize(const P &, char *b, char *) const { return b; }
                const char *deserialize(P &, const char *b, const char *) const { return b; }
        };
@@ -149,6 +156,7 @@ private:
                virtual ~PacketDefBase() { }
                virtual unsigned get_class_id() const = 0;
                unsigned get_id() const { return id; }
+               virtual std::string describe() const = 0;
                virtual const char *dispatch(ReceiverBase &, const char *, const char *) const = 0;
        };
 
@@ -172,6 +180,7 @@ private:
 
                const CompoundTypeDef<P> &get_compound() const { return *compound; }
 
+               virtual std::string describe() const;
                char *serialize(const P &, char *, char *) const;
                const char *deserialize(P &, const char *, const char *) const;
                virtual const char *dispatch(ReceiverBase &, const char *, const char *) const;
@@ -233,7 +242,11 @@ public:
        unsigned get_packet_size(const char *, unsigned) const;
        unsigned dispatch(ReceiverBase &, const char *, unsigned) const;
 
-       unsigned get_hash() const;
+       UInt64 get_hash() const;
+
+private:
+       template<typename T>
+       static std::string get_type_signature();
 };
 
 
@@ -262,32 +275,46 @@ unsigned Protocol::serialize(const P &pkt, char *buf, unsigned size) const
        return size;
 }
 
+template<typename T>
+std::string Protocol::get_type_signature()
+{
+       const UInt16 sig = Traits<T>::signature;
+       std::string result;
+       result += sig&0xFF;
+       if(sig>=0x100)
+               result += '0'+(sig>>8);
+       return result;
+}
+
 
 template<typename T, UInt8 K>
 struct Protocol::BasicTraits
 {
+       static const UInt16 signature = K | (sizeof(T)<<8);
        typedef BasicSerializer<T> Serializer;
 };
 
-template<> struct Protocol::Traits<Int8>: BasicTraits<Int8> { };
-template<> struct Protocol::Traits<UInt8>: BasicTraits<UInt8> { };
-template<> struct Protocol::Traits<Int16>: BasicTraits<Int16> { };
-template<> struct Protocol::Traits<UInt16>: BasicTraits<UInt16> { };
-template<> struct Protocol::Traits<Int32>: BasicTraits<Int32> { };
-template<> struct Protocol::Traits<UInt32>: BasicTraits<UInt32> { };
-template<> struct Protocol::Traits<Int64>: BasicTraits<Int64> { };
-template<> struct Protocol::Traits<UInt64>: BasicTraits<UInt64> { };
-template<> struct Protocol::Traits<float>: BasicTraits<float> { };
-template<> struct Protocol::Traits<double>: BasicTraits<double> { };
+template<> struct Protocol::Traits<Int8>: BasicTraits<Int8, 'I'> { };
+template<> struct Protocol::Traits<UInt8>: BasicTraits<UInt8, 'U'> { };
+template<> struct Protocol::Traits<Int16>: BasicTraits<Int16, 'I'> { };
+template<> struct Protocol::Traits<UInt16>: BasicTraits<UInt16, 'U'> { };
+template<> struct Protocol::Traits<Int32>: BasicTraits<Int32, 'I'> { };
+template<> struct Protocol::Traits<UInt32>: BasicTraits<UInt32, 'U'> { };
+template<> struct Protocol::Traits<Int64>: BasicTraits<Int64, 'I'> { };
+template<> struct Protocol::Traits<UInt64>: BasicTraits<UInt64, 'U'> { };
+template<> struct Protocol::Traits<float>: BasicTraits<float, 'F'> { };
+template<> struct Protocol::Traits<double>: BasicTraits<double, 'F'> { };
 
 template<> struct Protocol::Traits<std::string>
 {
+       static const UInt16 signature = 'S';
        typedef StringSerializer Serializer;
 };
 
 template<typename T>
 struct Protocol::Traits<std::vector<T> >
 {
+       static const UInt16 signature = 'A';
        typedef ArraySerializer<std::vector<T> > Serializer;
 };
 
@@ -304,6 +331,12 @@ Protocol::CompoundDef<C, S>::~CompoundDef()
        delete serializer;
 }
 
+template<typename C, typename S>
+std::string Protocol::CompoundDef<C, S>::describe() const
+{
+       return "{"+serializer->describe()+"}";
+}
+
 template<typename C, typename S>
 char *Protocol::CompoundDef<C, S>::serialize(const C &com, char *buf, char *end) const
 {
@@ -323,6 +356,12 @@ Protocol::ArraySerializer<A>::ArraySerializer(const Protocol &proto):
        element_serializer(proto)
 { }
 
+template<typename A>
+std::string Protocol::ArraySerializer<A>::describe() const
+{
+       return "["+element_serializer.describe()+"]";
+}
+
 template<typename A>
 char *Protocol::ArraySerializer<A>::serialize(const A &array, char *buf, char *end) const
 {
@@ -351,6 +390,12 @@ Protocol::Serializer<P, Head, S>::Serializer(const Head &h, Pointer p, const Pro
        ser(proto)
 { }
 
+template<typename P, typename Head, typename S>
+std::string Protocol::Serializer<P, Head, S>::describe() const
+{
+       return Head::describe()+ser.describe();
+}
+
 template<typename P, typename Head, typename S>
 char * Protocol::Serializer<P, Head, S>::serialize(const P &pkt, char *buf, char *end) const
 {
@@ -392,6 +437,12 @@ void Protocol::PacketTypeDef<P>::set_serializer(const S &ser)
        compound = new CompoundDef<P, S>(ser);
 }
 
+template<typename P>
+std::string Protocol::PacketTypeDef<P>::describe() const
+{
+       return compound->describe();
+}
+
 template<typename P>
 char *Protocol::PacketTypeDef<P>::serialize(const P &pkt, char *buf, char *end) const
 {