|
|
| version 1.17.6.3, 2013/08/22 13:09:53 | version 1.18.2.2, 2013/08/23 13:34:52 |
|---|---|
| Line 164 txPacket(sched_task_t *task) | Line 164 txPacket(sched_task_t *task) |
| 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, estlen, wlen = sizeof(struct tagRPCCall); | int ret, estlen, wlen = sizeof(struct tagRPCCall); |
| struct pollfd pfd; | |
| if (rpc->call_argc) { | if (rpc->call_argc) { |
| f = rpc_srv_getCall(s, ntohs(rpc->call_tag)); | f = rpc_srv_getCall(s, ntohs(rpc->call_tag)); |
| Line 179 txPacket(sched_task_t *task) | Line 180 txPacket(sched_task_t *task) |
| 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); |
| buf = AIT_GET_BUF(&c->cli_buf); | 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 */ |
| Line 188 txPacket(sched_task_t *task) | Line 190 txPacket(sched_task_t *task) |
| ait_freeVars(&c->cli_vars); | ait_freeVars(&c->cli_vars); |
| if (ret == -1) { | if (ret == -1) { |
| rpc_SetErr(EBADRPC, "Prepare RPC packet failed"); | rpc_SetErr(EBADRPC, "Prepare RPC packet failed"); |
| rpc->call_argc ^= rpc->call_argc; | rpc->call_argc ^= rpc->call_argc; |
| 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); |
| Line 205 txPacket(sched_task_t *task) | Line 208 txPacket(sched_task_t *task) |
| #endif | #endif |
| /* send reply */ | /* send reply */ |
| ret = send(TASK_FD(task), buf, wlen, MSG_NOSIGNAL); | pfd.fd = TASK_FD(task); |
| if (ret == -1 || ret != wlen) { | pfd.events = POLLOUT; |
| /* close connection */ | for (; wlen > 0; wlen -= ret, buf += ret) { |
| schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], | if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || |
| TASK_ARG(task), 0, NULL, 0); | pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) { |
| if (ret) | |
| LOGERR; | |
| else | |
| rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond"); | |
| /* close connection */ | |
| schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], | |
| TASK_ARG(task), 0, NULL, 0); | |
| return NULL; | |
| } | |
| ret = send(TASK_FD(task), buf, MIN(wlen, s->srv_netbuf), MSG_NOSIGNAL); | |
| if (ret == -1) { | |
| /* close connection */ | |
| schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], | |
| TASK_ARG(task), 0, NULL, 0); | |
| return NULL; | |
| } | |
| } | } |
| return NULL; | return NULL; |
| Line 282 rxPacket(sched_task_t *task) | Line 301 rxPacket(sched_task_t *task) |
| #endif | #endif |
| 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; |
| struct pollfd pfd; | |
| memset(buf, 0, sizeof(struct tagRPCCall)); | memset(buf, 0, sizeof(struct tagRPCCall)); |
| rlen = recv(TASK_FD(task), rpc, sizeof(struct tagRPCCall), 0); | rlen = recv(TASK_FD(task), rpc, sizeof(struct tagRPCCall), MSG_PEEK); |
| if (rlen < sizeof(struct tagRPCCall)) { | if (rlen < sizeof(struct tagRPCCall)) { |
| /* close connection */ | /* close connection */ |
| schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], | schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], |
| Line 295 rxPacket(sched_task_t *task) | Line 315 rxPacket(sched_task_t *task) |
| 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); |
| rpc = (struct tagRPCCall*) AIT_GET_BUF(&c->cli_buf); | rpc = (struct tagRPCCall*) AIT_GET_BUF(&c->cli_buf); |
| buf = AIT_GET_BUF(&c->cli_buf) + sizeof(struct tagRPCCall); | buf = AIT_GET_BUF(&c->cli_buf); |
| len = estlen - sizeof(struct tagRPCCall); | len = estlen; |
| } | } |
| if (len) { | /* get next part of packet */ |
| /* get next part of packet */ | memset(buf, 0, len); |
| memset(buf, 0, len); | pfd.fd = TASK_FD(task); |
| pfd.events = POLLIN | POLLPRI; | |
| for (; len > 0; len -= rlen, buf += rlen) { | |
| if ((rlen = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || | |
| pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) { | |
| if (rlen) | |
| LOGERR; | |
| else | |
| rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond"); | |
| schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], | |
| TASK_ARG(task), 0, NULL, 0); | |
| return NULL; | |
| } | |
| rlen = recv(TASK_FD(task), buf, len, 0); | rlen = recv(TASK_FD(task), buf, len, 0); |
| if (rlen == -1) { | if (rlen == -1) { |
| /* close connection */ | /* close connection */ |
| Line 380 txUDPPacket(sched_task_t *task) | Line 412 txUDPPacket(sched_task_t *task) |
| rpc_cli_t *c = TASK_ARG(task); | rpc_cli_t *c = TASK_ARG(task); |
| rpc_srv_t *s = c->cli_parent; | rpc_srv_t *s = c->cli_parent; |
| rpc_func_t *f = NULL; | rpc_func_t *f = NULL; |
| u_char buf[USHRT_MAX] = { 0 }; | 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, wlen = sizeof(struct tagRPCCall); |
| struct timespec ts = { DEF_RPC_TIMEOUT, 0 }; | struct timespec ts = { DEF_RPC_TIMEOUT, 0 }; |
| Line 389 txUDPPacket(sched_task_t *task) | Line 421 txUDPPacket(sched_task_t *task) |
| schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], | schedTimer(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], |
| TASK_ARG(task), ts, TASK_ARG(task), 0); | TASK_ARG(task), ts, TASK_ARG(task), 0); |
| /* copy RPC header */ | |
| memcpy(buf, TASK_DATA(task), wlen); | |
| if (rpc->call_argc) { | if (rpc->call_argc) { |
| f = rpc_srv_getCall(s, ntohs(rpc->call_tag)); | f = rpc_srv_getCall(s, ntohs(rpc->call_tag)); |
| if (!f) { | if (!f) { |
| Line 402 txUDPPacket(sched_task_t *task) | Line 431 txUDPPacket(sched_task_t *task) |
| } else { | } else { |
| 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, sizeof buf - wlen, RPC_RETVARS(c)); | ret = ait_vars2buffer(buf + wlen, AIT_LEN(&c->cli_buf) - wlen, |
| RPC_RETVARS(c)); | |
| /* Free return values */ | /* Free return values */ |
| ait_freeVars(&c->cli_vars); | ait_freeVars(&c->cli_vars); |
| if (ret == -1) { | if (ret == -1) { |