]> andersk Git - openssh.git/blob - clientloop.c
- djm@cvs.openbsd.org 2008/05/09 14:18:44
[openssh.git] / clientloop.c
1 /* $OpenBSD: clientloop.c,v 1.192 2008/05/09 14:18:44 djm Exp $ */
2 /*
3  * Author: Tatu Ylonen <ylo@cs.hut.fi>
4  * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
5  *                    All rights reserved
6  * The main loop for the interactive session (client side).
7  *
8  * As far as I am concerned, the code I have written for this software
9  * can be used freely for any purpose.  Any derived versions of this
10  * software must be clearly marked as such, and if the derived work is
11  * incompatible with the protocol description in the RFC file, it must be
12  * called by a name other than "ssh" or "Secure Shell".
13  *
14  *
15  * Copyright (c) 1999 Theo de Raadt.  All rights reserved.
16  *
17  * Redistribution and use in source and binary forms, with or without
18  * modification, are permitted provided that the following conditions
19  * are met:
20  * 1. Redistributions of source code must retain the above copyright
21  *    notice, this list of conditions and the following disclaimer.
22  * 2. Redistributions in binary form must reproduce the above copyright
23  *    notice, this list of conditions and the following disclaimer in the
24  *    documentation and/or other materials provided with the distribution.
25  *
26  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
27  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
28  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
29  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
30  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
31  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
32  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
33  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
34  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
35  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36  *
37  *
38  * SSH2 support added by Markus Friedl.
39  * Copyright (c) 1999, 2000, 2001 Markus Friedl.  All rights reserved.
40  *
41  * Redistribution and use in source and binary forms, with or without
42  * modification, are permitted provided that the following conditions
43  * are met:
44  * 1. Redistributions of source code must retain the above copyright
45  *    notice, this list of conditions and the following disclaimer.
46  * 2. Redistributions in binary form must reproduce the above copyright
47  *    notice, this list of conditions and the following disclaimer in the
48  *    documentation and/or other materials provided with the distribution.
49  *
50  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
51  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
52  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
53  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
54  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
55  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
56  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
57  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
58  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
59  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
60  */
61
62 #include "includes.h"
63
64 #include <sys/types.h>
65 #include <sys/ioctl.h>
66 #include <sys/param.h>
67 #ifdef HAVE_SYS_STAT_H
68 # include <sys/stat.h>
69 #endif
70 #ifdef HAVE_SYS_TIME_H
71 # include <sys/time.h>
72 #endif
73 #include <sys/socket.h>
74
75 #include <ctype.h>
76 #include <errno.h>
77 #ifdef HAVE_PATHS_H
78 #include <paths.h>
79 #endif
80 #include <signal.h>
81 #include <stdarg.h>
82 #include <stdio.h>
83 #include <stdlib.h>
84 #include <string.h>
85 #include <termios.h>
86 #include <pwd.h>
87 #include <unistd.h>
88
89 #include "openbsd-compat/sys-queue.h"
90 #include "xmalloc.h"
91 #include "ssh.h"
92 #include "ssh1.h"
93 #include "ssh2.h"
94 #include "packet.h"
95 #include "buffer.h"
96 #include "compat.h"
97 #include "channels.h"
98 #include "dispatch.h"
99 #include "key.h"
100 #include "cipher.h"
101 #include "kex.h"
102 #include "log.h"
103 #include "readconf.h"
104 #include "clientloop.h"
105 #include "sshconnect.h"
106 #include "authfd.h"
107 #include "atomicio.h"
108 #include "sshpty.h"
109 #include "misc.h"
110 #include "monitor_fdpass.h"
111 #include "match.h"
112 #include "msg.h"
113
114 /* import options */
115 extern Options options;
116
117 /* Flag indicating that stdin should be redirected from /dev/null. */
118 extern int stdin_null_flag;
119
120 /* Flag indicating that no shell has been requested */
121 extern int no_shell_flag;
122
123 /* Control socket */
124 extern int muxserver_sock;
125
126 /*
127  * Name of the host we are connecting to.  This is the name given on the
128  * command line, or the HostName specified for the user-supplied name in a
129  * configuration file.
130  */
131 extern char *host;
132
133 /*
134  * Flag to indicate that we have received a window change signal which has
135  * not yet been processed.  This will cause a message indicating the new
136  * window size to be sent to the server a little later.  This is volatile
137  * because this is updated in a signal handler.
138  */
139 static volatile sig_atomic_t received_window_change_signal = 0;
140 static volatile sig_atomic_t received_signal = 0;
141
142 /* Flag indicating whether the user's terminal is in non-blocking mode. */
143 static int in_non_blocking_mode = 0;
144
145 /* Common data for the client loop code. */
146 static volatile sig_atomic_t quit_pending; /* Set non-zero to quit the loop. */
147 static int escape_char;         /* Escape character. */
148 static int escape_pending;      /* Last character was the escape character */
149 static int last_was_cr;         /* Last character was a newline. */
150 static int exit_status;         /* Used to store the exit status of the command. */
151 static int stdin_eof;           /* EOF has been encountered on standard error. */
152 static Buffer stdin_buffer;     /* Buffer for stdin data. */
153 static Buffer stdout_buffer;    /* Buffer for stdout data. */
154 static Buffer stderr_buffer;    /* Buffer for stderr data. */
155 static u_long stdin_bytes, stdout_bytes, stderr_bytes;
156 static u_int buffer_high;/* Soft max buffer size. */
157 static int connection_in;       /* Connection to server (input). */
158 static int connection_out;      /* Connection to server (output). */
159 static int need_rekeying;       /* Set to non-zero if rekeying is requested. */
160 static int session_closed = 0;  /* In SSH2: login session closed. */
161
162 static void client_init_dispatch(void);
163 int     session_ident = -1;
164
165 struct channel_reply_ctx {
166         const char *request_type;
167         int id, do_close;
168 };
169
170 /*XXX*/
171 extern Kex *xxx_kex;
172
173 void ssh_process_session2_setup(int, int, int, Buffer *);
174
175 /* Restores stdin to blocking mode. */
176
177 static void
178 leave_non_blocking(void)
179 {
180         if (in_non_blocking_mode) {
181                 unset_nonblock(fileno(stdin));
182                 in_non_blocking_mode = 0;
183         }
184 }
185
186 /* Puts stdin terminal in non-blocking mode. */
187
188 static void
189 enter_non_blocking(void)
190 {
191         in_non_blocking_mode = 1;
192         set_nonblock(fileno(stdin));
193 }
194
195 /*
196  * Signal handler for the window change signal (SIGWINCH).  This just sets a
197  * flag indicating that the window has changed.
198  */
199 /*ARGSUSED */
200 static void
201 window_change_handler(int sig)
202 {
203         received_window_change_signal = 1;
204         signal(SIGWINCH, window_change_handler);
205 }
206
207 /*
208  * Signal handler for signals that cause the program to terminate.  These
209  * signals must be trapped to restore terminal modes.
210  */
211 /*ARGSUSED */
212 static void
213 signal_handler(int sig)
214 {
215         received_signal = sig;
216         quit_pending = 1;
217 }
218
219 /*
220  * Returns current time in seconds from Jan 1, 1970 with the maximum
221  * available resolution.
222  */
223
224 static double
225 get_current_time(void)
226 {
227         struct timeval tv;
228         gettimeofday(&tv, NULL);
229         return (double) tv.tv_sec + (double) tv.tv_usec / 1000000.0;
230 }
231
232 #define SSH_X11_PROTO "MIT-MAGIC-COOKIE-1"
233 void
234 client_x11_get_proto(const char *display, const char *xauth_path,
235     u_int trusted, char **_proto, char **_data)
236 {
237         char cmd[1024];
238         char line[512];
239         char xdisplay[512];
240         static char proto[512], data[512];
241         FILE *f;
242         int got_data = 0, generated = 0, do_unlink = 0, i;
243         char *xauthdir, *xauthfile;
244         struct stat st;
245
246         xauthdir = xauthfile = NULL;
247         *_proto = proto;
248         *_data = data;
249         proto[0] = data[0] = '\0';
250
251         if (xauth_path == NULL ||(stat(xauth_path, &st) == -1)) {
252                 debug("No xauth program.");
253         } else {
254                 if (display == NULL) {
255                         debug("x11_get_proto: DISPLAY not set");
256                         return;
257                 }
258                 /*
259                  * Handle FamilyLocal case where $DISPLAY does
260                  * not match an authorization entry.  For this we
261                  * just try "xauth list unix:displaynum.screennum".
262                  * XXX: "localhost" match to determine FamilyLocal
263                  *      is not perfect.
264                  */
265                 if (strncmp(display, "localhost:", 10) == 0) {
266                         snprintf(xdisplay, sizeof(xdisplay), "unix:%s",
267                             display + 10);
268                         display = xdisplay;
269                 }
270                 if (trusted == 0) {
271                         xauthdir = xmalloc(MAXPATHLEN);
272                         xauthfile = xmalloc(MAXPATHLEN);
273                         strlcpy(xauthdir, "/tmp/ssh-XXXXXXXXXX", MAXPATHLEN);
274                         if (mkdtemp(xauthdir) != NULL) {
275                                 do_unlink = 1;
276                                 snprintf(xauthfile, MAXPATHLEN, "%s/xauthfile",
277                                     xauthdir);
278                                 snprintf(cmd, sizeof(cmd),
279                                     "%s -f %s generate %s " SSH_X11_PROTO
280                                     " untrusted timeout 1200 2>" _PATH_DEVNULL,
281                                     xauth_path, xauthfile, display);
282                                 debug2("x11_get_proto: %s", cmd);
283                                 if (system(cmd) == 0)
284                                         generated = 1;
285                         }
286                 }
287
288                 /*
289                  * When in untrusted mode, we read the cookie only if it was
290                  * successfully generated as an untrusted one in the step
291                  * above.
292                  */
293                 if (trusted || generated) {
294                         snprintf(cmd, sizeof(cmd),
295                             "%s %s%s list %s 2>" _PATH_DEVNULL,
296                             xauth_path,
297                             generated ? "-f " : "" ,
298                             generated ? xauthfile : "",
299                             display);
300                         debug2("x11_get_proto: %s", cmd);
301                         f = popen(cmd, "r");
302                         if (f && fgets(line, sizeof(line), f) &&
303                             sscanf(line, "%*s %511s %511s", proto, data) == 2)
304                                 got_data = 1;
305                         if (f)
306                                 pclose(f);
307                 } else
308                         error("Warning: untrusted X11 forwarding setup failed: "
309                             "xauth key data not generated");
310         }
311
312         if (do_unlink) {
313                 unlink(xauthfile);
314                 rmdir(xauthdir);
315         }
316         if (xauthdir)
317                 xfree(xauthdir);
318         if (xauthfile)
319                 xfree(xauthfile);
320
321         /*
322          * If we didn't get authentication data, just make up some
323          * data.  The forwarding code will check the validity of the
324          * response anyway, and substitute this data.  The X11
325          * server, however, will ignore this fake data and use
326          * whatever authentication mechanisms it was using otherwise
327          * for the local connection.
328          */
329         if (!got_data) {
330                 u_int32_t rnd = 0;
331
332                 logit("Warning: No xauth data; "
333                     "using fake authentication data for X11 forwarding.");
334                 strlcpy(proto, SSH_X11_PROTO, sizeof proto);
335                 for (i = 0; i < 16; i++) {
336                         if (i % 4 == 0)
337                                 rnd = arc4random();
338                         snprintf(data + 2 * i, sizeof data - 2 * i, "%02x",
339                             rnd & 0xff);
340                         rnd >>= 8;
341                 }
342         }
343 }
344
345 /*
346  * This is called when the interactive is entered.  This checks if there is
347  * an EOF coming on stdin.  We must check this explicitly, as select() does
348  * not appear to wake up when redirecting from /dev/null.
349  */
350
351 static void
352 client_check_initial_eof_on_stdin(void)
353 {
354         int len;
355         char buf[1];
356
357         /*
358          * If standard input is to be "redirected from /dev/null", we simply
359          * mark that we have seen an EOF and send an EOF message to the
360          * server. Otherwise, we try to read a single character; it appears
361          * that for some files, such /dev/null, select() never wakes up for
362          * read for this descriptor, which means that we never get EOF.  This
363          * way we will get the EOF if stdin comes from /dev/null or similar.
364          */
365         if (stdin_null_flag) {
366                 /* Fake EOF on stdin. */
367                 debug("Sending eof.");
368                 stdin_eof = 1;
369                 packet_start(SSH_CMSG_EOF);
370                 packet_send();
371         } else {
372                 enter_non_blocking();
373
374                 /* Check for immediate EOF on stdin. */
375                 len = read(fileno(stdin), buf, 1);
376                 if (len == 0) {
377                         /* EOF.  Record that we have seen it and send EOF to server. */
378                         debug("Sending eof.");
379                         stdin_eof = 1;
380                         packet_start(SSH_CMSG_EOF);
381                         packet_send();
382                 } else if (len > 0) {
383                         /*
384                          * Got data.  We must store the data in the buffer,
385                          * and also process it as an escape character if
386                          * appropriate.
387                          */
388                         if ((u_char) buf[0] == escape_char)
389                                 escape_pending = 1;
390                         else
391                                 buffer_append(&stdin_buffer, buf, 1);
392                 }
393                 leave_non_blocking();
394         }
395 }
396
397
398 /*
399  * Make packets from buffered stdin data, and buffer them for sending to the
400  * connection.
401  */
402
403 static void
404 client_make_packets_from_stdin_data(void)
405 {
406         u_int len;
407
408         /* Send buffered stdin data to the server. */
409         while (buffer_len(&stdin_buffer) > 0 &&
410             packet_not_very_much_data_to_write()) {
411                 len = buffer_len(&stdin_buffer);
412                 /* Keep the packets at reasonable size. */
413                 if (len > packet_get_maxsize())
414                         len = packet_get_maxsize();
415                 packet_start(SSH_CMSG_STDIN_DATA);
416                 packet_put_string(buffer_ptr(&stdin_buffer), len);
417                 packet_send();
418                 buffer_consume(&stdin_buffer, len);
419                 stdin_bytes += len;
420                 /* If we have a pending EOF, send it now. */
421                 if (stdin_eof && buffer_len(&stdin_buffer) == 0) {
422                         packet_start(SSH_CMSG_EOF);
423                         packet_send();
424                 }
425         }
426 }
427
428 /*
429  * Checks if the client window has changed, and sends a packet about it to
430  * the server if so.  The actual change is detected elsewhere (by a software
431  * interrupt on Unix); this just checks the flag and sends a message if
432  * appropriate.
433  */
434
435 static void
436 client_check_window_change(void)
437 {
438         struct winsize ws;
439
440         if (! received_window_change_signal)
441                 return;
442         /** XXX race */
443         received_window_change_signal = 0;
444
445         debug2("client_check_window_change: changed");
446
447         if (compat20) {
448                 channel_send_window_changes();
449         } else {
450                 if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) < 0)
451                         return;
452                 packet_start(SSH_CMSG_WINDOW_SIZE);
453                 packet_put_int((u_int)ws.ws_row);
454                 packet_put_int((u_int)ws.ws_col);
455                 packet_put_int((u_int)ws.ws_xpixel);
456                 packet_put_int((u_int)ws.ws_ypixel);
457                 packet_send();
458         }
459 }
460
461 static void
462 client_global_request_reply(int type, u_int32_t seq, void *ctxt)
463 {
464         keep_alive_timeouts = 0;
465         client_global_request_reply_fwd(type, seq, ctxt);
466 }
467
468 static void
469 server_alive_check(void)
470 {
471         if (++keep_alive_timeouts > options.server_alive_count_max) {
472                 logit("Timeout, server not responding.");
473                 cleanup_exit(255);
474         }
475         packet_start(SSH2_MSG_GLOBAL_REQUEST);
476         packet_put_cstring("keepalive@openssh.com");
477         packet_put_char(1);     /* boolean: want reply */
478         packet_send();
479 }
480
481 /*
482  * Waits until the client can do something (some data becomes available on
483  * one of the file descriptors).
484  */
485 static void
486 client_wait_until_can_do_something(fd_set **readsetp, fd_set **writesetp,
487     int *maxfdp, u_int *nallocp, int rekeying)
488 {
489         struct timeval tv, *tvp;
490         int ret;
491
492         /* Add any selections by the channel mechanism. */
493         channel_prepare_select(readsetp, writesetp, maxfdp, nallocp, rekeying);
494
495         if (!compat20) {
496                 /* Read from the connection, unless our buffers are full. */
497                 if (buffer_len(&stdout_buffer) < buffer_high &&
498                     buffer_len(&stderr_buffer) < buffer_high &&
499                     channel_not_very_much_buffered_data())
500                         FD_SET(connection_in, *readsetp);
501                 /*
502                  * Read from stdin, unless we have seen EOF or have very much
503                  * buffered data to send to the server.
504                  */
505                 if (!stdin_eof && packet_not_very_much_data_to_write())
506                         FD_SET(fileno(stdin), *readsetp);
507
508                 /* Select stdout/stderr if have data in buffer. */
509                 if (buffer_len(&stdout_buffer) > 0)
510                         FD_SET(fileno(stdout), *writesetp);
511                 if (buffer_len(&stderr_buffer) > 0)
512                         FD_SET(fileno(stderr), *writesetp);
513         } else {
514                 /* channel_prepare_select could have closed the last channel */
515                 if (session_closed && !channel_still_open() &&
516                     !packet_have_data_to_write()) {
517                         /* clear mask since we did not call select() */
518                         memset(*readsetp, 0, *nallocp);
519                         memset(*writesetp, 0, *nallocp);
520                         return;
521                 } else {
522                         FD_SET(connection_in, *readsetp);
523                 }
524         }
525
526         /* Select server connection if have data to write to the server. */
527         if (packet_have_data_to_write())
528                 FD_SET(connection_out, *writesetp);
529
530         if (muxserver_sock != -1)
531                 FD_SET(muxserver_sock, *readsetp);
532
533         /*
534          * Wait for something to happen.  This will suspend the process until
535          * some selected descriptor can be read, written, or has some other
536          * event pending.
537          */
538
539         if (options.server_alive_interval == 0 || !compat20)
540                 tvp = NULL;
541         else {
542                 tv.tv_sec = options.server_alive_interval;
543                 tv.tv_usec = 0;
544                 tvp = &tv;
545         }
546         ret = select((*maxfdp)+1, *readsetp, *writesetp, NULL, tvp);
547         if (ret < 0) {
548                 char buf[100];
549
550                 /*
551                  * We have to clear the select masks, because we return.
552                  * We have to return, because the mainloop checks for the flags
553                  * set by the signal handlers.
554                  */
555                 memset(*readsetp, 0, *nallocp);
556                 memset(*writesetp, 0, *nallocp);
557
558                 if (errno == EINTR)
559                         return;
560                 /* Note: we might still have data in the buffers. */
561                 snprintf(buf, sizeof buf, "select: %s\r\n", strerror(errno));
562                 buffer_append(&stderr_buffer, buf, strlen(buf));
563                 quit_pending = 1;
564         } else if (ret == 0)
565                 server_alive_check();
566 }
567
568 static void
569 client_suspend_self(Buffer *bin, Buffer *bout, Buffer *berr)
570 {
571         /* Flush stdout and stderr buffers. */
572         if (buffer_len(bout) > 0)
573                 atomicio(vwrite, fileno(stdout), buffer_ptr(bout), buffer_len(bout));
574         if (buffer_len(berr) > 0)
575                 atomicio(vwrite, fileno(stderr), buffer_ptr(berr), buffer_len(berr));
576
577         leave_raw_mode();
578
579         /*
580          * Free (and clear) the buffer to reduce the amount of data that gets
581          * written to swap.
582          */
583         buffer_free(bin);
584         buffer_free(bout);
585         buffer_free(berr);
586
587         /* Send the suspend signal to the program itself. */
588         kill(getpid(), SIGTSTP);
589
590         /* Reset window sizes in case they have changed */
591         received_window_change_signal = 1;
592
593         /* OK, we have been continued by the user. Reinitialize buffers. */
594         buffer_init(bin);
595         buffer_init(bout);
596         buffer_init(berr);
597
598         enter_raw_mode();
599 }
600
601 static void
602 client_process_net_input(fd_set *readset)
603 {
604         int len;
605         char buf[8192];
606
607         /*
608          * Read input from the server, and add any such data to the buffer of
609          * the packet subsystem.
610          */
611         if (FD_ISSET(connection_in, readset)) {
612                 /* Read as much as possible. */
613                 len = read(connection_in, buf, sizeof(buf));
614                 if (len == 0) {
615                         /* Received EOF.  The remote host has closed the connection. */
616                         snprintf(buf, sizeof buf, "Connection to %.300s closed by remote host.\r\n",
617                                  host);
618                         buffer_append(&stderr_buffer, buf, strlen(buf));
619                         quit_pending = 1;
620                         return;
621                 }
622                 /*
623                  * There is a kernel bug on Solaris that causes select to
624                  * sometimes wake up even though there is no data available.
625                  */
626                 if (len < 0 && (errno == EAGAIN || errno == EINTR))
627                         len = 0;
628
629                 if (len < 0) {
630                         /* An error has encountered.  Perhaps there is a network problem. */
631                         snprintf(buf, sizeof buf, "Read from remote host %.300s: %.100s\r\n",
632                                  host, strerror(errno));
633                         buffer_append(&stderr_buffer, buf, strlen(buf));
634                         quit_pending = 1;
635                         return;
636                 }
637                 packet_process_incoming(buf, len);
638         }
639 }
640
641 static void
642 client_status_confirm(int type, Channel *c, void *ctx)
643 {
644         struct channel_reply_ctx *cr = (struct channel_reply_ctx *)ctx;
645         char errmsg[256];
646         int tochan;
647
648         /* XXX supress on mux _client_ quietmode */
649         tochan = options.log_level >= SYSLOG_LEVEL_ERROR &&
650             c->ctl_fd != -1 && c->extended_usage == CHAN_EXTENDED_WRITE;
651
652         if (type == SSH2_MSG_CHANNEL_SUCCESS) {
653                 debug2("%s request accepted on channel %d",
654                     cr->request_type, c->self);
655         } else if (type == SSH2_MSG_CHANNEL_FAILURE) {
656                 if (tochan) {
657                         snprintf(errmsg, sizeof(errmsg),
658                             "%s request failed\r\n", cr->request_type);
659                 } else {
660                         snprintf(errmsg, sizeof(errmsg),
661                             "%s request failed on channel %d",
662                             cr->request_type, c->self);
663                 }
664                 /* If error occurred on primary session channel, then exit */
665                 if (cr->do_close && c->self == session_ident)
666                         fatal("%s", errmsg);
667                 /* If error occurred on mux client, append to their stderr */
668                 if (tochan)
669                         buffer_append(&c->extended, errmsg, strlen(errmsg));
670                 else
671                         error("%s", errmsg);
672                 if (cr->do_close) {
673                         chan_read_failed(c);
674                         chan_write_failed(c);
675                 }
676         }
677         xfree(cr);
678 }
679
680 static void
681 client_abandon_status_confirm(Channel *c, void *ctx)
682 {
683         xfree(ctx);
684 }
685
686 static void
687 client_expect_confirm(int id, const char *request, int do_close)
688 {
689         struct channel_reply_ctx *cr = xmalloc(sizeof(*cr));
690
691         cr->request_type = request;
692         cr->do_close = do_close;
693
694         channel_register_status_confirm(id, client_status_confirm,
695             client_abandon_status_confirm, cr);
696 }
697
698 static void
699 process_cmdline(void)
700 {
701         void (*handler)(int);
702         char *s, *cmd, *cancel_host;
703         int delete = 0;
704         int local = 0;
705         u_short cancel_port;
706         Forward fwd;
707
708         bzero(&fwd, sizeof(fwd));
709         fwd.listen_host = fwd.connect_host = NULL;
710
711         leave_raw_mode();
712         handler = signal(SIGINT, SIG_IGN);
713         cmd = s = read_passphrase("\r\nssh> ", RP_ECHO);
714         if (s == NULL)
715                 goto out;
716         while (isspace(*s))
717                 s++;
718         if (*s == '-')
719                 s++;    /* Skip cmdline '-', if any */
720         if (*s == '\0')
721                 goto out;
722
723         if (*s == 'h' || *s == 'H' || *s == '?') {
724                 logit("Commands:");
725                 logit("      -L[bind_address:]port:host:hostport    "
726                     "Request local forward");
727                 logit("      -R[bind_address:]port:host:hostport    "
728                     "Request remote forward");
729                 logit("      -KR[bind_address:]port                 "
730                     "Cancel remote forward");
731                 if (!options.permit_local_command)
732                         goto out;
733                 logit("      !args                                  "
734                     "Execute local command");
735                 goto out;
736         }
737
738         if (*s == '!' && options.permit_local_command) {
739                 s++;
740                 ssh_local_cmd(s);
741                 goto out;
742         }
743
744         if (*s == 'K') {
745                 delete = 1;
746                 s++;
747         }
748         if (*s != 'L' && *s != 'R') {
749                 logit("Invalid command.");
750                 goto out;
751         }
752         if (*s == 'L')
753                 local = 1;
754         if (local && delete) {
755                 logit("Not supported.");
756                 goto out;
757         }
758         if ((!local || delete) && !compat20) {
759                 logit("Not supported for SSH protocol version 1.");
760                 goto out;
761         }
762
763         while (isspace(*++s))
764                 ;
765
766         if (delete) {
767                 cancel_port = 0;
768                 cancel_host = hpdelim(&s);      /* may be NULL */
769                 if (s != NULL) {
770                         cancel_port = a2port(s);
771                         cancel_host = cleanhostname(cancel_host);
772                 } else {
773                         cancel_port = a2port(cancel_host);
774                         cancel_host = NULL;
775                 }
776                 if (cancel_port == 0) {
777                         logit("Bad forwarding close port");
778                         goto out;
779                 }
780                 channel_request_rforward_cancel(cancel_host, cancel_port);
781         } else {
782                 if (!parse_forward(&fwd, s)) {
783                         logit("Bad forwarding specification.");
784                         goto out;
785                 }
786                 if (local) {
787                         if (channel_setup_local_fwd_listener(fwd.listen_host,
788                             fwd.listen_port, fwd.connect_host,
789                             fwd.connect_port, options.gateway_ports) < 0) {
790                                 logit("Port forwarding failed.");
791                                 goto out;
792                         }
793                 } else {
794                         if (channel_request_remote_forwarding(fwd.listen_host,
795                             fwd.listen_port, fwd.connect_host,
796                             fwd.connect_port) < 0) {
797                                 logit("Port forwarding failed.");
798                                 goto out;
799                         }
800                 }
801
802                 logit("Forwarding port.");
803         }
804
805 out:
806         signal(SIGINT, handler);
807         enter_raw_mode();
808         if (cmd)
809                 xfree(cmd);
810         if (fwd.listen_host != NULL)
811                 xfree(fwd.listen_host);
812         if (fwd.connect_host != NULL)
813                 xfree(fwd.connect_host);
814 }
815
816 /* process the characters one by one */
817 static int
818 process_escapes(Buffer *bin, Buffer *bout, Buffer *berr, char *buf, int len)
819 {
820         char string[1024];
821         pid_t pid;
822         int bytes = 0;
823         u_int i;
824         u_char ch;
825         char *s;
826
827         if (len <= 0)
828                 return (0);
829
830         for (i = 0; i < (u_int)len; i++) {
831                 /* Get one character at a time. */
832                 ch = buf[i];
833
834                 if (escape_pending) {
835                         /* We have previously seen an escape character. */
836                         /* Clear the flag now. */
837                         escape_pending = 0;
838
839                         /* Process the escaped character. */
840                         switch (ch) {
841                         case '.':
842                                 /* Terminate the connection. */
843                                 snprintf(string, sizeof string, "%c.\r\n", escape_char);
844                                 buffer_append(berr, string, strlen(string));
845
846                                 quit_pending = 1;
847                                 return -1;
848
849                         case 'Z' - 64:
850                                 /* Suspend the program. */
851                                 /* Print a message to that effect to the user. */
852                                 snprintf(string, sizeof string, "%c^Z [suspend ssh]\r\n", escape_char);
853                                 buffer_append(berr, string, strlen(string));
854
855                                 /* Restore terminal modes and suspend. */
856                                 client_suspend_self(bin, bout, berr);
857
858                                 /* We have been continued. */
859                                 continue;
860
861                         case 'B':
862                                 if (compat20) {
863                                         snprintf(string, sizeof string,
864                                             "%cB\r\n", escape_char);
865                                         buffer_append(berr, string,
866                                             strlen(string));
867                                         channel_request_start(session_ident,
868                                             "break", 0);
869                                         packet_put_int(1000);
870                                         packet_send();
871                                 }
872                                 continue;
873
874                         case 'R':
875                                 if (compat20) {
876                                         if (datafellows & SSH_BUG_NOREKEY)
877                                                 logit("Server does not support re-keying");
878                                         else
879                                                 need_rekeying = 1;
880                                 }
881                                 continue;
882
883                         case '&':
884                                 /*
885                                  * Detach the program (continue to serve connections,
886                                  * but put in background and no more new connections).
887                                  */
888                                 /* Restore tty modes. */
889                                 leave_raw_mode();
890
891                                 /* Stop listening for new connections. */
892                                 channel_stop_listening();
893
894                                 snprintf(string, sizeof string,
895                                     "%c& [backgrounded]\n", escape_char);
896                                 buffer_append(berr, string, strlen(string));
897
898                                 /* Fork into background. */
899                                 pid = fork();
900                                 if (pid < 0) {
901                                         error("fork: %.100s", strerror(errno));
902                                         continue;
903                                 }
904                                 if (pid != 0) { /* This is the parent. */
905                                         /* The parent just exits. */
906                                         exit(0);
907                                 }
908                                 /* The child continues serving connections. */
909                                 if (compat20) {
910                                         buffer_append(bin, "\004", 1);
911                                         /* fake EOF on stdin */
912                                         return -1;
913                                 } else if (!stdin_eof) {
914                                         /*
915                                          * Sending SSH_CMSG_EOF alone does not always appear
916                                          * to be enough.  So we try to send an EOF character
917                                          * first.
918                                          */
919                                         packet_start(SSH_CMSG_STDIN_DATA);
920                                         packet_put_string("\004", 1);
921                                         packet_send();
922                                         /* Close stdin. */
923                                         stdin_eof = 1;
924                                         if (buffer_len(bin) == 0) {
925                                                 packet_start(SSH_CMSG_EOF);
926                                                 packet_send();
927                                         }
928                                 }
929                                 continue;
930
931                         case '?':
932                                 snprintf(string, sizeof string,
933 "%c?\r\n\
934 Supported escape sequences:\r\n\
935 %c.  - terminate connection\r\n\
936 %cB  - send a BREAK to the remote system\r\n\
937 %cC  - open a command line\r\n\
938 %cR  - Request rekey (SSH protocol 2 only)\r\n\
939 %c^Z - suspend ssh\r\n\
940 %c#  - list forwarded connections\r\n\
941 %c&  - background ssh (when waiting for connections to terminate)\r\n\
942 %c?  - this message\r\n\
943 %c%c  - send the escape character by typing it twice\r\n\
944 (Note that escapes are only recognized immediately after newline.)\r\n",
945                                     escape_char, escape_char, escape_char, escape_char,
946                                     escape_char, escape_char, escape_char, escape_char,
947                                     escape_char, escape_char, escape_char);
948                                 buffer_append(berr, string, strlen(string));
949                                 continue;
950
951                         case '#':
952                                 snprintf(string, sizeof string, "%c#\r\n", escape_char);
953                                 buffer_append(berr, string, strlen(string));
954                                 s = channel_open_message();
955                                 buffer_append(berr, s, strlen(s));
956                                 xfree(s);
957                                 continue;
958
959                         case 'C':
960                                 process_cmdline();
961                                 continue;
962
963                         default:
964                                 if (ch != escape_char) {
965                                         buffer_put_char(bin, escape_char);
966                                         bytes++;
967                                 }
968                                 /* Escaped characters fall through here */
969                                 break;
970                         }
971                 } else {
972                         /*
973                          * The previous character was not an escape char. Check if this
974                          * is an escape.
975                          */
976                         if (last_was_cr && ch == escape_char) {
977                                 /* It is. Set the flag and continue to next character. */
978                                 escape_pending = 1;
979                                 continue;
980                         }
981                 }
982
983                 /*
984                  * Normal character.  Record whether it was a newline,
985                  * and append it to the buffer.
986                  */
987                 last_was_cr = (ch == '\r' || ch == '\n');
988                 buffer_put_char(bin, ch);
989                 bytes++;
990         }
991         return bytes;
992 }
993
994 static void
995 client_process_input(fd_set *readset)
996 {
997         int len;
998         char buf[8192];
999
1000         /* Read input from stdin. */
1001         if (FD_ISSET(fileno(stdin), readset)) {
1002                 /* Read as much as possible. */
1003                 len = read(fileno(stdin), buf, sizeof(buf));
1004                 if (len < 0 && (errno == EAGAIN || errno == EINTR))
1005                         return;         /* we'll try again later */
1006                 if (len <= 0) {
1007                         /*
1008                          * Received EOF or error.  They are treated
1009                          * similarly, except that an error message is printed
1010                          * if it was an error condition.
1011                          */
1012                         if (len < 0) {
1013                                 snprintf(buf, sizeof buf, "read: %.100s\r\n", strerror(errno));
1014                                 buffer_append(&stderr_buffer, buf, strlen(buf));
1015                         }
1016                         /* Mark that we have seen EOF. */
1017                         stdin_eof = 1;
1018                         /*
1019                          * Send an EOF message to the server unless there is
1020                          * data in the buffer.  If there is data in the
1021                          * buffer, no message will be sent now.  Code
1022                          * elsewhere will send the EOF when the buffer
1023                          * becomes empty if stdin_eof is set.
1024                          */
1025                         if (buffer_len(&stdin_buffer) == 0) {
1026                                 packet_start(SSH_CMSG_EOF);
1027                                 packet_send();
1028                         }
1029                 } else if (escape_char == SSH_ESCAPECHAR_NONE) {
1030                         /*
1031                          * Normal successful read, and no escape character.
1032                          * Just append the data to buffer.
1033                          */
1034                         buffer_append(&stdin_buffer, buf, len);
1035                 } else {
1036                         /*
1037                          * Normal, successful read.  But we have an escape character
1038                          * and have to process the characters one by one.
1039                          */
1040                         if (process_escapes(&stdin_buffer, &stdout_buffer,
1041                             &stderr_buffer, buf, len) == -1)
1042                                 return;
1043                 }
1044         }
1045 }
1046
1047 static void
1048 client_process_output(fd_set *writeset)
1049 {
1050         int len;
1051         char buf[100];
1052
1053         /* Write buffered output to stdout. */
1054         if (FD_ISSET(fileno(stdout), writeset)) {
1055                 /* Write as much data as possible. */
1056                 len = write(fileno(stdout), buffer_ptr(&stdout_buffer),
1057                     buffer_len(&stdout_buffer));
1058                 if (len <= 0) {
1059                         if (errno == EINTR || errno == EAGAIN)
1060                                 len = 0;
1061                         else {
1062                                 /*
1063                                  * An error or EOF was encountered.  Put an
1064                                  * error message to stderr buffer.
1065                                  */
1066                                 snprintf(buf, sizeof buf, "write stdout: %.50s\r\n", strerror(errno));
1067                                 buffer_append(&stderr_buffer, buf, strlen(buf));
1068                                 quit_pending = 1;
1069                                 return;
1070                         }
1071                 }
1072                 /* Consume printed data from the buffer. */
1073                 buffer_consume(&stdout_buffer, len);
1074                 stdout_bytes += len;
1075         }
1076         /* Write buffered output to stderr. */
1077         if (FD_ISSET(fileno(stderr), writeset)) {
1078                 /* Write as much data as possible. */
1079                 len = write(fileno(stderr), buffer_ptr(&stderr_buffer),
1080                     buffer_len(&stderr_buffer));
1081                 if (len <= 0) {
1082                         if (errno == EINTR || errno == EAGAIN)
1083                                 len = 0;
1084                         else {
1085                                 /* EOF or error, but can't even print error message. */
1086                                 quit_pending = 1;
1087                                 return;
1088                         }
1089                 }
1090                 /* Consume printed characters from the buffer. */
1091                 buffer_consume(&stderr_buffer, len);
1092                 stderr_bytes += len;
1093         }
1094 }
1095
1096 /*
1097  * Get packets from the connection input buffer, and process them as long as
1098  * there are packets available.
1099  *
1100  * Any unknown packets received during the actual
1101  * session cause the session to terminate.  This is
1102  * intended to make debugging easier since no
1103  * confirmations are sent.  Any compatible protocol
1104  * extensions must be negotiated during the
1105  * preparatory phase.
1106  */
1107
1108 static void
1109 client_process_buffered_input_packets(void)
1110 {
1111         dispatch_run(DISPATCH_NONBLOCK, &quit_pending, compat20 ? xxx_kex : NULL);
1112 }
1113
1114 /* scan buf[] for '~' before sending data to the peer */
1115
1116 static int
1117 simple_escape_filter(Channel *c, char *buf, int len)
1118 {
1119         if (c->extended_usage != CHAN_EXTENDED_WRITE)
1120                 return 0;
1121
1122         return process_escapes(&c->input, &c->output, &c->extended, buf, len);
1123 }
1124
1125 static void
1126 client_channel_closed(int id, void *arg)
1127 {
1128         channel_cancel_cleanup(id);
1129         session_closed = 1;
1130         leave_raw_mode();
1131 }
1132
1133 /*
1134  * Implements the interactive session with the server.  This is called after
1135  * the user has been authenticated, and a command has been started on the
1136  * remote host.  If escape_char != SSH_ESCAPECHAR_NONE, it is the character
1137  * used as an escape character for terminating or suspending the session.
1138  */
1139
1140 int
1141 client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
1142 {
1143         fd_set *readset = NULL, *writeset = NULL;
1144         double start_time, total_time;
1145         int max_fd = 0, max_fd2 = 0, len, rekeying = 0;
1146         u_int nalloc = 0;
1147         char buf[100];
1148
1149         debug("Entering interactive session.");
1150
1151         start_time = get_current_time();
1152
1153         /* Initialize variables. */
1154         escape_pending = 0;
1155         last_was_cr = 1;
1156         exit_status = -1;
1157         stdin_eof = 0;
1158         buffer_high = 64 * 1024;
1159         connection_in = packet_get_connection_in();
1160         connection_out = packet_get_connection_out();
1161         max_fd = MAX(connection_in, connection_out);
1162         if (muxserver_sock != -1)
1163                 max_fd = MAX(max_fd, muxserver_sock);
1164
1165         if (!compat20) {
1166                 /* enable nonblocking unless tty */
1167                 if (!isatty(fileno(stdin)))
1168                         set_nonblock(fileno(stdin));
1169                 if (!isatty(fileno(stdout)))
1170                         set_nonblock(fileno(stdout));
1171                 if (!isatty(fileno(stderr)))
1172                         set_nonblock(fileno(stderr));
1173                 max_fd = MAX(max_fd, fileno(stdin));
1174                 max_fd = MAX(max_fd, fileno(stdout));
1175                 max_fd = MAX(max_fd, fileno(stderr));
1176         }
1177         stdin_bytes = 0;
1178         stdout_bytes = 0;
1179         stderr_bytes = 0;
1180         quit_pending = 0;
1181         escape_char = escape_char_arg;
1182
1183         /* Initialize buffers. */
1184         buffer_init(&stdin_buffer);
1185         buffer_init(&stdout_buffer);
1186         buffer_init(&stderr_buffer);
1187
1188         client_init_dispatch();
1189
1190         /*
1191          * Set signal handlers, (e.g. to restore non-blocking mode)
1192          * but don't overwrite SIG_IGN, matches behaviour from rsh(1)
1193          */
1194         if (signal(SIGHUP, SIG_IGN) != SIG_IGN)
1195                 signal(SIGHUP, signal_handler);
1196         if (signal(SIGINT, SIG_IGN) != SIG_IGN)
1197                 signal(SIGINT, signal_handler);
1198         if (signal(SIGQUIT, SIG_IGN) != SIG_IGN)
1199                 signal(SIGQUIT, signal_handler);
1200         if (signal(SIGTERM, SIG_IGN) != SIG_IGN)
1201                 signal(SIGTERM, signal_handler);
1202         signal(SIGWINCH, window_change_handler);
1203
1204         if (have_pty)
1205                 enter_raw_mode();
1206
1207         if (compat20) {
1208                 session_ident = ssh2_chan_id;
1209                 if (escape_char != SSH_ESCAPECHAR_NONE)
1210                         channel_register_filter(session_ident,
1211                             simple_escape_filter, NULL);
1212                 if (session_ident != -1)
1213                         channel_register_cleanup(session_ident,
1214                             client_channel_closed, 0);
1215         } else {
1216                 /* Check if we should immediately send eof on stdin. */
1217                 client_check_initial_eof_on_stdin();
1218         }
1219
1220         /* Main loop of the client for the interactive session mode. */
1221         while (!quit_pending) {
1222
1223                 /* Process buffered packets sent by the server. */
1224                 client_process_buffered_input_packets();
1225
1226                 if (compat20 && session_closed && !channel_still_open())
1227                         break;
1228
1229                 rekeying = (xxx_kex != NULL && !xxx_kex->done);
1230
1231                 if (rekeying) {
1232                         debug("rekeying in progress");
1233                 } else {
1234                         /*
1235                          * Make packets of buffered stdin data, and buffer
1236                          * them for sending to the server.
1237                          */
1238                         if (!compat20)
1239                                 client_make_packets_from_stdin_data();
1240
1241                         /*
1242                          * Make packets from buffered channel data, and
1243                          * enqueue them for sending to the server.
1244                          */
1245                         if (packet_not_very_much_data_to_write())
1246                                 channel_output_poll();
1247
1248                         /*
1249                          * Check if the window size has changed, and buffer a
1250                          * message about it to the server if so.
1251                          */
1252                         client_check_window_change();
1253
1254                         if (quit_pending)
1255                                 break;
1256                 }
1257                 /*
1258                  * Wait until we have something to do (something becomes
1259                  * available on one of the descriptors).
1260                  */
1261                 max_fd2 = max_fd;
1262                 client_wait_until_can_do_something(&readset, &writeset,
1263                     &max_fd2, &nalloc, rekeying);
1264
1265                 if (quit_pending)
1266                         break;
1267
1268                 /* Do channel operations unless rekeying in progress. */
1269                 if (!rekeying) {
1270                         channel_after_select(readset, writeset);
1271                         if (need_rekeying || packet_need_rekeying()) {
1272                                 debug("need rekeying");
1273                                 xxx_kex->done = 0;
1274                                 kex_send_kexinit(xxx_kex);
1275                                 need_rekeying = 0;
1276                         }
1277                 }
1278
1279                 /* Buffer input from the connection.  */
1280                 client_process_net_input(readset);
1281
1282                 /* Accept control connections.  */
1283                 if (muxserver_sock != -1 &&FD_ISSET(muxserver_sock, readset)) {
1284                         if (muxserver_accept_control())
1285                                 quit_pending = 1;
1286                 }
1287
1288                 if (quit_pending)
1289                         break;
1290
1291                 if (!compat20) {
1292                         /* Buffer data from stdin */
1293                         client_process_input(readset);
1294                         /*
1295                          * Process output to stdout and stderr.  Output to
1296                          * the connection is processed elsewhere (above).
1297                          */
1298                         client_process_output(writeset);
1299                 }
1300
1301                 /* Send as much buffered packet data as possible to the sender. */
1302                 if (FD_ISSET(connection_out, writeset))
1303                         packet_write_poll();
1304         }
1305         if (readset)
1306                 xfree(readset);
1307         if (writeset)
1308                 xfree(writeset);
1309
1310         /* Terminate the session. */
1311
1312         /* Stop watching for window change. */
1313         signal(SIGWINCH, SIG_DFL);
1314
1315         channel_free_all();
1316
1317         if (have_pty)
1318                 leave_raw_mode();
1319
1320         /* restore blocking io */
1321         if (!isatty(fileno(stdin)))
1322                 unset_nonblock(fileno(stdin));
1323         if (!isatty(fileno(stdout)))
1324                 unset_nonblock(fileno(stdout));
1325         if (!isatty(fileno(stderr)))
1326                 unset_nonblock(fileno(stderr));
1327
1328         /*
1329          * If there was no shell or command requested, there will be no remote
1330          * exit status to be returned.  In that case, clear error code if the
1331          * connection was deliberately terminated at this end.
1332          */
1333         if (no_shell_flag && received_signal == SIGTERM) {
1334                 received_signal = 0;
1335                 exit_status = 0;
1336         }
1337
1338         if (received_signal)
1339                 fatal("Killed by signal %d.", (int) received_signal);
1340
1341         /*
1342          * In interactive mode (with pseudo tty) display a message indicating
1343          * that the connection has been closed.
1344          */
1345         if (have_pty && options.log_level != SYSLOG_LEVEL_QUIET) {
1346                 snprintf(buf, sizeof buf, "Connection to %.64s closed.\r\n", host);
1347                 buffer_append(&stderr_buffer, buf, strlen(buf));
1348         }
1349
1350         /* Output any buffered data for stdout. */
1351         while (buffer_len(&stdout_buffer) > 0) {
1352                 len = write(fileno(stdout), buffer_ptr(&stdout_buffer),
1353                     buffer_len(&stdout_buffer));
1354                 if (len <= 0) {
1355                         error("Write failed flushing stdout buffer.");
1356                         break;
1357                 }
1358                 buffer_consume(&stdout_buffer, len);
1359                 stdout_bytes += len;
1360         }
1361
1362         /* Output any buffered data for stderr. */
1363         while (buffer_len(&stderr_buffer) > 0) {
1364                 len = write(fileno(stderr), buffer_ptr(&stderr_buffer),
1365                     buffer_len(&stderr_buffer));
1366                 if (len <= 0) {
1367                         error("Write failed flushing stderr buffer.");
1368                         break;
1369                 }
1370                 buffer_consume(&stderr_buffer, len);
1371                 stderr_bytes += len;
1372         }
1373
1374         /* Clear and free any buffers. */
1375         memset(buf, 0, sizeof(buf));
1376         buffer_free(&stdin_buffer);
1377         buffer_free(&stdout_buffer);
1378         buffer_free(&stderr_buffer);
1379
1380         /* Report bytes transferred, and transfer rates. */
1381         total_time = get_current_time() - start_time;
1382         debug("Transferred: stdin %lu, stdout %lu, stderr %lu bytes in %.1f seconds",
1383             stdin_bytes, stdout_bytes, stderr_bytes, total_time);
1384         if (total_time > 0)
1385                 debug("Bytes per second: stdin %.1f, stdout %.1f, stderr %.1f",
1386                     stdin_bytes / total_time, stdout_bytes / total_time,
1387                     stderr_bytes / total_time);
1388
1389         /* Return the exit status of the program. */
1390         debug("Exit status %d", exit_status);
1391         return exit_status;
1392 }
1393
1394 /*********/
1395
1396 static void
1397 client_input_stdout_data(int type, u_int32_t seq, void *ctxt)
1398 {
1399         u_int data_len;
1400         char *data = packet_get_string(&data_len);
1401         packet_check_eom();
1402         buffer_append(&stdout_buffer, data, data_len);
1403         memset(data, 0, data_len);
1404         xfree(data);
1405 }
1406 static void
1407 client_input_stderr_data(int type, u_int32_t seq, void *ctxt)
1408 {
1409         u_int data_len;
1410         char *data = packet_get_string(&data_len);
1411         packet_check_eom();
1412         buffer_append(&stderr_buffer, data, data_len);
1413         memset(data, 0, data_len);
1414         xfree(data);
1415 }
1416 static void
1417 client_input_exit_status(int type, u_int32_t seq, void *ctxt)
1418 {
1419         exit_status = packet_get_int();
1420         packet_check_eom();
1421         /* Acknowledge the exit. */
1422         packet_start(SSH_CMSG_EXIT_CONFIRMATION);
1423         packet_send();
1424         /*
1425          * Must wait for packet to be sent since we are
1426          * exiting the loop.
1427          */
1428         packet_write_wait();
1429         /* Flag that we want to exit. */
1430         quit_pending = 1;
1431 }
1432 static void
1433 client_input_agent_open(int type, u_int32_t seq, void *ctxt)
1434 {
1435         Channel *c = NULL;
1436         int remote_id, sock;
1437
1438         /* Read the remote channel number from the message. */
1439         remote_id = packet_get_int();
1440         packet_check_eom();
1441
1442         /*
1443          * Get a connection to the local authentication agent (this may again
1444          * get forwarded).
1445          */
1446         sock = ssh_get_authentication_socket();
1447
1448         /*
1449          * If we could not connect the agent, send an error message back to
1450          * the server. This should never happen unless the agent dies,
1451          * because authentication forwarding is only enabled if we have an
1452          * agent.
1453          */
1454         if (sock >= 0) {
1455                 c = channel_new("", SSH_CHANNEL_OPEN, sock, sock,
1456                     -1, 0, 0, 0, "authentication agent connection", 1);
1457                 c->remote_id = remote_id;
1458                 c->force_drain = 1;
1459         }
1460         if (c == NULL) {
1461                 packet_start(SSH_MSG_CHANNEL_OPEN_FAILURE);
1462                 packet_put_int(remote_id);
1463         } else {
1464                 /* Send a confirmation to the remote host. */
1465                 debug("Forwarding authentication connection.");
1466                 packet_start(SSH_MSG_CHANNEL_OPEN_CONFIRMATION);
1467                 packet_put_int(remote_id);
1468                 packet_put_int(c->self);
1469         }
1470         packet_send();
1471 }
1472
1473 static Channel *
1474 client_request_forwarded_tcpip(const char *request_type, int rchan)
1475 {
1476         Channel *c = NULL;
1477         char *listen_address, *originator_address;
1478         int listen_port, originator_port;
1479
1480         /* Get rest of the packet */
1481         listen_address = packet_get_string(NULL);
1482         listen_port = packet_get_int();
1483         originator_address = packet_get_string(NULL);
1484         originator_port = packet_get_int();
1485         packet_check_eom();
1486
1487         debug("client_request_forwarded_tcpip: listen %s port %d, "
1488             "originator %s port %d", listen_address, listen_port,
1489             originator_address, originator_port);
1490
1491         c = channel_connect_by_listen_address(listen_port,
1492             "forwarded-tcpip", originator_address);
1493
1494         xfree(originator_address);
1495         xfree(listen_address);
1496         return c;
1497 }
1498
1499 static Channel *
1500 client_request_x11(const char *request_type, int rchan)
1501 {
1502         Channel *c = NULL;
1503         char *originator;
1504         int originator_port;
1505         int sock;
1506
1507         if (!options.forward_x11) {
1508                 error("Warning: ssh server tried X11 forwarding.");
1509                 error("Warning: this is probably a break-in attempt by a malicious server.");
1510                 return NULL;
1511         }
1512         originator = packet_get_string(NULL);
1513         if (datafellows & SSH_BUG_X11FWD) {
1514                 debug2("buggy server: x11 request w/o originator_port");
1515                 originator_port = 0;
1516         } else {
1517                 originator_port = packet_get_int();
1518         }
1519         packet_check_eom();
1520         /* XXX check permission */
1521         debug("client_request_x11: request from %s %d", originator,
1522             originator_port);
1523         xfree(originator);
1524         sock = x11_connect_display();
1525         if (sock < 0)
1526                 return NULL;
1527         c = channel_new("x11",
1528             SSH_CHANNEL_X11_OPEN, sock, sock, -1,
1529             CHAN_TCP_WINDOW_DEFAULT, CHAN_X11_PACKET_DEFAULT, 0, "x11", 1);
1530         c->force_drain = 1;
1531         return c;
1532 }
1533
1534 static Channel *
1535 client_request_agent(const char *request_type, int rchan)
1536 {
1537         Channel *c = NULL;
1538         int sock;
1539
1540         if (!options.forward_agent) {
1541                 error("Warning: ssh server tried agent forwarding.");
1542                 error("Warning: this is probably a break-in attempt by a malicious server.");
1543                 return NULL;
1544         }
1545         sock = ssh_get_authentication_socket();
1546         if (sock < 0)
1547                 return NULL;
1548         c = channel_new("authentication agent connection",
1549             SSH_CHANNEL_OPEN, sock, sock, -1,
1550             CHAN_X11_WINDOW_DEFAULT, CHAN_TCP_PACKET_DEFAULT, 0,
1551             "authentication agent connection", 1);
1552         c->force_drain = 1;
1553         return c;
1554 }
1555
1556 int
1557 client_request_tun_fwd(int tun_mode, int local_tun, int remote_tun)
1558 {
1559         Channel *c;
1560         int fd;
1561
1562         if (tun_mode == SSH_TUNMODE_NO)
1563                 return 0;
1564
1565         if (!compat20) {
1566                 error("Tunnel forwarding is not support for protocol 1");
1567                 return -1;
1568         }
1569
1570         debug("Requesting tun unit %d in mode %d", local_tun, tun_mode);
1571
1572         /* Open local tunnel device */
1573         if ((fd = tun_open(local_tun, tun_mode)) == -1) {
1574                 error("Tunnel device open failed.");
1575                 return -1;
1576         }
1577
1578         c = channel_new("tun", SSH_CHANNEL_OPENING, fd, fd, -1,
1579             CHAN_TCP_WINDOW_DEFAULT, CHAN_TCP_PACKET_DEFAULT, 0, "tun", 1);
1580         c->datagram = 1;
1581
1582 #if defined(SSH_TUN_FILTER)
1583         if (options.tun_open == SSH_TUNMODE_POINTOPOINT)
1584                 channel_register_filter(c->self, sys_tun_infilter,
1585                     sys_tun_outfilter);
1586 #endif
1587
1588         packet_start(SSH2_MSG_CHANNEL_OPEN);
1589         packet_put_cstring("tun@openssh.com");
1590         packet_put_int(c->self);
1591         packet_put_int(c->local_window_max);
1592         packet_put_int(c->local_maxpacket);
1593         packet_put_int(tun_mode);
1594         packet_put_int(remote_tun);
1595         packet_send();
1596
1597         return 0;
1598 }
1599
1600 /* XXXX move to generic input handler */
1601 static void
1602 client_input_channel_open(int type, u_int32_t seq, void *ctxt)
1603 {
1604         Channel *c = NULL;
1605         char *ctype;
1606         int rchan;
1607         u_int rmaxpack, rwindow, len;
1608
1609         ctype = packet_get_string(&len);
1610         rchan = packet_get_int();
1611         rwindow = packet_get_int();
1612         rmaxpack = packet_get_int();
1613
1614         debug("client_input_channel_open: ctype %s rchan %d win %d max %d",
1615             ctype, rchan, rwindow, rmaxpack);
1616
1617         if (strcmp(ctype, "forwarded-tcpip") == 0) {
1618                 c = client_request_forwarded_tcpip(ctype, rchan);
1619         } else if (strcmp(ctype, "x11") == 0) {
1620                 c = client_request_x11(ctype, rchan);
1621         } else if (strcmp(ctype, "auth-agent@openssh.com") == 0) {
1622                 c = client_request_agent(ctype, rchan);
1623         }
1624 /* XXX duplicate : */
1625         if (c != NULL) {
1626                 debug("confirm %s", ctype);
1627                 c->remote_id = rchan;
1628                 c->remote_window = rwindow;
1629                 c->remote_maxpacket = rmaxpack;
1630                 if (c->type != SSH_CHANNEL_CONNECTING) {
1631                         packet_start(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION);
1632                         packet_put_int(c->remote_id);
1633                         packet_put_int(c->self);
1634                         packet_put_int(c->local_window);
1635                         packet_put_int(c->local_maxpacket);
1636                         packet_send();
1637                 }
1638         } else {
1639                 debug("failure %s", ctype);
1640                 packet_start(SSH2_MSG_CHANNEL_OPEN_FAILURE);
1641                 packet_put_int(rchan);
1642                 packet_put_int(SSH2_OPEN_ADMINISTRATIVELY_PROHIBITED);
1643                 if (!(datafellows & SSH_BUG_OPENFAILURE)) {
1644                         packet_put_cstring("open failed");
1645                         packet_put_cstring("");
1646                 }
1647                 packet_send();
1648         }
1649         xfree(ctype);
1650 }
1651 static void
1652 client_input_channel_req(int type, u_int32_t seq, void *ctxt)
1653 {
1654         Channel *c = NULL;
1655         int exitval, id, reply, success = 0;
1656         char *rtype;
1657
1658         id = packet_get_int();
1659         rtype = packet_get_string(NULL);
1660         reply = packet_get_char();
1661
1662         debug("client_input_channel_req: channel %d rtype %s reply %d",
1663             id, rtype, reply);
1664
1665         if (id == -1) {
1666                 error("client_input_channel_req: request for channel -1");
1667         } else if ((c = channel_lookup(id)) == NULL) {
1668                 error("client_input_channel_req: channel %d: unknown channel", id);
1669         } else if (strcmp(rtype, "exit-status") == 0) {
1670                 exitval = packet_get_int();
1671                 if (id == session_ident) {
1672                         success = 1;
1673                         exit_status = exitval;
1674                 } else if (c->ctl_fd == -1) {
1675                         error("client_input_channel_req: unexpected channel %d",
1676                             session_ident);
1677                 } else {
1678                         atomicio(vwrite, c->ctl_fd, &exitval, sizeof(exitval));
1679                         success = 1;
1680                 }
1681                 packet_check_eom();
1682         }
1683         if (reply) {
1684                 packet_start(success ?
1685                     SSH2_MSG_CHANNEL_SUCCESS : SSH2_MSG_CHANNEL_FAILURE);
1686                 packet_put_int(id);
1687                 packet_send();
1688         }
1689         xfree(rtype);
1690 }
1691 static void
1692 client_input_global_request(int type, u_int32_t seq, void *ctxt)
1693 {
1694         char *rtype;
1695         int want_reply;
1696         int success = 0;
1697
1698         rtype = packet_get_string(NULL);
1699         want_reply = packet_get_char();
1700         debug("client_input_global_request: rtype %s want_reply %d",
1701             rtype, want_reply);
1702         if (want_reply) {
1703                 packet_start(success ?
1704                     SSH2_MSG_REQUEST_SUCCESS : SSH2_MSG_REQUEST_FAILURE);
1705                 packet_send();
1706                 packet_write_wait();
1707         }
1708         xfree(rtype);
1709 }
1710
1711 void
1712 client_session2_setup(int id, int want_tty, int want_subsystem,
1713     const char *term, struct termios *tiop, int in_fd, Buffer *cmd, char **env)
1714 {
1715         int len;
1716         Channel *c = NULL;
1717
1718         debug2("%s: id %d", __func__, id);
1719
1720         if ((c = channel_lookup(id)) == NULL)
1721                 fatal("client_session2_setup: channel %d: unknown channel", id);
1722
1723         if (want_tty) {
1724                 struct winsize ws;
1725                 struct termios tio;
1726
1727                 /* Store window size in the packet. */
1728                 if (ioctl(in_fd, TIOCGWINSZ, &ws) < 0)
1729                         memset(&ws, 0, sizeof(ws));
1730
1731                 channel_request_start(id, "pty-req", 1);
1732                 client_expect_confirm(id, "PTY allocation", 0);
1733                 packet_put_cstring(term != NULL ? term : "");
1734                 packet_put_int((u_int)ws.ws_col);
1735                 packet_put_int((u_int)ws.ws_row);
1736                 packet_put_int((u_int)ws.ws_xpixel);
1737                 packet_put_int((u_int)ws.ws_ypixel);
1738                 tio = get_saved_tio();
1739                 tty_make_modes(-1, tiop != NULL ? tiop : &tio);
1740                 packet_send();
1741                 /* XXX wait for reply */
1742                 c->client_tty = 1;
1743         }
1744
1745         /* Transfer any environment variables from client to server */
1746         if (options.num_send_env != 0 && env != NULL) {
1747                 int i, j, matched;
1748                 char *name, *val;
1749
1750                 debug("Sending environment.");
1751                 for (i = 0; env[i] != NULL; i++) {
1752                         /* Split */
1753                         name = xstrdup(env[i]);
1754                         if ((val = strchr(name, '=')) == NULL) {
1755                                 xfree(name);
1756                                 continue;
1757                         }
1758                         *val++ = '\0';
1759
1760                         matched = 0;
1761                         for (j = 0; j < options.num_send_env; j++) {
1762                                 if (match_pattern(name, options.send_env[j])) {
1763                                         matched = 1;
1764                                         break;
1765                                 }
1766                         }
1767                         if (!matched) {
1768                                 debug3("Ignored env %s", name);
1769                                 xfree(name);
1770                                 continue;
1771                         }
1772
1773                         debug("Sending env %s = %s", name, val);
1774                         channel_request_start(id, "env", 0);
1775                         packet_put_cstring(name);
1776                         packet_put_cstring(val);
1777                         packet_send();
1778                         xfree(name);
1779                 }
1780         }
1781
1782         len = buffer_len(cmd);
1783         if (len > 0) {
1784                 if (len > 900)
1785                         len = 900;
1786                 if (want_subsystem) {
1787                         debug("Sending subsystem: %.*s",
1788                             len, (u_char*)buffer_ptr(cmd));
1789                         channel_request_start(id, "subsystem", 1);
1790                         client_expect_confirm(id, "subsystem", 1);
1791                 } else {
1792                         debug("Sending command: %.*s",
1793                             len, (u_char*)buffer_ptr(cmd));
1794                         channel_request_start(id, "exec", 1);
1795                         client_expect_confirm(id, "exec", 1);
1796                 }
1797                 packet_put_string(buffer_ptr(cmd), buffer_len(cmd));
1798                 packet_send();
1799         } else {
1800                 channel_request_start(id, "shell", 1);
1801                 client_expect_confirm(id, "shell", 1);
1802                 packet_send();
1803         }
1804 }
1805
1806 static void
1807 client_init_dispatch_20(void)
1808 {
1809         dispatch_init(&dispatch_protocol_error);
1810
1811         dispatch_set(SSH2_MSG_CHANNEL_CLOSE, &channel_input_oclose);
1812         dispatch_set(SSH2_MSG_CHANNEL_DATA, &channel_input_data);
1813         dispatch_set(SSH2_MSG_CHANNEL_EOF, &channel_input_ieof);
1814         dispatch_set(SSH2_MSG_CHANNEL_EXTENDED_DATA, &channel_input_extended_data);
1815         dispatch_set(SSH2_MSG_CHANNEL_OPEN, &client_input_channel_open);
1816         dispatch_set(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION, &channel_input_open_confirmation);
1817         dispatch_set(SSH2_MSG_CHANNEL_OPEN_FAILURE, &channel_input_open_failure);
1818         dispatch_set(SSH2_MSG_CHANNEL_REQUEST, &client_input_channel_req);
1819         dispatch_set(SSH2_MSG_CHANNEL_WINDOW_ADJUST, &channel_input_window_adjust);
1820         dispatch_set(SSH2_MSG_CHANNEL_SUCCESS, &channel_input_status_confirm);
1821         dispatch_set(SSH2_MSG_CHANNEL_FAILURE, &channel_input_status_confirm);
1822         dispatch_set(SSH2_MSG_GLOBAL_REQUEST, &client_input_global_request);
1823
1824         /* rekeying */
1825         dispatch_set(SSH2_MSG_KEXINIT, &kex_input_kexinit);
1826
1827         /* global request reply messages */
1828         dispatch_set(SSH2_MSG_REQUEST_FAILURE, &client_global_request_reply);
1829         dispatch_set(SSH2_MSG_REQUEST_SUCCESS, &client_global_request_reply);
1830 }
1831 static void
1832 client_init_dispatch_13(void)
1833 {
1834         dispatch_init(NULL);
1835         dispatch_set(SSH_MSG_CHANNEL_CLOSE, &channel_input_close);
1836         dispatch_set(SSH_MSG_CHANNEL_CLOSE_CONFIRMATION, &channel_input_close_confirmation);
1837         dispatch_set(SSH_MSG_CHANNEL_DATA, &channel_input_data);
1838         dispatch_set(SSH_MSG_CHANNEL_OPEN_CONFIRMATION, &channel_input_open_confirmation);
1839         dispatch_set(SSH_MSG_CHANNEL_OPEN_FAILURE, &channel_input_open_failure);
1840         dispatch_set(SSH_MSG_PORT_OPEN, &channel_input_port_open);
1841         dispatch_set(SSH_SMSG_EXITSTATUS, &client_input_exit_status);
1842         dispatch_set(SSH_SMSG_STDERR_DATA, &client_input_stderr_data);
1843         dispatch_set(SSH_SMSG_STDOUT_DATA, &client_input_stdout_data);
1844
1845         dispatch_set(SSH_SMSG_AGENT_OPEN, options.forward_agent ?
1846             &client_input_agent_open : &deny_input_open);
1847         dispatch_set(SSH_SMSG_X11_OPEN, options.forward_x11 ?
1848             &x11_input_open : &deny_input_open);
1849 }
1850 static void
1851 client_init_dispatch_15(void)
1852 {
1853         client_init_dispatch_13();
1854         dispatch_set(SSH_MSG_CHANNEL_CLOSE, &channel_input_ieof);
1855         dispatch_set(SSH_MSG_CHANNEL_CLOSE_CONFIRMATION, & channel_input_oclose);
1856 }
1857 static void
1858 client_init_dispatch(void)
1859 {
1860         if (compat20)
1861                 client_init_dispatch_20();
1862         else if (compat13)
1863                 client_init_dispatch_13();
1864         else
1865                 client_init_dispatch_15();
1866 }
1867
1868 /* client specific fatal cleanup */
1869 void
1870 cleanup_exit(int i)
1871 {
1872         leave_raw_mode();
1873         leave_non_blocking();
1874         if (options.control_path != NULL && muxserver_sock != -1)
1875                 unlink(options.control_path);
1876         _exit(i);
1877 }
This page took 0.29707 seconds and 5 git commands to generate.