From 4d64fbb0ac4a250ffdbd5144d124c6795a90d543 Mon Sep 17 00:00:00 2001
From: Evgeniy Khramtsov <ekhramtsov@process-one.net>
Date: Tue, 3 Jan 2012 11:25:55 +1000
Subject: [PATCH] Also migrate C2S sessions with remote receivers (EJABS-1798)

---
 src/ejabberd_c2s.erl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/ejabberd_c2s.erl b/src/ejabberd_c2s.erl
index 995cbc4f7..6b0308d32 100644
--- a/src/ejabberd_c2s.erl
+++ b/src/ejabberd_c2s.erl
@@ -1748,7 +1748,8 @@ handle_info({migrate, Node}, StateName, StateData) ->
     end;
 handle_info({migrate_shutdown, Node, After}, StateName, StateData) ->
     if StateData#state.sockmod == ejabberd_frontend_socket orelse
-       StateData#state.xml_socket == true ->
+       StateData#state.xml_socket == true orelse
+       node(StateData#state.socket_monitor) /= node() ->
             migrate(self(), Node, After);
        true ->
             self() ! system_shutdown