# HG changeset patch
# User Matt Johnston <matt@ucc.asn.au>
# Date 1170490195 0
# Node ID e7c1a77d2921ab08da6cd768a56c9fd522becb68
# Parent  d6b8815a9b5e902e5a18575321cd894de8d1f545
disapproval of revision '2186a678a5f79ef432f6593a10e3e13df2a313d5'

diff -r d6b8815a9b5e -r e7c1a77d2921 common-channel.c
--- a/common-channel.c	Sun Jun 11 16:19:32 2006 +0000
+++ b/common-channel.c	Sat Feb 03 08:09:55 2007 +0000
@@ -181,7 +181,6 @@
 
 	struct Channel *channel;
 	unsigned int i;
-	int ret;
 
 	/* iterate through all the possible channels */
 	for (i = 0; i < ses.chansize; i++) {
@@ -205,8 +204,7 @@
 
 		/* if we can read from the writefd, it might be closed, so we try to
 		 * see if it has errors */
-		if (IS_DROPBEAR_SERVER && channel->writefd >= 0 
-				&& channel->writefd != channel->readfd
+		if (channel->writefd >= 0 && channel->writefd != channel->readfd
 				&& FD_ISSET(channel->writefd, readfds)) {
 			if (channel->initconn) {
 				/* Handling for "in progress" connection - this is needed
@@ -261,27 +259,27 @@
 				channel->writebuf,
 				channel->writebuf ? 0 : cbuf_getused(channel->extrabuf)))
 
-	/* server chansession channels are special, since readfd mightn't
-	 * close in the case of "sleep 4 & echo blah" until the sleep is up */
-	if (channel->type->checkclose) {
-		if (channel->type->checkclose(channel)) {
-			closewritefd(channel);
-			closereadfd(channel, channel->readfd);
-			closereadfd(channel, channel->errfd);
+	if (!channel->sentclosed) {
+
+		/* check for exited - currently only used for server sessions,
+		 * if the shell has exited etc */
+		if (channel->type->checkclose) {
+			if (channel->type->checkclose(channel)) {
+				closewritefd(channel);
+			}
 		}
-	}
 
-	if (!channel->senteof
-		&& channel->readfd == FD_CLOSED 
-		&& (channel->extrabuf == NULL || channel->errfd == FD_CLOSED)) {
-		send_msg_channel_eof(channel);
-	}
+		if (!channel->senteof
+			&& channel->readfd == FD_CLOSED 
+			&& (channel->extrabuf != NULL || channel->errfd == FD_CLOSED)) {
+			send_msg_channel_eof(channel);
+		}
 
-	if (!channel->sentclosed
-		&& channel->writefd == FD_CLOSED
-		&& channel->readfd == FD_CLOSED
-		&& (channel->extrabuf == NULL || channel->errfd == FD_CLOSED)) {
-		send_msg_channel_close(channel);
+		if (channel->writefd == FD_CLOSED
+			&& channel->readfd == FD_CLOSED
+			&& (channel->extrabuf != NULL || channel->errfd == FD_CLOSED)) {
+			send_msg_channel_close(channel);
+		}
 	}
 
 	/* When either party wishes to terminate the channel, it sends
@@ -446,22 +444,20 @@
 			}
 		}
 
+		/* For checking FD status (ie closure etc) - we don't actually
+		 * read data from writefd */
 		TRACE(("writefd = %d, readfd %d, errfd %d, bufused %d", 
 					channel->writefd, channel->readfd,
 					channel->errfd,
 					cbuf_getused(channel->writebuf) ))
-
-		/* For checking FD status (ie closure etc) - we don't actually
-		 * read data from writefd. We don't want to do this for the client,
-		 * since redirection to /dev/null will make it spin in the select */
-		if (IS_DROPBEAR_SERVER && channel->writefd >= 0 
-				&& channel->writefd != channel->readfd) {
+		if (channel->writefd >= 0 && channel->writefd != channel->readfd) {
 			FD_SET(channel->writefd, readfds);
 		}
 
-		/* Stuff from the wire */
+		/* Stuff from the wire, to local program/shell/user etc */
 		if ((channel->writefd >= 0 && cbuf_getused(channel->writebuf) > 0 )
 				|| channel->initconn) {
+
 				FD_SET(channel->writefd, writefds);
 		}
 
diff -r d6b8815a9b5e -r e7c1a77d2921 svr-chansession.c
--- a/svr-chansession.c	Sun Jun 11 16:19:32 2006 +0000
+++ b/svr-chansession.c	Sat Feb 03 08:09:55 2007 +0000
@@ -67,7 +67,8 @@
 extern char** environ;
 
 static int sesscheckclose(struct Channel *channel) {
-	return channel->writefd == -1;
+	struct ChanSess *chansess = (struct ChanSess*)channel->typedata;
+	return chansess->exit.exitpid >= 0;
 }
 
 /* Handler for childs exiting, store the state for return to the client */