--- libaitrpc/src/srv.c	2024/02/26 13:30:59	1.30.2.6
+++ libaitrpc/src/srv.c	2024/02/26 18:25:32	1.30.2.10
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: srv.c,v 1.30.2.6 2024/02/26 13:30:59 misho Exp $
+* $Id: srv.c,v 1.30.2.10 2024/02/26 18:25:32 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -87,7 +87,8 @@ rpc_freeCli(rpc_cli_t * __restrict c)
 {
 	rpc_srv_t *s = c->cli_parent;
 
-	schedCancelby(s->srv_root, taskMAX, CRITERIA_ARG, c, NULL);
+	if (s->srv_proto == SOCK_STREAM)
+		schedCancelby(s->srv_root, taskMAX, CRITERIA_ARG, c, NULL);
 
 	/* free buffer */
 	AIT_FREE_VAL(&c->cli_buf);
@@ -323,8 +324,15 @@ rxPacket(sched_task_t *task)
 
 	/* prepare rx */
 	len = recv(TASK_FD(task), &b, sizeof b, MSG_PEEK);
-	if (len == sizeof b)
+	if (len < 1) {
+		/* close connection */
+		schedEvent(TASK_ROOT(task), cbProto[s->srv_proto][CB_CLOSECLIENT], 
+				TASK_ARG(task), 0, NULL, 0);
+		taskExit(task, NULL);
+	} else if (len == sizeof b)
 		rlen = ntohl(b.call_len);
+	else
+		goto end;
 
 	rlen = recv(TASK_FD(task), buf, rlen, 0);
 	if (rlen == -1) {
@@ -390,7 +398,7 @@ err:
 	if (!noreply && (rpc->call_io & RPC_REQ))
 		schedWrite(TASK_ROOT(task), cbProto[s->srv_proto][CB_TXPACKET], 
 				TASK_ARG(task), TASK_FD(task), rpc, len);
-
+end:
 	/* lets get next packet */
 	schedReadSelf(task);
 	taskExit(task, NULL);
@@ -553,7 +561,7 @@ rxUDPPacket(sched_task_t *task)
 
 	c = _allocClient(srv, &sa);
 	if (!c) {
-		EVERBOSE(1, "RPC client quota exceeded! Connection will be shutdown!\n");
+		EVERBOSE(1, "RPC client quota exceeded!");
 		usleep(2000);	/* blocked client delay */
 		goto end;
 	} else {
@@ -1958,4 +1966,26 @@ rpc_srv_initServerExt(u_char InstID, int netBuf, int f
 	rpc_register_srvPing(srv);
 
 	return srv;
+}
+
+/*
+ * rpc_srv_Return() - Prepare IPC return answer to RPC client
+ *
+ * @c = RPC client 
+ * return: number of arguments in response
+ */
+int
+rpc_srv_Return(rpc_cli_t *c)
+{
+	rpc_srv_t *s = c->cli_parent;
+	u_char *buf = AIT_GET_BUF(&c->cli_buf);
+	struct tagRPCCall *rpc = (struct tagRPCCall*) buf;
+
+	if (!RPC_CHK_NOREPLY(rpc)) {
+		rpc->call_argc = (u_char) array_Size(RPC_RETVARS(c));
+		schedWrite(s->srv_root, cbProto[s->srv_proto][CB_TXPACKET], c, c->cli_sock, rpc, 0);
+	} else
+		rpc->call_argc ^= rpc->call_argc;
+
+	return rpc->call_argc;
 }