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