ifdef some functions that depend on other TWRP parts
End goal is to create a twrpTar binary that can run from the
command line.
Change-Id: I14f07a425b7755bd09dbce95a5918da1430964fc
diff --git a/twrp-functions.cpp b/twrp-functions.cpp
index bbd3c4c..5145b26 100644
--- a/twrp-functions.cpp
+++ b/twrp-functions.cpp
@@ -18,7 +18,7 @@
#include <stdio.h>
#include <stdlib.h>
-#include <string.h>
+#include <string>
#include <unistd.h>
#include <vector>
#include <dirent.h>
@@ -32,18 +32,20 @@
#include <sys/vfs.h>
#include <sys/types.h>
#include <sys/wait.h>
-#ifdef ANDROID_RB_POWEROFF
- #include "cutils/android_reboot.h"
-#endif
#include <iostream>
#include <fstream>
#include <sstream>
#include "twrp-functions.hpp"
-#include "partitions.hpp"
#include "twcommon.h"
+#ifndef BUILD_TWRPTAR_MAIN
#include "data.hpp"
+#include "partitions.hpp"
#include "variables.h"
#include "bootloader.h"
+#ifdef ANDROID_RB_POWEROFF
+ #include "cutils/android_reboot.h"
+#endif
+#endif // ndef BUILD_TWRPTAR_MAIN
#ifndef TW_EXCLUDE_ENCRYPTED_BACKUPS
#include "openaes/inc/oaes_lib.h"
#endif
@@ -115,6 +117,165 @@
return Path;
}
+int TWFunc::Wait_For_Child(pid_t pid, int *status, string Child_Name) {
+ pid_t rc_pid;
+
+ rc_pid = waitpid(pid, status, 0);
+ if (rc_pid > 0) {
+ if (WEXITSTATUS(*status) == 0)
+ LOGINFO("%s process ended with RC=%d\n", Child_Name.c_str(), WEXITSTATUS(*status)); // Success
+ else if (WIFSIGNALED(*status)) {
+ LOGINFO("%s process ended with signal: %d\n", Child_Name.c_str(), WTERMSIG(*status)); // Seg fault or some other non-graceful termination
+ return -1;
+ } else if (WEXITSTATUS(*status) != 0) {
+ LOGINFO("%s process ended with ERROR=%d\n", Child_Name.c_str(), WEXITSTATUS(*status)); // Graceful exit, but there was an error
+ return -1;
+ }
+ } else { // no PID returned
+ if (errno == ECHILD)
+ LOGINFO("%s no child process exist\n", Child_Name.c_str());
+ else {
+ LOGINFO("%s Unexpected error\n", Child_Name.c_str());
+ return -1;
+ }
+ }
+ return 0;
+}
+
+bool TWFunc::Path_Exists(string Path) {
+ // Check to see if the Path exists
+ struct stat st;
+ if (stat(Path.c_str(), &st) != 0)
+ return false;
+ else
+ return true;
+}
+
+int TWFunc::Get_File_Type(string fn) {
+ string::size_type i = 0;
+ int firstbyte = 0, secondbyte = 0;
+ char header[3];
+
+ ifstream f;
+ f.open(fn.c_str(), ios::in | ios::binary);
+ f.get(header, 3);
+ f.close();
+ firstbyte = header[i] & 0xff;
+ secondbyte = header[++i] & 0xff;
+
+ if (firstbyte == 0x1f && secondbyte == 0x8b)
+ return 1; // Compressed
+ else if (firstbyte == 0x4f && secondbyte == 0x41)
+ return 2; // Encrypted
+ else
+ return 0; // Unknown
+
+ return 0;
+}
+
+int TWFunc::Try_Decrypting_File(string fn, string password) {
+#ifndef TW_EXCLUDE_ENCRYPTED_BACKUPS
+ OAES_CTX * ctx = NULL;
+ uint8_t _key_data[32] = "";
+ FILE *f;
+ uint8_t buffer[4096];
+ uint8_t *buffer_out = NULL;
+ uint8_t *ptr = NULL;
+ size_t read_len = 0, out_len = 0;
+ int firstbyte = 0, secondbyte = 0, key_len;
+ size_t _j = 0;
+ size_t _key_data_len = 0;
+
+ // mostly kanged from OpenAES oaes.c
+ for( _j = 0; _j < 32; _j++ )
+ _key_data[_j] = _j + 1;
+ _key_data_len = password.size();
+ if( 16 >= _key_data_len )
+ _key_data_len = 16;
+ else if( 24 >= _key_data_len )
+ _key_data_len = 24;
+ else
+ _key_data_len = 32;
+ memcpy(_key_data, password.c_str(), password.size());
+
+ ctx = oaes_alloc();
+ if (ctx == NULL) {
+ LOGERR("Failed to allocate OAES\n");
+ return -1;
+ }
+
+ oaes_key_import_data(ctx, _key_data, _key_data_len);
+
+ f = fopen(fn.c_str(), "rb");
+ if (f == NULL) {
+ LOGERR("Failed to open '%s' to try decrypt\n", fn.c_str());
+ return -1;
+ }
+ read_len = fread(buffer, sizeof(uint8_t), 4096, f);
+ if (read_len <= 0) {
+ LOGERR("Read size during try decrypt failed\n");
+ fclose(f);
+ return -1;
+ }
+ if (oaes_decrypt(ctx, buffer, read_len, NULL, &out_len) != OAES_RET_SUCCESS) {
+ LOGERR("Error: Failed to retrieve required buffer size for trying decryption.\n");
+ fclose(f);
+ return -1;
+ }
+ buffer_out = (uint8_t *) calloc(out_len, sizeof(char));
+ if (buffer_out == NULL) {
+ LOGERR("Failed to allocate output buffer for try decrypt.\n");
+ fclose(f);
+ return -1;
+ }
+ if (oaes_decrypt(ctx, buffer, read_len, buffer_out, &out_len) != OAES_RET_SUCCESS) {
+ LOGERR("Failed to decrypt file '%s'\n", fn.c_str());
+ fclose(f);
+ free(buffer_out);
+ return 0;
+ }
+ fclose(f);
+ if (out_len < 2) {
+ LOGINFO("Successfully decrypted '%s' but read length %i too small.\n", fn.c_str(), out_len);
+ free(buffer_out);
+ return 1; // Decrypted successfully
+ }
+ ptr = buffer_out;
+ firstbyte = *ptr & 0xff;
+ ptr++;
+ secondbyte = *ptr & 0xff;
+ if (firstbyte == 0x1f && secondbyte == 0x8b) {
+ LOGINFO("Successfully decrypted '%s' and file is compressed.\n", fn.c_str());
+ free(buffer_out);
+ return 3; // Compressed
+ }
+ if (out_len >= 262) {
+ ptr = buffer_out + 257;
+ if (strncmp((char*)ptr, "ustar", 5) == 0) {
+ LOGINFO("Successfully decrypted '%s' and file is tar format.\n", fn.c_str());
+ free(buffer_out);
+ return 2; // Tar
+ }
+ }
+ free(buffer_out);
+ LOGINFO("No errors decrypting '%s' but no known file format.\n", fn.c_str());
+ return 1; // Decrypted successfully
+#else
+ LOGERR("Encrypted backup support not included.\n");
+ return -1;
+#endif
+}
+
+unsigned long TWFunc::Get_File_Size(string Path) {
+ struct stat st;
+
+ if (stat(Path.c_str(), &st) != 0)
+ return 0;
+ return st.st_size;
+}
+
+#ifndef BUILD_TWRPTAR_MAIN
+
// Returns "/path" from a full /path/to/file.name
string TWFunc::Get_Root_Path(string Path) {
string Local_Path = Path;
@@ -206,15 +367,6 @@
return true;
}
-bool TWFunc::Path_Exists(string Path) {
- // Check to see if the Path exists
- struct stat st;
- if (stat(Path.c_str(), &st) != 0)
- return false;
- else
- return true;
-}
-
void TWFunc::GUI_Operation_Text(string Read_Value, string Default_Text) {
string Display_Text;
@@ -237,14 +389,6 @@
DataManager::SetValue("tw_partition", Partition_Name);
}
-unsigned long TWFunc::Get_File_Size(string Path) {
- struct stat st;
-
- if (stat(Path.c_str(), &st) != 0)
- return 0;
- return st.st_size;
-}
-
void TWFunc::Copy_Log(string Source, string Destination) {
PartitionManager.Mount_By_Path(Destination, false);
FILE *destination_log = fopen(Destination.c_str(), "a");
@@ -478,30 +622,6 @@
return -1;
}
-vector<string> TWFunc::split_string(const string &in, char del, bool skip_empty) {
- vector<string> res;
-
- if (in.empty() || del == '\0')
- return res;
-
- string field;
- istringstream f(in);
- if (del == '\n') {
- while(getline(f, field)) {
- if (field.empty() && skip_empty)
- continue;
- res.push_back(field);
- }
- } else {
- while(getline(f, field, del)) {
- if (field.empty() && skip_empty)
- continue;
- res.push_back(field);
- }
- }
- return res;
-}
-
timespec TWFunc::timespec_diff(timespec& start, timespec& end)
{
timespec temp;
@@ -810,121 +930,6 @@
return true;
}
-int TWFunc::Get_File_Type(string fn) {
- string::size_type i = 0;
- int firstbyte = 0, secondbyte = 0;
- char header[3];
-
- ifstream f;
- f.open(fn.c_str(), ios::in | ios::binary);
- f.get(header, 3);
- f.close();
- firstbyte = header[i] & 0xff;
- secondbyte = header[++i] & 0xff;
-
- if (firstbyte == 0x1f && secondbyte == 0x8b)
- return 1; // Compressed
- else if (firstbyte == 0x4f && secondbyte == 0x41)
- return 2; // Encrypted
- else
- return 0; // Unknown
-
- return 0;
-}
-
-int TWFunc::Try_Decrypting_File(string fn, string password) {
-#ifndef TW_EXCLUDE_ENCRYPTED_BACKUPS
- OAES_CTX * ctx = NULL;
- uint8_t _key_data[32] = "";
- FILE *f;
- uint8_t buffer[4096];
- uint8_t *buffer_out = NULL;
- uint8_t *ptr = NULL;
- size_t read_len = 0, out_len = 0;
- int firstbyte = 0, secondbyte = 0, key_len;
- size_t _j = 0;
- size_t _key_data_len = 0;
-
- // mostly kanged from OpenAES oaes.c
- for( _j = 0; _j < 32; _j++ )
- _key_data[_j] = _j + 1;
- _key_data_len = password.size();
- if( 16 >= _key_data_len )
- _key_data_len = 16;
- else if( 24 >= _key_data_len )
- _key_data_len = 24;
- else
- _key_data_len = 32;
- memcpy(_key_data, password.c_str(), password.size());
-
- ctx = oaes_alloc();
- if (ctx == NULL) {
- LOGERR("Failed to allocate OAES\n");
- return -1;
- }
-
- oaes_key_import_data(ctx, _key_data, _key_data_len);
-
- f = fopen(fn.c_str(), "rb");
- if (f == NULL) {
- LOGERR("Failed to open '%s' to try decrypt\n", fn.c_str());
- return -1;
- }
- read_len = fread(buffer, sizeof(uint8_t), 4096, f);
- if (read_len <= 0) {
- LOGERR("Read size during try decrypt failed\n");
- fclose(f);
- return -1;
- }
- if (oaes_decrypt(ctx, buffer, read_len, NULL, &out_len) != OAES_RET_SUCCESS) {
- LOGERR("Error: Failed to retrieve required buffer size for trying decryption.\n");
- fclose(f);
- return -1;
- }
- buffer_out = (uint8_t *) calloc(out_len, sizeof(char));
- if (buffer_out == NULL) {
- LOGERR("Failed to allocate output buffer for try decrypt.\n");
- fclose(f);
- return -1;
- }
- if (oaes_decrypt(ctx, buffer, read_len, buffer_out, &out_len) != OAES_RET_SUCCESS) {
- LOGERR("Failed to decrypt file '%s'\n", fn.c_str());
- fclose(f);
- free(buffer_out);
- return 0;
- }
- fclose(f);
- if (out_len < 2) {
- LOGINFO("Successfully decrypted '%s' but read length %i too small.\n", fn.c_str(), out_len);
- free(buffer_out);
- return 1; // Decrypted successfully
- }
- ptr = buffer_out;
- firstbyte = *ptr & 0xff;
- ptr++;
- secondbyte = *ptr & 0xff;
- if (firstbyte == 0x1f && secondbyte == 0x8b) {
- LOGINFO("Successfully decrypted '%s' and file is compressed.\n", fn.c_str());
- free(buffer_out);
- return 3; // Compressed
- }
- if (out_len >= 262) {
- ptr = buffer_out + 257;
- if (strncmp((char*)ptr, "ustar", 5) == 0) {
- LOGINFO("Successfully decrypted '%s' and file is tar format.\n", fn.c_str());
- free(buffer_out);
- return 2; // Tar
- }
- }
- free(buffer_out);
- LOGINFO("No errors decrypting '%s' but no known file format.\n", fn.c_str());
- return 1; // Decrypted successfully
-#else
- LOGERR("Encrypted backup support not included.\n");
- return -1;
-#endif
-}
-
bool TWFunc::Try_Decrypting_Backup(string Restore_Path, string Password) {
DIR* d;
@@ -953,31 +958,6 @@
return true;
}
-int TWFunc::Wait_For_Child(pid_t pid, int *status, string Child_Name) {
- pid_t rc_pid;
-
- rc_pid = waitpid(pid, status, 0);
- if (rc_pid > 0) {
- if (WEXITSTATUS(*status) == 0)
- LOGINFO("%s process ended with RC=%d\n", Child_Name.c_str(), WEXITSTATUS(*status)); // Success
- else if (WIFSIGNALED(*status)) {
- LOGINFO("%s process ended with signal: %d\n", Child_Name.c_str(), WTERMSIG(*status)); // Seg fault or some other non-graceful termination
- return -1;
- } else if (WEXITSTATUS(*status) != 0) {
- LOGINFO("%s process ended with ERROR=%d\n", Child_Name.c_str(), WEXITSTATUS(*status)); // Graceful exit, but there was an error
- return -1;
- }
- } else { // no PID returned
- if (errno == ECHILD)
- LOGINFO("%s no child process exist\n", Child_Name.c_str());
- else {
- LOGINFO("%s Unexpected error\n", Child_Name.c_str());
- return -1;
- }
- }
- return 0;
-}
-
string TWFunc::Get_Current_Date() {
string Current_Date;
time_t seconds = time(0);
@@ -1132,3 +1112,5 @@
settimeofday(&tv, NULL);
#endif
}
+
+#endif // ndef BUILD_TWRPTAR_MAIN