X-Git-Url: https://git.proxmox.com/?p=pve-common.git;a=blobdiff_plain;f=src%2FPVE%2FTools.pm;h=d5373a4246062887bd778f35f4d2ac9cf2bb8bd1;hp=6d579d8f9cd1234cecafd986e60cf9eaec7a6c5b;hb=4fdc9659e9aa86fee1a88afd79e879333618832d;hpb=590b924e19426c93e0c7f4b969ff80eed66ab29e diff --git a/src/PVE/Tools.pm b/src/PVE/Tools.pm index 6d579d8..d5373a4 100644 --- a/src/PVE/Tools.pm +++ b/src/PVE/Tools.pm @@ -369,6 +369,7 @@ sub run_command { my $afterfork; my $noerr; my $keeplocale; + my $quiet; eval { @@ -395,6 +396,8 @@ sub run_command { $noerr = $param{$p}; } elsif ($p eq 'keeplocale') { $keeplocale = $param{$p}; + } elsif ($p eq 'quiet') { + $quiet = $param{$p}; } else { die "got unknown parameter '$p' for run_command\n"; } @@ -497,7 +500,7 @@ sub run_command { waitpid ($pid, 0); die $err; } - } else { + } elsif (!$quiet) { print $buf; *STDOUT->flush(); } @@ -517,7 +520,7 @@ sub run_command { waitpid ($pid, 0); die $err; } - } else { + } elsif (!$quiet) { print STDERR $buf; *STDERR->flush(); } @@ -978,49 +981,16 @@ sub run_fork { sub df { my ($path, $timeout) = @_; - my $res = { - total => 0, - used => 0, - avail => 0, - }; - - my $pipe = IO::Pipe->new(); - my $child = fork(); - if (!defined($child)) { - warn "fork failed: $!\n"; - return $res; - } + my $df = sub { return Filesys::Df::df($path, 1) }; - if (!$child) { - $pipe->writer(); - eval { - my $df = Filesys::Df::df($path, 1); - print {$pipe} "$df->{blocks}\n$df->{used}\n$df->{bavail}\n" - if defined($df); - $pipe->close(); - }; - if (my $err = $@) { - warn $err; - POSIX::_exit(1); - } - POSIX::_exit(0); - } - - $pipe->reader(); + my $res = eval { run_fork_with_timeout($timeout, $df) } // {}; + warn $@ if $@; - my $readvalues = sub { - $res->{total} = int(((<$pipe> // 0) =~ /^(\d*)$/)[0]); - $res->{used} = int(((<$pipe> // 0) =~ /^(\d*)$/)[0]); - $res->{avail} = int(((<$pipe> // 0) =~ /^(\d*)$/)[0]); - }; - eval { - run_with_timeout($timeout, $readvalues); + return { + total => $res->{blocks} // 0, + used => $res->{used} // 0, + avail => $res->{bavail} // 0, }; - warn $@ if $@; - $pipe->close(); - kill('KILL', $child); - waitpid($child, 0); - return $res; } # UPID helper @@ -1629,9 +1599,16 @@ sub encrypt_pw { } # intended usage: convert_size($val, "kb" => "gb") -# on reduction (converting to a bigger unit) we round up by default if -# information got lost. E.g. `convert_size(1023, "b" => "kb")` returns 1 +# we round up to the next integer by default +# E.g. `convert_size(1023, "b" => "kb")` returns 1 # use $no_round_up to switch this off, above example would then return 0 +# this is also true for converting down e.g. 0.0005 gb to mb returns 1 +# (0 if $no_round_up is true) +# allowed formats for value: +# 1234 +# 1234. +# 1234.1234 +# .1234 sub convert_size { my ($value, $from, $to, $no_round_up) = @_; @@ -1644,21 +1621,22 @@ sub convert_size { pb => 5, }; - $from = lc($from); $to = lc($to); + die "no value given" + if !defined($value) || $value eq ""; + + $from = lc($from // ''); $to = lc($to // ''); die "unknown 'from' and/or 'to' units ($from => $to)" - if !(defined($units->{$from}) && defined($units->{$to})); + if !defined($units->{$from}) || !defined($units->{$to}); - my $shift_amount = $units->{$from} - $units->{$to}; + die "value '$value' is not a valid, positive number" + if $value !~ m/^(?:[0-9]+\.?[0-9]*|[0-9]*\.[0-9]+)$/; - if ($shift_amount > 0) { - $value <<= ($shift_amount * 10); - } elsif ($shift_amount < 0) { - my $remainder = ($value & (1 << abs($shift_amount)*10) - 1); - $value >>= abs($shift_amount) * 10; - $value++ if $remainder && !$no_round_up; - } + my $shift_amount = ($units->{$from} - $units->{$to}) * 10; + + $value *= 2**$shift_amount; + $value++ if !$no_round_up && ($value - int($value)) > 0.0; - return $value; + return int($value); } 1;