* more checks on net system

git-svn-id: http://opensvn.csie.org/pttbbs/trunk/pttbbs@4606 63ad8ddf-47c3-0310-b6dd-a9e9d9715204

1 files changed, 16 insertions, 4 deletions

diff --git a/common/sys/net.c b/common/sys/net.c
index 232d5bb1..efeda16a 100644
--- a/common/sys/net.c
+++ b/common/sys/net.c
@@ -188,9 +188,11 @@ int toread(int fd, void *buf, int len)
 {
     int     l;
     for( l = 0 ; len > 0 ; )
-	if( (l = read(fd, buf, len)) <= 0 )
+	if( (l = read(fd, buf, len)) <= 0 ) {
+	    if (errno == EINTR || errno == EAGAIN)
+		continue;
 	    return -1;
-	else{
+	}else{
 	    buf += l;
 	    len -= l;
 	}
@@ -204,9 +206,11 @@ int towrite(int fd, const void *buf, int len)
 {
     int     l;
     for( l = 0 ; len > 0 ; )
-	if( (l = write(fd, buf, len)) <= 0 )
+	if( (l = write(fd, buf, len)) <= 0){
+	    if (errno == EINTR || errno == EAGAIN)
+		continue;
 	    return -1;
-	else{
+	}else{
 	    buf += l;
 	    len -= l;
 	}
@@ -293,9 +297,17 @@ int recv_remote_fd(int tunnel, const char *tunnel_path)
     }
 
     cmsg = CMSG_FIRSTHDR(&msg);
+    if (!cmsg) {
+	// kernel indicates there's no ancillary data
+	return -1;
+    }
+
     assert(cmsg->cmsg_type == SCM_RIGHTS);
     if (cmsg->cmsg_type != SCM_RIGHTS)
+    {
+	// invalid message!?
 	return -1;
+    }
 
     return *(int*)CMSG_DATA(cmsg);
 }