Ethan Yonker
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
105 lines
2.8 KiB
C++
105 lines
2.8 KiB
C++
/*
|
|
Copyright 2012 to 2016 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/>.
|
|
*/
|
|
|
|
extern "C" {
|
|
#include "libtar/libtar.h"
|
|
}
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <vector>
|
|
#include "twrpDU.hpp"
|
|
#include "progresstracking.hpp"
|
|
|
|
using namespace std;
|
|
|
|
struct TarListStruct {
|
|
std::string fn;
|
|
unsigned thread_id;
|
|
};
|
|
|
|
struct thread_data_struct {
|
|
std::vector<TarListStruct> *TarList;
|
|
unsigned thread_id;
|
|
};
|
|
|
|
class twrpTar {
|
|
public:
|
|
twrpTar();
|
|
virtual ~twrpTar();
|
|
int createTarFork(ProgressTracking *progress, pid_t &fork_pid);
|
|
int extractTarFork(ProgressTracking *progress);
|
|
void setfn(string fn);
|
|
void setdir(string dir);
|
|
void setsize(unsigned long long backup_size);
|
|
void setpassword(string pass);
|
|
unsigned long long get_size();
|
|
|
|
public:
|
|
int use_encryption;
|
|
int userdata_encryption;
|
|
int use_compression;
|
|
int split_archives;
|
|
int has_data_media;
|
|
string backup_name;
|
|
int progress_pipe_fd;
|
|
string partition_name;
|
|
string backup_folder;
|
|
|
|
private:
|
|
int extract();
|
|
int addFilesToExistingTar(vector <string> files, string tarFile);
|
|
int createTar();
|
|
int addFile(string fn, bool include_root);
|
|
int entryExists(string entry);
|
|
int closeTar();
|
|
int removeEOT(string tarFile);
|
|
int extractTar();
|
|
string Strip_Root_Dir(string Path);
|
|
int openTar();
|
|
int Generate_TarList(string Path, std::vector<TarListStruct> *TarList, unsigned long long *Target_Size, unsigned *thread_id);
|
|
static void* createList(void *cookie);
|
|
static void* extractMulti(void *cookie);
|
|
int tarList(std::vector<TarListStruct> *TarList, unsigned thread_id);
|
|
unsigned long long uncompressedSize(string filename, int *archive_type);
|
|
static void Signal_Kill(int signum);
|
|
|
|
int Archive_Current_Type;
|
|
unsigned long long Archive_Current_Size;
|
|
unsigned long long Total_Backup_Size;
|
|
bool include_root_dir;
|
|
TAR *t;
|
|
tartype_t tar_type; // Only used in createTar() but variable must persist while the tar is open
|
|
int fd;
|
|
pid_t pigz_pid;
|
|
pid_t oaes_pid;
|
|
unsigned long long file_count;
|
|
|
|
string tardir;
|
|
string tarfn;
|
|
string basefn;
|
|
string password;
|
|
|
|
std::vector<TarListStruct> *ItemList;
|
|
unsigned thread_id;
|
|
};
|