|
version 1.1.2.10, 2011/12/29 14:13:13
|
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 55 ConnectClient(int sock)
|
Line 56 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 76 CloseClient(int sock)
|
Line 78 CloseClient(int sock)
|
| return -1; |
return -1; |
| } |
} |
| |
|
| siz = send(sock, args->msg->msg_base, siz, 0); | 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 DISCONNECT %d bytes\n", siz); |
| |
|
| |
shutdown(sock, SHUT_RDWR); |
| close(sock); |
close(sock); |
| return siz; |
return siz; |
| } |
} |
| |
|
| int | void * |
| KAClient(int sock) | OpenFile(void) |
| { |
{ |
| int siz = 0; | int f, siz = 0; |
| struct pollfd pfd; | void *mem; |
| |
|
| siz = mqtt_msgPINGREQ(args->msg); | if (!args->isFile) |
| if (siz == -1) { | return NULL; |
| printf("Error:: msgPINGREQ #%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 PINGREQ %d bytes\n", siz); | 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.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 170 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); | |
| } |
} |