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

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.5.2.6 ! misho       6: * $Id: srv.c,v 1.5.2.5 2011/09/07 10:46:54 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: 
                     15: Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011
                     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: 
                     49: static void *
                     50: rpc_srv_dispatchCall(void *arg)
                     51: {
                     52:        rpc_cli_t *c = arg;
                     53:        rpc_srv_t *s;
1.3       misho      54:        rpc_func_t *f = NULL;
1.5       misho      55:        array_t *arr;
1.1       misho      56:        struct tagRPCCall *rpc;
1.3       misho      57:        struct tagRPCRet *rrpc;
                     58:        rpc_sess_t ses = { 0 };
1.1       misho      59:        fd_set fds;
1.5       misho      60:        u_char *buf;
1.2       misho      61:        int ret, argc = 0, Limit = 0;
1.1       misho      62:        register int i;
1.5.2.3   misho      63:        uint16_t tag = 0;
                     64:        uint32_t hash = 0;
1.1       misho      65: 
                     66:        if (!arg) {
                     67:                rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t procced RPC client ...\n");
                     68:                return NULL;
                     69:        } else
                     70:                s = c->cli_parent;
                     71: 
1.5       misho      72:        buf = malloc(s->srv_netbuf);
                     73:        if (!buf) {
                     74:                LOGERR;
                     75:                return NULL;
                     76:        }
                     77: 
1.1       misho      78:        do {
                     79:                FD_ZERO(&fds);
                     80:                FD_SET(c->cli_sock, &fds);
                     81:                ret = select(c->cli_sock + 1, &fds, NULL, NULL, NULL);
                     82:                if (ret == -1) {
1.4       misho      83:                        if (errno == EINTR && s->srv_kill != kill)
                     84:                                continue;
                     85: 
                     86:                        LOGERR;
1.1       misho      87:                        ret = -2;
1.4       misho      88:                        break;
1.1       misho      89:                }
1.5       misho      90:                memset(buf, 0, s->srv_netbuf);
                     91:                ret = recv(c->cli_sock, buf, s->srv_netbuf, 0);
1.4       misho      92:                if (ret == -1) {
1.1       misho      93:                        LOGERR;
                     94:                        ret = -3;
                     95:                        break;
                     96:                }
1.4       misho      97:                if (!ret) {             /* receive EOF */
1.1       misho      98:                        ret = 0;
                     99:                        break;
                    100:                }
                    101:                if (ret < sizeof(struct tagRPCCall)) {
1.5       misho     102:                        rpc_SetErr(ERPCMISMATCH, "Error:: too short RPC packet ...\n");
1.1       misho     103:                        ret = -4;
                    104:                        break;
                    105:                } else
                    106:                        rpc = (struct tagRPCCall*) buf;
1.4       misho     107:                /* check RPC packet session info */
1.5.2.1   misho     108:                if (rpc_chkPktSession(&rpc->call_session, &s->srv_session)) {
1.5       misho     109:                        rpc_SetErr(ERPCMISMATCH, "Error:: get invalid RPC session ...\n");
1.1       misho     110:                        ret = -5;
1.2       misho     111:                        goto makeReply;
                    112:                } else
                    113:                        Limit = sizeof(struct tagRPCCall);
1.5       misho     114: 
1.5.2.3   misho     115:                tag = rpc->call_tag;
                    116:                hash = rpc->call_hash;
                    117: 
1.4       misho     118:                /* RPC is OK! Go decapsulate variables ... */
1.5.2.2   misho     119:                if (ntohs(rpc->call_argc)) {
                    120:                        arr = io_buffer2vals(buf + Limit, s->srv_netbuf - Limit, 
                    121:                                        ntohs(rpc->call_argc), 1);
1.5       misho     122:                        if (!arr) {
1.2       misho     123:                                ret = -5;
                    124:                                goto makeReply;
1.1       misho     125:                        }
1.5       misho     126:                } else
                    127:                        arr = NULL;
1.1       misho     128: 
1.4       misho     129:                /* execute call */
1.1       misho     130:                argc = 0;
1.3       misho     131:                memcpy(&ses, &rpc->call_session, sizeof ses);
1.5.2.3   misho     132:                if (!(f = rpc_srv_getCall(s, ntohs(tag), ntohl(hash)))) {
1.5       misho     133:                        rpc_SetErr(EPROGUNAVAIL, "Error:: call not found into RPC server ...\n");
1.1       misho     134:                        ret = -6;
                    135:                } else
1.5       misho     136:                        if ((ret = rpc_srv_execCall(f, rpc, arr)) == -1)
1.2       misho     137:                                ret = -9;
1.5       misho     138:                        else {
                    139:                                if (arr)
                    140:                                        io_arrayDestroy(&arr);
                    141:                                argc = rpc_srv_getVars(f, &arr);
                    142:                                goto makeReply;         /* Call finish OK */
                    143:                        }
                    144: 
                    145:                if (arr)
                    146:                        io_arrayDestroy(&arr);
                    147: 
1.3       misho     148: makeReply:
1.5       misho     149:                /* Made reply */
                    150:                memset(buf, 0, s->srv_netbuf);
1.3       misho     151:                rrpc = (struct tagRPCRet*) buf;
                    152:                Limit = sizeof(struct tagRPCRet);
1.1       misho     153: 
1.3       misho     154:                memcpy(&rrpc->ret_session, &ses, sizeof(rpc_sess_t));
1.5.2.3   misho     155:                rrpc->ret_tag = tag;
                    156:                rrpc->ret_hash = hash;
1.5.2.2   misho     157:                rrpc->ret_errno = htonl(rpc_Errno);
                    158:                rrpc->ret_retcode = htonl(ret);
                    159:                rrpc->ret_argc = htons(argc);
1.1       misho     160: 
1.5       misho     161:                if (argc && arr) {
                    162:                        /* Go Encapsulate variables ... */
                    163:                        if ((i = io_vals2buffer(buf + Limit, s->srv_netbuf - Limit, arr)) == -1) {
                    164:                                rpc_srv_freeVals(f);
1.3       misho     165:                                argc = 0;
                    166:                                ret = -7;
1.5       misho     167:                                rpc_SetErr(EBADRPC, "Error:: in prepare RPC packet values (-7) ...\n");
1.3       misho     168:                                goto makeReply;
1.5       misho     169:                        } else {
                    170:                                Limit += i;
1.1       misho     171: 
1.5       misho     172:                                rpc_srv_freeVals(f);
1.1       misho     173:                        }
                    174:                }
                    175: 
1.4       misho     176:                ret = send(c->cli_sock, buf, Limit, 0);
                    177:                if (ret == -1) {
1.1       misho     178:                        LOGERR;
                    179:                        ret = -8;
                    180:                        break;
                    181:                }
                    182:                if (ret != Limit) {
1.5       misho     183:                        rpc_SetErr(EPROCUNAVAIL, "Error:: in send RPC request, should be send %d bytes, "
1.1       misho     184:                                        "really is %d\n", Limit, ret);
                    185:                        ret = -9;
                    186:                        break;
                    187:                }
1.4       misho     188:        } while (ret > -1 || s->srv_kill != kill);
1.1       misho     189: 
                    190:        shutdown(c->cli_sock, SHUT_RDWR);
                    191:        close(c->cli_sock);
                    192:        memset(c, 0, sizeof(rpc_cli_t));
1.5       misho     193:        free(buf);
1.2       misho     194:        return (void*) (long)ret;
                    195: }
                    196: 
                    197: 
                    198: static void *
                    199: rpc_srv_dispatchVars(void *arg)
                    200: {
                    201:        rpc_cli_t *c = arg;
                    202:        rpc_srv_t *s;
                    203:        rpc_blob_t *b;
1.4       misho     204:        int ret = 0;
1.2       misho     205:        fd_set fds;
                    206:        u_char buf[sizeof(struct tagBLOBHdr)];
                    207:        struct tagBLOBHdr *blob;
                    208: 
                    209:        if (!arg) {
                    210:                rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t procced BLOB client ...\n");
                    211:                return NULL;
                    212:        } else
                    213:                s = c->cli_parent;
                    214: 
                    215:        do {
1.4       misho     216:                /* check for disable service at this moment? */
                    217:                if (s->srv_blob.state == disable && s->srv_kill != kill) {
                    218:                        usleep(100000);
                    219: #ifdef HAVE_PTHREAD_YIELD
                    220:                        pthread_yield();
                    221: #endif
                    222:                        continue;
1.2       misho     223:                }
                    224: 
                    225:                FD_ZERO(&fds);
                    226:                FD_SET(c->cli_sock, &fds);
                    227:                ret = select(c->cli_sock + 1, &fds, NULL, NULL, NULL);
                    228:                if (ret == -1) {
1.4       misho     229:                        if (errno == EINTR && s->srv_kill != kill && s->srv_blob.state != kill)
                    230:                                continue;
                    231: 
                    232:                        LOGERR;
1.2       misho     233:                        ret = -2;
1.4       misho     234:                        break;
1.2       misho     235:                }
                    236: 
                    237:                memset(buf, 0, sizeof buf);
1.4       misho     238:                ret = recv(c->cli_sock, buf, sizeof buf, 0);
                    239:                if (ret == -1) {
1.2       misho     240:                        LOGERR;
                    241:                        ret = -3;
                    242:                        break;
                    243:                }
1.4       misho     244:                /* receive EOF, disable or kill service */
                    245:                if (!ret || s->srv_blob.state == kill || s->srv_kill == kill) {
1.2       misho     246:                        ret = 0;
                    247:                        break;
                    248:                }
                    249:                if (ret < sizeof(struct tagBLOBHdr)) {
1.5       misho     250:                        rpc_SetErr(ERPCMISMATCH, "Error:: too short BLOB packet ...\n");
1.2       misho     251:                        ret = -4;
                    252:                        break;
                    253:                } else
                    254:                        blob = (struct tagBLOBHdr*) buf;
1.4       misho     255:                /* check BLOB packet session info */
1.2       misho     256:                if (memcmp(&blob->hdr_session, &s->srv_session, sizeof blob->hdr_session)) {
                    257:                        rpc_SetErr(EINVAL, "Error:: get invalid BLOB session ...\n");
                    258:                        ret = -5;
                    259:                        goto makeReply;
                    260:                }
1.4       misho     261:                /* Go to proceed packet ... */
1.2       misho     262:                switch (blob->hdr_cmd) {
                    263:                        case get:
                    264:                                if (!(b = rpc_srv_getBLOB(s, blob->hdr_var))) {
                    265:                                        rpc_SetErr(EINVAL, "Error:: var (%x) not found into BLOB server ...\n", 
                    266:                                                        blob->hdr_var);
                    267:                                        ret = -6;
                    268:                                        break;
                    269:                                } else
                    270:                                        blob->hdr_len = b->blob_len;
                    271: 
                    272:                                if (rpc_srv_blobMap(s, b) != -1) {
                    273:                                        ret = rpc_srv_sendBLOB(c, b);
                    274:                                        rpc_srv_blobUnmap(b);
                    275:                                } else
                    276:                                        ret = -7;
                    277:                                break;
                    278:                        case set:
                    279:                                if ((b = rpc_srv_registerBLOB(s, blob->hdr_len))) {
1.4       misho     280:                                        /* set new BLOB variable for reply :) */
1.2       misho     281:                                        blob->hdr_var = b->blob_var;
                    282: 
                    283:                                        ret = rpc_srv_recvBLOB(c, b);
                    284:                                        rpc_srv_blobUnmap(b);
                    285:                                } else
                    286:                                        ret = -7;
                    287:                                break;
                    288:                        case unset:
                    289:                                ret = rpc_srv_unregisterBLOB(s, blob->hdr_var);
                    290:                                if (ret == -1)
                    291:                                        ret = -7;
                    292:                                break;
                    293:                        default:
1.5       misho     294:                                rpc_SetErr(EPROCUNAVAIL, "Error:: unsupported BLOB command (%d)...\n", 
1.2       misho     295:                                                blob->hdr_cmd);
                    296:                                ret = -7;
                    297:                }
                    298: 
                    299: makeReply:
1.4       misho     300:                /* Replay to client! */
1.2       misho     301:                blob->hdr_cmd = ret < 0 ? error : ok;
                    302:                blob->hdr_ret = ret;
1.4       misho     303:                ret = send(c->cli_sock, buf, sizeof buf, 0);
                    304:                if (ret == -1) {
1.2       misho     305:                        LOGERR;
                    306:                        ret = -8;
                    307:                        break;
                    308:                }
                    309:                if (ret != sizeof buf) {
1.5       misho     310:                        rpc_SetErr(EPROCUNAVAIL, "Error:: in send BLOB reply, should be send %d bytes, "
1.2       misho     311:                                        "really is %d\n", sizeof buf, ret);
                    312:                        ret = -9;
                    313:                        break;
                    314:                }
1.4       misho     315:        } while (ret > -1 || s->srv_kill != kill);
1.2       misho     316: 
                    317:        shutdown(c->cli_sock, SHUT_RDWR);
                    318:        close(c->cli_sock);
                    319:        memset(c, 0, sizeof(rpc_cli_t));
1.5       misho     320:        return (void*) ((long)ret);
1.1       misho     321: }
                    322: 
                    323: // -------------------------------------------------
                    324: 
                    325: /*
1.2       misho     326:  * rpc_srv_initBLOBServer() Init & create BLOB Server
1.4       misho     327:  * @srv = RPC server instance
1.2       misho     328:  * @Port = Port for bind server, if Port == 0 default port is selected
                    329:  * @diskDir = Disk place for BLOB file objects
                    330:  * return: -1 == error or 0 bind and created BLOB server instance
                    331:  */
                    332: int
                    333: rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_short Port, const char *diskDir)
                    334: {
                    335:        int n = 1;
1.5.2.6 ! misho     336:        io_sockaddr_t sa;
1.2       misho     337: 
                    338:        if (!srv) {
                    339:                rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t init BLOB server ...\n");
                    340:                return -1;
                    341:        }
                    342:        if (srv->srv_blob.state) {
                    343:                rpc_SetErr(EPERM, "Warning:: Already started BLOB server!\n");
                    344:                return 0;
                    345:        }
                    346:        if (!Port)
                    347:                Port = RPC_DEFPORT + 1;
                    348: 
                    349:        memset(&srv->srv_blob, 0, sizeof srv->srv_blob);
                    350:        if (access(diskDir, R_OK | W_OK) == -1) {
                    351:                LOGERR;
                    352:                return -1;
                    353:        } else
                    354:                strlcpy(srv->srv_blob.dir, diskDir, UCHAR_MAX + 1);
                    355: 
                    356:        srv->srv_blob.server.cli_tid = pthread_self();
                    357:        srv->srv_blob.server.cli_parent = srv;
1.4       misho     358: 
                    359:        memcpy(&sa, &srv->srv_server.cli_sa, sizeof sa);
1.5.2.6 ! misho     360:        switch (srv->srv_server.cli_sa.sa.sa_family) {
1.4       misho     361:                case AF_INET:
1.5.2.6 ! misho     362:                        sa.sin.sin_port = htons(Port);
        !           363:                        memcpy(&srv->srv_blob.server.cli_sa, &sa, sizeof sa);
1.4       misho     364:                        break;
                    365:                case AF_INET6:
1.5.2.6 ! misho     366:                        sa.sin6.sin6_port = htons(Port);
        !           367:                        memcpy(&srv->srv_blob.server.cli_sa, &sa, sizeof sa);
1.4       misho     368:                        break;
                    369:                case AF_LOCAL:
1.5.2.6 ! misho     370:                        strlcat(sa.sun.sun_path, ".blob", sizeof sa.sun.sun_path);
        !           371:                        memcpy(&srv->srv_blob.server.cli_sa, &sa, sizeof sa);
1.4       misho     372:                        break;
                    373:                default:
                    374:                        return -1;
1.2       misho     375:        }
                    376: 
1.4       misho     377:        /* create BLOB server socket */
1.5.2.6 ! misho     378:        srv->srv_blob.server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, SOCK_STREAM, 0);
1.2       misho     379:        if (srv->srv_blob.server.cli_sock == -1) {
                    380:                LOGERR;
                    381:                return -1;
                    382:        }
                    383:        if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
                    384:                LOGERR;
                    385:                close(srv->srv_blob.server.cli_sock);
                    386:                return -1;
                    387:        }
1.5       misho     388:        n = srv->srv_netbuf;
                    389:        if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) {
                    390:                LOGERR;
                    391:                close(srv->srv_blob.server.cli_sock);
                    392:                return -1;
                    393:        }
                    394:        if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) {
                    395:                LOGERR;
                    396:                close(srv->srv_blob.server.cli_sock);
                    397:                return -1;
                    398:        }
1.5.2.6 ! misho     399:        if (bind(srv->srv_blob.server.cli_sock, &srv->srv_blob.server.cli_sa.sa, 
1.2       misho     400:                                sizeof srv->srv_blob.server.cli_sa) == -1) {
                    401:                LOGERR;
                    402:                close(srv->srv_blob.server.cli_sock);
                    403:                return -1;
                    404:        }
                    405: 
1.4       misho     406:        /* allocate pool for concurent clients */
1.2       misho     407:        srv->srv_blob.clients = calloc(srv->srv_numcli, sizeof(rpc_cli_t));
                    408:        if (!srv->srv_blob.clients) {
                    409:                LOGERR;
                    410:                close(srv->srv_blob.server.cli_sock);
                    411:                return -1;
                    412:        } else
                    413:                memset(srv->srv_blob.clients, 0, srv->srv_numcli * sizeof(rpc_cli_t));
                    414: 
                    415:        pthread_mutex_init(&srv->srv_blob.mtx, NULL);
                    416: 
                    417:        rpc_srv_registerCall(srv, NULL, CALL_BLOBSHUTDOWN, 0);
1.5       misho     418:        rpc_srv_registerCall(srv, NULL, CALL_BLOBCLIENTS, 1);
                    419:        rpc_srv_registerCall(srv, NULL, CALL_BLOBVARS, 1);
                    420:        rpc_srv_registerCall(srv, NULL, CALL_BLOBSTATE, 0);
1.2       misho     421: 
1.4       misho     422:        srv->srv_blob.state = enable;   /* enable BLOB */
1.2       misho     423:        return 0;
                    424: }
                    425: 
                    426: /*
                    427:  * rpc_srv_endBLOBServer() Destroy BLOB server, close all opened sockets and free resources
                    428:  * @srv = RPC Server instance
                    429:  * return: none
                    430:  */
                    431: void
                    432: rpc_srv_endBLOBServer(rpc_srv_t * __restrict srv)
                    433: {
                    434:        rpc_cli_t *c;
                    435:        register int i;
                    436:        rpc_blob_t *f;
                    437: 
                    438:        if (!srv) {
                    439:                rpc_SetErr(EINVAL, "Error:: Can`t destroy server because parameter is null!\n");
                    440:                return;
                    441:        } else
1.4       misho     442:                srv->srv_blob.state = kill;
1.2       misho     443: 
                    444:        rpc_srv_unregisterCall(srv, NULL, CALL_BLOBSHUTDOWN);
                    445:        rpc_srv_unregisterCall(srv, NULL, CALL_BLOBCLIENTS);
                    446:        rpc_srv_unregisterCall(srv, NULL, CALL_BLOBVARS);
                    447:        rpc_srv_unregisterCall(srv, NULL, CALL_BLOBSTATE);
                    448: 
1.4       misho     449:        /* close all clients connections & server socket */
1.2       misho     450:        for (i = 0, c = srv->srv_blob.clients; i < srv->srv_numcli && c; i++, c++)
1.5.2.6 ! misho     451:                if (c->cli_sa.sa.sa_family)
1.2       misho     452:                        shutdown(c->cli_sock, SHUT_RDWR);
                    453:        close(srv->srv_blob.server.cli_sock);
                    454: 
1.5       misho     455:        pthread_mutex_lock(&srv->srv_blob.mtx);
1.2       misho     456:        if (srv->srv_blob.clients) {
                    457:                free(srv->srv_blob.clients);
                    458:                srv->srv_blob.clients = NULL;
                    459:        }
                    460: 
1.4       misho     461:        /* detach blobs */
1.2       misho     462:        while ((f = srv->srv_blob.blobs)) {
                    463:                srv->srv_blob.blobs = f->blob_next;
                    464:                rpc_srv_blobFree(srv, f);
                    465:                free(f);
                    466:        }
                    467:        pthread_mutex_unlock(&srv->srv_blob.mtx);
                    468: 
                    469:        while (pthread_mutex_trylock(&srv->srv_blob.mtx) == EBUSY);
                    470:        pthread_mutex_destroy(&srv->srv_blob.mtx);
                    471: }
                    472: 
                    473: /*
1.5       misho     474:  * rpc_srv_loopBLOB() Execute Main BLOB server loop and wait for clients requests
1.2       misho     475:  * @srv = RPC Server instance
                    476:  * return: -1 error or 0 ok, infinite loop ...
                    477:  */
                    478: int
1.5       misho     479: rpc_srv_loopBLOB(rpc_srv_t * __restrict srv)
1.2       misho     480: {
1.5.2.6 ! misho     481:        socklen_t salen = sizeof(io_sockaddr_t);
1.2       misho     482:        register int i;
                    483:        rpc_cli_t *c;
                    484:        fd_set fds;
                    485:        int ret;
                    486:        struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
                    487: 
1.4       misho     488:        if (!srv || srv->srv_blob.state == kill) {
1.2       misho     489:                rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t start BLOB server ...\n");
                    490:                return -1;
                    491:        }
                    492: 
                    493:        if (listen(srv->srv_blob.server.cli_sock, SOMAXCONN) == -1) {
                    494:                LOGERR;
                    495:                return -1;
                    496:        }
                    497: 
1.4       misho     498:        while (srv->srv_blob.state != kill && srv->srv_kill != kill) {
1.2       misho     499:                for (c = srv->srv_blob.clients, i = 0; i < srv->srv_numcli && c; i++, c++)
1.5.2.6 ! misho     500:                        if (!c->cli_sa.sa.sa_family)
1.2       misho     501:                                break;
                    502:                if (i >= srv->srv_numcli) {
1.5       misho     503: #ifdef HAVE_PTHREAD_YIELD
                    504:                        pthread_yield();
                    505: #else
1.2       misho     506:                        usleep(1000000);
1.5       misho     507: #endif
1.2       misho     508:                        continue;
                    509:                }
                    510: 
                    511:                FD_ZERO(&fds);
                    512:                FD_SET(srv->srv_blob.server.cli_sock, &fds);
                    513:                ret = select(srv->srv_blob.server.cli_sock + 1, &fds, NULL, NULL, &tv);
                    514:                if (ret == -1) {
                    515:                        LOGERR;
                    516:                        ret = 1;
                    517:                        break;
                    518:                }
                    519:                if (!ret)
                    520:                        continue;
                    521: 
1.5.2.6 ! misho     522:                c->cli_sock = accept(srv->srv_blob.server.cli_sock, &c->cli_sa.sa, &salen);
1.2       misho     523:                if (c->cli_sock == -1) {
                    524:                        LOGERR;
                    525:                        continue;
                    526:                } else
                    527:                        c->cli_parent = srv;
                    528: 
                    529:                if (pthread_create(&c->cli_tid, NULL, rpc_srv_dispatchVars, c)) {
                    530:                        LOGERR;
                    531:                        continue;
1.4       misho     532:                } else
                    533:                        pthread_detach(c->cli_tid);
1.2       misho     534:        }
                    535: 
1.5       misho     536:        srv->srv_blob.state = kill;
1.2       misho     537: 
                    538:        return 0;
                    539: }
                    540: 
                    541: 
                    542: /*
1.1       misho     543:  * rpc_srv_initServer() Init & create RPC Server
                    544:  * @regProgID = ProgramID for authentication & recognition
                    545:  * @regProcID = ProcessID for authentication & recognition
                    546:  * @concurentClients = Concurent clients at same time to this server
1.5       misho     547:  * @netBuf = Network buffer length, if =0 == BUFSIZ (also meaning max RPC packet)
1.4       misho     548:  * @family = Family type, AF_INET, AF_INET6 or AF_LOCAL
                    549:  * @csHost = Host name or address for bind server, if NULL any address
1.1       misho     550:  * @Port = Port for bind server, if Port == 0 default port is selected
                    551:  * return: NULL == error or !=NULL bind and created RPC server instance
                    552:  */
                    553: rpc_srv_t *
                    554: rpc_srv_initServer(u_int regProgID, u_int regProcID, int concurentClients, 
1.5       misho     555:                int netBuf, u_short family, const char *csHost, u_short Port)
1.1       misho     556: {
                    557:        rpc_srv_t *srv = NULL;
                    558:        int n = 1;
                    559:        struct hostent *host = NULL;
1.5.2.6 ! misho     560:        io_sockaddr_t sa;
1.1       misho     561: 
1.4       misho     562:        if (!concurentClients || !regProgID || 
                    563:                        (family != AF_INET && family != AF_INET6 && family != AF_LOCAL)) {
1.1       misho     564:                rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t init RPC server ...\n");
                    565:                return NULL;
                    566:        }
                    567:        if (!Port)
                    568:                Port = RPC_DEFPORT;
1.5       misho     569:        if (!netBuf)
                    570:                netBuf = BUFSIZ;
1.4       misho     571:        if (csHost && family != AF_LOCAL) {
1.1       misho     572:                host = gethostbyname2(csHost, family);
                    573:                if (!host) {
                    574:                        rpc_SetErr(h_errno, "Error:: %s\n", hstrerror(h_errno));
                    575:                        return NULL;
                    576:                }
                    577:        }
1.4       misho     578:        memset(&sa, 0, sizeof sa);
1.5.2.6 ! misho     579:        sa.sa.sa_family = family;
1.1       misho     580:        switch (family) {
                    581:                case AF_INET:
1.5.2.6 ! misho     582:                        sa.sin.sin_len = sizeof(struct sockaddr_in);
        !           583:                        sa.sin.sin_port = htons(Port);
1.1       misho     584:                        if (csHost)
1.5.2.6 ! misho     585:                                memcpy(&sa.sin.sin_addr, host->h_addr, host->h_length);
1.1       misho     586:                        break;
                    587:                case AF_INET6:
1.5.2.6 ! misho     588:                        sa.sin6.sin6_len = sizeof(struct sockaddr_in6);
        !           589:                        sa.sin6.sin6_port = htons(Port);
1.4       misho     590:                        if (csHost)
1.5.2.6 ! misho     591:                                memcpy(&sa.sin6.sin6_addr, host->h_addr, host->h_length);
1.4       misho     592:                        break;
                    593:                case AF_LOCAL:
1.5.2.6 ! misho     594:                        sa.sun.sun_len = sizeof(struct sockaddr_un);
1.1       misho     595:                        if (csHost)
1.5.2.6 ! misho     596:                                strlcpy(sa.sun.sun_path, csHost, sizeof sa.sun.sun_path);
        !           597:                        unlink(sa.sun.sun_path);
1.1       misho     598:                        break;
                    599:                default:
                    600:                        rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t start RPC server ...\n");
                    601:                        return NULL;
                    602:        }
                    603: 
                    604:        srv = malloc(sizeof(rpc_srv_t));
                    605:        if (!srv) {
                    606:                LOGERR;
                    607:                return NULL;
                    608:        } else
                    609:                memset(srv, 0, sizeof(rpc_srv_t));
                    610: 
1.5       misho     611:        srv->srv_netbuf = netBuf;
1.1       misho     612:        srv->srv_numcli = concurentClients;
                    613:        srv->srv_session.sess_version = RPC_VERSION;
                    614:        srv->srv_session.sess_program = regProgID;
                    615:        srv->srv_session.sess_process = regProcID;
                    616: 
1.2       misho     617:        srv->srv_server.cli_tid = pthread_self();
1.1       misho     618:        srv->srv_server.cli_parent = srv;
1.5.2.6 ! misho     619:        memcpy(&srv->srv_server.cli_sa, &sa, sizeof sa);
1.4       misho     620: 
                    621:        /* create server socket */
1.1       misho     622:        srv->srv_server.cli_sock = socket(family, SOCK_STREAM, 0);
                    623:        if (srv->srv_server.cli_sock == -1) {
                    624:                LOGERR;
                    625:                free(srv);
                    626:                return NULL;
                    627:        }
                    628:        if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
                    629:                LOGERR;
                    630:                close(srv->srv_server.cli_sock);
                    631:                free(srv);
                    632:                return NULL;
                    633:        }
1.5       misho     634:        n = srv->srv_netbuf;
                    635:        if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) {
                    636:                LOGERR;
                    637:                close(srv->srv_server.cli_sock);
                    638:                free(srv);
                    639:                return NULL;
                    640:        }
                    641:        if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) {
                    642:                LOGERR;
                    643:                close(srv->srv_server.cli_sock);
                    644:                free(srv);
                    645:                return NULL;
                    646:        }
1.5.2.6 ! misho     647:        if (bind(srv->srv_server.cli_sock, &srv->srv_server.cli_sa.sa, sizeof srv->srv_server.cli_sa) == -1) {
1.1       misho     648:                LOGERR;
                    649:                close(srv->srv_server.cli_sock);
                    650:                free(srv);
                    651:                return NULL;
                    652:        }
                    653: 
1.4       misho     654:        /* allocate pool for concurent clients */
1.1       misho     655:        srv->srv_clients = calloc(srv->srv_numcli, sizeof(rpc_cli_t));
                    656:        if (!srv->srv_clients) {
                    657:                LOGERR;
                    658:                close(srv->srv_server.cli_sock);
                    659:                free(srv);
                    660:                return NULL;
                    661:        } else
                    662:                memset(srv->srv_clients, 0, srv->srv_numcli * sizeof(rpc_cli_t));
                    663: 
1.2       misho     664:        pthread_mutex_init(&srv->srv_mtx, NULL);
                    665: 
                    666:        rpc_srv_registerCall(srv, NULL, CALL_SRVSHUTDOWN, 0);
1.5       misho     667:        rpc_srv_registerCall(srv, NULL, CALL_SRVCLIENTS, 1);
                    668:        rpc_srv_registerCall(srv, NULL, CALL_SRVSESSIONS, 4);
                    669:        rpc_srv_registerCall(srv, NULL, CALL_SRVCALLS, 1);
1.1       misho     670:        return srv;
                    671: }
                    672: 
                    673: /*
                    674:  * rpc_srv_endServer() Destroy RPC server, close all opened sockets and free resources
1.5.2.4   misho     675:  * @psrv = RPC Server instance
1.1       misho     676:  * return: none
                    677:  */
                    678: void
1.5.2.4   misho     679: rpc_srv_endServer(rpc_srv_t ** __restrict psrv)
1.1       misho     680: {
                    681:        rpc_cli_t *c;
                    682:        register int i;
                    683:        rpc_func_t *f;
                    684: 
1.5.2.4   misho     685:        if (!psrv || !*psrv) {
1.1       misho     686:                rpc_SetErr(EINVAL, "Error:: Can`t destroy server because parameter is null!\n");
                    687:                return;
                    688:        }
                    689: 
1.5.2.4   misho     690:        rpc_srv_endBLOBServer(*psrv);
1.1       misho     691: 
1.4       misho     692:        /* close all clients connections & server socket */
1.5.2.4   misho     693:        for (i = 0, c = (*psrv)->srv_clients; i < (*psrv)->srv_numcli && c; i++, c++)
1.5.2.6 ! misho     694:                if (c->cli_sa.sa.sa_family) {
1.1       misho     695:                        shutdown(c->cli_sock, SHUT_RDWR);
1.2       misho     696:                        close(c->cli_sock);
                    697:                }
1.5.2.4   misho     698:        close((*psrv)->srv_server.cli_sock);
1.1       misho     699: 
1.5.2.4   misho     700:        if ((*psrv)->srv_clients) {
                    701:                free((*psrv)->srv_clients);
                    702:                (*psrv)->srv_clients = NULL;
                    703:                (*psrv)->srv_numcli = 0;
1.1       misho     704:        }
                    705: 
1.4       misho     706:        /* detach exported calls */
1.5.2.4   misho     707:        pthread_mutex_lock(&(*psrv)->srv_mtx);
                    708:        while ((f = (*psrv)->srv_funcs)) {
                    709:                (*psrv)->srv_funcs = f->func_next;
1.5.2.5   misho     710:                rpc_srv_destroyVars(f);
1.2       misho     711:                free(f);
                    712:        }
1.5.2.4   misho     713:        pthread_mutex_unlock(&(*psrv)->srv_mtx);
1.2       misho     714: 
1.5.2.4   misho     715:        while (pthread_mutex_trylock(&(*psrv)->srv_mtx) == EBUSY);
                    716:        pthread_mutex_destroy(&(*psrv)->srv_mtx);
1.2       misho     717: 
1.5.2.4   misho     718:        free(*psrv);
                    719:        *psrv = NULL;
1.1       misho     720: }
                    721: 
                    722: /*
1.5       misho     723:  * rpc_srv_loopServer() Execute Main server loop and wait for clients requests
1.1       misho     724:  * @srv = RPC Server instance
                    725:  * return: -1 error or 0 ok, infinite loop ...
                    726:  */
                    727: int
1.5       misho     728: rpc_srv_loopServer(rpc_srv_t * __restrict srv)
1.1       misho     729: {
1.5.2.6 ! misho     730:        socklen_t salen = sizeof(io_sockaddr_t);
1.1       misho     731:        register int i;
                    732:        rpc_cli_t *c;
1.2       misho     733:        fd_set fds;
                    734:        int ret;
                    735:        struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
1.1       misho     736: 
                    737:        if (!srv) {
                    738:                rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t start RPC server ...\n");
                    739:                return -1;
                    740:        }
                    741: 
1.5       misho     742:        /* activate BLOB server worker if srv->srv_blob.state == enable */
                    743:        rpc_srv_execBLOBServer(srv);
                    744: 
1.1       misho     745:        if (listen(srv->srv_server.cli_sock, SOMAXCONN) == -1) {
                    746:                LOGERR;
                    747:                return -1;
                    748:        }
                    749: 
1.4       misho     750:        while (srv->srv_kill != kill) {
1.1       misho     751:                for (c = srv->srv_clients, i = 0; i < srv->srv_numcli && c; i++, c++)
1.5.2.6 ! misho     752:                        if (!c->cli_sa.sa.sa_family)
1.1       misho     753:                                break;
1.2       misho     754:                if (i >= srv->srv_numcli) {
1.5       misho     755: #ifdef HAVE_PTHREAD_YIELD
                    756:                        pthread_yield();
                    757: #else
1.1       misho     758:                        usleep(1000000);
1.5       misho     759: #endif
1.1       misho     760:                        continue;
                    761:                }
1.2       misho     762: 
                    763:                FD_ZERO(&fds);
                    764:                FD_SET(srv->srv_server.cli_sock, &fds);
                    765:                ret = select(srv->srv_server.cli_sock + 1, &fds, NULL, NULL, &tv);
                    766:                if (ret == -1) {
                    767:                        LOGERR;
                    768:                        ret = 1;
                    769:                        break;
                    770:                }
                    771:                if (!ret)
                    772:                        continue;
                    773: 
1.5.2.6 ! misho     774:                c->cli_sock = accept(srv->srv_server.cli_sock, &c->cli_sa.sa, &salen);
1.1       misho     775:                if (c->cli_sock == -1) {
                    776:                        LOGERR;
                    777:                        continue;
                    778:                } else
                    779:                        c->cli_parent = srv;
                    780: 
                    781:                if (pthread_create(&c->cli_tid, NULL, rpc_srv_dispatchCall, c)) {
                    782:                        LOGERR;
                    783:                        continue;
1.4       misho     784:                } else
                    785:                        pthread_detach(c->cli_tid);
1.1       misho     786:        }
                    787: 
                    788:        return 0;
                    789: }
                    790: 
                    791: // ---------------------------------------------------------
                    792: 
                    793: /*
                    794:  * rpc_srv_execCall() Execute registered call from RPC server
                    795:  * @call = Register RPC call
                    796:  * @rpc = IN RPC call structure
1.5       misho     797:  * @args = IN RPC calling arguments from RPC client
1.1       misho     798:  * return: -1 error, !=-1 ok
                    799:  */
                    800: int
1.2       misho     801: rpc_srv_execCall(rpc_func_t * __restrict call, struct tagRPCCall * __restrict rpc, 
1.5       misho     802:                array_t * __restrict args)
1.1       misho     803: {
                    804:        void *dl;
                    805:        rpc_callback_t func;
                    806:        int ret;
                    807: 
1.2       misho     808:        if (!call || !rpc || !call->func_parent) {
1.1       misho     809:                rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t exec call from RPC server ...\n");
                    810:                return -1;
                    811:        }
                    812: 
                    813:        dl = dlopen((char*) (*call->func_file ? call->func_file : NULL), RTLD_NOW);
                    814:        if (!dl) {
                    815:                rpc_SetErr(ENOENT, "Error:: Can`t attach module %s!\n", dlerror());
                    816:                return -1;
                    817:        }
                    818: 
                    819:        func = dlsym(dl, (char*) call->func_name);
                    820:        if (func)
1.5.2.2   misho     821:                ret = func(call, ntohs(rpc->call_argc), args);
1.1       misho     822:        else {
                    823:                rpc_SetErr(ENOEXEC, "Error:: Can`t find function %s!\n", dlerror());
                    824:                ret = -1;
                    825:        }
                    826: 
                    827:        dlclose(dl);
                    828:        return ret;
                    829: }

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