blob: d2a7ac97aa92684e17ea2b19e6d511dd0bafb07e [file] [log] [blame]
Doug Zongker9931f7f2009-06-10 14:11:53 -07001/*
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
17#include <stdio.h>
18#include <unistd.h>
19#include <stdlib.h>
Elliott Hughescd3c55a2015-01-29 20:50:08 -080020#include <string.h>
Doug Zongker9931f7f2009-06-10 14:11:53 -070021
Yabin Cuia029c9a2016-02-18 11:32:10 -080022#include <string>
23
Doug Zongker9931f7f2009-06-10 14:11:53 -070024#include "edify/expr.h"
25#include "updater.h"
26#include "install.h"
Doug Zongkerbc7ffed2014-08-15 14:31:52 -070027#include "blockimg.h"
Doug Zongker9931f7f2009-06-10 14:11:53 -070028#include "minzip/Zip.h"
Doug Zongker99916f02014-01-13 14:16:58 -080029#include "minzip/SysUtil.h"
Jed Estepd9408872015-12-15 16:04:53 -080030#include "config.h"
Doug Zongker9931f7f2009-06-10 14:11:53 -070031
Doug Zongker2b0fdc62009-06-19 11:22:06 -070032// Generated by the makefile, this function defines the
33// RegisterDeviceExtensions() function, which calls all the
34// registration functions for device-specific extensions.
35#include "register.inc"
36
Doug Zongker9931f7f2009-06-10 14:11:53 -070037// Where in the package we expect to find the edify script to execute.
38// (Note it's "updateR-script", not the older "update-script".)
39#define SCRIPT_NAME "META-INF/com/google/android/updater-script"
40
Tianjie Xu3c62b672016-02-05 18:25:58 -080041extern bool have_eio_error;
42
Stephen Smalley779701d2012-02-09 14:13:23 -050043struct selabel_handle *sehandle;
44
Doug Zongker9931f7f2009-06-10 14:11:53 -070045int main(int argc, char** argv) {
Doug Zongker512536a2010-02-17 16:11:44 -080046 // Various things log information to stdout or stderr more or less
Doug Zongkerfafc85b2013-07-09 12:29:45 -070047 // at random (though we've tried to standardize on stdout). The
48 // log file makes more sense if buffering is turned off so things
49 // appear in the right order.
Doug Zongker512536a2010-02-17 16:11:44 -080050 setbuf(stdout, NULL);
51 setbuf(stderr, NULL);
52
Doug Zongker9931f7f2009-06-10 14:11:53 -070053 if (argc != 4) {
Doug Zongkerfafc85b2013-07-09 12:29:45 -070054 printf("unexpected number of arguments (%d)\n", argc);
Doug Zongker9931f7f2009-06-10 14:11:53 -070055 return 1;
56 }
57
58 char* version = argv[1];
Doug Zongkere08991e2010-02-02 13:09:52 -080059 if ((version[0] != '1' && version[0] != '2' && version[0] != '3') ||
60 version[1] != '\0') {
61 // We support version 1, 2, or 3.
Doug Zongkerfafc85b2013-07-09 12:29:45 -070062 printf("wrong updater binary API; expected 1, 2, or 3; "
Doug Zongkere08991e2010-02-02 13:09:52 -080063 "got %s\n",
Doug Zongkerfb2e3af2009-06-17 17:29:40 -070064 argv[1]);
Doug Zongker9931f7f2009-06-10 14:11:53 -070065 return 2;
66 }
67
68 // Set up the pipe for sending commands back to the parent process.
69
70 int fd = atoi(argv[2]);
71 FILE* cmd_pipe = fdopen(fd, "wb");
72 setlinebuf(cmd_pipe);
73
74 // Extract the script from the package.
75
Doug Zongker99916f02014-01-13 14:16:58 -080076 const char* package_filename = argv[3];
77 MemMapping map;
78 if (sysMapFile(package_filename, &map) != 0) {
79 printf("failed to map package %s\n", argv[3]);
80 return 3;
81 }
Doug Zongker9931f7f2009-06-10 14:11:53 -070082 ZipArchive za;
83 int err;
Doug Zongker99916f02014-01-13 14:16:58 -080084 err = mzOpenZipArchive(map.addr, map.length, &za);
Doug Zongker9931f7f2009-06-10 14:11:53 -070085 if (err != 0) {
Doug Zongkerfafc85b2013-07-09 12:29:45 -070086 printf("failed to open package %s: %s\n",
Doug Zongker99916f02014-01-13 14:16:58 -080087 argv[3], strerror(err));
Doug Zongker9931f7f2009-06-10 14:11:53 -070088 return 3;
89 }
Jed Estepd9408872015-12-15 16:04:53 -080090 ota_io_init(&za);
Doug Zongker9931f7f2009-06-10 14:11:53 -070091
92 const ZipEntry* script_entry = mzFindZipEntry(&za, SCRIPT_NAME);
93 if (script_entry == NULL) {
Doug Zongker99916f02014-01-13 14:16:58 -080094 printf("failed to find %s in %s\n", SCRIPT_NAME, package_filename);
Doug Zongker9931f7f2009-06-10 14:11:53 -070095 return 4;
96 }
97
Yabin Cuia029c9a2016-02-18 11:32:10 -080098 std::string script(script_entry->uncompLen, '\0');
99 if (!mzReadZipEntry(&za, script_entry, &script[0], script_entry->uncompLen)) {
Doug Zongkerfafc85b2013-07-09 12:29:45 -0700100 printf("failed to read script from package\n");
Doug Zongker9931f7f2009-06-10 14:11:53 -0700101 return 5;
102 }
Doug Zongker9931f7f2009-06-10 14:11:53 -0700103
104 // Configure edify's functions.
105
106 RegisterBuiltins();
107 RegisterInstallFunctions();
Doug Zongkerbc7ffed2014-08-15 14:31:52 -0700108 RegisterBlockImageFunctions();
Doug Zongker2b0fdc62009-06-19 11:22:06 -0700109 RegisterDeviceExtensions();
Doug Zongker9931f7f2009-06-10 14:11:53 -0700110 FinishRegistration();
111
112 // Parse the script.
113
114 Expr* root;
Doug Zongker8edb00c2009-06-11 17:21:44 -0700115 int error_count = 0;
Yabin Cuia029c9a2016-02-18 11:32:10 -0800116 int error = parse_string(script.c_str(), &root, &error_count);
Doug Zongker8edb00c2009-06-11 17:21:44 -0700117 if (error != 0 || error_count > 0) {
Doug Zongkerfafc85b2013-07-09 12:29:45 -0700118 printf("%d parse errors\n", error_count);
Doug Zongker9931f7f2009-06-10 14:11:53 -0700119 return 6;
120 }
121
Stephen Smalley779701d2012-02-09 14:13:23 -0500122 struct selinux_opt seopts[] = {
123 { SELABEL_OPT_PATH, "/file_contexts" }
124 };
125
126 sehandle = selabel_open(SELABEL_CTX_FILE, seopts, 1);
127
128 if (!sehandle) {
Stephen Smalley779701d2012-02-09 14:13:23 -0500129 fprintf(cmd_pipe, "ui_print Warning: No file_contexts\n");
130 }
Stephen Smalley779701d2012-02-09 14:13:23 -0500131
Doug Zongker9931f7f2009-06-10 14:11:53 -0700132 // Evaluate the parsed script.
133
134 UpdaterInfo updater_info;
135 updater_info.cmd_pipe = cmd_pipe;
136 updater_info.package_zip = &za;
Doug Zongkere08991e2010-02-02 13:09:52 -0800137 updater_info.version = atoi(version);
Doug Zongkerbc7ffed2014-08-15 14:31:52 -0700138 updater_info.package_zip_addr = map.addr;
139 updater_info.package_zip_len = map.length;
Doug Zongker9931f7f2009-06-10 14:11:53 -0700140
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700141 State state;
142 state.cookie = &updater_info;
Yabin Cuia029c9a2016-02-18 11:32:10 -0800143 state.script = &script[0];
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700144 state.errmsg = NULL;
145
146 char* result = Evaluate(&state, root);
Tianjie Xu3c62b672016-02-05 18:25:58 -0800147
148 if (have_eio_error) {
149 fprintf(cmd_pipe, "retry_update\n");
150 }
151
Doug Zongker9931f7f2009-06-10 14:11:53 -0700152 if (result == NULL) {
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700153 if (state.errmsg == NULL) {
Doug Zongkerfafc85b2013-07-09 12:29:45 -0700154 printf("script aborted (no error message)\n");
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700155 fprintf(cmd_pipe, "ui_print script aborted (no error message)\n");
156 } else {
Doug Zongkerfafc85b2013-07-09 12:29:45 -0700157 printf("script aborted: %s\n", state.errmsg);
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700158 char* line = strtok(state.errmsg, "\n");
159 while (line) {
160 fprintf(cmd_pipe, "ui_print %s\n", line);
161 line = strtok(NULL, "\n");
162 }
163 fprintf(cmd_pipe, "ui_print\n");
164 }
165 free(state.errmsg);
Doug Zongker9931f7f2009-06-10 14:11:53 -0700166 return 7;
167 } else {
Doug Zongkerfafc85b2013-07-09 12:29:45 -0700168 fprintf(cmd_pipe, "ui_print script succeeded: result was [%s]\n", result);
Doug Zongker9931f7f2009-06-10 14:11:53 -0700169 free(result);
170 }
171
Doug Zongker8e5e4da2010-09-14 18:06:55 -0700172 if (updater_info.package_zip) {
173 mzCloseZipArchive(updater_info.package_zip);
174 }
Doug Zongker99916f02014-01-13 14:16:58 -0800175 sysReleaseMap(&map);
Doug Zongker9931f7f2009-06-10 14:11:53 -0700176 return 0;
177}