blob: 9b1999d90e3cd7a0f4f254600950add42a8e0912 [file] [log] [blame]
Doug Zongker9270a202012-01-09 15:16:13 -08001/*
2 * Copyright (C) 2007 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
xunchang95d67322019-04-05 16:16:07 -070017#include "minadbd_services.h"
18
Doug Zongker9270a202012-01-09 15:16:13 -080019#include <errno.h>
Dan Albert8f6eb5c2015-02-24 22:07:18 -080020#include <inttypes.h>
21#include <stdio.h>
22#include <stdlib.h>
23#include <string.h>
24#include <unistd.h>
Doug Zongker9270a202012-01-09 15:16:13 -080025
Josh Gao038c4a12018-07-27 11:18:30 -070026#include <functional>
xunchang5e6832a2019-03-15 16:04:32 -070027#include <memory>
Tianjie Xu6af51a02017-04-19 14:00:52 -070028#include <string>
Tao Bao5de19e22019-01-02 09:35:59 -080029#include <string_view>
Josh Gao2aa0d3a2017-04-13 12:45:55 -070030#include <thread>
Tao Baoe5c64462019-04-04 18:37:58 -070031#include <unordered_set>
Josh Gao2aa0d3a2017-04-13 12:45:55 -070032
xunchang95d67322019-04-05 16:16:07 -070033#include <android-base/file.h>
34#include <android-base/logging.h>
35#include <android-base/memory.h>
Tao Baoe5c64462019-04-04 18:37:58 -070036#include <android-base/parseint.h>
37#include <android-base/properties.h>
xunchang95d67322019-04-05 16:16:07 -070038#include <android-base/stringprintf.h>
39#include <android-base/strings.h>
40
Doug Zongker9270a202012-01-09 15:16:13 -080041#include "adb.h"
Josh Gao038c4a12018-07-27 11:18:30 -070042#include "adb_unique_fd.h"
Tao Baoe5c64462019-04-04 18:37:58 -070043#include "adb_utils.h"
Dan Albert8f6eb5c2015-02-24 22:07:18 -080044#include "fdevent.h"
45#include "fuse_adb_provider.h"
xunchang5e6832a2019-03-15 16:04:32 -070046#include "fuse_sideload.h"
xunchang95d67322019-04-05 16:16:07 -070047#include "minadbd_types.h"
Josh Gao038c4a12018-07-27 11:18:30 -070048#include "services.h"
Elliott Hughes24eb8a02016-06-15 15:12:17 -070049#include "sysdeps.h"
Doug Zongker9270a202012-01-09 15:16:13 -080050
xunchang95d67322019-04-05 16:16:07 -070051static int minadbd_socket = -1;
Tao Bao378bfbf2019-04-16 14:22:25 -070052static bool rescue_mode = false;
xunchang23f15fc2019-04-17 14:43:58 -070053static std::string sideload_mount_point = FUSE_SIDELOAD_HOST_MOUNTPOINT;
Tao Bao378bfbf2019-04-16 14:22:25 -070054
xunchang95d67322019-04-05 16:16:07 -070055void SetMinadbdSocketFd(int socket_fd) {
56 minadbd_socket = socket_fd;
57}
58
Tao Bao378bfbf2019-04-16 14:22:25 -070059void SetMinadbdRescueMode(bool rescue) {
60 rescue_mode = rescue;
61}
62
xunchang23f15fc2019-04-17 14:43:58 -070063void SetSideloadMountPoint(const std::string& path) {
64 sideload_mount_point = path;
65}
66
Tao Bao7b9b7db2019-04-19 15:22:15 -070067static bool WriteCommandToFd(MinadbdCommand cmd, int fd) {
xunchang95d67322019-04-05 16:16:07 -070068 char message[kMinadbdMessageSize];
69 memcpy(message, kMinadbdCommandPrefix, strlen(kMinadbdStatusPrefix));
70 android::base::put_unaligned(message + strlen(kMinadbdStatusPrefix), cmd);
71
72 if (!android::base::WriteFully(fd, message, kMinadbdMessageSize)) {
73 PLOG(ERROR) << "Failed to write message " << message;
74 return false;
75 }
76 return true;
77}
78
79// Blocks and reads the command status from |fd|. Returns false if the received message has a
80// format error.
81static bool WaitForCommandStatus(int fd, MinadbdCommandStatus* status) {
82 char buffer[kMinadbdMessageSize];
83 if (!android::base::ReadFully(fd, buffer, kMinadbdMessageSize)) {
84 PLOG(ERROR) << "Failed to response status from socket";
85 exit(kMinadbdSocketIOError);
86 }
87
88 std::string message(buffer, buffer + kMinadbdMessageSize);
89 if (!android::base::StartsWith(message, kMinadbdStatusPrefix)) {
90 LOG(ERROR) << "Failed to parse status in " << message;
91 return false;
92 }
93
94 *status = android::base::get_unaligned<MinadbdCommandStatus>(
95 message.substr(strlen(kMinadbdStatusPrefix)).c_str());
96 return true;
97}
98
Tao Baoe5c64462019-04-04 18:37:58 -070099static MinadbdErrorCode RunAdbFuseSideload(int sfd, const std::string& args,
100 MinadbdCommandStatus* status) {
101 auto pieces = android::base::Split(args, ":");
katao77d61732018-09-20 20:34:05 +0800102 int64_t file_size;
103 int block_size;
Tao Baoe5c64462019-04-04 18:37:58 -0700104 if (pieces.size() != 2 || !android::base::ParseInt(pieces[0], &file_size) || file_size <= 0 ||
105 !android::base::ParseInt(pieces[1], &block_size) || block_size <= 0) {
xunchang95d67322019-04-05 16:16:07 -0700106 LOG(ERROR) << "bad sideload-host arguments: " << args;
Tao Baoe5c64462019-04-04 18:37:58 -0700107 return kMinadbdPackageSizeError;
katao77d61732018-09-20 20:34:05 +0800108 }
Doug Zongker9270a202012-01-09 15:16:13 -0800109
xunchang95d67322019-04-05 16:16:07 -0700110 LOG(INFO) << "sideload-host file size " << file_size << ", block size " << block_size;
111
Tao Bao7b9b7db2019-04-19 15:22:15 -0700112 if (!WriteCommandToFd(MinadbdCommand::kInstall, minadbd_socket)) {
Tao Baoe5c64462019-04-04 18:37:58 -0700113 return kMinadbdSocketIOError;
xunchang95d67322019-04-05 16:16:07 -0700114 }
Doug Zongker9270a202012-01-09 15:16:13 -0800115
Tao Bao178cdd42019-04-15 12:45:50 -0700116 auto adb_data_reader = std::make_unique<FuseAdbDataProvider>(sfd, file_size, block_size);
xunchang23f15fc2019-04-17 14:43:58 -0700117 if (int result = run_fuse_sideload(std::move(adb_data_reader), sideload_mount_point.c_str());
118 result != 0) {
xunchang95d67322019-04-05 16:16:07 -0700119 LOG(ERROR) << "Failed to start fuse";
Tao Baoe5c64462019-04-04 18:37:58 -0700120 return kMinadbdFuseStartError;
xunchang95d67322019-04-05 16:16:07 -0700121 }
Doug Zongker9270a202012-01-09 15:16:13 -0800122
Tao Baoe5c64462019-04-04 18:37:58 -0700123 if (!WaitForCommandStatus(minadbd_socket, status)) {
124 return kMinadbdMessageFormatError;
125 }
126
127 // Signal host-side adb to stop. For sideload mode, we always send kSideloadServiceExitSuccess
128 // (i.e. "DONEDONE") regardless of the install result. For rescue mode, we send failure message on
129 // install error.
130 if (!rescue_mode || *status == MinadbdCommandStatus::kSuccess) {
131 if (!android::base::WriteFully(sfd, kSideloadServiceExitSuccess,
132 strlen(kSideloadServiceExitSuccess))) {
133 return kMinadbdHostSocketIOError;
134 }
135 } else {
136 if (!android::base::WriteFully(sfd, kSideloadServiceExitFailure,
137 strlen(kSideloadServiceExitFailure))) {
138 return kMinadbdHostSocketIOError;
139 }
140 }
141
142 return kMinadbdSuccess;
143}
144
145// Sideload service always exits after serving an install command.
146static void SideloadHostService(unique_fd sfd, const std::string& args) {
xunchang95d67322019-04-05 16:16:07 -0700147 MinadbdCommandStatus status;
Tao Baoe5c64462019-04-04 18:37:58 -0700148 exit(RunAdbFuseSideload(sfd.get(), args, &status));
149}
xunchang95d67322019-04-05 16:16:07 -0700150
Tao Baoe5c64462019-04-04 18:37:58 -0700151// Rescue service waits for the next command after an install command.
152static void RescueInstallHostService(unique_fd sfd, const std::string& args) {
153 MinadbdCommandStatus status;
154 if (auto result = RunAdbFuseSideload(sfd.get(), args, &status); result != kMinadbdSuccess) {
155 exit(result);
156 }
157}
158
159static void RescueGetpropHostService(unique_fd sfd, const std::string& prop) {
160 static const std::unordered_set<std::string> kGetpropAllowedProps = {
161 "ro.build.fingerprint",
162 "ro.build.date.utc",
163 };
164 auto allowed = kGetpropAllowedProps.find(prop) != kGetpropAllowedProps.end();
165 if (!allowed) {
166 return;
167 }
168
169 auto result = android::base::GetProperty(prop, "");
170 if (result.empty()) {
171 return;
172 }
173 if (!android::base::WriteFully(sfd, result.data(), result.size())) {
174 exit(kMinadbdHostSocketIOError);
175 }
Doug Zongker9270a202012-01-09 15:16:13 -0800176}
177
Tao Bao7b9b7db2019-04-19 15:22:15 -0700178// Reboots into the given target. We don't reboot directly from minadbd, but going through recovery
179// instead. This allows recovery to finish all the pending works (clear BCB, save logs etc) before
180// the reboot.
181static void RebootHostService(unique_fd /* sfd */, const std::string& target) {
182 MinadbdCommand command;
183 if (target == "bootloader") {
184 command = MinadbdCommand::kRebootBootloader;
185 } else if (target == "rescue") {
186 command = MinadbdCommand::kRebootRescue;
187 } else if (target == "recovery") {
188 command = MinadbdCommand::kRebootRecovery;
189 } else if (target == "fastboot") {
190 command = MinadbdCommand::kRebootFastboot;
191 } else {
192 command = MinadbdCommand::kRebootAndroid;
193 }
194 if (!WriteCommandToFd(command, minadbd_socket)) {
195 exit(kMinadbdSocketIOError);
196 }
197 MinadbdCommandStatus status;
198 if (!WaitForCommandStatus(minadbd_socket, &status)) {
199 exit(kMinadbdMessageFormatError);
200 }
201}
202
Tao Bao5de19e22019-01-02 09:35:59 -0800203unique_fd daemon_service_to_fd(std::string_view name, atransport* /* transport */) {
Tao Bao7b9b7db2019-04-19 15:22:15 -0700204 // Common services that are supported both in sideload and rescue modes.
205 if (ConsumePrefix(&name, "reboot:")) {
206 // "reboot:<target>", where target must be one of the following.
207 std::string args(name);
208 if (args.empty() || args == "bootloader" || args == "rescue" || args == "recovery" ||
209 args == "fastboot") {
210 return create_service_thread("reboot",
211 std::bind(RebootHostService, std::placeholders::_1, args));
212 }
213 return unique_fd{};
214 }
215
216 // Rescue-specific services.
Tao Baoe5c64462019-04-04 18:37:58 -0700217 if (rescue_mode) {
218 if (ConsumePrefix(&name, "rescue-install:")) {
219 // rescue-install:<file-size>:<block-size>
220 std::string args(name);
221 return create_service_thread(
222 "rescue-install", std::bind(RescueInstallHostService, std::placeholders::_1, args));
223 } else if (ConsumePrefix(&name, "rescue-getprop:")) {
224 // rescue-getprop:<prop>
225 std::string args(name);
226 return create_service_thread(
227 "rescue-getprop", std::bind(RescueGetpropHostService, std::placeholders::_1, args));
228 }
229 return unique_fd{};
230 }
231
Tao Bao7b9b7db2019-04-19 15:22:15 -0700232 // Sideload-specific services.
Tao Bao5de19e22019-01-02 09:35:59 -0800233 if (name.starts_with("sideload:")) {
234 // This exit status causes recovery to print a special error message saying to use a newer adb
235 // (that supports sideload-host).
xunchang95d67322019-04-05 16:16:07 -0700236 exit(kMinadbdAdbVersionError);
Tao Baoe5c64462019-04-04 18:37:58 -0700237 } else if (ConsumePrefix(&name, "sideload-host:")) {
238 // sideload-host:<file-size>:<block-size>
239 std::string args(name);
Josh Gao038c4a12018-07-27 11:18:30 -0700240 return create_service_thread("sideload-host",
Tao Baoe5c64462019-04-04 18:37:58 -0700241 std::bind(SideloadHostService, std::placeholders::_1, args));
Tao Bao99f0d9e2016-10-13 12:46:38 -0700242 }
Josh Gao038c4a12018-07-27 11:18:30 -0700243 return unique_fd{};
Doug Zongker9270a202012-01-09 15:16:13 -0800244}