From 7ddeac38b6bd0a16f46331cca9ba029492d8648a Mon Sep 17 00:00:00 2001 From: Holger Weiss Date: Tue, 21 Jun 2016 22:54:41 +0200 Subject: [PATCH] XEP-0198: Also count stanzas when socket is closed Don't forget to count stanzas received from the stream management client that are processed right after the connection was lost. --- src/ejabberd_c2s.erl | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/ejabberd_c2s.erl b/src/ejabberd_c2s.erl index 2a1018827..12838f2fe 100644 --- a/src/ejabberd_c2s.erl +++ b/src/ejabberd_c2s.erl @@ -2802,7 +2802,9 @@ check_h_attribute(#state{mgmt_stanzas_out = NumStanzasOut} = StateData, H) -> [jid:to_string(StateData#state.jid), H, NumStanzasOut]), mgmt_queue_drop(StateData, H). -update_num_stanzas_in(#state{mgmt_state = active} = StateData, El) -> +update_num_stanzas_in(StateData, El) + when StateData#state.mgmt_state == active; + StateData#state.mgmt_state == pending -> NewNum = case {is_stanza(El), StateData#state.mgmt_stanzas_in} of {true, 4294967295} -> 0;