From: Evgeniy Khramtsov <ekhramtsov@process-one.net>
Date: Fri, 14 Apr 2017 17:41:25 +0000 (+0300)
Subject: Better process session close
X-Git-Tag: 17.06-beta~156
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=0a7eb337720d03d4e34b2d70cbed2c6837ef1382;p=ejabberd

Better process session close
---

diff --git a/src/ejabberd_sm.erl b/src/ejabberd_sm.erl
index 1cd911e11..20614a58d 100644
--- a/src/ejabberd_sm.erl
+++ b/src/ejabberd_sm.erl
@@ -162,11 +162,12 @@ close_session(SID, User, Server, Resource) ->
     LServer = jid:nameprep(Server),
     LResource = jid:resourceprep(Resource),
     Mod = get_sm_backend(LServer),
-    Info = case get_sessions(Mod, LUser, LServer, LResource) of
-	       [#session{info = I} = Session|_] ->
+    Sessions = get_sessions(Mod, LUser, LServer, LResource),
+    Info = case lists:keyfind(SID, #session.sid, Sessions) of
+	       #session{info = I} = Session ->
 		   delete_session(Mod, Session),
 		   I;
-	       [] ->
+	       _ ->
 		   []
 	   end,
     JID = jid:make(User, Server, Resource),
@@ -472,7 +473,8 @@ host_down(Host) ->
     Mod = get_sm_backend(Host),
     lists:foreach(
       fun(#session{sid = {_, Pid}}) when node(Pid) == node() ->
-	      ejabberd_c2s:send(Pid, xmpp:serr_system_shutdown());
+	      ejabberd_c2s:send(Pid, xmpp:serr_system_shutdown()),
+	      ejabberd_c2s:stop(Pid);
 	 (_) ->
 	      ok
       end, get_sessions(Mod, Host)),