omg

2026-03-17 19:51:47 +01:00
parent 84c9fa2b60
commit e496cb0017
5 changed files with 42 additions and 19 deletions
@@ -68,7 +68,7 @@ func handleUnauthenticatedMessage(conn *websocket.Conn, client *Client, message
 		return false
 	}

-	err := SetClientFromToken(client, token)
+	clientId, err := GetClientIdFromToken(token)
 	if err != nil {
 		var errmsg = map[string]any{
 			"type":    WSServerResponse(InvalidCredentials),
@@ -77,6 +77,17 @@ func handleUnauthenticatedMessage(conn *websocket.Conn, client *Client, message
 		sendMessageCloseIfTimeout(conn, &errmsg)
 		return false
 	}
+
+	client, err = GetClientFromId(clientId)
+	if err != nil {
+		var errmsg = map[string]any{
+			"type":    WSServerResponse(InvalidCredentials),
+			"message": "bad token",
+		}
+		sendMessageCloseIfTimeout(conn, &errmsg)
+		return false
+	}
+
 	client.IsAuthenticated = true
 	return true
 }