From 4f69325e3c8f8ad842527d36385c3108436b94f3 Mon Sep 17 00:00:00 2001 From: Evgeny Khramtsov Date: Sat, 22 Jun 2019 10:19:17 +0300 Subject: Fix shaper selection in ejabberd_c2s --- src/ejabberd_c2s.erl | 4 ++-- src/ejabberd_s2s_in.erl | 2 +- src/ejabberd_s2s_out.erl | 2 +- src/ejabberd_service.erl | 2 +- 4 files changed, 5 insertions(+), 5 deletions(-) (limited to 'src') diff --git a/src/ejabberd_c2s.erl b/src/ejabberd_c2s.erl index 55f3111c0..820e9bbf7 100644 --- a/src/ejabberd_c2s.erl +++ b/src/ejabberd_c2s.erl @@ -55,7 +55,7 @@ -define(SETS, gb_sets). --type state() :: map(). +-type state() :: xmpp_stream_in:state(). -export_type([state/0]). %%%=================================================================== @@ -926,7 +926,7 @@ fix_from_to(Pkt, _State) -> Pkt. -spec change_shaper(state()) -> state(). -change_shaper(#{shaper := ShaperName, ip := IP, lserver := LServer, +change_shaper(#{shaper := ShaperName, ip := {IP, _}, lserver := LServer, user := U, server := S, resource := R} = State) -> JID = jid:make(U, S, R), Shaper = ejabberd_shaper:match(LServer, ShaperName, diff --git a/src/ejabberd_s2s_in.erl b/src/ejabberd_s2s_in.erl index 8b56d440e..b7b8fa1bf 100644 --- a/src/ejabberd_s2s_in.erl +++ b/src/ejabberd_s2s_in.erl @@ -46,7 +46,7 @@ -include("xmpp.hrl"). -include("logger.hrl"). --type state() :: map(). +-type state() :: xmpp_stream_in:state(). -export_type([state/0]). %%%=================================================================== diff --git a/src/ejabberd_s2s_out.erl b/src/ejabberd_s2s_out.erl index 5b27f96c4..818d16a5f 100644 --- a/src/ejabberd_s2s_out.erl +++ b/src/ejabberd_s2s_out.erl @@ -44,7 +44,7 @@ -include("xmpp.hrl"). -include("logger.hrl"). --type state() :: map(). +-type state() :: xmpp_stream_out:state(). -export_type([state/0]). %%%=================================================================== diff --git a/src/ejabberd_service.erl b/src/ejabberd_service.erl index 2662d04d0..2710422a5 100644 --- a/src/ejabberd_service.erl +++ b/src/ejabberd_service.erl @@ -38,7 +38,7 @@ -include("xmpp.hrl"). -include("logger.hrl"). --type state() :: map(). +-type state() :: xmpp_stream_in:state(). -export_type([state/0]). %%%=================================================================== -- cgit v1.2.3