version 1.1.2.8, 2011/12/20 15:03:42
|
version 1.2.2.1, 2012/01/30 22:58:24
|
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 63 try2Connect(int sock)
|
Line 64 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, 0); |
|
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; |
|
} |
|
|
|
int |
|
KAClient(int sock) |
|
{ |
|
int siz = 0; |
|
struct pollfd pfd; |
|
|
|
siz = mqtt_msgPINGREQ(args->msg); |
|
if (siz == -1) { |
|
printf("Error:: msgPINGREQ #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); |
|
return -1; |
|
} |
|
if (SendTo(sock, siz) == -1) |
|
return -1; |
|
|
|
pfd.fd = sock; |
|
pfd.events = POLLIN | POLLPRI; |
|
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); |
|
return -1; |
|
} |
|
if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) |
|
return -1; |
|
|
|
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 mqtt_readPINGRESP(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, 0); |
|
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; |
|
|
|
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 (KAClient(sock) == -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; |
} |
} |