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