blob: 584afb76ae5cd048830ffdd3e20dae2c676e0a39 [file] [log] [blame]
Damien Millerbab9bd42008-05-19 16:06:47 +10001/* $OpenBSD: clientloop.c,v 1.193 2008/05/09 16:21:13 markus Exp $ */
Damien Millerd4a8b7e1999-10-27 13:42:43 +10002/*
Damien Miller95def091999-11-25 00:26:21 +11003 * Author: Tatu Ylonen <ylo@cs.hut.fi>
Damien Miller95def091999-11-25 00:26:21 +11004 * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
5 * All rights reserved
Damien Miller95def091999-11-25 00:26:21 +11006 * The main loop for the interactive session (client side).
Damien Miller4af51302000-04-16 11:18:38 +10007 *
Damien Millere4340be2000-09-16 13:29:08 +11008 * 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 *
Damien Miller1383bd82000-04-06 12:32:37 +100038 * SSH2 support added by Markus Friedl.
Ben Lindstrom44697232001-07-04 03:32:30 +000039 * Copyright (c) 1999, 2000, 2001 Markus Friedl. All rights reserved.
Damien Millere4340be2000-09-16 13:29:08 +110040 *
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.
Damien Miller95def091999-11-25 00:26:21 +110060 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +100061
62#include "includes.h"
Damien Miller17e91c02006-03-15 11:28:34 +110063
Damien Millerf17883e2006-03-15 11:45:54 +110064#include <sys/types.h>
Damien Miller9aec9192006-08-05 10:57:45 +100065#include <sys/ioctl.h>
Damien Miller8dbffe72006-08-05 11:02:17 +100066#include <sys/param.h>
Damien Millerf17883e2006-03-15 11:45:54 +110067#ifdef HAVE_SYS_STAT_H
68# include <sys/stat.h>
69#endif
Damien Miller9aec9192006-08-05 10:57:45 +100070#ifdef HAVE_SYS_TIME_H
71# include <sys/time.h>
72#endif
Damien Millere3b60b52006-07-10 21:08:03 +100073#include <sys/socket.h>
Damien Miller99bd21e2006-03-15 11:11:28 +110074
Damien Millerc7b06362006-03-15 11:53:45 +110075#include <ctype.h>
Darren Tucker39972492006-07-12 22:22:46 +100076#include <errno.h>
Damien Miller6645e7a2006-03-15 14:42:54 +110077#ifdef HAVE_PATHS_H
Damien Miller03e20032006-03-15 11:16:59 +110078#include <paths.h>
Damien Miller6645e7a2006-03-15 14:42:54 +110079#endif
Damien Miller6ff3cad2006-03-15 11:52:09 +110080#include <signal.h>
Damien Millerded319c2006-09-01 15:38:36 +100081#include <stdarg.h>
Damien Millera7a73ee2006-08-05 11:37:59 +100082#include <stdio.h>
Damien Millere7a1e5c2006-08-05 11:34:19 +100083#include <stdlib.h>
Damien Millere3476ed2006-07-24 14:13:33 +100084#include <string.h>
Damien Miller99bd21e2006-03-15 11:11:28 +110085#include <termios.h>
Damien Millerd7834352006-08-05 12:39:39 +100086#include <pwd.h>
Damien Millere6b3b612006-07-24 14:01:23 +100087#include <unistd.h>
Damien Millerd4a8b7e1999-10-27 13:42:43 +100088
Damien Millerb84886b2008-05-19 15:05:07 +100089#include "openbsd-compat/sys-queue.h"
Damien Millerd7834352006-08-05 12:39:39 +100090#include "xmalloc.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100091#include "ssh.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +000092#include "ssh1.h"
93#include "ssh2.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100094#include "packet.h"
95#include "buffer.h"
Damien Millerb38eff82000-04-01 11:09:21 +100096#include "compat.h"
Ben Lindstromc7637672001-06-09 00:36:26 +000097#include "channels.h"
Damien Millerb38eff82000-04-01 11:09:21 +100098#include "dispatch.h"
Damien Miller0bc1bd82000-11-13 22:57:25 +110099#include "key.h"
Damien Millerd7834352006-08-05 12:39:39 +1000100#include "cipher.h"
Ben Lindstromf28f6342001-04-04 02:03:04 +0000101#include "kex.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000102#include "log.h"
103#include "readconf.h"
Ben Lindstrombf555ba2001-01-18 02:04:35 +0000104#include "clientloop.h"
Damien Milleraeb31d62005-12-13 19:29:36 +1100105#include "sshconnect.h"
Ben Lindstrom226cfa02001-01-22 05:34:40 +0000106#include "authfd.h"
107#include "atomicio.h"
Darren Tucker06f2bd82004-05-13 16:06:46 +1000108#include "sshpty.h"
Ben Lindstrom302ea6f2001-04-16 02:01:25 +0000109#include "misc.h"
Damien Miller0e220db2004-06-15 10:34:08 +1000110#include "monitor_fdpass.h"
111#include "match.h"
112#include "msg.h"
Damien Miller69b69aa2000-10-28 14:19:58 +1100113
114/* import options */
115extern Options options;
116
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000117/* Flag indicating that stdin should be redirected from /dev/null. */
118extern int stdin_null_flag;
119
Damien Millerd6965512003-12-17 16:31:53 +1100120/* Flag indicating that no shell has been requested */
121extern int no_shell_flag;
122
Damien Miller0e220db2004-06-15 10:34:08 +1000123/* Control socket */
Damien Millerb1cbfa22008-05-19 16:00:08 +1000124extern int muxserver_sock;
Damien Miller0e220db2004-06-15 10:34:08 +1000125
Damien Miller5428f641999-11-25 11:54:57 +1100126/*
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 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000131extern char *host;
132
Damien Miller5428f641999-11-25 11:54:57 +1100133/*
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 */
Ben Lindstrom5e71c542001-12-06 16:48:14 +0000139static volatile sig_atomic_t received_window_change_signal = 0;
140static volatile sig_atomic_t received_signal = 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000141
Damien Miller788f2122005-11-05 15:14:59 +1100142/* Flag indicating whether the user's terminal is in non-blocking mode. */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000143static int in_non_blocking_mode = 0;
144
145/* Common data for the client loop code. */
Damien Miller97c91f62006-04-23 12:08:37 +1000146static volatile sig_atomic_t quit_pending; /* Set non-zero to quit the loop. */
Damien Millerad833b32000-08-23 10:46:23 +1000147static int escape_char; /* Escape character. */
Damien Miller95def091999-11-25 00:26:21 +1100148static int escape_pending; /* Last character was the escape character */
149static int last_was_cr; /* Last character was a newline. */
150static int exit_status; /* Used to store the exit status of the command. */
151static int stdin_eof; /* EOF has been encountered on standard error. */
152static Buffer stdin_buffer; /* Buffer for stdin data. */
153static Buffer stdout_buffer; /* Buffer for stdout data. */
154static Buffer stderr_buffer; /* Buffer for stderr data. */
Ben Lindstrom46c16222000-12-22 01:43:59 +0000155static u_long stdin_bytes, stdout_bytes, stderr_bytes;
156static u_int buffer_high;/* Soft max buffer size. */
Damien Miller95def091999-11-25 00:26:21 +1100157static int connection_in; /* Connection to server (input). */
158static int connection_out; /* Connection to server (output). */
Ben Lindstrombe2cc432001-04-04 23:46:07 +0000159static int need_rekeying; /* Set to non-zero if rekeying is requested. */
Ben Lindstrom4c3f77d2001-04-05 23:37:36 +0000160static int session_closed = 0; /* In SSH2: login session closed. */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000161
Ben Lindstrombba81212001-06-25 05:01:22 +0000162static void client_init_dispatch(void);
Damien Miller1383bd82000-04-06 12:32:37 +1000163int session_ident = -1;
164
Damien Miller5771ed72008-05-19 15:35:33 +1000165struct channel_reply_ctx {
166 const char *request_type;
167 int id, do_close;
168};
169
Ben Lindstromf28f6342001-04-04 02:03:04 +0000170/*XXX*/
171extern Kex *xxx_kex;
172
Damien Miller0e220db2004-06-15 10:34:08 +1000173void ssh_process_session2_setup(int, int, int, Buffer *);
174
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000175/* Restores stdin to blocking mode. */
176
Ben Lindstrombba81212001-06-25 05:01:22 +0000177static void
Ben Lindstrom31ca54a2001-02-09 02:11:24 +0000178leave_non_blocking(void)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000179{
Damien Miller95def091999-11-25 00:26:21 +1100180 if (in_non_blocking_mode) {
Damien Miller03e66f62004-06-15 15:47:51 +1000181 unset_nonblock(fileno(stdin));
Damien Miller95def091999-11-25 00:26:21 +1100182 in_non_blocking_mode = 0;
Damien Miller95def091999-11-25 00:26:21 +1100183 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000184}
185
Damien Miller95def091999-11-25 00:26:21 +1100186/* Puts stdin terminal in non-blocking mode. */
187
Ben Lindstrombba81212001-06-25 05:01:22 +0000188static void
Ben Lindstrom31ca54a2001-02-09 02:11:24 +0000189enter_non_blocking(void)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000190{
Damien Miller95def091999-11-25 00:26:21 +1100191 in_non_blocking_mode = 1;
Damien Miller232711f2004-06-15 10:35:30 +1000192 set_nonblock(fileno(stdin));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000193}
194
Damien Miller5428f641999-11-25 11:54:57 +1100195/*
196 * Signal handler for the window change signal (SIGWINCH). This just sets a
197 * flag indicating that the window has changed.
198 */
Damien Millerf0b15df2006-03-26 13:59:20 +1100199/*ARGSUSED */
Ben Lindstrombba81212001-06-25 05:01:22 +0000200static void
Damien Miller95def091999-11-25 00:26:21 +1100201window_change_handler(int sig)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000202{
Damien Miller95def091999-11-25 00:26:21 +1100203 received_window_change_signal = 1;
204 signal(SIGWINCH, window_change_handler);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000205}
206
Damien Miller5428f641999-11-25 11:54:57 +1100207/*
208 * Signal handler for signals that cause the program to terminate. These
209 * signals must be trapped to restore terminal modes.
210 */
Damien Millerf0b15df2006-03-26 13:59:20 +1100211/*ARGSUSED */
Ben Lindstrombba81212001-06-25 05:01:22 +0000212static void
Damien Miller95def091999-11-25 00:26:21 +1100213signal_handler(int sig)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000214{
Ben Lindstromec46e0b2001-06-09 01:27:31 +0000215 received_signal = sig;
216 quit_pending = 1;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000217}
218
Damien Miller5428f641999-11-25 11:54:57 +1100219/*
220 * Returns current time in seconds from Jan 1, 1970 with the maximum
221 * available resolution.
222 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000223
Ben Lindstrombba81212001-06-25 05:01:22 +0000224static double
Ben Lindstrom31ca54a2001-02-09 02:11:24 +0000225get_current_time(void)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000226{
Damien Miller95def091999-11-25 00:26:21 +1100227 struct timeval tv;
228 gettimeofday(&tv, NULL);
229 return (double) tv.tv_sec + (double) tv.tv_usec / 1000000.0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000230}
231
Damien Miller17e7ed02005-06-17 12:54:33 +1000232#define SSH_X11_PROTO "MIT-MAGIC-COOKIE-1"
233void
234client_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 }
Darren Tucker513d13a2007-08-15 19:13:41 +1000287
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");
Damien Miller17e7ed02005-06-17 12:54:33 +1000310 }
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
Damien Miller5428f641999-11-25 11:54:57 +1100345/*
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 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000350
Ben Lindstrombba81212001-06-25 05:01:22 +0000351static void
Ben Lindstrom31ca54a2001-02-09 02:11:24 +0000352client_check_initial_eof_on_stdin(void)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000353{
Damien Miller95def091999-11-25 00:26:21 +1100354 int len;
355 char buf[1];
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000356
Damien Miller5428f641999-11-25 11:54:57 +1100357 /*
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 */
Damien Miller95def091999-11-25 00:26:21 +1100365 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 {
Damien Miller95def091999-11-25 00:26:21 +1100372 enter_non_blocking();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000373
Damien Miller95def091999-11-25 00:26:21 +1100374 /* Check for immediate EOF on stdin. */
375 len = read(fileno(stdin), buf, 1);
376 if (len == 0) {
Damien Miller5428f641999-11-25 11:54:57 +1100377 /* EOF. Record that we have seen it and send EOF to server. */
Damien Miller95def091999-11-25 00:26:21 +1100378 debug("Sending eof.");
379 stdin_eof = 1;
380 packet_start(SSH_CMSG_EOF);
381 packet_send();
382 } else if (len > 0) {
Damien Miller5428f641999-11-25 11:54:57 +1100383 /*
384 * Got data. We must store the data in the buffer,
385 * and also process it as an escape character if
386 * appropriate.
387 */
Ben Lindstrom46c16222000-12-22 01:43:59 +0000388 if ((u_char) buf[0] == escape_char)
Damien Miller95def091999-11-25 00:26:21 +1100389 escape_pending = 1;
Ben Lindstrome9613cf2001-03-05 06:14:02 +0000390 else
Damien Miller95def091999-11-25 00:26:21 +1100391 buffer_append(&stdin_buffer, buf, 1);
Damien Miller95def091999-11-25 00:26:21 +1100392 }
Damien Miller95def091999-11-25 00:26:21 +1100393 leave_non_blocking();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000394 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000395}
396
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000397
Damien Miller5428f641999-11-25 11:54:57 +1100398/*
399 * Make packets from buffered stdin data, and buffer them for sending to the
400 * connection.
401 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000402
Ben Lindstrombba81212001-06-25 05:01:22 +0000403static void
Ben Lindstrom31ca54a2001-02-09 02:11:24 +0000404client_make_packets_from_stdin_data(void)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000405{
Ben Lindstrom46c16222000-12-22 01:43:59 +0000406 u_int len;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000407
Damien Miller95def091999-11-25 00:26:21 +1100408 /* Send buffered stdin data to the server. */
409 while (buffer_len(&stdin_buffer) > 0 &&
Damien Miller9f0f5c62001-12-21 14:45:46 +1100410 packet_not_very_much_data_to_write()) {
Damien Miller95def091999-11-25 00:26:21 +1100411 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);
Ben Lindstrome9613cf2001-03-05 06:14:02 +0000419 stdin_bytes += len;
Damien Miller95def091999-11-25 00:26:21 +1100420 /* 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 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000425 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000426}
427
Damien Miller5428f641999-11-25 11:54:57 +1100428/*
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 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000434
Ben Lindstrombba81212001-06-25 05:01:22 +0000435static void
Ben Lindstrom31ca54a2001-02-09 02:11:24 +0000436client_check_window_change(void)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000437{
Damien Miller1383bd82000-04-06 12:32:37 +1000438 struct winsize ws;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000439
Damien Miller1383bd82000-04-06 12:32:37 +1000440 if (! received_window_change_signal)
441 return;
442 /** XXX race */
443 received_window_change_signal = 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000444
Damien Millerd3444942000-09-30 14:20:03 +1100445 debug2("client_check_window_change: changed");
Damien Miller1383bd82000-04-06 12:32:37 +1000446
447 if (compat20) {
Damien Miller0e220db2004-06-15 10:34:08 +1000448 channel_send_window_changes();
Damien Miller1383bd82000-04-06 12:32:37 +1000449 } else {
Damien Miller0e220db2004-06-15 10:34:08 +1000450 if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) < 0)
451 return;
Damien Miller1383bd82000-04-06 12:32:37 +1000452 packet_start(SSH_CMSG_WINDOW_SIZE);
Damien Miller71a73672006-03-26 14:04:36 +1100453 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);
Damien Miller1383bd82000-04-06 12:32:37 +1000457 packet_send();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000458 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000459}
460
Damien Miller509b0102003-12-17 16:33:10 +1100461static void
462client_global_request_reply(int type, u_int32_t seq, void *ctxt)
463{
Damien Miller58226f62008-03-07 18:33:30 +1100464 keep_alive_timeouts = 0;
Damien Miller509b0102003-12-17 16:33:10 +1100465 client_global_request_reply_fwd(type, seq, ctxt);
466}
467
468static void
469server_alive_check(void)
470{
Damien Miller58226f62008-03-07 18:33:30 +1100471 if (++keep_alive_timeouts > options.server_alive_count_max) {
Damien Miller985a4482006-10-24 03:02:41 +1000472 logit("Timeout, server not responding.");
473 cleanup_exit(255);
474 }
Damien Miller509b0102003-12-17 16:33:10 +1100475 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
Damien Miller5428f641999-11-25 11:54:57 +1100481/*
482 * Waits until the client can do something (some data becomes available on
483 * one of the file descriptors).
484 */
Ben Lindstrombba81212001-06-25 05:01:22 +0000485static void
Damien Miller5e953212001-01-30 09:14:00 +1100486client_wait_until_can_do_something(fd_set **readsetp, fd_set **writesetp,
Darren Tuckerc7a6fc42004-08-13 21:18:00 +1000487 int *maxfdp, u_int *nallocp, int rekeying)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000488{
Damien Miller509b0102003-12-17 16:33:10 +1100489 struct timeval tv, *tvp;
490 int ret;
491
Damien Miller5e953212001-01-30 09:14:00 +1100492 /* Add any selections by the channel mechanism. */
Ben Lindstrom16d29d52001-07-18 16:01:46 +0000493 channel_prepare_select(readsetp, writesetp, maxfdp, nallocp, rekeying);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000494
Damien Miller1383bd82000-04-06 12:32:37 +1000495 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())
Damien Miller5e953212001-01-30 09:14:00 +1100500 FD_SET(connection_in, *readsetp);
Damien Miller1383bd82000-04-06 12:32:37 +1000501 /*
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())
Damien Miller5e953212001-01-30 09:14:00 +1100506 FD_SET(fileno(stdin), *readsetp);
Damien Miller1383bd82000-04-06 12:32:37 +1000507
508 /* Select stdout/stderr if have data in buffer. */
509 if (buffer_len(&stdout_buffer) > 0)
Damien Miller5e953212001-01-30 09:14:00 +1100510 FD_SET(fileno(stdout), *writesetp);
Damien Miller1383bd82000-04-06 12:32:37 +1000511 if (buffer_len(&stderr_buffer) > 0)
Damien Miller5e953212001-01-30 09:14:00 +1100512 FD_SET(fileno(stderr), *writesetp);
Damien Miller1383bd82000-04-06 12:32:37 +1000513 } else {
Ben Lindstromc8b3f472001-05-17 03:19:40 +0000514 /* channel_prepare_select could have closed the last channel */
Damien Miller164a7f42001-10-12 11:36:09 +1000515 if (session_closed && !channel_still_open() &&
516 !packet_have_data_to_write()) {
517 /* clear mask since we did not call select() */
Damien Miller79faeff2001-11-12 11:06:32 +1100518 memset(*readsetp, 0, *nallocp);
519 memset(*writesetp, 0, *nallocp);
Damien Miller164a7f42001-10-12 11:36:09 +1000520 return;
Ben Lindstromc8b3f472001-05-17 03:19:40 +0000521 } else {
522 FD_SET(connection_in, *readsetp);
523 }
Damien Miller1383bd82000-04-06 12:32:37 +1000524 }
525
Damien Miller95def091999-11-25 00:26:21 +1100526 /* Select server connection if have data to write to the server. */
527 if (packet_have_data_to_write())
Damien Miller5e953212001-01-30 09:14:00 +1100528 FD_SET(connection_out, *writesetp);
Damien Miller95def091999-11-25 00:26:21 +1100529
Damien Millerb1cbfa22008-05-19 16:00:08 +1000530 if (muxserver_sock != -1)
531 FD_SET(muxserver_sock, *readsetp);
Damien Miller0e220db2004-06-15 10:34:08 +1000532
Damien Miller5428f641999-11-25 11:54:57 +1100533 /*
534 * Wait for something to happen. This will suspend the process until
535 * some selected descriptor can be read, written, or has some other
Damien Miller509b0102003-12-17 16:33:10 +1100536 * event pending.
Damien Miller5428f641999-11-25 11:54:57 +1100537 */
Damien Miller95def091999-11-25 00:26:21 +1100538
Damien Miller509b0102003-12-17 16:33:10 +1100539 if (options.server_alive_interval == 0 || !compat20)
540 tvp = NULL;
Darren Tuckerfc959702004-07-17 16:12:08 +1000541 else {
Damien Miller509b0102003-12-17 16:33:10 +1100542 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) {
Damien Miller95def091999-11-25 00:26:21 +1100548 char buf[100];
Ben Lindstromf9452512001-02-15 03:12:08 +0000549
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 */
Damien Miller79faeff2001-11-12 11:06:32 +1100555 memset(*readsetp, 0, *nallocp);
556 memset(*writesetp, 0, *nallocp);
Ben Lindstromf9452512001-02-15 03:12:08 +0000557
Damien Miller95def091999-11-25 00:26:21 +1100558 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));
Damien Miller95def091999-11-25 00:26:21 +1100563 quit_pending = 1;
Damien Miller509b0102003-12-17 16:33:10 +1100564 } else if (ret == 0)
565 server_alive_check();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000566}
567
Ben Lindstrombba81212001-06-25 05:01:22 +0000568static void
Damien Millerad833b32000-08-23 10:46:23 +1000569client_suspend_self(Buffer *bin, Buffer *bout, Buffer *berr)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000570{
Damien Miller95def091999-11-25 00:26:21 +1100571 /* Flush stdout and stderr buffers. */
Damien Millerad833b32000-08-23 10:46:23 +1000572 if (buffer_len(bout) > 0)
Darren Tucker9f63f222003-07-03 13:46:56 +1000573 atomicio(vwrite, fileno(stdout), buffer_ptr(bout), buffer_len(bout));
Damien Millerad833b32000-08-23 10:46:23 +1000574 if (buffer_len(berr) > 0)
Darren Tucker9f63f222003-07-03 13:46:56 +1000575 atomicio(vwrite, fileno(stderr), buffer_ptr(berr), buffer_len(berr));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000576
Damien Miller95def091999-11-25 00:26:21 +1100577 leave_raw_mode();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000578
Damien Miller5428f641999-11-25 11:54:57 +1100579 /*
580 * Free (and clear) the buffer to reduce the amount of data that gets
581 * written to swap.
582 */
Damien Millerad833b32000-08-23 10:46:23 +1000583 buffer_free(bin);
584 buffer_free(bout);
585 buffer_free(berr);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000586
Damien Miller95def091999-11-25 00:26:21 +1100587 /* Send the suspend signal to the program itself. */
588 kill(getpid(), SIGTSTP);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000589
Darren Tucker5d78de62004-11-05 20:35:44 +1100590 /* Reset window sizes in case they have changed */
591 received_window_change_signal = 1;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000592
Damien Miller95def091999-11-25 00:26:21 +1100593 /* OK, we have been continued by the user. Reinitialize buffers. */
Damien Millerad833b32000-08-23 10:46:23 +1000594 buffer_init(bin);
595 buffer_init(bout);
596 buffer_init(berr);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000597
Damien Miller95def091999-11-25 00:26:21 +1100598 enter_raw_mode();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000599}
600
Ben Lindstrombba81212001-06-25 05:01:22 +0000601static void
Damien Miller90fdfaf2006-03-26 14:25:37 +1100602client_process_net_input(fd_set *readset)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000603{
Damien Miller1383bd82000-04-06 12:32:37 +1000604 int len;
605 char buf[8192];
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000606
Damien Miller5428f641999-11-25 11:54:57 +1100607 /*
608 * Read input from the server, and add any such data to the buffer of
609 * the packet subsystem.
610 */
Damien Miller95def091999-11-25 00:26:21 +1100611 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);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000618 buffer_append(&stderr_buffer, buf, strlen(buf));
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000619 quit_pending = 1;
620 return;
Damien Miller95def091999-11-25 00:26:21 +1100621 }
Damien Miller5428f641999-11-25 11:54:57 +1100622 /*
623 * There is a kernel bug on Solaris that causes select to
624 * sometimes wake up even though there is no data available.
625 */
Ben Lindstromebc88272001-03-06 03:34:40 +0000626 if (len < 0 && (errno == EAGAIN || errno == EINTR))
Damien Miller95def091999-11-25 00:26:21 +1100627 len = 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000628
Damien Miller95def091999-11-25 00:26:21 +1100629 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));
Damien Miller95def091999-11-25 00:26:21 +1100634 quit_pending = 1;
635 return;
636 }
637 packet_process_incoming(buf, len);
638 }
Damien Miller1383bd82000-04-06 12:32:37 +1000639}
640
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000641static void
Damien Miller5771ed72008-05-19 15:35:33 +1000642client_status_confirm(int type, Channel *c, void *ctx)
Damien Miller0e220db2004-06-15 10:34:08 +1000643{
Damien Miller5771ed72008-05-19 15:35:33 +1000644 struct channel_reply_ctx *cr = (struct channel_reply_ctx *)ctx;
645 char errmsg[256];
646 int tochan;
Darren Tuckerfc959702004-07-17 16:12:08 +1000647
Damien Miller5771ed72008-05-19 15:35:33 +1000648 /* 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;
Damien Miller0e220db2004-06-15 10:34:08 +1000651
Damien Miller5771ed72008-05-19 15:35:33 +1000652 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 }
Damien Miller0e220db2004-06-15 10:34:08 +1000676 }
Damien Miller5771ed72008-05-19 15:35:33 +1000677 xfree(cr);
678}
Damien Miller0e220db2004-06-15 10:34:08 +1000679
Damien Miller5771ed72008-05-19 15:35:33 +1000680static void
681client_abandon_status_confirm(Channel *c, void *ctx)
682{
683 xfree(ctx);
684}
685
686static void
687client_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);
Damien Miller0e220db2004-06-15 10:34:08 +1000696}
697
698static void
Ben Lindstrom681d9322002-03-22 03:53:00 +0000699process_cmdline(void)
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000700{
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000701 void (*handler)(int);
Damien Millerf91ee4c2005-03-01 21:24:33 +1100702 char *s, *cmd, *cancel_host;
Darren Tuckere7066df2004-05-24 10:18:05 +1000703 int delete = 0;
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000704 int local = 0;
Damien Millerf91ee4c2005-03-01 21:24:33 +1100705 u_short cancel_port;
706 Forward fwd;
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000707
Darren Tucker23ae8ca2007-12-02 23:12:30 +1100708 bzero(&fwd, sizeof(fwd));
709 fwd.listen_host = fwd.connect_host = NULL;
710
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000711 leave_raw_mode();
Ben Lindstrom5a6abda2002-06-09 19:41:48 +0000712 handler = signal(SIGINT, SIG_IGN);
Ben Lindstrom681d9322002-03-22 03:53:00 +0000713 cmd = s = read_passphrase("\r\nssh> ", RP_ECHO);
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000714 if (s == NULL)
715 goto out;
Darren Tucker03b1cdb2007-03-21 20:46:03 +1100716 while (isspace(*s))
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000717 s++;
Darren Tuckere7066df2004-05-24 10:18:05 +1000718 if (*s == '-')
719 s++; /* Skip cmdline '-', if any */
Darren Tuckere1675822004-05-24 10:13:07 +1000720 if (*s == '\0')
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000721 goto out;
Darren Tuckere7066df2004-05-24 10:18:05 +1000722
Darren Tucker1973c882004-05-24 10:34:36 +1000723 if (*s == 'h' || *s == 'H' || *s == '?') {
Darren Tuckere7066df2004-05-24 10:18:05 +1000724 logit("Commands:");
Damien Miller43020952006-07-10 20:16:12 +1000725 logit(" -L[bind_address:]port:host:hostport "
726 "Request local forward");
727 logit(" -R[bind_address:]port:host:hostport "
728 "Request remote forward");
Damien Miller57e8ad32006-07-10 20:20:52 +1000729 logit(" -KR[bind_address:]port "
Damien Miller43020952006-07-10 20:16:12 +1000730 "Cancel remote forward");
Damien Millerd27b9472005-12-13 19:29:02 +1100731 if (!options.permit_local_command)
732 goto out;
Damien Miller43020952006-07-10 20:16:12 +1000733 logit(" !args "
734 "Execute local command");
Damien Millerd27b9472005-12-13 19:29:02 +1100735 goto out;
736 }
737
738 if (*s == '!' && options.permit_local_command) {
739 s++;
740 ssh_local_cmd(s);
Darren Tuckere7066df2004-05-24 10:18:05 +1000741 goto out;
742 }
743
744 if (*s == 'K') {
745 delete = 1;
746 s++;
747 }
748 if (*s != 'L' && *s != 'R') {
Damien Miller996acd22003-04-09 20:59:48 +1000749 logit("Invalid command.");
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000750 goto out;
751 }
Darren Tuckere7066df2004-05-24 10:18:05 +1000752 if (*s == 'L')
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000753 local = 1;
Darren Tuckere7066df2004-05-24 10:18:05 +1000754 if (local && delete) {
755 logit("Not supported.");
756 goto out;
757 }
758 if ((!local || delete) && !compat20) {
Damien Miller996acd22003-04-09 20:59:48 +1000759 logit("Not supported for SSH protocol version 1.");
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000760 goto out;
761 }
Darren Tuckere7066df2004-05-24 10:18:05 +1000762
Darren Tucker03b1cdb2007-03-21 20:46:03 +1100763 while (isspace(*++s))
764 ;
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000765
Darren Tuckere7066df2004-05-24 10:18:05 +1000766 if (delete) {
Damien Millerf91ee4c2005-03-01 21:24:33 +1100767 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");
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000778 goto out;
779 }
Damien Millerf91ee4c2005-03-01 21:24:33 +1100780 channel_request_rforward_cancel(cancel_host, cancel_port);
Darren Tuckere7066df2004-05-24 10:18:05 +1000781 } else {
Damien Millerf91ee4c2005-03-01 21:24:33 +1100782 if (!parse_forward(&fwd, s)) {
Darren Tuckere7066df2004-05-24 10:18:05 +1000783 logit("Bad forwarding specification.");
784 goto out;
785 }
Darren Tuckere7066df2004-05-24 10:18:05 +1000786 if (local) {
Damien Millerf91ee4c2005-03-01 21:24:33 +1100787 if (channel_setup_local_fwd_listener(fwd.listen_host,
788 fwd.listen_port, fwd.connect_host,
789 fwd.connect_port, options.gateway_ports) < 0) {
Darren Tuckere7066df2004-05-24 10:18:05 +1000790 logit("Port forwarding failed.");
791 goto out;
792 }
Damien Millerf91ee4c2005-03-01 21:24:33 +1100793 } else {
Darren Tuckere7d4b192006-07-12 22:17:10 +1000794 if (channel_request_remote_forwarding(fwd.listen_host,
Damien Millerf91ee4c2005-03-01 21:24:33 +1100795 fwd.listen_port, fwd.connect_host,
Darren Tuckere7d4b192006-07-12 22:17:10 +1000796 fwd.connect_port) < 0) {
797 logit("Port forwarding failed.");
798 goto out;
799 }
Damien Millerf91ee4c2005-03-01 21:24:33 +1100800 }
801
Darren Tuckere7066df2004-05-24 10:18:05 +1000802 logit("Forwarding port.");
803 }
804
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000805out:
806 signal(SIGINT, handler);
807 enter_raw_mode();
808 if (cmd)
809 xfree(cmd);
Darren Tucker23ae8ca2007-12-02 23:12:30 +1100810 if (fwd.listen_host != NULL)
811 xfree(fwd.listen_host);
812 if (fwd.connect_host != NULL)
813 xfree(fwd.connect_host);
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000814}
815
Damien Millerad833b32000-08-23 10:46:23 +1000816/* process the characters one by one */
Ben Lindstrombba81212001-06-25 05:01:22 +0000817static int
Damien Millerad833b32000-08-23 10:46:23 +1000818process_escapes(Buffer *bin, Buffer *bout, Buffer *berr, char *buf, int len)
819{
820 char string[1024];
821 pid_t pid;
822 int bytes = 0;
Ben Lindstrom46c16222000-12-22 01:43:59 +0000823 u_int i;
824 u_char ch;
Damien Millerad833b32000-08-23 10:46:23 +1000825 char *s;
826
Damien Millereccb9de2005-06-17 12:59:34 +1000827 if (len <= 0)
828 return (0);
829
830 for (i = 0; i < (u_int)len; i++) {
Damien Millerad833b32000-08-23 10:46:23 +1000831 /* 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));
Damien Millerad833b32000-08-23 10:46:23 +1000845
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));
Damien Millerad833b32000-08-23 10:46:23 +1000854
855 /* Restore terminal modes and suspend. */
856 client_suspend_self(bin, bout, berr);
857
858 /* We have been continued. */
859 continue;
860
Damien Miller54c45982003-05-15 10:20:13 +1000861 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
Ben Lindstromf28f6342001-04-04 02:03:04 +0000874 case 'R':
Ben Lindstrom11bd8992001-04-05 23:34:29 +0000875 if (compat20) {
876 if (datafellows & SSH_BUG_NOREKEY)
Damien Miller996acd22003-04-09 20:59:48 +1000877 logit("Server does not support re-keying");
Ben Lindstrom11bd8992001-04-05 23:34:29 +0000878 else
879 need_rekeying = 1;
880 }
Ben Lindstromf28f6342001-04-04 02:03:04 +0000881 continue;
882
Damien Millerad833b32000-08-23 10:46:23 +1000883 case '&':
Damien Millerad833b32000-08-23 10:46:23 +1000884 /*
885 * Detach the program (continue to serve connections,
886 * but put in background and no more new connections).
887 */
Damien Miller96507ef2001-11-12 10:52:25 +1100888 /* 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) {
Damien Millerad833b32000-08-23 10:46:23 +1000914 /*
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 }
Damien Miller96507ef2001-11-12 10:52:25 +1100929 continue;
Damien Millerad833b32000-08-23 10:46:23 +1000930
931 case '?':
932 snprintf(string, sizeof string,
933"%c?\r\n\
934Supported escape sequences:\r\n\
Damien Miller06692862002-09-04 16:32:10 +1000935%c. - terminate connection\r\n\
Damien Miller54c45982003-05-15 10:20:13 +1000936%cB - send a BREAK to the remote system\r\n\
Damien Miller06692862002-09-04 16:32:10 +1000937%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\
Damien Millerad833b32000-08-23 10:46:23 +1000944(Note that escapes are only recognized immediately after newline.)\r\n",
Damien Miller06692862002-09-04 16:32:10 +1000945 escape_char, escape_char, escape_char, escape_char,
946 escape_char, escape_char, escape_char, escape_char,
Damien Miller54c45982003-05-15 10:20:13 +1000947 escape_char, escape_char, escape_char);
Damien Millerad833b32000-08-23 10:46:23 +1000948 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
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000959 case 'C':
Ben Lindstrom681d9322002-03-22 03:53:00 +0000960 process_cmdline();
Ben Lindstrom5589f4b2002-03-22 03:24:32 +0000961 continue;
962
Damien Millerad833b32000-08-23 10:46:23 +1000963 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
Ben Lindstrombba81212001-06-25 05:01:22 +0000994static void
Damien Miller90fdfaf2006-03-26 14:25:37 +1100995client_process_input(fd_set *readset)
Damien Miller1383bd82000-04-06 12:32:37 +1000996{
Damien Miller166fca82000-04-20 07:42:21 +1000997 int len;
Damien Millerad833b32000-08-23 10:46:23 +1000998 char buf[8192];
Damien Miller1383bd82000-04-06 12:32:37 +1000999
Damien Miller95def091999-11-25 00:26:21 +11001000 /* 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));
Ben Lindstrom4c8cff12001-04-17 18:09:42 +00001004 if (len < 0 && (errno == EAGAIN || errno == EINTR))
1005 return; /* we'll try again later */
Damien Miller95def091999-11-25 00:26:21 +11001006 if (len <= 0) {
Damien Miller5428f641999-11-25 11:54:57 +11001007 /*
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 */
Damien Miller95def091999-11-25 00:26:21 +11001012 if (len < 0) {
1013 snprintf(buf, sizeof buf, "read: %.100s\r\n", strerror(errno));
1014 buffer_append(&stderr_buffer, buf, strlen(buf));
Damien Miller95def091999-11-25 00:26:21 +11001015 }
1016 /* Mark that we have seen EOF. */
1017 stdin_eof = 1;
Damien Miller5428f641999-11-25 11:54:57 +11001018 /*
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 */
Damien Miller95def091999-11-25 00:26:21 +11001025 if (buffer_len(&stdin_buffer) == 0) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001026 packet_start(SSH_CMSG_EOF);
1027 packet_send();
Damien Miller95def091999-11-25 00:26:21 +11001028 }
Ben Lindstrom2b1f71b2001-06-05 20:32:21 +00001029 } else if (escape_char == SSH_ESCAPECHAR_NONE) {
Damien Miller5428f641999-11-25 11:54:57 +11001030 /*
1031 * Normal successful read, and no escape character.
1032 * Just append the data to buffer.
1033 */
Damien Miller95def091999-11-25 00:26:21 +11001034 buffer_append(&stdin_buffer, buf, len);
Damien Miller95def091999-11-25 00:26:21 +11001035 } else {
Damien Miller5428f641999-11-25 11:54:57 +11001036 /*
1037 * Normal, successful read. But we have an escape character
1038 * and have to process the characters one by one.
1039 */
Ben Lindstrome9613cf2001-03-05 06:14:02 +00001040 if (process_escapes(&stdin_buffer, &stdout_buffer,
1041 &stderr_buffer, buf, len) == -1)
Damien Millerad833b32000-08-23 10:46:23 +10001042 return;
Damien Miller95def091999-11-25 00:26:21 +11001043 }
1044 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001045}
1046
Ben Lindstrombba81212001-06-25 05:01:22 +00001047static void
Damien Miller90fdfaf2006-03-26 14:25:37 +11001048client_process_output(fd_set *writeset)
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001049{
Damien Miller95def091999-11-25 00:26:21 +11001050 int len;
1051 char buf[100];
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001052
Damien Miller95def091999-11-25 00:26:21 +11001053 /* 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),
Damien Miller037a0dc1999-12-07 15:38:31 +11001057 buffer_len(&stdout_buffer));
Damien Miller95def091999-11-25 00:26:21 +11001058 if (len <= 0) {
Ben Lindstrom4c8cff12001-04-17 18:09:42 +00001059 if (errno == EINTR || errno == EAGAIN)
Damien Miller95def091999-11-25 00:26:21 +11001060 len = 0;
1061 else {
Damien Miller5428f641999-11-25 11:54:57 +11001062 /*
1063 * An error or EOF was encountered. Put an
1064 * error message to stderr buffer.
1065 */
Damien Miller95def091999-11-25 00:26:21 +11001066 snprintf(buf, sizeof buf, "write stdout: %.50s\r\n", strerror(errno));
1067 buffer_append(&stderr_buffer, buf, strlen(buf));
Damien Miller95def091999-11-25 00:26:21 +11001068 quit_pending = 1;
1069 return;
1070 }
1071 }
1072 /* Consume printed data from the buffer. */
1073 buffer_consume(&stdout_buffer, len);
Ben Lindstroma3700052001-04-05 23:26:32 +00001074 stdout_bytes += len;
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001075 }
Damien Miller95def091999-11-25 00:26:21 +11001076 /* 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),
Damien Miller037a0dc1999-12-07 15:38:31 +11001080 buffer_len(&stderr_buffer));
Damien Miller95def091999-11-25 00:26:21 +11001081 if (len <= 0) {
Ben Lindstrom4c8cff12001-04-17 18:09:42 +00001082 if (errno == EINTR || errno == EAGAIN)
Damien Miller95def091999-11-25 00:26:21 +11001083 len = 0;
1084 else {
Damien Miller5428f641999-11-25 11:54:57 +11001085 /* EOF or error, but can't even print error message. */
Damien Miller95def091999-11-25 00:26:21 +11001086 quit_pending = 1;
1087 return;
1088 }
1089 }
1090 /* Consume printed characters from the buffer. */
1091 buffer_consume(&stderr_buffer, len);
Ben Lindstroma3700052001-04-05 23:26:32 +00001092 stderr_bytes += len;
Damien Miller95def091999-11-25 00:26:21 +11001093 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001094}
1095
Damien Miller5428f641999-11-25 11:54:57 +11001096/*
Damien Millerb38eff82000-04-01 11:09:21 +10001097 * 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
Ben Lindstrombba81212001-06-25 05:01:22 +00001108static void
Ben Lindstrom31ca54a2001-02-09 02:11:24 +00001109client_process_buffered_input_packets(void)
Damien Millerb38eff82000-04-01 11:09:21 +10001110{
Ben Lindstromf28f6342001-04-04 02:03:04 +00001111 dispatch_run(DISPATCH_NONBLOCK, &quit_pending, compat20 ? xxx_kex : NULL);
Damien Millerb38eff82000-04-01 11:09:21 +10001112}
1113
Damien Millerad833b32000-08-23 10:46:23 +10001114/* scan buf[] for '~' before sending data to the peer */
1115
Ben Lindstrombba81212001-06-25 05:01:22 +00001116static int
Damien Millerad833b32000-08-23 10:46:23 +10001117simple_escape_filter(Channel *c, char *buf, int len)
1118{
Damien Miller5771ed72008-05-19 15:35:33 +10001119 if (c->extended_usage != CHAN_EXTENDED_WRITE)
1120 return 0;
1121
Damien Millerad833b32000-08-23 10:46:23 +10001122 return process_escapes(&c->input, &c->output, &c->extended, buf, len);
1123}
1124
Ben Lindstrombba81212001-06-25 05:01:22 +00001125static void
Ben Lindstrom4c3f77d2001-04-05 23:37:36 +00001126client_channel_closed(int id, void *arg)
1127{
Damien Miller3ec27592001-10-12 11:35:04 +10001128 channel_cancel_cleanup(id);
Ben Lindstrom4c3f77d2001-04-05 23:37:36 +00001129 session_closed = 1;
Darren Tucker9a2c4cd2003-09-22 21:16:05 +10001130 leave_raw_mode();
Ben Lindstrom4c3f77d2001-04-05 23:37:36 +00001131}
1132
Damien Millerb38eff82000-04-01 11:09:21 +10001133/*
Damien Miller5428f641999-11-25 11:54:57 +11001134 * 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
Ben Lindstrom2b1f71b2001-06-05 20:32:21 +00001136 * remote host. If escape_char != SSH_ESCAPECHAR_NONE, it is the character
1137 * used as an escape character for terminating or suspending the session.
Damien Miller5428f641999-11-25 11:54:57 +11001138 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001139
Damien Miller4af51302000-04-16 11:18:38 +10001140int
Damien Millerad833b32000-08-23 10:46:23 +10001141client_loop(int have_pty, int escape_char_arg, int ssh2_chan_id)
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001142{
Damien Miller5e953212001-01-30 09:14:00 +11001143 fd_set *readset = NULL, *writeset = NULL;
Damien Miller95def091999-11-25 00:26:21 +11001144 double start_time, total_time;
Darren Tuckerc7a6fc42004-08-13 21:18:00 +10001145 int max_fd = 0, max_fd2 = 0, len, rekeying = 0;
1146 u_int nalloc = 0;
Damien Miller95def091999-11-25 00:26:21 +11001147 char buf[100];
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001148
Damien Miller95def091999-11-25 00:26:21 +11001149 debug("Entering interactive session.");
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001150
Damien Miller95def091999-11-25 00:26:21 +11001151 start_time = get_current_time();
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001152
Damien Miller95def091999-11-25 00:26:21 +11001153 /* 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();
Damien Miller5e953212001-01-30 09:14:00 +11001161 max_fd = MAX(connection_in, connection_out);
Damien Millerb1cbfa22008-05-19 16:00:08 +10001162 if (muxserver_sock != -1)
1163 max_fd = MAX(max_fd, muxserver_sock);
Damien Miller5e953212001-01-30 09:14:00 +11001164
1165 if (!compat20) {
Ben Lindstrom302ea6f2001-04-16 02:01:25 +00001166 /* 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));
Damien Miller5e953212001-01-30 09:14:00 +11001173 max_fd = MAX(max_fd, fileno(stdin));
1174 max_fd = MAX(max_fd, fileno(stdout));
1175 max_fd = MAX(max_fd, fileno(stderr));
1176 }
Damien Miller95def091999-11-25 00:26:21 +11001177 stdin_bytes = 0;
1178 stdout_bytes = 0;
1179 stderr_bytes = 0;
1180 quit_pending = 0;
1181 escape_char = escape_char_arg;
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001182
Damien Miller95def091999-11-25 00:26:21 +11001183 /* Initialize buffers. */
1184 buffer_init(&stdin_buffer);
1185 buffer_init(&stdout_buffer);
1186 buffer_init(&stderr_buffer);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001187
Damien Millerb38eff82000-04-01 11:09:21 +10001188 client_init_dispatch();
1189
Ben Lindstromf49dbff2002-12-23 02:01:55 +00001190 /*
1191 * Set signal handlers, (e.g. to restore non-blocking mode)
1192 * but don't overwrite SIG_IGN, matches behaviour from rsh(1)
1193 */
Darren Tucker07336da2004-11-05 20:02:16 +11001194 if (signal(SIGHUP, SIG_IGN) != SIG_IGN)
1195 signal(SIGHUP, signal_handler);
Ben Lindstromf49dbff2002-12-23 02:01:55 +00001196 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);
Darren Tucker5d78de62004-11-05 20:35:44 +11001202 signal(SIGWINCH, window_change_handler);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001203
Damien Miller95def091999-11-25 00:26:21 +11001204 if (have_pty)
1205 enter_raw_mode();
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001206
Ben Lindstrom5b828322001-02-09 01:34:36 +00001207 if (compat20) {
1208 session_ident = ssh2_chan_id;
Ben Lindstrom2b1f71b2001-06-05 20:32:21 +00001209 if (escape_char != SSH_ESCAPECHAR_NONE)
Ben Lindstrom5b828322001-02-09 01:34:36 +00001210 channel_register_filter(session_ident,
Damien Miller077b2382005-12-31 16:22:32 +11001211 simple_escape_filter, NULL);
Ben Lindstrom4c3f77d2001-04-05 23:37:36 +00001212 if (session_ident != -1)
1213 channel_register_cleanup(session_ident,
Damien Miller39eda6e2005-11-05 14:52:50 +11001214 client_channel_closed, 0);
Ben Lindstrom5b828322001-02-09 01:34:36 +00001215 } else {
1216 /* Check if we should immediately send eof on stdin. */
Damien Miller1383bd82000-04-06 12:32:37 +10001217 client_check_initial_eof_on_stdin();
Ben Lindstrom5b828322001-02-09 01:34:36 +00001218 }
Damien Millerad833b32000-08-23 10:46:23 +10001219
Damien Miller95def091999-11-25 00:26:21 +11001220 /* Main loop of the client for the interactive session mode. */
1221 while (!quit_pending) {
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001222
Damien Miller5428f641999-11-25 11:54:57 +11001223 /* Process buffered packets sent by the server. */
Damien Miller95def091999-11-25 00:26:21 +11001224 client_process_buffered_input_packets();
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001225
Ben Lindstrom4c3f77d2001-04-05 23:37:36 +00001226 if (compat20 && session_closed && !channel_still_open())
Damien Miller1383bd82000-04-06 12:32:37 +10001227 break;
Ben Lindstrom4c3f77d2001-04-05 23:37:36 +00001228
1229 rekeying = (xxx_kex != NULL && !xxx_kex->done);
Damien Miller1383bd82000-04-06 12:32:37 +10001230
Ben Lindstrombe2cc432001-04-04 23:46:07 +00001231 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();
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001240
Ben Lindstrombe2cc432001-04-04 23:46:07 +00001241 /*
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();
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001247
Ben Lindstrombe2cc432001-04-04 23:46:07 +00001248 /*
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();
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001253
Ben Lindstrombe2cc432001-04-04 23:46:07 +00001254 if (quit_pending)
1255 break;
1256 }
Damien Miller5428f641999-11-25 11:54:57 +11001257 /*
1258 * Wait until we have something to do (something becomes
1259 * available on one of the descriptors).
1260 */
Ben Lindstrom16d29d52001-07-18 16:01:46 +00001261 max_fd2 = max_fd;
Ben Lindstrombe2cc432001-04-04 23:46:07 +00001262 client_wait_until_can_do_something(&readset, &writeset,
Ben Lindstrom16d29d52001-07-18 16:01:46 +00001263 &max_fd2, &nalloc, rekeying);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001264
Damien Miller95def091999-11-25 00:26:21 +11001265 if (quit_pending)
1266 break;
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001267
Ben Lindstrombe2cc432001-04-04 23:46:07 +00001268 /* Do channel operations unless rekeying in progress. */
1269 if (!rekeying) {
1270 channel_after_select(readset, writeset);
Damien Millera5539d22003-04-09 20:50:06 +10001271 if (need_rekeying || packet_need_rekeying()) {
1272 debug("need rekeying");
Ben Lindstrombe2cc432001-04-04 23:46:07 +00001273 xxx_kex->done = 0;
1274 kex_send_kexinit(xxx_kex);
1275 need_rekeying = 0;
1276 }
1277 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001278
Damien Miller1383bd82000-04-06 12:32:37 +10001279 /* Buffer input from the connection. */
Damien Miller5e953212001-01-30 09:14:00 +11001280 client_process_net_input(readset);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001281
Damien Miller0e220db2004-06-15 10:34:08 +10001282 /* Accept control connections. */
Damien Millerb1cbfa22008-05-19 16:00:08 +10001283 if (muxserver_sock != -1 &&FD_ISSET(muxserver_sock, readset)) {
1284 if (muxserver_accept_control())
1285 quit_pending = 1;
1286 }
Damien Miller0e220db2004-06-15 10:34:08 +10001287
Damien Miller1383bd82000-04-06 12:32:37 +10001288 if (quit_pending)
1289 break;
1290
1291 if (!compat20) {
1292 /* Buffer data from stdin */
Damien Miller5e953212001-01-30 09:14:00 +11001293 client_process_input(readset);
Damien Miller1383bd82000-04-06 12:32:37 +10001294 /*
1295 * Process output to stdout and stderr. Output to
1296 * the connection is processed elsewhere (above).
1297 */
Damien Miller5e953212001-01-30 09:14:00 +11001298 client_process_output(writeset);
Damien Miller1383bd82000-04-06 12:32:37 +10001299 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001300
Damien Miller5428f641999-11-25 11:54:57 +11001301 /* Send as much buffered packet data as possible to the sender. */
Damien Miller5e953212001-01-30 09:14:00 +11001302 if (FD_ISSET(connection_out, writeset))
Damien Miller95def091999-11-25 00:26:21 +11001303 packet_write_poll();
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001304 }
Damien Miller5e953212001-01-30 09:14:00 +11001305 if (readset)
1306 xfree(readset);
1307 if (writeset)
1308 xfree(writeset);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001309
Damien Miller95def091999-11-25 00:26:21 +11001310 /* Terminate the session. */
1311
1312 /* Stop watching for window change. */
Darren Tucker5d78de62004-11-05 20:35:44 +11001313 signal(SIGWINCH, SIG_DFL);
Damien Miller95def091999-11-25 00:26:21 +11001314
Ben Lindstrom601e4362001-06-21 03:19:23 +00001315 channel_free_all();
Damien Miller95def091999-11-25 00:26:21 +11001316
Ben Lindstromec46e0b2001-06-09 01:27:31 +00001317 if (have_pty)
1318 leave_raw_mode();
Ben Lindstromc93e84c2001-05-12 00:08:37 +00001319
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
Damien Millerd6965512003-12-17 16:31:53 +11001328 /*
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
Darren Tucker9a2c4cd2003-09-22 21:16:05 +10001338 if (received_signal)
Ben Lindstromf8f065b2001-12-06 17:52:16 +00001339 fatal("Killed by signal %d.", (int) received_signal);
Ben Lindstromec46e0b2001-06-09 01:27:31 +00001340
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
Damien Miller95def091999-11-25 00:26:21 +11001350 /* Output any buffered data for stdout. */
Ben Lindstromc93e84c2001-05-12 00:08:37 +00001351 while (buffer_len(&stdout_buffer) > 0) {
1352 len = write(fileno(stdout), buffer_ptr(&stdout_buffer),
Damien Miller037a0dc1999-12-07 15:38:31 +11001353 buffer_len(&stdout_buffer));
Ben Lindstromc93e84c2001-05-12 00:08:37 +00001354 if (len <= 0) {
Damien Miller95def091999-11-25 00:26:21 +11001355 error("Write failed flushing stdout buffer.");
Ben Lindstromc93e84c2001-05-12 00:08:37 +00001356 break;
1357 }
Damien Miller95def091999-11-25 00:26:21 +11001358 buffer_consume(&stdout_buffer, len);
Ben Lindstroma3700052001-04-05 23:26:32 +00001359 stdout_bytes += len;
Damien Miller95def091999-11-25 00:26:21 +11001360 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001361
Damien Miller95def091999-11-25 00:26:21 +11001362 /* Output any buffered data for stderr. */
Ben Lindstromc93e84c2001-05-12 00:08:37 +00001363 while (buffer_len(&stderr_buffer) > 0) {
1364 len = write(fileno(stderr), buffer_ptr(&stderr_buffer),
Damien Miller037a0dc1999-12-07 15:38:31 +11001365 buffer_len(&stderr_buffer));
Ben Lindstromc93e84c2001-05-12 00:08:37 +00001366 if (len <= 0) {
Damien Miller95def091999-11-25 00:26:21 +11001367 error("Write failed flushing stderr buffer.");
Ben Lindstromc93e84c2001-05-12 00:08:37 +00001368 break;
1369 }
Damien Miller95def091999-11-25 00:26:21 +11001370 buffer_consume(&stderr_buffer, len);
Ben Lindstroma3700052001-04-05 23:26:32 +00001371 stderr_bytes += len;
Damien Miller95def091999-11-25 00:26:21 +11001372 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001373
Damien Miller95def091999-11-25 00:26:21 +11001374 /* 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);
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001379
Damien Miller95def091999-11-25 00:26:21 +11001380 /* 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",
Damien Miller9f0f5c62001-12-21 14:45:46 +11001383 stdin_bytes, stdout_bytes, stderr_bytes, total_time);
Damien Miller95def091999-11-25 00:26:21 +11001384 if (total_time > 0)
1385 debug("Bytes per second: stdin %.1f, stdout %.1f, stderr %.1f",
Damien Miller9f0f5c62001-12-21 14:45:46 +11001386 stdin_bytes / total_time, stdout_bytes / total_time,
1387 stderr_bytes / total_time);
Damien Miller95def091999-11-25 00:26:21 +11001388
1389 /* Return the exit status of the program. */
1390 debug("Exit status %d", exit_status);
1391 return exit_status;
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001392}
Damien Millerb38eff82000-04-01 11:09:21 +10001393
1394/*********/
1395
Ben Lindstrombba81212001-06-25 05:01:22 +00001396static void
Damien Miller630d6f42002-01-22 23:17:30 +11001397client_input_stdout_data(int type, u_int32_t seq, void *ctxt)
Damien Millerb38eff82000-04-01 11:09:21 +10001398{
Ben Lindstrom46c16222000-12-22 01:43:59 +00001399 u_int data_len;
Damien Millerb38eff82000-04-01 11:09:21 +10001400 char *data = packet_get_string(&data_len);
Damien Miller48b03fc2002-01-22 23:11:40 +11001401 packet_check_eom();
Damien Millerb38eff82000-04-01 11:09:21 +10001402 buffer_append(&stdout_buffer, data, data_len);
Damien Millerb38eff82000-04-01 11:09:21 +10001403 memset(data, 0, data_len);
1404 xfree(data);
1405}
Ben Lindstrombba81212001-06-25 05:01:22 +00001406static void
Damien Miller630d6f42002-01-22 23:17:30 +11001407client_input_stderr_data(int type, u_int32_t seq, void *ctxt)
Damien Millerb38eff82000-04-01 11:09:21 +10001408{
Ben Lindstrom46c16222000-12-22 01:43:59 +00001409 u_int data_len;
Damien Millerb38eff82000-04-01 11:09:21 +10001410 char *data = packet_get_string(&data_len);
Damien Miller48b03fc2002-01-22 23:11:40 +11001411 packet_check_eom();
Damien Millerb38eff82000-04-01 11:09:21 +10001412 buffer_append(&stderr_buffer, data, data_len);
Damien Millerb38eff82000-04-01 11:09:21 +10001413 memset(data, 0, data_len);
1414 xfree(data);
1415}
Ben Lindstrombba81212001-06-25 05:01:22 +00001416static void
Damien Miller630d6f42002-01-22 23:17:30 +11001417client_input_exit_status(int type, u_int32_t seq, void *ctxt)
Damien Millerb38eff82000-04-01 11:09:21 +10001418{
Damien Millerb38eff82000-04-01 11:09:21 +10001419 exit_status = packet_get_int();
Damien Miller48b03fc2002-01-22 23:11:40 +11001420 packet_check_eom();
Damien Millerb38eff82000-04-01 11:09:21 +10001421 /* 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}
Darren Tucker5dcdd212003-10-02 16:17:00 +10001432static void
1433client_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}
Damien Millerb38eff82000-04-01 11:09:21 +10001472
Ben Lindstrombba81212001-06-25 05:01:22 +00001473static Channel *
Damien Miller0bc1bd82000-11-13 22:57:25 +11001474client_request_forwarded_tcpip(const char *request_type, int rchan)
1475{
Ben Lindstroma962c2f2002-07-04 00:14:17 +00001476 Channel *c = NULL;
Damien Miller0bc1bd82000-11-13 22:57:25 +11001477 char *listen_address, *originator_address;
1478 int listen_port, originator_port;
Damien Miller0bc1bd82000-11-13 22:57:25 +11001479
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();
Damien Miller48b03fc2002-01-22 23:11:40 +11001485 packet_check_eom();
Damien Miller0bc1bd82000-11-13 22:57:25 +11001486
Damien Millerbd740252008-05-19 15:37:09 +10001487 debug("client_request_forwarded_tcpip: listen %s port %d, "
1488 "originator %s port %d", listen_address, listen_port,
1489 originator_address, originator_port);
Damien Miller0bc1bd82000-11-13 22:57:25 +11001490
Damien Millerbd740252008-05-19 15:37:09 +10001491 c = channel_connect_by_listen_address(listen_port,
1492 "forwarded-tcpip", originator_address);
1493
Damien Miller0bc1bd82000-11-13 22:57:25 +11001494 xfree(originator_address);
1495 xfree(listen_address);
1496 return c;
1497}
1498
Ben Lindstroma962c2f2002-07-04 00:14:17 +00001499static Channel *
Damien Miller0bc1bd82000-11-13 22:57:25 +11001500client_request_x11(const char *request_type, int rchan)
1501{
1502 Channel *c = NULL;
1503 char *originator;
1504 int originator_port;
Ben Lindstrom99c73b32001-05-05 04:09:47 +00001505 int sock;
Damien Miller0bc1bd82000-11-13 22:57:25 +11001506
1507 if (!options.forward_x11) {
1508 error("Warning: ssh server tried X11 forwarding.");
Damien Miller5eb137c2005-12-31 16:19:53 +11001509 error("Warning: this is probably a break-in attempt by a malicious server.");
Damien Miller0bc1bd82000-11-13 22:57:25 +11001510 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 }
Damien Miller48b03fc2002-01-22 23:11:40 +11001519 packet_check_eom();
Damien Miller0bc1bd82000-11-13 22:57:25 +11001520 /* XXX check permission */
Damien Millerd83ff352001-01-30 09:19:34 +11001521 debug("client_request_x11: request from %s %d", originator,
1522 originator_port);
Damien Miller0bc1bd82000-11-13 22:57:25 +11001523 xfree(originator);
Ben Lindstrom99c73b32001-05-05 04:09:47 +00001524 sock = x11_connect_display();
1525 if (sock < 0)
1526 return NULL;
1527 c = channel_new("x11",
1528 SSH_CHANNEL_X11_OPEN, sock, sock, -1,
Damien Millerb1ca8bb2003-05-14 13:45:42 +10001529 CHAN_TCP_WINDOW_DEFAULT, CHAN_X11_PACKET_DEFAULT, 0, "x11", 1);
Ben Lindstrom944c4f02001-09-18 05:51:13 +00001530 c->force_drain = 1;
Damien Miller0bc1bd82000-11-13 22:57:25 +11001531 return c;
1532}
1533
Ben Lindstroma962c2f2002-07-04 00:14:17 +00001534static Channel *
Damien Miller0bc1bd82000-11-13 22:57:25 +11001535client_request_agent(const char *request_type, int rchan)
1536{
1537 Channel *c = NULL;
Ben Lindstrom99c73b32001-05-05 04:09:47 +00001538 int sock;
Damien Miller0bc1bd82000-11-13 22:57:25 +11001539
1540 if (!options.forward_agent) {
1541 error("Warning: ssh server tried agent forwarding.");
Damien Miller5eb137c2005-12-31 16:19:53 +11001542 error("Warning: this is probably a break-in attempt by a malicious server.");
Damien Miller0bc1bd82000-11-13 22:57:25 +11001543 return NULL;
1544 }
Darren Tucker82a3d2b2007-02-19 22:10:25 +11001545 sock = ssh_get_authentication_socket();
Ben Lindstrom99c73b32001-05-05 04:09:47 +00001546 if (sock < 0)
1547 return NULL;
1548 c = channel_new("authentication agent connection",
1549 SSH_CHANNEL_OPEN, sock, sock, -1,
Darren Tucker5baa1702007-12-29 09:37:10 +11001550 CHAN_X11_WINDOW_DEFAULT, CHAN_TCP_PACKET_DEFAULT, 0,
Damien Millerb1ca8bb2003-05-14 13:45:42 +10001551 "authentication agent connection", 1);
Ben Lindstrom944c4f02001-09-18 05:51:13 +00001552 c->force_drain = 1;
Damien Miller0bc1bd82000-11-13 22:57:25 +11001553 return c;
1554}
1555
Damien Millerb3ce9fe2007-08-08 14:32:41 +10001556int
1557client_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
Damien Millerbd483e72000-04-30 10:00:53 +10001600/* XXXX move to generic input handler */
Ben Lindstrombba81212001-06-25 05:01:22 +00001601static void
Damien Miller630d6f42002-01-22 23:17:30 +11001602client_input_channel_open(int type, u_int32_t seq, void *ctxt)
Damien Millerbd483e72000-04-30 10:00:53 +10001603{
1604 Channel *c = NULL;
1605 char *ctype;
Damien Millerbd483e72000-04-30 10:00:53 +10001606 int rchan;
Ben Lindstrom4fed2be2002-06-25 23:17:36 +00001607 u_int rmaxpack, rwindow, len;
Damien Millerbd483e72000-04-30 10:00:53 +10001608
1609 ctype = packet_get_string(&len);
1610 rchan = packet_get_int();
1611 rwindow = packet_get_int();
1612 rmaxpack = packet_get_int();
1613
Damien Millere247cc42000-05-07 12:03:14 +10001614 debug("client_input_channel_open: ctype %s rchan %d win %d max %d",
Damien Millerbd483e72000-04-30 10:00:53 +10001615 ctype, rchan, rwindow, rmaxpack);
1616
Damien Miller0bc1bd82000-11-13 22:57:25 +11001617 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);
Damien Millerbd483e72000-04-30 10:00:53 +10001623 }
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;
Ben Lindstroma69d89b2001-05-09 00:01:18 +00001630 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 }
Damien Millerbd483e72000-04-30 10:00:53 +10001638 } 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);
Ben Lindstromf3436742001-04-29 19:52:00 +00001643 if (!(datafellows & SSH_BUG_OPENFAILURE)) {
Ben Lindstroma69d89b2001-05-09 00:01:18 +00001644 packet_put_cstring("open failed");
Ben Lindstromf3436742001-04-29 19:52:00 +00001645 packet_put_cstring("");
1646 }
Damien Millerbd483e72000-04-30 10:00:53 +10001647 packet_send();
1648 }
1649 xfree(ctype);
1650}
Ben Lindstrombba81212001-06-25 05:01:22 +00001651static void
Damien Miller630d6f42002-01-22 23:17:30 +11001652client_input_channel_req(int type, u_int32_t seq, void *ctxt)
Ben Lindstrom5b828322001-02-09 01:34:36 +00001653{
1654 Channel *c = NULL;
Damien Miller0e220db2004-06-15 10:34:08 +10001655 int exitval, id, reply, success = 0;
Ben Lindstrom5b828322001-02-09 01:34:36 +00001656 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
Damien Miller3bbd8782004-06-18 22:23:22 +10001665 if (id == -1) {
1666 error("client_input_channel_req: request for channel -1");
1667 } else if ((c = channel_lookup(id)) == NULL) {
Ben Lindstrom5b828322001-02-09 01:34:36 +00001668 error("client_input_channel_req: channel %d: unknown channel", id);
Damien Millerbab9bd42008-05-19 16:06:47 +10001669 } else if (strcmp(rtype, "eow@openssh.com") == 0) {
1670 packet_check_eom();
1671 chan_rcvd_eow(c);
Ben Lindstrom5b828322001-02-09 01:34:36 +00001672 } else if (strcmp(rtype, "exit-status") == 0) {
Damien Miller0e220db2004-06-15 10:34:08 +10001673 exitval = packet_get_int();
1674 if (id == session_ident) {
1675 success = 1;
1676 exit_status = exitval;
1677 } else if (c->ctl_fd == -1) {
1678 error("client_input_channel_req: unexpected channel %d",
1679 session_ident);
1680 } else {
1681 atomicio(vwrite, c->ctl_fd, &exitval, sizeof(exitval));
1682 success = 1;
1683 }
Damien Miller48b03fc2002-01-22 23:11:40 +11001684 packet_check_eom();
Ben Lindstrom5b828322001-02-09 01:34:36 +00001685 }
1686 if (reply) {
1687 packet_start(success ?
1688 SSH2_MSG_CHANNEL_SUCCESS : SSH2_MSG_CHANNEL_FAILURE);
Damien Miller3bbd8782004-06-18 22:23:22 +10001689 packet_put_int(id);
Ben Lindstrom5b828322001-02-09 01:34:36 +00001690 packet_send();
1691 }
1692 xfree(rtype);
1693}
Damien Millerc3fa4072002-01-22 23:21:58 +11001694static void
1695client_input_global_request(int type, u_int32_t seq, void *ctxt)
1696{
1697 char *rtype;
1698 int want_reply;
1699 int success = 0;
1700
1701 rtype = packet_get_string(NULL);
1702 want_reply = packet_get_char();
Damien Miller509b0102003-12-17 16:33:10 +11001703 debug("client_input_global_request: rtype %s want_reply %d",
1704 rtype, want_reply);
Damien Millerc3fa4072002-01-22 23:21:58 +11001705 if (want_reply) {
1706 packet_start(success ?
1707 SSH2_MSG_REQUEST_SUCCESS : SSH2_MSG_REQUEST_FAILURE);
1708 packet_send();
1709 packet_write_wait();
1710 }
1711 xfree(rtype);
1712}
Damien Millerbd483e72000-04-30 10:00:53 +10001713
Damien Miller0e220db2004-06-15 10:34:08 +10001714void
Darren Tuckerfc959702004-07-17 16:12:08 +10001715client_session2_setup(int id, int want_tty, int want_subsystem,
Damien Miller5771ed72008-05-19 15:35:33 +10001716 const char *term, struct termios *tiop, int in_fd, Buffer *cmd, char **env)
Damien Miller0e220db2004-06-15 10:34:08 +10001717{
1718 int len;
Darren Tucker5d78de62004-11-05 20:35:44 +11001719 Channel *c = NULL;
Damien Miller0e220db2004-06-15 10:34:08 +10001720
1721 debug2("%s: id %d", __func__, id);
1722
Darren Tucker5d78de62004-11-05 20:35:44 +11001723 if ((c = channel_lookup(id)) == NULL)
1724 fatal("client_session2_setup: channel %d: unknown channel", id);
1725
Damien Miller0e220db2004-06-15 10:34:08 +10001726 if (want_tty) {
1727 struct winsize ws;
1728 struct termios tio;
1729
1730 /* Store window size in the packet. */
1731 if (ioctl(in_fd, TIOCGWINSZ, &ws) < 0)
1732 memset(&ws, 0, sizeof(ws));
1733
Damien Miller5771ed72008-05-19 15:35:33 +10001734 channel_request_start(id, "pty-req", 1);
1735 client_expect_confirm(id, "PTY allocation", 0);
Damien Miller0e220db2004-06-15 10:34:08 +10001736 packet_put_cstring(term != NULL ? term : "");
Damien Miller71a73672006-03-26 14:04:36 +11001737 packet_put_int((u_int)ws.ws_col);
1738 packet_put_int((u_int)ws.ws_row);
1739 packet_put_int((u_int)ws.ws_xpixel);
1740 packet_put_int((u_int)ws.ws_ypixel);
Damien Miller0e220db2004-06-15 10:34:08 +10001741 tio = get_saved_tio();
1742 tty_make_modes(-1, tiop != NULL ? tiop : &tio);
1743 packet_send();
1744 /* XXX wait for reply */
Darren Tucker5d78de62004-11-05 20:35:44 +11001745 c->client_tty = 1;
Damien Miller0e220db2004-06-15 10:34:08 +10001746 }
1747
1748 /* Transfer any environment variables from client to server */
Damien Miller3756dce2004-06-18 01:17:29 +10001749 if (options.num_send_env != 0 && env != NULL) {
Damien Miller0e220db2004-06-15 10:34:08 +10001750 int i, j, matched;
Damien Miller0e220db2004-06-15 10:34:08 +10001751 char *name, *val;
1752
1753 debug("Sending environment.");
Damien Miller3756dce2004-06-18 01:17:29 +10001754 for (i = 0; env[i] != NULL; i++) {
Damien Miller0e220db2004-06-15 10:34:08 +10001755 /* Split */
Damien Miller3756dce2004-06-18 01:17:29 +10001756 name = xstrdup(env[i]);
Damien Miller0e220db2004-06-15 10:34:08 +10001757 if ((val = strchr(name, '=')) == NULL) {
Damien Miller0a0176e2005-11-05 15:07:59 +11001758 xfree(name);
Damien Miller0e220db2004-06-15 10:34:08 +10001759 continue;
1760 }
1761 *val++ = '\0';
1762
1763 matched = 0;
1764 for (j = 0; j < options.num_send_env; j++) {
1765 if (match_pattern(name, options.send_env[j])) {
1766 matched = 1;
1767 break;
1768 }
1769 }
1770 if (!matched) {
1771 debug3("Ignored env %s", name);
Damien Miller0a0176e2005-11-05 15:07:59 +11001772 xfree(name);
Damien Miller0e220db2004-06-15 10:34:08 +10001773 continue;
1774 }
1775
1776 debug("Sending env %s = %s", name, val);
1777 channel_request_start(id, "env", 0);
1778 packet_put_cstring(name);
1779 packet_put_cstring(val);
1780 packet_send();
Damien Miller0a0176e2005-11-05 15:07:59 +11001781 xfree(name);
Damien Miller0e220db2004-06-15 10:34:08 +10001782 }
1783 }
1784
1785 len = buffer_len(cmd);
1786 if (len > 0) {
1787 if (len > 900)
1788 len = 900;
1789 if (want_subsystem) {
Damien Miller5771ed72008-05-19 15:35:33 +10001790 debug("Sending subsystem: %.*s",
1791 len, (u_char*)buffer_ptr(cmd));
1792 channel_request_start(id, "subsystem", 1);
1793 client_expect_confirm(id, "subsystem", 1);
Damien Miller0e220db2004-06-15 10:34:08 +10001794 } else {
Damien Miller5771ed72008-05-19 15:35:33 +10001795 debug("Sending command: %.*s",
1796 len, (u_char*)buffer_ptr(cmd));
1797 channel_request_start(id, "exec", 1);
1798 client_expect_confirm(id, "exec", 1);
Damien Miller0e220db2004-06-15 10:34:08 +10001799 }
1800 packet_put_string(buffer_ptr(cmd), buffer_len(cmd));
1801 packet_send();
1802 } else {
Damien Miller5771ed72008-05-19 15:35:33 +10001803 channel_request_start(id, "shell", 1);
1804 client_expect_confirm(id, "shell", 1);
Damien Miller0e220db2004-06-15 10:34:08 +10001805 packet_send();
1806 }
1807}
1808
Ben Lindstrombba81212001-06-25 05:01:22 +00001809static void
Ben Lindstrom31ca54a2001-02-09 02:11:24 +00001810client_init_dispatch_20(void)
Damien Miller1383bd82000-04-06 12:32:37 +10001811{
Ben Lindstrom8ac91062001-04-04 17:57:54 +00001812 dispatch_init(&dispatch_protocol_error);
Damien Miller2797f7f2002-04-23 21:09:44 +10001813
Damien Miller1383bd82000-04-06 12:32:37 +10001814 dispatch_set(SSH2_MSG_CHANNEL_CLOSE, &channel_input_oclose);
1815 dispatch_set(SSH2_MSG_CHANNEL_DATA, &channel_input_data);
1816 dispatch_set(SSH2_MSG_CHANNEL_EOF, &channel_input_ieof);
1817 dispatch_set(SSH2_MSG_CHANNEL_EXTENDED_DATA, &channel_input_extended_data);
Damien Millerbd483e72000-04-30 10:00:53 +10001818 dispatch_set(SSH2_MSG_CHANNEL_OPEN, &client_input_channel_open);
Damien Miller1383bd82000-04-06 12:32:37 +10001819 dispatch_set(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION, &channel_input_open_confirmation);
1820 dispatch_set(SSH2_MSG_CHANNEL_OPEN_FAILURE, &channel_input_open_failure);
Ben Lindstrom5b828322001-02-09 01:34:36 +00001821 dispatch_set(SSH2_MSG_CHANNEL_REQUEST, &client_input_channel_req);
Damien Miller1383bd82000-04-06 12:32:37 +10001822 dispatch_set(SSH2_MSG_CHANNEL_WINDOW_ADJUST, &channel_input_window_adjust);
Damien Millerb84886b2008-05-19 15:05:07 +10001823 dispatch_set(SSH2_MSG_CHANNEL_SUCCESS, &channel_input_status_confirm);
1824 dispatch_set(SSH2_MSG_CHANNEL_FAILURE, &channel_input_status_confirm);
Damien Millerc3fa4072002-01-22 23:21:58 +11001825 dispatch_set(SSH2_MSG_GLOBAL_REQUEST, &client_input_global_request);
Ben Lindstrom8ac91062001-04-04 17:57:54 +00001826
1827 /* rekeying */
1828 dispatch_set(SSH2_MSG_KEXINIT, &kex_input_kexinit);
Damien Miller2797f7f2002-04-23 21:09:44 +10001829
1830 /* global request reply messages */
1831 dispatch_set(SSH2_MSG_REQUEST_FAILURE, &client_global_request_reply);
1832 dispatch_set(SSH2_MSG_REQUEST_SUCCESS, &client_global_request_reply);
Damien Miller1383bd82000-04-06 12:32:37 +10001833}
Ben Lindstrombba81212001-06-25 05:01:22 +00001834static void
Ben Lindstrom31ca54a2001-02-09 02:11:24 +00001835client_init_dispatch_13(void)
Damien Millerb38eff82000-04-01 11:09:21 +10001836{
1837 dispatch_init(NULL);
1838 dispatch_set(SSH_MSG_CHANNEL_CLOSE, &channel_input_close);
1839 dispatch_set(SSH_MSG_CHANNEL_CLOSE_CONFIRMATION, &channel_input_close_confirmation);
1840 dispatch_set(SSH_MSG_CHANNEL_DATA, &channel_input_data);
Damien Millerb38eff82000-04-01 11:09:21 +10001841 dispatch_set(SSH_MSG_CHANNEL_OPEN_CONFIRMATION, &channel_input_open_confirmation);
1842 dispatch_set(SSH_MSG_CHANNEL_OPEN_FAILURE, &channel_input_open_failure);
1843 dispatch_set(SSH_MSG_PORT_OPEN, &channel_input_port_open);
Damien Millerb38eff82000-04-01 11:09:21 +10001844 dispatch_set(SSH_SMSG_EXITSTATUS, &client_input_exit_status);
1845 dispatch_set(SSH_SMSG_STDERR_DATA, &client_input_stderr_data);
1846 dispatch_set(SSH_SMSG_STDOUT_DATA, &client_input_stdout_data);
Damien Miller69b69aa2000-10-28 14:19:58 +11001847
1848 dispatch_set(SSH_SMSG_AGENT_OPEN, options.forward_agent ?
Darren Tucker5dcdd212003-10-02 16:17:00 +10001849 &client_input_agent_open : &deny_input_open);
Damien Miller69b69aa2000-10-28 14:19:58 +11001850 dispatch_set(SSH_SMSG_X11_OPEN, options.forward_x11 ?
1851 &x11_input_open : &deny_input_open);
Damien Millerb38eff82000-04-01 11:09:21 +10001852}
Ben Lindstrombba81212001-06-25 05:01:22 +00001853static void
Ben Lindstrom31ca54a2001-02-09 02:11:24 +00001854client_init_dispatch_15(void)
Damien Millerb38eff82000-04-01 11:09:21 +10001855{
1856 client_init_dispatch_13();
1857 dispatch_set(SSH_MSG_CHANNEL_CLOSE, &channel_input_ieof);
1858 dispatch_set(SSH_MSG_CHANNEL_CLOSE_CONFIRMATION, & channel_input_oclose);
1859}
Ben Lindstromdb47f382001-07-04 05:10:27 +00001860static void
Ben Lindstrom31ca54a2001-02-09 02:11:24 +00001861client_init_dispatch(void)
Damien Millerb38eff82000-04-01 11:09:21 +10001862{
Damien Miller1383bd82000-04-06 12:32:37 +10001863 if (compat20)
1864 client_init_dispatch_20();
1865 else if (compat13)
Damien Millerb38eff82000-04-01 11:09:21 +10001866 client_init_dispatch_13();
1867 else
1868 client_init_dispatch_15();
1869}
Darren Tucker9a2c4cd2003-09-22 21:16:05 +10001870
1871/* client specific fatal cleanup */
1872void
Darren Tucker3e33cec2003-10-02 16:12:36 +10001873cleanup_exit(int i)
Darren Tucker9a2c4cd2003-09-22 21:16:05 +10001874{
Darren Tucker9a2c4cd2003-09-22 21:16:05 +10001875 leave_raw_mode();
1876 leave_non_blocking();
Damien Millerb1cbfa22008-05-19 16:00:08 +10001877 if (options.control_path != NULL && muxserver_sock != -1)
Damien Miller0e220db2004-06-15 10:34:08 +10001878 unlink(options.control_path);
Darren Tucker3e33cec2003-10-02 16:12:36 +10001879 _exit(i);
Darren Tucker9a2c4cd2003-09-22 21:16:05 +10001880}