]> asedeno.scripts.mit.edu Git - PuTTY.git/blob - ssh.c
Makefile fix: we've been compiling ssh.c separately because it needs
[PuTTY.git] / ssh.c
1 #include <windows.h>
2 #include <stdio.h>
3 #include <stdlib.h>
4 #include <stdarg.h>
5 #include <assert.h>
6
7 #include "putty.h"
8 #include "tree234.h"
9 #include "ssh.h"
10
11 #ifndef FALSE
12 #define FALSE 0
13 #endif
14 #ifndef TRUE
15 #define TRUE 1
16 #endif
17
18 /* uncomment this for packet level debugging */
19 /* #define DUMP_PACKETS */
20
21 #define logevent(s) { logevent(s); \
22                       if ((flags & FLAG_STDERR) && (flags & FLAG_VERBOSE)) \
23                       { fprintf(stderr, "%s\n", s); fflush(stderr); } }
24
25 #define bombout(msg) ( ssh_state = SSH_STATE_CLOSED, \
26                           (s ? sk_close(s), s = NULL : 0), \
27                           connection_fatal msg )
28
29 #define SSH1_MSG_DISCONNECT                       1     /* 0x1 */
30 #define SSH1_SMSG_PUBLIC_KEY                      2     /* 0x2 */
31 #define SSH1_CMSG_SESSION_KEY                     3     /* 0x3 */
32 #define SSH1_CMSG_USER                            4     /* 0x4 */
33 #define SSH1_CMSG_AUTH_RSA                        6     /* 0x6 */
34 #define SSH1_SMSG_AUTH_RSA_CHALLENGE              7     /* 0x7 */
35 #define SSH1_CMSG_AUTH_RSA_RESPONSE               8     /* 0x8 */
36 #define SSH1_CMSG_AUTH_PASSWORD                   9     /* 0x9 */
37 #define SSH1_CMSG_REQUEST_PTY                     10    /* 0xa */
38 #define SSH1_CMSG_WINDOW_SIZE                     11    /* 0xb */
39 #define SSH1_CMSG_EXEC_SHELL                      12    /* 0xc */
40 #define SSH1_CMSG_EXEC_CMD                        13    /* 0xd */
41 #define SSH1_SMSG_SUCCESS                         14    /* 0xe */
42 #define SSH1_SMSG_FAILURE                         15    /* 0xf */
43 #define SSH1_CMSG_STDIN_DATA                      16    /* 0x10 */
44 #define SSH1_SMSG_STDOUT_DATA                     17    /* 0x11 */
45 #define SSH1_SMSG_STDERR_DATA                     18    /* 0x12 */
46 #define SSH1_CMSG_EOF                             19    /* 0x13 */
47 #define SSH1_SMSG_EXIT_STATUS                     20    /* 0x14 */
48 #define SSH1_MSG_CHANNEL_OPEN_CONFIRMATION        21    /* 0x15 */
49 #define SSH1_MSG_CHANNEL_OPEN_FAILURE             22    /* 0x16 */
50 #define SSH1_MSG_CHANNEL_DATA                     23    /* 0x17 */
51 #define SSH1_MSG_CHANNEL_CLOSE                    24    /* 0x18 */
52 #define SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION       25    /* 0x19 */
53 #define SSH1_SMSG_X11_OPEN                        27    /* 0x1b */
54 #define SSH1_CMSG_PORT_FORWARD_REQUEST            28    /* 0x1c */
55 #define SSH1_MSG_PORT_OPEN                        29    /* 0x1d */
56 #define SSH1_CMSG_AGENT_REQUEST_FORWARDING        30    /* 0x1e */
57 #define SSH1_SMSG_AGENT_OPEN                      31    /* 0x1f */
58 #define SSH1_MSG_IGNORE                           32    /* 0x20 */
59 #define SSH1_CMSG_EXIT_CONFIRMATION               33    /* 0x21 */
60 #define SSH1_CMSG_X11_REQUEST_FORWARDING          34    /* 0x22 */
61 #define SSH1_CMSG_AUTH_RHOSTS_RSA                 35    /* 0x23 */
62 #define SSH1_MSG_DEBUG                            36    /* 0x24 */
63 #define SSH1_CMSG_REQUEST_COMPRESSION             37    /* 0x25 */
64 #define SSH1_CMSG_AUTH_TIS                        39    /* 0x27 */
65 #define SSH1_SMSG_AUTH_TIS_CHALLENGE              40    /* 0x28 */
66 #define SSH1_CMSG_AUTH_TIS_RESPONSE               41    /* 0x29 */
67 #define SSH1_CMSG_AUTH_CCARD                      70    /* 0x46 */
68 #define SSH1_SMSG_AUTH_CCARD_CHALLENGE            71    /* 0x47 */
69 #define SSH1_CMSG_AUTH_CCARD_RESPONSE             72    /* 0x48 */
70
71 #define SSH1_AUTH_TIS                             5     /* 0x5 */
72 #define SSH1_AUTH_CCARD                           16    /* 0x10 */
73
74 #define SSH1_PROTOFLAG_SCREEN_NUMBER              1     /* 0x1 */
75 /* Mask for protoflags we will echo back to server if seen */
76 #define SSH1_PROTOFLAGS_SUPPORTED                 0     /* 0x1 */
77
78 #define SSH2_MSG_DISCONNECT                       1     /* 0x1 */
79 #define SSH2_MSG_IGNORE                           2     /* 0x2 */
80 #define SSH2_MSG_UNIMPLEMENTED                    3     /* 0x3 */
81 #define SSH2_MSG_DEBUG                            4     /* 0x4 */
82 #define SSH2_MSG_SERVICE_REQUEST                  5     /* 0x5 */
83 #define SSH2_MSG_SERVICE_ACCEPT                   6     /* 0x6 */
84 #define SSH2_MSG_KEXINIT                          20    /* 0x14 */
85 #define SSH2_MSG_NEWKEYS                          21    /* 0x15 */
86 #define SSH2_MSG_KEXDH_INIT                       30    /* 0x1e */
87 #define SSH2_MSG_KEXDH_REPLY                      31    /* 0x1f */
88 #define SSH2_MSG_KEX_DH_GEX_REQUEST               30    /* 0x1e */
89 #define SSH2_MSG_KEX_DH_GEX_GROUP                 31    /* 0x1f */
90 #define SSH2_MSG_KEX_DH_GEX_INIT                  32    /* 0x20 */
91 #define SSH2_MSG_KEX_DH_GEX_REPLY                 33    /* 0x21 */
92 #define SSH2_MSG_USERAUTH_REQUEST                 50    /* 0x32 */
93 #define SSH2_MSG_USERAUTH_FAILURE                 51    /* 0x33 */
94 #define SSH2_MSG_USERAUTH_SUCCESS                 52    /* 0x34 */
95 #define SSH2_MSG_USERAUTH_BANNER                  53    /* 0x35 */
96 #define SSH2_MSG_USERAUTH_PK_OK                   60    /* 0x3c */
97 #define SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ        60    /* 0x3c */
98 #define SSH2_MSG_USERAUTH_INFO_REQUEST            60    /* 0x3c */
99 #define SSH2_MSG_USERAUTH_INFO_RESPONSE           61    /* 0x3d */
100 #define SSH2_MSG_GLOBAL_REQUEST                   80    /* 0x50 */
101 #define SSH2_MSG_REQUEST_SUCCESS                  81    /* 0x51 */
102 #define SSH2_MSG_REQUEST_FAILURE                  82    /* 0x52 */
103 #define SSH2_MSG_CHANNEL_OPEN                     90    /* 0x5a */
104 #define SSH2_MSG_CHANNEL_OPEN_CONFIRMATION        91    /* 0x5b */
105 #define SSH2_MSG_CHANNEL_OPEN_FAILURE             92    /* 0x5c */
106 #define SSH2_MSG_CHANNEL_WINDOW_ADJUST            93    /* 0x5d */
107 #define SSH2_MSG_CHANNEL_DATA                     94    /* 0x5e */
108 #define SSH2_MSG_CHANNEL_EXTENDED_DATA            95    /* 0x5f */
109 #define SSH2_MSG_CHANNEL_EOF                      96    /* 0x60 */
110 #define SSH2_MSG_CHANNEL_CLOSE                    97    /* 0x61 */
111 #define SSH2_MSG_CHANNEL_REQUEST                  98    /* 0x62 */
112 #define SSH2_MSG_CHANNEL_SUCCESS                  99    /* 0x63 */
113 #define SSH2_MSG_CHANNEL_FAILURE                  100   /* 0x64 */
114
115 #define SSH2_DISCONNECT_HOST_NOT_ALLOWED_TO_CONNECT 1   /* 0x1 */
116 #define SSH2_DISCONNECT_PROTOCOL_ERROR            2     /* 0x2 */
117 #define SSH2_DISCONNECT_KEY_EXCHANGE_FAILED       3     /* 0x3 */
118 #define SSH2_DISCONNECT_HOST_AUTHENTICATION_FAILED 4    /* 0x4 */
119 #define SSH2_DISCONNECT_MAC_ERROR                 5     /* 0x5 */
120 #define SSH2_DISCONNECT_COMPRESSION_ERROR         6     /* 0x6 */
121 #define SSH2_DISCONNECT_SERVICE_NOT_AVAILABLE     7     /* 0x7 */
122 #define SSH2_DISCONNECT_PROTOCOL_VERSION_NOT_SUPPORTED 8        /* 0x8 */
123 #define SSH2_DISCONNECT_HOST_KEY_NOT_VERIFIABLE   9     /* 0x9 */
124 #define SSH2_DISCONNECT_CONNECTION_LOST           10    /* 0xa */
125 #define SSH2_DISCONNECT_BY_APPLICATION            11    /* 0xb */
126 #define SSH2_DISCONNECT_TOO_MANY_CONNECTIONS      12    /* 0xc */
127 #define SSH2_DISCONNECT_AUTH_CANCELLED_BY_USER    13    /* 0xd */
128 #define SSH2_DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE 14       /* 0xe */
129 #define SSH2_DISCONNECT_ILLEGAL_USER_NAME         15    /* 0xf */
130
131 static const char *const ssh2_disconnect_reasons[] = {
132     NULL,
133     "SSH_DISCONNECT_HOST_NOT_ALLOWED_TO_CONNECT",
134     "SSH_DISCONNECT_PROTOCOL_ERROR",
135     "SSH_DISCONNECT_KEY_EXCHANGE_FAILED",
136     "SSH_DISCONNECT_HOST_AUTHENTICATION_FAILED",
137     "SSH_DISCONNECT_MAC_ERROR",
138     "SSH_DISCONNECT_COMPRESSION_ERROR",
139     "SSH_DISCONNECT_SERVICE_NOT_AVAILABLE",
140     "SSH_DISCONNECT_PROTOCOL_VERSION_NOT_SUPPORTED",
141     "SSH_DISCONNECT_HOST_KEY_NOT_VERIFIABLE",
142     "SSH_DISCONNECT_CONNECTION_LOST",
143     "SSH_DISCONNECT_BY_APPLICATION",
144     "SSH_DISCONNECT_TOO_MANY_CONNECTIONS",
145     "SSH_DISCONNECT_AUTH_CANCELLED_BY_USER",
146     "SSH_DISCONNECT_NO_MORE_AUTH_METHODS_AVAILABLE",
147     "SSH_DISCONNECT_ILLEGAL_USER_NAME",
148 };
149
150 #define SSH2_OPEN_ADMINISTRATIVELY_PROHIBITED     1     /* 0x1 */
151 #define SSH2_OPEN_CONNECT_FAILED                  2     /* 0x2 */
152 #define SSH2_OPEN_UNKNOWN_CHANNEL_TYPE            3     /* 0x3 */
153 #define SSH2_OPEN_RESOURCE_SHORTAGE               4     /* 0x4 */
154
155 #define SSH2_EXTENDED_DATA_STDERR                 1     /* 0x1 */
156
157 /*
158  * Various remote-bug flags.
159  */
160 #define BUG_CHOKES_ON_SSH1_IGNORE                 1
161 #define BUG_SSH2_HMAC                             2
162 #define BUG_NEEDS_SSH1_PLAIN_PASSWORD             4
163
164
165 #define GET_32BIT(cp) \
166     (((unsigned long)(unsigned char)(cp)[0] << 24) | \
167     ((unsigned long)(unsigned char)(cp)[1] << 16) | \
168     ((unsigned long)(unsigned char)(cp)[2] << 8) | \
169     ((unsigned long)(unsigned char)(cp)[3]))
170
171 #define PUT_32BIT(cp, value) { \
172     (cp)[0] = (unsigned char)((value) >> 24); \
173     (cp)[1] = (unsigned char)((value) >> 16); \
174     (cp)[2] = (unsigned char)((value) >> 8); \
175     (cp)[3] = (unsigned char)(value); }
176
177 enum { PKT_END, PKT_INT, PKT_CHAR, PKT_DATA, PKT_STR, PKT_BIGNUM };
178
179 /* Coroutine mechanics for the sillier bits of the code */
180 #define crBegin1        static int crLine = 0;
181 #define crBegin2        switch(crLine) { case 0:;
182 #define crBegin         crBegin1; crBegin2;
183 #define crFinish(z)     } crLine = 0; return (z)
184 #define crFinishV       } crLine = 0; return
185 #define crReturn(z)     \
186         do {\
187             crLine=__LINE__; return (z); case __LINE__:;\
188         } while (0)
189 #define crReturnV       \
190         do {\
191             crLine=__LINE__; return; case __LINE__:;\
192         } while (0)
193 #define crStop(z)       do{ crLine = 0; return (z); }while(0)
194 #define crStopV         do{ crLine = 0; return; }while(0)
195 #define crWaitUntil(c)  do { crReturn(0); } while (!(c))
196 #define crWaitUntilV(c) do { crReturnV; } while (!(c))
197
198 extern char *x11_init(Socket *, char *, void *);
199 extern void x11_close(Socket);
200 extern int x11_send(Socket, char *, int);
201 extern void x11_invent_auth(char *, int, char *, int);
202 extern void x11_unthrottle(Socket s);
203 extern void x11_override_throttle(Socket s, int enable);
204
205 extern char *pfd_newconnect(Socket * s, char *hostname, int port, void *c);
206 extern char *pfd_addforward(char *desthost, int destport, int port);
207 extern void pfd_close(Socket s);
208 extern int pfd_send(Socket s, char *data, int len);
209 extern void pfd_confirm(Socket s);
210 extern void pfd_unthrottle(Socket s);
211 extern void pfd_override_throttle(Socket s, int enable);
212
213 /*
214  * Buffer management constants. There are several of these for
215  * various different purposes:
216  * 
217  *  - SSH1_BUFFER_LIMIT is the amount of backlog that must build up
218  *    on a local data stream before we throttle the whole SSH
219  *    connection (in SSH1 only). Throttling the whole connection is
220  *    pretty drastic so we set this high in the hope it won't
221  *    happen very often.
222  * 
223  *  - SSH_MAX_BACKLOG is the amount of backlog that must build up
224  *    on the SSH connection itself before we defensively throttle
225  *    _all_ local data streams. This is pretty drastic too (though
226  *    thankfully unlikely in SSH2 since the window mechanism should
227  *    ensure that the server never has any need to throttle its end
228  *    of the connection), so we set this high as well.
229  * 
230  *  - OUR_V2_WINSIZE is the maximum window size we present on SSH2
231  *    channels.
232  */
233
234 #define SSH1_BUFFER_LIMIT 32768
235 #define SSH_MAX_BACKLOG 32768
236 #define OUR_V2_WINSIZE 16384
237
238 /*
239  * Ciphers for SSH2. We miss out single-DES because it isn't
240  * supported; also 3DES and Blowfish are both done differently from
241  * SSH1. (3DES uses outer chaining; Blowfish has the opposite
242  * endianness and different-sized keys.)
243  */
244 const static struct ssh2_ciphers *ciphers[] = {
245     &ssh2_aes,
246     &ssh2_blowfish,
247     &ssh2_3des,
248 };
249
250 const static struct ssh_kex *kex_algs[] = {
251     &ssh_diffiehellman_gex,
252     &ssh_diffiehellman
253 };
254
255 const static struct ssh_signkey *hostkey_algs[] = { &ssh_rsa, &ssh_dss };
256
257 static void nullmac_key(unsigned char *key)
258 {
259 }
260 static void nullmac_generate(unsigned char *blk, int len,
261                              unsigned long seq)
262 {
263 }
264 static int nullmac_verify(unsigned char *blk, int len, unsigned long seq)
265 {
266     return 1;
267 }
268 const static struct ssh_mac ssh_mac_none = {
269     nullmac_key, nullmac_key, nullmac_generate, nullmac_verify, "none", 0
270 };
271 const static struct ssh_mac *macs[] = {
272     &ssh_sha1, &ssh_md5, &ssh_mac_none
273 };
274 const static struct ssh_mac *buggymacs[] = {
275     &ssh_sha1_buggy, &ssh_md5, &ssh_mac_none
276 };
277
278 static void ssh_comp_none_init(void)
279 {
280 }
281 static int ssh_comp_none_block(unsigned char *block, int len,
282                                unsigned char **outblock, int *outlen)
283 {
284     return 0;
285 }
286 static int ssh_comp_none_disable(void)
287 {
288     return 0;
289 }
290 const static struct ssh_compress ssh_comp_none = {
291     "none",
292     ssh_comp_none_init, ssh_comp_none_block,
293     ssh_comp_none_init, ssh_comp_none_block,
294     ssh_comp_none_disable
295 };
296 extern const struct ssh_compress ssh_zlib;
297 const static struct ssh_compress *compressions[] = {
298     &ssh_zlib, &ssh_comp_none
299 };
300
301 enum {                                 /* channel types */
302     CHAN_MAINSESSION,
303     CHAN_X11,
304     CHAN_AGENT,
305     CHAN_SOCKDATA,
306     CHAN_SOCKDATA_DORMANT              /* one the remote hasn't confirmed */
307 };
308
309 /*
310  * 2-3-4 tree storing channels.
311  */
312 struct ssh_channel {
313     unsigned remoteid, localid;
314     int type;
315     int closes;
316     union {
317         struct ssh1_data_channel {
318             int throttling;
319         } v1;
320         struct ssh2_data_channel {
321             bufchain outbuffer;
322             unsigned remwindow, remmaxpkt;
323             unsigned locwindow;
324         } v2;
325     } v;
326     union {
327         struct ssh_agent_channel {
328             unsigned char *message;
329             unsigned char msglen[4];
330             int lensofar, totallen;
331         } a;
332         struct ssh_x11_channel {
333             Socket s;
334         } x11;
335         struct ssh_pfd_channel {
336             Socket s;
337         } pfd;
338     } u;
339 };
340
341 /*
342  * 2-3-4 tree storing remote->local port forwardings. SSH 1 and SSH
343  * 2 use this structure in different ways, reflecting SSH 2's
344  * altogether saner approach to port forwarding.
345  * 
346  * In SSH 1, you arrange a remote forwarding by sending the server
347  * the remote port number, and the local destination host:port.
348  * When a connection comes in, the server sends you back that
349  * host:port pair, and you connect to it. This is a ready-made
350  * security hole if you're not on the ball: a malicious server
351  * could send you back _any_ host:port pair, so if you trustingly
352  * connect to the address it gives you then you've just opened the
353  * entire inside of your corporate network just by connecting
354  * through it to a dodgy SSH server. Hence, we must store a list of
355  * host:port pairs we _are_ trying to forward to, and reject a
356  * connection request from the server if it's not in the list.
357  * 
358  * In SSH 2, each side of the connection minds its own business and
359  * doesn't send unnecessary information to the other. You arrange a
360  * remote forwarding by sending the server just the remote port
361  * number. When a connection comes in, the server tells you which
362  * of its ports was connected to; and _you_ have to remember what
363  * local host:port pair went with that port number.
364  * 
365  * Hence: in SSH 1 this structure stores host:port pairs we intend
366  * to allow connections to, and is indexed by those host:port
367  * pairs. In SSH 2 it stores a mapping from source port to
368  * destination host:port pair, and is indexed by source port.
369  */
370 struct ssh_rportfwd {
371     unsigned sport, dport;
372     char dhost[256];
373 };
374
375 struct Packet {
376     long length;
377     int type;
378     unsigned char *data;
379     unsigned char *body;
380     long savedpos;
381     long maxlen;
382 };
383
384 static SHA_State exhash, exhashbase;
385
386 static Socket s = NULL;
387
388 static unsigned char session_key[32];
389 static int ssh1_compressing;
390 static int ssh1_remote_protoflags;
391 static int ssh1_local_protoflags;
392 static int ssh_agentfwd_enabled;
393 static int ssh_X11_fwd_enabled;
394 static int ssh_remote_bugs;
395 static const struct ssh_cipher *cipher = NULL;
396 static const struct ssh2_cipher *cscipher = NULL;
397 static const struct ssh2_cipher *sccipher = NULL;
398 static const struct ssh_mac *csmac = NULL;
399 static const struct ssh_mac *scmac = NULL;
400 static const struct ssh_compress *cscomp = NULL;
401 static const struct ssh_compress *sccomp = NULL;
402 static const struct ssh_kex *kex = NULL;
403 static const struct ssh_signkey *hostkey = NULL;
404 static unsigned char ssh2_session_id[20];
405 int (*ssh_get_line) (const char *prompt, char *str, int maxlen,
406                      int is_pw) = NULL;
407
408 static char *savedhost;
409 static int savedport;
410 static int ssh_send_ok;
411 static int ssh_echoing, ssh_editing;
412
413 static tree234 *ssh_channels;          /* indexed by local id */
414 static struct ssh_channel *mainchan;   /* primary session channel */
415
416 static tree234 *ssh_rportfwds;
417
418 static enum {
419     SSH_STATE_PREPACKET,
420     SSH_STATE_BEFORE_SIZE,
421     SSH_STATE_INTERMED,
422     SSH_STATE_SESSION,
423     SSH_STATE_CLOSED
424 } ssh_state = SSH_STATE_PREPACKET;
425
426 static int size_needed = FALSE, eof_needed = FALSE;
427
428 static struct Packet pktin = { 0, 0, NULL, NULL, 0 };
429 static struct Packet pktout = { 0, 0, NULL, NULL, 0 };
430 static unsigned char *deferred_send_data = NULL;
431 static int deferred_len = 0, deferred_size = 0;
432
433 /*
434  * Gross hack: pscp will try to start SFTP but fall back to scp1 if
435  * that fails. This variable is the means by which scp.c can reach
436  * into the SSH code and find out which one it got.
437  */
438 int ssh_fallback_cmd = 0;
439
440 static int ssh_version;
441 static int ssh1_throttle_count;
442 static int ssh_overall_bufsize;
443 static int ssh_throttled_all;
444 static int ssh1_stdout_throttling;
445 static void (*ssh_protocol) (unsigned char *in, int inlen, int ispkt);
446 static void ssh1_protocol(unsigned char *in, int inlen, int ispkt);
447 static void ssh2_protocol(unsigned char *in, int inlen, int ispkt);
448 static void ssh_size(void);
449 static void ssh_special(Telnet_Special);
450 static int ssh2_try_send(struct ssh_channel *c);
451 static void ssh2_add_channel_data(struct ssh_channel *c, char *buf,
452                                   int len);
453 static void ssh_throttle_all(int enable, int bufsize);
454 static void ssh2_set_window(struct ssh_channel *c, unsigned newwin);
455 static int (*s_rdpkt) (unsigned char **data, int *datalen);
456 static int ssh_sendbuffer(void);
457
458 static struct rdpkt1_state_tag {
459     long len, pad, biglen, to_read;
460     unsigned long realcrc, gotcrc;
461     unsigned char *p;
462     int i;
463     int chunk;
464 } rdpkt1_state;
465
466 static struct rdpkt2_state_tag {
467     long len, pad, payload, packetlen, maclen;
468     int i;
469     int cipherblk;
470     unsigned long incoming_sequence;
471 } rdpkt2_state;
472
473 static int ssh_channelcmp(void *av, void *bv)
474 {
475     struct ssh_channel *a = (struct ssh_channel *) av;
476     struct ssh_channel *b = (struct ssh_channel *) bv;
477     if (a->localid < b->localid)
478         return -1;
479     if (a->localid > b->localid)
480         return +1;
481     return 0;
482 }
483 static int ssh_channelfind(void *av, void *bv)
484 {
485     unsigned *a = (unsigned *) av;
486     struct ssh_channel *b = (struct ssh_channel *) bv;
487     if (*a < b->localid)
488         return -1;
489     if (*a > b->localid)
490         return +1;
491     return 0;
492 }
493
494 static int ssh_rportcmp_ssh1(void *av, void *bv)
495 {
496     struct ssh_rportfwd *a = (struct ssh_rportfwd *) av;
497     struct ssh_rportfwd *b = (struct ssh_rportfwd *) bv;
498     int i;
499     if ( (i = strcmp(a->dhost, b->dhost)) != 0)
500         return i < 0 ? -1 : +1;
501     if (a->dport > b->dport)
502         return +1;
503     if (a->dport < b->dport)
504         return -1;
505     return 0;
506 }
507
508 static int ssh_rportcmp_ssh2(void *av, void *bv)
509 {
510     struct ssh_rportfwd *a = (struct ssh_rportfwd *) av;
511     struct ssh_rportfwd *b = (struct ssh_rportfwd *) bv;
512
513     if (a->sport > b->sport)
514         return +1;
515     if (a->sport < b->sport)
516         return -1;
517     return 0;
518 }
519
520 static int alloc_channel_id(void)
521 {
522     const unsigned CHANNEL_NUMBER_OFFSET = 256;
523     unsigned low, high, mid;
524     int tsize;
525     struct ssh_channel *c;
526
527     /*
528      * First-fit allocation of channel numbers: always pick the
529      * lowest unused one. To do this, binary-search using the
530      * counted B-tree to find the largest channel ID which is in a
531      * contiguous sequence from the beginning. (Precisely
532      * everything in that sequence must have ID equal to its tree
533      * index plus CHANNEL_NUMBER_OFFSET.)
534      */
535     tsize = count234(ssh_channels);
536
537     low = -1;
538     high = tsize;
539     while (high - low > 1) {
540         mid = (high + low) / 2;
541         c = index234(ssh_channels, mid);
542         if (c->localid == mid + CHANNEL_NUMBER_OFFSET)
543             low = mid;                 /* this one is fine */
544         else
545             high = mid;                /* this one is past it */
546     }
547     /*
548      * Now low points to either -1, or the tree index of the
549      * largest ID in the initial sequence.
550      */
551     {
552         unsigned i = low + 1 + CHANNEL_NUMBER_OFFSET;
553         assert(NULL == find234(ssh_channels, &i, ssh_channelfind));
554     }
555     return low + 1 + CHANNEL_NUMBER_OFFSET;
556 }
557
558 static void c_write(char *buf, int len)
559 {
560     if ((flags & FLAG_STDERR)) {
561         int i;
562         for (i = 0; i < len; i++)
563             if (buf[i] != '\r')
564                 fputc(buf[i], stderr);
565         return;
566     }
567     from_backend(1, buf, len);
568 }
569
570 static void c_write_untrusted(char *buf, int len)
571 {
572     int i;
573     for (i = 0; i < len; i++) {
574         if (buf[i] == '\n')
575             c_write("\r\n", 2);
576         else if ((buf[i] & 0x60) || (buf[i] == '\r'))
577             c_write(buf + i, 1);
578     }
579 }
580
581 static void c_write_str(char *buf)
582 {
583     c_write(buf, strlen(buf));
584 }
585
586 /*
587  * Collect incoming data in the incoming packet buffer.
588  * Decipher and verify the packet when it is completely read.
589  * Drop SSH1_MSG_DEBUG and SSH1_MSG_IGNORE packets.
590  * Update the *data and *datalen variables.
591  * Return the additional nr of bytes needed, or 0 when
592  * a complete packet is available.
593  */
594 static int ssh1_rdpkt(unsigned char **data, int *datalen)
595 {
596     struct rdpkt1_state_tag *st = &rdpkt1_state;
597
598     crBegin;
599
600   next_packet:
601
602     pktin.type = 0;
603     pktin.length = 0;
604
605     for (st->i = st->len = 0; st->i < 4; st->i++) {
606         while ((*datalen) == 0)
607             crReturn(4 - st->i);
608         st->len = (st->len << 8) + **data;
609         (*data)++, (*datalen)--;
610     }
611
612     st->pad = 8 - (st->len % 8);
613     st->biglen = st->len + st->pad;
614     pktin.length = st->len - 5;
615
616     if (pktin.maxlen < st->biglen) {
617         pktin.maxlen = st->biglen;
618         pktin.data = (pktin.data == NULL ? smalloc(st->biglen + APIEXTRA) :
619                       srealloc(pktin.data, st->biglen + APIEXTRA));
620         if (!pktin.data)
621             fatalbox("Out of memory");
622     }
623
624     st->to_read = st->biglen;
625     st->p = pktin.data;
626     while (st->to_read > 0) {
627         st->chunk = st->to_read;
628         while ((*datalen) == 0)
629             crReturn(st->to_read);
630         if (st->chunk > (*datalen))
631             st->chunk = (*datalen);
632         memcpy(st->p, *data, st->chunk);
633         *data += st->chunk;
634         *datalen -= st->chunk;
635         st->p += st->chunk;
636         st->to_read -= st->chunk;
637     }
638
639     if (cipher)
640         cipher->decrypt(pktin.data, st->biglen);
641 #ifdef DUMP_PACKETS
642     debug(("Got packet len=%d pad=%d\n", st->len, st->pad));
643     dmemdump(pktin.data, st->biglen);
644 #endif
645
646     st->realcrc = crc32(pktin.data, st->biglen - 4);
647     st->gotcrc = GET_32BIT(pktin.data + st->biglen - 4);
648     if (st->gotcrc != st->realcrc) {
649         bombout(("Incorrect CRC received on packet"));
650         crReturn(0);
651     }
652
653     pktin.body = pktin.data + st->pad + 1;
654
655     if (ssh1_compressing) {
656         unsigned char *decompblk;
657         int decomplen;
658 #ifdef DUMP_PACKETS
659         debug(("Packet payload pre-decompression:\n"));
660         dmemdump(pktin.body - 1, pktin.length + 1);
661 #endif
662         zlib_decompress_block(pktin.body - 1, pktin.length + 1,
663                               &decompblk, &decomplen);
664
665         if (pktin.maxlen < st->pad + decomplen) {
666             pktin.maxlen = st->pad + decomplen;
667             pktin.data = srealloc(pktin.data, pktin.maxlen + APIEXTRA);
668             pktin.body = pktin.data + st->pad + 1;
669             if (!pktin.data)
670                 fatalbox("Out of memory");
671         }
672
673         memcpy(pktin.body - 1, decompblk, decomplen);
674         sfree(decompblk);
675         pktin.length = decomplen - 1;
676 #ifdef DUMP_PACKETS
677         debug(("Packet payload post-decompression:\n"));
678         dmemdump(pktin.body - 1, pktin.length + 1);
679 #endif
680     }
681
682     if (pktin.type == SSH1_SMSG_STDOUT_DATA ||
683         pktin.type == SSH1_SMSG_STDERR_DATA ||
684         pktin.type == SSH1_MSG_DEBUG ||
685         pktin.type == SSH1_SMSG_AUTH_TIS_CHALLENGE ||
686         pktin.type == SSH1_SMSG_AUTH_CCARD_CHALLENGE) {
687         long strlen = GET_32BIT(pktin.body);
688         if (strlen + 4 != pktin.length) {
689             bombout(("Received data packet with bogus string length"));
690             crReturn(0);
691         }
692     }
693
694     pktin.type = pktin.body[-1];
695
696     if (pktin.type == SSH1_MSG_DEBUG) {
697         /* log debug message */
698         char buf[80];
699         int strlen = GET_32BIT(pktin.body);
700         strcpy(buf, "Remote: ");
701         if (strlen > 70)
702             strlen = 70;
703         memcpy(buf + 8, pktin.body + 4, strlen);
704         buf[8 + strlen] = '\0';
705         logevent(buf);
706         goto next_packet;
707     } else if (pktin.type == SSH1_MSG_IGNORE) {
708         /* do nothing */
709         goto next_packet;
710     }
711
712     if (pktin.type == SSH1_MSG_DISCONNECT) {
713         /* log reason code in disconnect message */
714         char buf[256];
715         unsigned msglen = GET_32BIT(pktin.body);
716         unsigned nowlen;
717         strcpy(buf, "Remote sent disconnect: ");
718         nowlen = strlen(buf);
719         if (msglen > sizeof(buf) - nowlen - 1)
720             msglen = sizeof(buf) - nowlen - 1;
721         memcpy(buf + nowlen, pktin.body + 4, msglen);
722         buf[nowlen + msglen] = '\0';
723         logevent(buf);
724         bombout(("Server sent disconnect message:\n\"%s\"", buf+nowlen));
725         crReturn(0);
726     }
727
728     crFinish(0);
729 }
730
731 static int ssh2_rdpkt(unsigned char **data, int *datalen)
732 {
733     struct rdpkt2_state_tag *st = &rdpkt2_state;
734
735     crBegin;
736
737   next_packet:
738     pktin.type = 0;
739     pktin.length = 0;
740     if (sccipher)
741         st->cipherblk = sccipher->blksize;
742     else
743         st->cipherblk = 8;
744     if (st->cipherblk < 8)
745         st->cipherblk = 8;
746
747     if (pktin.maxlen < st->cipherblk) {
748         pktin.maxlen = st->cipherblk;
749         pktin.data =
750             (pktin.data ==
751              NULL ? smalloc(st->cipherblk +
752                             APIEXTRA) : srealloc(pktin.data,
753                                                  st->cipherblk +
754                                                  APIEXTRA));
755         if (!pktin.data)
756             fatalbox("Out of memory");
757     }
758
759     /*
760      * Acquire and decrypt the first block of the packet. This will
761      * contain the length and padding details.
762      */
763     for (st->i = st->len = 0; st->i < st->cipherblk; st->i++) {
764         while ((*datalen) == 0)
765             crReturn(st->cipherblk - st->i);
766         pktin.data[st->i] = *(*data)++;
767         (*datalen)--;
768     }
769
770     if (sccipher)
771         sccipher->decrypt(pktin.data, st->cipherblk);
772
773     /*
774      * Now get the length and padding figures.
775      */
776     st->len = GET_32BIT(pktin.data);
777     st->pad = pktin.data[4];
778
779     /*
780      * This enables us to deduce the payload length.
781      */
782     st->payload = st->len - st->pad - 1;
783
784     pktin.length = st->payload + 5;
785
786     /*
787      * So now we can work out the total packet length.
788      */
789     st->packetlen = st->len + 4;
790     st->maclen = scmac ? scmac->len : 0;
791
792     /*
793      * Adjust memory allocation if packet is too big.
794      */
795     if (pktin.maxlen < st->packetlen + st->maclen) {
796         pktin.maxlen = st->packetlen + st->maclen;
797         pktin.data =
798             (pktin.data ==
799              NULL ? smalloc(pktin.maxlen + APIEXTRA) : srealloc(pktin.data,
800                                                                 pktin.maxlen
801                                                                 +
802                                                                 APIEXTRA));
803         if (!pktin.data)
804             fatalbox("Out of memory");
805     }
806
807     /*
808      * Read and decrypt the remainder of the packet.
809      */
810     for (st->i = st->cipherblk; st->i < st->packetlen + st->maclen;
811          st->i++) {
812         while ((*datalen) == 0)
813             crReturn(st->packetlen + st->maclen - st->i);
814         pktin.data[st->i] = *(*data)++;
815         (*datalen)--;
816     }
817     /* Decrypt everything _except_ the MAC. */
818     if (sccipher)
819         sccipher->decrypt(pktin.data + st->cipherblk,
820                           st->packetlen - st->cipherblk);
821
822 #ifdef DUMP_PACKETS
823     debug(("Got packet len=%d pad=%d\n", st->len, st->pad));
824     dmemdump(pktin.data, st->packetlen);
825 #endif
826
827     /*
828      * Check the MAC.
829      */
830     if (scmac
831         && !scmac->verify(pktin.data, st->len + 4,
832                           st->incoming_sequence)) {
833         bombout(("Incorrect MAC received on packet"));
834         crReturn(0);
835     }
836     st->incoming_sequence++;           /* whether or not we MACed */
837
838     /*
839      * Decompress packet payload.
840      */
841     {
842         unsigned char *newpayload;
843         int newlen;
844         if (sccomp && sccomp->decompress(pktin.data + 5, pktin.length - 5,
845                                          &newpayload, &newlen)) {
846             if (pktin.maxlen < newlen + 5) {
847                 pktin.maxlen = newlen + 5;
848                 pktin.data =
849                     (pktin.data ==
850                      NULL ? smalloc(pktin.maxlen +
851                                     APIEXTRA) : srealloc(pktin.data,
852                                                          pktin.maxlen +
853                                                          APIEXTRA));
854                 if (!pktin.data)
855                     fatalbox("Out of memory");
856             }
857             pktin.length = 5 + newlen;
858             memcpy(pktin.data + 5, newpayload, newlen);
859 #ifdef DUMP_PACKETS
860             debug(("Post-decompression payload:\n"));
861             dmemdump(pktin.data + 5, newlen);
862 #endif
863
864             sfree(newpayload);
865         }
866     }
867
868     pktin.savedpos = 6;
869     pktin.type = pktin.data[5];
870
871     if (pktin.type == SSH2_MSG_IGNORE || pktin.type == SSH2_MSG_DEBUG)
872         goto next_packet;              /* FIXME: print DEBUG message */
873
874     if (pktin.type == SSH2_MSG_DISCONNECT) {
875         /* log reason code in disconnect message */
876         char buf[256];
877         int reason = GET_32BIT(pktin.data + 6);
878         unsigned msglen = GET_32BIT(pktin.data + 10);
879         unsigned nowlen;
880         if (reason > 0 && reason < lenof(ssh2_disconnect_reasons)) {
881             sprintf(buf, "Received disconnect message (%s)",
882                     ssh2_disconnect_reasons[reason]);
883         } else {
884             sprintf(buf, "Received disconnect message (unknown type %d)",
885                     reason);
886         }
887         logevent(buf);
888         strcpy(buf, "Disconnection message text: ");
889         nowlen = strlen(buf);
890         if (msglen > sizeof(buf) - nowlen - 1)
891             msglen = sizeof(buf) - nowlen - 1;
892         memcpy(buf + nowlen, pktin.data + 14, msglen);
893         buf[nowlen + msglen] = '\0';
894         logevent(buf);
895         bombout(("Server sent disconnect message\ntype %d (%s):\n\"%s\"",
896                  reason,
897                  (reason > 0 && reason < lenof(ssh2_disconnect_reasons)) ?
898                  ssh2_disconnect_reasons[reason] : "unknown",
899                  buf+nowlen));
900         crReturn(0);
901     }
902
903     crFinish(0);
904 }
905
906 static void ssh1_pktout_size(int len)
907 {
908     int pad, biglen;
909
910     len += 5;                          /* type and CRC */
911     pad = 8 - (len % 8);
912     biglen = len + pad;
913
914     pktout.length = len - 5;
915     if (pktout.maxlen < biglen) {
916         pktout.maxlen = biglen;
917 #ifdef MSCRYPTOAPI
918         /* Allocate enough buffer space for extra block
919          * for MS CryptEncrypt() */
920         pktout.data = (pktout.data == NULL ? smalloc(biglen + 12) :
921                        srealloc(pktout.data, biglen + 12));
922 #else
923         pktout.data = (pktout.data == NULL ? smalloc(biglen + 4) :
924                        srealloc(pktout.data, biglen + 4));
925 #endif
926         if (!pktout.data)
927             fatalbox("Out of memory");
928     }
929     pktout.body = pktout.data + 4 + pad + 1;
930 }
931
932 static void s_wrpkt_start(int type, int len)
933 {
934     ssh1_pktout_size(len);
935     pktout.type = type;
936 }
937
938 static int s_wrpkt_prepare(void)
939 {
940     int pad, len, biglen, i;
941     unsigned long crc;
942
943     pktout.body[-1] = pktout.type;
944
945     if (ssh1_compressing) {
946         unsigned char *compblk;
947         int complen;
948 #ifdef DUMP_PACKETS
949         debug(("Packet payload pre-compression:\n"));
950         dmemdump(pktout.body - 1, pktout.length + 1);
951 #endif
952         zlib_compress_block(pktout.body - 1, pktout.length + 1,
953                             &compblk, &complen);
954         ssh1_pktout_size(complen - 1);
955         memcpy(pktout.body - 1, compblk, complen);
956         sfree(compblk);
957     }
958
959     len = pktout.length + 5;           /* type and CRC */
960     pad = 8 - (len % 8);
961     biglen = len + pad;
962
963     for (i = 0; i < pad; i++)
964         pktout.data[i + 4] = random_byte();
965     crc = crc32(pktout.data + 4, biglen - 4);
966     PUT_32BIT(pktout.data + biglen, crc);
967     PUT_32BIT(pktout.data, len);
968
969 #ifdef DUMP_PACKETS
970     debug(("Sending packet len=%d\n", biglen + 4));
971     dmemdump(pktout.data, biglen + 4);
972 #endif
973     if (cipher)
974         cipher->encrypt(pktout.data + 4, biglen);
975
976     return biglen + 4;
977 }
978
979 static void s_wrpkt(void)
980 {
981     int len, backlog;
982     len = s_wrpkt_prepare();
983     backlog = sk_write(s, pktout.data, len);
984     if (backlog > SSH_MAX_BACKLOG)
985         ssh_throttle_all(1, backlog);
986 }
987
988 static void s_wrpkt_defer(void)
989 {
990     int len;
991     len = s_wrpkt_prepare();
992     if (deferred_len + len > deferred_size) {
993         deferred_size = deferred_len + len + 128;
994         deferred_send_data = srealloc(deferred_send_data, deferred_size);
995     }
996     memcpy(deferred_send_data + deferred_len, pktout.data, len);
997     deferred_len += len;
998 }
999
1000 /*
1001  * Construct a packet with the specified contents.
1002  */
1003 static void construct_packet(int pkttype, va_list ap1, va_list ap2)
1004 {
1005     unsigned char *p, *argp, argchar;
1006     unsigned long argint;
1007     int pktlen, argtype, arglen;
1008     Bignum bn;
1009
1010     pktlen = 0;
1011     while ((argtype = va_arg(ap1, int)) != PKT_END) {
1012         switch (argtype) {
1013           case PKT_INT:
1014             (void) va_arg(ap1, int);
1015             pktlen += 4;
1016             break;
1017           case PKT_CHAR:
1018             (void) va_arg(ap1, char);
1019             pktlen++;
1020             break;
1021           case PKT_DATA:
1022             (void) va_arg(ap1, unsigned char *);
1023             arglen = va_arg(ap1, int);
1024             pktlen += arglen;
1025             break;
1026           case PKT_STR:
1027             argp = va_arg(ap1, unsigned char *);
1028             arglen = strlen(argp);
1029             pktlen += 4 + arglen;
1030             break;
1031           case PKT_BIGNUM:
1032             bn = va_arg(ap1, Bignum);
1033             pktlen += ssh1_bignum_length(bn);
1034             break;
1035           default:
1036             assert(0);
1037         }
1038     }
1039
1040     s_wrpkt_start(pkttype, pktlen);
1041     p = pktout.body;
1042
1043     while ((argtype = va_arg(ap2, int)) != PKT_END) {
1044         switch (argtype) {
1045           case PKT_INT:
1046             argint = va_arg(ap2, int);
1047             PUT_32BIT(p, argint);
1048             p += 4;
1049             break;
1050           case PKT_CHAR:
1051             argchar = va_arg(ap2, unsigned char);
1052             *p = argchar;
1053             p++;
1054             break;
1055           case PKT_DATA:
1056             argp = va_arg(ap2, unsigned char *);
1057             arglen = va_arg(ap2, int);
1058             memcpy(p, argp, arglen);
1059             p += arglen;
1060             break;
1061           case PKT_STR:
1062             argp = va_arg(ap2, unsigned char *);
1063             arglen = strlen(argp);
1064             PUT_32BIT(p, arglen);
1065             memcpy(p + 4, argp, arglen);
1066             p += 4 + arglen;
1067             break;
1068           case PKT_BIGNUM:
1069             bn = va_arg(ap2, Bignum);
1070             p += ssh1_write_bignum(p, bn);
1071             break;
1072         }
1073     }
1074 }
1075
1076 static void send_packet(int pkttype, ...)
1077 {
1078     va_list ap1, ap2;
1079     va_start(ap1, pkttype);
1080     va_start(ap2, pkttype);
1081     construct_packet(pkttype, ap1, ap2);
1082     s_wrpkt();
1083 }
1084
1085 static void defer_packet(int pkttype, ...)
1086 {
1087     va_list ap1, ap2;
1088     va_start(ap1, pkttype);
1089     va_start(ap2, pkttype);
1090     construct_packet(pkttype, ap1, ap2);
1091     s_wrpkt_defer();
1092 }
1093
1094 static int ssh_versioncmp(char *a, char *b)
1095 {
1096     char *ae, *be;
1097     unsigned long av, bv;
1098
1099     av = strtoul(a, &ae, 10);
1100     bv = strtoul(b, &be, 10);
1101     if (av != bv)
1102         return (av < bv ? -1 : +1);
1103     if (*ae == '.')
1104         ae++;
1105     if (*be == '.')
1106         be++;
1107     av = strtoul(ae, &ae, 10);
1108     bv = strtoul(be, &be, 10);
1109     if (av != bv)
1110         return (av < bv ? -1 : +1);
1111     return 0;
1112 }
1113
1114
1115 /*
1116  * Utility routines for putting an SSH-protocol `string' and
1117  * `uint32' into a SHA state.
1118  */
1119 #include <stdio.h>
1120 static void sha_string(SHA_State * s, void *str, int len)
1121 {
1122     unsigned char lenblk[4];
1123     PUT_32BIT(lenblk, len);
1124     SHA_Bytes(s, lenblk, 4);
1125     SHA_Bytes(s, str, len);
1126 }
1127
1128 static void sha_uint32(SHA_State * s, unsigned i)
1129 {
1130     unsigned char intblk[4];
1131     PUT_32BIT(intblk, i);
1132     SHA_Bytes(s, intblk, 4);
1133 }
1134
1135 /*
1136  * SSH2 packet construction functions.
1137  */
1138 static void ssh2_pkt_ensure(int length)
1139 {
1140     if (pktout.maxlen < length) {
1141         pktout.maxlen = length + 256;
1142         pktout.data =
1143             (pktout.data ==
1144              NULL ? smalloc(pktout.maxlen +
1145                             APIEXTRA) : srealloc(pktout.data,
1146                                                  pktout.maxlen +
1147                                                  APIEXTRA));
1148         if (!pktout.data)
1149             fatalbox("Out of memory");
1150     }
1151 }
1152 static void ssh2_pkt_adddata(void *data, int len)
1153 {
1154     pktout.length += len;
1155     ssh2_pkt_ensure(pktout.length);
1156     memcpy(pktout.data + pktout.length - len, data, len);
1157 }
1158 static void ssh2_pkt_addbyte(unsigned char byte)
1159 {
1160     ssh2_pkt_adddata(&byte, 1);
1161 }
1162 static void ssh2_pkt_init(int pkt_type)
1163 {
1164     pktout.length = 5;
1165     ssh2_pkt_addbyte((unsigned char) pkt_type);
1166 }
1167 static void ssh2_pkt_addbool(unsigned char value)
1168 {
1169     ssh2_pkt_adddata(&value, 1);
1170 }
1171 static void ssh2_pkt_adduint32(unsigned long value)
1172 {
1173     unsigned char x[4];
1174     PUT_32BIT(x, value);
1175     ssh2_pkt_adddata(x, 4);
1176 }
1177 static void ssh2_pkt_addstring_start(void)
1178 {
1179     ssh2_pkt_adduint32(0);
1180     pktout.savedpos = pktout.length;
1181 }
1182 static void ssh2_pkt_addstring_str(char *data)
1183 {
1184     ssh2_pkt_adddata(data, strlen(data));
1185     PUT_32BIT(pktout.data + pktout.savedpos - 4,
1186               pktout.length - pktout.savedpos);
1187 }
1188 static void ssh2_pkt_addstring_data(char *data, int len)
1189 {
1190     ssh2_pkt_adddata(data, len);
1191     PUT_32BIT(pktout.data + pktout.savedpos - 4,
1192               pktout.length - pktout.savedpos);
1193 }
1194 static void ssh2_pkt_addstring(char *data)
1195 {
1196     ssh2_pkt_addstring_start();
1197     ssh2_pkt_addstring_str(data);
1198 }
1199 static char *ssh2_mpint_fmt(Bignum b, int *len)
1200 {
1201     unsigned char *p;
1202     int i, n = (bignum_bitcount(b) + 7) / 8;
1203     p = smalloc(n + 1);
1204     if (!p)
1205         fatalbox("out of memory");
1206     p[0] = 0;
1207     for (i = 1; i <= n; i++)
1208         p[i] = bignum_byte(b, n - i);
1209     i = 0;
1210     while (i <= n && p[i] == 0 && (p[i + 1] & 0x80) == 0)
1211         i++;
1212     memmove(p, p + i, n + 1 - i);
1213     *len = n + 1 - i;
1214     return p;
1215 }
1216 static void ssh2_pkt_addmp(Bignum b)
1217 {
1218     unsigned char *p;
1219     int len;
1220     p = ssh2_mpint_fmt(b, &len);
1221     ssh2_pkt_addstring_start();
1222     ssh2_pkt_addstring_data(p, len);
1223     sfree(p);
1224 }
1225
1226 /*
1227  * Construct an SSH2 final-form packet: compress it, encrypt it,
1228  * put the MAC on it. Final packet, ready to be sent, is stored in
1229  * pktout.data. Total length is returned.
1230  */
1231 static int ssh2_pkt_construct(void)
1232 {
1233     int cipherblk, maclen, padding, i;
1234     static unsigned long outgoing_sequence = 0;
1235
1236     /*
1237      * Compress packet payload.
1238      */
1239     {
1240         unsigned char *newpayload;
1241         int newlen;
1242 #ifdef DUMP_PACKETS
1243         if (cscomp && cscomp != &ssh_comp_none) {
1244             debug(("Pre-compression payload:\n"));
1245             dmemdump(pktout.data + 5, pktout.length - 5);
1246         }
1247 #endif
1248         if (cscomp && cscomp->compress(pktout.data + 5, pktout.length - 5,
1249                                        &newpayload, &newlen)) {
1250             pktout.length = 5;
1251             ssh2_pkt_adddata(newpayload, newlen);
1252             sfree(newpayload);
1253         }
1254     }
1255
1256     /*
1257      * Add padding. At least four bytes, and must also bring total
1258      * length (minus MAC) up to a multiple of the block size.
1259      */
1260     cipherblk = cscipher ? cscipher->blksize : 8;       /* block size */
1261     cipherblk = cipherblk < 8 ? 8 : cipherblk;  /* or 8 if blksize < 8 */
1262     padding = 4;
1263     padding +=
1264         (cipherblk - (pktout.length + padding) % cipherblk) % cipherblk;
1265     maclen = csmac ? csmac->len : 0;
1266     ssh2_pkt_ensure(pktout.length + padding + maclen);
1267     pktout.data[4] = padding;
1268     for (i = 0; i < padding; i++)
1269         pktout.data[pktout.length + i] = random_byte();
1270     PUT_32BIT(pktout.data, pktout.length + padding - 4);
1271     if (csmac)
1272         csmac->generate(pktout.data, pktout.length + padding,
1273                         outgoing_sequence);
1274     outgoing_sequence++;               /* whether or not we MACed */
1275
1276 #ifdef DUMP_PACKETS
1277     debug(("Sending packet len=%d\n", pktout.length + padding));
1278     dmemdump(pktout.data, pktout.length + padding);
1279 #endif
1280
1281     if (cscipher)
1282         cscipher->encrypt(pktout.data, pktout.length + padding);
1283
1284     /* Ready-to-send packet starts at pktout.data. We return length. */
1285     return pktout.length + padding + maclen;
1286 }
1287
1288 /*
1289  * Construct and send an SSH2 packet immediately.
1290  */
1291 static void ssh2_pkt_send(void)
1292 {
1293     int len;
1294     int backlog;
1295     len = ssh2_pkt_construct();
1296     backlog = sk_write(s, pktout.data, len);
1297     if (backlog > SSH_MAX_BACKLOG)
1298         ssh_throttle_all(1, backlog);
1299 }
1300
1301 /*
1302  * Construct an SSH2 packet and add it to a deferred data block.
1303  * Useful for sending multiple packets in a single sk_write() call,
1304  * to prevent a traffic-analysing listener from being able to work
1305  * out the length of any particular packet (such as the password
1306  * packet).
1307  * 
1308  * Note that because SSH2 sequence-numbers its packets, this can
1309  * NOT be used as an m4-style `defer' allowing packets to be
1310  * constructed in one order and sent in another.
1311  */
1312 static void ssh2_pkt_defer(void)
1313 {
1314     int len = ssh2_pkt_construct();
1315     if (deferred_len + len > deferred_size) {
1316         deferred_size = deferred_len + len + 128;
1317         deferred_send_data = srealloc(deferred_send_data, deferred_size);
1318     }
1319     memcpy(deferred_send_data + deferred_len, pktout.data, len);
1320     deferred_len += len;
1321 }
1322
1323 /*
1324  * Send the whole deferred data block constructed by
1325  * ssh2_pkt_defer() or SSH1's defer_packet().
1326  */
1327 static void ssh_pkt_defersend(void)
1328 {
1329     int backlog;
1330     backlog = sk_write(s, deferred_send_data, deferred_len);
1331     deferred_len = deferred_size = 0;
1332     sfree(deferred_send_data);
1333     deferred_send_data = NULL;
1334     if (backlog > SSH_MAX_BACKLOG)
1335         ssh_throttle_all(1, backlog);
1336 }
1337
1338 #if 0
1339 void bndebug(char *string, Bignum b)
1340 {
1341     unsigned char *p;
1342     int i, len;
1343     p = ssh2_mpint_fmt(b, &len);
1344     debug(("%s", string));
1345     for (i = 0; i < len; i++)
1346         debug((" %02x", p[i]));
1347     debug(("\n"));
1348     sfree(p);
1349 }
1350 #endif
1351
1352 static void sha_mpint(SHA_State * s, Bignum b)
1353 {
1354     unsigned char *p;
1355     int len;
1356     p = ssh2_mpint_fmt(b, &len);
1357     sha_string(s, p, len);
1358     sfree(p);
1359 }
1360
1361 /*
1362  * SSH2 packet decode functions.
1363  */
1364 static unsigned long ssh2_pkt_getuint32(void)
1365 {
1366     unsigned long value;
1367     if (pktin.length - pktin.savedpos < 4)
1368         return 0;                      /* arrgh, no way to decline (FIXME?) */
1369     value = GET_32BIT(pktin.data + pktin.savedpos);
1370     pktin.savedpos += 4;
1371     return value;
1372 }
1373 static int ssh2_pkt_getbool(void)
1374 {
1375     unsigned long value;
1376     if (pktin.length - pktin.savedpos < 1)
1377         return 0;                      /* arrgh, no way to decline (FIXME?) */
1378     value = pktin.data[pktin.savedpos] != 0;
1379     pktin.savedpos++;
1380     return value;
1381 }
1382 static void ssh2_pkt_getstring(char **p, int *length)
1383 {
1384     *p = NULL;
1385     if (pktin.length - pktin.savedpos < 4)
1386         return;
1387     *length = GET_32BIT(pktin.data + pktin.savedpos);
1388     pktin.savedpos += 4;
1389     if (pktin.length - pktin.savedpos < *length)
1390         return;
1391     *p = pktin.data + pktin.savedpos;
1392     pktin.savedpos += *length;
1393 }
1394 static Bignum ssh2_pkt_getmp(void)
1395 {
1396     char *p;
1397     int length;
1398     Bignum b;
1399
1400     ssh2_pkt_getstring(&p, &length);
1401     if (!p)
1402         return NULL;
1403     if (p[0] & 0x80) {
1404         bombout(("internal error: Can't handle negative mpints"));
1405         return NULL;
1406     }
1407     b = bignum_from_bytes(p, length);
1408     return b;
1409 }
1410
1411 /*
1412  * Examine the remote side's version string and compare it against
1413  * a list of known buggy implementations.
1414  */
1415 static void ssh_detect_bugs(char *vstring)
1416 {
1417     char *imp;                         /* pointer to implementation part */
1418     imp = vstring;
1419     imp += strcspn(imp, "-");
1420     if (*imp) imp++;
1421     imp += strcspn(imp, "-");
1422     if (*imp) imp++;
1423
1424     ssh_remote_bugs = 0;
1425
1426     if (!strcmp(imp, "1.2.18") || !strcmp(imp, "1.2.19") ||
1427         !strcmp(imp, "1.2.20") || !strcmp(imp, "1.2.21") ||
1428         !strcmp(imp, "1.2.22") || !strcmp(imp, "Cisco-1.25")) {
1429         /*
1430          * These versions don't support SSH1_MSG_IGNORE, so we have
1431          * to use a different defence against password length
1432          * sniffing.
1433          */
1434         ssh_remote_bugs |= BUG_CHOKES_ON_SSH1_IGNORE;
1435         logevent("We believe remote version has SSH1 ignore bug");
1436     }
1437
1438     if (!strcmp(imp, "Cisco-1.25")) {
1439         /*
1440          * These versions need a plain password sent; they can't
1441          * handle having a null and a random length of data after
1442          * the password.
1443          */
1444         ssh_remote_bugs |= BUG_NEEDS_SSH1_PLAIN_PASSWORD;
1445         logevent("We believe remote version needs a plain SSH1 password");
1446     }
1447
1448     if (!strncmp(imp, "2.1.0", 5) || !strncmp(imp, "2.0.", 4) ||
1449         !strncmp(imp, "2.2.0", 5) || !strncmp(imp, "2.3.0", 5) ||
1450         !strncmp(imp, "2.1 ", 4)) {
1451         /*
1452          * These versions have the HMAC bug.
1453          */
1454         ssh_remote_bugs |= BUG_SSH2_HMAC;
1455         logevent("We believe remote version has SSH2 HMAC bug");
1456     }
1457 }
1458
1459 static int do_ssh_init(unsigned char c)
1460 {
1461     static int vslen;
1462     static char version[10];
1463     static char *vstring;
1464     static int vstrsize;
1465     static char *vlog;
1466     static int i;
1467
1468     crBegin;
1469
1470     /* Search for the string "SSH-" in the input. */
1471     i = 0;
1472     while (1) {
1473         static const int transS[] = { 1, 2, 2, 1 };
1474         static const int transH[] = { 0, 0, 3, 0 };
1475         static const int transminus[] = { 0, 0, 0, -1 };
1476         if (c == 'S')
1477             i = transS[i];
1478         else if (c == 'H')
1479             i = transH[i];
1480         else if (c == '-')
1481             i = transminus[i];
1482         else
1483             i = 0;
1484         if (i < 0)
1485             break;
1486         crReturn(1);                   /* get another character */
1487     }
1488
1489     vstrsize = 16;
1490     vstring = smalloc(vstrsize);
1491     strcpy(vstring, "SSH-");
1492     vslen = 4;
1493     i = 0;
1494     while (1) {
1495         crReturn(1);                   /* get another char */
1496         if (vslen >= vstrsize - 1) {
1497             vstrsize += 16;
1498             vstring = srealloc(vstring, vstrsize);
1499         }
1500         vstring[vslen++] = c;
1501         if (i >= 0) {
1502             if (c == '-') {
1503                 version[i] = '\0';
1504                 i = -1;
1505             } else if (i < sizeof(version) - 1)
1506                 version[i++] = c;
1507         } else if (c == '\n')
1508             break;
1509     }
1510
1511     ssh_agentfwd_enabled = FALSE;
1512     rdpkt2_state.incoming_sequence = 0;
1513
1514     vstring[vslen] = 0;
1515     vlog = smalloc(20 + vslen);
1516     vstring[strcspn (vstring, "\r\n")] = '\0'; /* remove end-of-line chars */
1517     sprintf(vlog, "Server version: %s", vstring);
1518     logevent(vlog);
1519     ssh_detect_bugs(vstring);
1520     sfree(vlog);
1521
1522     /*
1523      * Server version "1.99" means we can choose whether we use v1
1524      * or v2 protocol. Choice is based on cfg.sshprot.
1525      */
1526     if (ssh_versioncmp(version, cfg.sshprot == 1 ? "2.0" : "1.99") >= 0) {
1527         /*
1528          * This is a v2 server. Begin v2 protocol.
1529          */
1530         char verstring[80], vlog[100];
1531         sprintf(verstring, "SSH-2.0-%s", sshver);
1532         SHA_Init(&exhashbase);
1533         /*
1534          * Hash our version string and their version string.
1535          */
1536         sha_string(&exhashbase, verstring, strlen(verstring));
1537         sha_string(&exhashbase, vstring, strcspn(vstring, "\r\n"));
1538         sprintf(vlog, "We claim version: %s", verstring);
1539         logevent(vlog);
1540         strcat(verstring, "\n");
1541         logevent("Using SSH protocol version 2");
1542         sk_write(s, verstring, strlen(verstring));
1543         ssh_protocol = ssh2_protocol;
1544         ssh_version = 2;
1545         s_rdpkt = ssh2_rdpkt;
1546     } else {
1547         /*
1548          * This is a v1 server. Begin v1 protocol.
1549          */
1550         char verstring[80], vlog[100];
1551         sprintf(verstring, "SSH-%s-%s",
1552                 (ssh_versioncmp(version, "1.5") <= 0 ? version : "1.5"),
1553                 sshver);
1554         sprintf(vlog, "We claim version: %s", verstring);
1555         logevent(vlog);
1556         strcat(verstring, "\n");
1557         logevent("Using SSH protocol version 1");
1558         sk_write(s, verstring, strlen(verstring));
1559         ssh_protocol = ssh1_protocol;
1560         ssh_version = 1;
1561         s_rdpkt = ssh1_rdpkt;
1562     }
1563     ssh_state = SSH_STATE_BEFORE_SIZE;
1564
1565     sfree(vstring);
1566
1567     crFinish(0);
1568 }
1569
1570 static void ssh_gotdata(unsigned char *data, int datalen)
1571 {
1572     crBegin;
1573
1574     /*
1575      * To begin with, feed the characters one by one to the
1576      * protocol initialisation / selection function do_ssh_init().
1577      * When that returns 0, we're done with the initial greeting
1578      * exchange and can move on to packet discipline.
1579      */
1580     while (1) {
1581         int ret;
1582         if (datalen == 0)
1583             crReturnV;                 /* more data please */
1584         ret = do_ssh_init(*data);
1585         data++;
1586         datalen--;
1587         if (ret == 0)
1588             break;
1589     }
1590
1591     /*
1592      * We emerge from that loop when the initial negotiation is
1593      * over and we have selected an s_rdpkt function. Now pass
1594      * everything to s_rdpkt, and then pass the resulting packets
1595      * to the proper protocol handler.
1596      */
1597     if (datalen == 0)
1598         crReturnV;
1599     while (1) {
1600         while (datalen > 0) {
1601             if (s_rdpkt(&data, &datalen) == 0) {
1602                 if (ssh_state == SSH_STATE_CLOSED) {
1603                     return;
1604                 }
1605                 ssh_protocol(NULL, 0, 1);
1606                 if (ssh_state == SSH_STATE_CLOSED) {
1607                     return;
1608                 }
1609             }
1610         }
1611         crReturnV;
1612     }
1613     crFinishV;
1614 }
1615
1616 static int ssh_closing(Plug plug, char *error_msg, int error_code,
1617                        int calling_back)
1618 {
1619     ssh_state = SSH_STATE_CLOSED;
1620     if (s) {
1621         sk_close(s);
1622         s = NULL;
1623     }
1624     if (error_msg) {
1625         /* A socket error has occurred. */
1626         connection_fatal(error_msg);
1627     } else {
1628         /* Otherwise, the remote side closed the connection normally. */
1629     }
1630     return 0;
1631 }
1632
1633 static int ssh_receive(Plug plug, int urgent, char *data, int len)
1634 {
1635     ssh_gotdata(data, len);
1636     if (ssh_state == SSH_STATE_CLOSED) {
1637         if (s) {
1638             sk_close(s);
1639             s = NULL;
1640         }
1641         return 0;
1642     }
1643     return 1;
1644 }
1645
1646 static void ssh_sent(Plug plug, int bufsize)
1647 {
1648     /*
1649      * If the send backlog on the SSH socket itself clears, we
1650      * should unthrottle the whole world if it was throttled.
1651      */
1652     if (bufsize < SSH_MAX_BACKLOG)
1653         ssh_throttle_all(0, bufsize);
1654 }
1655
1656 /*
1657  * Connect to specified host and port.
1658  * Returns an error message, or NULL on success.
1659  * Also places the canonical host name into `realhost'. It must be
1660  * freed by the caller.
1661  */
1662 static char *connect_to_host(char *host, int port, char **realhost)
1663 {
1664     static struct plug_function_table fn_table = {
1665         ssh_closing,
1666         ssh_receive,
1667         ssh_sent,
1668         NULL
1669     }, *fn_table_ptr = &fn_table;
1670
1671     SockAddr addr;
1672     char *err;
1673 #ifdef FWHACK
1674     char *FWhost;
1675     int FWport;
1676 #endif
1677
1678     savedhost = smalloc(1 + strlen(host));
1679     if (!savedhost)
1680         fatalbox("Out of memory");
1681     strcpy(savedhost, host);
1682
1683     if (port < 0)
1684         port = 22;                     /* default ssh port */
1685     savedport = port;
1686
1687 #ifdef FWHACK
1688     FWhost = host;
1689     FWport = port;
1690     host = FWSTR;
1691     port = 23;
1692 #endif
1693
1694     /*
1695      * Try to find host.
1696      */
1697     {
1698         char buf[200];
1699         sprintf(buf, "Looking up host \"%.170s\"", host);
1700         logevent(buf);
1701     }
1702     addr = sk_namelookup(host, realhost);
1703     if ((err = sk_addr_error(addr)))
1704         return err;
1705
1706 #ifdef FWHACK
1707     *realhost = strdup(FWhost);
1708 #endif
1709
1710     /*
1711      * Open socket.
1712      */
1713     {
1714         char buf[200], addrbuf[100];
1715         sk_getaddr(addr, addrbuf, 100);
1716         sprintf(buf, "Connecting to %.100s port %d", addrbuf, port);
1717         logevent(buf);
1718     }
1719     s = sk_new(addr, port, 0, 1, &fn_table_ptr);
1720     if ((err = sk_socket_error(s)))
1721         return err;
1722
1723 #ifdef FWHACK
1724     sk_write(s, "connect ", 8);
1725     sk_write(s, FWhost, strlen(FWhost));
1726     {
1727         char buf[20];
1728         sprintf(buf, " %d\n", FWport);
1729         sk_write(s, buf, strlen(buf));
1730     }
1731 #endif
1732
1733     return NULL;
1734 }
1735
1736 /*
1737  * Throttle or unthrottle the SSH connection.
1738  */
1739 static void ssh1_throttle(int adjust)
1740 {
1741     int old_count = ssh1_throttle_count;
1742     ssh1_throttle_count += adjust;
1743     assert(ssh1_throttle_count >= 0);
1744     if (ssh1_throttle_count && !old_count) {
1745         sk_set_frozen(s, 1);
1746     } else if (!ssh1_throttle_count && old_count) {
1747         sk_set_frozen(s, 0);
1748     }
1749 }
1750
1751 /*
1752  * Throttle or unthrottle _all_ local data streams (for when sends
1753  * on the SSH connection itself back up).
1754  */
1755 static void ssh_throttle_all(int enable, int bufsize)
1756 {
1757     int i;
1758     struct ssh_channel *c;
1759
1760     if (enable == ssh_throttled_all)
1761         return;
1762     ssh_throttled_all = enable;
1763     ssh_overall_bufsize = bufsize;
1764     if (!ssh_channels)
1765         return;
1766     for (i = 0; NULL != (c = index234(ssh_channels, i)); i++) {
1767         switch (c->type) {
1768           case CHAN_MAINSESSION:
1769             /*
1770              * This is treated separately, outside the switch.
1771              */
1772             break;
1773           case CHAN_X11:
1774             x11_override_throttle(c->u.x11.s, enable);
1775             break;
1776           case CHAN_AGENT:
1777             /* Agent channels require no buffer management. */
1778             break;
1779           case CHAN_SOCKDATA:
1780             pfd_override_throttle(c->u.x11.s, enable);
1781             break;
1782         }
1783     }
1784 }
1785
1786 /*
1787  * Handle the key exchange and user authentication phases.
1788  */
1789 static int do_ssh1_login(unsigned char *in, int inlen, int ispkt)
1790 {
1791     int i, j;
1792     static int len;
1793     static unsigned char *rsabuf, *keystr1, *keystr2;
1794     unsigned char cookie[8];
1795     struct RSAKey servkey, hostkey;
1796     struct MD5Context md5c;
1797     static unsigned long supported_ciphers_mask, supported_auths_mask;
1798     static int tried_publickey;
1799     static int tis_auth_refused, ccard_auth_refused;
1800     static unsigned char session_id[16];
1801     static int cipher_type;
1802     static char username[100];
1803
1804     crBegin;
1805
1806     if (!ispkt)
1807         crWaitUntil(ispkt);
1808
1809     if (pktin.type != SSH1_SMSG_PUBLIC_KEY) {
1810         bombout(("Public key packet not received"));
1811         crReturn(0);
1812     }
1813
1814     logevent("Received public keys");
1815
1816     memcpy(cookie, pktin.body, 8);
1817
1818     i = makekey(pktin.body + 8, &servkey, &keystr1, 0);
1819     j = makekey(pktin.body + 8 + i, &hostkey, &keystr2, 0);
1820
1821     /*
1822      * Log the host key fingerprint.
1823      */
1824     {
1825         char logmsg[80];
1826         logevent("Host key fingerprint is:");
1827         strcpy(logmsg, "      ");
1828         hostkey.comment = NULL;
1829         rsa_fingerprint(logmsg + strlen(logmsg),
1830                         sizeof(logmsg) - strlen(logmsg), &hostkey);
1831         logevent(logmsg);
1832     }
1833
1834     ssh1_remote_protoflags = GET_32BIT(pktin.body + 8 + i + j);
1835     supported_ciphers_mask = GET_32BIT(pktin.body + 12 + i + j);
1836     supported_auths_mask = GET_32BIT(pktin.body + 16 + i + j);
1837
1838     ssh1_local_protoflags =
1839         ssh1_remote_protoflags & SSH1_PROTOFLAGS_SUPPORTED;
1840     ssh1_local_protoflags |= SSH1_PROTOFLAG_SCREEN_NUMBER;
1841
1842     MD5Init(&md5c);
1843     MD5Update(&md5c, keystr2, hostkey.bytes);
1844     MD5Update(&md5c, keystr1, servkey.bytes);
1845     MD5Update(&md5c, pktin.body, 8);
1846     MD5Final(session_id, &md5c);
1847
1848     for (i = 0; i < 32; i++)
1849         session_key[i] = random_byte();
1850
1851     len = (hostkey.bytes > servkey.bytes ? hostkey.bytes : servkey.bytes);
1852
1853     rsabuf = smalloc(len);
1854     if (!rsabuf)
1855         fatalbox("Out of memory");
1856
1857     /*
1858      * Verify the host key.
1859      */
1860     {
1861         /*
1862          * First format the key into a string.
1863          */
1864         int len = rsastr_len(&hostkey);
1865         char fingerprint[100];
1866         char *keystr = smalloc(len);
1867         if (!keystr)
1868             fatalbox("Out of memory");
1869         rsastr_fmt(keystr, &hostkey);
1870         rsa_fingerprint(fingerprint, sizeof(fingerprint), &hostkey);
1871         verify_ssh_host_key(savedhost, savedport, "rsa", keystr,
1872                             fingerprint);
1873         sfree(keystr);
1874     }
1875
1876     for (i = 0; i < 32; i++) {
1877         rsabuf[i] = session_key[i];
1878         if (i < 16)
1879             rsabuf[i] ^= session_id[i];
1880     }
1881
1882     if (hostkey.bytes > servkey.bytes) {
1883         rsaencrypt(rsabuf, 32, &servkey);
1884         rsaencrypt(rsabuf, servkey.bytes, &hostkey);
1885     } else {
1886         rsaencrypt(rsabuf, 32, &hostkey);
1887         rsaencrypt(rsabuf, hostkey.bytes, &servkey);
1888     }
1889
1890     logevent("Encrypted session key");
1891
1892     {
1893         int cipher_chosen = 0, warn = 0;
1894         char *cipher_string = NULL;
1895         for (i = 0; !cipher_chosen && i < CIPHER_MAX; i++) {
1896             int next_cipher = cfg.ssh_cipherlist[i];
1897             if (next_cipher == CIPHER_WARN) {
1898                 /* If/when we choose a cipher, warn about it */
1899                 warn = 1;
1900             } else if (next_cipher == CIPHER_AES) {
1901                 /* XXX Probably don't need to mention this. */
1902                 logevent("AES not supported in SSH1, skipping");
1903             } else {
1904                 switch (next_cipher) {
1905                   case CIPHER_3DES:     cipher_type = SSH_CIPHER_3DES;
1906                                         cipher_string = "3DES"; break;
1907                   case CIPHER_BLOWFISH: cipher_type = SSH_CIPHER_BLOWFISH;
1908                                         cipher_string = "Blowfish"; break;
1909                   case CIPHER_DES:      cipher_type = SSH_CIPHER_DES;
1910                                         cipher_string = "single-DES"; break;
1911                 }
1912                 if (supported_ciphers_mask & (1 << cipher_type))
1913                     cipher_chosen = 1;
1914             }
1915         }
1916         if (!cipher_chosen) {
1917             if ((supported_ciphers_mask & (1 << SSH_CIPHER_3DES)) == 0)
1918                 bombout(("Server violates SSH 1 protocol by not "
1919                          "supporting 3DES encryption"));
1920             else
1921                 /* shouldn't happen */
1922                 bombout(("No supported ciphers found"));
1923             crReturn(0);
1924         }
1925
1926         /* Warn about chosen cipher if necessary. */
1927         if (warn)
1928             askcipher(cipher_string, 0);
1929     }
1930
1931     switch (cipher_type) {
1932       case SSH_CIPHER_3DES:
1933         logevent("Using 3DES encryption");
1934         break;
1935       case SSH_CIPHER_DES:
1936         logevent("Using single-DES encryption");
1937         break;
1938       case SSH_CIPHER_BLOWFISH:
1939         logevent("Using Blowfish encryption");
1940         break;
1941     }
1942
1943     send_packet(SSH1_CMSG_SESSION_KEY,
1944                 PKT_CHAR, cipher_type,
1945                 PKT_DATA, cookie, 8,
1946                 PKT_CHAR, (len * 8) >> 8, PKT_CHAR, (len * 8) & 0xFF,
1947                 PKT_DATA, rsabuf, len,
1948                 PKT_INT, ssh1_local_protoflags, PKT_END);
1949
1950     logevent("Trying to enable encryption...");
1951
1952     sfree(rsabuf);
1953
1954     cipher = cipher_type == SSH_CIPHER_BLOWFISH ? &ssh_blowfish_ssh1 :
1955         cipher_type == SSH_CIPHER_DES ? &ssh_des : &ssh_3des;
1956     cipher->sesskey(session_key);
1957
1958     crWaitUntil(ispkt);
1959
1960     if (pktin.type != SSH1_SMSG_SUCCESS) {
1961         bombout(("Encryption not successfully enabled"));
1962         crReturn(0);
1963     }
1964
1965     logevent("Successfully started encryption");
1966
1967     fflush(stdout);
1968     {
1969         static int pos = 0;
1970         static char c;
1971         if ((flags & FLAG_INTERACTIVE) && !*cfg.username) {
1972             if (ssh_get_line) {
1973                 if (!ssh_get_line("login as: ",
1974                                   username, sizeof(username), FALSE)) {
1975                     /*
1976                      * get_line failed to get a username.
1977                      * Terminate.
1978                      */
1979                     logevent("No username provided. Abandoning session.");
1980                     ssh_state = SSH_STATE_CLOSED;
1981                     crReturn(1);
1982                 }
1983             } else {
1984                 c_write_str("login as: ");
1985                 ssh_send_ok = 1;
1986                 while (pos >= 0) {
1987                     crWaitUntil(!ispkt);
1988                     while (inlen--)
1989                         switch (c = *in++) {
1990                           case 10:
1991                           case 13:
1992                             username[pos] = 0;
1993                             pos = -1;
1994                             break;
1995                           case 8:
1996                           case 127:
1997                             if (pos > 0) {
1998                                 c_write_str("\b \b");
1999                                 pos--;
2000                             }
2001                             break;
2002                           case 21:
2003                           case 27:
2004                             while (pos > 0) {
2005                                 c_write_str("\b \b");
2006                                 pos--;
2007                             }
2008                             break;
2009                           case 3:
2010                           case 4:
2011                             random_save_seed();
2012                             exit(0);
2013                             break;
2014                           default:
2015                             if (((c >= ' ' && c <= '~') ||
2016                                  ((unsigned char) c >= 160))
2017                                 && pos < sizeof(username)-1) {
2018                                 username[pos++] = c;
2019                                 c_write(&c, 1);
2020                             }
2021                             break;
2022                         }
2023                 }
2024                 c_write_str("\r\n");
2025                 username[strcspn(username, "\n\r")] = '\0';
2026             }
2027         } else {
2028             strncpy(username, cfg.username, 99);
2029             username[99] = '\0';
2030         }
2031
2032         send_packet(SSH1_CMSG_USER, PKT_STR, username, PKT_END);
2033         {
2034             char userlog[22 + sizeof(username)];
2035             sprintf(userlog, "Sent username \"%s\"", username);
2036             logevent(userlog);
2037             if (flags & FLAG_INTERACTIVE &&
2038                 (!((flags & FLAG_STDERR) && (flags & FLAG_VERBOSE)))) {
2039                 strcat(userlog, "\r\n");
2040                 c_write_str(userlog);
2041             }
2042         }
2043     }
2044
2045     crWaitUntil(ispkt);
2046
2047     tried_publickey = 0;
2048     tis_auth_refused = ccard_auth_refused = 0;
2049
2050     while (pktin.type == SSH1_SMSG_FAILURE) {
2051         static char password[100];
2052         static char prompt[200];
2053         static int pos;
2054         static char c;
2055         static int pwpkt_type;
2056         pwpkt_type = SSH1_CMSG_AUTH_PASSWORD;
2057
2058         if (agent_exists()) {
2059             /*
2060              * Attempt RSA authentication using Pageant.
2061              */
2062             static unsigned char request[5], *response, *p;
2063             static int responselen;
2064             static int i, nkeys;
2065             static int authed = FALSE;
2066             void *r;
2067
2068             logevent("Pageant is running. Requesting keys.");
2069
2070             /* Request the keys held by the agent. */
2071             PUT_32BIT(request, 1);
2072             request[4] = SSH1_AGENTC_REQUEST_RSA_IDENTITIES;
2073             agent_query(request, 5, &r, &responselen);
2074             response = (unsigned char *) r;
2075             if (response && responselen >= 5 &&
2076                 response[4] == SSH1_AGENT_RSA_IDENTITIES_ANSWER) {
2077                 p = response + 5;
2078                 nkeys = GET_32BIT(p);
2079                 p += 4;
2080                 {
2081                     char buf[64];
2082                     sprintf(buf, "Pageant has %d SSH1 keys", nkeys);
2083                     logevent(buf);
2084                 }
2085                 for (i = 0; i < nkeys; i++) {
2086                     static struct RSAKey key;
2087                     static Bignum challenge;
2088                     static char *commentp;
2089                     static int commentlen;
2090
2091                     {
2092                         char buf[64];
2093                         sprintf(buf, "Trying Pageant key #%d", i);
2094                         logevent(buf);
2095                     }
2096                     p += 4;
2097                     p += ssh1_read_bignum(p, &key.exponent);
2098                     p += ssh1_read_bignum(p, &key.modulus);
2099                     commentlen = GET_32BIT(p);
2100                     p += 4;
2101                     commentp = p;
2102                     p += commentlen;
2103                     send_packet(SSH1_CMSG_AUTH_RSA,
2104                                 PKT_BIGNUM, key.modulus, PKT_END);
2105                     crWaitUntil(ispkt);
2106                     if (pktin.type != SSH1_SMSG_AUTH_RSA_CHALLENGE) {
2107                         logevent("Key refused");
2108                         continue;
2109                     }
2110                     logevent("Received RSA challenge");
2111                     ssh1_read_bignum(pktin.body, &challenge);
2112                     {
2113                         char *agentreq, *q, *ret;
2114                         void *vret;
2115                         int len, retlen;
2116                         len = 1 + 4;   /* message type, bit count */
2117                         len += ssh1_bignum_length(key.exponent);
2118                         len += ssh1_bignum_length(key.modulus);
2119                         len += ssh1_bignum_length(challenge);
2120                         len += 16;     /* session id */
2121                         len += 4;      /* response format */
2122                         agentreq = smalloc(4 + len);
2123                         PUT_32BIT(agentreq, len);
2124                         q = agentreq + 4;
2125                         *q++ = SSH1_AGENTC_RSA_CHALLENGE;
2126                         PUT_32BIT(q, bignum_bitcount(key.modulus));
2127                         q += 4;
2128                         q += ssh1_write_bignum(q, key.exponent);
2129                         q += ssh1_write_bignum(q, key.modulus);
2130                         q += ssh1_write_bignum(q, challenge);
2131                         memcpy(q, session_id, 16);
2132                         q += 16;
2133                         PUT_32BIT(q, 1);        /* response format */
2134                         agent_query(agentreq, len + 4, &vret, &retlen);
2135                         ret = vret;
2136                         sfree(agentreq);
2137                         if (ret) {
2138                             if (ret[4] == SSH1_AGENT_RSA_RESPONSE) {
2139                                 logevent("Sending Pageant's response");
2140                                 send_packet(SSH1_CMSG_AUTH_RSA_RESPONSE,
2141                                             PKT_DATA, ret + 5, 16,
2142                                             PKT_END);
2143                                 sfree(ret);
2144                                 crWaitUntil(ispkt);
2145                                 if (pktin.type == SSH1_SMSG_SUCCESS) {
2146                                     logevent
2147                                         ("Pageant's response accepted");
2148                                     if (flags & FLAG_VERBOSE) {
2149                                         c_write_str
2150                                             ("Authenticated using RSA key \"");
2151                                         c_write(commentp, commentlen);
2152                                         c_write_str("\" from agent\r\n");
2153                                     }
2154                                     authed = TRUE;
2155                                 } else
2156                                     logevent
2157                                         ("Pageant's response not accepted");
2158                             } else {
2159                                 logevent
2160                                     ("Pageant failed to answer challenge");
2161                                 sfree(ret);
2162                             }
2163                         } else {
2164                             logevent("No reply received from Pageant");
2165                         }
2166                     }
2167                     freebn(key.exponent);
2168                     freebn(key.modulus);
2169                     freebn(challenge);
2170                     if (authed)
2171                         break;
2172                 }
2173             }
2174             if (authed)
2175                 break;
2176         }
2177         if (*cfg.keyfile && !tried_publickey)
2178             pwpkt_type = SSH1_CMSG_AUTH_RSA;
2179
2180         if (cfg.try_tis_auth &&
2181             (supported_auths_mask & (1 << SSH1_AUTH_TIS)) &&
2182             !tis_auth_refused) {
2183             pwpkt_type = SSH1_CMSG_AUTH_TIS_RESPONSE;
2184             logevent("Requested TIS authentication");
2185             send_packet(SSH1_CMSG_AUTH_TIS, PKT_END);
2186             crWaitUntil(ispkt);
2187             if (pktin.type != SSH1_SMSG_AUTH_TIS_CHALLENGE) {
2188                 logevent("TIS authentication declined");
2189                 if (flags & FLAG_INTERACTIVE)
2190                     c_write_str("TIS authentication refused.\r\n");
2191                 tis_auth_refused = 1;
2192                 continue;
2193             } else {
2194                 int challengelen = ((pktin.body[0] << 24) |
2195                                     (pktin.body[1] << 16) |
2196                                     (pktin.body[2] << 8) |
2197                                     (pktin.body[3]));
2198                 logevent("Received TIS challenge");
2199                 if (challengelen > sizeof(prompt) - 1)
2200                     challengelen = sizeof(prompt) - 1;  /* prevent overrun */
2201                 memcpy(prompt, pktin.body + 4, challengelen);
2202                 /* Prompt heuristic comes from OpenSSH */
2203                 strncpy(prompt + challengelen,
2204                         memchr(prompt, '\n', challengelen) ?
2205                         "": "\r\nResponse: ",
2206                         (sizeof prompt) - challengelen);
2207                 prompt[(sizeof prompt) - 1] = '\0';
2208             }
2209         }
2210         if (cfg.try_tis_auth &&
2211             (supported_auths_mask & (1 << SSH1_AUTH_CCARD)) &&
2212             !ccard_auth_refused) {
2213             pwpkt_type = SSH1_CMSG_AUTH_CCARD_RESPONSE;
2214             logevent("Requested CryptoCard authentication");
2215             send_packet(SSH1_CMSG_AUTH_CCARD, PKT_END);
2216             crWaitUntil(ispkt);
2217             if (pktin.type != SSH1_SMSG_AUTH_CCARD_CHALLENGE) {
2218                 logevent("CryptoCard authentication declined");
2219                 c_write_str("CryptoCard authentication refused.\r\n");
2220                 ccard_auth_refused = 1;
2221                 continue;
2222             } else {
2223                 int challengelen = ((pktin.body[0] << 24) |
2224                                     (pktin.body[1] << 16) |
2225                                     (pktin.body[2] << 8) |
2226                                     (pktin.body[3]));
2227                 logevent("Received CryptoCard challenge");
2228                 if (challengelen > sizeof(prompt) - 1)
2229                     challengelen = sizeof(prompt) - 1;  /* prevent overrun */
2230                 memcpy(prompt, pktin.body + 4, challengelen);
2231                 strncpy(prompt + challengelen,
2232                         memchr(prompt, '\n', challengelen) ?
2233                         "" : "\r\nResponse: ",
2234                         sizeof(prompt) - challengelen);
2235                 prompt[sizeof(prompt) - 1] = '\0';
2236             }
2237         }
2238         if (pwpkt_type == SSH1_CMSG_AUTH_PASSWORD) {
2239             sprintf(prompt, "%.90s@%.90s's password: ",
2240                     username, savedhost);
2241         }
2242         if (pwpkt_type == SSH1_CMSG_AUTH_RSA) {
2243             char *comment = NULL;
2244             if (flags & FLAG_VERBOSE)
2245                 c_write_str("Trying public key authentication.\r\n");
2246             if (!rsakey_encrypted(cfg.keyfile, &comment)) {
2247                 if (flags & FLAG_VERBOSE)
2248                     c_write_str("No passphrase required.\r\n");
2249                 goto tryauth;
2250             }
2251             sprintf(prompt, "Passphrase for key \"%.100s\": ", comment);
2252             sfree(comment);
2253         }
2254
2255         /*
2256          * Show password prompt, having first obtained it via a TIS
2257          * or CryptoCard exchange if we're doing TIS or CryptoCard
2258          * authentication.
2259          */
2260         if (ssh_get_line) {
2261             if (!ssh_get_line(prompt, password, sizeof(password), TRUE)) {
2262                 /*
2263                  * get_line failed to get a password (for example
2264                  * because one was supplied on the command line
2265                  * which has already failed to work). Terminate.
2266                  */
2267                 send_packet(SSH1_MSG_DISCONNECT,
2268                             PKT_STR, "No more passwords available to try",
2269                             PKT_END);
2270                 connection_fatal("Unable to authenticate");
2271                 ssh_state = SSH_STATE_CLOSED;
2272                 crReturn(1);
2273             }
2274         } else {
2275             /* Prompt may have come from server. We've munged it a bit, so
2276              * we know it to be zero-terminated at least once. */
2277             c_write_untrusted(prompt, strlen(prompt));
2278             pos = 0;
2279             ssh_send_ok = 1;
2280             while (pos >= 0) {
2281                 crWaitUntil(!ispkt);
2282                 while (inlen--)
2283                     switch (c = *in++) {
2284                       case 10:
2285                       case 13:
2286                         password[pos] = 0;
2287                         pos = -1;
2288                         break;
2289                       case 8:
2290                       case 127:
2291                         if (pos > 0)
2292                             pos--;
2293                         break;
2294                       case 21:
2295                       case 27:
2296                         pos = 0;
2297                         break;
2298                       case 3:
2299                       case 4:
2300                         random_save_seed();
2301                         exit(0);
2302                         break;
2303                       default:
2304                         if (pos < sizeof(password)-1)
2305                             password[pos++] = c;
2306                         break;
2307                     }
2308             }
2309             c_write_str("\r\n");
2310         }
2311
2312       tryauth:
2313         if (pwpkt_type == SSH1_CMSG_AUTH_RSA) {
2314             /*
2315              * Try public key authentication with the specified
2316              * key file.
2317              */
2318             static struct RSAKey pubkey;
2319             static Bignum challenge, response;
2320             static int i;
2321             static unsigned char buffer[32];
2322
2323             tried_publickey = 1;
2324             i = loadrsakey(cfg.keyfile, &pubkey, password);
2325             if (i == 0) {
2326                 c_write_str("Couldn't load private key from ");
2327                 c_write_str(cfg.keyfile);
2328                 c_write_str(".\r\n");
2329                 continue;              /* go and try password */
2330             }
2331             if (i == -1) {
2332                 c_write_str("Wrong passphrase.\r\n");
2333                 tried_publickey = 0;
2334                 continue;              /* try again */
2335             }
2336
2337             /*
2338              * Send a public key attempt.
2339              */
2340             send_packet(SSH1_CMSG_AUTH_RSA,
2341                         PKT_BIGNUM, pubkey.modulus, PKT_END);
2342
2343             crWaitUntil(ispkt);
2344             if (pktin.type == SSH1_SMSG_FAILURE) {
2345                 c_write_str("Server refused our public key.\r\n");
2346                 continue;              /* go and try password */
2347             }
2348             if (pktin.type != SSH1_SMSG_AUTH_RSA_CHALLENGE) {
2349                 bombout(("Bizarre response to offer of public key"));
2350                 crReturn(0);
2351             }
2352             ssh1_read_bignum(pktin.body, &challenge);
2353             response = rsadecrypt(challenge, &pubkey);
2354             freebn(pubkey.private_exponent);    /* burn the evidence */
2355
2356             for (i = 0; i < 32; i++) {
2357                 buffer[i] = bignum_byte(response, 31 - i);
2358             }
2359
2360             MD5Init(&md5c);
2361             MD5Update(&md5c, buffer, 32);
2362             MD5Update(&md5c, session_id, 16);
2363             MD5Final(buffer, &md5c);
2364
2365             send_packet(SSH1_CMSG_AUTH_RSA_RESPONSE,
2366                         PKT_DATA, buffer, 16, PKT_END);
2367
2368             crWaitUntil(ispkt);
2369             if (pktin.type == SSH1_SMSG_FAILURE) {
2370                 if (flags & FLAG_VERBOSE)
2371                     c_write_str
2372                         ("Failed to authenticate with our public key.\r\n");
2373                 continue;              /* go and try password */
2374             } else if (pktin.type != SSH1_SMSG_SUCCESS) {
2375                 bombout(
2376                         ("Bizarre response to RSA authentication response"));
2377                 crReturn(0);
2378             }
2379
2380             break;                     /* we're through! */
2381         } else {
2382             if (pwpkt_type == SSH1_CMSG_AUTH_PASSWORD) {
2383                 /*
2384                  * Defence against traffic analysis: we send a
2385                  * whole bunch of packets containing strings of
2386                  * different lengths. One of these strings is the
2387                  * password, in a SSH1_CMSG_AUTH_PASSWORD packet.
2388                  * The others are all random data in
2389                  * SSH1_MSG_IGNORE packets. This way a passive
2390                  * listener can't tell which is the password, and
2391                  * hence can't deduce the password length.
2392                  * 
2393                  * Anybody with a password length greater than 16
2394                  * bytes is going to have enough entropy in their
2395                  * password that a listener won't find it _that_
2396                  * much help to know how long it is. So what we'll
2397                  * do is:
2398                  * 
2399                  *  - if password length < 16, we send 15 packets
2400                  *    containing string lengths 1 through 15
2401                  * 
2402                  *  - otherwise, we let N be the nearest multiple
2403                  *    of 8 below the password length, and send 8
2404                  *    packets containing string lengths N through
2405                  *    N+7. This won't obscure the order of
2406                  *    magnitude of the password length, but it will
2407                  *    introduce a bit of extra uncertainty.
2408                  * 
2409                  * A few servers (the old 1.2.18 through 1.2.22)
2410                  * can't deal with SSH1_MSG_IGNORE. For these
2411                  * servers, we need an alternative defence. We make
2412                  * use of the fact that the password is interpreted
2413                  * as a C string: so we can append a NUL, then some
2414                  * random data.
2415                  * 
2416                  * One server (a Cisco one) can deal with neither
2417                  * SSH1_MSG_IGNORE _nor_ a padded password string.
2418                  * For this server we are left with no defences
2419                  * against password length sniffing.
2420                  */
2421                 if (!(ssh_remote_bugs & BUG_CHOKES_ON_SSH1_IGNORE)) {
2422                     /*
2423                      * The server can deal with SSH1_MSG_IGNORE, so
2424                      * we can use the primary defence.
2425                      */
2426                     int bottom, top, pwlen, i;
2427                     char *randomstr;
2428
2429                     pwlen = strlen(password);
2430                     if (pwlen < 16) {
2431                         bottom = 0;    /* zero length passwords are OK! :-) */
2432                         top = 15;
2433                     } else {
2434                         bottom = pwlen & ~7;
2435                         top = bottom + 7;
2436                     }
2437
2438                     assert(pwlen >= bottom && pwlen <= top);
2439
2440                     randomstr = smalloc(top + 1);
2441
2442                     for (i = bottom; i <= top; i++) {
2443                         if (i == pwlen)
2444                             defer_packet(pwpkt_type, PKT_STR, password,
2445                                          PKT_END);
2446                         else {
2447                             for (j = 0; j < i; j++) {
2448                                 do {
2449                                     randomstr[j] = random_byte();
2450                                 } while (randomstr[j] == '\0');
2451                             }
2452                             randomstr[i] = '\0';
2453                             defer_packet(SSH1_MSG_IGNORE,
2454                                          PKT_STR, randomstr, PKT_END);
2455                         }
2456                     }
2457                     logevent("Sending password with camouflage packets");
2458                     ssh_pkt_defersend();
2459                 } 
2460                 else if (!(ssh_remote_bugs & BUG_NEEDS_SSH1_PLAIN_PASSWORD)) {
2461                     /*
2462                      * The server can't deal with SSH1_MSG_IGNORE
2463                      * but can deal with padded passwords, so we
2464                      * can use the secondary defence.
2465                      */
2466                     char string[64];
2467                     char *s;
2468                     int len;
2469
2470                     len = strlen(password);
2471                     if (len < sizeof(string)) {
2472                         s = string;
2473                         strcpy(string, password);
2474                         len++;         /* cover the zero byte */
2475                         while (len < sizeof(string)) {
2476                             string[len++] = (char) random_byte();
2477                         }
2478                     } else {
2479                         s = password;
2480                     }
2481                     logevent("Sending length-padded password");
2482                     send_packet(pwpkt_type, PKT_INT, len,
2483                                 PKT_DATA, s, len, PKT_END);
2484                 } else {
2485                     /*
2486                      * The server has _both_
2487                      * BUG_CHOKES_ON_SSH1_IGNORE and
2488                      * BUG_NEEDS_SSH1_PLAIN_PASSWORD. There is
2489                      * therefore nothing we can do.
2490                      */
2491                     int len;
2492                     len = strlen(password);
2493                     logevent("Sending unpadded password");
2494                     send_packet(pwpkt_type, PKT_INT, len,
2495                                 PKT_DATA, password, len, PKT_END);
2496                 }
2497             } else {
2498                 send_packet(pwpkt_type, PKT_STR, password, PKT_END);
2499             }
2500         }
2501         logevent("Sent password");
2502         memset(password, 0, strlen(password));
2503         crWaitUntil(ispkt);
2504         if (pktin.type == SSH1_SMSG_FAILURE) {
2505             if (flags & FLAG_VERBOSE)
2506                 c_write_str("Access denied\r\n");
2507             logevent("Authentication refused");
2508         } else if (pktin.type == SSH1_MSG_DISCONNECT) {
2509             logevent("Received disconnect request");
2510             ssh_state = SSH_STATE_CLOSED;
2511             crReturn(1);
2512         } else if (pktin.type != SSH1_SMSG_SUCCESS) {
2513             bombout(("Strange packet received, type %d", pktin.type));
2514             crReturn(0);
2515         }
2516     }
2517
2518     logevent("Authentication successful");
2519
2520     crFinish(1);
2521 }
2522
2523 void sshfwd_close(struct ssh_channel *c)
2524 {
2525     if (c && !c->closes) {
2526         /*
2527          * If the channel's remoteid is -1, we have sent
2528          * CHANNEL_OPEN for this channel, but it hasn't even been
2529          * acknowledged by the server. So we must set a close flag
2530          * on it now, and then when the server acks the channel
2531          * open, we can close it then.
2532          */
2533         if (c->remoteid != -1) {
2534             if (ssh_version == 1) {
2535                 send_packet(SSH1_MSG_CHANNEL_CLOSE, PKT_INT, c->remoteid,
2536                             PKT_END);
2537             } else {
2538                 ssh2_pkt_init(SSH2_MSG_CHANNEL_CLOSE);
2539                 ssh2_pkt_adduint32(c->remoteid);
2540                 ssh2_pkt_send();
2541             }
2542         }
2543         c->closes = 1;
2544         if (c->type == CHAN_X11) {
2545             c->u.x11.s = NULL;
2546             logevent("Forwarded X11 connection terminated");
2547         } else if (c->type == CHAN_SOCKDATA ||
2548                    c->type == CHAN_SOCKDATA_DORMANT) {
2549             c->u.pfd.s = NULL;
2550             logevent("Forwarded port closed");
2551         }
2552     }
2553 }
2554
2555 int sshfwd_write(struct ssh_channel *c, char *buf, int len)
2556 {
2557     if (ssh_version == 1) {
2558         send_packet(SSH1_MSG_CHANNEL_DATA,
2559                     PKT_INT, c->remoteid,
2560                     PKT_INT, len, PKT_DATA, buf, len, PKT_END);
2561         /*
2562          * In SSH1 we can return 0 here - implying that forwarded
2563          * connections are never individually throttled - because
2564          * the only circumstance that can cause throttling will be
2565          * the whole SSH connection backing up, in which case
2566          * _everything_ will be throttled as a whole.
2567          */
2568         return 0;
2569     } else {
2570         ssh2_add_channel_data(c, buf, len);
2571         return ssh2_try_send(c);
2572     }
2573 }
2574
2575 void sshfwd_unthrottle(struct ssh_channel *c, int bufsize)
2576 {
2577     if (ssh_version == 1) {
2578         if (c->v.v1.throttling && bufsize < SSH1_BUFFER_LIMIT) {
2579             c->v.v1.throttling = 0;
2580             ssh1_throttle(-1);
2581         }
2582     } else {
2583         ssh2_set_window(c, OUR_V2_WINSIZE - bufsize);
2584     }
2585 }
2586
2587 static void ssh1_protocol(unsigned char *in, int inlen, int ispkt)
2588 {
2589     crBegin;
2590
2591     random_init();
2592
2593     while (!do_ssh1_login(in, inlen, ispkt)) {
2594         crReturnV;
2595     }
2596     if (ssh_state == SSH_STATE_CLOSED)
2597         crReturnV;
2598
2599     if (cfg.agentfwd && agent_exists()) {
2600         logevent("Requesting agent forwarding");
2601         send_packet(SSH1_CMSG_AGENT_REQUEST_FORWARDING, PKT_END);
2602         do {
2603             crReturnV;
2604         } while (!ispkt);
2605         if (pktin.type != SSH1_SMSG_SUCCESS
2606             && pktin.type != SSH1_SMSG_FAILURE) {
2607             bombout(("Protocol confusion"));
2608             crReturnV;
2609         } else if (pktin.type == SSH1_SMSG_FAILURE) {
2610             logevent("Agent forwarding refused");
2611         } else {
2612             logevent("Agent forwarding enabled");
2613             ssh_agentfwd_enabled = TRUE;
2614         }
2615     }
2616
2617     if (cfg.x11_forward) {
2618         char proto[20], data[64];
2619         logevent("Requesting X11 forwarding");
2620         x11_invent_auth(proto, sizeof(proto), data, sizeof(data));
2621         if (ssh1_local_protoflags & SSH1_PROTOFLAG_SCREEN_NUMBER) {
2622             send_packet(SSH1_CMSG_X11_REQUEST_FORWARDING,
2623                         PKT_STR, proto, PKT_STR, data,
2624                         PKT_INT, 0, PKT_END);
2625         } else {
2626             send_packet(SSH1_CMSG_X11_REQUEST_FORWARDING,
2627                         PKT_STR, proto, PKT_STR, data, PKT_END);
2628         }
2629         do {
2630             crReturnV;
2631         } while (!ispkt);
2632         if (pktin.type != SSH1_SMSG_SUCCESS
2633             && pktin.type != SSH1_SMSG_FAILURE) {
2634             bombout(("Protocol confusion"));
2635             crReturnV;
2636         } else if (pktin.type == SSH1_SMSG_FAILURE) {
2637             logevent("X11 forwarding refused");
2638         } else {
2639             logevent("X11 forwarding enabled");
2640             ssh_X11_fwd_enabled = TRUE;
2641         }
2642     }
2643
2644     {
2645         char type, *e;
2646         int n;
2647         int sport,dport;
2648         char sports[256], dports[256], host[256];
2649         char buf[1024];
2650
2651         ssh_rportfwds = newtree234(ssh_rportcmp_ssh1);
2652         /* Add port forwardings. */
2653         e = cfg.portfwd;
2654         while (*e) {
2655             type = *e++;
2656             n = 0;
2657             while (*e && *e != '\t')
2658                 sports[n++] = *e++;
2659             sports[n] = 0;
2660             if (*e == '\t')
2661                 e++;
2662             n = 0;
2663             while (*e && *e != ':')
2664                 host[n++] = *e++;
2665             host[n] = 0;
2666             if (*e == ':')
2667                 e++;
2668             n = 0;
2669             while (*e)
2670                 dports[n++] = *e++;
2671             dports[n] = 0;
2672             e++;
2673             dport = atoi(dports);
2674             sport = atoi(sports);
2675             if (sport && dport) {
2676                 if (type == 'L') {
2677                     pfd_addforward(host, dport, sport);
2678                     sprintf(buf, "Local port %d forwarding to %s:%d",
2679                             sport, host, dport);
2680                     logevent(buf);
2681                 } else {
2682                     struct ssh_rportfwd *pf;
2683                     pf = smalloc(sizeof(*pf));
2684                     strcpy(pf->dhost, host);
2685                     pf->dport = dport;
2686                     if (add234(ssh_rportfwds, pf) != pf) {
2687                         sprintf(buf, 
2688                                 "Duplicate remote port forwarding to %s:%d",
2689                                 host, dport);
2690                         logevent(buf);
2691                         sfree(pf);
2692                     } else {
2693                         sprintf(buf, "Requesting remote port %d forward to %s:%d",
2694                                 sport, host, dport);
2695                         logevent(buf);
2696                         send_packet(SSH1_CMSG_PORT_FORWARD_REQUEST,
2697                                     PKT_INT, sport,
2698                                     PKT_STR, host,
2699                                     PKT_INT, dport,
2700                                     PKT_END);
2701                     }
2702                 }
2703             }
2704         }
2705     }
2706
2707     if (!cfg.nopty) {
2708         send_packet(SSH1_CMSG_REQUEST_PTY,
2709                     PKT_STR, cfg.termtype,
2710                     PKT_INT, rows, PKT_INT, cols,
2711                     PKT_INT, 0, PKT_INT, 0, PKT_CHAR, 0, PKT_END);
2712         ssh_state = SSH_STATE_INTERMED;
2713         do {
2714             crReturnV;
2715         } while (!ispkt);
2716         if (pktin.type != SSH1_SMSG_SUCCESS
2717             && pktin.type != SSH1_SMSG_FAILURE) {
2718             bombout(("Protocol confusion"));
2719             crReturnV;
2720         } else if (pktin.type == SSH1_SMSG_FAILURE) {
2721             c_write_str("Server refused to allocate pty\r\n");
2722             ssh_editing = ssh_echoing = 1;
2723         }
2724         logevent("Allocated pty");
2725     } else {
2726         ssh_editing = ssh_echoing = 1;
2727     }
2728
2729     if (cfg.compression) {
2730         send_packet(SSH1_CMSG_REQUEST_COMPRESSION, PKT_INT, 6, PKT_END);
2731         do {
2732             crReturnV;
2733         } while (!ispkt);
2734         if (pktin.type != SSH1_SMSG_SUCCESS
2735             && pktin.type != SSH1_SMSG_FAILURE) {
2736             bombout(("Protocol confusion"));
2737             crReturnV;
2738         } else if (pktin.type == SSH1_SMSG_FAILURE) {
2739             c_write_str("Server refused to compress\r\n");
2740         }
2741         logevent("Started compression");
2742         ssh1_compressing = TRUE;
2743         zlib_compress_init();
2744         zlib_decompress_init();
2745     }
2746
2747     /*
2748      * Start the shell or command.
2749      * 
2750      * Special case: if the first-choice command is an SSH2
2751      * subsystem (hence not usable here) and the second choice
2752      * exists, we fall straight back to that.
2753      */
2754     {
2755         char *cmd = cfg.remote_cmd_ptr;
2756         
2757         if (cfg.ssh_subsys && cfg.remote_cmd_ptr2) {
2758             cmd = cfg.remote_cmd_ptr2;
2759             ssh_fallback_cmd = TRUE;
2760         }
2761         if (*cmd)
2762             send_packet(SSH1_CMSG_EXEC_CMD, PKT_STR, cmd, PKT_END);
2763         else
2764             send_packet(SSH1_CMSG_EXEC_SHELL, PKT_END);
2765         logevent("Started session");
2766     }
2767
2768     ssh_state = SSH_STATE_SESSION;
2769     if (size_needed)
2770         ssh_size();
2771     if (eof_needed)
2772         ssh_special(TS_EOF);
2773
2774     ldisc_send(NULL, 0, 0);            /* cause ldisc to notice changes */
2775     ssh_send_ok = 1;
2776     ssh_channels = newtree234(ssh_channelcmp);
2777     while (1) {
2778         crReturnV;
2779         if (ispkt) {
2780             if (pktin.type == SSH1_SMSG_STDOUT_DATA ||
2781                 pktin.type == SSH1_SMSG_STDERR_DATA) {
2782                 long len = GET_32BIT(pktin.body);
2783                 int bufsize =
2784                     from_backend(pktin.type == SSH1_SMSG_STDERR_DATA,
2785                                  pktin.body + 4, len);
2786                 if (!ssh1_stdout_throttling && bufsize > SSH1_BUFFER_LIMIT) {
2787                     ssh1_stdout_throttling = 1;
2788                     ssh1_throttle(+1);
2789                 }
2790             } else if (pktin.type == SSH1_MSG_DISCONNECT) {
2791                 ssh_state = SSH_STATE_CLOSED;
2792                 logevent("Received disconnect request");
2793                 crReturnV;
2794             } else if (pktin.type == SSH1_SMSG_X11_OPEN) {
2795                 /* Remote side is trying to open a channel to talk to our
2796                  * X-Server. Give them back a local channel number. */
2797                 struct ssh_channel *c;
2798
2799                 logevent("Received X11 connect request");
2800                 /* Refuse if X11 forwarding is disabled. */
2801                 if (!ssh_X11_fwd_enabled) {
2802                     send_packet(SSH1_MSG_CHANNEL_OPEN_FAILURE,
2803                                 PKT_INT, GET_32BIT(pktin.body), PKT_END);
2804                     logevent("Rejected X11 connect request");
2805                 } else {
2806                     c = smalloc(sizeof(struct ssh_channel));
2807
2808                     if (x11_init(&c->u.x11.s, cfg.x11_display, c) != NULL) {
2809                         logevent("opening X11 forward connection failed");
2810                         sfree(c);
2811                         send_packet(SSH1_MSG_CHANNEL_OPEN_FAILURE,
2812                                     PKT_INT, GET_32BIT(pktin.body),
2813                                     PKT_END);
2814                     } else {
2815                         logevent
2816                             ("opening X11 forward connection succeeded");
2817                         c->remoteid = GET_32BIT(pktin.body);
2818                         c->localid = alloc_channel_id();
2819                         c->closes = 0;
2820                         c->v.v1.throttling = 0;
2821                         c->type = CHAN_X11;     /* identify channel type */
2822                         add234(ssh_channels, c);
2823                         send_packet(SSH1_MSG_CHANNEL_OPEN_CONFIRMATION,
2824                                     PKT_INT, c->remoteid, PKT_INT,
2825                                     c->localid, PKT_END);
2826                         logevent("Opened X11 forward channel");
2827                     }
2828                 }
2829             } else if (pktin.type == SSH1_SMSG_AGENT_OPEN) {
2830                 /* Remote side is trying to open a channel to talk to our
2831                  * agent. Give them back a local channel number. */
2832                 struct ssh_channel *c;
2833
2834                 /* Refuse if agent forwarding is disabled. */
2835                 if (!ssh_agentfwd_enabled) {
2836                     send_packet(SSH1_MSG_CHANNEL_OPEN_FAILURE,
2837                                 PKT_INT, GET_32BIT(pktin.body), PKT_END);
2838                 } else {
2839                     c = smalloc(sizeof(struct ssh_channel));
2840                     c->remoteid = GET_32BIT(pktin.body);
2841                     c->localid = alloc_channel_id();
2842                     c->closes = 0;
2843                     c->v.v1.throttling = 0;
2844                     c->type = CHAN_AGENT;       /* identify channel type */
2845                     c->u.a.lensofar = 0;
2846                     add234(ssh_channels, c);
2847                     send_packet(SSH1_MSG_CHANNEL_OPEN_CONFIRMATION,
2848                                 PKT_INT, c->remoteid, PKT_INT, c->localid,
2849                                 PKT_END);
2850                 }
2851             } else if (pktin.type == SSH1_MSG_PORT_OPEN) {
2852                 /* Remote side is trying to open a channel to talk to a
2853                  * forwarded port. Give them back a local channel number. */
2854                 struct ssh_channel *c;
2855                 struct ssh_rportfwd pf;
2856                 int hostsize, port;
2857                 char host[256], buf[1024];
2858                 char *p, *h, *e;
2859                 c = smalloc(sizeof(struct ssh_channel));
2860
2861                 hostsize = GET_32BIT(pktin.body+4);
2862                 for(h = host, p = pktin.body+8; hostsize != 0; hostsize--) {
2863                     if (h+1 < host+sizeof(host))
2864                         *h++ = *p;
2865                     p++;
2866                 }
2867                 *h = 0;
2868                 port = GET_32BIT(p);
2869
2870                 strcpy(pf.dhost, host);
2871                 pf.dport = port;
2872
2873                 if (find234(ssh_rportfwds, &pf, NULL) == NULL) {
2874                     sprintf(buf, "Rejected remote port open request for %s:%d",
2875                             host, port);
2876                     logevent(buf);
2877                     send_packet(SSH1_MSG_CHANNEL_OPEN_FAILURE,
2878                                 PKT_INT, GET_32BIT(pktin.body), PKT_END);
2879                 } else {
2880                     sprintf(buf, "Received remote port open request for %s:%d",
2881                             host, port);
2882                     logevent(buf);
2883                     e = pfd_newconnect(&c->u.pfd.s, host, port, c);
2884                     if (e != NULL) {
2885                         char buf[256];
2886                         sprintf(buf, "Port open failed: %s", e);
2887                         logevent(buf);
2888                         sfree(c);
2889                         send_packet(SSH1_MSG_CHANNEL_OPEN_FAILURE,
2890                                     PKT_INT, GET_32BIT(pktin.body),
2891                                     PKT_END);
2892                     } else {
2893                         c->remoteid = GET_32BIT(pktin.body);
2894                         c->localid = alloc_channel_id();
2895                         c->closes = 0;
2896                         c->v.v1.throttling = 0;
2897                         c->type = CHAN_SOCKDATA;        /* identify channel type */
2898                         add234(ssh_channels, c);
2899                         send_packet(SSH1_MSG_CHANNEL_OPEN_CONFIRMATION,
2900                                     PKT_INT, c->remoteid, PKT_INT,
2901                                     c->localid, PKT_END);
2902                         logevent("Forwarded port opened successfully");
2903                     }
2904                 }
2905
2906             } else if (pktin.type == SSH1_MSG_CHANNEL_OPEN_CONFIRMATION) {
2907                 unsigned int remoteid = GET_32BIT(pktin.body);
2908                 unsigned int localid = GET_32BIT(pktin.body+4);
2909                 struct ssh_channel *c;
2910
2911                 c = find234(ssh_channels, &remoteid, ssh_channelfind);
2912                 if (c && c->type == CHAN_SOCKDATA_DORMANT) {
2913                     c->remoteid = localid;
2914                     c->type = CHAN_SOCKDATA;
2915                     pfd_confirm(c->u.pfd.s);
2916                 }
2917
2918                 if (c && c->closes) {
2919                     /*
2920                      * We have a pending close on this channel,
2921                      * which we decided on before the server acked
2922                      * the channel open. So now we know the
2923                      * remoteid, we can close it again.
2924                      */
2925                     send_packet(SSH1_MSG_CHANNEL_CLOSE, PKT_INT, c->remoteid,
2926                                 PKT_END);
2927                 }
2928
2929             } else if (pktin.type == SSH1_MSG_CHANNEL_OPEN_FAILURE) {
2930                 unsigned int remoteid = GET_32BIT(pktin.body);
2931                 unsigned int localid = GET_32BIT(pktin.body+4);
2932                 struct ssh_channel *c;
2933
2934                 c = find234(ssh_channels, &remoteid, ssh_channelfind);
2935                 if (c && c->type == CHAN_SOCKDATA_DORMANT) {
2936                     logevent("Forwarded connection refused by server");
2937                     pfd_close(c->u.pfd.s);
2938                     del234(ssh_channels, c);
2939                     sfree(c);
2940                 }
2941
2942             } else if (pktin.type == SSH1_MSG_CHANNEL_CLOSE ||
2943                        pktin.type == SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION) {
2944                 /* Remote side closes a channel. */
2945                 unsigned i = GET_32BIT(pktin.body);
2946                 struct ssh_channel *c;
2947                 c = find234(ssh_channels, &i, ssh_channelfind);
2948                 if (c) {
2949                     int closetype;
2950                     closetype =
2951                         (pktin.type == SSH1_MSG_CHANNEL_CLOSE ? 1 : 2);
2952                     if (!(c->closes & closetype))
2953                         send_packet(pktin.type, PKT_INT, c->remoteid,
2954                                     PKT_END);
2955                     if ((c->closes == 0) && (c->type == CHAN_X11)) {
2956                         logevent("Forwarded X11 connection terminated");
2957                         assert(c->u.x11.s != NULL);
2958                         x11_close(c->u.x11.s);
2959                         c->u.x11.s = NULL;
2960                     }
2961                     if ((c->closes == 0) && (c->type == CHAN_SOCKDATA)) {
2962                         logevent("Forwarded port closed");
2963                         assert(c->u.pfd.s != NULL);
2964                         pfd_close(c->u.pfd.s);
2965                         c->u.pfd.s = NULL;
2966                     }
2967                     c->closes |= closetype;
2968                     if (c->closes == 3) {
2969                         del234(ssh_channels, c);
2970                         sfree(c);
2971                     }
2972                 }
2973             } else if (pktin.type == SSH1_MSG_CHANNEL_DATA) {
2974                 /* Data sent down one of our channels. */
2975                 int i = GET_32BIT(pktin.body);
2976                 int len = GET_32BIT(pktin.body + 4);
2977                 unsigned char *p = pktin.body + 8;
2978                 struct ssh_channel *c;
2979                 c = find234(ssh_channels, &i, ssh_channelfind);
2980                 if (c) {
2981                     int bufsize;
2982                     switch (c->type) {
2983                       case CHAN_X11:
2984                         bufsize = x11_send(c->u.x11.s, p, len);
2985                         break;
2986                       case CHAN_SOCKDATA:
2987                         bufsize = pfd_send(c->u.pfd.s, p, len);
2988                         break;
2989                       case CHAN_AGENT:
2990                         /* Data for an agent message. Buffer it. */
2991                         while (len > 0) {
2992                             if (c->u.a.lensofar < 4) {
2993                                 int l = min(4 - c->u.a.lensofar, len);
2994                                 memcpy(c->u.a.msglen + c->u.a.lensofar, p,
2995                                        l);
2996                                 p += l;
2997                                 len -= l;
2998                                 c->u.a.lensofar += l;
2999                             }
3000                             if (c->u.a.lensofar == 4) {
3001                                 c->u.a.totallen =
3002                                     4 + GET_32BIT(c->u.a.msglen);
3003                                 c->u.a.message = smalloc(c->u.a.totallen);
3004                                 memcpy(c->u.a.message, c->u.a.msglen, 4);
3005                             }
3006                             if (c->u.a.lensofar >= 4 && len > 0) {
3007                                 int l =
3008                                     min(c->u.a.totallen - c->u.a.lensofar,
3009                                         len);
3010                                 memcpy(c->u.a.message + c->u.a.lensofar, p,
3011                                        l);
3012                                 p += l;
3013                                 len -= l;
3014                                 c->u.a.lensofar += l;
3015                             }
3016                             if (c->u.a.lensofar == c->u.a.totallen) {
3017                                 void *reply, *sentreply;
3018                                 int replylen;
3019                                 agent_query(c->u.a.message,
3020                                             c->u.a.totallen, &reply,
3021                                             &replylen);
3022                                 if (reply)
3023                                     sentreply = reply;
3024                                 else {
3025                                     /* Fake SSH_AGENT_FAILURE. */
3026                                     sentreply = "\0\0\0\1\5";
3027                                     replylen = 5;
3028                                 }
3029                                 send_packet(SSH1_MSG_CHANNEL_DATA,
3030                                             PKT_INT, c->remoteid,
3031                                             PKT_INT, replylen,
3032                                             PKT_DATA, sentreply, replylen,
3033                                             PKT_END);
3034                                 if (reply)
3035                                     sfree(reply);
3036                                 sfree(c->u.a.message);
3037                                 c->u.a.lensofar = 0;
3038                             }
3039                         }
3040                         bufsize = 0;   /* agent channels never back up */
3041                         break;
3042                     }
3043                     if (!c->v.v1.throttling && bufsize > SSH1_BUFFER_LIMIT) {
3044                         c->v.v1.throttling = 1;
3045                         ssh1_throttle(+1);
3046                     }
3047                 }
3048             } else if (pktin.type == SSH1_SMSG_SUCCESS) {
3049                 /* may be from EXEC_SHELL on some servers */
3050             } else if (pktin.type == SSH1_SMSG_FAILURE) {
3051                 /* may be from EXEC_SHELL on some servers
3052                  * if no pty is available or in other odd cases. Ignore */
3053             } else if (pktin.type == SSH1_SMSG_EXIT_STATUS) {
3054                 send_packet(SSH1_CMSG_EXIT_CONFIRMATION, PKT_END);
3055                 /*
3056                  * In case `helpful' firewalls or proxies tack
3057                  * extra human-readable text on the end of the
3058                  * session which we might mistake for another
3059                  * encrypted packet, we close the session once
3060                  * we've sent EXIT_CONFIRMATION.
3061                  */
3062                 ssh_state = SSH_STATE_CLOSED;
3063                 crReturnV;
3064             } else {
3065                 bombout(("Strange packet received: type %d", pktin.type));
3066                 crReturnV;
3067             }
3068         } else {
3069             while (inlen > 0) {
3070                 int len = min(inlen, 512);
3071                 send_packet(SSH1_CMSG_STDIN_DATA,
3072                             PKT_INT, len, PKT_DATA, in, len, PKT_END);
3073                 in += len;
3074                 inlen -= len;
3075             }
3076         }
3077     }
3078
3079     crFinishV;
3080 }
3081
3082 /*
3083  * Utility routine for decoding comma-separated strings in KEXINIT.
3084  */
3085 static int in_commasep_string(char *needle, char *haystack, int haylen)
3086 {
3087     int needlen = strlen(needle);
3088     while (1) {
3089         /*
3090          * Is it at the start of the string?
3091          */
3092         if (haylen >= needlen &&       /* haystack is long enough */
3093             !memcmp(needle, haystack, needlen) &&       /* initial match */
3094             (haylen == needlen || haystack[needlen] == ',')
3095             /* either , or EOS follows */
3096             )
3097             return 1;
3098         /*
3099          * If not, search for the next comma and resume after that.
3100          * If no comma found, terminate.
3101          */
3102         while (haylen > 0 && *haystack != ',')
3103             haylen--, haystack++;
3104         if (haylen == 0)
3105             return 0;
3106         haylen--, haystack++;          /* skip over comma itself */
3107     }
3108 }
3109
3110 /*
3111  * SSH2 key creation method.
3112  */
3113 static void ssh2_mkkey(Bignum K, char *H, char *sessid, char chr,
3114                        char *keyspace)
3115 {
3116     SHA_State s;
3117     /* First 20 bytes. */
3118     SHA_Init(&s);
3119     sha_mpint(&s, K);
3120     SHA_Bytes(&s, H, 20);
3121     SHA_Bytes(&s, &chr, 1);
3122     SHA_Bytes(&s, sessid, 20);
3123     SHA_Final(&s, keyspace);
3124     /* Next 20 bytes. */
3125     SHA_Init(&s);
3126     sha_mpint(&s, K);
3127     SHA_Bytes(&s, H, 20);
3128     SHA_Bytes(&s, keyspace, 20);
3129     SHA_Final(&s, keyspace + 20);
3130 }
3131
3132 /*
3133  * Handle the SSH2 transport layer.
3134  */
3135 static int do_ssh2_transport(unsigned char *in, int inlen, int ispkt)
3136 {
3137     static int i, j, len, nbits, pbits, warn;
3138     static char *str;
3139     static Bignum p, g, e, f, K;
3140     static int kex_init_value, kex_reply_value;
3141     static const struct ssh_mac **maclist;
3142     static int nmacs;
3143     static const struct ssh2_cipher *cscipher_tobe = NULL;
3144     static const struct ssh2_cipher *sccipher_tobe = NULL;
3145     static const struct ssh_mac *csmac_tobe = NULL;
3146     static const struct ssh_mac *scmac_tobe = NULL;
3147     static const struct ssh_compress *cscomp_tobe = NULL;
3148     static const struct ssh_compress *sccomp_tobe = NULL;
3149     static char *hostkeydata, *sigdata, *keystr, *fingerprint;
3150     static int hostkeylen, siglen;
3151     static void *hkey;                 /* actual host key */
3152     static unsigned char exchange_hash[20];
3153     static unsigned char keyspace[40];
3154     static int n_preferred_ciphers;
3155     static const struct ssh2_ciphers *preferred_ciphers[CIPHER_MAX];
3156     static const struct ssh_compress *preferred_comp;
3157     static int first_kex;
3158
3159     crBegin;
3160     random_init();
3161     first_kex = 1;
3162
3163     /*
3164      * Set up the preferred ciphers. (NULL => warn below here)
3165      */
3166     n_preferred_ciphers = 0;
3167     for (i = 0; i < CIPHER_MAX; i++) {
3168         switch (cfg.ssh_cipherlist[i]) {
3169           case CIPHER_BLOWFISH:
3170             preferred_ciphers[n_preferred_ciphers] = &ssh2_blowfish;
3171             n_preferred_ciphers++;
3172             break;
3173           case CIPHER_DES:
3174             /* Not supported in SSH2; silently drop */
3175             break;
3176           case CIPHER_3DES:
3177             preferred_ciphers[n_preferred_ciphers] = &ssh2_3des;
3178             n_preferred_ciphers++;
3179             break;
3180           case CIPHER_AES:
3181             preferred_ciphers[n_preferred_ciphers] = &ssh2_aes;
3182             n_preferred_ciphers++;
3183             break;
3184           case CIPHER_WARN:
3185             /* Flag for later. Don't bother if it's the last in
3186              * the list. */
3187             if (i < CIPHER_MAX - 1) {
3188                 preferred_ciphers[n_preferred_ciphers] = NULL;
3189                 n_preferred_ciphers++;
3190             }
3191             break;
3192         }
3193     }
3194
3195     /*
3196      * Set up preferred compression.
3197      */
3198     if (cfg.compression)
3199         preferred_comp = &ssh_zlib;
3200     else
3201         preferred_comp = &ssh_comp_none;
3202
3203     /*
3204      * Be prepared to work around the buggy MAC problem.
3205      */
3206     if (cfg.buggymac || (ssh_remote_bugs & BUG_SSH2_HMAC))
3207         maclist = buggymacs, nmacs = lenof(buggymacs);
3208     else
3209         maclist = macs, nmacs = lenof(macs);
3210
3211   begin_key_exchange:
3212     /*
3213      * Construct and send our key exchange packet.
3214      */
3215     ssh2_pkt_init(SSH2_MSG_KEXINIT);
3216     for (i = 0; i < 16; i++)
3217         ssh2_pkt_addbyte((unsigned char) random_byte());
3218     /* List key exchange algorithms. */
3219     ssh2_pkt_addstring_start();
3220     for (i = 0; i < lenof(kex_algs); i++) {
3221         ssh2_pkt_addstring_str(kex_algs[i]->name);
3222         if (i < lenof(kex_algs) - 1)
3223             ssh2_pkt_addstring_str(",");
3224     }
3225     /* List server host key algorithms. */
3226     ssh2_pkt_addstring_start();
3227     for (i = 0; i < lenof(hostkey_algs); i++) {
3228         ssh2_pkt_addstring_str(hostkey_algs[i]->name);
3229         if (i < lenof(hostkey_algs) - 1)
3230             ssh2_pkt_addstring_str(",");
3231     }
3232     /* List client->server encryption algorithms. */
3233     ssh2_pkt_addstring_start();
3234     for (i = 0; i < n_preferred_ciphers; i++) {
3235         const struct ssh2_ciphers *c = preferred_ciphers[i];
3236         if (!c) continue;              /* warning flag */
3237         for (j = 0; j < c->nciphers; j++) {
3238             ssh2_pkt_addstring_str(c->list[j]->name);
3239             if (i < n_preferred_ciphers || j < c->nciphers - 1)
3240                 ssh2_pkt_addstring_str(",");
3241         }
3242     }
3243     /* List server->client encryption algorithms. */
3244     ssh2_pkt_addstring_start();
3245     for (i = 0; i < n_preferred_ciphers; i++) {
3246         const struct ssh2_ciphers *c = preferred_ciphers[i];
3247         if (!c) continue; /* warning flag */
3248         for (j = 0; j < c->nciphers; j++) {
3249             ssh2_pkt_addstring_str(c->list[j]->name);
3250             if (i < n_preferred_ciphers || j < c->nciphers - 1)
3251                 ssh2_pkt_addstring_str(",");
3252         }
3253     }
3254     /* List client->server MAC algorithms. */
3255     ssh2_pkt_addstring_start();
3256     for (i = 0; i < nmacs; i++) {
3257         ssh2_pkt_addstring_str(maclist[i]->name);
3258         if (i < nmacs - 1)
3259             ssh2_pkt_addstring_str(",");
3260     }
3261     /* List server->client MAC algorithms. */
3262     ssh2_pkt_addstring_start();
3263     for (i = 0; i < nmacs; i++) {
3264         ssh2_pkt_addstring_str(maclist[i]->name);
3265         if (i < nmacs - 1)
3266             ssh2_pkt_addstring_str(",");
3267     }
3268     /* List client->server compression algorithms. */
3269     ssh2_pkt_addstring_start();
3270     for (i = 0; i < lenof(compressions) + 1; i++) {
3271         const struct ssh_compress *c =
3272             i == 0 ? preferred_comp : compressions[i - 1];
3273         ssh2_pkt_addstring_str(c->name);
3274         if (i < lenof(compressions))
3275             ssh2_pkt_addstring_str(",");
3276     }
3277     /* List server->client compression algorithms. */
3278     ssh2_pkt_addstring_start();
3279     for (i = 0; i < lenof(compressions) + 1; i++) {
3280         const struct ssh_compress *c =
3281             i == 0 ? preferred_comp : compressions[i - 1];
3282         ssh2_pkt_addstring_str(c->name);
3283         if (i < lenof(compressions))
3284             ssh2_pkt_addstring_str(",");
3285     }
3286     /* List client->server languages. Empty list. */
3287     ssh2_pkt_addstring_start();
3288     /* List server->client languages. Empty list. */
3289     ssh2_pkt_addstring_start();
3290     /* First KEX packet does _not_ follow, because we're not that brave. */
3291     ssh2_pkt_addbool(FALSE);
3292     /* Reserved. */
3293     ssh2_pkt_adduint32(0);
3294
3295     exhash = exhashbase;
3296     sha_string(&exhash, pktout.data + 5, pktout.length - 5);
3297
3298     ssh2_pkt_send();
3299
3300     if (!ispkt)
3301         crWaitUntil(ispkt);
3302     sha_string(&exhash, pktin.data + 5, pktin.length - 5);
3303
3304     /*
3305      * Now examine the other side's KEXINIT to see what we're up
3306      * to.
3307      */
3308     if (pktin.type != SSH2_MSG_KEXINIT) {
3309         bombout(("expected key exchange packet from server"));
3310         crReturn(0);
3311     }
3312     kex = NULL;
3313     hostkey = NULL;
3314     cscipher_tobe = NULL;
3315     sccipher_tobe = NULL;
3316     csmac_tobe = NULL;
3317     scmac_tobe = NULL;
3318     cscomp_tobe = NULL;
3319     sccomp_tobe = NULL;
3320     pktin.savedpos += 16;              /* skip garbage cookie */
3321     ssh2_pkt_getstring(&str, &len);    /* key exchange algorithms */
3322     for (i = 0; i < lenof(kex_algs); i++) {
3323         if (in_commasep_string(kex_algs[i]->name, str, len)) {
3324             kex = kex_algs[i];
3325             break;
3326         }
3327     }
3328     ssh2_pkt_getstring(&str, &len);    /* host key algorithms */
3329     for (i = 0; i < lenof(hostkey_algs); i++) {
3330         if (in_commasep_string(hostkey_algs[i]->name, str, len)) {
3331             hostkey = hostkey_algs[i];
3332             break;
3333         }
3334     }
3335     ssh2_pkt_getstring(&str, &len);    /* client->server cipher */
3336     warn = 0;
3337     for (i = 0; i < n_preferred_ciphers; i++) {
3338         const struct ssh2_ciphers *c = preferred_ciphers[i];
3339         if (!c) {
3340             warn = 1;
3341         } else {
3342             for (j = 0; j < c->nciphers; j++) {
3343                 if (in_commasep_string(c->list[j]->name, str, len)) {
3344                     cscipher_tobe = c->list[j];
3345                     break;
3346                 }
3347             }
3348         }
3349         if (cscipher_tobe) {
3350             if (warn)
3351                 askcipher(cscipher_tobe->name, 1);
3352             break;
3353         }
3354     }
3355     ssh2_pkt_getstring(&str, &len);    /* server->client cipher */
3356     warn = 0;
3357     for (i = 0; i < n_preferred_ciphers; i++) {
3358         const struct ssh2_ciphers *c = preferred_ciphers[i];
3359         if (!c) {
3360             warn = 1;
3361         } else {
3362             for (j = 0; j < c->nciphers; j++) {
3363                 if (in_commasep_string(c->list[j]->name, str, len)) {
3364                     sccipher_tobe = c->list[j];
3365                     break;
3366                 }
3367             }
3368         }
3369         if (sccipher_tobe) {
3370             if (warn)
3371                 askcipher(sccipher_tobe->name, 2);
3372             break;
3373         }
3374     }
3375     ssh2_pkt_getstring(&str, &len);    /* client->server mac */
3376     for (i = 0; i < nmacs; i++) {
3377         if (in_commasep_string(maclist[i]->name, str, len)) {
3378             csmac_tobe = maclist[i];
3379             break;
3380         }
3381     }
3382     ssh2_pkt_getstring(&str, &len);    /* server->client mac */
3383     for (i = 0; i < nmacs; i++) {
3384         if (in_commasep_string(maclist[i]->name, str, len)) {
3385             scmac_tobe = maclist[i];
3386             break;
3387         }
3388     }
3389     ssh2_pkt_getstring(&str, &len);    /* client->server compression */
3390     for (i = 0; i < lenof(compressions) + 1; i++) {
3391         const struct ssh_compress *c =
3392             i == 0 ? preferred_comp : compressions[i - 1];
3393         if (in_commasep_string(c->name, str, len)) {
3394             cscomp_tobe = c;
3395             break;
3396         }
3397     }
3398     ssh2_pkt_getstring(&str, &len);    /* server->client compression */
3399     for (i = 0; i < lenof(compressions) + 1; i++) {
3400         const struct ssh_compress *c =
3401             i == 0 ? preferred_comp : compressions[i - 1];
3402         if (in_commasep_string(c->name, str, len)) {
3403             sccomp_tobe = c;
3404             break;
3405         }
3406     }
3407
3408     /*
3409      * Work out the number of bits of key we will need from the key
3410      * exchange. We start with the maximum key length of either
3411      * cipher...
3412      */
3413     {
3414         int csbits, scbits;
3415
3416         csbits = cscipher_tobe->keylen;
3417         scbits = sccipher_tobe->keylen;
3418         nbits = (csbits > scbits ? csbits : scbits);
3419     }
3420     /* The keys only have 160-bit entropy, since they're based on
3421      * a SHA-1 hash. So cap the key size at 160 bits. */
3422     if (nbits > 160)
3423         nbits = 160;
3424
3425     /*
3426      * If we're doing Diffie-Hellman group exchange, start by
3427      * requesting a group.
3428      */
3429     if (kex == &ssh_diffiehellman_gex) {
3430         logevent("Doing Diffie-Hellman group exchange");
3431         /*
3432          * Work out how big a DH group we will need to allow that
3433          * much data.
3434          */
3435         pbits = 512 << ((nbits - 1) / 64);
3436         ssh2_pkt_init(SSH2_MSG_KEX_DH_GEX_REQUEST);
3437         ssh2_pkt_adduint32(pbits);
3438         ssh2_pkt_send();
3439
3440         crWaitUntil(ispkt);
3441         if (pktin.type != SSH2_MSG_KEX_DH_GEX_GROUP) {
3442             bombout(("expected key exchange group packet from server"));
3443             crReturn(0);
3444         }
3445         p = ssh2_pkt_getmp();
3446         g = ssh2_pkt_getmp();
3447         dh_setup_group(p, g);
3448         kex_init_value = SSH2_MSG_KEX_DH_GEX_INIT;
3449         kex_reply_value = SSH2_MSG_KEX_DH_GEX_REPLY;
3450     } else {
3451         dh_setup_group1();
3452         kex_init_value = SSH2_MSG_KEXDH_INIT;
3453         kex_reply_value = SSH2_MSG_KEXDH_REPLY;
3454     }
3455
3456     logevent("Doing Diffie-Hellman key exchange");
3457     /*
3458      * Now generate and send e for Diffie-Hellman.
3459      */
3460     e = dh_create_e(nbits * 2);
3461     ssh2_pkt_init(kex_init_value);
3462     ssh2_pkt_addmp(e);
3463     ssh2_pkt_send();
3464
3465     crWaitUntil(ispkt);
3466     if (pktin.type != kex_reply_value) {
3467         bombout(("expected key exchange reply packet from server"));
3468         crReturn(0);
3469     }
3470     ssh2_pkt_getstring(&hostkeydata, &hostkeylen);
3471     f = ssh2_pkt_getmp();
3472     ssh2_pkt_getstring(&sigdata, &siglen);
3473
3474     K = dh_find_K(f);
3475
3476     sha_string(&exhash, hostkeydata, hostkeylen);
3477     if (kex == &ssh_diffiehellman_gex) {
3478         sha_uint32(&exhash, pbits);
3479         sha_mpint(&exhash, p);
3480         sha_mpint(&exhash, g);
3481     }
3482     sha_mpint(&exhash, e);
3483     sha_mpint(&exhash, f);
3484     sha_mpint(&exhash, K);
3485     SHA_Final(&exhash, exchange_hash);
3486
3487     dh_cleanup();
3488
3489 #if 0
3490     debug(("Exchange hash is:\n"));
3491     dmemdump(exchange_hash, 20);
3492 #endif
3493
3494     hkey = hostkey->newkey(hostkeydata, hostkeylen);
3495     if (!hkey ||
3496         !hostkey->verifysig(hkey, sigdata, siglen, exchange_hash, 20)) {
3497         bombout(("Server's host key did not match the signature supplied"));
3498         crReturn(0);
3499     }
3500
3501     /*
3502      * Expect SSH2_MSG_NEWKEYS from server.
3503      */
3504     crWaitUntil(ispkt);
3505     if (pktin.type != SSH2_MSG_NEWKEYS) {
3506         bombout(("expected new-keys packet from server"));
3507         crReturn(0);
3508     }
3509
3510     /*
3511      * Authenticate remote host: verify host key. (We've already
3512      * checked the signature of the exchange hash.)
3513      */
3514     keystr = hostkey->fmtkey(hkey);
3515     fingerprint = hostkey->fingerprint(hkey);
3516     verify_ssh_host_key(savedhost, savedport, hostkey->keytype,
3517                         keystr, fingerprint);
3518     if (first_kex) {                   /* don't bother logging this in rekeys */
3519         logevent("Host key fingerprint is:");
3520         logevent(fingerprint);
3521     }
3522     sfree(fingerprint);
3523     sfree(keystr);
3524     hostkey->freekey(hkey);
3525
3526     /*
3527      * Send SSH2_MSG_NEWKEYS.
3528      */
3529     ssh2_pkt_init(SSH2_MSG_NEWKEYS);
3530     ssh2_pkt_send();
3531
3532     /*
3533      * Create and initialise session keys.
3534      */
3535     cscipher = cscipher_tobe;
3536     sccipher = sccipher_tobe;
3537     csmac = csmac_tobe;
3538     scmac = scmac_tobe;
3539     cscomp = cscomp_tobe;
3540     sccomp = sccomp_tobe;
3541     cscomp->compress_init();
3542     sccomp->decompress_init();
3543     /*
3544      * Set IVs after keys. Here we use the exchange hash from the
3545      * _first_ key exchange.
3546      */
3547     if (first_kex)
3548         memcpy(ssh2_session_id, exchange_hash, sizeof(exchange_hash));
3549     ssh2_mkkey(K, exchange_hash, ssh2_session_id, 'C', keyspace);
3550     cscipher->setcskey(keyspace);
3551     ssh2_mkkey(K, exchange_hash, ssh2_session_id, 'D', keyspace);
3552     sccipher->setsckey(keyspace);
3553     ssh2_mkkey(K, exchange_hash, ssh2_session_id, 'A', keyspace);
3554     cscipher->setcsiv(keyspace);
3555     ssh2_mkkey(K, exchange_hash, ssh2_session_id, 'B', keyspace);
3556     sccipher->setsciv(keyspace);
3557     ssh2_mkkey(K, exchange_hash, ssh2_session_id, 'E', keyspace);
3558     csmac->setcskey(keyspace);
3559     ssh2_mkkey(K, exchange_hash, ssh2_session_id, 'F', keyspace);
3560     scmac->setsckey(keyspace);
3561
3562     /*
3563      * If this is the first key exchange phase, we must pass the
3564      * SSH2_MSG_NEWKEYS packet to the next layer, not because it
3565      * wants to see it but because it will need time to initialise
3566      * itself before it sees an actual packet. In subsequent key
3567      * exchange phases, we don't pass SSH2_MSG_NEWKEYS on, because
3568      * it would only confuse the layer above.
3569      */
3570     if (!first_kex) {
3571         crReturn(0);
3572     }
3573     first_kex = 0;
3574
3575     /*
3576      * Now we're encrypting. Begin returning 1 to the protocol main
3577      * function so that other things can run on top of the
3578      * transport. If we ever see a KEXINIT, we must go back to the
3579      * start.
3580      */
3581     while (!(ispkt && pktin.type == SSH2_MSG_KEXINIT)) {
3582         crReturn(1);
3583     }
3584     logevent("Server initiated key re-exchange");
3585     goto begin_key_exchange;
3586
3587     crFinish(1);
3588 }
3589
3590 /*
3591  * Add data to an SSH2 channel output buffer.
3592  */
3593 static void ssh2_add_channel_data(struct ssh_channel *c, char *buf,
3594                                   int len)
3595 {
3596     bufchain_add(&c->v.v2.outbuffer, buf, len);
3597 }
3598
3599 /*
3600  * Attempt to send data on an SSH2 channel.
3601  */
3602 static int ssh2_try_send(struct ssh_channel *c)
3603 {
3604     while (c->v.v2.remwindow > 0 && bufchain_size(&c->v.v2.outbuffer) > 0) {
3605         int len;
3606         void *data;
3607         bufchain_prefix(&c->v.v2.outbuffer, &data, &len);
3608         if ((unsigned)len > c->v.v2.remwindow)
3609             len = c->v.v2.remwindow;
3610         if ((unsigned)len > c->v.v2.remmaxpkt)
3611             len = c->v.v2.remmaxpkt;
3612         ssh2_pkt_init(SSH2_MSG_CHANNEL_DATA);
3613         ssh2_pkt_adduint32(c->remoteid);
3614         ssh2_pkt_addstring_start();
3615         ssh2_pkt_addstring_data(data, len);
3616         ssh2_pkt_send();
3617         bufchain_consume(&c->v.v2.outbuffer, len);
3618         c->v.v2.remwindow -= len;
3619     }
3620
3621     /*
3622      * After having sent as much data as we can, return the amount
3623      * still buffered.
3624      */
3625     return bufchain_size(&c->v.v2.outbuffer);
3626 }
3627
3628 /*
3629  * Potentially enlarge the window on an SSH2 channel.
3630  */
3631 static void ssh2_set_window(struct ssh_channel *c, unsigned newwin)
3632 {
3633     if (newwin > c->v.v2.locwindow) {
3634         ssh2_pkt_init(SSH2_MSG_CHANNEL_WINDOW_ADJUST);
3635         ssh2_pkt_adduint32(c->remoteid);
3636         ssh2_pkt_adduint32(newwin - c->v.v2.locwindow);
3637         ssh2_pkt_send();
3638         c->v.v2.locwindow = newwin;
3639     }
3640 }
3641
3642 /*
3643  * Handle the SSH2 userauth and connection layers.
3644  */
3645 static void do_ssh2_authconn(unsigned char *in, int inlen, int ispkt)
3646 {
3647     static enum {
3648         AUTH_INVALID, AUTH_PUBLICKEY_AGENT, AUTH_PUBLICKEY_FILE,
3649         AUTH_PASSWORD,
3650         AUTH_KEYBOARD_INTERACTIVE
3651     } method;
3652     static enum {
3653         AUTH_TYPE_NONE,
3654         AUTH_TYPE_PUBLICKEY,
3655         AUTH_TYPE_PUBLICKEY_OFFER_LOUD,
3656         AUTH_TYPE_PUBLICKEY_OFFER_QUIET,
3657         AUTH_TYPE_PASSWORD,
3658         AUTH_TYPE_KEYBOARD_INTERACTIVE,
3659         AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET
3660     } type;
3661     static int gotit, need_pw, can_pubkey, can_passwd, can_keyb_inter;
3662     static int tried_pubkey_config, tried_agent, tried_keyb_inter;
3663     static int kbd_inter_running;
3664     static int we_are_in;
3665     static int num_prompts, echo;
3666     static char username[100];
3667     static char pwprompt[200];
3668     static char password[100];
3669
3670     crBegin;
3671
3672     /*
3673      * Request userauth protocol, and await a response to it.
3674      */
3675     ssh2_pkt_init(SSH2_MSG_SERVICE_REQUEST);
3676     ssh2_pkt_addstring("ssh-userauth");
3677     ssh2_pkt_send();
3678     crWaitUntilV(ispkt);
3679     if (pktin.type != SSH2_MSG_SERVICE_ACCEPT) {
3680         bombout(("Server refused user authentication protocol"));
3681         crReturnV;
3682     }
3683
3684     /*
3685      * We repeat this whole loop, including the username prompt,
3686      * until we manage a successful authentication. If the user
3687      * types the wrong _password_, they are sent back to the
3688      * beginning to try another username. (If they specify a
3689      * username in the config, they are never asked, even if they
3690      * do give a wrong password.)
3691      * 
3692      * I think this best serves the needs of
3693      * 
3694      *  - the people who have no configuration, no keys, and just
3695      *    want to try repeated (username,password) pairs until they
3696      *    type both correctly
3697      * 
3698      *  - people who have keys and configuration but occasionally
3699      *    need to fall back to passwords
3700      * 
3701      *  - people with a key held in Pageant, who might not have
3702      *    logged in to a particular machine before; so they want to
3703      *    type a username, and then _either_ their key will be
3704      *    accepted, _or_ they will type a password. If they mistype
3705      *    the username they will want to be able to get back and
3706      *    retype it!
3707      */
3708     do {
3709         static int pos;
3710         static char c;
3711
3712         /*
3713          * Get a username.
3714          */
3715         pos = 0;
3716         if ((flags & FLAG_INTERACTIVE) && !*cfg.username) {
3717             if (ssh_get_line) {
3718                 if (!ssh_get_line("login as: ",
3719                                   username, sizeof(username), FALSE)) {
3720                     /*
3721                      * get_line failed to get a username.
3722                      * Terminate.
3723                      */
3724                     logevent("No username provided. Abandoning session.");
3725                     ssh_state = SSH_STATE_CLOSED;
3726                     crReturnV;
3727                 }
3728             } else {
3729                 c_write_str("login as: ");
3730                 ssh_send_ok = 1;
3731                 while (pos >= 0) {
3732                     crWaitUntilV(!ispkt);
3733                     while (inlen--)
3734                         switch (c = *in++) {
3735                           case 10:
3736                           case 13:
3737                             username[pos] = 0;
3738                             pos = -1;
3739                             break;
3740                           case 8:
3741                           case 127:
3742                             if (pos > 0) {
3743                                 c_write_str("\b \b");
3744                                 pos--;
3745                             }
3746                             break;
3747                           case 21:
3748                           case 27:
3749                             while (pos > 0) {
3750                                 c_write_str("\b \b");
3751                                 pos--;
3752                             }
3753                             break;
3754                           case 3:
3755                           case 4:
3756                             random_save_seed();
3757                             exit(0);
3758                             break;
3759                           default:
3760                             if (((c >= ' ' && c <= '~') ||
3761                                  ((unsigned char) c >= 160))
3762                                 && pos < sizeof(username)-1) {
3763                                 username[pos++] = c;
3764                                 c_write(&c, 1);
3765                             }
3766                             break;
3767                         }
3768                 }
3769             }
3770             c_write_str("\r\n");
3771             username[strcspn(username, "\n\r")] = '\0';
3772         } else {
3773             char stuff[200];
3774             strncpy(username, cfg.username, 99);
3775             username[99] = '\0';
3776             if ((flags & FLAG_VERBOSE) || (flags & FLAG_INTERACTIVE)) {
3777                 sprintf(stuff, "Using username \"%s\".\r\n", username);
3778                 c_write_str(stuff);
3779             }
3780         }
3781
3782         /*
3783          * Send an authentication request using method "none": (a)
3784          * just in case it succeeds, and (b) so that we know what
3785          * authentication methods we can usefully try next.
3786          */
3787         ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
3788         ssh2_pkt_addstring(username);
3789         ssh2_pkt_addstring("ssh-connection");   /* service requested */
3790         ssh2_pkt_addstring("none");    /* method */
3791         ssh2_pkt_send();
3792         type = AUTH_TYPE_NONE;
3793         gotit = FALSE;
3794         we_are_in = FALSE;
3795
3796         tried_pubkey_config = FALSE;
3797         tried_agent = FALSE;
3798         tried_keyb_inter = FALSE;
3799         kbd_inter_running = FALSE;
3800
3801         while (1) {
3802             /*
3803              * Wait for the result of the last authentication request.
3804              */
3805             if (!gotit)
3806                 crWaitUntilV(ispkt);
3807             while (pktin.type == SSH2_MSG_USERAUTH_BANNER) {
3808                 char *banner;
3809                 int size;
3810                 /*
3811                  * Don't show the banner if we're operating in
3812                  * non-verbose non-interactive mode. (It's probably
3813                  * a script, which means nobody will read the
3814                  * banner _anyway_, and moreover the printing of
3815                  * the banner will screw up processing on the
3816                  * output of (say) plink.)
3817                  */
3818                 if (flags & (FLAG_VERBOSE | FLAG_INTERACTIVE)) {
3819                     ssh2_pkt_getstring(&banner, &size);
3820                     if (banner)
3821                         c_write_untrusted(banner, size);
3822                 }
3823                 crWaitUntilV(ispkt);
3824             }
3825             if (pktin.type == SSH2_MSG_USERAUTH_SUCCESS) {
3826                 logevent("Access granted");
3827                 we_are_in = TRUE;
3828                 break;
3829             }
3830
3831             if (kbd_inter_running &&
3832                 pktin.type == SSH2_MSG_USERAUTH_INFO_REQUEST) {
3833                 /*
3834                  * This is a further prompt in keyboard-interactive
3835                  * authentication. Do nothing.
3836                  */
3837             } else if (pktin.type != SSH2_MSG_USERAUTH_FAILURE) {
3838                 bombout(("Strange packet received during authentication: type %d",
3839                          pktin.type));
3840                 crReturnV;
3841             }
3842
3843             gotit = FALSE;
3844
3845             /*
3846              * OK, we're now sitting on a USERAUTH_FAILURE message, so
3847              * we can look at the string in it and know what we can
3848              * helpfully try next.
3849              */
3850             if (pktin.type == SSH2_MSG_USERAUTH_FAILURE) {
3851                 char *methods;
3852                 int methlen;
3853                 ssh2_pkt_getstring(&methods, &methlen);
3854                 kbd_inter_running = FALSE;
3855                 if (!ssh2_pkt_getbool()) {
3856                     /*
3857                      * We have received an unequivocal Access
3858                      * Denied. This can translate to a variety of
3859                      * messages:
3860                      * 
3861                      *  - if we'd just tried "none" authentication,
3862                      *    it's not worth printing anything at all
3863                      * 
3864                      *  - if we'd just tried a public key _offer_,
3865                      *    the message should be "Server refused our
3866                      *    key" (or no message at all if the key
3867                      *    came from Pageant)
3868                      * 
3869                      *  - if we'd just tried anything else, the
3870                      *    message really should be "Access denied".
3871                      * 
3872                      * Additionally, if we'd just tried password
3873                      * authentication, we should break out of this
3874                      * whole loop so as to go back to the username
3875                      * prompt.
3876                      */
3877                     if (type == AUTH_TYPE_NONE) {
3878                         /* do nothing */
3879                     } else if (type == AUTH_TYPE_PUBLICKEY_OFFER_LOUD ||
3880                                type == AUTH_TYPE_PUBLICKEY_OFFER_QUIET) {
3881                         if (type == AUTH_TYPE_PUBLICKEY_OFFER_LOUD)
3882                             c_write_str("Server refused our key\r\n");
3883                         logevent("Server refused public key");
3884                     } else if (type == AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET) {
3885                         /* server declined keyboard-interactive; ignore */
3886                     } else {
3887                         c_write_str("Access denied\r\n");
3888                         logevent("Access denied");
3889                         if (type == AUTH_TYPE_PASSWORD) {
3890                             we_are_in = FALSE;
3891                             break;
3892                         }
3893                     }
3894                 } else {
3895                     c_write_str("Further authentication required\r\n");
3896                     logevent("Further authentication required");
3897                 }
3898
3899                 can_pubkey =
3900                     in_commasep_string("publickey", methods, methlen);
3901                 can_passwd =
3902                     in_commasep_string("password", methods, methlen);
3903                 can_keyb_inter = 
3904                     in_commasep_string("keyboard-interactive", methods, methlen);
3905             }
3906
3907             method = 0;
3908
3909             if (!method && can_pubkey && agent_exists() && !tried_agent) {
3910                 /*
3911                  * Attempt public-key authentication using Pageant.
3912                  */
3913                 static unsigned char request[5], *response, *p;
3914                 static int responselen;
3915                 static int i, nkeys;
3916                 static int authed = FALSE;
3917                 void *r;
3918
3919                 tried_agent = TRUE;
3920
3921                 logevent("Pageant is running. Requesting keys.");
3922
3923                 /* Request the keys held by the agent. */
3924                 PUT_32BIT(request, 1);
3925                 request[4] = SSH2_AGENTC_REQUEST_IDENTITIES;
3926                 agent_query(request, 5, &r, &responselen);
3927                 response = (unsigned char *) r;
3928                 if (response && responselen >= 5 &&
3929                     response[4] == SSH2_AGENT_IDENTITIES_ANSWER) {
3930                     p = response + 5;
3931                     nkeys = GET_32BIT(p);
3932                     p += 4;
3933                     {
3934                         char buf[64];
3935                         sprintf(buf, "Pageant has %d SSH2 keys", nkeys);
3936                         logevent(buf);
3937                     }
3938                     for (i = 0; i < nkeys; i++) {
3939                         static char *pkblob, *alg, *commentp;
3940                         static int pklen, alglen, commentlen;
3941                         static int siglen, retlen, len;
3942                         static char *q, *agentreq, *ret;
3943                         void *vret;
3944
3945                         {
3946                             char buf[64];
3947                             sprintf(buf, "Trying Pageant key #%d", i);
3948                             logevent(buf);
3949                         }
3950                         pklen = GET_32BIT(p);
3951                         p += 4;
3952                         pkblob = p;
3953                         p += pklen;
3954                         alglen = GET_32BIT(pkblob);
3955                         alg = pkblob + 4;
3956                         commentlen = GET_32BIT(p);
3957                         p += 4;
3958                         commentp = p;
3959                         p += commentlen;
3960                         ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
3961                         ssh2_pkt_addstring(username);
3962                         ssh2_pkt_addstring("ssh-connection");   /* service requested */
3963                         ssh2_pkt_addstring("publickey");        /* method */
3964                         ssh2_pkt_addbool(FALSE);        /* no signature included */
3965                         ssh2_pkt_addstring_start();
3966                         ssh2_pkt_addstring_data(alg, alglen);
3967                         ssh2_pkt_addstring_start();
3968                         ssh2_pkt_addstring_data(pkblob, pklen);
3969                         ssh2_pkt_send();
3970
3971                         crWaitUntilV(ispkt);
3972                         if (pktin.type != SSH2_MSG_USERAUTH_PK_OK) {
3973                             logevent("Key refused");
3974                             continue;
3975                         }
3976
3977                         if (flags & FLAG_VERBOSE) {
3978                             c_write_str
3979                                 ("Authenticating with public key \"");
3980                             c_write(commentp, commentlen);
3981                             c_write_str("\" from agent\r\n");
3982                         }
3983
3984                         /*
3985                          * Server is willing to accept the key.
3986                          * Construct a SIGN_REQUEST.
3987                          */
3988                         ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
3989                         ssh2_pkt_addstring(username);
3990                         ssh2_pkt_addstring("ssh-connection");   /* service requested */
3991                         ssh2_pkt_addstring("publickey");        /* method */
3992                         ssh2_pkt_addbool(TRUE);
3993                         ssh2_pkt_addstring_start();
3994                         ssh2_pkt_addstring_data(alg, alglen);
3995                         ssh2_pkt_addstring_start();
3996                         ssh2_pkt_addstring_data(pkblob, pklen);
3997
3998                         siglen = pktout.length - 5 + 4 + 20;
3999                         len = 1;       /* message type */
4000                         len += 4 + pklen;       /* key blob */
4001                         len += 4 + siglen;      /* data to sign */
4002                         len += 4;      /* flags */
4003                         agentreq = smalloc(4 + len);
4004                         PUT_32BIT(agentreq, len);
4005                         q = agentreq + 4;
4006                         *q++ = SSH2_AGENTC_SIGN_REQUEST;
4007                         PUT_32BIT(q, pklen);
4008                         q += 4;
4009                         memcpy(q, pkblob, pklen);
4010                         q += pklen;
4011                         PUT_32BIT(q, siglen);
4012                         q += 4;
4013                         /* Now the data to be signed... */
4014                         PUT_32BIT(q, 20);
4015                         q += 4;
4016                         memcpy(q, ssh2_session_id, 20);
4017                         q += 20;
4018                         memcpy(q, pktout.data + 5, pktout.length - 5);
4019                         q += pktout.length - 5;
4020                         /* And finally the (zero) flags word. */
4021                         PUT_32BIT(q, 0);
4022                         agent_query(agentreq, len + 4, &vret, &retlen);
4023                         ret = vret;
4024                         sfree(agentreq);
4025                         if (ret) {
4026                             if (ret[4] == SSH2_AGENT_SIGN_RESPONSE) {
4027                                 logevent("Sending Pageant's response");
4028                                 ssh2_pkt_addstring_start();
4029                                 ssh2_pkt_addstring_data(ret + 9,
4030                                                         GET_32BIT(ret +
4031                                                                   5));
4032                                 ssh2_pkt_send();
4033                                 authed = TRUE;
4034                                 break;
4035                             } else {
4036                                 logevent
4037                                     ("Pageant failed to answer challenge");
4038                                 sfree(ret);
4039                             }
4040                         }
4041                     }
4042                     if (authed)
4043                         continue;
4044                 }
4045             }
4046
4047             if (!method && can_pubkey && *cfg.keyfile
4048                 && !tried_pubkey_config) {
4049                 unsigned char *pub_blob;
4050                 char *algorithm, *comment;
4051                 int pub_blob_len;
4052
4053                 tried_pubkey_config = TRUE;
4054
4055                 /*
4056                  * Try the public key supplied in the configuration.
4057                  *
4058                  * First, offer the public blob to see if the server is
4059                  * willing to accept it.
4060                  */
4061                 pub_blob = ssh2_userkey_loadpub(cfg.keyfile, &algorithm,
4062                                                 &pub_blob_len);
4063                 if (pub_blob) {
4064                     ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
4065                     ssh2_pkt_addstring(username);
4066                     ssh2_pkt_addstring("ssh-connection");       /* service requested */
4067                     ssh2_pkt_addstring("publickey");    /* method */
4068                     ssh2_pkt_addbool(FALSE);    /* no signature included */
4069                     ssh2_pkt_addstring(algorithm);
4070                     ssh2_pkt_addstring_start();
4071                     ssh2_pkt_addstring_data(pub_blob, pub_blob_len);
4072                     ssh2_pkt_send();
4073                     logevent("Offered public key");     /* FIXME */
4074
4075                     crWaitUntilV(ispkt);
4076                     if (pktin.type != SSH2_MSG_USERAUTH_PK_OK) {
4077                         gotit = TRUE;
4078                         type = AUTH_TYPE_PUBLICKEY_OFFER_LOUD;
4079                         continue;      /* key refused; give up on it */
4080                     }
4081
4082                     logevent("Offer of public key accepted");
4083                     /*
4084                      * Actually attempt a serious authentication using
4085                      * the key.
4086                      */
4087                     if (ssh2_userkey_encrypted(cfg.keyfile, &comment)) {
4088                         sprintf(pwprompt,
4089                                 "Passphrase for key \"%.100s\": ",
4090                                 comment);
4091                         need_pw = TRUE;
4092                     } else {
4093                         need_pw = FALSE;
4094                     }
4095                     c_write_str("Authenticating with public key \"");
4096                     c_write_str(comment);
4097                     c_write_str("\"\r\n");
4098                     method = AUTH_PUBLICKEY_FILE;
4099                 }
4100             }
4101
4102             if (!method && can_keyb_inter && !tried_keyb_inter) {
4103                 method = AUTH_KEYBOARD_INTERACTIVE;
4104                 type = AUTH_TYPE_KEYBOARD_INTERACTIVE;
4105                 tried_keyb_inter = TRUE;
4106
4107                 ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
4108                 ssh2_pkt_addstring(username);
4109                 ssh2_pkt_addstring("ssh-connection");   /* service requested */
4110                 ssh2_pkt_addstring("keyboard-interactive");     /* method */
4111                 ssh2_pkt_addstring(""); /* lang */
4112                 ssh2_pkt_addstring("");
4113                 ssh2_pkt_send();
4114
4115                 crWaitUntilV(ispkt);
4116                 if (pktin.type != SSH2_MSG_USERAUTH_INFO_REQUEST) {
4117                     if (pktin.type == SSH2_MSG_USERAUTH_FAILURE)
4118                         gotit = TRUE;
4119                     logevent("Keyboard-interactive authentication refused");
4120                     type = AUTH_TYPE_KEYBOARD_INTERACTIVE_QUIET;
4121                     continue;
4122                 }
4123
4124                 kbd_inter_running = TRUE;
4125             }
4126
4127             if (kbd_inter_running) {
4128                 method = AUTH_KEYBOARD_INTERACTIVE;
4129                 type = AUTH_TYPE_KEYBOARD_INTERACTIVE;
4130                 tried_keyb_inter = TRUE;
4131
4132                 /* We've got packet with that "interactive" info
4133                    dump banners, and set its prompt as ours */
4134                 {
4135                     char *name, *inst, *lang, *prompt;
4136                     int name_len, inst_len, lang_len, prompt_len;
4137                     ssh2_pkt_getstring(&name, &name_len);
4138                     ssh2_pkt_getstring(&inst, &inst_len);
4139                     ssh2_pkt_getstring(&lang, &lang_len);
4140                     if (name_len > 0)
4141                         c_write_untrusted(name, name_len);
4142                     if (inst_len > 0)
4143                         c_write_untrusted(inst, inst_len);
4144                     num_prompts = ssh2_pkt_getuint32();
4145
4146                     ssh2_pkt_getstring(&prompt, &prompt_len);
4147                     strncpy(pwprompt, prompt, sizeof(pwprompt));
4148                     need_pw = TRUE;
4149
4150                     echo = ssh2_pkt_getbool();
4151                 }
4152             }
4153
4154             if (!method && can_passwd) {
4155                 method = AUTH_PASSWORD;
4156                 sprintf(pwprompt, "%.90s@%.90s's password: ", username,
4157                         savedhost);
4158                 need_pw = TRUE;
4159             }
4160
4161             if (need_pw) {
4162                 if (ssh_get_line) {
4163                     if (!ssh_get_line(pwprompt, password,
4164                                       sizeof(password), TRUE)) {
4165                         /*
4166                          * get_line failed to get a password (for
4167                          * example because one was supplied on the
4168                          * command line which has already failed to
4169                          * work). Terminate.
4170                          */
4171                         ssh2_pkt_init(SSH2_MSG_DISCONNECT);
4172                         ssh2_pkt_adduint32(SSH2_DISCONNECT_BY_APPLICATION);
4173                         ssh2_pkt_addstring
4174                             ("No more passwords available to try");
4175                         ssh2_pkt_addstring("en");       /* language tag */
4176                         ssh2_pkt_send();
4177                         connection_fatal("Unable to authenticate");
4178                         ssh_state = SSH_STATE_CLOSED;
4179                         crReturnV;
4180                     }
4181                 } else {
4182                     static int pos = 0;
4183                     static char c;
4184
4185                     c_write_untrusted(pwprompt, strlen(pwprompt));
4186                     ssh_send_ok = 1;
4187
4188                     pos = 0;
4189                     while (pos >= 0) {
4190                         crWaitUntilV(!ispkt);
4191                         while (inlen--)
4192                             switch (c = *in++) {
4193                               case 10:
4194                               case 13:
4195                                 password[pos] = 0;
4196                                 pos = -1;
4197                                 break;
4198                               case 8:
4199                               case 127:
4200                                 if (pos > 0)
4201                                     pos--;
4202                                 break;
4203                               case 21:
4204                               case 27:
4205                                 pos = 0;
4206                                 break;
4207                               case 3:
4208                               case 4:
4209                                 random_save_seed();
4210                                 exit(0);
4211                                 break;
4212                               default:
4213                                 if (pos < sizeof(password)-1)
4214                                     password[pos++] = c;
4215                                 break;
4216                             }
4217                     }
4218                     c_write_str("\r\n");
4219                 }
4220             }
4221
4222             if (method == AUTH_PUBLICKEY_FILE) {
4223                 /*
4224                  * We have our passphrase. Now try the actual authentication.
4225                  */
4226                 struct ssh2_userkey *key;
4227
4228                 key = ssh2_load_userkey(cfg.keyfile, password);
4229                 if (key == SSH2_WRONG_PASSPHRASE || key == NULL) {
4230                     if (key == SSH2_WRONG_PASSPHRASE) {
4231                         c_write_str("Wrong passphrase\r\n");
4232                         tried_pubkey_config = FALSE;
4233                     } else {
4234                         c_write_str("Unable to load private key\r\n");
4235                         tried_pubkey_config = TRUE;
4236                     }
4237                     /* Send a spurious AUTH_NONE to return to the top. */
4238                     ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
4239                     ssh2_pkt_addstring(username);
4240                     ssh2_pkt_addstring("ssh-connection");       /* service requested */
4241                     ssh2_pkt_addstring("none"); /* method */
4242                     ssh2_pkt_send();
4243                     type = AUTH_TYPE_NONE;
4244                 } else {
4245                     unsigned char *blob, *sigdata;
4246                     int blob_len, sigdata_len;
4247
4248                     /*
4249                      * We have loaded the private key and the server
4250                      * has announced that it's willing to accept it.
4251                      * Hallelujah. Generate a signature and send it.
4252                      */
4253                     ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
4254                     ssh2_pkt_addstring(username);
4255                     ssh2_pkt_addstring("ssh-connection");       /* service requested */
4256                     ssh2_pkt_addstring("publickey");    /* method */
4257                     ssh2_pkt_addbool(TRUE);
4258                     ssh2_pkt_addstring(key->alg->name);
4259                     blob = key->alg->public_blob(key->data, &blob_len);
4260                     ssh2_pkt_addstring_start();
4261                     ssh2_pkt_addstring_data(blob, blob_len);
4262                     sfree(blob);
4263
4264                     /*
4265                      * The data to be signed is:
4266                      *
4267                      *   string  session-id
4268                      *
4269                      * followed by everything so far placed in the
4270                      * outgoing packet.
4271                      */
4272                     sigdata_len = pktout.length - 5 + 4 + 20;
4273                     sigdata = smalloc(sigdata_len);
4274                     PUT_32BIT(sigdata, 20);
4275                     memcpy(sigdata + 4, ssh2_session_id, 20);
4276                     memcpy(sigdata + 24, pktout.data + 5,
4277                            pktout.length - 5);
4278                     blob =
4279                         key->alg->sign(key->data, sigdata, sigdata_len,
4280                                        &blob_len);
4281                     ssh2_pkt_addstring_start();
4282                     ssh2_pkt_addstring_data(blob, blob_len);
4283                     sfree(blob);
4284                     sfree(sigdata);
4285
4286                     ssh2_pkt_send();
4287                     type = AUTH_TYPE_PUBLICKEY;
4288                 }
4289             } else if (method == AUTH_PASSWORD) {
4290                 /*
4291                  * We send the password packet lumped tightly together with
4292                  * an SSH_MSG_IGNORE packet. The IGNORE packet contains a
4293                  * string long enough to make the total length of the two
4294                  * packets constant. This should ensure that a passive
4295                  * listener doing traffic analyis can't work out the length
4296                  * of the password.
4297                  *
4298                  * For this to work, we need an assumption about the
4299                  * maximum length of the password packet. I think 256 is
4300                  * pretty conservative. Anyone using a password longer than
4301                  * that probably doesn't have much to worry about from
4302                  * people who find out how long their password is!
4303                  */
4304                 ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
4305                 ssh2_pkt_addstring(username);
4306                 ssh2_pkt_addstring("ssh-connection");   /* service requested */
4307                 ssh2_pkt_addstring("password");
4308                 ssh2_pkt_addbool(FALSE);
4309                 ssh2_pkt_addstring(password);
4310                 ssh2_pkt_defer();
4311                 /*
4312                  * We'll include a string that's an exact multiple of the
4313                  * cipher block size. If the cipher is NULL for some
4314                  * reason, we don't do this trick at all because we gain
4315                  * nothing by it.
4316                  */
4317                 if (cscipher) {
4318                     int stringlen, i;
4319
4320                     stringlen = (256 - deferred_len);
4321                     stringlen += cscipher->blksize - 1;
4322                     stringlen -= (stringlen % cscipher->blksize);
4323                     if (cscomp) {
4324                         /*
4325                          * Temporarily disable actual compression,
4326                          * so we can guarantee to get this string
4327                          * exactly the length we want it. The
4328                          * compression-disabling routine should
4329                          * return an integer indicating how many
4330                          * bytes we should adjust our string length
4331                          * by.
4332                          */
4333                         stringlen -= cscomp->disable_compression();
4334                     }
4335                     ssh2_pkt_init(SSH2_MSG_IGNORE);
4336                     ssh2_pkt_addstring_start();
4337                     for (i = 0; i < stringlen; i++) {
4338                         char c = (char) random_byte();
4339                         ssh2_pkt_addstring_data(&c, 1);
4340                     }
4341                     ssh2_pkt_defer();
4342                 }
4343                 ssh_pkt_defersend();
4344                 logevent("Sent password");
4345                 type = AUTH_TYPE_PASSWORD;
4346             } else if (method == AUTH_KEYBOARD_INTERACTIVE) {
4347                 ssh2_pkt_init(SSH2_MSG_USERAUTH_INFO_RESPONSE);
4348                 ssh2_pkt_adduint32(num_prompts);
4349                 ssh2_pkt_addstring(password);
4350                 memset(password, 0, sizeof(password));
4351                 ssh2_pkt_send();
4352                 type = AUTH_TYPE_KEYBOARD_INTERACTIVE;
4353             } else {
4354                 c_write_str
4355                     ("No supported authentication methods left to try!\r\n");
4356                 logevent
4357                     ("No supported authentications offered. Disconnecting");
4358                 ssh2_pkt_init(SSH2_MSG_DISCONNECT);
4359                 ssh2_pkt_adduint32(SSH2_DISCONNECT_BY_APPLICATION);
4360                 ssh2_pkt_addstring
4361                     ("No supported authentication methods available");
4362                 ssh2_pkt_addstring("en");       /* language tag */
4363                 ssh2_pkt_send();
4364                 ssh_state = SSH_STATE_CLOSED;
4365                 crReturnV;
4366             }
4367         }
4368     } while (!we_are_in);
4369
4370     /*
4371      * Now we're authenticated for the connection protocol. The
4372      * connection protocol will automatically have started at this
4373      * point; there's no need to send SERVICE_REQUEST.
4374      */
4375
4376     /*
4377      * So now create a channel with a session in it.
4378      */
4379     ssh_channels = newtree234(ssh_channelcmp);
4380     mainchan = smalloc(sizeof(struct ssh_channel));
4381     mainchan->localid = alloc_channel_id();
4382     ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN);
4383     ssh2_pkt_addstring("session");
4384     ssh2_pkt_adduint32(mainchan->localid);
4385     mainchan->v.v2.locwindow = OUR_V2_WINSIZE;
4386     ssh2_pkt_adduint32(mainchan->v.v2.locwindow);      /* our window size */
4387     ssh2_pkt_adduint32(0x4000UL);      /* our max pkt size */
4388     ssh2_pkt_send();
4389     crWaitUntilV(ispkt);
4390     if (pktin.type != SSH2_MSG_CHANNEL_OPEN_CONFIRMATION) {
4391         bombout(("Server refused to open a session"));
4392         crReturnV;
4393         /* FIXME: error data comes back in FAILURE packet */
4394     }
4395     if (ssh2_pkt_getuint32() != mainchan->localid) {
4396         bombout(("Server's channel confirmation cited wrong channel"));
4397         crReturnV;
4398     }
4399     mainchan->remoteid = ssh2_pkt_getuint32();
4400     mainchan->type = CHAN_MAINSESSION;
4401     mainchan->closes = 0;
4402     mainchan->v.v2.remwindow = ssh2_pkt_getuint32();
4403     mainchan->v.v2.remmaxpkt = ssh2_pkt_getuint32();
4404     bufchain_init(&mainchan->v.v2.outbuffer);
4405     add234(ssh_channels, mainchan);
4406     logevent("Opened channel for session");
4407
4408     /*
4409      * Potentially enable X11 forwarding.
4410      */
4411     if (cfg.x11_forward) {
4412         char proto[20], data[64];
4413         logevent("Requesting X11 forwarding");
4414         x11_invent_auth(proto, sizeof(proto), data, sizeof(data));
4415         ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
4416         ssh2_pkt_adduint32(mainchan->remoteid);
4417         ssh2_pkt_addstring("x11-req");
4418         ssh2_pkt_addbool(1);           /* want reply */
4419         ssh2_pkt_addbool(0);           /* many connections */
4420         ssh2_pkt_addstring(proto);
4421         ssh2_pkt_addstring(data);
4422         ssh2_pkt_adduint32(0);         /* screen number */
4423         ssh2_pkt_send();
4424
4425         do {
4426             crWaitUntilV(ispkt);
4427             if (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST) {
4428                 unsigned i = ssh2_pkt_getuint32();
4429                 struct ssh_channel *c;
4430                 c = find234(ssh_channels, &i, ssh_channelfind);
4431                 if (!c)
4432                     continue;          /* nonexistent channel */
4433                 c->v.v2.remwindow += ssh2_pkt_getuint32();
4434             }
4435         } while (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST);
4436
4437         if (pktin.type != SSH2_MSG_CHANNEL_SUCCESS) {
4438             if (pktin.type != SSH2_MSG_CHANNEL_FAILURE) {
4439                 bombout(("Unexpected response to X11 forwarding request:"
4440                          " packet type %d", pktin.type));
4441                 crReturnV;
4442             }
4443             logevent("X11 forwarding refused");
4444         } else {
4445             logevent("X11 forwarding enabled");
4446             ssh_X11_fwd_enabled = TRUE;
4447         }
4448     }
4449
4450     /*
4451      * Enable port forwardings.
4452      */
4453     {
4454         static char *e;                /* preserve across crReturn */
4455         char type;
4456         int n;
4457         int sport,dport;
4458         char sports[256], dports[256], host[256];
4459         char buf[1024];
4460
4461         ssh_rportfwds = newtree234(ssh_rportcmp_ssh2);
4462         /* Add port forwardings. */
4463         e = cfg.portfwd;
4464         while (*e) {
4465             type = *e++;
4466             n = 0;
4467             while (*e && *e != '\t')
4468                 sports[n++] = *e++;
4469             sports[n] = 0;
4470             if (*e == '\t')
4471                 e++;
4472             n = 0;
4473             while (*e && *e != ':')
4474                 host[n++] = *e++;
4475             host[n] = 0;
4476             if (*e == ':')
4477                 e++;
4478             n = 0;
4479             while (*e)
4480                 dports[n++] = *e++;
4481             dports[n] = 0;
4482             e++;
4483             dport = atoi(dports);
4484             sport = atoi(sports);
4485             if (sport && dport) {
4486                 if (type == 'L') {
4487                     pfd_addforward(host, dport, sport);
4488                     sprintf(buf, "Local port %d forwarding to %s:%d",
4489                             sport, host, dport);
4490                     logevent(buf);
4491                 } else {
4492                     struct ssh_rportfwd *pf;
4493                     pf = smalloc(sizeof(*pf));
4494                     strcpy(pf->dhost, host);
4495                     pf->dport = dport;
4496                     pf->sport = sport;
4497                     if (add234(ssh_rportfwds, pf) != pf) {
4498                         sprintf(buf, 
4499                                 "Duplicate remote port forwarding to %s:%d",
4500                                 host, dport);
4501                         logevent(buf);
4502                         sfree(pf);
4503                     } else {
4504                         sprintf(buf, "Requesting remote port %d (forwarded to %s:%d)",
4505                                 sport, host, dport);
4506                         logevent(buf);
4507                         ssh2_pkt_init(SSH2_MSG_GLOBAL_REQUEST);
4508                         ssh2_pkt_addstring("tcpip-forward");
4509                         ssh2_pkt_addbool(1);/* want reply */
4510                         ssh2_pkt_addstring("127.0.0.1");
4511                         ssh2_pkt_adduint32(sport);
4512                         ssh2_pkt_send();
4513
4514                         do {
4515                             crWaitUntilV(ispkt);
4516                             if (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST) {
4517                                 unsigned i = ssh2_pkt_getuint32();
4518                                 struct ssh_channel *c;
4519                                 c = find234(ssh_channels, &i, ssh_channelfind);
4520                                 if (!c)
4521                                     continue;/* nonexistent channel */
4522                                 c->v.v2.remwindow += ssh2_pkt_getuint32();
4523                             }
4524                         } while (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST);
4525
4526                         if (pktin.type != SSH2_MSG_REQUEST_SUCCESS) {
4527                             if (pktin.type != SSH2_MSG_REQUEST_FAILURE) {
4528                                 bombout(("Unexpected response to port "
4529                                          "forwarding request: packet type %d",
4530                                          pktin.type));
4531                                 crReturnV;
4532                             }
4533                             logevent("Server refused this port forwarding");
4534                         } else {
4535                             logevent("Remote port forwarding enabled");
4536                         }
4537                     }
4538                 }
4539             }
4540         }
4541     }
4542
4543     /*
4544      * Potentially enable agent forwarding.
4545      */
4546     if (cfg.agentfwd && agent_exists()) {
4547         logevent("Requesting OpenSSH-style agent forwarding");
4548         ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
4549         ssh2_pkt_adduint32(mainchan->remoteid);
4550         ssh2_pkt_addstring("auth-agent-req@openssh.com");
4551         ssh2_pkt_addbool(1);           /* want reply */
4552         ssh2_pkt_send();
4553
4554         do {
4555             crWaitUntilV(ispkt);
4556             if (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST) {
4557                 unsigned i = ssh2_pkt_getuint32();
4558                 struct ssh_channel *c;
4559                 c = find234(ssh_channels, &i, ssh_channelfind);
4560                 if (!c)
4561                     continue;          /* nonexistent channel */
4562                 c->v.v2.remwindow += ssh2_pkt_getuint32();
4563             }
4564         } while (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST);
4565
4566         if (pktin.type != SSH2_MSG_CHANNEL_SUCCESS) {
4567             if (pktin.type != SSH2_MSG_CHANNEL_FAILURE) {
4568                 bombout(("Unexpected response to agent forwarding request:"
4569                          " packet type %d", pktin.type));
4570                 crReturnV;
4571             }
4572             logevent("Agent forwarding refused");
4573         } else {
4574             logevent("Agent forwarding enabled");
4575             ssh_agentfwd_enabled = TRUE;
4576         }
4577     }
4578
4579     /*
4580      * Now allocate a pty for the session.
4581      */
4582     if (!cfg.nopty) {
4583         ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
4584         ssh2_pkt_adduint32(mainchan->remoteid); /* recipient channel */
4585         ssh2_pkt_addstring("pty-req");
4586         ssh2_pkt_addbool(1);           /* want reply */
4587         ssh2_pkt_addstring(cfg.termtype);
4588         ssh2_pkt_adduint32(cols);
4589         ssh2_pkt_adduint32(rows);
4590         ssh2_pkt_adduint32(0);         /* pixel width */
4591         ssh2_pkt_adduint32(0);         /* pixel height */
4592         ssh2_pkt_addstring_start();
4593         ssh2_pkt_addstring_data("\0", 1);       /* TTY_OP_END, no special options */
4594         ssh2_pkt_send();
4595         ssh_state = SSH_STATE_INTERMED;
4596
4597         do {
4598             crWaitUntilV(ispkt);
4599             if (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST) {
4600                 unsigned i = ssh2_pkt_getuint32();
4601                 struct ssh_channel *c;
4602                 c = find234(ssh_channels, &i, ssh_channelfind);
4603                 if (!c)
4604                     continue;          /* nonexistent channel */
4605                 c->v.v2.remwindow += ssh2_pkt_getuint32();
4606             }
4607         } while (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST);
4608
4609         if (pktin.type != SSH2_MSG_CHANNEL_SUCCESS) {
4610             if (pktin.type != SSH2_MSG_CHANNEL_FAILURE) {
4611                 bombout(("Unexpected response to pty request:"
4612                          " packet type %d", pktin.type));
4613                 crReturnV;
4614             }
4615             c_write_str("Server refused to allocate pty\r\n");
4616             ssh_editing = ssh_echoing = 1;
4617         } else {
4618             logevent("Allocated pty");
4619         }
4620     } else {
4621         ssh_editing = ssh_echoing = 1;
4622     }
4623
4624     /*
4625      * Start a shell or a remote command. We may have to attempt
4626      * this twice if the config data has provided a second choice
4627      * of command.
4628      */
4629     while (1) {
4630         int subsys;
4631         char *cmd;
4632
4633         if (ssh_fallback_cmd) {
4634             subsys = cfg.ssh_subsys2;
4635             cmd = cfg.remote_cmd_ptr2;
4636         } else {
4637             subsys = cfg.ssh_subsys;
4638             cmd = cfg.remote_cmd_ptr;
4639         }
4640
4641         ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
4642         ssh2_pkt_adduint32(mainchan->remoteid); /* recipient channel */
4643         if (subsys) {
4644             ssh2_pkt_addstring("subsystem");
4645             ssh2_pkt_addbool(1);               /* want reply */
4646             ssh2_pkt_addstring(cmd);
4647         } else if (*cmd) {
4648             ssh2_pkt_addstring("exec");
4649             ssh2_pkt_addbool(1);               /* want reply */
4650             ssh2_pkt_addstring(cmd);
4651         } else {
4652             ssh2_pkt_addstring("shell");
4653             ssh2_pkt_addbool(1);               /* want reply */
4654         }
4655         ssh2_pkt_send();
4656         do {
4657             crWaitUntilV(ispkt);
4658             if (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST) {
4659                 unsigned i = ssh2_pkt_getuint32();
4660                 struct ssh_channel *c;
4661                 c = find234(ssh_channels, &i, ssh_channelfind);
4662                 if (!c)
4663                     continue;          /* nonexistent channel */
4664                 c->v.v2.remwindow += ssh2_pkt_getuint32();
4665             }
4666         } while (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST);
4667         if (pktin.type != SSH2_MSG_CHANNEL_SUCCESS) {
4668             if (pktin.type != SSH2_MSG_CHANNEL_FAILURE) {
4669                 bombout(("Unexpected response to shell/command request:"
4670                          " packet type %d", pktin.type));
4671                 crReturnV;
4672             }
4673             /*
4674              * We failed to start the command. If this is the
4675              * fallback command, we really are finished; if it's
4676              * not, and if the fallback command exists, try falling
4677              * back to it before complaining.
4678              */
4679             if (!ssh_fallback_cmd && cfg.remote_cmd_ptr2 != NULL) {
4680                 logevent("Primary command failed; attempting fallback");
4681                 ssh_fallback_cmd = TRUE;
4682                 continue;
4683             }
4684             bombout(("Server refused to start a shell/command"));
4685             crReturnV;
4686         } else {
4687             logevent("Started a shell/command");
4688         }
4689         break;
4690     }
4691
4692     ssh_state = SSH_STATE_SESSION;
4693     if (size_needed)
4694         ssh_size();
4695     if (eof_needed)
4696         ssh_special(TS_EOF);
4697
4698     /*
4699      * Transfer data!
4700      */
4701     ldisc_send(NULL, 0, 0);            /* cause ldisc to notice changes */
4702     ssh_send_ok = 1;
4703     while (1) {
4704         static int try_send;
4705         crReturnV;
4706         try_send = FALSE;
4707         if (ispkt) {
4708             if (pktin.type == SSH2_MSG_CHANNEL_DATA ||
4709                 pktin.type == SSH2_MSG_CHANNEL_EXTENDED_DATA) {
4710                 char *data;
4711                 int length;
4712                 unsigned i = ssh2_pkt_getuint32();
4713                 struct ssh_channel *c;
4714                 c = find234(ssh_channels, &i, ssh_channelfind);
4715                 if (!c)
4716                     continue;          /* nonexistent channel */
4717                 if (pktin.type == SSH2_MSG_CHANNEL_EXTENDED_DATA &&
4718                     ssh2_pkt_getuint32() != SSH2_EXTENDED_DATA_STDERR)
4719                     continue;          /* extended but not stderr */
4720                 ssh2_pkt_getstring(&data, &length);
4721                 if (data) {
4722                     int bufsize;
4723                     c->v.v2.locwindow -= length;
4724                     switch (c->type) {
4725                       case CHAN_MAINSESSION:
4726                         bufsize =
4727                             from_backend(pktin.type ==
4728                                          SSH2_MSG_CHANNEL_EXTENDED_DATA,
4729                                          data, length);
4730                         break;
4731                       case CHAN_X11:
4732                         bufsize = x11_send(c->u.x11.s, data, length);
4733                         break;
4734                       case CHAN_SOCKDATA:
4735                         bufsize = pfd_send(c->u.pfd.s, data, length);
4736                         break;
4737                       case CHAN_AGENT:
4738                         while (length > 0) {
4739                             if (c->u.a.lensofar < 4) {
4740                                 int l = min(4 - c->u.a.lensofar, length);
4741                                 memcpy(c->u.a.msglen + c->u.a.lensofar,
4742                                        data, l);
4743                                 data += l;
4744                                 length -= l;
4745                                 c->u.a.lensofar += l;
4746                             }
4747                             if (c->u.a.lensofar == 4) {
4748                                 c->u.a.totallen =
4749                                     4 + GET_32BIT(c->u.a.msglen);
4750                                 c->u.a.message = smalloc(c->u.a.totallen);
4751                                 memcpy(c->u.a.message, c->u.a.msglen, 4);
4752                             }
4753                             if (c->u.a.lensofar >= 4 && length > 0) {
4754                                 int l =
4755                                     min(c->u.a.totallen - c->u.a.lensofar,
4756                                         length);
4757                                 memcpy(c->u.a.message + c->u.a.lensofar,
4758                                        data, l);
4759                                 data += l;
4760                                 length -= l;
4761                                 c->u.a.lensofar += l;
4762                             }
4763                             if (c->u.a.lensofar == c->u.a.totallen) {
4764                                 void *reply, *sentreply;
4765                                 int replylen;
4766                                 agent_query(c->u.a.message,
4767                                             c->u.a.totallen, &reply,
4768                                             &replylen);
4769                                 if (reply)
4770                                     sentreply = reply;
4771                                 else {
4772                                     /* Fake SSH_AGENT_FAILURE. */
4773                                     sentreply = "\0\0\0\1\5";
4774                                     replylen = 5;
4775                                 }
4776                                 ssh2_add_channel_data(c, sentreply,
4777                                                       replylen);
4778                                 try_send = TRUE;
4779                                 if (reply)
4780                                     sfree(reply);
4781                                 sfree(c->u.a.message);
4782                                 c->u.a.lensofar = 0;
4783                             }
4784                         }
4785                         bufsize = 0;
4786                         break;
4787                     }
4788                     /*
4789                      * If we are not buffering too much data,
4790                      * enlarge the window again at the remote side.
4791                      */
4792                     if (bufsize < OUR_V2_WINSIZE)
4793                         ssh2_set_window(c, OUR_V2_WINSIZE - bufsize);
4794                 }
4795             } else if (pktin.type == SSH2_MSG_DISCONNECT) {
4796                 ssh_state = SSH_STATE_CLOSED;
4797                 logevent("Received disconnect message");
4798                 crReturnV;
4799             } else if (pktin.type == SSH2_MSG_CHANNEL_REQUEST) {
4800                 continue;              /* exit status et al; ignore (FIXME?) */
4801             } else if (pktin.type == SSH2_MSG_CHANNEL_EOF) {
4802                 unsigned i = ssh2_pkt_getuint32();
4803                 struct ssh_channel *c;
4804
4805                 c = find234(ssh_channels, &i, ssh_channelfind);
4806                 if (!c)
4807                     continue;          /* nonexistent channel */
4808
4809                 if (c->type == CHAN_X11) {
4810                     /*
4811                      * Remote EOF on an X11 channel means we should
4812                      * wrap up and close the channel ourselves.
4813                      */
4814                     x11_close(c->u.x11.s);
4815                     sshfwd_close(c);
4816                 } else if (c->type == CHAN_AGENT) {
4817                     sshfwd_close(c);
4818                 } else if (c->type == CHAN_SOCKDATA) {
4819                     pfd_close(c->u.pfd.s);
4820                     sshfwd_close(c);
4821                 }
4822             } else if (pktin.type == SSH2_MSG_CHANNEL_CLOSE) {
4823                 unsigned i = ssh2_pkt_getuint32();
4824                 struct ssh_channel *c;
4825
4826                 c = find234(ssh_channels, &i, ssh_channelfind);
4827                 if (!c)
4828                     continue;          /* nonexistent channel */
4829                 /* Do pre-close processing on the channel. */
4830                 switch (c->type) {
4831                   case CHAN_MAINSESSION:
4832                     break;             /* nothing to see here, move along */
4833                   case CHAN_X11:
4834                     if (c->u.x11.s != NULL)
4835                         x11_close(c->u.x11.s);
4836                     sshfwd_close(c);
4837                     break;
4838                   case CHAN_AGENT:
4839                     sshfwd_close(c);
4840                     break;
4841                   case CHAN_SOCKDATA:
4842                     if (c->u.pfd.s != NULL)
4843                         pfd_close(c->u.pfd.s);
4844                     sshfwd_close(c);
4845                     break;
4846                 }
4847                 if (c->closes == 0) {
4848                     ssh2_pkt_init(SSH2_MSG_CHANNEL_CLOSE);
4849                     ssh2_pkt_adduint32(c->remoteid);
4850                     ssh2_pkt_send();
4851                 }
4852                 del234(ssh_channels, c);
4853                 bufchain_clear(&c->v.v2.outbuffer);
4854                 sfree(c);
4855
4856                 /*
4857                  * See if that was the last channel left open.
4858                  */
4859                 if (count234(ssh_channels) == 0) {
4860                     logevent("All channels closed. Disconnecting");
4861                     ssh2_pkt_init(SSH2_MSG_DISCONNECT);
4862                     ssh2_pkt_adduint32(SSH2_DISCONNECT_BY_APPLICATION);
4863                     ssh2_pkt_addstring("All open channels closed");
4864                     ssh2_pkt_addstring("en");   /* language tag */
4865                     ssh2_pkt_send();
4866                     ssh_state = SSH_STATE_CLOSED;
4867                     crReturnV;
4868                 }
4869                 continue;              /* remote sends close; ignore (FIXME) */
4870             } else if (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST) {
4871                 unsigned i = ssh2_pkt_getuint32();
4872                 struct ssh_channel *c;
4873                 c = find234(ssh_channels, &i, ssh_channelfind);
4874                 if (!c)
4875                     continue;          /* nonexistent channel */
4876                 c->v.v2.remwindow += ssh2_pkt_getuint32();
4877                 try_send = TRUE;
4878             } else if (pktin.type == SSH2_MSG_CHANNEL_OPEN_CONFIRMATION) {
4879                 unsigned i = ssh2_pkt_getuint32();
4880                 struct ssh_channel *c;
4881                 c = find234(ssh_channels, &i, ssh_channelfind);
4882                 if (!c)
4883                     continue;          /* nonexistent channel */
4884                 if (c->type != CHAN_SOCKDATA_DORMANT)
4885                     continue;          /* dunno why they're confirming this */
4886                 c->remoteid = ssh2_pkt_getuint32();
4887                 c->type = CHAN_SOCKDATA;
4888                 c->v.v2.remwindow = ssh2_pkt_getuint32();
4889                 c->v.v2.remmaxpkt = ssh2_pkt_getuint32();
4890                 bufchain_init(&c->v.v2.outbuffer);
4891                 if (c->u.pfd.s)
4892                     pfd_confirm(c->u.pfd.s);
4893                 if (c->closes) {
4894                     /*
4895                      * We have a pending close on this channel,
4896                      * which we decided on before the server acked
4897                      * the channel open. So now we know the
4898                      * remoteid, we can close it again.
4899                      */
4900                     ssh2_pkt_init(SSH2_MSG_CHANNEL_CLOSE);
4901                     ssh2_pkt_adduint32(c->remoteid);
4902                     ssh2_pkt_send();
4903                 }
4904             } else if (pktin.type == SSH2_MSG_CHANNEL_OPEN_FAILURE) {
4905                 unsigned i = ssh2_pkt_getuint32();
4906                 struct ssh_channel *c;
4907                 c = find234(ssh_channels, &i, ssh_channelfind);
4908                 if (!c)
4909                     continue;          /* nonexistent channel */
4910                 if (c->type != CHAN_SOCKDATA_DORMANT)
4911                     continue;          /* dunno why they're failing this */
4912
4913                 logevent("Forwarded connection refused by server");
4914
4915                 pfd_close(c->u.pfd.s);
4916
4917                 del234(ssh_channels, c);
4918                 sfree(c);
4919             } else if (pktin.type == SSH2_MSG_CHANNEL_OPEN) {
4920                 char *type;
4921                 int typelen;
4922                 char *error = NULL;
4923                 struct ssh_channel *c;
4924                 unsigned remid, winsize, pktsize;
4925                 ssh2_pkt_getstring(&type, &typelen);
4926                 c = smalloc(sizeof(struct ssh_channel));
4927
4928                 remid = ssh2_pkt_getuint32();
4929                 winsize = ssh2_pkt_getuint32();
4930                 pktsize = ssh2_pkt_getuint32();
4931
4932                 if (typelen == 3 && !memcmp(type, "x11", 3)) {
4933                     if (!ssh_X11_fwd_enabled)
4934                         error = "X11 forwarding is not enabled";
4935                     else if (x11_init(&c->u.x11.s, cfg.x11_display, c) !=
4936                              NULL) {
4937                         error = "Unable to open an X11 connection";
4938                     } else {
4939                         c->type = CHAN_X11;
4940                     }
4941                 } else if (typelen == 15 &&
4942                            !memcmp(type, "forwarded-tcpip", 15)) {
4943                     struct ssh_rportfwd pf, *realpf;
4944                     char *dummy;
4945                     int dummylen;
4946                     ssh2_pkt_getstring(&dummy, &dummylen);/* skip address */
4947                     pf.sport = ssh2_pkt_getuint32();
4948                     realpf = find234(ssh_rportfwds, &pf, NULL);
4949                     if (realpf == NULL) {
4950                         error = "Remote port is not recognised";
4951                     } else {
4952                         char *e = pfd_newconnect(&c->u.pfd.s, realpf->dhost,
4953                                                  realpf->dport, c);
4954                         char buf[1024];
4955                         sprintf(buf, "Received remote port open request for %s:%d",
4956                                 realpf->dhost, realpf->dport);
4957                         logevent(buf);
4958                         if (e != NULL) {
4959                             sprintf(buf, "Port open failed: %s", e);
4960                             logevent(buf);
4961                             error = "Port open failed";
4962                         } else {
4963                             logevent("Forwarded port opened successfully");
4964                             c->type = CHAN_SOCKDATA;
4965                         }
4966                     }
4967                 } else if (typelen == 22 &&
4968                            !memcmp(type, "auth-agent@openssh.com", 3)) {
4969                     if (!ssh_agentfwd_enabled)
4970                         error = "Agent forwarding is not enabled";
4971                     else {
4972                         c->type = CHAN_AGENT;   /* identify channel type */
4973                         c->u.a.lensofar = 0;
4974                     }
4975                 } else {
4976                     error = "Unsupported channel type requested";
4977                 }
4978
4979                 c->remoteid = remid;
4980                 if (error) {
4981                     ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN_FAILURE);
4982                     ssh2_pkt_adduint32(c->remoteid);
4983                     ssh2_pkt_adduint32(SSH2_OPEN_CONNECT_FAILED);
4984                     ssh2_pkt_addstring(error);
4985                     ssh2_pkt_addstring("en");   /* language tag */
4986                     ssh2_pkt_send();
4987                     sfree(c);
4988                 } else {
4989                     c->localid = alloc_channel_id();
4990                     c->closes = 0;
4991                     c->v.v2.locwindow = OUR_V2_WINSIZE;
4992                     c->v.v2.remwindow = winsize;
4993                     c->v.v2.remmaxpkt = pktsize;
4994                     bufchain_init(&c->v.v2.outbuffer);
4995                     add234(ssh_channels, c);
4996                     ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN_CONFIRMATION);
4997                     ssh2_pkt_adduint32(c->remoteid);
4998                     ssh2_pkt_adduint32(c->localid);
4999                     ssh2_pkt_adduint32(c->v.v2.locwindow);
5000                     ssh2_pkt_adduint32(0x4000UL);       /* our max pkt size */
5001                     ssh2_pkt_send();
5002                 }
5003             } else {
5004                 bombout(("Strange packet received: type %d", pktin.type));
5005                 crReturnV;
5006             }
5007         } else {
5008             /*
5009              * We have spare data. Add it to the channel buffer.
5010              */
5011             ssh2_add_channel_data(mainchan, in, inlen);
5012             try_send = TRUE;
5013         }
5014         if (try_send) {
5015             int i;
5016             struct ssh_channel *c;
5017             /*
5018              * Try to send data on all channels if we can.
5019              */
5020             for (i = 0; NULL != (c = index234(ssh_channels, i)); i++) {
5021                 int bufsize = ssh2_try_send(c);
5022                 if (bufsize == 0) {
5023                     switch (c->type) {
5024                       case CHAN_MAINSESSION:
5025                         /* stdin need not receive an unthrottle
5026                          * notification since it will be polled */
5027                         break;
5028                       case CHAN_X11:
5029                         x11_unthrottle(c->u.x11.s);
5030                         break;
5031                       case CHAN_AGENT:
5032                         /* agent sockets are request/response and need no
5033                          * buffer management */
5034                         break;
5035                       case CHAN_SOCKDATA:
5036                         pfd_unthrottle(c->u.pfd.s);
5037                         break;
5038                     }
5039                 }
5040             }
5041         }
5042     }
5043
5044     crFinishV;
5045 }
5046
5047 /*
5048  * Handle the top-level SSH2 protocol.
5049  */
5050 static void ssh2_protocol(unsigned char *in, int inlen, int ispkt)
5051 {
5052     if (do_ssh2_transport(in, inlen, ispkt) == 0)
5053         return;
5054     do_ssh2_authconn(in, inlen, ispkt);
5055 }
5056
5057 /*
5058  * Called to set up the connection.
5059  *
5060  * Returns an error message, or NULL on success.
5061  */
5062 static char *ssh_init(char *host, int port, char **realhost)
5063 {
5064     char *p;
5065
5066 #ifdef MSCRYPTOAPI
5067     if (crypto_startup() == 0)
5068         return "Microsoft high encryption pack not installed!";
5069 #endif
5070
5071     ssh_send_ok = 0;
5072     ssh_editing = 0;
5073     ssh_echoing = 0;
5074     ssh1_throttle_count = 0;
5075     ssh_overall_bufsize = 0;
5076     ssh_fallback_cmd = 0;
5077
5078     p = connect_to_host(host, port, realhost);
5079     if (p != NULL)
5080         return p;
5081
5082     return NULL;
5083 }
5084
5085 /*
5086  * Called to send data down the Telnet connection.
5087  */
5088 static int ssh_send(char *buf, int len)
5089 {
5090     if (s == NULL || ssh_protocol == NULL)
5091         return 0;
5092
5093     ssh_protocol(buf, len, 0);
5094
5095     return ssh_sendbuffer();
5096 }
5097
5098 /*
5099  * Called to query the current amount of buffered stdin data.
5100  */
5101 static int ssh_sendbuffer(void)
5102 {
5103     int override_value;
5104
5105     if (s == NULL || ssh_protocol == NULL)
5106         return 0;
5107
5108     /*
5109      * If the SSH socket itself has backed up, add the total backup
5110      * size on that to any individual buffer on the stdin channel.
5111      */
5112     override_value = 0;
5113     if (ssh_throttled_all)
5114         override_value = ssh_overall_bufsize;
5115
5116     if (ssh_version == 1) {
5117         return override_value;
5118     } else if (ssh_version == 2) {
5119         if (!mainchan || mainchan->closes > 0)
5120             return override_value;
5121         else
5122             return override_value + bufchain_size(&mainchan->v.v2.outbuffer);
5123     }
5124
5125     return 0;
5126 }
5127
5128 /*
5129  * Called to set the size of the window from SSH's POV.
5130  */
5131 static void ssh_size(void)
5132 {
5133     switch (ssh_state) {
5134       case SSH_STATE_BEFORE_SIZE:
5135       case SSH_STATE_PREPACKET:
5136       case SSH_STATE_CLOSED:
5137         break;                         /* do nothing */
5138       case SSH_STATE_INTERMED:
5139         size_needed = TRUE;            /* buffer for later */
5140         break;
5141       case SSH_STATE_SESSION:
5142         if (!cfg.nopty) {
5143             if (ssh_version == 1) {
5144                 send_packet(SSH1_CMSG_WINDOW_SIZE,
5145                             PKT_INT, rows, PKT_INT, cols,
5146                             PKT_INT, 0, PKT_INT, 0, PKT_END);
5147             } else {
5148                 ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
5149                 ssh2_pkt_adduint32(mainchan->remoteid);
5150                 ssh2_pkt_addstring("window-change");
5151                 ssh2_pkt_addbool(0);
5152                 ssh2_pkt_adduint32(cols);
5153                 ssh2_pkt_adduint32(rows);
5154                 ssh2_pkt_adduint32(0);
5155                 ssh2_pkt_adduint32(0);
5156                 ssh2_pkt_send();
5157             }
5158         }
5159         break;
5160     }
5161 }
5162
5163 /*
5164  * Send Telnet special codes. TS_EOF is useful for `plink', so you
5165  * can send an EOF and collect resulting output (e.g. `plink
5166  * hostname sort').
5167  */
5168 static void ssh_special(Telnet_Special code)
5169 {
5170     if (code == TS_EOF) {
5171         if (ssh_state != SSH_STATE_SESSION) {
5172             /*
5173              * Buffer the EOF in case we are pre-SESSION, so we can
5174              * send it as soon as we reach SESSION.
5175              */
5176             if (code == TS_EOF)
5177                 eof_needed = TRUE;
5178             return;
5179         }
5180         if (ssh_version == 1) {
5181             send_packet(SSH1_CMSG_EOF, PKT_END);
5182         } else {
5183             ssh2_pkt_init(SSH2_MSG_CHANNEL_EOF);
5184             ssh2_pkt_adduint32(mainchan->remoteid);
5185             ssh2_pkt_send();
5186         }
5187         logevent("Sent EOF message");
5188     } else if (code == TS_PING) {
5189         if (ssh_state == SSH_STATE_CLOSED
5190             || ssh_state == SSH_STATE_PREPACKET) return;
5191         if (ssh_version == 1) {
5192             send_packet(SSH1_MSG_IGNORE, PKT_STR, "", PKT_END);
5193         } else {
5194             ssh2_pkt_init(SSH2_MSG_IGNORE);
5195             ssh2_pkt_addstring_start();
5196             ssh2_pkt_send();
5197         }
5198     } else {
5199         /* do nothing */
5200     }
5201 }
5202
5203 void *new_sock_channel(Socket s)
5204 {
5205     struct ssh_channel *c;
5206     c = smalloc(sizeof(struct ssh_channel));
5207
5208     if (c) {
5209         c->remoteid = -1;              /* to be set when open confirmed */
5210         c->localid = alloc_channel_id();
5211         c->closes = 0;
5212         c->type = CHAN_SOCKDATA_DORMANT;/* identify channel type */
5213         c->u.pfd.s = s;
5214         add234(ssh_channels, c);
5215     }
5216     return c;
5217 }
5218
5219 /*
5220  * This is called when stdout/stderr (the entity to which
5221  * from_backend sends data) manages to clear some backlog.
5222  */
5223 void ssh_unthrottle(int bufsize)
5224 {
5225     if (ssh_version == 1) {
5226         if (ssh1_stdout_throttling && bufsize < SSH1_BUFFER_LIMIT) {
5227             ssh1_stdout_throttling = 0;
5228             ssh1_throttle(-1);
5229         }
5230     } else {
5231         if (mainchan && mainchan->closes == 0)
5232             ssh2_set_window(mainchan, OUR_V2_WINSIZE - bufsize);
5233     }
5234 }
5235
5236 void ssh_send_port_open(void *channel, char *hostname, int port, char *org)
5237 {
5238     struct ssh_channel *c = (struct ssh_channel *)channel;
5239     char buf[1024];
5240
5241     sprintf(buf, "Opening forwarded connection to %.512s:%d", hostname, port);
5242     logevent(buf);
5243
5244     if (ssh_version == 1) {
5245         send_packet(SSH1_MSG_PORT_OPEN,
5246                     PKT_INT, c->localid,
5247                     PKT_STR, hostname,
5248                     PKT_INT, port,
5249                     //PKT_STR, <org:orgport>,
5250                     PKT_END);
5251     } else {
5252         ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN);
5253         ssh2_pkt_addstring("direct-tcpip");
5254         ssh2_pkt_adduint32(c->localid);
5255         c->v.v2.locwindow = OUR_V2_WINSIZE;
5256         ssh2_pkt_adduint32(c->v.v2.locwindow);/* our window size */
5257         ssh2_pkt_adduint32(0x4000UL);      /* our max pkt size */
5258         ssh2_pkt_addstring(hostname);
5259         ssh2_pkt_adduint32(port);
5260         /*
5261          * We make up values for the originator data; partly it's
5262          * too much hassle to keep track, and partly I'm not
5263          * convinced the server should be told details like that
5264          * about my local network configuration.
5265          */
5266         ssh2_pkt_addstring("client-side-connection");
5267         ssh2_pkt_adduint32(0);
5268         ssh2_pkt_send();
5269     }
5270 }
5271
5272
5273 static Socket ssh_socket(void)
5274 {
5275     return s;
5276 }
5277
5278 static int ssh_sendok(void)
5279 {
5280     return ssh_send_ok;
5281 }
5282
5283 static int ssh_ldisc(int option)
5284 {
5285     if (option == LD_ECHO)
5286         return ssh_echoing;
5287     if (option == LD_EDIT)
5288         return ssh_editing;
5289     return FALSE;
5290 }
5291
5292 Backend ssh_backend = {
5293     ssh_init,
5294     ssh_send,
5295     ssh_sendbuffer,
5296     ssh_size,
5297     ssh_special,
5298     ssh_socket,
5299     ssh_sendok,
5300     ssh_ldisc,
5301     ssh_unthrottle,
5302     22
5303 };