Annotation of libaitrpc/src/srv.c, revision 1.28.2.5

1.1       misho       1: /*************************************************************************
                      2: * (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
                      3: *  by Michael Pounov <misho@openbsd-bg.org>
                      4: *
                      5: * $Author: misho $
1.28.2.5! misho       6: * $Id: srv.c,v 1.28.2.4 2016/08/02 15:18:41 misho Exp $
1.1       misho       7: *
1.2       misho       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.28.2.2  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: */
1.1       misho      46: #include "global.h"
                     47: 
                     48: 
1.13      misho      49: /* SOCK_STREAM */
                     50: static void *acceptClients(sched_task_t *);
                     51: static void *closeClient(sched_task_t *);
                     52: static void *rxPacket(sched_task_t *);
                     53: static void *txPacket(sched_task_t *);
                     54: 
                     55: /* SOCK_DGRAM */
                     56: static void *freeClient(sched_task_t *);
                     57: static void *rxUDPPacket(sched_task_t *);
                     58: static void *txUDPPacket(sched_task_t *);
                     59: 
                     60: /* SOCK_RAW */
1.26      misho      61: static void *rxRAWPacket(sched_task_t *);
                     62: static void *txRAWPacket(sched_task_t *);
1.13      misho      63: 
1.24      misho      64: /* SOCK_BPF */
                     65: static void *rxBPFPacket(sched_task_t *);
                     66: static void *txBPFPacket(sched_task_t *);
                     67: 
1.25      misho      68: /* SOCK_EXT */
                     69: static void *rxEXTPacket(sched_task_t *);
                     70: static void *txEXTPacket(sched_task_t *);
                     71: 
                     72: static sched_task_func_t cbProto[SOCK_MAX_SUPPORT][4] = {
1.27      misho      73:        { acceptClients, closeClient, rxPacket, txPacket },             /* SOCK_STREAM */
                     74:        { acceptClients, closeClient, rxPacket, txPacket },             /* SOCK_STREAM */
                     75:        { rxUDPPacket, freeClient, NULL /*rxUDPPacket*/, txUDPPacket }, /* SOCK_DGRAM */
                     76:        { rxRAWPacket, freeClient, NULL /*rxRAWPacket*/, txRAWPacket }, /* SOCK_RAW */
                     77:        { rxBPFPacket, freeClient, NULL /*rxBPFPacket*/, txBPFPacket }, /* SOCK_BPF */
                     78:        { rxEXTPacket, freeClient, NULL /*rxEXTPacket*/, txEXTPacket }  /* SOCK_EXT */
1.13      misho      79: };
                     80: 
1.23      misho      81: /* Global Signal Argument when kqueue support disabled */
                     82: 
                     83: static volatile uintptr_t _glSigArg = 0;
                     84: 
1.16      misho      85: void
1.13      misho      86: rpc_freeCli(rpc_cli_t * __restrict c)
1.10      misho      87: {
                     88:        rpc_srv_t *s = c->cli_parent;
                     89: 
1.13      misho      90:        schedCancelby(s->srv_root, taskMAX, CRITERIA_ARG, c, NULL);
1.10      misho      91: 
                     92:        /* free buffer */
                     93:        AIT_FREE_VAL(&c->cli_buf);
                     94: 
1.14      misho      95:        array_Del(s->srv_clients, c->cli_id, 0);
1.10      misho      96:        if (c)
1.14      misho      97:                e_free(c);
1.13      misho      98: }
                     99: 
                    100: 
                    101: static inline int
1.14      misho     102: _check4freeslot(rpc_srv_t * __restrict srv, sockaddr_t * __restrict sa)
1.13      misho     103: {
                    104:        rpc_cli_t *c = NULL;
                    105:        register int i;
                    106: 
                    107:        /* check free slots for connect */
1.14      misho     108:        for (i = 0; i < array_Size(srv->srv_clients) && 
                    109:                        (c = array(srv->srv_clients, i, rpc_cli_t*)); i++)
1.13      misho     110:                /* check for duplicates */
1.14      misho     111:                if (sa && !e_addrcmp(&c->cli_sa, sa, 42))
1.13      misho     112:                        break;
1.14      misho     113:        if (i >= array_Size(srv->srv_clients))
1.13      misho     114:                return -1;      /* no more free slots! */
                    115: 
                    116:        return i;
                    117: }
                    118: 
                    119: static rpc_cli_t *
1.14      misho     120: _allocClient(rpc_srv_t * __restrict srv, sockaddr_t * __restrict sa)
1.13      misho     121: {
                    122:        rpc_cli_t *c = NULL;
                    123:        int n;
                    124: 
1.25      misho     125:        if (srv->srv_proto != SOCK_EXT)
                    126:                n = _check4freeslot(srv, sa);
                    127:        else
                    128:                n = 0;
1.13      misho     129:        if (n == -1)
                    130:                return NULL;
                    131:        else
1.14      misho     132:                c = array(srv->srv_clients, n, rpc_cli_t*);
1.13      misho     133: 
                    134:        if (!c) {
1.14      misho     135:                c = e_malloc(sizeof(rpc_cli_t));
1.13      misho     136:                if (!c) {
                    137:                        LOGERR;
                    138:                        srv->srv_kill = 1;
                    139:                        return NULL;
                    140:                } else {
                    141:                        memset(c, 0, sizeof(rpc_cli_t));
1.14      misho     142:                        array_Set(srv->srv_clients, n, c);
1.13      misho     143:                        c->cli_id = n;
                    144:                        c->cli_parent = srv;
                    145:                }
                    146: 
                    147:                /* alloc empty buffer */
1.14      misho     148:                AIT_SET_BUFSIZ(&c->cli_buf, 0, srv->srv_netbuf);
1.13      misho     149:        }
                    150: 
                    151:        return c;
                    152: }
                    153: 
                    154: 
                    155: static void *
                    156: freeClient(sched_task_t *task)
                    157: {
                    158:        rpc_freeCli(TASK_ARG(task));
                    159: 
1.27      misho     160:        taskExit(task, NULL);
1.13      misho     161: }
                    162: 
                    163: static void *
                    164: closeClient(sched_task_t *task)
                    165: {
                    166:        int sock = ((rpc_cli_t*) TASK_ARG(task))->cli_sock;
                    167: 
                    168:        rpc_freeCli(TASK_ARG(task));
                    169: 
                    170:        /* close client socket */
                    171:        shutdown(sock, SHUT_RDWR);
                    172:        close(sock);
1.27      misho     173:        taskExit(task, NULL);
1.10      misho     174: }
1.7       misho     175: 
                    176: static void *
                    177: txPacket(sched_task_t *task)
                    178: {
                    179:        rpc_cli_t *c = TASK_ARG(task);
                    180:        rpc_srv_t *s = c->cli_parent;
                    181:        rpc_func_t *f = NULL;
1.18      misho     182:        u_char *buf = AIT_GET_BUF(&c->cli_buf);
1.7       misho     183:        struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
1.27      misho     184:        int ret, wlen = sizeof(struct tagRPCCall);
1.21      misho     185: #ifdef TCP_SESSION_TIMEOUT
                    186:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
                    187: 
                    188:        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
                    189:        schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    190:                         TASK_ARG(task), ts, TASK_ARG(task), 0);
                    191: #endif
1.7       misho     192: 
                    193:        if (rpc->call_argc) {
1.10      misho     194:                f = rpc_srv_getCall(s, ntohs(rpc->call_tag));
1.7       misho     195:                if (!f) {
1.10      misho     196:                        rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server");
1.18      misho     197: 
1.7       misho     198:                        rpc->call_argc ^= rpc->call_argc;
1.27      misho     199:                        RPC_SET_RETURN(rpc, -1);
                    200:                        RPC_SET_ERRNO(rpc, rpc_Errno);
                    201:                } else if (rpc_pktFreeSpace(c) > s->srv_netbuf) {
                    202:                        rpc_SetErr(EMSGSIZE, "Message too long");
                    203: 
                    204:                        rpc->call_argc ^= rpc->call_argc;
                    205:                        RPC_SET_RETURN(rpc, -1);
                    206:                        RPC_SET_ERRNO(rpc, rpc_Errno);
1.7       misho     207:                } else {
1.25      misho     208:                        rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
1.7       misho     209:                        /* Go Encapsulate variables */
1.18      misho     210:                        ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, 
                    211:                                        RPC_RETVARS(c));
1.7       misho     212:                        if (ret == -1) {
                    213:                                rpc_SetErr(EBADRPC, "Prepare RPC packet failed");
1.19      misho     214: 
1.7       misho     215:                                rpc->call_argc ^= rpc->call_argc;
1.27      misho     216:                                RPC_SET_RETURN(rpc, -1);
                    217:                                RPC_SET_ERRNO(rpc, rpc_Errno);
1.7       misho     218:                        } else
                    219:                                wlen += ret;
                    220:                }
                    221:        }
                    222: 
1.27      misho     223:        /* Free return values */
                    224:        ait_freeVars(&c->cli_vars);
                    225: 
1.18      misho     226:        rpc->call_len = htonl(wlen);
1.25      misho     227:        rpc->call_io = RPC_ACK;
1.8       misho     228: 
1.15      misho     229: #if 0
1.7       misho     230:        /* calculate CRC */
                    231:        rpc->call_crc ^= rpc->call_crc;
1.8       misho     232:        rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));
1.15      misho     233: #endif
1.7       misho     234: 
                    235:        /* send reply */
1.27      misho     236:        ret = send(TASK_FD(task), buf, wlen, MSG_NOSIGNAL);
                    237:        if (ret == -1) {
                    238:                /* close connection */
                    239:                schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    240:                                TASK_ARG(task), 0, NULL, 0);
1.10      misho     241:        }
1.7       misho     242: 
1.27      misho     243:        taskExit(task, NULL);
1.7       misho     244: }
                    245: 
                    246: static void *
                    247: execCall(sched_task_t *task)
                    248: {
                    249:        rpc_cli_t *c = TASK_ARG(task);
                    250:        rpc_srv_t *s = c->cli_parent;
                    251:        rpc_func_t *f = NULL;
                    252:        array_t *arr = NULL;
1.18      misho     253:        u_char *buf = AIT_GET_BUF(&c->cli_buf);
1.7       misho     254:        struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
1.25      misho     255:        int argc = rpc->call_argc;
1.7       misho     256: 
                    257:        /* Go decapsulate variables ... */
1.8       misho     258:        if (argc) {
1.14      misho     259:                arr = ait_buffer2vars(buf + sizeof(struct tagRPCCall), 
1.18      misho     260:                                AIT_LEN(&c->cli_buf) - sizeof(struct tagRPCCall), argc, 42);
1.7       misho     261:                if (!arr) {
1.14      misho     262:                        rpc_SetErr(ERPCMISMATCH, "#%d - %s", elwix_GetErrno(), elwix_GetError());
1.18      misho     263: 
1.7       misho     264:                        rpc->call_argc ^= rpc->call_argc;
1.27      misho     265:                        RPC_SET_RETURN(rpc, -1);
                    266:                        RPC_SET_ERRNO(rpc, rpc_Errno);
                    267:                        taskExit(task, NULL);
1.7       misho     268:                }
1.10      misho     269:        } else
                    270:                arr = NULL;
1.7       misho     271: 
1.10      misho     272:        if (!(f = rpc_srv_getCall(s, ntohs(rpc->call_tag)))) {
1.7       misho     273:                rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server");
1.18      misho     274: 
1.7       misho     275:                rpc->call_argc ^= rpc->call_argc;
1.27      misho     276:                RPC_SET_RETURN(rpc, -1);
                    277:                RPC_SET_ERRNO(rpc, rpc_Errno);
1.7       misho     278:        } else {
1.8       misho     279:                /* if client doesn't want reply */
1.27      misho     280:                RPC_SET_RETURN(rpc, rpc_srv_execCall(c, rpc, f->func_name, arr));
1.7       misho     281:                if (rpc->call_rep.ret == htonl(-1)) {
1.20      misho     282:                        if (!rpc->call_rep.eno) {
                    283:                                LOGERR;
1.27      misho     284:                                RPC_SET_ERRNO(rpc, rpc_Errno);
1.20      misho     285:                        }
1.7       misho     286:                        rpc->call_argc ^= rpc->call_argc;
1.27      misho     287:                        ait_freeVars(&c->cli_vars);
1.7       misho     288:                } else {
                    289:                        rpc->call_rep.eno ^= rpc->call_rep.eno;
1.27      misho     290:                        rpc->call_argc ^= rpc->call_argc;
                    291:                        if (TASK_VAL(task)) {
1.8       misho     292:                                /* without reply */
1.14      misho     293:                                ait_freeVars(&c->cli_vars);
1.10      misho     294:                        } else {
                    295:                                /* reply */
1.25      misho     296:                                rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
1.10      misho     297:                        }
1.7       misho     298:                }
                    299:        }
                    300: 
1.14      misho     301:        array_Destroy(&arr);
1.27      misho     302:        taskExit(task, NULL);
1.7       misho     303: }
                    304: 
                    305: static void *
                    306: rxPacket(sched_task_t *task)
                    307: {
                    308:        rpc_cli_t *c = TASK_ARG(task);
                    309:        rpc_srv_t *s = c->cli_parent;
1.27      misho     310:        int len, noreply = 0, rlen = AIT_LEN(&c->cli_buf);
1.15      misho     311: #if 0
                    312:        u_short crc;
                    313: #endif
1.10      misho     314:        u_char *buf = AIT_GET_BUF(&c->cli_buf);
1.28.2.1  misho     315:        struct tagRPCCall b, *rpc = (struct tagRPCCall*) buf;
1.21      misho     316: #ifdef TCP_SESSION_TIMEOUT
                    317:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
                    318: 
                    319:        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
                    320:        schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    321:                         TASK_ARG(task), ts, TASK_ARG(task), 0);
                    322: #endif
1.7       misho     323: 
1.27      misho     324:        /* prepare rx */
1.28.2.1  misho     325:        len = recv(TASK_FD(task), &b, sizeof b, MSG_PEEK);
1.27      misho     326:        if (len == sizeof b)
1.28.2.1  misho     327:                rlen = ntohl(b.call_len);
1.27      misho     328: 
                    329:        rlen = recv(TASK_FD(task), buf, rlen, 0);
                    330:        if (rlen == -1) {
1.10      misho     331:                /* close connection */
1.13      misho     332:                schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    333:                                TASK_ARG(task), 0, NULL, 0);
1.27      misho     334:                taskExit(task, NULL);
1.18      misho     335:        }
1.27      misho     336:        if (rlen < sizeof(struct tagRPCCall)) {
                    337:                rpc_SetErr(ERPCMISMATCH, "Short RPC packet");
1.18      misho     338: 
1.27      misho     339:                rpc->call_argc ^= rpc->call_argc;
                    340:                RPC_SET_RETURN(rpc, -1);
                    341:                RPC_SET_ERRNO(rpc, rpc_Errno);
                    342:                goto err;
                    343:        } else
                    344:                len = ntohl(rpc->call_len);
                    345:        if (rlen < len || len > AIT_LEN(&c->cli_buf)) {
                    346:                rpc_SetErr(ERPCMISMATCH, "Short RPC packet");
                    347: 
                    348:                rpc->call_argc ^= rpc->call_argc;
                    349:                RPC_SET_RETURN(rpc, -1);
                    350:                RPC_SET_ERRNO(rpc, rpc_Errno);
                    351:                goto err;
1.18      misho     352:        }
1.8       misho     353: 
1.25      misho     354:        /* skip loop packet */
                    355:        if (rpc->call_io & RPC_ACK) {
                    356:                schedReadSelf(task);
1.27      misho     357:                taskExit(task, NULL);
1.25      misho     358:        }
                    359: 
1.15      misho     360: #if 0
1.18      misho     361:        /* check integrity of packet */
                    362:        crc = ntohs(rpc->call_crc);
                    363:        rpc->call_crc ^= rpc->call_crc;
                    364:        if (crc != crcFletcher16((u_short*) rpc, len / 2)) {
                    365:                rpc_SetErr(ERPCMISMATCH, "Bad CRC RPC packet");
1.27      misho     366: 
                    367:                rpc->call_argc ^= rpc->call_argc;
                    368:                RPC_SET_RETURN(rpc, -1);
                    369:                RPC_SET_ERRNO(rpc, rpc_Errno);
                    370:                goto err;
1.18      misho     371:        }
1.15      misho     372: #endif
1.7       misho     373: 
1.18      misho     374:        /* check RPC packet session info */
                    375:        if (rpc_chkPktSession(&rpc->call_session, &s->srv_session)) {
                    376:                rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session");
1.7       misho     377: 
1.18      misho     378:                rpc->call_argc ^= rpc->call_argc;
1.27      misho     379:                RPC_SET_RETURN(rpc, -1);
                    380:                RPC_SET_ERRNO(rpc, rpc_Errno);
                    381:                goto err;
1.18      misho     382:        }
1.7       misho     383: 
1.27      misho     384:        noreply = RPC_CHK_NOREPLY(rpc);
                    385: 
                    386:        /* execute RPC call */
                    387:        schedEvent(TASK_ROOT(task), execCall, TASK_ARG(task), (int) noreply, rpc, len);
                    388: err:
1.18      misho     389:        /* send RPC reply */
                    390:        if (!noreply)
                    391:                schedWrite(TASK_ROOT(task), cbProto[s->srv_proto][CB_TXPACKET], 
                    392:                                TASK_ARG(task), TASK_FD(task), rpc, len);
1.7       misho     393: 
                    394:        /* lets get next packet */
1.18      misho     395:        schedReadSelf(task);
1.27      misho     396:        taskExit(task, NULL);
1.7       misho     397: }
                    398: 
1.1       misho     399: static void *
1.10      misho     400: acceptClients(sched_task_t *task)
1.1       misho     401: {
1.10      misho     402:        rpc_srv_t *srv = TASK_ARG(task);
                    403:        rpc_cli_t *c = NULL;
1.28.2.3  misho     404:        socklen_t salen = E_SOCKADDR_MAX;
1.21      misho     405:        int sock;
                    406: #ifdef TCP_SESSION_TIMEOUT
                    407:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
                    408: #endif
1.7       misho     409: 
1.13      misho     410:        c = _allocClient(srv, NULL);
1.21      misho     411:        if (!c) {
                    412:                EVERBOSE(1, "RPC client quota exceeded! Connection will be shutdown!\n");
                    413:                if ((sock = accept(TASK_FD(task), NULL, NULL)) != -1) {
                    414:                        shutdown(sock, SHUT_RDWR);
                    415:                        close(sock);
                    416:                }
1.10      misho     417:                goto end;
1.21      misho     418:        }
1.10      misho     419: 
                    420:        /* accept client */
                    421:        c->cli_sock = accept(TASK_FD(task), &c->cli_sa.sa, &salen);
                    422:        if (c->cli_sock == -1) {
                    423:                LOGERR;
                    424:                AIT_FREE_VAL(&c->cli_buf);
1.14      misho     425:                array_Del(srv->srv_clients, c->cli_id, 42);
1.10      misho     426:                goto end;
1.1       misho     427:        } else
1.10      misho     428:                fcntl(c->cli_sock, F_SETFL, fcntl(c->cli_sock, F_GETFL) | O_NONBLOCK);
1.1       misho     429: 
1.21      misho     430: #ifdef TCP_SESSION_TIMEOUT
                    431:        /* armed timer for close stateless connection */
                    432:        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL);
                    433:        schedTimer(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], c, 
                    434:                        ts, c, 0);
                    435: #endif
1.13      misho     436:        schedRead(TASK_ROOT(task), cbProto[srv->srv_proto][CB_RXPACKET], c, 
                    437:                        c->cli_sock, NULL, 0);
1.10      misho     438: end:
                    439:        schedReadSelf(task);
1.27      misho     440:        taskExit(task, NULL);
1.10      misho     441: }
1.5       misho     442: 
1.7       misho     443: 
1.10      misho     444: static void *
1.13      misho     445: txUDPPacket(sched_task_t *task)
1.10      misho     446: {
                    447:        rpc_cli_t *c = TASK_ARG(task);
                    448:        rpc_srv_t *s = c->cli_parent;
1.13      misho     449:        rpc_func_t *f = NULL;
1.18      misho     450:        u_char *buf = AIT_GET_BUF(&c->cli_buf);
1.13      misho     451:        struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
1.27      misho     452:        int ret, wlen = sizeof(struct tagRPCCall);
1.13      misho     453:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
                    454: 
                    455:        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
                    456:        schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    457:                         TASK_ARG(task), ts, TASK_ARG(task), 0);
                    458: 
                    459:        if (rpc->call_argc) {
                    460:                f = rpc_srv_getCall(s, ntohs(rpc->call_tag));
                    461:                if (!f) {
                    462:                        rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server");
1.27      misho     463: 
1.13      misho     464:                        rpc->call_argc ^= rpc->call_argc;
1.27      misho     465:                        RPC_SET_RETURN(rpc, -1);
                    466:                        RPC_SET_ERRNO(rpc, rpc_Errno);
                    467:                } else if (rpc_pktFreeSpace(c) > s->srv_netbuf) {
                    468:                        rpc_SetErr(EMSGSIZE, "Message too long");
                    469: 
                    470:                        rpc->call_argc ^= rpc->call_argc;
                    471:                        RPC_SET_RETURN(rpc, -1);
                    472:                        RPC_SET_ERRNO(rpc, rpc_Errno);
1.13      misho     473:                } else {
1.25      misho     474:                        rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
1.13      misho     475:                        /* Go Encapsulate variables */
1.18      misho     476:                        ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, 
                    477:                                        RPC_RETVARS(c));
1.13      misho     478:                        if (ret == -1) {
                    479:                                rpc_SetErr(EBADRPC, "Prepare RPC packet failed");
1.27      misho     480: 
1.13      misho     481:                                rpc->call_argc ^= rpc->call_argc;
1.27      misho     482:                                RPC_SET_RETURN(rpc, -1);
                    483:                                RPC_SET_ERRNO(rpc, rpc_Errno);
1.13      misho     484:                        } else
                    485:                                wlen += ret;
                    486:                }
                    487:        }
1.7       misho     488: 
1.27      misho     489:        /* Free return values */
                    490:        ait_freeVars(&c->cli_vars);
                    491: 
1.18      misho     492:        rpc->call_len = htonl(wlen);
1.25      misho     493:        rpc->call_io = RPC_ACK;
1.7       misho     494: 
1.13      misho     495:        /* calculate CRC */
                    496:        rpc->call_crc ^= rpc->call_crc;
                    497:        rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));
                    498: 
                    499:        /* send reply */
1.27      misho     500:        ret = sendto(TASK_FD(task), buf, wlen, MSG_NOSIGNAL, 
1.28.2.4  misho     501:                        &c->cli_sa.sa, e_addrlen(&c->cli_sa));
1.27      misho     502:        if (ret == -1) {
                    503:                /* close connection */
                    504:                schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    505:                                TASK_ARG(task), 0, NULL, 0);
1.13      misho     506:        }
                    507: 
1.27      misho     508:        taskExit(task, NULL);
1.13      misho     509: }
                    510: 
                    511: static void *
                    512: rxUDPPacket(sched_task_t *task)
                    513: {
                    514:        rpc_srv_t *srv = TASK_ARG(task);
                    515:        rpc_cli_t *c = NULL;
1.27      misho     516:        int len, noreply = 0, rlen;
1.22      misho     517:        u_short crc;
1.27      misho     518:        struct tagRPCCall *rpc;
1.14      misho     519:        sockaddr_t sa;
1.28.2.3  misho     520:        socklen_t salen = E_SOCKADDR_MAX;
1.13      misho     521:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
1.27      misho     522:        ait_val_t b = AIT_VAL_INIT;
1.13      misho     523: 
                    524:        /* receive connect packet */
1.27      misho     525:        AIT_SET_BUF(&b, NULL, srv->srv_netbuf);
1.28.2.2  misho     526: #ifndef __linux__
                    527:        sa.ss.ss_len = salen;
                    528: #endif
1.27      misho     529:        rlen = recvfrom(TASK_FD(task), AIT_GET_BUF(&b), AIT_LEN(&b), 0, &sa.sa, &salen);
                    530:        rpc = (struct tagRPCCall*) AIT_GET_BUF(&b);
                    531:        if (rlen < sizeof(struct tagRPCCall))
                    532:                goto end;
                    533:        else
                    534:                len = ntohl(rpc->call_len);
                    535:        if (rlen < len || len > srv->srv_netbuf)
                    536:                goto end;
                    537: 
                    538:        /* skip loop packet */
                    539:        if (rpc->call_io & RPC_ACK)
                    540:                goto end;
                    541: 
                    542:        /* check integrity of packet */
                    543:        crc = ntohs(rpc->call_crc);
                    544:        rpc->call_crc ^= rpc->call_crc;
                    545:        if (crc != crcFletcher16((u_short*) rpc, len / 2))
                    546:                goto end;
                    547: 
                    548:        /* check RPC packet session info */
                    549:        if (rpc_chkPktSession(&rpc->call_session, &srv->srv_session))
1.13      misho     550:                goto end;
                    551: 
                    552:        c = _allocClient(srv, &sa);
1.22      misho     553:        if (!c) {
                    554:                EVERBOSE(1, "RPC client quota exceeded! Connection will be shutdown!\n");
                    555:                usleep(2000);   /* blocked client delay */
1.13      misho     556:                goto end;
1.22      misho     557:        } else {
1.27      misho     558:                memcpy(AIT_GET_BUF(&c->cli_buf), AIT_GET_BUF(&b), len);
1.22      misho     559:                rpc = (struct tagRPCCall*) AIT_GET_BUF(&c->cli_buf);
                    560: 
1.13      misho     561:                c->cli_sock = TASK_FD(task);
                    562:                memcpy(&c->cli_sa, &sa, sizeof c->cli_sa);
1.22      misho     563: 
1.13      misho     564:                /* armed timer for close stateless connection */
                    565:                schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL);
                    566:                schedTimer(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], 
                    567:                                c, ts, c, 0);
                    568:        }
                    569: 
1.24      misho     570:        noreply = RPC_CHK_NOREPLY(rpc);
                    571: 
1.27      misho     572:        /* execute RPC call */
                    573:        schedEvent(TASK_ROOT(task), execCall, c, (int) noreply, rpc, len);
1.24      misho     574: 
                    575:        /* send RPC reply */
                    576:        if (!noreply)
                    577:                schedWrite(TASK_ROOT(task), cbProto[srv->srv_proto][CB_TXPACKET], 
                    578:                                c, TASK_FD(task), rpc, len);
                    579: end:
1.27      misho     580:        AIT_FREE_VAL(&b);
1.24      misho     581:        schedReadSelf(task);
1.27      misho     582:        taskExit(task, NULL);
1.24      misho     583: }
                    584: 
                    585: 
                    586: static void *
1.26      misho     587: txRAWPacket(sched_task_t *task)
                    588: {
                    589:        rpc_cli_t *c = TASK_ARG(task);
                    590:        rpc_srv_t *s = c->cli_parent;
                    591:        rpc_func_t *f = NULL;
                    592:        u_char *buf = AIT_GET_BUF(&c->cli_buf);
                    593:        struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
1.27      misho     594:        int ret, wlen = sizeof(struct tagRPCCall);
1.26      misho     595:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
                    596: 
                    597:        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
                    598:        schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    599:                         TASK_ARG(task), ts, TASK_ARG(task), 0);
                    600: 
                    601:        if (rpc->call_argc) {
                    602:                f = rpc_srv_getCall(s, ntohs(rpc->call_tag));
                    603:                if (!f) {
                    604:                        rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server");
1.27      misho     605: 
1.26      misho     606:                        rpc->call_argc ^= rpc->call_argc;
1.27      misho     607:                        RPC_SET_RETURN(rpc, -1);
                    608:                        RPC_SET_ERRNO(rpc, rpc_Errno);
                    609:                } else if (rpc_pktFreeSpace(c) > s->srv_netbuf) {
                    610:                        rpc_SetErr(EMSGSIZE, "Message too long");
                    611: 
                    612:                        rpc->call_argc ^= rpc->call_argc;
                    613:                        RPC_SET_RETURN(rpc, -1);
                    614:                        RPC_SET_ERRNO(rpc, rpc_Errno);
1.26      misho     615:                } else {
                    616:                        rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
                    617:                        /* Go Encapsulate variables */
                    618:                        ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, 
                    619:                                        RPC_RETVARS(c));
                    620:                        if (ret == -1) {
                    621:                                rpc_SetErr(EBADRPC, "Prepare RPC packet failed");
1.27      misho     622: 
1.26      misho     623:                                rpc->call_argc ^= rpc->call_argc;
1.27      misho     624:                                RPC_SET_RETURN(rpc, -1);
                    625:                                RPC_SET_ERRNO(rpc, rpc_Errno);
1.26      misho     626:                        } else
                    627:                                wlen += ret;
                    628:                }
                    629:        }
                    630: 
1.27      misho     631:        /* Free return values */
                    632:        ait_freeVars(&c->cli_vars);
                    633: 
1.26      misho     634:        rpc->call_len = htonl(wlen);
                    635:        rpc->call_io = RPC_ACK;
                    636: 
                    637:        /* calculate CRC */
                    638:        rpc->call_crc ^= rpc->call_crc;
                    639:        rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));
                    640: 
                    641:        /* send reply */
1.27      misho     642:        ret = sendto(TASK_FD(task), buf, wlen, MSG_NOSIGNAL, 
1.28.2.4  misho     643:                        &c->cli_sa.sa, e_addrlen(&c->cli_sa));
1.27      misho     644:        if (ret == -1) {
                    645:                /* close connection */
                    646:                schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    647:                                TASK_ARG(task), 0, NULL, 0);
1.26      misho     648:        }
                    649: 
1.27      misho     650:        taskExit(task, NULL);
1.26      misho     651: }
                    652: 
                    653: static void *
                    654: rxRAWPacket(sched_task_t *task)
                    655: {
                    656:        rpc_srv_t *srv = TASK_ARG(task);
                    657:        rpc_cli_t *c = NULL;
1.27      misho     658:        int len, noreply = 0, rlen;
1.26      misho     659:        u_short crc;
                    660:        struct tagRPCCall *rpc;
                    661:        sockaddr_t sa;
1.28.2.3  misho     662:        socklen_t salen = E_SOCKADDR_MAX;
1.26      misho     663:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
1.27      misho     664:        ait_val_t b = AIT_VAL_INIT;
1.26      misho     665: 
                    666:        /* receive connect packet */
1.27      misho     667:        AIT_SET_BUF(&b, NULL, srv->srv_netbuf);
1.28.2.2  misho     668: #ifndef __linux__
                    669:        sa.ss.ss_len = salen;
                    670: #endif
1.27      misho     671:        rlen = recvfrom(TASK_FD(task), AIT_GET_BUF(&b), AIT_LEN(&b), 0, &sa.sa, &salen);
                    672:        if (sa.sa.sa_family == AF_INET) {
                    673:                struct ip *h;
                    674:                h = (struct ip*) AIT_GET_BUF(&b);
                    675:                if (rlen < ntohs(h->ip_len) || h->ip_p != IPPROTO_ERPC)
                    676:                        goto end;
                    677:                else {
                    678:                        rlen -= sizeof(struct ip);
                    679:                        rpc = (struct tagRPCCall*) (h + 1);
                    680:                }
                    681:        } else {
1.28.2.4  misho     682: #ifdef IPV6_REMOVE_HEADER
1.27      misho     683:                struct ip6_hdr *h;
                    684:                h = (struct ip6_hdr*) AIT_GET_BUF(&b);
1.28.2.4  misho     685:                if (rlen < ntohs(h->ip6_plen) || h->ip6_nxt != IPPROTO_ERPC)
1.27      misho     686:                        goto end;
                    687:                else {
                    688:                        rlen -= sizeof(struct ip6_hdr);
                    689:                        rpc = (struct tagRPCCall*) (h + 1);
                    690:                }
1.28.2.4  misho     691: #else
                    692:                rpc = (struct tagRPCCall*) AIT_GET_BUF(&b);
                    693: #endif
1.27      misho     694:        }
                    695:        if (rlen < sizeof(struct tagRPCCall))
                    696:                goto end;
                    697:        else
                    698:                len = ntohl(rpc->call_len);
                    699:        if (rlen < len || len > srv->srv_netbuf)
                    700:                goto end;
                    701: 
                    702:        /* skip loop packet */
                    703:        if (rpc->call_io & RPC_ACK)
                    704:                goto end;
                    705: 
                    706:        /* check integrity of packet */
                    707:        crc = ntohs(rpc->call_crc);
                    708:        rpc->call_crc ^= rpc->call_crc;
                    709:        if (crc != crcFletcher16((u_short*) rpc, len / 2))
                    710:                goto end;
                    711: 
                    712:        /* check RPC packet session info */
                    713:        if (rpc_chkPktSession(&rpc->call_session, &srv->srv_session))
1.26      misho     714:                goto end;
                    715: 
                    716:        c = _allocClient(srv, &sa);
                    717:        if (!c) {
                    718:                EVERBOSE(1, "RPC client quota exceeded! Connection will be shutdown!\n");
                    719:                usleep(2000);   /* blocked client delay */
                    720:                goto end;
                    721:        } else {
1.27      misho     722:                memcpy(AIT_GET_BUF(&c->cli_buf), rpc, len);
                    723:                rpc = (struct tagRPCCall*) AIT_GET_BUF(&c->cli_buf);
1.26      misho     724: 
                    725:                c->cli_sock = TASK_FD(task);
                    726:                memcpy(&c->cli_sa, &sa, sizeof c->cli_sa);
                    727: 
                    728:                /* armed timer for close stateless connection */
                    729:                schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL);
                    730:                schedTimer(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], 
                    731:                                c, ts, c, 0);
                    732:        }
                    733: 
                    734:        noreply = RPC_CHK_NOREPLY(rpc);
                    735: 
1.27      misho     736:        /* execute RPC call */
                    737:        schedEvent(TASK_ROOT(task), execCall, c, (int) noreply, rpc, len);
1.26      misho     738: 
                    739:        /* send RPC reply */
                    740:        if (!noreply)
                    741:                schedWrite(TASK_ROOT(task), cbProto[srv->srv_proto][CB_TXPACKET], 
                    742:                                c, TASK_FD(task), rpc, len);
                    743: end:
1.27      misho     744:        AIT_FREE_VAL(&b);
1.26      misho     745:        schedReadSelf(task);
1.27      misho     746:        taskExit(task, NULL);
1.26      misho     747: }
                    748: 
                    749: 
                    750: static void *
1.24      misho     751: txBPFPacket(sched_task_t *task)
                    752: {
1.28.2.2  misho     753: #ifndef __linux__
1.24      misho     754:        rpc_cli_t *c = TASK_ARG(task);
                    755:        rpc_srv_t *s = c->cli_parent;
                    756:        rpc_func_t *f = NULL;
                    757:        u_char *buf = AIT_GET_BUF(&c->cli_buf);
                    758:        struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
1.27      misho     759:        int ret, wlen = sizeof(struct tagRPCCall);
1.24      misho     760:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
                    761:        struct ether_header *eh;
                    762:        ait_val_t b = AIT_VAL_INIT;
                    763: 
                    764:        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
                    765:        schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    766:                         TASK_ARG(task), ts, TASK_ARG(task), 0);
                    767: 
                    768:        if (rpc->call_argc) {
                    769:                f = rpc_srv_getCall(s, ntohs(rpc->call_tag));
                    770:                if (!f) {
                    771:                        rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server");
1.27      misho     772: 
1.24      misho     773:                        rpc->call_argc ^= rpc->call_argc;
1.27      misho     774:                        RPC_SET_RETURN(rpc, -1);
                    775:                        RPC_SET_ERRNO(rpc, rpc_Errno);
                    776:                } else if (rpc_pktFreeSpace(c) > s->srv_netbuf) {
                    777:                        rpc_SetErr(EMSGSIZE, "Message too long");
                    778: 
                    779:                        rpc->call_argc ^= rpc->call_argc;
                    780:                        RPC_SET_RETURN(rpc, -1);
                    781:                        RPC_SET_ERRNO(rpc, rpc_Errno);
1.24      misho     782:                } else {
1.25      misho     783:                        rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
1.24      misho     784:                        /* Go Encapsulate variables */
                    785:                        ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, 
                    786:                                        RPC_RETVARS(c));
                    787:                        if (ret == -1) {
                    788:                                rpc_SetErr(EBADRPC, "Prepare RPC packet failed");
1.27      misho     789: 
1.24      misho     790:                                rpc->call_argc ^= rpc->call_argc;
1.27      misho     791:                                RPC_SET_RETURN(rpc, -1);
                    792:                                RPC_SET_ERRNO(rpc, rpc_Errno);
1.24      misho     793:                        } else
                    794:                                wlen += ret;
                    795:                }
                    796:        }
                    797: 
1.27      misho     798:        /* Free return values */
                    799:        ait_freeVars(&RPC_RETVARS(c));
                    800: 
1.24      misho     801:        rpc->call_len = htonl(wlen);
1.25      misho     802:        rpc->call_io = RPC_ACK;
1.24      misho     803: 
                    804:        /* calculate CRC */
                    805:        rpc->call_crc ^= rpc->call_crc;
                    806:        rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2));
                    807: 
                    808:        /* send reply */
1.27      misho     809:        AIT_SET_BUF(&b, NULL, wlen + ETHER_HDR_LEN);
1.24      misho     810:        eh = (struct ether_header*) AIT_GET_BUF(&b);
                    811:        memcpy(eh->ether_dhost, LLADDR(&c->cli_sa.sdl), ETHER_ADDR_LEN);
                    812:        eh->ether_type = htons(RPC_DEFPORT);
1.27      misho     813:        memcpy(eh + 1, buf, wlen);
1.24      misho     814: 
                    815:        ret = write(TASK_FD(task), AIT_GET_BUF(&b), AIT_LEN(&b));
                    816:        AIT_FREE_VAL(&b);
                    817:        if (ret == -1) {
                    818:                /* close connection */
                    819:                schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    820:                                TASK_ARG(task), 0, NULL, 0);
1.22      misho     821:        }
1.28.2.2  misho     822: #else
                    823:        rpc_SetErr(ENOTSUP, "Feature isn't supported on Linux!");
                    824: #endif
1.13      misho     825: 
1.27      misho     826:        taskExit(task, NULL);
1.24      misho     827: }
                    828: 
                    829: static void *
                    830: rxBPFPacket(sched_task_t *task)
                    831: {
1.28.2.2  misho     832: #ifndef __linux__
1.24      misho     833:        rpc_srv_t *srv = TASK_ARG(task);
                    834:        rpc_cli_t *c = NULL;
                    835:        int len, rlen, noreply;
                    836:        u_short crc;
                    837:        struct tagRPCCall *rpc;
                    838:        sockaddr_t sa;
                    839:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
                    840:        struct bpf_hdr *h;
                    841:        struct ether_header *eh;
                    842:        ait_val_t b = AIT_VAL_INIT;
                    843: 
                    844:        /* receive connect packet */
                    845:        AIT_SET_BUF(&b, NULL, srv->srv_netbuf);
                    846:        rlen = read(TASK_FD(task), AIT_GET_BUF(&b), AIT_LEN(&b));
                    847:        h = (struct bpf_hdr*) AIT_GET_BUF(&b);
                    848:        rlen -= h->bh_hdrlen;
                    849:        if (rlen < h->bh_datalen || h->bh_caplen != h->bh_datalen || 
1.27      misho     850:                        rlen < ETHER_HDR_LEN + sizeof(struct tagRPCCall))
1.24      misho     851:                goto end;
1.27      misho     852:        else {
1.24      misho     853:                rlen = h->bh_caplen;
                    854:                eh = (struct ether_header*) (AIT_GET_BUF(&b) + h->bh_hdrlen);
                    855:                rlen -= ETHER_HDR_LEN;
                    856:                rpc = (struct tagRPCCall*) (eh + 1);
1.25      misho     857: 
1.24      misho     858:                if (eh->ether_type != ntohs(RPC_DEFPORT))
                    859:                        goto end;
                    860:                else
                    861:                        e_getlinkbymac((const ether_addr_t*) eh->ether_shost, &sa);
                    862:        }
1.27      misho     863:        if (rlen < sizeof(struct tagRPCCall))
                    864:                goto end;
                    865:        else
                    866:                len = ntohl(rpc->call_len);
                    867:        if (rlen < len || len > srv->srv_netbuf)
                    868:                goto end;
                    869: 
                    870: #ifdef CHECK_ETHACK
                    871:        /* skip loop packet */
                    872:        if (rpc->call_io & RPC_ACK)
                    873:                goto end;
                    874: #endif
                    875: 
                    876:        /* check integrity of packet */
                    877:        crc = ntohs(rpc->call_crc);
                    878:        rpc->call_crc ^= rpc->call_crc;
                    879:        if (crc != crcFletcher16((u_short*) rpc, len / 2))
                    880:                goto end;
                    881: 
                    882:        /* check RPC packet session info */
                    883:        if (rpc_chkPktSession(&rpc->call_session, &srv->srv_session))
                    884:                goto end;
1.24      misho     885: 
                    886:        c = _allocClient(srv, &sa);
                    887:        if (!c) {
                    888:                EVERBOSE(1, "RPC client quota exceeded! Connection will be shutdown!\n");
                    889:                usleep(2000);   /* blocked client delay */
                    890:                goto end;
                    891:        } else {
1.27      misho     892:                memcpy(AIT_GET_BUF(&c->cli_buf), rpc, len);
1.24      misho     893:                rpc = (struct tagRPCCall*) AIT_GET_BUF(&c->cli_buf);
                    894: 
                    895:                c->cli_sock = TASK_FD(task);
                    896:                memcpy(&c->cli_sa, &sa, sizeof c->cli_sa);
                    897: 
                    898:                /* armed timer for close stateless connection */
                    899:                schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL);
                    900:                schedTimer(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], 
                    901:                                c, ts, c, 0);
                    902:        }
                    903: 
1.22      misho     904:        noreply = RPC_CHK_NOREPLY(rpc);
1.18      misho     905: 
1.27      misho     906:        /* execute RPC call */
                    907:        schedEvent(TASK_ROOT(task), execCall, c, (int) noreply, rpc, len);
1.13      misho     908: 
1.22      misho     909:        /* send RPC reply */
                    910:        if (!noreply)
1.24      misho     911:                schedEvent(TASK_ROOT(task), cbProto[srv->srv_proto][CB_TXPACKET], 
1.22      misho     912:                                c, TASK_FD(task), rpc, len);
1.13      misho     913: end:
1.24      misho     914:        AIT_FREE_VAL(&b);
1.13      misho     915:        schedReadSelf(task);
1.28.2.2  misho     916: #else
                    917:        rpc_SetErr(ENOTSUP, "Feature isn't supported on Linux!");
                    918: #endif
                    919: 
1.27      misho     920:        taskExit(task, NULL);
1.13      misho     921: }
                    922: 
1.25      misho     923: 
                    924: static void *
                    925: txEXTPacket(sched_task_t *task)
                    926: {
                    927:        rpc_cli_t *c = TASK_ARG(task);
                    928:        rpc_srv_t *s = c->cli_parent;
                    929:        rpc_func_t *f = NULL;
                    930:        u_char *buf = AIT_GET_BUF(&c->cli_buf);
                    931:        struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
1.27      misho     932:        int ret, wlen = sizeof(struct tagRPCCall);
1.25      misho     933:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
                    934: 
                    935:        schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL);
                    936:        schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    937:                         TASK_ARG(task), ts, TASK_ARG(task), 0);
                    938: 
                    939:        if (rpc->call_argc) {
                    940:                f = rpc_srv_getCall(s, ntohs(rpc->call_tag));
                    941:                if (!f) {
                    942:                        rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server");
1.27      misho     943: 
                    944:                        rpc->call_argc ^= rpc->call_argc;
                    945:                        RPC_SET_RETURN(rpc, -1);
                    946:                        RPC_SET_ERRNO(rpc, rpc_Errno);
                    947:                } else if (rpc_pktFreeSpace(c) > s->srv_netbuf) {
                    948:                        rpc_SetErr(EMSGSIZE, "Message too long");
                    949: 
1.25      misho     950:                        rpc->call_argc ^= rpc->call_argc;
1.27      misho     951:                        RPC_SET_RETURN(rpc, -1);
                    952:                        RPC_SET_ERRNO(rpc, rpc_Errno);
1.25      misho     953:                } else {
                    954:                        rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
                    955:                        /* Go Encapsulate variables */
                    956:                        ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, 
                    957:                                        RPC_RETVARS(c));
                    958:                        if (ret == -1) {
                    959:                                rpc_SetErr(EBADRPC, "Prepare RPC packet failed");
1.27      misho     960: 
1.25      misho     961:                                rpc->call_argc ^= rpc->call_argc;
1.27      misho     962:                                RPC_SET_RETURN(rpc, -1);
                    963:                                RPC_SET_ERRNO(rpc, rpc_Errno);
1.25      misho     964:                        } else
                    965:                                wlen += ret;
                    966:                }
                    967:        }
                    968: 
1.27      misho     969:        /* Free return values */
                    970:        ait_freeVars(&RPC_RETVARS(c));
                    971: 
1.25      misho     972:        rpc->call_len = htonl(wlen);
                    973:        rpc->call_io = RPC_ACK;
                    974: 
                    975:        /* send reply */
1.27      misho     976:        ret = write(TASK_FD(task), buf, wlen);
1.25      misho     977:        if (ret == -1) {
                    978:                /* close connection */
                    979:                schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
                    980:                                TASK_ARG(task), 0, NULL, 0);
                    981:        }
                    982: 
1.27      misho     983:        taskExit(task, NULL);
1.25      misho     984: }
                    985: 
                    986: static void *
                    987: rxEXTPacket(sched_task_t *task)
                    988: {
                    989:        rpc_srv_t *srv = TASK_ARG(task);
                    990:        rpc_cli_t *c = NULL;
1.27      misho     991:        int len, noreply = 0, rlen = AIT_LEN(&c->cli_buf);
1.25      misho     992:        struct tagRPCCall *rpc;
                    993:        struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
1.27      misho     994:        sockaddr_t sa;
1.25      misho     995:        ait_val_t b = AIT_VAL_INIT;
                    996: 
                    997:        memset(&sa, 0, sizeof sa);
                    998:        /* receive connect packet */
                    999:        AIT_SET_BUF(&b, NULL, srv->srv_netbuf);
                   1000:        rlen = read(TASK_FD(task), AIT_GET_BUF(&b), AIT_LEN(&b));
1.27      misho    1001:        rpc = (struct tagRPCCall*) AIT_GET_BUF(&b);
                   1002:        if (rlen < sizeof(struct tagRPCCall))
                   1003:                goto end;
                   1004:        else
                   1005:                len = ntohl(rpc->call_len);
                   1006:        if (rlen < len || len > srv->srv_netbuf)
                   1007:                goto end;
                   1008: 
                   1009:        /* skip loop packet */
                   1010:        if (rpc->call_io & RPC_ACK)
1.25      misho    1011:                goto end;
                   1012: 
1.27      misho    1013:        /* check RPC packet session info */
                   1014:        if (rpc_chkPktSession(&rpc->call_session, &srv->srv_session))
                   1015:                goto end;
1.25      misho    1016: 
                   1017:        c = _allocClient(srv, &sa);
                   1018:        if (!c) {
                   1019:                EVERBOSE(1, "RPC client quota exceeded! Connection will be shutdown!\n");
                   1020:                usleep(2000);   /* blocked client delay */
                   1021:                goto end;
                   1022:        } else {
1.27      misho    1023:                memcpy(AIT_GET_BUF(&c->cli_buf), AIT_GET_BUF(&b), len);
1.25      misho    1024:                rpc = (struct tagRPCCall*) AIT_GET_BUF(&c->cli_buf);
                   1025: 
                   1026:                c->cli_sock = TASK_FD(task);
                   1027: 
                   1028:                /* armed timer for close stateless connection */
                   1029:                schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL);
                   1030:                schedTimer(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], 
                   1031:                                c, ts, c, 0);
                   1032:        }
                   1033: 
                   1034:        noreply = RPC_CHK_NOREPLY(rpc);
                   1035: 
1.27      misho    1036:        /* execute RPC call */
                   1037:        schedEvent(TASK_ROOT(task), execCall, c, (int) noreply, rpc, len);
1.25      misho    1038: 
                   1039:        /* send RPC reply */
                   1040:        if (!noreply)
                   1041:                schedWrite(TASK_ROOT(task), cbProto[srv->srv_proto][CB_TXPACKET], 
                   1042:                                c, TASK_FD(task), rpc, len);
                   1043: end:
                   1044:        AIT_FREE_VAL(&b);
                   1045:        schedReadSelf(task);
1.27      misho    1046:        taskExit(task, NULL);
1.25      misho    1047: }
                   1048: 
1.13      misho    1049: /* ------------------------------------------------------ */
                   1050: 
1.16      misho    1051: void
1.13      misho    1052: rpc_freeBLOBCli(rpc_cli_t * __restrict c)
                   1053: {
                   1054:        rpc_srv_t *s = c->cli_parent;
                   1055: 
                   1056:        schedCancelby(s->srv_blob.root, taskMAX, CRITERIA_ARG, c, NULL);
1.10      misho    1057: 
                   1058:        /* free buffer */
                   1059:        AIT_FREE_VAL(&c->cli_buf);
                   1060: 
1.14      misho    1061:        array_Del(s->srv_blob.clients, c->cli_id, 0);
1.10      misho    1062:        if (c)
1.14      misho    1063:                e_free(c);
1.13      misho    1064: }
                   1065: 
                   1066: 
                   1067: static void *
                   1068: closeBLOBClient(sched_task_t *task)
                   1069: {
                   1070:        int sock = ((rpc_cli_t*) TASK_ARG(task))->cli_sock;
                   1071: 
                   1072:        rpc_freeBLOBCli(TASK_ARG(task));
                   1073: 
                   1074:        /* close client socket */
                   1075:        shutdown(sock, SHUT_RDWR);
                   1076:        close(sock);
1.27      misho    1077:        taskExit(task, NULL);
1.7       misho    1078: }
                   1079: 
                   1080: static void *
                   1081: txBLOB(sched_task_t *task)
                   1082: {
1.10      misho    1083:        rpc_cli_t *c = TASK_ARG(task);
                   1084:        u_char *buf = AIT_GET_BUF(&c->cli_buf);
1.7       misho    1085:        int wlen = sizeof(struct tagBLOBHdr);
                   1086: 
                   1087:        /* send reply */
1.10      misho    1088:        wlen = send(TASK_FD(task), buf, wlen, MSG_NOSIGNAL);
                   1089:        if (wlen == -1 || wlen != sizeof(struct tagBLOBHdr)) {
                   1090:                /* close blob connection */
                   1091:                schedEvent(TASK_ROOT(task), closeBLOBClient, c, 42, NULL, 0);
                   1092:        }
1.7       misho    1093: 
1.27      misho    1094:        taskExit(task, NULL);
1.7       misho    1095: }
1.4       misho    1096: 
1.7       misho    1097: static void *
                   1098: rxBLOB(sched_task_t *task)
                   1099: {
                   1100:        rpc_cli_t *c = TASK_ARG(task);
                   1101:        rpc_srv_t *s = c->cli_parent;
                   1102:        rpc_blob_t *b;
1.10      misho    1103:        struct tagBLOBHdr blob;
1.7       misho    1104:        int rlen;
                   1105: 
1.10      misho    1106:        memset(&blob, 0, sizeof blob);
                   1107:        rlen = recv(TASK_FD(task), &blob, sizeof blob, 0);
                   1108:        if (rlen < 1) {
                   1109:                /* close blob connection */
                   1110:                schedEvent(TASK_ROOT(task), closeBLOBClient, c, 42, NULL, 0);
1.27      misho    1111:                taskExit(task, NULL);
1.7       misho    1112:        }
1.5       misho    1113: 
1.10      misho    1114:        /* check BLOB packet */
                   1115:        if (rlen < sizeof(struct tagBLOBHdr)) {
                   1116:                rpc_SetErr(ERPCMISMATCH, "Short BLOB packet");
1.6       misho    1117: 
1.10      misho    1118:                schedReadSelf(task);
1.27      misho    1119:                taskExit(task, NULL);
1.7       misho    1120:        }
1.1       misho    1121: 
1.7       misho    1122:        /* check RPC packet session info */
1.15      misho    1123:        if (rpc_chkPktSession(&blob.hdr_session, &s->srv_session)) {
1.7       misho    1124:                rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session");
1.10      misho    1125:                blob.hdr_cmd = error;
1.7       misho    1126:                goto end;
                   1127:        }
                   1128: 
                   1129:        /* Go to proceed packet ... */
1.10      misho    1130:        switch (blob.hdr_cmd) {
1.7       misho    1131:                case get:
1.10      misho    1132:                        if (!(b = rpc_srv_getBLOB(s, ntohl(blob.hdr_var)))) {
                   1133:                                rpc_SetErr(EINVAL, "Var=%x not found", ntohl(blob.hdr_var));
                   1134:                                blob.hdr_cmd = no;
1.27      misho    1135:                                RPC_SET_BLOB_RET(&blob, -1);
1.7       misho    1136:                                break;
                   1137:                        } else
1.10      misho    1138:                                blob.hdr_len = htonl(b->blob_len);
1.5       misho    1139: 
1.7       misho    1140:                        if (rpc_srv_blobMap(s, b) != -1) {
                   1141:                                /* deliver BLOB variable to client */
1.10      misho    1142:                                blob.hdr_ret = htonl(rpc_srv_sendBLOB(c, b));
1.7       misho    1143:                                rpc_srv_blobUnmap(b);
                   1144:                        } else {
1.10      misho    1145:                                blob.hdr_cmd = error;
1.27      misho    1146:                                RPC_SET_BLOB_RET(&blob, -1);
1.7       misho    1147:                        }
                   1148:                        break;
                   1149:                case set:
1.17      misho    1150:                        if ((b = rpc_srv_registerBLOB(s, ntohl(blob.hdr_len), 
                   1151:                                                        ntohl(blob.hdr_ret)))) {
1.7       misho    1152:                                /* set new BLOB variable for reply :) */
1.10      misho    1153:                                blob.hdr_var = htonl(b->blob_var);
1.7       misho    1154: 
                   1155:                                /* receive BLOB from client */
1.10      misho    1156:                                blob.hdr_ret = htonl(rpc_srv_recvBLOB(c, b));
1.7       misho    1157:                                rpc_srv_blobUnmap(b);
1.5       misho    1158:                        } else {
1.10      misho    1159:                                blob.hdr_cmd = error;
1.27      misho    1160:                                RPC_SET_BLOB_RET(&blob, -1);
1.7       misho    1161:                        }
                   1162:                        break;
                   1163:                case unset:
1.11      misho    1164:                        if (rpc_srv_unregisterBLOB(s, ntohl(blob.hdr_var)) == -1) {
1.10      misho    1165:                                blob.hdr_cmd = error;
1.27      misho    1166:                                RPC_SET_BLOB_RET(&blob, -1);
1.1       misho    1167:                        }
                   1168:                        break;
1.7       misho    1169:                default:
1.10      misho    1170:                        rpc_SetErr(EPROCUNAVAIL, "Unsupported BLOB command %d", blob.hdr_cmd);
                   1171:                        blob.hdr_cmd = error;
1.27      misho    1172:                        RPC_SET_BLOB_RET(&blob, -1);
1.7       misho    1173:        }
1.1       misho    1174: 
1.7       misho    1175: end:
1.10      misho    1176:        memcpy(AIT_ADDR(&c->cli_buf), &blob, sizeof blob);
                   1177:        schedWrite(TASK_ROOT(task), txBLOB, TASK_ARG(task), TASK_FD(task), NULL, 0);
                   1178:        schedReadSelf(task);
1.27      misho    1179:        taskExit(task, NULL);
1.2       misho    1180: }
                   1181: 
                   1182: static void *
1.17      misho    1183: flushBLOB(sched_task_t *task)
                   1184: {
1.28      misho    1185: #ifdef atomic_load_acq_ptr
1.23      misho    1186:        uintptr_t sigArg = atomic_load_acq_ptr(&_glSigArg);
1.28      misho    1187: #else
                   1188:        uintptr_t sigArg = *((volatile uintptr_t*) &_glSigArg);
                   1189: #endif
1.23      misho    1190:        rpc_srv_t *srv = sigArg ? (void*) sigArg : TASK_ARG(task);
1.17      misho    1191:        rpc_blob_t *b, *tmp;
                   1192: 
                   1193:        TAILQ_FOREACH_SAFE(b, &srv->srv_blob.blobs, blob_node, tmp) {
                   1194:                TAILQ_REMOVE(&srv->srv_blob.blobs, b, blob_node);
                   1195: 
                   1196:                rpc_srv_blobFree(srv, b);
                   1197:                e_free(b);
                   1198:        }
                   1199: 
1.28.2.5! misho    1200:        if (sigArg) {
1.23      misho    1201:                /* disabled kqueue support in libaitsched */
                   1202:                struct sigaction sa;
                   1203: 
                   1204:                memset(&sa, 0, sizeof sa);
                   1205:                sigemptyset(&sa.sa_mask);
                   1206:                sa.sa_handler = (void (*)(int)) flushBLOB;
                   1207:                sa.sa_flags = SA_RESTART | SA_RESETHAND;
                   1208:                sigaction(SIGFBLOB, &sa, NULL);
1.28.2.5! misho    1209:                return NULL;
        !          1210:        } else {
        !          1211:                schedSignalSelf(task);
        !          1212:                taskExit(task, NULL);
1.23      misho    1213:        }
1.17      misho    1214: }
                   1215: 
                   1216: static void *
1.10      misho    1217: acceptBLOBClients(sched_task_t *task)
1.2       misho    1218: {
1.10      misho    1219:        rpc_srv_t *srv = TASK_ARG(task);
                   1220:        rpc_cli_t *c = NULL;
                   1221:        register int i;
1.28.2.3  misho    1222:        socklen_t salen = E_SOCKADDR_MAX;
1.21      misho    1223:        int sock;
1.12      misho    1224: #ifdef TCP_NOPUSH
                   1225:        int n = 1;
                   1226: #endif
1.7       misho    1227: 
1.10      misho    1228:        /* check free slots for connect */
1.14      misho    1229:        for (i = 0; i < array_Size(srv->srv_blob.clients) && 
                   1230:                        (c = array(srv->srv_blob.clients, i, rpc_cli_t*)); i++);
1.21      misho    1231:        if (c) {        /* no more free slots! */
                   1232:                EVERBOSE(1, "BLOB client quota exceeded! Connection will be shutdown!\n");
                   1233:                if ((sock = accept(TASK_FD(task), NULL, NULL)) != -1) {
                   1234:                        shutdown(sock, SHUT_RDWR);
                   1235:                        close(sock);
                   1236:                }
1.10      misho    1237:                goto end;
1.21      misho    1238:        }
                   1239: 
1.14      misho    1240:        c = e_malloc(sizeof(rpc_cli_t));
1.10      misho    1241:        if (!c) {
1.7       misho    1242:                LOGERR;
1.10      misho    1243:                srv->srv_kill = srv->srv_blob.kill = 1;
1.27      misho    1244:                taskExit(task, NULL);
1.7       misho    1245:        } else {
1.10      misho    1246:                memset(c, 0, sizeof(rpc_cli_t));
1.14      misho    1247:                array_Set(srv->srv_blob.clients, i, c);
1.10      misho    1248:                c->cli_id = i;
                   1249:                c->cli_parent = srv;
1.7       misho    1250:        }
1.4       misho    1251: 
1.10      misho    1252:        /* alloc empty buffer */
1.14      misho    1253:        AIT_SET_BUFSIZ(&c->cli_buf, 0, srv->srv_netbuf);
1.2       misho    1254: 
1.10      misho    1255:        /* accept client */
                   1256:        c->cli_sock = accept(TASK_FD(task), &c->cli_sa.sa, &salen);
                   1257:        if (c->cli_sock == -1) {
                   1258:                LOGERR;
                   1259:                AIT_FREE_VAL(&c->cli_buf);
1.14      misho    1260:                array_Del(srv->srv_blob.clients, i, 42);
1.10      misho    1261:                goto end;
1.12      misho    1262:        } else {
                   1263: #ifdef TCP_NOPUSH
                   1264:                setsockopt(c->cli_sock, IPPROTO_TCP, TCP_NOPUSH, &n, sizeof n);
                   1265: #endif
1.10      misho    1266:                fcntl(c->cli_sock, F_SETFL, fcntl(c->cli_sock, F_GETFL) | O_NONBLOCK);
1.12      misho    1267:        }
1.2       misho    1268: 
1.10      misho    1269:        schedRead(TASK_ROOT(task), rxBLOB, c, c->cli_sock, NULL, 0);
                   1270: end:
                   1271:        schedReadSelf(task);
1.27      misho    1272:        taskExit(task, NULL);
1.1       misho    1273: }
                   1274: 
1.10      misho    1275: /* ------------------------------------------------------ */
1.1       misho    1276: 
                   1277: /*
1.7       misho    1278:  * rpc_srv_initBLOBServer() - Init & create BLOB Server
                   1279:  *
1.4       misho    1280:  * @srv = RPC server instance
1.2       misho    1281:  * @Port = Port for bind server, if Port == 0 default port is selected
                   1282:  * @diskDir = Disk place for BLOB file objects
                   1283:  * return: -1 == error or 0 bind and created BLOB server instance
                   1284:  */
                   1285: int
                   1286: rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_short Port, const char *diskDir)
                   1287: {
                   1288:        int n = 1;
1.28.2.3  misho    1289:        socklen_t salen;
1.2       misho    1290: 
1.10      misho    1291:        if (!srv || srv->srv_kill) {
1.7       misho    1292:                rpc_SetErr(EINVAL, "Invalid parameters can`t init BLOB server");
1.2       misho    1293:                return -1;
                   1294:        }
                   1295: 
                   1296:        memset(&srv->srv_blob, 0, sizeof srv->srv_blob);
                   1297:        if (access(diskDir, R_OK | W_OK) == -1) {
                   1298:                LOGERR;
                   1299:                return -1;
                   1300:        } else
1.7       misho    1301:                AIT_SET_STR(&srv->srv_blob.dir, diskDir);
1.2       misho    1302: 
1.10      misho    1303:        /* init blob list */
                   1304:        TAILQ_INIT(&srv->srv_blob.blobs);
                   1305: 
1.2       misho    1306:        srv->srv_blob.server.cli_parent = srv;
1.4       misho    1307: 
1.28.2.4  misho    1308:        memcpy(&srv->srv_blob.server.cli_sa, &srv->srv_server.cli_sa, sizeof srv->srv_blob.server.cli_sa);
1.10      misho    1309:        switch (srv->srv_blob.server.cli_sa.sa.sa_family) {
1.4       misho    1310:                case AF_INET:
1.10      misho    1311:                        srv->srv_blob.server.cli_sa.sin.sin_port = 
                   1312:                                htons(Port ? Port : ntohs(srv->srv_blob.server.cli_sa.sin.sin_port) + 1);
1.28.2.3  misho    1313:                        salen = sizeof srv->srv_blob.server.cli_sa.sin;
1.4       misho    1314:                        break;
                   1315:                case AF_INET6:
1.10      misho    1316:                        srv->srv_blob.server.cli_sa.sin6.sin6_port = 
                   1317:                                htons(Port ? Port : ntohs(srv->srv_blob.server.cli_sa.sin6.sin6_port) + 1);
1.28.2.3  misho    1318:                        salen = sizeof srv->srv_blob.server.cli_sa.sin6;
1.4       misho    1319:                        break;
                   1320:                case AF_LOCAL:
1.10      misho    1321:                        strlcat(srv->srv_blob.server.cli_sa.sun.sun_path, ".blob", 
                   1322:                                        sizeof srv->srv_blob.server.cli_sa.sun.sun_path);
1.28.2.3  misho    1323:                        salen = sizeof srv->srv_blob.server.cli_sa.sun;
1.4       misho    1324:                        break;
                   1325:                default:
1.7       misho    1326:                        AIT_FREE_VAL(&srv->srv_blob.dir);
1.4       misho    1327:                        return -1;
1.2       misho    1328:        }
                   1329: 
1.4       misho    1330:        /* create BLOB server socket */
1.6       misho    1331:        srv->srv_blob.server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, SOCK_STREAM, 0);
1.2       misho    1332:        if (srv->srv_blob.server.cli_sock == -1) {
                   1333:                LOGERR;
1.7       misho    1334:                AIT_FREE_VAL(&srv->srv_blob.dir);
1.2       misho    1335:                return -1;
                   1336:        }
                   1337:        if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
                   1338:                LOGERR;
                   1339:                close(srv->srv_blob.server.cli_sock);
1.7       misho    1340:                AIT_FREE_VAL(&srv->srv_blob.dir);
1.2       misho    1341:                return -1;
                   1342:        }
1.5       misho    1343:        n = srv->srv_netbuf;
                   1344:        if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) {
                   1345:                LOGERR;
                   1346:                close(srv->srv_blob.server.cli_sock);
1.7       misho    1347:                AIT_FREE_VAL(&srv->srv_blob.dir);
1.5       misho    1348:                return -1;
                   1349:        }
                   1350:        if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) {
                   1351:                LOGERR;
                   1352:                close(srv->srv_blob.server.cli_sock);
1.7       misho    1353:                AIT_FREE_VAL(&srv->srv_blob.dir);
1.5       misho    1354:                return -1;
                   1355:        }
1.28.2.3  misho    1356:        if (bind(srv->srv_blob.server.cli_sock, &srv->srv_blob.server.cli_sa.sa, salen) == -1) {
1.2       misho    1357:                LOGERR;
                   1358:                close(srv->srv_blob.server.cli_sock);
1.7       misho    1359:                AIT_FREE_VAL(&srv->srv_blob.dir);
1.2       misho    1360:                return -1;
1.13      misho    1361:        } else
                   1362:                fcntl(srv->srv_blob.server.cli_sock, F_SETFL, 
                   1363:                                fcntl(srv->srv_blob.server.cli_sock, F_GETFL) | O_NONBLOCK);
                   1364: 
1.2       misho    1365: 
1.10      misho    1366:        /* allocate pool for concurent blob clients */
1.14      misho    1367:        srv->srv_blob.clients = array_Init(array_Size(srv->srv_clients));
1.2       misho    1368:        if (!srv->srv_blob.clients) {
1.14      misho    1369:                rpc_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
1.2       misho    1370:                close(srv->srv_blob.server.cli_sock);
1.7       misho    1371:                AIT_FREE_VAL(&srv->srv_blob.dir);
1.2       misho    1372:                return -1;
1.10      misho    1373:        }
1.2       misho    1374: 
1.10      misho    1375:        /* init blob scheduler */
                   1376:        srv->srv_blob.root = schedBegin();
                   1377:        if (!srv->srv_blob.root) {
                   1378:                rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
1.14      misho    1379:                array_Destroy(&srv->srv_blob.clients);
1.10      misho    1380:                close(srv->srv_blob.server.cli_sock);
                   1381:                AIT_FREE_VAL(&srv->srv_blob.dir);
                   1382:                return -1;
                   1383:        }
1.2       misho    1384: 
                   1385:        return 0;
                   1386: }
                   1387: 
                   1388: /*
1.7       misho    1389:  * rpc_srv_endBLOBServer() - Destroy BLOB server, close all opened sockets and free resources
                   1390:  *
1.2       misho    1391:  * @srv = RPC Server instance
                   1392:  * return: none
                   1393:  */
1.16      misho    1394: void
1.2       misho    1395: rpc_srv_endBLOBServer(rpc_srv_t * __restrict srv)
                   1396: {
1.10      misho    1397:        if (!srv)
1.2       misho    1398:                return;
                   1399: 
1.10      misho    1400:        srv->srv_blob.kill = 1;
1.17      misho    1401: 
1.27      misho    1402:        if (srv->srv_blob.server.cli_sa.sa.sa_family == AF_LOCAL)
                   1403:                unlink(srv->srv_blob.server.cli_sa.sun.sun_path);
1.28      misho    1404: 
                   1405:        schedEnd(&srv->srv_blob.root);
1.2       misho    1406: }
                   1407: 
                   1408: /*
1.10      misho    1409:  * rpc_srv_loopBLOBServer() - Execute Main BLOB server loop and wait for clients requests
1.7       misho    1410:  *
1.2       misho    1411:  * @srv = RPC Server instance
                   1412:  * return: -1 error or 0 ok, infinite loop ...
                   1413:  */
                   1414: int
1.10      misho    1415: rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv)
1.2       misho    1416: {
1.10      misho    1417:        rpc_cli_t *c;
1.2       misho    1418:        register int i;
1.10      misho    1419:        rpc_blob_t *b, *tmp;
                   1420:        struct timespec ts = { RPC_SCHED_POLLING, 0 };
1.2       misho    1421: 
1.10      misho    1422:        if (!srv || srv->srv_kill) {
1.7       misho    1423:                rpc_SetErr(EINVAL, "Invalid parameter can`t start BLOB server");
1.2       misho    1424:                return -1;
                   1425:        }
                   1426: 
1.14      misho    1427:        if (listen(srv->srv_blob.server.cli_sock, array_Size(srv->srv_blob.clients)) == -1) {
1.2       misho    1428:                LOGERR;
                   1429:                return -1;
1.10      misho    1430:        }
                   1431: 
1.23      misho    1432:        if (!schedSignal(srv->srv_blob.root, flushBLOB, srv, SIGFBLOB, NULL, 0)) {
                   1433:                /* disabled kqueue support in libaitsched */
                   1434:                struct sigaction sa;
                   1435: 
1.28      misho    1436: #ifdef atomic_store_rel_ptr
1.23      misho    1437:                atomic_store_rel_ptr(&_glSigArg, (uintptr_t) srv);
1.28      misho    1438: #else
                   1439:                *((volatile uintptr_t*) &_glSigArg) = (uintptr_t) srv;
                   1440: #endif
1.23      misho    1441: 
                   1442:                memset(&sa, 0, sizeof sa);
                   1443:                sigemptyset(&sa.sa_mask);
                   1444:                sa.sa_handler = (void (*)(int)) flushBLOB;
                   1445:                sa.sa_flags = SA_RESTART | SA_RESETHAND;
                   1446:                sigaction(SIGFBLOB, &sa, NULL);
                   1447:        }
                   1448: 
1.10      misho    1449:        if (!schedRead(srv->srv_blob.root, acceptBLOBClients, srv, 
                   1450:                                srv->srv_blob.server.cli_sock, NULL, 0)) {
                   1451:                rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                   1452:                return -1;
                   1453:        }
1.2       misho    1454: 
1.10      misho    1455:        schedPolling(srv->srv_blob.root, &ts, NULL);
                   1456:        /* main rpc loop */
                   1457:        schedRun(srv->srv_blob.root, &srv->srv_blob.kill);
1.7       misho    1458: 
1.17      misho    1459:        /* detach blobs */
                   1460:        TAILQ_FOREACH_SAFE(b, &srv->srv_blob.blobs, blob_node, tmp) {
                   1461:                TAILQ_REMOVE(&srv->srv_blob.blobs, b, blob_node);
                   1462: 
                   1463:                rpc_srv_blobFree(srv, b);
                   1464:                e_free(b);
                   1465:        }
                   1466: 
1.10      misho    1467:        /* close all clients connections & server socket */
1.14      misho    1468:        for (i = 0; i < array_Size(srv->srv_blob.clients); i++) {
                   1469:                c = array(srv->srv_blob.clients, i, rpc_cli_t*);
1.10      misho    1470:                if (c) {
                   1471:                        shutdown(c->cli_sock, SHUT_RDWR);
                   1472:                        close(c->cli_sock);
                   1473: 
                   1474:                        schedCancelby(srv->srv_blob.root, taskMAX, CRITERIA_ARG, c, NULL);
                   1475:                        AIT_FREE_VAL(&c->cli_buf);
1.2       misho    1476:                }
1.14      misho    1477:                array_Del(srv->srv_blob.clients, i, 42);
1.10      misho    1478:        }
1.14      misho    1479:        array_Destroy(&srv->srv_blob.clients);
1.2       misho    1480: 
1.10      misho    1481:        close(srv->srv_blob.server.cli_sock);
1.2       misho    1482: 
1.10      misho    1483:        AIT_FREE_VAL(&srv->srv_blob.dir);
1.2       misho    1484:        return 0;
                   1485: }
                   1486: 
                   1487: 
                   1488: /*
1.7       misho    1489:  * rpc_srv_initServer() - Init & create RPC Server
                   1490:  *
1.15      misho    1491:  * @InstID = Instance for authentication & recognition
1.1       misho    1492:  * @concurentClients = Concurent clients at same time to this server
1.10      misho    1493:  * @netBuf = Network buffer length (min:512 bytes), if =0 == BUFSIZ (also meaning max RPC packet)
1.4       misho    1494:  * @csHost = Host name or address for bind server, if NULL any address
1.1       misho    1495:  * @Port = Port for bind server, if Port == 0 default port is selected
1.13      misho    1496:  * @proto = Protocol, if == 0 choose SOCK_STREAM
1.1       misho    1497:  * return: NULL == error or !=NULL bind and created RPC server instance
                   1498:  */
                   1499: rpc_srv_t *
1.15      misho    1500: rpc_srv_initServer(u_char InstID, int concurentClients, int netBuf, 
                   1501:                const char *csHost, u_short Port, int proto)
1.1       misho    1502: {
1.10      misho    1503:        int n = 1;
1.1       misho    1504:        rpc_srv_t *srv = NULL;
1.14      misho    1505:        sockaddr_t sa = E_SOCKADDR_INIT;
1.28.2.3  misho    1506:        socklen_t salen;
1.1       misho    1507: 
1.25      misho    1508:        if (!concurentClients || (proto < 0 || proto > SOCK_RAW)) {
1.10      misho    1509:                rpc_SetErr(EINVAL, "Invalid parameters can`t init RPC server");
1.1       misho    1510:                return NULL;
                   1511:        }
1.27      misho    1512:        if (!Port && proto < SOCK_RAW)
                   1513:                Port = RPC_DEFPORT;
1.28.2.3  misho    1514:        if (!(salen = e_gethostbyname(csHost, Port, &sa)))
1.10      misho    1515:                return NULL;
1.13      misho    1516:        if (!proto)
                   1517:                proto = SOCK_STREAM;
1.10      misho    1518:        if (netBuf < RPC_MIN_BUFSIZ)
1.5       misho    1519:                netBuf = BUFSIZ;
1.7       misho    1520:        else
1.14      misho    1521:                netBuf = E_ALIGN(netBuf, 2);    /* align netBuf length */
1.10      misho    1522: 
                   1523: #ifdef HAVE_SRANDOMDEV
                   1524:        srandomdev();
                   1525: #else
                   1526:        time_t tim;
                   1527: 
                   1528:        srandom((time(&tim) ^ getpid()));
                   1529: #endif
1.1       misho    1530: 
1.14      misho    1531:        srv = e_malloc(sizeof(rpc_srv_t));
1.1       misho    1532:        if (!srv) {
                   1533:                LOGERR;
                   1534:                return NULL;
                   1535:        } else
                   1536:                memset(srv, 0, sizeof(rpc_srv_t));
                   1537: 
1.13      misho    1538:        srv->srv_proto = proto;
1.5       misho    1539:        srv->srv_netbuf = netBuf;
1.1       misho    1540:        srv->srv_session.sess_version = RPC_VERSION;
1.15      misho    1541:        srv->srv_session.sess_instance = InstID;
1.1       misho    1542: 
                   1543:        srv->srv_server.cli_parent = srv;
1.10      misho    1544:        memcpy(&srv->srv_server.cli_sa, &sa, sizeof srv->srv_server.cli_sa);
                   1545: 
1.12      misho    1546:        /* init functions */
                   1547:        pthread_mutex_init(&srv->srv_funcs.mtx, NULL);
                   1548:        SLIST_INIT(&srv->srv_funcs);
                   1549:        AVL_INIT(&srv->srv_funcs);
1.10      misho    1550: 
                   1551:        /* init scheduler */
                   1552:        srv->srv_root = schedBegin();
                   1553:        if (!srv->srv_root) {
                   1554:                rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
1.12      misho    1555:                pthread_mutex_destroy(&srv->srv_funcs.mtx);
1.14      misho    1556:                e_free(srv);
1.10      misho    1557:                return NULL;
                   1558:        }
                   1559: 
                   1560:        /* init pool for clients */
1.14      misho    1561:        srv->srv_clients = array_Init(concurentClients);
1.10      misho    1562:        if (!srv->srv_clients) {
1.14      misho    1563:                rpc_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
1.10      misho    1564:                schedEnd(&srv->srv_root);
1.12      misho    1565:                pthread_mutex_destroy(&srv->srv_funcs.mtx);
1.14      misho    1566:                e_free(srv);
1.10      misho    1567:                return NULL;
                   1568:        }
1.4       misho    1569: 
                   1570:        /* create server socket */
1.26      misho    1571:        srv->srv_server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, 
                   1572:                        srv->srv_proto, srv->srv_proto == SOCK_RAW ? IPPROTO_ERPC : 0);
1.1       misho    1573:        if (srv->srv_server.cli_sock == -1) {
                   1574:                LOGERR;
1.14      misho    1575:                array_Destroy(&srv->srv_clients);
1.10      misho    1576:                schedEnd(&srv->srv_root);
1.12      misho    1577:                pthread_mutex_destroy(&srv->srv_funcs.mtx);
1.14      misho    1578:                e_free(srv);
1.1       misho    1579:                return NULL;
                   1580:        }
                   1581:        if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
                   1582:                LOGERR;
1.10      misho    1583:                goto err;
1.1       misho    1584:        }
1.27      misho    1585:        if (srv->srv_proto == SOCK_STREAM)
                   1586:                setsockopt(srv->srv_server.cli_sock, IPPROTO_TCP, TCP_NODELAY, &n, sizeof n);
1.5       misho    1587:        n = srv->srv_netbuf;
                   1588:        if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) {
                   1589:                LOGERR;
1.10      misho    1590:                goto err;
1.5       misho    1591:        }
                   1592:        if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) {
                   1593:                LOGERR;
1.10      misho    1594:                goto err;
1.5       misho    1595:        }
1.28.2.3  misho    1596:        if (bind(srv->srv_server.cli_sock, &srv->srv_server.cli_sa.sa, salen) == -1) {
1.1       misho    1597:                LOGERR;
1.10      misho    1598:                goto err;
1.13      misho    1599:        } else
                   1600:                fcntl(srv->srv_server.cli_sock, F_SETFL, 
                   1601:                                fcntl(srv->srv_server.cli_sock, F_GETFL) | O_NONBLOCK);
1.1       misho    1602: 
1.10      misho    1603:        rpc_register_srvPing(srv);
1.8       misho    1604: 
1.1       misho    1605:        return srv;
1.10      misho    1606: err:   /* error condition */
                   1607:        close(srv->srv_server.cli_sock);
1.14      misho    1608:        array_Destroy(&srv->srv_clients);
1.10      misho    1609:        schedEnd(&srv->srv_root);
1.12      misho    1610:        pthread_mutex_destroy(&srv->srv_funcs.mtx);
1.14      misho    1611:        e_free(srv);
1.10      misho    1612:        return NULL;
1.1       misho    1613: }
                   1614: 
                   1615: /*
1.7       misho    1616:  * rpc_srv_endServer() - Destroy RPC server, close all opened sockets and free resources
                   1617:  *
1.6       misho    1618:  * @psrv = RPC Server instance
1.1       misho    1619:  * return: none
                   1620:  */
1.16      misho    1621: void
1.6       misho    1622: rpc_srv_endServer(rpc_srv_t ** __restrict psrv)
1.1       misho    1623: {
1.10      misho    1624:        if (!psrv || !*psrv)
1.1       misho    1625:                return;
                   1626: 
1.10      misho    1627:        /* if send kill to blob server */
1.17      misho    1628:        rpc_srv_endBLOBServer(*psrv);
1.28      misho    1629:        /* wait for BLOB server done */
                   1630:        while (*(&(*psrv)->srv_blob.root))
                   1631:                usleep(1000);
1.1       misho    1632: 
1.10      misho    1633:        (*psrv)->srv_kill = 1;
                   1634:        sleep(RPC_SCHED_POLLING);
1.2       misho    1635: 
1.27      misho    1636:        if ((*psrv)->srv_server.cli_sa.sa.sa_family == AF_LOCAL)
                   1637:                unlink((*psrv)->srv_server.cli_sa.sun.sun_path);
                   1638: 
1.28      misho    1639:        schedEnd(&(*psrv)->srv_root);
                   1640: 
1.12      misho    1641:        pthread_mutex_destroy(&(*psrv)->srv_funcs.mtx);
1.14      misho    1642:        e_free(*psrv);
1.6       misho    1643:        *psrv = NULL;
1.1       misho    1644: }
                   1645: 
                   1646: /*
1.7       misho    1647:  * rpc_srv_loopServer() - Execute Main server loop and wait for clients requests
                   1648:  *
1.1       misho    1649:  * @srv = RPC Server instance
                   1650:  * return: -1 error or 0 ok, infinite loop ...
                   1651:  */
                   1652: int
1.5       misho    1653: rpc_srv_loopServer(rpc_srv_t * __restrict srv)
1.1       misho    1654: {
1.10      misho    1655:        rpc_cli_t *c;
1.1       misho    1656:        register int i;
1.12      misho    1657:        rpc_func_t *f;
1.10      misho    1658:        struct timespec ts = { RPC_SCHED_POLLING, 0 };
1.1       misho    1659: 
                   1660:        if (!srv) {
1.10      misho    1661:                rpc_SetErr(EINVAL, "Invalid parameter can`t start RPC server");
1.1       misho    1662:                return -1;
                   1663:        }
                   1664: 
1.13      misho    1665:        if (srv->srv_proto == SOCK_STREAM)
1.14      misho    1666:                if (listen(srv->srv_server.cli_sock, array_Size(srv->srv_clients)) == -1) {
1.13      misho    1667:                        LOGERR;
                   1668:                        return -1;
                   1669:                }
1.1       misho    1670: 
1.13      misho    1671:        if (!schedRead(srv->srv_root, cbProto[srv->srv_proto][CB_ACCEPTCLIENT], srv, 
                   1672:                                srv->srv_server.cli_sock, NULL, 0)) {
1.10      misho    1673:                rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                   1674:                return -1;
                   1675:        }
1.7       misho    1676: 
1.10      misho    1677:        schedPolling(srv->srv_root, &ts, NULL);
1.7       misho    1678:        /* main rpc loop */
1.10      misho    1679:        schedRun(srv->srv_root, &srv->srv_kill);
                   1680: 
                   1681:        /* close all clients connections & server socket */
1.14      misho    1682:        for (i = 0; i < array_Size(srv->srv_clients); i++) {
                   1683:                c = array(srv->srv_clients, i, rpc_cli_t*);
1.10      misho    1684:                if (c) {
1.24      misho    1685:                        if (srv->srv_proto == SOCK_STREAM) {
                   1686:                                shutdown(c->cli_sock, SHUT_RDWR);
                   1687:                                close(c->cli_sock);
                   1688:                        }
1.10      misho    1689: 
                   1690:                        schedCancelby(srv->srv_root, taskMAX, CRITERIA_ARG, c, NULL);
1.14      misho    1691:                        ait_freeVars(&RPC_RETVARS(c));
1.10      misho    1692:                        AIT_FREE_VAL(&c->cli_buf);
1.1       misho    1693:                }
1.14      misho    1694:                array_Del(srv->srv_clients, i, 42);
1.10      misho    1695:        }
1.14      misho    1696:        array_Destroy(&srv->srv_clients);
1.2       misho    1697: 
1.25      misho    1698:        if (srv->srv_proto != SOCK_EXT)
                   1699:                close(srv->srv_server.cli_sock);
1.2       misho    1700: 
1.10      misho    1701:        /* detach exported calls */
1.12      misho    1702:        RPC_FUNCS_LOCK(&srv->srv_funcs);
                   1703:        while ((f = SLIST_FIRST(&srv->srv_funcs))) {
                   1704:                SLIST_REMOVE_HEAD(&srv->srv_funcs, func_next);
1.1       misho    1705: 
1.10      misho    1706:                AIT_FREE_VAL(&f->func_name);
1.14      misho    1707:                e_free(f);
1.1       misho    1708:        }
1.12      misho    1709:        srv->srv_funcs.avlh_root = NULL;
                   1710:        RPC_FUNCS_UNLOCK(&srv->srv_funcs);
1.1       misho    1711: 
                   1712:        return 0;
                   1713: }
                   1714: 
                   1715: 
                   1716: /*
                   1717:  * rpc_srv_execCall() Execute registered call from RPC server
1.7       misho    1718:  *
1.10      misho    1719:  * @cli = RPC client
1.1       misho    1720:  * @rpc = IN RPC call structure
1.10      misho    1721:  * @funcname = Execute RPC function
1.5       misho    1722:  * @args = IN RPC calling arguments from RPC client
1.1       misho    1723:  * return: -1 error, !=-1 ok
                   1724:  */
                   1725: int
1.10      misho    1726: rpc_srv_execCall(rpc_cli_t * __restrict cli, struct tagRPCCall * __restrict rpc, 
                   1727:                ait_val_t funcname, array_t * __restrict args)
1.1       misho    1728: {
                   1729:        rpc_callback_t func;
                   1730: 
1.10      misho    1731:        if (!cli || !rpc || !AIT_ADDR(&funcname)) {
1.7       misho    1732:                rpc_SetErr(EINVAL, "Invalid parameter can`t exec function");
1.1       misho    1733:                return -1;
                   1734:        }
                   1735: 
1.10      misho    1736:        func = AIT_GET_LIKE(&funcname, rpc_callback_t);
                   1737:        return func(cli, rpc, args);
1.1       misho    1738: }
1.24      misho    1739: 
                   1740: 
                   1741: /*
                   1742:  * rpc_srv_initServer2() - Init & create layer2 RPC Server
                   1743:  *
                   1744:  * @InstID = Instance for authentication & recognition
                   1745:  * @concurentClients = Concurent clients at same time to this server
                   1746:  * @netBuf = Network buffer length (min:512 bytes), if =0 == BUFSIZ (also meaning max RPC packet)
                   1747:  * @csIface = Interface name for bind server, if NULL first interface on host
                   1748:  * return: NULL == error or !=NULL bind and created RPC server instance
                   1749:  */
                   1750: rpc_srv_t *
                   1751: rpc_srv_initServer2(u_char InstID, int concurentClients, int netBuf, const char *csIface)
                   1752: {
1.28.2.2  misho    1753: #ifndef __linux__
1.24      misho    1754:        int n = 1;
                   1755:        rpc_srv_t *srv = NULL;
                   1756:        sockaddr_t sa = E_SOCKADDR_INIT;
                   1757:        char szIface[64], szStr[STRSIZ];
                   1758:        register int i;
                   1759:        struct ifreq ifr;
                   1760:        struct bpf_insn insns[] = {
                   1761:                BPF_STMT(BPF_LD + BPF_H + BPF_ABS, 12),
                   1762:                BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, RPC_DEFPORT, 0, 1),
                   1763:                BPF_STMT(BPF_RET + BPF_K, -1),
                   1764:                BPF_STMT(BPF_RET + BPF_K, 0),
                   1765:        };
                   1766:        struct bpf_program fcode = { 
                   1767:                .bf_len = sizeof(insns) / sizeof(struct bpf_insn), 
                   1768:                        .bf_insns = insns
                   1769:        };
                   1770: 
                   1771:        if (!concurentClients) {
                   1772:                rpc_SetErr(EINVAL, "Invalid parameters can`t init RPC server");
                   1773:                return NULL;
                   1774:        }
                   1775:        if (!csIface) {
                   1776:                if (e_get1stiface(szIface, sizeof szIface))
                   1777:                        return NULL;
                   1778:        } else
                   1779:                strlcpy(szIface, csIface, sizeof szIface);
                   1780:        if (!e_getifacebyname(szIface, &sa))
                   1781:                return NULL;
                   1782: 
                   1783: #ifdef HAVE_SRANDOMDEV
                   1784:        srandomdev();
                   1785: #else
                   1786:        time_t tim;
                   1787: 
                   1788:        srandom((time(&tim) ^ getpid()));
                   1789: #endif
                   1790: 
                   1791:        srv = e_malloc(sizeof(rpc_srv_t));
                   1792:        if (!srv) {
                   1793:                LOGERR;
                   1794:                return NULL;
                   1795:        } else
                   1796:                memset(srv, 0, sizeof(rpc_srv_t));
                   1797: 
                   1798:        srv->srv_proto = SOCK_BPF;
                   1799:        srv->srv_netbuf = netBuf;
                   1800:        srv->srv_session.sess_version = RPC_VERSION;
                   1801:        srv->srv_session.sess_instance = InstID;
                   1802: 
                   1803:        srv->srv_server.cli_parent = srv;
                   1804:        memcpy(&srv->srv_server.cli_sa, &sa, sizeof srv->srv_server.cli_sa);
                   1805: 
                   1806:        /* init functions */
                   1807:        pthread_mutex_init(&srv->srv_funcs.mtx, NULL);
                   1808:        SLIST_INIT(&srv->srv_funcs);
                   1809:        AVL_INIT(&srv->srv_funcs);
                   1810: 
                   1811:        /* init scheduler */
                   1812:        srv->srv_root = schedBegin();
                   1813:        if (!srv->srv_root) {
                   1814:                rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                   1815:                pthread_mutex_destroy(&srv->srv_funcs.mtx);
                   1816:                e_free(srv);
                   1817:                return NULL;
                   1818:        }
                   1819: 
                   1820:        /* init pool for clients */
                   1821:        srv->srv_clients = array_Init(concurentClients);
                   1822:        if (!srv->srv_clients) {
                   1823:                rpc_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                   1824:                schedEnd(&srv->srv_root);
                   1825:                pthread_mutex_destroy(&srv->srv_funcs.mtx);
                   1826:                e_free(srv);
                   1827:                return NULL;
                   1828:        }
                   1829: 
                   1830:        /* create server handler */
                   1831:        for (i = 0; i < 10; i++) {
                   1832:                memset(szStr, 0, sizeof szStr);
                   1833:                snprintf(szStr, sizeof szStr, "/dev/bpf%d", i);
                   1834:                srv->srv_server.cli_sock = open(szStr, O_RDWR);
                   1835:                if (srv->srv_server.cli_sock > STDERR_FILENO)
                   1836:                        break;
                   1837:        }
                   1838:        if (srv->srv_server.cli_sock < 3) {
                   1839:                LOGERR;
                   1840:                array_Destroy(&srv->srv_clients);
                   1841:                schedEnd(&srv->srv_root);
                   1842:                pthread_mutex_destroy(&srv->srv_funcs.mtx);
                   1843:                e_free(srv);
                   1844:                return NULL;
                   1845:        }
                   1846: 
                   1847:        if (ioctl(srv->srv_server.cli_sock, BIOCIMMEDIATE, &n) == -1) {
                   1848:                LOGERR;
                   1849:                goto err;
                   1850:        }
                   1851:        if (ioctl(srv->srv_server.cli_sock, BIOCSETF, &fcode) == -1) {
                   1852:                LOGERR;
                   1853:                goto err;
                   1854:        }
                   1855:        n = (netBuf < RPC_MIN_BUFSIZ) ? getpagesize() : E_ALIGN(netBuf, 2);
                   1856:        if (ioctl(srv->srv_server.cli_sock, BIOCSBLEN, &n) == -1) {
                   1857:                LOGERR;
                   1858:                goto err;
                   1859:        } else
                   1860:                srv->srv_netbuf = n;
                   1861: 
                   1862:        memset(&ifr, 0, sizeof ifr);
                   1863:        strlcpy(ifr.ifr_name, szIface, sizeof ifr.ifr_name);
                   1864:        if (ioctl(srv->srv_server.cli_sock, BIOCSETIF, &ifr) == -1) {
                   1865:                LOGERR;
                   1866:                goto err;
                   1867:        } else
                   1868:                fcntl(srv->srv_server.cli_sock, F_SETFL, 
                   1869:                                fcntl(srv->srv_server.cli_sock, F_GETFL) | O_NONBLOCK);
                   1870: 
                   1871:        rpc_register_srvPing(srv);
                   1872: 
                   1873:        return srv;
                   1874: err:   /* error condition */
                   1875:        close(srv->srv_server.cli_sock);
                   1876:        array_Destroy(&srv->srv_clients);
                   1877:        schedEnd(&srv->srv_root);
                   1878:        pthread_mutex_destroy(&srv->srv_funcs.mtx);
                   1879:        e_free(srv);
1.28.2.2  misho    1880: #else
                   1881:        rpc_SetErr(ENOTSUP, "Feature isn't supported on Linux!");
                   1882: #endif
                   1883: 
1.24      misho    1884:        return NULL;
                   1885: }
1.25      misho    1886: 
                   1887: /*
                   1888:  * rpc_srv_initServerExt() - Init & create pipe RPC Server
                   1889:  *
                   1890:  * @InstID = Instance for authentication & recognition
                   1891:  * @netBuf = Network buffer length (min:512 bytes), if =0 == BUFSIZ (also meaning max RPC packet)
                   1892:  * @fd = File descriptor
                   1893:  * return: NULL == error or !=NULL bind and created RPC server instance
                   1894:  */
                   1895: rpc_srv_t *
                   1896: rpc_srv_initServerExt(u_char InstID, int netBuf, int fd)
                   1897: {
                   1898:        rpc_srv_t *srv = NULL;
                   1899: 
                   1900: #ifdef HAVE_SRANDOMDEV
                   1901:        srandomdev();
                   1902: #else
                   1903:        time_t tim;
                   1904: 
                   1905:        srandom((time(&tim) ^ getpid()));
                   1906: #endif
                   1907: 
                   1908:        srv = e_malloc(sizeof(rpc_srv_t));
                   1909:        if (!srv) {
                   1910:                LOGERR;
                   1911:                return NULL;
                   1912:        } else
                   1913:                memset(srv, 0, sizeof(rpc_srv_t));
                   1914: 
                   1915:        srv->srv_proto = SOCK_EXT;
                   1916:        srv->srv_netbuf = (netBuf < RPC_MIN_BUFSIZ) ? 
                   1917:                getpagesize() : E_ALIGN(netBuf, 2);
                   1918:        srv->srv_session.sess_version = RPC_VERSION;
                   1919:        srv->srv_session.sess_instance = InstID;
                   1920: 
                   1921:        srv->srv_server.cli_parent = srv;
                   1922:        srv->srv_server.cli_sock = fd;
                   1923: 
                   1924:        /* init functions */
                   1925:        pthread_mutex_init(&srv->srv_funcs.mtx, NULL);
                   1926:        SLIST_INIT(&srv->srv_funcs);
                   1927:        AVL_INIT(&srv->srv_funcs);
                   1928: 
                   1929:        /* init scheduler */
                   1930:        srv->srv_root = schedBegin();
                   1931:        if (!srv->srv_root) {
                   1932:                rpc_SetErr(sched_GetErrno(), "%s", sched_GetError());
                   1933:                pthread_mutex_destroy(&srv->srv_funcs.mtx);
                   1934:                e_free(srv);
                   1935:                return NULL;
                   1936:        }
                   1937: 
                   1938:        /* init pool for clients */
                   1939:        srv->srv_clients = array_Init(1);
                   1940:        if (!srv->srv_clients) {
                   1941:                rpc_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                   1942:                schedEnd(&srv->srv_root);
                   1943:                pthread_mutex_destroy(&srv->srv_funcs.mtx);
                   1944:                e_free(srv);
                   1945:                return NULL;
                   1946:        }
                   1947: 
                   1948:        fcntl(srv->srv_server.cli_sock, F_SETFL, 
                   1949:                        fcntl(srv->srv_server.cli_sock, F_GETFL) | O_NONBLOCK);
                   1950: 
                   1951:        rpc_register_srvPing(srv);
                   1952: 
                   1953:        return srv;
                   1954: }

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