]> asedeno.scripts.mit.edu Git - PuTTY.git/blob - ssh.c
Switch SSH-1 channel message handlers to use ssh_channel_msg().
[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 ? "half-open" : "nonexistent", localid);
7993         ssh_disconnect(ssh, NULL, buf, SSH2_DISCONNECT_PROTOCOL_ERROR, FALSE);
7994         sfree(buf);
7995         return NULL;
7996     }
7997     return c;
7998 }
7999
8000 static void ssh2_handle_winadj_response(struct ssh_channel *c,
8001                                         struct Packet *pktin, void *ctx)
8002 {
8003     unsigned *sizep = ctx;
8004
8005     /*
8006      * Winadj responses should always be failures. However, at least
8007      * one server ("boks_sshd") is known to return SUCCESS for channel
8008      * requests it's never heard of, such as "winadj@putty". Raised
8009      * with foxt.com as bug 090916-090424, but for the sake of a quiet
8010      * life, we don't worry about what kind of response we got.
8011      */
8012
8013     c->v.v2.remlocwin += *sizep;
8014     sfree(sizep);
8015     /*
8016      * winadj messages are only sent when the window is fully open, so
8017      * if we get an ack of one, we know any pending unthrottle is
8018      * complete.
8019      */
8020     if (c->v.v2.throttle_state == UNTHROTTLING)
8021         c->v.v2.throttle_state = UNTHROTTLED;
8022 }
8023
8024 static void ssh2_msg_channel_response(Ssh ssh, struct Packet *pktin)
8025 {
8026     struct ssh_channel *c = ssh_channel_msg(ssh, pktin);
8027     struct outstanding_channel_request *ocr;
8028
8029     if (!c) return;
8030     if (c->type == CHAN_SHARING) {
8031         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8032                                   pktin->body, pktin->length);
8033         return;
8034     }
8035     ocr = c->v.v2.chanreq_head;
8036     if (!ocr) {
8037         ssh2_msg_unexpected(ssh, pktin);
8038         return;
8039     }
8040     ocr->handler(c, pktin, ocr->ctx);
8041     c->v.v2.chanreq_head = ocr->next;
8042     sfree(ocr);
8043     /*
8044      * We may now initiate channel-closing procedures, if that
8045      * CHANNEL_REQUEST was the last thing outstanding before we send
8046      * CHANNEL_CLOSE.
8047      */
8048     ssh2_channel_check_close(c);
8049 }
8050
8051 static void ssh2_msg_channel_window_adjust(Ssh ssh, struct Packet *pktin)
8052 {
8053     struct ssh_channel *c;
8054     c = ssh_channel_msg(ssh, pktin);
8055     if (!c)
8056         return;
8057     if (c->type == CHAN_SHARING) {
8058         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8059                                   pktin->body, pktin->length);
8060         return;
8061     }
8062     if (!(c->closes & CLOSES_SENT_EOF)) {
8063         c->v.v2.remwindow += ssh_pkt_getuint32(pktin);
8064         ssh2_try_send_and_unthrottle(ssh, c);
8065     }
8066 }
8067
8068 static void ssh2_msg_channel_data(Ssh ssh, struct Packet *pktin)
8069 {
8070     char *data;
8071     int length;
8072     struct ssh_channel *c;
8073     c = ssh_channel_msg(ssh, pktin);
8074     if (!c)
8075         return;
8076     if (c->type == CHAN_SHARING) {
8077         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8078                                   pktin->body, pktin->length);
8079         return;
8080     }
8081     if (pktin->type == SSH2_MSG_CHANNEL_EXTENDED_DATA &&
8082         ssh_pkt_getuint32(pktin) != SSH2_EXTENDED_DATA_STDERR)
8083         return;                        /* extended but not stderr */
8084     ssh_pkt_getstring(pktin, &data, &length);
8085     if (data) {
8086         int bufsize;
8087         c->v.v2.locwindow -= length;
8088         c->v.v2.remlocwin -= length;
8089         bufsize = ssh_channel_data(c, pktin->type ==
8090                                    SSH2_MSG_CHANNEL_EXTENDED_DATA,
8091                                    data, length);
8092         /*
8093          * If it looks like the remote end hit the end of its window,
8094          * and we didn't want it to do that, think about using a
8095          * larger window.
8096          */
8097         if (c->v.v2.remlocwin <= 0 && c->v.v2.throttle_state == UNTHROTTLED &&
8098             c->v.v2.locmaxwin < 0x40000000)
8099             c->v.v2.locmaxwin += OUR_V2_WINSIZE;
8100         /*
8101          * If we are not buffering too much data,
8102          * enlarge the window again at the remote side.
8103          * If we are buffering too much, we may still
8104          * need to adjust the window if the server's
8105          * sent excess data.
8106          */
8107         if (bufsize < c->v.v2.locmaxwin)
8108             ssh2_set_window(c, c->v.v2.locmaxwin - bufsize);
8109         /*
8110          * If we're either buffering way too much data, or if we're
8111          * buffering anything at all and we're in "simple" mode,
8112          * throttle the whole channel.
8113          */
8114         if ((bufsize > c->v.v2.locmaxwin || (ssh_is_simple(ssh) && bufsize>0))
8115             && !c->throttling_conn) {
8116             c->throttling_conn = 1;
8117             ssh_throttle_conn(ssh, +1);
8118         }
8119     }
8120 }
8121
8122 static void ssh_check_termination(Ssh ssh)
8123 {
8124     if (ssh->version == 2 &&
8125         !conf_get_int(ssh->conf, CONF_ssh_no_shell) &&
8126         (ssh->channels && count234(ssh->channels) == 0) &&
8127         !(ssh->connshare && share_ndownstreams(ssh->connshare) > 0)) {
8128         /*
8129          * We used to send SSH_MSG_DISCONNECT here, because I'd
8130          * believed that _every_ conforming SSH-2 connection had to
8131          * end with a disconnect being sent by at least one side;
8132          * apparently I was wrong and it's perfectly OK to
8133          * unceremoniously slam the connection shut when you're done,
8134          * and indeed OpenSSH feels this is more polite than sending a
8135          * DISCONNECT. So now we don't.
8136          */
8137         ssh_disconnect(ssh, "All channels closed", NULL, 0, TRUE);
8138     }
8139 }
8140
8141 void ssh_sharing_downstream_connected(Ssh ssh, unsigned id,
8142                                       const char *peerinfo)
8143 {
8144     if (peerinfo)
8145         logeventf(ssh, "Connection sharing downstream #%u connected from %s",
8146                   id, peerinfo);
8147     else
8148         logeventf(ssh, "Connection sharing downstream #%u connected", id);
8149 }
8150
8151 void ssh_sharing_downstream_disconnected(Ssh ssh, unsigned id)
8152 {
8153     logeventf(ssh, "Connection sharing downstream #%u disconnected", id);
8154     ssh_check_termination(ssh);
8155 }
8156
8157 void ssh_sharing_logf(Ssh ssh, unsigned id, const char *logfmt, ...)
8158 {
8159     va_list ap;
8160     char *buf;
8161
8162     va_start(ap, logfmt);
8163     buf = dupvprintf(logfmt, ap);
8164     va_end(ap);
8165     if (id)
8166         logeventf(ssh, "Connection sharing downstream #%u: %s", id, buf);
8167     else
8168         logeventf(ssh, "Connection sharing: %s", buf);
8169     sfree(buf);
8170 }
8171
8172 static void ssh_channel_destroy(struct ssh_channel *c)
8173 {
8174     Ssh ssh = c->ssh;
8175
8176     switch (c->type) {
8177       case CHAN_MAINSESSION:
8178         ssh->mainchan = NULL;
8179         update_specials_menu(ssh->frontend);
8180         break;
8181       case CHAN_X11:
8182         if (c->u.x11.xconn != NULL)
8183             x11_close(c->u.x11.xconn);
8184         logevent("Forwarded X11 connection terminated");
8185         break;
8186       case CHAN_AGENT:
8187         sfree(c->u.a.message);
8188         break;
8189       case CHAN_SOCKDATA:
8190         if (c->u.pfd.pf != NULL)
8191             pfd_close(c->u.pfd.pf);
8192         logevent("Forwarded port closed");
8193         break;
8194     }
8195
8196     del234(ssh->channels, c);
8197     if (ssh->version == 2) {
8198         bufchain_clear(&c->v.v2.outbuffer);
8199         assert(c->v.v2.chanreq_head == NULL);
8200     }
8201     sfree(c);
8202
8203     /*
8204      * If that was the last channel left open, we might need to
8205      * terminate.
8206      */
8207     ssh_check_termination(ssh);
8208 }
8209
8210 static void ssh2_channel_check_close(struct ssh_channel *c)
8211 {
8212     Ssh ssh = c->ssh;
8213     struct Packet *pktout;
8214
8215     assert(ssh->version == 2);
8216     if (c->halfopen) {
8217         /*
8218          * If we've sent out our own CHANNEL_OPEN but not yet seen
8219          * either OPEN_CONFIRMATION or OPEN_FAILURE in response, then
8220          * it's too early to be sending close messages of any kind.
8221          */
8222         return;
8223     }
8224
8225     if ((!((CLOSES_SENT_EOF | CLOSES_RCVD_EOF) & ~c->closes) ||
8226          c->type == CHAN_ZOMBIE) &&
8227         !c->v.v2.chanreq_head &&
8228         !(c->closes & CLOSES_SENT_CLOSE)) {
8229         /*
8230          * We have both sent and received EOF (or the channel is a
8231          * zombie), and we have no outstanding channel requests, which
8232          * means the channel is in final wind-up. But we haven't sent
8233          * CLOSE, so let's do so now.
8234          */
8235         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_CLOSE);
8236         ssh2_pkt_adduint32(pktout, c->remoteid);
8237         ssh2_pkt_send(ssh, pktout);
8238         c->closes |= CLOSES_SENT_EOF | CLOSES_SENT_CLOSE;
8239     }
8240
8241     if (!((CLOSES_SENT_CLOSE | CLOSES_RCVD_CLOSE) & ~c->closes)) {
8242         assert(c->v.v2.chanreq_head == NULL);
8243         /*
8244          * We have both sent and received CLOSE, which means we're
8245          * completely done with the channel.
8246          */
8247         ssh_channel_destroy(c);
8248     }
8249 }
8250
8251 static void ssh2_channel_got_eof(struct ssh_channel *c)
8252 {
8253     if (c->closes & CLOSES_RCVD_EOF)
8254         return;                        /* already seen EOF */
8255     c->closes |= CLOSES_RCVD_EOF;
8256
8257     if (c->type == CHAN_X11) {
8258         x11_send_eof(c->u.x11.xconn);
8259     } else if (c->type == CHAN_AGENT) {
8260         if (c->u.a.outstanding_requests == 0) {
8261             /* Manufacture an outgoing EOF in response to the incoming one. */
8262             sshfwd_write_eof(c);
8263         }
8264     } else if (c->type == CHAN_SOCKDATA) {
8265         pfd_send_eof(c->u.pfd.pf);
8266     } else if (c->type == CHAN_MAINSESSION) {
8267         Ssh ssh = c->ssh;
8268
8269         if (!ssh->sent_console_eof &&
8270             (from_backend_eof(ssh->frontend) || ssh->got_pty)) {
8271             /*
8272              * Either from_backend_eof told us that the front end
8273              * wants us to close the outgoing side of the connection
8274              * as soon as we see EOF from the far end, or else we've
8275              * unilaterally decided to do that because we've allocated
8276              * a remote pty and hence EOF isn't a particularly
8277              * meaningful concept.
8278              */
8279             sshfwd_write_eof(c);
8280         }
8281         ssh->sent_console_eof = TRUE;
8282     }
8283
8284     ssh2_channel_check_close(c);
8285 }
8286
8287 static void ssh2_msg_channel_eof(Ssh ssh, struct Packet *pktin)
8288 {
8289     struct ssh_channel *c;
8290
8291     c = ssh_channel_msg(ssh, pktin);
8292     if (!c)
8293         return;
8294     if (c->type == CHAN_SHARING) {
8295         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8296                                   pktin->body, pktin->length);
8297         return;
8298     }
8299     ssh2_channel_got_eof(c);
8300 }
8301
8302 static void ssh2_msg_channel_close(Ssh ssh, struct Packet *pktin)
8303 {
8304     struct ssh_channel *c;
8305
8306     c = ssh_channel_msg(ssh, pktin);
8307     if (!c)
8308         return;
8309     if (c->type == CHAN_SHARING) {
8310         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8311                                   pktin->body, pktin->length);
8312         return;
8313     }
8314
8315     /*
8316      * When we receive CLOSE on a channel, we assume it comes with an
8317      * implied EOF if we haven't seen EOF yet.
8318      */
8319     ssh2_channel_got_eof(c);
8320
8321     if (!(ssh->remote_bugs & BUG_SENDS_LATE_REQUEST_REPLY)) {
8322         /*
8323          * It also means we stop expecting to see replies to any
8324          * outstanding channel requests, so clean those up too.
8325          * (ssh_chanreq_init will enforce by assertion that we don't
8326          * subsequently put anything back on this list.)
8327          */
8328         while (c->v.v2.chanreq_head) {
8329             struct outstanding_channel_request *ocr = c->v.v2.chanreq_head;
8330             ocr->handler(c, NULL, ocr->ctx);
8331             c->v.v2.chanreq_head = ocr->next;
8332             sfree(ocr);
8333         }
8334     }
8335
8336     /*
8337      * And we also send an outgoing EOF, if we haven't already, on the
8338      * assumption that CLOSE is a pretty forceful announcement that
8339      * the remote side is doing away with the entire channel. (If it
8340      * had wanted to send us EOF and continue receiving data from us,
8341      * it would have just sent CHANNEL_EOF.)
8342      */
8343     if (!(c->closes & CLOSES_SENT_EOF)) {
8344         /*
8345          * Make sure we don't read any more from whatever our local
8346          * data source is for this channel.
8347          */
8348         switch (c->type) {
8349           case CHAN_MAINSESSION:
8350             ssh->send_ok = 0;     /* stop trying to read from stdin */
8351             break;
8352           case CHAN_X11:
8353             x11_override_throttle(c->u.x11.xconn, 1);
8354             break;
8355           case CHAN_SOCKDATA:
8356             pfd_override_throttle(c->u.pfd.pf, 1);
8357             break;
8358         }
8359
8360         /*
8361          * Abandon any buffered data we still wanted to send to this
8362          * channel. Receiving a CHANNEL_CLOSE is an indication that
8363          * the server really wants to get on and _destroy_ this
8364          * channel, and it isn't going to send us any further
8365          * WINDOW_ADJUSTs to permit us to send pending stuff.
8366          */
8367         bufchain_clear(&c->v.v2.outbuffer);
8368
8369         /*
8370          * Send outgoing EOF.
8371          */
8372         sshfwd_write_eof(c);
8373     }
8374
8375     /*
8376      * Now process the actual close.
8377      */
8378     if (!(c->closes & CLOSES_RCVD_CLOSE)) {
8379         c->closes |= CLOSES_RCVD_CLOSE;
8380         ssh2_channel_check_close(c);
8381     }
8382 }
8383
8384 static void ssh2_msg_channel_open_confirmation(Ssh ssh, struct Packet *pktin)
8385 {
8386     struct ssh_channel *c;
8387
8388     c = ssh_channel_msg(ssh, pktin);
8389     if (!c)
8390         return;
8391     if (c->type == CHAN_SHARING) {
8392         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8393                                   pktin->body, pktin->length);
8394         return;
8395     }
8396     assert(c->halfopen); /* ssh_channel_msg will have enforced this */
8397     c->remoteid = ssh_pkt_getuint32(pktin);
8398     c->halfopen = FALSE;
8399     c->v.v2.remwindow = ssh_pkt_getuint32(pktin);
8400     c->v.v2.remmaxpkt = ssh_pkt_getuint32(pktin);
8401
8402     if (c->type == CHAN_SOCKDATA_DORMANT) {
8403         c->type = CHAN_SOCKDATA;
8404         if (c->u.pfd.pf)
8405             pfd_confirm(c->u.pfd.pf);
8406     } else if (c->type == CHAN_ZOMBIE) {
8407         /*
8408          * This case can occur if a local socket error occurred
8409          * between us sending out CHANNEL_OPEN and receiving
8410          * OPEN_CONFIRMATION. In this case, all we can do is
8411          * immediately initiate close proceedings now that we know the
8412          * server's id to put in the close message.
8413          */
8414         ssh2_channel_check_close(c);
8415     } else {
8416         /*
8417          * We never expect to receive OPEN_CONFIRMATION for any
8418          * *other* channel type (since only local-to-remote port
8419          * forwardings cause us to send CHANNEL_OPEN after the main
8420          * channel is live - all other auxiliary channel types are
8421          * initiated from the server end). It's safe to enforce this
8422          * by assertion rather than by ssh_disconnect, because the
8423          * real point is that we never constructed a half-open channel
8424          * structure in the first place with any type other than the
8425          * above.
8426          */
8427         assert(!"Funny channel type in ssh2_msg_channel_open_confirmation");
8428     }
8429
8430     if (c->pending_eof)
8431         ssh_channel_try_eof(c);        /* in case we had a pending EOF */
8432 }
8433
8434 static void ssh2_msg_channel_open_failure(Ssh ssh, struct Packet *pktin)
8435 {
8436     static const char *const reasons[] = {
8437         "<unknown reason code>",
8438             "Administratively prohibited",
8439             "Connect failed",
8440             "Unknown channel type",
8441             "Resource shortage",
8442     };
8443     unsigned reason_code;
8444     char *reason_string;
8445     int reason_length;
8446     struct ssh_channel *c;
8447
8448     c = ssh_channel_msg(ssh, pktin);
8449     if (!c)
8450         return;
8451     if (c->type == CHAN_SHARING) {
8452         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8453                                   pktin->body, pktin->length);
8454         return;
8455     }
8456     assert(c->halfopen); /* ssh_channel_msg will have enforced this */
8457
8458     if (c->type == CHAN_SOCKDATA_DORMANT) {
8459         reason_code = ssh_pkt_getuint32(pktin);
8460         if (reason_code >= lenof(reasons))
8461             reason_code = 0; /* ensure reasons[reason_code] in range */
8462         ssh_pkt_getstring(pktin, &reason_string, &reason_length);
8463         logeventf(ssh, "Forwarded connection refused by server: %s [%.*s]",
8464                   reasons[reason_code], reason_length,
8465                   NULLTOEMPTY(reason_string));
8466
8467         pfd_close(c->u.pfd.pf);
8468     } else if (c->type == CHAN_ZOMBIE) {
8469         /*
8470          * This case can occur if a local socket error occurred
8471          * between us sending out CHANNEL_OPEN and receiving
8472          * OPEN_FAILURE. In this case, we need do nothing except allow
8473          * the code below to throw the half-open channel away.
8474          */
8475     } else {
8476         /*
8477          * We never expect to receive OPEN_FAILURE for any *other*
8478          * channel type (since only local-to-remote port forwardings
8479          * cause us to send CHANNEL_OPEN after the main channel is
8480          * live - all other auxiliary channel types are initiated from
8481          * the server end). It's safe to enforce this by assertion
8482          * rather than by ssh_disconnect, because the real point is
8483          * that we never constructed a half-open channel structure in
8484          * the first place with any type other than the above.
8485          */
8486         assert(!"Funny channel type in ssh2_msg_channel_open_failure");
8487     }
8488
8489     del234(ssh->channels, c);
8490     sfree(c);
8491 }
8492
8493 static void ssh2_msg_channel_request(Ssh ssh, struct Packet *pktin)
8494 {
8495     char *type;
8496     int typelen, want_reply;
8497     int reply = SSH2_MSG_CHANNEL_FAILURE; /* default */
8498     struct ssh_channel *c;
8499     struct Packet *pktout;
8500
8501     c = ssh_channel_msg(ssh, pktin);
8502     if (!c)
8503         return;
8504     if (c->type == CHAN_SHARING) {
8505         share_got_pkt_from_server(c->u.sharing.ctx, pktin->type,
8506                                   pktin->body, pktin->length);
8507         return;
8508     }
8509     ssh_pkt_getstring(pktin, &type, &typelen);
8510     want_reply = ssh2_pkt_getbool(pktin);
8511
8512     if (c->closes & CLOSES_SENT_CLOSE) {
8513         /*
8514          * We don't reply to channel requests after we've sent
8515          * CHANNEL_CLOSE for the channel, because our reply might
8516          * cross in the network with the other side's CHANNEL_CLOSE
8517          * and arrive after they have wound the channel up completely.
8518          */
8519         want_reply = FALSE;
8520     }
8521
8522     /*
8523      * Having got the channel number, we now look at
8524      * the request type string to see if it's something
8525      * we recognise.
8526      */
8527     if (c == ssh->mainchan) {
8528         /*
8529          * We recognise "exit-status" and "exit-signal" on
8530          * the primary channel.
8531          */
8532         if (typelen == 11 &&
8533             !memcmp(type, "exit-status", 11)) {
8534
8535             ssh->exitcode = ssh_pkt_getuint32(pktin);
8536             logeventf(ssh, "Server sent command exit status %d",
8537                       ssh->exitcode);
8538             reply = SSH2_MSG_CHANNEL_SUCCESS;
8539
8540         } else if (typelen == 11 &&
8541                    !memcmp(type, "exit-signal", 11)) {
8542
8543             int is_plausible = TRUE, is_int = FALSE;
8544             char *fmt_sig = NULL, *fmt_msg = NULL;
8545             char *msg;
8546             int msglen = 0, core = FALSE;
8547             /* ICK: older versions of OpenSSH (e.g. 3.4p1)
8548              * provide an `int' for the signal, despite its
8549              * having been a `string' in the drafts of RFC 4254 since at
8550              * least 2001. (Fixed in session.c 1.147.) Try to
8551              * infer which we can safely parse it as. */
8552             {
8553                 unsigned char *p = pktin->body +
8554                     pktin->savedpos;
8555                 long len = pktin->length - pktin->savedpos;
8556                 unsigned long num = GET_32BIT(p); /* what is it? */
8557                 /* If it's 0, it hardly matters; assume string */
8558                 if (num == 0) {
8559                     is_int = FALSE;
8560                 } else {
8561                     int maybe_int = FALSE, maybe_str = FALSE;
8562 #define CHECK_HYPOTHESIS(offset, result)                                \
8563                     do                                                  \
8564                     {                                                   \
8565                         int q = toint(offset);                          \
8566                         if (q >= 0 && q+4 <= len) {                     \
8567                             q = toint(q + 4 + GET_32BIT(p+q));          \
8568                             if (q >= 0 && q+4 <= len &&                 \
8569                                 ((q = toint(q + 4 + GET_32BIT(p+q))) != 0) && \
8570                                 q == len)                               \
8571                                 result = TRUE;                          \
8572                         }                                               \
8573                     } while(0)
8574                     CHECK_HYPOTHESIS(4+1, maybe_int);
8575                     CHECK_HYPOTHESIS(4+num+1, maybe_str);
8576 #undef CHECK_HYPOTHESIS
8577                     if (maybe_int && !maybe_str)
8578                         is_int = TRUE;
8579                     else if (!maybe_int && maybe_str)
8580                         is_int = FALSE;
8581                     else
8582                         /* Crikey. Either or neither. Panic. */
8583                         is_plausible = FALSE;
8584                 }
8585             }
8586             ssh->exitcode = 128;       /* means `unknown signal' */
8587             if (is_plausible) {
8588                 if (is_int) {
8589                     /* Old non-standard OpenSSH. */
8590                     int signum = ssh_pkt_getuint32(pktin);
8591                     fmt_sig = dupprintf(" %d", signum);
8592                     ssh->exitcode = 128 + signum;
8593                 } else {
8594                     /* As per RFC 4254. */
8595                     char *sig;
8596                     int siglen;
8597                     ssh_pkt_getstring(pktin, &sig, &siglen);
8598                     /* Signal name isn't supposed to be blank, but
8599                      * let's cope gracefully if it is. */
8600                     if (siglen) {
8601                         fmt_sig = dupprintf(" \"%.*s\"",
8602                                             siglen, sig);
8603                     }
8604
8605                     /*
8606                      * Really hideous method of translating the
8607                      * signal description back into a locally
8608                      * meaningful number.
8609                      */
8610
8611                     if (0)
8612                         ;
8613 #define TRANSLATE_SIGNAL(s) \
8614     else if (siglen == lenof(#s)-1 && !memcmp(sig, #s, siglen)) \
8615         ssh->exitcode = 128 + SIG ## s
8616 #ifdef SIGABRT
8617                     TRANSLATE_SIGNAL(ABRT);
8618 #endif
8619 #ifdef SIGALRM
8620                     TRANSLATE_SIGNAL(ALRM);
8621 #endif
8622 #ifdef SIGFPE
8623                     TRANSLATE_SIGNAL(FPE);
8624 #endif
8625 #ifdef SIGHUP
8626                     TRANSLATE_SIGNAL(HUP);
8627 #endif
8628 #ifdef SIGILL
8629                     TRANSLATE_SIGNAL(ILL);
8630 #endif
8631 #ifdef SIGINT
8632                     TRANSLATE_SIGNAL(INT);
8633 #endif
8634 #ifdef SIGKILL
8635                     TRANSLATE_SIGNAL(KILL);
8636 #endif
8637 #ifdef SIGPIPE
8638                     TRANSLATE_SIGNAL(PIPE);
8639 #endif
8640 #ifdef SIGQUIT
8641                     TRANSLATE_SIGNAL(QUIT);
8642 #endif
8643 #ifdef SIGSEGV
8644                     TRANSLATE_SIGNAL(SEGV);
8645 #endif
8646 #ifdef SIGTERM
8647                     TRANSLATE_SIGNAL(TERM);
8648 #endif
8649 #ifdef SIGUSR1
8650                     TRANSLATE_SIGNAL(USR1);
8651 #endif
8652 #ifdef SIGUSR2
8653                     TRANSLATE_SIGNAL(USR2);
8654 #endif
8655 #undef TRANSLATE_SIGNAL
8656                     else
8657                         ssh->exitcode = 128;
8658                 }
8659                 core = ssh2_pkt_getbool(pktin);
8660                 ssh_pkt_getstring(pktin, &msg, &msglen);
8661                 if (msglen) {
8662                     fmt_msg = dupprintf(" (\"%.*s\")", msglen, msg);
8663                 }
8664                 /* ignore lang tag */
8665             } /* else don't attempt to parse */
8666             logeventf(ssh, "Server exited on signal%s%s%s",
8667                       fmt_sig ? fmt_sig : "",
8668                       core ? " (core dumped)" : "",
8669                       fmt_msg ? fmt_msg : "");
8670             sfree(fmt_sig);
8671             sfree(fmt_msg);
8672             reply = SSH2_MSG_CHANNEL_SUCCESS;
8673
8674         }
8675     } else {
8676         /*
8677          * This is a channel request we don't know
8678          * about, so we now either ignore the request
8679          * or respond with CHANNEL_FAILURE, depending
8680          * on want_reply.
8681          */
8682         reply = SSH2_MSG_CHANNEL_FAILURE;
8683     }
8684     if (want_reply) {
8685         pktout = ssh2_pkt_init(reply);
8686         ssh2_pkt_adduint32(pktout, c->remoteid);
8687         ssh2_pkt_send(ssh, pktout);
8688     }
8689 }
8690
8691 static void ssh2_msg_global_request(Ssh ssh, struct Packet *pktin)
8692 {
8693     char *type;
8694     int typelen, want_reply;
8695     struct Packet *pktout;
8696
8697     ssh_pkt_getstring(pktin, &type, &typelen);
8698     want_reply = ssh2_pkt_getbool(pktin);
8699
8700     /*
8701      * We currently don't support any global requests
8702      * at all, so we either ignore the request or
8703      * respond with REQUEST_FAILURE, depending on
8704      * want_reply.
8705      */
8706     if (want_reply) {
8707         pktout = ssh2_pkt_init(SSH2_MSG_REQUEST_FAILURE);
8708         ssh2_pkt_send(ssh, pktout);
8709     }
8710 }
8711
8712 struct X11FakeAuth *ssh_sharing_add_x11_display(Ssh ssh, int authtype,
8713                                                 void *share_cs,
8714                                                 void *share_chan)
8715 {
8716     struct X11FakeAuth *auth;
8717
8718     /*
8719      * Make up a new set of fake X11 auth data, and add it to the tree
8720      * of currently valid ones with an indication of the sharing
8721      * context that it's relevant to.
8722      */
8723     auth = x11_invent_fake_auth(ssh->x11authtree, authtype);
8724     auth->share_cs = share_cs;
8725     auth->share_chan = share_chan;
8726
8727     return auth;
8728 }
8729
8730 void ssh_sharing_remove_x11_display(Ssh ssh, struct X11FakeAuth *auth)
8731 {
8732     del234(ssh->x11authtree, auth);
8733     x11_free_fake_auth(auth);
8734 }
8735
8736 static void ssh2_msg_channel_open(Ssh ssh, struct Packet *pktin)
8737 {
8738     char *type;
8739     int typelen;
8740     char *peeraddr;
8741     int peeraddrlen;
8742     int peerport;
8743     const char *error = NULL;
8744     struct ssh_channel *c;
8745     unsigned remid, winsize, pktsize;
8746     unsigned our_winsize_override = 0;
8747     struct Packet *pktout;
8748
8749     ssh_pkt_getstring(pktin, &type, &typelen);
8750     c = snew(struct ssh_channel);
8751     c->ssh = ssh;
8752
8753     remid = ssh_pkt_getuint32(pktin);
8754     winsize = ssh_pkt_getuint32(pktin);
8755     pktsize = ssh_pkt_getuint32(pktin);
8756
8757     if (typelen == 3 && !memcmp(type, "x11", 3)) {
8758         char *addrstr;
8759
8760         ssh_pkt_getstring(pktin, &peeraddr, &peeraddrlen);
8761         addrstr = dupprintf("%.*s", peeraddrlen, NULLTOEMPTY(peeraddr));
8762         peerport = ssh_pkt_getuint32(pktin);
8763
8764         logeventf(ssh, "Received X11 connect request from %s:%d",
8765                   addrstr, peerport);
8766
8767         if (!ssh->X11_fwd_enabled && !ssh->connshare)
8768             error = "X11 forwarding is not enabled";
8769         else {
8770             c->u.x11.xconn = x11_init(ssh->x11authtree, c,
8771                                       addrstr, peerport);
8772             c->type = CHAN_X11;
8773             c->u.x11.initial = TRUE;
8774
8775             /*
8776              * If we are a connection-sharing upstream, then we should
8777              * initially present a very small window, adequate to take
8778              * the X11 initial authorisation packet but not much more.
8779              * Downstream will then present us a larger window (by
8780              * fiat of the connection-sharing protocol) and we can
8781              * guarantee to send a positive-valued WINDOW_ADJUST.
8782              */
8783             if (ssh->connshare)
8784                 our_winsize_override = 128;
8785
8786             logevent("Opened X11 forward channel");
8787         }
8788
8789         sfree(addrstr);
8790     } else if (typelen == 15 &&
8791                !memcmp(type, "forwarded-tcpip", 15)) {
8792         struct ssh_rportfwd pf, *realpf;
8793         char *shost;
8794         int shostlen;
8795         ssh_pkt_getstring(pktin, &shost, &shostlen);/* skip address */
8796         pf.shost = dupprintf("%.*s", shostlen, NULLTOEMPTY(shost));
8797         pf.sport = ssh_pkt_getuint32(pktin);
8798         ssh_pkt_getstring(pktin, &peeraddr, &peeraddrlen);
8799         peerport = ssh_pkt_getuint32(pktin);
8800         realpf = find234(ssh->rportfwds, &pf, NULL);
8801         logeventf(ssh, "Received remote port %s:%d open request "
8802                   "from %.*s:%d", pf.shost, pf.sport,
8803                   peeraddrlen, NULLTOEMPTY(peeraddr), peerport);
8804         sfree(pf.shost);
8805
8806         if (realpf == NULL) {
8807             error = "Remote port is not recognised";
8808         } else {
8809             char *err;
8810
8811             if (realpf->share_ctx) {
8812                 /*
8813                  * This port forwarding is on behalf of a
8814                  * connection-sharing downstream, so abandon our own
8815                  * channel-open procedure and just pass the message on
8816                  * to sshshare.c.
8817                  */
8818                 share_got_pkt_from_server(realpf->share_ctx, pktin->type,
8819                                           pktin->body, pktin->length);
8820                 sfree(c);
8821                 return;
8822             }
8823
8824             err = pfd_connect(&c->u.pfd.pf, realpf->dhost, realpf->dport,
8825                               c, ssh->conf, realpf->pfrec->addressfamily);
8826             logeventf(ssh, "Attempting to forward remote port to "
8827                       "%s:%d", realpf->dhost, realpf->dport);
8828             if (err != NULL) {
8829                 logeventf(ssh, "Port open failed: %s", err);
8830                 sfree(err);
8831                 error = "Port open failed";
8832             } else {
8833                 logevent("Forwarded port opened successfully");
8834                 c->type = CHAN_SOCKDATA;
8835             }
8836         }
8837     } else if (typelen == 22 &&
8838                !memcmp(type, "auth-agent@openssh.com", 22)) {
8839         if (!ssh->agentfwd_enabled)
8840             error = "Agent forwarding is not enabled";
8841         else {
8842             c->type = CHAN_AGENT;       /* identify channel type */
8843             c->u.a.lensofar = 0;
8844             c->u.a.message = NULL;
8845             c->u.a.outstanding_requests = 0;
8846         }
8847     } else {
8848         error = "Unsupported channel type requested";
8849     }
8850
8851     c->remoteid = remid;
8852     c->halfopen = FALSE;
8853     if (error) {
8854         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN_FAILURE);
8855         ssh2_pkt_adduint32(pktout, c->remoteid);
8856         ssh2_pkt_adduint32(pktout, SSH2_OPEN_CONNECT_FAILED);
8857         ssh2_pkt_addstring(pktout, error);
8858         ssh2_pkt_addstring(pktout, "en");       /* language tag */
8859         ssh2_pkt_send(ssh, pktout);
8860         logeventf(ssh, "Rejected channel open: %s", error);
8861         sfree(c);
8862     } else {
8863         ssh_channel_init(c);
8864         c->v.v2.remwindow = winsize;
8865         c->v.v2.remmaxpkt = pktsize;
8866         if (our_winsize_override) {
8867             c->v.v2.locwindow = c->v.v2.locmaxwin = c->v.v2.remlocwin =
8868                 our_winsize_override;
8869         }
8870         pktout = ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION);
8871         ssh2_pkt_adduint32(pktout, c->remoteid);
8872         ssh2_pkt_adduint32(pktout, c->localid);
8873         ssh2_pkt_adduint32(pktout, c->v.v2.locwindow);
8874         ssh2_pkt_adduint32(pktout, OUR_V2_MAXPKT);      /* our max pkt size */
8875         ssh2_pkt_send(ssh, pktout);
8876     }
8877 }
8878
8879 void sshfwd_x11_sharing_handover(struct ssh_channel *c,
8880                                  void *share_cs, void *share_chan,
8881                                  const char *peer_addr, int peer_port,
8882                                  int endian, int protomajor, int protominor,
8883                                  const void *initial_data, int initial_len)
8884 {
8885     /*
8886      * This function is called when we've just discovered that an X
8887      * forwarding channel on which we'd been handling the initial auth
8888      * ourselves turns out to be destined for a connection-sharing
8889      * downstream. So we turn the channel into a CHAN_SHARING, meaning
8890      * that we completely stop tracking windows and buffering data and
8891      * just pass more or less unmodified SSH messages back and forth.
8892      */
8893     c->type = CHAN_SHARING;
8894     c->u.sharing.ctx = share_cs;
8895     share_setup_x11_channel(share_cs, share_chan,
8896                             c->localid, c->remoteid, c->v.v2.remwindow,
8897                             c->v.v2.remmaxpkt, c->v.v2.locwindow,
8898                             peer_addr, peer_port, endian,
8899                             protomajor, protominor,
8900                             initial_data, initial_len);
8901 }
8902
8903 void sshfwd_x11_is_local(struct ssh_channel *c)
8904 {
8905     /*
8906      * This function is called when we've just discovered that an X
8907      * forwarding channel is _not_ destined for a connection-sharing
8908      * downstream but we're going to handle it ourselves. We stop
8909      * presenting a cautiously small window and go into ordinary data
8910      * exchange mode.
8911      */
8912     c->u.x11.initial = FALSE;
8913     ssh2_set_window(c, ssh_is_simple(c->ssh) ? OUR_V2_BIGWIN : OUR_V2_WINSIZE);
8914 }
8915
8916 /*
8917  * Buffer banner messages for later display at some convenient point,
8918  * if we're going to display them.
8919  */
8920 static void ssh2_msg_userauth_banner(Ssh ssh, struct Packet *pktin)
8921 {
8922     /* Arbitrary limit to prevent unbounded inflation of buffer */
8923     if (conf_get_int(ssh->conf, CONF_ssh_show_banner) &&
8924         bufchain_size(&ssh->banner) <= 131072) {
8925         char *banner = NULL;
8926         int size = 0;
8927         ssh_pkt_getstring(pktin, &banner, &size);
8928         if (banner)
8929             bufchain_add(&ssh->banner, banner, size);
8930     }
8931 }
8932
8933 /* Helper function to deal with sending tty modes for "pty-req" */
8934 static void ssh2_send_ttymode(void *data,
8935                               const struct ssh_ttymode *mode, char *val)
8936 {
8937     struct Packet *pktout = (struct Packet *)data;
8938     unsigned int arg = 0;
8939
8940     switch (mode->type) {
8941       case TTY_OP_CHAR:
8942         arg = ssh_tty_parse_specchar(val);
8943         break;
8944       case TTY_OP_BOOL:
8945         arg = ssh_tty_parse_boolean(val);
8946         break;
8947     }
8948     ssh2_pkt_addbyte(pktout, mode->opcode);
8949     ssh2_pkt_adduint32(pktout, arg);
8950 }
8951
8952 static void ssh2_setup_x11(struct ssh_channel *c, struct Packet *pktin,
8953                            void *ctx)
8954 {
8955     struct ssh2_setup_x11_state {
8956         int crLine;
8957     };
8958     Ssh ssh = c->ssh;
8959     struct Packet *pktout;
8960     crStateP(ssh2_setup_x11_state, ctx);
8961
8962     crBeginState;
8963
8964     logevent("Requesting X11 forwarding");
8965     pktout = ssh2_chanreq_init(ssh->mainchan, "x11-req",
8966                                ssh2_setup_x11, s);
8967     ssh2_pkt_addbool(pktout, 0);               /* many connections */
8968     ssh2_pkt_addstring(pktout, ssh->x11auth->protoname);
8969     ssh2_pkt_addstring(pktout, ssh->x11auth->datastring);
8970     ssh2_pkt_adduint32(pktout, ssh->x11disp->screennum);
8971     ssh2_pkt_send(ssh, pktout);
8972
8973     /* Wait to be called back with either a response packet, or NULL
8974      * meaning clean up and free our data */
8975     crReturnV;
8976
8977     if (pktin) {
8978         if (pktin->type == SSH2_MSG_CHANNEL_SUCCESS) {
8979             logevent("X11 forwarding enabled");
8980             ssh->X11_fwd_enabled = TRUE;
8981         } else
8982             logevent("X11 forwarding refused");
8983     }
8984
8985     crFinishFreeV;
8986 }
8987
8988 static void ssh2_setup_agent(struct ssh_channel *c, struct Packet *pktin,
8989                                    void *ctx)
8990 {
8991     struct ssh2_setup_agent_state {
8992         int crLine;
8993     };
8994     Ssh ssh = c->ssh;
8995     struct Packet *pktout;
8996     crStateP(ssh2_setup_agent_state, ctx);
8997
8998     crBeginState;
8999
9000     logevent("Requesting OpenSSH-style agent forwarding");
9001     pktout = ssh2_chanreq_init(ssh->mainchan, "auth-agent-req@openssh.com",
9002                                ssh2_setup_agent, s);
9003     ssh2_pkt_send(ssh, pktout);
9004
9005     /* Wait to be called back with either a response packet, or NULL
9006      * meaning clean up and free our data */
9007     crReturnV;
9008
9009     if (pktin) {
9010         if (pktin->type == SSH2_MSG_CHANNEL_SUCCESS) {
9011             logevent("Agent forwarding enabled");
9012             ssh->agentfwd_enabled = TRUE;
9013         } else
9014             logevent("Agent forwarding refused");
9015     }
9016
9017     crFinishFreeV;
9018 }
9019
9020 static void ssh2_setup_pty(struct ssh_channel *c, struct Packet *pktin,
9021                                  void *ctx)
9022 {
9023     struct ssh2_setup_pty_state {
9024         int crLine;
9025     };
9026     Ssh ssh = c->ssh;
9027     struct Packet *pktout;
9028     crStateP(ssh2_setup_pty_state, ctx);
9029
9030     crBeginState;
9031
9032     /* Unpick the terminal-speed string. */
9033     /* XXX perhaps we should allow no speeds to be sent. */
9034     ssh->ospeed = 38400; ssh->ispeed = 38400; /* last-resort defaults */
9035     sscanf(conf_get_str(ssh->conf, CONF_termspeed), "%d,%d", &ssh->ospeed, &ssh->ispeed);
9036     /* Build the pty request. */
9037     pktout = ssh2_chanreq_init(ssh->mainchan, "pty-req",
9038                                ssh2_setup_pty, s);
9039     ssh2_pkt_addstring(pktout, conf_get_str(ssh->conf, CONF_termtype));
9040     ssh2_pkt_adduint32(pktout, ssh->term_width);
9041     ssh2_pkt_adduint32(pktout, ssh->term_height);
9042     ssh2_pkt_adduint32(pktout, 0);             /* pixel width */
9043     ssh2_pkt_adduint32(pktout, 0);             /* pixel height */
9044     ssh2_pkt_addstring_start(pktout);
9045     parse_ttymodes(ssh, ssh2_send_ttymode, (void *)pktout);
9046     ssh2_pkt_addbyte(pktout, SSH2_TTY_OP_ISPEED);
9047     ssh2_pkt_adduint32(pktout, ssh->ispeed);
9048     ssh2_pkt_addbyte(pktout, SSH2_TTY_OP_OSPEED);
9049     ssh2_pkt_adduint32(pktout, ssh->ospeed);
9050     ssh2_pkt_addstring_data(pktout, "\0", 1); /* TTY_OP_END */
9051     ssh2_pkt_send(ssh, pktout);
9052     ssh->state = SSH_STATE_INTERMED;
9053
9054     /* Wait to be called back with either a response packet, or NULL
9055      * meaning clean up and free our data */
9056     crReturnV;
9057
9058     if (pktin) {
9059         if (pktin->type == SSH2_MSG_CHANNEL_SUCCESS) {
9060             logeventf(ssh, "Allocated pty (ospeed %dbps, ispeed %dbps)",
9061                       ssh->ospeed, ssh->ispeed);
9062             ssh->got_pty = TRUE;
9063         } else {
9064             c_write_str(ssh, "Server refused to allocate pty\r\n");
9065             ssh->editing = ssh->echoing = 1;
9066         }
9067     }
9068
9069     crFinishFreeV;
9070 }
9071
9072 static void ssh2_setup_env(struct ssh_channel *c, struct Packet *pktin,
9073                            void *ctx)
9074 {
9075     struct ssh2_setup_env_state {
9076         int crLine;
9077         int num_env, env_left, env_ok;
9078     };
9079     Ssh ssh = c->ssh;
9080     struct Packet *pktout;
9081     crStateP(ssh2_setup_env_state, ctx);
9082
9083     crBeginState;
9084
9085     /*
9086      * Send environment variables.
9087      * 
9088      * Simplest thing here is to send all the requests at once, and
9089      * then wait for a whole bunch of successes or failures.
9090      */
9091     s->num_env = 0;
9092     {
9093         char *key, *val;
9094
9095         for (val = conf_get_str_strs(ssh->conf, CONF_environmt, NULL, &key);
9096              val != NULL;
9097              val = conf_get_str_strs(ssh->conf, CONF_environmt, key, &key)) {
9098             pktout = ssh2_chanreq_init(ssh->mainchan, "env", ssh2_setup_env, s);
9099             ssh2_pkt_addstring(pktout, key);
9100             ssh2_pkt_addstring(pktout, val);
9101             ssh2_pkt_send(ssh, pktout);
9102
9103             s->num_env++;
9104         }
9105         if (s->num_env)
9106             logeventf(ssh, "Sent %d environment variables", s->num_env);
9107     }
9108
9109     if (s->num_env) {
9110         s->env_ok = 0;
9111         s->env_left = s->num_env;
9112
9113         while (s->env_left > 0) {
9114             /* Wait to be called back with either a response packet,
9115              * or NULL meaning clean up and free our data */
9116             crReturnV;
9117             if (!pktin) goto out;
9118             if (pktin->type == SSH2_MSG_CHANNEL_SUCCESS)
9119                 s->env_ok++;
9120             s->env_left--;
9121         }
9122
9123         if (s->env_ok == s->num_env) {
9124             logevent("All environment variables successfully set");
9125         } else if (s->env_ok == 0) {
9126             logevent("All environment variables refused");
9127             c_write_str(ssh, "Server refused to set environment variables\r\n");
9128         } else {
9129             logeventf(ssh, "%d environment variables refused",
9130                       s->num_env - s->env_ok);
9131             c_write_str(ssh, "Server refused to set all environment variables\r\n");
9132         }
9133     }
9134   out:;
9135     crFinishFreeV;
9136 }
9137
9138 /*
9139  * Handle the SSH-2 userauth and connection layers.
9140  */
9141 static void ssh2_msg_authconn(Ssh ssh, struct Packet *pktin)
9142 {
9143     do_ssh2_authconn(ssh, NULL, 0, pktin);
9144 }
9145
9146 static void ssh2_response_authconn(struct ssh_channel *c, struct Packet *pktin,
9147                                    void *ctx)
9148 {
9149     if (pktin)
9150         do_ssh2_authconn(c->ssh, NULL, 0, pktin);
9151 }
9152
9153 static void do_ssh2_authconn(Ssh ssh, const unsigned char *in, int inlen,
9154                              struct Packet *pktin)
9155 {
9156     struct do_ssh2_authconn_state {
9157         int crLine;
9158         enum {
9159             AUTH_TYPE_NONE,
9160                 AUTH_TYPE_PUBLICKEY,
9161                 AUTH_TYPE_PUBLICKEY_OFFER_LOUD,
9162                 AUTH_TYPE_PUBLICKEY_OFFER_QUIET,
9163                 AUTH_TYPE_PASSWORD,
9164                 AUTH_TYPE_GSSAPI,      /* always QUIET */
9165                 AUTH_TYPE_KEYBOARD_INTERACTIVE,
9166                 AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET
9167         } type;
9168         int done_service_req;
9169         int gotit, need_pw, can_pubkey, can_passwd, can_keyb_inter;
9170         int tried_pubkey_config, done_agent;
9171 #ifndef NO_GSSAPI
9172         int can_gssapi;
9173         int tried_gssapi;
9174 #endif
9175         int kbd_inter_refused;
9176         int we_are_in, userauth_success;
9177         prompts_t *cur_prompt;
9178         int num_prompts;
9179         char *username;
9180         char *password;
9181         int got_username;
9182         void *publickey_blob;
9183         int publickey_bloblen;
9184         int privatekey_available, privatekey_encrypted;
9185         char *publickey_algorithm;
9186         char *publickey_comment;
9187         unsigned char agent_request[5], *agent_response, *agentp;
9188         int agent_responselen;
9189         unsigned char *pkblob_in_agent;
9190         int keyi, nkeys;
9191         char *pkblob, *alg, *commentp;
9192         int pklen, alglen, commentlen;
9193         int siglen, retlen, len;
9194         char *q, *agentreq, *ret;
9195         struct Packet *pktout;
9196         Filename *keyfile;
9197 #ifndef NO_GSSAPI
9198         struct ssh_gss_library *gsslib;
9199         Ssh_gss_ctx gss_ctx;
9200         Ssh_gss_buf gss_buf;
9201         Ssh_gss_buf gss_rcvtok, gss_sndtok;
9202         Ssh_gss_name gss_srv_name;
9203         Ssh_gss_stat gss_stat;
9204 #endif
9205     };
9206     crState(do_ssh2_authconn_state);
9207
9208     crBeginState;
9209
9210     /* Register as a handler for all the messages this coroutine handles. */
9211     ssh->packet_dispatch[SSH2_MSG_SERVICE_ACCEPT] = ssh2_msg_authconn;
9212     ssh->packet_dispatch[SSH2_MSG_USERAUTH_REQUEST] = ssh2_msg_authconn;
9213     ssh->packet_dispatch[SSH2_MSG_USERAUTH_FAILURE] = ssh2_msg_authconn;
9214     ssh->packet_dispatch[SSH2_MSG_USERAUTH_SUCCESS] = ssh2_msg_authconn;
9215     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] = ssh2_msg_authconn;
9216     ssh->packet_dispatch[SSH2_MSG_USERAUTH_PK_OK] = ssh2_msg_authconn;
9217     /* ssh->packet_dispatch[SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ] = ssh2_msg_authconn; duplicate case value */
9218     /* ssh->packet_dispatch[SSH2_MSG_USERAUTH_INFO_REQUEST] = ssh2_msg_authconn; duplicate case value */
9219     ssh->packet_dispatch[SSH2_MSG_USERAUTH_INFO_RESPONSE] = ssh2_msg_authconn;
9220     ssh->packet_dispatch[SSH2_MSG_GLOBAL_REQUEST] = ssh2_msg_authconn;
9221     ssh->packet_dispatch[SSH2_MSG_REQUEST_SUCCESS] = ssh2_msg_authconn;
9222     ssh->packet_dispatch[SSH2_MSG_REQUEST_FAILURE] = ssh2_msg_authconn;
9223     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN] = ssh2_msg_authconn;
9224     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_CONFIRMATION] = ssh2_msg_authconn;
9225     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_FAILURE] = ssh2_msg_authconn;
9226     ssh->packet_dispatch[SSH2_MSG_CHANNEL_WINDOW_ADJUST] = ssh2_msg_authconn;
9227     ssh->packet_dispatch[SSH2_MSG_CHANNEL_DATA] = ssh2_msg_authconn;
9228     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EXTENDED_DATA] = ssh2_msg_authconn;
9229     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EOF] = ssh2_msg_authconn;
9230     ssh->packet_dispatch[SSH2_MSG_CHANNEL_CLOSE] = ssh2_msg_authconn;
9231     
9232     s->done_service_req = FALSE;
9233     s->we_are_in = s->userauth_success = FALSE;
9234     s->agent_response = NULL;
9235 #ifndef NO_GSSAPI
9236     s->tried_gssapi = FALSE;
9237 #endif
9238
9239     if (!ssh->bare_connection) {
9240         if (!conf_get_int(ssh->conf, CONF_ssh_no_userauth)) {
9241             /*
9242              * Request userauth protocol, and await a response to it.
9243              */
9244             s->pktout = ssh2_pkt_init(SSH2_MSG_SERVICE_REQUEST);
9245             ssh2_pkt_addstring(s->pktout, "ssh-userauth");
9246             ssh2_pkt_send(ssh, s->pktout);
9247             crWaitUntilV(pktin);
9248             if (pktin->type == SSH2_MSG_SERVICE_ACCEPT)
9249                 s->done_service_req = TRUE;
9250         }
9251         if (!s->done_service_req) {
9252             /*
9253              * Request connection protocol directly, without authentication.
9254              */
9255             s->pktout = ssh2_pkt_init(SSH2_MSG_SERVICE_REQUEST);
9256             ssh2_pkt_addstring(s->pktout, "ssh-connection");
9257             ssh2_pkt_send(ssh, s->pktout);
9258             crWaitUntilV(pktin);
9259             if (pktin->type == SSH2_MSG_SERVICE_ACCEPT) {
9260                 s->we_are_in = TRUE; /* no auth required */
9261             } else {
9262                 bombout(("Server refused service request"));
9263                 crStopV;
9264             }
9265         }
9266     } else {
9267         s->we_are_in = TRUE;
9268     }
9269
9270     /* Arrange to be able to deal with any BANNERs that come in.
9271      * (We do this now as packets may come in during the next bit.) */
9272     bufchain_init(&ssh->banner);
9273     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] =
9274         ssh2_msg_userauth_banner;
9275
9276     /*
9277      * Misc one-time setup for authentication.
9278      */
9279     s->publickey_blob = NULL;
9280     if (!s->we_are_in) {
9281
9282         /*
9283          * Load the public half of any configured public key file
9284          * for later use.
9285          */
9286         s->keyfile = conf_get_filename(ssh->conf, CONF_keyfile);
9287         if (!filename_is_null(s->keyfile)) {
9288             int keytype;
9289             logeventf(ssh, "Reading key file \"%.150s\"",
9290                       filename_to_str(s->keyfile));
9291             keytype = key_type(s->keyfile);
9292             if (keytype == SSH_KEYTYPE_SSH2 ||
9293                 keytype == SSH_KEYTYPE_SSH2_PUBLIC_RFC4716 ||
9294                 keytype == SSH_KEYTYPE_SSH2_PUBLIC_OPENSSH) {
9295                 const char *error;
9296                 s->publickey_blob =
9297                     ssh2_userkey_loadpub(s->keyfile,
9298                                          &s->publickey_algorithm,
9299                                          &s->publickey_bloblen, 
9300                                          &s->publickey_comment, &error);
9301                 if (s->publickey_blob) {
9302                     s->privatekey_available = (keytype == SSH_KEYTYPE_SSH2);
9303                     if (!s->privatekey_available)
9304                         logeventf(ssh, "Key file contains public key only");
9305                     s->privatekey_encrypted =
9306                         ssh2_userkey_encrypted(s->keyfile, NULL);
9307                 } else {
9308                     char *msgbuf;
9309                     logeventf(ssh, "Unable to load key (%s)", 
9310                               error);
9311                     msgbuf = dupprintf("Unable to load key file "
9312                                        "\"%.150s\" (%s)\r\n",
9313                                        filename_to_str(s->keyfile),
9314                                        error);
9315                     c_write_str(ssh, msgbuf);
9316                     sfree(msgbuf);
9317                 }
9318             } else {
9319                 char *msgbuf;
9320                 logeventf(ssh, "Unable to use this key file (%s)",
9321                           key_type_to_str(keytype));
9322                 msgbuf = dupprintf("Unable to use key file \"%.150s\""
9323                                    " (%s)\r\n",
9324                                    filename_to_str(s->keyfile),
9325                                    key_type_to_str(keytype));
9326                 c_write_str(ssh, msgbuf);
9327                 sfree(msgbuf);
9328                 s->publickey_blob = NULL;
9329             }
9330         }
9331
9332         /*
9333          * Find out about any keys Pageant has (but if there's a
9334          * public key configured, filter out all others).
9335          */
9336         s->nkeys = 0;
9337         s->agent_response = NULL;
9338         s->pkblob_in_agent = NULL;
9339         if (conf_get_int(ssh->conf, CONF_tryagent) && agent_exists()) {
9340
9341             void *r;
9342
9343             logevent("Pageant is running. Requesting keys.");
9344
9345             /* Request the keys held by the agent. */
9346             PUT_32BIT(s->agent_request, 1);
9347             s->agent_request[4] = SSH2_AGENTC_REQUEST_IDENTITIES;
9348             if (!agent_query(s->agent_request, 5, &r, &s->agent_responselen,
9349                              ssh_agent_callback, ssh)) {
9350                 do {
9351                     crReturnV;
9352                     if (pktin) {
9353                         bombout(("Unexpected data from server while"
9354                                  " waiting for agent response"));
9355                         crStopV;
9356                     }
9357                 } while (pktin || inlen > 0);
9358                 r = ssh->agent_response;
9359                 s->agent_responselen = ssh->agent_response_len;
9360             }
9361             s->agent_response = (unsigned char *) r;
9362             if (s->agent_response && s->agent_responselen >= 5 &&
9363                 s->agent_response[4] == SSH2_AGENT_IDENTITIES_ANSWER) {
9364                 int keyi;
9365                 unsigned char *p;
9366                 p = s->agent_response + 5;
9367                 s->nkeys = toint(GET_32BIT(p));
9368
9369                 /*
9370                  * Vet the Pageant response to ensure that the key
9371                  * count and blob lengths make sense.
9372                  */
9373                 if (s->nkeys < 0) {
9374                     logeventf(ssh, "Pageant response contained a negative"
9375                               " key count %d", s->nkeys);
9376                     s->nkeys = 0;
9377                     goto done_agent_query;
9378                 } else {
9379                     unsigned char *q = p + 4;
9380                     int lenleft = s->agent_responselen - 5 - 4;
9381
9382                     for (keyi = 0; keyi < s->nkeys; keyi++) {
9383                         int bloblen, commentlen;
9384                         if (lenleft < 4) {
9385                             logeventf(ssh, "Pageant response was truncated");
9386                             s->nkeys = 0;
9387                             goto done_agent_query;
9388                         }
9389                         bloblen = toint(GET_32BIT(q));
9390                         if (bloblen < 0 || bloblen > lenleft) {
9391                             logeventf(ssh, "Pageant response was truncated");
9392                             s->nkeys = 0;
9393                             goto done_agent_query;
9394                         }
9395                         lenleft -= 4 + bloblen;
9396                         q += 4 + bloblen;
9397                         commentlen = toint(GET_32BIT(q));
9398                         if (commentlen < 0 || commentlen > lenleft) {
9399                             logeventf(ssh, "Pageant response was truncated");
9400                             s->nkeys = 0;
9401                             goto done_agent_query;
9402                         }
9403                         lenleft -= 4 + commentlen;
9404                         q += 4 + commentlen;
9405                     }
9406                 }
9407
9408                 p += 4;
9409                 logeventf(ssh, "Pageant has %d SSH-2 keys", s->nkeys);
9410                 if (s->publickey_blob) {
9411                     /* See if configured key is in agent. */
9412                     for (keyi = 0; keyi < s->nkeys; keyi++) {
9413                         s->pklen = toint(GET_32BIT(p));
9414                         if (s->pklen == s->publickey_bloblen &&
9415                             !memcmp(p+4, s->publickey_blob,
9416                                     s->publickey_bloblen)) {
9417                             logeventf(ssh, "Pageant key #%d matches "
9418                                       "configured key file", keyi);
9419                             s->keyi = keyi;
9420                             s->pkblob_in_agent = p;
9421                             break;
9422                         }
9423                         p += 4 + s->pklen;
9424                         p += toint(GET_32BIT(p)) + 4; /* comment */
9425                     }
9426                     if (!s->pkblob_in_agent) {
9427                         logevent("Configured key file not in Pageant");
9428                         s->nkeys = 0;
9429                     }
9430                 }
9431             } else {
9432                 logevent("Failed to get reply from Pageant");
9433             }
9434           done_agent_query:;
9435         }
9436
9437     }
9438
9439     /*
9440      * We repeat this whole loop, including the username prompt,
9441      * until we manage a successful authentication. If the user
9442      * types the wrong _password_, they can be sent back to the
9443      * beginning to try another username, if this is configured on.
9444      * (If they specify a username in the config, they are never
9445      * asked, even if they do give a wrong password.)
9446      * 
9447      * I think this best serves the needs of
9448      * 
9449      *  - the people who have no configuration, no keys, and just
9450      *    want to try repeated (username,password) pairs until they
9451      *    type both correctly
9452      * 
9453      *  - people who have keys and configuration but occasionally
9454      *    need to fall back to passwords
9455      * 
9456      *  - people with a key held in Pageant, who might not have
9457      *    logged in to a particular machine before; so they want to
9458      *    type a username, and then _either_ their key will be
9459      *    accepted, _or_ they will type a password. If they mistype
9460      *    the username they will want to be able to get back and
9461      *    retype it!
9462      */
9463     s->got_username = FALSE;
9464     while (!s->we_are_in) {
9465         /*
9466          * Get a username.
9467          */
9468         if (s->got_username && !conf_get_int(ssh->conf, CONF_change_username)) {
9469             /*
9470              * We got a username last time round this loop, and
9471              * with change_username turned off we don't try to get
9472              * it again.
9473              */
9474         } else if ((ssh->username = get_remote_username(ssh->conf)) == NULL) {
9475             int ret; /* need not be kept over crReturn */
9476             s->cur_prompt = new_prompts(ssh->frontend);
9477             s->cur_prompt->to_server = TRUE;
9478             s->cur_prompt->name = dupstr("SSH login name");
9479             add_prompt(s->cur_prompt, dupstr("login as: "), TRUE); 
9480             ret = get_userpass_input(s->cur_prompt, NULL, 0);
9481             while (ret < 0) {
9482                 ssh->send_ok = 1;
9483                 crWaitUntilV(!pktin);
9484                 ret = get_userpass_input(s->cur_prompt, in, inlen);
9485                 ssh->send_ok = 0;
9486             }
9487             if (!ret) {
9488                 /*
9489                  * get_userpass_input() failed to get a username.
9490                  * Terminate.
9491                  */
9492                 free_prompts(s->cur_prompt);
9493                 ssh_disconnect(ssh, "No username provided", NULL, 0, TRUE);
9494                 crStopV;
9495             }
9496             ssh->username = dupstr(s->cur_prompt->prompts[0]->result);
9497             free_prompts(s->cur_prompt);
9498         } else {
9499             char *stuff;
9500             if ((flags & FLAG_VERBOSE) || (flags & FLAG_INTERACTIVE)) {
9501                 stuff = dupprintf("Using username \"%s\".\r\n", ssh->username);
9502                 c_write_str(ssh, stuff);
9503                 sfree(stuff);
9504             }
9505         }
9506         s->got_username = TRUE;
9507
9508         /*
9509          * Send an authentication request using method "none": (a)
9510          * just in case it succeeds, and (b) so that we know what
9511          * authentication methods we can usefully try next.
9512          */
9513         ssh->pkt_actx = SSH2_PKTCTX_NOAUTH;
9514
9515         s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
9516         ssh2_pkt_addstring(s->pktout, ssh->username);
9517         ssh2_pkt_addstring(s->pktout, "ssh-connection");/* service requested */
9518         ssh2_pkt_addstring(s->pktout, "none");    /* method */
9519         ssh2_pkt_send(ssh, s->pktout);
9520         s->type = AUTH_TYPE_NONE;
9521         s->gotit = FALSE;
9522         s->we_are_in = FALSE;
9523
9524         s->tried_pubkey_config = FALSE;
9525         s->kbd_inter_refused = FALSE;
9526
9527         /* Reset agent request state. */
9528         s->done_agent = FALSE;
9529         if (s->agent_response) {
9530             if (s->pkblob_in_agent) {
9531                 s->agentp = s->pkblob_in_agent;
9532             } else {
9533                 s->agentp = s->agent_response + 5 + 4;
9534                 s->keyi = 0;
9535             }
9536         }
9537
9538         while (1) {
9539             char *methods = NULL;
9540             int methlen = 0;
9541
9542             /*
9543              * Wait for the result of the last authentication request.
9544              */
9545             if (!s->gotit)
9546                 crWaitUntilV(pktin);
9547             /*
9548              * Now is a convenient point to spew any banner material
9549              * that we've accumulated. (This should ensure that when
9550              * we exit the auth loop, we haven't any left to deal
9551              * with.)
9552              */
9553             {
9554                 int size = bufchain_size(&ssh->banner);
9555                 /*
9556                  * Don't show the banner if we're operating in
9557                  * non-verbose non-interactive mode. (It's probably
9558                  * a script, which means nobody will read the
9559                  * banner _anyway_, and moreover the printing of
9560                  * the banner will screw up processing on the
9561                  * output of (say) plink.)
9562                  */
9563                 if (size && (flags & (FLAG_VERBOSE | FLAG_INTERACTIVE))) {
9564                     char *banner = snewn(size, char);
9565                     bufchain_fetch(&ssh->banner, banner, size);
9566                     c_write_untrusted(ssh, banner, size);
9567                     sfree(banner);
9568                 }
9569                 bufchain_clear(&ssh->banner);
9570             }
9571             if (pktin->type == SSH2_MSG_USERAUTH_SUCCESS) {
9572                 logevent("Access granted");
9573                 s->we_are_in = s->userauth_success = TRUE;
9574                 break;
9575             }
9576
9577             if (pktin->type != SSH2_MSG_USERAUTH_FAILURE && s->type != AUTH_TYPE_GSSAPI) {
9578                 bombout(("Strange packet received during authentication: "
9579                          "type %d", pktin->type));
9580                 crStopV;
9581             }
9582
9583             s->gotit = FALSE;
9584
9585             /*
9586              * OK, we're now sitting on a USERAUTH_FAILURE message, so
9587              * we can look at the string in it and know what we can
9588              * helpfully try next.
9589              */
9590             if (pktin->type == SSH2_MSG_USERAUTH_FAILURE) {
9591                 ssh_pkt_getstring(pktin, &methods, &methlen);
9592                 if (!ssh2_pkt_getbool(pktin)) {
9593                     /*
9594                      * We have received an unequivocal Access
9595                      * Denied. This can translate to a variety of
9596                      * messages, or no message at all.
9597                      *
9598                      * For forms of authentication which are attempted
9599                      * implicitly, by which I mean without printing
9600                      * anything in the window indicating that we're
9601                      * trying them, we should never print 'Access
9602                      * denied'.
9603                      *
9604                      * If we do print a message saying that we're
9605                      * attempting some kind of authentication, it's OK
9606                      * to print a followup message saying it failed -
9607                      * but the message may sometimes be more specific
9608                      * than simply 'Access denied'.
9609                      *
9610                      * Additionally, if we'd just tried password
9611                      * authentication, we should break out of this
9612                      * whole loop so as to go back to the username
9613                      * prompt (iff we're configured to allow
9614                      * username change attempts).
9615                      */
9616                     if (s->type == AUTH_TYPE_NONE) {
9617                         /* do nothing */
9618                     } else if (s->type == AUTH_TYPE_PUBLICKEY_OFFER_LOUD ||
9619                                s->type == AUTH_TYPE_PUBLICKEY_OFFER_QUIET) {
9620                         if (s->type == AUTH_TYPE_PUBLICKEY_OFFER_LOUD)
9621                             c_write_str(ssh, "Server refused our key\r\n");
9622                         logevent("Server refused our key");
9623                     } else if (s->type == AUTH_TYPE_PUBLICKEY) {
9624                         /* This _shouldn't_ happen except by a
9625                          * protocol bug causing client and server to
9626                          * disagree on what is a correct signature. */
9627                         c_write_str(ssh, "Server refused public-key signature"
9628                                     " despite accepting key!\r\n");
9629                         logevent("Server refused public-key signature"
9630                                  " despite accepting key!");
9631                     } else if (s->type==AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET) {
9632                         /* quiet, so no c_write */
9633                         logevent("Server refused keyboard-interactive authentication");
9634                     } else if (s->type==AUTH_TYPE_GSSAPI) {
9635                         /* always quiet, so no c_write */
9636                         /* also, the code down in the GSSAPI block has
9637                          * already logged this in the Event Log */
9638                     } else if (s->type == AUTH_TYPE_KEYBOARD_INTERACTIVE) {
9639                         logevent("Keyboard-interactive authentication failed");
9640                         c_write_str(ssh, "Access denied\r\n");
9641                     } else {
9642                         assert(s->type == AUTH_TYPE_PASSWORD);
9643                         logevent("Password authentication failed");
9644                         c_write_str(ssh, "Access denied\r\n");
9645
9646                         if (conf_get_int(ssh->conf, CONF_change_username)) {
9647                             /* XXX perhaps we should allow
9648                              * keyboard-interactive to do this too? */
9649                             s->we_are_in = FALSE;
9650                             break;
9651                         }
9652                     }
9653                 } else {
9654                     c_write_str(ssh, "Further authentication required\r\n");
9655                     logevent("Further authentication required");
9656                 }
9657
9658                 s->can_pubkey =
9659                     in_commasep_string("publickey", methods, methlen);
9660                 s->can_passwd =
9661                     in_commasep_string("password", methods, methlen);
9662                 s->can_keyb_inter = conf_get_int(ssh->conf, CONF_try_ki_auth) &&
9663                     in_commasep_string("keyboard-interactive", methods, methlen);
9664 #ifndef NO_GSSAPI
9665                 if (conf_get_int(ssh->conf, CONF_try_gssapi_auth) &&
9666                     in_commasep_string("gssapi-with-mic", methods, methlen)) {
9667                     /* Try loading the GSS libraries and see if we
9668                      * have any. */
9669                     if (!ssh->gsslibs)
9670                         ssh->gsslibs = ssh_gss_setup(ssh->conf);
9671                     s->can_gssapi = (ssh->gsslibs->nlibraries > 0);
9672                 } else {
9673                     /* No point in even bothering to try to load the
9674                      * GSS libraries, if the user configuration and
9675                      * server aren't both prepared to attempt GSSAPI
9676                      * auth in the first place. */
9677                     s->can_gssapi = FALSE;
9678                 }
9679 #endif
9680             }
9681
9682             ssh->pkt_actx = SSH2_PKTCTX_NOAUTH;
9683
9684             if (s->can_pubkey && !s->done_agent && s->nkeys) {
9685
9686                 /*
9687                  * Attempt public-key authentication using a key from Pageant.
9688                  */
9689
9690                 ssh->pkt_actx = SSH2_PKTCTX_PUBLICKEY;
9691
9692                 logeventf(ssh, "Trying Pageant key #%d", s->keyi);
9693
9694                 /* Unpack key from agent response */
9695                 s->pklen = toint(GET_32BIT(s->agentp));
9696                 s->agentp += 4;
9697                 s->pkblob = (char *)s->agentp;
9698                 s->agentp += s->pklen;
9699                 s->alglen = toint(GET_32BIT(s->pkblob));
9700                 s->alg = s->pkblob + 4;
9701                 s->commentlen = toint(GET_32BIT(s->agentp));
9702                 s->agentp += 4;
9703                 s->commentp = (char *)s->agentp;
9704                 s->agentp += s->commentlen;
9705                 /* s->agentp now points at next key, if any */
9706
9707                 /* See if server will accept it */
9708                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
9709                 ssh2_pkt_addstring(s->pktout, ssh->username);
9710                 ssh2_pkt_addstring(s->pktout, "ssh-connection");
9711                                                     /* service requested */
9712                 ssh2_pkt_addstring(s->pktout, "publickey");
9713                                                     /* method */
9714                 ssh2_pkt_addbool(s->pktout, FALSE); /* no signature included */
9715                 ssh2_pkt_addstring_start(s->pktout);
9716                 ssh2_pkt_addstring_data(s->pktout, s->alg, s->alglen);
9717                 ssh2_pkt_addstring_start(s->pktout);
9718                 ssh2_pkt_addstring_data(s->pktout, s->pkblob, s->pklen);
9719                 ssh2_pkt_send(ssh, s->pktout);
9720                 s->type = AUTH_TYPE_PUBLICKEY_OFFER_QUIET;
9721
9722                 crWaitUntilV(pktin);
9723                 if (pktin->type != SSH2_MSG_USERAUTH_PK_OK) {
9724
9725                     /* Offer of key refused. */
9726                     s->gotit = TRUE;
9727
9728                 } else {
9729                     
9730                     void *vret;
9731
9732                     if (flags & FLAG_VERBOSE) {
9733                         c_write_str(ssh, "Authenticating with "
9734                                     "public key \"");
9735                         c_write(ssh, s->commentp, s->commentlen);
9736                         c_write_str(ssh, "\" from agent\r\n");
9737                     }
9738
9739                     /*
9740                      * Server is willing to accept the key.
9741                      * Construct a SIGN_REQUEST.
9742                      */
9743                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
9744                     ssh2_pkt_addstring(s->pktout, ssh->username);
9745                     ssh2_pkt_addstring(s->pktout, "ssh-connection");
9746                                                         /* service requested */
9747                     ssh2_pkt_addstring(s->pktout, "publickey");
9748                                                         /* method */
9749                     ssh2_pkt_addbool(s->pktout, TRUE);  /* signature included */
9750                     ssh2_pkt_addstring_start(s->pktout);
9751                     ssh2_pkt_addstring_data(s->pktout, s->alg, s->alglen);
9752                     ssh2_pkt_addstring_start(s->pktout);
9753                     ssh2_pkt_addstring_data(s->pktout, s->pkblob, s->pklen);
9754
9755                     /* Ask agent for signature. */
9756                     s->siglen = s->pktout->length - 5 + 4 +
9757                         ssh->v2_session_id_len;
9758                     if (ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)
9759                         s->siglen -= 4;
9760                     s->len = 1;       /* message type */
9761                     s->len += 4 + s->pklen;     /* key blob */
9762                     s->len += 4 + s->siglen;    /* data to sign */
9763                     s->len += 4;      /* flags */
9764                     s->agentreq = snewn(4 + s->len, char);
9765                     PUT_32BIT(s->agentreq, s->len);
9766                     s->q = s->agentreq + 4;
9767                     *s->q++ = SSH2_AGENTC_SIGN_REQUEST;
9768                     PUT_32BIT(s->q, s->pklen);
9769                     s->q += 4;
9770                     memcpy(s->q, s->pkblob, s->pklen);
9771                     s->q += s->pklen;
9772                     PUT_32BIT(s->q, s->siglen);
9773                     s->q += 4;
9774                     /* Now the data to be signed... */
9775                     if (!(ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)) {
9776                         PUT_32BIT(s->q, ssh->v2_session_id_len);
9777                         s->q += 4;
9778                     }
9779                     memcpy(s->q, ssh->v2_session_id,
9780                            ssh->v2_session_id_len);
9781                     s->q += ssh->v2_session_id_len;
9782                     memcpy(s->q, s->pktout->data + 5,
9783                            s->pktout->length - 5);
9784                     s->q += s->pktout->length - 5;
9785                     /* And finally the (zero) flags word. */
9786                     PUT_32BIT(s->q, 0);
9787                     if (!agent_query(s->agentreq, s->len + 4,
9788                                      &vret, &s->retlen,
9789                                      ssh_agent_callback, ssh)) {
9790                         do {
9791                             crReturnV;
9792                             if (pktin) {
9793                                 bombout(("Unexpected data from server"
9794                                          " while waiting for agent"
9795                                          " response"));
9796                                 crStopV;
9797                             }
9798                         } while (pktin || inlen > 0);
9799                         vret = ssh->agent_response;
9800                         s->retlen = ssh->agent_response_len;
9801                     }
9802                     s->ret = vret;
9803                     sfree(s->agentreq);
9804                     if (s->ret) {
9805                         if (s->retlen >= 9 &&
9806                             s->ret[4] == SSH2_AGENT_SIGN_RESPONSE &&
9807                             GET_32BIT(s->ret + 5) <= (unsigned)(s->retlen-9)) {
9808                             logevent("Sending Pageant's response");
9809                             ssh2_add_sigblob(ssh, s->pktout,
9810                                              s->pkblob, s->pklen,
9811                                              s->ret + 9,
9812                                              GET_32BIT(s->ret + 5));
9813                             ssh2_pkt_send(ssh, s->pktout);
9814                             s->type = AUTH_TYPE_PUBLICKEY;
9815                         } else {
9816                             /* FIXME: less drastic response */
9817                             bombout(("Pageant failed to answer challenge"));
9818                             crStopV;
9819                         }
9820                     }
9821                 }
9822
9823                 /* Do we have any keys left to try? */
9824                 if (s->pkblob_in_agent) {
9825                     s->done_agent = TRUE;
9826                     s->tried_pubkey_config = TRUE;
9827                 } else {
9828                     s->keyi++;
9829                     if (s->keyi >= s->nkeys)
9830                         s->done_agent = TRUE;
9831                 }
9832
9833             } else if (s->can_pubkey && s->publickey_blob &&
9834                        s->privatekey_available && !s->tried_pubkey_config) {
9835
9836                 struct ssh2_userkey *key;   /* not live over crReturn */
9837                 char *passphrase;           /* not live over crReturn */
9838
9839                 ssh->pkt_actx = SSH2_PKTCTX_PUBLICKEY;
9840
9841                 s->tried_pubkey_config = TRUE;
9842
9843                 /*
9844                  * Try the public key supplied in the configuration.
9845                  *
9846                  * First, offer the public blob to see if the server is
9847                  * willing to accept it.
9848                  */
9849                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
9850                 ssh2_pkt_addstring(s->pktout, ssh->username);
9851                 ssh2_pkt_addstring(s->pktout, "ssh-connection");
9852                                                 /* service requested */
9853                 ssh2_pkt_addstring(s->pktout, "publickey");     /* method */
9854                 ssh2_pkt_addbool(s->pktout, FALSE);
9855                                                 /* no signature included */
9856                 ssh2_pkt_addstring(s->pktout, s->publickey_algorithm);
9857                 ssh2_pkt_addstring_start(s->pktout);
9858                 ssh2_pkt_addstring_data(s->pktout,
9859                                         (char *)s->publickey_blob,
9860                                         s->publickey_bloblen);
9861                 ssh2_pkt_send(ssh, s->pktout);
9862                 logevent("Offered public key");
9863
9864                 crWaitUntilV(pktin);
9865                 if (pktin->type != SSH2_MSG_USERAUTH_PK_OK) {
9866                     /* Key refused. Give up. */
9867                     s->gotit = TRUE; /* reconsider message next loop */
9868                     s->type = AUTH_TYPE_PUBLICKEY_OFFER_LOUD;
9869                     continue; /* process this new message */
9870                 }
9871                 logevent("Offer of public key accepted");
9872
9873                 /*
9874                  * Actually attempt a serious authentication using
9875                  * the key.
9876                  */
9877                 if (flags & FLAG_VERBOSE) {
9878                     c_write_str(ssh, "Authenticating with public key \"");
9879                     c_write_str(ssh, s->publickey_comment);
9880                     c_write_str(ssh, "\"\r\n");
9881                 }
9882                 key = NULL;
9883                 while (!key) {
9884                     const char *error;  /* not live over crReturn */
9885                     if (s->privatekey_encrypted) {
9886                         /*
9887                          * Get a passphrase from the user.
9888                          */
9889                         int ret; /* need not be kept over crReturn */
9890                         s->cur_prompt = new_prompts(ssh->frontend);
9891                         s->cur_prompt->to_server = FALSE;
9892                         s->cur_prompt->name = dupstr("SSH key passphrase");
9893                         add_prompt(s->cur_prompt,
9894                                    dupprintf("Passphrase for key \"%.100s\": ",
9895                                              s->publickey_comment),
9896                                    FALSE);
9897                         ret = get_userpass_input(s->cur_prompt, NULL, 0);
9898                         while (ret < 0) {
9899                             ssh->send_ok = 1;
9900                             crWaitUntilV(!pktin);
9901                             ret = get_userpass_input(s->cur_prompt,
9902                                                      in, inlen);
9903                             ssh->send_ok = 0;
9904                         }
9905                         if (!ret) {
9906                             /* Failed to get a passphrase. Terminate. */
9907                             free_prompts(s->cur_prompt);
9908                             ssh_disconnect(ssh, NULL,
9909                                            "Unable to authenticate",
9910                                            SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER,
9911                                            TRUE);
9912                             crStopV;
9913                         }
9914                         passphrase =
9915                             dupstr(s->cur_prompt->prompts[0]->result);
9916                         free_prompts(s->cur_prompt);
9917                     } else {
9918                         passphrase = NULL; /* no passphrase needed */
9919                     }
9920
9921                     /*
9922                      * Try decrypting the key.
9923                      */
9924                     s->keyfile = conf_get_filename(ssh->conf, CONF_keyfile);
9925                     key = ssh2_load_userkey(s->keyfile, passphrase, &error);
9926                     if (passphrase) {
9927                         /* burn the evidence */
9928                         smemclr(passphrase, strlen(passphrase));
9929                         sfree(passphrase);
9930                     }
9931                     if (key == SSH2_WRONG_PASSPHRASE || key == NULL) {
9932                         if (passphrase &&
9933                             (key == SSH2_WRONG_PASSPHRASE)) {
9934                             c_write_str(ssh, "Wrong passphrase\r\n");
9935                             key = NULL;
9936                             /* and loop again */
9937                         } else {
9938                             c_write_str(ssh, "Unable to load private key (");
9939                             c_write_str(ssh, error);
9940                             c_write_str(ssh, ")\r\n");
9941                             key = NULL;
9942                             break; /* try something else */
9943                         }
9944                     }
9945                 }
9946
9947                 if (key) {
9948                     unsigned char *pkblob, *sigblob, *sigdata;
9949                     int pkblob_len, sigblob_len, sigdata_len;
9950                     int p;
9951
9952                     /*
9953                      * We have loaded the private key and the server
9954                      * has announced that it's willing to accept it.
9955                      * Hallelujah. Generate a signature and send it.
9956                      */
9957                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
9958                     ssh2_pkt_addstring(s->pktout, ssh->username);
9959                     ssh2_pkt_addstring(s->pktout, "ssh-connection");
9960                                                     /* service requested */
9961                     ssh2_pkt_addstring(s->pktout, "publickey");
9962                                                     /* method */
9963                     ssh2_pkt_addbool(s->pktout, TRUE);
9964                                                     /* signature follows */
9965                     ssh2_pkt_addstring(s->pktout, key->alg->name);
9966                     pkblob = key->alg->public_blob(key->data,
9967                                                    &pkblob_len);
9968                     ssh2_pkt_addstring_start(s->pktout);
9969                     ssh2_pkt_addstring_data(s->pktout, (char *)pkblob,
9970                                             pkblob_len);
9971
9972                     /*
9973                      * The data to be signed is:
9974                      *
9975                      *   string  session-id
9976                      *
9977                      * followed by everything so far placed in the
9978                      * outgoing packet.
9979                      */
9980                     sigdata_len = s->pktout->length - 5 + 4 +
9981                         ssh->v2_session_id_len;
9982                     if (ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)
9983                         sigdata_len -= 4;
9984                     sigdata = snewn(sigdata_len, unsigned char);
9985                     p = 0;
9986                     if (!(ssh->remote_bugs & BUG_SSH2_PK_SESSIONID)) {
9987                         PUT_32BIT(sigdata+p, ssh->v2_session_id_len);
9988                         p += 4;
9989                     }
9990                     memcpy(sigdata+p, ssh->v2_session_id,
9991                            ssh->v2_session_id_len);
9992                     p += ssh->v2_session_id_len;
9993                     memcpy(sigdata+p, s->pktout->data + 5,
9994                            s->pktout->length - 5);
9995                     p += s->pktout->length - 5;
9996                     assert(p == sigdata_len);
9997                     sigblob = key->alg->sign(key->data, (char *)sigdata,
9998                                              sigdata_len, &sigblob_len);
9999                     ssh2_add_sigblob(ssh, s->pktout, pkblob, pkblob_len,
10000                                      sigblob, sigblob_len);
10001                     sfree(pkblob);
10002                     sfree(sigblob);
10003                     sfree(sigdata);
10004
10005                     ssh2_pkt_send(ssh, s->pktout);
10006                     logevent("Sent public key signature");
10007                     s->type = AUTH_TYPE_PUBLICKEY;
10008                     key->alg->freekey(key->data);
10009                     sfree(key->comment);
10010                     sfree(key);
10011                 }
10012
10013 #ifndef NO_GSSAPI
10014             } else if (s->can_gssapi && !s->tried_gssapi) {
10015
10016                 /* GSSAPI Authentication */
10017
10018                 int micoffset, len;
10019                 char *data;
10020                 Ssh_gss_buf mic;
10021                 s->type = AUTH_TYPE_GSSAPI;
10022                 s->tried_gssapi = TRUE;
10023                 s->gotit = TRUE;
10024                 ssh->pkt_actx = SSH2_PKTCTX_GSSAPI;
10025
10026                 /*
10027                  * Pick the highest GSS library on the preference
10028                  * list.
10029                  */
10030                 {
10031                     int i, j;
10032                     s->gsslib = NULL;
10033                     for (i = 0; i < ngsslibs; i++) {
10034                         int want_id = conf_get_int_int(ssh->conf,
10035                                                        CONF_ssh_gsslist, i);
10036                         for (j = 0; j < ssh->gsslibs->nlibraries; j++)
10037                             if (ssh->gsslibs->libraries[j].id == want_id) {
10038                                 s->gsslib = &ssh->gsslibs->libraries[j];
10039                                 goto got_gsslib;   /* double break */
10040                             }
10041                     }
10042                     got_gsslib:
10043                     /*
10044                      * We always expect to have found something in
10045                      * the above loop: we only came here if there
10046                      * was at least one viable GSS library, and the
10047                      * preference list should always mention
10048                      * everything and only change the order.
10049                      */
10050                     assert(s->gsslib);
10051                 }
10052
10053                 if (s->gsslib->gsslogmsg)
10054                     logevent(s->gsslib->gsslogmsg);
10055
10056                 /* Sending USERAUTH_REQUEST with "gssapi-with-mic" method */
10057                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
10058                 ssh2_pkt_addstring(s->pktout, ssh->username);
10059                 ssh2_pkt_addstring(s->pktout, "ssh-connection");
10060                 ssh2_pkt_addstring(s->pktout, "gssapi-with-mic");
10061                 logevent("Attempting GSSAPI authentication");
10062
10063                 /* add mechanism info */
10064                 s->gsslib->indicate_mech(s->gsslib, &s->gss_buf);
10065
10066                 /* number of GSSAPI mechanisms */
10067                 ssh2_pkt_adduint32(s->pktout,1);
10068
10069                 /* length of OID + 2 */
10070                 ssh2_pkt_adduint32(s->pktout, s->gss_buf.length + 2);
10071                 ssh2_pkt_addbyte(s->pktout, SSH2_GSS_OIDTYPE);
10072
10073                 /* length of OID */
10074                 ssh2_pkt_addbyte(s->pktout, (unsigned char) s->gss_buf.length);
10075
10076                 ssh_pkt_adddata(s->pktout, s->gss_buf.value,
10077                                 s->gss_buf.length);
10078                 ssh2_pkt_send(ssh, s->pktout);
10079                 crWaitUntilV(pktin);
10080                 if (pktin->type != SSH2_MSG_USERAUTH_GSSAPI_RESPONSE) {
10081                     logevent("GSSAPI authentication request refused");
10082                     continue;
10083                 }
10084
10085                 /* check returned packet ... */
10086
10087                 ssh_pkt_getstring(pktin, &data, &len);
10088                 s->gss_rcvtok.value = data;
10089                 s->gss_rcvtok.length = len;
10090                 if (s->gss_rcvtok.length != s->gss_buf.length + 2 ||
10091                     ((char *)s->gss_rcvtok.value)[0] != SSH2_GSS_OIDTYPE ||
10092                     ((char *)s->gss_rcvtok.value)[1] != s->gss_buf.length ||
10093                     memcmp((char *)s->gss_rcvtok.value + 2,
10094                            s->gss_buf.value,s->gss_buf.length) ) {
10095                     logevent("GSSAPI authentication - wrong response from server");
10096                     continue;
10097                 }
10098
10099                 /* now start running */
10100                 s->gss_stat = s->gsslib->import_name(s->gsslib,
10101                                                      ssh->fullhostname,
10102                                                      &s->gss_srv_name);
10103                 if (s->gss_stat != SSH_GSS_OK) {
10104                     if (s->gss_stat == SSH_GSS_BAD_HOST_NAME)
10105                         logevent("GSSAPI import name failed - Bad service name");
10106                     else
10107                         logevent("GSSAPI import name failed");
10108                     continue;
10109                 }
10110
10111                 /* fetch TGT into GSS engine */
10112                 s->gss_stat = s->gsslib->acquire_cred(s->gsslib, &s->gss_ctx);
10113
10114                 if (s->gss_stat != SSH_GSS_OK) {
10115                     logevent("GSSAPI authentication failed to get credentials");
10116                     s->gsslib->release_name(s->gsslib, &s->gss_srv_name);
10117                     continue;
10118                 }
10119
10120                 /* initial tokens are empty */
10121                 SSH_GSS_CLEAR_BUF(&s->gss_rcvtok);
10122                 SSH_GSS_CLEAR_BUF(&s->gss_sndtok);
10123
10124                 /* now enter the loop */
10125                 do {
10126                     s->gss_stat = s->gsslib->init_sec_context
10127                         (s->gsslib,
10128                          &s->gss_ctx,
10129                          s->gss_srv_name,
10130                          conf_get_int(ssh->conf, CONF_gssapifwd),
10131                          &s->gss_rcvtok,
10132                          &s->gss_sndtok);
10133
10134                     if (s->gss_stat!=SSH_GSS_S_COMPLETE &&
10135                         s->gss_stat!=SSH_GSS_S_CONTINUE_NEEDED) {
10136                         logevent("GSSAPI authentication initialisation failed");
10137
10138                         if (s->gsslib->display_status(s->gsslib, s->gss_ctx,
10139                                                       &s->gss_buf) == SSH_GSS_OK) {
10140                             logevent(s->gss_buf.value);
10141                             sfree(s->gss_buf.value);
10142                         }
10143
10144                         break;
10145                     }
10146                     logevent("GSSAPI authentication initialised");
10147
10148                     /* Client and server now exchange tokens until GSSAPI
10149                      * no longer says CONTINUE_NEEDED */
10150
10151                     if (s->gss_sndtok.length != 0) {
10152                         s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_GSSAPI_TOKEN);
10153                         ssh_pkt_addstring_start(s->pktout);
10154                         ssh_pkt_addstring_data(s->pktout,s->gss_sndtok.value,s->gss_sndtok.length);
10155                         ssh2_pkt_send(ssh, s->pktout);
10156                         s->gsslib->free_tok(s->gsslib, &s->gss_sndtok);
10157                     }
10158
10159                     if (s->gss_stat == SSH_GSS_S_CONTINUE_NEEDED) {
10160                         crWaitUntilV(pktin);
10161                         if (pktin->type != SSH2_MSG_USERAUTH_GSSAPI_TOKEN) {
10162                             logevent("GSSAPI authentication - bad server response");
10163                             s->gss_stat = SSH_GSS_FAILURE;
10164                             break;
10165                         }
10166                         ssh_pkt_getstring(pktin, &data, &len);
10167                         s->gss_rcvtok.value = data;
10168                         s->gss_rcvtok.length = len;
10169                     }
10170                 } while (s-> gss_stat == SSH_GSS_S_CONTINUE_NEEDED);
10171
10172                 if (s->gss_stat != SSH_GSS_OK) {
10173                     s->gsslib->release_name(s->gsslib, &s->gss_srv_name);
10174                     s->gsslib->release_cred(s->gsslib, &s->gss_ctx);
10175                     continue;
10176                 }
10177                 logevent("GSSAPI authentication loop finished OK");
10178
10179                 /* Now send the MIC */
10180
10181                 s->pktout = ssh2_pkt_init(0);
10182                 micoffset = s->pktout->length;
10183                 ssh_pkt_addstring_start(s->pktout);
10184                 ssh_pkt_addstring_data(s->pktout, (char *)ssh->v2_session_id, ssh->v2_session_id_len);
10185                 ssh_pkt_addbyte(s->pktout, SSH2_MSG_USERAUTH_REQUEST);
10186                 ssh_pkt_addstring(s->pktout, ssh->username);
10187                 ssh_pkt_addstring(s->pktout, "ssh-connection");
10188                 ssh_pkt_addstring(s->pktout, "gssapi-with-mic");
10189
10190                 s->gss_buf.value = (char *)s->pktout->data + micoffset;
10191                 s->gss_buf.length = s->pktout->length - micoffset;
10192
10193                 s->gsslib->get_mic(s->gsslib, s->gss_ctx, &s->gss_buf, &mic);
10194                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_GSSAPI_MIC);
10195                 ssh_pkt_addstring_start(s->pktout);
10196                 ssh_pkt_addstring_data(s->pktout, mic.value, mic.length);
10197                 ssh2_pkt_send(ssh, s->pktout);
10198                 s->gsslib->free_mic(s->gsslib, &mic);
10199
10200                 s->gotit = FALSE;
10201
10202                 s->gsslib->release_name(s->gsslib, &s->gss_srv_name);
10203                 s->gsslib->release_cred(s->gsslib, &s->gss_ctx);
10204                 continue;
10205 #endif
10206             } else if (s->can_keyb_inter && !s->kbd_inter_refused) {
10207
10208                 /*
10209                  * Keyboard-interactive authentication.
10210                  */
10211
10212                 s->type = AUTH_TYPE_KEYBOARD_INTERACTIVE;
10213
10214                 ssh->pkt_actx = SSH2_PKTCTX_KBDINTER;
10215
10216                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
10217                 ssh2_pkt_addstring(s->pktout, ssh->username);
10218                 ssh2_pkt_addstring(s->pktout, "ssh-connection");
10219                                                         /* service requested */
10220                 ssh2_pkt_addstring(s->pktout, "keyboard-interactive");
10221                                                         /* method */
10222                 ssh2_pkt_addstring(s->pktout, "");      /* lang */
10223                 ssh2_pkt_addstring(s->pktout, "");      /* submethods */
10224                 ssh2_pkt_send(ssh, s->pktout);
10225                 
10226                 logevent("Attempting keyboard-interactive authentication");
10227
10228                 crWaitUntilV(pktin);
10229                 if (pktin->type != SSH2_MSG_USERAUTH_INFO_REQUEST) {
10230                     /* Server is not willing to do keyboard-interactive
10231                      * at all (or, bizarrely but legally, accepts the
10232                      * user without actually issuing any prompts).
10233                      * Give up on it entirely. */
10234                     s->gotit = TRUE;
10235                     s->type = AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET;
10236                     s->kbd_inter_refused = TRUE; /* don't try it again */
10237                     continue;
10238                 }
10239
10240                 /*
10241                  * Loop while the server continues to send INFO_REQUESTs.
10242                  */
10243                 while (pktin->type == SSH2_MSG_USERAUTH_INFO_REQUEST) {
10244
10245                     char *name, *inst, *lang;
10246                     int name_len, inst_len, lang_len;
10247                     int i;
10248
10249                     /*
10250                      * We've got a fresh USERAUTH_INFO_REQUEST.
10251                      * Get the preamble and start building a prompt.
10252                      */
10253                     ssh_pkt_getstring(pktin, &name, &name_len);
10254                     ssh_pkt_getstring(pktin, &inst, &inst_len);
10255                     ssh_pkt_getstring(pktin, &lang, &lang_len);
10256                     s->cur_prompt = new_prompts(ssh->frontend);
10257                     s->cur_prompt->to_server = TRUE;
10258
10259                     /*
10260                      * Get any prompt(s) from the packet.
10261                      */
10262                     s->num_prompts = ssh_pkt_getuint32(pktin);
10263                     for (i = 0; i < s->num_prompts; i++) {
10264                         char *prompt;
10265                         int prompt_len;
10266                         int echo;
10267                         static char noprompt[] =
10268                             "<server failed to send prompt>: ";
10269
10270                         ssh_pkt_getstring(pktin, &prompt, &prompt_len);
10271                         echo = ssh2_pkt_getbool(pktin);
10272                         if (!prompt_len) {
10273                             prompt = noprompt;
10274                             prompt_len = lenof(noprompt)-1;
10275                         }
10276                         add_prompt(s->cur_prompt,
10277                                    dupprintf("%.*s", prompt_len, prompt),
10278                                    echo);
10279                     }
10280
10281                     if (name_len) {
10282                         /* FIXME: better prefix to distinguish from
10283                          * local prompts? */
10284                         s->cur_prompt->name =
10285                             dupprintf("SSH server: %.*s", name_len, name);
10286                         s->cur_prompt->name_reqd = TRUE;
10287                     } else {
10288                         s->cur_prompt->name =
10289                             dupstr("SSH server authentication");
10290                         s->cur_prompt->name_reqd = FALSE;
10291                     }
10292                     /* We add a prefix to try to make it clear that a prompt
10293                      * has come from the server.
10294                      * FIXME: ugly to print "Using..." in prompt _every_
10295                      * time round. Can this be done more subtly? */
10296                     /* Special case: for reasons best known to themselves,
10297                      * some servers send k-i requests with no prompts and
10298                      * nothing to display. Keep quiet in this case. */
10299                     if (s->num_prompts || name_len || inst_len) {
10300                         s->cur_prompt->instruction =
10301                             dupprintf("Using keyboard-interactive authentication.%s%.*s",
10302                                       inst_len ? "\n" : "", inst_len, inst);
10303                         s->cur_prompt->instr_reqd = TRUE;
10304                     } else {
10305                         s->cur_prompt->instr_reqd = FALSE;
10306                     }
10307
10308                     /*
10309                      * Display any instructions, and get the user's
10310                      * response(s).
10311                      */
10312                     {
10313                         int ret; /* not live over crReturn */
10314                         ret = get_userpass_input(s->cur_prompt, NULL, 0);
10315                         while (ret < 0) {
10316                             ssh->send_ok = 1;
10317                             crWaitUntilV(!pktin);
10318                             ret = get_userpass_input(s->cur_prompt, in, inlen);
10319                             ssh->send_ok = 0;
10320                         }
10321                         if (!ret) {
10322                             /*
10323                              * Failed to get responses. Terminate.
10324                              */
10325                             free_prompts(s->cur_prompt);
10326                             ssh_disconnect(ssh, NULL, "Unable to authenticate",
10327                                            SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER,
10328                                            TRUE);
10329                             crStopV;
10330                         }
10331                     }
10332
10333                     /*
10334                      * Send the response(s) to the server.
10335                      */
10336                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_INFO_RESPONSE);
10337                     ssh2_pkt_adduint32(s->pktout, s->num_prompts);
10338                     for (i=0; i < s->num_prompts; i++) {
10339                         ssh2_pkt_addstring(s->pktout,
10340                                            s->cur_prompt->prompts[i]->result);
10341                     }
10342                     ssh2_pkt_send_with_padding(ssh, s->pktout, 256);
10343
10344                     /*
10345                      * Free the prompts structure from this iteration.
10346                      * If there's another, a new one will be allocated
10347                      * when we return to the top of this while loop.
10348                      */
10349                     free_prompts(s->cur_prompt);
10350
10351                     /*
10352                      * Get the next packet in case it's another
10353                      * INFO_REQUEST.
10354                      */
10355                     crWaitUntilV(pktin);
10356
10357                 }
10358
10359                 /*
10360                  * We should have SUCCESS or FAILURE now.
10361                  */
10362                 s->gotit = TRUE;
10363
10364             } else if (s->can_passwd) {
10365
10366                 /*
10367                  * Plain old password authentication.
10368                  */
10369                 int ret; /* not live over crReturn */
10370                 int changereq_first_time; /* not live over crReturn */
10371
10372                 ssh->pkt_actx = SSH2_PKTCTX_PASSWORD;
10373
10374                 s->cur_prompt = new_prompts(ssh->frontend);
10375                 s->cur_prompt->to_server = TRUE;
10376                 s->cur_prompt->name = dupstr("SSH password");
10377                 add_prompt(s->cur_prompt, dupprintf("%s@%s's password: ",
10378                                                     ssh->username,
10379                                                     ssh->savedhost),
10380                            FALSE);
10381
10382                 ret = get_userpass_input(s->cur_prompt, NULL, 0);
10383                 while (ret < 0) {
10384                     ssh->send_ok = 1;
10385                     crWaitUntilV(!pktin);
10386                     ret = get_userpass_input(s->cur_prompt, in, inlen);
10387                     ssh->send_ok = 0;
10388                 }
10389                 if (!ret) {
10390                     /*
10391                      * Failed to get responses. Terminate.
10392                      */
10393                     free_prompts(s->cur_prompt);
10394                     ssh_disconnect(ssh, NULL, "Unable to authenticate",
10395                                    SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER,
10396                                    TRUE);
10397                     crStopV;
10398                 }
10399                 /*
10400                  * Squirrel away the password. (We may need it later if
10401                  * asked to change it.)
10402                  */
10403                 s->password = dupstr(s->cur_prompt->prompts[0]->result);
10404                 free_prompts(s->cur_prompt);
10405
10406                 /*
10407                  * Send the password packet.
10408                  *
10409                  * We pad out the password packet to 256 bytes to make
10410                  * it harder for an attacker to find the length of the
10411                  * user's password.
10412                  *
10413                  * Anyone using a password longer than 256 bytes
10414                  * probably doesn't have much to worry about from
10415                  * people who find out how long their password is!
10416                  */
10417                 s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
10418                 ssh2_pkt_addstring(s->pktout, ssh->username);
10419                 ssh2_pkt_addstring(s->pktout, "ssh-connection");
10420                                                         /* service requested */
10421                 ssh2_pkt_addstring(s->pktout, "password");
10422                 ssh2_pkt_addbool(s->pktout, FALSE);
10423                 ssh2_pkt_addstring(s->pktout, s->password);
10424                 ssh2_pkt_send_with_padding(ssh, s->pktout, 256);
10425                 logevent("Sent password");
10426                 s->type = AUTH_TYPE_PASSWORD;
10427
10428                 /*
10429                  * Wait for next packet, in case it's a password change
10430                  * request.
10431                  */
10432                 crWaitUntilV(pktin);
10433                 changereq_first_time = TRUE;
10434
10435                 while (pktin->type == SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ) {
10436
10437                     /* 
10438                      * We're being asked for a new password
10439                      * (perhaps not for the first time).
10440                      * Loop until the server accepts it.
10441                      */
10442
10443                     int got_new = FALSE; /* not live over crReturn */
10444                     char *prompt;   /* not live over crReturn */
10445                     int prompt_len; /* not live over crReturn */
10446                     
10447                     {
10448                         const char *msg;
10449                         if (changereq_first_time)
10450                             msg = "Server requested password change";
10451                         else
10452                             msg = "Server rejected new password";
10453                         logevent(msg);
10454                         c_write_str(ssh, msg);
10455                         c_write_str(ssh, "\r\n");
10456                     }
10457
10458                     ssh_pkt_getstring(pktin, &prompt, &prompt_len);
10459
10460                     s->cur_prompt = new_prompts(ssh->frontend);
10461                     s->cur_prompt->to_server = TRUE;
10462                     s->cur_prompt->name = dupstr("New SSH password");
10463                     s->cur_prompt->instruction =
10464                         dupprintf("%.*s", prompt_len, NULLTOEMPTY(prompt));
10465                     s->cur_prompt->instr_reqd = TRUE;
10466                     /*
10467                      * There's no explicit requirement in the protocol
10468                      * for the "old" passwords in the original and
10469                      * password-change messages to be the same, and
10470                      * apparently some Cisco kit supports password change
10471                      * by the user entering a blank password originally
10472                      * and the real password subsequently, so,
10473                      * reluctantly, we prompt for the old password again.
10474                      *
10475                      * (On the other hand, some servers don't even bother
10476                      * to check this field.)
10477                      */
10478                     add_prompt(s->cur_prompt,
10479                                dupstr("Current password (blank for previously entered password): "),
10480                                FALSE);
10481                     add_prompt(s->cur_prompt, dupstr("Enter new password: "),
10482                                FALSE);
10483                     add_prompt(s->cur_prompt, dupstr("Confirm new password: "),
10484                                FALSE);
10485
10486                     /*
10487                      * Loop until the user manages to enter the same
10488                      * password twice.
10489                      */
10490                     while (!got_new) {
10491
10492                         ret = get_userpass_input(s->cur_prompt, NULL, 0);
10493                         while (ret < 0) {
10494                             ssh->send_ok = 1;
10495                             crWaitUntilV(!pktin);
10496                             ret = get_userpass_input(s->cur_prompt, in, inlen);
10497                             ssh->send_ok = 0;
10498                         }
10499                         if (!ret) {
10500                             /*
10501                              * Failed to get responses. Terminate.
10502                              */
10503                             /* burn the evidence */
10504                             free_prompts(s->cur_prompt);
10505                             smemclr(s->password, strlen(s->password));
10506                             sfree(s->password);
10507                             ssh_disconnect(ssh, NULL, "Unable to authenticate",
10508                                            SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER,
10509                                            TRUE);
10510                             crStopV;
10511                         }
10512
10513                         /*
10514                          * If the user specified a new original password
10515                          * (IYSWIM), overwrite any previously specified
10516                          * one.
10517                          * (A side effect is that the user doesn't have to
10518                          * re-enter it if they louse up the new password.)
10519                          */
10520                         if (s->cur_prompt->prompts[0]->result[0]) {
10521                             smemclr(s->password, strlen(s->password));
10522                                 /* burn the evidence */
10523                             sfree(s->password);
10524                             s->password =
10525                                 dupstr(s->cur_prompt->prompts[0]->result);
10526                         }
10527
10528                         /*
10529                          * Check the two new passwords match.
10530                          */
10531                         got_new = (strcmp(s->cur_prompt->prompts[1]->result,
10532                                           s->cur_prompt->prompts[2]->result)
10533                                    == 0);
10534                         if (!got_new)
10535                             /* They don't. Silly user. */
10536                             c_write_str(ssh, "Passwords do not match\r\n");
10537
10538                     }
10539
10540                     /*
10541                      * Send the new password (along with the old one).
10542                      * (see above for padding rationale)
10543                      */
10544                     s->pktout = ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
10545                     ssh2_pkt_addstring(s->pktout, ssh->username);
10546                     ssh2_pkt_addstring(s->pktout, "ssh-connection");
10547                                                         /* service requested */
10548                     ssh2_pkt_addstring(s->pktout, "password");
10549                     ssh2_pkt_addbool(s->pktout, TRUE);
10550                     ssh2_pkt_addstring(s->pktout, s->password);
10551                     ssh2_pkt_addstring(s->pktout,
10552                                        s->cur_prompt->prompts[1]->result);
10553                     free_prompts(s->cur_prompt);
10554                     ssh2_pkt_send_with_padding(ssh, s->pktout, 256);
10555                     logevent("Sent new password");
10556                     
10557                     /*
10558                      * Now see what the server has to say about it.
10559                      * (If it's CHANGEREQ again, it's not happy with the
10560                      * new password.)
10561                      */
10562                     crWaitUntilV(pktin);
10563                     changereq_first_time = FALSE;
10564
10565                 }
10566
10567                 /*
10568                  * We need to reexamine the current pktin at the top
10569                  * of the loop. Either:
10570                  *  - we weren't asked to change password at all, in
10571                  *    which case it's a SUCCESS or FAILURE with the
10572                  *    usual meaning
10573                  *  - we sent a new password, and the server was
10574                  *    either OK with it (SUCCESS or FAILURE w/partial
10575                  *    success) or unhappy with the _old_ password
10576                  *    (FAILURE w/o partial success)
10577                  * In any of these cases, we go back to the top of
10578                  * the loop and start again.
10579                  */
10580                 s->gotit = TRUE;
10581
10582                 /*
10583                  * We don't need the old password any more, in any
10584                  * case. Burn the evidence.
10585                  */
10586                 smemclr(s->password, strlen(s->password));
10587                 sfree(s->password);
10588
10589             } else {
10590                 char *str = dupprintf("No supported authentication methods available"
10591                                       " (server sent: %.*s)",
10592                                       methlen, methods);
10593
10594                 ssh_disconnect(ssh, str,
10595                                "No supported authentication methods available",
10596                                SSH2_DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE,
10597                                FALSE);
10598                 sfree(str);
10599
10600                 crStopV;
10601
10602             }
10603
10604         }
10605     }
10606     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] = NULL;
10607
10608     /* Clear up various bits and pieces from authentication. */
10609     if (s->publickey_blob) {
10610         sfree(s->publickey_algorithm);
10611         sfree(s->publickey_blob);
10612         sfree(s->publickey_comment);
10613     }
10614     if (s->agent_response)
10615         sfree(s->agent_response);
10616
10617     if (s->userauth_success && !ssh->bare_connection) {
10618         /*
10619          * We've just received USERAUTH_SUCCESS, and we haven't sent any
10620          * packets since. Signal the transport layer to consider enacting
10621          * delayed compression.
10622          *
10623          * (Relying on we_are_in is not sufficient, as
10624          * draft-miller-secsh-compression-delayed is quite clear that it
10625          * triggers on USERAUTH_SUCCESS specifically, and we_are_in can
10626          * become set for other reasons.)
10627          */
10628         do_ssh2_transport(ssh, "enabling delayed compression", -2, NULL);
10629     }
10630
10631     ssh->channels = newtree234(ssh_channelcmp);
10632
10633     /*
10634      * Set up handlers for some connection protocol messages, so we
10635      * don't have to handle them repeatedly in this coroutine.
10636      */
10637     ssh->packet_dispatch[SSH2_MSG_CHANNEL_WINDOW_ADJUST] =
10638         ssh2_msg_channel_window_adjust;
10639     ssh->packet_dispatch[SSH2_MSG_GLOBAL_REQUEST] =
10640         ssh2_msg_global_request;
10641
10642     /*
10643      * Create the main session channel.
10644      */
10645     if (conf_get_int(ssh->conf, CONF_ssh_no_shell)) {
10646         ssh->mainchan = NULL;
10647     } else {
10648         ssh->mainchan = snew(struct ssh_channel);
10649         ssh->mainchan->ssh = ssh;
10650         ssh_channel_init(ssh->mainchan);
10651
10652         if (*conf_get_str(ssh->conf, CONF_ssh_nc_host)) {
10653             /*
10654              * Just start a direct-tcpip channel and use it as the main
10655              * channel.
10656              */
10657             ssh_send_port_open(ssh->mainchan,
10658                                conf_get_str(ssh->conf, CONF_ssh_nc_host),
10659                                conf_get_int(ssh->conf, CONF_ssh_nc_port),
10660                                "main channel");
10661             ssh->ncmode = TRUE;
10662         } else {
10663             s->pktout = ssh2_chanopen_init(ssh->mainchan, "session");
10664             logevent("Opening session as main channel");
10665             ssh2_pkt_send(ssh, s->pktout);
10666             ssh->ncmode = FALSE;
10667         }
10668         crWaitUntilV(pktin);
10669         if (pktin->type != SSH2_MSG_CHANNEL_OPEN_CONFIRMATION) {
10670             bombout(("Server refused to open channel"));
10671             crStopV;
10672             /* FIXME: error data comes back in FAILURE packet */
10673         }
10674         if (ssh_pkt_getuint32(pktin) != ssh->mainchan->localid) {
10675             bombout(("Server's channel confirmation cited wrong channel"));
10676             crStopV;
10677         }
10678         ssh->mainchan->remoteid = ssh_pkt_getuint32(pktin);
10679         ssh->mainchan->halfopen = FALSE;
10680         ssh->mainchan->type = CHAN_MAINSESSION;
10681         ssh->mainchan->v.v2.remwindow = ssh_pkt_getuint32(pktin);
10682         ssh->mainchan->v.v2.remmaxpkt = ssh_pkt_getuint32(pktin);
10683         update_specials_menu(ssh->frontend);
10684         logevent("Opened main channel");
10685     }
10686
10687     /*
10688      * Now we have a channel, make dispatch table entries for
10689      * general channel-based messages.
10690      */
10691     ssh->packet_dispatch[SSH2_MSG_CHANNEL_DATA] =
10692     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EXTENDED_DATA] =
10693         ssh2_msg_channel_data;
10694     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EOF] = ssh2_msg_channel_eof;
10695     ssh->packet_dispatch[SSH2_MSG_CHANNEL_CLOSE] = ssh2_msg_channel_close;
10696     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_CONFIRMATION] =
10697         ssh2_msg_channel_open_confirmation;
10698     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_FAILURE] =
10699         ssh2_msg_channel_open_failure;
10700     ssh->packet_dispatch[SSH2_MSG_CHANNEL_REQUEST] =
10701         ssh2_msg_channel_request;
10702     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN] =
10703         ssh2_msg_channel_open;
10704     ssh->packet_dispatch[SSH2_MSG_CHANNEL_SUCCESS] = ssh2_msg_channel_response;
10705     ssh->packet_dispatch[SSH2_MSG_CHANNEL_FAILURE] = ssh2_msg_channel_response;
10706
10707     /*
10708      * Now the connection protocol is properly up and running, with
10709      * all those dispatch table entries, so it's safe to let
10710      * downstreams start trying to open extra channels through us.
10711      */
10712     if (ssh->connshare)
10713         share_activate(ssh->connshare, ssh->v_s);
10714
10715     if (ssh->mainchan && ssh_is_simple(ssh)) {
10716         /*
10717          * This message indicates to the server that we promise
10718          * not to try to run any other channel in parallel with
10719          * this one, so it's safe for it to advertise a very large
10720          * window and leave the flow control to TCP.
10721          */
10722         s->pktout = ssh2_chanreq_init(ssh->mainchan,
10723                                       "simple@putty.projects.tartarus.org",
10724                                       NULL, NULL);
10725         ssh2_pkt_send(ssh, s->pktout);
10726     }
10727
10728     /*
10729      * Enable port forwardings.
10730      */
10731     ssh_setup_portfwd(ssh, ssh->conf);
10732
10733     if (ssh->mainchan && !ssh->ncmode) {
10734         /*
10735          * Send the CHANNEL_REQUESTS for the main session channel.
10736          * Each one is handled by its own little asynchronous
10737          * co-routine.
10738          */
10739
10740         /* Potentially enable X11 forwarding. */
10741         if (conf_get_int(ssh->conf, CONF_x11_forward)) {
10742             ssh->x11disp =
10743                 x11_setup_display(conf_get_str(ssh->conf, CONF_x11_display),
10744                                   ssh->conf);
10745             if (!ssh->x11disp) {
10746                 /* FIXME: return an error message from x11_setup_display */
10747                 logevent("X11 forwarding not enabled: unable to"
10748                          " initialise X display");
10749             } else {
10750                 ssh->x11auth = x11_invent_fake_auth
10751                     (ssh->x11authtree, conf_get_int(ssh->conf, CONF_x11_auth));
10752                 ssh->x11auth->disp = ssh->x11disp;
10753
10754                 ssh2_setup_x11(ssh->mainchan, NULL, NULL);
10755             }
10756         }
10757
10758         /* Potentially enable agent forwarding. */
10759         if (ssh_agent_forwarding_permitted(ssh))
10760             ssh2_setup_agent(ssh->mainchan, NULL, NULL);
10761
10762         /* Now allocate a pty for the session. */
10763         if (!conf_get_int(ssh->conf, CONF_nopty))
10764             ssh2_setup_pty(ssh->mainchan, NULL, NULL);
10765
10766         /* Send environment variables. */
10767         ssh2_setup_env(ssh->mainchan, NULL, NULL);
10768
10769         /*
10770          * Start a shell or a remote command. We may have to attempt
10771          * this twice if the config data has provided a second choice
10772          * of command.
10773          */
10774         while (1) {
10775             int subsys;
10776             char *cmd;
10777
10778             if (ssh->fallback_cmd) {
10779                 subsys = conf_get_int(ssh->conf, CONF_ssh_subsys2);
10780                 cmd = conf_get_str(ssh->conf, CONF_remote_cmd2);
10781             } else {
10782                 subsys = conf_get_int(ssh->conf, CONF_ssh_subsys);
10783                 cmd = conf_get_str(ssh->conf, CONF_remote_cmd);
10784             }
10785
10786             if (subsys) {
10787                 s->pktout = ssh2_chanreq_init(ssh->mainchan, "subsystem",
10788                                               ssh2_response_authconn, NULL);
10789                 ssh2_pkt_addstring(s->pktout, cmd);
10790             } else if (*cmd) {
10791                 s->pktout = ssh2_chanreq_init(ssh->mainchan, "exec",
10792                                               ssh2_response_authconn, NULL);
10793                 ssh2_pkt_addstring(s->pktout, cmd);
10794             } else {
10795                 s->pktout = ssh2_chanreq_init(ssh->mainchan, "shell",
10796                                               ssh2_response_authconn, NULL);
10797             }
10798             ssh2_pkt_send(ssh, s->pktout);
10799
10800             crWaitUntilV(pktin);
10801
10802             if (pktin->type != SSH2_MSG_CHANNEL_SUCCESS) {
10803                 if (pktin->type != SSH2_MSG_CHANNEL_FAILURE) {
10804                     bombout(("Unexpected response to shell/command request:"
10805                              " packet type %d", pktin->type));
10806                     crStopV;
10807                 }
10808                 /*
10809                  * We failed to start the command. If this is the
10810                  * fallback command, we really are finished; if it's
10811                  * not, and if the fallback command exists, try falling
10812                  * back to it before complaining.
10813                  */
10814                 if (!ssh->fallback_cmd &&
10815                     *conf_get_str(ssh->conf, CONF_remote_cmd2)) {
10816                     logevent("Primary command failed; attempting fallback");
10817                     ssh->fallback_cmd = TRUE;
10818                     continue;
10819                 }
10820                 bombout(("Server refused to start a shell/command"));
10821                 crStopV;
10822             } else {
10823                 logevent("Started a shell/command");
10824             }
10825             break;
10826         }
10827     } else {
10828         ssh->editing = ssh->echoing = TRUE;
10829     }
10830
10831     ssh->state = SSH_STATE_SESSION;
10832     if (ssh->size_needed)
10833         ssh_size(ssh, ssh->term_width, ssh->term_height);
10834     if (ssh->eof_needed)
10835         ssh_special(ssh, TS_EOF);
10836
10837     /*
10838      * Transfer data!
10839      */
10840     if (ssh->ldisc)
10841         ldisc_echoedit_update(ssh->ldisc);  /* cause ldisc to notice changes */
10842     if (ssh->mainchan)
10843         ssh->send_ok = 1;
10844     while (1) {
10845         crReturnV;
10846         if (pktin) {
10847
10848             /*
10849              * _All_ the connection-layer packets we expect to
10850              * receive are now handled by the dispatch table.
10851              * Anything that reaches here must be bogus.
10852              */
10853
10854             bombout(("Strange packet received: type %d", pktin->type));
10855             crStopV;
10856         } else if (ssh->mainchan) {
10857             /*
10858              * We have spare data. Add it to the channel buffer.
10859              */
10860             ssh_send_channel_data(ssh->mainchan, (char *)in, inlen);
10861         }
10862     }
10863
10864     crFinishV;
10865 }
10866
10867 /*
10868  * Handlers for SSH-2 messages that might arrive at any moment.
10869  */
10870 static void ssh2_msg_disconnect(Ssh ssh, struct Packet *pktin)
10871 {
10872     /* log reason code in disconnect message */
10873     char *buf, *msg;
10874     int reason, msglen;
10875
10876     reason = ssh_pkt_getuint32(pktin);
10877     ssh_pkt_getstring(pktin, &msg, &msglen);
10878
10879     if (reason > 0 && reason < lenof(ssh2_disconnect_reasons)) {
10880         buf = dupprintf("Received disconnect message (%s)",
10881                         ssh2_disconnect_reasons[reason]);
10882     } else {
10883         buf = dupprintf("Received disconnect message (unknown"
10884                         " type %d)", reason);
10885     }
10886     logevent(buf);
10887     sfree(buf);
10888     buf = dupprintf("Disconnection message text: %.*s",
10889                     msglen, NULLTOEMPTY(msg));
10890     logevent(buf);
10891     bombout(("Server sent disconnect message\ntype %d (%s):\n\"%.*s\"",
10892              reason,
10893              (reason > 0 && reason < lenof(ssh2_disconnect_reasons)) ?
10894              ssh2_disconnect_reasons[reason] : "unknown",
10895              msglen, NULLTOEMPTY(msg)));
10896     sfree(buf);
10897 }
10898
10899 static void ssh2_msg_debug(Ssh ssh, struct Packet *pktin)
10900 {
10901     /* log the debug message */
10902     char *msg;
10903     int msglen;
10904
10905     /* XXX maybe we should actually take notice of the return value */
10906     ssh2_pkt_getbool(pktin);
10907     ssh_pkt_getstring(pktin, &msg, &msglen);
10908
10909     logeventf(ssh, "Remote debug message: %.*s", msglen, NULLTOEMPTY(msg));
10910 }
10911
10912 static void ssh2_msg_transport(Ssh ssh, struct Packet *pktin)
10913 {
10914     do_ssh2_transport(ssh, NULL, 0, pktin);
10915 }
10916
10917 /*
10918  * Called if we receive a packet that isn't allowed by the protocol.
10919  * This only applies to packets whose meaning PuTTY understands.
10920  * Entirely unknown packets are handled below.
10921  */
10922 static void ssh2_msg_unexpected(Ssh ssh, struct Packet *pktin)
10923 {
10924     char *buf = dupprintf("Server protocol violation: unexpected %s packet",
10925                           ssh2_pkt_type(ssh->pkt_kctx, ssh->pkt_actx,
10926                                         pktin->type));
10927     ssh_disconnect(ssh, NULL, buf, SSH2_DISCONNECT_PROTOCOL_ERROR, FALSE);
10928     sfree(buf);
10929 }
10930
10931 static void ssh2_msg_something_unimplemented(Ssh ssh, struct Packet *pktin)
10932 {
10933     struct Packet *pktout;
10934     pktout = ssh2_pkt_init(SSH2_MSG_UNIMPLEMENTED);
10935     ssh2_pkt_adduint32(pktout, pktin->sequence);
10936     /*
10937      * UNIMPLEMENTED messages MUST appear in the same order as the
10938      * messages they respond to. Hence, never queue them.
10939      */
10940     ssh2_pkt_send_noqueue(ssh, pktout);
10941 }
10942
10943 /*
10944  * Handle the top-level SSH-2 protocol.
10945  */
10946 static void ssh2_protocol_setup(Ssh ssh)
10947 {
10948     int i;
10949
10950     /*
10951      * Most messages cause SSH2_MSG_UNIMPLEMENTED.
10952      */
10953     for (i = 0; i < 256; i++)
10954         ssh->packet_dispatch[i] = ssh2_msg_something_unimplemented;
10955
10956     /*
10957      * Initially, we only accept transport messages (and a few generic
10958      * ones).  do_ssh2_authconn will add more when it starts.
10959      * Messages that are understood but not currently acceptable go to
10960      * ssh2_msg_unexpected.
10961      */
10962     ssh->packet_dispatch[SSH2_MSG_UNIMPLEMENTED] = ssh2_msg_unexpected;
10963     ssh->packet_dispatch[SSH2_MSG_SERVICE_REQUEST] = ssh2_msg_unexpected;
10964     ssh->packet_dispatch[SSH2_MSG_SERVICE_ACCEPT] = ssh2_msg_unexpected;
10965     ssh->packet_dispatch[SSH2_MSG_KEXINIT] = ssh2_msg_transport;
10966     ssh->packet_dispatch[SSH2_MSG_NEWKEYS] = ssh2_msg_transport;
10967     ssh->packet_dispatch[SSH2_MSG_KEXDH_INIT] = ssh2_msg_transport;
10968     ssh->packet_dispatch[SSH2_MSG_KEXDH_REPLY] = ssh2_msg_transport;
10969     /* ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_REQUEST] = ssh2_msg_transport; duplicate case value */
10970     /* ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_GROUP] = ssh2_msg_transport; duplicate case value */
10971     ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_INIT] = ssh2_msg_transport;
10972     ssh->packet_dispatch[SSH2_MSG_KEX_DH_GEX_REPLY] = ssh2_msg_transport;
10973     ssh->packet_dispatch[SSH2_MSG_USERAUTH_REQUEST] = ssh2_msg_unexpected;
10974     ssh->packet_dispatch[SSH2_MSG_USERAUTH_FAILURE] = ssh2_msg_unexpected;
10975     ssh->packet_dispatch[SSH2_MSG_USERAUTH_SUCCESS] = ssh2_msg_unexpected;
10976     ssh->packet_dispatch[SSH2_MSG_USERAUTH_BANNER] = ssh2_msg_unexpected;
10977     ssh->packet_dispatch[SSH2_MSG_USERAUTH_PK_OK] = ssh2_msg_unexpected;
10978     /* ssh->packet_dispatch[SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ] = ssh2_msg_unexpected; duplicate case value */
10979     /* ssh->packet_dispatch[SSH2_MSG_USERAUTH_INFO_REQUEST] = ssh2_msg_unexpected; duplicate case value */
10980     ssh->packet_dispatch[SSH2_MSG_USERAUTH_INFO_RESPONSE] = ssh2_msg_unexpected;
10981     ssh->packet_dispatch[SSH2_MSG_GLOBAL_REQUEST] = ssh2_msg_unexpected;
10982     ssh->packet_dispatch[SSH2_MSG_REQUEST_SUCCESS] = ssh2_msg_unexpected;
10983     ssh->packet_dispatch[SSH2_MSG_REQUEST_FAILURE] = ssh2_msg_unexpected;
10984     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN] = ssh2_msg_unexpected;
10985     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_CONFIRMATION] = ssh2_msg_unexpected;
10986     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_FAILURE] = ssh2_msg_unexpected;
10987     ssh->packet_dispatch[SSH2_MSG_CHANNEL_WINDOW_ADJUST] = ssh2_msg_unexpected;
10988     ssh->packet_dispatch[SSH2_MSG_CHANNEL_DATA] = ssh2_msg_unexpected;
10989     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EXTENDED_DATA] = ssh2_msg_unexpected;
10990     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EOF] = ssh2_msg_unexpected;
10991     ssh->packet_dispatch[SSH2_MSG_CHANNEL_CLOSE] = ssh2_msg_unexpected;
10992     ssh->packet_dispatch[SSH2_MSG_CHANNEL_REQUEST] = ssh2_msg_unexpected;
10993     ssh->packet_dispatch[SSH2_MSG_CHANNEL_SUCCESS] = ssh2_msg_unexpected;
10994     ssh->packet_dispatch[SSH2_MSG_CHANNEL_FAILURE] = ssh2_msg_unexpected;
10995
10996     /*
10997      * These messages have a special handler from the start.
10998      */
10999     ssh->packet_dispatch[SSH2_MSG_DISCONNECT] = ssh2_msg_disconnect;
11000     ssh->packet_dispatch[SSH2_MSG_IGNORE] = ssh_msg_ignore; /* shared with SSH-1 */
11001     ssh->packet_dispatch[SSH2_MSG_DEBUG] = ssh2_msg_debug;
11002 }
11003
11004 static void ssh2_bare_connection_protocol_setup(Ssh ssh)
11005 {
11006     int i;
11007
11008     /*
11009      * Most messages cause SSH2_MSG_UNIMPLEMENTED.
11010      */
11011     for (i = 0; i < 256; i++)
11012         ssh->packet_dispatch[i] = ssh2_msg_something_unimplemented;
11013
11014     /*
11015      * Initially, we set all ssh-connection messages to 'unexpected';
11016      * do_ssh2_authconn will fill things in properly. We also handle a
11017      * couple of messages from the transport protocol which aren't
11018      * related to key exchange (UNIMPLEMENTED, IGNORE, DEBUG,
11019      * DISCONNECT).
11020      */
11021     ssh->packet_dispatch[SSH2_MSG_GLOBAL_REQUEST] = ssh2_msg_unexpected;
11022     ssh->packet_dispatch[SSH2_MSG_REQUEST_SUCCESS] = ssh2_msg_unexpected;
11023     ssh->packet_dispatch[SSH2_MSG_REQUEST_FAILURE] = ssh2_msg_unexpected;
11024     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN] = ssh2_msg_unexpected;
11025     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_CONFIRMATION] = ssh2_msg_unexpected;
11026     ssh->packet_dispatch[SSH2_MSG_CHANNEL_OPEN_FAILURE] = ssh2_msg_unexpected;
11027     ssh->packet_dispatch[SSH2_MSG_CHANNEL_WINDOW_ADJUST] = ssh2_msg_unexpected;
11028     ssh->packet_dispatch[SSH2_MSG_CHANNEL_DATA] = ssh2_msg_unexpected;
11029     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EXTENDED_DATA] = ssh2_msg_unexpected;
11030     ssh->packet_dispatch[SSH2_MSG_CHANNEL_EOF] = ssh2_msg_unexpected;
11031     ssh->packet_dispatch[SSH2_MSG_CHANNEL_CLOSE] = ssh2_msg_unexpected;
11032     ssh->packet_dispatch[SSH2_MSG_CHANNEL_REQUEST] = ssh2_msg_unexpected;
11033     ssh->packet_dispatch[SSH2_MSG_CHANNEL_SUCCESS] = ssh2_msg_unexpected;
11034     ssh->packet_dispatch[SSH2_MSG_CHANNEL_FAILURE] = ssh2_msg_unexpected;
11035
11036     ssh->packet_dispatch[SSH2_MSG_UNIMPLEMENTED] = ssh2_msg_unexpected;
11037
11038     /*
11039      * These messages have a special handler from the start.
11040      */
11041     ssh->packet_dispatch[SSH2_MSG_DISCONNECT] = ssh2_msg_disconnect;
11042     ssh->packet_dispatch[SSH2_MSG_IGNORE] = ssh_msg_ignore;
11043     ssh->packet_dispatch[SSH2_MSG_DEBUG] = ssh2_msg_debug;
11044 }
11045
11046 static void ssh2_timer(void *ctx, unsigned long now)
11047 {
11048     Ssh ssh = (Ssh)ctx;
11049
11050     if (ssh->state == SSH_STATE_CLOSED)
11051         return;
11052
11053     if (!ssh->kex_in_progress && !ssh->bare_connection &&
11054         conf_get_int(ssh->conf, CONF_ssh_rekey_time) != 0 &&
11055         now == ssh->next_rekey) {
11056         do_ssh2_transport(ssh, "timeout", -1, NULL);
11057     }
11058 }
11059
11060 static void ssh2_protocol(Ssh ssh, const void *vin, int inlen,
11061                           struct Packet *pktin)
11062 {
11063     const unsigned char *in = (const unsigned char *)vin;
11064     if (ssh->state == SSH_STATE_CLOSED)
11065         return;
11066
11067     if (pktin) {
11068         ssh->incoming_data_size += pktin->encrypted_len;
11069         if (!ssh->kex_in_progress &&
11070             ssh->max_data_size != 0 &&
11071             ssh->incoming_data_size > ssh->max_data_size)
11072             do_ssh2_transport(ssh, "too much data received", -1, NULL);
11073     }
11074
11075     if (pktin)
11076         ssh->packet_dispatch[pktin->type](ssh, pktin);
11077     else if (!ssh->protocol_initial_phase_done)
11078         do_ssh2_transport(ssh, in, inlen, pktin);
11079     else
11080         do_ssh2_authconn(ssh, in, inlen, pktin);
11081 }
11082
11083 static void ssh2_bare_connection_protocol(Ssh ssh, const void *vin, int inlen,
11084                                           struct Packet *pktin)
11085 {
11086     const unsigned char *in = (const unsigned char *)vin;
11087     if (ssh->state == SSH_STATE_CLOSED)
11088         return;
11089
11090     if (pktin)
11091         ssh->packet_dispatch[pktin->type](ssh, pktin);
11092     else
11093         do_ssh2_authconn(ssh, in, inlen, pktin);
11094 }
11095
11096 static void ssh_cache_conf_values(Ssh ssh)
11097 {
11098     ssh->logomitdata = conf_get_int(ssh->conf, CONF_logomitdata);
11099 }
11100
11101 /*
11102  * Called to set up the connection.
11103  *
11104  * Returns an error message, or NULL on success.
11105  */
11106 static const char *ssh_init(void *frontend_handle, void **backend_handle,
11107                             Conf *conf,
11108                             const char *host, int port, char **realhost,
11109                             int nodelay, int keepalive)
11110 {
11111     const char *p;
11112     Ssh ssh;
11113
11114     ssh = snew(struct ssh_tag);
11115     ssh->conf = conf_copy(conf);
11116     ssh_cache_conf_values(ssh);
11117     ssh->version = 0;                  /* when not ready yet */
11118     ssh->s = NULL;
11119     ssh->cipher = NULL;
11120     ssh->v1_cipher_ctx = NULL;
11121     ssh->crcda_ctx = NULL;
11122     ssh->cscipher = NULL;
11123     ssh->cs_cipher_ctx = NULL;
11124     ssh->sccipher = NULL;
11125     ssh->sc_cipher_ctx = NULL;
11126     ssh->csmac = NULL;
11127     ssh->cs_mac_ctx = NULL;
11128     ssh->scmac = NULL;
11129     ssh->sc_mac_ctx = NULL;
11130     ssh->cscomp = NULL;
11131     ssh->cs_comp_ctx = NULL;
11132     ssh->sccomp = NULL;
11133     ssh->sc_comp_ctx = NULL;
11134     ssh->kex = NULL;
11135     ssh->kex_ctx = NULL;
11136     ssh->hostkey = NULL;
11137     ssh->hostkey_str = NULL;
11138     ssh->exitcode = -1;
11139     ssh->close_expected = FALSE;
11140     ssh->clean_exit = FALSE;
11141     ssh->state = SSH_STATE_PREPACKET;
11142     ssh->size_needed = FALSE;
11143     ssh->eof_needed = FALSE;
11144     ssh->ldisc = NULL;
11145     ssh->logctx = NULL;
11146     ssh->deferred_send_data = NULL;
11147     ssh->deferred_len = 0;
11148     ssh->deferred_size = 0;
11149     ssh->fallback_cmd = 0;
11150     ssh->pkt_kctx = SSH2_PKTCTX_NOKEX;
11151     ssh->pkt_actx = SSH2_PKTCTX_NOAUTH;
11152     ssh->x11disp = NULL;
11153     ssh->x11auth = NULL;
11154     ssh->x11authtree = newtree234(x11_authcmp);
11155     ssh->v1_compressing = FALSE;
11156     ssh->v2_outgoing_sequence = 0;
11157     ssh->ssh1_rdpkt_crstate = 0;
11158     ssh->ssh2_rdpkt_crstate = 0;
11159     ssh->ssh2_bare_rdpkt_crstate = 0;
11160     ssh->ssh_gotdata_crstate = 0;
11161     ssh->do_ssh1_connection_crstate = 0;
11162     ssh->do_ssh_init_state = NULL;
11163     ssh->do_ssh_connection_init_state = NULL;
11164     ssh->do_ssh1_login_state = NULL;
11165     ssh->do_ssh2_transport_state = NULL;
11166     ssh->do_ssh2_authconn_state = NULL;
11167     ssh->v_c = NULL;
11168     ssh->v_s = NULL;
11169     ssh->mainchan = NULL;
11170     ssh->throttled_all = 0;
11171     ssh->v1_stdout_throttling = 0;
11172     ssh->queue = NULL;
11173     ssh->queuelen = ssh->queuesize = 0;
11174     ssh->queueing = FALSE;
11175     ssh->qhead = ssh->qtail = NULL;
11176     ssh->deferred_rekey_reason = NULL;
11177     bufchain_init(&ssh->queued_incoming_data);
11178     ssh->frozen = FALSE;
11179     ssh->username = NULL;
11180     ssh->sent_console_eof = FALSE;
11181     ssh->got_pty = FALSE;
11182     ssh->bare_connection = FALSE;
11183     ssh->X11_fwd_enabled = FALSE;
11184     ssh->connshare = NULL;
11185     ssh->attempting_connshare = FALSE;
11186     ssh->session_started = FALSE;
11187     ssh->specials = NULL;
11188     ssh->n_uncert_hostkeys = 0;
11189     ssh->cross_certifying = FALSE;
11190
11191     *backend_handle = ssh;
11192
11193 #ifdef MSCRYPTOAPI
11194     if (crypto_startup() == 0)
11195         return "Microsoft high encryption pack not installed!";
11196 #endif
11197
11198     ssh->frontend = frontend_handle;
11199     ssh->term_width = conf_get_int(ssh->conf, CONF_width);
11200     ssh->term_height = conf_get_int(ssh->conf, CONF_height);
11201
11202     ssh->channels = NULL;
11203     ssh->rportfwds = NULL;
11204     ssh->portfwds = NULL;
11205
11206     ssh->send_ok = 0;
11207     ssh->editing = 0;
11208     ssh->echoing = 0;
11209     ssh->conn_throttle_count = 0;
11210     ssh->overall_bufsize = 0;
11211     ssh->fallback_cmd = 0;
11212
11213     ssh->protocol = NULL;
11214
11215     ssh->protocol_initial_phase_done = FALSE;
11216
11217     ssh->pinger = NULL;
11218
11219     ssh->incoming_data_size = ssh->outgoing_data_size =
11220         ssh->deferred_data_size = 0L;
11221     ssh->max_data_size = parse_blocksize(conf_get_str(ssh->conf,
11222                                                       CONF_ssh_rekey_data));
11223     ssh->kex_in_progress = FALSE;
11224
11225 #ifndef NO_GSSAPI
11226     ssh->gsslibs = NULL;
11227 #endif
11228
11229     random_ref(); /* do this now - may be needed by sharing setup code */
11230
11231     p = connect_to_host(ssh, host, port, realhost, nodelay, keepalive);
11232     if (p != NULL) {
11233         random_unref();
11234         return p;
11235     }
11236
11237     return NULL;
11238 }
11239
11240 static void ssh_free(void *handle)
11241 {
11242     Ssh ssh = (Ssh) handle;
11243     struct ssh_channel *c;
11244     struct ssh_rportfwd *pf;
11245     struct X11FakeAuth *auth;
11246
11247     if (ssh->v1_cipher_ctx)
11248         ssh->cipher->free_context(ssh->v1_cipher_ctx);
11249     if (ssh->cs_cipher_ctx)
11250         ssh->cscipher->free_context(ssh->cs_cipher_ctx);
11251     if (ssh->sc_cipher_ctx)
11252         ssh->sccipher->free_context(ssh->sc_cipher_ctx);
11253     if (ssh->cs_mac_ctx)
11254         ssh->csmac->free_context(ssh->cs_mac_ctx);
11255     if (ssh->sc_mac_ctx)
11256         ssh->scmac->free_context(ssh->sc_mac_ctx);
11257     if (ssh->cs_comp_ctx) {
11258         if (ssh->cscomp)
11259             ssh->cscomp->compress_cleanup(ssh->cs_comp_ctx);
11260         else
11261             zlib_compress_cleanup(ssh->cs_comp_ctx);
11262     }
11263     if (ssh->sc_comp_ctx) {
11264         if (ssh->sccomp)
11265             ssh->sccomp->decompress_cleanup(ssh->sc_comp_ctx);
11266         else
11267             zlib_decompress_cleanup(ssh->sc_comp_ctx);
11268     }
11269     if (ssh->kex_ctx)
11270         dh_cleanup(ssh->kex_ctx);
11271     sfree(ssh->savedhost);
11272
11273     while (ssh->queuelen-- > 0)
11274         ssh_free_packet(ssh->queue[ssh->queuelen]);
11275     sfree(ssh->queue);
11276
11277     while (ssh->qhead) {
11278         struct queued_handler *qh = ssh->qhead;
11279         ssh->qhead = qh->next;
11280         sfree(qh);
11281     }
11282     ssh->qhead = ssh->qtail = NULL;
11283
11284     if (ssh->channels) {
11285         while ((c = delpos234(ssh->channels, 0)) != NULL) {
11286             switch (c->type) {
11287               case CHAN_X11:
11288                 if (c->u.x11.xconn != NULL)
11289                     x11_close(c->u.x11.xconn);
11290                 break;
11291               case CHAN_SOCKDATA:
11292               case CHAN_SOCKDATA_DORMANT:
11293                 if (c->u.pfd.pf != NULL)
11294                     pfd_close(c->u.pfd.pf);
11295                 break;
11296             }
11297             if (ssh->version == 2) {
11298                 struct outstanding_channel_request *ocr, *nocr;
11299                 ocr = c->v.v2.chanreq_head;
11300                 while (ocr) {
11301                     ocr->handler(c, NULL, ocr->ctx);
11302                     nocr = ocr->next;
11303                     sfree(ocr);
11304                     ocr = nocr;
11305                 }
11306                 bufchain_clear(&c->v.v2.outbuffer);
11307             }
11308             sfree(c);
11309         }
11310         freetree234(ssh->channels);
11311         ssh->channels = NULL;
11312     }
11313
11314     if (ssh->connshare)
11315         sharestate_free(ssh->connshare);
11316
11317     if (ssh->rportfwds) {
11318         while ((pf = delpos234(ssh->rportfwds, 0)) != NULL)
11319             free_rportfwd(pf);
11320         freetree234(ssh->rportfwds);
11321         ssh->rportfwds = NULL;
11322     }
11323     sfree(ssh->deferred_send_data);
11324     if (ssh->x11disp)
11325         x11_free_display(ssh->x11disp);
11326     while ((auth = delpos234(ssh->x11authtree, 0)) != NULL)
11327         x11_free_fake_auth(auth);
11328     freetree234(ssh->x11authtree);
11329     sfree(ssh->do_ssh_init_state);
11330     sfree(ssh->do_ssh1_login_state);
11331     sfree(ssh->do_ssh2_transport_state);
11332     sfree(ssh->do_ssh2_authconn_state);
11333     sfree(ssh->v_c);
11334     sfree(ssh->v_s);
11335     sfree(ssh->fullhostname);
11336     sfree(ssh->hostkey_str);
11337     sfree(ssh->specials);
11338     if (ssh->crcda_ctx) {
11339         crcda_free_context(ssh->crcda_ctx);
11340         ssh->crcda_ctx = NULL;
11341     }
11342     if (ssh->s)
11343         ssh_do_close(ssh, TRUE);
11344     expire_timer_context(ssh);
11345     if (ssh->pinger)
11346         pinger_free(ssh->pinger);
11347     bufchain_clear(&ssh->queued_incoming_data);
11348     sfree(ssh->username);
11349     conf_free(ssh->conf);
11350 #ifndef NO_GSSAPI
11351     if (ssh->gsslibs)
11352         ssh_gss_cleanup(ssh->gsslibs);
11353 #endif
11354     sfree(ssh);
11355
11356     random_unref();
11357 }
11358
11359 /*
11360  * Reconfigure the SSH backend.
11361  */
11362 static void ssh_reconfig(void *handle, Conf *conf)
11363 {
11364     Ssh ssh = (Ssh) handle;
11365     const char *rekeying = NULL;
11366     int rekey_mandatory = FALSE;
11367     unsigned long old_max_data_size;
11368     int i, rekey_time;
11369
11370     pinger_reconfig(ssh->pinger, ssh->conf, conf);
11371     if (ssh->portfwds)
11372         ssh_setup_portfwd(ssh, conf);
11373
11374     rekey_time = conf_get_int(conf, CONF_ssh_rekey_time);
11375     if (conf_get_int(ssh->conf, CONF_ssh_rekey_time) != rekey_time &&
11376         rekey_time != 0) {
11377         unsigned long new_next = ssh->last_rekey + rekey_time*60*TICKSPERSEC;
11378         unsigned long now = GETTICKCOUNT();
11379
11380         if (now - ssh->last_rekey > rekey_time*60*TICKSPERSEC) {
11381             rekeying = "timeout shortened";
11382         } else {
11383             ssh->next_rekey = schedule_timer(new_next - now, ssh2_timer, ssh);
11384         }
11385     }
11386
11387     old_max_data_size = ssh->max_data_size;
11388     ssh->max_data_size = parse_blocksize(conf_get_str(ssh->conf,
11389                                                       CONF_ssh_rekey_data));
11390     if (old_max_data_size != ssh->max_data_size &&
11391         ssh->max_data_size != 0) {
11392         if (ssh->outgoing_data_size > ssh->max_data_size ||
11393             ssh->incoming_data_size > ssh->max_data_size)
11394             rekeying = "data limit lowered";
11395     }
11396
11397     if (conf_get_int(ssh->conf, CONF_compression) !=
11398         conf_get_int(conf, CONF_compression)) {
11399         rekeying = "compression setting changed";
11400         rekey_mandatory = TRUE;
11401     }
11402
11403     for (i = 0; i < CIPHER_MAX; i++)
11404         if (conf_get_int_int(ssh->conf, CONF_ssh_cipherlist, i) !=
11405             conf_get_int_int(conf, CONF_ssh_cipherlist, i)) {
11406         rekeying = "cipher settings changed";
11407         rekey_mandatory = TRUE;
11408     }
11409     if (conf_get_int(ssh->conf, CONF_ssh2_des_cbc) !=
11410         conf_get_int(conf, CONF_ssh2_des_cbc)) {
11411         rekeying = "cipher settings changed";
11412         rekey_mandatory = TRUE;
11413     }
11414
11415     conf_free(ssh->conf);
11416     ssh->conf = conf_copy(conf);
11417     ssh_cache_conf_values(ssh);
11418
11419     if (!ssh->bare_connection && rekeying) {
11420         if (!ssh->kex_in_progress) {
11421             do_ssh2_transport(ssh, rekeying, -1, NULL);
11422         } else if (rekey_mandatory) {
11423             ssh->deferred_rekey_reason = rekeying;
11424         }
11425     }
11426 }
11427
11428 /*
11429  * Called to send data down the SSH connection.
11430  */
11431 static int ssh_send(void *handle, const char *buf, int len)
11432 {
11433     Ssh ssh = (Ssh) handle;
11434
11435     if (ssh == NULL || ssh->s == NULL || ssh->protocol == NULL)
11436         return 0;
11437
11438     ssh->protocol(ssh, (const unsigned char *)buf, len, 0);
11439
11440     return ssh_sendbuffer(ssh);
11441 }
11442
11443 /*
11444  * Called to query the current amount of buffered stdin data.
11445  */
11446 static int ssh_sendbuffer(void *handle)
11447 {
11448     Ssh ssh = (Ssh) handle;
11449     int override_value;
11450
11451     if (ssh == NULL || ssh->s == NULL || ssh->protocol == NULL)
11452         return 0;
11453
11454     /*
11455      * If the SSH socket itself has backed up, add the total backup
11456      * size on that to any individual buffer on the stdin channel.
11457      */
11458     override_value = 0;
11459     if (ssh->throttled_all)
11460         override_value = ssh->overall_bufsize;
11461
11462     if (ssh->version == 1) {
11463         return override_value;
11464     } else if (ssh->version == 2) {
11465         if (!ssh->mainchan)
11466             return override_value;
11467         else
11468             return (override_value +
11469                     bufchain_size(&ssh->mainchan->v.v2.outbuffer));
11470     }
11471
11472     return 0;
11473 }
11474
11475 /*
11476  * Called to set the size of the window from SSH's POV.
11477  */
11478 static void ssh_size(void *handle, int width, int height)
11479 {
11480     Ssh ssh = (Ssh) handle;
11481     struct Packet *pktout;
11482
11483     ssh->term_width = width;
11484     ssh->term_height = height;
11485
11486     switch (ssh->state) {
11487       case SSH_STATE_BEFORE_SIZE:
11488       case SSH_STATE_PREPACKET:
11489       case SSH_STATE_CLOSED:
11490         break;                         /* do nothing */
11491       case SSH_STATE_INTERMED:
11492         ssh->size_needed = TRUE;       /* buffer for later */
11493         break;
11494       case SSH_STATE_SESSION:
11495         if (!conf_get_int(ssh->conf, CONF_nopty)) {
11496             if (ssh->version == 1) {
11497                 send_packet(ssh, SSH1_CMSG_WINDOW_SIZE,
11498                             PKT_INT, ssh->term_height,
11499                             PKT_INT, ssh->term_width,
11500                             PKT_INT, 0, PKT_INT, 0, PKT_END);
11501             } else if (ssh->mainchan) {
11502                 pktout = ssh2_chanreq_init(ssh->mainchan, "window-change",
11503                                            NULL, NULL);
11504                 ssh2_pkt_adduint32(pktout, ssh->term_width);
11505                 ssh2_pkt_adduint32(pktout, ssh->term_height);
11506                 ssh2_pkt_adduint32(pktout, 0);
11507                 ssh2_pkt_adduint32(pktout, 0);
11508                 ssh2_pkt_send(ssh, pktout);
11509             }
11510         }
11511         break;
11512     }
11513 }
11514
11515 /*
11516  * Return a list of the special codes that make sense in this
11517  * protocol.
11518  */
11519 static const struct telnet_special *ssh_get_specials(void *handle)
11520 {
11521     static const struct telnet_special ssh1_ignore_special[] = {
11522         {"IGNORE message", TS_NOP}
11523     };
11524     static const struct telnet_special ssh2_ignore_special[] = {
11525         {"IGNORE message", TS_NOP},
11526     };
11527     static const struct telnet_special ssh2_rekey_special[] = {
11528         {"Repeat key exchange", TS_REKEY},
11529     };
11530     static const struct telnet_special ssh2_session_specials[] = {
11531         {NULL, TS_SEP},
11532         {"Break", TS_BRK},
11533         /* These are the signal names defined by RFC 4254.
11534          * They include all the ISO C signals, but are a subset of the POSIX
11535          * required signals. */
11536         {"SIGINT (Interrupt)", TS_SIGINT},
11537         {"SIGTERM (Terminate)", TS_SIGTERM},
11538         {"SIGKILL (Kill)", TS_SIGKILL},
11539         {"SIGQUIT (Quit)", TS_SIGQUIT},
11540         {"SIGHUP (Hangup)", TS_SIGHUP},
11541         {"More signals", TS_SUBMENU},
11542           {"SIGABRT", TS_SIGABRT}, {"SIGALRM", TS_SIGALRM},
11543           {"SIGFPE",  TS_SIGFPE},  {"SIGILL",  TS_SIGILL},
11544           {"SIGPIPE", TS_SIGPIPE}, {"SIGSEGV", TS_SIGSEGV},
11545           {"SIGUSR1", TS_SIGUSR1}, {"SIGUSR2", TS_SIGUSR2},
11546         {NULL, TS_EXITMENU}
11547     };
11548     static const struct telnet_special specials_end[] = {
11549         {NULL, TS_EXITMENU}
11550     };
11551
11552     struct telnet_special *specials = NULL;
11553     int nspecials = 0, specialsize = 0;
11554
11555     Ssh ssh = (Ssh) handle;
11556
11557     sfree(ssh->specials);
11558
11559 #define ADD_SPECIALS(name) do                                           \
11560     {                                                                   \
11561         int len = lenof(name);                                          \
11562         if (nspecials + len > specialsize) {                            \
11563             specialsize = (nspecials + len) * 5 / 4 + 32;               \
11564             specials = sresize(specials, specialsize, struct telnet_special); \
11565         }                                                               \
11566         memcpy(specials+nspecials, name, len*sizeof(struct telnet_special)); \
11567         nspecials += len;                                               \
11568     } while (0)
11569
11570     if (ssh->version == 1) {
11571         /* Don't bother offering IGNORE if we've decided the remote
11572          * won't cope with it, since we wouldn't bother sending it if
11573          * asked anyway. */
11574         if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH1_IGNORE))
11575             ADD_SPECIALS(ssh1_ignore_special);
11576     } else if (ssh->version == 2) {
11577         if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH2_IGNORE))
11578             ADD_SPECIALS(ssh2_ignore_special);
11579         if (!(ssh->remote_bugs & BUG_SSH2_REKEY) && !ssh->bare_connection)
11580             ADD_SPECIALS(ssh2_rekey_special);
11581         if (ssh->mainchan)
11582             ADD_SPECIALS(ssh2_session_specials);
11583
11584         if (ssh->n_uncert_hostkeys) {
11585             static const struct telnet_special uncert_start[] = {
11586                 {NULL, TS_SEP},
11587                 {"Cache new host key type", TS_SUBMENU},
11588             };
11589             static const struct telnet_special uncert_end[] = {
11590                 {NULL, TS_EXITMENU},
11591             };
11592             int i;
11593
11594             ADD_SPECIALS(uncert_start);
11595             for (i = 0; i < ssh->n_uncert_hostkeys; i++) {
11596                 struct telnet_special uncert[1];
11597                 const struct ssh_signkey *alg =
11598                     hostkey_algs[ssh->uncert_hostkeys[i]].alg;
11599                 uncert[0].name = alg->name;
11600                 uncert[0].code = TS_LOCALSTART + ssh->uncert_hostkeys[i];
11601                 ADD_SPECIALS(uncert);
11602             }
11603             ADD_SPECIALS(uncert_end);
11604         }
11605     } /* else we're not ready yet */
11606
11607     if (nspecials)
11608         ADD_SPECIALS(specials_end);
11609
11610     ssh->specials = specials;
11611
11612     if (nspecials) {
11613         return specials;
11614     } else {
11615         return NULL;
11616     }
11617 #undef ADD_SPECIALS
11618 }
11619
11620 /*
11621  * Send special codes. TS_EOF is useful for `plink', so you
11622  * can send an EOF and collect resulting output (e.g. `plink
11623  * hostname sort').
11624  */
11625 static void ssh_special(void *handle, Telnet_Special code)
11626 {
11627     Ssh ssh = (Ssh) handle;
11628     struct Packet *pktout;
11629
11630     if (code == TS_EOF) {
11631         if (ssh->state != SSH_STATE_SESSION) {
11632             /*
11633              * Buffer the EOF in case we are pre-SESSION, so we can
11634              * send it as soon as we reach SESSION.
11635              */
11636             if (code == TS_EOF)
11637                 ssh->eof_needed = TRUE;
11638             return;
11639         }
11640         if (ssh->version == 1) {
11641             send_packet(ssh, SSH1_CMSG_EOF, PKT_END);
11642         } else if (ssh->mainchan) {
11643             sshfwd_write_eof(ssh->mainchan);
11644             ssh->send_ok = 0;          /* now stop trying to read from stdin */
11645         }
11646         logevent("Sent EOF message");
11647     } else if (code == TS_PING || code == TS_NOP) {
11648         if (ssh->state == SSH_STATE_CLOSED
11649             || ssh->state == SSH_STATE_PREPACKET) return;
11650         if (ssh->version == 1) {
11651             if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH1_IGNORE))
11652                 send_packet(ssh, SSH1_MSG_IGNORE, PKT_STR, "", PKT_END);
11653         } else {
11654             if (!(ssh->remote_bugs & BUG_CHOKES_ON_SSH2_IGNORE)) {
11655                 pktout = ssh2_pkt_init(SSH2_MSG_IGNORE);
11656                 ssh2_pkt_addstring_start(pktout);
11657                 ssh2_pkt_send_noqueue(ssh, pktout);
11658             }
11659         }
11660     } else if (code == TS_REKEY) {
11661         if (!ssh->kex_in_progress && !ssh->bare_connection &&
11662             ssh->version == 2) {
11663             do_ssh2_transport(ssh, "at user request", -1, NULL);
11664         }
11665     } else if (code >= TS_LOCALSTART) {
11666         ssh->hostkey = hostkey_algs[code - TS_LOCALSTART].alg;
11667         ssh->cross_certifying = TRUE;
11668         if (!ssh->kex_in_progress && !ssh->bare_connection &&
11669             ssh->version == 2) {
11670             do_ssh2_transport(ssh, "cross-certifying new host key", -1, NULL);
11671         }
11672     } else if (code == TS_BRK) {
11673         if (ssh->state == SSH_STATE_CLOSED
11674             || ssh->state == SSH_STATE_PREPACKET) return;
11675         if (ssh->version == 1) {
11676             logevent("Unable to send BREAK signal in SSH-1");
11677         } else if (ssh->mainchan) {
11678             pktout = ssh2_chanreq_init(ssh->mainchan, "break", NULL, NULL);
11679             ssh2_pkt_adduint32(pktout, 0);   /* default break length */
11680             ssh2_pkt_send(ssh, pktout);
11681         }
11682     } else {
11683         /* Is is a POSIX signal? */
11684         const char *signame = NULL;
11685         if (code == TS_SIGABRT) signame = "ABRT";
11686         if (code == TS_SIGALRM) signame = "ALRM";
11687         if (code == TS_SIGFPE)  signame = "FPE";
11688         if (code == TS_SIGHUP)  signame = "HUP";
11689         if (code == TS_SIGILL)  signame = "ILL";
11690         if (code == TS_SIGINT)  signame = "INT";
11691         if (code == TS_SIGKILL) signame = "KILL";
11692         if (code == TS_SIGPIPE) signame = "PIPE";
11693         if (code == TS_SIGQUIT) signame = "QUIT";
11694         if (code == TS_SIGSEGV) signame = "SEGV";
11695         if (code == TS_SIGTERM) signame = "TERM";
11696         if (code == TS_SIGUSR1) signame = "USR1";
11697         if (code == TS_SIGUSR2) signame = "USR2";
11698         /* The SSH-2 protocol does in principle support arbitrary named
11699          * signals, including signame@domain, but we don't support those. */
11700         if (signame) {
11701             /* It's a signal. */
11702             if (ssh->version == 2 && ssh->mainchan) {
11703                 pktout = ssh2_chanreq_init(ssh->mainchan, "signal", NULL, NULL);
11704                 ssh2_pkt_addstring(pktout, signame);
11705                 ssh2_pkt_send(ssh, pktout);
11706                 logeventf(ssh, "Sent signal SIG%s", signame);
11707             }
11708         } else {
11709             /* Never heard of it. Do nothing */
11710         }
11711     }
11712 }
11713
11714 void *new_sock_channel(void *handle, struct PortForwarding *pf)
11715 {
11716     Ssh ssh = (Ssh) handle;
11717     struct ssh_channel *c;
11718     c = snew(struct ssh_channel);
11719
11720     c->ssh = ssh;
11721     ssh_channel_init(c);
11722     c->halfopen = TRUE;
11723     c->type = CHAN_SOCKDATA_DORMANT;/* identify channel type */
11724     c->u.pfd.pf = pf;
11725     return c;
11726 }
11727
11728 unsigned ssh_alloc_sharing_channel(Ssh ssh, void *sharing_ctx)
11729 {
11730     struct ssh_channel *c;
11731     c = snew(struct ssh_channel);
11732
11733     c->ssh = ssh;
11734     ssh_channel_init(c);
11735     c->type = CHAN_SHARING;
11736     c->u.sharing.ctx = sharing_ctx;
11737     return c->localid;
11738 }
11739
11740 void ssh_delete_sharing_channel(Ssh ssh, unsigned localid)
11741 {
11742     struct ssh_channel *c;
11743
11744     c = find234(ssh->channels, &localid, ssh_channelfind);
11745     if (c)
11746         ssh_channel_destroy(c);
11747 }
11748
11749 void ssh_send_packet_from_downstream(Ssh ssh, unsigned id, int type,
11750                                      const void *data, int datalen,
11751                                      const char *additional_log_text)
11752 {
11753     struct Packet *pkt;
11754
11755     pkt = ssh2_pkt_init(type);
11756     pkt->downstream_id = id;
11757     pkt->additional_log_text = additional_log_text;
11758     ssh2_pkt_adddata(pkt, data, datalen);
11759     ssh2_pkt_send(ssh, pkt);
11760 }
11761
11762 /*
11763  * This is called when stdout/stderr (the entity to which
11764  * from_backend sends data) manages to clear some backlog.
11765  */
11766 static void ssh_unthrottle(void *handle, int bufsize)
11767 {
11768     Ssh ssh = (Ssh) handle;
11769
11770     if (ssh->version == 1) {
11771         if (ssh->v1_stdout_throttling && bufsize < SSH1_BUFFER_LIMIT) {
11772             ssh->v1_stdout_throttling = 0;
11773             ssh_throttle_conn(ssh, -1);
11774         }
11775     } else {
11776         if (ssh->mainchan)
11777             ssh_channel_unthrottle(ssh->mainchan, bufsize);
11778     }
11779
11780     /*
11781      * Now process any SSH connection data that was stashed in our
11782      * queue while we were frozen.
11783      */
11784     ssh_process_queued_incoming_data(ssh);
11785 }
11786
11787 void ssh_send_port_open(void *channel, const char *hostname, int port,
11788                         const char *org)
11789 {
11790     struct ssh_channel *c = (struct ssh_channel *)channel;
11791     Ssh ssh = c->ssh;
11792     struct Packet *pktout;
11793
11794     logeventf(ssh, "Opening connection to %s:%d for %s", hostname, port, org);
11795
11796     if (ssh->version == 1) {
11797         send_packet(ssh, SSH1_MSG_PORT_OPEN,
11798                     PKT_INT, c->localid,
11799                     PKT_STR, hostname,
11800                     PKT_INT, port,
11801                     /* PKT_STR, <org:orgport>, */
11802                     PKT_END);
11803     } else {
11804         pktout = ssh2_chanopen_init(c, "direct-tcpip");
11805         {
11806             char *trimmed_host = host_strduptrim(hostname);
11807             ssh2_pkt_addstring(pktout, trimmed_host);
11808             sfree(trimmed_host);
11809         }
11810         ssh2_pkt_adduint32(pktout, port);
11811         /*
11812          * We make up values for the originator data; partly it's
11813          * too much hassle to keep track, and partly I'm not
11814          * convinced the server should be told details like that
11815          * about my local network configuration.
11816          * The "originator IP address" is syntactically a numeric
11817          * IP address, and some servers (e.g., Tectia) get upset
11818          * if it doesn't match this syntax.
11819          */
11820         ssh2_pkt_addstring(pktout, "0.0.0.0");
11821         ssh2_pkt_adduint32(pktout, 0);
11822         ssh2_pkt_send(ssh, pktout);
11823     }
11824 }
11825
11826 static int ssh_connected(void *handle)
11827 {
11828     Ssh ssh = (Ssh) handle;
11829     return ssh->s != NULL;
11830 }
11831
11832 static int ssh_sendok(void *handle)
11833 {
11834     Ssh ssh = (Ssh) handle;
11835     return ssh->send_ok;
11836 }
11837
11838 static int ssh_ldisc(void *handle, int option)
11839 {
11840     Ssh ssh = (Ssh) handle;
11841     if (option == LD_ECHO)
11842         return ssh->echoing;
11843     if (option == LD_EDIT)
11844         return ssh->editing;
11845     return FALSE;
11846 }
11847
11848 static void ssh_provide_ldisc(void *handle, void *ldisc)
11849 {
11850     Ssh ssh = (Ssh) handle;
11851     ssh->ldisc = ldisc;
11852 }
11853
11854 static void ssh_provide_logctx(void *handle, void *logctx)
11855 {
11856     Ssh ssh = (Ssh) handle;
11857     ssh->logctx = logctx;
11858 }
11859
11860 static int ssh_return_exitcode(void *handle)
11861 {
11862     Ssh ssh = (Ssh) handle;
11863     if (ssh->s != NULL)
11864         return -1;
11865     else
11866         return (ssh->exitcode >= 0 ? ssh->exitcode : INT_MAX);
11867 }
11868
11869 /*
11870  * cfg_info for SSH is the protocol running in this session.
11871  * (1 or 2 for the full SSH-1 or SSH-2 protocol; -1 for the bare
11872  * SSH-2 connection protocol, i.e. a downstream; 0 for not-decided-yet.)
11873  */
11874 static int ssh_cfg_info(void *handle)
11875 {
11876     Ssh ssh = (Ssh) handle;
11877     if (ssh->version == 0)
11878         return 0; /* don't know yet */
11879     else if (ssh->bare_connection)
11880         return -1;
11881     else
11882         return ssh->version;
11883 }
11884
11885 /*
11886  * Gross hack: pscp will try to start SFTP but fall back to scp1 if
11887  * that fails. This variable is the means by which scp.c can reach
11888  * into the SSH code and find out which one it got.
11889  */
11890 extern int ssh_fallback_cmd(void *handle)
11891 {
11892     Ssh ssh = (Ssh) handle;
11893     return ssh->fallback_cmd;
11894 }
11895
11896 Backend ssh_backend = {
11897     ssh_init,
11898     ssh_free,
11899     ssh_reconfig,
11900     ssh_send,
11901     ssh_sendbuffer,
11902     ssh_size,
11903     ssh_special,
11904     ssh_get_specials,
11905     ssh_connected,
11906     ssh_return_exitcode,
11907     ssh_sendok,
11908     ssh_ldisc,
11909     ssh_provide_ldisc,
11910     ssh_provide_logctx,
11911     ssh_unthrottle,
11912     ssh_cfg_info,
11913     ssh_test_for_upstream,
11914     "ssh",
11915     PROT_SSH,
11916     22
11917 };