--- libaitmqtt/src/srvside.c 2012/06/20 15:02:24 1.2 +++ libaitmqtt/src/srvside.c 2012/06/29 13:54:48 1.3 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: srvside.c,v 1.2 2012/06/20 15:02:24 misho Exp $ +* $Id: srvside.c,v 1.3 2012/06/29 13:54:48 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -47,20 +47,25 @@ SUCH DAMAGE. /* - * mqtt_srv_Init() Init MQTT server side support + * mqtt_srv_cliInit() Init MQTT server side support for clients * * @sock = Client socket from accept() * @buf = Message buffer * @timeout = timeout + * @nb = Non block socket * return: NULL error or !=NULL allocated server handle */ mqtt_srv_t * -mqtt_srv_Init(int sock, mqtt_msg_t * __restrict buf, u_short timeout) +mqtt_srv_cliInit(int sock, mqtt_msg_t * __restrict buf, u_short timeout, int nb) { mqtt_srv_t *srv = NULL; + struct timeval tv = { timeout, 0 }; + int n; if (!buf || !buf->msg_base || !buf->msg_len) return srv; + else + n = buf->msg_len; srv = malloc(sizeof(mqtt_srv_t)); if (!srv) { @@ -69,6 +74,13 @@ mqtt_srv_Init(int sock, mqtt_msg_t * __restrict buf, u } else memset(srv, 0, sizeof(mqtt_srv_t)); + /* set options of client socket */ + setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n); + setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n); + setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof tv); + setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof tv); + ioctl(sock, FIONBIO, nb); + srv->sock = sock; srv->buf = buf; srv->timeout = timeout; @@ -77,13 +89,13 @@ mqtt_srv_Init(int sock, mqtt_msg_t * __restrict buf, u } /* - * mqtt_srv_Fini() Finish MQTT server side support + * mqtt_srv_cliFini() Finish MQTT server side support for clients * * @psrv = Server handle * return: none */ void -mqtt_srv_Fini(mqtt_srv_t ** __restrict psrv) +mqtt_srv_cliFini(mqtt_srv_t ** __restrict psrv) { if (psrv && *psrv) { free(*psrv); @@ -92,7 +104,7 @@ mqtt_srv_Fini(mqtt_srv_t ** __restrict psrv) } /* - * mqtt_srv_Dispatch() MQTT server dispatcher + * mqtt_srv_cliDispatch() MQTT server dispatcher for clients * * @srv = Server handle * @rlen = Ready bytes to parse @@ -100,7 +112,7 @@ mqtt_srv_Fini(mqtt_srv_t ** __restrict psrv) * return: -1 error or 0 ok */ int -mqtt_srv_Dispatch(mqtt_srv_t * __restrict srv, int rlen, void *arg) +mqtt_srv_cliDispatch(mqtt_srv_t * __restrict srv, int rlen, void *arg) { struct mqtthdr *hdr; @@ -120,4 +132,75 @@ mqtt_srv_Dispatch(mqtt_srv_t * __restrict srv, int rle mqtt_SetErr(ENOTSUP, "Unsupported message type %d", hdr->mqtt_msg.type); return -1; +} + + +/* + * mqtt_srv_Create() - Create server socket + * + * @sa = Server bind address + * @salen = Server struct sockaddr size + * return: -1 error or >-1 server socket + */ +int +mqtt_srv_Create(struct sockaddr * __restrict sa, int salen) +{ + int s = -1, n = 1; + + if (!sa) + return -1; + + s = socket(sa->sa_family, SOCK_STREAM, 0); + if (s == -1) { + LOGERR; + return -1; + } + if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) { + LOGERR; + close(s); + return -1; + } + if (bind(s, sa, salen) == -1) { + LOGERR; + close(s); + return -1; + } + + return s; +} + +/* + * mqtt_srv_Destroy() - Close server socket + * + * @sock = Server socket + * return: -1 error or 0 ok + */ +int +mqtt_srv_Destroy(int sock) +{ + shutdown(sock, SHUT_RDWR); + return close(sock); +} + +/* + * mqtt_srv_Listen() - Listen server socket + * + * @sock = Server socket + * @maxconn = max number of pending connections + * @nb = Non block socket + * return: -1 error or 0 ok + */ +int +mqtt_srv_Listen(int sock, int maxconn, int nb) +{ + if (!maxconn) + maxconn = SOMAXCONN; + + if (listen(sock, maxconn) == -1) { + LOGERR; + return -1; + } else + ioctl(sock, FIONBIO, nb); + + return 0; }