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>