Annotation of libaitio/src/sock.c, revision 1.15.4.1

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.15.4.1! misho       6: * $Id: sock.c,v 1.15 2014/05/18 23:35:16 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: 
1.15.4.1! misho      15: Copyright 2004 - 2016
1.2       misho      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: 
1.5       misho      49: static void *
                     50: io_closeClient(sched_task_t *task)
                     51: {
                     52:        sock_cli_t *cli = (sock_cli_t*) TASK_ARG(task);
                     53:        sock_t *s = (sock_t*) cli->cli_parent;
1.7       misho      54:        int stat;
1.5       misho      55: 
1.15      misho      56:        schedCancelby(s->sock_root, taskMAX, CRITERIA_ARG, cli, NULL);
                     57: 
1.5       misho      58:        pthread_mutex_lock(&s->sock_mtx);
                     59:        TAILQ_REMOVE(&s->sock_cli, cli, cli_node);
                     60:        pthread_mutex_unlock(&s->sock_mtx);
                     61: 
1.9       misho      62:        if (*cli->cli_name)
                     63:                ioFreePTY(cli->cli_pty, cli->cli_name);
1.12      misho      64:        if (s->sock_prog) {
                     65:                io_progDetach(s->sock_prog, cli->cli_pty);
                     66: #if 0
                     67:                io_progCloseAt(s->sock_prog, cli->cli_pty);
                     68: #endif
                     69:                cli->cli_pty ^= cli->cli_pty;
                     70:                io_progCheck(s->sock_prog, 42);
                     71:        }
1.8       misho      72: 
1.5       misho      73:        if (s->sock_type == SOCK_STREAM) {
1.7       misho      74:                shutdown(cli->cli_fd, SHUT_RDWR);
                     75:                close(cli->cli_fd);
1.5       misho      76:        }
                     77:        AIT_FREE_VAL(&cli->cli_buf[1]);
                     78:        AIT_FREE_VAL(&cli->cli_buf[0]);
                     79: 
                     80:        if (cli->cli_pid > 0) {
1.7       misho      81:                kill(cli->cli_pid, SIGKILL);
1.5       misho      82:                while (waitpid(cli->cli_pid, &stat, WNOHANG) > 0) {
                     83:                        usleep(1000);
1.7       misho      84:                        kill(cli->cli_pid, SIGKILL);
1.5       misho      85:                }
                     86:        }
                     87: 
                     88:        e_free(cli);
                     89:        taskExit(task, NULL);
                     90: }
                     91: 
                     92: static void *
                     93: io_acceptClient(sched_task_t *task)
                     94: {
                     95:        int c, rlen;
                     96:        sockaddr_t sa;
                     97:        socklen_t salen = sizeof sa.ss;
                     98:        sock_cli_t *cli = NULL;
                     99:        sock_t *s = (sock_t*) TASK_ARG(task);
                    100: 
                    101:        if (s->sock_type == SOCK_STREAM) {
                    102:                if ((c = accept(TASK_FD(task), &sa.sa, &salen)) == -1) {
                    103:                        LOGERR;
                    104:                        goto end;
                    105:                }
                    106:        } else {
                    107:                if ((rlen = recvfrom(TASK_FD(task), 
                    108:                                                AIT_GET_BUF(&s->sock_buf), AIT_LEN(&s->sock_buf), 
                    109:                                                MSG_PEEK, &sa.sa, &salen)) == -1) {
                    110:                        LOGERR;
                    111:                        goto end;
                    112:                } else
                    113:                        c = TASK_FD(task);
                    114:        }
                    115: 
                    116:        cli = e_malloc(sizeof(sock_cli_t));
                    117:        if (!cli) {
                    118:                io_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                    119:                if (s->sock_type == SOCK_STREAM)
                    120:                        close(c);
                    121:                goto end;
                    122:        } else {
                    123:                memset(cli, 0, sizeof(sock_cli_t));
                    124:                pthread_mutex_lock(&s->sock_mtx);
                    125:                TAILQ_INSERT_TAIL(&s->sock_cli, cli, cli_node);
                    126:                pthread_mutex_unlock(&s->sock_mtx);
                    127:        }
                    128: 
                    129:        cli->cli_parent = TASK_ARG(task);
                    130:        cli->cli_fd = c;
                    131:        cli->cli_func = TASK_DATA(task);
                    132:        memcpy(&cli->cli_addr, &sa, sizeof cli->cli_addr);
                    133:        AIT_SET_BUFSIZ(&cli->cli_buf[0], 0, AIT_LEN(&s->sock_buf));
                    134:        AIT_SET_BUFSIZ(&cli->cli_buf[1], 0, AIT_LEN(&s->sock_buf));
                    135: 
                    136:        schedRead(TASK_ROOT(task), cli->cli_func, cli, cli->cli_fd, TASK_ARG(task), 0);
1.7       misho     137:        ioUpdTimerSocket(cli);
1.5       misho     138: end:
                    139:        schedReadSelf(task);
                    140:        taskExit(task, NULL);
                    141: }
                    142: 
                    143: static void *
                    144: io_txNet(sched_task_t *task)
                    145: {
1.10      misho     146:        int wlen, ret, len = TASK_DATLEN(task);
1.5       misho     147:        sock_cli_t *cli = TASK_ARG(task);
                    148:        sock_t *s = (sock_t*) cli->cli_parent;
1.10      misho     149:        u_char *buf = TASK_DATA(task);
                    150:        struct pollfd pfd[1];
1.5       misho     151: 
1.10      misho     152:        pfd->fd = TASK_FD(task);
                    153:        pfd->events = POLLOUT;
                    154:        pfd->revents = 0;
                    155:        for(; len > 0; len -= wlen, buf += wlen) {
                    156:                ioUpdTimerSocket(cli);
                    157: 
                    158:                if ((ret = poll(pfd, 1, s->sock_timeout.tv_sec * 1000)) < 1 || 
                    159:                                pfd->revents & (POLLNVAL | POLLERR | POLLHUP)) {
1.12      misho     160: #if 0
1.10      misho     161:                        if (!ret)
                    162:                                continue;
1.12      misho     163: #endif
1.10      misho     164:                        schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
                    165:                        break;
                    166:                }
1.5       misho     167: 
1.10      misho     168:                if (s->sock_type == SOCK_STREAM)
                    169:                        wlen = send(TASK_FD(task), buf, len, 0);
                    170:                else
1.15.4.1! misho     171: #ifndef __linux__
        !           172:                        wlen = sendto(TASK_FD(task), buf, len, 0, &cli->cli_addr.sa, cli->cli_addr.sa.sa_len);
        !           173: #else
        !           174:                        wlen = sendto(TASK_FD(task), buf, len, 0, &cli->cli_addr.sa, 
        !           175:                                        cli->cli_addr.sa.sa_family == AF_INET ? 
        !           176:                                        sizeof cli->cli_addr.sin : sizeof cli->cli_addr.sin6);
        !           177: #endif
1.10      misho     178:                if (wlen < 1) {
                    179:                        schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
                    180:                        break;
                    181:                }
                    182:        }
1.5       misho     183: 
                    184:        taskExit(task, NULL);
                    185: }
                    186: 
                    187: static void *
                    188: io_txPty(sched_task_t *task)
                    189: {
                    190:        int wlen;
                    191:        sock_cli_t *cli = TASK_ARG(task);
                    192: 
1.7       misho     193:        ioUpdTimerSocket(cli);
1.5       misho     194: 
                    195:        wlen = write(TASK_FD(task), TASK_DATA(task), TASK_DATLEN(task));
                    196:        if (wlen < 1)
1.7       misho     197:                schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
1.5       misho     198: 
                    199:        taskExit(task, NULL);
                    200: }
                    201: 
                    202: static void *
                    203: io_rxNet(sched_task_t *task)
                    204: {
                    205:        int rlen;
                    206:        sock_cli_t *cli = TASK_ARG(task);
                    207:        sock_t *s = (sock_t*) cli->cli_parent;
                    208:        sockaddr_t sa;
                    209:        socklen_t salen = sizeof sa.ss;
                    210: 
1.7       misho     211:        ioUpdTimerSocket(cli);
1.5       misho     212: 
                    213:        if (s->sock_type == SOCK_STREAM)
                    214:                rlen = recv(TASK_FD(task), AIT_GET_BUF(&cli->cli_buf[0]), 
                    215:                                AIT_LEN(&cli->cli_buf[0]), 0);
                    216:        else {
                    217:                rlen = recvfrom(TASK_FD(task), AIT_GET_BUF(&cli->cli_buf[0]), 
                    218:                                AIT_LEN(&cli->cli_buf[0]), 0, &sa.sa, &salen);
1.15      misho     219:                if (e_addrcmp(&cli->cli_addr, &sa, 42)) {
                    220:                        schedReadSelf(task);
                    221:                        taskExit(task, NULL);
                    222:                }
1.5       misho     223:        }
                    224:        if (rlen < 1)
1.7       misho     225:                schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
1.15      misho     226:        else {
1.5       misho     227:                schedEvent(TASK_ROOT(task), io_txPty, cli, cli->cli_pty, 
                    228:                                AIT_GET_BUF(&cli->cli_buf[0]), rlen);
1.15      misho     229:                schedReadSelf(task);
                    230:        }
                    231: 
1.5       misho     232:        taskExit(task, NULL);
                    233: }
                    234: 
                    235: static void *
                    236: io_rxPty(sched_task_t *task)
                    237: {
                    238:        int rlen;
                    239:        sock_cli_t *cli = TASK_ARG(task);
                    240: 
1.7       misho     241:        ioUpdTimerSocket(cli);
1.5       misho     242: 
                    243:        rlen = read(TASK_FD(task), AIT_GET_BUF(&cli->cli_buf[1]), AIT_LEN(&cli->cli_buf[1]));
                    244:        if (rlen < 1)
1.7       misho     245:                schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
1.15      misho     246:        else {
1.5       misho     247:                schedEvent(TASK_ROOT(task), io_txNet, cli, cli->cli_fd, 
                    248:                                AIT_GET_BUF(&cli->cli_buf[1]), rlen);
1.15      misho     249:                schedReadSelf(task);
                    250:        }
1.5       misho     251: 
                    252:        taskExit(task, NULL);
                    253: }
                    254: 
                    255: static void *
                    256: io_bridgeClient(sched_task_t *task)
                    257: {
1.13      misho     258:        int c, rlen, pty;
1.5       misho     259:        pid_t pid;
                    260:        sockaddr_t sa;
                    261:        socklen_t salen = sizeof sa.ss;
                    262:        sock_cli_t *cli = NULL;
                    263:        sock_t *s = (sock_t*) TASK_ARG(task);
                    264:        array_t *args = NULL;
                    265:        char **argv = NULL;
                    266: 
                    267:        if (s->sock_type == SOCK_STREAM) {
                    268:                if ((c = accept(TASK_FD(task), &sa.sa, &salen)) == -1) {
                    269:                        LOGERR;
                    270:                        goto end;
                    271:                }
                    272:        } else {
                    273:                if ((rlen = recvfrom(TASK_FD(task), 
                    274:                                                AIT_GET_BUF(&s->sock_buf), AIT_LEN(&s->sock_buf), 
                    275:                                                MSG_PEEK, &sa.sa, &salen)) == -1) {
                    276:                        LOGERR;
                    277:                        goto end;
                    278:                } else
                    279:                        c = TASK_FD(task);
                    280:        }
                    281: 
                    282:        cli = e_malloc(sizeof(sock_cli_t));
                    283:        if (!cli) {
                    284:                io_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                    285:                if (s->sock_type == SOCK_STREAM)
                    286:                        close(c);
                    287:                goto end;
                    288:        } else {
                    289:                memset(cli, 0, sizeof(sock_cli_t));
                    290:                pthread_mutex_lock(&s->sock_mtx);
                    291:                TAILQ_INSERT_TAIL(&s->sock_cli, cli, cli_node);
                    292:                pthread_mutex_unlock(&s->sock_mtx);
                    293:        }
                    294: 
                    295:        cli->cli_parent = TASK_ARG(task);
                    296:        cli->cli_fd = c;
                    297:        strlcpy(cli->cli_cmdline, TASK_DATA(task), sizeof cli->cli_cmdline);
                    298:        memcpy(&cli->cli_addr, &sa, sizeof cli->cli_addr);
                    299:        AIT_SET_BUFSIZ(&cli->cli_buf[0], 0, AIT_LEN(&s->sock_buf));
                    300:        AIT_SET_BUFSIZ(&cli->cli_buf[1], 0, AIT_LEN(&s->sock_buf));
                    301: 
1.13      misho     302:        switch ((pid = ioForkPTY(&pty, cli->cli_name, sizeof cli->cli_name, 
1.5       misho     303:                                NULL, NULL, NULL))) {
                    304:                case -1:
                    305:                        ELIBERR(io);
                    306:                        break;
                    307:                case 0:
1.13      misho     308:                        cli->cli_pty = pty;
1.15      misho     309:                        ioSetRAWMode(STDIN_FILENO, NULL);
1.13      misho     310: 
1.5       misho     311:                        array_Args(cli->cli_cmdline, 0, " \t", &args);
                    312:                        argv = array_To(args);
                    313:                        array_Destroy(&args);
                    314: 
1.11      misho     315: #if 0
1.5       misho     316:                        printf("Console %s\n", cli->cli_name);
1.11      misho     317: #endif
1.6       misho     318:                        rlen = execv(*argv, argv);
                    319:                        _exit(rlen);
1.5       misho     320:                        break;
                    321:                default:
1.13      misho     322:                        cli->cli_pty = pty;
1.5       misho     323:                        cli->cli_pid = pid;
                    324: 
                    325:                        schedRead(TASK_ROOT(task), io_rxPty, cli, cli->cli_pty, 
                    326:                                        TASK_ARG(task), 0);
                    327:                        schedRead(TASK_ROOT(task), io_rxNet, cli, cli->cli_fd, 
                    328:                                        TASK_ARG(task), 0);
1.7       misho     329:                        ioUpdTimerSocket(cli);
1.5       misho     330:                        break;
                    331:        }
                    332: end:
                    333:        schedReadSelf(task);
                    334:        taskExit(task, NULL);
                    335: }
                    336: 
1.12      misho     337: static void *
                    338: io_bridgeClient2Pool(sched_task_t *task)
                    339: {
                    340:        int c, rlen;
                    341:        sockaddr_t sa;
                    342:        socklen_t salen = sizeof sa.ss;
                    343:        sock_cli_t *cli = NULL;
                    344:        sock_t *s = (sock_t*) TASK_ARG(task);
                    345: 
                    346:        if (s->sock_type == SOCK_STREAM) {
                    347:                if ((c = accept(TASK_FD(task), &sa.sa, &salen)) == -1) {
                    348:                        LOGERR;
                    349:                        goto end;
                    350:                }
                    351:        } else {
                    352:                if ((rlen = recvfrom(TASK_FD(task), 
                    353:                                                AIT_GET_BUF(&s->sock_buf), AIT_LEN(&s->sock_buf), 
                    354:                                                MSG_PEEK, &sa.sa, &salen)) == -1) {
                    355:                        LOGERR;
                    356:                        goto end;
                    357:                } else
                    358:                        c = TASK_FD(task);
                    359:        }
                    360: 
                    361:        cli = e_malloc(sizeof(sock_cli_t));
                    362:        if (!cli) {
                    363:                io_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                    364:                if (s->sock_type == SOCK_STREAM)
                    365:                        close(c);
                    366:                goto end;
                    367:        } else {
                    368:                memset(cli, 0, sizeof(sock_cli_t));
                    369:                pthread_mutex_lock(&s->sock_mtx);
                    370:                TAILQ_INSERT_TAIL(&s->sock_cli, cli, cli_node);
                    371:                pthread_mutex_unlock(&s->sock_mtx);
                    372:        }
                    373: 
                    374:        io_progCheck(s->sock_prog, 42);
                    375: 
                    376:        cli->cli_parent = TASK_ARG(task);
                    377:        cli->cli_fd = c;
                    378:        cli->cli_pty = io_progAttach(s->sock_prog, 42);
                    379:        strlcpy(cli->cli_cmdline, TASK_DATA(task), sizeof cli->cli_cmdline);
                    380:        memcpy(&cli->cli_addr, &sa, sizeof cli->cli_addr);
                    381:        AIT_SET_BUFSIZ(&cli->cli_buf[0], 0, AIT_LEN(&s->sock_buf));
                    382:        AIT_SET_BUFSIZ(&cli->cli_buf[1], 0, AIT_LEN(&s->sock_buf));
                    383: 
                    384:        schedRead(TASK_ROOT(task), io_rxPty, cli, cli->cli_pty, TASK_ARG(task), 0);
                    385:        schedRead(TASK_ROOT(task), io_rxNet, cli, cli->cli_fd, TASK_ARG(task), 0);
                    386:        ioUpdTimerSocket(cli);
                    387: end:
                    388:        schedReadSelf(task);
                    389:        taskExit(task, NULL);
                    390: }
                    391: 
1.5       misho     392: 
1.2       misho     393: /*
                    394:  * ioInitSocket() - Init socket and allocate resources
                    395:  *
                    396:  * @role = Socket role
                    397:  * @type = Socket type
                    398:  * @proto = Socket protocol
                    399:  * @addr = Bind to address
                    400:  * @port = Bind to port
                    401:  * @buflen = Socket buffer, optional if =0 == BUFSIZ
                    402:  * return: NULL error or !=NULL created socket
                    403:  */
                    404: sock_t *
                    405: ioInitSocket(int role, int type, int proto, const char *addr, u_short port, size_t buflen)
                    406: {
                    407:        sock_t *s = NULL;
                    408:        int n = 1;
                    409: 
                    410:        if (!addr)
                    411:                return NULL;
                    412: 
                    413:        s = e_malloc(sizeof(sock_t));
                    414:        if (!s) {
                    415:                io_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                    416:                return NULL;
                    417:        } else
                    418:                memset(s, 0, sizeof(sock_t));
                    419: 
1.3       misho     420:        TAILQ_INIT(&s->sock_cli);
                    421: 
1.2       misho     422:        s->sock_role = role;
                    423:        s->sock_type = type;
                    424:        s->sock_proto = proto;
                    425:        if (!e_gethostbyname(addr, port, &s->sock_addr)) {
                    426:                io_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                    427:                e_free(s);
                    428:                return NULL;
                    429:        } else {
                    430:                buflen = buflen ? buflen : BUFSIZ;
1.5       misho     431:                buflen = E_ALIGN(buflen, 2);    /* align buflen length */
1.2       misho     432:                AIT_SET_BUFSIZ(&s->sock_buf, 0, buflen);
                    433:        }
                    434: 
                    435:        s->sock_fd = socket(s->sock_addr.sa.sa_family, s->sock_type, s->sock_proto);
                    436:        if (s->sock_fd == -1) {
                    437:                LOGERR;
                    438:                AIT_FREE_VAL(&s->sock_buf);
                    439:                e_free(s);
                    440:                return NULL;
                    441:        }
                    442:        if (setsockopt(s->sock_fd, SOL_SOCKET, SO_SNDBUF, &buflen, sizeof buflen) == -1) {
                    443:                LOGERR;
                    444:                AIT_FREE_VAL(&s->sock_buf);
                    445:                e_free(s);
                    446:                return NULL;
                    447:        }
                    448:        if (setsockopt(s->sock_fd, SOL_SOCKET, SO_RCVBUF, &buflen, sizeof buflen) == -1) {
                    449:                LOGERR;
                    450:                AIT_FREE_VAL(&s->sock_buf);
                    451:                e_free(s);
                    452:                return NULL;
                    453:        }
                    454:        if (setsockopt(s->sock_fd, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
                    455:                LOGERR;
                    456:                AIT_FREE_VAL(&s->sock_buf);
                    457:                e_free(s);
                    458:                return NULL;
                    459:        }
1.15.4.1! misho     460: #ifndef __linux__
1.2       misho     461:        if (bind(s->sock_fd, &s->sock_addr.sa, s->sock_addr.sa.sa_len) == -1) {
1.15.4.1! misho     462: #else
        !           463:        if (bind(s->sock_fd, &s->sock_addr.sa, s->sock_addr.sa.sa_family == AF_INET ? 
        !           464:                                sizeof s->sock_addr.sin : sizeof s->sock_addr.sin6) == -1) {
        !           465: #endif
1.2       misho     466:                LOGERR;
                    467:                AIT_FREE_VAL(&s->sock_buf);
                    468:                e_free(s);
                    469:                return NULL;
                    470:        }
                    471: 
1.5       misho     472:        s->sock_root = schedBegin();
                    473:        if (!s->sock_root) {
                    474:                io_SetErr(sched_GetErrno(), "%s", sched_GetError());
                    475:                AIT_FREE_VAL(&s->sock_buf);
                    476:                e_free(s);
                    477:                return NULL;
                    478:        }
                    479: 
1.3       misho     480:        pthread_mutex_init(&s->sock_mtx, NULL);
1.2       misho     481:        return s;
                    482: }
                    483: 
                    484: /*
                    485:  * ioCloseSocket() - Close socket and free resources
                    486:  *
                    487:  * @s = Socket
                    488:  * return: none
                    489:  */
                    490: void
                    491: ioCloseSocket(sock_t ** __restrict s)
                    492: {
1.5       misho     493:        sock_cli_t *cli;
                    494:        int stat;
1.3       misho     495: 
1.2       misho     496:        if (s && *s) {
1.3       misho     497:                pthread_mutex_lock(&(*s)->sock_mtx);
                    498:                while ((cli = TAILQ_FIRST(&(*s)->sock_cli))) {
                    499:                        TAILQ_REMOVE(&(*s)->sock_cli, cli, cli_node);
1.5       misho     500: 
                    501:                        schedCancelby((*s)->sock_root, taskMAX, CRITERIA_ARG, cli, NULL);
                    502: 
                    503:                        if ((*s)->sock_type == SOCK_STREAM) {
                    504:                                shutdown(cli->cli_fd, SHUT_RDWR);
                    505:                                close(cli->cli_fd);
                    506:                        }
                    507:                        AIT_FREE_VAL(&cli->cli_buf[1]);
                    508:                        AIT_FREE_VAL(&cli->cli_buf[0]);
                    509: 
                    510:                        if (cli->cli_pid > 0) {
1.7       misho     511:                                kill(cli->cli_pid, SIGKILL);
1.5       misho     512:                                while (waitpid(cli->cli_pid, &stat, WNOHANG) > 0) {
                    513:                                        usleep(1000);
1.7       misho     514:                                        kill(cli->cli_pid, SIGKILL);
1.5       misho     515:                                }
                    516:                        }
                    517: 
1.3       misho     518:                        e_free(cli);
                    519:                }
                    520:                pthread_mutex_unlock(&(*s)->sock_mtx);
                    521: 
1.2       misho     522:                shutdown((*s)->sock_fd, SHUT_RDWR);
                    523:                close((*s)->sock_fd);
                    524: 
                    525:                AIT_FREE_VAL(&(*s)->sock_buf);
1.3       misho     526: 
1.5       misho     527:                schedEnd(&(*s)->sock_root);
                    528: 
1.3       misho     529:                pthread_mutex_destroy(&(*s)->sock_mtx);
1.2       misho     530:                e_free(*s);
                    531:                *s = NULL;
                    532:        }
                    533: }
                    534: 
                    535: /*
                    536:  * ioUpSocket() - Setup socket for use
                    537:  *
                    538:  * @s = Socket
                    539:  * @arg = Server role = listen backlog queue and Client role = peer address
1.5       misho     540:  * @timeout = Socket timeout in sec (default -1 infinit)
1.2       misho     541:  * return: -1 error or 0 ok
                    542:  */
                    543: int
1.5       misho     544: ioUpSocket(sock_t * __restrict s, void *arg, int timeout)
1.2       misho     545: {
                    546:        int ret = 0;
                    547:        sockaddr_t *peer = (sockaddr_t*) arg;
                    548:        uintptr_t backlog = (uintptr_t) arg;
                    549: 
                    550:        if (!s || !arg)
                    551:                return -1;
1.5       misho     552:        else {
                    553:                s->sock_timeout.tv_sec = timeout;
                    554:                s->sock_timeout.tv_nsec = (timeout < 1) ? timeout : 0;
                    555:                schedPolling(s->sock_root, &s->sock_timeout, NULL);
                    556:        }
1.2       misho     557: 
                    558:        switch (s->sock_role) {
                    559:                case IO_SOCK_ROLE_CLIENT:
                    560:                        memcpy(&s->sock_peer, peer, sizeof s->sock_peer);
                    561: 
1.15.4.1! misho     562: #ifndef __linux__
        !           563:                        if (connect(s->sock_fd, &s->sock_peer.sa, s->sock_peer.sa.sa_len) == -1) {
        !           564: #else
        !           565:                        if (connect(s->sock_fd, &s->sock_peer.sa, s->sock_addr.sa.sa_family == AF_INET ? 
        !           566:                                                sizeof s->sock_peer.sin : sizeof s->sock_peer.sin6) == -1) {
        !           567: #endif
1.2       misho     568:                                LOGERR;
                    569:                                return -1;
                    570:                        }
                    571:                        break;
                    572:                case IO_SOCK_ROLE_SERVER:
                    573:                        if (s->sock_type == SOCK_STREAM) {
                    574:                                s->sock_backq = backlog;
                    575: 
                    576:                                if (listen(s->sock_fd, s->sock_backq) == -1) {
                    577:                                        LOGERR;
                    578:                                        return -1;
                    579:                                }
                    580:                        }
                    581:                        break;
                    582:                default:
                    583:                        io_SetErr(EINVAL, "Unsupported socket type");
                    584:                        return -1;
                    585:        }
                    586: 
                    587:        fcntl(s->sock_fd, F_SETFL, fcntl(s->sock_fd, F_GETFL) | O_NONBLOCK);
                    588:        return ret;
                    589: }
1.3       misho     590: 
1.5       misho     591: /*
                    592:  * ioUpdTimerSocket() - Update timeout of socket
                    593:  *
                    594:  * @c = Client socket
                    595:  * return:  none
                    596:  */
                    597: void
1.7       misho     598: ioUpdTimerSocket(sock_cli_t * __restrict c)
1.3       misho     599: {
1.5       misho     600:        sock_t *s;
1.3       misho     601: 
1.5       misho     602:        if (!c)
                    603:                return;
                    604:        else
                    605:                s = c->cli_parent;
1.3       misho     606: 
1.12      misho     607:        if (s->sock_prog)
                    608:                io_progCheck(s->sock_prog, 42);
                    609: 
1.7       misho     610:        schedCancelby(s->sock_root, taskTIMER, CRITERIA_ARG, c, NULL);
                    611:        schedTimer(s->sock_root, io_closeClient, c, s->sock_timeout, NULL, 0);
1.3       misho     612: }
                    613: 
1.5       misho     614: /*
                    615:  * ioCloseClient() - Close client socket
                    616:  *
                    617:  * @c = Client socket
                    618:  * return: 0 ok or !=0 error
                    619:  */
                    620: int
                    621: ioCloseClient(sock_cli_t * __restrict c)
1.3       misho     622: {
1.5       misho     623:        sock_t *s;
1.3       misho     624: 
1.5       misho     625:        if (!c)
                    626:                return -1;
                    627:        else
                    628:                s = c->cli_parent;
1.3       misho     629: 
1.7       misho     630:        return !schedEvent(s->sock_root, io_closeClient, c, 0, NULL, 0);
1.3       misho     631: }
                    632: 
                    633: /*
1.5       misho     634:  * ioLoopSocket() - Start socket scheduler
1.3       misho     635:  *
                    636:  * @s = Socket
1.5       misho     637:  * @rcb = Read callback
                    638:  * return: -1 error or return result from scheduler
1.3       misho     639:  */
                    640: int
1.5       misho     641: ioLoopSocket(sock_t * __restrict s, sched_task_func_t rcb)
1.3       misho     642: {
1.5       misho     643:        if (!s || !rcb || s->sock_kill)
1.3       misho     644:                return -1;
                    645: 
1.5       misho     646:        schedRead(s->sock_root, io_acceptClient, s, s->sock_fd, rcb, 0);
                    647:        return schedRun(s->sock_root, &s->sock_kill);
                    648: }
1.3       misho     649: 
1.12      misho     650: static void *
                    651: io_progPurge(sched_task_t *task)
                    652: {
                    653:        sock_t *s = (sock_t*) TASK_ARG(task);
                    654: 
                    655:        io_progVacuum(s->sock_prog, 0);
                    656: 
                    657:        schedTimer(TASK_ROOT(task), TASK_FUNC(task), TASK_ARG(task), 
                    658:                        s->sock_timeout, TASK_DATA(task), TASK_DATLEN(task));
                    659:        taskExit(task, NULL);
                    660: }
                    661: 
1.5       misho     662: /*
                    663:  * ioBridgeProg2Socket() - Start socket scheduler and bridge program to socket
                    664:  *
                    665:  * @s = Socket
                    666:  * @prgname = Program name
                    667:  * return: 0 ok or !=0 error
                    668:  */
                    669: int
                    670: ioBridgeProg2Socket(sock_t * __restrict s, const char *prgname)
                    671: {
                    672:        if (!s || !prgname || s->sock_kill)
                    673:                return -1;
1.3       misho     674: 
1.12      misho     675:        if (s->sock_prog) {
                    676:                schedRead(s->sock_root, io_bridgeClient2Pool, 
                    677:                                s, s->sock_fd, (void*) prgname, 0);
                    678:                schedTimer(s->sock_root, io_progPurge, s, s->sock_timeout, NULL, 0);
                    679:        } else
                    680:                schedRead(s->sock_root, io_bridgeClient, 
                    681:                                s, s->sock_fd, (void*) prgname, 0);
1.5       misho     682:        return schedRun(s->sock_root, &s->sock_kill);
1.3       misho     683: }
1.12      misho     684: 
                    685: /*
                    686:  * ioSetupProg2Socket() - Setup program pool to socket server
                    687:  *
                    688:  * @s = Socket
                    689:  * @p = Program pool
                    690:  * return: -1 error or 0 ok
                    691:  */
                    692: int
                    693: ioSetupProg2Socket(sock_t * __restrict s, prog_t * __restrict p)
                    694: {
                    695:        if (!s)
                    696:                return -1;
                    697: 
                    698:        s->sock_prog = p;
                    699:        return 0;
                    700: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>