blob: 7aa95e76187534f732410790edc608bb92c05ba2 [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>
Dees_Troy512376c2013-09-03 19:39:41 +000020#include <fcntl.h>
Doug Zongker9931f7f2009-06-10 14:11:53 -070021
22#include "edify/expr.h"
23#include "updater.h"
24#include "install.h"
25#include "minzip/Zip.h"
26
Doug Zongker2b0fdc62009-06-19 11:22:06 -070027// Generated by the makefile, this function defines the
28// RegisterDeviceExtensions() function, which calls all the
29// registration functions for device-specific extensions.
30#include "register.inc"
31
Doug Zongker9931f7f2009-06-10 14:11:53 -070032// Where in the package we expect to find the edify script to execute.
33// (Note it's "updateR-script", not the older "update-script".)
34#define SCRIPT_NAME "META-INF/com/google/android/updater-script"
Dees_Troy512376c2013-09-03 19:39:41 +000035#define SELINUX_CONTEXTS_ZIP "file_contexts"
36#define SELINUX_CONTEXTS_TMP "/tmp/file_contexts"
Doug Zongker9931f7f2009-06-10 14:11:53 -070037
Stephen Smalley779701d2012-02-09 14:13:23 -050038struct selabel_handle *sehandle;
39
Doug Zongker9931f7f2009-06-10 14:11:53 -070040int main(int argc, char** argv) {
Doug Zongker512536a2010-02-17 16:11:44 -080041 // Various things log information to stdout or stderr more or less
42 // at random. The log file makes more sense if buffering is
43 // turned off so things appear in the right order.
44 setbuf(stdout, NULL);
45 setbuf(stderr, NULL);
46
Doug Zongker9931f7f2009-06-10 14:11:53 -070047 if (argc != 4) {
48 fprintf(stderr, "unexpected number of arguments (%d)\n", argc);
49 return 1;
50 }
51
52 char* version = argv[1];
Doug Zongkere08991e2010-02-02 13:09:52 -080053 if ((version[0] != '1' && version[0] != '2' && version[0] != '3') ||
54 version[1] != '\0') {
55 // We support version 1, 2, or 3.
56 fprintf(stderr, "wrong updater binary API; expected 1, 2, or 3; "
57 "got %s\n",
Doug Zongkerfb2e3af2009-06-17 17:29:40 -070058 argv[1]);
Doug Zongker9931f7f2009-06-10 14:11:53 -070059 return 2;
60 }
61
62 // Set up the pipe for sending commands back to the parent process.
63
64 int fd = atoi(argv[2]);
65 FILE* cmd_pipe = fdopen(fd, "wb");
66 setlinebuf(cmd_pipe);
67
68 // Extract the script from the package.
69
70 char* package_data = argv[3];
71 ZipArchive za;
72 int err;
73 err = mzOpenZipArchive(package_data, &za);
74 if (err != 0) {
75 fprintf(stderr, "failed to open package %s: %s\n",
76 package_data, strerror(err));
77 return 3;
78 }
79
80 const ZipEntry* script_entry = mzFindZipEntry(&za, SCRIPT_NAME);
81 if (script_entry == NULL) {
82 fprintf(stderr, "failed to find %s in %s\n", SCRIPT_NAME, package_data);
83 return 4;
84 }
85
86 char* script = malloc(script_entry->uncompLen+1);
87 if (!mzReadZipEntry(&za, script_entry, script, script_entry->uncompLen)) {
88 fprintf(stderr, "failed to read script from package\n");
89 return 5;
90 }
91 script[script_entry->uncompLen] = '\0';
92
Dees_Troy512376c2013-09-03 19:39:41 +000093 const ZipEntry* file_contexts_entry = mzFindZipEntry(&za, SELINUX_CONTEXTS_ZIP);
Dees Troyd29b3362013-10-01 15:21:12 +000094 if (file_contexts_entry != NULL) {
Dees_Troy512376c2013-09-03 19:39:41 +000095 int file_contexts_fd = creat(SELINUX_CONTEXTS_TMP, 0644);
96 if (file_contexts_fd < 0) {
97 fprintf(stderr, "Could not extract %s to '%s'\n", SELINUX_CONTEXTS_ZIP, SELINUX_CONTEXTS_TMP);
98 return 3;
99 }
100
101 int ret_val = mzExtractZipEntryToFile(&za, file_contexts_entry, file_contexts_fd);
102 close(file_contexts_fd);
103
104 if (!ret_val) {
105 fprintf(stderr, "Could not extract '%s'\n", SELINUX_CONTEXTS_ZIP);
106 return 3;
107 }
108 }
109
Doug Zongker9931f7f2009-06-10 14:11:53 -0700110 // Configure edify's functions.
111
112 RegisterBuiltins();
113 RegisterInstallFunctions();
Doug Zongker2b0fdc62009-06-19 11:22:06 -0700114 RegisterDeviceExtensions();
Doug Zongker9931f7f2009-06-10 14:11:53 -0700115 FinishRegistration();
116
117 // Parse the script.
118
119 Expr* root;
Doug Zongker8edb00c2009-06-11 17:21:44 -0700120 int error_count = 0;
Doug Zongker9931f7f2009-06-10 14:11:53 -0700121 yy_scan_string(script);
Doug Zongker8edb00c2009-06-11 17:21:44 -0700122 int error = yyparse(&root, &error_count);
123 if (error != 0 || error_count > 0) {
124 fprintf(stderr, "%d parse errors\n", error_count);
Doug Zongker9931f7f2009-06-10 14:11:53 -0700125 return 6;
126 }
127
Dees_Troy512376c2013-09-03 19:39:41 +0000128 if (access(SELINUX_CONTEXTS_TMP, R_OK) == 0) {
129 struct selinux_opt seopts[] = {
130 { SELABEL_OPT_PATH, SELINUX_CONTEXTS_TMP }
131 };
Stephen Smalley779701d2012-02-09 14:13:23 -0500132
Dees_Troy512376c2013-09-03 19:39:41 +0000133 sehandle = selabel_open(SELABEL_CTX_FILE, seopts, 1);
134 } else {
135 struct selinux_opt seopts[] = {
136 { SELABEL_OPT_PATH, "/file_contexts" }
137 };
138
139 sehandle = selabel_open(SELABEL_CTX_FILE, seopts, 1);
140 }
Stephen Smalley779701d2012-02-09 14:13:23 -0500141
142 if (!sehandle) {
143 fprintf(stderr, "Warning: No file_contexts\n");
144 fprintf(cmd_pipe, "ui_print Warning: No file_contexts\n");
145 }
Stephen Smalley779701d2012-02-09 14:13:23 -0500146
Doug Zongker9931f7f2009-06-10 14:11:53 -0700147 // Evaluate the parsed script.
148
149 UpdaterInfo updater_info;
150 updater_info.cmd_pipe = cmd_pipe;
151 updater_info.package_zip = &za;
Doug Zongkere08991e2010-02-02 13:09:52 -0800152 updater_info.version = atoi(version);
Doug Zongker9931f7f2009-06-10 14:11:53 -0700153
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700154 State state;
155 state.cookie = &updater_info;
156 state.script = script;
157 state.errmsg = NULL;
158
159 char* result = Evaluate(&state, root);
Doug Zongker9931f7f2009-06-10 14:11:53 -0700160 if (result == NULL) {
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700161 if (state.errmsg == NULL) {
162 fprintf(stderr, "script aborted (no error message)\n");
163 fprintf(cmd_pipe, "ui_print script aborted (no error message)\n");
164 } else {
165 fprintf(stderr, "script aborted: %s\n", state.errmsg);
166 char* line = strtok(state.errmsg, "\n");
167 while (line) {
168 fprintf(cmd_pipe, "ui_print %s\n", line);
169 line = strtok(NULL, "\n");
170 }
171 fprintf(cmd_pipe, "ui_print\n");
172 }
173 free(state.errmsg);
Doug Zongker9931f7f2009-06-10 14:11:53 -0700174 return 7;
175 } else {
176 fprintf(stderr, "script result was [%s]\n", result);
177 free(result);
178 }
179
Doug Zongker8e5e4da2010-09-14 18:06:55 -0700180 if (updater_info.package_zip) {
181 mzCloseZipArchive(updater_info.package_zip);
182 }
Doug Zongkerd9c9d102009-06-12 12:24:39 -0700183 free(script);
Doug Zongker9931f7f2009-06-10 14:11:53 -0700184
185 return 0;
186}