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