1 files changed, 40 insertions, 35 deletions

diff --git a/clientloop.c b/clientloop.c
index 141e0fff5..e5ea74e26 100644
--- a/clientloop.c
+++ b/clientloop.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: clientloop.c,v 1.213 2009/07/05 19:28:33 stevesk Exp $ */
+/* $OpenBSD: clientloop.c,v 1.218 2010/01/28 00:21:18 djm Exp $ */
 /*
  * Author: Tatu Ylonen <ylo@cs.hut.fi>
  * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
@@ -125,7 +125,7 @@ extern int stdin_null_flag;
 extern int no_shell_flag;
 
 /* Control socket */
-extern int muxserver_sock;
+extern int muxserver_sock; /* XXX use mux_client_cleanup() instead */
 
 /*
  * Name of the host we are connecting to.  This is the name given on the
@@ -134,6 +134,9 @@ extern int muxserver_sock;
  */
 extern char *host;
 
+/* Force TTY allocation */
+extern int force_tty_flag;
+
 /*
  * Flag to indicate that we have received a window change signal which has
  * not yet been processed.  This will cause a message indicating the new
@@ -147,7 +150,7 @@ static volatile sig_atomic_t received_signal = 0;
 static int in_non_blocking_mode = 0;
 
 /* Common data for the client loop code. */
-static volatile sig_atomic_t quit_pending; /* Set non-zero to quit the loop. */
+volatile sig_atomic_t quit_pending; /* Set non-zero to quit the loop. */
 static int escape_char1;        /* Escape character. (proto1 only) */
 static int escape_pending1;     /* Last character was an escape (proto1 only) */
 static int last_was_cr;         /* Last character was a newline. */
@@ -165,6 +168,8 @@ static int session_closed = 0;	/* In SSH2: login session closed. */
 static void client_init_dispatch(void);
 int     session_ident = -1;
 
+int     session_resumed = 0;
+
 /* Track escape per proto2 channel */
 struct escape_filter_ctx {
         int escape_pending;
@@ -563,9 +568,6 @@ client_wait_until_can_do_something(fd_set **readsetp, fd_set **writesetp,
         if (packet_have_data_to_write())
                 FD_SET(connection_out, *writesetp);
 
-        if (muxserver_sock != -1)
-                FD_SET(muxserver_sock, *readsetp);
-
         /*
          * Wait for something to happen.  This will suspend the process until
          * some selected descriptor can be read, written, or has some other
@@ -612,7 +614,7 @@ client_suspend_self(Buffer *bin, Buffer *bout, Buffer *berr)
                 atomicio(vwrite, fileno(stderr), buffer_ptr(berr),
                     buffer_len(berr));
 
-        leave_raw_mode();
+        leave_raw_mode(force_tty_flag);
 
         /*
          * Free (and clear) the buffer to reduce the amount of data that gets
@@ -633,7 +635,7 @@ client_suspend_self(Buffer *bin, Buffer *bout, Buffer *berr)
         buffer_init(bout);
         buffer_init(berr);
 
-        enter_raw_mode();
+        enter_raw_mode(force_tty_flag);
 }
 
 static void
@@ -694,7 +696,7 @@ client_status_confirm(int type, Channel *c, void *ctx)
 
         /* XXX supress on mux _client_ quietmode */
         tochan = options.log_level >= SYSLOG_LEVEL_ERROR &&
-            c->ctl_fd != -1 && c->extended_usage == CHAN_EXTENDED_WRITE;
+            c->ctl_chan != -1 && c->extended_usage == CHAN_EXTENDED_WRITE;
 
         if (type == SSH2_MSG_CHANNEL_SUCCESS) {
                 debug2("%s request accepted on channel %d",
@@ -776,7 +778,7 @@ process_cmdline(void)
         bzero(&fwd, sizeof(fwd));
         fwd.listen_host = fwd.connect_host = NULL;
 
-        leave_raw_mode();
+        leave_raw_mode(force_tty_flag);
         handler = signal(SIGINT, SIG_IGN);
         cmd = s = read_passphrase("\r\nssh> ", RP_ECHO);
         if (s == NULL)
@@ -838,6 +840,7 @@ process_cmdline(void)
         while (isspace(*++s))
                 ;
 
+        /* XXX update list of forwards in options */
         if (delete) {
                 cancel_port = 0;
                 cancel_host = hpdelim(&s);      /* may be NULL */
@@ -879,7 +882,7 @@ process_cmdline(void)
 
 out:
         signal(SIGINT, handler);
-        enter_raw_mode();
+        enter_raw_mode(force_tty_flag);
         if (cmd)
                 xfree(cmd);
         if (fwd.listen_host != NULL)
@@ -935,7 +938,7 @@ process_escapes(Channel *c, Buffer *bin, Buffer *bout, Buffer *berr,
                                     escape_char);
                                 buffer_append(berr, string, strlen(string));
 
-                                if (c && c->ctl_fd != -1) {
+                                if (c && c->ctl_chan != -1) {
                                         chan_read_failed(c);
                                         chan_write_failed(c);
                                         return 0;
@@ -945,7 +948,7 @@ process_escapes(Channel *c, Buffer *bin, Buffer *bout, Buffer *berr,
 
                         case 'Z' - 64:
                                 /* XXX support this for mux clients */
-                                if (c && c->ctl_fd != -1) {
+                                if (c && c->ctl_chan != -1) {
  noescape:
                                         snprintf(string, sizeof string,
                                             "%c%c escape not available to "
@@ -990,7 +993,7 @@ process_escapes(Channel *c, Buffer *bin, Buffer *bout, Buffer *berr,
                                 continue;
 
                         case '&':
-                                if (c && c->ctl_fd != -1)
+                                if (c && c->ctl_chan != -1)
                                         goto noescape;
                                 /*
                                  * Detach the program (continue to serve
@@ -998,7 +1001,7 @@ process_escapes(Channel *c, Buffer *bin, Buffer *bout, Buffer *berr,
                                  * more new connections).
                                  */
                                 /* Restore tty modes. */
-                                leave_raw_mode();
+                                leave_raw_mode(force_tty_flag);
 
                                 /* Stop listening for new connections. */
                                 channel_stop_listening();
@@ -1041,7 +1044,7 @@ process_escapes(Channel *c, Buffer *bin, Buffer *bout, Buffer *berr,
                                 continue;
 
                         case '?':
-                                if (c && c->ctl_fd != -1) {
+                                if (c && c->ctl_chan != -1) {
                                         snprintf(string, sizeof string,
 "%c?\r\n\
 Supported escape sequences:\r\n\
@@ -1090,7 +1093,7 @@ Supported escape sequences:\r\n\
                                 continue;
 
                         case 'C':
-                                if (c && c->ctl_fd != -1)
+                                if (c && c->ctl_chan != -1)
                                         goto noescape;
                                 process_cmdline();
                                 continue;
@@ -1293,7 +1296,7 @@ client_channel_closed(int id, void *arg)
 {
         channel_cancel_cleanup(id);
         session_closed = 1;
-        leave_raw_mode();
+        leave_raw_mode(force_tty_flag);
 }
 
 /*
@@ -1326,8 +1329,6 @@ client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
         connection_in = packet_get_connection_in();
         connection_out = packet_get_connection_out();
         max_fd = MAX(connection_in, connection_out);
-        if (muxserver_sock != -1)
-                max_fd = MAX(max_fd, muxserver_sock);
 
         if (!compat20) {
                 /* enable nonblocking unless tty */
@@ -1366,7 +1367,7 @@ client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
         signal(SIGWINCH, window_change_handler);
 
         if (have_pty)
-                enter_raw_mode();
+                enter_raw_mode(force_tty_flag);
 
         if (compat20) {
                 session_ident = ssh2_chan_id;
@@ -1454,12 +1455,6 @@ client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
                 /* Buffer input from the connection.  */
                 client_process_net_input(readset);
 
-                /* Accept control connections.  */
-                if (muxserver_sock != -1 &&FD_ISSET(muxserver_sock, readset)) {
-                        if (muxserver_accept_control())
-                                quit_pending = 1;
-                }
-
                 if (quit_pending)
                         break;
 
@@ -1473,6 +1468,14 @@ client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
                         client_process_output(writeset);
                 }
 
+                if (session_resumed) {
+                        connection_in = packet_get_connection_in();
+                        connection_out = packet_get_connection_out();
+                        max_fd = MAX(max_fd, connection_out);
+                        max_fd = MAX(max_fd, connection_in);
+                        session_resumed = 0;
+                }
+
                 /*
                  * Send as much buffered packet data as possible to the
                  * sender.
@@ -1501,7 +1504,7 @@ client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
         channel_free_all();
 
         if (have_pty)
-                leave_raw_mode();
+                leave_raw_mode(force_tty_flag);
 
         /* restore blocking io */
         if (!isatty(fileno(stdin)))
@@ -1859,15 +1862,17 @@ client_input_channel_req(int type, u_int32_t seq, void *ctxt)
                 chan_rcvd_eow(c);
         } else if (strcmp(rtype, "exit-status") == 0) {
                 exitval = packet_get_int();
-                if (id == session_ident) {
+                if (c->ctl_chan != -1) {
+                        mux_exit_message(c, exitval);
+                        success = 1;
+                } else if (id == session_ident) {
+                        /* Record exit value of local session */
                         success = 1;
                         exit_status = exitval;
-                } else if (c->ctl_fd == -1) {
-                        error("client_input_channel_req: unexpected channel %d",
-                            session_ident);
                 } else {
-                        atomicio(vwrite, c->ctl_fd, &exitval, sizeof(exitval));
-                        success = 1;
+                        /* Probably for a mux channel that has already closed */
+                        debug("%s: no sink for exit-status on channel %d",
+                            __func__, id);
                 }
                 packet_check_eom();
         }
@@ -2063,7 +2068,7 @@ client_init_dispatch(void)
 void
 cleanup_exit(int i)
 {
-        leave_raw_mode();
+        leave_raw_mode(force_tty_flag);
         leave_non_blocking();
         if (options.control_path != NULL && muxserver_sock != -1)
                 unlink(options.control_path);