#include "global.h"
#include "mqtt.h"
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)
{
int siz = 0;
struct pollfd pfd;
siz = mqtt_msgCONNECT(args->msg, (char*) AIT_GET_STR(&args->ConnID), args->ka,
(char*) AIT_GET_STR(&args->User), (char*) AIT_GET_STR(&args->Pass),
(char*) args->Will.Topic.val.string, (char*) args->Will.Msg.val.string,
!args->notClear, args->QoS, args->Retain);
if (siz == -1) {
printf("Error:: msgCONNECT #%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 CONNECT %d bytes\n", siz);
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 (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);
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;
}
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.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 (u_char) mqtt_readPINGRESP(args->msg);
}
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);
}
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>