Merge pull request #30 from daftaupe/#26

Add support for X-Forwarded-For
2018-07-25 00:18:29 +02:00 · 2018-07-25 00:18:29 +02:00 · e2c6cedc27
commit e2c6cedc27
parent a4d2cf17aa 8ed27bf54b
1 changed files with 9 additions and 1 deletions
--- a/server/websocket_handler.go
+++ b/server/websocket_handler.go
@ -17,10 +17,18 @@ type wsHandler struct {
 }

 func newWSHandler(conn *websocket.Conn, state *State, r *http.Request) *wsHandler {
+         var address string
+
+         if r.Header.Get("X-Forwarded-For") != "" {
+		address = r.Header.Get("X-Forwarded-For")
+         } else {
+		address = conn.RemoteAddr().String()
+         }
+
 	h := &wsHandler{
 		ws:    newWSConn(conn),
 		state: state,
-		addr:  conn.RemoteAddr().String(),
+		addr:  address,
 	}
 	h.init(r)
 	h.initHandlers()