frp/utils/vhost/http.go

207 lines
5.7 KiB
Go
Raw Normal View History

2017-12-13 03:27:43 +08:00
// Copyright 2017 fatedier, fatedier@gmail.com
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package vhost
import (
"bytes"
2017-12-13 03:27:43 +08:00
"context"
"errors"
"fmt"
2017-12-13 03:27:43 +08:00
"log"
"net"
"net/http"
"strings"
"time"
frpLog "github.com/fatedier/frp/utils/log"
"github.com/fatedier/frp/utils/util"
2018-05-08 02:13:30 +08:00
"github.com/fatedier/golib/pool"
2017-12-13 03:27:43 +08:00
)
var (
2019-07-31 00:41:58 +08:00
ErrNoDomain = errors.New("no such domain")
2017-12-13 03:27:43 +08:00
)
2018-08-08 11:18:38 +08:00
type HttpReverseProxyOptions struct {
ResponseHeaderTimeoutS int64
}
2017-12-13 03:27:43 +08:00
type HttpReverseProxy struct {
2019-07-31 00:41:58 +08:00
proxy *ReverseProxy
2017-12-13 03:27:43 +08:00
vhostRouter *VhostRouters
2018-08-08 11:18:38 +08:00
responseHeaderTimeout time.Duration
2017-12-13 03:27:43 +08:00
}
2019-07-31 00:41:58 +08:00
func NewHttpReverseProxy(option HttpReverseProxyOptions, vhostRouter *VhostRouters) *HttpReverseProxy {
2018-08-08 11:18:38 +08:00
if option.ResponseHeaderTimeoutS <= 0 {
option.ResponseHeaderTimeoutS = 60
}
2017-12-13 03:27:43 +08:00
rp := &HttpReverseProxy{
2018-08-08 11:18:38 +08:00
responseHeaderTimeout: time.Duration(option.ResponseHeaderTimeoutS) * time.Second,
2019-07-31 00:41:58 +08:00
vhostRouter: vhostRouter,
2017-12-13 03:27:43 +08:00
}
proxy := &ReverseProxy{
Director: func(req *http.Request) {
req.URL.Scheme = "http"
url := req.Context().Value("url").(string)
oldHost := util.GetHostFromAddr(req.Context().Value("host").(string))
2018-05-20 23:22:07 +08:00
host := rp.GetRealHost(oldHost, url)
2017-12-13 03:27:43 +08:00
if host != "" {
req.Host = host
}
req.URL.Host = req.Host
2018-05-20 23:22:07 +08:00
headers := rp.GetHeaders(oldHost, url)
for k, v := range headers {
req.Header.Set(k, v)
}
2017-12-13 03:27:43 +08:00
},
Transport: &http.Transport{
2018-08-08 11:18:38 +08:00
ResponseHeaderTimeout: rp.responseHeaderTimeout,
2017-12-13 03:27:43 +08:00
DisableKeepAlives: true,
DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
url := ctx.Value("url").(string)
host := util.GetHostFromAddr(ctx.Value("host").(string))
2019-04-10 10:51:01 +08:00
remote := ctx.Value("remote").(string)
return rp.CreateConnection(host, url, remote)
2017-12-13 03:27:43 +08:00
},
},
BufferPool: newWrapPool(),
ErrorLog: log.New(newWrapLogger(), "", 0),
2019-08-06 16:49:22 +08:00
ErrorHandler: func(rw http.ResponseWriter, req *http.Request, err error) {
frpLog.Warn("do http proxy request error: %v", err)
2019-08-06 16:49:22 +08:00
rw.WriteHeader(http.StatusNotFound)
rw.Write(getNotFoundPageContent())
},
2017-12-13 03:27:43 +08:00
}
rp.proxy = proxy
return rp
}
2019-07-31 00:41:58 +08:00
// Register register the route config to reverse proxy
// reverse proxy will use CreateConnFn from routeCfg to create a connection to the remote service
2017-12-13 23:44:27 +08:00
func (rp *HttpReverseProxy) Register(routeCfg VhostRouteConfig) error {
2019-07-31 00:41:58 +08:00
err := rp.vhostRouter.Add(routeCfg.Domain, routeCfg.Location, &routeCfg)
if err != nil {
return err
2017-12-13 03:27:43 +08:00
}
return nil
}
2019-07-31 00:41:58 +08:00
// UnRegister unregister route config by domain and location
2017-12-13 03:27:43 +08:00
func (rp *HttpReverseProxy) UnRegister(domain string, location string) {
rp.vhostRouter.Del(domain, location)
}
func (rp *HttpReverseProxy) GetRealHost(domain string, location string) (host string) {
vr, ok := rp.getVhost(domain, location)
if ok {
2017-12-13 23:44:27 +08:00
host = vr.payload.(*VhostRouteConfig).RewriteHost
2017-12-13 03:27:43 +08:00
}
return
}
2018-05-20 23:22:07 +08:00
func (rp *HttpReverseProxy) GetHeaders(domain string, location string) (headers map[string]string) {
vr, ok := rp.getVhost(domain, location)
if ok {
headers = vr.payload.(*VhostRouteConfig).Headers
}
return
}
2019-07-31 00:41:58 +08:00
// CreateConnection create a new connection by route config
2019-04-10 10:51:01 +08:00
func (rp *HttpReverseProxy) CreateConnection(domain string, location string, remoteAddr string) (net.Conn, error) {
2017-12-13 03:27:43 +08:00
vr, ok := rp.getVhost(domain, location)
if ok {
2017-12-13 23:44:27 +08:00
fn := vr.payload.(*VhostRouteConfig).CreateConnFn
2017-12-13 03:27:43 +08:00
if fn != nil {
2019-04-10 10:51:01 +08:00
return fn(remoteAddr)
2017-12-13 03:27:43 +08:00
}
}
return nil, fmt.Errorf("%v: %s %s", ErrNoDomain, domain, location)
2017-12-13 03:27:43 +08:00
}
2017-12-13 23:44:27 +08:00
func (rp *HttpReverseProxy) CheckAuth(domain, location, user, passwd string) bool {
vr, ok := rp.getVhost(domain, location)
if ok {
checkUser := vr.payload.(*VhostRouteConfig).Username
checkPasswd := vr.payload.(*VhostRouteConfig).Password
if (checkUser != "" || checkPasswd != "") && (checkUser != user || checkPasswd != passwd) {
return false
}
}
return true
}
2019-07-31 00:41:58 +08:00
// getVhost get vhost router by domain and location
2017-12-13 03:27:43 +08:00
func (rp *HttpReverseProxy) getVhost(domain string, location string) (vr *VhostRouter, ok bool) {
// first we check the full hostname
// if not exist, then check the wildcard_domain such as *.example.com
vr, ok = rp.vhostRouter.Get(domain, location)
if ok {
return
}
domainSplit := strings.Split(domain, ".")
if len(domainSplit) < 3 {
return nil, false
}
for {
if len(domainSplit) < 3 {
return nil, false
}
domainSplit[0] = "*"
domain = strings.Join(domainSplit, ".")
vr, ok = rp.vhostRouter.Get(domain, location)
if ok {
return vr, true
}
domainSplit = domainSplit[1:]
2017-12-13 03:27:43 +08:00
}
}
func (rp *HttpReverseProxy) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
domain := util.GetHostFromAddr(req.Host)
2017-12-13 23:44:27 +08:00
location := req.URL.Path
user, passwd, _ := req.BasicAuth()
if !rp.CheckAuth(domain, location, user, passwd) {
rw.Header().Set("WWW-Authenticate", `Basic realm="Restricted"`)
http.Error(rw, http.StatusText(http.StatusUnauthorized), http.StatusUnauthorized)
return
}
2017-12-13 03:27:43 +08:00
rp.proxy.ServeHTTP(rw, req)
}
type wrapPool struct{}
func newWrapPool() *wrapPool { return &wrapPool{} }
func (p *wrapPool) Get() []byte { return pool.GetBuf(32 * 1024) }
func (p *wrapPool) Put(buf []byte) { pool.PutBuf(buf) }
type wrapLogger struct{}
func newWrapLogger() *wrapLogger { return &wrapLogger{} }
func (l *wrapLogger) Write(p []byte) (n int, err error) {
frpLog.Warn("%s", string(bytes.TrimRight(p, "\n")))
2017-12-13 03:27:43 +08:00
return len(p), nil
}