Annotation of libaitmqtt/src/srvside.c, revision 1.3
1.2 misho 1: /*************************************************************************
2: * (C) 2011 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
3: * by Michael Pounov <misho@openbsd-bg.org>
4: *
5: * $Author: misho $
1.3 ! misho 6: * $Id: srvside.c,v 1.2.4.2 2012/06/29 08:44:34 misho Exp $
1.2 misho 7: *
8: **************************************************************************
9: The ELWIX and AITNET software is distributed under the following
10: terms:
11:
12: All of the documentation and software included in the ELWIX and AITNET
13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
14:
15: Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
16: by Michael Pounov <misho@elwix.org>. All rights reserved.
17:
18: Redistribution and use in source and binary forms, with or without
19: modification, are permitted provided that the following conditions
20: are met:
21: 1. Redistributions of source code must retain the above copyright
22: notice, this list of conditions and the following disclaimer.
23: 2. Redistributions in binary form must reproduce the above copyright
24: notice, this list of conditions and the following disclaimer in the
25: documentation and/or other materials provided with the distribution.
26: 3. All advertising materials mentioning features or use of this software
27: must display the following acknowledgement:
28: This product includes software developed by Michael Pounov <misho@elwix.org>
29: ELWIX - Embedded LightWeight unIX and its contributors.
30: 4. Neither the name of AITNET nor the names of its contributors
31: may be used to endorse or promote products derived from this software
32: without specific prior written permission.
33:
34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
37: ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
44: SUCH DAMAGE.
45: */
46: #include "global.h"
47:
48:
49: /*
1.3 ! misho 50: * mqtt_srv_cliInit() Init MQTT server side support for clients
1.2 misho 51: *
52: * @sock = Client socket from accept()
53: * @buf = Message buffer
54: * @timeout = timeout
1.3 ! misho 55: * @nb = Non block socket
1.2 misho 56: * return: NULL error or !=NULL allocated server handle
57: */
58: mqtt_srv_t *
1.3 ! misho 59: mqtt_srv_cliInit(int sock, mqtt_msg_t * __restrict buf, u_short timeout, int nb)
1.2 misho 60: {
61: mqtt_srv_t *srv = NULL;
1.3 ! misho 62: struct timeval tv = { timeout, 0 };
! 63: int n;
1.2 misho 64:
65: if (!buf || !buf->msg_base || !buf->msg_len)
66: return srv;
1.3 ! misho 67: else
! 68: n = buf->msg_len;
1.2 misho 69:
70: srv = malloc(sizeof(mqtt_srv_t));
71: if (!srv) {
72: LOGERR;
73: return srv;
74: } else
75: memset(srv, 0, sizeof(mqtt_srv_t));
76:
1.3 ! misho 77: /* set options of client socket */
! 78: setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n);
! 79: setsockopt(sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n);
! 80: setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof tv);
! 81: setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof tv);
! 82: ioctl(sock, FIONBIO, nb);
! 83:
1.2 misho 84: srv->sock = sock;
85: srv->buf = buf;
86: srv->timeout = timeout;
87:
88: return srv;
89: }
90:
91: /*
1.3 ! misho 92: * mqtt_srv_cliFini() Finish MQTT server side support for clients
1.2 misho 93: *
94: * @psrv = Server handle
95: * return: none
96: */
97: void
1.3 ! misho 98: mqtt_srv_cliFini(mqtt_srv_t ** __restrict psrv)
1.2 misho 99: {
100: if (psrv && *psrv) {
101: free(*psrv);
102: *psrv = NULL;
103: }
104: }
105:
106: /*
1.3 ! misho 107: * mqtt_srv_cliDispatch() MQTT server dispatcher for clients
1.2 misho 108: *
109: * @srv = Server handle
110: * @rlen = Ready bytes to parse
111: * @arg = Argument pass to command callback
112: * return: -1 error or 0 ok
113: */
114: int
1.3 ! misho 115: mqtt_srv_cliDispatch(mqtt_srv_t * __restrict srv, int rlen, void *arg)
1.2 misho 116: {
117: struct mqtthdr *hdr;
118:
119: if (!srv || !srv->buf->msg_base || !srv->buf->msg_len)
120: return -1;
121: if (srv->buf->msg_len < sizeof(struct mqtthdr) || rlen < sizeof(struct mqtthdr)) {
122: mqtt_SetErr(EINVAL, "Message is too short ...");
123: return -1;
124: }
125: hdr = (struct mqtthdr*) srv->buf->msg_base;
126: if (hdr->mqtt_msg.type > MQTT_TYPE_MAX) {
127: mqtt_SetErr(ENOSYS, "Message type %d not implemented", hdr->mqtt_msg.type);
128: return -1;
129: }
130: if (srv->cmds[hdr->mqtt_msg.type])
131: return srv->cmds[hdr->mqtt_msg.type]((void*) srv, rlen, arg);
132:
133: mqtt_SetErr(ENOTSUP, "Unsupported message type %d", hdr->mqtt_msg.type);
134: return -1;
135: }
1.3 ! misho 136:
! 137:
! 138: /*
! 139: * mqtt_srv_Create() - Create server socket
! 140: *
! 141: * @sa = Server bind address
! 142: * @salen = Server struct sockaddr size
! 143: * return: -1 error or >-1 server socket
! 144: */
! 145: int
! 146: mqtt_srv_Create(struct sockaddr * __restrict sa, int salen)
! 147: {
! 148: int s = -1, n = 1;
! 149:
! 150: if (!sa)
! 151: return -1;
! 152:
! 153: s = socket(sa->sa_family, SOCK_STREAM, 0);
! 154: if (s == -1) {
! 155: LOGERR;
! 156: return -1;
! 157: }
! 158: if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
! 159: LOGERR;
! 160: close(s);
! 161: return -1;
! 162: }
! 163: if (bind(s, sa, salen) == -1) {
! 164: LOGERR;
! 165: close(s);
! 166: return -1;
! 167: }
! 168:
! 169: return s;
! 170: }
! 171:
! 172: /*
! 173: * mqtt_srv_Destroy() - Close server socket
! 174: *
! 175: * @sock = Server socket
! 176: * return: -1 error or 0 ok
! 177: */
! 178: int
! 179: mqtt_srv_Destroy(int sock)
! 180: {
! 181: shutdown(sock, SHUT_RDWR);
! 182: return close(sock);
! 183: }
! 184:
! 185: /*
! 186: * mqtt_srv_Listen() - Listen server socket
! 187: *
! 188: * @sock = Server socket
! 189: * @maxconn = max number of pending connections
! 190: * @nb = Non block socket
! 191: * return: -1 error or 0 ok
! 192: */
! 193: int
! 194: mqtt_srv_Listen(int sock, int maxconn, int nb)
! 195: {
! 196: if (!maxconn)
! 197: maxconn = SOMAXCONN;
! 198:
! 199: if (listen(sock, maxconn) == -1) {
! 200: LOGERR;
! 201: return -1;
! 202: } else
! 203: ioctl(sock, FIONBIO, nb);
! 204:
! 205: return 0;
! 206: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>