websockets: log if the connection is proxied

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
This commit is contained in:
Eric Callahan 2022-01-21 06:57:21 -05:00
parent e103accdfb
commit f5ceefbb8d
1 changed files with 7 additions and 0 deletions

View File

@ -394,7 +394,14 @@ class WebSocket(WebSocketHandler, Subscribable):
def open(self, *args, **kwargs) -> None: def open(self, *args, **kwargs) -> None:
self.set_nodelay(True) self.set_nodelay(True)
agent = self.request.headers.get("User-Agent", "") agent = self.request.headers.get("User-Agent", "")
is_proxy = False
if (
"X-Forwarded-For" in self.request.headers or
"X-Real-Ip" in self.request.headers
):
is_proxy = True
logging.info(f"Websocket Opened: ID: {self.uid}, " logging.info(f"Websocket Opened: ID: {self.uid}, "
f"Proxied: {is_proxy}, "
f"User Agent: {agent}") f"User Agent: {agent}")
self.wsm.add_websocket(self) self.wsm.add_websocket(self)