twrpTar.hpp

/*
        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/>.
*/

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>

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();
	int extractTarFork();
	int splitArchiveFork();
	void setexcl(string exclude);
	void setfn(string fn);
	void setdir(string dir);
	unsigned long long uncompressedSize();

public:
	int use_encryption;
	int userdata_encryption;
	int use_compression;
	int split_archives;
	int has_data_media;
	string backup_name;

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 create();
	int Split_Archive();
	int removeEOT(string tarFile);
	int extractTar();
	int tarDirs(bool include_root);
	int Generate_Multiple_Archives(string Path);
	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(bool include_root, std::vector<TarListStruct> *TarList, unsigned thread_id);

	int Archive_File_Count;
	int Archive_Current_Type;
	unsigned long long Archive_Current_Size;
	TAR *t;
	int fd;
	pid_t pigz_pid;
	pid_t oaes_pid;

	string tardir;
	string tarfn;
	string basefn;

	vector <string> tarexclude;
	vector<string> split;

	std::vector<TarListStruct> *ItemList;
	int thread_id;
};