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