3pp-mirror/neko
1
0
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2025-05-28 16:37:15 +02:00
Code Issues Projects Releases Packages Wiki Activity

reverse proxy mode

Browse source
This commit is contained in:
Craig 2020-04-06 20:14:30 +00:00
parent 414b5a8015
commit 584513de9b
2 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
server/internal/websocket/websocket.go
View file

@ -70,7 +70,7 @@ func (ws *WebSocketHandler) Start() error {
}
})
ws.sessions.OnDestroy(func(id string) {
ws.sessions.OnDestroy(func(id string, session types.Session) {
if err := ws.handler.SessionDestroyed(id); err != nil {
ws.logger.Warn().Str("id", id).Err(err).Msg("session destroyed with and error")
} else {
@ -191,7 +191,11 @@ func (ws *WebSocketHandler) Upgrade(w http.ResponseWriter, r *http.Request) erro
}
func (ws *WebSocketHandler) authenticate(r *http.Request) (string, string, bool, error) {
ip := utils.ReadUserIP(r)
ip := r.RemoteAddr
if ws.conf.Proxy {
ip = utils.ReadUserIP(r)
}
id, err := utils.NewUID(32)
if err != nil {