mirror of
https://git.proxmox.com/git/qemu-server
synced 2025-08-05 06:44:55 +00:00
use add_vm_to_pool/remove_vm_from_pool from PVE::AccessControl
This commit is contained in:
parent
c89dcd85bd
commit
be51704915
@ -249,37 +249,6 @@ __PACKAGE__->register_method({
|
|||||||
}});
|
}});
|
||||||
|
|
||||||
|
|
||||||
sub add_vm_to_pool {
|
|
||||||
my ($vmid, $pool) = @_;
|
|
||||||
|
|
||||||
my $addVMtoPoolFn = sub {
|
|
||||||
my $usercfg = cfs_read_file("user.cfg");
|
|
||||||
if (my $data = $usercfg->{pools}->{$pool}) {
|
|
||||||
$data->{vms}->{$vmid} = 1;
|
|
||||||
$usercfg->{vms}->{$vmid} = $pool;
|
|
||||||
cfs_write_file("user.cfg", $usercfg);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
PVE::AccessControl::lock_user_config($addVMtoPoolFn, "can't add VM $vmid to pool '$pool'");
|
|
||||||
};
|
|
||||||
|
|
||||||
sub remove_vm_from_pool {
|
|
||||||
my ($vmid) = @_;
|
|
||||||
|
|
||||||
my $delVMfromPoolFn = sub {
|
|
||||||
my $usercfg = cfs_read_file("user.cfg");
|
|
||||||
if (my $pool = $usercfg->{vms}->{$vmid}) {
|
|
||||||
if (my $data = $usercfg->{pools}->{$pool}) {
|
|
||||||
delete $data->{vms}->{$vmid};
|
|
||||||
delete $usercfg->{vms}->{$vmid};
|
|
||||||
cfs_write_file("user.cfg", $usercfg);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
PVE::AccessControl::lock_user_config($delVMfromPoolFn, "pool cleanup for VM $vmid failed");
|
|
||||||
}
|
|
||||||
|
|
||||||
__PACKAGE__->register_method({
|
__PACKAGE__->register_method({
|
||||||
name => 'create_vm',
|
name => 'create_vm',
|
||||||
@ -430,7 +399,7 @@ __PACKAGE__->register_method({
|
|||||||
pool => $pool,
|
pool => $pool,
|
||||||
unique => $unique });
|
unique => $unique });
|
||||||
|
|
||||||
add_vm_to_pool($vmid, $pool) if $pool;
|
PVE::AccessControl::add_vm_to_pool($vmid, $pool) if $pool;
|
||||||
};
|
};
|
||||||
|
|
||||||
return $rpcenv->fork_worker('qmrestore', $vmid, $authuser, $realcmd);
|
return $rpcenv->fork_worker('qmrestore', $vmid, $authuser, $realcmd);
|
||||||
@ -479,7 +448,7 @@ __PACKAGE__->register_method({
|
|||||||
die "create failed - $err";
|
die "create failed - $err";
|
||||||
}
|
}
|
||||||
|
|
||||||
add_vm_to_pool($vmid, $pool) if $pool;
|
PVE::AccessControl::add_vm_to_pool($vmid, $pool) if $pool;
|
||||||
};
|
};
|
||||||
|
|
||||||
return $rpcenv->fork_worker('qmcreate', $vmid, $authuser, $realcmd);
|
return $rpcenv->fork_worker('qmcreate', $vmid, $authuser, $realcmd);
|
||||||
@ -1102,7 +1071,7 @@ __PACKAGE__->register_method({
|
|||||||
|
|
||||||
PVE::QemuServer::vm_destroy($storecfg, $vmid, $skiplock);
|
PVE::QemuServer::vm_destroy($storecfg, $vmid, $skiplock);
|
||||||
|
|
||||||
remove_vm_from_pool($vmid);
|
PVE::AccessControl::remove_vm_from_pool($vmid);
|
||||||
};
|
};
|
||||||
|
|
||||||
return $rpcenv->fork_worker('qmdestroy', $vmid, $authuser, $realcmd);
|
return $rpcenv->fork_worker('qmdestroy', $vmid, $authuser, $realcmd);
|
||||||
@ -2097,7 +2066,7 @@ __PACKAGE__->register_method({
|
|||||||
if !rename($conffile, $newconffile);
|
if !rename($conffile, $newconffile);
|
||||||
}
|
}
|
||||||
|
|
||||||
add_vm_to_pool($newid, $pool) if $pool;
|
PVE::AccessControl::add_vm_to_pool($newid, $pool) if $pool;
|
||||||
};
|
};
|
||||||
if (my $err = $@) {
|
if (my $err = $@) {
|
||||||
unlink $conffile;
|
unlink $conffile;
|
||||||
|
Loading…
Reference in New Issue
Block a user