Annotation of libaitio/src/sock.c, revision 1.4.4.3
1.2 misho 1: /*************************************************************************
2: * (C) 2013 AITNET ltd - Sofia/Bulgaria - <misho@aitnet.org>
3: * by Michael Pounov <misho@elwix.org>
4: *
5: * $Author: misho $
1.4.4.3 ! misho 6: * $Id: sock.c,v 1.4.4.2 2013/11/21 14:43:53 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, 2013
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: /*
50: * ioInitSocket() - Init socket and allocate resources
51: *
52: * @role = Socket role
53: * @type = Socket type
54: * @proto = Socket protocol
55: * @addr = Bind to address
56: * @port = Bind to port
57: * @buflen = Socket buffer, optional if =0 == BUFSIZ
58: * return: NULL error or !=NULL created socket
59: */
60: sock_t *
61: ioInitSocket(int role, int type, int proto, const char *addr, u_short port, size_t buflen)
62: {
63: sock_t *s = NULL;
64: int n = 1;
65:
66: if (!addr)
67: return NULL;
68:
69: s = e_malloc(sizeof(sock_t));
70: if (!s) {
71: io_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
72: return NULL;
73: } else
74: memset(s, 0, sizeof(sock_t));
75:
1.3 misho 76: TAILQ_INIT(&s->sock_cli);
77:
1.2 misho 78: s->sock_role = role;
79: s->sock_type = type;
80: s->sock_proto = proto;
81: if (!e_gethostbyname(addr, port, &s->sock_addr)) {
82: io_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
83: e_free(s);
84: return NULL;
85: } else {
86: buflen = buflen ? buflen : BUFSIZ;
1.4.4.1 misho 87: buflen = E_ALIGN(buflen, 2); /* align buflen length */
1.2 misho 88: AIT_SET_BUFSIZ(&s->sock_buf, 0, buflen);
89: }
90:
91: s->sock_fd = socket(s->sock_addr.sa.sa_family, s->sock_type, s->sock_proto);
92: if (s->sock_fd == -1) {
93: LOGERR;
94: AIT_FREE_VAL(&s->sock_buf);
95: e_free(s);
96: return NULL;
97: }
98: if (setsockopt(s->sock_fd, SOL_SOCKET, SO_SNDBUF, &buflen, sizeof buflen) == -1) {
99: LOGERR;
100: AIT_FREE_VAL(&s->sock_buf);
101: e_free(s);
102: return NULL;
103: }
104: if (setsockopt(s->sock_fd, SOL_SOCKET, SO_RCVBUF, &buflen, sizeof buflen) == -1) {
105: LOGERR;
106: AIT_FREE_VAL(&s->sock_buf);
107: e_free(s);
108: return NULL;
109: }
110: if (setsockopt(s->sock_fd, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
111: LOGERR;
112: AIT_FREE_VAL(&s->sock_buf);
113: e_free(s);
114: return NULL;
115: }
116: if (bind(s->sock_fd, &s->sock_addr.sa, s->sock_addr.sa.sa_len) == -1) {
117: LOGERR;
118: AIT_FREE_VAL(&s->sock_buf);
119: e_free(s);
120: return NULL;
121: }
122:
1.4.4.1 misho 123: s->sock_root = schedBegin();
124: if (!s->sock_root) {
125: io_SetErr(sched_GetErrno(), "%s", sched_GetError());
126: AIT_FREE_VAL(&s->sock_buf);
127: e_free(s);
128: return NULL;
129: }
130:
1.3 misho 131: pthread_mutex_init(&s->sock_mtx, NULL);
1.2 misho 132: return s;
133: }
134:
135: /*
136: * ioCloseSocket() - Close socket and free resources
137: *
138: * @s = Socket
139: * return: none
140: */
141: void
142: ioCloseSocket(sock_t ** __restrict s)
143: {
1.3 misho 144: sock_cli_t *cli;
145:
1.2 misho 146: if (s && *s) {
1.3 misho 147: pthread_mutex_lock(&(*s)->sock_mtx);
148: while ((cli = TAILQ_FIRST(&(*s)->sock_cli))) {
149: TAILQ_REMOVE(&(*s)->sock_cli, cli, cli_node);
150: shutdown(cli->cli_fd, SHUT_RDWR);
151: close(cli->cli_fd);
152: AIT_FREE_VAL(&cli->cli_buf);
153: e_free(cli);
154: }
155: pthread_mutex_unlock(&(*s)->sock_mtx);
156:
1.2 misho 157: shutdown((*s)->sock_fd, SHUT_RDWR);
158: close((*s)->sock_fd);
159:
160: AIT_FREE_VAL(&(*s)->sock_buf);
1.3 misho 161:
1.4.4.1 misho 162: schedEnd(&(*s)->sock_root);
163:
1.3 misho 164: pthread_mutex_destroy(&(*s)->sock_mtx);
1.2 misho 165: e_free(*s);
166: *s = NULL;
167: }
168: }
169:
170: /*
171: * ioUpSocket() - Setup socket for use
172: *
173: * @s = Socket
174: * @arg = Server role = listen backlog queue and Client role = peer address
1.4.4.1 misho 175: * @timeout = Socket timeout in sec (default -1 infinit)
1.2 misho 176: * return: -1 error or 0 ok
177: */
178: int
1.4.4.1 misho 179: ioUpSocket(sock_t * __restrict s, void *arg, int timeout)
1.2 misho 180: {
181: int ret = 0;
182: sockaddr_t *peer = (sockaddr_t*) arg;
183: uintptr_t backlog = (uintptr_t) arg;
184:
185: if (!s || !arg)
186: return -1;
1.4.4.1 misho 187: else {
188: s->sock_timeout.tv_sec = timeout;
189: s->sock_timeout.tv_nsec = (timeout < 1) ? timeout : 0;
190: schedPolling(s->sock_root, &s->sock_timeout, NULL);
191: }
1.2 misho 192:
193: switch (s->sock_role) {
194: case IO_SOCK_ROLE_CLIENT:
195: memcpy(&s->sock_peer, peer, sizeof s->sock_peer);
196:
197: if (connect(s->sock_fd, &s->sock_peer.sa,
198: s->sock_peer.sa.sa_len) == -1) {
199: LOGERR;
200: return -1;
201: }
202: break;
203: case IO_SOCK_ROLE_SERVER:
204: if (s->sock_type == SOCK_STREAM) {
205: s->sock_backq = backlog;
206:
207: if (listen(s->sock_fd, s->sock_backq) == -1) {
208: LOGERR;
209: return -1;
210: }
211: }
212: break;
213: default:
214: io_SetErr(EINVAL, "Unsupported socket type");
215: return -1;
216: }
217:
218: fcntl(s->sock_fd, F_SETFL, fcntl(s->sock_fd, F_GETFL) | O_NONBLOCK);
219: return ret;
220: }
1.3 misho 221:
1.4.4.1 misho 222: static void *
1.4.4.2 misho 223: io_closeClient(sched_task_t *task)
1.4.4.1 misho 224: {
1.4.4.2 misho 225: sock_cli_t *cli = (sock_cli_t*) TASK_ARG(task);
1.3 misho 226: sock_t *s = (sock_t*) cli->cli_parent;
1.4.4.2 misho 227: int sock = (int) TASK_DATLEN(task);
1.3 misho 228:
229: if (s->sock_type == SOCK_STREAM) {
1.4.4.2 misho 230: shutdown(sock, SHUT_RDWR);
231: close(sock);
1.3 misho 232: }
233: AIT_FREE_VAL(&cli->cli_buf);
234:
235: pthread_mutex_lock(&s->sock_mtx);
236: TAILQ_REMOVE(&s->sock_cli, cli, cli_node);
237: pthread_mutex_unlock(&s->sock_mtx);
238:
239: e_free(cli);
1.4.4.2 misho 240: taskExit(task, NULL);
1.3 misho 241: }
242:
243: static void *
1.4.4.2 misho 244: io_acceptClient(sched_task_t *task)
1.3 misho 245: {
1.4.4.2 misho 246: int c, rlen;
247: sockaddr_t sa;
248: socklen_t salen = sizeof sa.ss;
249: sock_cli_t *cli = NULL;
250: sock_t *s = (sock_t*) TASK_ARG(task);
1.3 misho 251:
1.4.4.2 misho 252: if (s->sock_type == SOCK_STREAM) {
253: if ((c = accept(TASK_FD(task), &sa.sa, &salen)) == -1) {
254: LOGERR;
255: goto end;
256: }
257: } else {
258: if ((rlen = recvfrom(TASK_FD(task),
259: AIT_GET_BUF(&s->sock_buf), AIT_LEN(&s->sock_buf),
260: MSG_PEEK, &sa.sa, &salen)) == -1) {
261: LOGERR;
262: goto end;
263: } else
264: c = TASK_FD(task);
265: }
1.3 misho 266:
1.4.4.2 misho 267: cli = e_malloc(sizeof(sock_cli_t));
268: if (!cli) {
269: io_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
270: if (s->sock_type == SOCK_STREAM)
271: close(c);
272: goto end;
273: } else {
274: memset(cli, 0, sizeof(sock_cli_t));
275: pthread_mutex_lock(&s->sock_mtx);
276: TAILQ_INSERT_TAIL(&s->sock_cli, cli, cli_node);
277: pthread_mutex_unlock(&s->sock_mtx);
278: }
1.3 misho 279:
1.4.4.2 misho 280: cli->cli_parent = TASK_ARG(task);
281: cli->cli_fd = c;
282: cli->cli_func = TASK_DATA(task);
283: memcpy(&cli->cli_addr, &sa, sizeof cli->cli_addr);
284: AIT_SET_BUFSIZ(&cli->cli_buf, 0, AIT_LEN(&s->sock_buf));
285:
286: schedRead(TASK_ROOT(task), cli->cli_func, cli, cli->cli_fd, TASK_ARG(task), 0);
287: ioUpdTimerSocket(cli);
288: end:
289: schedReadSelf(task);
290: taskExit(task, NULL);
1.3 misho 291: }
292:
293: /*
1.4.4.2 misho 294: * ioUpdTimerSocket() - Update timeout of socket
1.3 misho 295: *
1.4.4.2 misho 296: * @c = Client socket
297: * return: none
1.3 misho 298: */
1.4.4.2 misho 299: void
300: ioUpdTimerSocket(sock_cli_t * __restrict c)
1.3 misho 301: {
1.4.4.2 misho 302: sock_t *s;
1.3 misho 303:
1.4.4.2 misho 304: if (!c)
305: return;
306: else
307: s = c->cli_parent;
1.3 misho 308:
1.4.4.2 misho 309: schedCancelby(s->sock_root, taskTIMER, CRITERIA_DATLEN, (void*) c->cli_fd, NULL);
310: schedTimer(s->sock_root, io_closeClient, c, s->sock_timeout, NULL, c->cli_fd);
311: }
1.3 misho 312:
1.4.4.2 misho 313: /*
314: * ioLoopSocket() - Start socket scheduler
315: *
316: * @s = Socket
317: * @rcb = Read callback
318: * return: -1 error or return result from scheduler
319: */
320: int
321: ioLoopSocket(sock_t * __restrict s, sched_task_func_t rcb)
322: {
323: schedRead(s->sock_root, io_acceptClient, s, s->sock_fd, rcb, 0);
324: return schedRun(s->sock_root, &s->sock_kill);
1.3 misho 325: }
1.4.4.3 ! misho 326:
! 327: /*
! 328: * ioCloseClient() - Close client socket
! 329: *
! 330: * @c = Client socket
! 331: * return: 0 ok or !=0 error
! 332: */
! 333: int
! 334: ioCloseClient(sock_cli_t * __restrict c)
! 335: {
! 336: sock_t *s;
! 337:
! 338: if (!c)
! 339: return -1;
! 340: else
! 341: s = c->cli_parent;
! 342:
! 343: return !schedEvent(s->sock_root, io_closeClient, c, 0, NULL, c->cli_fd);
! 344: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>