github
/
nps
mirror of https://github.com/ehang-io/nps
1
0
Fork 0
Code Issues Releases Wiki Activity

Update https.go

pull/825/head
bboysoul 2021-06-29 11:14:36 +08:00 committed by GitHub
parent 01f6a2f633
commit 646dd6f1ad
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server/proxy/https.go
View File

@ -125,8 +125,7 @@ func (https *HttpsServer) handleHttps(c net.Conn) {
if targetAddr, err = host.Target.GetRandomTarget(); err != nil { if targetAddr, err = host.Target.GetRandomTarget(); err != nil {
logs.Warn(err.Error()) logs.Warn(err.Error())
} }
r.Header.Set("X-Forwarded-For", c.RemoteAddr().String())
r.Header.Set("X-Real-IP", c.RemoteAddr().String())
logs.Trace("new https connection,clientId %d,host %s,remote address %s", host.Client.Id, r.Host, c.RemoteAddr().String()) logs.Trace("new https connection,clientId %d,host %s,remote address %s", host.Client.Id, r.Host, c.RemoteAddr().String())
https.DealClient(conn.NewConn(c), host.Client, targetAddr, rb, common.CONN_TCP, nil, host.Flow, host.Target.LocalProxy) https.DealClient(conn.NewConn(c), host.Client, targetAddr, rb, common.CONN_TCP, nil, host.Flow, host.Target.LocalProxy)
} }
@ -184,5 +183,7 @@ func buildHttpsRequest(hostName string) *http.Request {
r.URL = new(url.URL) r.URL = new(url.URL)
r.URL.Scheme = "https" r.URL.Scheme = "https"
r.Host = hostName r.Host = hostName
r.Header.Set("X-Forwarded-For", c.RemoteAddr().String())
r.Header.Set("X-Real-IP", c.RemoteAddr().String())
return r return r
} }