Passed
Push — master ( e15d84...f89732 )
by Patrick
02:08
created
library/Trapdirector/TrapsActions/TrapDBQuery.php 2 patches
Indentation   +175 added lines, -175 removed lines patch added patch discarded remove patch
@@ -19,207 +19,207 @@
 block discarded – undo
19 19
 trait TrapDBQuery
20 20
 {
21 21
     
22
-    /** @return TrapsController */
23
-    abstract protected function getTrapCtrl();
22
+	/** @return TrapsController */
23
+	abstract protected function getTrapCtrl();
24 24
 
25
-    /** @return Zend_Db_Adapter_Abstract : returns DB connexion or null on error */
26
-    abstract public function getDbConn();
25
+	/** @return Zend_Db_Adapter_Abstract : returns DB connexion or null on error */
26
+	abstract public function getDbConn();
27 27
     
28
-    /** Add handler rule in traps DB
29
-     *	@param array $params : array(<db item>=><value>)
30
-     *	@return int inserted id
31
-     */
32
-    public function addHandlerRule($params)
33
-    {
34
-        // TODO Check for rule consistency
28
+	/** Add handler rule in traps DB
29
+	 *	@param array $params : array(<db item>=><value>)
30
+	 *	@return int inserted id
31
+	 */
32
+	public function addHandlerRule($params)
33
+	{
34
+		// TODO Check for rule consistency
35 35
         
36
-        $dbConn = $this->getDbConn();
37
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
38
-        // Add last modified date = creation date and username
39
-        $params['created'] = new Zend_Db_Expr('NOW()');
40
-        $params['modified'] = new 	Zend_Db_Expr('NOW()');
41
-        $params['modifier'] = $this->getTrapCtrl()->Auth()->getUser()->getUsername();
36
+		$dbConn = $this->getDbConn();
37
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
38
+		// Add last modified date = creation date and username
39
+		$params['created'] = new Zend_Db_Expr('NOW()');
40
+		$params['modified'] = new 	Zend_Db_Expr('NOW()');
41
+		$params['modifier'] = $this->getTrapCtrl()->Auth()->getUser()->getUsername();
42 42
         
43
-        $query=$dbConn->insert(
44
-            $this->getTrapCtrl()->getModuleConfig()->getTrapRuleName(),
45
-            $params
46
-            );
47
-        if($query==false)
48
-        {
49
-            return null;
50
-        }
51
-        return $dbConn->lastInsertId();
52
-    }
43
+		$query=$dbConn->insert(
44
+			$this->getTrapCtrl()->getModuleConfig()->getTrapRuleName(),
45
+			$params
46
+			);
47
+		if($query==false)
48
+		{
49
+			return null;
50
+		}
51
+		return $dbConn->lastInsertId();
52
+	}
53 53
     
54
-    /** Update handler rule in traps DB
55
-     *	@param array $params : (<db item>=><value>)
56
-     *   @param integer $ruleID : rule id in db
57
-     *	@return array affected rows
58
-     */
59
-    public function updateHandlerRule($params,$ruleID)
60
-    {
61
-        // TODO Check for rule consistency
62
-        $dbConn = $this->getDbConn();
63
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
64
-        // Add last modified date = creation date and username
65
-        $params['modified'] = new 	Zend_Db_Expr('NOW()');
66
-        $params['modifier'] = $this->getTrapCtrl()->Auth()->getUser()->getUsername();
54
+	/** Update handler rule in traps DB
55
+	 *	@param array $params : (<db item>=><value>)
56
+	 *   @param integer $ruleID : rule id in db
57
+	 *	@return array affected rows
58
+	 */
59
+	public function updateHandlerRule($params,$ruleID)
60
+	{
61
+		// TODO Check for rule consistency
62
+		$dbConn = $this->getDbConn();
63
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
64
+		// Add last modified date = creation date and username
65
+		$params['modified'] = new 	Zend_Db_Expr('NOW()');
66
+		$params['modifier'] = $this->getTrapCtrl()->Auth()->getUser()->getUsername();
67 67
         
68
-        $numRows=$dbConn->update(
69
-            $this->getTrapCtrl()->getModuleConfig()->getTrapRuleName(),
70
-            $params,
71
-            'id='.$ruleID
72
-            );
73
-        return $numRows;
74
-    }
68
+		$numRows=$dbConn->update(
69
+			$this->getTrapCtrl()->getModuleConfig()->getTrapRuleName(),
70
+			$params,
71
+			'id='.$ruleID
72
+			);
73
+		return $numRows;
74
+	}
75 75
     
76
-    /** Delete rule by id
77
-     *	@param int $ruleID rule id
78
-     */
79
-    public function deleteRule($ruleID)
80
-    {
81
-        if (!preg_match('/^[0-9]+$/',$ruleID)) { throw new Exception('Invalid id');  }
76
+	/** Delete rule by id
77
+	 *	@param int $ruleID rule id
78
+	 */
79
+	public function deleteRule($ruleID)
80
+	{
81
+		if (!preg_match('/^[0-9]+$/',$ruleID)) { throw new Exception('Invalid id');  }
82 82
         
83
-        $dbConn = $this->getDbConn();
84
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
83
+		$dbConn = $this->getDbConn();
84
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
85 85
         
86
-        $query=$dbConn->delete(
87
-            $this->getTrapCtrl()->getModuleConfig()->getTrapRuleName(),
88
-            'id='.$ruleID
89
-            );
90
-        return $query;
91
-    }
86
+		$query=$dbConn->delete(
87
+			$this->getTrapCtrl()->getModuleConfig()->getTrapRuleName(),
88
+			'id='.$ruleID
89
+			);
90
+		return $query;
91
+	}
92 92
     
93
-    /** Delete trap by ip & oid
94
-     *	@param $ipAddr string source IP (v4 or v6)
95
-     *	@param $oid string oid
96
-     */
97
-    public function deleteTrap($ipAddr,$oid)
98
-    {
93
+	/** Delete trap by ip & oid
94
+	 *	@param $ipAddr string source IP (v4 or v6)
95
+	 *	@param $oid string oid
96
+	 */
97
+	public function deleteTrap($ipAddr,$oid)
98
+	{
99 99
         
100
-        $dbConn = $this->getDbConn();
101
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
102
-        $condition=null;
103
-        if ($ipAddr != null)
104
-        {
105
-            $condition="source_ip='$ipAddr'";
106
-        }
107
-        if ($oid != null)
108
-        {
109
-            $condition=($condition===null)?'':$condition.' AND ';
110
-            $condition.="trap_oid='$oid'";
111
-        }
112
-        if($condition === null) return null;
113
-        $query=$dbConn->delete(
114
-            $this->getTrapCtrl()->getModuleConfig()->getTrapTableName(),
115
-            $condition
116
-            );
117
-        // TODO test ret code etc...
118
-        return $query;
119
-    }
100
+		$dbConn = $this->getDbConn();
101
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
102
+		$condition=null;
103
+		if ($ipAddr != null)
104
+		{
105
+			$condition="source_ip='$ipAddr'";
106
+		}
107
+		if ($oid != null)
108
+		{
109
+			$condition=($condition===null)?'':$condition.' AND ';
110
+			$condition.="trap_oid='$oid'";
111
+		}
112
+		if($condition === null) return null;
113
+		$query=$dbConn->delete(
114
+			$this->getTrapCtrl()->getModuleConfig()->getTrapTableName(),
115
+			$condition
116
+			);
117
+		// TODO test ret code etc...
118
+		return $query;
119
+	}
120 120
     
121 121
     
122
-    /** count trap by ip & oid
123
-     *	@param $ipAddr string source IP (v4 or v6)
124
-     *	@param $oid string oid
125
-     */
126
-    public function countTrap($ipAddr,$oid)
127
-    {
122
+	/** count trap by ip & oid
123
+	 *	@param $ipAddr string source IP (v4 or v6)
124
+	 *	@param $oid string oid
125
+	 */
126
+	public function countTrap($ipAddr,$oid)
127
+	{
128 128
         
129
-        $dbConn = $this->getDbConn();
130
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
129
+		$dbConn = $this->getDbConn();
130
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
131 131
         
132
-        $condition=null;
133
-        if ($ipAddr != null)
134
-        {
135
-            $condition="source_ip='$ipAddr'";
136
-        }
137
-        if ($oid != null)
138
-        {
139
-            $condition=($condition===null)?'':$condition.' AND ';
140
-            $condition.="trap_oid='$oid'";
141
-        }
142
-        if($condition === null) return 0;
143
-        $query=$dbConn->select()
144
-            ->from(
145
-                $this->getTrapCtrl()->getModuleConfig()->getTrapTableName(),
146
-                array('num'=>'count(*)'))
147
-            ->where($condition);
148
-        $returnRow=$dbConn->fetchRow($query);
149
-        return $returnRow->num;
150
-    }
132
+		$condition=null;
133
+		if ($ipAddr != null)
134
+		{
135
+			$condition="source_ip='$ipAddr'";
136
+		}
137
+		if ($oid != null)
138
+		{
139
+			$condition=($condition===null)?'':$condition.' AND ';
140
+			$condition.="trap_oid='$oid'";
141
+		}
142
+		if($condition === null) return 0;
143
+		$query=$dbConn->select()
144
+			->from(
145
+				$this->getTrapCtrl()->getModuleConfig()->getTrapTableName(),
146
+				array('num'=>'count(*)'))
147
+			->where($condition);
148
+		$returnRow=$dbConn->fetchRow($query);
149
+		return $returnRow->num;
150
+	}
151 151
     
152
-    /** get configuration value
153
-     *	@param string $element : configuration name in db
154
-     */
155
-    public function getDBConfigValue($element)
156
-    {
152
+	/** get configuration value
153
+	 *	@param string $element : configuration name in db
154
+	 */
155
+	public function getDBConfigValue($element)
156
+	{
157 157
         
158
-        $dbConn = $this->getDbConn();
159
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
158
+		$dbConn = $this->getDbConn();
159
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
160 160
         
161
-        $query=$dbConn->select()
162
-        ->from(
163
-            $this->getTrapCtrl()->getModuleConfig()->getDbConfigTableName(),
164
-            array('value'=>'value'))
165
-            ->where('name=?',$element);
166
-            $returnRow=$dbConn->fetchRow($query);
167
-            if ($returnRow==null)  // value does not exists
168
-            {
169
-                $default=$this->getTrapCtrl()->getModuleConfig()->getDBConfigDefaults();
170
-                if ( ! isset($default[$element])) return null; // no default and not value
161
+		$query=$dbConn->select()
162
+		->from(
163
+			$this->getTrapCtrl()->getModuleConfig()->getDbConfigTableName(),
164
+			array('value'=>'value'))
165
+			->where('name=?',$element);
166
+			$returnRow=$dbConn->fetchRow($query);
167
+			if ($returnRow==null)  // value does not exists
168
+			{
169
+				$default=$this->getTrapCtrl()->getModuleConfig()->getDBConfigDefaults();
170
+				if ( ! isset($default[$element])) return null; // no default and not value
171 171
                 
172
-                $this->addDBConfigValue($element,$default[$element]);
173
-                return $default[$element];
174
-            }
175
-            if ($returnRow->value == null) // value id empty
176
-            {
177
-                $default=$this->getTrapCtrl()->getModuleConfig()->getDBConfigDefaults();
178
-                if ( ! isset($default[$element])) return null; // no default and not value
179
-                $this->setDBConfigValue($element,$default[$element]);
180
-                return $default[$element];
181
-            }
182
-            return $returnRow->value;
183
-    }
172
+				$this->addDBConfigValue($element,$default[$element]);
173
+				return $default[$element];
174
+			}
175
+			if ($returnRow->value == null) // value id empty
176
+			{
177
+				$default=$this->getTrapCtrl()->getModuleConfig()->getDBConfigDefaults();
178
+				if ( ! isset($default[$element])) return null; // no default and not value
179
+				$this->setDBConfigValue($element,$default[$element]);
180
+				return $default[$element];
181
+			}
182
+			return $returnRow->value;
183
+	}
184 184
     
185
-    /** add configuration value
186
-     *	@param string $element : name of config element
187
-     *   @param string $value : value
188
-     */
185
+	/** add configuration value
186
+	 *	@param string $element : name of config element
187
+	 *   @param string $value : value
188
+	 */
189 189
     
190
-    public function addDBConfigValue($element,$value)
191
-    {
190
+	public function addDBConfigValue($element,$value)
191
+	{
192 192
         
193
-        $dbConn = $this->getDbConn();
194
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
193
+		$dbConn = $this->getDbConn();
194
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
195 195
         
196
-        $query=$dbConn->insert(
197
-            $this->getTrapCtrl()->getModuleConfig()->getDbConfigTableName(),
198
-            array(
199
-                'name' => $element,
200
-                'value'=>$value
201
-            )
202
-            );
203
-        return $query;
204
-    }
196
+		$query=$dbConn->insert(
197
+			$this->getTrapCtrl()->getModuleConfig()->getDbConfigTableName(),
198
+			array(
199
+				'name' => $element,
200
+				'value'=>$value
201
+			)
202
+			);
203
+		return $query;
204
+	}
205 205
     
206
-    /** set configuration value
207
-     *	@param string $element : name of config element
208
-     *   @param string $value : value
209
-     */
210
-    public function setDBConfigValue($element,$value)
211
-    {
206
+	/** set configuration value
207
+	 *	@param string $element : name of config element
208
+	 *   @param string $value : value
209
+	 */
210
+	public function setDBConfigValue($element,$value)
211
+	{
212 212
         
213
-        $dbConn = $this->getDbConn();
214
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
213
+		$dbConn = $this->getDbConn();
214
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
215 215
         
216
-        $query=$dbConn->update(
217
-            $this->getTrapCtrl()->getModuleConfig()->getDbConfigTableName(),
218
-            array('value'=>$value),
219
-            'name=\''.$element.'\''
220
-            );
221
-        return $query;
222
-    }
216
+		$query=$dbConn->update(
217
+			$this->getTrapCtrl()->getModuleConfig()->getDbConfigTableName(),
218
+			array('value'=>$value),
219
+			'name=\''.$element.'\''
220
+			);
221
+		return $query;
222
+	}
223 223
     
224 224
     
225 225
 }
226 226
\ No newline at end of file
Please login to merge, or discard this patch.
Spacing   +30 added lines, -30 removed lines patch added patch discarded remove patch
@@ -33,18 +33,18 @@  discard block
 block discarded – undo
33 33
     {
34 34
         // TODO Check for rule consistency
35 35
         
36
-        $dbConn = $this->getDbConn();
36
+        $dbConn=$this->getDbConn();
37 37
         if ($dbConn === null) throw new \ErrorException('uncatched db error');
38 38
         // Add last modified date = creation date and username
39
-        $params['created'] = new Zend_Db_Expr('NOW()');
40
-        $params['modified'] = new 	Zend_Db_Expr('NOW()');
41
-        $params['modifier'] = $this->getTrapCtrl()->Auth()->getUser()->getUsername();
39
+        $params['created']=new Zend_Db_Expr('NOW()');
40
+        $params['modified']=new 	Zend_Db_Expr('NOW()');
41
+        $params['modifier']=$this->getTrapCtrl()->Auth()->getUser()->getUsername();
42 42
         
43 43
         $query=$dbConn->insert(
44 44
             $this->getTrapCtrl()->getModuleConfig()->getTrapRuleName(),
45 45
             $params
46 46
             );
47
-        if($query==false)
47
+        if ($query == false)
48 48
         {
49 49
             return null;
50 50
         }
@@ -56,14 +56,14 @@  discard block
 block discarded – undo
56 56
      *   @param integer $ruleID : rule id in db
57 57
      *	@return array affected rows
58 58
      */
59
-    public function updateHandlerRule($params,$ruleID)
59
+    public function updateHandlerRule($params, $ruleID)
60 60
     {
61 61
         // TODO Check for rule consistency
62
-        $dbConn = $this->getDbConn();
62
+        $dbConn=$this->getDbConn();
63 63
         if ($dbConn === null) throw new \ErrorException('uncatched db error');
64 64
         // Add last modified date = creation date and username
65
-        $params['modified'] = new 	Zend_Db_Expr('NOW()');
66
-        $params['modifier'] = $this->getTrapCtrl()->Auth()->getUser()->getUsername();
65
+        $params['modified']=new 	Zend_Db_Expr('NOW()');
66
+        $params['modifier']=$this->getTrapCtrl()->Auth()->getUser()->getUsername();
67 67
         
68 68
         $numRows=$dbConn->update(
69 69
             $this->getTrapCtrl()->getModuleConfig()->getTrapRuleName(),
@@ -78,9 +78,9 @@  discard block
 block discarded – undo
78 78
      */
79 79
     public function deleteRule($ruleID)
80 80
     {
81
-        if (!preg_match('/^[0-9]+$/',$ruleID)) { throw new Exception('Invalid id');  }
81
+        if (!preg_match('/^[0-9]+$/', $ruleID)) { throw new Exception('Invalid id'); }
82 82
         
83
-        $dbConn = $this->getDbConn();
83
+        $dbConn=$this->getDbConn();
84 84
         if ($dbConn === null) throw new \ErrorException('uncatched db error');
85 85
         
86 86
         $query=$dbConn->delete(
@@ -94,10 +94,10 @@  discard block
 block discarded – undo
94 94
      *	@param $ipAddr string source IP (v4 or v6)
95 95
      *	@param $oid string oid
96 96
      */
97
-    public function deleteTrap($ipAddr,$oid)
97
+    public function deleteTrap($ipAddr, $oid)
98 98
     {
99 99
         
100
-        $dbConn = $this->getDbConn();
100
+        $dbConn=$this->getDbConn();
101 101
         if ($dbConn === null) throw new \ErrorException('uncatched db error');
102 102
         $condition=null;
103 103
         if ($ipAddr != null)
@@ -106,10 +106,10 @@  discard block
 block discarded – undo
106 106
         }
107 107
         if ($oid != null)
108 108
         {
109
-            $condition=($condition===null)?'':$condition.' AND ';
109
+            $condition=($condition === null) ? '' : $condition.' AND ';
110 110
             $condition.="trap_oid='$oid'";
111 111
         }
112
-        if($condition === null) return null;
112
+        if ($condition === null) return null;
113 113
         $query=$dbConn->delete(
114 114
             $this->getTrapCtrl()->getModuleConfig()->getTrapTableName(),
115 115
             $condition
@@ -123,10 +123,10 @@  discard block
 block discarded – undo
123 123
      *	@param $ipAddr string source IP (v4 or v6)
124 124
      *	@param $oid string oid
125 125
      */
126
-    public function countTrap($ipAddr,$oid)
126
+    public function countTrap($ipAddr, $oid)
127 127
     {
128 128
         
129
-        $dbConn = $this->getDbConn();
129
+        $dbConn=$this->getDbConn();
130 130
         if ($dbConn === null) throw new \ErrorException('uncatched db error');
131 131
         
132 132
         $condition=null;
@@ -136,10 +136,10 @@  discard block
 block discarded – undo
136 136
         }
137 137
         if ($oid != null)
138 138
         {
139
-            $condition=($condition===null)?'':$condition.' AND ';
139
+            $condition=($condition === null) ? '' : $condition.' AND ';
140 140
             $condition.="trap_oid='$oid'";
141 141
         }
142
-        if($condition === null) return 0;
142
+        if ($condition === null) return 0;
143 143
         $query=$dbConn->select()
144 144
             ->from(
145 145
                 $this->getTrapCtrl()->getModuleConfig()->getTrapTableName(),
@@ -155,28 +155,28 @@  discard block
 block discarded – undo
155 155
     public function getDBConfigValue($element)
156 156
     {
157 157
         
158
-        $dbConn = $this->getDbConn();
158
+        $dbConn=$this->getDbConn();
159 159
         if ($dbConn === null) throw new \ErrorException('uncatched db error');
160 160
         
161 161
         $query=$dbConn->select()
162 162
         ->from(
163 163
             $this->getTrapCtrl()->getModuleConfig()->getDbConfigTableName(),
164 164
             array('value'=>'value'))
165
-            ->where('name=?',$element);
165
+            ->where('name=?', $element);
166 166
             $returnRow=$dbConn->fetchRow($query);
167
-            if ($returnRow==null)  // value does not exists
167
+            if ($returnRow == null)  // value does not exists
168 168
             {
169 169
                 $default=$this->getTrapCtrl()->getModuleConfig()->getDBConfigDefaults();
170
-                if ( ! isset($default[$element])) return null; // no default and not value
170
+                if (!isset($default[$element])) return null; // no default and not value
171 171
                 
172
-                $this->addDBConfigValue($element,$default[$element]);
172
+                $this->addDBConfigValue($element, $default[$element]);
173 173
                 return $default[$element];
174 174
             }
175 175
             if ($returnRow->value == null) // value id empty
176 176
             {
177 177
                 $default=$this->getTrapCtrl()->getModuleConfig()->getDBConfigDefaults();
178
-                if ( ! isset($default[$element])) return null; // no default and not value
179
-                $this->setDBConfigValue($element,$default[$element]);
178
+                if (!isset($default[$element])) return null; // no default and not value
179
+                $this->setDBConfigValue($element, $default[$element]);
180 180
                 return $default[$element];
181 181
             }
182 182
             return $returnRow->value;
@@ -187,10 +187,10 @@  discard block
 block discarded – undo
187 187
      *   @param string $value : value
188 188
      */
189 189
     
190
-    public function addDBConfigValue($element,$value)
190
+    public function addDBConfigValue($element, $value)
191 191
     {
192 192
         
193
-        $dbConn = $this->getDbConn();
193
+        $dbConn=$this->getDbConn();
194 194
         if ($dbConn === null) throw new \ErrorException('uncatched db error');
195 195
         
196 196
         $query=$dbConn->insert(
@@ -207,10 +207,10 @@  discard block
 block discarded – undo
207 207
      *	@param string $element : name of config element
208 208
      *   @param string $value : value
209 209
      */
210
-    public function setDBConfigValue($element,$value)
210
+    public function setDBConfigValue($element, $value)
211 211
     {
212 212
         
213
-        $dbConn = $this->getDbConn();
213
+        $dbConn=$this->getDbConn();
214 214
         if ($dbConn === null) throw new \ErrorException('uncatched db error');
215 215
         
216 216
         $query=$dbConn->update(
Please login to merge, or discard this patch.
library/Trapdirector/TrapsActions/UIDatabase.php 1 patch
Indentation   +260 added lines, -260 removed lines patch added patch discarded remove patch
@@ -19,30 +19,30 @@  discard block
 block discarded – undo
19 19
  */
20 20
 class DBException extends Exception
21 21
 {
22
-    /** @var array $returnArray */
23
-    private $returnArray;
22
+	/** @var array $returnArray */
23
+	private $returnArray;
24 24
     
25
-    /**
26
-     * Buil DBException
27
-     * @param array $retarray
28
-     * @param string $message
29
-     * @param int $code
30
-     * @param Exception $previous
31
-     */
32
-    public function __construct(array $retarray, string $message = null, int $code = 0, Exception $previous = null)
33
-    {
34
-        parent::__construct($message,$code,$previous);
35
-        $this->returnArray = $retarray;
36
-    }
25
+	/**
26
+	 * Buil DBException
27
+	 * @param array $retarray
28
+	 * @param string $message
29
+	 * @param int $code
30
+	 * @param Exception $previous
31
+	 */
32
+	public function __construct(array $retarray, string $message = null, int $code = 0, Exception $previous = null)
33
+	{
34
+		parent::__construct($message,$code,$previous);
35
+		$this->returnArray = $retarray;
36
+	}
37 37
     
38
-    /**
39
-     * Get exception array
40
-     * @return array
41
-     */
42
-    public function getArray()
43
-    {
44
-        return $this->returnArray;
45
-    }
38
+	/**
39
+	 * Get exception array
40
+	 * @return array
41
+	 */
42
+	public function getArray()
43
+	{
44
+		return $this->returnArray;
45
+	}
46 46
 }
47 47
 
48 48
 /**
@@ -56,278 +56,278 @@  discard block
 block discarded – undo
56 56
  */
57 57
 class UIDatabase //extends TrapDBQuery
58 58
 {
59
-    use TrapDBQuery,IdoDBQuery;
59
+	use TrapDBQuery,IdoDBQuery;
60 60
     
61
-    /** @var TrapsController $trapController TrapController 'parent' class */
62
-    protected  $trapController=null;
61
+	/** @var TrapsController $trapController TrapController 'parent' class */
62
+	protected  $trapController=null;
63 63
     
64
-    /** @var Selectable $trapDB Trap Database*/
65
-    protected $trapDB=null;
64
+	/** @var Selectable $trapDB Trap Database*/
65
+	protected $trapDB=null;
66 66
  
67
-    /** @var Selectable $trapDB Icinga IDO database*/
68
-    protected $idoDB=null;
67
+	/** @var Selectable $trapDB Icinga IDO database*/
68
+	protected $idoDB=null;
69 69
     
70
-    /** @var array $testResult */
71
-    protected $testResult;
70
+	/** @var array $testResult */
71
+	protected $testResult;
72 72
     
73
-    /**
74
-     * 
75
-     * @param TrapsController $trapCtrl
76
-     */
77
-    function __construct(TrapsController $trapCtrl)
78
-    {
79
-        $this->trapController=$trapCtrl;
80
-    }
73
+	/**
74
+	 * 
75
+	 * @param TrapsController $trapCtrl
76
+	 */
77
+	function __construct(TrapsController $trapCtrl)
78
+	{
79
+		$this->trapController=$trapCtrl;
80
+	}
81 81
     
82
-    /**
83
-     * Get TrapsController instance
84
-     * @return TrapsController
85
-     */
86
-    protected function getTrapCtrl()
87
-    {
88
-        return $this->trapController;
89
-    }
82
+	/**
83
+	 * Get TrapsController instance
84
+	 * @return TrapsController
85
+	 */
86
+	protected function getTrapCtrl()
87
+	{
88
+		return $this->trapController;
89
+	}
90 90
     
91
-    /**
92
-     * Test if database version >= min database version
93
-     * 
94
-     * @param \Zend_Db_Adapter_Abstract $dbConn
95
-     * @param int $min Minimum version
96
-     * @param bool $test Test mode
97
-     * @param string $DBname Name of DB
98
-     * @return bool true if OK, false if version < min version
99
-     * @throws Exception if error and test = true
100
-     */
101
-    protected function testDbVersion($dbAdapter,int $min,bool $test, string $DBname)
102
-    {
103
-        try
104
-        {
105
-            $query = $dbAdapter->select()
106
-            ->from($this->trapController->getModuleConfig()->getDbConfigTableName(),'value')
107
-            ->where('name=\'db_version\'');
108
-            $version=$dbAdapter->fetchRow($query);
109
-            if ( ($version == null) || ! property_exists($version,'value') )
110
-            {
111
-                if ($test === true) 
112
-                {
113
-                    $this->testResult = array(4,$DBname);
114
-                    return false;
115
-                }
116
-                $this->trapController->redirectNow('trapdirector/settings?dberror=4');
117
-                return false;
118
-            }
119
-            if ($version->value < $min)
120
-            {
121
-                if ($test === true) 
122
-                {
123
-                    $this->testResult = array(5,$version->value,$min);
124
-                    return false;
125
-                }
126
-                $this->trapController->redirectNow('trapdirector/settings?dberror=5');
127
-                return false;
128
-            }
129
-        }
130
-        catch (Exception $e)
131
-        {
132
-            if ($test === true) 
133
-            {
134
-                $this->testResult = array(3,$DBname,$e->getMessage());
135
-                return false;
136
-            }
137
-            $this->trapController->redirectNow('trapdirector/settings?dberror=4');
138
-            return false;
139
-        }
140
-        return true;
141
-    }
91
+	/**
92
+	 * Test if database version >= min database version
93
+	 * 
94
+	 * @param \Zend_Db_Adapter_Abstract $dbConn
95
+	 * @param int $min Minimum version
96
+	 * @param bool $test Test mode
97
+	 * @param string $DBname Name of DB
98
+	 * @return bool true if OK, false if version < min version
99
+	 * @throws Exception if error and test = true
100
+	 */
101
+	protected function testDbVersion($dbAdapter,int $min,bool $test, string $DBname)
102
+	{
103
+		try
104
+		{
105
+			$query = $dbAdapter->select()
106
+			->from($this->trapController->getModuleConfig()->getDbConfigTableName(),'value')
107
+			->where('name=\'db_version\'');
108
+			$version=$dbAdapter->fetchRow($query);
109
+			if ( ($version == null) || ! property_exists($version,'value') )
110
+			{
111
+				if ($test === true) 
112
+				{
113
+					$this->testResult = array(4,$DBname);
114
+					return false;
115
+				}
116
+				$this->trapController->redirectNow('trapdirector/settings?dberror=4');
117
+				return false;
118
+			}
119
+			if ($version->value < $min)
120
+			{
121
+				if ($test === true) 
122
+				{
123
+					$this->testResult = array(5,$version->value,$min);
124
+					return false;
125
+				}
126
+				$this->trapController->redirectNow('trapdirector/settings?dberror=5');
127
+				return false;
128
+			}
129
+		}
130
+		catch (Exception $e)
131
+		{
132
+			if ($test === true) 
133
+			{
134
+				$this->testResult = array(3,$DBname,$e->getMessage());
135
+				return false;
136
+			}
137
+			$this->trapController->redirectNow('trapdirector/settings?dberror=4');
138
+			return false;
139
+		}
140
+		return true;
141
+	}
142 142
     
143
-    /**	Get Database connexion
144
-     *	@param $DBname string DB name in resource.ini_ge
145
-     *	@param $test bool if set to true, returns error code and not database
146
-     *	@param $test_version bool if set to flase, does not test database version of trapDB
147
-     *  @throws DBException if test = true and error
148
-     *	@return Selectable|null : if test=false, returns DB connexion, else array(error_num,message) or null on error.
149
-     */
150
-    protected function getDbByName($DBname , $test = false , $test_version = true)
151
-    {
152
-        try
153
-        {
154
-            $dbconn = IcingaDbConnection::fromResourceName($DBname);
155
-        }
156
-        catch (Exception $e)
157
-        {
158
-            if ($test === true) 
159
-            {
160
-                throw new DBException(array(2,$DBname));
161
-            }
162
-            $this->trapController->redirectNow('trapdirector/settings?dberror=2');
163
-            return null;
164
-        }
143
+	/**	Get Database connexion
144
+	 *	@param $DBname string DB name in resource.ini_ge
145
+	 *	@param $test bool if set to true, returns error code and not database
146
+	 *	@param $test_version bool if set to flase, does not test database version of trapDB
147
+	 *  @throws DBException if test = true and error
148
+	 *	@return Selectable|null : if test=false, returns DB connexion, else array(error_num,message) or null on error.
149
+	 */
150
+	protected function getDbByName($DBname , $test = false , $test_version = true)
151
+	{
152
+		try
153
+		{
154
+			$dbconn = IcingaDbConnection::fromResourceName($DBname);
155
+		}
156
+		catch (Exception $e)
157
+		{
158
+			if ($test === true) 
159
+			{
160
+				throw new DBException(array(2,$DBname));
161
+			}
162
+			$this->trapController->redirectNow('trapdirector/settings?dberror=2');
163
+			return null;
164
+		}
165 165
         
166
-        try
167
-        {
168
-            $dbAdapter=$dbconn->getDbAdapter();
166
+		try
167
+		{
168
+			$dbAdapter=$dbconn->getDbAdapter();
169 169
             
170
-        }
171
-        catch (Exception $e)
172
-        {
173
-            if ($test === true)
174
-            {
175
-                throw new DBException(array(3,$DBname,$e->getMessage()));
176
-            }
177
-            $this->trapController->redirectNow('trapdirector/settings?dberror=3');
178
-            return null;
179
-        }
170
+		}
171
+		catch (Exception $e)
172
+		{
173
+			if ($test === true)
174
+			{
175
+				throw new DBException(array(3,$DBname,$e->getMessage()));
176
+			}
177
+			$this->trapController->redirectNow('trapdirector/settings?dberror=3');
178
+			return null;
179
+		}
180 180
         
181
-        if ($test_version == true) {
182
-            $testRet=$this->testDbVersion($dbAdapter, $this->trapController->getModuleConfig()->getDbMinVersion(), $test, $DBname);
183
-            if ($testRet !== true) 
184
-            {
185
-                throw new DBException($this->testResult);
186
-            }
187
-        }
181
+		if ($test_version == true) {
182
+			$testRet=$this->testDbVersion($dbAdapter, $this->trapController->getModuleConfig()->getDbMinVersion(), $test, $DBname);
183
+			if ($testRet !== true) 
184
+			{
185
+				throw new DBException($this->testResult);
186
+			}
187
+		}
188 188
         
189
-        return $dbconn;
190
-    }
189
+		return $dbconn;
190
+	}
191 191
 
192
-    /**
193
-     * Get Trap database
194
-     * @return Selectable|null : returns DB connexion or null on error.
195
-     */
196
-    public function getDb()
197
-    {
198
-        if ( $this->trapDB != null ) return $this->trapDB;
192
+	/**
193
+	 * Get Trap database
194
+	 * @return Selectable|null : returns DB connexion or null on error.
195
+	 */
196
+	public function getDb()
197
+	{
198
+		if ( $this->trapDB != null ) return $this->trapDB;
199 199
         
200 200
         
201
-        $dbresource=$this->trapController->Config()->get('config', 'database');
201
+		$dbresource=$this->trapController->Config()->get('config', 'database');
202 202
         
203
-        if ( ! $dbresource )
204
-        {
205
-            $this->trapController->redirectNow('trapdirector/settings?dberror=1');
206
-            return null;
207
-        }
203
+		if ( ! $dbresource )
204
+		{
205
+			$this->trapController->redirectNow('trapdirector/settings?dberror=1');
206
+			return null;
207
+		}
208 208
 
209
-        try {
210
-            $this->trapDB = $this->getDbByName($dbresource,false,true);
211
-        } catch (DBException $e) {
212
-            return null; // Should not happen as test = false
213
-        }
209
+		try {
210
+			$this->trapDB = $this->getDbByName($dbresource,false,true);
211
+		} catch (DBException $e) {
212
+			return null; // Should not happen as test = false
213
+		}
214 214
         
215
-        //$this->trapDB->getConnection();
215
+		//$this->trapDB->getConnection();
216 216
         
217
-        return $this->trapDB;
218
-    }
217
+		return $this->trapDB;
218
+	}
219 219
 
220
-    /**
221
-     * Get Zend adapter of DB.
222
-     * @return \Zend_Db_Adapter_Abstract|null
223
-     */
224
-    public function getDbConn()
225
-    {
226
-        if ($this->getDb() == null) return null;
227
-        return $this->getDb()->getConnection();
228
-    }
220
+	/**
221
+	 * Get Zend adapter of DB.
222
+	 * @return \Zend_Db_Adapter_Abstract|null
223
+	 */
224
+	public function getDbConn()
225
+	{
226
+		if ($this->getDb() == null) return null;
227
+		return $this->getDb()->getConnection();
228
+	}
229 229
     
230
-    /**
231
-     * Test Trap database
232
-     * @param boolean $test
233
-     * @throws DBException on error.
234
-     * @return \Zend_Db_Adapter_Abstract|array|null : if test=false, returns DB connexion, else array(error_num,message) or null on error.
235
-     */
236
-    public function testGetDb()
237
-    {       
238
-        $dbresource=$this->trapController->Config()->get('config', 'database');
230
+	/**
231
+	 * Test Trap database
232
+	 * @param boolean $test
233
+	 * @throws DBException on error.
234
+	 * @return \Zend_Db_Adapter_Abstract|array|null : if test=false, returns DB connexion, else array(error_num,message) or null on error.
235
+	 */
236
+	public function testGetDb()
237
+	{       
238
+		$dbresource=$this->trapController->Config()->get('config', 'database');
239 239
         
240
-        if ( ! $dbresource )
241
-        {
242
-                throw new DBException(array(1,''));
243
-        }
240
+		if ( ! $dbresource )
241
+		{
242
+				throw new DBException(array(1,''));
243
+		}
244 244
         
245
-        $this->trapDB = $this->getDbByName($dbresource,true,true);       
246
-        return;
247
-    }
245
+		$this->trapDB = $this->getDbByName($dbresource,true,true);       
246
+		return;
247
+	}
248 248
     
249 249
 
250
-    /**
251
-     * Get IDO Database
252
-     * @return \Zend_Db_Adapter_Abstract|NULL  returns DB connexion or null on error.
253
-     */
254
-    public function getIdoDb()
255
-    {
256
-        if ( $this->idoDB != null ) return $this->idoDB;
257
-        // TODO : get ido database directly from icingaweb2 config -> (or not if using only API)
258
-        $dbresource=$this->trapController->Config()->get('config', 'IDOdatabase');;
250
+	/**
251
+	 * Get IDO Database
252
+	 * @return \Zend_Db_Adapter_Abstract|NULL  returns DB connexion or null on error.
253
+	 */
254
+	public function getIdoDb()
255
+	{
256
+		if ( $this->idoDB != null ) return $this->idoDB;
257
+		// TODO : get ido database directly from icingaweb2 config -> (or not if using only API)
258
+		$dbresource=$this->trapController->Config()->get('config', 'IDOdatabase');;
259 259
         
260
-        if ( ! $dbresource )
261
-        {
262
-            $this->trapController->redirectNow('trapdirector/settings?idodberror=1');
263
-            return null;
264
-        }
260
+		if ( ! $dbresource )
261
+		{
262
+			$this->trapController->redirectNow('trapdirector/settings?idodberror=1');
263
+			return null;
264
+		}
265 265
         
266
-        try
267
-        {
268
-            $dbconn = IcingaDbConnection::fromResourceName($dbresource);
269
-        }
270
-        catch (Exception $e)
271
-        {
272
-            $this->trapController->redirectNow('trapdirector/settings?idodberror=2');
273
-            return null;
274
-        }
266
+		try
267
+		{
268
+			$dbconn = IcingaDbConnection::fromResourceName($dbresource);
269
+		}
270
+		catch (Exception $e)
271
+		{
272
+			$this->trapController->redirectNow('trapdirector/settings?idodberror=2');
273
+			return null;
274
+		}
275 275
 
276
-        $this->idoDB = $dbconn;
277
-        return $this->idoDB;
278
-    }
276
+		$this->idoDB = $dbconn;
277
+		return $this->idoDB;
278
+	}
279 279
 
280 280
 
281
-    /**
282
-     * Get Zend adapter of DB.
283
-     * @return \Zend_Db_Adapter_Abstract|null
284
-     */
285
-    public function getIdoDbConn()
286
-    {
287
-        if ($this->getIdoDb() == null) return null;
288
-        return $this->getIdoDb()->getConnection();
289
-    }
281
+	/**
282
+	 * Get Zend adapter of DB.
283
+	 * @return \Zend_Db_Adapter_Abstract|null
284
+	 */
285
+	public function getIdoDbConn()
286
+	{
287
+		if ($this->getIdoDb() == null) return null;
288
+		return $this->getIdoDb()->getConnection();
289
+	}
290 290
     
291
-    /**
292
-     * Get IDO Database
293
-     * @param boolean $test
294
-     * @throws DBException on error
295
-     */
296
-    public function testGetIdoDb()
297
-    {
298
-        // TODO : get ido database directly from icingaweb2 config -> (or not if using only API)
299
-        $dbresource=$this->trapController->Config()->get('config', 'IDOdatabase');;
291
+	/**
292
+	 * Get IDO Database
293
+	 * @param boolean $test
294
+	 * @throws DBException on error
295
+	 */
296
+	public function testGetIdoDb()
297
+	{
298
+		// TODO : get ido database directly from icingaweb2 config -> (or not if using only API)
299
+		$dbresource=$this->trapController->Config()->get('config', 'IDOdatabase');;
300 300
         
301
-        if ( ! $dbresource )
302
-        {
303
-            throw new DBException(array(1,'No database in config.ini'));
304
-        }
301
+		if ( ! $dbresource )
302
+		{
303
+			throw new DBException(array(1,'No database in config.ini'));
304
+		}
305 305
         
306
-        try
307
-        {
308
-            $dbconn = IcingaDbConnection::fromResourceName($dbresource);
309
-        }
310
-        catch (Exception $e)
311
-        {
312
-            throw new DBException( array(2,"Database $dbresource does not exists in IcingaWeb2") );
313
-        }
306
+		try
307
+		{
308
+			$dbconn = IcingaDbConnection::fromResourceName($dbresource);
309
+		}
310
+		catch (Exception $e)
311
+		{
312
+			throw new DBException( array(2,"Database $dbresource does not exists in IcingaWeb2") );
313
+		}
314 314
                
315
-        try
316
-        {
317
-            $query = $dbconn->select()
318
-            ->from('icinga_dbversion',array('version'));
319
-            $version=$dbconn->fetchRow($query);
320
-            if ( ($version == null) || ! property_exists($version,'version') )
321
-            {
322
-                throw new DBException( array(4,"$dbresource does not look like an IDO database"));
323
-            }
324
-        }
325
-        catch (Exception $e)
326
-        {
327
-            throw new DBException( array(3,"Error connecting to $dbresource : " . $e->getMessage()));
328
-        }
315
+		try
316
+		{
317
+			$query = $dbconn->select()
318
+			->from('icinga_dbversion',array('version'));
319
+			$version=$dbconn->fetchRow($query);
320
+			if ( ($version == null) || ! property_exists($version,'version') )
321
+			{
322
+				throw new DBException( array(4,"$dbresource does not look like an IDO database"));
323
+			}
324
+		}
325
+		catch (Exception $e)
326
+		{
327
+			throw new DBException( array(3,"Error connecting to $dbresource : " . $e->getMessage()));
328
+		}
329 329
         
330
-        return;
331
-    }
330
+		return;
331
+	}
332 332
     
333 333
 }
334 334
\ No newline at end of file
Please login to merge, or discard this patch.
library/Trapdirector/TrapsActions/IdoDBQuery.php 1 patch
Indentation   +222 added lines, -222 removed lines patch added patch discarded remove patch
@@ -18,247 +18,247 @@
 block discarded – undo
18 18
 trait IdoDBQuery
19 19
 {
20 20
 
21
-    /** @return TrapsController */
22
-    abstract protected function getTrapCtrl();
21
+	/** @return TrapsController */
22
+	abstract protected function getTrapCtrl();
23 23
     
24
-    /** @return Zend_Db_Adapter_Abstract : returns DB connexion or null on error */
25
-    abstract public function getIdoDbConn();
24
+	/** @return Zend_Db_Adapter_Abstract : returns DB connexion or null on error */
25
+	abstract public function getIdoDbConn();
26 26
     
27
-    /** Get host(s) by IP (v4 or v6) or by name in IDO database
28
-     *	does not catch exceptions
29
-     *	@return array of objects ( name, id (object_id), display_name)
30
-     */
31
-    public function getHostByIP($ip)
32
-    {
33
-        // 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
34
-        $dbConn = $this->getIdoDbConn();
35
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
27
+	/** Get host(s) by IP (v4 or v6) or by name in IDO database
28
+	 *	does not catch exceptions
29
+	 *	@return array of objects ( name, id (object_id), display_name)
30
+	 */
31
+	public function getHostByIP($ip)
32
+	{
33
+		// 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
34
+		$dbConn = $this->getIdoDbConn();
35
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
36 36
         
37
-        // TODO : check for SQL injections
38
-        $query=$dbConn->select()
39
-        ->from(
40
-            array('a' => 'icinga_objects'),
41
-            array('name' => 'a.name1','id' => 'object_id'))
42
-            ->join(
43
-                array('b' => 'icinga_hosts'),
44
-                'b.host_object_id=a.object_id',
45
-                array('display_name' => 'b.display_name'))
46
-                ->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");
47
-                return $dbConn->fetchAll($query);
48
-    }
37
+		// TODO : check for SQL injections
38
+		$query=$dbConn->select()
39
+		->from(
40
+			array('a' => 'icinga_objects'),
41
+			array('name' => 'a.name1','id' => 'object_id'))
42
+			->join(
43
+				array('b' => 'icinga_hosts'),
44
+				'b.host_object_id=a.object_id',
45
+				array('display_name' => 'b.display_name'))
46
+				->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");
47
+				return $dbConn->fetchAll($query);
48
+	}
49 49
     
50
-    /** Get host(s) by name in IDO database
51
-     *	does not catch exceptions
52
-     *	@return array of objects ( name, id (object_id), display_name)
53
-     */
54
-    public function getHostByName($name)
55
-    {
56
-        // 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
57
-        $dbConn = $this->getIdoDbConn();
58
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
50
+	/** Get host(s) by name in IDO database
51
+	 *	does not catch exceptions
52
+	 *	@return array of objects ( name, id (object_id), display_name)
53
+	 */
54
+	public function getHostByName($name)
55
+	{
56
+		// 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
57
+		$dbConn = $this->getIdoDbConn();
58
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
59 59
         
60
-        // TODO : check for SQL injections
61
-        $query=$dbConn->select()
62
-        ->from(
63
-            array('a' => 'icinga_objects'),
64
-            array('name' => 'a.name1','id' => 'object_id'))
65
-            ->join(
66
-                array('b' => 'icinga_hosts'),
67
-                'b.host_object_id=a.object_id',
68
-                array('display_name' => 'b.display_name'))
69
-                ->where("a.name1 = '$name'");
70
-                return $dbConn->fetchAll($query);
71
-    }
60
+		// TODO : check for SQL injections
61
+		$query=$dbConn->select()
62
+		->from(
63
+			array('a' => 'icinga_objects'),
64
+			array('name' => 'a.name1','id' => 'object_id'))
65
+			->join(
66
+				array('b' => 'icinga_hosts'),
67
+				'b.host_object_id=a.object_id',
68
+				array('display_name' => 'b.display_name'))
69
+				->where("a.name1 = '$name'");
70
+				return $dbConn->fetchAll($query);
71
+	}
72 72
     
73
-    /** Get host groups by  name in IDO database
74
-     *	does not catch exceptions
75
-     *	@return array of objects ( name, id (object_id), display_name)
76
-     */
77
-    public function getHostGroupByName($ip)
78
-    {
79
-        // 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
80
-        $dbConn = $this->getIdoDbConn();
81
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
82
-        // TODO : check for SQL injections
83
-        $query=$dbConn->select()
84
-        ->from(
85
-            array('a' => 'icinga_objects'),
86
-            array('name' => 'a.name1','id' => 'object_id'))
87
-            ->join(
88
-                array('b' => 'icinga_hostgroups'),
89
-                'b.hostgroup_object_id=a.object_id',
90
-                array('display_name' => 'b.alias'))
91
-                ->where("(a.name1 LIKE '%".$ip."%' OR b.alias LIKE '%".$ip."%') and a.is_active = 1");
92
-                return $dbConn->fetchAll($query);
93
-    }
73
+	/** Get host groups by  name in IDO database
74
+	 *	does not catch exceptions
75
+	 *	@return array of objects ( name, id (object_id), display_name)
76
+	 */
77
+	public function getHostGroupByName($ip)
78
+	{
79
+		// 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
80
+		$dbConn = $this->getIdoDbConn();
81
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
82
+		// TODO : check for SQL injections
83
+		$query=$dbConn->select()
84
+		->from(
85
+			array('a' => 'icinga_objects'),
86
+			array('name' => 'a.name1','id' => 'object_id'))
87
+			->join(
88
+				array('b' => 'icinga_hostgroups'),
89
+				'b.hostgroup_object_id=a.object_id',
90
+				array('display_name' => 'b.alias'))
91
+				->where("(a.name1 LIKE '%".$ip."%' OR b.alias LIKE '%".$ip."%') and a.is_active = 1");
92
+				return $dbConn->fetchAll($query);
93
+	}
94 94
  
95
-    /** Get host IP (v4 and v6) by name in IDO database
96
-     *	does not catch exceptions
97
-     *	@return array ( name, display_name, ip4, ip6)
98
-     */
99
-    public function getHostInfoByID($id)
100
-    {
101
-        if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
102
-        $dbConn = $this->getIdoDbConn();
103
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
104
-        $query=$dbConn->select()
105
-        ->from(
106
-            array('a' => 'icinga_objects'),
107
-            array('name' => 'a.name1'))
108
-            ->join(
109
-                array('b' => 'icinga_hosts'),
110
-                'b.host_object_id=a.object_id',
111
-                array('ip4' => 'b.address', 'ip6' => 'b.address6', 'display_name' => 'b.display_name'))
112
-                ->where("a.object_id = '".$id."'");
113
-                return $dbConn->fetchRow($query);
114
-    }
95
+	/** Get host IP (v4 and v6) by name in IDO database
96
+	 *	does not catch exceptions
97
+	 *	@return array ( name, display_name, ip4, ip6)
98
+	 */
99
+	public function getHostInfoByID($id)
100
+	{
101
+		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
102
+		$dbConn = $this->getIdoDbConn();
103
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
104
+		$query=$dbConn->select()
105
+		->from(
106
+			array('a' => 'icinga_objects'),
107
+			array('name' => 'a.name1'))
108
+			->join(
109
+				array('b' => 'icinga_hosts'),
110
+				'b.host_object_id=a.object_id',
111
+				array('ip4' => 'b.address', 'ip6' => 'b.address6', 'display_name' => 'b.display_name'))
112
+				->where("a.object_id = '".$id."'");
113
+				return $dbConn->fetchRow($query);
114
+	}
115 115
         
116
-    /** Get host by objectid  in IDO database
117
-     *	does not catch exceptions
118
-     *	@return array of objects ( id, name, display_name, ip, ip6,  )
119
-     */
120
-    public function getHostByObjectID($id) // TODO : duplicate of getHostInfoByID above
121
-    {
122
-        if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
123
-        $dbConn = $this->getIdoDbConn();
124
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
125
-        $query=$dbConn->select()
126
-        ->from(
127
-            array('a' => 'icinga_objects'),
128
-            array('name' => 'a.name1','id' => 'a.object_id'))
129
-            ->join(
130
-                array('b' => 'icinga_hosts'),
131
-                'b.host_object_id=a.object_id',
132
-                array('display_name' => 'b.display_name' , 'ip' => 'b.address', 'ip6' => 'b.address6'))
133
-                ->where('a.object_id = ?',$id);
134
-                return $dbConn->fetchRow($query);
135
-    }
116
+	/** Get host by objectid  in IDO database
117
+	 *	does not catch exceptions
118
+	 *	@return array of objects ( id, name, display_name, ip, ip6,  )
119
+	 */
120
+	public function getHostByObjectID($id) // TODO : duplicate of getHostInfoByID above
121
+	{
122
+		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
123
+		$dbConn = $this->getIdoDbConn();
124
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
125
+		$query=$dbConn->select()
126
+		->from(
127
+			array('a' => 'icinga_objects'),
128
+			array('name' => 'a.name1','id' => 'a.object_id'))
129
+			->join(
130
+				array('b' => 'icinga_hosts'),
131
+				'b.host_object_id=a.object_id',
132
+				array('display_name' => 'b.display_name' , 'ip' => 'b.address', 'ip6' => 'b.address6'))
133
+				->where('a.object_id = ?',$id);
134
+				return $dbConn->fetchRow($query);
135
+	}
136 136
     
137
-    /** Get services from object ( host_object_id) in IDO database
138
-     *	does not catch exceptions
139
-     *	@param $id	int object_id
140
-     *	@return array display_name (of service), service_object_id
141
-     */
142
-    public function getServicesByHostid($id)
143
-    {
144
-        // 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
145
-        if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
146
-        $dbConn = $this->getIdoDbConn();
147
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
148
-        $query=$dbConn->select()
149
-        ->from(
150
-            array('s' => 'icinga_services'),
151
-            array('name' => 's.display_name','id' => 's.service_object_id'))
152
-            ->join(
153
-                array('a' => 'icinga_objects'),
154
-                's.service_object_id=a.object_id',
155
-                array('is_active'=>'a.is_active','name2'=>'a.name2'))
156
-                ->where('s.host_object_id='.$id.' AND a.is_active = 1');
157
-                return $dbConn->fetchAll($query);
158
-    }
137
+	/** Get services from object ( host_object_id) in IDO database
138
+	 *	does not catch exceptions
139
+	 *	@param $id	int object_id
140
+	 *	@return array display_name (of service), service_object_id
141
+	 */
142
+	public function getServicesByHostid($id)
143
+	{
144
+		// 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
145
+		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
146
+		$dbConn = $this->getIdoDbConn();
147
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
148
+		$query=$dbConn->select()
149
+		->from(
150
+			array('s' => 'icinga_services'),
151
+			array('name' => 's.display_name','id' => 's.service_object_id'))
152
+			->join(
153
+				array('a' => 'icinga_objects'),
154
+				's.service_object_id=a.object_id',
155
+				array('is_active'=>'a.is_active','name2'=>'a.name2'))
156
+				->where('s.host_object_id='.$id.' AND a.is_active = 1');
157
+				return $dbConn->fetchAll($query);
158
+	}
159 159
     
160
-    /** Get services from hostgroup object id ( hostgroup_object_id) in IDO database
161
-     * 	gets all hosts in hostgroup and return common services
162
-     *	does not catch exceptions
163
-     *	@param $id	int object_id
164
-     *	@return array display_name (of service), service_object_id
165
-     */
166
-    public function getServicesByHostGroupid($id)
167
-    {
168
-        if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
169
-        $dbConn = $this->getIdoDbConn();
170
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
171
-        $query=$dbConn->select()
172
-        ->from(
173
-            array('s' => 'icinga_hostgroup_members'),
174
-            array('host_object_id' => 's.host_object_id'))
175
-            ->join(
176
-                array('a' => 'icinga_hostgroups'),
177
-                's.hostgroup_id=a.hostgroup_id',
178
-                'hostgroup_object_id')
179
-                ->where('a.hostgroup_object_id='.$id);
180
-                $hosts=$dbConn->fetchAll($query);
181
-                $common_services=array();
182
-                $num_hosts=count($hosts);
183
-                foreach ($hosts as $key => $host)
184
-                { // For each host, get all services and add in common_services if not found or add counter
185
-                    $host_services=$this->getServicesByHostid($host->host_object_id);
186
-                    foreach($host_services as $service)
187
-                    {
188
-                        if (isset($common_services[$service->name2]['num']))
189
-                        {
190
-                            $common_services[$service->name2]['num'] +=1;
191
-                        }
192
-                        else
193
-                        {
194
-                            $common_services[$service->name2]['num']=1;
195
-                            $common_services[$service->name2]['name']=$service->name;
196
-                        }
197
-                    }
198
-                }
199
-                $result=array();
160
+	/** Get services from hostgroup object id ( hostgroup_object_id) in IDO database
161
+	 * 	gets all hosts in hostgroup and return common services
162
+	 *	does not catch exceptions
163
+	 *	@param $id	int object_id
164
+	 *	@return array display_name (of service), service_object_id
165
+	 */
166
+	public function getServicesByHostGroupid($id)
167
+	{
168
+		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
169
+		$dbConn = $this->getIdoDbConn();
170
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
171
+		$query=$dbConn->select()
172
+		->from(
173
+			array('s' => 'icinga_hostgroup_members'),
174
+			array('host_object_id' => 's.host_object_id'))
175
+			->join(
176
+				array('a' => 'icinga_hostgroups'),
177
+				's.hostgroup_id=a.hostgroup_id',
178
+				'hostgroup_object_id')
179
+				->where('a.hostgroup_object_id='.$id);
180
+				$hosts=$dbConn->fetchAll($query);
181
+				$common_services=array();
182
+				$num_hosts=count($hosts);
183
+				foreach ($hosts as $key => $host)
184
+				{ // For each host, get all services and add in common_services if not found or add counter
185
+					$host_services=$this->getServicesByHostid($host->host_object_id);
186
+					foreach($host_services as $service)
187
+					{
188
+						if (isset($common_services[$service->name2]['num']))
189
+						{
190
+							$common_services[$service->name2]['num'] +=1;
191
+						}
192
+						else
193
+						{
194
+							$common_services[$service->name2]['num']=1;
195
+							$common_services[$service->name2]['name']=$service->name;
196
+						}
197
+					}
198
+				}
199
+				$result=array();
200 200
                 
201
-                //print_r($common_services);
202
-                foreach (array_keys($common_services) as $key)
203
-                {
204
-                    if ($common_services[$key]['num'] == $num_hosts)
205
-                    {
206
-                        array_push($result,array($key,$common_services[$key]['name']));
207
-                    }
208
-                }
201
+				//print_r($common_services);
202
+				foreach (array_keys($common_services) as $key)
203
+				{
204
+					if ($common_services[$key]['num'] == $num_hosts)
205
+					{
206
+						array_push($result,array($key,$common_services[$key]['name']));
207
+					}
208
+				}
209 209
                 
210
-                return $result;
211
-    }
210
+				return $result;
211
+	}
212 212
     
213
-    /** Get services object id by host name / service name in IDO database
214
-     *	does not catch exceptions
215
-     *	@param $hostname string host name
216
-     *	@param $name string service name
217
-     *	@return array  service id
218
-     */
219
-    public function getServiceIDByName($hostname,$name)
220
-    {
221
-        $dbConn = $this->getIdoDbConn();
222
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
213
+	/** Get services object id by host name / service name in IDO database
214
+	 *	does not catch exceptions
215
+	 *	@param $hostname string host name
216
+	 *	@param $name string service name
217
+	 *	@return array  service id
218
+	 */
219
+	public function getServiceIDByName($hostname,$name)
220
+	{
221
+		$dbConn = $this->getIdoDbConn();
222
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
223 223
         
224
-        if ($name == null)
225
-        {
226
-            return array();
227
-        }
224
+		if ($name == null)
225
+		{
226
+			return array();
227
+		}
228 228
         
229
-        $query=$dbConn->select()
230
-        ->from(
231
-            array('s' => 'icinga_services'),
232
-            array('name' => 's.display_name','id' => 's.service_object_id'))
233
-            ->join(
234
-                array('a' => 'icinga_objects'),
235
-                's.service_object_id=a.object_id',
236
-                'is_active')
237
-                ->where('a.name2=\''.$name.'\' AND a.name1=\''.$hostname.'\' AND a.is_active = 1');
229
+		$query=$dbConn->select()
230
+		->from(
231
+			array('s' => 'icinga_services'),
232
+			array('name' => 's.display_name','id' => 's.service_object_id'))
233
+			->join(
234
+				array('a' => 'icinga_objects'),
235
+				's.service_object_id=a.object_id',
236
+				'is_active')
237
+				->where('a.name2=\''.$name.'\' AND a.name1=\''.$hostname.'\' AND a.is_active = 1');
238 238
                 
239
-                return $dbConn->fetchAll($query);
240
-    }
239
+				return $dbConn->fetchAll($query);
240
+	}
241 241
     
242
-    /** Get object name from object_id  in IDO database
243
-     *	does not catch exceptions
244
-     *	@param int $id object_id (default to null, used first if not null)
245
-     *	@return array name1 (host) name2 (service)
246
-     */
247
-    public function getObjectNameByid($id)
248
-    {
249
-        // 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
250
-        if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
251
-        $dbConn = $this->getIdoDbConn();
252
-        if ($dbConn === null) throw new \ErrorException('uncatched db error');
242
+	/** Get object name from object_id  in IDO database
243
+	 *	does not catch exceptions
244
+	 *	@param int $id object_id (default to null, used first if not null)
245
+	 *	@return array name1 (host) name2 (service)
246
+	 */
247
+	public function getObjectNameByid($id)
248
+	{
249
+		// 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
250
+		if (!preg_match('/^[0-9]+$/',$id)) { throw new Exception('Invalid id');  }
251
+		$dbConn = $this->getIdoDbConn();
252
+		if ($dbConn === null) throw new \ErrorException('uncatched db error');
253 253
         
254
-        $query=$dbConn->select()
255
-        ->from(
256
-            array('a' => 'icinga_objects'),
257
-            array('name1' => 'a.name1','name2' => 'a.name2'))
258
-            ->where('a.object_id='.$id.' AND a.is_active = 1');
254
+		$query=$dbConn->select()
255
+		->from(
256
+			array('a' => 'icinga_objects'),
257
+			array('name1' => 'a.name1','name2' => 'a.name2'))
258
+			->where('a.object_id='.$id.' AND a.is_active = 1');
259 259
             
260
-            return $dbConn->fetchRow($query);
261
-    }
260
+			return $dbConn->fetchRow($query);
261
+	}
262 262
     
263 263
     
264 264
 }
265 265
\ No newline at end of file
Please login to merge, or discard this patch.