|
|
| version 1.1.1.1.2.1, 2010/06/18 13:36:01 | version 1.1.1.1.2.21, 2010/07/08 13:05:50 |
|---|---|
| Line 14 rpc_srv_dispatchCall(void *arg) | Line 14 rpc_srv_dispatchCall(void *arg) |
| { | { |
| rpc_cli_t *c = arg; | rpc_cli_t *c = arg; |
| rpc_srv_t *s; | rpc_srv_t *s; |
| rpc_val_t *vals, *v = NULL; | rpc_val_t *vals = NULL, *v = NULL; |
| rpc_func_t *f; | rpc_func_t *f; |
| struct tagRPCCall *rpc; | struct tagRPCCall *rpc; |
| struct tagRPCRet rrpc; | struct tagRPCRet rrpc; |
| fd_set fds; | fd_set fds; |
| u_char buf[BUFSIZ], *data; | u_char buf[BUFSIZ], *data; |
| int ret, argc, Limit = 0; | int ret, argc = 0, Limit = 0; |
| register int i; | register int i; |
| if (!arg) { | if (!arg) { |
| Line 57 rpc_srv_dispatchCall(void *arg) | Line 57 rpc_srv_dispatchCall(void *arg) |
| if (memcmp(&rpc->call_session, &s->srv_session, sizeof rpc->call_session)) { | if (memcmp(&rpc->call_session, &s->srv_session, sizeof rpc->call_session)) { |
| rpc_SetErr(EINVAL, "Error:: get invalid RPC session ...\n"); | rpc_SetErr(EINVAL, "Error:: get invalid RPC session ...\n"); |
| ret = -5; | ret = -5; |
| break; | goto makeReply; |
| } | } |
| // RPC is OK! Go decapsulate variables ... | // RPC is OK! Go decapsulate variables ... |
| if (rpc->call_argc) { | if (rpc->call_argc) { |
| Line 74 rpc_srv_dispatchCall(void *arg) | Line 74 rpc_srv_dispatchCall(void *arg) |
| v[i].val.string = (int8_t*) data; | v[i].val.string = (int8_t*) data; |
| data += v[i].val_len + 1; | data += v[i].val_len + 1; |
| break; | break; |
| case array: | case blob: |
| v[i].val.array = (int8_t**) data; | if (s->srv_blob.state == disable) { |
| data += v[i].val_len; | rpc_SetErr(ENOTSUP, "Error:: BLOB server is disabled\n"); |
| break; | ret = -5; |
| goto makeReply; | |
| } | |
| default: | default: |
| break; | break; |
| } | } |
| Line 90 rpc_srv_dispatchCall(void *arg) | Line 92 rpc_srv_dispatchCall(void *arg) |
| rpc_SetErr(EINVAL, "Error:: call not found into RPC server ...\n"); | rpc_SetErr(EINVAL, "Error:: call not found into RPC server ...\n"); |
| ret = -6; | ret = -6; |
| } else | } else |
| if ((ret = rpc_srv_execCall(s, f, rpc, v)) == -1) | if ((ret = rpc_srv_execCall(f, rpc, v)) == -1) |
| ret = -6; | ret = -9; |
| else | else |
| argc = rpc_srv_getValsCall(f, &vals); | argc = rpc_srv_getValsCall(f, &vals); |
| makeReply: | |
| memcpy(&rrpc.ret_session, &rpc->call_session, sizeof rrpc.ret_session); | memcpy(&rrpc.ret_session, &rpc->call_session, sizeof rrpc.ret_session); |
| rrpc.ret_tag = rpc->call_tag; | rrpc.ret_tag = rpc->call_tag; |
| rrpc.ret_hash = rpc->call_hash; | rrpc.ret_hash = rpc->call_hash; |
| Line 135 rpc_srv_dispatchCall(void *arg) | Line 138 rpc_srv_dispatchCall(void *arg) |
| data += vals[i].val_len + 1; | data += vals[i].val_len + 1; |
| Limit += vals[i].val_len + 1; | Limit += vals[i].val_len + 1; |
| break; | break; |
| case array: | case blob: |
| if (ret || Limit + vals[i].val_len > BUFSIZ) { | if (s->srv_blob.state == disable) { |
| rpc_SetErr(EMSGSIZE, "Error:: in prepare RPC packet (-7) ...\n"); | rpc_SetErr(ENOTSUP, "Error:: BLOB server is disabled\n"); |
| rrpc.ret_retcode = ret = -7; | rrpc.ret_retcode = ret = -5; |
| rrpc.ret_argc = 0; | rrpc.ret_argc = 0; |
| break; | break; |
| } | } |
| memcpy(data, vals[i].val.array, vals[i].val_len); | |
| data += vals[i].val_len; | |
| Limit += vals[i].val_len; | |
| break; | |
| default: | default: |
| break; | break; |
| } | } |
| Line 174 rpc_srv_dispatchCall(void *arg) | Line 172 rpc_srv_dispatchCall(void *arg) |
| return (void*) ret; | return (void*) ret; |
| } | } |
| static void * | |
| rpc_srv_dispatchVars(void *arg) | |
| { | |
| rpc_cli_t *c = arg; | |
| rpc_srv_t *s; | |
| rpc_blob_t *b; | |
| int ret; | |
| fd_set fds; | |
| u_char buf[sizeof(struct tagBLOBHdr)]; | |
| struct tagBLOBHdr *blob; | |
| if (!arg) { | |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t procced BLOB client ...\n"); | |
| return NULL; | |
| } else | |
| s = c->cli_parent; | |
| do { | |
| // check for disable service at this moment? | |
| if (s->srv_blob.state == disable) { | |
| ret = 0; | |
| break; | |
| } | |
| FD_ZERO(&fds); | |
| FD_SET(c->cli_sock, &fds); | |
| ret = select(c->cli_sock + 1, &fds, NULL, NULL, NULL); | |
| if (ret == -1) { | |
| ret = -2; | |
| } | |
| memset(buf, 0, sizeof buf); | |
| if ((ret = recv(c->cli_sock, buf, sizeof buf, 0)) == -1) { | |
| LOGERR; | |
| ret = -3; | |
| break; | |
| } | |
| if (!ret || s->srv_blob.state == disable) { // receive EOF or disable service | |
| ret = 0; | |
| break; | |
| } | |
| if (ret < sizeof(struct tagBLOBHdr)) { | |
| rpc_SetErr(EMSGSIZE, "Error:: too short BLOB packet ...\n"); | |
| ret = -4; | |
| break; | |
| } else | |
| blob = (struct tagBLOBHdr*) buf; | |
| // check BLOB packet session info | |
| if (memcmp(&blob->hdr_session, &s->srv_session, sizeof blob->hdr_session)) { | |
| rpc_SetErr(EINVAL, "Error:: get invalid BLOB session ...\n"); | |
| ret = -5; | |
| goto makeReply; | |
| } | |
| // Go to proceed packet ... | |
| switch (blob->hdr_cmd) { | |
| case get: | |
| if (!(b = rpc_srv_getBLOB(s, blob->hdr_var))) { | |
| rpc_SetErr(EINVAL, "Error:: var (%x) not found into BLOB server ...\n", | |
| blob->hdr_var); | |
| ret = -6; | |
| break; | |
| } else | |
| blob->hdr_len = b->blob_len; | |
| if (rpc_srv_blobMap(s, b) != -1) { | |
| ret = rpc_srv_sendBLOB(c, b); | |
| rpc_srv_blobUnmap(b); | |
| } else | |
| ret = -7; | |
| break; | |
| case set: | |
| if ((b = rpc_srv_registerBLOB(s, blob->hdr_len))) { | |
| // set new BLOB variable for reply :) | |
| blob->hdr_var = b->blob_var; | |
| ret = rpc_srv_recvBLOB(c, b); | |
| rpc_srv_blobUnmap(b); | |
| } else | |
| ret = -7; | |
| break; | |
| case unset: | |
| ret = rpc_srv_unregisterBLOB(s, blob->hdr_var); | |
| if (ret == -1) | |
| ret = -7; | |
| break; | |
| default: | |
| rpc_SetErr(EINVAL, "Error:: unsupported BLOB command (%d)...\n", | |
| blob->hdr_cmd); | |
| ret = -7; | |
| } | |
| makeReply: | |
| // Replay to client! | |
| blob->hdr_cmd = ret < 0 ? error : ok; | |
| blob->hdr_ret = ret; | |
| if ((ret = send(c->cli_sock, buf, sizeof buf, 0)) == -1) { | |
| LOGERR; | |
| ret = -8; | |
| break; | |
| } | |
| if (ret != sizeof buf) { | |
| rpc_SetErr(EBADMSG, "Error:: in send BLOB reply, should be send %d bytes, " | |
| "really is %d\n", sizeof buf, ret); | |
| ret = -9; | |
| break; | |
| } | |
| } while (ret > -1); | |
| shutdown(c->cli_sock, SHUT_RDWR); | |
| close(c->cli_sock); | |
| memset(c, 0, sizeof(rpc_cli_t)); | |
| return (void*) ret; | |
| } | |
| // ------------------------------------------------- | // ------------------------------------------------- |
| /* | /* |
| * rpc_srv_initBLOBServer() Init & create BLOB Server | |
| * @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 | |
| */ | |
| int | |
| rpc_srv_initBLOBServer(rpc_srv_t * __restrict srv, u_short Port, const char *diskDir) | |
| { | |
| int n = 1; | |
| struct sockaddr_in sin; | |
| struct sockaddr_in6 sin6; | |
| if (!srv) { | |
| rpc_SetErr(EINVAL, "Error:: Invalid parameters can`t init BLOB server ...\n"); | |
| 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); | |
| if (access(diskDir, R_OK | W_OK) == -1) { | |
| LOGERR; | |
| return -1; | |
| } else | |
| strlcpy(srv->srv_blob.dir, diskDir, UCHAR_MAX + 1); | |
| srv->srv_blob.server.cli_tid = pthread_self(); | |
| 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); | |
| sin.sin_port = htons(Port); | |
| memcpy(&srv->srv_blob.server.cli_sa, &sin, sizeof(struct sockaddr)); | |
| } else { | |
| memcpy(&sin6, &srv->srv_server.cli_sa, sizeof sin6); | |
| sin6.sin6_port = htons(Port); | |
| memcpy(&srv->srv_blob.server.cli_sa, &sin6, sizeof(struct sockaddr)); | |
| } | |
| srv->srv_blob.server.cli_sock = socket(srv->srv_server.cli_sa.sa_family, SOCK_STREAM, 0); | |
| if (srv->srv_blob.server.cli_sock == -1) { | |
| LOGERR; | |
| return -1; | |
| } | |
| if (setsockopt(srv->srv_blob.server.cli_sock, SOL_SOCKET, SO_REUSEADDR, &n, sizeof n) == -1) { | |
| LOGERR; | |
| close(srv->srv_blob.server.cli_sock); | |
| return -1; | |
| } | |
| if (bind(srv->srv_blob.server.cli_sock, &srv->srv_blob.server.cli_sa, | |
| sizeof srv->srv_blob.server.cli_sa) == -1) { | |
| LOGERR; | |
| close(srv->srv_blob.server.cli_sock); | |
| return -1; | |
| } | |
| srv->srv_blob.clients = calloc(srv->srv_numcli, sizeof(rpc_cli_t)); | |
| if (!srv->srv_blob.clients) { | |
| LOGERR; | |
| close(srv->srv_blob.server.cli_sock); | |
| return -1; | |
| } else | |
| memset(srv->srv_blob.clients, 0, srv->srv_numcli * sizeof(rpc_cli_t)); | |
| pthread_mutex_init(&srv->srv_blob.mtx, NULL); | |
| pthread_mutex_lock(&srv->srv_mtx); | |
| rpc_srv_registerCall(srv, NULL, CALL_BLOBSHUTDOWN, 0); | |
| rpc_srv_registerCall(srv, NULL, CALL_BLOBCLIENTS, 0); | |
| rpc_srv_registerCall(srv, NULL, CALL_BLOBVARS, 0); | |
| rpc_srv_registerCall(srv, NULL, CALL_BLOBSTATE, 1); | |
| pthread_mutex_unlock(&srv->srv_mtx); | |
| srv->srv_blob.state = enable; // enable BLOB | |
| return 0; | |
| } | |
| /* | |
| * rpc_srv_endBLOBServer() Destroy BLOB server, close all opened sockets and free resources | |
| * @srv = RPC Server instance | |
| * return: none | |
| */ | |
| void | |
| rpc_srv_endBLOBServer(rpc_srv_t * __restrict srv) | |
| { | |
| rpc_cli_t *c; | |
| register int i; | |
| rpc_blob_t *f; | |
| if (!srv) { | |
| rpc_SetErr(EINVAL, "Error:: Can`t destroy server because parameter is null!\n"); | |
| return; | |
| } else | |
| srv->srv_blob.state = disable; | |
| rpc_srv_unregisterCall(srv, NULL, CALL_BLOBSHUTDOWN); | |
| rpc_srv_unregisterCall(srv, NULL, CALL_BLOBCLIENTS); | |
| rpc_srv_unregisterCall(srv, NULL, CALL_BLOBVARS); | |
| rpc_srv_unregisterCall(srv, NULL, CALL_BLOBSTATE); | |
| for (i = 0, c = srv->srv_blob.clients; i < srv->srv_numcli && c; i++, c++) | |
| 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); | |
| srv->srv_blob.clients = NULL; | |
| } | |
| pthread_mutex_lock(&srv->srv_blob.mtx); | |
| while ((f = srv->srv_blob.blobs)) { | |
| srv->srv_blob.blobs = f->blob_next; | |
| rpc_srv_blobFree(srv, f); | |
| free(f); | |
| } | |
| pthread_mutex_unlock(&srv->srv_blob.mtx); | |
| while (pthread_mutex_trylock(&srv->srv_blob.mtx) == EBUSY); | |
| pthread_mutex_destroy(&srv->srv_blob.mtx); | |
| } | |
| /* | |
| * rpc_srv_execBLOBServer() Execute Main BLOB server loop and wait for clients requests | |
| * @srv = RPC Server instance | |
| * return: -1 error or 0 ok, infinite loop ... | |
| */ | |
| int | |
| rpc_srv_execBLOBServer(rpc_srv_t * __restrict srv) | |
| { | |
| socklen_t salen = sizeof(struct sockaddr); | |
| register int i; | |
| rpc_cli_t *c; | |
| fd_set fds; | |
| int ret; | |
| struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; | |
| if (!srv || srv->srv_blob.state == disable) { | |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t start BLOB server ...\n"); | |
| return -1; | |
| } | |
| if (listen(srv->srv_blob.server.cli_sock, SOMAXCONN) == -1) { | |
| LOGERR; | |
| return -1; | |
| } | |
| while (!blob_Kill && !rpc_Kill) { | |
| for (c = srv->srv_blob.clients, i = 0; i < srv->srv_numcli && c; i++, c++) | |
| if (!c->cli_sa.sa_family) | |
| break; | |
| if (i >= srv->srv_numcli) { | |
| usleep(1000000); | |
| continue; | |
| } | |
| FD_ZERO(&fds); | |
| FD_SET(srv->srv_blob.server.cli_sock, &fds); | |
| ret = select(srv->srv_blob.server.cli_sock + 1, &fds, NULL, NULL, &tv); | |
| if (ret == -1) { | |
| LOGERR; | |
| ret = 1; | |
| break; | |
| } | |
| if (!ret) | |
| continue; | |
| c->cli_sock = accept(srv->srv_blob.server.cli_sock, &c->cli_sa, &salen); | |
| if (c->cli_sock == -1) { | |
| LOGERR; | |
| continue; | |
| } else | |
| c->cli_parent = srv; | |
| if (pthread_create(&c->cli_tid, NULL, rpc_srv_dispatchVars, c)) { | |
| LOGERR; | |
| continue; | |
| } | |
| } | |
| srv->srv_blob.state = disable; | |
| return 0; | |
| } | |
| /* | |
| * rpc_srv_initServer() Init & create RPC Server | * rpc_srv_initServer() Init & create RPC Server |
| * @regProgID = ProgramID for authentication & recognition | * @regProgID = ProgramID for authentication & recognition |
| * @regProcID = ProcessID for authentication & recognition | * @regProcID = ProcessID for authentication & recognition |
| Line 243 rpc_srv_initServer(u_int regProgID, u_int regProcID, i | Line 546 rpc_srv_initServer(u_int regProgID, u_int regProcID, i |
| srv->srv_session.sess_program = regProgID; | srv->srv_session.sess_program = regProgID; |
| srv->srv_session.sess_process = regProcID; | 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) | if (family == AF_INET) |
| memcpy(&srv->srv_server.cli_sa, &sin, sizeof srv->srv_server.cli_sa); | memcpy(&srv->srv_server.cli_sa, &sin, sizeof srv->srv_server.cli_sa); |
| Line 276 rpc_srv_initServer(u_int regProgID, u_int regProcID, i | Line 580 rpc_srv_initServer(u_int regProgID, u_int regProcID, i |
| } else | } else |
| memset(srv->srv_clients, 0, srv->srv_numcli * sizeof(rpc_cli_t)); | memset(srv->srv_clients, 0, srv->srv_numcli * sizeof(rpc_cli_t)); |
| pthread_mutex_init(&srv->srv_mtx, NULL); | |
| rpc_srv_registerCall(srv, NULL, CALL_SRVSHUTDOWN, 0); | rpc_srv_registerCall(srv, NULL, CALL_SRVSHUTDOWN, 0); |
| rpc_srv_registerCall(srv, NULL, CALL_SRVCLIENTS, 0); | rpc_srv_registerCall(srv, NULL, CALL_SRVCLIENTS, 0); |
| rpc_srv_registerCall(srv, NULL, CALL_SRVCALLS, 0); | rpc_srv_registerCall(srv, NULL, CALL_SRVCALLS, 0); |
| rpc_srv_registerCall(srv, NULL, CALL_SRVSESSIONS, 0); | rpc_srv_registerCall(srv, NULL, CALL_SRVSESSIONS, 0); |
| pthread_mutex_init(&rpc_mtx, NULL); | |
| return srv; | return srv; |
| } | } |
| Line 302 rpc_srv_endServer(rpc_srv_t * __restrict srv) | Line 606 rpc_srv_endServer(rpc_srv_t * __restrict srv) |
| return; | return; |
| } | } |
| pthread_mutex_destroy(&rpc_mtx); | rpc_srv_endBLOBServer(srv); |
| while ((f = srv->srv_funcs)) { | |
| srv->srv_funcs = f->func_next; | |
| free(f); | |
| } | |
| for (i = 0, c = srv->srv_clients; i < srv->srv_numcli && c; i++, c++) | for (i = 0, c = srv->srv_clients; i < srv->srv_numcli && c; i++, c++) |
| if (c->cli_sa.sa_family) | if (c->cli_sa.sa_family) |
| shutdown(c->cli_sock, SHUT_RDWR); | shutdown(c->cli_sock, SHUT_RDWR); |
| Line 316 rpc_srv_endServer(rpc_srv_t * __restrict srv) | Line 615 rpc_srv_endServer(rpc_srv_t * __restrict srv) |
| if (srv->srv_clients) { | if (srv->srv_clients) { |
| free(srv->srv_clients); | free(srv->srv_clients); |
| srv->srv_clients = NULL; | |
| srv->srv_numcli = 0; | srv->srv_numcli = 0; |
| } | } |
| 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); | |
| while (pthread_mutex_trylock(&srv->srv_mtx) == EBUSY); | |
| pthread_mutex_destroy(&srv->srv_mtx); | |
| free(srv); | free(srv); |
| srv = NULL; | srv = NULL; |
| } | } |
| Line 334 rpc_srv_execServer(rpc_srv_t * __restrict srv) | Line 644 rpc_srv_execServer(rpc_srv_t * __restrict srv) |
| socklen_t salen = sizeof(struct sockaddr); | socklen_t salen = sizeof(struct sockaddr); |
| register int i; | register int i; |
| rpc_cli_t *c; | rpc_cli_t *c; |
| fd_set fds; | |
| int ret; | |
| struct timeval tv = { DEF_RPC_TIMEOUT, 0 }; | |
| if (!srv) { | if (!srv) { |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t start RPC server ...\n"); | rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t start RPC server ...\n"); |
| Line 349 rpc_srv_execServer(rpc_srv_t * __restrict srv) | Line 662 rpc_srv_execServer(rpc_srv_t * __restrict srv) |
| for (c = srv->srv_clients, i = 0; i < srv->srv_numcli && c; i++, c++) | for (c = srv->srv_clients, i = 0; i < srv->srv_numcli && c; i++, c++) |
| if (!c->cli_sa.sa_family) | if (!c->cli_sa.sa_family) |
| break; | break; |
| if (c && c->cli_sa.sa_family && c->cli_parent) { | if (i >= srv->srv_numcli) { |
| usleep(1000000); | usleep(1000000); |
| continue; | 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; | |
| ret = 1; | |
| break; | |
| } | |
| if (!ret) | |
| continue; | |
| c->cli_sock = accept(srv->srv_server.cli_sock, &c->cli_sa, &salen); | c->cli_sock = accept(srv->srv_server.cli_sock, &c->cli_sa, &salen); |
| if (c->cli_sock == -1) { | if (c->cli_sock == -1) { |
| printf("%s(%d): #%d - %s\n", __func__, __LINE__, errno, strerror(errno)); | |
| LOGERR; | LOGERR; |
| continue; | continue; |
| } else | } else |
| Line 367 rpc_srv_execServer(rpc_srv_t * __restrict srv) | Line 691 rpc_srv_execServer(rpc_srv_t * __restrict srv) |
| } | } |
| } | } |
| /* not reached !!! */ | |
| return 0; | return 0; |
| } | } |
| // --------------------------------------------------------- | // --------------------------------------------------------- |
| /* | /* |
| * rpc_srv_freeValsCall() Free return variables for RPC call | |
| * @call = RPC function call | |
| * 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 { | |
| ptr = realloc(call->func_vals, return_vals * sizeof(rpc_val_t)); | |
| if (!ptr) { | |
| LOGERR; | |
| return -1; | |
| } else | |
| call->func_vals = ptr; | |
| } | |
| return call->func_args; | |
| } | |
| /* | |
| * rpc_srv_delValsCall() Clean values from return variables of RPC call | |
| * @call = RPC function call | |
| * return: -1 error, !=-1 Returned number of cleaned RPC variables | |
| */ | |
| 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)); | |
| return call->func_args; | |
| } | |
| /* | |
| * 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; | |
| } | |
| *newvals = calloc(call->func_args, sizeof(rpc_val_t)); | |
| 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; | |
| } | |
| func->func_next = srv->srv_funcs; | |
| srv->srv_funcs = func; | |
| 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; | |
| 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); | |
| } | |
| 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 | * @call = Register RPC call |
| * @rpc = IN RPC call structure | * @rpc = IN RPC call structure |
| * @args = IN RPC call array of rpc values | * @args = IN RPC call array of rpc values |
| * 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_func_t * __restrict call, struct tagRPCCall * __restrict rpc, |
| struct tagRPCCall * __restrict rpc, rpc_val_t * __restrict args) | rpc_val_t * __restrict args) |
| { | { |
| void *dl; | void *dl; |
| rpc_callback_t func; | rpc_callback_t func; |
| int ret; | int ret; |
| if (!data || !call || !rpc) { | if (!call || !rpc || !call->func_parent) { |
| rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t exec call from RPC server ...\n"); | rpc_SetErr(EINVAL, "Error:: Invalid parameter can`t exec call from RPC server ...\n"); |
| return -1; | return -1; |
| } | } |
| Line 666 rpc_srv_execCall(void * const data, rpc_func_t * __res | Line 724 rpc_srv_execCall(void * const data, rpc_func_t * __res |
| func = dlsym(dl, (char*) call->func_name); | func = dlsym(dl, (char*) call->func_name); |
| if (func) | if (func) |
| ret = func(data, call, rpc->call_argc, args); | ret = func(call, rpc->call_argc, args); |
| else { | else { |
| rpc_SetErr(ENOEXEC, "Error:: Can`t find function %s!\n", dlerror()); | rpc_SetErr(ENOEXEC, "Error:: Can`t find function %s!\n", dlerror()); |
| ret = -1; | ret = -1; |