Merge pull request #162 from Tsu-ba-me/issues/151-vnc-on-ws
Web UI: add endpoint to handle building pipes from client (in browser) to server VMs' VNC servermain
Digimer
3 years ago
committed by
GitHub
commit
18839f5b7b
8 changed files with 1459 additions and 0 deletions
@ -0,0 +1,154 @@ |
|||||||
|
#!/usr/bin/perl |
||||||
|
# |
||||||
|
# Manages VNC ports for server VMs that have VNC enabled. |
||||||
|
# |
||||||
|
|
||||||
|
use strict; |
||||||
|
use warnings; |
||||||
|
use Anvil::Tools; |
||||||
|
use Data::Dumper; |
||||||
|
use JSON; |
||||||
|
|
||||||
|
$| = 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(); |
||||||
|
|
||||||
|
sub get_vnc_pipe_info |
||||||
|
{ |
||||||
|
my $parameters = shift; |
||||||
|
my $server_uuid = $parameters->{server_uuid}; |
||||||
|
my $host_uuid = $parameters->{host_uuid}; |
||||||
|
|
||||||
|
my $vnc_pipe_info = { protocol => "ws" }; |
||||||
|
|
||||||
|
my $query = " |
||||||
|
SELECT |
||||||
|
ssh_tunnel_forward_port |
||||||
|
FROM |
||||||
|
public.vnc_pipes |
||||||
|
WHERE |
||||||
|
server_uuid = ".$anvil->Database->quote($server_uuid)." |
||||||
|
AND |
||||||
|
ssh_tunnel_host_uuid = ".$anvil->Database->quote($host_uuid)." |
||||||
|
;"; |
||||||
|
|
||||||
|
my $forward_port = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ })->[0]->[0]; |
||||||
|
|
||||||
|
$vnc_pipe_info->{forward_port} = $forward_port; |
||||||
|
|
||||||
|
return $vnc_pipe_info; |
||||||
|
} |
||||||
|
|
||||||
|
sub is_job_incomplete |
||||||
|
{ |
||||||
|
my $parameters = shift; |
||||||
|
my $job_uuid = $parameters->{job_uuid}; |
||||||
|
|
||||||
|
my $query = " |
||||||
|
SELECT |
||||||
|
job_progress |
||||||
|
FROM |
||||||
|
public.jobs |
||||||
|
WHERE |
||||||
|
job_uuid = ".$anvil->Database->quote($job_uuid)." |
||||||
|
;"; |
||||||
|
|
||||||
|
my $job_progress = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ })->[0]->[0]; |
||||||
|
|
||||||
|
return $job_progress == 100 ? 0 : 1; |
||||||
|
} |
||||||
|
|
||||||
|
$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 }); |
||||||
|
} |
||||||
|
|
||||||
|
my $cookie_problem = $anvil->Account->read_cookies(); |
||||||
|
|
||||||
|
# Don't do anything data-related if the user is not logged in. |
||||||
|
if ($cookie_problem) |
||||||
|
{ |
||||||
|
$anvil->Log->entry({ source => $THIS_FILE, line => __LINE__, level => 0, 'print' => 1, priority => "err", key => "error_0307" }); |
||||||
|
$anvil->nice_exit({ exit_code => 1 }); |
||||||
|
} |
||||||
|
|
||||||
|
# Read in any CGI variables, if needed. |
||||||
|
$anvil->Get->cgi(); |
||||||
|
|
||||||
|
$anvil->Database->get_hosts(); |
||||||
|
$anvil->Database->get_anvils(); |
||||||
|
|
||||||
|
print $anvil->Template->get({ file => "shared.html", name => "json_headers", show_name => 0 })."\n"; |
||||||
|
|
||||||
|
my $response_body = {}; |
||||||
|
my $request_body; |
||||||
|
|
||||||
|
if (defined $anvil->data->{cgi}{PUTDATA}{value}) |
||||||
|
{ |
||||||
|
my $is_decode_json_success = eval { |
||||||
|
$request_body = decode_json($anvil->data->{cgi}{PUTDATA}{value}); |
||||||
|
}; |
||||||
|
|
||||||
|
if (not $is_decode_json_success) |
||||||
|
{ |
||||||
|
$anvil->Log->entry({ |
||||||
|
source => $THIS_FILE, |
||||||
|
line => __LINE__, |
||||||
|
level => 0, |
||||||
|
'print' => 1, |
||||||
|
priority => "err", |
||||||
|
key => "error_0304", |
||||||
|
variables => { request_body_string => $anvil->data->{cgi}{PUTDATA}{value}, json_decode_error => $@ } |
||||||
|
}); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
my $server_uuid = exists $request_body->{server_uuid} ? $request_body->{server_uuid} : $anvil->data->{switches}{'server-uuid'}; |
||||||
|
my $is_open = exists $request_body->{is_open} ? $request_body->{is_open} : $anvil->data->{switches}{'is-open'}; |
||||||
|
|
||||||
|
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { |
||||||
|
server_uuid => $server_uuid, |
||||||
|
is_open => $is_open |
||||||
|
} }); |
||||||
|
|
||||||
|
if ($server_uuid) |
||||||
|
{ |
||||||
|
my $host_uuid = $anvil->Get->host_uuid(); |
||||||
|
my $operation_string = defined $is_open ? "open" : "close"; |
||||||
|
my $open_string = defined $is_open ? $is_open : ""; |
||||||
|
my $job_uuid = $anvil->Database->insert_or_update_jobs({ |
||||||
|
job_command => $anvil->data->{path}{exe}{'striker-manage-vnc-pipes'}, |
||||||
|
job_data => "server-uuid=".$server_uuid."\nopen=".$open_string, |
||||||
|
job_host_uuid => $host_uuid, |
||||||
|
job_description => "job_0351,!!operation!".$operation_string."!!,!!server_uuid!".$server_uuid."!!,!!host_uuid!".$host_uuid."!!", |
||||||
|
job_name => "cgi-bin::manage_vnc_pipes::".$server_uuid."::".$operation_string, |
||||||
|
job_progress => 0, |
||||||
|
job_title => "job_0350" |
||||||
|
}); |
||||||
|
|
||||||
|
# Wait until the job is complete before continuing. |
||||||
|
while(is_job_incomplete({ job_uuid => $job_uuid })) |
||||||
|
{ |
||||||
|
sleep(2); |
||||||
|
} |
||||||
|
|
||||||
|
if ($is_open) |
||||||
|
{ |
||||||
|
$response_body = get_vnc_pipe_info({ server_uuid => $server_uuid, host_uuid => $host_uuid }); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
print JSON->new->utf8->encode($response_body)."\n"; |
||||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,137 @@ |
|||||||
|
#!/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; |
||||||
|
|
||||||
|
sub open_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", |
||||||
|
"-L0.0.0.0:".$forward_local_port.":0.0.0.0:".$forward_remote_port ] }); |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
sub close_ssh_tunnel |
||||||
|
{ |
||||||
|
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} = \&close_ssh_tunnel; |
||||||
|
$SIG->{TERM} = \&close_ssh_tunnel; |
||||||
|
|
||||||
|
$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 (open_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 }); |
||||||
|
} |
||||||
|
|
||||||
|
my $is_ssh_tunnel_alive = 1; |
||||||
|
|
||||||
|
while($is_ssh_tunnel_alive) |
||||||
|
{ |
||||||
|
$is_ssh_tunnel_alive = $ssh_fh->test('echo'); |
||||||
|
|
||||||
|
sleep(60); |
||||||
|
} |
||||||
|
|
||||||
|
close_ssh_tunnel(); |
||||||
Loading…
Reference in new issue