Dees Troy | 3be70a8 | 2013-10-22 14:25:12 +0000 | [diff] [blame] | 1 | /* |
| 2 | Copyright 2012 bigbiff/Dees_Troy TeamWin |
| 3 | This file is part of TWRP/TeamWin Recovery Project. |
| 4 | |
| 5 | TWRP is free software: you can redistribute it and/or modify |
| 6 | it under the terms of the GNU General Public License as published by |
| 7 | the Free Software Foundation, either version 3 of the License, or |
| 8 | (at your option) any later version. |
| 9 | |
| 10 | TWRP is distributed in the hope that it will be useful, |
| 11 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 13 | GNU General Public License for more details. |
| 14 | |
| 15 | You should have received a copy of the GNU General Public License |
| 16 | along with TWRP. If not, see <http://www.gnu.org/licenses/>. |
| 17 | */ |
| 18 | |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 19 | #ifndef _TWRPFUNCTIONS_HPP |
| 20 | #define _TWRPFUNCTIONS_HPP |
| 21 | |
| 22 | #include <string> |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 23 | #include <vector> |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 24 | |
| 25 | using namespace std; |
| 26 | |
| 27 | typedef enum |
| 28 | { |
Vojtech Bocek | fafb0c5 | 2013-07-25 22:53:02 +0200 | [diff] [blame] | 29 | rb_current = 0, |
| 30 | rb_system, |
| 31 | rb_recovery, |
| 32 | rb_poweroff, |
| 33 | rb_bootloader, // May also be fastboot |
| 34 | rb_download, |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 35 | } RebootCommand; |
| 36 | |
| 37 | // Partition class |
| 38 | class TWFunc |
| 39 | { |
| 40 | public: |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 41 | static string Get_Root_Path(string Path); // Trims any trailing folders or filenames from the path, also adds a leading / if not present |
| 42 | static string Get_Path(string Path); // Trims everything after the last / in the string |
| 43 | static string Get_Filename(string Path); // Trims the path off of a filename |
| 44 | |
Ethan Yonker | af2897c | 2014-02-10 13:07:14 -0600 | [diff] [blame] | 45 | static int Exec_Cmd(const string& cmd, string &result); //execute a command and return the result as a string by reference |
| 46 | static int Exec_Cmd(const string& cmd); //execute a command |
| 47 | static int Wait_For_Child(pid_t pid, int *status, string Child_Name); // Waits for pid to exit and checks exit status |
| 48 | static bool Path_Exists(string Path); // Returns true if the path exists |
| 49 | static int Get_File_Type(string fn); // Determines file type, 0 for unknown, 1 for gzip, 2 for OAES encrypted |
| 50 | 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 |
| 51 | static unsigned long Get_File_Size(string Path); // Returns the size of a file |
Vojtech Bocek | 05f87d6 | 2014-03-11 22:08:23 +0100 | [diff] [blame] | 52 | static std::string Remove_Trailing_Slashes(const std::string& path, bool leaveLast = false); // Normalizes the path, e.g /data//media/ -> /data/media |
Ethan Yonker | 1b7a31b | 2014-07-03 15:09:22 -0500 | [diff] [blame^] | 53 | static vector<string> split_string(const string &in, char del, bool skip_empty); |
Ethan Yonker | af2897c | 2014-02-10 13:07:14 -0600 | [diff] [blame] | 54 | |
| 55 | #ifndef BUILD_TWRPTAR_MAIN |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 56 | static void install_htc_dumlock(void); // Installs HTC Dumlock |
| 57 | static void htc_dumlock_restore_original_boot(void); // Restores the backup of boot from HTC Dumlock |
| 58 | static void htc_dumlock_reflash_recovery_to_boot(void); // Reflashes the current recovery to boot |
| 59 | static int Recursive_Mkdir(string Path); // Recursively makes the entire path |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 60 | static void GUI_Operation_Text(string Read_Value, string Default_Text); // Updates text for display in the GUI, e.g. Backing up %partition name% |
| 61 | static void GUI_Operation_Text(string Read_Value, string Partition_Name, string Default_Text); // Same as above but includes partition name |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 62 | static void Update_Log_File(void); // Writes the log to last_log |
| 63 | static void Update_Intent_File(string Intent); // Updates intent file |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 64 | static int tw_reboot(RebootCommand command); // Prepares the device for rebooting |
| 65 | 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 |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 66 | static int removeDir(const string path, bool removeParent); //recursively remove a directory |
| 67 | static int copy_file(string src, string dst, int mode); //copy file from src to dst with mode permissions |
| 68 | static unsigned int Get_D_Type_From_Stat(string Path); // Returns a dirent dt_type value using stat instead of dirent |
| 69 | static timespec timespec_diff(timespec& start, timespec& end); // Return a diff for 2 times |
Vojtech Bocek | e5ffcd1 | 2014-02-06 21:17:32 +0100 | [diff] [blame] | 70 | static int32_t timespec_diff_ms(timespec& start, timespec& end); // Returns diff in ms |
bigbiff bigbiff | cdcfee4 | 2013-02-27 21:11:26 -0500 | [diff] [blame] | 71 | static int read_file(string fn, vector<string>& results); //read from file |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 72 | static int read_file(string fn, string& results); //read from file |
| 73 | static int write_file(string fn, string& line); //write from file |
| 74 | static int drop_caches(void); //drop linux cache memory |
| 75 | static int Check_su_Perms(void); // check perms and owner of su binary in various locations |
| 76 | static bool Fix_su_Perms(void); // sets proper permissions for su binaries and superuser apk |
Vojtech Bocek | 37aeb8d | 2013-08-29 22:38:20 +0200 | [diff] [blame] | 77 | static int tw_chmod(const string& fn, const string& mode); // chmod function that converts a 3 or 4 char string into st_mode automatically |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 78 | static bool Install_SuperSU(void); // Installs su binary and apk and sets proper permissions |
Dees_Troy | 83bd483 | 2013-05-04 12:39:56 +0000 | [diff] [blame] | 79 | static bool Try_Decrypting_Backup(string Restore_Path, string Password); // true for success, false for failed to decrypt |
Ethan Yonker | b555789 | 2014-02-07 21:43:20 -0600 | [diff] [blame] | 80 | static string System_Property_Get(string Prop_Name); // Returns value of Prop_Name from reading /system/build.prop |
Dees Troy | b21cc64 | 2013-09-10 17:36:41 +0000 | [diff] [blame] | 81 | static string Get_Current_Date(void); // Returns the current date in ccyy-m-dd--hh-nn-ss format |
| 82 | 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 |
Vojtech Bocek | d0e38bc | 2014-02-03 23:36:57 +0100 | [diff] [blame] | 83 | static void Fixup_Time_On_Boot(); // Fixes time on devices which need it |
Vojtech Bocek | 0b7fe50 | 2014-03-13 17:36:52 +0100 | [diff] [blame] | 84 | static std::vector<std::string> Split_String(const std::string& str, const std::string& delimiter, bool removeEmpty = true); // Splits string by delimiter |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 85 | |
| 86 | private: |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 87 | static void Copy_Log(string Source, string Destination); |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 88 | |
| 89 | }; |
| 90 | |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 91 | extern int Log_Offset; |
Ethan Yonker | af2897c | 2014-02-10 13:07:14 -0600 | [diff] [blame] | 92 | #else |
| 93 | }; |
| 94 | #endif // ndef BUILD_TWRPTAR_MAIN |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 95 | |
Dees_Troy | 6ef6635 | 2013-02-21 08:26:57 -0600 | [diff] [blame] | 96 | #endif // _TWRPFUNCTIONS_HPP |