1 /*
2 * Copyright (C) 2012 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 <unistd.h>
18 #include <dirent.h>
19 #include <errno.h>
20 #include <stdlib.h>
21 #include <string.h>
22 #include <sys/types.h>
23 #include <sys/wait.h>
24 #include <sys/stat.h>
25 #include <signal.h>
26 #include <fcntl.h>
27
28 #include "ui.h"
29 #include "install.h"
30 #include "common.h"
31 #include "adb_install.h"
32 #include "minadbd/fuse_adb_provider.h"
33 #include "fuse_sideload.h"
34
35 #include <android-base/properties.h>
36
set_usb_driver(RecoveryUI * ui,bool enabled)37 static void set_usb_driver(RecoveryUI* ui, bool enabled) {
38 int fd = open("/sys/class/android_usb/android0/enable", O_WRONLY);
39 if (fd < 0) {
40 ui->Print("failed to open driver control: %s\n", strerror(errno));
41 return;
42 }
43 if (TEMP_FAILURE_RETRY(write(fd, enabled ? "1" : "0", 1)) == -1) {
44 ui->Print("failed to set driver control: %s\n", strerror(errno));
45 }
46 if (close(fd) < 0) {
47 ui->Print("failed to close driver control: %s\n", strerror(errno));
48 }
49 }
50
stop_adbd(RecoveryUI * ui)51 static void stop_adbd(RecoveryUI* ui) {
52 ui->Print("Stopping adbd...\n");
53 android::base::SetProperty("ctl.stop", "adbd");
54 set_usb_driver(ui, false);
55 }
56
maybe_restart_adbd(RecoveryUI * ui)57 static void maybe_restart_adbd(RecoveryUI* ui) {
58 if (is_ro_debuggable()) {
59 ui->Print("Restarting adbd...\n");
60 set_usb_driver(ui, true);
61 android::base::SetProperty("ctl.start", "adbd");
62 }
63 }
64
65 // How long (in seconds) we wait for the host to start sending us a
66 // package, before timing out.
67 #define ADB_INSTALL_TIMEOUT 300
68
apply_from_adb(RecoveryUI * ui,bool * wipe_cache,const char * install_file)69 int apply_from_adb(RecoveryUI* ui, bool* wipe_cache, const char* install_file) {
70 modified_flash = true;
71
72 stop_adbd(ui);
73 set_usb_driver(ui, true);
74
75 ui->Print("\n\nNow send the package you want to apply\n"
76 "to the device with \"adb sideload <filename>\"...\n");
77
78 pid_t child;
79 if ((child = fork()) == 0) {
80 execl("/sbin/recovery", "recovery", "--adbd", NULL);
81 _exit(EXIT_FAILURE);
82 }
83
84 // FUSE_SIDELOAD_HOST_PATHNAME will start to exist once the host
85 // connects and starts serving a package. Poll for its
86 // appearance. (Note that inotify doesn't work with FUSE.)
87 int result = INSTALL_ERROR;
88 int status;
89 bool waited = false;
90 struct stat st;
91 for (int i = 0; i < ADB_INSTALL_TIMEOUT; ++i) {
92 if (waitpid(child, &status, WNOHANG) != 0) {
93 result = INSTALL_ERROR;
94 waited = true;
95 break;
96 }
97
98 if (stat(FUSE_SIDELOAD_HOST_PATHNAME, &st) != 0) {
99 if (errno == ENOENT && i < ADB_INSTALL_TIMEOUT-1) {
100 sleep(1);
101 continue;
102 } else {
103 ui->Print("\nTimed out waiting for package.\n\n");
104 result = INSTALL_ERROR;
105 kill(child, SIGKILL);
106 break;
107 }
108 }
109 result = install_package(FUSE_SIDELOAD_HOST_PATHNAME, wipe_cache, install_file, false, 0);
110 break;
111 }
112
113 if (!waited) {
114 // Calling stat() on this magic filename signals the minadbd
115 // subprocess to shut down.
116 stat(FUSE_SIDELOAD_HOST_EXIT_PATHNAME, &st);
117
118 // TODO(dougz): there should be a way to cancel waiting for a
119 // package (by pushing some button combo on the device). For now
120 // you just have to 'adb sideload' a file that's not a valid
121 // package, like "/dev/null".
122 waitpid(child, &status, 0);
123 }
124
125 if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) {
126 if (WEXITSTATUS(status) == 3) {
127 ui->Print("\nYou need adb 1.0.32 or newer to sideload\nto this device.\n\n");
128 } else if (!WIFSIGNALED(status)) {
129 ui->Print("\n(adbd status %d)\n", WEXITSTATUS(status));
130 }
131 }
132
133 set_usb_driver(ui, false);
134 maybe_restart_adbd(ui);
135
136 return result;
137 }
138