patrickpr / trapdirector

application/clicommands/StatusCommand.php

Indentation +4 added lines, -4 removed lines

@@ -17,10 +17,10 @@
 class StatusCommand extends Command
 {
 	/**
-	* Get database counts
-	*
-	* Get number of traps, rules
-	*/
+	 * Get database counts
+	 *
+	 * Get number of traps, rules
+	 */
 	public function dbAction()
 	{
 		$db_prefix=$this->Config()->get('config', 'database_prefix');

application/clicommands/TrapsCommand.php

Indentation +17 added lines, -17 removed lines

@@ -17,16 +17,16 @@ 
 class TrapsCommand extends Command
 {
 	/**
-	*	Delete old traps
-	*
-	*	USAGE 
-	*
-	*	icingali trapdirector traps rotate [options]
-	*	
-	*	OPTIONS
-	*	
-	*	--days	remove traps older than <n> days
-	*/
+	 *	Delete old traps
+	 *
+	 *	USAGE 
+	 *
+	 *	icingali trapdirector traps rotate [options]
+	 *	
+	 *	OPTIONS
+	 *	
+	 *	--days	remove traps older than <n> days
+	 */
 	public function rotateAction()
 	{
 		$module=Icinga::app()->getModuleManager()->getModule($this->getModuleName());
@@ -56,13 +56,13 @@ 
 	}  	
 	
 	/**
-	*	Reset services to OK state if timeout is passed
-	*
-	*	USAGE
-	*
-	*	icingali trapdirector traps reset_services
-	*
-	*/
+	 *	Reset services to OK state if timeout is passed
+	 *
+	 *	USAGE
+	 *
+	 *	icingali trapdirector traps reset_services
+	 *
+	 */
 	public function resetOKAction()
 	{
 		$module=Icinga::app()->getModuleManager()->getModule($this->getModuleName());

application/forms/TrapsConfigForm.php

Indentation +82 added lines, -82 removed lines

@@ -32,13 +32,13 @@ 
 	
 	public function setDefaultIDODB($idoDbName)
 	{
-	    $this->idoDbName=$idoDbName;
-	    return $this;
+		$this->idoDbName=$idoDbName;
+		return $this;
 	}
 	
-    public function createElements(array $formData)
-    {
-        $this->addElement(
+	public function createElements(array $formData)
+	{
+		$this->addElement(
 			'select',
 			'config_database',
 			array(
@@ -72,14 +72,14 @@ 
 			 )
 		);		
 		$this->addElement(
-            'text',
-            'config_icingaweb2_etc',
-            array(
-                    'required'      => true,
-                    'label'             => $this->translate('icingaweb2 config dir'),
+			'text',
+			'config_icingaweb2_etc',
+			array(
+					'required'      => true,
+					'label'             => $this->translate('icingaweb2 config dir'),
 					'value'			=> $this->icinga_base_path,
-             )
-        );
+			 )
+		);
 		$retval=0;
 		$output=array();
 		$snmptranslate=exec('which snmptranslate',$output,$retval);
@@ -88,68 +88,68 @@ 
 			$snmptranslate='/usr/bin/snmptranslate';
 		}
 		$this->addElement(
-            'text',
-            'config_snmptranslate',
-            array(
-                    'required'      => true,
-                    'label'             => $this->translate('snmptranslate binary with path'),
+			'text',
+			'config_snmptranslate',
+			array(
+					'required'      => true,
+					'label'             => $this->translate('snmptranslate binary with path'),
 					'value'			=> $snmptranslate,
-             )
-        );
+			 )
+		);
 		$this->addElement(
-            'text',
-            'config_snmptranslate_dirs',
-            array(
-                    'required'      => true,
-                    'label'         => $this->translate('Path for mibs'),
+			'text',
+			'config_snmptranslate_dirs',
+			array(
+					'required'      => true,
+					'label'         => $this->translate('Path for mibs'),
 					'value'			=> $this->module_base_path.'/mibs',
-             )
-        );		
+			 )
+		);		
 		$this->addElement(
-            'text',
-            'config_icingacmd',
-            array(
-                    'required'      => true,
-                    'label'         => $this->translate('icingacmd with path'),
+			'text',
+			'config_icingacmd',
+			array(
+					'required'      => true,
+					'label'         => $this->translate('icingacmd with path'),
 					'value'			=> '/var/run/icinga2/cmd/icinga2.cmd',
-             )
-        );
+			 )
+		);
 		$this->addElement(
-		    'text',
-		    'config_icingaAPI_host',
-		    array(
-		        'required'      => false,
-		        'label'         => $this->translate('icinga2 API Host IP'),
-		        'value'			=> '',
-		    )
+			'text',
+			'config_icingaAPI_host',
+			array(
+				'required'      => false,
+				'label'         => $this->translate('icinga2 API Host IP'),
+				'value'			=> '',
+			)
 		);
 		$this->addElement(
-		    'text',
-		    'config_icingaAPI_port',
-		    array(
-		        'required'      => false,
-		        'label'         => $this->translate('icinga2 API TCP Port'),
-		        'value'			=> '5665',
-		    )
-		    );
+			'text',
+			'config_icingaAPI_port',
+			array(
+				'required'      => false,
+				'label'         => $this->translate('icinga2 API TCP Port'),
+				'value'			=> '5665',
+			)
+			);
 		$this->addElement(
-		    'text',
-		    'config_icingaAPI_user',
-		    array(
-		        'required'      => false,
-		        'label'         => $this->translate('icinga2 API username'),
-		        'value'			=> '',
-		    )
-        );
+			'text',
+			'config_icingaAPI_user',
+			array(
+				'required'      => false,
+				'label'         => $this->translate('icinga2 API username'),
+				'value'			=> '',
+			)
+		);
 		$this->addElement(
-		    'text',
-		    'config_icingaAPI_password',
-		    array(
-		        'required'      => false,
-		        'label'         => $this->translate('icinga2 API password'),
-		        'value'			=> '',
-		    )
-	    );
+			'text',
+			'config_icingaAPI_password',
+			array(
+				'required'      => false,
+				'label'         => $this->translate('icinga2 API password'),
+				'value'			=> '',
+			)
+		);
 		/*$this->addElement(
 		    'radio',
 		    'config_icinga2api_use',
@@ -159,7 +159,7 @@ 
 		        'value'			=> array('Yes', 'No'),//array('1' => 'Yes', '0' => 'No'),
 		    )
 		);*/  
-		    // For Icinga2 API connexion
+			// For Icinga2 API connexion
 	//	$icinga2API=new Fieldset('Icinga2 API connection');
 	/* 	$icinga2API_select=new radioBtn('icinga2API_use');
 		$icinga2API_select->setLabel('Use icinga2 API instead of cmd file?');
@@ -169,30 +169,30 @@ 
 		]);
 		$icinga2API->add($icinga2API_select);
 		$this->add($icinga2API); */
-    }	
+	}	
 
     
-    public function render(Zend_View_Interface $view = NULL)
-    {
+	public function render(Zend_View_Interface $view = NULL)
+	{
         
 
-        //return $html;
-        return parent::render($view);
-    }
+		//return $html;
+		return parent::render($view);
+	}
     
-    /**
-     * {@inheritdoc}
-     */
-    public function onSuccess()
-    {
-        parent::onSuccess();
-    }
+	/**
+	 * {@inheritdoc}
+	 */
+	public function onSuccess()
+	{
+		parent::onSuccess();
+	}
     
-    /**
-     * {@inheritdoc}
-     */
+	/**
+	 * {@inheritdoc}
+	 */
 	public function save()
-    {
+	{
 		parent::save();
 	}
 }

library/Trapdirector/Tables/TrapTableHostList.php

Indentation +81 added lines, -81 removed lines

@@ -21,13 +21,13 @@ 
 	protected $lastHost;
 	// Filters 
 	
-    protected $filter;
-    protected $enforcedFilters = array();
-    protected $searchColumns = array();
+	protected $filter;
+	protected $enforcedFilters = array();
+	protected $searchColumns = array();
 	
 	protected function getTitles() {
 		// TODO : check moduleconfig is set
-	    return $this->moduleConfig->getTrapHostListTitles();
+		return $this->moduleConfig->getTrapHostListTitles();
 	}
 		
 	// ******************  Render table in html 
@@ -35,45 +35,45 @@ 
 	// Host grouping
 	protected function renderHostIfNew($IP,$hostname)
 	{
-	    $view = $this->getView();
+		$view = $this->getView();
 	    
-	    if ($this->lastHost === $IP) {
-	        return;
-	    }
+		if ($this->lastHost === $IP) {
+			return;
+		}
 	    
-	    if ($this->lastHost === null) 
-	    {
-	        $htm = "<thead>\n  <tr>\n";
-	    } else {
-	        $htm = "</tbody>\n<thead>\n  <tr>\n";
-	    }
+		if ($this->lastHost === null) 
+		{
+			$htm = "<thead>\n  <tr>\n";
+		} else {
+			$htm = "</tbody>\n<thead>\n  <tr>\n";
+		}
 	    
-	    if ($this->columnCount === null) 
-	    {
-	        $this->columnCount = count($this->getTitles());
-	    }
+		if ($this->columnCount === null) 
+		{
+			$this->columnCount = count($this->getTitles());
+		}
 	    
-	    $htm .= '<th colspan="' . $this->columnCount . '">' . $view->escape($IP);
-	    if ($hostname != null)
-	    {
-	        $htm .= ' ('.$hostname.')';
-	    }
-	    $htm .= '</th>' . "\n";
-	    if ($this->lastHost === null) {
-	        $htm .= "  </tr>\n";
-	    } else {
-	        $htm .= "  </tr>\n</thead>\n";
-	    }
+		$htm .= '<th colspan="' . $this->columnCount . '">' . $view->escape($IP);
+		if ($hostname != null)
+		{
+			$htm .= ' ('.$hostname.')';
+		}
+		$htm .= '</th>' . "\n";
+		if ($this->lastHost === null) {
+			$htm .= "  </tr>\n";
+		} else {
+			$htm .= "  </tr>\n</thead>\n";
+		}
 	    
-	    $this->lastHost = $IP;
+		$this->lastHost = $IP;
 	    
-	    return $htm . "<tbody>\n";
+		return $htm . "<tbody>\n";
 	}		
 	
-    public function __toString()
-    {
-        return $this->render();
-    }
+	public function __toString()
+	{
+		return $this->render();
+	}
 	
 	public function render()
 	{
@@ -139,9 +139,9 @@ 
 
 	}
 
-    public function count()
-    {  
-        $db=$this->db();
+	public function count()
+	{  
+		$db=$this->db();
 		
 		$query = $this->getBaseQuery();
 		$this->applyFiltersToQuery($query);
@@ -149,16 +149,16 @@ 
 		
 		return count($values);
 		
-        //return $db->fetchOne($query);
-    }
+		//return $db->fetchOne($query);
+	}
 	
-    public function getPaginator()
-    {
-        $paginator = new Paginator();
-        $paginator->setQuery($this);
+	public function getPaginator()
+	{
+		$paginator = new Paginator();
+		$paginator->setQuery($this);
 
-        return $paginator;
-    }
+		return $paginator;
+	}
 	
 	// ****************** DB connection and query
 	
@@ -168,24 +168,24 @@ 
 		
 		$query = $this->getBaseQuery();
 		$this->applyFiltersToQuery($query);
-       if ($this->hasLimit() || $this->hasOffset()) {
-            $query->limit($this->getLimit(), $this->getOffset());
-        }		
+	   if ($this->hasLimit() || $this->hasOffset()) {
+			$query->limit($this->getLimit(), $this->getOffset());
+		}		
 		
 		return $db->fetchAll($query);
 	}
 	 
-    public function getBaseQuery()
-    {
+	public function getBaseQuery()
+	{
 		$db=$this->db();
 		$query = $db->select()->from(
-            $this->moduleConfig->getTrapTableName(),
-		    $this->moduleConfig->getTrapHostListDisplayColumns()
-		    )->group(array('t.source_ip','t.source_name','t.trap_oid'))
-		    ->order('t.source_ip');
+			$this->moduleConfig->getTrapTableName(),
+			$this->moduleConfig->getTrapHostListDisplayColumns()
+			)->group(array('t.source_ip','t.source_name','t.trap_oid'))
+			->order('t.source_ip');
 
-        return $query;
-    }	 
+		return $query;
+	}	 
 	
 	// ****************** Filters
 
@@ -215,39 +215,39 @@ 
 		$this->filter_done=(isset($filter['done']))?$this->filter_done=$filter['done']:0;
 	}
 	
-    protected function getSearchColumns()
-    {
-        return $this->getColumns();
-    }
+	protected function getSearchColumns()
+	{
+		return $this->getColumns();
+	}
 	
 	public function getColumns()
 	{
 		return $this->moduleConfig->getTrapListDisplayColumns();
 	}
 
-    public function setFilter($filter)
-    {
-        $this->filter = $filter;
-        return $this;
-    }
+	public function setFilter($filter)
+	{
+		$this->filter = $filter;
+		return $this;
+	}
 	
 	public function getFilterEditor(Request $request)
-    {
-        $filterEditor = Widget::create('filterEditor')
-            ->setColumns(array_keys($this->getColumns()))
-            ->setSearchColumns(array_keys($this->getSearchColumns()))
-            ->preserveParams('limit', 'sort', 'dir', 'view', 'backend')
-            ->ignoreParams('page')
-            ->handleRequest($request);
+	{
+		$filterEditor = Widget::create('filterEditor')
+			->setColumns(array_keys($this->getColumns()))
+			->setSearchColumns(array_keys($this->getSearchColumns()))
+			->preserveParams('limit', 'sort', 'dir', 'view', 'backend')
+			->ignoreParams('page')
+			->handleRequest($request);
 
-        $filter = $filterEditor->getFilter();
-        $this->setFilter($filter);
+		$filter = $filterEditor->getFilter();
+		$this->setFilter($filter);
 
-        return $filterEditor;
-    }
+		return $filterEditor;
+	}
 	
-    protected function applyFiltersToQuery($query)
-    {
+	protected function applyFiltersToQuery($query)
+	{
 		
 		$sql='';
 		if ($this->filter_query != '')
@@ -268,8 +268,8 @@ 
 			$sql.="(status != 'done')";
 		}
 		if ($sql != '') $query->where($sql);		
-        return $query;
-    }	
+		return $query;
+	}	
 
 }
 

tests/db_test.php

Indentation +29 added lines, -29 removed lines

@@ -14,40 +14,40 @@
 
 if (!array_key_exists('v',$options) || !array_key_exists('c',$options) || !array_key_exists('b',$options)|| !array_key_exists('a',$options))
 {
-    printf("Need version -v, path -a, database -b (mysql,pgsql) command -c (create/update)\n");
-    exit(1);
+	printf("Need version -v, path -a, database -b (mysql,pgsql) command -c (create/update)\n");
+	exit(1);
 }
 $command=$options['c'];
 $path=$options['a'];
 try {
-    switch($command)
-    {
-        case 'create':
-            $schema=($options['b']=='mysql')?'schema_v'.$options['v'].'.sql':'schema_v'.$options['v'].'.pgsql';
-            $schema=$path.'/SQL/'.$schema;
-            $trap->trapsDB->create_schema($schema, 'traps_');
-            break;
-        case 'update':
-            $message=$trap->trapsDB->update_schema($path."/SQL/",$options['v'], 'traps_',true);
-            printf("Update message : %s\n",$message);
-            if ($message == 'ERROR')
-            {
-                exit(1);
-            }
-            printf("Messages DONE, updating : \n");
-            $message=$trap->trapsDB->update_schema($path."/SQL/",$options['v'], 'traps_');
-            if ($message == 'ERROR')
-            {
-                exit(1);
-            }
-            break;
-        default:
-            printf("Unknown command\n");
-            exit(1);
-    }
+	switch($command)
+	{
+		case 'create':
+			$schema=($options['b']=='mysql')?'schema_v'.$options['v'].'.sql':'schema_v'.$options['v'].'.pgsql';
+			$schema=$path.'/SQL/'.$schema;
+			$trap->trapsDB->create_schema($schema, 'traps_');
+			break;
+		case 'update':
+			$message=$trap->trapsDB->update_schema($path."/SQL/",$options['v'], 'traps_',true);
+			printf("Update message : %s\n",$message);
+			if ($message == 'ERROR')
+			{
+				exit(1);
+			}
+			printf("Messages DONE, updating : \n");
+			$message=$trap->trapsDB->update_schema($path."/SQL/",$options['v'], 'traps_');
+			if ($message == 'ERROR')
+			{
+				exit(1);
+			}
+			break;
+		default:
+			printf("Unknown command\n");
+			exit(1);
+	}
 } catch (Exception $e) {
-    printf("Caught Exception %s\n",$e->getMessage());
-    exit (1);
+	printf("Caught Exception %s\n",$e->getMessage());
+	exit (1);
 }
 
 exit(0);

library/Trapdirector/Tables/HandlerTableList.php

Indentation +59 added lines, -59 removed lines

@@ -31,9 +31,9 @@ 
 	
 	// Filters 
 	
-    protected $filter;
-    protected $enforcedFilters = array();
-    protected $searchColumns = array();
+	protected $filter;
+	protected $enforcedFilters = array();
+	protected $searchColumns = array();
 	
 	protected function getTitles() {
 		return $this->moduleConfig->getHandlerListTitles();
@@ -45,10 +45,10 @@ 
 		$this->doTranslate=true;
 	}
 	// ******************  Render table in html  
-    public function __toString()
-    {
-        return $this->render();
-    }
+	public function __toString()
+	{
+		return $this->render();
+	}
 	
 	public function render()
 	{
@@ -153,30 +153,30 @@ 
 
 	}
 
-    public function count()
-    {
-        //$db = $this->connection()->getConnection();
-        $query = clone($this->getBaseQuery());
-        $query->reset('order')->columns(array('COUNT(*)'));
-        $this->applyFiltersToQuery($query);
+	public function count()
+	{
+		//$db = $this->connection()->getConnection();
+		$query = clone($this->getBaseQuery());
+		$query->reset('order')->columns(array('COUNT(*)'));
+		$this->applyFiltersToQuery($query);
 
 		$db=$this->db();
 		
 		$query = $db->select()->from(
-            $this->moduleConfig->getTrapRuleName(),
-            array('COUNT(*)')
-        );
+			$this->moduleConfig->getTrapRuleName(),
+			array('COUNT(*)')
+		);
 		
-        return $db->fetchOne($query);
-    }
+		return $db->fetchOne($query);
+	}
 	
-    public function getPaginator()
-    {
-        $paginator = new Paginator();
-        $paginator->setQuery($this);
+	public function getPaginator()
+	{
+		$paginator = new Paginator();
+		$paginator->setQuery($this);
 
-        return $paginator;
-    }
+		return $paginator;
+	}
 	
 	// ****************** DB connection and query
 	
@@ -186,63 +186,63 @@ 
 		
 		$query = $this->getBaseQuery();
 		
-       if ($this->hasLimit() || $this->hasOffset()) {
-            $query->limit($this->getLimit(), $this->getOffset());
-        }		
+	   if ($this->hasLimit() || $this->hasOffset()) {
+			$query->limit($this->getLimit(), $this->getOffset());
+		}		
 		
 		return $db->fetchAll($query);
 	}
 	 
-    public function getBaseQuery()
-    {
+	public function getBaseQuery()
+	{
 		$db=$this->db();
 		
 		$query = $db->select()->from(
-            $this->moduleConfig->getTrapRuleName(),
-            $this->moduleConfig->getHandlerListDisplayColumns()
-        )->order('host_name DESC,trap_oid DESC');
+			$this->moduleConfig->getTrapRuleName(),
+			$this->moduleConfig->getHandlerListDisplayColumns()
+		)->order('host_name DESC,trap_oid DESC');
 
-        return $query;
-    }	 
+		return $query;
+	}	 
 	
 	// ****************** Filters
 
-    protected function getSearchColumns()
-    {
-        return $this->getColumns();
-    }
+	protected function getSearchColumns()
+	{
+		return $this->getColumns();
+	}
 	
 	public function getColumns()
 	{
 		return $this->moduleConfig->getHandlerListDisplayColumns();
 	}
 
-    public function setFilter($filter)
-    {
-        $this->filter = $filter;
-        return $this;
-    }
+	public function setFilter($filter)
+	{
+		$this->filter = $filter;
+		return $this;
+	}
 	
 	public function getFilterEditor(Request $request)
-    {
-        $filterEditor = Widget::create('filterEditor')
-            ->setColumns(array_keys($this->getColumns()))
-            ->setSearchColumns(array_keys($this->getSearchColumns()))
-            ->preserveParams('limit', 'sort', 'dir', 'view', 'backend')
-            ->ignoreParams('page')
-            ->handleRequest($request);
+	{
+		$filterEditor = Widget::create('filterEditor')
+			->setColumns(array_keys($this->getColumns()))
+			->setSearchColumns(array_keys($this->getSearchColumns()))
+			->preserveParams('limit', 'sort', 'dir', 'view', 'backend')
+			->ignoreParams('page')
+			->handleRequest($request);
 
-        $filter = $filterEditor->getFilter();
-        $this->setFilter($filter);
+		$filter = $filterEditor->getFilter();
+		$this->setFilter($filter);
 
-        return $filterEditor;
-    }
+		return $filterEditor;
+	}
 	
-    protected function applyFiltersToQuery($query)
-    {
+	protected function applyFiltersToQuery($query)
+	{
 		// TODO : implement
 		
-        /*$filter = null;
+		/*$filter = null;
         $enforced = $this->enforcedFilters;
         if ($this->filter && ! $this->filter->isEmpty()) {
             $filter = $this->filter;
@@ -257,7 +257,7 @@ 
         }
 		*/
 		
-        return $query;
-    }	
+		return $query;
+	}	
 
 }

library/Trapdirector/TrapsProcess/Logging.php

Indentation +102 added lines, -102 removed lines

@@ -9,113 +9,113 @@
 class Logging
 {
 
-    //**** Options from config database
-    // Default values
-    public $debugLevel=2;  // 0=No output 1=critical 2=warning 3=trace 4=ALL
-    public $outputMode='syslog'; // alert type : file, syslog, display
-    public $outputFile="/tmp/trapdebug.txt";
-    protected $logLevels=array("","Error","Warning","Info","Debug");
-    protected $outputList=array('file', 'syslog', 'display');
+	//**** Options from config database
+	// Default values
+	public $debugLevel=2;  // 0=No output 1=critical 2=warning 3=trace 4=ALL
+	public $outputMode='syslog'; // alert type : file, syslog, display
+	public $outputFile="/tmp/trapdebug.txt";
+	protected $logLevels=array("","Error","Warning","Info","Debug");
+	protected $outputList=array('file', 'syslog', 'display');
     
-    /** Send log. Throws exception on critical error
-     *	@param	string $message Message to log
-     *	@param	int $level 1=critical 2=warning 3=trace 4=debug
-     *	@param  string $destination file/syslog/display
-     *	@return void
-     *  @throws Exception
-     **/
-    public function log( $message, $level, $destination ='')
-    {
-        if ($this->debugLevel >= $level)
-        {
-            $date = '['.  date("Y/m/d H:i:s") . '] '; // no date in syslog as already there
-            $message = '[TrapDirector] ['.$this->logLevels[$level].']: ' .$message . "\n";
+	/** Send log. Throws exception on critical error
+	 *	@param	string $message Message to log
+	 *	@param	int $level 1=critical 2=warning 3=trace 4=debug
+	 *	@param  string $destination file/syslog/display
+	 *	@return void
+	 *  @throws Exception
+	 **/
+	public function log( $message, $level, $destination ='')
+	{
+		if ($this->debugLevel >= $level)
+		{
+			$date = '['.  date("Y/m/d H:i:s") . '] '; // no date in syslog as already there
+			$message = '[TrapDirector] ['.$this->logLevels[$level].']: ' .$message . "\n";
             
-            $output = ( $destination != '' ) ? $destination : $this->outputMode;
-            switch ($output)
-            {
-                case 'file':
-                    file_put_contents ($this->outputFile, $date.$message , FILE_APPEND);
-                    break;
-                case 'syslog':
-                    switch($level)
-                    {
-                        case 1 : $prio = LOG_ERR;break;
-                        case 2 : $prio = LOG_WARNING;break;
-                        case 3 : $prio = LOG_INFO;break;
-                        case 4 : $prio = LOG_INFO;break; // LOG_DEBUG isn't always displayed in syslog
-                        default: $prio = LOG_ERR;
-                    }
-                    syslog($prio,$message);
-                    break;
-                case 'display':
-                    echo $date.$message;
-                    break;
-                default : // nothing we can do at this point
-                    throw new Exception($date.$message);
-            }
-        }
-        if ($level == 1)
-        {
-            throw new Exception($message);
-        }
-    }
+			$output = ( $destination != '' ) ? $destination : $this->outputMode;
+			switch ($output)
+			{
+				case 'file':
+					file_put_contents ($this->outputFile, $date.$message , FILE_APPEND);
+					break;
+				case 'syslog':
+					switch($level)
+					{
+						case 1 : $prio = LOG_ERR;break;
+						case 2 : $prio = LOG_WARNING;break;
+						case 3 : $prio = LOG_INFO;break;
+						case 4 : $prio = LOG_INFO;break; // LOG_DEBUG isn't always displayed in syslog
+						default: $prio = LOG_ERR;
+					}
+					syslog($prio,$message);
+					break;
+				case 'display':
+					echo $date.$message;
+					break;
+				default : // nothing we can do at this point
+					throw new Exception($date.$message);
+			}
+		}
+		if ($level == 1)
+		{
+			throw new Exception($message);
+		}
+	}
     
         
-    public function setLogging($debugLvl,$outputType,$outputFile=null)
-    {
-        $this->setLevel($debugLvl);
-        switch ($outputType)
-        {
-            case 'file':
-                if ($outputFile == null) throw new Exception("File logging without file !");
-                $this->setFile($outputFile);
-                $this->setDestination('file');
-                break;
-            default:
-                $this->setDestination($outputType);
-        }
-    }
+	public function setLogging($debugLvl,$outputType,$outputFile=null)
+	{
+		$this->setLevel($debugLvl);
+		switch ($outputType)
+		{
+			case 'file':
+				if ($outputFile == null) throw new Exception("File logging without file !");
+				$this->setFile($outputFile);
+				$this->setDestination('file');
+				break;
+			default:
+				$this->setDestination($outputType);
+		}
+	}
     
-    /**
-     * Set logging level
-     * @param integer $level
-     * @throws Exception
-     */
-    public function setLevel($level)
-    {
-        if (!is_integer($level) || $level < 0 || $level > 10)
-        {
-            throw new Exception('Invalid log level');
-        }
-        $this->debugLevel=$level;
-    }
+	/**
+	 * Set logging level
+	 * @param integer $level
+	 * @throws Exception
+	 */
+	public function setLevel($level)
+	{
+		if (!is_integer($level) || $level < 0 || $level > 10)
+		{
+			throw new Exception('Invalid log level');
+		}
+		$this->debugLevel=$level;
+	}
 
-    /**
-     * Set logging destination
-     * @param string $destination
-     * @throws Exception
-     */
-    public function setDestination($destination)
-    {
-        if (!is_string($destination) || ! in_array($destination, $this->outputList))
-        {
-            throw new Exception('Invalid log destination');
-        }
-        $this->outputMode=$destination;
-    }
-    /**
-     * Set file destination
-     * @param string $file
-     * @throws Exception
-     */
-    public function setFile($file)
-    {
-        if (!is_string($file))
-        {
-            throw new Exception('Invalid log file');
-        }
-        $this->outputFile=$file;
-    }
+	/**
+	 * Set logging destination
+	 * @param string $destination
+	 * @throws Exception
+	 */
+	public function setDestination($destination)
+	{
+		if (!is_string($destination) || ! in_array($destination, $this->outputList))
+		{
+			throw new Exception('Invalid log destination');
+		}
+		$this->outputMode=$destination;
+	}
+	/**
+	 * Set file destination
+	 * @param string $file
+	 * @throws Exception
+	 */
+	public function setFile($file)
+	{
+		if (!is_string($file))
+		{
+			throw new Exception('Invalid log file');
+		}
+		$this->outputFile=$file;
+	}
     
 }
\ No newline at end of file

library/Trapdirector/TrapsProcess/Database.php

Indentation +347 added lines, -347 removed lines

@@ -9,383 +9,383 @@
 class Database
 {
 
-    // Databases
-    /** @var \PDO $trapDB trap database */
-    protected $trapDB=null; 
-    protected $idoDB=null; //< ido database
-    public $trapDBType; //< Type of database for traps (mysql, pgsql)
-    public $idoDBType; //< Type of database for ido (mysql, pgsql)
+	// Databases
+	/** @var \PDO $trapDB trap database */
+	protected $trapDB=null; 
+	protected $idoDB=null; //< ido database
+	public $trapDBType; //< Type of database for traps (mysql, pgsql)
+	public $idoDBType; //< Type of database for ido (mysql, pgsql)
     
-    protected $trapDSN; //< trap database connection params
-    protected $trapUsername; //< trap database connection params
-    protected $trapPass; //< trap database connection params
-    public $dbPrefix; //< database tables prefix
+	protected $trapDSN; //< trap database connection params
+	protected $trapUsername; //< trap database connection params
+	protected $trapPass; //< trap database connection params
+	public $dbPrefix; //< database tables prefix
     
-    protected $idoSet; //< bool true is ido database set
-    protected $idoDSN; //< trap database connection params
-    protected $idoUsername; //< trap database connection params
-    protected $idoPass; //< trap database connection params
+	protected $idoSet; //< bool true is ido database set
+	protected $idoDSN; //< trap database connection params
+	protected $idoUsername; //< trap database connection params
+	protected $idoPass; //< trap database connection params
     
-    // Logging function
+	// Logging function
     
-    protected $logging; //< logging class
+	protected $logging; //< logging class
     
-    /**
-     * @param Logging $logClass : where to log
-     * @param array $dbParam : array of named params  type,host,dbname,username,[port],[password]
-     */
-    function __construct($logClass,$dbParam,$dbPrefix)
-    {
-        $this->logging=$logClass;
-        $this->dbPrefix=$dbPrefix;
+	/**
+	 * @param Logging $logClass : where to log
+	 * @param array $dbParam : array of named params  type,host,dbname,username,[port],[password]
+	 */
+	function __construct($logClass,$dbParam,$dbPrefix)
+	{
+		$this->logging=$logClass;
+		$this->dbPrefix=$dbPrefix;
         
-        $this->trapDSN=$this->setupDSN($dbParam);
-        $this->trapUsername = $dbParam['username'];
-        $this->trapPass = (array_key_exists('password', $dbParam)) ? $dbParam['password']:'';
-        $this->trapDBType=$dbParam['db'];
-        $this->logging->log('DSN : '.$this->trapDSN. ';user '.$this->trapUsername.' / prefix : '. $this->dbPrefix,INFO);
-        $this->db_connect_trap();
+		$this->trapDSN=$this->setupDSN($dbParam);
+		$this->trapUsername = $dbParam['username'];
+		$this->trapPass = (array_key_exists('password', $dbParam)) ? $dbParam['password']:'';
+		$this->trapDBType=$dbParam['db'];
+		$this->logging->log('DSN : '.$this->trapDSN. ';user '.$this->trapUsername.' / prefix : '. $this->dbPrefix,INFO);
+		$this->db_connect_trap();
         
-    }
+	}
     
-    /**
-     * Setup and connect to IDO database
-     * @param array $dbParam : array of named params
-     */
-    public function setupIDO($dbParam)
-    {
-        $this->idoDSN=$this->setupDSN($dbParam);
-        $this->idoUsername = $dbParam['username'];
-        $this->idoPass = (array_key_exists('password', $dbParam)) ? $dbParam['password']:'';
-        $this->logging->log('DSN : '.$this->idoDSN. ';user '.$this->idoUsername,INFO);
-        $this->idoDBType=$dbParam['db'];
-        $this->db_connect_ido();
-    }
+	/**
+	 * Setup and connect to IDO database
+	 * @param array $dbParam : array of named params
+	 */
+	public function setupIDO($dbParam)
+	{
+		$this->idoDSN=$this->setupDSN($dbParam);
+		$this->idoUsername = $dbParam['username'];
+		$this->idoPass = (array_key_exists('password', $dbParam)) ? $dbParam['password']:'';
+		$this->logging->log('DSN : '.$this->idoDSN. ';user '.$this->idoUsername,INFO);
+		$this->idoDBType=$dbParam['db'];
+		$this->db_connect_ido();
+	}
     
-    /**
-     * Connect to IDO database
-     * @return \PDO
-     */
-    public function db_connect_ido()
-    {
-        if ($this->idoDB != null) {
-            // Check if connection is still alive
-            try {
-                $this->idoDB->query('select 1')->fetchColumn();
-                return $this->idoDB;
-            } catch (Exception $e) {
-                // select 1 failed, try to reconnect.
-                $this->logging->log('Database IDO connection lost, reconnecting',WARN);
-            }
-        }
-        try {
-            $this->idoDB = new PDO($this->idoDSN,$this->idoUsername,$this->idoPass);
-        } catch (PDOException $e) {
-            $this->logging->log('Connection failed to IDO : ' . $e->getMessage(),ERROR,'');
-        }
-        return $this->idoDB;
-    }
+	/**
+	 * Connect to IDO database
+	 * @return \PDO
+	 */
+	public function db_connect_ido()
+	{
+		if ($this->idoDB != null) {
+			// Check if connection is still alive
+			try {
+				$this->idoDB->query('select 1')->fetchColumn();
+				return $this->idoDB;
+			} catch (Exception $e) {
+				// select 1 failed, try to reconnect.
+				$this->logging->log('Database IDO connection lost, reconnecting',WARN);
+			}
+		}
+		try {
+			$this->idoDB = new PDO($this->idoDSN,$this->idoUsername,$this->idoPass);
+		} catch (PDOException $e) {
+			$this->logging->log('Connection failed to IDO : ' . $e->getMessage(),ERROR,'');
+		}
+		return $this->idoDB;
+	}
     
-    /**
-     * Connect to Trap database
-     * @return \PDO
-     */
-    public function db_connect_trap()
-    {
-        if ($this->trapDB != null) {
-            // Check if connection is still alive
-            try {
-                $this->trapDB->query('select 1')->fetchColumn();
-                return $this->trapDB;
-            } catch (Exception $e) {
-                // select 1 failed, try to reconnect.
-                $this->logging->log('Database connection lost, reconnecting',WARN);
-            }           
-        }       
-        try {
-            $this->trapDB = new PDO($this->trapDSN,$this->trapUsername,$this->trapPass);
-        } catch (PDOException $e) {
-            $this->logging->log('Connection failed : ' . $e->getMessage(),ERROR,'');
-        }
-        return $this->trapDB;
-    }
+	/**
+	 * Connect to Trap database
+	 * @return \PDO
+	 */
+	public function db_connect_trap()
+	{
+		if ($this->trapDB != null) {
+			// Check if connection is still alive
+			try {
+				$this->trapDB->query('select 1')->fetchColumn();
+				return $this->trapDB;
+			} catch (Exception $e) {
+				// select 1 failed, try to reconnect.
+				$this->logging->log('Database connection lost, reconnecting',WARN);
+			}           
+		}       
+		try {
+			$this->trapDB = new PDO($this->trapDSN,$this->trapUsername,$this->trapPass);
+		} catch (PDOException $e) {
+			$this->logging->log('Connection failed : ' . $e->getMessage(),ERROR,'');
+		}
+		return $this->trapDB;
+	}
     
-    /**
-     * Setup dsn and check parameters
-     * @param array $configElmt
-     * @return string
-     */
-    protected function setupDSN($configElmt)  
-    {
-        if (!array_key_exists('db',$configElmt) ||
-            !array_key_exists('host',$configElmt) ||
-            !array_key_exists('dbname',$configElmt) ||
-            !array_key_exists('username',$configElmt))
-        {
-            $this->logging->log('Missing DB params',ERROR);
-            return ''; 
-        }
+	/**
+	 * Setup dsn and check parameters
+	 * @param array $configElmt
+	 * @return string
+	 */
+	protected function setupDSN($configElmt)  
+	{
+		if (!array_key_exists('db',$configElmt) ||
+			!array_key_exists('host',$configElmt) ||
+			!array_key_exists('dbname',$configElmt) ||
+			!array_key_exists('username',$configElmt))
+		{
+			$this->logging->log('Missing DB params',ERROR);
+			return ''; 
+		}
         
-        //	$dsn = 'mysql:dbname=traps;host=127.0.0.1';
-        $dsn= $configElmt['db'].':dbname='.$configElmt['dbname'].';host='.$configElmt['host'];
+		//	$dsn = 'mysql:dbname=traps;host=127.0.0.1';
+		$dsn= $configElmt['db'].':dbname='.$configElmt['dbname'].';host='.$configElmt['host'];
         
-        if (array_key_exists('port', $configElmt))
-        {
-            $dsn .= ';port='.$configElmt['port'];
-        }
-        return $dsn;
-    }
+		if (array_key_exists('port', $configElmt))
+		{
+			$dsn .= ';port='.$configElmt['port'];
+		}
+		return $dsn;
+	}
 
-    /** Set name=element in database config table
-     * @param string $name
-     * @param string $element
-     * @return boolean true on success, else false (error logged)
-     */
-    public function setDBConfig($name,$element)
-    {
-        $db_conn=$this->db_connect_trap();
-        $sql='SELECT id from '.$this->dbPrefix.'db_config WHERE ( name=\''.$name.'\' )';
-        if (($ret_code=$db_conn->query($sql)) === false) {
-            $this->logging->log('Error setting config element : ' . $sql,WARN,'');           
-            return false;
-        }
-        $value=$ret_code->fetch();
-        if ($value != null && isset($value['id']))
-        {   // Entry exists -> update
-            $sql='UPDATE '.$this->dbPrefix.'db_config SET value = \''.$element.'\' WHERE (id = '.$value['id'].')';
-        }
-        else
-        {   // Entry does no exists -> create
-            $sql='INSERT INTO '.$this->dbPrefix.'db_config (name,value) VALUES (\''.$name.'\' , \''.$element.'\' )';
-        }
-        if (($ret_code=$db_conn->query($sql)) === false) {
-            $this->logging->log('Error setting config element : ' . $sql,WARN,'');
-            return false;
-        }
-        $this->logging->log('Setting config '.$name.' = '.$element.' in database',INFO);
-        return true;
-    }
+	/** Set name=element in database config table
+	 * @param string $name
+	 * @param string $element
+	 * @return boolean true on success, else false (error logged)
+	 */
+	public function setDBConfig($name,$element)
+	{
+		$db_conn=$this->db_connect_trap();
+		$sql='SELECT id from '.$this->dbPrefix.'db_config WHERE ( name=\''.$name.'\' )';
+		if (($ret_code=$db_conn->query($sql)) === false) {
+			$this->logging->log('Error setting config element : ' . $sql,WARN,'');           
+			return false;
+		}
+		$value=$ret_code->fetch();
+		if ($value != null && isset($value['id']))
+		{   // Entry exists -> update
+			$sql='UPDATE '.$this->dbPrefix.'db_config SET value = \''.$element.'\' WHERE (id = '.$value['id'].')';
+		}
+		else
+		{   // Entry does no exists -> create
+			$sql='INSERT INTO '.$this->dbPrefix.'db_config (name,value) VALUES (\''.$name.'\' , \''.$element.'\' )';
+		}
+		if (($ret_code=$db_conn->query($sql)) === false) {
+			$this->logging->log('Error setting config element : ' . $sql,WARN,'');
+			return false;
+		}
+		$this->logging->log('Setting config '.$name.' = '.$element.' in database',INFO);
+		return true;
+	}
 
-    /**
-     *   Get data from db_config
-     *	@param $element string name of param
-     *	@return mixed : value (or null)
-     */
-    public function getDBConfig($element)
-    {
-        $db_conn=$this->db_connect_trap();
-        $sql='SELECT value from '.$this->dbPrefix.'db_config WHERE ( name=\''.$element.'\' )';
-        if (($ret_code=$db_conn->query($sql)) === false) {
-            $this->logging->log('No result in query : ' . $sql,WARN,'');
-            return null;
-        }
-        $value=$ret_code->fetch();
-        if ($value != null && isset($value['value']))
-        {
-            return $value['value'];
-        }
-        return null;
-    }
+	/**
+	 *   Get data from db_config
+	 *	@param $element string name of param
+	 *	@return mixed : value (or null)
+	 */
+	public function getDBConfig($element)
+	{
+		$db_conn=$this->db_connect_trap();
+		$sql='SELECT value from '.$this->dbPrefix.'db_config WHERE ( name=\''.$element.'\' )';
+		if (($ret_code=$db_conn->query($sql)) === false) {
+			$this->logging->log('No result in query : ' . $sql,WARN,'');
+			return null;
+		}
+		$value=$ret_code->fetch();
+		if ($value != null && isset($value['value']))
+		{
+			return $value['value'];
+		}
+		return null;
+	}
     
     
-    //*********    Schema Management *********************/
+	//*********    Schema Management *********************/
 
-    /** Create database schema
-     *	@param $schema_file	string File to read schema from
-     *	@param $table_prefix string to replace #PREFIX# in schema file by this
-     */
-    public function create_schema($schema_file,$table_prefix)
-    {
-        //Read data from snmptrapd from stdin
-        $input_stream=fopen($schema_file, 'r');
+	/** Create database schema
+	 *	@param $schema_file	string File to read schema from
+	 *	@param $table_prefix string to replace #PREFIX# in schema file by this
+	 */
+	public function create_schema($schema_file,$table_prefix)
+	{
+		//Read data from snmptrapd from stdin
+		$input_stream=fopen($schema_file, 'r');
         
-        if ($input_stream=== false)
-        {
-            $this->logging->log("Error reading schema !",ERROR,'');
-            return;
-        }
-        $newline='';
-        $cur_table='';
-        $cur_table_array=array();
-        $db_conn=$this->db_connect_trap();
+		if ($input_stream=== false)
+		{
+			$this->logging->log("Error reading schema !",ERROR,'');
+			return;
+		}
+		$newline='';
+		$cur_table='';
+		$cur_table_array=array();
+		$db_conn=$this->db_connect_trap();
         
-        while (($line=fgets($input_stream)) !== false)
-        {
-            $newline.=chop(preg_replace('/#PREFIX#/',$table_prefix,$line));
-            if (preg_match('/; *$/', $newline))
-            {
-                $sql= $newline;
-                if ($db_conn->query($sql) === false) {
-                    $this->logging->log('Error create schema : '.$sql,ERROR,'');
-                    return;
-                }
-                if (preg_match('/^ *CREATE TABLE ([^ ]+)/',$newline,$cur_table_array))
-                {
-                    $cur_table='table '.$cur_table_array[1];
-                }
-                else
-                {
-                    $cur_table='secret SQL stuff :-)';
-                }
-                $this->logging->log('Creating : ' . $cur_table,INFO );
-                $newline='';
-            }
-        }
+		while (($line=fgets($input_stream)) !== false)
+		{
+			$newline.=chop(preg_replace('/#PREFIX#/',$table_prefix,$line));
+			if (preg_match('/; *$/', $newline))
+			{
+				$sql= $newline;
+				if ($db_conn->query($sql) === false) {
+					$this->logging->log('Error create schema : '.$sql,ERROR,'');
+					return;
+				}
+				if (preg_match('/^ *CREATE TABLE ([^ ]+)/',$newline,$cur_table_array))
+				{
+					$cur_table='table '.$cur_table_array[1];
+				}
+				else
+				{
+					$cur_table='secret SQL stuff :-)';
+				}
+				$this->logging->log('Creating : ' . $cur_table,INFO );
+				$newline='';
+			}
+		}
         
-        $sql= $newline;
-        if ($sql != '' )
-        {
-            if ($db_conn->query($sql) === false) {
-                $this->logging->log('Error create schema : '.$sql,ERROR,'');
-                return;
-            }
-        }
-        $this->logging->log('Schema created',INFO);
-    }
+		$sql= $newline;
+		if ($sql != '' )
+		{
+			if ($db_conn->query($sql) === false) {
+				$this->logging->log('Error create schema : '.$sql,ERROR,'');
+				return;
+			}
+		}
+		$this->logging->log('Schema created',INFO);
+	}
     
-    /**
-     * Update database schema from current (as set in db) to $target_version
-     *     @param $prefix string file prefix of sql update File
-     *     @param $target_version int target db version number
-     *     @param $table_prefix string to replace #PREFIX# in schema file by this
-     *     @param bool $getmsg : only get messages from version upgrades
-     *     @return string : if $getmsg=true, return messages or 'ERROR' on error.
-     */
-    public function update_schema($prefix,$target_version,$table_prefix,$getmsg=false)
-    {
-        // Get current db number
-        $db_conn=$this->db_connect_trap();
-        $sql='SELECT value from '.$this->dbPrefix.'db_config WHERE name=\'db_version\' ';
-        $this->logging->log('SQL query : '.$sql,DEBUG );
-        if (($ret_code=$db_conn->query($sql)) === false) {
-            $this->logging->log('Cannot get db version. Query : ' . $sql,2,'');
-            return 'ERROR';
-        }
-        $version=$ret_code->fetchAll();
-        $cur_version=$version[0]['value'];
+	/**
+	 * Update database schema from current (as set in db) to $target_version
+	 *     @param $prefix string file prefix of sql update File
+	 *     @param $target_version int target db version number
+	 *     @param $table_prefix string to replace #PREFIX# in schema file by this
+	 *     @param bool $getmsg : only get messages from version upgrades
+	 *     @return string : if $getmsg=true, return messages or 'ERROR' on error.
+	 */
+	public function update_schema($prefix,$target_version,$table_prefix,$getmsg=false)
+	{
+		// Get current db number
+		$db_conn=$this->db_connect_trap();
+		$sql='SELECT value from '.$this->dbPrefix.'db_config WHERE name=\'db_version\' ';
+		$this->logging->log('SQL query : '.$sql,DEBUG );
+		if (($ret_code=$db_conn->query($sql)) === false) {
+			$this->logging->log('Cannot get db version. Query : ' . $sql,2,'');
+			return 'ERROR';
+		}
+		$version=$ret_code->fetchAll();
+		$cur_version=$version[0]['value'];
         
-        if ($this->trapDBType == 'pgsql')
-        {
-            $prefix .= 'update_pgsql/schema_';
-        }
-        else
-        {
-            $prefix .= 'update_sql/schema_';
-        }
-        //echo "version all :\n";print_r($version);echo " \n $cur_ver \n";
-        if ($getmsg === true)
-        {
-            return $this->update_schema_message($prefix, $cur_version, $target_version);
-        }
+		if ($this->trapDBType == 'pgsql')
+		{
+			$prefix .= 'update_pgsql/schema_';
+		}
+		else
+		{
+			$prefix .= 'update_sql/schema_';
+		}
+		//echo "version all :\n";print_r($version);echo " \n $cur_ver \n";
+		if ($getmsg === true)
+		{
+			return $this->update_schema_message($prefix, $cur_version, $target_version);
+		}
         
-        if ($this->update_schema_do($prefix, $cur_version, $target_version, $table_prefix) === true)
-        {
-            return 'ERROR';
-        }
-        return '';
+		if ($this->update_schema_do($prefix, $cur_version, $target_version, $table_prefix) === true)
+		{
+			return 'ERROR';
+		}
+		return '';
 
-    }
+	}
 
-    /**
-     * Update database schema from current (as set in db) to $target_version
-     *     @param string $prefix  file prefix of sql update File
-     *     @param int $cur_version  current db version number
-     *     @param int $target_version  target db version number
-     *     @param string $table_prefix   to replace #PREFIX# in schema file by this
-     *     @return bool : true on error
-     */
-    public function update_schema_do($prefix,$cur_version,$target_version,$table_prefix)
-    {
-        while($cur_version<$target_version)
-        { // TODO : execute pre & post scripts
-            $cur_version++;
-            $this->logging->log('Updating to version : ' .$cur_version ,INFO );
-            $updateFile=$prefix.'v'.($cur_version-1).'_v'.$cur_version.'.sql';
-            $input_stream=fopen($updateFile, 'r');
-            if ($input_stream=== false)
-            {
-                $this->logging->log("Error reading update file ". $updateFile,ERROR);
-                return true;
-            }
-            $newline='';
-            $db_conn=$this->db_connect_trap();
-            $db_conn->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
-            while (($line=fgets($input_stream)) !== false)
-            {
-                if (preg_match('/^#/', $line)) continue; // ignore comment lines
-                $newline.=chop(preg_replace('/#PREFIX#/',$table_prefix,$line));
-                if (preg_match('/; *$/', $newline))
-                {
-                    $sql_req=$db_conn->prepare($newline);
-                    if ($sql_req->execute() === false) {
-                        $this->logging->log('Error create schema : '.$newline,ERROR);
-                        return true;
-                    }
-                    $cur_table_array=array();
-                    if (preg_match('/^ *([^ ]+) TABLE ([^ ]+)/',$newline,$cur_table_array))
-                    {
-                        $cur_table=$cur_table_array[1] . ' SQL table '.$cur_table_array[2];
-                    }
-                    else
-                    {
-                        $cur_table='secret SQL stuff :-)';
-                        //$cur_table=$newline;
-                    }
-                    $this->logging->log('Doing : ' . $cur_table,INFO );
+	/**
+	 * Update database schema from current (as set in db) to $target_version
+	 *     @param string $prefix  file prefix of sql update File
+	 *     @param int $cur_version  current db version number
+	 *     @param int $target_version  target db version number
+	 *     @param string $table_prefix   to replace #PREFIX# in schema file by this
+	 *     @return bool : true on error
+	 */
+	public function update_schema_do($prefix,$cur_version,$target_version,$table_prefix)
+	{
+		while($cur_version<$target_version)
+		{ // TODO : execute pre & post scripts
+			$cur_version++;
+			$this->logging->log('Updating to version : ' .$cur_version ,INFO );
+			$updateFile=$prefix.'v'.($cur_version-1).'_v'.$cur_version.'.sql';
+			$input_stream=fopen($updateFile, 'r');
+			if ($input_stream=== false)
+			{
+				$this->logging->log("Error reading update file ". $updateFile,ERROR);
+				return true;
+			}
+			$newline='';
+			$db_conn=$this->db_connect_trap();
+			$db_conn->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
+			while (($line=fgets($input_stream)) !== false)
+			{
+				if (preg_match('/^#/', $line)) continue; // ignore comment lines
+				$newline.=chop(preg_replace('/#PREFIX#/',$table_prefix,$line));
+				if (preg_match('/; *$/', $newline))
+				{
+					$sql_req=$db_conn->prepare($newline);
+					if ($sql_req->execute() === false) {
+						$this->logging->log('Error create schema : '.$newline,ERROR);
+						return true;
+					}
+					$cur_table_array=array();
+					if (preg_match('/^ *([^ ]+) TABLE ([^ ]+)/',$newline,$cur_table_array))
+					{
+						$cur_table=$cur_table_array[1] . ' SQL table '.$cur_table_array[2];
+					}
+					else
+					{
+						$cur_table='secret SQL stuff :-)';
+						//$cur_table=$newline;
+					}
+					$this->logging->log('Doing : ' . $cur_table,INFO );
                     
-                    $newline='';
-                }
-            }
-            fclose($input_stream);
+					$newline='';
+				}
+			}
+			fclose($input_stream);
             
-            $sql='UPDATE '.$this->dbPrefix.'db_config SET value='.$cur_version.' WHERE ( name=\'db_version\' )';
-            $this->logging->log('SQL query : '.$sql,DEBUG );
-            if ($db_conn->query($sql) === false) {
-                $this->logging->log('Cannot update db version. Query : ' . $sql,WARN);
-                return true;
-            }
+			$sql='UPDATE '.$this->dbPrefix.'db_config SET value='.$cur_version.' WHERE ( name=\'db_version\' )';
+			$this->logging->log('SQL query : '.$sql,DEBUG );
+			if ($db_conn->query($sql) === false) {
+				$this->logging->log('Cannot update db version. Query : ' . $sql,WARN);
+				return true;
+			}
             
-            $this->logging->log('Schema updated to version : '.$cur_version ,INFO);
-        }
-        return false;
-    }
+			$this->logging->log('Schema updated to version : '.$cur_version ,INFO);
+		}
+		return false;
+	}
     
-    /**
-     * Get database message for update to $target_version
-     *     @param string $prefix  file prefix of sql update File
-     *     @param int $cur_version  current db version number
-     *     @param int $target_version  target db version number
-     *     @return string : return messages or 'ERROR'.
-     */
-    private function update_schema_message($prefix,$cur_version,$target_version)
-    {
+	/**
+	 * Get database message for update to $target_version
+	 *     @param string $prefix  file prefix of sql update File
+	 *     @param int $cur_version  current db version number
+	 *     @param int $target_version  target db version number
+	 *     @return string : return messages or 'ERROR'.
+	 */
+	private function update_schema_message($prefix,$cur_version,$target_version)
+	{
  
-        $message='';
-        $this->logging->log('getting message for upgrade',DEBUG );
-        while($cur_version<$target_version)
-        {
-            $cur_version++;
-            $updateFile=$prefix.'v'.($cur_version-1).'_v'.$cur_version.'.sql';
-            $input_stream=fopen($updateFile, 'r');
-            if ($input_stream=== false)
-            {
-                $this->logging->log("Error reading update file ". $updateFile,2,'');
-                return 'ERROR';
-            }
-            do 
-            { 
-                $line=fgets($input_stream); 
-            }
-            while ($line !== false && !preg_match('/#MESSAGE/',$line));
-            fclose($input_stream);
-            if ($line === false)
-            {
-                $this->logging->log("No message in file ". $updateFile,2,'');
-                return '';
-            }
-            $message .= ($cur_version-1) . '->' . $cur_version. ' : ' . preg_replace('/#MESSAGE : /','',$line)."\n";
-        }
-        return $message;
-    }
+		$message='';
+		$this->logging->log('getting message for upgrade',DEBUG );
+		while($cur_version<$target_version)
+		{
+			$cur_version++;
+			$updateFile=$prefix.'v'.($cur_version-1).'_v'.$cur_version.'.sql';
+			$input_stream=fopen($updateFile, 'r');
+			if ($input_stream=== false)
+			{
+				$this->logging->log("Error reading update file ". $updateFile,2,'');
+				return 'ERROR';
+			}
+			do 
+			{ 
+				$line=fgets($input_stream); 
+			}
+			while ($line !== false && !preg_match('/#MESSAGE/',$line));
+			fclose($input_stream);
+			if ($line === false)
+			{
+				$this->logging->log("No message in file ". $updateFile,2,'');
+				return '';
+			}
+			$message .= ($cur_version-1) . '->' . $cur_version. ' : ' . preg_replace('/#MESSAGE : /','',$line)."\n";
+		}
+		return $message;
+	}
     
 }
\ No newline at end of file

library/Trapdirector/Plugins/NetworkRule.php

Indentation +83 added lines, -83 removed lines

@@ -21,104 +21,104 @@
  */
 class NetworkRule extends PluginTemplate
 {        
-    /** @var string $description Description of plugin */
-    public $description='Network functions to use into rules
+	/** @var string $description Description of plugin */
+	public $description='Network functions to use into rules
 test test test';
     
-    /** @var array[] $functions Functions of this plugin for rule eval. 
-     * If no functions are declared, set to empty array
-     * $functions[<name>]['function'] : Name of the function to be called in this class
-     * $functions[<name>]['params'] : Description of input parameters of function.
-     * $functions[<name>]['description'] : Description. Can be multiline.
-    */
-    public $functions=array(
-        'inNetwork' => array( // The name of the function in rules
-            'function'      =>  'isInNetwork', // Name of the function 
-            'params'        =>  '<IP to test>,<Network IP>,<Network mask (CIDR)>', // parameters description
-            'description'   =>  'Test if IP is in network, ex : __inNetwork(192.168.123.5,192.168.123.0,24) returns true
+	/** @var array[] $functions Functions of this plugin for rule eval. 
+	 * If no functions are declared, set to empty array
+	 * $functions[<name>]['function'] : Name of the function to be called in this class
+	 * $functions[<name>]['params'] : Description of input parameters of function.
+	 * $functions[<name>]['description'] : Description. Can be multiline.
+	 */
+	public $functions=array(
+		'inNetwork' => array( // The name of the function in rules
+			'function'      =>  'isInNetwork', // Name of the function 
+			'params'        =>  '<IP to test>,<Network IP>,<Network mask (CIDR)>', // parameters description
+			'description'   =>  'Test if IP is in network, ex : __inNetwork(192.168.123.5,192.168.123.0,24) returns true
 Does not work with IPV6' // Description (can be multiline).
-        ),
-        'test' => array( // The name of the function in rules
-            'function'      =>  'testParam', // Name of the function
-            'params'        =>  '<boolean to return as string>', // parameters description
-            'description'   =>  'Returns value passed as argument' // Description (can be multiline).
-        )
-    );
+		),
+		'test' => array( // The name of the function in rules
+			'function'      =>  'testParam', // Name of the function
+			'params'        =>  '<boolean to return as string>', // parameters description
+			'description'   =>  'Returns value passed as argument' // Description (can be multiline).
+		)
+	);
     
-    /** @var boolean $catchAllTraps Set to true if all traps will be sent to the plugin NOT IMPLEMENTED */
-    public $catchAllTraps=false;
+	/** @var boolean $catchAllTraps Set to true if all traps will be sent to the plugin NOT IMPLEMENTED */
+	public $catchAllTraps=false;
     
-    /** @var boolean $processTraps Set to true if plugins can handle traps NOT IMPLEMENTED */
-    public $processTraps=false;
+	/** @var boolean $processTraps Set to true if plugins can handle traps NOT IMPLEMENTED */
+	public $processTraps=false;
     
-    /**
-     * Constructor. Can throw exceptions on error, but no logging at this point.
-     * @throws \Exception
-     * @return \Trapdirector\Plugins\NetworkRule
-     */
-    function __construct()
-    {
-        $this->name=basename(__FILE__,'.php');
-        return $this;
-    }
+	/**
+	 * Constructor. Can throw exceptions on error, but no logging at this point.
+	 * @throws \Exception
+	 * @return \Trapdirector\Plugins\NetworkRule
+	 */
+	function __construct()
+	{
+		$this->name=basename(__FILE__,'.php');
+		return $this;
+	}
     
-    /**
-     * Function called by trapdirector if found in rules
-     * Parameters check has to be done in function.
-     * @param array $params Function parameters
-     * @throws Exception
-     * @return bool Evaluation 
-     */
-    public function isInNetwork(array $params) : bool
-    {
-        // Check param numbers and thrown exception if not correct.
-        if (count($params)!=3)
-        {
-            throw new Exception('Invalid number of parameters : ' . count($params));
-        }
+	/**
+	 * Function called by trapdirector if found in rules
+	 * Parameters check has to be done in function.
+	 * @param array $params Function parameters
+	 * @throws Exception
+	 * @return bool Evaluation 
+	 */
+	public function isInNetwork(array $params) : bool
+	{
+		// Check param numbers and thrown exception if not correct.
+		if (count($params)!=3)
+		{
+			throw new Exception('Invalid number of parameters : ' . count($params));
+		}
         
-        $ip = $params[0];
-        $net = $params[1];
-        $masq = $params[2];
+		$ip = $params[0];
+		$net = $params[1];
+		$masq = $params[2];
         
         
-        $this->log('#'. $ip . '# / #' . $net . '# / #' . $masq,DEBUG);
+		$this->log('#'. $ip . '# / #' . $net . '# / #' . $masq,DEBUG);
         
-        $ip2 = ip2long($ip);
-        $net2 = ip2long($net);
+		$ip2 = ip2long($ip);
+		$net2 = ip2long($net);
         
-        if ($ip2 === false )
-        {
-            $this->log('Invalid IP : #' . $ip.'#',WARN);
-            throw new Exception('Invalid IP');
-        }
-        if ($net2 === false)
-        {
-            $this->log('Invalid network',WARN);
-            throw new Exception('Invalid net');
-        }
-        if ($masq<1 || $masq > 32)
-        {
-            $this->log('Invalid masq',WARN);
-            throw new Exception('Invalid net masq');
-        }
-        // $range is in IP/CIDR format eg 127.0.0.1/24
+		if ($ip2 === false )
+		{
+			$this->log('Invalid IP : #' . $ip.'#',WARN);
+			throw new Exception('Invalid IP');
+		}
+		if ($net2 === false)
+		{
+			$this->log('Invalid network',WARN);
+			throw new Exception('Invalid net');
+		}
+		if ($masq<1 || $masq > 32)
+		{
+			$this->log('Invalid masq',WARN);
+			throw new Exception('Invalid net masq');
+		}
+		// $range is in IP/CIDR format eg 127.0.0.1/24
 
-        $masq = pow( 2, ( 32 - $masq ) ) - 1;
-        $masq = ~ $masq;
-        return ( ( $ip2 & $masq ) == ( $net2 & $masq ) );
+		$masq = pow( 2, ( 32 - $masq ) ) - 1;
+		$masq = ~ $masq;
+		return ( ( $ip2 & $masq ) == ( $net2 & $masq ) );
         
-    }
+	}
     
-    public function testParam(array $param)
-    {
-        if (count($param)!=1)
-        {
-            throw new Exception('Invalid number of parameters : ' . count($param));
-        }
-        if ($param[0] == 'true') return true;
-        return false;
-    }
+	public function testParam(array $param)
+	{
+		if (count($param)!=1)
+		{
+			throw new Exception('Invalid number of parameters : ' . count($param));
+		}
+		if ($param[0] == 'true') return true;
+		return false;
+	}
 }