* src/ejabberd_frontend_socket.erl: Fixed SSL sockets

SVN Revision: 1664

diff --git a/ChangeLog b/ChangeLog
index 20c1934aad42a93c8eb8038603200a6d0601cea2..17b7e53ebe3c79c4f2482e07e767fbe8e55f4968 100644 (file)
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,7 @@
+2008-10-23  Alexey Shchepin  <alexey@process-one.net>
+
+       * src/ejabberd_frontend_socket.erl: Fixed SSL sockets
+
 2008-10-17  Badlop  <badlop@process-one.net>
 
        * src/Makefile.in: docdir should be prefixed with DESTDIR (thanks

diff --git a/src/ejabberd_frontend_socket.erl b/src/ejabberd_frontend_socket.erl
index 320968f70fa8c85500f39499c7592e215975fb9b..e2d244800321ee348e825a97aa2411ae13751a1b 100644 (file)
--- a/src/ejabberd_frontend_socket.erl
+++ b/src/ejabberd_frontend_socket.erl
@@ -93,7 +93,7 @@ start(Module, SockMod, Socket, Opts) ->
     end.
 
 starttls(FsmRef, TLSOpts) ->
-    gen_server:call(FsmRef, {starttls, TLSOpts}),
+    %gen_server:call(FsmRef, {starttls, TLSOpts}),
     FsmRef.
 
 starttls(FsmRef, TLSOpts, Data) ->
@@ -136,7 +136,8 @@ sockname(FsmRef) ->
     gen_server:call(FsmRef, sockname).
 
 peername(FsmRef) ->
-    gen_server:call(FsmRef, peername).
+    %gen_server:call(FsmRef, peername).
+    {ok, {{0, 0, 0, 0}, 0}}.
 
 
 %%====================================================================
@@ -153,11 +154,12 @@ peername(FsmRef) ->
 init([Module, SockMod, Socket, Opts, Receiver]) ->
     %% TODO: monitor the receiver
     Node = ejabberd_node_groups:get_closest_node(backend),
+    {SockMod2, Socket2} = check_starttls(SockMod, Socket, Receiver, Opts),
     {ok, Pid} =
        rpc:call(Node, Module, start, [{?MODULE, self()}, Opts]),
     ejabberd_receiver:become_controller(Receiver, Pid),
-    {ok, #state{sockmod = SockMod,
-               socket = Socket,
+    {ok, #state{sockmod = SockMod2,
+               socket = Socket2,
                receiver = Receiver}}.
 
 %%--------------------------------------------------------------------
@@ -298,3 +300,17 @@ code_change(_OldVsn, State, _Extra) ->
 %%--------------------------------------------------------------------
 %%% Internal functions
 %%--------------------------------------------------------------------
+check_starttls(SockMod, Socket, Receiver, Opts) ->
+    TLSEnabled = lists:member(tls, Opts),
+    TLSOpts = lists:filter(fun({certfile, _}) -> true;
+                             (_) -> false
+                          end, Opts),
+    if
+       TLSEnabled ->
+           {ok, TLSSocket} = tls:tcp_to_tls(Socket, TLSOpts),
+           ejabberd_receiver:starttls(Receiver, TLSSocket),
+           {tls, TLSSocket};
+       true ->
+           {SockMod, Socket}
+    end.
+