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 * Copyright (c) 1995 Tatu Ylonen , 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()); }