pandorafms/pandora_server/lib/PandoraFMS/Config.pm

package PandoraFMS::Config;
##########################################################################
# Pandora Config package
##########################################################################
# Copyright (c) 2004-2007 Sancho Lerena, slerena@gmail.com
# Copyright (c) 2005-2007 Artica Soluciones Tecnologicas S.L
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License 
# as published by the Free Software Foundation; at version 2.
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
##########################################################################

use warnings;
use Time::Local;
use Date::Manip;
use PandoraFMS::Tools;
use PandoraFMS::DB;
require Exporter;

our @ISA = ("Exporter");
our %EXPORT_TAGS = ( 'all' => [ qw( ) ] );
our @EXPORT_OK = ( @{ $EXPORT_TAGS{'all'} } );
our @EXPORT = qw( 	pandora_help_screen
			pandora_init
			pandora_loadconfig  );

# There is no global vars, all variables (setup) passed as hash reference

# version: Defines actual version of Pandora Server for this module only
my $pandora_version = "1.3-dev";
my $pandora_build="PS070731";
our $VERSION = $pandora_version." ".$pandora_build;

# Setup hash
my %pa_config;

# Public functions
##########################################################################
# SUB pandora_help_screen()
#  Show a help screen an exits
##########################################################################

sub help_screen {
	printf "\n\nSyntax: \n  pandora_server < fullpathname to pandora server configuration file > [ options ] \n\n";
	printf "Following options are optional : \n";
	printf "            -v  :  Verbose mode activated, give more information in logfile \n";
	printf "            -d  :  Debug mode activated, give extensive information in logfile \n";
	printf "            -D  :  Daemon mode (runs in backgroup)\n";
	printf "            -h  :  This screen, show a little help screen \n";
	printf " \n";
	exit;
}

##########################################################################
# SUB pandora_init ( %pandora_cfg )
# Makes the initial parameter parsing, initializing and error checking
##########################################################################

sub pandora_init {
	my $pa_config = $_[0];
	my $init_string = $_[1];
	printf "\n$init_string $pandora_version Build $pandora_build Copyright (c) 2004-2007 ArticaST\n";
	printf "This program is Free Software, licensed under the terms of GPL License v2.\n";
	printf "You can download latest versions and documentation at http://pandora.sourceforge.net. \n\n";

	# Check we are running GNU/Linux
	die "[ERROR] This isn't GNU/Linux. Pandora FMS Servers are only OFFICIALLY supported in GNU/Linux.\nContact us if you require assistance running Pandora FMS Server in other OS.\n\n" unless ($^O =~ m/linux/i);

	# Load config file from command line
	if ($#ARGV == -1 ){
		print "I Need at least one parameter: Complete path to Pandora FMS Server configuration file. \n";
		help_screen;
		exit;
	}
   	$pa_config->{"verbosity"}=1; 	# Verbose 1 by default
	$pa_config->{"daemon"}=0; 	# Daemon 0 by default

	# If there are not valid parameters
	my $parametro;
	my $ltotal=$#ARGV; my $ax;
	for ($ax=0;$ax<=$ltotal;$ax++){
		$parametro = $ARGV[$ax];
		if ($parametro =~ m/-h\z/i ) { help_screen();  }
		elsif ($parametro =~ m/-help\z/i ) { help_screen();  }
		elsif ($parametro =~ m/-help\z/i ) { help_screen();  }
		elsif ($parametro =~ m/-v\z/i) { $pa_config->{"verbosity"}=5; }
		elsif ($parametro =~ m/-d\z/) { $pa_config->{"verbosity"}=10; }
		elsif ($parametro =~ m/-D\z/) { $pa_config->{"daemon"}=1; }
		else { ($pa_config->{"pandora_path"} = $parametro); }
	}
	if ($pa_config->{"pandora_path"} eq ""){
		print "I Need at least one parameter: Complete path to Pandora FMS configuration file. \n";
		exit;
	}
}

##########################################################################
# Read external configuration file
##########################################################################

sub pandora_loadconfig {
	my $pa_config = $_[0];
	my $opmode = $_[1]; # 0 dataserver, 1 network server, 2 snmp console, 3 recon server
	my $archivo_cfg = $pa_config->{'pandora_path'};
	my $buffer_line;
	my @command_line;
	my $tbuf;

	# Default values
	$pa_config->{'version'} = $pandora_version;
	$pa_config->{'build'} = $pandora_build;
	$pa_config->{"dbuser"} ="pandora";
	$pa_config->{"dbpass"} = "pandora";
	$pa_config->{"dbhost"} = "localhost";
	$pa_config->{"dbname"} = "pandora";
	$pa_config->{"basepath"}=$pa_config->{'pandora_path'}; # Compatibility with Pandora 1.1
	$pa_config->{"incomingdir"}="/var/spool/pandor/data_in";
	$pa_config->{"server_threshold"}=30;
	$pa_config->{"alert_threshold"}=60;
	$pa_config->{"logfile"}="/var/log/pandora_server.log";
	$pa_config->{"errorlogfile"}="/var/log/pandora_server.error";
	$pa_config->{"networktimeout"}=15; 	# By default, not in config file yet
	$pa_config->{"pandora_master"}=1; 	# on by default
	$pa_config->{"pandora_check"}=0; 	# on by default
	$pa_config->{"snmpconsole"}=0; 	# off by default
	$pa_config->{"version"}=$pandora_version;
	$pa_config->{"build"}=$pandora_build;
	$pa_config->{"servername"}=`hostname`;
	$pa_config->{"servername"}=~ s/\s//g; # Replace ' ' chars
	$pa_config->{"networkserver"}=0;
	$pa_config->{"dataserver"}=0;
	$pa_config->{"reconserver"}=0;
	$pa_config->{"servermode"}="";
	$pa_config->{'pandora_snmp_logfile'}="/var/log/pandora_snmptrap.log";
	$pa_config->{"network_threads"}=5; # Fixed default
	$pa_config->{"keepalive"}=60; # 60 Seconds initially for server keepalive
	$pa_config->{"keepalive_orig"} = $pa_config->{"keepalive"};
	# Check for UID0
	if ($> == 0){
		printf " [W] It is not a good idea running Pandora FMS Server as root user, please DON'T DO IT!\n";
	}
	# Check for file
	if ( ! -e $archivo_cfg ) {
		printf "\n[ERROR] Cannot open configuration file at $archivo_cfg. \n";
		printf " Please specify a valid Pandora FMS Home Directory in command line. \n";
		exit 1;
	}
	# Collect items from config file and put in an array 
	open (CFG, "< $archivo_cfg");
	while (<CFG>){
		$buffer_line = $_;
		if ($buffer_line =~ /^[a-zA-Z]/){ # begins with letters
			if ($buffer_line =~ m/([\w\-\_\.]+)\s([0-9\w\-\_\.\/\?\&\=\)\(\_\-\!\*\@\#\%\$\~\"\']+)/){
				push @command_line,$buffer_line;
			}
		}
	}
 	close (CFG);

 	# Process this array with commandline like options 
 	# Process input parameters

 	my @args = @command_line;
 	my $parametro;
 	my $ltotal=$#args; 
	my $ax;

 	# Has read setup file ok ?
 	if ( $ltotal == 0 ) {
  		print "[ERROR] No valid setup tokens readed in $archivo_cfg ";
  		exit;
 	}
 
 	for ($ax=0;$ax<=$ltotal;$ax++){
  		$parametro = $args[$ax];
		if ($parametro =~ m/^incomingdir\s(.*)/i) {  
			$pa_config->{"incomingdir"} = $1; 
			$tbuf= $1;		
			if ($tbuf =~ m/^\.(.*)/){
				$pa_config->{"incomingdir"} =$pa_config->{"basepath"}.$1;
			} else {
				$pa_config->{"incomingdir"} = $tbuf;
			}
		}
		elsif ($parametro =~ m/^log_file\s(.*)/i) { 
			$tbuf= $1;		
			if ($tbuf =~ m/^\.(.*)/){
				$pa_config->{"logfile"} = $pa_config->{"basepath"}.$1;
			} else {
				$pa_config->{"logfile"} = $tbuf;
			}
		}
  		elsif ($parametro =~ m/^errorlog_file\s(.*)/i) { 
			$tbuf= $1;		
			if ($tbuf =~ m/^\.(.*)/){
				$pa_config->{"errorlogfile"} = $pa_config->{"basepath"}.$1;
			} else {
				$pa_config->{"errorlogfile"} = $tbuf;
			}
		}
		elsif ($parametro =~ m/^pandora_snmp_logfile\s(.*)/i) { $pa_config->{'pandora_snmp_logfile'}= $1; }
		elsif ($parametro =~ m/^dbname\s(.*)/i) { $pa_config->{'dbname'}= $1; }
		elsif ($parametro =~ m/^dbuser\s(.*)/i) { $pa_config->{'dbuser'}= $1; }
  		elsif ($parametro =~ m/^dbpass\s(.*)/i) { $pa_config->{'dbpass'}= $1; }
  		elsif ($parametro =~ m/^dbhost\s(.*)/i) { $pa_config->{'dbhost'}= $1; }
  		elsif ($parametro =~ m/^daemon\s([0-9]*)/i) { $pa_config->{'daemon'}= $1;}
		elsif ($parametro =~ m/^dataserver\s([0-9]*)/i) {
			$pa_config->{'dataserver'}= $1;
		}
		elsif ($parametro =~ m/^reconserver\s([0-9]*)/i) {
			$pa_config->{'reconserver'}= $1;
		}
		elsif ($parametro =~ m/^networkserver\s([0-9]*)/i) {
			$pa_config->{'networkserver'}= $1;
		}
		elsif ($parametro =~ m/^servername\s(.*)/i) { $pa_config->{'servername'}= $1; }
		elsif ($parametro =~ m/^checksum\s([0-9])/i) { $pa_config->{"pandora_check"} = $1; }
		elsif ($parametro =~ m/^master\s([0-9])/i) { $pa_config->{"pandora_master"} = $1; }
		elsif ($parametro =~ m/^snmpconsole\s([0-9])/i) {
			$pa_config->{"snmpconsole"} = $1;
		}
  		elsif ($parametro =~ m/^verbosity\s([0-9]*)/i) { $pa_config->{"verbosity"} = $1; } 
  		elsif ($parametro =~ m/^server_threshold\s([0-9]*)/i) { $pa_config->{"server_threshold"}  = $1; } 
		elsif ($parametro =~ m/^alert_threshold\s([0-9]*)/i) { $pa_config->{"alert_threshold"} = $1; } 
		elsif ($parametro =~ m/^network_timeout\s([0-9]*)/i) { $pa_config->{'networktimeout'}= $1; }
		elsif ($parametro =~ m/^network_threads\s([0-9]*)/i) { $pa_config->{'network_threads'}= $1; }
		elsif ($parametro =~ m/^server_keepalive\s([0-9]*)/i) { $pa_config->{"keepalive"} = $1; $pa_config->{"keepalive_orig"} = $1; }
 	}
	if ( $pa_config->{"verbosity"} > 0){
		print " [*] Server basepath is ".$pa_config->{'basepath'}."\n";
		print " [*] Server logfile at ".$pa_config->{"logfile"}."\n";
		print " [*] Server errorlogfile at ".$pa_config->{"errorlogfile"}."\n";
		print " [*] Server incoming directory at ".$pa_config->{"incomingdir"}."\n";
		print " [*] Server keepalive ".$pa_config->{"keepalive"}."\n";
		print " [*] Server threshold ".$pa_config->{"server_threshold"}."\n";
	}
 	# Check for valid token token values
 	if (( $pa_config->{"dbuser"} eq "" ) || ( $pa_config->{"basepath"} eq "" ) || ( $pa_config->{"incomingdir"} eq "" ) || ( $pa_config->{"logfile"} eq "" ) || ( $pa_config->{"dbhost"} eq "")  || ( $pa_config->{"pandora_master"} eq "") || ( $pa_config->{"dbpass"} eq "" ) ) {
		print "[ERROR] Bad Config values. Be sure that $archivo_cfg is a valid setup file. \n\n";
		exit;
	}
	if (($opmode ==0) && ($pa_config->{"dataserver"} ne 1)) {
		print " [ERROR] You must enable Dataserver in setup file to run Pandora FMS Data Server. \n\n";
		exit;
	} 
	if (($opmode ==1) && ($pa_config->{"networkserver"} ne 1)) {
		print " [ERROR] You must enable NetworkServer in setup file to run Pandora FMS Network Server. \n\n";
		exit;
	}
	if (($opmode ==2) && ($pa_config->{"snmpconsole"} ne 1)) {
		print " [ERROR] You must enable SnmpConsole in setup file to run Pandora FMS SNMP Console. \n\n";
		exit;
	}
	if (($opmode ==3) && ($pa_config->{"reconserver"} ne 1)) {
		print " [ERROR] You must enable Recon server in setup file to run Pandora FMS Recon server. \n\n";
		exit;
	}
	if ($opmode == 0){
		print " [*] You are running Pandora FMS Data Server. \n";
		$parametro ="Pandora FMS Data Server";
		$pa_config->{"servermode"}="_Data";
	}
	if ($opmode == 1){
		print " [*] You are running Pandora FMS Network Server. \n";
		$parametro ="Pandora FMS Network Server";
		$pa_config->{"servermode"}="_Net";
	}
	if ($opmode == 2){
		print " [*] You are running Pandora FMS SNMP Console. \n";
		$parametro ="Pandora FMS SNMP Console";
		$pa_config->{"servermode"}="_SNMP";
	}
	if ($opmode == 3){
		print " [*] You are running Pandora FMS Recon Server. \n";
		$parametro ="Pandora FMS Recon Server";
		$pa_config->{"servermode"}="_Recon";
	}
	if ($pa_config->{"pandora_check"} == 1) {
		print " [*] MD5 Security enabled.\n";
	}
	if ($pa_config->{"pandora_master"} == 1) {
		print " [*] This server is running in MASTER mode.\n";
	}
	logger ($pa_config, "Launching $parametro $pa_config->{'version'} $pa_config->{'build'}", 0);
	my $config_options = "Logfile at ".$pa_config->{"logfile"}.", Basepath is ".$pa_config->{"basepath"}.", Checksum is ".$pa_config->{"pandora_check"}.", Master is ".$pa_config->{"pandora_master"}.", SNMP Console is ".$pa_config->{"snmpconsole"}.", Server Threshold at ".$pa_config->{"server_threshold"}." sec, verbosity at ".$pa_config->{"verbosity"}.", Alert Threshold at $pa_config->{'alert_threshold'}, ServerName is '".$pa_config->{'servername'}.$pa_config->{"servermode"}."'";
	logger ($pa_config, "Config options: $config_options");
	my $dbh;
	# Check valid Database variables and update server status
	eval {
		$dbh = DBI->connect("DBI:mysql:$pa_config->{'dbname'}:$pa_config->{'dbhost'}:3306", $pa_config->{'dbuser'}, $pa_config->{'dbpass'}, { RaiseError => 1, AutoCommit => 1 });
		pandora_updateserver ($pa_config, $pa_config->{'servername'},1, $opmode, $dbh); # Alive status
	};
	if ($@) {

		logger ($pa_config, "Error connecting database in init Phase. Aborting startup.",0);
		print (" [E] Error connecting database in init Phase. Aborting startup. \n\n");
		print $@;
		exit;
	}
	print " [*] Pandora FMS Server [".$pa_config->{'servername'}.$pa_config->{"servermode"}."] is running and operative \n";
	$pa_config->{'server_id'} = dame_server_id ($pa_config, $pa_config->{'servername'}.$pa_config->{"servermode"}, $dbh);
	
	# Dump all errors to errorlog
	# DISABLED in DEBUGMODE
	# ENABLE FOR PRODUCTION
	open STDERR, ">>$pa_config->{'errorlogfile'}" or die "Can't write to Errorlog : $!";
}

# End of function declaration
# End of defined Code

1;
__END__