[pve-devel] [PATCH container v2 09/11] migration: prepare: use volume content type assertion helpers

[Daniel Kral]

Make use of newly introduced content type assertion helpers in the
preconditions check and preparation path of the migration API handlers.

Signed-off-by: Daniel Kral <d.kral at proxmox.com>
---
changes since v1:
- new!

 src/PVE/API2/LXC.pm    |  8 ++------
 src/PVE/LXC/Migrate.pm | 13 ++++++-------
 2 files changed, 8 insertions(+), 13 deletions(-)

diff --git a/src/PVE/API2/LXC.pm b/src/PVE/API2/LXC.pm
index 8e5e6a6..a945a08 100644
--- a/src/PVE/API2/LXC.pm
+++ b/src/PVE/API2/LXC.pm
@@ -43,13 +43,9 @@ BEGIN {
 my $check_storage_access_migrate = sub {
     my ($rpcenv, $authuser, $storecfg, $storage, $node) = @_;
 
-    PVE::Storage::storage_check_enabled($storecfg, $storage, $node);
-
     $rpcenv->check($authuser, "/storage/$storage", ['Datastore.AllocateSpace']);
-
-    my $scfg = PVE::Storage::storage_config($storecfg, $storage);
-    die "storage '$storage' does not support CT rootdirs\n"
-	if !$scfg->{content}->{rootdir};
+    PVE::Storage::storage_check_enabled($storecfg, $storage, $node);
+    PVE::Storage::assert_content_type_supported($storecfg, $storage, 'rootdir', $node);
 };
 
 __PACKAGE__->register_method ({
diff --git a/src/PVE/LXC/Migrate.pm b/src/PVE/LXC/Migrate.pm
index f06a10a..61d241a 100644
--- a/src/PVE/LXC/Migrate.pm
+++ b/src/PVE/LXC/Migrate.pm
@@ -73,12 +73,10 @@ sub prepare {
 	# check if storage is available on both nodes
 	my ($storecfg, $node) = $self->@{qw(storecfg node)};
 	my $scfg = PVE::Storage::storage_check_enabled($storecfg, $storage);
+	PVE::Storage::assert_content_type_supported($storecfg, $storage, 'rootdir');
 
 	my $targetsid = $storage;
 
-	die "content type 'rootdir' is not available on storage '$storage'\n"
-	    if !$scfg->{content}->{rootdir};
-
 	if ($scfg->{shared} && !$remote) {
 	    # PVE::Storage::activate_storage checks this for non-shared storages
 	    my $plugin = PVE::Storage::Plugin->lookup($scfg->{type});
@@ -93,10 +91,11 @@ sub prepare {
 	}
 
 	if (!$remote) {
-	    my $target_scfg = PVE::Storage::storage_check_enabled($storecfg, $targetsid, $node);
-
-	    die "$volid: content type 'rootdir' is not available on storage '$targetsid'\n"
-		if !$target_scfg->{content}->{rootdir};
+	    PVE::Storage::storage_check_enabled($storecfg, $targetsid, $node);
+	    eval {
+		PVE::Storage::assert_content_type_supported($storecfg, $targetsid, 'rootdir', $node)
+	    };
+	    die "$volid: $@" if $@;
 	}
 
 	$storages->{$targetsid} = 1;
-- 
2.39.5