netplaySniff – Diff between revs 5 and 6

Subversion Repositories:

?pathlinks?

Rev 5 – Rev 7 – Go to most recent revision – Details – Blame – Last modification – View Log

 const shortHash = require('short-hash')
 const { exec } = require("child_process")
 const Inotify = require('inotify-remastered').Inotify
 const inotify = new Inotify()
 const sqlite = require('sqlite3')
+const { Command } = require('commander')
-// load configuration file.
-let config = YAML.load('config.yml')
-// set up logger.
+// set up logger
 const logger = createLogger({
     format: format.combine(
         format.timestamp({
             format: 'YYYYMMDDHHmmss'
         }),
         format.printf(info =>
-            `${info.timestamp} ${info.level}: ${info.message}`+(info.splat !== undefined?`${info.splat}`:" ")
+            `${info.timestamp} ${info.level}: ${info.message}` + (info.splat !== undefined ? `${info.splat}` : " ")
+        )
     ),
     transports: [
         new transports.Console({
+            }
+        )
+    ]
 })
+const program = new Command()
-// set up packet capture
+program
-const cap = new Cap()
+    .name('netplaySniff')
+    .description(`
-const device = Cap.findDevice(`${config.router}`)
+  +--+
-const filter = `tcp and dst port ${config.netplay.port} and dst host ${config.netplay.host}`
+ |    | Monitor netplay traffic, sniff users and their
+ | || | IP addresses and store them in a database.
+  +--+
-const bufSize = 10 * 1024 * 1024
+`)
-const buffer = Buffer.alloc(65535)
-const linkType = cap.open(device, filter, bufSize, buffer)
+    .version('1.0')
+program
+    .command('run')
+    .option('-c, --config <path>', 'path to the configuration file', 'config.yml')
+    .option('-d, --database <path>', 'the path where to store a database', 'db/players.db')
+    .description('run the program as a daemon')
+    .action((options) => {
+        logger.info(`running as a daemon: ${options.database}`)
-cap.setMinBytes && cap.setMinBytes(0)
+        logger.info(`running as a daemon: ${options.config}`)
+        // load configuration file.
-let nickBanSet = new Set(config.bans.nicknames)
+        var config = YAML.load(options.config)
-// Watch the configuration file for changes.
+        // Watch the configuration file for changes.
-const configWatch = inotify.addWatch({
+        const configWatch = inotify.addWatch({
-    path: 'config.yml',
+            path: options.config,
-    watch_for: Inotify.IN_MODIFY,
+            watch_for: Inotify.IN_MODIFY,
-    callback: function(event) {
+            callback: function (event) {
-        logger.info(`Reloading configuration file config.yml`)
+                logger.info(`Reloading configuration file config.yml`)
-        config = YAML.load('config.yml')
+                config = YAML.load(options.config)
-        nickBanSet = new Set(config.bans.nicknames)
+            }
-    }
-})
+        })
-const mqttClient = mqtt.connect(config.mqtt.connect)
+        const mqttClient = mqtt.connect(config.mqtt.connect)
+        mqttClient.on('reconnect', () => {
+            logger.info('Reconnecting to MQTT server...')
+        })
+        mqttClient.on('connect', () => {
+            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 MQTT server.')
+                    return
+                }
+                logger.info('Subscribed to MQTT server.')
+            })
+        })
-mqttClient.on('reconnect', () => {
+        mqttClient.on('close', () => {
-    logger.info('Reconnecting to MQTT server...')
+            logger.error('Disconnected from MQTT server.')
+        })
+        mqttClient.on('error', (error) => {
+            logger.error(`MQTT ${error}`)
+            console.log(error)
+        })
-})
+        // set up packet capture
+        const cap = new Cap()
+        const device = Cap.findDevice(`${config.router}`)
+        const filter = `tcp and dst port ${config.netplay.port} and dst host ${config.netplay.host}`
+        const bufSize = 10 * 1024 * 1024
+        const buffer = Buffer.alloc(65535)
-mqttClient.on('connect', () => {
+        const linkType = cap.open(device, filter, bufSize, buffer)
-    logger.info('Connected to MQTT server.')
+        cap.setMinBytes && cap.setMinBytes(0)
-    // Subscribe to group message notifications with group name and password.
+        cap.on('packet', (bytes, truncated) => processPacket(bytes, truncated, config, mqttClient))
     mqttClient.subscribe(`${config.mqtt.topic}`, (error) => {
-        if (error) {
+        let netplay = {}
-            logger.info('Error subscribing to MQTT server.')
-            return
-        }
+        if (linkType !== 'ETHERNET') {
+            return
-        logger.info('Subscribed to MQTT server.')
+        }
-    })
-})
+        var ret = decoders.Ethernet(buffer)
+        if (ret.info.type !== PROTOCOL.ETHERNET.IPV4) {
+            return
+        }
+        ret = decoders.IPV4(buffer, ret.offset)
-mqttClient.on('close', () => {
+        netplay.ip = ret.info.srcaddr
+        if (ret.info.protocol !== PROTOCOL.IP.TCP) {
-    logger.error('Disconnected from MQTT server.')
+            return
-})
-mqttClient.on('error', (error) => {
-    logger.error(`MQTT ${error}`)
-    console.log(error)
-})
-cap.on('packet', function(bytes, truncated) {
-    let netplay = {}
-    if(linkType !== 'ETHERNET') {
-        return
-    }
-    var ret = decoders.Ethernet(buffer)
-    if (ret.info.type !== PROTOCOL.ETHERNET.IPV4) {
-        return
-    }
-    ret = decoders.IPV4(buffer, ret.offset)
-    netplay.ip = ret.info.srcaddr
-    if (ret.info.protocol !== PROTOCOL.IP.TCP) {
-        return
-    }
+        }
-    var dataLength = ret.info.totallen - ret.hdrlen
-    ret = decoders.TCP(buffer, ret.offset)
-    dataLength -= ret.hdrlen
-    var payload = buffer.subarray(ret.offset, ret.offset + dataLength)
-    // look for the NETPLAY_CMD_NICK in "netplay_private.h" data marker.
-    if(payload.indexOf('0020', 0, "hex") !== 2) {
-        return
-    }
-    // remove NULL and NETPLAY_CMD_NICK
-    netplay.nick = payload.toString().replace(/[\u0000\u0020]+/gi, '')
+        var dataLength = ret.info.totallen - ret.hdrlen
     netplay.hash = shortHash(`${netplay.nick}${netplay.ip}`)
-    netplay.time = new Date().toISOString()
+        ret = decoders.TCP(buffer, ret.offset)
+        dataLength -= ret.hdrlen
+        var payload = buffer.subarray(ret.offset, ret.offset + dataLength)
+        // look for the NETPLAY_CMD_NICK in "netplay_private.h" data marker.
+        if (payload.indexOf('0020', 0, "hex") !== 2) {
+            return
+        }
-    logger.info(`Player ${netplay.nick} joined via IP ${netplay.ip}`);
+        // remove NULL and NETPLAY_CMD_NICK
-    const db = new sqlite.Database(config.db.file, sqlite.OPEN_CREATE | sqlite.OPEN_READWRITE | sqlite.OPEN_FULLMUTEX, (error) => {
+        netplay.nick = payload.toString().replace(/[\u0000\u0020]+/gi, '')
-        if(error) {
+        netplay.hash = shortHash(`${netplay.nick}${netplay.ip}`)
+        netplay.time = new Date().toISOString()
-            logger.error(`failed to open database: ${config.db.file}`)
-            return
+        logger.info(`Player ${netplay.nick} joined via IP ${netplay.ip}`);
-        }
         const db = new sqlite.Database(config.db.file, sqlite.OPEN_CREATE | sqlite.OPEN_READWRITE | sqlite.OPEN_FULLMUTEX, (error) => {
-        db.run(`CREATE TABLE IF NOT EXISTS "players" ("nick" TEXT(15) NOT NULL, "ip" TEXT NOT NULL)`, (error, result) => {
+            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) {
+                if (error) {
-                logger.error(`could not create database table: ${error}`);
+                    logger.error(`could not create database table: ${error}`);
+                    return
                 return
                 db.run(`INSERT INTO "players" ("nick", "ip") VALUES ($nick, $ip)`, { $nick: netplay.nick, $ip: netplay.ip }, (error) => {
-            db.run(`INSERT INTO "players" ("nick", "ip") VALUES ($nick, $ip)`, { $nick: netplay.nick, $ip: netplay.ip }, (error) => {
-                if(error) {
+                    if (error) {
-                    logger.error(`could not insert player and IP into database: ${error}`)
+                        logger.error(`could not insert player and IP into database: ${error}`)
-                    return
+                        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, (error, packet) => {
-        logger.info(`player data sent to MQTT broker`)
-    })
-    // ban by nick.
-    if(nickBanSet.has(netplay.nick)) {
-        logger.info(`nick found to be banned: ${netplay.nick}`)
-        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
-            }
+                    logger.info(`player added to database`)
+                })
+            })
+        })
+        // send data to MQTT server
+        const data = JSON.stringify(netplay, null, 4)
+        mqttClient.publish(`${config.mqtt.topic}`, data, (error, packet) => {
+            logger.info(`player data sent to MQTT broker`)
+        })
+        // ban by nick.
+        let nickBanSet = new Set(config.bans.nicknames)
+        if (nickBanSet.has(netplay.nick)) {
+            logger.info(`nick found to be banned: ${netplay.nick}`)
+            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) {
+                if (stderr) {
+                    logger.error(`Standard error returned ${stderr}`)