patrickpr / trapdirector

library/Trapdirector/TrapsController.php

Indentation +103 added lines, -103 removed lines

@@ -46,8 +46,8 @@ 
 	
 	
 	/** Get instance of TrapModuleConfig class
-	*	@return TrapModuleConfig
-	*/
+	 *	@return TrapModuleConfig
+	 */
 	public function getModuleConfig() 
 	{
 		if ($this->moduleConfig == Null) 
@@ -79,12 +79,12 @@ 
 	 */
 	public function getTrapHostListTable()
 	{
-	    if ($this->trapTableHostList == Null) 
+		if ($this->trapTableHostList == Null) 
 		{
-	        $this->trapTableHostList = new TrapTableHostList();
-	        $this->trapTableHostList->setConfig($this->getModuleConfig());
-	    }
-	    return $this->trapTableHostList;
+			$this->trapTableHostList = new TrapTableHostList();
+			$this->trapTableHostList->setConfig($this->getModuleConfig());
+		}
+		return $this->trapTableHostList;
 	}
 	
 	/**
@@ -105,12 +105,12 @@ 
 	 */
 	public function getUIDatabase()
 	{
-	    if ($this->UIDatabase == Null)
-	    {
-	        $this->UIDatabase = new UIDatabase($this);
+		if ($this->UIDatabase == Null)
+		{
+			$this->UIDatabase = new UIDatabase($this);
 	       
-	    }
-	    return $this->UIDatabase;
+		}
+		return $this->UIDatabase;
 	}
 
 	/**
@@ -119,34 +119,34 @@ 
 	 */
 	public function getIdoConn()
 	{
-	    if ($this->icingaAPI === NULL)
-	    {
-    	    $host = $this->Config()->get('config', 'icingaAPI_host');
-    	    $port = $this->Config()->get('config', 'icingaAPI_port');
-    	    $user = $this->Config()->get('config', 'icingaAPI_user');
-    	    $pass = $this->Config()->get('config', 'icingaAPI_password');
-    	    $this->icingaAPI = new Icinga2API($host,$port);
-    	    $this->icingaAPI->setCredentials($user, $pass);
-    	    list($ret,$message) = $this->icingaAPI->test(array());
-    	    if ($ret === TRUE)
-    	    {
-    	        return $this->getUIDatabase();
-    	    }
+		if ($this->icingaAPI === NULL)
+		{
+			$host = $this->Config()->get('config', 'icingaAPI_host');
+			$port = $this->Config()->get('config', 'icingaAPI_port');
+			$user = $this->Config()->get('config', 'icingaAPI_user');
+			$pass = $this->Config()->get('config', 'icingaAPI_password');
+			$this->icingaAPI = new Icinga2API($host,$port);
+			$this->icingaAPI->setCredentials($user, $pass);
+			list($ret,$message) = $this->icingaAPI->test(array());
+			if ($ret === TRUE)
+			{
+				return $this->getUIDatabase();
+			}
     	    
-	    }
-	    return $this->icingaAPI;
+		}
+		return $this->icingaAPI;
 	    
 	}
 	
-    protected function applyPaginationLimits(Paginatable $paginatable, $limit = 25, $offset = null)
-    {
-        $limit = $this->params->get('limit', $limit);
-        $page = $this->params->get('page', $offset);
+	protected function applyPaginationLimits(Paginatable $paginatable, $limit = 25, $offset = null)
+	{
+		$limit = $this->params->get('limit', $limit);
+		$page = $this->params->get('page', $offset);
 
-        $paginatable->limit($limit, $page > 0 ? ($page - 1) * $limit : 0);
+		$paginatable->limit($limit, $page > 0 ? ($page - 1) * $limit : 0);
 
-        return $paginatable;
-    }	
+		return $paginatable;
+	}	
 	
 	public function displayExitError($source,$message)
 	{	// TODO : check better ways to transmit data (with POST ?)
@@ -155,33 +155,33 @@ 
 	
 	protected function checkReadPermission()
 	{
-        if (! $this->Auth()->hasPermission('trapdirector/view')) {
-            $this->displayExitError('Permissions','No permission fo view content');
-        }		
+		if (! $this->Auth()->hasPermission('trapdirector/view')) {
+			$this->displayExitError('Permissions','No permission fo view content');
+		}		
 	}
 
 	protected function checkConfigPermission()
 	{
-        if (! $this->Auth()->hasPermission('trapdirector/config')) {
-            $this->displayExitError('Permissions','No permission fo configure');
-        }		
+		if (! $this->Auth()->hasPermission('trapdirector/config')) {
+			$this->displayExitError('Permissions','No permission fo configure');
+		}		
 	}
 	
-    /**
-     * Check if user has write permission
-     * @param number $check optional : if set to 1, return true (user has permission) or false instead of displaying error page
-     * @return boolean : user has permission
-     */
+	/**
+	 * Check if user has write permission
+	 * @param number $check optional : if set to 1, return true (user has permission) or false instead of displaying error page
+	 * @return boolean : user has permission
+	 */
 	protected function checkModuleConfigPermission($check=0)
 	{
-        if (! $this->Auth()->hasPermission('trapdirector/module_config')) {
-            if ($check == 0)
-            {
-                $this->displayExitError('Permissions','No permission fo configure module');
-            }
-            return false;
-        }
-        return true;
+		if (! $this->Auth()->hasPermission('trapdirector/module_config')) {
+			if ($check == 0)
+			{
+				$this->displayExitError('Permissions','No permission fo configure module');
+			}
+			return false;
+		}
+		return true;
 	}
 
 	/*************************  Trap class get **********************/
@@ -201,18 +201,18 @@ 
 	/************************** MIB related **************************/
 	
 	/** Get MIBLoader class
-	*	@return MIBLoader class
-	*/
+	 *	@return MIBLoader class
+	 */
 	protected function getMIB()
 	{
 		if ($this->MIBData == null)
 		{
-		    $dbConn = $this->getUIDatabase()->getDbConn();
-		    if ($dbConn === null) throw new \ErrorException('uncatched db error');
+			$dbConn = $this->getUIDatabase()->getDbConn();
+			if ($dbConn === null) throw new \ErrorException('uncatched db error');
 			$this->MIBData=new MIBLoader(
 				$this->Config()->get('config', 'snmptranslate'),
 				$this->Config()->get('config', 'snmptranslate_dirs'),
-			    $dbConn,
+				$dbConn,
 				$this->getModuleConfig()
 			);
 		}
@@ -222,13 +222,13 @@ 
 	/**************************  Database queries *******************/		
 	
 	/** Check if director is installed
-	*	@return bool true/false
-	*/
+	 *	@return bool true/false
+	 */
 	protected function isDirectorInstalled()
 	{
-	    $output=array();
-	    exec('icingacli module list',$output);
-	    foreach ($output as $line)
+		$output=array();
+		exec('icingacli module list',$output);
+		foreach ($output as $line)
 		{
 			if (preg_match('/^director .*enabled/',$line))
 			{
@@ -241,72 +241,72 @@ 
 
 	/************************ UI elements **************************/
 	
-    /**
-     * get max rows to display before paging.
-     * @return number
-     */
+	/**
+	 * get max rows to display before paging.
+	 * @return number
+	 */
 	public function itemListDisplay()
 	{
-	    return $this->getUIDatabase()->getDBConfigValue('max_rows_in_list');
+		return $this->getUIDatabase()->getDBConfigValue('max_rows_in_list');
 	}
 
 	public function setitemListDisplay(int $maxRows)
 	{
-	    return $this->getUIDatabase()->setDBConfigValue('max_rows_in_list',$maxRows);
+		return $this->getUIDatabase()->setDBConfigValue('max_rows_in_list',$maxRows);
 	}
 	
-    /**
-     * get Handlers categories list (index => textvalue).
-     * @return array
-     */	
+	/**
+	 * get Handlers categories list (index => textvalue).
+	 * @return array
+	 */	
 	public function getHandlersCategory()
 	{
-	    //<index>:<name>!<index>:<name>
-	    $catList = $this->getUIDatabase()->getDBConfigValue('handler_categories');
-	    $catListArray=explode('!',$catList);
-	    $retArray=array();
-	    foreach ($catListArray as $category)
-	    {
-	        $catArray=explode(':',$category);
-	        $retArray[$catArray[0]] = $catArray[1];
-	    }
-	    return $retArray; 
+		//<index>:<name>!<index>:<name>
+		$catList = $this->getUIDatabase()->getDBConfigValue('handler_categories');
+		$catListArray=explode('!',$catList);
+		$retArray=array();
+		foreach ($catListArray as $category)
+		{
+			$catArray=explode(':',$category);
+			$retArray[$catArray[0]] = $catArray[1];
+		}
+		return $retArray; 
 	}
 
 	public function setHandlerCategory(array $catArray)
 	{
-	    $catString='';
-	    foreach ($catArray as $index => $value)
-	    {
-	        if ($catString != '' ) $catString .= '!';
-	        $catString .= $index . ':' . $value;
-	    }
-	    $this->getUIDatabase()->setDBConfigValue('handler_categories', $catString);
+		$catString='';
+		foreach ($catArray as $index => $value)
+		{
+			if ($catString != '' ) $catString .= '!';
+			$catString .= $index . ':' . $value;
+		}
+		$this->getUIDatabase()->setDBConfigValue('handler_categories', $catString);
 	}
 	
 	public function addHandlersCategory(string $catName)
 	{
-	    $catArray = $this->getHandlersCategory();
-	    $i=1;
-	    while (isset($catArray[$i]) && $i < 100) $i++;
-	    if ($i == 100) throw new ProgrammingError('Category array error');
-	    $catArray[$i] = $catName;
-	    $this->setHandlerCategory($catArray);
+		$catArray = $this->getHandlersCategory();
+		$i=1;
+		while (isset($catArray[$i]) && $i < 100) $i++;
+		if ($i == 100) throw new ProgrammingError('Category array error');
+		$catArray[$i] = $catName;
+		$this->setHandlerCategory($catArray);
 	}
 	
 	public function delHandlersCategory(int $catIndex)
 	{
-	    $catArray = $this->getHandlersCategory();
-	    unset($catArray[$catIndex]);
-	    $this->setHandlerCategory($catArray);
-	    $this->getUIDatabase()->updateHandlersOnCategoryDelete($catIndex);
+		$catArray = $this->getHandlersCategory();
+		unset($catArray[$catIndex]);
+		$this->setHandlerCategory($catArray);
+		$this->getUIDatabase()->updateHandlersOnCategoryDelete($catIndex);
 	}
 	
 	public function renameHandlersCategory(int $catIndex, string $catName)
 	{
-	    $catArray = $this->getHandlersCategory();
-	    $catArray[$catIndex] = $catName;
-	    $this->setHandlerCategory($catArray);
+		$catArray = $this->getHandlersCategory();
+		$catArray[$catIndex] = $catName;
+		$this->setHandlerCategory($catArray);
 	}
 	
 }

library/Trapdirector/Icinga2Api.php

Indentation +132 added lines, -132 removed lines

@@ -10,151 +10,151 @@
 class Icinga2API extends IcingaApiBase
 {
     
-    /**
-     * Creates Icinga2API object
-     * 
-     * @param string $host host name or IP
-     * @param number $port API port
-     */
-    public function __construct($host, $port = 5665)
-    {
-        parent::__construct($host,$port);
-    }
-    /**
+	/**
+	 * Creates Icinga2API object
+	 * 
+	 * @param string $host host name or IP
+	 * @param number $port API port
+	 */
+	public function __construct($host, $port = 5665)
+	{
+		parent::__construct($host,$port);
+	}
+	/**
 
 /************ Host query ************/  
   
-    /**
-     * return array of host by IP (4 or 6) or name
-     * @param string $ip
-     * @throws Exception
-     * @return array objects : array('__name','name','display_name')
-     */
-    public function getHostByIP(string $ip) 
-    {
-        return $this->standardQuery(
-            'host', 
-            'match("*' . $ip . '*",host.address) || match("*' . $ip . '*",host.address6) || match("*' . $ip . '*",host.name) || match("*' . $ip . '*",host.display_name)',
-            array('__name','name','display_name')
-        );
-    }
+	/**
+	 * return array of host by IP (4 or 6) or name
+	 * @param string $ip
+	 * @throws Exception
+	 * @return array objects : array('__name','name','display_name')
+	 */
+	public function getHostByIP(string $ip) 
+	{
+		return $this->standardQuery(
+			'host', 
+			'match("*' . $ip . '*",host.address) || match("*' . $ip . '*",host.address6) || match("*' . $ip . '*",host.name) || match("*' . $ip . '*",host.display_name)',
+			array('__name','name','display_name')
+		);
+	}
 
     
-    /**
-     * Get host(s) by name in API
-     * @param string $name
-     * @return array|NULL[] : see getHostByIP
-     */
-    public function getHostByName(string $name)
-    {
-        return $this->getHostByIP($name);
-    }
+	/**
+	 * Get host(s) by name in API
+	 * @param string $name
+	 * @return array|NULL[] : see getHostByIP
+	 */
+	public function getHostByName(string $name)
+	{
+		return $this->getHostByIP($name);
+	}
  
-    /**
-     * Get all host and IP from hostgroup
-     * @param string $hostGroup
-     * @throws Exception
-     * @return array : attributes : address, address6, name
-     */
-    public function getHostsIPByHostGroup($hostGroup)
-    {        
-        return $this->standardQuery(
-            'host', 
-            '"' . $hostGroup . '" in host.groups',
-            array('address','address6','name')
+	/**
+	 * Get all host and IP from hostgroup
+	 * @param string $hostGroup
+	 * @throws Exception
+	 * @return array : attributes : address, address6, name
+	 */
+	public function getHostsIPByHostGroup($hostGroup)
+	{        
+		return $this->standardQuery(
+			'host', 
+			'"' . $hostGroup . '" in host.groups',
+			array('address','address6','name')
                 
-        );
-    }
+		);
+	}
     
-    /**
-     * Get all host and IP from hostgroup
-     * @param string $hostGroup
-     * @throws Exception
-     * @return array : attributes : address, address6, name
-     */
-    public function getHostGroupByName($name)
-    {
-        return $this->standardQuery(
-            'hostgroup',
-            'match("*' . $name . '*",hostgroup.name)',
-            array('name','display_name')
+	/**
+	 * Get all host and IP from hostgroup
+	 * @param string $hostGroup
+	 * @throws Exception
+	 * @return array : attributes : address, address6, name
+	 */
+	public function getHostGroupByName($name)
+	{
+		return $this->standardQuery(
+			'hostgroup',
+			'match("*' . $name . '*",hostgroup.name)',
+			array('name','display_name')
             
-            );
-    }
+			);
+	}
 
-    /** Get services from host in API
-     *	
-     *  @throws Exception
-     *	@param $id string host name
-     *	@return array display_name (of service), service_object_id
-     */
-    public function getServicesByHostid(string $id, bool $active = TRUE, bool $passive_svc = TRUE)
-    {
-        $filter = 'match("' . $id . '!*", service.__name)';
-        if ($active === TRUE)
-        {
-            $filter .= ' && service.active==true';
-        }
-        if ($passive_svc === TRUE)
-        {
-            $filter .= ' && service.enable_passive_checks==true';
-        }
-        return $this->standardQuery(
-            'service',
-            $filter,
-            array('__name','name','display_name','active')
-            );
+	/** Get services from host in API
+	 *	
+	 *  @throws Exception
+	 *	@param $id string host name
+	 *	@return array display_name (of service), service_object_id
+	 */
+	public function getServicesByHostid(string $id, bool $active = TRUE, bool $passive_svc = TRUE)
+	{
+		$filter = 'match("' . $id . '!*", service.__name)';
+		if ($active === TRUE)
+		{
+			$filter .= ' && service.active==true';
+		}
+		if ($passive_svc === TRUE)
+		{
+			$filter .= ' && service.enable_passive_checks==true';
+		}
+		return $this->standardQuery(
+			'service',
+			$filter,
+			array('__name','name','display_name','active')
+			);
         
-    }
+	}
 
-    /**
-     * return array of host by IP (4 or 6) or name
-     * @param string $group Host group name
-     * @throws Exception
-     * @return array objects : array('name','display_name')
-     */
-    public function getHostsByGroup(string $group)
-    {
-         return $this->standardQuery(
-            'host',
-            '"' . $group . '" in host.groups',
-            array('name','display_name')
-            );
-    }
+	/**
+	 * return array of host by IP (4 or 6) or name
+	 * @param string $group Host group name
+	 * @throws Exception
+	 * @return array objects : array('name','display_name')
+	 */
+	public function getHostsByGroup(string $group)
+	{
+		 return $this->standardQuery(
+			'host',
+			'"' . $group . '" in host.groups',
+			array('name','display_name')
+			);
+	}
     
-    public function getServicesByHostGroupid(string $group)
-    {
-        $hostList = $this->getHostsByGroup($group);
-        //return $hostList;
-        $hostNum = count($hostList);
-        $serviceList=array();
-        foreach ($hostList as $curHost)
-        {
-            $services = $this->getServicesByHostid($curHost->name);
-            foreach ($services as $service)
-            {
-                //return $service;
-                if (! isset($serviceList[$service->name]))
-                {
-                    $serviceList[$service->name]=
-                        array('num'=> 1 ,'display_name' => $service->display_name);
-                }
-                else
-                {
-                    $serviceList[$service->name]['num']++;
-                }
-            }
-        }
-        $commonServices=array();
-        foreach ($serviceList as $serviceName => $values)
-        {
-            if ($values['num'] >= $hostNum)
-            {
-                $commonServices[$serviceName] = $values['display_name'];
-            }
-        }
-        return $commonServices;
-    }
+	public function getServicesByHostGroupid(string $group)
+	{
+		$hostList = $this->getHostsByGroup($group);
+		//return $hostList;
+		$hostNum = count($hostList);
+		$serviceList=array();
+		foreach ($hostList as $curHost)
+		{
+			$services = $this->getServicesByHostid($curHost->name);
+			foreach ($services as $service)
+			{
+				//return $service;
+				if (! isset($serviceList[$service->name]))
+				{
+					$serviceList[$service->name]=
+						array('num'=> 1 ,'display_name' => $service->display_name);
+				}
+				else
+				{
+					$serviceList[$service->name]['num']++;
+				}
+			}
+		}
+		$commonServices=array();
+		foreach ($serviceList as $serviceName => $values)
+		{
+			if ($values['num'] >= $hostNum)
+			{
+				$commonServices[$serviceName] = $values['display_name'];
+			}
+		}
+		return $commonServices;
+	}
     
 }
 

library/Trapdirector/IcingaApi/IcingaApiBase.php

Indentation +308 added lines, -308 removed lines

@@ -8,342 +8,342 @@
 
 class IcingaApiBase
 {
-    protected $version = 'v1';      //< icinga2 api version
+	protected $version = 'v1';      //< icinga2 api version
     
-    protected $host;                //< icinga2 host name or IP
-    protected $port;                //< icinga2 api port
+	protected $host;                //< icinga2 host name or IP
+	protected $port;                //< icinga2 api port
     
-    protected $user;                //< user name
-    protected $pass;                //< user password
-    protected $usercert;            //< user key for certificate auth (NOT IMPLEMENTED)
-    protected $authmethod='pass';   //< Authentication : 'pass' or 'cert'
+	protected $user;                //< user name
+	protected $pass;                //< user password
+	protected $usercert;            //< user key for certificate auth (NOT IMPLEMENTED)
+	protected $authmethod='pass';   //< Authentication : 'pass' or 'cert'
 
-    protected $queryURL=array(
-        'host'  => 'objects/hosts',
-        'hostgroup' => 'objects/hostgroups',
-        'service' => 'objects/services'
-    );
+	protected $queryURL=array(
+		'host'  => 'objects/hosts',
+		'hostgroup' => 'objects/hostgroups',
+		'service' => 'objects/services'
+	);
     
-    protected $curl;
-    // http://php.net/manual/de/function.json-last-error.php#119985
-    protected $errorReference = [
-        JSON_ERROR_NONE => 'No error has occurred.',
-        JSON_ERROR_DEPTH => 'The maximum stack depth has been exceeded.',
-        JSON_ERROR_STATE_MISMATCH => 'Invalid or malformed JSON.',
-        JSON_ERROR_CTRL_CHAR => 'Control character error, possibly incorrectly encoded.',
-        JSON_ERROR_SYNTAX => 'Syntax error.',
-        JSON_ERROR_UTF8 => 'Malformed UTF-8 characters, possibly incorrectly encoded.',
-        JSON_ERROR_RECURSION => 'One or more recursive references in the value to be encoded.',
-        JSON_ERROR_INF_OR_NAN => 'One or more NAN or INF values in the value to be encoded.',
-        JSON_ERROR_UNSUPPORTED_TYPE => 'A value of a type that cannot be encoded was given.',
-    ];
-    const JSON_UNKNOWN_ERROR = 'Unknown error.';
+	protected $curl;
+	// http://php.net/manual/de/function.json-last-error.php#119985
+	protected $errorReference = [
+		JSON_ERROR_NONE => 'No error has occurred.',
+		JSON_ERROR_DEPTH => 'The maximum stack depth has been exceeded.',
+		JSON_ERROR_STATE_MISMATCH => 'Invalid or malformed JSON.',
+		JSON_ERROR_CTRL_CHAR => 'Control character error, possibly incorrectly encoded.',
+		JSON_ERROR_SYNTAX => 'Syntax error.',
+		JSON_ERROR_UTF8 => 'Malformed UTF-8 characters, possibly incorrectly encoded.',
+		JSON_ERROR_RECURSION => 'One or more recursive references in the value to be encoded.',
+		JSON_ERROR_INF_OR_NAN => 'One or more NAN or INF values in the value to be encoded.',
+		JSON_ERROR_UNSUPPORTED_TYPE => 'A value of a type that cannot be encoded was given.',
+	];
+	const JSON_UNKNOWN_ERROR = 'Unknown error.';
     
-    /**
-     * Creates Icinga2API object
-     * 
-     * @param string $host host name or IP
-     * @param number $port API port
-     */
-    public function __construct($host, $port = 5665)
-    {
-        $this->host=$host;
-        $this->port=$port;
-    }
-    /**
-     * Set user & pass
-     * @param string $user
-     * @param string $pass
-     */
-    public function setCredentials($user,$pass)
-    {
-        $this->user=$user;
-        $this->pass=$pass;
-        $this->authmethod='pass';
-    }
+	/**
+	 * Creates Icinga2API object
+	 * 
+	 * @param string $host host name or IP
+	 * @param number $port API port
+	 */
+	public function __construct($host, $port = 5665)
+	{
+		$this->host=$host;
+		$this->port=$port;
+	}
+	/**
+	 * Set user & pass
+	 * @param string $user
+	 * @param string $pass
+	 */
+	public function setCredentials($user,$pass)
+	{
+		$this->user=$user;
+		$this->pass=$pass;
+		$this->authmethod='pass';
+	}
     
-    /**
-     * Set user & certificate (NOT IMPLEMENTED @throws RuntimeException)
-     * @param string $user
-     * @param string $usercert
-     */
-    public function setCredentialskey($user,$usercert)
-    {
-        $this->user=$user;
-        $this->usercert=$usercert;
-        $this->authmethod='cert';
-        throw new RuntimeException('Certificate auth not implemented');
-    }
+	/**
+	 * Set user & certificate (NOT IMPLEMENTED @throws RuntimeException)
+	 * @param string $user
+	 * @param string $usercert
+	 */
+	public function setCredentialskey($user,$usercert)
+	{
+		$this->user=$user;
+		$this->usercert=$usercert;
+		$this->authmethod='cert';
+		throw new RuntimeException('Certificate auth not implemented');
+	}
 
-    public function test(array $permissions)
-    {
-       try
-        {
-            $result=$this->request('GET', "", NULL, NULL);
-        } 
-        catch (Exception $e)
-        {
-            return array(true, 'Error with API : '.$e->getMessage());
-        }
-        //var_dump($result);
-        $permOk=1;
-        $permMissing='';
-        if ($permissions === NULL || count($permissions) == 0) // If no permission check return OK after connexion
-        {
-            return array(false,'OK');
-        }
-        if (property_exists($result, 'results') && property_exists($result->results[0], 'permissions'))
-        {
+	public function test(array $permissions)
+	{
+	   try
+		{
+			$result=$this->request('GET', "", NULL, NULL);
+		} 
+		catch (Exception $e)
+		{
+			return array(true, 'Error with API : '.$e->getMessage());
+		}
+		//var_dump($result);
+		$permOk=1;
+		$permMissing='';
+		if ($permissions === NULL || count($permissions) == 0) // If no permission check return OK after connexion
+		{
+			return array(false,'OK');
+		}
+		if (property_exists($result, 'results') && property_exists($result->results[0], 'permissions'))
+		{
             
-            foreach ( $permissions as $mustPermission)
-            {
-                $curPermOK=0;
-                foreach ( $result->results[0]->permissions as $curPermission)
-                {
-                    $curPermission=preg_replace('/\*/','.*',$curPermission); // put * as .* to created a regexp
-                    if (preg_match('#'.$curPermission.'#',$mustPermission))
-                    {
-                        $curPermOK=1;
-                        break;
-                    }
-                }
-                if ($curPermOK == 0)
-                {
-                    $permOk=0;
-                    $permMissing=$mustPermission;
-                    break;
-                }
-            }
-            if ($permOk == 0)
-            {
-                return array(true,'API connection OK, but missing permission : '.$permMissing);
-            }
-            return array(false,'API connection OK');
+			foreach ( $permissions as $mustPermission)
+			{
+				$curPermOK=0;
+				foreach ( $result->results[0]->permissions as $curPermission)
+				{
+					$curPermission=preg_replace('/\*/','.*',$curPermission); // put * as .* to created a regexp
+					if (preg_match('#'.$curPermission.'#',$mustPermission))
+					{
+						$curPermOK=1;
+						break;
+					}
+				}
+				if ($curPermOK == 0)
+				{
+					$permOk=0;
+					$permMissing=$mustPermission;
+					break;
+				}
+			}
+			if ($permOk == 0)
+			{
+				return array(true,'API connection OK, but missing permission : '.$permMissing);
+			}
+			return array(false,'API connection OK');
             
-        }
-        return array(true,'API connection OK, but cannot get permissions');
-    }
+		}
+		return array(true,'API connection OK, but cannot get permissions');
+	}
     
     
-    protected function url($url) {
-        return sprintf('https://%s:%d/%s/%s', $this->host, $this->port, $this->version, $url);
-    }
+	protected function url($url) {
+		return sprintf('https://%s:%d/%s/%s', $this->host, $this->port, $this->version, $url);
+	}
     
-    /**
-     * Create or return curl ressource
-     * @throws Exception
-     * @return resource
-     */
-    protected function curl() {
-        if ($this->curl === null) {
-            $this->curl = curl_init(sprintf('https://%s:%d', $this->host, $this->port));
-            if ($this->curl === false) {
-                throw new Exception('CURL INIT ERROR');
-            }
-        }
-        return $this->curl;
-    }
+	/**
+	 * Create or return curl ressource
+	 * @throws Exception
+	 * @return resource
+	 */
+	protected function curl() {
+		if ($this->curl === null) {
+			$this->curl = curl_init(sprintf('https://%s:%d', $this->host, $this->port));
+			if ($this->curl === false) {
+				throw new Exception('CURL INIT ERROR');
+			}
+		}
+		return $this->curl;
+	}
 
-    /**
-     * Send a passive service check
-     * @param string $host : host name 
-     * @param string $service : service name
-     * @param int $state : state of service
-     * @param string $display : service passive check output
-     * @param string $perfdata : performance data as string
-     * @return array (status = true (oK) or false (nok), string message)
-     */
-    public function serviceCheckResult($host,$service,$state,$display,$perfdata='')
-    {
-        //Send a POST request to the URL endpoint /v1/actions/process-check-result
-        //actions/process-check-result?service=example.localdomain!passive-ping6
-        $url='actions/process-check-result';
-        $body=array(
-            "filter"        => 'service.name=="'.$service.'" && service.host_name=="'.$host.'"',
-            'type'          => 'Service',
-            "exit_status"   => $state,
-            "plugin_output" => $display,
-            "performance_data" => $perfdata
-        );
-        try 
-        {
-            $result=$this->request('POST', $url, null, $body);
-        } catch (Exception $e) 
-        {
-            return array(false, $e->getMessage());
-        }
-        if (property_exists($result,'error') )
-        {
-            if (property_exists($result,'status'))
-            {
-                $message=$result->status;
-            }
-            else 
-            {
-                $message="Unkown status";
-            }
-            return array(false , 'Ret code ' .$result->error.' : '.$message);
-        }
-        if (property_exists($result, 'results'))
-        {
-            if (isset($result->results[0]))
-            {
-                return array(true,'code '.$result->results[0]->code.' : '.$result->results[0]->status);
-            }
-            else
-            {
-                return array(false,'Service not found');
-            }
+	/**
+	 * Send a passive service check
+	 * @param string $host : host name 
+	 * @param string $service : service name
+	 * @param int $state : state of service
+	 * @param string $display : service passive check output
+	 * @param string $perfdata : performance data as string
+	 * @return array (status = true (oK) or false (nok), string message)
+	 */
+	public function serviceCheckResult($host,$service,$state,$display,$perfdata='')
+	{
+		//Send a POST request to the URL endpoint /v1/actions/process-check-result
+		//actions/process-check-result?service=example.localdomain!passive-ping6
+		$url='actions/process-check-result';
+		$body=array(
+			"filter"        => 'service.name=="'.$service.'" && service.host_name=="'.$host.'"',
+			'type'          => 'Service',
+			"exit_status"   => $state,
+			"plugin_output" => $display,
+			"performance_data" => $perfdata
+		);
+		try 
+		{
+			$result=$this->request('POST', $url, null, $body);
+		} catch (Exception $e) 
+		{
+			return array(false, $e->getMessage());
+		}
+		if (property_exists($result,'error') )
+		{
+			if (property_exists($result,'status'))
+			{
+				$message=$result->status;
+			}
+			else 
+			{
+				$message="Unkown status";
+			}
+			return array(false , 'Ret code ' .$result->error.' : '.$message);
+		}
+		if (property_exists($result, 'results'))
+		{
+			if (isset($result->results[0]))
+			{
+				return array(true,'code '.$result->results[0]->code.' : '.$result->results[0]->status);
+			}
+			else
+			{
+				return array(false,'Service not found');
+			}
             
-        }
-        return array(false,'Unkown result, open issue with this : '.print_r($result,true));
-    }
+		}
+		return array(false,'Unkown result, open issue with this : '.print_r($result,true));
+	}
   
-    /**
-     * Check 'result' exists and check for errors/status
-     * @param \stdClass $result
-     * @throws Exception 
-     */
-    public function checkResultStatus(\stdClass $result) 
-    {
-        if (property_exists($result,'error') )
-        {
-            if (property_exists($result,'status'))
-            {
-                throw new Exception('Ret code ' .$result->error.' : ' . $result->status);
-            }
-            else
-            {
-                throw new Exception('Ret code ' .$result->error.' : Unkown status');
-            }
-        }
-        if ( ! property_exists($result, 'results'))
-        {
-            throw new Exception('Unkown result');
-        }
-    }
+	/**
+	 * Check 'result' exists and check for errors/status
+	 * @param \stdClass $result
+	 * @throws Exception 
+	 */
+	public function checkResultStatus(\stdClass $result) 
+	{
+		if (property_exists($result,'error') )
+		{
+			if (property_exists($result,'status'))
+			{
+				throw new Exception('Ret code ' .$result->error.' : ' . $result->status);
+			}
+			else
+			{
+				throw new Exception('Ret code ' .$result->error.' : Unkown status');
+			}
+		}
+		if ( ! property_exists($result, 'results'))
+		{
+			throw new Exception('Unkown result');
+		}
+	}
     
-    /**
-     * Send request to API
-     * @param string $method get/post/...
-     * @param string $url (after /v1/ )
-     * @param array $headers
-     * @param array $body 
-     * @throws Exception
-     * @return array
-     */
-    public function request($method, $url, $headers, $body) {
-        $auth = sprintf('%s:%s', $this->user, $this->pass);
-        $curlHeaders = array("Accept: application/json");
-        if ($body !== null) {
-            $body = json_encode($body);
-            array_push($curlHeaders, 'Content-Type: application/json');
-            //array_push($curlHeaders, 'X-HTTP-Method-Override: GET');
-        }
-        //var_dump($body);
-        //var_dump($this->url($url));
-        if ($headers !== null) {
-            $curlFinalHeaders = array_merge($curlHeaders, $headers);
-        } else 
-        {
-            $curlFinalHeaders=$curlHeaders;
-        }
-        $curl = $this->curl();
-        $opts = array(
-            CURLOPT_URL		=> $this->url($url),
-            CURLOPT_HTTPHEADER 	=> $curlFinalHeaders,
-            CURLOPT_USERPWD		=> $auth,
-            CURLOPT_CUSTOMREQUEST	=> strtoupper($method),
-            CURLOPT_RETURNTRANSFER 	=> true,
-            CURLOPT_CONNECTTIMEOUT 	=> 10,
-            CURLOPT_SSL_VERIFYHOST 	=> false,
-            CURLOPT_SSL_VERIFYPEER 	=> false,
-        );
-        if ($body !== null) {
-            $opts[CURLOPT_POSTFIELDS] = $body;
-        }
-        curl_setopt_array($curl, $opts);
-        $res = curl_exec($curl);
-        if ($res === false) {
-            throw new Exception('CURL ERROR: ' . curl_error($curl));
-        }
-        $statusCode = curl_getinfo($curl, CURLINFO_HTTP_CODE);
-        if ($statusCode === 401) {
-            throw new Exception('Unable to authenticate, please check your API credentials');
-        }
-        return $this->fromJsonResult($res);
-    }
+	/**
+	 * Send request to API
+	 * @param string $method get/post/...
+	 * @param string $url (after /v1/ )
+	 * @param array $headers
+	 * @param array $body 
+	 * @throws Exception
+	 * @return array
+	 */
+	public function request($method, $url, $headers, $body) {
+		$auth = sprintf('%s:%s', $this->user, $this->pass);
+		$curlHeaders = array("Accept: application/json");
+		if ($body !== null) {
+			$body = json_encode($body);
+			array_push($curlHeaders, 'Content-Type: application/json');
+			//array_push($curlHeaders, 'X-HTTP-Method-Override: GET');
+		}
+		//var_dump($body);
+		//var_dump($this->url($url));
+		if ($headers !== null) {
+			$curlFinalHeaders = array_merge($curlHeaders, $headers);
+		} else 
+		{
+			$curlFinalHeaders=$curlHeaders;
+		}
+		$curl = $this->curl();
+		$opts = array(
+			CURLOPT_URL		=> $this->url($url),
+			CURLOPT_HTTPHEADER 	=> $curlFinalHeaders,
+			CURLOPT_USERPWD		=> $auth,
+			CURLOPT_CUSTOMREQUEST	=> strtoupper($method),
+			CURLOPT_RETURNTRANSFER 	=> true,
+			CURLOPT_CONNECTTIMEOUT 	=> 10,
+			CURLOPT_SSL_VERIFYHOST 	=> false,
+			CURLOPT_SSL_VERIFYPEER 	=> false,
+		);
+		if ($body !== null) {
+			$opts[CURLOPT_POSTFIELDS] = $body;
+		}
+		curl_setopt_array($curl, $opts);
+		$res = curl_exec($curl);
+		if ($res === false) {
+			throw new Exception('CURL ERROR: ' . curl_error($curl));
+		}
+		$statusCode = curl_getinfo($curl, CURLINFO_HTTP_CODE);
+		if ($statusCode === 401) {
+			throw new Exception('Unable to authenticate, please check your API credentials');
+		}
+		return $this->fromJsonResult($res);
+	}
     
-    /**
-     * 
-     * @param string $json json encoded 
-     * @throws Exception
-     * @return array json decoded
-     */
-    protected function fromJsonResult($json) {
-        $result = @json_decode($json);
-        //var_dump($json);
-        if ($result === null) {
-            throw new Exception('Parsing JSON failed: '.$this->getLastJsonErrorMessage(json_last_error()));
-        }
-        return $result;
-    }
+	/**
+	 * 
+	 * @param string $json json encoded 
+	 * @throws Exception
+	 * @return array json decoded
+	 */
+	protected function fromJsonResult($json) {
+		$result = @json_decode($json);
+		//var_dump($json);
+		if ($result === null) {
+			throw new Exception('Parsing JSON failed: '.$this->getLastJsonErrorMessage(json_last_error()));
+		}
+		return $result;
+	}
     
-    /**
-     * Return text error no json error
-     * @param string $errorCode
-     * @return string
-     */
-    protected function getLastJsonErrorMessage($errorCode) {
-        if (!array_key_exists($errorCode, $this->errorReference)) {
-            return self::JSON_UNKNOWN_ERROR;
-        }
-        return $this->errorReference[$errorCode];
-    }
+	/**
+	 * Return text error no json error
+	 * @param string $errorCode
+	 * @return string
+	 */
+	protected function getLastJsonErrorMessage($errorCode) {
+		if (!array_key_exists($errorCode, $this->errorReference)) {
+			return self::JSON_UNKNOWN_ERROR;
+		}
+		return $this->errorReference[$errorCode];
+	}
 
-    public function standardQuery(string $urlType , string $filter, array $attributes)
-    {
-        /*
+	public function standardQuery(string $urlType , string $filter, array $attributes)
+	{
+		/*
          *  curl -k -s -u  trapdirector:trapdirector -H 'X-HTTP-Method-Override: GET' -X POST 'https://localhost:5665/v1/objects/hosts'
          *  -d '{"filter":"host.group==\"test_trap\"","attrs": ["address" ,"address6"]}'
          
          {"results":[{"attrs":{"__name":"Icinga host","address":"127.0.0.1","display_name":"Icinga host","name":"Icinga host"},"joins":{},"meta":{},"name":"Icinga host","type":"Host"}]}
          */
         
-        if (! isset($this->queryURL[$urlType] ))
-        {
-            throw new Exception("Unkown object type");
-        }
-        $url=$this->queryURL[$urlType];
-        $body=array();
-        if ($filter !== NULL)
-        {
-            $body['filter'] = $filter;
-        }
-        if (count($attributes) != 0)
-        {
-            $body['attrs'] = $attributes;
-        }
+		if (! isset($this->queryURL[$urlType] ))
+		{
+			throw new Exception("Unkown object type");
+		}
+		$url=$this->queryURL[$urlType];
+		$body=array();
+		if ($filter !== NULL)
+		{
+			$body['filter'] = $filter;
+		}
+		if (count($attributes) != 0)
+		{
+			$body['attrs'] = $attributes;
+		}
 
-        try
-        {
-            $result=$this->request('POST', $url, array('X-HTTP-Method-Override: GET'), $body);
-        } catch (Exception $e)
-        {
-            throw new Exception($e->getMessage());
-        }
+		try
+		{
+			$result=$this->request('POST', $url, array('X-HTTP-Method-Override: GET'), $body);
+		} catch (Exception $e)
+		{
+			throw new Exception($e->getMessage());
+		}
         
-        $this->checkResultStatus($result);
+		$this->checkResultStatus($result);
         
-        $numHost=0;
-        $hostArray=array();
-        while (isset($result->results[$numHost]) && property_exists ($result->results[$numHost],'attrs'))
-        {
-            $hostArray[$numHost] = $result->results[$numHost]->attrs;
-            $numHost++;
-        }
-        return $hostArray;
-    }
+		$numHost=0;
+		$hostArray=array();
+		while (isset($result->results[$numHost]) && property_exists ($result->results[$numHost],'attrs'))
+		{
+			$hostArray[$numHost] = $result->results[$numHost]->attrs;
+			$numHost++;
+		}
+		return $hostArray;
+	}
     
 }
 

library/Trapdirector/TrapsActions/IdoDBQuery.php

Indentation +222 added lines, -222 removed lines

@@ -18,247 +18,247 @@
 trait IdoDBQuery
 {
 
-    /** @return TrapsController */
-    abstract protected function getTrapCtrl();
+	/** @return TrapsController */
+	abstract protected function getTrapCtrl();
     
-    /** @return Zend_Db_Adapter_Abstract : returns DB connexion or null on error */
-    abstract public function getIdoDbConn();
+	/** @return Zend_Db_Adapter_Abstract : returns DB connexion or null on error */
+	abstract public function getIdoDbConn();
     
-    /** Get host(s) by IP (v4 or v6) or by name in IDO database
-     *	does not catch exceptions
-     *	@return array of objects ( name, id (object_id), display_name)
-     */
-    public function getHostByIP($ip)
-    {
-        // select a.name1, b.display_name from icinga.icinga_objects AS a , icinga.icinga_hosts AS b WHERE (b.address = '192.168.56.101' OR b.address6= '123456') and b.host_object_id=a.object_id
-        $dbConn = $this->getIdoDbConn();
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
+	/** Get host(s) by IP (v4 or v6) or by name in IDO database
+	 *	does not catch exceptions
+	 *	@return array of objects ( name, id (object_id), display_name)
+	 */
+	public function getHostByIP($ip)
+	{
+		// select a.name1, b.display_name from icinga.icinga_objects AS a , icinga.icinga_hosts AS b WHERE (b.address = '192.168.56.101' OR b.address6= '123456') and b.host_object_id=a.object_id
+		$dbConn = $this->getIdoDbConn();
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
         
-        // TODO : check for SQL injections
-        $query=$dbConn->select()
-        ->from(
-            array('a' => 'icinga_objects'),
-            array('name' => 'a.name1','id' => 'object_id'))
-            ->join(
-                array('b' => 'icinga_hosts'),
-                'b.host_object_id=a.object_id',
-                array('display_name' => 'b.display_name'))
-                ->where("(b.address LIKE '%".$ip."%' OR b.address6 LIKE '%".$ip."%' OR a.name1 LIKE '%".$ip."%' OR b.display_name LIKE '%".$ip."%') and a.is_active = 1");
-                return $dbConn->fetchAll($query);
-    }
+		// TODO : check for SQL injections
+		$query=$dbConn->select()
+		->from(
+			array('a' => 'icinga_objects'),
+			array('name' => 'a.name1','id' => 'object_id'))
+			->join(
+				array('b' => 'icinga_hosts'),
+				'b.host_object_id=a.object_id',
+				array('display_name' => 'b.display_name'))
+				->where("(b.address LIKE '%".$ip."%' OR b.address6 LIKE '%".$ip."%' OR a.name1 LIKE '%".$ip."%' OR b.display_name LIKE '%".$ip."%') and a.is_active = 1");
+				return $dbConn->fetchAll($query);
+	}
     
-    /** Get host(s) by name in IDO database
-     *	does not catch exceptions
-     *	@return array of objects ( name, id (object_id), display_name)
-     */
-    public function getHostByName($name)
-    {
-        // select a.name1, b.display_name from icinga.icinga_objects AS a , icinga.icinga_hosts AS b WHERE (b.address = '192.168.56.101' OR b.address6= '123456') and b.host_object_id=a.object_id
-        $dbConn = $this->getIdoDbConn();
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
+	/** Get host(s) by name in IDO database
+	 *	does not catch exceptions
+	 *	@return array of objects ( name, id (object_id), display_name)
+	 */
+	public function getHostByName($name)
+	{
+		// select a.name1, b.display_name from icinga.icinga_objects AS a , icinga.icinga_hosts AS b WHERE (b.address = '192.168.56.101' OR b.address6= '123456') and b.host_object_id=a.object_id
+		$dbConn = $this->getIdoDbConn();
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
         
-        // TODO : check for SQL injections
-        $query=$dbConn->select()
-        ->from(
-            array('a' => 'icinga_objects'),
-            array('name' => 'a.name1','id' => 'object_id'))
-            ->join(
-                array('b' => 'icinga_hosts'),
-                'b.host_object_id=a.object_id',
-                array('display_name' => 'b.display_name'))
-                ->where("a.name1 = '$name'");
-         return $dbConn->fetchAll($query);
-    }
+		// TODO : check for SQL injections
+		$query=$dbConn->select()
+		->from(
+			array('a' => 'icinga_objects'),
+			array('name' => 'a.name1','id' => 'object_id'))
+			->join(
+				array('b' => 'icinga_hosts'),
+				'b.host_object_id=a.object_id',
+				array('display_name' => 'b.display_name'))
+				->where("a.name1 = '$name'");
+		 return $dbConn->fetchAll($query);
+	}
     
-    /** Get host groups by  name in IDO database
-     *	does not catch exceptions
-     *	@return array of objects ( name, id (object_id), display_name)
-     */
-    public function getHostGroupByName($ip)
-    {
-        // select a.name1, b.display_name from icinga.icinga_objects AS a , icinga.icinga_hosts AS b WHERE (b.address = '192.168.56.101' OR b.address6= '123456') and b.host_object_id=a.object_id
-        $dbConn = $this->getIdoDbConn();
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
-        // TODO : check for SQL injections
-        $query=$dbConn->select()
-        ->from(
-            array('a' => 'icinga_objects'),
-            array('name' => 'a.name1','id' => 'object_id'))
-            ->join(
-                array('b' => 'icinga_hostgroups'),
-                'b.hostgroup_object_id=a.object_id',
-                array('display_name' => 'b.alias'))
-                ->where("(a.name1 LIKE '%".$ip."%' OR b.alias LIKE '%".$ip."%') and a.is_active = 1");
-                return $dbConn->fetchAll($query);
-    }
+	/** Get host groups by  name in IDO database
+	 *	does not catch exceptions
+	 *	@return array of objects ( name, id (object_id), display_name)
+	 */
+	public function getHostGroupByName($ip)
+	{
+		// select a.name1, b.display_name from icinga.icinga_objects AS a , icinga.icinga_hosts AS b WHERE (b.address = '192.168.56.101' OR b.address6= '123456') and b.host_object_id=a.object_id
+		$dbConn = $this->getIdoDbConn();
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
+		// TODO : check for SQL injections
+		$query=$dbConn->select()
+		->from(
+			array('a' => 'icinga_objects'),
+			array('name' => 'a.name1','id' => 'object_id'))
+			->join(
+				array('b' => 'icinga_hostgroups'),
+				'b.hostgroup_object_id=a.object_id',
+				array('display_name' => 'b.alias'))
+				->where("(a.name1 LIKE '%".$ip."%' OR b.alias LIKE '%".$ip."%') and a.is_active = 1");
+				return $dbConn->fetchAll($query);
+	}
  
-    /** Get host IP (v4 and v6) by name in IDO database
-     *	does not catch exceptions
-     *	@return array ( name, display_name, ip4, ip6)
-     */
-    public function getHostInfoByID($id)
-    {
-        if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
-        $dbConn = $this->getIdoDbConn();
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
-        $query=$dbConn->select()
-        ->from(
-            array('a' => 'icinga_objects'),
-            array('name' => 'a.name1'))
-            ->join(
-                array('b' => 'icinga_hosts'),
-                'b.host_object_id=a.object_id',
-                array('ip4' => 'b.address', 'ip6' => 'b.address6', 'display_name' => 'b.display_name'))
-                ->where("a.object_id = '".$id."'");
-                return $dbConn->fetchRow($query);
-    }
+	/** Get host IP (v4 and v6) by name in IDO database
+	 *	does not catch exceptions
+	 *	@return array ( name, display_name, ip4, ip6)
+	 */
+	public function getHostInfoByID($id)
+	{
+		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
+		$dbConn = $this->getIdoDbConn();
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
+		$query=$dbConn->select()
+		->from(
+			array('a' => 'icinga_objects'),
+			array('name' => 'a.name1'))
+			->join(
+				array('b' => 'icinga_hosts'),
+				'b.host_object_id=a.object_id',
+				array('ip4' => 'b.address', 'ip6' => 'b.address6', 'display_name' => 'b.display_name'))
+				->where("a.object_id = '".$id."'");
+				return $dbConn->fetchRow($query);
+	}
         
-    /** Get host by objectid  in IDO database
-     *	does not catch exceptions
-     *	@return array of objects ( id, name, display_name, ip, ip6,  )
-     */
-    public function getHostByObjectID($id) // TODO : duplicate of getHostInfoByID above
-    {
-        if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
-        $dbConn = $this->getIdoDbConn();
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
-        $query=$dbConn->select()
-        ->from(
-            array('a' => 'icinga_objects'),
-            array('name' => 'a.name1','id' => 'a.object_id'))
-            ->join(
-                array('b' => 'icinga_hosts'),
-                'b.host_object_id=a.object_id',
-                array('display_name' => 'b.display_name' , 'ip' => 'b.address', 'ip6' => 'b.address6'))
-                ->where('a.object_id = ?',$id);
-                return $dbConn->fetchRow($query);
-    }
+	/** Get host by objectid  in IDO database
+	 *	does not catch exceptions
+	 *	@return array of objects ( id, name, display_name, ip, ip6,  )
+	 */
+	public function getHostByObjectID($id) // TODO : duplicate of getHostInfoByID above
+	{
+		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
+		$dbConn = $this->getIdoDbConn();
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
+		$query=$dbConn->select()
+		->from(
+			array('a' => 'icinga_objects'),
+			array('name' => 'a.name1','id' => 'a.object_id'))
+			->join(
+				array('b' => 'icinga_hosts'),
+				'b.host_object_id=a.object_id',
+				array('display_name' => 'b.display_name' , 'ip' => 'b.address', 'ip6' => 'b.address6'))
+				->where('a.object_id = ?',$id);
+				return $dbConn->fetchRow($query);
+	}
     
-    /** Get services from object ( host_object_id) in IDO database
-     *	does not catch exceptions
-     *	@param $id	int object_id
-     *	@return array display_name (of service), service_object_id
-     */
-    public function getServicesByHostid($id)
-    {
-        // select a.name1, b.display_name from icinga.icinga_objects AS a , icinga.icinga_hosts AS b WHERE (b.address = '192.168.56.101' OR b.address6= '123456') and b.host_object_id=a.object_id
-        if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
-        $dbConn = $this->getIdoDbConn();
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
-        $query=$dbConn->select()
-        ->from(
-            array('s' => 'icinga_services'),
-            array('name' => 's.display_name','id' => 's.service_object_id'))
-            ->join(
-                array('a' => 'icinga_objects'),
-                's.service_object_id=a.object_id',
-                array('is_active'=>'a.is_active','name2'=>'a.name2'))
-                ->where('s.host_object_id='.$id.' AND a.is_active = 1');
-                return $dbConn->fetchAll($query);
-    }
+	/** Get services from object ( host_object_id) in IDO database
+	 *	does not catch exceptions
+	 *	@param $id	int object_id
+	 *	@return array display_name (of service), service_object_id
+	 */
+	public function getServicesByHostid($id)
+	{
+		// select a.name1, b.display_name from icinga.icinga_objects AS a , icinga.icinga_hosts AS b WHERE (b.address = '192.168.56.101' OR b.address6= '123456') and b.host_object_id=a.object_id
+		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
+		$dbConn = $this->getIdoDbConn();
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
+		$query=$dbConn->select()
+		->from(
+			array('s' => 'icinga_services'),
+			array('name' => 's.display_name','id' => 's.service_object_id'))
+			->join(
+				array('a' => 'icinga_objects'),
+				's.service_object_id=a.object_id',
+				array('is_active'=>'a.is_active','name2'=>'a.name2'))
+				->where('s.host_object_id='.$id.' AND a.is_active = 1');
+				return $dbConn->fetchAll($query);
+	}
     
-    /** Get services from hostgroup object id ( hostgroup_object_id) in IDO database
-     * 	gets all hosts in hostgroup and return common services
-     *	does not catch exceptions
-     *	@param $id	int object_id
-     *	@return array display_name (of service), service_object_id
-     */
-    public function getServicesByHostGroupid($id)
-    {
-        if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
-        $dbConn = $this->getIdoDbConn();
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
-        $query=$dbConn->select()
-        ->from(
-            array('s' => 'icinga_hostgroup_members'),
-            array('host_object_id' => 's.host_object_id'))
-            ->join(
-                array('a' => 'icinga_hostgroups'),
-                's.hostgroup_id=a.hostgroup_id',
-                'hostgroup_object_id')
-                ->where('a.hostgroup_object_id='.$id);
-                $hosts=$dbConn->fetchAll($query);
-                $common_services=array();
-                $num_hosts=count($hosts);
-                foreach ($hosts as $key => $host)
-                { // For each host, get all services and add in common_services if not found or add counter
-                    $host_services=$this->getServicesByHostid($host->host_object_id);
-                    foreach($host_services as $service)
-                    {
-                        if (isset($common_services[$service->name2]['num']))
-                        {
-                            $common_services[$service->name2]['num'] +=1;
-                        }
-                        else
-                        {
-                            $common_services[$service->name2]['num']=1;
-                            $common_services[$service->name2]['name']=$service->name;
-                        }
-                    }
-                }
-                $result=array();
+	/** Get services from hostgroup object id ( hostgroup_object_id) in IDO database
+	 * 	gets all hosts in hostgroup and return common services
+	 *	does not catch exceptions
+	 *	@param $id	int object_id
+	 *	@return array display_name (of service), service_object_id
+	 */
+	public function getServicesByHostGroupid($id)
+	{
+		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
+		$dbConn = $this->getIdoDbConn();
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
+		$query=$dbConn->select()
+		->from(
+			array('s' => 'icinga_hostgroup_members'),
+			array('host_object_id' => 's.host_object_id'))
+			->join(
+				array('a' => 'icinga_hostgroups'),
+				's.hostgroup_id=a.hostgroup_id',
+				'hostgroup_object_id')
+				->where('a.hostgroup_object_id='.$id);
+				$hosts=$dbConn->fetchAll($query);
+				$common_services=array();
+				$num_hosts=count($hosts);
+				foreach ($hosts as $key => $host)
+				{ // For each host, get all services and add in common_services if not found or add counter
+					$host_services=$this->getServicesByHostid($host->host_object_id);
+					foreach($host_services as $service)
+					{
+						if (isset($common_services[$service->name2]['num']))
+						{
+							$common_services[$service->name2]['num'] +=1;
+						}
+						else
+						{
+							$common_services[$service->name2]['num']=1;
+							$common_services[$service->name2]['name']=$service->name;
+						}
+					}
+				}
+				$result=array();
                 
-                //print_r($common_services);
-                foreach (array_keys($common_services) as $key)
-                {
-                    if ($common_services[$key]['num'] == $num_hosts)
-                    {
-                        array_push($result,array($key,$common_services[$key]['name']));
-                    }
-                }
+				//print_r($common_services);
+				foreach (array_keys($common_services) as $key)
+				{
+					if ($common_services[$key]['num'] == $num_hosts)
+					{
+						array_push($result,array($key,$common_services[$key]['name']));
+					}
+				}
                 
-                return $result;
-    }
+				return $result;
+	}
     
-    /** Get services object id by host name / service name in IDO database
-     *	does not catch exceptions
-     *	@param $hostname string host name
-     *	@param $name string service name
-     *	@return array  service id
-     */
-    public function getServiceIDByName($hostname,$name)
-    {
-        $dbConn = $this->getIdoDbConn();
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
+	/** Get services object id by host name / service name in IDO database
+	 *	does not catch exceptions
+	 *	@param $hostname string host name
+	 *	@param $name string service name
+	 *	@return array  service id
+	 */
+	public function getServiceIDByName($hostname,$name)
+	{
+		$dbConn = $this->getIdoDbConn();
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
         
-        if ($name == null)
-        {
-            return array();
-        }
+		if ($name == null)
+		{
+			return array();
+		}
         
-        $query=$dbConn->select()
-        ->from(
-            array('s' => 'icinga_services'),
-            array('name' => 's.display_name','id' => 's.service_object_id'))
-            ->join(
-                array('a' => 'icinga_objects'),
-                's.service_object_id=a.object_id',
-                'is_active')
-                ->where('a.name2=\''.$name.'\' AND a.name1=\''.$hostname.'\' AND a.is_active = 1');
+		$query=$dbConn->select()
+		->from(
+			array('s' => 'icinga_services'),
+			array('name' => 's.display_name','id' => 's.service_object_id'))
+			->join(
+				array('a' => 'icinga_objects'),
+				's.service_object_id=a.object_id',
+				'is_active')
+				->where('a.name2=\''.$name.'\' AND a.name1=\''.$hostname.'\' AND a.is_active = 1');
                 
-                return $dbConn->fetchAll($query);
-    }
+				return $dbConn->fetchAll($query);
+	}
     
-    /** Get object name from object_id  in IDO database
-     *	does not catch exceptions
-     *	@param int $id object_id (default to null, used first if not null)
-     *	@return array name1 (host) name2 (service)
-     */
-    public function getObjectNameByid($id)
-    {
-        // select a.name1, b.display_name from icinga.icinga_objects AS a , icinga.icinga_hosts AS b WHERE (b.address = '192.168.56.101' OR b.address6= '123456') and b.host_object_id=a.object_id
-        if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
-        $dbConn = $this->getIdoDbConn();
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
+	/** Get object name from object_id  in IDO database
+	 *	does not catch exceptions
+	 *	@param int $id object_id (default to null, used first if not null)
+	 *	@return array name1 (host) name2 (service)
+	 */
+	public function getObjectNameByid($id)
+	{
+		// select a.name1, b.display_name from icinga.icinga_objects AS a , icinga.icinga_hosts AS b WHERE (b.address = '192.168.56.101' OR b.address6= '123456') and b.host_object_id=a.object_id
+		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
+		$dbConn = $this->getIdoDbConn();
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
         
-        $query=$dbConn->select()
-        ->from(
-            array('a' => 'icinga_objects'),
-            array('name1' => 'a.name1','name2' => 'a.name2'))
-            ->where('a.object_id='.$id.' AND a.is_active = 1');
+		$query=$dbConn->select()
+		->from(
+			array('a' => 'icinga_objects'),
+			array('name1' => 'a.name1','name2' => 'a.name2'))
+			->where('a.object_id='.$id.' AND a.is_active = 1');
             
-            return $dbConn->fetchRow($query);
-    }
+			return $dbConn->fetchRow($query);
+	}
     
     
 }
\ No newline at end of file

configuration.php

Indentation +2 added lines, -2 removed lines

@@ -6,8 +6,8 @@
 $this->providePermission('trapdirector/config', $this->translate('Allow to create and modify traps services'));
 
 $this->provideConfigTab('config', array(
-    'title' => 'Configuration',
-    'url'   => 'settings'
+	'title' => 'Configuration',
+	'url'   => 'settings'
 ));
 
 /**

application/controllers/StatusController.php

Indentation +187 added lines, -187 removed lines

@@ -21,8 +21,8 @@ 
 		/************  Trapdb ***********/
 		try
 		{
-		    $dbConn = $this->getUIDatabase()->getDbConn();
-		    if ($dbConn === null) throw new \ErrorException('uncatched db error');
+			$dbConn = $this->getUIDatabase()->getDbConn();
+			if ($dbConn === null) throw new \ErrorException('uncatched db error');
 			$query = $dbConn->select()->from(
 				$this->getModuleConfig()->getTrapTableName(),
 				array('COUNT(*)')
@@ -51,7 +51,7 @@ 
 		
 		try
 		{		
-		    $this->view->currentLogDestination=$this->getUIDatabase()->getDBConfigValue('log_destination');
+			$this->view->currentLogDestination=$this->getUIDatabase()->getDBConfigValue('log_destination');
 			$this->view->logDestinations=$this->getModuleConfig()->getLogDestinations();
 			$this->view->currentLogFile=$this->getUIDatabase()->getDBConfigValue('log_file');
 			$this->view->logLevels=$this->getModuleConfig()->getlogLevels();
@@ -65,23 +65,23 @@ 
 		/*************** SNMP configuration ****************/
 		try
 		{
-		    $this->view->useSnmpTrapAddess= ( $this->getUIDatabase()->getDBConfigValue('use_SnmpTrapAddess') == 1 ) ? TRUE : FALSE;
-		    $this->view->SnmpTrapAddressOID=$this->getUIDatabase()->getDBConfigValue('SnmpTrapAddess_oid');
-		    $this->view->SnmpTrapAddressOIDDefault = ($this->view->SnmpTrapAddressOID == $this->getModuleConfig()->getDBConfigDefaults()['SnmpTrapAddess_oid'] ) ? TRUE : FALSE;
+			$this->view->useSnmpTrapAddess= ( $this->getUIDatabase()->getDBConfigValue('use_SnmpTrapAddess') == 1 ) ? TRUE : FALSE;
+			$this->view->SnmpTrapAddressOID=$this->getUIDatabase()->getDBConfigValue('SnmpTrapAddess_oid');
+			$this->view->SnmpTrapAddressOIDDefault = ($this->view->SnmpTrapAddressOID == $this->getModuleConfig()->getDBConfigDefaults()['SnmpTrapAddess_oid'] ) ? TRUE : FALSE;
 		    
 		}
 		catch (Exception $e)
 		{
-		    $this->displayExitError('status',$e->getMessage());
+			$this->displayExitError('status',$e->getMessage());
 		}		
 		
 	} 
   
 	/** Mib management
-	*	Post param : action=update_mib_db : update mib database
-	*	Post param : ation=check_update : check if mib update is finished
-	*	File post : mibfile -> save mib file
-	*/
+	 *	Post param : action=update_mib_db : update mib database
+	 *	Post param : ation=check_update : check if mib update is finished
+	 *	File post : mibfile -> save mib file
+	 */
 	public function mibAction()
 	{
 		$this->prepareTabs()->activate('mib');
@@ -100,22 +100,22 @@ 
 					$return=exec('icingacli trapdirector mib update --pid /tmp/trapdirector_update.pid');
 					if (preg_match('/OK/',$return))
 					{
-					    $this->_helper->json(array('status'=>'OK'));
+						$this->_helper->json(array('status'=>'OK'));
 					}
 					// Error
 					$this->_helper->json(array('status'=>$return));
 				}
 				if ($action == 'check_update')
 				{
-				    $file=@fopen('/tmp/trapdirector_update.pid','r');
-				    if ($file == false)
-				    {   // process is dead
-				        $this->_helper->json(array('status'=>'tu quoque fili','err'=>'Cannot open file'));
-				        return;
-				    }
-				    $pid=fgets($file);
-				    $output=array();
-				    $retVal=0;
+					$file=@fopen('/tmp/trapdirector_update.pid','r');
+					if ($file == false)
+					{   // process is dead
+						$this->_helper->json(array('status'=>'tu quoque fili','err'=>'Cannot open file'));
+						return;
+					}
+					$pid=fgets($file);
+					$output=array();
+					$retVal=0;
 					exec('ps '.$pid,$output,$retVal);
 					if ($retVal == 0)
 					{ // process is alive
@@ -123,7 +123,7 @@ 
 					}
 					else
 					{ // process is dead
-					    $this->_helper->json(array('status'=>'tu quoque fili','err'=>'no proc'.$pid));
+						$this->_helper->json(array('status'=>'tu quoque fili','err'=>'no proc'.$pid));
 					}
 				}
 				$this->_helper->json(array('status'=>'ERR : no '.$action.' action possible' ));
@@ -131,32 +131,32 @@ 
 			/** Check for mib file UPLOAD */
 			if (isset($_FILES['mibfile']))
 			{
-			    $name=filter_var($_FILES['mibfile']['name'],FILTER_SANITIZE_STRING);
+				$name=filter_var($_FILES['mibfile']['name'],FILTER_SANITIZE_STRING);
 				$DirConf=explode(':',$this->Config()->get('config', 'snmptranslate_dirs'));
 				$destDir=array_shift($DirConf);
 				if (!is_dir($destDir))
 				{
-				    $this->view->uploadStatus="ERROR : no $destDir directory, check module configuration";
+					$this->view->uploadStatus="ERROR : no $destDir directory, check module configuration";
 				}
 				else
 				{
-				    if (!is_writable($destDir))
-				    {
-				        $this->view->uploadStatus="ERROR : $destDir directory is not writable";
-				    }
-				    else
-				    {
-				        $destination = $destDir .'/'.$name; //$this->Module()->getBaseDir() . "/mibs/$name";
-				        $sourceTmpNam=filter_var($_FILES['mibfile']['tmp_name'],FILTER_SANITIZE_STRING);
-				        if (move_uploaded_file($sourceTmpNam,$destination)===false)
-    				    {
-    				        $this->view->uploadStatus="ERROR, file $destination not loaded. Check file and path name or selinux violations";
-    				    }
-    				    else
-    				    {
-    				        $this->view->uploadStatus="File $name uploaded in $destDir";
-    				    }
-				    }
+					if (!is_writable($destDir))
+					{
+						$this->view->uploadStatus="ERROR : $destDir directory is not writable";
+					}
+					else
+					{
+						$destination = $destDir .'/'.$name; //$this->Module()->getBaseDir() . "/mibs/$name";
+						$sourceTmpNam=filter_var($_FILES['mibfile']['tmp_name'],FILTER_SANITIZE_STRING);
+						if (move_uploaded_file($sourceTmpNam,$destination)===false)
+						{
+							$this->view->uploadStatus="ERROR, file $destination not loaded. Check file and path name or selinux violations";
+						}
+						else
+						{
+							$this->view->uploadStatus="File $name uploaded in $destDir";
+						}
+					}
 				}
 
 			}
@@ -172,16 +172,16 @@ 
 			$translate=exec($snmptranslate . ' 1');
 			if (preg_match('/iso/',$translate))
 			{
-			    $translate=exec($snmptranslate . ' 1.3.6.1.4');
-			    if (preg_match('/private/',$translate))
-			    {		    
-    				$this->view->snmptranslate='works fine';
-    				$this->view->snmptranslate_state='ok';
-			    }
-			    else
-			    {
-			        $this->view->snmptranslate='works fine but missing basic MIBs';
-			    }
+				$translate=exec($snmptranslate . ' 1.3.6.1.4');
+				if (preg_match('/private/',$translate))
+				{		    
+					$this->view->snmptranslate='works fine';
+					$this->view->snmptranslate_state='ok';
+				}
+				else
+				{
+					$this->view->snmptranslate='works fine but missing basic MIBs';
+				}
 			}
 			else
 			{
@@ -244,63 +244,63 @@ 
 	/** UI options */
 	public function uimgtAction()
 	{
-	    $this->prepareTabs()->activate('uimgt');
+		$this->prepareTabs()->activate('uimgt');
 	    
-	    $this->view->setError='';
-	    $this->view->setOKMsg='';
+		$this->view->setError='';
+		$this->view->setOKMsg='';
 	    
-	    //max_rows=25&row_update=update
-	    if ( $this->getRequest()->getParam('max_rows',NULL) !== NULL )
-	    {
-	        $maxRows = $this->getRequest()->getParam('max_rows');
-	        if (!preg_match('/^[0-9]+$/', $maxRows) || $maxRows < 1)
-	        {
-	            $this->view->setError='Max rows must be a number';
-	        }
-	        else
-	        {
-	            $this->setitemListDisplay($maxRows);
-	            $this->view->setOKMsg='Set max rows to ' . $maxRows;
-	        }
-	    }
+		//max_rows=25&row_update=update
+		if ( $this->getRequest()->getParam('max_rows',NULL) !== NULL )
+		{
+			$maxRows = $this->getRequest()->getParam('max_rows');
+			if (!preg_match('/^[0-9]+$/', $maxRows) || $maxRows < 1)
+			{
+				$this->view->setError='Max rows must be a number';
+			}
+			else
+			{
+				$this->setitemListDisplay($maxRows);
+				$this->view->setOKMsg='Set max rows to ' . $maxRows;
+			}
+		}
 	    
-	    if ( $this->getRequest()->getParam('add_category',NULL) !== NULL )
-	    {
-	        $addCat = $this->getRequest()->getParam('add_category');
-            $this->addHandlersCategory($addCat);
-	    }
+		if ( $this->getRequest()->getParam('add_category',NULL) !== NULL )
+		{
+			$addCat = $this->getRequest()->getParam('add_category');
+			$this->addHandlersCategory($addCat);
+		}
 	    
-	    if ( $this->getRequest()->getPost('type',NULL) !== NULL )
-	    {
-	        $type = $this->getRequest()->getPost('type',NULL);
-	        $index = $this->getRequest()->getPost('index',NULL);
-	        $newname = $this->getRequest()->getPost('newname',NULL);
+		if ( $this->getRequest()->getPost('type',NULL) !== NULL )
+		{
+			$type = $this->getRequest()->getPost('type',NULL);
+			$index = $this->getRequest()->getPost('index',NULL);
+			$newname = $this->getRequest()->getPost('newname',NULL);
 
-	        if (!preg_match('/^[0-9]+$/', $index) || $index < 1)
-	            $this->_helper->json(array('status'=>'Bad index'));
+			if (!preg_match('/^[0-9]+$/', $index) || $index < 1)
+				$this->_helper->json(array('status'=>'Bad index'));
 	        
-	        switch ($type)
-	        {
-	            case 'delete':
-	                $this->delHandlersCategory($index);
-	                $this->_helper->json(array('status'=>'OK'));
-	                return;
-	                break;
-	            case 'rename':
-	                $this->renameHandlersCategory($index, $newname);
-	                $this->_helper->json(array('status'=>'OK'));
-	                return;
-	                break;
-	            default:
-	                $this->_helper->json(array('status'=>'Unknwon command'));
-	                return;
-	                break;
-	        }
-	    }
+			switch ($type)
+			{
+				case 'delete':
+					$this->delHandlersCategory($index);
+					$this->_helper->json(array('status'=>'OK'));
+					return;
+					break;
+				case 'rename':
+					$this->renameHandlersCategory($index, $newname);
+					$this->_helper->json(array('status'=>'OK'));
+					return;
+					break;
+				default:
+					$this->_helper->json(array('status'=>'Unknwon command'));
+					return;
+					break;
+			}
+		}
 	    
-	    $this->view->maxRows = $this->itemListDisplay();
+		$this->view->maxRows = $this->itemListDisplay();
 	    
-	    $this->view->categories = $this->getHandlersCategory();
+		$this->view->categories = $this->getHandlersCategory();
 	    
 	    
 	    
@@ -349,47 +349,47 @@ 
 	 */
 	public function pluginsAction()
 	{
-	    $this->prepareTabs()->activate('plugins');
+		$this->prepareTabs()->activate('plugins');
 	    
-	    require_once($this->Module()->getBaseDir() .'/bin/trap_class.php');
-	    $icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
-	    $Trap = new Trap($icingaweb2_etc,4);
+		require_once($this->Module()->getBaseDir() .'/bin/trap_class.php');
+		$icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
+		$Trap = new Trap($icingaweb2_etc,4);
 	    
-	    $this->view->pluginLoaded = htmlentities($Trap->pluginClass->registerAllPlugins(false));
+		$this->view->pluginLoaded = htmlentities($Trap->pluginClass->registerAllPlugins(false));
 	    
-	    $enabledPlugins = $Trap->pluginClass->getEnabledPlugins();
+		$enabledPlugins = $Trap->pluginClass->getEnabledPlugins();
 
-	    $pluginList = $Trap->pluginClass->pluginList();
+		$pluginList = $Trap->pluginClass->pluginList();
 	    
-	    // Plugin list and fill function name list
-	    $functionList=array();
-	    $this->view->pluginArray=array();
-	    foreach ($pluginList as $plugin)
-	    {
-	        $pluginDetails=$Trap->pluginClass->pluginDetails($plugin);
-	        $pluginDetails->enabled =  (in_array($plugin, $enabledPlugins)) ? true : false;
-	        $pluginDetails->catchAllTraps = ($pluginDetails->catchAllTraps === true )? 'Yes' : 'No';
-	        $pluginDetails->processTraps = ($pluginDetails->processTraps === true )? 'Yes' : 'No';
-	        $pluginDetails->description = htmlentities($pluginDetails->description);
-	        $pluginDetails->description = preg_replace('/\n/','<br>',$pluginDetails->description);
-	        array_push($this->view->pluginArray, $pluginDetails);
-	        // Get functions for function details
-	        foreach ($pluginDetails->funcArray as $function)
-	        {
-	            array_push($functionList,$function);
-	        }
-	    }
+		// Plugin list and fill function name list
+		$functionList=array();
+		$this->view->pluginArray=array();
+		foreach ($pluginList as $plugin)
+		{
+			$pluginDetails=$Trap->pluginClass->pluginDetails($plugin);
+			$pluginDetails->enabled =  (in_array($plugin, $enabledPlugins)) ? true : false;
+			$pluginDetails->catchAllTraps = ($pluginDetails->catchAllTraps === true )? 'Yes' : 'No';
+			$pluginDetails->processTraps = ($pluginDetails->processTraps === true )? 'Yes' : 'No';
+			$pluginDetails->description = htmlentities($pluginDetails->description);
+			$pluginDetails->description = preg_replace('/\n/','<br>',$pluginDetails->description);
+			array_push($this->view->pluginArray, $pluginDetails);
+			// Get functions for function details
+			foreach ($pluginDetails->funcArray as $function)
+			{
+				array_push($functionList,$function);
+			}
+		}
 	    
-	    // Function list with details
-	    $this->view->functionList=array();
-	    foreach ($functionList as $function)
-	    {
-	        $functionDetail = $Trap->pluginClass->getFunctionDetails($function);
-	        $functionDetail->params = htmlentities($functionDetail->params);
-	        $functionDetail->description = htmlentities($functionDetail->description);
-	        $functionDetail->description = preg_replace('/\n/','<br>',$functionDetail->description);
-	        array_push($this->view->functionList, $functionDetail);
-	    }
+		// Function list with details
+		$this->view->functionList=array();
+		foreach ($functionList as $function)
+		{
+			$functionDetail = $Trap->pluginClass->getFunctionDetails($function);
+			$functionDetail->params = htmlentities($functionDetail->params);
+			$functionDetail->description = htmlentities($functionDetail->description);
+			$functionDetail->description = preg_replace('/\n/','<br>',$functionDetail->description);
+			array_push($this->view->functionList, $functionDetail);
+		}
 
 	}
 
@@ -398,47 +398,47 @@ 
 	 */
 	public function debugAction()
 	{
-	    $this->view->answer='No answer';
-	    $this->view->input1 = '';
+		$this->view->answer='No answer';
+		$this->view->input1 = '';
 	    
-	    $postData=$this->getRequest()->getPost();
-	    if (isset($postData['input1']))
-	    {
-	        $input1 = $postData['input1'];
-	        $input2 = $postData['input2'];
-	        $input3 = $postData['input3'];
-	        $this->view->input1 = $input1;
-	        //$this->view->answer=$input1 . '/' . $input2  . '/' . $input3;
-	        try {
-	            $this->view->answer = 'Answer : ';
-	            $API = $this->getIdoConn();
-	            $DB = $this->getUIDatabase();
+		$postData=$this->getRequest()->getPost();
+		if (isset($postData['input1']))
+		{
+			$input1 = $postData['input1'];
+			$input2 = $postData['input2'];
+			$input3 = $postData['input3'];
+			$this->view->input1 = $input1;
+			//$this->view->answer=$input1 . '/' . $input2  . '/' . $input3;
+			try {
+				$this->view->answer = 'Answer : ';
+				$API = $this->getIdoConn();
+				$DB = $this->getUIDatabase();
 
-	            //$hosts = $DB->getServicesByHostGroupid($input1);
-	            //$this->view->answer .= 'services : '.print_r($hosts,true);
+				//$hosts = $DB->getServicesByHostGroupid($input1);
+				//$this->view->answer .= 'services : '.print_r($hosts,true);
 	            
-	            $hosts = $API->getServicesByHostGroupid($input1);
-	            $this->view->answer .= 'services : '.print_r($hosts,true);
+				$hosts = $API->getServicesByHostGroupid($input1);
+				$this->view->answer .= 'services : '.print_r($hosts,true);
 	            
 
 	            
-	            return;
-	            $hosts = $API->getHostByIP($input1);
-	            $this->view->answer .= 'Host : '. print_r($hosts,true);
-	            $hosts = $API->getHostGroupByName($input1);
-	            $this->view->answer .= 'Group : '.print_r($hosts,true);
-	            $hosts = $API->getServicesByHostid($input1);
-	            $this->view->answer .= 'service : '.print_r($hosts,true);
+				return;
+				$hosts = $API->getHostByIP($input1);
+				$this->view->answer .= 'Host : '. print_r($hosts,true);
+				$hosts = $API->getHostGroupByName($input1);
+				$this->view->answer .= 'Group : '.print_r($hosts,true);
+				$hosts = $API->getServicesByHostid($input1);
+				$this->view->answer .= 'service : '.print_r($hosts,true);
 	            
-	            /* $hosts = $API->getHostsIPByHostGroup($input1);
+				/* $hosts = $API->getHostsIPByHostGroup($input1);
 	            $this->view->answer .= "\n Hostgrp : " . print_r($hosts,true); */
 	            
-	        } catch (Exception $e)
-	        {
-	            $this->view->answer = "Exception : " . print_r($e);
-	        }
+			} catch (Exception $e)
+			{
+				$this->view->answer = "Exception : " . print_r($e);
+			}
 	        
-	    }
+		}
 	    
 	}
 	
@@ -450,36 +450,36 @@ 
 		)->add('mib', array(
 			'label' => $this->translate('MIB Management'),
 			'url'   => $this->getModuleConfig()->urlPath() . '/status/mib')
-	    )->add('uimgt', array(
-	        'label' => $this->translate('UI Configuration'),
-	        'url'   => $this->getModuleConfig()->urlPath() . '/status/uimgt')
-        )->add('services', array(
+		)->add('uimgt', array(
+			'label' => $this->translate('UI Configuration'),
+			'url'   => $this->getModuleConfig()->urlPath() . '/status/uimgt')
+		)->add('services', array(
 			'label' => $this->translate('Services management'),
 			'url'   => $this->getModuleConfig()->urlPath() . '/status/services')
-	    )->add('plugins', array(
-	        'label' => $this->translate('Plugins management'),
-	        'url'   => $this->getModuleConfig()->urlPath() . '/status/plugins')
-	    );
+		)->add('plugins', array(
+			'label' => $this->translate('Plugins management'),
+			'url'   => $this->getModuleConfig()->urlPath() . '/status/plugins')
+		);
 	} 
 }
 
 // TODO : see if useless 
 class UploadForm extends Form
 { 
-    public function __construct($options = null) 
-    {
-        parent::__construct($options);
-        $this->addElements2();
-    }
+	public function __construct($options = null) 
+	{
+		parent::__construct($options);
+		$this->addElements2();
+	}
 
-    public function addElements2()
-    {
-        // File Input
-        $file = new File('mib-file');
-        $file->setLabel('Mib upload');
-             //->setAttrib('multiple', null);
-        $this->addElement($file);
+	public function addElements2()
+	{
+		// File Input
+		$file = new File('mib-file');
+		$file->setLabel('Mib upload');
+			 //->setAttrib('multiple', null);
+		$this->addElement($file);
 		$button = new Submit("upload",array('ignore'=>false));
 		$this->addElement($button);//->setIgnore(false);
-    }
+	}
 }

application/controllers/SettingsController.php

Indentation +231 added lines, -231 removed lines

@@ -25,16 +25,16 @@ 
    */
   private function get_param()
   {
-      $dberrorMsg=$this->params->get('dberror');
-      if ($dberrorMsg != '')
-      {
-          $this->view->errorDetected=$dberrorMsg;
-      }
-      $dberrorMsg=$this->params->get('idodberror');
-      if ($dberrorMsg != '')
-      {
-          $this->view->errorDetected=$dberrorMsg;
-      }
+	  $dberrorMsg=$this->params->get('dberror');
+	  if ($dberrorMsg != '')
+	  {
+		  $this->view->errorDetected=$dberrorMsg;
+	  }
+	  $dberrorMsg=$this->params->get('idodberror');
+	  if ($dberrorMsg != '')
+	  {
+		  $this->view->errorDetected=$dberrorMsg;
+	  }
   }
   
   /**
@@ -43,22 +43,22 @@ 
    */
   private function check_empty_config()
   {
-      $this->view->configErrorDetected == NULL; // Displayed error on various conifugration errors.
-      if ($this->Config()->isEmpty() == true)
-      {
-          $this->Config()->setSection('config'); // Set base config section.
-          try
-          {
-              $this->Config()->saveIni();
-              $this->view->configErrorDetected='Configuration is empty : you can run install script with parameters (see Automatic installation below)';
-              //$emptyConfig=1;
-          }
-          catch (Exception $e)
-          {
-              $this->view->configErrorDetected=$e->getMessage();
-          }
+	  $this->view->configErrorDetected == NULL; // Displayed error on various conifugration errors.
+	  if ($this->Config()->isEmpty() == true)
+	  {
+		  $this->Config()->setSection('config'); // Set base config section.
+		  try
+		  {
+			  $this->Config()->saveIni();
+			  $this->view->configErrorDetected='Configuration is empty : you can run install script with parameters (see Automatic installation below)';
+			  //$emptyConfig=1;
+		  }
+		  catch (Exception $e)
+		  {
+			  $this->view->configErrorDetected=$e->getMessage();
+		  }
           
-      }
+	  }
   }
   
   /**
@@ -71,56 +71,56 @@ 
    */
   private function check_db()
   {
-      $db_message=array( // index => ( message OK, message NOK, optional link if NOK )
-          0	=>	array('Database configuration OK','',''),
-          1	=>	array('Database set in config.ini','No database in config.ini',''),
-          2	=>	array('Database exists in Icingaweb2 config','Database does not exist in Icingaweb2 : ',
-              Url::fromPath('config/resource')),
-          3	=>	array('Database credentials OK','Database does not exist/invalid credentials/no schema : ',
-              Url::fromPath('trapdirector/settings/createschema')),
-          4	=>	array('Schema is set','Schema is not set for ',
-              Url::fromPath('trapdirector/settings/createschema')),
-          5	=>	array('Schema is up to date','Schema is outdated :',
-              Url::fromPath('trapdirector/settings/updateschema')),
-      );
+	  $db_message=array( // index => ( message OK, message NOK, optional link if NOK )
+		  0	=>	array('Database configuration OK','',''),
+		  1	=>	array('Database set in config.ini','No database in config.ini',''),
+		  2	=>	array('Database exists in Icingaweb2 config','Database does not exist in Icingaweb2 : ',
+			  Url::fromPath('config/resource')),
+		  3	=>	array('Database credentials OK','Database does not exist/invalid credentials/no schema : ',
+			  Url::fromPath('trapdirector/settings/createschema')),
+		  4	=>	array('Schema is set','Schema is not set for ',
+			  Url::fromPath('trapdirector/settings/createschema')),
+		  5	=>	array('Schema is up to date','Schema is outdated :',
+			  Url::fromPath('trapdirector/settings/updateschema')),
+	  );
       
-      try {
-          $this->getUIDatabase()->testGetDb(); // Get DB in test mode
-          $dberror=array(0,'');
-      } catch (DBException $e) {
-          $dberror = $e->getArray();
-      }
+	  try {
+		  $this->getUIDatabase()->testGetDb(); // Get DB in test mode
+		  $dberror=array(0,'');
+	  } catch (DBException $e) {
+		  $dberror = $e->getArray();
+	  }
       
-      $this->view->db_error=$dberror[0];
-      switch ($dberror[0])
-      {
-          case 2:
-          case 4:
-              $db_message[$dberror[0]][1] .= $dberror[1];
-              break;
-          case 3:
-              $db_message[$dberror[0]][1] .= $dberror[1] . ', Message : ' . $dberror[2];
-              break;
-          case 5:
-              $db_message[$dberror[0]][1] .= ' version '. $dberror[1] . ', version needed : ' .$dberror[2];
-              break;
-          case 0:
-          case 1:
-              break;
-          default:
-              new ProgrammingError('Out of bond result from database test');
-      }
-      $this->view->message=$db_message;
+	  $this->view->db_error=$dberror[0];
+	  switch ($dberror[0])
+	  {
+		  case 2:
+		  case 4:
+			  $db_message[$dberror[0]][1] .= $dberror[1];
+			  break;
+		  case 3:
+			  $db_message[$dberror[0]][1] .= $dberror[1] . ', Message : ' . $dberror[2];
+			  break;
+		  case 5:
+			  $db_message[$dberror[0]][1] .= ' version '. $dberror[1] . ', version needed : ' .$dberror[2];
+			  break;
+		  case 0:
+		  case 1:
+			  break;
+		  default:
+			  new ProgrammingError('Out of bond result from database test');
+	  }
+	  $this->view->message=$db_message;
       
-      try {
-          $this->getUIDatabase()->testGetIdoDb(); // Get DB in test mode
-          $dberror=array(0,'');
-      } catch (DBException $e) {
-          $dberror = $e->getArray();
-      }
+	  try {
+		  $this->getUIDatabase()->testGetIdoDb(); // Get DB in test mode
+		  $dberror=array(0,'');
+	  } catch (DBException $e) {
+		  $dberror = $e->getArray();
+	  }
       
-      $this->view->ido_db_error=$dberror[0];
-      $this->view->ido_message='IDO Database : ' . $dberror[1];
+	  $this->view->ido_db_error=$dberror[0];
+	  $this->view->ido_message='IDO Database : ' . $dberror[1];
   }
   
   /**
@@ -130,23 +130,23 @@ 
    */
   private function check_api()
   {
-      if ($this->Config()->get('config', 'icingaAPI_host') != '')
-      {
-          $apitest=new Icinga2API($this->Config()->get('config', 'icingaAPI_host'),$this->Config()->get('config', 'icingaAPI_port'));
-          $apitest->setCredentials($this->Config()->get('config', 'icingaAPI_user'), $this->Config()->get('config', 'icingaAPI_password'));
-          try {
-              list($this->view->apimessageError,$this->view->apimessage)=$apitest->test($this->getModuleConfig()::getapiUserPermissions());
-              //$this->view->apimessageError=false;
-          } catch (RuntimeException $e) {
-              $this->view->apimessage='API config : ' . $e->getMessage();
-              $this->view->apimessageError=true;
-          }
-      }
-      else
-      {
-          $this->view->apimessage='API parameters not configured';
-          $this->view->apimessageError=true;
-      }
+	  if ($this->Config()->get('config', 'icingaAPI_host') != '')
+	  {
+		  $apitest=new Icinga2API($this->Config()->get('config', 'icingaAPI_host'),$this->Config()->get('config', 'icingaAPI_port'));
+		  $apitest->setCredentials($this->Config()->get('config', 'icingaAPI_user'), $this->Config()->get('config', 'icingaAPI_password'));
+		  try {
+			  list($this->view->apimessageError,$this->view->apimessage)=$apitest->test($this->getModuleConfig()::getapiUserPermissions());
+			  //$this->view->apimessageError=false;
+		  } catch (RuntimeException $e) {
+			  $this->view->apimessage='API config : ' . $e->getMessage();
+			  $this->view->apimessageError=true;
+		  }
+	  }
+	  else
+	  {
+		  $this->view->apimessage='API parameters not configured';
+		  $this->view->apimessageError=true;
+	  }
   }
 
   /**
@@ -157,21 +157,21 @@ 
    */
   private function check_icingaweb_path()
   {
-      $this->view->icingaEtcWarn=0;
-      $icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
-      if ($icingaweb2_etc != "/etc/icingaweb2/" && $icingaweb2_etc != '')
-      {
-          $output=array();
+	  $this->view->icingaEtcWarn=0;
+	  $icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
+	  if ($icingaweb2_etc != "/etc/icingaweb2/" && $icingaweb2_etc != '')
+	  {
+		  $output=array();
           
-          exec('cat ' . $this->module->getBaseDir() .'/bin/trap_in.php | grep "\$icingaweb2Etc=" ',$output);
+		  exec('cat ' . $this->module->getBaseDir() .'/bin/trap_in.php | grep "\$icingaweb2Etc=" ',$output);
           
           
-          if (! isset($output[0]) || ! preg_match('#"'. $icingaweb2_etc .'"#',$output[0]))
-          {
-              $this->view->icingaEtcWarn=1;
-              $this->view->icingaweb2_etc=$icingaweb2_etc;
-          }
-      }
+		  if (! isset($output[0]) || ! preg_match('#"'. $icingaweb2_etc .'"#',$output[0]))
+		  {
+			  $this->view->icingaEtcWarn=1;
+			  $this->view->icingaweb2_etc=$icingaweb2_etc;
+		  }
+	  }
       
   }
   
@@ -182,15 +182,15 @@ 
    */
   private function get_db_list($allowed)
   {
-      $resources = array();
-      foreach (ResourceFactory::getResourceConfigs() as $name => $resource) 
-      {
-          if ($resource->get('type') === 'db' && in_array($resource->get('db'), $allowed)) 
-          {
-              $resources[$name] = $name;
-          }
-      }
-      return $resources;
+	  $resources = array();
+	  foreach (ResourceFactory::getResourceConfigs() as $name => $resource) 
+	  {
+		  if ($resource->get('type') === 'db' && in_array($resource->get('db'), $allowed)) 
+		  {
+			  $resources[$name] = $name;
+		  }
+	  }
+	  return $resources;
   }
   
   /**
@@ -199,23 +199,23 @@ 
    */
   private function get_php_binary()
   {
-      $phpBinary= array( PHP_BINARY, PHP_BINDIR . "/php", '/usr/bin/php');
+	  $phpBinary= array( PHP_BINARY, PHP_BINDIR . "/php", '/usr/bin/php');
 
-      foreach ($phpBinary as $phpBin )
-      {
-          $output=array();
-          $retCode=255;
-          $input="154865134987aaaa";
-          exec("$phpBin -r \"echo '$input';\"",$output,$retCode);
+	  foreach ($phpBinary as $phpBin )
+	  {
+		  $output=array();
+		  $retCode=255;
+		  $input="154865134987aaaa";
+		  exec("$phpBin -r \"echo '$input';\"",$output,$retCode);
           
-          if (! isset($output[0])) $output[0]="NO OUT";
+		  if (! isset($output[0])) $output[0]="NO OUT";
           
-          if ($retCode == 0 && preg_match("/$input/",$output[0]) == 1)
-          {
-              return $phpBin;
-          }          
-      }
-      return NULL;
+		  if ($retCode == 0 && preg_match("/$input/",$output[0]) == 1)
+		  {
+			  return $phpBin;
+		  }          
+	  }
+	  return NULL;
   }
   
   /**
@@ -234,7 +234,7 @@ 
   public function indexAction()
   {
       
-    // CHeck permissions : display tests in any case, but no configuration.
+	// CHeck permissions : display tests in any case, but no configuration.
 	$this->view->configPermission=$this->checkModuleConfigPermission(1);
 	// But check read permission
 	$this->checkReadPermission();
@@ -242,16 +242,16 @@ 
 	$this->view->tabs = $this->Module()->getConfigTabs()->activate('config');	
 	
 	// Get message : sent on configuration problems detected by controllers
-    $this->get_param();
+	$this->get_param();
     
-    // Test if configuration exists, if not create for installer script
+	// Test if configuration exists, if not create for installer script
 	$this->check_empty_config();
 
 	// Test Database
-    $this->check_db();
+	$this->check_db();
 	
 	//********* Test API
-    $this->check_api();
+	$this->check_api();
 	
 	//*********** Test snmptrapd alive and options
 	list ($this->view->snmptrapdError, $this->view->snmptrapdMessage) = $this->checkSnmpTrapd();
@@ -265,7 +265,7 @@ 
 	$phpBinary = $this->get_php_binary();
 	if ($phpBinary === null)
 	{
-	    $phpBinary = ' PHP BINARY NOT FOUND ';
+		$phpBinary = ' PHP BINARY NOT FOUND ';
 	    
 	}
 	
@@ -273,11 +273,11 @@ 
 	$this->view->traps_in_config= $phpBinary . ' ' . $this->Module()->getBaseDir() . '/bin/trap_in.php';
 	
 	$this->view->installer= $this->Module()->getBaseDir() . '/bin/installer.sh '
-	    . ' -c all ' 
-	    . ' -d ' . $this->Module()->getBaseDir()
-	    . ' -p ' . $phpBinary
-	    . ' -a ' . exec('whoami')
-	    . ' -w ' . Icinga::app()->getConfigDir();
+		. ' -c all ' 
+		. ' -d ' . $this->Module()->getBaseDir()
+		. ' -p ' . $phpBinary
+		. ' -a ' . exec('whoami')
+		. ' -w ' . Icinga::app()->getConfigDir();
 	        
 	// ******************* configuration form setup*******************
 	$this->view->form = $form = new TrapsConfigForm();
@@ -311,15 +311,15 @@ 
   public function satelliteAction()
   {
       
-      // CHeck permissions
-      $this->view->configPermission=$this->checkModuleConfigPermission();
+	  // CHeck permissions
+	  $this->view->configPermission=$this->checkModuleConfigPermission();
       
-      // Setup tabs
-      $this->view->tabs = $this->Module()->getConfigTabs()->activate('satellite');	
+	  // Setup tabs
+	  $this->view->tabs = $this->Module()->getConfigTabs()->activate('satellite');	
       
-      $this->view->masterHASet = FALSE;
+	  $this->view->masterHASet = FALSE;
       
-      $this->view->masterHAConf = array ('name' => 'masterHA1', 'icingaweb2 user'=>'User1');
+	  $this->view->masterHAConf = array ('name' => 'masterHA1', 'icingaweb2 user'=>'User1');
   }
   
   
@@ -335,8 +335,8 @@ 
 	
 	try 
 	{
-	    $this->getUIDatabase()->testGetDb(); // Get DB in test mode
-	    printf('Schema already exists');
+		$this->getUIDatabase()->testGetDb(); // Get DB in test mode
+		printf('Schema already exists');
 	    
 	} 
 	catch (DBException $e) 
@@ -347,19 +347,19 @@ 
 		// Get module database name
 		$dbName=$this->Config()->get('config', 'database');
 
-        $dbResource = ResourceFactory::getResourceConfig($dbName);
-        $dbType=$dbResource->get('db');
-        switch ($dbType) {
-          case 'mysql':
-              $dbFileExt='sql';
-              break;
-          case 'pgsql':
-              $dbFileExt='pgsql';
-              break;
-          default:
-              printf("Database configuration error : Unsuported DB");
-              return;
-        } 
+		$dbResource = ResourceFactory::getResourceConfig($dbName);
+		$dbType=$dbResource->get('db');
+		switch ($dbType) {
+		  case 'mysql':
+			  $dbFileExt='sql';
+			  break;
+		  case 'pgsql':
+			  $dbFileExt='pgsql';
+			  break;
+		  default:
+			  printf("Database configuration error : Unsuported DB");
+			  return;
+		} 
 
 		printf('<pre>');
 		require_once $this->Module()->getBaseDir() .'/bin/trap_class.php';
@@ -383,32 +383,32 @@ 
   public function updateschemaAction()
   {
 	  $this->checkModuleConfigPermission();
-      $this->getTabs()->add('get',array(
-    		'active'	=> true,
-    		'label'		=> $this->translate('Update Schema'),
-    		'url'		=> Url::fromRequest()
-    	));
+	  $this->getTabs()->add('get',array(
+			'active'	=> true,
+			'label'		=> $this->translate('Update Schema'),
+			'url'		=> Url::fromRequest()
+		));
 	  // check if needed
 	  $dberror=array();
-      try
-      {
-          $this->getUIDatabase()->testGetDb(); // Get DB in test mode
-          echo 'Schema already exists and is up to date<br>';
-          return;
-      }
-      catch (DBException $e)
-      {
-          $dberror=$e->getArray(); 
-      }
+	  try
+	  {
+		  $this->getUIDatabase()->testGetDb(); // Get DB in test mode
+		  echo 'Schema already exists and is up to date<br>';
+		  return;
+	  }
+	  catch (DBException $e)
+	  {
+		  $dberror=$e->getArray(); 
+	  }
 	  
 	  echo 'Return to <a href="' . Url::fromPath('trapdirector/settings') .'" class="link-button icon-wrench"> settings page </a><br><br>';
 	  
 	  if ($dberror[0] != 5)
 	  {
-	      echo 'Database does not exists or is not setup correctly<br>';
-	      return;
+		  echo 'Database does not exists or is not setup correctly<br>';
+		  return;
 	  }
-      // setup
+	  // setup
 	  require_once($this->Module()->getBaseDir() .'/bin/trap_class.php');
 	  $icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
 	  $debug_level=4;
@@ -421,20 +421,20 @@ 
 	  $target_version=$dberror[2];
 	  
 	  if ($this->params->get('msgok') == null) {
-	      // Check for messages and display if any
-              echo "Upgrade databse is going to start.<br>Don't forget to backup your database before update<br>";
-	      $Trap->setLogging(2,'syslog');
-	      $message = $Trap->trapsDB->update_schema($updateSchema,$target_version,$prefix,true);
-	      if ($message != '')
-	      {
-	          echo 'Note :<br><pre>';
-	          echo $message;
-	          echo '</pre>';
-	          echo '<br>';
-	          echo '<a  class="link-button" style="font-size:large;font-weight:bold" href="' . Url::fromPath('trapdirector/settings/updateschema') .'?msgok=1">Click here to update</a>';
-	          echo '<br>';
-	          return;
-	      }
+		  // Check for messages and display if any
+			  echo "Upgrade databse is going to start.<br>Don't forget to backup your database before update<br>";
+		  $Trap->setLogging(2,'syslog');
+		  $message = $Trap->trapsDB->update_schema($updateSchema,$target_version,$prefix,true);
+		  if ($message != '')
+		  {
+			  echo 'Note :<br><pre>';
+			  echo $message;
+			  echo '</pre>';
+			  echo '<br>';
+			  echo '<a  class="link-button" style="font-size:large;font-weight:bold" href="' . Url::fromPath('trapdirector/settings/updateschema') .'?msgok=1">Click here to update</a>';
+			  echo '<br>';
+			  return;
+		  }
 	  }
 	  
 	  $Trap->setLogging($debug_level,'display');
@@ -449,50 +449,50 @@ 
   
   private function checkSnmpTrapd()
   {
-      $psOutput=array();
-      // First check is someone is listening to port 162. As not root, we can't have pid... 
-      $sspath = exec('which ss 2>/dev/null');
-      if(empty($sspath))
-      {
-          // RHEL based systems
-          $sspath = '/usr/sbin/ss';
-      }
-      if(!is_executable("$sspath"))
-      {
-          return array(1,"Can not execute $sspath");
-      }
-      exec("$sspath -lun | grep ':162 '",$psOutput);
-      if (count($psOutput) == 0)
-      {
-          return array(1,'Port UDP/162 is not open : is snmptrapd running?');
-      }
-      $psOutput=array();
-      $selinux_state = '';
-      if(is_executable('/usr/sbin/getenforce'))
-      {
-          $selinux_state = exec('/usr/sbin/getenforce 2>/dev/null');
-      }
-      if($selinux_state !== 'Enforcing')
-      {
-          exec('ps --no-headers -o command -C snmptrapd',$psOutput);
-          if (count($psOutput) == 0)
-          {
-              return array(1,"UDP/162 : OK, but no snmptrapd process (?)");
-          }
-          // Assume there is only one line... TODO : see if there is a better way to do this
-          $line = preg_replace('/^.*snmptrapd /','',$psOutput[0]);
-          if (!preg_match('/-n/',$line))
-              return array(1,'snmptrapd has no -n option : '.$line);
-          if (!preg_match('/-O[^ ]*n/',$line))
-              return array(1,'snmptrapd has no -On option : '.$line);
-          if (!preg_match('/-O[^ ]*e/',$line))
-              return array(1,'snmptrapd has no -Oe option : '.$line);
+	  $psOutput=array();
+	  // First check is someone is listening to port 162. As not root, we can't have pid... 
+	  $sspath = exec('which ss 2>/dev/null');
+	  if(empty($sspath))
+	  {
+		  // RHEL based systems
+		  $sspath = '/usr/sbin/ss';
+	  }
+	  if(!is_executable("$sspath"))
+	  {
+		  return array(1,"Can not execute $sspath");
+	  }
+	  exec("$sspath -lun | grep ':162 '",$psOutput);
+	  if (count($psOutput) == 0)
+	  {
+		  return array(1,'Port UDP/162 is not open : is snmptrapd running?');
+	  }
+	  $psOutput=array();
+	  $selinux_state = '';
+	  if(is_executable('/usr/sbin/getenforce'))
+	  {
+		  $selinux_state = exec('/usr/sbin/getenforce 2>/dev/null');
+	  }
+	  if($selinux_state !== 'Enforcing')
+	  {
+		  exec('ps --no-headers -o command -C snmptrapd',$psOutput);
+		  if (count($psOutput) == 0)
+		  {
+			  return array(1,"UDP/162 : OK, but no snmptrapd process (?)");
+		  }
+		  // Assume there is only one line... TODO : see if there is a better way to do this
+		  $line = preg_replace('/^.*snmptrapd /','',$psOutput[0]);
+		  if (!preg_match('/-n/',$line))
+			  return array(1,'snmptrapd has no -n option : '.$line);
+		  if (!preg_match('/-O[^ ]*n/',$line))
+			  return array(1,'snmptrapd has no -On option : '.$line);
+		  if (!preg_match('/-O[^ ]*e/',$line))
+			  return array(1,'snmptrapd has no -Oe option : '.$line);
 
-          return array(0,'snmptrapd listening to UDP/162, options : '.$line);
-      }
-      else
-      {
-          return array(0,'A daemon (hidden by SELinux) is listening on UDP/162');
-      }
+		  return array(0,'snmptrapd listening to UDP/162, options : '.$line);
+	  }
+	  else
+	  {
+		  return array(0,'A daemon (hidden by SELinux) is listening on UDP/162');
+	  }
   }
 }