version 1.1.2.4, 2011/12/13 14:57:15
|
version 1.5, 2017/10/08 22:49:24
|
Line 1
|
Line 1
|
|
/************************************************************************* |
|
* (C) 2011 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com> |
|
* by Michael Pounov <misho@openbsd-bg.org> |
|
* |
|
* $Author$ |
|
* $Id$ |
|
* |
|
************************************************************************** |
|
The ELWIX and AITNET software is distributed under the following |
|
terms: |
|
|
|
All of the documentation and software included in the ELWIX and AITNET |
|
Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org> |
|
|
|
Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013 |
|
by Michael Pounov <misho@elwix.org>. All rights reserved. |
|
|
|
Redistribution and use in source and binary forms, with or without |
|
modification, are permitted provided that the following conditions |
|
are met: |
|
1. Redistributions of source code must retain the above copyright |
|
notice, this list of conditions and the following disclaimer. |
|
2. Redistributions in binary form must reproduce the above copyright |
|
notice, this list of conditions and the following disclaimer in the |
|
documentation and/or other materials provided with the distribution. |
|
3. All advertising materials mentioning features or use of this software |
|
must display the following acknowledgement: |
|
This product includes software developed by Michael Pounov <misho@elwix.org> |
|
ELWIX - Embedded LightWeight unIX and its contributors. |
|
4. Neither the name of AITNET nor the names of its contributors |
|
may be used to endorse or promote products derived from this software |
|
without specific prior written permission. |
|
|
|
THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND |
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE |
|
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
|
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
|
OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
|
HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
|
LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
|
OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
|
SUCH DAMAGE. |
|
*/ |
#include "global.h" |
#include "global.h" |
#include "mqtt.h" |
#include "mqtt.h" |
|
#include "client.h" |
|
|
|
|
int |
int |
InitClient(void) | ConnectClient(int sock) |
{ |
{ |
int sock; | int siz = 0; |
| struct pollfd pfd; |
|
|
sock = socket(args->addr.sa.sa_family, SOCK_STREAM, IPPROTO_TCP); | siz = mqtt_msgCONNECT(args->msg, (char*) AIT_GET_STR(&args->ConnID), args->ka, |
if (sock == -1) { | (char*) AIT_GET_STR(&args->User), (char*) AIT_GET_STR(&args->Pass), |
printf("Error:: socket() #%d - %s\n", errno, strerror(errno)); | (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; |
return -1; |
} |
} |
if (connect(sock, &args->addr.sa, args->addr.sa.sa_len) == -1) { | |
printf("Error:: connect() #%d - %s\n", errno, strerror(errno)); | siz = send(sock, args->msg->msg_base, siz, 0); |
| if (siz == -1) { |
| printf("Error:: send() #%d - %s\n", errno, strerror(errno)); |
return -1; |
return -1; |
|
} else |
|
EVERBS(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: |
|
EVERBS(3) printf("Timeout reached (%d) ...\n", args->ka * 1000); |
|
return -1; |
} |
} |
|
if (pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) |
|
return -1; |
|
|
return sock; | memset(args->msg->msg_base, 0, args->msg->msg_len); |
| 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 |
| EVERBS(3) printf("Received %d bytes\n", siz); |
| |
| return (u_char) mqtt_readCONNACK(args->msg); |
} |
} |
|
|
int | void * |
try2Connect(int sock) | OpenFile(void) |
{ |
{ |
int siz = 0; | int f, siz = 0; |
| void *mem; |
|
|
siz = mqtt_msgCONNECT(args->msg, AIT_GET_STR(&args->ConnID), AIT_GET_STR(&args->User), | if (!args->isFile) |
AIT_GET_STR(&args->Pass), args->Will.Topic.val.string, | return NULL; |
args->Will.Msg.val.string, args->Clear, args->QoS, args->Retain); | |
if (siz == -1) { | f = open(AIT_GET_STR(&args->Value), O_RDONLY); |
printf("Error:: msgCONNECT #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); | if (f == -1) { |
return -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); |
| } |
| } |
| |
| 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 |
VERB(3) printf("Sended CONNECT %d bytes\n", siz); | EVERBS(3) printf("Sended %d bytes\n", siz); |
| |
| return siz; |
| } |
| |
| 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: |
| EVERBS(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 |
| EVERBS(3) printf("Received %d bytes\n", siz); |
|
|
return siz; |
return siz; |
} |
} |