mirror of
https://github.com/meizu-m86/twrp_multirom_m86
synced 2025-11-03 21:45:40 +08:00
472f506817
The progress bar will now be updated during image backups, restores and during image flashing (except for sparse images which will require significant changes to libsparse, and except for mtd nand using flash_utils). The progress bar will now be updated mid-file for file systems (tar) so the user will see changes even during large file backup / restore. Add a new progress tracking class to simplify handling of progress bar updates. The class will only update the progress bar 5 times a second to reduce the CPU load from updating the GUI frequently which does affect backup times. Change-Id: Iff382faef3df1f86604af336c1a8ce8993cd12c5
114 lines
2.8 KiB
C
114 lines
2.8 KiB
C
/*
|
|
Copyright 2012 bigbiff/Dees_Troy TeamWin
|
|
This file is part of TWRP/TeamWin Recovery Project.
|
|
|
|
TWRP is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
TWRP is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with TWRP. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <fcntl.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include "libtar/libtar.h"
|
|
#include "twcommon.h"
|
|
|
|
int flush = 0, eot_count = -1;
|
|
unsigned char *write_buffer;
|
|
unsigned buffer_size = 4096;
|
|
unsigned buffer_loc = 0;
|
|
int buffer_status = 0;
|
|
int prog_pipe = -1;
|
|
const unsigned long long progress_size = (unsigned long long)(T_BLOCKSIZE);
|
|
|
|
void reinit_libtar_buffer(void) {
|
|
flush = 0;
|
|
eot_count = -1;
|
|
buffer_loc = 0;
|
|
buffer_status = 1;
|
|
}
|
|
|
|
void init_libtar_buffer(unsigned new_buff_size, int pipe_fd) {
|
|
if (new_buff_size != 0)
|
|
buffer_size = new_buff_size;
|
|
|
|
reinit_libtar_buffer();
|
|
write_buffer = (unsigned char*) malloc(sizeof(char *) * buffer_size);
|
|
prog_pipe = pipe_fd;
|
|
}
|
|
|
|
void free_libtar_buffer(void) {
|
|
if (buffer_status > 0)
|
|
free(write_buffer);
|
|
buffer_status = 0;
|
|
prog_pipe = -1;
|
|
}
|
|
|
|
ssize_t write_libtar_buffer(int fd, const void *buffer, size_t size) {
|
|
void* ptr;
|
|
|
|
if (flush == 0) {
|
|
ptr = write_buffer + buffer_loc;
|
|
memcpy(ptr, buffer, size);
|
|
buffer_loc += size;
|
|
if (eot_count >= 0 && eot_count < 2)
|
|
eot_count++;
|
|
/* At the end of the tar file, libtar will add 2 blank blocks.
|
|
Once we have received both EOT blocks, we will immediately
|
|
write anything in the buffer to the file.
|
|
*/
|
|
|
|
if (buffer_loc >= buffer_size || eot_count >= 2) {
|
|
flush = 1;
|
|
}
|
|
}
|
|
if (flush == 1) {
|
|
flush = 0;
|
|
if (buffer_loc == 0) {
|
|
// nothing to write
|
|
return 0;
|
|
}
|
|
if (write(fd, write_buffer, buffer_loc) != (int)buffer_loc) {
|
|
LOGERR("Error writing tar file!\n");
|
|
buffer_loc = 0;
|
|
return -1;
|
|
} else {
|
|
unsigned long long fs = (unsigned long long)(buffer_loc);
|
|
write(prog_pipe, &fs, sizeof(fs));
|
|
buffer_loc = 0;
|
|
return size;
|
|
}
|
|
} else {
|
|
return size;
|
|
}
|
|
// Shouldn't ever get here
|
|
return -1;
|
|
}
|
|
|
|
void flush_libtar_buffer(int fd) {
|
|
eot_count = 0;
|
|
if (buffer_status)
|
|
buffer_status = 2;
|
|
}
|
|
|
|
void init_libtar_no_buffer(int pipe_fd) {
|
|
buffer_size = T_BLOCKSIZE;
|
|
prog_pipe = pipe_fd;
|
|
buffer_status = 0;
|
|
}
|
|
|
|
ssize_t write_libtar_no_buffer(int fd, const void *buffer, size_t size) {
|
|
write(prog_pipe, &progress_size, sizeof(progress_size));
|
|
return write(fd, buffer, size);
|
|
}
|