--- libaitrpc/src/aitrpc.c	2013/08/23 13:53:15	1.11
+++ libaitrpc/src/aitrpc.c	2014/01/28 14:05:43	1.15
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@openbsd-bg.org>
 *
 * $Author: misho $
-* $Id: aitrpc.c,v 1.11 2013/08/23 13:53:15 misho Exp $
+* $Id: aitrpc.c,v 1.15 2014/01/28 14:05:43 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -12,7 +12,7 @@ terms:
 All of the documentation and software included in the ELWIX and AITNET
 Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
 
-Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013
+Copyright 2004 - 2014
 	by Michael Pounov <misho@elwix.org>.  All rights reserved.
 
 Redistribution and use in source and binary forms, with or without
@@ -149,7 +149,7 @@ rpc_addPktSession(rpc_sess_t *p, rpc_sess_t *s)
  * @sa = check client address, if you use udp protocol
  * @buf = buffer
  * @blen = buffer length
- * return: -1 error or >-1 readed bytes into buffer
+ * return: -1 error, 0 EOF or or >0 readed bytes into buffer
  */
 ssize_t
 rpc_Read(int sock, int type, int flags, sockaddr_t * __restrict sa, 
@@ -166,39 +166,33 @@ rpc_Read(int sock, int type, int flags, sockaddr_t * _
 	memset(&sa2, 0, sizeof sa2);
 	salen = sa2.ss.ss_len = sizeof(sockaddr_t);
 
-	if (type == SOCK_STREAM) {
-		for (len = blen; len > 0; len -= ret, buf += ret) {
-			if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
-					pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
+	for (len = blen; len > 0; len -= ret, buf += ret) {
+		if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
+				pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
+			if (ret)
 				LOGERR;
-				return -1;
-			} else
-				ret = recv(sock, buf, len, flags);
-			if (ret == -1) {
-				LOGERR;
-				return -1;
-			}
+			else
+				rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
+			return -1;
 		}
-		if (ret > -1)
-			ret = blen - len;
-	} else {
-		do {
-			if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
-					pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
-				LOGERR;
+		if (type == SOCK_STREAM)
+			ret = recv(sock, buf, len, flags);
+		else {
+			ret = recvfrom(sock, buf, len, flags, &sa2.sa, &salen);
+			if (ret > -1 && sa && e_addrcmp(sa, &sa2, 42)) {
+				rpc_SetErr(ERPCMISMATCH, 
+						"Received RPC response from unknown address");
 				return -1;
-			} else
-				ret = recvfrom(sock, buf, blen, flags, &sa2.sa, &salen);
-			if (ret < 0) {
-				LOGERR;
-				return -1;
 			}
-			if (e_addrcmp(sa, &sa2, 42))
-				rpc_SetErr(ERPCMISMATCH, "Received RPC response from unknown address");
-			else
-				break;
-		} while (42);
+		}
+		if (ret == -1) {
+			LOGERR;
+			return -1;
+		}
+		if (!ret)	/* EOF */
+			return 0;
 	}
+	ret = blen - len;
 
 	return ret;
 }
@@ -212,7 +206,7 @@ rpc_Read(int sock, int type, int flags, sockaddr_t * _
  * @sa = send to client address, if you use udp protocol
  * @buf = buffer
  * @blen = buffer length
- * return: -1 error or >-1 written bytes into buffer
+ * return: -1 error, 0 EOF or >0 written bytes into buffer
  */
 ssize_t
 rpc_Write(int sock, int type, int flags, sockaddr_t * __restrict sa, 
@@ -223,37 +217,36 @@ rpc_Write(int sock, int type, int flags, sockaddr_t * 
 
 	pfd.fd = sock;
 	pfd.events = POLLOUT;
-
-	if (type == SOCK_STREAM) {
-		for (len = blen; len > 0; len -= ret, buf += ret) {
-			if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
-					pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
-				LOGERR;
-				return -1;
-			} else
-				ret = send(sock, buf, len, flags);
-			if (ret == -1) {
-				LOGERR;
-				return -1;
-			}
-		}
-		if (ret > -1)
-			ret = blen - len;
-	} else if (sa) {
+	for (len = blen; len > 0; len -= ret, buf += ret) {
 		if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 || 
 				pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
+#if 0
+			if (!ret)
+				continue;
+#endif
+			if (ret > 0)
+				rpc_SetErr(EPIPE, "Disconnected RPC session\n");
+			else
+				LOGERR;
+			return -1;
+		}
+		if (type == SOCK_STREAM)
+			ret = send(sock, buf, len, flags);
+		else if (sa)
+			ret = sendto(sock, buf, len, flags, &sa->sa, sa->sa.sa_len);
+		else {
+			rpc_SetErr(EINVAL, "Invalid argument(s)!");
+			return -1;
+		}
+		if (ret == -1) {
 			LOGERR;
 			return -1;
-		} else
-			ret = sendto(sock, buf, blen, flags, &sa->sa, sa->sa.sa_len);
-	} else {
-		rpc_SetErr(EINVAL, "Invalid argument(s)!");
-		return -1;
+		}
+		if (!ret)	/* EOF */
+			return 0;
 	}
-	if (ret == -1) {
-		LOGERR;
-		return -1;
-	} else if (ret != blen) {
+	ret = blen - len;
+	if (ret != blen) {
 		rpc_SetErr(EPROCUNAVAIL, "RPC request, should be send %d bytes, "
 				"really sended %d bytes", len, ret);
 		return -1;