• nkk71's avatar
    Add MultiROM support to standard TWRP tree · 09e43081
    nkk71 authored
    Squashed commit that allows MultiROM TWRP to be built from
    within the standard TWRP environment.
    
    MultiROM build flags and TARGET_RECOVERY_IS_MULTIROM
    must be defined in device tree.
    
    -------------------------------------
    
    MultiROM-support:
    - Move files to separate folder
      * Move primary multirom files to /recovery/multirom
      * Move multirom prebuilt files to /recovery/multirom/prebuilt
      * Move cp_xattrs folder used by multirom to /recovery/multirom/cp_xattrs
    - Add Android.mk for mrom-prebuilt files
    - Add Android.mk for mrom-cp_xattrs files
    - Fix include paths
      * fix includes in multirom.h/c/hpp/cpp files to go to parent (main recovery folder)
    - Add read_data function
      * Add read_data function used by multirom
      * Don't append to minzip, use a seperate file
    - Remove deprecated 'flags' variable
      * Remove deprecated 'flags' in function call to mzExtractRecursive
      * Ref: https://github.com/omnirom/android_bootable_recovery/commit/9c0f5d6b348e37533bdcccf1166d6cbf1ca5c50b
    - gui: Add multirom functions
      * objects.hpp - Add multirom action definitions
      * action.cpp - Add multirom actions and functions
        Note: TW_ORS_IS_SECONDARY_ROM code moved to openrecoveryscript.cpp
      * Add TARGET_RECOVERY_IS_MULTIROM flag to Android.mk
      * Reorder code changes for easier merging
    - twrp-functions
      * Add functions needed by multirom
      * Reorder code changes for easier merging
    - partition.cpp and partitionmanager.cpp
      * Add and embed multirom specific code
      * Reorder code changes for easier merging
    - twrp.cpp
      * Add and embed multirom specific code
    - variables.h
      * Add multirom specific defines
    - Android.mk
      * update Android.mk for compartmentalized multirom
    - gui: Update fileselector
      * Add <excludes> tag
      * Add support for multiple file extensions
    - gui: Update listbox
      * Add support for dynamic <items> tag
    
    Thanks @CaptainThrowback and @z31s1g for your help
    09e43081
twrp-functions.hpp 7.44 KB
/*
	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/>.
*/

#ifndef _TWRPFUNCTIONS_HPP
#define _TWRPFUNCTIONS_HPP

#include <string>
#include <vector>

using namespace std;

typedef enum
{
	rb_current = 0,
	rb_system,
	rb_recovery,
	rb_poweroff,
	rb_bootloader,     // May also be fastboot
	rb_download,
} RebootCommand;

#ifdef TARGET_RECOVERY_IS_MULTIROM
#ifdef HAVE_SELINUX
struct selabel_handle;
#endif
#endif //TARGET_RECOVERY_IS_MULTIROM

// Partition class
class TWFunc
{
public:
	static string Get_Root_Path(string Path);                                   // Trims any trailing folders or filenames from the path, also adds a leading / if not present
	static string Get_Path(string Path);                                        // Trims everything after the last / in the string
	static string Get_Filename(string Path);                                    // Trims the path off of a filename

	static int Exec_Cmd(const string& cmd, string &result);                     //execute a command and return the result as a string by reference
	static int Exec_Cmd(const string& cmd);                                     //execute a command
#ifdef TARGET_RECOVERY_IS_MULTIROM
	static int Exec_Cmd_Show_Output(const string& cmd);
#endif //TARGET_RECOVERY_IS_MULTIROM
	static int Wait_For_Child(pid_t pid, int *status, string Child_Name);       // Waits for pid to exit and checks exit status
	static bool Path_Exists(string Path);                                       // Returns true if the path exists
	static int Get_File_Type(string fn); // Determines file type, 0 for unknown, 1 for gzip, 2 for OAES encrypted
	static int Try_Decrypting_File(string fn, string password); // -1 for some error, 0 for failed to decrypt, 1 for decrypted, 3 for decrypted and found gzip format
	static unsigned long Get_File_Size(string Path);                            // Returns the size of a file
	static std::string Remove_Trailing_Slashes(const std::string& path, bool leaveLast = false); // Normalizes the path, e.g /data//media/ -> /data/media
	static vector<string> split_string(const string &in, char del, bool skip_empty);

#ifndef BUILD_TWRPTAR_MAIN
	static void install_htc_dumlock(void);                                      // Installs HTC Dumlock
	static void htc_dumlock_restore_original_boot(void);                        // Restores the backup of boot from HTC Dumlock
	static void htc_dumlock_reflash_recovery_to_boot(void);                     // Reflashes the current recovery to boot
	static int Recursive_Mkdir(string Path);                                    // Recursively makes the entire path
	static void GUI_Operation_Text(string Read_Value, string Default_Text);     // Updates text for display in the GUI, e.g. Backing up %partition name%
	static void GUI_Operation_Text(string Read_Value, string Partition_Name, string Default_Text); // Same as above but includes partition name
	static void Update_Log_File(void);                                          // Writes the log to last_log
	static void Update_Intent_File(string Intent);                              // Updates intent file
	static int tw_reboot(RebootCommand command);                                // Prepares the device for rebooting
	static void check_and_run_script(const char* script_file, const char* display_name); // checks for the existence of a script, chmods it to 755, then runs it
	static int removeDir(const string path, bool removeParent); //recursively remove a directory
	static int copy_file(string src, string dst, int mode); //copy file from src to dst with mode permissions
	static unsigned int Get_D_Type_From_Stat(string Path);                      // Returns a dirent dt_type value using stat instead of dirent
	static timespec timespec_diff(timespec& start, timespec& end);	            // Return a diff for 2 times
	static int32_t timespec_diff_ms(timespec& start, timespec& end);            // Returns diff in ms
	static int read_file(string fn, vector<string>& results); //read from file
	static int read_file(string fn, string& results); //read from file
	static int read_file(string fn, uint64_t& results); //read from file
	static int write_file(string fn, string& line); //write from file
#ifdef TARGET_RECOVERY_IS_MULTIROM
	static int write_file(string fn, const string& line); //write from file
	static int write_file(string fn, const string& line, const char *mode); //write from file
#endif //TARGET_RECOVERY_IS_MULTIROM
	static bool Install_SuperSU(void); // Installs su binary and apk and sets proper permissions
	static bool Try_Decrypting_Backup(string Restore_Path, string Password); // true for success, false for failed to decrypt
	static string System_Property_Get(string Prop_Name);                // Returns value of Prop_Name from reading /system/build.prop
	static string Get_Current_Date(void);                               // Returns the current date in ccyy-m-dd--hh-nn-ss format
	static void Auto_Generate_Backup_Name();                            // Populates TW_BACKUP_NAME with a backup name based on current date and ro.build.display.id from /system/build.prop
	static void Fixup_Time_On_Boot(); // Fixes time on devices which need it
	static std::vector<std::string> Split_String(const std::string& str, const std::string& delimiter, bool removeEmpty = true); // Splits string by delimiter
	static bool Create_Dir_Recursive(const std::string& path, mode_t mode = 0755, uid_t uid = -1, gid_t gid = -1);  // Create directory and it's parents, if they don't exist. mode, uid and gid are set to all _newly_ created folders. If whole path exists, do nothing.
	static int Set_Brightness(std::string brightness_value); // Well, you can read, it does what it says, passing return int from TWFunc::Write_File ;)
	static bool Toggle_MTP(bool enable);                                        // Disables MTP if enable is false and re-enables MTP if enable is true and it was enabled the last time it was toggled off
	static std::string to_string(unsigned long value); //convert ul to string
	static void SetPerformanceMode(bool mode); // support recovery.perf.mode
	static void Disable_Stock_Recovery_Replace(); // Disable stock ROMs from replacing TWRP with stock recovery
#ifdef TARGET_RECOVERY_IS_MULTIROM
	static bool loadTheme();
	static bool reloadTheme();
	static std::string getDefaultThemePath(int rotation);
	static std::string getZIPThemePath(int rotation);
	static std::string getROMName();
	static void stringReplace(std::string& str, char before, char after);
	static void trim(std::string& str);
	static int64_t getFreeSpace(const std::string& path);
#ifdef HAVE_SELINUX
	static bool restorecon(const std::string& path, struct selabel_handle *sh);
#endif
#endif //TARGET_RECOVERY_IS_MULTIROM
	static unsigned long long IOCTL_Get_Block_Size(const char* block_device);

private:
	static void Copy_Log(string Source, string Destination);

};

extern int Log_Offset;
#else
};
#endif // ndef BUILD_TWRPTAR_MAIN

#endif // _TWRPFUNCTIONS_HPP