]> andersk Git - openssh.git/blobdiff - packet.c
- (dtucker) [openbsd-compat/readpassphrase.c] Update to OpenBSD's r1.21.
[openssh.git] / packet.c
index 56080cbb118c4feb40aa12b917cc28b625bcdbe4..994e35b6df50f926455475e6121fa6a16c252a6c 100644 (file)
--- a/packet.c
+++ b/packet.c
@@ -1,46 +1,83 @@
+/* $OpenBSD: packet.c,v 1.166 2009/06/27 09:29:06 andreas Exp $ */
 /*
- *
- * packet.c
- *
  * Author: Tatu Ylonen <ylo@cs.hut.fi>
- *
  * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
  *                    All rights reserved
- *
- * Created: Sat Mar 18 02:40:40 1995 ylo
- *
  * This file contains code implementing the packet protocol and communication
  * with the other side.  This same code is used both on client and server side.
  *
+ * As far as I am concerned, the code I have written for this software
+ * can be used freely for any purpose.  Any derived versions of this
+ * software must be clearly marked as such, and if the derived work is
+ * incompatible with the protocol description in the RFC file, it must be
+ * called by a name other than "ssh" or "Secure Shell".
+ *
+ *
  * SSH2 packet format added by Markus Friedl.
+ * Copyright (c) 2000, 2001 Markus Friedl.  All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
  *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "includes.h"
-RCSID("$OpenBSD: packet.c,v 1.33 2000/06/20 01:39:43 markus Exp $");
+#include <sys/types.h>
+#include "openbsd-compat/sys-queue.h"
+#include <sys/param.h>
+#include <sys/socket.h>
+#ifdef HAVE_SYS_TIME_H
+# include <sys/time.h>
+#endif
+
+#include <netinet/in.h>
+#include <netinet/ip.h>
+#include <arpa/inet.h>
+
+#include <errno.h>
+#include <stdarg.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <signal.h>
 
 #include "xmalloc.h"
 #include "buffer.h"
 #include "packet.h"
-#include "bufaux.h"
-#include "ssh.h"
 #include "crc32.h"
-#include "cipher.h"
-#include "getput.h"
-
 #include "compress.h"
 #include "deattack.h"
 #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 "key.h"
 #include "kex.h"
-#include "hmac.h"
+#include "mac.h"
+#include "log.h"
+#include "canohost.h"
+#include "misc.h"
+#include "ssh.h"
+#include "roaming.h"
 
 #ifdef PACKET_DEBUG
 #define DBG(x) x
@@ -48,90 +85,127 @@ RCSID("$OpenBSD: packet.c,v 1.33 2000/06/20 01:39:43 markus Exp $");
 #define DBG(x)
 #endif
 
-/*
- * This variable contains the file descriptors used for communicating with
- * the other side.  connection_in is used for reading; connection_out for
- * writing.  These can be the same descriptor, in which case it is assumed to
- * be a socket.
- */
-static int connection_in = -1;
-static int connection_out = -1;
+#define PACKET_MAX_SIZE (256 * 1024)
 
-/*
- * 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;
+struct packet_state {
+       u_int32_t seqnr;
+       u_int32_t packets;
+       u_int64_t blocks;
+       u_int64_t bytes;
+};
 
-/* Protocol flags for the remote side. */
-static unsigned int remote_protocol_flags = 0;
+struct packet {
+       TAILQ_ENTRY(packet) next;
+       u_char type;
+       Buffer payload;
+};
 
-/* Encryption context for receiving data.  This is only used for decryption. */
-static CipherContext receive_context;
+struct session_state {
+       /*
+        * This variable contains the file descriptors used for
+        * communicating with the other side.  connection_in is used for
+        * reading; connection_out for writing.  These can be the same
+        * descriptor, in which case it is assumed to be a socket.
+        */
+       int connection_in;
+       int connection_out;
 
-/* Encryption context for sending data.  This is only used for encryption. */
-static CipherContext send_context;
+       /* Protocol flags for the remote side. */
+       u_int remote_protocol_flags;
 
-/* Buffer for raw input data from the socket. */
-static Buffer input;
+       /* Encryption context for receiving data.  Only used for decryption. */
+       CipherContext receive_context;
 
-/* Buffer for raw output data going to the socket. */
-static Buffer output;
+       /* Encryption context for sending data.  Only used for encryption. */
+       CipherContext send_context;
 
-/* Buffer for the partial outgoing packet being constructed. */
-static Buffer outgoing_packet;
+       /* Buffer for raw input data from the socket. */
+       Buffer input;
 
-/* Buffer for the incoming packet currently being processed. */
-static Buffer incoming_packet;
+       /* Buffer for raw output data going to the socket. */
+       Buffer output;
 
-/* Scratch buffer for packet compression/decompression. */
-static Buffer compression_buffer;
+       /* Buffer for the partial outgoing packet being constructed. */
+       Buffer outgoing_packet;
 
-/* Flag indicating whether packet compression/decompression is enabled. */
-static int packet_compression = 0;
+       /* Buffer for the incoming packet currently being processed. */
+       Buffer incoming_packet;
 
-/* default maximum packet size */
-int max_packet_size = 32768;
+       /* Scratch buffer for packet compression/decompression. */
+       Buffer compression_buffer;
+       int compression_buffer_ready;
 
-/* Flag indicating whether this module has been initialized. */
-static int initialized = 0;
+       /*
+        * Flag indicating whether packet compression/decompression is
+        * enabled.
+        */
+       int packet_compression;
 
-/* Set to true if the connection is interactive. */
-static int interactive_mode = 0;
+       /* default maximum packet size */
+       u_int max_packet_size;
 
-/* True if SSH2 packet format is used */
-int use_ssh2_packet_format = 0;
+       /* Flag indicating whether this module has been initialized. */
+       int initialized;
 
-/* Session key information for Encryption and MAC */
-Kex    *kex = NULL;
+       /* Set to true if the connection is interactive. */
+       int interactive_mode;
 
-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)
+       /* Set to true if we are the server side. */
+       int server_side;
+
+       /* Set to true if we are authenticated. */
+       int after_authentication;
+
+       int keep_alive_timeouts;
+
+       /* The maximum time that we will wait to send or receive a packet */
+       int packet_timeout_ms;
+
+       /* Session key information for Encryption and MAC */
+       Newkeys *newkeys[MODE_MAX];
+       struct packet_state p_read, p_send;
+
+       u_int64_t max_blocks_in, max_blocks_out;
+       u_int32_t rekey_limit;
+
+       /* Session key for protocol v1 */
+       u_char ssh1_key[SSH_SESSION_KEY_LENGTH];
+       u_int ssh1_keylen;
+
+       /* roundup current message to extra_pad bytes */
+       u_char extra_pad;
+
+       /* XXX discard incoming data after MAC error */
+       u_int packet_discard;
+       Mac *packet_discard_mac;
+
+       /* Used in packet_read_poll2() */
+       u_int packlen;
+
+       /* Used in packet_send2 */
+       int rekeying;
+
+       /* Used in packet_set_interactive */
+       int set_interactive_called;
+
+       /* Used in packet_set_maxsize */
+       int set_maxsize_called;
+
+       TAILQ_HEAD(, packet) outgoing;
+};
+
+static struct session_state *active_state, *backup_state;
+
+static struct session_state *
+alloc_session_state(void)
 {
-       DBG(debug("use_ssh2_packet_format"));
-       use_ssh2_packet_format = 1;
+    struct session_state *s = xcalloc(1, sizeof(*s));
+
+    s->connection_in = -1;
+    s->connection_out = -1;
+    s->max_packet_size = 32768;
+    s->packet_timeout_ms = -1;
+    return s;
 }
 
 /*
@@ -141,40 +215,96 @@ packet_set_ssh2_format(void)
 void
 packet_set_connection(int fd_in, int fd_out)
 {
-       connection_in = fd_in;
-       connection_out = fd_out;
-       cipher_type = SSH_CIPHER_NONE;
-       cipher_set_key(&send_context, SSH_CIPHER_NONE, (unsigned char *) "", 0);
-       cipher_set_key(&receive_context, SSH_CIPHER_NONE, (unsigned char *) "", 0);
-       if (!initialized) {
-               initialized = 1;
-               buffer_init(&input);
-               buffer_init(&output);
-               buffer_init(&outgoing_packet);
-               buffer_init(&incoming_packet);
+       Cipher *none = cipher_by_name("none");
+
+       if (none == NULL)
+               fatal("packet_set_connection: cannot load cipher 'none'");
+       if (active_state == NULL)
+               active_state = alloc_session_state();
+       active_state->connection_in = fd_in;
+       active_state->connection_out = fd_out;
+       cipher_init(&active_state->send_context, none, (const u_char *)"",
+           0, NULL, 0, CIPHER_ENCRYPT);
+       cipher_init(&active_state->receive_context, none, (const u_char *)"",
+           0, NULL, 0, CIPHER_DECRYPT);
+       active_state->newkeys[MODE_IN] = active_state->newkeys[MODE_OUT] = NULL;
+       if (!active_state->initialized) {
+               active_state->initialized = 1;
+               buffer_init(&active_state->input);
+               buffer_init(&active_state->output);
+               buffer_init(&active_state->outgoing_packet);
+               buffer_init(&active_state->incoming_packet);
+               TAILQ_INIT(&active_state->outgoing);
+               active_state->p_send.packets = active_state->p_read.packets = 0;
+       }
+}
+
+void
+packet_set_timeout(int timeout, int count)
+{
+       if (timeout == 0 || count == 0) {
+               active_state->packet_timeout_ms = -1;
+               return;
+       }
+       if ((INT_MAX / 1000) / count < timeout)
+               active_state->packet_timeout_ms = INT_MAX;
+       else
+               active_state->packet_timeout_ms = timeout * count * 1000;
+}
+
+static void
+packet_stop_discard(void)
+{
+       if (active_state->packet_discard_mac) {
+               char buf[1024];
+               
+               memset(buf, 'a', sizeof(buf));
+               while (buffer_len(&active_state->incoming_packet) <
+                   PACKET_MAX_SIZE)
+                       buffer_append(&active_state->incoming_packet, buf,
+                           sizeof(buf));
+               (void) mac_compute(active_state->packet_discard_mac,
+                   active_state->p_read.seqnr,
+                   buffer_ptr(&active_state->incoming_packet),
+                   PACKET_MAX_SIZE);
        }
-       /* Kludge: arrange the close function to be called from fatal(). */
-       fatal_add_cleanup((void (*) (void *)) packet_close, NULL);
+       logit("Finished discarding for %.200s", get_remote_ipaddr());
+       cleanup_exit(255);
+}
+
+static void
+packet_start_discard(Enc *enc, Mac *mac, u_int packet_length, u_int discard)
+{
+       if (enc == NULL || !cipher_is_cbc(enc->cipher))
+               packet_disconnect("Packet corrupt");
+       if (packet_length != PACKET_MAX_SIZE && mac && mac->enabled)
+               active_state->packet_discard_mac = mac;
+       if (buffer_len(&active_state->input) >= discard)
+               packet_stop_discard();
+       active_state->packet_discard = discard -
+           buffer_len(&active_state->input);
 }
 
 /* Returns 1 if remote host is connected via socket, 0 if not. */
 
 int
-packet_connection_is_on_socket()
+packet_connection_is_on_socket(void)
 {
        struct sockaddr_storage from, to;
        socklen_t fromlen, tolen;
 
        /* filedescriptors in and out are the same, so it's a socket */
-       if (connection_in == connection_out)
+       if (active_state->connection_in == active_state->connection_out)
                return 1;
        fromlen = sizeof(from);
        memset(&from, 0, sizeof(from));
-       if (getpeername(connection_in, (struct sockaddr *)&from, &fromlen) < 0)
+       if (getpeername(active_state->connection_in, (struct sockaddr *)&from,
+           &fromlen) < 0)
                return 0;
        tolen = sizeof(to);
        memset(&to, 0, sizeof(to));
-       if (getpeername(connection_out, (struct sockaddr *)&to, &tolen) < 0)
+       if (getpeername(active_state->connection_out, (struct sockaddr *)&to,
+           &tolen) < 0)
                return 0;
        if (fromlen != tolen || memcmp(&from, &to, fromlen) != 0)
                return 0;
@@ -183,93 +313,206 @@ packet_connection_is_on_socket()
        return 1;
 }
 
+/*
+ * Exports an IV from the CipherContext required to export the key
+ * state back from the unprivileged child to the privileged parent
+ * process.
+ */
+
+void
+packet_get_keyiv(int mode, u_char *iv, u_int len)
+{
+       CipherContext *cc;
+
+       if (mode == MODE_OUT)
+               cc = &active_state->send_context;
+       else
+               cc = &active_state->receive_context;
+
+       cipher_get_keyiv(cc, iv, len);
+}
+
+int
+packet_get_keycontext(int mode, u_char *dat)
+{
+       CipherContext *cc;
+
+       if (mode == MODE_OUT)
+               cc = &active_state->send_context;
+       else
+               cc = &active_state->receive_context;
+
+       return (cipher_get_keycontext(cc, dat));
+}
+
+void
+packet_set_keycontext(int mode, u_char *dat)
+{
+       CipherContext *cc;
+
+       if (mode == MODE_OUT)
+               cc = &active_state->send_context;
+       else
+               cc = &active_state->receive_context;
+
+       cipher_set_keycontext(cc, dat);
+}
+
+int
+packet_get_keyiv_len(int mode)
+{
+       CipherContext *cc;
+
+       if (mode == MODE_OUT)
+               cc = &active_state->send_context;
+       else
+               cc = &active_state->receive_context;
+
+       return (cipher_get_keyiv_len(cc));
+}
+
+void
+packet_set_iv(int mode, u_char *dat)
+{
+       CipherContext *cc;
+
+       if (mode == MODE_OUT)
+               cc = &active_state->send_context;
+       else
+               cc = &active_state->receive_context;
+
+       cipher_set_keyiv(cc, dat);
+}
+
+int
+packet_get_ssh1_cipher(void)
+{
+       return (cipher_get_number(active_state->receive_context.cipher));
+}
+
+void
+packet_get_state(int mode, u_int32_t *seqnr, u_int64_t *blocks, u_int32_t *packets,
+    u_int64_t *bytes)
+{
+       struct packet_state *state;
+
+       state = (mode == MODE_IN) ?
+           &active_state->p_read : &active_state->p_send;
+       if (seqnr)
+               *seqnr = state->seqnr;
+       if (blocks)
+               *blocks = state->blocks;
+       if (packets)
+               *packets = state->packets;
+       if (bytes)
+               *bytes = state->bytes;
+}
+
+void
+packet_set_state(int mode, u_int32_t seqnr, u_int64_t blocks, u_int32_t packets,
+    u_int64_t bytes)
+{
+       struct packet_state *state;
+
+       state = (mode == MODE_IN) ?
+           &active_state->p_read : &active_state->p_send;
+       state->seqnr = seqnr;
+       state->blocks = blocks;
+       state->packets = packets;
+       state->bytes = bytes;
+}
+
 /* returns 1 if connection is via ipv4 */
 
 int
-packet_connection_is_ipv4()
+packet_connection_is_ipv4(void)
 {
        struct sockaddr_storage to;
        socklen_t tolen = sizeof(to);
 
        memset(&to, 0, sizeof(to));
-       if (getsockname(connection_out, (struct sockaddr *)&to, &tolen) < 0)
-               return 0;
-       if (to.ss_family != AF_INET)
+       if (getsockname(active_state->connection_out, (struct sockaddr *)&to,
+           &tolen) < 0)
                return 0;
-       return 1;
+       if (to.ss_family == AF_INET)
+               return 1;
+#ifdef IPV4_IN_IPV6
+       if (to.ss_family == AF_INET6 &&
+           IN6_IS_ADDR_V4MAPPED(&((struct sockaddr_in6 *)&to)->sin6_addr))
+               return 1;
+#endif
+       return 0;
 }
 
 /* Sets the connection into non-blocking mode. */
 
 void
-packet_set_nonblocking()
+packet_set_nonblocking(void)
 {
        /* Set the socket into non-blocking mode. */
-       if (fcntl(connection_in, F_SETFL, O_NONBLOCK) < 0)
-               error("fcntl O_NONBLOCK: %.100s", strerror(errno));
+       set_nonblock(active_state->connection_in);
 
-       if (connection_out != connection_in) {
-               if (fcntl(connection_out, F_SETFL, O_NONBLOCK) < 0)
-                       error("fcntl O_NONBLOCK: %.100s", strerror(errno));
-       }
+       if (active_state->connection_out != active_state->connection_in)
+               set_nonblock(active_state->connection_out);
 }
 
 /* Returns the socket used for reading. */
 
 int
-packet_get_connection_in()
+packet_get_connection_in(void)
 {
-       return connection_in;
+       return active_state->connection_in;
 }
 
 /* Returns the descriptor used for writing. */
 
 int
-packet_get_connection_out()
+packet_get_connection_out(void)
 {
-       return connection_out;
+       return active_state->connection_out;
 }
 
 /* Closes the connection and clears and frees internal data structures. */
 
 void
-packet_close()
+packet_close(void)
 {
-       if (!initialized)
+       if (!active_state->initialized)
                return;
-       initialized = 0;
-       if (connection_in == connection_out) {
-               shutdown(connection_out, SHUT_RDWR);
-               close(connection_out);
+       active_state->initialized = 0;
+       if (active_state->connection_in == active_state->connection_out) {
+               shutdown(active_state->connection_out, SHUT_RDWR);
+               close(active_state->connection_out);
        } else {
-               close(connection_in);
-               close(connection_out);
+               close(active_state->connection_in);
+               close(active_state->connection_out);
        }
-       buffer_free(&input);
-       buffer_free(&output);
-       buffer_free(&outgoing_packet);
-       buffer_free(&incoming_packet);
-       if (packet_compression) {
-               buffer_free(&compression_buffer);
+       buffer_free(&active_state->input);
+       buffer_free(&active_state->output);
+       buffer_free(&active_state->outgoing_packet);
+       buffer_free(&active_state->incoming_packet);
+       if (active_state->compression_buffer_ready) {
+               buffer_free(&active_state->compression_buffer);
                buffer_compress_uninit();
        }
+       cipher_cleanup(&active_state->send_context);
+       cipher_cleanup(&active_state->receive_context);
 }
 
 /* Sets remote side protocol flags. */
 
 void
-packet_set_protocol_flags(unsigned int protocol_flags)
+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);
+       active_state->remote_protocol_flags = protocol_flags;
 }
 
 /* Returns the remote protocol flags set earlier by the above function. */
 
-unsigned int
-packet_get_protocol_flags()
+u_int
+packet_get_protocol_flags(void)
 {
-       return remote_protocol_flags;
+       return active_state->remote_protocol_flags;
 }
 
 /*
@@ -277,57 +520,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)
-{
-       if (packet_compression)
-               fatal("Compression already enabled.");
-       packet_compression = 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,
-    unsigned int bytes)
+static void
+packet_init_compression(void)
 {
-       cipher_encrypt(cc, dest, src, bytes);
+       if (active_state->compression_buffer_ready == 1)
+               return;
+       active_state->compression_buffer_ready = 1;
+       buffer_init(&active_state->compression_buffer);
 }
 
-/*
- * Decrypts the given number of bytes, copying from src to dest. bytes is
- * known to be a multiple of 8.
- */
-
 void
-packet_decrypt(CipherContext * cc, void *dest, void *src,
-    unsigned int bytes)
+packet_start_compression(int level)
 {
-       int i;
-
-       if ((bytes % 8) != 0)
-               fatal("packet_decrypt: bad ciphertext length %d", bytes);
-
-       /*
-        * Cryptographic attack detector for ssh - Modifications for packet.c
-        * (C)1998 CORE-SDI, Buenos Aires Argentina Ariel Futoransky(futo@core-sdi.com)
-        */
-
-       if (cc->type == SSH_CIPHER_NONE || compat20) {
-               i = DEATTACK_OK;
-       } else {
-               i = detect_attack(src, bytes, NULL);
-       }
-       if (i == DEATTACK_DETECTED)
-               packet_disconnect("crc32 compensation attack: network attack detected");
-
-       cipher_decrypt(cc, dest, src, bytes);
+       if (active_state->packet_compression && !compat20)
+               fatal("Compression already enabled.");
+       active_state->packet_compression = 1;
+       packet_init_compression();
+       buffer_compress_init_send(level);
+       buffer_compress_init_recv();
 }
 
 /*
@@ -337,101 +547,98 @@ packet_decrypt(CipherContext * cc, void *dest, void *src,
  */
 
 void
-packet_set_encryption_key(const unsigned char *key, unsigned int keylen,
-    int cipher)
+packet_set_encryption_key(const u_char *key, u_int keylen,
+    int number)
 {
-       if (keylen < 20)
-               fatal("keylen too small: %d", keylen);
+       Cipher *cipher = cipher_by_number(number);
 
-       /* All other ciphers use the same key in both directions for now. */
-       cipher_set_key(&receive_context, cipher, key, keylen);
-       cipher_set_key(&send_context, cipher, key, keylen);
+       if (cipher == NULL)
+               fatal("packet_set_encryption_key: unknown cipher number %d", number);
+       if (keylen < 20)
+               fatal("packet_set_encryption_key: keylen too small: %d", keylen);
+       if (keylen > SSH_SESSION_KEY_LENGTH)
+               fatal("packet_set_encryption_key: keylen too big: %d", keylen);
+       memcpy(active_state->ssh1_key, key, keylen);
+       active_state->ssh1_keylen = keylen;
+       cipher_init(&active_state->send_context, cipher, key, keylen, NULL,
+           0, CIPHER_ENCRYPT);
+       cipher_init(&active_state->receive_context, cipher, key, keylen, NULL,
+           0, CIPHER_DECRYPT);
 }
 
-/* Starts constructing a packet to send. */
-
-void
-packet_start1(int type)
+u_int
+packet_get_encryption_key(u_char *key)
 {
-       char buf[9];
-
-       buffer_clear(&outgoing_packet);
-       memset(buf, 0, 8);
-       buf[8] = type;
-       buffer_append(&outgoing_packet, buf, 9);
+       if (key == NULL)
+               return (active_state->ssh1_keylen);
+       memcpy(key, active_state->ssh1_key, active_state->ssh1_keylen);
+       return (active_state->ssh1_keylen);
 }
 
+/* Start constructing a packet to send. */
 void
-packet_start2(int type)
+packet_start(u_char type)
 {
-       char buf[4+1+1];
-
-       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);
+       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(&active_state->outgoing_packet);
+       buffer_append(&active_state->outgoing_packet, buf, len);
 }
 
+/* Append payload. */
 void
-packet_start(int type)
+packet_put_char(int value)
 {
-       DBG(debug("packet_start[%d]",type));
-       if (use_ssh2_packet_format)
-               packet_start2(type);
-       else
-               packet_start1(type);
-}
+       char ch = value;
 
-/* Appends a character to the packet data. */
+       buffer_append(&active_state->outgoing_packet, &ch, 1);
+}
 
 void
-packet_put_char(int value)
+packet_put_int(u_int value)
 {
-       char ch = value;
-       buffer_append(&outgoing_packet, &ch, 1);
+       buffer_put_int(&active_state->outgoing_packet, value);
 }
 
-/* Appends an integer to the packet data. */
-
 void
-packet_put_int(unsigned int value)
+packet_put_int64(u_int64_t value)
 {
-       buffer_put_int(&outgoing_packet, value);
+       buffer_put_int64(&active_state->outgoing_packet, value);
 }
 
-/* Appends a string to packet data. */
-
 void
-packet_put_string(const char *buf, unsigned int len)
+packet_put_string(const void *buf, u_int len)
 {
-       buffer_put_string(&outgoing_packet, buf, len);
+       buffer_put_string(&active_state->outgoing_packet, buf, len);
 }
+
 void
 packet_put_cstring(const char *str)
 {
-       buffer_put_string(&outgoing_packet, str, strlen(str));
+       buffer_put_cstring(&active_state->outgoing_packet, str);
 }
 
 void
-packet_put_raw(const char *buf, unsigned int len)
+packet_put_raw(const void *buf, u_int len)
 {
-       buffer_append(&outgoing_packet, buf, len);
+       buffer_append(&active_state->outgoing_packet, buf, len);
 }
 
-
-/* Appends an arbitrary precision integer to packet data. */
-
 void
 packet_put_bignum(BIGNUM * value)
 {
-       buffer_put_bignum(&outgoing_packet, value);
+       buffer_put_bignum(&active_state->outgoing_packet, value);
 }
+
 void
 packet_put_bignum2(BIGNUM * value)
 {
-       buffer_put_bignum2(&outgoing_packet, value);
+       buffer_put_bignum2(&active_state->outgoing_packet, value);
 }
 
 /*
@@ -439,126 +646,241 @@ packet_put_bignum2(BIGNUM * value)
  * encrypts the packet before sending.
  */
 
-void
-packet_send1()
+static void
+packet_send1(void)
 {
-       char buf[8], *cp;
+       u_char buf[8], *cp;
        int i, padding, len;
-       unsigned int checksum;
-       u_int32_t rand = 0;
+       u_int checksum;
+       u_int32_t rnd = 0;
 
        /*
         * If using packet compression, compress the payload of the outgoing
         * packet.
         */
-       if (packet_compression) {
-               buffer_clear(&compression_buffer);
+       if (active_state->packet_compression) {
+               buffer_clear(&active_state->compression_buffer);
                /* Skip padding. */
-               buffer_consume(&outgoing_packet, 8);
+               buffer_consume(&active_state->outgoing_packet, 8);
                /* padding */
-               buffer_append(&compression_buffer, "\0\0\0\0\0\0\0\0", 8);
-               buffer_compress(&outgoing_packet, &compression_buffer);
-               buffer_clear(&outgoing_packet);
-               buffer_append(&outgoing_packet, buffer_ptr(&compression_buffer),
-                             buffer_len(&compression_buffer));
+               buffer_append(&active_state->compression_buffer,
+                   "\0\0\0\0\0\0\0\0", 8);
+               buffer_compress(&active_state->outgoing_packet,
+                   &active_state->compression_buffer);
+               buffer_clear(&active_state->outgoing_packet);
+               buffer_append(&active_state->outgoing_packet,
+                   buffer_ptr(&active_state->compression_buffer),
+                   buffer_len(&active_state->compression_buffer));
        }
        /* Compute packet length without padding (add checksum, remove padding). */
-       len = buffer_len(&outgoing_packet) + 4 - 8;
+       len = buffer_len(&active_state->outgoing_packet) + 4 - 8;
 
        /* Insert padding. Initialized to zero in packet_start1() */
        padding = 8 - len % 8;
-       if (cipher_type != SSH_CIPHER_NONE) {
-               cp = buffer_ptr(&outgoing_packet);
+       if (!active_state->send_context.plaintext) {
+               cp = buffer_ptr(&active_state->outgoing_packet);
                for (i = 0; i < padding; i++) {
                        if (i % 4 == 0)
-                               rand = arc4random();
-                       cp[7 - i] = rand & 0xff;
-                       rand >>= 8;
+                               rnd = arc4random();
+                       cp[7 - i] = rnd & 0xff;
+                       rnd >>= 8;
                }
        }
-       buffer_consume(&outgoing_packet, 8 - padding);
+       buffer_consume(&active_state->outgoing_packet, 8 - padding);
 
        /* Add check bytes. */
-       checksum = crc32((unsigned char *) buffer_ptr(&outgoing_packet),
-                        buffer_len(&outgoing_packet));
-       PUT_32BIT(buf, checksum);
-       buffer_append(&outgoing_packet, buf, 4);
+       checksum = ssh_crc32(buffer_ptr(&active_state->outgoing_packet),
+           buffer_len(&active_state->outgoing_packet));
+       put_u32(buf, checksum);
+       buffer_append(&active_state->outgoing_packet, buf, 4);
 
 #ifdef PACKET_DEBUG
        fprintf(stderr, "packet_send plain: ");
-       buffer_dump(&outgoing_packet);
+       buffer_dump(&active_state->outgoing_packet);
 #endif
 
        /* Append to output. */
-       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),
-                      buffer_len(&outgoing_packet));
+       put_u32(buf, len);
+       buffer_append(&active_state->output, buf, 4);
+       cp = buffer_append_space(&active_state->output,
+           buffer_len(&active_state->outgoing_packet));
+       cipher_crypt(&active_state->send_context, cp,
+           buffer_ptr(&active_state->outgoing_packet),
+           buffer_len(&active_state->outgoing_packet));
 
 #ifdef PACKET_DEBUG
        fprintf(stderr, "encrypted: ");
-       buffer_dump(&output);
+       buffer_dump(&active_state->output);
 #endif
-
-       buffer_clear(&outgoing_packet);
+       active_state->p_send.packets++;
+       active_state->p_send.bytes += len +
+           buffer_len(&active_state->outgoing_packet);
+       buffer_clear(&active_state->outgoing_packet);
 
        /*
-        * Note that the packet is now only buffered in output.  It won\'t be
+        * Note that the packet is now only buffered in output.  It won't be
         * actually sent until packet_write_wait or packet_write_poll is
         * called.
         */
 }
 
+void
+set_newkeys(int mode)
+{
+       Enc *enc;
+       Mac *mac;
+       Comp *comp;
+       CipherContext *cc;
+       u_int64_t *max_blocks;
+       int crypt_type;
+
+       debug2("set_newkeys: mode %d", mode);
+
+       if (mode == MODE_OUT) {
+               cc = &active_state->send_context;
+               crypt_type = CIPHER_ENCRYPT;
+               active_state->p_send.packets = active_state->p_send.blocks = 0;
+               max_blocks = &active_state->max_blocks_out;
+       } else {
+               cc = &active_state->receive_context;
+               crypt_type = CIPHER_DECRYPT;
+               active_state->p_read.packets = active_state->p_read.blocks = 0;
+               max_blocks = &active_state->max_blocks_in;
+       }
+       if (active_state->newkeys[mode] != NULL) {
+               debug("set_newkeys: rekeying");
+               cipher_cleanup(cc);
+               enc  = &active_state->newkeys[mode]->enc;
+               mac  = &active_state->newkeys[mode]->mac;
+               comp = &active_state->newkeys[mode]->comp;
+               mac_clear(mac);
+               xfree(enc->name);
+               xfree(enc->iv);
+               xfree(enc->key);
+               xfree(mac->name);
+               xfree(mac->key);
+               xfree(comp->name);
+               xfree(active_state->newkeys[mode]);
+       }
+       active_state->newkeys[mode] = kex_get_newkeys(mode);
+       if (active_state->newkeys[mode] == NULL)
+               fatal("newkeys: no keys for mode %d", mode);
+       enc  = &active_state->newkeys[mode]->enc;
+       mac  = &active_state->newkeys[mode]->mac;
+       comp = &active_state->newkeys[mode]->comp;
+       if (mac_init(mac) == 0)
+               mac->enabled = 1;
+       DBG(debug("cipher_init_context: %d", mode));
+       cipher_init(cc, enc->cipher, enc->key, enc->key_len,
+           enc->iv, enc->block_size, crypt_type);
+       /* Deleting the keys does not gain extra security */
+       /* memset(enc->iv,  0, enc->block_size);
+          memset(enc->key, 0, enc->key_len);
+          memset(mac->key, 0, mac->key_len); */
+       if ((comp->type == COMP_ZLIB ||
+           (comp->type == COMP_DELAYED &&
+            active_state->after_authentication)) && comp->enabled == 0) {
+               packet_init_compression();
+               if (mode == MODE_OUT)
+                       buffer_compress_init_send(6);
+               else
+                       buffer_compress_init_recv();
+               comp->enabled = 1;
+       }
+       /*
+        * The 2^(blocksize*2) limit is too expensive for 3DES,
+        * blowfish, etc, so enforce a 1GB limit for small blocksizes.
+        */
+       if (enc->block_size >= 16)
+               *max_blocks = (u_int64_t)1 << (enc->block_size*2);
+       else
+               *max_blocks = ((u_int64_t)1 << 30) / enc->block_size;
+       if (active_state->rekey_limit)
+               *max_blocks = MIN(*max_blocks,
+                   active_state->rekey_limit / enc->block_size);
+}
+
+/*
+ * Delayed compression for SSH2 is enabled after authentication:
+ * This happens on the server side after a SSH2_MSG_USERAUTH_SUCCESS is sent,
+ * and on the client side after a SSH2_MSG_USERAUTH_SUCCESS is received.
+ */
+static void
+packet_enable_delayed_compress(void)
+{
+       Comp *comp = NULL;
+       int mode;
+
+       /*
+        * Remember that we are past the authentication step, so rekeying
+        * with COMP_DELAYED will turn on compression immediately.
+        */
+       active_state->after_authentication = 1;
+       for (mode = 0; mode < MODE_MAX; mode++) {
+               /* protocol error: USERAUTH_SUCCESS received before NEWKEYS */
+               if (active_state->newkeys[mode] == NULL)
+                       continue;
+               comp = &active_state->newkeys[mode]->comp;
+               if (comp && !comp->enabled && comp->type == COMP_DELAYED) {
+                       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()
-{
-       unsigned char *macbuf = NULL;
-       char *cp;
-       unsigned int packet_length = 0;
-       unsigned int i, padlen, len;
-       u_int32_t rand = 0;
-       static unsigned int seqnr = 0;
-       int type;
+static void
+packet_send2_wrapped(void)
+{
+       u_char type, *cp, *macbuf = NULL;
+       u_char padlen, pad;
+       u_int packet_length = 0;
+       u_int i, len;
+       u_int32_t rnd = 0;
        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 (active_state->newkeys[MODE_OUT] != NULL) {
+               enc  = &active_state->newkeys[MODE_OUT]->enc;
+               mac  = &active_state->newkeys[MODE_OUT]->mac;
+               comp = &active_state->newkeys[MODE_OUT]->comp;
        }
        block_size = enc ? enc->block_size : 8;
 
-       cp = buffer_ptr(&outgoing_packet);
-       type = cp[5] & 0xff;
+       cp = buffer_ptr(&active_state->outgoing_packet);
+       type = cp[5];
 
 #ifdef PACKET_DEBUG
        fprintf(stderr, "plain:     ");
-       buffer_dump(&outgoing_packet);
+       buffer_dump(&active_state->outgoing_packet);
 #endif
 
        if (comp && comp->enabled) {
-               len = buffer_len(&outgoing_packet);
+               len = buffer_len(&active_state->outgoing_packet);
                /* skip header, compress only payload */
-               buffer_consume(&outgoing_packet, 5);
-               buffer_clear(&compression_buffer);
-               buffer_compress(&outgoing_packet, &compression_buffer);
-               buffer_clear(&outgoing_packet);
-               buffer_append(&outgoing_packet, "\0\0\0\0\0", 5);
-               buffer_append(&outgoing_packet, buffer_ptr(&compression_buffer),
-                   buffer_len(&compression_buffer));
+               buffer_consume(&active_state->outgoing_packet, 5);
+               buffer_clear(&active_state->compression_buffer);
+               buffer_compress(&active_state->outgoing_packet,
+                   &active_state->compression_buffer);
+               buffer_clear(&active_state->outgoing_packet);
+               buffer_append(&active_state->outgoing_packet, "\0\0\0\0\0", 5);
+               buffer_append(&active_state->outgoing_packet,
+                   buffer_ptr(&active_state->compression_buffer),
+                   buffer_len(&active_state->compression_buffer));
                DBG(debug("compression: raw %d compressed %d", len,
-                   buffer_len(&outgoing_packet)));
+                   buffer_len(&active_state->outgoing_packet)));
        }
 
        /* sizeof (packet_len + pad_len + payload) */
-       len = buffer_len(&outgoing_packet);
+       len = buffer_len(&active_state->outgoing_packet);
 
        /*
         * calc size of padding, alloc space, get random data,
@@ -567,73 +889,123 @@ packet_send2()
        padlen = block_size - (len % block_size);
        if (padlen < 4)
                padlen += block_size;
-       buffer_append_space(&outgoing_packet, &cp, padlen);
-       if (enc && enc->type != SSH_CIPHER_NONE) {
+       if (active_state->extra_pad) {
+               /* will wrap if extra_pad+padlen > 255 */
+               active_state->extra_pad =
+                   roundup(active_state->extra_pad, block_size);
+               pad = active_state->extra_pad -
+                   ((len + padlen) % active_state->extra_pad);
+               debug3("packet_send2: adding %d (len %d padlen %d extra_pad %d)",
+                   pad, len, padlen, active_state->extra_pad);
+               padlen += pad;
+               active_state->extra_pad = 0;
+       }
+       cp = buffer_append_space(&active_state->outgoing_packet, padlen);
+       if (enc && !active_state->send_context.plaintext) {
                /* random padding */
                for (i = 0; i < padlen; i++) {
                        if (i % 4 == 0)
-                               rand = arc4random();
-                       cp[i] = rand & 0xff;
-                       rand <<= 8;
+                               rnd = arc4random();
+                       cp[i] = rnd & 0xff;
+                       rnd >>= 8;
                }
        } else {
                /* clear padding */
                memset(cp, 0, padlen);
        }
        /* 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;
+       packet_length = buffer_len(&active_state->outgoing_packet) - 4;
+       cp = buffer_ptr(&active_state->outgoing_packet);
+       put_u32(cp, packet_length);
+       cp[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,
-                   (unsigned char *) buffer_ptr(&outgoing_packet),
-                   buffer_len(&outgoing_packet),
-                   mac->key, mac->key_len
-               );
-               DBG(debug("done calc HMAC out #%d", seqnr));
+               macbuf = mac_compute(mac, active_state->p_send.seqnr,
+                   buffer_ptr(&active_state->outgoing_packet),
+                   buffer_len(&active_state->outgoing_packet));
+               DBG(debug("done calc MAC out #%d", active_state->p_send.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),
-           buffer_len(&outgoing_packet));
+       cp = buffer_append_space(&active_state->output,
+           buffer_len(&active_state->outgoing_packet));
+       cipher_crypt(&active_state->send_context, cp,
+           buffer_ptr(&active_state->outgoing_packet),
+           buffer_len(&active_state->outgoing_packet));
        /* append unencrypted MAC */
        if (mac && mac->enabled)
-               buffer_append(&output, (char *)macbuf, mac->mac_len);
+               buffer_append(&active_state->output, macbuf, mac->mac_len);
 #ifdef PACKET_DEBUG
        fprintf(stderr, "encrypted: ");
-       buffer_dump(&output);
+       buffer_dump(&active_state->output);
 #endif
        /* increment sequence number for outgoing packets */
-       if (++seqnr == 0)
-               log("outgoing seqnr wraps around");
-       buffer_clear(&outgoing_packet);
+       if (++active_state->p_send.seqnr == 0)
+               logit("outgoing seqnr wraps around");
+       if (++active_state->p_send.packets == 0)
+               if (!(datafellows & SSH_BUG_NOREKEY))
+                       fatal("XXX too many packets with same key");
+       active_state->p_send.blocks += (packet_length + 4) / block_size;
+       active_state->p_send.bytes += packet_length + 4;
+       buffer_clear(&active_state->outgoing_packet);
+
+       if (type == SSH2_MSG_NEWKEYS)
+               set_newkeys(MODE_OUT);
+       else if (type == SSH2_MSG_USERAUTH_SUCCESS && active_state->server_side)
+               packet_enable_delayed_compress();
+}
+
+static void
+packet_send2(void)
+{
+       struct packet *p;
+       u_char type, *cp;
+
+       cp = buffer_ptr(&active_state->outgoing_packet);
+       type = cp[5];
+
+       /* during rekeying we can only send key exchange messages */
+       if (active_state->rekeying) {
+               if (!((type >= SSH2_MSG_TRANSPORT_MIN) &&
+                   (type <= SSH2_MSG_TRANSPORT_MAX))) {
+                       debug("enqueue packet: %u", type);
+                       p = xmalloc(sizeof(*p));
+                       p->type = type;
+                       memcpy(&p->payload, &active_state->outgoing_packet,
+                           sizeof(Buffer));
+                       buffer_init(&active_state->outgoing_packet);
+                       TAILQ_INSERT_TAIL(&active_state->outgoing, p, next);
+                       return;
+               }
+       }
+
+       /* rekeying starts with sending KEXINIT */
+       if (type == SSH2_MSG_KEXINIT)
+               active_state->rekeying = 1;
 
+       packet_send2_wrapped();
+
+       /* after a NEWKEYS message we can send the complete queue */
        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_set_key_iv send_context"));
-               cipher_set_key_iv(&send_context, enc->type,
-                   enc->key, enc->key_len,
-                   enc->iv, enc->iv_len);
-               clear_enc_keys(enc, kex->we_need);
-               if (comp->type != 0 && comp->enabled == 0) {
-                       comp->enabled = 1;
-                       if (! packet_compression)
-                               packet_start_compression(6);
+               active_state->rekeying = 0;
+               while ((p = TAILQ_FIRST(&active_state->outgoing))) {
+                       type = p->type;
+                       debug("dequeue packet: %u", type);
+                       buffer_free(&active_state->outgoing_packet);
+                       memcpy(&active_state->outgoing_packet, &p->payload,
+                           sizeof(Buffer));
+                       TAILQ_REMOVE(&active_state->outgoing, p, next);
+                       xfree(p);
+                       packet_send2_wrapped();
                }
        }
 }
 
 void
-packet_send()
+packet_send(void)
 {
-       if (use_ssh2_packet_format)
+       if (compat20)
                packet_send2();
        else
                packet_send1();
@@ -647,44 +1019,82 @@ packet_send()
  */
 
 int
-packet_read(int *payload_len_ptr)
+packet_read_seqnr(u_int32_t *seqnr_p)
 {
-       int type, len;
-       fd_set set;
+       int type, len, ret, ms_remain, cont;
+       fd_set *setp;
        char buf[8192];
+       struct timeval timeout, start, *timeoutp = NULL;
+
        DBG(debug("packet_read()"));
 
+       setp = (fd_set *)xcalloc(howmany(active_state->connection_in + 1,
+           NFDBITS), sizeof(fd_mask));
+
        /* Since we are blocking, ensure that all written packets have been sent. */
        packet_write_wait();
 
        /* Stay in the loop until we have received a complete packet. */
        for (;;) {
                /* Try to read a packet from the buffer. */
-               type = packet_read_poll(payload_len_ptr);
-               if (!use_ssh2_packet_format && (
+               type = packet_read_poll_seqnr(seqnr_p);
+               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);
+                       packet_check_eom();
                /* 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(active_state->connection_in + 1,
+                   NFDBITS) * sizeof(fd_mask));
+               FD_SET(active_state->connection_in, setp);
 
+               if (active_state->packet_timeout_ms > 0) {
+                       ms_remain = active_state->packet_timeout_ms;
+                       timeoutp = &timeout;
+               }
                /* Wait for some data to arrive. */
-               select(connection_in + 1, &set, NULL, NULL, NULL);
-
+               for (;;) {
+                       if (active_state->packet_timeout_ms != -1) {
+                               ms_to_timeval(&timeout, ms_remain);
+                               gettimeofday(&start, NULL);
+                       }
+                       if ((ret = select(active_state->connection_in + 1, setp,
+                           NULL, NULL, timeoutp)) >= 0)
+                               break;
+                       if (errno != EAGAIN && errno != EINTR &&
+                           errno != EWOULDBLOCK)
+                               break;
+                       if (active_state->packet_timeout_ms == -1)
+                               continue;
+                       ms_subtract_diff(&start, &ms_remain);
+                       if (ms_remain <= 0) {
+                               ret = 0;
+                               break;
+                       }
+               }
+               if (ret == 0) {
+                       logit("Connection to %.200s timed out while "
+                           "waiting to read", get_remote_ipaddr());
+                       cleanup_exit(255);
+               }
                /* Read data from the socket. */
-               len = read(connection_in, buf, sizeof(buf));
+               do {
+                       cont = 0;
+                       len = roaming_read(active_state->connection_in, buf,
+                           sizeof(buf), &cont);
+               } while (len == 0 && cont);
                if (len == 0) {
-                       log("Connection closed by %.200s", get_remote_ipaddr());
-                       fatal_cleanup();
+                       logit("Connection closed by %.200s", get_remote_ipaddr());
+                       cleanup_exit(255);
                }
                if (len < 0)
                        fatal("Read from socket failed: %.100s", strerror(errno));
@@ -694,17 +1104,23 @@ packet_read(int *payload_len_ptr)
        /* NOTREACHED */
 }
 
+int
+packet_read(void)
+{
+       return packet_read_seqnr(NULL);
+}
+
 /*
  * Waits until a packet has been received, verifies that its type matches
  * that given, and gives a fatal error and exits if there is a mismatch.
  */
 
 void
-packet_read_expect(int *payload_len_ptr, int expected_type)
+packet_read_expect(int expected_type)
 {
        int type;
 
-       type = packet_read(payload_len_ptr);
+       type = packet_read();
        if (type != expected_type)
                packet_disconnect("Protocol error: expected packet type %d, got %d",
                    expected_type, type);
@@ -717,242 +1133,266 @@ packet_read_expect(int *payload_len_ptr, int expected_type)
  * SSH_MSG_DISCONNECT is handled specially here.  Also,
  * SSH_MSG_IGNORE messages are skipped by this function and are never returned
  * to higher levels.
- *
- * The returned payload_len does include space consumed by:
- *     Packet length
- *     Padding
- *     Packet type
- *     Check bytes
  */
 
-int
-packet_read_poll1(int *payload_len_ptr)
+static int
+packet_read_poll1(void)
 {
-       unsigned int len, padded_len;
-       unsigned char *ucp;
-       char buf[8], *cp;
-       unsigned int checksum, stored_checksum;
+       u_int len, padded_len;
+       u_char *cp, type;
+       u_int checksum, stored_checksum;
 
        /* Check if input size is less than minimum packet size. */
-       if (buffer_len(&input) < 4 + 8)
+       if (buffer_len(&active_state->input) < 4 + 8)
                return SSH_MSG_NONE;
        /* Get length of incoming packet. */
-       ucp = (unsigned char *) buffer_ptr(&input);
-       len = GET_32BIT(ucp);
+       cp = buffer_ptr(&active_state->input);
+       len = get_u32(cp);
        if (len < 1 + 2 + 2 || len > 256 * 1024)
-               packet_disconnect("Bad packet length %d.", len);
+               packet_disconnect("Bad packet length %u.", len);
        padded_len = (len + 8) & ~7;
 
        /* Check if the packet has been entirely received. */
-       if (buffer_len(&input) < 4 + padded_len)
+       if (buffer_len(&active_state->input) < 4 + padded_len)
                return SSH_MSG_NONE;
 
        /* The entire packet is in buffer. */
 
        /* Consume packet length. */
-       buffer_consume(&input, 4);
+       buffer_consume(&active_state->input, 4);
+
+       /*
+        * Cryptographic attack detector for ssh
+        * (C)1998 CORE-SDI, Buenos Aires Argentina
+        * Ariel Futoransky(futo@core-sdi.com)
+        */
+       if (!active_state->receive_context.plaintext) {
+               switch (detect_attack(buffer_ptr(&active_state->input),
+                   padded_len)) {
+               case DEATTACK_DETECTED:
+                       packet_disconnect("crc32 compensation attack: "
+                           "network attack detected");
+               case DEATTACK_DOS_DETECTED:
+                       packet_disconnect("deattack denial of "
+                           "service detected");
+               }
+       }
+
+       /* Decrypt data to incoming_packet. */
+       buffer_clear(&active_state->incoming_packet);
+       cp = buffer_append_space(&active_state->incoming_packet, padded_len);
+       cipher_crypt(&active_state->receive_context, cp,
+           buffer_ptr(&active_state->input), padded_len);
 
-       /* Copy 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);
-       buffer_consume(&input, padded_len);
+       buffer_consume(&active_state->input, padded_len);
 
 #ifdef PACKET_DEBUG
        fprintf(stderr, "read_poll plain: ");
-       buffer_dump(&incoming_packet);
+       buffer_dump(&active_state->incoming_packet);
 #endif
 
        /* Compute packet checksum. */
-       checksum = crc32((unsigned char *) buffer_ptr(&incoming_packet),
-           buffer_len(&incoming_packet) - 4);
+       checksum = ssh_crc32(buffer_ptr(&active_state->incoming_packet),
+           buffer_len(&active_state->incoming_packet) - 4);
 
        /* Skip padding. */
-       buffer_consume(&incoming_packet, 8 - len % 8);
+       buffer_consume(&active_state->incoming_packet, 8 - len % 8);
 
        /* Test check bytes. */
+       if (len != buffer_len(&active_state->incoming_packet))
+               packet_disconnect("packet_read_poll1: len %d != buffer_len %d.",
+                   len, buffer_len(&active_state->incoming_packet));
 
-       if (len != buffer_len(&incoming_packet))
-               packet_disconnect("packet_read_poll: len %d != buffer_len %d.",
-                   len, buffer_len(&incoming_packet));
-
-       ucp = (unsigned char *) buffer_ptr(&incoming_packet) + len - 4;
-       stored_checksum = GET_32BIT(ucp);
+       cp = (u_char *)buffer_ptr(&active_state->incoming_packet) + len - 4;
+       stored_checksum = get_u32(cp);
        if (checksum != stored_checksum)
                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);
-               buffer_clear(&incoming_packet);
-               buffer_append(&incoming_packet, buffer_ptr(&compression_buffer),
-                   buffer_len(&compression_buffer));
+       buffer_consume_end(&active_state->incoming_packet, 4);
+
+       if (active_state->packet_compression) {
+               buffer_clear(&active_state->compression_buffer);
+               buffer_uncompress(&active_state->incoming_packet,
+                   &active_state->compression_buffer);
+               buffer_clear(&active_state->incoming_packet);
+               buffer_append(&active_state->incoming_packet,
+                   buffer_ptr(&active_state->compression_buffer),
+                   buffer_len(&active_state->compression_buffer));
        }
-       /* Get packet type. */
-       buffer_get(&incoming_packet, &buf[0], 1);
-
-       /* Return length of payload (without type field). */
-       *payload_len_ptr = buffer_len(&incoming_packet);
-
-       /* Return type. */
-       return (unsigned char) buf[0];
+       active_state->p_read.packets++;
+       active_state->p_read.bytes += padded_len + 4;
+       type = buffer_get_char(&active_state->incoming_packet);
+       if (type < SSH_MSG_MIN || type > SSH_MSG_MAX)
+               packet_disconnect("Invalid ssh1 packet type: %d", type);
+       return type;
 }
 
-int
-packet_read_poll2(int *payload_len_ptr)
-{
-       unsigned int padlen, need;
-       unsigned char buf[8], *macbuf;
-       unsigned char *ucp;
-       char *cp;
-       static unsigned int packet_length = 0;
-       static unsigned int seqnr = 0;
-       int type;
-       int maclen, block_size;
+static int
+packet_read_poll2(u_int32_t *seqnr_p)
+{
+       u_int padlen, need;
+       u_char *macbuf, *cp, type;
+       u_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 (active_state->packet_discard)
+               return SSH_MSG_NONE;
+
+       if (active_state->newkeys[MODE_IN] != NULL) {
+               enc  = &active_state->newkeys[MODE_IN]->enc;
+               mac  = &active_state->newkeys[MODE_IN]->mac;
+               comp = &active_state->newkeys[MODE_IN]->comp;
        }
        maclen = mac && mac->enabled ? mac->mac_len : 0;
        block_size = enc ? enc->block_size : 8;
 
-       if (packet_length == 0) {
+       if (active_state->packlen == 0) {
                /*
                 * check if input size is less than the cipher block size,
                 * decrypt first block and extract length of incoming packet
                 */
-               if (buffer_len(&input) < block_size)
+               if (buffer_len(&active_state->input) < block_size)
                        return SSH_MSG_NONE;
-               buffer_clear(&incoming_packet);
-               buffer_append_space(&incoming_packet, &cp, block_size);
-               packet_decrypt(&receive_context, cp, buffer_ptr(&input),
+               buffer_clear(&active_state->incoming_packet);
+               cp = buffer_append_space(&active_state->incoming_packet,
                    block_size);
-               ucp = (unsigned char *) buffer_ptr(&incoming_packet);
-               packet_length = GET_32BIT(ucp);
-               if (packet_length < 1 + 4 || packet_length > 256 * 1024) {
-                       buffer_dump(&incoming_packet);
-                       packet_disconnect("Bad packet length %d.", packet_length);
+               cipher_crypt(&active_state->receive_context, cp,
+                   buffer_ptr(&active_state->input), block_size);
+               cp = buffer_ptr(&active_state->incoming_packet);
+               active_state->packlen = get_u32(cp);
+               if (active_state->packlen < 1 + 4 ||
+                   active_state->packlen > PACKET_MAX_SIZE) {
+#ifdef PACKET_DEBUG
+                       buffer_dump(&active_state->incoming_packet);
+#endif
+                       logit("Bad packet length %u.", active_state->packlen);
+                       packet_start_discard(enc, mac, active_state->packlen,
+                           PACKET_MAX_SIZE);
+                       return SSH_MSG_NONE;
                }
-               DBG(debug("input: packet len %d", packet_length+4));
-               buffer_consume(&input, block_size);
+               DBG(debug("input: packet len %u", active_state->packlen+4));
+               buffer_consume(&active_state->input, block_size);
        }
        /* we have a partial packet of block_size bytes */
-       need = 4 + packet_length - block_size;
+       need = 4 + active_state->packlen - block_size;
        DBG(debug("partial packet %d, need %d, maclen %d", block_size,
            need, maclen));
-       if (need % block_size != 0)
-               fatal("padding error: need %d block %d mod %d",
+       if (need % block_size != 0) {
+               logit("padding error: need %d block %d mod %d",
                    need, block_size, need % block_size);
+               packet_start_discard(enc, mac, active_state->packlen,
+                   PACKET_MAX_SIZE - block_size);
+               return SSH_MSG_NONE;
+       }
        /*
         * check if the entire packet has been received and
         * decrypt into incoming_packet
         */
-       if (buffer_len(&input) < need + maclen)
+       if (buffer_len(&active_state->input) < need + maclen)
                return SSH_MSG_NONE;
 #ifdef PACKET_DEBUG
        fprintf(stderr, "read_poll enc/full: ");
-       buffer_dump(&input);
+       buffer_dump(&active_state->input);
 #endif
-       buffer_append_space(&incoming_packet, &cp, need);
-       packet_decrypt(&receive_context, cp, buffer_ptr(&input), need);
-       buffer_consume(&input, need);
+       cp = buffer_append_space(&active_state->incoming_packet, need);
+       cipher_crypt(&active_state->receive_context, cp,
+           buffer_ptr(&active_state->input), need);
+       buffer_consume(&active_state->input, need);
        /*
         * compute MAC over seqnr and packet,
         * increment sequence number for incoming packet
         */
        if (mac && mac->enabled) {
-               macbuf = hmac( mac->md, seqnr,
-                   (unsigned char *) buffer_ptr(&incoming_packet),
-                   buffer_len(&incoming_packet),
-                   mac->key, mac->key_len
-               );
-               if (memcmp(macbuf, buffer_ptr(&input), mac->mac_len) != 0)
-                       packet_disconnect("Corrupted HMAC on input.");
-               DBG(debug("HMAC #%d ok", seqnr));
-               buffer_consume(&input, mac->mac_len);
+               macbuf = mac_compute(mac, active_state->p_read.seqnr,
+                   buffer_ptr(&active_state->incoming_packet),
+                   buffer_len(&active_state->incoming_packet));
+               if (memcmp(macbuf, buffer_ptr(&active_state->input),
+                   mac->mac_len) != 0) {
+                       logit("Corrupted MAC on input.");
+                       if (need > PACKET_MAX_SIZE)
+                               fatal("internal error need %d", need);
+                       packet_start_discard(enc, mac, active_state->packlen,
+                           PACKET_MAX_SIZE - need);
+                       return SSH_MSG_NONE;
+               }
+                               
+               DBG(debug("MAC #%d ok", active_state->p_read.seqnr));
+               buffer_consume(&active_state->input, mac->mac_len);
        }
-       if (++seqnr == 0)
-               log("incoming seqnr wraps around");
+       /* XXX now it's safe to use fatal/packet_disconnect */
+       if (seqnr_p != NULL)
+               *seqnr_p = active_state->p_read.seqnr;
+       if (++active_state->p_read.seqnr == 0)
+               logit("incoming seqnr wraps around");
+       if (++active_state->p_read.packets == 0)
+               if (!(datafellows & SSH_BUG_NOREKEY))
+                       fatal("XXX too many packets with same key");
+       active_state->p_read.blocks += (active_state->packlen + 4) / block_size;
+       active_state->p_read.bytes += active_state->packlen + 4;
 
        /* get padlen */
-       cp = buffer_ptr(&incoming_packet) + 4;
-       padlen = *cp & 0xff;
+       cp = buffer_ptr(&active_state->incoming_packet);
+       padlen = cp[4];
        DBG(debug("input: padlen %d", padlen));
        if (padlen < 4)
                packet_disconnect("Corrupted padlen %d on input.", padlen);
 
        /* skip packet size + padlen, discard padding */
-       buffer_consume(&incoming_packet, 4 + 1);
-       buffer_consume_end(&incoming_packet, padlen);
+       buffer_consume(&active_state->incoming_packet, 4 + 1);
+       buffer_consume_end(&active_state->incoming_packet, padlen);
 
-       DBG(debug("input: len before de-compress %d", buffer_len(&incoming_packet)));
+       DBG(debug("input: len before de-compress %d",
+           buffer_len(&active_state->incoming_packet)));
        if (comp && comp->enabled) {
-               buffer_clear(&compression_buffer);
-               buffer_uncompress(&incoming_packet, &compression_buffer);
-               buffer_clear(&incoming_packet);
-               buffer_append(&incoming_packet, buffer_ptr(&compression_buffer),
-                   buffer_len(&compression_buffer));
-               DBG(debug("input: len after de-compress %d", buffer_len(&incoming_packet)));
+               buffer_clear(&active_state->compression_buffer);
+               buffer_uncompress(&active_state->incoming_packet,
+                   &active_state->compression_buffer);
+               buffer_clear(&active_state->incoming_packet);
+               buffer_append(&active_state->incoming_packet,
+                   buffer_ptr(&active_state->compression_buffer),
+                   buffer_len(&active_state->compression_buffer));
+               DBG(debug("input: len after de-compress %d",
+                   buffer_len(&active_state->incoming_packet)));
        }
        /*
         * get packet type, implies consume.
         * return length of payload (without type field)
         */
-       buffer_get(&incoming_packet, (char *)&buf[0], 1);
-       *payload_len_ptr = buffer_len(&incoming_packet);
-
-       /* reset for next packet */
-       packet_length = 0;
-
-       /* extract packet type */
-       type = (unsigned 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_set_key_iv receive_context"));
-               cipher_set_key_iv(&receive_context, enc->type,
-                   enc->key, enc->key_len,
-                   enc->iv, enc->iv_len);
-               clear_enc_keys(enc, kex->we_need);
-               if (comp->type != 0 && comp->enabled == 0) {
-                       comp->enabled = 1;
-                       if (! packet_compression)
-                               packet_start_compression(6);
-               }
-       }
-
+       type = buffer_get_char(&active_state->incoming_packet);
+       if (type < SSH2_MSG_MIN || type >= SSH2_MSG_LOCAL_MIN)
+               packet_disconnect("Invalid ssh2 packet type: %d", type);
+       if (type == SSH2_MSG_NEWKEYS)
+               set_newkeys(MODE_IN);
+       else if (type == SSH2_MSG_USERAUTH_SUCCESS &&
+           !active_state->server_side)
+               packet_enable_delayed_compress();
 #ifdef PACKET_DEBUG
-       fprintf(stderr, "read/plain[%d]:\r\n",type);
-       buffer_dump(&incoming_packet);
+       fprintf(stderr, "read/plain[%d]:\r\n", type);
+       buffer_dump(&active_state->incoming_packet);
 #endif
-       return (unsigned char)type;
+       /* reset for next packet */
+       active_state->packlen = 0;
+       return type;
 }
 
 int
-packet_read_poll(int *payload_len_ptr)
+packet_read_poll_seqnr(u_int32_t *seqnr_p)
 {
+       u_int reason, seqnr;
+       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(seqnr_p);
+                       if (type) {
+                               active_state->keep_alive_timeouts = 0;
                                DBG(debug("received packet type %d", type));
-                       switch(type) {
+                       }
+                       switch (type) {
                        case SSH2_MSG_IGNORE:
+                               debug3("Received SSH2_MSG_IGNORE");
                                break;
                        case SSH2_MSG_DEBUG:
                                packet_get_char();
@@ -965,16 +1405,22 @@ 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);
+                               logit("Received disconnect from %s: %u: %.400s",
+                                   get_remote_ipaddr(), reason, msg);
                                xfree(msg);
-                               fatal_cleanup();
+                               cleanup_exit(255);
+                               break;
+                       case SSH2_MSG_UNIMPLEMENTED:
+                               seqnr = packet_get_int();
+                               debug("Received SSH2_MSG_UNIMPLEMENTED for %u",
+                                   seqnr);
                                break;
                        default:
                                return type;
-                               break;
-                       }       
+                       }
                } else {
-                       switch(type) {
+                       type = packet_read_poll1();
+                       switch (type) {
                        case SSH_MSG_IGNORE:
                                break;
                        case SSH_MSG_DEBUG:
@@ -984,47 +1430,68 @@ packet_read_poll(int *payload_len_ptr)
                                break;
                        case SSH_MSG_DISCONNECT:
                                msg = packet_get_string(NULL);
-                               log("Received disconnect: %.900s", msg);
-                               fatal_cleanup();
-                               xfree(msg);
+                               logit("Received disconnect from %s: %.400s",
+                                   get_remote_ipaddr(), msg);
+                               cleanup_exit(255);
                                break;
                        default:
-                               if (type != 0)
+                               if (type)
                                        DBG(debug("received packet type %d", type));
                                return type;
-                               break;
-                       }       
+                       }
                }
        }
 }
 
+int
+packet_read_poll(void)
+{
+       return packet_read_poll_seqnr(NULL);
+}
+
 /*
  * Buffers the given amount of input characters.  This is intended to be used
  * together with packet_read_poll.
  */
 
 void
-packet_process_incoming(const char *buf, unsigned int len)
+packet_process_incoming(const char *buf, u_int len)
 {
-       buffer_append(&input, buf, len);
+       if (active_state->packet_discard) {
+               active_state->keep_alive_timeouts = 0; /* ?? */
+               if (len >= active_state->packet_discard)
+                       packet_stop_discard();
+               active_state->packet_discard -= len;
+               return;
+       }
+       buffer_append(&active_state->input, buf, len);
 }
 
 /* Returns a character from the packet. */
 
-unsigned int
-packet_get_char()
+u_int
+packet_get_char(void)
 {
        char ch;
-       buffer_get(&incoming_packet, &ch, 1);
-       return (unsigned char) ch;
+
+       buffer_get(&active_state->incoming_packet, &ch, 1);
+       return (u_char) ch;
 }
 
 /* Returns an integer from the packet data. */
 
-unsigned int
-packet_get_int()
+u_int
+packet_get_int(void)
+{
+       return buffer_get_int(&active_state->incoming_packet);
+}
+
+/* Returns an 64 bit integer from the packet data. */
+
+u_int64_t
+packet_get_int64(void)
 {
-       return buffer_get_int(&incoming_packet);
+       return buffer_get_int64(&active_state->incoming_packet);
 }
 
 /*
@@ -1033,30 +1500,31 @@ packet_get_int()
  */
 
 void
-packet_get_bignum(BIGNUM * value, int *length_ptr)
+packet_get_bignum(BIGNUM * value)
 {
-       *length_ptr = buffer_get_bignum(&incoming_packet, value);
+       buffer_get_bignum(&active_state->incoming_packet, value);
 }
 
 void
-packet_get_bignum2(BIGNUM * value, int *length_ptr)
+packet_get_bignum2(BIGNUM * value)
 {
-       *length_ptr = buffer_get_bignum2(&incoming_packet, value);
+       buffer_get_bignum2(&active_state->incoming_packet, value);
 }
 
-char *
-packet_get_raw(int *length_ptr)
+void *
+packet_get_raw(u_int *length_ptr)
 {
-       int bytes = buffer_len(&incoming_packet);
+       u_int bytes = buffer_len(&active_state->incoming_packet);
+
        if (length_ptr != NULL)
                *length_ptr = bytes;
-       return buffer_ptr(&incoming_packet);
+       return buffer_ptr(&active_state->incoming_packet);
 }
 
 int
 packet_remaining(void)
 {
-       return buffer_len(&incoming_packet);
+       return buffer_len(&active_state->incoming_packet);
 }
 
 /*
@@ -1066,10 +1534,16 @@ packet_remaining(void)
  * integer into which the length of the string is stored.
  */
 
-char *
-packet_get_string(unsigned int *length_ptr)
+void *
+packet_get_string(u_int *length_ptr)
+{
+       return buffer_get_string(&active_state->incoming_packet, length_ptr);
+}
+
+void *
+packet_get_string_ptr(u_int *length_ptr)
 {
-       return buffer_get_string(&incoming_packet, length_ptr);
+       return buffer_get_string_ptr(&active_state->incoming_packet, length_ptr);
 }
 
 /*
@@ -1087,6 +1561,9 @@ packet_send_debug(const char *fmt,...)
        char buf[1024];
        va_list args;
 
+       if (compat20 && (datafellows & SSH_BUG_DEBUG))
+               return;
+
        va_start(args, fmt);
        vsnprintf(buf, sizeof(buf), fmt, args);
        va_end(args);
@@ -1117,6 +1594,7 @@ packet_disconnect(const char *fmt,...)
        char buf[1024];
        va_list args;
        static int disconnecting = 0;
+
        if (disconnecting)      /* Guard against recursive invocations. */
                fatal("packet_disconnect called recursively.");
        disconnecting = 1;
@@ -1129,6 +1607,9 @@ packet_disconnect(const char *fmt,...)
        vsnprintf(buf, sizeof(buf), fmt, args);
        va_end(args);
 
+       /* Display the error locally */
+       logit("Disconnecting: %.100s", buf);
+
        /* Send the disconnect message to the other side, and wait for it to get sent. */
        if (compat20) {
                packet_start(SSH2_MSG_DISCONNECT);
@@ -1137,37 +1618,40 @@ 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();
 
        /* Stop listening for connections. */
-       channel_stop_listening();
+       channel_close_all();
 
        /* Close the connection. */
        packet_close();
-
-       /* Display the error locally and exit. */
-       log("Disconnecting: %.100s", buf);
-       fatal_cleanup();
+       cleanup_exit(255);
 }
 
 /* Checks if there is any buffered output, and tries to write some of the output. */
 
 void
-packet_write_poll()
+packet_write_poll(void)
 {
-       int len = buffer_len(&output);
+       int len = buffer_len(&active_state->output);
+       int cont;
+
        if (len > 0) {
-               len = write(connection_out, buffer_ptr(&output), len);
-               if (len <= 0) {
-                       if (errno == EAGAIN)
+               cont = 0;
+               len = roaming_write(active_state->connection_out,
+                   buffer_ptr(&active_state->output), len, &cont);
+               if (len == -1) {
+                       if (errno == EINTR || errno == EAGAIN ||
+                           errno == EWOULDBLOCK)
                                return;
-                       else
-                               fatal("Write failed: %.100s", strerror(errno));
+                       fatal("Write failed: %.100s", strerror(errno));
                }
-               buffer_consume(&output, len);
+               if (len == 0 && !cont)
+                       fatal("Write connection closed");
+               buffer_consume(&active_state->output, len);
        }
 }
 
@@ -1177,111 +1661,279 @@ packet_write_poll()
  */
 
 void
-packet_write_wait()
+packet_write_wait(void)
 {
+       fd_set *setp;
+       int ret, ms_remain;
+       struct timeval start, timeout, *timeoutp = NULL;
+
+       setp = (fd_set *)xcalloc(howmany(active_state->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(active_state->connection_out + 1,
+                   NFDBITS) * sizeof(fd_mask));
+               FD_SET(active_state->connection_out, setp);
+
+               if (active_state->packet_timeout_ms > 0) {
+                       ms_remain = active_state->packet_timeout_ms;
+                       timeoutp = &timeout;
+               }
+               for (;;) {
+                       if (active_state->packet_timeout_ms != -1) {
+                               ms_to_timeval(&timeout, ms_remain);
+                               gettimeofday(&start, NULL);
+                       }
+                       if ((ret = select(active_state->connection_out + 1,
+                           NULL, setp, NULL, timeoutp)) >= 0)
+                               break;
+                       if (errno != EAGAIN && errno != EINTR &&
+                           errno != EWOULDBLOCK)
+                               break;
+                       if (active_state->packet_timeout_ms == -1)
+                               continue;
+                       ms_subtract_diff(&start, &ms_remain);
+                       if (ms_remain <= 0) {
+                               ret = 0;
+                               break;
+                       }
+               }
+               if (ret == 0) {
+                       logit("Connection to %.200s timed out while "
+                           "waiting to write", get_remote_ipaddr());
+                       cleanup_exit(255);
+               }
                packet_write_poll();
        }
+       xfree(setp);
 }
 
 /* Returns true if there is buffered data to write to the connection. */
 
 int
-packet_have_data_to_write()
+packet_have_data_to_write(void)
 {
-       return buffer_len(&output) != 0;
+       return buffer_len(&active_state->output) != 0;
 }
 
 /* Returns true if there is not too much data to write to the connection. */
 
 int
-packet_not_very_much_data_to_write()
+packet_not_very_much_data_to_write(void)
 {
-       if (interactive_mode)
-               return buffer_len(&output) < 16384;
+       if (active_state->interactive_mode)
+               return buffer_len(&active_state->output) < 16384;
        else
-               return buffer_len(&output) < 128 * 1024;
+               return buffer_len(&active_state->output) < 128 * 1024;
+}
+
+static void
+packet_set_tos(int interactive)
+{
+#if defined(IP_TOS) && !defined(IP_TOS_IS_BROKEN)
+       int tos = interactive ? IPTOS_LOWDELAY : IPTOS_THROUGHPUT;
+
+       if (!packet_connection_is_on_socket() ||
+           !packet_connection_is_ipv4())
+               return;
+       if (setsockopt(active_state->connection_in, IPPROTO_IP, IP_TOS, &tos,
+           sizeof(tos)) < 0)
+               error("setsockopt IP_TOS %d: %.100s:",
+                   tos, strerror(errno));
+#endif
 }
 
 /* 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)
 {
-       int on = 1;
+       if (active_state->set_interactive_called)
+               return;
+       active_state->set_interactive_called = 1;
 
        /* Record that we are in interactive mode. */
-       interactive_mode = interactive;
+       active_state->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
-        */
-       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));
-#endif
-               if (setsockopt(connection_in, IPPROTO_TCP, TCP_NODELAY, (void *) &on,
-                   sizeof(on)) < 0)
-                       error("setsockopt TCP_NODELAY: %.100s", strerror(errno));
-       } else {
-               /*
-                * 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));
-#endif
-       }
+       set_nodelay(active_state->connection_in);
+       packet_set_tos(interactive);
 }
 
 /* Returns true if the current connection is interactive. */
 
 int
-packet_is_interactive()
+packet_is_interactive(void)
 {
-       return interactive_mode;
+       return active_state->interactive_mode;
 }
 
 int
-packet_set_maxsize(int s)
+packet_set_maxsize(u_int s)
 {
-       static int called = 0;
-       if (called) {
-               log("packet_set_maxsize: called twice: old %d new %d",
-                   max_packet_size, s);
+       if (active_state->set_maxsize_called) {
+               logit("packet_set_maxsize: called twice: old %d new %d",
+                   active_state->max_packet_size, s);
                return -1;
        }
        if (s < 4 * 1024 || s > 1024 * 1024) {
-               log("packet_set_maxsize: bad size %d", s);
+               logit("packet_set_maxsize: bad size %d", s);
                return -1;
        }
-       log("packet_set_maxsize: setting to %d", s);
-       max_packet_size = s;
+       active_state->set_maxsize_called = 1;
+       debug("packet_set_maxsize: setting to %d", s);
+       active_state->max_packet_size = s;
        return s;
 }
+
+int
+packet_inc_alive_timeouts(void)
+{
+       return ++active_state->keep_alive_timeouts;
+}
+
+void
+packet_set_alive_timeouts(int ka)
+{
+       active_state->keep_alive_timeouts = ka;
+}
+
+u_int
+packet_get_maxsize(void)
+{
+       return active_state->max_packet_size;
+}
+
+/* roundup current message to pad bytes */
+void
+packet_add_padding(u_char pad)
+{
+       active_state->extra_pad = pad;
+}
+
+/*
+ * 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.
+ */
+void
+packet_send_ignore(int nbytes)
+{
+       u_int32_t rnd = 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)
+                       rnd = arc4random();
+               packet_put_char((u_char)rnd & 0xff);
+               rnd >>= 8;
+       }
+}
+
+#define MAX_PACKETS    (1U<<31)
+int
+packet_need_rekeying(void)
+{
+       if (datafellows & SSH_BUG_NOREKEY)
+               return 0;
+       return
+           (active_state->p_send.packets > MAX_PACKETS) ||
+           (active_state->p_read.packets > MAX_PACKETS) ||
+           (active_state->max_blocks_out &&
+               (active_state->p_send.blocks > active_state->max_blocks_out)) ||
+           (active_state->max_blocks_in &&
+               (active_state->p_read.blocks > active_state->max_blocks_in));
+}
+
+void
+packet_set_rekey_limit(u_int32_t bytes)
+{
+       active_state->rekey_limit = bytes;
+}
+
+void
+packet_set_server(void)
+{
+       active_state->server_side = 1;
+}
+
+void
+packet_set_authenticated(void)
+{
+       active_state->after_authentication = 1;
+}
+
+void *
+packet_get_input(void)
+{
+       return (void *)&active_state->input;
+}
+
+void *
+packet_get_output(void)
+{
+       return (void *)&active_state->output;
+}
+
+void *
+packet_get_newkeys(int mode)
+{
+       return (void *)active_state->newkeys[mode];
+}
+
+/*
+ * Save the state for the real connection, and use a separate state when
+ * resuming a suspended connection.
+ */
+void
+packet_backup_state(void)
+{
+       struct session_state *tmp;
+
+       close(active_state->connection_in);
+       active_state->connection_in = -1;
+       close(active_state->connection_out);
+       active_state->connection_out = -1;
+       if (backup_state)
+               tmp = backup_state;
+       else
+               tmp = alloc_session_state();
+       backup_state = active_state;
+       active_state = tmp;
+}
+
+/*
+ * Swap in the old state when resuming a connecion.
+ */
+void
+packet_restore_state(void)
+{
+       struct session_state *tmp;
+       void *buf;
+       u_int len;
+
+       tmp = backup_state;
+       backup_state = active_state;
+       active_state = tmp;
+       active_state->connection_in = backup_state->connection_in;
+       backup_state->connection_in = -1;
+       active_state->connection_out = backup_state->connection_out;
+       backup_state->connection_out = -1;
+       len = buffer_len(&backup_state->input);
+       if (len > 0) {
+               buf = buffer_ptr(&backup_state->input);
+               buffer_append(&active_state->input, buf, len);
+               buffer_clear(&backup_state->input);
+               add_recv_bytes(len);
+       }
+}
This page took 0.154027 seconds and 4 git commands to generate.