Annotation of libaitrpc/src/builtin.c, revision 1.21.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.21.2.1! misho       6: * $Id: builtin.c,v 1.21 2015/05/18 15:09:59 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.8       misho     151:        srv->srv_kill = 1;
1.15      misho     152:        if (srv->srv_blob.tid)
                    153:                pthread_kill(srv->srv_blob.tid, SIGFBLOB);
1.2       misho     154:        return 0;
                    155: }
                    156: 
1.8       misho     157: static int
1.19      misho     158: rpcServerPing(RPC_CALL_STDARGS)
1.8       misho     159: {
1.17      misho     160:        if (ntohl((u_long) rpc->call_req.flags) == RPC_REPLY)
1.16      misho     161:                AIT_SET_U16(ait_getVars(&RPC_RETVARS(cli), 0), random() % USHRT_MAX);
1.8       misho     162:        return 0;
                    163: }
                    164: 
                    165: /* ---------------------------------------------------- */
1.2       misho     166: 
1.8       misho     167: static int
1.19      misho     168: rpcBLOBServerShutdown(RPC_CALL_STDARGS)
1.2       misho     169: {
                    170:        rpc_srv_t *srv;
                    171: 
1.8       misho     172:        RPC_CALLBACK_CHECK_INPUT(cli);
                    173:        srv = RPC_SRV_SERVER(cli);
1.2       misho     174: 
1.11      misho     175:        rpc_freeBLOBCli(cli);
                    176: 
1.8       misho     177:        srv->srv_blob.kill = 1;
1.2       misho     178:        return 0;
                    179: }
                    180: 
1.8       misho     181: static int
1.19      misho     182: rpcBLOBServerVars(RPC_CALL_STDARGS)
1.2       misho     183: {
                    184:        rpc_srv_t *srv;
                    185:        rpc_blob_t *b;
1.8       misho     186:        register int i = 0;
1.4       misho     187:        char *val, str[64];
                    188:        int len;
1.2       misho     189: 
1.8       misho     190:        RPC_CALLBACK_CHECK_INPUT(cli);
                    191:        srv = RPC_SRV_SERVER(cli);
1.2       misho     192: 
1.8       misho     193:        if (srv->srv_blob.kill) {
1.12      misho     194:                AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), "BLOB Server is killed");
1.4       misho     195:                return 1;
                    196:        }
                    197: 
1.8       misho     198:        TAILQ_FOREACH(b, &srv->srv_blob.blobs, blob_node)
                    199:                i++;
1.4       misho     200:        len = i * sizeof str;
                    201: 
                    202:        if (!len) {
1.12      misho     203:                AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), "");
1.2       misho     204:                return 0;
                    205:        }
                    206: 
1.12      misho     207:        if (!(val = e_malloc(len))) {
1.4       misho     208:                LOGERR;
                    209:                return -1;
                    210:        } else
                    211:                memset(val, 0, len);
                    212: 
1.8       misho     213:        TAILQ_FOREACH(b, &srv->srv_blob.blobs, blob_node) {
1.4       misho     214:                memset(str, 0, sizeof str);
                    215:                snprintf(str, sizeof str, "0x%0X(%lu)=%p ", b->blob_var, (u_long) b->blob_len, b->blob_data);
                    216:                strlcat(val, str, len);
                    217:        }
                    218: 
1.12      misho     219:        AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), val);
                    220:        e_free(val);
1.2       misho     221:        return 0;
                    222: }
                    223: 
1.8       misho     224: static int
1.19      misho     225: rpcBLOBServerClients(RPC_CALL_STDARGS)
1.2       misho     226: {
                    227:        rpc_srv_t *srv;
1.8       misho     228:        rpc_cli_t *c;
1.2       misho     229:        register int i;
1.4       misho     230:        int len;
1.3       misho     231:        const char *str = NULL;
1.8       misho     232:        char *val;
                    233:        ait_val_t v;
1.2       misho     234: 
1.8       misho     235:        RPC_CALLBACK_CHECK_INPUT(cli);
                    236:        srv = RPC_SRV_SERVER(cli);
1.4       misho     237: 
1.8       misho     238:        if (srv->srv_blob.kill) {
1.12      misho     239:                AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), "BLOB Server is killed");
1.4       misho     240:                return 1;
                    241:        }
                    242: 
1.12      misho     243:        len = array_Size(srv->srv_blob.clients) * STRSIZ;
                    244:        if (!(val = e_malloc(len))) {
1.4       misho     245:                LOGERR;
1.2       misho     246:                return -1;
1.4       misho     247:        } else
                    248:                memset(val, 0, len);
1.2       misho     249: 
1.12      misho     250:        for (i = 0; i < array_Size(srv->srv_clients); i++) {
                    251:                c = array(srv->srv_blob.clients, i, rpc_cli_t*);
1.8       misho     252:                if (!c)
1.2       misho     253:                        continue;
                    254: 
1.12      misho     255:                str = e_n2addr(&c->cli_sa, &v);
1.2       misho     256:                if (str)
1.4       misho     257:                        strlcat(val, (char*) str, len);
1.2       misho     258:                else
1.4       misho     259:                        strlcat(val, "0.0.0.0", len);
                    260:                strlcat(val, " ", len);
1.8       misho     261:                AIT_FREE_VAL(&v);
1.2       misho     262:        }
                    263: 
1.12      misho     264:        AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), val);
                    265:        e_free(val);
1.2       misho     266:        return 0;
                    267: }
1.8       misho     268: 
                    269: /* ----------------------------------------------------------------- */
                    270: 
                    271: /*
                    272:  * rpc_register_srvPing() - Register ping service function
                    273:  *
                    274:  * @srv = RPC server instance
                    275:  * return: -1 error or 0 ok
                    276:  */
1.14      misho     277: int
1.8       misho     278: rpc_register_srvPing(rpc_srv_t * __restrict srv)
                    279: {
                    280:        if (!srv)
                    281:                return -1;
                    282: 
                    283:        if (rpc_srv_registerCall(srv, CALL_SRVPING, rpcServerPing) < 1)
                    284:                return -1;
                    285: 
                    286:        return 0;
                    287: }
                    288: 
                    289: /*
                    290:  * rpc_register_srvServices() - Register internal service functions
                    291:  *
                    292:  * @srv = RPC server instance
                    293:  * return: -1 error or 0 ok
                    294:  */
                    295: int
                    296: rpc_register_srvServices(rpc_srv_t * __restrict srv)
                    297: {
                    298:        if (!srv)
                    299:                return -1;
                    300: 
                    301:        if (rpc_srv_registerCall(srv, CALL_SRVSHUTDOWN, rpcServerShutdown) < 1)
                    302:                return -1;
                    303:        if (rpc_srv_registerCall(srv, CALL_SRVCLIENTS, rpcServerClients) < 1)
                    304:                return -1;
                    305:        if (rpc_srv_registerCall(srv, CALL_SRVSESSIONS, rpcServerSessions) < 1)
                    306:                return -1;
                    307:        if (rpc_srv_registerCall(srv, CALL_SRVCALLS, rpcServerCalls) < 1)
                    308:                return -1;
                    309: 
                    310:        return 0;
                    311: }
                    312: 
                    313: /*
                    314:  * rpc_register_blobServices() - Register internal service functions
                    315:  *
                    316:  * @srv = RPC server instance
                    317:  * return: -1 error or 0 ok
                    318:  */
                    319: int
                    320: rpc_register_blobServices(rpc_srv_t * __restrict srv)
                    321: {
                    322:        if (!srv)
                    323:                return -1;
                    324: 
                    325:        if (rpc_srv_registerCall(srv, CALL_BLOBSHUTDOWN, rpcBLOBServerShutdown) < 1)
                    326:                return -1;
                    327:        if (rpc_srv_registerCall(srv, CALL_BLOBCLIENTS, rpcBLOBServerClients) < 1)
                    328:                return -1;
                    329:        if (rpc_srv_registerCall(srv, CALL_BLOBVARS, rpcBLOBServerVars) < 1)
                    330:                return -1;
                    331: 
                    332:        return 0;
                    333: }

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