blob: ab715291aa04bca1582d3df3063590788c3533d7 [file] [log] [blame]
Dees Troy3be70a82013-10-22 14:25:12 +00001/*
bigbiff bigbiff19fb79c2016-09-05 21:04:51 -04002 Copyright 2012 to 2017 bigbiff/Dees_Troy TeamWin
Dees Troy3be70a82013-10-22 14:25:12 +00003 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_Troy32c8eb82012-09-11 15:28:06 -040018
nkk71b4c35912017-10-11 23:39:10 +030019
20#ifndef _GNU_SOURCE
21#define _GNU_SOURCE
22#endif
23
Dees_Troy32c8eb82012-09-11 15:28:06 -040024#include <ctype.h>
25#include <errno.h>
26#include <fcntl.h>
27#include <limits.h>
nkk71b4c35912017-10-11 23:39:10 +030028#include <sys/mman.h>
Dees_Troy32c8eb82012-09-11 15:28:06 -040029#include <sys/stat.h>
30#include <sys/wait.h>
31#include <unistd.h>
32
33#include <string.h>
34#include <stdio.h>
35
Dees_Troy2673cec2013-04-02 20:22:16 +000036#include "twcommon.h"
Ethan Yonker75bf0412014-11-21 13:54:27 -060037#include "mtdutils/mounts.h"
38#include "mtdutils/mtdutils.h"
Ethan Yonker8373cfe2017-09-08 06:50:54 -050039
40#ifdef USE_MINZIP
Dees_Troy32c8eb82012-09-11 15:28:06 -040041#include "minzip/SysUtil.h"
Ethan Yonker8373cfe2017-09-08 06:50:54 -050042#else
43#include "otautil/SysUtil.h"
44#include <ziparchive/zip_archive.h>
45#endif
46#include "zipwrap.hpp"
Ethan Yonkerf1179622016-08-25 15:32:21 -050047#ifdef USE_OLD_VERIFIER
Ethan Yonker4bf259f2016-08-29 11:50:34 -050048#include "verifier24/verifier.h"
Ethan Yonkerf1179622016-08-25 15:32:21 -050049#else
Ethan Yonker75bf0412014-11-21 13:54:27 -060050#include "verifier.h"
Ethan Yonkerf1179622016-08-25 15:32:21 -050051#endif
Dees_Troy32c8eb82012-09-11 15:28:06 -040052#include "variables.h"
53#include "data.hpp"
54#include "partitions.hpp"
bigbiff bigbiff56cf5642016-08-19 17:43:45 -040055#include "twrpDigestDriver.hpp"
56#include "twrpDigest/twrpDigest.hpp"
57#include "twrpDigest/twrpMD5.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040058#include "twrp-functions.hpp"
Ethan Yonker74db1572015-10-28 12:44:49 -050059#include "gui/gui.hpp"
Ethan Yonker20eb0bc2016-03-22 14:23:28 -050060#include "gui/pages.hpp"
Ethan Yonkerf1179622016-08-25 15:32:21 -050061#include "legacy_property_service.h"
Ethan Yonker941a8992016-12-05 09:04:30 -060062#include "twinstall.h"
63#include "installcommand.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000064extern "C" {
65 #include "gui/gui.h"
that7e303cf2014-03-06 07:57:43 +010066}
67
Ethan Yonker941a8992016-12-05 09:04:30 -060068#define AB_OTA "payload_properties.txt"
69
that7e303cf2014-03-06 07:57:43 +010070static const char* properties_path = "/dev/__properties__";
71static const char* properties_path_renamed = "/dev/__properties_kk__";
Matt Mowercdd3b332014-03-27 14:38:48 -050072static bool legacy_props_env_initd = false;
73static bool legacy_props_path_modified = false;
that7e303cf2014-03-06 07:57:43 +010074
Ethan Yonker941a8992016-12-05 09:04:30 -060075enum zip_type {
76 UNKNOWN_ZIP_TYPE = 0,
77 UPDATE_BINARY_ZIP_TYPE,
78 AB_OTA_ZIP_TYPE,
79 TWRP_THEME_ZIP_TYPE
80};
81
that50640482015-08-30 12:08:05 +020082// to support pre-KitKat update-binaries that expect properties in the legacy format
Matt Mowercdd3b332014-03-27 14:38:48 -050083static int switch_to_legacy_properties()
that7e303cf2014-03-06 07:57:43 +010084{
Matt Mowercdd3b332014-03-27 14:38:48 -050085 if (!legacy_props_env_initd) {
86 if (legacy_properties_init() != 0)
87 return -1;
88
89 char tmp[32];
90 int propfd, propsz;
91 legacy_get_property_workspace(&propfd, &propsz);
92 sprintf(tmp, "%d,%d", dup(propfd), propsz);
93 setenv("ANDROID_PROPERTY_WORKSPACE", tmp, 1);
94 legacy_props_env_initd = true;
95 }
that7e303cf2014-03-06 07:57:43 +010096
97 if (TWFunc::Path_Exists(properties_path)) {
98 // hide real properties so that the updater uses the envvar to find the legacy format properties
Matt Mowercdd3b332014-03-27 14:38:48 -050099 if (rename(properties_path, properties_path_renamed) != 0) {
100 LOGERR("Renaming %s failed: %s\n", properties_path, strerror(errno));
101 return -1;
102 } else {
103 legacy_props_path_modified = true;
104 }
that7e303cf2014-03-06 07:57:43 +0100105 }
Matt Mowercdd3b332014-03-27 14:38:48 -0500106
107 return 0;
that7e303cf2014-03-06 07:57:43 +0100108}
109
Matt Mowercdd3b332014-03-27 14:38:48 -0500110static int switch_to_new_properties()
that7e303cf2014-03-06 07:57:43 +0100111{
112 if (TWFunc::Path_Exists(properties_path_renamed)) {
Matt Mowercdd3b332014-03-27 14:38:48 -0500113 if (rename(properties_path_renamed, properties_path) != 0) {
114 LOGERR("Renaming %s failed: %s\n", properties_path_renamed, strerror(errno));
115 return -1;
116 } else {
117 legacy_props_path_modified = false;
118 }
that7e303cf2014-03-06 07:57:43 +0100119 }
Matt Mowercdd3b332014-03-27 14:38:48 -0500120
121 return 0;
Dees_Troy2673cec2013-04-02 20:22:16 +0000122}
Dees_Troy32c8eb82012-09-11 15:28:06 -0400123
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500124static int Install_Theme(const char* path, ZipWrap *Zip) {
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500125#ifdef TW_OEM_BUILD // We don't do custom themes in OEM builds
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500126 Zip->Close();
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500127 return INSTALL_CORRUPT;
128#else
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500129 if (!Zip->EntryExists("ui.xml")) {
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500130 return INSTALL_CORRUPT;
131 }
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500132 Zip->Close();
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500133 if (!PartitionManager.Mount_Settings_Storage(true))
134 return INSTALL_ERROR;
135 string theme_path = DataManager::GetSettingsStoragePath();
136 theme_path += "/TWRP/theme";
137 if (!TWFunc::Path_Exists(theme_path)) {
138 if (!TWFunc::Recursive_Mkdir(theme_path)) {
139 return INSTALL_ERROR;
140 }
141 }
142 theme_path += "/ui.zip";
143 if (TWFunc::copy_file(path, theme_path, 0644) != 0) {
144 return INSTALL_ERROR;
145 }
146 LOGINFO("Installing custom theme '%s' to '%s'\n", path, theme_path.c_str());
147 PageManager::RequestReload();
148 return INSTALL_SUCCESS;
149#endif
150}
151
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500152static int Prepare_Update_Binary(const char *path, ZipWrap *Zip, int* wipe_cache) {
153 if (!Zip->ExtractEntry(ASSUMED_UPDATE_BINARY_NAME, TMP_UPDATER_BINARY_PATH, 0755)) {
154 Zip->Close();
Dees_Troy2673cec2013-04-02 20:22:16 +0000155 LOGERR("Could not extract '%s'\n", ASSUMED_UPDATE_BINARY_NAME);
156 return INSTALL_ERROR;
157 }
Dees_Troy32c8eb82012-09-11 15:28:06 -0400158
Dees_Troy512376c2013-09-03 19:39:41 +0000159 // If exists, extract file_contexts from the zip file
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500160 if (!Zip->EntryExists("file_contexts")) {
161 Zip->Close();
Dees_Troy512376c2013-09-03 19:39:41 +0000162 LOGINFO("Zip does not contain SELinux file_contexts file in its root.\n");
163 } else {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500164 const string output_filename = "/file_contexts";
Dees_Troy512376c2013-09-03 19:39:41 +0000165 LOGINFO("Zip contains SELinux file_contexts file in its root. Extracting to %s\n", output_filename.c_str());
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500166 if (!Zip->ExtractEntry("file_contexts", output_filename, 0644)) {
167 Zip->Close();
that50640482015-08-30 12:08:05 +0200168 LOGERR("Could not extract '%s'\n", output_filename.c_str());
Dees_Troy512376c2013-09-03 19:39:41 +0000169 return INSTALL_ERROR;
170 }
171 }
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500172 Zip->Close();
Ethan Yonker941a8992016-12-05 09:04:30 -0600173 return INSTALL_SUCCESS;
174}
175
nkk71b4c35912017-10-11 23:39:10 +0300176static bool update_binary_has_legacy_properties(const char *binary) {
177 const char str_to_match[] = "ANDROID_PROPERTY_WORKSPACE";
178 int len_to_match = sizeof(str_to_match) - 1;
179 bool found = false;
180
181 int fd = open(binary, O_RDONLY);
182 if (fd < 0) {
183 LOGINFO("has_legacy_properties: Could not open %s: %s!\n", binary, strerror(errno));
184 return false;
185 }
186
187 struct stat finfo;
188 if (fstat(fd, &finfo) < 0) {
189 LOGINFO("has_legacy_properties: Could not fstat %d: %s!\n", fd, strerror(errno));
190 close(fd);
191 return false;
192 }
193
194 void *data = mmap(NULL, finfo.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
195 if (data == MAP_FAILED) {
196 LOGINFO("has_legacy_properties: mmap (size=%lld) failed: %s!\n", finfo.st_size, strerror(errno));
197 } else {
198 if (memmem(data, finfo.st_size, str_to_match, len_to_match)) {
199 LOGINFO("has_legacy_properties: Found legacy property match!\n");
200 found = true;
201 }
202 munmap(data, finfo.st_size);
203 }
204 close(fd);
205
206 return found;
207}
208
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500209static int Run_Update_Binary(const char *path, ZipWrap *Zip, int* wipe_cache, zip_type ztype) {
Ethan Yonker941a8992016-12-05 09:04:30 -0600210 int ret_val, pipe_fd[2], status, zip_verify;
211 char buffer[1024];
212 FILE* child_data;
Dees_Troy512376c2013-09-03 19:39:41 +0000213
Matt Mower6883d732014-03-20 17:28:13 -0500214#ifndef TW_NO_LEGACY_PROPS
nkk71b4c35912017-10-11 23:39:10 +0300215 if (!update_binary_has_legacy_properties(TMP_UPDATER_BINARY_PATH)) {
216 LOGINFO("Legacy property environment not used in updater.\n");
217 } else if (switch_to_legacy_properties() != 0) { /* Set legacy properties */
218 LOGERR("Legacy property environment did not initialize successfully. Properties may not be detected.\n");
219 } else {
220 LOGINFO("Legacy property environment initialized.\n");
Matt Mowercdd3b332014-03-27 14:38:48 -0500221 }
Matt Mower6883d732014-03-20 17:28:13 -0500222#endif
Matt Mowercdd3b332014-03-27 14:38:48 -0500223
Dees_Troy2673cec2013-04-02 20:22:16 +0000224 pipe(pipe_fd);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400225
Ethan Yonker941a8992016-12-05 09:04:30 -0600226 std::vector<std::string> args;
227 if (ztype == UPDATE_BINARY_ZIP_TYPE) {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500228 ret_val = update_binary_command(path, 0, pipe_fd[1], &args);
Ethan Yonker941a8992016-12-05 09:04:30 -0600229 } else if (ztype == AB_OTA_ZIP_TYPE) {
230 ret_val = abupdate_binary_command(path, Zip, 0, pipe_fd[1], &args);
231 } else {
232 LOGERR("Unknown zip type %i\n", ztype);
233 ret_val = INSTALL_CORRUPT;
234 }
235 if (ret_val) {
236 close(pipe_fd[0]);
237 close(pipe_fd[1]);
238 return ret_val;
239 }
240
241 // Convert the vector to a NULL-terminated char* array suitable for execv.
242 const char* chr_args[args.size() + 1];
243 chr_args[args.size()] = NULL;
244 for (size_t i = 0; i < args.size(); i++)
245 chr_args[i] = args[i].c_str();
Dees_Troy32c8eb82012-09-11 15:28:06 -0400246
Dees_Troy2673cec2013-04-02 20:22:16 +0000247 pid_t pid = fork();
248 if (pid == 0) {
249 close(pipe_fd[0]);
Ethan Yonker941a8992016-12-05 09:04:30 -0600250 execve(chr_args[0], const_cast<char**>(chr_args), environ);
251 printf("E:Can't execute '%s': %s\n", chr_args[0], strerror(errno));
Dees_Troy2673cec2013-04-02 20:22:16 +0000252 _exit(-1);
253 }
254 close(pipe_fd[1]);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400255
Dees_Troy2673cec2013-04-02 20:22:16 +0000256 *wipe_cache = 0;
Dees_Troy32c8eb82012-09-11 15:28:06 -0400257
Dees_Troy2673cec2013-04-02 20:22:16 +0000258 DataManager::GetValue(TW_SIGNED_ZIP_VERIFY_VAR, zip_verify);
259 child_data = fdopen(pipe_fd[0], "r");
260 while (fgets(buffer, sizeof(buffer), child_data) != NULL) {
261 char* command = strtok(buffer, " \n");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200262 if (command == NULL) {
263 continue;
264 } else if (strcmp(command, "progress") == 0) {
265 char* fraction_char = strtok(NULL, " \n");
266 char* seconds_char = strtok(NULL, " \n");
Dees_Troy32c8eb82012-09-11 15:28:06 -0400267
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200268 float fraction_float = strtof(fraction_char, NULL);
269 int seconds_float = strtol(seconds_char, NULL, 10);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400270
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200271 if (zip_verify)
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500272 DataManager::ShowProgress(fraction_float * (1 - VERIFICATION_PROGRESS_FRAC), seconds_float);
Dees_Troy2673cec2013-04-02 20:22:16 +0000273 else
274 DataManager::ShowProgress(fraction_float, seconds_float);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200275 } else if (strcmp(command, "set_progress") == 0) {
276 char* fraction_char = strtok(NULL, " \n");
277 float fraction_float = strtof(fraction_char, NULL);
278 DataManager::SetProgress(fraction_float);
279 } else if (strcmp(command, "ui_print") == 0) {
280 char* display_value = strtok(NULL, "\n");
281 if (display_value) {
282 gui_print("%s", display_value);
283 } else {
284 gui_print("\n");
285 }
286 } else if (strcmp(command, "wipe_cache") == 0) {
287 *wipe_cache = 1;
288 } else if (strcmp(command, "clear_display") == 0) {
289 // Do nothing, not supported by TWRP
Ethan Yonker072c8d82016-08-26 22:22:24 -0500290 } else if (strcmp(command, "log") == 0) {
291 printf("%s\n", strtok(NULL, "\n"));
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200292 } else {
293 LOGERR("unknown command [%s]\n", command);
294 }
Dees_Troy2673cec2013-04-02 20:22:16 +0000295 }
296 fclose(child_data);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400297
that50640482015-08-30 12:08:05 +0200298 int waitrc = TWFunc::Wait_For_Child(pid, &status, "Updater");
Matt Mowercdd3b332014-03-27 14:38:48 -0500299
Matt Mower6883d732014-03-20 17:28:13 -0500300#ifndef TW_NO_LEGACY_PROPS
nkk71b4c35912017-10-11 23:39:10 +0300301 /* Unset legacy properties */
302 if (legacy_props_path_modified) {
303 if (switch_to_new_properties() != 0) {
304 LOGERR("Legacy property environment did not disable successfully. Legacy properties may still be in use.\n");
305 } else {
306 LOGINFO("Legacy property environment disabled.\n");
Matt Mowercdd3b332014-03-27 14:38:48 -0500307 }
308 }
Matt Mower6883d732014-03-20 17:28:13 -0500309#endif
Matt Mowercdd3b332014-03-27 14:38:48 -0500310
that50640482015-08-30 12:08:05 +0200311 if (waitrc != 0)
Dees_Troy2673cec2013-04-02 20:22:16 +0000312 return INSTALL_ERROR;
Dees_Troy32c8eb82012-09-11 15:28:06 -0400313
Dees_Troy2673cec2013-04-02 20:22:16 +0000314 return INSTALL_SUCCESS;
Dees_Troy32c8eb82012-09-11 15:28:06 -0400315}
316
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500317int TWinstall_zip(const char* path, int* wipe_cache) {
that50640482015-08-30 12:08:05 +0200318 int ret_val, zip_verify = 1;
Dees_Troy32c8eb82012-09-11 15:28:06 -0400319
Ethan Yonker24813422014-11-07 17:19:07 -0600320 if (strcmp(path, "error") == 0) {
321 LOGERR("Failed to get adb sideload file: '%s'\n", path);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400322 return INSTALL_CORRUPT;
Matt Mowerd5c1a922014-04-15 12:50:58 -0500323 }
Dees_Troy32c8eb82012-09-11 15:28:06 -0400324
Ethan Yonker74db1572015-10-28 12:44:49 -0500325 gui_msg(Msg("installing_zip=Installing zip file '{1}'")(path));
Ethan Yonker24813422014-11-07 17:19:07 -0600326 if (strlen(path) < 9 || strncmp(path, "/sideload", 9) != 0) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400327 string digest_str;
328 string Full_Filename = path;
329 string digest_file = path;
330 digest_file += ".md5";
331
332 gui_msg("check_for_digest=Checking for Digest file...");
333 if (!TWFunc::Path_Exists(digest_file)) {
334 gui_msg("no_digest=Skipping Digest check: no Digest file found");
335 }
336 else {
337 if (TWFunc::read_file(digest_file, digest_str) != 0) {
338 LOGERR("Skipping MD5 check: MD5 file unreadable\n");
339 }
340 else {
341 twrpDigest *digest = new twrpMD5();
342 if (!twrpDigestDriver::stream_file_to_digest(Full_Filename, digest)) {
343 delete digest;
344 return INSTALL_CORRUPT;
345 }
346 string digest_check = digest->return_digest_string();
347 if (digest_str == digest_check) {
348 gui_msg(Msg("digest_matched=Digest matched for '{1}'.")(path));
349 }
350 else {
351 LOGERR("Aborting zip install: Digest verification failed\n");
352 delete digest;
353 return INSTALL_CORRUPT;
354 }
355 delete digest;
356 }
Ethan Yonker24813422014-11-07 17:19:07 -0600357 }
358 }
359
Ethan Yonkerd5801c52014-04-14 08:59:35 -0500360#ifndef TW_OEM_BUILD
Dees_Troy32c8eb82012-09-11 15:28:06 -0400361 DataManager::GetValue(TW_SIGNED_ZIP_VERIFY_VAR, zip_verify);
Ethan Yonkerd5801c52014-04-14 08:59:35 -0500362#endif
Dees_Troy2673cec2013-04-02 20:22:16 +0000363 DataManager::SetProgress(0);
Ethan Yonkerf96087e2014-11-07 10:38:51 -0600364
365 MemMapping map;
366 if (sysMapFile(path, &map) != 0) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500367 gui_msg(Msg(msg::kError, "fail_sysmap=Failed to map file '{1}'")(path));
that50640482015-08-30 12:08:05 +0200368 return -1;
369 }
Ethan Yonkerf96087e2014-11-07 10:38:51 -0600370
Dees_Troy32c8eb82012-09-11 15:28:06 -0400371 if (zip_verify) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500372 gui_msg("verify_zip_sig=Verifying zip signature...");
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500373#ifdef USE_OLD_VERIFIER
Ethan Yonkerf96087e2014-11-07 10:38:51 -0600374 ret_val = verify_file(map.addr, map.length);
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500375#else
376 std::vector<Certificate> loadedKeys;
377 if (!load_keys("/res/keys", loadedKeys)) {
378 LOGINFO("Failed to load keys");
379 gui_err("verify_zip_fail=Zip signature verification failed!");
380 return -1;
381 }
382 ret_val = verify_file(map.addr, map.length, loadedKeys, std::bind(&DataManager::SetProgress, std::placeholders::_1));
383#endif
Dees_Troy2673cec2013-04-02 20:22:16 +0000384 if (ret_val != VERIFY_SUCCESS) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500385 LOGINFO("Zip signature verification failed: %i\n", ret_val);
386 gui_err("verify_zip_fail=Zip signature verification failed!");
Ethan Yonkerf9796a42014-11-08 07:28:03 -0600387 sysReleaseMap(&map);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400388 return -1;
Ethan Yonker738be7a2014-12-10 11:40:43 -0600389 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500390 gui_msg("verify_zip_done=Zip signature verified successfully.");
Dees_Troy32c8eb82012-09-11 15:28:06 -0400391 }
392 }
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500393 ZipWrap Zip;
394 if (!Zip.Open(path, &map)) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500395 gui_err("zip_corrupt=Zip file is corrupt!");
Ethan Yonkerf9796a42014-11-08 07:28:03 -0600396 sysReleaseMap(&map);
Dees_Troy2673cec2013-04-02 20:22:16 +0000397 return INSTALL_CORRUPT;
398 }
Ethan Yonker941a8992016-12-05 09:04:30 -0600399
Ethan Yonker072c8d82016-08-26 22:22:24 -0500400 time_t start, stop;
401 time(&start);
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500402 if (Zip.EntryExists(ASSUMED_UPDATE_BINARY_NAME)) {
Ethan Yonker941a8992016-12-05 09:04:30 -0600403 LOGINFO("Update binary zip\n");
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500404 // Additionally verify the compatibility of the package.
405 if (!verify_package_compatibility(&Zip)) {
406 gui_err("zip_compatible_err=Zip Treble compatibility error!");
407 sysReleaseMap(&map);
408 Zip.Close();
409 ret_val = INSTALL_CORRUPT;
410 } else {
411 ret_val = Prepare_Update_Binary(path, &Zip, wipe_cache);
412 if (ret_val == INSTALL_SUCCESS)
413 ret_val = Run_Update_Binary(path, &Zip, wipe_cache, UPDATE_BINARY_ZIP_TYPE);
414 }
Ethan Yonker941a8992016-12-05 09:04:30 -0600415 } else {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500416 if (Zip.EntryExists(AB_OTA)) {
Ethan Yonker941a8992016-12-05 09:04:30 -0600417 LOGINFO("AB zip\n");
418 ret_val = Run_Update_Binary(path, &Zip, wipe_cache, AB_OTA_ZIP_TYPE);
419 } else {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500420 if (Zip.EntryExists("ui.xml")) {
Ethan Yonker941a8992016-12-05 09:04:30 -0600421 LOGINFO("TWRP theme zip\n");
422 ret_val = Install_Theme(path, &Zip);
423 } else {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500424 Zip.Close();
Ethan Yonker941a8992016-12-05 09:04:30 -0600425 ret_val = INSTALL_CORRUPT;
426 }
427 }
428 }
Ethan Yonker072c8d82016-08-26 22:22:24 -0500429 time(&stop);
430 int total_time = (int) difftime(stop, start);
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500431 if (ret_val == INSTALL_CORRUPT) {
Ethan Yonker941a8992016-12-05 09:04:30 -0600432 gui_err("invalid_zip_format=Invalid zip file format!");
Ethan Yonker072c8d82016-08-26 22:22:24 -0500433 } else {
434 LOGINFO("Install took %i second(s).\n", total_time);
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500435 }
Ethan Yonkerf9796a42014-11-08 07:28:03 -0600436 sysReleaseMap(&map);
437 return ret_val;
Dees_Troy32c8eb82012-09-11 15:28:06 -0400438}