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

version 1.1.2.8, 2011/12/20 15:03:42 version 1.3, 2012/07/03 09:02:50
Line 1 Line 1
 #include "global.h"  #include "global.h"
 #include "mqtt.h"  #include "mqtt.h"
   #include "client.h"
   
   
 int  int
InitClient(void)ConnectClient(int sock)
 {  {
         int sock;  
   
         sock = socket(args->addr.sa.sa_family, SOCK_STREAM, IPPROTO_TCP);  
         if (sock == -1) {  
                 printf("Error:: socket() #%d - %s\n", errno, strerror(errno));  
                 return -1;  
         }  
         if (connect(sock, &args->addr.sa, args->addr.sa.sa_len) == -1) {  
                 printf("Error:: connect() #%d - %s\n", errno, strerror(errno));  
                 return -1;  
         }  
   
         return sock;  
 }  
   
 int  
 try2Connect(int sock)  
 {  
         int siz = 0;          int siz = 0;
         struct pollfd pfd;          struct pollfd pfd;
   
Line 43  try2Connect(int sock) Line 26  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 38  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 47  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);
   }
   
   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.3


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