proxmox-mirrors/pve-common
1
0
Fork 0
mirror of https://git.proxmox.com/git/pve-common synced 2025-08-14 15:16:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

tools: download from url: add option to allow overriding existing files

Browse Source 
Signed-off-by: Thomas Lamprecht <t.lamprecht@proxmox.com>
This commit is contained in:
Thomas Lamprecht 2021-06-16 12:41:33 +02:00
parent f52ecff957
commit 06c1c13f1c
1 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/PVE/Tools.pm
View File

@ -1842,6 +1842,7 @@ sub safe_compare {
# opts is a hash ref with the following known properties
# allow_overwrite - if 1, overwriting existing files is allowed, use with care. Default to false
# hash_required - if 1, at least one checksum has to be specified otherwise an error will be thrown
# http_proxy
# https_proxy
@ -1862,17 +1863,22 @@ sub download_file_from_url {
print "downloading $url to $dest\n";
if (-f $dest && $checksum_algorithm) {
print "calculating checksum of existing file...";
my $checksum_got = get_file_hash($checksum_algorithm, $dest);
if (-f $dest) {
if ($checksum_algorithm) {
print "calculating checksum of existing file...";
my $checksum_got = get_file_hash($checksum_algorithm, $dest);
if (lc($checksum_got) eq lc($checksum_expected)) {
print "OK, got correct file already, no need to download\n";
return;
} else {
# we could re-download, but may not be safe so just abort for now..
print "\n"; # the front end expects the error to reside at the last line without any noise
die "checksum mismatch: got '$checksum_got' != expect '$checksum_expected', aborting\n";
if (lc($checksum_got) eq lc($checksum_expected)) {
print "OK, got correct file already, no need to download\n";
return;
} elsif ($opts->{allow_overwrite}) {
print "checksum mismatch: got '$checksum_got' != expect '$checksum_expected', re-download\n";
} else {
print "\n"; # the front end expects the error to reside at the last line without any noise
die "checksum mismatch: got '$checksum_got' != expect '$checksum_expected', aborting\n";
}
} elsif (!$opts->{allow_overwrite}) {
die "refusing to override existing file '$dest'\n";
}
}