blob: fb639a7e19494452b1d588f280eb2ca167858e81 [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>
Ethan Yonker193befe2019-04-02 16:01:31 -050035#include <cutils/properties.h>
Dees_Troy32c8eb82012-09-11 15:28:06 -040036
Dees_Troy2673cec2013-04-02 20:22:16 +000037#include "twcommon.h"
Ethan Yonker75bf0412014-11-21 13:54:27 -060038#include "mtdutils/mounts.h"
39#include "mtdutils/mtdutils.h"
Ethan Yonker8373cfe2017-09-08 06:50:54 -050040
41#ifdef USE_MINZIP
Dees_Troy32c8eb82012-09-11 15:28:06 -040042#include "minzip/SysUtil.h"
Ethan Yonker8373cfe2017-09-08 06:50:54 -050043#else
bigbiffd58ba182020-03-23 10:02:29 -040044#include "otautil/sysutil.h"
Ethan Yonker8373cfe2017-09-08 06:50:54 -050045#include <ziparchive/zip_archive.h>
46#endif
47#include "zipwrap.hpp"
Ethan Yonkerf1179622016-08-25 15:32:21 -050048#ifdef USE_OLD_VERIFIER
Ethan Yonker4bf259f2016-08-29 11:50:34 -050049#include "verifier24/verifier.h"
bigbiffd58ba182020-03-23 10:02:29 -040050#elif USE_28_VERIFIER
51#include "verifier28/verifier.h"
Ethan Yonkerf1179622016-08-25 15:32:21 -050052#else
bigbiffd58ba182020-03-23 10:02:29 -040053#include "install/verifier.h"
Ethan Yonkerf1179622016-08-25 15:32:21 -050054#endif
Dees_Troy32c8eb82012-09-11 15:28:06 -040055#include "variables.h"
56#include "data.hpp"
57#include "partitions.hpp"
bigbiff bigbiff56cf5642016-08-19 17:43:45 -040058#include "twrpDigestDriver.hpp"
59#include "twrpDigest/twrpDigest.hpp"
60#include "twrpDigest/twrpMD5.hpp"
Dees_Troy38bd7602012-09-14 13:33:53 -040061#include "twrp-functions.hpp"
Ethan Yonker74db1572015-10-28 12:44:49 -050062#include "gui/gui.hpp"
Ethan Yonker20eb0bc2016-03-22 14:23:28 -050063#include "gui/pages.hpp"
Ethan Yonkerf1179622016-08-25 15:32:21 -050064#include "legacy_property_service.h"
Ethan Yonker941a8992016-12-05 09:04:30 -060065#include "twinstall.h"
66#include "installcommand.h"
Dees_Troy2673cec2013-04-02 20:22:16 +000067extern "C" {
68 #include "gui/gui.h"
that7e303cf2014-03-06 07:57:43 +010069}
70
Ethan Yonker941a8992016-12-05 09:04:30 -060071#define AB_OTA "payload_properties.txt"
72
lambdadroidc4faea82018-09-26 22:56:51 +020073#ifndef TW_NO_LEGACY_PROPS
that7e303cf2014-03-06 07:57:43 +010074static const char* properties_path = "/dev/__properties__";
75static const char* properties_path_renamed = "/dev/__properties_kk__";
Matt Mowercdd3b332014-03-27 14:38:48 -050076static bool legacy_props_env_initd = false;
77static bool legacy_props_path_modified = false;
lambdadroidc4faea82018-09-26 22:56:51 +020078#endif
that7e303cf2014-03-06 07:57:43 +010079
Ethan Yonker941a8992016-12-05 09:04:30 -060080enum zip_type {
81 UNKNOWN_ZIP_TYPE = 0,
82 UPDATE_BINARY_ZIP_TYPE,
83 AB_OTA_ZIP_TYPE,
84 TWRP_THEME_ZIP_TYPE
85};
86
lambdadroidc4faea82018-09-26 22:56:51 +020087#ifndef TW_NO_LEGACY_PROPS
that50640482015-08-30 12:08:05 +020088// to support pre-KitKat update-binaries that expect properties in the legacy format
Matt Mowercdd3b332014-03-27 14:38:48 -050089static int switch_to_legacy_properties()
that7e303cf2014-03-06 07:57:43 +010090{
Matt Mowercdd3b332014-03-27 14:38:48 -050091 if (!legacy_props_env_initd) {
92 if (legacy_properties_init() != 0)
93 return -1;
94
95 char tmp[32];
96 int propfd, propsz;
97 legacy_get_property_workspace(&propfd, &propsz);
98 sprintf(tmp, "%d,%d", dup(propfd), propsz);
99 setenv("ANDROID_PROPERTY_WORKSPACE", tmp, 1);
100 legacy_props_env_initd = true;
101 }
that7e303cf2014-03-06 07:57:43 +0100102
103 if (TWFunc::Path_Exists(properties_path)) {
104 // hide real properties so that the updater uses the envvar to find the legacy format properties
Matt Mowercdd3b332014-03-27 14:38:48 -0500105 if (rename(properties_path, properties_path_renamed) != 0) {
106 LOGERR("Renaming %s failed: %s\n", properties_path, strerror(errno));
107 return -1;
108 } else {
109 legacy_props_path_modified = true;
110 }
that7e303cf2014-03-06 07:57:43 +0100111 }
Matt Mowercdd3b332014-03-27 14:38:48 -0500112
113 return 0;
that7e303cf2014-03-06 07:57:43 +0100114}
115
Matt Mowercdd3b332014-03-27 14:38:48 -0500116static int switch_to_new_properties()
that7e303cf2014-03-06 07:57:43 +0100117{
118 if (TWFunc::Path_Exists(properties_path_renamed)) {
Matt Mowercdd3b332014-03-27 14:38:48 -0500119 if (rename(properties_path_renamed, properties_path) != 0) {
120 LOGERR("Renaming %s failed: %s\n", properties_path_renamed, strerror(errno));
121 return -1;
122 } else {
123 legacy_props_path_modified = false;
124 }
that7e303cf2014-03-06 07:57:43 +0100125 }
Matt Mowercdd3b332014-03-27 14:38:48 -0500126
127 return 0;
Dees_Troy2673cec2013-04-02 20:22:16 +0000128}
lambdadroidc4faea82018-09-26 22:56:51 +0200129#endif
Dees_Troy32c8eb82012-09-11 15:28:06 -0400130
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500131static int Install_Theme(const char* path, ZipWrap *Zip) {
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500132#ifdef TW_OEM_BUILD // We don't do custom themes in OEM builds
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500133 Zip->Close();
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500134 return INSTALL_CORRUPT;
135#else
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500136 if (!Zip->EntryExists("ui.xml")) {
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500137 return INSTALL_CORRUPT;
138 }
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500139 Zip->Close();
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500140 if (!PartitionManager.Mount_Settings_Storage(true))
141 return INSTALL_ERROR;
142 string theme_path = DataManager::GetSettingsStoragePath();
143 theme_path += "/TWRP/theme";
144 if (!TWFunc::Path_Exists(theme_path)) {
145 if (!TWFunc::Recursive_Mkdir(theme_path)) {
146 return INSTALL_ERROR;
147 }
148 }
149 theme_path += "/ui.zip";
150 if (TWFunc::copy_file(path, theme_path, 0644) != 0) {
151 return INSTALL_ERROR;
152 }
153 LOGINFO("Installing custom theme '%s' to '%s'\n", path, theme_path.c_str());
154 PageManager::RequestReload();
155 return INSTALL_SUCCESS;
156#endif
157}
158
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500159static int Prepare_Update_Binary(const char *path, ZipWrap *Zip, int* wipe_cache) {
Ethan Yonker193befe2019-04-02 16:01:31 -0500160 char arches[PATH_MAX];
161 std::string binary_name = ASSUMED_UPDATE_BINARY_NAME;
162 property_get("ro.product.cpu.abilist", arches, "error");
163 if (strcmp(arches, "error") == 0)
164 property_get("ro.product.cpu.abi", arches, "error");
165 vector<string> split = TWFunc::split_string(arches, ',', true);
166 std::vector<string>::iterator arch;
167 std::string base_name = binary_name;
168 base_name += "-";
169 for (arch = split.begin(); arch != split.end(); arch++) {
170 std::string temp = base_name + *arch;
171 if (Zip->EntryExists(temp)) {
172 binary_name = temp;
173 break;
174 }
175 }
176 LOGINFO("Extracting updater binary '%s'\n", binary_name.c_str());
177 if (!Zip->ExtractEntry(binary_name.c_str(), TMP_UPDATER_BINARY_PATH, 0755)) {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500178 Zip->Close();
Dees_Troy2673cec2013-04-02 20:22:16 +0000179 LOGERR("Could not extract '%s'\n", ASSUMED_UPDATE_BINARY_NAME);
180 return INSTALL_ERROR;
181 }
Dees_Troy32c8eb82012-09-11 15:28:06 -0400182
Dees_Troy512376c2013-09-03 19:39:41 +0000183 // If exists, extract file_contexts from the zip file
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500184 if (!Zip->EntryExists("file_contexts")) {
185 Zip->Close();
Dees_Troy512376c2013-09-03 19:39:41 +0000186 LOGINFO("Zip does not contain SELinux file_contexts file in its root.\n");
187 } else {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500188 const string output_filename = "/file_contexts";
Dees_Troy512376c2013-09-03 19:39:41 +0000189 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 -0500190 if (!Zip->ExtractEntry("file_contexts", output_filename, 0644)) {
191 Zip->Close();
that50640482015-08-30 12:08:05 +0200192 LOGERR("Could not extract '%s'\n", output_filename.c_str());
Dees_Troy512376c2013-09-03 19:39:41 +0000193 return INSTALL_ERROR;
194 }
195 }
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500196 Zip->Close();
Ethan Yonker941a8992016-12-05 09:04:30 -0600197 return INSTALL_SUCCESS;
198}
199
lambdadroidc4faea82018-09-26 22:56:51 +0200200#ifndef TW_NO_LEGACY_PROPS
nkk71b4c35912017-10-11 23:39:10 +0300201static bool update_binary_has_legacy_properties(const char *binary) {
202 const char str_to_match[] = "ANDROID_PROPERTY_WORKSPACE";
203 int len_to_match = sizeof(str_to_match) - 1;
204 bool found = false;
205
206 int fd = open(binary, O_RDONLY);
207 if (fd < 0) {
208 LOGINFO("has_legacy_properties: Could not open %s: %s!\n", binary, strerror(errno));
209 return false;
210 }
211
212 struct stat finfo;
213 if (fstat(fd, &finfo) < 0) {
214 LOGINFO("has_legacy_properties: Could not fstat %d: %s!\n", fd, strerror(errno));
215 close(fd);
216 return false;
217 }
218
219 void *data = mmap(NULL, finfo.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
220 if (data == MAP_FAILED) {
Simon Shi8ae2b0d2019-03-09 13:01:55 +0800221 LOGINFO("has_legacy_properties: mmap (size=%zu) failed: %s!\n", (size_t)finfo.st_size, strerror(errno));
nkk71b4c35912017-10-11 23:39:10 +0300222 } else {
223 if (memmem(data, finfo.st_size, str_to_match, len_to_match)) {
224 LOGINFO("has_legacy_properties: Found legacy property match!\n");
225 found = true;
226 }
227 munmap(data, finfo.st_size);
228 }
229 close(fd);
230
231 return found;
232}
lambdadroidc4faea82018-09-26 22:56:51 +0200233#endif
nkk71b4c35912017-10-11 23:39:10 +0300234
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500235static int Run_Update_Binary(const char *path, ZipWrap *Zip, int* wipe_cache, zip_type ztype) {
Ethan Yonker941a8992016-12-05 09:04:30 -0600236 int ret_val, pipe_fd[2], status, zip_verify;
237 char buffer[1024];
238 FILE* child_data;
Dees_Troy512376c2013-09-03 19:39:41 +0000239
Matt Mower6883d732014-03-20 17:28:13 -0500240#ifndef TW_NO_LEGACY_PROPS
nkk71b4c35912017-10-11 23:39:10 +0300241 if (!update_binary_has_legacy_properties(TMP_UPDATER_BINARY_PATH)) {
242 LOGINFO("Legacy property environment not used in updater.\n");
243 } else if (switch_to_legacy_properties() != 0) { /* Set legacy properties */
244 LOGERR("Legacy property environment did not initialize successfully. Properties may not be detected.\n");
245 } else {
246 LOGINFO("Legacy property environment initialized.\n");
Matt Mowercdd3b332014-03-27 14:38:48 -0500247 }
Matt Mower6883d732014-03-20 17:28:13 -0500248#endif
Matt Mowercdd3b332014-03-27 14:38:48 -0500249
Dees_Troy2673cec2013-04-02 20:22:16 +0000250 pipe(pipe_fd);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400251
Ethan Yonker941a8992016-12-05 09:04:30 -0600252 std::vector<std::string> args;
253 if (ztype == UPDATE_BINARY_ZIP_TYPE) {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500254 ret_val = update_binary_command(path, 0, pipe_fd[1], &args);
Ethan Yonker941a8992016-12-05 09:04:30 -0600255 } else if (ztype == AB_OTA_ZIP_TYPE) {
256 ret_val = abupdate_binary_command(path, Zip, 0, pipe_fd[1], &args);
257 } else {
258 LOGERR("Unknown zip type %i\n", ztype);
259 ret_val = INSTALL_CORRUPT;
260 }
261 if (ret_val) {
262 close(pipe_fd[0]);
263 close(pipe_fd[1]);
264 return ret_val;
265 }
266
267 // Convert the vector to a NULL-terminated char* array suitable for execv.
268 const char* chr_args[args.size() + 1];
269 chr_args[args.size()] = NULL;
270 for (size_t i = 0; i < args.size(); i++)
271 chr_args[i] = args[i].c_str();
Dees_Troy32c8eb82012-09-11 15:28:06 -0400272
Dees_Troy2673cec2013-04-02 20:22:16 +0000273 pid_t pid = fork();
274 if (pid == 0) {
275 close(pipe_fd[0]);
Ethan Yonker941a8992016-12-05 09:04:30 -0600276 execve(chr_args[0], const_cast<char**>(chr_args), environ);
277 printf("E:Can't execute '%s': %s\n", chr_args[0], strerror(errno));
Dees_Troy2673cec2013-04-02 20:22:16 +0000278 _exit(-1);
279 }
280 close(pipe_fd[1]);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400281
Dees_Troy2673cec2013-04-02 20:22:16 +0000282 *wipe_cache = 0;
Dees_Troy32c8eb82012-09-11 15:28:06 -0400283
Dees_Troy2673cec2013-04-02 20:22:16 +0000284 DataManager::GetValue(TW_SIGNED_ZIP_VERIFY_VAR, zip_verify);
285 child_data = fdopen(pipe_fd[0], "r");
286 while (fgets(buffer, sizeof(buffer), child_data) != NULL) {
287 char* command = strtok(buffer, " \n");
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200288 if (command == NULL) {
289 continue;
290 } else if (strcmp(command, "progress") == 0) {
291 char* fraction_char = strtok(NULL, " \n");
292 char* seconds_char = strtok(NULL, " \n");
Dees_Troy32c8eb82012-09-11 15:28:06 -0400293
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200294 float fraction_float = strtof(fraction_char, NULL);
295 int seconds_float = strtol(seconds_char, NULL, 10);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400296
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200297 if (zip_verify)
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500298 DataManager::ShowProgress(fraction_float * (1 - VERIFICATION_PROGRESS_FRAC), seconds_float);
Dees_Troy2673cec2013-04-02 20:22:16 +0000299 else
300 DataManager::ShowProgress(fraction_float, seconds_float);
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200301 } else if (strcmp(command, "set_progress") == 0) {
302 char* fraction_char = strtok(NULL, " \n");
303 float fraction_float = strtof(fraction_char, NULL);
304 DataManager::SetProgress(fraction_float);
305 } else if (strcmp(command, "ui_print") == 0) {
306 char* display_value = strtok(NULL, "\n");
307 if (display_value) {
308 gui_print("%s", display_value);
309 } else {
310 gui_print("\n");
311 }
312 } else if (strcmp(command, "wipe_cache") == 0) {
313 *wipe_cache = 1;
314 } else if (strcmp(command, "clear_display") == 0) {
315 // Do nothing, not supported by TWRP
Ethan Yonker072c8d82016-08-26 22:22:24 -0500316 } else if (strcmp(command, "log") == 0) {
317 printf("%s\n", strtok(NULL, "\n"));
Vojtech Bocekfafb0c52013-07-25 22:53:02 +0200318 } else {
319 LOGERR("unknown command [%s]\n", command);
320 }
Dees_Troy2673cec2013-04-02 20:22:16 +0000321 }
322 fclose(child_data);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400323
that50640482015-08-30 12:08:05 +0200324 int waitrc = TWFunc::Wait_For_Child(pid, &status, "Updater");
Matt Mowercdd3b332014-03-27 14:38:48 -0500325
Matt Mower6883d732014-03-20 17:28:13 -0500326#ifndef TW_NO_LEGACY_PROPS
nkk71b4c35912017-10-11 23:39:10 +0300327 /* Unset legacy properties */
328 if (legacy_props_path_modified) {
329 if (switch_to_new_properties() != 0) {
330 LOGERR("Legacy property environment did not disable successfully. Legacy properties may still be in use.\n");
331 } else {
332 LOGINFO("Legacy property environment disabled.\n");
Matt Mowercdd3b332014-03-27 14:38:48 -0500333 }
334 }
Matt Mower6883d732014-03-20 17:28:13 -0500335#endif
Matt Mowercdd3b332014-03-27 14:38:48 -0500336
that50640482015-08-30 12:08:05 +0200337 if (waitrc != 0)
Dees_Troy2673cec2013-04-02 20:22:16 +0000338 return INSTALL_ERROR;
Dees_Troy32c8eb82012-09-11 15:28:06 -0400339
Dees_Troy2673cec2013-04-02 20:22:16 +0000340 return INSTALL_SUCCESS;
Dees_Troy32c8eb82012-09-11 15:28:06 -0400341}
342
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500343int TWinstall_zip(const char* path, int* wipe_cache) {
that50640482015-08-30 12:08:05 +0200344 int ret_val, zip_verify = 1;
Dees_Troy32c8eb82012-09-11 15:28:06 -0400345
Ethan Yonker24813422014-11-07 17:19:07 -0600346 if (strcmp(path, "error") == 0) {
347 LOGERR("Failed to get adb sideload file: '%s'\n", path);
Dees_Troy32c8eb82012-09-11 15:28:06 -0400348 return INSTALL_CORRUPT;
Matt Mowerd5c1a922014-04-15 12:50:58 -0500349 }
Dees_Troy32c8eb82012-09-11 15:28:06 -0400350
Ethan Yonker74db1572015-10-28 12:44:49 -0500351 gui_msg(Msg("installing_zip=Installing zip file '{1}'")(path));
Ethan Yonker24813422014-11-07 17:19:07 -0600352 if (strlen(path) < 9 || strncmp(path, "/sideload", 9) != 0) {
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400353 string digest_str;
354 string Full_Filename = path;
bigbiff bigbiff56cf5642016-08-19 17:43:45 -0400355
356 gui_msg("check_for_digest=Checking for Digest file...");
bigbiff bigbiff718ab392019-03-28 19:46:56 -0400357
Ethan Yonker1da568f2019-04-09 15:29:51 -0500358 if (*path != '@' && !twrpDigestDriver::Check_File_Digest(Full_Filename)) {
359 LOGERR("Aborting zip install: Digest verification failed\n");
360 return INSTALL_CORRUPT;
Ethan Yonker24813422014-11-07 17:19:07 -0600361 }
362 }
363
Ethan Yonkerd5801c52014-04-14 08:59:35 -0500364#ifndef TW_OEM_BUILD
Dees_Troy32c8eb82012-09-11 15:28:06 -0400365 DataManager::GetValue(TW_SIGNED_ZIP_VERIFY_VAR, zip_verify);
Ethan Yonkerd5801c52014-04-14 08:59:35 -0500366#endif
Dees_Troy2673cec2013-04-02 20:22:16 +0000367 DataManager::SetProgress(0);
Ethan Yonkerf96087e2014-11-07 10:38:51 -0600368
369 MemMapping map;
Ethan Yonkerecbd3e82017-12-14 14:43:59 -0600370#ifdef USE_MINZIP
Ethan Yonkerf96087e2014-11-07 10:38:51 -0600371 if (sysMapFile(path, &map) != 0) {
Ethan Yonkerecbd3e82017-12-14 14:43:59 -0600372#else
373 if (!map.MapFile(path)) {
374#endif
Ethan Yonker74db1572015-10-28 12:44:49 -0500375 gui_msg(Msg(msg::kError, "fail_sysmap=Failed to map file '{1}'")(path));
that50640482015-08-30 12:08:05 +0200376 return -1;
377 }
Ethan Yonkerf96087e2014-11-07 10:38:51 -0600378
Dees_Troy32c8eb82012-09-11 15:28:06 -0400379 if (zip_verify) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500380 gui_msg("verify_zip_sig=Verifying zip signature...");
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500381#ifdef USE_OLD_VERIFIER
Ethan Yonkerf96087e2014-11-07 10:38:51 -0600382 ret_val = verify_file(map.addr, map.length);
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500383#else
384 std::vector<Certificate> loadedKeys;
385 if (!load_keys("/res/keys", loadedKeys)) {
386 LOGINFO("Failed to load keys");
387 gui_err("verify_zip_fail=Zip signature verification failed!");
Ethan Yonkerecbd3e82017-12-14 14:43:59 -0600388#ifdef USE_MINZIP
389 sysReleaseMap(&map);
390#endif
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500391 return -1;
392 }
393 ret_val = verify_file(map.addr, map.length, loadedKeys, std::bind(&DataManager::SetProgress, std::placeholders::_1));
394#endif
Dees_Troy2673cec2013-04-02 20:22:16 +0000395 if (ret_val != VERIFY_SUCCESS) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500396 LOGINFO("Zip signature verification failed: %i\n", ret_val);
397 gui_err("verify_zip_fail=Zip signature verification failed!");
Ethan Yonkerecbd3e82017-12-14 14:43:59 -0600398#ifdef USE_MINZIP
Ethan Yonkerf9796a42014-11-08 07:28:03 -0600399 sysReleaseMap(&map);
Ethan Yonkerecbd3e82017-12-14 14:43:59 -0600400#endif
Dees_Troy32c8eb82012-09-11 15:28:06 -0400401 return -1;
Ethan Yonker738be7a2014-12-10 11:40:43 -0600402 } else {
Ethan Yonker74db1572015-10-28 12:44:49 -0500403 gui_msg("verify_zip_done=Zip signature verified successfully.");
Dees_Troy32c8eb82012-09-11 15:28:06 -0400404 }
405 }
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500406 ZipWrap Zip;
407 if (!Zip.Open(path, &map)) {
Ethan Yonker74db1572015-10-28 12:44:49 -0500408 gui_err("zip_corrupt=Zip file is corrupt!");
Ethan Yonkerecbd3e82017-12-14 14:43:59 -0600409#ifdef USE_MINZIP
410 sysReleaseMap(&map);
411#endif
Dees_Troy2673cec2013-04-02 20:22:16 +0000412 return INSTALL_CORRUPT;
413 }
Ethan Yonker941a8992016-12-05 09:04:30 -0600414
Ethan Yonker072c8d82016-08-26 22:22:24 -0500415 time_t start, stop;
416 time(&start);
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500417 if (Zip.EntryExists(ASSUMED_UPDATE_BINARY_NAME)) {
Ethan Yonker941a8992016-12-05 09:04:30 -0600418 LOGINFO("Update binary zip\n");
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500419 // Additionally verify the compatibility of the package.
420 if (!verify_package_compatibility(&Zip)) {
421 gui_err("zip_compatible_err=Zip Treble compatibility error!");
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500422 Zip.Close();
Ethan Yonkerecbd3e82017-12-14 14:43:59 -0600423#ifdef USE_MINZIP
424 sysReleaseMap(&map);
425#endif
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500426 ret_val = INSTALL_CORRUPT;
427 } else {
428 ret_val = Prepare_Update_Binary(path, &Zip, wipe_cache);
429 if (ret_val == INSTALL_SUCCESS)
430 ret_val = Run_Update_Binary(path, &Zip, wipe_cache, UPDATE_BINARY_ZIP_TYPE);
431 }
Ethan Yonker941a8992016-12-05 09:04:30 -0600432 } else {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500433 if (Zip.EntryExists(AB_OTA)) {
Ethan Yonker941a8992016-12-05 09:04:30 -0600434 LOGINFO("AB zip\n");
435 ret_val = Run_Update_Binary(path, &Zip, wipe_cache, AB_OTA_ZIP_TYPE);
436 } else {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500437 if (Zip.EntryExists("ui.xml")) {
Ethan Yonker941a8992016-12-05 09:04:30 -0600438 LOGINFO("TWRP theme zip\n");
439 ret_val = Install_Theme(path, &Zip);
440 } else {
Ethan Yonker8373cfe2017-09-08 06:50:54 -0500441 Zip.Close();
Ethan Yonker941a8992016-12-05 09:04:30 -0600442 ret_val = INSTALL_CORRUPT;
443 }
444 }
445 }
Ethan Yonker072c8d82016-08-26 22:22:24 -0500446 time(&stop);
447 int total_time = (int) difftime(stop, start);
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500448 if (ret_val == INSTALL_CORRUPT) {
Ethan Yonker941a8992016-12-05 09:04:30 -0600449 gui_err("invalid_zip_format=Invalid zip file format!");
Ethan Yonker072c8d82016-08-26 22:22:24 -0500450 } else {
451 LOGINFO("Install took %i second(s).\n", total_time);
Ethan Yonker20eb0bc2016-03-22 14:23:28 -0500452 }
Ethan Yonkerecbd3e82017-12-14 14:43:59 -0600453#ifdef USE_MINZIP
Ethan Yonkerf9796a42014-11-08 07:28:03 -0600454 sysReleaseMap(&map);
Ethan Yonkerecbd3e82017-12-14 14:43:59 -0600455#endif
Ethan Yonkerf9796a42014-11-08 07:28:03 -0600456 return ret_val;
Dees_Troy32c8eb82012-09-11 15:28:06 -0400457}