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