123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203 |
- package proxy
- import (
- "io"
- "net"
- "reflect"
- "strings"
- libio "github.com/fatedier/golib/io"
- v1 "github.com/fatedier/frp/pkg/config/v1"
- "github.com/fatedier/frp/pkg/util/limit"
- netpkg "github.com/fatedier/frp/pkg/util/net"
- "github.com/fatedier/frp/pkg/util/util"
- "github.com/fatedier/frp/pkg/util/vhost"
- "github.com/fatedier/frp/server/metrics"
- )
- func init() {
- RegisterProxyFactory(reflect.TypeOf(&v1.HTTPProxyConfig{}), NewHTTPProxy)
- }
- type HTTPProxy struct {
- *BaseProxy
- cfg *v1.HTTPProxyConfig
- closeFuncs []func()
- }
- func NewHTTPProxy(baseProxy *BaseProxy) Proxy {
- unwrapped, ok := baseProxy.GetConfigurer().(*v1.HTTPProxyConfig)
- if !ok {
- return nil
- }
- return &HTTPProxy{
- BaseProxy: baseProxy,
- cfg: unwrapped,
- }
- }
- func (pxy *HTTPProxy) Run() (remoteAddr string, err error) {
- xl := pxy.xl
- routeConfig := vhost.RouteConfig{
- RewriteHost: pxy.cfg.HostHeaderRewrite,
- RouteByHTTPUser: pxy.cfg.RouteByHTTPUser,
- Headers: pxy.cfg.RequestHeaders.Set,
- ResponseHeaders: pxy.cfg.ResponseHeaders.Set,
- Username: pxy.cfg.HTTPUser,
- Password: pxy.cfg.HTTPPassword,
- CreateConnFn: pxy.GetRealConn,
- }
- locations := pxy.cfg.Locations
- if len(locations) == 0 {
- locations = []string{""}
- }
- defer func() {
- if err != nil {
- pxy.Close()
- }
- }()
- addrs := make([]string, 0)
- for _, domain := range pxy.cfg.CustomDomains {
- if domain == "" {
- continue
- }
- routeConfig.Domain = domain
- for _, location := range locations {
- routeConfig.Location = location
- tmpRouteConfig := routeConfig
-
- if pxy.cfg.LoadBalancer.Group != "" {
- err = pxy.rc.HTTPGroupCtl.Register(pxy.name, pxy.cfg.LoadBalancer.Group, pxy.cfg.LoadBalancer.GroupKey, routeConfig)
- if err != nil {
- return
- }
- pxy.closeFuncs = append(pxy.closeFuncs, func() {
- pxy.rc.HTTPGroupCtl.UnRegister(pxy.name, pxy.cfg.LoadBalancer.Group, tmpRouteConfig)
- })
- } else {
-
- err = pxy.rc.HTTPReverseProxy.Register(routeConfig)
- if err != nil {
- return
- }
- pxy.closeFuncs = append(pxy.closeFuncs, func() {
- pxy.rc.HTTPReverseProxy.UnRegister(tmpRouteConfig)
- })
- }
- addrs = append(addrs, util.CanonicalAddr(routeConfig.Domain, pxy.serverCfg.VhostHTTPPort))
- xl.Infof("http proxy listen for host [%s] location [%s] group [%s], routeByHTTPUser [%s]",
- routeConfig.Domain, routeConfig.Location, pxy.cfg.LoadBalancer.Group, pxy.cfg.RouteByHTTPUser)
- }
- }
- if pxy.cfg.SubDomain != "" {
- routeConfig.Domain = pxy.cfg.SubDomain + "." + pxy.serverCfg.SubDomainHost
- for _, location := range locations {
- routeConfig.Location = location
- tmpRouteConfig := routeConfig
-
- if pxy.cfg.LoadBalancer.Group != "" {
- err = pxy.rc.HTTPGroupCtl.Register(pxy.name, pxy.cfg.LoadBalancer.Group, pxy.cfg.LoadBalancer.GroupKey, routeConfig)
- if err != nil {
- return
- }
- pxy.closeFuncs = append(pxy.closeFuncs, func() {
- pxy.rc.HTTPGroupCtl.UnRegister(pxy.name, pxy.cfg.LoadBalancer.Group, tmpRouteConfig)
- })
- } else {
- err = pxy.rc.HTTPReverseProxy.Register(routeConfig)
- if err != nil {
- return
- }
- pxy.closeFuncs = append(pxy.closeFuncs, func() {
- pxy.rc.HTTPReverseProxy.UnRegister(tmpRouteConfig)
- })
- }
- addrs = append(addrs, util.CanonicalAddr(tmpRouteConfig.Domain, pxy.serverCfg.VhostHTTPPort))
- xl.Infof("http proxy listen for host [%s] location [%s] group [%s], routeByHTTPUser [%s]",
- routeConfig.Domain, routeConfig.Location, pxy.cfg.LoadBalancer.Group, pxy.cfg.RouteByHTTPUser)
- }
- }
- remoteAddr = strings.Join(addrs, ",")
- return
- }
- func (pxy *HTTPProxy) GetRealConn(remoteAddr string) (workConn net.Conn, err error) {
- xl := pxy.xl
- rAddr, errRet := net.ResolveTCPAddr("tcp", remoteAddr)
- if errRet != nil {
- xl.Warnf("resolve TCP addr [%s] error: %v", remoteAddr, errRet)
-
- }
- tmpConn, errRet := pxy.GetWorkConnFromPool(rAddr, nil)
- if errRet != nil {
- err = errRet
- return
- }
- var rwc io.ReadWriteCloser = tmpConn
- if pxy.cfg.Transport.UseEncryption {
- rwc, err = libio.WithEncryption(rwc, []byte(pxy.serverCfg.Auth.Token))
- if err != nil {
- xl.Errorf("create encryption stream error: %v", err)
- return
- }
- }
- if pxy.cfg.Transport.UseCompression {
- rwc = libio.WithCompression(rwc)
- }
- if pxy.GetLimiter() != nil {
- rwc = libio.WrapReadWriteCloser(limit.NewReader(rwc, pxy.GetLimiter()), limit.NewWriter(rwc, pxy.GetLimiter()), func() error {
- return rwc.Close()
- })
- }
- workConn = netpkg.WrapReadWriteCloserToConn(rwc, tmpConn)
- workConn = netpkg.WrapStatsConn(workConn, pxy.updateStatsAfterClosedConn)
- metrics.Server.OpenConnection(pxy.GetName(), pxy.GetConfigurer().GetBaseConfig().Type)
- return
- }
- func (pxy *HTTPProxy) updateStatsAfterClosedConn(totalRead, totalWrite int64) {
- name := pxy.GetName()
- proxyType := pxy.GetConfigurer().GetBaseConfig().Type
- metrics.Server.CloseConnection(name, proxyType)
- metrics.Server.AddTrafficIn(name, proxyType, totalWrite)
- metrics.Server.AddTrafficOut(name, proxyType, totalRead)
- }
- func (pxy *HTTPProxy) Close() {
- pxy.BaseProxy.Close()
- for _, closeFn := range pxy.closeFuncs {
- closeFn()
- }
- }
|