From 28b24a32b82c6429462569be14e00bec65a850ee Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= <essen@ninenines.eu>
Date: Wed, 18 Sep 2019 13:55:37 +0200
Subject: Support connecting to HTTPS server via TCP Socks5

---
 src/gun_socks.erl | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

(limited to 'src/gun_socks.erl')

diff --git a/src/gun_socks.erl b/src/gun_socks.erl
index 74f1767..16684f8 100644
--- a/src/gun_socks.erl
+++ b/src/gun_socks.erl
@@ -17,6 +17,7 @@
 -export([check_options/1]).
 -export([name/0]).
 -export([init/4]).
+-export([switch_transport/3]).
 -export([handle/4]).
 -export([closing/4]).
 -export([close/4]).
@@ -89,6 +90,9 @@ init(Owner, Socket, Transport, Opts) ->
 	#socks_state{owner=Owner, socket=Socket, transport=Transport, opts=Opts,
 		version=Version, status=auth_method_select}.
 
+switch_transport(Transport, Socket, State) ->
+	State#socks_state{socket=Socket, transport=Transport}.
+
 handle(Data, State, _, EvHandlerState) ->
 	{handle(Data, State), EvHandlerState}.
 
@@ -128,22 +132,21 @@ handle(<<5, 0, 0, Rest0/bits>>, State=#socks_state{owner=Owner, socket=Socket, t
 	%% @todo Maybe an event indicating success.
 	#{host := NewHost, port := NewPort} = Opts,
 	case Opts of
-		%% @todo TLS over TLS here as well.
-%		#{protocols := Protocols, transport := tls} ->
-%			TLSOpts = maps:get(tls_opts, Destination, []),
-%			TLSTimeout = maps:get(tls_handshake_timeout, Destination, infinity),
-			%%
+		#{transport := tls} ->
+			HandshakeEvent = #{
+				tls_opts => maps:get(tls_opts, Opts, []),
+				timeout => maps:get(tls_handshake_timeout, Opts, infinity)
+			},
+			[{origin, <<"https">>, NewHost, NewPort, socks5},
+				{tls_handshake, HandshakeEvent, maps:get(protocols, Opts, [http])}];
 		#{protocols := [{socks, SockOpts}]} ->
-			Owner ! {gun_socks_connected, self(), name()},
 			[{origin, <<"http">>, NewHost, NewPort, socks5},
 				{switch_protocol, ?MODULE, init(Owner, Socket, Transport, SockOpts)}];
 		#{protocols := [http2]} ->
-			Owner ! {gun_socks_connected, self(), gun_http2:name()},
 			[{origin, <<"http">>, NewHost, NewPort, socks5},
 				{switch_protocol, gun_http2, State},
 				{mode, http}];
 		_ ->
-			Owner ! {gun_socks_connected, self(), gun_http:name()},
 			[{origin, <<"http">>, NewHost, NewPort, socks5},
 				{switch_protocol, gun_http, State},
 				{mode, http}]
-- 
cgit v1.2.3