Robot
/
croc
mirror of https://github.com/schollz/croc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

compute addresses

This commit is contained in:
Zack Scholl 2018-07-01 11:40:00 -07:00
parent ca9fafed55
commit 6164bc37d5
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/server.go
View File

@ -20,9 +20,16 @@ func (c *Croc) startServer() (err error) {
var upgrader = websocket.Upgrader{} // use default options
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
// incoming websocket request
log.Debugf("connecting remote addr: %+v", r.Header)
log.Debugf("connecting remote addr: %+v", r)
ws, err := upgrader.Upgrade(w, r, nil)
log.Debugf("connecting remote addr: %s", ws.RemoteAddr().String())
address := ws.RemoteAddr().String()
if _, ok := r.Header["X-Forwarded-For"]; ok {
address = r.Header["X-Forwarded-For"][0]
}
if _, ok := r.Header["X-Real-Ip"]; ok {
address = r.Header["X-Real-Ip"][0]
}
log.Debugf("connecting remote addr: %s", address)
if err != nil {
log.Error("upgrade:", err)
return