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

version 1.1.2.10, 2011/12/29 14:13:13 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)  
 {  
         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  
 ConnectClient(int sock)  ConnectClient(int sock)
 {  {
         int siz = 0;          int siz = 0;
Line 55  ConnectClient(int sock) Line 38  ConnectClient(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 65  ConnectClient(int sock) Line 49  ConnectClient(int sock)
         return (u_char) mqtt_readCONNACK(args->msg);          return (u_char) mqtt_readCONNACK(args->msg);
 }  }
   
intvoid *
CloseClient(int sock)OpenFile(void)
 {  {
        int siz = 0;        int f, siz = 0;
         void *mem;
   
        siz = mqtt_msgDISCONNECT(args->msg);        if (!args->isFile)
        if (siz == -1) {                return NULL;
                printf("Error:: msgDISCONNECT #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
                return -1;        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);
   
        siz = send(sock, args->msg->msg_base, siz, 0);        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) {          if (siz == -1) {
                 printf("Error:: send() #%d - %s\n", errno, strerror(errno));                  printf("Error:: send() #%d - %s\n", errno, strerror(errno));
                 return -1;                  return -1;
         } else          } else
                ioVERBOSE(3) printf("Sended DISCONNECT %d bytes\n", siz);                ioVERBOSE(3) printf("Sended %d bytes\n", siz);
   
         close(sock);  
         return siz;          return siz;
 }  }
   
intinline int
KAClient(int sock)RecvFrom(int sock)
 {  {
         int siz = 0;  
         struct pollfd pfd;          struct pollfd pfd;
           int siz = 0;
   
        siz = mqtt_msgPINGREQ(args->msg);        memset(args->msg->msg_base, 0, args->msg->msg_len);
        if (siz == -1) { 
                printf("Error:: msgPINGREQ #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); 
                return -1; 
        } 
   
         siz = send(sock, args->msg->msg_base, siz, 0);  
         if (siz == -1) {  
                 printf("Error:: send() #%d - %s\n", errno, strerror(errno));  
                 return -1;  
         } else  
                 ioVERBOSE(3) printf("Sended PINGREQ %d bytes\n", siz);  
   
         pfd.fd = sock;          pfd.fd = sock;
         pfd.events = POLLIN | POLLPRI;          pfd.events = POLLIN | POLLPRI;
        switch (poll(&pfd, 1, args->ka * 1000)) {        do {
                case -1:                switch (poll(&pfd, 1, args->ka * 1000)) {
                        printf("Error:: poll() #%d - %s\n", errno, strerror(errno));                        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;                          return -1;
                case 0:        } while (0);
                        ioVERBOSE(3) printf("Timeout reached (%d) ...\n", args->ka * 1000); 
                        return -1; 
        } 
        if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) 
                return -1; 
   
         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) {
Line 126  KAClient(int sock) Line 129  KAClient(int sock)
         } else          } else
                 ioVERBOSE(3) printf("Received %d bytes\n", siz);                  ioVERBOSE(3) printf("Received %d bytes\n", siz);
   
        return (u_char) mqtt_readPINGRESP(args->msg);        return siz;
} 
 
FILE * 
OpenFile(void) 
{ 
        if (!args->isFile) 
                return NULL; 
 
        args->isFile = fopen(AIT_GET_STR(&args->Value), "r"); 
        if (!args->isFile) { 
                printf("Error:: in open file #%d - %s\n", errno, strerror(errno)); 
                return NULL; 
        } 
 
        return args->isFile; 
} 
 
void 
CloseFile(void) 
{ 
        if (args->isFile) 
                fclose(args->isFile); 
 }  }

Removed from v.1.1.2.10  
changed lines
  Added in v.1.3


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