channels.c revision 262566
1/* $OpenBSD: channels.c,v 1.328 2013/12/19 01:04:36 djm Exp $ */
2/* $FreeBSD: stable/10/crypto/openssh/channels.c 262566 2014-02-27 17:29:02Z des $ */
3/*
4 * Author: Tatu Ylonen <ylo@cs.hut.fi>
5 * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
6 *                    All rights reserved
7 * This file contains functions for generic socket connection forwarding.
8 * There is also code for initiating connection forwarding for X11 connections,
9 * arbitrary tcp/ip connections, and the authentication agent connection.
10 *
11 * As far as I am concerned, the code I have written for this software
12 * can be used freely for any purpose.  Any derived versions of this
13 * software must be clearly marked as such, and if the derived work is
14 * incompatible with the protocol description in the RFC file, it must be
15 * called by a name other than "ssh" or "Secure Shell".
16 *
17 * SSH2 support added by Markus Friedl.
18 * Copyright (c) 1999, 2000, 2001, 2002 Markus Friedl.  All rights reserved.
19 * Copyright (c) 1999 Dug Song.  All rights reserved.
20 * Copyright (c) 1999 Theo de Raadt.  All rights reserved.
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#include <sys/ioctl.h>
47#include <sys/un.h>
48#include <sys/socket.h>
49#ifdef HAVE_SYS_TIME_H
50# include <sys/time.h>
51#endif
52
53#include <netinet/in.h>
54#include <arpa/inet.h>
55
56#include <errno.h>
57#include <fcntl.h>
58#include <netdb.h>
59#include <stdio.h>
60#include <stdlib.h>
61#include <string.h>
62#include <termios.h>
63#include <unistd.h>
64#include <stdarg.h>
65
66#include "openbsd-compat/sys-queue.h"
67#include "xmalloc.h"
68#include "ssh.h"
69#include "ssh1.h"
70#include "ssh2.h"
71#include "packet.h"
72#include "log.h"
73#include "misc.h"
74#include "buffer.h"
75#include "channels.h"
76#include "compat.h"
77#include "canohost.h"
78#include "key.h"
79#include "authfd.h"
80#include "pathnames.h"
81
82/* -- channel core */
83
84/*
85 * Pointer to an array containing all allocated channels.  The array is
86 * dynamically extended as needed.
87 */
88static Channel **channels = NULL;
89
90/*
91 * Size of the channel array.  All slots of the array must always be
92 * initialized (at least the type field); unused slots set to NULL
93 */
94static u_int channels_alloc = 0;
95
96/*
97 * Maximum file descriptor value used in any of the channels.  This is
98 * updated in channel_new.
99 */
100static int channel_max_fd = 0;
101
102
103/* -- tcp forwarding */
104
105/*
106 * Data structure for storing which hosts are permitted for forward requests.
107 * The local sides of any remote forwards are stored in this array to prevent
108 * a corrupt remote server from accessing arbitrary TCP/IP ports on our local
109 * network (which might be behind a firewall).
110 */
111typedef struct {
112	char *host_to_connect;		/* Connect to 'host'. */
113	u_short port_to_connect;	/* Connect to 'port'. */
114	u_short listen_port;		/* Remote side should listen port number. */
115} ForwardPermission;
116
117/* List of all permitted host/port pairs to connect by the user. */
118static ForwardPermission *permitted_opens = NULL;
119
120/* List of all permitted host/port pairs to connect by the admin. */
121static ForwardPermission *permitted_adm_opens = NULL;
122
123/* Number of permitted host/port pairs in the array permitted by the user. */
124static int num_permitted_opens = 0;
125
126/* Number of permitted host/port pair in the array permitted by the admin. */
127static int num_adm_permitted_opens = 0;
128
129/* special-case port number meaning allow any port */
130#define FWD_PERMIT_ANY_PORT	0
131
132/*
133 * If this is true, all opens are permitted.  This is the case on the server
134 * on which we have to trust the client anyway, and the user could do
135 * anything after logging in anyway.
136 */
137static int all_opens_permitted = 0;
138
139
140/* -- X11 forwarding */
141
142/* Maximum number of fake X11 displays to try. */
143#define MAX_DISPLAYS  1000
144
145/* Saved X11 local (client) display. */
146static char *x11_saved_display = NULL;
147
148/* Saved X11 authentication protocol name. */
149static char *x11_saved_proto = NULL;
150
151/* Saved X11 authentication data.  This is the real data. */
152static char *x11_saved_data = NULL;
153static u_int x11_saved_data_len = 0;
154
155/*
156 * Fake X11 authentication data.  This is what the server will be sending us;
157 * we should replace any occurrences of this by the real data.
158 */
159static u_char *x11_fake_data = NULL;
160static u_int x11_fake_data_len;
161
162
163/* -- agent forwarding */
164
165#define	NUM_SOCKS	10
166
167/* AF_UNSPEC or AF_INET or AF_INET6 */
168static int IPv4or6 = AF_UNSPEC;
169
170/* helper */
171static void port_open_helper(Channel *c, char *rtype);
172
173/* non-blocking connect helpers */
174static int connect_next(struct channel_connect *);
175static void channel_connect_ctx_free(struct channel_connect *);
176
177/* -- HPN */
178
179static int hpn_disabled = 0;
180static u_int buffer_size = CHAN_HPN_MIN_WINDOW_DEFAULT;
181
182/* -- channel core */
183
184Channel *
185channel_by_id(int id)
186{
187	Channel *c;
188
189	if (id < 0 || (u_int)id >= channels_alloc) {
190		logit("channel_by_id: %d: bad id", id);
191		return NULL;
192	}
193	c = channels[id];
194	if (c == NULL) {
195		logit("channel_by_id: %d: bad id: channel free", id);
196		return NULL;
197	}
198	return c;
199}
200
201/*
202 * Returns the channel if it is allowed to receive protocol messages.
203 * Private channels, like listening sockets, may not receive messages.
204 */
205Channel *
206channel_lookup(int id)
207{
208	Channel *c;
209
210	if ((c = channel_by_id(id)) == NULL)
211		return (NULL);
212
213	switch (c->type) {
214	case SSH_CHANNEL_X11_OPEN:
215	case SSH_CHANNEL_LARVAL:
216	case SSH_CHANNEL_CONNECTING:
217	case SSH_CHANNEL_DYNAMIC:
218	case SSH_CHANNEL_OPENING:
219	case SSH_CHANNEL_OPEN:
220	case SSH_CHANNEL_INPUT_DRAINING:
221	case SSH_CHANNEL_OUTPUT_DRAINING:
222	case SSH_CHANNEL_ABANDONED:
223		return (c);
224	}
225	logit("Non-public channel %d, type %d.", id, c->type);
226	return (NULL);
227}
228
229/*
230 * Register filedescriptors for a channel, used when allocating a channel or
231 * when the channel consumer/producer is ready, e.g. shell exec'd
232 */
233static void
234channel_register_fds(Channel *c, int rfd, int wfd, int efd,
235    int extusage, int nonblock, int is_tty)
236{
237	/* Update the maximum file descriptor value. */
238	channel_max_fd = MAX(channel_max_fd, rfd);
239	channel_max_fd = MAX(channel_max_fd, wfd);
240	channel_max_fd = MAX(channel_max_fd, efd);
241
242	if (rfd != -1)
243		fcntl(rfd, F_SETFD, FD_CLOEXEC);
244	if (wfd != -1 && wfd != rfd)
245		fcntl(wfd, F_SETFD, FD_CLOEXEC);
246	if (efd != -1 && efd != rfd && efd != wfd)
247		fcntl(efd, F_SETFD, FD_CLOEXEC);
248
249	c->rfd = rfd;
250	c->wfd = wfd;
251	c->sock = (rfd == wfd) ? rfd : -1;
252	c->efd = efd;
253	c->extended_usage = extusage;
254
255	if ((c->isatty = is_tty) != 0)
256		debug2("channel %d: rfd %d isatty", c->self, c->rfd);
257#ifdef _AIX
258	/* XXX: Later AIX versions can't push as much data to tty */
259	c->wfd_isatty = is_tty || isatty(c->wfd);
260#endif
261
262	/* enable nonblocking mode */
263	if (nonblock) {
264		if (rfd != -1)
265			set_nonblock(rfd);
266		if (wfd != -1)
267			set_nonblock(wfd);
268		if (efd != -1)
269			set_nonblock(efd);
270	}
271}
272
273/*
274 * Allocate a new channel object and set its type and socket. This will cause
275 * remote_name to be freed.
276 */
277Channel *
278channel_new(char *ctype, int type, int rfd, int wfd, int efd,
279    u_int window, u_int maxpack, int extusage, char *remote_name, int nonblock)
280{
281	int found;
282	u_int i;
283	Channel *c;
284
285	/* Do initial allocation if this is the first call. */
286	if (channels_alloc == 0) {
287		channels_alloc = 10;
288		channels = xcalloc(channels_alloc, sizeof(Channel *));
289		for (i = 0; i < channels_alloc; i++)
290			channels[i] = NULL;
291	}
292	/* Try to find a free slot where to put the new channel. */
293	for (found = -1, i = 0; i < channels_alloc; i++)
294		if (channels[i] == NULL) {
295			/* Found a free slot. */
296			found = (int)i;
297			break;
298		}
299	if (found < 0) {
300		/* There are no free slots.  Take last+1 slot and expand the array.  */
301		found = channels_alloc;
302		if (channels_alloc > 10000)
303			fatal("channel_new: internal error: channels_alloc %d "
304			    "too big.", channels_alloc);
305		channels = xrealloc(channels, channels_alloc + 10,
306		    sizeof(Channel *));
307		channels_alloc += 10;
308		debug2("channel: expanding %d", channels_alloc);
309		for (i = found; i < channels_alloc; i++)
310			channels[i] = NULL;
311	}
312	/* Initialize and return new channel. */
313	c = channels[found] = xcalloc(1, sizeof(Channel));
314	buffer_init(&c->input);
315	buffer_init(&c->output);
316	buffer_init(&c->extended);
317	c->path = NULL;
318	c->listening_addr = NULL;
319	c->listening_port = 0;
320	c->ostate = CHAN_OUTPUT_OPEN;
321	c->istate = CHAN_INPUT_OPEN;
322	c->flags = 0;
323	channel_register_fds(c, rfd, wfd, efd, extusage, nonblock, 0);
324	c->notbefore = 0;
325	c->self = found;
326	c->type = type;
327	c->ctype = ctype;
328	c->dynamic_window = 0;
329	c->local_window = window;
330	c->local_window_max = window;
331	c->local_consumed = 0;
332	c->local_maxpacket = maxpack;
333	c->remote_id = -1;
334	c->remote_name = xstrdup(remote_name);
335	c->remote_window = 0;
336	c->remote_maxpacket = 0;
337	c->force_drain = 0;
338	c->single_connection = 0;
339	c->detach_user = NULL;
340	c->detach_close = 0;
341	c->open_confirm = NULL;
342	c->open_confirm_ctx = NULL;
343	c->input_filter = NULL;
344	c->output_filter = NULL;
345	c->filter_ctx = NULL;
346	c->filter_cleanup = NULL;
347	c->ctl_chan = -1;
348	c->mux_rcb = NULL;
349	c->mux_ctx = NULL;
350	c->mux_pause = 0;
351	c->delayed = 1;		/* prevent call to channel_post handler */
352	TAILQ_INIT(&c->status_confirms);
353	debug("channel %d: new [%s]", found, remote_name);
354	return c;
355}
356
357static int
358channel_find_maxfd(void)
359{
360	u_int i;
361	int max = 0;
362	Channel *c;
363
364	for (i = 0; i < channels_alloc; i++) {
365		c = channels[i];
366		if (c != NULL) {
367			max = MAX(max, c->rfd);
368			max = MAX(max, c->wfd);
369			max = MAX(max, c->efd);
370		}
371	}
372	return max;
373}
374
375int
376channel_close_fd(int *fdp)
377{
378	int ret = 0, fd = *fdp;
379
380	if (fd != -1) {
381		ret = close(fd);
382		*fdp = -1;
383		if (fd == channel_max_fd)
384			channel_max_fd = channel_find_maxfd();
385	}
386	return ret;
387}
388
389/* Close all channel fd/socket. */
390static void
391channel_close_fds(Channel *c)
392{
393	channel_close_fd(&c->sock);
394	channel_close_fd(&c->rfd);
395	channel_close_fd(&c->wfd);
396	channel_close_fd(&c->efd);
397}
398
399/* Free the channel and close its fd/socket. */
400void
401channel_free(Channel *c)
402{
403	char *s;
404	u_int i, n;
405	struct channel_confirm *cc;
406
407	for (n = 0, i = 0; i < channels_alloc; i++)
408		if (channels[i])
409			n++;
410	debug("channel %d: free: %s, nchannels %u", c->self,
411	    c->remote_name ? c->remote_name : "???", n);
412
413	s = channel_open_message();
414	debug3("channel %d: status: %s", c->self, s);
415	free(s);
416
417	if (c->sock != -1)
418		shutdown(c->sock, SHUT_RDWR);
419	channel_close_fds(c);
420	buffer_free(&c->input);
421	buffer_free(&c->output);
422	buffer_free(&c->extended);
423	free(c->remote_name);
424	c->remote_name = NULL;
425	free(c->path);
426	c->path = NULL;
427	free(c->listening_addr);
428	c->listening_addr = NULL;
429	while ((cc = TAILQ_FIRST(&c->status_confirms)) != NULL) {
430		if (cc->abandon_cb != NULL)
431			cc->abandon_cb(c, cc->ctx);
432		TAILQ_REMOVE(&c->status_confirms, cc, entry);
433		bzero(cc, sizeof(*cc));
434		free(cc);
435	}
436	if (c->filter_cleanup != NULL && c->filter_ctx != NULL)
437		c->filter_cleanup(c->self, c->filter_ctx);
438	channels[c->self] = NULL;
439	free(c);
440}
441
442void
443channel_free_all(void)
444{
445	u_int i;
446
447	for (i = 0; i < channels_alloc; i++)
448		if (channels[i] != NULL)
449			channel_free(channels[i]);
450}
451
452/*
453 * Closes the sockets/fds of all channels.  This is used to close extra file
454 * descriptors after a fork.
455 */
456void
457channel_close_all(void)
458{
459	u_int i;
460
461	for (i = 0; i < channels_alloc; i++)
462		if (channels[i] != NULL)
463			channel_close_fds(channels[i]);
464}
465
466/*
467 * Stop listening to channels.
468 */
469void
470channel_stop_listening(void)
471{
472	u_int i;
473	Channel *c;
474
475	for (i = 0; i < channels_alloc; i++) {
476		c = channels[i];
477		if (c != NULL) {
478			switch (c->type) {
479			case SSH_CHANNEL_AUTH_SOCKET:
480			case SSH_CHANNEL_PORT_LISTENER:
481			case SSH_CHANNEL_RPORT_LISTENER:
482			case SSH_CHANNEL_X11_LISTENER:
483				channel_close_fd(&c->sock);
484				channel_free(c);
485				break;
486			}
487		}
488	}
489}
490
491/*
492 * Returns true if no channel has too much buffered data, and false if one or
493 * more channel is overfull.
494 */
495int
496channel_not_very_much_buffered_data(void)
497{
498	u_int i;
499	Channel *c;
500
501	for (i = 0; i < channels_alloc; i++) {
502		c = channels[i];
503		if (c != NULL && c->type == SSH_CHANNEL_OPEN) {
504#if 0
505			if (!compat20 &&
506			    buffer_len(&c->input) > packet_get_maxsize()) {
507				debug2("channel %d: big input buffer %d",
508				    c->self, buffer_len(&c->input));
509				return 0;
510			}
511#endif
512			if (buffer_len(&c->output) > packet_get_maxsize()) {
513				debug2("channel %d: big output buffer %u > %u",
514				    c->self, buffer_len(&c->output),
515				    packet_get_maxsize());
516				return 0;
517			}
518		}
519	}
520	return 1;
521}
522
523/* Returns true if any channel is still open. */
524int
525channel_still_open(void)
526{
527	u_int i;
528	Channel *c;
529
530	for (i = 0; i < channels_alloc; i++) {
531		c = channels[i];
532		if (c == NULL)
533			continue;
534		switch (c->type) {
535		case SSH_CHANNEL_X11_LISTENER:
536		case SSH_CHANNEL_PORT_LISTENER:
537		case SSH_CHANNEL_RPORT_LISTENER:
538		case SSH_CHANNEL_MUX_LISTENER:
539		case SSH_CHANNEL_CLOSED:
540		case SSH_CHANNEL_AUTH_SOCKET:
541		case SSH_CHANNEL_DYNAMIC:
542		case SSH_CHANNEL_CONNECTING:
543		case SSH_CHANNEL_ZOMBIE:
544		case SSH_CHANNEL_ABANDONED:
545			continue;
546		case SSH_CHANNEL_LARVAL:
547			if (!compat20)
548				fatal("cannot happen: SSH_CHANNEL_LARVAL");
549			continue;
550		case SSH_CHANNEL_OPENING:
551		case SSH_CHANNEL_OPEN:
552		case SSH_CHANNEL_X11_OPEN:
553		case SSH_CHANNEL_MUX_CLIENT:
554			return 1;
555		case SSH_CHANNEL_INPUT_DRAINING:
556		case SSH_CHANNEL_OUTPUT_DRAINING:
557			if (!compat13)
558				fatal("cannot happen: OUT_DRAIN");
559			return 1;
560		default:
561			fatal("channel_still_open: bad channel type %d", c->type);
562			/* NOTREACHED */
563		}
564	}
565	return 0;
566}
567
568/* Returns the id of an open channel suitable for keepaliving */
569int
570channel_find_open(void)
571{
572	u_int i;
573	Channel *c;
574
575	for (i = 0; i < channels_alloc; i++) {
576		c = channels[i];
577		if (c == NULL || c->remote_id < 0)
578			continue;
579		switch (c->type) {
580		case SSH_CHANNEL_CLOSED:
581		case SSH_CHANNEL_DYNAMIC:
582		case SSH_CHANNEL_X11_LISTENER:
583		case SSH_CHANNEL_PORT_LISTENER:
584		case SSH_CHANNEL_RPORT_LISTENER:
585		case SSH_CHANNEL_MUX_LISTENER:
586		case SSH_CHANNEL_MUX_CLIENT:
587		case SSH_CHANNEL_OPENING:
588		case SSH_CHANNEL_CONNECTING:
589		case SSH_CHANNEL_ZOMBIE:
590		case SSH_CHANNEL_ABANDONED:
591			continue;
592		case SSH_CHANNEL_LARVAL:
593		case SSH_CHANNEL_AUTH_SOCKET:
594		case SSH_CHANNEL_OPEN:
595		case SSH_CHANNEL_X11_OPEN:
596			return i;
597		case SSH_CHANNEL_INPUT_DRAINING:
598		case SSH_CHANNEL_OUTPUT_DRAINING:
599			if (!compat13)
600				fatal("cannot happen: OUT_DRAIN");
601			return i;
602		default:
603			fatal("channel_find_open: bad channel type %d", c->type);
604			/* NOTREACHED */
605		}
606	}
607	return -1;
608}
609
610
611/*
612 * Returns a message describing the currently open forwarded connections,
613 * suitable for sending to the client.  The message contains crlf pairs for
614 * newlines.
615 */
616char *
617channel_open_message(void)
618{
619	Buffer buffer;
620	Channel *c;
621	char buf[1024], *cp;
622	u_int i;
623
624	buffer_init(&buffer);
625	snprintf(buf, sizeof buf, "The following connections are open:\r\n");
626	buffer_append(&buffer, buf, strlen(buf));
627	for (i = 0; i < channels_alloc; i++) {
628		c = channels[i];
629		if (c == NULL)
630			continue;
631		switch (c->type) {
632		case SSH_CHANNEL_X11_LISTENER:
633		case SSH_CHANNEL_PORT_LISTENER:
634		case SSH_CHANNEL_RPORT_LISTENER:
635		case SSH_CHANNEL_CLOSED:
636		case SSH_CHANNEL_AUTH_SOCKET:
637		case SSH_CHANNEL_ZOMBIE:
638		case SSH_CHANNEL_ABANDONED:
639		case SSH_CHANNEL_MUX_CLIENT:
640		case SSH_CHANNEL_MUX_LISTENER:
641			continue;
642		case SSH_CHANNEL_LARVAL:
643		case SSH_CHANNEL_OPENING:
644		case SSH_CHANNEL_CONNECTING:
645		case SSH_CHANNEL_DYNAMIC:
646		case SSH_CHANNEL_OPEN:
647		case SSH_CHANNEL_X11_OPEN:
648		case SSH_CHANNEL_INPUT_DRAINING:
649		case SSH_CHANNEL_OUTPUT_DRAINING:
650			snprintf(buf, sizeof buf,
651			    "  #%d %.300s (t%d r%d i%d/%d o%d/%d fd %d/%d cc %d)\r\n",
652			    c->self, c->remote_name,
653			    c->type, c->remote_id,
654			    c->istate, buffer_len(&c->input),
655			    c->ostate, buffer_len(&c->output),
656			    c->rfd, c->wfd, c->ctl_chan);
657			buffer_append(&buffer, buf, strlen(buf));
658			continue;
659		default:
660			fatal("channel_open_message: bad channel type %d", c->type);
661			/* NOTREACHED */
662		}
663	}
664	buffer_append(&buffer, "\0", 1);
665	cp = xstrdup(buffer_ptr(&buffer));
666	buffer_free(&buffer);
667	return cp;
668}
669
670void
671channel_send_open(int id)
672{
673	Channel *c = channel_lookup(id);
674
675	if (c == NULL) {
676		logit("channel_send_open: %d: bad id", id);
677		return;
678	}
679	debug2("channel %d: send open", id);
680	packet_start(SSH2_MSG_CHANNEL_OPEN);
681	packet_put_cstring(c->ctype);
682	packet_put_int(c->self);
683	packet_put_int(c->local_window);
684	packet_put_int(c->local_maxpacket);
685	packet_send();
686}
687
688void
689channel_request_start(int id, char *service, int wantconfirm)
690{
691	Channel *c = channel_lookup(id);
692
693	if (c == NULL) {
694		logit("channel_request_start: %d: unknown channel id", id);
695		return;
696	}
697	debug2("channel %d: request %s confirm %d", id, service, wantconfirm);
698	packet_start(SSH2_MSG_CHANNEL_REQUEST);
699	packet_put_int(c->remote_id);
700	packet_put_cstring(service);
701	packet_put_char(wantconfirm);
702}
703
704void
705channel_register_status_confirm(int id, channel_confirm_cb *cb,
706    channel_confirm_abandon_cb *abandon_cb, void *ctx)
707{
708	struct channel_confirm *cc;
709	Channel *c;
710
711	if ((c = channel_lookup(id)) == NULL)
712		fatal("channel_register_expect: %d: bad id", id);
713
714	cc = xcalloc(1, sizeof(*cc));
715	cc->cb = cb;
716	cc->abandon_cb = abandon_cb;
717	cc->ctx = ctx;
718	TAILQ_INSERT_TAIL(&c->status_confirms, cc, entry);
719}
720
721void
722channel_register_open_confirm(int id, channel_open_fn *fn, void *ctx)
723{
724	Channel *c = channel_lookup(id);
725
726	if (c == NULL) {
727		logit("channel_register_open_confirm: %d: bad id", id);
728		return;
729	}
730	c->open_confirm = fn;
731	c->open_confirm_ctx = ctx;
732}
733
734void
735channel_register_cleanup(int id, channel_callback_fn *fn, int do_close)
736{
737	Channel *c = channel_by_id(id);
738
739	if (c == NULL) {
740		logit("channel_register_cleanup: %d: bad id", id);
741		return;
742	}
743	c->detach_user = fn;
744	c->detach_close = do_close;
745}
746
747void
748channel_cancel_cleanup(int id)
749{
750	Channel *c = channel_by_id(id);
751
752	if (c == NULL) {
753		logit("channel_cancel_cleanup: %d: bad id", id);
754		return;
755	}
756	c->detach_user = NULL;
757	c->detach_close = 0;
758}
759
760void
761channel_register_filter(int id, channel_infilter_fn *ifn,
762    channel_outfilter_fn *ofn, channel_filter_cleanup_fn *cfn, void *ctx)
763{
764	Channel *c = channel_lookup(id);
765
766	if (c == NULL) {
767		logit("channel_register_filter: %d: bad id", id);
768		return;
769	}
770	c->input_filter = ifn;
771	c->output_filter = ofn;
772	c->filter_ctx = ctx;
773	c->filter_cleanup = cfn;
774}
775
776void
777channel_set_fds(int id, int rfd, int wfd, int efd,
778    int extusage, int nonblock, int is_tty, u_int window_max)
779{
780	Channel *c = channel_lookup(id);
781
782	if (c == NULL || c->type != SSH_CHANNEL_LARVAL)
783		fatal("channel_activate for non-larval channel %d.", id);
784	channel_register_fds(c, rfd, wfd, efd, extusage, nonblock, is_tty);
785	c->type = SSH_CHANNEL_OPEN;
786	c->local_window = c->local_window_max = window_max;
787	packet_start(SSH2_MSG_CHANNEL_WINDOW_ADJUST);
788	packet_put_int(c->remote_id);
789	packet_put_int(c->local_window);
790	packet_send();
791}
792
793/*
794 * 'channel_pre*' are called just before select() to add any bits relevant to
795 * channels in the select bitmasks.
796 */
797/*
798 * 'channel_post*': perform any appropriate operations for channels which
799 * have events pending.
800 */
801typedef void chan_fn(Channel *c, fd_set *readset, fd_set *writeset);
802chan_fn *channel_pre[SSH_CHANNEL_MAX_TYPE];
803chan_fn *channel_post[SSH_CHANNEL_MAX_TYPE];
804
805/* ARGSUSED */
806static void
807channel_pre_listener(Channel *c, fd_set *readset, fd_set *writeset)
808{
809	FD_SET(c->sock, readset);
810}
811
812/* ARGSUSED */
813static void
814channel_pre_connecting(Channel *c, fd_set *readset, fd_set *writeset)
815{
816	debug3("channel %d: waiting for connection", c->self);
817	FD_SET(c->sock, writeset);
818}
819
820static void
821channel_pre_open_13(Channel *c, fd_set *readset, fd_set *writeset)
822{
823	if (buffer_len(&c->input) < packet_get_maxsize())
824		FD_SET(c->sock, readset);
825	if (buffer_len(&c->output) > 0)
826		FD_SET(c->sock, writeset);
827}
828
829static u_int
830channel_tcpwinsz(void)
831{
832	u_int32_t tcpwinsz;
833	socklen_t optsz;
834	int ret, sd;
835	u_int maxlen;
836
837	/* If we are not on a socket return 128KB. */
838	if (!packet_connection_is_on_socket())
839		return (128 * 1024);
840
841	tcpwinsz = 0;
842	optsz = sizeof(tcpwinsz);
843	sd = packet_get_connection_in();
844	ret = getsockopt(sd, SOL_SOCKET, SO_RCVBUF, &tcpwinsz, &optsz);
845
846	/* Return no more than the maximum buffer size. */
847	maxlen = buffer_get_max_len();
848	if ((ret == 0) && tcpwinsz > maxlen)
849		tcpwinsz = maxlen;
850	/* In case getsockopt() failed return a minimum. */
851	if (tcpwinsz == 0)
852		tcpwinsz = CHAN_TCP_WINDOW_DEFAULT;
853	debug2("tcpwinsz: %d for connection: %d", tcpwinsz, sd);
854	return (tcpwinsz);
855}
856
857static void
858channel_pre_open(Channel *c, fd_set *readset, fd_set *writeset)
859{
860	u_int limit;
861
862	/* Check buffer limits. */
863	if (!c->tcpwinsz || c->dynamic_window > 0)
864		c->tcpwinsz = channel_tcpwinsz();
865
866	limit = MIN(compat20 ? c->remote_window : packet_get_maxsize(),
867	    2 * c->tcpwinsz);
868
869	if (c->istate == CHAN_INPUT_OPEN &&
870	    limit > 0 &&
871	    buffer_len(&c->input) < limit &&
872	    buffer_check_alloc(&c->input, CHAN_RBUF))
873		FD_SET(c->rfd, readset);
874	if (c->ostate == CHAN_OUTPUT_OPEN ||
875	    c->ostate == CHAN_OUTPUT_WAIT_DRAIN) {
876		if (buffer_len(&c->output) > 0) {
877			FD_SET(c->wfd, writeset);
878		} else if (c->ostate == CHAN_OUTPUT_WAIT_DRAIN) {
879			if (CHANNEL_EFD_OUTPUT_ACTIVE(c))
880				debug2("channel %d: obuf_empty delayed efd %d/(%d)",
881				    c->self, c->efd, buffer_len(&c->extended));
882			else
883				chan_obuf_empty(c);
884		}
885	}
886	/** XXX check close conditions, too */
887	if (compat20 && c->efd != -1 &&
888	    !(c->istate == CHAN_INPUT_CLOSED && c->ostate == CHAN_OUTPUT_CLOSED)) {
889		if (c->extended_usage == CHAN_EXTENDED_WRITE &&
890		    buffer_len(&c->extended) > 0)
891			FD_SET(c->efd, writeset);
892		else if (c->efd != -1 && !(c->flags & CHAN_EOF_SENT) &&
893		    (c->extended_usage == CHAN_EXTENDED_READ ||
894		    c->extended_usage == CHAN_EXTENDED_IGNORE) &&
895		    buffer_len(&c->extended) < c->remote_window)
896			FD_SET(c->efd, readset);
897	}
898	/* XXX: What about efd? races? */
899}
900
901/* ARGSUSED */
902static void
903channel_pre_input_draining(Channel *c, fd_set *readset, fd_set *writeset)
904{
905	if (buffer_len(&c->input) == 0) {
906		packet_start(SSH_MSG_CHANNEL_CLOSE);
907		packet_put_int(c->remote_id);
908		packet_send();
909		c->type = SSH_CHANNEL_CLOSED;
910		debug2("channel %d: closing after input drain.", c->self);
911	}
912}
913
914/* ARGSUSED */
915static void
916channel_pre_output_draining(Channel *c, fd_set *readset, fd_set *writeset)
917{
918	if (buffer_len(&c->output) == 0)
919		chan_mark_dead(c);
920	else
921		FD_SET(c->sock, writeset);
922}
923
924/*
925 * This is a special state for X11 authentication spoofing.  An opened X11
926 * connection (when authentication spoofing is being done) remains in this
927 * state until the first packet has been completely read.  The authentication
928 * data in that packet is then substituted by the real data if it matches the
929 * fake data, and the channel is put into normal mode.
930 * XXX All this happens at the client side.
931 * Returns: 0 = need more data, -1 = wrong cookie, 1 = ok
932 */
933static int
934x11_open_helper(Buffer *b)
935{
936	u_char *ucp;
937	u_int proto_len, data_len;
938
939	/* Check if the fixed size part of the packet is in buffer. */
940	if (buffer_len(b) < 12)
941		return 0;
942
943	/* Parse the lengths of variable-length fields. */
944	ucp = buffer_ptr(b);
945	if (ucp[0] == 0x42) {	/* Byte order MSB first. */
946		proto_len = 256 * ucp[6] + ucp[7];
947		data_len = 256 * ucp[8] + ucp[9];
948	} else if (ucp[0] == 0x6c) {	/* Byte order LSB first. */
949		proto_len = ucp[6] + 256 * ucp[7];
950		data_len = ucp[8] + 256 * ucp[9];
951	} else {
952		debug2("Initial X11 packet contains bad byte order byte: 0x%x",
953		    ucp[0]);
954		return -1;
955	}
956
957	/* Check if the whole packet is in buffer. */
958	if (buffer_len(b) <
959	    12 + ((proto_len + 3) & ~3) + ((data_len + 3) & ~3))
960		return 0;
961
962	/* Check if authentication protocol matches. */
963	if (proto_len != strlen(x11_saved_proto) ||
964	    memcmp(ucp + 12, x11_saved_proto, proto_len) != 0) {
965		debug2("X11 connection uses different authentication protocol.");
966		return -1;
967	}
968	/* Check if authentication data matches our fake data. */
969	if (data_len != x11_fake_data_len ||
970	    timingsafe_bcmp(ucp + 12 + ((proto_len + 3) & ~3),
971		x11_fake_data, x11_fake_data_len) != 0) {
972		debug2("X11 auth data does not match fake data.");
973		return -1;
974	}
975	/* Check fake data length */
976	if (x11_fake_data_len != x11_saved_data_len) {
977		error("X11 fake_data_len %d != saved_data_len %d",
978		    x11_fake_data_len, x11_saved_data_len);
979		return -1;
980	}
981	/*
982	 * Received authentication protocol and data match
983	 * our fake data. Substitute the fake data with real
984	 * data.
985	 */
986	memcpy(ucp + 12 + ((proto_len + 3) & ~3),
987	    x11_saved_data, x11_saved_data_len);
988	return 1;
989}
990
991static void
992channel_pre_x11_open_13(Channel *c, fd_set *readset, fd_set *writeset)
993{
994	int ret = x11_open_helper(&c->output);
995
996	if (ret == 1) {
997		/* Start normal processing for the channel. */
998		c->type = SSH_CHANNEL_OPEN;
999		channel_pre_open_13(c, readset, writeset);
1000	} else if (ret == -1) {
1001		/*
1002		 * We have received an X11 connection that has bad
1003		 * authentication information.
1004		 */
1005		logit("X11 connection rejected because of wrong authentication.");
1006		buffer_clear(&c->input);
1007		buffer_clear(&c->output);
1008		channel_close_fd(&c->sock);
1009		c->sock = -1;
1010		c->type = SSH_CHANNEL_CLOSED;
1011		packet_start(SSH_MSG_CHANNEL_CLOSE);
1012		packet_put_int(c->remote_id);
1013		packet_send();
1014	}
1015}
1016
1017static void
1018channel_pre_x11_open(Channel *c, fd_set *readset, fd_set *writeset)
1019{
1020	int ret = x11_open_helper(&c->output);
1021
1022	/* c->force_drain = 1; */
1023
1024	if (ret == 1) {
1025		c->type = SSH_CHANNEL_OPEN;
1026		channel_pre_open(c, readset, writeset);
1027	} else if (ret == -1) {
1028		logit("X11 connection rejected because of wrong authentication.");
1029		debug2("X11 rejected %d i%d/o%d", c->self, c->istate, c->ostate);
1030		chan_read_failed(c);
1031		buffer_clear(&c->input);
1032		chan_ibuf_empty(c);
1033		buffer_clear(&c->output);
1034		/* for proto v1, the peer will send an IEOF */
1035		if (compat20)
1036			chan_write_failed(c);
1037		else
1038			c->type = SSH_CHANNEL_OPEN;
1039		debug2("X11 closed %d i%d/o%d", c->self, c->istate, c->ostate);
1040	}
1041}
1042
1043static void
1044channel_pre_mux_client(Channel *c, fd_set *readset, fd_set *writeset)
1045{
1046	if (c->istate == CHAN_INPUT_OPEN && !c->mux_pause &&
1047	    buffer_check_alloc(&c->input, CHAN_RBUF))
1048		FD_SET(c->rfd, readset);
1049	if (c->istate == CHAN_INPUT_WAIT_DRAIN) {
1050		/* clear buffer immediately (discard any partial packet) */
1051		buffer_clear(&c->input);
1052		chan_ibuf_empty(c);
1053		/* Start output drain. XXX just kill chan? */
1054		chan_rcvd_oclose(c);
1055	}
1056	if (c->ostate == CHAN_OUTPUT_OPEN ||
1057	    c->ostate == CHAN_OUTPUT_WAIT_DRAIN) {
1058		if (buffer_len(&c->output) > 0)
1059			FD_SET(c->wfd, writeset);
1060		else if (c->ostate == CHAN_OUTPUT_WAIT_DRAIN)
1061			chan_obuf_empty(c);
1062	}
1063}
1064
1065/* try to decode a socks4 header */
1066/* ARGSUSED */
1067static int
1068channel_decode_socks4(Channel *c, fd_set *readset, fd_set *writeset)
1069{
1070	char *p, *host;
1071	u_int len, have, i, found, need;
1072	char username[256];
1073	struct {
1074		u_int8_t version;
1075		u_int8_t command;
1076		u_int16_t dest_port;
1077		struct in_addr dest_addr;
1078	} s4_req, s4_rsp;
1079
1080	debug2("channel %d: decode socks4", c->self);
1081
1082	have = buffer_len(&c->input);
1083	len = sizeof(s4_req);
1084	if (have < len)
1085		return 0;
1086	p = buffer_ptr(&c->input);
1087
1088	need = 1;
1089	/* SOCKS4A uses an invalid IP address 0.0.0.x */
1090	if (p[4] == 0 && p[5] == 0 && p[6] == 0 && p[7] != 0) {
1091		debug2("channel %d: socks4a request", c->self);
1092		/* ... and needs an extra string (the hostname) */
1093		need = 2;
1094	}
1095	/* Check for terminating NUL on the string(s) */
1096	for (found = 0, i = len; i < have; i++) {
1097		if (p[i] == '\0') {
1098			found++;
1099			if (found == need)
1100				break;
1101		}
1102		if (i > 1024) {
1103			/* the peer is probably sending garbage */
1104			debug("channel %d: decode socks4: too long",
1105			    c->self);
1106			return -1;
1107		}
1108	}
1109	if (found < need)
1110		return 0;
1111	buffer_get(&c->input, (char *)&s4_req.version, 1);
1112	buffer_get(&c->input, (char *)&s4_req.command, 1);
1113	buffer_get(&c->input, (char *)&s4_req.dest_port, 2);
1114	buffer_get(&c->input, (char *)&s4_req.dest_addr, 4);
1115	have = buffer_len(&c->input);
1116	p = buffer_ptr(&c->input);
1117	len = strlen(p);
1118	debug2("channel %d: decode socks4: user %s/%d", c->self, p, len);
1119	len++;					/* trailing '\0' */
1120	if (len > have)
1121		fatal("channel %d: decode socks4: len %d > have %d",
1122		    c->self, len, have);
1123	strlcpy(username, p, sizeof(username));
1124	buffer_consume(&c->input, len);
1125
1126	free(c->path);
1127	c->path = NULL;
1128	if (need == 1) {			/* SOCKS4: one string */
1129		host = inet_ntoa(s4_req.dest_addr);
1130		c->path = xstrdup(host);
1131	} else {				/* SOCKS4A: two strings */
1132		have = buffer_len(&c->input);
1133		p = buffer_ptr(&c->input);
1134		len = strlen(p);
1135		debug2("channel %d: decode socks4a: host %s/%d",
1136		    c->self, p, len);
1137		len++;				/* trailing '\0' */
1138		if (len > have)
1139			fatal("channel %d: decode socks4a: len %d > have %d",
1140			    c->self, len, have);
1141		if (len > NI_MAXHOST) {
1142			error("channel %d: hostname \"%.100s\" too long",
1143			    c->self, p);
1144			return -1;
1145		}
1146		c->path = xstrdup(p);
1147		buffer_consume(&c->input, len);
1148	}
1149	c->host_port = ntohs(s4_req.dest_port);
1150
1151	debug2("channel %d: dynamic request: socks4 host %s port %u command %u",
1152	    c->self, c->path, c->host_port, s4_req.command);
1153
1154	if (s4_req.command != 1) {
1155		debug("channel %d: cannot handle: %s cn %d",
1156		    c->self, need == 1 ? "SOCKS4" : "SOCKS4A", s4_req.command);
1157		return -1;
1158	}
1159	s4_rsp.version = 0;			/* vn: 0 for reply */
1160	s4_rsp.command = 90;			/* cd: req granted */
1161	s4_rsp.dest_port = 0;			/* ignored */
1162	s4_rsp.dest_addr.s_addr = INADDR_ANY;	/* ignored */
1163	buffer_append(&c->output, &s4_rsp, sizeof(s4_rsp));
1164	return 1;
1165}
1166
1167/* try to decode a socks5 header */
1168#define SSH_SOCKS5_AUTHDONE	0x1000
1169#define SSH_SOCKS5_NOAUTH	0x00
1170#define SSH_SOCKS5_IPV4		0x01
1171#define SSH_SOCKS5_DOMAIN	0x03
1172#define SSH_SOCKS5_IPV6		0x04
1173#define SSH_SOCKS5_CONNECT	0x01
1174#define SSH_SOCKS5_SUCCESS	0x00
1175
1176/* ARGSUSED */
1177static int
1178channel_decode_socks5(Channel *c, fd_set *readset, fd_set *writeset)
1179{
1180	struct {
1181		u_int8_t version;
1182		u_int8_t command;
1183		u_int8_t reserved;
1184		u_int8_t atyp;
1185	} s5_req, s5_rsp;
1186	u_int16_t dest_port;
1187	char dest_addr[255+1], ntop[INET6_ADDRSTRLEN];
1188	u_char *p;
1189	u_int have, need, i, found, nmethods, addrlen, af;
1190
1191	debug2("channel %d: decode socks5", c->self);
1192	p = buffer_ptr(&c->input);
1193	if (p[0] != 0x05)
1194		return -1;
1195	have = buffer_len(&c->input);
1196	if (!(c->flags & SSH_SOCKS5_AUTHDONE)) {
1197		/* format: ver | nmethods | methods */
1198		if (have < 2)
1199			return 0;
1200		nmethods = p[1];
1201		if (have < nmethods + 2)
1202			return 0;
1203		/* look for method: "NO AUTHENTICATION REQUIRED" */
1204		for (found = 0, i = 2; i < nmethods + 2; i++) {
1205			if (p[i] == SSH_SOCKS5_NOAUTH) {
1206				found = 1;
1207				break;
1208			}
1209		}
1210		if (!found) {
1211			debug("channel %d: method SSH_SOCKS5_NOAUTH not found",
1212			    c->self);
1213			return -1;
1214		}
1215		buffer_consume(&c->input, nmethods + 2);
1216		buffer_put_char(&c->output, 0x05);		/* version */
1217		buffer_put_char(&c->output, SSH_SOCKS5_NOAUTH);	/* method */
1218		FD_SET(c->sock, writeset);
1219		c->flags |= SSH_SOCKS5_AUTHDONE;
1220		debug2("channel %d: socks5 auth done", c->self);
1221		return 0;				/* need more */
1222	}
1223	debug2("channel %d: socks5 post auth", c->self);
1224	if (have < sizeof(s5_req)+1)
1225		return 0;			/* need more */
1226	memcpy(&s5_req, p, sizeof(s5_req));
1227	if (s5_req.version != 0x05 ||
1228	    s5_req.command != SSH_SOCKS5_CONNECT ||
1229	    s5_req.reserved != 0x00) {
1230		debug2("channel %d: only socks5 connect supported", c->self);
1231		return -1;
1232	}
1233	switch (s5_req.atyp){
1234	case SSH_SOCKS5_IPV4:
1235		addrlen = 4;
1236		af = AF_INET;
1237		break;
1238	case SSH_SOCKS5_DOMAIN:
1239		addrlen = p[sizeof(s5_req)];
1240		af = -1;
1241		break;
1242	case SSH_SOCKS5_IPV6:
1243		addrlen = 16;
1244		af = AF_INET6;
1245		break;
1246	default:
1247		debug2("channel %d: bad socks5 atyp %d", c->self, s5_req.atyp);
1248		return -1;
1249	}
1250	need = sizeof(s5_req) + addrlen + 2;
1251	if (s5_req.atyp == SSH_SOCKS5_DOMAIN)
1252		need++;
1253	if (have < need)
1254		return 0;
1255	buffer_consume(&c->input, sizeof(s5_req));
1256	if (s5_req.atyp == SSH_SOCKS5_DOMAIN)
1257		buffer_consume(&c->input, 1);    /* host string length */
1258	buffer_get(&c->input, &dest_addr, addrlen);
1259	buffer_get(&c->input, (char *)&dest_port, 2);
1260	dest_addr[addrlen] = '\0';
1261	free(c->path);
1262	c->path = NULL;
1263	if (s5_req.atyp == SSH_SOCKS5_DOMAIN) {
1264		if (addrlen >= NI_MAXHOST) {
1265			error("channel %d: dynamic request: socks5 hostname "
1266			    "\"%.100s\" too long", c->self, dest_addr);
1267			return -1;
1268		}
1269		c->path = xstrdup(dest_addr);
1270	} else {
1271		if (inet_ntop(af, dest_addr, ntop, sizeof(ntop)) == NULL)
1272			return -1;
1273		c->path = xstrdup(ntop);
1274	}
1275	c->host_port = ntohs(dest_port);
1276
1277	debug2("channel %d: dynamic request: socks5 host %s port %u command %u",
1278	    c->self, c->path, c->host_port, s5_req.command);
1279
1280	s5_rsp.version = 0x05;
1281	s5_rsp.command = SSH_SOCKS5_SUCCESS;
1282	s5_rsp.reserved = 0;			/* ignored */
1283	s5_rsp.atyp = SSH_SOCKS5_IPV4;
1284	dest_port = 0;				/* ignored */
1285
1286	buffer_append(&c->output, &s5_rsp, sizeof(s5_rsp));
1287	buffer_put_int(&c->output, ntohl(INADDR_ANY)); /* bind address */
1288	buffer_append(&c->output, &dest_port, sizeof(dest_port));
1289	return 1;
1290}
1291
1292Channel *
1293channel_connect_stdio_fwd(const char *host_to_connect, u_short port_to_connect,
1294    int in, int out)
1295{
1296	Channel *c;
1297
1298	debug("channel_connect_stdio_fwd %s:%d", host_to_connect,
1299	    port_to_connect);
1300
1301	c = channel_new("stdio-forward", SSH_CHANNEL_OPENING, in, out,
1302	    -1, CHAN_TCP_WINDOW_DEFAULT, CHAN_TCP_PACKET_DEFAULT,
1303	    0, "stdio-forward", /*nonblock*/0);
1304
1305	c->path = xstrdup(host_to_connect);
1306	c->host_port = port_to_connect;
1307	c->listening_port = 0;
1308	c->force_drain = 1;
1309
1310	channel_register_fds(c, in, out, -1, 0, 1, 0);
1311	port_open_helper(c, "direct-tcpip");
1312
1313	return c;
1314}
1315
1316/* dynamic port forwarding */
1317static void
1318channel_pre_dynamic(Channel *c, fd_set *readset, fd_set *writeset)
1319{
1320	u_char *p;
1321	u_int have;
1322	int ret;
1323
1324	have = buffer_len(&c->input);
1325	debug2("channel %d: pre_dynamic: have %d", c->self, have);
1326	/* buffer_dump(&c->input); */
1327	/* check if the fixed size part of the packet is in buffer. */
1328	if (have < 3) {
1329		/* need more */
1330		FD_SET(c->sock, readset);
1331		return;
1332	}
1333	/* try to guess the protocol */
1334	p = buffer_ptr(&c->input);
1335	switch (p[0]) {
1336	case 0x04:
1337		ret = channel_decode_socks4(c, readset, writeset);
1338		break;
1339	case 0x05:
1340		ret = channel_decode_socks5(c, readset, writeset);
1341		break;
1342	default:
1343		ret = -1;
1344		break;
1345	}
1346	if (ret < 0) {
1347		chan_mark_dead(c);
1348	} else if (ret == 0) {
1349		debug2("channel %d: pre_dynamic: need more", c->self);
1350		/* need more */
1351		FD_SET(c->sock, readset);
1352	} else {
1353		/* switch to the next state */
1354		c->type = SSH_CHANNEL_OPENING;
1355		port_open_helper(c, "direct-tcpip");
1356	}
1357}
1358
1359/* This is our fake X11 server socket. */
1360/* ARGSUSED */
1361static void
1362channel_post_x11_listener(Channel *c, fd_set *readset, fd_set *writeset)
1363{
1364	Channel *nc;
1365	struct sockaddr_storage addr;
1366	int newsock, oerrno;
1367	socklen_t addrlen;
1368	char buf[16384], *remote_ipaddr;
1369	int remote_port;
1370
1371	if (FD_ISSET(c->sock, readset)) {
1372		debug("X11 connection requested.");
1373		addrlen = sizeof(addr);
1374		newsock = accept(c->sock, (struct sockaddr *)&addr, &addrlen);
1375		if (c->single_connection) {
1376			oerrno = errno;
1377			debug2("single_connection: closing X11 listener.");
1378			channel_close_fd(&c->sock);
1379			chan_mark_dead(c);
1380			errno = oerrno;
1381		}
1382		if (newsock < 0) {
1383			if (errno != EINTR && errno != EWOULDBLOCK &&
1384			    errno != ECONNABORTED)
1385				error("accept: %.100s", strerror(errno));
1386			if (errno == EMFILE || errno == ENFILE)
1387				c->notbefore = monotime() + 1;
1388			return;
1389		}
1390		set_nodelay(newsock);
1391		remote_ipaddr = get_peer_ipaddr(newsock);
1392		remote_port = get_peer_port(newsock);
1393		snprintf(buf, sizeof buf, "X11 connection from %.200s port %d",
1394		    remote_ipaddr, remote_port);
1395
1396		nc = channel_new("accepted x11 socket",
1397		    SSH_CHANNEL_OPENING, newsock, newsock, -1,
1398		    c->local_window_max, c->local_maxpacket, 0, buf, 1);
1399		if (compat20) {
1400			packet_start(SSH2_MSG_CHANNEL_OPEN);
1401			packet_put_cstring("x11");
1402			packet_put_int(nc->self);
1403			packet_put_int(nc->local_window_max);
1404			packet_put_int(nc->local_maxpacket);
1405			/* originator ipaddr and port */
1406			packet_put_cstring(remote_ipaddr);
1407			if (datafellows & SSH_BUG_X11FWD) {
1408				debug2("ssh2 x11 bug compat mode");
1409			} else {
1410				packet_put_int(remote_port);
1411			}
1412			packet_send();
1413		} else {
1414			packet_start(SSH_SMSG_X11_OPEN);
1415			packet_put_int(nc->self);
1416			if (packet_get_protocol_flags() &
1417			    SSH_PROTOFLAG_HOST_IN_FWD_OPEN)
1418				packet_put_cstring(buf);
1419			packet_send();
1420		}
1421		free(remote_ipaddr);
1422	}
1423}
1424
1425static void
1426port_open_helper(Channel *c, char *rtype)
1427{
1428	int direct;
1429	char buf[1024];
1430	char *local_ipaddr = get_local_ipaddr(c->sock);
1431	int local_port = get_sock_port(c->sock, 1);
1432	char *remote_ipaddr = get_peer_ipaddr(c->sock);
1433	int remote_port = get_peer_port(c->sock);
1434
1435	if (remote_port == -1) {
1436		/* Fake addr/port to appease peers that validate it (Tectia) */
1437		free(remote_ipaddr);
1438		remote_ipaddr = xstrdup("127.0.0.1");
1439		remote_port = 65535;
1440	}
1441
1442	direct = (strcmp(rtype, "direct-tcpip") == 0);
1443
1444	snprintf(buf, sizeof buf,
1445	    "%s: listening port %d for %.100s port %d, "
1446	    "connect from %.200s port %d to %.100s port %d",
1447	    rtype, c->listening_port, c->path, c->host_port,
1448	    remote_ipaddr, remote_port, local_ipaddr, local_port);
1449
1450	free(c->remote_name);
1451	c->remote_name = xstrdup(buf);
1452
1453	if (compat20) {
1454		packet_start(SSH2_MSG_CHANNEL_OPEN);
1455		packet_put_cstring(rtype);
1456		packet_put_int(c->self);
1457		packet_put_int(c->local_window_max);
1458		packet_put_int(c->local_maxpacket);
1459		if (direct) {
1460			/* target host, port */
1461			packet_put_cstring(c->path);
1462			packet_put_int(c->host_port);
1463		} else {
1464			/* listen address, port */
1465			packet_put_cstring(c->path);
1466			packet_put_int(local_port);
1467		}
1468		/* originator host and port */
1469		packet_put_cstring(remote_ipaddr);
1470		packet_put_int((u_int)remote_port);
1471		packet_send();
1472	} else {
1473		packet_start(SSH_MSG_PORT_OPEN);
1474		packet_put_int(c->self);
1475		packet_put_cstring(c->path);
1476		packet_put_int(c->host_port);
1477		if (packet_get_protocol_flags() &
1478		    SSH_PROTOFLAG_HOST_IN_FWD_OPEN)
1479			packet_put_cstring(c->remote_name);
1480		packet_send();
1481	}
1482	free(remote_ipaddr);
1483	free(local_ipaddr);
1484}
1485
1486static void
1487channel_set_reuseaddr(int fd)
1488{
1489	int on = 1;
1490
1491	/*
1492	 * Set socket options.
1493	 * Allow local port reuse in TIME_WAIT.
1494	 */
1495	if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) == -1)
1496		error("setsockopt SO_REUSEADDR fd %d: %s", fd, strerror(errno));
1497}
1498
1499/*
1500 * This socket is listening for connections to a forwarded TCP/IP port.
1501 */
1502/* ARGSUSED */
1503static void
1504channel_post_port_listener(Channel *c, fd_set *readset, fd_set *writeset)
1505{
1506	Channel *nc;
1507	struct sockaddr_storage addr;
1508	int newsock, nextstate;
1509	socklen_t addrlen;
1510	char *rtype;
1511
1512	if (FD_ISSET(c->sock, readset)) {
1513		debug("Connection to port %d forwarding "
1514		    "to %.100s port %d requested.",
1515		    c->listening_port, c->path, c->host_port);
1516
1517		if (c->type == SSH_CHANNEL_RPORT_LISTENER) {
1518			nextstate = SSH_CHANNEL_OPENING;
1519			rtype = "forwarded-tcpip";
1520		} else {
1521			if (c->host_port == 0) {
1522				nextstate = SSH_CHANNEL_DYNAMIC;
1523				rtype = "dynamic-tcpip";
1524			} else {
1525				nextstate = SSH_CHANNEL_OPENING;
1526				rtype = "direct-tcpip";
1527			}
1528		}
1529
1530		addrlen = sizeof(addr);
1531		newsock = accept(c->sock, (struct sockaddr *)&addr, &addrlen);
1532		if (newsock < 0) {
1533			if (errno != EINTR && errno != EWOULDBLOCK &&
1534			    errno != ECONNABORTED)
1535				error("accept: %.100s", strerror(errno));
1536			if (errno == EMFILE || errno == ENFILE)
1537				c->notbefore = monotime() + 1;
1538			return;
1539		}
1540		set_nodelay(newsock);
1541		nc = channel_new(rtype, nextstate, newsock, newsock, -1,
1542		    c->local_window_max, c->local_maxpacket, 0, rtype, 1);
1543		nc->listening_port = c->listening_port;
1544		nc->host_port = c->host_port;
1545		if (c->path != NULL)
1546			nc->path = xstrdup(c->path);
1547
1548		if (nextstate != SSH_CHANNEL_DYNAMIC)
1549			port_open_helper(nc, rtype);
1550	}
1551}
1552
1553/*
1554 * This is the authentication agent socket listening for connections from
1555 * clients.
1556 */
1557/* ARGSUSED */
1558static void
1559channel_post_auth_listener(Channel *c, fd_set *readset, fd_set *writeset)
1560{
1561	Channel *nc;
1562	int newsock;
1563	struct sockaddr_storage addr;
1564	socklen_t addrlen;
1565
1566	if (FD_ISSET(c->sock, readset)) {
1567		addrlen = sizeof(addr);
1568		newsock = accept(c->sock, (struct sockaddr *)&addr, &addrlen);
1569		if (newsock < 0) {
1570			error("accept from auth socket: %.100s",
1571			    strerror(errno));
1572			if (errno == EMFILE || errno == ENFILE)
1573				c->notbefore = monotime() + 1;
1574			return;
1575		}
1576		nc = channel_new("accepted auth socket",
1577		    SSH_CHANNEL_OPENING, newsock, newsock, -1,
1578		    c->local_window_max, c->local_maxpacket,
1579		    0, "accepted auth socket", 1);
1580		if (compat20) {
1581			packet_start(SSH2_MSG_CHANNEL_OPEN);
1582			packet_put_cstring("auth-agent@openssh.com");
1583			packet_put_int(nc->self);
1584			packet_put_int(c->local_window_max);
1585			packet_put_int(c->local_maxpacket);
1586		} else {
1587			packet_start(SSH_SMSG_AGENT_OPEN);
1588			packet_put_int(nc->self);
1589		}
1590		packet_send();
1591	}
1592}
1593
1594/* ARGSUSED */
1595static void
1596channel_post_connecting(Channel *c, fd_set *readset, fd_set *writeset)
1597{
1598	int err = 0, sock;
1599	socklen_t sz = sizeof(err);
1600
1601	if (FD_ISSET(c->sock, writeset)) {
1602		if (getsockopt(c->sock, SOL_SOCKET, SO_ERROR, &err, &sz) < 0) {
1603			err = errno;
1604			error("getsockopt SO_ERROR failed");
1605		}
1606		if (err == 0) {
1607			debug("channel %d: connected to %s port %d",
1608			    c->self, c->connect_ctx.host, c->connect_ctx.port);
1609			channel_connect_ctx_free(&c->connect_ctx);
1610			c->type = SSH_CHANNEL_OPEN;
1611			if (compat20) {
1612				packet_start(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION);
1613				packet_put_int(c->remote_id);
1614				packet_put_int(c->self);
1615				packet_put_int(c->local_window);
1616				packet_put_int(c->local_maxpacket);
1617			} else {
1618				packet_start(SSH_MSG_CHANNEL_OPEN_CONFIRMATION);
1619				packet_put_int(c->remote_id);
1620				packet_put_int(c->self);
1621			}
1622		} else {
1623			debug("channel %d: connection failed: %s",
1624			    c->self, strerror(err));
1625			/* Try next address, if any */
1626			if ((sock = connect_next(&c->connect_ctx)) > 0) {
1627				close(c->sock);
1628				c->sock = c->rfd = c->wfd = sock;
1629				channel_max_fd = channel_find_maxfd();
1630				return;
1631			}
1632			/* Exhausted all addresses */
1633			error("connect_to %.100s port %d: failed.",
1634			    c->connect_ctx.host, c->connect_ctx.port);
1635			channel_connect_ctx_free(&c->connect_ctx);
1636			if (compat20) {
1637				packet_start(SSH2_MSG_CHANNEL_OPEN_FAILURE);
1638				packet_put_int(c->remote_id);
1639				packet_put_int(SSH2_OPEN_CONNECT_FAILED);
1640				if (!(datafellows & SSH_BUG_OPENFAILURE)) {
1641					packet_put_cstring(strerror(err));
1642					packet_put_cstring("");
1643				}
1644			} else {
1645				packet_start(SSH_MSG_CHANNEL_OPEN_FAILURE);
1646				packet_put_int(c->remote_id);
1647			}
1648			chan_mark_dead(c);
1649		}
1650		packet_send();
1651	}
1652}
1653
1654/* ARGSUSED */
1655static int
1656channel_handle_rfd(Channel *c, fd_set *readset, fd_set *writeset)
1657{
1658	char buf[CHAN_RBUF];
1659	int len, force;
1660
1661	force = c->isatty && c->detach_close && c->istate != CHAN_INPUT_CLOSED;
1662	if (c->rfd != -1 && (force || FD_ISSET(c->rfd, readset))) {
1663		errno = 0;
1664		len = read(c->rfd, buf, sizeof(buf));
1665		if (len < 0 && (errno == EINTR ||
1666		    ((errno == EAGAIN || errno == EWOULDBLOCK) && !force)))
1667			return 1;
1668#ifndef PTY_ZEROREAD
1669		if (len <= 0) {
1670#else
1671		if ((!c->isatty && len <= 0) ||
1672		    (c->isatty && (len < 0 || (len == 0 && errno != 0)))) {
1673#endif
1674			debug2("channel %d: read<=0 rfd %d len %d",
1675			    c->self, c->rfd, len);
1676			if (c->type != SSH_CHANNEL_OPEN) {
1677				debug2("channel %d: not open", c->self);
1678				chan_mark_dead(c);
1679				return -1;
1680			} else if (compat13) {
1681				buffer_clear(&c->output);
1682				c->type = SSH_CHANNEL_INPUT_DRAINING;
1683				debug2("channel %d: input draining.", c->self);
1684			} else {
1685				chan_read_failed(c);
1686			}
1687			return -1;
1688		}
1689		if (c->input_filter != NULL) {
1690			if (c->input_filter(c, buf, len) == -1) {
1691				debug2("channel %d: filter stops", c->self);
1692				chan_read_failed(c);
1693			}
1694		} else if (c->datagram) {
1695			buffer_put_string(&c->input, buf, len);
1696		} else {
1697			buffer_append(&c->input, buf, len);
1698		}
1699	}
1700	return 1;
1701}
1702
1703/* ARGSUSED */
1704static int
1705channel_handle_wfd(Channel *c, fd_set *readset, fd_set *writeset)
1706{
1707	struct termios tio;
1708	u_char *data = NULL, *buf;
1709	u_int dlen, olen = 0;
1710	int len;
1711
1712	/* Send buffered output data to the socket. */
1713	if (c->wfd != -1 &&
1714	    FD_ISSET(c->wfd, writeset) &&
1715	    buffer_len(&c->output) > 0) {
1716		olen = buffer_len(&c->output);
1717		if (c->output_filter != NULL) {
1718			if ((buf = c->output_filter(c, &data, &dlen)) == NULL) {
1719				debug2("channel %d: filter stops", c->self);
1720				if (c->type != SSH_CHANNEL_OPEN)
1721					chan_mark_dead(c);
1722				else
1723					chan_write_failed(c);
1724				return -1;
1725			}
1726		} else if (c->datagram) {
1727			buf = data = buffer_get_string(&c->output, &dlen);
1728		} else {
1729			buf = data = buffer_ptr(&c->output);
1730			dlen = buffer_len(&c->output);
1731		}
1732
1733		if (c->datagram) {
1734			/* ignore truncated writes, datagrams might get lost */
1735			len = write(c->wfd, buf, dlen);
1736			free(data);
1737			if (len < 0 && (errno == EINTR || errno == EAGAIN ||
1738			    errno == EWOULDBLOCK))
1739				return 1;
1740			if (len <= 0) {
1741				if (c->type != SSH_CHANNEL_OPEN)
1742					chan_mark_dead(c);
1743				else
1744					chan_write_failed(c);
1745				return -1;
1746			}
1747			goto out;
1748		}
1749#ifdef _AIX
1750		/* XXX: Later AIX versions can't push as much data to tty */
1751		if (compat20 && c->wfd_isatty)
1752			dlen = MIN(dlen, 8*1024);
1753#endif
1754
1755		len = write(c->wfd, buf, dlen);
1756		if (len < 0 &&
1757		    (errno == EINTR || errno == EAGAIN || errno == EWOULDBLOCK))
1758			return 1;
1759		if (len <= 0) {
1760			if (c->type != SSH_CHANNEL_OPEN) {
1761				debug2("channel %d: not open", c->self);
1762				chan_mark_dead(c);
1763				return -1;
1764			} else if (compat13) {
1765				buffer_clear(&c->output);
1766				debug2("channel %d: input draining.", c->self);
1767				c->type = SSH_CHANNEL_INPUT_DRAINING;
1768			} else {
1769				chan_write_failed(c);
1770			}
1771			return -1;
1772		}
1773#ifndef BROKEN_TCGETATTR_ICANON
1774		if (compat20 && c->isatty && dlen >= 1 && buf[0] != '\r') {
1775			if (tcgetattr(c->wfd, &tio) == 0 &&
1776			    !(tio.c_lflag & ECHO) && (tio.c_lflag & ICANON)) {
1777				/*
1778				 * Simulate echo to reduce the impact of
1779				 * traffic analysis. We need to match the
1780				 * size of a SSH2_MSG_CHANNEL_DATA message
1781				 * (4 byte channel id + buf)
1782				 */
1783				packet_send_ignore(4 + len);
1784				packet_send();
1785			}
1786		}
1787#endif
1788		buffer_consume(&c->output, len);
1789	}
1790 out:
1791	if (compat20 && olen > 0)
1792		c->local_consumed += olen - buffer_len(&c->output);
1793	return 1;
1794}
1795
1796static int
1797channel_handle_efd(Channel *c, fd_set *readset, fd_set *writeset)
1798{
1799	char buf[CHAN_RBUF];
1800	int len;
1801
1802/** XXX handle drain efd, too */
1803	if (c->efd != -1) {
1804		if (c->extended_usage == CHAN_EXTENDED_WRITE &&
1805		    FD_ISSET(c->efd, writeset) &&
1806		    buffer_len(&c->extended) > 0) {
1807			len = write(c->efd, buffer_ptr(&c->extended),
1808			    buffer_len(&c->extended));
1809			debug2("channel %d: written %d to efd %d",
1810			    c->self, len, c->efd);
1811			if (len < 0 && (errno == EINTR || errno == EAGAIN ||
1812			    errno == EWOULDBLOCK))
1813				return 1;
1814			if (len <= 0) {
1815				debug2("channel %d: closing write-efd %d",
1816				    c->self, c->efd);
1817				channel_close_fd(&c->efd);
1818			} else {
1819				buffer_consume(&c->extended, len);
1820				c->local_consumed += len;
1821			}
1822		} else if (c->efd != -1 &&
1823		    (c->extended_usage == CHAN_EXTENDED_READ ||
1824		    c->extended_usage == CHAN_EXTENDED_IGNORE) &&
1825		    (c->detach_close || FD_ISSET(c->efd, readset))) {
1826			len = read(c->efd, buf, sizeof(buf));
1827			debug2("channel %d: read %d from efd %d",
1828			    c->self, len, c->efd);
1829			if (len < 0 && (errno == EINTR || ((errno == EAGAIN ||
1830			    errno == EWOULDBLOCK) && !c->detach_close)))
1831				return 1;
1832			if (len <= 0) {
1833				debug2("channel %d: closing read-efd %d",
1834				    c->self, c->efd);
1835				channel_close_fd(&c->efd);
1836			} else {
1837				if (c->extended_usage == CHAN_EXTENDED_IGNORE) {
1838					debug3("channel %d: discard efd",
1839					    c->self);
1840				} else
1841					buffer_append(&c->extended, buf, len);
1842			}
1843		}
1844	}
1845	return 1;
1846}
1847
1848static int
1849channel_check_window(Channel *c)
1850{
1851	if (c->type == SSH_CHANNEL_OPEN &&
1852	    !(c->flags & (CHAN_CLOSE_SENT|CHAN_CLOSE_RCVD)) &&
1853	    ((c->local_window_max - c->local_window >
1854	    c->local_maxpacket*3) ||
1855	    c->local_window < c->local_window_max/2) &&
1856	    c->local_consumed > 0) {
1857		u_int addition = 0;
1858
1859		/* Adjust max window size if we are in a dynamic environment. */
1860		if (c->dynamic_window && c->tcpwinsz > c->local_window_max) {
1861			/*
1862			 * Grow the window somewhat aggressively to maintain
1863			 * pressure.
1864			 */
1865			addition = 1.5 * (c->tcpwinsz - c->local_window_max);
1866			c->local_window_max += addition;
1867		}
1868		packet_start(SSH2_MSG_CHANNEL_WINDOW_ADJUST);
1869		packet_put_int(c->remote_id);
1870		packet_put_int(c->local_consumed + addition);
1871		packet_send();
1872		debug2("channel %d: window %d sent adjust %d",
1873		    c->self, c->local_window,
1874		    c->local_consumed);
1875		c->local_window += c->local_consumed + addition;
1876		c->local_consumed = 0;
1877	}
1878	return 1;
1879}
1880
1881static void
1882channel_post_open(Channel *c, fd_set *readset, fd_set *writeset)
1883{
1884	channel_handle_rfd(c, readset, writeset);
1885	channel_handle_wfd(c, readset, writeset);
1886	if (!compat20)
1887		return;
1888	channel_handle_efd(c, readset, writeset);
1889	channel_check_window(c);
1890}
1891
1892static u_int
1893read_mux(Channel *c, u_int need)
1894{
1895	char buf[CHAN_RBUF];
1896	int len;
1897	u_int rlen;
1898
1899	if (buffer_len(&c->input) < need) {
1900		rlen = need - buffer_len(&c->input);
1901		len = read(c->rfd, buf, MIN(rlen, CHAN_RBUF));
1902		if (len <= 0) {
1903			if (errno != EINTR && errno != EAGAIN) {
1904				debug2("channel %d: ctl read<=0 rfd %d len %d",
1905				    c->self, c->rfd, len);
1906				chan_read_failed(c);
1907				return 0;
1908			}
1909		} else
1910			buffer_append(&c->input, buf, len);
1911	}
1912	return buffer_len(&c->input);
1913}
1914
1915static void
1916channel_post_mux_client(Channel *c, fd_set *readset, fd_set *writeset)
1917{
1918	u_int need;
1919	ssize_t len;
1920
1921	if (!compat20)
1922		fatal("%s: entered with !compat20", __func__);
1923
1924	if (c->rfd != -1 && !c->mux_pause && FD_ISSET(c->rfd, readset) &&
1925	    (c->istate == CHAN_INPUT_OPEN ||
1926	    c->istate == CHAN_INPUT_WAIT_DRAIN)) {
1927		/*
1928		 * Don't not read past the precise end of packets to
1929		 * avoid disrupting fd passing.
1930		 */
1931		if (read_mux(c, 4) < 4) /* read header */
1932			return;
1933		need = get_u32(buffer_ptr(&c->input));
1934#define CHANNEL_MUX_MAX_PACKET	(256 * 1024)
1935		if (need > CHANNEL_MUX_MAX_PACKET) {
1936			debug2("channel %d: packet too big %u > %u",
1937			    c->self, CHANNEL_MUX_MAX_PACKET, need);
1938			chan_rcvd_oclose(c);
1939			return;
1940		}
1941		if (read_mux(c, need + 4) < need + 4) /* read body */
1942			return;
1943		if (c->mux_rcb(c) != 0) {
1944			debug("channel %d: mux_rcb failed", c->self);
1945			chan_mark_dead(c);
1946			return;
1947		}
1948	}
1949
1950	if (c->wfd != -1 && FD_ISSET(c->wfd, writeset) &&
1951	    buffer_len(&c->output) > 0) {
1952		len = write(c->wfd, buffer_ptr(&c->output),
1953		    buffer_len(&c->output));
1954		if (len < 0 && (errno == EINTR || errno == EAGAIN))
1955			return;
1956		if (len <= 0) {
1957			chan_mark_dead(c);
1958			return;
1959		}
1960		buffer_consume(&c->output, len);
1961	}
1962}
1963
1964static void
1965channel_post_mux_listener(Channel *c, fd_set *readset, fd_set *writeset)
1966{
1967	Channel *nc;
1968	struct sockaddr_storage addr;
1969	socklen_t addrlen;
1970	int newsock;
1971	uid_t euid;
1972	gid_t egid;
1973
1974	if (!FD_ISSET(c->sock, readset))
1975		return;
1976
1977	debug("multiplexing control connection");
1978
1979	/*
1980	 * Accept connection on control socket
1981	 */
1982	memset(&addr, 0, sizeof(addr));
1983	addrlen = sizeof(addr);
1984	if ((newsock = accept(c->sock, (struct sockaddr*)&addr,
1985	    &addrlen)) == -1) {
1986		error("%s accept: %s", __func__, strerror(errno));
1987		if (errno == EMFILE || errno == ENFILE)
1988			c->notbefore = monotime() + 1;
1989		return;
1990	}
1991
1992	if (getpeereid(newsock, &euid, &egid) < 0) {
1993		error("%s getpeereid failed: %s", __func__,
1994		    strerror(errno));
1995		close(newsock);
1996		return;
1997	}
1998	if ((euid != 0) && (getuid() != euid)) {
1999		error("multiplex uid mismatch: peer euid %u != uid %u",
2000		    (u_int)euid, (u_int)getuid());
2001		close(newsock);
2002		return;
2003	}
2004	nc = channel_new("multiplex client", SSH_CHANNEL_MUX_CLIENT,
2005	    newsock, newsock, -1, c->local_window_max,
2006	    c->local_maxpacket, 0, "mux-control", 1);
2007	nc->mux_rcb = c->mux_rcb;
2008	debug3("%s: new mux channel %d fd %d", __func__,
2009	    nc->self, nc->sock);
2010	/* establish state */
2011	nc->mux_rcb(nc);
2012	/* mux state transitions must not elicit protocol messages */
2013	nc->flags |= CHAN_LOCAL;
2014}
2015
2016/* ARGSUSED */
2017static void
2018channel_post_output_drain_13(Channel *c, fd_set *readset, fd_set *writeset)
2019{
2020	int len;
2021
2022	/* Send buffered output data to the socket. */
2023	if (FD_ISSET(c->sock, writeset) && buffer_len(&c->output) > 0) {
2024		len = write(c->sock, buffer_ptr(&c->output),
2025			    buffer_len(&c->output));
2026		if (len <= 0)
2027			buffer_clear(&c->output);
2028		else
2029			buffer_consume(&c->output, len);
2030	}
2031}
2032
2033static void
2034channel_handler_init_20(void)
2035{
2036	channel_pre[SSH_CHANNEL_OPEN] =			&channel_pre_open;
2037	channel_pre[SSH_CHANNEL_X11_OPEN] =		&channel_pre_x11_open;
2038	channel_pre[SSH_CHANNEL_PORT_LISTENER] =	&channel_pre_listener;
2039	channel_pre[SSH_CHANNEL_RPORT_LISTENER] =	&channel_pre_listener;
2040	channel_pre[SSH_CHANNEL_X11_LISTENER] =		&channel_pre_listener;
2041	channel_pre[SSH_CHANNEL_AUTH_SOCKET] =		&channel_pre_listener;
2042	channel_pre[SSH_CHANNEL_CONNECTING] =		&channel_pre_connecting;
2043	channel_pre[SSH_CHANNEL_DYNAMIC] =		&channel_pre_dynamic;
2044	channel_pre[SSH_CHANNEL_MUX_LISTENER] =		&channel_pre_listener;
2045	channel_pre[SSH_CHANNEL_MUX_CLIENT] =		&channel_pre_mux_client;
2046
2047	channel_post[SSH_CHANNEL_OPEN] =		&channel_post_open;
2048	channel_post[SSH_CHANNEL_PORT_LISTENER] =	&channel_post_port_listener;
2049	channel_post[SSH_CHANNEL_RPORT_LISTENER] =	&channel_post_port_listener;
2050	channel_post[SSH_CHANNEL_X11_LISTENER] =	&channel_post_x11_listener;
2051	channel_post[SSH_CHANNEL_AUTH_SOCKET] =		&channel_post_auth_listener;
2052	channel_post[SSH_CHANNEL_CONNECTING] =		&channel_post_connecting;
2053	channel_post[SSH_CHANNEL_DYNAMIC] =		&channel_post_open;
2054	channel_post[SSH_CHANNEL_MUX_LISTENER] =	&channel_post_mux_listener;
2055	channel_post[SSH_CHANNEL_MUX_CLIENT] =		&channel_post_mux_client;
2056}
2057
2058static void
2059channel_handler_init_13(void)
2060{
2061	channel_pre[SSH_CHANNEL_OPEN] =			&channel_pre_open_13;
2062	channel_pre[SSH_CHANNEL_X11_OPEN] =		&channel_pre_x11_open_13;
2063	channel_pre[SSH_CHANNEL_X11_LISTENER] =		&channel_pre_listener;
2064	channel_pre[SSH_CHANNEL_PORT_LISTENER] =	&channel_pre_listener;
2065	channel_pre[SSH_CHANNEL_AUTH_SOCKET] =		&channel_pre_listener;
2066	channel_pre[SSH_CHANNEL_INPUT_DRAINING] =	&channel_pre_input_draining;
2067	channel_pre[SSH_CHANNEL_OUTPUT_DRAINING] =	&channel_pre_output_draining;
2068	channel_pre[SSH_CHANNEL_CONNECTING] =		&channel_pre_connecting;
2069	channel_pre[SSH_CHANNEL_DYNAMIC] =		&channel_pre_dynamic;
2070
2071	channel_post[SSH_CHANNEL_OPEN] =		&channel_post_open;
2072	channel_post[SSH_CHANNEL_X11_LISTENER] =	&channel_post_x11_listener;
2073	channel_post[SSH_CHANNEL_PORT_LISTENER] =	&channel_post_port_listener;
2074	channel_post[SSH_CHANNEL_AUTH_SOCKET] =		&channel_post_auth_listener;
2075	channel_post[SSH_CHANNEL_OUTPUT_DRAINING] =	&channel_post_output_drain_13;
2076	channel_post[SSH_CHANNEL_CONNECTING] =		&channel_post_connecting;
2077	channel_post[SSH_CHANNEL_DYNAMIC] =		&channel_post_open;
2078}
2079
2080static void
2081channel_handler_init_15(void)
2082{
2083	channel_pre[SSH_CHANNEL_OPEN] =			&channel_pre_open;
2084	channel_pre[SSH_CHANNEL_X11_OPEN] =		&channel_pre_x11_open;
2085	channel_pre[SSH_CHANNEL_X11_LISTENER] =		&channel_pre_listener;
2086	channel_pre[SSH_CHANNEL_PORT_LISTENER] =	&channel_pre_listener;
2087	channel_pre[SSH_CHANNEL_AUTH_SOCKET] =		&channel_pre_listener;
2088	channel_pre[SSH_CHANNEL_CONNECTING] =		&channel_pre_connecting;
2089	channel_pre[SSH_CHANNEL_DYNAMIC] =		&channel_pre_dynamic;
2090
2091	channel_post[SSH_CHANNEL_X11_LISTENER] =	&channel_post_x11_listener;
2092	channel_post[SSH_CHANNEL_PORT_LISTENER] =	&channel_post_port_listener;
2093	channel_post[SSH_CHANNEL_AUTH_SOCKET] =		&channel_post_auth_listener;
2094	channel_post[SSH_CHANNEL_OPEN] =		&channel_post_open;
2095	channel_post[SSH_CHANNEL_CONNECTING] =		&channel_post_connecting;
2096	channel_post[SSH_CHANNEL_DYNAMIC] =		&channel_post_open;
2097}
2098
2099static void
2100channel_handler_init(void)
2101{
2102	int i;
2103
2104	for (i = 0; i < SSH_CHANNEL_MAX_TYPE; i++) {
2105		channel_pre[i] = NULL;
2106		channel_post[i] = NULL;
2107	}
2108	if (compat20)
2109		channel_handler_init_20();
2110	else if (compat13)
2111		channel_handler_init_13();
2112	else
2113		channel_handler_init_15();
2114}
2115
2116/* gc dead channels */
2117static void
2118channel_garbage_collect(Channel *c)
2119{
2120	if (c == NULL)
2121		return;
2122	if (c->detach_user != NULL) {
2123		if (!chan_is_dead(c, c->detach_close))
2124			return;
2125		debug2("channel %d: gc: notify user", c->self);
2126		c->detach_user(c->self, NULL);
2127		/* if we still have a callback */
2128		if (c->detach_user != NULL)
2129			return;
2130		debug2("channel %d: gc: user detached", c->self);
2131	}
2132	if (!chan_is_dead(c, 1))
2133		return;
2134	debug2("channel %d: garbage collecting", c->self);
2135	channel_free(c);
2136}
2137
2138static void
2139channel_handler(chan_fn *ftab[], fd_set *readset, fd_set *writeset,
2140    time_t *unpause_secs)
2141{
2142	static int did_init = 0;
2143	u_int i, oalloc;
2144	Channel *c;
2145	time_t now;
2146
2147	if (!did_init) {
2148		channel_handler_init();
2149		did_init = 1;
2150	}
2151	now = monotime();
2152	if (unpause_secs != NULL)
2153		*unpause_secs = 0;
2154	for (i = 0, oalloc = channels_alloc; i < oalloc; i++) {
2155		c = channels[i];
2156		if (c == NULL)
2157			continue;
2158		if (c->delayed) {
2159			if (ftab == channel_pre)
2160				c->delayed = 0;
2161			else
2162				continue;
2163		}
2164		if (ftab[c->type] != NULL) {
2165			/*
2166			 * Run handlers that are not paused.
2167			 */
2168			if (c->notbefore <= now)
2169				(*ftab[c->type])(c, readset, writeset);
2170			else if (unpause_secs != NULL) {
2171				/*
2172				 * Collect the time that the earliest
2173				 * channel comes off pause.
2174				 */
2175				debug3("%s: chan %d: skip for %d more seconds",
2176				    __func__, c->self,
2177				    (int)(c->notbefore - now));
2178				if (*unpause_secs == 0 ||
2179				    (c->notbefore - now) < *unpause_secs)
2180					*unpause_secs = c->notbefore - now;
2181			}
2182		}
2183		channel_garbage_collect(c);
2184	}
2185	if (unpause_secs != NULL && *unpause_secs != 0)
2186		debug3("%s: first channel unpauses in %d seconds",
2187		    __func__, (int)*unpause_secs);
2188}
2189
2190/*
2191 * Allocate/update select bitmasks and add any bits relevant to channels in
2192 * select bitmasks.
2193 */
2194void
2195channel_prepare_select(fd_set **readsetp, fd_set **writesetp, int *maxfdp,
2196    u_int *nallocp, time_t *minwait_secs, int rekeying)
2197{
2198	u_int n, sz, nfdset;
2199
2200	n = MAX(*maxfdp, channel_max_fd);
2201
2202	nfdset = howmany(n+1, NFDBITS);
2203	/* Explicitly test here, because xrealloc isn't always called */
2204	if (nfdset && SIZE_T_MAX / nfdset < sizeof(fd_mask))
2205		fatal("channel_prepare_select: max_fd (%d) is too large", n);
2206	sz = nfdset * sizeof(fd_mask);
2207
2208	/* perhaps check sz < nalloc/2 and shrink? */
2209	if (*readsetp == NULL || sz > *nallocp) {
2210		*readsetp = xrealloc(*readsetp, nfdset, sizeof(fd_mask));
2211		*writesetp = xrealloc(*writesetp, nfdset, sizeof(fd_mask));
2212		*nallocp = sz;
2213	}
2214	*maxfdp = n;
2215	memset(*readsetp, 0, sz);
2216	memset(*writesetp, 0, sz);
2217
2218	if (!rekeying)
2219		channel_handler(channel_pre, *readsetp, *writesetp,
2220		    minwait_secs);
2221}
2222
2223/*
2224 * After select, perform any appropriate operations for channels which have
2225 * events pending.
2226 */
2227void
2228channel_after_select(fd_set *readset, fd_set *writeset)
2229{
2230	channel_handler(channel_post, readset, writeset, NULL);
2231}
2232
2233
2234/* If there is data to send to the connection, enqueue some of it now. */
2235void
2236channel_output_poll(void)
2237{
2238	Channel *c;
2239	u_int i, len;
2240
2241	for (i = 0; i < channels_alloc; i++) {
2242		c = channels[i];
2243		if (c == NULL)
2244			continue;
2245
2246		/*
2247		 * We are only interested in channels that can have buffered
2248		 * incoming data.
2249		 */
2250		if (compat13) {
2251			if (c->type != SSH_CHANNEL_OPEN &&
2252			    c->type != SSH_CHANNEL_INPUT_DRAINING)
2253				continue;
2254		} else {
2255			if (c->type != SSH_CHANNEL_OPEN)
2256				continue;
2257		}
2258		if (compat20 &&
2259		    (c->flags & (CHAN_CLOSE_SENT|CHAN_CLOSE_RCVD))) {
2260			/* XXX is this true? */
2261			debug3("channel %d: will not send data after close", c->self);
2262			continue;
2263		}
2264
2265		/* Get the amount of buffered data for this channel. */
2266		if ((c->istate == CHAN_INPUT_OPEN ||
2267		    c->istate == CHAN_INPUT_WAIT_DRAIN) &&
2268		    (len = buffer_len(&c->input)) > 0) {
2269			if (c->datagram) {
2270				if (len > 0) {
2271					u_char *data;
2272					u_int dlen;
2273
2274					data = buffer_get_string(&c->input,
2275					    &dlen);
2276					if (dlen > c->remote_window ||
2277					    dlen > c->remote_maxpacket) {
2278						debug("channel %d: datagram "
2279						    "too big for channel",
2280						    c->self);
2281						free(data);
2282						continue;
2283					}
2284					packet_start(SSH2_MSG_CHANNEL_DATA);
2285					packet_put_int(c->remote_id);
2286					packet_put_string(data, dlen);
2287					packet_send();
2288					c->remote_window -= dlen + 4;
2289					free(data);
2290				}
2291				continue;
2292			}
2293			/*
2294			 * Send some data for the other side over the secure
2295			 * connection.
2296			 */
2297			if (compat20) {
2298				if (len > c->remote_window)
2299					len = c->remote_window;
2300				if (len > c->remote_maxpacket)
2301					len = c->remote_maxpacket;
2302			} else {
2303				if (packet_is_interactive()) {
2304					if (len > 1024)
2305						len = 512;
2306				} else {
2307					/* Keep the packets at reasonable size. */
2308					if (len > packet_get_maxsize()/2)
2309						len = packet_get_maxsize()/2;
2310				}
2311			}
2312			if (len > 0) {
2313				packet_start(compat20 ?
2314				    SSH2_MSG_CHANNEL_DATA : SSH_MSG_CHANNEL_DATA);
2315				packet_put_int(c->remote_id);
2316				packet_put_string(buffer_ptr(&c->input), len);
2317				packet_send();
2318				buffer_consume(&c->input, len);
2319				c->remote_window -= len;
2320			}
2321		} else if (c->istate == CHAN_INPUT_WAIT_DRAIN) {
2322			if (compat13)
2323				fatal("cannot happen: istate == INPUT_WAIT_DRAIN for proto 1.3");
2324			/*
2325			 * input-buffer is empty and read-socket shutdown:
2326			 * tell peer, that we will not send more data: send IEOF.
2327			 * hack for extended data: delay EOF if EFD still in use.
2328			 */
2329			if (CHANNEL_EFD_INPUT_ACTIVE(c))
2330				debug2("channel %d: ibuf_empty delayed efd %d/(%d)",
2331				    c->self, c->efd, buffer_len(&c->extended));
2332			else
2333				chan_ibuf_empty(c);
2334		}
2335		/* Send extended data, i.e. stderr */
2336		if (compat20 &&
2337		    !(c->flags & CHAN_EOF_SENT) &&
2338		    c->remote_window > 0 &&
2339		    (len = buffer_len(&c->extended)) > 0 &&
2340		    c->extended_usage == CHAN_EXTENDED_READ) {
2341			debug2("channel %d: rwin %u elen %u euse %d",
2342			    c->self, c->remote_window, buffer_len(&c->extended),
2343			    c->extended_usage);
2344			if (len > c->remote_window)
2345				len = c->remote_window;
2346			if (len > c->remote_maxpacket)
2347				len = c->remote_maxpacket;
2348			packet_start(SSH2_MSG_CHANNEL_EXTENDED_DATA);
2349			packet_put_int(c->remote_id);
2350			packet_put_int(SSH2_EXTENDED_DATA_STDERR);
2351			packet_put_string(buffer_ptr(&c->extended), len);
2352			packet_send();
2353			buffer_consume(&c->extended, len);
2354			c->remote_window -= len;
2355			debug2("channel %d: sent ext data %d", c->self, len);
2356		}
2357	}
2358}
2359
2360
2361/* -- protocol input */
2362
2363/* ARGSUSED */
2364void
2365channel_input_data(int type, u_int32_t seq, void *ctxt)
2366{
2367	int id;
2368	char *data;
2369	u_int data_len, win_len;
2370	Channel *c;
2371
2372	/* Get the channel number and verify it. */
2373	id = packet_get_int();
2374	c = channel_lookup(id);
2375	if (c == NULL)
2376		packet_disconnect("Received data for nonexistent channel %d.", id);
2377
2378	/* Ignore any data for non-open channels (might happen on close) */
2379	if (c->type != SSH_CHANNEL_OPEN &&
2380	    c->type != SSH_CHANNEL_X11_OPEN)
2381		return;
2382
2383	/* Get the data. */
2384	data = packet_get_string_ptr(&data_len);
2385	win_len = data_len;
2386	if (c->datagram)
2387		win_len += 4;  /* string length header */
2388
2389	/*
2390	 * Ignore data for protocol > 1.3 if output end is no longer open.
2391	 * For protocol 2 the sending side is reducing its window as it sends
2392	 * data, so we must 'fake' consumption of the data in order to ensure
2393	 * that window updates are sent back.  Otherwise the connection might
2394	 * deadlock.
2395	 */
2396	if (!compat13 && c->ostate != CHAN_OUTPUT_OPEN) {
2397		if (compat20) {
2398			c->local_window -= win_len;
2399			c->local_consumed += win_len;
2400		}
2401		return;
2402	}
2403
2404	if (compat20) {
2405		if (win_len > c->local_maxpacket) {
2406			logit("channel %d: rcvd big packet %d, maxpack %d",
2407			    c->self, win_len, c->local_maxpacket);
2408		}
2409		if (win_len > c->local_window) {
2410			logit("channel %d: rcvd too much data %d, win %d",
2411			    c->self, win_len, c->local_window);
2412			return;
2413		}
2414		c->local_window -= win_len;
2415	}
2416	if (c->datagram)
2417		buffer_put_string(&c->output, data, data_len);
2418	else
2419		buffer_append(&c->output, data, data_len);
2420	packet_check_eom();
2421}
2422
2423/* ARGSUSED */
2424void
2425channel_input_extended_data(int type, u_int32_t seq, void *ctxt)
2426{
2427	int id;
2428	char *data;
2429	u_int data_len, tcode;
2430	Channel *c;
2431
2432	/* Get the channel number and verify it. */
2433	id = packet_get_int();
2434	c = channel_lookup(id);
2435
2436	if (c == NULL)
2437		packet_disconnect("Received extended_data for bad channel %d.", id);
2438	if (c->type != SSH_CHANNEL_OPEN) {
2439		logit("channel %d: ext data for non open", id);
2440		return;
2441	}
2442	if (c->flags & CHAN_EOF_RCVD) {
2443		if (datafellows & SSH_BUG_EXTEOF)
2444			debug("channel %d: accepting ext data after eof", id);
2445		else
2446			packet_disconnect("Received extended_data after EOF "
2447			    "on channel %d.", id);
2448	}
2449	tcode = packet_get_int();
2450	if (c->efd == -1 ||
2451	    c->extended_usage != CHAN_EXTENDED_WRITE ||
2452	    tcode != SSH2_EXTENDED_DATA_STDERR) {
2453		logit("channel %d: bad ext data", c->self);
2454		return;
2455	}
2456	data = packet_get_string(&data_len);
2457	packet_check_eom();
2458	if (data_len > c->local_window) {
2459		logit("channel %d: rcvd too much extended_data %d, win %d",
2460		    c->self, data_len, c->local_window);
2461		free(data);
2462		return;
2463	}
2464	debug2("channel %d: rcvd ext data %d", c->self, data_len);
2465	c->local_window -= data_len;
2466	buffer_append(&c->extended, data, data_len);
2467	free(data);
2468}
2469
2470/* ARGSUSED */
2471void
2472channel_input_ieof(int type, u_int32_t seq, void *ctxt)
2473{
2474	int id;
2475	Channel *c;
2476
2477	id = packet_get_int();
2478	packet_check_eom();
2479	c = channel_lookup(id);
2480	if (c == NULL)
2481		packet_disconnect("Received ieof for nonexistent channel %d.", id);
2482	chan_rcvd_ieof(c);
2483
2484	/* XXX force input close */
2485	if (c->force_drain && c->istate == CHAN_INPUT_OPEN) {
2486		debug("channel %d: FORCE input drain", c->self);
2487		c->istate = CHAN_INPUT_WAIT_DRAIN;
2488		if (buffer_len(&c->input) == 0)
2489			chan_ibuf_empty(c);
2490	}
2491
2492}
2493
2494/* ARGSUSED */
2495void
2496channel_input_close(int type, u_int32_t seq, void *ctxt)
2497{
2498	int id;
2499	Channel *c;
2500
2501	id = packet_get_int();
2502	packet_check_eom();
2503	c = channel_lookup(id);
2504	if (c == NULL)
2505		packet_disconnect("Received close for nonexistent channel %d.", id);
2506
2507	/*
2508	 * Send a confirmation that we have closed the channel and no more
2509	 * data is coming for it.
2510	 */
2511	packet_start(SSH_MSG_CHANNEL_CLOSE_CONFIRMATION);
2512	packet_put_int(c->remote_id);
2513	packet_send();
2514
2515	/*
2516	 * If the channel is in closed state, we have sent a close request,
2517	 * and the other side will eventually respond with a confirmation.
2518	 * Thus, we cannot free the channel here, because then there would be
2519	 * no-one to receive the confirmation.  The channel gets freed when
2520	 * the confirmation arrives.
2521	 */
2522	if (c->type != SSH_CHANNEL_CLOSED) {
2523		/*
2524		 * Not a closed channel - mark it as draining, which will
2525		 * cause it to be freed later.
2526		 */
2527		buffer_clear(&c->input);
2528		c->type = SSH_CHANNEL_OUTPUT_DRAINING;
2529	}
2530}
2531
2532/* proto version 1.5 overloads CLOSE_CONFIRMATION with OCLOSE */
2533/* ARGSUSED */
2534void
2535channel_input_oclose(int type, u_int32_t seq, void *ctxt)
2536{
2537	int id = packet_get_int();
2538	Channel *c = channel_lookup(id);
2539
2540	packet_check_eom();
2541	if (c == NULL)
2542		packet_disconnect("Received oclose for nonexistent channel %d.", id);
2543	chan_rcvd_oclose(c);
2544}
2545
2546/* ARGSUSED */
2547void
2548channel_input_close_confirmation(int type, u_int32_t seq, void *ctxt)
2549{
2550	int id = packet_get_int();
2551	Channel *c = channel_lookup(id);
2552
2553	packet_check_eom();
2554	if (c == NULL)
2555		packet_disconnect("Received close confirmation for "
2556		    "out-of-range channel %d.", id);
2557	if (c->type != SSH_CHANNEL_CLOSED && c->type != SSH_CHANNEL_ABANDONED)
2558		packet_disconnect("Received close confirmation for "
2559		    "non-closed channel %d (type %d).", id, c->type);
2560	channel_free(c);
2561}
2562
2563/* ARGSUSED */
2564void
2565channel_input_open_confirmation(int type, u_int32_t seq, void *ctxt)
2566{
2567	int id, remote_id;
2568	Channel *c;
2569
2570	id = packet_get_int();
2571	c = channel_lookup(id);
2572
2573	if (c==NULL || c->type != SSH_CHANNEL_OPENING)
2574		packet_disconnect("Received open confirmation for "
2575		    "non-opening channel %d.", id);
2576	remote_id = packet_get_int();
2577	/* Record the remote channel number and mark that the channel is now open. */
2578	c->remote_id = remote_id;
2579	c->type = SSH_CHANNEL_OPEN;
2580
2581	if (compat20) {
2582		c->remote_window = packet_get_int();
2583		c->remote_maxpacket = packet_get_int();
2584		if (c->open_confirm) {
2585			debug2("callback start");
2586			c->open_confirm(c->self, 1, c->open_confirm_ctx);
2587			debug2("callback done");
2588		}
2589		debug2("channel %d: open confirm rwindow %u rmax %u", c->self,
2590		    c->remote_window, c->remote_maxpacket);
2591	}
2592	packet_check_eom();
2593}
2594
2595static char *
2596reason2txt(int reason)
2597{
2598	switch (reason) {
2599	case SSH2_OPEN_ADMINISTRATIVELY_PROHIBITED:
2600		return "administratively prohibited";
2601	case SSH2_OPEN_CONNECT_FAILED:
2602		return "connect failed";
2603	case SSH2_OPEN_UNKNOWN_CHANNEL_TYPE:
2604		return "unknown channel type";
2605	case SSH2_OPEN_RESOURCE_SHORTAGE:
2606		return "resource shortage";
2607	}
2608	return "unknown reason";
2609}
2610
2611/* ARGSUSED */
2612void
2613channel_input_open_failure(int type, u_int32_t seq, void *ctxt)
2614{
2615	int id, reason;
2616	char *msg = NULL, *lang = NULL;
2617	Channel *c;
2618
2619	id = packet_get_int();
2620	c = channel_lookup(id);
2621
2622	if (c==NULL || c->type != SSH_CHANNEL_OPENING)
2623		packet_disconnect("Received open failure for "
2624		    "non-opening channel %d.", id);
2625	if (compat20) {
2626		reason = packet_get_int();
2627		if (!(datafellows & SSH_BUG_OPENFAILURE)) {
2628			msg  = packet_get_string(NULL);
2629			lang = packet_get_string(NULL);
2630		}
2631		logit("channel %d: open failed: %s%s%s", id,
2632		    reason2txt(reason), msg ? ": ": "", msg ? msg : "");
2633		free(msg);
2634		free(lang);
2635		if (c->open_confirm) {
2636			debug2("callback start");
2637			c->open_confirm(c->self, 0, c->open_confirm_ctx);
2638			debug2("callback done");
2639		}
2640	}
2641	packet_check_eom();
2642	/* Schedule the channel for cleanup/deletion. */
2643	chan_mark_dead(c);
2644}
2645
2646/* ARGSUSED */
2647void
2648channel_input_window_adjust(int type, u_int32_t seq, void *ctxt)
2649{
2650	Channel *c;
2651	int id;
2652	u_int adjust;
2653
2654	if (!compat20)
2655		return;
2656
2657	/* Get the channel number and verify it. */
2658	id = packet_get_int();
2659	c = channel_lookup(id);
2660
2661	if (c == NULL) {
2662		logit("Received window adjust for non-open channel %d.", id);
2663		return;
2664	}
2665	adjust = packet_get_int();
2666	packet_check_eom();
2667	debug2("channel %d: rcvd adjust %u", id, adjust);
2668	c->remote_window += adjust;
2669}
2670
2671/* ARGSUSED */
2672void
2673channel_input_port_open(int type, u_int32_t seq, void *ctxt)
2674{
2675	Channel *c = NULL;
2676	u_short host_port;
2677	char *host, *originator_string;
2678	int remote_id;
2679
2680	remote_id = packet_get_int();
2681	host = packet_get_string(NULL);
2682	host_port = packet_get_int();
2683
2684	if (packet_get_protocol_flags() & SSH_PROTOFLAG_HOST_IN_FWD_OPEN) {
2685		originator_string = packet_get_string(NULL);
2686	} else {
2687		originator_string = xstrdup("unknown (remote did not supply name)");
2688	}
2689	packet_check_eom();
2690	c = channel_connect_to(host, host_port,
2691	    "connected socket", originator_string);
2692	free(originator_string);
2693	free(host);
2694	if (c == NULL) {
2695		packet_start(SSH_MSG_CHANNEL_OPEN_FAILURE);
2696		packet_put_int(remote_id);
2697		packet_send();
2698	} else
2699		c->remote_id = remote_id;
2700}
2701
2702/* ARGSUSED */
2703void
2704channel_input_status_confirm(int type, u_int32_t seq, void *ctxt)
2705{
2706	Channel *c;
2707	struct channel_confirm *cc;
2708	int id;
2709
2710	/* Reset keepalive timeout */
2711	packet_set_alive_timeouts(0);
2712
2713	id = packet_get_int();
2714	packet_check_eom();
2715
2716	debug2("channel_input_status_confirm: type %d id %d", type, id);
2717
2718	if ((c = channel_lookup(id)) == NULL) {
2719		logit("channel_input_status_confirm: %d: unknown", id);
2720		return;
2721	}
2722	;
2723	if ((cc = TAILQ_FIRST(&c->status_confirms)) == NULL)
2724		return;
2725	cc->cb(type, c, cc->ctx);
2726	TAILQ_REMOVE(&c->status_confirms, cc, entry);
2727	bzero(cc, sizeof(*cc));
2728	free(cc);
2729}
2730
2731/* -- tcp forwarding */
2732
2733void
2734channel_set_af(int af)
2735{
2736	IPv4or6 = af;
2737}
2738
2739void
2740channel_set_hpn(int disabled, u_int buf_size)
2741{
2742	hpn_disabled = disabled;
2743	buffer_size = buf_size;
2744	debug("HPN Disabled: %d, HPN Buffer Size: %d",
2745	    hpn_disabled, buffer_size);
2746}
2747
2748/*
2749 * Determine whether or not a port forward listens to loopback, the
2750 * specified address or wildcard. On the client, a specified bind
2751 * address will always override gateway_ports. On the server, a
2752 * gateway_ports of 1 (``yes'') will override the client's specification
2753 * and force a wildcard bind, whereas a value of 2 (``clientspecified'')
2754 * will bind to whatever address the client asked for.
2755 *
2756 * Special-case listen_addrs are:
2757 *
2758 * "0.0.0.0"               -> wildcard v4/v6 if SSH_OLD_FORWARD_ADDR
2759 * "" (empty string), "*"  -> wildcard v4/v6
2760 * "localhost"             -> loopback v4/v6
2761 */
2762static const char *
2763channel_fwd_bind_addr(const char *listen_addr, int *wildcardp,
2764    int is_client, int gateway_ports)
2765{
2766	const char *addr = NULL;
2767	int wildcard = 0;
2768
2769	if (listen_addr == NULL) {
2770		/* No address specified: default to gateway_ports setting */
2771		if (gateway_ports)
2772			wildcard = 1;
2773	} else if (gateway_ports || is_client) {
2774		if (((datafellows & SSH_OLD_FORWARD_ADDR) &&
2775		    strcmp(listen_addr, "0.0.0.0") == 0 && is_client == 0) ||
2776		    *listen_addr == '\0' || strcmp(listen_addr, "*") == 0 ||
2777		    (!is_client && gateway_ports == 1)) {
2778			wildcard = 1;
2779			/*
2780			 * Notify client if they requested a specific listen
2781			 * address and it was overridden.
2782			 */
2783			if (*listen_addr != '\0' &&
2784			    strcmp(listen_addr, "0.0.0.0") != 0 &&
2785			    strcmp(listen_addr, "*") != 0) {
2786				packet_send_debug("Forwarding listen address "
2787				    "\"%s\" overridden by server "
2788				    "GatewayPorts", listen_addr);
2789			}
2790		}
2791		else if (strcmp(listen_addr, "localhost") != 0)
2792			addr = listen_addr;
2793	}
2794	if (wildcardp != NULL)
2795		*wildcardp = wildcard;
2796	return addr;
2797}
2798
2799static int
2800channel_setup_fwd_listener(int type, const char *listen_addr,
2801    u_short listen_port, int *allocated_listen_port,
2802    const char *host_to_connect, u_short port_to_connect, int gateway_ports)
2803{
2804	Channel *c;
2805	int sock, r, success = 0, wildcard = 0, is_client;
2806	struct addrinfo hints, *ai, *aitop;
2807	const char *host, *addr;
2808	char ntop[NI_MAXHOST], strport[NI_MAXSERV];
2809	in_port_t *lport_p;
2810
2811	host = (type == SSH_CHANNEL_RPORT_LISTENER) ?
2812	    listen_addr : host_to_connect;
2813	is_client = (type == SSH_CHANNEL_PORT_LISTENER);
2814
2815	if (host == NULL) {
2816		error("No forward host name.");
2817		return 0;
2818	}
2819	if (strlen(host) >= NI_MAXHOST) {
2820		error("Forward host name too long.");
2821		return 0;
2822	}
2823
2824	/* Determine the bind address, cf. channel_fwd_bind_addr() comment */
2825	addr = channel_fwd_bind_addr(listen_addr, &wildcard,
2826	    is_client, gateway_ports);
2827	debug3("channel_setup_fwd_listener: type %d wildcard %d addr %s",
2828	    type, wildcard, (addr == NULL) ? "NULL" : addr);
2829
2830	/*
2831	 * getaddrinfo returns a loopback address if the hostname is
2832	 * set to NULL and hints.ai_flags is not AI_PASSIVE
2833	 */
2834	memset(&hints, 0, sizeof(hints));
2835	hints.ai_family = IPv4or6;
2836	hints.ai_flags = wildcard ? AI_PASSIVE : 0;
2837	hints.ai_socktype = SOCK_STREAM;
2838	snprintf(strport, sizeof strport, "%d", listen_port);
2839	if ((r = getaddrinfo(addr, strport, &hints, &aitop)) != 0) {
2840		if (addr == NULL) {
2841			/* This really shouldn't happen */
2842			packet_disconnect("getaddrinfo: fatal error: %s",
2843			    ssh_gai_strerror(r));
2844		} else {
2845			error("channel_setup_fwd_listener: "
2846			    "getaddrinfo(%.64s): %s", addr,
2847			    ssh_gai_strerror(r));
2848		}
2849		return 0;
2850	}
2851	if (allocated_listen_port != NULL)
2852		*allocated_listen_port = 0;
2853	for (ai = aitop; ai; ai = ai->ai_next) {
2854		switch (ai->ai_family) {
2855		case AF_INET:
2856			lport_p = &((struct sockaddr_in *)ai->ai_addr)->
2857			    sin_port;
2858			break;
2859		case AF_INET6:
2860			lport_p = &((struct sockaddr_in6 *)ai->ai_addr)->
2861			    sin6_port;
2862			break;
2863		default:
2864			continue;
2865		}
2866		/*
2867		 * If allocating a port for -R forwards, then use the
2868		 * same port for all address families.
2869		 */
2870		if (type == SSH_CHANNEL_RPORT_LISTENER && listen_port == 0 &&
2871		    allocated_listen_port != NULL && *allocated_listen_port > 0)
2872			*lport_p = htons(*allocated_listen_port);
2873
2874		if (getnameinfo(ai->ai_addr, ai->ai_addrlen, ntop, sizeof(ntop),
2875		    strport, sizeof(strport), NI_NUMERICHOST|NI_NUMERICSERV) != 0) {
2876			error("channel_setup_fwd_listener: getnameinfo failed");
2877			continue;
2878		}
2879		/* Create a port to listen for the host. */
2880		sock = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
2881		if (sock < 0) {
2882			/* this is no error since kernel may not support ipv6 */
2883			verbose("socket: %.100s", strerror(errno));
2884			continue;
2885		}
2886
2887		channel_set_reuseaddr(sock);
2888		if (ai->ai_family == AF_INET6)
2889			sock_set_v6only(sock);
2890
2891		debug("Local forwarding listening on %s port %s.",
2892		    ntop, strport);
2893
2894		/* Bind the socket to the address. */
2895		if (bind(sock, ai->ai_addr, ai->ai_addrlen) < 0) {
2896			/* address can be in use ipv6 address is already bound */
2897			if (!ai->ai_next)
2898				error("bind: %.100s", strerror(errno));
2899			else
2900				verbose("bind: %.100s", strerror(errno));
2901
2902			close(sock);
2903			continue;
2904		}
2905		/* Start listening for connections on the socket. */
2906		if (listen(sock, SSH_LISTEN_BACKLOG) < 0) {
2907			error("listen: %.100s", strerror(errno));
2908			close(sock);
2909			continue;
2910		}
2911
2912		/*
2913		 * listen_port == 0 requests a dynamically allocated port -
2914		 * record what we got.
2915		 */
2916		if (type == SSH_CHANNEL_RPORT_LISTENER && listen_port == 0 &&
2917		    allocated_listen_port != NULL &&
2918		    *allocated_listen_port == 0) {
2919			*allocated_listen_port = get_sock_port(sock, 1);
2920			debug("Allocated listen port %d",
2921			    *allocated_listen_port);
2922		}
2923
2924		/*
2925		 * Allocate a channel number for the socket.  Explicitly test
2926		 * for hpn disabled option.  If true use smaller window size.
2927		 */
2928		if (hpn_disabled)
2929			c = channel_new("port listener", type, sock, sock, -1,
2930			    CHAN_TCP_WINDOW_DEFAULT, CHAN_TCP_PACKET_DEFAULT,
2931			    0, "port listener", 1);
2932		else
2933			c = channel_new("port listener", type, sock, sock, -1,
2934			    buffer_size, CHAN_TCP_PACKET_DEFAULT,
2935			    0, "port listener", 1);
2936		c->path = xstrdup(host);
2937		c->host_port = port_to_connect;
2938		c->listening_addr = addr == NULL ? NULL : xstrdup(addr);
2939		if (listen_port == 0 && allocated_listen_port != NULL &&
2940		    !(datafellows & SSH_BUG_DYNAMIC_RPORT))
2941			c->listening_port = *allocated_listen_port;
2942		else
2943			c->listening_port = listen_port;
2944		success = 1;
2945	}
2946	if (success == 0)
2947		error("channel_setup_fwd_listener: cannot listen to port: %d",
2948		    listen_port);
2949	freeaddrinfo(aitop);
2950	return success;
2951}
2952
2953int
2954channel_cancel_rport_listener(const char *host, u_short port)
2955{
2956	u_int i;
2957	int found = 0;
2958
2959	for (i = 0; i < channels_alloc; i++) {
2960		Channel *c = channels[i];
2961		if (c == NULL || c->type != SSH_CHANNEL_RPORT_LISTENER)
2962			continue;
2963		if (strcmp(c->path, host) == 0 && c->listening_port == port) {
2964			debug2("%s: close channel %d", __func__, i);
2965			channel_free(c);
2966			found = 1;
2967		}
2968	}
2969
2970	return (found);
2971}
2972
2973int
2974channel_cancel_lport_listener(const char *lhost, u_short lport,
2975    int cport, int gateway_ports)
2976{
2977	u_int i;
2978	int found = 0;
2979	const char *addr = channel_fwd_bind_addr(lhost, NULL, 1, gateway_ports);
2980
2981	for (i = 0; i < channels_alloc; i++) {
2982		Channel *c = channels[i];
2983		if (c == NULL || c->type != SSH_CHANNEL_PORT_LISTENER)
2984			continue;
2985		if (c->listening_port != lport)
2986			continue;
2987		if (cport == CHANNEL_CANCEL_PORT_STATIC) {
2988			/* skip dynamic forwardings */
2989			if (c->host_port == 0)
2990				continue;
2991		} else {
2992			if (c->host_port != cport)
2993				continue;
2994		}
2995		if ((c->listening_addr == NULL && addr != NULL) ||
2996		    (c->listening_addr != NULL && addr == NULL))
2997			continue;
2998		if (addr == NULL || strcmp(c->listening_addr, addr) == 0) {
2999			debug2("%s: close channel %d", __func__, i);
3000			channel_free(c);
3001			found = 1;
3002		}
3003	}
3004
3005	return (found);
3006}
3007
3008/* protocol local port fwd, used by ssh (and sshd in v1) */
3009int
3010channel_setup_local_fwd_listener(const char *listen_host, u_short listen_port,
3011    const char *host_to_connect, u_short port_to_connect, int gateway_ports)
3012{
3013	return channel_setup_fwd_listener(SSH_CHANNEL_PORT_LISTENER,
3014	    listen_host, listen_port, NULL, host_to_connect, port_to_connect,
3015	    gateway_ports);
3016}
3017
3018/* protocol v2 remote port fwd, used by sshd */
3019int
3020channel_setup_remote_fwd_listener(const char *listen_address,
3021    u_short listen_port, int *allocated_listen_port, int gateway_ports)
3022{
3023	return channel_setup_fwd_listener(SSH_CHANNEL_RPORT_LISTENER,
3024	    listen_address, listen_port, allocated_listen_port,
3025	    NULL, 0, gateway_ports);
3026}
3027
3028/*
3029 * Translate the requested rfwd listen host to something usable for
3030 * this server.
3031 */
3032static const char *
3033channel_rfwd_bind_host(const char *listen_host)
3034{
3035	if (listen_host == NULL) {
3036		if (datafellows & SSH_BUG_RFWD_ADDR)
3037			return "127.0.0.1";
3038		else
3039			return "localhost";
3040	} else if (*listen_host == '\0' || strcmp(listen_host, "*") == 0) {
3041		if (datafellows & SSH_BUG_RFWD_ADDR)
3042			return "0.0.0.0";
3043		else
3044			return "";
3045	} else
3046		return listen_host;
3047}
3048
3049/*
3050 * Initiate forwarding of connections to port "port" on remote host through
3051 * the secure channel to host:port from local side.
3052 * Returns handle (index) for updating the dynamic listen port with
3053 * channel_update_permitted_opens().
3054 */
3055int
3056channel_request_remote_forwarding(const char *listen_host, u_short listen_port,
3057    const char *host_to_connect, u_short port_to_connect)
3058{
3059	int type, success = 0, idx = -1;
3060
3061	/* Send the forward request to the remote side. */
3062	if (compat20) {
3063		packet_start(SSH2_MSG_GLOBAL_REQUEST);
3064		packet_put_cstring("tcpip-forward");
3065		packet_put_char(1);		/* boolean: want reply */
3066		packet_put_cstring(channel_rfwd_bind_host(listen_host));
3067		packet_put_int(listen_port);
3068		packet_send();
3069		packet_write_wait();
3070		/* Assume that server accepts the request */
3071		success = 1;
3072	} else {
3073		packet_start(SSH_CMSG_PORT_FORWARD_REQUEST);
3074		packet_put_int(listen_port);
3075		packet_put_cstring(host_to_connect);
3076		packet_put_int(port_to_connect);
3077		packet_send();
3078		packet_write_wait();
3079
3080		/* Wait for response from the remote side. */
3081		type = packet_read();
3082		switch (type) {
3083		case SSH_SMSG_SUCCESS:
3084			success = 1;
3085			break;
3086		case SSH_SMSG_FAILURE:
3087			break;
3088		default:
3089			/* Unknown packet */
3090			packet_disconnect("Protocol error for port forward request:"
3091			    "received packet type %d.", type);
3092		}
3093	}
3094	if (success) {
3095		/* Record that connection to this host/port is permitted. */
3096		permitted_opens = xrealloc(permitted_opens,
3097		    num_permitted_opens + 1, sizeof(*permitted_opens));
3098		idx = num_permitted_opens++;
3099		permitted_opens[idx].host_to_connect = xstrdup(host_to_connect);
3100		permitted_opens[idx].port_to_connect = port_to_connect;
3101		permitted_opens[idx].listen_port = listen_port;
3102	}
3103	return (idx);
3104}
3105
3106/*
3107 * Request cancellation of remote forwarding of connection host:port from
3108 * local side.
3109 */
3110int
3111channel_request_rforward_cancel(const char *host, u_short port)
3112{
3113	int i;
3114
3115	if (!compat20)
3116		return -1;
3117
3118	for (i = 0; i < num_permitted_opens; i++) {
3119		if (permitted_opens[i].host_to_connect != NULL &&
3120		    permitted_opens[i].listen_port == port)
3121			break;
3122	}
3123	if (i >= num_permitted_opens) {
3124		debug("%s: requested forward not found", __func__);
3125		return -1;
3126	}
3127	packet_start(SSH2_MSG_GLOBAL_REQUEST);
3128	packet_put_cstring("cancel-tcpip-forward");
3129	packet_put_char(0);
3130	packet_put_cstring(channel_rfwd_bind_host(host));
3131	packet_put_int(port);
3132	packet_send();
3133
3134	permitted_opens[i].listen_port = 0;
3135	permitted_opens[i].port_to_connect = 0;
3136	free(permitted_opens[i].host_to_connect);
3137	permitted_opens[i].host_to_connect = NULL;
3138
3139	return 0;
3140}
3141
3142/*
3143 * This is called after receiving CHANNEL_FORWARDING_REQUEST.  This initates
3144 * listening for the port, and sends back a success reply (or disconnect
3145 * message if there was an error).
3146 */
3147int
3148channel_input_port_forward_request(int is_root, int gateway_ports)
3149{
3150	u_short port, host_port;
3151	int success = 0;
3152	char *hostname;
3153
3154	/* Get arguments from the packet. */
3155	port = packet_get_int();
3156	hostname = packet_get_string(NULL);
3157	host_port = packet_get_int();
3158
3159#ifndef HAVE_CYGWIN
3160	/*
3161	 * Check that an unprivileged user is not trying to forward a
3162	 * privileged port.
3163	 */
3164	if (port < IPPORT_RESERVED && !is_root)
3165		packet_disconnect(
3166		    "Requested forwarding of port %d but user is not root.",
3167		    port);
3168	if (host_port == 0)
3169		packet_disconnect("Dynamic forwarding denied.");
3170#endif
3171
3172	/* Initiate forwarding */
3173	success = channel_setup_local_fwd_listener(NULL, port, hostname,
3174	    host_port, gateway_ports);
3175
3176	/* Free the argument string. */
3177	free(hostname);
3178
3179	return (success ? 0 : -1);
3180}
3181
3182/*
3183 * Permits opening to any host/port if permitted_opens[] is empty.  This is
3184 * usually called by the server, because the user could connect to any port
3185 * anyway, and the server has no way to know but to trust the client anyway.
3186 */
3187void
3188channel_permit_all_opens(void)
3189{
3190	if (num_permitted_opens == 0)
3191		all_opens_permitted = 1;
3192}
3193
3194void
3195channel_add_permitted_opens(char *host, int port)
3196{
3197	debug("allow port forwarding to host %s port %d", host, port);
3198
3199	permitted_opens = xrealloc(permitted_opens,
3200	    num_permitted_opens + 1, sizeof(*permitted_opens));
3201	permitted_opens[num_permitted_opens].host_to_connect = xstrdup(host);
3202	permitted_opens[num_permitted_opens].port_to_connect = port;
3203	num_permitted_opens++;
3204
3205	all_opens_permitted = 0;
3206}
3207
3208/*
3209 * Update the listen port for a dynamic remote forward, after
3210 * the actual 'newport' has been allocated. If 'newport' < 0 is
3211 * passed then they entry will be invalidated.
3212 */
3213void
3214channel_update_permitted_opens(int idx, int newport)
3215{
3216	if (idx < 0 || idx >= num_permitted_opens) {
3217		debug("channel_update_permitted_opens: index out of range:"
3218		    " %d num_permitted_opens %d", idx, num_permitted_opens);
3219		return;
3220	}
3221	debug("%s allowed port %d for forwarding to host %s port %d",
3222	    newport > 0 ? "Updating" : "Removing",
3223	    newport,
3224	    permitted_opens[idx].host_to_connect,
3225	    permitted_opens[idx].port_to_connect);
3226	if (newport >= 0)  {
3227		permitted_opens[idx].listen_port =
3228		    (datafellows & SSH_BUG_DYNAMIC_RPORT) ? 0 : newport;
3229	} else {
3230		permitted_opens[idx].listen_port = 0;
3231		permitted_opens[idx].port_to_connect = 0;
3232		free(permitted_opens[idx].host_to_connect);
3233		permitted_opens[idx].host_to_connect = NULL;
3234	}
3235}
3236
3237int
3238channel_add_adm_permitted_opens(char *host, int port)
3239{
3240	debug("config allows port forwarding to host %s port %d", host, port);
3241
3242	permitted_adm_opens = xrealloc(permitted_adm_opens,
3243	    num_adm_permitted_opens + 1, sizeof(*permitted_adm_opens));
3244	permitted_adm_opens[num_adm_permitted_opens].host_to_connect
3245	     = xstrdup(host);
3246	permitted_adm_opens[num_adm_permitted_opens].port_to_connect = port;
3247	return ++num_adm_permitted_opens;
3248}
3249
3250void
3251channel_disable_adm_local_opens(void)
3252{
3253	channel_clear_adm_permitted_opens();
3254	permitted_adm_opens = xmalloc(sizeof(*permitted_adm_opens));
3255	permitted_adm_opens[num_adm_permitted_opens].host_to_connect = NULL;
3256	num_adm_permitted_opens = 1;
3257}
3258
3259void
3260channel_clear_permitted_opens(void)
3261{
3262	int i;
3263
3264	for (i = 0; i < num_permitted_opens; i++)
3265		free(permitted_opens[i].host_to_connect);
3266	free(permitted_opens);
3267	permitted_opens = NULL;
3268	num_permitted_opens = 0;
3269}
3270
3271void
3272channel_clear_adm_permitted_opens(void)
3273{
3274	int i;
3275
3276	for (i = 0; i < num_adm_permitted_opens; i++)
3277		free(permitted_adm_opens[i].host_to_connect);
3278	free(permitted_adm_opens);
3279	permitted_adm_opens = NULL;
3280	num_adm_permitted_opens = 0;
3281}
3282
3283void
3284channel_print_adm_permitted_opens(void)
3285{
3286	int i;
3287
3288	printf("permitopen");
3289	if (num_adm_permitted_opens == 0) {
3290		printf(" any\n");
3291		return;
3292	}
3293	for (i = 0; i < num_adm_permitted_opens; i++)
3294		if (permitted_adm_opens[i].host_to_connect == NULL)
3295			printf(" none");
3296		else
3297			printf(" %s:%d", permitted_adm_opens[i].host_to_connect,
3298			    permitted_adm_opens[i].port_to_connect);
3299	printf("\n");
3300}
3301
3302/* returns port number, FWD_PERMIT_ANY_PORT or -1 on error */
3303int
3304permitopen_port(const char *p)
3305{
3306	int port;
3307
3308	if (strcmp(p, "*") == 0)
3309		return FWD_PERMIT_ANY_PORT;
3310	if ((port = a2port(p)) > 0)
3311		return port;
3312	return -1;
3313}
3314
3315static int
3316port_match(u_short allowedport, u_short requestedport)
3317{
3318	if (allowedport == FWD_PERMIT_ANY_PORT ||
3319	    allowedport == requestedport)
3320		return 1;
3321	return 0;
3322}
3323
3324/* Try to start non-blocking connect to next host in cctx list */
3325static int
3326connect_next(struct channel_connect *cctx)
3327{
3328	int sock, saved_errno;
3329	char ntop[NI_MAXHOST], strport[NI_MAXSERV];
3330
3331	for (; cctx->ai; cctx->ai = cctx->ai->ai_next) {
3332		if (cctx->ai->ai_family != AF_INET &&
3333		    cctx->ai->ai_family != AF_INET6)
3334			continue;
3335		if (getnameinfo(cctx->ai->ai_addr, cctx->ai->ai_addrlen,
3336		    ntop, sizeof(ntop), strport, sizeof(strport),
3337		    NI_NUMERICHOST|NI_NUMERICSERV) != 0) {
3338			error("connect_next: getnameinfo failed");
3339			continue;
3340		}
3341		if ((sock = socket(cctx->ai->ai_family, cctx->ai->ai_socktype,
3342		    cctx->ai->ai_protocol)) == -1) {
3343			if (cctx->ai->ai_next == NULL)
3344				error("socket: %.100s", strerror(errno));
3345			else
3346				verbose("socket: %.100s", strerror(errno));
3347			continue;
3348		}
3349		if (set_nonblock(sock) == -1)
3350			fatal("%s: set_nonblock(%d)", __func__, sock);
3351		if (connect(sock, cctx->ai->ai_addr,
3352		    cctx->ai->ai_addrlen) == -1 && errno != EINPROGRESS) {
3353			debug("connect_next: host %.100s ([%.100s]:%s): "
3354			    "%.100s", cctx->host, ntop, strport,
3355			    strerror(errno));
3356			saved_errno = errno;
3357			close(sock);
3358			errno = saved_errno;
3359			continue;	/* fail -- try next */
3360		}
3361		debug("connect_next: host %.100s ([%.100s]:%s) "
3362		    "in progress, fd=%d", cctx->host, ntop, strport, sock);
3363		cctx->ai = cctx->ai->ai_next;
3364		set_nodelay(sock);
3365		return sock;
3366	}
3367	return -1;
3368}
3369
3370static void
3371channel_connect_ctx_free(struct channel_connect *cctx)
3372{
3373	free(cctx->host);
3374	if (cctx->aitop)
3375		freeaddrinfo(cctx->aitop);
3376	bzero(cctx, sizeof(*cctx));
3377	cctx->host = NULL;
3378	cctx->ai = cctx->aitop = NULL;
3379}
3380
3381/* Return CONNECTING channel to remote host, port */
3382static Channel *
3383connect_to(const char *host, u_short port, char *ctype, char *rname)
3384{
3385	struct addrinfo hints;
3386	int gaierr;
3387	int sock = -1;
3388	char strport[NI_MAXSERV];
3389	struct channel_connect cctx;
3390	Channel *c;
3391
3392	memset(&cctx, 0, sizeof(cctx));
3393	memset(&hints, 0, sizeof(hints));
3394	hints.ai_family = IPv4or6;
3395	hints.ai_socktype = SOCK_STREAM;
3396	snprintf(strport, sizeof strport, "%d", port);
3397	if ((gaierr = getaddrinfo(host, strport, &hints, &cctx.aitop)) != 0) {
3398		error("connect_to %.100s: unknown host (%s)", host,
3399		    ssh_gai_strerror(gaierr));
3400		return NULL;
3401	}
3402
3403	cctx.host = xstrdup(host);
3404	cctx.port = port;
3405	cctx.ai = cctx.aitop;
3406
3407	if ((sock = connect_next(&cctx)) == -1) {
3408		error("connect to %.100s port %d failed: %s",
3409		    host, port, strerror(errno));
3410		channel_connect_ctx_free(&cctx);
3411		return NULL;
3412	}
3413	c = channel_new(ctype, SSH_CHANNEL_CONNECTING, sock, sock, -1,
3414	    CHAN_TCP_WINDOW_DEFAULT, CHAN_TCP_PACKET_DEFAULT, 0, rname, 1);
3415	c->connect_ctx = cctx;
3416	return c;
3417}
3418
3419Channel *
3420channel_connect_by_listen_address(u_short listen_port, char *ctype, char *rname)
3421{
3422	int i;
3423
3424	for (i = 0; i < num_permitted_opens; i++) {
3425		if (permitted_opens[i].host_to_connect != NULL &&
3426		    port_match(permitted_opens[i].listen_port, listen_port)) {
3427			return connect_to(
3428			    permitted_opens[i].host_to_connect,
3429			    permitted_opens[i].port_to_connect, ctype, rname);
3430		}
3431	}
3432	error("WARNING: Server requests forwarding for unknown listen_port %d",
3433	    listen_port);
3434	return NULL;
3435}
3436
3437/* Check if connecting to that port is permitted and connect. */
3438Channel *
3439channel_connect_to(const char *host, u_short port, char *ctype, char *rname)
3440{
3441	int i, permit, permit_adm = 1;
3442
3443	permit = all_opens_permitted;
3444	if (!permit) {
3445		for (i = 0; i < num_permitted_opens; i++)
3446			if (permitted_opens[i].host_to_connect != NULL &&
3447			    port_match(permitted_opens[i].port_to_connect, port) &&
3448			    strcmp(permitted_opens[i].host_to_connect, host) == 0)
3449				permit = 1;
3450	}
3451
3452	if (num_adm_permitted_opens > 0) {
3453		permit_adm = 0;
3454		for (i = 0; i < num_adm_permitted_opens; i++)
3455			if (permitted_adm_opens[i].host_to_connect != NULL &&
3456			    port_match(permitted_adm_opens[i].port_to_connect, port) &&
3457			    strcmp(permitted_adm_opens[i].host_to_connect, host)
3458			    == 0)
3459				permit_adm = 1;
3460	}
3461
3462	if (!permit || !permit_adm) {
3463		logit("Received request to connect to host %.100s port %d, "
3464		    "but the request was denied.", host, port);
3465		return NULL;
3466	}
3467	return connect_to(host, port, ctype, rname);
3468}
3469
3470void
3471channel_send_window_changes(void)
3472{
3473	u_int i;
3474	struct winsize ws;
3475
3476	for (i = 0; i < channels_alloc; i++) {
3477		if (channels[i] == NULL || !channels[i]->client_tty ||
3478		    channels[i]->type != SSH_CHANNEL_OPEN)
3479			continue;
3480		if (ioctl(channels[i]->rfd, TIOCGWINSZ, &ws) < 0)
3481			continue;
3482		channel_request_start(i, "window-change", 0);
3483		packet_put_int((u_int)ws.ws_col);
3484		packet_put_int((u_int)ws.ws_row);
3485		packet_put_int((u_int)ws.ws_xpixel);
3486		packet_put_int((u_int)ws.ws_ypixel);
3487		packet_send();
3488	}
3489}
3490
3491/* -- X11 forwarding */
3492
3493/*
3494 * Creates an internet domain socket for listening for X11 connections.
3495 * Returns 0 and a suitable display number for the DISPLAY variable
3496 * stored in display_numberp , or -1 if an error occurs.
3497 */
3498int
3499x11_create_display_inet(int x11_display_offset, int x11_use_localhost,
3500    int single_connection, u_int *display_numberp, int **chanids)
3501{
3502	Channel *nc = NULL;
3503	int display_number, sock;
3504	u_short port;
3505	struct addrinfo hints, *ai, *aitop;
3506	char strport[NI_MAXSERV];
3507	int gaierr, n, num_socks = 0, socks[NUM_SOCKS];
3508
3509	if (chanids == NULL)
3510		return -1;
3511
3512	for (display_number = x11_display_offset;
3513	    display_number < MAX_DISPLAYS;
3514	    display_number++) {
3515		port = 6000 + display_number;
3516		memset(&hints, 0, sizeof(hints));
3517		hints.ai_family = IPv4or6;
3518		hints.ai_flags = x11_use_localhost ? 0: AI_PASSIVE;
3519		hints.ai_socktype = SOCK_STREAM;
3520		snprintf(strport, sizeof strport, "%d", port);
3521		if ((gaierr = getaddrinfo(NULL, strport, &hints, &aitop)) != 0) {
3522			error("getaddrinfo: %.100s", ssh_gai_strerror(gaierr));
3523			return -1;
3524		}
3525		for (ai = aitop; ai; ai = ai->ai_next) {
3526			if (ai->ai_family != AF_INET && ai->ai_family != AF_INET6)
3527				continue;
3528			sock = socket(ai->ai_family, ai->ai_socktype,
3529			    ai->ai_protocol);
3530			if (sock < 0) {
3531				if ((errno != EINVAL) && (errno != EAFNOSUPPORT)
3532#ifdef EPFNOSUPPORT
3533				    && (errno != EPFNOSUPPORT)
3534#endif
3535				    ) {
3536					error("socket: %.100s", strerror(errno));
3537					freeaddrinfo(aitop);
3538					return -1;
3539				} else {
3540					debug("x11_create_display_inet: Socket family %d not supported",
3541						 ai->ai_family);
3542					continue;
3543				}
3544			}
3545			if (ai->ai_family == AF_INET6)
3546				sock_set_v6only(sock);
3547			if (x11_use_localhost)
3548				channel_set_reuseaddr(sock);
3549			if (bind(sock, ai->ai_addr, ai->ai_addrlen) < 0) {
3550				debug2("bind port %d: %.100s", port, strerror(errno));
3551				close(sock);
3552
3553				for (n = 0; n < num_socks; n++) {
3554					close(socks[n]);
3555				}
3556				num_socks = 0;
3557				break;
3558			}
3559			socks[num_socks++] = sock;
3560			if (num_socks == NUM_SOCKS)
3561				break;
3562		}
3563		freeaddrinfo(aitop);
3564		if (num_socks > 0)
3565			break;
3566	}
3567	if (display_number >= MAX_DISPLAYS) {
3568		error("Failed to allocate internet-domain X11 display socket.");
3569		return -1;
3570	}
3571	/* Start listening for connections on the socket. */
3572	for (n = 0; n < num_socks; n++) {
3573		sock = socks[n];
3574		if (listen(sock, SSH_LISTEN_BACKLOG) < 0) {
3575			error("listen: %.100s", strerror(errno));
3576			close(sock);
3577			return -1;
3578		}
3579	}
3580
3581	/* Allocate a channel for each socket. */
3582	*chanids = xcalloc(num_socks + 1, sizeof(**chanids));
3583	for (n = 0; n < num_socks; n++) {
3584		sock = socks[n];
3585		if (hpn_disabled)
3586			nc = channel_new("x11 listener",
3587			    SSH_CHANNEL_X11_LISTENER, sock, sock, -1,
3588			    CHAN_X11_WINDOW_DEFAULT, CHAN_X11_PACKET_DEFAULT,
3589			    0, "X11 inet listener", 1);
3590		else
3591			nc = channel_new("x11 listener",
3592			    SSH_CHANNEL_X11_LISTENER, sock, sock, -1,
3593			    buffer_size, CHAN_X11_PACKET_DEFAULT,
3594			    0, "X11 inet listener", 1);
3595		nc->single_connection = single_connection;
3596		(*chanids)[n] = nc->self;
3597	}
3598	(*chanids)[n] = -1;
3599
3600	/* Return the display number for the DISPLAY environment variable. */
3601	*display_numberp = display_number;
3602	return (0);
3603}
3604
3605static int
3606connect_local_xsocket_path(const char *pathname)
3607{
3608	int sock;
3609	struct sockaddr_un addr;
3610
3611	sock = socket(AF_UNIX, SOCK_STREAM, 0);
3612	if (sock < 0)
3613		error("socket: %.100s", strerror(errno));
3614	memset(&addr, 0, sizeof(addr));
3615	addr.sun_family = AF_UNIX;
3616	strlcpy(addr.sun_path, pathname, sizeof addr.sun_path);
3617	if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) == 0)
3618		return sock;
3619	close(sock);
3620	error("connect %.100s: %.100s", addr.sun_path, strerror(errno));
3621	return -1;
3622}
3623
3624static int
3625connect_local_xsocket(u_int dnr)
3626{
3627	char buf[1024];
3628	snprintf(buf, sizeof buf, _PATH_UNIX_X, dnr);
3629	return connect_local_xsocket_path(buf);
3630}
3631
3632int
3633x11_connect_display(void)
3634{
3635	u_int display_number;
3636	const char *display;
3637	char buf[1024], *cp;
3638	struct addrinfo hints, *ai, *aitop;
3639	char strport[NI_MAXSERV];
3640	int gaierr, sock = 0;
3641
3642	/* Try to open a socket for the local X server. */
3643	display = getenv("DISPLAY");
3644	if (!display) {
3645		error("DISPLAY not set.");
3646		return -1;
3647	}
3648	/*
3649	 * Now we decode the value of the DISPLAY variable and make a
3650	 * connection to the real X server.
3651	 */
3652
3653	/* Check if the display is from launchd. */
3654#ifdef __APPLE__
3655	if (strncmp(display, "/tmp/launch", 11) == 0) {
3656		sock = connect_local_xsocket_path(display);
3657		if (sock < 0)
3658			return -1;
3659
3660		/* OK, we now have a connection to the display. */
3661		return sock;
3662	}
3663#endif
3664	/*
3665	 * Check if it is a unix domain socket.  Unix domain displays are in
3666	 * one of the following formats: unix:d[.s], :d[.s], ::d[.s]
3667	 */
3668	if (strncmp(display, "unix:", 5) == 0 ||
3669	    display[0] == ':') {
3670		/* Connect to the unix domain socket. */
3671		if (sscanf(strrchr(display, ':') + 1, "%u", &display_number) != 1) {
3672			error("Could not parse display number from DISPLAY: %.100s",
3673			    display);
3674			return -1;
3675		}
3676		/* Create a socket. */
3677		sock = connect_local_xsocket(display_number);
3678		if (sock < 0)
3679			return -1;
3680
3681		/* OK, we now have a connection to the display. */
3682		return sock;
3683	}
3684	/*
3685	 * Connect to an inet socket.  The DISPLAY value is supposedly
3686	 * hostname:d[.s], where hostname may also be numeric IP address.
3687	 */
3688	strlcpy(buf, display, sizeof(buf));
3689	cp = strchr(buf, ':');
3690	if (!cp) {
3691		error("Could not find ':' in DISPLAY: %.100s", display);
3692		return -1;
3693	}
3694	*cp = 0;
3695	/* buf now contains the host name.  But first we parse the display number. */
3696	if (sscanf(cp + 1, "%u", &display_number) != 1) {
3697		error("Could not parse display number from DISPLAY: %.100s",
3698		    display);
3699		return -1;
3700	}
3701
3702	/* Look up the host address */
3703	memset(&hints, 0, sizeof(hints));
3704	hints.ai_family = IPv4or6;
3705	hints.ai_socktype = SOCK_STREAM;
3706	snprintf(strport, sizeof strport, "%u", 6000 + display_number);
3707	if ((gaierr = getaddrinfo(buf, strport, &hints, &aitop)) != 0) {
3708		error("%.100s: unknown host. (%s)", buf,
3709		ssh_gai_strerror(gaierr));
3710		return -1;
3711	}
3712	for (ai = aitop; ai; ai = ai->ai_next) {
3713		/* Create a socket. */
3714		sock = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
3715		if (sock < 0) {
3716			debug2("socket: %.100s", strerror(errno));
3717			continue;
3718		}
3719		/* Connect it to the display. */
3720		if (connect(sock, ai->ai_addr, ai->ai_addrlen) < 0) {
3721			debug2("connect %.100s port %u: %.100s", buf,
3722			    6000 + display_number, strerror(errno));
3723			close(sock);
3724			continue;
3725		}
3726		/* Success */
3727		break;
3728	}
3729	freeaddrinfo(aitop);
3730	if (!ai) {
3731		error("connect %.100s port %u: %.100s", buf, 6000 + display_number,
3732		    strerror(errno));
3733		return -1;
3734	}
3735	set_nodelay(sock);
3736	return sock;
3737}
3738
3739/*
3740 * This is called when SSH_SMSG_X11_OPEN is received.  The packet contains
3741 * the remote channel number.  We should do whatever we want, and respond
3742 * with either SSH_MSG_OPEN_CONFIRMATION or SSH_MSG_OPEN_FAILURE.
3743 */
3744
3745/* ARGSUSED */
3746void
3747x11_input_open(int type, u_int32_t seq, void *ctxt)
3748{
3749	Channel *c = NULL;
3750	int remote_id, sock = 0;
3751	char *remote_host;
3752
3753	debug("Received X11 open request.");
3754
3755	remote_id = packet_get_int();
3756
3757	if (packet_get_protocol_flags() & SSH_PROTOFLAG_HOST_IN_FWD_OPEN) {
3758		remote_host = packet_get_string(NULL);
3759	} else {
3760		remote_host = xstrdup("unknown (remote did not supply name)");
3761	}
3762	packet_check_eom();
3763
3764	/* Obtain a connection to the real X display. */
3765	sock = x11_connect_display();
3766	if (sock != -1) {
3767		/* Allocate a channel for this connection. */
3768		c = channel_new("connected x11 socket",
3769		    SSH_CHANNEL_X11_OPEN, sock, sock, -1, 0, 0, 0,
3770		    remote_host, 1);
3771		c->remote_id = remote_id;
3772		c->force_drain = 1;
3773	}
3774	free(remote_host);
3775	if (c == NULL) {
3776		/* Send refusal to the remote host. */
3777		packet_start(SSH_MSG_CHANNEL_OPEN_FAILURE);
3778		packet_put_int(remote_id);
3779	} else {
3780		/* Send a confirmation to the remote host. */
3781		packet_start(SSH_MSG_CHANNEL_OPEN_CONFIRMATION);
3782		packet_put_int(remote_id);
3783		packet_put_int(c->self);
3784	}
3785	packet_send();
3786}
3787
3788/* dummy protocol handler that denies SSH-1 requests (agent/x11) */
3789/* ARGSUSED */
3790void
3791deny_input_open(int type, u_int32_t seq, void *ctxt)
3792{
3793	int rchan = packet_get_int();
3794
3795	switch (type) {
3796	case SSH_SMSG_AGENT_OPEN:
3797		error("Warning: ssh server tried agent forwarding.");
3798		break;
3799	case SSH_SMSG_X11_OPEN:
3800		error("Warning: ssh server tried X11 forwarding.");
3801		break;
3802	default:
3803		error("deny_input_open: type %d", type);
3804		break;
3805	}
3806	error("Warning: this is probably a break-in attempt by a malicious server.");
3807	packet_start(SSH_MSG_CHANNEL_OPEN_FAILURE);
3808	packet_put_int(rchan);
3809	packet_send();
3810}
3811
3812/*
3813 * Requests forwarding of X11 connections, generates fake authentication
3814 * data, and enables authentication spoofing.
3815 * This should be called in the client only.
3816 */
3817void
3818x11_request_forwarding_with_spoofing(int client_session_id, const char *disp,
3819    const char *proto, const char *data, int want_reply)
3820{
3821	u_int data_len = (u_int) strlen(data) / 2;
3822	u_int i, value;
3823	char *new_data;
3824	int screen_number;
3825	const char *cp;
3826	u_int32_t rnd = 0;
3827
3828	if (x11_saved_display == NULL)
3829		x11_saved_display = xstrdup(disp);
3830	else if (strcmp(disp, x11_saved_display) != 0) {
3831		error("x11_request_forwarding_with_spoofing: different "
3832		    "$DISPLAY already forwarded");
3833		return;
3834	}
3835
3836	cp = strchr(disp, ':');
3837	if (cp)
3838		cp = strchr(cp, '.');
3839	if (cp)
3840		screen_number = (u_int)strtonum(cp + 1, 0, 400, NULL);
3841	else
3842		screen_number = 0;
3843
3844	if (x11_saved_proto == NULL) {
3845		/* Save protocol name. */
3846		x11_saved_proto = xstrdup(proto);
3847		/*
3848		 * Extract real authentication data and generate fake data
3849		 * of the same length.
3850		 */
3851		x11_saved_data = xmalloc(data_len);
3852		x11_fake_data = xmalloc(data_len);
3853		for (i = 0; i < data_len; i++) {
3854			if (sscanf(data + 2 * i, "%2x", &value) != 1)
3855				fatal("x11_request_forwarding: bad "
3856				    "authentication data: %.100s", data);
3857			if (i % 4 == 0)
3858				rnd = arc4random();
3859			x11_saved_data[i] = value;
3860			x11_fake_data[i] = rnd & 0xff;
3861			rnd >>= 8;
3862		}
3863		x11_saved_data_len = data_len;
3864		x11_fake_data_len = data_len;
3865	}
3866
3867	/* Convert the fake data into hex. */
3868	new_data = tohex(x11_fake_data, data_len);
3869
3870	/* Send the request packet. */
3871	if (compat20) {
3872		channel_request_start(client_session_id, "x11-req", want_reply);
3873		packet_put_char(0);	/* XXX bool single connection */
3874	} else {
3875		packet_start(SSH_CMSG_X11_REQUEST_FORWARDING);
3876	}
3877	packet_put_cstring(proto);
3878	packet_put_cstring(new_data);
3879	packet_put_int(screen_number);
3880	packet_send();
3881	packet_write_wait();
3882	free(new_data);
3883}
3884
3885
3886/* -- agent forwarding */
3887
3888/* Sends a message to the server to request authentication fd forwarding. */
3889
3890void
3891auth_request_forwarding(void)
3892{
3893	packet_start(SSH_CMSG_AGENT_REQUEST_FORWARDING);
3894	packet_send();
3895	packet_write_wait();
3896}
3897