|
|
| version 1.21.2.2, 2013/11/14 23:38:41 | version 1.22.6.1, 2014/01/28 13:56:25 |
|---|---|
| Line 66 static sched_task_func_t cbProto[SOCK_RAW + 1][4] = { | Line 66 static sched_task_func_t cbProto[SOCK_RAW + 1][4] = { |
| { NULL, NULL, NULL, NULL } /* SOCK_RAW */ | { NULL, NULL, NULL, NULL } /* SOCK_RAW */ |
| }; | }; |
| /* Global Signal Argument when kqueue support disabled */ | |
| static volatile uintptr_t _glSigArg = 0; | |
| void | void |
| rpc_freeCli(rpc_cli_t * __restrict c) | rpc_freeCli(rpc_cli_t * __restrict c) |
| { | { |
| Line 223 txPacket(sched_task_t *task) | Line 227 txPacket(sched_task_t *task) |
| if (ret) | if (ret) |
| LOGERR; | LOGERR; |
| else | else |
| rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond"); | rpc_SetErr(ETIMEDOUT, "Timeout reached! Client not respond"); |
| /* close connection */ | /* close connection */ |
| schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], | schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], |
| TASK_ARG(task), 0, NULL, 0); | TASK_ARG(task), 0, NULL, 0); |
| Line 346 rxPacket(sched_task_t *task) | Line 350 rxPacket(sched_task_t *task) |
| if (rlen) | if (rlen) |
| LOGERR; | LOGERR; |
| else | else |
| rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond"); | rpc_SetErr(ETIMEDOUT, "Timeout reached! Client not respond"); |
| schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], | schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], |
| TASK_ARG(task), 0, NULL, 0); | TASK_ARG(task), 0, NULL, 0); |
| return NULL; | return NULL; |
| Line 448 txUDPPacket(sched_task_t *task) | Line 452 txUDPPacket(sched_task_t *task) |
| rpc_func_t *f = NULL; | rpc_func_t *f = NULL; |
| u_char *buf = AIT_GET_BUF(&c->cli_buf); | u_char *buf = AIT_GET_BUF(&c->cli_buf); |
| struct tagRPCCall *rpc = (struct tagRPCCall*) buf; | struct tagRPCCall *rpc = (struct tagRPCCall*) buf; |
| int ret, wlen = sizeof(struct tagRPCCall); | int ret, estlen, wlen = sizeof(struct tagRPCCall); |
| struct timespec ts = { DEF_RPC_TIMEOUT, 0 }; | struct timespec ts = { DEF_RPC_TIMEOUT, 0 }; |
| struct pollfd pfd; | struct pollfd pfd; |
| Line 464 txUDPPacket(sched_task_t *task) | Line 468 txUDPPacket(sched_task_t *task) |
| rpc->call_rep.ret = RPC_ERROR(-1); | rpc->call_rep.ret = RPC_ERROR(-1); |
| rpc->call_rep.eno = RPC_ERROR(rpc_Errno); | rpc->call_rep.eno = RPC_ERROR(rpc_Errno); |
| } else { | } else { |
| /* calc estimated length */ | |
| estlen = ait_resideVars(RPC_RETVARS(c)) + wlen; | |
| if (estlen > AIT_LEN(&c->cli_buf)) | |
| AIT_RE_BUF(&c->cli_buf, estlen); | |
| buf = AIT_GET_BUF(&c->cli_buf); | |
| rpc = (struct tagRPCCall*) buf; | |
| rpc->call_argc = htons(array_Size(RPC_RETVARS(c))); | rpc->call_argc = htons(array_Size(RPC_RETVARS(c))); |
| /* Go Encapsulate variables */ | /* Go Encapsulate variables */ |
| ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, | ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, |
| Line 495 txUDPPacket(sched_task_t *task) | Line 506 txUDPPacket(sched_task_t *task) |
| if (ret) | if (ret) |
| LOGERR; | LOGERR; |
| else | else |
| rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond"); | rpc_SetErr(ETIMEDOUT, "Timeout reached! Client not respond"); |
| /* close connection */ | /* close connection */ |
| schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], | schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], |
| TASK_ARG(task), 0, NULL, 0); | TASK_ARG(task), 0, NULL, 0); |
| Line 537 rxUDPPacket(sched_task_t *task) | Line 548 rxUDPPacket(sched_task_t *task) |
| } | } |
| c = _allocClient(srv, &sa); | c = _allocClient(srv, &sa); |
| if (!c) | if (!c) { |
| EVERBOSE(1, "RPC client quota exceeded! Connection will be shutdown!\n"); | |
| usleep(2000); /* blocked client delay */ | |
| goto end; | goto end; |
| else { | } else { |
| estlen = ntohl(rpc->call_len); | estlen = ntohl(rpc->call_len); |
| if (estlen > AIT_LEN(&c->cli_buf)) | if (estlen > AIT_LEN(&c->cli_buf)) |
| AIT_RE_BUF(&c->cli_buf, estlen); | AIT_RE_BUF(&c->cli_buf, estlen); |
| Line 566 rxUDPPacket(sched_task_t *task) | Line 579 rxUDPPacket(sched_task_t *task) |
| if (rlen) | if (rlen) |
| LOGERR; | LOGERR; |
| else | else |
| rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond"); | rpc_SetErr(ETIMEDOUT, "Timeout reached! Client not respond"); |
| schedEvent(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], | schedEvent(TASK_ROOT(task), cbProto[srv->srv_proto][CB_CLOSECLIENT], |
| c, 0, NULL, 0); | c, 0, NULL, 0); |
| return NULL; | return NULL; |
| } | } |
| salen = sa.ss.ss_len = sizeof(sockaddr_t); | |
| rlen = recvfrom(TASK_FD(task), buf, len, 0, &sa.sa, &salen); | rlen = recvfrom(TASK_FD(task), buf, len, 0, &sa.sa, &salen); |
| if (rlen == -1) { | if (rlen == -1) { |
| /* close connection */ | /* close connection */ |
| Line 578 rxUDPPacket(sched_task_t *task) | Line 592 rxUDPPacket(sched_task_t *task) |
| c, 0, NULL, 0); | c, 0, NULL, 0); |
| return NULL; | return NULL; |
| } | } |
| if (e_addrcmp(&c->cli_sa, &sa, 42)) | |
| rlen ^= rlen; /* skip if arrive from different address */ | |
| } | } |
| len = estlen; | len = estlen; |
| Line 748 end: | Line 764 end: |
| static void * | static void * |
| flushBLOB(sched_task_t *task) | flushBLOB(sched_task_t *task) |
| { | { |
| rpc_srv_t *srv = TASK_ARG(task); | uintptr_t sigArg = atomic_load_acq_ptr(&_glSigArg); |
| rpc_srv_t *srv = sigArg ? (void*) sigArg : TASK_ARG(task); | |
| rpc_blob_t *b, *tmp; | rpc_blob_t *b, *tmp; |
| TAILQ_FOREACH_SAFE(b, &srv->srv_blob.blobs, blob_node, tmp) { | TAILQ_FOREACH_SAFE(b, &srv->srv_blob.blobs, blob_node, tmp) { |
| Line 758 flushBLOB(sched_task_t *task) | Line 775 flushBLOB(sched_task_t *task) |
| e_free(b); | e_free(b); |
| } | } |
| schedSignalSelf(task); | if (!schedSignalSelf(task)) { |
| /* disabled kqueue support in libaitsched */ | |
| struct sigaction sa; | |
| memset(&sa, 0, sizeof sa); | |
| sigemptyset(&sa.sa_mask); | |
| sa.sa_handler = (void (*)(int)) flushBLOB; | |
| sa.sa_flags = SA_RESTART | SA_RESETHAND; | |
| sigaction(SIGFBLOB, &sa, NULL); | |
| } | |
| return NULL; | return NULL; |
| } | } |
| Line 972 rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv) | Line 999 rpc_srv_loopBLOBServer(rpc_srv_t * __restrict srv) |
| return -1; | return -1; |
| } | } |
| schedSignal(srv->srv_blob.root, flushBLOB, srv, SIGFBLOB, NULL, 0); | if (!schedSignal(srv->srv_blob.root, flushBLOB, srv, SIGFBLOB, NULL, 0)) { |
| /* disabled kqueue support in libaitsched */ | |
| struct sigaction sa; | |
| atomic_store_rel_ptr(&_glSigArg, (uintptr_t) srv); | |
| memset(&sa, 0, sizeof sa); | |
| sigemptyset(&sa.sa_mask); | |
| sa.sa_handler = (void (*)(int)) flushBLOB; | |
| sa.sa_flags = SA_RESTART | SA_RESETHAND; | |
| sigaction(SIGFBLOB, &sa, NULL); | |
| } | |
| if (!schedRead(srv->srv_blob.root, acceptBLOBClients, srv, | if (!schedRead(srv->srv_blob.root, acceptBLOBClients, srv, |
| srv->srv_blob.server.cli_sock, NULL, 0)) { | srv->srv_blob.server.cli_sock, NULL, 0)) { |
| rpc_SetErr(sched_GetErrno(), "%s", sched_GetError()); | rpc_SetErr(sched_GetErrno(), "%s", sched_GetError()); |