mirror of
				https://github.com/librenms/librenms.git
				synced 2024-10-07 16:52:45 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			734 lines
		
	
	
		
			25 KiB
		
	
	
	
		
			PHP
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			734 lines
		
	
	
		
			25 KiB
		
	
	
	
		
			PHP
		
	
	
		
			Executable File
		
	
	
	
	
| #!/usr/bin/env 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/>.
 | |
|  * */
 | |
| 
 | |
| chdir(dirname($argv[0]));
 | |
| require_once 'includes/defaults.inc.php';
 | |
| require_once 'config.php';
 | |
| require_once 'includes/definitions.inc.php';
 | |
| require_once 'includes/functions.php';
 | |
| require_once 'includes/discovery/functions.inc.php';
 | |
| error_reporting(E_ERROR);
 | |
| 
 | |
| class ircbot {
 | |
| 
 | |
|     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'];
 | |
|             }
 | |
|             else if (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->load_external();
 | |
|         $this->log('Starting IRC-Bot..');
 | |
|         $this->init();
 | |
| 
 | |
|     }//end __construct()
 | |
| 
 | |
| 
 | |
|     private function load_external() {
 | |
|         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->connect_alert();
 | |
|         }
 | |
| 
 | |
|         $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;
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             $this->getData();
 | |
|             if ($this->config['irc_alert']) {
 | |
|                 $this->alertData();
 | |
|             }
 | |
| 
 | |
|             usleep($this->tick);
 | |
|         }
 | |
| 
 | |
|         return $this->init();
 | |
| 
 | |
|     }//end init()
 | |
| 
 | |
| 
 | |
|     private function connect_alert() {
 | |
|         $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(chr(3).'8Warning', chr(3).'4Critical'), $alert['severity']).$severity_extended.chr(3).' ';
 | |
|             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->irc_raw('PRIVMSG '.$chan.' :'.$severity.trim($alert['title']).' - Rule: '.trim($alert['name'] ? $alert['name'] : $alert['rule']).(sizeof($alert['faults']) > 0 ? ' - Faults:' : ''));
 | |
|                     foreach ($alert['faults'] as $k => $v) {
 | |
|                         $this->irc_raw('PRIVMSG '.$chan.' :#'.$k.' '.$v['string']);
 | |
|                     }
 | |
|                 }
 | |
|             }
 | |
|             else {
 | |
|                 foreach ($this->authd as $nick => $data) {
 | |
|                     if ($data['expire'] >= time()) {
 | |
|                         $this->irc_raw('PRIVMSG '.$nick.' :'.$severity.trim($alert['title']).' - Rule: '.trim($alert['name'] ? $alert['name'] : $alert['rule']).(sizeof($alert['faults']) > 0 ? ' - Faults'.(sizeof($alert['faults']) > 3 ? ' (showing first 3 out of '.sizeof($alert['faults']).' )' : '' ).':' : ''));
 | |
|                         foreach ($alert['faults'] as $k => $v) {
 | |
|                             $this->irc_raw('PRIVMSG '.$nick.' :#'.$k.' '.$v['string']);
 | |
|                             if ($k >= 3) {
 | |
|                                 break;
 | |
|                             }
 | |
|                         }
 | |
|                     }
 | |
|                 }
 | |
|             }
 | |
|         }//end if
 | |
| 
 | |
|     }//end alertData()
 | |
| 
 | |
| 
 | |
|     private function getData() {
 | |
|         if (($data = $this->read('irc')) !== false) {
 | |
|             $this->data = $data;
 | |
|             $ex         = explode(' ', $this->data);
 | |
|             if ($ex[0] == 'PING') {
 | |
|                 return $this->irc_raw('PONG '.$ex[1]);
 | |
|             }
 | |
| 
 | |
|             if ($ex[1] == 376 || ($ex[1] == 'MODE' && $ex[2] == $this->nick)) {
 | |
|                 if ($this->j == 2) {
 | |
|                     $this->joinChan();
 | |
|                     $this->j = 0;
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             $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->irc_raw('JOIN '.$chan);
 | |
|         }
 | |
| 
 | |
|         return true;
 | |
| 
 | |
|     }//end joinChan()
 | |
| 
 | |
| 
 | |
|     private function handleCommand() {
 | |
|         $this->command = str_replace(':.', '', $this->command);
 | |
|         $tmp           = explode(':.'.$this->command.' ', $this->data);
 | |
|         $this->user    = $this->get_user();
 | |
|         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);
 | |
|         }
 | |
|         else if ($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 connect($try) {
 | |
|         if ($try > $this->max_retry) {
 | |
|             $this->log('Failed too many connection attempts, aborting');
 | |
|             return die();
 | |
|         }
 | |
| 
 | |
|         $this->log('Trying to connect to '.$this->server.':'.$this->port.($this->ssl ? ' (SSL)' : ''));
 | |
|         if ($this->socket['irc']) {
 | |
|             fclose($this->socket['irc']);
 | |
|         }
 | |
| 
 | |
|         if ($this->ssl) {
 | |
|             $server = 'ssl://'.$this->server;
 | |
|         }
 | |
|         else {
 | |
|             $server = $this->server;
 | |
|         }
 | |
| 
 | |
|         $this->socket['irc'] = fsockopen($server, $this->port);
 | |
|         if (!is_resource($this->socket['irc'])) {
 | |
|             return $this->connect($try + 1);
 | |
|         }
 | |
|         else {
 | |
|             stream_set_blocking($this->socket['irc'], false);
 | |
|             return true;
 | |
|         }
 | |
| 
 | |
|     }//end connect()
 | |
| 
 | |
| 
 | |
|     private function doAuth() {
 | |
|         if ($this->irc_raw('USER '.$this->nick.' 0 '.$this->nick.' :'.$this->nick) && $this->irc_raw('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->irc_raw('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)) {
 | |
|             if (($this->sql = mysql_connect($this->config['db_host'], $this->config['db_user'], $this->config['db_pass'])) != false && mysql_select_db($this->config['db_name'])) {
 | |
|                 return true;
 | |
|             }
 | |
|             else {
 | |
|                 $this->log('Cannot connect to MySQL');
 | |
|                 return die();
 | |
|             }
 | |
|         }
 | |
|         else {
 | |
|             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 get_user() {
 | |
|         return $this->authd[$this->getUser($this->data)];
 | |
| 
 | |
|     }//end get_user()
 | |
| 
 | |
| 
 | |
|     private function irc_raw($params) {
 | |
|         return fputs($this->socket['irc'], $params."\r\n");
 | |
| 
 | |
|     }//end irc_raw()
 | |
| 
 | |
| 
 | |
|     private function _auth($params) {
 | |
|         $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                  = dbFetchRow('SELECT level FROM users WHERE user_id = ?', array($this->user['id']));
 | |
|                 $this->user['level']  = $tmp['level'];
 | |
|                 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 = dbFetchRow('SELECT `user_id`,`username`,`email` FROM `users` WHERE `username` = ?', array(mres($params[0])));
 | |
|             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();
 | |
|             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) {
 | |
|             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) {
 | |
|         return $this->respond($this->config['project_name_version'].', PHP: '.PHP_VERSION);
 | |
| 
 | |
|     }//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']);
 | |
|         $bps_out = formatRates($port['ifOutOctets_rate']);
 | |
|         $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 = array_pop(dbFetchRow('SELECT count(*) FROM devices'.$d_w));
 | |
|             $devup    = array_pop(dbFetchRow("SELECT count(*) FROM devices  WHERE status = '1' AND `ignore` = '0'".$d_a));
 | |
|             $devdown  = array_pop(dbFetchRow("SELECT count(*) FROM devices WHERE status = '0' AND `ignore` = '0'".$d_a));
 | |
|             $devign   = array_pop(dbFetchRow("SELECT count(*) FROM devices WHERE `ignore` = '1'".$d_a));
 | |
|             $devdis   = array_pop(dbFetchRow("SELECT count(*) FROM devices WHERE `disabled` = '1'".$d_a));
 | |
|             $msg      = 'Devices: '.$devcount.' ('.$devup.' up, '.$devdown.' down, '.$devign.' ignored, '.$devdis.' disabled'.')';
 | |
|             break;
 | |
| 
 | |
|         case 'ports':
 | |
|         case 'port':
 | |
|         case 'prt':
 | |
|             $prtcount = array_pop(dbFetchRow('SELECT count(*) FROM ports'.$p_w));
 | |
|             $prtup    = array_pop(dbFetchRow("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  = array_pop(dbFetchRow("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   = array_pop(dbFetchRow("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   = array_pop(dbFetchRow("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   = array_pop(dbFetchRow("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));
 | |
|             $msg      = 'Ports: '.$prtcount.' ('.$prtup.' up, '.$prtdown.' down, '.$prtign.' ignored, '.$prtsht.' shutdown'.')';
 | |
|             break;
 | |
| 
 | |
|         case 'services':
 | |
|         case 'service':
 | |
|         case 'srv':
 | |
|             $srvcount = array_pop(dbFetchRow('SELECT count(service_id) FROM services'.$d_w));
 | |
|             $srvup    = array_pop(dbFetchRow("SELECT count(service_id) FROM services  WHERE service_status = '1' AND service_ignore ='0'".$d_a));
 | |
|             $srvdown  = array_pop(dbFetchRow("SELECT count(service_id) FROM services WHERE service_status = '0' AND service_ignore = '0'".$d_a));
 | |
|             $srvign   = array_pop(dbFetchRow("SELECT count(service_id) FROM services WHERE service_ignore = '1'".$d_a));
 | |
|             $srvdis   = array_pop(dbFetchRow("SELECT count(service_id) FROM services WHERE service_disabled = '1'".$d_a));
 | |
|             $msg      = 'Services: '.$srvcount.' ('.$srvup.' up, '.$srvdown.' down, '.$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()
 | |
| 
 | |
| 
 | |
| }//end class
 | |
| 
 | |
| $irc = new ircbot();
 |