From db2d1fce76b1c56a57238fe032be245206a78ffa Mon Sep 17 00:00:00 2001 From: fatedier Date: Sun, 20 May 2018 23:22:07 +0800 Subject: [PATCH] http: support setting headers --- conf/frpc_full.ini | 10 ++++++---- models/config/proxy.go | 31 ++++++++++++++++++++++++++----- models/msg/msg.go | 13 +++++++------ server/proxy.go | 1 + utils/vhost/newhttp.go | 17 +++++++++++++++-- utils/vhost/vhost.go | 1 + 6 files changed, 56 insertions(+), 17 deletions(-) diff --git a/conf/frpc_full.ini b/conf/frpc_full.ini index ca933f0..e5e5c46 100644 --- a/conf/frpc_full.ini +++ b/conf/frpc_full.ini @@ -56,10 +56,10 @@ dns_server = 8.8.8.8 # heartbeat_interval = 30 # heartbeat_timeout = 90 -# ssh is the proxy name same as server's configuration -# if user in [common] section is not empty, it will be changed to {user}.{proxy} such as your_name.ssh +# 'ssh' is the unique proxy name +# if user in [common] section is not empty, it will be changed to {user}.{proxy} such as 'your_name.ssh' [ssh] -# tcp | udp | http | https, default is tcp +# tcp | udp | http | https | stcp | xtcp, default is tcp type = tcp local_ip = 127.0.0.1 local_port = 22 @@ -120,6 +120,8 @@ custom_domains = web02.yourdomain.com # locations is only available for http type locations = /,/pic host_header_rewrite = example.com +# params with prefix "header_" will be used to update http request headers +header_X-From-Where = frp [web02] type = https @@ -136,7 +138,7 @@ remote_port = 6003 # if plugin is defined, local_ip and local_port is useless # plugin will handle connections got from frps plugin = unix_domain_socket -# params set with prefix "plugin_" that plugin needed +# params with prefix "plugin_" that plugin needed plugin_unix_path = /var/run/docker.sock [plugin_http_proxy] diff --git a/models/config/proxy.go b/models/config/proxy.go index b1d36a8..53a2a45 100644 --- a/models/config/proxy.go +++ b/models/config/proxy.go @@ -429,10 +429,11 @@ type HttpProxyConf struct { LocalSvrConf - Locations []string `json:"locations"` - HostHeaderRewrite string `json:"host_header_rewrite"` - HttpUser string `json:"http_user"` - HttpPwd string `json:"http_pwd"` + Locations []string `json:"locations"` + HttpUser string `json:"http_user"` + HttpPwd string `json:"http_pwd"` + HostHeaderRewrite string `json:"host_header_rewrite"` + Headers map[string]string `json:"headers"` } func (cfg *HttpProxyConf) Compare(cmp ProxyConf) bool { @@ -447,9 +448,20 @@ func (cfg *HttpProxyConf) Compare(cmp ProxyConf) bool { strings.Join(cfg.Locations, " ") != strings.Join(cmpConf.Locations, " ") || cfg.HostHeaderRewrite != cmpConf.HostHeaderRewrite || cfg.HttpUser != cmpConf.HttpUser || - cfg.HttpPwd != cmpConf.HttpPwd { + cfg.HttpPwd != cmpConf.HttpPwd || + len(cfg.Headers) != len(cmpConf.Headers) { return false } + + for k, v := range cfg.Headers { + if v2, ok := cmpConf.Headers[k]; !ok { + return false + } else { + if v != v2 { + return false + } + } + } return true } @@ -461,6 +473,7 @@ func (cfg *HttpProxyConf) UnmarshalFromMsg(pMsg *msg.NewProxy) { cfg.HostHeaderRewrite = pMsg.HostHeaderRewrite cfg.HttpUser = pMsg.HttpUser cfg.HttpPwd = pMsg.HttpPwd + cfg.Headers = pMsg.Headers } func (cfg *HttpProxyConf) UnmarshalFromIni(prefix string, name string, section ini.Section) (err error) { @@ -487,6 +500,13 @@ func (cfg *HttpProxyConf) UnmarshalFromIni(prefix string, name string, section i cfg.HostHeaderRewrite = section["host_header_rewrite"] cfg.HttpUser = section["http_user"] cfg.HttpPwd = section["http_pwd"] + cfg.Headers = make(map[string]string) + + for k, v := range section { + if strings.HasPrefix(k, "header_") { + cfg.Headers[strings.TrimPrefix(k, "header_")] = v + } + } return } @@ -498,6 +518,7 @@ func (cfg *HttpProxyConf) MarshalToMsg(pMsg *msg.NewProxy) { pMsg.HostHeaderRewrite = cfg.HostHeaderRewrite pMsg.HttpUser = cfg.HttpUser pMsg.HttpPwd = cfg.HttpPwd + pMsg.Headers = cfg.Headers } func (cfg *HttpProxyConf) CheckForCli() (err error) { diff --git a/models/msg/msg.go b/models/msg/msg.go index fd3e656..9669c6b 100644 --- a/models/msg/msg.go +++ b/models/msg/msg.go @@ -91,12 +91,13 @@ type NewProxy struct { RemotePort int `json:"remote_port"` // http and https only - CustomDomains []string `json:"custom_domains"` - SubDomain string `json:"subdomain"` - Locations []string `json:"locations"` - HostHeaderRewrite string `json:"host_header_rewrite"` - HttpUser string `json:"http_user"` - HttpPwd string `json:"http_pwd"` + CustomDomains []string `json:"custom_domains"` + SubDomain string `json:"subdomain"` + Locations []string `json:"locations"` + HttpUser string `json:"http_user"` + HttpPwd string `json:"http_pwd"` + HostHeaderRewrite string `json:"host_header_rewrite"` + Headers map[string]string `json:"headers"` // stcp Sk string `json:"sk"` diff --git a/server/proxy.go b/server/proxy.go index 5947700..4bf6f1f 100644 --- a/server/proxy.go +++ b/server/proxy.go @@ -225,6 +225,7 @@ type HttpProxy struct { func (pxy *HttpProxy) Run() (remoteAddr string, err error) { routeConfig := vhost.VhostRouteConfig{ RewriteHost: pxy.cfg.HostHeaderRewrite, + Headers: pxy.cfg.Headers, Username: pxy.cfg.HttpUser, Password: pxy.cfg.HttpPwd, CreateConnFn: pxy.GetRealConn, diff --git a/utils/vhost/newhttp.go b/utils/vhost/newhttp.go index 819ce1d..1351b22 100644 --- a/utils/vhost/newhttp.go +++ b/utils/vhost/newhttp.go @@ -63,12 +63,17 @@ func NewHttpReverseProxy() *HttpReverseProxy { Director: func(req *http.Request) { req.URL.Scheme = "http" url := req.Context().Value("url").(string) - host := getHostFromAddr(req.Context().Value("host").(string)) - host = rp.GetRealHost(host, url) + oldHost := getHostFromAddr(req.Context().Value("host").(string)) + host := rp.GetRealHost(oldHost, url) if host != "" { req.Host = host } req.URL.Host = req.Host + + headers := rp.GetHeaders(oldHost, url) + for k, v := range headers { + req.Header.Set(k, v) + } }, Transport: &http.Transport{ ResponseHeaderTimeout: responseHeaderTimeout, @@ -117,6 +122,14 @@ func (rp *HttpReverseProxy) GetRealHost(domain string, location string) (host st return } +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 +} + func (rp *HttpReverseProxy) CreateConnection(domain string, location string) (net.Conn, error) { vr, ok := rp.getVhost(domain, location) if ok { diff --git a/utils/vhost/vhost.go b/utils/vhost/vhost.go index fe0d1f2..84e57e9 100644 --- a/utils/vhost/vhost.go +++ b/utils/vhost/vhost.go @@ -59,6 +59,7 @@ type VhostRouteConfig struct { RewriteHost string Username string Password string + Headers map[string]string CreateConnFn CreateConnFunc }