From 0fe9caa3047aaabfd0e2cb30f4d9aae0e0d12138 Mon Sep 17 00:00:00 2001
From: Alexey Shchepin <alexey@process-one.net>
Date: Tue, 5 Sep 2006 04:36:01 +0000
Subject: [PATCH] * src/ejabberd_socket.erl: All XML socket operations moved
 here * src/ejabberd_s2s_in.erl: Likewise

SVN Revision: 600
---
 src/ejabberd_s2s_in.erl | 10 ++++++----
 src/ejabberd_socket.erl |  8 ++++++++
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/ejabberd_s2s_in.erl b/src/ejabberd_s2s_in.erl
index 527b8cefd..401996ec3 100644
--- a/src/ejabberd_s2s_in.erl
+++ b/src/ejabberd_s2s_in.erl
@@ -140,9 +140,10 @@ wait_for_stream({xmlstreamstart, _Name, Attrs}, StateData) ->
 	    SASL =
 		if
 		    StateData#state.tls_enabled ->
-			case tls:get_peer_certificate(StateData#state.socket) of
+			case ejabberd_socket:get_peer_certificate(
+			       StateData#state.socket) of
 			    {ok, _Cert} ->
-				case tls:get_verify_result(
+				case ejabberd_socket:get_verify_result(
 				       StateData#state.socket) of
 				    0 ->
 					[{xmlelement, "mechanisms",
@@ -222,9 +223,10 @@ wait_for_feature_request({xmlstreamelement, El}, StateData) ->
 		    Auth = jlib:decode_base64(xml:get_cdata(Els)),
 		    AuthDomain = jlib:nameprep(Auth),
 		    AuthRes =
-			case tls:get_peer_certificate(StateData#state.socket) of
+			case ejabberd_socket:get_peer_certificate(
+			       StateData#state.socket) of
 			    {ok, Cert} ->
-				case tls:get_verify_result(
+				case ejabberd_socket:get_verify_result(
 				       StateData#state.socket) of
 				    0 ->
 					case AuthDomain of
diff --git a/src/ejabberd_socket.erl b/src/ejabberd_socket.erl
index 06deac5e5..417d4f2eb 100644
--- a/src/ejabberd_socket.erl
+++ b/src/ejabberd_socket.erl
@@ -18,6 +18,8 @@
 	 send/2,
 	 change_shaper/2,
 	 get_sockmod/1,
+	 get_peer_certificate/1,
+	 get_verify_result/1,
 	 close/1]).
 
 -record(socket_state, {sockmod, socket, receiver}).
@@ -93,6 +95,12 @@ change_shaper(SocketData, Shaper) ->
 get_sockmod(SocketData) ->
     SocketData#socket_state.sockmod.
 
+get_peer_certificate(SocketData) ->
+    tls:get_peer_certificate(SocketData#socket_state.socket).
+
+get_verify_result(SocketData) ->
+    tls:get_verify_result(SocketData#socket_state.socket).
+
 close(SocketData) ->
     ejabberd_receiver:close(SocketData#socket_state.receiver).