Mercurial > dropbear
diff common-session.c @ 416:a01c0c8e543a channel-fix
Improve behaviour when flushing out after a process has exited.
author | Matt Johnston <matt@ucc.asn.au> |
---|---|
date | Fri, 09 Feb 2007 10:43:16 +0000 |
parents | 70caa99bfe3a |
children | 9c61e7af0156 |
line wrap: on
line diff
--- a/common-session.c Tue Feb 06 16:00:18 2007 +0000 +++ b/common-session.c Fri Feb 09 10:43:16 2007 +0000 @@ -61,6 +61,12 @@ ses.connecttimeout = 0; + if (pipe(ses.signal_pipe) < 0) { + dropbear_exit("signal pipe failed"); + } + setnonblocking(ses.signal_pipe[0]); + setnonblocking(ses.signal_pipe[1]); + kexfirstinitialise(); /* initialise the kex state */ ses.writepayload = buf_new(MAX_TRANS_PAYLOAD_LEN); @@ -108,7 +114,6 @@ ses.allowprivport = 0; - TRACE(("leave session_init")) } @@ -132,6 +137,10 @@ FD_SET(ses.sock, &writefd); } } + + /* We get woken up when signal handlers write to this pipe. + SIGCHLD in svr-chansession is the only one currently. */ + FD_SET(ses.signal_pipe[0], &readfd); /* set up for channels which require reading/writing */ if (ses.dataallowed) { @@ -155,6 +164,14 @@ FD_ZERO(&writefd); FD_ZERO(&readfd); } + + /* We'll just empty out the pipe if required. We don't do + any thing with the data, since the pipe's purpose is purely to + wake up the select() above. */ + if (FD_ISSET(ses.signal_pipe[0], &readfd)) { + char x; + while (read(ses.signal_pipe[0], &x, 1) > 0) {} + } /* check for auth timeout, rekeying required etc */ checktimeouts();