]> asedeno.scripts.mit.edu Git - PuTTY.git/blob - ssh.c
e2b2d0e3dd34e63f900df889a26e0889cc657903
[PuTTY.git] / ssh.c
1 /*
2  * SSH backend.
3  */
4
5 #include <stdio.h>
6 #include <stdlib.h>
7 #include <stdarg.h>
8 #include <assert.h>
9 #include <limits.h>
10 #include <signal.h>
11
12 #include "putty.h"
13 #include "tree234.h"
14 #include "storage.h"
15 #include "ssh.h"
16 #ifndef NO_GSSAPI
17 #include "sshgssc.h"
18 #include "sshgss.h"
19 #endif
20
21 #ifndef FALSE
22 #define FALSE 0
23 #endif
24 #ifndef TRUE
25 #define TRUE 1
26 #endif
27
28 /*
29  * Packet type contexts, so that ssh2_pkt_type can correctly decode
30  * the ambiguous type numbers back into the correct type strings.
31  */
32 typedef enum {
33     SSH2_PKTCTX_NOKEX,
34     SSH2_PKTCTX_DHGROUP,
35     SSH2_PKTCTX_DHGEX,
36     SSH2_PKTCTX_ECDHKEX,
37     SSH2_PKTCTX_RSAKEX
38 } Pkt_KCtx;
39 typedef enum {
40     SSH2_PKTCTX_NOAUTH,
41     SSH2_PKTCTX_PUBLICKEY,
42     SSH2_PKTCTX_PASSWORD,
43     SSH2_PKTCTX_GSSAPI,
44     SSH2_PKTCTX_KBDINTER
45 } Pkt_ACtx;
46
47 static const char *const ssh2_disconnect_reasons[] = {
48     NULL,
49     "host not allowed to connect",
50     "protocol error",
51     "key exchange failed",
52     "host authentication failed",
53     "MAC error",
54     "compression error",
55     "service not available",
56     "protocol version not supported",
57     "host key not verifiable",
58     "connection lost",
59     "by application",
60     "too many connections",
61     "auth cancelled by user",
62     "no more auth methods available",
63     "illegal user name",
64 };
65
66 /*
67  * Various remote-bug flags.
68  */
69 #define BUG_CHOKES_ON_SSH1_IGNORE                 1
70 #define BUG_SSH2_HMAC                             2
71 #define BUG_NEEDS_SSH1_PLAIN_PASSWORD             4
72 #define BUG_CHOKES_ON_RSA                         8
73 #define BUG_SSH2_RSA_PADDING                     16
74 #define BUG_SSH2_DERIVEKEY                       32
75 #define BUG_SSH2_REKEY                           64
76 #define BUG_SSH2_PK_SESSIONID                   128
77 #define BUG_SSH2_MAXPKT                         256
78 #define BUG_CHOKES_ON_SSH2_IGNORE               512
79 #define BUG_CHOKES_ON_WINADJ                   1024
80 #define BUG_SENDS_LATE_REQUEST_REPLY           2048
81 #define BUG_SSH2_OLDGEX                        4096
82
83 #define DH_MIN_SIZE 1024
84 #define DH_MAX_SIZE 8192
85
86 /*
87  * Codes for terminal modes.
88  * Most of these are the same in SSH-1 and SSH-2.
89  * This list is derived from RFC 4254 and
90  * SSH-1 RFC-1.2.31.
91  */
92 static const struct ssh_ttymode {
93     const char* const mode;
94     int opcode;
95     enum { TTY_OP_CHAR, TTY_OP_BOOL } type;
96 } ssh_ttymodes[] = {
97     /* "V" prefix discarded for special characters relative to SSH specs */
98     { "INTR",         1, TTY_OP_CHAR },
99     { "QUIT",         2, TTY_OP_CHAR },
100     { "ERASE",        3, TTY_OP_CHAR },
101     { "KILL",         4, TTY_OP_CHAR },
102     { "EOF",          5, TTY_OP_CHAR },
103     { "EOL",          6, TTY_OP_CHAR },
104     { "EOL2",         7, TTY_OP_CHAR },
105     { "START",        8, TTY_OP_CHAR },
106     { "STOP",         9, TTY_OP_CHAR },
107     { "SUSP",        10, TTY_OP_CHAR },
108     { "DSUSP",       11, TTY_OP_CHAR },
109     { "REPRINT",     12, TTY_OP_CHAR },
110     { "WERASE",      13, TTY_OP_CHAR },
111     { "LNEXT",       14, TTY_OP_CHAR },
112     { "FLUSH",       15, TTY_OP_CHAR },
113     { "SWTCH",       16, TTY_OP_CHAR },
114     { "STATUS",      17, TTY_OP_CHAR },
115     { "DISCARD",     18, TTY_OP_CHAR },
116     { "IGNPAR",      30, TTY_OP_BOOL },
117     { "PARMRK",      31, TTY_OP_BOOL },
118     { "INPCK",       32, TTY_OP_BOOL },
119     { "ISTRIP",      33, TTY_OP_BOOL },
120     { "INLCR",       34, TTY_OP_BOOL },
121     { "IGNCR",       35, TTY_OP_BOOL },
122     { "ICRNL",       36, TTY_OP_BOOL },
123     { "IUCLC",       37, TTY_OP_BOOL },
124     { "IXON",        38, TTY_OP_BOOL },
125     { "IXANY",       39, TTY_OP_BOOL },
126     { "IXOFF",       40, TTY_OP_BOOL },
127     { "IMAXBEL",     41, TTY_OP_BOOL },
128     { "IUTF8",       42, TTY_OP_BOOL },
129     { "ISIG",        50, TTY_OP_BOOL },
130     { "ICANON",      51, TTY_OP_BOOL },
131     { "XCASE",       52, TTY_OP_BOOL },
132     { "ECHO",        53, TTY_OP_BOOL },
133     { "ECHOE",       54, TTY_OP_BOOL },
134     { "ECHOK",       55, TTY_OP_BOOL },
135     { "ECHONL",      56, TTY_OP_BOOL },
136     { "NOFLSH",      57, TTY_OP_BOOL },
137     { "TOSTOP",      58, TTY_OP_BOOL },
138     { "IEXTEN",      59, TTY_OP_BOOL },
139     { "ECHOCTL",     60, TTY_OP_BOOL },
140     { "ECHOKE",      61, TTY_OP_BOOL },
141     { "PENDIN",      62, TTY_OP_BOOL }, /* XXX is this a real mode? */
142     { "OPOST",       70, TTY_OP_BOOL },
143     { "OLCUC",       71, TTY_OP_BOOL },
144     { "ONLCR",       72, TTY_OP_BOOL },
145     { "OCRNL",       73, TTY_OP_BOOL },
146     { "ONOCR",       74, TTY_OP_BOOL },
147     { "ONLRET",      75, TTY_OP_BOOL },
148     { "CS7",         90, TTY_OP_BOOL },
149     { "CS8",         91, TTY_OP_BOOL },
150     { "PARENB",      92, TTY_OP_BOOL },
151     { "PARODD",      93, TTY_OP_BOOL }
152 };
153
154 /* Miscellaneous other tty-related constants. */
155 #define SSH_TTY_OP_END            0
156 /* The opcodes for ISPEED/OSPEED differ between SSH-1 and SSH-2. */
157 #define SSH1_TTY_OP_ISPEED      192
158 #define SSH1_TTY_OP_OSPEED      193
159 #define SSH2_TTY_OP_ISPEED      128
160 #define SSH2_TTY_OP_OSPEED      129
161
162 /* Helper functions for parsing tty-related config. */
163 static unsigned int ssh_tty_parse_specchar(char *s)
164 {
165     unsigned int ret;
166     if (*s) {
167         char *next = NULL;
168         ret = ctrlparse(s, &next);
169         if (!next) ret = s[0];
170     } else {
171         ret = 255; /* special value meaning "don't set" */
172     }
173     return ret;
174 }
175 static unsigned int ssh_tty_parse_boolean(char *s)
176 {
177     if (stricmp(s, "yes") == 0 ||
178         stricmp(s, "on") == 0 ||
179         stricmp(s, "true") == 0 ||
180         stricmp(s, "+") == 0)
181         return 1; /* true */
182     else if (stricmp(s, "no") == 0 ||
183              stricmp(s, "off") == 0 ||
184              stricmp(s, "false") == 0 ||
185              stricmp(s, "-") == 0)
186         return 0; /* false */
187     else
188         return (atoi(s) != 0);
189 }
190
191 #define translate(x) if (type == x) return #x
192 #define translatek(x,ctx) if (type == x && (pkt_kctx == ctx)) return #x
193 #define translatea(x,ctx) if (type == x && (pkt_actx == ctx)) return #x
194 static const char *ssh1_pkt_type(int type)
195 {
196     translate(SSH1_MSG_DISCONNECT);
197     translate(SSH1_SMSG_PUBLIC_KEY);
198     translate(SSH1_CMSG_SESSION_KEY);
199     translate(SSH1_CMSG_USER);
200     translate(SSH1_CMSG_AUTH_RSA);
201     translate(SSH1_SMSG_AUTH_RSA_CHALLENGE);
202     translate(SSH1_CMSG_AUTH_RSA_RESPONSE);
203     translate(SSH1_CMSG_AUTH_PASSWORD);
204     translate(SSH1_CMSG_REQUEST_PTY);
205     translate(SSH1_CMSG_WINDOW_SIZE);
206     translate(SSH1_CMSG_EXEC_SHELL);
207     translate(SSH1_CMSG_EXEC_CMD);
208     translate(SSH1_SMSG_SUCCESS);
209     translate(SSH1_SMSG_FAILURE);
210     translate(SSH1_CMSG_STDIN_DATA);
211     translate(SSH1_SMSG_STDOUT_DATA);
212     translate(SSH1_SMSG_STDERR_DATA);
213     translate(SSH1_CMSG_EOF);
214     translate(SSH1_SMSG_EXIT_STATUS);
215     translate(SSH1_MSG_CHANNEL_OPEN_CONFIRMATION);
216     translate(SSH1_MSG_CHANNEL_OPEN_FAILURE);
217     translate(SSH1_MSG_CHANNEL_DATA);
218     translate(SSH1_MSG_CHANNEL_CLOSE);
219     translate(SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION);
220     translate(SSH1_SMSG_X11_OPEN);
221     translate(SSH1_CMSG_PORT_FORWARD_REQUEST);
222     translate(SSH1_MSG_PORT_OPEN);
223     translate(SSH1_CMSG_AGENT_REQUEST_FORWARDING);
224     translate(SSH1_SMSG_AGENT_OPEN);
225     translate(SSH1_MSG_IGNORE);
226     translate(SSH1_CMSG_EXIT_CONFIRMATION);
227     translate(SSH1_CMSG_X11_REQUEST_FORWARDING);
228     translate(SSH1_CMSG_AUTH_RHOSTS_RSA);
229     translate(SSH1_MSG_DEBUG);
230     translate(SSH1_CMSG_REQUEST_COMPRESSION);
231     translate(SSH1_CMSG_AUTH_TIS);
232     translate(SSH1_SMSG_AUTH_TIS_CHALLENGE);
233     translate(SSH1_CMSG_AUTH_TIS_RESPONSE);
234     translate(SSH1_CMSG_AUTH_CCARD);
235     translate(SSH1_SMSG_AUTH_CCARD_CHALLENGE);
236     translate(SSH1_CMSG_AUTH_CCARD_RESPONSE);
237     return "unknown";
238 }
239 static const char *ssh2_pkt_type(Pkt_KCtx pkt_kctx, Pkt_ACtx pkt_actx,
240                                  int type)
241 {
242     translatea(SSH2_MSG_USERAUTH_GSSAPI_RESPONSE,SSH2_PKTCTX_GSSAPI);
243     translatea(SSH2_MSG_USERAUTH_GSSAPI_TOKEN,SSH2_PKTCTX_GSSAPI);
244     translatea(SSH2_MSG_USERAUTH_GSSAPI_EXCHANGE_COMPLETE,SSH2_PKTCTX_GSSAPI);
245     translatea(SSH2_MSG_USERAUTH_GSSAPI_ERROR,SSH2_PKTCTX_GSSAPI);
246     translatea(SSH2_MSG_USERAUTH_GSSAPI_ERRTOK,SSH2_PKTCTX_GSSAPI);
247     translatea(SSH2_MSG_USERAUTH_GSSAPI_MIC, SSH2_PKTCTX_GSSAPI);
248     translate(SSH2_MSG_DISCONNECT);
249     translate(SSH2_MSG_IGNORE);
250     translate(SSH2_MSG_UNIMPLEMENTED);
251     translate(SSH2_MSG_DEBUG);
252     translate(SSH2_MSG_SERVICE_REQUEST);
253     translate(SSH2_MSG_SERVICE_ACCEPT);
254     translate(SSH2_MSG_KEXINIT);
255     translate(SSH2_MSG_NEWKEYS);
256     translatek(SSH2_MSG_KEXDH_INIT, SSH2_PKTCTX_DHGROUP);
257     translatek(SSH2_MSG_KEXDH_REPLY, SSH2_PKTCTX_DHGROUP);
258     translatek(SSH2_MSG_KEX_DH_GEX_REQUEST_OLD, SSH2_PKTCTX_DHGEX);
259     translatek(SSH2_MSG_KEX_DH_GEX_REQUEST, SSH2_PKTCTX_DHGEX);
260     translatek(SSH2_MSG_KEX_DH_GEX_GROUP, SSH2_PKTCTX_DHGEX);
261     translatek(SSH2_MSG_KEX_DH_GEX_INIT, SSH2_PKTCTX_DHGEX);
262     translatek(SSH2_MSG_KEX_DH_GEX_REPLY, SSH2_PKTCTX_DHGEX);
263     translatek(SSH2_MSG_KEXRSA_PUBKEY, SSH2_PKTCTX_RSAKEX);
264     translatek(SSH2_MSG_KEXRSA_SECRET, SSH2_PKTCTX_RSAKEX);
265     translatek(SSH2_MSG_KEXRSA_DONE, SSH2_PKTCTX_RSAKEX);
266     translatek(SSH2_MSG_KEX_ECDH_INIT, SSH2_PKTCTX_ECDHKEX);
267     translatek(SSH2_MSG_KEX_ECDH_REPLY, SSH2_PKTCTX_ECDHKEX);
268     translate(SSH2_MSG_USERAUTH_REQUEST);
269     translate(SSH2_MSG_USERAUTH_FAILURE);
270     translate(SSH2_MSG_USERAUTH_SUCCESS);
271     translate(SSH2_MSG_USERAUTH_BANNER);
272     translatea(SSH2_MSG_USERAUTH_PK_OK, SSH2_PKTCTX_PUBLICKEY);
273     translatea(SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ, SSH2_PKTCTX_PASSWORD);
274     translatea(SSH2_MSG_USERAUTH_INFO_REQUEST, SSH2_PKTCTX_KBDINTER);
275     translatea(SSH2_MSG_USERAUTH_INFO_RESPONSE, SSH2_PKTCTX_KBDINTER);
276     translate(SSH2_MSG_GLOBAL_REQUEST);
277     translate(SSH2_MSG_REQUEST_SUCCESS);
278     translate(SSH2_MSG_REQUEST_FAILURE);
279     translate(SSH2_MSG_CHANNEL_OPEN);
280     translate(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION);
281     translate(SSH2_MSG_CHANNEL_OPEN_FAILURE);
282     translate(SSH2_MSG_CHANNEL_WINDOW_ADJUST);
283     translate(SSH2_MSG_CHANNEL_DATA);
284     translate(SSH2_MSG_CHANNEL_EXTENDED_DATA);
285     translate(SSH2_MSG_CHANNEL_EOF);
286     translate(SSH2_MSG_CHANNEL_CLOSE);
287     translate(SSH2_MSG_CHANNEL_REQUEST);
288     translate(SSH2_MSG_CHANNEL_SUCCESS);
289     translate(SSH2_MSG_CHANNEL_FAILURE);
290     return "unknown";
291 }
292 #undef translate
293 #undef translatec
294
295 /* Enumeration values for fields in SSH-1 packets */
296 enum {
297     PKT_END, PKT_INT, PKT_CHAR, PKT_DATA, PKT_STR, PKT_BIGNUM,
298 };
299
300 /*
301  * Coroutine mechanics for the sillier bits of the code. If these
302  * macros look impenetrable to you, you might find it helpful to
303  * read
304  * 
305  *   http://www.chiark.greenend.org.uk/~sgtatham/coroutines.html
306  * 
307  * which explains the theory behind these macros.
308  * 
309  * In particular, if you are getting `case expression not constant'
310  * errors when building with MS Visual Studio, this is because MS's
311  * Edit and Continue debugging feature causes their compiler to
312  * violate ANSI C. To disable Edit and Continue debugging:
313  * 
314  *  - right-click ssh.c in the FileView
315  *  - click Settings
316  *  - select the C/C++ tab and the General category
317  *  - under `Debug info:', select anything _other_ than `Program
318  *    Database for Edit and Continue'.
319  */
320 #define crBegin(v)      { int *crLine = &v; switch(v) { case 0:;
321 #define crBeginState    crBegin(s->crLine)
322 #define crStateP(t, v)                          \
323     struct t *s;                                \
324     if (!(v)) { s = (v) = snew(struct t); s->crLine = 0; }      \
325     s = (v);
326 #define crState(t)      crStateP(t, ssh->t)
327 #define crFinish(z)     } *crLine = 0; return (z); }
328 #define crFinishV       } *crLine = 0; return; }
329 #define crFinishFree(z) } sfree(s); return (z); }
330 #define crFinishFreeV   } sfree(s); return; }
331 #define crReturn(z)     \
332         do {\
333             *crLine =__LINE__; return (z); case __LINE__:;\
334         } while (0)
335 #define crReturnV       \
336         do {\
337             *crLine=__LINE__; return; case __LINE__:;\
338         } while (0)
339 #define crStop(z)       do{ *crLine = 0; return (z); }while(0)
340 #define crStopV         do{ *crLine = 0; return; }while(0)
341 #define crWaitUntil(c)  do { crReturn(0); } while (!(c))
342 #define crWaitUntilV(c) do { crReturnV; } while (!(c))
343
344 struct Packet;
345
346 static struct Packet *ssh1_pkt_init(int pkt_type);
347 static struct Packet *ssh2_pkt_init(int pkt_type);
348 static void ssh_pkt_ensure(struct Packet *, int length);
349 static void ssh_pkt_adddata(struct Packet *, const void *data, int len);
350 static void ssh_pkt_addbyte(struct Packet *, unsigned char value);
351 static void ssh2_pkt_addbool(struct Packet *, unsigned char value);
352 static void ssh_pkt_adduint32(struct Packet *, unsigned long value);
353 static void ssh_pkt_addstring_start(struct Packet *);
354 static void ssh_pkt_addstring_str(struct Packet *, const char *data);
355 static void ssh_pkt_addstring_data(struct Packet *, const char *data, int len);
356 static void ssh_pkt_addstring(struct Packet *, const char *data);
357 static unsigned char *ssh2_mpint_fmt(Bignum b, int *len);
358 static void ssh1_pkt_addmp(struct Packet *, Bignum b);
359 static void ssh2_pkt_addmp(struct Packet *, Bignum b);
360 static int ssh2_pkt_construct(Ssh, struct Packet *);
361 static void ssh2_pkt_send(Ssh, struct Packet *);
362 static void ssh2_pkt_send_noqueue(Ssh, struct Packet *);
363 static int do_ssh1_login(Ssh ssh, const unsigned char *in, int inlen,
364                          struct Packet *pktin);
365 static void do_ssh2_authconn(Ssh ssh, const unsigned char *in, int inlen,
366                              struct Packet *pktin);
367 static void ssh_channel_init(struct ssh_channel *c);
368 static struct ssh_channel *ssh_channel_msg(Ssh ssh, struct Packet *pktin);
369 static void ssh2_channel_check_close(struct ssh_channel *c);
370 static void ssh_channel_destroy(struct ssh_channel *c);
371 static void ssh_channel_unthrottle(struct ssh_channel *c, int bufsize);
372 static void ssh2_msg_something_unimplemented(Ssh ssh, struct Packet *pktin);
373
374 /*
375  * Buffer management constants. There are several of these for
376  * various different purposes:
377  * 
378  *  - SSH1_BUFFER_LIMIT is the amount of backlog that must build up
379  *    on a local data stream before we throttle the whole SSH
380  *    connection (in SSH-1 only). Throttling the whole connection is
381  *    pretty drastic so we set this high in the hope it won't
382  *    happen very often.
383  * 
384  *  - SSH_MAX_BACKLOG is the amount of backlog that must build up
385  *    on the SSH connection itself before we defensively throttle
386  *    _all_ local data streams. This is pretty drastic too (though
387  *    thankfully unlikely in SSH-2 since the window mechanism should
388  *    ensure that the server never has any need to throttle its end
389  *    of the connection), so we set this high as well.
390  * 
391  *  - OUR_V2_WINSIZE is the default window size we present on SSH-2
392  *    channels.
393  *
394  *  - OUR_V2_BIGWIN is the window size we advertise for the only
395  *    channel in a simple connection.  It must be <= INT_MAX.
396  *
397  *  - OUR_V2_MAXPKT is the official "maximum packet size" we send
398  *    to the remote side. This actually has nothing to do with the
399  *    size of the _packet_, but is instead a limit on the amount
400  *    of data we're willing to receive in a single SSH2 channel
401  *    data message.
402  *
403  *  - OUR_V2_PACKETLIMIT is actually the maximum size of SSH
404  *    _packet_ we're prepared to cope with.  It must be a multiple
405  *    of the cipher block size, and must be at least 35000.
406  */
407
408 #define SSH1_BUFFER_LIMIT 32768
409 #define SSH_MAX_BACKLOG 32768
410 #define OUR_V2_WINSIZE 16384
411 #define OUR_V2_BIGWIN 0x7fffffff
412 #define OUR_V2_MAXPKT 0x4000UL
413 #define OUR_V2_PACKETLIMIT 0x9000UL
414
415 struct ssh_signkey_with_user_pref_id {
416     const struct ssh_signkey *alg;
417     int id;
418 };
419 const static struct ssh_signkey_with_user_pref_id hostkey_algs[] = {
420     { &ssh_ecdsa_ed25519, HK_ED25519 },
421     { &ssh_ecdsa_nistp256, HK_ECDSA },
422     { &ssh_ecdsa_nistp384, HK_ECDSA },
423     { &ssh_ecdsa_nistp521, HK_ECDSA },
424     { &ssh_dss, HK_DSA },
425     { &ssh_rsa, HK_RSA },
426 };
427
428 const static struct ssh_mac *const macs[] = {
429     &ssh_hmac_sha256, &ssh_hmac_sha1, &ssh_hmac_sha1_96, &ssh_hmac_md5
430 };
431 const static struct ssh_mac *const buggymacs[] = {
432     &ssh_hmac_sha1_buggy, &ssh_hmac_sha1_96_buggy, &ssh_hmac_md5
433 };
434
435 static void *ssh_comp_none_init(void)
436 {
437     return NULL;
438 }
439 static void ssh_comp_none_cleanup(void *handle)
440 {
441 }
442 static int ssh_comp_none_block(void *handle, unsigned char *block, int len,
443                                unsigned char **outblock, int *outlen)
444 {
445     return 0;
446 }
447 static int ssh_comp_none_disable(void *handle)
448 {
449     return 0;
450 }
451 const static struct ssh_compress ssh_comp_none = {
452     "none", NULL,
453     ssh_comp_none_init, ssh_comp_none_cleanup, ssh_comp_none_block,
454     ssh_comp_none_init, ssh_comp_none_cleanup, ssh_comp_none_block,
455     ssh_comp_none_disable, NULL
456 };
457 extern const struct ssh_compress ssh_zlib;
458 const static struct ssh_compress *const compressions[] = {
459     &ssh_zlib, &ssh_comp_none
460 };
461
462 enum {                                 /* channel types */
463     CHAN_MAINSESSION,
464     CHAN_X11,
465     CHAN_AGENT,
466     CHAN_SOCKDATA,
467     CHAN_SOCKDATA_DORMANT,             /* one the remote hasn't confirmed */
468     /*
469      * CHAN_SHARING indicates a channel which is tracked here on
470      * behalf of a connection-sharing downstream. We do almost nothing
471      * with these channels ourselves: all messages relating to them
472      * get thrown straight to sshshare.c and passed on almost
473      * unmodified to downstream.
474      */
475     CHAN_SHARING,
476     /*
477      * CHAN_ZOMBIE is used to indicate a channel for which we've
478      * already destroyed the local data source: for instance, if a
479      * forwarded port experiences a socket error on the local side, we
480      * immediately destroy its local socket and turn the SSH channel
481      * into CHAN_ZOMBIE.
482      */
483     CHAN_ZOMBIE
484 };
485
486 typedef void (*handler_fn_t)(Ssh ssh, struct Packet *pktin);
487 typedef void (*chandler_fn_t)(Ssh ssh, struct Packet *pktin, void *ctx);
488 typedef void (*cchandler_fn_t)(struct ssh_channel *, struct Packet *, void *);
489
490 /*
491  * Each channel has a queue of outstanding CHANNEL_REQUESTS and their
492  * handlers.
493  */
494 struct outstanding_channel_request {
495     cchandler_fn_t handler;
496     void *ctx;
497     struct outstanding_channel_request *next;
498 };
499
500 /*
501  * 2-3-4 tree storing channels.
502  */
503 struct ssh_channel {
504     Ssh ssh;                           /* pointer back to main context */
505     unsigned remoteid, localid;
506     int type;
507     /* True if we opened this channel but server hasn't confirmed. */
508     int halfopen;
509     /*
510      * In SSH-1, this value contains four bits:
511      * 
512      *   1   We have sent SSH1_MSG_CHANNEL_CLOSE.
513      *   2   We have sent SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION.
514      *   4   We have received SSH1_MSG_CHANNEL_CLOSE.
515      *   8   We have received SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION.
516      * 
517      * A channel is completely finished with when all four bits are set.
518      *
519      * In SSH-2, the four bits mean:
520      *
521      *   1   We have sent SSH2_MSG_CHANNEL_EOF.
522      *   2   We have sent SSH2_MSG_CHANNEL_CLOSE.
523      *   4   We have received SSH2_MSG_CHANNEL_EOF.
524      *   8   We have received SSH2_MSG_CHANNEL_CLOSE.
525      *
526      * A channel is completely finished with when we have both sent
527      * and received CLOSE.
528      *
529      * The symbolic constants below use the SSH-2 terminology, which
530      * is a bit confusing in SSH-1, but we have to use _something_.
531      */
532 #define CLOSES_SENT_EOF    1
533 #define CLOSES_SENT_CLOSE  2
534 #define CLOSES_RCVD_EOF    4
535 #define CLOSES_RCVD_CLOSE  8
536     int closes;
537
538     /*
539      * This flag indicates that an EOF is pending on the outgoing side
540      * of the channel: that is, wherever we're getting the data for
541      * this channel has sent us some data followed by EOF. We can't
542      * actually send the EOF until we've finished sending the data, so
543      * we set this flag instead to remind us to do so once our buffer
544      * is clear.
545      */
546     int pending_eof;
547
548     /*
549      * True if this channel is causing the underlying connection to be
550      * throttled.
551      */
552     int throttling_conn;
553     union {
554         struct ssh2_data_channel {
555             bufchain outbuffer;
556             unsigned remwindow, remmaxpkt;
557             /* locwindow is signed so we can cope with excess data. */
558             int locwindow, locmaxwin;
559             /*
560              * remlocwin is the amount of local window that we think
561              * the remote end had available to it after it sent the
562              * last data packet or window adjust ack.
563              */
564             int remlocwin;
565             /*
566              * These store the list of channel requests that haven't
567              * been acked.
568              */
569             struct outstanding_channel_request *chanreq_head, *chanreq_tail;
570             enum { THROTTLED, UNTHROTTLING, UNTHROTTLED } throttle_state;
571         } v2;
572     } v;
573     union {
574         struct ssh_agent_channel {
575             unsigned char *message;
576             unsigned char msglen[4];
577             unsigned lensofar, totallen;
578             int outstanding_requests;
579         } a;
580         struct ssh_x11_channel {
581             struct X11Connection *xconn;
582             int initial;
583         } x11;
584         struct ssh_pfd_channel {
585             struct PortForwarding *pf;
586         } pfd;
587         struct ssh_sharing_channel {
588             void *ctx;
589         } sharing;
590     } u;
591 };
592
593 /*
594  * 2-3-4 tree storing remote->local port forwardings. SSH-1 and SSH-2
595  * use this structure in different ways, reflecting SSH-2's
596  * altogether saner approach to port forwarding.
597  * 
598  * In SSH-1, you arrange a remote forwarding by sending the server
599  * the remote port number, and the local destination host:port.
600  * When a connection comes in, the server sends you back that
601  * host:port pair, and you connect to it. This is a ready-made
602  * security hole if you're not on the ball: a malicious server
603  * could send you back _any_ host:port pair, so if you trustingly
604  * connect to the address it gives you then you've just opened the
605  * entire inside of your corporate network just by connecting
606  * through it to a dodgy SSH server. Hence, we must store a list of
607  * host:port pairs we _are_ trying to forward to, and reject a
608  * connection request from the server if it's not in the list.
609  * 
610  * In SSH-2, each side of the connection minds its own business and
611  * doesn't send unnecessary information to the other. You arrange a
612  * remote forwarding by sending the server just the remote port
613  * number. When a connection comes in, the server tells you which
614  * of its ports was connected to; and _you_ have to remember what
615  * local host:port pair went with that port number.
616  * 
617  * Hence, in SSH-1 this structure is indexed by destination
618  * host:port pair, whereas in SSH-2 it is indexed by source port.
619  */
620 struct ssh_portfwd; /* forward declaration */
621
622 struct ssh_rportfwd {
623     unsigned sport, dport;
624     char *shost, *dhost;
625     char *sportdesc;
626     void *share_ctx;
627     struct ssh_portfwd *pfrec;
628 };
629
630 static void free_rportfwd(struct ssh_rportfwd *pf)
631 {
632     if (pf) {
633         sfree(pf->sportdesc);
634         sfree(pf->shost);
635         sfree(pf->dhost);
636         sfree(pf);
637     }
638 }
639
640 /*
641  * Separately to the rportfwd tree (which is for looking up port
642  * open requests from the server), a tree of _these_ structures is
643  * used to keep track of all the currently open port forwardings,
644  * so that we can reconfigure in mid-session if the user requests
645  * it.
646  */
647 struct ssh_portfwd {
648     enum { DESTROY, KEEP, CREATE } status;
649     int type;
650     unsigned sport, dport;
651     char *saddr, *daddr;
652     char *sserv, *dserv;
653     struct ssh_rportfwd *remote;
654     int addressfamily;
655     struct PortListener *local;
656 };
657 #define free_portfwd(pf) ( \
658     ((pf) ? (sfree((pf)->saddr), sfree((pf)->daddr), \
659              sfree((pf)->sserv), sfree((pf)->dserv)) : (void)0 ), sfree(pf) )
660
661 struct Packet {
662     long length;            /* length of packet: see below */
663     long forcepad;          /* SSH-2: force padding to at least this length */
664     int type;               /* only used for incoming packets */
665     unsigned long sequence; /* SSH-2 incoming sequence number */
666     unsigned char *data;    /* allocated storage */
667     unsigned char *body;    /* offset of payload within `data' */
668     long savedpos;          /* dual-purpose saved packet position: see below */
669     long maxlen;            /* amount of storage allocated for `data' */
670     long encrypted_len;     /* for SSH-2 total-size counting */
671
672     /*
673      * A note on the 'length' and 'savedpos' fields above.
674      *
675      * Incoming packets are set up so that pkt->length is measured
676      * relative to pkt->body, which itself points to a few bytes after
677      * pkt->data (skipping some uninteresting header fields including
678      * the packet type code). The ssh_pkt_get* functions all expect
679      * this setup, and they also use pkt->savedpos to indicate how far
680      * through the packet being decoded they've got - and that, too,
681      * is an offset from pkt->body rather than pkt->data.
682      *
683      * During construction of an outgoing packet, however, pkt->length
684      * is measured relative to the base pointer pkt->data, and
685      * pkt->body is not really used for anything until the packet is
686      * ready for sending. In this mode, pkt->savedpos is reused as a
687      * temporary variable by the addstring functions, which write out
688      * a string length field and then keep going back and updating it
689      * as more data is appended to the subsequent string data field;
690      * pkt->savedpos stores the offset (again relative to pkt->data)
691      * of the start of the string data field.
692      */
693
694     /* Extra metadata used in SSH packet logging mode, allowing us to
695      * log in the packet header line that the packet came from a
696      * connection-sharing downstream and what if anything unusual was
697      * done to it. The additional_log_text field is expected to be a
698      * static string - it will not be freed. */
699     unsigned downstream_id;
700     const char *additional_log_text;
701 };
702
703 static void ssh1_protocol(Ssh ssh, const void *vin, int inlen,
704                           struct Packet *pktin);
705 static void ssh2_protocol(Ssh ssh, const void *vin, int inlen,
706                           struct Packet *pktin);
707 static void ssh2_bare_connection_protocol(Ssh ssh, const void *vin, int inlen,
708                                           struct Packet *pktin);
709 static void ssh1_protocol_setup(Ssh ssh);
710 static void ssh2_protocol_setup(Ssh ssh);
711 static void ssh2_bare_connection_protocol_setup(Ssh ssh);
712 static void ssh_size(void *handle, int width, int height);
713 static void ssh_special(void *handle, Telnet_Special);
714 static int ssh2_try_send(struct ssh_channel *c);
715 static int ssh_send_channel_data(struct ssh_channel *c,
716                                  const char *buf, int len);
717 static void ssh_throttle_all(Ssh ssh, int enable, int bufsize);
718 static void ssh2_set_window(struct ssh_channel *c, int newwin);
719 static int ssh_sendbuffer(void *handle);
720 static int ssh_do_close(Ssh ssh, int notify_exit);
721 static unsigned long ssh_pkt_getuint32(struct Packet *pkt);
722 static int ssh2_pkt_getbool(struct Packet *pkt);
723 static void ssh_pkt_getstring(struct Packet *pkt, char **p, int *length);
724 static void ssh2_timer(void *ctx, unsigned long now);
725 static void do_ssh2_transport(Ssh ssh, const void *vin, int inlen,
726                               struct Packet *pktin);
727 static void ssh2_msg_unexpected(Ssh ssh, struct Packet *pktin);
728
729 struct rdpkt1_state_tag {
730     long len, pad, biglen, to_read;
731     unsigned long realcrc, gotcrc;
732     unsigned char *p;
733     int i;
734     int chunk;
735     struct Packet *pktin;
736 };
737
738 struct rdpkt2_state_tag {
739     long len, pad, payload, packetlen, maclen;
740     int i;
741     int cipherblk;
742     unsigned long incoming_sequence;
743     struct Packet *pktin;
744 };
745
746 struct rdpkt2_bare_state_tag {
747     char length[4];
748     long packetlen;
749     int i;
750     unsigned long incoming_sequence;
751     struct Packet *pktin;
752 };
753
754 struct queued_handler;
755 struct queued_handler {
756     int msg1, msg2;
757     chandler_fn_t handler;
758     void *ctx;
759     struct queued_handler *next;
760 };
761
762 struct ssh_tag {
763     const struct plug_function_table *fn;
764     /* the above field _must_ be first in the structure */
765
766     char *v_c, *v_s;
767     void *exhash;
768
769     Socket s;
770
771     void *ldisc;
772     void *logctx;
773
774     unsigned char session_key[32];
775     int v1_compressing;
776     int v1_remote_protoflags;
777     int v1_local_protoflags;
778     int agentfwd_enabled;
779     int X11_fwd_enabled;
780     int remote_bugs;
781     const struct ssh_cipher *cipher;
782     void *v1_cipher_ctx;
783     void *crcda_ctx;
784     const struct ssh2_cipher *cscipher, *sccipher;
785     void *cs_cipher_ctx, *sc_cipher_ctx;
786     const struct ssh_mac *csmac, *scmac;
787     int csmac_etm, scmac_etm;
788     void *cs_mac_ctx, *sc_mac_ctx;
789     const struct ssh_compress *cscomp, *sccomp;
790     void *cs_comp_ctx, *sc_comp_ctx;
791     const struct ssh_kex *kex;
792     const struct ssh_signkey *hostkey;
793     char *hostkey_str; /* string representation, for easy checking in rekeys */
794     unsigned char v2_session_id[SSH2_KEX_MAX_HASH_LEN];
795     int v2_session_id_len;
796     void *kex_ctx;
797
798     int bare_connection;
799     int attempting_connshare;
800     void *connshare;
801
802     char *savedhost;
803     int savedport;
804     int send_ok;
805     int echoing, editing;
806
807     int session_started;
808     void *frontend;
809
810     int ospeed, ispeed;                /* temporaries */
811     int term_width, term_height;
812
813     tree234 *channels;                 /* indexed by local id */
814     struct ssh_channel *mainchan;      /* primary session channel */
815     int ncmode;                        /* is primary channel direct-tcpip? */
816     int exitcode;
817     int close_expected;
818     int clean_exit;
819
820     tree234 *rportfwds, *portfwds;
821
822     enum {
823         SSH_STATE_PREPACKET,
824         SSH_STATE_BEFORE_SIZE,
825         SSH_STATE_INTERMED,
826         SSH_STATE_SESSION,
827         SSH_STATE_CLOSED
828     } state;
829
830     int size_needed, eof_needed;
831     int sent_console_eof;
832     int got_pty;           /* affects EOF behaviour on main channel */
833
834     struct Packet **queue;
835     int queuelen, queuesize;
836     int queueing;
837     unsigned char *deferred_send_data;
838     int deferred_len, deferred_size;
839
840     /*
841      * Gross hack: pscp will try to start SFTP but fall back to
842      * scp1 if that fails. This variable is the means by which
843      * scp.c can reach into the SSH code and find out which one it
844      * got.
845      */
846     int fallback_cmd;
847
848     bufchain banner;    /* accumulates banners during do_ssh2_authconn */
849
850     Pkt_KCtx pkt_kctx;
851     Pkt_ACtx pkt_actx;
852
853     struct X11Display *x11disp;
854     struct X11FakeAuth *x11auth;
855     tree234 *x11authtree;
856
857     int version;
858     int conn_throttle_count;
859     int overall_bufsize;
860     int throttled_all;
861     int v1_stdout_throttling;
862     unsigned long v2_outgoing_sequence;
863
864     int ssh1_rdpkt_crstate;
865     int ssh2_rdpkt_crstate;
866     int ssh2_bare_rdpkt_crstate;
867     int ssh_gotdata_crstate;
868     int do_ssh1_connection_crstate;
869
870     void *do_ssh_init_state;
871     void *do_ssh1_login_state;
872     void *do_ssh2_transport_state;
873     void *do_ssh2_authconn_state;
874     void *do_ssh_connection_init_state;
875
876     struct rdpkt1_state_tag rdpkt1_state;
877     struct rdpkt2_state_tag rdpkt2_state;
878     struct rdpkt2_bare_state_tag rdpkt2_bare_state;
879
880     /* SSH-1 and SSH-2 use this for different things, but both use it */
881     int protocol_initial_phase_done;
882
883     void (*protocol) (Ssh ssh, const void *vin, int inlen,
884                       struct Packet *pkt);
885     struct Packet *(*s_rdpkt) (Ssh ssh, const unsigned char **data,
886                                int *datalen);
887     int (*do_ssh_init)(Ssh ssh, unsigned char c);
888
889     /*
890      * We maintain our own copy of a Conf structure here. That way,
891      * when we're passed a new one for reconfiguration, we can check
892      * the differences and potentially reconfigure port forwardings
893      * etc in mid-session.
894      */
895     Conf *conf;
896
897     /*
898      * Values cached out of conf so as to avoid the tree234 lookup
899      * cost every time they're used.
900      */
901     int logomitdata;
902
903     /*
904      * Dynamically allocated username string created during SSH
905      * login. Stored in here rather than in the coroutine state so
906      * that it'll be reliably freed if we shut down the SSH session
907      * at some unexpected moment.
908      */
909     char *username;
910
911     /*
912      * Used to transfer data back from async callbacks.
913      */
914     void *agent_response;
915     int agent_response_len;
916     int user_response;
917
918     /*
919      * The SSH connection can be set as `frozen', meaning we are
920      * not currently accepting incoming data from the network. This
921      * is slightly more serious than setting the _socket_ as
922      * frozen, because we may already have had data passed to us
923      * from the network which we need to delay processing until
924      * after the freeze is lifted, so we also need a bufchain to
925      * store that data.
926      */
927     int frozen;
928     bufchain queued_incoming_data;
929
930     /*
931      * Dispatch table for packet types that we may have to deal
932      * with at any time.
933      */
934     handler_fn_t packet_dispatch[256];
935
936     /*
937      * Queues of one-off handler functions for success/failure
938      * indications from a request.
939      */
940     struct queued_handler *qhead, *qtail;
941     handler_fn_t q_saved_handler1, q_saved_handler2;
942
943     /*
944      * This module deals with sending keepalives.
945      */
946     Pinger pinger;
947
948     /*
949      * Track incoming and outgoing data sizes and time, for
950      * size-based rekeys.
951      */
952     unsigned long incoming_data_size, outgoing_data_size, deferred_data_size;
953     unsigned long max_data_size;
954     int kex_in_progress;
955     unsigned long next_rekey, last_rekey;
956     const char *deferred_rekey_reason;
957
958     /*
959      * Fully qualified host name, which we need if doing GSSAPI.
960      */
961     char *fullhostname;
962
963 #ifndef NO_GSSAPI
964     /*
965      * GSSAPI libraries for this session.
966      */
967     struct ssh_gss_liblist *gsslibs;
968 #endif
969
970     /*
971      * The last list returned from get_specials.
972      */
973     struct telnet_special *specials;
974
975     /*
976      * List of host key algorithms for which we _don't_ have a stored
977      * host key. These are indices into the main hostkey_algs[] array
978      */
979     int uncert_hostkeys[lenof(hostkey_algs)];
980     int n_uncert_hostkeys;
981
982     /*
983      * Flag indicating that the current rekey is intended to finish
984      * with a newly cross-certified host key.
985      */
986     int cross_certifying;
987 };
988
989 static const char *ssh_pkt_type(Ssh ssh, int type)
990 {
991     if (ssh->version == 1)
992         return ssh1_pkt_type(type);
993     else
994         return ssh2_pkt_type(ssh->pkt_kctx, ssh->pkt_actx, type);
995 }
996
997 #define logevent(s) logevent(ssh->frontend, s)
998
999 /* logevent, only printf-formatted. */
1000 static void logeventf(Ssh ssh, const char *fmt, ...)
1001 {
1002     va_list ap;
1003     char *buf;
1004
1005     va_start(ap, fmt);
1006     buf = dupvprintf(fmt, ap);
1007     va_end(ap);
1008     logevent(buf);
1009     sfree(buf);
1010 }
1011
1012 static void bomb_out(Ssh ssh, char *text)
1013 {
1014     ssh_do_close(ssh, FALSE);
1015     logevent(text);
1016     connection_fatal(ssh->frontend, "%s", text);
1017     sfree(text);
1018 }
1019
1020 #define bombout(msg) bomb_out(ssh, dupprintf msg)
1021
1022 /* Helper function for common bits of parsing ttymodes. */
1023 static void parse_ttymodes(Ssh ssh,
1024                            void (*do_mode)(void *data,
1025                                            const struct ssh_ttymode *mode,
1026                                            char *val),
1027                            void *data)
1028 {
1029     int i;
1030     const struct ssh_ttymode *mode;
1031     char *val;
1032     char default_val[2];
1033
1034     strcpy(default_val, "A");
1035
1036     for (i = 0; i < lenof(ssh_ttymodes); i++) {
1037         mode = ssh_ttymodes + i;
1038         val = conf_get_str_str_opt(ssh->conf, CONF_ttymodes, mode->mode);
1039         if (!val)
1040             val = default_val;
1041
1042         /*
1043          * val[0] is either 'V', indicating that an explicit value
1044          * follows it, or 'A' indicating that we should pass the
1045          * value through from the local environment via get_ttymode.
1046          */
1047         if (val[0] == 'A') {
1048             val = get_ttymode(ssh->frontend, mode->mode);
1049             if (val) {
1050                 do_mode(data, mode, val);
1051                 sfree(val);
1052             }
1053         } else
1054             do_mode(data, mode, val + 1);              /* skip the 'V' */
1055     }
1056 }
1057
1058 static int ssh_channelcmp(void *av, void *bv)
1059 {
1060     struct ssh_channel *a = (struct ssh_channel *) av;
1061     struct ssh_channel *b = (struct ssh_channel *) bv;
1062     if (a->localid < b->localid)
1063         return -1;
1064     if (a->localid > b->localid)
1065         return +1;
1066     return 0;
1067 }
1068 static int ssh_channelfind(void *av, void *bv)
1069 {
1070     unsigned *a = (unsigned *) av;
1071     struct ssh_channel *b = (struct ssh_channel *) bv;
1072     if (*a < b->localid)
1073         return -1;
1074     if (*a > b->localid)
1075         return +1;
1076     return 0;
1077 }
1078
1079 static int ssh_rportcmp_ssh1(void *av, void *bv)
1080 {
1081     struct ssh_rportfwd *a = (struct ssh_rportfwd *) av;
1082     struct ssh_rportfwd *b = (struct ssh_rportfwd *) bv;
1083     int i;
1084     if ( (i = strcmp(a->dhost, b->dhost)) != 0)
1085         return i < 0 ? -1 : +1;
1086     if (a->dport > b->dport)
1087         return +1;
1088     if (a->dport < b->dport)
1089         return -1;
1090     return 0;
1091 }
1092
1093 static int ssh_rportcmp_ssh2(void *av, void *bv)
1094 {
1095     struct ssh_rportfwd *a = (struct ssh_rportfwd *) av;
1096     struct ssh_rportfwd *b = (struct ssh_rportfwd *) bv;
1097     int i;
1098     if ( (i = strcmp(a->shost, b->shost)) != 0)
1099         return i < 0 ? -1 : +1;
1100     if (a->sport > b->sport)
1101         return +1;
1102     if (a->sport < b->sport)
1103         return -1;
1104     return 0;
1105 }
1106
1107 /*
1108  * Special form of strcmp which can cope with NULL inputs. NULL is
1109  * defined to sort before even the empty string.
1110  */
1111 static int nullstrcmp(const char *a, const char *b)
1112 {
1113     if (a == NULL && b == NULL)
1114         return 0;
1115     if (a == NULL)
1116         return -1;
1117     if (b == NULL)
1118         return +1;
1119     return strcmp(a, b);
1120 }
1121
1122 static int ssh_portcmp(void *av, void *bv)
1123 {
1124     struct ssh_portfwd *a = (struct ssh_portfwd *) av;
1125     struct ssh_portfwd *b = (struct ssh_portfwd *) bv;
1126     int i;
1127     if (a->type > b->type)
1128         return +1;
1129     if (a->type < b->type)
1130         return -1;
1131     if (a->addressfamily > b->addressfamily)
1132         return +1;
1133     if (a->addressfamily < b->addressfamily)
1134         return -1;
1135     if ( (i = nullstrcmp(a->saddr, b->saddr)) != 0)
1136         return i < 0 ? -1 : +1;
1137     if (a->sport > b->sport)
1138         return +1;
1139     if (a->sport < b->sport)
1140         return -1;
1141     if (a->type != 'D') {
1142         if ( (i = nullstrcmp(a->daddr, b->daddr)) != 0)
1143             return i < 0 ? -1 : +1;
1144         if (a->dport > b->dport)
1145             return +1;
1146         if (a->dport < b->dport)
1147             return -1;
1148     }
1149     return 0;
1150 }
1151
1152 static int alloc_channel_id(Ssh ssh)
1153 {
1154     const unsigned CHANNEL_NUMBER_OFFSET = 256;
1155     unsigned low, high, mid;
1156     int tsize;
1157     struct ssh_channel *c;
1158
1159     /*
1160      * First-fit allocation of channel numbers: always pick the
1161      * lowest unused one. To do this, binary-search using the
1162      * counted B-tree to find the largest channel ID which is in a
1163      * contiguous sequence from the beginning. (Precisely
1164      * everything in that sequence must have ID equal to its tree
1165      * index plus CHANNEL_NUMBER_OFFSET.)
1166      */
1167     tsize = count234(ssh->channels);
1168
1169     low = -1;
1170     high = tsize;
1171     while (high - low > 1) {
1172         mid = (high + low) / 2;
1173         c = index234(ssh->channels, mid);
1174         if (c->localid == mid + CHANNEL_NUMBER_OFFSET)
1175             low = mid;                 /* this one is fine */
1176         else
1177             high = mid;                /* this one is past it */
1178     }
1179     /*
1180      * Now low points to either -1, or the tree index of the
1181      * largest ID in the initial sequence.
1182      */
1183     {
1184         unsigned i = low + 1 + CHANNEL_NUMBER_OFFSET;
1185         assert(NULL == find234(ssh->channels, &i, ssh_channelfind));
1186     }
1187     return low + 1 + CHANNEL_NUMBER_OFFSET;
1188 }
1189
1190 static void c_write_stderr(int trusted, const char *buf, int len)
1191 {
1192     int i;
1193     for (i = 0; i < len; i++)
1194         if (buf[i] != '\r' && (trusted || buf[i] == '\n' || (buf[i] & 0x60)))
1195             fputc(buf[i], stderr);
1196 }
1197
1198 static void c_write(Ssh ssh, const char *buf, int len)
1199 {
1200     if (flags & FLAG_STDERR)
1201         c_write_stderr(1, buf, len);
1202     else
1203         from_backend(ssh->frontend, 1, buf, len);
1204 }
1205
1206 static void c_write_untrusted(Ssh ssh, const char *buf, int len)
1207 {
1208     if (flags & FLAG_STDERR)
1209         c_write_stderr(0, buf, len);
1210     else
1211         from_backend_untrusted(ssh->frontend, buf, len);
1212 }
1213
1214 static void c_write_str(Ssh ssh, const char *buf)
1215 {
1216     c_write(ssh, buf, strlen(buf));
1217 }
1218
1219 static void ssh_free_packet(struct Packet *pkt)
1220 {
1221     sfree(pkt->data);
1222     sfree(pkt);
1223 }
1224 static struct Packet *ssh_new_packet(void)
1225 {
1226     struct Packet *pkt = snew(struct Packet);
1227
1228     pkt->body = pkt->data = NULL;
1229     pkt->maxlen = 0;
1230
1231     return pkt;
1232 }
1233
1234 static void ssh1_log_incoming_packet(Ssh ssh, struct Packet *pkt)
1235 {
1236     int nblanks = 0;
1237     struct logblank_t blanks[4];
1238     char *str;
1239     int slen;
1240
1241     pkt->savedpos = 0;
1242
1243     if (ssh->logomitdata &&
1244         (pkt->type == SSH1_SMSG_STDOUT_DATA ||
1245          pkt->type == SSH1_SMSG_STDERR_DATA ||
1246          pkt->type == SSH1_MSG_CHANNEL_DATA)) {
1247         /* "Session data" packets - omit the data string. */
1248         if (pkt->type == SSH1_MSG_CHANNEL_DATA)
1249             ssh_pkt_getuint32(pkt);    /* skip channel id */
1250         blanks[nblanks].offset = pkt->savedpos + 4;
1251         blanks[nblanks].type = PKTLOG_OMIT;
1252         ssh_pkt_getstring(pkt, &str, &slen);
1253         if (str) {
1254             blanks[nblanks].len = slen;
1255             nblanks++;
1256         }
1257     }
1258     log_packet(ssh->logctx, PKT_INCOMING, pkt->type,
1259                ssh1_pkt_type(pkt->type),
1260                pkt->body, pkt->length, nblanks, blanks, NULL,
1261                0, NULL);
1262 }
1263
1264 static void ssh1_log_outgoing_packet(Ssh ssh, struct Packet *pkt)
1265 {
1266     int nblanks = 0;
1267     struct logblank_t blanks[4];
1268     char *str;
1269     int slen;
1270
1271     /*
1272      * For outgoing packets, pkt->length represents the length of the
1273      * whole packet starting at pkt->data (including some header), and
1274      * pkt->body refers to the point within that where the log-worthy
1275      * payload begins. However, incoming packets expect pkt->length to
1276      * represent only the payload length (that is, it's measured from
1277      * pkt->body not from pkt->data). Temporarily adjust our outgoing
1278      * packet to conform to the incoming-packet semantics, so that we
1279      * can analyse it with the ssh_pkt_get functions.
1280      */
1281     pkt->length -= (pkt->body - pkt->data);
1282     pkt->savedpos = 0;
1283
1284     if (ssh->logomitdata &&
1285         (pkt->type == SSH1_CMSG_STDIN_DATA ||
1286          pkt->type == SSH1_MSG_CHANNEL_DATA)) {
1287         /* "Session data" packets - omit the data string. */
1288         if (pkt->type == SSH1_MSG_CHANNEL_DATA)
1289             ssh_pkt_getuint32(pkt);    /* skip channel id */
1290         blanks[nblanks].offset = pkt->savedpos + 4;
1291         blanks[nblanks].type = PKTLOG_OMIT;
1292         ssh_pkt_getstring(pkt, &str, &slen);
1293         if (str) {
1294             blanks[nblanks].len = slen;
1295             nblanks++;
1296         }
1297     }
1298
1299     if ((pkt->type == SSH1_CMSG_AUTH_PASSWORD ||
1300          pkt->type == SSH1_CMSG_AUTH_TIS_RESPONSE ||
1301          pkt->type == SSH1_CMSG_AUTH_CCARD_RESPONSE) &&
1302         conf_get_int(ssh->conf, CONF_logomitpass)) {
1303         /* If this is a password or similar packet, blank the password(s). */
1304         blanks[nblanks].offset = 0;
1305         blanks[nblanks].len = pkt->length;
1306         blanks[nblanks].type = PKTLOG_BLANK;
1307         nblanks++;
1308     } else if (pkt->type == SSH1_CMSG_X11_REQUEST_FORWARDING &&
1309                conf_get_int(ssh->conf, CONF_logomitpass)) {
1310         /*
1311          * If this is an X forwarding request packet, blank the fake
1312          * auth data.
1313          *
1314          * Note that while we blank the X authentication data here, we
1315          * don't take any special action to blank the start of an X11
1316          * channel, so using MIT-MAGIC-COOKIE-1 and actually opening
1317          * an X connection without having session blanking enabled is
1318          * likely to leak your cookie into the log.
1319          */
1320         pkt->savedpos = 0;
1321         ssh_pkt_getstring(pkt, &str, &slen);
1322         blanks[nblanks].offset = pkt->savedpos;
1323         blanks[nblanks].type = PKTLOG_BLANK;
1324         ssh_pkt_getstring(pkt, &str, &slen);
1325         if (str) {
1326             blanks[nblanks].len = pkt->savedpos - blanks[nblanks].offset;
1327             nblanks++;
1328         }
1329     }
1330
1331     log_packet(ssh->logctx, PKT_OUTGOING, pkt->data[12],
1332                ssh1_pkt_type(pkt->data[12]),
1333                pkt->body, pkt->length,
1334                nblanks, blanks, NULL, 0, NULL);
1335
1336     /*
1337      * Undo the above adjustment of pkt->length, to put the packet
1338      * back in the state we found it.
1339      */
1340     pkt->length += (pkt->body - pkt->data);
1341 }
1342
1343 /*
1344  * Collect incoming data in the incoming packet buffer.
1345  * Decipher and verify the packet when it is completely read.
1346  * Drop SSH1_MSG_DEBUG and SSH1_MSG_IGNORE packets.
1347  * Update the *data and *datalen variables.
1348  * Return a Packet structure when a packet is completed.
1349  */
1350 static struct Packet *ssh1_rdpkt(Ssh ssh, const unsigned char **data,
1351                                  int *datalen)
1352 {
1353     struct rdpkt1_state_tag *st = &ssh->rdpkt1_state;
1354
1355     crBegin(ssh->ssh1_rdpkt_crstate);
1356
1357     st->pktin = ssh_new_packet();
1358
1359     st->pktin->type = 0;
1360     st->pktin->length = 0;
1361
1362     for (st->i = st->len = 0; st->i < 4; st->i++) {
1363         while ((*datalen) == 0)
1364             crReturn(NULL);
1365         st->len = (st->len << 8) + **data;
1366         (*data)++, (*datalen)--;
1367     }
1368
1369     st->pad = 8 - (st->len % 8);
1370     st->biglen = st->len + st->pad;
1371     st->pktin->length = st->len - 5;
1372
1373     if (st->biglen < 0) {
1374         bombout(("Extremely large packet length from server suggests"
1375                  " data stream corruption"));
1376         ssh_free_packet(st->pktin);
1377         crStop(NULL);
1378     }
1379
1380     st->pktin->maxlen = st->biglen;
1381     st->pktin->data = snewn(st->biglen + APIEXTRA, unsigned char);
1382
1383     st->to_read = st->biglen;
1384     st->p = st->pktin->data;
1385     while (st->to_read > 0) {
1386         st->chunk = st->to_read;
1387         while ((*datalen) == 0)
1388             crReturn(NULL);
1389         if (st->chunk > (*datalen))
1390             st->chunk = (*datalen);
1391         memcpy(st->p, *data, st->chunk);
1392         *data += st->chunk;
1393         *datalen -= st->chunk;
1394         st->p += st->chunk;
1395         st->to_read -= st->chunk;
1396     }
1397
1398     if (ssh->cipher && detect_attack(ssh->crcda_ctx, st->pktin->data,
1399                                      st->biglen, NULL)) {
1400         bombout(("Network attack (CRC compensation) detected!"));
1401         ssh_free_packet(st->pktin);
1402         crStop(NULL);
1403     }
1404
1405     if (ssh->cipher)
1406         ssh->cipher->decrypt(ssh->v1_cipher_ctx, st->pktin->data, st->biglen);
1407
1408     st->realcrc = crc32_compute(st->pktin->data, st->biglen - 4);
1409     st->gotcrc = GET_32BIT(st->pktin->data + st->biglen - 4);
1410     if (st->gotcrc != st->realcrc) {
1411         bombout(("Incorrect CRC received on packet"));
1412         ssh_free_packet(st->pktin);
1413         crStop(NULL);
1414     }
1415
1416     st->pktin->body = st->pktin->data + st->pad + 1;
1417
1418     if (ssh->v1_compressing) {
1419         unsigned char *decompblk;
1420         int decomplen;
1421         if (!zlib_decompress_block(ssh->sc_comp_ctx,
1422                                    st->pktin->body - 1, st->pktin->length + 1,
1423                                    &decompblk, &decomplen)) {
1424             bombout(("Zlib decompression encountered invalid data"));
1425             ssh_free_packet(st->pktin);
1426             crStop(NULL);
1427         }
1428
1429         if (st->pktin->maxlen < st->pad + decomplen) {
1430             st->pktin->maxlen = st->pad + decomplen;
1431             st->pktin->data = sresize(st->pktin->data,
1432                                       st->pktin->maxlen + APIEXTRA,
1433                                       unsigned char);
1434             st->pktin->body = st->pktin->data + st->pad + 1;
1435         }
1436
1437         memcpy(st->pktin->body - 1, decompblk, decomplen);
1438         sfree(decompblk);
1439         st->pktin->length = decomplen - 1;
1440     }
1441
1442     st->pktin->type = st->pktin->body[-1];
1443
1444     /*
1445      * Now pktin->body and pktin->length identify the semantic content
1446      * of the packet, excluding the initial type byte.
1447      */
1448
1449     if (ssh->logctx)
1450         ssh1_log_incoming_packet(ssh, st->pktin);
1451
1452     st->pktin->savedpos = 0;
1453
1454     crFinish(st->pktin);
1455 }
1456
1457 static void ssh2_log_incoming_packet(Ssh ssh, struct Packet *pkt)
1458 {
1459     int nblanks = 0;
1460     struct logblank_t blanks[4];
1461     char *str;
1462     int slen;
1463
1464     pkt->savedpos = 0;
1465
1466     if (ssh->logomitdata &&
1467         (pkt->type == SSH2_MSG_CHANNEL_DATA ||
1468          pkt->type == SSH2_MSG_CHANNEL_EXTENDED_DATA)) {
1469         /* "Session data" packets - omit the data string. */
1470         ssh_pkt_getuint32(pkt);    /* skip channel id */
1471         if (pkt->type == SSH2_MSG_CHANNEL_EXTENDED_DATA)
1472             ssh_pkt_getuint32(pkt);    /* skip extended data type */
1473         blanks[nblanks].offset = pkt->savedpos + 4;
1474         blanks[nblanks].type = PKTLOG_OMIT;
1475         ssh_pkt_getstring(pkt, &str, &slen);
1476         if (str) {
1477             blanks[nblanks].len = slen;
1478             nblanks++;
1479         }
1480     }
1481
1482     log_packet(ssh->logctx, PKT_INCOMING, pkt->type,
1483                ssh2_pkt_type(ssh->pkt_kctx, ssh->pkt_actx, pkt->type),
1484                pkt->body, pkt->length, nblanks, blanks, &pkt->sequence,
1485                0, NULL);
1486 }
1487
1488 static void ssh2_log_outgoing_packet(Ssh ssh, struct Packet *pkt)
1489 {
1490     int nblanks = 0;
1491     struct logblank_t blanks[4];
1492     char *str;
1493     int slen;
1494
1495     /*
1496      * For outgoing packets, pkt->length represents the length of the
1497      * whole packet starting at pkt->data (including some header), and
1498      * pkt->body refers to the point within that where the log-worthy
1499      * payload begins. However, incoming packets expect pkt->length to
1500      * represent only the payload length (that is, it's measured from
1501      * pkt->body not from pkt->data). Temporarily adjust our outgoing
1502      * packet to conform to the incoming-packet semantics, so that we
1503      * can analyse it with the ssh_pkt_get functions.
1504      */
1505     pkt->length -= (pkt->body - pkt->data);
1506     pkt->savedpos = 0;
1507
1508     if (ssh->logomitdata &&
1509         (pkt->type == SSH2_MSG_CHANNEL_DATA ||
1510          pkt->type == SSH2_MSG_CHANNEL_EXTENDED_DATA)) {
1511         /* "Session data" packets - omit the data string. */
1512         ssh_pkt_getuint32(pkt);    /* skip channel id */
1513         if (pkt->type == SSH2_MSG_CHANNEL_EXTENDED_DATA)
1514             ssh_pkt_getuint32(pkt);    /* skip extended data type */
1515         blanks[nblanks].offset = pkt->savedpos + 4;
1516         blanks[nblanks].type = PKTLOG_OMIT;
1517         ssh_pkt_getstring(pkt, &str, &slen);
1518         if (str) {
1519             blanks[nblanks].len = slen;
1520             nblanks++;
1521         }
1522     }
1523
1524     if (pkt->type == SSH2_MSG_USERAUTH_REQUEST &&
1525         conf_get_int(ssh->conf, CONF_logomitpass)) {
1526         /* If this is a password packet, blank the password(s). */
1527         pkt->savedpos = 0;
1528         ssh_pkt_getstring(pkt, &str, &slen);
1529         ssh_pkt_getstring(pkt, &str, &slen);
1530         ssh_pkt_getstring(pkt, &str, &slen);
1531         if (slen == 8 && !memcmp(str, "password", 8)) {
1532             ssh2_pkt_getbool(pkt);
1533             /* Blank the password field. */
1534             blanks[nblanks].offset = pkt->savedpos;
1535             blanks[nblanks].type = PKTLOG_BLANK;
1536             ssh_pkt_getstring(pkt, &str, &slen);
1537             if (str) {
1538                 blanks[nblanks].len = pkt->savedpos - blanks[nblanks].offset;
1539                 nblanks++;
1540                 /* If there's another password field beyond it (change of
1541                  * password), blank that too. */
1542                 ssh_pkt_getstring(pkt, &str, &slen);
1543                 if (str)
1544                     blanks[nblanks-1].len =
1545                         pkt->savedpos - blanks[nblanks].offset;
1546             }
1547         }
1548     } else if (ssh->pkt_actx == SSH2_PKTCTX_KBDINTER &&
1549                pkt->type == SSH2_MSG_USERAUTH_INFO_RESPONSE &&
1550                conf_get_int(ssh->conf, CONF_logomitpass)) {
1551         /* If this is a keyboard-interactive response packet, blank
1552          * the responses. */
1553         pkt->savedpos = 0;
1554         ssh_pkt_getuint32(pkt);
1555         blanks[nblanks].offset = pkt->savedpos;
1556         blanks[nblanks].type = PKTLOG_BLANK;
1557         while (1) {
1558             ssh_pkt_getstring(pkt, &str, &slen);
1559             if (!str)
1560                 break;
1561         }
1562         blanks[nblanks].len = pkt->savedpos - blanks[nblanks].offset;
1563         nblanks++;
1564     } else if (pkt->type == SSH2_MSG_CHANNEL_REQUEST &&
1565                conf_get_int(ssh->conf, CONF_logomitpass)) {
1566         /*
1567          * If this is an X forwarding request packet, blank the fake
1568          * auth data.
1569          *
1570          * Note that while we blank the X authentication data here, we
1571          * don't take any special action to blank the start of an X11
1572          * channel, so using MIT-MAGIC-COOKIE-1 and actually opening
1573          * an X connection without having session blanking enabled is
1574          * likely to leak your cookie into the log.
1575          */
1576         pkt->savedpos = 0;
1577         ssh_pkt_getuint32(pkt);
1578         ssh_pkt_getstring(pkt, &str, &slen);
1579         if (slen == 7 && !memcmp(str, "x11-req", 0)) {
1580             ssh2_pkt_getbool(pkt);
1581             ssh2_pkt_getbool(pkt);
1582             ssh_pkt_getstring(pkt, &str, &slen);
1583             blanks[nblanks].offset = pkt->savedpos;
1584             blanks[nblanks].type = PKTLOG_BLANK;
1585             ssh_pkt_getstring(pkt, &str, &slen);
1586             if (str) {
1587                 blanks[nblanks].len = pkt->savedpos - blanks[nblanks].offset;
1588                 nblanks++;
1589             }
1590         }
1591     }
1592
1593     log_packet(ssh->logctx, PKT_OUTGOING, pkt->data[5],
1594                ssh2_pkt_type(ssh->pkt_kctx, ssh->pkt_actx, pkt->data[5]),
1595                pkt->body, pkt->length, nblanks, blanks,
1596                &ssh->v2_outgoing_sequence,
1597                pkt->downstream_id, pkt->additional_log_text);
1598
1599     /*
1600      * Undo the above adjustment of pkt->length, to put the packet
1601      * back in the state we found it.
1602      */
1603     pkt->length += (pkt->body - pkt->data);
1604 }
1605
1606 static struct Packet *ssh2_rdpkt(Ssh ssh, const unsigned char **data,
1607                                  int *datalen)
1608 {
1609     struct rdpkt2_state_tag *st = &ssh->rdpkt2_state;
1610
1611     crBegin(ssh->ssh2_rdpkt_crstate);
1612
1613     st->pktin = ssh_new_packet();
1614
1615     st->pktin->type = 0;
1616     st->pktin->length = 0;
1617     if (ssh->sccipher)
1618         st->cipherblk = ssh->sccipher->blksize;
1619     else
1620         st->cipherblk = 8;
1621     if (st->cipherblk < 8)
1622         st->cipherblk = 8;
1623     st->maclen = ssh->scmac ? ssh->scmac->len : 0;
1624
1625     if (ssh->sccipher && (ssh->sccipher->flags & SSH_CIPHER_IS_CBC) &&
1626         ssh->scmac && !ssh->scmac_etm) {
1627         /*
1628          * When dealing with a CBC-mode cipher, we want to avoid the
1629          * possibility of an attacker's tweaking the ciphertext stream
1630          * so as to cause us to feed the same block to the block
1631          * cipher more than once and thus leak information
1632          * (VU#958563).  The way we do this is not to take any
1633          * decisions on the basis of anything we've decrypted until
1634          * we've verified it with a MAC.  That includes the packet
1635          * length, so we just read data and check the MAC repeatedly,
1636          * and when the MAC passes, see if the length we've got is
1637          * plausible.
1638          *
1639          * This defence is unnecessary in OpenSSH ETM mode, because
1640          * the whole point of ETM mode is that the attacker can't
1641          * tweak the ciphertext stream at all without the MAC
1642          * detecting it before we decrypt anything.
1643          */
1644
1645         /* May as well allocate the whole lot now. */
1646         st->pktin->data = snewn(OUR_V2_PACKETLIMIT + st->maclen + APIEXTRA,
1647                                 unsigned char);
1648
1649         /* Read an amount corresponding to the MAC. */
1650         for (st->i = 0; st->i < st->maclen; st->i++) {
1651             while ((*datalen) == 0)
1652                 crReturn(NULL);
1653             st->pktin->data[st->i] = *(*data)++;
1654             (*datalen)--;
1655         }
1656
1657         st->packetlen = 0;
1658         {
1659             unsigned char seq[4];
1660             ssh->scmac->start(ssh->sc_mac_ctx);
1661             PUT_32BIT(seq, st->incoming_sequence);
1662             ssh->scmac->bytes(ssh->sc_mac_ctx, seq, 4);
1663         }
1664
1665         for (;;) { /* Once around this loop per cipher block. */
1666             /* Read another cipher-block's worth, and tack it onto the end. */
1667             for (st->i = 0; st->i < st->cipherblk; st->i++) {
1668                 while ((*datalen) == 0)
1669                     crReturn(NULL);
1670                 st->pktin->data[st->packetlen+st->maclen+st->i] = *(*data)++;
1671                 (*datalen)--;
1672             }
1673             /* Decrypt one more block (a little further back in the stream). */
1674             ssh->sccipher->decrypt(ssh->sc_cipher_ctx,
1675                                    st->pktin->data + st->packetlen,
1676                                    st->cipherblk);
1677             /* Feed that block to the MAC. */
1678             ssh->scmac->bytes(ssh->sc_mac_ctx,
1679                               st->pktin->data + st->packetlen, st->cipherblk);
1680             st->packetlen += st->cipherblk;
1681             /* See if that gives us a valid packet. */
1682             if (ssh->scmac->verresult(ssh->sc_mac_ctx,
1683                                       st->pktin->data + st->packetlen) &&
1684                 ((st->len = toint(GET_32BIT(st->pktin->data))) ==
1685                  st->packetlen-4))
1686                     break;
1687             if (st->packetlen >= OUR_V2_PACKETLIMIT) {
1688                 bombout(("No valid incoming packet found"));
1689                 ssh_free_packet(st->pktin);
1690                 crStop(NULL);
1691             }       
1692         }
1693         st->pktin->maxlen = st->packetlen + st->maclen;
1694         st->pktin->data = sresize(st->pktin->data,
1695                                   st->pktin->maxlen + APIEXTRA,
1696                                   unsigned char);
1697     } else if (ssh->scmac && ssh->scmac_etm) {
1698         st->pktin->data = snewn(4 + APIEXTRA, unsigned char);
1699
1700         /*
1701          * OpenSSH encrypt-then-MAC mode: the packet length is
1702          * unencrypted, unless the cipher supports length encryption.
1703          */
1704         for (st->i = st->len = 0; st->i < 4; st->i++) {
1705             while ((*datalen) == 0)
1706                 crReturn(NULL);
1707             st->pktin->data[st->i] = *(*data)++;
1708             (*datalen)--;
1709         }
1710         /* Cipher supports length decryption, so do it */
1711         if (ssh->sccipher && (ssh->sccipher->flags & SSH_CIPHER_SEPARATE_LENGTH)) {
1712             /* Keep the packet the same though, so the MAC passes */
1713             unsigned char len[4];
1714             memcpy(len, st->pktin->data, 4);
1715             ssh->sccipher->decrypt_length(ssh->sc_cipher_ctx, len, 4, st->incoming_sequence);
1716             st->len = toint(GET_32BIT(len));
1717         } else {
1718             st->len = toint(GET_32BIT(st->pktin->data));
1719         }
1720
1721         /*
1722          * _Completely_ silly lengths should be stomped on before they
1723          * do us any more damage.
1724          */
1725         if (st->len < 0 || st->len > OUR_V2_PACKETLIMIT ||
1726             st->len % st->cipherblk != 0) {
1727             bombout(("Incoming packet length field was garbled"));
1728             ssh_free_packet(st->pktin);
1729             crStop(NULL);
1730         }
1731
1732         /*
1733          * So now we can work out the total packet length.
1734          */
1735         st->packetlen = st->len + 4;
1736
1737         /*
1738          * Allocate memory for the rest of the packet.
1739          */
1740         st->pktin->maxlen = st->packetlen + st->maclen;
1741         st->pktin->data = sresize(st->pktin->data,
1742                                   st->pktin->maxlen + APIEXTRA,
1743                                   unsigned char);
1744
1745         /*
1746          * Read the remainder of the packet.
1747          */
1748         for (st->i = 4; st->i < st->packetlen + st->maclen; st->i++) {
1749             while ((*datalen) == 0)
1750                 crReturn(NULL);
1751             st->pktin->data[st->i] = *(*data)++;
1752             (*datalen)--;
1753         }
1754
1755         /*
1756          * Check the MAC.
1757          */
1758         if (ssh->scmac
1759             && !ssh->scmac->verify(ssh->sc_mac_ctx, st->pktin->data,
1760                                    st->len + 4, st->incoming_sequence)) {
1761             bombout(("Incorrect MAC received on packet"));
1762             ssh_free_packet(st->pktin);
1763             crStop(NULL);
1764         }
1765
1766         /* Decrypt everything between the length field and the MAC. */
1767         if (ssh->sccipher)
1768             ssh->sccipher->decrypt(ssh->sc_cipher_ctx,
1769                                    st->pktin->data + 4,
1770                                    st->packetlen - 4);
1771     } else {
1772         st->pktin->data = snewn(st->cipherblk + APIEXTRA, unsigned char);
1773
1774         /*
1775          * Acquire and decrypt the first block of the packet. This will
1776          * contain the length and padding details.
1777          */
1778         for (st->i = st->len = 0; st->i < st->cipherblk; st->i++) {
1779             while ((*datalen) == 0)
1780                 crReturn(NULL);
1781             st->pktin->data[st->i] = *(*data)++;
1782             (*datalen)--;
1783         }
1784
1785         if (ssh->sccipher)
1786             ssh->sccipher->decrypt(ssh->sc_cipher_ctx,
1787                                    st->pktin->data, st->cipherblk);
1788
1789         /*
1790          * Now get the length figure.
1791          */
1792         st->len = toint(GET_32BIT(st->pktin->data));
1793
1794         /*
1795          * _Completely_ silly lengths should be stomped on before they
1796          * do us any more damage.
1797          */
1798         if (st->len < 0 || st->len > OUR_V2_PACKETLIMIT ||
1799             (st->len + 4) % st->cipherblk != 0) {
1800             bombout(("Incoming packet was garbled on decryption"));
1801             ssh_free_packet(st->pktin);
1802             crStop(NULL);
1803         }
1804
1805         /*
1806          * So now we can work out the total packet length.
1807          */
1808         st->packetlen = st->len + 4;
1809
1810         /*
1811          * Allocate memory for the rest of the packet.
1812          */
1813         st->pktin->maxlen = st->packetlen + st->maclen;
1814         st->pktin->data = sresize(st->pktin->data,
1815                                   st->pktin->maxlen + APIEXTRA,
1816                                   unsigned char);
1817
1818         /*
1819          * Read and decrypt the remainder of the packet.
1820          */
1821         for (st->i = st->cipherblk; st->i < st->packetlen + st->maclen;
1822              st->i++) {
1823             while ((*datalen) == 0)
1824                 crReturn(NULL);
1825             st->pktin->data[st->i] = *(*data)++;
1826             (*datalen)--;
1827         }
1828         /* Decrypt everything _except_ the MAC. */
1829         if (ssh->sccipher)
1830             ssh->sccipher->decrypt(ssh->sc_cipher_ctx,
1831                                    st->pktin->data + st->cipherblk,
1832                                    st->packetlen - st->cipherblk);
1833
1834         /*
1835          * Check the MAC.
1836          */
1837         if (ssh->scmac
1838             && !ssh->scmac->verify(ssh->sc_mac_ctx, st->pktin->data,
1839                                    st->len + 4, st->incoming_sequence)) {
1840             bombout(("Incorrect MAC received on packet"));
1841             ssh_free_packet(st->pktin);
1842             crStop(NULL);
1843         }
1844     }
1845     /* Get and sanity-check the amount of random padding. */
1846     st->pad = st->pktin->data[4];
1847     if (st->pad < 4 || st->len - st->pad < 1) {
1848         bombout(("Invalid padding length on received packet"));
1849         ssh_free_packet(st->pktin);
1850         crStop(NULL);
1851     }
1852     /*
1853      * This enables us to deduce the payload length.
1854      */
1855     st->payload = st->len - st->pad - 1;
1856
1857     st->pktin->length = st->payload + 5;
1858     st->pktin->encrypted_len = st->packetlen;
1859
1860     st->pktin->sequence = st->incoming_sequence++;
1861
1862     st->pktin->length = st->packetlen - st->pad;
1863     assert(st->pktin->length >= 0);
1864
1865     /*
1866      * Decompress packet payload.
1867      */
1868     {
1869         unsigned char *newpayload;
1870         int newlen;
1871         if (ssh->sccomp &&
1872             ssh->sccomp->decompress(ssh->sc_comp_ctx,
1873                                     st->pktin->data + 5, st->pktin->length - 5,
1874                                     &newpayload, &newlen)) {
1875             if (st->pktin->maxlen < newlen + 5) {
1876                 st->pktin->maxlen = newlen + 5;
1877                 st->pktin->data = sresize(st->pktin->data,
1878                                           st->pktin->maxlen + APIEXTRA,
1879                                           unsigned char);
1880             }
1881             st->pktin->length = 5 + newlen;
1882             memcpy(st->pktin->data + 5, newpayload, newlen);
1883             sfree(newpayload);
1884         }
1885     }
1886
1887     /*
1888      * RFC 4253 doesn't explicitly say that completely empty packets
1889      * with no type byte are forbidden, so treat them as deserving
1890      * an SSH_MSG_UNIMPLEMENTED.
1891      */
1892     if (st->pktin->length <= 5) { /* == 5 we hope, but robustness */
1893         ssh2_msg_something_unimplemented(ssh, st->pktin);
1894         crStop(NULL);
1895     }
1896     /*
1897      * pktin->body and pktin->length should identify the semantic
1898      * content of the packet, excluding the initial type byte.
1899      */
1900     st->pktin->type = st->pktin->data[5];
1901     st->pktin->body = st->pktin->data + 6;
1902     st->pktin->length -= 6;
1903     assert(st->pktin->length >= 0);    /* one last double-check */
1904
1905     if (ssh->logctx)
1906         ssh2_log_incoming_packet(ssh, st->pktin);
1907
1908     st->pktin->savedpos = 0;
1909
1910     crFinish(st->pktin);
1911 }
1912
1913 static struct Packet *ssh2_bare_connection_rdpkt(Ssh ssh,
1914                                                  const unsigned char **data,
1915                                                  int *datalen)
1916 {
1917     struct rdpkt2_bare_state_tag *st = &ssh->rdpkt2_bare_state;
1918
1919     crBegin(ssh->ssh2_bare_rdpkt_crstate);
1920
1921     /*
1922      * Read the packet length field.
1923      */
1924     for (st->i = 0; st->i < 4; st->i++) {
1925         while ((*datalen) == 0)
1926             crReturn(NULL);
1927         st->length[st->i] = *(*data)++;
1928         (*datalen)--;
1929     }
1930
1931     st->packetlen = toint(GET_32BIT_MSB_FIRST(st->length));
1932     if (st->packetlen <= 0 || st->packetlen >= OUR_V2_PACKETLIMIT) {
1933         bombout(("Invalid packet length received"));
1934         crStop(NULL);
1935     }
1936
1937     st->pktin = ssh_new_packet();
1938     st->pktin->data = snewn(st->packetlen, unsigned char);
1939
1940     st->pktin->encrypted_len = st->packetlen;
1941
1942     st->pktin->sequence = st->incoming_sequence++;
1943
1944     /*
1945      * Read the remainder of the packet.
1946      */
1947     for (st->i = 0; st->i < st->packetlen; st->i++) {
1948         while ((*datalen) == 0)
1949             crReturn(NULL);
1950         st->pktin->data[st->i] = *(*data)++;
1951         (*datalen)--;
1952     }
1953
1954     /*
1955      * pktin->body and pktin->length should identify the semantic
1956      * content of the packet, excluding the initial type byte.
1957      */
1958     st->pktin->type = st->pktin->data[0];
1959     st->pktin->body = st->pktin->data + 1;
1960     st->pktin->length = st->packetlen - 1;
1961
1962     /*
1963      * Log incoming packet, possibly omitting sensitive fields.
1964      */
1965     if (ssh->logctx)
1966         ssh2_log_incoming_packet(ssh, st->pktin);
1967
1968     st->pktin->savedpos = 0;
1969
1970     crFinish(st->pktin);
1971 }
1972
1973 static int s_wrpkt_prepare(Ssh ssh, struct Packet *pkt, int *offset_p)
1974 {
1975     int pad, biglen, i, pktoffs;
1976     unsigned long crc;
1977 #ifdef __SC__
1978     /*
1979      * XXX various versions of SC (including 8.8.4) screw up the
1980      * register allocation in this function and use the same register
1981      * (D6) for len and as a temporary, with predictable results.  The
1982      * following sledgehammer prevents this.
1983      */
1984     volatile
1985 #endif
1986     int len;
1987
1988     if (ssh->logctx)
1989         ssh1_log_outgoing_packet(ssh, pkt);
1990
1991     if (ssh->v1_compressing) {
1992         unsigned char *compblk;
1993         int complen;
1994         zlib_compress_block(ssh->cs_comp_ctx,
1995                             pkt->data + 12, pkt->length - 12,
1996                             &compblk, &complen);
1997         ssh_pkt_ensure(pkt, complen + 2);   /* just in case it's got bigger */
1998         memcpy(pkt->data + 12, compblk, complen);
1999         sfree(compblk);
2000         pkt->length = complen + 12;
2001     }
2002
2003     ssh_pkt_ensure(pkt, pkt->length + 4); /* space for CRC */
2004     pkt->length += 4;
2005     len = pkt->length - 4 - 8;  /* len(type+data+CRC) */
2006     pad = 8 - (len % 8);
2007     pktoffs = 8 - pad;
2008     biglen = len + pad;         /* len(padding+type+data+CRC) */
2009
2010     for (i = pktoffs; i < 4+8; i++)
2011         pkt->data[i] = random_byte();
2012     crc = crc32_compute(pkt->data + pktoffs + 4, biglen - 4); /* all ex len */
2013     PUT_32BIT(pkt->data + pktoffs + 4 + biglen - 4, crc);
2014     PUT_32BIT(pkt->data + pktoffs, len);
2015
2016     if (ssh->cipher)
2017         ssh->cipher->encrypt(ssh->v1_cipher_ctx,
2018                              pkt->data + pktoffs + 4, biglen);
2019
2020     if (offset_p) *offset_p = pktoffs;
2021     return biglen + 4;          /* len(length+padding+type+data+CRC) */
2022 }
2023
2024 static int s_write(Ssh ssh, void *data, int len)
2025 {
2026     if (ssh->logctx)
2027         log_packet(ssh->logctx, PKT_OUTGOING, -1, NULL, data, len,
2028                    0, NULL, NULL, 0, NULL);
2029     if (!ssh->s)
2030         return 0;
2031     return sk_write(ssh->s, (char *)data, len);
2032 }
2033
2034 static void s_wrpkt(Ssh ssh, struct Packet *pkt)
2035 {
2036     int len, backlog, offset;
2037     len = s_wrpkt_prepare(ssh, pkt, &offset);
2038     backlog = s_write(ssh, pkt->data + offset, len);
2039     if (backlog > SSH_MAX_BACKLOG)
2040         ssh_throttle_all(ssh, 1, backlog);
2041     ssh_free_packet(pkt);
2042 }
2043
2044 static void s_wrpkt_defer(Ssh ssh, struct Packet *pkt)
2045 {
2046     int len, offset;
2047     len = s_wrpkt_prepare(ssh, pkt, &offset);
2048     if (ssh->deferred_len + len > ssh->deferred_size) {
2049         ssh->deferred_size = ssh->deferred_len + len + 128;
2050         ssh->deferred_send_data = sresize(ssh->deferred_send_data,
2051                                           ssh->deferred_size,
2052                                           unsigned char);
2053     }
2054     memcpy(ssh->deferred_send_data + ssh->deferred_len,
2055            pkt->data + offset, len);
2056     ssh->deferred_len += len;
2057     ssh_free_packet(pkt);
2058 }
2059
2060 /*
2061  * Construct a SSH-1 packet with the specified contents.
2062  * (This all-at-once interface used to be the only one, but now SSH-1
2063  * packets can also be constructed incrementally.)
2064  */
2065 static struct Packet *construct_packet(Ssh ssh, int pkttype, va_list ap)
2066 {
2067     int argtype;
2068     Bignum bn;
2069     struct Packet *pkt;
2070
2071     pkt = ssh1_pkt_init(pkttype);
2072
2073     while ((argtype = va_arg(ap, int)) != PKT_END) {
2074         unsigned char *argp, argchar;
2075         char *sargp;
2076         unsigned long argint;
2077         int arglen;
2078         switch (argtype) {
2079           /* Actual fields in the packet */
2080           case PKT_INT:
2081             argint = va_arg(ap, int);
2082             ssh_pkt_adduint32(pkt, argint);
2083             break;
2084           case PKT_CHAR:
2085             argchar = (unsigned char) va_arg(ap, int);
2086             ssh_pkt_addbyte(pkt, argchar);
2087             break;
2088           case PKT_DATA:
2089             argp = va_arg(ap, unsigned char *);
2090             arglen = va_arg(ap, int);
2091             ssh_pkt_adddata(pkt, argp, arglen);
2092             break;
2093           case PKT_STR:
2094             sargp = va_arg(ap, char *);
2095             ssh_pkt_addstring(pkt, sargp);
2096             break;
2097           case PKT_BIGNUM:
2098             bn = va_arg(ap, Bignum);
2099             ssh1_pkt_addmp(pkt, bn);
2100             break;
2101         }
2102     }
2103
2104     return pkt;
2105 }
2106
2107 static void send_packet(Ssh ssh, int pkttype, ...)
2108 {
2109     struct Packet *pkt;
2110     va_list ap;
2111     va_start(ap, pkttype);
2112     pkt = construct_packet(ssh, pkttype, ap);
2113     va_end(ap);
2114     s_wrpkt(ssh, pkt);
2115 }
2116
2117 static void defer_packet(Ssh ssh, int pkttype, ...)
2118 {
2119     struct Packet *pkt;
2120     va_list ap;
2121     va_start(ap, pkttype);
2122     pkt = construct_packet(ssh, pkttype, ap);
2123     va_end(ap);
2124     s_wrpkt_defer(ssh, pkt);
2125 }
2126
2127 static int ssh_versioncmp(const char *a, const char *b)
2128 {
2129     char *ae, *be;
2130     unsigned long av, bv;
2131
2132     av = strtoul(a, &ae, 10);
2133     bv = strtoul(b, &be, 10);
2134     if (av != bv)
2135         return (av < bv ? -1 : +1);
2136     if (*ae == '.')
2137         ae++;
2138     if (*be == '.')
2139         be++;
2140     av = strtoul(ae, &ae, 10);
2141     bv = strtoul(be, &be, 10);
2142     if (av != bv)
2143         return (av < bv ? -1 : +1);
2144     return 0;
2145 }
2146
2147 /*
2148  * Utility routines for putting an SSH-protocol `string' and
2149  * `uint32' into a hash state.
2150  */
2151 static void hash_string(const struct ssh_hash *h, void *s, void *str, int len)
2152 {
2153     unsigned char lenblk[4];
2154     PUT_32BIT(lenblk, len);
2155     h->bytes(s, lenblk, 4);
2156     h->bytes(s, str, len);
2157 }
2158
2159 static void hash_uint32(const struct ssh_hash *h, void *s, unsigned i)
2160 {
2161     unsigned char intblk[4];
2162     PUT_32BIT(intblk, i);
2163     h->bytes(s, intblk, 4);
2164 }
2165
2166 /*
2167  * Packet construction functions. Mostly shared between SSH-1 and SSH-2.
2168  */
2169 static void ssh_pkt_ensure(struct Packet *pkt, int length)
2170 {
2171     if (pkt->maxlen < length) {
2172         unsigned char *body = pkt->body;
2173         int offset = body ? body - pkt->data : 0;
2174         pkt->maxlen = length + 256;
2175         pkt->data = sresize(pkt->data, pkt->maxlen + APIEXTRA, unsigned char);
2176         if (body) pkt->body = pkt->data + offset;
2177     }
2178 }
2179 static void ssh_pkt_adddata(struct Packet *pkt, const void *data, int len)
2180 {
2181     pkt->length += len;
2182     ssh_pkt_ensure(pkt, pkt->length);
2183     memcpy(pkt->data + pkt->length - len, data, len);
2184 }
2185 static void ssh_pkt_addbyte(struct Packet *pkt, unsigned char byte)
2186 {
2187     ssh_pkt_adddata(pkt, &byte, 1);
2188 }
2189 static void ssh2_pkt_addbool(struct Packet *pkt, unsigned char value)
2190 {
2191     ssh_pkt_adddata(pkt, &value, 1);
2192 }
2193 static void ssh_pkt_adduint32(struct Packet *pkt, unsigned long value)
2194 {
2195     unsigned char x[4];
2196     PUT_32BIT(x, value);
2197     ssh_pkt_adddata(pkt, x, 4);
2198 }
2199 static void ssh_pkt_addstring_start(struct Packet *pkt)
2200 {
2201     ssh_pkt_adduint32(pkt, 0);
2202     pkt->savedpos = pkt->length;
2203 }
2204 static void ssh_pkt_addstring_data(struct Packet *pkt, const char *data,
2205                                    int len)
2206 {
2207     ssh_pkt_adddata(pkt, data, len);
2208     PUT_32BIT(pkt->data + pkt->savedpos - 4, pkt->length - pkt->savedpos);
2209 }
2210 static void ssh_pkt_addstring_str(struct Packet *pkt, const char *data)
2211 {
2212   ssh_pkt_addstring_data(pkt, data, strlen(data));
2213 }
2214 static void ssh_pkt_addstring(struct Packet *pkt, const char *data)
2215 {
2216     ssh_pkt_addstring_start(pkt);
2217     ssh_pkt_addstring_str(pkt, data);
2218 }
2219 static void ssh1_pkt_addmp(struct Packet *pkt, Bignum b)
2220 {
2221     int len = ssh1_bignum_length(b);
2222     unsigned char *data = snewn(len, unsigned char);
2223     (void) ssh1_write_bignum(data, b);
2224     ssh_pkt_adddata(pkt, data, len);
2225     sfree(data);
2226 }
2227 static unsigned char *ssh2_mpint_fmt(Bignum b, int *len)
2228 {
2229     unsigned char *p;
2230     int i, n = (bignum_bitcount(b) + 7) / 8;
2231     p = snewn(n + 1, unsigned char);
2232     p[0] = 0;
2233     for (i = 1; i <= n; i++)
2234         p[i] = bignum_byte(b, n - i);
2235     i = 0;
2236     while (i <= n && p[i] == 0 && (p[i + 1] & 0x80) == 0)
2237         i++;
2238     memmove(p, p + i, n + 1 - i);
2239     *len = n + 1 - i;
2240     return p;
2241 }
2242 static void ssh2_pkt_addmp(struct Packet *pkt, Bignum b)
2243 {
2244     unsigned char *p;
2245     int len;
2246     p = ssh2_mpint_fmt(b, &len);
2247     ssh_pkt_addstring_start(pkt);
2248     ssh_pkt_addstring_data(pkt, (char *)p, len);
2249     sfree(p);
2250 }
2251
2252 static struct Packet *ssh1_pkt_init(int pkt_type)
2253 {
2254     struct Packet *pkt = ssh_new_packet();
2255     pkt->length = 4 + 8;            /* space for length + max padding */
2256     ssh_pkt_addbyte(pkt, pkt_type);
2257     pkt->body = pkt->data + pkt->length;
2258     pkt->type = pkt_type;
2259     pkt->downstream_id = 0;
2260     pkt->additional_log_text = NULL;
2261     return pkt;
2262 }
2263
2264 /* For legacy code (SSH-1 and -2 packet construction used to be separate) */
2265 #define ssh2_pkt_ensure(pkt, length) ssh_pkt_ensure(pkt, length)
2266 #define ssh2_pkt_adddata(pkt, data, len) ssh_pkt_adddata(pkt, data, len)
2267 #define ssh2_pkt_addbyte(pkt, byte) ssh_pkt_addbyte(pkt, byte)
2268 #define ssh2_pkt_adduint32(pkt, value) ssh_pkt_adduint32(pkt, value)
2269 #define ssh2_pkt_addstring_start(pkt) ssh_pkt_addstring_start(pkt)
2270 #define ssh2_pkt_addstring_str(pkt, data) ssh_pkt_addstring_str(pkt, data)
2271 #define ssh2_pkt_addstring_data(pkt, data, len) ssh_pkt_addstring_data(pkt, data, len)
2272 #define ssh2_pkt_addstring(pkt, data) ssh_pkt_addstring(pkt, data)
2273
2274 static struct Packet *ssh2_pkt_init(int pkt_type)
2275 {
2276     struct Packet *pkt = ssh_new_packet();
2277     pkt->length = 5; /* space for packet length + padding length */
2278     pkt->forcepad = 0;
2279     pkt->type = pkt_type;
2280     ssh_pkt_addbyte(pkt, (unsigned char) pkt_type);
2281     pkt->body = pkt->data + pkt->length; /* after packet type */
2282     pkt->downstream_id = 0;
2283     pkt->additional_log_text = NULL;
2284     return pkt;
2285 }
2286
2287 /*
2288  * Construct an SSH-2 final-form packet: compress it, encrypt it,
2289  * put the MAC on it. Final packet, ready to be sent, is stored in
2290  * pkt->data. Total length is returned.
2291  */
2292 static int ssh2_pkt_construct(Ssh ssh, struct Packet *pkt)
2293 {
2294     int cipherblk, maclen, padding, unencrypted_prefix, i;
2295
2296     if (ssh->logctx)
2297         ssh2_log_outgoing_packet(ssh, pkt);
2298
2299     if (ssh->bare_connection) {
2300         /*
2301          * Trivial packet construction for the bare connection
2302          * protocol.
2303          */
2304         PUT_32BIT(pkt->data + 1, pkt->length - 5);
2305         pkt->body = pkt->data + 1;
2306         ssh->v2_outgoing_sequence++;   /* only for diagnostics, really */
2307         return pkt->length - 1;
2308     }
2309
2310     /*
2311      * Compress packet payload.
2312      */
2313     {
2314         unsigned char *newpayload;
2315         int newlen;
2316         if (ssh->cscomp &&
2317             ssh->cscomp->compress(ssh->cs_comp_ctx, pkt->data + 5,
2318                                   pkt->length - 5,
2319                                   &newpayload, &newlen)) {
2320             pkt->length = 5;
2321             ssh2_pkt_adddata(pkt, newpayload, newlen);
2322             sfree(newpayload);
2323         }
2324     }
2325
2326     /*
2327      * Add padding. At least four bytes, and must also bring total
2328      * length (minus MAC) up to a multiple of the block size.
2329      * If pkt->forcepad is set, make sure the packet is at least that size
2330      * after padding.
2331      */
2332     cipherblk = ssh->cscipher ? ssh->cscipher->blksize : 8;  /* block size */
2333     cipherblk = cipherblk < 8 ? 8 : cipherblk;  /* or 8 if blksize < 8 */
2334     padding = 4;
2335     unencrypted_prefix = (ssh->csmac && ssh->csmac_etm) ? 4 : 0;
2336     if (pkt->length + padding < pkt->forcepad)
2337         padding = pkt->forcepad - pkt->length;
2338     padding +=
2339         (cipherblk - (pkt->length - unencrypted_prefix + padding) % cipherblk)
2340         % cipherblk;
2341     assert(padding <= 255);
2342     maclen = ssh->csmac ? ssh->csmac->len : 0;
2343     ssh2_pkt_ensure(pkt, pkt->length + padding + maclen);
2344     pkt->data[4] = padding;
2345     for (i = 0; i < padding; i++)
2346         pkt->data[pkt->length + i] = random_byte();
2347     PUT_32BIT(pkt->data, pkt->length + padding - 4);
2348
2349     /* Encrypt length if the scheme requires it */
2350     if (ssh->cscipher && (ssh->cscipher->flags & SSH_CIPHER_SEPARATE_LENGTH)) {
2351         ssh->cscipher->encrypt_length(ssh->cs_cipher_ctx, pkt->data, 4,
2352                                       ssh->v2_outgoing_sequence);
2353     }
2354
2355     if (ssh->csmac && ssh->csmac_etm) {
2356         /*
2357          * OpenSSH-defined encrypt-then-MAC protocol.
2358          */
2359         if (ssh->cscipher)
2360             ssh->cscipher->encrypt(ssh->cs_cipher_ctx,
2361                                    pkt->data + 4, pkt->length + padding - 4);
2362         ssh->csmac->generate(ssh->cs_mac_ctx, pkt->data,
2363                              pkt->length + padding,
2364                              ssh->v2_outgoing_sequence);
2365     } else {
2366         /*
2367          * SSH-2 standard protocol.
2368          */
2369         if (ssh->csmac)
2370             ssh->csmac->generate(ssh->cs_mac_ctx, pkt->data,
2371                                  pkt->length + padding,
2372                                  ssh->v2_outgoing_sequence);
2373         if (ssh->cscipher)
2374             ssh->cscipher->encrypt(ssh->cs_cipher_ctx,
2375                                    pkt->data, pkt->length + padding);
2376     }
2377
2378     ssh->v2_outgoing_sequence++;       /* whether or not we MACed */
2379     pkt->encrypted_len = pkt->length + padding;
2380
2381     /* Ready-to-send packet starts at pkt->data. We return length. */
2382     pkt->body = pkt->data;
2383     return pkt->length + padding + maclen;
2384 }
2385
2386 /*
2387  * Routines called from the main SSH code to send packets. There
2388  * are quite a few of these, because we have two separate
2389  * mechanisms for delaying the sending of packets:
2390  * 
2391  *  - In order to send an IGNORE message and a password message in
2392  *    a single fixed-length blob, we require the ability to
2393  *    concatenate the encrypted forms of those two packets _into_ a
2394  *    single blob and then pass it to our <network.h> transport
2395  *    layer in one go. Hence, there's a deferment mechanism which
2396  *    works after packet encryption.
2397  * 
2398  *  - In order to avoid sending any connection-layer messages
2399  *    during repeat key exchange, we have to queue up any such
2400  *    outgoing messages _before_ they are encrypted (and in
2401  *    particular before they're allocated sequence numbers), and
2402  *    then send them once we've finished.
2403  * 
2404  * I call these mechanisms `defer' and `queue' respectively, so as
2405  * to distinguish them reasonably easily.
2406  * 
2407  * The functions send_noqueue() and defer_noqueue() free the packet
2408  * structure they are passed. Every outgoing packet goes through
2409  * precisely one of these functions in its life; packets passed to
2410  * ssh2_pkt_send() or ssh2_pkt_defer() either go straight to one of
2411  * these or get queued, and then when the queue is later emptied
2412  * the packets are all passed to defer_noqueue().
2413  *
2414  * When using a CBC-mode cipher, it's necessary to ensure that an
2415  * attacker can't provide data to be encrypted using an IV that they
2416  * know.  We ensure this by prefixing each packet that might contain
2417  * user data with an SSH_MSG_IGNORE.  This is done using the deferral
2418  * mechanism, so in this case send_noqueue() ends up redirecting to
2419  * defer_noqueue().  If you don't like this inefficiency, don't use
2420  * CBC.
2421  */
2422
2423 static void ssh2_pkt_defer_noqueue(Ssh, struct Packet *, int);
2424 static void ssh_pkt_defersend(Ssh);
2425
2426 /*
2427  * Send an SSH-2 packet immediately, without queuing or deferring.
2428  */
2429 static void ssh2_pkt_send_noqueue(Ssh ssh, struct Packet *pkt)
2430 {
2431     int len;
2432     int backlog;
2433     if (ssh->cscipher != NULL && (ssh->cscipher->flags & SSH_CIPHER_IS_CBC)) {
2434         /* We need to send two packets, so use the deferral mechanism. */
2435         ssh2_pkt_defer_noqueue(ssh, pkt, FALSE);
2436         ssh_pkt_defersend(ssh);
2437         return;
2438     }
2439     len = ssh2_pkt_construct(ssh, pkt);
2440     backlog = s_write(ssh, pkt->body, len);
2441     if (backlog > SSH_MAX_BACKLOG)
2442         ssh_throttle_all(ssh, 1, backlog);
2443
2444     ssh->outgoing_data_size += pkt->encrypted_len;
2445     if (!ssh->kex_in_progress &&
2446         !ssh->bare_connection &&
2447         ssh->max_data_size != 0 &&
2448         ssh->outgoing_data_size > ssh->max_data_size)
2449         do_ssh2_transport(ssh, "too much data sent", -1, NULL);
2450
2451     ssh_free_packet(pkt);
2452 }
2453
2454 /*
2455  * Defer an SSH-2 packet.
2456  */
2457 static void ssh2_pkt_defer_noqueue(Ssh ssh, struct Packet *pkt, int noignore)
2458 {
2459     int len;
2460     if (ssh->cscipher != NULL && (ssh->cscipher->flags & SSH_CIPHER_IS_CBC) &&
2461         ssh->deferred_len == 0 && !noignore &&
2462         !(ssh->remote_bugs & BUG_CHOKES_ON_SSH2_IGNORE)) {
2463         /*
2464          * Interpose an SSH_MSG_IGNORE to ensure that user data don't
2465          * get encrypted with a known IV.
2466          */
2467         struct Packet *ipkt = ssh2_pkt_init(SSH2_MSG_IGNORE);
2468         ssh2_pkt_addstring_start(ipkt);
2469         ssh2_pkt_defer_noqueue(ssh, ipkt, TRUE);
2470     }
2471     len = ssh2_pkt_construct(ssh, pkt);
2472     if (ssh->deferred_len + len > ssh->deferred_size) {
2473         ssh->deferred_size = ssh->deferred_len + len + 128;
2474         ssh->deferred_send_data = sresize(ssh->deferred_send_data,
2475                                           ssh->deferred_size,
2476                                           unsigned char);
2477     }
2478     memcpy(ssh->deferred_send_data + ssh->deferred_len, pkt->body, len);
2479     ssh->deferred_len += len;
2480     ssh->deferred_data_size += pkt->encrypted_len;
2481     ssh_free_packet(pkt);
2482 }
2483
2484 /*
2485  * Queue an SSH-2 packet.
2486  */
2487 static void ssh2_pkt_queue(Ssh ssh, struct Packet *pkt)
2488 {
2489     assert(ssh->queueing);
2490
2491     if (ssh->queuelen >= ssh->queuesize) {
2492         ssh->queuesize = ssh->queuelen + 32;
2493         ssh->queue = sresize(ssh->queue, ssh->queuesize, struct Packet *);
2494     }
2495
2496     ssh->queue[ssh->queuelen++] = pkt;
2497 }
2498
2499 /*
2500  * Either queue or send a packet, depending on whether queueing is
2501  * set.
2502  */
2503 static void ssh2_pkt_send(Ssh ssh, struct Packet *pkt)
2504 {
2505     if (ssh->queueing)
2506         ssh2_pkt_queue(ssh, pkt);
2507     else
2508         ssh2_pkt_send_noqueue(ssh, pkt);
2509 }
2510
2511 /*
2512  * Either queue or defer a packet, depending on whether queueing is
2513  * set.
2514  */
2515 static void ssh2_pkt_defer(Ssh ssh, struct Packet *pkt)
2516 {
2517     if (ssh->queueing)
2518         ssh2_pkt_queue(ssh, pkt);
2519     else
2520         ssh2_pkt_defer_noqueue(ssh, pkt, FALSE);
2521 }
2522
2523 /*
2524  * Send the whole deferred data block constructed by
2525  * ssh2_pkt_defer() or SSH-1's defer_packet().
2526  * 
2527  * The expected use of the defer mechanism is that you call
2528  * ssh2_pkt_defer() a few times, then call ssh_pkt_defersend(). If
2529  * not currently queueing, this simply sets up deferred_send_data
2530  * and then sends it. If we _are_ currently queueing, the calls to
2531  * ssh2_pkt_defer() put the deferred packets on to the queue
2532  * instead, and therefore ssh_pkt_defersend() has no deferred data
2533  * to send. Hence, there's no need to make it conditional on
2534  * ssh->queueing.
2535  */
2536 static void ssh_pkt_defersend(Ssh ssh)
2537 {
2538     int backlog;
2539     backlog = s_write(ssh, ssh->deferred_send_data, ssh->deferred_len);
2540     ssh->deferred_len = ssh->deferred_size = 0;
2541     sfree(ssh->deferred_send_data);
2542     ssh->deferred_send_data = NULL;
2543     if (backlog > SSH_MAX_BACKLOG)
2544         ssh_throttle_all(ssh, 1, backlog);
2545
2546     if (ssh->version == 2) {
2547         ssh->outgoing_data_size += ssh->deferred_data_size;
2548         ssh->deferred_data_size = 0;
2549         if (!ssh->kex_in_progress &&
2550             !ssh->bare_connection &&
2551             ssh->max_data_size != 0 &&
2552             ssh->outgoing_data_size > ssh->max_data_size)
2553             do_ssh2_transport(ssh, "too much data sent", -1, NULL);
2554     }
2555 }
2556
2557 /*
2558  * Send a packet whose length needs to be disguised (typically
2559  * passwords or keyboard-interactive responses).
2560  */
2561 static void ssh2_pkt_send_with_padding(Ssh ssh, struct Packet *pkt,
2562                                        int padsize)
2563 {
2564 #if 0
2565     if (0) {
2566         /*
2567          * The simplest way to do this is to adjust the
2568          * variable-length padding field in the outgoing packet.
2569          * 
2570          * Currently compiled out, because some Cisco SSH servers
2571          * don't like excessively padded packets (bah, why's it
2572          * always Cisco?)
2573          */
2574         pkt->forcepad = padsize;
2575         ssh2_pkt_send(ssh, pkt);
2576     } else
2577 #endif
2578     {
2579         /*
2580          * If we can't do that, however, an alternative approach is
2581          * to use the pkt_defer mechanism to bundle the packet
2582          * tightly together with an SSH_MSG_IGNORE such that their
2583          * combined length is a constant. So first we construct the
2584          * final form of this packet and defer its sending.
2585          */
2586         ssh2_pkt_defer(ssh, pkt);
2587
2588         /*
2589          * Now construct an SSH_MSG_IGNORE which includes a string
2590          * that's an exact multiple of the cipher block size. (If
2591          * the cipher is NULL so that the block size is
2592          * unavailable, we don't do this trick at all, because we
2593          * gain nothing by it.)
2594          */
2595         if (ssh->cscipher &&
2596             !(ssh->remote_bugs & BUG_CHOKES_ON_SSH2_IGNORE)) {
2597             int stringlen, i;
2598
2599             stringlen = (256 - ssh->deferred_len);
2600             stringlen += ssh->cscipher->blksize - 1;
2601             stringlen -= (stringlen % ssh->cscipher->blksize);
2602             if (ssh->cscomp) {
2603                 /*
2604                  * Temporarily disable actual compression, so we
2605                  * can guarantee to get this string exactly the
2606                  * length we want it. The compression-disabling
2607                  * routine should return an integer indicating how
2608                  * many bytes we should adjust our string length
2609                  * by.
2610                  */
2611                 stringlen -=
2612                     ssh->cscomp->disable_compression(ssh->cs_comp_ctx);
2613             }
2614             pkt = ssh2_pkt_init(SSH2_MSG_IGNORE);
2615             ssh2_pkt_addstring_start(pkt);
2616             for (i = 0; i < stringlen; i++) {
2617                 char c = (char) random_byte();
2618                 ssh2_pkt_addstring_data(pkt, &c, 1);
2619             }
2620             ssh2_pkt_defer(ssh, pkt);
2621         }
2622         ssh_pkt_defersend(ssh);
2623     }
2624 }
2625
2626 /*
2627  * Send all queued SSH-2 packets. We send them by means of
2628  * ssh2_pkt_defer_noqueue(), in case they included a pair of
2629  * packets that needed to be lumped together.
2630  */
2631 static void ssh2_pkt_queuesend(Ssh ssh)
2632 {
2633     int i;
2634
2635     assert(!ssh->queueing);
2636
2637     for (i = 0; i < ssh->queuelen; i++)
2638         ssh2_pkt_defer_noqueue(ssh, ssh->queue[i], FALSE);
2639     ssh->queuelen = 0;
2640
2641     ssh_pkt_defersend(ssh);
2642 }
2643
2644 #if 0
2645 void bndebug(char *string, Bignum b)
2646 {
2647     unsigned char *p;
2648     int i, len;
2649     p = ssh2_mpint_fmt(b, &len);
2650     debug(("%s", string));
2651     for (i = 0; i < len; i++)
2652         debug((" %02x", p[i]));
2653     debug(("\n"));
2654     sfree(p);
2655 }
2656 #endif
2657
2658 static void hash_mpint(const struct ssh_hash *h, void *s, Bignum b)
2659 {
2660     unsigned char *p;
2661     int len;
2662     p = ssh2_mpint_fmt(b, &len);
2663     hash_string(h, s, p, len);
2664     sfree(p);
2665 }
2666
2667 /*
2668  * Packet decode functions for both SSH-1 and SSH-2.
2669  */
2670 static unsigned long ssh_pkt_getuint32(struct Packet *pkt)
2671 {
2672     unsigned long value;
2673     if (pkt->length - pkt->savedpos < 4)
2674         return 0;                      /* arrgh, no way to decline (FIXME?) */
2675     value = GET_32BIT(pkt->body + pkt->savedpos);
2676     pkt->savedpos += 4;
2677     return value;
2678 }
2679 static int ssh2_pkt_getbool(struct Packet *pkt)
2680 {
2681     unsigned long value;
2682     if (pkt->length - pkt->savedpos < 1)
2683         return 0;                      /* arrgh, no way to decline (FIXME?) */
2684     value = pkt->body[pkt->savedpos] != 0;
2685     pkt->savedpos++;
2686     return value;
2687 }
2688 static void ssh_pkt_getstring(struct Packet *pkt, char **p, int *length)
2689 {
2690     int len;
2691     *p = NULL;
2692     *length = 0;
2693     if (pkt->length - pkt->savedpos < 4)
2694         return;
2695     len = toint(GET_32BIT(pkt->body + pkt->savedpos));
2696     if (len < 0)
2697         return;
2698     *length = len;
2699     pkt->savedpos += 4;
2700     if (pkt->length - pkt->savedpos < *length)
2701         return;
2702     *p = (char *)(pkt->body + pkt->savedpos);
2703     pkt->savedpos += *length;
2704 }
2705 static void *ssh_pkt_getdata(struct Packet *pkt, int length)
2706 {
2707     if (pkt->length - pkt->savedpos < length)
2708         return NULL;
2709     pkt->savedpos += length;
2710     return pkt->body + (pkt->savedpos - length);
2711 }
2712 static int ssh1_pkt_getrsakey(struct Packet *pkt, struct RSAKey *key,
2713                               const unsigned char **keystr)
2714 {
2715     int j;
2716
2717     j = makekey(pkt->body + pkt->savedpos,
2718                 pkt->length - pkt->savedpos,
2719                 key, keystr, 0);
2720
2721     if (j < 0)
2722         return FALSE;
2723     
2724     pkt->savedpos += j;
2725     assert(pkt->savedpos < pkt->length);
2726
2727     return TRUE;
2728 }
2729 static Bignum ssh1_pkt_getmp(struct Packet *pkt)
2730 {
2731     int j;
2732     Bignum b;
2733
2734     j = ssh1_read_bignum(pkt->body + pkt->savedpos,
2735                          pkt->length - pkt->savedpos, &b);
2736
2737     if (j < 0)
2738         return NULL;
2739
2740     pkt->savedpos += j;
2741     return b;
2742 }
2743 static Bignum ssh2_pkt_getmp(struct Packet *pkt)
2744 {
2745     char *p;
2746     int length;
2747     Bignum b;
2748
2749     ssh_pkt_getstring(pkt, &p, &length);
2750     if (!p)
2751         return NULL;
2752     if (p[0] & 0x80)
2753         return NULL;
2754     b = bignum_from_bytes((unsigned char *)p, length);
2755     return b;
2756 }
2757
2758 /*
2759  * Helper function to add an SSH-2 signature blob to a packet.
2760  * Expects to be shown the public key blob as well as the signature
2761  * blob. Normally works just like ssh2_pkt_addstring, but will
2762  * fiddle with the signature packet if necessary for
2763  * BUG_SSH2_RSA_PADDING.
2764  */
2765 static void ssh2_add_sigblob(Ssh ssh, struct Packet *pkt,
2766                              void *pkblob_v, int pkblob_len,
2767                              void *sigblob_v, int sigblob_len)
2768 {
2769     unsigned char *pkblob = (unsigned char *)pkblob_v;
2770     unsigned char *sigblob = (unsigned char *)sigblob_v;
2771
2772     /* dmemdump(pkblob, pkblob_len); */
2773     /* dmemdump(sigblob, sigblob_len); */
2774
2775     /*
2776      * See if this is in fact an ssh-rsa signature and a buggy
2777      * server; otherwise we can just do this the easy way.
2778      */
2779     if ((ssh->remote_bugs & BUG_SSH2_RSA_PADDING) && pkblob_len > 4+7+4 &&
2780         (GET_32BIT(pkblob) == 7 && !memcmp(pkblob+4, "ssh-rsa", 7))) {
2781         int pos, len, siglen;
2782
2783         /*
2784          * Find the byte length of the modulus.
2785          */
2786
2787         pos = 4+7;                     /* skip over "ssh-rsa" */
2788         len = toint(GET_32BIT(pkblob+pos)); /* get length of exponent */
2789         if (len < 0 || len > pkblob_len - pos - 4)
2790             goto give_up;
2791         pos += 4 + len;                /* skip over exponent */
2792         if (pkblob_len - pos < 4)
2793             goto give_up;
2794         len = toint(GET_32BIT(pkblob+pos)); /* find length of modulus */
2795         if (len < 0 || len > pkblob_len - pos - 4)
2796             goto give_up;
2797         pos += 4;                      /* find modulus itself */
2798         while (len > 0 && pkblob[pos] == 0)
2799             len--, pos++;
2800         /* debug(("modulus length is %d\n", len)); */
2801
2802         /*
2803          * Now find the signature integer.
2804          */
2805         pos = 4+7;                     /* skip over "ssh-rsa" */
2806         if (sigblob_len < pos+4)
2807             goto give_up;
2808         siglen = toint(GET_32BIT(sigblob+pos));
2809         if (siglen != sigblob_len - pos - 4)
2810             goto give_up;
2811         /* debug(("signature length is %d\n", siglen)); */
2812
2813         if (len != siglen) {
2814             unsigned char newlen[4];
2815             ssh2_pkt_addstring_start(pkt);
2816             ssh2_pkt_addstring_data(pkt, (char *)sigblob, pos);
2817             /* dmemdump(sigblob, pos); */
2818             pos += 4;                  /* point to start of actual sig */
2819             PUT_32BIT(newlen, len);
2820             ssh2_pkt_addstring_data(pkt, (char *)newlen, 4);
2821             /* dmemdump(newlen, 4); */
2822             newlen[0] = 0;
2823             while (len-- > siglen) {
2824                 ssh2_pkt_addstring_data(pkt, (char *)newlen, 1);
2825                 /* dmemdump(newlen, 1); */
2826             }
2827             ssh2_pkt_addstring_data(pkt, (char *)(sigblob+pos), siglen);
2828             /* dmemdump(sigblob+pos, siglen); */
2829             return;
2830         }
2831
2832         /* Otherwise fall through and do it the easy way. We also come
2833          * here as a fallback if we discover above that the key blob
2834          * is misformatted in some way. */
2835       give_up:;
2836     }
2837
2838     ssh2_pkt_addstring_start(pkt);
2839     ssh2_pkt_addstring_data(pkt, (char *)sigblob, sigblob_len);
2840 }
2841
2842 /*
2843  * Examine the remote side's version string and compare it against
2844  * a list of known buggy implementations.
2845  */
2846 static void ssh_detect_bugs(Ssh ssh, char *vstring)
2847 {
2848     char *imp;                         /* pointer to implementation part */
2849     imp = vstring;
2850     imp += strcspn(imp, "-");
2851     if (*imp) imp++;
2852     imp += strcspn(imp, "-");
2853     if (*imp) imp++;
2854
2855     ssh->remote_bugs = 0;
2856
2857     /*
2858      * General notes on server version strings:
2859      *  - Not all servers reporting "Cisco-1.25" have all the bugs listed
2860      *    here -- in particular, we've heard of one that's perfectly happy
2861      *    with SSH1_MSG_IGNOREs -- but this string never seems to change,
2862      *    so we can't distinguish them.
2863      */
2864     if (conf_get_int(ssh->conf, CONF_sshbug_ignore1) == FORCE_ON ||
2865         (conf_get_int(ssh->conf, CONF_sshbug_ignore1) == AUTO &&
2866          (!strcmp(imp, "1.2.18") || !strcmp(imp, "1.2.19") ||
2867           !strcmp(imp, "1.2.20") || !strcmp(imp, "1.2.21") ||
2868           !strcmp(imp, "1.2.22") || !strcmp(imp, "Cisco-1.25") ||
2869           !strcmp(imp, "OSU_1.4alpha3") || !strcmp(imp, "OSU_1.5alpha4")))) {
2870         /*
2871          * These versions don't support SSH1_MSG_IGNORE, so we have
2872          * to use a different defence against password length
2873          * sniffing.
2874          */
2875         ssh->remote_bugs |= BUG_CHOKES_ON_SSH1_IGNORE;
2876         logevent("We believe remote version has SSH-1 ignore bug");
2877     }
2878
2879     if (conf_get_int(ssh->conf, CONF_sshbug_plainpw1) == FORCE_ON ||
2880         (conf_get_int(ssh->conf, CONF_sshbug_plainpw1) == AUTO &&
2881          (!strcmp(imp, "Cisco-1.25") || !strcmp(imp, "OSU_1.4alpha3")))) {
2882         /*
2883          * These versions need a plain password sent; they can't
2884          * handle having a null and a random length of data after
2885          * the password.
2886          */
2887         ssh->remote_bugs |= BUG_NEEDS_SSH1_PLAIN_PASSWORD;
2888         logevent("We believe remote version needs a plain SSH-1 password");
2889     }
2890
2891     if (conf_get_int(ssh->conf, CONF_sshbug_rsa1) == FORCE_ON ||
2892         (conf_get_int(ssh->conf, CONF_sshbug_rsa1) == AUTO &&
2893          (!strcmp(imp, "Cisco-1.25")))) {
2894         /*
2895          * These versions apparently have no clue whatever about
2896          * RSA authentication and will panic and die if they see
2897          * an AUTH_RSA message.
2898          */
2899         ssh->remote_bugs |= BUG_CHOKES_ON_RSA;
2900         logevent("We believe remote version can't handle SSH-1 RSA authentication");
2901     }
2902
2903     if (conf_get_int(ssh->conf, CONF_sshbug_hmac2) == FORCE_ON ||
2904         (conf_get_int(ssh->conf, CONF_sshbug_hmac2) == AUTO &&
2905          !wc_match("* VShell", imp) &&
2906          (wc_match("2.1.0*", imp) || wc_match("2.0.*", imp) ||
2907           wc_match("2.2.0*", imp) || wc_match("2.3.0*", imp) ||
2908           wc_match("2.1 *", imp)))) {
2909         /*
2910          * These versions have the HMAC bug.
2911          */
2912         ssh->remote_bugs |= BUG_SSH2_HMAC;
2913         logevent("We believe remote version has SSH-2 HMAC bug");
2914     }
2915
2916     if (conf_get_int(ssh->conf, CONF_sshbug_derivekey2) == FORCE_ON ||
2917         (conf_get_int(ssh->conf, CONF_sshbug_derivekey2) == AUTO &&
2918          !wc_match("* VShell", imp) &&
2919          (wc_match("2.0.0*", imp) || wc_match("2.0.10*", imp) ))) {
2920         /*
2921          * These versions have the key-derivation bug (failing to
2922          * include the literal shared secret in the hashes that
2923          * generate the keys).
2924          */
2925         ssh->remote_bugs |= BUG_SSH2_DERIVEKEY;
2926         logevent("We believe remote version has SSH-2 key-derivation bug");
2927     }
2928
2929     if (conf_get_int(ssh->conf, CONF_sshbug_rsapad2) == FORCE_ON ||
2930         (conf_get_int(ssh->conf, CONF_sshbug_rsapad2) == AUTO &&
2931          (wc_match("OpenSSH_2.[5-9]*", imp) ||
2932           wc_match("OpenSSH_3.[0-2]*", imp) ||
2933           wc_match("mod_sftp/0.[0-8]*", imp) ||
2934           wc_match("mod_sftp/0.9.[0-8]", imp)))) {
2935         /*
2936          * These versions have the SSH-2 RSA padding bug.
2937          */
2938         ssh->remote_bugs |= BUG_SSH2_RSA_PADDING;
2939         logevent("We believe remote version has SSH-2 RSA padding bug");
2940     }
2941
2942     if (conf_get_int(ssh->conf, CONF_sshbug_pksessid2) == FORCE_ON ||
2943         (conf_get_int(ssh->conf, CONF_sshbug_pksessid2) == AUTO &&
2944          wc_match("OpenSSH_2.[0-2]*", imp))) {
2945         /*
2946          * These versions have the SSH-2 session-ID bug in
2947          * public-key authentication.
2948          */
2949         ssh->remote_bugs |= BUG_SSH2_PK_SESSIONID;
2950         logevent("We believe remote version has SSH-2 public-key-session-ID bug");
2951     }
2952
2953     if (conf_get_int(ssh->conf, CONF_sshbug_rekey2) == FORCE_ON ||
2954         (conf_get_int(ssh->conf, CONF_sshbug_rekey2) == AUTO &&
2955          (wc_match("DigiSSH_2.0", imp) ||
2956           wc_match("OpenSSH_2.[0-4]*", imp) ||
2957           wc_match("OpenSSH_2.5.[0-3]*", imp) ||
2958           wc_match("Sun_SSH_1.0", imp) ||
2959           wc_match("Sun_SSH_1.0.1", imp) ||
2960           /* All versions <= 1.2.6 (they changed their format in 1.2.7) */
2961           wc_match("WeOnlyDo-*", imp)))) {
2962         /*
2963          * These versions have the SSH-2 rekey bug.
2964          */
2965         ssh->remote_bugs |= BUG_SSH2_REKEY;
2966         logevent("We believe remote version has SSH-2 rekey bug");
2967     }
2968
2969     if (conf_get_int(ssh->conf, CONF_sshbug_maxpkt2) == FORCE_ON ||
2970         (conf_get_int(ssh->conf, CONF_sshbug_maxpkt2) == AUTO &&
2971          (wc_match("1.36_sshlib GlobalSCAPE", imp) ||
2972           wc_match("1.36 sshlib: GlobalScape", imp)))) {
2973         /*
2974          * This version ignores our makpkt and needs to be throttled.
2975          */
2976         ssh->remote_bugs |= BUG_SSH2_MAXPKT;
2977         logevent("We believe remote version ignores SSH-2 maximum packet size");
2978     }
2979
2980     if (conf_get_int(ssh->conf, CONF_sshbug_ignore2) == FORCE_ON) {
2981         /*
2982          * Servers that don't support SSH2_MSG_IGNORE. Currently,
2983          * none detected automatically.
2984          */
2985         ssh->remote_bugs |= BUG_CHOKES_ON_SSH2_IGNORE;
2986         logevent("We believe remote version has SSH-2 ignore bug");
2987     }
2988
2989     if (conf_get_int(ssh->conf, CONF_sshbug_oldgex2) == FORCE_ON ||
2990         (conf_get_int(ssh->conf, CONF_sshbug_oldgex2) == AUTO &&
2991          (wc_match("OpenSSH_2.[235]*", imp)))) {
2992         /*
2993          * These versions only support the original (pre-RFC4419)
2994          * SSH-2 GEX request, and disconnect with a protocol error if
2995          * we use the newer version.
2996          */
2997         ssh->remote_bugs |= BUG_SSH2_OLDGEX;
2998         logevent("We believe remote version has outdated SSH-2 GEX");
2999     }
3000
3001     if (conf_get_int(ssh->conf, CONF_sshbug_winadj) == FORCE_ON) {
3002         /*
3003          * Servers that don't support our winadj request for one
3004          * reason or another. Currently, none detected automatically.
3005          */
3006         ssh->remote_bugs |= BUG_CHOKES_ON_WINADJ;
3007         logevent("We believe remote version has winadj bug");
3008     }
3009
3010     if (conf_get_int(ssh->conf, CONF_sshbug_chanreq) == FORCE_ON ||
3011         (conf_get_int(ssh->conf, CONF_sshbug_chanreq) == AUTO &&
3012          (wc_match("OpenSSH_[2-5].*", imp) ||
3013           wc_match("OpenSSH_6.[0-6]*", imp) ||
3014           wc_match("dropbear_0.[2-4][0-9]*", imp) ||
3015           wc_match("dropbear_0.5[01]*", imp)))) {
3016         /*
3017          * These versions have the SSH-2 channel request bug.
3018          * OpenSSH 6.7 and above do not:
3019          * https://bugzilla.mindrot.org/show_bug.cgi?id=1818
3020          * dropbear_0.52 and above do not:
3021          * https://secure.ucc.asn.au/hg/dropbear/rev/cd02449b709c
3022          */
3023         ssh->remote_bugs |= BUG_SENDS_LATE_REQUEST_REPLY;
3024         logevent("We believe remote version has SSH-2 channel request bug");
3025     }
3026 }
3027
3028 /*
3029  * The `software version' part of an SSH version string is required
3030  * to contain no spaces or minus signs.
3031  */
3032 static void ssh_fix_verstring(char *str)
3033 {
3034     /* Eat "<protoversion>-". */
3035     while (*str && *str != '-') str++;
3036     assert(*str == '-'); str++;
3037
3038     /* Convert minus signs and spaces in the remaining string into
3039      * underscores. */
3040     while (*str) {
3041         if (*str == '-' || *str == ' ')
3042             *str = '_';
3043         str++;
3044     }
3045 }
3046
3047 /*
3048  * Send an appropriate SSH version string.
3049  */
3050 static void ssh_send_verstring(Ssh ssh, const char *protoname, char *svers)
3051 {
3052     char *verstring;
3053
3054     if (ssh->version == 2) {
3055         /*
3056          * Construct a v2 version string.
3057          */
3058         verstring = dupprintf("%s2.0-%s\015\012", protoname, sshver);
3059     } else {
3060         /*
3061          * Construct a v1 version string.
3062          */
3063         assert(!strcmp(protoname, "SSH-")); /* no v1 bare connection protocol */
3064         verstring = dupprintf("SSH-%s-%s\012",
3065                               (ssh_versioncmp(svers, "1.5") <= 0 ?
3066                                svers : "1.5"),
3067                               sshver);
3068     }
3069
3070     ssh_fix_verstring(verstring + strlen(protoname));
3071 #ifdef FUZZING
3072     /* FUZZING make PuTTY insecure, so make live use difficult. */
3073     verstring[0] = 'I';
3074 #endif
3075
3076     if (ssh->version == 2) {
3077         size_t len;
3078         /*
3079          * Record our version string.
3080          */
3081         len = strcspn(verstring, "\015\012");
3082         ssh->v_c = snewn(len + 1, char);
3083         memcpy(ssh->v_c, verstring, len);
3084         ssh->v_c[len] = 0;
3085     }
3086
3087     logeventf(ssh, "We claim version: %.*s",
3088               strcspn(verstring, "\015\012"), verstring);
3089     s_write(ssh, verstring, strlen(verstring));
3090     sfree(verstring);
3091 }
3092
3093 static int do_ssh_init(Ssh ssh, unsigned char c)
3094 {
3095     static const char protoname[] = "SSH-";
3096
3097     struct do_ssh_init_state {
3098         int crLine;
3099         int vslen;
3100         char version[10];
3101         char *vstring;
3102         int vstrsize;
3103         int i;
3104         int proto1, proto2;
3105     };
3106     crState(do_ssh_init_state);
3107     
3108     crBeginState;
3109
3110     /* Search for a line beginning with the protocol name prefix in
3111      * the input. */
3112     for (;;) {
3113         for (s->i = 0; protoname[s->i]; s->i++) {
3114             if ((char)c != protoname[s->i]) goto no;
3115             crReturn(1);
3116         }
3117         break;
3118       no:
3119         while (c != '\012')
3120             crReturn(1);
3121         crReturn(1);
3122     }
3123
3124     ssh->session_started = TRUE;
3125
3126     s->vstrsize = sizeof(protoname) + 16;
3127     s->vstring = snewn(s->vstrsize, char);
3128     strcpy(s->vstring, protoname);
3129     s->vslen = strlen(protoname);
3130     s->i = 0;
3131     while (1) {
3132         if (s->vslen >= s->vstrsize - 1) {
3133             s->vstrsize += 16;
3134             s->vstring = sresize(s->vstring, s->vstrsize, char);
3135         }
3136         s->vstring[s->vslen++] = c;
3137         if (s->i >= 0) {
3138             if (c == '-') {
3139                 s->version[s->i] = '\0';
3140                 s->i = -1;
3141             } else if (s->i < sizeof(s->version) - 1)
3142                 s->version[s->i++] = c;
3143         } else if (c == '\012')
3144             break;
3145         crReturn(1);                   /* get another char */
3146     }
3147
3148     ssh->agentfwd_enabled = FALSE;
3149     ssh->rdpkt2_state.incoming_sequence = 0;
3150
3151     s->vstring[s->vslen] = 0;
3152     s->vstring[strcspn(s->vstring, "\015\012")] = '\0';/* remove EOL chars */
3153     logeventf(ssh, "Server version: %s", s->vstring);
3154     ssh_detect_bugs(ssh, s->vstring);
3155
3156     /*
3157      * Decide which SSH protocol version to support.
3158      */
3159
3160     /* Anything strictly below "2.0" means protocol 1 is supported. */
3161     s->proto1 = ssh_versioncmp(s->version, "2.0") < 0;
3162     /* Anything greater or equal to "1.99" means protocol 2 is supported. */
3163     s->proto2 = ssh_versioncmp(s->version, "1.99") >= 0;
3164
3165     if (conf_get_int(ssh->conf, CONF_sshprot) == 0) {
3166         if (!s->proto1) {
3167             bombout(("SSH protocol version 1 required by our configuration "
3168                      "but not provided by server"));
3169             crStop(0);
3170         }
3171     } else if (conf_get_int(ssh->conf, CONF_sshprot) == 3) {
3172         if (!s->proto2) {
3173             bombout(("SSH protocol version 2 required by our configuration "
3174                      "but server only provides (old, insecure) SSH-1"));
3175             crStop(0);
3176         }
3177     } else {
3178         /* No longer support values 1 or 2 for CONF_sshprot */
3179         assert(!"Unexpected value for CONF_sshprot");
3180     }
3181
3182     if (s->proto2 && (conf_get_int(ssh->conf, CONF_sshprot) >= 2 || !s->proto1))
3183         ssh->version = 2;
3184     else
3185         ssh->version = 1;
3186
3187     logeventf(ssh, "Using SSH protocol version %d", ssh->version);
3188
3189     /* Send the version string, if we haven't already */
3190     if (conf_get_int(ssh->conf, CONF_sshprot) != 3)
3191         ssh_send_verstring(ssh, protoname, s->version);
3192
3193     if (ssh->version == 2) {
3194         size_t len;
3195         /*
3196          * Record their version string.
3197          */
3198         len = strcspn(s->vstring, "\015\012");
3199         ssh->v_s = snewn(len + 1, char);
3200         memcpy(ssh->v_s, s->vstring, len);
3201         ssh->v_s[len] = 0;
3202             
3203         /*
3204          * Initialise SSH-2 protocol.
3205          */
3206         ssh->protocol = ssh2_protocol;
3207         ssh2_protocol_setup(ssh);
3208         ssh->s_rdpkt = ssh2_rdpkt;
3209     } else {
3210         /*
3211          * Initialise SSH-1 protocol.
3212          */
3213         ssh->protocol = ssh1_protocol;
3214         ssh1_protocol_setup(ssh);
3215         ssh->s_rdpkt = ssh1_rdpkt;
3216     }
3217     if (ssh->version == 2)
3218         do_ssh2_transport(ssh, NULL, -1, NULL);
3219
3220     update_specials_menu(ssh->frontend);
3221     ssh->state = SSH_STATE_BEFORE_SIZE;
3222     ssh->pinger = pinger_new(ssh->conf, &ssh_backend, ssh);
3223
3224     sfree(s->vstring);
3225
3226     crFinish(0);
3227 }
3228
3229 static int do_ssh_connection_init(Ssh ssh, unsigned char c)
3230 {
3231     /*
3232      * Ordinary SSH begins with the banner "SSH-x.y-...". This is just
3233      * the ssh-connection part, extracted and given a trivial binary
3234      * packet protocol, so we replace 'SSH-' at the start with a new
3235      * name. In proper SSH style (though of course this part of the
3236      * proper SSH protocol _isn't_ subject to this kind of
3237      * DNS-domain-based extension), we define the new name in our
3238      * extension space.
3239      */
3240     static const char protoname[] =
3241         "SSHCONNECTION@putty.projects.tartarus.org-";
3242
3243     struct do_ssh_connection_init_state {
3244         int crLine;
3245         int vslen;
3246         char version[10];
3247         char *vstring;
3248         int vstrsize;
3249         int i;
3250     };
3251     crState(do_ssh_connection_init_state);
3252     
3253     crBeginState;
3254
3255     /* Search for a line beginning with the protocol name prefix in
3256      * the input. */
3257     for (;;) {
3258         for (s->i = 0; protoname[s->i]; s->i++) {
3259             if ((char)c != protoname[s->i]) goto no;
3260             crReturn(1);
3261         }
3262         break;
3263       no:
3264         while (c != '\012')
3265             crReturn(1);
3266         crReturn(1);
3267     }
3268
3269     s->vstrsize = sizeof(protoname) + 16;
3270     s->vstring = snewn(s->vstrsize, char);
3271     strcpy(s->vstring, protoname);
3272     s->vslen = strlen(protoname);
3273     s->i = 0;
3274     while (1) {
3275         if (s->vslen >= s->vstrsize - 1) {
3276             s->vstrsize += 16;
3277             s->vstring = sresize(s->vstring, s->vstrsize, char);
3278         }
3279         s->vstring[s->vslen++] = c;
3280         if (s->i >= 0) {
3281             if (c == '-') {
3282                 s->version[s->i] = '\0';
3283                 s->i = -1;
3284             } else if (s->i < sizeof(s->version) - 1)
3285                 s->version[s->i++] = c;
3286         } else if (c == '\012')
3287             break;
3288         crReturn(1);                   /* get another char */
3289     }
3290
3291     ssh->agentfwd_enabled = FALSE;
3292     ssh->rdpkt2_bare_state.incoming_sequence = 0;
3293
3294     s->vstring[s->vslen] = 0;
3295     s->vstring[strcspn(s->vstring, "\015\012")] = '\0';/* remove EOL chars */
3296     logeventf(ssh, "Server version: %s", s->vstring);
3297     ssh_detect_bugs(ssh, s->vstring);
3298
3299     /*
3300      * Decide which SSH protocol version to support. This is easy in
3301      * bare ssh-connection mode: only 2.0 is legal.
3302      */
3303     if (ssh_versioncmp(s->version, "2.0") < 0) {
3304         bombout(("Server announces compatibility with SSH-1 in bare ssh-connection protocol"));
3305         crStop(0);
3306     }
3307     if (conf_get_int(ssh->conf, CONF_sshprot) == 0) {
3308         bombout(("Bare ssh-connection protocol cannot be run in SSH-1-only mode"));
3309         crStop(0);
3310     }
3311
3312     ssh->version = 2;
3313
3314     logeventf(ssh, "Using bare ssh-connection protocol");
3315
3316     /* Send the version string, if we haven't already */
3317     ssh_send_verstring(ssh, protoname, s->version);
3318
3319     /*
3320      * Initialise bare connection protocol.
3321      */
3322     ssh->protocol = ssh2_bare_connection_protocol;
3323     ssh2_bare_connection_protocol_setup(ssh);
3324     ssh->s_rdpkt = ssh2_bare_connection_rdpkt;
3325
3326     update_specials_menu(ssh->frontend);
3327     ssh->state = SSH_STATE_BEFORE_SIZE;
3328     ssh->pinger = pinger_new(ssh->conf, &ssh_backend, ssh);
3329
3330     /*
3331      * Get authconn (really just conn) under way.
3332      */
3333     do_ssh2_authconn(ssh, NULL, 0, NULL);
3334
3335     sfree(s->vstring);
3336
3337     crFinish(0);
3338 }
3339
3340 static void ssh_process_incoming_data(Ssh ssh,
3341                                       const unsigned char **data, int *datalen)
3342 {
3343     struct Packet *pktin;
3344
3345     pktin = ssh->s_rdpkt(ssh, data, datalen);
3346     if (pktin) {
3347         ssh->protocol(ssh, NULL, 0, pktin);
3348         ssh_free_packet(pktin);
3349     }
3350 }
3351
3352 static void ssh_queue_incoming_data(Ssh ssh,
3353                                     const unsigned char **data, int *datalen)
3354 {
3355     bufchain_add(&ssh->queued_incoming_data, *data, *datalen);
3356     *data += *datalen;
3357     *datalen = 0;
3358 }
3359
3360 static void ssh_process_queued_incoming_data(Ssh ssh)
3361 {
3362     void *vdata;
3363     const unsigned char *data;
3364     int len, origlen;
3365
3366     while (!ssh->frozen && bufchain_size(&ssh->queued_incoming_data)) {
3367         bufchain_prefix(&ssh->queued_incoming_data, &vdata, &len);
3368         data = vdata;
3369         origlen = len;
3370
3371         while (!ssh->frozen && len > 0)
3372             ssh_process_incoming_data(ssh, &data, &len);
3373
3374         if (origlen > len)
3375             bufchain_consume(&ssh->queued_incoming_data, origlen - len);
3376     }
3377 }
3378
3379 static void ssh_set_frozen(Ssh ssh, int frozen)
3380 {
3381     if (ssh->s)
3382         sk_set_frozen(ssh->s, frozen);
3383     ssh->frozen = frozen;
3384 }
3385
3386 static void ssh_gotdata(Ssh ssh, const unsigned char *data, int datalen)
3387 {
3388     /* Log raw data, if we're in that mode. */
3389     if (ssh->logctx)
3390         log_packet(ssh->logctx, PKT_INCOMING, -1, NULL, data, datalen,
3391                    0, NULL, NULL, 0, NULL);
3392
3393     crBegin(ssh->ssh_gotdata_crstate);
3394
3395     /*
3396      * To begin with, feed the characters one by one to the
3397      * protocol initialisation / selection function do_ssh_init().
3398      * When that returns 0, we're done with the initial greeting
3399      * exchange and can move on to packet discipline.
3400      */
3401     while (1) {
3402         int ret;                       /* need not be kept across crReturn */
3403         if (datalen == 0)
3404             crReturnV;                 /* more data please */
3405         ret = ssh->do_ssh_init(ssh, *data);
3406         data++;
3407         datalen--;
3408         if (ret == 0)
3409             break;
3410     }
3411
3412     /*
3413      * We emerge from that loop when the initial negotiation is
3414      * over and we have selected an s_rdpkt function. Now pass
3415      * everything to s_rdpkt, and then pass the resulting packets
3416      * to the proper protocol handler.
3417      */
3418
3419     while (1) {
3420         while (bufchain_size(&ssh->queued_incoming_data) > 0 || datalen > 0) {
3421             if (ssh->frozen) {
3422                 ssh_queue_incoming_data(ssh, &data, &datalen);
3423                 /* This uses up all data and cannot cause anything interesting
3424                  * to happen; indeed, for anything to happen at all, we must
3425                  * return, so break out. */
3426                 break;
3427             } else if (bufchain_size(&ssh->queued_incoming_data) > 0) {
3428                 /* This uses up some or all data, and may freeze the
3429                  * session. */
3430                 ssh_process_queued_incoming_data(ssh);
3431             } else {
3432                 /* This uses up some or all data, and may freeze the
3433                  * session. */
3434                 ssh_process_incoming_data(ssh, &data, &datalen);
3435             }
3436             /* FIXME this is probably EBW. */
3437             if (ssh->state == SSH_STATE_CLOSED)
3438                 return;
3439         }
3440         /* We're out of data. Go and get some more. */
3441         crReturnV;
3442     }
3443     crFinishV;
3444 }
3445
3446 static int ssh_do_close(Ssh ssh, int notify_exit)
3447 {
3448     int ret = 0;
3449     struct ssh_channel *c;
3450
3451     ssh->state = SSH_STATE_CLOSED;
3452     expire_timer_context(ssh);
3453     if (ssh->s) {
3454         sk_close(ssh->s);
3455         ssh->s = NULL;
3456         if (notify_exit)
3457             notify_remote_exit(ssh->frontend);
3458         else
3459             ret = 1;
3460     }
3461     /*
3462      * Now we must shut down any port- and X-forwarded channels going
3463      * through this connection.
3464      */
3465     if (ssh->channels) {
3466         while (NULL != (c = index234(ssh->channels, 0))) {
3467             switch (c->type) {
3468               case CHAN_X11:
3469                 x11_close(c->u.x11.xconn);
3470                 break;
3471               case CHAN_SOCKDATA:
3472               case CHAN_SOCKDATA_DORMANT:
3473                 pfd_close(c->u.pfd.pf);
3474                 break;
3475             }
3476             del234(ssh->channels, c); /* moving next one to index 0 */
3477             if (ssh->version == 2)
3478                 bufchain_clear(&c->v.v2.outbuffer);
3479             sfree(c);
3480         }
3481     }
3482     /*
3483      * Go through port-forwardings, and close any associated
3484      * listening sockets.
3485      */
3486     if (ssh->portfwds) {
3487         struct ssh_portfwd *pf;
3488         while (NULL != (pf = index234(ssh->portfwds, 0))) {
3489             /* Dispose of any listening socket. */
3490             if (pf->local)
3491                 pfl_terminate(pf->local);
3492             del234(ssh->portfwds, pf); /* moving next one to index 0 */
3493             free_portfwd(pf);
3494         }
3495         freetree234(ssh->portfwds);
3496         ssh->portfwds = NULL;
3497     }
3498
3499     /*
3500      * Also stop attempting to connection-share.
3501      */
3502     if (ssh->connshare) {
3503         sharestate_free(ssh->connshare);
3504         ssh->connshare = NULL;
3505     }
3506
3507     return ret;
3508 }
3509
3510 static void ssh_socket_log(Plug plug, int type, SockAddr addr, int port,
3511                            const char *error_msg, int error_code)
3512 {
3513     Ssh ssh = (Ssh) plug;
3514
3515     /*
3516      * While we're attempting connection sharing, don't loudly log
3517      * everything that happens. Real TCP connections need to be logged
3518      * when we _start_ trying to connect, because it might be ages
3519      * before they respond if something goes wrong; but connection
3520      * sharing is local and quick to respond, and it's sufficient to
3521      * simply wait and see whether it worked afterwards.
3522      */
3523
3524     if (!ssh->attempting_connshare)
3525         backend_socket_log(ssh->frontend, type, addr, port,
3526                            error_msg, error_code, ssh->conf,
3527                            ssh->session_started);
3528 }
3529
3530 void ssh_connshare_log(Ssh ssh, int event, const char *logtext,
3531                        const char *ds_err, const char *us_err)
3532 {
3533     if (event == SHARE_NONE) {
3534         /* In this case, 'logtext' is an error message indicating a
3535          * reason why connection sharing couldn't be set up _at all_.
3536          * Failing that, ds_err and us_err indicate why we couldn't be
3537          * a downstream and an upstream respectively. */
3538         if (logtext) {
3539             logeventf(ssh, "Could not set up connection sharing: %s", logtext);
3540         } else {
3541             if (ds_err)
3542                 logeventf(ssh, "Could not set up connection sharing"
3543                           " as downstream: %s", ds_err);
3544             if (us_err)
3545                 logeventf(ssh, "Could not set up connection sharing"
3546                           " as upstream: %s", us_err);
3547         }
3548     } else if (event == SHARE_DOWNSTREAM) {
3549         /* In this case, 'logtext' is a local endpoint address */
3550         logeventf(ssh, "Using existing shared connection at %s", logtext);
3551         /* Also we should mention this in the console window to avoid
3552          * confusing users as to why this window doesn't behave the
3553          * usual way. */
3554         if ((flags & FLAG_VERBOSE) || (flags & FLAG_INTERACTIVE)) {
3555             c_write_str(ssh,"Reusing a shared connection to this server.\r\n");
3556         }
3557     } else if (event == SHARE_UPSTREAM) {
3558         /* In this case, 'logtext' is a local endpoint address too */
3559         logeventf(ssh, "Sharing this connection at %s", logtext);
3560     }
3561 }
3562
3563 static int ssh_closing(Plug plug, const char *error_msg, int error_code,
3564                        int calling_back)
3565 {
3566     Ssh ssh = (Ssh) plug;
3567     int need_notify = ssh_do_close(ssh, FALSE);
3568
3569     if (!error_msg) {
3570         if (!ssh->close_expected)
3571             error_msg = "Server unexpectedly closed network connection";
3572         else
3573             error_msg = "Server closed network connection";
3574     }
3575
3576     if (ssh->close_expected && ssh->clean_exit && ssh->exitcode < 0)
3577         ssh->exitcode = 0;
3578
3579     if (need_notify)
3580         notify_remote_exit(ssh->frontend);
3581
3582     if (error_msg)
3583         logevent(error_msg);
3584     if (!ssh->close_expected || !ssh->clean_exit)
3585         connection_fatal(ssh->frontend, "%s", error_msg);
3586     return 0;
3587 }
3588
3589 static int ssh_receive(Plug plug, int urgent, char *data, int len)
3590 {
3591     Ssh ssh = (Ssh) plug;
3592     ssh_gotdata(ssh, (unsigned char *)data, len);
3593     if (ssh->state == SSH_STATE_CLOSED) {
3594         ssh_do_close(ssh, TRUE);
3595         return 0;
3596     }
3597     return 1;
3598 }
3599
3600 static void ssh_sent(Plug plug, int bufsize)
3601 {
3602     Ssh ssh = (Ssh) plug;
3603     /*
3604      * If the send backlog on the SSH socket itself clears, we
3605      * should unthrottle the whole world if it was throttled.
3606      */
3607     if (bufsize < SSH_MAX_BACKLOG)
3608         ssh_throttle_all(ssh, 0, bufsize);
3609 }
3610
3611 static void ssh_hostport_setup(const char *host, int port, Conf *conf,
3612                                char **savedhost, int *savedport,
3613                                char **loghost_ret)
3614 {
3615     char *loghost = conf_get_str(conf, CONF_loghost);
3616     if (loghost_ret)
3617         *loghost_ret = loghost;
3618
3619     if (*loghost) {
3620         char *tmphost;
3621         char *colon;
3622
3623         tmphost = dupstr(loghost);
3624         *savedport = 22;               /* default ssh port */
3625
3626         /*
3627          * A colon suffix on the hostname string also lets us affect
3628          * savedport. (Unless there are multiple colons, in which case
3629          * we assume this is an unbracketed IPv6 literal.)
3630          */
3631         colon = host_strrchr(tmphost, ':');
3632         if (colon && colon == host_strchr(tmphost, ':')) {
3633             *colon++ = '\0';
3634             if (*colon)
3635                 *savedport = atoi(colon);
3636         }
3637
3638         *savedhost = host_strduptrim(tmphost);
3639         sfree(tmphost);
3640     } else {
3641         *savedhost = host_strduptrim(host);
3642         if (port < 0)
3643             port = 22;                 /* default ssh port */
3644         *savedport = port;
3645     }
3646 }
3647
3648 static int ssh_test_for_upstream(const char *host, int port, Conf *conf)
3649 {
3650     char *savedhost;
3651     int savedport;
3652     int ret;
3653
3654     random_ref(); /* platform may need this to determine share socket name */
3655     ssh_hostport_setup(host, port, conf, &savedhost, &savedport, NULL);
3656     ret = ssh_share_test_for_upstream(savedhost, savedport, conf);
3657     sfree(savedhost);
3658     random_unref();
3659
3660     return ret;
3661 }
3662
3663 /*
3664  * Connect to specified host and port.
3665  * Returns an error message, or NULL on success.
3666  * Also places the canonical host name into `realhost'. It must be
3667  * freed by the caller.
3668  */
3669 static const char *connect_to_host(Ssh ssh, const char *host, int port,
3670                                    char **realhost, int nodelay, int keepalive)
3671 {
3672     static const struct plug_function_table fn_table = {
3673         ssh_socket_log,
3674         ssh_closing,
3675         ssh_receive,
3676         ssh_sent,
3677         NULL
3678     };
3679
3680     SockAddr addr;
3681     const char *err;
3682     char *loghost;
3683     int addressfamily, sshprot;
3684
3685     ssh_hostport_setup(host, port, ssh->conf,
3686                        &ssh->savedhost, &ssh->savedport, &loghost);
3687
3688     ssh->fn = &fn_table;               /* make 'ssh' usable as a Plug */
3689
3690     /*
3691      * Try connection-sharing, in case that means we don't open a
3692      * socket after all. ssh_connection_sharing_init will connect to a
3693      * previously established upstream if it can, and failing that,
3694      * establish a listening socket for _us_ to be the upstream. In
3695      * the latter case it will return NULL just as if it had done
3696      * nothing, because here we only need to care if we're a
3697      * downstream and need to do our connection setup differently.
3698      */
3699     ssh->connshare = NULL;
3700     ssh->attempting_connshare = TRUE;  /* affects socket logging behaviour */
3701     ssh->s = ssh_connection_sharing_init(ssh->savedhost, ssh->savedport,
3702                                          ssh->conf, ssh, &ssh->connshare);
3703     ssh->attempting_connshare = FALSE;
3704     if (ssh->s != NULL) {
3705         /*
3706          * We are a downstream.
3707          */
3708         ssh->bare_connection = TRUE;
3709         ssh->do_ssh_init = do_ssh_connection_init;
3710         ssh->fullhostname = NULL;
3711         *realhost = dupstr(host);      /* best we can do */
3712     } else {
3713         /*
3714          * We're not a downstream, so open a normal socket.
3715          */
3716         ssh->do_ssh_init = do_ssh_init;
3717
3718         /*
3719          * Try to find host.
3720          */
3721         addressfamily = conf_get_int(ssh->conf, CONF_addressfamily);
3722         addr = name_lookup(host, port, realhost, ssh->conf, addressfamily,
3723                            ssh->frontend, "SSH connection");
3724         if ((err = sk_addr_error(addr)) != NULL) {
3725             sk_addr_free(addr);
3726             return err;
3727         }
3728         ssh->fullhostname = dupstr(*realhost);   /* save in case of GSSAPI */
3729
3730         ssh->s = new_connection(addr, *realhost, port,
3731                                 0, 1, nodelay, keepalive,
3732                                 (Plug) ssh, ssh->conf);
3733         if ((err = sk_socket_error(ssh->s)) != NULL) {
3734             ssh->s = NULL;
3735             notify_remote_exit(ssh->frontend);
3736             return err;
3737         }
3738     }
3739
3740     /*
3741      * The SSH version number is always fixed (since we no longer support
3742      * fallback between versions), so set it now, and if it's SSH-2,
3743      * send the version string now too.
3744      */
3745     sshprot = conf_get_int(ssh->conf, CONF_sshprot);
3746     assert(sshprot == 0 || sshprot == 3);
3747     if (sshprot == 0)
3748         /* SSH-1 only */
3749         ssh->version = 1;
3750     if (sshprot == 3 && !ssh->bare_connection) {
3751         /* SSH-2 only */
3752         ssh->version = 2;
3753         ssh_send_verstring(ssh, "SSH-", NULL);
3754     }
3755
3756     /*
3757      * loghost, if configured, overrides realhost.
3758      */
3759     if (*loghost) {
3760         sfree(*realhost);
3761         *realhost = dupstr(loghost);
3762     }
3763
3764     return NULL;
3765 }
3766
3767 /*
3768  * Throttle or unthrottle the SSH connection.
3769  */
3770 static void ssh_throttle_conn(Ssh ssh, int adjust)
3771 {
3772     int old_count = ssh->conn_throttle_count;
3773     ssh->conn_throttle_count += adjust;
3774     assert(ssh->conn_throttle_count >= 0);
3775     if (ssh->conn_throttle_count && !old_count) {
3776         ssh_set_frozen(ssh, 1);
3777     } else if (!ssh->conn_throttle_count && old_count) {
3778         ssh_set_frozen(ssh, 0);
3779     }
3780 }
3781
3782 /*
3783  * Throttle or unthrottle _all_ local data streams (for when sends
3784  * on the SSH connection itself back up).
3785  */
3786 static void ssh_throttle_all(Ssh ssh, int enable, int bufsize)
3787 {
3788     int i;
3789     struct ssh_channel *c;
3790
3791     if (enable == ssh->throttled_all)
3792         return;
3793     ssh->throttled_all = enable;
3794     ssh->overall_bufsize = bufsize;
3795     if (!ssh->channels)
3796         return;
3797     for (i = 0; NULL != (c = index234(ssh->channels, i)); i++) {
3798         switch (c->type) {
3799           case CHAN_MAINSESSION:
3800             /*
3801              * This is treated separately, outside the switch.
3802              */
3803             break;
3804           case CHAN_X11:
3805             x11_override_throttle(c->u.x11.xconn, enable);
3806             break;
3807           case CHAN_AGENT:
3808             /* Agent channels require no buffer management. */
3809             break;
3810           case CHAN_SOCKDATA:
3811             pfd_override_throttle(c->u.pfd.pf, enable);
3812             break;
3813         }
3814     }
3815 }
3816
3817 static void ssh_agent_callback(void *sshv, void *reply, int replylen)
3818 {
3819     Ssh ssh = (Ssh) sshv;
3820
3821     ssh->agent_response = reply;
3822     ssh->agent_response_len = replylen;
3823
3824     if (ssh->version == 1)
3825         do_ssh1_login(ssh, NULL, -1, NULL);
3826     else
3827         do_ssh2_authconn(ssh, NULL, -1, NULL);
3828 }
3829
3830 static void ssh_dialog_callback(void *sshv, int ret)
3831 {
3832     Ssh ssh = (Ssh) sshv;
3833
3834     ssh->user_response = ret;
3835
3836     if (ssh->version == 1)
3837         do_ssh1_login(ssh, NULL, -1, NULL);
3838     else
3839         do_ssh2_transport(ssh, NULL, -1, NULL);
3840
3841     /*
3842      * This may have unfrozen the SSH connection, so do a
3843      * queued-data run.
3844      */
3845     ssh_process_queued_incoming_data(ssh);
3846 }
3847
3848 static void ssh_agentf_callback(void *cv, void *reply, int replylen)
3849 {
3850     struct ssh_channel *c = (struct ssh_channel *)cv;
3851     const void *sentreply = reply;
3852
3853     c->u.a.outstanding_requests--;
3854     if (!sentreply) {
3855         /* Fake SSH_AGENT_FAILURE. */
3856         sentreply = "\0\0\0\1\5";
3857         replylen = 5;
3858     }
3859     ssh_send_channel_data(c, sentreply, replylen);
3860     if (reply)
3861         sfree(reply);
3862     /*
3863      * If we've already seen an incoming EOF but haven't sent an
3864      * outgoing one, this may be the moment to send it.
3865      */
3866     if (c->u.a.outstanding_requests == 0 && (c->closes & CLOSES_RCVD_EOF))
3867         sshfwd_write_eof(c);
3868 }
3869
3870 /*
3871  * Client-initiated disconnection. Send a DISCONNECT if `wire_reason'
3872  * non-NULL, otherwise just close the connection. `client_reason' == NULL
3873  * => log `wire_reason'.
3874  */
3875 static void ssh_disconnect(Ssh ssh, const char *client_reason,
3876                            const char *wire_reason,
3877                            int code, int clean_exit)
3878 {
3879     char *error;
3880     if (!client_reason)
3881         client_reason = wire_reason;
3882     if (client_reason)
3883         error = dupprintf("Disconnected: %s", client_reason);
3884     else
3885         error = dupstr("Disconnected");
3886     if (wire_reason) {
3887         if (ssh->version == 1) {
3888             send_packet(ssh, SSH1_MSG_DISCONNECT, PKT_STR, wire_reason,
3889                         PKT_END);
3890         } else if (ssh->version == 2) {
3891             struct Packet *pktout = ssh2_pkt_init(SSH2_MSG_DISCONNECT);
3892             ssh2_pkt_adduint32(pktout, code);
3893             ssh2_pkt_addstring(pktout, wire_reason);
3894             ssh2_pkt_addstring(pktout, "en");   /* language tag */
3895             ssh2_pkt_send_noqueue(ssh, pktout);
3896         }
3897     }
3898     ssh->close_expected = TRUE;
3899     ssh->clean_exit = clean_exit;
3900     ssh_closing((Plug)ssh, error, 0, 0);
3901     sfree(error);
3902 }
3903
3904 int verify_ssh_manual_host_key(Ssh ssh, const char *fingerprint,
3905                                const struct ssh_signkey *ssh2keytype,
3906                                void *ssh2keydata)
3907 {
3908     if (!conf_get_str_nthstrkey(ssh->conf, CONF_ssh_manual_hostkeys, 0)) {
3909         return -1;                     /* no manual keys configured */
3910     }
3911
3912     if (fingerprint) {
3913         /*
3914          * The fingerprint string we've been given will have things
3915          * like 'ssh-rsa 2048' at the front of it. Strip those off and
3916          * narrow down to just the colon-separated hex block at the
3917          * end of the string.
3918          */
3919         const char *p = strrchr(fingerprint, ' ');
3920         fingerprint = p ? p+1 : fingerprint;
3921         /* Quick sanity checks, including making sure it's in lowercase */
3922         assert(strlen(fingerprint) == 16*3 - 1);
3923         assert(fingerprint[2] == ':');
3924         assert(fingerprint[strspn(fingerprint, "0123456789abcdef:")] == 0);
3925
3926         if (conf_get_str_str_opt(ssh->conf, CONF_ssh_manual_hostkeys,
3927                                  fingerprint))
3928             return 1;                  /* success */
3929     }
3930
3931     if (ssh2keydata) {
3932         /*
3933          * Construct the base64-encoded public key blob and see if
3934          * that's listed.
3935          */
3936         unsigned char *binblob;
3937         char *base64blob;
3938         int binlen, atoms, i;
3939         binblob = ssh2keytype->public_blob(ssh2keydata, &binlen);
3940         atoms = (binlen + 2) / 3;
3941         base64blob = snewn(atoms * 4 + 1, char);
3942         for (i = 0; i < atoms; i++)
3943             base64_encode_atom(binblob + 3*i, binlen - 3*i, base64blob + 4*i);
3944         base64blob[atoms * 4] = '\0';
3945         sfree(binblob);
3946         if (conf_get_str_str_opt(ssh->conf, CONF_ssh_manual_hostkeys,
3947                                  base64blob)) {
3948             sfree(base64blob);
3949             return 1;                  /* success */
3950         }
3951         sfree(base64blob);
3952     }
3953
3954     return 0;
3955 }
3956
3957 /*
3958  * Handle the key exchange and user authentication phases.
3959  */
3960 static int do_ssh1_login(Ssh ssh, const unsigned char *in, int inlen,
3961                          struct Packet *pktin)
3962 {
3963     int i, j, ret;
3964     unsigned char cookie[8], *ptr;
3965     struct MD5Context md5c;
3966     struct do_ssh1_login_state {
3967         int crLine;
3968         int len;
3969         unsigned char *rsabuf;
3970         const unsigned char *keystr1, *keystr2;
3971         unsigned long supported_ciphers_mask, supported_auths_mask;
3972         int tried_publickey, tried_agent;
3973         int tis_auth_refused, ccard_auth_refused;
3974         unsigned char session_id[16];
3975         int cipher_type;
3976         void *publickey_blob;
3977         int publickey_bloblen;
3978         char *publickey_comment;
3979         int privatekey_available, privatekey_encrypted;
3980         prompts_t *cur_prompt;
3981         char c;
3982         int pwpkt_type;
3983         unsigned char request[5], *response, *p;
3984         int responselen;
3985         int keyi, nkeys;
3986         int authed;
3987         struct RSAKey key;
3988         Bignum challenge;
3989         char *commentp;
3990         int commentlen;
3991         int dlgret;
3992         Filename *keyfile;
3993         struct RSAKey servkey, hostkey;
3994     };
3995     crState(do_ssh1_login_state);
3996
3997     crBeginState;
3998
3999     if (!pktin)
4000         crWaitUntil(pktin);
4001
4002     if (pktin->type != SSH1_SMSG_PUBLIC_KEY) {
4003         bombout(("Public key packet not received"));
4004         crStop(0);
4005     }
4006
4007     logevent("Received public keys");
4008
4009     ptr = ssh_pkt_getdata(pktin, 8);
4010     if (!ptr) {
4011         bombout(("SSH-1 public key packet stopped before random cookie"));
4012         crStop(0);
4013     }
4014     memcpy(cookie, ptr, 8);
4015
4016     if (!ssh1_pkt_getrsakey(pktin, &s->servkey, &s->keystr1) ||
4017         !ssh1_pkt_getrsakey(pktin, &s->hostkey, &s->keystr2)) { 
4018         bombout(("Failed to read SSH-1 public keys from public key packet"));
4019         crStop(0);
4020     }
4021
4022     /*
4023      * Log the host key fingerprint.
4024      */
4025     {
4026         char logmsg[80];
4027         logevent("Host key fingerprint is:");
4028         strcpy(logmsg, "      ");
4029         s->hostkey.comment = NULL;
4030         rsa_fingerprint(logmsg + strlen(logmsg),
4031                         sizeof(logmsg) - strlen(logmsg), &s->hostkey);
4032         logevent(logmsg);
4033     }
4034
4035     ssh->v1_remote_protoflags = ssh_pkt_getuint32(pktin);
4036     s->supported_ciphers_mask = ssh_pkt_getuint32(pktin);
4037     s->supported_auths_mask = ssh_pkt_getuint32(pktin);
4038     if ((ssh->remote_bugs & BUG_CHOKES_ON_RSA))
4039         s->supported_auths_mask &= ~(1 << SSH1_AUTH_RSA);
4040
4041     ssh->v1_local_protoflags =
4042         ssh->v1_remote_protoflags & SSH1_PROTOFLAGS_SUPPORTED;
4043     ssh->v1_local_protoflags |= SSH1_PROTOFLAG_SCREEN_NUMBER;
4044
4045     MD5Init(&md5c);
4046     MD5Update(&md5c, s->keystr2, s->hostkey.bytes);
4047     MD5Update(&md5c, s->keystr1, s->servkey.bytes);
4048     MD5Update(&md5c, cookie, 8);
4049     MD5Final(s->session_id, &md5c);
4050
4051     for (i = 0; i < 32; i++)
4052         ssh->session_key[i] = random_byte();
4053
4054     /*
4055      * Verify that the `bits' and `bytes' parameters match.
4056      */
4057     if (s->hostkey.bits > s->hostkey.bytes * 8 ||
4058         s->servkey.bits > s->servkey.bytes * 8) {
4059         bombout(("SSH-1 public keys were badly formatted"));
4060         crStop(0);
4061     }
4062
4063     s->len = (s->hostkey.bytes > s->servkey.bytes ?
4064               s->hostkey.bytes : s->servkey.bytes);
4065
4066     s->rsabuf = snewn(s->len, unsigned char);
4067
4068     /*
4069      * Verify the host key.
4070      */
4071     {
4072         /*
4073          * First format the key into a string.
4074          */
4075         int len = rsastr_len(&s->hostkey);
4076         char fingerprint[100];
4077         char *keystr = snewn(len, char);
4078         rsastr_fmt(keystr, &s->hostkey);
4079         rsa_fingerprint(fingerprint, sizeof(fingerprint), &s->hostkey);
4080
4081         /* First check against manually configured host keys. */
4082         s->dlgret = verify_ssh_manual_host_key(ssh, fingerprint, NULL, NULL);
4083         if (s->dlgret == 0) {          /* did not match */
4084             bombout(("Host key did not appear in manually configured list"));
4085             sfree(keystr);
4086             crStop(0);
4087         } else if (s->dlgret < 0) { /* none configured; use standard handling */
4088             ssh_set_frozen(ssh, 1);
4089             s->dlgret = verify_ssh_host_key(ssh->frontend,
4090                                             ssh->savedhost, ssh->savedport,
4091                                             "rsa", keystr, fingerprint,
4092                                             ssh_dialog_callback, ssh);
4093             sfree(keystr);
4094 #ifdef FUZZING
4095             s->dlgret = 1;
4096 #endif
4097             if (s->dlgret < 0) {
4098                 do {
4099                     crReturn(0);
4100                     if (pktin) {
4101                         bombout(("Unexpected data from server while waiting"
4102                                  " for user host key response"));
4103                         crStop(0);
4104                     }
4105                 } while (pktin || inlen > 0);
4106                 s->dlgret = ssh->user_response;
4107             }
4108             ssh_set_frozen(ssh, 0);
4109
4110             if (s->dlgret == 0) {
4111                 ssh_disconnect(ssh, "User aborted at host key verification",
4112                                NULL, 0, TRUE);
4113                 crStop(0);
4114             }
4115         } else {
4116             sfree(keystr);
4117         }
4118     }
4119
4120     for (i = 0; i < 32; i++) {
4121         s->rsabuf[i] = ssh->session_key[i];
4122         if (i < 16)
4123             s->rsabuf[i] ^= s->session_id[i];
4124     }
4125
4126     if (s->hostkey.bytes > s->servkey.bytes) {
4127         ret = rsaencrypt(s->rsabuf, 32, &s->servkey);
4128         if (ret)
4129             ret = rsaencrypt(s->rsabuf, s->servkey.bytes, &s->hostkey);
4130     } else {
4131         ret = rsaencrypt(s->rsabuf, 32, &s->hostkey);
4132         if (ret)
4133             ret = rsaencrypt(s->rsabuf, s->hostkey.bytes, &s->servkey);
4134     }
4135     if (!ret) {
4136         bombout(("SSH-1 public key encryptions failed due to bad formatting"));
4137         crStop(0);      
4138     }
4139
4140     logevent("Encrypted session key");
4141
4142     {
4143         int cipher_chosen = 0, warn = 0;
4144         const char *cipher_string = NULL;
4145         int i;
4146         for (i = 0; !cipher_chosen && i < CIPHER_MAX; i++) {
4147             int next_cipher = conf_get_int_int(ssh->conf,
4148                                                CONF_ssh_cipherlist, i);
4149             if (next_cipher == CIPHER_WARN) {
4150                 /* If/when we choose a cipher, warn about it */
4151                 warn = 1;
4152             } else if (next_cipher == CIPHER_AES) {
4153                 /* XXX Probably don't need to mention this. */
4154                 logevent("AES not supported in SSH-1, skipping");
4155             } else {
4156                 switch (next_cipher) {
4157                   case CIPHER_3DES:     s->cipher_type = SSH_CIPHER_3DES;
4158                                         cipher_string = "3DES"; break;
4159                   case CIPHER_BLOWFISH: s->cipher_type = SSH_CIPHER_BLOWFISH;
4160                                         cipher_string = "Blowfish"; break;
4161                   case CIPHER_DES:      s->cipher_type = SSH_CIPHER_DES;
4162                                         cipher_string = "single-DES"; break;
4163                 }
4164                 if (s->supported_ciphers_mask & (1 << s->cipher_type))
4165                     cipher_chosen = 1;
4166             }
4167         }
4168         if (!cipher_chosen) {
4169             if ((s->supported_ciphers_mask & (1 << SSH_CIPHER_3DES)) == 0)
4170                 bombout(("Server violates SSH-1 protocol by not "
4171                          "supporting 3DES encryption"));
4172             else
4173                 /* shouldn't happen */
4174                 bombout(("No supported ciphers found"));
4175             crStop(0);
4176         }
4177
4178         /* Warn about chosen cipher if necessary. */
4179         if (warn) {
4180             ssh_set_frozen(ssh, 1);
4181             s->dlgret = askalg(ssh->frontend, "cipher", cipher_string,
4182                                ssh_dialog_callback, ssh);
4183             if (s->dlgret < 0) {
4184                 do {
4185                     crReturn(0);
4186                     if (pktin) {
4187                         bombout(("Unexpected data from server while waiting"
4188                                  " for user response"));
4189                         crStop(0);
4190                     }
4191                 } while (pktin || inlen > 0);
4192                 s->dlgret = ssh->user_response;
4193             }
4194             ssh_set_frozen(ssh, 0);
4195             if (s->dlgret == 0) {
4196                 ssh_disconnect(ssh, "User aborted at cipher warning", NULL,
4197                                0, TRUE);
4198                 crStop(0);
4199             }
4200         }
4201     }
4202
4203     switch (s->cipher_type) {
4204       case SSH_CIPHER_3DES:
4205         logevent("Using 3DES encryption");
4206         break;
4207       case SSH_CIPHER_DES:
4208         logevent("Using single-DES encryption");
4209         break;
4210       case SSH_CIPHER_BLOWFISH:
4211         logevent("Using Blowfish encryption");
4212         break;
4213     }
4214
4215     send_packet(ssh, SSH1_CMSG_SESSION_KEY,
4216                 PKT_CHAR, s->cipher_type,
4217                 PKT_DATA, cookie, 8,
4218                 PKT_CHAR, (s->len * 8) >> 8, PKT_CHAR, (s->len * 8) & 0xFF,
4219                 PKT_DATA, s->rsabuf, s->len,
4220                 PKT_INT, ssh->v1_local_protoflags, PKT_END);
4221
4222     logevent("Trying to enable encryption...");
4223
4224     sfree(s->rsabuf);
4225
4226     ssh->cipher = (s->cipher_type == SSH_CIPHER_BLOWFISH ? &ssh_blowfish_ssh1 :
4227                    s->cipher_type == SSH_CIPHER_DES ? &ssh_des :
4228                    &ssh_3des);
4229     ssh->v1_cipher_ctx = ssh->cipher->make_context();
4230     ssh->cipher->sesskey(ssh->v1_cipher_ctx, ssh->session_key);
4231     logeventf(ssh, "Initialised %s encryption", ssh->cipher->text_name);
4232
4233     ssh->crcda_ctx = crcda_make_context();
4234     logevent("Installing CRC compensation attack detector");
4235
4236     if (s->servkey.modulus) {
4237         sfree(s->servkey.modulus);
4238         s->servkey.modulus = NULL;
4239     }
4240     if (s->servkey.exponent) {
4241         sfree(s->servkey.exponent);
4242         s->servkey.exponent = NULL;
4243     }
4244     if (s->hostkey.modulus) {
4245         sfree(s->hostkey.modulus);
4246         s->hostkey.modulus = NULL;
4247     }
4248     if (s->hostkey.exponent) {
4249         sfree(s->hostkey.exponent);
4250         s->hostkey.exponent = NULL;
4251     }
4252     crWaitUntil(pktin);
4253
4254     if (pktin->type != SSH1_SMSG_SUCCESS) {
4255         bombout(("Encryption not successfully enabled"));
4256         crStop(0);
4257     }
4258
4259     logevent("Successfully started encryption");
4260
4261     fflush(stdout); /* FIXME eh? */
4262     {
4263         if ((ssh->username = get_remote_username(ssh->conf)) == NULL) {
4264             int ret; /* need not be kept over crReturn */
4265             s->cur_prompt = new_prompts(ssh->frontend);
4266             s->cur_prompt->to_server = TRUE;
4267             s->cur_prompt->name = dupstr("SSH login name");
4268             add_prompt(s->cur_prompt, dupstr("login as: "), TRUE);
4269             ret = get_userpass_input(s->cur_prompt, NULL, 0);
4270             while (ret < 0) {
4271                 ssh->send_ok = 1;
4272                 crWaitUntil(!pktin);
4273                 ret = get_userpass_input(s->cur_prompt, in, inlen);
4274                 ssh->send_ok = 0;
4275             }
4276             if (!ret) {
4277                 /*
4278                  * Failed to get a username. Terminate.
4279                  */
4280                 free_prompts(s->cur_prompt);
4281                 ssh_disconnect(ssh, "No username provided", NULL, 0, TRUE);
4282                 crStop(0);
4283             }
4284             ssh->username = dupstr(s->cur_prompt->prompts[0]->result);
4285             free_prompts(s->cur_prompt);
4286         }
4287
4288         send_packet(ssh, SSH1_CMSG_USER, PKT_STR, ssh->username, PKT_END);
4289         {
4290             char *userlog = dupprintf("Sent username \"%s\"", ssh->username);
4291             logevent(userlog);
4292             if (flags & FLAG_INTERACTIVE &&
4293                 (!((flags & FLAG_STDERR) && (flags & FLAG_VERBOSE)))) {
4294                 c_write_str(ssh, userlog);
4295                 c_write_str(ssh, "\r\n");
4296             }
4297             sfree(userlog);
4298         }
4299     }
4300
4301     crWaitUntil(pktin);
4302
4303     if ((s->supported_auths_mask & (1 << SSH1_AUTH_RSA)) == 0) {
4304         /* We must not attempt PK auth. Pretend we've already tried it. */
4305         s->tried_publickey = s->tried_agent = 1;
4306     } else {
4307         s->tried_publickey = s->tried_agent = 0;
4308     }
4309     s->tis_auth_refused = s->ccard_auth_refused = 0;
4310     /*
4311      * Load the public half of any configured keyfile for later use.
4312      */
4313     s->keyfile = conf_get_filename(ssh->conf, CONF_keyfile);
4314     if (!filename_is_null(s->keyfile)) {
4315         int keytype;
4316         logeventf(ssh, "Reading key file \"%.150s\"",
4317                   filename_to_str(s->keyfile));
4318         keytype = key_type(s->keyfile);
4319         if (keytype == SSH_KEYTYPE_SSH1 ||
4320             keytype == SSH_KEYTYPE_SSH1_PUBLIC) {
4321             const char *error;
4322             if (rsakey_pubblob(s->keyfile,
4323                                &s->publickey_blob, &s->publickey_bloblen,
4324                                &s->publickey_comment, &error)) {
4325                 s->privatekey_available = (keytype == SSH_KEYTYPE_SSH1);
4326                 if (!s->privatekey_available)
4327                     logeventf(ssh, "Key file contains public key only");
4328                 s->privatekey_encrypted = rsakey_encrypted(s->keyfile,
4329                                                            NULL);
4330             } else {
4331                 char *msgbuf;
4332                 logeventf(ssh, "Unable to load key (%s)", error);
4333                 msgbuf = dupprintf("Unable to load key file "
4334                                    "\"%.150s\" (%s)\r\n",
4335                                    filename_to_str(s->keyfile),
4336                                    error);
4337                 c_write_str(ssh, msgbuf);
4338                 sfree(msgbuf);
4339                 s->publickey_blob = NULL;
4340             }
4341         } else {
4342             char *msgbuf;
4343             logeventf(ssh, "Unable to use this key file (%s)",
4344                       key_type_to_str(keytype));
4345             msgbuf = dupprintf("Unable to use key file \"%.150s\""
4346                                " (%s)\r\n",
4347                                filename_to_str(s->keyfile),
4348                                key_type_to_str(keytype));
4349             c_write_str(ssh, msgbuf);
4350             sfree(msgbuf);
4351             s->publickey_blob = NULL;
4352         }
4353     } else
4354         s->publickey_blob = NULL;
4355
4356     while (pktin->type == SSH1_SMSG_FAILURE) {
4357         s->pwpkt_type = SSH1_CMSG_AUTH_PASSWORD;
4358
4359         if (conf_get_int(ssh->conf, CONF_tryagent) && agent_exists() && !s->tried_agent) {
4360             /*
4361              * Attempt RSA authentication using Pageant.
4362              */
4363             void *r;
4364
4365             s->authed = FALSE;
4366             s->tried_agent = 1;
4367             logevent("Pageant is running. Requesting keys.");
4368
4369             /* Request the keys held by the agent. */
4370             PUT_32BIT(s->request, 1);
4371             s->request[4] = SSH1_AGENTC_REQUEST_RSA_IDENTITIES;
4372             if (!agent_query(s->request, 5, &r, &s->responselen,
4373                              ssh_agent_callback, ssh)) {
4374                 do {
4375                     crReturn(0);
4376                     if (pktin) {
4377                         bombout(("Unexpected data from server while waiting"
4378                                  " for agent response"));
4379                         crStop(0);
4380                     }
4381                 } while (pktin || inlen > 0);
4382                 r = ssh->agent_response;
4383                 s->responselen = ssh->agent_response_len;
4384             }
4385             s->response = (unsigned char *) r;
4386             if (s->response && s->responselen >= 5 &&
4387                 s->response[4] == SSH1_AGENT_RSA_IDENTITIES_ANSWER) {
4388                 s->p = s->response + 5;
4389                 s->nkeys = toint(GET_32BIT(s->p));
4390                 if (s->nkeys < 0) {
4391                     logeventf(ssh, "Pageant reported negative key count %d",
4392                               s->nkeys);
4393                     s->nkeys = 0;
4394                 }
4395                 s->p += 4;
4396                 logeventf(ssh, "Pageant has %d SSH-1 keys", s->nkeys);
4397                 for (s->keyi = 0; s->keyi < s->nkeys; s->keyi++) {
4398                     unsigned char *pkblob = s->p;
4399                     s->p += 4;
4400                     {
4401                         int n, ok = FALSE;
4402                         do {           /* do while (0) to make breaking easy */
4403                             n = ssh1_read_bignum
4404                                 (s->p, toint(s->responselen-(s->p-s->response)),
4405                                  &s->key.exponent);
4406                             if (n < 0)
4407                                 break;
4408                             s->p += n;
4409                             n = ssh1_read_bignum
4410                                 (s->p, toint(s->responselen-(s->p-s->response)),
4411                                  &s->key.modulus);
4412                             if (n < 0)
4413                                 break;
4414                             s->p += n;
4415                             if (s->responselen - (s->p-s->response) < 4)
4416                                 break;
4417                             s->commentlen = toint(GET_32BIT(s->p));
4418                             s->p += 4;
4419                             if (s->commentlen < 0 ||
4420                                 toint(s->responselen - (s->p-s->response)) <
4421                                 s->commentlen)
4422                                 break;
4423                             s->commentp = (char *)s->p;
4424                             s->p += s->commentlen;
4425                             ok = TRUE;
4426                         } while (0);
4427                         if (!ok) {
4428                             logevent("Pageant key list packet was truncated");
4429                             break;
4430                         }
4431                     }
4432                     if (s->publickey_blob) {
4433                         if (!memcmp(pkblob, s->publickey_blob,
4434                                     s->publickey_bloblen)) {
4435                             logeventf(ssh, "Pageant key #%d matches "
4436                                       "configured key file", s->keyi);
4437                             s->tried_publickey = 1;
4438                         } else
4439                             /* Skip non-configured key */
4440                             continue;
4441                     }
4442                     logeventf(ssh, "Trying Pageant key #%d", s->keyi);
4443                     send_packet(ssh, SSH1_CMSG_AUTH_RSA,
4444                                 PKT_BIGNUM, s->key.modulus, PKT_END);
4445                     crWaitUntil(pktin);
4446                     if (pktin->type != SSH1_SMSG_AUTH_RSA_CHALLENGE) {
4447                         logevent("Key refused");
4448                         continue;
4449                     }
4450                     logevent("Received RSA challenge");
4451                     if ((s->challenge = ssh1_pkt_getmp(pktin)) == NULL) {
4452                         bombout(("Server's RSA challenge was badly formatted"));
4453                         crStop(0);
4454                     }
4455
4456                     {
4457                         char *agentreq, *q, *ret;
4458                         void *vret;
4459                         int len, retlen;
4460                         len = 1 + 4;   /* message type, bit count */
4461                         len += ssh1_bignum_length(s->key.exponent);
4462                         len += ssh1_bignum_length(s->key.modulus);
4463                         len += ssh1_bignum_length(s->challenge);
4464                         len += 16;     /* session id */
4465                         len += 4;      /* response format */
4466                         agentreq = snewn(4 + len, char);
4467                         PUT_32BIT(agentreq, len);
4468                         q = agentreq + 4;
4469                         *q++ = SSH1_AGENTC_RSA_CHALLENGE;
4470                         PUT_32BIT(q, bignum_bitcount(s->key.modulus));
4471                         q += 4;
4472                         q += ssh1_write_bignum(q, s->key.exponent);
4473                         q += ssh1_write_bignum(q, s->key.modulus);
4474                         q += ssh1_write_bignum(q, s->challenge);
4475                         memcpy(q, s->session_id, 16);
4476                         q += 16;
4477                         PUT_32BIT(q, 1);        /* response format */
4478                         if (!agent_query(agentreq, len + 4, &vret, &retlen,
4479                                          ssh_agent_callback, ssh)) {
4480                             sfree(agentreq);
4481                             do {
4482                                 crReturn(0);
4483                                 if (pktin) {
4484                                     bombout(("Unexpected data from server"
4485                                              " while waiting for agent"
4486                                              " response"));
4487                                     crStop(0);
4488                                 }
4489                             } while (pktin || inlen > 0);
4490                             vret = ssh->agent_response;
4491                             retlen = ssh->agent_response_len;
4492                         } else
4493                             sfree(agentreq);
4494                         ret = vret;
4495                         if (ret) {
4496                             if (ret[4] == SSH1_AGENT_RSA_RESPONSE) {
4497                                 logevent("Sending Pageant's response");
4498                                 send_packet(ssh, SSH1_CMSG_AUTH_RSA_RESPONSE,
4499                                             PKT_DATA, ret + 5, 16,
4500                                             PKT_END);
4501                                 sfree(ret);
4502                                 crWaitUntil(pktin);
4503                                 if (pktin->type == SSH1_SMSG_SUCCESS) {
4504                                     logevent
4505                                         ("Pageant's response accepted");
4506                                     if (flags & FLAG_VERBOSE) {
4507                                         c_write_str(ssh, "Authenticated using"
4508                                                     " RSA key \"");
4509                                         c_write(ssh, s->commentp,
4510                                                 s->commentlen);
4511                                         c_write_str(ssh, "\" from agent\r\n");
4512                                     }
4513                                     s->authed = TRUE;
4514                                 } else
4515                                     logevent
4516                                         ("Pageant's response not accepted");
4517                             } else {
4518                                 logevent
4519                                     ("Pageant failed to answer challenge");
4520                                 sfree(ret);
4521                             }
4522                         } else {
4523                             logevent("No reply received from Pageant");
4524                         }
4525                     }
4526                     freebn(s->key.exponent);
4527                     freebn(s->key.modulus);
4528                     freebn(s->challenge);
4529                     if (s->authed)
4530                         break;
4531                 }
4532                 sfree(s->response);
4533                 if (s->publickey_blob && !s->tried_publickey)
4534                     logevent("Configured key file not in Pageant");
4535             } else {
4536                 logevent("Failed to get reply from Pageant");
4537             }
4538             if (s->authed)
4539                 break;
4540         }
4541         if (s->publickey_blob && s->privatekey_available &&
4542             !s->tried_publickey) {
4543             /*
4544              * Try public key authentication with the specified
4545              * key file.
4546              */
4547             int got_passphrase; /* need not be kept over crReturn */
4548             if (flags & FLAG_VERBOSE)
4549                 c_write_str(ssh, "Trying public key authentication.\r\n");
4550             s->keyfile = conf_get_filename(ssh->conf, CONF_keyfile);
4551             logeventf(ssh, "Trying public key \"%s\"",
4552                       filename_to_str(s->keyfile));
4553             s->tried_publickey = 1;
4554             got_passphrase = FALSE;
4555             while (!got_passphrase) {
4556                 /*
4557                  * Get a passphrase, if necessary.
4558                  */
4559                 char *passphrase = NULL;    /* only written after crReturn */
4560                 const char *error;
4561                 if (!s->privatekey_encrypted) {
4562                     if (flags & FLAG_VERBOSE)
4563                         c_write_str(ssh, "No passphrase required.\r\n");
4564                     passphrase = NULL;
4565                 } else {
4566                     int ret; /* need not be kept over crReturn */
4567                     s->cur_prompt = new_prompts(ssh->frontend);
4568                     s->cur_prompt->to_server = FALSE;
4569                     s->cur_prompt->name = dupstr("SSH key passphrase");
4570                     add_prompt(s->cur_prompt,
4571                                dupprintf("Passphrase for key \"%.100s\": ",
4572                                          s->publickey_comment), FALSE);
4573                     ret = get_userpass_input(s->cur_prompt, NULL, 0);
4574                     while (ret < 0) {
4575                         ssh->send_ok = 1;
4576                         crWaitUntil(!pktin);
4577                         ret = get_userpass_input(s->cur_prompt, in, inlen);
4578                         ssh->send_ok = 0;
4579                     }
4580                     if (!ret) {
4581                         /* Failed to get a passphrase. Terminate. */
4582                         free_prompts(s->cur_prompt);
4583                         ssh_disconnect(ssh, NULL, "Unable to authenticate",
4584                                        0, TRUE);
4585                         crStop(0);
4586                     }
4587                     passphrase = dupstr(s->cur_prompt->prompts[0]->result);
4588                     free_prompts(s->cur_prompt);
4589                 }
4590                 /*
4591                  * Try decrypting key with passphrase.
4592                  */
4593                 s->keyfile = conf_get_filename(ssh->conf, CONF_keyfile);
4594                 ret = loadrsakey(s->keyfile, &s->key, passphrase,
4595                                  &error);
4596                 if (passphrase) {
4597                     smemclr(passphrase, strlen(passphrase));
4598                     sfree(passphrase);
4599                 }
4600                 if (ret == 1) {
4601                     /* Correct passphrase. */
4602                     got_passphrase = TRUE;
4603                 } else if (ret == 0) {
4604                     c_write_str(ssh, "Couldn't load private key from ");
4605                     c_write_str(ssh, filename_to_str(s->keyfile));
4606                     c_write_str(ssh, " (");
4607                     c_write_str(ssh, error);
4608                     c_write_str(ssh, ").\r\n");
4609                     got_passphrase = FALSE;
4610                     break;             /* go and try something else */
4611                 } else if (ret == -1) {
4612                     c_write_str(ssh, "Wrong passphrase.\r\n"); /* FIXME */
4613                     got_passphrase = FALSE;
4614                     /* and try again */
4615                 } else {
4616                     assert(0 && "unexpected return from loadrsakey()");
4617                     got_passphrase = FALSE;   /* placate optimisers */
4618                 }
4619             }
4620
4621             if (got_passphrase) {
4622
4623                 /*
4624                  * Send a public key attempt.
4625                  */
4626                 send_packet(ssh, SSH1_CMSG_AUTH_RSA,
4627                             PKT_BIGNUM, s->key.modulus, PKT_END);
4628
4629                 crWaitUntil(pktin);
4630                 if (pktin->type == SSH1_SMSG_FAILURE) {
4631                     c_write_str(ssh, "Server refused our public key.\r\n");
4632                     continue;          /* go and try something else */
4633                 }
4634                 if (pktin->type != SSH1_SMSG_AUTH_RSA_CHALLENGE) {
4635                     bombout(("Bizarre response to offer of public key"));
4636                     crStop(0);
4637                 }
4638
4639                 {
4640                     int i;
4641                     unsigned char buffer[32];
4642                     Bignum challenge, response;
4643
4644                     if ((challenge = ssh1_pkt_getmp(pktin)) == NULL) {
4645                         bombout(("Server's RSA challenge was badly formatted"));
4646                         crStop(0);
4647                     }
4648                     response = rsadecrypt(challenge, &s->key);
4649                     freebn(s->key.private_exponent);/* burn the evidence */
4650
4651                     for (i = 0; i < 32; i++) {
4652                         buffer[i] = bignum_byte(response, 31 - i);
4653                     }
4654
4655                     MD5Init(&md5c);
4656                     MD5Update(&md5c, buffer, 32);
4657                     MD5Update(&md5c, s->session_id, 16);
4658                     MD5Final(buffer, &md5c);
4659
4660                     send_packet(ssh, SSH1_CMSG_AUTH_RSA_RESPONSE,
4661                                 PKT_DATA, buffer, 16, PKT_END);
4662
4663                     freebn(challenge);
4664                     freebn(response);
4665                 }
4666
4667                 crWaitUntil(pktin);
4668                 if (pktin->type == SSH1_SMSG_FAILURE) {
4669                     if (flags & FLAG_VERBOSE)
4670                         c_write_str(ssh, "Failed to authenticate with"
4671                                     " our public key.\r\n");
4672                     continue;          /* go and try something else */
4673                 } else if (pktin->type != SSH1_SMSG_SUCCESS) {
4674                     bombout(("Bizarre response to RSA authentication response"));
4675                     crStop(0);
4676                 }
4677
4678                 break;                 /* we're through! */
4679             }
4680
4681         }
4682
4683         /*
4684          * Otherwise, try various forms of password-like authentication.
4685          */
4686         s->cur_prompt = new_prompts(ssh->frontend);
4687
4688         if (conf_get_int(ssh->conf, CONF_try_tis_auth) &&
4689             (s->supported_auths_mask & (1 << SSH1_AUTH_TIS)) &&
4690             !s->tis_auth_refused) {
4691             s->pwpkt_type = SSH1_CMSG_AUTH_TIS_RESPONSE;
4692             logevent("Requested TIS authentication");
4693             send_packet(ssh, SSH1_CMSG_AUTH_TIS, PKT_END);
4694             crWaitUntil(pktin);
4695             if (pktin->type != SSH1_SMSG_AUTH_TIS_CHALLENGE) {
4696                 logevent("TIS authentication declined");
4697                 if (flags & FLAG_INTERACTIVE)
4698                     c_write_str(ssh, "TIS authentication refused.\r\n");
4699                 s->tis_auth_refused = 1;
4700                 continue;
4701             } else {
4702                 char *challenge;
4703                 int challengelen;
4704                 char *instr_suf, *prompt;
4705
4706                 ssh_pkt_getstring(pktin, &challenge, &challengelen);
4707                 if (!challenge) {
4708                     bombout(("TIS challenge packet was badly formed"));
4709                     crStop(0);
4710                 }
4711                 logevent("Received TIS challenge");
4712                 s->cur_prompt->to_server = TRUE;
4713                 s->cur_prompt->name = dupstr("SSH TIS authentication");
4714                 /* Prompt heuristic comes from OpenSSH */
4715                 if (memchr(challenge, '\n', challengelen)) {
4716                     instr_suf = dupstr("");
4717                     prompt = dupprintf("%.*s", challengelen, challenge);
4718                 } else {
4719                     instr_suf = dupprintf("%.*s", challengelen, challenge);
4720                     prompt = dupstr("Response: ");
4721                 }
4722                 s->cur_prompt->instruction =
4723                     dupprintf("Using TIS authentication.%s%s",
4724                               (*instr_suf) ? "\n" : "",
4725                               instr_suf);
4726                 s->cur_prompt->instr_reqd = TRUE;
4727                 add_prompt(s->cur_prompt, prompt, FALSE);
4728                 sfree(instr_suf);
4729             }
4730         }
4731         if (conf_get_int(ssh->conf, CONF_try_tis_auth) &&
4732             (s->supported_auths_mask & (1 << SSH1_AUTH_CCARD)) &&
4733             !s->ccard_auth_refused) {
4734             s->pwpkt_type = SSH1_CMSG_AUTH_CCARD_RESPONSE;
4735             logevent("Requested CryptoCard authentication");
4736             send_packet(ssh, SSH1_CMSG_AUTH_CCARD, PKT_END);
4737             crWaitUntil(pktin);
4738             if (pktin->type != SSH1_SMSG_AUTH_CCARD_CHALLENGE) {
4739                 logevent("CryptoCard authentication declined");
4740                 c_write_str(ssh, "CryptoCard authentication refused.\r\n");
4741                 s->ccard_auth_refused = 1;
4742                 continue;
4743             } else {
4744                 char *challenge;
4745                 int challengelen;
4746                 char *instr_suf, *prompt;
4747
4748                 ssh_pkt_getstring(pktin, &challenge, &challengelen);
4749                 if (!challenge) {
4750                     bombout(("CryptoCard challenge packet was badly formed"));
4751                     crStop(0);
4752                 }
4753                 logevent("Received CryptoCard challenge");
4754                 s->cur_prompt->to_server = TRUE;
4755                 s->cur_prompt->name = dupstr("SSH CryptoCard authentication");
4756                 s->cur_prompt->name_reqd = FALSE;
4757                 /* Prompt heuristic comes from OpenSSH */
4758                 if (memchr(challenge, '\n', challengelen)) {
4759                     instr_suf = dupstr("");
4760                     prompt = dupprintf("%.*s", challengelen, challenge);
4761                 } else {
4762                     instr_suf = dupprintf("%.*s", challengelen, challenge);
4763                     prompt = dupstr("Response: ");
4764                 }
4765                 s->cur_prompt->instruction =
4766                     dupprintf("Using CryptoCard authentication.%s%s",
4767                               (*instr_suf) ? "\n" : "",
4768                               instr_suf);
4769                 s->cur_prompt->instr_reqd = TRUE;
4770                 add_prompt(s->cur_prompt, prompt, FALSE);
4771                 sfree(instr_suf);
4772             }
4773         }
4774         if (s->pwpkt_type == SSH1_CMSG_AUTH_PASSWORD) {
4775             if ((s->supported_auths_mask & (1 << SSH1_AUTH_PASSWORD)) == 0) {
4776                 bombout(("No supported authentication methods available"));
4777                 crStop(0);
4778             }
4779             s->cur_prompt->to_server = TRUE;
4780             s->cur_prompt->name = dupstr("SSH password");
4781             add_prompt(s->cur_prompt, dupprintf("%s@%s's password: ",
4782                                                 ssh->username, ssh->savedhost),
4783                        FALSE);
4784         }
4785
4786         /*
4787          * Show password prompt, having first obtained it via a TIS
4788          * or CryptoCard exchange if we're doing TIS or CryptoCard
4789          * authentication.
4790          */
4791         {
4792             int ret; /* need not be kept over crReturn */
4793             ret = get_userpass_input(s->cur_prompt, NULL, 0);
4794             while (ret < 0) {
4795                 ssh->send_ok = 1;
4796                 crWaitUntil(!pktin);
4797                 ret = get_userpass_input(s->cur_prompt, in, inlen);
4798                 ssh->send_ok = 0;
4799             }
4800             if (!ret) {
4801                 /*
4802                  * Failed to get a password (for example
4803                  * because one was supplied on the command line
4804                  * which has already failed to work). Terminate.
4805                  */
4806                 free_prompts(s->cur_prompt);
4807                 ssh_disconnect(ssh, NULL, "Unable to authenticate", 0, TRUE);
4808                 crStop(0);
4809             }
4810         }
4811
4812         if (s->pwpkt_type == SSH1_CMSG_AUTH_PASSWORD) {
4813             /*
4814              * Defence against traffic analysis: we send a
4815              * whole bunch of packets containing strings of
4816              * different lengths. One of these strings is the
4817              * password, in a SSH1_CMSG_AUTH_PASSWORD packet.
4818              * The others are all random data in
4819              * SSH1_MSG_IGNORE packets. This way a passive
4820              * listener can't tell which is the password, and
4821              * hence can't deduce the password length.
4822              * 
4823              * Anybody with a password length greater than 16
4824              * bytes is going to have enough entropy in their
4825              * password that a listener won't find it _that_
4826              * much help to know how long it is. So what we'll
4827              * do is:
4828              * 
4829              *  - if password length < 16, we send 15 packets
4830              *    containing string lengths 1 through 15
4831              * 
4832              *  - otherwise, we let N be the nearest multiple
4833              *    of 8 below the password length, and send 8
4834              *    packets containing string lengths N through
4835              *    N+7. This won't obscure the order of
4836              *    magnitude of the password length, but it will
4837              *    introduce a bit of extra uncertainty.
4838              * 
4839              * A few servers can't deal with SSH1_MSG_IGNORE, at
4840              * least in this context. For these servers, we need
4841              * an alternative defence. We make use of the fact
4842              * that the password is interpreted as a C string:
4843              * so we can append a NUL, then some random data.
4844              * 
4845              * A few servers can deal with neither SSH1_MSG_IGNORE
4846              * here _nor_ a padded password string.
4847              * For these servers we are left with no defences
4848              * against password length sniffing.
4849              */
4850             if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH1_IGNORE) &&
4851                 !(ssh->remote_bugs & BUG_NEEDS_SSH1_PLAIN_PASSWORD)) {
4852                 /*
4853                  * The server can deal with SSH1_MSG_IGNORE, so
4854                  * we can use the primary defence.
4855                  */
4856                 int bottom, top, pwlen, i;
4857                 char *randomstr;
4858
4859                 pwlen = strlen(s->cur_prompt->prompts[0]->result);
4860                 if (pwlen < 16) {
4861                     bottom = 0;    /* zero length passwords are OK! :-) */
4862                     top = 15;
4863                 } else {
4864                     bottom = pwlen & ~7;
4865                     top = bottom + 7;
4866                 }
4867
4868                 assert(pwlen >= bottom && pwlen <= top);
4869
4870                 randomstr = snewn(top + 1, char);
4871
4872                 for (i = bottom; i <= top; i++) {
4873                     if (i == pwlen) {
4874                         defer_packet(ssh, s->pwpkt_type,
4875                                      PKT_STR,s->cur_prompt->prompts[0]->result,
4876                                      PKT_END);
4877                     } else {
4878                         for (j = 0; j < i; j++) {
4879                             do {
4880                                 randomstr[j] = random_byte();
4881                             } while (randomstr[j] == '\0');
4882                         }
4883                         randomstr[i] = '\0';
4884                         defer_packet(ssh, SSH1_MSG_IGNORE,
4885                                      PKT_STR, randomstr, PKT_END);
4886                     }
4887                 }
4888                 logevent("Sending password with camouflage packets");
4889                 ssh_pkt_defersend(ssh);
4890                 sfree(randomstr);
4891             } 
4892             else if (!(ssh->remote_bugs & BUG_NEEDS_SSH1_PLAIN_PASSWORD)) {
4893                 /*
4894                  * The server can't deal with SSH1_MSG_IGNORE
4895                  * but can deal with padded passwords, so we
4896                  * can use the secondary defence.
4897                  */
4898                 char string[64];
4899                 char *ss;
4900                 int len;
4901
4902                 len = strlen(s->cur_prompt->prompts[0]->result);
4903                 if (len < sizeof(string)) {
4904                     ss = string;
4905                     strcpy(string, s->cur_prompt->prompts[0]->result);
4906                     len++;             /* cover the zero byte */
4907                     while (len < sizeof(string)) {
4908                         string[len++] = (char) random_byte();
4909                     }
4910                 } else {
4911                     ss = s->cur_prompt->prompts[0]->result;
4912                 }
4913                 logevent("Sending length-padded password");
4914                 send_packet(ssh, s->pwpkt_type,
4915                             PKT_INT, len, PKT_DATA, ss, len,
4916                             PKT_END);
4917             } else {
4918                 /*
4919                  * The server is believed unable to cope with
4920                  * any of our password camouflage methods.
4921                  */
4922                 int len;
4923                 len = strlen(s->cur_prompt->prompts[0]->result);
4924                 logevent("Sending unpadded password");
4925                 send_packet(ssh, s->pwpkt_type,
4926                             PKT_INT, len,
4927                             PKT_DATA, s->cur_prompt->prompts[0]->result, len,
4928                             PKT_END);
4929             }
4930         } else {
4931             send_packet(ssh, s->pwpkt_type,
4932                         PKT_STR, s->cur_prompt->prompts[0]->result,
4933                         PKT_END);
4934         }
4935         logevent("Sent password");
4936         free_prompts(s->cur_prompt);
4937         crWaitUntil(pktin);
4938         if (pktin->type == SSH1_SMSG_FAILURE) {
4939             if (flags & FLAG_VERBOSE)
4940                 c_write_str(ssh, "Access denied\r\n");
4941             logevent("Authentication refused");
4942         } else if (pktin->type != SSH1_SMSG_SUCCESS) {
4943             bombout(("Strange packet received, type %d", pktin->type));
4944             crStop(0);
4945         }
4946     }
4947
4948     /* Clear up */
4949     if (s->publickey_blob) {
4950         sfree(s->publickey_blob);
4951         sfree(s->publickey_comment);
4952     }
4953
4954     logevent("Authentication successful");
4955
4956     crFinish(1);
4957 }
4958
4959 static void ssh_channel_try_eof(struct ssh_channel *c)
4960 {
4961     Ssh ssh = c->ssh;
4962     assert(c->pending_eof);          /* precondition for calling us */
4963     if (c->halfopen)
4964         return;                 /* can't close: not even opened yet */
4965     if (ssh->version == 2 && bufchain_size(&c->v.v2.outbuffer) > 0)
4966         return;              /* can't send EOF: pending outgoing data */
4967
4968     c->pending_eof = FALSE;            /* we're about to send it */
4969     if (ssh->version == 1) {
4970         send_packet(ssh, SSH1_MSG_CHANNEL_CLOSE, PKT_INT, c->remoteid,
4971                     PKT_END);
4972         c->closes |= CLOSES_SENT_EOF;
4973     } else {
4974         struct Packet *pktout;
4975         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_EOF);
4976         ssh2_pkt_adduint32(pktout, c->remoteid);
4977         ssh2_pkt_send(ssh, pktout);
4978         c->closes |= CLOSES_SENT_EOF;
4979         ssh2_channel_check_close(c);
4980     }
4981 }
4982
4983 Conf *sshfwd_get_conf(struct ssh_channel *c)
4984 {
4985     Ssh ssh = c->ssh;
4986     return ssh->conf;
4987 }
4988
4989 void sshfwd_write_eof(struct ssh_channel *c)
4990 {
4991     Ssh ssh = c->ssh;
4992
4993     if (ssh->state == SSH_STATE_CLOSED)
4994         return;
4995
4996     if (c->closes & CLOSES_SENT_EOF)
4997         return;
4998
4999     c->pending_eof = TRUE;
5000     ssh_channel_try_eof(c);
5001 }
5002
5003 void sshfwd_unclean_close(struct ssh_channel *c, const char *err)
5004 {
5005     Ssh ssh = c->ssh;
5006
5007     if (ssh->state == SSH_STATE_CLOSED)
5008         return;
5009
5010     switch (c->type) {
5011       case CHAN_X11:
5012         x11_close(c->u.x11.xconn);
5013         logeventf(ssh, "Forwarded X11 connection terminated due to local "
5014                   "error: %s", err);
5015         break;
5016       case CHAN_SOCKDATA:
5017       case CHAN_SOCKDATA_DORMANT:
5018         pfd_close(c->u.pfd.pf);
5019         logeventf(ssh, "Forwarded port closed due to local error: %s", err);
5020         break;
5021     }
5022     c->type = CHAN_ZOMBIE;
5023     c->pending_eof = FALSE;   /* this will confuse a zombie channel */
5024
5025     ssh2_channel_check_close(c);
5026 }
5027
5028 int sshfwd_write(struct ssh_channel *c, char *buf, int len)
5029 {
5030     Ssh ssh = c->ssh;
5031
5032     if (ssh->state == SSH_STATE_CLOSED)
5033         return 0;
5034
5035     return ssh_send_channel_data(c, buf, len);
5036 }
5037
5038 void sshfwd_unthrottle(struct ssh_channel *c, int bufsize)
5039 {
5040     Ssh ssh = c->ssh;
5041
5042     if (ssh->state == SSH_STATE_CLOSED)
5043         return;
5044
5045     ssh_channel_unthrottle(c, bufsize);
5046 }
5047
5048 static void ssh_queueing_handler(Ssh ssh, struct Packet *pktin)
5049 {
5050     struct queued_handler *qh = ssh->qhead;
5051
5052     assert(qh != NULL);
5053
5054     assert(pktin->type == qh->msg1 || pktin->type == qh->msg2);
5055
5056     if (qh->msg1 > 0) {
5057         assert(ssh->packet_dispatch[qh->msg1] == ssh_queueing_handler);
5058         ssh->packet_dispatch[qh->msg1] = ssh->q_saved_handler1;
5059     }
5060     if (qh->msg2 > 0) {
5061         assert(ssh->packet_dispatch[qh->msg2] == ssh_queueing_handler);
5062         ssh->packet_dispatch[qh->msg2] = ssh->q_saved_handler2;
5063     }
5064
5065     if (qh->next) {
5066         ssh->qhead = qh->next;
5067
5068         if (ssh->qhead->msg1 > 0) {
5069             ssh->q_saved_handler1 = ssh->packet_dispatch[ssh->qhead->msg1];
5070             ssh->packet_dispatch[ssh->qhead->msg1] = ssh_queueing_handler;
5071         }
5072         if (ssh->qhead->msg2 > 0) {
5073             ssh->q_saved_handler2 = ssh->packet_dispatch[ssh->qhead->msg2];
5074             ssh->packet_dispatch[ssh->qhead->msg2] = ssh_queueing_handler;
5075         }
5076     } else {
5077         ssh->qhead = ssh->qtail = NULL;
5078     }
5079
5080     qh->handler(ssh, pktin, qh->ctx);
5081
5082     sfree(qh);
5083 }
5084
5085 static void ssh_queue_handler(Ssh ssh, int msg1, int msg2,
5086                               chandler_fn_t handler, void *ctx)
5087 {
5088     struct queued_handler *qh;
5089
5090     qh = snew(struct queued_handler);
5091     qh->msg1 = msg1;
5092     qh->msg2 = msg2;
5093     qh->handler = handler;
5094     qh->ctx = ctx;
5095     qh->next = NULL;
5096
5097     if (ssh->qtail == NULL) {
5098         ssh->qhead = qh;
5099
5100         if (qh->msg1 > 0) {
5101             ssh->q_saved_handler1 = ssh->packet_dispatch[ssh->qhead->msg1];
5102             ssh->packet_dispatch[qh->msg1] = ssh_queueing_handler;
5103         }
5104         if (qh->msg2 > 0) {
5105             ssh->q_saved_handler2 = ssh->packet_dispatch[ssh->qhead->msg2];
5106             ssh->packet_dispatch[qh->msg2] = ssh_queueing_handler;
5107         }
5108     } else {
5109         ssh->qtail->next = qh;
5110     }
5111     ssh->qtail = qh;
5112 }
5113
5114 static void ssh_rportfwd_succfail(Ssh ssh, struct Packet *pktin, void *ctx)
5115 {
5116     struct ssh_rportfwd *rpf, *pf = (struct ssh_rportfwd *)ctx;
5117
5118     if (pktin->type == (ssh->version == 1 ? SSH1_SMSG_SUCCESS :
5119                         SSH2_MSG_REQUEST_SUCCESS)) {
5120         logeventf(ssh, "Remote port forwarding from %s enabled",
5121                   pf->sportdesc);
5122     } else {
5123         logeventf(ssh, "Remote port forwarding from %s refused",
5124                   pf->sportdesc);
5125
5126         rpf = del234(ssh->rportfwds, pf);
5127         assert(rpf == pf);
5128         pf->pfrec->remote = NULL;
5129         free_rportfwd(pf);
5130     }
5131 }
5132
5133 int ssh_alloc_sharing_rportfwd(Ssh ssh, const char *shost, int sport,
5134                                void *share_ctx)
5135 {
5136     struct ssh_rportfwd *pf = snew(struct ssh_rportfwd);
5137     pf->dhost = NULL;
5138     pf->dport = 0;
5139     pf->share_ctx = share_ctx;
5140     pf->shost = dupstr(shost);
5141     pf->sport = sport;
5142     pf->sportdesc = NULL;
5143     if (!ssh->rportfwds) {
5144         assert(ssh->version == 2);
5145         ssh->rportfwds = newtree234(ssh_rportcmp_ssh2);
5146     }
5147     if (add234(ssh->rportfwds, pf) != pf) {
5148         sfree(pf->shost);
5149         sfree(pf);
5150         return FALSE;
5151     }
5152     return TRUE;
5153 }
5154
5155 static void ssh_sharing_global_request_response(Ssh ssh, struct Packet *pktin,
5156                                                 void *ctx)
5157 {
5158     share_got_pkt_from_server(ctx, pktin->type,
5159                               pktin->body, pktin->length);
5160 }
5161
5162 void ssh_sharing_queue_global_request(Ssh ssh, void *share_ctx)
5163 {
5164     ssh_queue_handler(ssh, SSH2_MSG_REQUEST_SUCCESS, SSH2_MSG_REQUEST_FAILURE,
5165                       ssh_sharing_global_request_response, share_ctx);
5166 }
5167
5168 static void ssh_setup_portfwd(Ssh ssh, Conf *conf)
5169 {
5170     struct ssh_portfwd *epf;
5171     int i;
5172     char *key, *val;
5173
5174     if (!ssh->portfwds) {
5175         ssh->portfwds = newtree234(ssh_portcmp);
5176     } else {
5177         /*
5178          * Go through the existing port forwardings and tag them
5179          * with status==DESTROY. Any that we want to keep will be
5180          * re-enabled (status==KEEP) as we go through the
5181          * configuration and find out which bits are the same as
5182          * they were before.
5183          */
5184         struct ssh_portfwd *epf;
5185         int i;
5186         for (i = 0; (epf = index234(ssh->portfwds, i)) != NULL; i++)
5187             epf->status = DESTROY;
5188     }
5189
5190     for (val = conf_get_str_strs(conf, CONF_portfwd, NULL, &key);
5191          val != NULL;
5192          val = conf_get_str_strs(conf, CONF_portfwd, key, &key)) {
5193         char *kp, *kp2, *vp, *vp2;
5194         char address_family, type;
5195         int sport,dport,sserv,dserv;
5196         char *sports, *dports, *saddr, *host;
5197
5198         kp = key;
5199
5200         address_family = 'A';
5201         type = 'L';
5202         if (*kp == 'A' || *kp == '4' || *kp == '6')
5203             address_family = *kp++;
5204         if (*kp == 'L' || *kp == 'R')
5205             type = *kp++;
5206
5207         if ((kp2 = host_strchr(kp, ':')) != NULL) {
5208             /*
5209              * There's a colon in the middle of the source port
5210              * string, which means that the part before it is
5211              * actually a source address.
5212              */
5213             char *saddr_tmp = dupprintf("%.*s", (int)(kp2 - kp), kp);
5214             saddr = host_strduptrim(saddr_tmp);
5215             sfree(saddr_tmp);
5216             sports = kp2+1;
5217         } else {
5218             saddr = NULL;
5219             sports = kp;
5220         }
5221         sport = atoi(sports);
5222         sserv = 0;
5223         if (sport == 0) {
5224             sserv = 1;
5225             sport = net_service_lookup(sports);
5226             if (!sport) {
5227                 logeventf(ssh, "Service lookup failed for source"
5228                           " port \"%s\"", sports);
5229             }
5230         }
5231
5232         if (type == 'L' && !strcmp(val, "D")) {
5233             /* dynamic forwarding */
5234             host = NULL;
5235             dports = NULL;
5236             dport = -1;
5237             dserv = 0;
5238             type = 'D';
5239         } else {
5240             /* ordinary forwarding */
5241             vp = val;
5242             vp2 = vp + host_strcspn(vp, ":");
5243             host = dupprintf("%.*s", (int)(vp2 - vp), vp);
5244             if (*vp2)
5245                 vp2++;
5246             dports = vp2;
5247             dport = atoi(dports);
5248             dserv = 0;
5249             if (dport == 0) {
5250                 dserv = 1;
5251                 dport = net_service_lookup(dports);
5252                 if (!dport) {
5253                     logeventf(ssh, "Service lookup failed for destination"
5254                               " port \"%s\"", dports);
5255                 }
5256             }
5257         }
5258
5259         if (sport && dport) {
5260             /* Set up a description of the source port. */
5261             struct ssh_portfwd *pfrec, *epfrec;
5262
5263             pfrec = snew(struct ssh_portfwd);
5264             pfrec->type = type;
5265             pfrec->saddr = saddr;
5266             pfrec->sserv = sserv ? dupstr(sports) : NULL;
5267             pfrec->sport = sport;
5268             pfrec->daddr = host;
5269             pfrec->dserv = dserv ? dupstr(dports) : NULL;
5270             pfrec->dport = dport;
5271             pfrec->local = NULL;
5272             pfrec->remote = NULL;
5273             pfrec->addressfamily = (address_family == '4' ? ADDRTYPE_IPV4 :
5274                                     address_family == '6' ? ADDRTYPE_IPV6 :
5275                                     ADDRTYPE_UNSPEC);
5276
5277             epfrec = add234(ssh->portfwds, pfrec);
5278             if (epfrec != pfrec) {
5279                 if (epfrec->status == DESTROY) {
5280                     /*
5281                      * We already have a port forwarding up and running
5282                      * with precisely these parameters. Hence, no need
5283                      * to do anything; simply re-tag the existing one
5284                      * as KEEP.
5285                      */
5286                     epfrec->status = KEEP;
5287                 }
5288                 /*
5289                  * Anything else indicates that there was a duplicate
5290                  * in our input, which we'll silently ignore.
5291                  */
5292                 free_portfwd(pfrec);
5293             } else {
5294                 pfrec->status = CREATE;
5295             }
5296         } else {
5297             sfree(saddr);
5298             sfree(host);
5299         }
5300     }
5301
5302     /*
5303      * Now go through and destroy any port forwardings which were
5304      * not re-enabled.
5305      */
5306     for (i = 0; (epf = index234(ssh->portfwds, i)) != NULL; i++)
5307         if (epf->status == DESTROY) {
5308             char *message;
5309
5310             message = dupprintf("%s port forwarding from %s%s%d",
5311                                 epf->type == 'L' ? "local" :
5312                                 epf->type == 'R' ? "remote" : "dynamic",
5313                                 epf->saddr ? epf->saddr : "",
5314                                 epf->saddr ? ":" : "",
5315                                 epf->sport);
5316
5317             if (epf->type != 'D') {
5318                 char *msg2 = dupprintf("%s to %s:%d", message,
5319                                        epf->daddr, epf->dport);
5320                 sfree(message);
5321                 message = msg2;
5322             }
5323
5324             logeventf(ssh, "Cancelling %s", message);
5325             sfree(message);
5326
5327             /* epf->remote or epf->local may be NULL if setting up a
5328              * forwarding failed. */
5329             if (epf->remote) {
5330                 struct ssh_rportfwd *rpf = epf->remote;
5331                 struct Packet *pktout;
5332
5333                 /*
5334                  * Cancel the port forwarding at the server
5335                  * end.
5336                  */
5337                 if (ssh->version == 1) {
5338                     /*
5339                      * We cannot cancel listening ports on the
5340                      * server side in SSH-1! There's no message
5341                      * to support it. Instead, we simply remove
5342                      * the rportfwd record from the local end
5343                      * so that any connections the server tries
5344                      * to make on it are rejected.
5345                      */
5346                 } else {
5347                     pktout = ssh2_pkt_init(SSH2_MSG_GLOBAL_REQUEST);
5348                     ssh2_pkt_addstring(pktout, "cancel-tcpip-forward");
5349                     ssh2_pkt_addbool(pktout, 0);/* _don't_ want reply */
5350                     if (epf->saddr) {
5351                         ssh2_pkt_addstring(pktout, epf->saddr);
5352                     } else if (conf_get_int(conf, CONF_rport_acceptall)) {
5353                         /* XXX: rport_acceptall may not represent
5354                          * what was used to open the original connection,
5355                          * since it's reconfigurable. */
5356                         ssh2_pkt_addstring(pktout, "");
5357                     } else {
5358                         ssh2_pkt_addstring(pktout, "localhost");
5359                     }
5360                     ssh2_pkt_adduint32(pktout, epf->sport);
5361                     ssh2_pkt_send(ssh, pktout);
5362                 }
5363
5364                 del234(ssh->rportfwds, rpf);
5365                 free_rportfwd(rpf);
5366             } else if (epf->local) {
5367                 pfl_terminate(epf->local);
5368             }
5369
5370             delpos234(ssh->portfwds, i);
5371             free_portfwd(epf);
5372             i--;                       /* so we don't skip one in the list */
5373         }
5374
5375     /*
5376      * And finally, set up any new port forwardings (status==CREATE).
5377      */
5378     for (i = 0; (epf = index234(ssh->portfwds, i)) != NULL; i++)
5379         if (epf->status == CREATE) {
5380             char *sportdesc, *dportdesc;
5381             sportdesc = dupprintf("%s%s%s%s%d%s",
5382                                   epf->saddr ? epf->saddr : "",
5383                                   epf->saddr ? ":" : "",
5384                                   epf->sserv ? epf->sserv : "",
5385                                   epf->sserv ? "(" : "",
5386                                   epf->sport,
5387                                   epf->sserv ? ")" : "");
5388             if (epf->type == 'D') {
5389                 dportdesc = NULL;
5390             } else {
5391                 dportdesc = dupprintf("%s:%s%s%d%s",
5392                                       epf->daddr,
5393                                       epf->dserv ? epf->dserv : "",
5394                                       epf->dserv ? "(" : "",
5395                                       epf->dport,
5396                                       epf->dserv ? ")" : "");
5397             }
5398
5399             if (epf->type == 'L') {
5400                 char *err = pfl_listen(epf->daddr, epf->dport,
5401                                        epf->saddr, epf->sport,
5402                                        ssh, conf, &epf->local,
5403                                        epf->addressfamily);
5404
5405                 logeventf(ssh, "Local %sport %s forwarding to %s%s%s",
5406                           epf->addressfamily == ADDRTYPE_IPV4 ? "IPv4 " :
5407                           epf->addressfamily == ADDRTYPE_IPV6 ? "IPv6 " : "",
5408                           sportdesc, dportdesc,
5409                           err ? " failed: " : "", err ? err : "");
5410                 if (err)
5411                     sfree(err);
5412             } else if (epf->type == 'D') {
5413                 char *err = pfl_listen(NULL, -1, epf->saddr, epf->sport,
5414                                        ssh, conf, &epf->local,
5415                                        epf->addressfamily);
5416
5417                 logeventf(ssh, "Local %sport %s SOCKS dynamic forwarding%s%s",
5418                           epf->addressfamily == ADDRTYPE_IPV4 ? "IPv4 " :
5419                           epf->addressfamily == ADDRTYPE_IPV6 ? "IPv6 " : "",
5420                           sportdesc,
5421                           err ? " failed: " : "", err ? err : "");
5422
5423                 if (err)
5424                     sfree(err);
5425             } else {
5426                 struct ssh_rportfwd *pf;
5427
5428                 /*
5429                  * Ensure the remote port forwardings tree exists.
5430                  */
5431                 if (!ssh->rportfwds) {
5432                     if (ssh->version == 1)
5433                         ssh->rportfwds = newtree234(ssh_rportcmp_ssh1);
5434                     else
5435                         ssh->rportfwds = newtree234(ssh_rportcmp_ssh2);
5436                 }
5437
5438                 pf = snew(struct ssh_rportfwd);
5439                 pf->share_ctx = NULL;
5440                 pf->dhost = dupstr(epf->daddr);
5441                 pf->dport = epf->dport;
5442                 if (epf->saddr) {
5443                     pf->shost = dupstr(epf->saddr);
5444                 } else if (conf_get_int(conf, CONF_rport_acceptall)) {
5445                     pf->shost = dupstr("");
5446                 } else {
5447                     pf->shost = dupstr("localhost");
5448                 }
5449                 pf->sport = epf->sport;
5450                 if (add234(ssh->rportfwds, pf) != pf) {
5451                     logeventf(ssh, "Duplicate remote port forwarding to %s:%d",
5452                               epf->daddr, epf->dport);
5453                     sfree(pf);
5454                 } else {
5455                     logeventf(ssh, "Requesting remote port %s"
5456                               " forward to %s", sportdesc, dportdesc);
5457
5458                     pf->sportdesc = sportdesc;
5459                     sportdesc = NULL;
5460                     epf->remote = pf;
5461                     pf->pfrec = epf;
5462
5463                     if (ssh->version == 1) {
5464                         send_packet(ssh, SSH1_CMSG_PORT_FORWARD_REQUEST,
5465                                     PKT_INT, epf->sport,
5466                                     PKT_STR, epf->daddr,
5467                                     PKT_INT, epf->dport,
5468                                     PKT_END);
5469                         ssh_queue_handler(ssh, SSH1_SMSG_SUCCESS,
5470                                           SSH1_SMSG_FAILURE,
5471                                           ssh_rportfwd_succfail, pf);
5472                     } else {
5473                         struct Packet *pktout;
5474                         pktout = ssh2_pkt_init(SSH2_MSG_GLOBAL_REQUEST);
5475                         ssh2_pkt_addstring(pktout, "tcpip-forward");
5476                         ssh2_pkt_addbool(pktout, 1);/* want reply */
5477                         ssh2_pkt_addstring(pktout, pf->shost);
5478                         ssh2_pkt_adduint32(pktout, pf->sport);
5479                         ssh2_pkt_send(ssh, pktout);
5480
5481                         ssh_queue_handler(ssh, SSH2_MSG_REQUEST_SUCCESS,
5482                                           SSH2_MSG_REQUEST_FAILURE,
5483                                           ssh_rportfwd_succfail, pf);
5484                     }
5485                 }
5486             }
5487             sfree(sportdesc);
5488             sfree(dportdesc);
5489         }
5490 }
5491
5492 static void ssh1_smsg_stdout_stderr_data(Ssh ssh, struct Packet *pktin)
5493 {
5494     char *string;
5495     int stringlen, bufsize;
5496
5497     ssh_pkt_getstring(pktin, &string, &stringlen);
5498     if (string == NULL) {
5499         bombout(("Incoming terminal data packet was badly formed"));
5500         return;
5501     }
5502
5503     bufsize = from_backend(ssh->frontend, pktin->type == SSH1_SMSG_STDERR_DATA,
5504                            string, stringlen);
5505     if (!ssh->v1_stdout_throttling && bufsize > SSH1_BUFFER_LIMIT) {
5506         ssh->v1_stdout_throttling = 1;
5507         ssh_throttle_conn(ssh, +1);
5508     }
5509 }
5510
5511 static void ssh1_smsg_x11_open(Ssh ssh, struct Packet *pktin)
5512 {
5513     /* Remote side is trying to open a channel to talk to our
5514      * X-Server. Give them back a local channel number. */
5515     struct ssh_channel *c;
5516     int remoteid = ssh_pkt_getuint32(pktin);
5517
5518     logevent("Received X11 connect request");
5519     /* Refuse if X11 forwarding is disabled. */
5520     if (!ssh->X11_fwd_enabled) {
5521         send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_FAILURE,
5522                     PKT_INT, remoteid, PKT_END);
5523         logevent("Rejected X11 connect request");
5524     } else {
5525         c = snew(struct ssh_channel);
5526         c->ssh = ssh;
5527
5528         ssh_channel_init(c);
5529         c->u.x11.xconn = x11_init(ssh->x11authtree, c, NULL, -1);
5530         c->remoteid = remoteid;
5531         c->halfopen = FALSE;
5532         c->type = CHAN_X11;     /* identify channel type */
5533         send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_CONFIRMATION,
5534                     PKT_INT, c->remoteid, PKT_INT,
5535                     c->localid, PKT_END);
5536         logevent("Opened X11 forward channel");
5537     }
5538 }
5539
5540 static void ssh1_smsg_agent_open(Ssh ssh, struct Packet *pktin)
5541 {
5542     /* Remote side is trying to open a channel to talk to our
5543      * agent. Give them back a local channel number. */
5544     struct ssh_channel *c;
5545     int remoteid = ssh_pkt_getuint32(pktin);
5546
5547     /* Refuse if agent forwarding is disabled. */
5548     if (!ssh->agentfwd_enabled) {
5549         send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_FAILURE,
5550                     PKT_INT, remoteid, PKT_END);
5551     } else {
5552         c = snew(struct ssh_channel);
5553         c->ssh = ssh;
5554         ssh_channel_init(c);
5555         c->remoteid = remoteid;
5556         c->halfopen = FALSE;
5557         c->type = CHAN_AGENT;   /* identify channel type */
5558         c->u.a.lensofar = 0;
5559         c->u.a.message = NULL;
5560         c->u.a.outstanding_requests = 0;
5561         send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_CONFIRMATION,
5562                     PKT_INT, c->remoteid, PKT_INT, c->localid,
5563                     PKT_END);
5564     }
5565 }
5566
5567 static void ssh1_msg_port_open(Ssh ssh, struct Packet *pktin)
5568 {
5569     /* Remote side is trying to open a channel to talk to a
5570      * forwarded port. Give them back a local channel number. */
5571     struct ssh_rportfwd pf, *pfp;
5572     int remoteid;
5573     int hostsize, port;
5574     char *host;
5575     char *err;
5576
5577     remoteid = ssh_pkt_getuint32(pktin);
5578     ssh_pkt_getstring(pktin, &host, &hostsize);
5579     port = ssh_pkt_getuint32(pktin);
5580
5581     pf.dhost = dupprintf("%.*s", hostsize, NULLTOEMPTY(host));
5582     pf.dport = port;
5583     pfp = find234(ssh->rportfwds, &pf, NULL);
5584
5585     if (pfp == NULL) {
5586         logeventf(ssh, "Rejected remote port open request for %s:%d",
5587                   pf.dhost, port);
5588         send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_FAILURE,
5589                     PKT_INT, remoteid, PKT_END);
5590     } else {
5591         struct ssh_channel *c = snew(struct ssh_channel);
5592         c->ssh = ssh;
5593
5594         logeventf(ssh, "Received remote port open request for %s:%d",
5595                   pf.dhost, port);
5596         err = pfd_connect(&c->u.pfd.pf, pf.dhost, port,
5597                           c, ssh->conf, pfp->pfrec->addressfamily);
5598         if (err != NULL) {
5599             logeventf(ssh, "Port open failed: %s", err);
5600             sfree(err);
5601             sfree(c);
5602             send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_FAILURE,
5603                         PKT_INT, remoteid, PKT_END);
5604         } else {
5605             ssh_channel_init(c);
5606             c->remoteid = remoteid;
5607             c->halfopen = FALSE;
5608             c->type = CHAN_SOCKDATA;    /* identify channel type */
5609             send_packet(ssh, SSH1_MSG_CHANNEL_OPEN_CONFIRMATION,
5610                         PKT_INT, c->remoteid, PKT_INT,
5611                         c->localid, PKT_END);
5612             logevent("Forwarded port opened successfully");
5613         }
5614     }
5615
5616     sfree(pf.dhost);
5617 }
5618
5619 static void ssh1_msg_channel_open_confirmation(Ssh ssh, struct Packet *pktin)
5620 {
5621     struct ssh_channel *c;
5622
5623     c = ssh_channel_msg(ssh, pktin);
5624     if (c && c->type == CHAN_SOCKDATA_DORMANT) {
5625         c->remoteid = ssh_pkt_getuint32(pktin);
5626         c->halfopen = FALSE;
5627         c->type = CHAN_SOCKDATA;
5628         c->throttling_conn = 0;
5629         pfd_confirm(c->u.pfd.pf);
5630     }
5631
5632     if (c && c->pending_eof) {
5633         /*
5634          * We have a pending close on this channel,
5635          * which we decided on before the server acked
5636          * the channel open. So now we know the
5637          * remoteid, we can close it again.
5638          */
5639         ssh_channel_try_eof(c);
5640     }
5641 }
5642
5643 static void ssh1_msg_channel_open_failure(Ssh ssh, struct Packet *pktin)
5644 {
5645     struct ssh_channel *c;
5646
5647     c = ssh_channel_msg(ssh, pktin);
5648     if (c && c->type == CHAN_SOCKDATA_DORMANT) {
5649         logevent("Forwarded connection refused by server");
5650         pfd_close(c->u.pfd.pf);
5651         del234(ssh->channels, c);
5652         sfree(c);
5653     }
5654 }
5655
5656 static void ssh1_msg_channel_close(Ssh ssh, struct Packet *pktin)
5657 {
5658     /* Remote side closes a channel. */
5659     struct ssh_channel *c;
5660
5661     c = ssh_channel_msg(ssh, pktin);
5662     if (c) {
5663
5664         if (pktin->type == SSH1_MSG_CHANNEL_CLOSE &&
5665             !(c->closes & CLOSES_RCVD_EOF)) {
5666             /*
5667              * Received CHANNEL_CLOSE, which we translate into
5668              * outgoing EOF.
5669              */
5670             int send_close = FALSE;
5671
5672             c->closes |= CLOSES_RCVD_EOF;
5673
5674             switch (c->type) {
5675               case CHAN_X11:
5676                 if (c->u.x11.xconn)
5677                     x11_send_eof(c->u.x11.xconn);
5678                 else
5679                     send_close = TRUE;
5680                 break;
5681               case CHAN_SOCKDATA:
5682                 if (c->u.pfd.pf)
5683                     pfd_send_eof(c->u.pfd.pf);
5684                 else
5685                     send_close = TRUE;
5686                 break;
5687               case CHAN_AGENT:
5688                 send_close = TRUE;
5689                 break;
5690             }
5691
5692             if (send_close && !(c->closes & CLOSES_SENT_EOF)) {
5693                 send_packet(ssh, SSH1_MSG_CHANNEL_CLOSE, PKT_INT, c->remoteid,
5694                             PKT_END);
5695                 c->closes |= CLOSES_SENT_EOF;
5696             }
5697         }
5698
5699         if (pktin->type == SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION &&
5700             !(c->closes & CLOSES_RCVD_CLOSE)) {
5701
5702             if (!(c->closes & CLOSES_SENT_EOF)) {
5703                 bombout(("Received CHANNEL_CLOSE_CONFIRMATION for channel %u"
5704                          " for which we never sent CHANNEL_CLOSE\n",
5705                          c->localid));
5706             }
5707
5708             c->closes |= CLOSES_RCVD_CLOSE;
5709         }
5710
5711         if (!((CLOSES_SENT_EOF | CLOSES_RCVD_EOF) & ~c->closes) &&
5712             !(c->closes & CLOSES_SENT_CLOSE)) {
5713             send_packet(ssh, SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION,
5714                         PKT_INT, c->remoteid, PKT_END);
5715             c->closes |= CLOSES_SENT_CLOSE;
5716         }
5717
5718         if (!((CLOSES_SENT_CLOSE | CLOSES_RCVD_CLOSE) & ~c->closes))
5719             ssh_channel_destroy(c);
5720     }
5721 }
5722
5723 /*
5724  * Handle incoming data on an SSH-1 or SSH-2 agent-forwarding channel.
5725  */
5726 static int ssh_agent_channel_data(struct ssh_channel *c, char *data,
5727                                   int length)
5728 {
5729     while (length > 0) {
5730         if (c->u.a.lensofar < 4) {
5731             unsigned int l = min(4 - c->u.a.lensofar, (unsigned)length);
5732             memcpy(c->u.a.msglen + c->u.a.lensofar, data, l);
5733             data += l;
5734             length -= l;
5735             c->u.a.lensofar += l;
5736         }
5737         if (c->u.a.lensofar == 4) {
5738             c->u.a.totallen = 4 + GET_32BIT(c->u.a.msglen);
5739             c->u.a.message = snewn(c->u.a.totallen, unsigned char);
5740             memcpy(c->u.a.message, c->u.a.msglen, 4);
5741         }
5742         if (c->u.a.lensofar >= 4 && length > 0) {
5743             unsigned int l = min(c->u.a.totallen - c->u.a.lensofar,
5744                                  (unsigned)length);
5745             memcpy(c->u.a.message + c->u.a.lensofar, data, l);
5746             data += l;
5747             length -= l;
5748             c->u.a.lensofar += l;
5749         }
5750         if (c->u.a.lensofar == c->u.a.totallen) {
5751             void *reply;
5752             int replylen;
5753             c->u.a.outstanding_requests++;
5754             if (agent_query(c->u.a.message, c->u.a.totallen, &reply, &replylen,
5755                             ssh_agentf_callback, c))
5756                 ssh_agentf_callback(c, reply, replylen);
5757             sfree(c->u.a.message);
5758             c->u.a.message = NULL;
5759             c->u.a.lensofar = 0;
5760         }
5761     }
5762     return 0;   /* agent channels never back up */
5763 }
5764
5765 static int ssh_channel_data(struct ssh_channel *c, int is_stderr,
5766                             char *data,  int length)
5767 {
5768     switch (c->type) {
5769       case CHAN_MAINSESSION:
5770         return from_backend(c->ssh->frontend, is_stderr, data, length);
5771       case CHAN_X11:
5772         return x11_send(c->u.x11.xconn, data, length);
5773       case CHAN_SOCKDATA:
5774         return pfd_send(c->u.pfd.pf, data, length);
5775       case CHAN_AGENT:
5776         return ssh_agent_channel_data(c, data, length);
5777     }
5778     return 0;
5779 }
5780
5781 static void ssh1_msg_channel_data(Ssh ssh, struct Packet *pktin)
5782 {
5783     /* Data sent down one of our channels. */
5784     char *p;
5785     int len;
5786     struct ssh_channel *c;
5787
5788     c = ssh_channel_msg(ssh, pktin);
5789     ssh_pkt_getstring(pktin, &p, &len);
5790
5791     if (c) {
5792         int bufsize = ssh_channel_data(c, FALSE, p, len);
5793         if (!c->throttling_conn && bufsize > SSH1_BUFFER_LIMIT) {
5794             c->throttling_conn = 1;
5795             ssh_throttle_conn(ssh, +1);
5796         }
5797     }
5798 }
5799
5800 static void ssh1_smsg_exit_status(Ssh ssh, struct Packet *pktin)
5801 {
5802     ssh->exitcode = ssh_pkt_getuint32(pktin);
5803     logeventf(ssh, "Server sent command exit status %d", ssh->exitcode);
5804     send_packet(ssh, SSH1_CMSG_EXIT_CONFIRMATION, PKT_END);
5805     /*
5806      * In case `helpful' firewalls or proxies tack
5807      * extra human-readable text on the end of the
5808      * session which we might mistake for another
5809      * encrypted packet, we close the session once
5810      * we've sent EXIT_CONFIRMATION.
5811      */
5812     ssh_disconnect(ssh, NULL, NULL, 0, TRUE);
5813 }
5814
5815 /* Helper function to deal with sending tty modes for REQUEST_PTY */
5816 static void ssh1_send_ttymode(void *data,
5817                               const struct ssh_ttymode *mode, char *val)
5818 {
5819     struct Packet *pktout = (struct Packet *)data;
5820     unsigned int arg = 0;
5821
5822     switch (mode->type) {
5823       case TTY_OP_CHAR:
5824         arg = ssh_tty_parse_specchar(val);
5825         break;
5826       case TTY_OP_BOOL:
5827         arg = ssh_tty_parse_boolean(val);
5828         break;
5829     }
5830     ssh2_pkt_addbyte(pktout, mode->opcode);
5831     ssh2_pkt_addbyte(pktout, arg);
5832 }
5833
5834 int ssh_agent_forwarding_permitted(Ssh ssh)
5835 {
5836     return conf_get_int(ssh->conf, CONF_agentfwd) && agent_exists();
5837 }
5838
5839 static void do_ssh1_connection(Ssh ssh, const unsigned char *in, int inlen,
5840                                struct Packet *pktin)
5841 {
5842     crBegin(ssh->do_ssh1_connection_crstate);
5843
5844     ssh->packet_dispatch[SSH1_SMSG_STDOUT_DATA] = 
5845         ssh->packet_dispatch[SSH1_SMSG_STDERR_DATA] =
5846         ssh1_smsg_stdout_stderr_data;
5847
5848     ssh->packet_dispatch[SSH1_MSG_CHANNEL_OPEN_CONFIRMATION] =
5849         ssh1_msg_channel_open_confirmation;
5850     ssh->packet_dispatch[SSH1_MSG_CHANNEL_OPEN_FAILURE] =
5851         ssh1_msg_channel_open_failure;
5852     ssh->packet_dispatch[SSH1_MSG_CHANNEL_CLOSE] =
5853         ssh->packet_dispatch[SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION] =
5854         ssh1_msg_channel_close;
5855     ssh->packet_dispatch[SSH1_MSG_CHANNEL_DATA] = ssh1_msg_channel_data;
5856     ssh->packet_dispatch[SSH1_SMSG_EXIT_STATUS] = ssh1_smsg_exit_status;
5857
5858     if (ssh_agent_forwarding_permitted(ssh)) {
5859         logevent("Requesting agent forwarding");
5860         send_packet(ssh, SSH1_CMSG_AGENT_REQUEST_FORWARDING, PKT_END);
5861         do {
5862             crReturnV;
5863         } while (!pktin);
5864         if (pktin->type != SSH1_SMSG_SUCCESS
5865             && pktin->type != SSH1_SMSG_FAILURE) {
5866             bombout(("Protocol confusion"));
5867             crStopV;
5868         } else if (pktin->type == SSH1_SMSG_FAILURE) {
5869             logevent("Agent forwarding refused");
5870         } else {
5871             logevent("Agent forwarding enabled");
5872             ssh->agentfwd_enabled = TRUE;
5873             ssh->packet_dispatch[SSH1_SMSG_AGENT_OPEN] = ssh1_smsg_agent_open;
5874         }
5875     }
5876
5877     if (conf_get_int(ssh->conf, CONF_x11_forward)) {
5878         ssh->x11disp =
5879             x11_setup_display(conf_get_str(ssh->conf, CONF_x11_display),
5880                               ssh->conf);
5881         if (!ssh->x11disp) {
5882             /* FIXME: return an error message from x11_setup_display */
5883             logevent("X11 forwarding not enabled: unable to"
5884                      " initialise X display");
5885         } else {
5886             ssh->x11auth = x11_invent_fake_auth
5887                 (ssh->x11authtree, conf_get_int(ssh->conf, CONF_x11_auth));
5888             ssh->x11auth->disp = ssh->x11disp;
5889
5890             logevent("Requesting X11 forwarding");
5891             if (ssh->v1_local_protoflags & SSH1_PROTOFLAG_SCREEN_NUMBER) {
5892                 send_packet(ssh, SSH1_CMSG_X11_REQUEST_FORWARDING,
5893                             PKT_STR, ssh->x11auth->protoname,
5894                             PKT_STR, ssh->x11auth->datastring,
5895                             PKT_INT, ssh->x11disp->screennum,
5896                             PKT_END);
5897             } else {
5898                 send_packet(ssh, SSH1_CMSG_X11_REQUEST_FORWARDING,
5899                             PKT_STR, ssh->x11auth->protoname,
5900                             PKT_STR, ssh->x11auth->datastring,
5901                             PKT_END);
5902             }
5903             do {
5904                 crReturnV;
5905             } while (!pktin);
5906             if (pktin->type != SSH1_SMSG_SUCCESS
5907                 && pktin->type != SSH1_SMSG_FAILURE) {
5908                 bombout(("Protocol confusion"));
5909                 crStopV;
5910             } else if (pktin->type == SSH1_SMSG_FAILURE) {
5911                 logevent("X11 forwarding refused");
5912             } else {
5913                 logevent("X11 forwarding enabled");
5914                 ssh->X11_fwd_enabled = TRUE;
5915                 ssh->packet_dispatch[SSH1_SMSG_X11_OPEN] = ssh1_smsg_x11_open;
5916             }
5917         }
5918     }
5919
5920     ssh_setup_portfwd(ssh, ssh->conf);
5921     ssh->packet_dispatch[SSH1_MSG_PORT_OPEN] = ssh1_msg_port_open;
5922
5923     if (!conf_get_int(ssh->conf, CONF_nopty)) {
5924         struct Packet *pkt;
5925         /* Unpick the terminal-speed string. */
5926         /* XXX perhaps we should allow no speeds to be sent. */
5927         ssh->ospeed = 38400; ssh->ispeed = 38400; /* last-resort defaults */
5928         sscanf(conf_get_str(ssh->conf, CONF_termspeed), "%d,%d", &ssh->ospeed, &ssh->ispeed);
5929         /* Send the pty request. */
5930         pkt = ssh1_pkt_init(SSH1_CMSG_REQUEST_PTY);
5931         ssh_pkt_addstring(pkt, conf_get_str(ssh->conf, CONF_termtype));
5932         ssh_pkt_adduint32(pkt, ssh->term_height);
5933         ssh_pkt_adduint32(pkt, ssh->term_width);
5934         ssh_pkt_adduint32(pkt, 0); /* width in pixels */
5935         ssh_pkt_adduint32(pkt, 0); /* height in pixels */
5936         parse_ttymodes(ssh, ssh1_send_ttymode, (void *)pkt);
5937         ssh_pkt_addbyte(pkt, SSH1_TTY_OP_ISPEED);
5938         ssh_pkt_adduint32(pkt, ssh->ispeed);
5939         ssh_pkt_addbyte(pkt, SSH1_TTY_OP_OSPEED);
5940         ssh_pkt_adduint32(pkt, ssh->ospeed);
5941         ssh_pkt_addbyte(pkt, SSH_TTY_OP_END);
5942         s_wrpkt(ssh, pkt);
5943         ssh->state = SSH_STATE_INTERMED;
5944         do {
5945             crReturnV;
5946         } while (!pktin);
5947         if (pktin->type != SSH1_SMSG_SUCCESS
5948             && pktin->type != SSH1_SMSG_FAILURE) {
5949             bombout(("Protocol confusion"));
5950             crStopV;
5951         } else if (pktin->type == SSH1_SMSG_FAILURE) {
5952             c_write_str(ssh, "Server refused to allocate pty\r\n");
5953             ssh->editing = ssh->echoing = 1;
5954         } else {
5955             logeventf(ssh, "Allocated pty (ospeed %dbps, ispeed %dbps)",
5956                       ssh->ospeed, ssh->ispeed);
5957             ssh->got_pty = TRUE;
5958         }
5959     } else {
5960         ssh->editing = ssh->echoing = 1;
5961     }
5962
5963     if (conf_get_int(ssh->conf, CONF_compression)) {
5964         send_packet(ssh, SSH1_CMSG_REQUEST_COMPRESSION, PKT_INT, 6, PKT_END);
5965         do {
5966             crReturnV;
5967         } while (!pktin);
5968         if (pktin->type != SSH1_SMSG_SUCCESS
5969             && pktin->type != SSH1_SMSG_FAILURE) {
5970             bombout(("Protocol confusion"));
5971             crStopV;
5972         } else if (pktin->type == SSH1_SMSG_FAILURE) {
5973             c_write_str(ssh, "Server refused to compress\r\n");
5974         }
5975         logevent("Started compression");
5976         ssh->v1_compressing = TRUE;
5977         ssh->cs_comp_ctx = zlib_compress_init();
5978         logevent("Initialised zlib (RFC1950) compression");
5979         ssh->sc_comp_ctx = zlib_decompress_init();
5980         logevent("Initialised zlib (RFC1950) decompression");
5981     }
5982
5983     /*
5984      * Start the shell or command.
5985      * 
5986      * Special case: if the first-choice command is an SSH-2
5987      * subsystem (hence not usable here) and the second choice
5988      * exists, we fall straight back to that.
5989      */
5990     {
5991         char *cmd = conf_get_str(ssh->conf, CONF_remote_cmd);
5992         
5993         if (conf_get_int(ssh->conf, CONF_ssh_subsys) &&
5994             conf_get_str(ssh->conf, CONF_remote_cmd2)) {
5995             cmd = conf_get_str(ssh->conf, CONF_remote_cmd2);
5996             ssh->fallback_cmd = TRUE;
5997         }
5998         if (*cmd)
5999             send_packet(ssh, SSH1_CMSG_EXEC_CMD, PKT_STR, cmd, PKT_END);
6000         else
6001             send_packet(ssh, SSH1_CMSG_EXEC_SHELL, PKT_END);
6002         logevent("Started session");
6003     }
6004
6005     ssh->state = SSH_STATE_SESSION;
6006     if (ssh->size_needed)
6007         ssh_size(ssh, ssh->term_width, ssh->term_height);
6008     if (ssh->eof_needed)
6009         ssh_special(ssh, TS_EOF);
6010
6011     if (ssh->ldisc)
6012         ldisc_echoedit_update(ssh->ldisc);  /* cause ldisc to notice changes */
6013     ssh->send_ok = 1;
6014     ssh->channels = newtree234(ssh_channelcmp);
6015     while (1) {
6016
6017         /*
6018          * By this point, most incoming packets are already being
6019          * handled by the dispatch table, and we need only pay
6020          * attention to the unusual ones.
6021          */
6022
6023         crReturnV;
6024         if (pktin) {
6025             if (pktin->type == SSH1_SMSG_SUCCESS) {
6026                 /* may be from EXEC_SHELL on some servers */
6027             } else if (pktin->type == SSH1_SMSG_FAILURE) {
6028                 /* may be from EXEC_SHELL on some servers
6029                  * if no pty is available or in other odd cases. Ignore */
6030             } else {
6031                 bombout(("Strange packet received: type %d", pktin->type));
6032                 crStopV;
6033             }
6034         } else {
6035             while (inlen > 0) {
6036                 int len = min(inlen, 512);
6037                 send_packet(ssh, SSH1_CMSG_STDIN_DATA,
6038                             PKT_INT, len, PKT_DATA, in, len,
6039                             PKT_END);
6040                 in += len;
6041                 inlen -= len;
6042             }
6043         }
6044     }
6045
6046     crFinishV;
6047 }
6048
6049 /*
6050  * Handle the top-level SSH-2 protocol.
6051  */
6052 static void ssh1_msg_debug(Ssh ssh, struct Packet *pktin)
6053 {
6054     char *msg;
6055     int msglen;
6056
6057     ssh_pkt_getstring(pktin, &msg, &msglen);
6058     logeventf(ssh, "Remote debug message: %.*s", msglen, NULLTOEMPTY(msg));
6059 }
6060
6061 static void ssh1_msg_disconnect(Ssh ssh, struct Packet *pktin)
6062 {
6063     /* log reason code in disconnect message */
6064     char *msg;
6065     int msglen;
6066
6067     ssh_pkt_getstring(pktin, &msg, &msglen);
6068     bombout(("Server sent disconnect message:\n\"%.*s\"",
6069              msglen, NULLTOEMPTY(msg)));
6070 }
6071
6072 static void ssh_msg_ignore(Ssh ssh, struct Packet *pktin)
6073 {
6074     /* Do nothing, because we're ignoring it! Duhh. */
6075 }
6076
6077 static void ssh1_protocol_setup(Ssh ssh)
6078 {
6079     int i;
6080
6081     /*
6082      * Most messages are handled by the coroutines.
6083      */
6084     for (i = 0; i < 256; i++)
6085         ssh->packet_dispatch[i] = NULL;
6086
6087     /*
6088      * These special message types we install handlers for.
6089      */
6090     ssh->packet_dispatch[SSH1_MSG_DISCONNECT] = ssh1_msg_disconnect;
6091     ssh->packet_dispatch[SSH1_MSG_IGNORE] = ssh_msg_ignore;
6092     ssh->packet_dispatch[SSH1_MSG_DEBUG] = ssh1_msg_debug;
6093 }
6094
6095 static void ssh1_protocol(Ssh ssh, const void *vin, int inlen,
6096                           struct Packet *pktin)
6097 {
6098     const unsigned char *in = (const unsigned char *)vin;
6099     if (ssh->state == SSH_STATE_CLOSED)
6100         return;
6101
6102     if (pktin && ssh->packet_dispatch[pktin->type]) {
6103         ssh->packet_dispatch[pktin->type](ssh, pktin);
6104         return;
6105     }
6106
6107     if (!ssh->protocol_initial_phase_done) {
6108         if (do_ssh1_login(ssh, in, inlen, pktin))
6109             ssh->protocol_initial_phase_done = TRUE;
6110         else
6111             return;
6112     }
6113
6114     do_ssh1_connection(ssh, in, inlen, pktin);
6115 }
6116
6117 /*
6118  * Utility routines for decoding comma-separated strings in KEXINIT.
6119  */
6120 static int first_in_commasep_string(char const *needle, char const *haystack,
6121                                     int haylen)
6122 {
6123     int needlen;
6124     if (!needle || !haystack)          /* protect against null pointers */
6125         return 0;
6126     needlen = strlen(needle);
6127
6128     if (haylen >= needlen &&       /* haystack is long enough */
6129         !memcmp(needle, haystack, needlen) &&   /* initial match */
6130         (haylen == needlen || haystack[needlen] == ',')
6131         /* either , or EOS follows */
6132         )
6133         return 1;
6134     return 0;
6135 }
6136
6137 static int in_commasep_string(char const *needle, char const *haystack,
6138                               int haylen)
6139 {
6140     char *p;
6141
6142     if (!needle || !haystack)          /* protect against null pointers */
6143         return 0;
6144     /*
6145      * Is it at the start of the string?
6146      */
6147     if (first_in_commasep_string(needle, haystack, haylen))
6148         return 1;
6149     /*
6150      * If not, search for the next comma and resume after that.
6151      * If no comma found, terminate.
6152      */
6153     p = memchr(haystack, ',', haylen);
6154     if (!p) return 0;
6155     /* + 1 to skip over comma */
6156     return in_commasep_string(needle, p + 1, haylen - (p + 1 - haystack));
6157 }
6158
6159 /*
6160  * Add a value to the comma-separated string at the end of the packet.
6161  */
6162 static void ssh2_pkt_addstring_commasep(struct Packet *pkt, const char *data)
6163 {
6164     if (pkt->length - pkt->savedpos > 0)
6165         ssh_pkt_addstring_str(pkt, ",");
6166     ssh_pkt_addstring_str(pkt, data);
6167 }
6168
6169
6170 /*
6171  * SSH-2 key derivation (RFC 4253 section 7.2).
6172  */
6173 static unsigned char *ssh2_mkkey(Ssh ssh, Bignum K, unsigned char *H,
6174                                  char chr, int keylen)
6175 {
6176     const struct ssh_hash *h = ssh->kex->hash;
6177     int keylen_padded;
6178     unsigned char *key;
6179     void *s, *s2;
6180
6181     if (keylen == 0)
6182         return NULL;
6183
6184     /* Round up to the next multiple of hash length. */
6185     keylen_padded = ((keylen + h->hlen - 1) / h->hlen) * h->hlen;
6186
6187     key = snewn(keylen_padded, unsigned char);
6188
6189     /* First hlen bytes. */
6190     s = h->init();
6191     if (!(ssh->remote_bugs & BUG_SSH2_DERIVEKEY))
6192         hash_mpint(h, s, K);
6193     h->bytes(s, H, h->hlen);
6194     h->bytes(s, &chr, 1);
6195     h->bytes(s, ssh->v2_session_id, ssh->v2_session_id_len);
6196     h->final(s, key);
6197
6198     /* Subsequent blocks of hlen bytes. */
6199     if (keylen_padded > h->hlen) {
6200         int offset;
6201
6202         s = h->init();
6203         if (!(ssh->remote_bugs & BUG_SSH2_DERIVEKEY))
6204             hash_mpint(h, s, K);
6205         h->bytes(s, H, h->hlen);
6206
6207         for (offset = h->hlen; offset < keylen_padded; offset += h->hlen) {
6208             h->bytes(s, key + offset - h->hlen, h->hlen);
6209             s2 = h->copy(s);
6210             h->final(s2, key + offset);
6211         }
6212
6213         h->free(s);
6214     }
6215
6216     /* Now clear any extra bytes of key material beyond the length
6217      * we're officially returning, because the caller won't know to
6218      * smemclr those. */
6219     if (keylen_padded > keylen)
6220         smemclr(key + keylen, keylen_padded - keylen);
6221
6222     return key;
6223 }
6224
6225 /*
6226  * Structure for constructing KEXINIT algorithm lists.
6227  */
6228 #define MAXKEXLIST 16
6229 struct kexinit_algorithm {
6230     const char *name;
6231     union {
6232         struct {
6233             const struct ssh_kex *kex;
6234             int warn;
6235         } kex;
6236         struct {
6237             const struct ssh_signkey *hostkey;
6238             int warn;
6239         } hk;
6240         struct {
6241             const struct ssh2_cipher *cipher;
6242             int warn;
6243         } cipher;
6244         struct {
6245             const struct ssh_mac *mac;
6246             int etm;
6247         } mac;
6248         const struct ssh_compress *comp;
6249     } u;
6250 };
6251
6252 /*
6253  * Find a slot in a KEXINIT algorithm list to use for a new algorithm.
6254  * If the algorithm is already in the list, return a pointer to its
6255  * entry, otherwise return an entry from the end of the list.
6256  * This assumes that every time a particular name is passed in, it
6257  * comes from the same string constant.  If this isn't true, this
6258  * function may need to be rewritten to use strcmp() instead.
6259  */
6260 static struct kexinit_algorithm *ssh2_kexinit_addalg(struct kexinit_algorithm
6261                                                      *list, const char *name)
6262 {
6263     int i;
6264
6265     for (i = 0; i < MAXKEXLIST; i++)
6266         if (list[i].name == NULL || list[i].name == name) {
6267             list[i].name = name;
6268             return &list[i];
6269         }
6270     assert(!"No space in KEXINIT list");
6271     return NULL;
6272 }
6273
6274 /*
6275  * Handle the SSH-2 transport layer.
6276  */
6277 static void do_ssh2_transport(Ssh ssh, const void *vin, int inlen,
6278                              struct Packet *pktin)
6279 {
6280     const unsigned char *in = (const unsigned char *)vin;
6281     enum kexlist {
6282         KEXLIST_KEX, KEXLIST_HOSTKEY, KEXLIST_CSCIPHER, KEXLIST_SCCIPHER,
6283         KEXLIST_CSMAC, KEXLIST_SCMAC, KEXLIST_CSCOMP, KEXLIST_SCCOMP,
6284         NKEXLIST
6285     };
6286     const char * kexlist_descr[NKEXLIST] = {
6287         "key exchange algorithm", "host key algorithm",
6288         "client-to-server cipher", "server-to-client cipher",
6289         "client-to-server MAC", "server-to-client MAC",
6290         "client-to-server compression method",
6291         "server-to-client compression method" };
6292     struct do_ssh2_transport_state {
6293         int crLine;
6294         int nbits, pbits, warn_kex, warn_hk, warn_cscipher, warn_sccipher;
6295         Bignum p, g, e, f, K;
6296         void *our_kexinit;
6297         int our_kexinitlen;
6298         int kex_init_value, kex_reply_value;
6299         const struct ssh_mac *const *maclist;
6300         int nmacs;
6301         const struct ssh2_cipher *cscipher_tobe;
6302         const struct ssh2_cipher *sccipher_tobe;
6303         const struct ssh_mac *csmac_tobe;
6304         const struct ssh_mac *scmac_tobe;
6305         int csmac_etm_tobe, scmac_etm_tobe;
6306         const struct ssh_compress *cscomp_tobe;
6307         const struct ssh_compress *sccomp_tobe;
6308         char *hostkeydata, *sigdata, *rsakeydata, *keystr, *fingerprint;
6309         int hostkeylen, siglen, rsakeylen;
6310         void *hkey;                    /* actual host key */
6311         void *rsakey;                  /* for RSA kex */
6312         void *eckey;                   /* for ECDH kex */
6313         unsigned char exchange_hash[SSH2_KEX_MAX_HASH_LEN];
6314         int n_preferred_kex;
6315         const struct ssh_kexes *preferred_kex[KEX_MAX];
6316         int n_preferred_hk;
6317         int preferred_hk[HK_MAX];
6318         int n_preferred_ciphers;
6319         const struct ssh2_ciphers *preferred_ciphers[CIPHER_MAX];
6320         const struct ssh_compress *preferred_comp;
6321         int userauth_succeeded;     /* for delayed compression */
6322         int pending_compression;
6323         int got_session_id, activated_authconn;
6324         struct Packet *pktout;
6325         int dlgret;
6326         int guessok;
6327         int ignorepkt;
6328         struct kexinit_algorithm kexlists[NKEXLIST][MAXKEXLIST];
6329     };
6330     crState(do_ssh2_transport_state);
6331
6332     assert(!ssh->bare_connection);
6333     assert(ssh->version == 2);
6334
6335     crBeginState;
6336
6337     s->cscipher_tobe = s->sccipher_tobe = NULL;
6338     s->csmac_tobe = s->scmac_tobe = NULL;
6339     s->cscomp_tobe = s->sccomp_tobe = NULL;
6340
6341     s->got_session_id = s->activated_authconn = FALSE;
6342     s->userauth_succeeded = FALSE;
6343     s->pending_compression = FALSE;
6344
6345     /*
6346      * Be prepared to work around the buggy MAC problem.
6347      */
6348     if (ssh->remote_bugs & BUG_SSH2_HMAC)
6349         s->maclist = buggymacs, s->nmacs = lenof(buggymacs);
6350     else
6351         s->maclist = macs, s->nmacs = lenof(macs);
6352
6353   begin_key_exchange:
6354     ssh->pkt_kctx = SSH2_PKTCTX_NOKEX;
6355     {
6356         int i, j, k, warn;
6357         struct kexinit_algorithm *alg;
6358
6359         /*
6360          * Set up the preferred key exchange. (NULL => warn below here)
6361          */
6362         s->n_preferred_kex = 0;
6363         for (i = 0; i < KEX_MAX; i++) {
6364             switch (conf_get_int_int(ssh->conf, CONF_ssh_kexlist, i)) {
6365               case KEX_DHGEX:
6366                 s->preferred_kex[s->n_preferred_kex++] =
6367                     &ssh_diffiehellman_gex;
6368                 break;
6369               case KEX_DHGROUP14:
6370                 s->preferred_kex[s->n_preferred_kex++] =
6371                     &ssh_diffiehellman_group14;
6372                 break;
6373               case KEX_DHGROUP1:
6374                 s->preferred_kex[s->n_preferred_kex++] =
6375                     &ssh_diffiehellman_group1;
6376                 break;
6377               case KEX_RSA:
6378                 s->preferred_kex[s->n_preferred_kex++] =
6379                     &ssh_rsa_kex;
6380                 break;
6381               case KEX_ECDH:
6382                 s->preferred_kex[s->n_preferred_kex++] =
6383                     &ssh_ecdh_kex;
6384                 break;
6385               case KEX_WARN:
6386                 /* Flag for later. Don't bother if it's the last in
6387                  * the list. */
6388                 if (i < KEX_MAX - 1) {
6389                     s->preferred_kex[s->n_preferred_kex++] = NULL;
6390                 }
6391                 break;
6392             }
6393         }
6394
6395         /*
6396          * Set up the preferred host key types. These are just the ids
6397          * in the enum in putty.h, so 'warn below here' is indicated
6398          * by HK_WARN.
6399          */
6400         s->n_preferred_hk = 0;
6401         for (i = 0; i < HK_MAX; i++) {
6402             int id = conf_get_int_int(ssh->conf, CONF_ssh_hklist, i);
6403             /* As above, don't bother with HK_WARN if it's last in the
6404              * list */
6405             if (id != HK_WARN || i < HK_MAX - 1)
6406                 s->preferred_hk[s->n_preferred_hk++] = id;
6407         }
6408
6409         /*
6410          * Set up the preferred ciphers. (NULL => warn below here)
6411          */
6412         s->n_preferred_ciphers = 0;
6413         for (i = 0; i < CIPHER_MAX; i++) {
6414             switch (conf_get_int_int(ssh->conf, CONF_ssh_cipherlist, i)) {
6415               case CIPHER_BLOWFISH:
6416                 s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_blowfish;
6417                 break;
6418               case CIPHER_DES:
6419                 if (conf_get_int(ssh->conf, CONF_ssh2_des_cbc)) {
6420                     s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_des;
6421                 }
6422                 break;
6423               case CIPHER_3DES:
6424                 s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_3des;
6425                 break;
6426               case CIPHER_AES:
6427                 s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_aes;
6428                 break;
6429               case CIPHER_ARCFOUR:
6430                 s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_arcfour;
6431                 break;
6432               case CIPHER_CHACHA20:
6433                 s->preferred_ciphers[s->n_preferred_ciphers++] = &ssh2_ccp;
6434                 break;
6435               case CIPHER_WARN:
6436                 /* Flag for later. Don't bother if it's the last in
6437                  * the list. */
6438                 if (i < CIPHER_MAX - 1) {
6439                     s->preferred_ciphers[s->n_preferred_ciphers++] = NULL;
6440                 }
6441                 break;
6442             }
6443         }
6444
6445         /*
6446          * Set up preferred compression.
6447          */
6448         if (conf_get_int(ssh->conf, CONF_compression))
6449             s->preferred_comp = &ssh_zlib;
6450         else
6451             s->preferred_comp = &ssh_comp_none;
6452
6453         /*
6454          * Enable queueing of outgoing auth- or connection-layer
6455          * packets while we are in the middle of a key exchange.
6456          */
6457         ssh->queueing = TRUE;
6458
6459         /*
6460          * Flag that KEX is in progress.
6461          */
6462         ssh->kex_in_progress = TRUE;
6463
6464         for (i = 0; i < NKEXLIST; i++)
6465             for (j = 0; j < MAXKEXLIST; j++)
6466                 s->kexlists[i][j].name = NULL;
6467         /* List key exchange algorithms. */
6468         warn = FALSE;
6469         for (i = 0; i < s->n_preferred_kex; i++) {
6470             const struct ssh_kexes *k = s->preferred_kex[i];
6471             if (!k) warn = TRUE;
6472             else for (j = 0; j < k->nkexes; j++) {
6473                 alg = ssh2_kexinit_addalg(s->kexlists[KEXLIST_KEX],
6474                                           k->list[j]->name);
6475                 alg->u.kex.kex = k->list[j];
6476                 alg->u.kex.warn = warn;
6477             }
6478         }
6479         /* List server host key algorithms. */
6480         if (!s->got_session_id) {
6481             /*
6482              * In the first key exchange, we list all the algorithms
6483              * we're prepared to cope with, but prefer those algorithms
6484              * for which we have a host key for this host.
6485              *
6486              * If the host key algorithm is below the warning
6487              * threshold, we warn even if we did already have a key
6488              * for it, on the basis that if the user has just
6489              * reconfigured that host key type to be warned about,
6490              * they surely _do_ want to be alerted that a server
6491              * they're actually connecting to is using it.
6492              */
6493             warn = FALSE;
6494             for (i = 0; i < s->n_preferred_hk; i++) {
6495                 if (s->preferred_hk[i] == HK_WARN)
6496                     warn = TRUE;
6497                 for (j = 0; j < lenof(hostkey_algs); j++) {
6498                     if (hostkey_algs[j].id != s->preferred_hk[i])
6499                         continue;
6500                     if (have_ssh_host_key(ssh->savedhost, ssh->savedport,
6501                                           hostkey_algs[j].alg->keytype)) {
6502                         alg = ssh2_kexinit_addalg(s->kexlists[KEXLIST_HOSTKEY],
6503                                                   hostkey_algs[j].alg->name);
6504                         alg->u.hk.hostkey = hostkey_algs[j].alg;
6505                         alg->u.hk.warn = warn;
6506                     }
6507                 }
6508             }
6509             warn = FALSE;
6510             for (i = 0; i < s->n_preferred_hk; i++) {
6511                 if (s->preferred_hk[i] == HK_WARN)
6512                     warn = TRUE;
6513                 for (j = 0; j < lenof(hostkey_algs); j++) {
6514                     if (hostkey_algs[j].id != s->preferred_hk[i])
6515                         continue;
6516                     alg = ssh2_kexinit_addalg(s->kexlists[KEXLIST_HOSTKEY],
6517                                               hostkey_algs[j].alg->name);
6518                     alg->u.hk.hostkey = hostkey_algs[j].alg;
6519                     alg->u.hk.warn = warn;
6520                 }
6521             }
6522         } else {
6523             /*
6524              * In subsequent key exchanges, we list only the kex
6525              * algorithm that was selected in the first key exchange,
6526              * so that we keep getting the same host key and hence
6527              * don't have to interrupt the user's session to ask for
6528              * reverification.
6529              */
6530             assert(ssh->kex);
6531             alg = ssh2_kexinit_addalg(s->kexlists[KEXLIST_HOSTKEY],
6532                                       ssh->hostkey->name);
6533             alg->u.hk.hostkey = ssh->hostkey;
6534             alg->u.hk.warn = FALSE;
6535         }
6536         /* List encryption algorithms (client->server then server->client). */
6537         for (k = KEXLIST_CSCIPHER; k <= KEXLIST_SCCIPHER; k++) {
6538             warn = FALSE;
6539 #ifdef FUZZING
6540             alg = ssh2_kexinit_addalg(s->kexlists[k], "none");
6541             alg->u.cipher.cipher = NULL;
6542             alg->u.cipher.warn = warn;
6543 #endif /* FUZZING */
6544             for (i = 0; i < s->n_preferred_ciphers; i++) {
6545                 const struct ssh2_ciphers *c = s->preferred_ciphers[i];
6546                 if (!c) warn = TRUE;
6547                 else for (j = 0; j < c->nciphers; j++) {
6548                     alg = ssh2_kexinit_addalg(s->kexlists[k],
6549                                               c->list[j]->name);
6550                     alg->u.cipher.cipher = c->list[j];
6551                     alg->u.cipher.warn = warn;
6552                 }
6553             }
6554         }
6555         /* List MAC algorithms (client->server then server->client). */
6556         for (j = KEXLIST_CSMAC; j <= KEXLIST_SCMAC; j++) {
6557 #ifdef FUZZING
6558             alg = ssh2_kexinit_addalg(s->kexlists[j], "none");
6559             alg->u.mac.mac = NULL;
6560             alg->u.mac.etm = FALSE;
6561 #endif /* FUZZING */
6562             for (i = 0; i < s->nmacs; i++) {
6563                 alg = ssh2_kexinit_addalg(s->kexlists[j], s->maclist[i]->name);
6564                 alg->u.mac.mac = s->maclist[i];
6565                 alg->u.mac.etm = FALSE;
6566             }
6567             for (i = 0; i < s->nmacs; i++)
6568                 /* For each MAC, there may also be an ETM version,
6569                  * which we list second. */
6570                 if (s->maclist[i]->etm_name) {
6571                     alg = ssh2_kexinit_addalg(s->kexlists[j],
6572                                               s->maclist[i]->etm_name);
6573                     alg->u.mac.mac = s->maclist[i];
6574                     alg->u.mac.etm = TRUE;
6575                 }
6576         }
6577         /* List client->server compression algorithms,
6578          * then server->client compression algorithms. (We use the
6579          * same set twice.) */
6580         for (j = KEXLIST_CSCOMP; j <= KEXLIST_SCCOMP; j++) {
6581             assert(lenof(compressions) > 1);
6582             /* Prefer non-delayed versions */
6583             alg = ssh2_kexinit_addalg(s->kexlists[j], s->preferred_comp->name);
6584             alg->u.comp = s->preferred_comp;
6585             /* We don't even list delayed versions of algorithms until
6586              * they're allowed to be used, to avoid a race. See the end of
6587              * this function. */
6588             if (s->userauth_succeeded && s->preferred_comp->delayed_name) {
6589                 alg = ssh2_kexinit_addalg(s->kexlists[j],
6590                                           s->preferred_comp->delayed_name);
6591                 alg->u.comp = s->preferred_comp;
6592             }
6593             for (i = 0; i < lenof(compressions); i++) {
6594                 const struct ssh_compress *c = compressions[i];
6595                 alg = ssh2_kexinit_addalg(s->kexlists[j], c->name);
6596                 alg->u.comp = c;
6597                 if (s->userauth_succeeded && c->delayed_name) {
6598                     alg = ssh2_kexinit_addalg(s->kexlists[j], c->delayed_name);
6599                     alg->u.comp = c;
6600                 }
6601             }
6602         }
6603         /*
6604          * Construct and send our key exchange packet.
6605          */
6606         s->pktout = ssh2_pkt_init(SSH2_MSG_KEXINIT);
6607         for (i = 0; i < 16; i++)
6608             ssh2_pkt_addbyte(s->pktout, (unsigned char) random_byte());
6609         for (i = 0; i < NKEXLIST; i++) {
6610             ssh2_pkt_addstring_start(s->pktout);
6611             for (j = 0; j < MAXKEXLIST; j++) {
6612                 if (s->kexlists[i][j].name == NULL) break;
6613                 ssh2_pkt_addstring_commasep(s->pktout, s->kexlists[i][j].name);
6614             }
6615         }
6616         /* List client->server languages. Empty list. */
6617         ssh2_pkt_addstring_start(s->pktout);
6618         /* List server->client languages. Empty list. */
6619         ssh2_pkt_addstring_start(s->pktout);
6620         /* First KEX packet does _not_ follow, because we're not that brave. */
6621         ssh2_pkt_addbool(s->pktout, FALSE);
6622         /* Reserved. */
6623         ssh2_pkt_adduint32(s->pktout, 0);
6624     }
6625
6626     s->our_kexinitlen = s->pktout->length - 5;
6627     s->our_kexinit = snewn(s->our_kexinitlen, unsigned char);
6628     memcpy(s->our_kexinit, s->pktout->data + 5, s->our_kexinitlen); 
6629
6630     ssh2_pkt_send_noqueue(ssh, s->pktout);
6631
6632     if (!pktin)
6633         crWaitUntilV(pktin);
6634
6635     /*
6636      * Now examine the other side's KEXINIT to see what we're up
6637      * to.
6638      */
6639     {
6640         char *str;
6641         int i, j, len;
6642
6643         if (pktin->type != SSH2_MSG_KEXINIT) {
6644             bombout(("expected key exchange packet from server"));
6645             crStopV;
6646         }
6647         ssh->kex = NULL;
6648         ssh->hostkey = NULL;
6649         s->cscipher_tobe = NULL;
6650         s->sccipher_tobe = NULL;
6651         s->csmac_tobe = NULL;
6652         s->scmac_tobe = NULL;
6653         s->cscomp_tobe = NULL;
6654         s->sccomp_tobe = NULL;
6655         s->warn_kex = s->warn_hk = FALSE;
6656         s->warn_cscipher = s->warn_sccipher = FALSE;
6657
6658         pktin->savedpos += 16;          /* skip garbage cookie */
6659
6660         s->guessok = FALSE;
6661         for (i = 0; i < NKEXLIST; i++) {
6662             ssh_pkt_getstring(pktin, &str, &len);
6663             if (!str) {
6664                 bombout(("KEXINIT packet was incomplete"));
6665                 crStopV;
6666             }
6667
6668             /* If we've already selected a cipher which requires a
6669              * particular MAC, then just select that, and don't even
6670              * bother looking through the server's KEXINIT string for
6671              * MACs. */
6672             if (i == KEXLIST_CSMAC && s->cscipher_tobe &&
6673                 s->cscipher_tobe->required_mac) {
6674                 s->csmac_tobe = s->cscipher_tobe->required_mac;
6675                 s->csmac_etm_tobe = !!(s->csmac_tobe->etm_name);
6676                 goto matched;
6677             }
6678             if (i == KEXLIST_SCMAC && s->sccipher_tobe &&
6679                 s->sccipher_tobe->required_mac) {
6680                 s->scmac_tobe = s->sccipher_tobe->required_mac;
6681                 s->scmac_etm_tobe = !!(s->scmac_tobe->etm_name);
6682                 goto matched;
6683             }
6684
6685             for (j = 0; j < MAXKEXLIST; j++) {
6686                 struct kexinit_algorithm *alg = &s->kexlists[i][j];
6687                 if (alg->name == NULL) break;
6688                 if (in_commasep_string(alg->name, str, len)) {
6689                     /* We've found a matching algorithm. */
6690                     if (i == KEXLIST_KEX || i == KEXLIST_HOSTKEY) {
6691                         /* Check if we might need to ignore first kex pkt */
6692                         if (j != 0 ||
6693                             !first_in_commasep_string(alg->name, str, len))
6694                             s->guessok = FALSE;
6695                     }
6696                     if (i == KEXLIST_KEX) {
6697                         ssh->kex = alg->u.kex.kex;
6698                         s->warn_kex = alg->u.kex.warn;
6699                     } else if (i == KEXLIST_HOSTKEY) {
6700                         ssh->hostkey = alg->u.hk.hostkey;
6701                         s->warn_hk = alg->u.hk.warn;
6702                     } else if (i == KEXLIST_CSCIPHER) {
6703                         s->cscipher_tobe = alg->u.cipher.cipher;
6704                         s->warn_cscipher = alg->u.cipher.warn;
6705                     } else if (i == KEXLIST_SCCIPHER) {
6706                         s->sccipher_tobe = alg->u.cipher.cipher;
6707                         s->warn_sccipher = alg->u.cipher.warn;
6708                     } else if (i == KEXLIST_CSMAC) {
6709                         s->csmac_tobe = alg->u.mac.mac;
6710                         s->csmac_etm_tobe = alg->u.mac.etm;
6711                     } else if (i == KEXLIST_SCMAC) {
6712                         s->scmac_tobe = alg->u.mac.mac;
6713                         s->scmac_etm_tobe = alg->u.mac.etm;
6714                     } else if (i == KEXLIST_CSCOMP) {
6715                         s->cscomp_tobe = alg->u.comp;
6716                     } else if (i == KEXLIST_SCCOMP) {
6717                         s->sccomp_tobe = alg->u.comp;
6718                     }
6719                     goto matched;
6720                 }
6721                 if ((i == KEXLIST_CSCOMP || i == KEXLIST_SCCOMP) &&
6722                     in_commasep_string(alg->u.comp->delayed_name, str, len))
6723                     s->pending_compression = TRUE;  /* try this later */
6724             }
6725             bombout(("Couldn't agree a %s (available: %.*s)",
6726                      kexlist_descr[i], len, str));
6727             crStopV;
6728           matched:;
6729
6730             if (i == KEXLIST_HOSTKEY) {
6731                 int j;
6732
6733                 /*
6734                  * In addition to deciding which host key we're
6735                  * actually going to use, we should make a list of the
6736                  * host keys offered by the server which we _don't_
6737                  * have cached. These will be offered as cross-
6738                  * certification options by ssh_get_specials.
6739                  *
6740                  * We also count the key we're currently using for KEX
6741                  * as one we've already got, because by the time this
6742                  * menu becomes visible, it will be.
6743                  */
6744                 ssh->n_uncert_hostkeys = 0;
6745
6746                 for (j = 0; j < lenof(hostkey_algs); j++) {
6747                     if (hostkey_algs[j].alg != ssh->hostkey &&
6748                         in_commasep_string(hostkey_algs[j].alg->name,
6749                                            str, len) &&
6750                         !have_ssh_host_key(ssh->savedhost, ssh->savedport,
6751                                            hostkey_algs[j].alg->keytype)) {
6752                         ssh->uncert_hostkeys[ssh->n_uncert_hostkeys++] = j;
6753                     }
6754                 }
6755             }
6756         }
6757
6758         if (s->pending_compression) {
6759             logevent("Server supports delayed compression; "
6760                      "will try this later");
6761         }
6762         ssh_pkt_getstring(pktin, &str, &len);  /* client->server language */
6763         ssh_pkt_getstring(pktin, &str, &len);  /* server->client language */
6764         s->ignorepkt = ssh2_pkt_getbool(pktin) && !s->guessok;
6765
6766         ssh->exhash = ssh->kex->hash->init();
6767         hash_string(ssh->kex->hash, ssh->exhash, ssh->v_c, strlen(ssh->v_c));
6768         hash_string(ssh->kex->hash, ssh->exhash, ssh->v_s, strlen(ssh->v_s));
6769         hash_string(ssh->kex->hash, ssh->exhash,
6770             s->our_kexinit, s->our_kexinitlen);
6771         sfree(s->our_kexinit);
6772         /* Include the type byte in the hash of server's KEXINIT */
6773         hash_string(ssh->kex->hash, ssh->exhash,
6774                     pktin->body - 1, pktin->length + 1);
6775
6776         if (s->warn_kex) {
6777             ssh_set_frozen(ssh, 1);
6778             s->dlgret = askalg(ssh->frontend, "key-exchange algorithm",
6779                                ssh->kex->name,
6780                                ssh_dialog_callback, ssh);
6781             if (s->dlgret < 0) {
6782                 do {
6783                     crReturnV;
6784                     if (pktin) {
6785                         bombout(("Unexpected data from server while"
6786                                  " waiting for user response"));
6787                         crStopV;
6788                     }
6789                 } while (pktin || inlen > 0);
6790                 s->dlgret = ssh->user_response;
6791             }
6792             ssh_set_frozen(ssh, 0);
6793             if (s->dlgret == 0) {
6794                 ssh_disconnect(ssh, "User aborted at kex warning", NULL,
6795                                0, TRUE);
6796                 crStopV;
6797             }
6798         }
6799
6800         if (s->warn_hk) {
6801             int j, k;
6802             char *betteralgs;
6803
6804             ssh_set_frozen(ssh, 1);
6805
6806             /*
6807              * Change warning box wording depending on why we chose a
6808              * warning-level host key algorithm. If it's because
6809              * that's all we have *cached*, use the askhk mechanism,
6810              * and list the host keys we could usefully cross-certify.
6811              * Otherwise, use askalg for the standard wording.
6812              */
6813             betteralgs = NULL;
6814             for (j = 0; j < ssh->n_uncert_hostkeys; j++) {
6815                 const struct ssh_signkey_with_user_pref_id *hktype =
6816                     &hostkey_algs[ssh->uncert_hostkeys[j]];
6817                 int better = FALSE;
6818                 for (k = 0; k < HK_MAX; k++) {
6819                     int id = conf_get_int_int(ssh->conf, CONF_ssh_hklist, k);
6820                     if (id == HK_WARN) {
6821                         break;
6822                     } else if (id == hktype->id) {
6823                         better = TRUE;
6824                         break;
6825                     }
6826                 }
6827                 if (better) {
6828                     if (betteralgs) {
6829                         char *old_ba = betteralgs;
6830                         betteralgs = dupcat(betteralgs, ",",
6831                                             hktype->alg->name,
6832                                             (const char *)NULL);
6833                         sfree(old_ba);
6834                     } else {
6835                         betteralgs = dupstr(hktype->alg->name);
6836                     }
6837                 }
6838             }
6839             if (betteralgs) {
6840                 s->dlgret = askhk(ssh->frontend, ssh->hostkey->name,
6841                                   betteralgs, ssh_dialog_callback, ssh);
6842                 sfree(betteralgs);
6843             } else {
6844                 s->dlgret = askalg(ssh->frontend, "host key type",
6845                                    ssh->hostkey->name,
6846                                    ssh_dialog_callback, ssh);
6847             }
6848             if (s->dlgret < 0) {
6849                 do {
6850                     crReturnV;
6851                     if (pktin) {
6852                         bombout(("Unexpected data from server while"
6853                                  " waiting for user response"));
6854                         crStopV;
6855                     }
6856                 } while (pktin || inlen > 0);
6857                 s->dlgret = ssh->user_response;
6858             }
6859             ssh_set_frozen(ssh, 0);
6860             if (s->dlgret == 0) {
6861                 ssh_disconnect(ssh, "User aborted at host key warning", NULL,
6862                                0, TRUE);
6863                 crStopV;
6864             }
6865         }
6866
6867         if (s->warn_cscipher) {
6868             ssh_set_frozen(ssh, 1);
6869             s->dlgret = askalg(ssh->frontend,
6870                                "client-to-server cipher",
6871                                s->cscipher_tobe->name,
6872                                ssh_dialog_callback, ssh);
6873             if (s->dlgret < 0) {
6874                 do {
6875                     crReturnV;
6876                     if (pktin) {
6877                         bombout(("Unexpected data from server while"
6878                                  " waiting for user response"));
6879                         crStopV;
6880                     }
6881                 } while (pktin || inlen > 0);
6882                 s->dlgret = ssh->user_response;
6883             }
6884             ssh_set_frozen(ssh, 0);
6885             if (s->dlgret == 0) {
6886                 ssh_disconnect(ssh, "User aborted at cipher warning", NULL,
6887                                0, TRUE);
6888                 crStopV;
6889             }
6890         }
6891
6892         if (s->warn_sccipher) {
6893             ssh_set_frozen(ssh, 1);
6894             s->dlgret = askalg(ssh->frontend,
6895                                "server-to-client cipher",
6896                                s->sccipher_tobe->name,
6897                                ssh_dialog_callback, ssh);
6898             if (s->dlgret < 0) {
6899                 do {
6900                     crReturnV;
6901                     if (pktin) {
6902                         bombout(("Unexpected data from server while"
6903                                  " waiting for user response"));
6904                         crStopV;
6905                     }
6906                 } while (pktin || inlen > 0);
6907                 s->dlgret = ssh->user_response;
6908             }
6909             ssh_set_frozen(ssh, 0);
6910             if (s->dlgret == 0) {
6911                 ssh_disconnect(ssh, "User aborted at cipher warning", NULL,
6912                                0, TRUE);
6913                 crStopV;
6914             }
6915         }
6916
6917         if (s->ignorepkt) /* first_kex_packet_follows */
6918             crWaitUntilV(pktin);                /* Ignore packet */
6919     }
6920
6921     if (ssh->kex->main_type == KEXTYPE_DH) {
6922         /*
6923          * Work out the number of bits of key we will need from the
6924          * key exchange. We start with the maximum key length of
6925          * either cipher...
6926          */
6927         {
6928             int csbits, scbits;
6929
6930             csbits = s->cscipher_tobe ? s->cscipher_tobe->real_keybits : 0;
6931             scbits = s->sccipher_tobe ? s->sccipher_tobe->real_keybits : 0;
6932             s->nbits = (csbits > scbits ? csbits : scbits);
6933         }
6934         /* The keys only have hlen-bit entropy, since they're based on
6935          * a hash. So cap the key size at hlen bits. */
6936         if (s->nbits > ssh->kex->hash->hlen * 8)
6937             s->nbits = ssh->kex->hash->hlen * 8;
6938
6939         /*
6940          * If we're doing Diffie-Hellman group exchange, start by
6941          * requesting a group.
6942          */
6943         if (dh_is_gex(ssh->kex)) {
6944             logevent("Doing Diffie-Hellman group exchange");
6945             ssh->pkt_kctx = SSH2_PKTCTX_DHGEX;
6946             /*
6947              * Work out how big a DH group we will need to allow that
6948              * much data.
6949              */
6950             s->pbits = 512 << ((s->nbits - 1) / 64);
6951             if (s->pbits < DH_MIN_SIZE)
6952                 s->pbits = DH_MIN_SIZE;
6953             if (s->pbits > DH_MAX_SIZE)
6954                 s->pbits = DH_MAX_SIZE;
6955             if ((ssh->remote_bugs & BUG_SSH2_OLDGEX)) {
6956                 s->pktout = ssh2_pkt_init(SSH2_MSG_KEX_DH_GEX_REQUEST_OLD);
6957                 ssh2_pkt_adduint32(s->pktout, s->pbits);
6958             } else {
6959                 s->pktout = ssh2_pkt_init(SSH2_MSG_KEX_DH_GEX_REQUEST);
6960                 ssh2_pkt_adduint32(s->pktout, DH_MIN_SIZE);
6961                 ssh2_pkt_adduint32(s->pktout, s->pbits);
6962                 ssh2_pkt_adduint32(s->pktout, DH_MAX_SIZE);
6963             }
6964             ssh2_pkt_send_noqueue(ssh, s->pktout);
6965
6966             crWaitUntilV(pktin);
6967             if (pktin->type != SSH2_MSG_KEX_DH_GEX_GROUP) {
6968                 bombout(("expected key exchange group packet from server"));
6969                 crStopV;
6970             }
6971             s->p = ssh2_pkt_getmp(pktin);
6972             s->g = ssh2_pkt_getmp(pktin);
6973             if (!s->p || !s->g) {
6974                 bombout(("unable to read mp-ints from incoming group packet"));
6975                 crStopV;
6976             }
6977             ssh->kex_ctx = dh_setup_gex(s->p, s->g);
6978             s->kex_init_value = SSH2_MSG_KEX_DH_GEX_INIT;
6979             s->kex_reply_value = SSH2_MSG_KEX_DH_GEX_REPLY;
6980         } else {
6981             ssh->pkt_kctx = SSH2_PKTCTX_DHGROUP;
6982             ssh->kex_ctx = dh_setup_group(ssh->kex);
6983             s->kex_init_value = SSH2_MSG_KEXDH_INIT;
6984             s->kex_reply_value = SSH2_MSG_KEXDH_REPLY;
6985             logeventf(ssh, "Using Diffie-Hellman with standard group \"%s\"",
6986                       ssh->kex->groupname);
6987         }
6988
6989         logeventf(ssh, "Doing Diffie-Hellman key exchange with hash %s",
6990                   ssh->kex->hash->text_name);
6991         /*
6992          * Now generate and send e for Diffie-Hellman.
6993          */
6994         set_busy_status(ssh->frontend, BUSY_CPU); /* this can take a while */
6995         s->e = dh_create_e(ssh->kex_ctx, s->nbits * 2);
6996         s->pktout = ssh2_pkt_init(s->kex_init_value);
6997         ssh2_pkt_addmp(s->pktout, s->e);
6998         ssh2_pkt_send_noqueue(ssh, s->pktout);
6999
7000         set_busy_status(ssh->frontend, BUSY_WAITING); /* wait for server */
7001         crWaitUntilV(pktin);
7002         if (pktin->type != s->kex_reply_value) {
7003             bombout(("expected key exchange reply packet from server"));
7004             crStopV;
7005         }
7006         set_busy_status(ssh->frontend, BUSY_CPU); /* cogitate */
7007         ssh_pkt_getstring(pktin, &s->hostkeydata, &s->hostkeylen);
7008         if (!s->hostkeydata) {
7009             bombout(("unable to parse key exchange reply packet"));
7010             crStopV;
7011         }
7012         s->hkey = ssh->hostkey->newkey(ssh->hostkey,
7013                                        s->hostkeydata, s->hostkeylen);
7014         s->f = ssh2_pkt_getmp(pktin);
7015         if (!s->f) {
7016             bombout(("unable to parse key exchange reply packet"));
7017             crStopV;
7018         }
7019         ssh_pkt_getstring(pktin, &s->sigdata, &s->siglen);
7020         if (!s->sigdata) {
7021             bombout(("unable to parse key exchange reply packet"));
7022             crStopV;
7023         }
7024
7025         {
7026             const char *err = dh_validate_f(ssh->kex_ctx, s->f);
7027             if (err) {
7028                 bombout(("key exchange reply failed validation: %s", err));
7029                 crStopV;
7030             }
7031         }
7032         s->K = dh_find_K(ssh->kex_ctx, s->f);
7033
7034         /* We assume everything from now on will be quick, and it might
7035          * involve user interaction. */
7036         set_busy_status(ssh->frontend, BUSY_NOT);
7037
7038         hash_string(ssh->kex->hash, ssh->exhash, s->hostkeydata, s->hostkeylen);
7039         if (dh_is_gex(ssh->kex)) {
7040             if (!(ssh->remote_bugs & BUG_SSH2_OLDGEX))
7041                 hash_uint32(ssh->kex->hash, ssh->exhash, DH_MIN_SIZE);
7042             hash_uint32(ssh->kex->hash, ssh->exhash, s->pbits);
7043             if (!(ssh->remote_bugs & BUG_SSH2_OLDGEX))
7044                 hash_uint32(ssh->kex->hash, ssh->exhash, DH_MAX_SIZE);
7045             hash_mpint(ssh->kex->hash, ssh->exhash, s->p);
7046             hash_mpint(ssh->kex->hash, ssh->exhash, s->g);
7047         }
7048         hash_mpint(ssh->kex->hash, ssh->exhash, s->e);
7049         hash_mpint(ssh->kex->hash, ssh->exhash, s->f);
7050
7051         dh_cleanup(ssh->kex_ctx);
7052         freebn(s->f);
7053         if (dh_is_gex(ssh->kex)) {
7054             freebn(s->g);
7055             freebn(s->p);
7056         }
7057     } else if (ssh->kex->main_type == KEXTYPE_ECDH) {
7058
7059         logeventf(ssh, "Doing ECDH key exchange with curve %s and hash %s",
7060                   ssh_ecdhkex_curve_textname(ssh->kex),
7061                   ssh->kex->hash->text_name);
7062         ssh->pkt_kctx = SSH2_PKTCTX_ECDHKEX;
7063
7064         s->eckey = ssh_ecdhkex_newkey(ssh->kex);
7065         if (!s->eckey) {
7066             bombout(("Unable to generate key for ECDH"));
7067             crStopV;
7068         }
7069
7070         {
7071             char *publicPoint;
7072             int publicPointLength;
7073             publicPoint = ssh_ecdhkex_getpublic(s->eckey, &publicPointLength);
7074             if (!publicPoint) {
7075                 ssh_ecdhkex_freekey(s->eckey);
7076                 bombout(("Unable to encode public key for ECDH"));
7077                 crStopV;
7078             }
7079             s->pktout = ssh2_pkt_init(SSH2_MSG_KEX_ECDH_INIT);
7080             ssh2_pkt_addstring_start(s->pktout);
7081             ssh2_pkt_addstring_data(s->pktout, publicPoint, publicPointLength);
7082             sfree(publicPoint);
7083         }
7084
7085         ssh2_pkt_send_noqueue(ssh, s->pktout);
7086
7087         crWaitUntilV(pktin);
7088         if (pktin->type != SSH2_MSG_KEX_ECDH_REPLY) {
7089             ssh_ecdhkex_freekey(s->eckey);
7090             bombout(("expected ECDH reply packet from server"));
7091             crStopV;
7092         }
7093
7094         ssh_pkt_getstring(pktin, &s->hostkeydata, &s->hostkeylen);
7095         if (!s->hostkeydata) {
7096             bombout(("unable to parse ECDH reply packet"));
7097             crStopV;
7098         }
7099         hash_string(ssh->kex->hash, ssh->exhash, s->hostkeydata, s->hostkeylen);
7100         s->hkey = ssh->hostkey->newkey(ssh->hostkey,
7101                                        s->hostkeydata, s->hostkeylen);
7102
7103         {
7104             char *publicPoint;
7105             int publicPointLength;
7106             publicPoint = ssh_ecdhkex_getpublic(s->eckey, &publicPointLength);
7107             if (!publicPoint) {
7108                 ssh_ecdhkex_freekey(s->eckey);
7109                 bombout(("Unable to encode public key for ECDH hash"));
7110                 crStopV;
7111             }
7112             hash_string(ssh->kex->hash, ssh->exhash,
7113                         publicPoint, publicPointLength);
7114             sfree(publicPoint);
7115         }
7116
7117         {
7118             char *keydata;
7119             int keylen;
7120             ssh_pkt_getstring(pktin, &keydata, &keylen);
7121             if (!keydata) {
7122                 bombout(("unable to parse ECDH reply packet"));
7123                 crStopV;
7124             }
7125             hash_string(ssh->kex->hash, ssh->exhash, keydata, keylen);
7126             s->K = ssh_ecdhkex_getkey(s->eckey, keydata, keylen);
7127             if (!s->K) {
7128                 ssh_ecdhkex_freekey(s->eckey);
7129                 bombout(("point received in ECDH was not valid"));
7130                 crStopV;
7131             }
7132         }
7133
7134         ssh_pkt_getstring(pktin, &s->sigdata, &s->siglen);
7135         if (!s->sigdata) {
7136             bombout(("unable to parse key exchange reply packet"));
7137             crStopV;
7138         }
7139
7140         ssh_ecdhkex_freekey(s->eckey);
7141     } else {
7142         logeventf(ssh, "Doing RSA key exchange with hash %s",
7143                   ssh->kex->hash->text_name);
7144         ssh->pkt_kctx = SSH2_PKTCTX_RSAKEX;
7145         /*
7146          * RSA key exchange. First expect a KEXRSA_PUBKEY packet
7147          * from the server.
7148          */
7149         crWaitUntilV(pktin);
7150         if (pktin->type != SSH2_MSG_KEXRSA_PUBKEY) {
7151             bombout(("expected RSA public key packet from server"));
7152             crStopV;
7153         }
7154
7155         ssh_pkt_getstring(pktin, &s->hostkeydata, &s->hostkeylen);
7156         if (!s->hostkeydata) {
7157             bombout(("unable to parse RSA public key packet"));
7158             crStopV;
7159         }
7160         hash_string(ssh->kex->hash, ssh->exhash,
7161                     s->hostkeydata, s->hostkeylen);
7162         s->hkey = ssh->hostkey->newkey(ssh->hostkey,
7163                                        s->hostkeydata, s->hostkeylen);
7164
7165         {
7166             char *keydata;
7167             ssh_pkt_getstring(pktin, &keydata, &s->rsakeylen);
7168             if (!keydata) {
7169                 bombout(("unable to parse RSA public key packet"));
7170                 crStopV;
7171             }
7172             s->rsakeydata = snewn(s->rsakeylen, char);
7173             memcpy(s->rsakeydata, keydata, s->rsakeylen);
7174         }
7175
7176         s->rsakey = ssh_rsakex_newkey(s->rsakeydata, s->rsakeylen);
7177         if (!s->rsakey) {
7178             sfree(s->rsakeydata);
7179             bombout(("unable to parse RSA public key from server"));
7180             crStopV;
7181         }
7182
7183         hash_string(ssh->kex->hash, ssh->exhash, s->rsakeydata, s->rsakeylen);
7184
7185         /*
7186          * Next, set up a shared secret K, of precisely KLEN -
7187          * 2*HLEN - 49 bits, where KLEN is the bit length of the
7188          * RSA key modulus and HLEN is the bit length of the hash
7189          * we're using.
7190          */
7191         {
7192             int klen = ssh_rsakex_klen(s->rsakey);
7193             int nbits = klen - (2*ssh->kex->hash->hlen*8 + 49);
7194             int i, byte = 0;
7195             unsigned char *kstr1, *kstr2, *outstr;
7196             int kstr1len, kstr2len, outstrlen;
7197
7198             s->K = bn_power_2(nbits - 1);
7199
7200             for (i = 0; i < nbits; i++) {
7201                 if ((i & 7) == 0) {
7202                     byte = random_byte();
7203                 }
7204                 bignum_set_bit(s->K, i, (byte >> (i & 7)) & 1);
7205             }
7206
7207             /*
7208              * Encode this as an mpint.
7209              */
7210             kstr1 = ssh2_mpint_fmt(s->K, &kstr1len);
7211             kstr2 = snewn(kstr2len = 4 + kstr1len, unsigned char);
7212             PUT_32BIT(kstr2, kstr1len);
7213             memcpy(kstr2 + 4, kstr1, kstr1len);
7214
7215             /*
7216              * Encrypt it with the given RSA key.
7217              */
7218             outstrlen = (klen + 7) / 8;
7219             outstr = snewn(outstrlen, unsigned char);
7220             ssh_rsakex_encrypt(ssh->kex->hash, kstr2, kstr2len,
7221                                outstr, outstrlen, s->rsakey);
7222
7223             /*
7224              * And send it off in a return packet.
7225              */
7226             s->pktout = ssh2_pkt_init(SSH2_MSG_KEXRSA_SECRET);
7227             ssh2_pkt_addstring_start(s->pktout);
7228             ssh2_pkt_addstring_data(s->pktout, (char *)outstr, outstrlen);
7229             ssh2_pkt_send_noqueue(ssh, s->pktout);
7230
7231             hash_string(ssh->kex->hash, ssh->exhash, outstr, outstrlen);
7232
7233             sfree(kstr2);
7234             sfree(kstr1);
7235             sfree(outstr);
7236         }
7237
7238         ssh_rsakex_freekey(s->rsakey);
7239
7240         crWaitUntilV(pktin);
7241         if (pktin->type != SSH2_MSG_KEXRSA_DONE) {
7242             sfree(s->rsakeydata);
7243             bombout(("expected signature packet from server"));
7244             crStopV;
7245         }
7246
7247         ssh_pkt_getstring(pktin, &s->sigdata, &s->siglen);
7248         if (!s->sigdata) {
7249             bombout(("unable to parse signature packet"));
7250             crStopV;
7251         }
7252
7253         sfree(s->rsakeydata);
7254     }
7255
7256     hash_mpint(ssh->kex->hash, ssh->exhash, s->K);
7257     assert(ssh->kex->hash->hlen <= sizeof(s->exchange_hash));
7258     ssh->kex->hash->final(ssh->exhash, s->exchange_hash);
7259
7260     ssh->kex_ctx = NULL;
7261
7262 #if 0
7263     debug(("Exchange hash is:\n"));
7264     dmemdump(s->exchange_hash, ssh->kex->hash->hlen);
7265 #endif
7266
7267     if (!s->hkey) {
7268         bombout(("Server's host key is invalid"));
7269         crStopV;
7270     }
7271
7272     if (!ssh->hostkey->verifysig(s->hkey, s->sigdata, s->siglen,
7273                                  (char *)s->exchange_hash,
7274                                  ssh->kex->hash->hlen)) {
7275 #ifndef FUZZING
7276         bombout(("Server's host key did not match the signature supplied"));
7277         crStopV;
7278 #endif
7279     }
7280
7281     s->keystr = ssh->hostkey->fmtkey(s->hkey);
7282     if (!s->got_session_id) {
7283         /*
7284          * Make a note of any other host key formats that are available.
7285          */
7286         {
7287             int i, j, nkeys = 0;
7288             char *list = NULL;
7289             for (i = 0; i < lenof(hostkey_algs); i++) {
7290                 if (hostkey_algs[i].alg == ssh->hostkey)
7291                     continue;
7292
7293                 for (j = 0; j < ssh->n_uncert_hostkeys; j++)
7294                     if (ssh->uncert_hostkeys[j] == i)
7295                         break;
7296
7297                 if (j < ssh->n_uncert_hostkeys) {
7298                     char *newlist;
7299                     if (list)
7300                         newlist = dupprintf("%s/%s", list,
7301                                             hostkey_algs[i].alg->name);
7302                     else
7303                         newlist = dupprintf("%s", hostkey_algs[i].alg->name);
7304                     sfree(list);
7305                     list = newlist;
7306                     nkeys++;
7307                 }
7308             }
7309             if (list) {
7310                 logeventf(ssh,
7311                           "Server also has %s host key%s, but we "
7312                           "don't know %s", list,
7313                           nkeys > 1 ? "s" : "",
7314                           nkeys > 1 ? "any of them" : "it");
7315                 sfree(list);
7316             }
7317         }
7318
7319         /*
7320          * Authenticate remote host: verify host key. (We've already
7321          * checked the signature of the exchange hash.)
7322          */
7323         s->fingerprint = ssh2_fingerprint(ssh->hostkey, s->hkey);
7324         logevent("Host key fingerprint is:");
7325         logevent(s->fingerprint);
7326         /* First check against manually configured host keys. */
7327         s->dlgret = verify_ssh_manual_host_key(ssh, s->fingerprint,
7328                                                ssh->hostkey, s->hkey);
7329         if (s->dlgret == 0) {          /* did not match */
7330             bombout(("Host key did not appear in manually configured list"));
7331             crStopV;
7332         } else if (s->dlgret < 0) { /* none configured; use standard handling */
7333             ssh_set_frozen(ssh, 1);
7334             s->dlgret = verify_ssh_host_key(ssh->frontend,
7335                                             ssh->savedhost, ssh->savedport,
7336                                             ssh->hostkey->keytype, s->keystr,
7337                                             s->fingerprint,
7338                                             ssh_dialog_callback, ssh);
7339 #ifdef FUZZING
7340             s->dlgret = 1;
7341 #endif
7342             if (s->dlgret < 0) {
7343                 do {
7344                     crReturnV;
7345                     if (pktin) {
7346                         bombout(("Unexpected data from server while waiting"
7347                                  " for user host key response"));
7348                         crStopV;
7349                     }
7350                 } while (pktin || inlen > 0);
7351                 s->dlgret = ssh->user_response;
7352             }
7353             ssh_set_frozen(ssh, 0);
7354             if (s->dlgret == 0) {
7355                 ssh_disconnect(ssh, "Aborted at host key verification", NULL,
7356                                0, TRUE);
7357                 crStopV;
7358             }
7359         }
7360         sfree(s->fingerprint);
7361         /*
7362          * Save this host key, to check against the one presented in
7363          * subsequent rekeys.
7364          */
7365         ssh->hostkey_str = s->keystr;
7366     } else if (ssh->cross_certifying) {
7367         s->fingerprint = ssh2_fingerprint(ssh->hostkey, s->hkey);
7368         logevent("Storing additional host key for this host:");
7369         logevent(s->fingerprint);
7370         store_host_key(ssh->savedhost, ssh->savedport,
7371                        ssh->hostkey->keytype, s->keystr);
7372         ssh->cross_certifying = FALSE;
7373         /*
7374          * Don't forget to store the new key as the one we'll be
7375          * re-checking in future normal rekeys.
7376          */
7377         ssh->hostkey_str = s->keystr;
7378     } else {
7379         /*
7380          * In a rekey, we never present an interactive host key
7381          * verification request to the user. Instead, we simply
7382          * enforce that the key we're seeing this time is identical to
7383          * the one we saw before.
7384          */
7385         if (strcmp(ssh->hostkey_str, s->keystr)) {
7386 #ifndef FUZZING
7387             bombout(("Host key was different in repeat key exchange"));
7388             crStopV;
7389 #endif
7390         }
7391         sfree(s->keystr);
7392     }
7393     ssh->hostkey->freekey(s->hkey);
7394
7395     /*
7396      * The exchange hash from the very first key exchange is also
7397      * the session id, used in session key construction and
7398      * authentication.
7399      */
7400     if (!s->got_session_id) {
7401         assert(sizeof(s->exchange_hash) <= sizeof(ssh->v2_session_id));
7402         memcpy(ssh->v2_session_id, s->exchange_hash,
7403                sizeof(s->exchange_hash));
7404         ssh->v2_session_id_len = ssh->kex->hash->hlen;
7405         assert(ssh->v2_session_id_len <= sizeof(ssh->v2_session_id));
7406         s->got_session_id = TRUE;
7407     }
7408
7409     /*
7410      * Send SSH2_MSG_NEWKEYS.
7411      */
7412     s->pktout = ssh2_pkt_init(SSH2_MSG_NEWKEYS);
7413     ssh2_pkt_send_noqueue(ssh, s->pktout);
7414     ssh->outgoing_data_size = 0;       /* start counting from here */
7415
7416     /*
7417      * We've sent client NEWKEYS, so create and initialise
7418      * client-to-server session keys.
7419      */
7420     if (ssh->cs_cipher_ctx)
7421         ssh->cscipher->free_context(ssh->cs_cipher_ctx);
7422     ssh->cscipher = s->cscipher_tobe;
7423     if (ssh->cscipher) ssh->cs_cipher_ctx = ssh->cscipher->make_context();
7424
7425     if (ssh->cs_mac_ctx)
7426         ssh->csmac->free_context(ssh->cs_mac_ctx);
7427     ssh->csmac = s->csmac_tobe;
7428     ssh->csmac_etm = s->csmac_etm_tobe;
7429     if (ssh->csmac)
7430         ssh->cs_mac_ctx = ssh->csmac->make_context(ssh->cs_cipher_ctx);
7431
7432     if (ssh->cs_comp_ctx)
7433         ssh->cscomp->compress_cleanup(ssh->cs_comp_ctx);
7434     ssh->cscomp = s->cscomp_tobe;
7435     ssh->cs_comp_ctx = ssh->cscomp->compress_init();
7436
7437     /*
7438      * Set IVs on client-to-server keys. Here we use the exchange
7439      * hash from the _first_ key exchange.
7440      */
7441     if (ssh->cscipher) {
7442         unsigned char *key;
7443
7444         key = ssh2_mkkey(ssh, s->K, s->exchange_hash, 'C',
7445                          ssh->cscipher->padded_keybytes);
7446         ssh->cscipher->setkey(ssh->cs_cipher_ctx, key);
7447         smemclr(key, ssh->cscipher->padded_keybytes);
7448         sfree(key);
7449
7450         key = ssh2_mkkey(ssh, s->K, s->exchange_hash, 'A',
7451                          ssh->cscipher->blksize);
7452         ssh->cscipher->setiv(ssh->cs_cipher_ctx, key);
7453         smemclr(key, ssh->cscipher->blksize);
7454         sfree(key);
7455     }
7456     if (ssh->csmac) {
7457         unsigned char *key;
7458
7459         key = ssh2_mkkey(ssh, s->K, s->exchange_hash, 'E',
7460                          ssh->csmac->keylen);
7461         ssh->csmac->setkey(ssh->cs_mac_ctx, key);
7462         smemclr(key, ssh->csmac->keylen);
7463         sfree(key);
7464     }
7465
7466     if (ssh->cscipher)
7467         logeventf(ssh, "Initialised %.200s client->server encryption",
7468                   ssh->cscipher->text_name);
7469     if (ssh->csmac)
7470         logeventf(ssh, "Initialised %.200s client->server MAC algorithm%s%s",
7471                   ssh->csmac->text_name,
7472                   ssh->csmac_etm ? " (in ETM mode)" : "",
7473                   ssh->cscipher->required_mac ? " (required by cipher)" : "");
7474     if (ssh->cscomp->text_name)
7475         logeventf(ssh, "Initialised %s compression",
7476                   ssh->cscomp->text_name);
7477
7478     /*
7479      * Now our end of the key exchange is complete, we can send all
7480      * our queued higher-layer packets.
7481      */
7482     ssh->queueing = FALSE;
7483     ssh2_pkt_queuesend(ssh);
7484
7485     /*
7486      * Expect SSH2_MSG_NEWKEYS from server.
7487      */
7488     crWaitUntilV(pktin);
7489     if (pktin->type != SSH2_MSG_NEWKEYS) {
7490         bombout(("expected new-keys packet from server"));
7491         crStopV;
7492     }
7493     ssh->incoming_data_size = 0;       /* start counting from here */
7494
7495     /*
7496      * We've seen server NEWKEYS, so create and initialise
7497      * server-to-client session keys.
7498      */
7499     if (ssh->sc_cipher_ctx)
7500         ssh->sccipher->free_context(ssh->sc_cipher_ctx);
7501     if (s->sccipher_tobe) {
7502         ssh->sccipher = s->sccipher_tobe;
7503         ssh->sc_cipher_ctx = ssh->sccipher->make_context();
7504     }
7505
7506     if (ssh->sc_mac_ctx)
7507         ssh->scmac->free_context(ssh->sc_mac_ctx);
7508     if (s->scmac_tobe) {
7509         ssh->scmac = s->scmac_tobe;
7510         ssh->scmac_etm = s->scmac_etm_tobe;
7511         ssh->sc_mac_ctx = ssh->scmac->make_context(ssh->sc_cipher_ctx);
7512     }
7513
7514     if (ssh->sc_comp_ctx)
7515         ssh->sccomp->decompress_cleanup(ssh->sc_comp_ctx);
7516     ssh->sccomp = s->sccomp_tobe;
7517     ssh->sc_comp_ctx = ssh->sccomp->decompress_init();
7518
7519     /*
7520      * Set IVs on server-to-client keys. Here we use the exchange
7521      * hash from the _first_ key exchange.
7522      */
7523     if (ssh->sccipher) {
7524         unsigned char *key;
7525
7526         key = ssh2_mkkey(ssh, s->K, s->exchange_hash, 'D',
7527                          ssh->sccipher->padded_keybytes);
7528         ssh->sccipher->setkey(ssh->sc_cipher_ctx, key);
7529         smemclr(key, ssh->sccipher->padded_keybytes);
7530         sfree(key);
7531
7532         key = ssh2_mkkey(ssh, s->K, s->exchange_hash, 'B',
7533                          ssh->sccipher->blksize);
7534         ssh->sccipher->setiv(ssh->sc_cipher_ctx, key);
7535         smemclr(key, ssh->sccipher->blksize);
7536         sfree(key);
7537     }
7538     if (ssh->scmac) {
7539         unsigned char *key;
7540
7541         key = ssh2_mkkey(ssh, s->K, s->exchange_hash, 'F',
7542                          ssh->scmac->keylen);
7543         ssh->scmac->setkey(ssh->sc_mac_ctx, key);
7544         smemclr(key, ssh->scmac->keylen);
7545         sfree(key);
7546     }
7547     if (ssh->sccipher)
7548         logeventf(ssh, "Initialised %.200s server->client encryption",
7549                   ssh->sccipher->text_name);
7550     if (ssh->scmac)
7551         logeventf(ssh, "Initialised %.200s server->client MAC algorithm%s%s",
7552                   ssh->scmac->text_name,
7553                   ssh->scmac_etm ? " (in ETM mode)" : "",
7554                   ssh->sccipher->required_mac ? " (required by cipher)" : "");
7555     if (ssh->sccomp->text_name)
7556         logeventf(ssh, "Initialised %s decompression",
7557                   ssh->sccomp->text_name);
7558
7559     /*
7560      * Free shared secret.
7561      */
7562     freebn(s->K);
7563
7564     /*
7565      * Update the specials menu to list the remaining uncertified host
7566      * keys.
7567      */
7568     update_specials_menu(ssh->frontend);
7569
7570     /*
7571      * Key exchange is over. Loop straight back round if we have a
7572      * deferred rekey reason.
7573      */
7574     if (ssh->deferred_rekey_reason) {
7575         logevent(ssh->deferred_rekey_reason);
7576         pktin = NULL;
7577         ssh->deferred_rekey_reason = NULL;
7578         goto begin_key_exchange;
7579     }
7580
7581     /*
7582      * Otherwise, schedule a timer for our next rekey.
7583      */
7584     ssh->kex_in_progress = FALSE;
7585     ssh->last_rekey = GETTICKCOUNT();
7586     if (conf_get_int(ssh->conf, CONF_ssh_rekey_time) != 0)
7587         ssh->next_rekey = schedule_timer(conf_get_int(ssh->conf, CONF_ssh_rekey_time)*60*TICKSPERSEC,
7588                                          ssh2_timer, ssh);
7589
7590     /*
7591      * Now we're encrypting. Begin returning 1 to the protocol main
7592      * function so that other things can run on top of the
7593      * transport. If we ever see a KEXINIT, we must go back to the
7594      * start.
7595      * 
7596      * We _also_ go back to the start if we see pktin==NULL and
7597      * inlen negative, because this is a special signal meaning
7598      * `initiate client-driven rekey', and `in' contains a message
7599      * giving the reason for the rekey.
7600      *
7601      * inlen==-1 means always initiate a rekey;
7602      * inlen==-2 means that userauth has completed successfully and
7603      *   we should consider rekeying (for delayed compression).
7604      */
7605     while (!((pktin && pktin->type == SSH2_MSG_KEXINIT) ||
7606              (!pktin && inlen < 0))) {
7607         wait_for_rekey:
7608         if (!ssh->protocol_initial_phase_done) {
7609             ssh->protocol_initial_phase_done = TRUE;
7610             /*
7611              * Allow authconn to initialise itself.
7612              */
7613             do_ssh2_authconn(ssh, NULL, 0, NULL);
7614         }
7615         crReturnV;
7616     }
7617     if (pktin) {
7618         logevent("Server initiated key re-exchange");
7619     } else {
7620         if (inlen == -2) {
7621             /* 
7622              * authconn has seen a USERAUTH_SUCCEEDED. Time to enable
7623              * delayed compression, if it's available.
7624              *
7625              * draft-miller-secsh-compression-delayed-00 says that you
7626              * negotiate delayed compression in the first key exchange, and
7627              * both sides start compressing when the server has sent
7628              * USERAUTH_SUCCESS. This has a race condition -- the server
7629              * can't know when the client has seen it, and thus which incoming
7630              * packets it should treat as compressed.
7631              *
7632              * Instead, we do the initial key exchange without offering the
7633              * delayed methods, but note if the server offers them; when we
7634              * get here, if a delayed method was available that was higher
7635              * on our list than what we got, we initiate a rekey in which we
7636              * _do_ list the delayed methods (and hopefully get it as a
7637              * result). Subsequent rekeys will do the same.
7638              */
7639             assert(!s->userauth_succeeded); /* should only happen once */
7640             s->userauth_succeeded = TRUE;
7641             if (!s->pending_compression)
7642                 /* Can't see any point rekeying. */
7643                 goto wait_for_rekey;       /* this is utterly horrid */
7644             /* else fall through to rekey... */
7645             s->pending_compression = FALSE;
7646         }
7647         /*
7648          * Now we've decided to rekey.
7649          *
7650          * Special case: if the server bug is set that doesn't
7651          * allow rekeying, we give a different log message and
7652          * continue waiting. (If such a server _initiates_ a rekey,
7653          * we process it anyway!)
7654          */
7655         if ((ssh->remote_bugs & BUG_SSH2_REKEY)) {
7656             logeventf(ssh, "Server bug prevents key re-exchange (%s)",
7657                       (char *)in);
7658             /* Reset the counters, so that at least this message doesn't
7659              * hit the event log _too_ often. */
7660             ssh->outgoing_data_size = 0;
7661             ssh->incoming_data_size = 0;
7662             if (conf_get_int(ssh->conf, CONF_ssh_rekey_time) != 0) {
7663                 ssh->next_rekey =
7664                     schedule_timer(conf_get_int(ssh->conf, CONF_ssh_rekey_time)*60*TICKSPERSEC,
7665                                    ssh2_timer, ssh);
7666             }
7667             goto wait_for_rekey;       /* this is still utterly horrid */
7668         } else {
7669             logeventf(ssh, "Initiating key re-exchange (%s)", (char *)in);
7670         }
7671     }
7672     goto begin_key_exchange;
7673
7674     crFinishV;
7675 }
7676
7677 /*
7678  * Send data on an SSH channel.  In SSH-2, this involves buffering it
7679  * first.
7680  */
7681 static int ssh_send_channel_data(struct ssh_channel *c, const char *buf,
7682                                    int len)
7683 {
7684     if (c->ssh->version == 2) {
7685         bufchain_add(&c->v.v2.outbuffer, buf, len);
7686         return ssh2_try_send(c);
7687     } else {
7688         send_packet(c->ssh, SSH1_MSG_CHANNEL_DATA,
7689                     PKT_INT, c->remoteid,
7690                     PKT_INT, len,
7691                     PKT_DATA, buf, len,
7692                     PKT_END);
7693         /*
7694          * In SSH-1 we can return 0 here - implying that channels are
7695          * never individually throttled - because the only
7696          * circumstance that can cause throttling will be the whole
7697          * SSH connection backing up, in which case _everything_ will
7698          * be throttled as a whole.
7699          */
7700         return 0;
7701     }
7702 }
7703
7704 /*
7705  * Attempt to send data on an SSH-2 channel.
7706  */
7707 static int ssh2_try_send(struct ssh_channel *c)
7708 {
7709     Ssh ssh = c->ssh;
7710     struct Packet *pktout;
7711     int ret;
7712
7713     while (c->v.v2.remwindow > 0 && bufchain_size(&c->v.v2.outbuffer) > 0) {
7714         int len;
7715         void *data;
7716         bufchain_prefix(&c->v.v2.outbuffer, &data, &len);
7717         if ((unsigned)len > c->v.v2.remwindow)
7718             len = c->v.v2.remwindow;
7719         if ((unsigned)len > c->v.v2.remmaxpkt)
7720             len = c->v.v2.remmaxpkt;
7721         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_DATA);
7722         ssh2_pkt_adduint32(pktout, c->remoteid);
7723         ssh2_pkt_addstring_start(pktout);
7724         ssh2_pkt_addstring_data(pktout, data, len);
7725         ssh2_pkt_send(ssh, pktout);
7726         bufchain_consume(&c->v.v2.outbuffer, len);
7727         c->v.v2.remwindow -= len;
7728     }
7729
7730     /*
7731      * After having sent as much data as we can, return the amount
7732      * still buffered.
7733      */
7734     ret = bufchain_size(&c->v.v2.outbuffer);
7735
7736     /*
7737      * And if there's no data pending but we need to send an EOF, send
7738      * it.
7739      */
7740     if (!ret && c->pending_eof)
7741         ssh_channel_try_eof(c);
7742
7743     return ret;
7744 }
7745
7746 static void ssh2_try_send_and_unthrottle(Ssh ssh, struct ssh_channel *c)
7747 {
7748     int bufsize;
7749     if (c->closes & CLOSES_SENT_EOF)
7750         return;                   /* don't send on channels we've EOFed */
7751     bufsize = ssh2_try_send(c);
7752     if (bufsize == 0) {
7753         switch (c->type) {
7754           case CHAN_MAINSESSION:
7755             /* stdin need not receive an unthrottle
7756              * notification since it will be polled */
7757             break;
7758           case CHAN_X11:
7759             x11_unthrottle(c->u.x11.xconn);
7760             break;
7761           case CHAN_AGENT:
7762             /* agent sockets are request/response and need no
7763              * buffer management */
7764             break;
7765           case CHAN_SOCKDATA:
7766             pfd_unthrottle(c->u.pfd.pf);
7767             break;
7768         }
7769     }
7770 }
7771
7772 static int ssh_is_simple(Ssh ssh)
7773 {
7774     /*
7775      * We use the 'simple' variant of the SSH protocol if we're asked
7776      * to, except not if we're also doing connection-sharing (either
7777      * tunnelling our packets over an upstream or expecting to be
7778      * tunnelled over ourselves), since then the assumption that we
7779      * have only one channel to worry about is not true after all.
7780      */
7781     return (conf_get_int(ssh->conf, CONF_ssh_simple) &&
7782             !ssh->bare_connection && !ssh->connshare);
7783 }
7784
7785 /*
7786  * Set up most of a new ssh_channel.
7787  */
7788 static void ssh_channel_init(struct ssh_channel *c)
7789 {
7790     Ssh ssh = c->ssh;
7791     c->localid = alloc_channel_id(ssh);
7792     c->closes = 0;
7793     c->pending_eof = FALSE;
7794     c->throttling_conn = FALSE;
7795     if (ssh->version == 2) {
7796         c->v.v2.locwindow = c->v.v2.locmaxwin = c->v.v2.remlocwin =
7797             ssh_is_simple(ssh) ? OUR_V2_BIGWIN : OUR_V2_WINSIZE;
7798         c->v.v2.chanreq_head = NULL;
7799         c->v.v2.throttle_state = UNTHROTTLED;
7800         bufchain_init(&c->v.v2.outbuffer);
7801     }
7802     add234(ssh->channels, c);
7803 }
7804
7805 /*
7806  * Construct the common parts of a CHANNEL_OPEN.
7807  */
7808 static struct Packet *ssh2_chanopen_init(struct ssh_channel *c,
7809                                          const char *type)
7810 {
7811     struct Packet *pktout;
7812
7813     pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN);
7814     ssh2_pkt_addstring(pktout, type);
7815     ssh2_pkt_adduint32(pktout, c->localid);
7816     ssh2_pkt_adduint32(pktout, c->v.v2.locwindow);/* our window size */
7817     ssh2_pkt_adduint32(pktout, OUR_V2_MAXPKT);      /* our max pkt size */
7818     return pktout;
7819 }
7820
7821 /*
7822  * CHANNEL_FAILURE doesn't come with any indication of what message
7823  * caused it, so we have to keep track of the outstanding
7824  * CHANNEL_REQUESTs ourselves.
7825  */
7826 static void ssh2_queue_chanreq_handler(struct ssh_channel *c,
7827                                        cchandler_fn_t handler, void *ctx)
7828 {
7829     struct outstanding_channel_request *ocr =
7830         snew(struct outstanding_channel_request);
7831
7832     assert(!(c->closes & (CLOSES_SENT_CLOSE | CLOSES_RCVD_CLOSE)));
7833     ocr->handler = handler;
7834     ocr->ctx = ctx;
7835     ocr->next = NULL;
7836     if (!c->v.v2.chanreq_head)
7837         c->v.v2.chanreq_head = ocr;
7838     else
7839         c->v.v2.chanreq_tail->next = ocr;
7840     c->v.v2.chanreq_tail = ocr;
7841 }
7842
7843 /*
7844  * Construct the common parts of a CHANNEL_REQUEST.  If handler is not
7845  * NULL then a reply will be requested and the handler will be called
7846  * when it arrives.  The returned packet is ready to have any
7847  * request-specific data added and be sent.  Note that if a handler is
7848  * provided, it's essential that the request actually be sent.
7849  *
7850  * The handler will usually be passed the response packet in pktin. If
7851  * pktin is NULL, this means that no reply will ever be forthcoming
7852  * (e.g. because the entire connection is being destroyed, or because
7853  * the server initiated channel closure before we saw the response)
7854  * and the handler should free any storage it's holding.
7855  */
7856 static struct Packet *ssh2_chanreq_init(struct ssh_channel *c,
7857                                         const char *type,
7858                                         cchandler_fn_t handler, void *ctx)
7859 {
7860     struct Packet *pktout;
7861
7862     assert(!(c->closes & (CLOSES_SENT_CLOSE | CLOSES_RCVD_CLOSE)));
7863     pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
7864     ssh2_pkt_adduint32(pktout, c->remoteid);
7865     ssh2_pkt_addstring(pktout, type);
7866     ssh2_pkt_addbool(pktout, handler != NULL);
7867     if (handler != NULL)
7868         ssh2_queue_chanreq_handler(c, handler, ctx);
7869     return pktout;
7870 }
7871
7872 static void ssh_channel_unthrottle(struct ssh_channel *c, int bufsize)
7873 {
7874     Ssh ssh = c->ssh;
7875     int buflimit;
7876
7877     if (ssh->version == 1) {
7878         buflimit = SSH1_BUFFER_LIMIT;
7879     } else {
7880         if (ssh_is_simple(ssh))
7881             buflimit = 0;
7882         else
7883             buflimit = c->v.v2.locmaxwin;
7884         if (bufsize < buflimit)
7885             ssh2_set_window(c, buflimit - bufsize);
7886     }
7887     if (c->throttling_conn && bufsize <= buflimit) {
7888         c->throttling_conn = 0;
7889         ssh_throttle_conn(ssh, -1);
7890     }
7891 }
7892
7893 /*
7894  * Potentially enlarge the window on an SSH-2 channel.
7895  */
7896 static void ssh2_handle_winadj_response(struct ssh_channel *, struct Packet *,
7897                                         void *);
7898 static void ssh2_set_window(struct ssh_channel *c, int newwin)
7899 {
7900     Ssh ssh = c->ssh;
7901
7902     /*
7903      * Never send WINDOW_ADJUST for a channel that the remote side has
7904      * already sent EOF on; there's no point, since it won't be
7905      * sending any more data anyway. Ditto if _we've_ already sent
7906      * CLOSE.
7907      */
7908     if (c->closes & (CLOSES_RCVD_EOF | CLOSES_SENT_CLOSE))
7909         return;
7910
7911     /*
7912      * Also, never widen the window for an X11 channel when we're
7913      * still waiting to see its initial auth and may yet hand it off
7914      * to a downstream.
7915      */
7916     if (c->type == CHAN_X11 && c->u.x11.initial)
7917         return;
7918
7919     /*
7920      * If the remote end has a habit of ignoring maxpkt, limit the
7921      * window so that it has no choice (assuming it doesn't ignore the
7922      * window as well).
7923      */
7924     if ((ssh->remote_bugs & BUG_SSH2_MAXPKT) && newwin > OUR_V2_MAXPKT)
7925         newwin = OUR_V2_MAXPKT;
7926
7927     /*
7928      * Only send a WINDOW_ADJUST if there's significantly more window
7929      * available than the other end thinks there is.  This saves us
7930      * sending a WINDOW_ADJUST for every character in a shell session.
7931      *
7932      * "Significant" is arbitrarily defined as half the window size.
7933      */
7934     if (newwin / 2 >= c->v.v2.locwindow) {
7935         struct Packet *pktout;
7936         unsigned *up;
7937
7938         /*
7939          * In order to keep track of how much window the client
7940          * actually has available, we'd like it to acknowledge each
7941          * WINDOW_ADJUST.  We can't do that directly, so we accompany
7942          * it with a CHANNEL_REQUEST that has to be acknowledged.
7943          *
7944          * This is only necessary if we're opening the window wide.
7945          * If we're not, then throughput is being constrained by
7946          * something other than the maximum window size anyway.
7947          */
7948         if (newwin == c->v.v2.locmaxwin &&
7949             !(ssh->remote_bugs & BUG_CHOKES_ON_WINADJ)) {
7950             up = snew(unsigned);
7951             *up = newwin - c->v.v2.locwindow;
7952             pktout = ssh2_chanreq_init(c, "winadj@putty.projects.tartarus.org",
7953                                        ssh2_handle_winadj_response, up);
7954             ssh2_pkt_send(ssh, pktout);
7955
7956             if (c->v.v2.throttle_state != UNTHROTTLED)
7957                 c->v.v2.throttle_state = UNTHROTTLING;
7958         } else {
7959             /* Pretend the WINDOW_ADJUST was acked immediately. */
7960             c->v.v2.remlocwin = newwin;
7961             c->v.v2.throttle_state = THROTTLED;
7962         }
7963         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_WINDOW_ADJUST);
7964         ssh2_pkt_adduint32(pktout, c->remoteid);
7965         ssh2_pkt_adduint32(pktout, newwin - c->v.v2.locwindow);
7966         ssh2_pkt_send(ssh, pktout);
7967         c->v.v2.locwindow = newwin;
7968     }
7969 }
7970
7971 /*
7972  * Find the channel associated with a message.  If there's no channel,
7973  * or it's not properly open, make a noise about it and return NULL.
7974  */
7975 static struct ssh_channel *ssh_channel_msg(Ssh ssh, struct Packet *pktin)
7976 {
7977     unsigned localid = ssh_pkt_getuint32(pktin);
7978     struct ssh_channel *c;
7979     int halfopen_ok;
7980
7981     /* Is this message OK on a half-open connection? */
7982     if (ssh->version == 1)
7983         halfopen_ok = (pktin->type == SSH1_MSG_CHANNEL_OPEN_CONFIRMATION ||
7984                        pktin->type == SSH1_MSG_CHANNEL_OPEN_FAILURE);
7985     else
7986         halfopen_ok = (pktin->type == SSH2_MSG_CHANNEL_OPEN_CONFIRMATION ||
7987                        pktin->type == SSH2_MSG_CHANNEL_OPEN_FAILURE);
7988     c = find234(ssh->channels, &localid, ssh_channelfind);
7989     if (!c || (c->type != CHAN_SHARING && (c->halfopen != halfopen_ok))) {
7990         char *buf = dupprintf("Received %s for %s channel %u",
7991                               ssh_pkt_type(ssh, pktin->type),
7992                               !c ? "nonexistent" :
7993                               c->halfopen ? "half-open" : "open",
7994                               localid);
7995         ssh_disconnect(ssh, NULL, buf, SSH2_DISCONNECT_PROTOCOL_ERROR, FALSE);
7996         sfree(buf);
7997         return NULL;
7998     }
7999     return c;
8000 }
8001
8002 static void ssh2_handle_winadj_response(struct ssh_channel *c,
8003                                         struct Packet *pktin, void *ctx)
8004 {
8005     unsigned *sizep = ctx;
8006
8007     /*
8008      * Winadj responses should always be failures. However, at least
8009      * one server ("boks_sshd") is known to return SUCCESS for channel
8010      * requests it's never heard of, such as "winadj@putty". Raised
8011      * with foxt.com as bug 090916-090424, but for the sake of a quiet
8012      * life, we don't worry about what kind of response we got.
8013      */
8014
8015     c->v.v2.remlocwin += *sizep;
8016     sfree(sizep);
8017     /*
8018      * winadj messages are only sent when the window is fully open, so
8019      * if we get an ack of one, we know any pending unthrottle is
8020      * complete.
8021      */
8022     if (c->v.v2.throttle_state == UNTHROTTLING)
8023         c->v.v2.throttle_state = UNTHROTTLED;
8024 }
8025
8026 static void ssh2_msg_channel_response(Ssh ssh, struct Packet *pktin)
8027 {
8028     struct ssh_channel *c = ssh_channel_msg(ssh, pktin);
8029     struct outstanding_channel_request *ocr;
8030
8031     if (!c) return;
8032     if (c->type == CHAN_SHARING) {
8033         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8034                                   pktin->body, pktin->length);
8035         return;
8036     }
8037     ocr = c->v.v2.chanreq_head;
8038     if (!ocr) {
8039         ssh2_msg_unexpected(ssh, pktin);
8040         return;
8041     }
8042     ocr->handler(c, pktin, ocr->ctx);
8043     c->v.v2.chanreq_head = ocr->next;
8044     sfree(ocr);
8045     /*
8046      * We may now initiate channel-closing procedures, if that
8047      * CHANNEL_REQUEST was the last thing outstanding before we send
8048      * CHANNEL_CLOSE.
8049      */
8050     ssh2_channel_check_close(c);
8051 }
8052
8053 static void ssh2_msg_channel_window_adjust(Ssh ssh, struct Packet *pktin)
8054 {
8055     struct ssh_channel *c;
8056     c = ssh_channel_msg(ssh, pktin);
8057     if (!c)
8058         return;
8059     if (c->type == CHAN_SHARING) {
8060         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8061                                   pktin->body, pktin->length);
8062         return;
8063     }
8064     if (!(c->closes & CLOSES_SENT_EOF)) {
8065         c->v.v2.remwindow += ssh_pkt_getuint32(pktin);
8066         ssh2_try_send_and_unthrottle(ssh, c);
8067     }
8068 }
8069
8070 static void ssh2_msg_channel_data(Ssh ssh, struct Packet *pktin)
8071 {
8072     char *data;
8073     int length;
8074     struct ssh_channel *c;
8075     c = ssh_channel_msg(ssh, pktin);
8076     if (!c)
8077         return;
8078     if (c->type == CHAN_SHARING) {
8079         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8080                                   pktin->body, pktin->length);
8081         return;
8082     }
8083     if (pktin->type == SSH2_MSG_CHANNEL_EXTENDED_DATA &&
8084         ssh_pkt_getuint32(pktin) != SSH2_EXTENDED_DATA_STDERR)
8085         return;                        /* extended but not stderr */
8086     ssh_pkt_getstring(pktin, &data, &length);
8087     if (data) {
8088         int bufsize;
8089         c->v.v2.locwindow -= length;
8090         c->v.v2.remlocwin -= length;
8091         bufsize = ssh_channel_data(c, pktin->type ==
8092                                    SSH2_MSG_CHANNEL_EXTENDED_DATA,
8093                                    data, length);
8094         /*
8095          * If it looks like the remote end hit the end of its window,
8096          * and we didn't want it to do that, think about using a
8097          * larger window.
8098          */
8099         if (c->v.v2.remlocwin <= 0 && c->v.v2.throttle_state == UNTHROTTLED &&
8100             c->v.v2.locmaxwin < 0x40000000)
8101             c->v.v2.locmaxwin += OUR_V2_WINSIZE;
8102         /*
8103          * If we are not buffering too much data,
8104          * enlarge the window again at the remote side.
8105          * If we are buffering too much, we may still
8106          * need to adjust the window if the server's
8107          * sent excess data.
8108          */
8109         if (bufsize < c->v.v2.locmaxwin)
8110             ssh2_set_window(c, c->v.v2.locmaxwin - bufsize);
8111         /*
8112          * If we're either buffering way too much data, or if we're
8113          * buffering anything at all and we're in "simple" mode,
8114          * throttle the whole channel.
8115          */
8116         if ((bufsize > c->v.v2.locmaxwin || (ssh_is_simple(ssh) && bufsize>0))
8117             && !c->throttling_conn) {
8118             c->throttling_conn = 1;
8119             ssh_throttle_conn(ssh, +1);
8120         }
8121     }
8122 }
8123
8124 static void ssh_check_termination(Ssh ssh)
8125 {
8126     if (ssh->version == 2 &&
8127         !conf_get_int(ssh->conf, CONF_ssh_no_shell) &&
8128         (ssh->channels && count234(ssh->channels) == 0) &&
8129         !(ssh->connshare && share_ndownstreams(ssh->connshare) > 0)) {
8130         /*
8131          * We used to send SSH_MSG_DISCONNECT here, because I'd
8132          * believed that _every_ conforming SSH-2 connection had to
8133          * end with a disconnect being sent by at least one side;
8134          * apparently I was wrong and it's perfectly OK to
8135          * unceremoniously slam the connection shut when you're done,
8136          * and indeed OpenSSH feels this is more polite than sending a
8137          * DISCONNECT. So now we don't.
8138          */
8139         ssh_disconnect(ssh, "All channels closed", NULL, 0, TRUE);
8140     }
8141 }
8142
8143 void ssh_sharing_downstream_connected(Ssh ssh, unsigned id,
8144                                       const char *peerinfo)
8145 {
8146     if (peerinfo)
8147         logeventf(ssh, "Connection sharing downstream #%u connected from %s",
8148                   id, peerinfo);
8149     else
8150         logeventf(ssh, "Connection sharing downstream #%u connected", id);
8151 }
8152
8153 void ssh_sharing_downstream_disconnected(Ssh ssh, unsigned id)
8154 {
8155     logeventf(ssh, "Connection sharing downstream #%u disconnected", id);
8156     ssh_check_termination(ssh);
8157 }
8158
8159 void ssh_sharing_logf(Ssh ssh, unsigned id, const char *logfmt, ...)
8160 {
8161     va_list ap;
8162     char *buf;
8163
8164     va_start(ap, logfmt);
8165     buf = dupvprintf(logfmt, ap);
8166     va_end(ap);
8167     if (id)
8168         logeventf(ssh, "Connection sharing downstream #%u: %s", id, buf);
8169     else
8170         logeventf(ssh, "Connection sharing: %s", buf);
8171     sfree(buf);
8172 }
8173
8174 static void ssh_channel_destroy(struct ssh_channel *c)
8175 {
8176     Ssh ssh = c->ssh;
8177
8178     switch (c->type) {
8179       case CHAN_MAINSESSION:
8180         ssh->mainchan = NULL;
8181         update_specials_menu(ssh->frontend);
8182         break;
8183       case CHAN_X11:
8184         if (c->u.x11.xconn != NULL)
8185             x11_close(c->u.x11.xconn);
8186         logevent("Forwarded X11 connection terminated");
8187         break;
8188       case CHAN_AGENT:
8189         sfree(c->u.a.message);
8190         break;
8191       case CHAN_SOCKDATA:
8192         if (c->u.pfd.pf != NULL)
8193             pfd_close(c->u.pfd.pf);
8194         logevent("Forwarded port closed");
8195         break;
8196     }
8197
8198     del234(ssh->channels, c);
8199     if (ssh->version == 2) {
8200         bufchain_clear(&c->v.v2.outbuffer);
8201         assert(c->v.v2.chanreq_head == NULL);
8202     }
8203     sfree(c);
8204
8205     /*
8206      * If that was the last channel left open, we might need to
8207      * terminate.
8208      */
8209     ssh_check_termination(ssh);
8210 }
8211
8212 static void ssh2_channel_check_close(struct ssh_channel *c)
8213 {
8214     Ssh ssh = c->ssh;
8215     struct Packet *pktout;
8216
8217     assert(ssh->version == 2);
8218     if (c->halfopen) {
8219         /*
8220          * If we've sent out our own CHANNEL_OPEN but not yet seen
8221          * either OPEN_CONFIRMATION or OPEN_FAILURE in response, then
8222          * it's too early to be sending close messages of any kind.
8223          */
8224         return;
8225     }
8226
8227     if ((!((CLOSES_SENT_EOF | CLOSES_RCVD_EOF) & ~c->closes) ||
8228          c->type == CHAN_ZOMBIE) &&
8229         !c->v.v2.chanreq_head &&
8230         !(c->closes & CLOSES_SENT_CLOSE)) {
8231         /*
8232          * We have both sent and received EOF (or the channel is a
8233          * zombie), and we have no outstanding channel requests, which
8234          * means the channel is in final wind-up. But we haven't sent
8235          * CLOSE, so let's do so now.
8236          */
8237         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_CLOSE);
8238         ssh2_pkt_adduint32(pktout, c->remoteid);
8239         ssh2_pkt_send(ssh, pktout);
8240         c->closes |= CLOSES_SENT_EOF | CLOSES_SENT_CLOSE;
8241     }
8242
8243     if (!((CLOSES_SENT_CLOSE | CLOSES_RCVD_CLOSE) & ~c->closes)) {
8244         assert(c->v.v2.chanreq_head == NULL);
8245         /*
8246          * We have both sent and received CLOSE, which means we're
8247          * completely done with the channel.
8248          */
8249         ssh_channel_destroy(c);
8250     }
8251 }
8252
8253 static void ssh2_channel_got_eof(struct ssh_channel *c)
8254 {
8255     if (c->closes & CLOSES_RCVD_EOF)
8256         return;                        /* already seen EOF */
8257     c->closes |= CLOSES_RCVD_EOF;
8258
8259     if (c->type == CHAN_X11) {
8260         x11_send_eof(c->u.x11.xconn);
8261     } else if (c->type == CHAN_AGENT) {
8262         if (c->u.a.outstanding_requests == 0) {
8263             /* Manufacture an outgoing EOF in response to the incoming one. */
8264             sshfwd_write_eof(c);
8265         }
8266     } else if (c->type == CHAN_SOCKDATA) {
8267         pfd_send_eof(c->u.pfd.pf);
8268     } else if (c->type == CHAN_MAINSESSION) {
8269         Ssh ssh = c->ssh;
8270
8271         if (!ssh->sent_console_eof &&
8272             (from_backend_eof(ssh->frontend) || ssh->got_pty)) {
8273             /*
8274              * Either from_backend_eof told us that the front end
8275              * wants us to close the outgoing side of the connection
8276              * as soon as we see EOF from the far end, or else we've
8277              * unilaterally decided to do that because we've allocated
8278              * a remote pty and hence EOF isn't a particularly
8279              * meaningful concept.
8280              */
8281             sshfwd_write_eof(c);
8282         }
8283         ssh->sent_console_eof = TRUE;
8284     }
8285
8286     ssh2_channel_check_close(c);
8287 }
8288
8289 static void ssh2_msg_channel_eof(Ssh ssh, struct Packet *pktin)
8290 {
8291     struct ssh_channel *c;
8292
8293     c = ssh_channel_msg(ssh, pktin);
8294     if (!c)
8295         return;
8296     if (c->type == CHAN_SHARING) {
8297         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8298                                   pktin->body, pktin->length);
8299         return;
8300     }
8301     ssh2_channel_got_eof(c);
8302 }
8303
8304 static void ssh2_msg_channel_close(Ssh ssh, struct Packet *pktin)
8305 {
8306     struct ssh_channel *c;
8307
8308     c = ssh_channel_msg(ssh, pktin);
8309     if (!c)
8310         return;
8311     if (c->type == CHAN_SHARING) {
8312         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8313                                   pktin->body, pktin->length);
8314         return;
8315     }
8316
8317     /*
8318      * When we receive CLOSE on a channel, we assume it comes with an
8319      * implied EOF if we haven't seen EOF yet.
8320      */
8321     ssh2_channel_got_eof(c);
8322
8323     if (!(ssh->remote_bugs & BUG_SENDS_LATE_REQUEST_REPLY)) {
8324         /*
8325          * It also means we stop expecting to see replies to any
8326          * outstanding channel requests, so clean those up too.
8327          * (ssh_chanreq_init will enforce by assertion that we don't
8328          * subsequently put anything back on this list.)
8329          */
8330         while (c->v.v2.chanreq_head) {
8331             struct outstanding_channel_request *ocr = c->v.v2.chanreq_head;
8332             ocr->handler(c, NULL, ocr->ctx);
8333             c->v.v2.chanreq_head = ocr->next;
8334             sfree(ocr);
8335         }
8336     }
8337
8338     /*
8339      * And we also send an outgoing EOF, if we haven't already, on the
8340      * assumption that CLOSE is a pretty forceful announcement that
8341      * the remote side is doing away with the entire channel. (If it
8342      * had wanted to send us EOF and continue receiving data from us,
8343      * it would have just sent CHANNEL_EOF.)
8344      */
8345     if (!(c->closes & CLOSES_SENT_EOF)) {
8346         /*
8347          * Make sure we don't read any more from whatever our local
8348          * data source is for this channel.
8349          */
8350         switch (c->type) {
8351           case CHAN_MAINSESSION:
8352             ssh->send_ok = 0;     /* stop trying to read from stdin */
8353             break;
8354           case CHAN_X11:
8355             x11_override_throttle(c->u.x11.xconn, 1);
8356             break;
8357           case CHAN_SOCKDATA:
8358             pfd_override_throttle(c->u.pfd.pf, 1);
8359             break;
8360         }
8361
8362         /*
8363          * Abandon any buffered data we still wanted to send to this
8364          * channel. Receiving a CHANNEL_CLOSE is an indication that
8365          * the server really wants to get on and _destroy_ this
8366          * channel, and it isn't going to send us any further
8367          * WINDOW_ADJUSTs to permit us to send pending stuff.
8368          */
8369         bufchain_clear(&c->v.v2.outbuffer);
8370
8371         /*
8372          * Send outgoing EOF.
8373          */
8374         sshfwd_write_eof(c);
8375     }
8376
8377     /*
8378      * Now process the actual close.
8379      */
8380     if (!(c->closes & CLOSES_RCVD_CLOSE)) {
8381         c->closes |= CLOSES_RCVD_CLOSE;
8382         ssh2_channel_check_close(c);
8383     }
8384 }
8385
8386 static void ssh2_msg_channel_open_confirmation(Ssh ssh, struct Packet *pktin)
8387 {
8388     struct ssh_channel *c;
8389
8390     c = ssh_channel_msg(ssh, pktin);
8391     if (!c)
8392         return;
8393     if (c->type == CHAN_SHARING) {
8394         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8395                                   pktin->body, pktin->length);
8396         return;
8397     }
8398     assert(c->halfopen); /* ssh_channel_msg will have enforced this */
8399     c->remoteid = ssh_pkt_getuint32(pktin);
8400     c->halfopen = FALSE;
8401     c->v.v2.remwindow = ssh_pkt_getuint32(pktin);
8402     c->v.v2.remmaxpkt = ssh_pkt_getuint32(pktin);
8403
8404     if (c->type == CHAN_SOCKDATA_DORMANT) {
8405         c->type = CHAN_SOCKDATA;
8406         if (c->u.pfd.pf)
8407             pfd_confirm(c->u.pfd.pf);
8408     } else if (c->type == CHAN_ZOMBIE) {
8409         /*
8410          * This case can occur if a local socket error occurred
8411          * between us sending out CHANNEL_OPEN and receiving
8412          * OPEN_CONFIRMATION. In this case, all we can do is
8413          * immediately initiate close proceedings now that we know the
8414          * server's id to put in the close message.
8415          */
8416         ssh2_channel_check_close(c);
8417     } else {
8418         /*
8419          * We never expect to receive OPEN_CONFIRMATION for any
8420          * *other* channel type (since only local-to-remote port
8421          * forwardings cause us to send CHANNEL_OPEN after the main
8422          * channel is live - all other auxiliary channel types are
8423          * initiated from the server end). It's safe to enforce this
8424          * by assertion rather than by ssh_disconnect, because the
8425          * real point is that we never constructed a half-open channel
8426          * structure in the first place with any type other than the
8427          * above.
8428          */
8429         assert(!"Funny channel type in ssh2_msg_channel_open_confirmation");
8430     }
8431
8432     if (c->pending_eof)
8433         ssh_channel_try_eof(c);        /* in case we had a pending EOF */
8434 }
8435
8436 static void ssh2_msg_channel_open_failure(Ssh ssh, struct Packet *pktin)
8437 {
8438     static const char *const reasons[] = {
8439         "<unknown reason code>",
8440             "Administratively prohibited",
8441             "Connect failed",
8442             "Unknown channel type",
8443             "Resource shortage",
8444     };
8445     unsigned reason_code;
8446     char *reason_string;
8447     int reason_length;
8448     struct ssh_channel *c;
8449
8450     c = ssh_channel_msg(ssh, pktin);
8451     if (!c)
8452         return;
8453     if (c->type == CHAN_SHARING) {
8454         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8455                                   pktin->body, pktin->length);
8456         return;
8457     }
8458     assert(c->halfopen); /* ssh_channel_msg will have enforced this */
8459
8460     if (c->type == CHAN_SOCKDATA_DORMANT) {
8461         reason_code = ssh_pkt_getuint32(pktin);
8462         if (reason_code >= lenof(reasons))
8463             reason_code = 0; /* ensure reasons[reason_code] in range */
8464         ssh_pkt_getstring(pktin, &reason_string, &reason_length);
8465         logeventf(ssh, "Forwarded connection refused by server: %s [%.*s]",
8466                   reasons[reason_code], reason_length,
8467                   NULLTOEMPTY(reason_string));
8468
8469         pfd_close(c->u.pfd.pf);
8470     } else if (c->type == CHAN_ZOMBIE) {
8471         /*
8472          * This case can occur if a local socket error occurred
8473          * between us sending out CHANNEL_OPEN and receiving
8474          * OPEN_FAILURE. In this case, we need do nothing except allow
8475          * the code below to throw the half-open channel away.
8476          */
8477     } else {
8478         /*
8479          * We never expect to receive OPEN_FAILURE for any *other*
8480          * channel type (since only local-to-remote port forwardings
8481          * cause us to send CHANNEL_OPEN after the main channel is
8482          * live - all other auxiliary channel types are initiated from
8483          * the server end). It's safe to enforce this by assertion
8484          * rather than by ssh_disconnect, because the real point is
8485          * that we never constructed a half-open channel structure in
8486          * the first place with any type other than the above.
8487          */
8488         assert(!"Funny channel type in ssh2_msg_channel_open_failure");
8489     }
8490
8491     del234(ssh->channels, c);
8492     sfree(c);
8493 }
8494
8495 static void ssh2_msg_channel_request(Ssh ssh, struct Packet *pktin)
8496 {
8497     char *type;
8498     int typelen, want_reply;
8499     int reply = SSH2_MSG_CHANNEL_FAILURE; /* default */
8500     struct ssh_channel *c;
8501     struct Packet *pktout;
8502
8503     c = ssh_channel_msg(ssh, pktin);
8504     if (!c)
8505         return;
8506     if (c->type == CHAN_SHARING) {
8507         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8508                                   pktin->body, pktin->length);
8509         return;
8510     }
8511     ssh_pkt_getstring(pktin, &type, &typelen);
8512     want_reply = ssh2_pkt_getbool(pktin);
8513
8514     if (c->closes & CLOSES_SENT_CLOSE) {
8515         /*
8516          * We don't reply to channel requests after we've sent
8517          * CHANNEL_CLOSE for the channel, because our reply might
8518          * cross in the network with the other side's CHANNEL_CLOSE
8519          * and arrive after they have wound the channel up completely.
8520          */
8521         want_reply = FALSE;
8522     }
8523
8524     /*
8525      * Having got the channel number, we now look at
8526      * the request type string to see if it's something
8527      * we recognise.
8528      */
8529     if (c == ssh->mainchan) {
8530         /*
8531          * We recognise "exit-status" and "exit-signal" on
8532          * the primary channel.
8533          */
8534         if (typelen == 11 &&
8535             !memcmp(type, "exit-status", 11)) {
8536
8537             ssh->exitcode = ssh_pkt_getuint32(pktin);
8538             logeventf(ssh, "Server sent command exit status %d",
8539                       ssh->exitcode);
8540             reply = SSH2_MSG_CHANNEL_SUCCESS;
8541
8542         } else if (typelen == 11 &&
8543                    !memcmp(type, "exit-signal", 11)) {
8544
8545             int is_plausible = TRUE, is_int = FALSE;
8546             char *fmt_sig = NULL, *fmt_msg = NULL;
8547             char *msg;
8548             int msglen = 0, core = FALSE;
8549             /* ICK: older versions of OpenSSH (e.g. 3.4p1)
8550              * provide an `int' for the signal, despite its
8551              * having been a `string' in the drafts of RFC 4254 since at
8552              * least 2001. (Fixed in session.c 1.147.) Try to
8553              * infer which we can safely parse it as. */
8554             {
8555                 unsigned char *p = pktin->body +
8556                     pktin->savedpos;
8557                 long len = pktin->length - pktin->savedpos;
8558                 unsigned long num = GET_32BIT(p); /* what is it? */
8559                 /* If it's 0, it hardly matters; assume string */
8560                 if (num == 0) {
8561                     is_int = FALSE;
8562                 } else {
8563                     int maybe_int = FALSE, maybe_str = FALSE;
8564 #define CHECK_HYPOTHESIS(offset, result)                                \
8565                     do                                                  \
8566                     {                                                   \
8567                         int q = toint(offset);                          \
8568                         if (q >= 0 && q+4 <= len) {                     \
8569                             q = toint(q + 4 + GET_32BIT(p+q));          \
8570                             if (q >= 0 && q+4 <= len &&                 \
8571                                 ((q = toint(q + 4 + GET_32BIT(p+q))) != 0) && \
8572                                 q == len)                               \
8573                                 result = TRUE;                          \
8574                         }                                               \
8575                     } while(0)
8576                     CHECK_HYPOTHESIS(4+1, maybe_int);
8577                     CHECK_HYPOTHESIS(4+num+1, maybe_str);
8578 #undef CHECK_HYPOTHESIS
8579                     if (maybe_int && !maybe_str)
8580                         is_int = TRUE;
8581                     else if (!maybe_int && maybe_str)
8582                         is_int = FALSE;
8583                     else
8584                         /* Crikey. Either or neither. Panic. */
8585                         is_plausible = FALSE;
8586                 }
8587             }
8588             ssh->exitcode = 128;       /* means `unknown signal' */
8589             if (is_plausible) {
8590                 if (is_int) {
8591                     /* Old non-standard OpenSSH. */
8592                     int signum = ssh_pkt_getuint32(pktin);
8593                     fmt_sig = dupprintf(" %d", signum);
8594                     ssh->exitcode = 128 + signum;
8595                 } else {
8596                     /* As per RFC 4254. */
8597                     char *sig;
8598                     int siglen;
8599                     ssh_pkt_getstring(pktin, &sig, &siglen);
8600                     /* Signal name isn't supposed to be blank, but
8601                      * let's cope gracefully if it is. */
8602                     if (siglen) {
8603                         fmt_sig = dupprintf(" \"%.*s\"",
8604                                             siglen, sig);
8605                     }
8606
8607                     /*
8608                      * Really hideous method of translating the
8609                      * signal description back into a locally
8610                      * meaningful number.
8611                      */
8612
8613                     if (0)
8614                         ;
8615 #define TRANSLATE_SIGNAL(s) \
8616     else if (siglen == lenof(#s)-1 && !memcmp(sig, #s, siglen)) \
8617         ssh->exitcode = 128 + SIG ## s
8618 #ifdef SIGABRT
8619                     TRANSLATE_SIGNAL(ABRT);
8620 #endif
8621 #ifdef SIGALRM
8622                     TRANSLATE_SIGNAL(ALRM);
8623 #endif
8624 #ifdef SIGFPE
8625                     TRANSLATE_SIGNAL(FPE);
8626 #endif
8627 #ifdef SIGHUP
8628                     TRANSLATE_SIGNAL(HUP);
8629 #endif
8630 #ifdef SIGILL
8631                     TRANSLATE_SIGNAL(ILL);
8632 #endif
8633 #ifdef SIGINT
8634                     TRANSLATE_SIGNAL(INT);
8635 #endif
8636 #ifdef SIGKILL
8637                     TRANSLATE_SIGNAL(KILL);
8638 #endif
8639 #ifdef SIGPIPE
8640                     TRANSLATE_SIGNAL(PIPE);
8641 #endif
8642 #ifdef SIGQUIT
8643                     TRANSLATE_SIGNAL(QUIT);
8644 #endif
8645 #ifdef SIGSEGV
8646                     TRANSLATE_SIGNAL(SEGV);
8647 #endif
8648 #ifdef SIGTERM
8649                     TRANSLATE_SIGNAL(TERM);
8650 #endif
8651 #ifdef SIGUSR1
8652                     TRANSLATE_SIGNAL(USR1);
8653 #endif
8654 #ifdef SIGUSR2
8655                     TRANSLATE_SIGNAL(USR2);
8656 #endif
8657 #undef TRANSLATE_SIGNAL
8658                     else
8659                         ssh->exitcode = 128;
8660                 }
8661                 core = ssh2_pkt_getbool(pktin);
8662                 ssh_pkt_getstring(pktin, &msg, &msglen);
8663                 if (msglen) {
8664                     fmt_msg = dupprintf(" (\"%.*s\")", msglen, msg);
8665                 }
8666                 /* ignore lang tag */
8667             } /* else don't attempt to parse */
8668             logeventf(ssh, "Server exited on signal%s%s%s",
8669                       fmt_sig ? fmt_sig : "",
8670                       core ? " (core dumped)" : "",
8671                       fmt_msg ? fmt_msg : "");
8672             sfree(fmt_sig);
8673             sfree(fmt_msg);
8674             reply = SSH2_MSG_CHANNEL_SUCCESS;
8675
8676         }
8677     } else {
8678         /*
8679          * This is a channel request we don't know
8680          * about, so we now either ignore the request
8681          * or respond with CHANNEL_FAILURE, depending
8682          * on want_reply.
8683          */
8684         reply = SSH2_MSG_CHANNEL_FAILURE;
8685     }
8686     if (want_reply) {
8687         pktout = ssh2_pkt_init(reply);
8688         ssh2_pkt_adduint32(pktout, c->remoteid);
8689         ssh2_pkt_send(ssh, pktout);
8690     }
8691 }
8692
8693 static void ssh2_msg_global_request(Ssh ssh, struct Packet *pktin)
8694 {
8695     char *type;
8696     int typelen, want_reply;
8697     struct Packet *pktout;
8698
8699     ssh_pkt_getstring(pktin, &type, &typelen);
8700     want_reply = ssh2_pkt_getbool(pktin);
8701
8702     /*
8703      * We currently don't support any global requests
8704      * at all, so we either ignore the request or
8705      * respond with REQUEST_FAILURE, depending on
8706      * want_reply.
8707      */
8708     if (want_reply) {
8709         pktout = ssh2_pkt_init(SSH2_MSG_REQUEST_FAILURE);
8710         ssh2_pkt_send(ssh, pktout);
8711     }
8712 }
8713
8714 struct X11FakeAuth *ssh_sharing_add_x11_display(Ssh ssh, int authtype,
8715                                                 void *share_cs,
8716                                                 void *share_chan)
8717 {
8718     struct X11FakeAuth *auth;
8719
8720     /*
8721      * Make up a new set of fake X11 auth data, and add it to the tree
8722      * of currently valid ones with an indication of the sharing
8723      * context that it's relevant to.
8724      */
8725     auth = x11_invent_fake_auth(ssh->x11authtree, authtype);
8726     auth->share_cs = share_cs;
8727     auth->share_chan = share_chan;
8728
8729     return auth;
8730 }
8731
8732 void ssh_sharing_remove_x11_display(Ssh ssh, struct X11FakeAuth *auth)
8733 {
8734     del234(ssh->x11authtree, auth);
8735     x11_free_fake_auth(auth);
8736 }
8737
8738 static void ssh2_msg_channel_open(Ssh ssh, struct Packet *pktin)
8739 {
8740     char *type;
8741     int typelen;
8742     char *peeraddr;
8743     int peeraddrlen;
8744     int peerport;
8745     const char *error = NULL;
8746     struct ssh_channel *c;
8747     unsigned remid, winsize, pktsize;
8748     unsigned our_winsize_override = 0;
8749     struct Packet *pktout;
8750
8751     ssh_pkt_getstring(pktin, &type, &typelen);
8752     c = snew(struct ssh_channel);
8753     c->ssh = ssh;
8754
8755     remid = ssh_pkt_getuint32(pktin);
8756     winsize = ssh_pkt_getuint32(pktin);
8757     pktsize = ssh_pkt_getuint32(pktin);
8758
8759     if (typelen == 3 && !memcmp(type, "x11", 3)) {
8760         char *addrstr;
8761
8762         ssh_pkt_getstring(pktin, &peeraddr, &peeraddrlen);
8763         addrstr = dupprintf("%.*s", peeraddrlen, NULLTOEMPTY(peeraddr));
8764         peerport = ssh_pkt_getuint32(pktin);
8765
8766         logeventf(ssh, "Received X11 connect request from %s:%d",
8767                   addrstr, peerport);
8768
8769         if (!ssh->X11_fwd_enabled && !ssh->connshare)
8770             error = "X11 forwarding is not enabled";
8771         else {
8772             c->u.x11.xconn = x11_init(ssh->x11authtree, c,
8773                                       addrstr, peerport);
8774             c->type = CHAN_X11;
8775             c->u.x11.initial = TRUE;
8776
8777             /*
8778              * If we are a connection-sharing upstream, then we should
8779              * initially present a very small window, adequate to take
8780              * the X11 initial authorisation packet but not much more.
8781              * Downstream will then present us a larger window (by
8782              * fiat of the connection-sharing protocol) and we can
8783              * guarantee to send a positive-valued WINDOW_ADJUST.
8784              */
8785             if (ssh->connshare)
8786                 our_winsize_override = 128;
8787
8788             logevent("Opened X11 forward channel");
8789         }
8790
8791         sfree(addrstr);
8792     } else if (typelen == 15 &&
8793                !memcmp(type, "forwarded-tcpip", 15)) {
8794         struct ssh_rportfwd pf, *realpf;
8795         char *shost;
8796         int shostlen;
8797         ssh_pkt_getstring(pktin, &shost, &shostlen);/* skip address */
8798         pf.shost = dupprintf("%.*s", shostlen, NULLTOEMPTY(shost));
8799         pf.sport = ssh_pkt_getuint32(pktin);
8800         ssh_pkt_getstring(pktin, &peeraddr, &peeraddrlen);
8801         peerport = ssh_pkt_getuint32(pktin);
8802         realpf = find234(ssh->rportfwds, &pf, NULL);
8803         logeventf(ssh, "Received remote port %s:%d open request "
8804                   "from %.*s:%d", pf.shost, pf.sport,
8805                   peeraddrlen, NULLTOEMPTY(peeraddr), peerport);
8806         sfree(pf.shost);
8807
8808         if (realpf == NULL) {
8809             error = "Remote port is not recognised";
8810         } else {
8811             char *err;
8812
8813             if (realpf->share_ctx) {
8814                 /*
8815                  * This port forwarding is on behalf of a
8816                  * connection-sharing downstream, so abandon our own
8817                  * channel-open procedure and just pass the message on
8818                  * to sshshare.c.
8819                  */
8820                 share_got_pkt_from_server(realpf->share_ctx, pktin->type,
8821                                           pktin->body, pktin->length);
8822                 sfree(c);
8823                 return;
8824             }
8825
8826             err = pfd_connect(&c->u.pfd.pf, realpf->dhost, realpf->dport,
8827                               c, ssh->conf, realpf->pfrec->addressfamily);
8828             logeventf(ssh, "Attempting to forward remote port to "
8829                       "%s:%d", realpf->dhost, realpf->dport);
8830             if (err != NULL) {
8831                 logeventf(ssh, "Port open failed: %s", err);
8832                 sfree(err);
8833                 error = "Port open failed";
8834             } else {
8835                 logevent("Forwarded port opened successfully");
8836                 c->type = CHAN_SOCKDATA;
8837             }
8838         }
8839     } else if (typelen == 22 &&
8840                !memcmp(type, "auth-agent@openssh.com", 22)) {
8841         if (!ssh->agentfwd_enabled)
8842             error = "Agent forwarding is not enabled";
8843         else {
8844             c->type = CHAN_AGENT;       /* identify channel type */
8845             c->u.a.lensofar = 0;
8846             c->u.a.message = NULL;
8847             c->u.a.outstanding_requests = 0;
8848         }
8849     } else {
8850         error = "Unsupported channel type requested";
8851     }
8852
8853     c->remoteid = remid;
8854     c->halfopen = FALSE;
8855     if (error) {
8856         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN_FAILURE);
8857         ssh2_pkt_adduint32(pktout, c->remoteid);
8858         ssh2_pkt_adduint32(pktout, SSH2_OPEN_CONNECT_FAILED);
8859         ssh2_pkt_addstring(pktout, error);
8860         ssh2_pkt_addstring(pktout, "en");       /* language tag */
8861         ssh2_pkt_send(ssh, pktout);
8862         logeventf(ssh, "Rejected channel open: %s", error);
8863         sfree(c);
8864     } else {
8865         ssh_channel_init(c);
8866         c->v.v2.remwindow = winsize;
8867         c->v.v2.remmaxpkt = pktsize;
8868         if (our_winsize_override) {
8869             c->v.v2.locwindow = c->v.v2.locmaxwin = c->v.v2.remlocwin =
8870                 our_winsize_override;
8871         }
8872         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION);
8873         ssh2_pkt_adduint32(pktout, c->remoteid);
8874         ssh2_pkt_adduint32(pktout, c->localid);
8875         ssh2_pkt_adduint32(pktout, c->v.v2.locwindow);
8876         ssh2_pkt_adduint32(pktout, OUR_V2_MAXPKT);      /* our max pkt size */
8877         ssh2_pkt_send(ssh, pktout);
8878     }
8879 }
8880
8881 void sshfwd_x11_sharing_handover(struct ssh_channel *c,
8882                                  void *share_cs, void *share_chan,
8883                                  const char *peer_addr, int peer_port,
8884                                  int endian, int protomajor, int protominor,
8885                                  const void *initial_data, int initial_len)
8886 {
8887     /*
8888      * This function is called when we've just discovered that an X
8889      * forwarding channel on which we'd been handling the initial auth
8890      * ourselves turns out to be destined for a connection-sharing
8891      * downstream. So we turn the channel into a CHAN_SHARING, meaning
8892      * that we completely stop tracking windows and buffering data and
8893      * just pass more or less unmodified SSH messages back and forth.
8894      */
8895     c->type = CHAN_SHARING;
8896     c->u.sharing.ctx = share_cs;
8897     share_setup_x11_channel(share_cs, share_chan,
8898                             c->localid, c->remoteid, c->v.v2.remwindow,
8899                             c->v.v2.remmaxpkt, c->v.v2.locwindow,
8900                             peer_addr, peer_port, endian,
8901                             protomajor, protominor,
8902                             initial_data, initial_len);
8903 }
8904
8905 void sshfwd_x11_is_local(struct ssh_channel *c)
8906 {
8907     /*
8908      * This function is called when we've just discovered that an X
8909      * forwarding channel is _not_ destined for a connection-sharing
8910      * downstream but we're going to handle it ourselves. We stop
8911      * presenting a cautiously small window and go into ordinary data
8912      * exchange mode.
8913      */
8914     c->u.x11.initial = FALSE;
8915     ssh2_set_window(c, ssh_is_simple(c->ssh) ? OUR_V2_BIGWIN : OUR_V2_WINSIZE);
8916 }
8917
8918 /*
8919  * Buffer banner messages for later display at some convenient point,
8920  * if we're going to display them.
8921  */
8922 static void ssh2_msg_userauth_banner(Ssh ssh, struct Packet *pktin)
8923 {
8924     /* Arbitrary limit to prevent unbounded inflation of buffer */
8925     if (conf_get_int(ssh->conf, CONF_ssh_show_banner) &&
8926         bufchain_size(&ssh->banner) <= 131072) {
8927         char *banner = NULL;
8928         int size = 0;
8929         ssh_pkt_getstring(pktin, &banner, &size);
8930         if (banner)
8931             bufchain_add(&ssh->banner, banner, size);
8932     }
8933 }
8934
8935 /* Helper function to deal with sending tty modes for "pty-req" */
8936 static void ssh2_send_ttymode(void *data,
8937                               const struct ssh_ttymode *mode, char *val)
8938 {
8939     struct Packet *pktout = (struct Packet *)data;
8940     unsigned int arg = 0;
8941
8942     switch (mode->type) {
8943       case TTY_OP_CHAR:
8944         arg = ssh_tty_parse_specchar(val);
8945         break;
8946       case TTY_OP_BOOL:
8947         arg = ssh_tty_parse_boolean(val);
8948         break;
8949     }
8950     ssh2_pkt_addbyte(pktout, mode->opcode);
8951     ssh2_pkt_adduint32(pktout, arg);
8952 }
8953
8954 static void ssh2_setup_x11(struct ssh_channel *c, struct Packet *pktin,
8955                            void *ctx)
8956 {
8957     struct ssh2_setup_x11_state {
8958         int crLine;
8959     };
8960     Ssh ssh = c->ssh;
8961     struct Packet *pktout;
8962     crStateP(ssh2_setup_x11_state, ctx);
8963
8964     crBeginState;
8965
8966     logevent("Requesting X11 forwarding");
8967     pktout = ssh2_chanreq_init(ssh->mainchan, "x11-req",
8968                                ssh2_setup_x11, s);
8969     ssh2_pkt_addbool(pktout, 0);               /* many connections */
8970     ssh2_pkt_addstring(pktout, ssh->x11auth->protoname);
8971     ssh2_pkt_addstring(pktout, ssh->x11auth->datastring);
8972     ssh2_pkt_adduint32(pktout, ssh->x11disp->screennum);
8973     ssh2_pkt_send(ssh, pktout);
8974
8975     /* Wait to be called back with either a response packet, or NULL
8976      * meaning clean up and free our data */
8977     crReturnV;
8978
8979     if (pktin) {
8980         if (pktin->type == SSH2_MSG_CHANNEL_SUCCESS) {
8981             logevent("X11 forwarding enabled");
8982             ssh->X11_fwd_enabled = TRUE;
8983         } else
8984             logevent("X11 forwarding refused");
8985     }
8986
8987     crFinishFreeV;
8988 }
8989
8990 static void ssh2_setup_agent(struct ssh_channel *c, struct Packet *pktin,
8991                                    void *ctx)
8992 {
8993     struct ssh2_setup_agent_state {
8994         int crLine;
8995     };
8996     Ssh ssh = c->ssh;
8997     struct Packet *pktout;
8998     crStateP(ssh2_setup_agent_state, ctx);
8999
9000     crBeginState;
9001
9002     logevent("Requesting OpenSSH-style agent forwarding");
9003     pktout = ssh2_chanreq_init(ssh->mainchan, "auth-agent-req@openssh.com",
9004                                ssh2_setup_agent, s);
9005     ssh2_pkt_send(ssh, pktout);
9006
9007     /* Wait to be called back with either a response packet, or NULL
9008      * meaning clean up and free our data */
9009     crReturnV;
9010
9011     if (pktin) {
9012         if (pktin->type == SSH2_MSG_CHANNEL_SUCCESS) {
9013             logevent("Agent forwarding enabled");
9014             ssh->agentfwd_enabled = TRUE;
9015         } else
9016             logevent("Agent forwarding refused");
9017     }
9018
9019     crFinishFreeV;
9020 }
9021
9022 static void ssh2_setup_pty(struct ssh_channel *c, struct Packet *pktin,
9023                                  void *ctx)
9024 {
9025     struct ssh2_setup_pty_state {
9026         int crLine;
9027     };
9028     Ssh ssh = c->ssh;
9029     struct Packet *pktout;
9030     crStateP(ssh2_setup_pty_state, ctx);
9031
9032     crBeginState;
9033
9034     /* Unpick the terminal-speed string. */
9035     /* XXX perhaps we should allow no speeds to be sent. */
9036     ssh->ospeed = 38400; ssh->ispeed = 38400; /* last-resort defaults */
9037     sscanf(conf_get_str(ssh->conf, CONF_termspeed), "%d,%d", &ssh->ospeed, &ssh->ispeed);
9038     /* Build the pty request. */
9039     pktout = ssh2_chanreq_init(ssh->mainchan, "pty-req",
9040                                ssh2_setup_pty, s);
9041     ssh2_pkt_addstring(pktout, conf_get_str(ssh->conf, CONF_termtype));
9042     ssh2_pkt_adduint32(pktout, ssh->term_width);
9043     ssh2_pkt_adduint32(pktout, ssh->term_height);
9044     ssh2_pkt_adduint32(pktout, 0);             /* pixel width */
9045     ssh2_pkt_adduint32(pktout, 0);             /* pixel height */
9046     ssh2_pkt_addstring_start(pktout);
9047     parse_ttymodes(ssh, ssh2_send_ttymode, (void *)pktout);
9048     ssh2_pkt_addbyte(pktout, SSH2_TTY_OP_ISPEED);
9049     ssh2_pkt_adduint32(pktout, ssh->ispeed);
9050     ssh2_pkt_addbyte(pktout, SSH2_TTY_OP_OSPEED);
9051     ssh2_pkt_adduint32(pktout, ssh->ospeed);
9052     ssh2_pkt_addstring_data(pktout, "\0", 1); /* TTY_OP_END */
9053     ssh2_pkt_send(ssh, pktout);
9054     ssh->state = SSH_STATE_INTERMED;
9055
9056     /* Wait to be called back with either a response packet, or NULL
9057      * meaning clean up and free our data */
9058     crReturnV;
9059
9060     if (pktin) {
9061         if (pktin->type == SSH2_MSG_CHANNEL_SUCCESS) {
9062             logeventf(ssh, "Allocated pty (ospeed %dbps, ispeed %dbps)",
9063                       ssh->ospeed, ssh->ispeed);
9064             ssh->got_pty = TRUE;
9065         } else {
9066             c_write_str(ssh, "Server refused to allocate pty\r\n");
9067             ssh->editing = ssh->echoing = 1;
9068         }
9069     }
9070
9071     crFinishFreeV;
9072 }
9073
9074 static void ssh2_setup_env(struct ssh_channel *c, struct Packet *pktin,
9075                            void *ctx)
9076 {
9077     struct ssh2_setup_env_state {
9078         int crLine;
9079         int num_env, env_left, env_ok;
9080     };
9081     Ssh ssh = c->ssh;
9082     struct Packet *pktout;
9083     crStateP(ssh2_setup_env_state, ctx);
9084
9085     crBeginState;
9086
9087     /*
9088      * Send environment variables.
9089      * 
9090      * Simplest thing here is to send all the requests at once, and
9091      * then wait for a whole bunch of successes or failures.
9092      */
9093     s->num_env = 0;
9094     {
9095         char *key, *val;
9096
9097         for (val = conf_get_str_strs(ssh->conf, CONF_environmt, NULL, &key);
9098              val != NULL;
9099              val = conf_get_str_strs(ssh->conf, CONF_environmt, key, &key)) {
9100             pktout = ssh2_chanreq_init(ssh->mainchan, "env", ssh2_setup_env, s);
9101             ssh2_pkt_addstring(pktout, key);
9102             ssh2_pkt_addstring(pktout, val);
9103             ssh2_pkt_send(ssh, pktout);
9104
9105             s->num_env++;
9106         }
9107         if (s->num_env)
9108             logeventf(ssh, "Sent %d environment variables", s->num_env);
9109     }
9110
9111     if (s->num_env) {
9112         s->env_ok = 0;
9113         s->env_left = s->num_env;
9114
9115         while (s->env_left > 0) {
9116             /* Wait to be called back with either a response packet,
9117              * or NULL meaning clean up and free our data */
9118             crReturnV;
9119             if (!pktin) goto out;
9120             if (pktin->type == SSH2_MSG_CHANNEL_SUCCESS)
9121                 s->env_ok++;
9122             s->env_left--;
9123         }
9124
9125         if (s->env_ok == s->num_env) {
9126             logevent("All environment variables successfully set");
9127         } else if (s->env_ok == 0) {
9128             logevent("All environment variables refused");
9129             c_write_str(ssh, "Server refused to set environment variables\r\n");
9130         } else {
9131             logeventf(ssh, "%d environment variables refused",
9132                       s->num_env - s->env_ok);
9133             c_write_str(ssh, "Server refused to set all environment variables\r\n");
9134         }
9135     }
9136   out:;
9137     crFinishFreeV;
9138 }
9139
9140 /*
9141  * Handle the SSH-2 userauth and connection layers.
9142  */
9143 static void ssh2_msg_authconn(Ssh ssh, struct Packet *pktin)
9144 {
9145     do_ssh2_authconn(ssh, NULL, 0, pktin);
9146 }
9147
9148 static void ssh2_response_authconn(struct ssh_channel *c, struct Packet *pktin,
9149                                    void *ctx)
9150 {
9151     if (pktin)
9152         do_ssh2_authconn(c->ssh, NULL, 0, pktin);
9153 }
9154
9155 static void do_ssh2_authconn(Ssh ssh, const unsigned char *in, int inlen,
9156                              struct Packet *pktin)
9157 {
9158     struct do_ssh2_authconn_state {
9159         int crLine;
9160         enum {
9161             AUTH_TYPE_NONE,
9162                 AUTH_TYPE_PUBLICKEY,
9163                 AUTH_TYPE_PUBLICKEY_OFFER_LOUD,
9164                 AUTH_TYPE_PUBLICKEY_OFFER_QUIET,
9165                 AUTH_TYPE_PASSWORD,
9166                 AUTH_TYPE_GSSAPI,      /* always QUIET */
9167                 AUTH_TYPE_KEYBOARD_INTERACTIVE,
9168                 AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET
9169         } type;
9170         int done_service_req;
9171         int gotit, need_pw, can_pubkey, can_passwd, can_keyb_inter;
9172         int tried_pubkey_config, done_agent;
9173 #ifndef NO_GSSAPI
9174         int can_gssapi;
9175         int tried_gssapi;
9176 #endif
9177         int kbd_inter_refused;
9178         int we_are_in, userauth_success;
9179         prompts_t *cur_prompt;
9180         int num_prompts;
9181         char *username;
9182         char *password;
9183         int got_username;
9184         void *publickey_blob;
9185         int publickey_bloblen;
9186         int privatekey_available, privatekey_encrypted;
9187         char *publickey_algorithm;
9188         char *publickey_comment;
9189         unsigned char agent_request[5], *agent_response, *agentp;
9190         int agent_responselen;
9191         unsigned char *pkblob_in_agent;
9192         int keyi, nkeys;
9193         char *pkblob, *alg, *commentp;
9194         int pklen, alglen, commentlen;
9195         int siglen, retlen, len;
9196         char *q, *agentreq, *ret;
9197         struct Packet *pktout;
9198         Filename *keyfile;
9199 #ifndef NO_GSSAPI
9200         struct ssh_gss_library *gsslib;
9201         Ssh_gss_ctx gss_ctx;
9202         Ssh_gss_buf gss_buf;
9203         Ssh_gss_buf gss_rcvtok, gss_sndtok;
9204         Ssh_gss_name gss_srv_name;
9205         Ssh_gss_stat gss_stat;
9206 #endif
9207     };
9208     crState(do_ssh2_authconn_state);
9209
9210     crBeginState;
9211
9212     /* Register as a handler for all the messages this coroutine handles. */
9213     ssh->packet_dispatch[SSH2_MSG_SERVICE_ACCEPT] = ssh2_msg_authconn;
9214     ssh->packet_dispatch[SSH2_MSG_USERAUTH_REQUEST] = ssh2_msg_authconn;
9215     ssh->packet_dispatch[SSH2_MSG_USERAUTH_FAILURE] = ssh2_msg_authconn;
9216     ssh->packet_dispatch[SSH2_MSG_USERAUTH_SUCCESS] = ssh2_msg_authconn;
9217     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] = ssh2_msg_authconn;
9218     ssh->packet_dispatch[SSH2_MSG_USERAUTH_PK_OK] = ssh2_msg_authconn;
9219     /* ssh->packet_dispatch[SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ] = ssh2_msg_authconn; duplicate case value */
9220     /* ssh->packet_dispatch[SSH2_MSG_USERAUTH_INFO_REQUEST] = ssh2_msg_authconn; duplicate case value */
9221     ssh->packet_dispatch[SSH2_MSG_USERAUTH_INFO_RESPONSE] = ssh2_msg_authconn;
9222     ssh->packet_dispatch[SSH2_MSG_GLOBAL_REQUEST] = ssh2_msg_authconn;
9223     ssh->packet_dispatch[SSH2_MSG_REQUEST_SUCCESS] = ssh2_msg_authconn;
9224     ssh->packet_dispatch[SSH2_MSG_REQUEST_FAILURE] = ssh2_msg_authconn;
9225     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN] = ssh2_msg_authconn;
9226     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_CONFIRMATION] = ssh2_msg_authconn;
9227     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_FAILURE] = ssh2_msg_authconn;
9228     ssh->packet_dispatch[SSH2_MSG_CHANNEL_WINDOW_ADJUST] = ssh2_msg_authconn;
9229     ssh->packet_dispatch[SSH2_MSG_CHANNEL_DATA] = ssh2_msg_authconn;
9230     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EXTENDED_DATA] = ssh2_msg_authconn;
9231     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EOF] = ssh2_msg_authconn;
9232     ssh->packet_dispatch[SSH2_MSG_CHANNEL_CLOSE] = ssh2_msg_authconn;
9233     
9234     s->done_service_req = FALSE;
9235     s->we_are_in = s->userauth_success = FALSE;
9236     s->agent_response = NULL;
9237 #ifndef NO_GSSAPI
9238     s->tried_gssapi = FALSE;
9239 #endif
9240
9241     if (!ssh->bare_connection) {
9242         if (!conf_get_int(ssh->conf, CONF_ssh_no_userauth)) {
9243             /*
9244              * Request userauth protocol, and await a response to it.
9245              */
9246             s->pktout = ssh2_pkt_init(SSH2_MSG_SERVICE_REQUEST);
9247             ssh2_pkt_addstring(s->pktout, "ssh-userauth");
9248             ssh2_pkt_send(ssh, s->pktout);
9249             crWaitUntilV(pktin);
9250             if (pktin->type == SSH2_MSG_SERVICE_ACCEPT)
9251                 s->done_service_req = TRUE;
9252         }
9253         if (!s->done_service_req) {
9254             /*
9255              * Request connection protocol directly, without authentication.
9256              */
9257             s->pktout = ssh2_pkt_init(SSH2_MSG_SERVICE_REQUEST);
9258             ssh2_pkt_addstring(s->pktout, "ssh-connection");
9259             ssh2_pkt_send(ssh, s->pktout);
9260             crWaitUntilV(pktin);
9261             if (pktin->type == SSH2_MSG_SERVICE_ACCEPT) {
9262                 s->we_are_in = TRUE; /* no auth required */
9263             } else {
9264                 bombout(("Server refused service request"));
9265                 crStopV;
9266             }
9267         }
9268     } else {
9269         s->we_are_in = TRUE;
9270     }
9271
9272     /* Arrange to be able to deal with any BANNERs that come in.
9273      * (We do this now as packets may come in during the next bit.) */
9274     bufchain_init(&ssh->banner);
9275     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] =
9276         ssh2_msg_userauth_banner;
9277
9278     /*
9279      * Misc one-time setup for authentication.
9280      */
9281     s->publickey_blob = NULL;
9282     if (!s->we_are_in) {
9283
9284         /*
9285          * Load the public half of any configured public key file
9286          * for later use.
9287          */
9288         s->keyfile = conf_get_filename(ssh->conf, CONF_keyfile);
9289         if (!filename_is_null(s->keyfile)) {
9290             int keytype;
9291             logeventf(ssh, "Reading key file \"%.150s\"",
9292                       filename_to_str(s->keyfile));
9293             keytype = key_type(s->keyfile);
9294             if (keytype == SSH_KEYTYPE_SSH2 ||
9295                 keytype == SSH_KEYTYPE_SSH2_PUBLIC_RFC4716 ||
9296                 keytype == SSH_KEYTYPE_SSH2_PUBLIC_OPENSSH) {
9297                 const char *error;
9298                 s->publickey_blob =
9299                     ssh2_userkey_loadpub(s->keyfile,
9300                                          &s->publickey_algorithm,
9301                                          &s->publickey_bloblen, 
9302                                          &s->publickey_comment, &error);
9303                 if (s->publickey_blob) {
9304                     s->privatekey_available = (keytype == SSH_KEYTYPE_SSH2);
9305                     if (!s->privatekey_available)
9306                         logeventf(ssh, "Key file contains public key only");
9307                     s->privatekey_encrypted =
9308                         ssh2_userkey_encrypted(s->keyfile, NULL);
9309                 } else {
9310                     char *msgbuf;
9311                     logeventf(ssh, "Unable to load key (%s)", 
9312                               error);
9313                     msgbuf = dupprintf("Unable to load key file "
9314                                        "\"%.150s\" (%s)\r\n",
9315                                        filename_to_str(s->keyfile),
9316                                        error);
9317                     c_write_str(ssh, msgbuf);
9318                     sfree(msgbuf);
9319                 }
9320             } else {
9321                 char *msgbuf;
9322                 logeventf(ssh, "Unable to use this key file (%s)",
9323                           key_type_to_str(keytype));
9324                 msgbuf = dupprintf("Unable to use key file \"%.150s\""
9325                                    " (%s)\r\n",
9326                                    filename_to_str(s->keyfile),
9327                                    key_type_to_str(keytype));
9328                 c_write_str(ssh, msgbuf);
9329                 sfree(msgbuf);
9330                 s->publickey_blob = NULL;
9331             }
9332         }
9333
9334         /*
9335          * Find out about any keys Pageant has (but if there's a
9336          * public key configured, filter out all others).
9337          */
9338         s->nkeys = 0;
9339         s->agent_response = NULL;
9340         s->pkblob_in_agent = NULL;
9341         if (conf_get_int(ssh->conf, CONF_tryagent) && agent_exists()) {
9342
9343             void *r;
9344
9345             logevent("Pageant is running. Requesting keys.");
9346
9347             /* Request the keys held by the agent. */
9348             PUT_32BIT(s->agent_request, 1);
9349             s->agent_request[4] = SSH2_AGENTC_REQUEST_IDENTITIES;
9350             if (!agent_query(s->agent_request, 5, &r, &s->agent_responselen,
9351                              ssh_agent_callback, ssh)) {
9352                 do {
9353                     crReturnV;
9354                     if (pktin) {
9355                         bombout(("Unexpected data from server while"
9356                                  " waiting for agent response"));
9357                         crStopV;
9358                     }
9359                 } while (pktin || inlen > 0);
9360                 r = ssh->agent_response;
9361                 s->agent_responselen = ssh->agent_response_len;
9362             }
9363             s->agent_response = (unsigned char *) r;
9364             if (s->agent_response && s->agent_responselen >= 5 &&
9365                 s->agent_response[4] == SSH2_AGENT_IDENTITIES_ANSWER) {
9366                 int keyi;
9367                 unsigned char *p;
9368                 p = s->agent_response + 5;
9369                 s->nkeys = toint(GET_32BIT(p));
9370
9371                 /*
9372                  * Vet the Pageant response to ensure that the key
9373                  * count and blob lengths make sense.
9374                  */
9375                 if (s->nkeys < 0) {
9376                     logeventf(ssh, "Pageant response contained a negative"
9377                               " key count %d", s->nkeys);
9378                     s->nkeys = 0;
9379                     goto done_agent_query;
9380                 } else {
9381                     unsigned char *q = p + 4;
9382                     int lenleft = s->agent_responselen - 5 - 4;
9383
9384                     for (keyi = 0; keyi < s->nkeys; keyi++) {
9385                         int bloblen, commentlen;
9386                         if (lenleft < 4) {
9387                             logeventf(ssh, "Pageant response was truncated");
9388                             s->nkeys = 0;
9389                             goto done_agent_query;
9390                         }
9391                         bloblen = toint(GET_32BIT(q));
9392                         if (bloblen < 0 || bloblen > lenleft) {
9393                             logeventf(ssh, "Pageant response was truncated");
9394                             s->nkeys = 0;
9395                             goto done_agent_query;
9396                         }
9397                         lenleft -= 4 + bloblen;
9398                         q += 4 + bloblen;
9399                         commentlen = toint(GET_32BIT(q));
9400                         if (commentlen < 0 || commentlen > lenleft) {
9401                             logeventf(ssh, "Pageant response was truncated");
9402                             s->nkeys = 0;
9403                             goto done_agent_query;
9404                         }
9405                         lenleft -= 4 + commentlen;
9406                         q += 4 + commentlen;
9407                     }
9408                 }
9409
9410                 p += 4;
9411                 logeventf(ssh, "Pageant has %d SSH-2 keys", s->nkeys);
9412                 if (s->publickey_blob) {
9413                     /* See if configured key is in agent. */
9414                     for (keyi = 0; keyi < s->nkeys; keyi++) {
9415                         s->pklen = toint(GET_32BIT(p));
9416                         if (s->pklen == s->publickey_bloblen &&
9417                             !memcmp(p+4, s->publickey_blob,
9418                                     s->publickey_bloblen)) {
9419                             logeventf(ssh, "Pageant key #%d matches "
9420                                       "configured key file", keyi);
9421                             s->keyi = keyi;
9422                             s->pkblob_in_agent = p;
9423                             break;
9424                         }
9425                         p += 4 + s->pklen;
9426                         p += toint(GET_32BIT(p)) + 4; /* comment */
9427                     }
9428                     if (!s->pkblob_in_agent) {
9429                         logevent("Configured key file not in Pageant");
9430                         s->nkeys = 0;
9431                     }
9432                 }
9433             } else {
9434                 logevent("Failed to get reply from Pageant");
9435             }
9436           done_agent_query:;
9437         }
9438
9439     }
9440
9441     /*
9442      * We repeat this whole loop, including the username prompt,
9443      * until we manage a successful authentication. If the user
9444      * types the wrong _password_, they can be sent back to the
9445      * beginning to try another username, if this is configured on.
9446      * (If they specify a username in the config, they are never
9447      * asked, even if they do give a wrong password.)
9448      * 
9449      * I think this best serves the needs of
9450      * 
9451      *  - the people who have no configuration, no keys, and just
9452      *    want to try repeated (username,password) pairs until they
9453      *    type both correctly
9454      * 
9455      *  - people who have keys and configuration but occasionally
9456      *    need to fall back to passwords
9457      * 
9458      *  - people with a key held in Pageant, who might not have
9459      *    logged in to a particular machine before; so they want to
9460      *    type a username, and then _either_ their key will be
9461      *    accepted, _or_ they will type a password. If they mistype
9462      *    the username they will want to be able to get back and
9463      *    retype it!
9464      */
9465     s->got_username = FALSE;
9466     while (!s->we_are_in) {
9467         /*
9468          * Get a username.
9469          */
9470         if (s->got_username && !conf_get_int(ssh->conf, CONF_change_username)) {
9471             /*
9472              * We got a username last time round this loop, and
9473              * with change_username turned off we don't try to get
9474              * it again.
9475              */
9476         } else if ((ssh->username = get_remote_username(ssh->conf)) == NULL) {
9477             int ret; /* need not be kept over crReturn */
9478             s->cur_prompt = new_prompts(ssh->frontend);
9479             s->cur_prompt->to_server = TRUE;
9480             s->cur_prompt->name = dupstr("SSH login name");
9481             add_prompt(s->cur_prompt, dupstr("login as: "), TRUE); 
9482             ret = get_userpass_input(s->cur_prompt, NULL, 0);
9483             while (ret < 0) {
9484                 ssh->send_ok = 1;
9485                 crWaitUntilV(!pktin);
9486                 ret = get_userpass_input(s->cur_prompt, in, inlen);
9487                 ssh->send_ok = 0;
9488             }
9489             if (!ret) {
9490                 /*
9491                  * get_userpass_input() failed to get a username.
9492                  * Terminate.
9493                  */
9494                 free_prompts(s->cur_prompt);
9495                 ssh_disconnect(ssh, "No username provided", NULL, 0, TRUE);
9496                 crStopV;
9497             }
9498             ssh->username = dupstr(s->cur_prompt->prompts[0]->result);
9499             free_prompts(s->cur_prompt);
9500         } else {
9501             char *stuff;
9502             if ((flags & FLAG_VERBOSE) || (flags & FLAG_INTERACTIVE)) {
9503                 stuff = dupprintf("Using username \"%s\".\r\n", ssh->username);
9504                 c_write_str(ssh, stuff);
9505                 sfree(stuff);
9506             }
9507         }
9508         s->got_username = TRUE;
9509
9510         /*
9511          * Send an authentication request using method "none": (a)
9512          * just in case it succeeds, and (b) so that we know what
9513          * authentication methods we can usefully try next.
9514          */
9515         ssh->pkt_actx = SSH2_PKTCTX_NOAUTH;
9516
9517         s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
9518         ssh2_pkt_addstring(s->pktout, ssh->username);
9519         ssh2_pkt_addstring(s->pktout, "ssh-connection");/* service requested */
9520         ssh2_pkt_addstring(s->pktout, "none");    /* method */
9521         ssh2_pkt_send(ssh, s->pktout);
9522         s->type = AUTH_TYPE_NONE;
9523         s->gotit = FALSE;
9524         s->we_are_in = FALSE;
9525
9526         s->tried_pubkey_config = FALSE;
9527         s->kbd_inter_refused = FALSE;
9528
9529         /* Reset agent request state. */
9530         s->done_agent = FALSE;
9531         if (s->agent_response) {
9532             if (s->pkblob_in_agent) {
9533                 s->agentp = s->pkblob_in_agent;
9534             } else {
9535                 s->agentp = s->agent_response + 5 + 4;
9536                 s->keyi = 0;
9537             }
9538         }
9539
9540         while (1) {
9541             char *methods = NULL;
9542             int methlen = 0;
9543
9544             /*
9545              * Wait for the result of the last authentication request.
9546              */
9547             if (!s->gotit)
9548                 crWaitUntilV(pktin);
9549             /*
9550              * Now is a convenient point to spew any banner material
9551              * that we've accumulated. (This should ensure that when
9552              * we exit the auth loop, we haven't any left to deal
9553              * with.)
9554              */
9555             {
9556                 int size = bufchain_size(&ssh->banner);
9557                 /*
9558                  * Don't show the banner if we're operating in
9559                  * non-verbose non-interactive mode. (It's probably
9560                  * a script, which means nobody will read the
9561                  * banner _anyway_, and moreover the printing of
9562                  * the banner will screw up processing on the
9563                  * output of (say) plink.)
9564                  */
9565                 if (size && (flags & (FLAG_VERBOSE | FLAG_INTERACTIVE))) {
9566                     char *banner = snewn(size, char);
9567                     bufchain_fetch(&ssh->banner, banner, size);
9568                     c_write_untrusted(ssh, banner, size);
9569                     sfree(banner);
9570                 }
9571                 bufchain_clear(&ssh->banner);
9572             }
9573             if (pktin->type == SSH2_MSG_USERAUTH_SUCCESS) {
9574                 logevent("Access granted");
9575                 s->we_are_in = s->userauth_success = TRUE;
9576                 break;
9577             }
9578
9579             if (pktin->type != SSH2_MSG_USERAUTH_FAILURE && s->type != AUTH_TYPE_GSSAPI) {
9580                 bombout(("Strange packet received during authentication: "
9581                          "type %d", pktin->type));
9582                 crStopV;
9583             }
9584
9585             s->gotit = FALSE;
9586
9587             /*
9588              * OK, we're now sitting on a USERAUTH_FAILURE message, so
9589              * we can look at the string in it and know what we can
9590              * helpfully try next.
9591              */
9592             if (pktin->type == SSH2_MSG_USERAUTH_FAILURE) {
9593                 ssh_pkt_getstring(pktin, &methods, &methlen);
9594                 if (!ssh2_pkt_getbool(pktin)) {
9595                     /*
9596                      * We have received an unequivocal Access
9597                      * Denied. This can translate to a variety of
9598                      * messages, or no message at all.
9599                      *
9600                      * For forms of authentication which are attempted
9601                      * implicitly, by which I mean without printing
9602                      * anything in the window indicating that we're
9603                      * trying them, we should never print 'Access
9604                      * denied'.
9605                      *
9606                      * If we do print a message saying that we're
9607                      * attempting some kind of authentication, it's OK
9608                      * to print a followup message saying it failed -
9609                      * but the message may sometimes be more specific
9610                      * than simply 'Access denied'.
9611                      *
9612                      * Additionally, if we'd just tried password
9613                      * authentication, we should break out of this
9614                      * whole loop so as to go back to the username
9615                      * prompt (iff we're configured to allow
9616                      * username change attempts).
9617                      */
9618                     if (s->type == AUTH_TYPE_NONE) {
9619                         /* do nothing */
9620                     } else if (s->type == AUTH_TYPE_PUBLICKEY_OFFER_LOUD ||
9621                                s->type == AUTH_TYPE_PUBLICKEY_OFFER_QUIET) {
9622                         if (s->type == AUTH_TYPE_PUBLICKEY_OFFER_LOUD)
9623                             c_write_str(ssh, "Server refused our key\r\n");
9624                         logevent("Server refused our key");
9625                     } else if (s->type == AUTH_TYPE_PUBLICKEY) {
9626                         /* This _shouldn't_ happen except by a
9627                          * protocol bug causing client and server to
9628                          * disagree on what is a correct signature. */
9629                         c_write_str(ssh, "Server refused public-key signature"
9630                                     " despite accepting key!\r\n");
9631                         logevent("Server refused public-key signature"
9632                                  " despite accepting key!");
9633                     } else if (s->type==AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET) {
9634                         /* quiet, so no c_write */
9635                         logevent("Server refused keyboard-interactive authentication");
9636                     } else if (s->type==AUTH_TYPE_GSSAPI) {
9637                         /* always quiet, so no c_write */
9638                         /* also, the code down in the GSSAPI block has
9639                          * already logged this in the Event Log */
9640                     } else if (s->type == AUTH_TYPE_KEYBOARD_INTERACTIVE) {
9641                         logevent("Keyboard-interactive authentication failed");
9642                         c_write_str(ssh, "Access denied\r\n");
9643                     } else {
9644                         assert(s->type == AUTH_TYPE_PASSWORD);
9645                         logevent("Password authentication failed");
9646                         c_write_str(ssh, "Access denied\r\n");
9647
9648                         if (conf_get_int(ssh->conf, CONF_change_username)) {
9649                             /* XXX perhaps we should allow
9650                              * keyboard-interactive to do this too? */
9651                             s->we_are_in = FALSE;
9652                             break;
9653                         }
9654                     }
9655                 } else {
9656                     c_write_str(ssh, "Further authentication required\r\n");
9657                     logevent("Further authentication required");
9658                 }
9659
9660                 s->can_pubkey =
9661                     in_commasep_string("publickey", methods, methlen);
9662                 s->can_passwd =
9663                     in_commasep_string("password", methods, methlen);
9664                 s->can_keyb_inter = conf_get_int(ssh->conf, CONF_try_ki_auth) &&
9665                     in_commasep_string("keyboard-interactive", methods, methlen);
9666 #ifndef NO_GSSAPI
9667                 if (conf_get_int(ssh->conf, CONF_try_gssapi_auth) &&
9668                     in_commasep_string("gssapi-with-mic", methods, methlen)) {
9669                     /* Try loading the GSS libraries and see if we
9670                      * have any. */
9671                     if (!ssh->gsslibs)
9672                         ssh->gsslibs = ssh_gss_setup(ssh->conf);
9673                     s->can_gssapi = (ssh->gsslibs->nlibraries > 0);
9674                 } else {
9675                     /* No point in even bothering to try to load the
9676                      * GSS libraries, if the user configuration and
9677                      * server aren't both prepared to attempt GSSAPI
9678                      * auth in the first place. */
9679                     s->can_gssapi = FALSE;
9680                 }
9681 #endif
9682             }
9683
9684             ssh->pkt_actx = SSH2_PKTCTX_NOAUTH;
9685
9686             if (s->can_pubkey && !s->done_agent && s->nkeys) {
9687
9688                 /*
9689                  * Attempt public-key authentication using a key from Pageant.
9690                  */
9691
9692                 ssh->pkt_actx = SSH2_PKTCTX_PUBLICKEY;
9693
9694                 logeventf(ssh, "Trying Pageant key #%d", s->keyi);
9695
9696                 /* Unpack key from agent response */
9697                 s->pklen = toint(GET_32BIT(s->agentp));
9698                 s->agentp += 4;
9699                 s->pkblob = (char *)s->agentp;
9700                 s->agentp += s->pklen;
9701                 s->alglen = toint(GET_32BIT(s->pkblob));
9702                 s->alg = s->pkblob + 4;
9703                 s->commentlen = toint(GET_32BIT(s->agentp));
9704                 s->agentp += 4;
9705                 s->commentp = (char *)s->agentp;
9706                 s->agentp += s->commentlen;
9707                 /* s->agentp now points at next key, if any */
9708
9709                 /* See if server will accept it */
9710                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
9711                 ssh2_pkt_addstring(s->pktout, ssh->username);
9712                 ssh2_pkt_addstring(s->pktout, "ssh-connection");
9713                                                     /* service requested */
9714                 ssh2_pkt_addstring(s->pktout, "publickey");
9715                                                     /* method */
9716                 ssh2_pkt_addbool(s->pktout, FALSE); /* no signature included */
9717                 ssh2_pkt_addstring_start(s->pktout);
9718                 ssh2_pkt_addstring_data(s->pktout, s->alg, s->alglen);
9719                 ssh2_pkt_addstring_start(s->pktout);
9720                 ssh2_pkt_addstring_data(s->pktout, s->pkblob, s->pklen);
9721                 ssh2_pkt_send(ssh, s->pktout);
9722                 s->type = AUTH_TYPE_PUBLICKEY_OFFER_QUIET;
9723
9724                 crWaitUntilV(pktin);
9725                 if (pktin->type != SSH2_MSG_USERAUTH_PK_OK) {
9726
9727                     /* Offer of key refused. */
9728                     s->gotit = TRUE;
9729
9730                 } else {
9731                     
9732                     void *vret;
9733
9734                     if (flags & FLAG_VERBOSE) {
9735                         c_write_str(ssh, "Authenticating with "
9736                                     "public key \"");
9737                         c_write(ssh, s->commentp, s->commentlen);
9738                         c_write_str(ssh, "\" from agent\r\n");
9739                     }
9740
9741                     /*
9742                      * Server is willing to accept the key.
9743                      * Construct a SIGN_REQUEST.
9744                      */
9745                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
9746                     ssh2_pkt_addstring(s->pktout, ssh->username);
9747                     ssh2_pkt_addstring(s->pktout, "ssh-connection");
9748                                                         /* service requested */
9749                     ssh2_pkt_addstring(s->pktout, "publickey");
9750                                                         /* method */
9751                     ssh2_pkt_addbool(s->pktout, TRUE);  /* signature included */
9752                     ssh2_pkt_addstring_start(s->pktout);
9753                     ssh2_pkt_addstring_data(s->pktout, s->alg, s->alglen);
9754                     ssh2_pkt_addstring_start(s->pktout);
9755                     ssh2_pkt_addstring_data(s->pktout, s->pkblob, s->pklen);
9756
9757                     /* Ask agent for signature. */
9758                     s->siglen = s->pktout->length - 5 + 4 +
9759                         ssh->v2_session_id_len;
9760                     if (ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)
9761                         s->siglen -= 4;
9762                     s->len = 1;       /* message type */
9763                     s->len += 4 + s->pklen;     /* key blob */
9764                     s->len += 4 + s->siglen;    /* data to sign */
9765                     s->len += 4;      /* flags */
9766                     s->agentreq = snewn(4 + s->len, char);
9767                     PUT_32BIT(s->agentreq, s->len);
9768                     s->q = s->agentreq + 4;
9769                     *s->q++ = SSH2_AGENTC_SIGN_REQUEST;
9770                     PUT_32BIT(s->q, s->pklen);
9771                     s->q += 4;
9772                     memcpy(s->q, s->pkblob, s->pklen);
9773                     s->q += s->pklen;
9774                     PUT_32BIT(s->q, s->siglen);
9775                     s->q += 4;
9776                     /* Now the data to be signed... */
9777                     if (!(ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)) {
9778                         PUT_32BIT(s->q, ssh->v2_session_id_len);
9779                         s->q += 4;
9780                     }
9781                     memcpy(s->q, ssh->v2_session_id,
9782                            ssh->v2_session_id_len);
9783                     s->q += ssh->v2_session_id_len;
9784                     memcpy(s->q, s->pktout->data + 5,
9785                            s->pktout->length - 5);
9786                     s->q += s->pktout->length - 5;
9787                     /* And finally the (zero) flags word. */
9788                     PUT_32BIT(s->q, 0);
9789                     if (!agent_query(s->agentreq, s->len + 4,
9790                                      &vret, &s->retlen,
9791                                      ssh_agent_callback, ssh)) {
9792                         do {
9793                             crReturnV;
9794                             if (pktin) {
9795                                 bombout(("Unexpected data from server"
9796                                          " while waiting for agent"
9797                                          " response"));
9798                                 crStopV;
9799                             }
9800                         } while (pktin || inlen > 0);
9801                         vret = ssh->agent_response;
9802                         s->retlen = ssh->agent_response_len;
9803                     }
9804                     s->ret = vret;
9805                     sfree(s->agentreq);
9806                     if (s->ret) {
9807                         if (s->retlen >= 9 &&
9808                             s->ret[4] == SSH2_AGENT_SIGN_RESPONSE &&
9809                             GET_32BIT(s->ret + 5) <= (unsigned)(s->retlen-9)) {
9810                             logevent("Sending Pageant's response");
9811                             ssh2_add_sigblob(ssh, s->pktout,
9812                                              s->pkblob, s->pklen,
9813                                              s->ret + 9,
9814                                              GET_32BIT(s->ret + 5));
9815                             ssh2_pkt_send(ssh, s->pktout);
9816                             s->type = AUTH_TYPE_PUBLICKEY;
9817                         } else {
9818                             /* FIXME: less drastic response */
9819                             bombout(("Pageant failed to answer challenge"));
9820                             crStopV;
9821                         }
9822                     }
9823                 }
9824
9825                 /* Do we have any keys left to try? */
9826                 if (s->pkblob_in_agent) {
9827                     s->done_agent = TRUE;
9828                     s->tried_pubkey_config = TRUE;
9829                 } else {
9830                     s->keyi++;
9831                     if (s->keyi >= s->nkeys)
9832                         s->done_agent = TRUE;
9833                 }
9834
9835             } else if (s->can_pubkey && s->publickey_blob &&
9836                        s->privatekey_available && !s->tried_pubkey_config) {
9837
9838                 struct ssh2_userkey *key;   /* not live over crReturn */
9839                 char *passphrase;           /* not live over crReturn */
9840
9841                 ssh->pkt_actx = SSH2_PKTCTX_PUBLICKEY;
9842
9843                 s->tried_pubkey_config = TRUE;
9844
9845                 /*
9846                  * Try the public key supplied in the configuration.
9847                  *
9848                  * First, offer the public blob to see if the server is
9849                  * willing to accept it.
9850                  */
9851                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
9852                 ssh2_pkt_addstring(s->pktout, ssh->username);
9853                 ssh2_pkt_addstring(s->pktout, "ssh-connection");
9854                                                 /* service requested */
9855                 ssh2_pkt_addstring(s->pktout, "publickey");     /* method */
9856                 ssh2_pkt_addbool(s->pktout, FALSE);
9857                                                 /* no signature included */
9858                 ssh2_pkt_addstring(s->pktout, s->publickey_algorithm);
9859                 ssh2_pkt_addstring_start(s->pktout);
9860                 ssh2_pkt_addstring_data(s->pktout,
9861                                         (char *)s->publickey_blob,
9862                                         s->publickey_bloblen);
9863                 ssh2_pkt_send(ssh, s->pktout);
9864                 logevent("Offered public key");
9865
9866                 crWaitUntilV(pktin);
9867                 if (pktin->type != SSH2_MSG_USERAUTH_PK_OK) {
9868                     /* Key refused. Give up. */
9869                     s->gotit = TRUE; /* reconsider message next loop */
9870                     s->type = AUTH_TYPE_PUBLICKEY_OFFER_LOUD;
9871                     continue; /* process this new message */
9872                 }
9873                 logevent("Offer of public key accepted");
9874
9875                 /*
9876                  * Actually attempt a serious authentication using
9877                  * the key.
9878                  */
9879                 if (flags & FLAG_VERBOSE) {
9880                     c_write_str(ssh, "Authenticating with public key \"");
9881                     c_write_str(ssh, s->publickey_comment);
9882                     c_write_str(ssh, "\"\r\n");
9883                 }
9884                 key = NULL;
9885                 while (!key) {
9886                     const char *error;  /* not live over crReturn */
9887                     if (s->privatekey_encrypted) {
9888                         /*
9889                          * Get a passphrase from the user.
9890                          */
9891                         int ret; /* need not be kept over crReturn */
9892                         s->cur_prompt = new_prompts(ssh->frontend);
9893                         s->cur_prompt->to_server = FALSE;
9894                         s->cur_prompt->name = dupstr("SSH key passphrase");
9895                         add_prompt(s->cur_prompt,
9896                                    dupprintf("Passphrase for key \"%.100s\": ",
9897                                              s->publickey_comment),
9898                                    FALSE);
9899                         ret = get_userpass_input(s->cur_prompt, NULL, 0);
9900                         while (ret < 0) {
9901                             ssh->send_ok = 1;
9902                             crWaitUntilV(!pktin);
9903                             ret = get_userpass_input(s->cur_prompt,
9904                                                      in, inlen);
9905                             ssh->send_ok = 0;
9906                         }
9907                         if (!ret) {
9908                             /* Failed to get a passphrase. Terminate. */
9909                             free_prompts(s->cur_prompt);
9910                             ssh_disconnect(ssh, NULL,
9911                                            "Unable to authenticate",
9912                                            SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER,
9913                                            TRUE);
9914                             crStopV;
9915                         }
9916                         passphrase =
9917                             dupstr(s->cur_prompt->prompts[0]->result);
9918                         free_prompts(s->cur_prompt);
9919                     } else {
9920                         passphrase = NULL; /* no passphrase needed */
9921                     }
9922
9923                     /*
9924                      * Try decrypting the key.
9925                      */
9926                     s->keyfile = conf_get_filename(ssh->conf, CONF_keyfile);
9927                     key = ssh2_load_userkey(s->keyfile, passphrase, &error);
9928                     if (passphrase) {
9929                         /* burn the evidence */
9930                         smemclr(passphrase, strlen(passphrase));
9931                         sfree(passphrase);
9932                     }
9933                     if (key == SSH2_WRONG_PASSPHRASE || key == NULL) {
9934                         if (passphrase &&
9935                             (key == SSH2_WRONG_PASSPHRASE)) {
9936                             c_write_str(ssh, "Wrong passphrase\r\n");
9937                             key = NULL;
9938                             /* and loop again */
9939                         } else {
9940                             c_write_str(ssh, "Unable to load private key (");
9941                             c_write_str(ssh, error);
9942                             c_write_str(ssh, ")\r\n");
9943                             key = NULL;
9944                             break; /* try something else */
9945                         }
9946                     }
9947                 }
9948
9949                 if (key) {
9950                     unsigned char *pkblob, *sigblob, *sigdata;
9951                     int pkblob_len, sigblob_len, sigdata_len;
9952                     int p;
9953
9954                     /*
9955                      * We have loaded the private key and the server
9956                      * has announced that it's willing to accept it.
9957                      * Hallelujah. Generate a signature and send it.
9958                      */
9959                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
9960                     ssh2_pkt_addstring(s->pktout, ssh->username);
9961                     ssh2_pkt_addstring(s->pktout, "ssh-connection");
9962                                                     /* service requested */
9963                     ssh2_pkt_addstring(s->pktout, "publickey");
9964                                                     /* method */
9965                     ssh2_pkt_addbool(s->pktout, TRUE);
9966                                                     /* signature follows */
9967                     ssh2_pkt_addstring(s->pktout, key->alg->name);
9968                     pkblob = key->alg->public_blob(key->data,
9969                                                    &pkblob_len);
9970                     ssh2_pkt_addstring_start(s->pktout);
9971                     ssh2_pkt_addstring_data(s->pktout, (char *)pkblob,
9972                                             pkblob_len);
9973
9974                     /*
9975                      * The data to be signed is:
9976                      *
9977                      *   string  session-id
9978                      *
9979                      * followed by everything so far placed in the
9980                      * outgoing packet.
9981                      */
9982                     sigdata_len = s->pktout->length - 5 + 4 +
9983                         ssh->v2_session_id_len;
9984                     if (ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)
9985                         sigdata_len -= 4;
9986                     sigdata = snewn(sigdata_len, unsigned char);
9987                     p = 0;
9988                     if (!(ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)) {
9989                         PUT_32BIT(sigdata+p, ssh->v2_session_id_len);
9990                         p += 4;
9991                     }
9992                     memcpy(sigdata+p, ssh->v2_session_id,
9993                            ssh->v2_session_id_len);
9994                     p += ssh->v2_session_id_len;
9995                     memcpy(sigdata+p, s->pktout->data + 5,
9996                            s->pktout->length - 5);
9997                     p += s->pktout->length - 5;
9998                     assert(p == sigdata_len);
9999                     sigblob = key->alg->sign(key->data, (char *)sigdata,
10000                                              sigdata_len, &sigblob_len);
10001                     ssh2_add_sigblob(ssh, s->pktout, pkblob, pkblob_len,
10002                                      sigblob, sigblob_len);
10003                     sfree(pkblob);
10004                     sfree(sigblob);
10005                     sfree(sigdata);
10006
10007                     ssh2_pkt_send(ssh, s->pktout);
10008                     logevent("Sent public key signature");
10009                     s->type = AUTH_TYPE_PUBLICKEY;
10010                     key->alg->freekey(key->data);
10011                     sfree(key->comment);
10012                     sfree(key);
10013                 }
10014
10015 #ifndef NO_GSSAPI
10016             } else if (s->can_gssapi && !s->tried_gssapi) {
10017
10018                 /* GSSAPI Authentication */
10019
10020                 int micoffset, len;
10021                 char *data;
10022                 Ssh_gss_buf mic;
10023                 s->type = AUTH_TYPE_GSSAPI;
10024                 s->tried_gssapi = TRUE;
10025                 s->gotit = TRUE;
10026                 ssh->pkt_actx = SSH2_PKTCTX_GSSAPI;
10027
10028                 /*
10029                  * Pick the highest GSS library on the preference
10030                  * list.
10031                  */
10032                 {
10033                     int i, j;
10034                     s->gsslib = NULL;
10035                     for (i = 0; i < ngsslibs; i++) {
10036                         int want_id = conf_get_int_int(ssh->conf,
10037                                                        CONF_ssh_gsslist, i);
10038                         for (j = 0; j < ssh->gsslibs->nlibraries; j++)
10039                             if (ssh->gsslibs->libraries[j].id == want_id) {
10040                                 s->gsslib = &ssh->gsslibs->libraries[j];
10041                                 goto got_gsslib;   /* double break */
10042                             }
10043                     }
10044                     got_gsslib:
10045                     /*
10046                      * We always expect to have found something in
10047                      * the above loop: we only came here if there
10048                      * was at least one viable GSS library, and the
10049                      * preference list should always mention
10050                      * everything and only change the order.
10051                      */
10052                     assert(s->gsslib);
10053                 }
10054
10055                 if (s->gsslib->gsslogmsg)
10056                     logevent(s->gsslib->gsslogmsg);
10057
10058                 /* Sending USERAUTH_REQUEST with "gssapi-with-mic" method */
10059                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
10060                 ssh2_pkt_addstring(s->pktout, ssh->username);
10061                 ssh2_pkt_addstring(s->pktout, "ssh-connection");
10062                 ssh2_pkt_addstring(s->pktout, "gssapi-with-mic");
10063                 logevent("Attempting GSSAPI authentication");
10064
10065                 /* add mechanism info */
10066                 s->gsslib->indicate_mech(s->gsslib, &s->gss_buf);
10067
10068                 /* number of GSSAPI mechanisms */
10069                 ssh2_pkt_adduint32(s->pktout,1);
10070
10071                 /* length of OID + 2 */
10072                 ssh2_pkt_adduint32(s->pktout, s->gss_buf.length + 2);
10073                 ssh2_pkt_addbyte(s->pktout, SSH2_GSS_OIDTYPE);
10074
10075                 /* length of OID */
10076                 ssh2_pkt_addbyte(s->pktout, (unsigned char) s->gss_buf.length);
10077
10078                 ssh_pkt_adddata(s->pktout, s->gss_buf.value,
10079                                 s->gss_buf.length);
10080                 ssh2_pkt_send(ssh, s->pktout);
10081                 crWaitUntilV(pktin);
10082                 if (pktin->type != SSH2_MSG_USERAUTH_GSSAPI_RESPONSE) {
10083                     logevent("GSSAPI authentication request refused");
10084                     continue;
10085                 }
10086
10087                 /* check returned packet ... */
10088
10089                 ssh_pkt_getstring(pktin, &data, &len);
10090                 s->gss_rcvtok.value = data;
10091                 s->gss_rcvtok.length = len;
10092                 if (s->gss_rcvtok.length != s->gss_buf.length + 2 ||
10093                     ((char *)s->gss_rcvtok.value)[0] != SSH2_GSS_OIDTYPE ||
10094                     ((char *)s->gss_rcvtok.value)[1] != s->gss_buf.length ||
10095                     memcmp((char *)s->gss_rcvtok.value + 2,
10096                            s->gss_buf.value,s->gss_buf.length) ) {
10097                     logevent("GSSAPI authentication - wrong response from server");
10098                     continue;
10099                 }
10100
10101                 /* now start running */
10102                 s->gss_stat = s->gsslib->import_name(s->gsslib,
10103                                                      ssh->fullhostname,
10104                                                      &s->gss_srv_name);
10105                 if (s->gss_stat != SSH_GSS_OK) {
10106                     if (s->gss_stat == SSH_GSS_BAD_HOST_NAME)
10107                         logevent("GSSAPI import name failed - Bad service name");
10108                     else
10109                         logevent("GSSAPI import name failed");
10110                     continue;
10111                 }
10112
10113                 /* fetch TGT into GSS engine */
10114                 s->gss_stat = s->gsslib->acquire_cred(s->gsslib, &s->gss_ctx);
10115
10116                 if (s->gss_stat != SSH_GSS_OK) {
10117                     logevent("GSSAPI authentication failed to get credentials");
10118                     s->gsslib->release_name(s->gsslib, &s->gss_srv_name);
10119                     continue;
10120                 }
10121
10122                 /* initial tokens are empty */
10123                 SSH_GSS_CLEAR_BUF(&s->gss_rcvtok);
10124                 SSH_GSS_CLEAR_BUF(&s->gss_sndtok);
10125
10126                 /* now enter the loop */
10127                 do {
10128                     s->gss_stat = s->gsslib->init_sec_context
10129                         (s->gsslib,
10130                          &s->gss_ctx,
10131                          s->gss_srv_name,
10132                          conf_get_int(ssh->conf, CONF_gssapifwd),
10133                          &s->gss_rcvtok,
10134                          &s->gss_sndtok);
10135
10136                     if (s->gss_stat!=SSH_GSS_S_COMPLETE &&
10137                         s->gss_stat!=SSH_GSS_S_CONTINUE_NEEDED) {
10138                         logevent("GSSAPI authentication initialisation failed");
10139
10140                         if (s->gsslib->display_status(s->gsslib, s->gss_ctx,
10141                                                       &s->gss_buf) == SSH_GSS_OK) {
10142                             logevent(s->gss_buf.value);
10143                             sfree(s->gss_buf.value);
10144                         }
10145
10146                         break;
10147                     }
10148                     logevent("GSSAPI authentication initialised");
10149
10150                     /* Client and server now exchange tokens until GSSAPI
10151                      * no longer says CONTINUE_NEEDED */
10152
10153                     if (s->gss_sndtok.length != 0) {
10154                         s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_GSSAPI_TOKEN);
10155                         ssh_pkt_addstring_start(s->pktout);
10156                         ssh_pkt_addstring_data(s->pktout,s->gss_sndtok.value,s->gss_sndtok.length);
10157                         ssh2_pkt_send(ssh, s->pktout);
10158                         s->gsslib->free_tok(s->gsslib, &s->gss_sndtok);
10159                     }
10160
10161                     if (s->gss_stat == SSH_GSS_S_CONTINUE_NEEDED) {
10162                         crWaitUntilV(pktin);
10163                         if (pktin->type != SSH2_MSG_USERAUTH_GSSAPI_TOKEN) {
10164                             logevent("GSSAPI authentication - bad server response");
10165                             s->gss_stat = SSH_GSS_FAILURE;
10166                             break;
10167                         }
10168                         ssh_pkt_getstring(pktin, &data, &len);
10169                         s->gss_rcvtok.value = data;
10170                         s->gss_rcvtok.length = len;
10171                     }
10172                 } while (s-> gss_stat == SSH_GSS_S_CONTINUE_NEEDED);
10173
10174                 if (s->gss_stat != SSH_GSS_OK) {
10175                     s->gsslib->release_name(s->gsslib, &s->gss_srv_name);
10176                     s->gsslib->release_cred(s->gsslib, &s->gss_ctx);
10177                     continue;
10178                 }
10179                 logevent("GSSAPI authentication loop finished OK");
10180
10181                 /* Now send the MIC */
10182
10183                 s->pktout = ssh2_pkt_init(0);
10184                 micoffset = s->pktout->length;
10185                 ssh_pkt_addstring_start(s->pktout);
10186                 ssh_pkt_addstring_data(s->pktout, (char *)ssh->v2_session_id, ssh->v2_session_id_len);
10187                 ssh_pkt_addbyte(s->pktout, SSH2_MSG_USERAUTH_REQUEST);
10188                 ssh_pkt_addstring(s->pktout, ssh->username);
10189                 ssh_pkt_addstring(s->pktout, "ssh-connection");
10190                 ssh_pkt_addstring(s->pktout, "gssapi-with-mic");
10191
10192                 s->gss_buf.value = (char *)s->pktout->data + micoffset;
10193                 s->gss_buf.length = s->pktout->length - micoffset;
10194
10195                 s->gsslib->get_mic(s->gsslib, s->gss_ctx, &s->gss_buf, &mic);
10196                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_GSSAPI_MIC);
10197                 ssh_pkt_addstring_start(s->pktout);
10198                 ssh_pkt_addstring_data(s->pktout, mic.value, mic.length);
10199                 ssh2_pkt_send(ssh, s->pktout);
10200                 s->gsslib->free_mic(s->gsslib, &mic);
10201
10202                 s->gotit = FALSE;
10203
10204                 s->gsslib->release_name(s->gsslib, &s->gss_srv_name);
10205                 s->gsslib->release_cred(s->gsslib, &s->gss_ctx);
10206                 continue;
10207 #endif
10208             } else if (s->can_keyb_inter && !s->kbd_inter_refused) {
10209
10210                 /*
10211                  * Keyboard-interactive authentication.
10212                  */
10213
10214                 s->type = AUTH_TYPE_KEYBOARD_INTERACTIVE;
10215
10216                 ssh->pkt_actx = SSH2_PKTCTX_KBDINTER;
10217
10218                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
10219                 ssh2_pkt_addstring(s->pktout, ssh->username);
10220                 ssh2_pkt_addstring(s->pktout, "ssh-connection");
10221                                                         /* service requested */
10222                 ssh2_pkt_addstring(s->pktout, "keyboard-interactive");
10223                                                         /* method */
10224                 ssh2_pkt_addstring(s->pktout, "");      /* lang */
10225                 ssh2_pkt_addstring(s->pktout, "");      /* submethods */
10226                 ssh2_pkt_send(ssh, s->pktout);
10227                 
10228                 logevent("Attempting keyboard-interactive authentication");
10229
10230                 crWaitUntilV(pktin);
10231                 if (pktin->type != SSH2_MSG_USERAUTH_INFO_REQUEST) {
10232                     /* Server is not willing to do keyboard-interactive
10233                      * at all (or, bizarrely but legally, accepts the
10234                      * user without actually issuing any prompts).
10235                      * Give up on it entirely. */
10236                     s->gotit = TRUE;
10237                     s->type = AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET;
10238                     s->kbd_inter_refused = TRUE; /* don't try it again */
10239                     continue;
10240                 }
10241
10242                 /*
10243                  * Loop while the server continues to send INFO_REQUESTs.
10244                  */
10245                 while (pktin->type == SSH2_MSG_USERAUTH_INFO_REQUEST) {
10246
10247                     char *name, *inst, *lang;
10248                     int name_len, inst_len, lang_len;
10249                     int i;
10250
10251                     /*
10252                      * We've got a fresh USERAUTH_INFO_REQUEST.
10253                      * Get the preamble and start building a prompt.
10254                      */
10255                     ssh_pkt_getstring(pktin, &name, &name_len);
10256                     ssh_pkt_getstring(pktin, &inst, &inst_len);
10257                     ssh_pkt_getstring(pktin, &lang, &lang_len);
10258                     s->cur_prompt = new_prompts(ssh->frontend);
10259                     s->cur_prompt->to_server = TRUE;
10260
10261                     /*
10262                      * Get any prompt(s) from the packet.
10263                      */
10264                     s->num_prompts = ssh_pkt_getuint32(pktin);
10265                     for (i = 0; i < s->num_prompts; i++) {
10266                         char *prompt;
10267                         int prompt_len;
10268                         int echo;
10269                         static char noprompt[] =
10270                             "<server failed to send prompt>: ";
10271
10272                         ssh_pkt_getstring(pktin, &prompt, &prompt_len);
10273                         echo = ssh2_pkt_getbool(pktin);
10274                         if (!prompt_len) {
10275                             prompt = noprompt;
10276                             prompt_len = lenof(noprompt)-1;
10277                         }
10278                         add_prompt(s->cur_prompt,
10279                                    dupprintf("%.*s", prompt_len, prompt),
10280                                    echo);
10281                     }
10282
10283                     if (name_len) {
10284                         /* FIXME: better prefix to distinguish from
10285                          * local prompts? */
10286                         s->cur_prompt->name =
10287                             dupprintf("SSH server: %.*s", name_len, name);
10288                         s->cur_prompt->name_reqd = TRUE;
10289                     } else {
10290                         s->cur_prompt->name =
10291                             dupstr("SSH server authentication");
10292                         s->cur_prompt->name_reqd = FALSE;
10293                     }
10294                     /* We add a prefix to try to make it clear that a prompt
10295                      * has come from the server.
10296                      * FIXME: ugly to print "Using..." in prompt _every_
10297                      * time round. Can this be done more subtly? */
10298                     /* Special case: for reasons best known to themselves,
10299                      * some servers send k-i requests with no prompts and
10300                      * nothing to display. Keep quiet in this case. */
10301                     if (s->num_prompts || name_len || inst_len) {
10302                         s->cur_prompt->instruction =
10303                             dupprintf("Using keyboard-interactive authentication.%s%.*s",
10304                                       inst_len ? "\n" : "", inst_len, inst);
10305                         s->cur_prompt->instr_reqd = TRUE;
10306                     } else {
10307                         s->cur_prompt->instr_reqd = FALSE;
10308                     }
10309
10310                     /*
10311                      * Display any instructions, and get the user's
10312                      * response(s).
10313                      */
10314                     {
10315                         int ret; /* not live over crReturn */
10316                         ret = get_userpass_input(s->cur_prompt, NULL, 0);
10317                         while (ret < 0) {
10318                             ssh->send_ok = 1;
10319                             crWaitUntilV(!pktin);
10320                             ret = get_userpass_input(s->cur_prompt, in, inlen);
10321                             ssh->send_ok = 0;
10322                         }
10323                         if (!ret) {
10324                             /*
10325                              * Failed to get responses. Terminate.
10326                              */
10327                             free_prompts(s->cur_prompt);
10328                             ssh_disconnect(ssh, NULL, "Unable to authenticate",
10329                                            SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER,
10330                                            TRUE);
10331                             crStopV;
10332                         }
10333                     }
10334
10335                     /*
10336                      * Send the response(s) to the server.
10337                      */
10338                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_INFO_RESPONSE);
10339                     ssh2_pkt_adduint32(s->pktout, s->num_prompts);
10340                     for (i=0; i < s->num_prompts; i++) {
10341                         ssh2_pkt_addstring(s->pktout,
10342                                            s->cur_prompt->prompts[i]->result);
10343                     }
10344                     ssh2_pkt_send_with_padding(ssh, s->pktout, 256);
10345
10346                     /*
10347                      * Free the prompts structure from this iteration.
10348                      * If there's another, a new one will be allocated
10349                      * when we return to the top of this while loop.
10350                      */
10351                     free_prompts(s->cur_prompt);
10352
10353                     /*
10354                      * Get the next packet in case it's another
10355                      * INFO_REQUEST.
10356                      */
10357                     crWaitUntilV(pktin);
10358
10359                 }
10360
10361                 /*
10362                  * We should have SUCCESS or FAILURE now.
10363                  */
10364                 s->gotit = TRUE;
10365
10366             } else if (s->can_passwd) {
10367
10368                 /*
10369                  * Plain old password authentication.
10370                  */
10371                 int ret; /* not live over crReturn */
10372                 int changereq_first_time; /* not live over crReturn */
10373
10374                 ssh->pkt_actx = SSH2_PKTCTX_PASSWORD;
10375
10376                 s->cur_prompt = new_prompts(ssh->frontend);
10377                 s->cur_prompt->to_server = TRUE;
10378                 s->cur_prompt->name = dupstr("SSH password");
10379                 add_prompt(s->cur_prompt, dupprintf("%s@%s's password: ",
10380                                                     ssh->username,
10381                                                     ssh->savedhost),
10382                            FALSE);
10383
10384                 ret = get_userpass_input(s->cur_prompt, NULL, 0);
10385                 while (ret < 0) {
10386                     ssh->send_ok = 1;
10387                     crWaitUntilV(!pktin);
10388                     ret = get_userpass_input(s->cur_prompt, in, inlen);
10389                     ssh->send_ok = 0;
10390                 }
10391                 if (!ret) {
10392                     /*
10393                      * Failed to get responses. Terminate.
10394                      */
10395                     free_prompts(s->cur_prompt);
10396                     ssh_disconnect(ssh, NULL, "Unable to authenticate",
10397                                    SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER,
10398                                    TRUE);
10399                     crStopV;
10400                 }
10401                 /*
10402                  * Squirrel away the password. (We may need it later if
10403                  * asked to change it.)
10404                  */
10405                 s->password = dupstr(s->cur_prompt->prompts[0]->result);
10406                 free_prompts(s->cur_prompt);
10407
10408                 /*
10409                  * Send the password packet.
10410                  *
10411                  * We pad out the password packet to 256 bytes to make
10412                  * it harder for an attacker to find the length of the
10413                  * user's password.
10414                  *
10415                  * Anyone using a password longer than 256 bytes
10416                  * probably doesn't have much to worry about from
10417                  * people who find out how long their password is!
10418                  */
10419                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
10420                 ssh2_pkt_addstring(s->pktout, ssh->username);
10421                 ssh2_pkt_addstring(s->pktout, "ssh-connection");
10422                                                         /* service requested */
10423                 ssh2_pkt_addstring(s->pktout, "password");
10424                 ssh2_pkt_addbool(s->pktout, FALSE);
10425                 ssh2_pkt_addstring(s->pktout, s->password);
10426                 ssh2_pkt_send_with_padding(ssh, s->pktout, 256);
10427                 logevent("Sent password");
10428                 s->type = AUTH_TYPE_PASSWORD;
10429
10430                 /*
10431                  * Wait for next packet, in case it's a password change
10432                  * request.
10433                  */
10434                 crWaitUntilV(pktin);
10435                 changereq_first_time = TRUE;
10436
10437                 while (pktin->type == SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ) {
10438
10439                     /* 
10440                      * We're being asked for a new password
10441                      * (perhaps not for the first time).
10442                      * Loop until the server accepts it.
10443                      */
10444
10445                     int got_new = FALSE; /* not live over crReturn */
10446                     char *prompt;   /* not live over crReturn */
10447                     int prompt_len; /* not live over crReturn */
10448                     
10449                     {
10450                         const char *msg;
10451                         if (changereq_first_time)
10452                             msg = "Server requested password change";
10453                         else
10454                             msg = "Server rejected new password";
10455                         logevent(msg);
10456                         c_write_str(ssh, msg);
10457                         c_write_str(ssh, "\r\n");
10458                     }
10459
10460                     ssh_pkt_getstring(pktin, &prompt, &prompt_len);
10461
10462                     s->cur_prompt = new_prompts(ssh->frontend);
10463                     s->cur_prompt->to_server = TRUE;
10464                     s->cur_prompt->name = dupstr("New SSH password");
10465                     s->cur_prompt->instruction =
10466                         dupprintf("%.*s", prompt_len, NULLTOEMPTY(prompt));
10467                     s->cur_prompt->instr_reqd = TRUE;
10468                     /*
10469                      * There's no explicit requirement in the protocol
10470                      * for the "old" passwords in the original and
10471                      * password-change messages to be the same, and
10472                      * apparently some Cisco kit supports password change
10473                      * by the user entering a blank password originally
10474                      * and the real password subsequently, so,
10475                      * reluctantly, we prompt for the old password again.
10476                      *
10477                      * (On the other hand, some servers don't even bother
10478                      * to check this field.)
10479                      */
10480                     add_prompt(s->cur_prompt,
10481                                dupstr("Current password (blank for previously entered password): "),
10482                                FALSE);
10483                     add_prompt(s->cur_prompt, dupstr("Enter new password: "),
10484                                FALSE);
10485                     add_prompt(s->cur_prompt, dupstr("Confirm new password: "),
10486                                FALSE);
10487
10488                     /*
10489                      * Loop until the user manages to enter the same
10490                      * password twice.
10491                      */
10492                     while (!got_new) {
10493
10494                         ret = get_userpass_input(s->cur_prompt, NULL, 0);
10495                         while (ret < 0) {
10496                             ssh->send_ok = 1;
10497                             crWaitUntilV(!pktin);
10498                             ret = get_userpass_input(s->cur_prompt, in, inlen);
10499                             ssh->send_ok = 0;
10500                         }
10501                         if (!ret) {
10502                             /*
10503                              * Failed to get responses. Terminate.
10504                              */
10505                             /* burn the evidence */
10506                             free_prompts(s->cur_prompt);
10507                             smemclr(s->password, strlen(s->password));
10508                             sfree(s->password);
10509                             ssh_disconnect(ssh, NULL, "Unable to authenticate",
10510                                            SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER,
10511                                            TRUE);
10512                             crStopV;
10513                         }
10514
10515                         /*
10516                          * If the user specified a new original password
10517                          * (IYSWIM), overwrite any previously specified
10518                          * one.
10519                          * (A side effect is that the user doesn't have to
10520                          * re-enter it if they louse up the new password.)
10521                          */
10522                         if (s->cur_prompt->prompts[0]->result[0]) {
10523                             smemclr(s->password, strlen(s->password));
10524                                 /* burn the evidence */
10525                             sfree(s->password);
10526                             s->password =
10527                                 dupstr(s->cur_prompt->prompts[0]->result);
10528                         }
10529
10530                         /*
10531                          * Check the two new passwords match.
10532                          */
10533                         got_new = (strcmp(s->cur_prompt->prompts[1]->result,
10534                                           s->cur_prompt->prompts[2]->result)
10535                                    == 0);
10536                         if (!got_new)
10537                             /* They don't. Silly user. */
10538                             c_write_str(ssh, "Passwords do not match\r\n");
10539
10540                     }
10541
10542                     /*
10543                      * Send the new password (along with the old one).
10544                      * (see above for padding rationale)
10545                      */
10546                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
10547                     ssh2_pkt_addstring(s->pktout, ssh->username);
10548                     ssh2_pkt_addstring(s->pktout, "ssh-connection");
10549                                                         /* service requested */
10550                     ssh2_pkt_addstring(s->pktout, "password");
10551                     ssh2_pkt_addbool(s->pktout, TRUE);
10552                     ssh2_pkt_addstring(s->pktout, s->password);
10553                     ssh2_pkt_addstring(s->pktout,
10554                                        s->cur_prompt->prompts[1]->result);
10555                     free_prompts(s->cur_prompt);
10556                     ssh2_pkt_send_with_padding(ssh, s->pktout, 256);
10557                     logevent("Sent new password");
10558                     
10559                     /*
10560                      * Now see what the server has to say about it.
10561                      * (If it's CHANGEREQ again, it's not happy with the
10562                      * new password.)
10563                      */
10564                     crWaitUntilV(pktin);
10565                     changereq_first_time = FALSE;
10566
10567                 }
10568
10569                 /*
10570                  * We need to reexamine the current pktin at the top
10571                  * of the loop. Either:
10572                  *  - we weren't asked to change password at all, in
10573                  *    which case it's a SUCCESS or FAILURE with the
10574                  *    usual meaning
10575                  *  - we sent a new password, and the server was
10576                  *    either OK with it (SUCCESS or FAILURE w/partial
10577                  *    success) or unhappy with the _old_ password
10578                  *    (FAILURE w/o partial success)
10579                  * In any of these cases, we go back to the top of
10580                  * the loop and start again.
10581                  */
10582                 s->gotit = TRUE;
10583
10584                 /*
10585                  * We don't need the old password any more, in any
10586                  * case. Burn the evidence.
10587                  */
10588                 smemclr(s->password, strlen(s->password));
10589                 sfree(s->password);
10590
10591             } else {
10592                 char *str = dupprintf("No supported authentication methods available"
10593                                       " (server sent: %.*s)",
10594                                       methlen, methods);
10595
10596                 ssh_disconnect(ssh, str,
10597                                "No supported authentication methods available",
10598                                SSH2_DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE,
10599                                FALSE);
10600                 sfree(str);
10601
10602                 crStopV;
10603
10604             }
10605
10606         }
10607     }
10608     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] = NULL;
10609
10610     /* Clear up various bits and pieces from authentication. */
10611     if (s->publickey_blob) {
10612         sfree(s->publickey_algorithm);
10613         sfree(s->publickey_blob);
10614         sfree(s->publickey_comment);
10615     }
10616     if (s->agent_response)
10617         sfree(s->agent_response);
10618
10619     if (s->userauth_success && !ssh->bare_connection) {
10620         /*
10621          * We've just received USERAUTH_SUCCESS, and we haven't sent any
10622          * packets since. Signal the transport layer to consider enacting
10623          * delayed compression.
10624          *
10625          * (Relying on we_are_in is not sufficient, as
10626          * draft-miller-secsh-compression-delayed is quite clear that it
10627          * triggers on USERAUTH_SUCCESS specifically, and we_are_in can
10628          * become set for other reasons.)
10629          */
10630         do_ssh2_transport(ssh, "enabling delayed compression", -2, NULL);
10631     }
10632
10633     ssh->channels = newtree234(ssh_channelcmp);
10634
10635     /*
10636      * Set up handlers for some connection protocol messages, so we
10637      * don't have to handle them repeatedly in this coroutine.
10638      */
10639     ssh->packet_dispatch[SSH2_MSG_CHANNEL_WINDOW_ADJUST] =
10640         ssh2_msg_channel_window_adjust;
10641     ssh->packet_dispatch[SSH2_MSG_GLOBAL_REQUEST] =
10642         ssh2_msg_global_request;
10643
10644     /*
10645      * Create the main session channel.
10646      */
10647     if (conf_get_int(ssh->conf, CONF_ssh_no_shell)) {
10648         ssh->mainchan = NULL;
10649     } else {
10650         ssh->mainchan = snew(struct ssh_channel);
10651         ssh->mainchan->ssh = ssh;
10652         ssh_channel_init(ssh->mainchan);
10653
10654         if (*conf_get_str(ssh->conf, CONF_ssh_nc_host)) {
10655             /*
10656              * Just start a direct-tcpip channel and use it as the main
10657              * channel.
10658              */
10659             ssh_send_port_open(ssh->mainchan,
10660                                conf_get_str(ssh->conf, CONF_ssh_nc_host),
10661                                conf_get_int(ssh->conf, CONF_ssh_nc_port),
10662                                "main channel");
10663             ssh->ncmode = TRUE;
10664         } else {
10665             s->pktout = ssh2_chanopen_init(ssh->mainchan, "session");
10666             logevent("Opening session as main channel");
10667             ssh2_pkt_send(ssh, s->pktout);
10668             ssh->ncmode = FALSE;
10669         }
10670         crWaitUntilV(pktin);
10671         if (pktin->type != SSH2_MSG_CHANNEL_OPEN_CONFIRMATION) {
10672             bombout(("Server refused to open channel"));
10673             crStopV;
10674             /* FIXME: error data comes back in FAILURE packet */
10675         }
10676         if (ssh_pkt_getuint32(pktin) != ssh->mainchan->localid) {
10677             bombout(("Server's channel confirmation cited wrong channel"));
10678             crStopV;
10679         }
10680         ssh->mainchan->remoteid = ssh_pkt_getuint32(pktin);
10681         ssh->mainchan->halfopen = FALSE;
10682         ssh->mainchan->type = CHAN_MAINSESSION;
10683         ssh->mainchan->v.v2.remwindow = ssh_pkt_getuint32(pktin);
10684         ssh->mainchan->v.v2.remmaxpkt = ssh_pkt_getuint32(pktin);
10685         update_specials_menu(ssh->frontend);
10686         logevent("Opened main channel");
10687     }
10688
10689     /*
10690      * Now we have a channel, make dispatch table entries for
10691      * general channel-based messages.
10692      */
10693     ssh->packet_dispatch[SSH2_MSG_CHANNEL_DATA] =
10694     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EXTENDED_DATA] =
10695         ssh2_msg_channel_data;
10696     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EOF] = ssh2_msg_channel_eof;
10697     ssh->packet_dispatch[SSH2_MSG_CHANNEL_CLOSE] = ssh2_msg_channel_close;
10698     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_CONFIRMATION] =
10699         ssh2_msg_channel_open_confirmation;
10700     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_FAILURE] =
10701         ssh2_msg_channel_open_failure;
10702     ssh->packet_dispatch[SSH2_MSG_CHANNEL_REQUEST] =
10703         ssh2_msg_channel_request;
10704     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN] =
10705         ssh2_msg_channel_open;
10706     ssh->packet_dispatch[SSH2_MSG_CHANNEL_SUCCESS] = ssh2_msg_channel_response;
10707     ssh->packet_dispatch[SSH2_MSG_CHANNEL_FAILURE] = ssh2_msg_channel_response;
10708
10709     /*
10710      * Now the connection protocol is properly up and running, with
10711      * all those dispatch table entries, so it's safe to let
10712      * downstreams start trying to open extra channels through us.
10713      */
10714     if (ssh->connshare)
10715         share_activate(ssh->connshare, ssh->v_s);
10716
10717     if (ssh->mainchan && ssh_is_simple(ssh)) {
10718         /*
10719          * This message indicates to the server that we promise
10720          * not to try to run any other channel in parallel with
10721          * this one, so it's safe for it to advertise a very large
10722          * window and leave the flow control to TCP.
10723          */
10724         s->pktout = ssh2_chanreq_init(ssh->mainchan,
10725                                       "simple@putty.projects.tartarus.org",
10726                                       NULL, NULL);
10727         ssh2_pkt_send(ssh, s->pktout);
10728     }
10729
10730     /*
10731      * Enable port forwardings.
10732      */
10733     ssh_setup_portfwd(ssh, ssh->conf);
10734
10735     if (ssh->mainchan && !ssh->ncmode) {
10736         /*
10737          * Send the CHANNEL_REQUESTS for the main session channel.
10738          * Each one is handled by its own little asynchronous
10739          * co-routine.
10740          */
10741
10742         /* Potentially enable X11 forwarding. */
10743         if (conf_get_int(ssh->conf, CONF_x11_forward)) {
10744             ssh->x11disp =
10745                 x11_setup_display(conf_get_str(ssh->conf, CONF_x11_display),
10746                                   ssh->conf);
10747             if (!ssh->x11disp) {
10748                 /* FIXME: return an error message from x11_setup_display */
10749                 logevent("X11 forwarding not enabled: unable to"
10750                          " initialise X display");
10751             } else {
10752                 ssh->x11auth = x11_invent_fake_auth
10753                     (ssh->x11authtree, conf_get_int(ssh->conf, CONF_x11_auth));
10754                 ssh->x11auth->disp = ssh->x11disp;
10755
10756                 ssh2_setup_x11(ssh->mainchan, NULL, NULL);
10757             }
10758         }
10759
10760         /* Potentially enable agent forwarding. */
10761         if (ssh_agent_forwarding_permitted(ssh))
10762             ssh2_setup_agent(ssh->mainchan, NULL, NULL);
10763
10764         /* Now allocate a pty for the session. */
10765         if (!conf_get_int(ssh->conf, CONF_nopty))
10766             ssh2_setup_pty(ssh->mainchan, NULL, NULL);
10767
10768         /* Send environment variables. */
10769         ssh2_setup_env(ssh->mainchan, NULL, NULL);
10770
10771         /*
10772          * Start a shell or a remote command. We may have to attempt
10773          * this twice if the config data has provided a second choice
10774          * of command.
10775          */
10776         while (1) {
10777             int subsys;
10778             char *cmd;
10779
10780             if (ssh->fallback_cmd) {
10781                 subsys = conf_get_int(ssh->conf, CONF_ssh_subsys2);
10782                 cmd = conf_get_str(ssh->conf, CONF_remote_cmd2);
10783             } else {
10784                 subsys = conf_get_int(ssh->conf, CONF_ssh_subsys);
10785                 cmd = conf_get_str(ssh->conf, CONF_remote_cmd);
10786             }
10787
10788             if (subsys) {
10789                 s->pktout = ssh2_chanreq_init(ssh->mainchan, "subsystem",
10790                                               ssh2_response_authconn, NULL);
10791                 ssh2_pkt_addstring(s->pktout, cmd);
10792             } else if (*cmd) {
10793                 s->pktout = ssh2_chanreq_init(ssh->mainchan, "exec",
10794                                               ssh2_response_authconn, NULL);
10795                 ssh2_pkt_addstring(s->pktout, cmd);
10796             } else {
10797                 s->pktout = ssh2_chanreq_init(ssh->mainchan, "shell",
10798                                               ssh2_response_authconn, NULL);
10799             }
10800             ssh2_pkt_send(ssh, s->pktout);
10801
10802             crWaitUntilV(pktin);
10803
10804             if (pktin->type != SSH2_MSG_CHANNEL_SUCCESS) {
10805                 if (pktin->type != SSH2_MSG_CHANNEL_FAILURE) {
10806                     bombout(("Unexpected response to shell/command request:"
10807                              " packet type %d", pktin->type));
10808                     crStopV;
10809                 }
10810                 /*
10811                  * We failed to start the command. If this is the
10812                  * fallback command, we really are finished; if it's
10813                  * not, and if the fallback command exists, try falling
10814                  * back to it before complaining.
10815                  */
10816                 if (!ssh->fallback_cmd &&
10817                     *conf_get_str(ssh->conf, CONF_remote_cmd2)) {
10818                     logevent("Primary command failed; attempting fallback");
10819                     ssh->fallback_cmd = TRUE;
10820                     continue;
10821                 }
10822                 bombout(("Server refused to start a shell/command"));
10823                 crStopV;
10824             } else {
10825                 logevent("Started a shell/command");
10826             }
10827             break;
10828         }
10829     } else {
10830         ssh->editing = ssh->echoing = TRUE;
10831     }
10832
10833     ssh->state = SSH_STATE_SESSION;
10834     if (ssh->size_needed)
10835         ssh_size(ssh, ssh->term_width, ssh->term_height);
10836     if (ssh->eof_needed)
10837         ssh_special(ssh, TS_EOF);
10838
10839     /*
10840      * Transfer data!
10841      */
10842     if (ssh->ldisc)
10843         ldisc_echoedit_update(ssh->ldisc);  /* cause ldisc to notice changes */
10844     if (ssh->mainchan)
10845         ssh->send_ok = 1;
10846     while (1) {
10847         crReturnV;
10848         if (pktin) {
10849
10850             /*
10851              * _All_ the connection-layer packets we expect to
10852              * receive are now handled by the dispatch table.
10853              * Anything that reaches here must be bogus.
10854              */
10855
10856             bombout(("Strange packet received: type %d", pktin->type));
10857             crStopV;
10858         } else if (ssh->mainchan) {
10859             /*
10860              * We have spare data. Add it to the channel buffer.
10861              */
10862             ssh_send_channel_data(ssh->mainchan, (char *)in, inlen);
10863         }
10864     }
10865
10866     crFinishV;
10867 }
10868
10869 /*
10870  * Handlers for SSH-2 messages that might arrive at any moment.
10871  */
10872 static void ssh2_msg_disconnect(Ssh ssh, struct Packet *pktin)
10873 {
10874     /* log reason code in disconnect message */
10875     char *buf, *msg;
10876     int reason, msglen;
10877
10878     reason = ssh_pkt_getuint32(pktin);
10879     ssh_pkt_getstring(pktin, &msg, &msglen);
10880
10881     if (reason > 0 && reason < lenof(ssh2_disconnect_reasons)) {
10882         buf = dupprintf("Received disconnect message (%s)",
10883                         ssh2_disconnect_reasons[reason]);
10884     } else {
10885         buf = dupprintf("Received disconnect message (unknown"
10886                         " type %d)", reason);
10887     }
10888     logevent(buf);
10889     sfree(buf);
10890     buf = dupprintf("Disconnection message text: %.*s",
10891                     msglen, NULLTOEMPTY(msg));
10892     logevent(buf);
10893     bombout(("Server sent disconnect message\ntype %d (%s):\n\"%.*s\"",
10894              reason,
10895              (reason > 0 && reason < lenof(ssh2_disconnect_reasons)) ?
10896              ssh2_disconnect_reasons[reason] : "unknown",
10897              msglen, NULLTOEMPTY(msg)));
10898     sfree(buf);
10899 }
10900
10901 static void ssh2_msg_debug(Ssh ssh, struct Packet *pktin)
10902 {
10903     /* log the debug message */
10904     char *msg;
10905     int msglen;
10906
10907     /* XXX maybe we should actually take notice of the return value */
10908     ssh2_pkt_getbool(pktin);
10909     ssh_pkt_getstring(pktin, &msg, &msglen);
10910
10911     logeventf(ssh, "Remote debug message: %.*s", msglen, NULLTOEMPTY(msg));
10912 }
10913
10914 static void ssh2_msg_transport(Ssh ssh, struct Packet *pktin)
10915 {
10916     do_ssh2_transport(ssh, NULL, 0, pktin);
10917 }
10918
10919 /*
10920  * Called if we receive a packet that isn't allowed by the protocol.
10921  * This only applies to packets whose meaning PuTTY understands.
10922  * Entirely unknown packets are handled below.
10923  */
10924 static void ssh2_msg_unexpected(Ssh ssh, struct Packet *pktin)
10925 {
10926     char *buf = dupprintf("Server protocol violation: unexpected %s packet",
10927                           ssh2_pkt_type(ssh->pkt_kctx, ssh->pkt_actx,
10928                                         pktin->type));
10929     ssh_disconnect(ssh, NULL, buf, SSH2_DISCONNECT_PROTOCOL_ERROR, FALSE);
10930     sfree(buf);
10931 }
10932
10933 static void ssh2_msg_something_unimplemented(Ssh ssh, struct Packet *pktin)
10934 {
10935     struct Packet *pktout;
10936     pktout = ssh2_pkt_init(SSH2_MSG_UNIMPLEMENTED);
10937     ssh2_pkt_adduint32(pktout, pktin->sequence);
10938     /*
10939      * UNIMPLEMENTED messages MUST appear in the same order as the
10940      * messages they respond to. Hence, never queue them.
10941      */
10942     ssh2_pkt_send_noqueue(ssh, pktout);
10943 }
10944
10945 /*
10946  * Handle the top-level SSH-2 protocol.
10947  */
10948 static void ssh2_protocol_setup(Ssh ssh)
10949 {
10950     int i;
10951
10952     /*
10953      * Most messages cause SSH2_MSG_UNIMPLEMENTED.
10954      */
10955     for (i = 0; i < 256; i++)
10956         ssh->packet_dispatch[i] = ssh2_msg_something_unimplemented;
10957
10958     /*
10959      * Initially, we only accept transport messages (and a few generic
10960      * ones).  do_ssh2_authconn will add more when it starts.
10961      * Messages that are understood but not currently acceptable go to
10962      * ssh2_msg_unexpected.
10963      */
10964     ssh->packet_dispatch[SSH2_MSG_UNIMPLEMENTED] = ssh2_msg_unexpected;
10965     ssh->packet_dispatch[SSH2_MSG_SERVICE_REQUEST] = ssh2_msg_unexpected;
10966     ssh->packet_dispatch[SSH2_MSG_SERVICE_ACCEPT] = ssh2_msg_unexpected;
10967     ssh->packet_dispatch[SSH2_MSG_KEXINIT] = ssh2_msg_transport;
10968     ssh->packet_dispatch[SSH2_MSG_NEWKEYS] = ssh2_msg_transport;
10969     ssh->packet_dispatch[SSH2_MSG_KEXDH_INIT] = ssh2_msg_transport;
10970     ssh->packet_dispatch[SSH2_MSG_KEXDH_REPLY] = ssh2_msg_transport;
10971     /* ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_REQUEST] = ssh2_msg_transport; duplicate case value */
10972     /* ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_GROUP] = ssh2_msg_transport; duplicate case value */
10973     ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_INIT] = ssh2_msg_transport;
10974     ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_REPLY] = ssh2_msg_transport;
10975     ssh->packet_dispatch[SSH2_MSG_USERAUTH_REQUEST] = ssh2_msg_unexpected;
10976     ssh->packet_dispatch[SSH2_MSG_USERAUTH_FAILURE] = ssh2_msg_unexpected;
10977     ssh->packet_dispatch[SSH2_MSG_USERAUTH_SUCCESS] = ssh2_msg_unexpected;
10978     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] = ssh2_msg_unexpected;
10979     ssh->packet_dispatch[SSH2_MSG_USERAUTH_PK_OK] = ssh2_msg_unexpected;
10980     /* ssh->packet_dispatch[SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ] = ssh2_msg_unexpected; duplicate case value */
10981     /* ssh->packet_dispatch[SSH2_MSG_USERAUTH_INFO_REQUEST] = ssh2_msg_unexpected; duplicate case value */
10982     ssh->packet_dispatch[SSH2_MSG_USERAUTH_INFO_RESPONSE] = ssh2_msg_unexpected;
10983     ssh->packet_dispatch[SSH2_MSG_GLOBAL_REQUEST] = ssh2_msg_unexpected;
10984     ssh->packet_dispatch[SSH2_MSG_REQUEST_SUCCESS] = ssh2_msg_unexpected;
10985     ssh->packet_dispatch[SSH2_MSG_REQUEST_FAILURE] = ssh2_msg_unexpected;
10986     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN] = ssh2_msg_unexpected;
10987     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_CONFIRMATION] = ssh2_msg_unexpected;
10988     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_FAILURE] = ssh2_msg_unexpected;
10989     ssh->packet_dispatch[SSH2_MSG_CHANNEL_WINDOW_ADJUST] = ssh2_msg_unexpected;
10990     ssh->packet_dispatch[SSH2_MSG_CHANNEL_DATA] = ssh2_msg_unexpected;
10991     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EXTENDED_DATA] = ssh2_msg_unexpected;
10992     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EOF] = ssh2_msg_unexpected;
10993     ssh->packet_dispatch[SSH2_MSG_CHANNEL_CLOSE] = ssh2_msg_unexpected;
10994     ssh->packet_dispatch[SSH2_MSG_CHANNEL_REQUEST] = ssh2_msg_unexpected;
10995     ssh->packet_dispatch[SSH2_MSG_CHANNEL_SUCCESS] = ssh2_msg_unexpected;
10996     ssh->packet_dispatch[SSH2_MSG_CHANNEL_FAILURE] = ssh2_msg_unexpected;
10997
10998     /*
10999      * These messages have a special handler from the start.
11000      */
11001     ssh->packet_dispatch[SSH2_MSG_DISCONNECT] = ssh2_msg_disconnect;
11002     ssh->packet_dispatch[SSH2_MSG_IGNORE] = ssh_msg_ignore; /* shared with SSH-1 */
11003     ssh->packet_dispatch[SSH2_MSG_DEBUG] = ssh2_msg_debug;
11004 }
11005
11006 static void ssh2_bare_connection_protocol_setup(Ssh ssh)
11007 {
11008     int i;
11009
11010     /*
11011      * Most messages cause SSH2_MSG_UNIMPLEMENTED.
11012      */
11013     for (i = 0; i < 256; i++)
11014         ssh->packet_dispatch[i] = ssh2_msg_something_unimplemented;
11015
11016     /*
11017      * Initially, we set all ssh-connection messages to 'unexpected';
11018      * do_ssh2_authconn will fill things in properly. We also handle a
11019      * couple of messages from the transport protocol which aren't
11020      * related to key exchange (UNIMPLEMENTED, IGNORE, DEBUG,
11021      * DISCONNECT).
11022      */
11023     ssh->packet_dispatch[SSH2_MSG_GLOBAL_REQUEST] = ssh2_msg_unexpected;
11024     ssh->packet_dispatch[SSH2_MSG_REQUEST_SUCCESS] = ssh2_msg_unexpected;
11025     ssh->packet_dispatch[SSH2_MSG_REQUEST_FAILURE] = ssh2_msg_unexpected;
11026     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN] = ssh2_msg_unexpected;
11027     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_CONFIRMATION] = ssh2_msg_unexpected;
11028     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_FAILURE] = ssh2_msg_unexpected;
11029     ssh->packet_dispatch[SSH2_MSG_CHANNEL_WINDOW_ADJUST] = ssh2_msg_unexpected;
11030     ssh->packet_dispatch[SSH2_MSG_CHANNEL_DATA] = ssh2_msg_unexpected;
11031     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EXTENDED_DATA] = ssh2_msg_unexpected;
11032     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EOF] = ssh2_msg_unexpected;
11033     ssh->packet_dispatch[SSH2_MSG_CHANNEL_CLOSE] = ssh2_msg_unexpected;
11034     ssh->packet_dispatch[SSH2_MSG_CHANNEL_REQUEST] = ssh2_msg_unexpected;
11035     ssh->packet_dispatch[SSH2_MSG_CHANNEL_SUCCESS] = ssh2_msg_unexpected;
11036     ssh->packet_dispatch[SSH2_MSG_CHANNEL_FAILURE] = ssh2_msg_unexpected;
11037
11038     ssh->packet_dispatch[SSH2_MSG_UNIMPLEMENTED] = ssh2_msg_unexpected;
11039
11040     /*
11041      * These messages have a special handler from the start.
11042      */
11043     ssh->packet_dispatch[SSH2_MSG_DISCONNECT] = ssh2_msg_disconnect;
11044     ssh->packet_dispatch[SSH2_MSG_IGNORE] = ssh_msg_ignore;
11045     ssh->packet_dispatch[SSH2_MSG_DEBUG] = ssh2_msg_debug;
11046 }
11047
11048 static void ssh2_timer(void *ctx, unsigned long now)
11049 {
11050     Ssh ssh = (Ssh)ctx;
11051
11052     if (ssh->state == SSH_STATE_CLOSED)
11053         return;
11054
11055     if (!ssh->kex_in_progress && !ssh->bare_connection &&
11056         conf_get_int(ssh->conf, CONF_ssh_rekey_time) != 0 &&
11057         now == ssh->next_rekey) {
11058         do_ssh2_transport(ssh, "timeout", -1, NULL);
11059     }
11060 }
11061
11062 static void ssh2_protocol(Ssh ssh, const void *vin, int inlen,
11063                           struct Packet *pktin)
11064 {
11065     const unsigned char *in = (const unsigned char *)vin;
11066     if (ssh->state == SSH_STATE_CLOSED)
11067         return;
11068
11069     if (pktin) {
11070         ssh->incoming_data_size += pktin->encrypted_len;
11071         if (!ssh->kex_in_progress &&
11072             ssh->max_data_size != 0 &&
11073             ssh->incoming_data_size > ssh->max_data_size)
11074             do_ssh2_transport(ssh, "too much data received", -1, NULL);
11075     }
11076
11077     if (pktin)
11078         ssh->packet_dispatch[pktin->type](ssh, pktin);
11079     else if (!ssh->protocol_initial_phase_done)
11080         do_ssh2_transport(ssh, in, inlen, pktin);
11081     else
11082         do_ssh2_authconn(ssh, in, inlen, pktin);
11083 }
11084
11085 static void ssh2_bare_connection_protocol(Ssh ssh, const void *vin, int inlen,
11086                                           struct Packet *pktin)
11087 {
11088     const unsigned char *in = (const unsigned char *)vin;
11089     if (ssh->state == SSH_STATE_CLOSED)
11090         return;
11091
11092     if (pktin)
11093         ssh->packet_dispatch[pktin->type](ssh, pktin);
11094     else
11095         do_ssh2_authconn(ssh, in, inlen, pktin);
11096 }
11097
11098 static void ssh_cache_conf_values(Ssh ssh)
11099 {
11100     ssh->logomitdata = conf_get_int(ssh->conf, CONF_logomitdata);
11101 }
11102
11103 /*
11104  * Called to set up the connection.
11105  *
11106  * Returns an error message, or NULL on success.
11107  */
11108 static const char *ssh_init(void *frontend_handle, void **backend_handle,
11109                             Conf *conf,
11110                             const char *host, int port, char **realhost,
11111                             int nodelay, int keepalive)
11112 {
11113     const char *p;
11114     Ssh ssh;
11115
11116     ssh = snew(struct ssh_tag);
11117     ssh->conf = conf_copy(conf);
11118     ssh_cache_conf_values(ssh);
11119     ssh->version = 0;                  /* when not ready yet */
11120     ssh->s = NULL;
11121     ssh->cipher = NULL;
11122     ssh->v1_cipher_ctx = NULL;
11123     ssh->crcda_ctx = NULL;
11124     ssh->cscipher = NULL;
11125     ssh->cs_cipher_ctx = NULL;
11126     ssh->sccipher = NULL;
11127     ssh->sc_cipher_ctx = NULL;
11128     ssh->csmac = NULL;
11129     ssh->cs_mac_ctx = NULL;
11130     ssh->scmac = NULL;
11131     ssh->sc_mac_ctx = NULL;
11132     ssh->cscomp = NULL;
11133     ssh->cs_comp_ctx = NULL;
11134     ssh->sccomp = NULL;
11135     ssh->sc_comp_ctx = NULL;
11136     ssh->kex = NULL;
11137     ssh->kex_ctx = NULL;
11138     ssh->hostkey = NULL;
11139     ssh->hostkey_str = NULL;
11140     ssh->exitcode = -1;
11141     ssh->close_expected = FALSE;
11142     ssh->clean_exit = FALSE;
11143     ssh->state = SSH_STATE_PREPACKET;
11144     ssh->size_needed = FALSE;
11145     ssh->eof_needed = FALSE;
11146     ssh->ldisc = NULL;
11147     ssh->logctx = NULL;
11148     ssh->deferred_send_data = NULL;
11149     ssh->deferred_len = 0;
11150     ssh->deferred_size = 0;
11151     ssh->fallback_cmd = 0;
11152     ssh->pkt_kctx = SSH2_PKTCTX_NOKEX;
11153     ssh->pkt_actx = SSH2_PKTCTX_NOAUTH;
11154     ssh->x11disp = NULL;
11155     ssh->x11auth = NULL;
11156     ssh->x11authtree = newtree234(x11_authcmp);
11157     ssh->v1_compressing = FALSE;
11158     ssh->v2_outgoing_sequence = 0;
11159     ssh->ssh1_rdpkt_crstate = 0;
11160     ssh->ssh2_rdpkt_crstate = 0;
11161     ssh->ssh2_bare_rdpkt_crstate = 0;
11162     ssh->ssh_gotdata_crstate = 0;
11163     ssh->do_ssh1_connection_crstate = 0;
11164     ssh->do_ssh_init_state = NULL;
11165     ssh->do_ssh_connection_init_state = NULL;
11166     ssh->do_ssh1_login_state = NULL;
11167     ssh->do_ssh2_transport_state = NULL;
11168     ssh->do_ssh2_authconn_state = NULL;
11169     ssh->v_c = NULL;
11170     ssh->v_s = NULL;
11171     ssh->mainchan = NULL;
11172     ssh->throttled_all = 0;
11173     ssh->v1_stdout_throttling = 0;
11174     ssh->queue = NULL;
11175     ssh->queuelen = ssh->queuesize = 0;
11176     ssh->queueing = FALSE;
11177     ssh->qhead = ssh->qtail = NULL;
11178     ssh->deferred_rekey_reason = NULL;
11179     bufchain_init(&ssh->queued_incoming_data);
11180     ssh->frozen = FALSE;
11181     ssh->username = NULL;
11182     ssh->sent_console_eof = FALSE;
11183     ssh->got_pty = FALSE;
11184     ssh->bare_connection = FALSE;
11185     ssh->X11_fwd_enabled = FALSE;
11186     ssh->connshare = NULL;
11187     ssh->attempting_connshare = FALSE;
11188     ssh->session_started = FALSE;
11189     ssh->specials = NULL;
11190     ssh->n_uncert_hostkeys = 0;
11191     ssh->cross_certifying = FALSE;
11192
11193     *backend_handle = ssh;
11194
11195 #ifdef MSCRYPTOAPI
11196     if (crypto_startup() == 0)
11197         return "Microsoft high encryption pack not installed!";
11198 #endif
11199
11200     ssh->frontend = frontend_handle;
11201     ssh->term_width = conf_get_int(ssh->conf, CONF_width);
11202     ssh->term_height = conf_get_int(ssh->conf, CONF_height);
11203
11204     ssh->channels = NULL;
11205     ssh->rportfwds = NULL;
11206     ssh->portfwds = NULL;
11207
11208     ssh->send_ok = 0;
11209     ssh->editing = 0;
11210     ssh->echoing = 0;
11211     ssh->conn_throttle_count = 0;
11212     ssh->overall_bufsize = 0;
11213     ssh->fallback_cmd = 0;
11214
11215     ssh->protocol = NULL;
11216
11217     ssh->protocol_initial_phase_done = FALSE;
11218
11219     ssh->pinger = NULL;
11220
11221     ssh->incoming_data_size = ssh->outgoing_data_size =
11222         ssh->deferred_data_size = 0L;
11223     ssh->max_data_size = parse_blocksize(conf_get_str(ssh->conf,
11224                                                       CONF_ssh_rekey_data));
11225     ssh->kex_in_progress = FALSE;
11226
11227 #ifndef NO_GSSAPI
11228     ssh->gsslibs = NULL;
11229 #endif
11230
11231     random_ref(); /* do this now - may be needed by sharing setup code */
11232
11233     p = connect_to_host(ssh, host, port, realhost, nodelay, keepalive);
11234     if (p != NULL) {
11235         random_unref();
11236         return p;
11237     }
11238
11239     return NULL;
11240 }
11241
11242 static void ssh_free(void *handle)
11243 {
11244     Ssh ssh = (Ssh) handle;
11245     struct ssh_channel *c;
11246     struct ssh_rportfwd *pf;
11247     struct X11FakeAuth *auth;
11248
11249     if (ssh->v1_cipher_ctx)
11250         ssh->cipher->free_context(ssh->v1_cipher_ctx);
11251     if (ssh->cs_cipher_ctx)
11252         ssh->cscipher->free_context(ssh->cs_cipher_ctx);
11253     if (ssh->sc_cipher_ctx)
11254         ssh->sccipher->free_context(ssh->sc_cipher_ctx);
11255     if (ssh->cs_mac_ctx)
11256         ssh->csmac->free_context(ssh->cs_mac_ctx);
11257     if (ssh->sc_mac_ctx)
11258         ssh->scmac->free_context(ssh->sc_mac_ctx);
11259     if (ssh->cs_comp_ctx) {
11260         if (ssh->cscomp)
11261             ssh->cscomp->compress_cleanup(ssh->cs_comp_ctx);
11262         else
11263             zlib_compress_cleanup(ssh->cs_comp_ctx);
11264     }
11265     if (ssh->sc_comp_ctx) {
11266         if (ssh->sccomp)
11267             ssh->sccomp->decompress_cleanup(ssh->sc_comp_ctx);
11268         else
11269             zlib_decompress_cleanup(ssh->sc_comp_ctx);
11270     }
11271     if (ssh->kex_ctx)
11272         dh_cleanup(ssh->kex_ctx);
11273     sfree(ssh->savedhost);
11274
11275     while (ssh->queuelen-- > 0)
11276         ssh_free_packet(ssh->queue[ssh->queuelen]);
11277     sfree(ssh->queue);
11278
11279     while (ssh->qhead) {
11280         struct queued_handler *qh = ssh->qhead;
11281         ssh->qhead = qh->next;
11282         sfree(qh);
11283     }
11284     ssh->qhead = ssh->qtail = NULL;
11285
11286     if (ssh->channels) {
11287         while ((c = delpos234(ssh->channels, 0)) != NULL) {
11288             switch (c->type) {
11289               case CHAN_X11:
11290                 if (c->u.x11.xconn != NULL)
11291                     x11_close(c->u.x11.xconn);
11292                 break;
11293               case CHAN_SOCKDATA:
11294               case CHAN_SOCKDATA_DORMANT:
11295                 if (c->u.pfd.pf != NULL)
11296                     pfd_close(c->u.pfd.pf);
11297                 break;
11298             }
11299             if (ssh->version == 2) {
11300                 struct outstanding_channel_request *ocr, *nocr;
11301                 ocr = c->v.v2.chanreq_head;
11302                 while (ocr) {
11303                     ocr->handler(c, NULL, ocr->ctx);
11304                     nocr = ocr->next;
11305                     sfree(ocr);
11306                     ocr = nocr;
11307                 }
11308                 bufchain_clear(&c->v.v2.outbuffer);
11309             }
11310             sfree(c);
11311         }
11312         freetree234(ssh->channels);
11313         ssh->channels = NULL;
11314     }
11315
11316     if (ssh->connshare)
11317         sharestate_free(ssh->connshare);
11318
11319     if (ssh->rportfwds) {
11320         while ((pf = delpos234(ssh->rportfwds, 0)) != NULL)
11321             free_rportfwd(pf);
11322         freetree234(ssh->rportfwds);
11323         ssh->rportfwds = NULL;
11324     }
11325     sfree(ssh->deferred_send_data);
11326     if (ssh->x11disp)
11327         x11_free_display(ssh->x11disp);
11328     while ((auth = delpos234(ssh->x11authtree, 0)) != NULL)
11329         x11_free_fake_auth(auth);
11330     freetree234(ssh->x11authtree);
11331     sfree(ssh->do_ssh_init_state);
11332     sfree(ssh->do_ssh1_login_state);
11333     sfree(ssh->do_ssh2_transport_state);
11334     sfree(ssh->do_ssh2_authconn_state);
11335     sfree(ssh->v_c);
11336     sfree(ssh->v_s);
11337     sfree(ssh->fullhostname);
11338     sfree(ssh->hostkey_str);
11339     sfree(ssh->specials);
11340     if (ssh->crcda_ctx) {
11341         crcda_free_context(ssh->crcda_ctx);
11342         ssh->crcda_ctx = NULL;
11343     }
11344     if (ssh->s)
11345         ssh_do_close(ssh, TRUE);
11346     expire_timer_context(ssh);
11347     if (ssh->pinger)
11348         pinger_free(ssh->pinger);
11349     bufchain_clear(&ssh->queued_incoming_data);
11350     sfree(ssh->username);
11351     conf_free(ssh->conf);
11352 #ifndef NO_GSSAPI
11353     if (ssh->gsslibs)
11354         ssh_gss_cleanup(ssh->gsslibs);
11355 #endif
11356     sfree(ssh);
11357
11358     random_unref();
11359 }
11360
11361 /*
11362  * Reconfigure the SSH backend.
11363  */
11364 static void ssh_reconfig(void *handle, Conf *conf)
11365 {
11366     Ssh ssh = (Ssh) handle;
11367     const char *rekeying = NULL;
11368     int rekey_mandatory = FALSE;
11369     unsigned long old_max_data_size;
11370     int i, rekey_time;
11371
11372     pinger_reconfig(ssh->pinger, ssh->conf, conf);
11373     if (ssh->portfwds)
11374         ssh_setup_portfwd(ssh, conf);
11375
11376     rekey_time = conf_get_int(conf, CONF_ssh_rekey_time);
11377     if (conf_get_int(ssh->conf, CONF_ssh_rekey_time) != rekey_time &&
11378         rekey_time != 0) {
11379         unsigned long new_next = ssh->last_rekey + rekey_time*60*TICKSPERSEC;
11380         unsigned long now = GETTICKCOUNT();
11381
11382         if (now - ssh->last_rekey > rekey_time*60*TICKSPERSEC) {
11383             rekeying = "timeout shortened";
11384         } else {
11385             ssh->next_rekey = schedule_timer(new_next - now, ssh2_timer, ssh);
11386         }
11387     }
11388
11389     old_max_data_size = ssh->max_data_size;
11390     ssh->max_data_size = parse_blocksize(conf_get_str(ssh->conf,
11391                                                       CONF_ssh_rekey_data));
11392     if (old_max_data_size != ssh->max_data_size &&
11393         ssh->max_data_size != 0) {
11394         if (ssh->outgoing_data_size > ssh->max_data_size ||
11395             ssh->incoming_data_size > ssh->max_data_size)
11396             rekeying = "data limit lowered";
11397     }
11398
11399     if (conf_get_int(ssh->conf, CONF_compression) !=
11400         conf_get_int(conf, CONF_compression)) {
11401         rekeying = "compression setting changed";
11402         rekey_mandatory = TRUE;
11403     }
11404
11405     for (i = 0; i < CIPHER_MAX; i++)
11406         if (conf_get_int_int(ssh->conf, CONF_ssh_cipherlist, i) !=
11407             conf_get_int_int(conf, CONF_ssh_cipherlist, i)) {
11408         rekeying = "cipher settings changed";
11409         rekey_mandatory = TRUE;
11410     }
11411     if (conf_get_int(ssh->conf, CONF_ssh2_des_cbc) !=
11412         conf_get_int(conf, CONF_ssh2_des_cbc)) {
11413         rekeying = "cipher settings changed";
11414         rekey_mandatory = TRUE;
11415     }
11416
11417     conf_free(ssh->conf);
11418     ssh->conf = conf_copy(conf);
11419     ssh_cache_conf_values(ssh);
11420
11421     if (!ssh->bare_connection && rekeying) {
11422         if (!ssh->kex_in_progress) {
11423             do_ssh2_transport(ssh, rekeying, -1, NULL);
11424         } else if (rekey_mandatory) {
11425             ssh->deferred_rekey_reason = rekeying;
11426         }
11427     }
11428 }
11429
11430 /*
11431  * Called to send data down the SSH connection.
11432  */
11433 static int ssh_send(void *handle, const char *buf, int len)
11434 {
11435     Ssh ssh = (Ssh) handle;
11436
11437     if (ssh == NULL || ssh->s == NULL || ssh->protocol == NULL)
11438         return 0;
11439
11440     ssh->protocol(ssh, (const unsigned char *)buf, len, 0);
11441
11442     return ssh_sendbuffer(ssh);
11443 }
11444
11445 /*
11446  * Called to query the current amount of buffered stdin data.
11447  */
11448 static int ssh_sendbuffer(void *handle)
11449 {
11450     Ssh ssh = (Ssh) handle;
11451     int override_value;
11452
11453     if (ssh == NULL || ssh->s == NULL || ssh->protocol == NULL)
11454         return 0;
11455
11456     /*
11457      * If the SSH socket itself has backed up, add the total backup
11458      * size on that to any individual buffer on the stdin channel.
11459      */
11460     override_value = 0;
11461     if (ssh->throttled_all)
11462         override_value = ssh->overall_bufsize;
11463
11464     if (ssh->version == 1) {
11465         return override_value;
11466     } else if (ssh->version == 2) {
11467         if (!ssh->mainchan)
11468             return override_value;
11469         else
11470             return (override_value +
11471                     bufchain_size(&ssh->mainchan->v.v2.outbuffer));
11472     }
11473
11474     return 0;
11475 }
11476
11477 /*
11478  * Called to set the size of the window from SSH's POV.
11479  */
11480 static void ssh_size(void *handle, int width, int height)
11481 {
11482     Ssh ssh = (Ssh) handle;
11483     struct Packet *pktout;
11484
11485     ssh->term_width = width;
11486     ssh->term_height = height;
11487
11488     switch (ssh->state) {
11489       case SSH_STATE_BEFORE_SIZE:
11490       case SSH_STATE_PREPACKET:
11491       case SSH_STATE_CLOSED:
11492         break;                         /* do nothing */
11493       case SSH_STATE_INTERMED:
11494         ssh->size_needed = TRUE;       /* buffer for later */
11495         break;
11496       case SSH_STATE_SESSION:
11497         if (!conf_get_int(ssh->conf, CONF_nopty)) {
11498             if (ssh->version == 1) {
11499                 send_packet(ssh, SSH1_CMSG_WINDOW_SIZE,
11500                             PKT_INT, ssh->term_height,
11501                             PKT_INT, ssh->term_width,
11502                             PKT_INT, 0, PKT_INT, 0, PKT_END);
11503             } else if (ssh->mainchan) {
11504                 pktout = ssh2_chanreq_init(ssh->mainchan, "window-change",
11505                                            NULL, NULL);
11506                 ssh2_pkt_adduint32(pktout, ssh->term_width);
11507                 ssh2_pkt_adduint32(pktout, ssh->term_height);
11508                 ssh2_pkt_adduint32(pktout, 0);
11509                 ssh2_pkt_adduint32(pktout, 0);
11510                 ssh2_pkt_send(ssh, pktout);
11511             }
11512         }
11513         break;
11514     }
11515 }
11516
11517 /*
11518  * Return a list of the special codes that make sense in this
11519  * protocol.
11520  */
11521 static const struct telnet_special *ssh_get_specials(void *handle)
11522 {
11523     static const struct telnet_special ssh1_ignore_special[] = {
11524         {"IGNORE message", TS_NOP}
11525     };
11526     static const struct telnet_special ssh2_ignore_special[] = {
11527         {"IGNORE message", TS_NOP},
11528     };
11529     static const struct telnet_special ssh2_rekey_special[] = {
11530         {"Repeat key exchange", TS_REKEY},
11531     };
11532     static const struct telnet_special ssh2_session_specials[] = {
11533         {NULL, TS_SEP},
11534         {"Break", TS_BRK},
11535         /* These are the signal names defined by RFC 4254.
11536          * They include all the ISO C signals, but are a subset of the POSIX
11537          * required signals. */
11538         {"SIGINT (Interrupt)", TS_SIGINT},
11539         {"SIGTERM (Terminate)", TS_SIGTERM},
11540         {"SIGKILL (Kill)", TS_SIGKILL},
11541         {"SIGQUIT (Quit)", TS_SIGQUIT},
11542         {"SIGHUP (Hangup)", TS_SIGHUP},
11543         {"More signals", TS_SUBMENU},
11544           {"SIGABRT", TS_SIGABRT}, {"SIGALRM", TS_SIGALRM},
11545           {"SIGFPE",  TS_SIGFPE},  {"SIGILL",  TS_SIGILL},
11546           {"SIGPIPE", TS_SIGPIPE}, {"SIGSEGV", TS_SIGSEGV},
11547           {"SIGUSR1", TS_SIGUSR1}, {"SIGUSR2", TS_SIGUSR2},
11548         {NULL, TS_EXITMENU}
11549     };
11550     static const struct telnet_special specials_end[] = {
11551         {NULL, TS_EXITMENU}
11552     };
11553
11554     struct telnet_special *specials = NULL;
11555     int nspecials = 0, specialsize = 0;
11556
11557     Ssh ssh = (Ssh) handle;
11558
11559     sfree(ssh->specials);
11560
11561 #define ADD_SPECIALS(name) do                                           \
11562     {                                                                   \
11563         int len = lenof(name);                                          \
11564         if (nspecials + len > specialsize) {                            \
11565             specialsize = (nspecials + len) * 5 / 4 + 32;               \
11566             specials = sresize(specials, specialsize, struct telnet_special); \
11567         }                                                               \
11568         memcpy(specials+nspecials, name, len*sizeof(struct telnet_special)); \
11569         nspecials += len;                                               \
11570     } while (0)
11571
11572     if (ssh->version == 1) {
11573         /* Don't bother offering IGNORE if we've decided the remote
11574          * won't cope with it, since we wouldn't bother sending it if
11575          * asked anyway. */
11576         if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH1_IGNORE))
11577             ADD_SPECIALS(ssh1_ignore_special);
11578     } else if (ssh->version == 2) {
11579         if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH2_IGNORE))
11580             ADD_SPECIALS(ssh2_ignore_special);
11581         if (!(ssh->remote_bugs & BUG_SSH2_REKEY) && !ssh->bare_connection)
11582             ADD_SPECIALS(ssh2_rekey_special);
11583         if (ssh->mainchan)
11584             ADD_SPECIALS(ssh2_session_specials);
11585
11586         if (ssh->n_uncert_hostkeys) {
11587             static const struct telnet_special uncert_start[] = {
11588                 {NULL, TS_SEP},
11589                 {"Cache new host key type", TS_SUBMENU},
11590             };
11591             static const struct telnet_special uncert_end[] = {
11592                 {NULL, TS_EXITMENU},
11593             };
11594             int i;
11595
11596             ADD_SPECIALS(uncert_start);
11597             for (i = 0; i < ssh->n_uncert_hostkeys; i++) {
11598                 struct telnet_special uncert[1];
11599                 const struct ssh_signkey *alg =
11600                     hostkey_algs[ssh->uncert_hostkeys[i]].alg;
11601                 uncert[0].name = alg->name;
11602                 uncert[0].code = TS_LOCALSTART + ssh->uncert_hostkeys[i];
11603                 ADD_SPECIALS(uncert);
11604             }
11605             ADD_SPECIALS(uncert_end);
11606         }
11607     } /* else we're not ready yet */
11608
11609     if (nspecials)
11610         ADD_SPECIALS(specials_end);
11611
11612     ssh->specials = specials;
11613
11614     if (nspecials) {
11615         return specials;
11616     } else {
11617         return NULL;
11618     }
11619 #undef ADD_SPECIALS
11620 }
11621
11622 /*
11623  * Send special codes. TS_EOF is useful for `plink', so you
11624  * can send an EOF and collect resulting output (e.g. `plink
11625  * hostname sort').
11626  */
11627 static void ssh_special(void *handle, Telnet_Special code)
11628 {
11629     Ssh ssh = (Ssh) handle;
11630     struct Packet *pktout;
11631
11632     if (code == TS_EOF) {
11633         if (ssh->state != SSH_STATE_SESSION) {
11634             /*
11635              * Buffer the EOF in case we are pre-SESSION, so we can
11636              * send it as soon as we reach SESSION.
11637              */
11638             if (code == TS_EOF)
11639                 ssh->eof_needed = TRUE;
11640             return;
11641         }
11642         if (ssh->version == 1) {
11643             send_packet(ssh, SSH1_CMSG_EOF, PKT_END);
11644         } else if (ssh->mainchan) {
11645             sshfwd_write_eof(ssh->mainchan);
11646             ssh->send_ok = 0;          /* now stop trying to read from stdin */
11647         }
11648         logevent("Sent EOF message");
11649     } else if (code == TS_PING || code == TS_NOP) {
11650         if (ssh->state == SSH_STATE_CLOSED
11651             || ssh->state == SSH_STATE_PREPACKET) return;
11652         if (ssh->version == 1) {
11653             if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH1_IGNORE))
11654                 send_packet(ssh, SSH1_MSG_IGNORE, PKT_STR, "", PKT_END);
11655         } else {
11656             if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH2_IGNORE)) {
11657                 pktout = ssh2_pkt_init(SSH2_MSG_IGNORE);
11658                 ssh2_pkt_addstring_start(pktout);
11659                 ssh2_pkt_send_noqueue(ssh, pktout);
11660             }
11661         }
11662     } else if (code == TS_REKEY) {
11663         if (!ssh->kex_in_progress && !ssh->bare_connection &&
11664             ssh->version == 2) {
11665             do_ssh2_transport(ssh, "at user request", -1, NULL);
11666         }
11667     } else if (code >= TS_LOCALSTART) {
11668         ssh->hostkey = hostkey_algs[code - TS_LOCALSTART].alg;
11669         ssh->cross_certifying = TRUE;
11670         if (!ssh->kex_in_progress && !ssh->bare_connection &&
11671             ssh->version == 2) {
11672             do_ssh2_transport(ssh, "cross-certifying new host key", -1, NULL);
11673         }
11674     } else if (code == TS_BRK) {
11675         if (ssh->state == SSH_STATE_CLOSED
11676             || ssh->state == SSH_STATE_PREPACKET) return;
11677         if (ssh->version == 1) {
11678             logevent("Unable to send BREAK signal in SSH-1");
11679         } else if (ssh->mainchan) {
11680             pktout = ssh2_chanreq_init(ssh->mainchan, "break", NULL, NULL);
11681             ssh2_pkt_adduint32(pktout, 0);   /* default break length */
11682             ssh2_pkt_send(ssh, pktout);
11683         }
11684     } else {
11685         /* Is is a POSIX signal? */
11686         const char *signame = NULL;
11687         if (code == TS_SIGABRT) signame = "ABRT";
11688         if (code == TS_SIGALRM) signame = "ALRM";
11689         if (code == TS_SIGFPE)  signame = "FPE";
11690         if (code == TS_SIGHUP)  signame = "HUP";
11691         if (code == TS_SIGILL)  signame = "ILL";
11692         if (code == TS_SIGINT)  signame = "INT";
11693         if (code == TS_SIGKILL) signame = "KILL";
11694         if (code == TS_SIGPIPE) signame = "PIPE";
11695         if (code == TS_SIGQUIT) signame = "QUIT";
11696         if (code == TS_SIGSEGV) signame = "SEGV";
11697         if (code == TS_SIGTERM) signame = "TERM";
11698         if (code == TS_SIGUSR1) signame = "USR1";
11699         if (code == TS_SIGUSR2) signame = "USR2";
11700         /* The SSH-2 protocol does in principle support arbitrary named
11701          * signals, including signame@domain, but we don't support those. */
11702         if (signame) {
11703             /* It's a signal. */
11704             if (ssh->version == 2 && ssh->mainchan) {
11705                 pktout = ssh2_chanreq_init(ssh->mainchan, "signal", NULL, NULL);
11706                 ssh2_pkt_addstring(pktout, signame);
11707                 ssh2_pkt_send(ssh, pktout);
11708                 logeventf(ssh, "Sent signal SIG%s", signame);
11709             }
11710         } else {
11711             /* Never heard of it. Do nothing */
11712         }
11713     }
11714 }
11715
11716 void *new_sock_channel(void *handle, struct PortForwarding *pf)
11717 {
11718     Ssh ssh = (Ssh) handle;
11719     struct ssh_channel *c;
11720     c = snew(struct ssh_channel);
11721
11722     c->ssh = ssh;
11723     ssh_channel_init(c);
11724     c->halfopen = TRUE;
11725     c->type = CHAN_SOCKDATA_DORMANT;/* identify channel type */
11726     c->u.pfd.pf = pf;
11727     return c;
11728 }
11729
11730 unsigned ssh_alloc_sharing_channel(Ssh ssh, void *sharing_ctx)
11731 {
11732     struct ssh_channel *c;
11733     c = snew(struct ssh_channel);
11734
11735     c->ssh = ssh;
11736     ssh_channel_init(c);
11737     c->type = CHAN_SHARING;
11738     c->u.sharing.ctx = sharing_ctx;
11739     return c->localid;
11740 }
11741
11742 void ssh_delete_sharing_channel(Ssh ssh, unsigned localid)
11743 {
11744     struct ssh_channel *c;
11745
11746     c = find234(ssh->channels, &localid, ssh_channelfind);
11747     if (c)
11748         ssh_channel_destroy(c);
11749 }
11750
11751 void ssh_send_packet_from_downstream(Ssh ssh, unsigned id, int type,
11752                                      const void *data, int datalen,
11753                                      const char *additional_log_text)
11754 {
11755     struct Packet *pkt;
11756
11757     pkt = ssh2_pkt_init(type);
11758     pkt->downstream_id = id;
11759     pkt->additional_log_text = additional_log_text;
11760     ssh2_pkt_adddata(pkt, data, datalen);
11761     ssh2_pkt_send(ssh, pkt);
11762 }
11763
11764 /*
11765  * This is called when stdout/stderr (the entity to which
11766  * from_backend sends data) manages to clear some backlog.
11767  */
11768 static void ssh_unthrottle(void *handle, int bufsize)
11769 {
11770     Ssh ssh = (Ssh) handle;
11771
11772     if (ssh->version == 1) {
11773         if (ssh->v1_stdout_throttling && bufsize < SSH1_BUFFER_LIMIT) {
11774             ssh->v1_stdout_throttling = 0;
11775             ssh_throttle_conn(ssh, -1);
11776         }
11777     } else {
11778         if (ssh->mainchan)
11779             ssh_channel_unthrottle(ssh->mainchan, bufsize);
11780     }
11781
11782     /*
11783      * Now process any SSH connection data that was stashed in our
11784      * queue while we were frozen.
11785      */
11786     ssh_process_queued_incoming_data(ssh);
11787 }
11788
11789 void ssh_send_port_open(void *channel, const char *hostname, int port,
11790                         const char *org)
11791 {
11792     struct ssh_channel *c = (struct ssh_channel *)channel;
11793     Ssh ssh = c->ssh;
11794     struct Packet *pktout;
11795
11796     logeventf(ssh, "Opening connection to %s:%d for %s", hostname, port, org);
11797
11798     if (ssh->version == 1) {
11799         send_packet(ssh, SSH1_MSG_PORT_OPEN,
11800                     PKT_INT, c->localid,
11801                     PKT_STR, hostname,
11802                     PKT_INT, port,
11803                     /* PKT_STR, <org:orgport>, */
11804                     PKT_END);
11805     } else {
11806         pktout = ssh2_chanopen_init(c, "direct-tcpip");
11807         {
11808             char *trimmed_host = host_strduptrim(hostname);
11809             ssh2_pkt_addstring(pktout, trimmed_host);
11810             sfree(trimmed_host);
11811         }
11812         ssh2_pkt_adduint32(pktout, port);
11813         /*
11814          * We make up values for the originator data; partly it's
11815          * too much hassle to keep track, and partly I'm not
11816          * convinced the server should be told details like that
11817          * about my local network configuration.
11818          * The "originator IP address" is syntactically a numeric
11819          * IP address, and some servers (e.g., Tectia) get upset
11820          * if it doesn't match this syntax.
11821          */
11822         ssh2_pkt_addstring(pktout, "0.0.0.0");
11823         ssh2_pkt_adduint32(pktout, 0);
11824         ssh2_pkt_send(ssh, pktout);
11825     }
11826 }
11827
11828 static int ssh_connected(void *handle)
11829 {
11830     Ssh ssh = (Ssh) handle;
11831     return ssh->s != NULL;
11832 }
11833
11834 static int ssh_sendok(void *handle)
11835 {
11836     Ssh ssh = (Ssh) handle;
11837     return ssh->send_ok;
11838 }
11839
11840 static int ssh_ldisc(void *handle, int option)
11841 {
11842     Ssh ssh = (Ssh) handle;
11843     if (option == LD_ECHO)
11844         return ssh->echoing;
11845     if (option == LD_EDIT)
11846         return ssh->editing;
11847     return FALSE;
11848 }
11849
11850 static void ssh_provide_ldisc(void *handle, void *ldisc)
11851 {
11852     Ssh ssh = (Ssh) handle;
11853     ssh->ldisc = ldisc;
11854 }
11855
11856 static void ssh_provide_logctx(void *handle, void *logctx)
11857 {
11858     Ssh ssh = (Ssh) handle;
11859     ssh->logctx = logctx;
11860 }
11861
11862 static int ssh_return_exitcode(void *handle)
11863 {
11864     Ssh ssh = (Ssh) handle;
11865     if (ssh->s != NULL)
11866         return -1;
11867     else
11868         return (ssh->exitcode >= 0 ? ssh->exitcode : INT_MAX);
11869 }
11870
11871 /*
11872  * cfg_info for SSH is the protocol running in this session.
11873  * (1 or 2 for the full SSH-1 or SSH-2 protocol; -1 for the bare
11874  * SSH-2 connection protocol, i.e. a downstream; 0 for not-decided-yet.)
11875  */
11876 static int ssh_cfg_info(void *handle)
11877 {
11878     Ssh ssh = (Ssh) handle;
11879     if (ssh->version == 0)
11880         return 0; /* don't know yet */
11881     else if (ssh->bare_connection)
11882         return -1;
11883     else
11884         return ssh->version;
11885 }
11886
11887 /*
11888  * Gross hack: pscp will try to start SFTP but fall back to scp1 if
11889  * that fails. This variable is the means by which scp.c can reach
11890  * into the SSH code and find out which one it got.
11891  */
11892 extern int ssh_fallback_cmd(void *handle)
11893 {
11894     Ssh ssh = (Ssh) handle;
11895     return ssh->fallback_cmd;
11896 }
11897
11898 Backend ssh_backend = {
11899     ssh_init,
11900     ssh_free,
11901     ssh_reconfig,
11902     ssh_send,
11903     ssh_sendbuffer,
11904     ssh_size,
11905     ssh_special,
11906     ssh_get_specials,
11907     ssh_connected,
11908     ssh_return_exitcode,
11909     ssh_sendok,
11910     ssh_ldisc,
11911     ssh_provide_ldisc,
11912     ssh_provide_logctx,
11913     ssh_unthrottle,
11914     ssh_cfg_info,
11915     ssh_test_for_upstream,
11916     "ssh",
11917     PROT_SSH,
11918     22
11919 };