fumiama/yamaih
1
0
Fork 0
mirror of https://github.com/fumiama/yamaih.git synced 2026-06-22 11:30:26 +08:00
Code Activity

feat(proxy): remove localhost judgement

Browse Source
This commit is contained in:
源文雨
2025-03-31 01:54:29 +09:00
parent 2e71be8be1
commit 7e3b2e189b
1 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
procuratio.go
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 // extractIP parse real IP addr to r.RemoteAddr from proxy
func extractIP(r *http.Request) { func extractIP(r *http.Request) {
raddr := r.RemoteAddr raddr := r.RemoteAddr
if strings.Contains(raddr, "127.0.0.1") || realr := r.Header.Get("X-Forwarded-For")
strings.Contains(raddr, "localhost") || if len(realr) > 0 && !strings.Contains(realr, "@") {
strings.Contains(raddr, "@") { raddr = realr
realr := r.Header.Get("X-Forwarded-For") } else {
realr = r.Header.Get("X-Real-IP")
if len(realr) > 0 && !strings.Contains(realr, "@") { if len(realr) > 0 && !strings.Contains(realr, "@") {
raddr = realr raddr = realr
} else {
realr = r.Header.Get("X-Real-IP")
if len(realr) > 0 && !strings.Contains(realr, "@") {
raddr = realr
}
} }
} }
r.RemoteAddr = raddr r.RemoteAddr = raddr