Diff for /mqtt/src/client.c between versions 1.1.2.8 and 1.2.2.4

version 1.1.2.8, 2011/12/20 15:03:42 version 1.2.2.4, 2012/04/27 16:41:56
Line 1 Line 1
 #include "global.h"  #include "global.h"
 #include "mqtt.h"  #include "mqtt.h"
   #include "client.h"
   
   
 int  int
Line 21  InitClient(void) Line 22  InitClient(void)
 }  }
   
 int  int
try2Connect(int sock)ConnectClient(int sock)
 {  {
         int siz = 0;          int siz = 0;
         struct pollfd pfd;          struct pollfd pfd;
Line 43  try2Connect(int sock) Line 44  try2Connect(int sock)
                 ioVERBOSE(3) printf("Sended CONNECT %d bytes\n", siz);                  ioVERBOSE(3) printf("Sended CONNECT %d bytes\n", siz);
   
         pfd.fd = sock;          pfd.fd = sock;
        pfd.events = POLLIN;        pfd.events = POLLIN | POLLPRI;
         switch (poll(&pfd, 1, args->ka * 1000)) {          switch (poll(&pfd, 1, args->ka * 1000)) {
                 case -1:                  case -1:
                         printf("Error:: poll() #%d - %s\n", errno, strerror(errno));                          printf("Error:: poll() #%d - %s\n", errno, strerror(errno));
Line 55  try2Connect(int sock) Line 56  try2Connect(int sock)
         if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL))          if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL))
                 return -1;                  return -1;
   
           memset(args->msg->msg_base, 0, args->msg->msg_len);
         siz = recv(sock, args->msg->msg_base, args->msg->msg_len, 0);          siz = recv(sock, args->msg->msg_base, args->msg->msg_len, 0);
         if (siz == -1) {          if (siz == -1) {
                 printf("Error:: recv() #%d - %s\n", errno, strerror(errno));                  printf("Error:: recv() #%d - %s\n", errno, strerror(errno));
Line 63  try2Connect(int sock) Line 65  try2Connect(int sock)
                 ioVERBOSE(3) printf("Received %d bytes\n", siz);                  ioVERBOSE(3) printf("Received %d bytes\n", siz);
   
         return (u_char) mqtt_readCONNACK(args->msg);          return (u_char) mqtt_readCONNACK(args->msg);
   }
   
   int
   CloseClient(int sock)
   {
           int siz = 0;
   
           siz = mqtt_msgDISCONNECT(args->msg);
           if (siz == -1) {
                   printf("Error:: msgDISCONNECT #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
                   return -1;
           }
   
           siz = send(sock, args->msg->msg_base, siz, MSG_NOSIGNAL);
           if (siz == -1) {
                   printf("Error:: send() #%d - %s\n", errno, strerror(errno));
                   return -1;
           } else
                   ioVERBOSE(3) printf("Sended DISCONNECT %d bytes\n", siz);
   
           shutdown(sock, SHUT_RDWR);
           close(sock);
           return siz;
   }
   
   void *
   OpenFile(void)
   {
           int f, siz = 0;
           void *mem;
   
           if (!args->isFile)
                   return NULL;
   
           f = open(AIT_GET_STR(&args->Value), O_RDONLY);
           if (f == -1) {
                   printf("Error:: in open file #%d - %s\n", errno, strerror(errno));
                   return NULL;
           }
           mem = mmap(NULL, siz, PROT_READ, MAP_PRIVATE, f, 0);
           if (mem == MAP_FAILED) {
                   printf("Error:: in map file #%d - %s\n", errno, strerror(errno));
                   close(f);
                   return NULL;
           } else
                   close(f);
   
           AIT_SET_PTR(&args->Value, mem, siz);
           return mem;
   }
   
   void
   CloseFile(void)
   {
           if (args->isFile) {
                   munmap(AIT_GET_PTR(&args->Value), AIT_LEN(&args->Value));
                   AIT_FREE_VAL(&args->Value);
           }
   }
   
   inline int
   SendTo(int sock, int siz)
   {
           siz = send(sock, args->msg->msg_base, siz, MSG_NOSIGNAL);
           if (siz == -1) {
                   printf("Error:: send() #%d - %s\n", errno, strerror(errno));
                   return -1;
           } else
                   ioVERBOSE(3) printf("Sended %d bytes\n", siz);
   
           return siz;
   }
   
   inline int
   RecvFrom(int sock)
   {
           struct pollfd pfd;
           int siz = 0;
   
           memset(args->msg->msg_base, 0, args->msg->msg_len);
   
           pfd.fd = sock;
           pfd.events = POLLIN | POLLPRI;
           do {
                   switch (poll(&pfd, 1, args->ka * 1000)) {
                           case -1:
                                   printf("Error:: poll() #%d - %s\n", errno, strerror(errno));
                                   return -1;
                           case 0:
                                   ioVERBOSE(3) printf("Timeout reached (%d) ...\n", args->ka * 1000);
                                   if (mqtt_KeepAlive(sock, args->ka, 1) == -1)
                                           return -1;
                                   continue;
                   }
                   if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL))
                           return -1;
           } while (0);
   
           siz = recv(sock, args->msg->msg_base, args->msg->msg_len, 0);
           if (siz == -1) {
                   printf("Error:: recv() #%d - %s\n", errno, strerror(errno));
                   return -1;
           } else
                   ioVERBOSE(3) printf("Received %d bytes\n", siz);
   
           return siz;
 }  }

Removed from v.1.1.2.8  
changed lines
  Added in v.1.2.2.4


FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>