TrapsController::countTrap() - Code Metrics - Inspection of "Push Trap DB query in class" - patrickpr/trapdirector - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( 3d7e39...d44cfc )

by Patrick

created 2020-02-13 16:00 UTC

TrapsController::countTrap() A

↳ Parent: Project

Complexity

Conditions	6
Paths	13

Size

Total Lines	24
Code Lines	16

Duplication

Lines	0
Ratio	0 %

Importance

Changes	2
Bugs	0	Features	0

Metric	Value
cc	6
eloc	16
c	2
b	0
f	0
nc	13
nop	2
dl	0
loc	24
rs	9.1111

<?php

namespace Icinga\Module\Trapdirector;

use Icinga\Web\Controller;

use Icinga\Data\Paginatable;

use Exception;

use Icinga\Module\Trapdirector\Config\TrapModuleConfig;
use Icinga\Module\Trapdirector\Tables\TrapTableList;
use Icinga\Module\Trapdirector\Tables\TrapTableHostList;
use Icinga\Module\Trapdirector\Tables\HandlerTableList;
use Icinga\Module\Trapdirector\Config\MIBLoader;
use Icinga\Module\Trapdirector\TrapsActions\UIDatabase;

use Trapdirector\Trap;

use Icinga\Data\ConfigObject;
filter:
    dependency_paths: ["lib/*"]

class TrapsController extends Controller
{
	/** @var TrapModuleConfig $moduleConfig TrapModuleConfig instance */
	protected $moduleConfig;
	/** @var TrapTableList $trapTableList (by date)*/
	protected $trapTableList;
	/** @var TrapTableHostList $trapTableHostList TrapTableList (by hosts)*/
	protected $trapTableHostList;
	/** @var HandlerTableList $handlerTableList HandlerTableList instance*/
	protected $handlerTableList;
	/** @var ConfigObject $trapDB Trap database */
	protected $trapDB;
	/** @var ConfigObject $icingaDB Icinga IDO database */
	protected $icingaDB;
	/** @var MIBLoader $MIBData MIBLoader class */
	protected $MIBData;
	/** @var Trap $trapClass Trap class for bin/trap_class.php */
	protected $trapClass;
	/** @var UIDatabase $UIDatabase */
	protected $UIDatabase;
	
	
	
	/** Get instance of TrapModuleConfig class
	*	@return TrapModuleConfig
	*/
	public function getModuleConfig() 
	{
		if ($this->moduleConfig == Null) 
		{
			$db_prefix=$this->Config()->get('config', 'database_prefix');
			if ($db_prefix === null) 
			{
				$this->redirectNow('trapdirector/settings?message=No database prefix');
			}
			$this->moduleConfig = new TrapModuleConfig($db_prefix);
		}
		return $this->moduleConfig;
	}
	
	/**
	 * Get instance of TrapTableList
	 * @return \Icinga\Module\Trapdirector\Tables\TrapTableList
	 */
	public function getTrapListTable() {
		if ($this->trapTableList == Null) {
			$this->trapTableList = new TrapTableList();
			$this->trapTableList->setConfig($this->getModuleConfig());
		}
		return $this->trapTableList;
	}
	
	/**
	 * @return \Icinga\Module\Trapdirector\Tables\TrapTableHostList
	 */
	public function getTrapHostListTable()
	{
	    if ($this->trapTableHostList == Null) 
		{
	        $this->trapTableHostList = new TrapTableHostList();
	        $this->trapTableHostList->setConfig($this->getModuleConfig());
	    }
	    return $this->trapTableHostList;
	}
	
	/**
	 * @return \Icinga\Module\Trapdirector\Tables\HandlerTableList
	 */
	public function getHandlerListTable() 
	{
		if ($this->handlerTableList == Null) 
		{
			$this->handlerTableList = new HandlerTableList();
			$this->handlerTableList->setConfig($this->getModuleConfig());
		}
		return $this->handlerTableList;
	}	

	/**
	 * @return UIDatabase
	 */
	public function getUIDatabase()
	{
	    if ($this->UIDatabase == Null)
	    {
	        $this->UIDatabase = new UIDatabase($this);
	       
	    }
	    return $this->UIDatabase;
	}
	
    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);

        return $paginatable;
    }	
	
	public function displayExitError($source,$message)
	{	// TODO : check better ways to transmit data (with POST ?)
		$this->redirectNow('trapdirector/error?source='.$source.'&message='.$message);
	}
	
	protected function checkReadPermission()
	{
        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');
        }		
	}
	
    /**
     * 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;
	}

	/*************************  Trap class get **********************/
	public function getTrapClass()
	{ // TODO : try/catch here ? or within caller
		if ($this->trapClass == null)
		{
			require_once($this->Module()->getBaseDir() .'/bin/trap_class.php');
			$icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
			//$debug_level=4;
			$this->trapClass = new Trap($icingaweb2_etc);
			//$Trap->setLogging($debug_level,'syslog');
		}
		return $this->trapClass;
	}
	
	/************************** MIB related **************************/
	
	/** Get 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');
			$this->MIBData=new MIBLoader(
				$this->Config()->get('config', 'snmptranslate'),
				$this->Config()->get('config', 'snmptranslate_dirs'),
			    $dbConn,
				$this->getModuleConfig()
			);
		}
		return $this->MIBData;
	}	
	
	/**************************  Database queries *******************/
	
	/** 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)
	*/
	protected 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->getUIDatabase()->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);
	}

	/** Get host(s) by name in IDO database
	*	does not catch exceptions
	*	@return array of objects ( name, id (object_id), display_name)
	*/
	protected 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->getUIDatabase()->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);
	}	
	
	/** Get host groups by  name in IDO database
	*	does not catch exceptions
	*	@return array of objects ( name, id (object_id), display_name)
	*/
	protected 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->getUIDatabase()->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)
	*/
	protected function getHostInfoByID($id) 
	{
		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
		$dbConn = $this->getUIDatabase()->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,  )
	*/
	protected function getHostByObjectID($id) // TODO : duplicate of getHostInfoByID above
	{
		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
		$dbConn = $this->getUIDatabase()->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
	*/
	protected 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->getUIDatabase()->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
	*/
	protected function getServicesByHostGroupid($id) 
	{		
		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
		$dbConn = $this->getUIDatabase()->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']));
			}
		}
		
		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
	*/
	protected function getServiceIDByName($hostname,$name) 
	{
	    $dbConn = $this->getUIDatabase()->getIdoDbConn();
	    if ($dbConn === null) throw new \ErrorException('uncatched db error');
	    
		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');

		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)
	*/
	protected 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->getUIDatabase()->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');

		return $dbConn->fetchRow($query);
	}		
	
	/** Check if director is installed
	*	@return bool true/false
	*/
	protected function isDirectorInstalled()
	{
	    $output=array();
	    exec('icingacli module list',$output);
	    foreach ($output as $line)
		{
			if (preg_match('/^director .*enabled/',$line))
			{
				return true;
			}
		}
		return false;
	}
	
}



1			<?php
2
3			namespace Icinga\Module\Trapdirector;
4
5			use Icinga\Web\Controller;
6
7			use Icinga\Data\Paginatable;
8
9			use Exception;
10
11			use Icinga\Module\Trapdirector\Config\TrapModuleConfig;
12			use Icinga\Module\Trapdirector\Tables\TrapTableList;
13			use Icinga\Module\Trapdirector\Tables\TrapTableHostList;
14			use Icinga\Module\Trapdirector\Tables\HandlerTableList;
15			use Icinga\Module\Trapdirector\Config\MIBLoader;
16			use Icinga\Module\Trapdirector\TrapsActions\UIDatabase;
17
18			use Trapdirector\Trap;
19
20			use Icinga\Data\ConfigObject;
			1 ignored issue – show Bug introduced 2020-02-11 19:59 UTC by Report Bug Copy Issue Report The type `Icinga\Data\ConfigObject` was not found. Maybe you did not declare it correctly or list all dependencies? The issue could also be caused by a filter entry in the build configuration. If the path has been excluded in your configuration, e.g. `excluded_paths: ["lib/"]`, you can move it to the dependency path list as follows: filter: dependency_paths: ["lib/"] For further information see https://scrutinizer-ci.com/docs/tools/php/php-scrutinizer/#list-dependency-paths Loading history...
21
22			class TrapsController extends Controller
23			{
24			/** @var TrapModuleConfig $moduleConfig TrapModuleConfig instance */
25			protected $moduleConfig;
26			/** @var TrapTableList $trapTableList (by date)*/
27			protected $trapTableList;
28			/** @var TrapTableHostList $trapTableHostList TrapTableList (by hosts)*/
29			protected $trapTableHostList;
30			/** @var HandlerTableList $handlerTableList HandlerTableList instance*/
31			protected $handlerTableList;
32			/** @var ConfigObject $trapDB Trap database */
33			protected $trapDB;
34			/** @var ConfigObject $icingaDB Icinga IDO database */
35			protected $icingaDB;
36			/** @var MIBLoader $MIBData MIBLoader class */
37			protected $MIBData;
38			/** @var Trap $trapClass Trap class for bin/trap_class.php */
39			protected $trapClass;
40			/** @var UIDatabase $UIDatabase */
41			protected $UIDatabase;
42
43
44
45			/** Get instance of TrapModuleConfig class
46			* @return TrapModuleConfig
47			*/
48			public function getModuleConfig()
49			{
50			if ($this->moduleConfig == Null)
51			{
52			$db_prefix=$this->Config()->get('config', 'database_prefix');
53			if ($db_prefix === null)
54			{
55			$this->redirectNow('trapdirector/settings?message=No database prefix');
56			}
57			$this->moduleConfig = new TrapModuleConfig($db_prefix);
58			}
59			return $this->moduleConfig;
60			}
61
62			/**
63			* Get instance of TrapTableList
64			* @return \Icinga\Module\Trapdirector\Tables\TrapTableList
65			*/
66			public function getTrapListTable() {
67			if ($this->trapTableList == Null) {
68			$this->trapTableList = new TrapTableList();
69			$this->trapTableList->setConfig($this->getModuleConfig());
70			}
71			return $this->trapTableList;
72			}
73
74			/**
75			* @return \Icinga\Module\Trapdirector\Tables\TrapTableHostList
76			*/
77			public function getTrapHostListTable()
78			{
79			if ($this->trapTableHostList == Null)
80			{
81			$this->trapTableHostList = new TrapTableHostList();
82			$this->trapTableHostList->setConfig($this->getModuleConfig());
83			}
84			return $this->trapTableHostList;
85			}
86
87			/**
88			* @return \Icinga\Module\Trapdirector\Tables\HandlerTableList
89			*/
90			public function getHandlerListTable()
91			{
92			if ($this->handlerTableList == Null)
93			{
94			$this->handlerTableList = new HandlerTableList();
95			$this->handlerTableList->setConfig($this->getModuleConfig());
96			}
97			return $this->handlerTableList;
98			}
99
100			/**
101			* @return UIDatabase
102			*/
103			public function getUIDatabase()
104			{
105			if ($this->UIDatabase == Null)
106			{
107			$this->UIDatabase = new UIDatabase($this);
108
109			}
110			return $this->UIDatabase;
111			}
112
113			protected function applyPaginationLimits(Paginatable $paginatable, $limit = 25, $offset = null)
114			{
115			$limit = $this->params->get('limit', $limit);
116			$page = $this->params->get('page', $offset);
117
118			$paginatable->limit($limit, $page > 0 ? ($page - 1) * $limit : 0);
119
120			return $paginatable;
121			}
122
123			public function displayExitError($source,$message)
124			{ // TODO : check better ways to transmit data (with POST ?)
125			$this->redirectNow('trapdirector/error?source='.$source.'&message='.$message);
126			}
127
128			protected function checkReadPermission()
129			{
130			if (! $this->Auth()->hasPermission('trapdirector/view')) {
131			$this->displayExitError('Permissions','No permission fo view content');
132			}
133			}
134
135			protected function checkConfigPermission()
136			{
137			if (! $this->Auth()->hasPermission('trapdirector/config')) {
138			$this->displayExitError('Permissions','No permission fo configure');
139			}
140			}
141
142			/**
143			* Check if user has write permission
144			* @param number $check optional : if set to 1, return true (user has permission) or false instead of displaying error page
145			* @return boolean : user has permission
146			*/
147			protected function checkModuleConfigPermission($check=0)
148			{
149			if (! $this->Auth()->hasPermission('trapdirector/module_config')) {
150			if ($check == 0)
151			{
152			$this->displayExitError('Permissions','No permission fo configure module');
153			}
154			return false;
155			}
156			return true;
157			}
158
159			/*********************** Trap class get ********************/
160			public function getTrapClass()
161			{ // TODO : try/catch here ? or within caller
162			if ($this->trapClass == null)
163			{
164			require_once($this->Module()->getBaseDir() .'/bin/trap_class.php');
165			$icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
166			//$debug_level=4;
167			$this->trapClass = new Trap($icingaweb2_etc);
168			//$Trap->setLogging($debug_level,'syslog');
169			}
170			return $this->trapClass;
171			}
172
173			/************************ MIB related ************************/
174
175			/** Get MIBLoader class
176			* @return MIBLoader class
177			*/
178			protected function getMIB()
179			{
180			if ($this->MIBData == null)
181			{
182			$dbConn = $this->getUIDatabase()->getDbConn();
183			if ($dbConn === null) throw new \ErrorException('uncatched db error');
184			$this->MIBData=new MIBLoader(
185			$this->Config()->get('config', 'snmptranslate'),
186			$this->Config()->get('config', 'snmptranslate_dirs'),
187			$dbConn,
188			$this->getModuleConfig()
189			);
190			}
191			return $this->MIBData;
192			}
193
194			/************************ Database queries *****************/
195
196			/** Get host(s) by IP (v4 or v6) or by name in IDO database
197			* does not catch exceptions
198			* @return array of objects ( name, id (object_id), display_name)
199			*/
200			protected function getHostByIP($ip)
201			{
202			// 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
203			$dbConn = $this->getUIDatabase()->getIdoDbConn();
204			if ($dbConn === null) throw new \ErrorException('uncatched db error');
205
206			// TODO : check for SQL injections
207			$query=$dbConn->select()
208			->from(
209			array('a' => 'icinga_objects'),
210			array('name' => 'a.name1','id' => 'object_id'))
211			->join(
212			array('b' => 'icinga_hosts'),
213			'b.host_object_id=a.object_id',
214			array('display_name' => 'b.display_name'))
215			->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");
216			return $dbConn->fetchAll($query);
217			}
218
219			/** Get host(s) by name in IDO database
220			* does not catch exceptions
221			* @return array of objects ( name, id (object_id), display_name)
222			*/
223			protected function getHostByName($name)
224			{
225			// 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
226			$dbConn = $this->getUIDatabase()->getIdoDbConn();
227			if ($dbConn === null) throw new \ErrorException('uncatched db error');
228
229			// TODO : check for SQL injections
230			$query=$dbConn->select()
231			->from(
232			array('a' => 'icinga_objects'),
233			array('name' => 'a.name1','id' => 'object_id'))
234			->join(
235			array('b' => 'icinga_hosts'),
236			'b.host_object_id=a.object_id',
237			array('display_name' => 'b.display_name'))
238			->where("a.name1 = '$name'");
239			return $dbConn->fetchAll($query);
240			}
241
242			/** Get host groups by name in IDO database
243			* does not catch exceptions
244			* @return array of objects ( name, id (object_id), display_name)
245			*/
246			protected function getHostGroupByName($ip)
247			{
248			// 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
249			$dbConn = $this->getUIDatabase()->getIdoDbConn();
250			if ($dbConn === null) throw new \ErrorException('uncatched db error');
251			// TODO : check for SQL injections
252			$query=$dbConn->select()
253			->from(
254			array('a' => 'icinga_objects'),
255			array('name' => 'a.name1','id' => 'object_id'))
256			->join(
257			array('b' => 'icinga_hostgroups'),
258			'b.hostgroup_object_id=a.object_id',
259			array('display_name' => 'b.alias'))
260			->where("(a.name1 LIKE '%".$ip."%' OR b.alias LIKE '%".$ip."%') and a.is_active = 1");
261			return $dbConn->fetchAll($query);
262			}
263
264
265			/** Get host IP (v4 and v6) by name in IDO database
266			* does not catch exceptions
267			* @return array ( name, display_name, ip4, ip6)
268			*/
269			protected function getHostInfoByID($id)
270			{
271			if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id'); }
272			$dbConn = $this->getUIDatabase()->getIdoDbConn();
273			if ($dbConn === null) throw new \ErrorException('uncatched db error');
274			$query=$dbConn->select()
275			->from(
276			array('a' => 'icinga_objects'),
277			array('name' => 'a.name1'))
278			->join(
279			array('b' => 'icinga_hosts'),
280			'b.host_object_id=a.object_id',
281			array('ip4' => 'b.address', 'ip6' => 'b.address6', 'display_name' => 'b.display_name'))
282			->where("a.object_id = '".$id."'");
283			return $dbConn->fetchRow($query);
284			}
285
286
287			/** Get host by objectid in IDO database
288			* does not catch exceptions
289			* @return array of objects ( id, name, display_name, ip, ip6, )
290			*/
291			protected function getHostByObjectID($id) // TODO : duplicate of getHostInfoByID above
292			{
293			if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id'); }
294			$dbConn = $this->getUIDatabase()->getIdoDbConn();
295			if ($dbConn === null) throw new \ErrorException('uncatched db error');
296			$query=$dbConn->select()
297			->from(
298			array('a' => 'icinga_objects'),
299			array('name' => 'a.name1','id' => 'a.object_id'))
300			->join(
301			array('b' => 'icinga_hosts'),
302			'b.host_object_id=a.object_id',
303			array('display_name' => 'b.display_name' , 'ip' => 'b.address', 'ip6' => 'b.address6'))
304			->where('a.object_id = ?',$id);
305			return $dbConn->fetchRow($query);
306			}
307
308			/** Get services from object ( host_object_id) in IDO database
309			* does not catch exceptions
310			* @param $id int object_id
311			* @return array display_name (of service), service_object_id
312			*/
313			protected function getServicesByHostid($id)
314			{
315			// 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
316			if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id'); }
317			$dbConn = $this->getUIDatabase()->getIdoDbConn();
318			if ($dbConn === null) throw new \ErrorException('uncatched db error');
319			$query=$dbConn->select()
320			->from(
321			array('s' => 'icinga_services'),
322			array('name' => 's.display_name','id' => 's.service_object_id'))
323			->join(
324			array('a' => 'icinga_objects'),
325			's.service_object_id=a.object_id',
326			array('is_active'=>'a.is_active','name2'=>'a.name2'))
327			->where('s.host_object_id='.$id.' AND a.is_active = 1');
328			return $dbConn->fetchAll($query);
329			}
330
331			/** Get services from hostgroup object id ( hostgroup_object_id) in IDO database
332			* gets all hosts in hostgroup and return common services
333			* does not catch exceptions
334			* @param $id int object_id
335			* @return array display_name (of service), service_object_id
336			*/
337			protected function getServicesByHostGroupid($id)
338			{
339			if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id'); }
340			$dbConn = $this->getUIDatabase()->getIdoDbConn();
341			if ($dbConn === null) throw new \ErrorException('uncatched db error');
342			$query=$dbConn->select()
343			->from(
344			array('s' => 'icinga_hostgroup_members'),
345			array('host_object_id' => 's.host_object_id'))
346			->join(
347			array('a' => 'icinga_hostgroups'),
348			's.hostgroup_id=a.hostgroup_id',
349			'hostgroup_object_id')
350			->where('a.hostgroup_object_id='.$id);
351			$hosts=$dbConn->fetchAll($query);
352			$common_services=array();
353			$num_hosts=count($hosts);
354			foreach ($hosts as $key => $host)
355			{ // For each host, get all services and add in common_services if not found or add counter
356			$host_services=$this->getServicesByHostid($host->host_object_id);
357			foreach($host_services as $service)
358			{
359			if (isset($common_services[$service->name2]['num']))
360			{
361			$common_services[$service->name2]['num'] +=1;
362			}
363			else
364			{
365			$common_services[$service->name2]['num']=1;
366			$common_services[$service->name2]['name']=$service->name;
367			}
368			}
369			}
370			$result=array();
371
372			//print_r($common_services);
373			foreach (array_keys($common_services) as $key)
374			{
375			if ($common_services[$key]['num'] == $num_hosts)
376			{
377			array_push($result,array($key,$common_services[$key]['name']));
378			}
379			}
380
381			return $result;
382			}
383
384			/** Get services object id by host name / service name in IDO database
385			* does not catch exceptions
386			* @param $hostname string host name
387			* @param $name string service name
388			* @return array service id
389			*/
390			protected function getServiceIDByName($hostname,$name)
391			{
392			$dbConn = $this->getUIDatabase()->getIdoDbConn();
393			if ($dbConn === null) throw new \ErrorException('uncatched db error');
394
395			if ($name == null)
396			{
397			return array();
398			}
399
400			$query=$dbConn->select()
401			->from(
402			array('s' => 'icinga_services'),
403			array('name' => 's.display_name','id' => 's.service_object_id'))
404			->join(
405			array('a' => 'icinga_objects'),
406			's.service_object_id=a.object_id',
407			'is_active')
408			->where('a.name2=\''.$name.'\' AND a.name1=\''.$hostname.'\' AND a.is_active = 1');
409
410			return $dbConn->fetchAll($query);
411			}
412
413			/** Get object name from object_id in IDO database
414			* does not catch exceptions
415			* @param int $id object_id (default to null, used first if not null)
416			* @return array name1 (host) name2 (service)
417			*/
418			protected function getObjectNameByid($id)
419			{
420			// 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
421			if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id'); }
422			$dbConn = $this->getUIDatabase()->getIdoDbConn();
423			if ($dbConn === null) throw new \ErrorException('uncatched db error');
424
425			$query=$dbConn->select()
426			->from(
427			array('a' => 'icinga_objects'),
428			array('name1' => 'a.name1','name2' => 'a.name2'))
429			->where('a.object_id='.$id.' AND a.is_active = 1');
430
431			return $dbConn->fetchRow($query);
432			}
433
434			/** Check if director is installed
435			* @return bool true/false
436			*/
437			protected function isDirectorInstalled()
438			{
439			$output=array();
440			exec('icingacli module list',$output);
441			foreach ($output as $line)
442			{
443			if (preg_match('/^director .*enabled/',$line))
444			{
445			return true;
446			}
447			}
448			return false;
449			}
450
451			}
452
453

patrickpr / trapdirector

Push — master ( 3d7e39...d44cfc )

TrapsController::countTrap() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like