]> asedeno.scripts.mit.edu Git - PuTTY.git/blob - ssh.c
A major purpose of PuTTY's memory-allocation functions is to succeed or die
[PuTTY.git] / ssh.c
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <stdarg.h>
4 #include <assert.h>
5
6 #include "putty.h"
7 #include "tree234.h"
8 #include "ssh.h"
9
10 #ifndef FALSE
11 #define FALSE 0
12 #endif
13 #ifndef TRUE
14 #define TRUE 1
15 #endif
16
17 #define SSH1_MSG_DISCONNECT                       1     /* 0x1 */
18 #define SSH1_SMSG_PUBLIC_KEY                      2     /* 0x2 */
19 #define SSH1_CMSG_SESSION_KEY                     3     /* 0x3 */
20 #define SSH1_CMSG_USER                            4     /* 0x4 */
21 #define SSH1_CMSG_AUTH_RSA                        6     /* 0x6 */
22 #define SSH1_SMSG_AUTH_RSA_CHALLENGE              7     /* 0x7 */
23 #define SSH1_CMSG_AUTH_RSA_RESPONSE               8     /* 0x8 */
24 #define SSH1_CMSG_AUTH_PASSWORD                   9     /* 0x9 */
25 #define SSH1_CMSG_REQUEST_PTY                     10    /* 0xa */
26 #define SSH1_CMSG_WINDOW_SIZE                     11    /* 0xb */
27 #define SSH1_CMSG_EXEC_SHELL                      12    /* 0xc */
28 #define SSH1_CMSG_EXEC_CMD                        13    /* 0xd */
29 #define SSH1_SMSG_SUCCESS                         14    /* 0xe */
30 #define SSH1_SMSG_FAILURE                         15    /* 0xf */
31 #define SSH1_CMSG_STDIN_DATA                      16    /* 0x10 */
32 #define SSH1_SMSG_STDOUT_DATA                     17    /* 0x11 */
33 #define SSH1_SMSG_STDERR_DATA                     18    /* 0x12 */
34 #define SSH1_CMSG_EOF                             19    /* 0x13 */
35 #define SSH1_SMSG_EXIT_STATUS                     20    /* 0x14 */
36 #define SSH1_MSG_CHANNEL_OPEN_CONFIRMATION        21    /* 0x15 */
37 #define SSH1_MSG_CHANNEL_OPEN_FAILURE             22    /* 0x16 */
38 #define SSH1_MSG_CHANNEL_DATA                     23    /* 0x17 */
39 #define SSH1_MSG_CHANNEL_CLOSE                    24    /* 0x18 */
40 #define SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION       25    /* 0x19 */
41 #define SSH1_SMSG_X11_OPEN                        27    /* 0x1b */
42 #define SSH1_CMSG_PORT_FORWARD_REQUEST            28    /* 0x1c */
43 #define SSH1_MSG_PORT_OPEN                        29    /* 0x1d */
44 #define SSH1_CMSG_AGENT_REQUEST_FORWARDING        30    /* 0x1e */
45 #define SSH1_SMSG_AGENT_OPEN                      31    /* 0x1f */
46 #define SSH1_MSG_IGNORE                           32    /* 0x20 */
47 #define SSH1_CMSG_EXIT_CONFIRMATION               33    /* 0x21 */
48 #define SSH1_CMSG_X11_REQUEST_FORWARDING          34    /* 0x22 */
49 #define SSH1_CMSG_AUTH_RHOSTS_RSA                 35    /* 0x23 */
50 #define SSH1_MSG_DEBUG                            36    /* 0x24 */
51 #define SSH1_CMSG_REQUEST_COMPRESSION             37    /* 0x25 */
52 #define SSH1_CMSG_AUTH_TIS                        39    /* 0x27 */
53 #define SSH1_SMSG_AUTH_TIS_CHALLENGE              40    /* 0x28 */
54 #define SSH1_CMSG_AUTH_TIS_RESPONSE               41    /* 0x29 */
55 #define SSH1_CMSG_AUTH_CCARD                      70    /* 0x46 */
56 #define SSH1_SMSG_AUTH_CCARD_CHALLENGE            71    /* 0x47 */
57 #define SSH1_CMSG_AUTH_CCARD_RESPONSE             72    /* 0x48 */
58
59 #define SSH1_AUTH_TIS                             5     /* 0x5 */
60 #define SSH1_AUTH_CCARD                           16    /* 0x10 */
61
62 #define SSH1_PROTOFLAG_SCREEN_NUMBER              1     /* 0x1 */
63 /* Mask for protoflags we will echo back to server if seen */
64 #define SSH1_PROTOFLAGS_SUPPORTED                 0     /* 0x1 */
65
66 #define SSH2_MSG_DISCONNECT                       1     /* 0x1 */
67 #define SSH2_MSG_IGNORE                           2     /* 0x2 */
68 #define SSH2_MSG_UNIMPLEMENTED                    3     /* 0x3 */
69 #define SSH2_MSG_DEBUG                            4     /* 0x4 */
70 #define SSH2_MSG_SERVICE_REQUEST                  5     /* 0x5 */
71 #define SSH2_MSG_SERVICE_ACCEPT                   6     /* 0x6 */
72 #define SSH2_MSG_KEXINIT                          20    /* 0x14 */
73 #define SSH2_MSG_NEWKEYS                          21    /* 0x15 */
74 #define SSH2_MSG_KEXDH_INIT                       30    /* 0x1e */
75 #define SSH2_MSG_KEXDH_REPLY                      31    /* 0x1f */
76 #define SSH2_MSG_KEX_DH_GEX_REQUEST               30    /* 0x1e */
77 #define SSH2_MSG_KEX_DH_GEX_GROUP                 31    /* 0x1f */
78 #define SSH2_MSG_KEX_DH_GEX_INIT                  32    /* 0x20 */
79 #define SSH2_MSG_KEX_DH_GEX_REPLY                 33    /* 0x21 */
80 #define SSH2_MSG_USERAUTH_REQUEST                 50    /* 0x32 */
81 #define SSH2_MSG_USERAUTH_FAILURE                 51    /* 0x33 */
82 #define SSH2_MSG_USERAUTH_SUCCESS                 52    /* 0x34 */
83 #define SSH2_MSG_USERAUTH_BANNER                  53    /* 0x35 */
84 #define SSH2_MSG_USERAUTH_PK_OK                   60    /* 0x3c */
85 #define SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ        60    /* 0x3c */
86 #define SSH2_MSG_USERAUTH_INFO_REQUEST            60    /* 0x3c */
87 #define SSH2_MSG_USERAUTH_INFO_RESPONSE           61    /* 0x3d */
88 #define SSH2_MSG_GLOBAL_REQUEST                   80    /* 0x50 */
89 #define SSH2_MSG_REQUEST_SUCCESS                  81    /* 0x51 */
90 #define SSH2_MSG_REQUEST_FAILURE                  82    /* 0x52 */
91 #define SSH2_MSG_CHANNEL_OPEN                     90    /* 0x5a */
92 #define SSH2_MSG_CHANNEL_OPEN_CONFIRMATION        91    /* 0x5b */
93 #define SSH2_MSG_CHANNEL_OPEN_FAILURE             92    /* 0x5c */
94 #define SSH2_MSG_CHANNEL_WINDOW_ADJUST            93    /* 0x5d */
95 #define SSH2_MSG_CHANNEL_DATA                     94    /* 0x5e */
96 #define SSH2_MSG_CHANNEL_EXTENDED_DATA            95    /* 0x5f */
97 #define SSH2_MSG_CHANNEL_EOF                      96    /* 0x60 */
98 #define SSH2_MSG_CHANNEL_CLOSE                    97    /* 0x61 */
99 #define SSH2_MSG_CHANNEL_REQUEST                  98    /* 0x62 */
100 #define SSH2_MSG_CHANNEL_SUCCESS                  99    /* 0x63 */
101 #define SSH2_MSG_CHANNEL_FAILURE                  100   /* 0x64 */
102
103 /*
104  * Packet type contexts, so that ssh2_pkt_type can correctly decode
105  * the ambiguous type numbers back into the correct type strings.
106  */
107 #define SSH2_PKTCTX_DHGROUP          0x0001
108 #define SSH2_PKTCTX_DHGEX            0x0002
109 #define SSH2_PKTCTX_KEX_MASK         0x000F
110 #define SSH2_PKTCTX_PUBLICKEY        0x0010
111 #define SSH2_PKTCTX_PASSWORD         0x0020
112 #define SSH2_PKTCTX_KBDINTER         0x0040
113 #define SSH2_PKTCTX_AUTH_MASK        0x00F0
114
115 #define SSH2_DISCONNECT_HOST_NOT_ALLOWED_TO_CONNECT 1   /* 0x1 */
116 #define SSH2_DISCONNECT_PROTOCOL_ERROR            2     /* 0x2 */
117 #define SSH2_DISCONNECT_KEY_EXCHANGE_FAILED       3     /* 0x3 */
118 #define SSH2_DISCONNECT_HOST_AUTHENTICATION_FAILED 4    /* 0x4 */
119 #define SSH2_DISCONNECT_MAC_ERROR                 5     /* 0x5 */
120 #define SSH2_DISCONNECT_COMPRESSION_ERROR         6     /* 0x6 */
121 #define SSH2_DISCONNECT_SERVICE_NOT_AVAILABLE     7     /* 0x7 */
122 #define SSH2_DISCONNECT_PROTOCOL_VERSION_NOT_SUPPORTED 8        /* 0x8 */
123 #define SSH2_DISCONNECT_HOST_KEY_NOT_VERIFIABLE   9     /* 0x9 */
124 #define SSH2_DISCONNECT_CONNECTION_LOST           10    /* 0xa */
125 #define SSH2_DISCONNECT_BY_APPLICATION            11    /* 0xb */
126 #define SSH2_DISCONNECT_TOO_MANY_CONNECTIONS      12    /* 0xc */
127 #define SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER    13    /* 0xd */
128 #define SSH2_DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE 14       /* 0xe */
129 #define SSH2_DISCONNECT_ILLEGAL_USER_NAME         15    /* 0xf */
130
131 static const char *const ssh2_disconnect_reasons[] = {
132     NULL,
133     "SSH_DISCONNECT_HOST_NOT_ALLOWED_TO_CONNECT",
134     "SSH_DISCONNECT_PROTOCOL_ERROR",
135     "SSH_DISCONNECT_KEY_EXCHANGE_FAILED",
136     "SSH_DISCONNECT_HOST_AUTHENTICATION_FAILED",
137     "SSH_DISCONNECT_MAC_ERROR",
138     "SSH_DISCONNECT_COMPRESSION_ERROR",
139     "SSH_DISCONNECT_SERVICE_NOT_AVAILABLE",
140     "SSH_DISCONNECT_PROTOCOL_VERSION_NOT_SUPPORTED",
141     "SSH_DISCONNECT_HOST_KEY_NOT_VERIFIABLE",
142     "SSH_DISCONNECT_CONNECTION_LOST",
143     "SSH_DISCONNECT_BY_APPLICATION",
144     "SSH_DISCONNECT_TOO_MANY_CONNECTIONS",
145     "SSH_DISCONNECT_AUTH_CANCELLED_BY_USER",
146     "SSH_DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE",
147     "SSH_DISCONNECT_ILLEGAL_USER_NAME",
148 };
149
150 #define SSH2_OPEN_ADMINISTRATIVELY_PROHIBITED     1     /* 0x1 */
151 #define SSH2_OPEN_CONNECT_FAILED                  2     /* 0x2 */
152 #define SSH2_OPEN_UNKNOWN_CHANNEL_TYPE            3     /* 0x3 */
153 #define SSH2_OPEN_RESOURCE_SHORTAGE               4     /* 0x4 */
154
155 #define SSH2_EXTENDED_DATA_STDERR                 1     /* 0x1 */
156
157 /*
158  * Various remote-bug flags.
159  */
160 #define BUG_CHOKES_ON_SSH1_IGNORE                 1
161 #define BUG_SSH2_HMAC                             2
162 #define BUG_NEEDS_SSH1_PLAIN_PASSWORD             4
163 #define BUG_CHOKES_ON_RSA                         8
164 #define BUG_SSH2_RSA_PADDING                     16
165 #define BUG_SSH2_DERIVEKEY                       32
166 #define BUG_SSH2_REKEY                           64
167 #define BUG_SSH2_PK_SESSIONID                   128
168
169 /*
170  * Codes for terminal modes.
171  * Most of these are the same in SSH-1 and SSH-2.
172  * This list is derived from draft-ietf-secsh-connect-25 and
173  * SSH-1 RFC-1.2.31.
174  */
175 static const struct {
176     const char* const mode;
177     int opcode;
178     enum { TTY_OP_CHAR, TTY_OP_BOOL } type;
179 } ssh_ttymodes[] = {
180     /* "V" prefix discarded for special characters relative to SSH specs */
181     { "INTR",         1, TTY_OP_CHAR },
182     { "QUIT",         2, TTY_OP_CHAR },
183     { "ERASE",        3, TTY_OP_CHAR },
184     { "KILL",         4, TTY_OP_CHAR },
185     { "EOF",          5, TTY_OP_CHAR },
186     { "EOL",          6, TTY_OP_CHAR },
187     { "EOL2",         7, TTY_OP_CHAR },
188     { "START",        8, TTY_OP_CHAR },
189     { "STOP",         9, TTY_OP_CHAR },
190     { "SUSP",        10, TTY_OP_CHAR },
191     { "DSUSP",       11, TTY_OP_CHAR },
192     { "REPRINT",     12, TTY_OP_CHAR },
193     { "WERASE",      13, TTY_OP_CHAR },
194     { "LNEXT",       14, TTY_OP_CHAR },
195     { "FLUSH",       15, TTY_OP_CHAR },
196     { "SWTCH",       16, TTY_OP_CHAR },
197     { "STATUS",      17, TTY_OP_CHAR },
198     { "DISCARD",     18, TTY_OP_CHAR },
199     { "IGNPAR",      30, TTY_OP_BOOL },
200     { "PARMRK",      31, TTY_OP_BOOL },
201     { "INPCK",       32, TTY_OP_BOOL },
202     { "ISTRIP",      33, TTY_OP_BOOL },
203     { "INLCR",       34, TTY_OP_BOOL },
204     { "IGNCR",       35, TTY_OP_BOOL },
205     { "ICRNL",       36, TTY_OP_BOOL },
206     { "IUCLC",       37, TTY_OP_BOOL },
207     { "IXON",        38, TTY_OP_BOOL },
208     { "IXANY",       39, TTY_OP_BOOL },
209     { "IXOFF",       40, TTY_OP_BOOL },
210     { "IMAXBEL",     41, TTY_OP_BOOL },
211     { "ISIG",        50, TTY_OP_BOOL },
212     { "ICANON",      51, TTY_OP_BOOL },
213     { "XCASE",       52, TTY_OP_BOOL },
214     { "ECHO",        53, TTY_OP_BOOL },
215     { "ECHOE",       54, TTY_OP_BOOL },
216     { "ECHOK",       55, TTY_OP_BOOL },
217     { "ECHONL",      56, TTY_OP_BOOL },
218     { "NOFLSH",      57, TTY_OP_BOOL },
219     { "TOSTOP",      58, TTY_OP_BOOL },
220     { "IEXTEN",      59, TTY_OP_BOOL },
221     { "ECHOCTL",     60, TTY_OP_BOOL },
222     { "ECHOKE",      61, TTY_OP_BOOL },
223     { "PENDIN",      62, TTY_OP_BOOL }, /* XXX is this a real mode? */
224     { "OPOST",       70, TTY_OP_BOOL },
225     { "OLCUC",       71, TTY_OP_BOOL },
226     { "ONLCR",       72, TTY_OP_BOOL },
227     { "OCRNL",       73, TTY_OP_BOOL },
228     { "ONOCR",       74, TTY_OP_BOOL },
229     { "ONLRET",      75, TTY_OP_BOOL },
230     { "CS7",         90, TTY_OP_BOOL },
231     { "CS8",         91, TTY_OP_BOOL },
232     { "PARENB",      92, TTY_OP_BOOL },
233     { "PARODD",      93, TTY_OP_BOOL }
234 };
235
236 /* Miscellaneous other tty-related constants. */
237 #define SSH_TTY_OP_END            0
238 /* The opcodes for ISPEED/OSPEED differ between SSH-1 and SSH-2. */
239 #define SSH1_TTY_OP_ISPEED      192
240 #define SSH1_TTY_OP_OSPEED      193
241 #define SSH2_TTY_OP_ISPEED      128
242 #define SSH2_TTY_OP_OSPEED      129
243
244 /* Helper functions for parsing tty-related config. */
245 static unsigned int ssh_tty_parse_specchar(char *s)
246 {
247     unsigned int ret;
248     if (*s) {
249         char *next = NULL;
250         ret = ctrlparse(s, &next);
251         if (!next) ret = s[0];
252     } else {
253         ret = 255; /* special value meaning "don't set" */
254     }
255     return ret;
256 }
257 static unsigned int ssh_tty_parse_boolean(char *s)
258 {
259     if (stricmp(s, "yes") == 0 ||
260         stricmp(s, "on") == 0 ||
261         stricmp(s, "true") == 0 ||
262         stricmp(s, "+") == 0)
263         return 1; /* true */
264     else if (stricmp(s, "no") == 0 ||
265              stricmp(s, "off") == 0 ||
266              stricmp(s, "false") == 0 ||
267              stricmp(s, "-") == 0)
268         return 0; /* false */
269     else
270         return (atoi(s) != 0);
271 }
272
273 #define translate(x) if (type == x) return #x
274 #define translatec(x,ctx) if (type == x && (pkt_ctx & ctx)) return #x
275 static char *ssh1_pkt_type(int type)
276 {
277     translate(SSH1_MSG_DISCONNECT);
278     translate(SSH1_SMSG_PUBLIC_KEY);
279     translate(SSH1_CMSG_SESSION_KEY);
280     translate(SSH1_CMSG_USER);
281     translate(SSH1_CMSG_AUTH_RSA);
282     translate(SSH1_SMSG_AUTH_RSA_CHALLENGE);
283     translate(SSH1_CMSG_AUTH_RSA_RESPONSE);
284     translate(SSH1_CMSG_AUTH_PASSWORD);
285     translate(SSH1_CMSG_REQUEST_PTY);
286     translate(SSH1_CMSG_WINDOW_SIZE);
287     translate(SSH1_CMSG_EXEC_SHELL);
288     translate(SSH1_CMSG_EXEC_CMD);
289     translate(SSH1_SMSG_SUCCESS);
290     translate(SSH1_SMSG_FAILURE);
291     translate(SSH1_CMSG_STDIN_DATA);
292     translate(SSH1_SMSG_STDOUT_DATA);
293     translate(SSH1_SMSG_STDERR_DATA);
294     translate(SSH1_CMSG_EOF);
295     translate(SSH1_SMSG_EXIT_STATUS);
296     translate(SSH1_MSG_CHANNEL_OPEN_CONFIRMATION);
297     translate(SSH1_MSG_CHANNEL_OPEN_FAILURE);
298     translate(SSH1_MSG_CHANNEL_DATA);
299     translate(SSH1_MSG_CHANNEL_CLOSE);
300     translate(SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION);
301     translate(SSH1_SMSG_X11_OPEN);
302     translate(SSH1_CMSG_PORT_FORWARD_REQUEST);
303     translate(SSH1_MSG_PORT_OPEN);
304     translate(SSH1_CMSG_AGENT_REQUEST_FORWARDING);
305     translate(SSH1_SMSG_AGENT_OPEN);
306     translate(SSH1_MSG_IGNORE);
307     translate(SSH1_CMSG_EXIT_CONFIRMATION);
308     translate(SSH1_CMSG_X11_REQUEST_FORWARDING);
309     translate(SSH1_CMSG_AUTH_RHOSTS_RSA);
310     translate(SSH1_MSG_DEBUG);
311     translate(SSH1_CMSG_REQUEST_COMPRESSION);
312     translate(SSH1_CMSG_AUTH_TIS);
313     translate(SSH1_SMSG_AUTH_TIS_CHALLENGE);
314     translate(SSH1_CMSG_AUTH_TIS_RESPONSE);
315     translate(SSH1_CMSG_AUTH_CCARD);
316     translate(SSH1_SMSG_AUTH_CCARD_CHALLENGE);
317     translate(SSH1_CMSG_AUTH_CCARD_RESPONSE);
318     return "unknown";
319 }
320 static char *ssh2_pkt_type(int pkt_ctx, int type)
321 {
322     translate(SSH2_MSG_DISCONNECT);
323     translate(SSH2_MSG_IGNORE);
324     translate(SSH2_MSG_UNIMPLEMENTED);
325     translate(SSH2_MSG_DEBUG);
326     translate(SSH2_MSG_SERVICE_REQUEST);
327     translate(SSH2_MSG_SERVICE_ACCEPT);
328     translate(SSH2_MSG_KEXINIT);
329     translate(SSH2_MSG_NEWKEYS);
330     translatec(SSH2_MSG_KEXDH_INIT, SSH2_PKTCTX_DHGROUP);
331     translatec(SSH2_MSG_KEXDH_REPLY, SSH2_PKTCTX_DHGROUP);
332     translatec(SSH2_MSG_KEX_DH_GEX_REQUEST, SSH2_PKTCTX_DHGEX);
333     translatec(SSH2_MSG_KEX_DH_GEX_GROUP, SSH2_PKTCTX_DHGEX);
334     translatec(SSH2_MSG_KEX_DH_GEX_INIT, SSH2_PKTCTX_DHGEX);
335     translatec(SSH2_MSG_KEX_DH_GEX_REPLY, SSH2_PKTCTX_DHGEX);
336     translate(SSH2_MSG_USERAUTH_REQUEST);
337     translate(SSH2_MSG_USERAUTH_FAILURE);
338     translate(SSH2_MSG_USERAUTH_SUCCESS);
339     translate(SSH2_MSG_USERAUTH_BANNER);
340     translatec(SSH2_MSG_USERAUTH_PK_OK, SSH2_PKTCTX_PUBLICKEY);
341     translatec(SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ, SSH2_PKTCTX_PASSWORD);
342     translatec(SSH2_MSG_USERAUTH_INFO_REQUEST, SSH2_PKTCTX_KBDINTER);
343     translatec(SSH2_MSG_USERAUTH_INFO_RESPONSE, SSH2_PKTCTX_KBDINTER);
344     translate(SSH2_MSG_GLOBAL_REQUEST);
345     translate(SSH2_MSG_REQUEST_SUCCESS);
346     translate(SSH2_MSG_REQUEST_FAILURE);
347     translate(SSH2_MSG_CHANNEL_OPEN);
348     translate(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION);
349     translate(SSH2_MSG_CHANNEL_OPEN_FAILURE);
350     translate(SSH2_MSG_CHANNEL_WINDOW_ADJUST);
351     translate(SSH2_MSG_CHANNEL_DATA);
352     translate(SSH2_MSG_CHANNEL_EXTENDED_DATA);
353     translate(SSH2_MSG_CHANNEL_EOF);
354     translate(SSH2_MSG_CHANNEL_CLOSE);
355     translate(SSH2_MSG_CHANNEL_REQUEST);
356     translate(SSH2_MSG_CHANNEL_SUCCESS);
357     translate(SSH2_MSG_CHANNEL_FAILURE);
358     return "unknown";
359 }
360 #undef translate
361 #undef translatec
362
363 /* Enumeration values for fields in SSH-1 packets */
364 enum {
365     PKT_END, PKT_INT, PKT_CHAR, PKT_DATA, PKT_STR, PKT_BIGNUM,
366     /* These values are for communicating relevant semantics of
367      * fields to the packet logging code. */
368     PKTT_OTHER, PKTT_PASSWORD, PKTT_DATA
369 };
370
371 /*
372  * Coroutine mechanics for the sillier bits of the code. If these
373  * macros look impenetrable to you, you might find it helpful to
374  * read
375  * 
376  *   http://www.chiark.greenend.org.uk/~sgtatham/coroutines.html
377  * 
378  * which explains the theory behind these macros.
379  * 
380  * In particular, if you are getting `case expression not constant'
381  * errors when building with MS Visual Studio, this is because MS's
382  * Edit and Continue debugging feature causes their compiler to
383  * violate ANSI C. To disable Edit and Continue debugging:
384  * 
385  *  - right-click ssh.c in the FileView
386  *  - click Settings
387  *  - select the C/C++ tab and the General category
388  *  - under `Debug info:', select anything _other_ than `Program
389  *    Database for Edit and Continue'.
390  */
391 #define crBegin(v)      { int *crLine = &v; switch(v) { case 0:;
392 #define crState(t) \
393     struct t *s; \
394     if (!ssh->t) ssh->t = snew(struct t); \
395     s = ssh->t;
396 #define crFinish(z)     } *crLine = 0; return (z); }
397 #define crFinishV       } *crLine = 0; return; }
398 #define crReturn(z)     \
399         do {\
400             *crLine =__LINE__; return (z); case __LINE__:;\
401         } while (0)
402 #define crReturnV       \
403         do {\
404             *crLine=__LINE__; return; case __LINE__:;\
405         } while (0)
406 #define crStop(z)       do{ *crLine = 0; return (z); }while(0)
407 #define crStopV         do{ *crLine = 0; return; }while(0)
408 #define crWaitUntil(c)  do { crReturn(0); } while (!(c))
409 #define crWaitUntilV(c) do { crReturnV; } while (!(c))
410
411 typedef struct ssh_tag *Ssh;
412 struct Packet;
413
414 static struct Packet *ssh1_pkt_init(int pkt_type);
415 static struct Packet *ssh2_pkt_init(int pkt_type);
416 static void ssh_pkt_ensure(struct Packet *, int length);
417 static void ssh_pkt_adddata(struct Packet *, void *data, int len);
418 static void ssh_pkt_addbyte(struct Packet *, unsigned char value);
419 static void ssh2_pkt_addbool(struct Packet *, unsigned char value);
420 static void ssh_pkt_adduint32(struct Packet *, unsigned long value);
421 static void ssh_pkt_addstring_start(struct Packet *);
422 static void ssh_pkt_addstring_str(struct Packet *, char *data);
423 static void ssh_pkt_addstring_data(struct Packet *, char *data, int len);
424 static void ssh_pkt_addstring(struct Packet *, char *data);
425 static unsigned char *ssh2_mpint_fmt(Bignum b, int *len);
426 static void ssh1_pkt_addmp(struct Packet *, Bignum b);
427 static void ssh2_pkt_addmp(struct Packet *, Bignum b);
428 static int ssh2_pkt_construct(Ssh, struct Packet *);
429 static void ssh2_pkt_send(Ssh, struct Packet *);
430 static void ssh2_pkt_send_noqueue(Ssh, struct Packet *);
431 static int do_ssh1_login(Ssh ssh, unsigned char *in, int inlen,
432                          struct Packet *pktin);
433 static void do_ssh2_authconn(Ssh ssh, unsigned char *in, int inlen,
434                              struct Packet *pktin);
435
436 /*
437  * Buffer management constants. There are several of these for
438  * various different purposes:
439  * 
440  *  - SSH1_BUFFER_LIMIT is the amount of backlog that must build up
441  *    on a local data stream before we throttle the whole SSH
442  *    connection (in SSH-1 only). Throttling the whole connection is
443  *    pretty drastic so we set this high in the hope it won't
444  *    happen very often.
445  * 
446  *  - SSH_MAX_BACKLOG is the amount of backlog that must build up
447  *    on the SSH connection itself before we defensively throttle
448  *    _all_ local data streams. This is pretty drastic too (though
449  *    thankfully unlikely in SSH-2 since the window mechanism should
450  *    ensure that the server never has any need to throttle its end
451  *    of the connection), so we set this high as well.
452  * 
453  *  - OUR_V2_WINSIZE is the maximum window size we present on SSH-2
454  *    channels.
455  */
456
457 #define SSH1_BUFFER_LIMIT 32768
458 #define SSH_MAX_BACKLOG 32768
459 #define OUR_V2_WINSIZE 16384
460 #define OUR_V2_MAXPKT 0x4000UL
461
462 const static struct ssh_signkey *hostkey_algs[] = { &ssh_rsa, &ssh_dss };
463
464 const static struct ssh_mac *macs[] = {
465     &ssh_sha1, &ssh_md5
466 };
467 const static struct ssh_mac *buggymacs[] = {
468     &ssh_sha1_buggy, &ssh_md5
469 };
470
471 static void *ssh_comp_none_init(void)
472 {
473     return NULL;
474 }
475 static void ssh_comp_none_cleanup(void *handle)
476 {
477 }
478 static int ssh_comp_none_block(void *handle, unsigned char *block, int len,
479                                unsigned char **outblock, int *outlen)
480 {
481     return 0;
482 }
483 static int ssh_comp_none_disable(void *handle)
484 {
485     return 0;
486 }
487 const static struct ssh_compress ssh_comp_none = {
488     "none",
489     ssh_comp_none_init, ssh_comp_none_cleanup, ssh_comp_none_block,
490     ssh_comp_none_init, ssh_comp_none_cleanup, ssh_comp_none_block,
491     ssh_comp_none_disable, NULL
492 };
493 extern const struct ssh_compress ssh_zlib;
494 const static struct ssh_compress *compressions[] = {
495     &ssh_zlib, &ssh_comp_none
496 };
497
498 enum {                                 /* channel types */
499     CHAN_MAINSESSION,
500     CHAN_X11,
501     CHAN_AGENT,
502     CHAN_SOCKDATA,
503     CHAN_SOCKDATA_DORMANT              /* one the remote hasn't confirmed */
504 };
505
506 /*
507  * 2-3-4 tree storing channels.
508  */
509 struct ssh_channel {
510     Ssh ssh;                           /* pointer back to main context */
511     unsigned remoteid, localid;
512     int type;
513     /* True if we opened this channel but server hasn't confirmed. */
514     int halfopen;
515     /*
516      * In SSH-1, this value contains four bits:
517      * 
518      *   1   We have sent SSH1_MSG_CHANNEL_CLOSE.
519      *   2   We have sent SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION.
520      *   4   We have received SSH1_MSG_CHANNEL_CLOSE.
521      *   8   We have received SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION.
522      * 
523      * A channel is completely finished with when all four bits are set.
524      */
525     int closes;
526     union {
527         struct ssh1_data_channel {
528             int throttling;
529         } v1;
530         struct ssh2_data_channel {
531             bufchain outbuffer;
532             unsigned remwindow, remmaxpkt;
533             unsigned locwindow;
534         } v2;
535     } v;
536     union {
537         struct ssh_agent_channel {
538             unsigned char *message;
539             unsigned char msglen[4];
540             unsigned lensofar, totallen;
541         } a;
542         struct ssh_x11_channel {
543             Socket s;
544         } x11;
545         struct ssh_pfd_channel {
546             Socket s;
547         } pfd;
548     } u;
549 };
550
551 /*
552  * 2-3-4 tree storing remote->local port forwardings. SSH-1 and SSH-2
553  * use this structure in different ways, reflecting SSH-2's
554  * altogether saner approach to port forwarding.
555  * 
556  * In SSH-1, you arrange a remote forwarding by sending the server
557  * the remote port number, and the local destination host:port.
558  * When a connection comes in, the server sends you back that
559  * host:port pair, and you connect to it. This is a ready-made
560  * security hole if you're not on the ball: a malicious server
561  * could send you back _any_ host:port pair, so if you trustingly
562  * connect to the address it gives you then you've just opened the
563  * entire inside of your corporate network just by connecting
564  * through it to a dodgy SSH server. Hence, we must store a list of
565  * host:port pairs we _are_ trying to forward to, and reject a
566  * connection request from the server if it's not in the list.
567  * 
568  * In SSH-2, each side of the connection minds its own business and
569  * doesn't send unnecessary information to the other. You arrange a
570  * remote forwarding by sending the server just the remote port
571  * number. When a connection comes in, the server tells you which
572  * of its ports was connected to; and _you_ have to remember what
573  * local host:port pair went with that port number.
574  * 
575  * Hence, in SSH-1 this structure is indexed by destination
576  * host:port pair, whereas in SSH-2 it is indexed by source port.
577  */
578 struct ssh_portfwd; /* forward declaration */
579
580 struct ssh_rportfwd {
581     unsigned sport, dport;
582     char dhost[256];
583     char *sportdesc;
584     struct ssh_portfwd *pfrec;
585 };
586 #define free_rportfwd(pf) ( \
587     ((pf) ? (sfree((pf)->sportdesc)) : (void)0 ), sfree(pf) )
588
589 /*
590  * Separately to the rportfwd tree (which is for looking up port
591  * open requests from the server), a tree of _these_ structures is
592  * used to keep track of all the currently open port forwardings,
593  * so that we can reconfigure in mid-session if the user requests
594  * it.
595  */
596 struct ssh_portfwd {
597     enum { DESTROY, KEEP, CREATE } status;
598     int type;
599     unsigned sport, dport;
600     char *saddr, *daddr;
601     char *sserv, *dserv;
602     struct ssh_rportfwd *remote;
603     int addressfamily;
604     void *local;
605 };
606 #define free_portfwd(pf) ( \
607     ((pf) ? (sfree((pf)->saddr), sfree((pf)->daddr), \
608              sfree((pf)->sserv), sfree((pf)->dserv)) : (void)0 ), sfree(pf) )
609
610 struct Packet {
611     long length;            /* length of `data' actually used */
612     long forcepad;          /* SSH-2: force padding to at least this length */
613     int type;               /* only used for incoming packets */
614     unsigned long sequence; /* SSH-2 incoming sequence number */
615     unsigned char *data;    /* allocated storage */
616     unsigned char *body;    /* offset of payload within `data' */
617     long savedpos;          /* temporary index into `data' (for strings) */
618     long maxlen;            /* amount of storage allocated for `data' */
619     long encrypted_len;     /* for SSH-2 total-size counting */
620
621     /*
622      * State associated with packet logging
623      */
624     int logmode;
625     int nblanks;
626     struct logblank_t *blanks;
627 };
628
629 static void ssh1_protocol(Ssh ssh, void *vin, int inlen,
630                           struct Packet *pktin);
631 static void ssh2_protocol(Ssh ssh, void *vin, int inlen,
632                           struct Packet *pktin);
633 static void ssh1_protocol_setup(Ssh ssh);
634 static void ssh2_protocol_setup(Ssh ssh);
635 static void ssh_size(void *handle, int width, int height);
636 static void ssh_special(void *handle, Telnet_Special);
637 static int ssh2_try_send(struct ssh_channel *c);
638 static void ssh2_add_channel_data(struct ssh_channel *c, char *buf, int len);
639 static void ssh_throttle_all(Ssh ssh, int enable, int bufsize);
640 static void ssh2_set_window(struct ssh_channel *c, unsigned newwin);
641 static int ssh_sendbuffer(void *handle);
642 static int ssh_do_close(Ssh ssh, int notify_exit);
643 static unsigned long ssh_pkt_getuint32(struct Packet *pkt);
644 static int ssh2_pkt_getbool(struct Packet *pkt);
645 static void ssh_pkt_getstring(struct Packet *pkt, char **p, int *length);
646 static void ssh2_timer(void *ctx, long now);
647 static int do_ssh2_transport(Ssh ssh, void *vin, int inlen,
648                              struct Packet *pktin);
649
650 struct rdpkt1_state_tag {
651     long len, pad, biglen, to_read;
652     unsigned long realcrc, gotcrc;
653     unsigned char *p;
654     int i;
655     int chunk;
656     struct Packet *pktin;
657 };
658
659 struct rdpkt2_state_tag {
660     long len, pad, payload, packetlen, maclen;
661     int i;
662     int cipherblk;
663     unsigned long incoming_sequence;
664     struct Packet *pktin;
665 };
666
667 typedef void (*handler_fn_t)(Ssh ssh, struct Packet *pktin);
668 typedef void (*chandler_fn_t)(Ssh ssh, struct Packet *pktin, void *ctx);
669
670 struct queued_handler;
671 struct queued_handler {
672     int msg1, msg2;
673     chandler_fn_t handler;
674     void *ctx;
675     struct queued_handler *next;
676 };
677
678 struct ssh_tag {
679     const struct plug_function_table *fn;
680     /* the above field _must_ be first in the structure */
681
682     SHA_State exhash, exhashbase;
683
684     Socket s;
685
686     void *ldisc;
687     void *logctx;
688
689     unsigned char session_key[32];
690     int v1_compressing;
691     int v1_remote_protoflags;
692     int v1_local_protoflags;
693     int agentfwd_enabled;
694     int X11_fwd_enabled;
695     int remote_bugs;
696     const struct ssh_cipher *cipher;
697     void *v1_cipher_ctx;
698     void *crcda_ctx;
699     const struct ssh2_cipher *cscipher, *sccipher;
700     void *cs_cipher_ctx, *sc_cipher_ctx;
701     const struct ssh_mac *csmac, *scmac;
702     void *cs_mac_ctx, *sc_mac_ctx;
703     const struct ssh_compress *cscomp, *sccomp;
704     void *cs_comp_ctx, *sc_comp_ctx;
705     const struct ssh_kex *kex;
706     const struct ssh_signkey *hostkey;
707     unsigned char v2_session_id[20];
708     void *kex_ctx;
709
710     char *savedhost;
711     int savedport;
712     int send_ok;
713     int echoing, editing;
714
715     void *frontend;
716
717     int ospeed, ispeed;                /* temporaries */
718     int term_width, term_height;
719
720     tree234 *channels;                 /* indexed by local id */
721     struct ssh_channel *mainchan;      /* primary session channel */
722     int exitcode;
723     int close_expected;
724     int clean_exit;
725
726     tree234 *rportfwds, *portfwds;
727
728     enum {
729         SSH_STATE_PREPACKET,
730         SSH_STATE_BEFORE_SIZE,
731         SSH_STATE_INTERMED,
732         SSH_STATE_SESSION,
733         SSH_STATE_CLOSED
734     } state;
735
736     int size_needed, eof_needed;
737
738     struct Packet **queue;
739     int queuelen, queuesize;
740     int queueing;
741     unsigned char *deferred_send_data;
742     int deferred_len, deferred_size;
743
744     /*
745      * Gross hack: pscp will try to start SFTP but fall back to
746      * scp1 if that fails. This variable is the means by which
747      * scp.c can reach into the SSH code and find out which one it
748      * got.
749      */
750     int fallback_cmd;
751
752     bufchain banner;    /* accumulates banners during do_ssh2_authconn */
753     /*
754      * Used for username and password input.
755      */
756     char *userpass_input_buffer;
757     int userpass_input_buflen;
758     int userpass_input_bufpos;
759     int userpass_input_echo;
760
761     int pkt_ctx;
762
763     void *x11auth;
764
765     int version;
766     int v1_throttle_count;
767     int overall_bufsize;
768     int throttled_all;
769     int v1_stdout_throttling;
770     unsigned long v2_outgoing_sequence;
771
772     int ssh1_rdpkt_crstate;
773     int ssh2_rdpkt_crstate;
774     int do_ssh_init_crstate;
775     int ssh_gotdata_crstate;
776     int do_ssh1_login_crstate;
777     int do_ssh1_connection_crstate;
778     int do_ssh2_transport_crstate;
779     int do_ssh2_authconn_crstate;
780
781     void *do_ssh_init_state;
782     void *do_ssh1_login_state;
783     void *do_ssh2_transport_state;
784     void *do_ssh2_authconn_state;
785
786     struct rdpkt1_state_tag rdpkt1_state;
787     struct rdpkt2_state_tag rdpkt2_state;
788
789     /* SSH-1 and SSH-2 use this for different things, but both use it */
790     int protocol_initial_phase_done;
791
792     void (*protocol) (Ssh ssh, void *vin, int inlen,
793                       struct Packet *pkt);
794     struct Packet *(*s_rdpkt) (Ssh ssh, unsigned char **data, int *datalen);
795
796     /*
797      * We maintain a full _copy_ of a Config structure here, not
798      * merely a pointer to it. That way, when we're passed a new
799      * one for reconfiguration, we can check the differences and
800      * potentially reconfigure port forwardings etc in mid-session.
801      */
802     Config cfg;
803
804     /*
805      * Used to transfer data back from async callbacks.
806      */
807     void *agent_response;
808     int agent_response_len;
809     int user_response;
810
811     /*
812      * The SSH connection can be set as `frozen', meaning we are
813      * not currently accepting incoming data from the network. This
814      * is slightly more serious than setting the _socket_ as
815      * frozen, because we may already have had data passed to us
816      * from the network which we need to delay processing until
817      * after the freeze is lifted, so we also need a bufchain to
818      * store that data.
819      */
820     int frozen;
821     bufchain queued_incoming_data;
822
823     /*
824      * Dispatch table for packet types that we may have to deal
825      * with at any time.
826      */
827     handler_fn_t packet_dispatch[256];
828
829     /*
830      * Queues of one-off handler functions for success/failure
831      * indications from a request.
832      */
833     struct queued_handler *qhead, *qtail;
834
835     /*
836      * This module deals with sending keepalives.
837      */
838     Pinger pinger;
839
840     /*
841      * Track incoming and outgoing data sizes and time, for
842      * size-based rekeys.
843      */
844     unsigned long incoming_data_size, outgoing_data_size, deferred_data_size;
845     unsigned long max_data_size;
846     int kex_in_progress;
847     long next_rekey, last_rekey;
848     char *deferred_rekey_reason;    /* points to STATIC string; don't free */
849 };
850
851 #define logevent(s) logevent(ssh->frontend, s)
852
853 /* logevent, only printf-formatted. */
854 static void logeventf(Ssh ssh, const char *fmt, ...)
855 {
856     va_list ap;
857     char *buf;
858
859     va_start(ap, fmt);
860     buf = dupvprintf(fmt, ap);
861     va_end(ap);
862     logevent(buf);
863     sfree(buf);
864 }
865
866 #define bombout(msg) \
867     do { \
868         char *text = dupprintf msg; \
869         ssh_do_close(ssh, FALSE); \
870         logevent(text); \
871         connection_fatal(ssh->frontend, "%s", text); \
872         sfree(text); \
873     } while (0)
874
875 /* Functions to leave bits out of the SSH packet log file. */
876
877 static void dont_log_password(Ssh ssh, struct Packet *pkt, int blanktype)
878 {
879     if (ssh->cfg.logomitpass)
880         pkt->logmode = blanktype;
881 }
882
883 static void dont_log_data(Ssh ssh, struct Packet *pkt, int blanktype)
884 {
885     if (ssh->cfg.logomitdata)
886         pkt->logmode = blanktype;
887 }
888
889 static void end_log_omission(Ssh ssh, struct Packet *pkt)
890 {
891     pkt->logmode = PKTLOG_EMIT;
892 }
893
894 /* Helper function for common bits of parsing cfg.ttymodes. */
895 static void parse_ttymodes(Ssh ssh, char *modes,
896                            void (*do_mode)(void *data, char *mode, char *val),
897                            void *data)
898 {
899     while (*modes) {
900         char *t = strchr(modes, '\t');
901         char *m = snewn(t-modes+1, char);
902         char *val;
903         strncpy(m, modes, t-modes);
904         m[t-modes] = '\0';
905         if (*(t+1) == 'A')
906             val = get_ttymode(ssh->frontend, m);
907         else
908             val = dupstr(t+2);
909         if (val)
910             do_mode(data, m, val);
911         sfree(m);
912         sfree(val);
913         modes += strlen(modes) + 1;
914     }
915 }
916
917 static int ssh_channelcmp(void *av, void *bv)
918 {
919     struct ssh_channel *a = (struct ssh_channel *) av;
920     struct ssh_channel *b = (struct ssh_channel *) bv;
921     if (a->localid < b->localid)
922         return -1;
923     if (a->localid > b->localid)
924         return +1;
925     return 0;
926 }
927 static int ssh_channelfind(void *av, void *bv)
928 {
929     unsigned *a = (unsigned *) av;
930     struct ssh_channel *b = (struct ssh_channel *) bv;
931     if (*a < b->localid)
932         return -1;
933     if (*a > b->localid)
934         return +1;
935     return 0;
936 }
937
938 static int ssh_rportcmp_ssh1(void *av, void *bv)
939 {
940     struct ssh_rportfwd *a = (struct ssh_rportfwd *) av;
941     struct ssh_rportfwd *b = (struct ssh_rportfwd *) bv;
942     int i;
943     if ( (i = strcmp(a->dhost, b->dhost)) != 0)
944         return i < 0 ? -1 : +1;
945     if (a->dport > b->dport)
946         return +1;
947     if (a->dport < b->dport)
948         return -1;
949     return 0;
950 }
951
952 static int ssh_rportcmp_ssh2(void *av, void *bv)
953 {
954     struct ssh_rportfwd *a = (struct ssh_rportfwd *) av;
955     struct ssh_rportfwd *b = (struct ssh_rportfwd *) bv;
956
957     if (a->sport > b->sport)
958         return +1;
959     if (a->sport < b->sport)
960         return -1;
961     return 0;
962 }
963
964 /*
965  * Special form of strcmp which can cope with NULL inputs. NULL is
966  * defined to sort before even the empty string.
967  */
968 static int nullstrcmp(const char *a, const char *b)
969 {
970     if (a == NULL && b == NULL)
971         return 0;
972     if (a == NULL)
973         return -1;
974     if (b == NULL)
975         return +1;
976     return strcmp(a, b);
977 }
978
979 static int ssh_portcmp(void *av, void *bv)
980 {
981     struct ssh_portfwd *a = (struct ssh_portfwd *) av;
982     struct ssh_portfwd *b = (struct ssh_portfwd *) bv;
983     int i;
984     if (a->type > b->type)
985         return +1;
986     if (a->type < b->type)
987         return -1;
988     if (a->addressfamily > b->addressfamily)
989         return +1;
990     if (a->addressfamily < b->addressfamily)
991         return -1;
992     if ( (i = nullstrcmp(a->saddr, b->saddr)) != 0)
993         return i < 0 ? -1 : +1;
994     if (a->sport > b->sport)
995         return +1;
996     if (a->sport < b->sport)
997         return -1;
998     if (a->type != 'D') {
999         if ( (i = nullstrcmp(a->daddr, b->daddr)) != 0)
1000             return i < 0 ? -1 : +1;
1001         if (a->dport > b->dport)
1002             return +1;
1003         if (a->dport < b->dport)
1004             return -1;
1005     }
1006     return 0;
1007 }
1008
1009 static int alloc_channel_id(Ssh ssh)
1010 {
1011     const unsigned CHANNEL_NUMBER_OFFSET = 256;
1012     unsigned low, high, mid;
1013     int tsize;
1014     struct ssh_channel *c;
1015
1016     /*
1017      * First-fit allocation of channel numbers: always pick the
1018      * lowest unused one. To do this, binary-search using the
1019      * counted B-tree to find the largest channel ID which is in a
1020      * contiguous sequence from the beginning. (Precisely
1021      * everything in that sequence must have ID equal to its tree
1022      * index plus CHANNEL_NUMBER_OFFSET.)
1023      */
1024     tsize = count234(ssh->channels);
1025
1026     low = -1;
1027     high = tsize;
1028     while (high - low > 1) {
1029         mid = (high + low) / 2;
1030         c = index234(ssh->channels, mid);
1031         if (c->localid == mid + CHANNEL_NUMBER_OFFSET)
1032             low = mid;                 /* this one is fine */
1033         else
1034             high = mid;                /* this one is past it */
1035     }
1036     /*
1037      * Now low points to either -1, or the tree index of the
1038      * largest ID in the initial sequence.
1039      */
1040     {
1041         unsigned i = low + 1 + CHANNEL_NUMBER_OFFSET;
1042         assert(NULL == find234(ssh->channels, &i, ssh_channelfind));
1043     }
1044     return low + 1 + CHANNEL_NUMBER_OFFSET;
1045 }
1046
1047 static void c_write(Ssh ssh, const char *buf, int len)
1048 {
1049     if ((flags & FLAG_STDERR)) {
1050         int i;
1051         for (i = 0; i < len; i++)
1052             if (buf[i] != '\r')
1053                 fputc(buf[i], stderr);
1054         return;
1055     }
1056     from_backend(ssh->frontend, 1, buf, len);
1057 }
1058
1059 static void c_write_untrusted(Ssh ssh, const char *buf, int len)
1060 {
1061     int i;
1062     for (i = 0; i < len; i++) {
1063         if (buf[i] == '\n')
1064             c_write(ssh, "\r\n", 2);
1065         else if ((buf[i] & 0x60) || (buf[i] == '\r'))
1066             c_write(ssh, buf + i, 1);
1067     }
1068 }
1069
1070 static void c_write_str(Ssh ssh, const char *buf)
1071 {
1072     c_write(ssh, buf, strlen(buf));
1073 }
1074
1075 static void ssh_free_packet(struct Packet *pkt)
1076 {
1077     sfree(pkt->data);
1078     sfree(pkt);
1079 }
1080 static struct Packet *ssh_new_packet(void)
1081 {
1082     struct Packet *pkt = snew(struct Packet);
1083
1084     pkt->body = pkt->data = NULL;
1085     pkt->maxlen = 0;
1086     pkt->logmode = PKTLOG_EMIT;
1087     pkt->nblanks = 0;
1088     pkt->blanks = NULL;
1089
1090     return pkt;
1091 }
1092
1093 /*
1094  * Collect incoming data in the incoming packet buffer.
1095  * Decipher and verify the packet when it is completely read.
1096  * Drop SSH1_MSG_DEBUG and SSH1_MSG_IGNORE packets.
1097  * Update the *data and *datalen variables.
1098  * Return a Packet structure when a packet is completed.
1099  */
1100 static struct Packet *ssh1_rdpkt(Ssh ssh, unsigned char **data, int *datalen)
1101 {
1102     struct rdpkt1_state_tag *st = &ssh->rdpkt1_state;
1103
1104     crBegin(ssh->ssh1_rdpkt_crstate);
1105
1106     st->pktin = ssh_new_packet();
1107
1108     st->pktin->type = 0;
1109     st->pktin->length = 0;
1110
1111     for (st->i = st->len = 0; st->i < 4; st->i++) {
1112         while ((*datalen) == 0)
1113             crReturn(NULL);
1114         st->len = (st->len << 8) + **data;
1115         (*data)++, (*datalen)--;
1116     }
1117
1118     st->pad = 8 - (st->len % 8);
1119     st->biglen = st->len + st->pad;
1120     st->pktin->length = st->len - 5;
1121
1122     if (st->biglen < 0) {
1123         bombout(("Extremely large packet length from server suggests"
1124                  " data stream corruption"));
1125         ssh_free_packet(st->pktin);
1126         crStop(NULL);
1127     }
1128
1129     st->pktin->maxlen = st->biglen;
1130     st->pktin->data = snewn(st->biglen + APIEXTRA, unsigned char);
1131
1132     st->to_read = st->biglen;
1133     st->p = st->pktin->data;
1134     while (st->to_read > 0) {
1135         st->chunk = st->to_read;
1136         while ((*datalen) == 0)
1137             crReturn(NULL);
1138         if (st->chunk > (*datalen))
1139             st->chunk = (*datalen);
1140         memcpy(st->p, *data, st->chunk);
1141         *data += st->chunk;
1142         *datalen -= st->chunk;
1143         st->p += st->chunk;
1144         st->to_read -= st->chunk;
1145     }
1146
1147     if (ssh->cipher && detect_attack(ssh->crcda_ctx, st->pktin->data,
1148                                      st->biglen, NULL)) {
1149         bombout(("Network attack (CRC compensation) detected!"));
1150         ssh_free_packet(st->pktin);
1151         crStop(NULL);
1152     }
1153
1154     if (ssh->cipher)
1155         ssh->cipher->decrypt(ssh->v1_cipher_ctx, st->pktin->data, st->biglen);
1156
1157     st->realcrc = crc32_compute(st->pktin->data, st->biglen - 4);
1158     st->gotcrc = GET_32BIT(st->pktin->data + st->biglen - 4);
1159     if (st->gotcrc != st->realcrc) {
1160         bombout(("Incorrect CRC received on packet"));
1161         ssh_free_packet(st->pktin);
1162         crStop(NULL);
1163     }
1164
1165     st->pktin->body = st->pktin->data + st->pad + 1;
1166     st->pktin->savedpos = 0;
1167
1168     if (ssh->v1_compressing) {
1169         unsigned char *decompblk;
1170         int decomplen;
1171         if (!zlib_decompress_block(ssh->sc_comp_ctx,
1172                                    st->pktin->body - 1, st->pktin->length + 1,
1173                                    &decompblk, &decomplen)) {
1174             bombout(("Zlib decompression encountered invalid data"));
1175             ssh_free_packet(st->pktin);
1176             crStop(NULL);
1177         }
1178
1179         if (st->pktin->maxlen < st->pad + decomplen) {
1180             st->pktin->maxlen = st->pad + decomplen;
1181             st->pktin->data = sresize(st->pktin->data,
1182                                       st->pktin->maxlen + APIEXTRA,
1183                                       unsigned char);
1184             st->pktin->body = st->pktin->data + st->pad + 1;
1185         }
1186
1187         memcpy(st->pktin->body - 1, decompblk, decomplen);
1188         sfree(decompblk);
1189         st->pktin->length = decomplen - 1;
1190     }
1191
1192     st->pktin->type = st->pktin->body[-1];
1193
1194     /*
1195      * Log incoming packet, possibly omitting sensitive fields.
1196      */
1197     if (ssh->logctx) {
1198         int nblanks = 0;
1199         struct logblank_t blank;
1200         if (ssh->cfg.logomitdata) {
1201             int do_blank = FALSE, blank_prefix = 0;
1202             /* "Session data" packets - omit the data field */
1203             if ((st->pktin->type == SSH1_SMSG_STDOUT_DATA) ||
1204                 (st->pktin->type == SSH1_SMSG_STDERR_DATA)) {
1205                 do_blank = TRUE; blank_prefix = 0;
1206             } else if (st->pktin->type == SSH1_MSG_CHANNEL_DATA) {
1207                 do_blank = TRUE; blank_prefix = 4;
1208             }
1209             if (do_blank) {
1210                 blank.offset = blank_prefix;
1211                 blank.len = st->pktin->length;
1212                 blank.type = PKTLOG_OMIT;
1213                 nblanks = 1;
1214             }
1215         }
1216         log_packet(ssh->logctx,
1217                    PKT_INCOMING, st->pktin->type,
1218                    ssh1_pkt_type(st->pktin->type),
1219                    st->pktin->body, st->pktin->length,
1220                    nblanks, &blank);
1221     }
1222
1223     crFinish(st->pktin);
1224 }
1225
1226 static struct Packet *ssh2_rdpkt(Ssh ssh, unsigned char **data, int *datalen)
1227 {
1228     struct rdpkt2_state_tag *st = &ssh->rdpkt2_state;
1229
1230     crBegin(ssh->ssh2_rdpkt_crstate);
1231
1232     st->pktin = ssh_new_packet();
1233
1234     st->pktin->type = 0;
1235     st->pktin->length = 0;
1236     if (ssh->sccipher)
1237         st->cipherblk = ssh->sccipher->blksize;
1238     else
1239         st->cipherblk = 8;
1240     if (st->cipherblk < 8)
1241         st->cipherblk = 8;
1242
1243     st->pktin->data = snewn(st->cipherblk + APIEXTRA, unsigned char);
1244
1245     /*
1246      * Acquire and decrypt the first block of the packet. This will
1247      * contain the length and padding details.
1248      */
1249     for (st->i = st->len = 0; st->i < st->cipherblk; st->i++) {
1250         while ((*datalen) == 0)
1251             crReturn(NULL);
1252         st->pktin->data[st->i] = *(*data)++;
1253         (*datalen)--;
1254     }
1255
1256     if (ssh->sccipher)
1257         ssh->sccipher->decrypt(ssh->sc_cipher_ctx,
1258                                st->pktin->data, st->cipherblk);
1259
1260     /*
1261      * Now get the length and padding figures.
1262      */
1263     st->len = GET_32BIT(st->pktin->data);
1264     st->pad = st->pktin->data[4];
1265
1266     /*
1267      * _Completely_ silly lengths should be stomped on before they
1268      * do us any more damage.
1269      */
1270     if (st->len < 0 || st->pad < 0 || st->len + st->pad < 0) {
1271         bombout(("Incoming packet was garbled on decryption"));
1272         ssh_free_packet(st->pktin);
1273         crStop(NULL);
1274     }
1275
1276     /*
1277      * This enables us to deduce the payload length.
1278      */
1279     st->payload = st->len - st->pad - 1;
1280
1281     st->pktin->length = st->payload + 5;
1282
1283     /*
1284      * So now we can work out the total packet length.
1285      */
1286     st->packetlen = st->len + 4;
1287     st->maclen = ssh->scmac ? ssh->scmac->len : 0;
1288
1289     /*
1290      * Allocate memory for the rest of the packet.
1291      */
1292     st->pktin->maxlen = st->packetlen + st->maclen;
1293     st->pktin->data = sresize(st->pktin->data,
1294                               st->pktin->maxlen + APIEXTRA,
1295                               unsigned char);
1296
1297     /*
1298      * Read and decrypt the remainder of the packet.
1299      */
1300     for (st->i = st->cipherblk; st->i < st->packetlen + st->maclen;
1301          st->i++) {
1302         while ((*datalen) == 0)
1303             crReturn(NULL);
1304         st->pktin->data[st->i] = *(*data)++;
1305         (*datalen)--;
1306     }
1307     /* Decrypt everything _except_ the MAC. */
1308     if (ssh->sccipher)
1309         ssh->sccipher->decrypt(ssh->sc_cipher_ctx,
1310                                st->pktin->data + st->cipherblk,
1311                                st->packetlen - st->cipherblk);
1312
1313     st->pktin->encrypted_len = st->packetlen;
1314
1315     /*
1316      * Check the MAC.
1317      */
1318     if (ssh->scmac
1319         && !ssh->scmac->verify(ssh->sc_mac_ctx, st->pktin->data, st->len + 4,
1320                                st->incoming_sequence)) {
1321         bombout(("Incorrect MAC received on packet"));
1322         ssh_free_packet(st->pktin);
1323         crStop(NULL);
1324     }
1325
1326     st->pktin->sequence = st->incoming_sequence++;
1327
1328     /*
1329      * Decompress packet payload.
1330      */
1331     {
1332         unsigned char *newpayload;
1333         int newlen;
1334         if (ssh->sccomp &&
1335             ssh->sccomp->decompress(ssh->sc_comp_ctx,
1336                                     st->pktin->data + 5, st->pktin->length - 5,
1337                                     &newpayload, &newlen)) {
1338             if (st->pktin->maxlen < newlen + 5) {
1339                 st->pktin->maxlen = newlen + 5;
1340                 st->pktin->data = sresize(st->pktin->data,
1341                                           st->pktin->maxlen + APIEXTRA,
1342                                           unsigned char);
1343             }
1344             st->pktin->length = 5 + newlen;
1345             memcpy(st->pktin->data + 5, newpayload, newlen);
1346             sfree(newpayload);
1347         }
1348     }
1349
1350     st->pktin->savedpos = 6;
1351     st->pktin->body = st->pktin->data;
1352     st->pktin->type = st->pktin->data[5];
1353
1354     /*
1355      * Log incoming packet, possibly omitting sensitive fields.
1356      */
1357     if (ssh->logctx) {
1358         int nblanks = 0;
1359         struct logblank_t blank;
1360         if (ssh->cfg.logomitdata) {
1361             int do_blank = FALSE, blank_prefix = 0;
1362             /* "Session data" packets - omit the data field */
1363             if (st->pktin->type == SSH2_MSG_CHANNEL_DATA) {
1364                 do_blank = TRUE; blank_prefix = 4;
1365             } else if (st->pktin->type == SSH2_MSG_CHANNEL_EXTENDED_DATA) {
1366                 do_blank = TRUE; blank_prefix = 8;
1367             }
1368             if (do_blank) {
1369                 blank.offset = blank_prefix;
1370                 blank.len = (st->pktin->length-6) - blank_prefix;
1371                 blank.type = PKTLOG_OMIT;
1372                 nblanks = 1;
1373             }
1374         }
1375         log_packet(ssh->logctx, PKT_INCOMING, st->pktin->type,
1376                    ssh2_pkt_type(ssh->pkt_ctx, st->pktin->type),
1377                    st->pktin->data+6, st->pktin->length-6,
1378                    nblanks, &blank);
1379     }
1380
1381     crFinish(st->pktin);
1382 }
1383
1384 static int s_wrpkt_prepare(Ssh ssh, struct Packet *pkt, int *offset_p)
1385 {
1386     int pad, biglen, i, pktoffs;
1387     unsigned long crc;
1388 #ifdef __SC__
1389     /*
1390      * XXX various versions of SC (including 8.8.4) screw up the
1391      * register allocation in this function and use the same register
1392      * (D6) for len and as a temporary, with predictable results.  The
1393      * following sledgehammer prevents this.
1394      */
1395     volatile
1396 #endif
1397     int len;
1398
1399     if (ssh->logctx)
1400         log_packet(ssh->logctx, PKT_OUTGOING, pkt->data[12],
1401                    ssh1_pkt_type(pkt->data[12]),
1402                    pkt->body, pkt->length - (pkt->body - pkt->data),
1403                    pkt->nblanks, pkt->blanks);
1404     sfree(pkt->blanks); pkt->blanks = NULL;
1405     pkt->nblanks = 0;
1406
1407     if (ssh->v1_compressing) {
1408         unsigned char *compblk;
1409         int complen;
1410         zlib_compress_block(ssh->cs_comp_ctx,
1411                             pkt->data + 12, pkt->length - 12,
1412                             &compblk, &complen);
1413         memcpy(pkt->data + 12, compblk, complen);
1414         sfree(compblk);
1415         pkt->length = complen + 12;
1416     }
1417
1418     ssh_pkt_ensure(pkt, pkt->length + 4); /* space for CRC */
1419     pkt->length += 4;
1420     len = pkt->length - 4 - 8;  /* len(type+data+CRC) */
1421     pad = 8 - (len % 8);
1422     pktoffs = 8 - pad;
1423     biglen = len + pad;         /* len(padding+type+data+CRC) */
1424
1425     for (i = pktoffs; i < 4+8; i++)
1426         pkt->data[i] = random_byte();
1427     crc = crc32_compute(pkt->data + pktoffs + 4, biglen - 4); /* all ex len */
1428     PUT_32BIT(pkt->data + pktoffs + 4 + biglen - 4, crc);
1429     PUT_32BIT(pkt->data + pktoffs, len);
1430
1431     if (ssh->cipher)
1432         ssh->cipher->encrypt(ssh->v1_cipher_ctx,
1433                              pkt->data + pktoffs + 4, biglen);
1434
1435     if (offset_p) *offset_p = pktoffs;
1436     return biglen + 4;          /* len(length+padding+type+data+CRC) */
1437 }
1438
1439 static void s_wrpkt(Ssh ssh, struct Packet *pkt)
1440 {
1441     int len, backlog, offset;
1442     len = s_wrpkt_prepare(ssh, pkt, &offset);
1443     backlog = sk_write(ssh->s, (char *)pkt->data + offset, len);
1444     if (backlog > SSH_MAX_BACKLOG)
1445         ssh_throttle_all(ssh, 1, backlog);
1446     ssh_free_packet(pkt);
1447 }
1448
1449 static void s_wrpkt_defer(Ssh ssh, struct Packet *pkt)
1450 {
1451     int len, offset;
1452     len = s_wrpkt_prepare(ssh, pkt, &offset);
1453     if (ssh->deferred_len + len > ssh->deferred_size) {
1454         ssh->deferred_size = ssh->deferred_len + len + 128;
1455         ssh->deferred_send_data = sresize(ssh->deferred_send_data,
1456                                           ssh->deferred_size,
1457                                           unsigned char);
1458     }
1459     memcpy(ssh->deferred_send_data + ssh->deferred_len,
1460            pkt->data + offset, len);
1461     ssh->deferred_len += len;
1462     ssh_free_packet(pkt);
1463 }
1464
1465 /*
1466  * Construct a SSH-1 packet with the specified contents.
1467  * (This all-at-once interface used to be the only one, but now SSH-1
1468  * packets can also be constructed incrementally.)
1469  */
1470 static struct Packet *construct_packet(Ssh ssh, int pkttype, va_list ap)
1471 {
1472     int argtype;
1473     Bignum bn;
1474     struct Packet *pkt;
1475
1476     pkt = ssh1_pkt_init(pkttype);
1477
1478     while ((argtype = va_arg(ap, int)) != PKT_END) {
1479         unsigned char *argp, argchar;
1480         unsigned long argint;
1481         int arglen;
1482         switch (argtype) {
1483           /* Actual fields in the packet */
1484           case PKT_INT:
1485             argint = va_arg(ap, int);
1486             ssh_pkt_adduint32(pkt, argint);
1487             break;
1488           case PKT_CHAR:
1489             argchar = (unsigned char) va_arg(ap, int);
1490             ssh_pkt_addbyte(pkt, argchar);
1491             break;
1492           case PKT_DATA:
1493             argp = va_arg(ap, unsigned char *);
1494             arglen = va_arg(ap, int);
1495             ssh_pkt_adddata(pkt, argp, arglen);
1496             break;
1497           case PKT_STR:
1498             argp = va_arg(ap, unsigned char *);
1499             ssh_pkt_addstring(pkt, argp);
1500             break;
1501           case PKT_BIGNUM:
1502             bn = va_arg(ap, Bignum);
1503             ssh1_pkt_addmp(pkt, bn);
1504             break;
1505           /* Tokens for modifications to packet logging */
1506           case PKTT_PASSWORD:
1507             dont_log_password(ssh, pkt, PKTLOG_BLANK);
1508             break;
1509           case PKTT_DATA:
1510             dont_log_data(ssh, pkt, PKTLOG_OMIT);
1511             break;
1512           case PKTT_OTHER:
1513             end_log_omission(ssh, pkt);
1514             break;
1515         }
1516     }
1517
1518     return pkt;
1519 }
1520
1521 static void send_packet(Ssh ssh, int pkttype, ...)
1522 {
1523     struct Packet *pkt;
1524     va_list ap;
1525     va_start(ap, pkttype);
1526     pkt = construct_packet(ssh, pkttype, ap);
1527     va_end(ap);
1528     s_wrpkt(ssh, pkt);
1529 }
1530
1531 static void defer_packet(Ssh ssh, int pkttype, ...)
1532 {
1533     struct Packet *pkt;
1534     va_list ap;
1535     va_start(ap, pkttype);
1536     pkt = construct_packet(ssh, pkttype, ap);
1537     va_end(ap);
1538     s_wrpkt_defer(ssh, pkt);
1539 }
1540
1541 static int ssh_versioncmp(char *a, char *b)
1542 {
1543     char *ae, *be;
1544     unsigned long av, bv;
1545
1546     av = strtoul(a, &ae, 10);
1547     bv = strtoul(b, &be, 10);
1548     if (av != bv)
1549         return (av < bv ? -1 : +1);
1550     if (*ae == '.')
1551         ae++;
1552     if (*be == '.')
1553         be++;
1554     av = strtoul(ae, &ae, 10);
1555     bv = strtoul(be, &be, 10);
1556     if (av != bv)
1557         return (av < bv ? -1 : +1);
1558     return 0;
1559 }
1560
1561 /*
1562  * Utility routines for putting an SSH-protocol `string' and
1563  * `uint32' into a SHA state.
1564  */
1565 static void sha_string(SHA_State * s, void *str, int len)
1566 {
1567     unsigned char lenblk[4];
1568     PUT_32BIT(lenblk, len);
1569     SHA_Bytes(s, lenblk, 4);
1570     SHA_Bytes(s, str, len);
1571 }
1572
1573 static void sha_uint32(SHA_State * s, unsigned i)
1574 {
1575     unsigned char intblk[4];
1576     PUT_32BIT(intblk, i);
1577     SHA_Bytes(s, intblk, 4);
1578 }
1579
1580 /*
1581  * Packet construction functions. Mostly shared between SSH-1 and SSH-2.
1582  */
1583 static void ssh_pkt_ensure(struct Packet *pkt, int length)
1584 {
1585     if (pkt->maxlen < length) {
1586         unsigned char *body = pkt->body;
1587         int offset = body ? pkt->data - body : 0;
1588         pkt->maxlen = length + 256;
1589         pkt->data = sresize(pkt->data, pkt->maxlen + APIEXTRA, unsigned char);
1590         if (body) pkt->body = pkt->data + offset;
1591     }
1592 }
1593 static void ssh_pkt_adddata(struct Packet *pkt, void *data, int len)
1594 {
1595     if (pkt->logmode != PKTLOG_EMIT) {
1596         pkt->nblanks++;
1597         pkt->blanks = sresize(pkt->blanks, pkt->nblanks, struct logblank_t);
1598         assert(pkt->body);
1599         pkt->blanks[pkt->nblanks-1].offset = pkt->length -
1600                                              (pkt->body - pkt->data);
1601         pkt->blanks[pkt->nblanks-1].len = len;
1602         pkt->blanks[pkt->nblanks-1].type = pkt->logmode;
1603     }
1604     pkt->length += len;
1605     ssh_pkt_ensure(pkt, pkt->length);
1606     memcpy(pkt->data + pkt->length - len, data, len);
1607 }
1608 static void ssh_pkt_addbyte(struct Packet *pkt, unsigned char byte)
1609 {
1610     ssh_pkt_adddata(pkt, &byte, 1);
1611 }
1612 static void ssh2_pkt_addbool(struct Packet *pkt, unsigned char value)
1613 {
1614     ssh_pkt_adddata(pkt, &value, 1);
1615 }
1616 static void ssh_pkt_adduint32(struct Packet *pkt, unsigned long value)
1617 {
1618     unsigned char x[4];
1619     PUT_32BIT(x, value);
1620     ssh_pkt_adddata(pkt, x, 4);
1621 }
1622 static void ssh_pkt_addstring_start(struct Packet *pkt)
1623 {
1624     ssh_pkt_adduint32(pkt, 0);
1625     pkt->savedpos = pkt->length;
1626 }
1627 static void ssh_pkt_addstring_str(struct Packet *pkt, char *data)
1628 {
1629     ssh_pkt_adddata(pkt, data, strlen(data));
1630     PUT_32BIT(pkt->data + pkt->savedpos - 4, pkt->length - pkt->savedpos);
1631 }
1632 static void ssh_pkt_addstring_data(struct Packet *pkt, char *data, int len)
1633 {
1634     ssh_pkt_adddata(pkt, data, len);
1635     PUT_32BIT(pkt->data + pkt->savedpos - 4, pkt->length - pkt->savedpos);
1636 }
1637 static void ssh_pkt_addstring(struct Packet *pkt, char *data)
1638 {
1639     ssh_pkt_addstring_start(pkt);
1640     ssh_pkt_addstring_str(pkt, data);
1641 }
1642 static void ssh1_pkt_addmp(struct Packet *pkt, Bignum b)
1643 {
1644     int len = ssh1_bignum_length(b);
1645     unsigned char *data = snewn(len, char);
1646     (void) ssh1_write_bignum(data, b);
1647     ssh_pkt_adddata(pkt, data, len);
1648     sfree(data);
1649 }
1650 static unsigned char *ssh2_mpint_fmt(Bignum b, int *len)
1651 {
1652     unsigned char *p;
1653     int i, n = (bignum_bitcount(b) + 7) / 8;
1654     p = snewn(n + 1, unsigned char);
1655     p[0] = 0;
1656     for (i = 1; i <= n; i++)
1657         p[i] = bignum_byte(b, n - i);
1658     i = 0;
1659     while (i <= n && p[i] == 0 && (p[i + 1] & 0x80) == 0)
1660         i++;
1661     memmove(p, p + i, n + 1 - i);
1662     *len = n + 1 - i;
1663     return p;
1664 }
1665 static void ssh2_pkt_addmp(struct Packet *pkt, Bignum b)
1666 {
1667     unsigned char *p;
1668     int len;
1669     p = ssh2_mpint_fmt(b, &len);
1670     ssh_pkt_addstring_start(pkt);
1671     ssh_pkt_addstring_data(pkt, (char *)p, len);
1672     sfree(p);
1673 }
1674
1675 static struct Packet *ssh1_pkt_init(int pkt_type)
1676 {
1677     struct Packet *pkt = ssh_new_packet();
1678     pkt->length = 4 + 8;            /* space for length + max padding */
1679     ssh_pkt_addbyte(pkt, pkt_type);
1680     pkt->body = pkt->data + pkt->length;
1681     return pkt;
1682 }
1683
1684 /* For legacy code (SSH-1 and -2 packet construction used to be separate) */
1685 #define ssh2_pkt_ensure(pkt, length) ssh_pkt_ensure(pkt, length)
1686 #define ssh2_pkt_adddata(pkt, data, len) ssh_pkt_adddata(pkt, data, len)
1687 #define ssh2_pkt_addbyte(pkt, byte) ssh_pkt_addbyte(pkt, byte)
1688 #define ssh2_pkt_adduint32(pkt, value) ssh_pkt_adduint32(pkt, value)
1689 #define ssh2_pkt_addstring_start(pkt) ssh_pkt_addstring_start(pkt)
1690 #define ssh2_pkt_addstring_str(pkt, data) ssh_pkt_addstring_str(pkt, data)
1691 #define ssh2_pkt_addstring_data(pkt, data, len) ssh_pkt_addstring_data(pkt, data, len)
1692 #define ssh2_pkt_addstring(pkt, data) ssh_pkt_addstring(pkt, data)
1693
1694 static struct Packet *ssh2_pkt_init(int pkt_type)
1695 {
1696     struct Packet *pkt = ssh_new_packet();
1697     pkt->length = 5;
1698     pkt->forcepad = 0;
1699     ssh_pkt_addbyte(pkt, (unsigned char) pkt_type);
1700     pkt->body = pkt->data + pkt->length;
1701     return pkt;
1702 }
1703
1704 /*
1705  * Construct an SSH-2 final-form packet: compress it, encrypt it,
1706  * put the MAC on it. Final packet, ready to be sent, is stored in
1707  * pkt->data. Total length is returned.
1708  */
1709 static int ssh2_pkt_construct(Ssh ssh, struct Packet *pkt)
1710 {
1711     int cipherblk, maclen, padding, i;
1712
1713     if (ssh->logctx)
1714         log_packet(ssh->logctx, PKT_OUTGOING, pkt->data[5],
1715                    ssh2_pkt_type(ssh->pkt_ctx, pkt->data[5]),
1716                    pkt->body, pkt->length - (pkt->body - pkt->data),
1717                    pkt->nblanks, pkt->blanks);
1718     sfree(pkt->blanks); pkt->blanks = NULL;
1719     pkt->nblanks = 0;
1720
1721     /*
1722      * Compress packet payload.
1723      */
1724     {
1725         unsigned char *newpayload;
1726         int newlen;
1727         if (ssh->cscomp &&
1728             ssh->cscomp->compress(ssh->cs_comp_ctx, pkt->data + 5,
1729                                   pkt->length - 5,
1730                                   &newpayload, &newlen)) {
1731             pkt->length = 5;
1732             ssh2_pkt_adddata(pkt, newpayload, newlen);
1733             sfree(newpayload);
1734         }
1735     }
1736
1737     /*
1738      * Add padding. At least four bytes, and must also bring total
1739      * length (minus MAC) up to a multiple of the block size.
1740      * If pkt->forcepad is set, make sure the packet is at least that size
1741      * after padding.
1742      */
1743     cipherblk = ssh->cscipher ? ssh->cscipher->blksize : 8;  /* block size */
1744     cipherblk = cipherblk < 8 ? 8 : cipherblk;  /* or 8 if blksize < 8 */
1745     padding = 4;
1746     if (pkt->length + padding < pkt->forcepad)
1747         padding = pkt->forcepad - pkt->length;
1748     padding +=
1749         (cipherblk - (pkt->length + padding) % cipherblk) % cipherblk;
1750     assert(padding <= 255);
1751     maclen = ssh->csmac ? ssh->csmac->len : 0;
1752     ssh2_pkt_ensure(pkt, pkt->length + padding + maclen);
1753     pkt->data[4] = padding;
1754     for (i = 0; i < padding; i++)
1755         pkt->data[pkt->length + i] = random_byte();
1756     PUT_32BIT(pkt->data, pkt->length + padding - 4);
1757     if (ssh->csmac)
1758         ssh->csmac->generate(ssh->cs_mac_ctx, pkt->data,
1759                              pkt->length + padding,
1760                              ssh->v2_outgoing_sequence);
1761     ssh->v2_outgoing_sequence++;       /* whether or not we MACed */
1762
1763     if (ssh->cscipher)
1764         ssh->cscipher->encrypt(ssh->cs_cipher_ctx,
1765                                pkt->data, pkt->length + padding);
1766
1767     pkt->encrypted_len = pkt->length + padding;
1768
1769     /* Ready-to-send packet starts at pkt->data. We return length. */
1770     return pkt->length + padding + maclen;
1771 }
1772
1773 /*
1774  * Routines called from the main SSH code to send packets. There
1775  * are quite a few of these, because we have two separate
1776  * mechanisms for delaying the sending of packets:
1777  * 
1778  *  - In order to send an IGNORE message and a password message in
1779  *    a single fixed-length blob, we require the ability to
1780  *    concatenate the encrypted forms of those two packets _into_ a
1781  *    single blob and then pass it to our <network.h> transport
1782  *    layer in one go. Hence, there's a deferment mechanism which
1783  *    works after packet encryption.
1784  * 
1785  *  - In order to avoid sending any connection-layer messages
1786  *    during repeat key exchange, we have to queue up any such
1787  *    outgoing messages _before_ they are encrypted (and in
1788  *    particular before they're allocated sequence numbers), and
1789  *    then send them once we've finished.
1790  * 
1791  * I call these mechanisms `defer' and `queue' respectively, so as
1792  * to distinguish them reasonably easily.
1793  * 
1794  * The functions send_noqueue() and defer_noqueue() free the packet
1795  * structure they are passed. Every outgoing packet goes through
1796  * precisely one of these functions in its life; packets passed to
1797  * ssh2_pkt_send() or ssh2_pkt_defer() either go straight to one of
1798  * these or get queued, and then when the queue is later emptied
1799  * the packets are all passed to defer_noqueue().
1800  *
1801  * When using a CBC-mode cipher, it's necessary to ensure that an
1802  * attacker can't provide data to be encrypted using an IV that they
1803  * know.  We ensure this by prefixing each packet that might contain
1804  * user data with an SSH_MSG_IGNORE.  This is done using the deferral
1805  * mechanism, so in this case send_noqueue() ends up redirecting to
1806  * defer_noqueue().  If you don't like this inefficiency, don't use
1807  * CBC.
1808  */
1809
1810 static void ssh2_pkt_defer_noqueue(Ssh, struct Packet *, int);
1811 static void ssh_pkt_defersend(Ssh);
1812
1813 /*
1814  * Send an SSH-2 packet immediately, without queuing or deferring.
1815  */
1816 static void ssh2_pkt_send_noqueue(Ssh ssh, struct Packet *pkt)
1817 {
1818     int len;
1819     int backlog;
1820     if (ssh->cscipher != NULL && (ssh->cscipher->flags & SSH_CIPHER_IS_CBC)) {
1821         /* We need to send two packets, so use the deferral mechanism. */
1822         ssh2_pkt_defer_noqueue(ssh, pkt, FALSE);
1823         ssh_pkt_defersend(ssh);
1824         return;
1825     }
1826     len = ssh2_pkt_construct(ssh, pkt);
1827     backlog = sk_write(ssh->s, (char *)pkt->data, len);
1828     if (backlog > SSH_MAX_BACKLOG)
1829         ssh_throttle_all(ssh, 1, backlog);
1830
1831     ssh->outgoing_data_size += pkt->encrypted_len;
1832     if (!ssh->kex_in_progress &&
1833         ssh->max_data_size != 0 &&
1834         ssh->outgoing_data_size > ssh->max_data_size)
1835         do_ssh2_transport(ssh, "too much data sent", -1, NULL);
1836
1837     ssh_free_packet(pkt);
1838 }
1839
1840 /*
1841  * Defer an SSH-2 packet.
1842  */
1843 static void ssh2_pkt_defer_noqueue(Ssh ssh, struct Packet *pkt, int noignore)
1844 {
1845     int len;
1846     if (ssh->cscipher != NULL && (ssh->cscipher->flags & SSH_CIPHER_IS_CBC) &&
1847         ssh->deferred_len == 0 && !noignore) {
1848         /*
1849          * Interpose an SSH_MSG_IGNORE to ensure that user data don't
1850          * get encrypted with a known IV.
1851          */
1852         struct Packet *ipkt = ssh2_pkt_init(SSH2_MSG_IGNORE);
1853         ssh2_pkt_defer_noqueue(ssh, ipkt, TRUE);
1854     }
1855     len = ssh2_pkt_construct(ssh, pkt);
1856     if (ssh->deferred_len + len > ssh->deferred_size) {
1857         ssh->deferred_size = ssh->deferred_len + len + 128;
1858         ssh->deferred_send_data = sresize(ssh->deferred_send_data,
1859                                           ssh->deferred_size,
1860                                           unsigned char);
1861     }
1862     memcpy(ssh->deferred_send_data + ssh->deferred_len, pkt->data, len);
1863     ssh->deferred_len += len;
1864     ssh->deferred_data_size += pkt->encrypted_len;
1865     ssh_free_packet(pkt);
1866 }
1867
1868 /*
1869  * Queue an SSH-2 packet.
1870  */
1871 static void ssh2_pkt_queue(Ssh ssh, struct Packet *pkt)
1872 {
1873     assert(ssh->queueing);
1874
1875     if (ssh->queuelen >= ssh->queuesize) {
1876         ssh->queuesize = ssh->queuelen + 32;
1877         ssh->queue = sresize(ssh->queue, ssh->queuesize, struct Packet *);
1878     }
1879
1880     ssh->queue[ssh->queuelen++] = pkt;
1881 }
1882
1883 /*
1884  * Either queue or send a packet, depending on whether queueing is
1885  * set.
1886  */
1887 static void ssh2_pkt_send(Ssh ssh, struct Packet *pkt)
1888 {
1889     if (ssh->queueing)
1890         ssh2_pkt_queue(ssh, pkt);
1891     else
1892         ssh2_pkt_send_noqueue(ssh, pkt);
1893 }
1894
1895 #if 0 /* disused */
1896 /*
1897  * Either queue or defer a packet, depending on whether queueing is
1898  * set.
1899  */
1900 static void ssh2_pkt_defer(Ssh ssh, struct Packet *pkt)
1901 {
1902     if (ssh->queueing)
1903         ssh2_pkt_queue(ssh, pkt);
1904     else
1905         ssh2_pkt_defer_noqueue(ssh, pkt, FALSE);
1906 }
1907 #endif
1908
1909 /*
1910  * Send the whole deferred data block constructed by
1911  * ssh2_pkt_defer() or SSH-1's defer_packet().
1912  * 
1913  * The expected use of the defer mechanism is that you call
1914  * ssh2_pkt_defer() a few times, then call ssh_pkt_defersend(). If
1915  * not currently queueing, this simply sets up deferred_send_data
1916  * and then sends it. If we _are_ currently queueing, the calls to
1917  * ssh2_pkt_defer() put the deferred packets on to the queue
1918  * instead, and therefore ssh_pkt_defersend() has no deferred data
1919  * to send. Hence, there's no need to make it conditional on
1920  * ssh->queueing.
1921  */
1922 static void ssh_pkt_defersend(Ssh ssh)
1923 {
1924     int backlog;
1925     backlog = sk_write(ssh->s, (char *)ssh->deferred_send_data,
1926                        ssh->deferred_len);
1927     ssh->deferred_len = ssh->deferred_size = 0;
1928     sfree(ssh->deferred_send_data);
1929     ssh->deferred_send_data = NULL;
1930     if (backlog > SSH_MAX_BACKLOG)
1931         ssh_throttle_all(ssh, 1, backlog);
1932
1933     ssh->outgoing_data_size += ssh->deferred_data_size;
1934     if (!ssh->kex_in_progress &&
1935         ssh->max_data_size != 0 &&
1936         ssh->outgoing_data_size > ssh->max_data_size)
1937         do_ssh2_transport(ssh, "too much data sent", -1, NULL);
1938     ssh->deferred_data_size = 0;
1939 }
1940
1941 /*
1942  * Send all queued SSH-2 packets. We send them by means of
1943  * ssh2_pkt_defer_noqueue(), in case they included a pair of
1944  * packets that needed to be lumped together.
1945  */
1946 static void ssh2_pkt_queuesend(Ssh ssh)
1947 {
1948     int i;
1949
1950     assert(!ssh->queueing);
1951
1952     for (i = 0; i < ssh->queuelen; i++)
1953         ssh2_pkt_defer_noqueue(ssh, ssh->queue[i], FALSE);
1954     ssh->queuelen = 0;
1955
1956     ssh_pkt_defersend(ssh);
1957 }
1958
1959 #if 0
1960 void bndebug(char *string, Bignum b)
1961 {
1962     unsigned char *p;
1963     int i, len;
1964     p = ssh2_mpint_fmt(b, &len);
1965     debug(("%s", string));
1966     for (i = 0; i < len; i++)
1967         debug((" %02x", p[i]));
1968     debug(("\n"));
1969     sfree(p);
1970 }
1971 #endif
1972
1973 static void sha_mpint(SHA_State * s, Bignum b)
1974 {
1975     unsigned char *p;
1976     int len;
1977     p = ssh2_mpint_fmt(b, &len);
1978     sha_string(s, p, len);
1979     sfree(p);
1980 }
1981
1982 /*
1983  * Packet decode functions for both SSH-1 and SSH-2.
1984  */
1985 static unsigned long ssh_pkt_getuint32(struct Packet *pkt)
1986 {
1987     unsigned long value;
1988     if (pkt->length - pkt->savedpos < 4)
1989         return 0;                      /* arrgh, no way to decline (FIXME?) */
1990     value = GET_32BIT(pkt->body + pkt->savedpos);
1991     pkt->savedpos += 4;
1992     return value;
1993 }
1994 static int ssh2_pkt_getbool(struct Packet *pkt)
1995 {
1996     unsigned long value;
1997     if (pkt->length - pkt->savedpos < 1)
1998         return 0;                      /* arrgh, no way to decline (FIXME?) */
1999     value = pkt->body[pkt->savedpos] != 0;
2000     pkt->savedpos++;
2001     return value;
2002 }
2003 static void ssh_pkt_getstring(struct Packet *pkt, char **p, int *length)
2004 {
2005     int len;
2006     *p = NULL;
2007     *length = 0;
2008     if (pkt->length - pkt->savedpos < 4)
2009         return;
2010     len = GET_32BIT(pkt->body + pkt->savedpos);
2011     if (len < 0)
2012         return;
2013     *length = len;
2014     pkt->savedpos += 4;
2015     if (pkt->length - pkt->savedpos < *length)
2016         return;
2017     *p = (char *)(pkt->body + pkt->savedpos);
2018     pkt->savedpos += *length;
2019 }
2020 static void *ssh_pkt_getdata(struct Packet *pkt, int length)
2021 {
2022     if (pkt->length - pkt->savedpos < length)
2023         return NULL;
2024     pkt->savedpos += length;
2025     return pkt->body + (pkt->savedpos - length);
2026 }
2027 static int ssh1_pkt_getrsakey(struct Packet *pkt, struct RSAKey *key,
2028                               unsigned char **keystr)
2029 {
2030     int j;
2031
2032     j = makekey(pkt->body + pkt->savedpos,
2033                 pkt->length - pkt->savedpos,
2034                 key, keystr, 0);
2035
2036     if (j < 0)
2037         return FALSE;
2038     
2039     pkt->savedpos += j;
2040     assert(pkt->savedpos < pkt->length);
2041
2042     return TRUE;
2043 }
2044 static Bignum ssh1_pkt_getmp(struct Packet *pkt)
2045 {
2046     int j;
2047     Bignum b;
2048
2049     j = ssh1_read_bignum(pkt->body + pkt->savedpos,
2050                          pkt->length - pkt->savedpos, &b);
2051
2052     if (j < 0)
2053         return NULL;
2054
2055     pkt->savedpos += j;
2056     return b;
2057 }
2058 static Bignum ssh2_pkt_getmp(struct Packet *pkt)
2059 {
2060     char *p;
2061     int length;
2062     Bignum b;
2063
2064     ssh_pkt_getstring(pkt, &p, &length);
2065     if (!p)
2066         return NULL;
2067     if (p[0] & 0x80)
2068         return NULL;
2069     b = bignum_from_bytes((unsigned char *)p, length);
2070     return b;
2071 }
2072
2073 /*
2074  * Helper function to add an SSH-2 signature blob to a packet.
2075  * Expects to be shown the public key blob as well as the signature
2076  * blob. Normally works just like ssh2_pkt_addstring, but will
2077  * fiddle with the signature packet if necessary for
2078  * BUG_SSH2_RSA_PADDING.
2079  */
2080 static void ssh2_add_sigblob(Ssh ssh, struct Packet *pkt,
2081                              void *pkblob_v, int pkblob_len,
2082                              void *sigblob_v, int sigblob_len)
2083 {
2084     unsigned char *pkblob = (unsigned char *)pkblob_v;
2085     unsigned char *sigblob = (unsigned char *)sigblob_v;
2086
2087     /* dmemdump(pkblob, pkblob_len); */
2088     /* dmemdump(sigblob, sigblob_len); */
2089
2090     /*
2091      * See if this is in fact an ssh-rsa signature and a buggy
2092      * server; otherwise we can just do this the easy way.
2093      */
2094     if ((ssh->remote_bugs & BUG_SSH2_RSA_PADDING) &&
2095         (GET_32BIT(pkblob) == 7 && !memcmp(pkblob+4, "ssh-rsa", 7))) {
2096         int pos, len, siglen;
2097
2098         /*
2099          * Find the byte length of the modulus.
2100          */
2101
2102         pos = 4+7;                     /* skip over "ssh-rsa" */
2103         pos += 4 + GET_32BIT(pkblob+pos);   /* skip over exponent */
2104         len = GET_32BIT(pkblob+pos);   /* find length of modulus */
2105         pos += 4;                      /* find modulus itself */
2106         while (len > 0 && pkblob[pos] == 0)
2107             len--, pos++;
2108         /* debug(("modulus length is %d\n", len)); */
2109
2110         /*
2111          * Now find the signature integer.
2112          */
2113         pos = 4+7;                     /* skip over "ssh-rsa" */
2114         siglen = GET_32BIT(sigblob+pos);
2115         /* debug(("signature length is %d\n", siglen)); */
2116
2117         if (len != siglen) {
2118             unsigned char newlen[4];
2119             ssh2_pkt_addstring_start(pkt);
2120             ssh2_pkt_addstring_data(pkt, (char *)sigblob, pos);
2121             /* dmemdump(sigblob, pos); */
2122             pos += 4;                  /* point to start of actual sig */
2123             PUT_32BIT(newlen, len);
2124             ssh2_pkt_addstring_data(pkt, (char *)newlen, 4);
2125             /* dmemdump(newlen, 4); */
2126             newlen[0] = 0;
2127             while (len-- > siglen) {
2128                 ssh2_pkt_addstring_data(pkt, (char *)newlen, 1);
2129                 /* dmemdump(newlen, 1); */
2130             }
2131             ssh2_pkt_addstring_data(pkt, (char *)(sigblob+pos), siglen);
2132             /* dmemdump(sigblob+pos, siglen); */
2133             return;
2134         }
2135
2136         /* Otherwise fall through and do it the easy way. */
2137     }
2138
2139     ssh2_pkt_addstring_start(pkt);
2140     ssh2_pkt_addstring_data(pkt, (char *)sigblob, sigblob_len);
2141 }
2142
2143 /*
2144  * Examine the remote side's version string and compare it against
2145  * a list of known buggy implementations.
2146  */
2147 static void ssh_detect_bugs(Ssh ssh, char *vstring)
2148 {
2149     char *imp;                         /* pointer to implementation part */
2150     imp = vstring;
2151     imp += strcspn(imp, "-");
2152     if (*imp) imp++;
2153     imp += strcspn(imp, "-");
2154     if (*imp) imp++;
2155
2156     ssh->remote_bugs = 0;
2157
2158     if (ssh->cfg.sshbug_ignore1 == FORCE_ON ||
2159         (ssh->cfg.sshbug_ignore1 == AUTO &&
2160          (!strcmp(imp, "1.2.18") || !strcmp(imp, "1.2.19") ||
2161           !strcmp(imp, "1.2.20") || !strcmp(imp, "1.2.21") ||
2162           !strcmp(imp, "1.2.22") || !strcmp(imp, "Cisco-1.25") ||
2163           !strcmp(imp, "OSU_1.4alpha3") || !strcmp(imp, "OSU_1.5alpha4")))) {
2164         /*
2165          * These versions don't support SSH1_MSG_IGNORE, so we have
2166          * to use a different defence against password length
2167          * sniffing.
2168          */
2169         ssh->remote_bugs |= BUG_CHOKES_ON_SSH1_IGNORE;
2170         logevent("We believe remote version has SSH-1 ignore bug");
2171     }
2172
2173     if (ssh->cfg.sshbug_plainpw1 == FORCE_ON ||
2174         (ssh->cfg.sshbug_plainpw1 == AUTO &&
2175          (!strcmp(imp, "Cisco-1.25") || !strcmp(imp, "OSU_1.4alpha3")))) {
2176         /*
2177          * These versions need a plain password sent; they can't
2178          * handle having a null and a random length of data after
2179          * the password.
2180          */
2181         ssh->remote_bugs |= BUG_NEEDS_SSH1_PLAIN_PASSWORD;
2182         logevent("We believe remote version needs a plain SSH-1 password");
2183     }
2184
2185     if (ssh->cfg.sshbug_rsa1 == FORCE_ON ||
2186         (ssh->cfg.sshbug_rsa1 == AUTO &&
2187          (!strcmp(imp, "Cisco-1.25")))) {
2188         /*
2189          * These versions apparently have no clue whatever about
2190          * RSA authentication and will panic and die if they see
2191          * an AUTH_RSA message.
2192          */
2193         ssh->remote_bugs |= BUG_CHOKES_ON_RSA;
2194         logevent("We believe remote version can't handle SSH-1 RSA authentication");
2195     }
2196
2197     if (ssh->cfg.sshbug_hmac2 == FORCE_ON ||
2198         (ssh->cfg.sshbug_hmac2 == AUTO &&
2199          !wc_match("* VShell", imp) &&
2200          (wc_match("2.1.0*", imp) || wc_match("2.0.*", imp) ||
2201           wc_match("2.2.0*", imp) || wc_match("2.3.0*", imp) ||
2202           wc_match("2.1 *", imp)))) {
2203         /*
2204          * These versions have the HMAC bug.
2205          */
2206         ssh->remote_bugs |= BUG_SSH2_HMAC;
2207         logevent("We believe remote version has SSH-2 HMAC bug");
2208     }
2209
2210     if (ssh->cfg.sshbug_derivekey2 == FORCE_ON ||
2211         (ssh->cfg.sshbug_derivekey2 == AUTO &&
2212          !wc_match("* VShell", imp) &&
2213          (wc_match("2.0.0*", imp) || wc_match("2.0.10*", imp) ))) {
2214         /*
2215          * These versions have the key-derivation bug (failing to
2216          * include the literal shared secret in the hashes that
2217          * generate the keys).
2218          */
2219         ssh->remote_bugs |= BUG_SSH2_DERIVEKEY;
2220         logevent("We believe remote version has SSH-2 key-derivation bug");
2221     }
2222
2223     if (ssh->cfg.sshbug_rsapad2 == FORCE_ON ||
2224         (ssh->cfg.sshbug_rsapad2 == AUTO &&
2225          (wc_match("OpenSSH_2.[5-9]*", imp) ||
2226           wc_match("OpenSSH_3.[0-2]*", imp)))) {
2227         /*
2228          * These versions have the SSH-2 RSA padding bug.
2229          */
2230         ssh->remote_bugs |= BUG_SSH2_RSA_PADDING;
2231         logevent("We believe remote version has SSH-2 RSA padding bug");
2232     }
2233
2234     if (ssh->cfg.sshbug_pksessid2 == FORCE_ON ||
2235         (ssh->cfg.sshbug_pksessid2 == AUTO &&
2236          wc_match("OpenSSH_2.[0-2]*", imp))) {
2237         /*
2238          * These versions have the SSH-2 session-ID bug in
2239          * public-key authentication.
2240          */
2241         ssh->remote_bugs |= BUG_SSH2_PK_SESSIONID;
2242         logevent("We believe remote version has SSH-2 public-key-session-ID bug");
2243     }
2244
2245     if (ssh->cfg.sshbug_rekey2 == FORCE_ON ||
2246         (ssh->cfg.sshbug_rekey2 == AUTO &&
2247          (wc_match("OpenSSH_2.[0-4]*", imp) ||
2248           wc_match("OpenSSH_2.5.[0-3]*", imp) ||
2249           wc_match("Sun_SSH_1.0", imp) ||
2250           wc_match("Sun_SSH_1.0.1", imp)))) {
2251         /*
2252          * These versions have the SSH-2 rekey bug.
2253          */
2254         ssh->remote_bugs |= BUG_SSH2_REKEY;
2255         logevent("We believe remote version has SSH-2 rekey bug");
2256     }
2257 }
2258
2259 /*
2260  * The `software version' part of an SSH version string is required
2261  * to contain no spaces or minus signs.
2262  */
2263 static void ssh_fix_verstring(char *str)
2264 {
2265     /* Eat "SSH-<protoversion>-". */
2266     assert(*str == 'S'); str++;
2267     assert(*str == 'S'); str++;
2268     assert(*str == 'H'); str++;
2269     assert(*str == '-'); str++;
2270     while (*str && *str != '-') str++;
2271     assert(*str == '-'); str++;
2272
2273     /* Convert minus signs and spaces in the remaining string into
2274      * underscores. */
2275     while (*str) {
2276         if (*str == '-' || *str == ' ')
2277             *str = '_';
2278         str++;
2279     }
2280 }
2281
2282 static int do_ssh_init(Ssh ssh, unsigned char c)
2283 {
2284     struct do_ssh_init_state {
2285         int vslen;
2286         char version[10];
2287         char *vstring;
2288         int vstrsize;
2289         int i;
2290         int proto1, proto2;
2291     };
2292     crState(do_ssh_init_state);
2293
2294     crBegin(ssh->do_ssh_init_crstate);
2295
2296     /* Search for the string "SSH-" in the input. */
2297     s->i = 0;
2298     while (1) {
2299         static const int transS[] = { 1, 2, 2, 1 };
2300         static const int transH[] = { 0, 0, 3, 0 };
2301         static const int transminus[] = { 0, 0, 0, -1 };
2302         if (c == 'S')
2303             s->i = transS[s->i];
2304         else if (c == 'H')
2305             s->i = transH[s->i];
2306         else if (c == '-')
2307             s->i = transminus[s->i];
2308         else
2309             s->i = 0;
2310         if (s->i < 0)
2311             break;
2312         crReturn(1);                   /* get another character */
2313     }
2314
2315     s->vstrsize = 16;
2316     s->vstring = snewn(s->vstrsize, char);
2317     strcpy(s->vstring, "SSH-");
2318     s->vslen = 4;
2319     s->i = 0;
2320     while (1) {
2321         crReturn(1);                   /* get another char */
2322         if (s->vslen >= s->vstrsize - 1) {
2323             s->vstrsize += 16;
2324             s->vstring = sresize(s->vstring, s->vstrsize, char);
2325         }
2326         s->vstring[s->vslen++] = c;
2327         if (s->i >= 0) {
2328             if (c == '-') {
2329                 s->version[s->i] = '\0';
2330                 s->i = -1;
2331             } else if (s->i < sizeof(s->version) - 1)
2332                 s->version[s->i++] = c;
2333         } else if (c == '\012')
2334             break;
2335     }
2336
2337     ssh->agentfwd_enabled = FALSE;
2338     ssh->rdpkt2_state.incoming_sequence = 0;
2339
2340     s->vstring[s->vslen] = 0;
2341     s->vstring[strcspn(s->vstring, "\015\012")] = '\0';/* remove EOL chars */
2342     logeventf(ssh, "Server version: %s", s->vstring);
2343     ssh_detect_bugs(ssh, s->vstring);
2344
2345     /*
2346      * Decide which SSH protocol version to support.
2347      */
2348
2349     /* Anything strictly below "2.0" means protocol 1 is supported. */
2350     s->proto1 = ssh_versioncmp(s->version, "2.0") < 0;
2351     /* Anything greater or equal to "1.99" means protocol 2 is supported. */
2352     s->proto2 = ssh_versioncmp(s->version, "1.99") >= 0;
2353
2354     if (ssh->cfg.sshprot == 0 && !s->proto1) {
2355         bombout(("SSH protocol version 1 required by user but not provided by server"));
2356         crStop(0);
2357     }
2358     if (ssh->cfg.sshprot == 3 && !s->proto2) {
2359         bombout(("SSH protocol version 2 required by user but not provided by server"));
2360         crStop(0);
2361     }
2362
2363     {
2364         char *verstring;
2365
2366         if (s->proto2 && (ssh->cfg.sshprot >= 2 || !s->proto1)) {
2367             /*
2368              * Construct a v2 version string.
2369              */
2370             verstring = dupprintf("SSH-2.0-%s\015\012", sshver);
2371             ssh->version = 2;
2372         } else {
2373             /*
2374              * Construct a v1 version string.
2375              */
2376             verstring = dupprintf("SSH-%s-%s\012",
2377                                   (ssh_versioncmp(s->version, "1.5") <= 0 ?
2378                                    s->version : "1.5"),
2379                                   sshver);
2380             ssh->version = 1;
2381         }
2382
2383         ssh_fix_verstring(verstring);
2384
2385         if (ssh->version == 2) {
2386             /*
2387              * Hash our version string and their version string.
2388              */
2389             SHA_Init(&ssh->exhashbase);
2390             sha_string(&ssh->exhashbase, verstring,
2391                        strcspn(verstring, "\015\012"));
2392             sha_string(&ssh->exhashbase, s->vstring,
2393                        strcspn(s->vstring, "\015\012"));
2394
2395             /*
2396              * Initialise SSH-2 protocol.
2397              */
2398             ssh->protocol = ssh2_protocol;
2399             ssh2_protocol_setup(ssh);
2400             ssh->s_rdpkt = ssh2_rdpkt;
2401         } else {
2402             /*
2403              * Initialise SSH-1 protocol.
2404              */
2405             ssh->protocol = ssh1_protocol;
2406             ssh1_protocol_setup(ssh);
2407             ssh->s_rdpkt = ssh1_rdpkt;
2408         }
2409         logeventf(ssh, "We claim version: %.*s",
2410                   strcspn(verstring, "\015\012"), verstring);
2411         sk_write(ssh->s, verstring, strlen(verstring));
2412         sfree(verstring);
2413         if (ssh->version == 2)
2414             do_ssh2_transport(ssh, NULL, -1, NULL);
2415     }
2416
2417     logeventf(ssh, "Using SSH protocol version %d", ssh->version);
2418
2419     update_specials_menu(ssh->frontend);
2420     ssh->state = SSH_STATE_BEFORE_SIZE;
2421     ssh->pinger = pinger_new(&ssh->cfg, &ssh_backend, ssh);
2422
2423     sfree(s->vstring);
2424
2425     crFinish(0);
2426 }
2427
2428 static void ssh_process_incoming_data(Ssh ssh,
2429                                       unsigned char **data, int *datalen)
2430 {
2431     struct Packet *pktin = ssh->s_rdpkt(ssh, data, datalen);
2432     if (pktin) {
2433         ssh->protocol(ssh, NULL, 0, pktin);
2434         ssh_free_packet(pktin);
2435     }
2436 }
2437
2438 static void ssh_queue_incoming_data(Ssh ssh,
2439                                     unsigned char **data, int *datalen)
2440 {
2441     bufchain_add(&ssh->queued_incoming_data, *data, *datalen);
2442     *data += *datalen;
2443     *datalen = 0;
2444 }
2445
2446 static void ssh_process_queued_incoming_data(Ssh ssh)
2447 {
2448     void *vdata;
2449     unsigned char *data;
2450     int len, origlen;
2451
2452     while (!ssh->frozen && bufchain_size(&ssh->queued_incoming_data)) {
2453         bufchain_prefix(&ssh->queued_incoming_data, &vdata, &len);
2454         data = vdata;
2455         origlen = len;
2456
2457         while (!ssh->frozen && len > 0)
2458             ssh_process_incoming_data(ssh, &data, &len);
2459
2460         if (origlen > len)
2461             bufchain_consume(&ssh->queued_incoming_data, origlen - len);
2462     }
2463 }
2464
2465 static void ssh_set_frozen(Ssh ssh, int frozen)
2466 {
2467     if (ssh->s)
2468         sk_set_frozen(ssh->s, frozen);
2469     ssh->frozen = frozen;
2470 }
2471
2472 static void ssh_gotdata(Ssh ssh, unsigned char *data, int datalen)
2473 {
2474     crBegin(ssh->ssh_gotdata_crstate);
2475
2476     /*
2477      * To begin with, feed the characters one by one to the
2478      * protocol initialisation / selection function do_ssh_init().
2479      * When that returns 0, we're done with the initial greeting
2480      * exchange and can move on to packet discipline.
2481      */
2482     while (1) {
2483         int ret;                       /* need not be kept across crReturn */
2484         if (datalen == 0)
2485             crReturnV;                 /* more data please */
2486         ret = do_ssh_init(ssh, *data);
2487         data++;
2488         datalen--;
2489         if (ret == 0)
2490             break;
2491     }
2492
2493     /*
2494      * We emerge from that loop when the initial negotiation is
2495      * over and we have selected an s_rdpkt function. Now pass
2496      * everything to s_rdpkt, and then pass the resulting packets
2497      * to the proper protocol handler.
2498      */
2499     if (datalen == 0)
2500         crReturnV;
2501
2502     /*
2503      * Process queued data if there is any.
2504      */
2505     ssh_process_queued_incoming_data(ssh);
2506
2507     while (1) {
2508         while (datalen > 0) {
2509             if (ssh->frozen)
2510                 ssh_queue_incoming_data(ssh, &data, &datalen);
2511
2512             ssh_process_incoming_data(ssh, &data, &datalen);
2513
2514             if (ssh->state == SSH_STATE_CLOSED)
2515                 return;
2516         }
2517         crReturnV;
2518     }
2519     crFinishV;
2520 }
2521
2522 static int ssh_do_close(Ssh ssh, int notify_exit)
2523 {
2524     int ret = 0;
2525     struct ssh_channel *c;
2526
2527     ssh->state = SSH_STATE_CLOSED;
2528     expire_timer_context(ssh);
2529     if (ssh->s) {
2530         sk_close(ssh->s);
2531         ssh->s = NULL;
2532         if (notify_exit)
2533             notify_remote_exit(ssh->frontend);
2534         else
2535             ret = 1;
2536     }
2537     /*
2538      * Now we must shut down any port- and X-forwarded channels going
2539      * through this connection.
2540      */
2541     if (ssh->channels) {
2542         while (NULL != (c = index234(ssh->channels, 0))) {
2543             switch (c->type) {
2544               case CHAN_X11:
2545                 x11_close(c->u.x11.s);
2546                 break;
2547               case CHAN_SOCKDATA:
2548                 pfd_close(c->u.pfd.s);
2549                 break;
2550             }
2551             del234(ssh->channels, c); /* moving next one to index 0 */
2552             if (ssh->version == 2)
2553                 bufchain_clear(&c->v.v2.outbuffer);
2554             sfree(c);
2555         }
2556     }
2557     /*
2558      * Go through port-forwardings, and close any associated
2559      * listening sockets.
2560      */
2561     if (ssh->portfwds) {
2562         struct ssh_portfwd *pf;
2563         while (NULL != (pf = index234(ssh->portfwds, 0))) {
2564             /* Dispose of any listening socket. */
2565             if (pf->local)
2566                 pfd_terminate(pf->local);
2567             del234(ssh->portfwds, pf); /* moving next one to index 0 */
2568             free_portfwd(pf);
2569         }
2570     }
2571
2572     return ret;
2573 }
2574
2575 static void ssh_log(Plug plug, int type, SockAddr addr, int port,
2576                     const char *error_msg, int error_code)
2577 {
2578     Ssh ssh = (Ssh) plug;
2579     char addrbuf[256], *msg;
2580
2581     sk_getaddr(addr, addrbuf, lenof(addrbuf));
2582
2583     if (type == 0)
2584         msg = dupprintf("Connecting to %s port %d", addrbuf, port);
2585     else
2586         msg = dupprintf("Failed to connect to %s: %s", addrbuf, error_msg);
2587
2588     logevent(msg);
2589     sfree(msg);
2590 }
2591
2592 static int ssh_closing(Plug plug, const char *error_msg, int error_code,
2593                        int calling_back)
2594 {
2595     Ssh ssh = (Ssh) plug;
2596     int need_notify = ssh_do_close(ssh, FALSE);
2597
2598     if (!error_msg) {
2599         if (!ssh->close_expected)
2600             error_msg = "Server unexpectedly closed network connection";
2601         else
2602             error_msg = "Server closed network connection";
2603     }
2604
2605     if (need_notify)
2606         notify_remote_exit(ssh->frontend);
2607
2608     if (error_msg)
2609         logevent(error_msg);
2610     if (!ssh->close_expected || !ssh->clean_exit)
2611         connection_fatal(ssh->frontend, "%s", error_msg);
2612     return 0;
2613 }
2614
2615 static int ssh_receive(Plug plug, int urgent, char *data, int len)
2616 {
2617     Ssh ssh = (Ssh) plug;
2618     ssh_gotdata(ssh, (unsigned char *)data, len);
2619     if (ssh->state == SSH_STATE_CLOSED) {
2620         ssh_do_close(ssh, TRUE);
2621         return 0;
2622     }
2623     return 1;
2624 }
2625
2626 static void ssh_sent(Plug plug, int bufsize)
2627 {
2628     Ssh ssh = (Ssh) plug;
2629     /*
2630      * If the send backlog on the SSH socket itself clears, we
2631      * should unthrottle the whole world if it was throttled.
2632      */
2633     if (bufsize < SSH_MAX_BACKLOG)
2634         ssh_throttle_all(ssh, 0, bufsize);
2635 }
2636
2637 /*
2638  * Connect to specified host and port.
2639  * Returns an error message, or NULL on success.
2640  * Also places the canonical host name into `realhost'. It must be
2641  * freed by the caller.
2642  */
2643 static const char *connect_to_host(Ssh ssh, char *host, int port,
2644                                    char **realhost, int nodelay, int keepalive)
2645 {
2646     static const struct plug_function_table fn_table = {
2647         ssh_log,
2648         ssh_closing,
2649         ssh_receive,
2650         ssh_sent,
2651         NULL
2652     };
2653
2654     SockAddr addr;
2655     const char *err;
2656
2657     ssh->savedhost = snewn(1 + strlen(host), char);
2658     strcpy(ssh->savedhost, host);
2659
2660     if (port < 0)
2661         port = 22;                     /* default ssh port */
2662     ssh->savedport = port;
2663
2664     /*
2665      * Try to find host.
2666      */
2667     logeventf(ssh, "Looking up host \"%s\"%s", host,
2668               (ssh->cfg.addressfamily == ADDRTYPE_IPV4 ? " (IPv4)" :
2669                (ssh->cfg.addressfamily == ADDRTYPE_IPV6 ? " (IPv6)" : "")));
2670     addr = name_lookup(host, port, realhost, &ssh->cfg,
2671                        ssh->cfg.addressfamily);
2672     if ((err = sk_addr_error(addr)) != NULL) {
2673         sk_addr_free(addr);
2674         return err;
2675     }
2676
2677     /*
2678      * Open socket.
2679      */
2680     ssh->fn = &fn_table;
2681     ssh->s = new_connection(addr, *realhost, port,
2682                             0, 1, nodelay, keepalive, (Plug) ssh, &ssh->cfg);
2683     if ((err = sk_socket_error(ssh->s)) != NULL) {
2684         ssh->s = NULL;
2685         notify_remote_exit(ssh->frontend);
2686         return err;
2687     }
2688
2689     return NULL;
2690 }
2691
2692 /*
2693  * Throttle or unthrottle the SSH connection.
2694  */
2695 static void ssh1_throttle(Ssh ssh, int adjust)
2696 {
2697     int old_count = ssh->v1_throttle_count;
2698     ssh->v1_throttle_count += adjust;
2699     assert(ssh->v1_throttle_count >= 0);
2700     if (ssh->v1_throttle_count && !old_count) {
2701         ssh_set_frozen(ssh, 1);
2702     } else if (!ssh->v1_throttle_count && old_count) {
2703         ssh_set_frozen(ssh, 0);
2704     }
2705 }
2706
2707 /*
2708  * Throttle or unthrottle _all_ local data streams (for when sends
2709  * on the SSH connection itself back up).
2710  */
2711 static void ssh_throttle_all(Ssh ssh, int enable, int bufsize)
2712 {
2713     int i;
2714     struct ssh_channel *c;
2715
2716     if (enable == ssh->throttled_all)
2717         return;
2718     ssh->throttled_all = enable;
2719     ssh->overall_bufsize = bufsize;
2720     if (!ssh->channels)
2721         return;
2722     for (i = 0; NULL != (c = index234(ssh->channels, i)); i++) {
2723         switch (c->type) {
2724           case CHAN_MAINSESSION:
2725             /*
2726              * This is treated separately, outside the switch.
2727              */
2728             break;
2729           case CHAN_X11:
2730             x11_override_throttle(c->u.x11.s, enable);
2731             break;
2732           case CHAN_AGENT:
2733             /* Agent channels require no buffer management. */
2734             break;
2735           case CHAN_SOCKDATA:
2736             pfd_override_throttle(c->u.pfd.s, enable);
2737             break;
2738         }
2739     }
2740 }
2741
2742 /*
2743  * Username and password input, abstracted off into routines
2744  * reusable in several places - even between SSH-1 and SSH-2.
2745  */
2746
2747 /* Set up a username or password input loop on a given buffer. */
2748 static void setup_userpass_input(Ssh ssh, char *buffer, int buflen, int echo)
2749 {
2750     ssh->userpass_input_buffer = buffer;
2751     ssh->userpass_input_buflen = buflen;
2752     ssh->userpass_input_bufpos = 0;
2753     ssh->userpass_input_echo = echo;
2754 }
2755
2756 /*
2757  * Process some terminal data in the course of username/password
2758  * input. Returns >0 for success (line of input returned in
2759  * buffer), <0 for failure (user hit ^C/^D, bomb out and exit), 0
2760  * for inconclusive (keep waiting for more input please).
2761  */
2762 static int process_userpass_input(Ssh ssh, unsigned char *in, int inlen)
2763 {
2764     char c;
2765
2766     while (inlen--) {
2767         switch (c = *in++) {
2768           case 10:
2769           case 13:
2770             ssh->userpass_input_buffer[ssh->userpass_input_bufpos] = 0;
2771             ssh->userpass_input_buffer[ssh->userpass_input_buflen-1] = 0;
2772             return +1;
2773             break;
2774           case 8:
2775           case 127:
2776             if (ssh->userpass_input_bufpos > 0) {
2777                 if (ssh->userpass_input_echo)
2778                     c_write_str(ssh, "\b \b");
2779                 ssh->userpass_input_bufpos--;
2780             }
2781             break;
2782           case 21:
2783           case 27:
2784             while (ssh->userpass_input_bufpos > 0) {
2785                 if (ssh->userpass_input_echo)
2786                     c_write_str(ssh, "\b \b");
2787                 ssh->userpass_input_bufpos--;
2788             }
2789             break;
2790           case 3:
2791           case 4:
2792             return -1;
2793             break;
2794           default:
2795             /*
2796              * This simplistic check for printability is disabled
2797              * when we're doing password input, because some people
2798              * have control characters in their passwords.o
2799              */
2800             if ((!ssh->userpass_input_echo ||
2801                  (c >= ' ' && c <= '~') ||
2802                  ((unsigned char) c >= 160))
2803                 && ssh->userpass_input_bufpos < ssh->userpass_input_buflen-1) {
2804                 ssh->userpass_input_buffer[ssh->userpass_input_bufpos++] = c;
2805                 if (ssh->userpass_input_echo)
2806                     c_write(ssh, &c, 1);
2807             }
2808             break;
2809         }
2810     }
2811     return 0;
2812 }
2813
2814 static void ssh_agent_callback(void *sshv, void *reply, int replylen)
2815 {
2816     Ssh ssh = (Ssh) sshv;
2817
2818     ssh->agent_response = reply;
2819     ssh->agent_response_len = replylen;
2820
2821     if (ssh->version == 1)
2822         do_ssh1_login(ssh, NULL, -1, NULL);
2823     else
2824         do_ssh2_authconn(ssh, NULL, -1, NULL);
2825 }
2826
2827 static void ssh_dialog_callback(void *sshv, int ret)
2828 {
2829     Ssh ssh = (Ssh) sshv;
2830
2831     ssh->user_response = ret;
2832
2833     if (ssh->version == 1)
2834         do_ssh1_login(ssh, NULL, -1, NULL);
2835     else
2836         do_ssh2_transport(ssh, NULL, -1, NULL);
2837
2838     /*
2839      * This may have unfrozen the SSH connection, so do a
2840      * queued-data run.
2841      */
2842     ssh_process_queued_incoming_data(ssh);
2843 }
2844
2845 static void ssh_agentf_callback(void *cv, void *reply, int replylen)
2846 {
2847     struct ssh_channel *c = (struct ssh_channel *)cv;
2848     Ssh ssh = c->ssh;
2849     void *sentreply = reply;
2850
2851     if (!sentreply) {
2852         /* Fake SSH_AGENT_FAILURE. */
2853         sentreply = "\0\0\0\1\5";
2854         replylen = 5;
2855     }
2856     if (ssh->version == 2) {
2857         ssh2_add_channel_data(c, sentreply, replylen);
2858         ssh2_try_send(c);
2859     } else {
2860         send_packet(ssh, SSH1_MSG_CHANNEL_DATA,
2861                     PKT_INT, c->remoteid,
2862                     PKTT_DATA,
2863                     PKT_INT, replylen,
2864                     PKT_DATA, sentreply, replylen,
2865                     PKTT_OTHER,
2866                     PKT_END);
2867     }
2868     if (reply)
2869         sfree(reply);
2870 }
2871
2872 /*
2873  * Client-initiated disconnection. Send a DISCONNECT if `wire_reason'
2874  * non-NULL, otherwise just close the connection. `client_reason' == NULL
2875  * => log `wire_reason'.
2876  */
2877 static void ssh_disconnect(Ssh ssh, char *client_reason, char *wire_reason,
2878                            int code, int clean_exit)
2879 {
2880     char *error;
2881     if (!client_reason)
2882         client_reason = wire_reason;
2883     if (client_reason)
2884         error = dupprintf("Disconnected: %s", client_reason);
2885     else
2886         error = dupstr("Disconnected");
2887     if (wire_reason) {
2888         if (ssh->version == 1) {
2889             send_packet(ssh, SSH1_MSG_DISCONNECT, PKT_STR, wire_reason,
2890                         PKT_END);
2891         } else if (ssh->version == 2) {
2892             struct Packet *pktout = ssh2_pkt_init(SSH2_MSG_DISCONNECT);
2893             ssh2_pkt_adduint32(pktout, code);
2894             ssh2_pkt_addstring(pktout, wire_reason);
2895             ssh2_pkt_addstring(pktout, "en");   /* language tag */
2896             ssh2_pkt_send_noqueue(ssh, pktout);
2897         }
2898     }
2899     ssh->close_expected = TRUE;
2900     ssh->clean_exit = clean_exit;
2901     ssh_closing((Plug)ssh, error, 0, 0);
2902     sfree(error);
2903 }
2904
2905 /*
2906  * Handle the key exchange and user authentication phases.
2907  */
2908 static int do_ssh1_login(Ssh ssh, unsigned char *in, int inlen,
2909                          struct Packet *pktin)
2910 {
2911     int i, j, ret;
2912     unsigned char cookie[8], *ptr;
2913     struct RSAKey servkey, hostkey;
2914     struct MD5Context md5c;
2915     struct do_ssh1_login_state {
2916         int len;
2917         unsigned char *rsabuf, *keystr1, *keystr2;
2918         unsigned long supported_ciphers_mask, supported_auths_mask;
2919         int tried_publickey, tried_agent;
2920         int tis_auth_refused, ccard_auth_refused;
2921         unsigned char session_id[16];
2922         int cipher_type;
2923         char username[100];
2924         void *publickey_blob;
2925         int publickey_bloblen;
2926         char password[100];
2927         char prompt[200];
2928         int pos;
2929         char c;
2930         int pwpkt_type;
2931         unsigned char request[5], *response, *p;
2932         int responselen;
2933         int keyi, nkeys;
2934         int authed;
2935         struct RSAKey key;
2936         Bignum challenge;
2937         char *commentp;
2938         int commentlen;
2939         int dlgret;
2940     };
2941     crState(do_ssh1_login_state);
2942
2943     crBegin(ssh->do_ssh1_login_crstate);
2944
2945     if (!pktin)
2946         crWaitUntil(pktin);
2947
2948     if (pktin->type != SSH1_SMSG_PUBLIC_KEY) {
2949         bombout(("Public key packet not received"));
2950         crStop(0);
2951     }
2952
2953     logevent("Received public keys");
2954
2955     ptr = ssh_pkt_getdata(pktin, 8);
2956     if (!ptr) {
2957         bombout(("SSH-1 public key packet stopped before random cookie"));
2958         crStop(0);
2959     }
2960     memcpy(cookie, ptr, 8);
2961
2962     if (!ssh1_pkt_getrsakey(pktin, &servkey, &s->keystr1) ||
2963         !ssh1_pkt_getrsakey(pktin, &hostkey, &s->keystr2)) {    
2964         bombout(("Failed to read SSH-1 public keys from public key packet"));
2965         crStop(0);
2966     }
2967
2968     /*
2969      * Log the host key fingerprint.
2970      */
2971     {
2972         char logmsg[80];
2973         logevent("Host key fingerprint is:");
2974         strcpy(logmsg, "      ");
2975         hostkey.comment = NULL;
2976         rsa_fingerprint(logmsg + strlen(logmsg),
2977                         sizeof(logmsg) - strlen(logmsg), &hostkey);
2978         logevent(logmsg);
2979     }
2980
2981     ssh->v1_remote_protoflags = ssh_pkt_getuint32(pktin);
2982     s->supported_ciphers_mask = ssh_pkt_getuint32(pktin);
2983     s->supported_auths_mask = ssh_pkt_getuint32(pktin);
2984
2985     ssh->v1_local_protoflags =
2986         ssh->v1_remote_protoflags & SSH1_PROTOFLAGS_SUPPORTED;
2987     ssh->v1_local_protoflags |= SSH1_PROTOFLAG_SCREEN_NUMBER;
2988
2989     MD5Init(&md5c);
2990     MD5Update(&md5c, s->keystr2, hostkey.bytes);
2991     MD5Update(&md5c, s->keystr1, servkey.bytes);
2992     MD5Update(&md5c, cookie, 8);
2993     MD5Final(s->session_id, &md5c);
2994
2995     for (i = 0; i < 32; i++)
2996         ssh->session_key[i] = random_byte();
2997
2998     /*
2999      * Verify that the `bits' and `bytes' parameters match.
3000      */
3001     if (hostkey.bits > hostkey.bytes * 8 ||
3002         servkey.bits > servkey.bytes * 8) {
3003         bombout(("SSH-1 public keys were badly formatted"));
3004         crStop(0);
3005     }
3006
3007     s->len = (hostkey.bytes > servkey.bytes ? hostkey.bytes : servkey.bytes);
3008
3009     s->rsabuf = snewn(s->len, unsigned char);
3010
3011     /*
3012      * Verify the host key.
3013      */
3014     {
3015         /*
3016          * First format the key into a string.
3017          */
3018         int len = rsastr_len(&hostkey);
3019         char fingerprint[100];
3020         char *keystr = snewn(len, char);
3021         rsastr_fmt(keystr, &hostkey);
3022         rsa_fingerprint(fingerprint, sizeof(fingerprint), &hostkey);
3023
3024         ssh_set_frozen(ssh, 1);
3025         s->dlgret = verify_ssh_host_key(ssh->frontend,
3026                                         ssh->savedhost, ssh->savedport,
3027                                         "rsa", keystr, fingerprint,
3028                                         ssh_dialog_callback, ssh);
3029         sfree(keystr);
3030         if (s->dlgret < 0) {
3031             do {
3032                 crReturn(0);
3033                 if (pktin) {
3034                     bombout(("Unexpected data from server while waiting"
3035                              " for user host key response"));
3036                     crStop(0);
3037                 }
3038             } while (pktin || inlen > 0);
3039             s->dlgret = ssh->user_response;
3040         }
3041         ssh_set_frozen(ssh, 0);
3042
3043         if (s->dlgret == 0) {
3044             ssh_disconnect(ssh, "User aborted at host key verification",
3045                            NULL, 0, TRUE);
3046             crStop(0);
3047         }
3048     }
3049
3050     for (i = 0; i < 32; i++) {
3051         s->rsabuf[i] = ssh->session_key[i];
3052         if (i < 16)
3053             s->rsabuf[i] ^= s->session_id[i];
3054     }
3055
3056     if (hostkey.bytes > servkey.bytes) {
3057         ret = rsaencrypt(s->rsabuf, 32, &servkey);
3058         if (ret)
3059             ret = rsaencrypt(s->rsabuf, servkey.bytes, &hostkey);
3060     } else {
3061         ret = rsaencrypt(s->rsabuf, 32, &hostkey);
3062         if (ret)
3063             ret = rsaencrypt(s->rsabuf, hostkey.bytes, &servkey);
3064     }
3065     if (!ret) {
3066         bombout(("SSH-1 public key encryptions failed due to bad formatting"));
3067         crStop(0);      
3068     }
3069
3070     logevent("Encrypted session key");
3071
3072     {
3073         int cipher_chosen = 0, warn = 0;
3074         char *cipher_string = NULL;
3075         int i;
3076         for (i = 0; !cipher_chosen && i < CIPHER_MAX; i++) {
3077             int next_cipher = ssh->cfg.ssh_cipherlist[i];
3078             if (next_cipher == CIPHER_WARN) {
3079                 /* If/when we choose a cipher, warn about it */
3080                 warn = 1;
3081             } else if (next_cipher == CIPHER_AES) {
3082                 /* XXX Probably don't need to mention this. */
3083                 logevent("AES not supported in SSH-1, skipping");
3084             } else {
3085                 switch (next_cipher) {
3086                   case CIPHER_3DES:     s->cipher_type = SSH_CIPHER_3DES;
3087                                         cipher_string = "3DES"; break;
3088                   case CIPHER_BLOWFISH: s->cipher_type = SSH_CIPHER_BLOWFISH;
3089                                         cipher_string = "Blowfish"; break;
3090                   case CIPHER_DES:      s->cipher_type = SSH_CIPHER_DES;
3091                                         cipher_string = "single-DES"; break;
3092                 }
3093                 if (s->supported_ciphers_mask & (1 << s->cipher_type))
3094                     cipher_chosen = 1;
3095             }
3096         }
3097         if (!cipher_chosen) {
3098             if ((s->supported_ciphers_mask & (1 << SSH_CIPHER_3DES)) == 0)
3099                 bombout(("Server violates SSH-1 protocol by not "
3100                          "supporting 3DES encryption"));
3101             else
3102                 /* shouldn't happen */
3103                 bombout(("No supported ciphers found"));
3104             crStop(0);
3105         }
3106
3107         /* Warn about chosen cipher if necessary. */
3108         if (warn) {
3109             ssh_set_frozen(ssh, 1);
3110             s->dlgret = askalg(ssh->frontend, "cipher", cipher_string,
3111                                ssh_dialog_callback, ssh);
3112             if (s->dlgret < 0) {
3113                 do {
3114                     crReturn(0);
3115                     if (pktin) {
3116                         bombout(("Unexpected data from server while waiting"
3117                                  " for user response"));
3118                         crStop(0);
3119                     }
3120                 } while (pktin || inlen > 0);
3121                 s->dlgret = ssh->user_response;
3122             }
3123             ssh_set_frozen(ssh, 0);
3124             if (s->dlgret == 0) {
3125                 ssh_disconnect(ssh, "User aborted at cipher warning", NULL,
3126                                0, TRUE);
3127                 crStop(0);
3128             }
3129         }
3130     }
3131
3132     switch (s->cipher_type) {
3133       case SSH_CIPHER_3DES:
3134         logevent("Using 3DES encryption");
3135         break;
3136       case SSH_CIPHER_DES:
3137         logevent("Using single-DES encryption");
3138         break;
3139       case SSH_CIPHER_BLOWFISH:
3140         logevent("Using Blowfish encryption");
3141         break;
3142     }
3143
3144     send_packet(ssh, SSH1_CMSG_SESSION_KEY,
3145                 PKT_CHAR, s->cipher_type,
3146                 PKT_DATA, cookie, 8,
3147                 PKT_CHAR, (s->len * 8) >> 8, PKT_CHAR, (s->len * 8) & 0xFF,
3148                 PKT_DATA, s->rsabuf, s->len,
3149                 PKT_INT, ssh->v1_local_protoflags, PKT_END);
3150
3151     logevent("Trying to enable encryption...");
3152
3153     sfree(s->rsabuf);
3154
3155     ssh->cipher = (s->cipher_type == SSH_CIPHER_BLOWFISH ? &ssh_blowfish_ssh1 :
3156                    s->cipher_type == SSH_CIPHER_DES ? &ssh_des :
3157                    &ssh_3des);
3158     ssh->v1_cipher_ctx = ssh->cipher->make_context();
3159     ssh->cipher->sesskey(ssh->v1_cipher_ctx, ssh->session_key);
3160     logeventf(ssh, "Initialised %s encryption", ssh->cipher->text_name);
3161
3162     ssh->crcda_ctx = crcda_make_context();
3163     logevent("Installing CRC compensation attack detector");
3164
3165     if (servkey.modulus) {
3166         sfree(servkey.modulus);
3167         servkey.modulus = NULL;
3168     }
3169     if (servkey.exponent) {
3170         sfree(servkey.exponent);
3171         servkey.exponent = NULL;
3172     }
3173     if (hostkey.modulus) {
3174         sfree(hostkey.modulus);
3175         hostkey.modulus = NULL;
3176     }
3177     if (hostkey.exponent) {
3178         sfree(hostkey.exponent);
3179         hostkey.exponent = NULL;
3180     }
3181     crWaitUntil(pktin);
3182
3183     if (pktin->type != SSH1_SMSG_SUCCESS) {
3184         bombout(("Encryption not successfully enabled"));
3185         crStop(0);
3186     }
3187
3188     logevent("Successfully started encryption");
3189
3190     fflush(stdout);
3191     {
3192         if (!*ssh->cfg.username) {
3193             if (ssh_get_line && !ssh_getline_pw_only) {
3194                 if (!ssh_get_line("login as: ",
3195                                   s->username, sizeof(s->username), FALSE)) {
3196                     /*
3197                      * get_line failed to get a username.
3198                      * Terminate.
3199                      */
3200                     ssh_disconnect(ssh, "No username provided", NULL, 0, TRUE);
3201                     crStop(1);
3202                 }
3203             } else {
3204                 int ret;               /* need not be kept over crReturn */
3205                 c_write_str(ssh, "login as: ");
3206                 ssh->send_ok = 1;
3207
3208                 setup_userpass_input(ssh, s->username, sizeof(s->username), 1);
3209                 do {
3210                     crWaitUntil(!pktin);
3211                     ret = process_userpass_input(ssh, in, inlen);
3212                 } while (ret == 0);
3213                 if (ret < 0)
3214                     cleanup_exit(0);
3215                 c_write_str(ssh, "\r\n");
3216             }
3217         } else {
3218             strncpy(s->username, ssh->cfg.username, sizeof(s->username));
3219             s->username[sizeof(s->username)-1] = '\0';
3220         }
3221
3222         send_packet(ssh, SSH1_CMSG_USER, PKT_STR, s->username, PKT_END);
3223         {
3224             char userlog[22 + sizeof(s->username)];
3225             sprintf(userlog, "Sent username \"%s\"", s->username);
3226             logevent(userlog);
3227             if (flags & FLAG_INTERACTIVE &&
3228                 (!((flags & FLAG_STDERR) && (flags & FLAG_VERBOSE)))) {
3229                 strcat(userlog, "\r\n");
3230                 c_write_str(ssh, userlog);
3231             }
3232         }
3233     }
3234
3235     crWaitUntil(pktin);
3236
3237     if ((ssh->remote_bugs & BUG_CHOKES_ON_RSA)) {
3238         /* We must not attempt PK auth. Pretend we've already tried it. */
3239         s->tried_publickey = s->tried_agent = 1;
3240     } else {
3241         s->tried_publickey = s->tried_agent = 0;
3242     }
3243     s->tis_auth_refused = s->ccard_auth_refused = 0;
3244     /* Load the public half of ssh->cfg.keyfile so we notice if it's in Pageant */
3245     if (!filename_is_null(ssh->cfg.keyfile)) {
3246         if (!rsakey_pubblob(&ssh->cfg.keyfile,
3247                             &s->publickey_blob, &s->publickey_bloblen, NULL))
3248             s->publickey_blob = NULL;
3249     } else
3250         s->publickey_blob = NULL;
3251
3252     while (pktin->type == SSH1_SMSG_FAILURE) {
3253         s->pwpkt_type = SSH1_CMSG_AUTH_PASSWORD;
3254
3255         if (agent_exists() && !s->tried_agent) {
3256             /*
3257              * Attempt RSA authentication using Pageant.
3258              */
3259             void *r;
3260
3261             s->authed = FALSE;
3262             s->tried_agent = 1;
3263             logevent("Pageant is running. Requesting keys.");
3264
3265             /* Request the keys held by the agent. */
3266             PUT_32BIT(s->request, 1);
3267             s->request[4] = SSH1_AGENTC_REQUEST_RSA_IDENTITIES;
3268             if (!agent_query(s->request, 5, &r, &s->responselen,
3269                              ssh_agent_callback, ssh)) {
3270                 do {
3271                     crReturn(0);
3272                     if (pktin) {
3273                         bombout(("Unexpected data from server while waiting"
3274                                  " for agent response"));
3275                         crStop(0);
3276                     }
3277                 } while (pktin || inlen > 0);
3278                 r = ssh->agent_response;
3279                 s->responselen = ssh->agent_response_len;
3280             }
3281             s->response = (unsigned char *) r;
3282             if (s->response && s->responselen >= 5 &&
3283                 s->response[4] == SSH1_AGENT_RSA_IDENTITIES_ANSWER) {
3284                 s->p = s->response + 5;
3285                 s->nkeys = GET_32BIT(s->p);
3286                 s->p += 4;
3287                 logeventf(ssh, "Pageant has %d SSH-1 keys", s->nkeys);
3288                 for (s->keyi = 0; s->keyi < s->nkeys; s->keyi++) {
3289                     logeventf(ssh, "Trying Pageant key #%d", s->keyi);
3290                     if (s->publickey_blob &&
3291                         !memcmp(s->p, s->publickey_blob,
3292                                 s->publickey_bloblen)) {
3293                         logevent("This key matches configured key file");
3294                         s->tried_publickey = 1;
3295                     }
3296                     s->p += 4;
3297                     {
3298                         int n, ok = FALSE;
3299                         do {           /* do while (0) to make breaking easy */
3300                             n = ssh1_read_bignum
3301                                 (s->p, s->responselen-(s->p-s->response),
3302                                  &s->key.exponent);
3303                             if (n < 0)
3304                                 break;
3305                             s->p += n;
3306                             n = ssh1_read_bignum
3307                                 (s->p, s->responselen-(s->p-s->response),
3308                                  &s->key.modulus);
3309                             if (n < 0)
3310                             break;
3311                             s->p += n;
3312                             if (s->responselen - (s->p-s->response) < 4)
3313                                 break;
3314                             s->commentlen = GET_32BIT(s->p);
3315                             s->p += 4;
3316                             if (s->responselen - (s->p-s->response) <
3317                                 s->commentlen)
3318                                 break;
3319                             s->commentp = (char *)s->p;
3320                             s->p += s->commentlen;
3321                             ok = TRUE;
3322                         } while (0);
3323                         if (!ok) {
3324                             logevent("Pageant key list packet was truncated");
3325                             break;
3326                         }
3327                     }
3328                     send_packet(ssh, SSH1_CMSG_AUTH_RSA,
3329                                 PKT_BIGNUM, s->key.modulus, PKT_END);
3330                     crWaitUntil(pktin);
3331                     if (pktin->type != SSH1_SMSG_AUTH_RSA_CHALLENGE) {
3332                         logevent("Key refused");
3333                         continue;
3334                     }
3335                     logevent("Received RSA challenge");
3336                     if ((s->challenge = ssh1_pkt_getmp(pktin)) == NULL) {
3337                         bombout(("Server's RSA challenge was badly formatted"));
3338                         crStop(0);
3339                     }
3340
3341                     {
3342                         char *agentreq, *q, *ret;
3343                         void *vret;
3344                         int len, retlen;
3345                         len = 1 + 4;   /* message type, bit count */
3346                         len += ssh1_bignum_length(s->key.exponent);
3347                         len += ssh1_bignum_length(s->key.modulus);
3348                         len += ssh1_bignum_length(s->challenge);
3349                         len += 16;     /* session id */
3350                         len += 4;      /* response format */
3351                         agentreq = snewn(4 + len, char);
3352                         PUT_32BIT(agentreq, len);
3353                         q = agentreq + 4;
3354                         *q++ = SSH1_AGENTC_RSA_CHALLENGE;
3355                         PUT_32BIT(q, bignum_bitcount(s->key.modulus));
3356                         q += 4;
3357                         q += ssh1_write_bignum(q, s->key.exponent);
3358                         q += ssh1_write_bignum(q, s->key.modulus);
3359                         q += ssh1_write_bignum(q, s->challenge);
3360                         memcpy(q, s->session_id, 16);
3361                         q += 16;
3362                         PUT_32BIT(q, 1);        /* response format */
3363                         if (!agent_query(agentreq, len + 4, &vret, &retlen,
3364                                          ssh_agent_callback, ssh)) {
3365                             sfree(agentreq);
3366                             do {
3367                                 crReturn(0);
3368                                 if (pktin) {
3369                                     bombout(("Unexpected data from server"
3370                                              " while waiting for agent"
3371                                              " response"));
3372                                     crStop(0);
3373                                 }
3374                             } while (pktin || inlen > 0);
3375                             vret = ssh->agent_response;
3376                             retlen = ssh->agent_response_len;
3377                         } else
3378                             sfree(agentreq);
3379                         ret = vret;
3380                         if (ret) {
3381                             if (ret[4] == SSH1_AGENT_RSA_RESPONSE) {
3382                                 logevent("Sending Pageant's response");
3383                                 send_packet(ssh, SSH1_CMSG_AUTH_RSA_RESPONSE,
3384                                             PKT_DATA, ret + 5, 16,
3385                                             PKT_END);
3386                                 sfree(ret);
3387                                 crWaitUntil(pktin);
3388                                 if (pktin->type == SSH1_SMSG_SUCCESS) {
3389                                     logevent
3390                                         ("Pageant's response accepted");
3391                                     if (flags & FLAG_VERBOSE) {
3392                                         c_write_str(ssh, "Authenticated using"
3393                                                     " RSA key \"");
3394                                         c_write(ssh, s->commentp,
3395                                                 s->commentlen);
3396                                         c_write_str(ssh, "\" from agent\r\n");
3397                                     }
3398                                     s->authed = TRUE;
3399                                 } else
3400                                     logevent
3401                                         ("Pageant's response not accepted");
3402                             } else {
3403                                 logevent
3404                                     ("Pageant failed to answer challenge");
3405                                 sfree(ret);
3406                             }
3407                         } else {
3408                             logevent("No reply received from Pageant");
3409                         }
3410                     }
3411                     freebn(s->key.exponent);
3412                     freebn(s->key.modulus);
3413                     freebn(s->challenge);
3414                     if (s->authed)
3415                         break;
3416                 }
3417                 sfree(s->response);
3418             }
3419             if (s->authed)
3420                 break;
3421         }
3422         if (!filename_is_null(ssh->cfg.keyfile) && !s->tried_publickey)
3423             s->pwpkt_type = SSH1_CMSG_AUTH_RSA;
3424
3425         if (ssh->cfg.try_tis_auth &&
3426             (s->supported_auths_mask & (1 << SSH1_AUTH_TIS)) &&
3427             !s->tis_auth_refused) {
3428             s->pwpkt_type = SSH1_CMSG_AUTH_TIS_RESPONSE;
3429             logevent("Requested TIS authentication");
3430             send_packet(ssh, SSH1_CMSG_AUTH_TIS, PKT_END);
3431             crWaitUntil(pktin);
3432             if (pktin->type != SSH1_SMSG_AUTH_TIS_CHALLENGE) {
3433                 logevent("TIS authentication declined");
3434                 if (flags & FLAG_INTERACTIVE)
3435                     c_write_str(ssh, "TIS authentication refused.\r\n");
3436                 s->tis_auth_refused = 1;
3437                 continue;
3438             } else {
3439                 char *challenge;
3440                 int challengelen;
3441
3442                 ssh_pkt_getstring(pktin, &challenge, &challengelen);
3443                 if (!challenge) {
3444                     bombout(("TIS challenge packet was badly formed"));
3445                     crStop(0);
3446                 }
3447                 c_write_str(ssh, "Using TIS authentication.\r\n");
3448                 logevent("Received TIS challenge");
3449                 if (challengelen > sizeof(s->prompt) - 1)
3450                     challengelen = sizeof(s->prompt) - 1;/* prevent overrun */
3451                 memcpy(s->prompt, challenge, challengelen);
3452                 /* Prompt heuristic comes from OpenSSH */
3453                 strncpy(s->prompt + challengelen,
3454                         memchr(s->prompt, '\n', challengelen) ?
3455                         "": "\r\nResponse: ",
3456                         (sizeof s->prompt) - challengelen);
3457                 s->prompt[(sizeof s->prompt) - 1] = '\0';
3458             }
3459         }
3460         if (ssh->cfg.try_tis_auth &&
3461             (s->supported_auths_mask & (1 << SSH1_AUTH_CCARD)) &&
3462             !s->ccard_auth_refused) {
3463             s->pwpkt_type = SSH1_CMSG_AUTH_CCARD_RESPONSE;
3464             logevent("Requested CryptoCard authentication");
3465             send_packet(ssh, SSH1_CMSG_AUTH_CCARD, PKT_END);
3466             crWaitUntil(pktin);
3467             if (pktin->type != SSH1_SMSG_AUTH_CCARD_CHALLENGE) {
3468                 logevent("CryptoCard authentication declined");
3469                 c_write_str(ssh, "CryptoCard authentication refused.\r\n");
3470                 s->ccard_auth_refused = 1;
3471                 continue;
3472             } else {
3473                 char *challenge;
3474                 int challengelen;
3475
3476                 ssh_pkt_getstring(pktin, &challenge, &challengelen);
3477                 if (!challenge) {
3478                     bombout(("CryptoCard challenge packet was badly formed"));
3479                     crStop(0);
3480                 }
3481                 c_write_str(ssh, "Using CryptoCard authentication.\r\n");
3482                 logevent("Received CryptoCard challenge");
3483                 if (challengelen > sizeof(s->prompt) - 1)
3484                     challengelen = sizeof(s->prompt) - 1;/* prevent overrun */
3485                 memcpy(s->prompt, challenge, challengelen);
3486                 strncpy(s->prompt + challengelen,
3487                         memchr(s->prompt, '\n', challengelen) ?
3488                         "" : "\r\nResponse: ",
3489                         sizeof(s->prompt) - challengelen);
3490                 s->prompt[sizeof(s->prompt) - 1] = '\0';
3491             }
3492         }
3493         if (s->pwpkt_type == SSH1_CMSG_AUTH_PASSWORD) {
3494             sprintf(s->prompt, "%.90s@%.90s's password: ",
3495                     s->username, ssh->savedhost);
3496         }
3497         if (s->pwpkt_type == SSH1_CMSG_AUTH_RSA) {
3498             char *comment = NULL;
3499             int type;
3500             if (flags & FLAG_VERBOSE)
3501                 c_write_str(ssh, "Trying public key authentication.\r\n");
3502             logeventf(ssh, "Trying public key \"%s\"",
3503                       filename_to_str(&ssh->cfg.keyfile));
3504             type = key_type(&ssh->cfg.keyfile);
3505             if (type != SSH_KEYTYPE_SSH1) {
3506                 char *msg = dupprintf("Key is of wrong type (%s)",
3507                                       key_type_to_str(type));
3508                 logevent(msg);
3509                 c_write_str(ssh, msg);
3510                 c_write_str(ssh, "\r\n");
3511                 sfree(msg);
3512                 s->tried_publickey = 1;
3513                 continue;
3514             }
3515             if (!rsakey_encrypted(&ssh->cfg.keyfile, &comment)) {
3516                 if (flags & FLAG_VERBOSE)
3517                     c_write_str(ssh, "No passphrase required.\r\n");
3518                 goto tryauth;
3519             }
3520             sprintf(s->prompt, "Passphrase for key \"%.100s\": ", comment);
3521             sfree(comment);
3522         }
3523
3524         /*
3525          * Show password prompt, having first obtained it via a TIS
3526          * or CryptoCard exchange if we're doing TIS or CryptoCard
3527          * authentication.
3528          */
3529         if (ssh_get_line) {
3530             if (!ssh_get_line(s->prompt, s->password,
3531                               sizeof(s->password), TRUE)) {
3532                 /*
3533                  * get_line failed to get a password (for example
3534                  * because one was supplied on the command line
3535                  * which has already failed to work). Terminate.
3536                  */
3537                 ssh_disconnect(ssh, NULL, "Unable to authenticate", 0, FALSE);
3538                 crStop(1);
3539             }
3540         } else {
3541             /* Prompt may have come from server. We've munged it a bit, so
3542              * we know it to be zero-terminated at least once. */
3543             int ret;                   /* need not be saved over crReturn */
3544             c_write_untrusted(ssh, s->prompt, strlen(s->prompt));
3545             s->pos = 0;
3546
3547             setup_userpass_input(ssh, s->password, sizeof(s->password), 0);
3548             do {
3549                 crWaitUntil(!pktin);
3550                 ret = process_userpass_input(ssh, in, inlen);
3551             } while (ret == 0);
3552             if (ret < 0)
3553                 cleanup_exit(0);
3554             c_write_str(ssh, "\r\n");
3555         }
3556
3557       tryauth:
3558         if (s->pwpkt_type == SSH1_CMSG_AUTH_RSA) {
3559             /*
3560              * Try public key authentication with the specified
3561              * key file.
3562              */
3563             s->tried_publickey = 1;
3564             
3565             {
3566                 const char *error = NULL;
3567                 int ret = loadrsakey(&ssh->cfg.keyfile, &s->key, s->password,
3568                                      &error);
3569                 if (ret == 0) {
3570                     c_write_str(ssh, "Couldn't load private key from ");
3571                     c_write_str(ssh, filename_to_str(&ssh->cfg.keyfile));
3572                     c_write_str(ssh, " (");
3573                     c_write_str(ssh, error);
3574                     c_write_str(ssh, ").\r\n");
3575                     continue;          /* go and try password */
3576                 }
3577                 if (ret == -1) {
3578                     c_write_str(ssh, "Wrong passphrase.\r\n");
3579                     s->tried_publickey = 0;
3580                     continue;          /* try again */
3581                 }
3582             }
3583
3584             /*
3585              * Send a public key attempt.
3586              */
3587             send_packet(ssh, SSH1_CMSG_AUTH_RSA,
3588                         PKT_BIGNUM, s->key.modulus, PKT_END);
3589
3590             crWaitUntil(pktin);
3591             if (pktin->type == SSH1_SMSG_FAILURE) {
3592                 c_write_str(ssh, "Server refused our public key.\r\n");
3593                 continue;              /* go and try password */
3594             }
3595             if (pktin->type != SSH1_SMSG_AUTH_RSA_CHALLENGE) {
3596                 bombout(("Bizarre response to offer of public key"));
3597                 crStop(0);
3598             }
3599
3600             {
3601                 int i;
3602                 unsigned char buffer[32];
3603                 Bignum challenge, response;
3604
3605                 if ((challenge = ssh1_pkt_getmp(pktin)) == NULL) {
3606                     bombout(("Server's RSA challenge was badly formatted"));
3607                     crStop(0);
3608                 }
3609                 response = rsadecrypt(challenge, &s->key);
3610                 freebn(s->key.private_exponent);/* burn the evidence */
3611
3612                 for (i = 0; i < 32; i++) {
3613                     buffer[i] = bignum_byte(response, 31 - i);
3614                 }
3615
3616                 MD5Init(&md5c);
3617                 MD5Update(&md5c, buffer, 32);
3618                 MD5Update(&md5c, s->session_id, 16);
3619                 MD5Final(buffer, &md5c);
3620
3621                 send_packet(ssh, SSH1_CMSG_AUTH_RSA_RESPONSE,
3622                             PKT_DATA, buffer, 16, PKT_END);
3623
3624                 freebn(challenge);
3625                 freebn(response);
3626             }
3627
3628             crWaitUntil(pktin);
3629             if (pktin->type == SSH1_SMSG_FAILURE) {
3630                 if (flags & FLAG_VERBOSE)
3631                     c_write_str(ssh, "Failed to authenticate with"
3632                                 " our public key.\r\n");
3633                 continue;              /* go and try password */
3634             } else if (pktin->type != SSH1_SMSG_SUCCESS) {
3635                 bombout(("Bizarre response to RSA authentication response"));
3636                 crStop(0);
3637             }
3638
3639             break;                     /* we're through! */
3640         } else {
3641             if (s->pwpkt_type == SSH1_CMSG_AUTH_PASSWORD) {
3642                 /*
3643                  * Defence against traffic analysis: we send a
3644                  * whole bunch of packets containing strings of
3645                  * different lengths. One of these strings is the
3646                  * password, in a SSH1_CMSG_AUTH_PASSWORD packet.
3647                  * The others are all random data in
3648                  * SSH1_MSG_IGNORE packets. This way a passive
3649                  * listener can't tell which is the password, and
3650                  * hence can't deduce the password length.
3651                  * 
3652                  * Anybody with a password length greater than 16
3653                  * bytes is going to have enough entropy in their
3654                  * password that a listener won't find it _that_
3655                  * much help to know how long it is. So what we'll
3656                  * do is:
3657                  * 
3658                  *  - if password length < 16, we send 15 packets
3659                  *    containing string lengths 1 through 15
3660                  * 
3661                  *  - otherwise, we let N be the nearest multiple
3662                  *    of 8 below the password length, and send 8
3663                  *    packets containing string lengths N through
3664                  *    N+7. This won't obscure the order of
3665                  *    magnitude of the password length, but it will
3666                  *    introduce a bit of extra uncertainty.
3667                  * 
3668                  * A few servers (the old 1.2.18 through 1.2.22)
3669                  * can't deal with SSH1_MSG_IGNORE. For these
3670                  * servers, we need an alternative defence. We make
3671                  * use of the fact that the password is interpreted
3672                  * as a C string: so we can append a NUL, then some
3673                  * random data.
3674                  * 
3675                  * One server (a Cisco one) can deal with neither
3676                  * SSH1_MSG_IGNORE _nor_ a padded password string.
3677                  * For this server we are left with no defences
3678                  * against password length sniffing.
3679                  */
3680                 if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH1_IGNORE)) {
3681                     /*
3682                      * The server can deal with SSH1_MSG_IGNORE, so
3683                      * we can use the primary defence.
3684                      */
3685                     int bottom, top, pwlen, i;
3686                     char *randomstr;
3687
3688                     pwlen = strlen(s->password);
3689                     if (pwlen < 16) {
3690                         bottom = 0;    /* zero length passwords are OK! :-) */
3691                         top = 15;
3692                     } else {
3693                         bottom = pwlen & ~7;
3694                         top = bottom + 7;
3695                     }
3696
3697                     assert(pwlen >= bottom && pwlen <= top);
3698
3699                     randomstr = snewn(top + 1, char);
3700
3701                     for (i = bottom; i <= top; i++) {
3702                         if (i == pwlen) {
3703                             defer_packet(ssh, s->pwpkt_type,
3704                                          PKTT_PASSWORD, PKT_STR, s->password,
3705                                          PKTT_OTHER, PKT_END);
3706                         } else {
3707                             for (j = 0; j < i; j++) {
3708                                 do {
3709                                     randomstr[j] = random_byte();
3710                                 } while (randomstr[j] == '\0');
3711                             }
3712                             randomstr[i] = '\0';
3713                             defer_packet(ssh, SSH1_MSG_IGNORE,
3714                                          PKT_STR, randomstr, PKT_END);
3715                         }
3716                     }
3717                     logevent("Sending password with camouflage packets");
3718                     ssh_pkt_defersend(ssh);
3719                     sfree(randomstr);
3720                 } 
3721                 else if (!(ssh->remote_bugs & BUG_NEEDS_SSH1_PLAIN_PASSWORD)) {
3722                     /*
3723                      * The server can't deal with SSH1_MSG_IGNORE
3724                      * but can deal with padded passwords, so we
3725                      * can use the secondary defence.
3726                      */
3727                     char string[64];
3728                     char *ss;
3729                     int len;
3730
3731                     len = strlen(s->password);
3732                     if (len < sizeof(string)) {
3733                         ss = string;
3734                         strcpy(string, s->password);
3735                         len++;         /* cover the zero byte */
3736                         while (len < sizeof(string)) {
3737                             string[len++] = (char) random_byte();
3738                         }
3739                     } else {
3740                         ss = s->password;
3741                     }
3742                     logevent("Sending length-padded password");
3743                     send_packet(ssh, s->pwpkt_type, PKTT_PASSWORD,
3744                                 PKT_INT, len, PKT_DATA, ss, len,
3745                                 PKTT_OTHER, PKT_END);
3746                 } else {
3747                     /*
3748                      * The server has _both_
3749                      * BUG_CHOKES_ON_SSH1_IGNORE and
3750                      * BUG_NEEDS_SSH1_PLAIN_PASSWORD. There is
3751                      * therefore nothing we can do.
3752                      */
3753                     int len;
3754                     len = strlen(s->password);
3755                     logevent("Sending unpadded password");
3756                     send_packet(ssh, s->pwpkt_type,
3757                                 PKTT_PASSWORD, PKT_INT, len,
3758                                 PKT_DATA, s->password, len,
3759                                 PKTT_OTHER, PKT_END);
3760                 }
3761             } else {
3762                 send_packet(ssh, s->pwpkt_type, PKTT_PASSWORD,
3763                             PKT_STR, s->password, PKTT_OTHER, PKT_END);
3764             }
3765         }
3766         logevent("Sent password");
3767         memset(s->password, 0, strlen(s->password));
3768         crWaitUntil(pktin);
3769         if (pktin->type == SSH1_SMSG_FAILURE) {
3770             if (flags & FLAG_VERBOSE)
3771                 c_write_str(ssh, "Access denied\r\n");
3772             logevent("Authentication refused");
3773         } else if (pktin->type != SSH1_SMSG_SUCCESS) {
3774             bombout(("Strange packet received, type %d", pktin->type));
3775             crStop(0);
3776         }
3777     }
3778
3779     logevent("Authentication successful");
3780
3781     crFinish(1);
3782 }
3783
3784 void sshfwd_close(struct ssh_channel *c)
3785 {
3786     Ssh ssh = c->ssh;
3787
3788     if (ssh->state == SSH_STATE_CLOSED)
3789         return;
3790
3791     if (c && !c->closes) {
3792         /*
3793          * If halfopen is true, we have sent
3794          * CHANNEL_OPEN for this channel, but it hasn't even been
3795          * acknowledged by the server. So we must set a close flag
3796          * on it now, and then when the server acks the channel
3797          * open, we can close it then.
3798          */
3799         if (!c->halfopen) {
3800             if (ssh->version == 1) {
3801                 send_packet(ssh, SSH1_MSG_CHANNEL_CLOSE, PKT_INT, c->remoteid,
3802                             PKT_END);
3803             } else {
3804                 struct Packet *pktout;
3805                 pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_CLOSE);
3806                 ssh2_pkt_adduint32(pktout, c->remoteid);
3807                 ssh2_pkt_send(ssh, pktout);
3808             }
3809         }
3810         c->closes = 1;                 /* sent MSG_CLOSE */
3811         if (c->type == CHAN_X11) {
3812             c->u.x11.s = NULL;
3813             logevent("Forwarded X11 connection terminated");
3814         } else if (c->type == CHAN_SOCKDATA ||
3815                    c->type == CHAN_SOCKDATA_DORMANT) {
3816             c->u.pfd.s = NULL;
3817             logevent("Forwarded port closed");
3818         }
3819     }
3820 }
3821
3822 int sshfwd_write(struct ssh_channel *c, char *buf, int len)
3823 {
3824     Ssh ssh = c->ssh;
3825
3826     if (ssh->state == SSH_STATE_CLOSED)
3827         return 0;
3828
3829     if (ssh->version == 1) {
3830         send_packet(ssh, SSH1_MSG_CHANNEL_DATA,
3831                     PKT_INT, c->remoteid,
3832                     PKTT_DATA,
3833                     PKT_INT, len, PKT_DATA, buf, len,
3834                     PKTT_OTHER, PKT_END);
3835         /*
3836          * In SSH-1 we can return 0 here - implying that forwarded
3837          * connections are never individually throttled - because
3838          * the only circumstance that can cause throttling will be
3839          * the whole SSH connection backing up, in which case
3840          * _everything_ will be throttled as a whole.
3841          */
3842         return 0;
3843     } else {
3844         ssh2_add_channel_data(c, buf, len);
3845         return ssh2_try_send(c);
3846     }
3847 }
3848
3849 void sshfwd_unthrottle(struct ssh_channel *c, int bufsize)
3850 {
3851     Ssh ssh = c->ssh;
3852
3853     if (ssh->state == SSH_STATE_CLOSED)
3854         return;
3855
3856     if (ssh->version == 1) {
3857         if (c->v.v1.throttling && bufsize < SSH1_BUFFER_LIMIT) {
3858             c->v.v1.throttling = 0;
3859             ssh1_throttle(ssh, -1);
3860         }
3861     } else {
3862         ssh2_set_window(c, OUR_V2_WINSIZE - bufsize);
3863     }
3864 }
3865
3866 static void ssh_queueing_handler(Ssh ssh, struct Packet *pktin)
3867 {
3868     struct queued_handler *qh = ssh->qhead;
3869
3870     assert(qh != NULL);
3871
3872     assert(pktin->type == qh->msg1 || pktin->type == qh->msg2);
3873
3874     if (qh->msg1 > 0) {
3875         assert(ssh->packet_dispatch[qh->msg1] == ssh_queueing_handler);
3876         ssh->packet_dispatch[qh->msg1] = NULL;
3877     }
3878     if (qh->msg2 > 0) {
3879         assert(ssh->packet_dispatch[qh->msg2] == ssh_queueing_handler);
3880         ssh->packet_dispatch[qh->msg2] = NULL;
3881     }
3882
3883     if (qh->next) {
3884         ssh->qhead = qh->next;
3885
3886         if (ssh->qhead->msg1 > 0) {
3887             assert(ssh->packet_dispatch[ssh->qhead->msg1] == NULL);
3888             ssh->packet_dispatch[ssh->qhead->msg1] = ssh_queueing_handler;
3889         }
3890         if (ssh->qhead->msg2 > 0) {
3891             assert(ssh->packet_dispatch[ssh->qhead->msg2] == NULL);
3892             ssh->packet_dispatch[ssh->qhead->msg2] = ssh_queueing_handler;
3893         }
3894     } else {
3895         ssh->qhead = ssh->qtail = NULL;
3896         ssh->packet_dispatch[pktin->type] = NULL;
3897     }
3898
3899     qh->handler(ssh, pktin, qh->ctx);
3900
3901     sfree(qh);
3902 }
3903
3904 static void ssh_queue_handler(Ssh ssh, int msg1, int msg2,
3905                               chandler_fn_t handler, void *ctx)
3906 {
3907     struct queued_handler *qh;
3908
3909     qh = snew(struct queued_handler);
3910     qh->msg1 = msg1;
3911     qh->msg2 = msg2;
3912     qh->handler = handler;
3913     qh->ctx = ctx;
3914     qh->next = NULL;
3915
3916     if (ssh->qtail == NULL) {
3917         ssh->qhead = qh;
3918
3919         if (qh->msg1 > 0) {
3920             assert(ssh->packet_dispatch[qh->msg1] == NULL);
3921             ssh->packet_dispatch[qh->msg1] = ssh_queueing_handler;
3922         }
3923         if (qh->msg2 > 0) {
3924             assert(ssh->packet_dispatch[qh->msg2] == NULL);
3925             ssh->packet_dispatch[qh->msg2] = ssh_queueing_handler;
3926         }
3927     } else {
3928         ssh->qtail->next = qh;
3929     }
3930     ssh->qtail = qh;
3931 }
3932
3933 static void ssh_rportfwd_succfail(Ssh ssh, struct Packet *pktin, void *ctx)
3934 {
3935     struct ssh_rportfwd *rpf, *pf = (struct ssh_rportfwd *)ctx;
3936
3937     if (pktin->type == (ssh->version == 1 ? SSH1_SMSG_SUCCESS :
3938                         SSH2_MSG_REQUEST_SUCCESS)) {
3939         logeventf(ssh, "Remote port forwarding from %s enabled",
3940                   pf->sportdesc);
3941     } else {
3942         logeventf(ssh, "Remote port forwarding from %s refused",
3943                   pf->sportdesc);
3944
3945         rpf = del234(ssh->rportfwds, pf);
3946         assert(rpf == pf);
3947         free_rportfwd(pf);
3948     }
3949 }
3950
3951 static void ssh_setup_portfwd(Ssh ssh, const Config *cfg)
3952 {
3953     const char *portfwd_strptr = cfg->portfwd;
3954     struct ssh_portfwd *epf;
3955     int i;
3956
3957     if (!ssh->portfwds) {
3958         ssh->portfwds = newtree234(ssh_portcmp);
3959     } else {
3960         /*
3961          * Go through the existing port forwardings and tag them
3962          * with status==DESTROY. Any that we want to keep will be
3963          * re-enabled (status==KEEP) as we go through the
3964          * configuration and find out which bits are the same as
3965          * they were before.
3966          */
3967         struct ssh_portfwd *epf;
3968         int i;
3969         for (i = 0; (epf = index234(ssh->portfwds, i)) != NULL; i++)
3970             epf->status = DESTROY;
3971     }
3972
3973     while (*portfwd_strptr) {
3974         char address_family, type;
3975         int sport,dport,sserv,dserv;
3976         char sports[256], dports[256], saddr[256], host[256];
3977         int n;
3978
3979         address_family = 'A';
3980         type = 'L';
3981         if (*portfwd_strptr == 'A' ||
3982             *portfwd_strptr == '4' ||
3983             *portfwd_strptr == '6')
3984             address_family = *portfwd_strptr++;
3985         if (*portfwd_strptr == 'L' ||
3986             *portfwd_strptr == 'R' ||
3987             *portfwd_strptr == 'D')
3988             type = *portfwd_strptr++;
3989
3990         saddr[0] = '\0';
3991
3992         n = 0;
3993         while (*portfwd_strptr && *portfwd_strptr != '\t') {
3994             if (*portfwd_strptr == ':') {
3995                 /*
3996                  * We've seen a colon in the middle of the
3997                  * source port number. This means that
3998                  * everything we've seen until now is the
3999                  * source _address_, so we'll move it into
4000                  * saddr and start sports from the beginning
4001                  * again.
4002                  */
4003                 portfwd_strptr++;
4004                 sports[n] = '\0';
4005                 if (ssh->version == 1 && type == 'R') {
4006                     logeventf(ssh, "SSH-1 cannot handle remote source address "
4007                               "spec \"%s\"; ignoring", sports);
4008                 } else
4009                     strcpy(saddr, sports);
4010                 n = 0;
4011             }
4012             if (n < lenof(sports)-1) sports[n++] = *portfwd_strptr++;
4013         }
4014         sports[n] = 0;
4015         if (type != 'D') {
4016             if (*portfwd_strptr == '\t')
4017                 portfwd_strptr++;
4018             n = 0;
4019             while (*portfwd_strptr && *portfwd_strptr != ':') {
4020                 if (n < lenof(host)-1) host[n++] = *portfwd_strptr++;
4021             }
4022             host[n] = 0;
4023             if (*portfwd_strptr == ':')
4024                 portfwd_strptr++;
4025             n = 0;
4026             while (*portfwd_strptr) {
4027                 if (n < lenof(dports)-1) dports[n++] = *portfwd_strptr++;
4028             }
4029             dports[n] = 0;
4030             portfwd_strptr++;
4031             dport = atoi(dports);
4032             dserv = 0;
4033             if (dport == 0) {
4034                 dserv = 1;
4035                 dport = net_service_lookup(dports);
4036                 if (!dport) {
4037                     logeventf(ssh, "Service lookup failed for destination"
4038                               " port \"%s\"", dports);
4039                 }
4040             }
4041         } else {
4042             while (*portfwd_strptr) portfwd_strptr++;
4043             host[0] = 0;
4044             dports[0] = 0;
4045             dport = dserv = -1;
4046             portfwd_strptr++;          /* eat the NUL and move to next one */
4047         }
4048         sport = atoi(sports);
4049         sserv = 0;
4050         if (sport == 0) {
4051             sserv = 1;
4052             sport = net_service_lookup(sports);
4053             if (!sport) {
4054                 logeventf(ssh, "Service lookup failed for source"
4055                           " port \"%s\"", sports);
4056             }
4057         }
4058         if (sport && dport) {
4059             /* Set up a description of the source port. */
4060             struct ssh_portfwd *pfrec, *epfrec;
4061
4062             pfrec = snew(struct ssh_portfwd);
4063             pfrec->type = type;
4064             pfrec->saddr = *saddr ? dupstr(saddr) : NULL;
4065             pfrec->sserv = sserv ? dupstr(sports) : NULL;
4066             pfrec->sport = sport;
4067             pfrec->daddr = *host ? dupstr(host) : NULL;
4068             pfrec->dserv = dserv ? dupstr(dports) : NULL;
4069             pfrec->dport = dport;
4070             pfrec->local = NULL;
4071             pfrec->remote = NULL;
4072             pfrec->addressfamily = (address_family == '4' ? ADDRTYPE_IPV4 :
4073                                     address_family == '6' ? ADDRTYPE_IPV6 :
4074                                     ADDRTYPE_UNSPEC);
4075
4076             epfrec = add234(ssh->portfwds, pfrec);
4077             if (epfrec != pfrec) {
4078                 /*
4079                  * We already have a port forwarding with precisely
4080                  * these parameters. Hence, no need to do anything;
4081                  * simply tag the existing one as KEEP.
4082                  */
4083                 epfrec->status = KEEP;
4084                 free_portfwd(pfrec);
4085             } else {
4086                 pfrec->status = CREATE;
4087             }
4088         }
4089     }
4090
4091     /*
4092      * Now go through and destroy any port forwardings which were
4093      * not re-enabled.
4094      */
4095     for (i = 0; (epf = index234(ssh->portfwds, i)) != NULL; i++)
4096         if (epf->status == DESTROY) {
4097             char *message;
4098
4099             message = dupprintf("%s port forwarding from %s%s%d",
4100                                 epf->type == 'L' ? "local" :
4101                                 epf->type == 'R' ? "remote" : "dynamic",
4102                                 epf->saddr ? epf->saddr : "",
4103                                 epf->saddr ? ":" : "",
4104                                 epf->sport);
4105
4106             if (epf->type != 'D') {
4107                 char *msg2 = dupprintf("%s to %s:%d", message,
4108                                        epf->daddr, epf->dport);
4109                 sfree(message);
4110                 message = msg2;
4111             }
4112
4113             logeventf(ssh, "Cancelling %s", message);
4114             sfree(message);
4115
4116             if (epf->remote) {
4117                 struct ssh_rportfwd *rpf = epf->remote;
4118                 struct Packet *pktout;
4119
4120                 /*
4121                  * Cancel the port forwarding at the server
4122                  * end.
4123                  */
4124                 if (ssh->version == 1) {
4125                     /*
4126                      * We cannot cancel listening ports on the
4127                      * server side in SSH-1! There's no message
4128                      * to support it. Instead, we simply remove
4129                      * the rportfwd record from the local end
4130                      * so that any connections the server tries
4131                      * to make on it are rejected.
4132                      */
4133                 } else {
4134                     pktout = ssh2_pkt_init(SSH2_MSG_GLOBAL_REQUEST);
4135                     ssh2_pkt_addstring(pktout, "cancel-tcpip-forward");
4136                     ssh2_pkt_addbool(pktout, 0);/* _don't_ want reply */
4137                     if (epf->saddr) {
4138                         ssh2_pkt_addstring(pktout, epf->saddr);
4139                     } else if (ssh->cfg.rport_acceptall) {
4140                         /* XXX: ssh->cfg.rport_acceptall may not represent
4141                          * what was used to open the original connection,
4142                          * since it's reconfigurable. */
4143                         ssh2_pkt_addstring(pktout, "0.0.0.0");
4144                     } else {
4145                         ssh2_pkt_addstring(pktout, "127.0.0.1");
4146                     }
4147                     ssh2_pkt_adduint32(pktout, epf->sport);
4148                     ssh2_pkt_send(ssh, pktout);
4149                 }
4150
4151                 del234(ssh->rportfwds, rpf);
4152                 free_rportfwd(rpf);
4153             } else if (epf->local) {
4154                 pfd_terminate(epf->local);
4155             }
4156
4157             delpos234(ssh->portfwds, i);
4158             free_portfwd(epf);
4159             i--;                       /* so we don't skip one in the list */
4160         }
4161
4162     /*
4163      * And finally, set up any new port forwardings (status==CREATE).
4164      */
4165     for (i = 0; (epf = index234(ssh->portfwds, i)) != NULL; i++)
4166         if (epf->status == CREATE) {
4167             char *sportdesc, *dportdesc;
4168             sportdesc = dupprintf("%s%s%s%s%d%s",
4169                                   epf->saddr ? epf->saddr : "",
4170                                   epf->saddr ? ":" : "",
4171                                   epf->sserv ? epf->sserv : "",
4172                                   epf->sserv ? "(" : "",
4173                                   epf->sport,
4174                                   epf->sserv ? ")" : "");
4175             if (epf->type == 'D') {
4176                 dportdesc = NULL;
4177             } else {
4178                 dportdesc = dupprintf("%s:%s%s%d%s",
4179                                       epf->daddr,
4180                                       epf->dserv ? epf->dserv : "",
4181                                       epf->dserv ? "(" : "",
4182                                       epf->dport,
4183                                       epf->dserv ? ")" : "");
4184             }
4185
4186             if (epf->type == 'L') {
4187                 const char *err = pfd_addforward(epf->daddr, epf->dport,
4188                                                  epf->saddr, epf->sport,
4189                                                  ssh, cfg,
4190                                                  &epf->local,
4191                                                  epf->addressfamily);
4192
4193                 logeventf(ssh, "Local %sport %s forwarding to %s%s%s",
4194                           epf->addressfamily == ADDRTYPE_IPV4 ? "IPv4 " :
4195                           epf->addressfamily == ADDRTYPE_IPV6 ? "IPv6 " : "",
4196                           sportdesc, dportdesc,
4197                           err ? " failed: " : "", err ? err : "");
4198             } else if (epf->type == 'D') {
4199                 const char *err = pfd_addforward(NULL, -1,
4200                                                  epf->saddr, epf->sport,
4201                                                  ssh, cfg,
4202                                                  &epf->local,
4203                                                  epf->addressfamily);
4204
4205                 logeventf(ssh, "Local %sport %s SOCKS dynamic forwarding%s%s",
4206                           epf->addressfamily == ADDRTYPE_IPV4 ? "IPv4 " :
4207                           epf->addressfamily == ADDRTYPE_IPV6 ? "IPv6 " : "",
4208                           sportdesc,
4209                           err ? " failed: " : "", err ? err : "");
4210             } else {
4211                 struct ssh_rportfwd *pf;
4212
4213                 /*
4214                  * Ensure the remote port forwardings tree exists.
4215                  */
4216                 if (!ssh->rportfwds) {
4217                     if (ssh->version == 1)
4218                         ssh->rportfwds = newtree234(ssh_rportcmp_ssh1);
4219                     else
4220                         ssh->rportfwds = newtree234(ssh_rportcmp_ssh2);
4221                 }
4222
4223                 pf = snew(struct ssh_rportfwd);
4224                 strncpy(pf->dhost, epf->daddr, lenof(pf->dhost)-1);
4225                 pf->dhost[lenof(pf->dhost)-1] = '\0';
4226                 pf->dport = epf->dport;
4227                 pf->sport = epf->sport;
4228                 if (add234(ssh->rportfwds, pf) != pf) {
4229                     logeventf(ssh, "Duplicate remote port forwarding to %s:%d",
4230                               epf->daddr, epf->dport);
4231                     sfree(pf);
4232                 } else {
4233                     logeventf(ssh, "Requesting remote port %s"
4234                               " forward to %s", sportdesc, dportdesc);
4235
4236                     pf->sportdesc = sportdesc;
4237                     sportdesc = NULL;
4238                     epf->remote = pf;
4239                     pf->pfrec = epf;
4240
4241                     if (ssh->version == 1) {
4242                         send_packet(ssh, SSH1_CMSG_PORT_FORWARD_REQUEST,
4243                                     PKT_INT, epf->sport,
4244                                     PKT_STR, epf->daddr,
4245                                     PKT_INT, epf->dport,
4246                                     PKT_END);
4247                         ssh_queue_handler(ssh, SSH1_SMSG_SUCCESS,
4248                                           SSH1_SMSG_FAILURE,
4249                                           ssh_rportfwd_succfail, pf);
4250                     } else {
4251                         struct Packet *pktout;
4252                         pktout = ssh2_pkt_init(SSH2_MSG_GLOBAL_REQUEST);
4253                         ssh2_pkt_addstring(pktout, "tcpip-forward");
4254                         ssh2_pkt_addbool(pktout, 1);/* want reply */
4255                         if (epf->saddr) {
4256                             ssh2_pkt_addstring(pktout, epf->saddr);
4257                         } else if (cfg->rport_acceptall) {
4258                             ssh2_pkt_addstring(pktout, "0.0.0.0");
4259                         } else {
4260                             ssh2_pkt_addstring(pktout, "127.0.0.1");
4261                         }
4262                         ssh2_pkt_adduint32(pktout, epf->sport);
4263                         ssh2_pkt_send(ssh, pktout);
4264
4265                         ssh_queue_handler(ssh, SSH2_MSG_REQUEST_SUCCESS,
4266                                           SSH2_MSG_REQUEST_FAILURE,
4267                                           ssh_rportfwd_succfail, pf);
4268                     }
4269                 }
4270             }
4271             sfree(sportdesc);
4272             sfree(dportdesc);
4273         }
4274 }
4275
4276 static void ssh1_smsg_stdout_stderr_data(Ssh ssh, struct Packet *pktin)
4277 {
4278     char *string;
4279     int stringlen, bufsize;
4280
4281     ssh_pkt_getstring(pktin, &string, &stringlen);
4282     if (string == NULL) {
4283         bombout(("Incoming terminal data packet was badly formed"));
4284         return;
4285     }
4286
4287     bufsize = from_backend(ssh->frontend, pktin->type == SSH1_SMSG_STDERR_DATA,
4288                            string, stringlen);
4289     if (!ssh->v1_stdout_throttling && bufsize > SSH1_BUFFER_LIMIT) {
4290         ssh->v1_stdout_throttling = 1;
4291         ssh1_throttle(ssh, +1);
4292     }
4293 }
4294
4295 static void ssh1_smsg_x11_open(Ssh ssh, struct Packet *pktin)
4296 {
4297     /* Remote side is trying to open a channel to talk to our
4298      * X-Server. Give them back a local channel number. */
4299     struct ssh_channel *c;
4300     int remoteid = ssh_pkt_getuint32(pktin);
4301
4302     logevent("Received X11 connect request");
4303     /* Refuse if X11 forwarding is disabled. */
4304     if (!ssh->X11_fwd_enabled) {
4305         send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_FAILURE,
4306                     PKT_INT, remoteid, PKT_END);
4307         logevent("Rejected X11 connect request");
4308     } else {
4309         c = snew(struct ssh_channel);
4310         c->ssh = ssh;
4311
4312         if (x11_init(&c->u.x11.s, ssh->cfg.x11_display, c,
4313                      ssh->x11auth, NULL, -1, &ssh->cfg) != NULL) {
4314             logevent("Opening X11 forward connection failed");
4315             sfree(c);
4316             send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_FAILURE,
4317                         PKT_INT, remoteid, PKT_END);
4318         } else {
4319             logevent
4320                 ("Opening X11 forward connection succeeded");
4321             c->remoteid = remoteid;
4322             c->halfopen = FALSE;
4323             c->localid = alloc_channel_id(ssh);
4324             c->closes = 0;
4325             c->v.v1.throttling = 0;
4326             c->type = CHAN_X11; /* identify channel type */
4327             add234(ssh->channels, c);
4328             send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_CONFIRMATION,
4329                         PKT_INT, c->remoteid, PKT_INT,
4330                         c->localid, PKT_END);
4331             logevent("Opened X11 forward channel");
4332         }
4333     }
4334 }
4335
4336 static void ssh1_smsg_agent_open(Ssh ssh, struct Packet *pktin)
4337 {
4338     /* Remote side is trying to open a channel to talk to our
4339      * agent. Give them back a local channel number. */
4340     struct ssh_channel *c;
4341     int remoteid = ssh_pkt_getuint32(pktin);
4342
4343     /* Refuse if agent forwarding is disabled. */
4344     if (!ssh->agentfwd_enabled) {
4345         send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_FAILURE,
4346                     PKT_INT, remoteid, PKT_END);
4347     } else {
4348         c = snew(struct ssh_channel);
4349         c->ssh = ssh;
4350         c->remoteid = remoteid;
4351         c->halfopen = FALSE;
4352         c->localid = alloc_channel_id(ssh);
4353         c->closes = 0;
4354         c->v.v1.throttling = 0;
4355         c->type = CHAN_AGENT;   /* identify channel type */
4356         c->u.a.lensofar = 0;
4357         add234(ssh->channels, c);
4358         send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_CONFIRMATION,
4359                     PKT_INT, c->remoteid, PKT_INT, c->localid,
4360                     PKT_END);
4361     }
4362 }
4363
4364 static void ssh1_msg_port_open(Ssh ssh, struct Packet *pktin)
4365 {
4366     /* Remote side is trying to open a channel to talk to a
4367      * forwarded port. Give them back a local channel number. */
4368     struct ssh_channel *c;
4369     struct ssh_rportfwd pf, *pfp;
4370     int remoteid;
4371     int hostsize, port;
4372     char *host;
4373     const char *e;
4374     c = snew(struct ssh_channel);
4375     c->ssh = ssh;
4376
4377     remoteid = ssh_pkt_getuint32(pktin);
4378     ssh_pkt_getstring(pktin, &host, &hostsize);
4379     port = ssh_pkt_getuint32(pktin);
4380
4381     if (hostsize >= lenof(pf.dhost))
4382         hostsize = lenof(pf.dhost)-1;
4383     memcpy(pf.dhost, host, hostsize);
4384     pf.dhost[hostsize] = '\0';
4385     pf.dport = port;
4386     pfp = find234(ssh->rportfwds, &pf, NULL);
4387
4388     if (pfp == NULL) {
4389         logeventf(ssh, "Rejected remote port open request for %s:%d",
4390                   pf.dhost, port);
4391         send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_FAILURE,
4392                     PKT_INT, remoteid, PKT_END);
4393     } else {
4394         logeventf(ssh, "Received remote port open request for %s:%d",
4395                   pf.dhost, port);
4396         e = pfd_newconnect(&c->u.pfd.s, pf.dhost, port,
4397                            c, &ssh->cfg, pfp->pfrec->addressfamily);
4398         if (e != NULL) {
4399             logeventf(ssh, "Port open failed: %s", e);
4400             sfree(c);
4401             send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_FAILURE,
4402                         PKT_INT, remoteid, PKT_END);
4403         } else {
4404             c->remoteid = remoteid;
4405             c->halfopen = FALSE;
4406             c->localid = alloc_channel_id(ssh);
4407             c->closes = 0;
4408             c->v.v1.throttling = 0;
4409             c->type = CHAN_SOCKDATA;    /* identify channel type */
4410             add234(ssh->channels, c);
4411             send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_CONFIRMATION,
4412                         PKT_INT, c->remoteid, PKT_INT,
4413                         c->localid, PKT_END);
4414             logevent("Forwarded port opened successfully");
4415         }
4416     }
4417 }
4418
4419 static void ssh1_msg_channel_open_confirmation(Ssh ssh, struct Packet *pktin)
4420 {
4421     unsigned int remoteid = ssh_pkt_getuint32(pktin);
4422     unsigned int localid = ssh_pkt_getuint32(pktin);
4423     struct ssh_channel *c;
4424
4425     c = find234(ssh->channels, &remoteid, ssh_channelfind);
4426     if (c && c->type == CHAN_SOCKDATA_DORMANT) {
4427         c->remoteid = localid;
4428         c->halfopen = FALSE;
4429         c->type = CHAN_SOCKDATA;
4430         c->v.v1.throttling = 0;
4431         pfd_confirm(c->u.pfd.s);
4432     }
4433
4434     if (c && c->closes) {
4435         /*
4436          * We have a pending close on this channel,
4437          * which we decided on before the server acked
4438          * the channel open. So now we know the
4439          * remoteid, we can close it again.
4440          */
4441         send_packet(ssh, SSH1_MSG_CHANNEL_CLOSE,
4442                     PKT_INT, c->remoteid, PKT_END);
4443     }
4444 }
4445
4446 static void ssh1_msg_channel_open_failure(Ssh ssh, struct Packet *pktin)
4447 {
4448     unsigned int remoteid = ssh_pkt_getuint32(pktin);
4449     struct ssh_channel *c;
4450
4451     c = find234(ssh->channels, &remoteid, ssh_channelfind);
4452     if (c && c->type == CHAN_SOCKDATA_DORMANT) {
4453         logevent("Forwarded connection refused by server");
4454         pfd_close(c->u.pfd.s);
4455         del234(ssh->channels, c);
4456         sfree(c);
4457     }
4458 }
4459
4460 static void ssh1_msg_channel_close(Ssh ssh, struct Packet *pktin)
4461 {
4462     /* Remote side closes a channel. */
4463     unsigned i = ssh_pkt_getuint32(pktin);
4464     struct ssh_channel *c;
4465     c = find234(ssh->channels, &i, ssh_channelfind);
4466     if (c && !c->halfopen) {
4467         int closetype;
4468         closetype =
4469             (pktin->type == SSH1_MSG_CHANNEL_CLOSE ? 1 : 2);
4470
4471         if ((c->closes == 0) && (c->type == CHAN_X11)) {
4472             logevent("Forwarded X11 connection terminated");
4473             assert(c->u.x11.s != NULL);
4474             x11_close(c->u.x11.s);
4475             c->u.x11.s = NULL;
4476         }
4477         if ((c->closes == 0) && (c->type == CHAN_SOCKDATA)) {
4478             logevent("Forwarded port closed");
4479             assert(c->u.pfd.s != NULL);
4480             pfd_close(c->u.pfd.s);
4481             c->u.pfd.s = NULL;
4482         }
4483
4484         c->closes |= (closetype << 2);   /* seen this message */
4485         if (!(c->closes & closetype)) {
4486             send_packet(ssh, pktin->type, PKT_INT, c->remoteid,
4487                         PKT_END);
4488             c->closes |= closetype;      /* sent it too */
4489         }
4490
4491         if (c->closes == 15) {
4492             del234(ssh->channels, c);
4493             sfree(c);
4494         }
4495     } else {
4496         bombout(("Received CHANNEL_CLOSE%s for %s channel %d\n",
4497                  pktin->type == SSH1_MSG_CHANNEL_CLOSE ? "" :
4498                  "_CONFIRMATION", c ? "half-open" : "nonexistent",
4499                  i));
4500     }
4501 }
4502
4503 static void ssh1_msg_channel_data(Ssh ssh, struct Packet *pktin)
4504 {
4505     /* Data sent down one of our channels. */
4506     int i = ssh_pkt_getuint32(pktin);
4507     char *p;
4508     int len;
4509     struct ssh_channel *c;
4510
4511     ssh_pkt_getstring(pktin, &p, &len);
4512
4513     c = find234(ssh->channels, &i, ssh_channelfind);
4514     if (c) {
4515         int bufsize = 0;
4516         switch (c->type) {
4517           case CHAN_X11:
4518             bufsize = x11_send(c->u.x11.s, p, len);
4519             break;
4520           case CHAN_SOCKDATA:
4521             bufsize = pfd_send(c->u.pfd.s, p, len);
4522             break;
4523           case CHAN_AGENT:
4524             /* Data for an agent message. Buffer it. */
4525             while (len > 0) {
4526                 if (c->u.a.lensofar < 4) {
4527                     unsigned int l = min(4 - c->u.a.lensofar, len);
4528                     memcpy(c->u.a.msglen + c->u.a.lensofar, p,
4529                            l);
4530                     p += l;
4531                     len -= l;
4532                     c->u.a.lensofar += l;
4533                 }
4534                 if (c->u.a.lensofar == 4) {
4535                     c->u.a.totallen =
4536                         4 + GET_32BIT(c->u.a.msglen);
4537                     c->u.a.message = snewn(c->u.a.totallen,
4538                                            unsigned char);
4539                     memcpy(c->u.a.message, c->u.a.msglen, 4);
4540                 }
4541                 if (c->u.a.lensofar >= 4 && len > 0) {
4542                     unsigned int l =
4543                         min(c->u.a.totallen - c->u.a.lensofar,
4544                             len);
4545                     memcpy(c->u.a.message + c->u.a.lensofar, p,
4546                            l);
4547                     p += l;
4548                     len -= l;
4549                     c->u.a.lensofar += l;
4550                 }
4551                 if (c->u.a.lensofar == c->u.a.totallen) {
4552                     void *reply;
4553                     int replylen;
4554                     if (agent_query(c->u.a.message,
4555                                     c->u.a.totallen,
4556                                     &reply, &replylen,
4557                                     ssh_agentf_callback, c))
4558                         ssh_agentf_callback(c, reply, replylen);
4559                     sfree(c->u.a.message);
4560                     c->u.a.lensofar = 0;
4561                 }
4562             }
4563             bufsize = 0;   /* agent channels never back up */
4564             break;
4565         }
4566         if (!c->v.v1.throttling && bufsize > SSH1_BUFFER_LIMIT) {
4567             c->v.v1.throttling = 1;
4568             ssh1_throttle(ssh, +1);
4569         }
4570     }
4571 }
4572
4573 static void ssh1_smsg_exit_status(Ssh ssh, struct Packet *pktin)
4574 {
4575     ssh->exitcode = ssh_pkt_getuint32(pktin);
4576     logeventf(ssh, "Server sent command exit status %d", ssh->exitcode);
4577     send_packet(ssh, SSH1_CMSG_EXIT_CONFIRMATION, PKT_END);
4578     /*
4579      * In case `helpful' firewalls or proxies tack
4580      * extra human-readable text on the end of the
4581      * session which we might mistake for another
4582      * encrypted packet, we close the session once
4583      * we've sent EXIT_CONFIRMATION.
4584      */
4585     ssh_disconnect(ssh, NULL, NULL, 0, TRUE);
4586 }
4587
4588 /* Helper function to deal with sending tty modes for REQUEST_PTY */
4589 static void ssh1_send_ttymode(void *data, char *mode, char *val)
4590 {
4591     struct Packet *pktout = (struct Packet *)data;
4592     int i = 0;
4593     unsigned int arg = 0;
4594     while (strcmp(mode, ssh_ttymodes[i].mode) != 0) i++;
4595     if (i == lenof(ssh_ttymodes)) return;
4596     switch (ssh_ttymodes[i].type) {
4597       case TTY_OP_CHAR:
4598         arg = ssh_tty_parse_specchar(val);
4599         break;
4600       case TTY_OP_BOOL:
4601         arg = ssh_tty_parse_boolean(val);
4602         break;
4603     }
4604     ssh2_pkt_addbyte(pktout, ssh_ttymodes[i].opcode);
4605     ssh2_pkt_addbyte(pktout, arg);
4606 }
4607
4608
4609 static void do_ssh1_connection(Ssh ssh, unsigned char *in, int inlen,
4610                                struct Packet *pktin)
4611 {
4612     crBegin(ssh->do_ssh1_connection_crstate);
4613
4614     ssh->packet_dispatch[SSH1_SMSG_STDOUT_DATA] = 
4615         ssh->packet_dispatch[SSH1_SMSG_STDERR_DATA] =
4616         ssh1_smsg_stdout_stderr_data;
4617
4618     ssh->packet_dispatch[SSH1_MSG_CHANNEL_OPEN_CONFIRMATION] =
4619         ssh1_msg_channel_open_confirmation;
4620     ssh->packet_dispatch[SSH1_MSG_CHANNEL_OPEN_FAILURE] =
4621         ssh1_msg_channel_open_failure;
4622     ssh->packet_dispatch[SSH1_MSG_CHANNEL_CLOSE] =
4623         ssh->packet_dispatch[SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION] =
4624         ssh1_msg_channel_close;
4625     ssh->packet_dispatch[SSH1_MSG_CHANNEL_DATA] = ssh1_msg_channel_data;
4626     ssh->packet_dispatch[SSH1_SMSG_EXIT_STATUS] = ssh1_smsg_exit_status;
4627
4628     if (ssh->cfg.agentfwd && agent_exists()) {
4629         logevent("Requesting agent forwarding");
4630         send_packet(ssh, SSH1_CMSG_AGENT_REQUEST_FORWARDING, PKT_END);
4631         do {
4632             crReturnV;
4633         } while (!pktin);
4634         if (pktin->type != SSH1_SMSG_SUCCESS
4635             && pktin->type != SSH1_SMSG_FAILURE) {
4636             bombout(("Protocol confusion"));
4637             crStopV;
4638         } else if (pktin->type == SSH1_SMSG_FAILURE) {
4639             logevent("Agent forwarding refused");
4640         } else {
4641             logevent("Agent forwarding enabled");
4642             ssh->agentfwd_enabled = TRUE;
4643             ssh->packet_dispatch[SSH1_SMSG_AGENT_OPEN] = ssh1_smsg_agent_open;
4644         }
4645     }
4646
4647     if (ssh->cfg.x11_forward) {
4648         char proto[20], data[64];
4649         logevent("Requesting X11 forwarding");
4650         ssh->x11auth = x11_invent_auth(proto, sizeof(proto),
4651                                        data, sizeof(data), ssh->cfg.x11_auth);
4652         x11_get_real_auth(ssh->x11auth, ssh->cfg.x11_display);
4653         /*
4654          * Note that while we blank the X authentication data here, we don't
4655          * take any special action to blank the start of an X11 channel,
4656          * so using MIT-MAGIC-COOKIE-1 and actually opening an X connection
4657          * without having session blanking enabled is likely to leak your
4658          * cookie into the log.
4659          */
4660         if (ssh->v1_local_protoflags & SSH1_PROTOFLAG_SCREEN_NUMBER) {
4661             send_packet(ssh, SSH1_CMSG_X11_REQUEST_FORWARDING,
4662                         PKT_STR, proto,
4663                         PKTT_PASSWORD, PKT_STR, data, PKTT_OTHER,
4664                         PKT_INT, x11_get_screen_number(ssh->cfg.x11_display),
4665                         PKT_END);
4666         } else {
4667             send_packet(ssh, SSH1_CMSG_X11_REQUEST_FORWARDING,
4668                         PKT_STR, proto,
4669                         PKTT_PASSWORD, PKT_STR, data, PKTT_OTHER, PKT_END);
4670         }
4671         do {
4672             crReturnV;
4673         } while (!pktin);
4674         if (pktin->type != SSH1_SMSG_SUCCESS
4675             && pktin->type != SSH1_SMSG_FAILURE) {
4676             bombout(("Protocol confusion"));
4677             crStopV;
4678         } else if (pktin->type == SSH1_SMSG_FAILURE) {
4679             logevent("X11 forwarding refused");
4680         } else {
4681             logevent("X11 forwarding enabled");
4682             ssh->X11_fwd_enabled = TRUE;
4683             ssh->packet_dispatch[SSH1_SMSG_X11_OPEN] = ssh1_smsg_x11_open;
4684         }
4685     }
4686
4687     ssh_setup_portfwd(ssh, &ssh->cfg);
4688     ssh->packet_dispatch[SSH1_MSG_PORT_OPEN] = ssh1_msg_port_open;
4689
4690     if (!ssh->cfg.nopty) {
4691         struct Packet *pkt;
4692         /* Unpick the terminal-speed string. */
4693         /* XXX perhaps we should allow no speeds to be sent. */
4694         ssh->ospeed = 38400; ssh->ispeed = 38400; /* last-resort defaults */
4695         sscanf(ssh->cfg.termspeed, "%d,%d", &ssh->ospeed, &ssh->ispeed);
4696         /* Send the pty request. */
4697         pkt = ssh1_pkt_init(SSH1_CMSG_REQUEST_PTY);
4698         ssh_pkt_addstring(pkt, ssh->cfg.termtype);
4699         ssh_pkt_adduint32(pkt, ssh->term_height);
4700         ssh_pkt_adduint32(pkt, ssh->term_width);
4701         ssh_pkt_adduint32(pkt, 0); /* width in pixels */
4702         ssh_pkt_adduint32(pkt, 0); /* height in pixels */
4703         parse_ttymodes(ssh, ssh->cfg.ttymodes,
4704                        ssh1_send_ttymode, (void *)pkt);
4705         ssh_pkt_addbyte(pkt, SSH1_TTY_OP_ISPEED);
4706         ssh_pkt_adduint32(pkt, ssh->ispeed);
4707         ssh_pkt_addbyte(pkt, SSH1_TTY_OP_OSPEED);
4708         ssh_pkt_adduint32(pkt, ssh->ospeed);
4709         ssh_pkt_addbyte(pkt, SSH_TTY_OP_END);
4710         s_wrpkt(ssh, pkt);
4711         ssh->state = SSH_STATE_INTERMED;
4712         do {
4713             crReturnV;
4714         } while (!pktin);
4715         if (pktin->type != SSH1_SMSG_SUCCESS
4716             && pktin->type != SSH1_SMSG_FAILURE) {
4717             bombout(("Protocol confusion"));
4718             crStopV;
4719         } else if (pktin->type == SSH1_SMSG_FAILURE) {
4720             c_write_str(ssh, "Server refused to allocate pty\r\n");
4721             ssh->editing = ssh->echoing = 1;
4722         }
4723         logeventf(ssh, "Allocated pty (ospeed %dbps, ispeed %dbps)",
4724                   ssh->ospeed, ssh->ispeed);
4725     } else {
4726         ssh->editing = ssh->echoing = 1;
4727     }
4728
4729     if (ssh->cfg.compression) {
4730         send_packet(ssh, SSH1_CMSG_REQUEST_COMPRESSION, PKT_INT, 6, PKT_END);
4731         do {
4732             crReturnV;
4733         } while (!pktin);
4734         if (pktin->type != SSH1_SMSG_SUCCESS
4735             && pktin->type != SSH1_SMSG_FAILURE) {
4736             bombout(("Protocol confusion"));
4737             crStopV;
4738         } else if (pktin->type == SSH1_SMSG_FAILURE) {
4739             c_write_str(ssh, "Server refused to compress\r\n");
4740         }
4741         logevent("Started compression");
4742         ssh->v1_compressing = TRUE;
4743         ssh->cs_comp_ctx = zlib_compress_init();
4744         logevent("Initialised zlib (RFC1950) compression");
4745         ssh->sc_comp_ctx = zlib_decompress_init();
4746         logevent("Initialised zlib (RFC1950) decompression");
4747     }
4748
4749     /*
4750      * Start the shell or command.
4751      * 
4752      * Special case: if the first-choice command is an SSH-2
4753      * subsystem (hence not usable here) and the second choice
4754      * exists, we fall straight back to that.
4755      */
4756     {
4757         char *cmd = ssh->cfg.remote_cmd_ptr;
4758
4759         if (!cmd) cmd = ssh->cfg.remote_cmd;
4760         
4761         if (ssh->cfg.ssh_subsys && ssh->cfg.remote_cmd_ptr2) {
4762             cmd = ssh->cfg.remote_cmd_ptr2;
4763             ssh->fallback_cmd = TRUE;
4764         }
4765         if (*cmd)
4766             send_packet(ssh, SSH1_CMSG_EXEC_CMD, PKT_STR, cmd, PKT_END);
4767         else
4768             send_packet(ssh, SSH1_CMSG_EXEC_SHELL, PKT_END);
4769         logevent("Started session");
4770     }
4771
4772     ssh->state = SSH_STATE_SESSION;
4773     if (ssh->size_needed)
4774         ssh_size(ssh, ssh->term_width, ssh->term_height);
4775     if (ssh->eof_needed)
4776         ssh_special(ssh, TS_EOF);
4777
4778     if (ssh->ldisc)
4779         ldisc_send(ssh->ldisc, NULL, 0, 0);/* cause ldisc to notice changes */
4780     ssh->send_ok = 1;
4781     ssh->channels = newtree234(ssh_channelcmp);
4782     while (1) {
4783
4784         /*
4785          * By this point, most incoming packets are already being
4786          * handled by the dispatch table, and we need only pay
4787          * attention to the unusual ones.
4788          */
4789
4790         crReturnV;
4791         if (pktin) {
4792             if (pktin->type == SSH1_SMSG_SUCCESS) {
4793                 /* may be from EXEC_SHELL on some servers */
4794             } else if (pktin->type == SSH1_SMSG_FAILURE) {
4795                 /* may be from EXEC_SHELL on some servers
4796                  * if no pty is available or in other odd cases. Ignore */
4797             } else {
4798                 bombout(("Strange packet received: type %d", pktin->type));
4799                 crStopV;
4800             }
4801         } else {
4802             while (inlen > 0) {
4803                 int len = min(inlen, 512);
4804                 send_packet(ssh, SSH1_CMSG_STDIN_DATA, PKTT_DATA,
4805                             PKT_INT, len, PKT_DATA, in, len,
4806                             PKTT_OTHER, PKT_END);
4807                 in += len;
4808                 inlen -= len;
4809             }
4810         }
4811     }
4812
4813     crFinishV;
4814 }
4815
4816 /*
4817  * Handle the top-level SSH-2 protocol.
4818  */
4819 static void ssh1_msg_debug(Ssh ssh, struct Packet *pktin)
4820 {
4821     char *msg;
4822     int msglen;
4823
4824     ssh_pkt_getstring(pktin, &msg, &msglen);
4825     logeventf(ssh, "Remote debug message: %.*s", msglen, msg);
4826 }
4827
4828 static void ssh1_msg_disconnect(Ssh ssh, struct Packet *pktin)
4829 {
4830     /* log reason code in disconnect message */
4831     char *msg;
4832     int msglen;
4833
4834     ssh_pkt_getstring(pktin, &msg, &msglen);
4835     bombout(("Server sent disconnect message:\n\"%.*s\"", msglen, msg));
4836 }
4837
4838 static void ssh_msg_ignore(Ssh ssh, struct Packet *pktin)
4839 {
4840     /* Do nothing, because we're ignoring it! Duhh. */
4841 }
4842
4843 static void ssh1_protocol_setup(Ssh ssh)
4844 {
4845     int i;
4846
4847     /*
4848      * Most messages are handled by the coroutines.
4849      */
4850     for (i = 0; i < 256; i++)
4851         ssh->packet_dispatch[i] = NULL;
4852
4853     /*
4854      * These special message types we install handlers for.
4855      */
4856     ssh->packet_dispatch[SSH1_MSG_DISCONNECT] = ssh1_msg_disconnect;
4857     ssh->packet_dispatch[SSH1_MSG_IGNORE] = ssh_msg_ignore;
4858     ssh->packet_dispatch[SSH1_MSG_DEBUG] = ssh1_msg_debug;
4859 }
4860
4861 static void ssh1_protocol(Ssh ssh, void *vin, int inlen,
4862                           struct Packet *pktin)
4863 {
4864     unsigned char *in=(unsigned char*)vin;
4865     if (ssh->state == SSH_STATE_CLOSED)
4866         return;
4867
4868     if (pktin && ssh->packet_dispatch[pktin->type]) {
4869         ssh->packet_dispatch[pktin->type](ssh, pktin);
4870         return;
4871     }
4872
4873     if (!ssh->protocol_initial_phase_done) {
4874         if (do_ssh1_login(ssh, in, inlen, pktin))
4875             ssh->protocol_initial_phase_done = TRUE;
4876         else
4877             return;
4878     }
4879
4880     do_ssh1_connection(ssh, in, inlen, pktin);
4881 }
4882
4883 /*
4884  * Utility routine for decoding comma-separated strings in KEXINIT.
4885  */
4886 static int in_commasep_string(char *needle, char *haystack, int haylen)
4887 {
4888     int needlen;
4889     if (!needle || !haystack)          /* protect against null pointers */
4890         return 0;
4891     needlen = strlen(needle);
4892     while (1) {
4893         /*
4894          * Is it at the start of the string?
4895          */
4896         if (haylen >= needlen &&       /* haystack is long enough */
4897             !memcmp(needle, haystack, needlen) &&       /* initial match */
4898             (haylen == needlen || haystack[needlen] == ',')
4899             /* either , or EOS follows */
4900             )
4901             return 1;
4902         /*
4903          * If not, search for the next comma and resume after that.
4904          * If no comma found, terminate.
4905          */
4906         while (haylen > 0 && *haystack != ',')
4907             haylen--, haystack++;
4908         if (haylen == 0)
4909             return 0;
4910         haylen--, haystack++;          /* skip over comma itself */
4911     }
4912 }
4913
4914 /*
4915  * Similar routine for checking whether we have the first string in a list.
4916  */
4917 static int first_in_commasep_string(char *needle, char *haystack, int haylen)
4918 {
4919     int needlen;
4920     if (!needle || !haystack)          /* protect against null pointers */
4921         return 0;
4922     needlen = strlen(needle);
4923     /*
4924      * Is it at the start of the string?
4925      */
4926     if (haylen >= needlen &&       /* haystack is long enough */
4927         !memcmp(needle, haystack, needlen) &&   /* initial match */
4928         (haylen == needlen || haystack[needlen] == ',')
4929         /* either , or EOS follows */
4930         )
4931         return 1;
4932     return 0;
4933 }
4934
4935
4936 /*
4937  * SSH-2 key creation method.
4938  */
4939 static void ssh2_mkkey(Ssh ssh, Bignum K, unsigned char *H,
4940                        unsigned char *sessid, char chr,
4941                        unsigned char *keyspace)
4942 {
4943     SHA_State s;
4944     /* First 20 bytes. */
4945     SHA_Init(&s);
4946     if (!(ssh->remote_bugs & BUG_SSH2_DERIVEKEY))
4947         sha_mpint(&s, K);
4948     SHA_Bytes(&s, H, 20);
4949     SHA_Bytes(&s, &chr, 1);
4950     SHA_Bytes(&s, sessid, 20);
4951     SHA_Final(&s, keyspace);
4952     /* Next 20 bytes. */
4953     SHA_Init(&s);
4954     if (!(ssh->remote_bugs & BUG_SSH2_DERIVEKEY))
4955         sha_mpint(&s, K);
4956     SHA_Bytes(&s, H, 20);
4957     SHA_Bytes(&s, keyspace, 20);
4958     SHA_Final(&s, keyspace + 20);
4959 }
4960
4961 /*
4962  * Handle the SSH-2 transport layer.
4963  */
4964 static int do_ssh2_transport(Ssh ssh, void *vin, int inlen,
4965                              struct Packet *pktin)
4966 {
4967     unsigned char *in = (unsigned char *)vin;
4968     struct do_ssh2_transport_state {
4969         int nbits, pbits, warn_kex, warn_cscipher, warn_sccipher;
4970         Bignum p, g, e, f, K;
4971         int kex_init_value, kex_reply_value;
4972         const struct ssh_mac **maclist;
4973         int nmacs;
4974         const struct ssh2_cipher *cscipher_tobe;
4975         const struct ssh2_cipher *sccipher_tobe;
4976         const struct ssh_mac *csmac_tobe;
4977         const struct ssh_mac *scmac_tobe;
4978         const struct ssh_compress *cscomp_tobe;
4979         const struct ssh_compress *sccomp_tobe;
4980         char *hostkeydata, *sigdata, *keystr, *fingerprint;
4981         int hostkeylen, siglen;
4982         void *hkey;                    /* actual host key */
4983         unsigned char exchange_hash[20];
4984         int n_preferred_kex;
4985         const struct ssh_kex *preferred_kex[KEX_MAX];
4986         int n_preferred_ciphers;
4987         const struct ssh2_ciphers *preferred_ciphers[CIPHER_MAX];
4988         const struct ssh_compress *preferred_comp;
4989         int got_session_id, activated_authconn;
4990         struct Packet *pktout;
4991         int dlgret;
4992         int guessok;
4993         int ignorepkt;
4994     };
4995     crState(do_ssh2_transport_state);
4996
4997     crBegin(ssh->do_ssh2_transport_crstate);
4998
4999     s->cscipher_tobe = s->sccipher_tobe = NULL;
5000     s->csmac_tobe = s->scmac_tobe = NULL;
5001     s->cscomp_tobe = s->sccomp_tobe = NULL;
5002
5003     s->got_session_id = s->activated_authconn = FALSE;
5004
5005     /*
5006      * Be prepared to work around the buggy MAC problem.
5007      */
5008     if (ssh->remote_bugs & BUG_SSH2_HMAC)
5009         s->maclist = buggymacs, s->nmacs = lenof(buggymacs);
5010     else
5011         s->maclist = macs, s->nmacs = lenof(macs);
5012
5013   begin_key_exchange:
5014     ssh->pkt_ctx &= ~SSH2_PKTCTX_KEX_MASK;
5015     {
5016         int i, j, commalist_started;
5017
5018         /*
5019          * Set up the preferred key exchange. (NULL => warn below here)
5020          */
5021         s->n_preferred_kex = 0;
5022         for (i = 0; i < KEX_MAX; i++) {
5023             switch (ssh->cfg.ssh_kexlist[i]) {
5024               case KEX_DHGEX:
5025                 s->preferred_kex[s->n_preferred_kex++] =
5026                     &ssh_diffiehellman_gex;
5027                 break;
5028               case KEX_DHGROUP14:
5029                 s->preferred_kex[s->n_preferred_kex++] =
5030                     &ssh_diffiehellman_group14;
5031                 break;
5032               case KEX_DHGROUP1:
5033                 s->preferred_kex[s->n_preferred_kex++] =
5034                     &ssh_diffiehellman_group1;
5035                 break;
5036               case CIPHER_WARN:
5037                 /* Flag for later. Don't bother if it's the last in
5038                  * the list. */
5039                 if (i < KEX_MAX - 1) {
5040                     s->preferred_kex[s->n_preferred_kex++] = NULL;
5041                 }
5042                 break;
5043             }
5044         }
5045
5046         /*
5047          * Set up the preferred ciphers. (NULL => warn below here)
5048          */
5049         s->n_preferred_ciphers = 0;
5050         for (i = 0; i < CIPHER_MAX; i++) {
5051             switch (ssh->cfg.ssh_cipherlist[i]) {
5052               case CIPHER_BLOWFISH:
5053                 s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_blowfish;
5054                 break;
5055               case CIPHER_DES:
5056                 if (ssh->cfg.ssh2_des_cbc) {
5057                     s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_des;
5058                 }
5059                 break;
5060               case CIPHER_3DES:
5061                 s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_3des;
5062                 break;
5063               case CIPHER_AES:
5064                 s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_aes;
5065                 break;
5066               case CIPHER_ARCFOUR:
5067                 s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_arcfour;
5068                 break;
5069               case CIPHER_WARN:
5070                 /* Flag for later. Don't bother if it's the last in
5071                  * the list. */
5072                 if (i < CIPHER_MAX - 1) {
5073                     s->preferred_ciphers[s->n_preferred_ciphers++] = NULL;
5074                 }
5075                 break;
5076             }
5077         }
5078
5079         /*
5080          * Set up preferred compression.
5081          */
5082         if (ssh->cfg.compression)
5083             s->preferred_comp = &ssh_zlib;
5084         else
5085             s->preferred_comp = &ssh_comp_none;
5086
5087         /*
5088          * Enable queueing of outgoing auth- or connection-layer
5089          * packets while we are in the middle of a key exchange.
5090          */
5091         ssh->queueing = TRUE;
5092
5093         /*
5094          * Flag that KEX is in progress.
5095          */
5096         ssh->kex_in_progress = TRUE;
5097
5098         /*
5099          * Construct and send our key exchange packet.
5100          */
5101         s->pktout = ssh2_pkt_init(SSH2_MSG_KEXINIT);
5102         for (i = 0; i < 16; i++)
5103             ssh2_pkt_addbyte(s->pktout, (unsigned char) random_byte());
5104         /* List key exchange algorithms. */
5105         ssh2_pkt_addstring_start(s->pktout);
5106         commalist_started = 0;
5107         for (i = 0; i < s->n_preferred_kex; i++) {
5108             const struct ssh_kex *k = s->preferred_kex[i];
5109             if (!k) continue;          /* warning flag */
5110             if (commalist_started)
5111                 ssh2_pkt_addstring_str(s->pktout, ",");
5112             ssh2_pkt_addstring_str(s->pktout, s->preferred_kex[i]->name);
5113             commalist_started = 1;
5114         }
5115         /* List server host key algorithms. */
5116         ssh2_pkt_addstring_start(s->pktout);
5117         for (i = 0; i < lenof(hostkey_algs); i++) {
5118             ssh2_pkt_addstring_str(s->pktout, hostkey_algs[i]->name);
5119             if (i < lenof(hostkey_algs) - 1)
5120                 ssh2_pkt_addstring_str(s->pktout, ",");
5121         }
5122         /* List client->server encryption algorithms. */
5123         ssh2_pkt_addstring_start(s->pktout);
5124         commalist_started = 0;
5125         for (i = 0; i < s->n_preferred_ciphers; i++) {
5126             const struct ssh2_ciphers *c = s->preferred_ciphers[i];
5127             if (!c) continue;          /* warning flag */
5128             for (j = 0; j < c->nciphers; j++) {
5129                 if (commalist_started)
5130                     ssh2_pkt_addstring_str(s->pktout, ",");
5131                 ssh2_pkt_addstring_str(s->pktout, c->list[j]->name);
5132                 commalist_started = 1;
5133             }
5134         }
5135         /* List server->client encryption algorithms. */
5136         ssh2_pkt_addstring_start(s->pktout);
5137         commalist_started = 0;
5138         for (i = 0; i < s->n_preferred_ciphers; i++) {
5139             const struct ssh2_ciphers *c = s->preferred_ciphers[i];
5140             if (!c) continue; /* warning flag */
5141             for (j = 0; j < c->nciphers; j++) {
5142                 if (commalist_started)
5143                     ssh2_pkt_addstring_str(s->pktout, ",");
5144                 ssh2_pkt_addstring_str(s->pktout, c->list[j]->name);
5145                 commalist_started = 1;
5146             }
5147         }
5148         /* List client->server MAC algorithms. */
5149         ssh2_pkt_addstring_start(s->pktout);
5150         for (i = 0; i < s->nmacs; i++) {
5151             ssh2_pkt_addstring_str(s->pktout, s->maclist[i]->name);
5152             if (i < s->nmacs - 1)
5153                 ssh2_pkt_addstring_str(s->pktout, ",");
5154         }
5155         /* List server->client MAC algorithms. */
5156         ssh2_pkt_addstring_start(s->pktout);
5157         for (i = 0; i < s->nmacs; i++) {
5158             ssh2_pkt_addstring_str(s->pktout, s->maclist[i]->name);
5159             if (i < s->nmacs - 1)
5160                 ssh2_pkt_addstring_str(s->pktout, ",");
5161         }
5162         /* List client->server compression algorithms. */
5163         ssh2_pkt_addstring_start(s->pktout);
5164         assert(lenof(compressions) > 1);
5165         ssh2_pkt_addstring_str(s->pktout, s->preferred_comp->name);
5166         for (i = 0; i < lenof(compressions); i++) {
5167             const struct ssh_compress *c = compressions[i];
5168             if (c != s->preferred_comp) {
5169                 ssh2_pkt_addstring_str(s->pktout, ",");
5170                 ssh2_pkt_addstring_str(s->pktout, c->name);
5171             }
5172         }
5173         /* List server->client compression algorithms. */
5174         ssh2_pkt_addstring_start(s->pktout);
5175         assert(lenof(compressions) > 1);
5176         ssh2_pkt_addstring_str(s->pktout, s->preferred_comp->name);
5177         for (i = 0; i < lenof(compressions); i++) {
5178             const struct ssh_compress *c = compressions[i];
5179             if (c != s->preferred_comp) {
5180                 ssh2_pkt_addstring_str(s->pktout, ",");
5181                 ssh2_pkt_addstring_str(s->pktout, c->name);
5182             }
5183         }
5184         /* List client->server languages. Empty list. */
5185         ssh2_pkt_addstring_start(s->pktout);
5186         /* List server->client languages. Empty list. */
5187         ssh2_pkt_addstring_start(s->pktout);
5188         /* First KEX packet does _not_ follow, because we're not that brave. */
5189         ssh2_pkt_addbool(s->pktout, FALSE);
5190         /* Reserved. */
5191         ssh2_pkt_adduint32(s->pktout, 0);
5192     }
5193
5194     ssh->exhash = ssh->exhashbase;
5195     sha_string(&ssh->exhash, s->pktout->data + 5, s->pktout->length - 5);
5196
5197     ssh2_pkt_send_noqueue(ssh, s->pktout);
5198
5199     if (!pktin)
5200         crWaitUntil(pktin);
5201     if (pktin->length > 5)
5202         sha_string(&ssh->exhash, pktin->data + 5, pktin->length - 5);
5203
5204     /*
5205      * Now examine the other side's KEXINIT to see what we're up
5206      * to.
5207      */
5208     {
5209         char *str, *preferred;
5210         int i, j, len;
5211
5212         if (pktin->type != SSH2_MSG_KEXINIT) {
5213             bombout(("expected key exchange packet from server"));
5214             crStop(0);
5215         }
5216         ssh->kex = NULL;
5217         ssh->hostkey = NULL;
5218         s->cscipher_tobe = NULL;
5219         s->sccipher_tobe = NULL;
5220         s->csmac_tobe = NULL;
5221         s->scmac_tobe = NULL;
5222         s->cscomp_tobe = NULL;
5223         s->sccomp_tobe = NULL;
5224         s->warn_kex = s->warn_cscipher = s->warn_sccipher = FALSE;
5225
5226         pktin->savedpos += 16;          /* skip garbage cookie */
5227         ssh_pkt_getstring(pktin, &str, &len);    /* key exchange algorithms */
5228
5229         preferred = NULL;
5230         for (i = 0; i < s->n_preferred_kex; i++) {
5231             const struct ssh_kex *k = s->preferred_kex[i];
5232             if (!k) {
5233                 s->warn_kex = TRUE;
5234             } else {
5235                 if (!preferred) preferred = k->name;
5236                 if (in_commasep_string(k->name, str, len))
5237                     ssh->kex = k;
5238             }
5239             if (ssh->kex)
5240                 break;
5241         }
5242         if (!ssh->kex) {
5243             bombout(("Couldn't agree a key exchange algorithm (available: %s)",
5244                      str ? str : "(null)"));
5245             crStop(0);
5246         }
5247         /*
5248          * Note that the server's guess is considered wrong if it doesn't match
5249          * the first algorithm in our list, even if it's still the algorithm
5250          * we end up using.
5251          */
5252         s->guessok = first_in_commasep_string(preferred, str, len);
5253         ssh_pkt_getstring(pktin, &str, &len);    /* host key algorithms */
5254         for (i = 0; i < lenof(hostkey_algs); i++) {
5255             if (in_commasep_string(hostkey_algs[i]->name, str, len)) {
5256                 ssh->hostkey = hostkey_algs[i];
5257                 break;
5258             }
5259         }
5260         s->guessok = s->guessok &&
5261             first_in_commasep_string(hostkey_algs[0]->name, str, len);
5262         ssh_pkt_getstring(pktin, &str, &len);    /* client->server cipher */
5263         for (i = 0; i < s->n_preferred_ciphers; i++) {
5264             const struct ssh2_ciphers *c = s->preferred_ciphers[i];
5265             if (!c) {
5266                 s->warn_cscipher = TRUE;
5267             } else {
5268                 for (j = 0; j < c->nciphers; j++) {
5269                     if (in_commasep_string(c->list[j]->name, str, len)) {
5270                         s->cscipher_tobe = c->list[j];
5271                         break;
5272                     }
5273                 }
5274             }
5275             if (s->cscipher_tobe)
5276                 break;
5277         }
5278         if (!s->cscipher_tobe) {
5279             bombout(("Couldn't agree a client-to-server cipher (available: %s)",
5280                      str ? str : "(null)"));
5281             crStop(0);
5282         }
5283
5284         ssh_pkt_getstring(pktin, &str, &len);    /* server->client cipher */
5285         for (i = 0; i < s->n_preferred_ciphers; i++) {
5286             const struct ssh2_ciphers *c = s->preferred_ciphers[i];
5287             if (!c) {
5288                 s->warn_sccipher = TRUE;
5289             } else {
5290                 for (j = 0; j < c->nciphers; j++) {
5291                     if (in_commasep_string(c->list[j]->name, str, len)) {
5292                         s->sccipher_tobe = c->list[j];
5293                         break;
5294                     }
5295                 }
5296             }
5297             if (s->sccipher_tobe)
5298                 break;
5299         }
5300         if (!s->sccipher_tobe) {
5301             bombout(("Couldn't agree a server-to-client cipher (available: %s)",
5302                      str ? str : "(null)"));
5303             crStop(0);
5304         }
5305
5306         ssh_pkt_getstring(pktin, &str, &len);    /* client->server mac */
5307         for (i = 0; i < s->nmacs; i++) {
5308             if (in_commasep_string(s->maclist[i]->name, str, len)) {
5309                 s->csmac_tobe = s->maclist[i];
5310                 break;
5311             }
5312         }
5313         ssh_pkt_getstring(pktin, &str, &len);    /* server->client mac */
5314         for (i = 0; i < s->nmacs; i++) {
5315             if (in_commasep_string(s->maclist[i]->name, str, len)) {
5316                 s->scmac_tobe = s->maclist[i];
5317                 break;
5318             }
5319         }
5320         ssh_pkt_getstring(pktin, &str, &len);  /* client->server compression */
5321         for (i = 0; i < lenof(compressions) + 1; i++) {
5322             const struct ssh_compress *c =
5323                 i == 0 ? s->preferred_comp : compressions[i - 1];
5324             if (in_commasep_string(c->name, str, len)) {
5325                 s->cscomp_tobe = c;
5326                 break;
5327             }
5328         }
5329         ssh_pkt_getstring(pktin, &str, &len);  /* server->client compression */
5330         for (i = 0; i < lenof(compressions) + 1; i++) {
5331             const struct ssh_compress *c =
5332                 i == 0 ? s->preferred_comp : compressions[i - 1];
5333             if (in_commasep_string(c->name, str, len)) {
5334                 s->sccomp_tobe = c;
5335                 break;
5336             }
5337         }
5338         ssh_pkt_getstring(pktin, &str, &len);  /* client->server language */
5339         ssh_pkt_getstring(pktin, &str, &len);  /* server->client language */
5340         s->ignorepkt = ssh2_pkt_getbool(pktin) && !s->guessok;
5341
5342         if (s->warn_kex) {
5343             ssh_set_frozen(ssh, 1);
5344             s->dlgret = askalg(ssh->frontend, "key-exchange algorithm",
5345                                ssh->kex->name,
5346                                ssh_dialog_callback, ssh);
5347             if (s->dlgret < 0) {
5348                 do {
5349                     crReturn(0);
5350                     if (pktin) {
5351                         bombout(("Unexpected data from server while"
5352                                  " waiting for user response"));
5353                         crStop(0);
5354                     }
5355                 } while (pktin || inlen > 0);
5356                 s->dlgret = ssh->user_response;
5357             }
5358             ssh_set_frozen(ssh, 0);
5359             if (s->dlgret == 0) {
5360                 ssh_disconnect(ssh, "User aborted at kex warning", NULL,
5361                                0, TRUE);
5362                 crStop(0);
5363             }
5364         }
5365
5366         if (s->warn_cscipher) {
5367             ssh_set_frozen(ssh, 1);
5368             s->dlgret = askalg(ssh->frontend,
5369                                "client-to-server cipher",
5370                                s->cscipher_tobe->name,
5371                                ssh_dialog_callback, ssh);
5372             if (s->dlgret < 0) {
5373                 do {
5374                     crReturn(0);
5375                     if (pktin) {
5376                         bombout(("Unexpected data from server while"
5377                                  " waiting for user response"));
5378                         crStop(0);
5379                     }
5380                 } while (pktin || inlen > 0);
5381                 s->dlgret = ssh->user_response;
5382             }
5383             ssh_set_frozen(ssh, 0);
5384             if (s->dlgret == 0) {
5385                 ssh_disconnect(ssh, "User aborted at cipher warning", NULL,
5386                                0, TRUE);
5387                 crStop(0);
5388             }
5389         }
5390
5391         if (s->warn_sccipher) {
5392             ssh_set_frozen(ssh, 1);
5393             s->dlgret = askalg(ssh->frontend,
5394                                "server-to-client cipher",
5395                                s->sccipher_tobe->name,
5396                                ssh_dialog_callback, ssh);
5397             if (s->dlgret < 0) {
5398                 do {
5399                     crReturn(0);
5400                     if (pktin) {
5401                         bombout(("Unexpected data from server while"
5402                                  " waiting for user response"));
5403                         crStop(0);
5404                     }
5405                 } while (pktin || inlen > 0);
5406                 s->dlgret = ssh->user_response;
5407             }
5408             ssh_set_frozen(ssh, 0);
5409             if (s->dlgret == 0) {
5410                 ssh_disconnect(ssh, "User aborted at cipher warning", NULL,
5411                                0, TRUE);
5412                 crStop(0);
5413             }
5414         }
5415
5416         if (s->ignorepkt) /* first_kex_packet_follows */
5417             crWaitUntil(pktin);                /* Ignore packet */
5418     }
5419
5420     /*
5421      * Work out the number of bits of key we will need from the key
5422      * exchange. We start with the maximum key length of either
5423      * cipher...
5424      */
5425     {
5426         int csbits, scbits;
5427
5428         csbits = s->cscipher_tobe->keylen;
5429         scbits = s->sccipher_tobe->keylen;
5430         s->nbits = (csbits > scbits ? csbits : scbits);
5431     }
5432     /* The keys only have 160-bit entropy, since they're based on
5433      * a SHA-1 hash. So cap the key size at 160 bits. */
5434     if (s->nbits > 160)
5435         s->nbits = 160;
5436
5437     /*
5438      * If we're doing Diffie-Hellman group exchange, start by
5439      * requesting a group.
5440      */
5441     if (!ssh->kex->pdata) {
5442         logevent("Doing Diffie-Hellman group exchange");
5443         ssh->pkt_ctx |= SSH2_PKTCTX_DHGEX;
5444         /*
5445          * Work out how big a DH group we will need to allow that
5446          * much data.
5447          */
5448         s->pbits = 512 << ((s->nbits - 1) / 64);
5449         s->pktout = ssh2_pkt_init(SSH2_MSG_KEX_DH_GEX_REQUEST);
5450         ssh2_pkt_adduint32(s->pktout, s->pbits);
5451         ssh2_pkt_send_noqueue(ssh, s->pktout);
5452
5453         crWaitUntil(pktin);
5454         if (pktin->type != SSH2_MSG_KEX_DH_GEX_GROUP) {
5455             bombout(("expected key exchange group packet from server"));
5456             crStop(0);
5457         }
5458         s->p = ssh2_pkt_getmp(pktin);
5459         s->g = ssh2_pkt_getmp(pktin);
5460         if (!s->p || !s->g) {
5461             bombout(("unable to read mp-ints from incoming group packet"));
5462             crStop(0);
5463         }
5464         ssh->kex_ctx = dh_setup_gex(s->p, s->g);
5465         s->kex_init_value = SSH2_MSG_KEX_DH_GEX_INIT;
5466         s->kex_reply_value = SSH2_MSG_KEX_DH_GEX_REPLY;
5467     } else {
5468         ssh->pkt_ctx |= SSH2_PKTCTX_DHGROUP;
5469         ssh->kex_ctx = dh_setup_group(ssh->kex);
5470         s->kex_init_value = SSH2_MSG_KEXDH_INIT;
5471         s->kex_reply_value = SSH2_MSG_KEXDH_REPLY;
5472         logeventf(ssh, "Using Diffie-Hellman with standard group \"%s\"",
5473                   ssh->kex->groupname);
5474     }
5475
5476     logevent("Doing Diffie-Hellman key exchange");
5477     /*
5478      * Now generate and send e for Diffie-Hellman.
5479      */
5480     set_busy_status(ssh->frontend, BUSY_CPU); /* this can take a while */
5481     s->e = dh_create_e(ssh->kex_ctx, s->nbits * 2);
5482     s->pktout = ssh2_pkt_init(s->kex_init_value);
5483     ssh2_pkt_addmp(s->pktout, s->e);
5484     ssh2_pkt_send_noqueue(ssh, s->pktout);
5485
5486     set_busy_status(ssh->frontend, BUSY_WAITING); /* wait for server */
5487     crWaitUntil(pktin);
5488     if (pktin->type != s->kex_reply_value) {
5489         bombout(("expected key exchange reply packet from server"));
5490         crStop(0);
5491     }
5492     set_busy_status(ssh->frontend, BUSY_CPU); /* cogitate */
5493     ssh_pkt_getstring(pktin, &s->hostkeydata, &s->hostkeylen);
5494     s->f = ssh2_pkt_getmp(pktin);
5495     if (!s->f) {
5496         bombout(("unable to parse key exchange reply packet"));
5497         crStop(0);
5498     }
5499     ssh_pkt_getstring(pktin, &s->sigdata, &s->siglen);
5500
5501     s->K = dh_find_K(ssh->kex_ctx, s->f);
5502
5503     /* We assume everything from now on will be quick, and it might
5504      * involve user interaction. */
5505     set_busy_status(ssh->frontend, BUSY_NOT);
5506
5507     sha_string(&ssh->exhash, s->hostkeydata, s->hostkeylen);
5508     if (ssh->kex == &ssh_diffiehellman_gex) {
5509         sha_uint32(&ssh->exhash, s->pbits);
5510         sha_mpint(&ssh->exhash, s->p);
5511         sha_mpint(&ssh->exhash, s->g);
5512     }
5513     sha_mpint(&ssh->exhash, s->e);
5514     sha_mpint(&ssh->exhash, s->f);
5515     sha_mpint(&ssh->exhash, s->K);
5516     SHA_Final(&ssh->exhash, s->exchange_hash);
5517
5518     dh_cleanup(ssh->kex_ctx);
5519     ssh->kex_ctx = NULL;
5520
5521 #if 0
5522     debug(("Exchange hash is:\n"));
5523     dmemdump(s->exchange_hash, 20);
5524 #endif
5525
5526     s->hkey = ssh->hostkey->newkey(s->hostkeydata, s->hostkeylen);
5527     if (!s->hkey ||
5528         !ssh->hostkey->verifysig(s->hkey, s->sigdata, s->siglen,
5529                                  (char *)s->exchange_hash, 20)) {
5530         bombout(("Server's host key did not match the signature supplied"));
5531         crStop(0);
5532     }
5533
5534     /*
5535      * Authenticate remote host: verify host key. (We've already
5536      * checked the signature of the exchange hash.)
5537      */
5538     s->keystr = ssh->hostkey->fmtkey(s->hkey);
5539     s->fingerprint = ssh->hostkey->fingerprint(s->hkey);
5540     ssh_set_frozen(ssh, 1);
5541     s->dlgret = verify_ssh_host_key(ssh->frontend,
5542                                     ssh->savedhost, ssh->savedport,
5543                                     ssh->hostkey->keytype, s->keystr,
5544                                     s->fingerprint,
5545                                     ssh_dialog_callback, ssh);
5546     if (s->dlgret < 0) {
5547         do {
5548             crReturn(0);
5549             if (pktin) {
5550                 bombout(("Unexpected data from server while waiting"
5551                          " for user host key response"));
5552                     crStop(0);
5553             }
5554         } while (pktin || inlen > 0);
5555         s->dlgret = ssh->user_response;
5556     }
5557     ssh_set_frozen(ssh, 0);
5558     if (s->dlgret == 0) {
5559         ssh_disconnect(ssh, "User aborted at host key verification", NULL,
5560                        0, TRUE);
5561         crStop(0);
5562     }
5563     if (!s->got_session_id) {     /* don't bother logging this in rekeys */
5564         logevent("Host key fingerprint is:");
5565         logevent(s->fingerprint);
5566     }
5567     sfree(s->fingerprint);
5568     sfree(s->keystr);
5569     ssh->hostkey->freekey(s->hkey);
5570
5571     /*
5572      * The exchange hash from the very first key exchange is also
5573      * the session id, used in session key construction and
5574      * authentication.
5575      */
5576     if (!s->got_session_id) {
5577         memcpy(ssh->v2_session_id, s->exchange_hash,
5578                sizeof(s->exchange_hash));
5579         s->got_session_id = TRUE;
5580     }
5581
5582     /*
5583      * Send SSH2_MSG_NEWKEYS.
5584      */
5585     s->pktout = ssh2_pkt_init(SSH2_MSG_NEWKEYS);
5586     ssh2_pkt_send_noqueue(ssh, s->pktout);
5587     ssh->outgoing_data_size = 0;       /* start counting from here */
5588
5589     /*
5590      * We've sent client NEWKEYS, so create and initialise
5591      * client-to-server session keys.
5592      */
5593     if (ssh->cs_cipher_ctx)
5594         ssh->cscipher->free_context(ssh->cs_cipher_ctx);
5595     ssh->cscipher = s->cscipher_tobe;
5596     ssh->cs_cipher_ctx = ssh->cscipher->make_context();
5597
5598     if (ssh->cs_mac_ctx)
5599         ssh->csmac->free_context(ssh->cs_mac_ctx);
5600     ssh->csmac = s->csmac_tobe;
5601     ssh->cs_mac_ctx = ssh->csmac->make_context();
5602
5603     if (ssh->cs_comp_ctx)
5604         ssh->cscomp->compress_cleanup(ssh->cs_comp_ctx);
5605     ssh->cscomp = s->cscomp_tobe;
5606     ssh->cs_comp_ctx = ssh->cscomp->compress_init();
5607
5608     /*
5609      * Set IVs on client-to-server keys. Here we use the exchange
5610      * hash from the _first_ key exchange.
5611      */
5612     {
5613         unsigned char keyspace[40];
5614         ssh2_mkkey(ssh,s->K,s->exchange_hash,ssh->v2_session_id,'C',keyspace);
5615         ssh->cscipher->setkey(ssh->cs_cipher_ctx, keyspace);
5616         ssh2_mkkey(ssh,s->K,s->exchange_hash,ssh->v2_session_id,'A',keyspace);
5617         ssh->cscipher->setiv(ssh->cs_cipher_ctx, keyspace);
5618         ssh2_mkkey(ssh,s->K,s->exchange_hash,ssh->v2_session_id,'E',keyspace);
5619         ssh->csmac->setkey(ssh->cs_mac_ctx, keyspace);
5620     }
5621
5622     logeventf(ssh, "Initialised %.200s client->server encryption",
5623               ssh->cscipher->text_name);
5624     logeventf(ssh, "Initialised %.200s client->server MAC algorithm",
5625               ssh->csmac->text_name);
5626     if (ssh->cscomp->text_name)
5627         logeventf(ssh, "Initialised %s compression",
5628                   ssh->cscomp->text_name);
5629
5630     /*
5631      * Now our end of the key exchange is complete, we can send all
5632      * our queued higher-layer packets.
5633      */
5634     ssh->queueing = FALSE;
5635     ssh2_pkt_queuesend(ssh);
5636
5637     /*
5638      * Expect SSH2_MSG_NEWKEYS from server.
5639      */
5640     crWaitUntil(pktin);
5641     if (pktin->type != SSH2_MSG_NEWKEYS) {
5642         bombout(("expected new-keys packet from server"));
5643         crStop(0);
5644     }
5645     ssh->incoming_data_size = 0;       /* start counting from here */
5646
5647     /*
5648      * We've seen server NEWKEYS, so create and initialise
5649      * server-to-client session keys.
5650      */
5651     if (ssh->sc_cipher_ctx)
5652         ssh->sccipher->free_context(ssh->sc_cipher_ctx);
5653     ssh->sccipher = s->sccipher_tobe;
5654     ssh->sc_cipher_ctx = ssh->sccipher->make_context();
5655
5656     if (ssh->sc_mac_ctx)
5657         ssh->scmac->free_context(ssh->sc_mac_ctx);
5658     ssh->scmac = s->scmac_tobe;
5659     ssh->sc_mac_ctx = ssh->scmac->make_context();
5660
5661     if (ssh->sc_comp_ctx)
5662         ssh->sccomp->decompress_cleanup(ssh->sc_comp_ctx);
5663     ssh->sccomp = s->sccomp_tobe;
5664     ssh->sc_comp_ctx = ssh->sccomp->decompress_init();
5665
5666     /*
5667      * Set IVs on server-to-client keys. Here we use the exchange
5668      * hash from the _first_ key exchange.
5669      */
5670     {
5671         unsigned char keyspace[40];
5672         ssh2_mkkey(ssh,s->K,s->exchange_hash,ssh->v2_session_id,'D',keyspace);
5673         ssh->sccipher->setkey(ssh->sc_cipher_ctx, keyspace);
5674         ssh2_mkkey(ssh,s->K,s->exchange_hash,ssh->v2_session_id,'B',keyspace);
5675         ssh->sccipher->setiv(ssh->sc_cipher_ctx, keyspace);
5676         ssh2_mkkey(ssh,s->K,s->exchange_hash,ssh->v2_session_id,'F',keyspace);
5677         ssh->scmac->setkey(ssh->sc_mac_ctx, keyspace);
5678     }
5679     logeventf(ssh, "Initialised %.200s server->client encryption",
5680               ssh->sccipher->text_name);
5681     logeventf(ssh, "Initialised %.200s server->client MAC algorithm",
5682               ssh->scmac->text_name);
5683     if (ssh->sccomp->text_name)
5684         logeventf(ssh, "Initialised %s decompression",
5685                   ssh->sccomp->text_name);
5686
5687     /*
5688      * Free key exchange data.
5689      */
5690     freebn(s->f);
5691     freebn(s->K);
5692     if (ssh->kex == &ssh_diffiehellman_gex) {
5693         freebn(s->g);
5694         freebn(s->p);
5695     }
5696
5697     /*
5698      * Key exchange is over. Loop straight back round if we have a
5699      * deferred rekey reason.
5700      */
5701     if (ssh->deferred_rekey_reason) {
5702         logevent(ssh->deferred_rekey_reason);
5703         pktin = NULL;
5704         ssh->deferred_rekey_reason = NULL;
5705         goto begin_key_exchange;
5706     }
5707
5708     /*
5709      * Otherwise, schedule a timer for our next rekey.
5710      */
5711     ssh->kex_in_progress = FALSE;
5712     ssh->last_rekey = GETTICKCOUNT();
5713     if (ssh->cfg.ssh_rekey_time != 0)
5714         ssh->next_rekey = schedule_timer(ssh->cfg.ssh_rekey_time*60*TICKSPERSEC,
5715                                          ssh2_timer, ssh);
5716
5717     /*
5718      * If this is the first key exchange phase, we must pass the
5719      * SSH2_MSG_NEWKEYS packet to the next layer, not because it
5720      * wants to see it but because it will need time to initialise
5721      * itself before it sees an actual packet. In subsequent key
5722      * exchange phases, we don't pass SSH2_MSG_NEWKEYS on, because
5723      * it would only confuse the layer above.
5724      */
5725     if (s->activated_authconn) {
5726         crReturn(1);
5727     }
5728     s->activated_authconn = TRUE;
5729
5730     /*
5731      * Now we're encrypting. Begin returning 1 to the protocol main
5732      * function so that other things can run on top of the
5733      * transport. If we ever see a KEXINIT, we must go back to the
5734      * start.
5735      * 
5736      * We _also_ go back to the start if we see pktin==NULL and
5737      * inlen==-1, because this is a special signal meaning
5738      * `initiate client-driven rekey', and `in' contains a message
5739      * giving the reason for the rekey.
5740      */
5741     while (!((pktin && pktin->type == SSH2_MSG_KEXINIT) ||
5742              (!pktin && inlen == -1))) {
5743         wait_for_rekey:
5744         crReturn(1);
5745     }
5746     if (pktin) {
5747         logevent("Server initiated key re-exchange");
5748     } else {
5749         /*
5750          * Special case: if the server bug is set that doesn't
5751          * allow rekeying, we give a different log message and
5752          * continue waiting. (If such a server _initiates_ a rekey,
5753          * we process it anyway!)
5754          */
5755         if ((ssh->remote_bugs & BUG_SSH2_REKEY)) {
5756             logeventf(ssh, "Server bug prevents key re-exchange (%s)",
5757                       (char *)in);
5758             /* Reset the counters, so that at least this message doesn't
5759              * hit the event log _too_ often. */
5760             ssh->outgoing_data_size = 0;
5761             ssh->incoming_data_size = 0;
5762             if (ssh->cfg.ssh_rekey_time != 0) {
5763                 ssh->next_rekey =
5764                     schedule_timer(ssh->cfg.ssh_rekey_time*60*TICKSPERSEC,
5765                                    ssh2_timer, ssh);
5766             }
5767             goto wait_for_rekey;       /* this is utterly horrid */
5768         } else {
5769             logeventf(ssh, "Initiating key re-exchange (%s)", (char *)in);
5770         }
5771     }
5772     goto begin_key_exchange;
5773
5774     crFinish(1);
5775 }
5776
5777 /*
5778  * Add data to an SSH-2 channel output buffer.
5779  */
5780 static void ssh2_add_channel_data(struct ssh_channel *c, char *buf,
5781                                   int len)
5782 {
5783     bufchain_add(&c->v.v2.outbuffer, buf, len);
5784 }
5785
5786 /*
5787  * Attempt to send data on an SSH-2 channel.
5788  */
5789 static int ssh2_try_send(struct ssh_channel *c)
5790 {
5791     Ssh ssh = c->ssh;
5792     struct Packet *pktout;
5793
5794     while (c->v.v2.remwindow > 0 && bufchain_size(&c->v.v2.outbuffer) > 0) {
5795         int len;
5796         void *data;
5797         bufchain_prefix(&c->v.v2.outbuffer, &data, &len);
5798         if ((unsigned)len > c->v.v2.remwindow)
5799             len = c->v.v2.remwindow;
5800         if ((unsigned)len > c->v.v2.remmaxpkt)
5801             len = c->v.v2.remmaxpkt;
5802         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_DATA);
5803         ssh2_pkt_adduint32(pktout, c->remoteid);
5804         dont_log_data(ssh, pktout, PKTLOG_OMIT);
5805         ssh2_pkt_addstring_start(pktout);
5806         ssh2_pkt_addstring_data(pktout, data, len);
5807         end_log_omission(ssh, pktout);
5808         ssh2_pkt_send(ssh, pktout);
5809         bufchain_consume(&c->v.v2.outbuffer, len);
5810         c->v.v2.remwindow -= len;
5811     }
5812
5813     /*
5814      * After having sent as much data as we can, return the amount
5815      * still buffered.
5816      */
5817     return bufchain_size(&c->v.v2.outbuffer);
5818 }
5819
5820 static void ssh2_try_send_and_unthrottle(struct ssh_channel *c)
5821 {
5822     int bufsize;
5823     if (c->closes)
5824         return;                        /* don't send on closing channels */
5825     bufsize = ssh2_try_send(c);
5826     if (bufsize == 0) {
5827         switch (c->type) {
5828           case CHAN_MAINSESSION:
5829             /* stdin need not receive an unthrottle
5830              * notification since it will be polled */
5831             break;
5832           case CHAN_X11:
5833             x11_unthrottle(c->u.x11.s);
5834             break;
5835           case CHAN_AGENT:
5836             /* agent sockets are request/response and need no
5837              * buffer management */
5838             break;
5839           case CHAN_SOCKDATA:
5840             pfd_unthrottle(c->u.pfd.s);
5841             break;
5842         }
5843     }
5844 }
5845
5846 /*
5847  * Potentially enlarge the window on an SSH-2 channel.
5848  */
5849 static void ssh2_set_window(struct ssh_channel *c, unsigned newwin)
5850 {
5851     Ssh ssh = c->ssh;
5852
5853     /*
5854      * Never send WINDOW_ADJUST for a channel that the remote side
5855      * already thinks it's closed; there's no point, since it won't
5856      * be sending any more data anyway.
5857      */
5858     if (c->closes != 0)
5859         return;
5860
5861     /*
5862      * Only send a WINDOW_ADJUST if there's significantly more window
5863      * available than the other end thinks there is.  This saves us
5864      * sending a WINDOW_ADJUST for every character in a shell session.
5865      *
5866      * "Significant" is arbitrarily defined as half the window size.
5867      */
5868     if (newwin > c->v.v2.locwindow * 2) {
5869         struct Packet *pktout;
5870
5871         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_WINDOW_ADJUST);
5872         ssh2_pkt_adduint32(pktout, c->remoteid);
5873         ssh2_pkt_adduint32(pktout, newwin - c->v.v2.locwindow);
5874         ssh2_pkt_send(ssh, pktout);
5875         c->v.v2.locwindow = newwin;
5876     }
5877 }
5878
5879 static void ssh2_msg_channel_window_adjust(Ssh ssh, struct Packet *pktin)
5880 {
5881     unsigned i = ssh_pkt_getuint32(pktin);
5882     struct ssh_channel *c;
5883     c = find234(ssh->channels, &i, ssh_channelfind);
5884     if (c && !c->closes) {
5885         c->v.v2.remwindow += ssh_pkt_getuint32(pktin);
5886         ssh2_try_send_and_unthrottle(c);
5887     }
5888 }
5889
5890 static void ssh2_msg_channel_data(Ssh ssh, struct Packet *pktin)
5891 {
5892     char *data;
5893     int length;
5894     unsigned i = ssh_pkt_getuint32(pktin);
5895     struct ssh_channel *c;
5896     c = find234(ssh->channels, &i, ssh_channelfind);
5897     if (!c)
5898         return;                        /* nonexistent channel */
5899     if (pktin->type == SSH2_MSG_CHANNEL_EXTENDED_DATA &&
5900         ssh_pkt_getuint32(pktin) != SSH2_EXTENDED_DATA_STDERR)
5901         return;                        /* extended but not stderr */
5902     ssh_pkt_getstring(pktin, &data, &length);
5903     if (data) {
5904         int bufsize = 0;
5905         c->v.v2.locwindow -= length;
5906         switch (c->type) {
5907           case CHAN_MAINSESSION:
5908             bufsize =
5909                 from_backend(ssh->frontend, pktin->type ==
5910                              SSH2_MSG_CHANNEL_EXTENDED_DATA,
5911                              data, length);
5912             break;
5913           case CHAN_X11:
5914             bufsize = x11_send(c->u.x11.s, data, length);
5915             break;
5916           case CHAN_SOCKDATA:
5917             bufsize = pfd_send(c->u.pfd.s, data, length);
5918             break;
5919           case CHAN_AGENT:
5920             while (length > 0) {
5921                 if (c->u.a.lensofar < 4) {
5922                     unsigned int l = min(4 - c->u.a.lensofar, length);
5923                     memcpy(c->u.a.msglen + c->u.a.lensofar,
5924                            data, l);
5925                     data += l;
5926                     length -= l;
5927                     c->u.a.lensofar += l;
5928                 }
5929                 if (c->u.a.lensofar == 4) {
5930                     c->u.a.totallen =
5931                         4 + GET_32BIT(c->u.a.msglen);
5932                     c->u.a.message = snewn(c->u.a.totallen,
5933                                            unsigned char);
5934                     memcpy(c->u.a.message, c->u.a.msglen, 4);
5935                 }
5936                 if (c->u.a.lensofar >= 4 && length > 0) {
5937                     unsigned int l =
5938                         min(c->u.a.totallen - c->u.a.lensofar,
5939                             length);
5940                     memcpy(c->u.a.message + c->u.a.lensofar,
5941                            data, l);
5942                     data += l;
5943                     length -= l;
5944                     c->u.a.lensofar += l;
5945                 }
5946                 if (c->u.a.lensofar == c->u.a.totallen) {
5947                     void *reply;
5948                     int replylen;
5949                     if (agent_query(c->u.a.message,
5950                                     c->u.a.totallen,
5951                                     &reply, &replylen,
5952                                     ssh_agentf_callback, c))
5953                         ssh_agentf_callback(c, reply, replylen);
5954                     sfree(c->u.a.message);
5955                     c->u.a.lensofar = 0;
5956                 }
5957             }
5958             bufsize = 0;
5959             break;
5960         }
5961         /*
5962          * If we are not buffering too much data,
5963          * enlarge the window again at the remote side.
5964          */
5965         if (bufsize < OUR_V2_WINSIZE)
5966             ssh2_set_window(c, OUR_V2_WINSIZE - bufsize);
5967     }
5968 }
5969
5970 static void ssh2_msg_channel_eof(Ssh ssh, struct Packet *pktin)
5971 {
5972     unsigned i = ssh_pkt_getuint32(pktin);
5973     struct ssh_channel *c;
5974
5975     c = find234(ssh->channels, &i, ssh_channelfind);
5976     if (!c)
5977         return;                        /* nonexistent channel */
5978
5979     if (c->type == CHAN_X11) {
5980         /*
5981          * Remote EOF on an X11 channel means we should
5982          * wrap up and close the channel ourselves.
5983          */
5984         x11_close(c->u.x11.s);
5985         sshfwd_close(c);
5986     } else if (c->type == CHAN_AGENT) {
5987         sshfwd_close(c);
5988     } else if (c->type == CHAN_SOCKDATA) {
5989         pfd_close(c->u.pfd.s);
5990         sshfwd_close(c);
5991     }
5992 }
5993
5994 static void ssh2_msg_channel_close(Ssh ssh, struct Packet *pktin)
5995 {
5996     unsigned i = ssh_pkt_getuint32(pktin);
5997     struct ssh_channel *c;
5998     struct Packet *pktout;
5999
6000     c = find234(ssh->channels, &i, ssh_channelfind);
6001     if (!c || c->halfopen) {
6002         bombout(("Received CHANNEL_CLOSE for %s channel %d\n",
6003                  c ? "half-open" : "nonexistent", i));
6004         return;
6005     }
6006     /* Do pre-close processing on the channel. */
6007     switch (c->type) {
6008       case CHAN_MAINSESSION:
6009         ssh->mainchan = NULL;
6010         update_specials_menu(ssh->frontend);
6011         break;
6012       case CHAN_X11:
6013         if (c->u.x11.s != NULL)
6014             x11_close(c->u.x11.s);
6015         sshfwd_close(c);
6016         break;
6017       case CHAN_AGENT:
6018         sshfwd_close(c);
6019         break;
6020       case CHAN_SOCKDATA:
6021         if (c->u.pfd.s != NULL)
6022             pfd_close(c->u.pfd.s);
6023         sshfwd_close(c);
6024         break;
6025     }
6026     if (c->closes == 0) {
6027         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_CLOSE);
6028         ssh2_pkt_adduint32(pktout, c->remoteid);
6029         ssh2_pkt_send(ssh, pktout);
6030     }
6031     del234(ssh->channels, c);
6032     bufchain_clear(&c->v.v2.outbuffer);
6033     sfree(c);
6034
6035     /*
6036      * See if that was the last channel left open.
6037      * (This is only our termination condition if we're
6038      * not running in -N mode.)
6039      */
6040     if (!ssh->cfg.ssh_no_shell && count234(ssh->channels) == 0) {
6041         /*
6042          * We used to send SSH_MSG_DISCONNECT here,
6043          * because I'd believed that _every_ conforming
6044          * SSH-2 connection had to end with a disconnect
6045          * being sent by at least one side; apparently
6046          * I was wrong and it's perfectly OK to
6047          * unceremoniously slam the connection shut
6048          * when you're done, and indeed OpenSSH feels
6049          * this is more polite than sending a
6050          * DISCONNECT. So now we don't.
6051          */
6052         ssh_disconnect(ssh, "All channels closed", NULL, 0, TRUE);
6053     }
6054 }
6055
6056 static void ssh2_msg_channel_open_confirmation(Ssh ssh, struct Packet *pktin)
6057 {
6058     unsigned i = ssh_pkt_getuint32(pktin);
6059     struct ssh_channel *c;
6060     struct Packet *pktout;
6061
6062     c = find234(ssh->channels, &i, ssh_channelfind);
6063     if (!c)
6064         return;                        /* nonexistent channel */
6065     if (c->type != CHAN_SOCKDATA_DORMANT)
6066         return;                        /* dunno why they're confirming this */
6067     c->remoteid = ssh_pkt_getuint32(pktin);
6068     c->halfopen = FALSE;
6069     c->type = CHAN_SOCKDATA;
6070     c->v.v2.remwindow = ssh_pkt_getuint32(pktin);
6071     c->v.v2.remmaxpkt = ssh_pkt_getuint32(pktin);
6072     if (c->u.pfd.s)
6073         pfd_confirm(c->u.pfd.s);
6074     if (c->closes) {
6075         /*
6076          * We have a pending close on this channel,
6077          * which we decided on before the server acked
6078          * the channel open. So now we know the
6079          * remoteid, we can close it again.
6080          */
6081         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_CLOSE);
6082         ssh2_pkt_adduint32(pktout, c->remoteid);
6083         ssh2_pkt_send(ssh, pktout);
6084     }
6085 }
6086
6087 static void ssh2_msg_channel_open_failure(Ssh ssh, struct Packet *pktin)
6088 {
6089     static const char *const reasons[] = {
6090         "<unknown reason code>",
6091             "Administratively prohibited",
6092             "Connect failed",
6093             "Unknown channel type",
6094             "Resource shortage",
6095     };
6096     unsigned i = ssh_pkt_getuint32(pktin);
6097     unsigned reason_code;
6098     char *reason_string;
6099     int reason_length;
6100     struct ssh_channel *c;
6101     c = find234(ssh->channels, &i, ssh_channelfind);
6102     if (!c)
6103         return;                        /* nonexistent channel */
6104     if (c->type != CHAN_SOCKDATA_DORMANT)
6105         return;                        /* dunno why they're failing this */
6106
6107     reason_code = ssh_pkt_getuint32(pktin);
6108     if (reason_code >= lenof(reasons))
6109         reason_code = 0; /* ensure reasons[reason_code] in range */
6110     ssh_pkt_getstring(pktin, &reason_string, &reason_length);
6111     logeventf(ssh, "Forwarded connection refused by server: %s [%.*s]",
6112               reasons[reason_code], reason_length, reason_string);
6113
6114     pfd_close(c->u.pfd.s);
6115
6116     del234(ssh->channels, c);
6117     sfree(c);
6118 }
6119
6120 static void ssh2_msg_channel_request(Ssh ssh, struct Packet *pktin)
6121 {
6122     unsigned localid;
6123     char *type;
6124     int typelen, want_reply;
6125     int reply = SSH2_MSG_CHANNEL_FAILURE; /* default */
6126     struct ssh_channel *c;
6127     struct Packet *pktout;
6128
6129     localid = ssh_pkt_getuint32(pktin);
6130     ssh_pkt_getstring(pktin, &type, &typelen);
6131     want_reply = ssh2_pkt_getbool(pktin);
6132
6133     /*
6134      * First, check that the channel exists. Otherwise,
6135      * we can instantly disconnect with a rude message.
6136      */
6137     c = find234(ssh->channels, &localid, ssh_channelfind);
6138     if (!c) {
6139         char *buf = dupprintf("Received channel request for nonexistent"
6140                               " channel %d", localid);
6141         ssh_disconnect(ssh, NULL, buf, SSH2_DISCONNECT_PROTOCOL_ERROR, FALSE);
6142         sfree(buf);
6143         return;
6144     }
6145
6146     /*
6147      * Having got the channel number, we now look at
6148      * the request type string to see if it's something
6149      * we recognise.
6150      */
6151     if (c == ssh->mainchan) {
6152         /*
6153          * We recognise "exit-status" and "exit-signal" on
6154          * the primary channel.
6155          */
6156         if (typelen == 11 &&
6157             !memcmp(type, "exit-status", 11)) {
6158
6159             ssh->exitcode = ssh_pkt_getuint32(pktin);
6160             logeventf(ssh, "Server sent command exit status %d",
6161                       ssh->exitcode);
6162             reply = SSH2_MSG_CHANNEL_SUCCESS;
6163
6164         } else if (typelen == 11 &&
6165                    !memcmp(type, "exit-signal", 11)) {
6166
6167             int is_plausible = TRUE, is_int = FALSE;
6168             char *fmt_sig = "", *fmt_msg = "";
6169             char *msg;
6170             int msglen = 0, core = FALSE;
6171             /* ICK: older versions of OpenSSH (e.g. 3.4p1)
6172              * provide an `int' for the signal, despite its
6173              * having been a `string' in the drafts since at
6174              * least 2001. (Fixed in session.c 1.147.) Try to
6175              * infer which we can safely parse it as. */
6176             {
6177                 unsigned char *p = pktin->body +
6178                     pktin->savedpos;
6179                 long len = pktin->length - pktin->savedpos;
6180                 unsigned long num = GET_32BIT(p); /* what is it? */
6181                 /* If it's 0, it hardly matters; assume string */
6182                 if (num == 0) {
6183                     is_int = FALSE;
6184                 } else {
6185                     int maybe_int = FALSE, maybe_str = FALSE;
6186 #define CHECK_HYPOTHESIS(offset, result) \
6187     do { \
6188         long q = offset; \
6189         if (q >= 0 && q+4 <= len) { \
6190             q = q + 4 + GET_32BIT(p+q); \
6191             if (q >= 0 && q+4 <= len && \
6192                     ((q = q + 4 + GET_32BIT(p+q))!= 0) && q == len) \
6193                 result = TRUE; \
6194         } \
6195     } while(0)
6196                     CHECK_HYPOTHESIS(4+1, maybe_int);
6197                     CHECK_HYPOTHESIS(4+num+1, maybe_str);
6198 #undef CHECK_HYPOTHESIS
6199                     if (maybe_int && !maybe_str)
6200                         is_int = TRUE;
6201                     else if (!maybe_int && maybe_str)
6202                         is_int = FALSE;
6203                     else
6204                         /* Crikey. Either or neither. Panic. */
6205                         is_plausible = FALSE;
6206                 }
6207             }
6208             if (is_plausible) {
6209                 if (is_int) {
6210                     /* Old non-standard OpenSSH. */
6211                     int signum = ssh_pkt_getuint32(pktin);
6212                     fmt_sig = dupprintf(" %d", signum);
6213                 } else {
6214                     /* As per the drafts. */
6215                     char *sig;
6216                     int siglen;
6217                     ssh_pkt_getstring(pktin, &sig, &siglen);
6218                     /* Signal name isn't supposed to be blank, but
6219                      * let's cope gracefully if it is. */
6220                     if (siglen) {
6221                         fmt_sig = dupprintf(" \"%.*s\"",
6222                                             siglen, sig);
6223                     }
6224                 }
6225                 core = ssh2_pkt_getbool(pktin);
6226                 ssh_pkt_getstring(pktin, &msg, &msglen);
6227                 if (msglen) {
6228                     fmt_msg = dupprintf(" (\"%.*s\")", msglen, msg);
6229                 }
6230                 /* ignore lang tag */
6231             } /* else don't attempt to parse */
6232             logeventf(ssh, "Server exited on signal%s%s%s",
6233                       fmt_sig, core ? " (core dumped)" : "",
6234                       fmt_msg);
6235             if (*fmt_sig) sfree(fmt_sig);
6236             if (*fmt_msg) sfree(fmt_msg);
6237             reply = SSH2_MSG_CHANNEL_SUCCESS;
6238
6239         }
6240     } else {
6241         /*
6242          * This is a channel request we don't know
6243          * about, so we now either ignore the request
6244          * or respond with CHANNEL_FAILURE, depending
6245          * on want_reply.
6246          */
6247         reply = SSH2_MSG_CHANNEL_FAILURE;
6248     }
6249     if (want_reply) {
6250         pktout = ssh2_pkt_init(reply);
6251         ssh2_pkt_adduint32(pktout, c->remoteid);
6252         ssh2_pkt_send(ssh, pktout);
6253     }
6254 }
6255
6256 static void ssh2_msg_global_request(Ssh ssh, struct Packet *pktin)
6257 {
6258     char *type;
6259     int typelen, want_reply;
6260     struct Packet *pktout;
6261
6262     ssh_pkt_getstring(pktin, &type, &typelen);
6263     want_reply = ssh2_pkt_getbool(pktin);
6264
6265     /*
6266      * We currently don't support any global requests
6267      * at all, so we either ignore the request or
6268      * respond with REQUEST_FAILURE, depending on
6269      * want_reply.
6270      */
6271     if (want_reply) {
6272         pktout = ssh2_pkt_init(SSH2_MSG_REQUEST_FAILURE);
6273         ssh2_pkt_send(ssh, pktout);
6274     }
6275 }
6276
6277 static void ssh2_msg_channel_open(Ssh ssh, struct Packet *pktin)
6278 {
6279     char *type;
6280     int typelen;
6281     char *peeraddr;
6282     int peeraddrlen;
6283     int peerport;
6284     char *error = NULL;
6285     struct ssh_channel *c;
6286     unsigned remid, winsize, pktsize;
6287     struct Packet *pktout;
6288
6289     ssh_pkt_getstring(pktin, &type, &typelen);
6290     c = snew(struct ssh_channel);
6291     c->ssh = ssh;
6292
6293     remid = ssh_pkt_getuint32(pktin);
6294     winsize = ssh_pkt_getuint32(pktin);
6295     pktsize = ssh_pkt_getuint32(pktin);
6296
6297     if (typelen == 3 && !memcmp(type, "x11", 3)) {
6298         char *addrstr;
6299
6300         ssh_pkt_getstring(pktin, &peeraddr, &peeraddrlen);
6301         addrstr = snewn(peeraddrlen+1, char);
6302         memcpy(addrstr, peeraddr, peeraddrlen);
6303         addrstr[peeraddrlen] = '\0';
6304         peerport = ssh_pkt_getuint32(pktin);
6305
6306         logeventf(ssh, "Received X11 connect request from %s:%d",
6307                   addrstr, peerport);
6308
6309         if (!ssh->X11_fwd_enabled)
6310             error = "X11 forwarding is not enabled";
6311         else if (x11_init(&c->u.x11.s, ssh->cfg.x11_display, c,
6312                           ssh->x11auth, addrstr, peerport,
6313                           &ssh->cfg) != NULL) {
6314             error = "Unable to open an X11 connection";
6315         } else {
6316             logevent("Opening X11 forward connection succeeded");
6317             c->type = CHAN_X11;
6318         }
6319
6320         sfree(addrstr);
6321     } else if (typelen == 15 &&
6322                !memcmp(type, "forwarded-tcpip", 15)) {
6323         struct ssh_rportfwd pf, *realpf;
6324         char *dummy;
6325         int dummylen;
6326         ssh_pkt_getstring(pktin, &dummy, &dummylen);/* skip address */
6327         pf.sport = ssh_pkt_getuint32(pktin);
6328         ssh_pkt_getstring(pktin, &peeraddr, &peeraddrlen);
6329         peerport = ssh_pkt_getuint32(pktin);
6330         realpf = find234(ssh->rportfwds, &pf, NULL);
6331         logeventf(ssh, "Received remote port %d open request "
6332                   "from %s:%d", pf.sport, peeraddr, peerport);
6333         if (realpf == NULL) {
6334             error = "Remote port is not recognised";
6335         } else {
6336             const char *e = pfd_newconnect(&c->u.pfd.s,
6337                                            realpf->dhost,
6338                                            realpf->dport, c,
6339                                            &ssh->cfg,
6340                                            realpf->pfrec->addressfamily);
6341             logeventf(ssh, "Attempting to forward remote port to "
6342                       "%s:%d", realpf->dhost, realpf->dport);
6343             if (e != NULL) {
6344                 logeventf(ssh, "Port open failed: %s", e);
6345                 error = "Port open failed";
6346             } else {
6347                 logevent("Forwarded port opened successfully");
6348                 c->type = CHAN_SOCKDATA;
6349             }
6350         }
6351     } else if (typelen == 22 &&
6352                !memcmp(type, "auth-agent@openssh.com", 22)) {
6353         if (!ssh->agentfwd_enabled)
6354             error = "Agent forwarding is not enabled";
6355         else {
6356             c->type = CHAN_AGENT;       /* identify channel type */
6357             c->u.a.lensofar = 0;
6358         }
6359     } else {
6360         error = "Unsupported channel type requested";
6361     }
6362
6363     c->remoteid = remid;
6364     c->halfopen = FALSE;
6365     if (error) {
6366         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN_FAILURE);
6367         ssh2_pkt_adduint32(pktout, c->remoteid);
6368         ssh2_pkt_adduint32(pktout, SSH2_OPEN_CONNECT_FAILED);
6369         ssh2_pkt_addstring(pktout, error);
6370         ssh2_pkt_addstring(pktout, "en");       /* language tag */
6371         ssh2_pkt_send(ssh, pktout);
6372         logeventf(ssh, "Rejected channel open: %s", error);
6373         sfree(c);
6374     } else {
6375         c->localid = alloc_channel_id(ssh);
6376         c->closes = 0;
6377         c->v.v2.locwindow = OUR_V2_WINSIZE;
6378         c->v.v2.remwindow = winsize;
6379         c->v.v2.remmaxpkt = pktsize;
6380         bufchain_init(&c->v.v2.outbuffer);
6381         add234(ssh->channels, c);
6382         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION);
6383         ssh2_pkt_adduint32(pktout, c->remoteid);
6384         ssh2_pkt_adduint32(pktout, c->localid);
6385         ssh2_pkt_adduint32(pktout, c->v.v2.locwindow);
6386         ssh2_pkt_adduint32(pktout, OUR_V2_MAXPKT);      /* our max pkt size */
6387         ssh2_pkt_send(ssh, pktout);
6388     }
6389 }
6390
6391 /*
6392  * Buffer banner messages for later display at some convenient point.
6393  */
6394 static void ssh2_msg_userauth_banner(Ssh ssh, struct Packet *pktin)
6395 {
6396     /* Arbitrary limit to prevent unbounded inflation of buffer */
6397     if (bufchain_size(&ssh->banner) <= 131072) {
6398         char *banner = NULL;
6399         int size = 0;
6400         ssh_pkt_getstring(pktin, &banner, &size);
6401         if (banner)
6402             bufchain_add(&ssh->banner, banner, size);
6403     }
6404 }
6405
6406 /* Helper function to deal with sending tty modes for "pty-req" */
6407 static void ssh2_send_ttymode(void *data, char *mode, char *val)
6408 {
6409     struct Packet *pktout = (struct Packet *)data;
6410     int i = 0;
6411     unsigned int arg = 0;
6412     while (strcmp(mode, ssh_ttymodes[i].mode) != 0) i++;
6413     if (i == lenof(ssh_ttymodes)) return;
6414     switch (ssh_ttymodes[i].type) {
6415       case TTY_OP_CHAR:
6416         arg = ssh_tty_parse_specchar(val);
6417         break;
6418       case TTY_OP_BOOL:
6419         arg = ssh_tty_parse_boolean(val);
6420         break;
6421     }
6422     ssh2_pkt_addbyte(pktout, ssh_ttymodes[i].opcode);
6423     ssh2_pkt_adduint32(pktout, arg);
6424 }
6425
6426 /*
6427  * Handle the SSH-2 userauth and connection layers.
6428  */
6429 static void do_ssh2_authconn(Ssh ssh, unsigned char *in, int inlen,
6430                              struct Packet *pktin)
6431 {
6432     struct do_ssh2_authconn_state {
6433         enum {
6434             AUTH_INVALID, AUTH_PUBLICKEY_AGENT, AUTH_PUBLICKEY_FILE,
6435                 AUTH_PASSWORD,
6436                 AUTH_KEYBOARD_INTERACTIVE
6437         } method;
6438         enum {
6439             AUTH_TYPE_NONE,
6440                 AUTH_TYPE_PUBLICKEY,
6441                 AUTH_TYPE_PUBLICKEY_OFFER_LOUD,
6442                 AUTH_TYPE_PUBLICKEY_OFFER_QUIET,
6443                 AUTH_TYPE_PASSWORD,
6444                 AUTH_TYPE_KEYBOARD_INTERACTIVE,
6445                 AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET
6446         } type;
6447         int done_service_req;
6448         int gotit, need_pw, can_pubkey, can_passwd, can_keyb_inter;
6449         int tried_pubkey_config, tried_agent;
6450         int kbd_inter_running, kbd_inter_refused;
6451         int we_are_in;
6452         int num_prompts, curr_prompt, echo;
6453         char username[100];
6454         int got_username;
6455         char pwprompt[512];
6456         char password[100];
6457         void *publickey_blob;
6458         int publickey_bloblen;
6459         unsigned char request[5], *response, *p;
6460         int responselen;
6461         int keyi, nkeys;
6462         int authed;
6463         char *pkblob, *alg, *commentp;
6464         int pklen, alglen, commentlen;
6465         int siglen, retlen, len;
6466         char *q, *agentreq, *ret;
6467         int try_send;
6468         int num_env, env_left, env_ok;
6469         struct Packet *pktout;
6470     };
6471     crState(do_ssh2_authconn_state);
6472
6473     crBegin(ssh->do_ssh2_authconn_crstate);
6474
6475     s->done_service_req = FALSE;
6476     s->we_are_in = FALSE;
6477     if (!ssh->cfg.ssh_no_userauth) {
6478         /*
6479          * Request userauth protocol, and await a response to it.
6480          */
6481         s->pktout = ssh2_pkt_init(SSH2_MSG_SERVICE_REQUEST);
6482         ssh2_pkt_addstring(s->pktout, "ssh-userauth");
6483         ssh2_pkt_send(ssh, s->pktout);
6484         crWaitUntilV(pktin);
6485         if (pktin->type == SSH2_MSG_SERVICE_ACCEPT)
6486             s->done_service_req = TRUE;
6487     }
6488     if (!s->done_service_req) {
6489         /*
6490          * Request connection protocol directly, without authentication.
6491          */
6492         s->pktout = ssh2_pkt_init(SSH2_MSG_SERVICE_REQUEST);
6493         ssh2_pkt_addstring(s->pktout, "ssh-connection");
6494         ssh2_pkt_send(ssh, s->pktout);
6495         crWaitUntilV(pktin);
6496         if (pktin->type == SSH2_MSG_SERVICE_ACCEPT) {
6497             s->we_are_in = TRUE; /* no auth required */
6498         } else {
6499             bombout(("Server refused service request"));
6500             crStopV;
6501         }
6502     }
6503
6504     /*
6505      * We repeat this whole loop, including the username prompt,
6506      * until we manage a successful authentication. If the user
6507      * types the wrong _password_, they can be sent back to the
6508      * beginning to try another username, if this is configured on.
6509      * (If they specify a username in the config, they are never
6510      * asked, even if they do give a wrong password.)
6511      * 
6512      * I think this best serves the needs of
6513      * 
6514      *  - the people who have no configuration, no keys, and just
6515      *    want to try repeated (username,password) pairs until they
6516      *    type both correctly
6517      * 
6518      *  - people who have keys and configuration but occasionally
6519      *    need to fall back to passwords
6520      * 
6521      *  - people with a key held in Pageant, who might not have
6522      *    logged in to a particular machine before; so they want to
6523      *    type a username, and then _either_ their key will be
6524      *    accepted, _or_ they will type a password. If they mistype
6525      *    the username they will want to be able to get back and
6526      *    retype it!
6527      */
6528     s->username[0] = '\0';
6529     s->got_username = FALSE;
6530     bufchain_init(&ssh->banner);
6531     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] =
6532         ssh2_msg_userauth_banner;
6533     while (!s->we_are_in) {
6534         /*
6535          * Get a username.
6536          */
6537         if (s->got_username && !ssh->cfg.change_username) {
6538             /*
6539              * We got a username last time round this loop, and
6540              * with change_username turned off we don't try to get
6541              * it again.
6542              */
6543         } else if (!*ssh->cfg.username) {
6544             if (ssh_get_line && !ssh_getline_pw_only) {
6545                 if (!ssh_get_line("login as: ",
6546                                   s->username, sizeof(s->username), FALSE)) {
6547                     /*
6548                      * get_line failed to get a username.
6549                      * Terminate.
6550                      */
6551                     ssh_disconnect(ssh, "No username provided", NULL, 0, TRUE);
6552                     crStopV;
6553                 }
6554             } else {
6555                 int ret;               /* need not be saved across crReturn */
6556                 c_write_str(ssh, "login as: ");
6557                 ssh->send_ok = 1;
6558                 setup_userpass_input(ssh, s->username, sizeof(s->username), 1);
6559                 do {
6560                     crWaitUntilV(!pktin);
6561                     ret = process_userpass_input(ssh, in, inlen);
6562                 } while (ret == 0);
6563                 if (ret < 0)
6564                     cleanup_exit(0);
6565                 c_write_str(ssh, "\r\n");
6566             }
6567             s->username[strcspn(s->username, "\n\r")] = '\0';
6568         } else {
6569             char *stuff;
6570             strncpy(s->username, ssh->cfg.username, sizeof(s->username));
6571             s->username[sizeof(s->username)-1] = '\0';
6572             if ((flags & FLAG_VERBOSE) || (flags & FLAG_INTERACTIVE)) {
6573                 stuff = dupprintf("Using username \"%s\".\r\n", s->username);
6574                 c_write_str(ssh, stuff);
6575                 sfree(stuff);
6576             }
6577         }
6578         s->got_username = TRUE;
6579
6580         /*
6581          * Send an authentication request using method "none": (a)
6582          * just in case it succeeds, and (b) so that we know what
6583          * authentication methods we can usefully try next.
6584          */
6585         ssh->pkt_ctx &= ~SSH2_PKTCTX_AUTH_MASK;
6586
6587         s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
6588         ssh2_pkt_addstring(s->pktout, s->username);
6589         ssh2_pkt_addstring(s->pktout, "ssh-connection");/* service requested */
6590         ssh2_pkt_addstring(s->pktout, "none");    /* method */
6591         ssh2_pkt_send(ssh, s->pktout);
6592         s->type = AUTH_TYPE_NONE;
6593         s->gotit = FALSE;
6594         s->we_are_in = FALSE;
6595
6596         s->tried_pubkey_config = FALSE;
6597         s->tried_agent = FALSE;
6598         s->kbd_inter_running = FALSE;
6599         s->kbd_inter_refused = FALSE;
6600         /* Load the pub half of ssh->cfg.keyfile so we notice if it's in Pageant */
6601         if (!filename_is_null(ssh->cfg.keyfile)) {
6602             int keytype;
6603             logeventf(ssh, "Reading private key file \"%.150s\"",
6604                       filename_to_str(&ssh->cfg.keyfile));
6605             keytype = key_type(&ssh->cfg.keyfile);
6606             if (keytype == SSH_KEYTYPE_SSH2) {
6607                 s->publickey_blob =
6608                     ssh2_userkey_loadpub(&ssh->cfg.keyfile, NULL,
6609                                          &s->publickey_bloblen, NULL);
6610             } else {
6611                 char *msgbuf;
6612                 logeventf(ssh, "Unable to use this key file (%s)",
6613                           key_type_to_str(keytype));
6614                 msgbuf = dupprintf("Unable to use key file \"%.150s\""
6615                                    " (%s)\r\n",
6616                                    filename_to_str(&ssh->cfg.keyfile),
6617                                    key_type_to_str(keytype));
6618                 c_write_str(ssh, msgbuf);
6619                 sfree(msgbuf);
6620                 s->publickey_blob = NULL;
6621             }
6622         } else
6623             s->publickey_blob = NULL;
6624
6625         while (1) {
6626             /*
6627              * Wait for the result of the last authentication request.
6628              */
6629             if (!s->gotit)
6630                 crWaitUntilV(pktin);
6631             /*
6632              * Now is a convenient point to spew any banner material
6633              * that we've accumulated. (This should ensure that when
6634              * we exit the auth loop, we haven't any left to deal
6635              * with.)
6636              */
6637             {
6638                 int size = bufchain_size(&ssh->banner);
6639                 /*
6640                  * Don't show the banner if we're operating in
6641                  * non-verbose non-interactive mode. (It's probably
6642                  * a script, which means nobody will read the
6643                  * banner _anyway_, and moreover the printing of
6644                  * the banner will screw up processing on the
6645                  * output of (say) plink.)
6646                  */
6647                 if (size && (flags & (FLAG_VERBOSE | FLAG_INTERACTIVE))) {
6648                     char *banner = snewn(size, char);
6649                     bufchain_fetch(&ssh->banner, banner, size);
6650                     c_write_untrusted(ssh, banner, size);
6651                     sfree(banner);
6652                 }
6653                 bufchain_clear(&ssh->banner);
6654             }
6655             if (pktin->type == SSH2_MSG_USERAUTH_SUCCESS) {
6656                 logevent("Access granted");
6657                 s->we_are_in = TRUE;
6658                 break;
6659             }
6660
6661             if (s->kbd_inter_running &&
6662                 pktin->type == SSH2_MSG_USERAUTH_INFO_REQUEST) {
6663                 /*
6664                  * This is either a further set-of-prompts packet
6665                  * in keyboard-interactive authentication, or it's
6666                  * the same one and we came back here with `gotit'
6667                  * set. In the former case, we must reset the
6668                  * curr_prompt variable.
6669                  */
6670                 if (!s->gotit)
6671                     s->curr_prompt = 0;
6672             } else if (pktin->type == SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ) {
6673                 /* FIXME: perhaps we should support this? */
6674                 bombout(("PASSWD_CHANGEREQ not yet supported"));
6675                 crStopV;
6676             } else if (pktin->type != SSH2_MSG_USERAUTH_FAILURE) {
6677                 bombout(("Strange packet received during authentication: type %d",
6678                          pktin->type));
6679                 crStopV;
6680             }
6681
6682             s->gotit = FALSE;
6683
6684             /*
6685              * OK, we're now sitting on a USERAUTH_FAILURE message, so
6686              * we can look at the string in it and know what we can
6687              * helpfully try next.
6688              */
6689             if (pktin->type == SSH2_MSG_USERAUTH_FAILURE) {
6690                 char *methods;
6691                 int methlen;
6692                 ssh_pkt_getstring(pktin, &methods, &methlen);
6693                 s->kbd_inter_running = FALSE;
6694                 if (!ssh2_pkt_getbool(pktin)) {
6695                     /*
6696                      * We have received an unequivocal Access
6697                      * Denied. This can translate to a variety of
6698                      * messages:
6699                      * 
6700                      *  - if we'd just tried "none" authentication,
6701                      *    it's not worth printing anything at all
6702                      * 
6703                      *  - if we'd just tried a public key _offer_,
6704                      *    the message should be "Server refused our
6705                      *    key" (or no message at all if the key
6706                      *    came from Pageant)
6707                      * 
6708                      *  - if we'd just tried anything else, the
6709                      *    message really should be "Access denied".
6710                      * 
6711                      * Additionally, if we'd just tried password
6712                      * authentication, we should break out of this
6713                      * whole loop so as to go back to the username
6714                      * prompt (iff we're configured to allow
6715                      * username change attempts).
6716                      */
6717                     if (s->type == AUTH_TYPE_NONE) {
6718                         /* do nothing */
6719                     } else if (s->type == AUTH_TYPE_PUBLICKEY_OFFER_LOUD ||
6720                                s->type == AUTH_TYPE_PUBLICKEY_OFFER_QUIET) {
6721                         if (s->type == AUTH_TYPE_PUBLICKEY_OFFER_LOUD)
6722                             c_write_str(ssh, "Server refused our key\r\n");
6723                         logevent("Server refused public key");
6724                     } else if (s->type==AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET) {
6725                         /* server declined keyboard-interactive; ignore */
6726                     } else {
6727                         c_write_str(ssh, "Access denied\r\n");
6728                         logevent("Access denied");
6729                         if (s->type == AUTH_TYPE_PASSWORD &&
6730                             ssh->cfg.change_username) {
6731                             /* XXX perhaps we should allow
6732                              * keyboard-interactive to do this too? */
6733                             s->we_are_in = FALSE;
6734                             break;
6735                         }
6736                     }
6737                 } else {
6738                     c_write_str(ssh, "Further authentication required\r\n");
6739                     logevent("Further authentication required");
6740                 }
6741
6742                 s->can_pubkey =
6743                     in_commasep_string("publickey", methods, methlen);
6744                 s->can_passwd =
6745                     in_commasep_string("password", methods, methlen);
6746                 s->can_keyb_inter = ssh->cfg.try_ki_auth &&
6747                     in_commasep_string("keyboard-interactive", methods, methlen);
6748             }
6749
6750             s->method = 0;
6751             ssh->pkt_ctx &= ~SSH2_PKTCTX_AUTH_MASK;
6752             s->need_pw = FALSE;
6753
6754             /*
6755              * Most password/passphrase prompts will be
6756              * non-echoing, so we set this to 0 by default.
6757              * Exception is that some keyboard-interactive prompts
6758              * can be echoing, in which case we'll set this to 1.
6759              */
6760             s->echo = 0;
6761
6762             if (!s->method && s->can_pubkey &&
6763                 agent_exists() && !s->tried_agent) {
6764                 /*
6765                  * Attempt public-key authentication using Pageant.
6766                  */
6767                 void *r;
6768                 s->authed = FALSE;
6769
6770                 ssh->pkt_ctx &= ~SSH2_PKTCTX_AUTH_MASK;
6771                 ssh->pkt_ctx |= SSH2_PKTCTX_PUBLICKEY;
6772
6773                 s->tried_agent = TRUE;
6774
6775                 logevent("Pageant is running. Requesting keys.");
6776
6777                 /* Request the keys held by the agent. */
6778                 PUT_32BIT(s->request, 1);
6779                 s->request[4] = SSH2_AGENTC_REQUEST_IDENTITIES;
6780                 if (!agent_query(s->request, 5, &r, &s->responselen,
6781                                  ssh_agent_callback, ssh)) {
6782                     do {
6783                         crReturnV;
6784                         if (pktin) {
6785                             bombout(("Unexpected data from server while"
6786                                      " waiting for agent response"));
6787                             crStopV;
6788                         }
6789                     } while (pktin || inlen > 0);
6790                     r = ssh->agent_response;
6791                     s->responselen = ssh->agent_response_len;
6792                 }
6793                 s->response = (unsigned char *) r;
6794                 if (s->response && s->responselen >= 5 &&
6795                     s->response[4] == SSH2_AGENT_IDENTITIES_ANSWER) {
6796                     s->p = s->response + 5;
6797                     s->nkeys = GET_32BIT(s->p);
6798                     s->p += 4;
6799                     logeventf(ssh, "Pageant has %d SSH-2 keys", s->nkeys);
6800                     for (s->keyi = 0; s->keyi < s->nkeys; s->keyi++) {
6801                         void *vret;
6802
6803                         logeventf(ssh, "Trying Pageant key #%d", s->keyi);
6804                         s->pklen = GET_32BIT(s->p);
6805                         s->p += 4;
6806                         if (s->publickey_blob &&
6807                             s->pklen == s->publickey_bloblen &&
6808                             !memcmp(s->p, s->publickey_blob,
6809                                     s->publickey_bloblen)) {
6810                             logevent("This key matches configured key file");
6811                             s->tried_pubkey_config = 1;
6812                         }
6813                         s->pkblob = (char *)s->p;
6814                         s->p += s->pklen;
6815                         s->alglen = GET_32BIT(s->pkblob);
6816                         s->alg = s->pkblob + 4;
6817                         s->commentlen = GET_32BIT(s->p);
6818                         s->p += 4;
6819                         s->commentp = (char *)s->p;
6820                         s->p += s->commentlen;
6821                         s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
6822                         ssh2_pkt_addstring(s->pktout, s->username);
6823                         ssh2_pkt_addstring(s->pktout, "ssh-connection");        /* service requested */
6824                         ssh2_pkt_addstring(s->pktout, "publickey");     /* method */
6825                         ssh2_pkt_addbool(s->pktout, FALSE);     /* no signature included */
6826                         ssh2_pkt_addstring_start(s->pktout);
6827                         ssh2_pkt_addstring_data(s->pktout, s->alg, s->alglen);
6828                         ssh2_pkt_addstring_start(s->pktout);
6829                         ssh2_pkt_addstring_data(s->pktout, s->pkblob, s->pklen);
6830                         ssh2_pkt_send(ssh, s->pktout);
6831
6832                         crWaitUntilV(pktin);
6833                         if (pktin->type != SSH2_MSG_USERAUTH_PK_OK) {
6834                             logevent("Key refused");
6835                             continue;
6836                         }
6837
6838                         if (flags & FLAG_VERBOSE) {
6839                             c_write_str(ssh, "Authenticating with "
6840                                         "public key \"");
6841                             c_write(ssh, s->commentp, s->commentlen);
6842                             c_write_str(ssh, "\" from agent\r\n");
6843                         }
6844
6845                         /*
6846                          * Server is willing to accept the key.
6847                          * Construct a SIGN_REQUEST.
6848                          */
6849                         s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
6850                         ssh2_pkt_addstring(s->pktout, s->username);
6851                         ssh2_pkt_addstring(s->pktout, "ssh-connection");        /* service requested */
6852                         ssh2_pkt_addstring(s->pktout, "publickey");     /* method */
6853                         ssh2_pkt_addbool(s->pktout, TRUE);
6854                         ssh2_pkt_addstring_start(s->pktout);
6855                         ssh2_pkt_addstring_data(s->pktout, s->alg, s->alglen);
6856                         ssh2_pkt_addstring_start(s->pktout);
6857                         ssh2_pkt_addstring_data(s->pktout, s->pkblob, s->pklen);
6858
6859                         s->siglen = s->pktout->length - 5 + 4 + 20;
6860                         if (ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)
6861                             s->siglen -= 4;
6862                         s->len = 1;       /* message type */
6863                         s->len += 4 + s->pklen; /* key blob */
6864                         s->len += 4 + s->siglen;        /* data to sign */
6865                         s->len += 4;      /* flags */
6866                         s->agentreq = snewn(4 + s->len, char);
6867                         PUT_32BIT(s->agentreq, s->len);
6868                         s->q = s->agentreq + 4;
6869                         *s->q++ = SSH2_AGENTC_SIGN_REQUEST;
6870                         PUT_32BIT(s->q, s->pklen);
6871                         s->q += 4;
6872                         memcpy(s->q, s->pkblob, s->pklen);
6873                         s->q += s->pklen;
6874                         PUT_32BIT(s->q, s->siglen);
6875                         s->q += 4;
6876                         /* Now the data to be signed... */
6877                         if (!(ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)) {
6878                             PUT_32BIT(s->q, 20);
6879                             s->q += 4;
6880                         }
6881                         memcpy(s->q, ssh->v2_session_id, 20);
6882                         s->q += 20;
6883                         memcpy(s->q, s->pktout->data + 5,
6884                                s->pktout->length - 5);
6885                         s->q += s->pktout->length - 5;
6886                         /* And finally the (zero) flags word. */
6887                         PUT_32BIT(s->q, 0);
6888                         if (!agent_query(s->agentreq, s->len + 4,
6889                                          &vret, &s->retlen,
6890                                          ssh_agent_callback, ssh)) {
6891                             do {
6892                                 crReturnV;
6893                                 if (pktin) {
6894                                     bombout(("Unexpected data from server"
6895                                              " while waiting for agent"
6896                                              " response"));
6897                                     crStopV;
6898                                 }
6899                             } while (pktin || inlen > 0);
6900                             vret = ssh->agent_response;
6901                             s->retlen = ssh->agent_response_len;
6902                         }
6903                         s->ret = vret;
6904                         sfree(s->agentreq);
6905                         if (s->ret) {
6906                             if (s->ret[4] == SSH2_AGENT_SIGN_RESPONSE) {
6907                                 logevent("Sending Pageant's response");
6908                                 ssh2_add_sigblob(ssh, s->pktout,
6909                                                  s->pkblob, s->pklen,
6910                                                  s->ret + 9,
6911                                                  GET_32BIT(s->ret + 5));
6912                                 ssh2_pkt_send(ssh, s->pktout);
6913                                 s->authed = TRUE;
6914                                 break;
6915                             } else {
6916                                 logevent
6917                                     ("Pageant failed to answer challenge");
6918                                 sfree(s->ret);
6919                             }
6920                         }
6921                     }
6922                     if (s->authed)
6923                         continue;
6924                 }
6925                 sfree(s->response);
6926             }
6927
6928             if (!s->method && s->can_pubkey && s->publickey_blob
6929                 && !s->tried_pubkey_config) {
6930                 unsigned char *pub_blob;
6931                 char *algorithm, *comment;
6932                 int pub_blob_len;
6933
6934                 s->tried_pubkey_config = TRUE;
6935
6936                 ssh->pkt_ctx &= ~SSH2_PKTCTX_AUTH_MASK;
6937                 ssh->pkt_ctx |= SSH2_PKTCTX_PUBLICKEY;
6938
6939                 /*
6940                  * Try the public key supplied in the configuration.
6941                  *
6942                  * First, offer the public blob to see if the server is
6943                  * willing to accept it.
6944                  */
6945                 pub_blob =
6946                     (unsigned char *)ssh2_userkey_loadpub(&ssh->cfg.keyfile,
6947                                                           &algorithm,
6948                                                           &pub_blob_len,
6949                                                           NULL);
6950                 if (pub_blob) {
6951                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
6952                     ssh2_pkt_addstring(s->pktout, s->username);
6953                     ssh2_pkt_addstring(s->pktout, "ssh-connection");    /* service requested */
6954                     ssh2_pkt_addstring(s->pktout, "publickey"); /* method */
6955                     ssh2_pkt_addbool(s->pktout, FALSE); /* no signature included */
6956                     ssh2_pkt_addstring(s->pktout, algorithm);
6957                     ssh2_pkt_addstring_start(s->pktout);
6958                     ssh2_pkt_addstring_data(s->pktout, (char *)pub_blob,
6959                                             pub_blob_len);
6960                     ssh2_pkt_send(ssh, s->pktout);
6961                     logevent("Offered public key");
6962
6963                     crWaitUntilV(pktin);
6964                     if (pktin->type != SSH2_MSG_USERAUTH_PK_OK) {
6965                         s->gotit = TRUE;
6966                         s->type = AUTH_TYPE_PUBLICKEY_OFFER_LOUD;
6967                         continue;      /* key refused; give up on it */
6968                     }
6969
6970                     logevent("Offer of public key accepted");
6971                     /*
6972                      * Actually attempt a serious authentication using
6973                      * the key.
6974                      */
6975                     if (ssh2_userkey_encrypted(&ssh->cfg.keyfile, &comment)) {
6976                         sprintf(s->pwprompt,
6977                                 "Passphrase for key \"%.100s\": ",
6978                                 comment);
6979                         s->need_pw = TRUE;
6980                     } else {
6981                         s->need_pw = FALSE;
6982                     }
6983                     if (flags & FLAG_VERBOSE) {
6984                         c_write_str(ssh, "Authenticating with public key \"");
6985                         c_write_str(ssh, comment);
6986                         c_write_str(ssh, "\"\r\n");
6987                     }
6988                     s->method = AUTH_PUBLICKEY_FILE;
6989                 }
6990             }
6991
6992             if (!s->method && s->can_keyb_inter && !s->kbd_inter_refused &&
6993                 !s->kbd_inter_running) {
6994                 s->method = AUTH_KEYBOARD_INTERACTIVE;
6995                 s->type = AUTH_TYPE_KEYBOARD_INTERACTIVE;
6996
6997                 ssh->pkt_ctx &= ~SSH2_PKTCTX_AUTH_MASK;
6998                 ssh->pkt_ctx |= SSH2_PKTCTX_KBDINTER;
6999
7000                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
7001                 ssh2_pkt_addstring(s->pktout, s->username);
7002                 ssh2_pkt_addstring(s->pktout, "ssh-connection");        /* service requested */
7003                 ssh2_pkt_addstring(s->pktout, "keyboard-interactive");  /* method */
7004                 ssh2_pkt_addstring(s->pktout, ""); /* lang */
7005                 ssh2_pkt_addstring(s->pktout, "");
7006                 ssh2_pkt_send(ssh, s->pktout);
7007
7008                 crWaitUntilV(pktin);
7009                 if (pktin->type != SSH2_MSG_USERAUTH_INFO_REQUEST) {
7010                     if (pktin->type == SSH2_MSG_USERAUTH_FAILURE)
7011                         s->gotit = TRUE;
7012                     logevent("Keyboard-interactive authentication refused");
7013                     s->type = AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET;
7014                     s->kbd_inter_refused = TRUE; /* don't try it again */
7015                     continue;
7016                 }
7017
7018                 c_write_str(ssh, "Using keyboard-interactive authentication.\r\n");
7019                 s->kbd_inter_running = TRUE;
7020                 s->curr_prompt = 0;
7021             }
7022
7023             if (s->kbd_inter_running) {
7024                 s->method = AUTH_KEYBOARD_INTERACTIVE;
7025                 s->type = AUTH_TYPE_KEYBOARD_INTERACTIVE;
7026
7027                 ssh->pkt_ctx &= ~SSH2_PKTCTX_AUTH_MASK;
7028                 ssh->pkt_ctx |= SSH2_PKTCTX_KBDINTER;
7029
7030                 if (s->curr_prompt == 0) {
7031                     /*
7032                      * We've got a fresh USERAUTH_INFO_REQUEST.
7033                      * Display header data, and start going through
7034                      * the prompts.
7035                      */
7036                     char *name, *inst, *lang;
7037                     int name_len, inst_len, lang_len;
7038
7039                     ssh_pkt_getstring(pktin, &name, &name_len);
7040                     ssh_pkt_getstring(pktin, &inst, &inst_len);
7041                     ssh_pkt_getstring(pktin, &lang, &lang_len);
7042                     if (name_len > 0) {
7043                         c_write_untrusted(ssh, name, name_len);
7044                         c_write_str(ssh, "\r\n");
7045                     }
7046                     if (inst_len > 0) {
7047                         c_write_untrusted(ssh, inst, inst_len);
7048                         c_write_str(ssh, "\r\n");
7049                     }
7050                     s->num_prompts = ssh_pkt_getuint32(pktin);
7051                 }
7052
7053                 /*
7054                  * If there are prompts remaining in the packet,
7055                  * display one and get a response.
7056                  */
7057                 if (s->curr_prompt < s->num_prompts) {
7058                     char *prompt;
7059                     int prompt_len;
7060
7061                     ssh_pkt_getstring(pktin, &prompt, &prompt_len);
7062                     if (prompt_len > 0) {
7063                         static const char trunc[] = "<prompt truncated>: ";
7064                         static const int prlen = sizeof(s->pwprompt) -
7065                                                  lenof(trunc);
7066                         if (prompt_len > prlen) {
7067                             memcpy(s->pwprompt, prompt, prlen);
7068                             strcpy(s->pwprompt + prlen, trunc);
7069                         } else {
7070                             memcpy(s->pwprompt, prompt, prompt_len);
7071                             s->pwprompt[prompt_len] = '\0';
7072                         }
7073                     } else {
7074                         strcpy(s->pwprompt,
7075                                "<server failed to send prompt>: ");
7076                     }
7077                     s->echo = ssh2_pkt_getbool(pktin);
7078                     s->need_pw = TRUE;
7079                 } else
7080                     s->need_pw = FALSE;
7081             }
7082
7083             if (!s->method && s->can_passwd) {
7084                 s->method = AUTH_PASSWORD;
7085                 ssh->pkt_ctx &= ~SSH2_PKTCTX_AUTH_MASK;
7086                 ssh->pkt_ctx |= SSH2_PKTCTX_PASSWORD;
7087                 sprintf(s->pwprompt, "%.90s@%.90s's password: ", s->username,
7088                         ssh->savedhost);
7089                 s->need_pw = TRUE;
7090             }
7091
7092             if (s->need_pw) {
7093                 if (ssh_get_line) {
7094                     if (!ssh_get_line(s->pwprompt, s->password,
7095                                       sizeof(s->password), TRUE)) {
7096                         /*
7097                          * get_line failed to get a password (for
7098                          * example because one was supplied on the
7099                          * command line which has already failed to
7100                          * work). Terminate.
7101                          */
7102                         ssh_disconnect(ssh, NULL, "Unable to authenticate",
7103                                        SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER,
7104                                        FALSE);
7105                         crStopV;
7106                     }
7107                 } else {
7108                     int ret;           /* need not be saved across crReturn */
7109                     c_write_untrusted(ssh, s->pwprompt, strlen(s->pwprompt));
7110                     ssh->send_ok = 1;
7111
7112                     setup_userpass_input(ssh, s->password,
7113                                          sizeof(s->password), s->echo);
7114                     do {
7115                         crWaitUntilV(!pktin);
7116                         ret = process_userpass_input(ssh, in, inlen);
7117                     } while (ret == 0);
7118                     if (ret < 0)
7119                         cleanup_exit(0);
7120                     c_write_str(ssh, "\r\n");
7121                 }
7122             }
7123
7124             if (s->method == AUTH_PUBLICKEY_FILE) {
7125                 /*
7126                  * We have our passphrase. Now try the actual authentication.
7127                  */
7128                 struct ssh2_userkey *key;
7129                 const char *error = NULL;
7130
7131                 key = ssh2_load_userkey(&ssh->cfg.keyfile, s->password,
7132                                         &error);
7133                 if (key == SSH2_WRONG_PASSPHRASE || key == NULL) {
7134                     if (key == SSH2_WRONG_PASSPHRASE) {
7135                         c_write_str(ssh, "Wrong passphrase\r\n");
7136                         s->tried_pubkey_config = FALSE;
7137                     } else {
7138                         c_write_str(ssh, "Unable to load private key (");
7139                         c_write_str(ssh, error);
7140                         c_write_str(ssh, ")\r\n");
7141                         s->tried_pubkey_config = TRUE;
7142                     }
7143                     /* Send a spurious AUTH_NONE to return to the top. */
7144                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
7145                     ssh2_pkt_addstring(s->pktout, s->username);
7146                     ssh2_pkt_addstring(s->pktout, "ssh-connection");    /* service requested */
7147                     ssh2_pkt_addstring(s->pktout, "none");      /* method */
7148                     ssh2_pkt_send(ssh, s->pktout);
7149                     s->type = AUTH_TYPE_NONE;
7150                 } else {
7151                     unsigned char *pkblob, *sigblob, *sigdata;
7152                     int pkblob_len, sigblob_len, sigdata_len;
7153                     int p;
7154
7155                     /*
7156                      * We have loaded the private key and the server
7157                      * has announced that it's willing to accept it.
7158                      * Hallelujah. Generate a signature and send it.
7159                      */
7160                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
7161                     ssh2_pkt_addstring(s->pktout, s->username);
7162                     ssh2_pkt_addstring(s->pktout, "ssh-connection");    /* service requested */
7163                     ssh2_pkt_addstring(s->pktout, "publickey"); /* method */
7164                     ssh2_pkt_addbool(s->pktout, TRUE);
7165                     ssh2_pkt_addstring(s->pktout, key->alg->name);
7166                     pkblob = key->alg->public_blob(key->data, &pkblob_len);
7167                     ssh2_pkt_addstring_start(s->pktout);
7168                     ssh2_pkt_addstring_data(s->pktout, (char *)pkblob, pkblob_len);
7169
7170                     /*
7171                      * The data to be signed is:
7172                      *
7173                      *   string  session-id
7174                      *
7175                      * followed by everything so far placed in the
7176                      * outgoing packet.
7177                      */
7178                     sigdata_len = s->pktout->length - 5 + 4 + 20;
7179                     if (ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)
7180                         sigdata_len -= 4;
7181                     sigdata = snewn(sigdata_len, unsigned char);
7182                     p = 0;
7183                     if (!(ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)) {
7184                         PUT_32BIT(sigdata+p, 20);
7185                         p += 4;
7186                     }
7187                     memcpy(sigdata+p, ssh->v2_session_id, 20); p += 20;
7188                     memcpy(sigdata+p, s->pktout->data + 5,
7189                            s->pktout->length - 5);
7190                     p += s->pktout->length - 5;
7191                     assert(p == sigdata_len);
7192                     sigblob = key->alg->sign(key->data, (char *)sigdata,
7193                                              sigdata_len, &sigblob_len);
7194                     ssh2_add_sigblob(ssh, s->pktout, pkblob, pkblob_len,
7195                                      sigblob, sigblob_len);
7196                     sfree(pkblob);
7197                     sfree(sigblob);
7198                     sfree(sigdata);
7199
7200                     ssh2_pkt_send(ssh, s->pktout);
7201                     s->type = AUTH_TYPE_PUBLICKEY;
7202                     key->alg->freekey(key->data);
7203                 }
7204             } else if (s->method == AUTH_PASSWORD) {
7205                 /*
7206                  * We pad out the password packet to 256 bytes to make
7207                  * it harder for an attacker to find the length of the
7208                  * user's password.
7209                  *
7210                  * Anyone using a password longer than 256 bytes
7211                  * probably doesn't have much to worry about from
7212                  * people who find out how long their password is!
7213                  */
7214                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
7215                 s->pktout->forcepad = 256;
7216                 ssh2_pkt_addstring(s->pktout, s->username);
7217                 ssh2_pkt_addstring(s->pktout, "ssh-connection");        /* service requested */
7218                 ssh2_pkt_addstring(s->pktout, "password");
7219                 ssh2_pkt_addbool(s->pktout, FALSE);
7220                 dont_log_password(ssh, s->pktout, PKTLOG_BLANK);
7221                 ssh2_pkt_addstring(s->pktout, s->password);
7222                 memset(s->password, 0, sizeof(s->password));
7223                 end_log_omission(ssh, s->pktout);
7224                 ssh2_pkt_send(ssh, s->pktout);
7225                 logevent("Sent password");
7226                 s->type = AUTH_TYPE_PASSWORD;
7227             } else if (s->method == AUTH_KEYBOARD_INTERACTIVE) {
7228                 if (s->curr_prompt == 0) {
7229                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_INFO_RESPONSE);
7230                     s->pktout->forcepad = 256;
7231                     ssh2_pkt_adduint32(s->pktout, s->num_prompts);
7232                 }
7233                 if (s->need_pw) {      /* only add pw if we just got one! */
7234                     dont_log_password(ssh, s->pktout, PKTLOG_BLANK);
7235                     ssh2_pkt_addstring(s->pktout, s->password);
7236                     memset(s->password, 0, sizeof(s->password));
7237                     end_log_omission(ssh, s->pktout);
7238                     s->curr_prompt++;
7239                 }
7240                 if (s->curr_prompt >= s->num_prompts) {
7241                     ssh2_pkt_send(ssh, s->pktout);
7242                 } else {
7243                     /*
7244                      * If there are prompts remaining, we set
7245                      * `gotit' so that we won't attempt to get
7246                      * another packet. Then we go back round the
7247                      * loop and will end up retrieving another
7248                      * prompt out of the existing packet. Funky or
7249                      * what?
7250                      */
7251                     s->gotit = TRUE;
7252                 }
7253                 s->type = AUTH_TYPE_KEYBOARD_INTERACTIVE;
7254             } else {
7255                 ssh_disconnect(ssh, NULL,
7256                                "No supported authentication methods available",
7257                                SSH2_DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE,
7258                                FALSE);
7259                 crStopV;
7260             }
7261         }
7262     }
7263     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] = NULL;
7264
7265     /*
7266      * Now the connection protocol has started, one way or another.
7267      */
7268
7269     ssh->channels = newtree234(ssh_channelcmp);
7270
7271     /*
7272      * Set up handlers for some connection protocol messages, so we
7273      * don't have to handle them repeatedly in this coroutine.
7274      */
7275     ssh->packet_dispatch[SSH2_MSG_CHANNEL_WINDOW_ADJUST] =
7276         ssh2_msg_channel_window_adjust;
7277     ssh->packet_dispatch[SSH2_MSG_GLOBAL_REQUEST] =
7278         ssh2_msg_global_request;
7279
7280     /*
7281      * Create the main session channel.
7282      */
7283     if (!ssh->cfg.ssh_no_shell) {
7284         ssh->mainchan = snew(struct ssh_channel);
7285         ssh->mainchan->ssh = ssh;
7286         ssh->mainchan->localid = alloc_channel_id(ssh);
7287         s->pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN);
7288         ssh2_pkt_addstring(s->pktout, "session");
7289         ssh2_pkt_adduint32(s->pktout, ssh->mainchan->localid);
7290         ssh->mainchan->v.v2.locwindow = OUR_V2_WINSIZE;
7291         ssh2_pkt_adduint32(s->pktout, ssh->mainchan->v.v2.locwindow);/* our window size */
7292         ssh2_pkt_adduint32(s->pktout, OUR_V2_MAXPKT);    /* our max pkt size */
7293         ssh2_pkt_send(ssh, s->pktout);
7294         crWaitUntilV(pktin);
7295         if (pktin->type != SSH2_MSG_CHANNEL_OPEN_CONFIRMATION) {
7296             bombout(("Server refused to open a session"));
7297             crStopV;
7298             /* FIXME: error data comes back in FAILURE packet */
7299         }
7300         if (ssh_pkt_getuint32(pktin) != ssh->mainchan->localid) {
7301             bombout(("Server's channel confirmation cited wrong channel"));
7302             crStopV;
7303         }
7304         ssh->mainchan->remoteid = ssh_pkt_getuint32(pktin);
7305         ssh->mainchan->halfopen = FALSE;
7306         ssh->mainchan->type = CHAN_MAINSESSION;
7307         ssh->mainchan->closes = 0;
7308         ssh->mainchan->v.v2.remwindow = ssh_pkt_getuint32(pktin);
7309         ssh->mainchan->v.v2.remmaxpkt = ssh_pkt_getuint32(pktin);
7310         bufchain_init(&ssh->mainchan->v.v2.outbuffer);
7311         add234(ssh->channels, ssh->mainchan);
7312         update_specials_menu(ssh->frontend);
7313         logevent("Opened channel for session");
7314     } else
7315         ssh->mainchan = NULL;
7316
7317     /*
7318      * Now we have a channel, make dispatch table entries for
7319      * general channel-based messages.
7320      */
7321     ssh->packet_dispatch[SSH2_MSG_CHANNEL_DATA] =
7322     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EXTENDED_DATA] =
7323         ssh2_msg_channel_data;
7324     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EOF] = ssh2_msg_channel_eof;
7325     ssh->packet_dispatch[SSH2_MSG_CHANNEL_CLOSE] = ssh2_msg_channel_close;
7326     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_CONFIRMATION] =
7327         ssh2_msg_channel_open_confirmation;
7328     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_FAILURE] =
7329         ssh2_msg_channel_open_failure;
7330     ssh->packet_dispatch[SSH2_MSG_CHANNEL_REQUEST] =
7331         ssh2_msg_channel_request;
7332     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN] =
7333         ssh2_msg_channel_open;
7334
7335     /*
7336      * Potentially enable X11 forwarding.
7337      */
7338     if (ssh->mainchan && ssh->cfg.x11_forward) {
7339         char proto[20], data[64];
7340         logevent("Requesting X11 forwarding");
7341         ssh->x11auth = x11_invent_auth(proto, sizeof(proto),
7342                                        data, sizeof(data), ssh->cfg.x11_auth);
7343         x11_get_real_auth(ssh->x11auth, ssh->cfg.x11_display);
7344         s->pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
7345         ssh2_pkt_adduint32(s->pktout, ssh->mainchan->remoteid);
7346         ssh2_pkt_addstring(s->pktout, "x11-req");
7347         ssh2_pkt_addbool(s->pktout, 1);        /* want reply */
7348         ssh2_pkt_addbool(s->pktout, 0);        /* many connections */
7349         ssh2_pkt_addstring(s->pktout, proto);
7350         /*
7351          * Note that while we blank the X authentication data here, we don't
7352          * take any special action to blank the start of an X11 channel,
7353          * so using MIT-MAGIC-COOKIE-1 and actually opening an X connection
7354          * without having session blanking enabled is likely to leak your
7355          * cookie into the log.
7356          */
7357         dont_log_password(ssh, s->pktout, PKTLOG_BLANK);
7358         ssh2_pkt_addstring(s->pktout, data);
7359         end_log_omission(ssh, s->pktout);
7360         ssh2_pkt_adduint32(s->pktout, x11_get_screen_number(ssh->cfg.x11_display));
7361         ssh2_pkt_send(ssh, s->pktout);
7362
7363         crWaitUntilV(pktin);
7364
7365         if (pktin->type != SSH2_MSG_CHANNEL_SUCCESS) {
7366             if (pktin->type != SSH2_MSG_CHANNEL_FAILURE) {
7367                 bombout(("Unexpected response to X11 forwarding request:"
7368                          " packet type %d", pktin->type));
7369                 crStopV;
7370             }
7371             logevent("X11 forwarding refused");
7372         } else {
7373             logevent("X11 forwarding enabled");
7374             ssh->X11_fwd_enabled = TRUE;
7375         }
7376     }
7377
7378     /*
7379      * Enable port forwardings.
7380      */
7381     ssh_setup_portfwd(ssh, &ssh->cfg);
7382
7383     /*
7384      * Potentially enable agent forwarding.
7385      */
7386     if (ssh->mainchan && ssh->cfg.agentfwd && agent_exists()) {
7387         logevent("Requesting OpenSSH-style agent forwarding");
7388         s->pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
7389         ssh2_pkt_adduint32(s->pktout, ssh->mainchan->remoteid);
7390         ssh2_pkt_addstring(s->pktout, "auth-agent-req@openssh.com");
7391         ssh2_pkt_addbool(s->pktout, 1);        /* want reply */
7392         ssh2_pkt_send(ssh, s->pktout);
7393
7394         crWaitUntilV(pktin);
7395
7396         if (pktin->type != SSH2_MSG_CHANNEL_SUCCESS) {
7397             if (pktin->type != SSH2_MSG_CHANNEL_FAILURE) {
7398                 bombout(("Unexpected response to agent forwarding request:"
7399                          " packet type %d", pktin->type));
7400                 crStopV;
7401             }
7402             logevent("Agent forwarding refused");
7403         } else {
7404             logevent("Agent forwarding enabled");
7405             ssh->agentfwd_enabled = TRUE;
7406         }
7407     }
7408
7409     /*
7410      * Now allocate a pty for the session.
7411      */
7412     if (ssh->mainchan && !ssh->cfg.nopty) {
7413         /* Unpick the terminal-speed string. */
7414         /* XXX perhaps we should allow no speeds to be sent. */
7415         ssh->ospeed = 38400; ssh->ispeed = 38400; /* last-resort defaults */
7416         sscanf(ssh->cfg.termspeed, "%d,%d", &ssh->ospeed, &ssh->ispeed);
7417         /* Build the pty request. */
7418         s->pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
7419         ssh2_pkt_adduint32(s->pktout, ssh->mainchan->remoteid); /* recipient channel */
7420         ssh2_pkt_addstring(s->pktout, "pty-req");
7421         ssh2_pkt_addbool(s->pktout, 1);        /* want reply */
7422         ssh2_pkt_addstring(s->pktout, ssh->cfg.termtype);
7423         ssh2_pkt_adduint32(s->pktout, ssh->term_width);
7424         ssh2_pkt_adduint32(s->pktout, ssh->term_height);
7425         ssh2_pkt_adduint32(s->pktout, 0);              /* pixel width */
7426         ssh2_pkt_adduint32(s->pktout, 0);              /* pixel height */
7427         ssh2_pkt_addstring_start(s->pktout);
7428         parse_ttymodes(ssh, ssh->cfg.ttymodes,
7429                        ssh2_send_ttymode, (void *)s->pktout);
7430         ssh2_pkt_addbyte(s->pktout, SSH2_TTY_OP_ISPEED);
7431         ssh2_pkt_adduint32(s->pktout, ssh->ispeed);
7432         ssh2_pkt_addbyte(s->pktout, SSH2_TTY_OP_OSPEED);
7433         ssh2_pkt_adduint32(s->pktout, ssh->ospeed);
7434         ssh2_pkt_addstring_data(s->pktout, "\0", 1); /* TTY_OP_END */
7435         ssh2_pkt_send(ssh, s->pktout);
7436         ssh->state = SSH_STATE_INTERMED;
7437
7438         crWaitUntilV(pktin);
7439
7440         if (pktin->type != SSH2_MSG_CHANNEL_SUCCESS) {
7441             if (pktin->type != SSH2_MSG_CHANNEL_FAILURE) {
7442                 bombout(("Unexpected response to pty request:"
7443                          " packet type %d", pktin->type));
7444                 crStopV;
7445             }
7446             c_write_str(ssh, "Server refused to allocate pty\r\n");
7447             ssh->editing = ssh->echoing = 1;
7448         } else {
7449             logeventf(ssh, "Allocated pty (ospeed %dbps, ispeed %dbps)",
7450                       ssh->ospeed, ssh->ispeed);
7451         }
7452     } else {
7453         ssh->editing = ssh->echoing = 1;
7454     }
7455
7456     /*
7457      * Send environment variables.
7458      * 
7459      * Simplest thing here is to send all the requests at once, and
7460      * then wait for a whole bunch of successes or failures.
7461      */
7462     if (ssh->mainchan && *ssh->cfg.environmt) {
7463         char *e = ssh->cfg.environmt;
7464         char *var, *varend, *val;
7465
7466         s->num_env = 0;
7467
7468         while (*e) {
7469             var = e;
7470             while (*e && *e != '\t') e++;
7471             varend = e;
7472             if (*e == '\t') e++;
7473             val = e;
7474             while (*e) e++;
7475             e++;
7476
7477             s->pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
7478             ssh2_pkt_adduint32(s->pktout, ssh->mainchan->remoteid);
7479             ssh2_pkt_addstring(s->pktout, "env");
7480             ssh2_pkt_addbool(s->pktout, 1);            /* want reply */
7481             ssh2_pkt_addstring_start(s->pktout);
7482             ssh2_pkt_addstring_data(s->pktout, var, varend-var);
7483             ssh2_pkt_addstring(s->pktout, val);
7484             ssh2_pkt_send(ssh, s->pktout);
7485
7486             s->num_env++;
7487         }
7488
7489         logeventf(ssh, "Sent %d environment variables", s->num_env);
7490
7491         s->env_ok = 0;
7492         s->env_left = s->num_env;
7493
7494         while (s->env_left > 0) {
7495             crWaitUntilV(pktin);
7496
7497             if (pktin->type != SSH2_MSG_CHANNEL_SUCCESS) {
7498                 if (pktin->type != SSH2_MSG_CHANNEL_FAILURE) {
7499                     bombout(("Unexpected response to environment request:"
7500                              " packet type %d", pktin->type));
7501                     crStopV;
7502                 }
7503             } else {
7504                 s->env_ok++;
7505             }
7506
7507             s->env_left--;
7508         }
7509
7510         if (s->env_ok == s->num_env) {
7511             logevent("All environment variables successfully set");
7512         } else if (s->env_ok == 0) {
7513             logevent("All environment variables refused");
7514             c_write_str(ssh, "Server refused to set environment variables\r\n");
7515         } else {
7516             logeventf(ssh, "%d environment variables refused",
7517                       s->num_env - s->env_ok);
7518             c_write_str(ssh, "Server refused to set all environment variables\r\n");
7519         }
7520     }
7521
7522     /*
7523      * Start a shell or a remote command. We may have to attempt
7524      * this twice if the config data has provided a second choice
7525      * of command.
7526      */
7527     if (ssh->mainchan) while (1) {
7528         int subsys;
7529         char *cmd;
7530
7531         if (ssh->fallback_cmd) {
7532             subsys = ssh->cfg.ssh_subsys2;
7533             cmd = ssh->cfg.remote_cmd_ptr2;
7534         } else {
7535             subsys = ssh->cfg.ssh_subsys;
7536             cmd = ssh->cfg.remote_cmd_ptr;
7537             if (!cmd) cmd = ssh->cfg.remote_cmd;
7538         }
7539
7540         s->pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
7541         ssh2_pkt_adduint32(s->pktout, ssh->mainchan->remoteid); /* recipient channel */
7542         if (subsys) {
7543             ssh2_pkt_addstring(s->pktout, "subsystem");
7544             ssh2_pkt_addbool(s->pktout, 1);            /* want reply */
7545             ssh2_pkt_addstring(s->pktout, cmd);
7546         } else if (*cmd) {
7547             ssh2_pkt_addstring(s->pktout, "exec");
7548             ssh2_pkt_addbool(s->pktout, 1);            /* want reply */
7549             ssh2_pkt_addstring(s->pktout, cmd);
7550         } else {
7551             ssh2_pkt_addstring(s->pktout, "shell");
7552             ssh2_pkt_addbool(s->pktout, 1);            /* want reply */
7553         }
7554         ssh2_pkt_send(ssh, s->pktout);
7555
7556         crWaitUntilV(pktin);
7557
7558         if (pktin->type != SSH2_MSG_CHANNEL_SUCCESS) {
7559             if (pktin->type != SSH2_MSG_CHANNEL_FAILURE) {
7560                 bombout(("Unexpected response to shell/command request:"
7561                          " packet type %d", pktin->type));
7562                 crStopV;
7563             }
7564             /*
7565              * We failed to start the command. If this is the
7566              * fallback command, we really are finished; if it's
7567              * not, and if the fallback command exists, try falling
7568              * back to it before complaining.
7569              */
7570             if (!ssh->fallback_cmd && ssh->cfg.remote_cmd_ptr2 != NULL) {
7571                 logevent("Primary command failed; attempting fallback");
7572                 ssh->fallback_cmd = TRUE;
7573                 continue;
7574             }
7575             bombout(("Server refused to start a shell/command"));
7576             crStopV;
7577         } else {
7578             logevent("Started a shell/command");
7579         }
7580         break;
7581     }
7582
7583     ssh->state = SSH_STATE_SESSION;
7584     if (ssh->size_needed)
7585         ssh_size(ssh, ssh->term_width, ssh->term_height);
7586     if (ssh->eof_needed)
7587         ssh_special(ssh, TS_EOF);
7588
7589     /*
7590      * Transfer data!
7591      */
7592     if (ssh->ldisc)
7593         ldisc_send(ssh->ldisc, NULL, 0, 0);/* cause ldisc to notice changes */
7594     if (ssh->mainchan)
7595         ssh->send_ok = 1;
7596     while (1) {
7597         crReturnV;
7598         s->try_send = FALSE;
7599         if (pktin) {
7600
7601             /*
7602              * _All_ the connection-layer packets we expect to
7603              * receive are now handled by the dispatch table.
7604              * Anything that reaches here must be bogus.
7605              */
7606
7607             bombout(("Strange packet received: type %d", pktin->type));
7608             crStopV;
7609         } else if (ssh->mainchan) {
7610             /*
7611              * We have spare data. Add it to the channel buffer.
7612              */
7613             ssh2_add_channel_data(ssh->mainchan, (char *)in, inlen);
7614             s->try_send = TRUE;
7615         }
7616         if (s->try_send) {
7617             int i;
7618             struct ssh_channel *c;
7619             /*
7620              * Try to send data on all channels if we can.
7621              */
7622             for (i = 0; NULL != (c = index234(ssh->channels, i)); i++)
7623                 ssh2_try_send_and_unthrottle(c);
7624         }
7625     }
7626
7627     crFinishV;
7628 }
7629
7630 /*
7631  * Handlers for SSH-2 messages that might arrive at any moment.
7632  */
7633 static void ssh2_msg_disconnect(Ssh ssh, struct Packet *pktin)
7634 {
7635     /* log reason code in disconnect message */
7636     char *buf, *msg;
7637     int nowlen, reason, msglen;
7638
7639     reason = ssh_pkt_getuint32(pktin);
7640     ssh_pkt_getstring(pktin, &msg, &msglen);
7641
7642     if (reason > 0 && reason < lenof(ssh2_disconnect_reasons)) {
7643         buf = dupprintf("Received disconnect message (%s)",
7644                         ssh2_disconnect_reasons[reason]);
7645     } else {
7646         buf = dupprintf("Received disconnect message (unknown"
7647                         " type %d)", reason);
7648     }
7649     logevent(buf);
7650     sfree(buf);
7651     buf = dupprintf("Disconnection message text: %n%.*s",
7652                     &nowlen, msglen, msg);
7653     logevent(buf);
7654     bombout(("Server sent disconnect message\ntype %d (%s):\n\"%s\"",
7655              reason,
7656              (reason > 0 && reason < lenof(ssh2_disconnect_reasons)) ?
7657              ssh2_disconnect_reasons[reason] : "unknown",
7658              buf+nowlen));
7659     sfree(buf);
7660 }
7661
7662 static void ssh2_msg_debug(Ssh ssh, struct Packet *pktin)
7663 {
7664     /* log the debug message */
7665     char *msg;
7666     int msglen;
7667     int always_display;
7668
7669     /* XXX maybe we should actually take notice of this */
7670     always_display = ssh2_pkt_getbool(pktin);
7671     ssh_pkt_getstring(pktin, &msg, &msglen);
7672
7673     logeventf(ssh, "Remote debug message: %.*s", msglen, msg);
7674 }
7675
7676 static void ssh2_msg_something_unimplemented(Ssh ssh, struct Packet *pktin)
7677 {
7678     struct Packet *pktout;
7679     pktout = ssh2_pkt_init(SSH2_MSG_UNIMPLEMENTED);
7680     ssh2_pkt_adduint32(pktout, pktin->sequence);
7681     /*
7682      * UNIMPLEMENTED messages MUST appear in the same order as the
7683      * messages they respond to. Hence, never queue them.
7684      */
7685     ssh2_pkt_send_noqueue(ssh, pktout);
7686 }
7687
7688 /*
7689  * Handle the top-level SSH-2 protocol.
7690  */
7691 static void ssh2_protocol_setup(Ssh ssh)
7692 {
7693     int i;
7694
7695     /*
7696      * Most messages cause SSH2_MSG_UNIMPLEMENTED.
7697      */
7698     for (i = 0; i < 256; i++)
7699         ssh->packet_dispatch[i] = ssh2_msg_something_unimplemented;
7700
7701     /*
7702      * Any message we actually understand, we set to NULL so that
7703      * the coroutines will get it.
7704      */
7705     ssh->packet_dispatch[SSH2_MSG_UNIMPLEMENTED] = NULL;
7706     ssh->packet_dispatch[SSH2_MSG_SERVICE_REQUEST] = NULL;
7707     ssh->packet_dispatch[SSH2_MSG_SERVICE_ACCEPT] = NULL;
7708     ssh->packet_dispatch[SSH2_MSG_KEXINIT] = NULL;
7709     ssh->packet_dispatch[SSH2_MSG_NEWKEYS] = NULL;
7710     ssh->packet_dispatch[SSH2_MSG_KEXDH_INIT] = NULL;
7711     ssh->packet_dispatch[SSH2_MSG_KEXDH_REPLY] = NULL;
7712     /* ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_REQUEST] = NULL; duplicate case value */
7713     /* ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_GROUP] = NULL; duplicate case value */
7714     ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_INIT] = NULL;
7715     ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_REPLY] = NULL;
7716     ssh->packet_dispatch[SSH2_MSG_USERAUTH_REQUEST] = NULL;
7717     ssh->packet_dispatch[SSH2_MSG_USERAUTH_FAILURE] = NULL;
7718     ssh->packet_dispatch[SSH2_MSG_USERAUTH_SUCCESS] = NULL;
7719     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] = NULL;
7720     ssh->packet_dispatch[SSH2_MSG_USERAUTH_PK_OK] = NULL;
7721     /* ssh->packet_dispatch[SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ] = NULL; duplicate case value */
7722     /* ssh->packet_dispatch[SSH2_MSG_USERAUTH_INFO_REQUEST] = NULL; duplicate case value */
7723     ssh->packet_dispatch[SSH2_MSG_USERAUTH_INFO_RESPONSE] = NULL;
7724     ssh->packet_dispatch[SSH2_MSG_GLOBAL_REQUEST] = NULL;
7725     ssh->packet_dispatch[SSH2_MSG_REQUEST_SUCCESS] = NULL;
7726     ssh->packet_dispatch[SSH2_MSG_REQUEST_FAILURE] = NULL;
7727     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN] = NULL;
7728     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_CONFIRMATION] = NULL;
7729     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_FAILURE] = NULL;
7730     ssh->packet_dispatch[SSH2_MSG_CHANNEL_WINDOW_ADJUST] = NULL;
7731     ssh->packet_dispatch[SSH2_MSG_CHANNEL_DATA] = NULL;
7732     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EXTENDED_DATA] = NULL;
7733     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EOF] = NULL;
7734     ssh->packet_dispatch[SSH2_MSG_CHANNEL_CLOSE] = NULL;
7735     ssh->packet_dispatch[SSH2_MSG_CHANNEL_REQUEST] = NULL;
7736     ssh->packet_dispatch[SSH2_MSG_CHANNEL_SUCCESS] = NULL;
7737     ssh->packet_dispatch[SSH2_MSG_CHANNEL_FAILURE] = NULL;
7738
7739     /*
7740      * These special message types we install handlers for.
7741      */
7742     ssh->packet_dispatch[SSH2_MSG_DISCONNECT] = ssh2_msg_disconnect;
7743     ssh->packet_dispatch[SSH2_MSG_IGNORE] = ssh_msg_ignore; /* shared with SSH-1 */
7744     ssh->packet_dispatch[SSH2_MSG_DEBUG] = ssh2_msg_debug;
7745 }
7746
7747 static void ssh2_timer(void *ctx, long now)
7748 {
7749     Ssh ssh = (Ssh)ctx;
7750
7751     if (ssh->state == SSH_STATE_CLOSED)
7752         return;
7753
7754     if (!ssh->kex_in_progress && ssh->cfg.ssh_rekey_time != 0 &&
7755         now - ssh->next_rekey >= 0) {
7756         do_ssh2_transport(ssh, "timeout", -1, NULL);
7757     }
7758 }
7759
7760 static void ssh2_protocol(Ssh ssh, void *vin, int inlen,
7761                           struct Packet *pktin)
7762 {
7763     unsigned char *in = (unsigned char *)vin;
7764     if (ssh->state == SSH_STATE_CLOSED)
7765         return;
7766
7767     if (pktin) {
7768         ssh->incoming_data_size += pktin->encrypted_len;
7769         if (!ssh->kex_in_progress &&
7770             ssh->max_data_size != 0 &&
7771             ssh->incoming_data_size > ssh->max_data_size)
7772             do_ssh2_transport(ssh, "too much data received", -1, NULL);
7773     }
7774
7775     if (pktin && ssh->packet_dispatch[pktin->type]) {
7776         ssh->packet_dispatch[pktin->type](ssh, pktin);
7777         return;
7778     }
7779
7780     if (!ssh->protocol_initial_phase_done ||
7781         (pktin && pktin->type >= 20 && pktin->type < 50)) {
7782         if (do_ssh2_transport(ssh, in, inlen, pktin) &&
7783             !ssh->protocol_initial_phase_done) {
7784             ssh->protocol_initial_phase_done = TRUE;
7785             /*
7786              * Allow authconn to initialise itself.
7787              */
7788             do_ssh2_authconn(ssh, NULL, 0, NULL);
7789         }
7790     } else {
7791         do_ssh2_authconn(ssh, in, inlen, pktin);
7792     }
7793 }
7794
7795 /*
7796  * Called to set up the connection.
7797  *
7798  * Returns an error message, or NULL on success.
7799  */
7800 static const char *ssh_init(void *frontend_handle, void **backend_handle,
7801                             Config *cfg,
7802                             char *host, int port, char **realhost, int nodelay,
7803                             int keepalive)
7804 {
7805     const char *p;
7806     Ssh ssh;
7807
7808     ssh = snew(struct ssh_tag);
7809     ssh->cfg = *cfg;                   /* STRUCTURE COPY */
7810     ssh->version = 0;                  /* when not ready yet */
7811     ssh->s = NULL;
7812     ssh->cipher = NULL;
7813     ssh->v1_cipher_ctx = NULL;
7814     ssh->crcda_ctx = NULL;
7815     ssh->cscipher = NULL;
7816     ssh->cs_cipher_ctx = NULL;
7817     ssh->sccipher = NULL;
7818     ssh->sc_cipher_ctx = NULL;
7819     ssh->csmac = NULL;
7820     ssh->cs_mac_ctx = NULL;
7821     ssh->scmac = NULL;
7822     ssh->sc_mac_ctx = NULL;
7823     ssh->cscomp = NULL;
7824     ssh->cs_comp_ctx = NULL;
7825     ssh->sccomp = NULL;
7826     ssh->sc_comp_ctx = NULL;
7827     ssh->kex = NULL;
7828     ssh->kex_ctx = NULL;
7829     ssh->hostkey = NULL;
7830     ssh->exitcode = -1;
7831     ssh->close_expected = FALSE;
7832     ssh->clean_exit = FALSE;
7833     ssh->state = SSH_STATE_PREPACKET;
7834     ssh->size_needed = FALSE;
7835     ssh->eof_needed = FALSE;
7836     ssh->ldisc = NULL;
7837     ssh->logctx = NULL;
7838     ssh->deferred_send_data = NULL;
7839     ssh->deferred_len = 0;
7840     ssh->deferred_size = 0;
7841     ssh->fallback_cmd = 0;
7842     ssh->pkt_ctx = 0;
7843     ssh->x11auth = NULL;
7844     ssh->v1_compressing = FALSE;
7845     ssh->v2_outgoing_sequence = 0;
7846     ssh->ssh1_rdpkt_crstate = 0;
7847     ssh->ssh2_rdpkt_crstate = 0;
7848     ssh->do_ssh_init_crstate = 0;
7849     ssh->ssh_gotdata_crstate = 0;
7850     ssh->do_ssh1_connection_crstate = 0;
7851     ssh->do_ssh1_login_crstate = 0;
7852     ssh->do_ssh2_transport_crstate = 0;
7853     ssh->do_ssh2_authconn_crstate = 0;
7854     ssh->do_ssh_init_state = NULL;
7855     ssh->do_ssh1_login_state = NULL;
7856     ssh->do_ssh2_transport_state = NULL;
7857     ssh->do_ssh2_authconn_state = NULL;
7858     ssh->mainchan = NULL;
7859     ssh->throttled_all = 0;
7860     ssh->v1_stdout_throttling = 0;
7861     ssh->queue = NULL;
7862     ssh->queuelen = ssh->queuesize = 0;
7863     ssh->queueing = FALSE;
7864     ssh->qhead = ssh->qtail = NULL;
7865     ssh->deferred_rekey_reason = NULL;
7866     bufchain_init(&ssh->queued_incoming_data);
7867     ssh->frozen = FALSE;
7868
7869     *backend_handle = ssh;
7870
7871 #ifdef MSCRYPTOAPI
7872     if (crypto_startup() == 0)
7873         return "Microsoft high encryption pack not installed!";
7874 #endif
7875
7876     ssh->frontend = frontend_handle;
7877     ssh->term_width = ssh->cfg.width;
7878     ssh->term_height = ssh->cfg.height;
7879
7880     ssh->channels = NULL;
7881     ssh->rportfwds = NULL;
7882     ssh->portfwds = NULL;
7883
7884     ssh->send_ok = 0;
7885     ssh->editing = 0;
7886     ssh->echoing = 0;
7887     ssh->v1_throttle_count = 0;
7888     ssh->overall_bufsize = 0;
7889     ssh->fallback_cmd = 0;
7890
7891     ssh->protocol = NULL;
7892
7893     ssh->protocol_initial_phase_done = FALSE;
7894
7895     ssh->pinger = NULL;
7896
7897     ssh->incoming_data_size = ssh->outgoing_data_size =
7898         ssh->deferred_data_size = 0L;
7899     ssh->max_data_size = parse_blocksize(ssh->cfg.ssh_rekey_data);
7900     ssh->kex_in_progress = FALSE;
7901
7902     p = connect_to_host(ssh, host, port, realhost, nodelay, keepalive);
7903     if (p != NULL)
7904         return p;
7905
7906     random_ref();
7907
7908     return NULL;
7909 }
7910
7911 static void ssh_free(void *handle)
7912 {
7913     Ssh ssh = (Ssh) handle;
7914     struct ssh_channel *c;
7915     struct ssh_rportfwd *pf;
7916
7917     if (ssh->v1_cipher_ctx)
7918         ssh->cipher->free_context(ssh->v1_cipher_ctx);
7919     if (ssh->cs_cipher_ctx)
7920         ssh->cscipher->free_context(ssh->cs_cipher_ctx);
7921     if (ssh->sc_cipher_ctx)
7922         ssh->sccipher->free_context(ssh->sc_cipher_ctx);
7923     if (ssh->cs_mac_ctx)
7924         ssh->csmac->free_context(ssh->cs_mac_ctx);
7925     if (ssh->sc_mac_ctx)
7926         ssh->scmac->free_context(ssh->sc_mac_ctx);
7927     if (ssh->cs_comp_ctx) {
7928         if (ssh->cscomp)
7929             ssh->cscomp->compress_cleanup(ssh->cs_comp_ctx);
7930         else
7931             zlib_compress_cleanup(ssh->cs_comp_ctx);
7932     }
7933     if (ssh->sc_comp_ctx) {
7934         if (ssh->sccomp)
7935             ssh->sccomp->decompress_cleanup(ssh->sc_comp_ctx);
7936         else
7937             zlib_decompress_cleanup(ssh->sc_comp_ctx);
7938     }
7939     if (ssh->kex_ctx)
7940         dh_cleanup(ssh->kex_ctx);
7941     sfree(ssh->savedhost);
7942
7943     while (ssh->queuelen-- > 0)
7944         ssh_free_packet(ssh->queue[ssh->queuelen]);
7945     sfree(ssh->queue);
7946
7947     while (ssh->qhead) {
7948         struct queued_handler *qh = ssh->qhead;
7949         ssh->qhead = qh->next;
7950         sfree(ssh->qhead);
7951     }
7952     ssh->qhead = ssh->qtail = NULL;
7953
7954     if (ssh->channels) {
7955         while ((c = delpos234(ssh->channels, 0)) != NULL) {
7956             switch (c->type) {
7957               case CHAN_X11:
7958                 if (c->u.x11.s != NULL)
7959                     x11_close(c->u.x11.s);
7960                 break;
7961               case CHAN_SOCKDATA:
7962                 if (c->u.pfd.s != NULL)
7963                     pfd_close(c->u.pfd.s);
7964                 break;
7965             }
7966             sfree(c);
7967         }
7968         freetree234(ssh->channels);
7969         ssh->channels = NULL;
7970     }
7971
7972     if (ssh->rportfwds) {
7973         while ((pf = delpos234(ssh->rportfwds, 0)) != NULL)
7974             sfree(pf);
7975         freetree234(ssh->rportfwds);
7976         ssh->rportfwds = NULL;
7977     }
7978     sfree(ssh->deferred_send_data);
7979     if (ssh->x11auth)
7980         x11_free_auth(ssh->x11auth);
7981     sfree(ssh->do_ssh_init_state);
7982     sfree(ssh->do_ssh1_login_state);
7983     sfree(ssh->do_ssh2_transport_state);
7984     sfree(ssh->do_ssh2_authconn_state);
7985     if (ssh->crcda_ctx) {
7986         crcda_free_context(ssh->crcda_ctx);
7987         ssh->crcda_ctx = NULL;
7988     }
7989     if (ssh->s)
7990         ssh_do_close(ssh, TRUE);
7991     expire_timer_context(ssh);
7992     if (ssh->pinger)
7993         pinger_free(ssh->pinger);
7994     bufchain_clear(&ssh->queued_incoming_data);
7995     sfree(ssh);
7996
7997     random_unref();
7998 }
7999
8000 /*
8001  * Reconfigure the SSH backend.
8002  */
8003 static void ssh_reconfig(void *handle, Config *cfg)
8004 {
8005     Ssh ssh = (Ssh) handle;
8006     char *rekeying = NULL, rekey_mandatory = FALSE;
8007     unsigned long old_max_data_size;
8008
8009     pinger_reconfig(ssh->pinger, &ssh->cfg, cfg);
8010     if (ssh->portfwds)
8011         ssh_setup_portfwd(ssh, cfg);
8012
8013     if (ssh->cfg.ssh_rekey_time != cfg->ssh_rekey_time &&
8014         cfg->ssh_rekey_time != 0) {
8015         long new_next = ssh->last_rekey + cfg->ssh_rekey_time*60*TICKSPERSEC;
8016         long now = GETTICKCOUNT();
8017
8018         if (new_next - now < 0) {
8019             rekeying = "timeout shortened";
8020         } else {
8021             ssh->next_rekey = schedule_timer(new_next - now, ssh2_timer, ssh);
8022         }
8023     }
8024
8025     old_max_data_size = ssh->max_data_size;
8026     ssh->max_data_size = parse_blocksize(cfg->ssh_rekey_data);
8027     if (old_max_data_size != ssh->max_data_size &&
8028         ssh->max_data_size != 0) {
8029         if (ssh->outgoing_data_size > ssh->max_data_size ||
8030             ssh->incoming_data_size > ssh->max_data_size)
8031             rekeying = "data limit lowered";
8032     }
8033
8034     if (ssh->cfg.compression != cfg->compression) {
8035         rekeying = "compression setting changed";
8036         rekey_mandatory = TRUE;
8037     }
8038
8039     if (ssh->cfg.ssh2_des_cbc != cfg->ssh2_des_cbc ||
8040         memcmp(ssh->cfg.ssh_cipherlist, cfg->ssh_cipherlist,
8041                sizeof(ssh->cfg.ssh_cipherlist))) {
8042         rekeying = "cipher settings changed";
8043         rekey_mandatory = TRUE;
8044     }
8045
8046     ssh->cfg = *cfg;                   /* STRUCTURE COPY */
8047
8048     if (rekeying) {
8049         if (!ssh->kex_in_progress) {
8050             do_ssh2_transport(ssh, rekeying, -1, NULL);
8051         } else if (rekey_mandatory) {
8052             ssh->deferred_rekey_reason = rekeying;
8053         }
8054     }
8055 }
8056
8057 /*
8058  * Called to send data down the SSH connection.
8059  */
8060 static int ssh_send(void *handle, char *buf, int len)
8061 {
8062     Ssh ssh = (Ssh) handle;
8063
8064     if (ssh == NULL || ssh->s == NULL || ssh->protocol == NULL)
8065         return 0;
8066
8067     ssh->protocol(ssh, (unsigned char *)buf, len, 0);
8068
8069     return ssh_sendbuffer(ssh);
8070 }
8071
8072 /*
8073  * Called to query the current amount of buffered stdin data.
8074  */
8075 static int ssh_sendbuffer(void *handle)
8076 {
8077     Ssh ssh = (Ssh) handle;
8078     int override_value;
8079
8080     if (ssh == NULL || ssh->s == NULL || ssh->protocol == NULL)
8081         return 0;
8082
8083     /*
8084      * If the SSH socket itself has backed up, add the total backup
8085      * size on that to any individual buffer on the stdin channel.
8086      */
8087     override_value = 0;
8088     if (ssh->throttled_all)
8089         override_value = ssh->overall_bufsize;
8090
8091     if (ssh->version == 1) {
8092         return override_value;
8093     } else if (ssh->version == 2) {
8094         if (!ssh->mainchan || ssh->mainchan->closes > 0)
8095             return override_value;
8096         else
8097             return (override_value +
8098                     bufchain_size(&ssh->mainchan->v.v2.outbuffer));
8099     }
8100
8101     return 0;
8102 }
8103
8104 /*
8105  * Called to set the size of the window from SSH's POV.
8106  */
8107 static void ssh_size(void *handle, int width, int height)
8108 {
8109     Ssh ssh = (Ssh) handle;
8110     struct Packet *pktout;
8111
8112     ssh->term_width = width;
8113     ssh->term_height = height;
8114
8115     switch (ssh->state) {
8116       case SSH_STATE_BEFORE_SIZE:
8117       case SSH_STATE_PREPACKET:
8118       case SSH_STATE_CLOSED:
8119         break;                         /* do nothing */
8120       case SSH_STATE_INTERMED:
8121         ssh->size_needed = TRUE;       /* buffer for later */
8122         break;
8123       case SSH_STATE_SESSION:
8124         if (!ssh->cfg.nopty) {
8125             if (ssh->version == 1) {
8126                 send_packet(ssh, SSH1_CMSG_WINDOW_SIZE,
8127                             PKT_INT, ssh->term_height,
8128                             PKT_INT, ssh->term_width,
8129                             PKT_INT, 0, PKT_INT, 0, PKT_END);
8130             } else if (ssh->mainchan) {
8131                 pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
8132                 ssh2_pkt_adduint32(pktout, ssh->mainchan->remoteid);
8133                 ssh2_pkt_addstring(pktout, "window-change");
8134                 ssh2_pkt_addbool(pktout, 0);
8135                 ssh2_pkt_adduint32(pktout, ssh->term_width);
8136                 ssh2_pkt_adduint32(pktout, ssh->term_height);
8137                 ssh2_pkt_adduint32(pktout, 0);
8138                 ssh2_pkt_adduint32(pktout, 0);
8139                 ssh2_pkt_send(ssh, pktout);
8140             }
8141         }
8142         break;
8143     }
8144 }
8145
8146 /*
8147  * Return a list of the special codes that make sense in this
8148  * protocol.
8149  */
8150 static const struct telnet_special *ssh_get_specials(void *handle)
8151 {
8152     static const struct telnet_special ssh1_ignore_special[] = {
8153         {"IGNORE message", TS_NOP}
8154     };
8155     static const struct telnet_special ssh2_transport_specials[] = {
8156         {"IGNORE message", TS_NOP},
8157         {"Repeat key exchange", TS_REKEY},
8158     };
8159     static const struct telnet_special ssh2_session_specials[] = {
8160         {NULL, TS_SEP},
8161         {"Break", TS_BRK},
8162         /* These are the signal names defined by draft-ietf-secsh-connect-23.
8163          * They include all the ISO C signals, but are a subset of the POSIX
8164          * required signals. */
8165         {"SIGINT (Interrupt)", TS_SIGINT},
8166         {"SIGTERM (Terminate)", TS_SIGTERM},
8167         {"SIGKILL (Kill)", TS_SIGKILL},
8168         {"SIGQUIT (Quit)", TS_SIGQUIT},
8169         {"SIGHUP (Hangup)", TS_SIGHUP},
8170         {"More signals", TS_SUBMENU},
8171           {"SIGABRT", TS_SIGABRT}, {"SIGALRM", TS_SIGALRM},
8172           {"SIGFPE",  TS_SIGFPE},  {"SIGILL",  TS_SIGILL},
8173           {"SIGPIPE", TS_SIGPIPE}, {"SIGSEGV", TS_SIGSEGV},
8174           {"SIGUSR1", TS_SIGUSR1}, {"SIGUSR2", TS_SIGUSR2},
8175         {NULL, TS_EXITMENU}
8176     };
8177     static const struct telnet_special specials_end[] = {
8178         {NULL, TS_EXITMENU}
8179     };
8180     /* XXX review this length for any changes: */
8181     static struct telnet_special ssh_specials[lenof(ssh2_transport_specials) +
8182                                               lenof(ssh2_session_specials) +
8183                                               lenof(specials_end)];
8184     Ssh ssh = (Ssh) handle;
8185     int i = 0;
8186 #define ADD_SPECIALS(name) \
8187     do { \
8188         assert((i + lenof(name)) <= lenof(ssh_specials)); \
8189         memcpy(&ssh_specials[i], name, sizeof name); \
8190         i += lenof(name); \
8191     } while(0)
8192
8193     if (ssh->version == 1) {
8194         /* Don't bother offering IGNORE if we've decided the remote
8195          * won't cope with it, since we wouldn't bother sending it if
8196          * asked anyway. */
8197         if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH1_IGNORE))
8198             ADD_SPECIALS(ssh1_ignore_special);
8199     } else if (ssh->version == 2) {
8200         ADD_SPECIALS(ssh2_transport_specials);
8201         if (ssh->mainchan)
8202             ADD_SPECIALS(ssh2_session_specials);
8203     } /* else we're not ready yet */
8204
8205     if (i) {
8206         ADD_SPECIALS(specials_end);
8207         return ssh_specials;
8208     } else {
8209         return NULL;
8210     }
8211 #undef ADD_SPECIALS
8212 }
8213
8214 /*
8215  * Send special codes. TS_EOF is useful for `plink', so you
8216  * can send an EOF and collect resulting output (e.g. `plink
8217  * hostname sort').
8218  */
8219 static void ssh_special(void *handle, Telnet_Special code)
8220 {
8221     Ssh ssh = (Ssh) handle;
8222     struct Packet *pktout;
8223
8224     if (code == TS_EOF) {
8225         if (ssh->state != SSH_STATE_SESSION) {
8226             /*
8227              * Buffer the EOF in case we are pre-SESSION, so we can
8228              * send it as soon as we reach SESSION.
8229              */
8230             if (code == TS_EOF)
8231                 ssh->eof_needed = TRUE;
8232             return;
8233         }
8234         if (ssh->version == 1) {
8235             send_packet(ssh, SSH1_CMSG_EOF, PKT_END);
8236         } else if (ssh->mainchan) {
8237             struct Packet *pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_EOF);
8238             ssh2_pkt_adduint32(pktout, ssh->mainchan->remoteid);
8239             ssh2_pkt_send(ssh, pktout);
8240             ssh->send_ok = 0;          /* now stop trying to read from stdin */
8241         }
8242         logevent("Sent EOF message");
8243     } else if (code == TS_PING || code == TS_NOP) {
8244         if (ssh->state == SSH_STATE_CLOSED
8245             || ssh->state == SSH_STATE_PREPACKET) return;
8246         if (ssh->version == 1) {
8247             if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH1_IGNORE))
8248                 send_packet(ssh, SSH1_MSG_IGNORE, PKT_STR, "", PKT_END);
8249         } else {
8250             pktout = ssh2_pkt_init(SSH2_MSG_IGNORE);
8251             ssh2_pkt_addstring_start(pktout);
8252             ssh2_pkt_send_noqueue(ssh, pktout);
8253         }
8254     } else if (code == TS_REKEY) {
8255         if (!ssh->kex_in_progress && ssh->version == 2) {
8256             do_ssh2_transport(ssh, "at user request", -1, NULL);
8257         }
8258     } else if (code == TS_BRK) {
8259         if (ssh->state == SSH_STATE_CLOSED
8260             || ssh->state == SSH_STATE_PREPACKET) return;
8261         if (ssh->version == 1) {
8262             logevent("Unable to send BREAK signal in SSH-1");
8263         } else if (ssh->mainchan) {
8264             pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
8265             ssh2_pkt_adduint32(pktout, ssh->mainchan->remoteid);
8266             ssh2_pkt_addstring(pktout, "break");
8267             ssh2_pkt_addbool(pktout, 0);
8268             ssh2_pkt_adduint32(pktout, 0);   /* default break length */
8269             ssh2_pkt_send(ssh, pktout);
8270         }
8271     } else {
8272         /* Is is a POSIX signal? */
8273         char *signame = NULL;
8274         if (code == TS_SIGABRT) signame = "ABRT";
8275         if (code == TS_SIGALRM) signame = "ALRM";
8276         if (code == TS_SIGFPE)  signame = "FPE";
8277         if (code == TS_SIGHUP)  signame = "HUP";
8278         if (code == TS_SIGILL)  signame = "ILL";
8279         if (code == TS_SIGINT)  signame = "INT";
8280         if (code == TS_SIGKILL) signame = "KILL";
8281         if (code == TS_SIGPIPE) signame = "PIPE";
8282         if (code == TS_SIGQUIT) signame = "QUIT";
8283         if (code == TS_SIGSEGV) signame = "SEGV";
8284         if (code == TS_SIGTERM) signame = "TERM";
8285         if (code == TS_SIGUSR1) signame = "USR1";
8286         if (code == TS_SIGUSR2) signame = "USR2";
8287         /* The SSH-2 protocol does in principle support arbitrary named
8288          * signals, including signame@domain, but we don't support those. */
8289         if (signame) {
8290             /* It's a signal. */
8291             if (ssh->version == 2 && ssh->mainchan) {
8292                 pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
8293                 ssh2_pkt_adduint32(pktout, ssh->mainchan->remoteid);
8294                 ssh2_pkt_addstring(pktout, "signal");
8295                 ssh2_pkt_addbool(pktout, 0);
8296                 ssh2_pkt_addstring(pktout, signame);
8297                 ssh2_pkt_send(ssh, pktout);
8298                 logeventf(ssh, "Sent signal SIG%s", signame);
8299             }
8300         } else {
8301             /* Never heard of it. Do nothing */
8302         }
8303     }
8304 }
8305
8306 void *new_sock_channel(void *handle, Socket s)
8307 {
8308     Ssh ssh = (Ssh) handle;
8309     struct ssh_channel *c;
8310     c = snew(struct ssh_channel);
8311     c->ssh = ssh;
8312
8313     if (c) {
8314         c->halfopen = TRUE;
8315         c->localid = alloc_channel_id(ssh);
8316         c->closes = 0;
8317         c->type = CHAN_SOCKDATA_DORMANT;/* identify channel type */
8318         c->u.pfd.s = s;
8319         bufchain_init(&c->v.v2.outbuffer);
8320         add234(ssh->channels, c);
8321     }
8322     return c;
8323 }
8324
8325 /*
8326  * This is called when stdout/stderr (the entity to which
8327  * from_backend sends data) manages to clear some backlog.
8328  */
8329 static void ssh_unthrottle(void *handle, int bufsize)
8330 {
8331     Ssh ssh = (Ssh) handle;
8332     if (ssh->version == 1) {
8333         if (ssh->v1_stdout_throttling && bufsize < SSH1_BUFFER_LIMIT) {
8334             ssh->v1_stdout_throttling = 0;
8335             ssh1_throttle(ssh, -1);
8336         }
8337     } else {
8338         if (ssh->mainchan && ssh->mainchan->closes == 0)
8339             ssh2_set_window(ssh->mainchan, OUR_V2_WINSIZE - bufsize);
8340     }
8341 }
8342
8343 void ssh_send_port_open(void *channel, char *hostname, int port, char *org)
8344 {
8345     struct ssh_channel *c = (struct ssh_channel *)channel;
8346     Ssh ssh = c->ssh;
8347     struct Packet *pktout;
8348
8349     logeventf(ssh, "Opening forwarded connection to %s:%d", hostname, port);
8350
8351     if (ssh->version == 1) {
8352         send_packet(ssh, SSH1_MSG_PORT_OPEN,
8353                     PKT_INT, c->localid,
8354                     PKT_STR, hostname,
8355                     PKT_INT, port,
8356                     /* PKT_STR, <org:orgport>, */
8357                     PKT_END);
8358     } else {
8359         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN);
8360         ssh2_pkt_addstring(pktout, "direct-tcpip");
8361         ssh2_pkt_adduint32(pktout, c->localid);
8362         c->v.v2.locwindow = OUR_V2_WINSIZE;
8363         ssh2_pkt_adduint32(pktout, c->v.v2.locwindow);/* our window size */
8364         ssh2_pkt_adduint32(pktout, OUR_V2_MAXPKT);      /* our max pkt size */
8365         ssh2_pkt_addstring(pktout, hostname);
8366         ssh2_pkt_adduint32(pktout, port);
8367         /*
8368          * We make up values for the originator data; partly it's
8369          * too much hassle to keep track, and partly I'm not
8370          * convinced the server should be told details like that
8371          * about my local network configuration.
8372          */
8373         ssh2_pkt_addstring(pktout, "client-side-connection");
8374         ssh2_pkt_adduint32(pktout, 0);
8375         ssh2_pkt_send(ssh, pktout);
8376     }
8377 }
8378
8379 static Socket ssh_socket(void *handle)
8380 {
8381     Ssh ssh = (Ssh) handle;
8382     return ssh->s;
8383 }
8384
8385 static int ssh_sendok(void *handle)
8386 {
8387     Ssh ssh = (Ssh) handle;
8388     return ssh->send_ok;
8389 }
8390
8391 static int ssh_ldisc(void *handle, int option)
8392 {
8393     Ssh ssh = (Ssh) handle;
8394     if (option == LD_ECHO)
8395         return ssh->echoing;
8396     if (option == LD_EDIT)
8397         return ssh->editing;
8398     return FALSE;
8399 }
8400
8401 static void ssh_provide_ldisc(void *handle, void *ldisc)
8402 {
8403     Ssh ssh = (Ssh) handle;
8404     ssh->ldisc = ldisc;
8405 }
8406
8407 static void ssh_provide_logctx(void *handle, void *logctx)
8408 {
8409     Ssh ssh = (Ssh) handle;
8410     ssh->logctx = logctx;
8411 }
8412
8413 static int ssh_return_exitcode(void *handle)
8414 {
8415     Ssh ssh = (Ssh) handle;
8416     if (ssh->s != NULL)
8417         return -1;
8418     else
8419         return (ssh->exitcode >= 0 ? ssh->exitcode : 0);
8420 }
8421
8422 /*
8423  * cfg_info for SSH is the currently running version of the
8424  * protocol. (1 for 1; 2 for 2; 0 for not-decided-yet.)
8425  */
8426 static int ssh_cfg_info(void *handle)
8427 {
8428     Ssh ssh = (Ssh) handle;
8429     return ssh->version;
8430 }
8431
8432 /*
8433  * Gross hack: pscp will try to start SFTP but fall back to scp1 if
8434  * that fails. This variable is the means by which scp.c can reach
8435  * into the SSH code and find out which one it got.
8436  */
8437 extern int ssh_fallback_cmd(void *handle)
8438 {
8439     Ssh ssh = (Ssh) handle;
8440     return ssh->fallback_cmd;
8441 }
8442
8443 Backend ssh_backend = {
8444     ssh_init,
8445     ssh_free,
8446     ssh_reconfig,
8447     ssh_send,
8448     ssh_sendbuffer,
8449     ssh_size,
8450     ssh_special,
8451     ssh_get_specials,
8452     ssh_socket,
8453     ssh_return_exitcode,
8454     ssh_sendok,
8455     ssh_ldisc,
8456     ssh_provide_ldisc,
8457     ssh_provide_logctx,
8458     ssh_unthrottle,
8459     ssh_cfg_info,
8460     22
8461 };