ppsguard.js

#!/usr/bin/env node
var dgram = require('dgram');

var Measured = require('measured')
var stats_rx = new Measured.Meter('rx')
var stats_cpu = new Measured.Gauge(function () { return os.loadavg() })

var os = require('os'), fs = require('fs')


var argv = require('minimist')(process.argv.slice(2))
var rxinterface = argv.interface || 'eth0'

var warning = argv.message || 'WARNING!'
var max_pps = argv.max_pps || 500
var max_mean = argv.max_mean || 1000
var max_cpu = argv.max_cpu || 2
var json = argv.json || false
var udp = argv.udp || false
var limit_amount = argv.limit_amount || 1
var limit_every = argv.limit_every || "second"

var RateLimiter = require('limiter').RateLimiter;
// 'second', 'minute', 'day', or a number of milliseconds
var limiter = new RateLimiter(limit_amount,limit_every);


if(udp) {
  var HOST = udp.split(':')[0];
  var PORT = udp.split(':')[1];
  var client = dgram.createSocket('udp4');
}

var lastR = 0
var getPPS = function () {
  var RX1 = fs.readFileSync('/sys/class/net/' + rxinterface + '/statistics/rx_packets', 'utf8')
  if (lastR != 0) stats_rx.mark((RX1 || 0) - (lastR))
  lastR = RX1 || 0
}

var sendUDP = function (message) {
  if (argv.debug) console.log('Sending Message',message,HOST,PORT);
  client.send(message, 0, message.length, PORT, HOST, function(err, bytes) {
    if (err) { throw err; client.close(); }
  });
}

setInterval(function () {
  getPPS()
  var pps = stats_rx.toJSON()
  var cpu = stats_cpu.toJSON()
  var res = {
    timestamp: new Date(),
    message: warning,
    pps: pps.currentRate,
    mean: pps.mean,
    cpu: cpu[0]
  }
  if (argv.debug) console.log('Dump:',res.cpu > max_cpu,res.pps > max_pps,res.mean > max_mean);
  if (res.cpu >= max_cpu || res.mean >= max_mean || res.pps >= max_pps) {
   // rate limiter
   if (argv.debug) console.log('Emit:',res);
   limiter.removeTokens(1, function() {
    if (udp) { sendUDP(JSON.stringify(res)); return; }
    else if (json) {
	console.log(res);
    } else {
	console.log(new Date(), JSON.stringify(res))
    }
   });
  }
}, argv.interval || 1000)

console.warn('PPS/CPU Counter started...', argv)