Better process session close

diff --git a/src/ejabberd_sm.erl b/src/ejabberd_sm.erl
index 1cd911e11ea70760b5c44a8830403ef6e00fb810..20614a58d61c574ea5623fcf7c884bb4fbe40764 100644 (file)
--- 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)),