67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
package tunnel
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/Dreamacro/clash/adapters/local"
|
|
C "github.com/Dreamacro/clash/constant"
|
|
)
|
|
|
|
func (t *Tunnel) handleHTTP(request *adapters.HttpAdapter, proxy C.ProxyAdapter) {
|
|
req := http.Transport{
|
|
Dial: func(string, string) (net.Conn, error) {
|
|
conn := newTrafficTrack(proxy.Conn(), t.traffic)
|
|
return conn, nil
|
|
},
|
|
// from http.DefaultTransport
|
|
MaxIdleConns: 100,
|
|
IdleConnTimeout: 90 * time.Second,
|
|
ExpectContinueTimeout: 1 * time.Second,
|
|
}
|
|
resp, err := req.RoundTrip(request.R)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
header := request.W.Header()
|
|
for k, vv := range resp.Header {
|
|
for _, v := range vv {
|
|
header.Add(k, v)
|
|
}
|
|
}
|
|
request.W.WriteHeader(resp.StatusCode)
|
|
var writer io.Writer = request.W
|
|
if len(resp.TransferEncoding) > 0 && resp.TransferEncoding[0] == "chunked" {
|
|
writer = ChunkWriter{Writer: request.W}
|
|
}
|
|
io.Copy(writer, resp.Body)
|
|
}
|
|
|
|
func (t *Tunnel) handleHTTPS(request *adapters.HttpsAdapter, proxy C.ProxyAdapter) {
|
|
conn := newTrafficTrack(proxy.Conn(), t.traffic)
|
|
go io.Copy(request.Conn(), conn)
|
|
io.Copy(conn, request.Conn())
|
|
}
|
|
|
|
func (t *Tunnel) handleSOCKS(request *adapters.SocksAdapter, proxy C.ProxyAdapter) {
|
|
conn := newTrafficTrack(proxy.Conn(), t.traffic)
|
|
go io.Copy(request.Conn(), conn)
|
|
io.Copy(conn, request.Conn())
|
|
}
|
|
|
|
// ChunkWriter is a writer wrapper and used when TransferEncoding is chunked
|
|
type ChunkWriter struct {
|
|
io.Writer
|
|
}
|
|
|
|
func (cw ChunkWriter) Write(b []byte) (int, error) {
|
|
n, err := cw.Writer.Write(b)
|
|
if err == nil {
|
|
cw.Writer.(http.Flusher).Flush()
|
|
}
|
|
return n, err
|
|
}
|