add support for https proxy

gorilla#548
costanic · Jan 13, 2021 · 9b1460a · 9b1460a
1 parent c3dd95a
commit 9b1460a
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/proxy.go b/proxy.go
@@ -6,6 +6,7 @@ package websocket
 
 import (
 	"bufio"
+	"crypto/tls"
 	"encoding/base64"
 	"errors"
 	"net"
@@ -20,10 +21,21 @@ func (fn netDialerFunc) Dial(network, addr string) (net.Conn, error) {
 	return fn(network, addr)
 }
 
+type tlsDialer struct {
+}
+
+func (t *tlsDialer) Dial(network, addr string) (c net.Conn, err error) {
+	return tls.DialWithDialer(&net.Dialer{}, network, addr, &tls.Config{})
+}
+
 func init() {
 	proxy_RegisterDialerType("http", func(proxyURL *url.URL, forwardDialer proxy_Dialer) (proxy_Dialer, error) {
 		return &httpProxyDialer{proxyURL: proxyURL, forwardDial: forwardDialer.Dial}, nil
 	})
+	proxy_RegisterDialerType("https", func(proxyURL *url.URL, forwardDialer proxy_Dialer) (proxy_Dialer, error) {
+		dialer := &tlsDialer{}
+		return &httpProxyDialer{proxyURL: proxyURL, forwardDial: dialer.Dial}, nil
+	})
 }
 
 type httpProxyDialer struct {