netplaySniff – Diff between revs 1 and 2

Subversion Repositories:

?pathlinks?

Rev 1 – Rev 3 – Go to most recent revision – Details – Blame – Last modification – View Log

 #!/usr/bin/env node
 ///////////////////////////////////////////////////////////////////////////
-//    Copyright (C) 2023 Wizardry and Steamworks - License: MIT          //
+//    Copyright (C) 2024 Wizardry and Steamworks - License: MIT          //
 ///////////////////////////////////////////////////////////////////////////
 const fs = require('fs')
 const path = require('path')
 const PROTOCOL = decoders.PROTOCOL
 const shortHash = require('short-hash')
 const { exec } = require("child_process")
 const Inotify = require('inotify-remastered').Inotify
 const inotify = new Inotify()
+const sqlite = require('sqlite')
 // load configuration file.
 let config = YAML.load('config.yml')
 // set up logger.
 const logger = createLogger({
+    format: format.combine(
+        format.timestamp({
-    format: format.combine(
+            format: 'YYYYMMDDHHmmss'
+        }),
+        format.printf(info =>
-        format.splat(),
+            `${info.timestamp} ${info.level}: ${info.message}`+(info.splat !== undefined?`${info.splat}`:" ")
         format.simple()
     ),
     transports: [
         new transports.Console({
 })
 const mqttClient = mqtt.connect(config.mqtt.connect)
 mqttClient.on('reconnect', () => {
-    logger.info('Reconnecting to Corrade MQTT server...')
+    logger.info('Reconnecting to MQTT server...')
 })
 mqttClient.on('connect', () => {
-    logger.info('Connected to Corrade MQTT server.')
+    logger.info('Connected to MQTT server.')
     // Subscribe to group message notifications with group name and password.
     mqttClient.subscribe(`${config.mqtt.topic}`, (error) => {
         if (error) {
-            logger.info('Error subscribing to Corrade MQTT group messages.')
+            logger.info('Error subscribing to MQTT server.')
             return
+        }
-        logger.info('Subscribed to Corrade MQTT group messages.')
+        logger.info('Subscribed to MQTT server.')
     })
 })
 mqttClient.on('close', () => {
-    logger.error('Disconnected from Corrade MQTT server...')
+    logger.error('Disconnected from MQTT server.')
 })
     // remove NULL and NETPLAY_CMD_NICK
     netplay.nick = payload.toString().replace(/[\u0000\u0020]+/gi, '')
     netplay.hash = shortHash(`${netplay.nick}${netplay.ip}`)
     netplay.time = new Date().toISOString()
+    logger.info(`Player ${netplay.nick} joined via IP ${netplay.ip}`);
+    const db = new sqlite.Database({ filename: config.db.file }, sqlite.OPEN_CREATE | sqlite.OPEN_READWRITE | sqlite.OPEN_FULLMUTEX, (error) => {
+        if(error) {
+            logger.error(`failed to open database ${config.db.file}`)
+            return
+        }
+        db.run(`CREATE TABLE IF NOT EXISTS "players" ("nick" TEXT(15) NOT NULL, "ip" TEXT NOT NULL)`, (error, result) => {
+            if(error) {
+                logger.error(`could not create database table`);
+                return
+            }
+            db.run(`INSERT INTO "players" ("nick", "ip") VALUES (:nick, :ip)`, { nick: netplay.nick, $ip: netplay.ip }, (error) => {
+                if(error) {
+                    logger.error(`could not insert player and IP into database`)
+                    return
+                }
+                logger.info(`player added to database`)
+            })
+        })
+    })
+    // send data to MQTT server
+    const data = JSON.stringify(netplay, null, 4)
+    mqttClient.publish(`${config.mqtt.topic}`, data)
     // ban by nick.
     if(nickBanSet.has(netplay.nick)) {
         logger.info(`nick found to be banned: ${netplay.nick}`)
-        exec(`/usr/sbin/iptables -t mangle -A PREROUTING -p tcp --src ${netplay.ip} --dport ${config.netplay.port} -j DROP`, (error, stdout, stderr) => {
+        exec(`iptables -t mangle -A PREROUTING -p tcp --src ${netplay.ip} --dport ${config.netplay.port} -j DROP`, (error, stdout, stderr) => {
             if (error) {
                 logger.error(`Error returned while banning connecting client ${error.message}`)
                 return
+            }
             if (stderr) {
                 logger.error(`Standard error returned ${stderr}`)
                 return
             }
-            if(stdout) {
+            if (stdout) {
                 logger.info(`Standard error reported while banning ${typeof stdout}`)
                 return
-            }
-            // only send notification if the connecting nick isn't banned
-            const data = JSON.stringify(netplay, null, 4)
             mqttClient.publish(`${config.mqtt.topic}`, data)
         })
+    }