]> andersk Git - openssh.git/blobdiff - clientloop.c
- millert@cvs.openbsd.org 2001/03/04 17:42:28
[openssh.git] / clientloop.c
index 49a943a7399529efdd745a27a89f76d22b40a3d4..cdd32dbbebb076cec96ce67d22ebe1de1a8d9d85 100644 (file)
@@ -59,7 +59,7 @@
  */
 
 #include "includes.h"
-RCSID("$OpenBSD: clientloop.c,v 1.46 2001/01/29 16:55:36 markus Exp $");
+RCSID("$OpenBSD: clientloop.c,v 1.52 2001/02/28 08:45:39 markus Exp $");
 
 #include "ssh.h"
 #include "ssh1.h"
@@ -127,14 +127,13 @@ static u_int buffer_high;/* Soft max buffer size. */
 static int connection_in;      /* Connection to server (input). */
 static int connection_out;     /* Connection to server (output). */
 
-
 void   client_init_dispatch(void);
 int    session_ident = -1;
 
 /* Returns the user\'s terminal to normal mode if it had been put in raw mode. */
 
 void
-leave_raw_mode()
+leave_raw_mode(void)
 {
        if (!in_raw_mode)
                return;
@@ -148,7 +147,7 @@ leave_raw_mode()
 /* Puts the user\'s terminal in raw mode. */
 
 void
-enter_raw_mode()
+enter_raw_mode(void)
 {
        struct termios tio;
 
@@ -174,7 +173,7 @@ enter_raw_mode()
 /* Restores stdin to blocking mode. */
 
 void
-leave_non_blocking()
+leave_non_blocking(void)
 {
        if (in_non_blocking_mode) {
                (void) fcntl(fileno(stdin), F_SETFL, 0);
@@ -186,7 +185,7 @@ leave_non_blocking()
 /* Puts stdin terminal in non-blocking mode. */
 
 void
-enter_non_blocking()
+enter_non_blocking(void)
 {
        in_non_blocking_mode = 1;
        (void) fcntl(fileno(stdin), F_SETFL, O_NONBLOCK);
@@ -228,7 +227,7 @@ signal_handler(int sig)
  */
 
 double
-get_current_time()
+get_current_time(void)
 {
        struct timeval tv;
        gettimeofday(&tv, NULL);
@@ -242,7 +241,7 @@ get_current_time()
  */
 
 void
-client_check_initial_eof_on_stdin()
+client_check_initial_eof_on_stdin(void)
 {
        int len;
        char buf[1];
@@ -280,10 +279,8 @@ client_check_initial_eof_on_stdin()
                         */
                        if ((u_char) buf[0] == escape_char)
                                escape_pending = 1;
-                       else {
+                       else
                                buffer_append(&stdin_buffer, buf, 1);
-                               stdin_bytes += 1;
-                       }
                }
                leave_non_blocking();
        }
@@ -296,7 +293,7 @@ client_check_initial_eof_on_stdin()
  */
 
 void
-client_make_packets_from_stdin_data()
+client_make_packets_from_stdin_data(void)
 {
        u_int len;
 
@@ -311,6 +308,7 @@ client_make_packets_from_stdin_data()
                packet_put_string(buffer_ptr(&stdin_buffer), len);
                packet_send();
                buffer_consume(&stdin_buffer, len);
+               stdin_bytes += len;
                /* If we have a pending EOF, send it now. */
                if (stdin_eof && buffer_len(&stdin_buffer) == 0) {
                        packet_start(SSH_CMSG_EOF);
@@ -327,7 +325,7 @@ client_make_packets_from_stdin_data()
  */
 
 void
-client_check_window_change()
+client_check_window_change(void)
 {
        struct winsize ws;
 
@@ -407,12 +405,20 @@ client_wait_until_can_do_something(fd_set **readsetp, fd_set **writesetp,
 
        if (select((*maxfdp)+1, *readsetp, *writesetp, NULL, NULL) < 0) {
                char buf[100];
+
+               /*
+                * We have to clear the select masks, because we return.
+                * We have to return, because the mainloop checks for the flags
+                * set by the signal handlers.
+                */
+               memset(*readsetp, 0, *maxfdp);
+               memset(*writesetp, 0, *maxfdp);
+
                if (errno == EINTR)
                        return;
                /* Note: we might still have data in the buffers. */
                snprintf(buf, sizeof buf, "select: %s\r\n", strerror(errno));
                buffer_append(&stderr_buffer, buf, strlen(buf));
-               stderr_bytes += strlen(buf);
                quit_pending = 1;
        }
 }
@@ -478,7 +484,6 @@ client_process_net_input(fd_set * readset)
                        snprintf(buf, sizeof buf, "Connection to %.300s closed by remote host.\r\n",
                                 host);
                        buffer_append(&stderr_buffer, buf, strlen(buf));
-                       stderr_bytes += strlen(buf);
                        quit_pending = 1;
                        return;
                }
@@ -494,7 +499,6 @@ client_process_net_input(fd_set * readset)
                        snprintf(buf, sizeof buf, "Read from remote host %.300s: %.100s\r\n",
                                 host, strerror(errno));
                        buffer_append(&stderr_buffer, buf, strlen(buf));
-                       stderr_bytes += strlen(buf);
                        quit_pending = 1;
                        return;
                }
@@ -528,7 +532,6 @@ process_escapes(Buffer *bin, Buffer *bout, Buffer *berr, char *buf, int len)
                                /* Terminate the connection. */
                                snprintf(string, sizeof string, "%c.\r\n", escape_char);
                                buffer_append(berr, string, strlen(string));
-                               /*stderr_bytes += strlen(string); XXX*/
 
                                quit_pending = 1;
                                return -1;
@@ -538,7 +541,6 @@ process_escapes(Buffer *bin, Buffer *bout, Buffer *berr, char *buf, int len)
                                /* Print a message to that effect to the user. */
                                snprintf(string, sizeof string, "%c^Z [suspend ssh]\r\n", escape_char);
                                buffer_append(berr, string, strlen(string));
-                               /*stderr_bytes += strlen(string); XXX*/
 
                                /* Restore terminal modes and suspend. */
                                client_suspend_self(bin, bout, berr);
@@ -648,7 +650,6 @@ Supported escape sequences:\r\n\
 void
 client_process_input(fd_set * readset)
 {
-       int ret;
        int len;
        char buf[8192];
 
@@ -665,7 +666,6 @@ client_process_input(fd_set * readset)
                        if (len < 0) {
                                snprintf(buf, sizeof buf, "read: %.100s\r\n", strerror(errno));
                                buffer_append(&stderr_buffer, buf, strlen(buf));
-                               stderr_bytes += strlen(buf);
                        }
                        /* Mark that we have seen EOF. */
                        stdin_eof = 1;
@@ -686,16 +686,14 @@ client_process_input(fd_set * readset)
                         * Just append the data to buffer.
                         */
                        buffer_append(&stdin_buffer, buf, len);
-                       stdin_bytes += len;
                } else {
                        /*
                         * Normal, successful read.  But we have an escape character
                         * and have to process the characters one by one.
                         */
-                       ret = process_escapes(&stdin_buffer, &stdout_buffer, &stderr_buffer, buf, len);
-                       if (ret == -1)
+                       if (process_escapes(&stdin_buffer, &stdout_buffer,
+                           &stderr_buffer, buf, len) == -1)
                                return;
-                       stdout_bytes += ret;
                }
        }
 }
@@ -721,13 +719,13 @@ client_process_output(fd_set * writeset)
                                 */
                                snprintf(buf, sizeof buf, "write stdout: %.50s\r\n", strerror(errno));
                                buffer_append(&stderr_buffer, buf, strlen(buf));
-                               stderr_bytes += strlen(buf);
                                quit_pending = 1;
                                return;
                        }
                }
                /* Consume printed data from the buffer. */
                buffer_consume(&stdout_buffer, len);
+               stdout_bytes += len; 
        }
        /* Write buffered output to stderr. */
        if (FD_ISSET(fileno(stderr), writeset)) {
@@ -745,6 +743,7 @@ client_process_output(fd_set * writeset)
                }
                /* Consume printed characters from the buffer. */
                buffer_consume(&stderr_buffer, len);
+               stderr_bytes += len; 
        }
 }
 
@@ -761,7 +760,7 @@ client_process_output(fd_set * writeset)
  */
 
 void
-client_process_buffered_input_packets()
+client_process_buffered_input_packets(void)
 {
        dispatch_run(DISPATCH_NONBLOCK, &quit_pending, NULL);
 }
@@ -834,12 +833,15 @@ client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
        if (have_pty)
                enter_raw_mode();
 
-       /* Check if we should immediately send eof on stdin. */
-       if (!compat20)
+       if (compat20) {
+               session_ident = ssh2_chan_id;
+               if (escape_char != -1)
+                       channel_register_filter(session_ident,
+                           simple_escape_filter);
+       } else {
+               /* Check if we should immediately send eof on stdin. */
                client_check_initial_eof_on_stdin();
-
-       if (compat20 && escape_char != -1)
-               channel_register_filter(ssh2_chan_id, simple_escape_filter);
+       }
 
        /* Main loop of the client for the interactive session mode. */
        while (!quit_pending) {
@@ -928,7 +930,6 @@ client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
        if (have_pty && options.log_level != SYSLOG_LEVEL_QUIET) {
                snprintf(buf, sizeof buf, "Connection to %.64s closed.\r\n", host);
                buffer_append(&stderr_buffer, buf, strlen(buf));
-               stderr_bytes += strlen(buf);
        }
        /* Output any buffered data for stdout. */
        while (buffer_len(&stdout_buffer) > 0) {
@@ -939,6 +940,7 @@ client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
                        break;
                }
                buffer_consume(&stdout_buffer, len);
+               stdout_bytes += len; 
        }
 
        /* Output any buffered data for stderr. */
@@ -950,6 +952,7 @@ client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
                        break;
                }
                buffer_consume(&stderr_buffer, len);
+               stderr_bytes += len; 
        }
 
        if (have_pty)
@@ -984,7 +987,6 @@ client_input_stdout_data(int type, int plen, void *ctxt)
        char *data = packet_get_string(&data_len);
        packet_integrity_check(plen, 4 + data_len, type);
        buffer_append(&stdout_buffer, data, data_len);
-       stdout_bytes += data_len;
        memset(data, 0, data_len);
        xfree(data);
 }
@@ -995,7 +997,6 @@ client_input_stderr_data(int type, int plen, void *ctxt)
        char *data = packet_get_string(&data_len);
        packet_integrity_check(plen, 4 + data_len, type);
        buffer_append(&stderr_buffer, data, data_len);
-       stdout_bytes += data_len;
        memset(data, 0, data_len);
        xfree(data);
 }
@@ -1069,6 +1070,8 @@ client_request_x11(const char *request_type, int rchan)
        }
        packet_done();
        /* XXX check permission */
+       debug("client_request_x11: request from %s %d", originator,
+           originator_port);
        sock = x11_connect_display();
        if (sock >= 0) {
                newch = channel_new("x11",
@@ -1153,9 +1156,45 @@ client_input_channel_open(int type, int plen, void *ctxt)
        }
        xfree(ctype);
 }
+void
+client_input_channel_req(int type, int plen, void *ctxt)
+{
+       Channel *c = NULL;
+       int id, reply, success = 0;
+       char *rtype;
+
+       id = packet_get_int();
+       rtype = packet_get_string(NULL);
+       reply = packet_get_char();
+
+       debug("client_input_channel_req: channel %d rtype %s reply %d",
+           id, rtype, reply);
+
+       if (session_ident == -1) {
+               error("client_input_channel_req: no channel %d", session_ident);
+       } else if (id != session_ident) {
+               error("client_input_channel_req: channel %d: wrong channel: %d",
+                   session_ident, id);
+       }
+       c = channel_lookup(id);
+       if (c == NULL) {
+               error("client_input_channel_req: channel %d: unknown channel", id);
+       } else if (strcmp(rtype, "exit-status") == 0) {
+               success = 1;
+               exit_status = packet_get_int();
+               packet_done();
+       }
+       if (reply) {
+               packet_start(success ?
+                   SSH2_MSG_CHANNEL_SUCCESS : SSH2_MSG_CHANNEL_FAILURE);
+               packet_put_int(c->remote_id);
+               packet_send();
+       }
+       xfree(rtype);
+}
 
 void
-client_init_dispatch_20()
+client_init_dispatch_20(void)
 {
        dispatch_init(&dispatch_protocol_error);
        dispatch_set(SSH2_MSG_CHANNEL_CLOSE, &channel_input_oclose);
@@ -1165,11 +1204,11 @@ client_init_dispatch_20()
        dispatch_set(SSH2_MSG_CHANNEL_OPEN, &client_input_channel_open);
        dispatch_set(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION, &channel_input_open_confirmation);
        dispatch_set(SSH2_MSG_CHANNEL_OPEN_FAILURE, &channel_input_open_failure);
-       dispatch_set(SSH2_MSG_CHANNEL_REQUEST, &channel_input_channel_request);
+       dispatch_set(SSH2_MSG_CHANNEL_REQUEST, &client_input_channel_req);
        dispatch_set(SSH2_MSG_CHANNEL_WINDOW_ADJUST, &channel_input_window_adjust);
 }
 void
-client_init_dispatch_13()
+client_init_dispatch_13(void)
 {
        dispatch_init(NULL);
        dispatch_set(SSH_MSG_CHANNEL_CLOSE, &channel_input_close);
@@ -1188,14 +1227,14 @@ client_init_dispatch_13()
            &x11_input_open : &deny_input_open);
 }
 void
-client_init_dispatch_15()
+client_init_dispatch_15(void)
 {
        client_init_dispatch_13();
        dispatch_set(SSH_MSG_CHANNEL_CLOSE, &channel_input_ieof);
        dispatch_set(SSH_MSG_CHANNEL_CLOSE_CONFIRMATION, & channel_input_oclose);
 }
 void
-client_init_dispatch()
+client_init_dispatch(void)
 {
        if (compat20)
                client_init_dispatch_20();
@@ -1204,49 +1243,3 @@ client_init_dispatch()
        else
                client_init_dispatch_15();
 }
-
-void
-client_input_channel_req(int id, void *arg)
-{
-       Channel *c = NULL;
-       u_int len;
-       int success = 0;
-       int reply;
-       char *rtype;
-
-       rtype = packet_get_string(&len);
-       reply = packet_get_char();
-
-       debug("client_input_channel_req: rtype %s reply %d", rtype, reply);
-
-       c = channel_lookup(id);
-       if (c == NULL)
-               fatal("client_input_channel_req: channel %d: bad channel", id);
-
-       if (session_ident == -1) {
-               error("client_input_channel_req: no channel %d", id);
-       } else if (id != session_ident) {
-               error("client_input_channel_req: bad channel %d != %d",
-                   id, session_ident);
-       } else if (strcmp(rtype, "exit-status") == 0) {
-               success = 1;
-               exit_status = packet_get_int();
-               packet_done();
-       }
-       if (reply) {
-               packet_start(success ?
-                   SSH2_MSG_CHANNEL_SUCCESS : SSH2_MSG_CHANNEL_FAILURE);
-               packet_put_int(c->remote_id);
-               packet_send();
-       }
-       xfree(rtype);
-}
-
-void
-clientloop_set_session_ident(int id)
-{
-       debug2("clientloop_set_session_ident: id %d", id);
-       session_ident = id;
-       channel_register_callback(id, SSH2_MSG_CHANNEL_REQUEST,
-           client_input_channel_req, (void *)0);
-}
This page took 0.050677 seconds and 4 git commands to generate.