blob: 4a7b340a5e1324755c06b3015694c0f61b09f33f [file] [log] [blame]
Dees Troy3be70a82013-10-22 14:25:12 +00001/*
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_Troy38bd7602012-09-14 13:33:53 -040019#include <stdio.h>
20#include <stdlib.h>
Ethan Yonkeraf2897c2014-02-10 13:07:14 -060021#include <string>
Dees_Troy38bd7602012-09-14 13:33:53 -040022#include <unistd.h>
23#include <vector>
24#include <dirent.h>
25#include <time.h>
Dees_Troy43d8b002012-09-17 16:00:01 -040026#include <errno.h>
bigbiff bigbiff9c754052013-01-09 09:09:08 -050027#include <fcntl.h>
28#include <sys/mount.h>
Dees_Troya58bead2012-09-27 09:49:29 -040029#include <sys/reboot.h>
bigbiff bigbiff9c754052013-01-09 09:09:08 -050030#include <sys/sendfile.h>
31#include <sys/stat.h>
32#include <sys/vfs.h>
Dees_Troy83bd4832013-05-04 12:39:56 +000033#include <sys/types.h>
34#include <sys/wait.h>
bigbiff bigbiff9c754052013-01-09 09:09:08 -050035#include <iostream>
36#include <fstream>
Dees_Troy83bd4832013-05-04 12:39:56 +000037#include <sstream>
Dees_Troy38bd7602012-09-14 13:33:53 -040038#include "twrp-functions.hpp"
Dees_Troy2673cec2013-04-02 20:22:16 +000039#include "twcommon.h"
Ethan Yonkeraf2897c2014-02-10 13:07:14 -060040#ifndef BUILD_TWRPTAR_MAIN
Dees_Troyb46a6842012-09-25 11:06:46 -040041#include "data.hpp"
Ethan Yonkeraf2897c2014-02-10 13:07:14 -060042#include "partitions.hpp"
Dees_Troy3477d712012-09-27 15:44:01 -040043#include "variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000044#include "bootloader.h"
Tom Hite5a926722014-09-15 01:31:03 +000045#include "cutils/properties.h"
Ethan Yonkeraf2897c2014-02-10 13:07:14 -060046#ifdef ANDROID_RB_POWEROFF
47 #include "cutils/android_reboot.h"
48#endif
49#endif // ndef BUILD_TWRPTAR_MAIN
Dees_Troy83bd4832013-05-04 12:39:56 +000050#ifndef TW_EXCLUDE_ENCRYPTED_BACKUPS
51 #include "openaes/inc/oaes_lib.h"
52#endif
Ethan Yonkerd7f20922014-11-07 10:33:08 -060053#include "cutils/android_reboot.h"
Dees_Troy38bd7602012-09-14 13:33:53 -040054
Dees_Troyb05ddee2013-01-28 20:24:50 +000055extern "C" {
56 #include "libcrecovery/common.h"
Ethan Yonker4b94cfd2014-12-11 10:00:45 -060057 #include "set_metadata.h"
Dees_Troyb05ddee2013-01-28 20:24:50 +000058}
59
bigbiff bigbiff9c754052013-01-09 09:09:08 -050060/* Execute a command */
Vojtech Bocek05534202013-09-11 08:11:56 +020061int TWFunc::Exec_Cmd(const string& cmd, string &result) {
Dees_Troy29a06352013-08-24 12:06:47 +000062 FILE* exec;
63 char buffer[130];
64 int ret = 0;
65 exec = __popen(cmd.c_str(), "r");
66 if (!exec) return -1;
67 while(!feof(exec)) {
68 memset(&buffer, 0, sizeof(buffer));
69 if (fgets(buffer, 128, exec) != NULL) {
70 buffer[128] = '\n';
thatd43bf2d2014-09-21 23:13:02 +020071 buffer[129] = 0;
bigbiff bigbiff9c754052013-01-09 09:09:08 -050072 result += buffer;
Dees_Troyb05ddee2013-01-28 20:24:50 +000073 }
bigbiff bigbiff9c754052013-01-09 09:09:08 -050074 }
Dees_Troy29a06352013-08-24 12:06:47 +000075 ret = __pclose(exec);
76 return ret;
bigbiff bigbiff9c754052013-01-09 09:09:08 -050077}
78
Vojtech Bocek05534202013-09-11 08:11:56 +020079int TWFunc::Exec_Cmd(const string& cmd) {
80 pid_t pid;
81 int status;
82 switch(pid = fork())
83 {
84 case -1:
bigbiff bigbiff731df792014-02-20 18:26:13 -050085 LOGERR("Exec_Cmd(): vfork failed: %d!\n", errno);
Vojtech Bocek05534202013-09-11 08:11:56 +020086 return -1;
87 case 0: // child
88 execl("/sbin/sh", "sh", "-c", cmd.c_str(), NULL);
89 _exit(127);
90 break;
91 default:
92 {
93 if (TWFunc::Wait_For_Child(pid, &status, cmd) != 0)
94 return -1;
95 else
96 return 0;
97 }
98 }
99}
100
Dees_Troy38bd7602012-09-14 13:33:53 -0400101// Returns "file.name" from a full /path/to/file.name
102string TWFunc::Get_Filename(string Path) {
103 size_t pos = Path.find_last_of("/");
104 if (pos != string::npos) {
105 string Filename;
106 Filename = Path.substr(pos + 1, Path.size() - pos - 1);
107 return Filename;
108 } else
109 return Path;
110}
111
112// Returns "/path/to/" from a full /path/to/file.name
113string TWFunc::Get_Path(string Path) {
114 size_t pos = Path.find_last_of("/");
115 if (pos != string::npos) {
116 string Pathonly;
117 Pathonly = Path.substr(0, pos + 1);
118 return Pathonly;
119 } else
120 return Path;
121}
122
Ethan Yonkeraf2897c2014-02-10 13:07:14 -0600123int TWFunc::Wait_For_Child(pid_t pid, int *status, string Child_Name) {
124 pid_t rc_pid;
125
126 rc_pid = waitpid(pid, status, 0);
127 if (rc_pid > 0) {
128 if (WEXITSTATUS(*status) == 0)
129 LOGINFO("%s process ended with RC=%d\n", Child_Name.c_str(), WEXITSTATUS(*status)); // Success
130 else if (WIFSIGNALED(*status)) {
131 LOGINFO("%s process ended with signal: %d\n", Child_Name.c_str(), WTERMSIG(*status)); // Seg fault or some other non-graceful termination
132 return -1;
133 } else if (WEXITSTATUS(*status) != 0) {
134 LOGINFO("%s process ended with ERROR=%d\n", Child_Name.c_str(), WEXITSTATUS(*status)); // Graceful exit, but there was an error
135 return -1;
136 }
137 } else { // no PID returned
138 if (errno == ECHILD)
139 LOGINFO("%s no child process exist\n", Child_Name.c_str());
140 else {
141 LOGINFO("%s Unexpected error\n", Child_Name.c_str());
142 return -1;
143 }
144 }
145 return 0;
146}
147
148bool TWFunc::Path_Exists(string Path) {
149 // Check to see if the Path exists
150 struct stat st;
151 if (stat(Path.c_str(), &st) != 0)
152 return false;
153 else
154 return true;
155}
156
157int TWFunc::Get_File_Type(string fn) {
158 string::size_type i = 0;
159 int firstbyte = 0, secondbyte = 0;
160 char header[3];
161
162 ifstream f;
163 f.open(fn.c_str(), ios::in | ios::binary);
164 f.get(header, 3);
165 f.close();
166 firstbyte = header[i] & 0xff;
167 secondbyte = header[++i] & 0xff;
168
169 if (firstbyte == 0x1f && secondbyte == 0x8b)
170 return 1; // Compressed
171 else if (firstbyte == 0x4f && secondbyte == 0x41)
172 return 2; // Encrypted
173 else
174 return 0; // Unknown
175
176 return 0;
177}
178
179int TWFunc::Try_Decrypting_File(string fn, string password) {
180#ifndef TW_EXCLUDE_ENCRYPTED_BACKUPS
181 OAES_CTX * ctx = NULL;
182 uint8_t _key_data[32] = "";
183 FILE *f;
184 uint8_t buffer[4096];
185 uint8_t *buffer_out = NULL;
186 uint8_t *ptr = NULL;
187 size_t read_len = 0, out_len = 0;
188 int firstbyte = 0, secondbyte = 0, key_len;
189 size_t _j = 0;
190 size_t _key_data_len = 0;
191
192 // mostly kanged from OpenAES oaes.c
193 for( _j = 0; _j < 32; _j++ )
194 _key_data[_j] = _j + 1;
195 _key_data_len = password.size();
196 if( 16 >= _key_data_len )
197 _key_data_len = 16;
198 else if( 24 >= _key_data_len )
199 _key_data_len = 24;
200 else
201 _key_data_len = 32;
202 memcpy(_key_data, password.c_str(), password.size());
203
204 ctx = oaes_alloc();
205 if (ctx == NULL) {
206 LOGERR("Failed to allocate OAES\n");
207 return -1;
208 }
209
210 oaes_key_import_data(ctx, _key_data, _key_data_len);
211
212 f = fopen(fn.c_str(), "rb");
213 if (f == NULL) {
214 LOGERR("Failed to open '%s' to try decrypt\n", fn.c_str());
215 return -1;
216 }
217 read_len = fread(buffer, sizeof(uint8_t), 4096, f);
218 if (read_len <= 0) {
219 LOGERR("Read size during try decrypt failed\n");
220 fclose(f);
221 return -1;
222 }
223 if (oaes_decrypt(ctx, buffer, read_len, NULL, &out_len) != OAES_RET_SUCCESS) {
224 LOGERR("Error: Failed to retrieve required buffer size for trying decryption.\n");
225 fclose(f);
226 return -1;
227 }
228 buffer_out = (uint8_t *) calloc(out_len, sizeof(char));
229 if (buffer_out == NULL) {
230 LOGERR("Failed to allocate output buffer for try decrypt.\n");
231 fclose(f);
232 return -1;
233 }
234 if (oaes_decrypt(ctx, buffer, read_len, buffer_out, &out_len) != OAES_RET_SUCCESS) {
235 LOGERR("Failed to decrypt file '%s'\n", fn.c_str());
236 fclose(f);
237 free(buffer_out);
238 return 0;
239 }
240 fclose(f);
241 if (out_len < 2) {
242 LOGINFO("Successfully decrypted '%s' but read length %i too small.\n", fn.c_str(), out_len);
243 free(buffer_out);
244 return 1; // Decrypted successfully
245 }
246 ptr = buffer_out;
247 firstbyte = *ptr & 0xff;
248 ptr++;
249 secondbyte = *ptr & 0xff;
250 if (firstbyte == 0x1f && secondbyte == 0x8b) {
251 LOGINFO("Successfully decrypted '%s' and file is compressed.\n", fn.c_str());
252 free(buffer_out);
253 return 3; // Compressed
254 }
255 if (out_len >= 262) {
256 ptr = buffer_out + 257;
257 if (strncmp((char*)ptr, "ustar", 5) == 0) {
258 LOGINFO("Successfully decrypted '%s' and file is tar format.\n", fn.c_str());
259 free(buffer_out);
260 return 2; // Tar
261 }
262 }
263 free(buffer_out);
264 LOGINFO("No errors decrypting '%s' but no known file format.\n", fn.c_str());
265 return 1; // Decrypted successfully
266#else
267 LOGERR("Encrypted backup support not included.\n");
268 return -1;
269#endif
270}
271
272unsigned long TWFunc::Get_File_Size(string Path) {
273 struct stat st;
274
275 if (stat(Path.c_str(), &st) != 0)
276 return 0;
277 return st.st_size;
278}
279
Vojtech Bocek05f87d62014-03-11 22:08:23 +0100280std::string TWFunc::Remove_Trailing_Slashes(const std::string& path, bool leaveLast)
281{
282 std::string res;
283 size_t last_idx = 0, idx = 0;
284
285 while(last_idx != std::string::npos)
286 {
287 if(last_idx != 0)
288 res += '/';
289
290 idx = path.find_first_of('/', last_idx);
291 if(idx == std::string::npos) {
292 res += path.substr(last_idx, idx);
293 break;
294 }
295
296 res += path.substr(last_idx, idx-last_idx);
297 last_idx = path.find_first_not_of('/', idx);
298 }
299
300 if(leaveLast)
301 res += '/';
302 return res;
303}
304
Ethan Yonker1b7a31b2014-07-03 15:09:22 -0500305vector<string> TWFunc::split_string(const string &in, char del, bool skip_empty) {
306 vector<string> res;
307
308 if (in.empty() || del == '\0')
309 return res;
310
311 string field;
312 istringstream f(in);
313 if (del == '\n') {
314 while(getline(f, field)) {
315 if (field.empty() && skip_empty)
316 continue;
317 res.push_back(field);
318 }
319 } else {
320 while(getline(f, field, del)) {
321 if (field.empty() && skip_empty)
322 continue;
323 res.push_back(field);
324 }
325 }
326 return res;
327}
328
Ethan Yonkeraf2897c2014-02-10 13:07:14 -0600329#ifndef BUILD_TWRPTAR_MAIN
330
Dees_Troy38bd7602012-09-14 13:33:53 -0400331// Returns "/path" from a full /path/to/file.name
332string TWFunc::Get_Root_Path(string Path) {
333 string Local_Path = Path;
334
335 // Make sure that we have a leading slash
336 if (Local_Path.substr(0, 1) != "/")
337 Local_Path = "/" + Local_Path;
338
339 // Trim the path to get the root path only
340 size_t position = Local_Path.find("/", 2);
341 if (position != string::npos) {
342 Local_Path.resize(position);
343 }
344 return Local_Path;
345}
346
347void TWFunc::install_htc_dumlock(void) {
Dees_Troy38bd7602012-09-14 13:33:53 -0400348 int need_libs = 0;
349
350 if (!PartitionManager.Mount_By_Path("/system", true))
351 return;
352
353 if (!PartitionManager.Mount_By_Path("/data", true))
354 return;
355
Dees_Troy2673cec2013-04-02 20:22:16 +0000356 gui_print("Installing HTC Dumlock to system...\n");
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500357 copy_file("/res/htcd/htcdumlocksys", "/system/bin/htcdumlock", 0755);
Dees_Troy8170a922012-09-18 15:40:25 -0400358 if (!Path_Exists("/system/bin/flash_image")) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000359 gui_print("Installing flash_image...\n");
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500360 copy_file("/res/htcd/flash_imagesys", "/system/bin/flash_image", 0755);
Dees_Troy38bd7602012-09-14 13:33:53 -0400361 need_libs = 1;
362 } else
Dees_Troy2673cec2013-04-02 20:22:16 +0000363 gui_print("flash_image is already installed, skipping...\n");
Dees_Troy8170a922012-09-18 15:40:25 -0400364 if (!Path_Exists("/system/bin/dump_image")) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000365 gui_print("Installing dump_image...\n");
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500366 copy_file("/res/htcd/dump_imagesys", "/system/bin/dump_image", 0755);
Dees_Troy38bd7602012-09-14 13:33:53 -0400367 need_libs = 1;
368 } else
Dees_Troy2673cec2013-04-02 20:22:16 +0000369 gui_print("dump_image is already installed, skipping...\n");
Dees_Troy38bd7602012-09-14 13:33:53 -0400370 if (need_libs) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000371 gui_print("Installing libs needed for flash_image and dump_image...\n");
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500372 copy_file("/res/htcd/libbmlutils.so", "/system/lib/libbmlutils.so", 0755);
373 copy_file("/res/htcd/libflashutils.so", "/system/lib/libflashutils.so", 0755);
374 copy_file("/res/htcd/libmmcutils.so", "/system/lib/libmmcutils.so", 0755);
375 copy_file("/res/htcd/libmtdutils.so", "/system/lib/libmtdutils.so", 0755);
Dees_Troy38bd7602012-09-14 13:33:53 -0400376 }
Dees_Troy2673cec2013-04-02 20:22:16 +0000377 gui_print("Installing HTC Dumlock app...\n");
Dees_Troy38bd7602012-09-14 13:33:53 -0400378 mkdir("/data/app", 0777);
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500379 unlink("/data/app/com.teamwin.htcdumlock*");
380 copy_file("/res/htcd/HTCDumlock.apk", "/data/app/com.teamwin.htcdumlock.apk", 0777);
Dees_Troy38bd7602012-09-14 13:33:53 -0400381 sync();
Dees_Troy2673cec2013-04-02 20:22:16 +0000382 gui_print("HTC Dumlock is installed.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -0400383}
384
385void TWFunc::htc_dumlock_restore_original_boot(void) {
386 if (!PartitionManager.Mount_By_Path("/sdcard", true))
387 return;
388
Dees_Troy2673cec2013-04-02 20:22:16 +0000389 gui_print("Restoring original boot...\n");
Vojtech Bocek05534202013-09-11 08:11:56 +0200390 Exec_Cmd("htcdumlock restore");
Dees_Troy2673cec2013-04-02 20:22:16 +0000391 gui_print("Original boot restored.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -0400392}
393
394void TWFunc::htc_dumlock_reflash_recovery_to_boot(void) {
395 if (!PartitionManager.Mount_By_Path("/sdcard", true))
396 return;
Dees_Troy2673cec2013-04-02 20:22:16 +0000397 gui_print("Reflashing recovery to boot...\n");
Vojtech Bocek05534202013-09-11 08:11:56 +0200398 Exec_Cmd("htcdumlock recovery noreboot");
Dees_Troy2673cec2013-04-02 20:22:16 +0000399 gui_print("Recovery is flashed to boot.\n");
Dees_Troy38bd7602012-09-14 13:33:53 -0400400}
Dees_Troy43d8b002012-09-17 16:00:01 -0400401
402int TWFunc::Recursive_Mkdir(string Path) {
403 string pathCpy = Path;
404 string wholePath;
405 size_t pos = pathCpy.find("/", 2);
406
407 while (pos != string::npos)
408 {
409 wholePath = pathCpy.substr(0, pos);
Ethan Yonker4b94cfd2014-12-11 10:00:45 -0600410 if (!TWFunc::Path_Exists(wholePath)) {
411 if (mkdir(wholePath.c_str(), 0777)) {
412 LOGERR("Unable to create folder: %s (errno=%d)\n", wholePath.c_str(), errno);
413 return false;
414 } else {
415 tw_set_default_metadata(wholePath.c_str());
416 }
Dees_Troy43d8b002012-09-17 16:00:01 -0400417 }
418
419 pos = pathCpy.find("/", pos + 1);
420 }
421 if (mkdir(wholePath.c_str(), 0777) && errno != EEXIST)
422 return false;
423 return true;
424}
425
Dees_Troyb46a6842012-09-25 11:06:46 -0400426void TWFunc::GUI_Operation_Text(string Read_Value, string Default_Text) {
427 string Display_Text;
428
429 DataManager::GetValue(Read_Value, Display_Text);
430 if (Display_Text.empty())
431 Display_Text = Default_Text;
432
433 DataManager::SetValue("tw_operation", Display_Text);
434 DataManager::SetValue("tw_partition", "");
435}
436
437void TWFunc::GUI_Operation_Text(string Read_Value, string Partition_Name, string Default_Text) {
438 string Display_Text;
439
440 DataManager::GetValue(Read_Value, Display_Text);
441 if (Display_Text.empty())
442 Display_Text = Default_Text;
443
444 DataManager::SetValue("tw_operation", Display_Text);
445 DataManager::SetValue("tw_partition", Partition_Name);
Dees_Troy7c2dec82012-09-26 09:49:14 -0400446}
447
Dees_Troy2673cec2013-04-02 20:22:16 +0000448void TWFunc::Copy_Log(string Source, string Destination) {
Dees Troy9d7fdf52013-09-19 20:49:25 +0000449 PartitionManager.Mount_By_Path(Destination, false);
Dees_Troy2673cec2013-04-02 20:22:16 +0000450 FILE *destination_log = fopen(Destination.c_str(), "a");
451 if (destination_log == NULL) {
452 LOGERR("TWFunc::Copy_Log -- Can't open destination log file: '%s'\n", Destination.c_str());
Dees_Troy6ef66352013-02-21 08:26:57 -0600453 } else {
Dees_Troy2673cec2013-04-02 20:22:16 +0000454 FILE *source_log = fopen(Source.c_str(), "r");
455 if (source_log != NULL) {
456 fseek(source_log, Log_Offset, SEEK_SET);
457 char buffer[4096];
458 while (fgets(buffer, sizeof(buffer), source_log))
459 fputs(buffer, destination_log); // Buffered write of log file
460 Log_Offset = ftell(source_log);
461 fflush(source_log);
462 fclose(source_log);
Dees_Troy6ef66352013-02-21 08:26:57 -0600463 }
Dees_Troy2673cec2013-04-02 20:22:16 +0000464 fflush(destination_log);
465 fclose(destination_log);
Dees_Troy6ef66352013-02-21 08:26:57 -0600466 }
Dees_Troya58bead2012-09-27 09:49:29 -0400467}
468
Dees_Troy2673cec2013-04-02 20:22:16 +0000469void TWFunc::Update_Log_File(void) {
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500470 // Copy logs to cache so the system can find out what happened.
Dees Troy9d7fdf52013-09-19 20:49:25 +0000471 if (PartitionManager.Mount_By_Path("/cache", false)) {
472 if (!TWFunc::Path_Exists("/cache/recovery/.")) {
473 LOGINFO("Recreating /cache/recovery folder.\n");
474 if (mkdir("/cache/recovery", S_IRWXU | S_IRWXG | S_IWGRP | S_IXGRP) != 0)
475 LOGINFO("Unable to create /cache/recovery folder.\n");
476 }
477 Copy_Log(TMP_LOG_FILE, "/cache/recovery/log");
478 copy_file("/cache/recovery/log", "/cache/recovery/last_log", 600);
479 chown("/cache/recovery/log", 1000, 1000);
480 chmod("/cache/recovery/log", 0600);
481 chmod("/cache/recovery/last_log", 0640);
482 } else {
483 LOGINFO("Failed to mount /cache for TWFunc::Update_Log_File\n");
484 }
Dees_Troya58bead2012-09-27 09:49:29 -0400485
Dees_Troy2673cec2013-04-02 20:22:16 +0000486 // Reset bootloader message
487 TWPartition* Part = PartitionManager.Find_Partition_By_Path("/misc");
488 if (Part != NULL) {
489 struct bootloader_message boot;
490 memset(&boot, 0, sizeof(boot));
491 if (Part->Current_File_System == "mtd") {
492 if (set_bootloader_message_mtd_name(&boot, Part->MTD_Name.c_str()) != 0)
493 LOGERR("Unable to set MTD bootloader message.\n");
494 } else if (Part->Current_File_System == "emmc") {
495 if (set_bootloader_message_block_name(&boot, Part->Actual_Block_Device.c_str()) != 0)
496 LOGERR("Unable to set emmc bootloader message.\n");
497 } else {
498 LOGERR("Unknown file system for /misc: '%s'\n", Part->Current_File_System.c_str());
499 }
500 }
Dees_Troya58bead2012-09-27 09:49:29 -0400501
Dees Troy9d7fdf52013-09-19 20:49:25 +0000502 if (PartitionManager.Mount_By_Path("/cache", true)) {
503 if (unlink("/cache/recovery/command") && errno != ENOENT) {
504 LOGINFO("Can't unlink %s\n", "/cache/recovery/command");
505 }
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500506 }
Dees_Troya58bead2012-09-27 09:49:29 -0400507
Dees_Troy2673cec2013-04-02 20:22:16 +0000508 sync();
509}
510
511void TWFunc::Update_Intent_File(string Intent) {
512 if (PartitionManager.Mount_By_Path("/cache", false) && !Intent.empty()) {
513 TWFunc::write_file("/cache/recovery/intent", Intent);
514 }
Dees_Troya58bead2012-09-27 09:49:29 -0400515}
516
517// reboot: Reboot the system. Return -1 on error, no return on success
518int TWFunc::tw_reboot(RebootCommand command)
519{
520 // Always force a sync before we reboot
Dees_Troy6ef66352013-02-21 08:26:57 -0600521 sync();
Dees_Troya58bead2012-09-27 09:49:29 -0400522
Dees_Troy2673cec2013-04-02 20:22:16 +0000523 switch (command) {
524 case rb_current:
525 case rb_system:
526 Update_Log_File();
527 Update_Intent_File("s");
528 sync();
529 check_and_run_script("/sbin/rebootsystem.sh", "reboot system");
530 return reboot(RB_AUTOBOOT);
531 case rb_recovery:
532 check_and_run_script("/sbin/rebootrecovery.sh", "reboot recovery");
Ethan Yonker75bf0412014-11-21 13:54:27 -0600533#ifdef ANDROID_RB_PROPERTY
Ethan Yonkerd7f20922014-11-07 10:33:08 -0600534 property_set(ANDROID_RB_PROPERTY, "reboot,recovery");
Ethan Yonker75bf0412014-11-21 13:54:27 -0600535#else
536 return __reboot(LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2, LINUX_REBOOT_CMD_RESTART2, (void*) "recovery");
537#endif
Ethan Yonkerd7f20922014-11-07 10:33:08 -0600538 sleep(5);
539 return 0;
Dees_Troy2673cec2013-04-02 20:22:16 +0000540 case rb_bootloader:
541 check_and_run_script("/sbin/rebootbootloader.sh", "reboot bootloader");
Ethan Yonker75bf0412014-11-21 13:54:27 -0600542#ifdef ANDROID_RB_PROPERTY
Ethan Yonkerd7f20922014-11-07 10:33:08 -0600543 property_set(ANDROID_RB_PROPERTY, "reboot,bootloader");
Ethan Yonker75bf0412014-11-21 13:54:27 -0600544#else
545 return __reboot(LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2, LINUX_REBOOT_CMD_RESTART2, (void*) "bootloader");
546#endif
Ethan Yonkerd7f20922014-11-07 10:33:08 -0600547 sleep(5);
548 return 0;
Dees_Troy2673cec2013-04-02 20:22:16 +0000549 case rb_poweroff:
550 check_and_run_script("/sbin/poweroff.sh", "power off");
Dees_Troya4438782013-02-22 18:44:00 +0000551#ifdef ANDROID_RB_POWEROFF
Dees_Troy2673cec2013-04-02 20:22:16 +0000552 android_reboot(ANDROID_RB_POWEROFF, 0, 0);
Dees_Troya4438782013-02-22 18:44:00 +0000553#endif
Dees_Troy2673cec2013-04-02 20:22:16 +0000554 return reboot(RB_POWER_OFF);
555 case rb_download:
556 check_and_run_script("/sbin/rebootdownload.sh", "reboot download");
Ethan Yonker75bf0412014-11-21 13:54:27 -0600557#ifdef ANDROID_RB_PROPERTY
Ethan Yonkerd7f20922014-11-07 10:33:08 -0600558 property_set(ANDROID_RB_PROPERTY, "reboot,download");
Ethan Yonker75bf0412014-11-21 13:54:27 -0600559#else
560 return __reboot(LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2, LINUX_REBOOT_CMD_RESTART2, (void*) "download");
561#endif
Ethan Yonkerd7f20922014-11-07 10:33:08 -0600562 sleep(5);
563 return 0;
Dees_Troy2673cec2013-04-02 20:22:16 +0000564 default:
565 return -1;
Dees_Troy6ef66352013-02-21 08:26:57 -0600566 }
567 return -1;
Dees_Troya58bead2012-09-27 09:49:29 -0400568}
569
570void TWFunc::check_and_run_script(const char* script_file, const char* display_name)
571{
572 // Check for and run startup script if script exists
573 struct stat st;
574 if (stat(script_file, &st) == 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000575 gui_print("Running %s script...\n", display_name);
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500576 chmod(script_file, S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH);
Vojtech Bocek05534202013-09-11 08:11:56 +0200577 TWFunc::Exec_Cmd(script_file);
Dees_Troy2673cec2013-04-02 20:22:16 +0000578 gui_print("\nFinished running %s script.\n", display_name);
Dees_Troya58bead2012-09-27 09:49:29 -0400579 }
Dees_Troy3477d712012-09-27 15:44:01 -0400580}
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500581
582int TWFunc::removeDir(const string path, bool skipParent) {
Dees_Troyce675462013-01-09 19:48:21 +0000583 DIR *d = opendir(path.c_str());
584 int r = 0;
585 string new_path;
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500586
Dees_Troyce675462013-01-09 19:48:21 +0000587 if (d == NULL) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000588 LOGERR("Error opening '%s'\n", path.c_str());
Dees_Troyce675462013-01-09 19:48:21 +0000589 return -1;
590 }
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500591
Dees_Troyce675462013-01-09 19:48:21 +0000592 if (d) {
593 struct dirent *p;
594 while (!r && (p = readdir(d))) {
Dees_Troyce675462013-01-09 19:48:21 +0000595 if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
596 continue;
597 new_path = path + "/";
598 new_path.append(p->d_name);
599 if (p->d_type == DT_DIR) {
600 r = removeDir(new_path, true);
601 if (!r) {
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500602 if (p->d_type == DT_DIR)
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500603 r = rmdir(new_path.c_str());
604 else
Dees_Troy2673cec2013-04-02 20:22:16 +0000605 LOGINFO("Unable to removeDir '%s': %s\n", new_path.c_str(), strerror(errno));
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500606 }
bigbiff bigbiff98f1f902013-01-19 18:46:13 -0500607 } else if (p->d_type == DT_REG || p->d_type == DT_LNK || p->d_type == DT_FIFO || p->d_type == DT_SOCK) {
Dees_Troyce675462013-01-09 19:48:21 +0000608 r = unlink(new_path.c_str());
Dees_Troye34c1332013-02-06 19:13:00 +0000609 if (r != 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000610 LOGINFO("Unable to unlink '%s: %s'\n", new_path.c_str(), strerror(errno));
Dees_Troye34c1332013-02-06 19:13:00 +0000611 }
Dees_Troyce675462013-01-09 19:48:21 +0000612 }
613 }
614 closedir(d);
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500615
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500616 if (!r) {
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500617 if (skipParent)
618 return 0;
619 else
620 r = rmdir(path.c_str());
621 }
Dees_Troyce675462013-01-09 19:48:21 +0000622 }
623 return r;
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500624}
625
626int TWFunc::copy_file(string src, string dst, int mode) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000627 LOGINFO("Copying file %s to %s\n", src.c_str(), dst.c_str());
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500628 ifstream srcfile(src.c_str(), ios::binary);
629 ofstream dstfile(dst.c_str(), ios::binary);
630 dstfile << srcfile.rdbuf();
631 srcfile.close();
632 dstfile.close();
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500633 if (chmod(dst.c_str(), mode) != 0)
634 return -1;
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500635 return 0;
636}
Dees_Troy3ee47bc2013-01-25 21:47:37 +0000637
638unsigned int TWFunc::Get_D_Type_From_Stat(string Path) {
639 struct stat st;
640
641 stat(Path.c_str(), &st);
642 if (st.st_mode & S_IFDIR)
643 return DT_DIR;
644 else if (st.st_mode & S_IFBLK)
645 return DT_BLK;
646 else if (st.st_mode & S_IFCHR)
647 return DT_CHR;
648 else if (st.st_mode & S_IFIFO)
649 return DT_FIFO;
650 else if (st.st_mode & S_IFLNK)
651 return DT_LNK;
652 else if (st.st_mode & S_IFREG)
653 return DT_REG;
654 else if (st.st_mode & S_IFSOCK)
655 return DT_SOCK;
656 return DT_UNKNOWN;
Dees_Troye34c1332013-02-06 19:13:00 +0000657}
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500658
659int TWFunc::read_file(string fn, string& results) {
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200660 ifstream file;
661 file.open(fn.c_str(), ios::in);
662
663 if (file.is_open()) {
664 file >> results;
665 file.close();
666 return 0;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500667 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200668
669 LOGINFO("Cannot find file %s\n", fn.c_str());
670 return -1;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500671}
672
673int TWFunc::read_file(string fn, vector<string>& results) {
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500674 ifstream file;
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500675 string line;
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500676 file.open(fn.c_str(), ios::in);
677 if (file.is_open()) {
bigbiff bigbiffcdcfee42013-02-27 21:11:26 -0500678 while (getline(file, line))
679 results.push_back(line);
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500680 file.close();
681 return 0;
682 }
Dees_Troy2673cec2013-04-02 20:22:16 +0000683 LOGINFO("Cannot find file %s\n", fn.c_str());
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500684 return -1;
685}
686
xNUTxe85f02d2014-07-18 01:30:58 +0200687int TWFunc::read_file(string fn, uint64_t& results) {
688 ifstream file;
689 file.open(fn.c_str(), ios::in);
690
691 if (file.is_open()) {
692 file >> results;
693 file.close();
694 return 0;
695 }
696
697 LOGINFO("Cannot find file %s\n", fn.c_str());
698 return -1;
699}
700
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500701int TWFunc::write_file(string fn, string& line) {
702 FILE *file;
703 file = fopen(fn.c_str(), "w");
704 if (file != NULL) {
705 fwrite(line.c_str(), line.size(), 1, file);
706 fclose(file);
707 return 0;
708 }
Dees_Troy2673cec2013-04-02 20:22:16 +0000709 LOGINFO("Cannot find file %s\n", fn.c_str());
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500710 return -1;
711}
712
713timespec TWFunc::timespec_diff(timespec& start, timespec& end)
714{
Dees_Troy6ef66352013-02-21 08:26:57 -0600715 timespec temp;
716 if ((end.tv_nsec-start.tv_nsec)<0) {
717 temp.tv_sec = end.tv_sec-start.tv_sec-1;
718 temp.tv_nsec = 1000000000+end.tv_nsec-start.tv_nsec;
719 } else {
720 temp.tv_sec = end.tv_sec-start.tv_sec;
721 temp.tv_nsec = end.tv_nsec-start.tv_nsec;
722 }
723 return temp;
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500724}
725
Vojtech Boceke5ffcd12014-02-06 21:17:32 +0100726int32_t TWFunc::timespec_diff_ms(timespec& start, timespec& end)
727{
728 return ((end.tv_sec * 1000) + end.tv_nsec/1000000) -
729 ((start.tv_sec * 1000) + start.tv_nsec/1000000);
730}
731
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200732int TWFunc::drop_caches(void) {
Dees_Troy6ef66352013-02-21 08:26:57 -0600733 string file = "/proc/sys/vm/drop_caches";
734 string value = "3";
735 if (write_file(file, value) != 0)
736 return -1;
737 return 0;
738}
739
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200740int TWFunc::tw_chmod(const string& fn, const string& mode) {
Dees_Troy6ef66352013-02-21 08:26:57 -0600741 long mask = 0;
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200742 std::string::size_type n = mode.length();
743 int cls = 0;
Dees_Troy6ef66352013-02-21 08:26:57 -0600744
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200745 if(n == 3)
746 ++cls;
747 else if(n != 4)
748 {
749 LOGERR("TWFunc::tw_chmod used with %u long mode string (should be 3 or 4)!\n", mode.length());
750 return -1;
751 }
752
753 for (n = 0; n < mode.length(); ++n, ++cls) {
754 if (cls == 0) {
Dees_Troy6ef66352013-02-21 08:26:57 -0600755 if (mode[n] == '0')
756 continue;
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200757 else if (mode[n] == '1')
Dees_Troy6ef66352013-02-21 08:26:57 -0600758 mask |= S_ISVTX;
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200759 else if (mode[n] == '2')
Dees_Troy6ef66352013-02-21 08:26:57 -0600760 mask |= S_ISGID;
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200761 else if (mode[n] == '4')
Dees_Troy6ef66352013-02-21 08:26:57 -0600762 mask |= S_ISUID;
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200763 else if (mode[n] == '5') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600764 mask |= S_ISVTX;
765 mask |= S_ISUID;
766 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200767 else if (mode[n] == '6') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600768 mask |= S_ISGID;
769 mask |= S_ISUID;
770 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200771 else if (mode[n] == '7') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600772 mask |= S_ISVTX;
773 mask |= S_ISGID;
774 mask |= S_ISUID;
775 }
776 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200777 else if (cls == 1) {
Dees_Troy6ef66352013-02-21 08:26:57 -0600778 if (mode[n] == '7') {
779 mask |= S_IRWXU;
780 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200781 else if (mode[n] == '6') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600782 mask |= S_IRUSR;
783 mask |= S_IWUSR;
784 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200785 else if (mode[n] == '5') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600786 mask |= S_IRUSR;
787 mask |= S_IXUSR;
788 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200789 else if (mode[n] == '4')
Dees_Troy6ef66352013-02-21 08:26:57 -0600790 mask |= S_IRUSR;
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200791 else if (mode[n] == '3') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600792 mask |= S_IWUSR;
793 mask |= S_IRUSR;
794 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200795 else if (mode[n] == '2')
Dees_Troy6ef66352013-02-21 08:26:57 -0600796 mask |= S_IWUSR;
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200797 else if (mode[n] == '1')
Dees_Troy6ef66352013-02-21 08:26:57 -0600798 mask |= S_IXUSR;
799 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200800 else if (cls == 2) {
Dees_Troy6ef66352013-02-21 08:26:57 -0600801 if (mode[n] == '7') {
802 mask |= S_IRWXG;
803 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200804 else if (mode[n] == '6') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600805 mask |= S_IRGRP;
806 mask |= S_IWGRP;
807 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200808 else if (mode[n] == '5') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600809 mask |= S_IRGRP;
810 mask |= S_IXGRP;
811 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200812 else if (mode[n] == '4')
Dees_Troy6ef66352013-02-21 08:26:57 -0600813 mask |= S_IRGRP;
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200814 else if (mode[n] == '3') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600815 mask |= S_IWGRP;
816 mask |= S_IXGRP;
817 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200818 else if (mode[n] == '2')
Dees_Troy6ef66352013-02-21 08:26:57 -0600819 mask |= S_IWGRP;
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200820 else if (mode[n] == '1')
Dees_Troy6ef66352013-02-21 08:26:57 -0600821 mask |= S_IXGRP;
822 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200823 else if (cls == 3) {
Dees_Troy6ef66352013-02-21 08:26:57 -0600824 if (mode[n] == '7') {
825 mask |= S_IRWXO;
826 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200827 else if (mode[n] == '6') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600828 mask |= S_IROTH;
829 mask |= S_IWOTH;
830 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200831 else if (mode[n] == '5') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600832 mask |= S_IROTH;
833 mask |= S_IXOTH;
834 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200835 else if (mode[n] == '4')
836 mask |= S_IROTH;
837 else if (mode[n] == '3') {
Dees_Troy6ef66352013-02-21 08:26:57 -0600838 mask |= S_IWOTH;
839 mask |= S_IXOTH;
840 }
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200841 else if (mode[n] == '2')
Dees_Troy6ef66352013-02-21 08:26:57 -0600842 mask |= S_IWOTH;
Vojtech Bocek37aeb8d2013-08-29 22:38:20 +0200843 else if (mode[n] == '1')
Dees_Troy6ef66352013-02-21 08:26:57 -0600844 mask |= S_IXOTH;
845 }
846 }
847
848 if (chmod(fn.c_str(), mask) != 0) {
Dees_Troy2673cec2013-04-02 20:22:16 +0000849 LOGERR("Unable to chmod '%s' %l\n", fn.c_str(), mask);
Dees_Troy6ef66352013-02-21 08:26:57 -0600850 return -1;
851 }
852
853 return 0;
854}
855
856bool TWFunc::Install_SuperSU(void) {
857 if (!PartitionManager.Mount_By_Path("/system", true))
858 return false;
859
Ethan Yonkere3e88292014-12-10 16:17:55 -0600860 check_and_run_script("/supersu/install-supersu.sh", "SuperSU");
Dees_Troy6ef66352013-02-21 08:26:57 -0600861 return true;
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500862}
Dees_Troy83bd4832013-05-04 12:39:56 +0000863
Dees_Troy83bd4832013-05-04 12:39:56 +0000864bool TWFunc::Try_Decrypting_Backup(string Restore_Path, string Password) {
865 DIR* d;
866
867 string Filename;
868 Restore_Path += "/";
869 d = opendir(Restore_Path.c_str());
870 if (d == NULL) {
871 LOGERR("Error opening '%s'\n", Restore_Path.c_str());
872 return false;
873 }
874
875 struct dirent* de;
876 while ((de = readdir(d)) != NULL) {
877 Filename = Restore_Path;
878 Filename += de->d_name;
879 if (TWFunc::Get_File_Type(Filename) == 2) {
880 if (TWFunc::Try_Decrypting_File(Filename, Password) < 2) {
881 DataManager::SetValue("tw_restore_password", ""); // Clear the bad password
882 DataManager::SetValue("tw_restore_display", ""); // Also clear the display mask
883 closedir(d);
884 return false;
885 }
886 }
887 }
888 closedir(d);
889 return true;
890}
891
Dees Troyb21cc642013-09-10 17:36:41 +0000892string TWFunc::Get_Current_Date() {
893 string Current_Date;
894 time_t seconds = time(0);
895 struct tm *t = localtime(&seconds);
896 char timestamp[255];
897 sprintf(timestamp,"%04d-%02d-%02d--%02d-%02d-%02d",t->tm_year+1900,t->tm_mon+1,t->tm_mday,t->tm_hour,t->tm_min,t->tm_sec);
898 Current_Date = timestamp;
899 return Current_Date;
900}
901
Ethan Yonkerb5557892014-02-07 21:43:20 -0600902string TWFunc::System_Property_Get(string Prop_Name) {
Dees Troyb21cc642013-09-10 17:36:41 +0000903 bool mount_state = PartitionManager.Is_Mounted_By_Path("/system");
904 std::vector<string> buildprop;
Ethan Yonkerb5557892014-02-07 21:43:20 -0600905 string propvalue;
906 if (!PartitionManager.Mount_By_Path("/system", true))
907 return propvalue;
Dees Troyb21cc642013-09-10 17:36:41 +0000908 if (TWFunc::read_file("/system/build.prop", buildprop) != 0) {
Ethan Yonkerb5557892014-02-07 21:43:20 -0600909 LOGINFO("Unable to open /system/build.prop for getting '%s'.\n", Prop_Name.c_str());
Vojtech Boceka2e70162013-09-17 17:05:10 +0200910 DataManager::SetValue(TW_BACKUP_NAME, Get_Current_Date());
Dees Troyb21cc642013-09-10 17:36:41 +0000911 if (!mount_state)
912 PartitionManager.UnMount_By_Path("/system", false);
Ethan Yonkerb5557892014-02-07 21:43:20 -0600913 return propvalue;
Dees Troyb21cc642013-09-10 17:36:41 +0000914 }
915 int line_count = buildprop.size();
916 int index;
917 size_t start_pos = 0, end_pos;
Ethan Yonkerb5557892014-02-07 21:43:20 -0600918 string propname;
Dees Troyb21cc642013-09-10 17:36:41 +0000919 for (index = 0; index < line_count; index++) {
920 end_pos = buildprop.at(index).find("=", start_pos);
921 propname = buildprop.at(index).substr(start_pos, end_pos);
Ethan Yonkerb5557892014-02-07 21:43:20 -0600922 if (propname == Prop_Name) {
Dees Troyb21cc642013-09-10 17:36:41 +0000923 propvalue = buildprop.at(index).substr(end_pos + 1, buildprop.at(index).size());
Ethan Yonkerb5557892014-02-07 21:43:20 -0600924 if (!mount_state)
925 PartitionManager.UnMount_By_Path("/system", false);
926 return propvalue;
Dees Troyb21cc642013-09-10 17:36:41 +0000927 }
928 }
Dees Troyb21cc642013-09-10 17:36:41 +0000929 if (!mount_state)
930 PartitionManager.UnMount_By_Path("/system", false);
Ethan Yonkerb5557892014-02-07 21:43:20 -0600931 return propvalue;
932}
933
934void TWFunc::Auto_Generate_Backup_Name() {
935 string propvalue = System_Property_Get("ro.build.display.id");
936 if (propvalue.empty()) {
937 DataManager::SetValue(TW_BACKUP_NAME, Get_Current_Date());
938 return;
939 }
940 string Backup_Name = Get_Current_Date();
941 Backup_Name += " " + propvalue;
942 if (Backup_Name.size() > MAX_BACKUP_NAME_LEN)
943 Backup_Name.resize(MAX_BACKUP_NAME_LEN);
944 // Trailing spaces cause problems on some file systems, so remove them
945 string space_check, space = " ";
946 space_check = Backup_Name.substr(Backup_Name.size() - 1, 1);
947 while (space_check == space) {
948 Backup_Name.resize(Backup_Name.size() - 1);
949 space_check = Backup_Name.substr(Backup_Name.size() - 1, 1);
950 }
951 DataManager::SetValue(TW_BACKUP_NAME, Backup_Name);
Ethan Yonker92d48e02014-02-26 12:05:55 -0600952 if (PartitionManager.Check_Backup_Name(false) != 0) {
953 LOGINFO("Auto generated backup name '%s' contains invalid characters, using date instead.\n", Backup_Name.c_str());
954 DataManager::SetValue(TW_BACKUP_NAME, Get_Current_Date());
955 }
Vojtech Bocek05534202013-09-11 08:11:56 +0200956}
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +0100957
958void TWFunc::Fixup_Time_On_Boot()
959{
960#ifdef QCOM_RTC_FIX
xNUTxe85f02d2014-07-18 01:30:58 +0200961
962 LOGINFO("TWFunc::Fixup_Time: Pre-fix date and time: %s\n", TWFunc::Get_Current_Date().c_str());
963
964 struct timeval tv;
965 uint64_t offset = 0;
966 std::string sepoch = "/sys/class/rtc/rtc0/since_epoch";
967
968 if (TWFunc::read_file(sepoch, offset) == 0) {
969
970 LOGINFO("TWFunc::Fixup_Time: Setting time offset from file %s\n", sepoch.c_str());
971
972 tv.tv_sec = offset;
973 tv.tv_usec = 0;
974 settimeofday(&tv, NULL);
975
976 gettimeofday(&tv, NULL);
977
978 if (tv.tv_sec > 1405209403) { // Anything older then 12 Jul 2014 23:56:43 GMT will do nicely thank you ;)
979
980 LOGINFO("TWFunc::Fixup_Time: Date and time corrected: %s\n", TWFunc::Get_Current_Date().c_str());
981 return;
982
983 }
984
985 } else {
986
987 LOGINFO("TWFunc::Fixup_Time: opening %s failed\n", sepoch.c_str());
988
989 }
990
991 LOGINFO("TWFunc::Fixup_Time: will attempt to use the ats files now.\n", sepoch.c_str());
992
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +0100993 // Devices with Qualcomm Snapdragon 800 do some shenanigans with RTC.
994 // They never set it, it just ticks forward from 1970-01-01 00:00,
995 // and then they have files /data/system/time/ats_* with 64bit offset
996 // in miliseconds which, when added to the RTC, gives the correct time.
997 // So, the time is: (offset_from_ats + value_from_RTC)
998 // There are multiple ats files, they are for different systems? Bases?
999 // Like, ats_1 is for modem and ats_2 is for TOD (time of day?).
1000 // Look at file time_genoff.h in CodeAurora, qcom-opensource/time-services
1001
1002 static const char *paths[] = { "/data/system/time/", "/data/time/" };
1003
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +01001004 FILE *f;
xNUTxe85f02d2014-07-18 01:30:58 +02001005 DIR *d;
1006 offset = 0;
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +01001007 struct dirent *dt;
1008 std::string ats_path;
1009
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +01001010 if(!PartitionManager.Mount_By_Path("/data", false))
1011 return;
1012
1013 // Prefer ats_2, it seems to be the one we want according to logcat on hammerhead
1014 // - it is the one for ATS_TOD (time of day?).
1015 // However, I never saw a device where the offset differs between ats files.
1016 for(size_t i = 0; i < (sizeof(paths)/sizeof(paths[0])); ++i)
1017 {
1018 DIR *d = opendir(paths[i]);
1019 if(!d)
1020 continue;
1021
1022 while((dt = readdir(d)))
1023 {
1024 if(dt->d_type != DT_REG || strncmp(dt->d_name, "ats_", 4) != 0)
1025 continue;
1026
1027 if(ats_path.empty() || strcmp(dt->d_name, "ats_2") == 0)
1028 ats_path = std::string(paths[i]).append(dt->d_name);
1029 }
1030
1031 closedir(d);
1032 }
1033
1034 if(ats_path.empty())
1035 {
xNUTxe85f02d2014-07-18 01:30:58 +02001036 LOGINFO("TWFunc::Fixup_Time: no ats files found, leaving untouched!\n");
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +01001037 return;
1038 }
1039
1040 f = fopen(ats_path.c_str(), "r");
1041 if(!f)
1042 {
Dees Troy3e254b92014-03-06 20:24:54 +00001043 LOGINFO("TWFunc::Fixup_Time: failed to open file %s\n", ats_path.c_str());
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +01001044 return;
1045 }
1046
1047 if(fread(&offset, sizeof(offset), 1, f) != 1)
1048 {
Dees Troy3e254b92014-03-06 20:24:54 +00001049 LOGINFO("TWFunc::Fixup_Time: failed load uint64 from file %s\n", ats_path.c_str());
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +01001050 fclose(f);
1051 return;
1052 }
1053 fclose(f);
1054
1055 LOGINFO("TWFunc::Fixup_Time: Setting time offset from file %s, offset %llu\n", ats_path.c_str(), offset);
1056
1057 gettimeofday(&tv, NULL);
1058
1059 tv.tv_sec += offset/1000;
1060 tv.tv_usec += (offset%1000)*1000;
1061
1062 while(tv.tv_usec >= 1000000)
1063 {
1064 ++tv.tv_sec;
1065 tv.tv_usec -= 1000000;
1066 }
1067
1068 settimeofday(&tv, NULL);
xNUTxe85f02d2014-07-18 01:30:58 +02001069
1070 LOGINFO("TWFunc::Fixup_Time: Date and time corrected: %s\n", TWFunc::Get_Current_Date().c_str());
1071
Vojtech Bocekd0e38bc2014-02-03 23:36:57 +01001072#endif
1073}
Ethan Yonkeraf2897c2014-02-10 13:07:14 -06001074
Vojtech Bocek0b7fe502014-03-13 17:36:52 +01001075std::vector<std::string> TWFunc::Split_String(const std::string& str, const std::string& delimiter, bool removeEmpty)
1076{
1077 std::vector<std::string> res;
1078 size_t idx = 0, idx_last = 0;
1079
1080 while(idx < str.size())
1081 {
1082 idx = str.find_first_of(delimiter, idx_last);
1083 if(idx == std::string::npos)
1084 idx = str.size();
1085
1086 if(idx-idx_last != 0 || !removeEmpty)
1087 res.push_back(str.substr(idx_last, idx-idx_last));
1088
1089 idx_last = idx + delimiter.size();
1090 }
1091
1092 return res;
1093}
1094
Vojtech Bocek03fd6c52014-03-13 18:46:34 +01001095bool TWFunc::Create_Dir_Recursive(const std::string& path, mode_t mode, uid_t uid, gid_t gid)
1096{
1097 std::vector<std::string> parts = Split_String(path, "/");
1098 std::string cur_path;
1099 struct stat info;
1100 for(size_t i = 0; i < parts.size(); ++i)
1101 {
1102 cur_path += "/" + parts[i];
1103 if(stat(cur_path.c_str(), &info) < 0 || !S_ISDIR(info.st_mode))
1104 {
1105 if(mkdir(cur_path.c_str(), mode) < 0)
1106 return false;
1107 chown(cur_path.c_str(), uid, gid);
1108 }
1109 }
1110 return true;
1111}
1112
xNUTxe85f02d2014-07-18 01:30:58 +02001113int TWFunc::Set_Brightness(std::string brightness_value)
1114{
1115
1116 std::string brightness_file = DataManager::GetStrValue("tw_brightness_file");;
1117
1118 if (brightness_file.compare("/nobrightness") != 0) {
1119 std::string secondary_brightness_file = DataManager::GetStrValue("tw_secondary_brightness_file");
1120 LOGINFO("TWFunc::Set_Brightness: Setting brightness control to %s\n", brightness_value.c_str());
1121 int result = TWFunc::write_file(brightness_file, brightness_value);
1122 if (secondary_brightness_file != "") {
1123 LOGINFO("TWFunc::Set_Brightness: Setting SECONDARY brightness control to %s\n", brightness_value.c_str());
1124 TWFunc::write_file(secondary_brightness_file, brightness_value);
1125 }
1126 return result;
1127 }
1128 return -1;
1129}
1130
bigbiff bigbiffc7eee6f2014-09-02 18:59:01 -04001131bool TWFunc::Toggle_MTP(bool enable) {
1132#ifdef TW_HAS_MTP
1133 static int was_enabled = false;
1134
1135 if (enable && was_enabled) {
1136 if (!PartitionManager.Enable_MTP())
1137 PartitionManager.Disable_MTP();
1138 } else {
1139 was_enabled = DataManager::GetIntValue("tw_mtp_enabled");
1140 PartitionManager.Disable_MTP();
1141 usleep(500);
1142 }
1143 return was_enabled;
1144#else
1145 return false;
1146#endif
1147}
1148
Tom Hite5a926722014-09-15 01:31:03 +00001149void TWFunc::SetPerformanceMode(bool mode) {
1150 if (mode) {
1151 property_set("recovery.perf.mode", "1");
1152 } else {
1153 property_set("recovery.perf.mode", "0");
1154 }
1155 // Some time for events to catch up to init handlers
1156 usleep(500000);
1157}
1158
Jenkins1710bf22014-10-02 20:22:21 -04001159std::string TWFunc::to_string(unsigned long value) {
1160 std::ostringstream os;
1161 os << value;
1162 return os.str();
1163}
1164
Ethan Yonkeraf2897c2014-02-10 13:07:14 -06001165#endif // ndef BUILD_TWRPTAR_MAIN