1c062697b9
Work on abstracting the CollabVMServer so it now calls into a interface for sending/recieving protocol messages. This will allow cleaner bringup of a fully binary protocol, and generally is just cleaner code. Mostly everything is parsd/running through this new layer, although there are some TODO items: - NetworkClient/... should just spit out a Buffer or something that eventually turns into or has one - TCP protocol will need to be revamped so we can support an actual binary protocol on top of it. The current thing is line based - More admin op stuff needs to be handled - The handlers are a bit jumbled around atm - There is still a good amount of code which assumes guacamole which needs to be rewritten dont use this branch fuckers
51 lines
1.6 KiB
TypeScript
51 lines
1.6 KiB
TypeScript
// TODO: replace tcp protocol with smth like
|
|
// struct msg { beu32 len; char data[len] }
|
|
// (along with a length cap obviously)
|
|
import EventEmitter from 'events';
|
|
import NetworkServer from '../NetworkServer.js';
|
|
import { Server, Socket } from 'net';
|
|
import IConfig from '../IConfig.js';
|
|
import TCPClient from './TCPClient.js';
|
|
import { IPDataManager } from '../IPData.js';
|
|
import { User } from '../User.js';
|
|
import pino from 'pino';
|
|
import { BanManager } from '../BanManager.js';
|
|
|
|
export default class TCPServer extends EventEmitter implements NetworkServer {
|
|
listener: Server;
|
|
Config: IConfig;
|
|
logger = pino({ name: 'CVMTS.TCPServer' });
|
|
clients: TCPClient[];
|
|
private banmgr: BanManager;
|
|
|
|
constructor(config: IConfig, banmgr: BanManager) {
|
|
super();
|
|
this.Config = config;
|
|
this.listener = new Server();
|
|
this.clients = [];
|
|
this.listener.on('connection', (socket) => this.onConnection(socket));
|
|
this.banmgr = banmgr;
|
|
}
|
|
|
|
private async onConnection(socket: Socket) {
|
|
this.logger.info(`New TCP connection from ${socket.remoteAddress}`);
|
|
if (await this.banmgr.isIPBanned(socket.remoteAddress!)) {
|
|
socket.write('6.banned;');
|
|
socket.destroy();
|
|
return;
|
|
}
|
|
var client = new TCPClient(socket);
|
|
this.clients.push(client);
|
|
this.emit('connect', new User(client, IPDataManager.GetIPData(client.getIP()), this.Config));
|
|
}
|
|
|
|
start(): void {
|
|
this.listener.listen(this.Config.tcp.port, this.Config.tcp.host, () => {
|
|
this.logger.info(`TCP server listening on ${this.Config.tcp.host}:${this.Config.tcp.port}`);
|
|
});
|
|
}
|
|
stop(): void {
|
|
this.listener.close();
|
|
}
|
|
}
|