import log from "./log"; import fs from "fs"; import net, {Socket} from "net"; import colors from "chalk"; import Helper from "./helper"; import Config from "./config"; type Connection = { socket: Socket; user: string; }; class Identification { private connectionId: number; private connections: Map; private oidentdFile?: string; constructor(startedCallback: (identHandler: Identification, err?: Error) => void) { this.connectionId = 0; this.connections = new Map(); if (typeof Config.values.oidentd === "string") { this.oidentdFile = Helper.expandHome(Config.values.oidentd); log.info(`Oidentd file: ${colors.green(this.oidentdFile)}`); this.refresh(); } if (Config.values.identd.enable) { if (this.oidentdFile) { log.warn( "Using both identd and oidentd at the same time, this is most likely not intended." ); } const server = net.createServer(this.serverConnection.bind(this)); server.on("error", (err) => { startedCallback(this, err); }); server.listen( { port: Config.values.identd.port || 113, host: Config.values.bind, }, () => { const address = server.address(); if (typeof address === "string") { log.info(`Identd server available on ${colors.green(address)}`); } else if (address?.address) { log.info( `Identd server available on ${colors.green( address.address + ":" + address.port.toString() )}` ); } startedCallback(this); } ); } else { startedCallback(this); } } serverConnection(socket: Socket) { socket.on("error", (err: string) => log.error(`Identd socket error: ${err}`)); socket.on("data", (data) => { this.respondToIdent(socket, data); socket.end(); }); } respondToIdent(socket: Socket, buffer: Buffer) { const data = buffer.toString().split(","); const lport = parseInt(data[0], 10) || 0; const fport = parseInt(data[1], 10) || 0; if (lport < 1 || fport < 1 || lport > 65535 || fport > 65535) { return; } for (const connection of this.connections.values()) { if (connection.socket.remotePort === fport && connection.socket.localPort === lport) { return socket.write( `${lport}, ${fport} : USERID : TheLounge : ${connection.user}\r\n` ); } } socket.write(`${lport}, ${fport} : ERROR : NO-USER\r\n`); } addSocket(socket: Socket, user: string) { const id = ++this.connectionId; this.connections.set(id, {socket, user}); if (this.oidentdFile) { this.refresh(); } return id; } removeSocket(id: number) { this.connections.delete(id); if (this.oidentdFile) { this.refresh(); } } refresh() { let file = "# Warning: file generated by Hard Lounge: changes will be overwritten!\n"; this.connections.forEach((connection, id) => { if (!connection.socket.remotePort || !connection.socket.localPort) { // Race condition: this can happen when more than one socket gets disconnected at // once, since we `refresh()` for each one being added/removed. This results // in there possibly being one or more disconnected sockets remaining when we get here. log.warn( `oidentd: socket has no remote or local port (id=${id}). See https://github.com/thelounge/thelounge/pull/4695.` ); return; } file += `fport ${connection.socket.remotePort}` + ` lport ${connection.socket.localPort}` + ` { reply "${connection.user}" }\n`; }); if (this.oidentdFile) { fs.writeFile(this.oidentdFile, file, {flag: "w+"}, function (err) { if (err) { log.error("Failed to update oidentd file!", err.message); } }); } } } export default Identification;