1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-22 18:17:52 -05:00
v2fly/proxy/http/server.go

327 lines
9.3 KiB
Go
Raw Normal View History

2015-10-28 07:13:27 -04:00
package http
import (
2017-02-15 16:51:01 -05:00
"bufio"
2017-01-14 18:57:06 -05:00
"context"
2017-10-28 08:25:22 -04:00
"encoding/base64"
2015-12-15 10:00:47 -05:00
"io"
2015-12-14 11:26:29 -05:00
"net/http"
"strings"
2017-01-31 06:42:05 -05:00
"time"
2015-10-28 07:13:27 -04:00
core "github.com/v2fly/v2ray-core/v5"
"github.com/v2fly/v2ray-core/v5/common"
"github.com/v2fly/v2ray-core/v5/common/buf"
"github.com/v2fly/v2ray-core/v5/common/errors"
"github.com/v2fly/v2ray-core/v5/common/log"
"github.com/v2fly/v2ray-core/v5/common/net"
"github.com/v2fly/v2ray-core/v5/common/protocol"
http_proto "github.com/v2fly/v2ray-core/v5/common/protocol/http"
"github.com/v2fly/v2ray-core/v5/common/session"
"github.com/v2fly/v2ray-core/v5/common/signal"
"github.com/v2fly/v2ray-core/v5/common/task"
"github.com/v2fly/v2ray-core/v5/features/policy"
"github.com/v2fly/v2ray-core/v5/features/routing"
"github.com/v2fly/v2ray-core/v5/transport/internet"
2015-10-28 07:13:27 -04:00
)
2018-04-02 04:12:35 -04:00
// Server is an HTTP proxy server.
2016-05-29 10:46:31 -04:00
type Server struct {
2018-10-21 04:27:13 -04:00
config *ServerConfig
policyManager policy.Manager
2015-10-28 07:13:27 -04:00
}
2016-12-22 07:54:11 -05:00
// NewServer creates a new HTTP inbound handler.
func NewServer(ctx context.Context, config *ServerConfig) (*Server, error) {
2018-10-21 04:27:13 -04:00
v := core.MustFromContext(ctx)
2017-01-06 09:32:36 -05:00
s := &Server{
2018-10-21 04:27:13 -04:00
config: config,
policyManager: v.GetFeature(policy.ManagerType()).(policy.Manager),
}
return s, nil
2015-10-28 07:13:27 -04:00
}
2018-10-11 16:34:31 -04:00
func (s *Server) policy() policy.Session {
config := s.config
2018-10-21 04:27:13 -04:00
p := s.policyManager.ForLevel(config.UserLevel)
if config.Timeout > 0 && config.UserLevel == 0 {
p.Timeouts.ConnectionIdle = time.Duration(config.Timeout) * time.Second
}
return p
}
2018-12-31 15:25:10 -05:00
// Network implements proxy.Inbound.
2018-11-20 10:58:26 -05:00
func (*Server) Network() []net.Network {
2020-10-29 03:30:38 -04:00
return []net.Network{net.Network_TCP, net.Network_UNIX}
2017-01-14 18:57:06 -05:00
}
2017-04-11 18:21:46 -04:00
func isTimeout(err error) bool {
2017-11-16 06:01:42 -05:00
nerr, ok := errors.Cause(err).(net.Error)
2017-04-11 18:21:46 -04:00
return ok && nerr.Timeout()
}
2017-10-28 08:25:22 -04:00
func parseBasicAuth(auth string) (username, password string, ok bool) {
const prefix = "Basic "
if !strings.HasPrefix(auth, prefix) {
return
}
c, err := base64.StdEncoding.DecodeString(auth[len(prefix):])
if err != nil {
return
}
cs := string(c)
s := strings.IndexByte(cs, ':')
if s < 0 {
return
}
return cs[:s], cs[s+1:], true
}
2017-11-24 17:24:12 -05:00
type readerOnly struct {
io.Reader
}
func (s *Server) Process(ctx context.Context, network net.Network, conn internet.Connection, dispatcher routing.Dispatcher) error {
inbound := session.InboundFromContext(ctx)
if inbound != nil {
inbound.User = &protocol.MemoryUser{
Level: s.config.UserLevel,
}
}
2017-11-24 17:24:12 -05:00
reader := bufio.NewReaderSize(readerOnly{conn}, buf.Size)
Start:
2018-04-20 10:03:42 -04:00
if err := conn.SetReadDeadline(time.Now().Add(s.policy().Timeouts.Handshake)); err != nil {
newError("failed to set read deadline").Base(err).WriteToLog(session.ExportIDToError(ctx))
2018-04-20 10:03:42 -04:00
}
request, err := http.ReadRequest(reader)
if err != nil {
trace := newError("failed to read http request").Base(err)
2017-04-11 18:21:46 -04:00
if errors.Cause(err) != io.EOF && !isTimeout(errors.Cause(err)) {
trace.AtWarning()
2016-06-03 14:21:46 -04:00
}
return trace
}
2017-10-26 15:44:13 -04:00
2017-12-05 17:49:16 -05:00
if len(s.config.Accounts) > 0 {
2017-10-28 08:25:22 -04:00
user, pass, ok := parseBasicAuth(request.Header.Get("Proxy-Authorization"))
2017-12-02 04:07:42 -05:00
if !ok || !s.config.HasAccount(user, pass) {
2018-04-11 10:15:29 -04:00
return common.Error2(conn.Write([]byte("HTTP/1.1 407 Proxy Authentication Required\r\nProxy-Authenticate: Basic realm=\"proxy\"\r\n\r\n")))
2017-10-26 15:44:13 -04:00
}
if inbound != nil {
inbound.User.Email = user
}
2017-10-26 15:44:13 -04:00
}
newError("request to Method [", request.Method, "] Host [", request.Host, "] with URL [", request.URL, "]").WriteToLog(session.ExportIDToError(ctx))
2018-04-20 10:03:42 -04:00
if err := conn.SetReadDeadline(time.Time{}); err != nil {
newError("failed to clear read deadline").Base(err).WriteToLog(session.ExportIDToError(ctx))
2018-04-20 10:03:42 -04:00
}
2017-01-31 10:49:59 -05:00
defaultPort := net.Port(80)
if strings.EqualFold(request.URL.Scheme, "https") {
defaultPort = net.Port(443)
}
host := request.Host
if host == "" {
host = request.URL.Host
}
2018-12-10 17:08:16 -05:00
dest, err := http_proto.ParseHost(host, defaultPort)
if err != nil {
2017-04-08 19:43:25 -04:00
return newError("malformed proxy host: ", host).AtWarning().Base(err)
}
2019-06-12 01:04:34 -04:00
ctx = log.ContextWithAccessMessage(ctx, &log.AccessMessage{
2017-12-19 15:28:12 -05:00
From: conn.RemoteAddr(),
To: request.URL,
Status: log.AccessAccepted,
2019-06-12 01:04:34 -04:00
Reason: "",
2017-12-19 15:28:12 -05:00
})
if strings.EqualFold(request.Method, "CONNECT") {
return s.handleConnect(ctx, request, reader, conn, dest, dispatcher)
2015-12-16 09:52:40 -05:00
}
keepAlive := (strings.TrimSpace(strings.ToLower(request.Header.Get("Proxy-Connection"))) == "keep-alive")
2017-11-17 14:54:24 -05:00
err = s.handlePlainHTTP(ctx, request, conn, dest, dispatcher)
if err == errWaitAnother {
if keepAlive {
goto Start
}
err = nil
}
return err
2015-12-16 09:52:40 -05:00
}
2015-12-15 16:13:09 -05:00
func (s *Server) handleConnect(ctx context.Context, _ *http.Request, reader *bufio.Reader, conn internet.Connection, dest net.Destination, dispatcher routing.Dispatcher) error {
2017-11-17 14:54:24 -05:00
_, err := conn.Write([]byte("HTTP/1.1 200 Connection established\r\n\r\n"))
2017-04-18 07:30:26 -04:00
if err != nil {
2017-04-08 19:43:25 -04:00
return newError("failed to write back OK response").Base(err)
2016-12-22 07:54:11 -05:00
}
2015-12-15 10:00:47 -05:00
2018-05-25 06:08:28 -04:00
plcy := s.policy()
2017-11-14 18:36:14 -05:00
ctx, cancel := context.WithCancel(ctx)
2018-05-25 06:08:28 -04:00
timer := signal.CancelAfterInactivity(ctx, cancel, plcy.Timeouts.ConnectionIdle)
2018-10-11 16:34:31 -04:00
ctx = policy.ContextWithBufferPolicy(ctx, plcy.Buffer)
2018-04-16 18:31:10 -04:00
link, err := dispatcher.Dispatch(ctx, dest)
if err != nil {
return err
}
2015-12-15 10:00:47 -05:00
2017-11-17 14:54:24 -05:00
if reader.Buffered() > 0 {
2018-11-18 13:36:36 -05:00
payload, err := buf.ReadFrom(io.LimitReader(reader, int64(reader.Buffered())))
2018-04-01 18:25:42 -04:00
if err != nil {
return err
}
2018-04-16 18:31:10 -04:00
if err := link.Writer.WriteMultiBuffer(payload); err != nil {
2017-11-17 14:54:24 -05:00
return err
}
2017-11-17 14:58:46 -05:00
reader = nil
2017-11-17 14:54:24 -05:00
}
2018-04-11 10:45:09 -04:00
requestDone := func() error {
defer timer.SetTimeout(plcy.Timeouts.DownlinkOnly)
2015-12-15 10:00:47 -05:00
return buf.Copy(buf.NewReader(conn), link.Writer, buf.UpdateActivity(timer))
2018-04-11 10:45:09 -04:00
}
2015-12-15 10:00:47 -05:00
2018-04-11 10:45:09 -04:00
responseDone := func() error {
defer timer.SetTimeout(plcy.Timeouts.UplinkOnly)
2018-02-23 11:07:45 -05:00
2017-11-17 14:54:24 -05:00
v2writer := buf.NewWriter(conn)
2018-04-16 18:31:10 -04:00
if err := buf.Copy(link.Reader, v2writer, buf.UpdateActivity(timer)); err != nil {
2016-12-29 18:32:20 -05:00
return err
2016-11-21 18:17:49 -05:00
}
2018-02-23 11:07:45 -05:00
2016-12-29 18:32:20 -05:00
return nil
2018-04-11 10:45:09 -04:00
}
2016-12-29 18:32:20 -05:00
2021-05-19 17:28:52 -04:00
closeWriter := task.OnSuccess(requestDone, task.Close(link.Writer))
2018-12-06 05:35:02 -05:00
if err := task.Run(ctx, closeWriter, responseDone); err != nil {
2018-12-31 15:25:10 -05:00
common.Interrupt(link.Reader)
common.Interrupt(link.Writer)
2017-04-08 19:43:25 -04:00
return newError("connection ends").Base(err)
2017-01-10 08:22:42 -05:00
}
return nil
}
2015-12-15 10:00:47 -05:00
var errWaitAnother = newError("keep alive")
2016-06-01 18:57:08 -04:00
func (s *Server) handlePlainHTTP(ctx context.Context, request *http.Request, writer io.Writer, dest net.Destination, dispatcher routing.Dispatcher) error {
if !s.config.AllowTransparent && request.URL.Host == "" {
2017-11-05 16:13:15 -05:00
// RFC 2068 (HTTP/1.1) requires URL to be absolute URL in HTTP proxy.
response := &http.Response{
Status: "Bad Request",
StatusCode: 400,
Proto: "HTTP/1.1",
ProtoMajor: 1,
ProtoMinor: 1,
Header: http.Header(make(map[string][]string)),
Body: nil,
ContentLength: 0,
Close: true,
}
response.Header.Set("Proxy-Connection", "close")
response.Header.Set("Connection", "close")
return response.Write(writer)
}
2017-11-05 16:13:15 -05:00
if len(request.URL.Host) > 0 {
request.Host = request.URL.Host
}
http_proto.RemoveHopByHopHeaders(request.Header)
// Prevent UA from being set to golang's default ones
if request.Header.Get("User-Agent") == "" {
request.Header.Set("User-Agent", "")
}
content := &session.Content{}
2019-02-28 05:45:06 -05:00
2019-02-28 08:36:31 -05:00
content.SetAttribute(":method", strings.ToUpper(request.Method))
2019-02-28 05:45:06 -05:00
content.SetAttribute(":path", request.URL.Path)
for key := range request.Header {
value := request.Header.Get(key)
content.SetAttribute(strings.ToLower(key), value)
}
ctx = session.ContextWithContent(ctx, content)
2019-02-28 04:37:27 -05:00
2018-04-16 18:31:10 -04:00
link, err := dispatcher.Dispatch(ctx, dest)
if err != nil {
return err
}
2016-12-29 18:51:39 -05:00
2021-12-26 23:33:08 -05:00
// Plain HTTP request is not a stream. The request always finishes before response. Hence, request has to be closed later.
defer common.Close(link.Writer)
var result error = errWaitAnother
2018-04-11 10:45:09 -04:00
requestDone := func() error {
request.Header.Set("Connection", "close")
2018-04-16 18:31:10 -04:00
requestWriter := buf.NewBufferedWriter(link.Writer)
2017-11-15 05:23:25 -05:00
common.Must(requestWriter.SetBuffered(false))
if err := request.Write(requestWriter); err != nil {
return newError("failed to write whole request").Base(err).AtWarning()
}
return nil
2018-04-11 10:45:09 -04:00
}
2016-12-29 18:32:20 -05:00
2018-04-11 10:45:09 -04:00
responseDone := func() error {
2018-04-20 18:54:53 -04:00
responseReader := bufio.NewReaderSize(&buf.BufferedReader{Reader: link.Reader}, buf.Size)
2020-12-03 03:07:41 -05:00
response, err := http.ReadResponse(responseReader, request)
if err == nil {
http_proto.RemoveHopByHopHeaders(response.Header)
if response.ContentLength >= 0 {
response.Header.Set("Proxy-Connection", "keep-alive")
response.Header.Set("Connection", "keep-alive")
response.Header.Set("Keep-Alive", "timeout=4")
response.Close = false
} else {
response.Close = true
result = nil
}
2020-12-03 03:07:41 -05:00
defer response.Body.Close()
} else {
newError("failed to read response from ", request.Host).Base(err).AtWarning().WriteToLog(session.ExportIDToError(ctx))
response = &http.Response{
Status: "Service Unavailable",
StatusCode: 503,
Proto: "HTTP/1.1",
ProtoMajor: 1,
ProtoMinor: 1,
Header: http.Header(make(map[string][]string)),
Body: nil,
ContentLength: 0,
Close: true,
}
response.Header.Set("Connection", "close")
response.Header.Set("Proxy-Connection", "close")
}
2017-04-17 16:35:20 -04:00
if err := response.Write(writer); err != nil {
return newError("failed to write response").Base(err).AtWarning()
}
2016-12-29 18:32:20 -05:00
return nil
2018-04-11 10:45:09 -04:00
}
2016-12-29 18:32:20 -05:00
2018-12-06 05:35:02 -05:00
if err := task.Run(ctx, requestDone, responseDone); err != nil {
2018-12-31 15:25:10 -05:00
common.Interrupt(link.Reader)
common.Interrupt(link.Writer)
2017-04-08 19:43:25 -04:00
return newError("connection ends").Base(err)
2016-12-29 18:32:20 -05:00
}
return result
2015-10-28 07:13:27 -04:00
}
2016-05-07 08:08:27 -04:00
func init() {
common.Must(common.RegisterConfig((*ServerConfig)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
return NewServer(ctx, config.(*ServerConfig))
}))
2016-05-07 08:08:27 -04:00
}