cthrace/collabvm-1.2.ts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cvmts: reimplement connection limit using ipdata

Browse Source
This commit is contained in:
modeco80
2024-06-19 17:56:55 -04:00
parent ba8743f461
commit b485e7f689
2 changed files with 22 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
cvmts/src/WebSocket/WSServer.ts
View File

@@ -119,15 +119,15 @@ export default class WSServer extends EventEmitter implements NetworkServer {
ip = req.socket.remoteAddress;
}
// TODO: Implement
let ipdata = IPDataManager.GetIPDataMaybe(ip);
// Get the amount of active connections coming from the requesting IP.
//let connections = this.clients.filter((client) => client.IP.address == ip);
// If it exceeds the limit set in the config, reject the connection with a 429.
//if (connections.length + 1 > this.Config.http.maxConnections) {
// socket.write('HTTP/1.1 429 Too Many Requests\n\n429 Too Many Requests');
// socket.destroy();
//}
if(ipdata != null) {
let connections = ipdata.refCount;
if (connections + 1 > this.Config.collabvm.maxConnections) {
socket.write('HTTP/1.1 429 Too Many Requests\n\n429 Too Many Requests');
socket.destroy();
}
}
this.wsServer.handleUpgrade(req, socket, head, (ws: WebSocket) => {
this.wsServer.emit('connection', ws, req);
@@ -149,4 +149,4 @@ export default class WSServer extends EventEmitter implements NetworkServer {
this.logger.Info(`New WebSocket connection from ${user.IP.address}`);
}
}
}