xref: /freebsd/crypto/openssh/serverloop.c (revision a04a10f8915401e0ac20dce0ade6c5b6e1bb13da)
1511b41d2SMark Murray /*
2511b41d2SMark Murray  * Author: Tatu Ylonen <ylo@cs.hut.fi>
3511b41d2SMark Murray  * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
4511b41d2SMark Murray  *                    All rights reserved
5511b41d2SMark Murray  * Created: Sun Sep 10 00:30:37 1995 ylo
6511b41d2SMark Murray  * Server main loop for handling the interactive session.
7511b41d2SMark Murray  */
8a04a10f8SKris Kennaway /*
9a04a10f8SKris Kennaway  * SSH2 support by Markus Friedl.
10a04a10f8SKris Kennaway  * Copyright (c) 2000 Markus Friedl. All rights reserved.
11a04a10f8SKris Kennaway  */
12511b41d2SMark Murray 
13511b41d2SMark Murray #include "includes.h"
14511b41d2SMark Murray #include "xmalloc.h"
15511b41d2SMark Murray #include "ssh.h"
16511b41d2SMark Murray #include "packet.h"
17511b41d2SMark Murray #include "buffer.h"
18511b41d2SMark Murray #include "servconf.h"
19511b41d2SMark Murray #include "pty.h"
20a04a10f8SKris Kennaway #include "channels.h"
21a04a10f8SKris Kennaway 
22a04a10f8SKris Kennaway #include "compat.h"
23a04a10f8SKris Kennaway #include "ssh2.h"
24a04a10f8SKris Kennaway #include "session.h"
25a04a10f8SKris Kennaway #include "dispatch.h"
26511b41d2SMark Murray 
27511b41d2SMark Murray static Buffer stdin_buffer;	/* Buffer for stdin data. */
28511b41d2SMark Murray static Buffer stdout_buffer;	/* Buffer for stdout data. */
29511b41d2SMark Murray static Buffer stderr_buffer;	/* Buffer for stderr data. */
30511b41d2SMark Murray static int fdin;		/* Descriptor for stdin (for writing) */
31511b41d2SMark Murray static int fdout;		/* Descriptor for stdout (for reading);
32511b41d2SMark Murray 				   May be same number as fdin. */
33511b41d2SMark Murray static int fderr;		/* Descriptor for stderr.  May be -1. */
34511b41d2SMark Murray static long stdin_bytes = 0;	/* Number of bytes written to stdin. */
35511b41d2SMark Murray static long stdout_bytes = 0;	/* Number of stdout bytes sent to client. */
36511b41d2SMark Murray static long stderr_bytes = 0;	/* Number of stderr bytes sent to client. */
37511b41d2SMark Murray static long fdout_bytes = 0;	/* Number of stdout bytes read from program. */
38511b41d2SMark Murray static int stdin_eof = 0;	/* EOF message received from client. */
39511b41d2SMark Murray static int fdout_eof = 0;	/* EOF encountered reading from fdout. */
40511b41d2SMark Murray static int fderr_eof = 0;	/* EOF encountered readung from fderr. */
41511b41d2SMark Murray static int connection_in;	/* Connection to client (input). */
42511b41d2SMark Murray static int connection_out;	/* Connection to client (output). */
43511b41d2SMark Murray static unsigned int buffer_high;/* "Soft" max buffer size. */
44511b41d2SMark Murray static int max_fd;		/* Max file descriptor number for select(). */
45511b41d2SMark Murray 
46511b41d2SMark Murray /*
47511b41d2SMark Murray  * This SIGCHLD kludge is used to detect when the child exits.  The server
48511b41d2SMark Murray  * will exit after that, as soon as forwarded connections have terminated.
49511b41d2SMark Murray  */
50511b41d2SMark Murray 
51a04a10f8SKris Kennaway static pid_t child_pid;			/* Pid of the child. */
52511b41d2SMark Murray static volatile int child_terminated;	/* The child has terminated. */
53511b41d2SMark Murray static volatile int child_wait_status;	/* Status from wait(). */
54511b41d2SMark Murray 
55a04a10f8SKris Kennaway void	server_init_dispatch(void);
56a04a10f8SKris Kennaway 
57511b41d2SMark Murray void
58511b41d2SMark Murray sigchld_handler(int sig)
59511b41d2SMark Murray {
60511b41d2SMark Murray 	int save_errno = errno;
61a04a10f8SKris Kennaway 	pid_t wait_pid;
62a04a10f8SKris Kennaway 
63511b41d2SMark Murray 	debug("Received SIGCHLD.");
64511b41d2SMark Murray 	wait_pid = wait((int *) &child_wait_status);
65511b41d2SMark Murray 	if (wait_pid != -1) {
66511b41d2SMark Murray 		if (wait_pid != child_pid)
67511b41d2SMark Murray 			error("Strange, got SIGCHLD and wait returned pid %d but child is %d",
68511b41d2SMark Murray 			      wait_pid, child_pid);
69511b41d2SMark Murray 		if (WIFEXITED(child_wait_status) ||
70511b41d2SMark Murray 		    WIFSIGNALED(child_wait_status))
71511b41d2SMark Murray 			child_terminated = 1;
72511b41d2SMark Murray 	}
73511b41d2SMark Murray 	signal(SIGCHLD, sigchld_handler);
74511b41d2SMark Murray 	errno = save_errno;
75511b41d2SMark Murray }
76511b41d2SMark Murray void
77a04a10f8SKris Kennaway sigchld_handler2(int sig)
78511b41d2SMark Murray {
79a04a10f8SKris Kennaway 	int save_errno = errno;
80a04a10f8SKris Kennaway 	debug("Received SIGCHLD.");
81a04a10f8SKris Kennaway 	child_terminated = 1;
82a04a10f8SKris Kennaway 	signal(SIGCHLD, sigchld_handler2);
83a04a10f8SKris Kennaway 	errno = save_errno;
84511b41d2SMark Murray }
85511b41d2SMark Murray 
86511b41d2SMark Murray /*
87511b41d2SMark Murray  * Make packets from buffered stderr data, and buffer it for sending
88511b41d2SMark Murray  * to the client.
89511b41d2SMark Murray  */
90511b41d2SMark Murray void
91511b41d2SMark Murray make_packets_from_stderr_data()
92511b41d2SMark Murray {
93511b41d2SMark Murray 	int len;
94511b41d2SMark Murray 
95511b41d2SMark Murray 	/* Send buffered stderr data to the client. */
96511b41d2SMark Murray 	while (buffer_len(&stderr_buffer) > 0 &&
97511b41d2SMark Murray 	    packet_not_very_much_data_to_write()) {
98511b41d2SMark Murray 		len = buffer_len(&stderr_buffer);
99511b41d2SMark Murray 		if (packet_is_interactive()) {
100511b41d2SMark Murray 			if (len > 512)
101511b41d2SMark Murray 				len = 512;
102511b41d2SMark Murray 		} else {
103511b41d2SMark Murray 			/* Keep the packets at reasonable size. */
104511b41d2SMark Murray 			if (len > packet_get_maxsize())
105511b41d2SMark Murray 				len = packet_get_maxsize();
106511b41d2SMark Murray 		}
107511b41d2SMark Murray 		packet_start(SSH_SMSG_STDERR_DATA);
108511b41d2SMark Murray 		packet_put_string(buffer_ptr(&stderr_buffer), len);
109511b41d2SMark Murray 		packet_send();
110511b41d2SMark Murray 		buffer_consume(&stderr_buffer, len);
111511b41d2SMark Murray 		stderr_bytes += len;
112511b41d2SMark Murray 	}
113511b41d2SMark Murray }
114511b41d2SMark Murray 
115511b41d2SMark Murray /*
116511b41d2SMark Murray  * Make packets from buffered stdout data, and buffer it for sending to the
117511b41d2SMark Murray  * client.
118511b41d2SMark Murray  */
119511b41d2SMark Murray void
120511b41d2SMark Murray make_packets_from_stdout_data()
121511b41d2SMark Murray {
122511b41d2SMark Murray 	int len;
123511b41d2SMark Murray 
124511b41d2SMark Murray 	/* Send buffered stdout data to the client. */
125511b41d2SMark Murray 	while (buffer_len(&stdout_buffer) > 0 &&
126511b41d2SMark Murray 	    packet_not_very_much_data_to_write()) {
127511b41d2SMark Murray 		len = buffer_len(&stdout_buffer);
128511b41d2SMark Murray 		if (packet_is_interactive()) {
129511b41d2SMark Murray 			if (len > 512)
130511b41d2SMark Murray 				len = 512;
131511b41d2SMark Murray 		} else {
132511b41d2SMark Murray 			/* Keep the packets at reasonable size. */
133511b41d2SMark Murray 			if (len > packet_get_maxsize())
134511b41d2SMark Murray 				len = packet_get_maxsize();
135511b41d2SMark Murray 		}
136511b41d2SMark Murray 		packet_start(SSH_SMSG_STDOUT_DATA);
137511b41d2SMark Murray 		packet_put_string(buffer_ptr(&stdout_buffer), len);
138511b41d2SMark Murray 		packet_send();
139511b41d2SMark Murray 		buffer_consume(&stdout_buffer, len);
140511b41d2SMark Murray 		stdout_bytes += len;
141511b41d2SMark Murray 	}
142511b41d2SMark Murray }
143511b41d2SMark Murray 
144511b41d2SMark Murray /*
145511b41d2SMark Murray  * Sleep in select() until we can do something.  This will initialize the
146511b41d2SMark Murray  * select masks.  Upon return, the masks will indicate which descriptors
147511b41d2SMark Murray  * have data or can accept data.  Optionally, a maximum time can be specified
148511b41d2SMark Murray  * for the duration of the wait (0 = infinite).
149511b41d2SMark Murray  */
150511b41d2SMark Murray void
151511b41d2SMark Murray wait_until_can_do_something(fd_set * readset, fd_set * writeset,
152511b41d2SMark Murray 			    unsigned int max_time_milliseconds)
153511b41d2SMark Murray {
154511b41d2SMark Murray 	struct timeval tv, *tvp;
155511b41d2SMark Murray 	int ret;
156511b41d2SMark Murray 
157511b41d2SMark Murray 	/* When select fails we restart from here. */
158511b41d2SMark Murray retry_select:
159511b41d2SMark Murray 
160511b41d2SMark Murray 	/* Initialize select() masks. */
161511b41d2SMark Murray 	FD_ZERO(readset);
162511b41d2SMark Murray 
163511b41d2SMark Murray 	/*
164511b41d2SMark Murray 	 * Read packets from the client unless we have too much buffered
165511b41d2SMark Murray 	 * stdin or channel data.
166511b41d2SMark Murray 	 */
167a04a10f8SKris Kennaway 	if (compat20) {
168a04a10f8SKris Kennaway 		/* wrong: bad condition XXX */
169a04a10f8SKris Kennaway 		if (channel_not_very_much_buffered_data())
170a04a10f8SKris Kennaway 			FD_SET(connection_in, readset);
171a04a10f8SKris Kennaway 	} else {
172511b41d2SMark Murray 		if (buffer_len(&stdin_buffer) < 4096 &&
173511b41d2SMark Murray 		    channel_not_very_much_buffered_data())
174511b41d2SMark Murray 			FD_SET(connection_in, readset);
175a04a10f8SKris Kennaway 	}
176511b41d2SMark Murray 
177511b41d2SMark Murray 	/*
178511b41d2SMark Murray 	 * If there is not too much data already buffered going to the
179511b41d2SMark Murray 	 * client, try to get some more data from the program.
180511b41d2SMark Murray 	 */
181a04a10f8SKris Kennaway 	if (!compat20 && packet_not_very_much_data_to_write()) {
182511b41d2SMark Murray 		if (!fdout_eof)
183511b41d2SMark Murray 			FD_SET(fdout, readset);
184511b41d2SMark Murray 		if (!fderr_eof)
185511b41d2SMark Murray 			FD_SET(fderr, readset);
186511b41d2SMark Murray 	}
187511b41d2SMark Murray 	FD_ZERO(writeset);
188511b41d2SMark Murray 
189511b41d2SMark Murray 	/* Set masks for channel descriptors. */
190511b41d2SMark Murray 	channel_prepare_select(readset, writeset);
191511b41d2SMark Murray 
192511b41d2SMark Murray 	/*
193511b41d2SMark Murray 	 * If we have buffered packet data going to the client, mark that
194511b41d2SMark Murray 	 * descriptor.
195511b41d2SMark Murray 	 */
196511b41d2SMark Murray 	if (packet_have_data_to_write())
197511b41d2SMark Murray 		FD_SET(connection_out, writeset);
198511b41d2SMark Murray 
199511b41d2SMark Murray 	/* If we have buffered data, try to write some of that data to the
200511b41d2SMark Murray 	   program. */
201a04a10f8SKris Kennaway 	if (!compat20 && fdin != -1 && buffer_len(&stdin_buffer) > 0)
202511b41d2SMark Murray 		FD_SET(fdin, writeset);
203511b41d2SMark Murray 
204511b41d2SMark Murray 	/* Update the maximum descriptor number if appropriate. */
205511b41d2SMark Murray 	if (channel_max_fd() > max_fd)
206511b41d2SMark Murray 		max_fd = channel_max_fd();
207511b41d2SMark Murray 
208511b41d2SMark Murray 	/*
209511b41d2SMark Murray 	 * If child has terminated and there is enough buffer space to read
210511b41d2SMark Murray 	 * from it, then read as much as is available and exit.
211511b41d2SMark Murray 	 */
212511b41d2SMark Murray 	if (child_terminated && packet_not_very_much_data_to_write())
213511b41d2SMark Murray 		if (max_time_milliseconds == 0)
214511b41d2SMark Murray 			max_time_milliseconds = 100;
215511b41d2SMark Murray 
216511b41d2SMark Murray 	if (max_time_milliseconds == 0)
217511b41d2SMark Murray 		tvp = NULL;
218511b41d2SMark Murray 	else {
219511b41d2SMark Murray 		tv.tv_sec = max_time_milliseconds / 1000;
220511b41d2SMark Murray 		tv.tv_usec = 1000 * (max_time_milliseconds % 1000);
221511b41d2SMark Murray 		tvp = &tv;
222511b41d2SMark Murray 	}
223a04a10f8SKris Kennaway 	if (tvp!=NULL)
224a04a10f8SKris Kennaway 		debug("tvp!=NULL kid %d mili %d", child_terminated, max_time_milliseconds);
225511b41d2SMark Murray 
226511b41d2SMark Murray 	/* Wait for something to happen, or the timeout to expire. */
227511b41d2SMark Murray 	ret = select(max_fd + 1, readset, writeset, NULL, tvp);
228511b41d2SMark Murray 
229511b41d2SMark Murray 	if (ret < 0) {
230511b41d2SMark Murray 		if (errno != EINTR)
231511b41d2SMark Murray 			error("select: %.100s", strerror(errno));
232511b41d2SMark Murray 		else
233511b41d2SMark Murray 			goto retry_select;
234511b41d2SMark Murray 	}
235511b41d2SMark Murray }
236511b41d2SMark Murray 
237511b41d2SMark Murray /*
238511b41d2SMark Murray  * Processes input from the client and the program.  Input data is stored
239511b41d2SMark Murray  * in buffers and processed later.
240511b41d2SMark Murray  */
241511b41d2SMark Murray void
242511b41d2SMark Murray process_input(fd_set * readset)
243511b41d2SMark Murray {
244511b41d2SMark Murray 	int len;
245511b41d2SMark Murray 	char buf[16384];
246511b41d2SMark Murray 
247511b41d2SMark Murray 	/* Read and buffer any input data from the client. */
248511b41d2SMark Murray 	if (FD_ISSET(connection_in, readset)) {
249511b41d2SMark Murray 		len = read(connection_in, buf, sizeof(buf));
250511b41d2SMark Murray 		if (len == 0) {
251511b41d2SMark Murray 			verbose("Connection closed by remote host.");
252511b41d2SMark Murray 			fatal_cleanup();
253511b41d2SMark Murray 		}
254511b41d2SMark Murray 		/*
255511b41d2SMark Murray 		 * There is a kernel bug on Solaris that causes select to
256511b41d2SMark Murray 		 * sometimes wake up even though there is no data available.
257511b41d2SMark Murray 		 */
258511b41d2SMark Murray 		if (len < 0 && errno == EAGAIN)
259511b41d2SMark Murray 			len = 0;
260511b41d2SMark Murray 
261511b41d2SMark Murray 		if (len < 0) {
262511b41d2SMark Murray 			verbose("Read error from remote host: %.100s", strerror(errno));
263511b41d2SMark Murray 			fatal_cleanup();
264511b41d2SMark Murray 		}
265511b41d2SMark Murray 		/* Buffer any received data. */
266511b41d2SMark Murray 		packet_process_incoming(buf, len);
267511b41d2SMark Murray 	}
268a04a10f8SKris Kennaway 	if (compat20)
269a04a10f8SKris Kennaway 		return;
270a04a10f8SKris Kennaway 
271511b41d2SMark Murray 	/* Read and buffer any available stdout data from the program. */
272511b41d2SMark Murray 	if (!fdout_eof && FD_ISSET(fdout, readset)) {
273511b41d2SMark Murray 		len = read(fdout, buf, sizeof(buf));
274511b41d2SMark Murray 		if (len <= 0)
275511b41d2SMark Murray 			fdout_eof = 1;
276511b41d2SMark Murray 		else {
277511b41d2SMark Murray 			buffer_append(&stdout_buffer, buf, len);
278511b41d2SMark Murray 			fdout_bytes += len;
279511b41d2SMark Murray 		}
280511b41d2SMark Murray 	}
281511b41d2SMark Murray 	/* Read and buffer any available stderr data from the program. */
282511b41d2SMark Murray 	if (!fderr_eof && FD_ISSET(fderr, readset)) {
283511b41d2SMark Murray 		len = read(fderr, buf, sizeof(buf));
284511b41d2SMark Murray 		if (len <= 0)
285511b41d2SMark Murray 			fderr_eof = 1;
286511b41d2SMark Murray 		else
287511b41d2SMark Murray 			buffer_append(&stderr_buffer, buf, len);
288511b41d2SMark Murray 	}
289511b41d2SMark Murray }
290511b41d2SMark Murray 
291511b41d2SMark Murray /*
292511b41d2SMark Murray  * Sends data from internal buffers to client program stdin.
293511b41d2SMark Murray  */
294511b41d2SMark Murray void
295511b41d2SMark Murray process_output(fd_set * writeset)
296511b41d2SMark Murray {
297511b41d2SMark Murray 	int len;
298511b41d2SMark Murray 
299511b41d2SMark Murray 	/* Write buffered data to program stdin. */
300a04a10f8SKris Kennaway 	if (!compat20 && fdin != -1 && FD_ISSET(fdin, writeset)) {
301511b41d2SMark Murray 		len = write(fdin, buffer_ptr(&stdin_buffer),
302511b41d2SMark Murray 		    buffer_len(&stdin_buffer));
303511b41d2SMark Murray 		if (len <= 0) {
304511b41d2SMark Murray #ifdef USE_PIPES
305511b41d2SMark Murray 			close(fdin);
306511b41d2SMark Murray #else
307a04a10f8SKris Kennaway 			if (fdin != fdout)
308511b41d2SMark Murray 				close(fdin);
309511b41d2SMark Murray 			else
310511b41d2SMark Murray 				shutdown(fdin, SHUT_WR); /* We will no longer send. */
311511b41d2SMark Murray #endif
312511b41d2SMark Murray 			fdin = -1;
313511b41d2SMark Murray 		} else {
314511b41d2SMark Murray 			/* Successful write.  Consume the data from the buffer. */
315511b41d2SMark Murray 			buffer_consume(&stdin_buffer, len);
316511b41d2SMark Murray 			/* Update the count of bytes written to the program. */
317511b41d2SMark Murray 			stdin_bytes += len;
318511b41d2SMark Murray 		}
319511b41d2SMark Murray 	}
320511b41d2SMark Murray 	/* Send any buffered packet data to the client. */
321511b41d2SMark Murray 	if (FD_ISSET(connection_out, writeset))
322511b41d2SMark Murray 		packet_write_poll();
323511b41d2SMark Murray }
324511b41d2SMark Murray 
325511b41d2SMark Murray /*
326511b41d2SMark Murray  * Wait until all buffered output has been sent to the client.
327511b41d2SMark Murray  * This is used when the program terminates.
328511b41d2SMark Murray  */
329511b41d2SMark Murray void
330511b41d2SMark Murray drain_output()
331511b41d2SMark Murray {
332511b41d2SMark Murray 	/* Send any buffered stdout data to the client. */
333511b41d2SMark Murray 	if (buffer_len(&stdout_buffer) > 0) {
334511b41d2SMark Murray 		packet_start(SSH_SMSG_STDOUT_DATA);
335511b41d2SMark Murray 		packet_put_string(buffer_ptr(&stdout_buffer),
336511b41d2SMark Murray 				  buffer_len(&stdout_buffer));
337511b41d2SMark Murray 		packet_send();
338511b41d2SMark Murray 		/* Update the count of sent bytes. */
339511b41d2SMark Murray 		stdout_bytes += buffer_len(&stdout_buffer);
340511b41d2SMark Murray 	}
341511b41d2SMark Murray 	/* Send any buffered stderr data to the client. */
342511b41d2SMark Murray 	if (buffer_len(&stderr_buffer) > 0) {
343511b41d2SMark Murray 		packet_start(SSH_SMSG_STDERR_DATA);
344511b41d2SMark Murray 		packet_put_string(buffer_ptr(&stderr_buffer),
345511b41d2SMark Murray 				  buffer_len(&stderr_buffer));
346511b41d2SMark Murray 		packet_send();
347511b41d2SMark Murray 		/* Update the count of sent bytes. */
348511b41d2SMark Murray 		stderr_bytes += buffer_len(&stderr_buffer);
349511b41d2SMark Murray 	}
350511b41d2SMark Murray 	/* Wait until all buffered data has been written to the client. */
351511b41d2SMark Murray 	packet_write_wait();
352511b41d2SMark Murray }
353511b41d2SMark Murray 
354a04a10f8SKris Kennaway void
355a04a10f8SKris Kennaway process_buffered_input_packets()
356a04a10f8SKris Kennaway {
357a04a10f8SKris Kennaway 	dispatch_run(DISPATCH_NONBLOCK, NULL);
358a04a10f8SKris Kennaway }
359a04a10f8SKris Kennaway 
360511b41d2SMark Murray /*
361511b41d2SMark Murray  * Performs the interactive session.  This handles data transmission between
362511b41d2SMark Murray  * the client and the program.  Note that the notion of stdin, stdout, and
363511b41d2SMark Murray  * stderr in this function is sort of reversed: this function writes to
364511b41d2SMark Murray  * stdin (of the child program), and reads from stdout and stderr (of the
365511b41d2SMark Murray  * child program).
366511b41d2SMark Murray  */
367511b41d2SMark Murray void
368a04a10f8SKris Kennaway server_loop(pid_t pid, int fdin_arg, int fdout_arg, int fderr_arg)
369511b41d2SMark Murray {
370a04a10f8SKris Kennaway 	int wait_status;	/* Status returned by wait(). */
371a04a10f8SKris Kennaway 	pid_t wait_pid;		/* pid returned by wait(). */
372511b41d2SMark Murray 	int waiting_termination = 0;	/* Have displayed waiting close message. */
373511b41d2SMark Murray 	unsigned int max_time_milliseconds;
374511b41d2SMark Murray 	unsigned int previous_stdout_buffer_bytes;
375511b41d2SMark Murray 	unsigned int stdout_buffer_bytes;
376511b41d2SMark Murray 	int type;
377511b41d2SMark Murray 
378511b41d2SMark Murray 	debug("Entering interactive session.");
379511b41d2SMark Murray 
380511b41d2SMark Murray 	/* Initialize the SIGCHLD kludge. */
381511b41d2SMark Murray 	child_pid = pid;
382511b41d2SMark Murray 	child_terminated = 0;
383511b41d2SMark Murray 	signal(SIGCHLD, sigchld_handler);
384511b41d2SMark Murray 
385511b41d2SMark Murray 	/* Initialize our global variables. */
386511b41d2SMark Murray 	fdin = fdin_arg;
387511b41d2SMark Murray 	fdout = fdout_arg;
388511b41d2SMark Murray 	fderr = fderr_arg;
389511b41d2SMark Murray 	connection_in = packet_get_connection_in();
390511b41d2SMark Murray 	connection_out = packet_get_connection_out();
391511b41d2SMark Murray 
392511b41d2SMark Murray 	previous_stdout_buffer_bytes = 0;
393511b41d2SMark Murray 
394511b41d2SMark Murray 	/* Set approximate I/O buffer size. */
395511b41d2SMark Murray 	if (packet_is_interactive())
396511b41d2SMark Murray 		buffer_high = 4096;
397511b41d2SMark Murray 	else
398511b41d2SMark Murray 		buffer_high = 64 * 1024;
399511b41d2SMark Murray 
400511b41d2SMark Murray 	/* Initialize max_fd to the maximum of the known file descriptors. */
401511b41d2SMark Murray 	max_fd = fdin;
402511b41d2SMark Murray 	if (fdout > max_fd)
403511b41d2SMark Murray 		max_fd = fdout;
404511b41d2SMark Murray 	if (fderr != -1 && fderr > max_fd)
405511b41d2SMark Murray 		max_fd = fderr;
406511b41d2SMark Murray 	if (connection_in > max_fd)
407511b41d2SMark Murray 		max_fd = connection_in;
408511b41d2SMark Murray 	if (connection_out > max_fd)
409511b41d2SMark Murray 		max_fd = connection_out;
410511b41d2SMark Murray 
411511b41d2SMark Murray 	/* Initialize Initialize buffers. */
412511b41d2SMark Murray 	buffer_init(&stdin_buffer);
413511b41d2SMark Murray 	buffer_init(&stdout_buffer);
414511b41d2SMark Murray 	buffer_init(&stderr_buffer);
415511b41d2SMark Murray 
416511b41d2SMark Murray 	/*
417511b41d2SMark Murray 	 * If we have no separate fderr (which is the case when we have a pty
418511b41d2SMark Murray 	 * - there we cannot make difference between data sent to stdout and
419511b41d2SMark Murray 	 * stderr), indicate that we have seen an EOF from stderr.  This way
420511b41d2SMark Murray 	 * we don\'t need to check the descriptor everywhere.
421511b41d2SMark Murray 	 */
422511b41d2SMark Murray 	if (fderr == -1)
423511b41d2SMark Murray 		fderr_eof = 1;
424511b41d2SMark Murray 
425a04a10f8SKris Kennaway 	server_init_dispatch();
426a04a10f8SKris Kennaway 
427511b41d2SMark Murray 	/* Main loop of the server for the interactive session mode. */
428511b41d2SMark Murray 	for (;;) {
429511b41d2SMark Murray 		fd_set readset, writeset;
430511b41d2SMark Murray 
431511b41d2SMark Murray 		/* Process buffered packets from the client. */
432511b41d2SMark Murray 		process_buffered_input_packets();
433511b41d2SMark Murray 
434511b41d2SMark Murray 		/*
435511b41d2SMark Murray 		 * If we have received eof, and there is no more pending
436511b41d2SMark Murray 		 * input data, cause a real eof by closing fdin.
437511b41d2SMark Murray 		 */
438511b41d2SMark Murray 		if (stdin_eof && fdin != -1 && buffer_len(&stdin_buffer) == 0) {
439511b41d2SMark Murray #ifdef USE_PIPES
440511b41d2SMark Murray 			close(fdin);
441511b41d2SMark Murray #else
442a04a10f8SKris Kennaway 			if (fdin != fdout)
443511b41d2SMark Murray 				close(fdin);
444511b41d2SMark Murray 			else
445511b41d2SMark Murray 				shutdown(fdin, SHUT_WR); /* We will no longer send. */
446511b41d2SMark Murray #endif
447511b41d2SMark Murray 			fdin = -1;
448511b41d2SMark Murray 		}
449511b41d2SMark Murray 		/* Make packets from buffered stderr data to send to the client. */
450511b41d2SMark Murray 		make_packets_from_stderr_data();
451511b41d2SMark Murray 
452511b41d2SMark Murray 		/*
453511b41d2SMark Murray 		 * Make packets from buffered stdout data to send to the
454511b41d2SMark Murray 		 * client. If there is very little to send, this arranges to
455511b41d2SMark Murray 		 * not send them now, but to wait a short while to see if we
456511b41d2SMark Murray 		 * are getting more data. This is necessary, as some systems
457511b41d2SMark Murray 		 * wake up readers from a pty after each separate character.
458511b41d2SMark Murray 		 */
459511b41d2SMark Murray 		max_time_milliseconds = 0;
460511b41d2SMark Murray 		stdout_buffer_bytes = buffer_len(&stdout_buffer);
461511b41d2SMark Murray 		if (stdout_buffer_bytes != 0 && stdout_buffer_bytes < 256 &&
462511b41d2SMark Murray 		    stdout_buffer_bytes != previous_stdout_buffer_bytes) {
463511b41d2SMark Murray 			/* try again after a while */
464511b41d2SMark Murray 			max_time_milliseconds = 10;
465511b41d2SMark Murray 		} else {
466511b41d2SMark Murray 			/* Send it now. */
467511b41d2SMark Murray 			make_packets_from_stdout_data();
468511b41d2SMark Murray 		}
469511b41d2SMark Murray 		previous_stdout_buffer_bytes = buffer_len(&stdout_buffer);
470511b41d2SMark Murray 
471511b41d2SMark Murray 		/* Send channel data to the client. */
472511b41d2SMark Murray 		if (packet_not_very_much_data_to_write())
473511b41d2SMark Murray 			channel_output_poll();
474511b41d2SMark Murray 
475511b41d2SMark Murray 		/*
476511b41d2SMark Murray 		 * Bail out of the loop if the program has closed its output
477511b41d2SMark Murray 		 * descriptors, and we have no more data to send to the
478511b41d2SMark Murray 		 * client, and there is no pending buffered data.
479511b41d2SMark Murray 		 */
480511b41d2SMark Murray 		if (fdout_eof && fderr_eof && !packet_have_data_to_write() &&
481511b41d2SMark Murray 		    buffer_len(&stdout_buffer) == 0 && buffer_len(&stderr_buffer) == 0) {
482511b41d2SMark Murray 			if (!channel_still_open())
483a04a10f8SKris Kennaway 				break;
484511b41d2SMark Murray 			if (!waiting_termination) {
485511b41d2SMark Murray 				const char *s = "Waiting for forwarded connections to terminate...\r\n";
486511b41d2SMark Murray 				char *cp;
487511b41d2SMark Murray 				waiting_termination = 1;
488511b41d2SMark Murray 				buffer_append(&stderr_buffer, s, strlen(s));
489511b41d2SMark Murray 
490511b41d2SMark Murray 				/* Display list of open channels. */
491511b41d2SMark Murray 				cp = channel_open_message();
492511b41d2SMark Murray 				buffer_append(&stderr_buffer, cp, strlen(cp));
493511b41d2SMark Murray 				xfree(cp);
494511b41d2SMark Murray 			}
495511b41d2SMark Murray 		}
496511b41d2SMark Murray 		/* Sleep in select() until we can do something. */
497511b41d2SMark Murray 		wait_until_can_do_something(&readset, &writeset,
498511b41d2SMark Murray 					    max_time_milliseconds);
499511b41d2SMark Murray 
500511b41d2SMark Murray 		/* Process any channel events. */
501511b41d2SMark Murray 		channel_after_select(&readset, &writeset);
502511b41d2SMark Murray 
503511b41d2SMark Murray 		/* Process input from the client and from program stdout/stderr. */
504511b41d2SMark Murray 		process_input(&readset);
505511b41d2SMark Murray 
506511b41d2SMark Murray 		/* Process output to the client and to program stdin. */
507511b41d2SMark Murray 		process_output(&writeset);
508511b41d2SMark Murray 	}
509511b41d2SMark Murray 
510511b41d2SMark Murray 	/* Cleanup and termination code. */
511511b41d2SMark Murray 
512511b41d2SMark Murray 	/* Wait until all output has been sent to the client. */
513511b41d2SMark Murray 	drain_output();
514511b41d2SMark Murray 
515511b41d2SMark Murray 	debug("End of interactive session; stdin %ld, stdout (read %ld, sent %ld), stderr %ld bytes.",
516511b41d2SMark Murray 	      stdin_bytes, fdout_bytes, stdout_bytes, stderr_bytes);
517511b41d2SMark Murray 
518511b41d2SMark Murray 	/* Free and clear the buffers. */
519511b41d2SMark Murray 	buffer_free(&stdin_buffer);
520511b41d2SMark Murray 	buffer_free(&stdout_buffer);
521511b41d2SMark Murray 	buffer_free(&stderr_buffer);
522511b41d2SMark Murray 
523511b41d2SMark Murray 	/* Close the file descriptors. */
524511b41d2SMark Murray 	if (fdout != -1)
525511b41d2SMark Murray 		close(fdout);
526511b41d2SMark Murray 	fdout = -1;
527511b41d2SMark Murray 	fdout_eof = 1;
528511b41d2SMark Murray 	if (fderr != -1)
529511b41d2SMark Murray 		close(fderr);
530511b41d2SMark Murray 	fderr = -1;
531511b41d2SMark Murray 	fderr_eof = 1;
532511b41d2SMark Murray 	if (fdin != -1)
533511b41d2SMark Murray 		close(fdin);
534511b41d2SMark Murray 	fdin = -1;
535511b41d2SMark Murray 
536511b41d2SMark Murray 	/* Stop listening for channels; this removes unix domain sockets. */
537511b41d2SMark Murray 	channel_stop_listening();
538511b41d2SMark Murray 
539511b41d2SMark Murray 	/* Wait for the child to exit.  Get its exit status. */
540511b41d2SMark Murray 	wait_pid = wait(&wait_status);
541511b41d2SMark Murray 	if (wait_pid < 0) {
542511b41d2SMark Murray 		/*
543511b41d2SMark Murray 		 * It is possible that the wait was handled by SIGCHLD
544511b41d2SMark Murray 		 * handler.  This may result in either: this call
545511b41d2SMark Murray 		 * returning with EINTR, or: this call returning ECHILD.
546511b41d2SMark Murray 		 */
547511b41d2SMark Murray 		if (child_terminated)
548511b41d2SMark Murray 			wait_status = child_wait_status;
549511b41d2SMark Murray 		else
550511b41d2SMark Murray 			packet_disconnect("wait: %.100s", strerror(errno));
551511b41d2SMark Murray 	} else {
552511b41d2SMark Murray 		/* Check if it matches the process we forked. */
553511b41d2SMark Murray 		if (wait_pid != pid)
554511b41d2SMark Murray 			error("Strange, wait returned pid %d, expected %d",
555511b41d2SMark Murray 			       wait_pid, pid);
556511b41d2SMark Murray 	}
557511b41d2SMark Murray 
558511b41d2SMark Murray 	/* We no longer want our SIGCHLD handler to be called. */
559511b41d2SMark Murray 	signal(SIGCHLD, SIG_DFL);
560511b41d2SMark Murray 
561511b41d2SMark Murray 	/* Check if it exited normally. */
562511b41d2SMark Murray 	if (WIFEXITED(wait_status)) {
563511b41d2SMark Murray 		/* Yes, normal exit.  Get exit status and send it to the client. */
564511b41d2SMark Murray 		debug("Command exited with status %d.", WEXITSTATUS(wait_status));
565511b41d2SMark Murray 		packet_start(SSH_SMSG_EXITSTATUS);
566511b41d2SMark Murray 		packet_put_int(WEXITSTATUS(wait_status));
567511b41d2SMark Murray 		packet_send();
568511b41d2SMark Murray 		packet_write_wait();
569511b41d2SMark Murray 
570511b41d2SMark Murray 		/*
571511b41d2SMark Murray 		 * Wait for exit confirmation.  Note that there might be
572511b41d2SMark Murray 		 * other packets coming before it; however, the program has
573511b41d2SMark Murray 		 * already died so we just ignore them.  The client is
574511b41d2SMark Murray 		 * supposed to respond with the confirmation when it receives
575511b41d2SMark Murray 		 * the exit status.
576511b41d2SMark Murray 		 */
577511b41d2SMark Murray 		do {
578511b41d2SMark Murray 			int plen;
579511b41d2SMark Murray 			type = packet_read(&plen);
580511b41d2SMark Murray 		}
581511b41d2SMark Murray 		while (type != SSH_CMSG_EXIT_CONFIRMATION);
582511b41d2SMark Murray 
583511b41d2SMark Murray 		debug("Received exit confirmation.");
584511b41d2SMark Murray 		return;
585511b41d2SMark Murray 	}
586511b41d2SMark Murray 	/* Check if the program terminated due to a signal. */
587511b41d2SMark Murray 	if (WIFSIGNALED(wait_status))
588511b41d2SMark Murray 		packet_disconnect("Command terminated on signal %d.",
589511b41d2SMark Murray 				  WTERMSIG(wait_status));
590511b41d2SMark Murray 
591511b41d2SMark Murray 	/* Some weird exit cause.  Just exit. */
592511b41d2SMark Murray 	packet_disconnect("wait returned status %04x.", wait_status);
593511b41d2SMark Murray 	/* NOTREACHED */
594511b41d2SMark Murray }
595a04a10f8SKris Kennaway 
596a04a10f8SKris Kennaway void
597a04a10f8SKris Kennaway server_loop2(void)
598a04a10f8SKris Kennaway {
599a04a10f8SKris Kennaway 	fd_set readset, writeset;
600a04a10f8SKris Kennaway 	int had_channel = 0;
601a04a10f8SKris Kennaway 	int status;
602a04a10f8SKris Kennaway 	pid_t pid;
603a04a10f8SKris Kennaway 
604a04a10f8SKris Kennaway 	debug("Entering interactive session for SSH2.");
605a04a10f8SKris Kennaway 
606a04a10f8SKris Kennaway 	signal(SIGCHLD, sigchld_handler2);
607a04a10f8SKris Kennaway 	child_terminated = 0;
608a04a10f8SKris Kennaway 	connection_in = packet_get_connection_in();
609a04a10f8SKris Kennaway 	connection_out = packet_get_connection_out();
610a04a10f8SKris Kennaway 	max_fd = connection_in;
611a04a10f8SKris Kennaway 	if (connection_out > max_fd)
612a04a10f8SKris Kennaway 		max_fd = connection_out;
613a04a10f8SKris Kennaway 	server_init_dispatch();
614a04a10f8SKris Kennaway 
615a04a10f8SKris Kennaway 	for (;;) {
616a04a10f8SKris Kennaway 		process_buffered_input_packets();
617a04a10f8SKris Kennaway 		if (!had_channel && channel_still_open())
618a04a10f8SKris Kennaway 			had_channel = 1;
619a04a10f8SKris Kennaway 		if (had_channel && !channel_still_open()) {
620a04a10f8SKris Kennaway 			debug("!channel_still_open.");
621a04a10f8SKris Kennaway 			break;
622a04a10f8SKris Kennaway 		}
623a04a10f8SKris Kennaway 		if (packet_not_very_much_data_to_write())
624a04a10f8SKris Kennaway 			channel_output_poll();
625a04a10f8SKris Kennaway 		wait_until_can_do_something(&readset, &writeset, 0);
626a04a10f8SKris Kennaway 		if (child_terminated) {
627a04a10f8SKris Kennaway 			while ((pid = waitpid(-1, &status, WNOHANG)) > 0)
628a04a10f8SKris Kennaway 				session_close_by_pid(pid, status);
629a04a10f8SKris Kennaway 			child_terminated = 0;
630a04a10f8SKris Kennaway 		}
631a04a10f8SKris Kennaway 		channel_after_select(&readset, &writeset);
632a04a10f8SKris Kennaway 		process_input(&readset);
633a04a10f8SKris Kennaway 		process_output(&writeset);
634a04a10f8SKris Kennaway 	}
635a04a10f8SKris Kennaway 	signal(SIGCHLD, SIG_DFL);
636a04a10f8SKris Kennaway 	while ((pid = waitpid(-1, &status, WNOHANG)) > 0)
637a04a10f8SKris Kennaway 		session_close_by_pid(pid, status);
638a04a10f8SKris Kennaway 	channel_stop_listening();
639a04a10f8SKris Kennaway }
640a04a10f8SKris Kennaway 
641a04a10f8SKris Kennaway void
642a04a10f8SKris Kennaway server_input_stdin_data(int type, int plen)
643a04a10f8SKris Kennaway {
644a04a10f8SKris Kennaway 	char *data;
645a04a10f8SKris Kennaway 	unsigned int data_len;
646a04a10f8SKris Kennaway 
647a04a10f8SKris Kennaway 	/* Stdin data from the client.  Append it to the buffer. */
648a04a10f8SKris Kennaway 	/* Ignore any data if the client has closed stdin. */
649a04a10f8SKris Kennaway 	if (fdin == -1)
650a04a10f8SKris Kennaway 		return;
651a04a10f8SKris Kennaway 	data = packet_get_string(&data_len);
652a04a10f8SKris Kennaway 	packet_integrity_check(plen, (4 + data_len), type);
653a04a10f8SKris Kennaway 	buffer_append(&stdin_buffer, data, data_len);
654a04a10f8SKris Kennaway 	memset(data, 0, data_len);
655a04a10f8SKris Kennaway 	xfree(data);
656a04a10f8SKris Kennaway }
657a04a10f8SKris Kennaway 
658a04a10f8SKris Kennaway void
659a04a10f8SKris Kennaway server_input_eof(int type, int plen)
660a04a10f8SKris Kennaway {
661a04a10f8SKris Kennaway 	/*
662a04a10f8SKris Kennaway 	 * Eof from the client.  The stdin descriptor to the
663a04a10f8SKris Kennaway 	 * program will be closed when all buffered data has
664a04a10f8SKris Kennaway 	 * drained.
665a04a10f8SKris Kennaway 	 */
666a04a10f8SKris Kennaway 	debug("EOF received for stdin.");
667a04a10f8SKris Kennaway 	packet_integrity_check(plen, 0, type);
668a04a10f8SKris Kennaway 	stdin_eof = 1;
669a04a10f8SKris Kennaway }
670a04a10f8SKris Kennaway 
671a04a10f8SKris Kennaway void
672a04a10f8SKris Kennaway server_input_window_size(int type, int plen)
673a04a10f8SKris Kennaway {
674a04a10f8SKris Kennaway 	int row = packet_get_int();
675a04a10f8SKris Kennaway 	int col = packet_get_int();
676a04a10f8SKris Kennaway 	int xpixel = packet_get_int();
677a04a10f8SKris Kennaway 	int ypixel = packet_get_int();
678a04a10f8SKris Kennaway 
679a04a10f8SKris Kennaway 	debug("Window change received.");
680a04a10f8SKris Kennaway 	packet_integrity_check(plen, 4 * 4, type);
681a04a10f8SKris Kennaway 	if (fdin != -1)
682a04a10f8SKris Kennaway 		pty_change_window_size(fdin, row, col, xpixel, ypixel);
683a04a10f8SKris Kennaway }
684a04a10f8SKris Kennaway 
685a04a10f8SKris Kennaway int
686a04a10f8SKris Kennaway input_direct_tcpip(void)
687a04a10f8SKris Kennaway {
688a04a10f8SKris Kennaway 	int sock;
689a04a10f8SKris Kennaway 	char *target, *originator;
690a04a10f8SKris Kennaway 	int target_port, originator_port;
691a04a10f8SKris Kennaway 
692a04a10f8SKris Kennaway 	target = packet_get_string(NULL);
693a04a10f8SKris Kennaway 	target_port = packet_get_int();
694a04a10f8SKris Kennaway 	originator = packet_get_string(NULL);
695a04a10f8SKris Kennaway 	originator_port = packet_get_int();
696a04a10f8SKris Kennaway 	packet_done();
697a04a10f8SKris Kennaway 	/* XXX check permission */
698a04a10f8SKris Kennaway 	sock = channel_connect_to(target, target_port);
699a04a10f8SKris Kennaway 	xfree(target);
700a04a10f8SKris Kennaway 	xfree(originator);
701a04a10f8SKris Kennaway 	if (sock < 0)
702a04a10f8SKris Kennaway 		return -1;
703a04a10f8SKris Kennaway 	return channel_new("direct-tcpip", SSH_CHANNEL_OPEN,
704a04a10f8SKris Kennaway 	    sock, sock, -1, 4*1024, 32*1024, 0, xstrdup("direct-tcpip"));
705a04a10f8SKris Kennaway }
706a04a10f8SKris Kennaway 
707a04a10f8SKris Kennaway void
708a04a10f8SKris Kennaway server_input_channel_open(int type, int plen)
709a04a10f8SKris Kennaway {
710a04a10f8SKris Kennaway 	Channel *c = NULL;
711a04a10f8SKris Kennaway 	char *ctype;
712a04a10f8SKris Kennaway 	int id;
713a04a10f8SKris Kennaway 	unsigned int len;
714a04a10f8SKris Kennaway 	int rchan;
715a04a10f8SKris Kennaway 	int rmaxpack;
716a04a10f8SKris Kennaway 	int rwindow;
717a04a10f8SKris Kennaway 
718a04a10f8SKris Kennaway 	ctype = packet_get_string(&len);
719a04a10f8SKris Kennaway 	rchan = packet_get_int();
720a04a10f8SKris Kennaway 	rwindow = packet_get_int();
721a04a10f8SKris Kennaway 	rmaxpack = packet_get_int();
722a04a10f8SKris Kennaway 
723a04a10f8SKris Kennaway 	debug("channel_input_open: ctype %s rchan %d win %d max %d",
724a04a10f8SKris Kennaway 	    ctype, rchan, rwindow, rmaxpack);
725a04a10f8SKris Kennaway 
726a04a10f8SKris Kennaway 	if (strcmp(ctype, "session") == 0) {
727a04a10f8SKris Kennaway 		debug("open session");
728a04a10f8SKris Kennaway 		packet_done();
729a04a10f8SKris Kennaway 		/*
730a04a10f8SKris Kennaway 		 * A server session has no fd to read or write
731a04a10f8SKris Kennaway 		 * until a CHANNEL_REQUEST for a shell is made,
732a04a10f8SKris Kennaway 		 * so we set the type to SSH_CHANNEL_LARVAL.
733a04a10f8SKris Kennaway 		 * Additionally, a callback for handling all
734a04a10f8SKris Kennaway 		 * CHANNEL_REQUEST messages is registered.
735a04a10f8SKris Kennaway 		 */
736a04a10f8SKris Kennaway 		id = channel_new(ctype, SSH_CHANNEL_LARVAL,
737a04a10f8SKris Kennaway 		    -1, -1, -1, 0, 32*1024, 0, xstrdup("server-session"));
738a04a10f8SKris Kennaway 		if (session_open(id) == 1) {
739a04a10f8SKris Kennaway 			channel_register_callback(id, SSH2_MSG_CHANNEL_REQUEST,
740a04a10f8SKris Kennaway 			    session_input_channel_req, (void *)0);
741a04a10f8SKris Kennaway 			channel_register_cleanup(id, session_close_by_channel);
742a04a10f8SKris Kennaway 			c = channel_lookup(id);
743a04a10f8SKris Kennaway 		} else {
744a04a10f8SKris Kennaway 			debug("session open failed, free channel %d", id);
745a04a10f8SKris Kennaway 			channel_free(id);
746a04a10f8SKris Kennaway 		}
747a04a10f8SKris Kennaway 	} else if (strcmp(ctype, "direct-tcpip") == 0) {
748a04a10f8SKris Kennaway 		debug("open direct-tcpip");
749a04a10f8SKris Kennaway 		id = input_direct_tcpip();
750a04a10f8SKris Kennaway 		if (id >= 0)
751a04a10f8SKris Kennaway 			c = channel_lookup(id);
752a04a10f8SKris Kennaway 	}
753a04a10f8SKris Kennaway 	if (c != NULL) {
754a04a10f8SKris Kennaway 		debug("confirm %s", ctype);
755a04a10f8SKris Kennaway 		c->remote_id = rchan;
756a04a10f8SKris Kennaway 		c->remote_window = rwindow;
757a04a10f8SKris Kennaway 		c->remote_maxpacket = rmaxpack;
758a04a10f8SKris Kennaway 
759a04a10f8SKris Kennaway 		packet_start(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION);
760a04a10f8SKris Kennaway 		packet_put_int(c->remote_id);
761a04a10f8SKris Kennaway 		packet_put_int(c->self);
762a04a10f8SKris Kennaway 		packet_put_int(c->local_window);
763a04a10f8SKris Kennaway 		packet_put_int(c->local_maxpacket);
764a04a10f8SKris Kennaway 		packet_send();
765a04a10f8SKris Kennaway 	} else {
766a04a10f8SKris Kennaway 		debug("failure %s", ctype);
767a04a10f8SKris Kennaway 		packet_start(SSH2_MSG_CHANNEL_OPEN_FAILURE);
768a04a10f8SKris Kennaway 		packet_put_int(rchan);
769a04a10f8SKris Kennaway 		packet_put_int(SSH2_OPEN_ADMINISTRATIVELY_PROHIBITED);
770a04a10f8SKris Kennaway 		packet_put_cstring("bla bla");
771a04a10f8SKris Kennaway 		packet_put_cstring("");
772a04a10f8SKris Kennaway 		packet_send();
773a04a10f8SKris Kennaway 	}
774a04a10f8SKris Kennaway 	xfree(ctype);
775a04a10f8SKris Kennaway }
776a04a10f8SKris Kennaway 
777a04a10f8SKris Kennaway void
778a04a10f8SKris Kennaway server_init_dispatch_20()
779a04a10f8SKris Kennaway {
780a04a10f8SKris Kennaway 	debug("server_init_dispatch_20");
781a04a10f8SKris Kennaway 	dispatch_init(&dispatch_protocol_error);
782a04a10f8SKris Kennaway 	dispatch_set(SSH2_MSG_CHANNEL_CLOSE, &channel_input_oclose);
783a04a10f8SKris Kennaway 	dispatch_set(SSH2_MSG_CHANNEL_DATA, &channel_input_data);
784a04a10f8SKris Kennaway 	dispatch_set(SSH2_MSG_CHANNEL_EOF, &channel_input_ieof);
785a04a10f8SKris Kennaway 	dispatch_set(SSH2_MSG_CHANNEL_EXTENDED_DATA, &channel_input_extended_data);
786a04a10f8SKris Kennaway 	dispatch_set(SSH2_MSG_CHANNEL_OPEN, &server_input_channel_open);
787a04a10f8SKris Kennaway 	dispatch_set(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION, &channel_input_open_confirmation);
788a04a10f8SKris Kennaway 	dispatch_set(SSH2_MSG_CHANNEL_OPEN_FAILURE, &channel_input_open_failure);
789a04a10f8SKris Kennaway 	dispatch_set(SSH2_MSG_CHANNEL_REQUEST, &channel_input_channel_request);
790a04a10f8SKris Kennaway 	dispatch_set(SSH2_MSG_CHANNEL_WINDOW_ADJUST, &channel_input_window_adjust);
791a04a10f8SKris Kennaway }
792a04a10f8SKris Kennaway void
793a04a10f8SKris Kennaway server_init_dispatch_13()
794a04a10f8SKris Kennaway {
795a04a10f8SKris Kennaway 	debug("server_init_dispatch_13");
796a04a10f8SKris Kennaway 	dispatch_init(NULL);
797a04a10f8SKris Kennaway 	dispatch_set(SSH_CMSG_EOF, &server_input_eof);
798a04a10f8SKris Kennaway 	dispatch_set(SSH_CMSG_STDIN_DATA, &server_input_stdin_data);
799a04a10f8SKris Kennaway 	dispatch_set(SSH_CMSG_WINDOW_SIZE, &server_input_window_size);
800a04a10f8SKris Kennaway 	dispatch_set(SSH_MSG_CHANNEL_CLOSE, &channel_input_close);
801a04a10f8SKris Kennaway 	dispatch_set(SSH_MSG_CHANNEL_CLOSE_CONFIRMATION, &channel_input_close_confirmation);
802a04a10f8SKris Kennaway 	dispatch_set(SSH_MSG_CHANNEL_DATA, &channel_input_data);
803a04a10f8SKris Kennaway 	dispatch_set(SSH_MSG_CHANNEL_OPEN_CONFIRMATION, &channel_input_open_confirmation);
804a04a10f8SKris Kennaway 	dispatch_set(SSH_MSG_CHANNEL_OPEN_FAILURE, &channel_input_open_failure);
805a04a10f8SKris Kennaway 	dispatch_set(SSH_MSG_PORT_OPEN, &channel_input_port_open);
806a04a10f8SKris Kennaway }
807a04a10f8SKris Kennaway void
808a04a10f8SKris Kennaway server_init_dispatch_15()
809a04a10f8SKris Kennaway {
810a04a10f8SKris Kennaway 	server_init_dispatch_13();
811a04a10f8SKris Kennaway 	debug("server_init_dispatch_15");
812a04a10f8SKris Kennaway 	dispatch_set(SSH_MSG_CHANNEL_CLOSE, &channel_input_ieof);
813a04a10f8SKris Kennaway 	dispatch_set(SSH_MSG_CHANNEL_CLOSE_CONFIRMATION, &channel_input_oclose);
814a04a10f8SKris Kennaway }
815a04a10f8SKris Kennaway void
816a04a10f8SKris Kennaway server_init_dispatch()
817a04a10f8SKris Kennaway {
818a04a10f8SKris Kennaway 	if (compat20)
819a04a10f8SKris Kennaway 		server_init_dispatch_20();
820a04a10f8SKris Kennaway 	else if (compat13)
821a04a10f8SKris Kennaway 		server_init_dispatch_13();
822a04a10f8SKris Kennaway 	else
823a04a10f8SKris Kennaway 		server_init_dispatch_15();
824a04a10f8SKris Kennaway }
825