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

PBSClient: add file_restore_extract function

Browse Source 
*_prepare creates a fifo for streaming data back to clients directly,
filefile_restore_extract blocks and should be called from a background
worker - while it is running outcoming data can be read from the FIFO.

Signed-off-by: Stefan Reiter <s.reiter@proxmox.com>
This commit is contained in:
Stefan Reiter 2021-04-22 17:34:50 +02:00 committed by Thomas Lamprecht
parent 67252649d9
commit 77e402f085
1 changed files with 55 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
src/PVE/PBSClient.pm
View File

@ -5,9 +5,10 @@ use strict;
use warnings;
use Fcntl qw(F_GETFD F_SETFD FD_CLOEXEC);
use File::Temp qw(tempdir);
use IO::File;
use JSON;
use POSIX qw(strftime ENOENT);
use POSIX qw(mkfifo strftime ENOENT);
use PVE::JSONSchema qw(get_standard_option);
use PVE::Tools qw(run_command file_set_contents file_get_contents file_read_firstline $IPV6RE);
@ -346,4 +347,57 @@ sub file_restore_list {
);
}
# call sync from API, returns a fifo path for streaming data to clients,
# pass it to file_restore_extract to start transfering data
sub file_restore_extract_prepare {
my ($self) = @_;
my $tmpdir = tempdir();
mkfifo("$tmpdir/fifo", 0600)
or die "creating file download fifo '$tmpdir/fifo' failed: $!\n";
# allow reading data for proxy user
my $wwwid = getpwnam('www-data') ||
die "getpwnam failed";
chown $wwwid, -1, "$tmpdir"
or die "changing permission on fifo dir '$tmpdir' failed: $!\n";
chown $wwwid, -1, "$tmpdir/fifo"
or die "changing permission on fifo '$tmpdir/fifo' failed: $!\n";
return "$tmpdir/fifo";
}
# this blocks while data is transfered, call this from a background worker
sub file_restore_extract {
my ($self, $output_file, $snapshot, $filepath, $base64) = @_;
my $ret = eval {
local $SIG{ALRM} = sub { die "got timeout\n" };
alarm(30);
sysopen(my $fh, "$output_file", O_WRONLY)
or die "open target '$output_file' for writing failed: $!\n";
alarm(0);
my $fn = fileno($fh);
my $errfunc = sub { print $_[0], "\n"; };
return run_raw_client_cmd(
$self,
"extract",
[ $snapshot, $filepath, "-", "--base64", $base64 ? 1 : 0 ],
binary => "proxmox-file-restore",
errfunc => $errfunc,
output => ">&$fn",
);
};
my $err = $@;
unlink($output_file);
$output_file =~ s/fifo$//;
rmdir($output_file) if -d $output_file;
die "file restore task failed: $err" if $err;
return $ret;
}
1;