Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2009 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
Tao Bao | 0c7839a | 2016-10-10 15:48:37 -0700 | [diff] [blame] | 17 | #include "updater/updater.h" |
Tao Bao | 876b939 | 2016-10-03 18:06:46 -0700 | [diff] [blame] | 18 | |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 19 | #include <stdio.h> |
| 20 | #include <unistd.h> |
| 21 | #include <stdlib.h> |
Elliott Hughes | cd3c55a | 2015-01-29 20:50:08 -0800 | [diff] [blame] | 22 | #include <string.h> |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 23 | |
Yabin Cui | a029c9a | 2016-02-18 11:32:10 -0800 | [diff] [blame] | 24 | #include <string> |
| 25 | |
Tao Bao | 876b939 | 2016-10-03 18:06:46 -0700 | [diff] [blame] | 26 | #include <android-base/strings.h> |
Elliott Hughes | 4bbd5bf | 2016-04-01 18:24:39 -0700 | [diff] [blame] | 27 | #include <selinux/label.h> |
| 28 | #include <selinux/selinux.h> |
| 29 | |
Tao Bao | 876b939 | 2016-10-03 18:06:46 -0700 | [diff] [blame] | 30 | #include "config.h" |
| 31 | #include "edify/expr.h" |
Tao Bao | 876b939 | 2016-10-03 18:06:46 -0700 | [diff] [blame] | 32 | #include "minzip/SysUtil.h" |
Tao Bao | 0c7839a | 2016-10-10 15:48:37 -0700 | [diff] [blame] | 33 | #include "minzip/Zip.h" |
| 34 | #include "updater/blockimg.h" |
| 35 | #include "updater/install.h" |
Tao Bao | 876b939 | 2016-10-03 18:06:46 -0700 | [diff] [blame] | 36 | |
Doug Zongker | 2b0fdc6 | 2009-06-19 11:22:06 -0700 | [diff] [blame] | 37 | // Generated by the makefile, this function defines the |
| 38 | // RegisterDeviceExtensions() function, which calls all the |
| 39 | // registration functions for device-specific extensions. |
| 40 | #include "register.inc" |
| 41 | |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 42 | // Where in the package we expect to find the edify script to execute. |
| 43 | // (Note it's "updateR-script", not the older "update-script".) |
| 44 | #define SCRIPT_NAME "META-INF/com/google/android/updater-script" |
| 45 | |
Tianjie Xu | 3c62b67 | 2016-02-05 18:25:58 -0800 | [diff] [blame] | 46 | extern bool have_eio_error; |
| 47 | |
Stephen Smalley | 779701d | 2012-02-09 14:13:23 -0500 | [diff] [blame] | 48 | struct selabel_handle *sehandle; |
| 49 | |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 50 | int main(int argc, char** argv) { |
Doug Zongker | 512536a | 2010-02-17 16:11:44 -0800 | [diff] [blame] | 51 | // Various things log information to stdout or stderr more or less |
Doug Zongker | fafc85b | 2013-07-09 12:29:45 -0700 | [diff] [blame] | 52 | // at random (though we've tried to standardize on stdout). The |
| 53 | // log file makes more sense if buffering is turned off so things |
| 54 | // appear in the right order. |
Doug Zongker | 512536a | 2010-02-17 16:11:44 -0800 | [diff] [blame] | 55 | setbuf(stdout, NULL); |
| 56 | setbuf(stderr, NULL); |
| 57 | |
Tianjie Xu | 7ce287d | 2016-05-31 09:29:49 -0700 | [diff] [blame] | 58 | if (argc != 4 && argc != 5) { |
Doug Zongker | fafc85b | 2013-07-09 12:29:45 -0700 | [diff] [blame] | 59 | printf("unexpected number of arguments (%d)\n", argc); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 60 | return 1; |
| 61 | } |
| 62 | |
| 63 | char* version = argv[1]; |
Doug Zongker | e08991e | 2010-02-02 13:09:52 -0800 | [diff] [blame] | 64 | if ((version[0] != '1' && version[0] != '2' && version[0] != '3') || |
| 65 | version[1] != '\0') { |
| 66 | // We support version 1, 2, or 3. |
Doug Zongker | fafc85b | 2013-07-09 12:29:45 -0700 | [diff] [blame] | 67 | printf("wrong updater binary API; expected 1, 2, or 3; " |
Doug Zongker | e08991e | 2010-02-02 13:09:52 -0800 | [diff] [blame] | 68 | "got %s\n", |
Doug Zongker | fb2e3af | 2009-06-17 17:29:40 -0700 | [diff] [blame] | 69 | argv[1]); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 70 | return 2; |
| 71 | } |
| 72 | |
| 73 | // Set up the pipe for sending commands back to the parent process. |
| 74 | |
| 75 | int fd = atoi(argv[2]); |
| 76 | FILE* cmd_pipe = fdopen(fd, "wb"); |
| 77 | setlinebuf(cmd_pipe); |
| 78 | |
| 79 | // Extract the script from the package. |
| 80 | |
Doug Zongker | 99916f0 | 2014-01-13 14:16:58 -0800 | [diff] [blame] | 81 | const char* package_filename = argv[3]; |
| 82 | MemMapping map; |
| 83 | if (sysMapFile(package_filename, &map) != 0) { |
| 84 | printf("failed to map package %s\n", argv[3]); |
| 85 | return 3; |
| 86 | } |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 87 | ZipArchive za; |
| 88 | int err; |
Doug Zongker | 99916f0 | 2014-01-13 14:16:58 -0800 | [diff] [blame] | 89 | err = mzOpenZipArchive(map.addr, map.length, &za); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 90 | if (err != 0) { |
Doug Zongker | fafc85b | 2013-07-09 12:29:45 -0700 | [diff] [blame] | 91 | printf("failed to open package %s: %s\n", |
Doug Zongker | 99916f0 | 2014-01-13 14:16:58 -0800 | [diff] [blame] | 92 | argv[3], strerror(err)); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 93 | return 3; |
| 94 | } |
Jed Estep | ff6df89 | 2015-12-15 16:04:53 -0800 | [diff] [blame] | 95 | ota_io_init(&za); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 96 | |
| 97 | const ZipEntry* script_entry = mzFindZipEntry(&za, SCRIPT_NAME); |
| 98 | if (script_entry == NULL) { |
Doug Zongker | 99916f0 | 2014-01-13 14:16:58 -0800 | [diff] [blame] | 99 | printf("failed to find %s in %s\n", SCRIPT_NAME, package_filename); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 100 | return 4; |
| 101 | } |
| 102 | |
Yabin Cui | a029c9a | 2016-02-18 11:32:10 -0800 | [diff] [blame] | 103 | std::string script(script_entry->uncompLen, '\0'); |
| 104 | if (!mzReadZipEntry(&za, script_entry, &script[0], script_entry->uncompLen)) { |
Doug Zongker | fafc85b | 2013-07-09 12:29:45 -0700 | [diff] [blame] | 105 | printf("failed to read script from package\n"); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 106 | return 5; |
| 107 | } |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 108 | |
| 109 | // Configure edify's functions. |
| 110 | |
| 111 | RegisterBuiltins(); |
| 112 | RegisterInstallFunctions(); |
Doug Zongker | bc7ffed | 2014-08-15 14:31:52 -0700 | [diff] [blame] | 113 | RegisterBlockImageFunctions(); |
Doug Zongker | 2b0fdc6 | 2009-06-19 11:22:06 -0700 | [diff] [blame] | 114 | RegisterDeviceExtensions(); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 115 | FinishRegistration(); |
| 116 | |
| 117 | // Parse the script. |
| 118 | |
| 119 | Expr* root; |
Doug Zongker | 8edb00c | 2009-06-11 17:21:44 -0700 | [diff] [blame] | 120 | int error_count = 0; |
Yabin Cui | a029c9a | 2016-02-18 11:32:10 -0800 | [diff] [blame] | 121 | int error = parse_string(script.c_str(), &root, &error_count); |
Doug Zongker | 8edb00c | 2009-06-11 17:21:44 -0700 | [diff] [blame] | 122 | if (error != 0 || error_count > 0) { |
Doug Zongker | fafc85b | 2013-07-09 12:29:45 -0700 | [diff] [blame] | 123 | printf("%d parse errors\n", error_count); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 124 | return 6; |
| 125 | } |
| 126 | |
Stephen Smalley | 779701d | 2012-02-09 14:13:23 -0500 | [diff] [blame] | 127 | struct selinux_opt seopts[] = { |
| 128 | { SELABEL_OPT_PATH, "/file_contexts" } |
| 129 | }; |
| 130 | |
| 131 | sehandle = selabel_open(SELABEL_CTX_FILE, seopts, 1); |
| 132 | |
| 133 | if (!sehandle) { |
Stephen Smalley | 779701d | 2012-02-09 14:13:23 -0500 | [diff] [blame] | 134 | fprintf(cmd_pipe, "ui_print Warning: No file_contexts\n"); |
| 135 | } |
Stephen Smalley | 779701d | 2012-02-09 14:13:23 -0500 | [diff] [blame] | 136 | |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 137 | // Evaluate the parsed script. |
| 138 | |
| 139 | UpdaterInfo updater_info; |
| 140 | updater_info.cmd_pipe = cmd_pipe; |
| 141 | updater_info.package_zip = &za; |
Doug Zongker | e08991e | 2010-02-02 13:09:52 -0800 | [diff] [blame] | 142 | updater_info.version = atoi(version); |
Doug Zongker | bc7ffed | 2014-08-15 14:31:52 -0700 | [diff] [blame] | 143 | updater_info.package_zip_addr = map.addr; |
| 144 | updater_info.package_zip_len = map.length; |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 145 | |
Tao Bao | 876b939 | 2016-10-03 18:06:46 -0700 | [diff] [blame] | 146 | State state(script, &updater_info); |
Doug Zongker | d9c9d10 | 2009-06-12 12:24:39 -0700 | [diff] [blame] | 147 | |
Tianjie Xu | 7ce287d | 2016-05-31 09:29:49 -0700 | [diff] [blame] | 148 | if (argc == 5) { |
| 149 | if (strcmp(argv[4], "retry") == 0) { |
| 150 | state.is_retry = true; |
| 151 | } else { |
| 152 | printf("unexpected argument: %s", argv[4]); |
| 153 | } |
| 154 | } |
| 155 | |
Doug Zongker | d9c9d10 | 2009-06-12 12:24:39 -0700 | [diff] [blame] | 156 | char* result = Evaluate(&state, root); |
Tianjie Xu | 3c62b67 | 2016-02-05 18:25:58 -0800 | [diff] [blame] | 157 | |
| 158 | if (have_eio_error) { |
| 159 | fprintf(cmd_pipe, "retry_update\n"); |
| 160 | } |
| 161 | |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 162 | if (result == NULL) { |
Tao Bao | 876b939 | 2016-10-03 18:06:46 -0700 | [diff] [blame] | 163 | if (state.errmsg.empty()) { |
Doug Zongker | fafc85b | 2013-07-09 12:29:45 -0700 | [diff] [blame] | 164 | printf("script aborted (no error message)\n"); |
Doug Zongker | d9c9d10 | 2009-06-12 12:24:39 -0700 | [diff] [blame] | 165 | fprintf(cmd_pipe, "ui_print script aborted (no error message)\n"); |
| 166 | } else { |
Tao Bao | 876b939 | 2016-10-03 18:06:46 -0700 | [diff] [blame] | 167 | printf("script aborted: %s\n", state.errmsg.c_str()); |
| 168 | const std::vector<std::string> lines = android::base::Split(state.errmsg, "\n"); |
| 169 | for (const std::string& line : lines) { |
Tianjie Xu | 1625583 | 2016-04-30 11:49:59 -0700 | [diff] [blame] | 170 | // Parse the error code in abort message. |
| 171 | // Example: "E30: This package is for bullhead devices." |
Tao Bao | 876b939 | 2016-10-03 18:06:46 -0700 | [diff] [blame] | 172 | if (!line.empty() && line[0] == 'E') { |
| 173 | if (sscanf(line.c_str(), "E%u: ", &state.error_code) != 1) { |
| 174 | printf("Failed to parse error code: [%s]\n", line.c_str()); |
Tianjie Xu | 1625583 | 2016-04-30 11:49:59 -0700 | [diff] [blame] | 175 | } |
| 176 | } |
Tao Bao | 876b939 | 2016-10-03 18:06:46 -0700 | [diff] [blame] | 177 | fprintf(cmd_pipe, "ui_print %s\n", line.c_str()); |
Doug Zongker | d9c9d10 | 2009-06-12 12:24:39 -0700 | [diff] [blame] | 178 | } |
| 179 | fprintf(cmd_pipe, "ui_print\n"); |
| 180 | } |
Tianjie Xu | 1625583 | 2016-04-30 11:49:59 -0700 | [diff] [blame] | 181 | |
| 182 | if (state.error_code != kNoError) { |
| 183 | fprintf(cmd_pipe, "log error: %d\n", state.error_code); |
| 184 | // Cause code should provide additional information about the abort; |
| 185 | // report only when an error exists. |
| 186 | if (state.cause_code != kNoCause) { |
| 187 | fprintf(cmd_pipe, "log cause: %d\n", state.cause_code); |
| 188 | } |
| 189 | } |
| 190 | |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 191 | return 7; |
| 192 | } else { |
Doug Zongker | fafc85b | 2013-07-09 12:29:45 -0700 | [diff] [blame] | 193 | fprintf(cmd_pipe, "ui_print script succeeded: result was [%s]\n", result); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 194 | free(result); |
| 195 | } |
| 196 | |
Doug Zongker | 8e5e4da | 2010-09-14 18:06:55 -0700 | [diff] [blame] | 197 | if (updater_info.package_zip) { |
| 198 | mzCloseZipArchive(updater_info.package_zip); |
| 199 | } |
Doug Zongker | 99916f0 | 2014-01-13 14:16:58 -0800 | [diff] [blame] | 200 | sysReleaseMap(&map); |
Doug Zongker | 9931f7f | 2009-06-10 14:11:53 -0700 | [diff] [blame] | 201 | return 0; |
| 202 | } |