* Created the new anvil-manage-host that can check/set if a host is configured. On Strikers, it can age out data, resync data, and check/set if the local database is active.
* Updated striker-prep-database to again enable the postgresql service. Signed-off-by: Digimer <digimer@alteeve.ca>main
Digimer
3 years ago
parent
b575b1ebc1
commit
74b7719cf5
6 changed files with 466 additions and 67 deletions
@ -0,0 +1,310 @@ |
|||||||
|
#!/usr/bin/perl |
||||||
|
# |
||||||
|
# This tool provides a way to perform actions on a host that normally are handled by other processes, like |
||||||
|
# marking a system as configure, etc. This will likely grow constantly over time. |
||||||
|
# |
||||||
|
# Supported switches; |
||||||
|
# --mark-configured, --mark-unconfigured |
||||||
|
# - Set / unset the 'system::configured' flag for the host. |
||||||
|
# --database-active, --database-inactive |
||||||
|
# - On Striker, mark the local database as active / inactive |
||||||
|
# |
||||||
|
|
||||||
|
use strict; |
||||||
|
use warnings; |
||||||
|
use Anvil::Tools; |
||||||
|
|
||||||
|
my $THIS_FILE = ($0 =~ /^.*\/(.*)$/)[0]; |
||||||
|
my $running_directory = ($0 =~ /^(.*?)\/$THIS_FILE$/)[0]; |
||||||
|
if (($running_directory =~ /^\./) && ($ENV{PWD})) |
||||||
|
{ |
||||||
|
$running_directory =~ s/^\./$ENV{PWD}/; |
||||||
|
} |
||||||
|
|
||||||
|
my $anvil = Anvil::Tools->new(); |
||||||
|
|
||||||
|
# Read switches |
||||||
|
$anvil->data->{switches}{'age-out-database'} = ""; |
||||||
|
$anvil->data->{switches}{'check-configured'} = ""; |
||||||
|
$anvil->data->{switches}{'check-database'} = ""; |
||||||
|
$anvil->data->{switches}{'database-active'} = ""; |
||||||
|
$anvil->data->{switches}{'database-inactive'} = ""; |
||||||
|
$anvil->data->{switches}{'mark-configured'} = ""; |
||||||
|
$anvil->data->{switches}{'mark-unconfigured'} = ""; |
||||||
|
$anvil->data->{switches}{'resync-database'} = ""; |
||||||
|
$anvil->Get->switches; |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 2, secure => 0, key => "log_0115", variables => { program => $THIS_FILE }}); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||||
|
'switches::age-out-database' => $anvil->data->{switches}{'age-out-database'}, |
||||||
|
'switches::check-configured' => $anvil->data->{switches}{'check-configured'}, |
||||||
|
'switches::check-database' => $anvil->data->{switches}{'check-database'}, |
||||||
|
'switches::database-active' => $anvil->data->{switches}{'database-active'}, |
||||||
|
'switches::database-inactive' => $anvil->data->{switches}{'database-inactive'}, |
||||||
|
'switches::mark-configured' => $anvil->data->{switches}{'mark-configured'}, |
||||||
|
'switches::mark-unconfigured' => $anvil->data->{switches}{'mark-unconfigured'}, |
||||||
|
'switches::resync-database' => $anvil->data->{switches}{'resync-database'}, |
||||||
|
}}); |
||||||
|
|
||||||
|
$anvil->Database->connect({all => 1}); |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 2, secure => 0, key => "log_0132"}); |
||||||
|
if (not $anvil->data->{sys}{database}{connections}) |
||||||
|
{ |
||||||
|
# No databases, we can't do anything. |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, priority => "err", key => "error_0077"}); |
||||||
|
$anvil->nice_exit({exit_code => 1}); |
||||||
|
} |
||||||
|
|
||||||
|
if (($anvil->data->{switches}{'mark-configured'}) or ($anvil->data->{switches}{'mark-unconfigured'}) or ($anvil->data->{switches}{'check-configured'})) |
||||||
|
{ |
||||||
|
update_config($anvil); |
||||||
|
} |
||||||
|
elsif (($anvil->data->{switches}{'database-active'}) or ($anvil->data->{switches}{'database-inactive'}) or ($anvil->data->{switches}{'check-database'})) |
||||||
|
{ |
||||||
|
update_database($anvil); |
||||||
|
} |
||||||
|
elsif ($anvil->data->{switches}{'age-out-database'}) |
||||||
|
{ |
||||||
|
age_out_data($anvil); |
||||||
|
} |
||||||
|
elsif ($anvil->data->{switches}{'resync-database'}) |
||||||
|
{ |
||||||
|
resync_database($anvil); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
# Show the options. |
||||||
|
print $anvil->Words->string({key => "message_0282"})."\n"; |
||||||
|
} |
||||||
|
|
||||||
|
$anvil->nice_exit({exit_code => 0}); |
||||||
|
|
||||||
|
|
||||||
|
############################################################################################################# |
||||||
|
# Functions # |
||||||
|
############################################################################################################# |
||||||
|
|
||||||
|
sub age_out_data |
||||||
|
{ |
||||||
|
my ($anvil) = @_; |
||||||
|
|
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0285"}); |
||||||
|
$anvil->Database->_age_out_data({debug => 2}); |
||||||
|
|
||||||
|
return(0); |
||||||
|
} |
||||||
|
|
||||||
|
sub resync_database |
||||||
|
{ |
||||||
|
my ($anvil) = @_; |
||||||
|
|
||||||
|
if ($anvil->data->{sys}{database}{connections} == 1) |
||||||
|
{ |
||||||
|
# No reason to proceed. |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0283"}); |
||||||
|
return(0); |
||||||
|
} |
||||||
|
|
||||||
|
# Before we reconnect, make sure all scan agent tables are loaded. |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0286"}); |
||||||
|
$anvil->ScanCore->_scan_directory({directory => $anvil->data->{path}{directories}{scan_agents}}); |
||||||
|
my $tables = ""; |
||||||
|
foreach my $agent_name (sort {$a cmp $b} keys %{$anvil->data->{scancore}{agent}}) |
||||||
|
{ |
||||||
|
my $agent_path = $anvil->data->{scancore}{agent}{$agent_name}; |
||||||
|
my $agent_words = $agent_path.".xml"; |
||||||
|
my $schema_file = $agent_path.".sql"; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||||
|
agent_name => $agent_name, |
||||||
|
agent_path => $agent_path, |
||||||
|
agent_words => $agent_words, |
||||||
|
schema_file => $schema_file, |
||||||
|
}}); |
||||||
|
|
||||||
|
if (-e $schema_file) |
||||||
|
{ |
||||||
|
# See that it's loaded. |
||||||
|
$tables = $anvil->Database->get_tables_from_schema({schema_file => $schema_file}); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { tables => $tables }}); |
||||||
|
foreach my $table (@{$tables}) |
||||||
|
{ |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { table => $table }}); |
||||||
|
} |
||||||
|
$anvil->Database->check_agent_data({ |
||||||
|
agent => $agent_name, |
||||||
|
tables => $tables, |
||||||
|
}); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
# Disconnect and reconnect. |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0284"}); |
||||||
|
$anvil->Database->disconnect(); |
||||||
|
$anvil->refresh(); |
||||||
|
$anvil->Database->connect({ |
||||||
|
all => 1, |
||||||
|
check_for_resync => 1, |
||||||
|
}); |
||||||
|
|
||||||
|
return(0); |
||||||
|
} |
||||||
|
|
||||||
|
sub update_database |
||||||
|
{ |
||||||
|
my ($anvil) = @_; |
||||||
|
|
||||||
|
# Are we a striker? |
||||||
|
my $host_type = $anvil->Get->host_type(); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { host_type => $host_type }}); |
||||||
|
if ($host_type ne "striker") |
||||||
|
{ |
||||||
|
# Nope. |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 0, priority => "err", key => "error_0358"}); |
||||||
|
$anvil->nice_exit({exit_code => 1}); |
||||||
|
} |
||||||
|
|
||||||
|
my $host_uuid = $anvil->Get->host_uuid(); |
||||||
|
my $variable_name = "database::".$host_uuid."::active"; |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||||
|
host_uuid => $host_uuid, |
||||||
|
variable_name => $variable_name, |
||||||
|
}}); |
||||||
|
|
||||||
|
|
||||||
|
# Read if it's active or inactive yet. |
||||||
|
my ($active_value, undef, undef) = $anvil->Database->read_variable({variable_name => "database::".$host_uuid."::active"}); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { active_value => $active_value }}); |
||||||
|
if ($anvil->data->{switches}{'check-database'}) |
||||||
|
{ |
||||||
|
if ($active_value) |
||||||
|
{ |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0277"}); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0276"}); |
||||||
|
} |
||||||
|
return(0); |
||||||
|
} |
||||||
|
|
||||||
|
# Are we enabling or disabling the database? |
||||||
|
if ($active_value) |
||||||
|
{ |
||||||
|
if ($anvil->data->{switches}{'database-active'}) |
||||||
|
{ |
||||||
|
# Already active. |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0279"}); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
# Mark inactive |
||||||
|
my $variable_uuid = $anvil->Database->insert_or_update_variables({ |
||||||
|
variable_name => $variable_name, |
||||||
|
variable_value => "0", |
||||||
|
variable_default => "0", |
||||||
|
variable_description => "striker_0294", |
||||||
|
variable_section => "database", |
||||||
|
variable_source_uuid => "NULL", |
||||||
|
variable_source_table => "", |
||||||
|
}); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { variable_uuid => $variable_uuid }}); |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0280"}); |
||||||
|
} |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
if ($anvil->data->{switches}{'database-inactive'}) |
||||||
|
{ |
||||||
|
# Already inactive |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0281"}); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
# Mark active |
||||||
|
my $variable_uuid = $anvil->Database->insert_or_update_variables({ |
||||||
|
variable_name => $variable_name, |
||||||
|
variable_value => "1", |
||||||
|
variable_default => "0", |
||||||
|
variable_description => "striker_0294", |
||||||
|
variable_section => "database", |
||||||
|
variable_source_uuid => "NULL", |
||||||
|
variable_source_table => "", |
||||||
|
}); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { variable_uuid => $variable_uuid }}); |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0278"}); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return(0); |
||||||
|
} |
||||||
|
|
||||||
|
sub update_config |
||||||
|
{ |
||||||
|
my ($anvil) = @_; |
||||||
|
|
||||||
|
# Check if it's already configured. |
||||||
|
my $configured = $anvil->System->check_if_configured({debug => 2}); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { configured => $configured }}); |
||||||
|
|
||||||
|
if ($anvil->data->{switches}{'check-configured'}) |
||||||
|
{ |
||||||
|
if ($configured) |
||||||
|
{ |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0275"}); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0274"}); |
||||||
|
} |
||||||
|
return(0); |
||||||
|
} |
||||||
|
|
||||||
|
if ($configured) |
||||||
|
{ |
||||||
|
if ($anvil->data->{switches}{'mark-configured'}) |
||||||
|
{ |
||||||
|
# Nothing to do. |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0271"}); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
# Mark UN configures. |
||||||
|
my $variable_uuid = $anvil->Database->insert_or_update_variables({ |
||||||
|
variable_name => "system::configured", |
||||||
|
variable_value => 0, |
||||||
|
variable_default => "", |
||||||
|
variable_description => "striker_0048", |
||||||
|
variable_section => "system", |
||||||
|
variable_source_uuid => $anvil->Get->host_uuid, |
||||||
|
variable_source_table => "hosts", |
||||||
|
}); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { variable_uuid => $variable_uuid }}); |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0272"}); |
||||||
|
} |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
if ($anvil->data->{switches}{'mark-unconfigured'}) |
||||||
|
{ |
||||||
|
# Nothing to do. |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0273"}); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
# Mark configures. |
||||||
|
my $variable_uuid = $anvil->Database->insert_or_update_variables({ |
||||||
|
variable_name => "system::configured", |
||||||
|
variable_value => 1, |
||||||
|
variable_default => "", |
||||||
|
variable_description => "striker_0048", |
||||||
|
variable_section => "system", |
||||||
|
variable_source_uuid => $anvil->Get->host_uuid, |
||||||
|
variable_source_table => "hosts", |
||||||
|
}); |
||||||
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { variable_uuid => $variable_uuid }}); |
||||||
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 1, key => "message_0270"}); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
return(0); |
||||||
|
} |
||||||
Loading…
Reference in new issue