mirror of
git://git.proxmox.com/git/qemu-server.git
synced 2025-01-20 14:03:50 +03:00
131 lines
2.7 KiB
Perl
131 lines
2.7 KiB
Perl
package PVE::QemuServer::Agent;
|
|
|
|
use strict;
|
|
use warnings;
|
|
|
|
use PVE::QemuServer;
|
|
use PVE::QemuServer::Monitor;
|
|
use MIME::Base64 qw(decode_base64 encode_base64);
|
|
use JSON;
|
|
use base 'Exporter';
|
|
|
|
our @EXPORT_OK = qw(
|
|
check_agent_error
|
|
agent_available
|
|
agent_cmd
|
|
);
|
|
|
|
sub check_agent_error {
|
|
my ($result, $errmsg, $noerr) = @_;
|
|
|
|
$errmsg //= '';
|
|
my $error = '';
|
|
if (ref($result) eq 'HASH' && $result->{error} && $result->{error}->{desc}) {
|
|
$error = "Agent error: $result->{error}->{desc}\n";
|
|
} elsif (!defined($result)) {
|
|
$error = "Agent error: $errmsg\n";
|
|
}
|
|
|
|
if ($error) {
|
|
die $error if !$noerr;
|
|
|
|
warn $error;
|
|
return;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
sub agent_available {
|
|
my ($vmid, $conf, $noerr) = @_;
|
|
|
|
eval {
|
|
die "No QEMU guest agent configured\n" if !defined($conf->{agent});
|
|
die "VM $vmid is not running\n" if !PVE::QemuServer::check_running($vmid);
|
|
die "QEMU guest agent is not running\n" if !PVE::QemuServer::qga_check_running($vmid, 1);
|
|
};
|
|
|
|
if (my $err = $@) {
|
|
die $err if !$noerr;
|
|
return;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
# loads config, checks if available, executes command, checks for errors
|
|
sub agent_cmd {
|
|
my ($vmid, $cmd, $params, $errormsg, $noerr) = @_;
|
|
|
|
my $conf = PVE::QemuConfig->load_config($vmid); # also checks if VM exists
|
|
agent_available($vmid, $conf, $noerr);
|
|
|
|
my $res = PVE::QemuServer::Monitor::mon_cmd($vmid, "guest-$cmd", %$params);
|
|
check_agent_error($res, $errormsg, $noerr);
|
|
|
|
return $res;
|
|
}
|
|
|
|
sub qemu_exec {
|
|
my ($vmid, $input_data, $cmd) = @_;
|
|
|
|
my $args = {
|
|
'capture-output' => JSON::true,
|
|
};
|
|
|
|
if ($cmd) {
|
|
$args->{path} = shift @$cmd;
|
|
$args->{arg} = $cmd;
|
|
}
|
|
|
|
$args->{'input-data'} = encode_base64($input_data, '') if defined($input_data);
|
|
|
|
die "command or input-data (or both) required\n"
|
|
if !defined($args->{'input-data'}) && !defined($args->{path});
|
|
|
|
my $errmsg = "can't execute command";
|
|
if ($cmd) {
|
|
$errmsg .= " ($args->{path} $args->{arg})";
|
|
}
|
|
if (defined($input_data)) {
|
|
$errmsg .= " (input-data given)";
|
|
}
|
|
|
|
my $res = agent_cmd($vmid, "exec", $args, $errmsg);
|
|
|
|
return $res;
|
|
}
|
|
|
|
sub qemu_exec_status {
|
|
my ($vmid, $pid) = @_;
|
|
|
|
my $res = agent_cmd($vmid, "exec-status", { pid => $pid }, "can't get exec status for '$pid'");
|
|
|
|
if ($res->{'out-data'}) {
|
|
my $decoded = eval { decode_base64($res->{'out-data'}) };
|
|
warn $@ if $@;
|
|
if (defined($decoded)) {
|
|
$res->{'out-data'} = $decoded;
|
|
}
|
|
}
|
|
|
|
if ($res->{'err-data'}) {
|
|
my $decoded = eval { decode_base64($res->{'err-data'}) };
|
|
warn $@ if $@;
|
|
if (defined($decoded)) {
|
|
$res->{'err-data'} = $decoded;
|
|
}
|
|
}
|
|
|
|
# convert JSON::Boolean to 1/0
|
|
foreach my $d (keys %$res) {
|
|
if (JSON::is_bool($res->{$d})) {
|
|
$res->{$d} = ($res->{$d})? 1 : 0;
|
|
}
|
|
}
|
|
|
|
return $res;
|
|
}
|
|
|
|
1;
|