TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-07 18:50:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

可以配置是否在反向代理中添加X-Real-IP和X-Forwarded-*

Browse Source
This commit is contained in:
GoEdgeLab
2021-01-26 20:28:12 +08:00
parent a1d8c1bbf4
commit e85ce96ca5
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/nodes/http_request.go
View File

@@ -864,8 +864,8 @@ func (this *HTTPRequest) requestServerPort() int {
// 设置代理相关头部信息 // 设置代理相关头部信息
// 参考：https://tools.ietf.org/html/rfc7239 // 参考：https://tools.ietf.org/html/rfc7239
// TODO X-Forwarded-* 系列做成可选项，避免有些源站屏蔽此项
func (this *HTTPRequest) setForwardHeaders(header http.Header) { func (this *HTTPRequest) setForwardHeaders(header http.Header) {
// TODO 做成可选项
if this.RawReq.Header.Get("Connection") == "close" { if this.RawReq.Header.Get("Connection") == "close" {
this.RawReq.Header.Set("Connection", "keep-alive") this.RawReq.Header.Set("Connection", "keep-alive")
} }
@@ -877,7 +877,7 @@ func (this *HTTPRequest) setForwardHeaders(header http.Header) {
} }
// x-real-ip // x-real-ip
{ if this.reverseProxy != nil && this.reverseProxy.ShouldAddXRealIPHeader() {
_, ok1 := header["X-Real-IP"] _, ok1 := header["X-Real-IP"]
_, ok2 := header["X-Real-Ip"] _, ok2 := header["X-Real-Ip"]
if !ok1 && !ok2 { if !ok1 && !ok2 {
@@ -886,7 +886,7 @@ func (this *HTTPRequest) setForwardHeaders(header http.Header) {
} }
// X-Forwarded-For // X-Forwarded-For
{ if this.reverseProxy != nil && this.reverseProxy.ShouldAddXForwardedForHeader() {
forwardedFor, ok := header["X-Forwarded-For"] forwardedFor, ok := header["X-Forwarded-For"]
if ok { if ok {
_, hasForwardHeader := this.RawReq.Header["X-Forwarded-For"] _, hasForwardHeader := this.RawReq.Header["X-Forwarded-For"]
@@ -909,15 +909,21 @@ func (this *HTTPRequest) setForwardHeaders(header http.Header) {
}**/ }**/
// others // others
if this.reverseProxy != nil && this.reverseProxy.ShouldAddXForwardedByHeader() {
this.RawReq.Header.Set("X-Forwarded-By", this.ServerAddr) this.RawReq.Header.Set("X-Forwarded-By", this.ServerAddr)
}
if this.reverseProxy != nil && this.reverseProxy.ShouldAddXForwardedHostHeader() {
if _, ok := header["X-Forwarded-Host"]; !ok { if _, ok := header["X-Forwarded-Host"]; !ok {
this.RawReq.Header.Set("X-Forwarded-Host", this.Host) this.RawReq.Header.Set("X-Forwarded-Host", this.Host)
} }
}
if this.reverseProxy != nil && this.reverseProxy.ShouldAddXForwardedProtoHeader() {
if _, ok := header["X-Forwarded-Proto"]; !ok { if _, ok := header["X-Forwarded-Proto"]; !ok {
this.RawReq.Header.Set("X-Forwarded-Proto", this.requestScheme()) this.RawReq.Header.Set("X-Forwarded-Proto", this.requestScheme())
} }
}
} }
// 处理自定义Request Header // 处理自定义Request Header