921 lines
33 KiB
PHP
921 lines
33 KiB
PHP
<?php
|
|
/*
|
|
* Copyright (C) 2014 <singh@devilcode.org>
|
|
* Modified and Relicensed by <f0o@devilcode.org> under the expressed
|
|
* permission by the Copyright-Holder <singh@devilcode.org>.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
* */
|
|
|
|
namespace LibreNMS;
|
|
|
|
use LibreNMS\DB\Eloquent;
|
|
use LibreNMS\Authentication\LegacyAuth;
|
|
|
|
class IRCBot
|
|
{
|
|
|
|
private $last_activity = '';
|
|
|
|
private $data = '';
|
|
|
|
private $authd = array();
|
|
|
|
private $debug = false;
|
|
|
|
private $server = '';
|
|
|
|
private $port = '';
|
|
|
|
private $ssl = false;
|
|
|
|
private $pass = '';
|
|
|
|
private $nick = 'LibreNMS';
|
|
|
|
private $chan = array();
|
|
|
|
private $commands = array(
|
|
'auth',
|
|
'quit',
|
|
'listdevices',
|
|
'device',
|
|
'port',
|
|
'down',
|
|
'version',
|
|
'status',
|
|
'log',
|
|
'help',
|
|
'reload',
|
|
'join',
|
|
);
|
|
|
|
private $external = array();
|
|
|
|
private $tick = 62500;
|
|
|
|
|
|
public function __construct()
|
|
{
|
|
global $config;
|
|
$this->log('Setting up IRC-Bot..');
|
|
|
|
$this->config = $config;
|
|
$this->debug = $this->config['irc_debug'];
|
|
$this->config['irc_authtime'] = $this->config['irc_authtime'] ? $this->config['irc_authtime'] : 3;
|
|
$this->max_retry = $this->config['irc_maxretry'];
|
|
$this->server = $this->config['irc_host'];
|
|
if ($this->config['irc_port'][0] == '+') {
|
|
$this->ssl = true;
|
|
$this->port = substr($this->config['irc_port'], 1);
|
|
} else {
|
|
$this->port = $this->config['irc_port'];
|
|
}
|
|
|
|
if ($this->config['irc_nick']) {
|
|
$this->nick = $this->config['irc_nick'];
|
|
}
|
|
|
|
if ($this->config['irc_chan']) {
|
|
if (is_array($this->config['irc_chan'])) {
|
|
$this->chan = $this->config['irc_chan'];
|
|
} elseif (strstr($this->config['irc_chan'], ',')) {
|
|
$this->chan = explode(',', $this->config['irc_chan']);
|
|
} else {
|
|
$this->chan = array($this->config['irc_chan']);
|
|
}
|
|
}
|
|
|
|
if ($this->config['irc_alert_chan']) {
|
|
if (strstr($this->config['irc_alert_chan'], ',')) {
|
|
$this->config['irc_alert_chan'] = explode(',', $this->config['irc_alert_chan']);
|
|
} else {
|
|
$this->config['irc_alert_chan'] = array($this->config['irc_alert_chan']);
|
|
}
|
|
}
|
|
|
|
if ($this->config['irc_pass']) {
|
|
$this->pass = $this->config['irc_pass'];
|
|
}
|
|
|
|
$this->loadExternal();
|
|
$this->log('Starting IRC-Bot..');
|
|
$this->init();
|
|
}//end __construct()
|
|
|
|
|
|
private function loadExternal()
|
|
{
|
|
if (!$this->config['irc_external']) {
|
|
return true;
|
|
}
|
|
|
|
$this->log('Caching external commands...');
|
|
if (!is_array($this->config['irc_external'])) {
|
|
$this->config['irc_external'] = explode(',', $this->config['irc_external']);
|
|
}
|
|
|
|
foreach ($this->config['irc_external'] as $ext) {
|
|
if (($this->external[$ext] = file_get_contents('includes/ircbot/'.$ext.'.inc.php')) == '') {
|
|
unset($this->external[$ext]);
|
|
}
|
|
}
|
|
|
|
return $this->log('Cached '.sizeof($this->external).' commands.');
|
|
}//end load_external()
|
|
|
|
|
|
private function init()
|
|
{
|
|
if ($this->config['irc_alert']) {
|
|
$this->connectAlert();
|
|
}
|
|
|
|
$this->last_activity = time();
|
|
|
|
$this->j = 2;
|
|
|
|
$this->connect();
|
|
$this->log('Connected');
|
|
if ($this->pass) {
|
|
fwrite($this->socket['irc'], 'PASS '.$this->pass."\n\r");
|
|
}
|
|
|
|
$this->doAuth();
|
|
while (true) {
|
|
foreach ($this->socket as $n => $socket) {
|
|
if (!is_resource($socket) || feof($socket)) {
|
|
$this->log("Socket '$n' closed. Restarting.");
|
|
break 2;
|
|
}
|
|
}
|
|
|
|
if (isset($this->tempnick)) {
|
|
$this->ircRaw('NICK '.$this->nick);
|
|
}
|
|
|
|
$this->getData();
|
|
if ($this->config['irc_alert']) {
|
|
$this->alertData();
|
|
}
|
|
|
|
if ($this->config['irc_conn_timeout']) {
|
|
$inactive_seconds = time() - $this->last_activity;
|
|
$max_inactive = $this->config['irc_conn_timeout'];
|
|
if ($inactive_seconds > $max_inactive) {
|
|
$this->log("No data from server since " . $max_inactive . " seconds. Restarting.");
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
usleep($this->tick);
|
|
}
|
|
|
|
return $this->init();
|
|
}//end init()
|
|
|
|
|
|
private function connectAlert()
|
|
{
|
|
$f = $this->config['install_dir'].'/.ircbot.alert';
|
|
if (( file_exists($f) && filetype($f) != 'fifo' && !unlink($f) ) || ( !file_exists($f) && !shell_exec("mkfifo $f && echo 1") )) {
|
|
$this->log('Error - Cannot create Alert-File');
|
|
return false;
|
|
}
|
|
|
|
if (($this->socket['alert'] = fopen($f, 'r+'))) {
|
|
$this->log('Opened Alert-File');
|
|
stream_set_blocking($this->socket['alert'], false);
|
|
return true;
|
|
}
|
|
|
|
$this->log('Error - Cannot open Alert-File');
|
|
return false;
|
|
}//end connect_alert()
|
|
|
|
|
|
private function read($buff)
|
|
{
|
|
$r = fread($this->socket[$buff], 64);
|
|
$this->buff[$buff] .= $r;
|
|
$r = strlen($r);
|
|
if (strstr($this->buff[$buff], "\n")) {
|
|
$tmp = explode("\n", $this->buff[$buff], 2);
|
|
$this->buff[$buff] = substr($this->buff[$buff], (strlen($tmp[0]) + 1));
|
|
if ($this->debug) {
|
|
$this->log("Returning buffer '$buff': '".trim($tmp[0])."'");
|
|
}
|
|
|
|
return $tmp[0];
|
|
}
|
|
|
|
if ($this->debug && $r > 0) {
|
|
$this->log("Expanding buffer '$buff' = '".trim($this->buff[$buff])."'");
|
|
}
|
|
|
|
return false;
|
|
}//end read()
|
|
|
|
|
|
private function alertData()
|
|
{
|
|
if (($alert = $this->read('alert')) !== false) {
|
|
$alert = json_decode($alert, true);
|
|
if (!is_array($alert)) {
|
|
return false;
|
|
}
|
|
|
|
switch ($alert['state']) :
|
|
case 3:
|
|
$severity_extended = '+';
|
|
break;
|
|
case 4:
|
|
$severity_extended = '-';
|
|
break;
|
|
default:
|
|
$severity_extended = '';
|
|
endswitch;
|
|
|
|
$severity = str_replace(array('warning', 'critical'), array($this->_color('Warning', 'yellow'), $this->_color('Critical', 'red')), $alert['severity']).$severity_extended.' ';
|
|
if ($alert['state'] == 0 and $this->config['irc_alert_utf8']) {
|
|
$severity = str_replace(array('Warning', 'Critical'), array('̶W̶a̶r̶n̶i̶n̶g', '̶C̶r̶i̶t̶i̶c̶a̶l'), $severity);
|
|
}
|
|
|
|
if ($this->config['irc_alert_chan']) {
|
|
foreach ($this->config['irc_alert_chan'] as $chan) {
|
|
$this->ircRaw('PRIVMSG '.$chan.' :'.$severity.trim($alert['title']));
|
|
if (!$this->config['irc_alert_short']) { // Only send the title if set to short
|
|
foreach (explode("\n", $alert['msg']) as $line) {
|
|
// We don't need to repeat the title
|
|
$line = strip_tags($line);
|
|
if (trim($line) != trim($alert['title'])) {
|
|
$this->ircRaw('PRIVMSG '.$chan.' :'.$line);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
foreach ($this->authd as $nick => $data) {
|
|
if ($data['expire'] >= time()) {
|
|
$this->ircRaw('PRIVMSG '.$nick.' :'.$severity.trim($alert['title']));
|
|
if (!$this->config['irc_alert_short']) { // Only send the title if set to short
|
|
foreach (explode("\n", $alert['msg']) as $line) {
|
|
// We don't need to repeat the title
|
|
$line = strip_tags($line);
|
|
if (trim($line) != trim($alert['title'])) {
|
|
$this->ircRaw('PRIVMSG '.$nick.' :'.$line);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}//end if
|
|
}//end alertData()
|
|
|
|
|
|
private function getData()
|
|
{
|
|
if (($data = $this->read('irc')) !== false) {
|
|
$this->last_activity = time();
|
|
$this->data = $data;
|
|
$ex = explode(' ', $this->data);
|
|
if ($ex[0] == 'PING') {
|
|
return $this->ircRaw('PONG '.$ex[1]);
|
|
}
|
|
|
|
if ($ex[1] == 376 || $ex[1] == 422 || ($ex[1] == 'MODE' && $ex[2] == $this->nick)) {
|
|
if ($this->j == 2) {
|
|
$this->joinChan();
|
|
$this->j = 0;
|
|
}
|
|
}
|
|
|
|
if (($this->config['irc_ctcp']) && (preg_match("/^:".chr(1).".*/", $ex[3]))) {
|
|
// Handle CTCP
|
|
$ctcp = trim(preg_replace("/[^A-Z]/", "", $ex[3]));
|
|
$ctcp_reply = null;
|
|
$this->log("Received irc CTCP: ".$ctcp." from ".$this->getUser($this->data));
|
|
switch ($ctcp) {
|
|
case 'VERSION':
|
|
$ctcp_reply = chr(1)."$ctcp ".$this->config['irc_ctcp_version'].chr(1);
|
|
break;
|
|
case 'PING':
|
|
$ctcp_reply = chr(1)."$ctcp ".$ex[4]. " ".$ex[5].chr(1);
|
|
break;
|
|
case 'TIME':
|
|
$ctcp_reply = chr(1)."$ctcp ".date('c').chr(1);
|
|
break;
|
|
}
|
|
if ($ctcp_reply !== null) {
|
|
$this->log("Sending irc CTCP: ".'NOTICE '.$this->getUser($this->data)." :".$ctcp_reply);
|
|
return $this->ircRaw('NOTICE '.$this->getUser($this->data)." :".$ctcp_reply);
|
|
}
|
|
}
|
|
|
|
if (($ex[1] == 'NICK') && (preg_replace("/^:/", "", $ex[2]) == $this->nick)) {
|
|
// Nickname changed successfully
|
|
if ($this->debug) {
|
|
$this->log("Regained our real nick");
|
|
}
|
|
unset($this->tempnick);
|
|
}
|
|
if (($ex[1] == 433) || ($ex[1] == 437)) {
|
|
// Nickname already in use / temp unavailable
|
|
if ($this->debug) {
|
|
$this->log("Nickname already in use...");
|
|
}
|
|
if ($ex[2] != "*") {
|
|
$this->tempnick = $ex[2];
|
|
}
|
|
if (!isset($this->tempnick)) {
|
|
$this->tempnick = $this->nick.rand(0, 99);
|
|
}
|
|
if ($this->debug) {
|
|
$this->log("Using temp nick ".$this->tempnick);
|
|
}
|
|
return $this->ircRaw('NICK '.$this->tempnick);
|
|
}
|
|
|
|
$this->command = str_replace(array(chr(10), chr(13)), '', $ex[3]);
|
|
if (strstr($this->command, ':.')) {
|
|
$this->handleCommand();
|
|
}
|
|
}
|
|
}//end getData()
|
|
|
|
|
|
private function joinChan($chan = false)
|
|
{
|
|
if ($chan) {
|
|
$this->chan[] = $chan;
|
|
}
|
|
|
|
foreach ($this->chan as $chan) {
|
|
$this->ircRaw('JOIN '.$chan);
|
|
}
|
|
|
|
return true;
|
|
}//end joinChan()
|
|
|
|
|
|
private function handleCommand()
|
|
{
|
|
$this->command = str_replace(':.', '', $this->command);
|
|
$tmp = explode(':.'.$this->command.' ', $this->data);
|
|
$this->user = $this->getAuthdUser();
|
|
if (!$this->isAuthd() && (isset($this->config['irc_auth']))) {
|
|
$this->hostAuth();
|
|
}
|
|
if ($this->isAuthd() || trim($this->command) == 'auth') {
|
|
$this->proceedCommand(str_replace("\n", '', trim($this->command)), trim($tmp[1]));
|
|
}
|
|
|
|
$this->authd[$this->getUser($this->data)] = $this->user;
|
|
return false;
|
|
}//end handleCommand()
|
|
|
|
|
|
private function proceedCommand($command, $params)
|
|
{
|
|
$command = strtolower($command);
|
|
if (in_array($command, $this->commands)) {
|
|
$this->chkdb();
|
|
$this->log($command." ( '".$params."' )");
|
|
return $this->{'_'.$command}($params);
|
|
} elseif ($this->external[$command]) {
|
|
$this->chkdb();
|
|
$this->log($command." ( '".$params."' ) [Ext]");
|
|
return eval($this->external[$command]);
|
|
}
|
|
|
|
return false;
|
|
}//end proceedCommand()
|
|
|
|
|
|
private function respond($msg)
|
|
{
|
|
$chan = $this->getChan($this->data);
|
|
return $this->sendMessage($msg, strstr($chan, '#') ? $chan : $this->getUser($this->data));
|
|
}//end respond()
|
|
|
|
|
|
private function getChan($param)
|
|
{
|
|
$data = explode('PRIVMSG ', $this->data, 3);
|
|
$data = explode(' ', $data[1], 2);
|
|
return $data[0];
|
|
}//end getChan()
|
|
|
|
|
|
private function getUser($param)
|
|
{
|
|
$arrData = explode('!', $param, 2);
|
|
return str_replace(':', '', $arrData[0]);
|
|
}//end getUser()
|
|
|
|
private function getUserHost($param)
|
|
{
|
|
$arrData = explode(' ', $param, 2);
|
|
return str_replace(':', '', $arrData[0]);
|
|
}//end getUserHost()
|
|
|
|
private function connect($try = 0)
|
|
{
|
|
if ($try > $this->max_retry) {
|
|
$this->log('Failed too many connection attempts, aborting');
|
|
return die();
|
|
}
|
|
|
|
$this->log('Trying to connect ('.($try + 1).') to '.$this->server.':'.$this->port.($this->ssl ? ' (SSL)' : ''));
|
|
if ($this->socket['irc']) {
|
|
$this->ircRaw('QUIT :Reloading');
|
|
fclose($this->socket['irc']);
|
|
}
|
|
|
|
if ($this->ssl) {
|
|
$server = 'ssl://'.$this->server;
|
|
} else {
|
|
$server = $this->server;
|
|
}
|
|
|
|
if ($this->ssl && $this->config['irc_disable_ssl_check']) {
|
|
$ssl_context_params = array('ssl'=>array('allow_self_signed'=> true, 'verify_peer' => false, 'verify_peer_name' => false ));
|
|
$ssl_context = stream_context_create($ssl_context_params);
|
|
$this->socket['irc'] = stream_socket_client($server.':'.$this->port, $errno, $errstr, 30, STREAM_CLIENT_CONNECT, $ssl_context);
|
|
} else {
|
|
$this->socket['irc'] = fsockopen($server, $this->port);
|
|
}
|
|
|
|
if (!is_resource($this->socket['irc'])) {
|
|
if ($try < 5) {
|
|
sleep(5);
|
|
} elseif ($try < 10) {
|
|
sleep(60);
|
|
} else {
|
|
sleep(300);
|
|
}
|
|
return $this->connect($try + 1);
|
|
} else {
|
|
stream_set_blocking($this->socket['irc'], false);
|
|
return true;
|
|
}
|
|
}//end connect()
|
|
|
|
|
|
private function doAuth()
|
|
{
|
|
if ($this->ircRaw('USER '.$this->nick.' 0 '.$this->nick.' :'.$this->nick) && $this->ircRaw('NICK '.$this->nick)) {
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}//end doAuth()
|
|
|
|
|
|
private function sendMessage($message, $chan)
|
|
{
|
|
if ($this->debug) {
|
|
$this->log("Sending 'PRIVMSG ".trim($chan).' :'.trim($message)."'");
|
|
}
|
|
|
|
return $this->ircRaw('PRIVMSG '.trim($chan).' :'.trim($message));
|
|
}//end sendMessage()
|
|
|
|
|
|
private function log($msg)
|
|
{
|
|
echo '['.date('r').'] '.trim($msg)."\n";
|
|
return true;
|
|
}//end log()
|
|
|
|
|
|
private function chkdb()
|
|
{
|
|
if (!Eloquent::isConnected()) {
|
|
try {
|
|
Eloquent::boot();
|
|
} catch (\PDOException $e) {
|
|
$this->log('Cannot connect to MySQL: ' . $e->getMessage());
|
|
return die();
|
|
}
|
|
}
|
|
return true;
|
|
}//end chkdb()
|
|
|
|
|
|
private function isAuthd()
|
|
{
|
|
if ($this->user['expire'] >= time()) {
|
|
$this->user['expire'] = (time() + ($this->config['irc_authtime'] * 3600));
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}//end isAuthd()
|
|
|
|
|
|
private function getAuthdUser()
|
|
{
|
|
return $this->authd[$this->getUser($this->data)];
|
|
}//end getAuthUser()
|
|
|
|
private function hostAuth()
|
|
{
|
|
global $authorizer;
|
|
foreach ($this->config['irc_auth'] as $nms_user => $hosts) {
|
|
foreach ($hosts as $host) {
|
|
$host = preg_replace("/\*/", ".*", $host);
|
|
if (preg_match("/$host/", $this->getUserHost($this->data))) {
|
|
$user_id = LegacyAuth::get()->getUserid(mres($nms_user));
|
|
$user = LegacyAuth::get()->getUser($user_id);
|
|
$this->user['name'] = $user['username'];
|
|
$this->user['id'] = $user_id;
|
|
$this->user['level'] = LegacyAuth::get()->getUserlevel($user['username']);
|
|
$this->user['expire'] = (time() + ($this->config['irc_authtime'] * 3600));
|
|
if ($this->user['level'] < 5) {
|
|
foreach (dbFetchRows('SELECT device_id FROM devices_perms WHERE user_id = ?', array($this->user['id'])) as $tmp) {
|
|
$this->user['devices'][] = $tmp['device_id'];
|
|
}
|
|
|
|
foreach (dbFetchRows('SELECT port_id FROM ports_perms WHERE user_id = ?', array($this->user['id'])) as $tmp) {
|
|
$this->user['ports'][] = $tmp['port_id'];
|
|
}
|
|
}
|
|
if ($this->debug) {
|
|
$this->log("HostAuth on irc for '".$user['username']."', ID: '".$user_id."', Host: '".$host);
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}//end hostAuth
|
|
|
|
|
|
private function ircRaw($params)
|
|
{
|
|
return fputs($this->socket['irc'], $params."\r\n");
|
|
}//end irc_raw()
|
|
|
|
|
|
private function _auth($params)
|
|
{
|
|
global $authorizer;
|
|
$params = explode(' ', $params, 2);
|
|
if (strlen($params[0]) == 64) {
|
|
if ($this->tokens[$this->getUser($this->data)] == $params[0]) {
|
|
$this->user['expire'] = (time() + ($this->config['irc_authtime'] * 3600));
|
|
$tmp_user = LegacyAuth::get()->getUser($this->user['id']);
|
|
$tmp = LegacyAuth::get()->getUserlevel($tmp_user['username']);
|
|
$this->user['level'] = $tmp;
|
|
if ($this->user['level'] < 5) {
|
|
foreach (dbFetchRows('SELECT device_id FROM devices_perms WHERE user_id = ?', array($this->user['id'])) as $tmp) {
|
|
$this->user['devices'][] = $tmp['device_id'];
|
|
}
|
|
|
|
foreach (dbFetchRows('SELECT port_id FROM ports_perms WHERE user_id = ?', array($this->user['id'])) as $tmp) {
|
|
$this->user['ports'][] = $tmp['port_id'];
|
|
}
|
|
}
|
|
|
|
return $this->respond('Authenticated.');
|
|
} else {
|
|
return $this->respond('Nope.');
|
|
}
|
|
} else {
|
|
$user_id = LegacyAuth::get()->getUserid(mres($params[0]));
|
|
$user = LegacyAuth::get()->getUser($user_id);
|
|
if ($user['email'] && $user['username'] == $params[0]) {
|
|
$token = hash('gost', openssl_random_pseudo_bytes(1024));
|
|
$this->tokens[$this->getUser($this->data)] = $token;
|
|
$this->user['name'] = $params[0];
|
|
$this->user['id'] = $user['user_id'];
|
|
if ($this->debug) {
|
|
$this->log("Auth for '".$params[0]."', ID: '".$user['user_id']."', Token: '".$token."', Mail: '".$user['email']."'");
|
|
}
|
|
|
|
if (send_mail($user['email'], 'LibreNMS IRC-Bot Authtoken', "Your Authtoken for the IRC-Bot:\r\n\r\n".$token."\r\n\r\n") === true) {
|
|
return $this->respond('Token sent!');
|
|
} else {
|
|
return $this->respond('Sorry, seems like mail doesnt like us.');
|
|
}
|
|
} else {
|
|
return $this->respond('Who are you again?');
|
|
}
|
|
}//end if
|
|
return false;
|
|
}//end _auth()
|
|
|
|
|
|
private function _reload()
|
|
{
|
|
if ($this->user['level'] == 10) {
|
|
global $config;
|
|
$config = array();
|
|
$config['install_dir'] = $this->config['install_dir'];
|
|
chdir($config['install_dir']);
|
|
include 'includes/defaults.inc.php';
|
|
include 'config.php';
|
|
include 'includes/definitions.inc.php';
|
|
$this->respond('Reloading configuration & defaults');
|
|
if ($config != $this->config) {
|
|
return $this->__construct();
|
|
}
|
|
} else {
|
|
return $this->respond('Permission denied.');
|
|
}
|
|
}//end _reload()
|
|
|
|
|
|
private function _join($params)
|
|
{
|
|
if ($this->user['level'] == 10) {
|
|
return $this->joinChan($params);
|
|
} else {
|
|
return $this->respond('Permission denied.');
|
|
}
|
|
}//end _join()
|
|
|
|
|
|
private function _quit($params)
|
|
{
|
|
if ($this->user['level'] == 10) {
|
|
$this->ircRaw("QUIT :Requested");
|
|
return die();
|
|
} else {
|
|
return $this->respond('Permission denied.');
|
|
}
|
|
}//end _quit()
|
|
|
|
|
|
private function _help($params)
|
|
{
|
|
foreach ($this->commands as $cmd) {
|
|
$msg .= ', '.$cmd;
|
|
}
|
|
|
|
$msg = substr($msg, 2);
|
|
return $this->respond("Available commands: $msg");
|
|
}//end _help()
|
|
|
|
|
|
private function _version($params)
|
|
{
|
|
$versions = version_info();
|
|
$schema_version = $versions['db_schema'];
|
|
$version = substr($versions['local_sha'], 0, 7);
|
|
|
|
$msg = $this->config['project_name_version'].', Version: '.$version.', DB schema: #'.$schema_version.', PHP: '.PHP_VERSION;
|
|
return $this->respond($msg);
|
|
}//end _version()
|
|
|
|
|
|
private function _log($params)
|
|
{
|
|
$num = 1;
|
|
if ($params > 1) {
|
|
$num = $params;
|
|
}
|
|
|
|
if ($this->user['level'] < 5) {
|
|
$tmp = dbFetchRows('SELECT `event_id`,`host`,`datetime`,`message`,`type` FROM `eventlog` WHERE `host` IN ('.implode(',', $this->user['devices']).') ORDER BY `event_id` DESC LIMIT '.mres($num));
|
|
} else {
|
|
$tmp = dbFetchRows('SELECT `event_id`,`host`,`datetime`,`message`,`type` FROM `eventlog` ORDER BY `event_id` DESC LIMIT '.mres($num));
|
|
}
|
|
|
|
foreach ($tmp as $device) {
|
|
$hostid = dbFetchRow('SELECT `hostname` FROM `devices` WHERE `device_id` = '.$device['host']);
|
|
$this->respond($device['event_id'].' '.$hostid['hostname'].' '.$device['datetime'].' '.$device['message'].' '.$device['type']);
|
|
}
|
|
|
|
if (!$hostid) {
|
|
$this->respond('Nothing to see, maybe a bug?');
|
|
}
|
|
|
|
return true;
|
|
}//end _log()
|
|
|
|
|
|
private function _down($params)
|
|
{
|
|
if ($this->user['level'] < 5) {
|
|
$tmp = dbFetchRows('SELECT `hostname` FROM `devices` WHERE status=0 AND `device_id` IN ('.implode(',', $this->user['devices']).')');
|
|
} else {
|
|
$tmp = dbFetchRows('SELECT `hostname` FROM `devices` WHERE status=0');
|
|
}
|
|
|
|
foreach ($tmp as $db) {
|
|
if ($db['hostname']) {
|
|
$msg .= ', '.$db['hostname'];
|
|
}
|
|
}
|
|
|
|
$msg = substr($msg, 2);
|
|
$msg = $msg ? $msg : 'Nothing to show :)';
|
|
return $this->respond($msg);
|
|
}//end _down()
|
|
|
|
|
|
private function _device($params)
|
|
{
|
|
$params = explode(' ', $params);
|
|
$hostname = $params[0];
|
|
$device = dbFetchRow('SELECT * FROM `devices` WHERE `hostname` = ?', array($hostname));
|
|
if (!$device) {
|
|
return $this->respond('Error: Bad or Missing hostname, use .listdevices to show all devices.');
|
|
}
|
|
|
|
if ($this->user['level'] < 5 && !in_array($device['device_id'], $this->user['devices'])) {
|
|
return $this->respond('Error: Permission denied.');
|
|
}
|
|
|
|
$status = $device['status'] ? 'Up '.formatUptime($device['uptime']) : 'Down';
|
|
$status .= $device['ignore'] ? '*Ignored*' : '';
|
|
$status .= $device['disabled'] ? '*Disabled*' : '';
|
|
return $this->respond($device['os'].' '.$device['version'].' '.$device['features'].' '.$status);
|
|
}//end _device()
|
|
|
|
|
|
private function _port($params)
|
|
{
|
|
$params = explode(' ', $params);
|
|
$hostname = $params[0];
|
|
$ifname = $params[1];
|
|
if (!$hostname || !$ifname) {
|
|
return $this->respond('Error: Missing hostname or ifname.');
|
|
}
|
|
|
|
$device = dbFetchRow('SELECT * FROM `devices` WHERE `hostname` = ?', array($hostname));
|
|
$port = dbFetchRow('SELECT * FROM `ports` WHERE (`ifName` = ? OR `ifDescr` = ?) AND device_id = ?', array($ifname, $ifname, $device['device_id']));
|
|
if ($this->user['level'] < 5 && !in_array($port['port_id'], $this->user['ports']) && !in_array($device['device_id'], $this->user['devices'])) {
|
|
return $this->respond('Error: Permission denied.');
|
|
}
|
|
|
|
$bps_in = formatRates($port['ifInOctets_rate'] * 8);
|
|
$bps_out = formatRates($port['ifOutOctets_rate'] * 8);
|
|
$pps_in = format_bi($port['ifInUcastPkts_rate']);
|
|
$pps_out = format_bi($port['ifOutUcastPkts_rate']);
|
|
return $this->respond($port['ifAdminStatus'].'/'.$port['ifOperStatus'].' '.$bps_in.' > bps > '.$bps_out.' | '.$pps_in.'pps > PPS > '.$pps_out.'pps');
|
|
}//end _port()
|
|
|
|
|
|
private function _listdevices($params)
|
|
{
|
|
if ($this->user['level'] < 5) {
|
|
$tmp = dbFetchRows('SELECT `hostname` FROM `devices` WHERE `device_id` IN ('.implode(',', $this->user['devices']).')');
|
|
} else {
|
|
$tmp = dbFetchRows('SELECT `hostname` FROM `devices`');
|
|
}
|
|
|
|
foreach ($tmp as $device) {
|
|
$msg .= ', '.$device['hostname'];
|
|
}
|
|
|
|
$msg = substr($msg, 2);
|
|
$msg = $msg ? $msg : 'Nothing to show..?';
|
|
return $this->respond($msg);
|
|
}//end _listdevices()
|
|
|
|
|
|
private function _status($params)
|
|
{
|
|
$params = explode(' ', $params);
|
|
$statustype = $params[0];
|
|
if ($this->user['level'] < 5) {
|
|
$d_w = ' WHERE device_id IN ('.implode(',', $this->user['devices']).')';
|
|
$d_a = ' AND device_id IN ('.implode(',', $this->user['devices']).')';
|
|
$p_w = ' WHERE port_id IN ('.implode(',', $this->user['ports']).') OR device_id IN ('.implode(',', $this->user['devices']).')';
|
|
$p_a = ' AND (I.port_id IN ('.implode(',', $this->user['ports']).') OR I.device_id IN ('.implode(',', $this->user['devices']).'))';
|
|
}
|
|
|
|
switch ($statustype) {
|
|
case 'devices':
|
|
case 'device':
|
|
case 'dev':
|
|
$devcount = dbFetchCell('SELECT count(*) FROM devices'.$d_w);
|
|
$devup = dbFetchCell("SELECT count(*) FROM devices WHERE status = '1' AND `ignore` = '0'".$d_a);
|
|
$devdown = dbFetchCell("SELECT count(*) FROM devices WHERE status = '0' AND `ignore` = '0'".$d_a);
|
|
$devign = dbFetchCell("SELECT count(*) FROM devices WHERE `ignore` = '1'".$d_a);
|
|
$devdis = dbFetchCell("SELECT count(*) FROM devices WHERE `disabled` = '1'".$d_a);
|
|
if ($devup > 0) {
|
|
$devup = $this->_color($devup, 'green');
|
|
}
|
|
if ($devdown > 0) {
|
|
$devdown = $this->_color($devdown, 'red');
|
|
$devcount = $this->_color($devcount, 'yellow', null, 'bold');
|
|
} else {
|
|
$devcount = $this->_color($devcount, 'green', null, 'bold');
|
|
}
|
|
$msg = 'Devices: '.$devcount.' ('.$devup.' up, '.$devdown.' down, '.$devign.' ignored, '.$devdis.' disabled'.')';
|
|
break;
|
|
|
|
case 'ports':
|
|
case 'port':
|
|
case 'prt':
|
|
$prtcount = dbFetchCell('SELECT count(*) FROM ports'.$p_w);
|
|
$prtup = dbFetchCell("SELECT count(*) FROM ports AS I, devices AS D WHERE I.ifOperStatus = 'up' AND I.ignore = '0' AND I.device_id = D.device_id AND D.ignore = '0'".$p_a);
|
|
$prtdown = dbFetchCell("SELECT count(*) FROM ports AS I, devices AS D WHERE I.ifOperStatus = 'down' AND I.ifAdminStatus = 'up' AND I.ignore = '0' AND D.device_id = I.device_id AND D.ignore = '0'".$p_a);
|
|
$prtsht = dbFetchCell("SELECT count(*) FROM ports AS I, devices AS D WHERE I.ifAdminStatus = 'down' AND I.ignore = '0' AND D.device_id = I.device_id AND D.ignore = '0'".$p_a);
|
|
$prtign = dbFetchCell("SELECT count(*) FROM ports AS I, devices AS D WHERE D.device_id = I.device_id AND (I.ignore = '1' OR D.ignore = '1')".$p_a);
|
|
// $prterr = dbFetchCell("SELECT count(*) FROM ports AS I, devices AS D WHERE D.device_id = I.device_id AND (I.ignore = '0' OR D.ignore = '0') AND (I.ifInErrors_delta > '0' OR I.ifOutErrors_delta > '0')".$p_a);
|
|
if ($prtup > 0) {
|
|
$prtup = $this->_color($prtup, 'green');
|
|
}
|
|
if ($prtdown > 0) {
|
|
$prtdown = $this->_color($prtdown, 'red');
|
|
$prtcount = $this->_color($prtcount, 'yellow', null, 'bold');
|
|
} else {
|
|
$prtcount = $this->_color($prtcount, 'green', null, 'bold');
|
|
}
|
|
$msg = 'Ports: '.$prtcount.' ('.$prtup.' up, '.$prtdown.' down, '.$prtign.' ignored, '.$prtsht.' shutdown'.')';
|
|
break;
|
|
|
|
case 'services':
|
|
case 'service':
|
|
case 'srv':
|
|
$status_counts = array();
|
|
$status_colors = array(0 => 'green', 3 => 'lightblue', 1 => 'yellow', 2 => 'red');
|
|
$srvcount = dbFetchCell('SELECT COUNT(*) FROM services'.$d_w);
|
|
$srvign = dbFetchCell("SELECT COUNT(*) FROM services WHERE service_ignore = 1".$d_a);
|
|
$srvdis = dbFetchCell("SELECT COUNT(*) FROM services WHERE service_disabled = 1".$d_a);
|
|
$service_status = dbFetchRows("SELECT `service_status`, COUNT(*) AS `count` FROM `services` WHERE `service_disabled`=0 AND `service_ignore`=0 $d_a GROUP BY `service_status`");
|
|
$service_status = array_column($service_status, 'count', 'service_status'); // key by status
|
|
|
|
foreach ($status_colors as $status => $color) {
|
|
if (isset($service_status[$status])) {
|
|
$status_counts[$status] = $this->_color($service_status[$status], $color);
|
|
$srvcount = $this->_color($srvcount, $color, null, 'bold'); // upgrade the main count color
|
|
} else {
|
|
$status_counts[$status] = 0;
|
|
}
|
|
}
|
|
|
|
$msg = "Services: $srvcount ({$status_counts[0]} up, {$status_counts[2]} down, {$status_counts[1]} warning, {$status_counts[3]} unknown, $srvign ignored, $srvdis disabled)";
|
|
break;
|
|
|
|
default:
|
|
$msg = 'Error: STATUS requires one of the following: <devices|device|dev>|<ports|port|prt>|<services|service|src>';
|
|
break;
|
|
}//end switch
|
|
|
|
return $this->respond($msg);
|
|
}//end _status()
|
|
|
|
private function _color($text, $fg_color, $bg_color = null, $other = null)
|
|
{
|
|
$colors = array(
|
|
'white' => "00",
|
|
'black' => "01",
|
|
'blue' => "02",
|
|
'green' => "03",
|
|
'red' => "04",
|
|
'brown' => "05",
|
|
'purple' => "06",
|
|
'orange' => "07",
|
|
'yellow' => "08",
|
|
'lightgreen' => "09",
|
|
'cyan' => "10",
|
|
'lightcyan' => "11",
|
|
'lightblue' => "12",
|
|
'pink' => "13",
|
|
'grey' => "14",
|
|
'lightgrey' => "15",
|
|
);
|
|
$ret = chr(3);
|
|
if (array_key_exists($fg_color, $colors)) {
|
|
$ret .= $colors[$fg_color];
|
|
if (array_key_exists($bg_color, $colors)) {
|
|
$ret .= ",".$colors[$fg_color];
|
|
}
|
|
}
|
|
switch ($other) {
|
|
case 'bold':
|
|
$ret .= chr(2);
|
|
break;
|
|
case 'underline':
|
|
$ret .= chr(31);
|
|
break;
|
|
case 'italics':
|
|
case 'reverse':
|
|
$ret .= chr(22);
|
|
break;
|
|
}
|
|
$ret .= $text;
|
|
$ret .= chr(15);
|
|
return $ret;
|
|
}// end _color
|
|
}//end class
|