code refactoring
lqqyt2423
2 years ago
parent
c3dc20feb5
commit
7f55903797
@ -1,160 +0,0 @@
|
|||||||
package proxy
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bufio"
|
|
||||||
"context"
|
|
||||||
"crypto/tls"
|
|
||||||
"net"
|
|
||||||
"net/http"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/lqqyt2423/go-mitmproxy/cert"
|
|
||||||
log "github.com/sirupsen/logrus"
|
|
||||||
)
|
|
||||||
|
|
||||||
// 模拟了标准库中 server 运行，目的是仅通过当前进程内存转发 socket 数据，不需要经过 tcp 或 unix socket
|
|
||||||
|
|
||||||
type pipeAddr struct {
|
|
||||||
remoteAddr string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (pipeAddr) Network() string { return "pipe" }
|
|
||||||
func (a *pipeAddr) String() string { return a.remoteAddr }
|
|
||||||
|
|
||||||
// add Peek method for conn
|
|
||||||
type pipeConn struct {
|
|
||||||
net.Conn
|
|
||||||
r *bufio.Reader
|
|
||||||
host string
|
|
||||||
remoteAddr string
|
|
||||||
connContext *ConnContext
|
|
||||||
}
|
|
||||||
|
|
||||||
func newPipeConn(c net.Conn, req *http.Request) *pipeConn {
|
|
||||||
return &pipeConn{
|
|
||||||
Conn: c,
|
|
||||||
r: bufio.NewReader(c),
|
|
||||||
host: req.Host,
|
|
||||||
remoteAddr: req.RemoteAddr,
|
|
||||||
connContext: req.Context().Value(connContextKey).(*ConnContext),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *pipeConn) Peek(n int) ([]byte, error) {
|
|
||||||
return c.r.Peek(n)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *pipeConn) Read(data []byte) (int, error) {
|
|
||||||
return c.r.Read(data)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *pipeConn) RemoteAddr() net.Addr {
|
|
||||||
return &pipeAddr{remoteAddr: c.remoteAddr}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 建立客户端和服务端通信的通道
|
|
||||||
func newPipes(req *http.Request) (net.Conn, *pipeConn) {
|
|
||||||
client, srv := net.Pipe()
|
|
||||||
server := newPipeConn(srv, req)
|
|
||||||
return client, server
|
|
||||||
}
|
|
||||||
|
|
||||||
// mock net.Listener
|
|
||||||
type middleListener struct {
|
|
||||||
connChan chan net.Conn
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *middleListener) Accept() (net.Conn, error) { return <-l.connChan, nil }
|
|
||||||
func (l *middleListener) Close() error { return nil }
|
|
||||||
func (l *middleListener) Addr() net.Addr { return nil }
|
|
||||||
|
|
||||||
// middle: man-in-the-middle server
|
|
||||||
type middle struct {
|
|
||||||
proxy *Proxy
|
|
||||||
ca *cert.CA
|
|
||||||
listener *middleListener
|
|
||||||
server *http.Server
|
|
||||||
}
|
|
||||||
|
|
||||||
func newMiddle(proxy *Proxy) (interceptor, error) {
|
|
||||||
ca, err := cert.NewCA(proxy.Opts.CaRootPath)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
m := &middle{
|
|
||||||
proxy: proxy,
|
|
||||||
ca: ca,
|
|
||||||
listener: &middleListener{
|
|
||||||
connChan: make(chan net.Conn),
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
server := &http.Server{
|
|
||||||
Handler: m,
|
|
||||||
ConnContext: func(ctx context.Context, c net.Conn) context.Context {
|
|
||||||
return context.WithValue(ctx, connContextKey, c.(*tls.Conn).NetConn().(*pipeConn).connContext)
|
|
||||||
},
|
|
||||||
TLSNextProto: make(map[string]func(*http.Server, *tls.Conn, http.Handler)), // disable http2
|
|
||||||
TLSConfig: &tls.Config{
|
|
||||||
GetCertificate: func(chi *tls.ClientHelloInfo) (*tls.Certificate, error) {
|
|
||||||
log.Debugf("Middle GetCertificate ServerName: %v\n", chi.ServerName)
|
|
||||||
return ca.GetCert(chi.ServerName)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
m.server = server
|
|
||||||
return m, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *middle) Start() error {
|
|
||||||
return m.server.ServeTLS(m.listener, "", "")
|
|
||||||
}
|
|
||||||
|
|
||||||
// todo: should block until ServerConnected
|
|
||||||
func (m *middle) Dial(req *http.Request) (net.Conn, error) {
|
|
||||||
pipeClientConn, pipeServerConn := newPipes(req)
|
|
||||||
go m.intercept(pipeServerConn)
|
|
||||||
return pipeClientConn, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *middle) ServeHTTP(res http.ResponseWriter, req *http.Request) {
|
|
||||||
if strings.EqualFold(req.Header.Get("Connection"), "Upgrade") && strings.EqualFold(req.Header.Get("Upgrade"), "websocket") {
|
|
||||||
// wss
|
|
||||||
defaultWebSocket.wss(res, req)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if req.URL.Scheme == "" {
|
|
||||||
req.URL.Scheme = "https"
|
|
||||||
}
|
|
||||||
if req.URL.Host == "" {
|
|
||||||
req.URL.Host = req.Host
|
|
||||||
}
|
|
||||||
m.proxy.ServeHTTP(res, req)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 解析 connect 流量
|
|
||||||
// 如果是 tls 流量，则进入 listener.Accept => Middle.ServeHTTP
|
|
||||||
// 否则很可能是 ws 流量
|
|
||||||
func (m *middle) intercept(pipeServerConn *pipeConn) {
|
|
||||||
log := log.WithField("in", "middle.intercept").WithField("host", pipeServerConn.host)
|
|
||||||
|
|
||||||
buf, err := pipeServerConn.Peek(3)
|
|
||||||
if err != nil {
|
|
||||||
log.Errorf("Peek error: %v\n", err)
|
|
||||||
pipeServerConn.Close()
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// https://github.com/mitmproxy/mitmproxy/blob/main/mitmproxy/net/tls.py is_tls_record_magic
|
|
||||||
if buf[0] == 0x16 && buf[1] == 0x03 && buf[2] <= 0x03 {
|
|
||||||
// tls
|
|
||||||
pipeServerConn.connContext.ClientConn.Tls = true
|
|
||||||
pipeServerConn.connContext.initHttpsServerConn()
|
|
||||||
m.listener.connChan <- pipeServerConn
|
|
||||||
} else {
|
|
||||||
// ws
|
|
||||||
defaultWebSocket.ws(pipeServerConn, pipeServerConn.host)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue