From 555cd4c1c6b8c1e3d43fb0f097e51844ede29321 Mon Sep 17 00:00:00 2001
From: lqqyt2423 <974923609@qq.com>
Date: Fri, 4 Dec 2020 23:37:44 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E6=9E=90https=E5=88=9D=E6=AD=A5?=
 =?UTF-8?q?=E8=B0=83=E8=AF=95=E6=88=90=E5=8A=9F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cert/cert.go   |  2 +-
 proxy/proxy.go | 45 ++++++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 45 insertions(+), 2 deletions(-)

diff --git a/cert/cert.go b/cert/cert.go
index e07fd0c..a6f06e1 100644
--- a/cert/cert.go
+++ b/cert/cert.go
@@ -253,7 +253,7 @@ func (ca *CA) DummyCert(commonName string) (*tls.Certificate, error) {
 
 	cert := &tls.Certificate{
 		Certificate: [][]byte{certBytes},
-		PrivateKey:  ca.PrivateKey,
+		PrivateKey:  &ca.PrivateKey,
 	}
 
 	return cert, nil
diff --git a/proxy/proxy.go b/proxy/proxy.go
index 6086fc2..e237a56 100644
--- a/proxy/proxy.go
+++ b/proxy/proxy.go
@@ -1,11 +1,14 @@
 package proxy
 
 import (
+	"crypto/tls"
 	"io"
 	"log"
 	"net"
 	"net/http"
 	"time"
+
+	"github.com/lqqyt2423/go-mitmproxy/cert"
 )
 
 type Options struct {
@@ -14,9 +17,25 @@ type Options struct {
 
 type Proxy struct {
 	Server *http.Server
+
+	ca               *cert.CA
+	extraNetListener net.Listener
+	extraServer      *http.Server
 }
 
 func (proxy *Proxy) Start() error {
+	ln, err := net.Listen("tcp", "127.0.0.1:") // port number is automatically chosen
+	if err != nil {
+		return err
+	}
+	proxy.extraNetListener = ln
+	proxy.extraServer.Addr = ln.Addr().String()
+	log.Printf("Proxy extraServer Addr is %v\n", proxy.extraServer.Addr)
+	go func() {
+		defer ln.Close()
+		log.Fatal(proxy.extraServer.ServeTLS(ln, "", ""))
+	}()
+
 	log.Printf("Proxy start listen at %v\n", proxy.Server.Addr)
 	return proxy.Server.ListenAndServe()
 }
@@ -27,6 +46,8 @@ func (proxy *Proxy) ServeHTTP(res http.ResponseWriter, req *http.Request) {
 		return
 	}
 
+	log.Printf("url: %v\n", req.URL.String())
+
 	if !req.URL.IsAbs() || req.URL.Host == "" {
 		res.WriteHeader(400)
 		_, err := io.WriteString(res, "此为代理服务器，不能直接发起请求")
@@ -73,7 +94,12 @@ func (proxy *Proxy) ServeHTTP(res http.ResponseWriter, req *http.Request) {
 func (proxy *Proxy) handleConnect(res http.ResponseWriter, req *http.Request) {
 	log.Printf("CONNECT: %v\n", req.Host)
 
-	conn, err := net.Dial("tcp", req.Host)
+	// 直接转发
+	// conn, err := net.Dial("tcp", req.Host)
+
+	// 内部解析 HTTPS
+	conn, err := net.Dial("tcp", proxy.extraServer.Addr)
+
 	if err != nil {
 		log.Printf("error: %v, host: %v\n", err, req.Host)
 		res.WriteHeader(502)
@@ -118,5 +144,22 @@ func NewProxy(opts *Options) *Proxy {
 		Addr:    opts.Addr,
 		Handler: proxy,
 	}
+
+	ca, err := cert.NewCA("")
+	if err != nil {
+		panic(err)
+	}
+	proxy.ca = ca
+	proxy.extraServer = &http.Server{
+		Handler: proxy,
+		TLSConfig: &tls.Config{
+			PreferServerCipherSuites: true,
+			GetCertificate: func(chi *tls.ClientHelloInfo) (*tls.Certificate, error) {
+				log.Printf("GetCertificate ServerName: %v\n", chi.ServerName)
+				return proxy.ca.DummyCert(chi.ServerName)
+			},
+		},
+	}
+
 	return proxy
 }