Tsu-ba-me
3 years ago
parent
c7f63f4a4b
commit
23d818cfff
3 changed files with 204 additions and 17 deletions
@ -0,0 +1,133 @@ |
|||||||
|
#!/usr/bin/perl |
||||||
|
# |
||||||
|
# Open an SSH tunnel using the Net::OpenSSH module and keep it opened with an infinite loop. |
||||||
|
# |
||||||
|
# Note: this is a temporary solution to avoid directly calling the SSH command. |
||||||
|
# |
||||||
|
|
||||||
|
use strict; |
||||||
|
use warnings; |
||||||
|
use Anvil::Tools; |
||||||
|
use Net::OpenSSH; |
||||||
|
|
||||||
|
$| = 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(); |
||||||
|
my $ssh_fh; |
||||||
|
|
||||||
|
$anvil->Log->level({ set => 2 }); |
||||||
|
|
||||||
|
sub start_ssh_tunnel |
||||||
|
{ |
||||||
|
my $parameters = shift; |
||||||
|
|
||||||
|
# Required parameters: |
||||||
|
my $remote_user = $parameters->{remote_user}; |
||||||
|
my $target = $parameters->{target}; |
||||||
|
my $forward_local_port = $parameters->{forward_local_port}; |
||||||
|
my $forward_remote_port = $parameters->{forward_remote_port}; |
||||||
|
|
||||||
|
if ((not defined $remote_user) |
||||||
|
or (not defined $target) |
||||||
|
or (not defined $forward_local_port) |
||||||
|
or (not defined $forward_remote_port)) |
||||||
|
{ |
||||||
|
return 1; |
||||||
|
} |
||||||
|
|
||||||
|
# Optional parameters: |
||||||
|
my $port = $parameters->{port} ? $parameters->{port} : 22; |
||||||
|
|
||||||
|
my $ssh_fh_key = $remote_user."\@".$target.":".$port; |
||||||
|
my $query = " |
||||||
|
SELECT anv.anvil_password |
||||||
|
FROM hosts AS hos |
||||||
|
JOIN anvils AS anv |
||||||
|
ON hos.host_uuid = anv.anvil_node1_host_uuid |
||||||
|
OR hos.host_uuid = anv.anvil_node2_host_uuid |
||||||
|
OR hos.host_uuid = anv.anvil_dr1_host_uuid |
||||||
|
WHERE hos.host_name = ".$anvil->Database->quote($target)." |
||||||
|
;"; |
||||||
|
|
||||||
|
my $password = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ })->[0]->[0]; |
||||||
|
|
||||||
|
my ($output, $error, $return_code) = $anvil->Remote->call({ |
||||||
|
remote_user => $remote_user, |
||||||
|
target => $target, |
||||||
|
password => $password, |
||||||
|
shell_call => $anvil->data->{path}{exe}{echo}." 1", |
||||||
|
no_cache => 1, |
||||||
|
}); |
||||||
|
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||||
|
output => $output, |
||||||
|
error => $error, |
||||||
|
return_code => $return_code |
||||||
|
} }); |
||||||
|
|
||||||
|
if ($output eq "1") |
||||||
|
{ |
||||||
|
$ssh_fh = $anvil->data->{cache}{ssh_fh}{$ssh_fh_key}; |
||||||
|
|
||||||
|
delete $anvil->data->{cache}{ssh_fh}{$ssh_fh_key}; |
||||||
|
|
||||||
|
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||||
|
is_ssh_fh_defined => defined $ssh_fh ? 1 : 0 |
||||||
|
} }); |
||||||
|
} |
||||||
|
|
||||||
|
$ssh_fh->system({ ssh_opts => [ "-O", "forward", |
||||||
|
"-L".$forward_local_port.":localhost:".$forward_remote_port ] }); |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
sub handle_stop_signals |
||||||
|
{ |
||||||
|
if (defined $ssh_fh->disconnect) |
||||||
|
{ |
||||||
|
$ssh_fh->disconnect(); |
||||||
|
|
||||||
|
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||||
|
message => "SSH tunnel disconnected." |
||||||
|
} }); |
||||||
|
} |
||||||
|
|
||||||
|
$anvil->nice_exit({ exit_code => 0 }); |
||||||
|
} |
||||||
|
|
||||||
|
$SIG->{INT} = \&handle_stop_signals; |
||||||
|
$SIG->{TERM} = \&handle_stop_signals; |
||||||
|
|
||||||
|
$anvil->Get->switches; |
||||||
|
|
||||||
|
$anvil->Database->connect; |
||||||
|
$anvil->Log->entry({ source => $THIS_FILE, line => __LINE__, level => 2, secure => 0, key => "log_0132" }); |
||||||
|
if (not $anvil->data->{sys}{database}{connections}) |
||||||
|
{ |
||||||
|
# No databases, exit. |
||||||
|
$anvil->Log->entry({ source => $THIS_FILE, line => __LINE__, level => 0, 'print' => 1, priority => "err", key => "error_0003" }); |
||||||
|
$anvil->nice_exit({ exit_code => 1 }); |
||||||
|
} |
||||||
|
|
||||||
|
if (start_ssh_tunnel({ |
||||||
|
remote_user => $anvil->data->{switches}{'remote-user'}, |
||||||
|
target => $anvil->data->{switches}{'target'}, |
||||||
|
port => $anvil->data->{switches}{'port'}, |
||||||
|
forward_local_port => $anvil->data->{switches}{'forward-local-port'}, |
||||||
|
forward_remote_port => $anvil->data->{switches}{'forward-remote-port'} |
||||||
|
}) > 0) |
||||||
|
{ |
||||||
|
$anvil->nice_exit({ exit_code => 1 }); |
||||||
|
} |
||||||
|
|
||||||
|
while(1) |
||||||
|
{ |
||||||
|
sleep(1); |
||||||
|
} |
||||||
Loading…
Reference in new issue