ssh.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <stdarg.h>
   4 #include <assert.h>
   5 #include <winsock.h>
   6
   7 #include "putty.h"
   8 #include "tree234.h"
   9 #include "ssh.h"
  10 #include "scp.h"
  11
  12 #ifndef FALSE
  13 #define FALSE 0
  14 #endif
  15 #ifndef TRUE
  16 #define TRUE 1
  17 #endif
  18
  19 #define logevent(s) { logevent(s); \
  20                       if (!(flags & FLAG_CONNECTION) && (flags & FLAG_VERBOSE)) \
  21                       fprintf(stderr, "%s\n", s); }
  22
  23 #define SSH1_MSG_DISCONNECT     1
  24 #define SSH1_SMSG_PUBLIC_KEY    2
  25 #define SSH1_CMSG_SESSION_KEY   3
  26 #define SSH1_CMSG_USER          4
  27 #define SSH1_CMSG_AUTH_RSA      6
  28 #define SSH1_SMSG_AUTH_RSA_CHALLENGE 7
  29 #define SSH1_CMSG_AUTH_RSA_RESPONSE 8
  30 #define SSH1_CMSG_AUTH_PASSWORD 9
  31 #define SSH1_CMSG_REQUEST_PTY   10
  32 #define SSH1_CMSG_WINDOW_SIZE   11
  33 #define SSH1_CMSG_EXEC_SHELL    12
  34 #define SSH1_CMSG_EXEC_CMD      13
  35 #define SSH1_SMSG_SUCCESS       14
  36 #define SSH1_SMSG_FAILURE       15
  37 #define SSH1_CMSG_STDIN_DATA    16
  38 #define SSH1_SMSG_STDOUT_DATA   17
  39 #define SSH1_SMSG_STDERR_DATA   18
  40 #define SSH1_CMSG_EOF           19
  41 #define SSH1_SMSG_EXIT_STATUS   20
  42 #define SSH1_MSG_CHANNEL_OPEN_CONFIRMATION  21
  43 #define SSH1_MSG_CHANNEL_OPEN_FAILURE   22
  44 #define SSH1_MSG_CHANNEL_DATA   23
  45 #define SSH1_MSG_CHANNEL_CLOSE  24
  46 #define SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION 25
  47 #define SSH1_CMSG_AGENT_REQUEST_FORWARDING  30
  48 #define SSH1_SMSG_AGENT_OPEN    31
  49 #define SSH1_CMSG_EXIT_CONFIRMATION     33
  50 #define SSH1_MSG_IGNORE         32
  51 #define SSH1_MSG_DEBUG          36
  52 #define SSH1_CMSG_AUTH_TIS      39
  53 #define SSH1_SMSG_AUTH_TIS_CHALLENGE    40
  54 #define SSH1_CMSG_AUTH_TIS_RESPONSE     41
  55 #define SSH1_CMSG_AUTH_CCARD    70
  56 #define SSH1_SMSG_AUTH_CCARD_CHALLENGE  71
  57 #define SSH1_CMSG_AUTH_CCARD_RESPONSE   72
  58
  59 #define SSH1_AUTH_TIS           5
  60 #define SSH1_AUTH_CCARD         16
  61
  62 #define SSH_AGENTC_REQUEST_RSA_IDENTITIES    1
  63 #define SSH_AGENT_RSA_IDENTITIES_ANSWER      2
  64 #define SSH_AGENTC_RSA_CHALLENGE             3
  65 #define SSH_AGENT_RSA_RESPONSE               4
  66 #define SSH_AGENT_FAILURE                    5
  67 #define SSH_AGENT_SUCCESS                    6
  68 #define SSH_AGENTC_ADD_RSA_IDENTITY          7
  69 #define SSH_AGENTC_REMOVE_RSA_IDENTITY       8
  70
  71 #define SSH2_MSG_DISCONNECT             1
  72 #define SSH2_MSG_IGNORE                 2
  73 #define SSH2_MSG_UNIMPLEMENTED          3
  74 #define SSH2_MSG_DEBUG                  4
  75 #define SSH2_MSG_SERVICE_REQUEST        5
  76 #define SSH2_MSG_SERVICE_ACCEPT         6
  77 #define SSH2_MSG_KEXINIT                20
  78 #define SSH2_MSG_NEWKEYS                21
  79 #define SSH2_MSG_KEXDH_INIT             30
  80 #define SSH2_MSG_KEXDH_REPLY            31
  81 #define SSH2_MSG_USERAUTH_REQUEST            50
  82 #define SSH2_MSG_USERAUTH_FAILURE            51
  83 #define SSH2_MSG_USERAUTH_SUCCESS            52
  84 #define SSH2_MSG_USERAUTH_BANNER             53
  85 #define SSH2_MSG_USERAUTH_PK_OK              60
  86 #define SSH2_MSG_USERAUTH_PASSWD_CHANGEREQ   60
  87 #define SSH2_MSG_GLOBAL_REQUEST                  80
  88 #define SSH2_MSG_REQUEST_SUCCESS                 81
  89 #define SSH2_MSG_REQUEST_FAILURE                 82
  90 #define SSH2_MSG_CHANNEL_OPEN                    90
  91 #define SSH2_MSG_CHANNEL_OPEN_CONFIRMATION       91
  92 #define SSH2_MSG_CHANNEL_OPEN_FAILURE            92
  93 #define SSH2_MSG_CHANNEL_WINDOW_ADJUST           93
  94 #define SSH2_MSG_CHANNEL_DATA                    94
  95 #define SSH2_MSG_CHANNEL_EXTENDED_DATA           95
  96 #define SSH2_MSG_CHANNEL_EOF                     96
  97 #define SSH2_MSG_CHANNEL_CLOSE                   97
  98 #define SSH2_MSG_CHANNEL_REQUEST                 98
  99 #define SSH2_MSG_CHANNEL_SUCCESS                 99
 100 #define SSH2_MSG_CHANNEL_FAILURE                 100
 101
 102 #define SSH2_OPEN_ADMINISTRATIVELY_PROHIBITED    1
 103 #define SSH2_OPEN_CONNECT_FAILED                 2
 104 #define SSH2_OPEN_UNKNOWN_CHANNEL_TYPE           3
 105 #define SSH2_OPEN_RESOURCE_SHORTAGE              4
 106 #define SSH2_EXTENDED_DATA_STDERR                1
 107
 108 #define GET_32BIT(cp) \
 109     (((unsigned long)(unsigned char)(cp)[0] << 24) | \
 110     ((unsigned long)(unsigned char)(cp)[1] << 16) | \
 111     ((unsigned long)(unsigned char)(cp)[2] << 8) | \
 112     ((unsigned long)(unsigned char)(cp)[3]))
 113
 114 #define PUT_32BIT(cp, value) { \
 115     (cp)[0] = (unsigned char)((value) >> 24); \
 116     (cp)[1] = (unsigned char)((value) >> 16); \
 117     (cp)[2] = (unsigned char)((value) >> 8); \
 118     (cp)[3] = (unsigned char)(value); }
 119
 120 enum { PKT_END, PKT_INT, PKT_CHAR, PKT_DATA, PKT_STR, PKT_BIGNUM };
 121
 122 /* Coroutine mechanics for the sillier bits of the code */
 123 #define crBegin1        static int crLine = 0;
 124 #define crBegin2        switch(crLine) { case 0:;
 125 #define crBegin         crBegin1; crBegin2;
 126 #define crFinish(z)     } crLine = 0; return (z)
 127 #define crFinishV       } crLine = 0; return
 128 #define crReturn(z)     \
 129         do {\
 130             crLine=__LINE__; return (z); case __LINE__:;\
 131         } while (0)
 132 #define crReturnV       \
 133         do {\
 134             crLine=__LINE__; return; case __LINE__:;\
 135         } while (0)
 136 #define crStop(z)       do{ crLine = 0; return (z); }while(0)
 137 #define crStopV         do{ crLine = 0; return; }while(0)
 138 #define crWaitUntil(c)  do { crReturn(0); } while (!(c))
 139 #define crWaitUntilV(c) do { crReturnV; } while (!(c))
 140
 141 extern struct ssh_cipher ssh_3des;
 142 extern struct ssh_cipher ssh_3des_ssh2;
 143 extern struct ssh_cipher ssh_des;
 144 extern struct ssh_cipher ssh_blowfish;
 145
 146 /* for ssh 2; we miss out single-DES because it isn't supported */
 147 struct ssh_cipher *ciphers[] = { &ssh_3des_ssh2, &ssh_blowfish };
 148
 149 extern struct ssh_kex ssh_diffiehellman;
 150 struct ssh_kex *kex_algs[] = { &ssh_diffiehellman };
 151
 152 extern struct ssh_hostkey ssh_dss;
 153 struct ssh_hostkey *hostkey_algs[] = { &ssh_dss };
 154
 155 extern struct ssh_mac ssh_sha1;
 156
 157 static SHA_State exhash;
 158
 159 static void nullmac_key(unsigned char *key) { }
 160 static void nullmac_generate(unsigned char *blk, int len, unsigned long seq) { }
 161 static int nullmac_verify(unsigned char *blk, int len, unsigned long seq) { return 1; }
 162 struct ssh_mac ssh_mac_none = {
 163     nullmac_key, nullmac_key, nullmac_generate, nullmac_verify, "none", 0
 164 };
 165 struct ssh_mac *macs[] = { &ssh_sha1, &ssh_mac_none };
 166
 167 struct ssh_compress ssh_comp_none = {
 168     "none"
 169 };
 170 struct ssh_compress *compressions[] = { &ssh_comp_none };
 171
 172 static SOCKET s = INVALID_SOCKET;
 173
 174 static unsigned char session_key[32];
 175 static struct ssh_cipher *cipher = NULL;
 176 static struct ssh_cipher *cscipher = NULL;
 177 static struct ssh_cipher *sccipher = NULL;
 178 static struct ssh_mac *csmac = NULL;
 179 static struct ssh_mac *scmac = NULL;
 180 static struct ssh_compress *cscomp = NULL;
 181 static struct ssh_compress *sccomp = NULL;
 182 static struct ssh_kex *kex = NULL;
 183 static struct ssh_hostkey *hostkey = NULL;
 184 int (*ssh_get_password)(const char *prompt, char *str, int maxlen) = NULL;
 185
 186 static char *savedhost;
 187 static int ssh_send_ok;
 188
 189 /*
 190  * 2-3-4 tree storing channels.
 191  */
 192 struct ssh_channel {
 193     int remoteid, localid;
 194     int type;
 195     int closes;
 196     union {
 197         struct ssh_agent_channel {
 198             unsigned char *message;
 199             unsigned char msglen[4];
 200             int lensofar, totallen;
 201         } a;
 202     } u;
 203 };
 204 static tree234 *ssh_channels;           /* indexed by local id */
 205 static int ssh_channelcmp(void *av, void *bv) {
 206     struct ssh_channel *a = (struct ssh_channel *)av;
 207     struct ssh_channel *b = (struct ssh_channel *)bv;
 208     if (a->localid < b->localid) return -1;
 209     if (a->localid > b->localid) return +1;
 210     return 0;
 211 }
 212 static int ssh_channelfind(void *av, void *bv) {
 213     int *a = (int *)av;
 214     struct ssh_channel *b = (struct ssh_channel *)bv;
 215     if (*a < b->localid) return -1;
 216     if (*a > b->localid) return +1;
 217     return 0;
 218 }
 219
 220 static enum {
 221     SSH_STATE_BEFORE_SIZE,
 222     SSH_STATE_INTERMED,
 223     SSH_STATE_SESSION,
 224     SSH_STATE_CLOSED
 225 } ssh_state = SSH_STATE_BEFORE_SIZE;
 226
 227 static int size_needed = FALSE;
 228
 229 static void s_write (char *buf, int len) {
 230     while (len > 0) {
 231         int i = send (s, buf, len, 0);
 232         noise_ultralight(i);
 233         if (i <= 0)
 234             fatalbox("Lost connection while sending");
 235         if (i > 0)
 236             len -= i, buf += i;
 237     }
 238 }
 239
 240 static int s_read (char *buf, int len) {
 241     int ret = 0;
 242     while (len > 0) {
 243         int i = recv (s, buf, len, 0);
 244         noise_ultralight(i);
 245         if (i > 0)
 246             len -= i, buf += i, ret += i;
 247         else
 248             return i;
 249     }
 250     return ret;
 251 }
 252
 253 static void c_write (char *buf, int len) {
 254     if (!(flags & FLAG_CONNECTION)) {
 255         int i;
 256         for (i = 0; i < len; i++)
 257             if (buf[i] != '\r')
 258                 fputc(buf[i], stderr);
 259         return;
 260     }
 261     while (len--)
 262         c_write1(*buf++);
 263 }
 264
 265 struct Packet {
 266     long length;
 267     int type;
 268     unsigned char *data;
 269     unsigned char *body;
 270     long savedpos;
 271     long maxlen;
 272 };
 273
 274 static struct Packet pktin = { 0, 0, NULL, NULL, 0 };
 275 static struct Packet pktout = { 0, 0, NULL, NULL, 0 };
 276
 277 static int ssh_version;
 278 static void (*ssh_protocol)(unsigned char *in, int inlen, int ispkt);
 279 static void ssh1_protocol(unsigned char *in, int inlen, int ispkt);
 280 static void ssh2_protocol(unsigned char *in, int inlen, int ispkt);
 281 static void ssh_size(void);
 282
 283 static int (*s_rdpkt)(unsigned char **data, int *datalen);
 284
 285 /*
 286  * Collect incoming data in the incoming packet buffer.
 287  * Decipher and verify the packet when it is completely read.
 288  * Drop SSH1_MSG_DEBUG and SSH1_MSG_IGNORE packets.
 289  * Update the *data and *datalen variables.
 290  * Return the additional nr of bytes needed, or 0 when
 291  * a complete packet is available.
 292  */
 293 static int ssh1_rdpkt(unsigned char **data, int *datalen)
 294 {
 295     static long len, pad, biglen, to_read;
 296     static unsigned long realcrc, gotcrc;
 297     static unsigned char *p;
 298     static int i;
 299
 300     crBegin;
 301
 302 next_packet:
 303
 304     pktin.type = 0;
 305     pktin.length = 0;
 306
 307     for (i = len = 0; i < 4; i++) {
 308         while ((*datalen) == 0)
 309             crReturn(4-i);
 310         len = (len << 8) + **data;
 311         (*data)++, (*datalen)--;
 312     }
 313
 314 #ifdef FWHACK
 315     if (len == 0x52656d6f) {       /* "Remo"te server has closed ... */
 316         len = 0x300;               /* big enough to carry to end */
 317     }
 318 #endif
 319
 320     pad = 8 - (len % 8);
 321     biglen = len + pad;
 322     pktin.length = len - 5;
 323
 324     if (pktin.maxlen < biglen) {
 325         pktin.maxlen = biglen;
 326         pktin.data = (pktin.data == NULL ? malloc(biglen+APIEXTRA) :
 327                       realloc(pktin.data, biglen+APIEXTRA));
 328         if (!pktin.data)
 329             fatalbox("Out of memory");
 330     }
 331
 332     to_read = biglen;
 333     p = pktin.data;
 334     while (to_read > 0) {
 335         static int chunk;
 336         chunk = to_read;
 337         while ((*datalen) == 0)
 338             crReturn(to_read);
 339         if (chunk > (*datalen))
 340             chunk = (*datalen);
 341         memcpy(p, *data, chunk);
 342         *data += chunk;
 343         *datalen -= chunk;
 344         p += chunk;
 345         to_read -= chunk;
 346     }
 347
 348     if (cipher)
 349         cipher->decrypt(pktin.data, biglen);
 350
 351     pktin.type = pktin.data[pad];
 352     pktin.body = pktin.data + pad + 1;
 353
 354     realcrc = crc32(pktin.data, biglen-4);
 355     gotcrc = GET_32BIT(pktin.data+biglen-4);
 356     if (gotcrc != realcrc) {
 357         fatalbox("Incorrect CRC received on packet");
 358     }
 359
 360     if (pktin.type == SSH1_SMSG_STDOUT_DATA ||
 361         pktin.type == SSH1_SMSG_STDERR_DATA ||
 362         pktin.type == SSH1_MSG_DEBUG ||
 363         pktin.type == SSH1_SMSG_AUTH_TIS_CHALLENGE ||
 364         pktin.type == SSH1_SMSG_AUTH_CCARD_CHALLENGE) {
 365         long strlen = GET_32BIT(pktin.body);
 366         if (strlen + 4 != pktin.length)
 367             fatalbox("Received data packet with bogus string length");
 368     }
 369
 370     if (pktin.type == SSH1_MSG_DEBUG) {
 371         /* log debug message */
 372         char buf[80];
 373         int strlen = GET_32BIT(pktin.body);
 374         strcpy(buf, "Remote: ");
 375         if (strlen > 70) strlen = 70;
 376         memcpy(buf+8, pktin.body+4, strlen);
 377         buf[8+strlen] = '\0';
 378         logevent(buf);
 379         goto next_packet;
 380     } else if (pktin.type == SSH1_MSG_IGNORE) {
 381         /* do nothing */
 382         goto next_packet;
 383     }
 384
 385     crFinish(0);
 386 }
 387
 388 static int ssh2_rdpkt(unsigned char **data, int *datalen)
 389 {
 390     static long len, pad, payload, packetlen, maclen;
 391     static int i;
 392     static int cipherblk;
 393     static unsigned long incoming_sequence = 0;
 394
 395     crBegin;
 396
 397 next_packet:
 398
 399     pktin.type = 0;
 400     pktin.length = 0;
 401
 402     if (cipher)
 403         cipherblk = cipher->blksize;
 404     else
 405         cipherblk = 8;
 406     if (cipherblk < 8)
 407         cipherblk = 8;
 408
 409     if (pktin.maxlen < cipherblk) {
 410         pktin.maxlen = cipherblk;
 411         pktin.data = (pktin.data == NULL ? malloc(cipherblk+APIEXTRA) :
 412                       realloc(pktin.data, cipherblk+APIEXTRA));
 413         if (!pktin.data)
 414             fatalbox("Out of memory");
 415     }
 416
 417     /*
 418      * Acquire and decrypt the first block of the packet. This will
 419      * contain the length and padding details.
 420      */
 421      for (i = len = 0; i < cipherblk; i++) {
 422         while ((*datalen) == 0)
 423             crReturn(cipherblk-i);
 424         pktin.data[i] = *(*data)++;
 425         (*datalen)--;
 426     }
 427 #ifdef FWHACK
 428     if (!memcmp(pktin.data, "Remo", 4)) {/* "Remo"te server has closed ... */
 429         /* FIXME */
 430     }
 431 #endif
 432     if (sccipher)
 433         sccipher->decrypt(pktin.data, cipherblk);
 434
 435     /*
 436      * Now get the length and padding figures.
 437      */
 438     len = GET_32BIT(pktin.data);
 439     pad = pktin.data[4];
 440
 441     /*
 442      * This enables us to deduce the payload length.
 443      */
 444     payload = len - pad - 1;
 445
 446     pktin.length = payload + 5;
 447
 448     /*
 449      * So now we can work out the total packet length.
 450      */
 451     packetlen = len + 4;
 452     maclen = scmac ? scmac->len : 0;
 453
 454     /*
 455      * Adjust memory allocation if packet is too big.
 456      */
 457     if (pktin.maxlen < packetlen) {
 458         pktin.maxlen = packetlen;
 459         pktin.data = (pktin.data == NULL ? malloc(packetlen+APIEXTRA) :
 460                       realloc(pktin.data, packetlen+APIEXTRA));
 461         if (!pktin.data)
 462             fatalbox("Out of memory");
 463     }
 464
 465     /*
 466      * Read and decrypt the remainder of the packet.
 467      */
 468     for (i = cipherblk; i < packetlen + maclen; i++) {
 469         while ((*datalen) == 0)
 470             crReturn(packetlen + maclen - i);
 471         pktin.data[i] = *(*data)++;
 472         (*datalen)--;
 473     }
 474     /* Decrypt everything _except_ the MAC. */
 475     if (sccipher)
 476         sccipher->decrypt(pktin.data + cipherblk, packetlen - cipherblk);
 477
 478 #if 0
 479     debug(("Got packet len=%d pad=%d\r\n", len, pad));
 480     for (i = 0; i < packetlen; i++)
 481         debug(("  %02x", (unsigned char)pktin.data[i]));
 482     debug(("\r\n"));
 483 #endif
 484
 485     /*
 486      * Check the MAC.
 487      */
 488     if (scmac && !scmac->verify(pktin.data, len+4, incoming_sequence))
 489         fatalbox("Incorrect MAC received on packet");
 490     incoming_sequence++;               /* whether or not we MACed */
 491
 492     pktin.savedpos = 6;
 493     pktin.type = pktin.data[5];
 494
 495     if (pktin.type == SSH2_MSG_IGNORE || pktin.type == SSH2_MSG_DEBUG)
 496         goto next_packet;              /* FIXME: print DEBUG message */
 497
 498     crFinish(0);
 499 }
 500
 501 static void ssh_gotdata(unsigned char *data, int datalen)
 502 {
 503     while (datalen > 0) {
 504         if ( s_rdpkt(&data, &datalen) == 0 ) {
 505             ssh_protocol(NULL, 0, 1);
 506             if (ssh_state == SSH_STATE_CLOSED) {
 507                 return;
 508             }
 509         }
 510     }
 511 }
 512
 513
 514 static void s_wrpkt_start(int type, int len) {
 515     int pad, biglen;
 516
 517     len += 5;                          /* type and CRC */
 518     pad = 8 - (len%8);
 519     biglen = len + pad;
 520
 521     pktout.length = len-5;
 522     if (pktout.maxlen < biglen) {
 523         pktout.maxlen = biglen;
 524 #ifdef MSCRYPTOAPI
 525         /* Allocate enough buffer space for extra block
 526          * for MS CryptEncrypt() */
 527         pktout.data = (pktout.data == NULL ? malloc(biglen+12) :
 528                        realloc(pktout.data, biglen+12));
 529 #else
 530         pktout.data = (pktout.data == NULL ? malloc(biglen+4) :
 531                        realloc(pktout.data, biglen+4));
 532 #endif
 533         if (!pktout.data)
 534             fatalbox("Out of memory");
 535     }
 536
 537     pktout.type = type;
 538     pktout.body = pktout.data+4+pad+1;
 539 }
 540
 541 static void s_wrpkt(void) {
 542     int pad, len, biglen, i;
 543     unsigned long crc;
 544
 545     len = pktout.length + 5;           /* type and CRC */
 546     pad = 8 - (len%8);
 547     biglen = len + pad;
 548
 549     pktout.body[-1] = pktout.type;
 550     for (i=0; i<pad; i++)
 551         pktout.data[i+4] = random_byte();
 552     crc = crc32(pktout.data+4, biglen-4);
 553     PUT_32BIT(pktout.data+biglen, crc);
 554     PUT_32BIT(pktout.data, len);
 555
 556     if (cipher)
 557         cipher->encrypt(pktout.data+4, biglen);
 558
 559     s_write(pktout.data, biglen+4);
 560 }
 561
 562 /*
 563  * Construct a packet with the specified contents and
 564  * send it to the server.
 565  */
 566 static void send_packet(int pkttype, ...)
 567 {
 568     va_list args;
 569     unsigned char *p, *argp, argchar;
 570     unsigned long argint;
 571     int pktlen, argtype, arglen;
 572     Bignum bn;
 573
 574     pktlen = 0;
 575     va_start(args, pkttype);
 576     while ((argtype = va_arg(args, int)) != PKT_END) {
 577         switch (argtype) {
 578           case PKT_INT:
 579             (void) va_arg(args, int);
 580             pktlen += 4;
 581             break;
 582           case PKT_CHAR:
 583             (void) va_arg(args, char);
 584             pktlen++;
 585             break;
 586           case PKT_DATA:
 587             (void) va_arg(args, unsigned char *);
 588             arglen = va_arg(args, int);
 589             pktlen += arglen;
 590             break;
 591           case PKT_STR:
 592             argp = va_arg(args, unsigned char *);
 593             arglen = strlen(argp);
 594             pktlen += 4 + arglen;
 595             break;
 596           case PKT_BIGNUM:
 597             bn = va_arg(args, Bignum);
 598             pktlen += ssh1_bignum_length(bn);
 599             break;
 600           default:
 601             assert(0);
 602         }
 603     }
 604     va_end(args);
 605
 606     s_wrpkt_start(pkttype, pktlen);
 607     p = pktout.body;
 608
 609     va_start(args, pkttype);
 610     while ((argtype = va_arg(args, int)) != PKT_END) {
 611         switch (argtype) {
 612           case PKT_INT:
 613             argint = va_arg(args, int);
 614             PUT_32BIT(p, argint);
 615             p += 4;
 616             break;
 617           case PKT_CHAR:
 618             argchar = va_arg(args, unsigned char);
 619             *p = argchar;
 620             p++;
 621             break;
 622           case PKT_DATA:
 623             argp = va_arg(args, unsigned char *);
 624             arglen = va_arg(args, int);
 625             memcpy(p, argp, arglen);
 626             p += arglen;
 627             break;
 628           case PKT_STR:
 629             argp = va_arg(args, unsigned char *);
 630             arglen = strlen(argp);
 631             PUT_32BIT(p, arglen);
 632             memcpy(p + 4, argp, arglen);
 633             p += 4 + arglen;
 634             break;
 635           case PKT_BIGNUM:
 636             bn = va_arg(args, Bignum);
 637             p += ssh1_write_bignum(p, bn);
 638             break;
 639         }
 640     }
 641     va_end(args);
 642
 643     s_wrpkt();
 644 }
 645
 646
 647 /*
 648  * Connect to specified host and port.
 649  * Returns an error message, or NULL on success.
 650  * Also places the canonical host name into `realhost'.
 651  */
 652 static char *connect_to_host(char *host, int port, char **realhost)
 653 {
 654     SOCKADDR_IN addr;
 655     struct hostent *h;
 656     unsigned long a;
 657 #ifdef FWHACK
 658     char *FWhost;
 659     int FWport;
 660 #endif
 661
 662     savedhost = malloc(1+strlen(host));
 663     if (!savedhost)
 664         fatalbox("Out of memory");
 665     strcpy(savedhost, host);
 666
 667     if (port < 0)
 668         port = 22;                     /* default ssh port */
 669
 670 #ifdef FWHACK
 671     FWhost = host;
 672     FWport = port;
 673     host = FWSTR;
 674     port = 23;
 675 #endif
 676
 677     /*
 678      * Try to find host.
 679      */
 680     if ( (a = inet_addr(host)) == (unsigned long) INADDR_NONE) {
 681         if ( (h = gethostbyname(host)) == NULL)
 682             switch (WSAGetLastError()) {
 683               case WSAENETDOWN: return "Network is down";
 684               case WSAHOST_NOT_FOUND: case WSANO_DATA:
 685                 return "Host does not exist";
 686               case WSATRY_AGAIN: return "Host not found";
 687               default: return "gethostbyname: unknown error";
 688             }
 689         memcpy (&a, h->h_addr, sizeof(a));
 690         *realhost = h->h_name;
 691     } else
 692         *realhost = host;
 693 #ifdef FWHACK
 694     *realhost = FWhost;
 695 #endif
 696     a = ntohl(a);
 697
 698     /*
 699      * Open socket.
 700      */
 701     s = socket(AF_INET, SOCK_STREAM, 0);
 702     if (s == INVALID_SOCKET)
 703         switch (WSAGetLastError()) {
 704           case WSAENETDOWN: return "Network is down";
 705           case WSAEAFNOSUPPORT: return "TCP/IP support not present";
 706           default: return "socket(): unknown error";
 707         }
 708
 709     /*
 710      * Bind to local address.
 711      */
 712     addr.sin_family = AF_INET;
 713     addr.sin_addr.s_addr = htonl(INADDR_ANY);
 714     addr.sin_port = htons(0);
 715     if (bind (s, (struct sockaddr *)&addr, sizeof(addr)) == SOCKET_ERROR)
 716         switch (WSAGetLastError()) {
 717           case WSAENETDOWN: return "Network is down";
 718           default: return "bind(): unknown error";
 719         }
 720
 721     /*
 722      * Connect to remote address.
 723      */
 724     addr.sin_addr.s_addr = htonl(a);
 725     addr.sin_port = htons((short)port);
 726     if (connect (s, (struct sockaddr *)&addr, sizeof(addr)) == SOCKET_ERROR)
 727         switch (WSAGetLastError()) {
 728           case WSAENETDOWN: return "Network is down";
 729           case WSAECONNREFUSED: return "Connection refused";
 730           case WSAENETUNREACH: return "Network is unreachable";
 731           case WSAEHOSTUNREACH: return "No route to host";
 732           default: return "connect(): unknown error";
 733         }
 734
 735 #ifdef FWHACK
 736     send(s, "connect ", 8, 0);
 737     send(s, FWhost, strlen(FWhost), 0);
 738     {
 739         char buf[20];
 740         sprintf(buf, " %d\n", FWport);
 741         send (s, buf, strlen(buf), 0);
 742     }
 743 #endif
 744
 745     return NULL;
 746 }
 747
 748 static int ssh_versioncmp(char *a, char *b) {
 749     char *ae, *be;
 750     unsigned long av, bv;
 751
 752     av = strtoul(a, &ae, 10);
 753     bv = strtoul(b, &be, 10);
 754     if (av != bv) return (av < bv ? -1 : +1);
 755     if (*ae == '.') ae++;
 756     if (*be == '.') be++;
 757     av = strtoul(ae, &ae, 10);
 758     bv = strtoul(be, &be, 10);
 759     if (av != bv) return (av < bv ? -1 : +1);
 760     return 0;
 761 }
 762
 763
 764 /*
 765  * Utility routine for putting an SSH-protocol `string' into a SHA
 766  * state.
 767  */
 768 #include <stdio.h>
 769 void sha_string(SHA_State *s, void *str, int len) {
 770     unsigned char lenblk[4];
 771     PUT_32BIT(lenblk, len);
 772     SHA_Bytes(s, lenblk, 4);
 773     SHA_Bytes(s, str, len);
 774 }
 775
 776 /*
 777  * SSH2 packet construction functions.
 778  */
 779 void ssh2_pkt_adddata(void *data, int len) {
 780     pktout.length += len;
 781     if (pktout.maxlen < pktout.length) {
 782         pktout.maxlen = pktout.length + 256;
 783         pktout.data = (pktout.data == NULL ? malloc(pktout.maxlen+APIEXTRA) :
 784                        realloc(pktout.data, pktout.maxlen+APIEXTRA));
 785         if (!pktout.data)
 786             fatalbox("Out of memory");
 787     }
 788     memcpy(pktout.data+pktout.length-len, data, len);
 789 }
 790 void ssh2_pkt_addbyte(unsigned char byte) {
 791     ssh2_pkt_adddata(&byte, 1);
 792 }
 793 void ssh2_pkt_init(int pkt_type) {
 794     pktout.length = 5;
 795     ssh2_pkt_addbyte((unsigned char)pkt_type);
 796 }
 797 void ssh2_pkt_addbool(unsigned char value) {
 798     ssh2_pkt_adddata(&value, 1);
 799 }
 800 void ssh2_pkt_adduint32(unsigned long value) {
 801     unsigned char x[4];
 802     PUT_32BIT(x, value);
 803     ssh2_pkt_adddata(x, 4);
 804 }
 805 void ssh2_pkt_addstring_start(void) {
 806     ssh2_pkt_adduint32(0);
 807     pktout.savedpos = pktout.length;
 808 }
 809 void ssh2_pkt_addstring_str(char *data) {
 810     ssh2_pkt_adddata(data, strlen(data));
 811     PUT_32BIT(pktout.data + pktout.savedpos - 4,
 812               pktout.length - pktout.savedpos);
 813 }
 814 void ssh2_pkt_addstring_data(char *data, int len) {
 815     ssh2_pkt_adddata(data, len);
 816     PUT_32BIT(pktout.data + pktout.savedpos - 4,
 817               pktout.length - pktout.savedpos);
 818 }
 819 void ssh2_pkt_addstring(char *data) {
 820     ssh2_pkt_addstring_start();
 821     ssh2_pkt_addstring_str(data);
 822 }
 823 char *ssh2_mpint_fmt(Bignum b, int *len) {
 824     unsigned char *p;
 825     int i, n = b[0];
 826     p = malloc(n * 2 + 1);
 827     if (!p)
 828         fatalbox("out of memory");
 829     p[0] = 0;
 830     for (i = 0; i < n; i++) {
 831         p[i*2+1] = (b[n-i] >> 8) & 0xFF;
 832         p[i*2+2] = (b[n-i]     ) & 0xFF;
 833     }
 834     i = 0;
 835     while (p[i] == 0 && (p[i+1] & 0x80) == 0)
 836         i++;
 837     memmove(p, p+i, n*2+1-i);
 838     *len = n*2+1-i;
 839     return p;
 840 }
 841 void ssh2_pkt_addmp(Bignum b) {
 842     unsigned char *p;
 843     int len;
 844     p = ssh2_mpint_fmt(b, &len);
 845     ssh2_pkt_addstring_start();
 846     ssh2_pkt_addstring_data(p, len);
 847     free(p);
 848 }
 849 void ssh2_pkt_send(void) {
 850     int cipherblk, maclen, padding, i;
 851     static unsigned long outgoing_sequence = 0;
 852
 853     /*
 854      * Add padding. At least four bytes, and must also bring total
 855      * length (minus MAC) up to a multiple of the block size.
 856      */
 857     cipherblk = cipher ? cipher->blksize : 8;   /* block size */
 858     cipherblk = cipherblk < 8 ? 8 : cipherblk;   /* or 8 if blksize < 8 */
 859     padding = 4;
 860     padding += (cipherblk - (pktout.length + padding) % cipherblk) % cipherblk;
 861     pktout.data[4] = padding;
 862     for (i = 0; i < padding; i++)
 863         pktout.data[pktout.length + i] = random_byte();
 864     PUT_32BIT(pktout.data, pktout.length + padding - 4);
 865     if (csmac)
 866         csmac->generate(pktout.data, pktout.length + padding,
 867                         outgoing_sequence);
 868     outgoing_sequence++;               /* whether or not we MACed */
 869
 870 #if 0
 871     debug(("Sending packet len=%d\r\n", pktout.length+padding));
 872     for (i = 0; i < pktout.length+padding; i++)
 873         debug(("  %02x", (unsigned char)pktout.data[i]));
 874     debug(("\r\n"));
 875 #endif
 876
 877     if (cscipher)
 878         cscipher->encrypt(pktout.data, pktout.length + padding);
 879     maclen = csmac ? csmac->len : 0;
 880
 881     s_write(pktout.data, pktout.length + padding + maclen);
 882 }
 883
 884 #if 0
 885 void bndebug(char *string, Bignum b) {
 886     unsigned char *p;
 887     int i, len;
 888     p = ssh2_mpint_fmt(b, &len);
 889     debug(("%s", string));
 890     for (i = 0; i < len; i++)
 891         debug((" %02x", p[i]));
 892     debug(("\r\n"));
 893     free(p);
 894 }
 895 #endif
 896
 897 void sha_mpint(SHA_State *s, Bignum b) {
 898     unsigned char *p;
 899     int len;
 900     p = ssh2_mpint_fmt(b, &len);
 901     sha_string(s, p, len);
 902     free(p);
 903 }
 904
 905 /*
 906  * SSH2 packet decode functions.
 907  */
 908 unsigned long ssh2_pkt_getuint32(void) {
 909     unsigned long value;
 910     if (pktin.length - pktin.savedpos < 4)
 911         return 0;                      /* arrgh, no way to decline (FIXME?) */
 912     value = GET_32BIT(pktin.data+pktin.savedpos);
 913     pktin.savedpos += 4;
 914     return value;
 915 }
 916 void ssh2_pkt_getstring(char **p, int *length) {
 917     *p = NULL;
 918     if (pktin.length - pktin.savedpos < 4)
 919         return;
 920     *length = GET_32BIT(pktin.data+pktin.savedpos);
 921     pktin.savedpos += 4;
 922     if (pktin.length - pktin.savedpos < *length)
 923         return;
 924     *p = pktin.data+pktin.savedpos;
 925     pktin.savedpos += *length;
 926 }
 927 Bignum ssh2_pkt_getmp(void) {
 928     char *p;
 929     int i, j, length;
 930     Bignum b;
 931
 932     ssh2_pkt_getstring(&p, &length);
 933     if (!p)
 934         return NULL;
 935     if (p[0] & 0x80)
 936         fatalbox("internal error: Can't handle negative mpints");
 937     b = newbn((length+1)/2);
 938     for (i = 0; i < length; i++) {
 939         j = length - 1 - i;
 940         if (j & 1)
 941             b[j/2+1] |= ((unsigned char)p[i]) << 8;
 942         else
 943             b[j/2+1] |= ((unsigned char)p[i]);
 944     }
 945     return b;
 946 }
 947
 948 static int do_ssh_init(void) {
 949     char c, *vsp;
 950     char version[10];
 951     char vstring[80];
 952     char vlog[sizeof(vstring)+20];
 953     int i;
 954
 955 #ifdef FWHACK
 956     i = 0;
 957     while (s_read(&c, 1) == 1) {
 958         if (c == 'S' && i < 2) i++;
 959         else if (c == 'S' && i == 2) i = 2;
 960         else if (c == 'H' && i == 2) break;
 961         else i = 0;
 962     }
 963 #else
 964     if (s_read(&c,1) != 1 || c != 'S') return 0;
 965     if (s_read(&c,1) != 1 || c != 'S') return 0;
 966     if (s_read(&c,1) != 1 || c != 'H') return 0;
 967 #endif
 968     strcpy(vstring, "SSH-");
 969     vsp = vstring+4;
 970     if (s_read(&c,1) != 1 || c != '-') return 0;
 971     i = 0;
 972     while (1) {
 973         if (s_read(&c,1) != 1)
 974             return 0;
 975         if (vsp < vstring+sizeof(vstring)-1)
 976             *vsp++ = c;
 977         if (i >= 0) {
 978             if (c == '-') {
 979                 version[i] = '\0';
 980                 i = -1;
 981             } else if (i < sizeof(version)-1)
 982                 version[i++] = c;
 983         }
 984         else if (c == '\n')
 985             break;
 986     }
 987
 988     *vsp = 0;
 989     sprintf(vlog, "Server version: %s", vstring);
 990     vlog[strcspn(vlog, "\r\n")] = '\0';
 991     logevent(vlog);
 992
 993     /*
 994      * Server version "1.99" means we can choose whether we use v1
 995      * or v2 protocol. Choice is based on cfg.sshprot.
 996      */
 997     if (ssh_versioncmp(version, cfg.sshprot == 1 ? "2.0" : "1.99") >= 0) {
 998         /*
 999          * This is a v2 server. Begin v2 protocol.
1000          */
1001         char *verstring = "SSH-2.0-PuTTY";
1002         SHA_Init(&exhash);
1003         /*
1004          * Hash our version string and their version string.
1005          */
1006         sha_string(&exhash, verstring, strlen(verstring));
1007         sha_string(&exhash, vstring, strcspn(vstring, "\r\n"));
1008         sprintf(vstring, "%s\n", verstring);
1009         sprintf(vlog, "We claim version: %s", verstring);
1010         logevent(vlog);
1011         logevent("Using SSH protocol version 2");
1012         s_write(vstring, strlen(vstring));
1013         ssh_protocol = ssh2_protocol;
1014         ssh_version = 2;
1015         s_rdpkt = ssh2_rdpkt;
1016     } else {
1017         /*
1018          * This is a v1 server. Begin v1 protocol.
1019          */
1020         sprintf(vstring, "SSH-%s-PuTTY\n",
1021                 (ssh_versioncmp(version, "1.5") <= 0 ? version : "1.5"));
1022         sprintf(vlog, "We claim version: %s", vstring);
1023         vlog[strcspn(vlog, "\r\n")] = '\0';
1024         logevent(vlog);
1025         logevent("Using SSH protocol version 1");
1026         s_write(vstring, strlen(vstring));
1027         ssh_protocol = ssh1_protocol;
1028         ssh_version = 1;
1029         s_rdpkt = ssh1_rdpkt;
1030     }
1031     ssh_send_ok = 0;
1032     return 1;
1033 }
1034
1035 /*
1036  * Handle the key exchange and user authentication phases.
1037  */
1038 static int do_ssh1_login(unsigned char *in, int inlen, int ispkt)
1039 {
1040     int i, j, len;
1041     unsigned char *rsabuf, *keystr1, *keystr2;
1042     unsigned char cookie[8];
1043     struct RSAKey servkey, hostkey;
1044     struct MD5Context md5c;
1045     static unsigned long supported_ciphers_mask, supported_auths_mask;
1046     static int tried_publickey;
1047     static unsigned char session_id[16];
1048     int cipher_type;
1049
1050     crBegin;
1051
1052     if (!ispkt) crWaitUntil(ispkt);
1053
1054     if (pktin.type != SSH1_SMSG_PUBLIC_KEY)
1055         fatalbox("Public key packet not received");
1056
1057     logevent("Received public keys");
1058
1059     memcpy(cookie, pktin.body, 8);
1060
1061     i = makekey(pktin.body+8, &servkey, &keystr1, 0);
1062     j = makekey(pktin.body+8+i, &hostkey, &keystr2, 0);
1063
1064     /*
1065      * Hash the host key and print the hash in the log box. Just as
1066      * a last resort in case the registry's host key checking is
1067      * compromised, we'll allow the user some ability to verify
1068      * host keys by eye.
1069      */
1070     MD5Init(&md5c);
1071     MD5Update(&md5c, keystr2, hostkey.bytes);
1072     MD5Final(session_id, &md5c);
1073     {
1074         char logmsg[80];
1075         int i;
1076         logevent("Host key MD5 is:");
1077         strcpy(logmsg, "      ");
1078         for (i = 0; i < 16; i++)
1079             sprintf(logmsg+strlen(logmsg), "%02x", session_id[i]);
1080         logevent(logmsg);
1081     }
1082
1083     supported_ciphers_mask = GET_32BIT(pktin.body+12+i+j);
1084     supported_auths_mask = GET_32BIT(pktin.body+16+i+j);
1085
1086     MD5Init(&md5c);
1087     MD5Update(&md5c, keystr2, hostkey.bytes);
1088     MD5Update(&md5c, keystr1, servkey.bytes);
1089     MD5Update(&md5c, pktin.body, 8);
1090     MD5Final(session_id, &md5c);
1091
1092     for (i=0; i<32; i++)
1093         session_key[i] = random_byte();
1094
1095     len = (hostkey.bytes > servkey.bytes ? hostkey.bytes : servkey.bytes);
1096
1097     rsabuf = malloc(len);
1098     if (!rsabuf)
1099         fatalbox("Out of memory");
1100
1101     /*
1102      * Verify the host key.
1103      */
1104     {
1105         /*
1106          * First format the key into a string.
1107          */
1108         int len = rsastr_len(&hostkey);
1109         char *keystr = malloc(len);
1110         if (!keystr)
1111             fatalbox("Out of memory");
1112         rsastr_fmt(keystr, &hostkey);
1113         verify_ssh_host_key(savedhost, keystr);
1114         free(keystr);
1115     }
1116
1117     for (i=0; i<32; i++) {
1118         rsabuf[i] = session_key[i];
1119         if (i < 16)
1120             rsabuf[i] ^= session_id[i];
1121     }
1122
1123     if (hostkey.bytes > servkey.bytes) {
1124         rsaencrypt(rsabuf, 32, &servkey);
1125         rsaencrypt(rsabuf, servkey.bytes, &hostkey);
1126     } else {
1127         rsaencrypt(rsabuf, 32, &hostkey);
1128         rsaencrypt(rsabuf, hostkey.bytes, &servkey);
1129     }
1130
1131     logevent("Encrypted session key");
1132
1133     cipher_type = cfg.cipher == CIPHER_BLOWFISH ? SSH_CIPHER_BLOWFISH :
1134                   cfg.cipher == CIPHER_DES ? SSH_CIPHER_DES :
1135                   SSH_CIPHER_3DES;
1136     if ((supported_ciphers_mask & (1 << cipher_type)) == 0) {
1137         c_write("Selected cipher not supported, falling back to 3DES\r\n", 53);
1138         cipher_type = SSH_CIPHER_3DES;
1139     }
1140     switch (cipher_type) {
1141       case SSH_CIPHER_3DES: logevent("Using 3DES encryption"); break;
1142       case SSH_CIPHER_DES: logevent("Using single-DES encryption"); break;
1143       case SSH_CIPHER_BLOWFISH: logevent("Using Blowfish encryption"); break;
1144     }
1145
1146     send_packet(SSH1_CMSG_SESSION_KEY,
1147                 PKT_CHAR, cipher_type,
1148                 PKT_DATA, cookie, 8,
1149                 PKT_CHAR, (len*8) >> 8, PKT_CHAR, (len*8) & 0xFF,
1150                 PKT_DATA, rsabuf, len,
1151                 PKT_INT, 0,
1152                 PKT_END);
1153
1154     logevent("Trying to enable encryption...");
1155
1156     free(rsabuf);
1157
1158     cipher = cipher_type == SSH_CIPHER_BLOWFISH ? &ssh_blowfish :
1159              cipher_type == SSH_CIPHER_DES ? &ssh_des :
1160              &ssh_3des;
1161     cipher->sesskey(session_key);
1162
1163     crWaitUntil(ispkt);
1164
1165     if (pktin.type != SSH1_SMSG_SUCCESS)
1166         fatalbox("Encryption not successfully enabled");
1167
1168     logevent("Successfully started encryption");
1169
1170     fflush(stdout);
1171     {
1172         static char username[100];
1173         static int pos = 0;
1174         static char c;
1175         if ((flags & FLAG_CONNECTION) && !*cfg.username) {
1176             c_write("login as: ", 10);
1177             while (pos >= 0) {
1178                 crWaitUntil(!ispkt);
1179                 while (inlen--) switch (c = *in++) {
1180                   case 10: case 13:
1181                     username[pos] = 0;
1182                     pos = -1;
1183                     break;
1184                   case 8: case 127:
1185                     if (pos > 0) {
1186                         c_write("\b \b", 3);
1187                         pos--;
1188                     }
1189                     break;
1190                   case 21: case 27:
1191                     while (pos > 0) {
1192                         c_write("\b \b", 3);
1193                         pos--;
1194                     }
1195                     break;
1196                   case 3: case 4:
1197                     random_save_seed();
1198                     exit(0);
1199                     break;
1200                   default:
1201                     if (((c >= ' ' && c <= '~') ||
1202                          ((unsigned char)c >= 160)) && pos < 40) {
1203                         username[pos++] = c;
1204                         c_write(&c, 1);
1205                     }
1206                     break;
1207                 }
1208             }
1209             c_write("\r\n", 2);
1210             username[strcspn(username, "\n\r")] = '\0';
1211         } else {
1212             char stuff[200];
1213             strncpy(username, cfg.username, 99);
1214             username[99] = '\0';
1215             if (flags & FLAG_VERBOSE) {
1216                 sprintf(stuff, "Sent username \"%s\".\r\n", username);
1217                 c_write(stuff, strlen(stuff));
1218             }
1219         }
1220
1221         send_packet(SSH1_CMSG_USER, PKT_STR, username, PKT_END);
1222         {
1223             char userlog[20+sizeof(username)];
1224             sprintf(userlog, "Sent username \"%s\"", username);
1225             logevent(userlog);
1226         }
1227     }
1228
1229     crWaitUntil(ispkt);
1230
1231     tried_publickey = 0;
1232
1233     while (pktin.type == SSH1_SMSG_FAILURE) {
1234         static char password[100];
1235         static int pos;
1236         static char c;
1237         static int pwpkt_type;
1238         /*
1239          * Show password prompt, having first obtained it via a TIS
1240          * or CryptoCard exchange if we're doing TIS or CryptoCard
1241          * authentication.
1242          */
1243         pwpkt_type = SSH1_CMSG_AUTH_PASSWORD;
1244         if (agent_exists()) {
1245             /*
1246              * Attempt RSA authentication using Pageant.
1247              */
1248             static unsigned char request[5], *response, *p;
1249             static int responselen;
1250             static int i, nkeys;
1251             static int authed = FALSE;
1252             void *r;
1253
1254             logevent("Pageant is running. Requesting keys.");
1255
1256             /* Request the keys held by the agent. */
1257             PUT_32BIT(request, 1);
1258             request[4] = SSH_AGENTC_REQUEST_RSA_IDENTITIES;
1259             agent_query(request, 5, &r, &responselen);
1260             response = (unsigned char *)r;
1261             if (response) {
1262                 p = response + 5;
1263                 nkeys = GET_32BIT(p); p += 4;
1264                 { char buf[64]; sprintf(buf, "Pageant has %d keys", nkeys);
1265                     logevent(buf); }
1266                 for (i = 0; i < nkeys; i++) {
1267                     static struct RSAKey key;
1268                     static Bignum challenge;
1269
1270                     { char buf[64]; sprintf(buf, "Trying Pageant key #%d", i);
1271                         logevent(buf); }
1272                     p += 4;
1273                     p += ssh1_read_bignum(p, &key.exponent);
1274                     p += ssh1_read_bignum(p, &key.modulus);
1275                     send_packet(SSH1_CMSG_AUTH_RSA,
1276                                 PKT_BIGNUM, key.modulus, PKT_END);
1277                     crWaitUntil(ispkt);
1278                     if (pktin.type != SSH1_SMSG_AUTH_RSA_CHALLENGE) {
1279                         logevent("Key refused");
1280                         continue;
1281                     }
1282                     logevent("Received RSA challenge");
1283                     ssh1_read_bignum(pktin.body, &challenge);
1284                     {
1285                         char *agentreq, *q, *ret;
1286                         int len, retlen;
1287                         len = 1 + 4;   /* message type, bit count */
1288                         len += ssh1_bignum_length(key.exponent);
1289                         len += ssh1_bignum_length(key.modulus);
1290                         len += ssh1_bignum_length(challenge);
1291                         len += 16;     /* session id */
1292                         len += 4;      /* response format */
1293                         agentreq = malloc(4 + len);
1294                         PUT_32BIT(agentreq, len);
1295                         q = agentreq + 4;
1296                         *q++ = SSH_AGENTC_RSA_CHALLENGE;
1297                         PUT_32BIT(q, ssh1_bignum_bitcount(key.modulus));
1298                         q += 4;
1299                         q += ssh1_write_bignum(q, key.exponent);
1300                         q += ssh1_write_bignum(q, key.modulus);
1301                         q += ssh1_write_bignum(q, challenge);
1302                         memcpy(q, session_id, 16); q += 16;
1303                         PUT_32BIT(q, 1);   /* response format */
1304                         agent_query(agentreq, len+4, &ret, &retlen);
1305                         free(agentreq);
1306                         if (ret) {
1307                             if (ret[4] == SSH_AGENT_RSA_RESPONSE) {
1308                                 logevent("Sending Pageant's response");
1309                                 send_packet(SSH1_CMSG_AUTH_RSA_RESPONSE,
1310                                             PKT_DATA, ret+5, 16, PKT_END);
1311                                 free(ret);
1312                                 crWaitUntil(ispkt);
1313                                 if (pktin.type == SSH1_SMSG_SUCCESS) {
1314                                     logevent("Pageant's response accepted");
1315                                     authed = TRUE;
1316                                 } else
1317                                     logevent("Pageant's response not accepted");
1318                             } else {
1319                                 logevent("Pageant failed to answer challenge");
1320                                 free(ret);
1321                             }
1322                         } else {
1323                             logevent("No reply received from Pageant");
1324                         }
1325                     }
1326                     freebn(key.exponent);
1327                     freebn(key.modulus);
1328                     freebn(challenge);
1329                     if (authed)
1330                         break;
1331                 }
1332             }
1333             if (authed)
1334                 break;
1335         }
1336         if (*cfg.keyfile && !tried_publickey)
1337             pwpkt_type = SSH1_CMSG_AUTH_RSA;
1338
1339         if (pwpkt_type == SSH1_CMSG_AUTH_PASSWORD && !FLAG_WINDOWED) {
1340             char prompt[200];
1341             sprintf(prompt, "%s@%s's password: ", cfg.username, savedhost);
1342             if (!ssh_get_password(prompt, password, sizeof(password))) {
1343                 /*
1344                  * get_password failed to get a password (for
1345                  * example because one was supplied on the command
1346                  * line which has already failed to work).
1347                  * Terminate.
1348                  */
1349                 logevent("No more passwords to try");
1350                 ssh_state = SSH_STATE_CLOSED;
1351                 crReturn(1);
1352             }
1353         } else {
1354
1355             if (pktin.type == SSH1_SMSG_FAILURE &&
1356                 cfg.try_tis_auth &&
1357                 (supported_auths_mask & (1<<SSH1_AUTH_TIS))) {
1358                 pwpkt_type = SSH1_CMSG_AUTH_TIS_RESPONSE;
1359                 logevent("Requested TIS authentication");
1360                 send_packet(SSH1_CMSG_AUTH_TIS, PKT_END);
1361                 crWaitUntil(ispkt);
1362                 if (pktin.type != SSH1_SMSG_AUTH_TIS_CHALLENGE) {
1363                     logevent("TIS authentication declined");
1364                     c_write("TIS authentication refused.\r\n", 29);
1365                 } else {
1366                     int challengelen = ((pktin.body[0] << 24) |
1367                                         (pktin.body[1] << 16) |
1368                                         (pktin.body[2] << 8) |
1369                                         (pktin.body[3]));
1370                     logevent("Received TIS challenge");
1371                     c_write(pktin.body+4, challengelen);
1372                 }
1373             }
1374             if (pktin.type == SSH1_SMSG_FAILURE &&
1375                 cfg.try_tis_auth &&
1376                 (supported_auths_mask & (1<<SSH1_AUTH_CCARD))) {
1377                 pwpkt_type = SSH1_CMSG_AUTH_CCARD_RESPONSE;
1378                 logevent("Requested CryptoCard authentication");
1379                 send_packet(SSH1_CMSG_AUTH_CCARD, PKT_END);
1380                 crWaitUntil(ispkt);
1381                 if (pktin.type != SSH1_SMSG_AUTH_CCARD_CHALLENGE) {
1382                     logevent("CryptoCard authentication declined");
1383                     c_write("CryptoCard authentication refused.\r\n", 29);
1384                 } else {
1385                     int challengelen = ((pktin.body[0] << 24) |
1386                                         (pktin.body[1] << 16) |
1387                                         (pktin.body[2] << 8) |
1388                                         (pktin.body[3]));
1389                     logevent("Received CryptoCard challenge");
1390                     c_write(pktin.body+4, challengelen);
1391                     c_write("\r\nResponse : ", 13);
1392                 }
1393             }
1394             if (pwpkt_type == SSH1_CMSG_AUTH_PASSWORD)
1395                 c_write("password: ", 10);
1396             if (pwpkt_type == SSH1_CMSG_AUTH_RSA) {
1397                 if (flags & FLAG_VERBOSE)
1398                     c_write("Trying public key authentication.\r\n", 35);
1399                 if (!rsakey_encrypted(cfg.keyfile)) {
1400                     if (flags & FLAG_VERBOSE)
1401                         c_write("No passphrase required.\r\n", 25);
1402                     goto tryauth;
1403                 }
1404                 c_write("passphrase: ", 12);
1405             }
1406
1407             pos = 0;
1408             while (pos >= 0) {
1409                 crWaitUntil(!ispkt);
1410                 while (inlen--) switch (c = *in++) {
1411                   case 10: case 13:
1412                     password[pos] = 0;
1413                     pos = -1;
1414                     break;
1415                   case 8: case 127:
1416                     if (pos > 0)
1417                         pos--;
1418                     break;
1419                   case 21: case 27:
1420                     pos = 0;
1421                     break;
1422                   case 3: case 4:
1423                     random_save_seed();
1424                     exit(0);
1425                     break;
1426                   default:
1427                     if (((c >= ' ' && c <= '~') ||
1428                          ((unsigned char)c >= 160)) && pos < sizeof(password))
1429                         password[pos++] = c;
1430                     break;
1431                 }
1432             }
1433             c_write("\r\n", 2);
1434
1435         }
1436
1437         tryauth:
1438         if (pwpkt_type == SSH1_CMSG_AUTH_RSA) {
1439             /*
1440              * Try public key authentication with the specified
1441              * key file.
1442              */
1443             static struct RSAKey pubkey;
1444             static Bignum challenge, response;
1445             static int i;
1446             static unsigned char buffer[32];
1447
1448             tried_publickey = 1;
1449             i = loadrsakey(cfg.keyfile, &pubkey, password);
1450             if (i == 0) {
1451                 c_write("Couldn't load public key from ", 30);
1452                 c_write(cfg.keyfile, strlen(cfg.keyfile));
1453                 c_write(".\r\n", 3);
1454                 continue;              /* go and try password */
1455             }
1456             if (i == -1) {
1457                 c_write("Wrong passphrase.\r\n", 19);
1458                 tried_publickey = 0;
1459                 continue;              /* try again */
1460             }
1461
1462             /*
1463              * Send a public key attempt.
1464              */
1465             send_packet(SSH1_CMSG_AUTH_RSA,
1466                         PKT_BIGNUM, pubkey.modulus, PKT_END);
1467
1468             crWaitUntil(ispkt);
1469             if (pktin.type == SSH1_SMSG_FAILURE) {
1470                 if (flags & FLAG_VERBOSE)
1471                     c_write("Server refused our public key.\r\n", 32);
1472                 continue;              /* go and try password */
1473             }
1474             if (pktin.type != SSH1_SMSG_AUTH_RSA_CHALLENGE)
1475                 fatalbox("Bizarre response to offer of public key");
1476             ssh1_read_bignum(pktin.body, &challenge);
1477             response = rsadecrypt(challenge, &pubkey);
1478             freebn(pubkey.private_exponent);   /* burn the evidence */
1479
1480             for (i = 0; i < 32; i += 2) {
1481                 buffer[i] = response[16-i/2] >> 8;
1482                 buffer[i+1] = response[16-i/2] & 0xFF;
1483             }
1484
1485             MD5Init(&md5c);
1486             MD5Update(&md5c, buffer, 32);
1487             MD5Update(&md5c, session_id, 16);
1488             MD5Final(buffer, &md5c);
1489
1490             send_packet(SSH1_CMSG_AUTH_RSA_RESPONSE,
1491                         PKT_DATA, buffer, 16, PKT_END);
1492
1493             crWaitUntil(ispkt);
1494             if (pktin.type == SSH1_SMSG_FAILURE) {
1495                 if (flags & FLAG_VERBOSE)
1496                     c_write("Failed to authenticate with our public key.\r\n",
1497                             45);
1498                 continue;              /* go and try password */
1499             } else if (pktin.type != SSH1_SMSG_SUCCESS) {
1500                 fatalbox("Bizarre response to RSA authentication response");
1501             }
1502
1503             break;                     /* we're through! */
1504         } else {
1505             send_packet(pwpkt_type, PKT_STR, password, PKT_END);
1506         }
1507         logevent("Sent password");
1508         memset(password, 0, strlen(password));
1509         crWaitUntil(ispkt);
1510         if (pktin.type == SSH1_SMSG_FAILURE) {
1511             if (flags & FLAG_VERBOSE)
1512                 c_write("Access denied\r\n", 15);
1513             logevent("Authentication refused");
1514         } else if (pktin.type == SSH1_MSG_DISCONNECT) {
1515             logevent("Received disconnect request");
1516             ssh_state = SSH_STATE_CLOSED;
1517             crReturn(1);
1518         } else if (pktin.type != SSH1_SMSG_SUCCESS) {
1519             fatalbox("Strange packet received, type %d", pktin.type);
1520         }
1521     }
1522
1523     logevent("Authentication successful");
1524
1525     crFinish(1);
1526 }
1527
1528 static void ssh1_protocol(unsigned char *in, int inlen, int ispkt) {
1529     crBegin;
1530
1531     random_init();
1532
1533     while (!do_ssh1_login(in, inlen, ispkt)) {
1534         crReturnV;
1535     }
1536     if (ssh_state == SSH_STATE_CLOSED)
1537         crReturnV;
1538
1539     if (1 /* FIXME: agent exists && agent forwarding configured */ ) {
1540         logevent("Requesting agent forwarding");
1541         send_packet(SSH1_CMSG_AGENT_REQUEST_FORWARDING, PKT_END);
1542         do { crReturnV; } while (!ispkt);
1543         if (pktin.type != SSH1_SMSG_SUCCESS && pktin.type != SSH1_SMSG_FAILURE) {
1544             fatalbox("Protocol confusion");
1545         } else if (pktin.type == SSH1_SMSG_FAILURE) {
1546             logevent("Agent forwarding refused");
1547         } else
1548             logevent("Agent forwarding enabled");
1549     }
1550
1551     if (!cfg.nopty) {
1552         send_packet(SSH1_CMSG_REQUEST_PTY,
1553                     PKT_STR, cfg.termtype,
1554                     PKT_INT, rows, PKT_INT, cols,
1555                     PKT_INT, 0, PKT_INT, 0,
1556                     PKT_CHAR, 0,
1557                     PKT_END);
1558         ssh_state = SSH_STATE_INTERMED;
1559         do { crReturnV; } while (!ispkt);
1560         if (pktin.type != SSH1_SMSG_SUCCESS && pktin.type != SSH1_SMSG_FAILURE) {
1561             fatalbox("Protocol confusion");
1562         } else if (pktin.type == SSH1_SMSG_FAILURE) {
1563             c_write("Server refused to allocate pty\r\n", 32);
1564         }
1565         logevent("Allocated pty");
1566     }
1567
1568     if (*cfg.remote_cmd)
1569         send_packet(SSH1_CMSG_EXEC_CMD, PKT_STR, cfg.remote_cmd, PKT_END);
1570     else
1571         send_packet(SSH1_CMSG_EXEC_SHELL, PKT_END);
1572     logevent("Started session");
1573
1574     ssh_state = SSH_STATE_SESSION;
1575     if (size_needed)
1576         ssh_size();
1577
1578     ssh_send_ok = 1;
1579     ssh_channels = newtree234(ssh_channelcmp);
1580     while (1) {
1581         crReturnV;
1582         if (ispkt) {
1583             if (pktin.type == SSH1_SMSG_STDOUT_DATA ||
1584                 pktin.type == SSH1_SMSG_STDERR_DATA) {
1585                 long len = GET_32BIT(pktin.body);
1586                 c_write(pktin.body+4, len);
1587             } else if (pktin.type == SSH1_MSG_DISCONNECT) {
1588                 ssh_state = SSH_STATE_CLOSED;
1589                 logevent("Received disconnect request");
1590             } else if (pktin.type == SSH1_SMSG_AGENT_OPEN) {
1591                 /* Remote side is trying to open a channel to talk to our
1592                  * agent. Give them back a local channel number. */
1593                 int i = 1;
1594                 struct ssh_channel *c;
1595                 enum234 e;
1596                 for (c = first234(ssh_channels, &e); c; c = next234(&e)) {
1597                     if (c->localid > i)
1598                         break;         /* found a free number */
1599                     i = c->localid + 1;
1600                 }
1601                 c = malloc(sizeof(struct ssh_channel));
1602                 c->remoteid = GET_32BIT(pktin.body);
1603                 c->localid = i;
1604                 c->type = SSH1_SMSG_AGENT_OPEN;   /* identify channel type */
1605                 add234(ssh_channels, c);
1606                 send_packet(SSH1_MSG_CHANNEL_OPEN_CONFIRMATION,
1607                             PKT_INT, c->remoteid, PKT_INT, c->localid,
1608                             PKT_END);
1609             } else if (pktin.type == SSH1_MSG_CHANNEL_CLOSE ||
1610                        pktin.type == SSH1_MSG_CHANNEL_CLOSE_CONFIRMATION) {
1611                 /* Remote side closes a channel. */
1612                 int i = GET_32BIT(pktin.body);
1613                 struct ssh_channel *c;
1614                 c = find234(ssh_channels, &i, ssh_channelfind);
1615                 if (c) {
1616                     int closetype;
1617                     closetype = (pktin.type == SSH1_MSG_CHANNEL_CLOSE ? 1 : 2);
1618                     send_packet(pktin.type, PKT_INT, c->remoteid, PKT_END);
1619                     c->closes |= closetype;
1620                     if (c->closes == 3) {
1621                         del234(ssh_channels, c);
1622                         free(c);
1623                     }
1624                 }
1625             } else if (pktin.type == SSH1_MSG_CHANNEL_DATA) {
1626                 /* Data sent down one of our channels. */
1627                 int i = GET_32BIT(pktin.body);
1628                 int len = GET_32BIT(pktin.body+4);
1629                 unsigned char *p = pktin.body+8;
1630                 struct ssh_channel *c;
1631                 c = find234(ssh_channels, &i, ssh_channelfind);
1632                 if (c) {
1633                     switch(c->type) {
1634                       case SSH1_SMSG_AGENT_OPEN:
1635                         /* Data for an agent message. Buffer it. */
1636                         while (len > 0) {
1637                             if (c->u.a.lensofar < 4) {
1638                                 int l = min(4 - c->u.a.lensofar, len);
1639                                 memcpy(c->u.a.msglen + c->u.a.lensofar, p, l);
1640                                 p += l; len -= l; c->u.a.lensofar += l;
1641                             }
1642                             if (c->u.a.lensofar == 4) {
1643                                 c->u.a.totallen = 4 + GET_32BIT(c->u.a.msglen);
1644                                 c->u.a.message = malloc(c->u.a.totallen);
1645                                 memcpy(c->u.a.message, c->u.a.msglen, 4);
1646                             }
1647                             if (c->u.a.lensofar >= 4 && len > 0) {
1648                                 int l = min(c->u.a.totallen - c->u.a.lensofar, len);
1649                                 memcpy(c->u.a.message + c->u.a.lensofar, p, l);
1650                                 p += l; len -= l; c->u.a.lensofar += l;
1651                             }
1652                             if (c->u.a.lensofar == c->u.a.totallen) {
1653                                 void *reply, *sentreply;
1654                                 int replylen;
1655                                 agent_query(c->u.a.message, c->u.a.totallen,
1656                                             &reply, &replylen);
1657                                 if (reply)
1658                                     sentreply = reply;
1659                                 else {
1660                                     /* Fake SSH_AGENT_FAILURE. */
1661                                     sentreply = "\0\0\0\1\5";
1662                                     replylen = 5;
1663                                 }
1664                                 send_packet(SSH1_MSG_CHANNEL_DATA,
1665                                             PKT_INT, c->remoteid,
1666                                             PKT_INT, replylen,
1667                                             PKT_DATA, sentreply, replylen,
1668                                             PKT_END);
1669                                 if (reply)
1670                                     free(reply);
1671                                 free(c->u.a.message);
1672                                 c->u.a.lensofar = 0;
1673                             }
1674                         }
1675                         break;
1676                     }
1677                 }
1678             } else if (pktin.type == SSH1_SMSG_SUCCESS) {
1679                 /* may be from EXEC_SHELL on some servers */
1680             } else if (pktin.type == SSH1_SMSG_FAILURE) {
1681                 /* may be from EXEC_SHELL on some servers
1682                  * if no pty is available or in other odd cases. Ignore */
1683             } else if (pktin.type == SSH1_SMSG_EXIT_STATUS) {
1684                 send_packet(SSH1_CMSG_EXIT_CONFIRMATION, PKT_END);
1685             } else {
1686                 fatalbox("Strange packet received: type %d", pktin.type);
1687             }
1688         } else {
1689             send_packet(SSH1_CMSG_STDIN_DATA,
1690                         PKT_INT, inlen, PKT_DATA, in, inlen, PKT_END);
1691         }
1692     }
1693
1694     crFinishV;
1695 }
1696
1697 /*
1698  * Utility routine for decoding comma-separated strings in KEXINIT.
1699  */
1700 int in_commasep_string(char *needle, char *haystack, int haylen) {
1701     int needlen = strlen(needle);
1702     while (1) {
1703         /*
1704          * Is it at the start of the string?
1705          */
1706         if (haylen >= needlen &&       /* haystack is long enough */
1707             !memcmp(needle, haystack, needlen) &&    /* initial match */
1708             (haylen == needlen || haystack[needlen] == ',')
1709                                        /* either , or EOS follows */
1710             )
1711             return 1;
1712         /*
1713          * If not, search for the next comma and resume after that.
1714          * If no comma found, terminate.
1715          */
1716         while (haylen > 0 && *haystack != ',')
1717             haylen--, haystack++;
1718         if (haylen == 0)
1719             return 0;
1720         haylen--, haystack++;          /* skip over comma itself */
1721     }
1722 }
1723
1724 /*
1725  * SSH2 key creation method.
1726  */
1727 void ssh2_mkkey(Bignum K, char *H, char chr, char *keyspace) {
1728     SHA_State s;
1729     /* First 20 bytes. */
1730     SHA_Init(&s);
1731     sha_mpint(&s, K);
1732     SHA_Bytes(&s, H, 20);
1733     SHA_Bytes(&s, &chr, 1);
1734     SHA_Bytes(&s, H, 20);
1735     SHA_Final(&s, keyspace);
1736     /* Next 20 bytes. */
1737     SHA_Init(&s);
1738     sha_mpint(&s, K);
1739     SHA_Bytes(&s, H, 20);
1740     SHA_Bytes(&s, keyspace, 20);
1741     SHA_Final(&s, keyspace+20);
1742 }
1743
1744 /*
1745  * Handle the SSH2 transport layer.
1746  */
1747 static int do_ssh2_transport(unsigned char *in, int inlen, int ispkt)
1748 {
1749     static int i, len;
1750     static char *str;
1751     static Bignum e, f, K;
1752     static struct ssh_cipher *cscipher_tobe = NULL;
1753     static struct ssh_cipher *sccipher_tobe = NULL;
1754     static struct ssh_mac *csmac_tobe = NULL;
1755     static struct ssh_mac *scmac_tobe = NULL;
1756     static struct ssh_compress *cscomp_tobe = NULL;
1757     static struct ssh_compress *sccomp_tobe = NULL;
1758     static char *hostkeydata, *sigdata, *keystr;
1759     static int hostkeylen, siglen;
1760     static unsigned char exchange_hash[20];
1761     static unsigned char keyspace[40];
1762
1763     crBegin;
1764     random_init();
1765
1766     begin_key_exchange:
1767     /*
1768      * Construct and send our key exchange packet.
1769      */
1770     ssh2_pkt_init(SSH2_MSG_KEXINIT);
1771     for (i = 0; i < 16; i++)
1772         ssh2_pkt_addbyte((unsigned char)random_byte());
1773     /* List key exchange algorithms. */
1774     ssh2_pkt_addstring_start();
1775     for (i = 0; i < lenof(kex_algs); i++) {
1776         ssh2_pkt_addstring_str(kex_algs[i]->name);
1777         if (i < lenof(kex_algs)-1)
1778             ssh2_pkt_addstring_str(",");
1779     }
1780     /* List server host key algorithms. */
1781     ssh2_pkt_addstring_start();
1782     for (i = 0; i < lenof(hostkey_algs); i++) {
1783         ssh2_pkt_addstring_str(hostkey_algs[i]->name);
1784         if (i < lenof(hostkey_algs)-1)
1785             ssh2_pkt_addstring_str(",");
1786     }
1787     /* List client->server encryption algorithms. */
1788     ssh2_pkt_addstring_start();
1789     for (i = 0; i < lenof(ciphers); i++) {
1790         ssh2_pkt_addstring_str(ciphers[i]->name);
1791         if (i < lenof(ciphers)-1)
1792             ssh2_pkt_addstring_str(",");
1793     }
1794     /* List server->client encryption algorithms. */
1795     ssh2_pkt_addstring_start();
1796     for (i = 0; i < lenof(ciphers); i++) {
1797         ssh2_pkt_addstring_str(ciphers[i]->name);
1798         if (i < lenof(ciphers)-1)
1799             ssh2_pkt_addstring_str(",");
1800     }
1801     /* List client->server MAC algorithms. */
1802     ssh2_pkt_addstring_start();
1803     for (i = 0; i < lenof(macs); i++) {
1804         ssh2_pkt_addstring_str(macs[i]->name);
1805         if (i < lenof(macs)-1)
1806             ssh2_pkt_addstring_str(",");
1807     }
1808     /* List server->client MAC algorithms. */
1809     ssh2_pkt_addstring_start();
1810     for (i = 0; i < lenof(macs); i++) {
1811         ssh2_pkt_addstring_str(macs[i]->name);
1812         if (i < lenof(macs)-1)
1813             ssh2_pkt_addstring_str(",");
1814     }
1815     /* List client->server compression algorithms. */
1816     ssh2_pkt_addstring_start();
1817     for (i = 0; i < lenof(compressions); i++) {
1818         ssh2_pkt_addstring_str(compressions[i]->name);
1819         if (i < lenof(compressions)-1)
1820             ssh2_pkt_addstring_str(",");
1821     }
1822     /* List server->client compression algorithms. */
1823     ssh2_pkt_addstring_start();
1824     for (i = 0; i < lenof(compressions); i++) {
1825         ssh2_pkt_addstring_str(compressions[i]->name);
1826         if (i < lenof(compressions)-1)
1827             ssh2_pkt_addstring_str(",");
1828     }
1829     /* List client->server languages. Empty list. */
1830     ssh2_pkt_addstring_start();
1831     /* List server->client languages. Empty list. */
1832     ssh2_pkt_addstring_start();
1833     /* First KEX packet does _not_ follow, because we're not that brave. */
1834     ssh2_pkt_addbool(FALSE);
1835     /* Reserved. */
1836     ssh2_pkt_adduint32(0);
1837     sha_string(&exhash, pktout.data+5, pktout.length-5);
1838     ssh2_pkt_send();
1839
1840     if (!ispkt) crWaitUntil(ispkt);
1841     sha_string(&exhash, pktin.data+5, pktin.length-5);
1842
1843     /*
1844      * Now examine the other side's KEXINIT to see what we're up
1845      * to.
1846      */
1847     if (pktin.type != SSH2_MSG_KEXINIT) {
1848         fatalbox("expected key exchange packet from server");
1849     }
1850     kex = NULL; hostkey = NULL; cscipher_tobe = NULL; sccipher_tobe = NULL;
1851     csmac_tobe = NULL; scmac_tobe = NULL; cscomp_tobe = NULL; sccomp_tobe = NULL;
1852     pktin.savedpos += 16;              /* skip garbage cookie */
1853     ssh2_pkt_getstring(&str, &len);    /* key exchange algorithms */
1854     for (i = 0; i < lenof(kex_algs); i++) {
1855         if (in_commasep_string(kex_algs[i]->name, str, len)) {
1856             kex = kex_algs[i];
1857             break;
1858         }
1859     }
1860     ssh2_pkt_getstring(&str, &len);    /* host key algorithms */
1861     for (i = 0; i < lenof(hostkey_algs); i++) {
1862         if (in_commasep_string(hostkey_algs[i]->name, str, len)) {
1863             hostkey = hostkey_algs[i];
1864             break;
1865         }
1866     }
1867     ssh2_pkt_getstring(&str, &len);    /* client->server cipher */
1868     for (i = 0; i < lenof(ciphers); i++) {
1869         if (in_commasep_string(ciphers[i]->name, str, len)) {
1870             cscipher_tobe = ciphers[i];
1871             break;
1872         }
1873     }
1874     ssh2_pkt_getstring(&str, &len);    /* server->client cipher */
1875     for (i = 0; i < lenof(ciphers); i++) {
1876         if (in_commasep_string(ciphers[i]->name, str, len)) {
1877             sccipher_tobe = ciphers[i];
1878             break;
1879         }
1880     }
1881     ssh2_pkt_getstring(&str, &len);    /* client->server mac */
1882     for (i = 0; i < lenof(macs); i++) {
1883         if (in_commasep_string(macs[i]->name, str, len)) {
1884             csmac_tobe = macs[i];
1885             break;
1886         }
1887     }
1888     ssh2_pkt_getstring(&str, &len);    /* server->client mac */
1889     for (i = 0; i < lenof(macs); i++) {
1890         if (in_commasep_string(macs[i]->name, str, len)) {
1891             scmac_tobe = macs[i];
1892             break;
1893         }
1894     }
1895     ssh2_pkt_getstring(&str, &len);    /* client->server compression */
1896     for (i = 0; i < lenof(compressions); i++) {
1897         if (in_commasep_string(compressions[i]->name, str, len)) {
1898             cscomp_tobe = compressions[i];
1899             break;
1900         }
1901     }
1902     ssh2_pkt_getstring(&str, &len);    /* server->client compression */
1903     for (i = 0; i < lenof(compressions); i++) {
1904         if (in_commasep_string(compressions[i]->name, str, len)) {
1905             sccomp_tobe = compressions[i];
1906             break;
1907         }
1908     }
1909
1910     /*
1911      * Currently we only support Diffie-Hellman and DSS, so let's
1912      * bomb out if those aren't selected.
1913      */
1914     if (kex != &ssh_diffiehellman || hostkey != &ssh_dss)
1915         fatalbox("internal fault: chaos in SSH 2 transport layer");
1916
1917     /*
1918      * Now we begin the fun. Generate and send e for Diffie-Hellman.
1919      */
1920     e = dh_create_e();
1921     ssh2_pkt_init(SSH2_MSG_KEXDH_INIT);
1922     ssh2_pkt_addmp(e);
1923     ssh2_pkt_send();
1924
1925     crWaitUntil(ispkt);
1926     if (pktin.type != SSH2_MSG_KEXDH_REPLY) {
1927         fatalbox("expected key exchange packet from server");
1928     }
1929     ssh2_pkt_getstring(&hostkeydata, &hostkeylen);
1930     f = ssh2_pkt_getmp();
1931     ssh2_pkt_getstring(&sigdata, &siglen);
1932
1933     K = dh_find_K(f);
1934
1935     sha_string(&exhash, hostkeydata, hostkeylen);
1936     sha_mpint(&exhash, e);
1937     sha_mpint(&exhash, f);
1938     sha_mpint(&exhash, K);
1939     SHA_Final(&exhash, exchange_hash);
1940
1941 #if 0
1942     debug(("Exchange hash is:\r\n"));
1943     for (i = 0; i < 20; i++)
1944         debug((" %02x", exchange_hash[i]));
1945     debug(("\r\n"));
1946 #endif
1947
1948     hostkey->setkey(hostkeydata, hostkeylen);
1949     if (!hostkey->verifysig(sigdata, siglen, exchange_hash, 20))
1950         fatalbox("Server failed host key check");
1951
1952     /*
1953      * Expect SSH2_MSG_NEWKEYS from server.
1954      */
1955     crWaitUntil(ispkt);
1956     if (pktin.type != SSH2_MSG_NEWKEYS)
1957         fatalbox("expected new-keys packet from server");
1958
1959     /*
1960      * Authenticate remote host: verify host key. (We've already
1961      * checked the signature of the exchange hash.)
1962      */
1963     keystr = hostkey->fmtkey();
1964     verify_ssh_host_key(savedhost, keystr);
1965     free(keystr);
1966
1967     /*
1968      * Send SSH2_MSG_NEWKEYS.
1969      */
1970     ssh2_pkt_init(SSH2_MSG_NEWKEYS);
1971     ssh2_pkt_send();
1972
1973     /*
1974      * Create and initialise session keys.
1975      */
1976     cscipher = cscipher_tobe;
1977     sccipher = sccipher_tobe;
1978     csmac = csmac_tobe;
1979     scmac = scmac_tobe;
1980     cscomp = cscomp_tobe;
1981     sccomp = sccomp_tobe;
1982     /*
1983      * Set IVs after keys.
1984      */
1985     ssh2_mkkey(K, exchange_hash, 'C', keyspace); cscipher->setcskey(keyspace);
1986     ssh2_mkkey(K, exchange_hash, 'D', keyspace); cscipher->setsckey(keyspace);
1987     ssh2_mkkey(K, exchange_hash, 'A', keyspace); cscipher->setcsiv(keyspace);
1988     ssh2_mkkey(K, exchange_hash, 'B', keyspace); sccipher->setsciv(keyspace);
1989     ssh2_mkkey(K, exchange_hash, 'E', keyspace); csmac->setcskey(keyspace);
1990     ssh2_mkkey(K, exchange_hash, 'F', keyspace); scmac->setsckey(keyspace);
1991
1992     /*
1993      * Now we're encrypting. Begin returning 1 to the protocol main
1994      * function so that other things can run on top of the
1995      * transport. If we ever see a KEXINIT, we must go back to the
1996      * start.
1997      */
1998     do {
1999         crReturn(1);
2000     } while (!(ispkt && pktin.type == SSH2_MSG_KEXINIT));
2001     goto begin_key_exchange;
2002
2003     crFinish(1);
2004 }
2005
2006 /*
2007  * SSH2: remote identifier for the main session channel.
2008  */
2009 static unsigned long ssh_remote_channel;
2010
2011 /*
2012  * Handle the SSH2 userauth and connection layers.
2013  */
2014 static void do_ssh2_authconn(unsigned char *in, int inlen, int ispkt)
2015 {
2016     static unsigned long remote_winsize;
2017     static unsigned long remote_maxpkt;
2018
2019     crBegin;
2020
2021     /*
2022      * Request userauth protocol, and await a response to it.
2023      */
2024     ssh2_pkt_init(SSH2_MSG_SERVICE_REQUEST);
2025     ssh2_pkt_addstring("ssh-userauth");
2026     ssh2_pkt_send();
2027     crWaitUntilV(ispkt);
2028     if (pktin.type != SSH2_MSG_SERVICE_ACCEPT)
2029         fatalbox("Server refused user authentication protocol");
2030
2031     /*
2032      * FIXME: currently we support only password authentication.
2033      * (This places us technically in violation of the SSH2 spec.
2034      * We must fix this.)
2035      */
2036     while (1) {
2037         /*
2038          * Get a username and a password.
2039          */
2040         static char username[100];
2041         static char password[100];
2042         static int pos = 0;
2043         static char c;
2044
2045         if ((flags & FLAG_CONNECTION) && !*cfg.username) {
2046             c_write("login as: ", 10);
2047             while (pos >= 0) {
2048                 crWaitUntilV(!ispkt);
2049                 while (inlen--) switch (c = *in++) {
2050                   case 10: case 13:
2051                     username[pos] = 0;
2052                     pos = -1;
2053                     break;
2054                   case 8: case 127:
2055                     if (pos > 0) {
2056                         c_write("\b \b", 3);
2057                         pos--;
2058                     }
2059                     break;
2060                   case 21: case 27:
2061                     while (pos > 0) {
2062                         c_write("\b \b", 3);
2063                         pos--;
2064                     }
2065                     break;
2066                   case 3: case 4:
2067                     random_save_seed();
2068                     exit(0);
2069                     break;
2070                   default:
2071                     if (((c >= ' ' && c <= '~') ||
2072                          ((unsigned char)c >= 160)) && pos < 40) {
2073                         username[pos++] = c;
2074                         c_write(&c, 1);
2075                     }
2076                     break;
2077                 }
2078             }
2079             c_write("\r\n", 2);
2080             username[strcspn(username, "\n\r")] = '\0';
2081         } else {
2082             char stuff[200];
2083             strncpy(username, cfg.username, 99);
2084             username[99] = '\0';
2085             if (flags & FLAG_VERBOSE) {
2086                 sprintf(stuff, "Using username \"%s\".\r\n", username);
2087                 c_write(stuff, strlen(stuff));
2088             }
2089         }
2090
2091         if (!(flags & FLAG_WINDOWED)) {
2092             char prompt[200];
2093             sprintf(prompt, "%s@%s's password: ", cfg.username, savedhost);
2094             if (!ssh_get_password(prompt, password, sizeof(password))) {
2095                 /*
2096                  * get_password failed to get a password (for
2097                  * example because one was supplied on the command
2098                  * line which has already failed to work).
2099                  * Terminate.
2100                  */
2101                 logevent("No more passwords to try");
2102                 ssh_state = SSH_STATE_CLOSED;
2103                 crReturnV;
2104             }
2105         } else {
2106             c_write("password: ", 10);
2107
2108             pos = 0;
2109             while (pos >= 0) {
2110                 crWaitUntilV(!ispkt);
2111                 while (inlen--) switch (c = *in++) {
2112                   case 10: case 13:
2113                     password[pos] = 0;
2114                     pos = -1;
2115                     break;
2116                   case 8: case 127:
2117                     if (pos > 0)
2118                         pos--;
2119                     break;
2120                   case 21: case 27:
2121                     pos = 0;
2122                     break;
2123                   case 3: case 4:
2124                     random_save_seed();
2125                     exit(0);
2126                     break;
2127                   default:
2128                     if (((c >= ' ' && c <= '~') ||
2129                          ((unsigned char)c >= 160)) && pos < 40)
2130                         password[pos++] = c;
2131                     break;
2132                 }
2133             }
2134             c_write("\r\n", 2);
2135         }
2136
2137         ssh2_pkt_init(SSH2_MSG_USERAUTH_REQUEST);
2138         ssh2_pkt_addstring(username);
2139         ssh2_pkt_addstring("ssh-connection");   /* service requested */
2140         ssh2_pkt_addstring("password");
2141         ssh2_pkt_addbool(FALSE);
2142         ssh2_pkt_addstring(password);
2143         ssh2_pkt_send();
2144
2145         crWaitUntilV(ispkt);
2146         if (pktin.type != SSH2_MSG_USERAUTH_SUCCESS) {
2147             c_write("Access denied\r\n", 15);
2148             logevent("Authentication refused");
2149         } else
2150             break;
2151     }
2152
2153     /*
2154      * Now we're authenticated for the connection protocol. The
2155      * connection protocol will automatically have started at this
2156      * point; there's no need to send SERVICE_REQUEST.
2157      */
2158
2159     /*
2160      * So now create a channel with a session in it.
2161      */
2162     ssh2_pkt_init(SSH2_MSG_CHANNEL_OPEN);
2163     ssh2_pkt_addstring("session");
2164     ssh2_pkt_adduint32(100);           /* as good as any */
2165     ssh2_pkt_adduint32(0xFFFFFFFFUL);  /* very big window which we ignore */
2166     ssh2_pkt_adduint32(0xFFFFFFFFUL);  /* very big max pkt size */
2167     ssh2_pkt_send();
2168     crWaitUntilV(ispkt);
2169     if (pktin.type != SSH2_MSG_CHANNEL_OPEN_CONFIRMATION) {
2170         fatalbox("Server refused to open a session");
2171         /* FIXME: error data comes back in FAILURE packet */
2172     }
2173     if (ssh2_pkt_getuint32() != 100) {
2174         fatalbox("Server's channel confirmation cited wrong channel");
2175     }
2176     ssh_remote_channel = ssh2_pkt_getuint32();
2177     remote_winsize = ssh2_pkt_getuint32();
2178     remote_maxpkt = ssh2_pkt_getuint32();
2179     logevent("Opened channel for session");
2180
2181     /*
2182      * Now allocate a pty for the session.
2183      */
2184     ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
2185     ssh2_pkt_adduint32(ssh_remote_channel); /* recipient channel */
2186     ssh2_pkt_addstring("pty-req");
2187     ssh2_pkt_addbool(1);               /* want reply */
2188     ssh2_pkt_addstring(cfg.termtype);
2189     ssh2_pkt_adduint32(cols);
2190     ssh2_pkt_adduint32(rows);
2191     ssh2_pkt_adduint32(0);             /* pixel width */
2192     ssh2_pkt_adduint32(0);             /* pixel height */
2193     ssh2_pkt_addstring_start();
2194     ssh2_pkt_addstring_data("\0", 1);  /* TTY_OP_END, no special options */
2195     ssh2_pkt_send();
2196
2197     do {                               /* FIXME: pay attention to these */
2198         crWaitUntilV(ispkt);
2199     } while (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST);
2200
2201     if (pktin.type != SSH2_MSG_CHANNEL_SUCCESS) {
2202         if (pktin.type != SSH2_MSG_CHANNEL_FAILURE) {
2203             fatalbox("Server got confused by pty request");
2204         }
2205         c_write("Server refused to allocate pty\r\n", 32);
2206     } else {
2207         logevent("Allocated pty");
2208     }
2209
2210     /*
2211      * Start a shell.
2212      */
2213     ssh2_pkt_init(SSH2_MSG_CHANNEL_REQUEST);
2214     ssh2_pkt_adduint32(ssh_remote_channel); /* recipient channel */
2215     ssh2_pkt_addstring("shell");
2216     ssh2_pkt_addbool(1);               /* want reply */
2217     ssh2_pkt_send();
2218     do {                               /* FIXME: pay attention to these */
2219         crWaitUntilV(ispkt);
2220     } while (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST);
2221     if (pktin.type != SSH2_MSG_CHANNEL_SUCCESS) {
2222         if (pktin.type != SSH2_MSG_CHANNEL_FAILURE) {
2223             fatalbox("Server got confused by shell request");
2224         }
2225         fatalbox("Server refused to start a shell");
2226     } else {
2227         logevent("Started a shell");
2228     }
2229
2230     /*
2231      * Transfer data!
2232      */
2233     ssh_send_ok = 1;
2234     while (1) {
2235         crReturnV;
2236         if (ispkt) {
2237             if (pktin.type == SSH2_MSG_CHANNEL_DATA ||
2238                 pktin.type == SSH2_MSG_CHANNEL_EXTENDED_DATA) {
2239                 char *data;
2240                 int length;
2241                 if (ssh2_pkt_getuint32() != 100)
2242                     continue;          /* wrong channel */
2243                 if (pktin.type == SSH2_MSG_CHANNEL_EXTENDED_DATA &&
2244                     ssh2_pkt_getuint32() != SSH2_EXTENDED_DATA_STDERR)
2245                     continue;          /* extended but not stderr */
2246                 ssh2_pkt_getstring(&data, &length);
2247                 if (data)
2248                     c_write(data, length);
2249             } else if (pktin.type == SSH2_MSG_DISCONNECT) {
2250                 ssh_state = SSH_STATE_CLOSED;
2251                 logevent("Received disconnect request");
2252             } else if (pktin.type == SSH2_MSG_CHANNEL_REQUEST) {
2253                 continue;              /* exit status et al; ignore (FIXME?) */
2254             } else if (pktin.type == SSH2_MSG_CHANNEL_WINDOW_ADJUST) {
2255                 continue;              /* ignore for now (FIXME!) */
2256             } else {
2257                 fatalbox("Strange packet received: type %d", pktin.type);
2258             }
2259         } else {
2260             /* FIXME: for now, ignore window size */
2261             ssh2_pkt_init(SSH2_MSG_CHANNEL_DATA);
2262             ssh2_pkt_adduint32(ssh_remote_channel);
2263             ssh2_pkt_addstring_start();
2264             ssh2_pkt_addstring_data(in, inlen);
2265             ssh2_pkt_send();
2266         }
2267     }
2268
2269     crFinishV;
2270 }
2271
2272 /*
2273  * Handle the top-level SSH2 protocol.
2274  */
2275 static void ssh2_protocol(unsigned char *in, int inlen, int ispkt)
2276 {
2277     if (do_ssh2_transport(in, inlen, ispkt) == 0)
2278         return;
2279     do_ssh2_authconn(in, inlen, ispkt);
2280 }
2281
2282 /*
2283  * Called to set up the connection. Will arrange for WM_NETEVENT
2284  * messages to be passed to the specified window, whose window
2285  * procedure should then call telnet_msg().
2286  *
2287  * Returns an error message, or NULL on success.
2288  */
2289 static char *ssh_init (HWND hwnd, char *host, int port, char **realhost) {
2290     char *p;
2291
2292 #ifdef MSCRYPTOAPI
2293     if(crypto_startup() == 0)
2294         return "Microsoft high encryption pack not installed!";
2295 #endif
2296
2297     p = connect_to_host(host, port, realhost);
2298     if (p != NULL)
2299         return p;
2300
2301     if (!do_ssh_init())
2302         return "Protocol initialisation error";
2303
2304     if (hwnd && WSAAsyncSelect (s, hwnd, WM_NETEVENT, FD_READ | FD_CLOSE) == SOCKET_ERROR)
2305         switch (WSAGetLastError()) {
2306           case WSAENETDOWN: return "Network is down";
2307           default: return "WSAAsyncSelect(): unknown error";
2308         }
2309
2310     return NULL;
2311 }
2312
2313 /*
2314  * Process a WM_NETEVENT message. Will return 0 if the connection
2315  * has closed, or <0 for a socket error.
2316  */
2317 static int ssh_msg (WPARAM wParam, LPARAM lParam) {
2318     int ret;
2319     char buf[256];
2320
2321     /*
2322      * Because reading less than the whole of the available pending
2323      * data can generate an FD_READ event, we need to allow for the
2324      * possibility that FD_READ may arrive with FD_CLOSE already in
2325      * the queue; so it's possible that we can get here even with s
2326      * invalid. If so, we return 1 and don't worry about it.
2327      */
2328     if (s == INVALID_SOCKET)
2329         return 1;
2330
2331     if (WSAGETSELECTERROR(lParam) != 0)
2332         return -WSAGETSELECTERROR(lParam);
2333
2334     switch (WSAGETSELECTEVENT(lParam)) {
2335       case FD_READ:
2336       case FD_CLOSE:
2337         ret = recv(s, buf, sizeof(buf), 0);
2338         if (ret < 0 && WSAGetLastError() == WSAEWOULDBLOCK)
2339             return 1;
2340         if (ret < 0)                   /* any _other_ error */
2341             return -10000-WSAGetLastError();
2342         if (ret == 0) {
2343             s = INVALID_SOCKET;
2344             return 0;
2345         }
2346         ssh_gotdata (buf, ret);
2347         if (ssh_state == SSH_STATE_CLOSED) {
2348             closesocket(s);
2349             s = INVALID_SOCKET;
2350             return 0;
2351         }
2352         return 1;
2353     }
2354     return 1;                          /* shouldn't happen, but WTF */
2355 }
2356
2357 /*
2358  * Called to send data down the Telnet connection.
2359  */
2360 static void ssh_send (char *buf, int len) {
2361     if (s == INVALID_SOCKET)
2362         return;
2363
2364     ssh_protocol(buf, len, 0);
2365 }
2366
2367 /*
2368  * Called to set the size of the window from Telnet's POV.
2369  */
2370 static void ssh_size(void) {
2371     switch (ssh_state) {
2372       case SSH_STATE_BEFORE_SIZE:
2373       case SSH_STATE_CLOSED:
2374         break;                         /* do nothing */
2375       case SSH_STATE_INTERMED:
2376         size_needed = TRUE;            /* buffer for later */
2377         break;
2378       case SSH_STATE_SESSION:
2379         if (!cfg.nopty) {
2380             send_packet(SSH1_CMSG_WINDOW_SIZE,
2381                         PKT_INT, rows, PKT_INT, cols,
2382                         PKT_INT, 0, PKT_INT, 0, PKT_END);
2383         }
2384     }
2385 }
2386
2387 /*
2388  * Send Telnet special codes. TS_EOF is useful for `plink', so you
2389  * can send an EOF and collect resulting output (e.g. `plink
2390  * hostname sort').
2391  */
2392 static void ssh_special (Telnet_Special code) {
2393     if (code == TS_EOF) {
2394         if (ssh_version = 1) {
2395             send_packet(SSH1_CMSG_EOF, PKT_END);
2396         } else {
2397             ssh2_pkt_init(SSH2_MSG_CHANNEL_EOF);
2398             ssh2_pkt_adduint32(ssh_remote_channel);
2399             ssh2_pkt_send();
2400         }
2401     } else {
2402         /* do nothing */
2403     }
2404 }
2405
2406
2407 /*
2408  * Read and decrypt one incoming SSH packet.
2409  * (only used by pSCP)
2410  */
2411 static void get_packet(void)
2412 {
2413     unsigned char buf[4096], *p;
2414     long to_read;
2415     int len;
2416
2417     p = NULL;
2418     len = 0;
2419
2420     while ((to_read = s_rdpkt(&p, &len)) > 0) {
2421         if (to_read > sizeof(buf)) to_read = sizeof(buf);
2422         len = s_read(buf, to_read);
2423         if (len != to_read) {
2424             closesocket(s);
2425             s = INVALID_SOCKET;
2426             return;
2427         }
2428         p = buf;
2429     }
2430
2431     assert(len == 0);
2432 }
2433
2434 /*
2435  * Receive a block of data over the SSH link. Block until
2436  * all data is available. Return nr of bytes read (0 if lost connection).
2437  * (only used by pSCP)
2438  */
2439 int ssh_scp_recv(unsigned char *buf, int len)
2440 {
2441     static int pending_input_len = 0;
2442     static unsigned char *pending_input_ptr;
2443     int to_read = len;
2444
2445     if (pending_input_len >= to_read) {
2446         memcpy(buf, pending_input_ptr, to_read);
2447         pending_input_ptr += to_read;
2448         pending_input_len -= to_read;
2449         return len;
2450     }
2451
2452     if (pending_input_len > 0) {
2453         memcpy(buf, pending_input_ptr, pending_input_len);
2454         buf += pending_input_len;
2455         to_read -= pending_input_len;
2456         pending_input_len = 0;
2457     }
2458
2459     if (s == INVALID_SOCKET)
2460         return 0;
2461     while (to_read > 0) {
2462         get_packet();
2463         if (s == INVALID_SOCKET)
2464             return 0;
2465         if (pktin.type == SSH1_SMSG_STDOUT_DATA) {
2466             int plen = GET_32BIT(pktin.body);
2467             if (plen <= to_read) {
2468                 memcpy(buf, pktin.body + 4, plen);
2469                 buf += plen;
2470                 to_read -= plen;
2471             } else {
2472                 memcpy(buf, pktin.body + 4, to_read);
2473                 pending_input_len = plen - to_read;
2474                 pending_input_ptr = pktin.body + 4 + to_read;
2475                 to_read = 0;
2476             }
2477         } else if (pktin.type == SSH1_SMSG_STDERR_DATA) {
2478             int plen = GET_32BIT(pktin.body);
2479             fwrite(pktin.body + 4, plen, 1, stderr);
2480         } else if (pktin.type == SSH1_MSG_DISCONNECT) {
2481                 logevent("Received disconnect request");
2482         } else if (pktin.type == SSH1_SMSG_SUCCESS ||
2483                    pktin.type == SSH1_SMSG_FAILURE) {
2484                 /* ignore */
2485         } else if (pktin.type == SSH1_SMSG_EXIT_STATUS) {
2486             char logbuf[100];
2487             sprintf(logbuf, "Remote exit status: %d", GET_32BIT(pktin.body));
2488             logevent(logbuf);
2489             send_packet(SSH1_CMSG_EXIT_CONFIRMATION, PKT_END);
2490             logevent("Closing connection");
2491             closesocket(s);
2492             s = INVALID_SOCKET;
2493         } else {
2494             fatalbox("Strange packet received: type %d", pktin.type);
2495         }
2496     }
2497
2498     return len;
2499 }
2500
2501 /*
2502  * Send a block of data over the SSH link.
2503  * Block until all data is sent.
2504  * (only used by pSCP)
2505  */
2506 void ssh_scp_send(unsigned char *buf, int len)
2507 {
2508     if (s == INVALID_SOCKET)
2509         return;
2510     send_packet(SSH1_CMSG_STDIN_DATA,
2511                 PKT_INT, len, PKT_DATA, buf, len, PKT_END);
2512 }
2513
2514 /*
2515  * Send an EOF notification to the server.
2516  * (only used by pSCP)
2517  */
2518 void ssh_scp_send_eof(void)
2519 {
2520     if (s == INVALID_SOCKET)
2521         return;
2522     send_packet(SSH1_CMSG_EOF, PKT_END);
2523 }
2524
2525 /*
2526  * Set up the connection, login on the remote host and
2527  * start execution of a command.
2528  * Returns an error message, or NULL on success.
2529  * (only used by pSCP)
2530  */
2531 char *ssh_scp_init(char *host, int port, char *cmd, char **realhost)
2532 {
2533     char buf[160], *p;
2534
2535 #ifdef MSCRYPTOAPI
2536     if (crypto_startup() == 0)
2537         return "Microsoft high encryption pack not installed!";
2538 #endif
2539
2540     p = connect_to_host(host, port, realhost);
2541     if (p != NULL)
2542         return p;
2543
2544     random_init();
2545
2546     if (!do_ssh_init())
2547         return "Protocol initialisation error";
2548
2549     /* Exchange keys and login */
2550     do {
2551         get_packet();
2552         if (s == INVALID_SOCKET)
2553             return "Connection closed by remote host";
2554     } while (!do_ssh1_login(NULL, 0, 1));
2555
2556     if (ssh_state == SSH_STATE_CLOSED) {
2557         closesocket(s);
2558         s = INVALID_SOCKET;
2559         return "Session initialisation error";
2560     }
2561
2562     /* Execute command */
2563     sprintf(buf, "Sending command: %.100s", cmd);
2564     logevent(buf);
2565     send_packet(SSH1_CMSG_EXEC_CMD, PKT_STR, cmd, PKT_END);
2566
2567     return NULL;
2568 }
2569
2570 static SOCKET ssh_socket(void) { return s; }
2571
2572 static int ssh_sendok(void) { return ssh_send_ok; }
2573
2574 Backend ssh_backend = {
2575     ssh_init,
2576     ssh_msg,
2577     ssh_send,
2578     ssh_size,
2579     ssh_special,
2580     ssh_socket,
2581     ssh_sendok
2582 };