--- libaitrpc/src/srv.c	2015/06/29 22:39:33	1.26.2.9
+++ libaitrpc/src/srv.c	2015/07/02 17:52:53	1.26.2.11
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: srv.c,v 1.26.2.9 2015/06/29 22:39:33 misho Exp $
+* $Id: srv.c,v 1.26.2.11 2015/07/02 17:52:53 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -197,14 +197,14 @@ txPacket(sched_task_t *task)
 			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);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 		} else if (rpc_pktFreeSpace(c) > s->srv_netbuf) {
 			rpc_SetErr(EMSGSIZE, "Message too long");
 
 			rpc->call_argc ^= rpc->call_argc;
-			rpc->call_rep.ret = RPC_ERROR(-1);
-			rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 		} else {
 			rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
 			/* Go Encapsulate variables */
@@ -214,8 +214,8 @@ txPacket(sched_task_t *task)
 				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);
+				RPC_SET_RETURN(rpc, -1);
+				RPC_SET_ERRNO(rpc, rpc_Errno);
 			} else
 				wlen += ret;
 		}
@@ -263,8 +263,8 @@ execCall(sched_task_t *task)
 			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);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 			taskExit(task, NULL);
 		}
 	} else
@@ -274,15 +274,15 @@ execCall(sched_task_t *task)
 		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);
+		RPC_SET_RETURN(rpc, -1);
+		RPC_SET_ERRNO(rpc, rpc_Errno);
 	} else {
 		/* if client doesn't want reply */
-		rpc->call_rep.ret = RPC_ERROR(rpc_srv_execCall(c, rpc, f->func_name, arr));
+		RPC_SET_RETURN(rpc, rpc_srv_execCall(c, rpc, f->func_name, arr));
 		if (rpc->call_rep.ret == htonl(-1)) {
 			if (!rpc->call_rep.eno) {
 				LOGERR;
-				rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
+				RPC_SET_ERRNO(rpc, rpc_Errno);
 			}
 			rpc->call_argc ^= rpc->call_argc;
 			ait_freeVars(&c->cli_vars);
@@ -313,6 +313,7 @@ rxPacket(sched_task_t *task)
 	u_short crc;
 #endif
 	u_char *buf = AIT_GET_BUF(&c->cli_buf);
+	u_char b[sizeof(struct tagRPCCall)];
 	struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
 #ifdef TCP_SESSION_TIMEOUT
 	struct timespec ts = { DEF_RPC_TIMEOUT, 0 };
@@ -322,6 +323,11 @@ rxPacket(sched_task_t *task)
 			 TASK_ARG(task), ts, TASK_ARG(task), 0);
 #endif
 
+	/* prepare rx */
+	len = recv(TASK_FD(task), b, sizeof b, MSG_PEEK);
+	if (len == sizeof b)
+		rlen = ntohl(((struct tagRPCCall*) b)->call_len);
+
 	rlen = recv(TASK_FD(task), buf, rlen, 0);
 	if (rlen == -1) {
 		/* close connection */
@@ -333,8 +339,8 @@ rxPacket(sched_task_t *task)
 		rpc_SetErr(ERPCMISMATCH, "Short RPC packet");
 
 		rpc->call_argc ^= rpc->call_argc;
-		rpc->call_rep.ret = RPC_ERROR(-1);
-		rpc->call_rep.eno = RPC_ERROR(errno);
+		RPC_SET_RETURN(rpc, -1);
+		RPC_SET_ERRNO(rpc, rpc_Errno);
 		goto err;
 	} else
 		len = ntohl(rpc->call_len);
@@ -342,8 +348,8 @@ rxPacket(sched_task_t *task)
 		rpc_SetErr(ERPCMISMATCH, "Short RPC packet");
 
 		rpc->call_argc ^= rpc->call_argc;
-		rpc->call_rep.ret = RPC_ERROR(-1);
-		rpc->call_rep.eno = RPC_ERROR(errno);
+		RPC_SET_RETURN(rpc, -1);
+		RPC_SET_ERRNO(rpc, rpc_Errno);
 		goto err;
 	}
 
@@ -361,8 +367,8 @@ rxPacket(sched_task_t *task)
 		rpc_SetErr(ERPCMISMATCH, "Bad CRC RPC packet");
 
 		rpc->call_argc ^= rpc->call_argc;
-		rpc->call_rep.ret = RPC_ERROR(-1);
-		rpc->call_rep.eno = RPC_ERROR(errno);
+		RPC_SET_RETURN(rpc, -1);
+		RPC_SET_ERRNO(rpc, rpc_Errno);
 		goto err;
 	}
 #endif
@@ -372,8 +378,8 @@ rxPacket(sched_task_t *task)
 		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);
+		RPC_SET_RETURN(rpc, -1);
+		RPC_SET_ERRNO(rpc, rpc_Errno);
 		goto err;
 	}
 
@@ -458,14 +464,14 @@ txUDPPacket(sched_task_t *task)
 			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);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 		} else if (rpc_pktFreeSpace(c) > s->srv_netbuf) {
 			rpc_SetErr(EMSGSIZE, "Message too long");
 
 			rpc->call_argc ^= rpc->call_argc;
-			rpc->call_rep.ret = RPC_ERROR(-1);
-			rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 		} else {
 			rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
 			/* Go Encapsulate variables */
@@ -475,8 +481,8 @@ txUDPPacket(sched_task_t *task)
 				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);
+				RPC_SET_RETURN(rpc, -1);
+				RPC_SET_ERRNO(rpc, rpc_Errno);
 			} else
 				wlen += ret;
 		}
@@ -598,14 +604,14 @@ txRAWPacket(sched_task_t *task)
 			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);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 		} else if (rpc_pktFreeSpace(c) > s->srv_netbuf) {
 			rpc_SetErr(EMSGSIZE, "Message too long");
 
 			rpc->call_argc ^= rpc->call_argc;
-			rpc->call_rep.ret = RPC_ERROR(-1);
-			rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 		} else {
 			rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
 			/* Go Encapsulate variables */
@@ -613,9 +619,10 @@ txRAWPacket(sched_task_t *task)
 					RPC_RETVARS(c));
 			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);
+				RPC_SET_RETURN(rpc, -1);
+				RPC_SET_ERRNO(rpc, rpc_Errno);
 			} else
 				wlen += ret;
 		}
@@ -758,14 +765,14 @@ txBPFPacket(sched_task_t *task)
 			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);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 		} else if (rpc_pktFreeSpace(c) > s->srv_netbuf) {
 			rpc_SetErr(EMSGSIZE, "Message too long");
 
 			rpc->call_argc ^= rpc->call_argc;
-			rpc->call_rep.ret = RPC_ERROR(-1);
-			rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 		} else {
 			rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
 			/* Go Encapsulate variables */
@@ -775,8 +782,8 @@ txBPFPacket(sched_task_t *task)
 				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);
+				RPC_SET_RETURN(rpc, -1);
+				RPC_SET_ERRNO(rpc, rpc_Errno);
 			} else
 				wlen += ret;
 		}
@@ -921,14 +928,14 @@ txEXTPacket(sched_task_t *task)
 			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);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 		} else if (rpc_pktFreeSpace(c) > s->srv_netbuf) {
 			rpc_SetErr(EMSGSIZE, "Message too long");
 
 			rpc->call_argc ^= rpc->call_argc;
-			rpc->call_rep.ret = RPC_ERROR(-1);
-			rpc->call_rep.eno = RPC_ERROR(rpc_Errno);
+			RPC_SET_RETURN(rpc, -1);
+			RPC_SET_ERRNO(rpc, rpc_Errno);
 		} else {
 			rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
 			/* Go Encapsulate variables */
@@ -938,8 +945,8 @@ txEXTPacket(sched_task_t *task)
 				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);
+				RPC_SET_RETURN(rpc, -1);
+				RPC_SET_ERRNO(rpc, rpc_Errno);
 			} else
 				wlen += ret;
 		}
@@ -1111,7 +1118,7 @@ rxBLOB(sched_task_t *task)
 			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);
+				RPC_SET_BLOB_RET(&blob, -1);
 				break;
 			} else
 				blob.hdr_len = htonl(b->blob_len);
@@ -1122,7 +1129,7 @@ rxBLOB(sched_task_t *task)
 				rpc_srv_blobUnmap(b);
 			} else {
 				blob.hdr_cmd = error;
-				blob.hdr_ret = RPC_ERROR(-1);
+				RPC_SET_BLOB_RET(&blob, -1);
 			}
 			break;
 		case set:
@@ -1136,19 +1143,19 @@ rxBLOB(sched_task_t *task)
 				rpc_srv_blobUnmap(b);
 			} else {
 				blob.hdr_cmd = error;
-				blob.hdr_ret = RPC_ERROR(-1);
+				RPC_SET_BLOB_RET(&blob, -1);
 			}
 			break;
 		case unset:
 			if (rpc_srv_unregisterBLOB(s, ntohl(blob.hdr_var)) == -1) {
 				blob.hdr_cmd = error;
-				blob.hdr_ret = RPC_ERROR(-1);
+				RPC_SET_BLOB_RET(&blob, -1);
 			}
 			break;
 		default:
 			rpc_SetErr(EPROCUNAVAIL, "Unsupported BLOB command %d", blob.hdr_cmd);
 			blob.hdr_cmd = error;
-			blob.hdr_ret = RPC_ERROR(-1);
+			RPC_SET_BLOB_RET(&blob, -1);
 	}
 
 end: