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'] * 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 = 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();
 |