var express = require('express'); var path = require('path'); var favicon = require('serve-favicon'); var logger = require('morgan'); var cookieParser = require('cookie-parser'); var bodyParser = require('body-parser'); const execSync = require('child_process').execSync; var app = express(); /* Read Config */ var json_file = require('jsonfile'); var glass_config = json_file.readFileSync('config/glass_config.json'); // uncomment after placing your favicon in /public //app.use(favicon(path.join(__dirname, 'public', 'favicon.ico'))); app.use(logger('dev')); app.use(bodyParser.json()); app.use(bodyParser.urlencoded({ extended: false })); app.use(cookieParser()); app.use(express.static(path.join(__dirname, 'public'))); /* Normal Web Routes */ app.use('/', require('./routes/index')); app.use('/users', require('./routes/users')); app.use('/get_stats', require('./routes/get_stats')); app.use('/dhcp_leases', require('./routes/dhcp_leases')); app.use('/dhcp_log', require('./routes/dhcp_log')); app.use('/dhcp_config', require('./routes/dhcp_config')); app.use('/dhcp_config_snapshots', require('./routes/dhcp_config_snapshots')); app.use('/dhcp_config_snapshot_view', require('./routes/dhcp_config_snapshot_view')); app.use('/dhcp_config_save', require('./routes/dhcp_config_save')); app.use('/dhcp_start_stop_restart', require('./routes/dhcp_start_stop_restart')); app.use('/api_examples', require('./routes/api_examples')); app.use('/glass_settings', require('./routes/glass_settings')); app.use('/glass_alerts', require('./routes/glass_alerts')); app.use('/glass_alert_settings_save', require('./routes/glass_alert_settings_save')); app.use('/glass_settings_save', require('./routes/glass_settings_save')); /* API Routes */ app.use('/api/get_active_leases/', require('./api/get_active_leases')); app.use('/api/get_subnet_details/', require('./api/get_subnet_details')); app.set('view engine', 'html'); // catch 404 and forward to error handler app.use(function(req, res, next) { var err = new Error('Not Found'); err.status = 404; next(err); }); // error handler app.use(function(err, req, res, next) { // set locals, only providing error in development res.locals.message = err.message; res.locals.error = req.app.get('env') === 'development' ? err : {}; // render the error page res.status(err.status || 500); res.send(err.message); }); module.exports = app; /** * Global Variables */ leases_per_minute = 0; cpu_utilization = 0; total_leases = 0; current_time = 0; leases_per_second = 0; current_leases_per_second = 0; leases_last_update_time = 0; listening_to_log_file = 0; options = {}; options.interval = 1000; host_name = execSync("cat /etc/hostname").toString(); /** * Ingest Current Lease File */ var lease_parser = require('./lib/lease_parser.js'); dhcp_lease_data = {}; lease_read_buffer = ""; fs = require('fs'); fs.readFile(glass_config.leases_file, 'utf8', function (err,data) { if (err) { return console.log(err); } else { lease_parser.parse(data); } }); /** * Leases File Listener */ var tail_module = require('always-tail'); tail = new tail_module( glass_config.leases_file, "\n", options ); tail.on("line", function(data) { unix_time = Math.floor(new Date() / 1000); /* Buffering lines until we get full lease data */ lease_read_buffer = lease_read_buffer + data + "\n"; /* End of lease - cut off and parse the buffer */ if (/}/i.test(data)){ lease_parser.parse(lease_read_buffer); lease_read_buffer = ""; } /* Count leases per second */ if(/lease/.test(data)) { leases_per_second++; } if(current_time != unix_time) { current_time = unix_time; current_leases_per_second = leases_per_second; leases_last_update_time = unix_time; leases_per_second = 0; } }); /** * Watch DHCP Log File */ var json_file = require('jsonfile'); var glass_config = json_file.readFileSync('config/glass_config.json'); var options = {}; options.interval = 1000; var dashboard_timer = setInterval(function(){ // console.log("Checking timers..."); unix_time = Math.floor(new Date() / 1000); if((unix_time - 5) > leases_last_update_time){ current_leases_per_second = 0; } // console.log(JSON.stringify(dhcp_lease_data, null, 2)); }, 5000); /** * Calculate leases per minute */ var leases_per_minute_data = []; var leases_per_minute_counter = 0; leases_per_minute_counter_timer = setInterval(function(){ // console.log("leases per minute counter %i", leases_per_minute_counter); leases_per_minute_data[leases_per_minute_counter] = current_leases_per_second; leases_per_minute_counter++; /* Count how many actual data sets we walked that have values */ leases_per_minute = 0; for (i = 0; i < 59; i++){ if(leases_per_minute_data[i] > 0) { leases_per_minute += leases_per_minute_data[i]; // console.log("iteration " + i + " val: " + leases_per_minute_data[i] + " lpm: " + leases_per_minute); } else { // console.log("no data " + i); } } if (leases_per_minute_counter == 60) leases_per_minute_counter = 0; }, 1000); /** * Poll: CPU Utilization */ cpu_utilization_poll = setInterval(function(){ cpu_utilization = parseFloat(execSync("top -bn 1 | awk 'NR>7{s+=$9} END {print s/4}'").toString()) }, (15 * 1000)); /** * Clean Expired Leases */ lease_clean_timer = setInterval(function(){ lease_parser.clean(); }, (60 * 1000)); function get_socket_clients_connected_count() { wss.clients.forEach(function each(client) { if (client.readyState === WebSocket.OPEN) { socket_clients++; } }); return socket_clients; } /** * Websocket Server */ const WebSocket = require('ws'); const wss = new WebSocket.Server({ port: 8080 }); options.interval = 300; var tail_dhcp_log = new tail_module( glass_config.log_file, "\n", options ); tail_dhcp_log.on("line", function(data) { if(listening_to_log_file) { wss.broadcast_event(data, 'dhcp_log_subscription'); } }); wss.on('connection', function connection(ws) { socket_clients++; console.log("[WS] CLIENT_CONNECT: Socket clients (" + socket_clients + ")"); if (!listening_to_log_file) { /* Watch log file for new information */ var tail_module = require('always-tail'); listening_to_log_file = 1; } }); wss.on('close', function close() { socket_clients--; console.log("[WS] CLIENT_DISCONNECT: Socket clients (" + socket_clients + ")"); }); function heartbeat() { this.isAlive = true; } function isJson(str) { try { JSON.parse(str); } catch (e) { return false; } return true; } wss.on('connection', function connection(ws) { ws.isAlive = true; ws.on('pong', heartbeat); ws.event_subscription = []; ws.on('message', function incoming(data) { if(data != "" && isJson(data)) { var json = JSON.parse(data); if(typeof json["event_subscription"] !== "undefined"){ console.log("[WS] Incoming Subscription '%s'", json['event_subscription']); ws.event_subscription[json["event_subscription"]] = 1; } if(typeof json["event_unsubscribe"] !== "undefined"){ console.log("[WS] event_unsubscribe '%s'", json['event_unsubscribe']); delete ws.event_subscription[json["event_unsubscribe"]]; } } }); stale_connections_audit(); }); wss.broadcast = function broadcast(data) { wss.clients.forEach(function each(client) { if (client.readyState === WebSocket.OPEN) { client.send(data); } }); }; wss.broadcast_event = function broadcast(data, event) { wss.clients.forEach(function each(client) { if (client.readyState === WebSocket.OPEN) { if(client.event_subscription[event]) client.send(data); } }); }; function stale_connections_audit() { socket_clients = 0; wss.clients.forEach(function each(ws) { if (ws.isAlive === false) return ws.terminate(); ws.isAlive = false; ws.ping('', false, true); socket_clients++; }); console.log("[WS] STATUS: Socket clients (" + socket_clients + ")"); } /* Keepalive - kill stale connections (30s poll) */ const interval = setInterval(function ping() { stale_connections_audit(); }, 30000); var socket_clients = 0; /** * Slack Hooks */ var Slack = require('slack-node'); webhookUri = glass_config.slack_webhook_url; slack = new Slack(); slack.setWebhook(webhookUri); function slack_message(message) { slack.webhook({ channel: glass_config.slack_alert_channel, username: "Glass", icon_emoji: "https://i.imgur.com/VDzAuAq.png", text: message }, function (err, response) { console.log(response); }); } /** * Alert Checks */ alert_status = []; alert_status['leases_per_second'] = 0; alert_check_timer = setInterval(function(){ if(glass_config.lease_per_second_threshold > 0) { if (current_leases_per_second <= glass_config.lease_per_second_threshold && alert_status['leases_per_second'] == 0) { alert_status['leases_per_second'] = 1; // slack_message(":warning: WARNING: DHCP leases per second have dropped below critical threshold (" + glass_config.lease_per_second_threshold + ") Current LP/s (" + current_leases_per_second + ")"); } else if (current_leases_per_second >= glass_config.lease_per_second_threshold && alert_status['leases_per_second'] == 1) { alert_status['leases_per_second'] = 0; // slack_message(":white_check_mark: CLEAR: DHCP leases per second have returned to above the critical threshold (" + glass_config.lease_per_second_threshold + ") Current LP/s (" + current_leases_per_second + ")"); } } }, (5 * 1000));