1 files changed, 8 insertions, 3 deletions

diff --git a/packet.c b/packet.c
index 9cf200cc3..ffcd8eab9 100644
--- a/packet.c
+++ b/packet.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: packet.c,v 1.222 2016/01/14 16:17:40 markus Exp $ */
+/* $OpenBSD: packet.c,v 1.223 2016/01/29 02:42:46 dtucker Exp $ */
 /*
  * Author: Tatu Ylonen <ylo@cs.hut.fi>
  * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
@@ -2251,16 +2251,21 @@ int
 ssh_packet_need_rekeying(struct ssh *ssh)
 {
         struct session_state *state = ssh->state;
+        u_int32_t buf_in, buf_out;
 
         if (ssh->compat & SSH_BUG_NOREKEY)
                 return 0;
+        buf_in = roundup(sshbuf_len(state->input),
+            state->newkeys[MODE_IN]->enc.block_size);
+        buf_out = roundup(sshbuf_len(state->output),
+            state->newkeys[MODE_OUT]->enc.block_size);
         return
             (state->p_send.packets > MAX_PACKETS) ||
             (state->p_read.packets > MAX_PACKETS) ||
             (state->max_blocks_out &&
-                (state->p_send.blocks > state->max_blocks_out)) ||
+                (state->p_send.blocks + buf_out > state->max_blocks_out)) ||
             (state->max_blocks_in &&
-                (state->p_read.blocks > state->max_blocks_in)) ||
+                (state->p_read.blocks + buf_in > state->max_blocks_in)) ||
             (state->rekey_interval != 0 && state->rekey_time +
                  state->rekey_interval <= monotime());
 }