--- libaitio/src/sock.c	2014/02/08 22:06:17	1.14
+++ libaitio/src/sock.c	2014/04/29 01:26:21	1.14.4.1
@@ -3,7 +3,7 @@
 *  by Michael Pounov <misho@elwix.org>
 *
 * $Author: misho $
-* $Id: sock.c,v 1.14 2014/02/08 22:06:17 misho Exp $
+* $Id: sock.c,v 1.14.4.1 2014/04/29 01:26:21 misho Exp $
 *
 **************************************************************************
 The ELWIX and AITNET software is distributed under the following
@@ -211,16 +211,19 @@ io_rxNet(sched_task_t *task)
 	else {
 		rlen = recvfrom(TASK_FD(task), AIT_GET_BUF(&cli->cli_buf[0]), 
 				AIT_LEN(&cli->cli_buf[0]), 0, &sa.sa, &salen);
-		if (e_addrcmp(&cli->cli_addr, &sa, 42))
-			goto end;
+		if (e_addrcmp(&cli->cli_addr, &sa, 42)) {
+			schedReadSelf(task);
+			taskExit(task, NULL);
+		}
 	}
 	if (rlen < 1)
 		schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
-	else
+	else {
 		schedEvent(TASK_ROOT(task), io_txPty, cli, cli->cli_pty, 
 				AIT_GET_BUF(&cli->cli_buf[0]), rlen);
-end:
-	schedReadSelf(task);
+		schedReadSelf(task);
+	}
+
 	taskExit(task, NULL);
 }
 
@@ -235,11 +238,12 @@ io_rxPty(sched_task_t *task)
 	rlen = read(TASK_FD(task), AIT_GET_BUF(&cli->cli_buf[1]), AIT_LEN(&cli->cli_buf[1]));
 	if (rlen < 1)
 		schedEvent(TASK_ROOT(task), io_closeClient, cli, 0, NULL, 0);
-	else
+	else {
 		schedEvent(TASK_ROOT(task), io_txNet, cli, cli->cli_fd, 
 				AIT_GET_BUF(&cli->cli_buf[1]), rlen);
+		schedReadSelf(task);
+	}
 
-	schedReadSelf(task);
 	taskExit(task, NULL);
 }