Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 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> |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 27 | #include <stdio.h> |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 28 | |
| 29 | #include "ui.h" |
| 30 | #include "cutils/properties.h" |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 31 | #include "adb_install.h" |
Doug Zongker | 18a78e0 | 2014-07-10 07:31:46 -0700 | [diff] [blame] | 32 | #include "minadbd/fuse_adb_provider.h" |
| 33 | #include "fuse_sideload.h" |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 34 | |
| 35 | static RecoveryUI* ui = NULL; |
| 36 | |
Ethan Yonker | 2481342 | 2014-11-07 17:19:07 -0600 | [diff] [blame] | 37 | void |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 38 | set_usb_driver(bool enabled) { |
| 39 | int fd = open("/sys/class/android_usb/android0/enable", O_WRONLY); |
| 40 | if (fd < 0) { |
Dees_Troy | e00c83a | 2012-09-21 10:00:52 -0400 | [diff] [blame] | 41 | /* These error messages show when built in older Android branches (e.g. Gingerbread) |
| 42 | It's not a critical error so we're disabling the error messages. |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 43 | ui->Print("failed to open driver control: %s\n", strerror(errno)); |
Dees_Troy | e00c83a | 2012-09-21 10:00:52 -0400 | [diff] [blame] | 44 | */ |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 45 | printf("failed to open driver control: %s\n", strerror(errno)); |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 46 | return; |
| 47 | } |
Ethan Yonker | c798c9c | 2015-10-09 11:15:26 -0500 | [diff] [blame^] | 48 | |
Elliott Hughes | 2f5feed | 2015-04-28 17:24:24 -0700 | [diff] [blame] | 49 | if (TEMP_FAILURE_RETRY(write(fd, enabled ? "1" : "0", 1)) == -1) { |
Dees_Troy | e00c83a | 2012-09-21 10:00:52 -0400 | [diff] [blame] | 50 | /* |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 51 | ui->Print("failed to set driver control: %s\n", strerror(errno)); |
Dees_Troy | e00c83a | 2012-09-21 10:00:52 -0400 | [diff] [blame] | 52 | */ |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 53 | printf("failed to set driver control: %s\n", strerror(errno)); |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 54 | } |
| 55 | if (close(fd) < 0) { |
Dees_Troy | e00c83a | 2012-09-21 10:00:52 -0400 | [diff] [blame] | 56 | /* |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 57 | ui->Print("failed to close driver control: %s\n", strerror(errno)); |
Dees_Troy | e00c83a | 2012-09-21 10:00:52 -0400 | [diff] [blame] | 58 | */ |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 59 | printf("failed to close driver control: %s\n", strerror(errno)); |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 60 | } |
| 61 | } |
| 62 | |
| 63 | static void |
| 64 | stop_adbd() { |
| 65 | property_set("ctl.stop", "adbd"); |
| 66 | set_usb_driver(false); |
| 67 | } |
| 68 | |
Ethan Yonker | c798c9c | 2015-10-09 11:15:26 -0500 | [diff] [blame^] | 69 | bool is_ro_debuggable() { |
| 70 | char value[PROPERTY_VALUE_MAX+1]; |
| 71 | return (property_get("ro.debuggable", value, NULL) == 1 && value[0] == '1'); |
| 72 | } |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 73 | |
Ethan Yonker | 2481342 | 2014-11-07 17:19:07 -0600 | [diff] [blame] | 74 | void |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 75 | maybe_restart_adbd() { |
| 76 | char value[PROPERTY_VALUE_MAX+1]; |
Elliott Hughes | f14af80 | 2015-02-10 14:46:14 -0800 | [diff] [blame] | 77 | if (is_ro_debuggable()) { |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 78 | printf("Restarting adbd...\n"); |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 79 | set_usb_driver(true); |
| 80 | property_set("ctl.start", "adbd"); |
| 81 | } |
| 82 | } |
| 83 | |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 84 | // How long (in seconds) we wait for the host to start sending us a |
| 85 | // package, before timing out. |
| 86 | #define ADB_INSTALL_TIMEOUT 300 |
| 87 | |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 88 | int |
that | cc8ddca | 2015-01-03 01:59:36 +0100 | [diff] [blame] | 89 | apply_from_adb(const char* install_file, pid_t* child_pid) { |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 90 | |
| 91 | stop_adbd(); |
| 92 | set_usb_driver(true); |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 93 | /* |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 94 | ui->Print("\n\nNow send the package you want to apply\n" |
| 95 | "to the device with \"adb sideload <filename>\"...\n"); |
Dees_Troy | 2673cec | 2013-04-02 20:22:16 +0000 | [diff] [blame] | 96 | */ |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 97 | pid_t child; |
| 98 | if ((child = fork()) == 0) { |
Dees_Troy | 9a4b569 | 2012-09-19 15:09:45 -0400 | [diff] [blame] | 99 | execl("/sbin/recovery", "recovery", "--adbd", install_file, NULL); |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 100 | _exit(-1); |
| 101 | } |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 102 | |
that | cc8ddca | 2015-01-03 01:59:36 +0100 | [diff] [blame] | 103 | *child_pid = child; |
| 104 | // caller can now kill the child thread from another thread |
Ethan Yonker | a167416 | 2014-11-06 08:35:10 -0600 | [diff] [blame] | 105 | |
Doug Zongker | 18a78e0 | 2014-07-10 07:31:46 -0700 | [diff] [blame] | 106 | // FUSE_SIDELOAD_HOST_PATHNAME will start to exist once the host |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 107 | // connects and starts serving a package. Poll for its |
| 108 | // appearance. (Note that inotify doesn't work with FUSE.) |
| 109 | int result; |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 110 | int status; |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 111 | bool waited = false; |
| 112 | struct stat st; |
| 113 | for (int i = 0; i < ADB_INSTALL_TIMEOUT; ++i) { |
| 114 | if (waitpid(child, &status, WNOHANG) != 0) { |
Ethan Yonker | 2686009 | 2014-11-06 09:49:25 -0600 | [diff] [blame] | 115 | result = -1; |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 116 | waited = true; |
| 117 | break; |
| 118 | } |
| 119 | |
Doug Zongker | 18a78e0 | 2014-07-10 07:31:46 -0700 | [diff] [blame] | 120 | if (stat(FUSE_SIDELOAD_HOST_PATHNAME, &st) != 0) { |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 121 | if (errno == ENOENT && i < ADB_INSTALL_TIMEOUT-1) { |
| 122 | sleep(1); |
| 123 | continue; |
| 124 | } else { |
that | cc8ddca | 2015-01-03 01:59:36 +0100 | [diff] [blame] | 125 | printf("\nTimed out waiting for package: %s\n\n", strerror(errno)); |
Ethan Yonker | 2686009 | 2014-11-06 09:49:25 -0600 | [diff] [blame] | 126 | result = -1; |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 127 | kill(child, SIGKILL); |
| 128 | break; |
| 129 | } |
| 130 | } |
Ethan Yonker | 2481342 | 2014-11-07 17:19:07 -0600 | [diff] [blame] | 131 | // Install is handled elsewhere in TWRP |
that | cc8ddca | 2015-01-03 01:59:36 +0100 | [diff] [blame] | 132 | //install_package(FUSE_SIDELOAD_HOST_PATHNAME, wipe_cache, install_file, false); |
| 133 | return 0; |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 134 | } |
| 135 | |
that | cc8ddca | 2015-01-03 01:59:36 +0100 | [diff] [blame] | 136 | // if we got here, something failed |
| 137 | *child_pid = 0; |
| 138 | |
| 139 | if (!waited) { |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 140 | // Calling stat() on this magic filename signals the minadbd |
| 141 | // subprocess to shut down. |
Doug Zongker | 18a78e0 | 2014-07-10 07:31:46 -0700 | [diff] [blame] | 142 | stat(FUSE_SIDELOAD_HOST_EXIT_PATHNAME, &st); |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 143 | |
| 144 | // TODO(dougz): there should be a way to cancel waiting for a |
| 145 | // package (by pushing some button combo on the device). For now |
| 146 | // you just have to 'adb sideload' a file that's not a valid |
| 147 | // package, like "/dev/null". |
| 148 | waitpid(child, &status, 0); |
that | cc8ddca | 2015-01-03 01:59:36 +0100 | [diff] [blame] | 149 | } |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 150 | |
that | cc8ddca | 2015-01-03 01:59:36 +0100 | [diff] [blame] | 151 | if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) { |
| 152 | if (WEXITSTATUS(status) == 3) { |
| 153 | printf("\nYou need adb 1.0.32 or newer to sideload\nto this device.\n\n"); |
| 154 | result = -2; |
| 155 | } else if (!WIFSIGNALED(status)) { |
| 156 | printf("status %d\n", WEXITSTATUS(status)); |
Doug Zongker | 075ad80 | 2014-06-26 15:35:51 -0700 | [diff] [blame] | 157 | } |
that | cc8ddca | 2015-01-03 01:59:36 +0100 | [diff] [blame] | 158 | } |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 159 | |
that | cc8ddca | 2015-01-03 01:59:36 +0100 | [diff] [blame] | 160 | set_usb_driver(false); |
| 161 | maybe_restart_adbd(); |
| 162 | |
| 163 | return result; |
Doug Zongker | 9270a20 | 2012-01-09 15:16:13 -0800 | [diff] [blame] | 164 | } |