zishuo/apinto
1
0
Fork 0
mirror of https://github.com/eolinker/apinto synced 2025-10-23 16:53:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

解决x-forwarded-for 为空的问题

Browse Source
This commit is contained in:
huangmengzhu
2023-05-25 17:10:03 +08:00
parent 515f942c1e
commit 8211bea252
4 changed files with 21 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
node/http-context/proxy.go
View File

@@ -1,7 +1,6 @@
package http_context
import (
"bytes"
"fmt"
"github.com/eolinker/eosc/log"
@@ -45,23 +44,16 @@ var (
)
func (r *ProxyRequest) reset(request *fasthttp.Request, remoteAddr string) {
//proxyRequest := fasthttp.AcquireRequest()
//request.CopyTo(proxyRequest)
r.req = request
r.RequestReader.reset(request, remoteAddr)
forwardedFor := r.req.Header.PeekBytes(xforwardedforKey)
if len(forwardedFor) > 0 {
if i := bytes.IndexByte(forwardedFor, ','); i > 0 {
r.realIP = string(forwardedFor[:i])
} else {
r.realIP = string(forwardedFor)
}
r.req.Header.Set("x-forwarded-for", fmt.Sprint(string(forwardedFor), ",", r.remoteAddr))
r.req.Header.Set("x-forwarded-for", fmt.Sprint(string(forwardedFor), ",", remoteAddr))
} else {
r.req.Header.Set("x-forwarded-for", r.remoteAddr)
r.realIP = r.remoteAddr
r.req.Header.Set("x-forwarded-for", remoteAddr)
}
r.RequestReader.reset(r.req, remoteAddr)
}
//func NewProxyRequest(request *fasthttp.Request, remoteAddr string) *ProxyRequest {