tyler.durden/pandorafms
1
0
Fork 0
mirror of https://github.com/pandorafms/pandorafms.git synced 2025-07-31 01:35:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

2011-05-09 Sergio Martin <sergio.martin@artica.es>

Browse Source 
* lib/PandoraFMS/DB.pm
	lib/PandoraFMS/Core.pm
	util/pandora_manage.pl: Moved some functions from CLI to common
	files



git-svn-id: https://svn.code.sf.net/p/pandora/code/trunk@4317 c3f86ba8-e40f-0410-aaad-9ba5e7f4b01f
This commit is contained in:
zarzuelo 2011-05-09 09:42:01 +00:00
parent 98c4bd8c68
commit 59c2600490
4 changed files with 249 additions and 219 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pandora_server/ChangeLog
View File

@ -1,3 +1,10 @@
2011-05-09 Sergio Martin <sergio.martin@artica.es>
* lib/PandoraFMS/DB.pm
lib/PandoraFMS/Core.pm
util/pandora_manage.pl: Moved some functions from CLI to common
files
2011-05-07 Sancho Lerena <slerena@artica.es> 2011-05-07 Sancho Lerena <slerena@artica.es>
* lib/PandoraFMS/ReconServer.pm: Ported changes from 3.2 on Recon. * lib/PandoraFMS/ReconServer.pm: Ported changes from 3.2 on Recon.

77
pandora_server/lib/PandoraFMS/Core.pm
View File

@ -132,7 +132,10 @@ our @EXPORT = qw(
pandora_create_incident pandora_create_incident
pandora_create_module pandora_create_module
pandora_create_module_from_hash pandora_create_module_from_hash
pandora_create_template_module
pandora_create_template_module_action
pandora_delete_agent pandora_delete_agent
pandora_delete_all_template_module_actions
pandora_delete_module pandora_delete_module
pandora_evaluate_alert pandora_evaluate_alert
pandora_evaluate_compound_alert pandora_evaluate_compound_alert
@ -156,6 +159,7 @@ our @EXPORT = qw(
pandora_update_module_on_error pandora_update_module_on_error
pandora_update_module_from_hash pandora_update_module_from_hash
pandora_update_server pandora_update_server
pandora_update_template_module
pandora_group_statistics pandora_group_statistics
pandora_server_statistics pandora_server_statistics
pandora_self_monitoring pandora_self_monitoring
@ -1030,6 +1034,79 @@ sub pandora_update_agent ($$$$$$$;$$$$$$) {
} }
##########################################################################
=head2 C<< pandora_create_template_module(I<$pa_config>, I<$id_agent_module>, I<$id_alert_template>, I<$dbh>, I<$id_policy_alerts>, I<$disabled>, I<$standby>) >>
Create a template module.
=cut
##########################################################################
sub pandora_create_template_module ($$$$;$$$) {
my ($pa_config, $id_agent_module, $id_alert_template, $dbh, $id_policy_alerts, $disabled, $standby) = @_;
$id_policy_alerts = 0 unless defined $id_policy_alerts;
$disabled = 0 unless defined $disabled;
$standby = 0 unless defined $standby;
my $module_name = get_module_name($dbh, $id_agent_module);
logger($pa_config, "Creating alert of template '$id_alert_template' on agent module '$module_name'.", 10);
$dbh->do("INSERT INTO talert_template_modules (`id_agent_module`, `id_alert_template`, `id_policy_alerts`, `disabled`, `standby`) VALUES ($id_agent_module, $id_alert_template, $id_policy_alerts, $disabled, $standby)");
return $dbh->{'mysql_insertid'};
}
##########################################################################
=head2 C<< pandora_update_template_module(I<$pa_config>, I<$id_alert>, I<$dbh>, I<$id_policy_alerts>, I<$disabled>, I<$standby>) >>
Update a template module.
=cut
##########################################################################
sub pandora_update_template_module ($$$;$$$) {
my ($pa_config, $id_alert, $dbh, $id_policy_alerts, $disabled, $standby) = @_;
$id_policy_alerts = 0 unless defined $id_policy_alerts;
$disabled = 0 unless defined $disabled;
$standby = 0 unless defined $standby;
#my $module_name = get_module_name($dbh, $id_agent_module);
#logger($pa_config, "Update alert of template '$id_alert_template' on agent module '$module_name'.", 10);
$dbh->do("UPDATE talert_template_modules SET `id_policy_alerts` = '$id_policy_alerts', `disabled` = '$disabled', `standby` = '$standby' WHERE id = $id_alert");
return $dbh->{'mysql_insertid'};
}
##########################################################################
=head2 C<< pandora_create_template_module(I<$pa_config>, I<$parameters>, I<$dbh>) >>
Create a template action.
=cut
##########################################################################
sub pandora_create_template_module_action ($$$) {
my ($pa_config, $parameters, $dbh) = @_;
logger($pa_config, "Creating module alert action to alert '$parameters->{'id_alert_template_module'}'.", 10);
my $action_id = db_process_insert($dbh, 'id', 'talert_template_module_actions', $parameters);
return $action_id;
}
##########################################################################
=head2 C<< pandora_delete_all_template_module_actions(I<$dbh>, I<$template_module_id>) >>
Delete all actions of policy template module.
=cut
##########################################################################
sub pandora_delete_all_template_module_actions ($$) {
my ($dbh, $template_module_id) = @_;
return db_do ($dbh, 'DELETE FROM talert_template_module_actions WHERE id_alert_template_module = ?', $template_module_id);
}
########################################################################## ##########################################################################
=head2 C<< pandora_update_agent_address(I<$pa_config>, I<$agent_id>, I<$address>, I<$dbh>) >> =head2 C<< pandora_update_agent_address(I<$pa_config>, I<$agent_id>, I<$address>, I<$dbh>) >>

11
pandora_server/lib/PandoraFMS/DB.pm
View File

@ -42,6 +42,7 @@ our @EXPORT = qw(
get_agent_id get_agent_id
get_agent_name get_agent_name
get_agent_module_id get_agent_module_id
get_alert_template_module_id
get_db_rows get_db_rows
get_db_single_row get_db_single_row
get_db_value get_db_value
@ -398,6 +399,16 @@ sub db_update ($$;@) {
return $rows; return $rows;
} }
##########################################################################
## Return alert template-module ID given the module and template ids.
##########################################################################
sub get_alert_template_module_id ($$$) {
my ($dbh, $id_module, $id_template) = @_;
my $rc = get_db_value ($dbh, "SELECT id FROM talert_template_modules WHERE id_agent_module = ? AND id_alert_template = ?", $id_module, $id_template);
return defined ($rc) ? $rc : -1;
}
########################################################################## ##########################################################################
## SQL insert. Returns the ID of the inserted row. ## SQL insert. Returns the ID of the inserted row.
########################################################################## ##########################################################################

373
pandora_server/util/pandora_manage.pl
View File

@ -174,41 +174,6 @@ sub pandora_manage_init ($) {
help_screen () if ($conf->{'pandora_path'} =~ m/--*h\w*\z/i ); help_screen () if ($conf->{'pandora_path'} =~ m/--*h\w*\z/i );
} }
##########################################################################
## Create a template module.
##########################################################################
sub pandora_create_template_module ($$$$;$$$) {
my ($pa_config, $id_agent_module, $id_alert_template, $dbh, $id_policy_alerts, $disabled, $standby) = @_;
$id_policy_alerts = 0 unless defined $id_policy_alerts;
$disabled = 0 unless defined $disabled;
$standby = 0 unless defined $standby;
my $module_name = get_module_name($dbh, $id_agent_module);
logger($pa_config, "Creating alert of template '$id_alert_template' on agent module '$module_name'.", 10);
$dbh->do("INSERT INTO talert_template_modules (`id_agent_module`, `id_alert_template`, `id_policy_alerts`, `disabled`, `standby`) VALUES ($id_agent_module, $id_alert_template, $id_policy_alerts, $disabled, $standby)");
return $dbh->{'mysql_insertid'};
}
##########################################################################
## Update a template module.
##########################################################################
sub pandora_update_template_module ($$$;$$$) {
my ($pa_config, $id_alert, $dbh, $id_policy_alerts, $disabled, $standby) = @_;
$id_policy_alerts = 0 unless defined $id_policy_alerts;
$disabled = 0 unless defined $disabled;
$standby = 0 unless defined $standby;
#my $module_name = get_module_name($dbh, $id_agent_module);
#logger($pa_config, "Update alert of template '$id_alert_template' on agent module '$module_name'.", 10);
$dbh->do("UPDATE talert_template_modules SET `id_policy_alerts` = '$id_policy_alerts', `disabled` = '$disabled', `standby` = '$standby' WHERE id = $id_alert");
return $dbh->{'mysql_insertid'};
}
########################################################################## ##########################################################################
## Delete a template module. ## Delete a template module.
########################################################################## ##########################################################################
@ -231,15 +196,6 @@ sub pandora_delete_template_module_action ($$$) {
return db_do ($dbh, 'DELETE FROM talert_template_module_actions WHERE id_alert_template_module = ? AND id_alert_action = ?', $template_module_id, $action_id); return db_do ($dbh, 'DELETE FROM talert_template_module_actions WHERE id_alert_template_module = ? AND id_alert_action = ?', $template_module_id, $action_id);
} }
##########################################################################
## Delete all actions of policy template module
##########################################################################
sub pandora_delete_all_template_module_actions ($$) {
my ($dbh, $template_module_id) = @_;
return db_do ($dbh, 'DELETE FROM talert_template_module_actions WHERE id_alert_template_module = ?', $template_module_id);
}
########################################################################## ##########################################################################
## Create a user. ## Create a user.
########################################################################## ##########################################################################
@ -270,19 +226,6 @@ else {
} }
} }
##########################################################################
## Create a template action.
##########################################################################
sub pandora_create_template_module_action ($$$) {
my ($pa_config, $parameters, $dbh) = @_;
logger($pa_config, "Creating module alert action to alert '$parameters->{'id_alert_template_module'}'.", 10);
my $action_id = db_process_insert($dbh, 'id', 'talert_template_module_actions', $parameters);
return $action_id;
}
########################################################################## ##########################################################################
## Assign a profile to the given user/group. ## Assign a profile to the given user/group.
########################################################################## ##########################################################################
@ -405,16 +348,6 @@ sub pandora_validate_event_filter ($$$$$$$$$) {
db_do ($dbh, "UPDATE tevento SET estado = 1 WHERE estado = 0".$filter); db_do ($dbh, "UPDATE tevento SET estado = 1 WHERE estado = 0".$filter);
} }
##########################################################################
## Return alert template-module ID given the module and template ids.
##########################################################################
sub get_alert_template_module_id ($$$) {
my ($dbh, $id_module, $id_template) = @_;
my $rc = get_db_value ($dbh, "SELECT id FROM talert_template_modules WHERE id_agent_module = ? AND id_alert_template = ?", $id_module, $id_template);
return defined ($rc) ? $rc : -1;
}
############################################################################### ###############################################################################
############################################################################### ###############################################################################
# PRINT HELP AND CHECK ERRORS FUNCTIONS # PRINT HELP AND CHECK ERRORS FUNCTIONS
@ -1295,159 +1228,161 @@ sub pandora_manage_main ($$$) {
my $policy_id = enterprise_hook('get_policy_id',[$dbh, $policy_name]); my $policy_id = enterprise_hook('get_policy_id',[$dbh, $policy_name]);
exist_check($policy_id,'policy',$policy_name); exist_check($policy_id,'policy',$policy_name);
my $blocked_policies = enterprise_hook('pandora_block_policies', [$dbh]); enterprise_hook('pandora_apply_policy', [$dbh, $conf, $policy_id]);
if($blocked_policies eq '0E0') { #~ my $blocked_policies = enterprise_hook('pandora_block_policies', [$dbh]);
print "[ERROR] The policies are blocked in other terminal.\n\n"; #~
exit; #~ if($blocked_policies eq '0E0') {
} #~ print "[ERROR] The policies are blocked in other terminal.\n\n";
#~ exit;
# Get the agents #~ }
my $array_pointer_ag = enterprise_hook('get_policy_agents',[$dbh, $policy_id]); #~
#~ # Get the agents
if(!defined($array_pointer_ag)) { #~ my $array_pointer_ag = enterprise_hook('get_policy_agents',[$dbh, $policy_id]);
print "[ERROR] This option is not available in OPEN version.\n\n"; #~
exit; #~ if(!defined($array_pointer_ag)) {
} #~ print "[ERROR] This option is not available in OPEN version.\n\n";
#~ exit;
print "[INFO] Applying policy '$policy_name'\n\n"; #~ }
#~
foreach my $agent (@{$array_pointer_ag}) { #~ print "[INFO] Applying policy '$policy_name'\n\n";
my $id_agent = $agent->{'id_agent'}; #~
my $agent_name = get_agent_name($dbh, $id_agent); #~ foreach my $agent (@{$array_pointer_ag}) {
#~ my $id_agent = $agent->{'id_agent'};
# Get the modules #~ my $agent_name = get_agent_name($dbh, $id_agent);
my $array_pointer_mod = enterprise_hook('get_policy_modules',[$dbh, $policy_id]); #~
#~ # Get the modules
if(!defined($array_pointer_mod)) { #~ my $array_pointer_mod = enterprise_hook('get_policy_modules',[$dbh, $policy_id]);
print "[ERROR] This option is not available in OPEN version.\n\n"; #~
exit; #~ if(!defined($array_pointer_mod)) {
} #~ print "[ERROR] This option is not available in OPEN version.\n\n";
#~ exit;
foreach my $module (@{$array_pointer_mod}) { #~ }
# Adapt the fields from tpolicy_modules to tagente_modulos #~
$module->{'id_agente'} = $id_agent; #~ foreach my $module (@{$array_pointer_mod}) {
#~ # Adapt the fields from tpolicy_modules to tagente_modulos
$module->{'id_policy_module'} = $module->{'id'}; #~ $module->{'id_agente'} = $id_agent;
delete $module->{'id'}; #~
#~ $module->{'id_policy_module'} = $module->{'id'};
$module->{'descripcion'} = $module->{'description'}; #~ delete $module->{'id'};
delete $module->{'description'}; #~
#~ $module->{'descripcion'} = $module->{'description'};
$module->{'nombre'} = $module->{'name'}; #~ delete $module->{'description'};
delete $module->{'name'}; #~
#~ $module->{'nombre'} = $module->{'name'};
$module->{'id_modulo'} = $module->{'id_module'}; #~ delete $module->{'name'};
delete $module->{'id_module'}; #~
#~ $module->{'id_modulo'} = $module->{'id_module'};
delete $module->{'id_policy'}; #~ delete $module->{'id_module'};
#~
#Store the conf data #~ delete $module->{'id_policy'};
$configuration_data .= safe_output("\n\n$module->{'configuration_data'}"); #~
#~ #Store the conf data
delete $module->{'configuration_data'}; #~ $configuration_data .= safe_output("\n\n$module->{'configuration_data'}");
#~
my $id_module = get_agent_module_id ($dbh, $module->{'nombre'}, $module->{'id_agente'}); #~ delete $module->{'configuration_data'};
#~
# If the module doesn't exist we create it, otherwise we update it #~ my $id_module = get_agent_module_id ($dbh, $module->{'nombre'}, $module->{'id_agente'});
#~
if($id_module == -1) { #~ # If the module doesn't exist we create it, otherwise we update it
# Create module #~
$id_module = pandora_create_module_from_hash ($conf, $module, $dbh); #~ if($id_module == -1) {
} #~ # Create module
else { #~ $id_module = pandora_create_module_from_hash ($conf, $module, $dbh);
# Update module #~ }
pandora_update_module_from_hash ($conf, $module, 'id_agente_modulo', $id_module, $dbh); #~ else {
} #~ # Update module
#~ pandora_update_module_from_hash ($conf, $module, 'id_agente_modulo', $id_module, $dbh);
# Get policy alerts and create it on created modules #~ }
my $array_pointer_ale = enterprise_hook('get_policy_module_alerts',[$dbh, $policy_id, $module->{'id_policy_module'}]); #~
#~ # Get policy alerts and create it on created modules
foreach my $alert (@{$array_pointer_ale}) { #~ my $array_pointer_ale = enterprise_hook('get_policy_module_alerts',[$dbh, $policy_id, $module->{'id_policy_module'}]);
my $id_alert_template_module = get_alert_template_module_id($dbh, $id_module, $alert->{'id_alert_template'}); #~
#~ foreach my $alert (@{$array_pointer_ale}) {
# Only if the template doesnt exist we create it. If exists we update it #~ my $id_alert_template_module = get_alert_template_module_id($dbh, $id_module, $alert->{'id_alert_template'});
if($id_alert_template_module == -1) { #~
$id_alert_template_module = pandora_create_template_module ($conf, $id_module, $alert->{'id_alert_template'}, $dbh, $alert->{'id'}, $alert->{'disabled'}, $alert->{'standby'}); #~ # Only if the template doesnt exist we create it. If exists we update it
} #~ if($id_alert_template_module == -1) {
else { #~ $id_alert_template_module = pandora_create_template_module ($conf, $id_module, $alert->{'id_alert_template'}, $dbh, $alert->{'id'}, $alert->{'disabled'}, $alert->{'standby'});
pandora_update_template_module ($conf, $id_alert_template_module, $dbh, $alert->{'id'}, $alert->{'disabled'}, $alert->{'standby'}); #~ }
} #~ else {
#~ pandora_update_template_module ($conf, $id_alert_template_module, $dbh, $alert->{'id'}, $alert->{'disabled'}, $alert->{'standby'});
# Get policy alert actions and create it on modules created #~ }
my $array_pointer_aleact = enterprise_hook('get_policy_alert_actions',[$dbh, $alert->{'id'}]); #~
#~ # Get policy alert actions and create it on modules created
pandora_delete_all_template_module_actions ($dbh, $id_alert_template_module); #~ my $array_pointer_aleact = enterprise_hook('get_policy_alert_actions',[$dbh, $alert->{'id'}]);
#~
foreach my $alert_action (@{$array_pointer_aleact}) { #~ pandora_delete_all_template_module_actions ($dbh, $id_alert_template_module);
delete $alert_action->{'id_policy_alert'}; #~
delete $alert_action->{'id'}; #~ foreach my $alert_action (@{$array_pointer_aleact}) {
$alert_action->{'id_alert_template_module'} = $id_alert_template_module; #~ delete $alert_action->{'id_policy_alert'};
#~ delete $alert_action->{'id'};
pandora_create_template_module_action ($conf, $alert_action, $dbh); #~ $alert_action->{'id_alert_template_module'} = $id_alert_template_module;
#~
} #~ pandora_create_template_module_action ($conf, $alert_action, $dbh);
} #~
#~ }
#Add the conf information to the agent conf file #~ }
enterprise_hook('pandora_create_policy_conf_info',[$conf, $policy_name, $configuration_data, $agent_name, $dbh]); #~
#~ #Add the conf information to the agent conf file
# Flag applyed the agent #~ enterprise_hook('pandora_create_policy_conf_info',[$conf, $policy_name, $configuration_data, $agent_name, $dbh]);
enterprise_hook('pandora_apply_agent_policy',[$policy_id, $id_agent, $dbh]); #~
#~ # Flag applyed the agent
} #~ enterprise_hook('pandora_apply_agent_policy',[$policy_id, $id_agent, $dbh]);
#~
# Get policy collections and link it on created modules #~ }
my $array_pointer_col = enterprise_hook('get_policy_collections',[$dbh, $policy_id]); #~
#~ # Get policy collections and link it on created modules
my $collection_data = ''; #~ my $array_pointer_col = enterprise_hook('get_policy_collections',[$dbh, $policy_id]);
#~
foreach my $collection (@{$array_pointer_col}) { #~ my $collection_data = '';
my $collection_name = enterprise_hook('get_collection_name',[$dbh, $collection->{'id_collection'}]); #~
#~ foreach my $collection (@{$array_pointer_col}) {
$collection_data = "\n#file_collection $collection_name\n"; #~ my $collection_name = enterprise_hook('get_collection_name',[$dbh, $collection->{'id_collection'}]);
$collection_data .= "\nfile_collection fc_$collection->{'id_collection'}\n\n"; #~
} #~ $collection_data = "\n#file_collection $collection_name\n";
#~ $collection_data .= "\nfile_collection fc_$collection->{'id_collection'}\n\n";
if($collection_data ne '') { #~ }
enterprise_hook('pandora_delete_collection_agent_from_info',[$conf, $agent_name, $policy_id]); #~
enterprise_hook('pandora_create_collection_conf_info',[$conf, $policy_name, $collection_data,$agent_name,$dbh]); #~ if($collection_data ne '') {
} #~ enterprise_hook('pandora_delete_collection_agent_from_info',[$conf, $agent_name, $policy_id]);
#~ enterprise_hook('pandora_create_collection_conf_info',[$conf, $policy_name, $collection_data,$agent_name,$dbh]);
$configuration_data = ''; #~ }
$collection_data = ''; #~
} #~ $configuration_data = '';
#~ $collection_data = '';
# Get extern policy alerts and create it on modules #~ }
my $array_pointer_ale_ext = enterprise_hook('get_policy_module_alerts',[$dbh, $policy_id, 0]); #~
#~ # Get extern policy alerts and create it on modules
foreach my $alert (@{$array_pointer_ale_ext}) { #~ my $array_pointer_ale_ext = enterprise_hook('get_policy_module_alerts',[$dbh, $policy_id, 0]);
#~
my $array_modules_id = enterprise_hook('get_policy_agents_modules_id',[$dbh, $policy_id, $alert->{'name_extern_module'}]); #~ foreach my $alert (@{$array_pointer_ale_ext}) {
#~
foreach my $module_id (@{$array_modules_id}) { #~ my $array_modules_id = enterprise_hook('get_policy_agents_modules_id',[$dbh, $policy_id, $alert->{'name_extern_module'}]);
my $id_alert_template_module = get_alert_template_module_id($dbh, $module_id->{'id_agente_modulo'}, $alert->{'id_alert_template'}); #~
#~ foreach my $module_id (@{$array_modules_id}) {
if($id_alert_template_module == -1) { #~ my $id_alert_template_module = get_alert_template_module_id($dbh, $module_id->{'id_agente_modulo'}, $alert->{'id_alert_template'});
$id_alert_template_module = pandora_create_template_module ($conf, $module_id->{'id_agente_modulo'}, $alert->{'id_alert_template'}, $dbh, $alert->{'id'}); #~
} #~ if($id_alert_template_module == -1) {
#~ $id_alert_template_module = pandora_create_template_module ($conf, $module_id->{'id_agente_modulo'}, $alert->{'id_alert_template'}, $dbh, $alert->{'id'});
# Get policy alert actions and create it on modules created #~ }
my $array_pointer_aleact = enterprise_hook('get_policy_alert_actions',[$dbh, $alert->{'id'}]); #~
#~ # Get policy alert actions and create it on modules created
pandora_delete_all_template_module_actions ($dbh, $id_alert_template_module); #~ my $array_pointer_aleact = enterprise_hook('get_policy_alert_actions',[$dbh, $alert->{'id'}]);
#~
foreach my $alert_action (@{$array_pointer_aleact}) { #~ pandora_delete_all_template_module_actions ($dbh, $id_alert_template_module);
delete $alert_action->{'id_policy_alert'}; #~
delete $alert_action->{'id'}; #~ foreach my $alert_action (@{$array_pointer_aleact}) {
#~ delete $alert_action->{'id_policy_alert'};
$alert_action->{'id_alert_template_module'} = $id_alert_template_module; #~ delete $alert_action->{'id'};
#~
pandora_create_template_module_action ($conf, $alert_action, $dbh); #~ $alert_action->{'id_alert_template_module'} = $id_alert_template_module;
} #~
} #~ pandora_create_template_module_action ($conf, $alert_action, $dbh);
} #~ }
enterprise_hook('pandora_unblock_policies', [$dbh]); #~ }
#~ }
#~ enterprise_hook('pandora_unblock_policies', [$dbh]);
} }
elsif ($param eq '--force_unblock_policies') { elsif ($param eq '--force_unblock_policies') {
enterprise_hook('pandora_unblock_policies', [$dbh]); enterprise_hook('pandora_unblock_policies', [$dbh]);