Changeset 344 in code

Timestamp:

Jun 29, 2020, 4:27:43 PM (5 years ago)

Author:

contact

Message:

Extract X-Forwarded-* headers for WebSocket connections

File:

• trunk/server.go (modified) (1 diff)

trunk/server.go

@@ -152 +152 @@
                 return
         }
-        s.handle(newWebsocketIRCConn(conn), req.RemoteAddr)
+        remoteAddr := req.RemoteAddr
+        forwardedHost := req.Header.Get("X-Forwarded-For")
+        forwardedPort := req.Header.Get("X-Forwarded-Port")
+        if forwardedHost != "" && forwardedPort != "" {
+                remoteAddr = net.JoinHostPort(forwardedHost, forwardedPort)
+        }
+        s.handle(newWebsocketIRCConn(conn), remoteAddr)
 }