|
version 1.1.1.1.2.4, 2010/06/23 11:33:38
|
version 1.16.2.3, 2013/07/15 14:54:40
|
|
Line 5
|
Line 5
|
| * $Author$ |
* $Author$ |
| * $Id$ |
* $Id$ |
| * |
* |
| *************************************************************************/ | ************************************************************************** |
| | The ELWIX and AITNET software is distributed under the following |
| | terms: |
| | |
| | All of the documentation and software included in the ELWIX and AITNET |
| | Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org> |
| | |
| | Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013 |
| | by Michael Pounov <misho@elwix.org>. All rights reserved. |
| | |
| | Redistribution and use in source and binary forms, with or without |
| | modification, are permitted provided that the following conditions |
| | are met: |
| | 1. Redistributions of source code must retain the above copyright |
| | notice, this list of conditions and the following disclaimer. |
| | 2. Redistributions in binary form must reproduce the above copyright |
| | notice, this list of conditions and the following disclaimer in the |
| | documentation and/or other materials provided with the distribution. |
| | 3. All advertising materials mentioning features or use of this software |
| | must display the following acknowledgement: |
| | This product includes software developed by Michael Pounov <misho@elwix.org> |
| | ELWIX - Embedded LightWeight unIX and its contributors. |
| | 4. Neither the name of AITNET nor the names of its contributors |
| | may be used to endorse or promote products derived from this software |
| | without specific prior written permission. |
| | |
| | THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND |
| | ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| | IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| | ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE |
| | FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| | DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
| | OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
| | HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
| | LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
| | OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
| | SUCH DAMAGE. |
| | */ |
| #include "global.h" |
#include "global.h" |
| |
|
| |
|
| static void * | /* SOCK_STREAM */ |
| rpc_srv_dispatchCall(void *arg) | static void *acceptClients(sched_task_t *); |
| | static void *closeClient(sched_task_t *); |
| | static void *rxPacket(sched_task_t *); |
| | static void *txPacket(sched_task_t *); |
| | |
| | /* SOCK_DGRAM */ |
| | static void *freeClient(sched_task_t *); |
| | static void *rxUDPPacket(sched_task_t *); |
| | static void *txUDPPacket(sched_task_t *); |
| | |
| | /* SOCK_RAW */ |
| | |
| | static sched_task_func_t cbProto[SOCK_RAW + 1][4] = { |
| | { acceptClients, closeClient, rxPacket, txPacket }, /* SOCK_STREAM */ |
| | { acceptClients, closeClient, rxPacket, txPacket }, /* SOCK_STREAM */ |
| | { rxUDPPacket, freeClient, rxUDPPacket, txUDPPacket }, /* SOCK_DGRAM */ |
| | { NULL, NULL, NULL, NULL } /* SOCK_RAW */ |
| | }; |
| | |
| | |
| | void |
| | rpc_freeCli(rpc_cli_t * __restrict c) |
| { |
{ |
| rpc_cli_t cli, *c = arg; | rpc_srv_t *s = c->cli_parent; |
| rpc_srv_t *s; | |
| rpc_val_t *vals, *v = NULL; | schedCancelby(s->srv_root, taskMAX, CRITERIA_ARG, c, NULL); |
| rpc_func_t *f; | |
| struct tagRPCCall *rpc; | /* free buffer */ |
| struct tagRPCRet rrpc; | AIT_FREE_VAL(&c->cli_buf); |
| fd_set fds; | |
| u_char buf[BUFSIZ], *data; | array_Del(s->srv_clients, c->cli_id, 0); |
| int ret, argc, Limit = 0; | if (c) |
| | e_free(c); |
| | } |
| | |
| | |
| | static inline int |
| | _check4freeslot(rpc_srv_t * __restrict srv, sockaddr_t * __restrict sa) |
| | { |
| | rpc_cli_t *c = NULL; |
| register int i; |
register int i; |
| |
|
| if (!arg) { | /* check free slots for connect */ |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t procced RPC client ...\n"); | for (i = 0; i < array_Size(srv->srv_clients) && |
| | (c = array(srv->srv_clients, i, rpc_cli_t*)); i++) |
| | /* check for duplicates */ |
| | if (sa && !e_addrcmp(&c->cli_sa, sa, 42)) |
| | break; |
| | if (i >= array_Size(srv->srv_clients)) |
| | return -1; /* no more free slots! */ |
| | |
| | return i; |
| | } |
| | |
| | static rpc_cli_t * |
| | _allocClient(rpc_srv_t * __restrict srv, sockaddr_t * __restrict sa) |
| | { |
| | rpc_cli_t *c = NULL; |
| | int n; |
| | |
| | n = _check4freeslot(srv, sa); |
| | if (n == -1) |
| return NULL; |
return NULL; |
| |
else |
| |
c = array(srv->srv_clients, n, rpc_cli_t*); |
| |
|
| |
if (!c) { |
| |
c = e_malloc(sizeof(rpc_cli_t)); |
| |
if (!c) { |
| |
LOGERR; |
| |
srv->srv_kill = 1; |
| |
return NULL; |
| |
} else { |
| |
memset(c, 0, sizeof(rpc_cli_t)); |
| |
array_Set(srv->srv_clients, n, c); |
| |
c->cli_id = n; |
| |
c->cli_parent = srv; |
| |
} |
| |
|
| |
/* alloc empty buffer */ |
| |
AIT_SET_BUFSIZ(&c->cli_buf, 0, srv->srv_netbuf); |
| |
} |
| |
|
| |
return c; |
| |
} |
| |
|
| |
|
| |
static void * |
| |
freeClient(sched_task_t *task) |
| |
{ |
| |
rpc_freeCli(TASK_ARG(task)); |
| |
|
| |
return NULL; |
| |
} |
| |
|
| |
static void * |
| |
closeClient(sched_task_t *task) |
| |
{ |
| |
int sock = ((rpc_cli_t*) TASK_ARG(task))->cli_sock; |
| |
|
| |
rpc_freeCli(TASK_ARG(task)); |
| |
|
| |
/* close client socket */ |
| |
shutdown(sock, SHUT_RDWR); |
| |
close(sock); |
| |
return NULL; |
| |
} |
| |
|
| |
static void * |
| |
txPacket(sched_task_t *task) |
| |
{ |
| |
rpc_cli_t *c = TASK_ARG(task); |
| |
rpc_srv_t *s = c->cli_parent; |
| |
rpc_func_t *f = NULL; |
| |
u_char buf[USHRT_MAX] = { 0 }; |
| |
struct tagRPCCall *rpc = (struct tagRPCCall*) buf; |
| |
int ret, wlen = sizeof(struct tagRPCCall); |
| |
|
| |
/* copy RPC header */ |
| |
memcpy(buf, TASK_DATA(task), wlen); |
| |
|
| |
if (rpc->call_argc) { |
| |
f = rpc_srv_getCall(s, ntohs(rpc->call_tag)); |
| |
if (!f) { |
| |
rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server"); |
| |
rpc->call_argc ^= rpc->call_argc; |
| |
rpc->call_rep.ret = RPC_ERROR(-1); |
| |
rpc->call_rep.eno = RPC_ERROR(rpc_Errno); |
| |
} else { |
| |
rpc->call_argc = htons(array_Size(RPC_RETVARS(c))); |
| |
/* Go Encapsulate variables */ |
| |
ret = ait_vars2buffer(buf + wlen, sizeof buf - wlen, RPC_RETVARS(c)); |
| |
/* Free return values */ |
| |
ait_freeVars(&c->cli_vars); |
| |
if (ret == -1) { |
| |
rpc_SetErr(EBADRPC, "Prepare RPC packet failed"); |
| |
rpc->call_argc ^= rpc->call_argc; |
| |
rpc->call_rep.ret = RPC_ERROR(-1); |
| |
rpc->call_rep.eno = RPC_ERROR(rpc_Errno); |
| |
} else |
| |
wlen += ret; |
| |
} |
| |
} |
| |
|
| |
rpc->call_len = htons(wlen); |
| |
|
| |
#if 0 |
| |
/* calculate CRC */ |
| |
rpc->call_crc ^= rpc->call_crc; |
| |
rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2)); |
| |
#endif |
| |
|
| |
/* send reply */ |
| |
ret = send(TASK_FD(task), buf, wlen, MSG_NOSIGNAL); |
| |
if (ret == -1 || ret != wlen) { |
| |
/* close connection */ |
| |
schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], |
| |
TASK_ARG(task), 0, NULL, 0); |
| |
} |
| |
|
| |
return NULL; |
| |
} |
| |
|
| |
static void * |
| |
execCall(sched_task_t *task) |
| |
{ |
| |
rpc_cli_t *c = TASK_ARG(task); |
| |
rpc_srv_t *s = c->cli_parent; |
| |
rpc_func_t *f = NULL; |
| |
array_t *arr = NULL; |
| |
u_char *buf = AIT_GET_BUF(&c->cli_buf) + TASK_VAL(task); |
| |
struct tagRPCCall *rpc = (struct tagRPCCall*) buf; |
| |
int argc = ntohs(rpc->call_argc); |
| |
|
| |
/* Go decapsulate variables ... */ |
| |
if (argc) { |
| |
arr = ait_buffer2vars(buf + sizeof(struct tagRPCCall), |
| |
AIT_LEN(&c->cli_buf) - TASK_VAL(task) - sizeof(struct tagRPCCall), |
| |
argc, 42); |
| |
if (!arr) { |
| |
rpc_SetErr(ERPCMISMATCH, "#%d - %s", elwix_GetErrno(), elwix_GetError()); |
| |
rpc->call_argc ^= rpc->call_argc; |
| |
rpc->call_rep.ret = RPC_ERROR(-1); |
| |
rpc->call_rep.eno = RPC_ERROR(rpc_Errno); |
| |
return NULL; |
| |
} |
| } else |
} else |
| s = c->cli_parent; | arr = NULL; |
| |
|
| do { | if (!(f = rpc_srv_getCall(s, ntohs(rpc->call_tag)))) { |
| FD_ZERO(&fds); | rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server"); |
| FD_SET(c->cli_sock, &fds); | rpc->call_argc ^= rpc->call_argc; |
| ret = select(c->cli_sock + 1, &fds, NULL, NULL, NULL); | rpc->call_rep.ret = RPC_ERROR(-1); |
| if (ret == -1) { | rpc->call_rep.eno = RPC_ERROR(rpc_Errno); |
| ret = -2; | } else { |
| | /* if client doesn't want reply */ |
| | argc = RPC_CHK_NOREPLY(rpc); |
| | rpc->call_rep.ret = RPC_ERROR(rpc_srv_execCall(c, rpc, f->func_name, arr)); |
| | if (rpc->call_rep.ret == htonl(-1)) { |
| | rpc->call_rep.eno = RPC_ERROR(errno); |
| | rpc->call_argc ^= rpc->call_argc; |
| | } else { |
| | rpc->call_rep.eno ^= rpc->call_rep.eno; |
| | if (argc) { |
| | /* without reply */ |
| | ait_freeVars(&c->cli_vars); |
| | rpc->call_argc ^= rpc->call_argc; |
| | } else { |
| | /* reply */ |
| | rpc->call_argc = htons(array_Size(RPC_RETVARS(c))); |
| | } |
| } |
} |
| memset(&rrpc, 0, sizeof rrpc); | } |
| memset(buf, 0, BUFSIZ); | |
| if ((ret = recv(c->cli_sock, buf, BUFSIZ, 0)) == -1) { | array_Destroy(&arr); |
| LOGERR; | return NULL; |
| ret = -3; | } |
| | |
| | static void * |
| | rxPacket(sched_task_t *task) |
| | { |
| | rpc_cli_t *c = TASK_ARG(task); |
| | rpc_srv_t *s = c->cli_parent; |
| | int len, rlen, noreply; |
| | u_short off = TASK_DATLEN(task); |
| | #if 0 |
| | u_short crc; |
| | #endif |
| | u_char *buf = AIT_GET_BUF(&c->cli_buf); |
| | struct tagRPCCall *rpc; |
| | |
| | if (!off) |
| | memset(buf, 0, AIT_LEN(&c->cli_buf)); |
| | rlen = recv(TASK_FD(task), buf + off, AIT_LEN(&c->cli_buf) - off, 0); |
| | if (rlen < 1) { |
| | /* close connection */ |
| | schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], |
| | TASK_ARG(task), 0, NULL, 0); |
| | return NULL; |
| | } else { |
| | rlen += off; /* add reminded bytes from previous rxPacket, if exists! */ |
| | off = 0; /* process buffer from start offset == 0 */ |
| | } |
| | |
| | do { |
| | /* check RPC packet */ |
| | if (rlen < sizeof(struct tagRPCCall)) { |
| | rpc_SetErr(ERPCMISMATCH, "Short RPC packet"); |
| | |
| | /* reminder received previous bytes ;) */ |
| | schedRead(TASK_ROOT(task), TASK_FUNC(task), TASK_ARG(task), |
| | TASK_FD(task), TASK_DATA(task), rlen); |
| | return NULL; |
| | } else |
| | rpc = (struct tagRPCCall*) (buf + off); |
| | |
| | len = ntohs(rpc->call_len); |
| | rlen -= len; |
| | |
| | /* check RPC packet lengths */ |
| | if (rlen < 0 || len < sizeof(struct tagRPCCall)) { |
| | rpc_SetErr(ERPCMISMATCH, "Broken RPC packet length"); |
| | /* skip entire packet */ |
| break; |
break; |
| } |
} |
| if (!ret) { // receive EOF | |
| ret = 0; | #if 0 |
| break; | /* check integrity of packet */ |
| | crc = ntohs(rpc->call_crc); |
| | rpc->call_crc ^= rpc->call_crc; |
| | if (crc != crcFletcher16((u_short*) rpc, len / 2)) { |
| | rpc_SetErr(ERPCMISMATCH, "Bad CRC RPC packet"); |
| | |
| | off += len; |
| | /* try next packet remaining into buffer */ |
| | continue; |
| } |
} |
| if (ret < sizeof(struct tagRPCCall)) { | #endif |
| rpc_SetErr(EMSGSIZE, "Error:: too short RPC packet ...\n"); | |
| ret = -4; | noreply = RPC_CHK_NOREPLY(rpc); |
| | |
| | /* check RPC packet session info */ |
| | if (rpc_chkPktSession(&rpc->call_session, &s->srv_session)) { |
| | rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session"); |
| | rpc->call_argc ^= rpc->call_argc; |
| | rpc->call_rep.ret = RPC_ERROR(-1); |
| | rpc->call_rep.eno = RPC_ERROR(errno); |
| | } else { |
| | /* execute RPC call */ |
| | schedEvent(TASK_ROOT(task), execCall, TASK_ARG(task), off, NULL, 0); |
| | } |
| | |
| | /* send RPC reply */ |
| | if (!noreply) |
| | schedWrite(TASK_ROOT(task), cbProto[s->srv_proto][CB_TXPACKET], |
| | TASK_ARG(task), TASK_FD(task), rpc, len); |
| | |
| | off += len; |
| | } while (rlen > 0); |
| | |
| | /* lets get next packet */ |
| | schedRead(TASK_ROOT(task), TASK_FUNC(task), TASK_ARG(task), TASK_FD(task), TASK_DATA(task), 0); |
| | return NULL; |
| | } |
| | |
| | static void * |
| | acceptClients(sched_task_t *task) |
| | { |
| | rpc_srv_t *srv = TASK_ARG(task); |
| | rpc_cli_t *c = NULL; |
| | socklen_t salen = sizeof(sockaddr_t); |
| | |
| | c = _allocClient(srv, NULL); |
| | if (!c) |
| | goto end; |
| | |
| | /* accept client */ |
| | c->cli_sock = accept(TASK_FD(task), &c->cli_sa.sa, &salen); |
| | if (c->cli_sock == -1) { |
| | LOGERR; |
| | AIT_FREE_VAL(&c->cli_buf); |
| | array_Del(srv->srv_clients, c->cli_id, 42); |
| | goto end; |
| | } else |
| | fcntl(c->cli_sock, F_SETFL, fcntl(c->cli_sock, F_GETFL) | O_NONBLOCK); |
| | |
| | schedRead(TASK_ROOT(task), cbProto[srv->srv_proto][CB_RXPACKET], c, |
| | c->cli_sock, NULL, 0); |
| | end: |
| | schedReadSelf(task); |
| | return NULL; |
| | } |
| | |
| | |
| | static void * |
| | txUDPPacket(sched_task_t *task) |
| | { |
| | rpc_cli_t *c = TASK_ARG(task); |
| | rpc_srv_t *s = c->cli_parent; |
| | rpc_func_t *f = NULL; |
| | u_char buf[USHRT_MAX] = { 0 }; |
| | struct tagRPCCall *rpc = (struct tagRPCCall*) buf; |
| | int ret, wlen = sizeof(struct tagRPCCall); |
| | struct timespec ts = { DEF_RPC_TIMEOUT, 0 }; |
| | |
| | schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, TASK_ARG(task), NULL); |
| | schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], |
| | TASK_ARG(task), ts, TASK_ARG(task), 0); |
| | |
| | /* copy RPC header */ |
| | memcpy(buf, TASK_DATA(task), wlen); |
| | |
| | if (rpc->call_argc) { |
| | f = rpc_srv_getCall(s, ntohs(rpc->call_tag)); |
| | if (!f) { |
| | rpc_SetErr(EPROGUNAVAIL, "Function not found at RPC server"); |
| | rpc->call_argc ^= rpc->call_argc; |
| | rpc->call_rep.ret = RPC_ERROR(-1); |
| | rpc->call_rep.eno = RPC_ERROR(rpc_Errno); |
| | } else { |
| | rpc->call_argc = htons(array_Size(RPC_RETVARS(c))); |
| | /* Go Encapsulate variables */ |
| | ret = ait_vars2buffer(buf + wlen, sizeof buf - wlen, RPC_RETVARS(c)); |
| | /* Free return values */ |
| | ait_freeVars(&c->cli_vars); |
| | if (ret == -1) { |
| | rpc_SetErr(EBADRPC, "Prepare RPC packet failed"); |
| | rpc->call_argc ^= rpc->call_argc; |
| | rpc->call_rep.ret = RPC_ERROR(-1); |
| | rpc->call_rep.eno = RPC_ERROR(rpc_Errno); |
| | } else |
| | wlen += ret; |
| | } |
| | } |
| | |
| | rpc->call_len = htons(wlen); |
| | |
| | /* calculate CRC */ |
| | rpc->call_crc ^= rpc->call_crc; |
| | rpc->call_crc = htons(crcFletcher16((u_short*) buf, wlen / 2)); |
| | |
| | /* send reply */ |
| | ret = sendto(TASK_FD(task), buf, wlen, MSG_NOSIGNAL, |
| | &c->cli_sa.sa, c->cli_sa.sa.sa_len); |
| | if (ret == -1 || ret != wlen) { |
| | /* close connection */ |
| | schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], |
| | TASK_ARG(task), 0, NULL, 0); |
| | } |
| | |
| | return NULL; |
| | } |
| | |
| | static void * |
| | rxUDPPacket(sched_task_t *task) |
| | { |
| | rpc_srv_t *srv = TASK_ARG(task); |
| | rpc_cli_t *c = NULL; |
| | int len, rlen, noreply; |
| | u_short crc, off = 0; |
| | u_char buf[USHRT_MAX + 1]; |
| | struct tagRPCCall *rpc = (struct tagRPCCall*) buf; |
| | sockaddr_t sa; |
| | socklen_t salen; |
| | struct timespec ts = { DEF_RPC_TIMEOUT, 0 }; |
| | |
| | /* receive connect packet */ |
| | salen = sa.ss.ss_len = sizeof(sockaddr_t); |
| | rlen = recvfrom(TASK_FD(task), buf, sizeof buf, 0, &sa.sa, &salen); |
| | if (rlen < 1) |
| | goto end; |
| | |
| | c = _allocClient(srv, &sa); |
| | if (!c) |
| | goto end; |
| | else { |
| | c->cli_sock = TASK_FD(task); |
| | memcpy(&c->cli_sa, &sa, sizeof c->cli_sa); |
| | memcpy(AIT_GET_BUF(&c->cli_buf), buf, AIT_LEN(&c->cli_buf)); |
| | /* armed timer for close stateless connection */ |
| | schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, c, NULL); |
| | schedTimer(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], |
| | c, ts, c, 0); |
| | } |
| | |
| | do { |
| | /* check RPC packet */ |
| | if (rlen < sizeof(struct tagRPCCall)) { |
| | rpc_SetErr(ERPCMISMATCH, "Short RPC packet"); |
| break; |
break; |
| } else |
} else |
| rpc = (struct tagRPCCall*) buf; | rpc = (struct tagRPCCall*) (AIT_GET_BUF(&c->cli_buf) + off); |
| // check RPC packet session info | |
| if (memcmp(&rpc->call_session, &s->srv_session, sizeof rpc->call_session)) { | len = ntohs(rpc->call_len); |
| rpc_SetErr(EINVAL, "Error:: get invalid RPC session ...\n"); | rlen -= len; |
| ret = -5; | |
| | /* check RPC packet lengths */ |
| | if (rlen < 0 || len < sizeof(struct tagRPCCall)) { |
| | rpc_SetErr(ERPCMISMATCH, "Broken RPC packet length"); |
| | /* skip entire packet */ |
| break; |
break; |
| } |
} |
| // RPC is OK! Go decapsulate variables ... | |
| if (rpc->call_argc) { | /* check integrity of packet */ |
| v = (rpc_val_t*) (buf + sizeof(struct tagRPCCall)); | crc = ntohs(rpc->call_crc); |
| // RPC received variables types OK! | rpc->call_crc ^= rpc->call_crc; |
| data = (u_char*) v + rpc->call_argc * sizeof(rpc_val_t); | if (crc != crcFletcher16((u_short*) rpc, len / 2)) { |
| for (i = 0; i < rpc->call_argc; i++) { | rpc_SetErr(ERPCMISMATCH, "Bad CRC RPC packet"); |
| switch (v[i].val_type) { | |
| case buffer: | off += len; |
| v[i].val.buffer = data; | /* try next packet remaining into buffer */ |
| data += v[i].val_len; | continue; |
| break; | |
| case string: | |
| v[i].val.string = (int8_t*) data; | |
| data += v[i].val_len + 1; | |
| break; | |
| case blob: | |
| rpc_srv_getBLOB(&v[i], data); | |
| data += sizeof(rpc_cli_t); | |
| break; | |
| default: | |
| break; | |
| } | |
| } | |
| } |
} |
| |
|
| argc = 0; | noreply = RPC_CHK_NOREPLY(rpc); |
| vals = NULL; | |
| if (!(f = rpc_srv_getCall(s, rpc->call_tag, rpc->call_hash))) { | |
| rpc_SetErr(EINVAL, "Error:: call not found into RPC server ...\n"); | |
| ret = -6; | |
| } else | |
| if ((ret = rpc_srv_execCall(s, f, rpc, v)) == -1) | |
| ret = -6; | |
| else | |
| argc = rpc_srv_getValsCall(f, &vals); | |
| |
|
| memcpy(&rrpc.ret_session, &rpc->call_session, sizeof rrpc.ret_session); | /* check RPC packet session info */ |
| rrpc.ret_tag = rpc->call_tag; | if (rpc_chkPktSession(&rpc->call_session, &srv->srv_session)) { |
| rrpc.ret_hash = rpc->call_hash; | rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session"); |
| rrpc.ret_errno = rpc_Errno; | rpc->call_argc ^= rpc->call_argc; |
| rrpc.ret_retcode = ret; | rpc->call_rep.ret = RPC_ERROR(-1); |
| rrpc.ret_argc = argc; | rpc->call_rep.eno = RPC_ERROR(errno); |
| | } else { |
| | /* execute RPC call */ |
| | schedEvent(TASK_ROOT(task), execCall, c, off, NULL, 0); |
| | } |
| |
|
| memset(buf, 0, BUFSIZ); | /* send RPC reply */ |
| memcpy(buf, &rrpc, (Limit = sizeof rrpc)); | if (!noreply) |
| if (argc && vals) { | schedWrite(TASK_ROOT(task), cbProto[srv->srv_proto][CB_TXPACKET], |
| v = (rpc_val_t*) (buf + sizeof rrpc); | c, TASK_FD(task), rpc, len); |
| memcpy(v, vals, argc * sizeof(rpc_val_t)); | |
| Limit += argc * sizeof(rpc_val_t); | |
| data = (u_char*) v + argc * sizeof(rpc_val_t); | |
| for (ret = i = 0; i < argc; i++) { | |
| switch (vals[i].val_type) { | |
| case buffer: | |
| if (ret || Limit + vals[i].val_len > BUFSIZ) { | |
| rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n"); | |
| rrpc.ret_retcode = ret = -7; | |
| rrpc.ret_argc = 0; | |
| break; | |
| } | |
| |
|
| memcpy(data, vals[i].val.buffer, vals[i].val_len); | off += len; |
| data += vals[i].val_len; | } while (rlen > 0); |
| Limit += vals[i].val_len; | |
| break; | |
| case string: | |
| if (ret || Limit + vals[i].val_len + 1 > BUFSIZ) { | |
| rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n"); | |
| rrpc.ret_retcode = ret = -7; | |
| rrpc.ret_argc = 0; | |
| break; | |
| } | |
| |
|
| memcpy(data, vals[i].val.string, vals[i].val_len + 1); | end: |
| data += vals[i].val_len + 1; | schedReadSelf(task); |
| Limit += vals[i].val_len + 1; | return NULL; |
| break; | } |
| case blob: | |
| if (ret || Limit + sizeof(rpc_cli_t) > BUFSIZ) { | |
| rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n"); | |
| rrpc.ret_retcode = ret = -7; | |
| rrpc.ret_argc = 0; | |
| break; | |
| } | |
| |
|
| memcpy(data, &cli, sizeof(rpc_cli_t)); | /* ------------------------------------------------------ */ |
| data += sizeof(rpc_cli_t); | |
| Limit += sizeof(rpc_cli_t); | |
| |
|
| rpc_srv_setBLOB(&vals[i], &cli); | void |
| break; | rpc_freeBLOBCli(rpc_cli_t * __restrict c) |
| default: | { |
| break; | rpc_srv_t *s = c->cli_parent; |
| } | |
| |
|
| RPC_FREE_VAL(&vals[i]); | schedCancelby(s->srv_blob.root, taskMAX, CRITERIA_ARG, c, NULL); |
| | |
| | /* free buffer */ |
| | AIT_FREE_VAL(&c->cli_buf); |
| | |
| | array_Del(s->srv_blob.clients, c->cli_id, 0); |
| | if (c) |
| | e_free(c); |
| | } |
| | |
| | |
| | static void * |
| | closeBLOBClient(sched_task_t *task) |
| | { |
| | int sock = ((rpc_cli_t*) TASK_ARG(task))->cli_sock; |
| | |
| | rpc_freeBLOBCli(TASK_ARG(task)); |
| | |
| | /* close client socket */ |
| | shutdown(sock, SHUT_RDWR); |
| | close(sock); |
| | return NULL; |
| | } |
| | |
| | static void * |
| | txBLOB(sched_task_t *task) |
| | { |
| | rpc_cli_t *c = TASK_ARG(task); |
| | u_char *buf = AIT_GET_BUF(&c->cli_buf); |
| | int wlen = sizeof(struct tagBLOBHdr); |
| | |
| | /* send reply */ |
| | wlen = send(TASK_FD(task), buf, wlen, MSG_NOSIGNAL); |
| | if (wlen == -1 || wlen != sizeof(struct tagBLOBHdr)) { |
| | /* close blob connection */ |
| | schedEvent(TASK_ROOT(task), closeBLOBClient, c, 42, NULL, 0); |
| | } |
| | |
| | return NULL; |
| | } |
| | |
| | static void * |
| | rxBLOB(sched_task_t *task) |
| | { |
| | rpc_cli_t *c = TASK_ARG(task); |
| | rpc_srv_t *s = c->cli_parent; |
| | rpc_blob_t *b; |
| | struct tagBLOBHdr blob; |
| | int rlen; |
| | |
| | memset(&blob, 0, sizeof blob); |
| | rlen = recv(TASK_FD(task), &blob, sizeof blob, 0); |
| | if (rlen < 1) { |
| | /* close blob connection */ |
| | schedEvent(TASK_ROOT(task), closeBLOBClient, c, 42, NULL, 0); |
| | return NULL; |
| | } |
| | |
| | /* check BLOB packet */ |
| | if (rlen < sizeof(struct tagBLOBHdr)) { |
| | rpc_SetErr(ERPCMISMATCH, "Short BLOB packet"); |
| | |
| | schedReadSelf(task); |
| | return NULL; |
| | } |
| | |
| | /* check RPC packet session info */ |
| | if (rpc_chkPktSession(&blob.hdr_session, &s->srv_session)) { |
| | rpc_SetErr(ERPCMISMATCH, "Get invalid RPC session"); |
| | blob.hdr_cmd = error; |
| | goto end; |
| | } |
| | |
| | /* Go to proceed packet ... */ |
| | switch (blob.hdr_cmd) { |
| | case get: |
| | if (!(b = rpc_srv_getBLOB(s, ntohl(blob.hdr_var)))) { |
| | rpc_SetErr(EINVAL, "Var=%x not found", ntohl(blob.hdr_var)); |
| | blob.hdr_cmd = no; |
| | blob.hdr_ret = RPC_ERROR(-1); |
| | break; |
| | } else |
| | blob.hdr_len = htonl(b->blob_len); |
| | |
| | if (rpc_srv_blobMap(s, b) != -1) { |
| | /* deliver BLOB variable to client */ |
| | blob.hdr_ret = htonl(rpc_srv_sendBLOB(c, b)); |
| | rpc_srv_blobUnmap(b); |
| | } else { |
| | blob.hdr_cmd = error; |
| | blob.hdr_ret = RPC_ERROR(-1); |
| } |
} |
| } | break; |
| | case set: |
| | if ((b = rpc_srv_registerBLOB(s, ntohl(blob.hdr_len), |
| | ntohl(blob.hdr_ret), task))) { |
| | /* set new BLOB variable for reply :) */ |
| | blob.hdr_var = htonl(b->blob_var); |
| |
|
| if ((ret = send(c->cli_sock, buf, Limit, 0)) == -1) { | /* receive BLOB from client */ |
| LOGERR; | blob.hdr_ret = htonl(rpc_srv_recvBLOB(c, b)); |
| ret = -8; | rpc_srv_blobUnmap(b); |
| | } else { |
| | blob.hdr_cmd = error; |
| | blob.hdr_ret = RPC_ERROR(-1); |
| | } |
| break; |
break; |
| } | case unset: |
| if (ret != Limit) { | if (rpc_srv_unregisterBLOB(s, ntohl(blob.hdr_var)) == -1) { |
| rpc_SetErr(EBADMSG, "Error:: in send RPC request, should be send %d bytes, " | blob.hdr_cmd = error; |
| "really is %d\n", Limit, ret); | blob.hdr_ret = RPC_ERROR(-1); |
| ret = -9; | } |
| | |
| | schedCancelby(TASK_ROOT(task), taskTIMER, CRITERIA_DATA, |
| | (void*) (intptr_t) ntohl(blob.hdr_var), NULL); |
| break; |
break; |
| } | default: |
| } while (ret > -1); | rpc_SetErr(EPROCUNAVAIL, "Unsupported BLOB command %d", blob.hdr_cmd); |
| | blob.hdr_cmd = error; |
| | blob.hdr_ret = RPC_ERROR(-1); |
| | } |
| |
|
| shutdown(c->cli_sock, SHUT_RDWR); | end: |
| close(c->cli_sock); | memcpy(AIT_ADDR(&c->cli_buf), &blob, sizeof blob); |
| memset(c, 0, sizeof(rpc_cli_t)); | schedWrite(TASK_ROOT(task), txBLOB, TASK_ARG(task), TASK_FD(task), NULL, 0); |
| return (void*) ret; | schedReadSelf(task); |
| | return NULL; |
| } |
} |
| |
|
| |
|
| static void * |
static void * |
| rpc_srv_dispatchVars(void *arg) | acceptBLOBClients(sched_task_t *task) |
| { |
{ |
| rpc_cli_t *c = arg; | rpc_srv_t *srv = TASK_ARG(task); |
| // rpc_srv_t *s; | rpc_cli_t *c = NULL; |
| int ret; | register int i; |
| | socklen_t salen = sizeof(sockaddr_t); |
| | #ifdef TCP_NOPUSH |
| | int n = 1; |
| | #endif |
| |
|
| return (void*) ret; | /* check free slots for connect */ |
| | for (i = 0; i < array_Size(srv->srv_blob.clients) && |
| | (c = array(srv->srv_blob.clients, i, rpc_cli_t*)); i++); |
| | if (c) /* no more free slots! */ |
| | goto end; |
| | c = e_malloc(sizeof(rpc_cli_t)); |
| | if (!c) { |
| | LOGERR; |
| | srv->srv_kill = srv->srv_blob.kill = 1; |
| | return NULL; |
| | } else { |
| | memset(c, 0, sizeof(rpc_cli_t)); |
| | array_Set(srv->srv_blob.clients, i, c); |
| | c->cli_id = i; |
| | c->cli_parent = srv; |
| | } |
| | |
| | /* alloc empty buffer */ |
| | AIT_SET_BUFSIZ(&c->cli_buf, 0, srv->srv_netbuf); |
| | |
| | /* accept client */ |
| | c->cli_sock = accept(TASK_FD(task), &c->cli_sa.sa, &salen); |
| | if (c->cli_sock == -1) { |
| | LOGERR; |
| | AIT_FREE_VAL(&c->cli_buf); |
| | array_Del(srv->srv_blob.clients, i, 42); |
| | goto end; |
| | } else { |
| | #ifdef TCP_NOPUSH |
| | setsockopt(c->cli_sock, IPPROTO_TCP, TCP_NOPUSH, &n, sizeof n); |
| | #endif |
| | fcntl(c->cli_sock, F_SETFL, fcntl(c->cli_sock, F_GETFL) | O_NONBLOCK); |
| | } |
| | |
| | schedRead(TASK_ROOT(task), rxBLOB, c, c->cli_sock, NULL, 0); |
| | end: |
| | schedReadSelf(task); |
| | return NULL; |
| } |
} |
| |
|
| // ------------------------------------------------- | /* ------------------------------------------------------ */ |
| |
|
| /* |
/* |
| * rpc_srv_initBLOBServer() Init & create BLOB Server | * rpc_srv_initBLOBServer() - Init & create BLOB Server |
| | * |
| | * @srv = RPC server instance |
| * @Port = Port for bind server, if Port == 0 default port is selected |
* @Port = Port for bind server, if Port == 0 default port is selected |
| |
* @diskDir = Disk place for BLOB file objects |
| * return: -1 == error or 0 bind and created BLOB server instance |
* return: -1 == error or 0 bind and created BLOB server instance |
| */ |
*/ |
| int |
int |
| rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_short Port) | rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_short Port, const char *diskDir) |
| { |
{ |
| int n = 1; |
int n = 1; |
| struct sockaddr_in sin; |
|
| struct sockaddr_in6 sin6; |
|
| |
|
| if (!srv) { | if (!srv || srv->srv_kill) { |
| rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t init BLOB server ...\n"); | rpc_SetErr(EINVAL, "Invalid parameters can`t init BLOB server"); |
| return -1; |
return -1; |
| } |
} |
| if (srv->srv_blob.state) { |
|
| rpc_SetErr(EPERM, "Warning:: Already started BLOB server!\n"); |
|
| return 0; |
|
| } |
|
| if (!Port) |
|
| Port = RPC_DEFPORT + 1; |
|
| |
|
| memset(&srv->srv_blob, 0, sizeof srv->srv_blob); |
memset(&srv->srv_blob, 0, sizeof srv->srv_blob); |
| srv->srv_blob.server.cli_tid = pthread_self(); | if (access(diskDir, R_OK | W_OK) == -1) { |
| | LOGERR; |
| | return -1; |
| | } else |
| | AIT_SET_STR(&srv->srv_blob.dir, diskDir); |
| | |
| | /* init blob list */ |
| | TAILQ_INIT(&srv->srv_blob.blobs); |
| | |
| srv->srv_blob.server.cli_parent = srv; |
srv->srv_blob.server.cli_parent = srv; |
| if (srv->srv_server.cli_sa.sa_family == AF_INET) { | |
| memcpy(&sin, &srv->srv_server.cli_sa, sizeof sin); | memcpy(&srv->srv_blob.server.cli_sa, &srv->srv_server.cli_sa, sizeof(sockaddr_t)); |
| sin.sin_port = htons(Port); | switch (srv->srv_blob.server.cli_sa.sa.sa_family) { |
| memcpy(&srv->srv_blob.server.cli_sa, &sin, sizeof(struct sockaddr)); | case AF_INET: |
| } else { | srv->srv_blob.server.cli_sa.sin.sin_port = |
| memcpy(&sin6, &srv->srv_server.cli_sa, sizeof sin6); | htons(Port ? Port : ntohs(srv->srv_blob.server.cli_sa.sin.sin_port) + 1); |
| sin6.sin6_port = htons(Port); | break; |
| memcpy(&srv->srv_blob.server.cli_sa, &sin6, sizeof(struct sockaddr)); | case AF_INET6: |
| | srv->srv_blob.server.cli_sa.sin6.sin6_port = |
| | htons(Port ? Port : ntohs(srv->srv_blob.server.cli_sa.sin6.sin6_port) + 1); |
| | break; |
| | case AF_LOCAL: |
| | strlcat(srv->srv_blob.server.cli_sa.sun.sun_path, ".blob", |
| | sizeof srv->srv_blob.server.cli_sa.sun.sun_path); |
| | break; |
| | default: |
| | AIT_FREE_VAL(&srv->srv_blob.dir); |
| | return -1; |
| } |
} |
| |
|
| srv->srv_blob.server.cli_sock = socket(srv->srv_server.cli_sa.sa_family, SOCK_STREAM, 0); | /* create BLOB server socket */ |
| | srv->srv_blob.server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, SOCK_STREAM, 0); |
| if (srv->srv_blob.server.cli_sock == -1) { |
if (srv->srv_blob.server.cli_sock == -1) { |
| LOGERR; |
LOGERR; |
| |
AIT_FREE_VAL(&srv->srv_blob.dir); |
| return -1; |
return -1; |
| } |
} |
| if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) { |
if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) { |
| LOGERR; |
LOGERR; |
| close(srv->srv_blob.server.cli_sock); |
close(srv->srv_blob.server.cli_sock); |
| |
AIT_FREE_VAL(&srv->srv_blob.dir); |
| return -1; |
return -1; |
| } |
} |
| if (bind(srv->srv_blob.server.cli_sock, &srv->srv_blob.server.cli_sa, | n = srv->srv_netbuf; |
| sizeof srv->srv_blob.server.cli_sa) == -1) { | if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) { |
| LOGERR; |
LOGERR; |
| close(srv->srv_blob.server.cli_sock); |
close(srv->srv_blob.server.cli_sock); |
| |
AIT_FREE_VAL(&srv->srv_blob.dir); |
| return -1; |
return -1; |
| } |
} |
| if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) { |
| srv->srv_blob.clients = calloc(srv->srv_numcli, sizeof(rpc_cli_t)); | |
| if (!srv->srv_blob.clients) { | |
| LOGERR; |
LOGERR; |
| close(srv->srv_blob.server.cli_sock); |
close(srv->srv_blob.server.cli_sock); |
| |
AIT_FREE_VAL(&srv->srv_blob.dir); |
| return -1; |
return -1; |
| |
} |
| |
if (bind(srv->srv_blob.server.cli_sock, &srv->srv_blob.server.cli_sa.sa, |
| |
srv->srv_blob.server.cli_sa.sa.sa_len) == -1) { |
| |
LOGERR; |
| |
close(srv->srv_blob.server.cli_sock); |
| |
AIT_FREE_VAL(&srv->srv_blob.dir); |
| |
return -1; |
| } else |
} else |
| memset(srv->srv_blob.clients, 0, srv->srv_numcli * sizeof(rpc_cli_t)); | fcntl(srv->srv_blob.server.cli_sock, F_SETFL, |
| | fcntl(srv->srv_blob.server.cli_sock, F_GETFL) | O_NONBLOCK); |
| |
|
| pthread_mutex_init(&srv->srv_blob.mtx, NULL); |
|
| |
|
| pthread_mutex_lock(&srv->srv_mtx); | /* allocate pool for concurent blob clients */ |
| rpc_srv_registerCall(srv, NULL, CALL_BLOBSHUTDOWN, 0); | srv->srv_blob.clients = array_Init(array_Size(srv->srv_clients)); |
| rpc_srv_registerCall(srv, NULL, CALL_BLOBCLIENTS, 0); | if (!srv->srv_blob.clients) { |
| rpc_srv_registerCall(srv, NULL, CALL_BLOBVARS, 0); | rpc_SetErr(elwix_GetErrno(), "%s", elwix_GetError()); |
| pthread_mutex_unlock(&srv->srv_mtx); | close(srv->srv_blob.server.cli_sock); |
| | AIT_FREE_VAL(&srv->srv_blob.dir); |
| | return -1; |
| | } |
| |
|
| srv->srv_blob.state = 1; // enable BLOB | /* init blob scheduler */ |
| | srv->srv_blob.root = schedBegin(); |
| | if (!srv->srv_blob.root) { |
| | rpc_SetErr(sched_GetErrno(), "%s", sched_GetError()); |
| | array_Destroy(&srv->srv_blob.clients); |
| | close(srv->srv_blob.server.cli_sock); |
| | AIT_FREE_VAL(&srv->srv_blob.dir); |
| | return -1; |
| | } |
| | |
| return 0; |
return 0; |
| } |
} |
| |
|
| /* |
/* |
| * rpc_srv_endBLOBServer() Destroy BLOB server, close all opened sockets and free resources | * rpc_srv_endBLOBServer() - Destroy BLOB server, close all opened sockets and free resources |
| | * |
| * @srv = RPC Server instance |
* @srv = RPC Server instance |
| * return: none |
* return: none |
| */ |
*/ |
| void |
void |
| rpc_srv_endBLOBServer(rpc_srv_t * __restrict srv) |
rpc_srv_endBLOBServer(rpc_srv_t * __restrict srv) |
| { |
{ |
| rpc_cli_t *c; | if (!srv) |
| register int i; | |
| rpc_blob_t *f; | |
| |
| if (!srv) { | |
| rpc_SetErr(EINVAL, "Error:: Can`t destroy server because parameter is null!\n"); | |
| return; |
return; |
| } else |
|
| srv->srv_blob.state = 0; |
|
| |
|
| for (i = 0, c = srv->srv_blob.clients; i < srv->srv_numcli && c; i++, c++) | srv->srv_blob.kill = 1; |
| if (c->cli_sa.sa_family) | |
| shutdown(c->cli_sock, SHUT_RDWR); | |
| close(srv->srv_blob.server.cli_sock); | |
| |
| if (srv->srv_blob.clients) | |
| free(srv->srv_blob.clients); | |
| |
| pthread_mutex_lock(&srv->srv_blob.mtx); | |
| while ((f = srv->srv_blob.blobs)) { | |
| srv->srv_blob.blobs = f->blob_next; | |
| free(f); | |
| } | |
| pthread_mutex_unlock(&srv->srv_blob.mtx); | |
| |
| pthread_mutex_destroy(&srv->srv_blob.mtx); | |
| } |
} |
| |
|
| /* |
/* |
| * rpc_srv_execBLOBServer() Execute Main BLOB server loop and wait for clients requests | * rpc_srv_loopBLOBServer() - Execute Main BLOB server loop and wait for clients requests |
| | * |
| * @srv = RPC Server instance |
* @srv = RPC Server instance |
| * return: -1 error or 0 ok, infinite loop ... |
* return: -1 error or 0 ok, infinite loop ... |
| */ |
*/ |
| int |
int |
| rpc_srv_execBLOBServer(rpc_srv_t * __restrict srv) | rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv) |
| { |
{ |
| socklen_t salen = sizeof(struct sockaddr); |
|
| register int i; |
|
| rpc_cli_t *c; |
rpc_cli_t *c; |
| fd_set fds; | register int i; |
| int ret; | rpc_blob_t *b, *tmp; |
| struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; | struct timespec ts = { RPC_SCHED_POLLING, 0 }; |
| |
|
| if (!srv || !srv->srv_blob.state) { | if (!srv || srv->srv_kill) { |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t start BLOB server ...\n"); | rpc_SetErr(EINVAL, "Invalid parameter can`t start BLOB server"); |
| return -1; |
return -1; |
| } |
} |
| |
|
| if (listen(srv->srv_blob.server.cli_sock, SOMAXCONN) == -1) { | if (listen(srv->srv_blob.server.cli_sock, array_Size(srv->srv_blob.clients)) == -1) { |
| LOGERR; |
LOGERR; |
| return -1; |
return -1; |
| } |
} |
| |
|
| while (!rpc_Kill) { | if (!schedRead(srv->srv_blob.root, acceptBLOBClients, srv, |
| for (c = srv->srv_blob.clients, i = 0; i < srv->srv_numcli && c; i++, c++) | srv->srv_blob.server.cli_sock, NULL, 0)) { |
| if (!c->cli_sa.sa_family) | rpc_SetErr(sched_GetErrno(), "%s", sched_GetError()); |
| break; | return -1; |
| if (c && c->cli_sa.sa_family && c->cli_parent) { | } |
| usleep(1000000); | |
| continue; | |
| } | |
| |
|
| FD_ZERO(&fds); | schedPolling(srv->srv_blob.root, &ts, NULL); |
| FD_SET(srv->srv_blob.server.cli_sock, &fds); | /* main rpc loop */ |
| ret = select(srv->srv_blob.server.cli_sock + 1, &fds, NULL, NULL, &tv); | schedRun(srv->srv_blob.root, &srv->srv_blob.kill); |
| if (ret == -1) { | |
| LOGERR; | |
| ret = 1; | |
| break; | |
| } | |
| if (!ret) | |
| continue; | |
| |
|
| c->cli_sock = accept(srv->srv_blob.server.cli_sock, &c->cli_sa, &salen); | /* close all clients connections & server socket */ |
| if (c->cli_sock == -1) { | for (i = 0; i < array_Size(srv->srv_blob.clients); i++) { |
| LOGERR; | c = array(srv->srv_blob.clients, i, rpc_cli_t*); |
| continue; | if (c) { |
| } else | shutdown(c->cli_sock, SHUT_RDWR); |
| c->cli_parent = srv; | close(c->cli_sock); |
| |
|
| if (pthread_create(&c->cli_tid, NULL, rpc_srv_dispatchVars, c)) { | schedCancelby(srv->srv_blob.root, taskMAX, CRITERIA_ARG, c, NULL); |
| LOGERR; | AIT_FREE_VAL(&c->cli_buf); |
| continue; | |
| } |
} |
| |
array_Del(srv->srv_blob.clients, i, 42); |
| } |
} |
| |
array_Destroy(&srv->srv_blob.clients); |
| |
|
| |
close(srv->srv_blob.server.cli_sock); |
| |
|
| |
/* detach blobs */ |
| |
TAILQ_FOREACH_SAFE(b, &srv->srv_blob.blobs, blob_node, tmp) { |
| |
TAILQ_REMOVE(&srv->srv_blob.blobs, b, blob_node); |
| |
|
| |
rpc_srv_blobFree(srv, b); |
| |
e_free(b); |
| |
} |
| |
|
| |
schedEnd(&srv->srv_blob.root); |
| |
AIT_FREE_VAL(&srv->srv_blob.dir); |
| return 0; |
return 0; |
| } |
} |
| |
|
| |
|
| /* |
/* |
| * rpc_srv_initServer() Init & create RPC Server | * rpc_srv_initServer() - Init & create RPC Server |
| * @regProgID = ProgramID for authentication & recognition | * |
| * @regProcID = ProcessID for authentication & recognition | * @InstID = Instance for authentication & recognition |
| * @concurentClients = Concurent clients at same time to this server |
* @concurentClients = Concurent clients at same time to this server |
| * @family = Family socket type, AF_INET or AF_INET6 | * @netBuf = Network buffer length (min:512 bytes), if =0 == BUFSIZ (also meaning max RPC packet) |
| * @csHost = Host name or IP address for bind server, if NULL any address | * @csHost = Host name or address for bind server, if NULL any address |
| * @Port = Port for bind server, if Port == 0 default port is selected |
* @Port = Port for bind server, if Port == 0 default port is selected |
| |
* @proto = Protocol, if == 0 choose SOCK_STREAM |
| * return: NULL == error or !=NULL bind and created RPC server instance |
* return: NULL == error or !=NULL bind and created RPC server instance |
| */ |
*/ |
| rpc_srv_t * |
rpc_srv_t * |
| rpc_srv_initServer(u_int regProgID, u_int regProcID, int concurentClients, | rpc_srv_initServer(u_char InstID, int concurentClients, int netBuf, |
| u_short family, const char *csHost, u_short Port) | const char *csHost, u_short Port, int proto) |
| { |
{ |
| rpc_srv_t *srv = NULL; |
|
| int n = 1; |
int n = 1; |
| struct hostent *host = NULL; | rpc_srv_t *srv = NULL; |
| struct sockaddr_in sin; | sockaddr_t sa = E_SOCKADDR_INIT; |
| struct sockaddr_in6 sin6; | |
| |
|
| if (!concurentClients || !regProgID || (family != AF_INET && family != AF_INET6)) { | if (!concurentClients || (proto < 0 || proto > SOCK_DGRAM)) { |
| rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t init RPC server ...\n"); | rpc_SetErr(EINVAL, "Invalid parameters can`t init RPC server"); |
| return NULL; |
return NULL; |
| } |
} |
| |
if (!e_gethostbyname(csHost, Port, &sa)) |
| |
return NULL; |
| if (!Port) |
if (!Port) |
| Port = RPC_DEFPORT; |
Port = RPC_DEFPORT; |
| if (csHost) { | if (!proto) |
| host = gethostbyname2(csHost, family); | proto = SOCK_STREAM; |
| if (!host) { | if (netBuf < RPC_MIN_BUFSIZ) |
| rpc_SetErr(h_errno, "Error:: %s\n", hstrerror(h_errno)); | netBuf = BUFSIZ; |
| return NULL; | else |
| } | netBuf = E_ALIGN(netBuf, 2); /* align netBuf length */ |
| } | |
| switch (family) { | |
| case AF_INET: | |
| memset(&sin, 0, sizeof sin); | |
| sin.sin_len = sizeof sin; | |
| sin.sin_family = family; | |
| sin.sin_port = htons(Port); | |
| if (csHost) | |
| memcpy(&sin.sin_addr, host->h_addr, host->h_length); | |
| break; | |
| case AF_INET6: | |
| memset(&sin6, 0, sizeof sin6); | |
| sin6.sin6_len = sizeof sin6; | |
| sin6.sin6_family = family; | |
| sin6.sin6_port = htons(Port); | |
| if (csHost) | |
| memcpy(&sin6.sin6_addr, host->h_addr, host->h_length); | |
| break; | |
| default: | |
| rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t start RPC server ...\n"); | |
| return NULL; | |
| } | |
| |
|
| srv = malloc(sizeof(rpc_srv_t)); | #ifdef HAVE_SRANDOMDEV |
| | srandomdev(); |
| | #else |
| | time_t tim; |
| | |
| | srandom((time(&tim) ^ getpid())); |
| | #endif |
| | |
| | srv = e_malloc(sizeof(rpc_srv_t)); |
| if (!srv) { |
if (!srv) { |
| LOGERR; |
LOGERR; |
| return NULL; |
return NULL; |
| } else |
} else |
| memset(srv, 0, sizeof(rpc_srv_t)); |
memset(srv, 0, sizeof(rpc_srv_t)); |
| |
|
| srv->srv_numcli = concurentClients; | srv->srv_proto = proto; |
| | srv->srv_netbuf = netBuf; |
| srv->srv_session.sess_version = RPC_VERSION; |
srv->srv_session.sess_version = RPC_VERSION; |
| srv->srv_session.sess_program = regProgID; | srv->srv_session.sess_instance = InstID; |
| srv->srv_session.sess_process = regProcID; | |
| |
|
| srv->srv_server.cli_tid = pthread_self(); |
|
| srv->srv_server.cli_parent = srv; |
srv->srv_server.cli_parent = srv; |
| if (family == AF_INET) | memcpy(&srv->srv_server.cli_sa, &sa, sizeof srv->srv_server.cli_sa); |
| memcpy(&srv->srv_server.cli_sa, &sin, sizeof srv->srv_server.cli_sa); | |
| else | /* init functions */ |
| memcpy(&srv->srv_server.cli_sa, &sin6, sizeof srv->srv_server.cli_sa); | pthread_mutex_init(&srv->srv_funcs.mtx, NULL); |
| srv->srv_server.cli_sock = socket(family, SOCK_STREAM, 0); | SLIST_INIT(&srv->srv_funcs); |
| | AVL_INIT(&srv->srv_funcs); |
| | |
| | /* init scheduler */ |
| | srv->srv_root = schedBegin(); |
| | if (!srv->srv_root) { |
| | rpc_SetErr(sched_GetErrno(), "%s", sched_GetError()); |
| | pthread_mutex_destroy(&srv->srv_funcs.mtx); |
| | e_free(srv); |
| | return NULL; |
| | } |
| | |
| | /* init pool for clients */ |
| | srv->srv_clients = array_Init(concurentClients); |
| | if (!srv->srv_clients) { |
| | rpc_SetErr(elwix_GetErrno(), "%s", elwix_GetError()); |
| | schedEnd(&srv->srv_root); |
| | pthread_mutex_destroy(&srv->srv_funcs.mtx); |
| | e_free(srv); |
| | return NULL; |
| | } |
| | |
| | /* create server socket */ |
| | srv->srv_server.cli_sock = socket(srv->srv_server.cli_sa.sa.sa_family, srv->srv_proto, 0); |
| if (srv->srv_server.cli_sock == -1) { |
if (srv->srv_server.cli_sock == -1) { |
| LOGERR; |
LOGERR; |
| free(srv); | array_Destroy(&srv->srv_clients); |
| | schedEnd(&srv->srv_root); |
| | pthread_mutex_destroy(&srv->srv_funcs.mtx); |
| | e_free(srv); |
| return NULL; |
return NULL; |
| } |
} |
| if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) { |
if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) { |
| LOGERR; |
LOGERR; |
| close(srv->srv_server.cli_sock); | goto err; |
| free(srv); | |
| return NULL; | |
| } |
} |
| if (bind(srv->srv_server.cli_sock, &srv->srv_server.cli_sa, sizeof srv->srv_server.cli_sa) == -1) { | n = srv->srv_netbuf; |
| | if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_SNDBUF, &n, sizeof n) == -1) { |
| LOGERR; |
LOGERR; |
| close(srv->srv_server.cli_sock); | goto err; |
| free(srv); | |
| return NULL; | |
| } |
} |
| if (setsockopt(srv->srv_server.cli_sock, SOL_SOCKET, SO_RCVBUF, &n, sizeof n) == -1) { |
| srv->srv_clients = calloc(srv->srv_numcli, sizeof(rpc_cli_t)); | |
| if (!srv->srv_clients) { | |
| LOGERR; |
LOGERR; |
| close(srv->srv_server.cli_sock); | goto err; |
| free(srv); | } |
| return NULL; | if (bind(srv->srv_server.cli_sock, &srv->srv_server.cli_sa.sa, |
| | srv->srv_server.cli_sa.sa.sa_len) == -1) { |
| | LOGERR; |
| | goto err; |
| } else |
} else |
| memset(srv->srv_clients, 0, srv->srv_numcli * sizeof(rpc_cli_t)); | fcntl(srv->srv_server.cli_sock, F_SETFL, |
| | fcntl(srv->srv_server.cli_sock, F_GETFL) | O_NONBLOCK); |
| |
|
| rpc_srv_registerCall(srv, NULL, CALL_SRVSHUTDOWN, 0); | rpc_register_srvPing(srv); |
| rpc_srv_registerCall(srv, NULL, CALL_SRVCLIENTS, 0); | |
| rpc_srv_registerCall(srv, NULL, CALL_SRVCALLS, 0); | |
| rpc_srv_registerCall(srv, NULL, CALL_SRVSESSIONS, 0); | |
| |
|
| pthread_mutex_init(&srv->srv_mtx, NULL); |
|
| return srv; |
return srv; |
| |
err: /* error condition */ |
| |
close(srv->srv_server.cli_sock); |
| |
array_Destroy(&srv->srv_clients); |
| |
schedEnd(&srv->srv_root); |
| |
pthread_mutex_destroy(&srv->srv_funcs.mtx); |
| |
e_free(srv); |
| |
return NULL; |
| } |
} |
| |
|
| /* |
/* |
| * rpc_srv_endServer() Destroy RPC server, close all opened sockets and free resources | * rpc_srv_endServer() - Destroy RPC server, close all opened sockets and free resources |
| * @srv = RPC Server instance | * |
| | * @psrv = RPC Server instance |
| * return: none |
* return: none |
| */ |
*/ |
| void |
void |
| rpc_srv_endServer(rpc_srv_t * __restrict srv) | rpc_srv_endServer(rpc_srv_t ** __restrict psrv) |
| { |
{ |
| rpc_cli_t *c; | if (!psrv || !*psrv) |
| register int i; | |
| rpc_func_t *f; | |
| |
| if (!srv) { | |
| rpc_SetErr(EINVAL, "Error:: Can`t destroy server because parameter is null!\n"); | |
| return; |
return; |
| } |
|
| |
|
| rpc_srv_endBLOBServer(srv); | /* if send kill to blob server */ |
| | if (!(*psrv)->srv_blob.kill) |
| | rpc_srv_endBLOBServer(*psrv); |
| |
|
| for (i = 0, c = srv->srv_clients; i < srv->srv_numcli && c; i++, c++) | (*psrv)->srv_kill = 1; |
| if (c->cli_sa.sa_family) | sleep(RPC_SCHED_POLLING); |
| shutdown(c->cli_sock, SHUT_RDWR); | |
| close(srv->srv_server.cli_sock); | |
| |
|
| if (srv->srv_clients) { | pthread_mutex_destroy(&(*psrv)->srv_funcs.mtx); |
| free(srv->srv_clients); | e_free(*psrv); |
| srv->srv_numcli = 0; | *psrv = NULL; |
| } | |
| |
| pthread_mutex_lock(&srv->srv_mtx); | |
| while ((f = srv->srv_funcs)) { | |
| srv->srv_funcs = f->func_next; | |
| free(f); | |
| } | |
| pthread_mutex_unlock(&srv->srv_mtx); | |
| |
| pthread_mutex_destroy(&srv->srv_mtx); | |
| |
| free(srv); | |
| srv = NULL; | |
| } |
} |
| |
|
| /* |
/* |
| * rpc_srv_execServer() Execute Main server loop and wait for clients requests | * rpc_srv_loopServer() - Execute Main server loop and wait for clients requests |
| | * |
| * @srv = RPC Server instance |
* @srv = RPC Server instance |
| * return: -1 error or 0 ok, infinite loop ... |
* return: -1 error or 0 ok, infinite loop ... |
| */ |
*/ |
| int |
int |
| rpc_srv_execServer(rpc_srv_t * __restrict srv) | rpc_srv_loopServer(rpc_srv_t * __restrict srv) |
| { |
{ |
| socklen_t salen = sizeof(struct sockaddr); |
|
| register int i; |
|
| rpc_cli_t *c; |
rpc_cli_t *c; |
| fd_set fds; | register int i; |
| int ret; | rpc_func_t *f; |
| struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; | struct timespec ts = { RPC_SCHED_POLLING, 0 }; |
| |
|
| if (!srv) { |
if (!srv) { |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t start RPC server ...\n"); | rpc_SetErr(EINVAL, "Invalid parameter can`t start RPC server"); |
| return -1; |
return -1; |
| } |
} |
| |
|
| if (listen(srv->srv_server.cli_sock, SOMAXCONN) == -1) { | if (srv->srv_proto == SOCK_STREAM) |
| LOGERR; | if (listen(srv->srv_server.cli_sock, array_Size(srv->srv_clients)) == -1) { |
| return -1; | |
| } | |
| |
| while (!rpc_Kill) { | |
| for (c = srv->srv_clients, i = 0; i < srv->srv_numcli && c; i++, c++) | |
| if (!c->cli_sa.sa_family) | |
| break; | |
| if (c && c->cli_sa.sa_family && c->cli_parent) { | |
| usleep(1000000); | |
| continue; | |
| } | |
| |
| FD_ZERO(&fds); | |
| FD_SET(srv->srv_server.cli_sock, &fds); | |
| ret = select(srv->srv_server.cli_sock + 1, &fds, NULL, NULL, &tv); | |
| if (ret == -1) { | |
| LOGERR; |
LOGERR; |
| ret = 1; | return -1; |
| break; | |
| } |
} |
| if (!ret) |
|
| continue; |
|
| |
|
| c->cli_sock = accept(srv->srv_server.cli_sock, &c->cli_sa, &salen); | if (!schedRead(srv->srv_root, cbProto[srv->srv_proto][CB_ACCEPTCLIENT], srv, |
| if (c->cli_sock == -1) { | srv->srv_server.cli_sock, NULL, 0)) { |
| LOGERR; | rpc_SetErr(sched_GetErrno(), "%s", sched_GetError()); |
| continue; | return -1; |
| } else | |
| c->cli_parent = srv; | |
| |
| if (pthread_create(&c->cli_tid, NULL, rpc_srv_dispatchCall, c)) { | |
| LOGERR; | |
| continue; | |
| } | |
| } |
} |
| |
|
| return 0; | schedPolling(srv->srv_root, &ts, NULL); |
| } | /* main rpc loop */ |
| | schedRun(srv->srv_root, &srv->srv_kill); |
| |
|
| // --------------------------------------------------------- | /* close all clients connections & server socket */ |
| | for (i = 0; i < array_Size(srv->srv_clients); i++) { |
| | c = array(srv->srv_clients, i, rpc_cli_t*); |
| | if (c) { |
| | shutdown(c->cli_sock, SHUT_RDWR); |
| | close(c->cli_sock); |
| |
|
| /* | schedCancelby(srv->srv_root, taskMAX, CRITERIA_ARG, c, NULL); |
| * rpc_srv_freeValsCall() Free return variables for RPC call | ait_freeVars(&RPC_RETVARS(c)); |
| * @call = RPC function call | AIT_FREE_VAL(&c->cli_buf); |
| * return: none | |
| */ | |
| inline void | |
| rpc_srv_freeValsCall(rpc_func_t * __restrict call) | |
| { | |
| rpc_srv_declValsCall(call, 0); | |
| } | |
| |
| /* | |
| * rpc_srv_declValsCall() Declare return variables for RPC call | |
| * @call = RPC function call | |
| * @return_vals = Number of return variables | |
| * return: -1 error, !=-1 ok | |
| */ | |
| inline int | |
| rpc_srv_declValsCall(rpc_func_t * __restrict call, int return_vals) | |
| { | |
| void *ptr; | |
| |
| if (!call || return_vals < 0) { | |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t declare return variables for RPC call...\n"); | |
| return -1; | |
| } else | |
| call->func_args = return_vals; | |
| |
| if (!return_vals) { | |
| if (call->func_vals) { | |
| free(call->func_vals); | |
| call->func_vals = NULL; | |
| } |
} |
| } else { | array_Del(srv->srv_clients, i, 42); |
| ptr = realloc(call->func_vals, return_vals * sizeof(rpc_val_t)); | |
| if (!ptr) { | |
| LOGERR; | |
| return -1; | |
| } else | |
| call->func_vals = ptr; | |
| } |
} |
| |
array_Destroy(&srv->srv_clients); |
| |
|
| return call->func_args; | close(srv->srv_server.cli_sock); |
| } | |
| |
|
| /* | /* detach exported calls */ |
| * rpc_srv_delValsCall() Clean values from return variables of RPC call | RPC_FUNCS_LOCK(&srv->srv_funcs); |
| * @call = RPC function call | while ((f = SLIST_FIRST(&srv->srv_funcs))) { |
| * return: -1 error, !=-1 Returned number of cleaned RPC variables | SLIST_REMOVE_HEAD(&srv->srv_funcs, func_next); |
| */ | |
| inline int | |
| rpc_srv_delValsCall(rpc_func_t * __restrict call) | |
| { | |
| if (!call) { | |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t delete return variables ...\n"); | |
| return -1; | |
| } | |
| |
|
| memset(call->func_vals, 0, call->func_args * sizeof(rpc_val_t)); | AIT_FREE_VAL(&f->func_name); |
| return call->func_args; | e_free(f); |
| } | |
| |
| /* | |
| * rpc_srv_copyValsCall() Copy return variables for RPC call to new variable | |
| * @call = RPC function call | |
| * @newvals = New allocated variables array, must be free after use | |
| * return: -1 error, !=-1 Returned number of copied RPC variables | |
| */ | |
| inline int | |
| rpc_srv_copyValsCall(rpc_func_t * __restrict call, rpc_val_t ** __restrict newvals) | |
| { | |
| if (!call || !newvals) { | |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t copy return variables to new array\n"); | |
| return -1; | |
| } |
} |
| |
srv->srv_funcs.avlh_root = NULL; |
| |
RPC_FUNCS_UNLOCK(&srv->srv_funcs); |
| |
|
| *newvals = calloc(call->func_args, sizeof(rpc_val_t)); | schedEnd(&srv->srv_root); |
| if (!*newvals) { | |
| LOGERR; | |
| return -1; | |
| } else | |
| memcpy(*newvals, call->func_vals, call->func_args * sizeof(rpc_val_t)); | |
| |
| return call->func_args; | |
| } | |
| |
| /* | |
| * rpc_srv_getValsCall() Get return variables for RPC call | |
| * @call = RPC function call | |
| * @vals = Returned variables, may be NULL | |
| * return: -1 error, !=-1 Number of returned variables | |
| */ | |
| inline int | |
| rpc_srv_getValsCall(rpc_func_t * __restrict call, rpc_val_t ** __restrict vals) | |
| { | |
| if (!call) { | |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t get return variables ...\n"); | |
| return -1; | |
| } | |
| |
| if (vals) | |
| *vals = call->func_vals; | |
| return call->func_args; | |
| } | |
| |
| // --------------------------------------------------------- | |
| |
| /* | |
| * rpc_srv_registerCall() Register call to RPC server | |
| * @srv = RPC Server instance | |
| * @csModule = Module name, if NULL self binary | |
| * @csFunc = Function name | |
| * @args = Number of function arguments | |
| * return: -1 error or 0 register ok | |
| */ | |
| int | |
| rpc_srv_registerCall(rpc_srv_t * __restrict srv, const char *csModule, const char *csFunc, u_char args) | |
| { | |
| rpc_func_t *func; | |
| u_char str[MAXPATHLEN + UCHAR_MAX + 1]; | |
| |
| memset(str, 0, MAXPATHLEN + UCHAR_MAX + 1); | |
| if (!srv || !csFunc) { | |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t register function to RPC server ...\n"); | |
| return -1; | |
| } | |
| if (!(func = malloc(sizeof(rpc_func_t)))) { | |
| LOGERR; | |
| return -1; | |
| } else { | |
| memset(func, 0, sizeof(rpc_func_t)); | |
| strlcpy((char*) func->func_name, csFunc, UCHAR_MAX + 1); | |
| } | |
| if (csModule) { | |
| strlcpy((char*) func->func_file, csModule, MAXPATHLEN); | |
| strlcpy((char*) str, csModule, MAXPATHLEN + UCHAR_MAX + 1); | |
| } | |
| strlcat((char*) str, "__", MAXPATHLEN + UCHAR_MAX + 1); | |
| strlcat((char*) str, csFunc, MAXPATHLEN + UCHAR_MAX + 1); | |
| |
| func->func_tag = crcFletcher16((u_short*) str, (MAXPATHLEN + UCHAR_MAX + 1) / 2); | |
| func->func_hash = hash_fnv((char*) str, MAXPATHLEN + UCHAR_MAX + 1); | |
| |
| if (rpc_srv_declValsCall(func, args) == -1) { | |
| free(func); | |
| return -1; | |
| } | |
| |
| pthread_mutex_lock(&srv->srv_mtx); | |
| func->func_next = srv->srv_funcs; | |
| srv->srv_funcs = func; | |
| pthread_mutex_unlock(&srv->srv_mtx); | |
| return 0; |
return 0; |
| } |
} |
| |
|
| /* |
|
| * rpc_srv_unregisterCall() Unregister call from RPC server |
|
| * @srv = RPC Server instance |
|
| * @csModule = Module name, if NULL self binary |
|
| * @csFunc = Function name |
|
| * return: -1 error, 0 not found call, 1 unregister ok |
|
| */ |
|
| int |
|
| rpc_srv_unregisterCall(rpc_srv_t * __restrict srv, const char *csModule, const char *csFunc) |
|
| { |
|
| rpc_func_t func, *f, *curr; |
|
| u_char str[MAXPATHLEN + UCHAR_MAX + 1]; |
|
| |
|
| memset(&func, 0, sizeof(rpc_func_t)); |
|
| memset(str, 0, MAXPATHLEN + UCHAR_MAX + 1); |
|
| if (!srv || !csFunc) { |
|
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t unregister function from RPC server ...\n"); |
|
| return -1; |
|
| } else |
|
| strlcpy((char*) func.func_name, csFunc, UCHAR_MAX + 1); |
|
| if (csModule) { |
|
| strlcpy((char*) func.func_file, csModule, MAXPATHLEN); |
|
| strlcpy((char*) str, csModule, MAXPATHLEN + UCHAR_MAX + 1); |
|
| } |
|
| strlcat((char*) str, "__", MAXPATHLEN + UCHAR_MAX + 1); |
|
| strlcat((char*) str, csFunc, MAXPATHLEN + UCHAR_MAX + 1); |
|
| |
|
| func.func_tag = crcFletcher16((u_short*) str, (MAXPATHLEN + UCHAR_MAX + 1) / 2); |
|
| func.func_hash = hash_fnv((char*) str, MAXPATHLEN + UCHAR_MAX + 1); |
|
| |
|
| f = rpc_srv_getCall(srv, func.func_tag, func.func_hash); |
|
| if (!f) // not found element for unregister |
|
| return 0; |
|
| |
|
| pthread_mutex_lock(&srv->srv_mtx); |
|
| if (srv->srv_funcs == f) { // if is 1st element |
|
| srv->srv_funcs = srv->srv_funcs->func_next; |
|
| |
|
| if (f->func_args && f->func_vals) |
|
| free(f->func_vals); |
|
| free(f); |
|
| } else { |
|
| for (curr = srv->srv_funcs; curr->func_next != f; curr = curr->func_next); |
|
| curr->func_next = curr->func_next->func_next; |
|
| |
|
| if (f->func_args && f->func_vals) |
|
| free(f->func_vals); |
|
| free(f); |
|
| } |
|
| pthread_mutex_unlock(&srv->srv_mtx); |
|
| |
|
| return 1; |
|
| } |
|
| |
|
| /* |
/* |
| * rpc_srv_getCall() Get registered call from RPC server |
|
| * @srv = RPC Server instance |
|
| * @tag = tag for function |
|
| * @hash = hash for function |
|
| * return: NULL not found call, !=NULL return call |
|
| */ |
|
| inline rpc_func_t * |
|
| rpc_srv_getCall(rpc_srv_t * __restrict srv, uint16_t tag, uint32_t hash) |
|
| { |
|
| rpc_func_t *f; |
|
| |
|
| if (!srv) { |
|
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t get function from RPC server ...\n"); |
|
| return NULL; |
|
| } |
|
| |
|
| for (f = srv->srv_funcs; f; f = f->func_next) |
|
| if (f->func_tag == tag && f->func_hash == hash) |
|
| break; |
|
| |
|
| return f; |
|
| } |
|
| |
|
| /* |
|
| * rpc_srv_getFunc() Get registered call from RPC server by Name |
|
| * @srv = RPC Server instance |
|
| * @csModule = Module name, if NULL self binary |
|
| * @csFunc = Function name |
|
| * return: NULL not found call, !=NULL return call |
|
| */ |
|
| rpc_func_t * |
|
| rpc_srv_getFunc(rpc_srv_t * __restrict srv, const char *csModule, const char *csFunc) |
|
| { |
|
| rpc_func_t func; |
|
| u_char str[MAXPATHLEN + UCHAR_MAX + 1]; |
|
| |
|
| memset(&func, 0, sizeof(rpc_func_t)); |
|
| memset(str, 0, MAXPATHLEN + UCHAR_MAX + 1); |
|
| if (!srv || !csFunc) { |
|
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t get function from RPC server ...\n"); |
|
| return NULL; |
|
| } else |
|
| strlcpy((char*) func.func_name, csFunc, UCHAR_MAX + 1); |
|
| if (csModule) { |
|
| strlcpy((char*) func.func_file, csModule, MAXPATHLEN); |
|
| strlcpy((char*) str, csModule, MAXPATHLEN + UCHAR_MAX + 1); |
|
| } |
|
| strlcat((char*) str, "__", MAXPATHLEN + UCHAR_MAX + 1); |
|
| strlcat((char*) str, csFunc, MAXPATHLEN + UCHAR_MAX + 1); |
|
| |
|
| func.func_tag = crcFletcher16((u_short*) str, (MAXPATHLEN + UCHAR_MAX + 1) / 2); |
|
| func.func_hash = hash_fnv((char*) str, MAXPATHLEN + UCHAR_MAX + 1); |
|
| |
|
| return rpc_srv_getCall(srv, func.func_tag, func.func_hash); |
|
| } |
|
| |
|
| // --------------------------------------------------------- |
|
| |
|
| /* |
|
| * rpc_srv_execCall() Execute registered call from RPC server |
* rpc_srv_execCall() Execute registered call from RPC server |
| * @data = RPC const data | * |
| * @call = Register RPC call | * @cli = RPC client |
| * @rpc = IN RPC call structure |
* @rpc = IN RPC call structure |
| * @args = IN RPC call array of rpc values | * @funcname = Execute RPC function |
| | * @args = IN RPC calling arguments from RPC client |
| * return: -1 error, !=-1 ok |
* return: -1 error, !=-1 ok |
| */ |
*/ |
| int |
int |
| rpc_srv_execCall(void * const data, rpc_func_t * __restrict call, | rpc_srv_execCall(rpc_cli_t * __restrict cli, struct tagRPCCall * __restrict rpc, |
| struct tagRPCCall * __restrict rpc, rpc_val_t * __restrict args) | ait_val_t funcname, array_t * __restrict args) |
| { |
{ |
| void *dl; |
|
| rpc_callback_t func; |
rpc_callback_t func; |
| int ret; |
|
| |
|
| if (!data || !call || !rpc) { | if (!cli || !rpc || !AIT_ADDR(&funcname)) { |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t exec call from RPC server ...\n"); | rpc_SetErr(EINVAL, "Invalid parameter can`t exec function"); |
| return -1; |
return -1; |
| } |
} |
| |
|
| dl = dlopen((char*) (*call->func_file ? call->func_file : NULL), RTLD_NOW); | func = AIT_GET_LIKE(&funcname, rpc_callback_t); |
| if (!dl) { | return func(cli, rpc, args); |
| rpc_SetErr(ENOENT, "Error:: Can`t attach module %s!\n", dlerror()); | |
| return -1; | |
| } | |
| |
| func = dlsym(dl, (char*) call->func_name); | |
| if (func) | |
| ret = func(data, call, rpc->call_argc, args); | |
| else { | |
| rpc_SetErr(ENOEXEC, "Error:: Can`t find function %s!\n", dlerror()); | |
| ret = -1; | |
| } | |
| |
| dlclose(dl); | |
| return ret; | |
| } |
} |