protocols/Message.cc

   1 #include "Message.h"
   2
   3 namespace remote { namespace protocols {
   4
   5 Message::Message()
   6 {
   7         data = NULL;
   8         datalen = 0;
   9         recvLen = 0;
  10 }
  11
  12 Message::~Message()
  13 {
  14         if (data) delete data;
  15         datalen = 0;
  16 }
  17
  18 bool Message::nonBlockingRecv(int fd)
  19 {
  20         char* p_datalen = (char*)&datalen;
  21         int r;
  22         uint32_t targetRecvLen;
  23
  24         if ( recvLen < sizeof(datalen))
  25         {
  26                 // still at the datalen part
  27                 r = read(fd,p_datalen+recvLen,sizeof(datalen)-recvLen);
  28                 if (r <= 0) {
  29                         Log::error("Recv error: %s", strerror(errno));
  30                         __THROW__("Unable to read message length!");
  31                 }
  32                 recvLen += r;
  33         }
  34         if ( recvLen < sizeof(datalen) )
  35         {
  36                 return false;
  37         }
  38         if ( recvLen == sizeof(datalen) )
  39         {
  40                 datalen = ntohl(datalen);
  41                 if (datalen > 512000) __THROW__("Data too long!");
  42                 if (data) delete data;
  43                 data = new uint8_t[datalen];
  44         }
  45         targetRecvLen = datalen + sizeof(datalen);
  46         if ( recvLen >= sizeof(datalen) && recvLen < targetRecvLen )
  47         {
  48                 r = read(fd,data+recvLen-sizeof(datalen),targetRecvLen-recvLen);
  49                 if (r <= 0) {
  50                         Log::error("Recv error: %s", strerror(errno));
  51                         __THROW__ ("Unable to read message data!");
  52                 }
  53                 recvLen += r;
  54 //              printf("Read %u of %u bytes.\n",recvLen-sizeof(datalen),datalen);
  55         }
  56         if (recvLen == targetRecvLen)
  57         {
  58                 recvLen = 0;
  59                 return true;
  60         }
  61         return false;
  62 }
  63
  64
  65 void Message::sendMsg(int fd,BaseMsg& msg)
  66 {
  67         uint32_t len;
  68         if (data) delete data;
  69         // allocate data
  70         datalen = msg.getLength();
  71         data = new uint8_t[datalen];
  72         len = datalen;
  73 //      printf("Writing %u bytes of message data\n",datalen);
  74
  75         uint8_t* d = data;
  76         msg.write(d,len);
  77 //      msg.print(stdout);
  78
  79
  80         // write length, then data
  81         if (!send(fd,datalen))
  82                 __THROW__ ("Unable to send datalen!");
  83         if (::send(fd, (char*)data, datalen, 0) != ((int64_t)datalen))
  84                 __THROW__ ("Unable to send data!");
  85 //      printf("Done writing %u bytes of message data\n\n\n",datalen);
  86 }
  87
  88 uint32_t Message::getLength()
  89 {
  90         return datalen;
  91 }
  92
  93 uint8_t* Message::getData()
  94 {
  95         return data;
  96 }
  97 }}