package router import ( "context" "encoding/json" "github.com/gin-gonic/gin" ws "github.com/gorilla/websocket" "github.com/pterodactyl/wings/router/websocket" ) // Upgrades a connection to a websocket and passes events along between. func getServerWebsocket(c *gin.Context) { s := GetServer(c.Param("server")) handler, err := websocket.GetHandler(s, c.Writer, c.Request) if err != nil { TrackedServerError(err, s).AbortWithServerError(c) return } defer handler.Connection.Close() // Create a context that can be canceled when the user disconnects from this // socket that will also cancel listeners running in separate threads. ctx, cancel := context.WithCancel(context.Background()) defer cancel() go handler.ListenForServerEvents(ctx) go handler.ListenForExpiration(ctx) for { j := websocket.Message{} _, p, err := handler.Connection.ReadMessage() if err != nil { if !ws.IsCloseError( err, ws.CloseNormalClosure, ws.CloseGoingAway, ws.CloseNoStatusReceived, ws.CloseServiceRestart, ws.CloseAbnormalClosure, ) { s.Log().WithField("error", err).Warn("error handling websocket message for server") } break } // Discard and JSON parse errors into the void and don't continue processing this // specific socket request. If we did a break here the client would get disconnected // from the socket, which is NOT what we want to do. if err := json.Unmarshal(p, &j); err != nil { continue } if err := handler.HandleInbound(j); err != nil { handler.SendErrorJson(j, err) } } }