clash/tunnel/connection.go

98 lines
2.2 KiB
Go
Raw Normal View History

package tunnel
import (
2018-08-27 00:06:40 +08:00
"bufio"
"io"
2018-09-17 00:15:58 +08:00
"net"
2018-08-27 00:06:40 +08:00
"net/http"
"strings"
"sync"
2018-09-21 11:33:29 +08:00
"time"
2018-12-22 23:56:42 +08:00
adapters "github.com/Dreamacro/clash/adapters/inbound"
)
const (
// io.Copy default buffer size is 32 KiB
// but the maximum packet size of vmess/shadowsocks is about 16 KiB
// so define a buffer of 20 KiB to reduce the memory of each TCP relay
bufferSize = 20 * 1024
)
var bufPool = sync.Pool{New: func() interface{} { return make([]byte, bufferSize) }}
2018-12-22 23:56:42 +08:00
func (t *Tunnel) handleHTTP(request *adapters.HTTPAdapter, outbound net.Conn) {
conn := newTrafficTrack(outbound, t.traffic)
2018-08-27 00:06:40 +08:00
req := request.R
host := req.Host
2018-08-27 00:06:40 +08:00
for {
keepAlive := strings.TrimSpace(strings.ToLower(req.Header.Get("Proxy-Connection"))) == "keep-alive"
2018-08-27 00:06:40 +08:00
req.Header.Set("Connection", "close")
req.RequestURI = ""
adapters.RemoveHopByHopHeaders(req.Header)
err := req.Write(conn)
if err != nil {
break
}
2018-08-27 00:06:40 +08:00
br := bufio.NewReader(conn)
resp, err := http.ReadResponse(br, req)
if err != nil {
break
}
adapters.RemoveHopByHopHeaders(resp.Header)
if resp.ContentLength >= 0 {
resp.Header.Set("Proxy-Connection", "keep-alive")
resp.Header.Set("Connection", "keep-alive")
resp.Header.Set("Keep-Alive", "timeout=4")
resp.Close = false
} else {
resp.Close = true
}
err = resp.Write(request.Conn())
if err != nil || resp.Close {
break
}
if !keepAlive {
break
}
2018-08-27 00:06:40 +08:00
req, err = http.ReadRequest(bufio.NewReader(request.Conn()))
if err != nil {
break
}
// Sometimes firefox just open a socket to process multiple domains in HTTP
// The temporary solution is close connection when encountering different HOST
if req.Host != host {
break
}
}
}
2018-12-22 23:56:42 +08:00
func (t *Tunnel) handleSOCKS(request *adapters.SocketAdapter, outbound net.Conn) {
conn := newTrafficTrack(outbound, t.traffic)
2018-09-17 00:15:58 +08:00
relay(request.Conn(), conn)
}
// relay copies between left and right bidirectionally.
func relay(leftConn, rightConn net.Conn) {
ch := make(chan error)
go func() {
buf := bufPool.Get().([]byte)
_, err := io.CopyBuffer(leftConn, rightConn, buf)
bufPool.Put(buf[:cap(buf)])
2018-09-21 11:33:29 +08:00
leftConn.SetReadDeadline(time.Now())
2018-09-17 00:15:58 +08:00
ch <- err
}()
buf := bufPool.Get().([]byte)
io.CopyBuffer(rightConn, leftConn, buf)
bufPool.Put(buf[:cap(buf)])
2018-09-21 11:33:29 +08:00
rightConn.SetReadDeadline(time.Now())
2018-09-17 00:15:58 +08:00
<-ch
}