Annotation of libaitrpc/src/builtin.c, revision 1.22.2.1

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.22.2.1! misho       6: * $Id: builtin.c,v 1.22 2015/07/02 22:28:15 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.20      misho      15: Copyright 2004 - 2015
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: 
                     49: /* builtin RPC server functions */
                     50: 
1.8       misho      51: static int
1.19      misho      52: rpcServerClients(RPC_CALL_STDARGS)
1.1       misho      53: {
1.2       misho      54:        rpc_srv_t *srv;
1.8       misho      55:        rpc_cli_t *c;
1.1       misho      56:        register int i;
1.4       misho      57:        int len;
1.3       misho      58:        const char *str = NULL;
1.8       misho      59:        char *val;
                     60:        ait_val_t v;
1.2       misho      61: 
1.8       misho      62:        RPC_CALLBACK_CHECK_INPUT(cli);
                     63:        srv = RPC_SRV_SERVER(cli);
1.4       misho      64: 
1.12      misho      65:        len = array_Size(srv->srv_clients) * STRSIZ;
                     66:        if (!(val = e_malloc(len))) {
1.4       misho      67:                LOGERR;
1.1       misho      68:                return -1;
1.4       misho      69:        } else
                     70:                memset(val, 0, len);
1.1       misho      71: 
1.12      misho      72:        for (i = 0; i < array_Size(srv->srv_clients); i++) {
                     73:                c = array(srv->srv_clients, i, rpc_cli_t*);
1.8       misho      74:                if (!c)
1.1       misho      75:                        continue;
                     76: 
1.12      misho      77:                str = e_n2addr(&c->cli_sa, &v);
1.2       misho      78:                if (str)
1.4       misho      79:                        strlcat(val, (char*) str, len);
1.2       misho      80:                else
1.4       misho      81:                        strlcat(val, "0.0.0.0", len);
                     82:                strlcat(val, " ", len);
1.8       misho      83:                AIT_FREE_VAL(&v);
1.1       misho      84:        }
                     85: 
1.8       misho      86:        /* return values */
1.12      misho      87:        AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), val);
                     88:        e_free(val);
1.1       misho      89:        return 0;
                     90: }
                     91: 
1.8       misho      92: static int
1.19      misho      93: rpcServerCalls(RPC_CALL_STDARGS)
1.1       misho      94: {
1.2       misho      95:        rpc_srv_t *srv;
1.1       misho      96:        rpc_func_t *f;
1.8       misho      97:        register int i = 0;
1.4       misho      98:        int len;
                     99:        char *val, str[MAXPATHLEN];
1.1       misho     100: 
1.8       misho     101:        RPC_CALLBACK_CHECK_INPUT(cli);
                    102:        srv = RPC_SRV_SERVER(cli);
1.1       misho     103: 
1.10      misho     104:        SLIST_FOREACH(f, &srv->srv_funcs, func_next)
1.8       misho     105:                i++;
1.4       misho     106:        len = i * STRSIZ;
                    107: 
1.12      misho     108:        if (!(val = e_malloc(len))) {
1.4       misho     109:                LOGERR;
1.1       misho     110:                return -1;
1.4       misho     111:        } else
                    112:                memset(val, 0, len);
1.1       misho     113: 
1.10      misho     114:        SLIST_FOREACH(f, &srv->srv_funcs, func_next)
1.8       misho     115:                if (AIT_ADDR(&f->func_name)) {
1.3       misho     116:                        memset(str, 0, sizeof str);
1.8       misho     117:                        snprintf(str, sizeof str, "/%hu/0x%p; ", AIT_KEY(&f->func_name), 
                    118:                                        AIT_ADDR(&f->func_name));
1.4       misho     119:                        strlcat(val, str, len);
1.1       misho     120:                }
                    121: 
1.8       misho     122:        /* return variables */
1.12      misho     123:        AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), val);
                    124:        e_free(val);
1.1       misho     125:        return 0;
                    126: }
                    127: 
1.8       misho     128: static int
1.19      misho     129: rpcServerSessions(RPC_CALL_STDARGS)
1.1       misho     130: {
1.2       misho     131:        rpc_srv_t *srv;
1.1       misho     132: 
1.8       misho     133:        RPC_CALLBACK_CHECK_INPUT(cli);
                    134:        srv = RPC_SRV_SERVER(cli);
1.1       misho     135: 
1.13      misho     136:        AIT_SET_I32(ait_getVars(&RPC_RETVARS(cli), 2), array_Size(srv->srv_clients));
1.12      misho     137:        AIT_SET_U8(ait_getVars(&RPC_RETVARS(cli), 0), srv->srv_session.sess_version);
1.13      misho     138:        AIT_SET_U8(ait_getVars(&RPC_RETVARS(cli), 1), srv->srv_session.sess_instance);
1.1       misho     139: 
                    140:        return 0;
                    141: }
1.2       misho     142: 
1.8       misho     143: static int
1.19      misho     144: rpcServerShutdown(RPC_CALL_STDARGS)
1.2       misho     145: {
                    146:        rpc_srv_t *srv;
                    147: 
1.8       misho     148:        RPC_CALLBACK_CHECK_INPUT(cli);
                    149:        srv = RPC_SRV_SERVER(cli);
1.11      misho     150:        
1.15      misho     151:        if (srv->srv_blob.tid)
                    152:                pthread_kill(srv->srv_blob.tid, SIGFBLOB);
1.22.2.1! misho     153:        while (srv->srv_blob.root) {
        !           154:                srv->srv_blob.kill = 1;
        !           155:                usleep(1000);
        !           156:        }
        !           157:        srv->srv_kill = 1;
1.2       misho     158:        return 0;
                    159: }
                    160: 
1.8       misho     161: static int
1.19      misho     162: rpcServerPing(RPC_CALL_STDARGS)
1.8       misho     163: {
1.17      misho     164:        if (ntohl((u_long) rpc->call_req.flags) == RPC_REPLY)
1.16      misho     165:                AIT_SET_U16(ait_getVars(&RPC_RETVARS(cli), 0), random() % USHRT_MAX);
1.8       misho     166:        return 0;
                    167: }
                    168: 
                    169: /* ---------------------------------------------------- */
1.2       misho     170: 
1.8       misho     171: static int
1.19      misho     172: rpcBLOBServerShutdown(RPC_CALL_STDARGS)
1.2       misho     173: {
                    174:        rpc_srv_t *srv;
                    175: 
1.8       misho     176:        RPC_CALLBACK_CHECK_INPUT(cli);
                    177:        srv = RPC_SRV_SERVER(cli);
1.2       misho     178: 
1.11      misho     179:        rpc_freeBLOBCli(cli);
                    180: 
1.8       misho     181:        srv->srv_blob.kill = 1;
1.2       misho     182:        return 0;
                    183: }
                    184: 
1.8       misho     185: static int
1.19      misho     186: rpcBLOBServerVars(RPC_CALL_STDARGS)
1.2       misho     187: {
                    188:        rpc_srv_t *srv;
                    189:        rpc_blob_t *b;
1.8       misho     190:        register int i = 0;
1.4       misho     191:        char *val, str[64];
                    192:        int len;
1.2       misho     193: 
1.8       misho     194:        RPC_CALLBACK_CHECK_INPUT(cli);
                    195:        srv = RPC_SRV_SERVER(cli);
1.2       misho     196: 
1.8       misho     197:        if (srv->srv_blob.kill) {
1.12      misho     198:                AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), "BLOB Server is killed");
1.4       misho     199:                return 1;
                    200:        }
                    201: 
1.8       misho     202:        TAILQ_FOREACH(b, &srv->srv_blob.blobs, blob_node)
                    203:                i++;
1.4       misho     204:        len = i * sizeof str;
                    205: 
                    206:        if (!len) {
1.12      misho     207:                AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), "");
1.2       misho     208:                return 0;
                    209:        }
                    210: 
1.12      misho     211:        if (!(val = e_malloc(len))) {
1.4       misho     212:                LOGERR;
                    213:                return -1;
                    214:        } else
                    215:                memset(val, 0, len);
                    216: 
1.8       misho     217:        TAILQ_FOREACH(b, &srv->srv_blob.blobs, blob_node) {
1.4       misho     218:                memset(str, 0, sizeof str);
                    219:                snprintf(str, sizeof str, "0x%0X(%lu)=%p ", b->blob_var, (u_long) b->blob_len, b->blob_data);
                    220:                strlcat(val, str, len);
                    221:        }
                    222: 
1.12      misho     223:        AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), val);
                    224:        e_free(val);
1.2       misho     225:        return 0;
                    226: }
                    227: 
1.8       misho     228: static int
1.19      misho     229: rpcBLOBServerClients(RPC_CALL_STDARGS)
1.2       misho     230: {
                    231:        rpc_srv_t *srv;
1.8       misho     232:        rpc_cli_t *c;
1.2       misho     233:        register int i;
1.4       misho     234:        int len;
1.3       misho     235:        const char *str = NULL;
1.8       misho     236:        char *val;
                    237:        ait_val_t v;
1.2       misho     238: 
1.8       misho     239:        RPC_CALLBACK_CHECK_INPUT(cli);
                    240:        srv = RPC_SRV_SERVER(cli);
1.4       misho     241: 
1.8       misho     242:        if (srv->srv_blob.kill) {
1.12      misho     243:                AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), "BLOB Server is killed");
1.4       misho     244:                return 1;
                    245:        }
                    246: 
1.12      misho     247:        len = array_Size(srv->srv_blob.clients) * STRSIZ;
                    248:        if (!(val = e_malloc(len))) {
1.4       misho     249:                LOGERR;
1.2       misho     250:                return -1;
1.4       misho     251:        } else
                    252:                memset(val, 0, len);
1.2       misho     253: 
1.12      misho     254:        for (i = 0; i < array_Size(srv->srv_clients); i++) {
                    255:                c = array(srv->srv_blob.clients, i, rpc_cli_t*);
1.8       misho     256:                if (!c)
1.2       misho     257:                        continue;
                    258: 
1.12      misho     259:                str = e_n2addr(&c->cli_sa, &v);
1.2       misho     260:                if (str)
1.4       misho     261:                        strlcat(val, (char*) str, len);
1.2       misho     262:                else
1.4       misho     263:                        strlcat(val, "0.0.0.0", len);
                    264:                strlcat(val, " ", len);
1.8       misho     265:                AIT_FREE_VAL(&v);
1.2       misho     266:        }
                    267: 
1.12      misho     268:        AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), val);
                    269:        e_free(val);
1.2       misho     270:        return 0;
                    271: }
1.8       misho     272: 
                    273: /* ----------------------------------------------------------------- */
                    274: 
                    275: /*
                    276:  * rpc_register_srvPing() - Register ping service function
                    277:  *
                    278:  * @srv = RPC server instance
                    279:  * return: -1 error or 0 ok
                    280:  */
1.14      misho     281: int
1.8       misho     282: rpc_register_srvPing(rpc_srv_t * __restrict srv)
                    283: {
                    284:        if (!srv)
                    285:                return -1;
                    286: 
                    287:        if (rpc_srv_registerCall(srv, CALL_SRVPING, rpcServerPing) < 1)
                    288:                return -1;
                    289: 
                    290:        return 0;
                    291: }
                    292: 
                    293: /*
                    294:  * rpc_register_srvServices() - Register internal service functions
                    295:  *
                    296:  * @srv = RPC server instance
                    297:  * return: -1 error or 0 ok
                    298:  */
                    299: int
                    300: rpc_register_srvServices(rpc_srv_t * __restrict srv)
                    301: {
                    302:        if (!srv)
                    303:                return -1;
                    304: 
                    305:        if (rpc_srv_registerCall(srv, CALL_SRVSHUTDOWN, rpcServerShutdown) < 1)
                    306:                return -1;
                    307:        if (rpc_srv_registerCall(srv, CALL_SRVCLIENTS, rpcServerClients) < 1)
                    308:                return -1;
                    309:        if (rpc_srv_registerCall(srv, CALL_SRVSESSIONS, rpcServerSessions) < 1)
                    310:                return -1;
                    311:        if (rpc_srv_registerCall(srv, CALL_SRVCALLS, rpcServerCalls) < 1)
                    312:                return -1;
                    313: 
                    314:        return 0;
                    315: }
                    316: 
                    317: /*
                    318:  * rpc_register_blobServices() - Register internal service functions
                    319:  *
                    320:  * @srv = RPC server instance
                    321:  * return: -1 error or 0 ok
                    322:  */
                    323: int
                    324: rpc_register_blobServices(rpc_srv_t * __restrict srv)
                    325: {
                    326:        if (!srv)
                    327:                return -1;
                    328: 
                    329:        if (rpc_srv_registerCall(srv, CALL_BLOBSHUTDOWN, rpcBLOBServerShutdown) < 1)
                    330:                return -1;
                    331:        if (rpc_srv_registerCall(srv, CALL_BLOBCLIENTS, rpcBLOBServerClients) < 1)
                    332:                return -1;
                    333:        if (rpc_srv_registerCall(srv, CALL_BLOBVARS, rpcBLOBServerVars) < 1)
                    334:                return -1;
                    335: 
                    336:        return 0;
                    337: }

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