defend against the client crapping out

This commit is contained in:
Holly Stubbs 2024-05-08 01:02:18 +01:00
parent 4947a445ea
commit ba0d062512
Signed by: tgpholly
GPG key ID: B8583C4B7D18119E
3 changed files with 45 additions and 32 deletions

View file

@ -461,6 +461,7 @@ kbd {
let timeLastFrame = performance.now();
let frameDeltaTime = 0;
let timeSinceLastPingReset = performance.now();
try {
function animate() {
frameDeltaTime = (performance.now() - timeLastFrame) * 0.001;
@ -490,6 +491,9 @@ kbd {
timeLastFrame = performance.now();
}
animate();
} catch (e) {
console.error(e);
}
let windowContainer = null;
@ -622,7 +626,9 @@ kbd {
setTimeout(() => doConnect(localStorage["mpapikey"]), 5000);
}
ws.onclose = onCloseAndError;
ws.onerror = onCloseAndError;
ws.onerror = (e) => {
console.error(e);
}
}
function createOnlineDialog() {

View file

@ -411,10 +411,13 @@ websocketServer.on("connection", (socket) => {
socket.on("message", async (data) => {
const reader = createReader(Endian.LE, data as Buffer);
// There is absolutely no reason we should ever get
// more than 50 bytes legit.
if (reader.length > 0 && reader.length < 1024) {
switch (reader.readByte()) {
case MessageType.KeepAlive:
{
user.lastKeepAliveTime = Date.now();
break;
}
case MessageType.ClientDetails:
{
if (user !== undefined) {
@ -449,9 +452,9 @@ websocketServer.on("connection", (socket) => {
usersToSend.writeUInt(otherUser.id).writeShortString(otherUser.username).writeFloat(otherUser.cursorX).writeInt(otherUser.cursorY);
}
if (dbParty) {
user = users.set(myUUID, new RemoteUser(socket, dbUser.Username, page, rawURL, dbUser.Id, dbParty.Id, dbParty.Name));
user = users.set(myUUID, new RemoteUser(socket, myUUID, dbUser.Username, page, rawURL, dbUser.Id, dbParty.Id, dbParty.Name));
} else {
user = users.set(myUUID, new RemoteUser(socket, dbUser.Username, page, rawURL, dbUser.Id, Number.MIN_VALUE, ""));
user = users.set(myUUID, new RemoteUser(socket, myUUID, dbUser.Username, page, rawURL, dbUser.Id, Number.MIN_VALUE, ""));
}
sendToAllButSelf(user, createWriter(Endian.LE, 6 + dbUser.Username.length).writeByte(MessageType.ClientJoined).writeUInt(user.id).writeShortString(dbUser.Username).toBuffer());
user.send(usersToSend.toBuffer());

View file

@ -3,7 +3,8 @@ import { WebSocket } from "ws";
export default class RemoteUser {
private static USER_IDS = 0;
private readonly socket:WebSocket;
public readonly socket:WebSocket;
public readonly connectionUUID:string;
public readonly id:number;
public readonly username:string;
public readonly currentURL:string;
@ -15,9 +16,11 @@ export default class RemoteUser {
public userId:number;
public groupId:number = Number.MIN_VALUE;
public groupName:string;
public lastKeepAliveTime:number;
constructor(socket:WebSocket, username:string, currentURL:string, rawURL:string, userId:number, groupId:number, groupName:string) {
constructor(socket:WebSocket, connectionUUID:string, username:string, currentURL:string, rawURL:string, userId:number, groupId:number, groupName:string) {
this.socket = socket;
this.connectionUUID = connectionUUID;
this.id = RemoteUser.USER_IDS++;
this.username = username;
this.currentURL = currentURL;
@ -27,6 +30,7 @@ export default class RemoteUser {
this.userId = userId;
this.groupId = groupId;
this.groupName = groupName;
this.lastKeepAliveTime = Date.now();
}
send(data:Buffer) {