mirror of
https://github.com/librenms/librenms.git
synced 2024-10-07 16:52:45 +00:00
d841625f12
* Refactor database and config init Connect to the database without loading full config Load config completely so post-processing is always done consistently. Erase existing $config when loading, fixes issues in case we load the config twice. If the database is not connected, don't try to load database settings. (Fixes some db errors on install) Attempt to remove $config access/modification before init.php Remove usage of db_name, that might not match the connected database. Centralize db config loading, so we consistently apply db_test database settings. Many of these changes are influenced by Laravel port. * Some safety so we don't assign strings to numeric port field Smooth out phpunit bootstrap * Fix a couple of scrutinizer warnings.
924 lines
33 KiB
PHP
924 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\Authentication\Auth;
|
|
use LibreNMS\Exceptions\DatabaseConnectException;
|
|
|
|
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, $database_link;
|
|
$this->log('Setting up IRC-Bot..');
|
|
if (is_resource($database_link)) {
|
|
$this->sql = $database_link;
|
|
}
|
|
|
|
$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 (!is_resource($this->sql)) {
|
|
try {
|
|
$this->sql = dbConnect();
|
|
} catch (DatabaseConnectException $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 = Auth::get()->getUserid(mres($nms_user));
|
|
$user = Auth::get()->getUser($user_id);
|
|
$this->user['name'] = $user['username'];
|
|
$this->user['id'] = $user_id;
|
|
$this->user['level'] = Auth::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 = Auth::get()->getUser($this->user['id']);
|
|
$tmp = Auth::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 = Auth::get()->getUserid(mres($params[0]));
|
|
$user = Auth::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
|