]> git.proxmox.com Git - pve-access-control.git/blobdiff - PVE/AccessControl.pm
change from dpkg-deb to dpkg-buildpackage
[pve-access-control.git] / PVE / AccessControl.pm
index 2239e1449cacc7122756e0da32b9c925804bca67..7d02cdf27494b64a237963db408a6a28eca38517 100644 (file)
@@ -1,11 +1,17 @@
 package PVE::AccessControl;
 
 use strict;
+use warnings;
 use Encode;
 use Crypt::OpenSSL::Random;
 use Crypt::OpenSSL::RSA;
+use Net::SSLeay;
+use Net::IP;
 use MIME::Base64;
 use Digest::SHA;
+
+use PVE::OTP;
+use PVE::Ticket;
 use PVE::Tools qw(run_command lock_file file_get_contents split_list safe_print);
 use PVE::Cluster qw(cfs_register_file cfs_read_file cfs_write_file cfs_lock_file);
 use PVE::JSONSchema;
@@ -37,8 +43,8 @@ my $ticket_lifetime = 3600*2; # 2 hours
 
 Crypt::OpenSSL::RSA->import_random_seed();
 
-cfs_register_file('user.cfg', 
-                 \&parse_user_config,  
+cfs_register_file('user.cfg',
+                 \&parse_user_config,
                  \&write_user_config);
 
 
@@ -60,11 +66,11 @@ sub lock_user_config {
 }
 
 my $pve_auth_pub_key;
-sub get_pubkey {    
+sub get_pubkey {
 
     return $pve_auth_pub_key if $pve_auth_pub_key;
 
-    my $input = PVE::Tools::file_get_contents($authpubkeyfn); 
+    my $input = PVE::Tools::file_get_contents($authpubkeyfn);
 
     $pve_auth_pub_key = Crypt::OpenSSL::RSA->new_public_key($input);
 
@@ -74,7 +80,7 @@ sub get_pubkey {
 my $csrf_prevention_secret;
 my $get_csrfr_secret = sub {
     if (!$csrf_prevention_secret) {
-       my $input = PVE::Tools::file_get_contents($pve_www_key_fn); 
+       my $input = PVE::Tools::file_get_contents($pve_www_key_fn);
        $csrf_prevention_secret = Digest::SHA::sha1_base64($input);
     }
     return $csrf_prevention_secret;
@@ -83,30 +89,18 @@ my $get_csrfr_secret = sub {
 sub assemble_csrf_prevention_token {
     my ($username) = @_;
 
-    my $timestamp = sprintf("%08X", time());
-
-    my $digest = Digest::SHA::sha1_base64("$timestamp:$username", &$get_csrfr_secret());
+    my $secret =  &$get_csrfr_secret();
 
-    return "$timestamp:$digest"; 
+    return PVE::Ticket::assemble_csrf_prevention_token ($secret, $username);
 }
 
 sub verify_csrf_prevention_token {
     my ($username, $token, $noerr) = @_;
 
-    if ($token =~ m/^([A-Z0-9]{8}):(\S+)$/) {
-       my $sig = $2;
-       my $timestamp = $1;
-       my $ttime = hex($timestamp);
-
-       my $digest = Digest::SHA::sha1_base64("$timestamp:$username", &$get_csrfr_secret());
-
-       my $age = time() - $ttime;
-       return if ($digest eq $sig) && ($age > -300) && ($age < $ticket_lifetime);
-    }
+    my $secret =  &$get_csrfr_secret();
 
-    die "Permission denied - invalid csrf token\n" if !$noerr;
-
-    return undef;
+    return PVE::Ticket::verify_csrf_prevention_token(
+       $secret, $username, $token, -300, $ticket_lifetime, $noerr);
 }
 
 my $pve_auth_priv_key;
@@ -114,7 +108,7 @@ sub get_privkey {
 
     return $pve_auth_priv_key if $pve_auth_priv_key;
 
-    my $input = PVE::Tools::file_get_contents($authprivkeyfn); 
+    my $input = PVE::Tools::file_get_contents($authprivkeyfn);
 
     $pve_auth_priv_key = Crypt::OpenSSL::RSA->new_private_key($input);
 
@@ -126,42 +120,22 @@ sub assemble_ticket {
 
     my $rsa_priv = get_privkey();
 
-    my $timestamp = sprintf("%08X", time());
-
-    my $plain = "PVE:$username:$timestamp";
-
-    my $ticket = $plain . "::" . encode_base64($rsa_priv->sign($plain), '');
-
-    return $ticket;
+    return PVE::Ticket::assemble_rsa_ticket($rsa_priv, 'PVE', $username);
 }
 
 sub verify_ticket {
     my ($ticket, $noerr) = @_;
 
-    if ($ticket && $ticket =~ m/^(PVE:\S+)::([^:\s]+)$/) {
-       my $plain = $1;
-       my $sig = $2;
-
-       my $rsa_pub = get_pubkey();
-       if ($rsa_pub->verify($plain, decode_base64($sig))) {
-           if ($plain =~ m/^PVE:(\S+):([A-Z0-9]{8})$/) {
-               my $username = $1;
-               my $timestamp = $2;
-               my $ttime = hex($timestamp);
+    my $rsa_pub = get_pubkey();
 
-               my $age = time() - $ttime;
+    my ($username, $age) = PVE::Ticket::verify_rsa_ticket(
+       $rsa_pub, 'PVE', $ticket, undef, -300, $ticket_lifetime, $noerr);
 
-               if (PVE::Auth::Plugin::verify_username($username, 1) &&
-                   ($age > -300) && ($age < $ticket_lifetime)) {
-                   return wantarray ? ($username, $age) : $username;
-               }
-           }
-       }
-    }
+    return undef if $noerr && !defined($username);
 
-    die "permission denied - invalid ticket\n" if !$noerr;
+    return undef if !PVE::Auth::Plugin::verify_username($username, $noerr);
 
-    return undef;
+    return wantarray ? ($username, $age) : $username;
 }
 
 # VNC tickets
@@ -172,45 +146,103 @@ sub assemble_vnc_ticket {
 
     my $rsa_priv = get_privkey();
 
-    my $timestamp = sprintf("%08X", time());
-
-    my $plain = "PVEVNC:$timestamp";
-
     $path = normalize_path($path);
 
-    my $full = "$plain:$username:$path";
-
-    my $ticket = $plain . "::" . encode_base64($rsa_priv->sign($full), '');
+    my $secret_data = "$username:$path";
 
-    return $ticket;
+    return PVE::Ticket::assemble_rsa_ticket(
+       $rsa_priv, 'PVEVNC', undef, $secret_data);
 }
 
 sub verify_vnc_ticket {
     my ($ticket, $username, $path, $noerr) = @_;
 
-    if ($ticket && $ticket =~ m/^(PVEVNC:\S+)::([^:\s]+)$/) {
-       my $plain = $1;
-       my $sig = $2;
-       my $full = "$plain:$username:$path";
+    my $rsa_pub = get_pubkey();
 
-       my $rsa_pub = get_pubkey();
-       # Note: sign only match if $username and  $path is correct
-       if ($rsa_pub->verify($full, decode_base64($sig))) {
-           if ($plain =~ m/^PVEVNC:([A-Z0-9]{8})$/) {
-               my $ttime = hex($1);
+    my $secret_data = "$username:$path";
 
-               my $age = time() - $ttime;
+    return PVE::Ticket::verify_rsa_ticket(
+       $rsa_pub, 'PVEVNC', $ticket, $secret_data, -20, 40, $noerr);
+}
 
-               if (($age > -20) && ($age < 40)) {
-                   return 1;
-               }
-           }
-       }
+sub assemble_spice_ticket {
+    my ($username, $vmid, $node) = @_;
+
+    my $secret = &$get_csrfr_secret();
+
+    return PVE::Ticket::assemble_spice_ticket(
+       $secret, $username, $vmid, $node);
+}
+
+
+sub verify_spice_connect_url {
+    my ($connect_str) = @_;
+
+    my $secret = &$get_csrfr_secret();
+
+    return PVE::Ticket::verify_spice_connect_url($secret, $connect_str);
+}
+
+sub read_x509_subject_spice {
+    my ($filename) = @_;
+
+    # read x509 subject
+    my $bio = Net::SSLeay::BIO_new_file($filename, 'r');
+    die "Could not open $filename using OpenSSL\n"
+       if !$bio;
+
+    my $x509 = Net::SSLeay::PEM_read_bio_X509($bio);
+    Net::SSLeay::BIO_free($bio);
+
+    die "Could not parse X509 certificate in $filename\n"
+       if !$x509;
+
+    my $nameobj = Net::SSLeay::X509_get_subject_name($x509);
+    my $subject = Net::SSLeay::X509_NAME_oneline($nameobj);
+    Net::SSLeay::X509_free($x509);
+
+    # remote-viewer wants comma as seperator (not '/')
+    $subject =~ s!^/!!;
+    $subject =~ s!/(\w+=)!,$1!g;
+
+    return $subject;
+}
+
+# helper to generate SPICE remote-viewer configuration
+sub remote_viewer_config {
+    my ($authuser, $vmid, $node, $proxy, $title, $port) = @_;
+
+    if (!$proxy) {
+       my $host = `hostname -f` || PVE::INotify::nodename();
+       chomp $host;
+       $proxy = $host;
     }
 
-    die "permission denied - invalid vnc ticket\n" if !$noerr;
+    my ($ticket, $proxyticket) = assemble_spice_ticket($authuser, $vmid, $node);
+
+    my $filename = "/etc/pve/local/pve-ssl.pem";
+    my $subject = read_x509_subject_spice($filename);
+
+    my $cacert = PVE::Tools::file_get_contents("/etc/pve/pve-root-ca.pem", 8192);
+    $cacert =~ s/\n/\\n/g;
+
+    $proxy = "[$proxy]" if Net::IP::ip_is_ipv6($proxy);
+    my $config = {
+       'secure-attention' => "Ctrl+Alt+Ins",
+       'toggle-fullscreen' => "Shift+F11",
+       'release-cursor' => "Ctrl+Alt+R",
+       type => 'spice',
+       title => $title,
+       host => $proxyticket, # this breaks tls hostname verification, so we need to use 'host-subject'
+       proxy => "http://$proxy:3128",
+       'tls-port' => $port,
+       'host-subject' => $subject,
+       ca => $cacert,
+       password => $ticket,
+       'delete-this-file' => 1,
+    };
 
-    return undef;
+    return ($ticket, $proxyticket, $config);
 }
 
 sub check_user_exist {
@@ -218,11 +250,11 @@ sub check_user_exist {
 
     $username = PVE::Auth::Plugin::verify_username($username, $noerr);
     return undef if !$username;
+
     return $usercfg->{users}->{$username} if $usercfg && $usercfg->{users}->{$username};
 
     die "no such user ('$username')\n" if !$noerr;
+
     return undef;
 }
 
@@ -234,50 +266,63 @@ sub check_user_enabled {
 
     return 1 if $data->{enable};
 
-    return 1 if $username eq 'root@pam'; # root is always enabled
-
     die "user '$username' is disabled\n" if !$noerr;
+
     return undef;
 }
 
+sub verify_one_time_pw {
+    my ($usercfg, $username, $tfa_cfg, $otp) = @_;
+
+    my $type = $tfa_cfg->{type};
+
+    die "missing one time password for two-factor authentication '$type'\n" if !$otp;
+
+    # fixme: proxy support?
+    my $proxy;
+
+    if ($type eq 'yubico') {
+       my $keys = $usercfg->{users}->{$username}->{keys};
+       PVE::OTP::yubico_verify_otp($otp, $keys, $tfa_cfg->{url},
+                                   $tfa_cfg->{id}, $tfa_cfg->{key}, $proxy);
+    } elsif ($type eq 'oath') {
+       my $keys = $usercfg->{users}->{$username}->{keys};
+       PVE::OTP::oath_verify_otp($otp, $keys, $tfa_cfg->{step}, $tfa_cfg->{digits});
+    } else {
+       die "unknown tfa type '$type'\n";
+    }
+}
+
 # password should be utf8 encoded
+# Note: some plugins delay/sleep if auth fails
 sub authenticate_user {
-    my ($username, $password) = @_;
+    my ($username, $password, $otp) = @_;
 
     die "no username specified\n" if !$username;
+
     my ($ruid, $realm);
 
     ($username, $ruid, $realm) = PVE::Auth::Plugin::verify_username($username);
 
     my $usercfg = cfs_read_file('user.cfg');
 
-    eval { check_user_enabled($usercfg, $username); };
-    if (my $err = $@) {
-       sleep(2);
-       die $err;
-    }
+    check_user_enabled($usercfg, $username);
 
     my $ctime = time();
     my $expire = $usercfg->{users}->{$username}->{expire};
 
-    if ($expire && ($expire < $ctime)) {
-       sleep(2);
-       die "account expired\n"
-    }
+    die "account expired\n" if $expire && ($expire < $ctime);
 
     my $domain_cfg = cfs_read_file('domains.cfg');
 
-    eval {
-       my $cfg = $domain_cfg->{ids}->{$realm};
-       die "auth domain '$realm' does not exists\n" if !$cfg;
-       my $plugin = PVE::Auth::Plugin->lookup($cfg->{type});
-       $plugin->authenticate_user($cfg, $realm, $ruid, $password);
-    };
-    if (my $err = $@) {
-       sleep(2); # timeout after failed auth
-       die $err;
+    my $cfg = $domain_cfg->{ids}->{$realm};
+    die "auth domain '$realm' does not exists\n" if !$cfg;
+    my $plugin = PVE::Auth::Plugin->lookup($cfg->{type});
+    $plugin->authenticate_user($cfg, $realm, $ruid, $password);
+
+    if ($cfg->{tfa}) {
+       my $tfa_cfg = PVE::Auth::Plugin::parse_tfa_config($cfg->{tfa});
+       verify_one_time_pw($usercfg, $username, $tfa_cfg, $otp);
     }
 
     return $username;
@@ -291,102 +336,95 @@ sub domain_set_password {
     my $domain_cfg = cfs_read_file('domains.cfg');
 
     my $cfg = $domain_cfg->{ids}->{$realm};
-    die "auth domain '$realm' does not exists\n" if !$cfg;
+    die "auth domain '$realm' does not exist\n" if !$cfg;
     my $plugin = PVE::Auth::Plugin->lookup($cfg->{type});
     $plugin->store_password($cfg, $realm, $username, $password);
 }
 
 sub add_user_group {
-
     my ($username, $usercfg, $group) = @_;
+
     $usercfg->{users}->{$username}->{groups}->{$group} = 1;
     $usercfg->{groups}->{$group}->{users}->{$username} = 1;
 }
 
 sub delete_user_group {
-
     my ($username, $usercfg) = @_;
-    
+
     foreach my $group (keys %{$usercfg->{groups}}) {
 
-       delete ($usercfg->{groups}->{$group}->{users}->{$username}) 
+       delete ($usercfg->{groups}->{$group}->{users}->{$username})
            if $usercfg->{groups}->{$group}->{users}->{$username};
     }
 }
 
 sub delete_user_acl {
-
     my ($username, $usercfg) = @_;
 
     foreach my $acl (keys %{$usercfg->{acl}}) {
 
-       delete ($usercfg->{acl}->{$acl}->{users}->{$username}) 
+       delete ($usercfg->{acl}->{$acl}->{users}->{$username})
            if $usercfg->{acl}->{$acl}->{users}->{$username};
     }
 }
 
 sub delete_group_acl {
-
     my ($group, $usercfg) = @_;
 
     foreach my $acl (keys %{$usercfg->{acl}}) {
 
-       delete ($usercfg->{acl}->{$acl}->{groups}->{$group}) 
+       delete ($usercfg->{acl}->{$acl}->{groups}->{$group})
            if $usercfg->{acl}->{$acl}->{groups}->{$group};
     }
 }
 
 sub delete_pool_acl {
-
     my ($pool, $usercfg) = @_;
 
     my $path = "/pool/$pool";
 
-    foreach my $aclpath (keys %{$usercfg->{acl}}) {
-       delete ($usercfg->{acl}->{$aclpath})
-           if $usercfg->{acl}->{$aclpath} eq 'path';
-    }
+    delete ($usercfg->{acl}->{$path})
 }
 
 # we automatically create some predefined roles by splitting privs
 # into 3 groups (per category)
 # root: only root is allowed to do that
 # admin: an administrator can to that
-# user: a normak user/customer can to that
+# user: a normal user/customer can to that
 my $privgroups = {
     VM => {
        root => [],
-       admin => [           
-           'VM.Config.Disk', 
-           'VM.Config.CPU', 
-           'VM.Config.Memory', 
-           'VM.Config.Network', 
+       admin => [
+           'VM.Config.Disk',
+           'VM.Config.CPU',
+           'VM.Config.Memory',
+           'VM.Config.Network',
            'VM.Config.HWType',
-           'VM.Config.Options', # covers all other things 
-           'VM.Allocate', 
-           'VM.Copy', 
+           'VM.Config.Options', # covers all other things
+           'VM.Allocate',
+           'VM.Clone',
            'VM.Migrate',
-           'VM.Monitor', 
-           'VM.Snapshot', 
+           'VM.Monitor',
+           'VM.Snapshot',
        ],
        user => [
            'VM.Config.CDROM', # change CDROM media
-           'VM.Console', 
+           'VM.Console',
            'VM.Backup',
            'VM.PowerMgmt',
        ],
-       audit => [ 
+       audit => [
            'VM.Audit',
        ],
     },
     Sys => {
        root => [
-           'Sys.PowerMgmt',     
+           'Sys.PowerMgmt',
            'Sys.Modify', # edit/change node settings
        ],
        admin => [
            'Permissions.Modify',
-           'Sys.Console',    
+           'Sys.Console',
            'Sys.Syslog',
        ],
        user => [],
@@ -414,7 +452,7 @@ my $privgroups = {
        admin => [
            'User.Modify',
            'Group.Allocate', # edit/change group settings
-           'Realm.AllocateUser', 
+           'Realm.AllocateUser',
        ],
        user => [],
        audit => [],
@@ -432,15 +470,15 @@ my $privgroups = {
 my $valid_privs = {};
 
 my $special_roles = {
-    'NoAccess' => {}, # no priviledges
-    'Administrator' => $valid_privs, # all priviledges
+    'NoAccess' => {}, # no privileges
+    'Administrator' => $valid_privs, # all privileges
 };
 
 sub create_roles {
 
     foreach my $cat (keys %$privgroups) {
        my $cd = $privgroups->{$cat};
-       foreach my $p (@{$cd->{root}}, @{$cd->{admin}}, 
+       foreach my $p (@{$cd->{root}}, @{$cd->{admin}},
                       @{$cd->{user}}, @{$cd->{audit}}) {
            $valid_privs->{$p} = 1;
        }
@@ -459,7 +497,7 @@ sub create_roles {
        }
     }
 
-    $special_roles->{"PVETemplateUser"} = { 'VM.Copy' => 1, 'VM.Audit' => 1 };
+    $special_roles->{"PVETemplateUser"} = { 'VM.Clone' => 1, 'VM.Audit' => 1 };
 };
 
 create_roles();
@@ -475,9 +513,9 @@ sub add_role_privs {
        if (defined ($valid_privs->{$priv})) {
            $usercfg->{roles}->{$role}->{$priv} = 1;
        } else {
-           die "invalid priviledge '$priv'\n";
-       } 
-    }  
+           die "invalid privilege '$priv'\n";
+       }
+    }
 }
 
 sub normalize_path {
@@ -491,10 +529,10 @@ sub normalize_path {
 
     $path = "/$path" if $path !~ m|^/|;
 
-    return undef if $path !~ m|^[[:alnum:]\-\_\/]+$|;
+    return undef if $path !~ m|^[[:alnum:]\.\-\_\/]+$|;
 
     return $path;
-} 
+}
 
 
 PVE::JSONSchema::register_format('pve-groupid', \&verify_groupname);
@@ -507,7 +545,7 @@ sub verify_groupname {
 
        return undef;
     }
-    
+
     return $groupname;
 }
 
@@ -521,7 +559,7 @@ sub verify_rolename {
 
        return undef;
     }
-    
+
     return $rolename;
 }
 
@@ -535,7 +573,7 @@ sub verify_poolname {
 
        return undef;
     }
-    
+
     return $poolname;
 }
 
@@ -544,11 +582,11 @@ sub verify_privname {
     my ($priv, $noerr) = @_;
 
     if (!$valid_privs->{$priv}) {
-       die "invalid priviledge '$priv'\n" if !$noerr;
+       die "invalid privilege '$priv'\n" if !$noerr;
 
        return undef;
     }
-    
+
     return $priv;
 }
 
@@ -559,10 +597,10 @@ sub userconfig_force_defaults {
        $cfg->{roles}->{$r} = $special_roles->{$r};
     }
 
-    # fixme: remove 'root' group (not required)?
-
-    # add root user 
-    $cfg->{users}->{'root@pam'}->{enable} = 1;
+    # add root user if not exists
+    if (!$cfg->{users}->{'root@pam'}) {
+       $cfg->{users}->{'root@pam'}->{enable} = 1;
+    }
 }
 
 sub parse_user_config {
@@ -572,15 +610,13 @@ sub parse_user_config {
 
     userconfig_force_defaults($cfg);
 
-    while ($raw && $raw =~ s/^(.*?)(\n|$)//) {
+    $raw = '' if !defined($raw);
+    while ($raw =~ /^\s*(.+?)\s*$/gm) {
        my $line = $1;
-
-       next if $line =~ m/^\s*$/; # skip empty lines
-
        my @data;
 
        foreach my $d (split (/:/, $line)) {
-           $d =~ s/^\s+//; 
+           $d =~ s/^\s+//;
            $d =~ s/\s+$//;
            push @data, $d
        }
@@ -588,7 +624,7 @@ sub parse_user_config {
        my $et = shift @data;
 
        if ($et eq 'user') {
-           my ($user, $enable, $expire, $firstname, $lastname, $email, $comment) = @data;
+           my ($user, $enable, $expire, $firstname, $lastname, $email, $comment, $keys) = @data;
 
            my (undef, undef, $realm) = PVE::Auth::Plugin::verify_username($user, 1);
            if (!$realm) {
@@ -620,6 +656,8 @@ sub parse_user_config {
            $cfg->{users}->{$user}->{email} = $email;
            $cfg->{users}->{$user}->{comment} = PVE::Tools::decode_text($comment) if $comment;
            $cfg->{users}->{$user}->{expire} = $expire;
+           # keys: allowed yubico key ids or oath secrets (base32 encoded)
+           $cfg->{users}->{$user}->{keys} = $keys if $keys;
 
            #$cfg->{users}->{$user}->{groups}->{$group} = 1;
            #$cfg->{groups}->{$group}->{$user} = 1;
@@ -644,7 +682,7 @@ sub parse_user_config {
                    next;
                }
 
-               if ($cfg->{users}->{$user}) { # user exists 
+               if ($cfg->{users}->{$user}) { # user exists
                    $cfg->{users}->{$user}->{groups}->{$group} = 1;
                    $cfg->{groups}->{$group}->{users}->{$user} = 1;
                } else {
@@ -654,7 +692,7 @@ sub parse_user_config {
 
        } elsif ($et eq 'role') {
            my ($role, $privlist) = @data;
-               
+
            if (!verify_rolename($role, 1)) {
                warn "user config - ignore role '$role' - invalid characters in role name\n";
                next;
@@ -668,15 +706,15 @@ sub parse_user_config {
                    $cfg->{roles}->{$role}->{$priv} = 1;
                } else {
                    warn "user config - ignore invalid priviledge '$priv'\n";
-               } 
+               }
            }
-           
+
        } elsif ($et eq 'acl') {
            my ($propagate, $pathtxt, $uglist, $rolelist) = @data;
 
            if (my $path = normalize_path($pathtxt)) {
                foreach my $role (split_list($rolelist)) {
-                       
+
                    if (!verify_rolename($role, 1)) {
                        warn "user config - ignore invalid role name '$role' in acl\n";
                        next;
@@ -685,13 +723,13 @@ sub parse_user_config {
                    foreach my $ug (split_list($uglist)) {
                        if ($ug =~ m/^@(\S+)$/) {
                            my $group = $1;
-                           if ($cfg->{groups}->{$group}) { # group exists 
+                           if ($cfg->{groups}->{$group}) { # group exists
                                $cfg->{acl}->{$path}->{groups}->{$group}->{$role} = $propagate;
                            } else {
                                warn "user config - ignore invalid acl group '$group'\n";
                            }
                        } elsif (PVE::Auth::Plugin::verify_username($ug, 1)) {
-                           if ($cfg->{users}->{$ug}) { # user exists 
+                           if ($cfg->{users}->{$ug}) { # user exists
                                $cfg->{acl}->{$path}->{users}->{$ug}->{$role} = $propagate;
                            } else {
                                warn "user config - ignore invalid acl member '$ug'\n";
@@ -730,7 +768,7 @@ sub parse_user_config {
                }
 
                $cfg->{pools}->{$pool}->{vms}->{$vmid} = 1;
-                   
+
                # record vmid ==> pool relation
                $cfg->{vms}->{$vmid} = $pool;
            }
@@ -765,7 +803,8 @@ sub write_user_config {
        my $comment = $d->{comment} ? PVE::Tools::encode_text($d->{comment}) : '';
        my $expire = int($d->{expire} || 0);
        my $enable = $d->{enable} ? 1 : 0;
-       $data .= "user:$user:$enable:$expire:$firstname:$lastname:$email:$comment:\n";
+       my $keys = $d->{keys} ? $d->{keys} : '';
+       $data .= "user:$user:$enable:$expire:$firstname:$lastname:$email:$comment:$keys:\n";
     }
 
     $data .= "\n";
@@ -773,7 +812,7 @@ sub write_user_config {
     foreach my $group (keys %{$cfg->{groups}}) {
        my $d = $cfg->{groups}->{$group};
        my $list = join (',', keys %{$d->{users}});
-       my $comment = $d->{comment} ? PVE::Tools::encode_text($d->{comment}) : '';      
+       my $comment = $d->{comment} ? PVE::Tools::encode_text($d->{comment}) : '';
        $data .= "group:$group:$list:$comment:\n";
     }
 
@@ -783,7 +822,7 @@ sub write_user_config {
        my $d = $cfg->{pools}->{$pool};
        my $vmlist = join (',', keys %{$d->{vms}});
        my $storelist = join (',', keys %{$d->{storage}});
-       my $comment = $d->{comment} ? PVE::Tools::encode_text($d->{comment}) : '';      
+       my $comment = $d->{comment} ? PVE::Tools::encode_text($d->{comment}) : '';
        $data .= "pool:$pool:$comment:$vmlist:$storelist:\n";
     }
 
@@ -822,8 +861,8 @@ sub write_user_config {
        }
 
        foreach my $user (keys %{$d->{users}}) {
-           # no need to save, because root is always 'Administartor'
-           next if $user eq 'root@pam'; 
+           # no need to save, because root is always 'Administrator'
+           next if $user eq 'root@pam';
 
            my $l0 = '';
            my $l1 = '';
@@ -857,7 +896,7 @@ sub write_user_config {
 sub roles {
     my ($cfg, $user, $path) = @_;
 
-    # NOTE: we do not consider pools here. 
+    # NOTE: we do not consider pools here.
     # You need to use $rpcenv->roles() instead if you want that.
 
     return 'Administrator' if $user eq 'root@pam'; # root can do anything
@@ -912,7 +951,7 @@ sub roles {
 
     return ('NoAccess') if defined ($perm->{NoAccess});
     #return () if defined ($perm->{NoAccess});
-   
+
     #print "permission $user $path = " . Dumper ($perm);
 
     my @ra = keys %$perm;
@@ -921,7 +960,7 @@ sub roles {
 
     return @ra;
 }
-    
+
 sub permission {
     my ($cfg, $user, $path) = @_;
 
@@ -929,7 +968,7 @@ sub permission {
     return {} if !$user;
 
     my @ra = roles($cfg, $user, $path);
-    
+
     my $privs = {};
 
     foreach my $role (@ra) {
@@ -959,4 +998,104 @@ sub check_permissions {
     return 1;
 }
 
+sub remove_vm_access {
+    my ($vmid) = @_;
+    my $delVMaccessFn = sub {
+        my $usercfg = cfs_read_file("user.cfg");
+       my $modified;
+
+        if (my $acl = $usercfg->{acl}->{"/vms/$vmid"}) {
+            delete $usercfg->{acl}->{"/vms/$vmid"};
+           $modified = 1;
+        }
+        if (my $pool = $usercfg->{vms}->{$vmid}) {
+            if (my $data = $usercfg->{pools}->{$pool}) {
+                delete $data->{vms}->{$vmid};
+                delete $usercfg->{vms}->{$vmid};
+               $modified = 1;
+            }
+        }
+       cfs_write_file("user.cfg", $usercfg) if $modified;
+    };
+
+    lock_user_config($delVMaccessFn, "access permissions cleanup for VM $vmid failed");
+}
+
+sub remove_storage_access {
+    my ($storeid) = @_;
+
+    my $deleteStorageAccessFn = sub {
+        my $usercfg = cfs_read_file("user.cfg");
+       my $modified;
+
+        if (my $storage = $usercfg->{acl}->{"/storage/$storeid"}) {
+            delete $usercfg->{acl}->{"/storage/$storeid"};
+            $modified = 1;
+        }
+       foreach my $pool (keys %{$usercfg->{pools}}) {
+           delete $usercfg->{pools}->{$pool}->{storage}->{$storeid};
+           $modified = 1;
+       }
+        cfs_write_file("user.cfg", $usercfg) if $modified;
+    };
+
+    lock_user_config($deleteStorageAccessFn,
+                    "access permissions cleanup for storage $storeid failed");
+}
+
+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);
+       }
+    };
+
+    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);
+           }
+       }
+    };
+
+    lock_user_config($delVMfromPoolFn, "pool cleanup for VM $vmid failed");
+}
+
+# bash completion helpers
+
+sub complete_username {
+
+    my $user_cfg = cfs_read_file('user.cfg');
+
+    return [ keys %{$user_cfg->{users}} ];
+}
+
+sub complete_group {
+
+    my $user_cfg = cfs_read_file('user.cfg');
+
+    return [ keys %{$user_cfg->{groups}} ];
+}
+
+sub complete_realm {
+
+    my $domain_cfg = cfs_read_file('domains.cfg');
+
+    return [ keys %{$domain_cfg->{ids}} ];
+}
+
 1;