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