File:  [ELWIX - Embedded LightWeight unIX -] / libaitrpc / src / builtin.c
Revision 1.22: download - view: text, annotated - select for diffs - revision graph
Thu Jul 2 22:28:15 2015 UTC (9 years ago) by misho
Branches: MAIN
CVS tags: rpc9_2, RPC9_1, HEAD
version 9.1

    1: /*************************************************************************
    2: * (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
    3: *  by Michael Pounov <misho@openbsd-bg.org>
    4: *
    5: * $Author: misho $
    6: * $Id: builtin.c,v 1.22 2015/07/02 22:28:15 misho Exp $
    7: *
    8: **************************************************************************
    9: The ELWIX and AITNET software is distributed under the following
   10: terms:
   11: 
   12: All of the documentation and software included in the ELWIX and AITNET
   13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
   14: 
   15: Copyright 2004 - 2015
   16: 	by Michael Pounov <misho@elwix.org>.  All rights reserved.
   17: 
   18: Redistribution and use in source and binary forms, with or without
   19: modification, are permitted provided that the following conditions
   20: are met:
   21: 1. Redistributions of source code must retain the above copyright
   22:    notice, this list of conditions and the following disclaimer.
   23: 2. Redistributions in binary form must reproduce the above copyright
   24:    notice, this list of conditions and the following disclaimer in the
   25:    documentation and/or other materials provided with the distribution.
   26: 3. All advertising materials mentioning features or use of this software
   27:    must display the following acknowledgement:
   28: This product includes software developed by Michael Pounov <misho@elwix.org>
   29: ELWIX - Embedded LightWeight unIX and its contributors.
   30: 4. Neither the name of AITNET nor the names of its contributors
   31:    may be used to endorse or promote products derived from this software
   32:    without specific prior written permission.
   33: 
   34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
   35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   37: ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
   38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
   39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
   40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
   41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
   42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
   43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
   44: SUCH DAMAGE.
   45: */
   46: #include "global.h"
   47: 
   48: 
   49: /* builtin RPC server functions */
   50: 
   51: static int
   52: rpcServerClients(RPC_CALL_STDARGS)
   53: {
   54: 	rpc_srv_t *srv;
   55: 	rpc_cli_t *c;
   56: 	register int i;
   57: 	int len;
   58: 	const char *str = NULL;
   59: 	char *val;
   60: 	ait_val_t v;
   61: 
   62: 	RPC_CALLBACK_CHECK_INPUT(cli);
   63: 	srv = RPC_SRV_SERVER(cli);
   64: 
   65: 	len = array_Size(srv->srv_clients) * STRSIZ;
   66: 	if (!(val = e_malloc(len))) {
   67: 		LOGERR;
   68: 		return -1;
   69: 	} else
   70: 		memset(val, 0, len);
   71: 
   72: 	for (i = 0; i < array_Size(srv->srv_clients); i++) {
   73: 		c = array(srv->srv_clients, i, rpc_cli_t*);
   74: 		if (!c)
   75: 			continue;
   76: 
   77: 		str = e_n2addr(&c->cli_sa, &v);
   78: 		if (str)
   79: 			strlcat(val, (char*) str, len);
   80: 		else
   81: 			strlcat(val, "0.0.0.0", len);
   82: 		strlcat(val, " ", len);
   83: 		AIT_FREE_VAL(&v);
   84: 	}
   85: 
   86: 	/* return values */
   87: 	AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), val);
   88: 	e_free(val);
   89: 	return 0;
   90: }
   91: 
   92: static int
   93: rpcServerCalls(RPC_CALL_STDARGS)
   94: {
   95: 	rpc_srv_t *srv;
   96: 	rpc_func_t *f;
   97: 	register int i = 0;
   98: 	int len;
   99: 	char *val, str[MAXPATHLEN];
  100: 
  101: 	RPC_CALLBACK_CHECK_INPUT(cli);
  102: 	srv = RPC_SRV_SERVER(cli);
  103: 
  104: 	SLIST_FOREACH(f, &srv->srv_funcs, func_next)
  105: 		i++;
  106: 	len = i * STRSIZ;
  107: 
  108: 	if (!(val = e_malloc(len))) {
  109: 		LOGERR;
  110: 		return -1;
  111: 	} else
  112: 		memset(val, 0, len);
  113: 
  114: 	SLIST_FOREACH(f, &srv->srv_funcs, func_next)
  115: 		if (AIT_ADDR(&f->func_name)) {
  116: 			memset(str, 0, sizeof str);
  117: 			snprintf(str, sizeof str, "/%hu/0x%p; ", AIT_KEY(&f->func_name), 
  118: 					AIT_ADDR(&f->func_name));
  119: 			strlcat(val, str, len);
  120: 		}
  121: 
  122: 	/* return variables */
  123: 	AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), val);
  124: 	e_free(val);
  125: 	return 0;
  126: }
  127: 
  128: static int
  129: rpcServerSessions(RPC_CALL_STDARGS)
  130: {
  131: 	rpc_srv_t *srv;
  132: 
  133: 	RPC_CALLBACK_CHECK_INPUT(cli);
  134: 	srv = RPC_SRV_SERVER(cli);
  135: 
  136: 	AIT_SET_I32(ait_getVars(&RPC_RETVARS(cli), 2), array_Size(srv->srv_clients));
  137: 	AIT_SET_U8(ait_getVars(&RPC_RETVARS(cli), 0), srv->srv_session.sess_version);
  138: 	AIT_SET_U8(ait_getVars(&RPC_RETVARS(cli), 1), srv->srv_session.sess_instance);
  139: 
  140: 	return 0;
  141: }
  142: 
  143: static int
  144: rpcServerShutdown(RPC_CALL_STDARGS)
  145: {
  146: 	rpc_srv_t *srv;
  147: 
  148: 	RPC_CALLBACK_CHECK_INPUT(cli);
  149: 	srv = RPC_SRV_SERVER(cli);
  150: 	
  151: 	srv->srv_kill = 1;
  152: 	if (srv->srv_blob.tid)
  153: 		pthread_kill(srv->srv_blob.tid, SIGFBLOB);
  154: 	return 0;
  155: }
  156: 
  157: static int
  158: rpcServerPing(RPC_CALL_STDARGS)
  159: {
  160: 	if (ntohl((u_long) rpc->call_req.flags) == RPC_REPLY)
  161: 		AIT_SET_U16(ait_getVars(&RPC_RETVARS(cli), 0), random() % USHRT_MAX);
  162: 	return 0;
  163: }
  164: 
  165: /* ---------------------------------------------------- */
  166: 
  167: static int
  168: rpcBLOBServerShutdown(RPC_CALL_STDARGS)
  169: {
  170: 	rpc_srv_t *srv;
  171: 
  172: 	RPC_CALLBACK_CHECK_INPUT(cli);
  173: 	srv = RPC_SRV_SERVER(cli);
  174: 
  175: 	rpc_freeBLOBCli(cli);
  176: 
  177: 	srv->srv_blob.kill = 1;
  178: 	return 0;
  179: }
  180: 
  181: static int
  182: rpcBLOBServerVars(RPC_CALL_STDARGS)
  183: {
  184: 	rpc_srv_t *srv;
  185: 	rpc_blob_t *b;
  186: 	register int i = 0;
  187: 	char *val, str[64];
  188: 	int len;
  189: 
  190: 	RPC_CALLBACK_CHECK_INPUT(cli);
  191: 	srv = RPC_SRV_SERVER(cli);
  192: 
  193: 	if (srv->srv_blob.kill) {
  194: 		AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), "BLOB Server is killed");
  195: 		return 1;
  196: 	}
  197: 
  198: 	TAILQ_FOREACH(b, &srv->srv_blob.blobs, blob_node)
  199: 		i++;
  200: 	len = i * sizeof str;
  201: 
  202: 	if (!len) {
  203: 		AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), "");
  204: 		return 0;
  205: 	}
  206: 
  207: 	if (!(val = e_malloc(len))) {
  208: 		LOGERR;
  209: 		return -1;
  210: 	} else
  211: 		memset(val, 0, len);
  212: 
  213: 	TAILQ_FOREACH(b, &srv->srv_blob.blobs, blob_node) {
  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: 
  219: 	AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), val);
  220: 	e_free(val);
  221: 	return 0;
  222: }
  223: 
  224: static int
  225: rpcBLOBServerClients(RPC_CALL_STDARGS)
  226: {
  227: 	rpc_srv_t *srv;
  228: 	rpc_cli_t *c;
  229: 	register int i;
  230: 	int len;
  231: 	const char *str = NULL;
  232: 	char *val;
  233: 	ait_val_t v;
  234: 
  235: 	RPC_CALLBACK_CHECK_INPUT(cli);
  236: 	srv = RPC_SRV_SERVER(cli);
  237: 
  238: 	if (srv->srv_blob.kill) {
  239: 		AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), "BLOB Server is killed");
  240: 		return 1;
  241: 	}
  242: 
  243: 	len = array_Size(srv->srv_blob.clients) * STRSIZ;
  244: 	if (!(val = e_malloc(len))) {
  245: 		LOGERR;
  246: 		return -1;
  247: 	} else
  248: 		memset(val, 0, len);
  249: 
  250: 	for (i = 0; i < array_Size(srv->srv_clients); i++) {
  251: 		c = array(srv->srv_blob.clients, i, rpc_cli_t*);
  252: 		if (!c)
  253: 			continue;
  254: 
  255: 		str = e_n2addr(&c->cli_sa, &v);
  256: 		if (str)
  257: 			strlcat(val, (char*) str, len);
  258: 		else
  259: 			strlcat(val, "0.0.0.0", len);
  260: 		strlcat(val, " ", len);
  261: 		AIT_FREE_VAL(&v);
  262: 	}
  263: 
  264: 	AIT_SET_STR(ait_getVars(&RPC_RETVARS(cli), 0), val);
  265: 	e_free(val);
  266: 	return 0;
  267: }
  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:  */
  277: int
  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>