From: Holger Weiss <holger@zedat.fu-berlin.de>
Date: Tue, 21 Jun 2016 20:54:41 +0000 (+0200)
Subject: XEP-0198: Also count stanzas when socket is closed
X-Git-Tag: 16.06~20
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=7ddeac38b6bd0a16f46331cca9ba029492d8648a;p=ejabberd

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.
---

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;