blob: 9bf66f47a967f3c244ca20e22635c2afe78c007f [file] [log] [blame]
Ethan Yonker6e8c27a2016-12-22 17:55:57 -06001/*
Dees_Troya13d74f2013-03-24 08:54:55 -05002 Copyright 2013 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*/
Dees_Troy51a0e822012-09-05 15:24:24 -040018
19#include <stdarg.h>
20#include <stdio.h>
21#include <stdlib.h>
22#include <string.h>
23#include <fcntl.h>
24#include <sys/stat.h>
25#include <sys/time.h>
26#include <sys/mman.h>
27#include <sys/types.h>
28#include <sys/ioctl.h>
29#include <linux/input.h>
30#include <time.h>
31#include <unistd.h>
32#include <stdlib.h>
Dees_Troy657c3092012-09-10 20:32:10 -040033#include <sys/wait.h>
Dees_Troy83bd4832013-05-04 12:39:56 +000034#include <dirent.h>
Matt Mower0c88b842017-01-15 16:00:49 -060035#include <private/android_filesystem_config.h>
bigbiffdf8436b2020-08-30 16:22:34 -040036#include <android-base/properties.h>
Dees_Troy51a0e822012-09-05 15:24:24 -040037
38#include <string>
39#include <sstream>
40#include "../partitions.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040041#include "../twrp-functions.hpp"
bigbiffad58e1b2020-07-06 20:24:34 -040042#include "../twrpRepacker.hpp"
Dees_Troy812660f2012-09-20 09:55:17 -040043#include "../openrecoveryscript.hpp"
Dees_Troy51a0e822012-09-05 15:24:24 -040044
bigbiffd58ba182020-03-23 10:02:29 -040045#include "install/adb_install.h"
46
47#include "fuse_sideload.h"
bigbiff bigbiff8a68c312013-02-10 14:28:30 -050048#include "blanktimer.hpp"
Ethan Yonker8373cfe2017-09-08 06:50:54 -050049#include "../twinstall.h"
Ethan Yonkerfbb43532015-12-28 21:54:50 +010050
Dees_Troy51a0e822012-09-05 15:24:24 -040051extern "C" {
Dees_Troy2673cec2013-04-02 20:22:16 +000052#include "../twcommon.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040053#include "../variables.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000054#include "cutils/properties.h"
bigbiffd58ba182020-03-23 10:02:29 -040055#include "install/adb_install.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040056};
bigbiffd58ba182020-03-23 10:02:29 -040057#include "set_metadata.h"
Ethan Yonkerfbb43532015-12-28 21:54:50 +010058#include "../minuitwrp/minui.h"
Dees_Troy51a0e822012-09-05 15:24:24 -040059
60#include "rapidxml.hpp"
61#include "objects.hpp"
bigbiffd58ba182020-03-23 10:02:29 -040062#include "tw_atomic.hpp"
Dees_Troy51a0e822012-09-05 15:24:24 -040063
that3f7b1ac2014-12-30 11:30:13 +010064GUIAction::mapFunc GUIAction::mf;
thatc6085482015-01-09 22:12:43 +010065std::set<string> GUIAction::setActionsRunningInCallerThread;
that3f7b1ac2014-12-30 11:30:13 +010066static string zip_queue[10];
67static int zip_queue_index;
thatcc8ddca2015-01-03 01:59:36 +010068pid_t sideload_child_pid;
Noah Jacobson81d638d2019-04-28 00:10:07 -040069extern std::vector<users_struct> Users_List;
that3f7b1ac2014-12-30 11:30:13 +010070
that73a52952015-01-28 23:07:34 +010071static void *ActionThread_work_wrapper(void *data);
72
73class ActionThread
74{
75public:
76 ActionThread();
77 ~ActionThread();
78
79 void threadActions(GUIAction *act);
80 void run(void *data);
81private:
82 friend void *ActionThread_work_wrapper(void*);
83 struct ThreadData
84 {
85 ActionThread *this_;
86 GUIAction *act;
87 ThreadData(ActionThread *this_, GUIAction *act) : this_(this_), act(act) {}
88 };
89
90 pthread_t m_thread;
91 bool m_thread_running;
92 pthread_mutex_t m_act_lock;
93};
94
95static ActionThread action_thread; // for all kinds of longer running actions
96static ActionThread cancel_thread; // for longer running "cancel" actions
thatc6085482015-01-09 22:12:43 +010097
98static void *ActionThread_work_wrapper(void *data)
99{
that73a52952015-01-28 23:07:34 +0100100 static_cast<ActionThread::ThreadData*>(data)->this_->run(data);
thatc6085482015-01-09 22:12:43 +0100101 return NULL;
102}
103
104ActionThread::ActionThread()
105{
106 m_thread_running = false;
107 pthread_mutex_init(&m_act_lock, NULL);
108}
109
110ActionThread::~ActionThread()
111{
112 pthread_mutex_lock(&m_act_lock);
Matt Mowera8a89d12016-12-30 18:10:37 -0600113 if (m_thread_running) {
thatc6085482015-01-09 22:12:43 +0100114 pthread_mutex_unlock(&m_act_lock);
115 pthread_join(m_thread, NULL);
116 } else {
117 pthread_mutex_unlock(&m_act_lock);
118 }
119 pthread_mutex_destroy(&m_act_lock);
120}
121
that7d3b54f2015-01-09 22:52:51 +0100122void ActionThread::threadActions(GUIAction *act)
thatc6085482015-01-09 22:12:43 +0100123{
124 pthread_mutex_lock(&m_act_lock);
125 if (m_thread_running) {
126 pthread_mutex_unlock(&m_act_lock);
that7d3b54f2015-01-09 22:52:51 +0100127 LOGERR("Another threaded action is already running -- not running %u actions starting with '%s'\n",
128 act->mActions.size(), act->mActions[0].mFunction.c_str());
thatc6085482015-01-09 22:12:43 +0100129 } else {
130 m_thread_running = true;
131 pthread_mutex_unlock(&m_act_lock);
that73a52952015-01-28 23:07:34 +0100132 ThreadData *d = new ThreadData(this, act);
thatc6085482015-01-09 22:12:43 +0100133 pthread_create(&m_thread, NULL, &ActionThread_work_wrapper, d);
134 }
135}
136
137void ActionThread::run(void *data)
138{
139 ThreadData *d = (ThreadData*)data;
140 GUIAction* act = d->act;
141
142 std::vector<GUIAction::Action>::iterator it;
that7d3b54f2015-01-09 22:52:51 +0100143 for (it = act->mActions.begin(); it != act->mActions.end(); ++it)
thatc6085482015-01-09 22:12:43 +0100144 act->doAction(*it);
145
146 pthread_mutex_lock(&m_act_lock);
147 m_thread_running = false;
148 pthread_mutex_unlock(&m_act_lock);
149 delete d;
150}
151
Dees_Troy51a0e822012-09-05 15:24:24 -0400152GUIAction::GUIAction(xml_node<>* node)
Vojtech Bocekede51c52014-02-07 23:58:09 +0100153 : GUIObject(node)
Dees_Troy51a0e822012-09-05 15:24:24 -0400154{
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200155 xml_node<>* child;
156 xml_node<>* actions;
157 xml_attribute<>* attr;
Dees_Troy51a0e822012-09-05 15:24:24 -0400158
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200159 if (!node) return;
Dees_Troy51a0e822012-09-05 15:24:24 -0400160
that3f7b1ac2014-12-30 11:30:13 +0100161 if (mf.empty()) {
Vojtech Bocekecd89182015-01-15 16:28:54 +0100162#define ADD_ACTION(n) mf[#n] = &GUIAction::n
163#define ADD_ACTION_EX(name, func) mf[name] = &GUIAction::func
thatc6085482015-01-09 22:12:43 +0100164 // These actions will be run in the caller's thread
Vojtech Bocekecd89182015-01-15 16:28:54 +0100165 ADD_ACTION(reboot);
166 ADD_ACTION(home);
167 ADD_ACTION(key);
168 ADD_ACTION(page);
169 ADD_ACTION(reload);
170 ADD_ACTION(readBackup);
171 ADD_ACTION(set);
172 ADD_ACTION(clear);
173 ADD_ACTION(mount);
174 ADD_ACTION(unmount);
175 ADD_ACTION_EX("umount", unmount);
176 ADD_ACTION(restoredefaultsettings);
177 ADD_ACTION(copylog);
178 ADD_ACTION(compute);
179 ADD_ACTION_EX("addsubtract", compute);
180 ADD_ACTION(setguitimezone);
181 ADD_ACTION(overlay);
182 ADD_ACTION(queuezip);
183 ADD_ACTION(cancelzip);
184 ADD_ACTION(queueclear);
185 ADD_ACTION(sleep);
Matt Mower9a2a2052016-05-31 21:31:22 -0500186 ADD_ACTION(sleepcounter);
Vojtech Bocekecd89182015-01-15 16:28:54 +0100187 ADD_ACTION(appenddatetobackupname);
188 ADD_ACTION(generatebackupname);
189 ADD_ACTION(checkpartitionlist);
190 ADD_ACTION(getpartitiondetails);
191 ADD_ACTION(screenshot);
192 ADD_ACTION(setbrightness);
193 ADD_ACTION(fileexists);
194 ADD_ACTION(killterminal);
195 ADD_ACTION(checkbackupname);
196 ADD_ACTION(adbsideloadcancel);
197 ADD_ACTION(fixsu);
198 ADD_ACTION(startmtp);
199 ADD_ACTION(stopmtp);
200 ADD_ACTION(cancelbackup);
Ethan Yonkereb32b1f2015-05-18 10:23:03 -0500201 ADD_ACTION(checkpartitionlifetimewrites);
202 ADD_ACTION(mountsystemtoggle);
Ethan Yonker74db1572015-10-28 12:44:49 -0500203 ADD_ACTION(setlanguage);
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -0600204 ADD_ACTION(checkforapp);
Matt Mower9472ba12016-01-20 18:12:47 -0600205 ADD_ACTION(togglebacklight);
bigbiffdf8436b2020-08-30 16:22:34 -0400206 ADD_ACTION(enableadb);
207 ADD_ACTION(enablefastboot);
thatc6085482015-01-09 22:12:43 +0100208
209 // remember actions that run in the caller thread
210 for (mapFunc::const_iterator it = mf.begin(); it != mf.end(); ++it)
211 setActionsRunningInCallerThread.insert(it->first);
212
213 // These actions will run in a separate thread
Vojtech Bocekecd89182015-01-15 16:28:54 +0100214 ADD_ACTION(flash);
215 ADD_ACTION(wipe);
216 ADD_ACTION(refreshsizes);
217 ADD_ACTION(nandroid);
Ethan Yonkerb5fab762016-01-28 14:03:33 -0600218 ADD_ACTION(fixcontexts);
Vojtech Bocekecd89182015-01-15 16:28:54 +0100219 ADD_ACTION(fixpermissions);
220 ADD_ACTION(dd);
221 ADD_ACTION(partitionsd);
222 ADD_ACTION(installhtcdumlock);
223 ADD_ACTION(htcdumlockrestoreboot);
224 ADD_ACTION(htcdumlockreflashrecovery);
225 ADD_ACTION(cmd);
226 ADD_ACTION(terminalcommand);
227 ADD_ACTION(reinjecttwrp);
228 ADD_ACTION(decrypt);
229 ADD_ACTION(adbsideload);
230 ADD_ACTION(openrecoveryscript);
231 ADD_ACTION(installsu);
232 ADD_ACTION(decrypt_backup);
233 ADD_ACTION(repair);
Ethan Yonkera2719152015-05-28 09:44:41 -0500234 ADD_ACTION(resize);
Vojtech Bocekecd89182015-01-15 16:28:54 +0100235 ADD_ACTION(changefilesystem);
236 ADD_ACTION(flashimage);
that10ae24f2015-12-26 20:53:51 +0100237 ADD_ACTION(twcmd);
Ethan Yonker1b190162016-12-05 15:25:19 -0600238 ADD_ACTION(setbootslot);
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -0600239 ADD_ACTION(installapp);
Ethan Yonker76bbd3a2019-05-10 10:50:04 -0500240 ADD_ACTION(uninstalltwrpsystemapp);
Ethan Yonker53796e72019-01-11 22:49:52 -0600241 ADD_ACTION(repackimage);
242 ADD_ACTION(fixabrecoverybootloop);
that3f7b1ac2014-12-30 11:30:13 +0100243 }
244
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200245 // First, get the action
Ethan Yonker21ff02a2015-02-18 14:35:00 -0600246 actions = FindNode(node, "actions");
247 if (actions) child = FindNode(actions, "action");
248 else child = FindNode(node, "action");
Dees_Troy51a0e822012-09-05 15:24:24 -0400249
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200250 if (!child) return;
Dees_Troy51a0e822012-09-05 15:24:24 -0400251
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200252 while (child)
253 {
254 Action action;
Dees_Troy51a0e822012-09-05 15:24:24 -0400255
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200256 attr = child->first_attribute("function");
257 if (!attr) return;
Matt Mowerfb1c4ff2014-04-16 13:43:36 -0500258
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200259 action.mFunction = attr->value();
260 action.mArg = child->value();
261 mActions.push_back(action);
Dees_Troy51a0e822012-09-05 15:24:24 -0400262
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200263 child = child->next_sibling("action");
264 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400265
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200266 // Now, let's get either the key or region
Ethan Yonker21ff02a2015-02-18 14:35:00 -0600267 child = FindNode(node, "touch");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200268 if (child)
269 {
270 attr = child->first_attribute("key");
271 if (attr)
272 {
Vojtech Bocek0b7fe502014-03-13 17:36:52 +0100273 std::vector<std::string> keys = TWFunc::Split_String(attr->value(), "+");
Matt Mowera8a89d12016-12-30 18:10:37 -0600274 for (size_t i = 0; i < keys.size(); ++i)
Vojtech Bocek0b7fe502014-03-13 17:36:52 +0100275 {
276 const int key = getKeyByName(keys[i]);
277 mKeys[key] = false;
278 }
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200279 }
280 else
281 {
282 attr = child->first_attribute("x");
283 if (!attr) return;
284 mActionX = atol(attr->value());
285 attr = child->first_attribute("y");
286 if (!attr) return;
287 mActionY = atol(attr->value());
288 attr = child->first_attribute("w");
289 if (!attr) return;
290 mActionW = atol(attr->value());
291 attr = child->first_attribute("h");
292 if (!attr) return;
293 mActionH = atol(attr->value());
294 }
295 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400296}
297
Matt Mower25036b72016-06-24 12:30:18 -0500298int GUIAction::NotifyTouch(TOUCH_STATE state, int x __unused, int y __unused)
Dees_Troy51a0e822012-09-05 15:24:24 -0400299{
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200300 if (state == TOUCH_RELEASE)
301 doActions();
Dees_Troy51a0e822012-09-05 15:24:24 -0400302
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200303 return 0;
Dees_Troy51a0e822012-09-05 15:24:24 -0400304}
305
Vojtech Bocek0b7fe502014-03-13 17:36:52 +0100306int GUIAction::NotifyKey(int key, bool down)
Dees_Troy51a0e822012-09-05 15:24:24 -0400307{
Vojtech Bocek0b7fe502014-03-13 17:36:52 +0100308 std::map<int, bool>::iterator itr = mKeys.find(key);
Matt Mowera8a89d12016-12-30 18:10:37 -0600309 if (itr == mKeys.end())
that8834a0f2016-01-05 23:29:30 +0100310 return 1;
Vojtech Bocek0b7fe502014-03-13 17:36:52 +0100311
312 bool prevState = itr->second;
313 itr->second = down;
314
315 // If there is only one key for this action, wait for key up so it
316 // doesn't trigger with multi-key actions.
317 // Else, check if all buttons are pressed, then consume their release events
318 // so they don't trigger one-button actions and reset mKeys pressed status
Matt Mowera8a89d12016-12-30 18:10:37 -0600319 if (mKeys.size() == 1) {
320 if (!down && prevState) {
Vojtech Bocek0b7fe502014-03-13 17:36:52 +0100321 doActions();
thatd4725ca2016-01-19 00:15:21 +0100322 return 0;
323 }
Matt Mowera8a89d12016-12-30 18:10:37 -0600324 } else if (down) {
325 for (itr = mKeys.begin(); itr != mKeys.end(); ++itr) {
326 if (!itr->second)
that8834a0f2016-01-05 23:29:30 +0100327 return 1;
Vojtech Bocek0b7fe502014-03-13 17:36:52 +0100328 }
329
330 // Passed, all req buttons are pressed, reset them and consume release events
331 HardwareKeyboard *kb = PageManager::GetHardwareKeyboard();
Matt Mowera8a89d12016-12-30 18:10:37 -0600332 for (itr = mKeys.begin(); itr != mKeys.end(); ++itr) {
Vojtech Bocek0b7fe502014-03-13 17:36:52 +0100333 kb->ConsumeKeyRelease(itr->first);
334 itr->second = false;
335 }
336
337 doActions();
thatd4725ca2016-01-19 00:15:21 +0100338 return 0;
Vojtech Bocek0b7fe502014-03-13 17:36:52 +0100339 }
340
thatd4725ca2016-01-19 00:15:21 +0100341 return 1;
Dees_Troy51a0e822012-09-05 15:24:24 -0400342}
343
Vojtech Bocek07220562014-02-08 02:05:33 +0100344int GUIAction::NotifyVarChange(const std::string& varName, const std::string& value)
Dees_Troy51a0e822012-09-05 15:24:24 -0400345{
Vojtech Bocek07220562014-02-08 02:05:33 +0100346 GUIObject::NotifyVarChange(varName, value);
347
Vojtech Bocek0b7fe502014-03-13 17:36:52 +0100348 if (varName.empty() && !isConditionValid() && mKeys.empty() && !mActionW)
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200349 doActions();
Matt Mowera8a89d12016-12-30 18:10:37 -0600350 else if ((varName.empty() || IsConditionVariable(varName)) && isConditionValid() && isConditionTrue())
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200351 doActions();
Dees_Troy51a0e822012-09-05 15:24:24 -0400352
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200353 return 0;
Dees_Troy51a0e822012-09-05 15:24:24 -0400354}
355
356void GUIAction::simulate_progress_bar(void)
357{
Ethan Yonker74db1572015-10-28 12:44:49 -0500358 gui_msg("simulating=Simulating actions...");
Dees_Troy51a0e822012-09-05 15:24:24 -0400359 for (int i = 0; i < 5; i++)
360 {
bigbiff7abc5fe2015-01-17 16:53:12 -0500361 if (PartitionManager.stop_backup.get_value()) {
362 DataManager::SetValue("tw_cancel_backup", 1);
Matt Mower3c366972015-12-25 19:28:31 -0600363 gui_msg("backup_cancel=Backup Cancelled");
bigbiff7abc5fe2015-01-17 16:53:12 -0500364 DataManager::SetValue("ui_progress", 0);
365 PartitionManager.stop_backup.set_value(0);
366 return;
367 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400368 usleep(500000);
369 DataManager::SetValue("ui_progress", i * 20);
370 }
371}
372
Ethan Yonker0d47eb52015-01-09 11:23:19 -0600373int GUIAction::flash_zip(std::string filename, int* wipe_cache)
Dees_Troy51a0e822012-09-05 15:24:24 -0400374{
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200375 int ret_val = 0;
Dees_Troy51a0e822012-09-05 15:24:24 -0400376
377 DataManager::SetValue("ui_progress", 0);
Chaosmasterd5364a02020-02-03 15:38:02 +0100378 DataManager::SetValue("ui_portion_size", 0);
379 DataManager::SetValue("ui_portion_start", 0);
Dees_Troy51a0e822012-09-05 15:24:24 -0400380
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200381 if (filename.empty())
382 {
383 LOGERR("No file specified.\n");
384 return -1;
385 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400386
Ethan Yonker9598c072016-01-15 21:54:34 -0600387 if (!TWFunc::Path_Exists(filename)) {
388 if (!PartitionManager.Mount_By_Path(filename, true)) {
389 return -1;
390 }
391 if (!TWFunc::Path_Exists(filename)) {
392 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")(filename));
393 return -1;
394 }
395 }
Dees_Troy657c3092012-09-10 20:32:10 -0400396
Dees_Troy51a0e822012-09-05 15:24:24 -0400397 if (simulate) {
398 simulate_progress_bar();
399 } else {
Dees_Troy657c3092012-09-10 20:32:10 -0400400 ret_val = TWinstall_zip(filename.c_str(), wipe_cache);
Dees_Troy51a0e822012-09-05 15:24:24 -0400401
402 // Now, check if we need to ensure TWRP remains installed...
403 struct stat st;
bigbiffad58e1b2020-07-06 20:24:34 -0400404 if (stat("/system/bin/installTwrp", &st) == 0)
Dees_Troy51a0e822012-09-05 15:24:24 -0400405 {
406 DataManager::SetValue("tw_operation", "Configuring TWRP");
407 DataManager::SetValue("tw_partition", "");
Ethan Yonker74db1572015-10-28 12:44:49 -0500408 gui_msg("config_twrp=Configuring TWRP...");
bigbiffad58e1b2020-07-06 20:24:34 -0400409 if (TWFunc::Exec_Cmd("/system/bin/installTwrp reinstall") < 0)
Dees_Troy51a0e822012-09-05 15:24:24 -0400410 {
Ethan Yonker74db1572015-10-28 12:44:49 -0500411 gui_msg("config_twrp_err=Unable to configure TWRP with this kernel.");
Dees_Troy51a0e822012-09-05 15:24:24 -0400412 }
413 }
414 }
415
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200416 // Done
417 DataManager::SetValue("ui_progress", 100);
418 DataManager::SetValue("ui_progress", 0);
Chaosmasterd5364a02020-02-03 15:38:02 +0100419 DataManager::SetValue("ui_portion_size", 0);
420 DataManager::SetValue("ui_portion_start", 0);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200421 return ret_val;
Dees_Troy51a0e822012-09-05 15:24:24 -0400422}
423
that73a52952015-01-28 23:07:34 +0100424GUIAction::ThreadType GUIAction::getThreadType(const GUIAction::Action& action)
thatc6085482015-01-09 22:12:43 +0100425{
that73a52952015-01-28 23:07:34 +0100426 string func = gui_parse_text(action.mFunction);
427 bool needsThread = setActionsRunningInCallerThread.find(func) == setActionsRunningInCallerThread.end();
428 if (needsThread) {
429 if (func == "cancelbackup")
430 return THREAD_CANCEL;
431 else
432 return THREAD_ACTION;
433 }
434 return THREAD_NONE;
thatc6085482015-01-09 22:12:43 +0100435}
436
Dees_Troy51a0e822012-09-05 15:24:24 -0400437int GUIAction::doActions()
438{
that3f7b1ac2014-12-30 11:30:13 +0100439 if (mActions.size() < 1)
440 return -1;
Dees_Troy51a0e822012-09-05 15:24:24 -0400441
that73a52952015-01-28 23:07:34 +0100442 // Determine in which thread to run the actions.
443 // Do it for all actions at once before starting, so that we can cancel the whole batch if the thread is already busy.
444 ThreadType threadType = THREAD_NONE;
that3f7b1ac2014-12-30 11:30:13 +0100445 std::vector<Action>::iterator it;
that73a52952015-01-28 23:07:34 +0100446 for (it = mActions.begin(); it != mActions.end(); ++it) {
447 ThreadType tt = getThreadType(*it);
448 if (tt == THREAD_NONE)
449 continue;
450 if (threadType == THREAD_NONE)
451 threadType = tt;
452 else if (threadType != tt) {
453 LOGERR("Can't mix normal and cancel actions in the same list.\n"
454 "Running the whole batch in the cancel thread.\n");
455 threadType = THREAD_CANCEL;
thatc6085482015-01-09 22:12:43 +0100456 break;
457 }
that7d3b54f2015-01-09 22:52:51 +0100458 }
that73a52952015-01-28 23:07:34 +0100459
460 // Now run the actions in the desired thread.
461 switch (threadType) {
462 case THREAD_ACTION:
463 action_thread.threadActions(this);
464 break;
465
466 case THREAD_CANCEL:
467 cancel_thread.threadActions(this);
468 break;
469
470 default: {
471 // no iterators here because theme reloading might kill our object
472 const size_t cnt = mActions.size();
473 for (size_t i = 0; i < cnt; ++i)
474 doAction(mActions[i]);
475 }
thatc6085482015-01-09 22:12:43 +0100476 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400477
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200478 return 0;
Dees_Troy51a0e822012-09-05 15:24:24 -0400479}
480
that3f7b1ac2014-12-30 11:30:13 +0100481int GUIAction::doAction(Action action)
Dees_Troy51a0e822012-09-05 15:24:24 -0400482{
that3f7b1ac2014-12-30 11:30:13 +0100483 DataManager::GetValue(TW_SIMULATE_ACTIONS, simulate);
Dees_Troy51a0e822012-09-05 15:24:24 -0400484
that3f7b1ac2014-12-30 11:30:13 +0100485 std::string function = gui_parse_text(action.mFunction);
486 std::string arg = gui_parse_text(action.mArg);
Dees_Troy51a0e822012-09-05 15:24:24 -0400487
that3f7b1ac2014-12-30 11:30:13 +0100488 // find function and execute it
489 mapFunc::const_iterator funcitr = mf.find(function);
490 if (funcitr != mf.end())
491 return (this->*funcitr->second)(arg);
492
493 LOGERR("Unknown action '%s'\n", function.c_str());
494 return -1;
Dees_Troy51a0e822012-09-05 15:24:24 -0400495}
496
497void GUIAction::operation_start(const string operation_name)
498{
that3f7b1ac2014-12-30 11:30:13 +0100499 LOGINFO("operation_start: '%s'\n", operation_name.c_str());
Samer Diab (S.a.M.e.R_d)71e9b042014-01-07 20:18:47 +0000500 time(&Start);
Dees_Troy51a0e822012-09-05 15:24:24 -0400501 DataManager::SetValue(TW_ACTION_BUSY, 1);
502 DataManager::SetValue("ui_progress", 0);
Chaosmasterd5364a02020-02-03 15:38:02 +0100503 DataManager::SetValue("ui_portion_size", 0);
504 DataManager::SetValue("ui_portion_start", 0);
Dees_Troy51a0e822012-09-05 15:24:24 -0400505 DataManager::SetValue("tw_operation", operation_name);
Dees_Troy51a0e822012-09-05 15:24:24 -0400506 DataManager::SetValue("tw_operation_state", 0);
Ethan Yonkerd83c9ea2015-01-02 15:22:31 -0600507 DataManager::SetValue("tw_operation_status", 0);
bigbiff25d25b92020-06-19 16:07:38 -0400508 bool tw_ab_device = TWFunc::get_log_dir() != CACHE_LOGS_DIR;
bigbiff bigbiff19874f12019-01-08 20:06:57 -0500509 DataManager::SetValue("tw_ab_device", tw_ab_device);
Dees_Troy51a0e822012-09-05 15:24:24 -0400510}
511
that3f7b1ac2014-12-30 11:30:13 +0100512void GUIAction::operation_end(const int operation_status)
Dees_Troy51a0e822012-09-05 15:24:24 -0400513{
Samer Diab (S.a.M.e.R_d)71e9b042014-01-07 20:18:47 +0000514 time_t Stop;
Dees_Troy51a0e822012-09-05 15:24:24 -0400515 int simulate_fail;
Dees_Troy51a0e822012-09-05 15:24:24 -0400516 DataManager::SetValue("ui_progress", 100);
517 if (simulate) {
518 DataManager::GetValue(TW_SIMULATE_FAIL, simulate_fail);
519 if (simulate_fail != 0)
520 DataManager::SetValue("tw_operation_status", 1);
521 else
522 DataManager::SetValue("tw_operation_status", 0);
523 } else {
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500524 if (operation_status != 0) {
Dees_Troy51a0e822012-09-05 15:24:24 -0400525 DataManager::SetValue("tw_operation_status", 1);
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500526 }
527 else {
Dees_Troy51a0e822012-09-05 15:24:24 -0400528 DataManager::SetValue("tw_operation_status", 0);
bigbiff bigbiff9c754052013-01-09 09:09:08 -0500529 }
Dees_Troy51a0e822012-09-05 15:24:24 -0400530 }
531 DataManager::SetValue("tw_operation_state", 1);
532 DataManager::SetValue(TW_ACTION_BUSY, 0);
bigbiff bigbiff8a68c312013-02-10 14:28:30 -0500533 blankTimer.resetTimerAndUnblank();
LuK133762326f42015-09-30 19:57:46 +0200534 property_set("twrp.action_complete", "1");
Samer Diab (S.a.M.e.R_d)71e9b042014-01-07 20:18:47 +0000535 time(&Stop);
bigbiff bigbiff3ed778a2019-03-12 19:28:31 -0400536
537#ifndef TW_NO_HAPTICS
Samer Diab (S.a.M.e.R_d)71e9b042014-01-07 20:18:47 +0000538 if ((int) difftime(Stop, Start) > 10)
Ethan Yonker03db3262014-02-05 08:02:06 -0600539 DataManager::Vibrate("tw_action_vibrate");
bigbiff bigbiff3ed778a2019-03-12 19:28:31 -0400540#endif
541
that3f7b1ac2014-12-30 11:30:13 +0100542 LOGINFO("operation_end - status=%d\n", operation_status);
Dees_Troy51a0e822012-09-05 15:24:24 -0400543}
544
that3f7b1ac2014-12-30 11:30:13 +0100545int GUIAction::reboot(std::string arg)
Dees_Troy51a0e822012-09-05 15:24:24 -0400546{
that3f7b1ac2014-12-30 11:30:13 +0100547 sync();
548 DataManager::SetValue("tw_gui_done", 1);
549 DataManager::SetValue("tw_reboot_arg", arg);
Dees_Troy51a0e822012-09-05 15:24:24 -0400550
that3f7b1ac2014-12-30 11:30:13 +0100551 return 0;
552}
Dees_Troy51a0e822012-09-05 15:24:24 -0400553
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500554int GUIAction::home(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100555{
that3f7b1ac2014-12-30 11:30:13 +0100556 gui_changePage("main");
557 return 0;
558}
Dees_Troy51a0e822012-09-05 15:24:24 -0400559
that3f7b1ac2014-12-30 11:30:13 +0100560int GUIAction::key(std::string arg)
561{
562 const int key = getKeyByName(arg);
563 PageManager::NotifyKey(key, true);
564 PageManager::NotifyKey(key, false);
565 return 0;
566}
567
568int GUIAction::page(std::string arg)
569{
LuK133762326f42015-09-30 19:57:46 +0200570 property_set("twrp.action_complete", "0");
that3f7b1ac2014-12-30 11:30:13 +0100571 std::string page_name = gui_parse_text(arg);
572 return gui_changePage(page_name);
573}
574
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500575int GUIAction::reload(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100576{
Ethan Yonkere0f1f3b2015-10-27 09:49:01 -0500577 PageManager::RequestReload();
578 // The actual reload is handled in pages.cpp in PageManager::RunReload()
579 // The reload will occur on the next Update or Render call and will
580 // be performed in the rendoer thread instead of the action thread
581 // to prevent crashing which could occur when we start deleting
582 // GUI resources in the action thread while we attempt to render
583 // with those same resources in another thread.
that3f7b1ac2014-12-30 11:30:13 +0100584 return 0;
585}
Dees_Troy51a0e822012-09-05 15:24:24 -0400586
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500587int GUIAction::readBackup(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100588{
589 string Restore_Name;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -0400590
that3f7b1ac2014-12-30 11:30:13 +0100591 DataManager::GetValue("tw_restore", Restore_Name);
592 PartitionManager.Set_Restore_Files(Restore_Name);
593 return 0;
594}
595
596int GUIAction::set(std::string arg)
597{
598 if (arg.find('=') != string::npos)
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200599 {
that3f7b1ac2014-12-30 11:30:13 +0100600 string varName = arg.substr(0, arg.find('='));
601 string value = arg.substr(arg.find('=') + 1, string::npos);
Dees_Troy51a0e822012-09-05 15:24:24 -0400602
that3f7b1ac2014-12-30 11:30:13 +0100603 DataManager::GetValue(value, value);
604 DataManager::SetValue(varName, value);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200605 }
that3f7b1ac2014-12-30 11:30:13 +0100606 else
607 DataManager::SetValue(arg, "1");
608 return 0;
609}
Dees_Troy51a0e822012-09-05 15:24:24 -0400610
that3f7b1ac2014-12-30 11:30:13 +0100611int GUIAction::clear(std::string arg)
612{
613 DataManager::SetValue(arg, "0");
614 return 0;
615}
Dees_Troy51a0e822012-09-05 15:24:24 -0400616
that3f7b1ac2014-12-30 11:30:13 +0100617int GUIAction::mount(std::string arg)
618{
619 if (arg == "usb") {
620 DataManager::SetValue(TW_ACTION_BUSY, 1);
Dees_Troy51a0e822012-09-05 15:24:24 -0400621 if (!simulate)
that3f7b1ac2014-12-30 11:30:13 +0100622 PartitionManager.usb_storage_enable();
623 else
Ethan Yonker74db1572015-10-28 12:44:49 -0500624 gui_msg("simulating=Simulating actions...");
that3f7b1ac2014-12-30 11:30:13 +0100625 } else if (!simulate) {
626 PartitionManager.Mount_By_Path(arg, true);
627 PartitionManager.Add_MTP_Storage(arg);
628 } else
Ethan Yonker74db1572015-10-28 12:44:49 -0500629 gui_msg("simulating=Simulating actions...");
that3f7b1ac2014-12-30 11:30:13 +0100630 return 0;
631}
632
633int GUIAction::unmount(std::string arg)
634{
635 if (arg == "usb") {
636 if (!simulate)
637 PartitionManager.usb_storage_disable();
638 else
Ethan Yonker74db1572015-10-28 12:44:49 -0500639 gui_msg("simulating=Simulating actions...");
that3f7b1ac2014-12-30 11:30:13 +0100640 DataManager::SetValue(TW_ACTION_BUSY, 0);
641 } else if (!simulate) {
642 PartitionManager.UnMount_By_Path(arg, true);
643 } else
Ethan Yonker74db1572015-10-28 12:44:49 -0500644 gui_msg("simulating=Simulating actions...");
that3f7b1ac2014-12-30 11:30:13 +0100645 return 0;
646}
647
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500648int GUIAction::restoredefaultsettings(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100649{
650 operation_start("Restore Defaults");
651 if (simulate) // Simulated so that people don't accidently wipe out the "simulation is on" setting
Ethan Yonker74db1572015-10-28 12:44:49 -0500652 gui_msg("simulating=Simulating actions...");
that3f7b1ac2014-12-30 11:30:13 +0100653 else {
654 DataManager::ResetDefaults();
655 PartitionManager.Update_System_Details();
656 PartitionManager.Mount_Current_Storage(true);
657 }
658 operation_end(0);
659 return 0;
660}
661
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500662int GUIAction::copylog(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100663{
664 operation_start("Copy Log");
665 if (!simulate)
666 {
bigbiff bigbiffbad332a2016-07-29 21:18:13 -0400667 string dst, curr_storage;
668 int copy_kernel_log = 0;
669
670 DataManager::GetValue("tw_include_kernel_log", copy_kernel_log);
that3f7b1ac2014-12-30 11:30:13 +0100671 PartitionManager.Mount_Current_Storage(true);
bigbiff bigbiffbad332a2016-07-29 21:18:13 -0400672 curr_storage = DataManager::GetCurrentStoragePath();
673 dst = curr_storage + "/recovery.log";
that3f7b1ac2014-12-30 11:30:13 +0100674 TWFunc::copy_file("/tmp/recovery.log", dst.c_str(), 0755);
675 tw_set_default_metadata(dst.c_str());
bigbiff bigbiffbad332a2016-07-29 21:18:13 -0400676 if (copy_kernel_log)
677 TWFunc::copy_kernel_log(curr_storage);
that3f7b1ac2014-12-30 11:30:13 +0100678 sync();
bigbiff bigbiffbad332a2016-07-29 21:18:13 -0400679 gui_msg(Msg("copy_log=Copied recovery log to {1}")(dst));
that3f7b1ac2014-12-30 11:30:13 +0100680 } else
681 simulate_progress_bar();
682 operation_end(0);
683 return 0;
684}
685
686
687int GUIAction::compute(std::string arg)
688{
689 if (arg.find("+") != string::npos)
690 {
691 string varName = arg.substr(0, arg.find('+'));
692 string string_to_add = arg.substr(arg.find('+') + 1, string::npos);
693 int amount_to_add = atoi(string_to_add.c_str());
694 int value;
695
696 DataManager::GetValue(varName, value);
697 DataManager::SetValue(varName, value + amount_to_add);
Dees_Troy51a0e822012-09-05 15:24:24 -0400698 return 0;
699 }
that3f7b1ac2014-12-30 11:30:13 +0100700 if (arg.find("-") != string::npos)
Dees_Troy51a0e822012-09-05 15:24:24 -0400701 {
that3f7b1ac2014-12-30 11:30:13 +0100702 string varName = arg.substr(0, arg.find('-'));
703 string string_to_subtract = arg.substr(arg.find('-') + 1, string::npos);
704 int amount_to_subtract = atoi(string_to_subtract.c_str());
705 int value;
Dees_Troy51a0e822012-09-05 15:24:24 -0400706
that3f7b1ac2014-12-30 11:30:13 +0100707 DataManager::GetValue(varName, value);
708 value -= amount_to_subtract;
709 if (value <= 0)
710 value = 0;
711 DataManager::SetValue(varName, value);
712 return 0;
713 }
714 if (arg.find("*") != string::npos)
715 {
716 string varName = arg.substr(0, arg.find('*'));
717 string multiply_by_str = gui_parse_text(arg.substr(arg.find('*') + 1, string::npos));
718 int multiply_by = atoi(multiply_by_str.c_str());
719 int value;
720
721 DataManager::GetValue(varName, value);
722 DataManager::SetValue(varName, value*multiply_by);
723 return 0;
724 }
725 if (arg.find("/") != string::npos)
726 {
727 string varName = arg.substr(0, arg.find('/'));
728 string divide_by_str = gui_parse_text(arg.substr(arg.find('/') + 1, string::npos));
729 int divide_by = atoi(divide_by_str.c_str());
730 int value;
731
Matt Mowera8a89d12016-12-30 18:10:37 -0600732 if (divide_by != 0)
that3f7b1ac2014-12-30 11:30:13 +0100733 {
Dees_Troy51a0e822012-09-05 15:24:24 -0400734 DataManager::GetValue(varName, value);
that3f7b1ac2014-12-30 11:30:13 +0100735 DataManager::SetValue(varName, value/divide_by);
Dees_Troy51a0e822012-09-05 15:24:24 -0400736 }
737 return 0;
738 }
that3f7b1ac2014-12-30 11:30:13 +0100739 LOGERR("Unable to perform compute '%s'\n", arg.c_str());
740 return -1;
741}
Dees_Troy51a0e822012-09-05 15:24:24 -0400742
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500743int GUIAction::setguitimezone(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100744{
745 string SelectedZone;
746 DataManager::GetValue(TW_TIME_ZONE_GUISEL, SelectedZone); // read the selected time zone into SelectedZone
747 string Zone = SelectedZone.substr(0, SelectedZone.find(';')); // parse to get time zone
748 string DSTZone = SelectedZone.substr(SelectedZone.find(';') + 1, string::npos); // parse to get DST component
749
750 int dst;
751 DataManager::GetValue(TW_TIME_ZONE_GUIDST, dst); // check wether user chose to use DST
752
753 string offset;
754 DataManager::GetValue(TW_TIME_ZONE_GUIOFFSET, offset); // pull in offset
755
756 string NewTimeZone = Zone;
757 if (offset != "0")
758 NewTimeZone += ":" + offset;
759
760 if (dst != 0)
761 NewTimeZone += DSTZone;
762
763 DataManager::SetValue(TW_TIME_ZONE_VAR, NewTimeZone);
764 DataManager::update_tz_environment_variables();
765 return 0;
766}
767
768int GUIAction::overlay(std::string arg)
769{
770 return gui_changeOverlay(arg);
771}
772
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500773int GUIAction::queuezip(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100774{
775 if (zip_queue_index >= 10) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500776 gui_msg("max_queue=Maximum zip queue reached!");
Dees_Troy51a0e822012-09-05 15:24:24 -0400777 return 0;
778 }
that3f7b1ac2014-12-30 11:30:13 +0100779 DataManager::GetValue("tw_filename", zip_queue[zip_queue_index]);
780 if (strlen(zip_queue[zip_queue_index].c_str()) > 0) {
781 zip_queue_index++;
Dees_Troy51a0e822012-09-05 15:24:24 -0400782 DataManager::SetValue(TW_ZIP_QUEUE_COUNT, zip_queue_index);
that3f7b1ac2014-12-30 11:30:13 +0100783 }
784 return 0;
785}
786
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500787int GUIAction::cancelzip(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100788{
789 if (zip_queue_index <= 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500790 gui_msg("min_queue=Minimum zip queue reached!");
Dees_Troy51a0e822012-09-05 15:24:24 -0400791 return 0;
that3f7b1ac2014-12-30 11:30:13 +0100792 } else {
793 zip_queue_index--;
794 DataManager::SetValue(TW_ZIP_QUEUE_COUNT, zip_queue_index);
Dees_Troy51a0e822012-09-05 15:24:24 -0400795 }
that3f7b1ac2014-12-30 11:30:13 +0100796 return 0;
797}
Dees_Troy51a0e822012-09-05 15:24:24 -0400798
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500799int GUIAction::queueclear(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100800{
801 zip_queue_index = 0;
802 DataManager::SetValue(TW_ZIP_QUEUE_COUNT, zip_queue_index);
803 return 0;
804}
Dees_Troy51a0e822012-09-05 15:24:24 -0400805
that3f7b1ac2014-12-30 11:30:13 +0100806int GUIAction::sleep(std::string arg)
807{
808 operation_start("Sleep");
809 usleep(atoi(arg.c_str()));
810 operation_end(0);
811 return 0;
812}
Dees Troyb21cc642013-09-10 17:36:41 +0000813
Matt Mower9a2a2052016-05-31 21:31:22 -0500814int GUIAction::sleepcounter(std::string arg)
815{
816 operation_start("SleepCounter");
817 // Ensure user notices countdown in case it needs to be cancelled
818 blankTimer.resetTimerAndUnblank();
819 int total = atoi(arg.c_str());
820 for (int t = total; t > 0; t--) {
821 int progress = (int)(((float)(total-t)/(float)total)*100.0);
822 DataManager::SetValue("ui_progress", progress);
823 ::sleep(1);
824 DataManager::SetValue("tw_sleep", t-1);
825 }
826 DataManager::SetValue("ui_progress", 100);
827 operation_end(0);
828 return 0;
829}
830
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500831int GUIAction::appenddatetobackupname(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100832{
833 operation_start("AppendDateToBackupName");
834 string Backup_Name;
835 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
836 Backup_Name += TWFunc::Get_Current_Date();
837 if (Backup_Name.size() > MAX_BACKUP_NAME_LEN)
838 Backup_Name.resize(MAX_BACKUP_NAME_LEN);
839 DataManager::SetValue(TW_BACKUP_NAME, Backup_Name);
Matt Mower76b8afc2016-06-24 12:04:27 -0500840 PageManager::NotifyKey(KEY_END, true);
841 PageManager::NotifyKey(KEY_END, false);
that3f7b1ac2014-12-30 11:30:13 +0100842 operation_end(0);
843 return 0;
844}
Ethan Yonker87c7bac2014-05-25 21:41:08 -0500845
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500846int GUIAction::generatebackupname(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100847{
848 operation_start("GenerateBackupName");
849 TWFunc::Auto_Generate_Backup_Name();
850 operation_end(0);
851 return 0;
852}
Ethan Yonker87c7bac2014-05-25 21:41:08 -0500853
Ethan Yonker483e9f42016-01-11 22:21:18 -0600854int GUIAction::checkpartitionlist(std::string arg)
that3f7b1ac2014-12-30 11:30:13 +0100855{
Ethan Yonker483e9f42016-01-11 22:21:18 -0600856 string List, part_path;
that3f7b1ac2014-12-30 11:30:13 +0100857 int count = 0;
Ethan Yonker87c7bac2014-05-25 21:41:08 -0500858
Ethan Yonker483e9f42016-01-11 22:21:18 -0600859 if (arg.empty())
860 arg = "tw_wipe_list";
861 DataManager::GetValue(arg, List);
862 LOGINFO("checkpartitionlist list '%s'\n", List.c_str());
863 if (!List.empty()) {
864 size_t start_pos = 0, end_pos = List.find(";", start_pos);
865 while (end_pos != string::npos && start_pos < List.size()) {
866 part_path = List.substr(start_pos, end_pos - start_pos);
867 LOGINFO("checkpartitionlist part_path '%s'\n", part_path.c_str());
868 if (part_path == "/and-sec" || part_path == "DALVIK" || part_path == "INTERNAL") {
that3f7b1ac2014-12-30 11:30:13 +0100869 // Do nothing
Dees_Troy51a0e822012-09-05 15:24:24 -0400870 } else {
that3f7b1ac2014-12-30 11:30:13 +0100871 count++;
872 }
873 start_pos = end_pos + 1;
Ethan Yonker483e9f42016-01-11 22:21:18 -0600874 end_pos = List.find(";", start_pos);
that3f7b1ac2014-12-30 11:30:13 +0100875 }
876 DataManager::SetValue("tw_check_partition_list", count);
877 } else {
878 DataManager::SetValue("tw_check_partition_list", 0);
879 }
Matt Mowera8a89d12016-12-30 18:10:37 -0600880 return 0;
that3f7b1ac2014-12-30 11:30:13 +0100881}
Dees_Troy51a0e822012-09-05 15:24:24 -0400882
Ethan Yonker483e9f42016-01-11 22:21:18 -0600883int GUIAction::getpartitiondetails(std::string arg)
that3f7b1ac2014-12-30 11:30:13 +0100884{
Ethan Yonker483e9f42016-01-11 22:21:18 -0600885 string List, part_path;
Dees_Troy51a0e822012-09-05 15:24:24 -0400886
Ethan Yonker483e9f42016-01-11 22:21:18 -0600887 if (arg.empty())
888 arg = "tw_wipe_list";
889 DataManager::GetValue(arg, List);
890 LOGINFO("getpartitiondetails list '%s'\n", List.c_str());
891 if (!List.empty()) {
892 size_t start_pos = 0, end_pos = List.find(";", start_pos);
893 part_path = List;
894 while (end_pos != string::npos && start_pos < List.size()) {
895 part_path = List.substr(start_pos, end_pos - start_pos);
896 LOGINFO("getpartitiondetails part_path '%s'\n", part_path.c_str());
897 if (part_path == "/and-sec" || part_path == "DALVIK" || part_path == "INTERNAL") {
that3f7b1ac2014-12-30 11:30:13 +0100898 // Do nothing
899 } else {
Ethan Yonker483e9f42016-01-11 22:21:18 -0600900 DataManager::SetValue("tw_partition_path", part_path);
that3f7b1ac2014-12-30 11:30:13 +0100901 break;
902 }
903 start_pos = end_pos + 1;
Ethan Yonker483e9f42016-01-11 22:21:18 -0600904 end_pos = List.find(";", start_pos);
that3f7b1ac2014-12-30 11:30:13 +0100905 }
Ethan Yonker483e9f42016-01-11 22:21:18 -0600906 if (!part_path.empty()) {
907 TWPartition* Part = PartitionManager.Find_Partition_By_Path(part_path);
that3f7b1ac2014-12-30 11:30:13 +0100908 if (Part) {
909 unsigned long long mb = 1048576;
Dees_Troya13d74f2013-03-24 08:54:55 -0500910
that3f7b1ac2014-12-30 11:30:13 +0100911 DataManager::SetValue("tw_partition_name", Part->Display_Name);
912 DataManager::SetValue("tw_partition_mount_point", Part->Mount_Point);
913 DataManager::SetValue("tw_partition_file_system", Part->Current_File_System);
914 DataManager::SetValue("tw_partition_size", Part->Size / mb);
915 DataManager::SetValue("tw_partition_used", Part->Used / mb);
916 DataManager::SetValue("tw_partition_free", Part->Free / mb);
917 DataManager::SetValue("tw_partition_backup_size", Part->Backup_Size / mb);
918 DataManager::SetValue("tw_partition_removable", Part->Removable);
919 DataManager::SetValue("tw_partition_is_present", Part->Is_Present);
920
921 if (Part->Can_Repair())
922 DataManager::SetValue("tw_partition_can_repair", 1);
923 else
924 DataManager::SetValue("tw_partition_can_repair", 0);
Ethan Yonkera2719152015-05-28 09:44:41 -0500925 if (Part->Can_Resize())
926 DataManager::SetValue("tw_partition_can_resize", 1);
927 else
928 DataManager::SetValue("tw_partition_can_resize", 0);
bigbiffad58e1b2020-07-06 20:24:34 -0400929 if (TWFunc::Path_Exists("/system/bin/mkfs.fat"))
that3f7b1ac2014-12-30 11:30:13 +0100930 DataManager::SetValue("tw_partition_vfat", 1);
931 else
932 DataManager::SetValue("tw_partition_vfat", 0);
bigbiffad58e1b2020-07-06 20:24:34 -0400933 if (TWFunc::Path_Exists("/system/bin/mkexfatfs"))
that3f7b1ac2014-12-30 11:30:13 +0100934 DataManager::SetValue("tw_partition_exfat", 1);
935 else
936 DataManager::SetValue("tw_partition_exfat", 0);
bigbiffad58e1b2020-07-06 20:24:34 -0400937 if (TWFunc::Path_Exists("/system/bin/mkfs.f2fs"))
that3f7b1ac2014-12-30 11:30:13 +0100938 DataManager::SetValue("tw_partition_f2fs", 1);
939 else
940 DataManager::SetValue("tw_partition_f2fs", 0);
bigbiffad58e1b2020-07-06 20:24:34 -0400941 if (TWFunc::Path_Exists("/system/bin/mke2fs"))
that3f7b1ac2014-12-30 11:30:13 +0100942 DataManager::SetValue("tw_partition_ext", 1);
943 else
944 DataManager::SetValue("tw_partition_ext", 0);
945 return 0;
946 } else {
Ethan Yonker483e9f42016-01-11 22:21:18 -0600947 LOGERR("Unable to locate partition: '%s'\n", part_path.c_str());
that3f7b1ac2014-12-30 11:30:13 +0100948 }
949 }
950 }
951 DataManager::SetValue("tw_partition_name", "");
952 DataManager::SetValue("tw_partition_file_system", "");
Ethan Yonker483e9f42016-01-11 22:21:18 -0600953 // Set this to 0 to prevent trying to partition this device, just in case
954 DataManager::SetValue("tw_partition_removable", 0);
that3f7b1ac2014-12-30 11:30:13 +0100955 return 0;
956}
957
Ethan Yonkerd0514ba2015-10-22 14:17:47 -0500958int GUIAction::screenshot(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +0100959{
960 time_t tm;
961 char path[256];
962 int path_len;
Matt Mower0c88b842017-01-15 16:00:49 -0600963 uid_t uid = AID_MEDIA_RW;
964 gid_t gid = AID_MEDIA_RW;
that3f7b1ac2014-12-30 11:30:13 +0100965
966 const std::string storage = DataManager::GetCurrentStoragePath();
Matt Mowera8a89d12016-12-30 18:10:37 -0600967 if (PartitionManager.Is_Mounted_By_Path(storage)) {
that3f7b1ac2014-12-30 11:30:13 +0100968 snprintf(path, sizeof(path), "%s/Pictures/Screenshots/", storage.c_str());
969 } else {
970 strcpy(path, "/tmp/");
971 }
972
Matt Mowera8a89d12016-12-30 18:10:37 -0600973 if (!TWFunc::Create_Dir_Recursive(path, 0775, uid, gid))
that3f7b1ac2014-12-30 11:30:13 +0100974 return 0;
975
976 tm = time(NULL);
977 path_len = strlen(path);
978
979 // Screenshot_2014-01-01-18-21-38.png
980 strftime(path+path_len, sizeof(path)-path_len, "Screenshot_%Y-%m-%d-%H-%M-%S.png", localtime(&tm));
981
982 int res = gr_save_screenshot(path);
Matt Mowera8a89d12016-12-30 18:10:37 -0600983 if (res == 0) {
that3f7b1ac2014-12-30 11:30:13 +0100984 chmod(path, 0666);
985 chown(path, uid, gid);
986
that677b13f2015-12-26 23:57:31 +0100987 gui_msg(Msg("screenshot_saved=Screenshot was saved to {1}")(path));
that3f7b1ac2014-12-30 11:30:13 +0100988
VDavid0032034a412019-10-18 22:43:20 +0200989 // blink to notify that the screenshot was taken
that3f7b1ac2014-12-30 11:30:13 +0100990 gr_color(255, 255, 255, 255);
991 gr_fill(0, 0, gr_fb_width(), gr_fb_height());
992 gr_flip();
993 gui_forceRender();
994 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500995 gui_err("screenshot_err=Failed to take a screenshot!");
that3f7b1ac2014-12-30 11:30:13 +0100996 }
997 return 0;
998}
999
1000int GUIAction::setbrightness(std::string arg)
1001{
1002 return TWFunc::Set_Brightness(arg);
1003}
1004
1005int GUIAction::fileexists(std::string arg)
1006{
1007 struct stat st;
1008 string newpath = arg + "/.";
1009
1010 operation_start("FileExists");
1011 if (stat(arg.c_str(), &st) == 0 || stat(newpath.c_str(), &st) == 0)
1012 operation_end(0);
1013 else
1014 operation_end(1);
1015 return 0;
1016}
1017
thatcc8ddca2015-01-03 01:59:36 +01001018void GUIAction::reinject_after_flash()
1019{
1020 if (DataManager::GetIntValue(TW_HAS_INJECTTWRP) == 1 && DataManager::GetIntValue(TW_INJECT_AFTER_ZIP) == 1) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001021 gui_msg("injecttwrp=Injecting TWRP into boot image...");
thatcc8ddca2015-01-03 01:59:36 +01001022 if (simulate) {
1023 simulate_progress_bar();
1024 } else {
1025 TWPartition* Boot = PartitionManager.Find_Partition_By_Path("/boot");
1026 if (Boot == NULL || Boot->Current_File_System != "emmc")
1027 TWFunc::Exec_Cmd("injecttwrp --dump /tmp/backup_recovery_ramdisk.img /tmp/injected_boot.img --flash");
1028 else {
1029 string injectcmd = "injecttwrp --dump /tmp/backup_recovery_ramdisk.img /tmp/injected_boot.img --flash bd=" + Boot->Actual_Block_Device;
1030 TWFunc::Exec_Cmd(injectcmd);
1031 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001032 gui_msg("done=Done.");
thatcc8ddca2015-01-03 01:59:36 +01001033 }
1034 }
1035}
1036
mauronofrio6d3bf892019-10-26 19:47:55 +02001037int GUIAction::ozip_decrypt(string zip_path)
1038{
bigbiffad58e1b2020-07-06 20:24:34 -04001039 if (!TWFunc::Path_Exists("/system/bin/ozip_decrypt")) {
mauronofrio6d3bf892019-10-26 19:47:55 +02001040 return 1;
1041 }
1042 gui_msg("ozip_decrypt_decryption=Starting Ozip Decryption...");
1043 TWFunc::Exec_Cmd("ozip_decrypt " + (string)TW_OZIP_DECRYPT_KEY + " '" + zip_path + "'");
1044 gui_msg("ozip_decrypt_finish=Ozip Decryption Finished!");
1045 return 0;
1046}
1047
that3f7b1ac2014-12-30 11:30:13 +01001048int GUIAction::flash(std::string arg)
1049{
1050 int i, ret_val = 0, wipe_cache = 0;
Ethan Yonker0d47eb52015-01-09 11:23:19 -06001051 // We're going to jump to this page first, like a loading page
1052 gui_changePage(arg);
that3f7b1ac2014-12-30 11:30:13 +01001053 for (i=0; i<zip_queue_index; i++) {
thatc7b631b2015-06-01 23:36:57 +02001054 string zip_path = zip_queue[i];
1055 size_t slashpos = zip_path.find_last_of('/');
1056 string zip_filename = (slashpos == string::npos) ? zip_path : zip_path.substr(slashpos + 1);
that3f7b1ac2014-12-30 11:30:13 +01001057 operation_start("Flashing");
mauronofrio6d3bf892019-10-26 19:47:55 +02001058 if((zip_path.substr(zip_path.size() - 4, 4)) == "ozip")
1059 {
1060 if((ozip_decrypt(zip_path)) != 0)
1061 {
1062 LOGERR("Unable to find ozip_decrypt!");
1063 break;
1064 }
1065 zip_filename = (zip_filename.substr(0, zip_filename.size() - 4)).append("zip");
1066 zip_path = (zip_path.substr(0, zip_path.size() - 4)).append("zip");
1067 if (!TWFunc::Path_Exists(zip_path)) {
1068 LOGERR("Unable to find decrypted zip");
1069 break;
1070 }
1071 }
thatc7b631b2015-06-01 23:36:57 +02001072 DataManager::SetValue("tw_filename", zip_path);
1073 DataManager::SetValue("tw_file", zip_filename);
that3f7b1ac2014-12-30 11:30:13 +01001074 DataManager::SetValue(TW_ZIP_INDEX, (i + 1));
1075
1076 TWFunc::SetPerformanceMode(true);
thatc7b631b2015-06-01 23:36:57 +02001077 ret_val = flash_zip(zip_path, &wipe_cache);
that3f7b1ac2014-12-30 11:30:13 +01001078 TWFunc::SetPerformanceMode(false);
1079 if (ret_val != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001080 gui_msg(Msg(msg::kError, "zip_err=Error installing zip file '{1}'")(zip_path));
that3f7b1ac2014-12-30 11:30:13 +01001081 ret_val = 1;
Ethan Yonker0d47eb52015-01-09 11:23:19 -06001082 break;
that3f7b1ac2014-12-30 11:30:13 +01001083 }
1084 }
1085 zip_queue_index = 0;
that3f7b1ac2014-12-30 11:30:13 +01001086
Ethan Yonker0d47eb52015-01-09 11:23:19 -06001087 if (wipe_cache) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001088 gui_msg("zip_wipe_cache=One or more zip requested a cache wipe -- Wiping cache now.");
that3f7b1ac2014-12-30 11:30:13 +01001089 PartitionManager.Wipe_By_Path("/cache");
Ethan Yonker0d47eb52015-01-09 11:23:19 -06001090 }
that3f7b1ac2014-12-30 11:30:13 +01001091
thatcc8ddca2015-01-03 01:59:36 +01001092 reinject_after_flash();
that3f7b1ac2014-12-30 11:30:13 +01001093 PartitionManager.Update_System_Details();
1094 operation_end(ret_val);
bigbiffa869fc72016-03-01 19:40:36 -05001095 // This needs to be after the operation_end call so we change pages before we change variables that we display on the screen
Ethan Yonker0d47eb52015-01-09 11:23:19 -06001096 DataManager::SetValue(TW_ZIP_QUEUE_COUNT, zip_queue_index);
that3f7b1ac2014-12-30 11:30:13 +01001097 return 0;
1098}
1099
1100int GUIAction::wipe(std::string arg)
1101{
1102 operation_start("Format");
1103 DataManager::SetValue("tw_partition", arg);
that3f7b1ac2014-12-30 11:30:13 +01001104 int ret_val = false;
1105
1106 if (simulate) {
1107 simulate_progress_bar();
1108 } else {
1109 if (arg == "data")
1110 ret_val = PartitionManager.Factory_Reset();
1111 else if (arg == "battery")
1112 ret_val = PartitionManager.Wipe_Battery_Stats();
1113 else if (arg == "rotate")
1114 ret_val = PartitionManager.Wipe_Rotate_Data();
1115 else if (arg == "dalvik")
1116 ret_val = PartitionManager.Wipe_Dalvik_Cache();
1117 else if (arg == "DATAMEDIA") {
1118 ret_val = PartitionManager.Format_Data();
1119 } else if (arg == "INTERNAL") {
Matt Mower23d8aae2017-01-06 14:30:33 -06001120 int has_datamedia;
that3f7b1ac2014-12-30 11:30:13 +01001121
1122 DataManager::GetValue(TW_HAS_DATA_MEDIA, has_datamedia);
1123 if (has_datamedia) {
1124 ret_val = PartitionManager.Wipe_Media_From_Data();
1125 } else {
1126 ret_val = PartitionManager.Wipe_By_Path(DataManager::GetSettingsStoragePath());
1127 }
1128 } else if (arg == "EXTERNAL") {
1129 string External_Path;
1130
1131 DataManager::GetValue(TW_EXTERNAL_PATH, External_Path);
1132 ret_val = PartitionManager.Wipe_By_Path(External_Path);
1133 } else if (arg == "ANDROIDSECURE") {
1134 ret_val = PartitionManager.Wipe_Android_Secure();
1135 } else if (arg == "LIST") {
1136 string Wipe_List, wipe_path;
1137 bool skip = false;
1138 ret_val = true;
that3f7b1ac2014-12-30 11:30:13 +01001139
1140 DataManager::GetValue("tw_wipe_list", Wipe_List);
1141 LOGINFO("wipe list '%s'\n", Wipe_List.c_str());
1142 if (!Wipe_List.empty()) {
1143 size_t start_pos = 0, end_pos = Wipe_List.find(";", start_pos);
1144 while (end_pos != string::npos && start_pos < Wipe_List.size()) {
1145 wipe_path = Wipe_List.substr(start_pos, end_pos - start_pos);
1146 LOGINFO("wipe_path '%s'\n", wipe_path.c_str());
1147 if (wipe_path == "/and-sec") {
1148 if (!PartitionManager.Wipe_Android_Secure()) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001149 gui_msg("and_sec_wipe_err=Unable to wipe android secure");
that3f7b1ac2014-12-30 11:30:13 +01001150 ret_val = false;
1151 break;
1152 } else {
1153 skip = true;
1154 }
1155 } else if (wipe_path == "DALVIK") {
1156 if (!PartitionManager.Wipe_Dalvik_Cache()) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001157 gui_err("dalvik_wipe_err=Failed to wipe dalvik");
that3f7b1ac2014-12-30 11:30:13 +01001158 ret_val = false;
1159 break;
1160 } else {
1161 skip = true;
1162 }
1163 } else if (wipe_path == "INTERNAL") {
1164 if (!PartitionManager.Wipe_Media_From_Data()) {
1165 ret_val = false;
1166 break;
1167 } else {
1168 skip = true;
Dees_Troya13d74f2013-03-24 08:54:55 -05001169 }
1170 }
that3f7b1ac2014-12-30 11:30:13 +01001171 if (!skip) {
1172 if (!PartitionManager.Wipe_By_Path(wipe_path)) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001173 gui_msg(Msg(msg::kError, "unable_to_wipe=Unable to wipe {1}.")(wipe_path));
that3f7b1ac2014-12-30 11:30:13 +01001174 ret_val = false;
1175 break;
1176 } else if (wipe_path == DataManager::GetSettingsStoragePath()) {
1177 arg = wipe_path;
1178 }
Dees_Troy38bd7602012-09-14 13:33:53 -04001179 } else {
that3f7b1ac2014-12-30 11:30:13 +01001180 skip = false;
Dees_Troy38bd7602012-09-14 13:33:53 -04001181 }
that3f7b1ac2014-12-30 11:30:13 +01001182 start_pos = end_pos + 1;
1183 end_pos = Wipe_List.find(";", start_pos);
Dees_Troy51a0e822012-09-05 15:24:24 -04001184 }
1185 }
that3f7b1ac2014-12-30 11:30:13 +01001186 } else
1187 ret_val = PartitionManager.Wipe_By_Path(arg);
that73a52952015-01-28 23:07:34 +01001188#ifndef TW_OEM_BUILD
that3f7b1ac2014-12-30 11:30:13 +01001189 if (arg == DataManager::GetSettingsStoragePath()) {
1190 // If we wiped the settings storage path, recreate the TWRP folder and dump the settings
1191 string Storage_Path = DataManager::GetSettingsStoragePath();
Dees_Troy51a0e822012-09-05 15:24:24 -04001192
that3f7b1ac2014-12-30 11:30:13 +01001193 if (PartitionManager.Mount_By_Path(Storage_Path, true)) {
1194 LOGINFO("Making TWRP folder and saving settings.\n");
1195 Storage_Path += "/TWRP";
1196 mkdir(Storage_Path.c_str(), 0777);
1197 DataManager::Flush();
Dees_Troy51a0e822012-09-05 15:24:24 -04001198 } else {
that3f7b1ac2014-12-30 11:30:13 +01001199 LOGERR("Unable to recreate TWRP folder and save settings.\n");
1200 }
1201 }
1202#endif
1203 }
1204 PartitionManager.Update_System_Details();
1205 if (ret_val)
1206 ret_val = 0; // 0 is success
1207 else
1208 ret_val = 1; // 1 is failure
1209 operation_end(ret_val);
1210 return 0;
1211}
Samer Diab (S.a.M.e.R_d)71e9b042014-01-07 20:18:47 +00001212
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001213int GUIAction::refreshsizes(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001214{
1215 operation_start("Refreshing Sizes");
1216 if (simulate) {
1217 simulate_progress_bar();
1218 } else
1219 PartitionManager.Update_System_Details();
1220 operation_end(0);
1221 return 0;
1222}
1223
1224int GUIAction::nandroid(std::string arg)
1225{
that3f7b1ac2014-12-30 11:30:13 +01001226 if (simulate) {
that73a52952015-01-28 23:07:34 +01001227 PartitionManager.stop_backup.set_value(0);
that3f7b1ac2014-12-30 11:30:13 +01001228 DataManager::SetValue("tw_partition", "Simulation");
1229 simulate_progress_bar();
that73a52952015-01-28 23:07:34 +01001230 operation_end(0);
that3f7b1ac2014-12-30 11:30:13 +01001231 } else {
bigbiff7abc5fe2015-01-17 16:53:12 -05001232 operation_start("Nandroid");
1233 int ret = 0;
1234
that3f7b1ac2014-12-30 11:30:13 +01001235 if (arg == "backup") {
1236 string Backup_Name;
1237 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
Ethan Yonker4adc33e2016-01-22 16:07:11 -06001238 string auto_gen = gui_lookup("auto_generate", "(Auto Generate)");
Ethan Yonker53796e72019-01-11 22:49:52 -06001239 if (Backup_Name == auto_gen || Backup_Name == gui_lookup("curr_date", "(Current Date)") || Backup_Name == "0" || Backup_Name == "(" || PartitionManager.Check_Backup_Name(Backup_Name, true, true) == 0) {
bigbiffce8f83c2015-12-12 18:30:21 -05001240 ret = PartitionManager.Run_Backup(false);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001241 DataManager::SetValue("tw_encrypt_backup", 0); // reset value so we don't encrypt every subsequent backup
1242 if (!PartitionManager.stop_backup.get_value()) {
1243 if (ret == false)
1244 ret = 1; // 1 for failure
1245 else
1246 ret = 0; // 0 for success
1247 DataManager::SetValue("tw_cancel_backup", 0);
1248 } else {
1249 DataManager::SetValue("tw_cancel_backup", 1);
1250 gui_msg("backup_cancel=Backup Cancelled");
1251 ret = 0;
1252 }
bigbiffce8f83c2015-12-12 18:30:21 -05001253 } else {
that3f7b1ac2014-12-30 11:30:13 +01001254 operation_end(1);
1255 return -1;
that3f7b1ac2014-12-30 11:30:13 +01001256 }
Ethan Yonker74db1572015-10-28 12:44:49 -05001257 DataManager::SetValue(TW_BACKUP_NAME, auto_gen);
that3f7b1ac2014-12-30 11:30:13 +01001258 } else if (arg == "restore") {
1259 string Restore_Name;
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001260 int gui_adb_backup;
1261
that3f7b1ac2014-12-30 11:30:13 +01001262 DataManager::GetValue("tw_restore", Restore_Name);
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001263 DataManager::GetValue("tw_enable_adb_backup", gui_adb_backup);
1264 if (gui_adb_backup) {
1265 DataManager::SetValue("tw_operation_state", 1);
1266 if (TWFunc::stream_adb_backup(Restore_Name) == 0)
1267 ret = 0; // success
1268 else
1269 ret = 1; // failure
1270 DataManager::SetValue("tw_enable_adb_backup", 0);
1271 ret = 0; // assume success???
1272 } else {
1273 if (PartitionManager.Run_Restore(Restore_Name))
1274 ret = 0; // success
1275 else
1276 ret = 1; // failure
1277 }
that3f7b1ac2014-12-30 11:30:13 +01001278 } else {
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04001279 operation_end(1); // invalid arg specified, fail
bigbiff7abc5fe2015-01-17 16:53:12 -05001280 return -1;
1281 }
that73a52952015-01-28 23:07:34 +01001282 operation_end(ret);
bigbiff7abc5fe2015-01-17 16:53:12 -05001283 return ret;
1284 }
1285 return 0;
1286}
1287
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001288int GUIAction::cancelbackup(std::string arg __unused) {
bigbiff7abc5fe2015-01-17 16:53:12 -05001289 if (simulate) {
bigbiff7abc5fe2015-01-17 16:53:12 -05001290 PartitionManager.stop_backup.set_value(1);
bigbiff7abc5fe2015-01-17 16:53:12 -05001291 }
1292 else {
bigbiff7abc5fe2015-01-17 16:53:12 -05001293 int op_status = PartitionManager.Cancel_Backup();
1294 if (op_status != 0)
1295 op_status = 1; // failure
bigbiff7abc5fe2015-01-17 16:53:12 -05001296 }
1297
1298 return 0;
that3f7b1ac2014-12-30 11:30:13 +01001299}
Dees_Troy51a0e822012-09-05 15:24:24 -04001300
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001301int GUIAction::fixcontexts(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001302{
that73a52952015-01-28 23:07:34 +01001303 int op_status = 0;
1304
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001305 operation_start("Fix Contexts");
1306 LOGINFO("fix contexts started!\n");
that3f7b1ac2014-12-30 11:30:13 +01001307 if (simulate) {
1308 simulate_progress_bar();
1309 } else {
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001310 op_status = PartitionManager.Fix_Contexts();
that3f7b1ac2014-12-30 11:30:13 +01001311 if (op_status != 0)
1312 op_status = 1; // failure
that3f7b1ac2014-12-30 11:30:13 +01001313 }
that73a52952015-01-28 23:07:34 +01001314 operation_end(op_status);
that3f7b1ac2014-12-30 11:30:13 +01001315 return 0;
1316}
1317
Ethan Yonkerb5fab762016-01-28 14:03:33 -06001318int GUIAction::fixpermissions(std::string arg)
1319{
1320 return fixcontexts(arg);
1321}
1322
that3f7b1ac2014-12-30 11:30:13 +01001323int GUIAction::dd(std::string arg)
1324{
1325 operation_start("imaging");
1326
1327 if (simulate) {
1328 simulate_progress_bar();
1329 } else {
1330 string cmd = "dd " + arg;
1331 TWFunc::Exec_Cmd(cmd);
1332 }
1333 operation_end(0);
1334 return 0;
1335}
1336
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001337int GUIAction::partitionsd(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001338{
1339 operation_start("Partition SD Card");
1340 int ret_val = 0;
1341
1342 if (simulate) {
1343 simulate_progress_bar();
1344 } else {
1345 int allow_partition;
1346 DataManager::GetValue(TW_ALLOW_PARTITION_SDCARD, allow_partition);
1347 if (allow_partition == 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001348 gui_err("no_real_sdcard=This device does not have a real SD Card! Aborting!");
that3f7b1ac2014-12-30 11:30:13 +01001349 } else {
1350 if (!PartitionManager.Partition_SDCard())
1351 ret_val = 1; // failed
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001352 }
that3f7b1ac2014-12-30 11:30:13 +01001353 }
1354 operation_end(ret_val);
1355 return 0;
Dees_Troy51a0e822012-09-05 15:24:24 -04001356
that3f7b1ac2014-12-30 11:30:13 +01001357}
Dees_Troy51a0e822012-09-05 15:24:24 -04001358
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001359int GUIAction::installhtcdumlock(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001360{
1361 operation_start("Install HTC Dumlock");
1362 if (simulate) {
1363 simulate_progress_bar();
1364 } else
1365 TWFunc::install_htc_dumlock();
Dees_Troy51a0e822012-09-05 15:24:24 -04001366
that3f7b1ac2014-12-30 11:30:13 +01001367 operation_end(0);
1368 return 0;
1369}
Dees_Troy51a0e822012-09-05 15:24:24 -04001370
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001371int GUIAction::htcdumlockrestoreboot(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001372{
1373 operation_start("HTC Dumlock Restore Boot");
1374 if (simulate) {
1375 simulate_progress_bar();
1376 } else
1377 TWFunc::htc_dumlock_restore_original_boot();
Dees_Troy51a0e822012-09-05 15:24:24 -04001378
that3f7b1ac2014-12-30 11:30:13 +01001379 operation_end(0);
1380 return 0;
1381}
Dees_Troy51a0e822012-09-05 15:24:24 -04001382
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001383int GUIAction::htcdumlockreflashrecovery(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001384{
1385 operation_start("HTC Dumlock Reflash Recovery");
1386 if (simulate) {
1387 simulate_progress_bar();
1388 } else
1389 TWFunc::htc_dumlock_reflash_recovery_to_boot();
Dees_Troy51a0e822012-09-05 15:24:24 -04001390
that3f7b1ac2014-12-30 11:30:13 +01001391 operation_end(0);
1392 return 0;
1393}
Dees_Troy51a0e822012-09-05 15:24:24 -04001394
that3f7b1ac2014-12-30 11:30:13 +01001395int GUIAction::cmd(std::string arg)
1396{
1397 int op_status = 0;
1398
1399 operation_start("Command");
1400 LOGINFO("Running command: '%s'\n", arg.c_str());
1401 if (simulate) {
1402 simulate_progress_bar();
1403 } else {
1404 op_status = TWFunc::Exec_Cmd(arg);
1405 if (op_status != 0)
1406 op_status = 1;
1407 }
1408
1409 operation_end(op_status);
1410 return 0;
1411}
1412
1413int GUIAction::terminalcommand(std::string arg)
1414{
1415 int op_status = 0;
1416 string cmdpath, command;
1417
1418 DataManager::GetValue("tw_terminal_location", cmdpath);
1419 operation_start("CommandOutput");
1420 gui_print("%s # %s\n", cmdpath.c_str(), arg.c_str());
1421 if (simulate) {
1422 simulate_progress_bar();
1423 operation_end(op_status);
Matt Mower5aa29ab2015-02-25 23:50:55 -06001424 } else if (arg == "exit") {
1425 LOGINFO("Exiting terminal\n");
1426 operation_end(op_status);
1427 page("main");
that3f7b1ac2014-12-30 11:30:13 +01001428 } else {
1429 command = "cd \"" + cmdpath + "\" && " + arg + " 2>&1";;
1430 LOGINFO("Actual command is: '%s'\n", command.c_str());
that3f7b1ac2014-12-30 11:30:13 +01001431 DataManager::SetValue("tw_terminal_state", 1);
1432 DataManager::SetValue("tw_background_thread_running", 1);
thatc6085482015-01-09 22:12:43 +01001433 FILE* fp;
1434 char line[512];
1435
1436 fp = popen(command.c_str(), "r");
1437 if (fp == NULL) {
Ethan Yonker74db1572015-10-28 12:44:49 -05001438 LOGERR("Error opening command to run (%s).\n", strerror(errno));
thatc6085482015-01-09 22:12:43 +01001439 } else {
Matt Mower23d8aae2017-01-06 14:30:33 -06001440 int fd = fileno(fp), has_data = 0, check = 0, keep_going = -1;
thatc6085482015-01-09 22:12:43 +01001441 struct timeval timeout;
1442 fd_set fdset;
1443
Matt Mowera8a89d12016-12-30 18:10:37 -06001444 while (keep_going)
thatc6085482015-01-09 22:12:43 +01001445 {
1446 FD_ZERO(&fdset);
1447 FD_SET(fd, &fdset);
1448 timeout.tv_sec = 0;
1449 timeout.tv_usec = 400000;
1450 has_data = select(fd+1, &fdset, NULL, NULL, &timeout);
1451 if (has_data == 0) {
1452 // Timeout reached
1453 DataManager::GetValue("tw_terminal_state", check);
1454 if (check == 0) {
1455 keep_going = 0;
1456 }
1457 } else if (has_data < 0) {
1458 // End of execution
1459 keep_going = 0;
1460 } else {
1461 // Try to read output
Matt Mowera8a89d12016-12-30 18:10:37 -06001462 if (fgets(line, sizeof(line), fp) != NULL)
thatc6085482015-01-09 22:12:43 +01001463 gui_print("%s", line); // Display output
1464 else
1465 keep_going = 0; // Done executing
1466 }
1467 }
1468 fclose(fp);
Dees_Troy51a0e822012-09-05 15:24:24 -04001469 }
thatc6085482015-01-09 22:12:43 +01001470 DataManager::SetValue("tw_operation_status", 0);
1471 DataManager::SetValue("tw_operation_state", 1);
1472 DataManager::SetValue("tw_terminal_state", 0);
1473 DataManager::SetValue("tw_background_thread_running", 0);
1474 DataManager::SetValue(TW_ACTION_BUSY, 0);
that3f7b1ac2014-12-30 11:30:13 +01001475 }
1476 return 0;
1477}
1478
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001479int GUIAction::killterminal(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001480{
that3f7b1ac2014-12-30 11:30:13 +01001481 LOGINFO("Sending kill command...\n");
1482 operation_start("KillCommand");
1483 DataManager::SetValue("tw_operation_status", 0);
1484 DataManager::SetValue("tw_operation_state", 1);
1485 DataManager::SetValue("tw_terminal_state", 0);
1486 DataManager::SetValue("tw_background_thread_running", 0);
1487 DataManager::SetValue(TW_ACTION_BUSY, 0);
1488 return 0;
1489}
1490
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001491int GUIAction::reinjecttwrp(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001492{
1493 int op_status = 0;
1494 operation_start("ReinjectTWRP");
Ethan Yonker74db1572015-10-28 12:44:49 -05001495 gui_msg("injecttwrp=Injecting TWRP into boot image...");
that3f7b1ac2014-12-30 11:30:13 +01001496 if (simulate) {
1497 simulate_progress_bar();
1498 } else {
1499 TWFunc::Exec_Cmd("injecttwrp --dump /tmp/backup_recovery_ramdisk.img /tmp/injected_boot.img --flash");
Ethan Yonker74db1572015-10-28 12:44:49 -05001500 gui_msg("done=Done.");
that3f7b1ac2014-12-30 11:30:13 +01001501 }
1502
1503 operation_end(op_status);
1504 return 0;
1505}
1506
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001507int GUIAction::checkbackupname(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001508{
1509 int op_status = 0;
1510
1511 operation_start("CheckBackupName");
1512 if (simulate) {
1513 simulate_progress_bar();
1514 } else {
Ethan Yonker53796e72019-01-11 22:49:52 -06001515 string Backup_Name;
1516 DataManager::GetValue(TW_BACKUP_NAME, Backup_Name);
1517 op_status = PartitionManager.Check_Backup_Name(Backup_Name, true, true);
that3f7b1ac2014-12-30 11:30:13 +01001518 if (op_status != 0)
1519 op_status = 1;
1520 }
1521
1522 operation_end(op_status);
1523 return 0;
1524}
1525
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001526int GUIAction::decrypt(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001527{
1528 int op_status = 0;
1529
1530 operation_start("Decrypt");
1531 if (simulate) {
1532 simulate_progress_bar();
1533 } else {
1534 string Password;
Noah Jacobson81d638d2019-04-28 00:10:07 -04001535 string userID;
that3f7b1ac2014-12-30 11:30:13 +01001536 DataManager::GetValue("tw_crypto_password", Password);
Noah Jacobson81d638d2019-04-28 00:10:07 -04001537
1538 if (DataManager::GetIntValue(TW_IS_FBE)) { // for FBE
1539 DataManager::GetValue("tw_crypto_user_id", userID);
1540 if (userID != "") {
1541 op_status = PartitionManager.Decrypt_Device(Password, atoi(userID.c_str()));
1542 if (userID != "0") {
1543 if (op_status != 0)
1544 op_status = 1;
1545 operation_end(op_status);
1546 return 0;
1547 }
1548 } else {
1549 LOGINFO("User ID not found\n");
1550 op_status = 1;
1551 }
1552 ::sleep(1);
1553 } else { // for FDE
1554 op_status = PartitionManager.Decrypt_Device(Password);
1555 }
1556
that3f7b1ac2014-12-30 11:30:13 +01001557 if (op_status != 0)
1558 op_status = 1;
1559 else {
that3f7b1ac2014-12-30 11:30:13 +01001560 DataManager::SetValue(TW_IS_ENCRYPTED, 0);
1561
Ethan Yonkercf50da52015-01-12 21:59:07 -06001562 int has_datamedia;
that3f7b1ac2014-12-30 11:30:13 +01001563
Ethan Yonkercf50da52015-01-12 21:59:07 -06001564 // Check for a custom theme and load it if exists
1565 DataManager::GetValue(TW_HAS_DATA_MEDIA, has_datamedia);
1566 if (has_datamedia != 0) {
1567 if (tw_get_default_metadata(DataManager::GetSettingsStoragePath().c_str()) != 0) {
Ethan Yonkerd9ff3c52015-01-21 21:51:20 -06001568 LOGINFO("Failed to get default contexts and file mode for storage files.\n");
Ethan Yonkercf50da52015-01-12 21:59:07 -06001569 } else {
1570 LOGINFO("Got default contexts and file mode for storage files.\n");
that3f7b1ac2014-12-30 11:30:13 +01001571 }
1572 }
Ethan Yonker66a19492015-12-10 10:19:45 -06001573 PartitionManager.Decrypt_Adopted();
that3f7b1ac2014-12-30 11:30:13 +01001574 }
1575 }
1576
1577 operation_end(op_status);
1578 return 0;
1579}
1580
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001581int GUIAction::adbsideload(std::string arg __unused)
thatcc8ddca2015-01-03 01:59:36 +01001582{
1583 operation_start("Sideload");
1584 if (simulate) {
1585 simulate_progress_bar();
1586 operation_end(0);
1587 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001588 gui_msg("start_sideload=Starting ADB sideload feature...");
thatc6085482015-01-09 22:12:43 +01001589 bool mtp_was_enabled = TWFunc::Toggle_MTP(false);
1590
1591 // wait for the adb connection
bigbiffd58ba182020-03-23 10:02:29 -04001592 // int ret = apply_from_adb("/", &sideload_child_pid);
1593 Device::BuiltinAction reboot_action = Device::REBOOT_BOOTLOADER;
1594 int ret = ApplyFromAdb("/", &reboot_action);
thatc6085482015-01-09 22:12:43 +01001595 DataManager::SetValue("tw_has_cancel", 0); // Remove cancel button from gui now that the zip install is going to start
1596
1597 if (ret != 0) {
1598 if (ret == -2)
Ethan Yonker74db1572015-10-28 12:44:49 -05001599 gui_msg("need_new_adb=You need adb 1.0.32 or newer to sideload to this device.");
thatc6085482015-01-09 22:12:43 +01001600 ret = 1; // failure
1601 } else {
1602 int wipe_cache = 0;
1603 int wipe_dalvik = 0;
1604 DataManager::GetValue("tw_wipe_dalvik", wipe_dalvik);
1605
1606 if (TWinstall_zip(FUSE_SIDELOAD_HOST_PATHNAME, &wipe_cache) == 0) {
1607 if (wipe_cache || DataManager::GetIntValue("tw_wipe_cache"))
1608 PartitionManager.Wipe_By_Path("/cache");
1609 if (wipe_dalvik)
1610 PartitionManager.Wipe_Dalvik_Cache();
1611 } else {
1612 ret = 1; // failure
1613 }
thatcc8ddca2015-01-03 01:59:36 +01001614 }
thatc6085482015-01-09 22:12:43 +01001615 if (sideload_child_pid) {
1616 LOGINFO("Signaling child sideload process to exit.\n");
1617 struct stat st;
1618 // Calling stat() on this magic filename signals the minadbd
1619 // subprocess to shut down.
1620 stat(FUSE_SIDELOAD_HOST_EXIT_PATHNAME, &st);
1621 int status;
1622 LOGINFO("Waiting for child sideload process to exit.\n");
1623 waitpid(sideload_child_pid, &status, 0);
1624 }
Dees Troy28a85d22015-04-28 02:03:16 +00001625 property_set("ctl.start", "adbd");
thatc6085482015-01-09 22:12:43 +01001626 TWFunc::Toggle_MTP(mtp_was_enabled);
1627 reinject_after_flash();
1628 operation_end(ret);
thatcc8ddca2015-01-03 01:59:36 +01001629 }
that3f7b1ac2014-12-30 11:30:13 +01001630 return 0;
1631}
1632
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001633int GUIAction::adbsideloadcancel(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001634{
that3f7b1ac2014-12-30 11:30:13 +01001635 struct stat st;
1636 DataManager::SetValue("tw_has_cancel", 0); // Remove cancel button from gui
Ethan Yonker74db1572015-10-28 12:44:49 -05001637 gui_msg("cancel_sideload=Cancelling ADB sideload...");
thatcc8ddca2015-01-03 01:59:36 +01001638 LOGINFO("Signaling child sideload process to exit.\n");
1639 // Calling stat() on this magic filename signals the minadbd
1640 // subprocess to shut down.
1641 stat(FUSE_SIDELOAD_HOST_EXIT_PATHNAME, &st);
1642 if (!sideload_child_pid) {
1643 LOGERR("Unable to get child ID\n");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001644 return 0;
1645 }
thatcc8ddca2015-01-03 01:59:36 +01001646 ::sleep(1);
1647 LOGINFO("Killing child sideload process.\n");
1648 kill(sideload_child_pid, SIGTERM);
1649 int status;
1650 LOGINFO("Waiting for child sideload process to exit.\n");
1651 waitpid(sideload_child_pid, &status, 0);
1652 sideload_child_pid = 0;
that3f7b1ac2014-12-30 11:30:13 +01001653 DataManager::SetValue("tw_page_done", "1"); // For OpenRecoveryScript support
1654 return 0;
1655}
1656
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001657int GUIAction::openrecoveryscript(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001658{
1659 operation_start("OpenRecoveryScript");
1660 if (simulate) {
1661 simulate_progress_bar();
Ethan Yonker24e7eb72015-01-03 16:13:06 -06001662 operation_end(0);
that3f7b1ac2014-12-30 11:30:13 +01001663 } else {
that10ae24f2015-12-26 20:53:51 +01001664 int op_status = OpenRecoveryScript::Run_OpenRecoveryScript_Action();
Ethan Yonkere1abe612015-06-09 11:09:32 -05001665 operation_end(op_status);
that3f7b1ac2014-12-30 11:30:13 +01001666 }
1667 return 0;
1668}
1669
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001670int GUIAction::installsu(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001671{
1672 int op_status = 0;
1673
1674 operation_start("Install SuperSU");
1675 if (simulate) {
1676 simulate_progress_bar();
1677 } else {
Ethan Yonkerfa67cbf2018-07-20 12:22:33 -05001678 LOGERR("Installing SuperSU was deprecated from TWRP.\n");
that3f7b1ac2014-12-30 11:30:13 +01001679 }
1680
1681 operation_end(op_status);
1682 return 0;
1683}
1684
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001685int GUIAction::fixsu(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001686{
1687 int op_status = 0;
1688
1689 operation_start("Fixing Superuser Permissions");
1690 if (simulate) {
1691 simulate_progress_bar();
1692 } else {
1693 LOGERR("Fixing su permissions was deprecated from TWRP.\n");
1694 LOGERR("4.3+ ROMs with SELinux will always lose su perms.\n");
1695 }
1696
1697 operation_end(op_status);
1698 return 0;
1699}
1700
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001701int GUIAction::decrypt_backup(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001702{
1703 int op_status = 0;
1704
1705 operation_start("Try Restore Decrypt");
1706 if (simulate) {
1707 simulate_progress_bar();
1708 } else {
1709 string Restore_Path, Filename, Password;
1710 DataManager::GetValue("tw_restore", Restore_Path);
1711 Restore_Path += "/";
1712 DataManager::GetValue("tw_restore_password", Password);
1713 TWFunc::SetPerformanceMode(true);
1714 if (TWFunc::Try_Decrypting_Backup(Restore_Path, Password))
1715 op_status = 0; // success
1716 else
1717 op_status = 1; // fail
1718 TWFunc::SetPerformanceMode(false);
1719 }
1720
1721 operation_end(op_status);
1722 return 0;
1723}
1724
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001725int GUIAction::repair(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001726{
1727 int op_status = 0;
1728
1729 operation_start("Repair Partition");
1730 if (simulate) {
1731 simulate_progress_bar();
1732 } else {
1733 string part_path;
1734 DataManager::GetValue("tw_partition_mount_point", part_path);
1735 if (PartitionManager.Repair_By_Path(part_path, true)) {
1736 op_status = 0; // success
1737 } else {
that3f7b1ac2014-12-30 11:30:13 +01001738 op_status = 1; // fail
1739 }
1740 }
1741
1742 operation_end(op_status);
1743 return 0;
1744}
1745
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001746int GUIAction::resize(std::string arg __unused)
Ethan Yonkera2719152015-05-28 09:44:41 -05001747{
1748 int op_status = 0;
1749
1750 operation_start("Resize Partition");
1751 if (simulate) {
1752 simulate_progress_bar();
1753 } else {
1754 string part_path;
1755 DataManager::GetValue("tw_partition_mount_point", part_path);
1756 if (PartitionManager.Resize_By_Path(part_path, true)) {
1757 op_status = 0; // success
1758 } else {
Ethan Yonkera2719152015-05-28 09:44:41 -05001759 op_status = 1; // fail
1760 }
1761 }
1762
1763 operation_end(op_status);
1764 return 0;
1765}
1766
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001767int GUIAction::changefilesystem(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001768{
1769 int op_status = 0;
1770
1771 operation_start("Change File System");
1772 if (simulate) {
1773 simulate_progress_bar();
1774 } else {
1775 string part_path, file_system;
1776 DataManager::GetValue("tw_partition_mount_point", part_path);
1777 DataManager::GetValue("tw_action_new_file_system", file_system);
1778 if (PartitionManager.Wipe_By_Path(part_path, file_system)) {
1779 op_status = 0; // success
1780 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -05001781 gui_err("change_fs_err=Error changing file system.");
that3f7b1ac2014-12-30 11:30:13 +01001782 op_status = 1; // fail
1783 }
1784 }
1785 PartitionManager.Update_System_Details();
1786 operation_end(op_status);
1787 return 0;
1788}
1789
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001790int GUIAction::startmtp(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001791{
1792 int op_status = 0;
1793
1794 operation_start("Start MTP");
1795 if (PartitionManager.Enable_MTP())
1796 op_status = 0; // success
1797 else
1798 op_status = 1; // fail
1799
1800 operation_end(op_status);
1801 return 0;
1802}
1803
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001804int GUIAction::stopmtp(std::string arg __unused)
that3f7b1ac2014-12-30 11:30:13 +01001805{
1806 int op_status = 0;
1807
1808 operation_start("Stop MTP");
1809 if (PartitionManager.Disable_MTP())
1810 op_status = 0; // success
1811 else
1812 op_status = 1; // fail
1813
1814 operation_end(op_status);
1815 return 0;
Dees_Troy51a0e822012-09-05 15:24:24 -04001816}
1817
Ethan Yonkerd0514ba2015-10-22 14:17:47 -05001818int GUIAction::flashimage(std::string arg __unused)
Ethan Yonker96af84a2015-01-05 14:58:36 -06001819{
1820 int op_status = 0;
1821
1822 operation_start("Flash Image");
Ethan Yonkerdcf2b672016-09-13 14:41:53 -05001823 string path, filename;
1824 DataManager::GetValue("tw_zip_location", path);
1825 DataManager::GetValue("tw_file", filename);
Ethan Yonkere080c1f2016-09-19 13:50:25 -05001826 if (PartitionManager.Flash_Image(path, filename))
Ethan Yonker96af84a2015-01-05 14:58:36 -06001827 op_status = 0; // success
1828 else
1829 op_status = 1; // fail
1830
1831 operation_end(op_status);
1832 return 0;
1833}
1834
that10ae24f2015-12-26 20:53:51 +01001835int GUIAction::twcmd(std::string arg)
1836{
1837 operation_start("TWRP CLI Command");
1838 if (simulate)
1839 simulate_progress_bar();
1840 else
1841 OpenRecoveryScript::Run_CLI_Command(arg.c_str());
1842 operation_end(0);
1843 return 0;
1844}
1845
Dees_Troy51a0e822012-09-05 15:24:24 -04001846int GUIAction::getKeyByName(std::string key)
1847{
that8834a0f2016-01-05 23:29:30 +01001848 if (key == "home") return KEY_HOMEPAGE; // note: KEY_HOME is cursor movement (like KEY_END)
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001849 else if (key == "menu") return KEY_MENU;
1850 else if (key == "back") return KEY_BACK;
1851 else if (key == "search") return KEY_SEARCH;
1852 else if (key == "voldown") return KEY_VOLUMEDOWN;
1853 else if (key == "volup") return KEY_VOLUMEUP;
1854 else if (key == "power") {
Dees_Troy51a0e822012-09-05 15:24:24 -04001855 int ret_val;
1856 DataManager::GetValue(TW_POWER_BUTTON, ret_val);
1857 if (!ret_val)
1858 return KEY_POWER;
1859 else
1860 return ret_val;
1861 }
1862
Vojtech Bocekfafb0c52013-07-25 22:53:02 +02001863 return atol(key.c_str());
Dees_Troy51a0e822012-09-05 15:24:24 -04001864}
Ethan Yonkereb32b1f2015-05-18 10:23:03 -05001865
1866int GUIAction::checkpartitionlifetimewrites(std::string arg)
1867{
1868 int op_status = 0;
1869 TWPartition* sys = PartitionManager.Find_Partition_By_Path(arg);
1870
1871 operation_start("Check Partition Lifetime Writes");
1872 if (sys) {
1873 if (sys->Check_Lifetime_Writes() != 0)
1874 DataManager::SetValue("tw_lifetime_writes", 1);
1875 else
1876 DataManager::SetValue("tw_lifetime_writes", 0);
1877 op_status = 0; // success
1878 } else {
1879 DataManager::SetValue("tw_lifetime_writes", 1);
1880 op_status = 1; // fail
1881 }
1882
1883 operation_end(op_status);
1884 return 0;
1885}
1886
1887int GUIAction::mountsystemtoggle(std::string arg)
1888{
1889 int op_status = 0;
Captain Throwback9d6feb52018-07-27 10:05:24 -04001890 bool remount_system = PartitionManager.Is_Mounted_By_Path(PartitionManager.Get_Android_Root_Path());
Ethan Yonker1673e3d2015-10-26 21:51:58 -05001891 bool remount_vendor = PartitionManager.Is_Mounted_By_Path("/vendor");
Ethan Yonkereb32b1f2015-05-18 10:23:03 -05001892
1893 operation_start("Toggle System Mount");
Captain Throwback9d6feb52018-07-27 10:05:24 -04001894 if (!PartitionManager.UnMount_By_Path(PartitionManager.Get_Android_Root_Path(), true)) {
Ethan Yonkereb32b1f2015-05-18 10:23:03 -05001895 op_status = 1; // fail
1896 } else {
Captain Throwback9d6feb52018-07-27 10:05:24 -04001897 TWPartition* Part = PartitionManager.Find_Partition_By_Path(PartitionManager.Get_Android_Root_Path());
Ethan Yonkereb32b1f2015-05-18 10:23:03 -05001898 if (Part) {
Ethan Yonkerd6966f42015-05-30 14:52:16 -05001899 if (arg == "0") {
Ethan Yonkereb32b1f2015-05-18 10:23:03 -05001900 DataManager::SetValue("tw_mount_system_ro", 0);
1901 Part->Change_Mount_Read_Only(false);
1902 } else {
1903 DataManager::SetValue("tw_mount_system_ro", 1);
1904 Part->Change_Mount_Read_Only(true);
1905 }
1906 if (remount_system) {
1907 Part->Mount(true);
1908 }
1909 op_status = 0; // success
1910 } else {
1911 op_status = 1; // fail
1912 }
Ethan Yonker1673e3d2015-10-26 21:51:58 -05001913 Part = PartitionManager.Find_Partition_By_Path("/vendor");
1914 if (Part) {
1915 if (arg == "0") {
1916 Part->Change_Mount_Read_Only(false);
1917 } else {
1918 Part->Change_Mount_Read_Only(true);
1919 }
1920 if (remount_vendor) {
1921 Part->Mount(true);
1922 }
1923 op_status = 0; // success
1924 } else {
1925 op_status = 1; // fail
1926 }
Ethan Yonkereb32b1f2015-05-18 10:23:03 -05001927 }
1928
1929 operation_end(op_status);
1930 return 0;
1931}
Ethan Yonker74db1572015-10-28 12:44:49 -05001932
1933int GUIAction::setlanguage(std::string arg __unused)
1934{
1935 int op_status = 0;
1936
1937 operation_start("Set Language");
1938 PageManager::LoadLanguage(DataManager::GetStrValue("tw_language"));
1939 PageManager::RequestReload();
1940 op_status = 0; // success
1941
1942 operation_end(op_status);
1943 return 0;
1944}
Ethan Yonker1b190162016-12-05 15:25:19 -06001945
Matt Mower9472ba12016-01-20 18:12:47 -06001946int GUIAction::togglebacklight(std::string arg __unused)
1947{
1948 blankTimer.toggleBlank();
1949 return 0;
1950}
1951
Ethan Yonker1b190162016-12-05 15:25:19 -06001952int GUIAction::setbootslot(std::string arg)
1953{
1954 operation_start("Set Boot Slot");
Captain Throwbackb60a2732020-10-13 17:55:43 -04001955 if (!simulate) {
1956 if (!PartitionManager.UnMount_By_Path("/vendor", false)) {
1957 // PartitionManager failed to unmount /vendor, this should not happen,
1958 // but in case it does, do a lazy unmount
1959 LOGINFO("WARNING: vendor partition could not be unmounted normally!\n");
1960 umount2("/vendor", MNT_DETACH);
1961 PartitionManager.Set_Active_Slot(arg);
1962 } else {
1963 PartitionManager.Set_Active_Slot(arg);
1964 }
1965 } else {
Ethan Yonker1b190162016-12-05 15:25:19 -06001966 simulate_progress_bar();
Captain Throwbackb60a2732020-10-13 17:55:43 -04001967 }
Ethan Yonker1b190162016-12-05 15:25:19 -06001968 operation_end(0);
1969 return 0;
1970}
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06001971
1972int GUIAction::checkforapp(std::string arg __unused)
1973{
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06001974 operation_start("Check for TWRP App");
1975 if (!simulate)
1976 {
1977 string sdkverstr = TWFunc::System_Property_Get("ro.build.version.sdk");
1978 int sdkver = 0;
1979 if (!sdkverstr.empty()) {
1980 sdkver = atoi(sdkverstr.c_str());
1981 }
1982 if (sdkver <= 13) {
1983 if (sdkver == 0)
1984 LOGINFO("Unable to read sdk version from build prop\n");
1985 else
1986 LOGINFO("SDK version too low for TWRP app (%i < 14)\n", sdkver);
Ethan Yonker64c5c0a2017-03-06 12:42:54 -06001987 DataManager::SetValue("tw_app_install_status", 1); // 0 = no status, 1 = not installed, 2 = already installed or do not install
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06001988 goto exit;
1989 }
Ethan Yonker76bbd3a2019-05-10 10:50:04 -05001990 if (TWFunc::Is_TWRP_App_In_System()) {
1991 DataManager::SetValue("tw_app_install_status", 2); // 0 = no status, 1 = not installed, 2 = already installed or do not install
1992 goto exit;
Ethan Yonker5e1a7f92017-01-18 16:44:54 -06001993 }
1994 if (PartitionManager.Mount_By_Path("/data", false)) {
Ethan Yonker5128d292017-02-04 19:52:56 -06001995 const char parent_path[] = "/data/app";
1996 const char app_prefix[] = "me.twrp.twrpapp-";
1997 DIR *d = opendir(parent_path);
1998 if (d) {
1999 struct dirent *p;
2000 while ((p = readdir(d))) {
2001 if (p->d_type != DT_DIR || strlen(p->d_name) < strlen(app_prefix) || strncmp(p->d_name, app_prefix, strlen(app_prefix)))
2002 continue;
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002003 closedir(d);
Ethan Yonker5128d292017-02-04 19:52:56 -06002004 LOGINFO("App found at '%s/%s'\n", parent_path, p->d_name);
Ethan Yonker64c5c0a2017-03-06 12:42:54 -06002005 DataManager::SetValue("tw_app_install_status", 2); // 0 = no status, 1 = not installed, 2 = already installed or do not install
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002006 goto exit;
2007 }
Ethan Yonker5128d292017-02-04 19:52:56 -06002008 closedir(d);
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002009 }
Ethan Yonker64c5c0a2017-03-06 12:42:54 -06002010 } else {
2011 LOGINFO("Data partition cannot be mounted during app check\n");
2012 DataManager::SetValue("tw_app_install_status", 2); // 0 = no status, 1 = not installed, 2 = already installed or do not install
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002013 }
2014 } else
2015 simulate_progress_bar();
2016 LOGINFO("App not installed\n");
2017 DataManager::SetValue("tw_app_install_status", 1); // 0 = no status, 1 = not installed, 2 = already installed
2018exit:
2019 operation_end(0);
2020 return 0;
2021}
2022
2023int GUIAction::installapp(std::string arg __unused)
2024{
2025 int op_status = 1;
2026 operation_start("Install TWRP App");
2027 if (!simulate)
2028 {
2029 if (DataManager::GetIntValue("tw_mount_system_ro") > 0 || DataManager::GetIntValue("tw_app_install_system") == 0) {
2030 if (PartitionManager.Mount_By_Path("/data", true)) {
2031 string install_path = "/data/app";
2032 string context = "u:object_r:apk_data_file:s0";
2033 if (!TWFunc::Path_Exists(install_path)) {
2034 if (mkdir(install_path.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH)) {
2035 LOGERR("Error making %s directory: %s\n", install_path.c_str(), strerror(errno));
2036 goto exit;
2037 }
2038 if (chown(install_path.c_str(), 1000, 1000)) {
2039 LOGERR("chown %s error: %s\n", install_path.c_str(), strerror(errno));
2040 goto exit;
2041 }
Ethan Yonker4767caf2017-01-11 10:45:04 -06002042 if (setfilecon(install_path.c_str(), (security_context_t)context.c_str()) < 0) {
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002043 LOGERR("setfilecon %s error: %s\n", install_path.c_str(), strerror(errno));
2044 goto exit;
2045 }
2046 }
2047 install_path += "/me.twrp.twrpapp-1";
2048 if (mkdir(install_path.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH)) {
2049 LOGERR("Error making %s directory: %s\n", install_path.c_str(), strerror(errno));
2050 goto exit;
2051 }
2052 if (chown(install_path.c_str(), 1000, 1000)) {
2053 LOGERR("chown %s error: %s\n", install_path.c_str(), strerror(errno));
2054 goto exit;
2055 }
Ethan Yonker4767caf2017-01-11 10:45:04 -06002056 if (setfilecon(install_path.c_str(), (security_context_t)context.c_str()) < 0) {
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002057 LOGERR("setfilecon %s error: %s\n", install_path.c_str(), strerror(errno));
2058 goto exit;
2059 }
2060 install_path += "/base.apk";
bigbiffad58e1b2020-07-06 20:24:34 -04002061 if (TWFunc::copy_file("/system/bin/me.twrp.twrpapp.apk", install_path, 0644)) {
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002062 LOGERR("Error copying apk file\n");
2063 goto exit;
2064 }
2065 if (chown(install_path.c_str(), 1000, 1000)) {
2066 LOGERR("chown %s error: %s\n", install_path.c_str(), strerror(errno));
2067 goto exit;
2068 }
Ethan Yonker4767caf2017-01-11 10:45:04 -06002069 if (setfilecon(install_path.c_str(), (security_context_t)context.c_str()) < 0) {
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002070 LOGERR("setfilecon %s error: %s\n", install_path.c_str(), strerror(errno));
2071 goto exit;
2072 }
2073 sync();
2074 sync();
2075 }
2076 } else {
Captain Throwback9d6feb52018-07-27 10:05:24 -04002077 if (PartitionManager.Mount_By_Path(PartitionManager.Get_Android_Root_Path(), true)) {
2078 string base_path = PartitionManager.Get_Android_Root_Path();
2079 if (TWFunc::Path_Exists(PartitionManager.Get_Android_Root_Path() + "/system"))
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002080 base_path += "/system"; // For devices with system as a root file system (e.g. Pixel)
2081 string install_path = base_path + "/priv-app";
2082 string context = "u:object_r:system_file:s0";
2083 if (!TWFunc::Path_Exists(install_path))
2084 install_path = base_path + "/app";
2085 if (TWFunc::Path_Exists(install_path)) {
2086 install_path += "/twrpapp";
2087 LOGINFO("Installing app to '%s'\n", install_path.c_str());
2088 if (mkdir(install_path.c_str(), S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH) == 0) {
Ethan Yonker4767caf2017-01-11 10:45:04 -06002089 if (setfilecon(install_path.c_str(), (security_context_t)context.c_str()) < 0) {
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002090 LOGERR("setfilecon %s error: %s\n", install_path.c_str(), strerror(errno));
2091 goto exit;
2092 }
2093 install_path += "/me.twrp.twrpapp.apk";
bigbiffad58e1b2020-07-06 20:24:34 -04002094 if (TWFunc::copy_file("/system/bin/me.twrp.twrpapp.apk", install_path, 0644)) {
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002095 LOGERR("Error copying apk file\n");
2096 goto exit;
2097 }
Ethan Yonker4767caf2017-01-11 10:45:04 -06002098 if (setfilecon(install_path.c_str(), (security_context_t)context.c_str()) < 0) {
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002099 LOGERR("setfilecon %s error: %s\n", install_path.c_str(), strerror(errno));
2100 goto exit;
2101 }
Stefan Tauner6ad14572020-01-11 22:28:53 +01002102
2103 // System apps require their permissions to be pre-set via an XML file in /etc/permissions
2104 string permission_path = base_path + "/etc/permissions/privapp-permissions-twrpapp.xml";
bigbiffad58e1b2020-07-06 20:24:34 -04002105 if (TWFunc::copy_file("/system/bin/privapp-permissions-twrpapp.xml", permission_path, 0644)) {
Stefan Tauner6ad14572020-01-11 22:28:53 +01002106 LOGERR("Error copying permission file\n");
2107 goto exit;
2108 }
2109 if (chown(permission_path.c_str(), 1000, 1000)) {
2110 LOGERR("chown %s error: %s\n", permission_path.c_str(), strerror(errno));
2111 goto exit;
2112 }
2113 if (setfilecon(permission_path.c_str(), (security_context_t)context.c_str()) < 0) {
2114 LOGERR("setfilecon %s error: %s\n", permission_path.c_str(), strerror(errno));
2115 goto exit;
2116 }
2117
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002118 sync();
2119 sync();
Captain Throwback9d6feb52018-07-27 10:05:24 -04002120 PartitionManager.UnMount_By_Path(PartitionManager.Get_Android_Root_Path(), true);
Ethan Yonkerb4bff5e2016-12-16 07:47:58 -06002121 op_status = 0;
2122 } else {
2123 LOGERR("Error making app directory '%s': %s\n", strerror(errno));
2124 }
2125 }
2126 }
2127 }
2128 } else
2129 simulate_progress_bar();
2130exit:
2131 operation_end(0);
2132 return 0;
2133}
Ethan Yonker53796e72019-01-11 22:49:52 -06002134
Ethan Yonker76bbd3a2019-05-10 10:50:04 -05002135int GUIAction::uninstalltwrpsystemapp(std::string arg __unused)
2136{
2137 int op_status = 1;
2138 operation_start("Uninstall TWRP System App");
2139 if (!simulate)
2140 {
2141 int Mount_System_RO = DataManager::GetIntValue("tw_mount_system_ro");
2142 TWPartition* Part = PartitionManager.Find_Partition_By_Path(PartitionManager.Get_Android_Root_Path());
2143 if (!Part) {
2144 LOGERR("Unabled to find system partition.\n");
2145 goto exit;
2146 }
2147 if (!Part->UnMount(true)) {
2148 goto exit;
2149 }
2150 if (Mount_System_RO > 0) {
2151 DataManager::SetValue("tw_mount_system_ro", 0);
2152 Part->Change_Mount_Read_Only(false);
2153 }
2154 if (Part->Mount(true)) {
2155 string base_path = PartitionManager.Get_Android_Root_Path();
2156 if (TWFunc::Path_Exists(PartitionManager.Get_Android_Root_Path() + "/system"))
2157 base_path += "/system"; // For devices with system as a root file system (e.g. Pixel)
2158 string uninstall_path = base_path + "/priv-app";
2159 if (!TWFunc::Path_Exists(uninstall_path))
2160 uninstall_path = base_path + "/app";
2161 uninstall_path += "/twrpapp";
2162 if (TWFunc::Path_Exists(uninstall_path)) {
2163 LOGINFO("Uninstalling TWRP App from '%s'\n", uninstall_path.c_str());
2164 if (TWFunc::removeDir(uninstall_path, false) == 0) {
2165 sync();
2166 op_status = 0;
2167 DataManager::SetValue("tw_app_installed_in_system", 0);
2168 DataManager::SetValue("tw_app_install_status", 0);
2169 } else {
2170 LOGERR("Unable to remove TWRP app from system.\n");
2171 }
2172 } else {
2173 LOGINFO("didn't find TWRP app in '%s'\n", uninstall_path.c_str());
2174 }
2175 }
2176 Part->UnMount(true);
2177 if (Mount_System_RO > 0) {
2178 DataManager::SetValue("tw_mount_system_ro", Mount_System_RO);
2179 Part->Change_Mount_Read_Only(true);
2180 }
2181 } else
2182 simulate_progress_bar();
2183exit:
2184 operation_end(0);
2185 return 0;
2186}
2187
Ethan Yonker53796e72019-01-11 22:49:52 -06002188int GUIAction::repackimage(std::string arg __unused)
2189{
2190 int op_status = 1;
bigbiffad58e1b2020-07-06 20:24:34 -04002191 twrpRepacker repacker;
2192
Ethan Yonker53796e72019-01-11 22:49:52 -06002193 operation_start("Repack Image");
2194 if (!simulate)
2195 {
2196 std::string path = DataManager::GetStrValue("tw_filename");
2197 Repack_Options_struct Repack_Options;
2198 Repack_Options.Disable_Verity = false;
2199 Repack_Options.Disable_Force_Encrypt = false;
2200 Repack_Options.Backup_First = DataManager::GetIntValue("tw_repack_backup_first") != 0;
2201 if (DataManager::GetIntValue("tw_repack_kernel") == 1)
2202 Repack_Options.Type = REPLACE_KERNEL;
2203 else
2204 Repack_Options.Type = REPLACE_RAMDISK;
bigbiffad58e1b2020-07-06 20:24:34 -04002205 if (!repacker.Repack_Image_And_Flash(path, Repack_Options))
Ethan Yonker53796e72019-01-11 22:49:52 -06002206 goto exit;
2207 } else
2208 simulate_progress_bar();
2209 op_status = 0;
2210exit:
2211 operation_end(op_status);
2212 return 0;
2213}
2214
2215int GUIAction::fixabrecoverybootloop(std::string arg __unused)
2216{
2217 int op_status = 1;
bigbiffad58e1b2020-07-06 20:24:34 -04002218 twrpRepacker repacker;
2219
Ethan Yonker53796e72019-01-11 22:49:52 -06002220 operation_start("Repack Image");
2221 if (!simulate)
2222 {
bigbiffad58e1b2020-07-06 20:24:34 -04002223 if (!TWFunc::Path_Exists("/system/bin/magiskboot")) {
Ethan Yonker53796e72019-01-11 22:49:52 -06002224 LOGERR("Image repacking tool not present in this TWRP build!");
2225 goto exit;
2226 }
2227 DataManager::SetProgress(0);
2228 TWPartition* part = PartitionManager.Find_Partition_By_Path("/boot");
2229 if (part)
2230 gui_msg(Msg("unpacking_image=Unpacking {1}...")(part->Display_Name));
2231 else {
2232 gui_msg(Msg(msg::kError, "unable_to_locate=Unable to locate {1}.")("/boot"));
2233 goto exit;
2234 }
bigbiffad58e1b2020-07-06 20:24:34 -04002235 if (!repacker.Backup_Image_For_Repack(part, REPACK_ORIG_DIR, DataManager::GetIntValue("tw_repack_backup_first") != 0, gui_lookup("repack", "Repack")))
Ethan Yonker53796e72019-01-11 22:49:52 -06002236 goto exit;
2237 DataManager::SetProgress(.25);
2238 gui_msg("fixing_recovery_loop_patch=Patching kernel...");
bigbiffad58e1b2020-07-06 20:24:34 -04002239 std::string command = "cd " REPACK_ORIG_DIR " && /system/bin/magiskboot hexpatch kernel 77616E745F696E697472616D667300 736B69705F696E697472616D667300";
Ethan Yonker53796e72019-01-11 22:49:52 -06002240 if (TWFunc::Exec_Cmd(command) != 0) {
2241 gui_msg(Msg(msg::kError, "fix_recovery_loop_patch_error=Error patching kernel."));
2242 goto exit;
2243 }
2244 std::string header_path = REPACK_ORIG_DIR;
2245 header_path += "header";
2246 if (TWFunc::Path_Exists(header_path)) {
2247 command = "cd " REPACK_ORIG_DIR " && sed -i \"s|$(grep '^cmdline=' header | cut -d= -f2-)|$(grep '^cmdline=' header | cut -d= -f2- | sed -e 's/skip_override//' -e 's/ */ /g' -e 's/[ \t]*$//')|\" header";
2248 if (TWFunc::Exec_Cmd(command) != 0) {
2249 gui_msg(Msg(msg::kError, "fix_recovery_loop_patch_error=Error patching kernel."));
2250 goto exit;
2251 }
2252 }
2253 DataManager::SetProgress(.5);
2254 gui_msg(Msg("repacking_image=Repacking {1}...")(part->Display_Name));
bigbiffad58e1b2020-07-06 20:24:34 -04002255 command = "cd " REPACK_ORIG_DIR " && /system/bin/magiskboot repack " REPACK_ORIG_DIR "boot.img";
Ethan Yonker53796e72019-01-11 22:49:52 -06002256 if (TWFunc::Exec_Cmd(command) != 0) {
2257 gui_msg(Msg(msg::kError, "repack_error=Error repacking image."));
2258 goto exit;
2259 }
2260 DataManager::SetProgress(.75);
2261 std::string path = REPACK_ORIG_DIR;
2262 std::string file = "new-boot.img";
2263 DataManager::SetValue("tw_flash_partition", "/boot;");
2264 if (!PartitionManager.Flash_Image(path, file)) {
2265 LOGINFO("Error flashing new image\n");
2266 goto exit;
2267 }
2268 DataManager::SetProgress(1);
2269 TWFunc::removeDir(REPACK_ORIG_DIR, false);
2270 } else
2271 simulate_progress_bar();
2272 op_status = 0;
2273exit:
2274 operation_end(op_status);
2275 return 0;
2276}
bigbiffdf8436b2020-08-30 16:22:34 -04002277
2278
2279int GUIAction::enableadb(std::string arg __unused) {
2280 android::base::SetProperty("sys.usb.config", "none");
2281 android::base::SetProperty("sys.usb.config", "adb");
2282 return 0;
2283}
2284
2285int GUIAction::enablefastboot(std::string arg __unused) {
2286 android::base::SetProperty("sys.usb.config", "none");
2287 android::base::SetProperty("sys.usb.config", "fastboot");
2288 return 0;
2289}