1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-23 02:26:49 -05:00
v2fly/proxy/http/http.go

264 lines
6.5 KiB
Go
Raw Normal View History

2015-10-28 07:13:27 -04:00
package http
import (
2015-12-14 18:53:40 -05:00
"bufio"
2015-12-15 10:00:47 -05:00
"io"
2015-10-28 07:13:27 -04:00
"net"
2015-12-14 11:26:29 -05:00
"net/http"
2015-12-14 18:53:40 -05:00
"strconv"
2015-12-14 11:26:29 -05:00
"strings"
2015-12-15 10:00:47 -05:00
"sync"
2015-10-28 07:13:27 -04:00
"github.com/v2ray/v2ray-core/app"
2015-12-14 18:53:40 -05:00
"github.com/v2ray/v2ray-core/common/alloc"
"github.com/v2ray/v2ray-core/common/log"
2015-12-02 15:44:01 -05:00
v2net "github.com/v2ray/v2ray-core/common/net"
"github.com/v2ray/v2ray-core/common/retry"
2016-01-18 06:58:04 -05:00
"github.com/v2ray/v2ray-core/common/serial"
2016-01-19 17:41:40 -05:00
"github.com/v2ray/v2ray-core/proxy"
2015-12-15 10:00:47 -05:00
"github.com/v2ray/v2ray-core/transport/ray"
2015-10-28 07:13:27 -04:00
)
type HttpProxyServer struct {
sync.Mutex
2016-01-19 17:41:40 -05:00
accepting bool
space app.Space
config *Config
tcpListener *net.TCPListener
listeningPort v2net.Port
2015-10-28 07:13:27 -04:00
}
2016-01-16 07:08:50 -05:00
func NewHttpProxyServer(space app.Space, config *Config) *HttpProxyServer {
2015-10-28 07:13:27 -04:00
return &HttpProxyServer{
2015-12-14 18:53:40 -05:00
space: space,
config: config,
2015-10-28 07:13:27 -04:00
}
}
2016-01-19 17:41:40 -05:00
func (this *HttpProxyServer) Port() v2net.Port {
return this.listeningPort
}
func (this *HttpProxyServer) Close() {
this.accepting = false
if this.tcpListener != nil {
this.tcpListener.Close()
2016-01-03 18:33:25 -05:00
this.Lock()
this.tcpListener = nil
this.Unlock()
}
}
2015-12-14 11:26:29 -05:00
func (this *HttpProxyServer) Listen(port v2net.Port) error {
2016-01-19 17:41:40 -05:00
if this.accepting {
if this.listeningPort == port {
return nil
} else {
return proxy.ErrorAlreadyListening
}
}
this.listeningPort = port
2015-12-14 18:53:40 -05:00
tcpListener, err := net.ListenTCP("tcp", &net.TCPAddr{
Port: int(port.Value()),
IP: []byte{0, 0, 0, 0},
})
if err != nil {
return err
2015-10-28 07:13:27 -04:00
}
2016-01-04 02:41:01 -05:00
this.Lock()
this.tcpListener = tcpListener
2016-01-04 02:41:01 -05:00
this.Unlock()
this.accepting = true
go this.accept()
2015-12-14 18:53:40 -05:00
return nil
2015-12-14 11:26:29 -05:00
}
func (this *HttpProxyServer) accept() {
2015-12-14 18:53:40 -05:00
for this.accepting {
retry.Timed(100 /* times */, 100 /* ms */).On(func() error {
2016-01-03 19:19:27 -05:00
this.Lock()
defer this.Unlock()
2016-01-03 19:13:43 -05:00
if !this.accepting {
return nil
}
2016-01-03 19:19:27 -05:00
tcpConn, err := this.tcpListener.AcceptTCP()
if err != nil {
2016-01-18 06:24:33 -05:00
log.Error("Failed to accept HTTP connection: ", err)
2016-01-03 19:19:27 -05:00
return err
}
2016-01-03 19:19:27 -05:00
go this.handleConnection(tcpConn)
return nil
})
2015-12-14 18:53:40 -05:00
}
}
2015-12-14 11:26:29 -05:00
2015-12-16 17:53:38 -05:00
func parseHost(rawHost string, defaultPort v2net.Port) (v2net.Destination, error) {
2015-12-15 10:00:47 -05:00
port := defaultPort
2015-12-14 18:53:40 -05:00
host, rawPort, err := net.SplitHostPort(rawHost)
if err != nil {
if addrError, ok := err.(*net.AddrError); ok && strings.Contains(addrError.Err, "missing port") {
host = rawHost
} else {
return nil, err
}
2015-12-15 10:00:47 -05:00
} else {
intPort, err := strconv.Atoi(rawPort)
if err != nil {
return nil, err
}
port = v2net.Port(intPort)
2015-12-14 11:26:29 -05:00
}
2015-12-15 10:00:47 -05:00
2015-12-14 18:53:40 -05:00
if ip := net.ParseIP(host); ip != nil {
2015-12-16 17:53:38 -05:00
return v2net.TCPDestination(v2net.IPAddress(ip), port), nil
2015-12-14 18:53:40 -05:00
}
2015-12-16 17:53:38 -05:00
return v2net.TCPDestination(v2net.DomainAddress(host), port), nil
2015-12-14 11:26:29 -05:00
}
2015-12-14 18:53:40 -05:00
func (this *HttpProxyServer) handleConnection(conn *net.TCPConn) {
2015-12-15 10:00:47 -05:00
defer conn.Close()
reader := bufio.NewReader(conn)
request, err := http.ReadRequest(reader)
if err != nil {
2016-01-18 06:24:33 -05:00
log.Warning("Failed to read http request: ", err)
return
}
2016-01-18 06:24:33 -05:00
log.Info("Request to Method [", request.Method, "] Host [", request.Host, "] with URL [", request.URL, "]")
defaultPort := v2net.Port(80)
if strings.ToLower(request.URL.Scheme) == "https" {
defaultPort = v2net.Port(443)
}
host := request.Host
if len(host) == 0 {
host = request.URL.Host
}
dest, err := parseHost(host, defaultPort)
if err != nil {
2016-01-18 06:24:33 -05:00
log.Warning("Malformed proxy host (", host, "): ", err)
2016-01-11 06:35:28 -05:00
return
}
if strings.ToUpper(request.Method) == "CONNECT" {
this.handleConnect(request, dest, reader, conn)
} else {
this.handlePlainHTTP(request, dest, reader, conn)
2015-12-16 09:52:40 -05:00
}
}
2015-12-15 16:13:09 -05:00
2015-12-16 17:53:38 -05:00
func (this *HttpProxyServer) handleConnect(request *http.Request, destination v2net.Destination, reader io.Reader, writer io.Writer) {
2015-12-16 09:52:40 -05:00
response := &http.Response{
Status: "200 OK",
StatusCode: 200,
Proto: "HTTP/1.1",
ProtoMajor: 1,
ProtoMinor: 1,
Header: http.Header(make(map[string][]string)),
Body: nil,
ContentLength: 0,
Close: false,
}
2015-12-15 16:13:09 -05:00
2015-12-16 09:52:40 -05:00
buffer := alloc.NewSmallBuffer().Clear()
response.Write(buffer)
writer.Write(buffer.Value)
2015-12-16 10:37:32 -05:00
buffer.Release()
2015-12-15 10:00:47 -05:00
2015-12-16 17:53:38 -05:00
packet := v2net.NewPacket(destination, nil, true)
2015-12-16 09:52:40 -05:00
ray := this.space.PacketDispatcher().DispatchToOutbound(packet)
this.transport(reader, writer, ray)
2015-12-15 10:00:47 -05:00
}
func (this *HttpProxyServer) transport(input io.Reader, output io.Writer, ray ray.InboundRay) {
var wg sync.WaitGroup
wg.Add(2)
defer wg.Wait()
2015-12-15 10:00:47 -05:00
2015-12-16 09:52:40 -05:00
go func() {
v2net.ReaderToChan(ray.InboundInput(), input)
close(ray.InboundInput())
wg.Done()
2015-12-16 09:52:40 -05:00
}()
2015-12-15 10:00:47 -05:00
go func() {
v2net.ChanToWriter(output, ray.InboundOutput())
wg.Done()
2015-12-15 10:00:47 -05:00
}()
}
2015-12-15 10:00:47 -05:00
func stripHopByHopHeaders(request *http.Request) {
// Strip hop-by-hop header basaed on RFC:
// http://www.w3.org/Protocols/rfc2616/rfc2616-sec13.html#sec13.5.1
// https://www.mnot.net/blog/2011/07/11/what_proxies_must_do
request.Header.Del("Proxy-Connection")
request.Header.Del("Proxy-Authenticate")
request.Header.Del("Proxy-Authorization")
request.Header.Del("TE")
request.Header.Del("Trailers")
request.Header.Del("Transfer-Encoding")
request.Header.Del("Upgrade")
// TODO: support keep-alive
connections := request.Header.Get("Connection")
request.Header.Set("Connection", "close")
if len(connections) == 0 {
return
}
for _, h := range strings.Split(connections, ",") {
request.Header.Del(strings.TrimSpace(h))
}
}
func (this *HttpProxyServer) handlePlainHTTP(request *http.Request, dest v2net.Destination, reader *bufio.Reader, writer io.Writer) {
if len(request.URL.Host) <= 0 {
hdr := http.Header(make(map[string][]string))
hdr.Set("Connection", "close")
response := &http.Response{
Status: "400 Bad Request",
StatusCode: 400,
Proto: "HTTP/1.1",
ProtoMajor: 1,
ProtoMinor: 1,
Header: hdr,
Body: nil,
ContentLength: 0,
Close: false,
}
buffer := alloc.NewSmallBuffer().Clear()
response.Write(buffer)
writer.Write(buffer.Value)
buffer.Release()
return
}
request.Host = request.URL.Host
stripHopByHopHeaders(request)
2016-01-04 02:41:01 -05:00
requestBuffer := alloc.NewBuffer().Clear() // Don't release this buffer as it is passed into a Packet.
request.Write(requestBuffer)
2016-01-18 06:58:04 -05:00
log.Debug("Request to remote:\n", serial.BytesLiteral(requestBuffer.Value))
packet := v2net.NewPacket(dest, requestBuffer, true)
ray := this.space.PacketDispatcher().DispatchToOutbound(packet)
defer close(ray.InboundInput())
var wg sync.WaitGroup
wg.Add(1)
go func() {
defer wg.Done()
responseReader := bufio.NewReader(NewChanReader(ray.InboundOutput()))
response, err := http.ReadResponse(responseReader, request)
if err != nil {
return
}
2016-01-04 02:41:01 -05:00
responseBuffer := alloc.NewBuffer().Clear()
defer responseBuffer.Release()
response.Write(responseBuffer)
writer.Write(responseBuffer.Value)
response.Body.Close()
}()
wg.Wait()
2015-10-28 07:13:27 -04:00
}