You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
2.8 KiB
102 lines
2.8 KiB
6 years ago
|
#!/usr/bin/perl
|
||
|
|
#
|
||
|
|
# This set, clear and report if the host needs to be rebooted.
|
||
|
|
#
|
||
|
|
# Examples;
|
||
|
|
# - Enable - anvil-reboot-needed --set 1
|
||
|
|
# - Disable - anvil-reboot-needed --set 0
|
||
|
|
# - Report - anvil-reboot-needed
|
||
|
|
#
|
||
|
|
# Exit codes;
|
||
|
|
# 0 = Normal exit.
|
||
|
|
# 1 = No database connections available.
|
||
|
|
|
||
|
|
use strict;
|
||
|
|
use warnings;
|
||
|
|
use Anvil::Tools;
|
||
|
|
|
||
|
|
# Disable buffering
|
||
|
|
$| = 1;
|
||
|
|
|
||
|
|
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({log_level => 2, log_secure => 1});
|
||
|
|
|
||
|
|
$anvil->Storage->read_config({file => "/etc/anvil/anvil.conf"});
|
||
|
|
|
||
|
|
# Read switches
|
||
|
|
$anvil->data->{switches}{set} = "";
|
||
|
|
$anvil->Get->switches;
|
||
|
|
|
||
|
|
# Connect to DBs.
|
||
|
|
$anvil->Database->connect;
|
||
|
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 3, secure => 0, key => "log_0132"});
|
||
|
|
if (not $anvil->data->{sys}{database}{connections})
|
||
|
|
{
|
||
|
|
# No databases, exit.
|
||
|
|
print $anvil->Words->string({key => "error_0003"})."\n";
|
||
|
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, level => 0, secure => 0, key => "error_0003"});
|
||
|
|
$anvil->nice_exit({exit_code => 1});
|
||
|
|
}
|
||
|
|
|
||
|
|
my $reboot_needed = $anvil->System->reboot_needed({debug => 2});
|
||
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { reboot_needed => $reboot_needed }});
|
||
|
|
|
||
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { "switches::set" => $anvil->data->{switches}{set} }});
|
||
|
|
if ($anvil->data->{switches}{set} eq "1")
|
||
|
|
{
|
||
|
|
# Enable
|
||
|
|
if (not $reboot_needed)
|
||
|
|
{
|
||
|
|
$reboot_needed = $anvil->System->reboot_needed({debug => 2, set => 1});
|
||
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { reboot_needed => $reboot_needed }});
|
||
|
|
print $anvil->Words->string({key => "message_0048"})."\n";
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
# Was already set, do nothing
|
||
|
|
print $anvil->Words->string({key => "message_0049"})."\n";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
elsif ($anvil->data->{switches}{set} eq "0")
|
||
|
|
{
|
||
|
|
# Disabled
|
||
|
|
if ($reboot_needed)
|
||
|
|
{
|
||
|
|
$reboot_needed = $anvil->System->reboot_needed({debug => 2, set => 0});
|
||
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { reboot_needed => $reboot_needed }});
|
||
|
|
print $anvil->Words->string({key => "message_0050"})."\n";
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
# Was already disabled, do nothing
|
||
|
|
print $anvil->Words->string({key => "message_0051"})."\n";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
elsif ($anvil->data->{switches}{set})
|
||
|
|
{
|
||
|
|
# Bad call
|
||
|
|
print $anvil->Words->string({key => "message_0052", variables => { program => $THIS_FILE }})."\n";
|
||
|
|
}
|
||
|
|
|
||
|
|
# Get the current state
|
||
|
|
|
||
|
|
if ($reboot_needed)
|
||
|
|
{
|
||
|
|
# Report that we're in mainteance mode
|
||
|
|
print $anvil->Words->string({key => "message_0053"})."\n";
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
# Report that we're not.
|
||
|
|
print $anvil->Words->string({key => "message_0054"})."\n";
|
||
|
|
}
|
||
|
|
|
||
|
|
# We're done
|
||
|
|
$anvil->nice_exit({exit_code => 0});
|