Passed
Push — master ( f94f6f...4a1327 )
by Patrick
03:06
created
library/Trapdirector/TrapsProcess/Rule.php 1 patch
Indentation   +342 added lines, -342 removed lines patch added patch discarded remove patch
@@ -7,396 +7,396 @@
 block discarded – undo
7 7
 class Rule
8 8
 {
9 9
     
10
-    /** @var Logging $logging logging class*/
11
-    protected $logging;
10
+	/** @var Logging $logging logging class*/
11
+	protected $logging;
12 12
     
13
-    /** @var Trap $trapClass */
14
-    protected $trapClass;
13
+	/** @var Trap $trapClass */
14
+	protected $trapClass;
15 15
     
16
-    /**
17
-     * Setup Rule Class
18
-     * @param Trap $trapClass : To get logging class & plugin class
19
-     */
20
-    function __construct($trapClass)
21
-    {
22
-        $this->trapClass=$trapClass;
23
-        $this->logging=$trapClass->logging;
24
-    }
16
+	/**
17
+	 * Setup Rule Class
18
+	 * @param Trap $trapClass : To get logging class & plugin class
19
+	 */
20
+	function __construct($trapClass)
21
+	{
22
+		$this->trapClass=$trapClass;
23
+		$this->logging=$trapClass->logging;
24
+	}
25 25
 
26 26
 /**
27 27
  * Get full number 
28 28
  * @return array<number,string>
29 29
  */
30
-    private function get_number($rule,&$item)
31
-    {
32
-        $item2=$item+1;
33
-        while (
34
-            ($item2!=strlen($rule)) 
35
-            && (preg_match('/[\-0-9\.]/',$rule[$item2]))) 
36
-        { 
37
-            $item2++ ;
38
-        }
39
-        $val=substr($rule,$item,$item2-$item);
40
-        $item=$item2;
41
-        //echo "number ".$val."\n";
30
+	private function get_number($rule,&$item)
31
+	{
32
+		$item2=$item+1;
33
+		while (
34
+			($item2!=strlen($rule)) 
35
+			&& (preg_match('/[\-0-9\.]/',$rule[$item2]))) 
36
+		{ 
37
+			$item2++ ;
38
+		}
39
+		$val=substr($rule,$item,$item2-$item);
40
+		$item=$item2;
41
+		//echo "number ".$val."\n";
42 42
         
43
-        return array(0,$val);
44
-    }
43
+		return array(0,$val);
44
+	}
45 45
 
46
-    private function get_string($rule,&$item)
47
-    {
48
-        $item++;
49
-        $item2=$this->eval_getNext($rule,$item,'"');
50
-        $val=substr($rule,$item,$item2-$item-1);
51
-        $item=$item2;
52
-        //echo "string : ".$val."\n";
53
-        return array(1,$val);
46
+	private function get_string($rule,&$item)
47
+	{
48
+		$item++;
49
+		$item2=$this->eval_getNext($rule,$item,'"');
50
+		$val=substr($rule,$item,$item2-$item-1);
51
+		$item=$item2;
52
+		//echo "string : ".$val."\n";
53
+		return array(1,$val);
54 54
         
55
-    }
55
+	}
56 56
     
57
-    /**
58
-     * Parse elements inside () : jumps over "" and count parenthesis.
59
-     * Ex : ( "test" != ")test" & (1==2) ) will return "test" != ")test" & (1==2)
60
-     * @param string $rule : the current rule
61
-     * @param int $item : actual position in rule
62
-     * @throws Exception
63
-     * @return string : everything inside parenthesis
64
-     */
65
-    private function parse_parenthesis(string $rule,int &$item) : string
66
-    {
67
-        $item++;
68
-        $start=$item;
69
-        $parenthesis_count=0;
70
-        while (($item < strlen($rule)) // Not end of string AND
71
-            && ( ($rule[$item] != ')' ) || $parenthesis_count > 0) ) // Closing ')' or embeded ()
72
-        {
73
-            if ($rule[$item] == '"' )
74
-            { // pass through string
75
-                $item++;
76
-                $item=$this->eval_getNext($rule,$item,'"');
77
-            }
78
-            else{
79
-                if ($rule[$item] == '(')
80
-                {
81
-                    $parenthesis_count++;
82
-                }
83
-                if ($rule[$item] == ')')
84
-                {
85
-                    $parenthesis_count--;
86
-                }
87
-                $item++;
88
-            }
89
-        }
57
+	/**
58
+	 * Parse elements inside () : jumps over "" and count parenthesis.
59
+	 * Ex : ( "test" != ")test" & (1==2) ) will return "test" != ")test" & (1==2)
60
+	 * @param string $rule : the current rule
61
+	 * @param int $item : actual position in rule
62
+	 * @throws Exception
63
+	 * @return string : everything inside parenthesis
64
+	 */
65
+	private function parse_parenthesis(string $rule,int &$item) : string
66
+	{
67
+		$item++;
68
+		$start=$item;
69
+		$parenthesis_count=0;
70
+		while (($item < strlen($rule)) // Not end of string AND
71
+			&& ( ($rule[$item] != ')' ) || $parenthesis_count > 0) ) // Closing ')' or embeded ()
72
+		{
73
+			if ($rule[$item] == '"' )
74
+			{ // pass through string
75
+				$item++;
76
+				$item=$this->eval_getNext($rule,$item,'"');
77
+			}
78
+			else{
79
+				if ($rule[$item] == '(')
80
+				{
81
+					$parenthesis_count++;
82
+				}
83
+				if ($rule[$item] == ')')
84
+				{
85
+					$parenthesis_count--;
86
+				}
87
+				$item++;
88
+			}
89
+		}
90 90
         
91
-        if ($item==strlen($rule)) {throw new Exception("no closing () in ".$rule ." at " .$item);}
92
-        $val=substr($rule,$start,$item-$start);
93
-        $item++;
94
-        return $val;
95
-    }
91
+		if ($item==strlen($rule)) {throw new Exception("no closing () in ".$rule ." at " .$item);}
92
+		$val=substr($rule,$start,$item-$start);
93
+		$item++;
94
+		return $val;
95
+	}
96 96
 
97 97
     
98
-    /**
99
-     * Get and eval a grouped condition - ex : (1==1)
100
-     * @param string $rule
101
-     * @param int $item
102
-     * @return array
103
-     */
104
-    private function get_group(string $rule,int &$item) : array
105
-    {
106
-        // gets eveything inside parenthesis
107
-        $val=$this->parse_parenthesis($rule, $item);
108
-        // Returns boolean with evaluation of all inside parenthesis
109
-        $start=0;
110
-        return array(2,$this->evaluation($val,$start));
111
-    }
98
+	/**
99
+	 * Get and eval a grouped condition - ex : (1==1)
100
+	 * @param string $rule
101
+	 * @param int $item
102
+	 * @return array
103
+	 */
104
+	private function get_group(string $rule,int &$item) : array
105
+	{
106
+		// gets eveything inside parenthesis
107
+		$val=$this->parse_parenthesis($rule, $item);
108
+		// Returns boolean with evaluation of all inside parenthesis
109
+		$start=0;
110
+		return array(2,$this->evaluation($val,$start));
111
+	}
112 112
 
113
-    private function get_function(string $rule,int &$item) : array
114
-    {
115
-        // function is : __function(param1,param2...)
116
-        $start=$item; 
117
-        while (($item < strlen($rule)) && ($rule[$item] != '(' )) // Not end of string AND not opening '('
118
-        {
119
-            $item++;
120
-        }        
121
-        if ($item==strlen($rule)) {throw new Exception("no opening () for function in ".$rule ." at " .$item);}
113
+	private function get_function(string $rule,int &$item) : array
114
+	{
115
+		// function is : __function(param1,param2...)
116
+		$start=$item; 
117
+		while (($item < strlen($rule)) && ($rule[$item] != '(' )) // Not end of string AND not opening '('
118
+		{
119
+			$item++;
120
+		}        
121
+		if ($item==strlen($rule)) {throw new Exception("no opening () for function in ".$rule ." at " .$item);}
122 122
         
123
-        // get parameters between parenthesis
123
+		// get parameters between parenthesis
124 124
         
125
-        $this->parse_parenthesis($rule, $item);
125
+		$this->parse_parenthesis($rule, $item);
126 126
         
127
-        $val=substr($rule,$start,$item-$start);
127
+		$val=substr($rule,$start,$item-$start);
128 128
         
129
-        $this->logging->log('got function ' . $val,DEBUG);
129
+		$this->logging->log('got function ' . $val,DEBUG);
130 130
         
131
-        return array(2,$this->trapClass->pluginClass->evaluateFunctionString($val));
131
+		return array(2,$this->trapClass->pluginClass->evaluateFunctionString($val));
132 132
         
133
-    }
133
+	}
134 134
     
135
-    protected function eval_getElement($rule,&$item)
136
-    {
137
-        if ($item >= strlen($rule))
138
-        {
139
-            throw new Exception("Early end of string ".$rule ." at " .$item );
140
-        }
141
-        while ($rule[$item]==' ') $item++;
142
-        if (preg_match('/[\-0-9\.]/',$rule[$item]))
143
-        { // number
144
-            return $this->get_number($rule, $item);
145
-        }
146
-        if ($rule[$item] == '"')
147
-        { // string
148
-            return $this->get_string($rule, $item);
149
-        }
135
+	protected function eval_getElement($rule,&$item)
136
+	{
137
+		if ($item >= strlen($rule))
138
+		{
139
+			throw new Exception("Early end of string ".$rule ." at " .$item );
140
+		}
141
+		while ($rule[$item]==' ') $item++;
142
+		if (preg_match('/[\-0-9\.]/',$rule[$item]))
143
+		{ // number
144
+			return $this->get_number($rule, $item);
145
+		}
146
+		if ($rule[$item] == '"')
147
+		{ // string
148
+			return $this->get_string($rule, $item);
149
+		}
150 150
         
151
-        if ($rule[$item] == '(')
152
-        { // grouping
153
-            return $this->get_group($rule, $item);
154
-        }
155
-        if ($rule[$item] == '_')
156
-        { // function
157
-            return $this->get_function($rule, $item);
158
-        }
159
-        throw new Exception("number/string not found in ".$rule ." at " .$item . ' : ' .$rule[$item]);
151
+		if ($rule[$item] == '(')
152
+		{ // grouping
153
+			return $this->get_group($rule, $item);
154
+		}
155
+		if ($rule[$item] == '_')
156
+		{ // function
157
+			return $this->get_function($rule, $item);
158
+		}
159
+		throw new Exception("number/string not found in ".$rule ." at " .$item . ' : ' .$rule[$item]);
160 160
         
161
-    }
161
+	}
162 162
     
163
-    protected function eval_getNext($rule,$item,$tok)
164
-    {
165
-        while (
166
-            ($rule[$item] != $tok ) 
167
-            && ($item < strlen($rule))) 
168
-        { 
169
-            $item++;
170
-        }
171
-        if ($item==strlen($rule)) {
172
-            throw new Exception("closing '".$tok."' not found in ".$rule ." at " .$item);
173
-        }
174
-        return $item+1;
175
-    }
163
+	protected function eval_getNext($rule,$item,$tok)
164
+	{
165
+		while (
166
+			($rule[$item] != $tok ) 
167
+			&& ($item < strlen($rule))) 
168
+		{ 
169
+			$item++;
170
+		}
171
+		if ($item==strlen($rule)) {
172
+			throw new Exception("closing '".$tok."' not found in ".$rule ." at " .$item);
173
+		}
174
+		return $item+1;
175
+	}
176 176
     
177
-    protected function eval_getOper($rule,&$item)
178
-    {
179
-        while ($rule[$item]==' ') $item++;
180
-        switch ($rule[$item])
181
-        {
182
-            case '<':
183
-                if ($rule[$item+1]=='=') { $item+=2; return array(0,"<=");}
184
-                $item++; return array(0,"<");
185
-            case '>':
186
-                if ($rule[$item+1]=='=') { $item+=2; return array(0,">=");}
187
-                $item++; return array(0,">");
188
-            case '=':
189
-                $item++; return array(0,"=");
190
-            case '!':
191
-                if ($rule[$item+1]=='=') { $item+=2; return array(0,"!=");}
192
-                throw new Exception("Erreur in expr - incorrect operator '!'  found in ".$rule ." at " .$item);
193
-            case '~':
194
-                $item++; return array(0,"~");
195
-            case '|':
196
-                $item++; return array(1,"|");
197
-            case '&':
198
-                $item++; return array(1,"&");
199
-            default	:
200
-                throw new Exception("Erreur in expr - operator not found in ".$rule ." at " .$item);
201
-        }
202
-    }
177
+	protected function eval_getOper($rule,&$item)
178
+	{
179
+		while ($rule[$item]==' ') $item++;
180
+		switch ($rule[$item])
181
+		{
182
+			case '<':
183
+				if ($rule[$item+1]=='=') { $item+=2; return array(0,"<=");}
184
+				$item++; return array(0,"<");
185
+			case '>':
186
+				if ($rule[$item+1]=='=') { $item+=2; return array(0,">=");}
187
+				$item++; return array(0,">");
188
+			case '=':
189
+				$item++; return array(0,"=");
190
+			case '!':
191
+				if ($rule[$item+1]=='=') { $item+=2; return array(0,"!=");}
192
+				throw new Exception("Erreur in expr - incorrect operator '!'  found in ".$rule ." at " .$item);
193
+			case '~':
194
+				$item++; return array(0,"~");
195
+			case '|':
196
+				$item++; return array(1,"|");
197
+			case '&':
198
+				$item++; return array(1,"&");
199
+			default	:
200
+				throw new Exception("Erreur in expr - operator not found in ".$rule ." at " .$item);
201
+		}
202
+	}
203 203
     
204
-    private function check_negate_first($rule,&$item)
205
-    {
206
-        if ( $rule[$item] == '!') // If '!' found, negate next expression.
207
-        {
208
-            $item++;
209
-            return true;
210
-        }
211
-        else
212
-        {
213
-            return false;
214
-        }
215
-    }
204
+	private function check_negate_first($rule,&$item)
205
+	{
206
+		if ( $rule[$item] == '!') // If '!' found, negate next expression.
207
+		{
208
+			$item++;
209
+			return true;
210
+		}
211
+		else
212
+		{
213
+			return false;
214
+		}
215
+	}
216 216
 
217
-    private function do_compare($val1,$val2,$comp,$negate)
218
-    {
219
-        switch ($comp){
220
-            case '<':	$retVal= ($val1 < $val2); break;
221
-            case '<=':	$retVal= ($val1 <= $val2); break;
222
-            case '>':	$retVal= ($val1 > $val2); break;
223
-            case '>=':	$retVal= ($val1 >= $val2); break;
224
-            case '=':	$retVal= ($val1 == $val2); break;
225
-            case '!=':	$retVal= ($val1 != $val2); break;
226
-            case '~':	$retVal= (preg_match('/'.preg_replace('/"/','',$val2).'/',$val1)); break;
227
-            case '|':	$retVal= ($val1 || $val2); break;
228
-            case '&':	$retVal= ($val1 && $val2); break;
229
-            default:  throw new Exception("Error in expression - unknown comp : ".$comp);
230
-        }
231
-        if ($negate === true) $retVal = ! $retVal; // Inverse result if negate before expression
217
+	private function do_compare($val1,$val2,$comp,$negate)
218
+	{
219
+		switch ($comp){
220
+			case '<':	$retVal= ($val1 < $val2); break;
221
+			case '<=':	$retVal= ($val1 <= $val2); break;
222
+			case '>':	$retVal= ($val1 > $val2); break;
223
+			case '>=':	$retVal= ($val1 >= $val2); break;
224
+			case '=':	$retVal= ($val1 == $val2); break;
225
+			case '!=':	$retVal= ($val1 != $val2); break;
226
+			case '~':	$retVal= (preg_match('/'.preg_replace('/"/','',$val2).'/',$val1)); break;
227
+			case '|':	$retVal= ($val1 || $val2); break;
228
+			case '&':	$retVal= ($val1 && $val2); break;
229
+			default:  throw new Exception("Error in expression - unknown comp : ".$comp);
230
+		}
231
+		if ($negate === true) $retVal = ! $retVal; // Inverse result if negate before expression
232 232
         
233
-        return $retVal;
234
-    }
233
+		return $retVal;
234
+	}
235 235
     
236
-    /** Evaluation : makes token and evaluate.
237
-     *	Public function for expressions testing
238
-     *	accepts : < > = <= >= !=  (typec = 0)
239
-     *	operators : & | (typec=1)
240
-     *	with : integers/float  (type 0) or strings "" (type 1) or results (type 2)
241
-     *   comparison int vs strings will return null (error)
242
-     *	return : bool or null on error
243
-     */
244
-    public function evaluation($rule,&$item)
245
-    {
246
-        //echo "Evaluation of ".substr($rule,$item)."\n";
247
-        $negate=$this->check_negate_first($rule, $item);
248
-        // First element : number, string or ()
249
-        list($type1,$val1) = $this->eval_getElement($rule,$item);
250
-        //echo "Elmt1: ".$val1."/".$type1." : ".substr($rule,$item)."\n";
236
+	/** Evaluation : makes token and evaluate.
237
+	 *	Public function for expressions testing
238
+	 *	accepts : < > = <= >= !=  (typec = 0)
239
+	 *	operators : & | (typec=1)
240
+	 *	with : integers/float  (type 0) or strings "" (type 1) or results (type 2)
241
+	 *   comparison int vs strings will return null (error)
242
+	 *	return : bool or null on error
243
+	 */
244
+	public function evaluation($rule,&$item)
245
+	{
246
+		//echo "Evaluation of ".substr($rule,$item)."\n";
247
+		$negate=$this->check_negate_first($rule, $item);
248
+		// First element : number, string or ()
249
+		list($type1,$val1) = $this->eval_getElement($rule,$item);
250
+		//echo "Elmt1: ".$val1."/".$type1." : ".substr($rule,$item)."\n";
251 251
         
252
-        if ($item==strlen($rule)) // If only element, return value, but only boolean
253
-        {
254
-            if ($type1 != 2) throw new Exception("Cannot use num/string as boolean : ".$rule);
255
-            if ($negate === true) $val1= ! $val1;
256
-            return $val1;
257
-        }
252
+		if ($item==strlen($rule)) // If only element, return value, but only boolean
253
+		{
254
+			if ($type1 != 2) throw new Exception("Cannot use num/string as boolean : ".$rule);
255
+			if ($negate === true) $val1= ! $val1;
256
+			return $val1;
257
+		}
258 258
         
259
-        // Second element : operator
260
-        list($typec,$comp) = $this->eval_getOper($rule,$item);
261
-        //echo "Comp : ".$comp." : ".substr($rule,$item)."\n";
259
+		// Second element : operator
260
+		list($typec,$comp) = $this->eval_getOper($rule,$item);
261
+		//echo "Comp : ".$comp." : ".substr($rule,$item)."\n";
262 262
         
263
-        // Third element : number, string or ()
264
-        if ( $rule[$item] == '!') // starts with a ! so evaluate whats next
265
-        {
266
-            $item++;
267
-            if ($typec != 1) throw new Exception("Mixing boolean and comparison : ".$rule);
268
-            $val2= ! $this->evaluation($rule,$item);
269
-            $type2=2; // result is a boolean
270
-        }
271
-        else
272
-        {
273
-            list($type2,$val2) = $this->eval_getElement($rule,$item);
274
-        }
275
-        //echo "Elmt2: ".$val2."/".$type2." : ".substr($rule,$item)."\n";
263
+		// Third element : number, string or ()
264
+		if ( $rule[$item] == '!') // starts with a ! so evaluate whats next
265
+		{
266
+			$item++;
267
+			if ($typec != 1) throw new Exception("Mixing boolean and comparison : ".$rule);
268
+			$val2= ! $this->evaluation($rule,$item);
269
+			$type2=2; // result is a boolean
270
+		}
271
+		else
272
+		{
273
+			list($type2,$val2) = $this->eval_getElement($rule,$item);
274
+		}
275
+		//echo "Elmt2: ".$val2."/".$type2." : ".substr($rule,$item)."\n";
276 276
         
277
-        if ($type1!=$type2)  // cannot compare different types
278
-        {
279
-            throw new Exception("Cannot compare string & number : ".$rule);
280
-        }
281
-        if ($typec==1 && $type1 !=2) // cannot use & or | with string/number
282
-        {
283
-            throw new Exception("Cannot use boolean operators with string & number : ".$rule);
284
-        }
277
+		if ($type1!=$type2)  // cannot compare different types
278
+		{
279
+			throw new Exception("Cannot compare string & number : ".$rule);
280
+		}
281
+		if ($typec==1 && $type1 !=2) // cannot use & or | with string/number
282
+		{
283
+			throw new Exception("Cannot use boolean operators with string & number : ".$rule);
284
+		}
285 285
         
286
-        $retVal = $this->do_compare($val1, $val2, $comp, $negate);
286
+		$retVal = $this->do_compare($val1, $val2, $comp, $negate);
287 287
         
288
-        if ($item==strlen($rule)) return $retVal; // End of string : return evaluation
289
-        // check for logical operator :
290
-        switch ($rule[$item])
291
-        {
292
-            case '|':	$item++; return ($retVal || $this->evaluation($rule,$item) );
293
-            case '&':	$item++; return ($retVal && $this->evaluation($rule,$item) );
288
+		if ($item==strlen($rule)) return $retVal; // End of string : return evaluation
289
+		// check for logical operator :
290
+		switch ($rule[$item])
291
+		{
292
+			case '|':	$item++; return ($retVal || $this->evaluation($rule,$item) );
293
+			case '&':	$item++; return ($retVal && $this->evaluation($rule,$item) );
294 294
             
295
-            default:  throw new Exception("Erreur in expr - garbadge at end of expression : ".$rule[$item]);
296
-        }
297
-    }
295
+			default:  throw new Exception("Erreur in expr - garbadge at end of expression : ".$rule[$item]);
296
+		}
297
+	}
298 298
     
299
-    // Remove all whitespaces (when not quoted)
300
-    public function eval_cleanup($rule)
301
-    {
302
-        $item=0;
303
-        $rule2='';
304
-        while ($item < strlen($rule))
305
-        {
306
-            if ($rule[$item]==' ') { $item++; continue; }
307
-            if ($rule[$item]=='"')
308
-            {
309
-                $rule2.=$rule[$item];
310
-                $item++;
311
-                while (($item < strlen($rule)) && ($rule[$item]!='"') )
312
-                {
313
-                    $rule2.=$rule[$item];
314
-                    $item++;
315
-                }
316
-                if ($item == strlen ($rule)) throw new Exception("closing '\"' not found in ".$rule ." at " .$item);
317
-                $rule2.=$rule[$item];
318
-                $item++;
319
-                continue;
320
-            }
299
+	// Remove all whitespaces (when not quoted)
300
+	public function eval_cleanup($rule)
301
+	{
302
+		$item=0;
303
+		$rule2='';
304
+		while ($item < strlen($rule))
305
+		{
306
+			if ($rule[$item]==' ') { $item++; continue; }
307
+			if ($rule[$item]=='"')
308
+			{
309
+				$rule2.=$rule[$item];
310
+				$item++;
311
+				while (($item < strlen($rule)) && ($rule[$item]!='"') )
312
+				{
313
+					$rule2.=$rule[$item];
314
+					$item++;
315
+				}
316
+				if ($item == strlen ($rule)) throw new Exception("closing '\"' not found in ".$rule ." at " .$item);
317
+				$rule2.=$rule[$item];
318
+				$item++;
319
+				continue;
320
+			}
321 321
             
322
-            $rule2.=$rule[$item];
323
-            $item++;
324
-        }
322
+			$rule2.=$rule[$item];
323
+			$item++;
324
+		}
325 325
         
326
-        return $rule2;
327
-    }
326
+		return $rule2;
327
+	}
328 328
 
329 329
 
330
-    /**
331
-     * Get '*' or '**' and transform in [0-9]+ or .* in return string
332
-     * @param string $oid OID in normal or regexp format. '*' will be escaped ('\*')
333
-     * @return string correct regexp format
334
-     */
335
-    public function regexp_eval(string &$oid)
336
-    {
337
-        // ** replaced by .*
338
-        $oidR=preg_replace('/\*\*/', '.*', $oid);
339
-        // * replaced by [0-9]+
340
-        $oidR=preg_replace('/\*/', '[0-9]+', $oidR);
330
+	/**
331
+	 * Get '*' or '**' and transform in [0-9]+ or .* in return string
332
+	 * @param string $oid OID in normal or regexp format. '*' will be escaped ('\*')
333
+	 * @return string correct regexp format
334
+	 */
335
+	public function regexp_eval(string &$oid)
336
+	{
337
+		// ** replaced by .*
338
+		$oidR=preg_replace('/\*\*/', '.*', $oid);
339
+		// * replaced by [0-9]+
340
+		$oidR=preg_replace('/\*/', '[0-9]+', $oidR);
341 341
         
342
-        // replace * with \* in oid for preg_replace
343
-        $oid=preg_replace('/\*/', '\*', $oid);
342
+		// replace * with \* in oid for preg_replace
343
+		$oid=preg_replace('/\*/', '\*', $oid);
344 344
         
345
-        $this->logging->log('Regexp eval : '.$oid.' / '.$oidR,DEBUG );
345
+		$this->logging->log('Regexp eval : '.$oid.' / '.$oidR,DEBUG );
346 346
         
347
-        return $oidR;
348
-    }
347
+		return $oidR;
348
+	}
349 349
     
350 350
     
351
-    /** Evaluation rule (uses eval_* functions recursively)
352
-     *	@param string $rule : rule ( _OID(.1.3.6.1.4.1.8072.2.3.2.1)=_OID(.1.3.6.1.2.1.1.3.0) )
353
-     *  @param array $oidList : OIDs values to sustitute.
354
-     *	@return bool : true : rule match, false : rule don't match , throw exception on error.
355
-     */   
356
-    public function eval_rule($rule,$oidList)
357
-    {
358
-        if ($rule==null || $rule == '') // Empty rule is always true
359
-        {
360
-            return true;
361
-        }
362
-        $matches=array();
363
-        while (preg_match('/_OID\(([0-9\.\*]+)\)/',$rule,$matches) == 1)
364
-        {
365
-            $oid=$matches[1];
366
-            $found=0;
367
-            // Test and transform regexp
368
-            $oidR = $this->regexp_eval($oid);
351
+	/** Evaluation rule (uses eval_* functions recursively)
352
+	 *	@param string $rule : rule ( _OID(.1.3.6.1.4.1.8072.2.3.2.1)=_OID(.1.3.6.1.2.1.1.3.0) )
353
+	 *  @param array $oidList : OIDs values to sustitute.
354
+	 *	@return bool : true : rule match, false : rule don't match , throw exception on error.
355
+	 */   
356
+	public function eval_rule($rule,$oidList)
357
+	{
358
+		if ($rule==null || $rule == '') // Empty rule is always true
359
+		{
360
+			return true;
361
+		}
362
+		$matches=array();
363
+		while (preg_match('/_OID\(([0-9\.\*]+)\)/',$rule,$matches) == 1)
364
+		{
365
+			$oid=$matches[1];
366
+			$found=0;
367
+			// Test and transform regexp
368
+			$oidR = $this->regexp_eval($oid);
369 369
             
370
-            foreach($oidList as $val)
371
-            {
372
-                if (preg_match("/^$oidR$/",$val->oid) == 1)
373
-                {
374
-                    if (!preg_match('/^-?[0-9]*\.?[0-9]+$/',$val->value))
375
-                    { // If not a number, change " to ' and put " around it
376
-                        $val->value=preg_replace('/"/',"'",$val->value);
377
-                        $val->value='"'.$val->value.'"';
378
-                    }
379
-                    $rep=0;
380
-                    $rule=preg_replace('/_OID\('.$oid.'\)/',$val->value,$rule,-1,$rep);
381
-                    if ($rep==0)
382
-                    {
383
-                        $this->logging->log("Error in rule_eval",WARN,'');
384
-                        return false;
385
-                    }
386
-                    $found=1;
387
-                    break;
388
-                }
389
-            }
390
-            if ($found==0)
391
-            {	// OID not found : throw error
392
-                throw new Exception('OID '.$oid.' not found in trap');
393
-            }
394
-        }
395
-        $item=0;
396
-        $rule=$this->eval_cleanup($rule);
397
-        $this->logging->log('Rule after clenup: '.$rule,INFO );
370
+			foreach($oidList as $val)
371
+			{
372
+				if (preg_match("/^$oidR$/",$val->oid) == 1)
373
+				{
374
+					if (!preg_match('/^-?[0-9]*\.?[0-9]+$/',$val->value))
375
+					{ // If not a number, change " to ' and put " around it
376
+						$val->value=preg_replace('/"/',"'",$val->value);
377
+						$val->value='"'.$val->value.'"';
378
+					}
379
+					$rep=0;
380
+					$rule=preg_replace('/_OID\('.$oid.'\)/',$val->value,$rule,-1,$rep);
381
+					if ($rep==0)
382
+					{
383
+						$this->logging->log("Error in rule_eval",WARN,'');
384
+						return false;
385
+					}
386
+					$found=1;
387
+					break;
388
+				}
389
+			}
390
+			if ($found==0)
391
+			{	// OID not found : throw error
392
+				throw new Exception('OID '.$oid.' not found in trap');
393
+			}
394
+		}
395
+		$item=0;
396
+		$rule=$this->eval_cleanup($rule);
397
+		$this->logging->log('Rule after clenup: '.$rule,INFO );
398 398
         
399
-        return  $this->evaluation($rule,$item);
400
-    }
399
+		return  $this->evaluation($rule,$item);
400
+	}
401 401
     
402 402
 }
403 403
\ No newline at end of file
Please login to merge, or discard this patch.
library/Trapdirector/TrapsProcess/Trap.php 1 patch
Indentation   +969 added lines, -969 removed lines patch added patch discarded remove patch
@@ -16,960 +16,960 @@  discard block
 block discarded – undo
16 16
  */
17 17
 class Trap
18 18
 {
19
-    // Configuration files and dirs
20
-    /** @var string Icinga etc path */
21
-    protected $icingaweb2Etc;
22
-    /** @var string $trapModuleConfig config.ini of module */
23
-    protected $trapModuleConfig;
24
-    /** @var string $icingaweb2Ressources resources.ini of icingaweb2 */
25
-    protected $icingaweb2Ressources;
26
-    // Options from config.ini (default values)
27
-    /** @var string $snmptranslate */
28
-    protected $snmptranslate='/usr/bin/snmptranslate';
29
-    /** @var string $snmptranslate_dirs */
30
-    protected $snmptranslate_dirs='/usr/share/icingaweb2/modules/trapdirector/mibs';
31
-    /** @var string $icinga2cmd */
32
-    protected $icinga2cmd='/var/run/icinga2/cmd/icinga2.cmd';
33
-    /** @var string $dbPrefix */
34
-    protected $dbPrefix='traps_';
35
-    
36
-    // API
37
-    /** @var boolean $apiUse */
38
-    protected $apiUse=false;
39
-    /** @var Icinga2API $icinga2api */
40
-    protected $icinga2api=null;
41
-    /** @var string $apiHostname */
42
-    protected $apiHostname='';
43
-    /** @var integer $apiPort */
44
-    protected $apiPort=0;
45
-    /** @var string $apiUsername */
46
-    protected $apiUsername='';
47
-    /** @var string $apiPassword */
48
-    protected $apiPassword='';
49
-    
50
-    // Logs
51
-    /** @var Logging Logging class. */
52
-    public $logging;    //< Logging class.
53
-    /** @var bool true if log was setup in constructor */
54
-    protected $logSetup;   //< bool true if log was setup in constructor
55
-    
56
-    // Databases
57
-    /** @var Database $trapsDB  Database class*/
58
-    public $trapsDB = null;
59
-    
60
-    // Trap received data
61
-    protected $receivingHost;
62
-    /** @var array	Main trap data (oid, source...) */
63
-    public $trapData=array();
64
-    /** @var array $trapDataExt Additional trap data objects (oid/value).*/
65
-    public $trapDataExt=array(); 
66
-    /** @var int $trapId trap_id after sql insert*/
67
-    public $trapId=null;
68
-    /** @var string $trapAction trap action for final write*/
69
-    public $trapAction=null;
70
-    /** @var boolean $trapToDb log trap to DB */
71
-    protected $trapToDb=true;
72
-    
73
-    /** @var Mib mib class */
74
-    public $mibClass = null;
75
-    
76
-    /** @var Rule rule class */
77
-    public $ruleClass = null;
78
-    
79
-    /** @var Plugins plugins manager **/
80
-    public $pluginClass = null;
81
-    
82
-    function __construct($etcDir='/etc/icingaweb2',$baseLogLevel=null,$baseLogMode='syslog',$baseLogFile='')
83
-    {
84
-        // Paths of ini files
85
-        $this->icingaweb2Etc=$etcDir;
86
-        $this->trapModuleConfig=$this->icingaweb2Etc."/modules/trapdirector/config.ini";
87
-        $this->icingaweb2Ressources=$this->icingaweb2Etc."/resources.ini";
88
-        
89
-        //************* Setup logging
90
-        $this->logging = new Logging();
91
-        if ($baseLogLevel != null)
92
-        {
93
-            $this->logging->setLogging($baseLogLevel, $baseLogMode,$baseLogFile);
94
-            $this->logSetup=true;
95
-        }
96
-        else
97
-        {
98
-            $this->logSetup=false;
99
-        }
100
-        $this->logging->log('Loggin started', INFO);
101
-        
102
-        //*************** Get options from ini files
103
-        if (! is_file($this->trapModuleConfig))
104
-        {
105
-            throw new Exception("Ini file ".$this->trapModuleConfig." does not exists");
106
-        }
107
-        $trapConfig=parse_ini_file($this->trapModuleConfig,true);
108
-        if ($trapConfig == false)
109
-        {
110
-            $this->logging->log("Error reading ini file : ".$this->trapModuleConfig,ERROR,'syslog');
111
-            throw new Exception("Error reading ini file : ".$this->trapModuleConfig);
112
-        }
113
-        $this->getMainOptions($trapConfig); // Get main options from ini file
114
-        
115
-        //*************** Setup database class & get options
116
-        $this->setupDatabase($trapConfig);
117
-        
118
-        $this->getDatabaseOptions(); // Get options in database
119
-        
120
-        //*************** Setup API
121
-        if ($this->apiUse === true) $this->getAPI(); // Setup API
122
-        
123
-        //*************** Setup MIB
124
-        $this->mibClass = new Mib($this->logging,$this->trapsDB,$this->snmptranslate,$this->snmptranslate_dirs); // Create Mib class
125
-        
126
-        //*************** Setup Rule
127
-        $this->ruleClass = new Rule($this); //< Create Rule class
128
-        
129
-        $this->trapData=array(  // TODO : put this in a reset function (DAEMON_MODE)
130
-            'source_ip'	=> 'unknown',
131
-            'source_port'	=> 'unknown',
132
-            'destination_ip'	=> 'unknown',
133
-            'destination_port'	=> 'unknown',
134
-            'trap_oid'	=> 'unknown'
135
-        );
136
-        
137
-        //*************** Setup Plugins
138
-        //Create plugin class. Plugins are not loaded here, but by calling registerAllPlugins
139
-        $this->pluginClass = new Plugins($this);
19
+	// Configuration files and dirs
20
+	/** @var string Icinga etc path */
21
+	protected $icingaweb2Etc;
22
+	/** @var string $trapModuleConfig config.ini of module */
23
+	protected $trapModuleConfig;
24
+	/** @var string $icingaweb2Ressources resources.ini of icingaweb2 */
25
+	protected $icingaweb2Ressources;
26
+	// Options from config.ini (default values)
27
+	/** @var string $snmptranslate */
28
+	protected $snmptranslate='/usr/bin/snmptranslate';
29
+	/** @var string $snmptranslate_dirs */
30
+	protected $snmptranslate_dirs='/usr/share/icingaweb2/modules/trapdirector/mibs';
31
+	/** @var string $icinga2cmd */
32
+	protected $icinga2cmd='/var/run/icinga2/cmd/icinga2.cmd';
33
+	/** @var string $dbPrefix */
34
+	protected $dbPrefix='traps_';
35
+    
36
+	// API
37
+	/** @var boolean $apiUse */
38
+	protected $apiUse=false;
39
+	/** @var Icinga2API $icinga2api */
40
+	protected $icinga2api=null;
41
+	/** @var string $apiHostname */
42
+	protected $apiHostname='';
43
+	/** @var integer $apiPort */
44
+	protected $apiPort=0;
45
+	/** @var string $apiUsername */
46
+	protected $apiUsername='';
47
+	/** @var string $apiPassword */
48
+	protected $apiPassword='';
49
+    
50
+	// Logs
51
+	/** @var Logging Logging class. */
52
+	public $logging;    //< Logging class.
53
+	/** @var bool true if log was setup in constructor */
54
+	protected $logSetup;   //< bool true if log was setup in constructor
55
+    
56
+	// Databases
57
+	/** @var Database $trapsDB  Database class*/
58
+	public $trapsDB = null;
59
+    
60
+	// Trap received data
61
+	protected $receivingHost;
62
+	/** @var array	Main trap data (oid, source...) */
63
+	public $trapData=array();
64
+	/** @var array $trapDataExt Additional trap data objects (oid/value).*/
65
+	public $trapDataExt=array(); 
66
+	/** @var int $trapId trap_id after sql insert*/
67
+	public $trapId=null;
68
+	/** @var string $trapAction trap action for final write*/
69
+	public $trapAction=null;
70
+	/** @var boolean $trapToDb log trap to DB */
71
+	protected $trapToDb=true;
72
+    
73
+	/** @var Mib mib class */
74
+	public $mibClass = null;
75
+    
76
+	/** @var Rule rule class */
77
+	public $ruleClass = null;
78
+    
79
+	/** @var Plugins plugins manager **/
80
+	public $pluginClass = null;
81
+    
82
+	function __construct($etcDir='/etc/icingaweb2',$baseLogLevel=null,$baseLogMode='syslog',$baseLogFile='')
83
+	{
84
+		// Paths of ini files
85
+		$this->icingaweb2Etc=$etcDir;
86
+		$this->trapModuleConfig=$this->icingaweb2Etc."/modules/trapdirector/config.ini";
87
+		$this->icingaweb2Ressources=$this->icingaweb2Etc."/resources.ini";
88
+        
89
+		//************* Setup logging
90
+		$this->logging = new Logging();
91
+		if ($baseLogLevel != null)
92
+		{
93
+			$this->logging->setLogging($baseLogLevel, $baseLogMode,$baseLogFile);
94
+			$this->logSetup=true;
95
+		}
96
+		else
97
+		{
98
+			$this->logSetup=false;
99
+		}
100
+		$this->logging->log('Loggin started', INFO);
101
+        
102
+		//*************** Get options from ini files
103
+		if (! is_file($this->trapModuleConfig))
104
+		{
105
+			throw new Exception("Ini file ".$this->trapModuleConfig." does not exists");
106
+		}
107
+		$trapConfig=parse_ini_file($this->trapModuleConfig,true);
108
+		if ($trapConfig == false)
109
+		{
110
+			$this->logging->log("Error reading ini file : ".$this->trapModuleConfig,ERROR,'syslog');
111
+			throw new Exception("Error reading ini file : ".$this->trapModuleConfig);
112
+		}
113
+		$this->getMainOptions($trapConfig); // Get main options from ini file
114
+        
115
+		//*************** Setup database class & get options
116
+		$this->setupDatabase($trapConfig);
117
+        
118
+		$this->getDatabaseOptions(); // Get options in database
119
+        
120
+		//*************** Setup API
121
+		if ($this->apiUse === true) $this->getAPI(); // Setup API
122
+        
123
+		//*************** Setup MIB
124
+		$this->mibClass = new Mib($this->logging,$this->trapsDB,$this->snmptranslate,$this->snmptranslate_dirs); // Create Mib class
125
+        
126
+		//*************** Setup Rule
127
+		$this->ruleClass = new Rule($this); //< Create Rule class
128
+        
129
+		$this->trapData=array(  // TODO : put this in a reset function (DAEMON_MODE)
130
+			'source_ip'	=> 'unknown',
131
+			'source_port'	=> 'unknown',
132
+			'destination_ip'	=> 'unknown',
133
+			'destination_port'	=> 'unknown',
134
+			'trap_oid'	=> 'unknown'
135
+		);
136
+        
137
+		//*************** Setup Plugins
138
+		//Create plugin class. Plugins are not loaded here, but by calling registerAllPlugins
139
+		$this->pluginClass = new Plugins($this);
140 140
             
141 141
             
142
-    }
143
-    
144
-    /**
145
-     * Get option from array of ini file, send message if empty
146
-     * @param string $option_array Array of ini file
147
-     * @param string $option_category category in ini file
148
-     * @param string $option_name name of option in category
149
-     * @param mixed $option_var variable to fill if found, left untouched if not found
150
-     * @param integer $log_level default 2 (warning)
151
-     * @param string $message warning message if not found
152
-     * @return boolean true if found, or false
153
-     */
154
-    protected function getOptionIfSet($option_array,$option_category,$option_name, &$option_var, $log_level = 2, $message = null)
155
-    {
156
-        if (!isset($option_array[$option_category][$option_name]))
157
-        {
158
-            if ($message === null)
159
-            {
160
-                $message='No ' . $option_name . ' in config file: '. $this->trapModuleConfig;
161
-            }
162
-            $this->logging->log($message,$log_level);
163
-            return false;
164
-        }
165
-        else
166
-        {
167
-            $option_var=$option_array[$option_category][$option_name];
168
-            return true;
169
-        }
170
-    }
171
-    
172
-    /**
173
-     * Get options from ini file
174
-     * @param array $trap_config : ini file array
175
-     */
176
-    protected function getMainOptions($trapConfig)
177
-    {
178
-        
179
-        // Snmptranslate binary path
180
-        $this->getOptionIfSet($trapConfig,'config','snmptranslate', $this->snmptranslate);
181
-        
182
-        // mibs path
183
-        $this->getOptionIfSet($trapConfig,'config','snmptranslate_dirs', $this->snmptranslate_dirs);
184
-        
185
-        // icinga2cmd path
186
-        $this->getOptionIfSet($trapConfig,'config','icingacmd', $this->icinga2cmd);
187
-        
188
-        // table prefix
189
-        $this->getOptionIfSet($trapConfig,'config','database_prefix', $this->dbPrefix);
190
-        
191
-        // API options
192
-        if ($this->getOptionIfSet($trapConfig,'config','icingaAPI_host', $this->apiHostname))
193
-        {
194
-            $this->apiUse=true;
195
-            $this->getOptionIfSet($trapConfig,'config','icingaAPI_port', $this->apiPort);
196
-            $this->getOptionIfSet($trapConfig,'config','icingaAPI_user', $this->apiUsername);
197
-            $this->getOptionIfSet($trapConfig,'config','icingaAPI_password', $this->apiPassword);
198
-        }
199
-    }
200
-    
201
-    /**
202
-     * Create and setup database class for trap & ido (if no api) db
203
-     * @param array $trap_config : ini file array
204
-     */
205
-    protected function setupDatabase($trapConfig)
206
-    {
207
-        // Trap database
208
-        if (!array_key_exists('database',$trapConfig['config']))
209
-        {
210
-            $this->logging->log("No database in config file: ".$this->trapModuleConfig,ERROR,'');
211
-            return;
212
-        }
213
-        $dbTrapName=$trapConfig['config']['database'];
214
-        $this->logging->log("Found database in config file: ".$dbTrapName,INFO );
215
-        
216
-        if ( ($dbConfig=parse_ini_file($this->icingaweb2Ressources,true)) === false)
217
-        {
218
-            $this->logging->log("Error reading ini file : ".$this->icingaweb2Ressources,ERROR,'');
219
-            return;
220
-        }
221
-        if (!array_key_exists($dbTrapName,$dbConfig))
222
-        {
223
-            $this->logging->log("No database '.$dbTrapName.' in config file: ".$this->icingaweb2Ressources,ERROR,'');
224
-            return;
225
-        }
226
-        
227
-        $this->trapsDB = new Database($this->logging,$dbConfig[$dbTrapName],$this->dbPrefix);
228
-        
229
-        if ($this->apiUse === true) return; // In case of API use, no IDO is necessary
230
-        
231
-        // IDO Database
232
-        if (!array_key_exists('IDOdatabase',$trapConfig['config']))
233
-        {
234
-            $this->logging->log("No IDOdatabase in config file: ".$this->trapModuleConfig,ERROR,'');
235
-        }
236
-        $dbIdoName=$trapConfig['config']['IDOdatabase'];
237
-        
238
-        $this->logging->log("Found IDO database in config file: ".$dbIdoName,INFO );
239
-        if (!array_key_exists($dbIdoName,$dbConfig))
240
-        {
241
-            $this->logging->log("No database '.$dbIdoName.' in config file: ".$this->icingaweb2Ressources,ERROR,'');
242
-            return;
243
-        }
244
-        
245
-        $this->trapsDB->setupIDO($dbConfig[$dbIdoName]);
246
-    }
247
-    
248
-    /**
249
-     * Get options in database
250
-     */
251
-    protected function getDatabaseOptions()
252
-    {
253
-        // Database options
254
-        if ($this->logSetup === false) // Only if logging was no setup in constructor
255
-        {
256
-            $this->getDBConfigIfSet('log_level',$this->logging->debugLevel);
257
-            $this->getDBConfigIfSet('log_destination',$this->logging->outputMode);
258
-            $this->getDBConfigIfSet('log_file',$this->logging->outputFile);
259
-        }
260
-    }
261
-    
262
-    /** Set $variable to value if $element found in database config table
263
-     * @param string $element
264
-     * @param string $variable
265
-     */
266
-    protected function getDBConfigIfSet($element,&$variable)
267
-    {
268
-        $value=$this->getDBConfig($element);
269
-        if ($value != null) $variable=$value;
270
-    }
271
-    
272
-    /**
273
-     *   Get data from db_config
274
-     *	@param $element string name of param
275
-     *	@return mixed : value (or null)
276
-     */
277
-    protected function getDBConfig($element)  // TODO : put this in DB class
278
-    {
279
-        $db_conn=$this->trapsDB->db_connect_trap();
280
-        $sql='SELECT value from '.$this->dbPrefix.'db_config WHERE ( name=\''.$element.'\' )';
281
-        if (($ret_code=$db_conn->query($sql)) === false) {
282
-            $this->logging->log('No result in query : ' . $sql,WARN,'');
283
-            return null;
284
-        }
285
-        $value=$ret_code->fetch();
286
-        if ($value != null && isset($value['value']))
287
-        {
288
-            return $value['value'];
289
-        }
290
-        return null;
291
-    }
292
-    
293
-    /** OBSOLETE Send log. Throws exception on critical error
294
-     *	@param	string $message Message to log
295
-     *	@param	int $level 1=critical 2=warning 3=trace 4=debug
296
-     *	@param  string $destination file/syslog/display
297
-     *	@return void
298
-     **/
299
-    public function trapLog( $message, $level, $destination ='') // OBSOLETE
300
-    {
301
-        // TODO : replace ref with $this->logging->log
302
-        $this->logging->log($message, $level, $destination);
303
-    }
304
-    
305
-    public function setLogging($debugLvl,$outputType,$outputOption=null)  // OBSOLETE
306
-    {
307
-        $this->logging->setLogging($debugLvl, $outputType,$outputOption);
308
-    }
309
-    
310
-    /**
311
-     * Returns or create new IcingaAPI object
312
-     * @return \Icinga\Module\Trapdirector\Icinga2API
313
-     */
314
-    protected function getAPI()
315
-    {
316
-        if ($this->icinga2api == null)
317
-        {
318
-            $this->icinga2api = new Icinga2API($this->apiHostname,$this->apiPort);
319
-        }
320
-        return $this->icinga2api;
321
-    }
322
-    
323
-    
324
-    /**
325
-     * read data from stream
326
-     *	@param $stream string input stream, defaults to "php://stdin"
327
-     *	@return mixed array trap data or exception with error
328
-     */
329
-    public function read_trap($stream='php://stdin')
330
-    {
331
-        //Read data from snmptrapd from stdin
332
-        $input_stream=fopen($stream, 'r');
333
-        
334
-        if ($input_stream === false)
335
-        {
336
-            $this->writeTrapErrorToDB("Error reading trap (code 1/Stdin)");
337
-            $this->logging->log("Error reading stdin !",ERROR,'');
338
-            return null; // note : exception thrown by logging
339
-        }
340
-        
341
-        // line 1 : host
342
-        $this->receivingHost=chop(fgets($input_stream));
343
-        if ($this->receivingHost === false)
344
-        {
345
-            $this->writeTrapErrorToDB("Error reading trap (code 1/Line Host)");
346
-            $this->logging->log("Error reading Host !",ERROR,'');
347
-        }
348
-        // line 2 IP:port=>IP:port
349
-        $IP=chop(fgets($input_stream));
350
-        if ($IP === false)
351
-        {
352
-            $this->writeTrapErrorToDB("Error reading trap (code 1/Line IP)");
353
-            $this->logging->log("Error reading IP !",ERROR,'');
354
-        }
355
-        $matches=array();
356
-        $ret_code=preg_match('/.DP: \[(.*)\]:(.*)->\[(.*)\]:(.*)/',$IP,$matches);
357
-        if ($ret_code===0 || $ret_code===false)
358
-        {
359
-            $this->writeTrapErrorToDB("Error parsing trap (code 2/IP)");
360
-            $this->logging->log('Error parsing IP : '.$IP,ERROR,'');
361
-        }
362
-        else
363
-        {
364
-            $this->trapData['source_ip']=$matches[1];
365
-            $this->trapData['destination_ip']=$matches[3];
366
-            $this->trapData['source_port']=$matches[2];
367
-            $this->trapData['destination_port']=$matches[4];
368
-        }
369
-        
370
-        while (($vars=fgets($input_stream)) !==false)
371
-        {
372
-            $vars=chop($vars);
373
-            $ret_code=preg_match('/^([^ ]+) (.*)$/',$vars,$matches);
374
-            if ($ret_code===0 || $ret_code===false)
375
-            {
376
-                $this->logging->log('No match on trap data : '.$vars,WARN,'');
377
-            }
378
-            else
379
-            {
380
-                if (($matches[1]=='.1.3.6.1.6.3.1.1.4.1.0') || ($matches[1]=='.1.3.6.1.6.3.1.1.4.1'))
381
-                {
382
-                    $this->trapData['trap_oid']=$matches[2];
383
-                }
384
-                else
385
-                {
386
-                    $object= new stdClass;
387
-                    $object->oid =$matches[1];
388
-                    $object->value = $matches[2];
389
-                    array_push($this->trapDataExt,$object);
390
-                }
391
-            }
392
-        }
393
-        
394
-        if ($this->trapData['trap_oid']=='unknown')
395
-        {
396
-            $this->writeTrapErrorToDB("No trap oid found : check snmptrapd configuration (code 3/OID)",$this->trapData['source_ip']);
397
-            $this->logging->log('no trap oid found',ERROR,'');
398
-        }
399
-        
400
-        // Translate oids.
401
-        
402
-        $retArray=$this->translateOID($this->trapData['trap_oid']);
403
-        if ($retArray != null)
404
-        {
405
-            $this->trapData['trap_name']=$retArray['trap_name'];
406
-            $this->trapData['trap_name_mib']=$retArray['trap_name_mib'];
407
-        }
408
-        foreach ($this->trapDataExt as $key => $val)
409
-        {
410
-            $retArray=$this->translateOID($val->oid);
411
-            if ($retArray != null)
412
-            {
413
-                $this->trapDataExt[$key]->oid_name=$retArray['trap_name'];
414
-                $this->trapDataExt[$key]->oid_name_mib=$retArray['trap_name_mib'];
415
-            }
416
-        }
417
-        
418
-        
419
-        $this->trapData['status']= 'waiting';
420
-        
421
-        return $this->trapData;
422
-    }
423
-    
424
-    /**
425
-     * Translate oid into array(MIB,Name)
426
-     * @param $oid string oid to translate
427
-     * @return mixed : null if not found or array(MIB,Name)
428
-     */
429
-    public function translateOID($oid)
430
-    {
431
-        // try from database
432
-        $db_conn=$this->trapsDB->db_connect_trap();
433
-        
434
-        $sql='SELECT mib,name from '.$this->dbPrefix.'mib_cache WHERE oid=\''.$oid.'\';';
435
-        $this->logging->log('SQL query : '.$sql,DEBUG );
436
-        if (($ret_code=$db_conn->query($sql)) === false) {
437
-            $this->logging->log('No result in query : ' . $sql,ERROR,'');
438
-        }
439
-        $name=$ret_code->fetch();
440
-        if ($name['name'] != null)
441
-        {
442
-            return array('trap_name_mib'=>$name['mib'],'trap_name'=>$name['name']);
443
-        }
444
-        
445
-        // Also check if it is an instance of OID
446
-        $oid_instance=preg_replace('/\.[0-9]+$/','',$oid);
447
-        
448
-        $sql='SELECT mib,name from '.$this->dbPrefix.'mib_cache WHERE oid=\''.$oid_instance.'\';';
449
-        $this->logging->log('SQL query : '.$sql,DEBUG );
450
-        if (($ret_code=$db_conn->query($sql)) === false) {
451
-            $this->logging->log('No result in query : ' . $sql,ERROR,'');
452
-        }
453
-        $name=$ret_code->fetch();
454
-        if ($name['name'] != null)
455
-        {
456
-            return array('trap_name_mib'=>$name['mib'],'trap_name'=>$name['name']);
457
-        }
458
-        
459
-        // Try to get oid name from snmptranslate
460
-        $translate=exec($this->snmptranslate . ' -m ALL -M +'.$this->snmptranslate_dirs.
461
-            ' '.$oid);
462
-        $matches=array();
463
-        $ret_code=preg_match('/(.*)::(.*)/',$translate,$matches);
464
-        if ($ret_code===0 || $ret_code === false) {
465
-            return NULL;
466
-        } else {
467
-            $this->logging->log('Found name with snmptrapd and not in DB for oid='.$oid,INFO);
468
-            return array('trap_name_mib'=>$matches[1],'trap_name'=>$matches[2]);
469
-        }
470
-    }
471
-    
472
-    /**
473
-     * Erase old trap records
474
-     *	@param integer $days : erase traps when more than $days old
475
-     *	@return integer : number of lines deleted
476
-     **/
477
-    public function eraseOldTraps($days=0)
478
-    {
479
-        if ($days==0)
480
-        {
481
-            if (($days=$this->getDBConfig('db_remove_days')) == null)
482
-            {
483
-                $this->logging->log('No days specified & no db value : no tap erase' ,WARN,'');
484
-                return;
485
-            }
486
-        }
487
-        $db_conn=$this->trapsDB->db_connect_trap();
488
-        $daysago = strtotime("-".$days." day");
489
-        $sql= 'delete from '.$this->dbPrefix.'received where date_received < \''.date("Y-m-d H:i:s",$daysago).'\';';
490
-        if ($db_conn->query($sql) === false) {
491
-            $this->logging->log('Error erasing traps : '.$sql,ERROR,'');
492
-        }
493
-        $this->logging->log('Erased traps older than '.$days.' day(s) : '.$sql,INFO);
494
-    }
495
-    
496
-    /** Write error to received trap database
497
-     */
498
-    public function writeTrapErrorToDB($message,$sourceIP=null,$trapoid=null)
499
-    {
500
-        
501
-        $db_conn=$this->trapsDB->db_connect_trap();
502
-        
503
-        // add date time
504
-        $insert_col ='date_received,status';
505
-        $insert_val = "'" . date("Y-m-d H:i:s")."','error'";
506
-        
507
-        if ($sourceIP !=null)
508
-        {
509
-            $insert_col .=',source_ip';
510
-            $insert_val .=",'". $sourceIP ."'";
511
-        }
512
-        if ($trapoid !=null)
513
-        {
514
-            $insert_col .=',trap_oid';
515
-            $insert_val .=",'". $trapoid ."'";
516
-        }
517
-        $insert_col .=',status_detail';
518
-        $insert_val .=",'". $message ."'";
519
-        
520
-        $sql= 'INSERT INTO '.$this->dbPrefix.'received (' . $insert_col . ') VALUES ('.$insert_val.')';
521
-        
522
-        switch ($this->trapsDB->trapDBType)
523
-        {
524
-            case 'pgsql':
525
-                $sql .= ' RETURNING id;';
526
-                $this->logging->log('sql : '.$sql,INFO);
527
-                if (($ret_code=$db_conn->query($sql)) === false) {
528
-                    $this->logging->log('Error SQL insert : '.$sql,1,'');
529
-                }
530
-                $this->logging->log('SQL insertion OK',INFO );
531
-                // Get last id to insert oid/values in secondary table
532
-                if (($inserted_id_ret=$ret_code->fetch(PDO::FETCH_ASSOC)) === false) {
142
+	}
143
+    
144
+	/**
145
+	 * Get option from array of ini file, send message if empty
146
+	 * @param string $option_array Array of ini file
147
+	 * @param string $option_category category in ini file
148
+	 * @param string $option_name name of option in category
149
+	 * @param mixed $option_var variable to fill if found, left untouched if not found
150
+	 * @param integer $log_level default 2 (warning)
151
+	 * @param string $message warning message if not found
152
+	 * @return boolean true if found, or false
153
+	 */
154
+	protected function getOptionIfSet($option_array,$option_category,$option_name, &$option_var, $log_level = 2, $message = null)
155
+	{
156
+		if (!isset($option_array[$option_category][$option_name]))
157
+		{
158
+			if ($message === null)
159
+			{
160
+				$message='No ' . $option_name . ' in config file: '. $this->trapModuleConfig;
161
+			}
162
+			$this->logging->log($message,$log_level);
163
+			return false;
164
+		}
165
+		else
166
+		{
167
+			$option_var=$option_array[$option_category][$option_name];
168
+			return true;
169
+		}
170
+	}
171
+    
172
+	/**
173
+	 * Get options from ini file
174
+	 * @param array $trap_config : ini file array
175
+	 */
176
+	protected function getMainOptions($trapConfig)
177
+	{
178
+        
179
+		// Snmptranslate binary path
180
+		$this->getOptionIfSet($trapConfig,'config','snmptranslate', $this->snmptranslate);
181
+        
182
+		// mibs path
183
+		$this->getOptionIfSet($trapConfig,'config','snmptranslate_dirs', $this->snmptranslate_dirs);
184
+        
185
+		// icinga2cmd path
186
+		$this->getOptionIfSet($trapConfig,'config','icingacmd', $this->icinga2cmd);
187
+        
188
+		// table prefix
189
+		$this->getOptionIfSet($trapConfig,'config','database_prefix', $this->dbPrefix);
190
+        
191
+		// API options
192
+		if ($this->getOptionIfSet($trapConfig,'config','icingaAPI_host', $this->apiHostname))
193
+		{
194
+			$this->apiUse=true;
195
+			$this->getOptionIfSet($trapConfig,'config','icingaAPI_port', $this->apiPort);
196
+			$this->getOptionIfSet($trapConfig,'config','icingaAPI_user', $this->apiUsername);
197
+			$this->getOptionIfSet($trapConfig,'config','icingaAPI_password', $this->apiPassword);
198
+		}
199
+	}
200
+    
201
+	/**
202
+	 * Create and setup database class for trap & ido (if no api) db
203
+	 * @param array $trap_config : ini file array
204
+	 */
205
+	protected function setupDatabase($trapConfig)
206
+	{
207
+		// Trap database
208
+		if (!array_key_exists('database',$trapConfig['config']))
209
+		{
210
+			$this->logging->log("No database in config file: ".$this->trapModuleConfig,ERROR,'');
211
+			return;
212
+		}
213
+		$dbTrapName=$trapConfig['config']['database'];
214
+		$this->logging->log("Found database in config file: ".$dbTrapName,INFO );
215
+        
216
+		if ( ($dbConfig=parse_ini_file($this->icingaweb2Ressources,true)) === false)
217
+		{
218
+			$this->logging->log("Error reading ini file : ".$this->icingaweb2Ressources,ERROR,'');
219
+			return;
220
+		}
221
+		if (!array_key_exists($dbTrapName,$dbConfig))
222
+		{
223
+			$this->logging->log("No database '.$dbTrapName.' in config file: ".$this->icingaweb2Ressources,ERROR,'');
224
+			return;
225
+		}
226
+        
227
+		$this->trapsDB = new Database($this->logging,$dbConfig[$dbTrapName],$this->dbPrefix);
228
+        
229
+		if ($this->apiUse === true) return; // In case of API use, no IDO is necessary
230
+        
231
+		// IDO Database
232
+		if (!array_key_exists('IDOdatabase',$trapConfig['config']))
233
+		{
234
+			$this->logging->log("No IDOdatabase in config file: ".$this->trapModuleConfig,ERROR,'');
235
+		}
236
+		$dbIdoName=$trapConfig['config']['IDOdatabase'];
237
+        
238
+		$this->logging->log("Found IDO database in config file: ".$dbIdoName,INFO );
239
+		if (!array_key_exists($dbIdoName,$dbConfig))
240
+		{
241
+			$this->logging->log("No database '.$dbIdoName.' in config file: ".$this->icingaweb2Ressources,ERROR,'');
242
+			return;
243
+		}
244
+        
245
+		$this->trapsDB->setupIDO($dbConfig[$dbIdoName]);
246
+	}
247
+    
248
+	/**
249
+	 * Get options in database
250
+	 */
251
+	protected function getDatabaseOptions()
252
+	{
253
+		// Database options
254
+		if ($this->logSetup === false) // Only if logging was no setup in constructor
255
+		{
256
+			$this->getDBConfigIfSet('log_level',$this->logging->debugLevel);
257
+			$this->getDBConfigIfSet('log_destination',$this->logging->outputMode);
258
+			$this->getDBConfigIfSet('log_file',$this->logging->outputFile);
259
+		}
260
+	}
261
+    
262
+	/** Set $variable to value if $element found in database config table
263
+	 * @param string $element
264
+	 * @param string $variable
265
+	 */
266
+	protected function getDBConfigIfSet($element,&$variable)
267
+	{
268
+		$value=$this->getDBConfig($element);
269
+		if ($value != null) $variable=$value;
270
+	}
271
+    
272
+	/**
273
+	 *   Get data from db_config
274
+	 *	@param $element string name of param
275
+	 *	@return mixed : value (or null)
276
+	 */
277
+	protected function getDBConfig($element)  // TODO : put this in DB class
278
+	{
279
+		$db_conn=$this->trapsDB->db_connect_trap();
280
+		$sql='SELECT value from '.$this->dbPrefix.'db_config WHERE ( name=\''.$element.'\' )';
281
+		if (($ret_code=$db_conn->query($sql)) === false) {
282
+			$this->logging->log('No result in query : ' . $sql,WARN,'');
283
+			return null;
284
+		}
285
+		$value=$ret_code->fetch();
286
+		if ($value != null && isset($value['value']))
287
+		{
288
+			return $value['value'];
289
+		}
290
+		return null;
291
+	}
292
+    
293
+	/** OBSOLETE Send log. Throws exception on critical error
294
+	 *	@param	string $message Message to log
295
+	 *	@param	int $level 1=critical 2=warning 3=trace 4=debug
296
+	 *	@param  string $destination file/syslog/display
297
+	 *	@return void
298
+	 **/
299
+	public function trapLog( $message, $level, $destination ='') // OBSOLETE
300
+	{
301
+		// TODO : replace ref with $this->logging->log
302
+		$this->logging->log($message, $level, $destination);
303
+	}
304
+    
305
+	public function setLogging($debugLvl,$outputType,$outputOption=null)  // OBSOLETE
306
+	{
307
+		$this->logging->setLogging($debugLvl, $outputType,$outputOption);
308
+	}
309
+    
310
+	/**
311
+	 * Returns or create new IcingaAPI object
312
+	 * @return \Icinga\Module\Trapdirector\Icinga2API
313
+	 */
314
+	protected function getAPI()
315
+	{
316
+		if ($this->icinga2api == null)
317
+		{
318
+			$this->icinga2api = new Icinga2API($this->apiHostname,$this->apiPort);
319
+		}
320
+		return $this->icinga2api;
321
+	}
322
+    
323
+    
324
+	/**
325
+	 * read data from stream
326
+	 *	@param $stream string input stream, defaults to "php://stdin"
327
+	 *	@return mixed array trap data or exception with error
328
+	 */
329
+	public function read_trap($stream='php://stdin')
330
+	{
331
+		//Read data from snmptrapd from stdin
332
+		$input_stream=fopen($stream, 'r');
333
+        
334
+		if ($input_stream === false)
335
+		{
336
+			$this->writeTrapErrorToDB("Error reading trap (code 1/Stdin)");
337
+			$this->logging->log("Error reading stdin !",ERROR,'');
338
+			return null; // note : exception thrown by logging
339
+		}
340
+        
341
+		// line 1 : host
342
+		$this->receivingHost=chop(fgets($input_stream));
343
+		if ($this->receivingHost === false)
344
+		{
345
+			$this->writeTrapErrorToDB("Error reading trap (code 1/Line Host)");
346
+			$this->logging->log("Error reading Host !",ERROR,'');
347
+		}
348
+		// line 2 IP:port=>IP:port
349
+		$IP=chop(fgets($input_stream));
350
+		if ($IP === false)
351
+		{
352
+			$this->writeTrapErrorToDB("Error reading trap (code 1/Line IP)");
353
+			$this->logging->log("Error reading IP !",ERROR,'');
354
+		}
355
+		$matches=array();
356
+		$ret_code=preg_match('/.DP: \[(.*)\]:(.*)->\[(.*)\]:(.*)/',$IP,$matches);
357
+		if ($ret_code===0 || $ret_code===false)
358
+		{
359
+			$this->writeTrapErrorToDB("Error parsing trap (code 2/IP)");
360
+			$this->logging->log('Error parsing IP : '.$IP,ERROR,'');
361
+		}
362
+		else
363
+		{
364
+			$this->trapData['source_ip']=$matches[1];
365
+			$this->trapData['destination_ip']=$matches[3];
366
+			$this->trapData['source_port']=$matches[2];
367
+			$this->trapData['destination_port']=$matches[4];
368
+		}
369
+        
370
+		while (($vars=fgets($input_stream)) !==false)
371
+		{
372
+			$vars=chop($vars);
373
+			$ret_code=preg_match('/^([^ ]+) (.*)$/',$vars,$matches);
374
+			if ($ret_code===0 || $ret_code===false)
375
+			{
376
+				$this->logging->log('No match on trap data : '.$vars,WARN,'');
377
+			}
378
+			else
379
+			{
380
+				if (($matches[1]=='.1.3.6.1.6.3.1.1.4.1.0') || ($matches[1]=='.1.3.6.1.6.3.1.1.4.1'))
381
+				{
382
+					$this->trapData['trap_oid']=$matches[2];
383
+				}
384
+				else
385
+				{
386
+					$object= new stdClass;
387
+					$object->oid =$matches[1];
388
+					$object->value = $matches[2];
389
+					array_push($this->trapDataExt,$object);
390
+				}
391
+			}
392
+		}
393
+        
394
+		if ($this->trapData['trap_oid']=='unknown')
395
+		{
396
+			$this->writeTrapErrorToDB("No trap oid found : check snmptrapd configuration (code 3/OID)",$this->trapData['source_ip']);
397
+			$this->logging->log('no trap oid found',ERROR,'');
398
+		}
399
+        
400
+		// Translate oids.
401
+        
402
+		$retArray=$this->translateOID($this->trapData['trap_oid']);
403
+		if ($retArray != null)
404
+		{
405
+			$this->trapData['trap_name']=$retArray['trap_name'];
406
+			$this->trapData['trap_name_mib']=$retArray['trap_name_mib'];
407
+		}
408
+		foreach ($this->trapDataExt as $key => $val)
409
+		{
410
+			$retArray=$this->translateOID($val->oid);
411
+			if ($retArray != null)
412
+			{
413
+				$this->trapDataExt[$key]->oid_name=$retArray['trap_name'];
414
+				$this->trapDataExt[$key]->oid_name_mib=$retArray['trap_name_mib'];
415
+			}
416
+		}
417
+        
418
+        
419
+		$this->trapData['status']= 'waiting';
420
+        
421
+		return $this->trapData;
422
+	}
423
+    
424
+	/**
425
+	 * Translate oid into array(MIB,Name)
426
+	 * @param $oid string oid to translate
427
+	 * @return mixed : null if not found or array(MIB,Name)
428
+	 */
429
+	public function translateOID($oid)
430
+	{
431
+		// try from database
432
+		$db_conn=$this->trapsDB->db_connect_trap();
433
+        
434
+		$sql='SELECT mib,name from '.$this->dbPrefix.'mib_cache WHERE oid=\''.$oid.'\';';
435
+		$this->logging->log('SQL query : '.$sql,DEBUG );
436
+		if (($ret_code=$db_conn->query($sql)) === false) {
437
+			$this->logging->log('No result in query : ' . $sql,ERROR,'');
438
+		}
439
+		$name=$ret_code->fetch();
440
+		if ($name['name'] != null)
441
+		{
442
+			return array('trap_name_mib'=>$name['mib'],'trap_name'=>$name['name']);
443
+		}
444
+        
445
+		// Also check if it is an instance of OID
446
+		$oid_instance=preg_replace('/\.[0-9]+$/','',$oid);
447
+        
448
+		$sql='SELECT mib,name from '.$this->dbPrefix.'mib_cache WHERE oid=\''.$oid_instance.'\';';
449
+		$this->logging->log('SQL query : '.$sql,DEBUG );
450
+		if (($ret_code=$db_conn->query($sql)) === false) {
451
+			$this->logging->log('No result in query : ' . $sql,ERROR,'');
452
+		}
453
+		$name=$ret_code->fetch();
454
+		if ($name['name'] != null)
455
+		{
456
+			return array('trap_name_mib'=>$name['mib'],'trap_name'=>$name['name']);
457
+		}
458
+        
459
+		// Try to get oid name from snmptranslate
460
+		$translate=exec($this->snmptranslate . ' -m ALL -M +'.$this->snmptranslate_dirs.
461
+			' '.$oid);
462
+		$matches=array();
463
+		$ret_code=preg_match('/(.*)::(.*)/',$translate,$matches);
464
+		if ($ret_code===0 || $ret_code === false) {
465
+			return NULL;
466
+		} else {
467
+			$this->logging->log('Found name with snmptrapd and not in DB for oid='.$oid,INFO);
468
+			return array('trap_name_mib'=>$matches[1],'trap_name'=>$matches[2]);
469
+		}
470
+	}
471
+    
472
+	/**
473
+	 * Erase old trap records
474
+	 *	@param integer $days : erase traps when more than $days old
475
+	 *	@return integer : number of lines deleted
476
+	 **/
477
+	public function eraseOldTraps($days=0)
478
+	{
479
+		if ($days==0)
480
+		{
481
+			if (($days=$this->getDBConfig('db_remove_days')) == null)
482
+			{
483
+				$this->logging->log('No days specified & no db value : no tap erase' ,WARN,'');
484
+				return;
485
+			}
486
+		}
487
+		$db_conn=$this->trapsDB->db_connect_trap();
488
+		$daysago = strtotime("-".$days." day");
489
+		$sql= 'delete from '.$this->dbPrefix.'received where date_received < \''.date("Y-m-d H:i:s",$daysago).'\';';
490
+		if ($db_conn->query($sql) === false) {
491
+			$this->logging->log('Error erasing traps : '.$sql,ERROR,'');
492
+		}
493
+		$this->logging->log('Erased traps older than '.$days.' day(s) : '.$sql,INFO);
494
+	}
495
+    
496
+	/** Write error to received trap database
497
+	 */
498
+	public function writeTrapErrorToDB($message,$sourceIP=null,$trapoid=null)
499
+	{
500
+        
501
+		$db_conn=$this->trapsDB->db_connect_trap();
502
+        
503
+		// add date time
504
+		$insert_col ='date_received,status';
505
+		$insert_val = "'" . date("Y-m-d H:i:s")."','error'";
506
+        
507
+		if ($sourceIP !=null)
508
+		{
509
+			$insert_col .=',source_ip';
510
+			$insert_val .=",'". $sourceIP ."'";
511
+		}
512
+		if ($trapoid !=null)
513
+		{
514
+			$insert_col .=',trap_oid';
515
+			$insert_val .=",'". $trapoid ."'";
516
+		}
517
+		$insert_col .=',status_detail';
518
+		$insert_val .=",'". $message ."'";
519
+        
520
+		$sql= 'INSERT INTO '.$this->dbPrefix.'received (' . $insert_col . ') VALUES ('.$insert_val.')';
521
+        
522
+		switch ($this->trapsDB->trapDBType)
523
+		{
524
+			case 'pgsql':
525
+				$sql .= ' RETURNING id;';
526
+				$this->logging->log('sql : '.$sql,INFO);
527
+				if (($ret_code=$db_conn->query($sql)) === false) {
528
+					$this->logging->log('Error SQL insert : '.$sql,1,'');
529
+				}
530
+				$this->logging->log('SQL insertion OK',INFO );
531
+				// Get last id to insert oid/values in secondary table
532
+				if (($inserted_id_ret=$ret_code->fetch(PDO::FETCH_ASSOC)) === false) {
533 533
                     
534
-                    $this->logging->log('Erreur recuperation id',1,'');
535
-                }
536
-                if (! isset($inserted_id_ret['id'])) {
537
-                    $this->logging->log('Error getting id',1,'');
538
-                }
539
-                $this->trapId=$inserted_id_ret['id'];
540
-                break;
541
-            case 'mysql':
542
-                $sql .= ';';
543
-                $this->logging->log('sql : '.$sql,INFO );
544
-                if ($db_conn->query($sql) === false) {
545
-                    $this->logging->log('Error SQL insert : '.$sql,1,'');
546
-                }
547
-                $this->logging->log('SQL insertion OK',INFO );
548
-                // Get last id to insert oid/values in secondary table
549
-                $sql='SELECT LAST_INSERT_ID();';
550
-                if (($ret_code=$db_conn->query($sql)) === false) {
551
-                    $this->logging->log('Erreur recuperation id',1,'');
552
-                }
534
+					$this->logging->log('Erreur recuperation id',1,'');
535
+				}
536
+				if (! isset($inserted_id_ret['id'])) {
537
+					$this->logging->log('Error getting id',1,'');
538
+				}
539
+				$this->trapId=$inserted_id_ret['id'];
540
+				break;
541
+			case 'mysql':
542
+				$sql .= ';';
543
+				$this->logging->log('sql : '.$sql,INFO );
544
+				if ($db_conn->query($sql) === false) {
545
+					$this->logging->log('Error SQL insert : '.$sql,1,'');
546
+				}
547
+				$this->logging->log('SQL insertion OK',INFO );
548
+				// Get last id to insert oid/values in secondary table
549
+				$sql='SELECT LAST_INSERT_ID();';
550
+				if (($ret_code=$db_conn->query($sql)) === false) {
551
+					$this->logging->log('Erreur recuperation id',1,'');
552
+				}
553 553
                 
554
-                $inserted_id=$ret_code->fetch(PDO::FETCH_ASSOC)['LAST_INSERT_ID()'];
555
-                if ($inserted_id==false) throw new Exception("Weird SQL error : last_insert_id returned false : open issue");
556
-                $this->trapId=$inserted_id;
557
-                break;
558
-            default:
559
-                $this->logging->log('Error SQL type unknown  : '.$this->trapsDB->trapDBType,1,'');
560
-        }
561
-        
562
-        $this->logging->log('id found: '. $this->trapId,INFO );
563
-    }
564
-    
565
-    /** Write trap data to trap database
566
-     */
567
-    public function writeTrapToDB()
568
-    {
569
-        
570
-        // If action is ignore -> don't send t DB
571
-        if ($this->trapToDb === false) return;
572
-        
573
-        
574
-        $db_conn=$this->trapsDB->db_connect_trap();
575
-        
576
-        $insert_col='';
577
-        $insert_val='';
578
-        // add date time
579
-        $this->trapData['date_received'] = date("Y-m-d H:i:s");
580
-        
581
-        $firstcol=1;
582
-        foreach ($this->trapData as $col => $val)
583
-        {
584
-            if ($firstcol==0)
585
-            {
586
-                $insert_col .=',';
587
-                $insert_val .=',';
588
-            }
589
-            $insert_col .= $col ;
590
-            $insert_val .= ($val==null)? 'NULL' : $db_conn->quote($val);
591
-            $firstcol=0;
592
-        }
593
-        
594
-        $sql= 'INSERT INTO '.$this->dbPrefix.'received (' . $insert_col . ') VALUES ('.$insert_val.')';
595
-        switch ($this->trapsDB->trapDBType)
596
-        {
597
-            case 'pgsql':
598
-                $sql .= ' RETURNING id;';
599
-                $this->logging->log('sql : '.$sql,INFO );
600
-                if (($ret_code=$db_conn->query($sql)) === false) {
601
-                    $this->logging->log('Error SQL insert : '.$sql,ERROR,'');
602
-                }
603
-                $this->logging->log('SQL insertion OK',INFO );
604
-                // Get last id to insert oid/values in secondary table
605
-                if (($inserted_id_ret=$ret_code->fetch(PDO::FETCH_ASSOC)) === false) {
554
+				$inserted_id=$ret_code->fetch(PDO::FETCH_ASSOC)['LAST_INSERT_ID()'];
555
+				if ($inserted_id==false) throw new Exception("Weird SQL error : last_insert_id returned false : open issue");
556
+				$this->trapId=$inserted_id;
557
+				break;
558
+			default:
559
+				$this->logging->log('Error SQL type unknown  : '.$this->trapsDB->trapDBType,1,'');
560
+		}
561
+        
562
+		$this->logging->log('id found: '. $this->trapId,INFO );
563
+	}
564
+    
565
+	/** Write trap data to trap database
566
+	 */
567
+	public function writeTrapToDB()
568
+	{
569
+        
570
+		// If action is ignore -> don't send t DB
571
+		if ($this->trapToDb === false) return;
572
+        
573
+        
574
+		$db_conn=$this->trapsDB->db_connect_trap();
575
+        
576
+		$insert_col='';
577
+		$insert_val='';
578
+		// add date time
579
+		$this->trapData['date_received'] = date("Y-m-d H:i:s");
580
+        
581
+		$firstcol=1;
582
+		foreach ($this->trapData as $col => $val)
583
+		{
584
+			if ($firstcol==0)
585
+			{
586
+				$insert_col .=',';
587
+				$insert_val .=',';
588
+			}
589
+			$insert_col .= $col ;
590
+			$insert_val .= ($val==null)? 'NULL' : $db_conn->quote($val);
591
+			$firstcol=0;
592
+		}
593
+        
594
+		$sql= 'INSERT INTO '.$this->dbPrefix.'received (' . $insert_col . ') VALUES ('.$insert_val.')';
595
+		switch ($this->trapsDB->trapDBType)
596
+		{
597
+			case 'pgsql':
598
+				$sql .= ' RETURNING id;';
599
+				$this->logging->log('sql : '.$sql,INFO );
600
+				if (($ret_code=$db_conn->query($sql)) === false) {
601
+					$this->logging->log('Error SQL insert : '.$sql,ERROR,'');
602
+				}
603
+				$this->logging->log('SQL insertion OK',INFO );
604
+				// Get last id to insert oid/values in secondary table
605
+				if (($inserted_id_ret=$ret_code->fetch(PDO::FETCH_ASSOC)) === false) {
606 606
                     
607
-                    $this->logging->log('Erreur recuperation id',ERROR,'');
608
-                }
609
-                if (! isset($inserted_id_ret['id'])) {
610
-                    $this->logging->log('Error getting id',ERROR,'');
611
-                }
612
-                $this->trapId=$inserted_id_ret['id'];
613
-                break;
614
-            case 'mysql':
615
-                $sql .= ';';
616
-                $this->logging->log('sql : '.$sql,INFO );
617
-                if ($db_conn->query($sql) === false) {
618
-                    $this->logging->log('Error SQL insert : '.$sql,ERROR,'');
619
-                }
620
-                $this->logging->log('SQL insertion OK',INFO );
621
-                // Get last id to insert oid/values in secondary table
622
-                $sql='SELECT LAST_INSERT_ID();';
623
-                if (($ret_code=$db_conn->query($sql)) === false) {
624
-                    $this->logging->log('Erreur recuperation id',ERROR,'');
625
-                }
607
+					$this->logging->log('Erreur recuperation id',ERROR,'');
608
+				}
609
+				if (! isset($inserted_id_ret['id'])) {
610
+					$this->logging->log('Error getting id',ERROR,'');
611
+				}
612
+				$this->trapId=$inserted_id_ret['id'];
613
+				break;
614
+			case 'mysql':
615
+				$sql .= ';';
616
+				$this->logging->log('sql : '.$sql,INFO );
617
+				if ($db_conn->query($sql) === false) {
618
+					$this->logging->log('Error SQL insert : '.$sql,ERROR,'');
619
+				}
620
+				$this->logging->log('SQL insertion OK',INFO );
621
+				// Get last id to insert oid/values in secondary table
622
+				$sql='SELECT LAST_INSERT_ID();';
623
+				if (($ret_code=$db_conn->query($sql)) === false) {
624
+					$this->logging->log('Erreur recuperation id',ERROR,'');
625
+				}
626 626
                 
627
-                $inserted_id=$ret_code->fetch(PDO::FETCH_ASSOC)['LAST_INSERT_ID()'];
628
-                if ($inserted_id==false) throw new Exception("Weird SQL error : last_insert_id returned false : open issue");
629
-                $this->trapId=$inserted_id;
630
-                break;
631
-            default:
632
-                $this->logging->log('Error SQL type unknown : '.$this->trapsDB->trapDBType,ERROR,'');
633
-        }
634
-        $this->logging->log('id found: '.$this->trapId,INFO );
635
-        
636
-        // Fill trap extended data table
637
-        foreach ($this->trapDataExt as $value) {
638
-            // TODO : detect if trap value is encoded and decode it to UTF-8 for database
639
-            $firstcol=1;
640
-            $value->trap_id = $this->trapId;
641
-            $insert_col='';
642
-            $insert_val='';
643
-            foreach ($value as $col => $val)
644
-            {
645
-                if ($firstcol==0)
646
-                {
647
-                    $insert_col .=',';
648
-                    $insert_val .=',';
649
-                }
650
-                $insert_col .= $col;
651
-                $insert_val .= ($val==null)? 'NULL' : $db_conn->quote($val);
652
-                $firstcol=0;
653
-            }
627
+				$inserted_id=$ret_code->fetch(PDO::FETCH_ASSOC)['LAST_INSERT_ID()'];
628
+				if ($inserted_id==false) throw new Exception("Weird SQL error : last_insert_id returned false : open issue");
629
+				$this->trapId=$inserted_id;
630
+				break;
631
+			default:
632
+				$this->logging->log('Error SQL type unknown : '.$this->trapsDB->trapDBType,ERROR,'');
633
+		}
634
+		$this->logging->log('id found: '.$this->trapId,INFO );
635
+        
636
+		// Fill trap extended data table
637
+		foreach ($this->trapDataExt as $value) {
638
+			// TODO : detect if trap value is encoded and decode it to UTF-8 for database
639
+			$firstcol=1;
640
+			$value->trap_id = $this->trapId;
641
+			$insert_col='';
642
+			$insert_val='';
643
+			foreach ($value as $col => $val)
644
+			{
645
+				if ($firstcol==0)
646
+				{
647
+					$insert_col .=',';
648
+					$insert_val .=',';
649
+				}
650
+				$insert_col .= $col;
651
+				$insert_val .= ($val==null)? 'NULL' : $db_conn->quote($val);
652
+				$firstcol=0;
653
+			}
654 654
             
655
-            $sql= 'INSERT INTO '.$this->dbPrefix.'received_data (' . $insert_col . ') VALUES ('.$insert_val.');';
655
+			$sql= 'INSERT INTO '.$this->dbPrefix.'received_data (' . $insert_col . ') VALUES ('.$insert_val.');';
656 656
             
657
-            if ($db_conn->query($sql) === false) {
658
-                $this->logging->log('Erreur insertion data : ' . $sql,WARN,'');
659
-            }
660
-        }
661
-    }
662
-    
663
-    /** Get rules from rule database with ip and oid
664
-     *	@param $ip string ipv4 or ipv6
665
-     *	@param $oid string oid in numeric
666
-     *	@return mixed|boolean : PDO object or false
667
-     */
668
-    protected function getRules($ip,$oid)
669
-    {
670
-        $db_conn=$this->trapsDB->db_connect_trap();
671
-        // fetch rules based on IP in rule and OID
672
-        $sql='SELECT * from '.$this->dbPrefix.'rules WHERE trap_oid=\''.$oid.'\' ';
673
-        $this->logging->log('SQL query : '.$sql,DEBUG );
674
-        if (($ret_code=$db_conn->query($sql)) === false) {
675
-            $this->logging->log('No result in query : ' . $sql,WARN,'');
676
-            return false;
677
-        }
678
-        $rules_all=$ret_code->fetchAll();
679
-        //echo "rule all :\n";print_r($rules_all);echo "\n";
680
-        $rules_ret=array();
681
-        $rule_ret_key=0;
682
-        foreach ($rules_all as $key => $rule)
683
-        {
684
-            if ($rule['ip4']==$ip || $rule['ip6']==$ip)
685
-            {
686
-                $rules_ret[$rule_ret_key]=$rules_all[$key];
687
-                //TODO : get host name by API (and check if correct in rule).
688
-                $rule_ret_key++;
689
-                continue;
690
-            }
691
-            // TODO : get hosts IP by API
692
-            if (isset($rule['host_group_name']) && $rule['host_group_name']!=null)
693
-            { // get ips of group members by oid
694
-                $db_conn2=$this->trapsDB->db_connect_ido();
695
-                $sql="SELECT m.host_object_id, a.address as ip4, a.address6 as ip6, b.name1 as host_name
657
+			if ($db_conn->query($sql) === false) {
658
+				$this->logging->log('Erreur insertion data : ' . $sql,WARN,'');
659
+			}
660
+		}
661
+	}
662
+    
663
+	/** Get rules from rule database with ip and oid
664
+	 *	@param $ip string ipv4 or ipv6
665
+	 *	@param $oid string oid in numeric
666
+	 *	@return mixed|boolean : PDO object or false
667
+	 */
668
+	protected function getRules($ip,$oid)
669
+	{
670
+		$db_conn=$this->trapsDB->db_connect_trap();
671
+		// fetch rules based on IP in rule and OID
672
+		$sql='SELECT * from '.$this->dbPrefix.'rules WHERE trap_oid=\''.$oid.'\' ';
673
+		$this->logging->log('SQL query : '.$sql,DEBUG );
674
+		if (($ret_code=$db_conn->query($sql)) === false) {
675
+			$this->logging->log('No result in query : ' . $sql,WARN,'');
676
+			return false;
677
+		}
678
+		$rules_all=$ret_code->fetchAll();
679
+		//echo "rule all :\n";print_r($rules_all);echo "\n";
680
+		$rules_ret=array();
681
+		$rule_ret_key=0;
682
+		foreach ($rules_all as $key => $rule)
683
+		{
684
+			if ($rule['ip4']==$ip || $rule['ip6']==$ip)
685
+			{
686
+				$rules_ret[$rule_ret_key]=$rules_all[$key];
687
+				//TODO : get host name by API (and check if correct in rule).
688
+				$rule_ret_key++;
689
+				continue;
690
+			}
691
+			// TODO : get hosts IP by API
692
+			if (isset($rule['host_group_name']) && $rule['host_group_name']!=null)
693
+			{ // get ips of group members by oid
694
+				$db_conn2=$this->trapsDB->db_connect_ido();
695
+				$sql="SELECT m.host_object_id, a.address as ip4, a.address6 as ip6, b.name1 as host_name
696 696
 						FROM icinga_objects as o
697 697
 						LEFT JOIN icinga_hostgroups as h ON o.object_id=h.hostgroup_object_id
698 698
 						LEFT JOIN icinga_hostgroup_members as m ON h.hostgroup_id=m.hostgroup_id
699 699
 						LEFT JOIN icinga_hosts as a ON a.host_object_id = m.host_object_id
700 700
 						LEFT JOIN icinga_objects as b ON b.object_id = a.host_object_id
701 701
 						WHERE o.name1='".$rule['host_group_name']."';";
702
-                if (($ret_code2=$db_conn2->query($sql)) === false) {
703
-                    $this->logging->log('No result in query : ' . $sql,WARN,'');
704
-                    continue;
705
-                }
706
-                $grouphosts=$ret_code2->fetchAll();
707
-                //echo "rule grp :\n";print_r($grouphosts);echo "\n";
708
-                foreach ( $grouphosts as $host)
709
-                {
710
-                    //echo $host['ip4']."\n";
711
-                    if ($host['ip4']==$ip || $host['ip6']==$ip)
712
-                    {
713
-                        //echo "Rule added \n";
714
-                        $rules_ret[$rule_ret_key]=$rules_all[$key];
715
-                        $rules_ret[$rule_ret_key]['host_name']=$host['host_name'];
716
-                        $rule_ret_key++;
717
-                    }
718
-                }
719
-            }
720
-        }
721
-        //echo "rule rest :\n";print_r($rules_ret);echo "\n";exit(0);
722
-        return $rules_ret;
723
-    }
724
-    
725
-    /** Add rule match to rule
726
-     *	@param id int : rule id
727
-     *   @param set int : value to set
728
-     */
729
-    protected function add_rule_match($id, $set)
730
-    {
731
-        $db_conn=$this->trapsDB->db_connect_trap();
732
-        $sql="UPDATE ".$this->dbPrefix."rules SET num_match = '".$set."' WHERE (id = '".$id."');";
733
-        if ($db_conn->query($sql) === false) {
734
-            $this->logging->log('Error in update query : ' . $sql,WARN,'');
735
-        }
736
-    }
737
-    
738
-    /** Send SERVICE_CHECK_RESULT with icinga2cmd or API
739
-     *
740
-     * @param string $host
741
-     * @param string $service
742
-     * @param integer $state numerical staus
743
-     * @param string $display
744
-     * @returnn bool true is service check was sent without error
745
-     */
746
-    public function serviceCheckResult($host,$service,$state,$display)
747
-    {
748
-        if ($this->apiUse === false)
749
-        {
750
-            $send = '[' . date('U') .'] PROCESS_SERVICE_CHECK_RESULT;' .
751
-                $host.';' .$service .';' . $state . ';'.$display;
752
-                $this->logging->log( $send." : to : " .$this->icinga2cmd,INFO );
702
+				if (($ret_code2=$db_conn2->query($sql)) === false) {
703
+					$this->logging->log('No result in query : ' . $sql,WARN,'');
704
+					continue;
705
+				}
706
+				$grouphosts=$ret_code2->fetchAll();
707
+				//echo "rule grp :\n";print_r($grouphosts);echo "\n";
708
+				foreach ( $grouphosts as $host)
709
+				{
710
+					//echo $host['ip4']."\n";
711
+					if ($host['ip4']==$ip || $host['ip6']==$ip)
712
+					{
713
+						//echo "Rule added \n";
714
+						$rules_ret[$rule_ret_key]=$rules_all[$key];
715
+						$rules_ret[$rule_ret_key]['host_name']=$host['host_name'];
716
+						$rule_ret_key++;
717
+					}
718
+				}
719
+			}
720
+		}
721
+		//echo "rule rest :\n";print_r($rules_ret);echo "\n";exit(0);
722
+		return $rules_ret;
723
+	}
724
+    
725
+	/** Add rule match to rule
726
+	 *	@param id int : rule id
727
+	 *   @param set int : value to set
728
+	 */
729
+	protected function add_rule_match($id, $set)
730
+	{
731
+		$db_conn=$this->trapsDB->db_connect_trap();
732
+		$sql="UPDATE ".$this->dbPrefix."rules SET num_match = '".$set."' WHERE (id = '".$id."');";
733
+		if ($db_conn->query($sql) === false) {
734
+			$this->logging->log('Error in update query : ' . $sql,WARN,'');
735
+		}
736
+	}
737
+    
738
+	/** Send SERVICE_CHECK_RESULT with icinga2cmd or API
739
+	 *
740
+	 * @param string $host
741
+	 * @param string $service
742
+	 * @param integer $state numerical staus
743
+	 * @param string $display
744
+	 * @returnn bool true is service check was sent without error
745
+	 */
746
+	public function serviceCheckResult($host,$service,$state,$display)
747
+	{
748
+		if ($this->apiUse === false)
749
+		{
750
+			$send = '[' . date('U') .'] PROCESS_SERVICE_CHECK_RESULT;' .
751
+				$host.';' .$service .';' . $state . ';'.$display;
752
+				$this->logging->log( $send." : to : " .$this->icinga2cmd,INFO );
753 753
                 
754
-                // TODO : file_put_contents & fopen (,'w' or 'a') does not work. See why. Or not as using API will be by default....
755
-                exec('echo "'.$send.'" > ' .$this->icinga2cmd);
756
-                return true;
757
-        }
758
-        else
759
-        {
760
-            // Get perfdata if found
761
-            $matches=array();
762
-            if (preg_match('/(.*)\|(.*)/',$display,$matches) == 1)
763
-            {
764
-                $display=$matches[1];
765
-                $perfdata=$matches[2];
766
-            }
767
-            else
768
-            {
769
-                $perfdata='';
770
-            }
754
+				// TODO : file_put_contents & fopen (,'w' or 'a') does not work. See why. Or not as using API will be by default....
755
+				exec('echo "'.$send.'" > ' .$this->icinga2cmd);
756
+				return true;
757
+		}
758
+		else
759
+		{
760
+			// Get perfdata if found
761
+			$matches=array();
762
+			if (preg_match('/(.*)\|(.*)/',$display,$matches) == 1)
763
+			{
764
+				$display=$matches[1];
765
+				$perfdata=$matches[2];
766
+			}
767
+			else
768
+			{
769
+				$perfdata='';
770
+			}
771 771
             
772
-            $api = $this->getAPI();
773
-            $api->setCredentials($this->apiUsername, $this->apiPassword);
774
-            list($retcode,$retmessage)=$api->serviceCheckResult($host,$service,$state,$display,$perfdata);
775
-            if ($retcode == false)
776
-            {
777
-                $this->logging->log( "Error sending result : " .$retmessage,WARN,'');
778
-                return false;
779
-            }
780
-            else
781
-            {
782
-                $this->logging->log( "Sent result : " .$retmessage,INFO );
783
-                return true;
784
-            }
785
-        }
786
-    }
787
-    
788
-    public function getHostByIP($ip)
789
-    {
790
-        $api = $this->getAPI();
791
-        $api->setCredentials($this->apiUsername, $this->apiPassword);
792
-        return $api->getHostByIP($ip);
793
-    }
794
-    
795
-    /** Resolve display.
796
-     *	Changes _OID(<oid>) to value if found or text "<not in trap>"
797
-     *	@param $display string
798
-     *	@return string display
799
-     */
800
-    protected function applyDisplay($display)
801
-    {
802
-        $matches=array();
803
-        while (preg_match('/_OID\(([0-9\.\*]+)\)/',$display,$matches) == 1)
804
-        {
805
-            $oid=$matches[1];
806
-            $found=0;
807
-            // Test and transform regexp
808
-            $oidR = $this->ruleClass->regexp_eval($oid);
772
+			$api = $this->getAPI();
773
+			$api->setCredentials($this->apiUsername, $this->apiPassword);
774
+			list($retcode,$retmessage)=$api->serviceCheckResult($host,$service,$state,$display,$perfdata);
775
+			if ($retcode == false)
776
+			{
777
+				$this->logging->log( "Error sending result : " .$retmessage,WARN,'');
778
+				return false;
779
+			}
780
+			else
781
+			{
782
+				$this->logging->log( "Sent result : " .$retmessage,INFO );
783
+				return true;
784
+			}
785
+		}
786
+	}
787
+    
788
+	public function getHostByIP($ip)
789
+	{
790
+		$api = $this->getAPI();
791
+		$api->setCredentials($this->apiUsername, $this->apiPassword);
792
+		return $api->getHostByIP($ip);
793
+	}
794
+    
795
+	/** Resolve display.
796
+	 *	Changes _OID(<oid>) to value if found or text "<not in trap>"
797
+	 *	@param $display string
798
+	 *	@return string display
799
+	 */
800
+	protected function applyDisplay($display)
801
+	{
802
+		$matches=array();
803
+		while (preg_match('/_OID\(([0-9\.\*]+)\)/',$display,$matches) == 1)
804
+		{
805
+			$oid=$matches[1];
806
+			$found=0;
807
+			// Test and transform regexp
808
+			$oidR = $this->ruleClass->regexp_eval($oid);
809 809
             
810
-            foreach($this->trapDataExt as $val)
811
-            {
812
-                if (preg_match("/^$oidR$/",$val->oid) == 1)
813
-                {
814
-                    $val->value=preg_replace('/"/','',$val->value);
815
-                    $rep=0;
816
-                    $display=preg_replace('/_OID\('.$oid.'\)/',$val->value,$display,-1,$rep);
817
-                    if ($rep==0)
818
-                    {
819
-                        $this->logging->log("Error in display",WARN,'');
820
-                        return $display;
821
-                    }
822
-                    $found=1;
823
-                    break;
824
-                }
825
-            }
826
-            if ($found==0)
827
-            {
828
-                $display=preg_replace('/_OID\('.$oid.'\)/','<not in trap>',$display,-1,$rep);
829
-                if ($rep==0)
830
-                {
831
-                    $this->logging->log("Error in display",WARN,'');
832
-                    return $display;
833
-                }
834
-            }
835
-        }
836
-        return $display;
837
-    }
838
-    
839
-    /** Match rules for current trap and do action
840
-     */
841
-    public function applyRules()
842
-    {
843
-        $rules = $this->getRules($this->trapData['source_ip'],$this->trapData['trap_oid']);
844
-        
845
-        if ($rules===false || count($rules)==0)
846
-        {
847
-            $this->logging->log('No rules found for this trap',INFO );
848
-            $this->trapData['status']='unknown';
849
-            $this->trapToDb=true;
850
-            return;
851
-        }
852
-        //print_r($rules);
853
-        // Evaluate all rules in sequence
854
-        $this->trapAction=null;
855
-        foreach ($rules as $rule)
856
-        {
810
+			foreach($this->trapDataExt as $val)
811
+			{
812
+				if (preg_match("/^$oidR$/",$val->oid) == 1)
813
+				{
814
+					$val->value=preg_replace('/"/','',$val->value);
815
+					$rep=0;
816
+					$display=preg_replace('/_OID\('.$oid.'\)/',$val->value,$display,-1,$rep);
817
+					if ($rep==0)
818
+					{
819
+						$this->logging->log("Error in display",WARN,'');
820
+						return $display;
821
+					}
822
+					$found=1;
823
+					break;
824
+				}
825
+			}
826
+			if ($found==0)
827
+			{
828
+				$display=preg_replace('/_OID\('.$oid.'\)/','<not in trap>',$display,-1,$rep);
829
+				if ($rep==0)
830
+				{
831
+					$this->logging->log("Error in display",WARN,'');
832
+					return $display;
833
+				}
834
+			}
835
+		}
836
+		return $display;
837
+	}
838
+    
839
+	/** Match rules for current trap and do action
840
+	 */
841
+	public function applyRules()
842
+	{
843
+		$rules = $this->getRules($this->trapData['source_ip'],$this->trapData['trap_oid']);
844
+        
845
+		if ($rules===false || count($rules)==0)
846
+		{
847
+			$this->logging->log('No rules found for this trap',INFO );
848
+			$this->trapData['status']='unknown';
849
+			$this->trapToDb=true;
850
+			return;
851
+		}
852
+		//print_r($rules);
853
+		// Evaluate all rules in sequence
854
+		$this->trapAction=null;
855
+		foreach ($rules as $rule)
856
+		{
857 857
             
858
-            $host_name=$rule['host_name'];
859
-            $service_name=$rule['service_name'];
858
+			$host_name=$rule['host_name'];
859
+			$service_name=$rule['service_name'];
860 860
             
861
-            $display=$this->applyDisplay($rule['display']);
862
-            $this->trapAction = ($this->trapAction==null)? '' : $this->trapAction . ', ';
863
-            try
864
-            {
865
-                $this->logging->log('Rule to eval : '.$rule['rule'],INFO );
866
-                $evalr=$this->ruleClass->eval_rule($rule['rule'], $this->trapDataExt) ;
867
-                //->eval_rule($rule['rule']);
861
+			$display=$this->applyDisplay($rule['display']);
862
+			$this->trapAction = ($this->trapAction==null)? '' : $this->trapAction . ', ';
863
+			try
864
+			{
865
+				$this->logging->log('Rule to eval : '.$rule['rule'],INFO );
866
+				$evalr=$this->ruleClass->eval_rule($rule['rule'], $this->trapDataExt) ;
867
+				//->eval_rule($rule['rule']);
868 868
                 
869
-                if ($evalr == true)
870
-                {
871
-                    //$this->logging->log('rules OOK: '.print_r($rule),INFO );
872
-                    $action=$rule['action_match'];
873
-                    $this->logging->log('action OK : '.$action,INFO );
874
-                    if ($action >= 0)
875
-                    {
876
-                        if ($this->serviceCheckResult($host_name,$service_name,$action,$display) == false)
877
-                        {
878
-                            $this->trapAction.='Error sending status : check cmd/API';
879
-                        }
880
-                        else
881
-                        {
882
-                            $this->add_rule_match($rule['id'],$rule['num_match']+1);
883
-                            $this->trapAction.='Status '.$action.' to '.$host_name.'/'.$service_name;
884
-                        }
885
-                    }
886
-                    else
887
-                    {
888
-                        $this->add_rule_match($rule['id'],$rule['num_match']+1);
889
-                    }
890
-                    $this->trapToDb=($action==-2)?false:true;
891
-                }
892
-                else
893
-                {
894
-                    //$this->logging->log('rules KOO : '.print_r($rule),INFO );
869
+				if ($evalr == true)
870
+				{
871
+					//$this->logging->log('rules OOK: '.print_r($rule),INFO );
872
+					$action=$rule['action_match'];
873
+					$this->logging->log('action OK : '.$action,INFO );
874
+					if ($action >= 0)
875
+					{
876
+						if ($this->serviceCheckResult($host_name,$service_name,$action,$display) == false)
877
+						{
878
+							$this->trapAction.='Error sending status : check cmd/API';
879
+						}
880
+						else
881
+						{
882
+							$this->add_rule_match($rule['id'],$rule['num_match']+1);
883
+							$this->trapAction.='Status '.$action.' to '.$host_name.'/'.$service_name;
884
+						}
885
+					}
886
+					else
887
+					{
888
+						$this->add_rule_match($rule['id'],$rule['num_match']+1);
889
+					}
890
+					$this->trapToDb=($action==-2)?false:true;
891
+				}
892
+				else
893
+				{
894
+					//$this->logging->log('rules KOO : '.print_r($rule),INFO );
895 895
                     
896
-                    $action=$rule['action_nomatch'];
897
-                    $this->logging->log('action NOK : '.$action,INFO );
898
-                    if ($action >= 0)
899
-                    {
900
-                        if ($this->serviceCheckResult($host_name,$service_name,$action,$display)==false)
901
-                        {
902
-                            $this->trapAction.='Error sending status : check cmd/API';
903
-                        }
904
-                        else
905
-                        {
906
-                            $this->add_rule_match($rule['id'],$rule['num_match']+1);
907
-                            $this->trapAction.='Status '.$action.' to '.$host_name.'/'.$service_name;
908
-                        }
909
-                    }
910
-                    else
911
-                    {
912
-                        $this->add_rule_match($rule['id'],$rule['num_match']+1);
913
-                    }
914
-                    $this->trapToDb=($action==-2)?false:true;
915
-                }
916
-                // Put name in source_name
917
-                if (!isset($this->trapData['source_name']))
918
-                {
919
-                    $this->trapData['source_name']=$rule['host_name'];
920
-                }
921
-                else
922
-                {
923
-                    if (!preg_match('/'.$rule['host_name'].'/',$this->trapData['source_name']))
924
-                    { // only add if not present
925
-                        $this->trapData['source_name'].=','.$rule['host_name'];
926
-                    }
927
-                }
928
-            }
929
-            catch (Exception $e)
930
-            {
931
-                $this->logging->log('Error in rule eval : '.$e->getMessage(),WARN,'');
932
-                $this->trapAction.=' ERR : '.$e->getMessage();
933
-                $this->trapData['status']='error';
934
-            }
896
+					$action=$rule['action_nomatch'];
897
+					$this->logging->log('action NOK : '.$action,INFO );
898
+					if ($action >= 0)
899
+					{
900
+						if ($this->serviceCheckResult($host_name,$service_name,$action,$display)==false)
901
+						{
902
+							$this->trapAction.='Error sending status : check cmd/API';
903
+						}
904
+						else
905
+						{
906
+							$this->add_rule_match($rule['id'],$rule['num_match']+1);
907
+							$this->trapAction.='Status '.$action.' to '.$host_name.'/'.$service_name;
908
+						}
909
+					}
910
+					else
911
+					{
912
+						$this->add_rule_match($rule['id'],$rule['num_match']+1);
913
+					}
914
+					$this->trapToDb=($action==-2)?false:true;
915
+				}
916
+				// Put name in source_name
917
+				if (!isset($this->trapData['source_name']))
918
+				{
919
+					$this->trapData['source_name']=$rule['host_name'];
920
+				}
921
+				else
922
+				{
923
+					if (!preg_match('/'.$rule['host_name'].'/',$this->trapData['source_name']))
924
+					{ // only add if not present
925
+						$this->trapData['source_name'].=','.$rule['host_name'];
926
+					}
927
+				}
928
+			}
929
+			catch (Exception $e)
930
+			{
931
+				$this->logging->log('Error in rule eval : '.$e->getMessage(),WARN,'');
932
+				$this->trapAction.=' ERR : '.$e->getMessage();
933
+				$this->trapData['status']='error';
934
+			}
935 935
             
936
-        }
937
-        if ($this->trapData['status']=='error')
938
-        {
939
-            $this->trapToDb=true; // Always put errors in DB for the use can see
940
-        }
941
-        else
942
-        {
943
-            $this->trapData['status']='done';
944
-        }
945
-    }
946
-    
947
-    /** Add Time a action to rule
948
-     *	@param string $time : time to process to insert in SQL
949
-     */
950
-    public function add_rule_final($time)
951
-    {
952
-        $db_conn=$this->trapsDB->db_connect_trap();
953
-        if ($this->trapAction==null)
954
-        {
955
-            $this->trapAction='No action';
956
-        }
957
-        $sql="UPDATE ".$this->dbPrefix."received SET process_time = '".$time."' , status_detail='".$this->trapAction."'  WHERE (id = '".$this->trapId."');";
958
-        if ($db_conn->query($sql) === false) {
959
-            $this->logging->log('Error in update query : ' . $sql,WARN,'');
960
-        }
961
-    }
962
-    
963
-    /*********** UTILITIES *********************/
964
-    
965
-    /** reset service to OK after time defined in rule
966
-     *	TODO logic is : get all service in error + all rules, see if getting all rules then select services is better
967
-     *	@return integer : not in use
968
-     **/
969
-    public function reset_services()
970
-    {
971
-        // Get all services not in 'ok' state
972
-        $sql_query="SELECT s.service_object_id,
936
+		}
937
+		if ($this->trapData['status']=='error')
938
+		{
939
+			$this->trapToDb=true; // Always put errors in DB for the use can see
940
+		}
941
+		else
942
+		{
943
+			$this->trapData['status']='done';
944
+		}
945
+	}
946
+    
947
+	/** Add Time a action to rule
948
+	 *	@param string $time : time to process to insert in SQL
949
+	 */
950
+	public function add_rule_final($time)
951
+	{
952
+		$db_conn=$this->trapsDB->db_connect_trap();
953
+		if ($this->trapAction==null)
954
+		{
955
+			$this->trapAction='No action';
956
+		}
957
+		$sql="UPDATE ".$this->dbPrefix."received SET process_time = '".$time."' , status_detail='".$this->trapAction."'  WHERE (id = '".$this->trapId."');";
958
+		if ($db_conn->query($sql) === false) {
959
+			$this->logging->log('Error in update query : ' . $sql,WARN,'');
960
+		}
961
+	}
962
+    
963
+	/*********** UTILITIES *********************/
964
+    
965
+	/** reset service to OK after time defined in rule
966
+	 *	TODO logic is : get all service in error + all rules, see if getting all rules then select services is better
967
+	 *	@return integer : not in use
968
+	 **/
969
+	public function reset_services()
970
+	{
971
+		// Get all services not in 'ok' state
972
+		$sql_query="SELECT s.service_object_id,
973 973
 	 UNIX_TIMESTAMP(s.last_check) AS last_check,
974 974
 	s.current_state as state,
975 975
 	v.name1 as host_name,
@@ -977,43 +977,43 @@  discard block
 block discarded – undo
977 977
 	FROM icinga_servicestatus AS s
978 978
     LEFT JOIN icinga_objects as v ON s.service_object_id=v.object_id
979 979
     WHERE s.current_state != 0;";
980
-        $db_conn=$this->trapsDB->db_connect_ido();
981
-        if (($services_db=$db_conn->query($sql_query)) === false) { // set err to 1 to throw exception.
982
-            $this->logging->log('No result in query : ' . $sql_query,ERROR,'');
983
-            return 0;
984
-        }
985
-        $services=$services_db->fetchAll();
986
-        
987
-        // Get all rules
988
-        $sql_query="SELECT host_name, service_name, revert_ok FROM ".$this->dbPrefix."rules where revert_ok != 0;";
989
-        $db_conn2=$this->trapsDB->db_connect_trap();
990
-        if (($rules_db=$db_conn2->query($sql_query)) === false) {
991
-            $this->logging->log('No result in query : ' . $sql_query,ERROR,'');
992
-            return 0;
993
-        }
994
-        $rules=$rules_db->fetchAll();
995
-        
996
-        $now=date('U');
997
-        
998
-        $numreset=0;
999
-        foreach ($rules as $rule)
1000
-        {
1001
-            foreach ($services as $service)
1002
-            {
1003
-                if ($service['service_name'] == $rule['service_name'] &&
1004
-                    $service['host_name'] == $rule['host_name'] &&
1005
-                    ($service['last_check'] + $rule['revert_ok']) < $now)
1006
-                {
1007
-                    $this->serviceCheckResult($service['host_name'],$service['service_name'],0,'Reset service to OK after '.$rule['revert_ok'].' seconds');
1008
-                    $numreset++;
1009
-                }
1010
-            }
1011
-        }
1012
-        echo "\n";
1013
-        echo $numreset . " service(s) reset to OK\n";
1014
-        return 0;
1015
-        
1016
-    }
980
+		$db_conn=$this->trapsDB->db_connect_ido();
981
+		if (($services_db=$db_conn->query($sql_query)) === false) { // set err to 1 to throw exception.
982
+			$this->logging->log('No result in query : ' . $sql_query,ERROR,'');
983
+			return 0;
984
+		}
985
+		$services=$services_db->fetchAll();
986
+        
987
+		// Get all rules
988
+		$sql_query="SELECT host_name, service_name, revert_ok FROM ".$this->dbPrefix."rules where revert_ok != 0;";
989
+		$db_conn2=$this->trapsDB->db_connect_trap();
990
+		if (($rules_db=$db_conn2->query($sql_query)) === false) {
991
+			$this->logging->log('No result in query : ' . $sql_query,ERROR,'');
992
+			return 0;
993
+		}
994
+		$rules=$rules_db->fetchAll();
995
+        
996
+		$now=date('U');
997
+        
998
+		$numreset=0;
999
+		foreach ($rules as $rule)
1000
+		{
1001
+			foreach ($services as $service)
1002
+			{
1003
+				if ($service['service_name'] == $rule['service_name'] &&
1004
+					$service['host_name'] == $rule['host_name'] &&
1005
+					($service['last_check'] + $rule['revert_ok']) < $now)
1006
+				{
1007
+					$this->serviceCheckResult($service['host_name'],$service['service_name'],0,'Reset service to OK after '.$rule['revert_ok'].' seconds');
1008
+					$numreset++;
1009
+				}
1010
+			}
1011
+		}
1012
+		echo "\n";
1013
+		echo $numreset . " service(s) reset to OK\n";
1014
+		return 0;
1015
+        
1016
+	}
1017 1017
     
1018 1018
     
1019 1019
 }
1020 1020
\ No newline at end of file
Please login to merge, or discard this patch.
application/controllers/HandlerController.php 1 patch
Indentation   +240 added lines, -241 removed lines patch added patch discarded remove patch
@@ -11,13 +11,12 @@  discard block
 block discarded – undo
11 11
 
12 12
 //use Icinga\Web\Form as Form;
13 13
 /** Rules management
14
-
15
-*/
14
+ */
16 15
 class HandlerController extends TrapsController
17 16
 {
18 17
 
19 18
 	/** index : list existing rules 
20
-	*/
19
+	 */
21 20
 	public function indexAction()
22 21
 	{	
23 22
 		$this->checkReadPermission();
@@ -39,23 +38,23 @@  discard block
 block discarded – undo
39 38
 	 */
40 39
 	public function testruleAction()
41 40
 	{
42
-	    $this->checkReadPermission();
43
-	    $this->getTabs()->add('get',array(
44
-	        'active'	=> true,
45
-	        'label'		=> $this->translate('Test Rule'),
46
-	        'url'		=> Url::fromRequest()
47
-	    ));
41
+		$this->checkReadPermission();
42
+		$this->getTabs()->add('get',array(
43
+			'active'	=> true,
44
+			'label'		=> $this->translate('Test Rule'),
45
+			'url'		=> Url::fromRequest()
46
+		));
48 47
 	    
49
-	    //$db = $this->getDb();
48
+		//$db = $this->getDb();
50 49
 
51
-	    if ($this->params->get('rule') !== null) 
52
-	    {
53
-	        $this->view->rule= $this->params->get('rule');
54
-	    }
55
-	    else
56
-	    {
57
-	        $this->view->rule='';
58
-	    }
50
+		if ($this->params->get('rule') !== null) 
51
+		{
52
+			$this->view->rule= $this->params->get('rule');
53
+		}
54
+		else
55
+		{
56
+			$this->view->rule='';
57
+		}
59 58
 	}
60 59
 	
61 60
 	/**
@@ -63,31 +62,31 @@  discard block
 block discarded – undo
63 62
 	 */
64 63
 	private function add_setup_vars()
65 64
 	{
66
-	    // variables to send to view
67
-	    $this->view->hostlist=array(); // host list to input datalist
68
-	    $this->view->hostname=''; // Host name in input text
69
-	    $this->view->serviceGet=false; // Set to true to get list of service if only one host set
70
-	    $this->view->serviceSet=null; // Select service in services select (must have serviceGet=true).
71
-	    $this->view->mainoid=''; // Trap OID
72
-	    $this->view->mib=''; // Trap mib
73
-	    $this->view->name=''; // Trap name
74
-	    $this->view->trapListForMIB=array(); // Trap list if mib exists for trap
75
-	    $this->view->objectList=array(); // objects sent with trap
76
-	    $this->view->display=''; // Initial display
77
-	    $this->view->rule=''; // rule display
78
-	    $this->view->revertOK=''; // revert OK in seconds
79
-	    $this->view->hostid=-1; // normally set by javascript serviceGet()
80
-	    $this->view->ruleid=-1; // Rule id in DB for update & delete
81
-	    $this->view->setToUpdate=false; // set form as update form
82
-	    $this->view->setRuleMatch=-1; // set action on rule match (default nothing)
83
-	    $this->view->setRuleNoMatch=-1; // set action on rule no match (default nothing)
65
+		// variables to send to view
66
+		$this->view->hostlist=array(); // host list to input datalist
67
+		$this->view->hostname=''; // Host name in input text
68
+		$this->view->serviceGet=false; // Set to true to get list of service if only one host set
69
+		$this->view->serviceSet=null; // Select service in services select (must have serviceGet=true).
70
+		$this->view->mainoid=''; // Trap OID
71
+		$this->view->mib=''; // Trap mib
72
+		$this->view->name=''; // Trap name
73
+		$this->view->trapListForMIB=array(); // Trap list if mib exists for trap
74
+		$this->view->objectList=array(); // objects sent with trap
75
+		$this->view->display=''; // Initial display
76
+		$this->view->rule=''; // rule display
77
+		$this->view->revertOK=''; // revert OK in seconds
78
+		$this->view->hostid=-1; // normally set by javascript serviceGet()
79
+		$this->view->ruleid=-1; // Rule id in DB for update & delete
80
+		$this->view->setToUpdate=false; // set form as update form
81
+		$this->view->setRuleMatch=-1; // set action on rule match (default nothing)
82
+		$this->view->setRuleNoMatch=-1; // set action on rule no match (default nothing)
84 83
 	    
85
-	    $this->view->selectGroup=false; // Select by group if true
86
-	    $this->view->hostgroupid=-1; // host group id
87
-	    $this->view->serviceGroupGet=false; // Get list of service for group (set serviceSet to select one)
84
+		$this->view->selectGroup=false; // Select by group if true
85
+		$this->view->hostgroupid=-1; // host group id
86
+		$this->view->serviceGroupGet=false; // Get list of service for group (set serviceSet to select one)
88 87
 	    
89
-	    $this->view->modifier=null;
90
-	    $this->view->modified=null;
88
+		$this->view->modifier=null;
89
+		$this->view->modified=null;
91 90
 	}
92 91
 	
93 92
 	/**
@@ -96,102 +95,102 @@  discard block
 block discarded – undo
96 95
 	 */
97 96
 	private function add_from_existing($trapid)
98 97
 	{
99
-	    /********** Setup from existing trap ***************/
100
-	    // Get the full trap info
101
-	    $trapDetail=$this->getTrapDetail($trapid);
98
+		/********** Setup from existing trap ***************/
99
+		// Get the full trap info
100
+		$trapDetail=$this->getTrapDetail($trapid);
102 101
 	    
103
-	    $hostfilter=$trapDetail->source_ip;
102
+		$hostfilter=$trapDetail->source_ip;
104 103
 	    
105
-	    // Get host
106
-	    try
107
-	    {
108
-	        $hosts=$this->getHostByIP($hostfilter);
109
-	    }
110
-	    catch (Exception $e)
111
-	    {
112
-	        $this->displayExitError('Add handler : get host by IP/Name ',$e->getMessage());
113
-	    }
104
+		// Get host
105
+		try
106
+		{
107
+			$hosts=$this->getHostByIP($hostfilter);
108
+		}
109
+		catch (Exception $e)
110
+		{
111
+			$this->displayExitError('Add handler : get host by IP/Name ',$e->getMessage());
112
+		}
114 113
 	    
115 114
 	    
116
-	    // if one unique host found -> put id text input
117
-	    if (count($hosts)==1) {
118
-	        $this->view->hostname=$hosts[0]->name;
119
-	        //$hostid=$hosts[0]->id;
120
-	        // Tell JS to get services when page is loaded
121
-	        $this->view->serviceGet=true;
115
+		// if one unique host found -> put id text input
116
+		if (count($hosts)==1) {
117
+			$this->view->hostname=$hosts[0]->name;
118
+			//$hostid=$hosts[0]->id;
119
+			// Tell JS to get services when page is loaded
120
+			$this->view->serviceGet=true;
122 121
 	        
123
-	    }
124
-	    else
125
-	    {
126
-	        foreach($hosts as $key=>$val)
127
-	        {
128
-	            array_push($this->view->hostlist,$hosts[$key]->name);
129
-	        }
130
-	    }
122
+		}
123
+		else
124
+		{
125
+			foreach($hosts as $key=>$val)
126
+			{
127
+				array_push($this->view->hostlist,$hosts[$key]->name);
128
+			}
129
+		}
131 130
 	    
132
-	    // set up trap oid and objects received by the trap
131
+		// set up trap oid and objects received by the trap
133 132
 	    
134
-	    $this->view->mainoid=$trapDetail->trap_oid;
135
-	    if ($trapDetail->trap_name_mib != null)
136
-	    {
137
-	        $this->view->mib=$trapDetail->trap_name_mib;
138
-	        $this->view->name=$trapDetail->trap_name;
139
-	        $this->view->trapListForMIB=$this->getMIB()
140
-	        ->getTrapList($trapDetail->trap_name_mib);
141
-	    }
133
+		$this->view->mainoid=$trapDetail->trap_oid;
134
+		if ($trapDetail->trap_name_mib != null)
135
+		{
136
+			$this->view->mib=$trapDetail->trap_name_mib;
137
+			$this->view->name=$trapDetail->trap_name;
138
+			$this->view->trapListForMIB=$this->getMIB()
139
+			->getTrapList($trapDetail->trap_name_mib);
140
+		}
142 141
 	    
143
-	    // Get all objects that can be in trap from MIB
144
-	    $allObjects=$this->getMIB()->getObjectList($trapDetail->trap_oid);
145
-	    // Get all objects in current Trap
146
-	    $currentTrapObjects=$this->getTrapobjects($trapid);
147
-	    $oid_index=1;
148
-	    foreach ($currentTrapObjects as $key => $val)
149
-	    {
150
-	        $currentObjectType='Unknown';
151
-	        $currentObjectTypeEnum='Unknown';
152
-	        if (isset($allObjects[$val->oid]['type']))
153
-	        {
154
-	            $currentObjectType=$allObjects[$val->oid]['type'];
155
-	            $currentObjectTypeEnum=$allObjects[$val->oid]['type_enum'];
156
-	        }
157
-	        $currentObject=array(
158
-	            $oid_index,
159
-	            $val->oid,
160
-	            $val->oid_name_mib,
161
-	            $val->oid_name,
162
-	            $val->value,
163
-	            $currentObjectType,
164
-	            $currentObjectTypeEnum
165
-	        );
166
-	        $oid_index++;
167
-	        array_push($this->view->objectList,$currentObject);
168
-	        // set currrent object to null in allObjects
169
-	        if (isset($allObjects[$val->oid]))
170
-	        {
171
-	            $allObjects[$val->oid]=null;
172
-	        }
173
-	    }
174
-	    if ($allObjects!=null) // in case trap doesn't have objects or is not resolved
175
-	    {
176
-	        foreach ($allObjects as $key => $val)
177
-	        {
178
-	            if ($val==null) { continue; }
179
-	            array_push($this->view->objectList, array(
180
-	                $oid_index,
181
-	                $key,
182
-	                $allObjects[$key]['mib'],
183
-	                $allObjects[$key]['name'],
184
-	                '',
185
-	                $allObjects[$key]['type'],
186
-	                $allObjects[$key]['type_enum']
187
-	            ));
188
-	            $oid_index++;
189
-	        }
190
-	    }
142
+		// Get all objects that can be in trap from MIB
143
+		$allObjects=$this->getMIB()->getObjectList($trapDetail->trap_oid);
144
+		// Get all objects in current Trap
145
+		$currentTrapObjects=$this->getTrapobjects($trapid);
146
+		$oid_index=1;
147
+		foreach ($currentTrapObjects as $key => $val)
148
+		{
149
+			$currentObjectType='Unknown';
150
+			$currentObjectTypeEnum='Unknown';
151
+			if (isset($allObjects[$val->oid]['type']))
152
+			{
153
+				$currentObjectType=$allObjects[$val->oid]['type'];
154
+				$currentObjectTypeEnum=$allObjects[$val->oid]['type_enum'];
155
+			}
156
+			$currentObject=array(
157
+				$oid_index,
158
+				$val->oid,
159
+				$val->oid_name_mib,
160
+				$val->oid_name,
161
+				$val->value,
162
+				$currentObjectType,
163
+				$currentObjectTypeEnum
164
+			);
165
+			$oid_index++;
166
+			array_push($this->view->objectList,$currentObject);
167
+			// set currrent object to null in allObjects
168
+			if (isset($allObjects[$val->oid]))
169
+			{
170
+				$allObjects[$val->oid]=null;
171
+			}
172
+		}
173
+		if ($allObjects!=null) // in case trap doesn't have objects or is not resolved
174
+		{
175
+			foreach ($allObjects as $key => $val)
176
+			{
177
+				if ($val==null) { continue; }
178
+				array_push($this->view->objectList, array(
179
+					$oid_index,
180
+					$key,
181
+					$allObjects[$key]['mib'],
182
+					$allObjects[$key]['name'],
183
+					'',
184
+					$allObjects[$key]['type'],
185
+					$allObjects[$key]['type_enum']
186
+				));
187
+				$oid_index++;
188
+			}
189
+		}
191 190
 	    
192
-	    // Add a simple display
193
-	    $this->view->display='Trap '.$trapDetail->trap_name.' received';
194
-	    $this->view->create_basic_rule=true;
191
+		// Add a simple display
192
+		$this->view->display='Trap '.$trapDetail->trap_name.' received';
193
+		$this->view->create_basic_rule=true;
195 194
 	}
196 195
 
197 196
 	/**
@@ -200,29 +199,29 @@  discard block
 block discarded – undo
200 199
 	 */
201 200
 	private function add_check_host_exists($ruleDetail)
202 201
 	{
203
-	    // Check if hostname still exists
204
-	    $host_get=$this->getHostByName($this->view->hostname);
202
+		// Check if hostname still exists
203
+		$host_get=$this->getHostByName($this->view->hostname);
205 204
 	    
206
-	    if (count($host_get)==0)
207
-	    {
208
-	        $this->view->warning_message='Host '.$this->view->hostname. ' doesn\'t exists anymore';
209
-	        $this->view->serviceGet=false;
210
-	    }
211
-	    else
212
-	    {
213
-	        // Tell JS to get services when page is loaded
214
-	        $this->view->serviceGet=true;
215
-	        // get service id for form to set :
216
-	        $serviceID=$this->getServiceIDByName($this->view->hostname,$ruleDetail->service_name);
217
-	        if (count($serviceID) ==0)
218
-	        {
219
-	            $this->view->warning_message=' Service '.$ruleDetail->service_name. ' doesn\'t exists anymore';
220
-	        }
221
-	        else
222
-	        {
223
-	            $this->view->serviceSet=$serviceID[0]->id;
224
-	        }
225
-	    }
205
+		if (count($host_get)==0)
206
+		{
207
+			$this->view->warning_message='Host '.$this->view->hostname. ' doesn\'t exists anymore';
208
+			$this->view->serviceGet=false;
209
+		}
210
+		else
211
+		{
212
+			// Tell JS to get services when page is loaded
213
+			$this->view->serviceGet=true;
214
+			// get service id for form to set :
215
+			$serviceID=$this->getServiceIDByName($this->view->hostname,$ruleDetail->service_name);
216
+			if (count($serviceID) ==0)
217
+			{
218
+				$this->view->warning_message=' Service '.$ruleDetail->service_name. ' doesn\'t exists anymore';
219
+			}
220
+			else
221
+			{
222
+				$this->view->serviceSet=$serviceID[0]->id;
223
+			}
224
+		}
226 225
 	}
227 226
 
228 227
 	/**
@@ -231,33 +230,33 @@  discard block
 block discarded – undo
231 230
 	 */
232 231
 	private function add_check_hostgroup_exists($ruleDetail)
233 232
 	{
234
-	    // Check if groupe exists
235
-	    $group_get=$this->getHostGroupByName($this->view->hostgroupname);
236
-	    if (count($group_get)==0)
237
-	    {
238
-	        $this->view->warning_message='HostGroup '.$this->view->hostgroupname. ' doesn\'t exists anymore';
239
-	        $this->view->serviceGroupGet=false;
240
-	    }
241
-	    else
242
-	    {
243
-	        $grpServices=$this->getServicesByHostGroupid($group_get[0]->id);
244
-	        $foundGrpService=0;
245
-	        foreach ($grpServices as $grpService)
246
-	        {
247
-	            if ($grpService[0] == $ruleDetail->service_name)
248
-	            {
249
-	                $foundGrpService=1;
250
-	                $this->view->serviceSet=$ruleDetail->service_name;
251
-	            }
252
-	        }
233
+		// Check if groupe exists
234
+		$group_get=$this->getHostGroupByName($this->view->hostgroupname);
235
+		if (count($group_get)==0)
236
+		{
237
+			$this->view->warning_message='HostGroup '.$this->view->hostgroupname. ' doesn\'t exists anymore';
238
+			$this->view->serviceGroupGet=false;
239
+		}
240
+		else
241
+		{
242
+			$grpServices=$this->getServicesByHostGroupid($group_get[0]->id);
243
+			$foundGrpService=0;
244
+			foreach ($grpServices as $grpService)
245
+			{
246
+				if ($grpService[0] == $ruleDetail->service_name)
247
+				{
248
+					$foundGrpService=1;
249
+					$this->view->serviceSet=$ruleDetail->service_name;
250
+				}
251
+			}
253 252
 	        
254
-	        // Tell JS to get services when page is loaded
255
-	        $this->view->serviceGroupGet=true;
256
-	        if ($foundGrpService==0)
257
-	        {
258
-	            $this->view->warning_message.=' Service '.$ruleDetail->service_name. ' doesn\'t exists anymore';
259
-	        }
260
-	    }
253
+			// Tell JS to get services when page is loaded
254
+			$this->view->serviceGroupGet=true;
255
+			if ($foundGrpService==0)
256
+			{
257
+				$this->view->warning_message.=' Service '.$ruleDetail->service_name. ' doesn\'t exists anymore';
258
+			}
259
+		}
261 260
 	}
262 261
 	
263 262
 	/**
@@ -269,52 +268,52 @@  discard block
 block discarded – undo
269 268
 	 */
270 269
 	private function add_create_trap_object_list(&$display, &$rule)
271 270
 	{
272
-	    $curObjectList=array();
273
-	    $index=1;
274
-	    // check in display & rule for : OID(<oid>)
275
-	    $matches=array();
276
-	    while ( preg_match('/_OID\(([\.0-9\*]+)\)/',$display,$matches) ||
277
-	        preg_match('/_OID\(([\.0-9\*]+)\)/',$rule,$matches))
278
-	    {
279
-	        $curOid=$matches[1];
271
+		$curObjectList=array();
272
+		$index=1;
273
+		// check in display & rule for : OID(<oid>)
274
+		$matches=array();
275
+		while ( preg_match('/_OID\(([\.0-9\*]+)\)/',$display,$matches) ||
276
+			preg_match('/_OID\(([\.0-9\*]+)\)/',$rule,$matches))
277
+		{
278
+			$curOid=$matches[1];
280 279
 	        
281
-	        if ( (preg_match('/\*/',$curOid) == 0 ) 
282
-	            && ($object=$this->getMIB()->translateOID($curOid)) != null)
283
-	        {
284
-	            array_push($curObjectList, array(
285
-	                $index,
286
-	                $curOid,
287
-	                $object['mib'],
288
-	                $object['name'],
289
-	                '',
290
-	                $object['type'],
291
-	                $object['type_enum']
292
-	            ));
293
-	        }
294
-	        else
295
-	        {
296
-	            array_push($curObjectList, array(
297
-	                $index,
298
-	                $curOid,
299
-	                'not found',
300
-	                'not found',
301
-	                '',
302
-	                'not found',
303
-	                'not found'
304
-	            ));
305
-	        }
306
-	        $curOid = preg_replace('/\*/','\*',$curOid);
307
-	        $display=preg_replace('/_OID\('.$curOid.'\)/','\$'.$index.'\$',$display);
308
-	        $rule=preg_replace('/_OID\('.$curOid.'\)/','\$'.$index.'\$',$rule);
309
-	        $index++;
310
-	    }
311
-	    return $curObjectList;
280
+			if ( (preg_match('/\*/',$curOid) == 0 ) 
281
+				&& ($object=$this->getMIB()->translateOID($curOid)) != null)
282
+			{
283
+				array_push($curObjectList, array(
284
+					$index,
285
+					$curOid,
286
+					$object['mib'],
287
+					$object['name'],
288
+					'',
289
+					$object['type'],
290
+					$object['type_enum']
291
+				));
292
+			}
293
+			else
294
+			{
295
+				array_push($curObjectList, array(
296
+					$index,
297
+					$curOid,
298
+					'not found',
299
+					'not found',
300
+					'',
301
+					'not found',
302
+					'not found'
303
+				));
304
+			}
305
+			$curOid = preg_replace('/\*/','\*',$curOid);
306
+			$display=preg_replace('/_OID\('.$curOid.'\)/','\$'.$index.'\$',$display);
307
+			$rule=preg_replace('/_OID\('.$curOid.'\)/','\$'.$index.'\$',$rule);
308
+			$index++;
309
+		}
310
+		return $curObjectList;
312 311
 	}
313 312
 	
314 313
 	/** Add a handler  
315
-	*	Get params fromid : setup from existing trap (id of trap table)
316
-	*	Get param ruleid : edit from existing handler (id of rule table)
317
-	*/
314
+	 *	Get params fromid : setup from existing trap (id of trap table)
315
+	 *	Get param ruleid : edit from existing handler (id of rule table)
316
+	 */
318 317
 	public function addAction()
319 318
 	{
320 319
 		$this->checkConfigPermission();
@@ -333,8 +332,8 @@  discard block
 block discarded – undo
333 332
 		//$this->view->trapvalues=false; // Set to true to display 'value' colum in objects
334 333
 		
335 334
 		if (($trapid = $this->params->get('fromid')) !== null) {
336
-		    /********** Setup from existing trap ***************/
337
-            $this->add_from_existing($trapid);
335
+			/********** Setup from existing trap ***************/
336
+			$this->add_from_existing($trapid);
338 337
 			return;
339 338
 		}
340 339
 		
@@ -359,14 +358,14 @@  discard block
 block discarded – undo
359 358
 			$this->view->warning_message='';
360 359
 			if ($this->view->hostname != null)
361 360
 			{
362
-			    $this->view->selectGroup=false;
363
-			    // Check if hostname still exists
364
-			    $this->add_check_host_exists($ruleDetail);
361
+				$this->view->selectGroup=false;
362
+				// Check if hostname still exists
363
+				$this->add_check_host_exists($ruleDetail);
365 364
 			}
366 365
 			else
367 366
 			{
368
-			    $this->view->selectGroup=true;
369
-			    $this->add_check_hostgroup_exists($ruleDetail); //  Check if groupe exists				
367
+				$this->view->selectGroup=true;
368
+				$this->add_check_hostgroup_exists($ruleDetail); //  Check if groupe exists				
370 369
 			}
371 370
 			
372 371
 			$this->view->mainoid=$ruleDetail->trap_oid;
@@ -394,9 +393,9 @@  discard block
 block discarded – undo
394 393
 	}
395 394
 	
396 395
 	/** Validate form and output message to user  
397
-	*	@param in postdata 
398
-	* 	@return string status : OK / <Message>
399
-	**/
396
+	 *	@param in postdata 
397
+	 * 	@return string status : OK / <Message>
398
+	 **/
400 399
 	protected function handlerformAction()
401 400
 	{
402 401
 		$postData=$this->getRequest()->getPost();
@@ -437,7 +436,7 @@  discard block
 block discarded – undo
437 436
 			//$this->Module()->
438 437
 			$this->_helper->json(array(
439 438
 				'status'=>'OK',
440
-			    'redirect'=>'trapdirector/handler'
439
+				'redirect'=>'trapdirector/handler'
441 440
 			      
442 441
 			));
443 442
 		}		
@@ -521,9 +520,9 @@  discard block
 block discarded – undo
521 520
 	}
522 521
 
523 522
 	/** Get trap detail by trapid. 
524
-	*	@param integer $trapid : id of trap in received table
525
-	*	@return array (objects)
526
-	*/
523
+	 *	@param integer $trapid : id of trap in received table
524
+	 *	@return array (objects)
525
+	 */
527 526
 	protected function getTrapDetail($trapid) 
528 527
 	{
529 528
 		if (!preg_match('/^[0-9]+$/',$trapid)) { throw new Exception('Invalid id');  }
@@ -544,8 +543,8 @@  discard block
 block discarded – undo
544 543
 			$trapDetail=$db->fetchRow($query);
545 544
 			if ( $trapDetail == null ) 
546 545
 			{
547
-			    $trapDetail = 'NULL';
548
-			    throw new Exception('No traps was found with id = '.$trapid);
546
+				$trapDetail = 'NULL';
547
+				throw new Exception('No traps was found with id = '.$trapid);
549 548
 			}
550 549
 		}
551 550
 		catch (Exception $e)
@@ -559,9 +558,9 @@  discard block
 block discarded – undo
559 558
 	}
560 559
 
561 560
 	/** Get trap objects
562
-	*	@param integer $trapid : trap id
563
-	* 	@return array : full column in db of trap id
564
-	*/
561
+	 *	@param integer $trapid : trap id
562
+	 * 	@return array : full column in db of trap id
563
+	 */
565 564
 	protected function getTrapobjects($trapid)
566 565
 	{	
567 566
 		if (!preg_match('/^[0-9]+$/',$trapid)) { throw new Exception('Invalid id');  }
@@ -592,9 +591,9 @@  discard block
 block discarded – undo
592 591
 	}
593 592
 
594 593
 	/** Get rule detail by ruleid.
595
-	*	@param integer $ruleid int id of rule in rule table
596
-	*	@return object : column objects in db
597
-	*/
594
+	 *	@param integer $ruleid int id of rule in rule table
595
+	 *	@return object : column objects in db
596
+	 */
598 597
 	protected function getRuleDetail($ruleid) 
599 598
 	{
600 599
 		if (!preg_match('/^[0-9]+$/',$ruleid)) { throw new Exception('Invalid id');  }
@@ -621,7 +620,7 @@  discard block
 block discarded – undo
621 620
 	}
622 621
 
623 622
 	/** Setup tabs for rules 
624
-	*/
623
+	 */
625 624
 	protected function prepareTabs()
626 625
 	{
627 626
 		return $this->getTabs()->add('status', array(
Please login to merge, or discard this patch.