patrickpr / trapdirector

application/controllers/HelperController.php

<?php

namespace Icinga\Module\Trapdirector\Controllers;

//use Icinga\Web\Controller;
//use Icinga\Web\Url;
use Exception;
use Icinga\Module\Trapdirector\TrapsController;
use Trapdirector\Trap;

class HelperController extends TrapsController
{
	
	/** Get host list with filter (IP or name) : host=<filter>
	*	returns in JSON : status=>OK/NOK  hosts=>array of hosts
	*/
	public function gethostsAction()
	{
		$postData=$this->getRequest()->getPost();
		
		$hostFilter = $this->checkPostVar($postData, 'hostFilter', '.*');
		
		$retHosts=array('status'=>'OK','hosts' => array());
		$this->getIdoConn(); // set apiMode to correct val
        if ($this->apiMode === TRUE)
        {
          $hosts=$this->getIdoConn()->getHostByNameOrIP($hostFilter);

The method getHostByNameOrIP() does not exist on Icinga\Module\Trapdirector\TrapsActions\UIDatabase. Did you maybe mean getHostByName()?

          $retHosts['test']=count($hosts);
        }
        else 
        {
		  $hosts=$this->getIdoConn()->getHostByIP($hostFilter);
        }
		foreach ($hosts as $val)
		{
			array_push($retHosts['hosts'],$val->name);
		}
		
		$this->_helper->json($retHosts);
	}
	
	/** Get hostgroup list with filter (name) : hostgroup=<hostFilter>
	*	returns in JSON : status=>OK/NOK  hosts=>array of hosts
	*/
	public function gethostgroupsAction()
	{
		$postData=$this->getRequest()->getPost();
		
		$hostFilter = $this->checkPostVar($postData, 'hostFilter', '.*');
		
		$retHosts=array('status'=>'OK','hosts' => array());

		$hosts=$this->getIdoConn()->getHostGroupByName($hostFilter);
		foreach ($hosts as $val)
		{
			array_push($retHosts['hosts'],$val->name);
		}
		
		$this->_helper->json($retHosts);
	}
	
	/** Get service list by host name ( host=<host> )
	*	returns in JSON : 
	*		status=>OK/No services found/More than one host matches
	*		services=>array of services (name)
	*		hostid = host object id or -1 if not found.
	*/
	public function getservicesAction()
	{
		$postData=$this->getRequest()->getPost();
		
		$host=$this->checkPostVar($postData, 'host', '.*');
		if (isset($postData['host']))
		{
			$host=$postData['host'];
		}
		else
		{
			$this->_helper->json(array('status'=>'No Hosts','hostid' => -1));
			return;
		}
		
		$hostArray=$this->getIdoConn()->getHostByName($host);
		if (count($hostArray) > 1)
		{	
			$this->_helper->json(array('status'=>'More than one host matches','hostid' => -1));
			return;
		}
		else if (count($hostArray) == 0)
		{
			$this->_helper->json(array('status'=>'No host matches','hostid' => -1));
			return;
		}
		$services=$this->getIdoConn()->getServicesByHostid($hostArray[0]->id);
		if (count($services) < 1)
		{
			$this->_helper->json(array('status'=>'No services found for host','hostid' => $hostArray[0]->id));
			return;
		}
		$retServices=array('status'=>'OK','services' => array(),'hostid' => $hostArray[0]->id);
		foreach ($services as $val)
		{
			array_push($retServices['services'],array($val->id , $val->name));
		}
		$this->_helper->json($retServices);
	}
	
	/** Get service list by host group ( name=<host> )
	*	returns in JSON : 
	*		status=>OK/No services found/More than one host matches
	*		services=>array of services (name)
	*		groupid = group object id or -1 if not found.
	*/
	public function gethostgroupservicesAction()
	{
		$postData=$this->getRequest()->getPost();
		
		$host = $this->checkPostVar($postData, 'host', '.+');
		
		$hostArray=$this->getIdoConn()->getHostGroupByName($host);
		if (count($hostArray) > 1)
		{	
			$this->_helper->json(array('status'=>'More than one hostgroup matches','hostid' => -1));
			return;
		}
		else if (count($hostArray) == 0)
		{
			$this->_helper->json(array('status'=>'No hostgroup matches','hostid' => -1));
			return;
		}
		$services=$this->getIdoConn()->getServicesByHostGroupid($hostArray[0]->id);
		if (count($services) < 1)
		{
			$this->_helper->json(array('status'=>'No services found for hostgroup','hostid' => $hostArray[0]->id));
			return;
		}
		$retServices=array('status'=>'OK','services' => $services,'hostid' => $hostArray[0]->id);
		
		$this->_helper->json($retServices);
	}

	/** Get traps from mib  : entry : mib=<mib>
	*	returns in JSON : 
	*		status=>OK/No mib/Error getting mibs
	*		traps=>array of array( oid -> name)
	*/
	public function gettrapsAction()
	{
		$postData=$this->getRequest()->getPost();
		
		$mib = $this->checkPostVar($postData, 'mib', '.*');

		try
		{
			$traplist=$this->getMIB()->getTrapList($mib);
			$retTraps=array('status'=>'OK','traps' => $traplist);
		} 
		catch (Exception $e) 
		{ 
			$retTraps=array('status' => 'Error getting mibs');
		}
		$this->_helper->json($retTraps);
	}	

	/** Get trap objects from mib  : entry : trap=<oid>
	*	returns in JSON : 
	*		status=>OK/no trap/not found
	*		objects=>array of array( oid -> name, oid->mib)
	*/
	public function gettrapobjectsAction()
	{
		$postData=$this->getRequest()->getPost();
		
		$trap = $this->checkPostVar($postData, 'trap', '.*');
		
		try
		{
			$objectlist=$this->getMIB()->getObjectList($trap);
			$retObjects=array('status'=>'OK','objects' => $objectlist);
		} 
		catch (Exception $e) 
		{ 
			$retObjects=array('status' => 'not found');
		}
		$this->_helper->json($retObjects);
	}	
	
	/** Get list of all loaded mibs : entry : none
	*	return : array of strings.
	*/
	public function getmiblistAction()
	{
		try
		{
			$miblist=$this->getMIB()->getMIBList();
		} 
		catch (Exception $e) 
		{ 
			$miblist=array('Error getting mibs');
		}
		$this->_helper->json($miblist);
	}
	
	/** Get MIB::Name from OID : entry : oid
	*		status=>OK/No oid/not found
	*		mib=>string
	*		name=>string
	*/	
	public function translateoidAction()
	{
		$postData=$this->getRequest()->getPost();
		
		$oid = $this->checkPostVar($postData, 'oid', '.*');
		
		// Try to get oid name from snmptranslate
		if (($object=$this->getMIB()->translateOID($oid)) == null)
		{
			$this->_helper->json(array('status'=>'Not found'));
			return;
		}
		else
		{
			$this->_helper->json(
				array('status'=>'OK',
					'mib' => $object['mib'], 
					'name' => $object['name'],
					'type' => $object['type'],
					'type_enum' => $object['type_enum'],
				    'description' => $object['description']
				)
			);
		}

	}
	
	/** Save or execute database purge of <n> days
	*	days=>int 
	*	action=>save/execute
	*	return : status=>OK/Message error
	*/
	public function dbmaintenanceAction()
	{
		
		$postData=$this->getRequest()->getPost();
		
		$days = $this->checkPostVar($postData, 'days', '^[0-9]+$');
		$days=intval($days);

		$action = $this->checkPostVar($postData, 'action', 'save|execute');
		
		if ($action == 'save')
		{
			try
			{
			    $this->getUIDatabase()->setDBConfigValue('db_remove_days',$days);
			}
			catch (Exception $e)
			{
				$this->_helper->json(array('status'=>'Save error : '.$e->getMessage() ));
				return;
			}
			$this->_helper->json(array('status'=>'OK'));
			return;
		}
		if ($action == 'execute')
		{
			try
			{
				require_once($this->Module()->getBaseDir() .'/bin/trap_class.php');
				$icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
				$debug_level=4;
				$trap = new Trap($icingaweb2_etc);
				$trap->setLogging($debug_level,'syslog');
				$trap->eraseOldTraps($days);
			}
			catch (Exception $e)
			{
				$this->_helper->json(array('status'=>'execute error : '.$e->getMessage() ));
				return;
			}			
			$this->_helper->json(array('status'=>'OK'));
		}
			
	}	

	/** Save snmp configuration to db
	 *	use=>1 (true) or 0
	 *	oid=>specific oid or '' for default
	 *	return : status=>OK/Message error
	 */
	public function snmpconfigAction()
	{
	    $postData=$this->getRequest()->getPost();
	    
	    $snmpUse = $this->checkPostVar($postData, 'useTrapAddr', '0|1');
	    
	    $snmpOID = $this->checkPostVar($postData, 'trapAddrOID', '^[\.0-9]+$');
	    	    
	    try
	    {
	        $this->getUIDatabase()->setDBConfigValue('use_SnmpTrapAddess',$snmpUse);
	        $this->getUIDatabase()->setDBConfigValue('SnmpTrapAddess_oid',$snmpOID);
	    }
	    catch (Exception $e)
	    {
	        $this->_helper->json(array('status'=>'Save error : '.$e->getMessage() ));
	        return;
	    }
	    $this->_helper->json(array('status'=>'OK'));
	    return;
	    
	}
	
	/** Save log output to db
	*	destination=>log destination 
	*	file=>file name
	*	level => int 
	*	return : status=>OK/Message error
	*/
	public function logdestinationAction()
	{
		$postData=$this->getRequest()->getPost();
		
		$destination = $this->checkPostVar($postData, 'destination', '.*');
		$logDest=$this->getModuleConfig()->getLogDestinations();
		if (!isset($logDest[$destination]))
		{
			$this->_helper->json(array('status'=>'invalid destination : '.$destination));
			return;
		}

		if (isset($postData['file']))
		{ 
			$file=$postData['file'];
			$fileHandler=@fopen($file,'w');
			if ($fileHandler == false)
			{   // File os note writabe / cannot create
			    $this->_helper->json(array('status'=>'File not writable :  '.$file));
			    return;
			}
		}
		else
		{
			if ($destination != 'file')
			{
				$file=null;
			}
			else
			{
				$this->_helper->json(array('status'=>'No file'));
				return;
			}
		}

		$level = $this->checkPostVar($postData, 'level', '[0-9]');
				
		try
		{
		    $this->getUIDatabase()->setDBConfigValue('log_destination',$destination);
		    $this->getUIDatabase()->setDBConfigValue('log_file',$file);
		    $this->getUIDatabase()->setDBConfigValue('log_level',$level);
		}
		catch (Exception $e)
		{
			$this->_helper->json(array('status'=>'Save error : '.$e->getMessage() ));
			return;
		}
		$this->_helper->json(array('status'=>'OK'));
		return;
			
	}	
	
	/** Test a rule evaluation
	 *	rule=>rule to evaluate
	 *	action=>'evaluate'
	 *	return : status=>OK/Message error & message : return of evaluation
	 */
	public function testruleAction()
	{
	    
	    $postData=$this->getRequest()->getPost();
	   
	    $rule = $this->checkPostVar($postData, 'rule', '.*');

	    $action = $this->checkPostVar($postData, 'action', 'evaluate');

	    if ($action == 'evaluate')
	    {
	        try
	        {
	            require_once($this->Module()->getBaseDir() .'/bin/trap_class.php');
	            $icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
	            $trap = new Trap($icingaweb2_etc);
	            // Cleanup spaces before eval
	            $rule=$trap->ruleClass->eval_cleanup($rule);
	            // Eval
	            $item=0;
	            $rule=$trap->ruleClass->evaluation($rule,$item);
	        }
	        catch (Exception $e)
	        {
	            $this->_helper->json(array('status'=>'Evaluation error : '.$e->getMessage() ));
	            return;
	        }
	        $return=($rule==true)?'true':'false';
	        $this->_helper->json(array('status'=>'OK', 'message' => $return));
	    }
	    
	}	

	/** Test a rule evaluation
	 *	name    => name of plugin
	 *	action  => enable | disable
	 *	return : status=>OK/Message error
	 */
	public function pluginAction()
	{
	    $postData=$this->getRequest()->getPost();
	    
	    $pluginName = $this->checkPostVar($postData, 'name', '.*');
	    
	    $action = $this->checkPostVar($postData, 'action', 'enable|disable');
	    
        try
        {
            require_once($this->Module()->getBaseDir() .'/bin/trap_class.php');
            $icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
            $trap = new Trap($icingaweb2_etc);
            // Enable plugin.
            $action=($action == 'enable') ? true : false;
            $retVal=$trap->pluginClass->enablePlugin($pluginName, $action);
            
        }
        catch (Exception $e)
        {
            $this->_helper->json(array('status'=>'Action error : '.$e->getMessage() ));
            return;
        }
        if ($retVal === true)
        {
            $this->_helper->json(array('status'=>'OK'));
        }
        else
        {
            $this->_helper->json(array('status'=>'Error, see logs'));
        }
	}
	
	/** Function evaluation
	 *	function    => name of function
	 *	action      => evaluate
	 *	return : status=>OK/Message error & message : return of evaluation ('true' or 'false' )
	 */
	public function functionAction()
	{
	    $postData=$this->getRequest()->getPost();
	    
	    $functionString = $this->checkPostVar($postData, 'function', '.*');
	    
	    $this->checkPostVar($postData, 'action', 'evaluate');
	    
	    // Only one action possible for now, no tests on action.
	    try
	    {
	        require_once($this->Module()->getBaseDir() .'/bin/trap_class.php');
	        $icingaweb2Etc=$this->Config()->get('config', 'icingaweb2_etc');
	        $trap = new Trap($icingaweb2Etc);
	        // load all plugins in case tested function is not enabled.
	        $trap->pluginClass->registerAllPlugins(false);
	        // Clean all spaces
	        $functionString = $trap->ruleClass->eval_cleanup($functionString);
	        // Eval functions
	        $result = $trap->pluginClass->evaluateFunctionString($functionString);	        
	    }
	    catch (Exception $e)
	    {
	        $this->_helper->json(array('status'=>'Action error : '.$e->getMessage() ));
	        return;
	    }
	    
        $result = ($result === true)?'True':'False';
        $this->_helper->json(array('status'=>'OK','message' => $result));
	}

    /**************   Utilities **********************/

	private function checkPostVar(array $postData,string $postVar, string $validRegexp) : string
	{
	    if (!isset ($postData[$postVar]))
	    {
	        $this->_helper->json(array('status'=>'No ' . $postVar));
	        return '';
	    }
	    if (preg_match('/'.$validRegexp.'/', $postData[$postVar]) != 1)
	    {
	        $this->_helper->json(array('status'=>'Unknown ' . $postVar . ' value '.$postData[$postVar]));
	        return '';
	    }
	    return $postData[$postVar];
	}
}