seasharp
/
anvil
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore(tools): hide open all components in manage vnc pipes

Browse Source
main
Tsu-ba-me 2 years ago
parent d9d0244f3f
commit d64e5ff17f
1 changed files with 244 additions and 246 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 490
      tools/striker-manage-vnc-pipes

490
tools/striker-manage-vnc-pipes
Unescape View File

@ -88,6 +88,8 @@ WHERE server_uuid = ".$anvil->Database->quote($server_uuid)."
my $results = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ }); my $results = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ });
my $count = @{$results}; my $count = @{$results};
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { query => $query, count => $count }, prefix => "get_server_info" });
if ($count == 1) if ($count == 1)
{ {
my $row = $results->[0]; my $row = $results->[0];
@ -205,34 +207,46 @@ sub is_websockify_exists
my $ws_host_uuid = $parameters->{ws_host_uuid}; my $ws_host_uuid = $parameters->{ws_host_uuid};
my $server_vnc_port = $parameters->{server_vnc_port}; my $server_vnc_port = $parameters->{server_vnc_port};
# Expect 0 to 1 record(s) because each VM server can only have 1
# websockify instance (not considering shared VNC sessions).
my $query = " my $query = "
SELECT SELECT
vnc.server_vnc_port, hos.host_name, vnc.ws_host_uuid, vnc.ws_pid, vnc.ws_source_port, vnc.ssh_tunnel_host_uuid a.server_vnc_port,
FROM b.host_name,
public.vnc_pipes AS vnc a.ws_host_uuid,
JOIN a.ws_pid,
public.hosts AS hos a.ws_dest_port,
ON a.ssh_tunnel_host_uuid
vnc.ws_host_uuid = hos.host_uuid FROM vnc_pipes AS a
JOIN hosts AS b ON a.ws_host_uuid = b.host_uuid
WHERE WHERE
vnc.server_uuid = ".$anvil->Database->quote($server_uuid)." a.server_uuid = ".$anvil->Database->quote($server_uuid)."
AND
a.ws_host_uuid = ".$anvil->Database->quote($ws_host_uuid)."
AND AND
vnc.ws_host_uuid = ".$anvil->Database->quote($ws_host_uuid)." a.server_vnc_port IS NOT NULL
ORDER BY a.ws_pid DESC
;"; ;";
my $results = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ }); my $results = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ });
my $count = @{$results};
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { query => $query, count => $count }, prefix => "is_websockify_exists" });
my $ws_exists_info = { exists_code => 0 }; my $ws_exists_info = { exists_code => 0 };
foreach my $row (@{$results}) foreach my $row (@{$results})
{ {
my $server_vnc_port_in_record = $row->[0] // 0; my $server_vnc_port_in_record = $row->[0];
my $ws_host_name = $row->[1]; my $ws_host_name = $row->[1];
my $ws_host_uuid_in_record = $row->[2] // ''; my $ws_host_uuid_in_record = $row->[2];
my $ws_pid = $row->[3]; my $ws_pid = $row->[3];
my $ws_source_port = $row->[4]; my $ws_dest_port = $row->[4];
my $ssh_tunnel_host_uuid = $row->[5]; my $ssh_tunnel_host_uuid = $row->[5];
my $clean_up_parameters = { host_name => $ws_host_name, ws_pid => $ws_pid }; my $clean_up_parameters = { host_name => $ws_host_name, ws_pid => $ws_pid };
$ws_exists_info->{exists_code} = 1;
if ($ws_host_uuid ne $ws_host_uuid_in_record) if ($ws_host_uuid ne $ws_host_uuid_in_record)
{ {
# VNC server host mismatch; try to stop the recorded instance. # VNC server host mismatch; try to stop the recorded instance.
@ -242,32 +256,25 @@ AND
# No need to preserve the websockify source port in # No need to preserve the websockify source port in
# this case because the tunnel will need to be replaced # this case because the tunnel will need to be replaced
# as well. # as well.
$ws_exists_info->{exists_code} = 1;
} }
elsif ($server_vnc_port != $server_vnc_port_in_record) elsif ($server_vnc_port != $server_vnc_port_in_record)
{ {
# VNC server port mismatch; try to stop the recorded instance. # VNC server port mismatch; try to stop the recorded instance.
stop_websockify($clean_up_parameters); stop_websockify($clean_up_parameters);
$ws_exists_info->{ws_source_port} = $ws_source_port; $ws_exists_info->{ws_dest_port} = $ws_dest_port;
$ws_exists_info->{exists_code} = 1;
} }
elsif (not is_websockify_process($clean_up_parameters)) elsif (not is_websockify_process($clean_up_parameters))
{ {
# The recorded instance died. # The recorded instance died.
$ws_exists_info->{ws_source_port} = $ws_source_port; $ws_exists_info->{ws_dest_port} = $ws_dest_port;
$ws_exists_info->{exists_code} = 1;
} }
else else
{ {
# Found one websockify instance that passed all tests. # Found one websockify instance that passed all tests.
$ws_exists_info->{ws_pid} = $ws_pid; $ws_exists_info->{exists_code} = 2;
$ws_exists_info->{ws_source_port} = $ws_source_port; $ws_exists_info->{ws_pid} = $ws_pid;
# Code 2: the websockify instance is not recorded for the pipe made from the current host, do update only. $ws_exists_info->{ws_dest_port} = $ws_dest_port;
# Code 3: the websockify instance is recorded for the pipe made from the current host, do nothing.
$ws_exists_info->{exists_code} = $host_uuid eq $ssh_tunnel_host_uuid ? 3 : 2;
# Don't continue the loop because all pipes should align to this verified instance.
last;
} }
} }
@ -282,66 +289,88 @@ sub is_ssh_tunnel_exists
my $server_uuid = $parameters->{server_uuid}; my $server_uuid = $parameters->{server_uuid};
my $ssh_tunnel_host_uuid = $parameters->{ssh_tunnel_host_uuid}; my $ssh_tunnel_host_uuid = $parameters->{ssh_tunnel_host_uuid};
my $ws_host_uuid = $parameters->{ws_host_uuid}; my $ws_host_uuid = $parameters->{ws_host_uuid};
my $ws_source_port = $parameters->{ws_source_port}; my $ws_dest_port = $parameters->{ws_dest_port};
# Expect 0 to n record(s), where n is the number of subnodes:
# n=0: no SSH tunnel connected to the server's ws instance
# n>0: at least one SSH tunnel is known, regardless of whether it is
# actually alive
#
# Order by clause ensures NULL record(s) are at first.
my $query = " my $query = "
SELECT SELECT
ws_host_uuid, ws_source_port, ssh_tunnel_pid, ssh_tunnel_forward_port a.ws_host_uuid,
FROM a.ws_pid,
public.vnc_pipes a.ws_dest_port,
a.ssh_tunnel_dest_port,
b.host_name,
a.ssh_tunnel_host_uuid,
a.ssh_tunnel_pid,
a.ssh_tunnel_source_port
FROM vnc_pipes AS a
JOIN hosts AS b ON a.ssh_tunnel_host_uuid = b.host_uuid
WHERE WHERE
server_uuid = ".$anvil->Database->quote($server_uuid)." a.server_uuid = ".$anvil->Database->quote($server_uuid)."
AND AND
ssh_tunnel_host_uuid = ".$anvil->Database->quote($ssh_tunnel_host_uuid)." a.ssh_tunnel_host_uuid IS NOT NULL
ORDER BY a.ws_pid DESC
;"; ;";
my $results = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ }); my $results = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ });
my $count = @{$results}; my $count = @{$results};
my $ssh_tunnel_exists_info = { exists_code => 0 };
if ($count == 1) $anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { query => $query, count => $count }, prefix => "is_ssh_tunnel_exists" });
{
my $row = $results->[0];
my $ws_host_uuid_in_record = $row->[0] // '';
my $ws_source_port_in_record = $row->[1] // 0;
my $ssh_tunnel_pid = $row->[2];
my $ssh_tunnel_forward_port = $row->[3];
my $clean_up_parameters = { ssh_tunnel_pid => $ssh_tunnel_pid };
$ssh_tunnel_exists_info->{ssh_tunnel_pid} = $ssh_tunnel_pid; my $ssh_tunnel_exists_info = { exists_code => 0 };
$ssh_tunnel_exists_info->{ssh_tunnel_forward_port} = $ssh_tunnel_forward_port;
$ssh_tunnel_exists_info->{exists_code} = 1;
if ($ws_host_uuid ne $ws_host_uuid_in_record) foreach my $row (@{$results})
{
my $row = $results->[0];
my $ws_host_uuid_in_record = $row->[0];
my $ws_pid_in_record = $row->[1];
my $ws_dest_port_in_record = $row->[2];
my $ssh_tunnel_dest_port = $row->[3];
my $ssh_tunnel_host_name = $row->[4];
my $ssh_tunnel_host_uuid_in_record = $row->[5];
my $ssh_tunnel_pid = $row->[6];
my $ssh_tunnel_source_port = $row->[7];
my $clean_up_parameters = { host_name => $ssh_tunnel_host_name, ssh_tunnel_pid => $ssh_tunnel_pid };
$ssh_tunnel_exists_info->{exists_code} = 1;
if (not defined $ws_pid_in_record and defined $ssh_tunnel_pid)
{ {
# Websockify host mismatch; try to stop the recorded instance. # Websockify instance doesn't exist but tunnel does.
# Likely happens after a server migration. # Remove the tunnel that's not connected to anything.
stop_ssh_tunnel($clean_up_parameters); stop_ssh_tunnel($clean_up_parameters);
# No need to preserve the SSH tunnel forward port in clear_vnc_pipe_st({ server_uuid => $server_uuid, ws_host_uuid => $ws_host_uuid_in_record });
# this case because the websockify instance will need
# to be replaced as well.
delete $ssh_tunnel_exists_info->{ssh_tunnel_forward_port};
return $ssh_tunnel_exists_info; # No need to preserve any tunnel info because this
# tunnel doesn't need to be recreated.
} }
elsif ($ssh_tunnel_source_port != $ws_dest_port_in_record)
if ($ws_source_port != $ws_source_port_in_record)
{ {
# Websockify source port mismatch; try to stop the recorded instance. # Tunnel is not pointing to the destination port of the
# websockify instance; stop it and recreate.
stop_ssh_tunnel($clean_up_parameters); stop_ssh_tunnel($clean_up_parameters);
return $ssh_tunnel_exists_info; # Port mismatch means we need to recreate the tunnel.
$ssh_tunnel_exists_info->{ssh_tunnel_dest_port} = $ssh_tunnel_dest_port;
} }
elsif (not is_ssh_process($clean_up_parameters))
if (not is_ssh_process($clean_up_parameters))
{ {
# The recorded tunnel died. # The tunnel pid doesn't point to an active tunnel;
return $ssh_tunnel_exists_info; # mark as need-to-be recreated.
$ssh_tunnel_exists_info->{ssh_tunnel_dest_port} = $ssh_tunnel_dest_port;
}
else
{
# Passed all tests; tunnel considered exists.
$ssh_tunnel_exists_info->{exists_code} = 2;
$ssh_tunnel_exists_info->{ssh_tunnel_dest_port} = $ssh_tunnel_dest_port;
$ssh_tunnel_exists_info->{ssh_tunnel_pid} = $ssh_tunnel_pid;
} }
# Passed all tests; tunnel considered exists.
$ssh_tunnel_exists_info->{exists_code} = 2;
} }
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => $ssh_tunnel_exists_info, prefix => "is_ssh_tunnel_exists" }); $anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => $ssh_tunnel_exists_info, prefix => "is_ssh_tunnel_exists" });
@ -388,24 +417,19 @@ sub start_websockify
server_vnc_port => $target_port server_vnc_port => $target_port
}); });
if ($ws_exists_info->{exists_code} =~ /^[23]$/) if ($ws_exists_info->{exists_code} == 2)
{ {
$ws_info = {}; $ws_info = {};
$ws_info->{pid} = $ws_exists_info->{ws_pid}; $ws_info->{pid} = $ws_exists_info->{ws_pid};
$ws_info->{source_port} = $ws_exists_info->{ws_source_port}; $ws_info->{source_port} = $ws_exists_info->{ws_dest_port};
if ($ws_exists_info->{exists_code} == 2)
{
$ws_info->{is_update} = 1;
}
} }
else else
{ {
if (not defined $source_port) if (not defined $source_port)
{ {
if (defined $ws_exists_info->{ws_source_port}) if (defined $ws_exists_info->{ws_dest_port})
{ {
$source_port = $ws_exists_info->{ws_source_port}; $source_port = $ws_exists_info->{ws_dest_port};
} }
else else
{ {
@ -467,58 +491,58 @@ sub stop_websockify
sub start_ssh_tunnel sub start_ssh_tunnel
{ {
my $parameters = shift; my $parameters = shift;
my $server_uuid = $parameters->{server_uuid}; my $server_uuid = $parameters->{server_uuid};
my $host_name = $parameters->{host_name}; my $host_name = $parameters->{host_name};
my $host_uuid = $parameters->{host_uuid}; my $host_uuid = $parameters->{host_uuid};
my $ws_host_name = $parameters->{ws_host_name}; my $ws_host_name = $parameters->{ws_host_name};
my $ws_host_uuid = $parameters->{ws_host_uuid}; my $ws_host_uuid = $parameters->{ws_host_uuid};
my $ws_source_port = $parameters->{ws_source_port}; my $ws_dest_port = $parameters->{ws_dest_port};
my $ssh_tunnel_forward_port = $parameters->{ssh_tunnel_forward_port}; my $ssh_tunnel_dest_port = $parameters->{ssh_tunnel_dest_port};
my $ssh_tunnel_info; my $ssh_tunnel_info;
my $ssh_tunnel_exists_info = is_ssh_tunnel_exists({ my $ssh_tunnel_exists_info = is_ssh_tunnel_exists({
server_uuid => $server_uuid, server_uuid => $server_uuid,
ssh_tunnel_host_uuid => $host_uuid, ssh_tunnel_host_uuid => $host_uuid,
ws_host_uuid => $ws_host_uuid, ws_host_uuid => $ws_host_uuid,
ws_source_port => $ws_source_port ws_dest_port => $ws_dest_port
}); });
if ($ssh_tunnel_exists_info->{exists_code} == 2) if ($ssh_tunnel_exists_info->{exists_code} == 2)
{ {
$ssh_tunnel_info = {}; $ssh_tunnel_info = {};
$ssh_tunnel_info->{pid} = $ssh_tunnel_exists_info->{ssh_tunnel_pid}; $ssh_tunnel_info->{pid} = $ssh_tunnel_exists_info->{ssh_tunnel_pid};
$ssh_tunnel_info->{forward_port} = $ssh_tunnel_exists_info->{ssh_tunnel_forward_port}; $ssh_tunnel_info->{forward_port} = $ssh_tunnel_exists_info->{ssh_tunnel_dest_port};
} }
else else
{ {
if (not defined $ssh_tunnel_forward_port) if (not defined $ssh_tunnel_dest_port)
{ {
if (defined $ssh_tunnel_exists_info->{ssh_tunnel_forward_port}) if (defined $ssh_tunnel_exists_info->{ssh_tunnel_dest_port})
{ {
$ssh_tunnel_forward_port = $ssh_tunnel_exists_info->{ssh_tunnel_forward_port}; $ssh_tunnel_dest_port = $ssh_tunnel_exists_info->{ssh_tunnel_dest_port};
} }
else else
{ {
$ssh_tunnel_forward_port = $ws_source_port; $ssh_tunnel_dest_port = $ws_dest_port;
} }
} }
$ssh_tunnel_forward_port = get_available_port({ start_port => $ssh_tunnel_forward_port }); $ssh_tunnel_dest_port = get_available_port({ start_port => $ssh_tunnel_dest_port });
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { $anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => {
ssh_tunnel_forward_port => $ssh_tunnel_forward_port ssh_tunnel_dest_port => $ssh_tunnel_dest_port
}, prefix => "start_ssh_tunnel" }); }, prefix => "start_ssh_tunnel" });
if (not defined $ssh_tunnel_forward_port) if (not defined $ssh_tunnel_dest_port)
{ {
return; return;
} }
my $shell_call = $anvil->data->{path}{exe}{'striker-open-ssh-tunnel'} my $shell_call = $anvil->data->{path}{exe}{'striker-open-ssh-tunnel'}
." --remote-user admin --target ".$ws_host_name ." --remote-user admin --target ".$ws_host_name
." --forward-local-port ".$ssh_tunnel_forward_port ." --forward-local-port ".$ssh_tunnel_dest_port
." --forward-remote-port ".$ws_source_port ." --forward-remote-port ".$ws_dest_port
." &>/dev/null & echo pid:\$!"; ." &>/dev/null & echo pid:\$!";
my ($shell_output, $shell_error, $shell_return_code) = call({ host_name => $host_name, shell_call => $shell_call }); my ($shell_output, $shell_error, $shell_return_code) = call({ host_name => $host_name, shell_call => $shell_call });
@ -529,7 +553,7 @@ sub start_ssh_tunnel
$ssh_tunnel_info = {}; $ssh_tunnel_info = {};
$ssh_tunnel_info->{pid} = $ssh_tunnel_pid; $ssh_tunnel_info->{pid} = $ssh_tunnel_pid;
$ssh_tunnel_info->{forward_port} = $ssh_tunnel_forward_port; $ssh_tunnel_info->{forward_port} = $ssh_tunnel_dest_port;
if ($ssh_tunnel_exists_info->{exists_code} == 1) if ($ssh_tunnel_exists_info->{exists_code} == 1)
{ {
@ -567,10 +591,11 @@ CREATE TABLE IF NOT EXISTS public.vnc_pipes (
server_vnc_port numeric, server_vnc_port numeric,
ws_host_uuid uuid, ws_host_uuid uuid,
ws_pid numeric, ws_pid numeric,
ws_source_port numeric, ws_dest_port numeric,
ssh_tunnel_dest_port numeric,
ssh_tunnel_host_uuid uuid, ssh_tunnel_host_uuid uuid,
ssh_tunnel_pid numeric, ssh_tunnel_pid numeric,
ssh_tunnel_forward_port numeric, ssh_tunnel_source_port numeric,
modified_date timestamp with time zone not null, modified_date timestamp with time zone not null,
unique(server_uuid, ws_host_uuid) unique(server_uuid, ws_host_uuid)
);"; );";
@ -592,14 +617,15 @@ sub insert_or_update_vnc_pipe
return (1) if (not defined $server_uuid); return (1) if (not defined $server_uuid);
my $server_vnc_port = $parameters->{server_vnc_port}; my $server_vnc_port = $parameters->{server_vnc_port};
my $ssh_tunnel_forward_port = $parameters->{ssh_tunnel_forward_port}; my $ssh_tunnel_dest_port = $parameters->{ssh_tunnel_dest_port};
my $ssh_tunnel_host_uuid = $parameters->{ssh_tunnel_host_uuid}; my $ssh_tunnel_host_uuid = $parameters->{ssh_tunnel_host_uuid};
my $ssh_tunnel_pid = $parameters->{ssh_tunnel_pid}; my $ssh_tunnel_pid = $parameters->{ssh_tunnel_pid};
my $vnc_pipe_uuid = $parameters->{vnc_pipe_uuid} // $anvil->Get->uuid(); my $ssh_tunnel_source_port = $parameters->{ssh_tunnel_source_port};
my $ws_host_uuid = $parameters->{ws_host_uuid}; my $vnc_pipe_uuid = $parameters->{vnc_pipe_uuid} // $anvil->Get->uuid();
my $ws_pid = $parameters->{ws_pid}; my $ws_host_uuid = $parameters->{ws_host_uuid};
my $ws_source_port = $parameters->{ws_source_port}; my $ws_pid = $parameters->{ws_pid};
my $ws_dest_port = $parameters->{ws_dest_port};
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => $parameters, prefix => "insert_or_update_vnc_pipe" }); $anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => $parameters, prefix => "insert_or_update_vnc_pipe" });
@ -614,10 +640,11 @@ sub insert_or_update_vnc_pipe
"server_vnc_port", "server_vnc_port",
"ws_host_uuid", "ws_host_uuid",
"ws_pid", "ws_pid",
"ws_source_port", "ws_dest_port",
"ssh_tunnel_dest_port",
"ssh_tunnel_host_uuid", "ssh_tunnel_host_uuid",
"ssh_tunnel_pid", "ssh_tunnel_pid",
"ssh_tunnel_forward_port" "ssh_tunnel_source_port"
) )
{ {
my $column_value = $parameters->{$column_name}; my $column_value = $parameters->{$column_name};
@ -649,6 +676,45 @@ INSERT INTO public.vnc_pipes (
modified_date = $quoted_vnc_pipe_mdate;"; modified_date = $quoted_vnc_pipe_mdate;";
$anvil->Database->write({ query => $query, source => $THIS_FILE, line => __LINE__ }); $anvil->Database->write({ query => $query, source => $THIS_FILE, line => __LINE__ });
return (0);
}
sub clear_vnc_pipe_ws
{
my $parameters = shift;
my $server_uuid = $parameters->{server_uuid};
my $ws_host_uuid = $parameters->{ws_host_uuid};
return (1) if (not defined $ws_host_uuid);
insert_or_update_vnc_pipe({
server_uuid => $server_uuid,
server_vnc_port => "NULL",
ws_dest_port => "NULL",
ws_host_uuid => $ws_host_uuid,
ws_pid => "NULL"
});
return (0);
}
sub clear_vnc_pipe_st
{
my $parameters = shift;
my $server_uuid = $parameters->{server_uuid};
my $ws_host_uuid = $parameters->{ws_host_uuid};
return (1) if (not defined $ws_host_uuid);
insert_or_update_vnc_pipe({
server_uuid => $server_uuid,
ssh_tunnel_dest_port => "NULL",
ssh_tunnel_host_uuid => "NULL",
ssh_tunnel_pid => "NULL",
ssh_tunnel_source_port => "NULL",
ws_host_uuid => $ws_host_uuid
});
} }
sub get_vnc_pipe sub get_vnc_pipe
@ -658,56 +724,62 @@ sub get_vnc_pipe
return if (not defined $server_uuid); return if (not defined $server_uuid);
my $is_ws_pid = $parameters->{is_ws_pid};
my $ssh_tunnel_host_uuid = $parameters->{ssh_tunnel_host_uuid}; my $ssh_tunnel_host_uuid = $parameters->{ssh_tunnel_host_uuid};
my $ws_host_uuid = $parameters->{ws_host_uuid}; my $ws_host_uuid = $parameters->{ws_host_uuid};
my $vnc_pipe_info; my $vnc_pipe_info;
my $cond_ws_pid = defined $is_ws_pid
? "AND a.ws_pid IS NOT NULL" : "";
my $cond_ssht_huuid = defined $ssh_tunnel_host_uuid my $cond_ssht_huuid = defined $ssh_tunnel_host_uuid
? "AND ssh_tunnel_host_uuid = ".$anvil->Database->quote($ssh_tunnel_host_uuid) : ""; ? "AND a.ssh_tunnel_host_uuid = ".$anvil->Database->quote($ssh_tunnel_host_uuid) : "";
my $cond_ws_huuid = defined $ws_host_uuid my $cond_ws_huuid = defined $ws_host_uuid
? "AND ws_host_uuid = ".$anvil->Database->quote($ws_host_uuid) : ""; ? "AND a.ws_host_uuid = ".$anvil->Database->quote($ws_host_uuid) : "";
my $query = " my $query = "
SELECT SELECT
vnc.server_vnc_port, a.server_vnc_port,
hos.host_name, b.host_name,
vnc.ws_host_uuid, a.ws_host_uuid,
vnc.ws_pid, a.ws_pid,
vnc.ws_source_port, a.ws_dest_port,
vnc.ssh_tunnel_host_uuid, a.ssh_tunnel_host_uuid,
vnc.ssh_tunnel_pid, a.ssh_tunnel_pid,
vnc.ssh_tunnel_forward_port a.ssh_tunnel_dest_port
FROM FROM
public.vnc_pipes AS vnc public.vnc_pipes AS a
JOIN JOIN
public.hosts AS hos public.hosts AS b
ON ON
vnc.ws_host_uuid = hos.host_uuid a.ws_host_uuid = b.host_uuid
WHERE WHERE
server_uuid = ".$anvil->Database->quote($server_uuid)." server_uuid = ".$anvil->Database->quote($server_uuid)."
$cond_ws_pid
$cond_ssht_huuid $cond_ssht_huuid
$cond_ws_huuid $cond_ws_huuid
ORDER BY ORDER BY
vnc.modified_date DESC a.modified_date DESC
;"; ;";
my $results = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ }); my $results = $anvil->Database->query({ query => $query, source => $THIS_FILE, line => __LINE__ });
my $count = @{$results}; my $count = @{$results};
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { query => $query, count => $count }, prefix => "get_vnc_pipe" });
if ($count == 1) if ($count == 1)
{ {
my $row = $results->[0]; my $row = $results->[0];
$vnc_pipe_info = {}; $vnc_pipe_info = {};
$vnc_pipe_info->{server_vnc_port} = $row->[0]; $vnc_pipe_info->{server_vnc_port} = $row->[0];
$vnc_pipe_info->{host_name} = $row->[1]; $vnc_pipe_info->{host_name} = $row->[1];
$vnc_pipe_info->{ws_host_uuid} = $row->[2]; $vnc_pipe_info->{ws_host_uuid} = $row->[2];
$vnc_pipe_info->{ws_pid} = $row->[3]; $vnc_pipe_info->{ws_pid} = $row->[3];
$vnc_pipe_info->{ws_source_port} = $row->[4]; $vnc_pipe_info->{ws_dest_port} = $row->[4];
$vnc_pipe_info->{ssh_tunnel_host_uuid} = $row->[5]; $vnc_pipe_info->{ssh_tunnel_host_uuid} = $row->[5];
$vnc_pipe_info->{ssh_tunnel_pid} = $row->[6]; $vnc_pipe_info->{ssh_tunnel_pid} = $row->[6];
$vnc_pipe_info->{ssh_tunnel_forward_port} = $row->[7]; $vnc_pipe_info->{ssh_tunnel_dest_port} = $row->[7];
} }
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => $vnc_pipe_info, prefix => "get_vnc_pipe" }); $anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => $vnc_pipe_info, prefix => "get_vnc_pipe" });
@ -753,7 +825,7 @@ sub open_ws
server_vnc_port => $vnc_info->{port}, server_vnc_port => $vnc_info->{port},
ws_host_uuid => $server_info->{host_uuid}, ws_host_uuid => $server_info->{host_uuid},
ws_pid => $ws_info->{pid}, ws_pid => $ws_info->{pid},
ws_source_port => $ws_info->{source_port} ws_dest_port => $ws_info->{source_port}
}); });
} }
@ -776,13 +848,7 @@ sub close_ws
stop_websockify({ host_name => $vnc_pipe_info->{host_name}, ws_pid => $vnc_pipe_info->{ws_pid} }); stop_websockify({ host_name => $vnc_pipe_info->{host_name}, ws_pid => $vnc_pipe_info->{ws_pid} });
insert_or_update_vnc_pipe({ clear_vnc_pipe_ws({ server_uuid => $server_uuid, ws_host_uuid => $vnc_pipe_info->{ws_host_uuid} });
server_uuid => $server_uuid,
server_vnc_port => "NULL",
ws_host_uuid => $vnc_pipe_info->{ws_host_uuid},
ws_pid => "NULL",
ws_source_port => "NULL"
});
return (0); return (0);
} }
@ -801,16 +867,16 @@ sub open_st
return (1, "error_0313") if (not defined $server_info); return (1, "error_0313") if (not defined $server_info);
my $vnc_pipe_info = get_vnc_pipe({ server_uuid => $server_uuid, ws_host_uuid => $server_info->{host_uuid} }); my $vnc_pipe_info = get_vnc_pipe({ is_ws_pid => 1, server_uuid => $server_uuid, ws_host_uuid => $server_info->{host_uuid} });
return (1, "error_0317") if (not defined $vnc_pipe_info); return (1, "error_0317") if (not defined $vnc_pipe_info);
my $ssh_tunnel_info = start_ssh_tunnel({ my $ssh_tunnel_info = start_ssh_tunnel({
server_uuid => $server_uuid, server_uuid => $server_uuid,
host_uuid => $host_uuid, host_uuid => $host_uuid,
ws_host_name => $vnc_pipe_info->{host_name}, ws_host_name => $vnc_pipe_info->{host_name},
ws_host_uuid => $vnc_pipe_info->{ws_host_uuid}, ws_host_uuid => $vnc_pipe_info->{ws_host_uuid},
ws_source_port => $vnc_pipe_info->{ws_source_port} ws_dest_port => $vnc_pipe_info->{ws_dest_port}
}); });
return (1, "error_0316") if (not defined $ssh_tunnel_info); return (1, "error_0316") if (not defined $ssh_tunnel_info);
@ -818,11 +884,12 @@ sub open_st
if ($ssh_tunnel_info->{is_new} or $ssh_tunnel_info->{is_update}) if ($ssh_tunnel_info->{is_new} or $ssh_tunnel_info->{is_update})
{ {
insert_or_update_vnc_pipe({ insert_or_update_vnc_pipe({
server_uuid => $server_uuid, server_uuid => $server_uuid,
ssh_tunnel_host_uuid => $host_uuid, ssh_tunnel_dest_port => $ssh_tunnel_info->{forward_port},
ssh_tunnel_pid => $ssh_tunnel_info->{pid}, ssh_tunnel_host_uuid => $host_uuid,
ssh_tunnel_forward_port => $ssh_tunnel_info->{forward_port}, ssh_tunnel_pid => $ssh_tunnel_info->{pid},
ws_host_uuid => $vnc_pipe_info->{ws_host_uuid} ssh_tunnel_source_port => $vnc_pipe_info->{ws_dest_port},
ws_host_uuid => $vnc_pipe_info->{ws_host_uuid}
}); });
} }
@ -852,104 +919,11 @@ sub close_st
stop_ssh_tunnel({ ssh_tunnel_pid => $vnc_pipe_info->{ssh_tunnel_pid} }); stop_ssh_tunnel({ ssh_tunnel_pid => $vnc_pipe_info->{ssh_tunnel_pid} });
insert_or_update_vnc_pipe({ clear_vnc_pipe_st({ server_uuid => $server_uuid, ws_host_uuid => $vnc_pipe_info->{ws_host_uuid} });
server_uuid => $server_uuid,
ssh_tunnel_host_uuid => "NULL",
ssh_tunnel_pid => "NULL",
ssh_tunnel_forward_port => "NULL",
ws_host_uuid => $vnc_pipe_info->{ws_host_uuid}
});
return (0); return (0);
} }
sub handle_vnc_pipe_error
{
my $error_key = shift;
my $parameters = shift;
my $host_uuid = $parameters->{host_uuid};
my $server_uuid = $parameters->{server_uuid};
$anvil->Log->entry({ source => $THIS_FILE, line => __LINE__, level => 1, key => $error_key, variables => {
server_uuid => $server_uuid,
host_uuid => $host_uuid
} });
$anvil->Job->update_progress({
progress => 100,
message => "$error_key,!!server_uuid!$server_uuid!!,!!host_uuid!$host_uuid!!",
job_status => "failed"
});
}
sub open_vnc_pipe
{
my $parameters = shift;
my $host_uuid = $parameters->{host_uuid};
my $is_print = $parameters->{print};
my $server_host_uuid = $parameters->{server_host_uuid};
my $server_uuid = $parameters->{server_uuid};
my $server_vnc_port = $parameters->{server_vnc_port};
my $is_error;
my $open_output;
my $open_params = {
host_uuid => $host_uuid,
print => $is_print,
server_info => get_server_info($parameters),
server_uuid => $server_uuid,
server_vnc_port => $server_vnc_port
};
($is_error, $open_output) = open_ws($open_params);
return handle_vnc_pipe_error($open_output, $open_params) if ($is_error);
($is_error, $open_output) = open_st($open_params);
return handle_vnc_pipe_error($open_output, $open_params) if ($is_error);
$anvil->Job->update_progress({
progress => 100,
message => "message_0260,!!operation!opening!!,!!server_uuid!".$server_uuid."!!,!!host_uuid!".$host_uuid."!!"
});
return (0, { forward_port => $open_params->{forward_port} });
}
sub close_vnc_pipe
{
my $parameters = shift;
my $host_uuid = $parameters->{host_uuid};
my $server_uuid = $parameters->{server_uuid};
my $is_error;
my $close_output;
my $close_params = {
host_uuid => $host_uuid,
server_uuid => $server_uuid,
vnc_pipe_info => get_vnc_pipe({
server_uuid => $server_uuid,
ssh_tunnel_host_uuid => $host_uuid
})
};
($is_error, $close_output) = close_ws($close_params);
return handle_vnc_pipe_error($close_output, $close_params) if ($is_error);
($is_error, $close_output) = close_st($close_params);
return handle_vnc_pipe_error($close_output, $close_params) if ($is_error);
$anvil->Job->update_progress({
progress => 100,
message => "message_0260,!!operation!closing!!,!!server_uuid!".$server_uuid."!!,!!host_uuid!".$host_uuid."!!"
});
}
$anvil->Get->switches; $anvil->Get->switches;
$anvil->Database->connect; $anvil->Database->connect;
@ -1036,7 +1010,6 @@ $anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, lis
} }); } });
my $map_to_operation = { my $map_to_operation = {
all => { close => \&close_vnc_pipe, open => \&open_vnc_pipe },
st => { close => \&close_st, open => \&open_st }, st => { close => \&close_st, open => \&open_st },
ws => { close => \&close_ws, open => \&open_ws }, ws => { close => \&close_ws, open => \&open_ws },
}; };
@ -1055,14 +1028,39 @@ if ($server_uuid =~ /[a-f0-9]{8}-[a-f0-9]{4}-[1-5][a-f0-9]{3}-[89ab][a-f0-9]{3}-
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { operation => $op } }); $anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 2, list => { operation => $op } });
my ($is_error) = $ops->{$op}({ my $host_uuid = $anvil->data->{sys}{host_uuid};
host_uuid => $anvil->data->{sys}{host_uuid},
my ($is_error, $error) = $ops->{$op}({
host_uuid => $host_uuid,
server_host_uuid => $server_host_uuid, server_host_uuid => $server_host_uuid,
server_uuid => $server_uuid, server_uuid => $server_uuid,
server_vnc_port => $server_vnc_port server_vnc_port => $server_vnc_port
}); });
$anvil->nice_exit({ exit_code => 2 }) if ($is_error); if ($is_error)
{
$anvil->Log->variables({ source => $THIS_FILE, line => __LINE__, level => 1, list => {
error => $error,
host_uuid => $host_uuid,
server_uuid => $server_uuid
} });
if (defined $error and $error =~ /^error_/)
{
$anvil->Job->update_progress({
progress => 100,
message => "$error,!!server_uuid!$server_uuid!!,!!host_uuid!$host_uuid!!",
job_status => "failed"
});
}
$anvil->nice_exit({ exit_code => 2 });
}
$anvil->Job->update_progress({
progress => 100,
message => "message_0260,!!operation!$op!!,!!server_uuid!$server_uuid!!,!!host_uuid!$host_uuid!!"
});
} }
elsif ($is_drop_table) elsif ($is_drop_table)
{ {

Write Preview
Loading…
Cancel
Save