1
0
mirror of https://github.com/fumiama/yamaih.git synced 2026-06-05 00:10:28 +08:00

feat(proxy): remove localhost judgement

This commit is contained in:
源文雨
2025-03-31 01:54:29 +09:00
parent 2e71be8be1
commit 7e3b2e189b

View File

@@ -96,17 +96,13 @@ func (g *Gemini) handler(w http.ResponseWriter, r *http.Request) {
// extractIP parse real IP addr to r.RemoteAddr from proxy
func extractIP(r *http.Request) {
raddr := r.RemoteAddr
if strings.Contains(raddr, "127.0.0.1") ||
strings.Contains(raddr, "localhost") ||
strings.Contains(raddr, "@") {
realr := r.Header.Get("X-Forwarded-For")
realr := r.Header.Get("X-Forwarded-For")
if len(realr) > 0 && !strings.Contains(realr, "@") {
raddr = realr
} else {
realr = r.Header.Get("X-Real-IP")
if len(realr) > 0 && !strings.Contains(realr, "@") {
raddr = realr
} else {
realr = r.Header.Get("X-Real-IP")
if len(realr) > 0 && !strings.Contains(realr, "@") {
raddr = realr
}
}
}
r.RemoteAddr = raddr