* Fixed bugs in tools/anvil-manage-striker-peers, basic testing done.

Signed-off-by: Digimer <digimer@alteeve.ca>
main
Digimer 6 years ago
parent f9ce6e1154
commit 1106d36e03
  1. 3
      share/words.xml
  2. 96
      tools/anvil-manage-striker-peers

@ -57,7 +57,7 @@ NOTE: You must update the password of any other system using this host's
</key>
<key name="message_0030">Failed to write the new password to the temporary file: [#!variable!file!#]. Please check the logs for details.</key>
<key name="message_0031">Beginning configuration of local system.</key>
<key name="message_0032"></key>
<key name="message_0032">Peer: [#!variable!peer!#], database: [#!variable!name!#]</key>
<key name="message_0033"></key>
<key name="message_0034"><![CDATA[Logged in as: [<span class="fixed_width">#!data!sys::users::user_name!#</span>]]]></key>
<key name="message_0035"></key>
@ -288,6 +288,7 @@ The database connection error was:
<key name="log_0187">UUID cache file: [#!data!path::data::host_uuid!#] doesn't exists and we're not running as root. Unable to proceed.</key>
<key name="log_0188">Database archive check skipped, not running as root.</key>
<key name="log_0189">Database archiving is disabled, skipping archive checks.</key>
<key name="log_0190">Peer: [#!variable!peer!#], database: [#!variable!name!#], password: [#!variable!password!#], host UUID: [#!variable!uuid!#]</key>
<!-- Test words. Do NOT change unless you update 't/Words.t' or tests will needlessly fail. -->
<key name="t_0000">Test</key>

@ -28,6 +28,7 @@ $| = 1;
my $anvil = Anvil::Tools->new({log_level => 2, log_secure => 1});
# Read switches
$anvil->data->{switches}{list} = "";
$anvil->Get->switches;
# Make sure we're running as 'root'
@ -44,11 +45,52 @@ $anvil->Storage->read_config({file => $anvil->data->{path}{configs}{'anvil.conf'
# Read in the anvil.conf, we're going to need it in any case.
$anvil->data->{body}{'anvil.conf'} = $anvil->Storage->read_file({file => $anvil->data->{path}{configs}{'anvil.conf'}});
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => 1, level => 2, list => { anvil_conf_body => $anvil_conf_body }});
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => 1, level => 3, list => { "body::anvil.conf" => $anvil->data->{body}{'anvil.conf'} }});
# If we don't find the entry, or if the entry exists but has changed, this will be set to '1' so we'll
# rewrite the file.
$anvil->data->{config}{rewrite} = 0;
# Am I adding, editing or deleting?
process_entry($anvil);
process_entry($anvil) if not $anvil->data->{switches}{list};
# Re-read the (new) config
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { "config::rewrite" => $anvil->data->{config}{rewrite} }});
if ($anvil->data->{config}{rewrite})
{
$anvil->Storage->read_config({file => $anvil->data->{path}{configs}{'anvil.conf'}});
}
### Report the peers.
# First sort by host name/ip
foreach my $uuid (keys %{$anvil->data->{database}})
{
my $host = $anvil->data->{database}{$uuid}{host};
$anvil->data->{sorted}{db}{$host} = $uuid;
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => 0, level => 2, list => {
host => $host,
"sorted::db::${host}" => $anvil->data->{sorted}{db}{$host},
}});
}
foreach my $host (sort {$a cmp $b} keys %{$anvil->data->{sorted}{db}})
{
my $uuid = $anvil->data->{sorted}{db}{$host};
my $port = $anvil->data->{database}{$uuid}{port} ? $anvil->data->{database}{$uuid}{port} : 5432;
my $name = $anvil->data->{database}{$uuid}{name} ? $anvil->data->{database}{$uuid}{name} : $anvil->data->{sys}{database}{name};
my $user = $anvil->data->{database}{$uuid}{user} ? $anvil->data->{database}{$uuid}{user} : $anvil->data->{sys}{database}{user};
my $password = $anvil->data->{database}{$uuid}{password} ? $anvil->data->{database}{$uuid}{password} : "";
print $anvil->Words->string({key => "message_0032", variables => {
peer => $user."\@".$host.":".$port,
name => $name,
}})."\n";
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 1, secure => 0, key => "log_0190", variables => {
peer => $user."\@".$host.":".$port,
name => $name,
password => $anvil->Log->secure ? $password : $anvil->Words->string({key => "log_0186"}),
uuid => $uuid,
}});
}
$anvil->nice_exit({code => 0});
@ -101,10 +143,6 @@ sub process_entry
my $password = $anvil->Storage->read_file({file => $password_file});
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => 1, level => 2, list => { password => $password }});
# If we don't find the entry, or if the entry exists but has changed, this will be set to '1' so
# we'll rewrite the file.
my $write = 0;
# If the config already exists, we'll look at each of the values to see if any changed (or are not defaults). If so, we'll rewrite
my $host_variable = "database::${host_uuid}::host";
my $host_different = 1;
@ -120,7 +158,7 @@ sub process_entry
my $insert = $host_variable." = ".$host."\n";
$insert .= $port_variable." = ".$port."\n";
$insert .= $password_variable." = ".$password."\n";
$insert .= $ping_variable." = ".$ping."\n\n";
$insert .= $ping_variable." = ".$ping."\n";
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => 1, level => 2, list => { insert => $insert }});
# Loop through the existing file.
@ -130,7 +168,7 @@ sub process_entry
foreach my $line (split/\n/, $anvil->data->{body}{'anvil.conf'})
{
# If I removed an entry, I also want to delete the white space after it.
if (($just_deleted) && ((not $line) or ($line eq =~ /^\s+$/))
if (($just_deleted) && ((not $line) or ($line =~ /^\s+$/)))
{
$just_deleted = 0;
next;
@ -164,10 +202,10 @@ sub process_entry
if ($anvil->data->{switches}{remove})
{
$just_deleted = 1;
$write = 1;
$anvil->data->{config}{rewrite} = 1;
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
just_deleted => $just_deleted.
'write' => $write,
"config::rewrite" => $anvil->data->{config}{rewrite},
}});
next;
}
@ -180,10 +218,10 @@ sub process_entry
else
{
$line = $variable.$left_space."=".$right_space.$host;
$write = 1;
$anvil->data->{config}{rewrite} = 1;
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
line => $line,
'write' => $write,
"config::rewrite" => $anvil->data->{config}{rewrite},
}});
}
}
@ -198,10 +236,10 @@ sub process_entry
if ($anvil->data->{switches}{remove})
{
$just_deleted = 1;
$write = 1;
$anvil->data->{config}{rewrite} = 1;
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
just_deleted => $just_deleted.
'write' => $write,
"config::rewrite" => $anvil->data->{config}{rewrite},
}});
next;
}
@ -214,10 +252,10 @@ sub process_entry
else
{
$line = $variable.$left_space."=".$right_space.$port;
$write = 1;
$anvil->data->{config}{rewrite} = 1;
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
line => $line,
'write' => $write,
"config::rewrite" => $anvil->data->{config}{rewrite},
}});
}
}
@ -236,10 +274,10 @@ sub process_entry
if ($anvil->data->{switches}{remove})
{
$just_deleted = 1;
$write = 1;
$anvil->data->{config}{rewrite} = 1;
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
just_deleted => $just_deleted.
'write' => $write,
"config::rewrite" => $anvil->data->{config}{rewrite},
}});
next;
}
@ -252,10 +290,10 @@ sub process_entry
else
{
$line = $variable.$left_space."=".$right_space.$password;
$write = 1;
$anvil->data->{config}{rewrite} = 1;
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
line => $anvil->Log->secure ? $line : $anvil->Words->string({key => "log_0186"}),
'write' => $write,
"config::rewrite" => $anvil->data->{config}{rewrite},
}});
}
}
@ -270,10 +308,10 @@ sub process_entry
if ($anvil->data->{switches}{remove})
{
$just_deleted = 1;
$write = 1;
$anvil->data->{config}{rewrite} = 1;
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
just_deleted => $just_deleted.
'write' => $write,
"config::rewrite" => $anvil->data->{config}{rewrite},
}});
next;
}
@ -286,10 +324,10 @@ sub process_entry
else
{
$line = $variable.$left_space."=".$right_space.$ping;
$write = 1;
$anvil->data->{config}{rewrite} = 1;
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
line => $line,
'write' => $write,
"config::rewrite" => $anvil->data->{config}{rewrite},
}});
}
}
@ -299,19 +337,19 @@ sub process_entry
# If I've not seen this DB, enter it.
if ((not $peer_seen) && ($anvil->data->{switches}{add}))
{
$new_body .= $insert;
$write = 1;
$new_body .= $insert."\n";
$anvil->data->{config}{rewrite} = 1;
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, secure => 1, level => 2, list => {
new_body => $new_body,
'write' => $write,
"config::rewrite" => $anvil->data->{config}{rewrite},
}});
}
}
$new_body .= $line."\n";
}
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { 'write' => $write }});
if ($write)
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { "config::rewrite" => $anvil->data->{config}{rewrite} }});
if ($anvil->data->{config}{rewrite})
{
# Backup the original
my $backup_file = $anvil->Storage->backup({secure => 1, file => $anvil->data->{path}{configs}{'anvil.conf'}});

Loading…
Cancel
Save