![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to srv.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / libaitrpc / src|
Return to srv.c CVS log | Up to [ELWIX - Embedded LightWeight unIX -] / libaitrpc / src |
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.1.1.1.2.25! misho 6: * $Id: srv.c,v 1.1.1.1.2.24 2011/03/15 16:48:31 misho Exp $
1.1 misho 7: *
8: *************************************************************************/
9: #include "global.h"
10:
11:
12: static void *
13: rpc_srv_dispatchCall(void *arg)
14: {
1.1.1.1.2.9 misho 15: rpc_cli_t *c = arg;
1.1 misho 16: rpc_srv_t *s;
1.1.1.1.2.12 misho 17: rpc_val_t *vals = NULL, *v = NULL;
1.1 misho 18: rpc_func_t *f;
19: struct tagRPCCall *rpc;
20: struct tagRPCRet rrpc;
21: fd_set fds;
22: u_char buf[BUFSIZ], *data;
1.1.1.1.2.12 misho 23: int ret, argc = 0, Limit = 0;
1.1 misho 24: register int i;
25:
26: if (!arg) {
27: rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t procced RPC client ...\n");
28: return NULL;
29: } else
30: s = c->cli_parent;
31:
32: do {
33: FD_ZERO(&fds);
34: FD_SET(c->cli_sock, &fds);
35: ret = select(c->cli_sock + 1, &fds, NULL, NULL, NULL);
36: if (ret == -1) {
37: ret = -2;
38: }
39: memset(&rrpc, 0, sizeof rrpc);
40: memset(buf, 0, BUFSIZ);
41: if ((ret = recv(c->cli_sock, buf, BUFSIZ, 0)) == -1) {
42: LOGERR;
43: ret = -3;
44: break;
45: }
46: if (!ret) { // receive EOF
47: ret = 0;
48: break;
49: }
50: if (ret < sizeof(struct tagRPCCall)) {
51: rpc_SetErr(EMSGSIZE, "Error:: too short RPC packet ...\n");
52: ret = -4;
53: break;
54: } else
55: rpc = (struct tagRPCCall*) buf;
56: // check RPC packet session info
57: if (memcmp(&rpc->call_session, &s->srv_session, sizeof rpc->call_session)) {
58: rpc_SetErr(EINVAL, "Error:: get invalid RPC session ...\n");
59: ret = -5;
1.1.1.1.2.12 misho 60: goto makeReply;
1.1.1.1.2.22 misho 61: } else
62: Limit = sizeof(struct tagRPCCall);
1.1 misho 63: // RPC is OK! Go decapsulate variables ...
64: if (rpc->call_argc) {
1.1.1.1.2.22 misho 65: v = (rpc_val_t*) (buf + Limit);
66: // check RPC packet length
67: if (rpc->call_argc * sizeof(rpc_val_t) > BUFSIZ - Limit) {
68: rpc_SetErr(EMSGSIZE, "Error:: Too big RPC packet ...\n");
69: ret = -5;
70: goto makeReply;
71: } else
72: Limit += rpc->call_argc * sizeof(rpc_val_t);
1.1 misho 73: // RPC received variables types OK!
74: data = (u_char*) v + rpc->call_argc * sizeof(rpc_val_t);
75: for (i = 0; i < rpc->call_argc; i++) {
76: switch (v[i].val_type) {
77: case buffer:
1.1.1.1.2.22 misho 78: if (v[i].val_len > BUFSIZ - Limit) {
79: rpc_SetErr(EMSGSIZE, "Error:: Too big RPC packet ...\n");
80: ret = -5;
81: goto makeReply;
82: } else
83: Limit += v[i].val_len;
84:
1.1 misho 85: v[i].val.buffer = data;
86: data += v[i].val_len;
87: break;
88: case string:
1.1.1.1.2.22 misho 89: if (v[i].val_len + 1 > BUFSIZ - Limit) {
90: rpc_SetErr(EMSGSIZE, "Error:: Too big RPC packet ...\n");
91: ret = -5;
92: goto makeReply;
93: } else
94: Limit += v[i].val_len;
95:
1.1 misho 96: v[i].val.string = (int8_t*) data;
97: data += v[i].val_len + 1;
98: break;
1.1.1.1.2.12 misho 99: case blob:
100: if (s->srv_blob.state == disable) {
101: rpc_SetErr(ENOTSUP, "Error:: BLOB server is disabled\n");
102: ret = -5;
103: goto makeReply;
104: }
1.1 misho 105: default:
106: break;
107: }
108: }
109: }
110:
111: argc = 0;
112: vals = NULL;
113: if (!(f = rpc_srv_getCall(s, rpc->call_tag, rpc->call_hash))) {
114: rpc_SetErr(EINVAL, "Error:: call not found into RPC server ...\n");
115: ret = -6;
116: } else
1.1.1.1.2.11 misho 117: if ((ret = rpc_srv_execCall(f, rpc, v)) == -1)
1.1.1.1.2.16 misho 118: ret = -9;
1.1 misho 119: else
120: argc = rpc_srv_getValsCall(f, &vals);
121:
1.1.1.1.2.12 misho 122: makeReply:
1.1 misho 123: memcpy(&rrpc.ret_session, &rpc->call_session, sizeof rrpc.ret_session);
124: rrpc.ret_tag = rpc->call_tag;
125: rrpc.ret_hash = rpc->call_hash;
126: rrpc.ret_errno = rpc_Errno;
127: rrpc.ret_retcode = ret;
128: rrpc.ret_argc = argc;
129:
130: memset(buf, 0, BUFSIZ);
131: memcpy(buf, &rrpc, (Limit = sizeof rrpc));
132: if (argc && vals) {
133: v = (rpc_val_t*) (buf + sizeof rrpc);
134: memcpy(v, vals, argc * sizeof(rpc_val_t));
135: Limit += argc * sizeof(rpc_val_t);
136: data = (u_char*) v + argc * sizeof(rpc_val_t);
137: for (ret = i = 0; i < argc; i++) {
138: switch (vals[i].val_type) {
139: case buffer:
140: if (ret || Limit + vals[i].val_len > BUFSIZ) {
141: rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n");
142: rrpc.ret_retcode = ret = -7;
143: rrpc.ret_argc = 0;
144: break;
145: }
146:
147: memcpy(data, vals[i].val.buffer, vals[i].val_len);
148: data += vals[i].val_len;
149: Limit += vals[i].val_len;
150: break;
151: case string:
152: if (ret || Limit + vals[i].val_len + 1 > BUFSIZ) {
153: rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n");
154: rrpc.ret_retcode = ret = -7;
155: rrpc.ret_argc = 0;
156: break;
157: }
158:
159: memcpy(data, vals[i].val.string, vals[i].val_len + 1);
160: data += vals[i].val_len + 1;
161: Limit += vals[i].val_len + 1;
162: break;
1.1.1.1.2.12 misho 163: case blob:
164: if (s->srv_blob.state == disable) {
165: rpc_SetErr(ENOTSUP, "Error:: BLOB server is disabled\n");
166: rrpc.ret_retcode = ret = -5;
167: rrpc.ret_argc = 0;
168: break;
169: }
1.1 misho 170: default:
171: break;
172: }
173:
174: RPC_FREE_VAL(&vals[i]);
175: }
176: }
177:
178: if ((ret = send(c->cli_sock, buf, Limit, 0)) == -1) {
179: LOGERR;
180: ret = -8;
181: break;
182: }
183: if (ret != Limit) {
1.1.1.1.2.23 misho 184: rpc_SetErr(ECANCELED, "Error:: in send RPC request, should be send %d bytes, "
1.1 misho 185: "really is %d\n", Limit, ret);
186: ret = -9;
187: break;
188: }
189: } while (ret > -1);
190:
191: shutdown(c->cli_sock, SHUT_RDWR);
192: close(c->cli_sock);
193: memset(c, 0, sizeof(rpc_cli_t));
1.1.1.1.2.25! misho 194: return (void*) (long)ret;
1.1 misho 195: }
196:
1.1.1.1.2.4 misho 197:
198: static void *
199: rpc_srv_dispatchVars(void *arg)
200: {
201: rpc_cli_t *c = arg;
1.1.1.1.2.5 misho 202: rpc_srv_t *s;
203: rpc_blob_t *b;
1.1.1.1.2.18 misho 204: int ret;
1.1.1.1.2.5 misho 205: fd_set fds;
1.1.1.1.2.6 misho 206: u_char buf[sizeof(struct tagBLOBHdr)];
1.1.1.1.2.5 misho 207: struct tagBLOBHdr *blob;
208:
209: if (!arg) {
210: rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t procced BLOB client ...\n");
211: return NULL;
212: } else
213: s = c->cli_parent;
214:
215: do {
1.1.1.1.2.12 misho 216: // check for disable service at this moment?
217: if (s->srv_blob.state == disable) {
218: ret = 0;
219: break;
220: }
221:
1.1.1.1.2.5 misho 222: FD_ZERO(&fds);
223: FD_SET(c->cli_sock, &fds);
224: ret = select(c->cli_sock + 1, &fds, NULL, NULL, NULL);
225: if (ret == -1) {
226: ret = -2;
1.1.1.1.2.18 misho 227: }
228:
1.1.1.1.2.6 misho 229: memset(buf, 0, sizeof buf);
230: if ((ret = recv(c->cli_sock, buf, sizeof buf, 0)) == -1) {
1.1.1.1.2.5 misho 231: LOGERR;
232: ret = -3;
233: break;
234: }
1.1.1.1.2.10 misho 235: if (!ret || s->srv_blob.state == disable) { // receive EOF or disable service
1.1.1.1.2.5 misho 236: ret = 0;
237: break;
238: }
239: if (ret < sizeof(struct tagBLOBHdr)) {
240: rpc_SetErr(EMSGSIZE, "Error:: too short BLOB packet ...\n");
241: ret = -4;
242: break;
243: } else
244: blob = (struct tagBLOBHdr*) buf;
245: // check BLOB packet session info
1.1.1.1.2.18 misho 246: if (memcmp(&blob->hdr_session, &s->srv_session, sizeof blob->hdr_session)) {
247: rpc_SetErr(EINVAL, "Error:: get invalid BLOB session ...\n");
1.1.1.1.2.5 misho 248: ret = -5;
1.1.1.1.2.12 misho 249: goto makeReply;
1.1.1.1.2.5 misho 250: }
1.1.1.1.2.8 misho 251: // Go to proceed packet ...
1.1.1.1.2.5 misho 252: switch (blob->hdr_cmd) {
253: case get:
1.1.1.1.2.9 misho 254: if (!(b = rpc_srv_getBLOB(s, blob->hdr_var))) {
255: rpc_SetErr(EINVAL, "Error:: var (%x) not found into BLOB server ...\n",
256: blob->hdr_var);
257: ret = -6;
258: break;
1.1.1.1.2.19 misho 259: } else
260: blob->hdr_len = b->blob_len;
1.1.1.1.2.9 misho 261:
1.1.1.1.2.6 misho 262: if (rpc_srv_blobMap(s, b) != -1) {
263: ret = rpc_srv_sendBLOB(c, b);
264: rpc_srv_blobUnmap(b);
265: } else
266: ret = -7;
1.1.1.1.2.5 misho 267: break;
268: case set:
1.1.1.1.2.9 misho 269: if ((b = rpc_srv_registerBLOB(s, blob->hdr_len))) {
270: // set new BLOB variable for reply :)
271: blob->hdr_var = b->blob_var;
1.1.1.1.2.8 misho 272:
1.1.1.1.2.9 misho 273: ret = rpc_srv_recvBLOB(c, b);
274: rpc_srv_blobUnmap(b);
275: } else
1.1.1.1.2.6 misho 276: ret = -7;
1.1.1.1.2.5 misho 277: break;
278: case unset:
1.1.1.1.2.9 misho 279: ret = rpc_srv_unregisterBLOB(s, blob->hdr_var);
1.1.1.1.2.6 misho 280: if (ret == -1)
281: ret = -7;
1.1.1.1.2.5 misho 282: break;
283: default:
284: rpc_SetErr(EINVAL, "Error:: unsupported BLOB command (%d)...\n",
285: blob->hdr_cmd);
1.1.1.1.2.7 misho 286: ret = -7;
1.1.1.1.2.5 misho 287: }
1.1.1.1.2.6 misho 288:
1.1.1.1.2.12 misho 289: makeReply:
1.1.1.1.2.6 misho 290: // Replay to client!
1.1.1.1.2.7 misho 291: blob->hdr_cmd = ret < 0 ? error : ok;
1.1.1.1.2.18 misho 292: blob->hdr_ret = ret;
1.1.1.1.2.7 misho 293: if ((ret = send(c->cli_sock, buf, sizeof buf, 0)) == -1) {
294: LOGERR;
295: ret = -8;
296: break;
297: }
298: if (ret != sizeof buf) {
1.1.1.1.2.23 misho 299: rpc_SetErr(ECANCELED, "Error:: in send BLOB reply, should be send %d bytes, "
1.1.1.1.2.7 misho 300: "really is %d\n", sizeof buf, ret);
301: ret = -9;
302: break;
303: }
1.1.1.1.2.5 misho 304: } while (ret > -1);
1.1.1.1.2.4 misho 305:
1.1.1.1.2.5 misho 306: shutdown(c->cli_sock, SHUT_RDWR);
307: close(c->cli_sock);
308: memset(c, 0, sizeof(rpc_cli_t));
1.1.1.1.2.25! misho 309: return (void*) (long)ret;
1.1.1.1.2.4 misho 310: }
311:
1.1 misho 312: // -------------------------------------------------
313:
314: /*
1.1.1.1.2.4 misho 315: * rpc_srv_initBLOBServer() Init & create BLOB Server
316: * @Port = Port for bind server, if Port == 0 default port is selected
1.1.1.1.2.5 misho 317: * @diskDir = Disk place for BLOB file objects
1.1.1.1.2.4 misho 318: * return: -1 == error or 0 bind and created BLOB server instance
319: */
320: int
1.1.1.1.2.5 misho 321: rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_short Port, const char *diskDir)
1.1.1.1.2.4 misho 322: {
323: int n = 1;
324: struct sockaddr_in sin;
325: struct sockaddr_in6 sin6;
326:
327: if (!srv) {
328: rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t init BLOB server ...\n");
329: return -1;
330: }
331: if (srv->srv_blob.state) {
332: rpc_SetErr(EPERM, "Warning:: Already started BLOB server!\n");
333: return 0;
334: }
335: if (!Port)
336: Port = RPC_DEFPORT + 1;
337:
338: memset(&srv->srv_blob, 0, sizeof srv->srv_blob);
1.1.1.1.2.5 misho 339: if (access(diskDir, R_OK | W_OK) == -1) {
340: LOGERR;
341: return -1;
342: } else
343: strlcpy(srv->srv_blob.dir, diskDir, UCHAR_MAX + 1);
344:
1.1.1.1.2.4 misho 345: srv->srv_blob.server.cli_tid = pthread_self();
346: srv->srv_blob.server.cli_parent = srv;
347: if (srv->srv_server.cli_sa.sa_family == AF_INET) {
348: memcpy(&sin, &srv->srv_server.cli_sa, sizeof sin);
349: sin.sin_port = htons(Port);
350: memcpy(&srv->srv_blob.server.cli_sa, &sin, sizeof(struct sockaddr));
351: } else {
352: memcpy(&sin6, &srv->srv_server.cli_sa, sizeof sin6);
353: sin6.sin6_port = htons(Port);
354: memcpy(&srv->srv_blob.server.cli_sa, &sin6, sizeof(struct sockaddr));
355: }
356:
357: srv->srv_blob.server.cli_sock = socket(srv->srv_server.cli_sa.sa_family, SOCK_STREAM, 0);
358: if (srv->srv_blob.server.cli_sock == -1) {
359: LOGERR;
360: return -1;
361: }
362: if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
363: LOGERR;
364: close(srv->srv_blob.server.cli_sock);
365: return -1;
366: }
367: if (bind(srv->srv_blob.server.cli_sock, &srv->srv_blob.server.cli_sa,
368: sizeof srv->srv_blob.server.cli_sa) == -1) {
369: LOGERR;
370: close(srv->srv_blob.server.cli_sock);
371: return -1;
372: }
373:
374: srv->srv_blob.clients = calloc(srv->srv_numcli, sizeof(rpc_cli_t));
375: if (!srv->srv_blob.clients) {
376: LOGERR;
377: close(srv->srv_blob.server.cli_sock);
378: return -1;
379: } else
380: memset(srv->srv_blob.clients, 0, srv->srv_numcli * sizeof(rpc_cli_t));
381:
382: pthread_mutex_init(&srv->srv_blob.mtx, NULL);
383:
384: pthread_mutex_lock(&srv->srv_mtx);
385: rpc_srv_registerCall(srv, NULL, CALL_BLOBSHUTDOWN, 0);
386: rpc_srv_registerCall(srv, NULL, CALL_BLOBCLIENTS, 0);
387: rpc_srv_registerCall(srv, NULL, CALL_BLOBVARS, 0);
1.1.1.1.2.13 misho 388: rpc_srv_registerCall(srv, NULL, CALL_BLOBSTATE, 1);
1.1.1.1.2.4 misho 389: pthread_mutex_unlock(&srv->srv_mtx);
390:
1.1.1.1.2.5 misho 391: srv->srv_blob.state = enable; // enable BLOB
1.1.1.1.2.4 misho 392: return 0;
393: }
394:
395: /*
396: * rpc_srv_endBLOBServer() Destroy BLOB server, close all opened sockets and free resources
397: * @srv = RPC Server instance
398: * return: none
399: */
400: void
401: rpc_srv_endBLOBServer(rpc_srv_t * __restrict srv)
402: {
403: rpc_cli_t *c;
404: register int i;
405: rpc_blob_t *f;
406:
407: if (!srv) {
408: rpc_SetErr(EINVAL, "Error:: Can`t destroy server because parameter is null!\n");
409: return;
410: } else
1.1.1.1.2.5 misho 411: srv->srv_blob.state = disable;
1.1.1.1.2.4 misho 412:
1.1.1.1.2.10 misho 413: rpc_srv_unregisterCall(srv, NULL, CALL_BLOBSHUTDOWN);
414: rpc_srv_unregisterCall(srv, NULL, CALL_BLOBCLIENTS);
415: rpc_srv_unregisterCall(srv, NULL, CALL_BLOBVARS);
1.1.1.1.2.13 misho 416: rpc_srv_unregisterCall(srv, NULL, CALL_BLOBSTATE);
1.1.1.1.2.10 misho 417:
1.1.1.1.2.4 misho 418: for (i = 0, c = srv->srv_blob.clients; i < srv->srv_numcli && c; i++, c++)
419: if (c->cli_sa.sa_family)
420: shutdown(c->cli_sock, SHUT_RDWR);
421: close(srv->srv_blob.server.cli_sock);
422:
1.1.1.1.2.13 misho 423: if (srv->srv_blob.clients) {
1.1.1.1.2.4 misho 424: free(srv->srv_blob.clients);
1.1.1.1.2.13 misho 425: srv->srv_blob.clients = NULL;
426: }
1.1.1.1.2.4 misho 427:
428: pthread_mutex_lock(&srv->srv_blob.mtx);
429: while ((f = srv->srv_blob.blobs)) {
430: srv->srv_blob.blobs = f->blob_next;
1.1.1.1.2.14 misho 431: rpc_srv_blobFree(srv, f);
1.1.1.1.2.4 misho 432: free(f);
433: }
434: pthread_mutex_unlock(&srv->srv_blob.mtx);
435:
1.1.1.1.2.20 misho 436: while (pthread_mutex_trylock(&srv->srv_blob.mtx) == EBUSY);
1.1.1.1.2.4 misho 437: pthread_mutex_destroy(&srv->srv_blob.mtx);
438: }
439:
440: /*
441: * rpc_srv_execBLOBServer() Execute Main BLOB server loop and wait for clients requests
442: * @srv = RPC Server instance
443: * return: -1 error or 0 ok, infinite loop ...
444: */
445: int
446: rpc_srv_execBLOBServer(rpc_srv_t * __restrict srv)
447: {
448: socklen_t salen = sizeof(struct sockaddr);
449: register int i;
450: rpc_cli_t *c;
451: fd_set fds;
452: int ret;
453: struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
454:
1.1.1.1.2.12 misho 455: if (!srv || srv->srv_blob.state == disable) {
1.1.1.1.2.4 misho 456: rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t start BLOB server ...\n");
457: return -1;
458: }
459:
460: if (listen(srv->srv_blob.server.cli_sock, SOMAXCONN) == -1) {
461: LOGERR;
462: return -1;
463: }
464:
1.1.1.1.2.11 misho 465: while (!blob_Kill && !rpc_Kill) {
1.1.1.1.2.4 misho 466: for (c = srv->srv_blob.clients, i = 0; i < srv->srv_numcli && c; i++, c++)
467: if (!c->cli_sa.sa_family)
468: break;
1.1.1.1.2.13 misho 469: if (i >= srv->srv_numcli) {
1.1.1.1.2.4 misho 470: usleep(1000000);
471: continue;
472: }
473:
474: FD_ZERO(&fds);
475: FD_SET(srv->srv_blob.server.cli_sock, &fds);
476: ret = select(srv->srv_blob.server.cli_sock + 1, &fds, NULL, NULL, &tv);
477: if (ret == -1) {
478: LOGERR;
479: ret = 1;
480: break;
481: }
482: if (!ret)
483: continue;
484:
485: c->cli_sock = accept(srv->srv_blob.server.cli_sock, &c->cli_sa, &salen);
486: if (c->cli_sock == -1) {
487: LOGERR;
488: continue;
489: } else
490: c->cli_parent = srv;
491:
492: if (pthread_create(&c->cli_tid, NULL, rpc_srv_dispatchVars, c)) {
493: LOGERR;
494: continue;
495: }
496: }
497:
1.1.1.1.2.11 misho 498: srv->srv_blob.state = disable;
499:
1.1.1.1.2.4 misho 500: return 0;
501: }
502:
503:
504: /*
1.1 misho 505: * rpc_srv_initServer() Init & create RPC Server
506: * @regProgID = ProgramID for authentication & recognition
507: * @regProcID = ProcessID for authentication & recognition
508: * @concurentClients = Concurent clients at same time to this server
509: * @family = Family socket type, AF_INET or AF_INET6
510: * @csHost = Host name or IP address for bind server, if NULL any address
511: * @Port = Port for bind server, if Port == 0 default port is selected
512: * return: NULL == error or !=NULL bind and created RPC server instance
513: */
514: rpc_srv_t *
515: rpc_srv_initServer(u_int regProgID, u_int regProcID, int concurentClients,
516: u_short family, const char *csHost, u_short Port)
517: {
518: rpc_srv_t *srv = NULL;
519: int n = 1;
520: struct hostent *host = NULL;
521: struct sockaddr_in sin;
522: struct sockaddr_in6 sin6;
523:
524: if (!concurentClients || !regProgID || (family != AF_INET && family != AF_INET6)) {
525: rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t init RPC server ...\n");
526: return NULL;
527: }
528: if (!Port)
529: Port = RPC_DEFPORT;
530: if (csHost) {
531: host = gethostbyname2(csHost, family);
532: if (!host) {
533: rpc_SetErr(h_errno, "Error:: %s\n", hstrerror(h_errno));
534: return NULL;
535: }
536: }
537: switch (family) {
538: case AF_INET:
539: memset(&sin, 0, sizeof sin);
540: sin.sin_len = sizeof sin;
541: sin.sin_family = family;
542: sin.sin_port = htons(Port);
543: if (csHost)
544: memcpy(&sin.sin_addr, host->h_addr, host->h_length);
545: break;
546: case AF_INET6:
547: memset(&sin6, 0, sizeof sin6);
548: sin6.sin6_len = sizeof sin6;
549: sin6.sin6_family = family;
550: sin6.sin6_port = htons(Port);
551: if (csHost)
552: memcpy(&sin6.sin6_addr, host->h_addr, host->h_length);
553: break;
554: default:
555: rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t start RPC server ...\n");
556: return NULL;
557: }
558:
559: srv = malloc(sizeof(rpc_srv_t));
560: if (!srv) {
561: LOGERR;
562: return NULL;
563: } else
564: memset(srv, 0, sizeof(rpc_srv_t));
565:
566: srv->srv_numcli = concurentClients;
567: srv->srv_session.sess_version = RPC_VERSION;
568: srv->srv_session.sess_program = regProgID;
569: srv->srv_session.sess_process = regProcID;
570:
1.1.1.1.2.2 misho 571: srv->srv_server.cli_tid = pthread_self();
1.1 misho 572: srv->srv_server.cli_parent = srv;
573: if (family == AF_INET)
574: memcpy(&srv->srv_server.cli_sa, &sin, sizeof srv->srv_server.cli_sa);
575: else
576: memcpy(&srv->srv_server.cli_sa, &sin6, sizeof srv->srv_server.cli_sa);
577: srv->srv_server.cli_sock = socket(family, SOCK_STREAM, 0);
578: if (srv->srv_server.cli_sock == -1) {
579: LOGERR;
580: free(srv);
581: return NULL;
582: }
583: if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) {
584: LOGERR;
585: close(srv->srv_server.cli_sock);
586: free(srv);
587: return NULL;
588: }
589: if (bind(srv->srv_server.cli_sock, &srv->srv_server.cli_sa, sizeof srv->srv_server.cli_sa) == -1) {
590: LOGERR;
591: close(srv->srv_server.cli_sock);
592: free(srv);
593: return NULL;
594: }
595:
596: srv->srv_clients = calloc(srv->srv_numcli, sizeof(rpc_cli_t));
597: if (!srv->srv_clients) {
598: LOGERR;
599: close(srv->srv_server.cli_sock);
600: free(srv);
601: return NULL;
602: } else
603: memset(srv->srv_clients, 0, srv->srv_numcli * sizeof(rpc_cli_t));
604:
1.1.1.1.2.21 misho 605: pthread_mutex_init(&srv->srv_mtx, NULL);
606:
1.1.1.1.2.1 misho 607: rpc_srv_registerCall(srv, NULL, CALL_SRVSHUTDOWN, 0);
1.1 misho 608: rpc_srv_registerCall(srv, NULL, CALL_SRVCLIENTS, 0);
609: rpc_srv_registerCall(srv, NULL, CALL_SRVCALLS, 0);
610: rpc_srv_registerCall(srv, NULL, CALL_SRVSESSIONS, 0);
611: return srv;
612: }
613:
614: /*
615: * rpc_srv_endServer() Destroy RPC server, close all opened sockets and free resources
616: * @srv = RPC Server instance
617: * return: none
618: */
619: void
620: rpc_srv_endServer(rpc_srv_t * __restrict srv)
621: {
622: rpc_cli_t *c;
623: register int i;
624: rpc_func_t *f;
625:
626: if (!srv) {
627: rpc_SetErr(EINVAL, "Error:: Can`t destroy server because parameter is null!\n");
628: return;
629: }
630:
1.1.1.1.2.4 misho 631: rpc_srv_endBLOBServer(srv);
1.1 misho 632:
633: for (i = 0, c = srv->srv_clients; i < srv->srv_numcli && c; i++, c++)
1.1.1.1.2.24 misho 634: if (c->cli_sa.sa_family) {
1.1 misho 635: shutdown(c->cli_sock, SHUT_RDWR);
1.1.1.1.2.24 misho 636: close(c->cli_sock);
637: }
1.1 misho 638: close(srv->srv_server.cli_sock);
639:
640: if (srv->srv_clients) {
641: free(srv->srv_clients);
1.1.1.1.2.13 misho 642: srv->srv_clients = NULL;
1.1 misho 643: srv->srv_numcli = 0;
644: }
645:
1.1.1.1.2.4 misho 646: pthread_mutex_lock(&srv->srv_mtx);
647: while ((f = srv->srv_funcs)) {
648: srv->srv_funcs = f->func_next;
649: free(f);
650: }
651: pthread_mutex_unlock(&srv->srv_mtx);
652:
1.1.1.1.2.20 misho 653: while (pthread_mutex_trylock(&srv->srv_mtx) == EBUSY);
1.1.1.1.2.4 misho 654: pthread_mutex_destroy(&srv->srv_mtx);
655:
1.1 misho 656: free(srv);
657: srv = NULL;
658: }
659:
660: /*
661: * rpc_srv_execServer() Execute Main server loop and wait for clients requests
662: * @srv = RPC Server instance
663: * return: -1 error or 0 ok, infinite loop ...
664: */
665: int
666: rpc_srv_execServer(rpc_srv_t * __restrict srv)
667: {
668: socklen_t salen = sizeof(struct sockaddr);
669: register int i;
670: rpc_cli_t *c;
1.1.1.1.2.2 misho 671: fd_set fds;
672: int ret;
673: struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
1.1 misho 674:
675: if (!srv) {
676: rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t start RPC server ...\n");
677: return -1;
678: }
679:
680: if (listen(srv->srv_server.cli_sock, SOMAXCONN) == -1) {
681: LOGERR;
682: return -1;
683: }
684:
1.1.1.1.2.1 misho 685: while (!rpc_Kill) {
1.1 misho 686: for (c = srv->srv_clients, i = 0; i < srv->srv_numcli && c; i++, c++)
687: if (!c->cli_sa.sa_family)
688: break;
1.1.1.1.2.13 misho 689: if (i >= srv->srv_numcli) {
1.1 misho 690: usleep(1000000);
691: continue;
692: }
1.1.1.1.2.2 misho 693:
694: FD_ZERO(&fds);
695: FD_SET(srv->srv_server.cli_sock, &fds);
696: ret = select(srv->srv_server.cli_sock + 1, &fds, NULL, NULL, &tv);
697: if (ret == -1) {
698: LOGERR;
699: ret = 1;
700: break;
701: }
702: if (!ret)
703: continue;
704:
1.1 misho 705: c->cli_sock = accept(srv->srv_server.cli_sock, &c->cli_sa, &salen);
706: if (c->cli_sock == -1) {
707: LOGERR;
708: continue;
709: } else
710: c->cli_parent = srv;
711:
712: if (pthread_create(&c->cli_tid, NULL, rpc_srv_dispatchCall, c)) {
713: LOGERR;
714: continue;
715: }
716: }
717:
718: return 0;
719: }
720:
721: // ---------------------------------------------------------
722:
723: /*
724: * rpc_srv_execCall() Execute registered call from RPC server
725: * @call = Register RPC call
726: * @rpc = IN RPC call structure
727: * @args = IN RPC call array of rpc values
728: * return: -1 error, !=-1 ok
729: */
730: int
1.1.1.1.2.11 misho 731: rpc_srv_execCall(rpc_func_t * __restrict call, struct tagRPCCall * __restrict rpc,
732: rpc_val_t * __restrict args)
1.1 misho 733: {
734: void *dl;
735: rpc_callback_t func;
736: int ret;
737:
1.1.1.1.2.11 misho 738: if (!call || !rpc || !call->func_parent) {
1.1 misho 739: rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t exec call from RPC server ...\n");
740: return -1;
741: }
742:
743: dl = dlopen((char*) (*call->func_file ? call->func_file : NULL), RTLD_NOW);
744: if (!dl) {
745: rpc_SetErr(ENOENT, "Error:: Can`t attach module %s!\n", dlerror());
746: return -1;
747: }
748:
749: func = dlsym(dl, (char*) call->func_name);
750: if (func)
1.1.1.1.2.11 misho 751: ret = func(call, rpc->call_argc, args);
1.1 misho 752: else {
753: rpc_SetErr(ENOEXEC, "Error:: Can`t find function %s!\n", dlerror());
754: ret = -1;
755: }
756:
757: dlclose(dl);
758: return ret;
759: }