services.cpp - LeafOS-Project/android_packages_modules_adb - Gitiles

 /*
  * Copyright (C) 2007 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 #define TRACE_TAG SERVICES

 #include "sysdeps.h"

 #include <errno.h>
 #include <stddef.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>

 #include <cstring>
 #include <thread>

 #include <android-base/stringprintf.h>
 #include <android-base/strings.h>
 #include <cutils/sockets.h>

 #include "adb.h"
 #include "adb_io.h"
 #include "adb_unique_fd.h"
 #include "adb_utils.h"
 #include "adb_wifi.h"
 #include "services.h"
 #include "socket_spec.h"
 #include "sysdeps.h"
 #include "transport.h"

 namespace {

 void service_bootstrap_func(std::string service_name, std::function<void(unique_fd)> func,
                             unique_fd fd) {
     adb_thread_setname(android::base::StringPrintf("%s svc %d", service_name.c_str(), fd.get()));
     func(std::move(fd));
 }

 }  // namespace

 unique_fd create_service_thread(const char* service_name, std::function<void(unique_fd)> func) {
     int s[2];
     if (adb_socketpair(s)) {
         printf("cannot create service socket pair\n");
         return unique_fd();
     }
     D("socketpair: (%d,%d)", s[0], s[1]);

 #if !ADB_HOST
     if (strcmp(service_name, "sync") == 0) {
         // Set file sync service socket to maximum size
         int max_buf = LINUX_MAX_SOCKET_SIZE;
         adb_setsockopt(s[0], SOL_SOCKET, SO_SNDBUF, &max_buf, sizeof(max_buf));
         adb_setsockopt(s[1], SOL_SOCKET, SO_SNDBUF, &max_buf, sizeof(max_buf));
     }
 #endif  // !ADB_HOST

     std::thread(service_bootstrap_func, service_name, func, unique_fd(s[1])).detach();

     D("service thread started, %d:%d", s[0], s[1]);
     return unique_fd(s[0]);
 }

 unique_fd service_to_fd(std::string_view name, atransport* transport) {
     unique_fd ret;

     if (is_socket_spec(name)) {
         std::string error;
         if (!socket_spec_connect(&ret, name, nullptr, nullptr, &error)) {
             LOG(ERROR) << "failed to connect to socket '" << name << "': " << error;
         }
     } else {
 #if !ADB_HOST
         ret = daemon_service_to_fd(name, transport);
 #endif
     }

     if (ret >= 0) {
         close_on_exec(ret.get());
     }
     return ret;
 }

 #if ADB_HOST
 void connect_emulator(const std::string& port_spec, std::string* response) {
     std::vector<std::string> pieces = android::base::Split(port_spec, ",");
     if (pieces.size() != 2) {
         *response = android::base::StringPrintf("unable to parse '%s' as <console port>,<adb port>",
                                                 port_spec.c_str());
         return;
     }

     int console_port = strtol(pieces[0].c_str(), nullptr, 0);
     int adb_port = strtol(pieces[1].c_str(), nullptr, 0);
     if (console_port <= 0 || adb_port <= 0) {
         *response = android::base::StringPrintf("Invalid port numbers: %s", port_spec.c_str());
         return;
     }

     // Check if the emulator is already known.
     // Note: There's a small but harmless race condition here: An emulator not
     // present just yet could be registered by another invocation right
     // after doing this check here. However, local_connect protects
     // against double-registration too. From here, a better error message
     // can be produced. In the case of the race condition, the very specific
     // error message won't be shown, but the data doesn't get corrupted.
     atransport* known_emulator = find_emulator_transport_by_adb_port(adb_port);
     if (known_emulator != nullptr) {
         *response = android::base::StringPrintf("Emulator already registered on port %d", adb_port);
         return;
     }

     // Preconditions met, try to connect to the emulator.
     std::string error;
     if (!local_connect_arbitrary_ports(console_port, adb_port, &error)) {
         *response = android::base::StringPrintf("Connected to emulator on ports %d,%d",
                                                 console_port, adb_port);
     } else {
         *response = android::base::StringPrintf("Could not connect to emulator on ports %d,%d: %s",
                                                 console_port, adb_port, error.c_str());
     }
 }

 static void connect_service(unique_fd fd, std::string host) {
     std::string response;
     if (!strncmp(host.c_str(), "emu:", 4)) {
         connect_emulator(host.c_str() + 4, &response);
     } else {
         connect_device(host, &response);
     }

     // Send response for emulator and device
     SendProtocolString(fd.get(), response);
 }

 static void pair_service(unique_fd fd, std::string host, std::string password) {
     std::string response;
     adb_wifi_pair_device(host, password, response);
     if (android::base::StartsWith(response, "Successful")) {
         SendProtocolString(fd.get(), response);
     } else {
         SendFail(fd, response);
     }
 }

 static void wait_service(unique_fd fd, std::string serial, TransportId transport_id,
                          std::string spec) {
     std::vector<std::string> components = android::base::Split(spec, "-");
     if (components.size() < 2) {
         SendFail(fd, "short wait-for-: " + spec);
         return;
     }

     TransportType transport_type;
     if (components[0] == "local") {
         transport_type = kTransportLocal;
     } else if (components[0] == "usb") {
         transport_type = kTransportUsb;
     } else if (components[0] == "any") {
         transport_type = kTransportAny;
     } else {
         SendFail(fd, "bad wait-for- transport: " + spec);
         return;
     }

     std::vector<ConnectionState> states;
     for (size_t i = 1; i < components.size(); ++i) {
         if (components[i] == "device") {
             states.push_back(kCsDevice);
         } else if (components[i] == "recovery") {
             states.push_back(kCsRecovery);
         } else if (components[i] == "rescue") {
             states.push_back(kCsRescue);
         } else if (components[i] == "sideload") {
             states.push_back(kCsSideload);
         } else if (components[i] == "bootloader") {
             states.push_back(kCsBootloader);
         } else if (components[i] == "any") {
             states.push_back(kCsAny);
         } else if (components[i] == "disconnect") {
             states.push_back(kCsOffline);
         } else {
             SendFail(fd, "bad wait-for- state: " + spec);
             return;
         }
     }

     while (true) {
         bool is_ambiguous = false;
         std::string error = "unknown error";
         atransport* t =
                 acquire_one_transport(transport_type, !serial.empty() ? serial.c_str() : nullptr,
                                       transport_id, &is_ambiguous, &error);

         // If the target transport disconnect (to wait-for) is unclear, punt
         // to the user for corrective action (e.g. specify `adb -s <>
         // wait-for-disconnect`).
         if (is_ambiguous) {
             SendFail(fd, error);
             return;
         }
         for (const auto& state : states) {
             if (state == kCsOffline) {
                 // Special case for wait-for-disconnect:
                 // We want to wait for USB devices to completely disappear, but
                 // TCP devices can go into the offline state, since we
                 // automatically reconnect (disabling WiFi or disconnecting
                 // from the WiFi connection would trigger transition from
                 // `device` to `offline`).
                 // If the transport is torn down (regardless of whether it is
                 // USB or wireless), unblock.
                 if (!t) {
                     SendOkay(fd);
                     return;
                 }
             } else {
                 if (t && (state == kCsAny || state == t->GetConnectionState())) {
                     SendOkay(fd);
                     return;
                 }
             }
         }

         // Sleep before retrying.
         adb_pollfd pfd = {.fd = fd.get(), .events = POLLIN};
         if (adb_poll(&pfd, 1, 100) != 0) {
             // The other end of the socket is closed, probably because the
             // client terminated. Bail out.
             SendFail(fd, error);
             return;
         }
     }
 }
 #endif

 #if ADB_HOST
 asocket* host_service_to_socket(std::string_view name, std::string_view serial,
                                 TransportId transport_id) {
     if (name == "track-devices") {
         return create_device_tracker(SHORT_TEXT);
     } else if (name == "track-devices-l") {
         return create_device_tracker(LONG_TEXT);
     } else if (name == "track-devices-proto-binary") {
         return create_device_tracker(PROTOBUF);
     } else if (name == "track-devices-proto-text") {
         return create_device_tracker(TEXT_PROTOBUF);
     } else if (android::base::ConsumePrefix(&name, "wait-for-")) {
         std::string spec(name);
         unique_fd fd =
                 create_service_thread("wait", std::bind(wait_service, std::placeholders::_1,
                                                         std::string(serial), transport_id, spec));
         return create_local_socket(std::move(fd));
     } else if (android::base::ConsumePrefix(&name, "connect:")) {
         std::string host(name);
         unique_fd fd = create_service_thread(
                 "connect", std::bind(connect_service, std::placeholders::_1, host));
         return create_local_socket(std::move(fd));
     } else if (android::base::ConsumePrefix(&name, "pair:")) {
         const char* divider = strchr(name.data(), ':');
         if (!divider) {
             return nullptr;
         }
         std::string password(name.data(), divider);
         std::string host(divider + 1);
         unique_fd fd = create_service_thread(
                 "pair", std::bind(pair_service, std::placeholders::_1, host, password));
         return create_local_socket(std::move(fd));
     }
     return nullptr;
 }
 #endif /* ADB_HOST */
	/*
	* Copyright (C) 2007 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	#define TRACE_TAG SERVICES

	#include "sysdeps.h"

	#include <errno.h>
	#include <stddef.h>
	#include <stdio.h>
	#include <stdlib.h>
	#include <string.h>

	#include <cstring>
	#include <thread>

	#include <android-base/stringprintf.h>
	#include <android-base/strings.h>
	#include <cutils/sockets.h>

	#include "adb.h"
	#include "adb_io.h"
	#include "adb_unique_fd.h"
	#include "adb_utils.h"
	#include "adb_wifi.h"
	#include "services.h"
	#include "socket_spec.h"
	#include "sysdeps.h"
	#include "transport.h"

	namespace {

	void service_bootstrap_func(std::string service_name, std::function<void(unique_fd)> func,
	unique_fd fd) {
	adb_thread_setname(android::base::StringPrintf("%s svc %d", service_name.c_str(), fd.get()));
	func(std::move(fd));
	}

	} // namespace

	unique_fd create_service_thread(const char* service_name, std::function<void(unique_fd)> func) {
	int s[2];
	if (adb_socketpair(s)) {
	printf("cannot create service socket pair\n");
	return unique_fd();
	}
	D("socketpair: (%d,%d)", s[0], s[1]);

	#if !ADB_HOST
	if (strcmp(service_name, "sync") == 0) {
	// Set file sync service socket to maximum size
	int max_buf = LINUX_MAX_SOCKET_SIZE;
	adb_setsockopt(s[0], SOL_SOCKET, SO_SNDBUF, &max_buf, sizeof(max_buf));
	adb_setsockopt(s[1], SOL_SOCKET, SO_SNDBUF, &max_buf, sizeof(max_buf));
	}
	#endif // !ADB_HOST

	std::thread(service_bootstrap_func, service_name, func, unique_fd(s[1])).detach();

	D("service thread started, %d:%d", s[0], s[1]);
	return unique_fd(s[0]);
	}

	unique_fd service_to_fd(std::string_view name, atransport* transport) {
	unique_fd ret;

	if (is_socket_spec(name)) {
	std::string error;
	if (!socket_spec_connect(&ret, name, nullptr, nullptr, &error)) {
	LOG(ERROR) << "failed to connect to socket '" << name << "': " << error;
	}
	} else {
	#if !ADB_HOST
	ret = daemon_service_to_fd(name, transport);
	#endif
	}

	if (ret >= 0) {
	close_on_exec(ret.get());
	}
	return ret;
	}

	#if ADB_HOST
	void connect_emulator(const std::string& port_spec, std::string* response) {
	std::vector<std::string> pieces = android::base::Split(port_spec, ",");
	if (pieces.size() != 2) {
	*response = android::base::StringPrintf("unable to parse '%s' as <console port>,<adb port>",
	port_spec.c_str());
	return;
	}

	int console_port = strtol(pieces[0].c_str(), nullptr, 0);
	int adb_port = strtol(pieces[1].c_str(), nullptr, 0);
	if (console_port <= 0 \|\| adb_port <= 0) {
	*response = android::base::StringPrintf("Invalid port numbers: %s", port_spec.c_str());
	return;
	}

	// Check if the emulator is already known.
	// Note: There's a small but harmless race condition here: An emulator not
	// present just yet could be registered by another invocation right
	// after doing this check here. However, local_connect protects
	// against double-registration too. From here, a better error message
	// can be produced. In the case of the race condition, the very specific
	// error message won't be shown, but the data doesn't get corrupted.
	atransport* known_emulator = find_emulator_transport_by_adb_port(adb_port);
	if (known_emulator != nullptr) {
	*response = android::base::StringPrintf("Emulator already registered on port %d", adb_port);
	return;
	}

	// Preconditions met, try to connect to the emulator.
	std::string error;
	if (!local_connect_arbitrary_ports(console_port, adb_port, &error)) {
	*response = android::base::StringPrintf("Connected to emulator on ports %d,%d",
	console_port, adb_port);
	} else {
	*response = android::base::StringPrintf("Could not connect to emulator on ports %d,%d: %s",
	console_port, adb_port, error.c_str());
	}
	}

	static void connect_service(unique_fd fd, std::string host) {
	std::string response;
	if (!strncmp(host.c_str(), "emu:", 4)) {
	connect_emulator(host.c_str() + 4, &response);
	} else {
	connect_device(host, &response);
	}

	// Send response for emulator and device
	SendProtocolString(fd.get(), response);
	}

	static void pair_service(unique_fd fd, std::string host, std::string password) {
	std::string response;
	adb_wifi_pair_device(host, password, response);
	if (android::base::StartsWith(response, "Successful")) {
	SendProtocolString(fd.get(), response);
	} else {
	SendFail(fd, response);
	}
	}

	static void wait_service(unique_fd fd, std::string serial, TransportId transport_id,
	std::string spec) {
	std::vector<std::string> components = android::base::Split(spec, "-");
	if (components.size() < 2) {
	SendFail(fd, "short wait-for-: " + spec);
	return;
	}

	TransportType transport_type;
	if (components[0] == "local") {
	transport_type = kTransportLocal;
	} else if (components[0] == "usb") {
	transport_type = kTransportUsb;
	} else if (components[0] == "any") {
	transport_type = kTransportAny;
	} else {
	SendFail(fd, "bad wait-for- transport: " + spec);
	return;
	}

	std::vector<ConnectionState> states;
	for (size_t i = 1; i < components.size(); ++i) {
	if (components[i] == "device") {
	states.push_back(kCsDevice);
	} else if (components[i] == "recovery") {
	states.push_back(kCsRecovery);
	} else if (components[i] == "rescue") {
	states.push_back(kCsRescue);
	} else if (components[i] == "sideload") {
	states.push_back(kCsSideload);
	} else if (components[i] == "bootloader") {
	states.push_back(kCsBootloader);
	} else if (components[i] == "any") {
	states.push_back(kCsAny);
	} else if (components[i] == "disconnect") {
	states.push_back(kCsOffline);
	} else {
	SendFail(fd, "bad wait-for- state: " + spec);
	return;
	}
	}

	while (true) {
	bool is_ambiguous = false;
	std::string error = "unknown error";
	atransport* t =
	acquire_one_transport(transport_type, !serial.empty() ? serial.c_str() : nullptr,
	transport_id, &is_ambiguous, &error);

	// If the target transport disconnect (to wait-for) is unclear, punt
	// to the user for corrective action (e.g. specify `adb -s <>
	// wait-for-disconnect`).
	if (is_ambiguous) {
	SendFail(fd, error);
	return;
	}
	for (const auto& state : states) {
	if (state == kCsOffline) {
	// Special case for wait-for-disconnect:
	// We want to wait for USB devices to completely disappear, but
	// TCP devices can go into the offline state, since we
	// automatically reconnect (disabling WiFi or disconnecting
	// from the WiFi connection would trigger transition from
	// `device` to `offline`).
	// If the transport is torn down (regardless of whether it is
	// USB or wireless), unblock.
	if (!t) {
	SendOkay(fd);
	return;
	}
	} else {
	if (t && (state == kCsAny \|\| state == t->GetConnectionState())) {
	SendOkay(fd);
	return;
	}
	}
	}

	// Sleep before retrying.
	adb_pollfd pfd = {.fd = fd.get(), .events = POLLIN};
	if (adb_poll(&pfd, 1, 100) != 0) {
	// The other end of the socket is closed, probably because the
	// client terminated. Bail out.
	SendFail(fd, error);
	return;
	}
	}
	}
	#endif

	#if ADB_HOST
	asocket* host_service_to_socket(std::string_view name, std::string_view serial,
	TransportId transport_id) {
	if (name == "track-devices") {
	return create_device_tracker(SHORT_TEXT);
	} else if (name == "track-devices-l") {
	return create_device_tracker(LONG_TEXT);
	} else if (name == "track-devices-proto-binary") {
	return create_device_tracker(PROTOBUF);
	} else if (name == "track-devices-proto-text") {
	return create_device_tracker(TEXT_PROTOBUF);
	} else if (android::base::ConsumePrefix(&name, "wait-for-")) {
	std::string spec(name);
	unique_fd fd =
	create_service_thread("wait", std::bind(wait_service, std::placeholders::_1,
	std::string(serial), transport_id, spec));
	return create_local_socket(std::move(fd));
	} else if (android::base::ConsumePrefix(&name, "connect:")) {
	std::string host(name);
	unique_fd fd = create_service_thread(
	"connect", std::bind(connect_service, std::placeholders::_1, host));
	return create_local_socket(std::move(fd));
	} else if (android::base::ConsumePrefix(&name, "pair:")) {
	const char* divider = strchr(name.data(), ':');
	if (!divider) {
	return nullptr;
	}
	std::string password(name.data(), divider);
	std::string host(divider + 1);
	unique_fd fd = create_service_thread(
	"pair", std::bind(pair_service, std::placeholders::_1, host, password));
	return create_local_socket(std::move(fd));
	}
	return nullptr;
	}
	#endif /* ADB_HOST */