]> andersk Git - openssh.git/blobdiff - packet.c
- markus@cvs.openbsd.org 2001/06/12 16:10:38
[openssh.git] / packet.c
index f055c4f2e244f0dab6e9c7342f393154fb6dcc04..d3a43609ba94b4b54f5aa2e080eaa84fdf5c0326 100644 (file)
--- a/packet.c
+++ b/packet.c
  */
 
 #include "includes.h"
-RCSID("$OpenBSD: packet.c,v 1.40 2000/12/19 23:17:57 markus Exp $");
+RCSID("$OpenBSD: packet.c,v 1.65 2001/06/07 20:23:04 markus Exp $");
 
 #include "xmalloc.h"
 #include "buffer.h"
 #include "packet.h"
 #include "bufaux.h"
-#include "ssh.h"
 #include "crc32.h"
 #include "getput.h"
 
@@ -52,15 +51,14 @@ RCSID("$OpenBSD: packet.c,v 1.40 2000/12/19 23:17:57 markus Exp $");
 #include "channels.h"
 
 #include "compat.h"
+#include "ssh1.h"
 #include "ssh2.h"
 
-#include <openssl/bn.h>
-#include <openssl/dh.h>
-#include <openssl/hmac.h>
-#include "buffer.h"
 #include "cipher.h"
 #include "kex.h"
-#include "hmac.h"
+#include "mac.h"
+#include "log.h"
+#include "canohost.h"
 
 #ifdef PACKET_DEBUG
 #define DBG(x) x
@@ -77,12 +75,6 @@ RCSID("$OpenBSD: packet.c,v 1.40 2000/12/19 23:17:57 markus Exp $");
 static int connection_in = -1;
 static int connection_out = -1;
 
-/*
- * Cipher type.  This value is only used to determine whether to pad the
- * packets with zeroes or random data.
- */
-static int cipher_type = SSH_CIPHER_NONE;
-
 /* Protocol flags for the remote side. */
 static u_int remote_protocol_flags = 0;
 
@@ -106,6 +98,7 @@ static Buffer incoming_packet;
 
 /* Scratch buffer for packet compression/decompression. */
 static Buffer compression_buffer;
+static int compression_buffer_ready = 0;
 
 /* Flag indicating whether packet compression/decompression is enabled. */
 static int packet_compression = 0;
@@ -119,40 +112,8 @@ static int initialized = 0;
 /* Set to true if the connection is interactive. */
 static int interactive_mode = 0;
 
-/* True if SSH2 packet format is used */
-int use_ssh2_packet_format = 0;
-
 /* Session key information for Encryption and MAC */
-Kex    *kex = NULL;
-
-void
-packet_set_kex(Kex *k)
-{
-       if( k->mac[MODE_IN ].key == NULL ||
-           k->enc[MODE_IN ].key == NULL ||
-           k->enc[MODE_IN ].iv  == NULL ||
-           k->mac[MODE_OUT].key == NULL ||
-           k->enc[MODE_OUT].key == NULL ||
-           k->enc[MODE_OUT].iv  == NULL)
-               fatal("bad KEX");
-       kex = k;
-}
-void
-clear_enc_keys(Enc *enc, int len)
-{
-       memset(enc->iv,  0, len);
-       memset(enc->key, 0, len);
-       xfree(enc->iv);
-       xfree(enc->key);
-       enc->iv = NULL;
-       enc->key = NULL;
-}
-void
-packet_set_ssh2_format(void)
-{
-       DBG(debug("use_ssh2_packet_format"));
-       use_ssh2_packet_format = 1;
-}
+Newkeys *newkeys[MODE_MAX];
 
 /*
  * Sets the descriptors used for communication.  Disables encryption until
@@ -166,9 +127,9 @@ packet_set_connection(int fd_in, int fd_out)
                fatal("packet_set_connection: cannot load cipher 'none'");
        connection_in = fd_in;
        connection_out = fd_out;
-       cipher_type = SSH_CIPHER_NONE;
        cipher_init(&send_context, none, (u_char *) "", 0, NULL, 0);
        cipher_init(&receive_context, none, (u_char *) "", 0, NULL, 0);
+       newkeys[MODE_IN] = newkeys[MODE_OUT] = NULL;
        if (!initialized) {
                initialized = 1;
                buffer_init(&input);
@@ -272,7 +233,7 @@ packet_close()
        buffer_free(&output);
        buffer_free(&outgoing_packet);
        buffer_free(&incoming_packet);
-       if (packet_compression) {
+       if (compression_buffer_ready) {
                buffer_free(&compression_buffer);
                buffer_compress_uninit();
        }
@@ -284,7 +245,6 @@ void
 packet_set_protocol_flags(u_int protocol_flags)
 {
        remote_protocol_flags = protocol_flags;
-       channel_set_options((protocol_flags & SSH_PROTOFLAG_HOST_IN_FWD_OPEN) != 0);
 }
 
 /* Returns the remote protocol flags set earlier by the above function. */
@@ -300,47 +260,24 @@ packet_get_protocol_flags()
  * Level is compression level 1 (fastest) - 9 (slow, best) as in gzip.
  */
 
-/*** XXXXX todo: kex means re-init */
 void
-packet_start_compression(int level)
+packet_init_compression()
 {
-       if (packet_compression)
-               fatal("Compression already enabled.");
-       packet_compression = 1;
+       if (compression_buffer_ready == 1)
+               return;
+       compression_buffer_ready = 1;
        buffer_init(&compression_buffer);
-       buffer_compress_init(level);
-}
-
-/*
- * Encrypts the given number of bytes, copying from src to dest. bytes is
- * known to be a multiple of 8.
- */
-
-void
-packet_encrypt(CipherContext * cc, void *dest, void *src,
-    u_int bytes)
-{
-       cipher_encrypt(cc, dest, src, bytes);
 }
 
-/*
- * Decrypts the given number of bytes, copying from src to dest. bytes is
- * known to be a multiple of 8.
- */
-
 void
-packet_decrypt(CipherContext *context, void *dest, void *src, u_int bytes)
+packet_start_compression(int level)
 {
-       /*
-        * Cryptographic attack detector for ssh - Modifications for packet.c
-        * (C)1998 CORE-SDI, Buenos Aires Argentina Ariel Futoransky(futo@core-sdi.com)
-        */
-       if (!compat20 &&
-           context->cipher->number != SSH_CIPHER_NONE &&
-           detect_attack(src, bytes, NULL) == DEATTACK_DETECTED)
-               packet_disconnect("crc32 compensation attack: network attack detected");
-
-       cipher_decrypt(context, dest, src, bytes);
+       if (packet_compression && !compat20)
+               fatal("Compression already enabled.");
+       packet_compression = 1;
+       packet_init_compression();
+       buffer_compress_init_send(level);
+       buffer_compress_init_recv();
 }
 
 /*
@@ -348,7 +285,6 @@ packet_decrypt(CipherContext *context, void *dest, void *src, u_int bytes)
  * key is used for both sending and reception.  However, both directions are
  * encrypted independently of each other.
  */
-
 void
 packet_set_encryption_key(const u_char *key, u_int keylen,
     int number)
@@ -362,61 +298,33 @@ packet_set_encryption_key(const u_char *key, u_int keylen,
        cipher_init(&send_context, cipher, key, keylen, NULL, 0);
 }
 
-/* Starts constructing a packet to send. */
-
-void
-packet_start1(int type)
-{
-       char buf[9];
-
-       buffer_clear(&outgoing_packet);
-       memset(buf, 0, 8);
-       buf[8] = type;
-       buffer_append(&outgoing_packet, buf, 9);
-}
-
+/* Start constructing a packet to send. */
 void
-packet_start2(int type)
+packet_start(u_char type)
 {
-       char buf[4+1+1];
+       u_char buf[9];
+       int len;
 
+       DBG(debug("packet_start[%d]", type));
+       len = compat20 ? 6 : 9;
+       memset(buf, 0, len - 1);
+       buf[len - 1] = type;
        buffer_clear(&outgoing_packet);
-       memset(buf, 0, sizeof buf);
-       /* buf[0..3] = payload_len; */
-       /* buf[4] =    pad_len; */
-       buf[5] = type & 0xff;
-       buffer_append(&outgoing_packet, buf, sizeof buf);
-}
-
-void
-packet_start(int type)
-{
-       DBG(debug("packet_start[%d]",type));
-       if (use_ssh2_packet_format)
-               packet_start2(type);
-       else
-               packet_start1(type);
+       buffer_append(&outgoing_packet, buf, len);
 }
 
-/* Appends a character to the packet data. */
-
+/* Append payload. */
 void
 packet_put_char(int value)
 {
        char ch = value;
        buffer_append(&outgoing_packet, &ch, 1);
 }
-
-/* Appends an integer to the packet data. */
-
 void
 packet_put_int(u_int value)
 {
        buffer_put_int(&outgoing_packet, value);
 }
-
-/* Appends a string to packet data. */
-
 void
 packet_put_string(const char *buf, u_int len)
 {
@@ -425,18 +333,13 @@ packet_put_string(const char *buf, u_int len)
 void
 packet_put_cstring(const char *str)
 {
-       buffer_put_string(&outgoing_packet, str, strlen(str));
+       buffer_put_cstring(&outgoing_packet, str);
 }
-
 void
 packet_put_raw(const char *buf, u_int len)
 {
        buffer_append(&outgoing_packet, buf, len);
 }
-
-
-/* Appends an arbitrary precision integer to packet data. */
-
 void
 packet_put_bignum(BIGNUM * value)
 {
@@ -454,7 +357,7 @@ packet_put_bignum2(BIGNUM * value)
  */
 
 void
-packet_send1()
+packet_send1(void)
 {
        char buf[8], *cp;
        int i, padding, len;
@@ -481,7 +384,7 @@ packet_send1()
 
        /* Insert padding. Initialized to zero in packet_start1() */
        padding = 8 - len % 8;
-       if (cipher_type != SSH_CIPHER_NONE) {
+       if (send_context.cipher->number != SSH_CIPHER_NONE) {
                cp = buffer_ptr(&outgoing_packet);
                for (i = 0; i < padding; i++) {
                        if (i % 4 == 0)
@@ -507,7 +410,7 @@ packet_send1()
        PUT_32BIT(buf, len);
        buffer_append(&output, buf, 4);
        buffer_append_space(&output, &cp, buffer_len(&outgoing_packet));
-       packet_encrypt(&send_context, cp, buffer_ptr(&outgoing_packet),
+       cipher_encrypt(&send_context, cp, buffer_ptr(&outgoing_packet),
                       buffer_len(&outgoing_packet));
 
 #ifdef PACKET_DEBUG
@@ -524,33 +427,82 @@ packet_send1()
         */
 }
 
+void
+set_newkeys(int mode)
+{
+       Enc *enc;
+       Mac *mac;
+       Comp *comp;
+       CipherContext *cc;
+
+       debug("newkeys: mode %d", mode);
+
+       cc = (mode == MODE_OUT) ? &send_context : &receive_context;
+       if (newkeys[mode] != NULL) {
+               debug("newkeys: rekeying");
+               /* todo: free old keys, reset compression/cipher-ctxt; */
+               memset(cc, 0, sizeof(*cc));
+               enc  = &newkeys[mode]->enc;
+               mac  = &newkeys[mode]->mac;
+               comp = &newkeys[mode]->comp;
+               memset(mac->key, 0, mac->key_len);
+               xfree(enc->name);
+               xfree(enc->iv);
+               xfree(enc->key);
+               xfree(mac->name);
+               xfree(mac->key);
+               xfree(comp->name);
+               xfree(newkeys[mode]);
+       }
+       newkeys[mode] = kex_get_newkeys(mode);
+       if (newkeys[mode] == NULL)
+               fatal("newkeys: no keys for mode %d", mode);
+       enc  = &newkeys[mode]->enc;
+       mac  = &newkeys[mode]->mac;
+       comp = &newkeys[mode]->comp;
+       if (mac->md != NULL)
+               mac->enabled = 1;
+       DBG(debug("cipher_init_context: %d", mode));
+       cipher_init(cc, enc->cipher, enc->key, enc->cipher->key_len,
+           enc->iv, enc->cipher->block_size);
+       memset(enc->iv,  0, enc->cipher->block_size);
+       memset(enc->key, 0, enc->cipher->key_len);
+       if (comp->type != 0 && comp->enabled == 0) {
+               packet_init_compression();
+               if (mode == MODE_OUT)
+                       buffer_compress_init_send(6);
+               else
+                       buffer_compress_init_recv();
+               comp->enabled = 1;
+       }
+}
+
 /*
  * Finalize packet in SSH2 format (compress, mac, encrypt, enqueue)
  */
 void
-packet_send2()
+packet_send2(void)
 {
-       u_char *macbuf = NULL;
+       static u_int32_t seqnr = 0;
+       u_char type, *ucp, *macbuf = NULL;
        char *cp;
        u_int packet_length = 0;
        u_int i, padlen, len;
        u_int32_t rand = 0;
-       static u_int seqnr = 0;
-       int type;
        Enc *enc   = NULL;
        Mac *mac   = NULL;
        Comp *comp = NULL;
        int block_size;
 
-       if (kex != NULL) {
-               enc  = &kex->enc[MODE_OUT];
-               mac  = &kex->mac[MODE_OUT];
-               comp = &kex->comp[MODE_OUT];
+       if (newkeys[MODE_OUT] != NULL) {
+               enc  = &newkeys[MODE_OUT]->enc;
+               mac  = &newkeys[MODE_OUT]->mac;
+               comp = &newkeys[MODE_OUT]->comp;
        }
        block_size = enc ? enc->cipher->block_size : 8;
 
-       cp = buffer_ptr(&outgoing_packet);
-       type = cp[5] & 0xff;
+       ucp = (u_char *) buffer_ptr(&outgoing_packet);
+       type = ucp[5];
 
 #ifdef PACKET_DEBUG
        fprintf(stderr, "plain:     ");
@@ -588,7 +540,7 @@ packet_send2()
                        if (i % 4 == 0)
                                rand = arc4random();
                        cp[i] = rand & 0xff;
-                       rand <<= 8;
+                       rand >>= 8;
                }
        } else {
                /* clear padding */
@@ -596,23 +548,21 @@ packet_send2()
        }
        /* packet_length includes payload, padding and padding length field */
        packet_length = buffer_len(&outgoing_packet) - 4;
-       cp = buffer_ptr(&outgoing_packet);
-       PUT_32BIT(cp, packet_length);
-       cp[4] = padlen & 0xff;
+       ucp = (u_char *)buffer_ptr(&outgoing_packet);
+       PUT_32BIT(ucp, packet_length);
+       ucp[4] = padlen;
        DBG(debug("send: len %d (includes padlen %d)", packet_length+4, padlen));
 
        /* compute MAC over seqnr and packet(length fields, payload, padding) */
        if (mac && mac->enabled) {
-               macbuf = hmac( mac->md, seqnr,
+               macbuf = mac_compute(mac, seqnr,
                    (u_char *) buffer_ptr(&outgoing_packet),
-                   buffer_len(&outgoing_packet),
-                   mac->key, mac->key_len
-               );
+                   buffer_len(&outgoing_packet));
                DBG(debug("done calc MAC out #%d", seqnr));
        }
        /* encrypt packet and append to output buffer. */
        buffer_append_space(&output, &cp, buffer_len(&outgoing_packet));
-       packet_encrypt(&send_context, cp, buffer_ptr(&outgoing_packet),
+       cipher_encrypt(&send_context, cp, buffer_ptr(&outgoing_packet),
            buffer_len(&outgoing_packet));
        /* append unencrypted MAC */
        if (mac && mac->enabled)
@@ -626,28 +576,14 @@ packet_send2()
                log("outgoing seqnr wraps around");
        buffer_clear(&outgoing_packet);
 
-       if (type == SSH2_MSG_NEWKEYS) {
-               if (kex==NULL || mac==NULL || enc==NULL || comp==NULL)
-                       fatal("packet_send2: no KEX");
-               if (mac->md != NULL)
-                       mac->enabled = 1;
-               DBG(debug("cipher_init send_context"));
-               cipher_init(&send_context, enc->cipher,
-                   enc->key, enc->cipher->key_len,
-                   enc->iv, enc->cipher->block_size);
-               clear_enc_keys(enc, kex->we_need);
-               if (comp->type != 0 && comp->enabled == 0) {
-                       comp->enabled = 1;
-                       if (! packet_compression)
-                               packet_start_compression(6);
-               }
-       }
+       if (type == SSH2_MSG_NEWKEYS)
+               set_newkeys(MODE_OUT);
 }
 
 void
 packet_send()
 {
-       if (use_ssh2_packet_format)
+       if (compat20)
                packet_send2();
        else
                packet_send1();
@@ -664,10 +600,13 @@ int
 packet_read(int *payload_len_ptr)
 {
        int type, len;
-       fd_set set;
+       fd_set *setp;
        char buf[8192];
        DBG(debug("packet_read()"));
 
+       setp = (fd_set *)xmalloc(howmany(connection_in+1, NFDBITS) *
+           sizeof(fd_mask));
+
        /* Since we are blocking, ensure that all written packets have been sent. */
        packet_write_wait();
 
@@ -675,24 +614,29 @@ packet_read(int *payload_len_ptr)
        for (;;) {
                /* Try to read a packet from the buffer. */
                type = packet_read_poll(payload_len_ptr);
-               if (!use_ssh2_packet_format && (
+               if (!compat20 && (
                    type == SSH_SMSG_SUCCESS
                    || type == SSH_SMSG_FAILURE
                    || type == SSH_CMSG_EOF
                    || type == SSH_CMSG_EXIT_CONFIRMATION))
                        packet_integrity_check(*payload_len_ptr, 0, type);
                /* If we got a packet, return it. */
-               if (type != SSH_MSG_NONE)
+               if (type != SSH_MSG_NONE) {
+                       xfree(setp);
                        return type;
+               }
                /*
                 * Otherwise, wait for some data to arrive, add it to the
                 * buffer, and try again.
                 */
-               FD_ZERO(&set);
-               FD_SET(connection_in, &set);
+               memset(setp, 0, howmany(connection_in + 1, NFDBITS) *
+                   sizeof(fd_mask));
+               FD_SET(connection_in, setp);
 
                /* Wait for some data to arrive. */
-               select(connection_in + 1, &set, NULL, NULL, NULL);
+               while (select(connection_in + 1, setp, NULL, NULL, NULL) == -1 &&
+                   (errno == EAGAIN || errno == EINTR))
+                       ;
 
                /* Read data from the socket. */
                len = read(connection_in, buf, sizeof(buf));
@@ -743,8 +687,8 @@ int
 packet_read_poll1(int *payload_len_ptr)
 {
        u_int len, padded_len;
-       u_char *ucp;
-       char buf[8], *cp;
+       u_char *ucp, type;
+       char *cp;
        u_int checksum, stored_checksum;
 
        /* Check if input size is less than minimum packet size. */
@@ -766,10 +710,20 @@ packet_read_poll1(int *payload_len_ptr)
        /* Consume packet length. */
        buffer_consume(&input, 4);
 
-       /* Copy data to incoming_packet. */
+       /*
+        * Cryptographic attack detector for ssh
+        * (C)1998 CORE-SDI, Buenos Aires Argentina
+        * Ariel Futoransky(futo@core-sdi.com)
+        */
+       if (receive_context.cipher->number != SSH_CIPHER_NONE &&
+           detect_attack(buffer_ptr(&input), padded_len, NULL) == DEATTACK_DETECTED)
+               packet_disconnect("crc32 compensation attack: network attack detected");
+
+       /* Decrypt data to incoming_packet. */
        buffer_clear(&incoming_packet);
        buffer_append_space(&incoming_packet, &cp, padded_len);
-       packet_decrypt(&receive_context, cp, buffer_ptr(&input), padded_len);
+       cipher_decrypt(&receive_context, cp, buffer_ptr(&input), padded_len);
+
        buffer_consume(&input, padded_len);
 
 #ifdef PACKET_DEBUG
@@ -785,7 +739,6 @@ packet_read_poll1(int *payload_len_ptr)
        buffer_consume(&incoming_packet, 8 - len % 8);
 
        /* Test check bytes. */
-
        if (len != buffer_len(&incoming_packet))
                packet_disconnect("packet_read_poll: len %d != buffer_len %d.",
                    len, buffer_len(&incoming_packet));
@@ -796,7 +749,6 @@ packet_read_poll1(int *payload_len_ptr)
                packet_disconnect("Corrupted check bytes on input.");
        buffer_consume_end(&incoming_packet, 4);
 
-       /* If using packet compression, decompress the packet. */
        if (packet_compression) {
                buffer_clear(&compression_buffer);
                buffer_uncompress(&incoming_packet, &compression_buffer);
@@ -804,35 +756,28 @@ packet_read_poll1(int *payload_len_ptr)
                buffer_append(&incoming_packet, buffer_ptr(&compression_buffer),
                    buffer_len(&compression_buffer));
        }
-       /* Get packet type. */
-       buffer_get(&incoming_packet, &buf[0], 1);
-
-       /* Return length of payload (without type field). */
+       type = buffer_get_char(&incoming_packet);
        *payload_len_ptr = buffer_len(&incoming_packet);
-
-       /* Return type. */
-       return (u_char) buf[0];
+       return type;
 }
 
 int
 packet_read_poll2(int *payload_len_ptr)
 {
+       static u_int32_t seqnr = 0;
+       static u_int packet_length = 0;
        u_int padlen, need;
-       u_char buf[8], *macbuf;
-       u_char *ucp;
+       u_char *macbuf, *ucp, type;
        char *cp;
-       static u_int packet_length = 0;
-       static u_int seqnr = 0;
-       int type;
        int maclen, block_size;
        Enc *enc   = NULL;
        Mac *mac   = NULL;
        Comp *comp = NULL;
 
-       if (kex != NULL) {
-               enc  = &kex->enc[MODE_IN];
-               mac  = &kex->mac[MODE_IN];
-               comp = &kex->comp[MODE_IN];
+       if (newkeys[MODE_IN] != NULL) {
+               enc  = &newkeys[MODE_IN]->enc;
+               mac  = &newkeys[MODE_IN]->mac;
+               comp = &newkeys[MODE_IN]->comp;
        }
        maclen = mac && mac->enabled ? mac->mac_len : 0;
        block_size = enc ? enc->cipher->block_size : 8;
@@ -846,7 +791,7 @@ packet_read_poll2(int *payload_len_ptr)
                        return SSH_MSG_NONE;
                buffer_clear(&incoming_packet);
                buffer_append_space(&incoming_packet, &cp, block_size);
-               packet_decrypt(&receive_context, cp, buffer_ptr(&input),
+               cipher_decrypt(&receive_context, cp, buffer_ptr(&input),
                    block_size);
                ucp = (u_char *) buffer_ptr(&incoming_packet);
                packet_length = GET_32BIT(ucp);
@@ -875,18 +820,16 @@ packet_read_poll2(int *payload_len_ptr)
        buffer_dump(&input);
 #endif
        buffer_append_space(&incoming_packet, &cp, need);
-       packet_decrypt(&receive_context, cp, buffer_ptr(&input), need);
+       cipher_decrypt(&receive_context, cp, buffer_ptr(&input), need);
        buffer_consume(&input, need);
        /*
         * compute MAC over seqnr and packet,
         * increment sequence number for incoming packet
         */
        if (mac && mac->enabled) {
-               macbuf = hmac( mac->md, seqnr,
+               macbuf = mac_compute(mac, seqnr,
                    (u_char *) buffer_ptr(&incoming_packet),
-                   buffer_len(&incoming_packet),
-                   mac->key, mac->key_len
-               );
+                   buffer_len(&incoming_packet));
                if (memcmp(macbuf, buffer_ptr(&input), mac->mac_len) != 0)
                        packet_disconnect("Corrupted MAC on input.");
                DBG(debug("MAC #%d ok", seqnr));
@@ -897,7 +840,7 @@ packet_read_poll2(int *payload_len_ptr)
 
        /* get padlen */
        cp = buffer_ptr(&incoming_packet) + 4;
-       padlen = *cp & 0xff;
+       padlen = (u_char) *cp;
        DBG(debug("input: padlen %d", padlen));
        if (padlen < 4)
                packet_disconnect("Corrupted padlen %d on input.", padlen);
@@ -919,51 +862,30 @@ packet_read_poll2(int *payload_len_ptr)
         * get packet type, implies consume.
         * return length of payload (without type field)
         */
-       buffer_get(&incoming_packet, (char *)&buf[0], 1);
+       type = buffer_get_char(&incoming_packet);
+       if (type == SSH2_MSG_NEWKEYS)
+               set_newkeys(MODE_IN);
        *payload_len_ptr = buffer_len(&incoming_packet);
-
-       /* reset for next packet */
-       packet_length = 0;
-
-       /* extract packet type */
-       type = (u_char)buf[0];
-
-       if (type == SSH2_MSG_NEWKEYS) {
-               if (kex==NULL || mac==NULL || enc==NULL || comp==NULL)
-                       fatal("packet_read_poll2: no KEX");
-               if (mac->md != NULL)
-                       mac->enabled = 1;
-               DBG(debug("cipher_init receive_context"));
-               cipher_init(&receive_context, enc->cipher,
-                   enc->key, enc->cipher->key_len,
-                   enc->iv, enc->cipher->block_size);
-               clear_enc_keys(enc, kex->we_need);
-               if (comp->type != 0 && comp->enabled == 0) {
-                       comp->enabled = 1;
-                       if (! packet_compression)
-                               packet_start_compression(6);
-               }
-       }
-
 #ifdef PACKET_DEBUG
-       fprintf(stderr, "read/plain[%d]:\r\n",type);
+       fprintf(stderr, "read/plain[%d]:\r\n", type);
        buffer_dump(&incoming_packet);
 #endif
-       return (u_char)type;
+       /* reset for next packet */
+       packet_length = 0;
+       return type;
 }
 
 int
 packet_read_poll(int *payload_len_ptr)
 {
+       int reason;
+       u_char type;
        char *msg;
-       for (;;) {
-               int type = use_ssh2_packet_format ?
-                   packet_read_poll2(payload_len_ptr):
-                   packet_read_poll1(payload_len_ptr);
 
-               if(compat20) {
-                       int reason;
-                       if (type != 0)
+       for (;;) {
+               if (compat20) {
+                       type = packet_read_poll2(payload_len_ptr);
+                       if (type)
                                DBG(debug("received packet type %d", type));
                        switch(type) {
                        case SSH2_MSG_IGNORE:
@@ -979,15 +901,17 @@ packet_read_poll(int *payload_len_ptr)
                        case SSH2_MSG_DISCONNECT:
                                reason = packet_get_int();
                                msg = packet_get_string(NULL);
-                               log("Received disconnect: %d: %.900s", reason, msg);
+                               log("Received disconnect from %s: %d: %.400s", get_remote_ipaddr(),
+                                       reason, msg);
                                xfree(msg);
                                fatal_cleanup();
                                break;
                        default:
                                return type;
                                break;
-                       }       
+                       }
                } else {
+                       type = packet_read_poll1(payload_len_ptr);
                        switch(type) {
                        case SSH_MSG_IGNORE:
                                break;
@@ -998,16 +922,17 @@ packet_read_poll(int *payload_len_ptr)
                                break;
                        case SSH_MSG_DISCONNECT:
                                msg = packet_get_string(NULL);
-                               log("Received disconnect: %.900s", msg);
+                               log("Received disconnect from %s: %.400s", get_remote_ipaddr(),
+                                       msg);
                                fatal_cleanup();
                                xfree(msg);
                                break;
                        default:
-                               if (type != 0)
+                               if (type)
                                        DBG(debug("received packet type %d", type));
                                return type;
                                break;
-                       }       
+                       }
                }
        }
 }
@@ -1154,7 +1079,7 @@ packet_disconnect(const char *fmt,...)
                packet_put_cstring("");
        } else {
                packet_start(SSH_MSG_DISCONNECT);
-               packet_put_string(buf, strlen(buf));
+               packet_put_cstring(buf);
        }
        packet_send();
        packet_write_wait();
@@ -1196,14 +1121,21 @@ packet_write_poll()
 void
 packet_write_wait()
 {
+       fd_set *setp;
+
+       setp = (fd_set *)xmalloc(howmany(connection_out + 1, NFDBITS) *
+           sizeof(fd_mask));
        packet_write_poll();
        while (packet_have_data_to_write()) {
-               fd_set set;
-               FD_ZERO(&set);
-               FD_SET(connection_out, &set);
-               select(connection_out + 1, NULL, &set, NULL, NULL);
+               memset(setp, 0, howmany(connection_out + 1, NFDBITS) *
+                   sizeof(fd_mask));
+               FD_SET(connection_out, setp);
+               while (select(connection_out + 1, NULL, setp, NULL, NULL) == -1 &&
+                   (errno == EAGAIN || errno == EINTR))
+                       ;
                packet_write_poll();
        }
+       xfree(setp);
 }
 
 /* Returns true if there is buffered data to write to the connection. */
@@ -1228,48 +1160,50 @@ packet_not_very_much_data_to_write()
 /* Informs that the current session is interactive.  Sets IP flags for that. */
 
 void
-packet_set_interactive(int interactive, int keepalives)
+packet_set_interactive(int interactive)
 {
+       static int called = 0;
+#if defined(IP_TOS) && !defined(IP_TOS_IS_BROKEN)
+       int lowdelay = IPTOS_LOWDELAY;
+       int throughput = IPTOS_THROUGHPUT;
+#endif
        int on = 1;
 
+       if (called)
+               return;
+       called = 1;
+
        /* Record that we are in interactive mode. */
        interactive_mode = interactive;
 
        /* Only set socket options if using a socket.  */
        if (!packet_connection_is_on_socket())
                return;
-       if (keepalives) {
-               /* Set keepalives if requested. */
-               if (setsockopt(connection_in, SOL_SOCKET, SO_KEEPALIVE, (void *) &on,
-                   sizeof(on)) < 0)
-                       error("setsockopt SO_KEEPALIVE: %.100s", strerror(errno));
-       }
        /*
-        * IPTOS_LOWDELAY, TCP_NODELAY and IPTOS_THROUGHPUT are IPv4 only
+        * IPTOS_LOWDELAY and IPTOS_THROUGHPUT are IPv4 only
         */
-       if (!packet_connection_is_ipv4())
-               return;
        if (interactive) {
                /*
                 * Set IP options for an interactive connection.  Use
                 * IPTOS_LOWDELAY and TCP_NODELAY.
                 */
 #if defined(IP_TOS) && !defined(IP_TOS_IS_BROKEN)
-               int lowdelay = IPTOS_LOWDELAY;
-               if (setsockopt(connection_in, IPPROTO_IP, IP_TOS, (void *) &lowdelay,
-                   sizeof(lowdelay)) < 0)
-                       error("setsockopt IPTOS_LOWDELAY: %.100s", strerror(errno));
+               if (packet_connection_is_ipv4()) {
+                       if (setsockopt(connection_in, IPPROTO_IP, IP_TOS,
+                           (void *) &lowdelay, sizeof(lowdelay)) < 0)
+                               error("setsockopt IPTOS_LOWDELAY: %.100s",
+                                   strerror(errno));
+               }
 #endif
                if (setsockopt(connection_in, IPPROTO_TCP, TCP_NODELAY, (void *) &on,
                    sizeof(on)) < 0)
                        error("setsockopt TCP_NODELAY: %.100s", strerror(errno));
-       } else {
+       } else if (packet_connection_is_ipv4()) {
                /*
                 * Set IP options for a non-interactive connection.  Use
                 * IPTOS_THROUGHPUT.
                 */
 #if defined(IP_TOS) && !defined(IP_TOS_IS_BROKEN)
-               int throughput = IPTOS_THROUGHPUT;
                if (setsockopt(connection_in, IPPROTO_IP, IP_TOS, (void *) &throughput,
                    sizeof(throughput)) < 0)
                        error("setsockopt IPTOS_THROUGHPUT: %.100s", strerror(errno));
@@ -1302,3 +1236,65 @@ packet_set_maxsize(int s)
        max_packet_size = s;
        return s;
 }
+
+/*
+ * 9.2.  Ignored Data Message
+ *
+ *   byte      SSH_MSG_IGNORE
+ *   string    data
+ *
+ * All implementations MUST understand (and ignore) this message at any
+ * time (after receiving the protocol version). No implementation is
+ * required to send them. This message can be used as an additional
+ * protection measure against advanced traffic analysis techniques.
+ */
+/* size of current + ignore message should be n*sumlen bytes (w/o mac) */
+void
+packet_inject_ignore(int sumlen)
+{
+       int blocksize, padlen, have, need, nb, mini, nbytes;
+       Enc *enc = NULL;
+
+       if (compat20 == 0)
+               return;
+
+       have = buffer_len(&outgoing_packet);
+       debug2("packet_inject_ignore: current %d", have);
+       if (newkeys[MODE_OUT] != NULL)
+               enc  = &newkeys[MODE_OUT]->enc;
+       blocksize = enc ? enc->cipher->block_size : 8;
+       padlen = blocksize - (have % blocksize);
+       if (padlen < 4)
+               padlen += blocksize;
+       have += padlen;
+       have /= blocksize;      /* # of blocks for current message */
+
+       nb   = roundup(sumlen,  blocksize) / blocksize; /* blocks for both */
+       mini = roundup(5+1+4+4, blocksize) / blocksize; /* minsize ignore msg */
+       need = nb - (have % nb);                        /* blocks for ignore */
+       if (need <= mini)
+               need += nb;
+       nbytes = (need - mini) * blocksize;     /* size of ignore payload */
+       debug2("packet_inject_ignore: block %d have %d nb %d mini %d need %d",
+           blocksize, have, nb, mini, need);
+
+       /* enqueue current message and append a ignore message */
+       packet_send();
+       packet_send_ignore(nbytes);
+}
+
+void
+packet_send_ignore(int nbytes)
+{
+       u_int32_t rand = 0;
+       int i;
+
+       packet_start(compat20 ? SSH2_MSG_IGNORE : SSH_MSG_IGNORE);
+       packet_put_int(nbytes);
+       for(i = 0; i < nbytes; i++) {
+               if (i % 4 == 0)
+                       rand = arc4random();
+               packet_put_char(rand & 0xff);
+               rand >>= 8;
+       }
+}
This page took 0.159298 seconds and 4 git commands to generate.