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.7.2.7 2012/05/17 08:42:32 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, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
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_func_t *call, struct tagRPCCall *rpc, array_t *iv)
53: {
54: rpc_srv_t *srv;
55: rpc_cli_t *cli;
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(call);
63: srv = RPC_FUNC_SERVER(call);
64:
65: len = io_arraySize(srv->srv_clients) * STRSIZ;
66: if (!(val = malloc(len))) {
67: LOGERR;
68: return -1;
69: } else
70: memset(val, 0, len);
71:
72: for (i = 0; i < io_arraySize(srv->srv_clients); i++) {
73: cli = io_array(srv->srv_clients, i, rpc_cli_t*);
74: if (!cli)
75: continue;
76:
77: str = io_n2addr(&cli->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: AIT_SET_STR(io_array(RPC_FUNC_RETVARS(call), 0, ait_val_t*), val);
87: free(val);
88: return 0;
89: }
90:
91: static int
92: rpcServerCalls(rpc_func_t *call, struct tagRPCCall *rpc, array_t *iv)
93: {
94: rpc_srv_t *srv;
95: rpc_func_t *f;
96: register int i = 0;
97: int len;
98: char *val, str[MAXPATHLEN];
99:
100: RPC_CALLBACK_CHECK_INPUT(call);
101: srv = RPC_FUNC_SERVER(call);
102:
103: TAILQ_FOREACH(f, &srv->srv_funcs, func_node)
104: i++;
105: len = i * STRSIZ;
106:
107: if (!(val = malloc(len))) {
108: LOGERR;
109: return -1;
110: } else
111: memset(val, 0, len);
112:
113: TAILQ_FOREACH(f, &srv->srv_funcs, func_node)
114: if (AIT_ADDR(&f->func_name)) {
115: memset(str, 0, sizeof str);
116: snprintf(str, sizeof str, "/%hu/0x%p(%d); ", AIT_KEY(&f->func_name),
117: AIT_ADDR(&f->func_name), io_arraySize(RPC_FUNC_RETVARS(f)));
118: strlcat(val, str, len);
119: }
120:
121: AIT_SET_STR(io_getVars(RPC_FUNC_RETVARS(call), 0), val);
122: free(val);
123: return 0;
124: }
125:
126: static int
127: rpcServerSessions(rpc_func_t *call, struct tagRPCCall *rpc, array_t *iv)
128: {
129: rpc_srv_t *srv;
130: ait_val_t *v;
131:
132: RPC_CALLBACK_CHECK_INPUT(call);
133: srv = RPC_FUNC_SERVER(call);
134:
135: v = io_array(RPC_FUNC_RETVARS(call), 0, ait_val_t*);
136: AIT_SET_U8(v, srv->srv_session.sess_version);
137: v = io_array(RPC_FUNC_RETVARS(call), 1, ait_val_t*);
138: AIT_SET_U32(v, srv->srv_session.sess_program);
139: v = io_array(RPC_FUNC_RETVARS(call), 2, ait_val_t*);
140: AIT_SET_U8(v, srv->srv_session.sess_process);
141: v = io_array(RPC_FUNC_RETVARS(call), 3, ait_val_t*);
142: AIT_SET_I32(v, io_arraySize(srv->srv_clients));
143:
144: return 0;
145: }
146:
147: static int
148: rpcServerShutdown(rpc_func_t *call, struct tagRPCCall *rpc, array_t *iv)
149: {
150: rpc_srv_t *srv;
151:
152: RPC_CALLBACK_CHECK_INPUT(call);
153: srv = RPC_FUNC_SERVER(call);
154:
155: srv->srv_kill = 1;
156:
157: return 0;
158: }
159:
160: static int
161: rpcServerPing(rpc_func_t *call, struct tagRPCCall *rpc, array_t *iv)
162: {
163: AIT_SET_U16(io_getVars(RPC_FUNC_RETVARS(call), 0), ntohs(rpc->call_seq));
164: return 0;
165: }
166:
167: /* ---------------------------------------------------- */
168:
169: static int
170: rpcBLOBServerShutdown(rpc_func_t *call, struct tagRPCCall *rpc, array_t *iv)
171: {
172: rpc_srv_t *srv;
173:
174: RPC_CALLBACK_CHECK_INPUT(call);
175: srv = RPC_FUNC_SERVER(call);
176:
177: srv->srv_blob.kill = 1;
178:
179: return 0;
180: }
181:
182: static int
183: rpcBLOBServerVars(rpc_func_t *call, struct tagRPCCall *rpc, array_t *iv)
184: {
185: rpc_srv_t *srv;
186: rpc_blob_t *b;
187: register int i = 0;
188: char *val, str[64];
189: int len;
190:
191: RPC_CALLBACK_CHECK_INPUT(call);
192: srv = RPC_FUNC_SERVER(call);
193:
194: if (srv->srv_blob.kill) {
195: AIT_SET_STR(io_array(RPC_FUNC_RETVARS(call), 0, ait_val_t*), "BLOB Server is killed");
196: return 1;
197: }
198:
199: TAILQ_FOREACH(b, &srv->srv_blob.blobs, blob_node)
200: i++;
201: len = i * sizeof str;
202:
203: if (!len) {
204: AIT_SET_STR(io_array(RPC_FUNC_RETVARS(call), 0, ait_val_t*), "");
205: return 0;
206: }
207:
208: if (!(val = malloc(len))) {
209: LOGERR;
210: return -1;
211: } else
212: memset(val, 0, len);
213:
214: TAILQ_FOREACH(b, &srv->srv_blob.blobs, blob_node) {
215: memset(str, 0, sizeof str);
216: snprintf(str, sizeof str, "0x%0X(%lu)=%p ", b->blob_var, (u_long) b->blob_len, b->blob_data);
217: strlcat(val, str, len);
218: }
219:
220: AIT_SET_STR(io_array(RPC_FUNC_RETVARS(call), 0, ait_val_t*), val);
221: free(val);
222: return 0;
223: }
224:
225: static int
226: rpcBLOBServerClients(rpc_func_t *call, struct tagRPCCall *rpc, array_t *iv)
227: {
228: rpc_srv_t *srv;
229: rpc_cli_t *cli;
230: register int i;
231: int len;
232: const char *str = NULL;
233: char *val;
234: ait_val_t v;
235:
236: RPC_CALLBACK_CHECK_INPUT(call);
237: srv = RPC_FUNC_SERVER(call);
238:
239: if (srv->srv_blob.kill) {
240: AIT_SET_STR(io_array(RPC_FUNC_RETVARS(call), 0, ait_val_t*), "BLOB Server is killed");
241: return 1;
242: }
243:
244: len = io_arraySize(srv->srv_blob.clients) * STRSIZ;
245: if (!(val = malloc(len))) {
246: LOGERR;
247: return -1;
248: } else
249: memset(val, 0, len);
250:
251: for (i = 0; i < io_arraySize(srv->srv_clients); i++) {
252: cli = io_array(srv->srv_blob.clients, i, rpc_cli_t*);
253: if (!cli)
254: continue;
255:
256: str = io_n2addr(&cli->cli_sa, &v);
257: if (str)
258: strlcat(val, (char*) str, len);
259: else
260: strlcat(val, "0.0.0.0", len);
261: strlcat(val, " ", len);
262: AIT_FREE_VAL(&v);
263: }
264:
265: AIT_SET_STR(io_array(RPC_FUNC_RETVARS(call), 0, ait_val_t*), val);
266: free(val);
267: return 0;
268: }
269:
270: /* ----------------------------------------------------------------- */
271:
272: /*
273: * rpc_register_srvPing() - Register ping service function
274: *
275: * @srv = RPC server instance
276: * return: -1 error or 0 ok
277: */
278: inline int
279: rpc_register_srvPing(rpc_srv_t * __restrict srv)
280: {
281: if (!srv)
282: return -1;
283:
284: if (rpc_srv_registerCall(srv, CALL_SRVPING, rpcServerPing, 1) < 1)
285: return -1;
286:
287: return 0;
288: }
289:
290: /*
291: * rpc_register_srvServices() - Register internal service functions
292: *
293: * @srv = RPC server instance
294: * return: -1 error or 0 ok
295: */
296: int
297: rpc_register_srvServices(rpc_srv_t * __restrict srv)
298: {
299: if (!srv)
300: return -1;
301:
302: if (rpc_srv_registerCall(srv, CALL_SRVSHUTDOWN, rpcServerShutdown, 0) < 1)
303: return -1;
304: if (rpc_srv_registerCall(srv, CALL_SRVCLIENTS, rpcServerClients, 1) < 1)
305: return -1;
306: if (rpc_srv_registerCall(srv, CALL_SRVSESSIONS, rpcServerSessions, 4) < 1)
307: return -1;
308: if (rpc_srv_registerCall(srv, CALL_SRVCALLS, rpcServerCalls, 1) < 1)
309: return -1;
310:
311: return 0;
312: }
313:
314: /*
315: * rpc_register_blobServices() - Register internal service functions
316: *
317: * @srv = RPC server instance
318: * return: -1 error or 0 ok
319: */
320: int
321: rpc_register_blobServices(rpc_srv_t * __restrict srv)
322: {
323: if (!srv)
324: return -1;
325:
326: if (rpc_srv_registerCall(srv, CALL_BLOBSHUTDOWN, rpcBLOBServerShutdown, 0) < 1)
327: return -1;
328: if (rpc_srv_registerCall(srv, CALL_BLOBCLIENTS, rpcBLOBServerClients, 1) < 1)
329: return -1;
330: if (rpc_srv_registerCall(srv, CALL_BLOBVARS, rpcBLOBServerVars, 1) < 1)
331: return -1;
332:
333: return 0;
334: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>