From ea2bceaf19bf13859a0c310d988485754c19f91c Mon Sep 17 00:00:00 2001
From: Dominik Csapak <d.csapak@proxmox.com>
Date: Fri, 16 Feb 2018 12:06:36 +0100
Subject: [PATCH] make an api call for each guest agent command

with a 'register_command' sub, which generates an api call
we call it for each command in the list, and one time for
the old general {vmid}/agent endpoint (for compatibility)

permissions/methods are the same as previously, but can
be overriden

Signed-off-by: Dominik Csapak <d.csapak@proxmox.com>
---
 PVE/API2/Qemu/Agent.pm | 160 +++++++++++++++++++++++++++++------------
 1 file changed, 113 insertions(+), 47 deletions(-)

diff --git a/PVE/API2/Qemu/Agent.pm b/PVE/API2/Qemu/Agent.pm
index 437d3f68..93d47bc6 100644
--- a/PVE/API2/Qemu/Agent.pm
+++ b/PVE/API2/Qemu/Agent.pm
@@ -9,68 +9,134 @@ use PVE::QemuServer;
 
 use base qw(PVE::RESTHandler);
 
-my $guest_agent_commands = [
-    'ping',
-    'get-time',
-    'info',
-    'fsfreeze-status',
-    'fsfreeze-freeze',
-    'fsfreeze-thaw',
-    'fstrim',
-    'network-get-interfaces',
-    'get-vcpus',
-    'get-fsinfo',
-    'get-memory-blocks',
-    'get-memory-block-info',
-    'suspend-hybrid',
-    'suspend-ram',
-    'suspend-disk',
-    'shutdown',
-    ];
-
-__PACKAGE__->register_method({
-    name => 'agent',
-    path => '',
-    method => 'POST',
-    protected => 1,
-    proxyto => 'node',
-    description => "Execute Qemu Guest Agent commands.",
-    permissions => {
-	check => ['perm', '/vms/{vmid}', [ 'VM.Monitor' ]],
+# list of commands
+# will generate one api endpoint per command
+# needs a 'method' property and optionally a 'perms' property (default VM.Monitor)
+my $guest_agent_commands = {
+    'ping' => {
+	method => 'POST',
     },
-    parameters => {
+    'get-time' => {
+	method => 'POST',
+    },
+    'info' => {
+	method => 'POST',
+    },
+    'fsfreeze-status' => {
+	method => 'POST',
+    },
+    'fsfreeze-freeze' => {
+	method => 'POST',
+    },
+    'fsfreeze-thaw' => {
+	method => 'POST',
+    },
+    'fstrim' => {
+	method => 'POST',
+    },
+    'network-get-interfaces' => {
+	method => 'POST',
+    },
+    'get-vcpus' => {
+	method => 'POST',
+    },
+    'get-fsinfo' => {
+	method => 'POST',
+    },
+    'get-memory-blocks' => {
+	method => 'POST',
+    },
+    'get-memory-block-info' => {
+	method => 'POST',
+    },
+    'suspend-hybrid' => {
+	method => 'POST',
+    },
+    'suspend-ram' => {
+	method => 'POST',
+    },
+    'suspend-disk' => {
+	method => 'POST',
+    },
+    'shutdown' => {
+	method => 'POST',
+    },
+};
+
+sub register_command {
+    my ($class, $command, $method, $perm) = @_;
+
+    die "no method given\n" if !$method;
+    die "no command given\n" if !defined($command);
+
+    my $permission;
+
+    if (ref($perm) eq 'HASH') {
+	$permission = $perm;
+    } else {
+	$perm //= 'VM.Monitor';
+	$permission = { check => [ 'perm', '/vms/{vmid}', [ $perm ]]};
+    }
+
+    my $parameters = {
 	additionalProperties => 0,
 	properties => {
 	    node => get_standard_option('pve-node'),
 	    vmid => get_standard_option('pve-vmid', {
-                   completion => \&PVE::QemuServer::complete_vmid_running }),
+		    completion => \&PVE::QemuServer::complete_vmid_running }),
 	    command => {
 		type => 'string',
 		description => "The QGA command.",
-		enum => $guest_agent_commands,
+		enum => [ sort keys %$guest_agent_commands ],
 	    },
 	},
-    },
-    returns => {
-	type => 'object',
-	description => "Returns an object with a single `result` property. The type of that
-property depends on the executed command.",
-    },
-    code => sub {
-	my ($param) = @_;
+    };
 
-	my $vmid = $param->{vmid};
+    my $description = "Execute Qemu Guest Agent commands.";
+    my $name = 'agent';
 
-	my $conf = PVE::QemuConfig->load_config ($vmid); # check if VM exists
+    if ($command ne '') {
+	$description = "Execute $command.";
+	$name = $command;
+	delete $parameters->{properties}->{command};
+    }
 
-	die "No Qemu Guest Agent\n" if !defined($conf->{agent});
-	die "VM $vmid is not running\n" if !PVE::QemuServer::check_running($vmid);
+    __PACKAGE__->register_method({
+	name => $name,
+	path => $command,
+	method => $method,
+	protected => 1,
+	proxyto => 'node',
+	description => $description,
+	permissions => $permission,
+	parameters => $parameters,
+	returns => {
+	    type => 'object',
+	    description => "Returns an object with a single `result` property.",
+	},
+	code => sub {
+	    my ($param) = @_;
 
-	my $cmd = $param->{command};
+	    my $vmid = $param->{vmid};
 
-	my $res = PVE::QemuServer::vm_mon_cmd($vmid, "guest-$cmd");
+	    my $conf = PVE::QemuConfig->load_config ($vmid); # check if VM exists
 
-	return { result => $res };
-    }});
+	    die "No Qemu Guest Agent\n" if !defined($conf->{agent});
+	    die "VM $vmid is not running\n" if !PVE::QemuServer::check_running($vmid);
+
+	    my $cmd = $param->{command} // $command;
+	    my $res = PVE::QemuServer::vm_mon_cmd($vmid, "guest-$cmd");
+
+	    return { result => $res };
+	}});
+}
+
+# old {vmid}/agent POST endpoint, here for compatibility
+__PACKAGE__->register_command('', 'POST');
+
+for my $cmd (sort keys %$guest_agent_commands) {
+    my $props = $guest_agent_commands->{$cmd};
+    __PACKAGE__->register_command($cmd, $props->{method}, $props->{perms});
+}
 
 1;