Merge from vendor branch LIBARCHIVE:
[dragonfly.git] / crypto / openssh-4 / ssh.c
1 /* $OpenBSD: ssh.c,v 1.301 2007/08/07 07:32:53 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  * Ssh client program.  This program can be used to log into a remote machine.
7  * The software supports strong authentication, encryption, and forwarding
8  * of X11, TCP/IP, and authentication connections.
9  *
10  * As far as I am concerned, the code I have written for this software
11  * can be used freely for any purpose.  Any derived versions of this
12  * software must be clearly marked as such, and if the derived work is
13  * incompatible with the protocol description in the RFC file, it must be
14  * called by a name other than "ssh" or "Secure Shell".
15  *
16  * Copyright (c) 1999 Niels Provos.  All rights reserved.
17  * Copyright (c) 2000, 2001, 2002, 2003 Markus Friedl.  All rights reserved.
18  *
19  * Modified to work with SSL by Niels Provos <provos@citi.umich.edu>
20  * in Canada (German citizen).
21  *
22  * Redistribution and use in source and binary forms, with or without
23  * modification, are permitted provided that the following conditions
24  * are met:
25  * 1. Redistributions of source code must retain the above copyright
26  *    notice, this list of conditions and the following disclaimer.
27  * 2. Redistributions in binary form must reproduce the above copyright
28  *    notice, this list of conditions and the following disclaimer in the
29  *    documentation and/or other materials provided with the distribution.
30  *
31  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
32  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
33  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
34  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
35  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
36  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
37  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
38  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
39  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
40  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
41  */
42
43 #include "includes.h"
44
45 #include <sys/types.h>
46 #ifdef HAVE_SYS_STAT_H
47 # include <sys/stat.h>
48 #endif
49 #include <sys/resource.h>
50 #include <sys/ioctl.h>
51 #include <sys/socket.h>
52 #include <sys/un.h>
53
54 #include <ctype.h>
55 #include <errno.h>
56 #include <fcntl.h>
57 #include <netdb.h>
58 #ifdef HAVE_PATHS_H
59 #include <paths.h>
60 #endif
61 #include <pwd.h>
62 #include <signal.h>
63 #include <stdarg.h>
64 #include <stddef.h>
65 #include <stdio.h>
66 #include <stdlib.h>
67 #include <string.h>
68 #include <unistd.h>
69
70 #include <netinet/in.h>
71 #include <arpa/inet.h>
72
73 #include <openssl/evp.h>
74 #include <openssl/err.h>
75
76 #include "xmalloc.h"
77 #include "ssh.h"
78 #include "ssh1.h"
79 #include "ssh2.h"
80 #include "compat.h"
81 #include "cipher.h"
82 #include "packet.h"
83 #include "buffer.h"
84 #include "channels.h"
85 #include "key.h"
86 #include "authfd.h"
87 #include "authfile.h"
88 #include "pathnames.h"
89 #include "dispatch.h"
90 #include "clientloop.h"
91 #include "log.h"
92 #include "readconf.h"
93 #include "sshconnect.h"
94 #include "misc.h"
95 #include "kex.h"
96 #include "mac.h"
97 #include "sshpty.h"
98 #include "match.h"
99 #include "msg.h"
100 #include "monitor_fdpass.h"
101 #include "uidswap.h"
102 #include "version.h"
103
104 #ifdef SMARTCARD
105 #include "scard.h"
106 #endif
107
108 extern char *__progname;
109
110 /* Flag indicating whether debug mode is on.  This can be set on the command line. */
111 int debug_flag = 0;
112
113 /* Flag indicating whether a tty should be allocated */
114 int tty_flag = 0;
115 int no_tty_flag = 0;
116 int force_tty_flag = 0;
117
118 /* don't exec a shell */
119 int no_shell_flag = 0;
120
121 /*
122  * Flag indicating that nothing should be read from stdin.  This can be set
123  * on the command line.
124  */
125 int stdin_null_flag = 0;
126
127 /*
128  * Flag indicating that ssh should fork after authentication.  This is useful
129  * so that the passphrase can be entered manually, and then ssh goes to the
130  * background.
131  */
132 int fork_after_authentication_flag = 0;
133
134 /*
135  * General data structure for command line options and options configurable
136  * in configuration files.  See readconf.h.
137  */
138 Options options;
139
140 /* optional user configfile */
141 char *config = NULL;
142
143 /*
144  * Name of the host we are connecting to.  This is the name given on the
145  * command line, or the HostName specified for the user-supplied name in a
146  * configuration file.
147  */
148 char *host;
149
150 /* socket address the host resolves to */
151 struct sockaddr_storage hostaddr;
152
153 /* Private host keys. */
154 Sensitive sensitive_data;
155
156 /* Original real UID. */
157 uid_t original_real_uid;
158 uid_t original_effective_uid;
159
160 /* command to be executed */
161 Buffer command;
162
163 /* Should we execute a command or invoke a subsystem? */
164 int subsystem_flag = 0;
165
166 /* # of replies received for global requests */
167 static int client_global_request_id = 0;
168
169 /* pid of proxycommand child process */
170 pid_t proxy_command_pid = 0;
171
172 /* fd to control socket */
173 int control_fd = -1;
174
175 /* Multiplexing control command */
176 static u_int mux_command = 0;
177
178 /* Only used in control client mode */
179 volatile sig_atomic_t control_client_terminate = 0;
180 u_int control_server_pid = 0;
181
182 /* Prints a help message to the user.  This function never returns. */
183
184 static void
185 usage(void)
186 {
187         fprintf(stderr,
188 "usage: ssh [-1246AaCfgKkMNnqsTtVvXxY] [-b bind_address] [-c cipher_spec]\n"
189 "           [-D [bind_address:]port] [-e escape_char] [-F configfile]\n"
190 "           [-i identity_file] [-L [bind_address:]port:host:hostport]\n"
191 "           [-l login_name] [-m mac_spec] [-O ctl_cmd] [-o option] [-p port]\n"
192 "           [-R [bind_address:]port:host:hostport] [-S ctl_path]\n"
193 "           [-w local_tun[:remote_tun]] [user@]hostname [command]\n"
194         );
195         exit(255);
196 }
197
198 static int ssh_session(void);
199 static int ssh_session2(void);
200 static void load_public_identity_files(void);
201 static void control_client(const char *path);
202
203 /*
204  * Main program for the ssh client.
205  */
206 int
207 main(int ac, char **av)
208 {
209         int i, opt, exit_status;
210         char *p, *cp, *line, buf[256];
211         struct stat st;
212         struct passwd *pw;
213         int dummy;
214         extern int optind, optreset;
215         extern char *optarg;
216         struct servent *sp;
217         Forward fwd;
218
219         /* Ensure that fds 0, 1 and 2 are open or directed to /dev/null */
220         sanitise_stdfd();
221
222         __progname = ssh_get_progname(av[0]);
223         init_rng();
224
225         /*
226          * Save the original real uid.  It will be needed later (uid-swapping
227          * may clobber the real uid).
228          */
229         original_real_uid = getuid();
230         original_effective_uid = geteuid();
231
232         /*
233          * Use uid-swapping to give up root privileges for the duration of
234          * option processing.  We will re-instantiate the rights when we are
235          * ready to create the privileged port, and will permanently drop
236          * them when the port has been created (actually, when the connection
237          * has been made, as we may need to create the port several times).
238          */
239         PRIV_END;
240
241 #ifdef HAVE_SETRLIMIT
242         /* If we are installed setuid root be careful to not drop core. */
243         if (original_real_uid != original_effective_uid) {
244                 struct rlimit rlim;
245                 rlim.rlim_cur = rlim.rlim_max = 0;
246                 if (setrlimit(RLIMIT_CORE, &rlim) < 0)
247                         fatal("setrlimit failed: %.100s", strerror(errno));
248         }
249 #endif
250         /* Get user data. */
251         pw = getpwuid(original_real_uid);
252         if (!pw) {
253                 logit("You don't exist, go away!");
254                 exit(255);
255         }
256         /* Take a copy of the returned structure. */
257         pw = pwcopy(pw);
258
259         /*
260          * Set our umask to something reasonable, as some files are created
261          * with the default umask.  This will make them world-readable but
262          * writable only by the owner, which is ok for all files for which we
263          * don't set the modes explicitly.
264          */
265         umask(022);
266
267         /* Initialize option structure to indicate that no values have been set. */
268         initialize_options(&options);
269
270         /* Parse command-line arguments. */
271         host = NULL;
272
273  again:
274         while ((opt = getopt(ac, av,
275             "1246ab:c:e:fgi:kl:m:no:p:qstvxACD:F:I:KL:MNO:PR:S:TVw:XY")) != -1) {
276                 switch (opt) {
277                 case '1':
278                         options.protocol = SSH_PROTO_1;
279                         break;
280                 case '2':
281                         options.protocol = SSH_PROTO_2;
282                         break;
283                 case '4':
284                         options.address_family = AF_INET;
285                         break;
286                 case '6':
287                         options.address_family = AF_INET6;
288                         break;
289                 case 'n':
290                         stdin_null_flag = 1;
291                         break;
292                 case 'f':
293                         fork_after_authentication_flag = 1;
294                         stdin_null_flag = 1;
295                         break;
296                 case 'x':
297                         options.forward_x11 = 0;
298                         break;
299                 case 'X':
300                         options.forward_x11 = 1;
301                         break;
302                 case 'Y':
303                         options.forward_x11 = 1;
304                         options.forward_x11_trusted = 1;
305                         break;
306                 case 'g':
307                         options.gateway_ports = 1;
308                         break;
309                 case 'O':
310                         if (strcmp(optarg, "check") == 0)
311                                 mux_command = SSHMUX_COMMAND_ALIVE_CHECK;
312                         else if (strcmp(optarg, "exit") == 0)
313                                 mux_command = SSHMUX_COMMAND_TERMINATE;
314                         else
315                                 fatal("Invalid multiplex command.");
316                         break;
317                 case 'P':       /* deprecated */
318                         options.use_privileged_port = 0;
319                         break;
320                 case 'a':
321                         options.forward_agent = 0;
322                         break;
323                 case 'A':
324                         options.forward_agent = 1;
325                         break;
326                 case 'k':
327                         options.gss_deleg_creds = 0;
328                         break;
329                 case 'K':
330                         options.gss_authentication = 1;
331                         options.gss_deleg_creds = 1;
332                         break;
333                 case 'i':
334                         if (stat(optarg, &st) < 0) {
335                                 fprintf(stderr, "Warning: Identity file %s "
336                                     "not accessible: %s.\n", optarg,
337                                     strerror(errno));
338                                 break;
339                         }
340                         if (options.num_identity_files >=
341                             SSH_MAX_IDENTITY_FILES)
342                                 fatal("Too many identity files specified "
343                                     "(max %d)", SSH_MAX_IDENTITY_FILES);
344                         options.identity_files[options.num_identity_files++] =
345                             xstrdup(optarg);
346                         break;
347                 case 'I':
348 #ifdef SMARTCARD
349                         options.smartcard_device = xstrdup(optarg);
350 #else
351                         fprintf(stderr, "no support for smartcards.\n");
352 #endif
353                         break;
354                 case 't':
355                         if (tty_flag)
356                                 force_tty_flag = 1;
357                         tty_flag = 1;
358                         break;
359                 case 'v':
360                         if (debug_flag == 0) {
361                                 debug_flag = 1;
362                                 options.log_level = SYSLOG_LEVEL_DEBUG1;
363                         } else {
364                                 if (options.log_level < SYSLOG_LEVEL_DEBUG3)
365                                         options.log_level++;
366                                 break;
367                         }
368                         /* FALLTHROUGH */
369                 case 'V':
370                         fprintf(stderr, "%s, %s\n",
371                             SSH_RELEASE, SSLeay_version(SSLEAY_VERSION));
372                         if (opt == 'V')
373                                 exit(0);
374                         break;
375                 case 'w':
376                         if (options.tun_open == -1)
377                                 options.tun_open = SSH_TUNMODE_DEFAULT;
378                         options.tun_local = a2tun(optarg, &options.tun_remote);
379                         if (options.tun_local == SSH_TUNID_ERR) {
380                                 fprintf(stderr, "Bad tun device '%s'\n", optarg);
381                                 exit(255);
382                         }
383                         break;
384                 case 'q':
385                         options.log_level = SYSLOG_LEVEL_QUIET;
386                         break;
387                 case 'e':
388                         if (optarg[0] == '^' && optarg[2] == 0 &&
389                             (u_char) optarg[1] >= 64 &&
390                             (u_char) optarg[1] < 128)
391                                 options.escape_char = (u_char) optarg[1] & 31;
392                         else if (strlen(optarg) == 1)
393                                 options.escape_char = (u_char) optarg[0];
394                         else if (strcmp(optarg, "none") == 0)
395                                 options.escape_char = SSH_ESCAPECHAR_NONE;
396                         else {
397                                 fprintf(stderr, "Bad escape character '%s'.\n",
398                                     optarg);
399                                 exit(255);
400                         }
401                         break;
402                 case 'c':
403                         if (ciphers_valid(optarg)) {
404                                 /* SSH2 only */
405                                 options.ciphers = xstrdup(optarg);
406                                 options.cipher = SSH_CIPHER_INVALID;
407                         } else {
408                                 /* SSH1 only */
409                                 options.cipher = cipher_number(optarg);
410                                 if (options.cipher == -1) {
411                                         fprintf(stderr,
412                                             "Unknown cipher type '%s'\n",
413                                             optarg);
414                                         exit(255);
415                                 }
416                                 if (options.cipher == SSH_CIPHER_3DES)
417                                         options.ciphers = "3des-cbc";
418                                 else if (options.cipher == SSH_CIPHER_BLOWFISH)
419                                         options.ciphers = "blowfish-cbc";
420                                 else
421                                         options.ciphers = (char *)-1;
422                         }
423                         break;
424                 case 'm':
425                         if (mac_valid(optarg))
426                                 options.macs = xstrdup(optarg);
427                         else {
428                                 fprintf(stderr, "Unknown mac type '%s'\n",
429                                     optarg);
430                                 exit(255);
431                         }
432                         break;
433                 case 'M':
434                         if (options.control_master == SSHCTL_MASTER_YES)
435                                 options.control_master = SSHCTL_MASTER_ASK;
436                         else
437                                 options.control_master = SSHCTL_MASTER_YES;
438                         break;
439                 case 'p':
440                         options.port = a2port(optarg);
441                         if (options.port == 0) {
442                                 fprintf(stderr, "Bad port '%s'\n", optarg);
443                                 exit(255);
444                         }
445                         break;
446                 case 'l':
447                         options.user = optarg;
448                         break;
449
450                 case 'L':
451                         if (parse_forward(&fwd, optarg))
452                                 add_local_forward(&options, &fwd);
453                         else {
454                                 fprintf(stderr,
455                                     "Bad local forwarding specification '%s'\n",
456                                     optarg);
457                                 exit(255);
458                         }
459                         break;
460
461                 case 'R':
462                         if (parse_forward(&fwd, optarg)) {
463                                 add_remote_forward(&options, &fwd);
464                         } else {
465                                 fprintf(stderr,
466                                     "Bad remote forwarding specification "
467                                     "'%s'\n", optarg);
468                                 exit(255);
469                         }
470                         break;
471
472                 case 'D':
473                         cp = p = xstrdup(optarg);
474                         memset(&fwd, '\0', sizeof(fwd));
475                         fwd.connect_host = "socks";
476                         if ((fwd.listen_host = hpdelim(&cp)) == NULL) {
477                                 fprintf(stderr, "Bad dynamic forwarding "
478                                     "specification '%.100s'\n", optarg);
479                                 exit(255);
480                         }
481                         if (cp != NULL) {
482                                 fwd.listen_port = a2port(cp);
483                                 fwd.listen_host = cleanhostname(fwd.listen_host);
484                         } else {
485                                 fwd.listen_port = a2port(fwd.listen_host);
486                                 fwd.listen_host = NULL;
487                         }
488
489                         if (fwd.listen_port == 0) {
490                                 fprintf(stderr, "Bad dynamic port '%s'\n",
491                                     optarg);
492                                 exit(255);
493                         }
494                         add_local_forward(&options, &fwd);
495                         xfree(p);
496                         break;
497
498                 case 'C':
499                         options.compression = 1;
500                         break;
501                 case 'N':
502                         no_shell_flag = 1;
503                         no_tty_flag = 1;
504                         break;
505                 case 'T':
506                         no_tty_flag = 1;
507                         break;
508                 case 'o':
509                         dummy = 1;
510                         line = xstrdup(optarg);
511                         if (process_config_line(&options, host ? host : "",
512                             line, "command-line", 0, &dummy) != 0)
513                                 exit(255);
514                         xfree(line);
515                         break;
516                 case 's':
517                         subsystem_flag = 1;
518                         break;
519                 case 'S':
520                         if (options.control_path != NULL)
521                                 free(options.control_path);
522                         options.control_path = xstrdup(optarg);
523                         break;
524                 case 'b':
525                         options.bind_address = optarg;
526                         break;
527                 case 'F':
528                         config = optarg;
529                         break;
530                 default:
531                         usage();
532                 }
533         }
534
535         ac -= optind;
536         av += optind;
537
538         if (ac > 0 && !host && **av != '-') {
539                 if (strrchr(*av, '@')) {
540                         p = xstrdup(*av);
541                         cp = strrchr(p, '@');
542                         if (cp == NULL || cp == p)
543                                 usage();
544                         options.user = p;
545                         *cp = '\0';
546                         host = ++cp;
547                 } else
548                         host = *av;
549                 if (ac > 1) {
550                         optind = optreset = 1;
551                         goto again;
552                 }
553                 ac--, av++;
554         }
555
556         /* Check that we got a host name. */
557         if (!host)
558                 usage();
559
560         SSLeay_add_all_algorithms();
561         ERR_load_crypto_strings();
562
563         /* Initialize the command to execute on remote host. */
564         buffer_init(&command);
565
566         /*
567          * Save the command to execute on the remote host in a buffer. There
568          * is no limit on the length of the command, except by the maximum
569          * packet size.  Also sets the tty flag if there is no command.
570          */
571         if (!ac) {
572                 /* No command specified - execute shell on a tty. */
573                 tty_flag = 1;
574                 if (subsystem_flag) {
575                         fprintf(stderr,
576                             "You must specify a subsystem to invoke.\n");
577                         usage();
578                 }
579         } else {
580                 /* A command has been specified.  Store it into the buffer. */
581                 for (i = 0; i < ac; i++) {
582                         if (i)
583                                 buffer_append(&command, " ", 1);
584                         buffer_append(&command, av[i], strlen(av[i]));
585                 }
586         }
587
588         /* Cannot fork to background if no command. */
589         if (fork_after_authentication_flag && buffer_len(&command) == 0 && !no_shell_flag)
590                 fatal("Cannot fork into background without a command to execute.");
591
592         /* Allocate a tty by default if no command specified. */
593         if (buffer_len(&command) == 0)
594                 tty_flag = 1;
595
596         /* Force no tty */
597         if (no_tty_flag)
598                 tty_flag = 0;
599         /* Do not allocate a tty if stdin is not a tty. */
600         if ((!isatty(fileno(stdin)) || stdin_null_flag) && !force_tty_flag) {
601                 if (tty_flag)
602                         logit("Pseudo-terminal will not be allocated because stdin is not a terminal.");
603                 tty_flag = 0;
604         }
605
606         /*
607          * Initialize "log" output.  Since we are the client all output
608          * actually goes to stderr.
609          */
610         log_init(av[0], options.log_level == -1 ? SYSLOG_LEVEL_INFO : options.log_level,
611             SYSLOG_FACILITY_USER, 1);
612
613         /*
614          * Read per-user configuration file.  Ignore the system wide config
615          * file if the user specifies a config file on the command line.
616          */
617         if (config != NULL) {
618                 if (!read_config_file(config, host, &options, 0))
619                         fatal("Can't open user config file %.100s: "
620                             "%.100s", config, strerror(errno));
621         } else {
622                 snprintf(buf, sizeof buf, "%.100s/%.100s", pw->pw_dir,
623                     _PATH_SSH_USER_CONFFILE);
624                 (void)read_config_file(buf, host, &options, 1);
625
626                 /* Read systemwide configuration file after use config. */
627                 (void)read_config_file(_PATH_HOST_CONFIG_FILE, host,
628                     &options, 0);
629         }
630
631         /* Fill configuration defaults. */
632         fill_default_options(&options);
633
634         channel_set_af(options.address_family);
635
636         /* reinit */
637         log_init(av[0], options.log_level, SYSLOG_FACILITY_USER, 1);
638
639         seed_rng();
640
641         if (options.user == NULL)
642                 options.user = xstrdup(pw->pw_name);
643
644         if (options.hostname != NULL)
645                 host = options.hostname;
646
647         /* force lowercase for hostkey matching */
648         if (options.host_key_alias != NULL) {
649                 for (p = options.host_key_alias; *p; p++)
650                         if (isupper(*p))
651                                 *p = (char)tolower(*p);
652         }
653
654         /* Get default port if port has not been set. */
655         if (options.port == 0) {
656                 sp = getservbyname(SSH_SERVICE_NAME, "tcp");
657                 options.port = sp ? ntohs(sp->s_port) : SSH_DEFAULT_PORT;
658         }
659
660         if (options.proxy_command != NULL &&
661             strcmp(options.proxy_command, "none") == 0)
662                 options.proxy_command = NULL;
663         if (options.control_path != NULL &&
664             strcmp(options.control_path, "none") == 0)
665                 options.control_path = NULL;
666
667         if (options.control_path != NULL) {
668                 char thishost[NI_MAXHOST];
669
670                 if (gethostname(thishost, sizeof(thishost)) == -1)
671                         fatal("gethostname: %s", strerror(errno));
672                 snprintf(buf, sizeof(buf), "%d", options.port);
673                 cp = tilde_expand_filename(options.control_path,
674                     original_real_uid);
675                 options.control_path = percent_expand(cp, "p", buf, "h", host,
676                     "r", options.user, "l", thishost, (char *)NULL);
677                 xfree(cp);
678         }
679         if (mux_command != 0 && options.control_path == NULL)
680                 fatal("No ControlPath specified for \"-O\" command");
681         if (options.control_path != NULL)
682                 control_client(options.control_path);
683
684         /* Open a connection to the remote host. */
685         if (ssh_connect(host, &hostaddr, options.port,
686             options.address_family, options.connection_attempts,
687 #ifdef HAVE_CYGWIN
688             options.use_privileged_port,
689 #else
690             original_effective_uid == 0 && options.use_privileged_port,
691 #endif
692             options.proxy_command) != 0)
693                 exit(255);
694
695         /*
696          * If we successfully made the connection, load the host private key
697          * in case we will need it later for combined rsa-rhosts
698          * authentication. This must be done before releasing extra
699          * privileges, because the file is only readable by root.
700          * If we cannot access the private keys, load the public keys
701          * instead and try to execute the ssh-keysign helper instead.
702          */
703         sensitive_data.nkeys = 0;
704         sensitive_data.keys = NULL;
705         sensitive_data.external_keysign = 0;
706         if (options.rhosts_rsa_authentication ||
707             options.hostbased_authentication) {
708                 sensitive_data.nkeys = 3;
709                 sensitive_data.keys = xcalloc(sensitive_data.nkeys,
710                     sizeof(Key));
711
712                 PRIV_START;
713                 sensitive_data.keys[0] = key_load_private_type(KEY_RSA1,
714                     _PATH_HOST_KEY_FILE, "", NULL, NULL);
715                 sensitive_data.keys[1] = key_load_private_type(KEY_DSA,
716                     _PATH_HOST_DSA_KEY_FILE, "", NULL, NULL);
717                 sensitive_data.keys[2] = key_load_private_type(KEY_RSA,
718                     _PATH_HOST_RSA_KEY_FILE, "", NULL, NULL);
719                 PRIV_END;
720
721                 if (options.hostbased_authentication == 1 &&
722                     sensitive_data.keys[0] == NULL &&
723                     sensitive_data.keys[1] == NULL &&
724                     sensitive_data.keys[2] == NULL) {
725                         sensitive_data.keys[1] = key_load_public(
726                             _PATH_HOST_DSA_KEY_FILE, NULL);
727                         sensitive_data.keys[2] = key_load_public(
728                             _PATH_HOST_RSA_KEY_FILE, NULL);
729                         sensitive_data.external_keysign = 1;
730                 }
731         }
732         /*
733          * Get rid of any extra privileges that we may have.  We will no
734          * longer need them.  Also, extra privileges could make it very hard
735          * to read identity files and other non-world-readable files from the
736          * user's home directory if it happens to be on a NFS volume where
737          * root is mapped to nobody.
738          */
739         if (original_effective_uid == 0) {
740                 PRIV_START;
741                 permanently_set_uid(pw);
742         }
743
744         /*
745          * Now that we are back to our own permissions, create ~/.ssh
746          * directory if it doesn't already exist.
747          */
748         snprintf(buf, sizeof buf, "%.100s%s%.100s", pw->pw_dir, strcmp(pw->pw_dir, "/") ? "/" : "", _PATH_SSH_USER_DIR);
749         if (stat(buf, &st) < 0)
750                 if (mkdir(buf, 0700) < 0)
751                         error("Could not create directory '%.200s'.", buf);
752
753         /* load options.identity_files */
754         load_public_identity_files();
755
756         /* Expand ~ in known host file names. */
757         /* XXX mem-leaks: */
758         options.system_hostfile =
759             tilde_expand_filename(options.system_hostfile, original_real_uid);
760         options.user_hostfile =
761             tilde_expand_filename(options.user_hostfile, original_real_uid);
762         options.system_hostfile2 =
763             tilde_expand_filename(options.system_hostfile2, original_real_uid);
764         options.user_hostfile2 =
765             tilde_expand_filename(options.user_hostfile2, original_real_uid);
766
767         signal(SIGPIPE, SIG_IGN); /* ignore SIGPIPE early */
768
769         /* Log into the remote system.  This never returns if the login fails. */
770         ssh_login(&sensitive_data, host, (struct sockaddr *)&hostaddr, pw);
771
772         /* We no longer need the private host keys.  Clear them now. */
773         if (sensitive_data.nkeys != 0) {
774                 for (i = 0; i < sensitive_data.nkeys; i++) {
775                         if (sensitive_data.keys[i] != NULL) {
776                                 /* Destroys contents safely */
777                                 debug3("clear hostkey %d", i);
778                                 key_free(sensitive_data.keys[i]);
779                                 sensitive_data.keys[i] = NULL;
780                         }
781                 }
782                 xfree(sensitive_data.keys);
783         }
784         for (i = 0; i < options.num_identity_files; i++) {
785                 if (options.identity_files[i]) {
786                         xfree(options.identity_files[i]);
787                         options.identity_files[i] = NULL;
788                 }
789                 if (options.identity_keys[i]) {
790                         key_free(options.identity_keys[i]);
791                         options.identity_keys[i] = NULL;
792                 }
793         }
794
795         exit_status = compat20 ? ssh_session2() : ssh_session();
796         packet_close();
797
798         if (options.control_path != NULL && control_fd != -1)
799                 unlink(options.control_path);
800
801         /*
802          * Send SIGHUP to proxy command if used. We don't wait() in
803          * case it hangs and instead rely on init to reap the child
804          */
805         if (proxy_command_pid > 1)
806                 kill(proxy_command_pid, SIGHUP);
807
808         return exit_status;
809 }
810
811 static void
812 ssh_init_forwarding(void)
813 {
814         int success = 0;
815         int i;
816
817         /* Initiate local TCP/IP port forwardings. */
818         for (i = 0; i < options.num_local_forwards; i++) {
819                 debug("Local connections to %.200s:%d forwarded to remote "
820                     "address %.200s:%d",
821                     (options.local_forwards[i].listen_host == NULL) ?
822                     (options.gateway_ports ? "*" : "LOCALHOST") :
823                     options.local_forwards[i].listen_host,
824                     options.local_forwards[i].listen_port,
825                     options.local_forwards[i].connect_host,
826                     options.local_forwards[i].connect_port);
827                 success += channel_setup_local_fwd_listener(
828                     options.local_forwards[i].listen_host,
829                     options.local_forwards[i].listen_port,
830                     options.local_forwards[i].connect_host,
831                     options.local_forwards[i].connect_port,
832                     options.gateway_ports);
833         }
834         if (i > 0 && success != i && options.exit_on_forward_failure)
835                 fatal("Could not request local forwarding.");
836         if (i > 0 && success == 0)
837                 error("Could not request local forwarding.");
838
839         /* Initiate remote TCP/IP port forwardings. */
840         for (i = 0; i < options.num_remote_forwards; i++) {
841                 debug("Remote connections from %.200s:%d forwarded to "
842                     "local address %.200s:%d",
843                     (options.remote_forwards[i].listen_host == NULL) ?
844                     "LOCALHOST" : options.remote_forwards[i].listen_host,
845                     options.remote_forwards[i].listen_port,
846                     options.remote_forwards[i].connect_host,
847                     options.remote_forwards[i].connect_port);
848                 if (channel_request_remote_forwarding(
849                     options.remote_forwards[i].listen_host,
850                     options.remote_forwards[i].listen_port,
851                     options.remote_forwards[i].connect_host,
852                     options.remote_forwards[i].connect_port) < 0) {
853                         if (options.exit_on_forward_failure)
854                                 fatal("Could not request remote forwarding.");
855                         else
856                                 logit("Warning: Could not request remote "
857                                     "forwarding.");
858                 }
859         }
860
861         /* Initiate tunnel forwarding. */
862         if (options.tun_open != SSH_TUNMODE_NO) {
863                 if (client_request_tun_fwd(options.tun_open,
864                     options.tun_local, options.tun_remote) == -1) {
865                         if (options.exit_on_forward_failure)
866                                 fatal("Could not request tunnel forwarding.");
867                         else
868                                 error("Could not request tunnel forwarding.");
869                 }
870         }                       
871 }
872
873 static void
874 check_agent_present(void)
875 {
876         if (options.forward_agent) {
877                 /* Clear agent forwarding if we don't have an agent. */
878                 if (!ssh_agent_present())
879                         options.forward_agent = 0;
880         }
881 }
882
883 static int
884 ssh_session(void)
885 {
886         int type;
887         int interactive = 0;
888         int have_tty = 0;
889         struct winsize ws;
890         char *cp;
891         const char *display;
892
893         /* Enable compression if requested. */
894         if (options.compression) {
895                 debug("Requesting compression at level %d.", options.compression_level);
896
897                 if (options.compression_level < 1 || options.compression_level > 9)
898                         fatal("Compression level must be from 1 (fast) to 9 (slow, best).");
899
900                 /* Send the request. */
901                 packet_start(SSH_CMSG_REQUEST_COMPRESSION);
902                 packet_put_int(options.compression_level);
903                 packet_send();
904                 packet_write_wait();
905                 type = packet_read();
906                 if (type == SSH_SMSG_SUCCESS)
907                         packet_start_compression(options.compression_level);
908                 else if (type == SSH_SMSG_FAILURE)
909                         logit("Warning: Remote host refused compression.");
910                 else
911                         packet_disconnect("Protocol error waiting for compression response.");
912         }
913         /* Allocate a pseudo tty if appropriate. */
914         if (tty_flag) {
915                 debug("Requesting pty.");
916
917                 /* Start the packet. */
918                 packet_start(SSH_CMSG_REQUEST_PTY);
919
920                 /* Store TERM in the packet.  There is no limit on the
921                    length of the string. */
922                 cp = getenv("TERM");
923                 if (!cp)
924                         cp = "";
925                 packet_put_cstring(cp);
926
927                 /* Store window size in the packet. */
928                 if (ioctl(fileno(stdin), TIOCGWINSZ, &ws) < 0)
929                         memset(&ws, 0, sizeof(ws));
930                 packet_put_int((u_int)ws.ws_row);
931                 packet_put_int((u_int)ws.ws_col);
932                 packet_put_int((u_int)ws.ws_xpixel);
933                 packet_put_int((u_int)ws.ws_ypixel);
934
935                 /* Store tty modes in the packet. */
936                 tty_make_modes(fileno(stdin), NULL);
937
938                 /* Send the packet, and wait for it to leave. */
939                 packet_send();
940                 packet_write_wait();
941
942                 /* Read response from the server. */
943                 type = packet_read();
944                 if (type == SSH_SMSG_SUCCESS) {
945                         interactive = 1;
946                         have_tty = 1;
947                 } else if (type == SSH_SMSG_FAILURE)
948                         logit("Warning: Remote host failed or refused to allocate a pseudo tty.");
949                 else
950                         packet_disconnect("Protocol error waiting for pty request response.");
951         }
952         /* Request X11 forwarding if enabled and DISPLAY is set. */
953         display = getenv("DISPLAY");
954         if (options.forward_x11 && display != NULL) {
955                 char *proto, *data;
956                 /* Get reasonable local authentication information. */
957                 client_x11_get_proto(display, options.xauth_location,
958                     options.forward_x11_trusted, &proto, &data);
959                 /* Request forwarding with authentication spoofing. */
960                 debug("Requesting X11 forwarding with authentication spoofing.");
961                 x11_request_forwarding_with_spoofing(0, display, proto, data);
962
963                 /* Read response from the server. */
964                 type = packet_read();
965                 if (type == SSH_SMSG_SUCCESS) {
966                         interactive = 1;
967                 } else if (type == SSH_SMSG_FAILURE) {
968                         logit("Warning: Remote host denied X11 forwarding.");
969                 } else {
970                         packet_disconnect("Protocol error waiting for X11 forwarding");
971                 }
972         }
973         /* Tell the packet module whether this is an interactive session. */
974         packet_set_interactive(interactive);
975
976         /* Request authentication agent forwarding if appropriate. */
977         check_agent_present();
978
979         if (options.forward_agent) {
980                 debug("Requesting authentication agent forwarding.");
981                 auth_request_forwarding();
982
983                 /* Read response from the server. */
984                 type = packet_read();
985                 packet_check_eom();
986                 if (type != SSH_SMSG_SUCCESS)
987                         logit("Warning: Remote host denied authentication agent forwarding.");
988         }
989
990         /* Initiate port forwardings. */
991         ssh_init_forwarding();
992
993         /* If requested, let ssh continue in the background. */
994         if (fork_after_authentication_flag)
995                 if (daemon(1, 1) < 0)
996                         fatal("daemon() failed: %.200s", strerror(errno));
997
998         /*
999          * If a command was specified on the command line, execute the
1000          * command now. Otherwise request the server to start a shell.
1001          */
1002         if (buffer_len(&command) > 0) {
1003                 int len = buffer_len(&command);
1004                 if (len > 900)
1005                         len = 900;
1006                 debug("Sending command: %.*s", len, (u_char *)buffer_ptr(&command));
1007                 packet_start(SSH_CMSG_EXEC_CMD);
1008                 packet_put_string(buffer_ptr(&command), buffer_len(&command));
1009                 packet_send();
1010                 packet_write_wait();
1011         } else {
1012                 debug("Requesting shell.");
1013                 packet_start(SSH_CMSG_EXEC_SHELL);
1014                 packet_send();
1015                 packet_write_wait();
1016         }
1017
1018         /* Enter the interactive session. */
1019         return client_loop(have_tty, tty_flag ?
1020             options.escape_char : SSH_ESCAPECHAR_NONE, 0);
1021 }
1022
1023 static void
1024 ssh_subsystem_reply(int type, u_int32_t seq, void *ctxt)
1025 {
1026         int id, len;
1027
1028         id = packet_get_int();
1029         len = buffer_len(&command);
1030         if (len > 900)
1031                 len = 900;
1032         packet_check_eom();
1033         if (type == SSH2_MSG_CHANNEL_FAILURE)
1034                 fatal("Request for subsystem '%.*s' failed on channel %d",
1035                     len, (u_char *)buffer_ptr(&command), id);
1036 }
1037
1038 void
1039 client_global_request_reply_fwd(int type, u_int32_t seq, void *ctxt)
1040 {
1041         int i;
1042
1043         i = client_global_request_id++;
1044         if (i >= options.num_remote_forwards)
1045                 return;
1046         debug("remote forward %s for: listen %d, connect %s:%d",
1047             type == SSH2_MSG_REQUEST_SUCCESS ? "success" : "failure",
1048             options.remote_forwards[i].listen_port,
1049             options.remote_forwards[i].connect_host,
1050             options.remote_forwards[i].connect_port);
1051         if (type == SSH2_MSG_REQUEST_FAILURE) {
1052                 if (options.exit_on_forward_failure)
1053                         fatal("Error: remote port forwarding failed for "
1054                             "listen port %d",
1055                             options.remote_forwards[i].listen_port);
1056                 else
1057                         logit("Warning: remote port forwarding failed for "
1058                             "listen port %d",
1059                             options.remote_forwards[i].listen_port);
1060         }
1061 }
1062
1063 static void
1064 ssh_control_listener(void)
1065 {
1066         struct sockaddr_un addr;
1067         mode_t old_umask;
1068         int addr_len;
1069
1070         if (options.control_path == NULL ||
1071             options.control_master == SSHCTL_MASTER_NO)
1072                 return;
1073
1074         debug("setting up multiplex master socket");
1075
1076         memset(&addr, '\0', sizeof(addr));
1077         addr.sun_family = AF_UNIX;
1078         addr_len = offsetof(struct sockaddr_un, sun_path) +
1079             strlen(options.control_path) + 1;
1080
1081         if (strlcpy(addr.sun_path, options.control_path,
1082             sizeof(addr.sun_path)) >= sizeof(addr.sun_path))
1083                 fatal("ControlPath too long");
1084
1085         if ((control_fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0)
1086                 fatal("%s socket(): %s", __func__, strerror(errno));
1087
1088         old_umask = umask(0177);
1089         if (bind(control_fd, (struct sockaddr *)&addr, addr_len) == -1) {
1090                 control_fd = -1;
1091                 if (errno == EINVAL || errno == EADDRINUSE)
1092                         fatal("ControlSocket %s already exists",
1093                             options.control_path);
1094                 else
1095                         fatal("%s bind(): %s", __func__, strerror(errno));
1096         }
1097         umask(old_umask);
1098
1099         if (listen(control_fd, 64) == -1)
1100                 fatal("%s listen(): %s", __func__, strerror(errno));
1101
1102         set_nonblock(control_fd);
1103 }
1104
1105 /* request pty/x11/agent/tcpfwd/shell for channel */
1106 static void
1107 ssh_session2_setup(int id, void *arg)
1108 {
1109         extern char **environ;
1110         const char *display;
1111         int interactive = tty_flag;
1112
1113         display = getenv("DISPLAY");
1114         if (options.forward_x11 && display != NULL) {
1115                 char *proto, *data;
1116                 /* Get reasonable local authentication information. */
1117                 client_x11_get_proto(display, options.xauth_location,
1118                     options.forward_x11_trusted, &proto, &data);
1119                 /* Request forwarding with authentication spoofing. */
1120                 debug("Requesting X11 forwarding with authentication spoofing.");
1121                 x11_request_forwarding_with_spoofing(id, display, proto, data);
1122                 interactive = 1;
1123                 /* XXX wait for reply */
1124         }
1125
1126         check_agent_present();
1127         if (options.forward_agent) {
1128                 debug("Requesting authentication agent forwarding.");
1129                 channel_request_start(id, "auth-agent-req@openssh.com", 0);
1130                 packet_send();
1131         }
1132
1133         client_session2_setup(id, tty_flag, subsystem_flag, getenv("TERM"),
1134             NULL, fileno(stdin), &command, environ, &ssh_subsystem_reply);
1135
1136         packet_set_interactive(interactive);
1137 }
1138
1139 /* open new channel for a session */
1140 static int
1141 ssh_session2_open(void)
1142 {
1143         Channel *c;
1144         int window, packetmax, in, out, err;
1145
1146         if (stdin_null_flag) {
1147                 in = open(_PATH_DEVNULL, O_RDONLY);
1148         } else {
1149                 in = dup(STDIN_FILENO);
1150         }
1151         out = dup(STDOUT_FILENO);
1152         err = dup(STDERR_FILENO);
1153
1154         if (in < 0 || out < 0 || err < 0)
1155                 fatal("dup() in/out/err failed");
1156
1157         /* enable nonblocking unless tty */
1158         if (!isatty(in))
1159                 set_nonblock(in);
1160         if (!isatty(out))
1161                 set_nonblock(out);
1162         if (!isatty(err))
1163                 set_nonblock(err);
1164
1165         window = CHAN_SES_WINDOW_DEFAULT;
1166         packetmax = CHAN_SES_PACKET_DEFAULT;
1167         if (tty_flag) {
1168                 window >>= 1;
1169                 packetmax >>= 1;
1170         }
1171         c = channel_new(
1172             "session", SSH_CHANNEL_OPENING, in, out, err,
1173             window, packetmax, CHAN_EXTENDED_WRITE,
1174             "client-session", /*nonblock*/0);
1175
1176         debug3("ssh_session2_open: channel_new: %d", c->self);
1177
1178         channel_send_open(c->self);
1179         if (!no_shell_flag)
1180                 channel_register_confirm(c->self, ssh_session2_setup, NULL);
1181
1182         return c->self;
1183 }
1184
1185 static int
1186 ssh_session2(void)
1187 {
1188         int id = -1;
1189
1190         /* XXX should be pre-session */
1191         ssh_init_forwarding();
1192
1193         if (!no_shell_flag || (datafellows & SSH_BUG_DUMMYCHAN))
1194                 id = ssh_session2_open();
1195
1196         /* Execute a local command */
1197         if (options.local_command != NULL &&
1198             options.permit_local_command)
1199                 ssh_local_cmd(options.local_command);
1200
1201         /* Start listening for multiplex clients */
1202         ssh_control_listener();
1203
1204         /* If requested, let ssh continue in the background. */
1205         if (fork_after_authentication_flag)
1206                 if (daemon(1, 1) < 0)
1207                         fatal("daemon() failed: %.200s", strerror(errno));
1208
1209         return client_loop(tty_flag, tty_flag ?
1210             options.escape_char : SSH_ESCAPECHAR_NONE, id);
1211 }
1212
1213 static void
1214 load_public_identity_files(void)
1215 {
1216         char *filename, *cp, thishost[NI_MAXHOST];
1217         int i = 0;
1218         Key *public;
1219         struct passwd *pw;
1220 #ifdef SMARTCARD
1221         Key **keys;
1222
1223         if (options.smartcard_device != NULL &&
1224             options.num_identity_files < SSH_MAX_IDENTITY_FILES &&
1225             (keys = sc_get_keys(options.smartcard_device, NULL)) != NULL) {
1226                 int count = 0;
1227                 for (i = 0; keys[i] != NULL; i++) {
1228                         count++;
1229                         memmove(&options.identity_files[1], &options.identity_files[0],
1230                             sizeof(char *) * (SSH_MAX_IDENTITY_FILES - 1));
1231                         memmove(&options.identity_keys[1], &options.identity_keys[0],
1232                             sizeof(Key *) * (SSH_MAX_IDENTITY_FILES - 1));
1233                         options.num_identity_files++;
1234                         options.identity_keys[0] = keys[i];
1235                         options.identity_files[0] = sc_get_key_label(keys[i]);
1236                 }
1237                 if (options.num_identity_files > SSH_MAX_IDENTITY_FILES)
1238                         options.num_identity_files = SSH_MAX_IDENTITY_FILES;
1239                 i = count;
1240                 xfree(keys);
1241         }
1242 #endif /* SMARTCARD */
1243         if ((pw = getpwuid(original_real_uid)) == NULL)
1244                 fatal("load_public_identity_files: getpwuid failed");
1245         if (gethostname(thishost, sizeof(thishost)) == -1)
1246                 fatal("load_public_identity_files: gethostname: %s",
1247                     strerror(errno));
1248         for (; i < options.num_identity_files; i++) {
1249                 cp = tilde_expand_filename(options.identity_files[i],
1250                     original_real_uid);
1251                 filename = percent_expand(cp, "d", pw->pw_dir,
1252                     "u", pw->pw_name, "l", thishost, "h", host,
1253                     "r", options.user, (char *)NULL);
1254                 xfree(cp);
1255                 public = key_load_public(filename, NULL);
1256                 debug("identity file %s type %d", filename,
1257                     public ? public->type : -1);
1258                 xfree(options.identity_files[i]);
1259                 options.identity_files[i] = filename;
1260                 options.identity_keys[i] = public;
1261         }
1262 }
1263
1264 static void
1265 control_client_sighandler(int signo)
1266 {
1267         control_client_terminate = signo;
1268 }
1269
1270 static void
1271 control_client_sigrelay(int signo)
1272 {
1273         if (control_server_pid > 1)
1274                 kill(control_server_pid, signo);
1275 }
1276
1277 static int
1278 env_permitted(char *env)
1279 {
1280         int i, ret;
1281         char name[1024], *cp;
1282
1283         if ((cp = strchr(env, '=')) == NULL || cp == env)
1284                 return (0);
1285         ret = snprintf(name, sizeof(name), "%.*s", (int)(cp - env), env);
1286         if (ret <= 0 || (size_t)ret >= sizeof(name))
1287                 fatal("env_permitted: name '%.100s...' too long", env);
1288
1289         for (i = 0; i < options.num_send_env; i++)
1290                 if (match_pattern(name, options.send_env[i]))
1291                         return (1);
1292
1293         return (0);
1294 }
1295
1296 static void
1297 control_client(const char *path)
1298 {
1299         struct sockaddr_un addr;
1300         int i, r, fd, sock, exitval[2], num_env, addr_len;
1301         Buffer m;
1302         char *term;
1303         extern char **environ;
1304         u_int  flags;
1305
1306         if (mux_command == 0)
1307                 mux_command = SSHMUX_COMMAND_OPEN;
1308
1309         switch (options.control_master) {
1310         case SSHCTL_MASTER_AUTO:
1311         case SSHCTL_MASTER_AUTO_ASK:
1312                 debug("auto-mux: Trying existing master");
1313                 /* FALLTHROUGH */
1314         case SSHCTL_MASTER_NO:
1315                 break;
1316         default:
1317                 return;
1318         }
1319
1320         memset(&addr, '\0', sizeof(addr));
1321         addr.sun_family = AF_UNIX;
1322         addr_len = offsetof(struct sockaddr_un, sun_path) +
1323             strlen(path) + 1;
1324
1325         if (strlcpy(addr.sun_path, path,
1326             sizeof(addr.sun_path)) >= sizeof(addr.sun_path))
1327                 fatal("ControlPath too long");
1328
1329         if ((sock = socket(PF_UNIX, SOCK_STREAM, 0)) < 0)
1330                 fatal("%s socket(): %s", __func__, strerror(errno));
1331
1332         if (connect(sock, (struct sockaddr *)&addr, addr_len) == -1) {
1333                 if (mux_command != SSHMUX_COMMAND_OPEN) {
1334                         fatal("Control socket connect(%.100s): %s", path,
1335                             strerror(errno));
1336                 }
1337                 if (errno == ENOENT)
1338                         debug("Control socket \"%.100s\" does not exist", path);
1339                 else {
1340                         error("Control socket connect(%.100s): %s", path,
1341                             strerror(errno));
1342                 }
1343                 close(sock);
1344                 return;
1345         }
1346
1347         if (stdin_null_flag) {
1348                 if ((fd = open(_PATH_DEVNULL, O_RDONLY)) == -1)
1349                         fatal("open(/dev/null): %s", strerror(errno));
1350                 if (dup2(fd, STDIN_FILENO) == -1)
1351                         fatal("dup2: %s", strerror(errno));
1352                 if (fd > STDERR_FILENO)
1353                         close(fd);
1354         }
1355
1356         term = getenv("TERM");
1357
1358         flags = 0;
1359         if (tty_flag)
1360                 flags |= SSHMUX_FLAG_TTY;
1361         if (subsystem_flag)
1362                 flags |= SSHMUX_FLAG_SUBSYS;
1363         if (options.forward_x11)
1364                 flags |= SSHMUX_FLAG_X11_FWD;
1365         if (options.forward_agent)
1366                 flags |= SSHMUX_FLAG_AGENT_FWD;
1367
1368         buffer_init(&m);
1369
1370         /* Send our command to server */
1371         buffer_put_int(&m, mux_command);
1372         buffer_put_int(&m, flags);
1373         if (ssh_msg_send(sock, SSHMUX_VER, &m) == -1)
1374                 fatal("%s: msg_send", __func__);
1375         buffer_clear(&m);
1376
1377         /* Get authorisation status and PID of controlee */
1378         if (ssh_msg_recv(sock, &m) == -1)
1379                 fatal("%s: msg_recv", __func__);
1380         if (buffer_get_char(&m) != SSHMUX_VER)
1381                 fatal("%s: wrong version", __func__);
1382         if (buffer_get_int(&m) != 1)
1383                 fatal("Connection to master denied");
1384         control_server_pid = buffer_get_int(&m);
1385
1386         buffer_clear(&m);
1387
1388         switch (mux_command) {
1389         case SSHMUX_COMMAND_ALIVE_CHECK:
1390                 fprintf(stderr, "Master running (pid=%d)\r\n",
1391                     control_server_pid);
1392                 exit(0);
1393         case SSHMUX_COMMAND_TERMINATE:
1394                 fprintf(stderr, "Exit request sent.\r\n");
1395                 exit(0);
1396         case SSHMUX_COMMAND_OPEN:
1397                 /* continue below */
1398                 break;
1399         default:
1400                 fatal("silly mux_command %d", mux_command);
1401         }
1402
1403         /* SSHMUX_COMMAND_OPEN */
1404         buffer_put_cstring(&m, term ? term : "");
1405         buffer_append(&command, "\0", 1);
1406         buffer_put_cstring(&m, buffer_ptr(&command));
1407
1408         if (options.num_send_env == 0 || environ == NULL) {
1409                 buffer_put_int(&m, 0);
1410         } else {
1411                 /* Pass environment */
1412                 num_env = 0;
1413                 for (i = 0; environ[i] != NULL; i++)
1414                         if (env_permitted(environ[i]))
1415                                 num_env++; /* Count */
1416
1417                 buffer_put_int(&m, num_env);
1418
1419                 for (i = 0; environ[i] != NULL && num_env >= 0; i++)
1420                         if (env_permitted(environ[i])) {
1421                                 num_env--;
1422                                 buffer_put_cstring(&m, environ[i]);
1423                         }
1424         }
1425
1426         if (ssh_msg_send(sock, SSHMUX_VER, &m) == -1)
1427                 fatal("%s: msg_send", __func__);
1428
1429         mm_send_fd(sock, STDIN_FILENO);
1430         mm_send_fd(sock, STDOUT_FILENO);
1431         mm_send_fd(sock, STDERR_FILENO);
1432
1433         /* Wait for reply, so master has a chance to gather ttymodes */
1434         buffer_clear(&m);
1435         if (ssh_msg_recv(sock, &m) == -1)
1436                 fatal("%s: msg_recv", __func__);
1437         if (buffer_get_char(&m) != SSHMUX_VER)
1438                 fatal("%s: wrong version", __func__);
1439         buffer_free(&m);
1440
1441         signal(SIGHUP, control_client_sighandler);
1442         signal(SIGINT, control_client_sighandler);
1443         signal(SIGTERM, control_client_sighandler);
1444         signal(SIGWINCH, control_client_sigrelay);
1445
1446         if (tty_flag)
1447                 enter_raw_mode();
1448
1449         /*
1450          * Stick around until the controlee closes the client_fd.
1451          * Before it does, it is expected to write this process' exit
1452          * value (one int). This process must read the value and wait for
1453          * the closure of the client_fd; if this one closes early, the 
1454          * multiplex master will terminate early too (possibly losing data).
1455          */
1456         exitval[0] = 0;
1457         for (i = 0; !control_client_terminate && i < (int)sizeof(exitval);) {
1458                 r = read(sock, (char *)exitval + i, sizeof(exitval) - i);
1459                 if (r == 0) {
1460                         debug2("Received EOF from master");
1461                         break;
1462                 }
1463                 if (r == -1) {
1464                         if (errno == EINTR)
1465                                 continue;
1466                         fatal("%s: read %s", __func__, strerror(errno));
1467                 }
1468                 i += r;
1469         }
1470
1471         close(sock);
1472         leave_raw_mode();
1473         if (i > (int)sizeof(int))
1474                 fatal("%s: master returned too much data (%d > %lu)",
1475                     __func__, i, sizeof(int));
1476         if (control_client_terminate) {
1477                 debug2("Exiting on signal %d", control_client_terminate);
1478                 exitval[0] = 255;
1479         } else if (i < (int)sizeof(int)) {
1480                 debug2("Control master terminated unexpectedly");
1481                 exitval[0] = 255;
1482         } else
1483                 debug2("Received exit status from master %d", exitval[0]);
1484
1485         if (tty_flag && options.log_level != SYSLOG_LEVEL_QUIET)
1486                 fprintf(stderr, "Shared connection to %s closed.\r\n", host);
1487
1488         exit(exitval[0]);
1489 }