source/net/protocol.cpp

   1 #include <cstring>
   2 #include <string>
   3 #include <msp/core/hash.h>
   4 #include <msp/core/maputils.h>
   5 #include <msp/strings/format.h>
   6 #include <msp/strings/lexicalcast.h>
   7 #include "protocol.h"
   8 #include "protocol_impl.h"
   9
  10 using namespace std;
  11
  12 namespace Msp {
  13 namespace Net {
  14
  15 Protocol::Protocol(unsigned npi):
  16         header_def(0),
  17         next_packet_id(npi)
  18 {
  19         PacketDefBuilder<PacketHeader, NullSerializer<PacketHeader> >(*this, header_def, NullSerializer<PacketHeader>())
  20                 (&PacketHeader::type)(&PacketHeader::length);
  21 }
  22
  23 Protocol::~Protocol()
  24 {
  25         for(map<unsigned, PacketDefBase *>::iterator i=packet_class_defs.begin(); i!=packet_class_defs.end(); ++i)
  26                 delete i->second;
  27 }
  28
  29 void Protocol::add_packet(PacketDefBase *pdef)
  30 {
  31         PacketDefBase *&ptr = packet_class_defs[pdef->get_class_id()];
  32         if(ptr)
  33         {
  34                 packet_id_defs.erase(ptr->get_id());
  35                 delete ptr;
  36         }
  37         ptr = pdef;
  38         if(unsigned id = pdef->get_id())
  39                 packet_id_defs[id] = pdef;
  40 }
  41
  42 const Protocol::PacketDefBase &Protocol::get_packet_by_class_id(unsigned id) const
  43 {
  44         return *get_item(packet_class_defs, id);
  45 }
  46
  47 const Protocol::PacketDefBase &Protocol::get_packet_by_id(unsigned id) const
  48 {
  49         return *get_item(packet_id_defs, id);
  50 }
  51
  52 unsigned Protocol::dispatch(ReceiverBase &rcv, const char *buf, unsigned size) const
  53 {
  54         PacketHeader header;
  55         buf = header_def.deserialize(header, buf, buf+size);
  56         if(header.length>size)
  57                 throw bad_packet("truncated");
  58         const PacketDefBase &pdef = get_packet_by_id(header.type);
  59         const char *ptr = pdef.dispatch(rcv, buf, buf+header.length);
  60         return ptr-buf;
  61 }
  62
  63 unsigned Protocol::get_packet_size(const char *buf, unsigned size) const
  64 {
  65         if(size<4)
  66                 return 0;
  67         PacketHeader header;
  68         header_def.deserialize(header, buf, buf+size);
  69         return header.length;
  70 }
  71
  72 UInt64 Protocol::get_hash() const
  73 {
  74         string description;
  75         for(PacketMap::const_iterator i=packet_id_defs.begin(); i!=packet_id_defs.end(); ++i)
  76                 description += format("%d:%s\n", i->first, i->second->describe());
  77         return hash64(description);
  78 }
  79
  80
  81 /* TODO These assumes the machine is little-endian; are there any relevant
  82 big-endian platforms these days? */
  83 template<typename T>
  84 char *Protocol::BasicSerializer<T>::serialize(const T &value, char *buf, char *end) const
  85 {
  86         if(end-buf<static_cast<int>(sizeof(T)))
  87                 throw buffer_error("overflow");
  88
  89         const char *ptr = reinterpret_cast<const char *>(&value)+sizeof(T);
  90         for(unsigned i=0; i<sizeof(T); ++i)
  91                 *buf++ = *--ptr;
  92
  93         return buf;
  94 }
  95
  96 template<typename T>
  97 const char *Protocol::BasicSerializer<T>::deserialize(T &value, const char *buf, const char *end) const
  98 {
  99         if(end-buf<static_cast<int>(sizeof(T)))
 100                 throw buffer_error("underflow");
 101
 102         char *ptr = reinterpret_cast<char *>(&value)+sizeof(T);
 103         for(unsigned i=0; i<sizeof(T); ++i)
 104                 *--ptr = *buf++;
 105
 106         return buf;
 107 }
 108
 109 template char *Protocol::BasicSerializer<Int8>::serialize(const Int8 &, char *, char *) const;
 110 template char *Protocol::BasicSerializer<Int16>::serialize(const Int16 &, char *, char *) const;
 111 template char *Protocol::BasicSerializer<Int32>::serialize(const Int32 &, char *, char *) const;
 112 template char *Protocol::BasicSerializer<Int64>::serialize(const Int64 &, char *, char *) const;
 113 template char *Protocol::BasicSerializer<UInt8>::serialize(const UInt8 &, char *, char *) const;
 114 template char *Protocol::BasicSerializer<UInt16>::serialize(const UInt16 &, char *, char *) const;
 115 template char *Protocol::BasicSerializer<UInt32>::serialize(const UInt32 &, char *, char *) const;
 116 template char *Protocol::BasicSerializer<UInt64>::serialize(const UInt64 &, char *, char *) const;
 117 template char *Protocol::BasicSerializer<float>::serialize(const float &, char *, char *) const;
 118 template char *Protocol::BasicSerializer<double>::serialize(const double &, char *, char *) const;
 119 template const char *Protocol::BasicSerializer<Int8>::deserialize(Int8 &, const char *, const char *) const;
 120 template const char *Protocol::BasicSerializer<Int16>::deserialize(Int16 &, const char *, const char *) const;
 121 template const char *Protocol::BasicSerializer<Int32>::deserialize(Int32 &, const char *, const char *) const;
 122 template const char *Protocol::BasicSerializer<Int64>::deserialize(Int64 &, const char *, const char *) const;
 123 template const char *Protocol::BasicSerializer<UInt8>::deserialize(UInt8 &, const char *, const char *) const;
 124 template const char *Protocol::BasicSerializer<UInt16>::deserialize(UInt16 &, const char *, const char *) const;
 125 template const char *Protocol::BasicSerializer<UInt32>::deserialize(UInt32 &, const char *, const char *) const;
 126 template const char *Protocol::BasicSerializer<UInt64>::deserialize(UInt64 &, const char *, const char *) const;
 127 template const char *Protocol::BasicSerializer<float>::deserialize(float &, const char *, const char *) const;
 128 template const char *Protocol::BasicSerializer<double>::deserialize(double &, const char *, const char *) const;
 129
 130
 131 Protocol::StringSerializer::StringSerializer(const Protocol &p):
 132         length_serializer(p)
 133 { }
 134
 135 char *Protocol::StringSerializer::serialize(const string &str, char *buf, char *end) const
 136 {
 137         buf = length_serializer.serialize(str.size(), buf, end);
 138         if(end-buf<static_cast<int>(str.size()))
 139                 throw buffer_error("overflow");
 140         copy(str.begin(), str.end(), buf);
 141         return buf+str.size();
 142 }
 143
 144 const char *Protocol::StringSerializer::deserialize(string &str, const char *buf, const char *end) const
 145 {
 146         UInt16 length;
 147         buf = length_serializer.deserialize(length, buf, end);
 148         if(end-buf<static_cast<int>(length))
 149                 throw buffer_error("underflow");
 150         str.assign(buf, buf+length);
 151         return buf+length;
 152 }
 153
 154
 155 unsigned Protocol::PacketDefBase::next_class_id = 1;
 156
 157 Protocol::PacketDefBase::PacketDefBase(unsigned i):
 158         id(i)
 159 { }
 160
 161
 162 Protocol::PacketHeader::PacketHeader():
 163         type(0),
 164         length(0)
 165 { }
 166
 167 Protocol::PacketHeader::PacketHeader(UInt16 t, UInt16 l):
 168         type(t),
 169         length(l)
 170 { }
 171
 172 } // namespace Net
 173 } // namespace Msp