adb.cpp revision 5200c6670f041550c23821fec8e8e49b30ef6d29
1dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project/*
2dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * Copyright (C) 2007 The Android Open Source Project
3dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project *
4dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * Licensed under the Apache License, Version 2.0 (the "License");
5dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * you may not use this file except in compliance with the License.
6dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * You may obtain a copy of the License at
7dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project *
8dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project *      http://www.apache.org/licenses/LICENSE-2.0
9dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project *
10dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * Unless required by applicable law or agreed to in writing, software
11dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * distributed under the License is distributed on an "AS IS" BASIS,
12dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * See the License for the specific language governing permissions and
14dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * limitations under the License.
15dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project */
16dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
173313426fad9eaaf53017cdbde889ebcec91358ecDan Albert#define TRACE_TAG TRACE_ADB
183313426fad9eaaf53017cdbde889ebcec91358ecDan Albert
193313426fad9eaaf53017cdbde889ebcec91358ecDan Albert#include "sysdeps.h"
203313426fad9eaaf53017cdbde889ebcec91358ecDan Albert#include "adb.h"
21dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
22dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <ctype.h>
23dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <errno.h>
24ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#include <stdarg.h>
25c7993af64baec271a238646bc20aaa846866c4a9Scott Anderson#include <stddef.h>
26ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#include <stdint.h>
27ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#include <stdio.h>
28ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#include <stdlib.h>
29dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <string.h>
301f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood#include <sys/time.h>
31ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#include <time.h>
32ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert
33ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#include <string>
34e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert#include <vector>
35e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert#include <unordered_map>
36dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
379313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert#include <base/logging.h>
387b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes#include <base/stringprintf.h>
398d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes#include <base/strings.h>
407b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes
41d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby#include "adb_auth.h"
42cc731cc76786b6bdc58764aad9924c0d0c8d645fDan Albert#include "adb_io.h"
43e9fca14c9e555a9d02e1d94ba6d3b290673ea26bDan Albert#include "adb_listeners.h"
443d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes#include "adb_utils.h"
457664901a355b959f312e9acff5a0fd31b7139623Dan Albert#include "transport.h"
46dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
47e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))
48e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
49dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if !ADB_HOST
50893a4a47e8c9290128b9254af0246e36f821c260Nick Kralevich#include <cutils/properties.h>
51e2864bf727d6a404bd871f063c03e36cc8eedeecNick Kralevich#include <sys/capability.h>
52885342a0f2c834a6b680284047c47c9d04b32565Jeff Sharkey#include <sys/mount.h>
53dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
54dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
559313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan AlbertADB_MUTEX_DEFINE(D_lock);
56dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
57dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectint HOST = 0;
58dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
59e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#if !ADB_HOST
609313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertconst char* adb_device_banner = "device";
619313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertstatic android::base::LogdLogger gLogdLogger;
62e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#endif
63dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
649313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertvoid AdbLogger(android::base::LogId id, android::base::LogSeverity severity,
659313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert               const char* tag, const char* file, unsigned int line,
669313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert               const char* message) {
679313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert    android::base::StderrLogger(id, severity, tag, file, line, message);
689313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert#if !ADB_HOST
699313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert    gLogdLogger(id, severity, tag, file, line, message);
709313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert#endif
719313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert}
729313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert
739313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertvoid fatal(const char *fmt, ...) {
74dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_list ap;
75dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_start(ap, fmt);
76dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "error: ");
77dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    vfprintf(stderr, fmt, ap);
78dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "\n");
79dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_end(ap);
80dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    exit(-1);
81dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
82dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
839313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertvoid fatal_errno(const char* fmt, ...) {
84dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_list ap;
85dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_start(ap, fmt);
86dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "error: %s: ", strerror(errno));
87dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    vfprintf(stderr, fmt, ap);
88dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "\n");
89dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_end(ap);
90dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    exit(-1);
91dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
92dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
93ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#if !ADB_HOST
949313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertstatic std::string get_log_file_name() {
95ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    struct tm now;
96ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    time_t t;
97ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    tzset();
98ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    time(&t);
99ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    localtime_r(&t, &now);
100ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert
1018743ef98414b336f222327253f2cde6bf6aee386Dan Albert    char timestamp[PATH_MAX];
1028743ef98414b336f222327253f2cde6bf6aee386Dan Albert    strftime(timestamp, sizeof(timestamp), "%Y-%m-%d-%H-%M-%S", &now);
1038743ef98414b336f222327253f2cde6bf6aee386Dan Albert
1049313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert    return android::base::StringPrintf("/data/adb/adb-%s-%d", timestamp,
1059313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert                                       getpid());
1069313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert}
1079313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert
1089313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertvoid start_device_log(void) {
1099313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert    int fd = unix_open(get_log_file_name().c_str(),
1109313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert                       O_WRONLY | O_CREAT | O_TRUNC | O_CLOEXEC, 0640);
111ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    if (fd == -1) {
112ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert        return;
113ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    }
114ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert
1159313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert    // Redirect stdout and stderr to the log file.
116ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    dup2(fd, STDOUT_FILENO);
117ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    dup2(fd, STDERR_FILENO);
118ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    fprintf(stderr, "--- adb starting (pid %d) ---\n", getpid());
1196ac5d7dc56a016718d2a26803435505e983e5d80Spencer Low    unix_close(fd);
120ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert}
121ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#endif
122ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert
123ea2175ab6bae74409620374a96fc9a0097a1845cDan Albertint adb_trace_mask;
124ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert
125ea2175ab6bae74409620374a96fc9a0097a1845cDan Albertstd::string get_trace_setting_from_env() {
126ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    const char* setting = getenv("ADB_TRACE");
127ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    if (setting == nullptr) {
128ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert        setting = "";
129ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    }
130ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert
131ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    return std::string(setting);
132ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert}
133ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert
134ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#if !ADB_HOST
135ea2175ab6bae74409620374a96fc9a0097a1845cDan Albertstd::string get_trace_setting_from_prop() {
136ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    char buf[PROPERTY_VALUE_MAX];
137ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    property_get("persist.adb.trace_mask", buf, "");
138ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    return std::string(buf);
139ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert}
140ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#endif
141ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert
142ea2175ab6bae74409620374a96fc9a0097a1845cDan Albertstd::string get_trace_setting() {
143ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#if ADB_HOST
144ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    return get_trace_setting_from_env();
145ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#else
146ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    return get_trace_setting_from_prop();
147ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#endif
148ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert}
149ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert
150e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert// Split the space separated list of tags from the trace setting and build the
151e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert// trace mask from it. note that '1' and 'all' are special cases to enable all
152e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert// tracing.
153ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert//
154ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert// adb's trace setting comes from the ADB_TRACE environment variable, whereas
155ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert// adbd's comes from the system property persist.adb.trace_mask.
1569313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertstatic void setup_trace_mask() {
157ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert    const std::string trace_setting = get_trace_setting();
158dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
159e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert    std::unordered_map<std::string, int> trace_flags = {
160e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"1", 0},
161e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"all", 0},
162e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"adb", TRACE_ADB},
163e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"sockets", TRACE_SOCKETS},
164e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"packets", TRACE_PACKETS},
165e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"rwx", TRACE_RWX},
166e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"usb", TRACE_USB},
167e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"sync", TRACE_SYNC},
168e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"sysdeps", TRACE_SYSDEPS},
169e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"transport", TRACE_TRANSPORT},
170e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"jdwp", TRACE_JDWP},
171e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"services", TRACE_SERVICES},
172e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        {"auth", TRACE_AUTH}};
173e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert
174e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert    std::vector<std::string> elements = android::base::Split(trace_setting, " ");
175e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert    for (const auto& elem : elements) {
176e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        const auto& flag = trace_flags.find(elem);
177e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        if (flag == trace_flags.end()) {
1788d8126a705dd3c5734a0894f88c2c758784bd469Spencer Low            D("Unknown trace flag: %s\n", flag->first.c_str());
179e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert            continue;
180dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
181e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert
182e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        if (flag->second == 0) {
183e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert            // 0 is used for the special values "1" and "all" that enable all
184e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert            // tracing.
185e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert            adb_trace_mask = ~0;
186e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert            return;
187e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert        } else {
188e246219d0c9897b1c08204d6c535d5d341c4e7eeDan Albert            adb_trace_mask |= 1 << flag->second;
189dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
190dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1919313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert}
192ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert
1939313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertvoid adb_trace_init(char** argv) {
1946f009d9496fc060ba76f97348aa0702a62fcf927Paul Lawrence    // Don't open log file if no tracing, since this will block
1956f009d9496fc060ba76f97348aa0702a62fcf927Paul Lawrence    // the crypto unmount of /data
1966f009d9496fc060ba76f97348aa0702a62fcf927Paul Lawrence    const std::string trace_setting = get_trace_setting();
1976f009d9496fc060ba76f97348aa0702a62fcf927Paul Lawrence    if (trace_setting.empty()) {
1986f009d9496fc060ba76f97348aa0702a62fcf927Paul Lawrence        return;
1996f009d9496fc060ba76f97348aa0702a62fcf927Paul Lawrence    }
2006f009d9496fc060ba76f97348aa0702a62fcf927Paul Lawrence
201ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#if !ADB_HOST
2029313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert    if (isatty(STDOUT_FILENO) == 0) {
2039313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert        start_device_log();
2049313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert    }
205ea2175ab6bae74409620374a96fc9a0097a1845cDan Albert#endif
2069313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert
2079313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert    setup_trace_mask();
2089313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert    android::base::InitLogging(argv, AdbLogger);
209dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
210dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
211bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albertapacket* get_apacket(void)
212dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
213bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albert    apacket* p = reinterpret_cast<apacket*>(malloc(sizeof(apacket)));
214bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albert    if (p == nullptr) {
215bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albert      fatal("failed to allocate an apacket");
216bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albert    }
217bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albert
218dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    memset(p, 0, sizeof(apacket) - MAX_PAYLOAD);
219dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return p;
220dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
221dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
222dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid put_apacket(apacket *p)
223dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
224dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    free(p);
225dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
226dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
227d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Gobyvoid handle_online(atransport *t)
228dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
229dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("adb: online\n");
230d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    t->online = 1;
231dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
232dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
233dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid handle_offline(atransport *t)
234dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
235dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("adb: offline\n");
236dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    //Close the associated usb
237d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    t->online = 0;
238dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    run_transport_disconnects(t);
239dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
240dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
241d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby#if DEBUG_PACKETS
242dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#define DUMPMAX 32
243dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid print_packet(const char *label, apacket *p)
244dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
245dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char *tag;
246dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char *x;
247dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    unsigned count;
248dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
249dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    switch(p->msg.command){
250dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_SYNC: tag = "SYNC"; break;
251dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_CNXN: tag = "CNXN" ; break;
252dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OPEN: tag = "OPEN"; break;
253dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OKAY: tag = "OKAY"; break;
254dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_CLSE: tag = "CLSE"; break;
255dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_WRTE: tag = "WRTE"; break;
256d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    case A_AUTH: tag = "AUTH"; break;
257dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    default: tag = "????"; break;
258dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
259dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
260dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "%s: %s %08x %08x %04x \"",
261dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            label, tag, p->msg.arg0, p->msg.arg1, p->msg.data_length);
262dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    count = p->msg.data_length;
263dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    x = (char*) p->data;
264dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(count > DUMPMAX) {
265dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        count = DUMPMAX;
266dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        tag = "\n";
267dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } else {
268dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        tag = "\"\n";
269dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
270dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    while(count-- > 0){
271dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if((*x >= ' ') && (*x < 127)) {
272dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fputc(*x, stderr);
273dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else {
274dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fputc('.', stderr);
275dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
276dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        x++;
277dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
278d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    fputs(tag, stderr);
279dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
280dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
281dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
282dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void send_ready(unsigned local, unsigned remote, atransport *t)
283dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
284dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("Calling send_ready \n");
285dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    apacket *p = get_apacket();
286dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.command = A_OKAY;
287dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg0 = local;
288dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg1 = remote;
289dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    send_packet(p, t);
290dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
291dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
292dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void send_close(unsigned local, unsigned remote, atransport *t)
293dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
294dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("Calling send_close \n");
295dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    apacket *p = get_apacket();
296dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.command = A_CLSE;
297dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg0 = local;
298dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg1 = remote;
299dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    send_packet(p, t);
300dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
301dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
302e82c2db05cae70a0490a1f84b7211ef42c329671Scott Andersonstatic size_t fill_connect_data(char *buf, size_t bufsize)
303e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson{
304e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#if ADB_HOST
305e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    return snprintf(buf, bufsize, "host::") + 1;
306e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#else
307e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    static const char *cnxn_props[] = {
308e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        "ro.product.name",
309e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        "ro.product.model",
310e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        "ro.product.device",
311e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    };
312e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    static const int num_cnxn_props = ARRAY_SIZE(cnxn_props);
313e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    int i;
314e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    size_t remaining = bufsize;
315e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    size_t len;
316e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
317e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    len = snprintf(buf, remaining, "%s::", adb_device_banner);
318e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    remaining -= len;
319e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    buf += len;
320e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    for (i = 0; i < num_cnxn_props; i++) {
321e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        char value[PROPERTY_VALUE_MAX];
322e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        property_get(cnxn_props[i], value, "");
323e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        len = snprintf(buf, remaining, "%s=%s;", cnxn_props[i], value);
324e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        remaining -= len;
325e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        buf += len;
326e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    }
327e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
328e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    return bufsize - remaining + 1;
329e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#endif
330e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson}
331e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
332ba3a251749c88210949008ae26f8bc04812084c1Dan Albertvoid send_connect(atransport *t)
333dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
334dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("Calling send_connect \n");
335dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    apacket *cp = get_apacket();
336dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    cp->msg.command = A_CNXN;
3373d2904cdf2371e26c0465184436bd063979a5d97Tamas Berghammer    cp->msg.arg0 = t->get_protocol_version();
3383d2904cdf2371e26c0465184436bd063979a5d97Tamas Berghammer    cp->msg.arg1 = t->get_max_payload();
339e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    cp->msg.data_length = fill_connect_data((char *)cp->data,
3403d2904cdf2371e26c0465184436bd063979a5d97Tamas Berghammer                                            MAX_PAYLOAD_V1);
341dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    send_packet(cp, t);
342d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby}
343d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
3448d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes// qual_overwrite is used to overwrite a qualifier string.  dst is a
3458d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes// pointer to a char pointer.  It is assumed that if *dst is non-NULL, it
3468d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes// was malloc'ed and needs to freed.  *dst will be set to a dup of src.
3478d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes// TODO: switch to std::string for these atransport fields instead.
3488d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughesstatic void qual_overwrite(char** dst, const std::string& src) {
349e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    free(*dst);
3508d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    *dst = strdup(src.c_str());
351e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson}
352e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
3538d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughesvoid parse_banner(const char* banner, atransport* t) {
354dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("parse_banner: %s\n", banner);
3558d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes
3568d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    // The format is something like:
3578d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    // "device::ro.product.name=x;ro.product.model=y;ro.product.device=z;".
3588d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    std::vector<std::string> pieces = android::base::Split(banner, ":");
3598d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes
3608d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    if (pieces.size() > 2) {
3618d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes        const std::string& props = pieces[2];
3628d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes        for (auto& prop : android::base::Split(props, ";")) {
3638d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            // The list of properties was traditionally ;-terminated rather than ;-separated.
3648d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            if (prop.empty()) continue;
3658d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes
3668d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            std::vector<std::string> key_value = android::base::Split(prop, "=");
3678d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            if (key_value.size() != 2) continue;
3688d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes
3698d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            const std::string& key = key_value[0];
3708d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            const std::string& value = key_value[1];
3718d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            if (key == "ro.product.name") {
3728d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes                qual_overwrite(&t->product, value);
3738d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            } else if (key == "ro.product.model") {
3748d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes                qual_overwrite(&t->model, value);
3758d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            } else if (key == "ro.product.device") {
3768d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes                qual_overwrite(&t->device, value);
377e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson            }
378e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        }
379dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
380dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
3818d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    const std::string& type = pieces[0];
3828d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    if (type == "bootloader") {
383dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        D("setting connection_state to kCsBootloader\n");
384dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        t->connection_state = kCsBootloader;
385dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        update_transports();
3868d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    } else if (type == "device") {
387dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        D("setting connection_state to kCsDevice\n");
388dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        t->connection_state = kCsDevice;
389dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        update_transports();
3908d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    } else if (type == "recovery") {
391dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        D("setting connection_state to kCsRecovery\n");
392dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        t->connection_state = kCsRecovery;
393dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        update_transports();
3948d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    } else if (type == "sideload") {
395dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        D("setting connection_state to kCsSideload\n");
396dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        t->connection_state = kCsSideload;
397447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker        update_transports();
3983ce9575af76fa6dc110506080434303b9459abf4Elliott Hughes    } else {
399dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        D("setting connection_state to kCsHost\n");
400dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        t->connection_state = kCsHost;
401447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker    }
402dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
403dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
404dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid handle_packet(apacket *p, atransport *t)
405dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
406dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    asocket *s;
407dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
408899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta    D("handle_packet() %c%c%c%c\n", ((char*) (&(p->msg.command)))[0],
409899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta            ((char*) (&(p->msg.command)))[1],
410899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta            ((char*) (&(p->msg.command)))[2],
411899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta            ((char*) (&(p->msg.command)))[3]);
412dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    print_packet("recv", p);
413dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
414dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    switch(p->msg.command){
415dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_SYNC:
416dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if(p->msg.arg0){
417dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            send_packet(p, t);
418dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            if(HOST) send_connect(t);
419dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else {
420dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert            t->connection_state = kCsOffline;
421dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            handle_offline(t);
422dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            send_packet(p, t);
423dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
424dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return;
425dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
426dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_CNXN: /* CONNECT(version, maxdata, "system-id-string") */
427dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        if(t->connection_state != kCsOffline) {
428dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert            t->connection_state = kCsOffline;
429dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            handle_offline(t);
430dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
431d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
4323d2904cdf2371e26c0465184436bd063979a5d97Tamas Berghammer        t->update_version(p->msg.arg0, p->msg.arg1);
4338d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes        parse_banner(reinterpret_cast<const char*>(p->data), t);
434d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
4355cba504215ea91187cc36ec7aec5dce1b0f4b0feElliott Hughes        if (HOST || !auth_required) {
436d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            handle_online(t);
4375cba504215ea91187cc36ec7aec5dce1b0f4b0feElliott Hughes            if (!HOST) send_connect(t);
438d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        } else {
439d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            send_auth_request(t);
440d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        }
441d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        break;
442d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
443d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    case A_AUTH:
444d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        if (p->msg.arg0 == ADB_AUTH_TOKEN) {
445dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert            t->connection_state = kCsUnauthorized;
446d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            t->key = adb_auth_nextkey(t->key);
447d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            if (t->key) {
448d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                send_auth_response(p->data, p->msg.data_length, t);
449d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            } else {
450d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                /* No more private keys to try, send the public key */
451d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                send_auth_publickey(t);
452d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            }
453d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        } else if (p->msg.arg0 == ADB_AUTH_SIGNATURE) {
454d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            if (adb_auth_verify(t->token, p->data, p->msg.data_length)) {
455d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                adb_auth_verified(t);
456d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                t->failed_auth_attempts = 0;
457d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            } else {
458d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                if (t->failed_auth_attempts++ > 10)
459d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                    adb_sleep_ms(1000);
460d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                send_auth_request(t);
461d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            }
462d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        } else if (p->msg.arg0 == ADB_AUTH_RSAPUBLICKEY) {
463d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            adb_auth_confirm_key(p->data, p->msg.data_length, t);
464d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        }
465dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
466dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
467dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OPEN: /* OPEN(local-id, 0, "destination") */
468818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg0 != 0 && p->msg.arg1 == 0) {
469dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            char *name = (char*) p->data;
470dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
471dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            s = create_local_service_socket(name);
472dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            if(s == 0) {
473dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                send_close(0, p->msg.arg0, t);
474dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            } else {
475dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                s->peer = create_remote_socket(p->msg.arg0, t);
476dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                s->peer->peer = s;
477dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                send_ready(s->id, s->peer->id, t);
478dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                s->ready(s);
479dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
480dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
481dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
482dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
483dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OKAY: /* READY(local-id, remote-id, "") */
484818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
485818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner            if((s = find_local_socket(p->msg.arg1, 0))) {
486dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                if(s->peer == 0) {
487818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    /* On first READY message, create the connection. */
488dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    s->peer = create_remote_socket(p->msg.arg0, t);
489dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    s->peer->peer = s;
490818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    s->ready(s);
491818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                } else if (s->peer->id == p->msg.arg0) {
492818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    /* Other READY messages must use the same local-id */
493818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    s->ready(s);
494818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                } else {
495818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    D("Invalid A_OKAY(%d,%d), expected A_OKAY(%d,%d) on transport %s\n",
496818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                      p->msg.arg0, p->msg.arg1, s->peer->id, p->msg.arg1, t->serial);
497dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                }
498dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
499dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
500dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
501dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
502818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner    case A_CLSE: /* CLOSE(local-id, remote-id, "") or CLOSE(0, remote-id, "") */
503818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg1 != 0) {
504818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner            if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
505818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                /* According to protocol.txt, p->msg.arg0 might be 0 to indicate
506818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * a failed OPEN only. However, due to a bug in previous ADB
507818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * versions, CLOSE(0, remote-id, "") was also used for normal
508818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * CLOSE() operations.
509818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 *
510818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * This is bad because it means a compromised adbd could
511818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * send packets to close connections between the host and
512818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * other devices. To avoid this, only allow this if the local
513818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * socket has a peer on the same transport.
514818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 */
515818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                if (p->msg.arg0 == 0 && s->peer && s->peer->transport != t) {
516818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    D("Invalid A_CLSE(0, %u) from transport %s, expected transport %s\n",
517818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                      p->msg.arg1, t->serial, s->peer->transport->serial);
518818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                } else {
519818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    s->close(s);
520818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                }
521dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
522dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
523dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
524dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
525818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner    case A_WRTE: /* WRITE(local-id, remote-id, <data>) */
526818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
527818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner            if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
528dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                unsigned rid = p->msg.arg0;
529dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                p->len = p->msg.data_length;
530dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
531dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                if(s->enqueue(s, p) == 0) {
532dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    D("Enqueue the socket\n");
533dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    send_ready(s->id, rid, t);
534dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                }
535dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                return;
536dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
537dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
538dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
539dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
540dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    default:
541dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        printf("handle_packet: what is %08x?!\n", p->msg.command);
542dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
543dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
544dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    put_apacket(p);
545dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
546dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
547dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if ADB_HOST
548571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall
549a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzingerint launch_server(int server_port)
550dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
551e77b6a08623bba383ce55cd6653bec76cdf57792Yabin Cui#if defined(_WIN32)
552dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* we need to start the server in the background                    */
553dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* we create a PIPE that will be used to wait for the server's "OK" */
554dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* message since the pipe handles must be inheritable, we use a     */
555dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* security attribute                                               */
556d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    HANDLE                nul_read, nul_write;
557dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    HANDLE                pipe_read, pipe_write;
558267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    HANDLE                stdout_handle, stderr_handle;
559dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    SECURITY_ATTRIBUTES   sa;
560dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    STARTUPINFO           startup;
561dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    PROCESS_INFORMATION   pinfo;
562dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char                  program_path[ MAX_PATH ];
563dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    int                   ret;
564dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
565dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    sa.nLength = sizeof(sa);
566dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    sa.lpSecurityDescriptor = NULL;
567dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    sa.bInheritHandle = TRUE;
568dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
569d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    /* Redirect stdin and stderr to Windows /dev/null. If we instead pass our
570d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low     * stdin/stderr handles and they are console handles, when the adb server
571d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low     * starts up, the C Runtime will see console handles for a process that
572d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low     * isn't connected to a console and it will configure stderr to be closed.
573d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low     * At that point, freopen() could be used to reopen stderr, but it would
574d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low     * take more massaging to fixup the file descriptor number that freopen()
575d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low     * uses. It's simplest to avoid all of this complexity by just redirecting
576d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low     * stdin/stderr to `nul' and then the C Runtime acts as expected.
577d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low     */
578d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    nul_read = CreateFileW(L"nul", GENERIC_READ,
579d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low                           FILE_SHARE_READ | FILE_SHARE_WRITE, &sa,
580d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low                           OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
581d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    if (nul_read == INVALID_HANDLE_VALUE) {
582d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low        fprintf(stderr, "CreateFileW(nul, GENERIC_READ) failure, error %ld\n",
583d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low                GetLastError());
584d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low        return -1;
585d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    }
586d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low
587d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    nul_write = CreateFileW(L"nul", GENERIC_WRITE,
588d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low                            FILE_SHARE_READ | FILE_SHARE_WRITE, &sa,
589d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low                            OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
590d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    if (nul_write == INVALID_HANDLE_VALUE) {
591d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low        fprintf(stderr, "CreateFileW(nul, GENERIC_WRITE) failure, error %ld\n",
592d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low                GetLastError());
593d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low        CloseHandle(nul_read);
594d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low        return -1;
595d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    }
596d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low
597dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* create pipe, and ensure its read handle isn't inheritable */
598dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    ret = CreatePipe( &pipe_read, &pipe_write, &sa, 0 );
599dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!ret) {
600dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "CreatePipe() failure, error %ld\n", GetLastError() );
601d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low        CloseHandle(nul_read);
602d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low        CloseHandle(nul_write);
603dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return -1;
604dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
605dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
606dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    SetHandleInformation( pipe_read, HANDLE_FLAG_INHERIT, 0 );
607dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
608267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    /* Some programs want to launch an adb command and collect its output by
609267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * calling CreateProcess with inheritable stdout/stderr handles, then
610267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * using read() to get its output. When this happens, the stdout/stderr
611267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * handles passed to the adb client process will also be inheritable.
612267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * When starting the adb server here, care must be taken to reset them
613267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * to non-inheritable.
614267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * Otherwise, something bad happens: even if the adb command completes,
615267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * the calling process is stuck while read()-ing from the stdout/stderr
616267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * descriptors, because they're connected to corresponding handles in the
617267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * adb server process (even if the latter never uses/writes to them).
618267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     */
619267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    stdout_handle = GetStdHandle( STD_OUTPUT_HANDLE );
620267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    stderr_handle = GetStdHandle( STD_ERROR_HANDLE );
621267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    if (stdout_handle != INVALID_HANDLE_VALUE) {
622267aa8b00eadf273321e492026af74b7baad890eRay Donnelly        SetHandleInformation( stdout_handle, HANDLE_FLAG_INHERIT, 0 );
623267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    }
624267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    if (stderr_handle != INVALID_HANDLE_VALUE) {
625267aa8b00eadf273321e492026af74b7baad890eRay Donnelly        SetHandleInformation( stderr_handle, HANDLE_FLAG_INHERIT, 0 );
626267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    }
627267aa8b00eadf273321e492026af74b7baad890eRay Donnelly
628dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    ZeroMemory( &startup, sizeof(startup) );
629dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    startup.cb = sizeof(startup);
630d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    startup.hStdInput  = nul_read;
631dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    startup.hStdOutput = pipe_write;
632d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    startup.hStdError  = nul_write;
633dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    startup.dwFlags    = STARTF_USESTDHANDLES;
634dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
635dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    ZeroMemory( &pinfo, sizeof(pinfo) );
636dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
637dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* get path of current program */
638dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    GetModuleFileName( NULL, program_path, sizeof(program_path) );
639a09558c2e205771e6830eeb6241c5c0e3b1cf4ceWenhao Li    char args[64];
640a09558c2e205771e6830eeb6241c5c0e3b1cf4ceWenhao Li    snprintf(args, sizeof(args), "adb -P %d fork-server server",  server_port);
641dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    ret = CreateProcess(
642dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            program_path,                              /* program path  */
643a09558c2e205771e6830eeb6241c5c0e3b1cf4ceWenhao Li            args,
644dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                                    /* the fork-server argument will set the
645dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                                       debug = 2 in the child           */
646dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                   /* process handle is not inheritable */
647dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                    /* thread handle is not inheritable */
648dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            TRUE,                          /* yes, inherit some handles */
649dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            DETACHED_PROCESS, /* the new process doesn't have a console */
650dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                     /* use parent's environment block */
651dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                    /* use parent's starting directory */
652dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            &startup,                 /* startup info, i.e. std handles */
653dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            &pinfo );
654dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
655d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    CloseHandle( nul_read );
656d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    CloseHandle( nul_write );
657dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    CloseHandle( pipe_write );
658dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
659dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!ret) {
660dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "CreateProcess failure, error %ld\n", GetLastError() );
661dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        CloseHandle( pipe_read );
662dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return -1;
663dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
664dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
665dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    CloseHandle( pinfo.hProcess );
666dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    CloseHandle( pinfo.hThread );
667dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
668dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* wait for the "OK\n" message */
669dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    {
670dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        char  temp[3];
671dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        DWORD  count;
672dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
673dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        ret = ReadFile( pipe_read, temp, 3, &count, NULL );
674dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        CloseHandle( pipe_read );
675dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if ( !ret ) {
676dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fprintf(stderr, "could not read ok from ADB Server, error = %ld\n", GetLastError() );
677dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return -1;
678dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
679dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (count != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
680dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fprintf(stderr, "ADB server didn't ACK\n" );
681dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return -1;
682dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
683dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
684e77b6a08623bba383ce55cd6653bec76cdf57792Yabin Cui#else /* !defined(_WIN32) */
685dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char    path[PATH_MAX];
686dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    int     fd[2];
687dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
688dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // set up a pipe so the child can tell us when it is ready.
689dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // fd[0] will be parent's end, and fd[1] will get mapped to stderr in the child.
690dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (pipe(fd)) {
691dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "pipe failed in launch_server, errno: %d\n", errno);
692dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return -1;
693dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
6943166410a82f43d39201be98a8d35c51baa86cb53Alexey Tarasov    get_my_path(path, PATH_MAX);
695dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    pid_t pid = fork();
696dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(pid < 0) return -1;
697dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
698dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (pid == 0) {
699dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // child side of the fork
700dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
701dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // redirect stderr to the pipe
702dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // we use stderr instead of stdout due to stdout's buffering behavior.
703dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[0]);
704dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        dup2(fd[1], STDERR_FILENO);
705dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[1]);
706dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
707d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel        char str_port[30];
708d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel        snprintf(str_port, sizeof(str_port), "%d",  server_port);
709dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // child process
710d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel        int result = execl(path, "adb", "-P", str_port, "fork-server", "server", NULL);
711dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // this should not return
712dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "OOPS! execl returned %d, errno: %d\n", result, errno);
713dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } else  {
714dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // parent side of the fork
715dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
716dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        char  temp[3];
717dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
718dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        temp[0] = 'A'; temp[1] = 'B'; temp[2] = 'C';
719dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // wait for the "OK\n" message
720dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[1]);
721dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int ret = adb_read(fd[0], temp, 3);
722408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall        int saved_errno = errno;
723dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[0]);
724dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (ret < 0) {
725408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall            fprintf(stderr, "could not read ok from ADB Server, errno = %d\n", saved_errno);
726dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return -1;
727dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
728dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (ret != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
729dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fprintf(stderr, "ADB server didn't ACK\n" );
730dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return -1;
731dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
732dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
733dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        setsid();
734dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
735e77b6a08623bba383ce55cd6653bec76cdf57792Yabin Cui#endif /* !defined(_WIN32) */
736dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return 0;
737dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
738e77b6a08623bba383ce55cd6653bec76cdf57792Yabin Cui#endif /* ADB_HOST */
739dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
740252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner// Try to handle a network forwarding request.
741252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner// This returns 1 on success, 0 on failure, and -1 to indicate this is not
742252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner// a forwarding-related request.
743aee80fb67bcc3fc5209050c40bd8efc1bbcc0610Elliott Hughesint handle_forward_request(const char* service, TransportType type, const char* serial, int reply_fd)
744252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner{
745252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    if (!strcmp(service, "list-forward")) {
746252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        // Create the list of forward redirections.
747e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        std::string listeners = format_listeners();
748252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#if ADB_HOST
749e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendOkay(reply_fd);
750252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#endif
751e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendProtocolString(reply_fd, listeners);
752252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        return 1;
753252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    }
754252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
755252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    if (!strcmp(service, "killforward-all")) {
756252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        remove_all_listeners();
757252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#if ADB_HOST
758252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        /* On the host: 1st OKAY is connect, 2nd OKAY is status */
759e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendOkay(reply_fd);
760252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#endif
761e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendOkay(reply_fd);
762252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        return 1;
763252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    }
764252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
765424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes    if (!strncmp(service, "forward:", 8) || !strncmp(service, "killforward:", 12)) {
766424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        // killforward:local
767424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        // forward:(norebind:)?local;remote
768424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        bool kill_forward = false;
769424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        bool no_rebind = false;
770424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        if (android::base::StartsWith(service, "killforward:")) {
771424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            kill_forward = true;
772424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            service += 12;
773a6b340af123b82e3aaa33cb2a0d45f637eeb4759Spencer Low        } else {
774a6b340af123b82e3aaa33cb2a0d45f637eeb4759Spencer Low            service += 8;   // skip past "forward:"
775424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            if (android::base::StartsWith(service, "norebind:")) {
776424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes                no_rebind = true;
777424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes                service += 9;
778424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            }
779252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
780252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
781424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        std::vector<std::string> pieces = android::base::Split(service, ";");
782252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
783424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        if (kill_forward) {
784424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            // Check killforward: parameter format: '<local>'
785424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            if (pieces.size() != 1 || pieces[0].empty()) {
786424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes                SendFail(reply_fd, android::base::StringPrintf("bad killforward: %s", service));
787252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                return 1;
788252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            }
789252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        } else {
790424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            // Check forward: parameter format: '<local>;<remote>'
791424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            if (pieces.size() != 2 || pieces[0].empty() || pieces[1].empty() || pieces[1][0] == '*') {
792424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes                SendFail(reply_fd, android::base::StringPrintf("bad forward: %s", service));
793252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                return 1;
794252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            }
795252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
796252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
7977be29c819b4fb8bf9b1d4b69a4333f8765de0281Elliott Hughes        std::string error_msg;
798424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        atransport* transport = acquire_one_transport(kCsAny, type, serial, &error_msg);
799252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        if (!transport) {
800e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            SendFail(reply_fd, error_msg);
801252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            return 1;
802252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
803252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
8045200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low        std::string error;
8053bd73c12c07dcefc965abeef535ac53c4754d682Elliott Hughes        InstallStatus r;
806424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        if (kill_forward) {
807424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            r = remove_listener(pieces[0].c_str(), transport);
808252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        } else {
8095200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low            r = install_listener(pieces[0], pieces[1].c_str(), transport,
8105200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low                                 no_rebind, &error);
811252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
8127b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes        if (r == INSTALL_STATUS_OK) {
813252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#if ADB_HOST
814252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            /* On the host: 1st OKAY is connect, 2nd OKAY is status */
815e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            SendOkay(reply_fd);
816252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#endif
817e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            SendOkay(reply_fd);
818252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            return 1;
819252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
820252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
8217b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes        std::string message;
8227b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes        switch (r) {
823424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes          case INSTALL_STATUS_OK: message = "success (!)"; break;
8247b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes          case INSTALL_STATUS_INTERNAL_ERROR: message = "internal error"; break;
8257b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes          case INSTALL_STATUS_CANNOT_BIND:
8265200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low            message = android::base::StringPrintf("cannot bind to socket: %s",
8275200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low                                                  error.c_str());
8287b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes            break;
8297b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes          case INSTALL_STATUS_CANNOT_REBIND:
830a6b340af123b82e3aaa33cb2a0d45f637eeb4759Spencer Low            message = android::base::StringPrintf("cannot rebind existing socket");
8317b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes            break;
832424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes          case INSTALL_STATUS_LISTENER_NOT_FOUND:
833424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            message = android::base::StringPrintf("listener '%s' not found", service);
834424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            break;
835252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
836e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendFail(reply_fd, message);
837252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        return 1;
838252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    }
839252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    return 0;
840252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner}
841252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
842e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes#if ADB_HOST
843e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughesstatic int SendOkay(int fd, const std::string& s) {
844e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    SendOkay(fd);
845e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    SendProtocolString(fd, s);
846e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    return 0;
847e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes}
848e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes#endif
849e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes
850c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertint handle_host_request(const char* service, TransportType type,
851c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert                        const char* serial, int reply_fd, asocket* s) {
852c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert    if (strcmp(service, "kill") == 0) {
853c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert        fprintf(stderr, "adb server killed by remote request\n");
854dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fflush(stdout);
855e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendOkay(reply_fd);
856dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        exit(0);
857dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
858dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
859dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if ADB_HOST
860dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport:" is used for switching transport with a specified serial number
861dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport-usb:" is used for switching transport to the only USB transport
862dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport-local:" is used for switching transport to the only local transport
863dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport-any:" is used for switching transport to the only transport
864dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!strncmp(service, "transport", strlen("transport"))) {
8653bd73c12c07dcefc965abeef535ac53c4754d682Elliott Hughes        TransportType type = kTransportAny;
866dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
867dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (!strncmp(service, "transport-usb", strlen("transport-usb"))) {
868dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            type = kTransportUsb;
869dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else if (!strncmp(service, "transport-local", strlen("transport-local"))) {
870dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            type = kTransportLocal;
871dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else if (!strncmp(service, "transport-any", strlen("transport-any"))) {
872dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            type = kTransportAny;
873dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else if (!strncmp(service, "transport:", strlen("transport:"))) {
874dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            service += strlen("transport:");
8753175c8e9bfcb01f181d7300755da15a999e5461cTom Marlin            serial = service;
876dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
877dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
878e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        std::string error_msg;
879e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        atransport* t = acquire_one_transport(kCsAny, type, serial, &error_msg);
880e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        if (t != nullptr) {
881e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes            s->transport = t;
882e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            SendOkay(reply_fd);
883dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else {
884e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            SendFail(reply_fd, error_msg);
885dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
886dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return 1;
887dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
888dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
889dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // return a list of all connected devices
890e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson    if (!strncmp(service, "devices", 7)) {
891e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        bool long_listing = (strcmp(service+7, "-l") == 0);
892e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        if (long_listing || service[7] == 0) {
893e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            D("Getting device list...\n");
894e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            std::string device_list = list_transports(long_listing);
895e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            D("Sending device list...\n");
896e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes            return SendOkay(reply_fd, device_list);
897e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson        }
898e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        return 1;
899dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
900dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
90174d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood    // remove TCP transport
90274d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood    if (!strncmp(service, "disconnect:", 11)) {
9033d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        const std::string address(service + 11);
9043d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        if (address.empty()) {
905cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            // disconnect from all TCP devices
906cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            unregister_all_tcp_transports();
9073d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes            return SendOkay(reply_fd, "disconnected everything");
90874d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood        }
90974d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood
9103d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        std::string serial;
9113d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        std::string host;
9123d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        int port = DEFAULT_ADB_LOCAL_TRANSPORT_PORT;
9133d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        std::string error;
9143d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        if (!parse_host_and_port(address, &serial, &host, &port, &error)) {
9153d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes            return SendFail(reply_fd, android::base::StringPrintf("couldn't parse '%s': %s",
9163d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes                                                                  address.c_str(), error.c_str()));
9173d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        }
9183d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        atransport* t = find_transport(serial.c_str());
9193d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        if (t == nullptr) {
9203d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes            return SendFail(reply_fd, android::base::StringPrintf("no such device '%s'",
9213d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes                                                                  serial.c_str()));
9223d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        }
9233d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        unregister_transport(t);
9243d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        return SendOkay(reply_fd, android::base::StringPrintf("disconnected %s", address.c_str()));
9252f38b699713dc2587a771bd5d4c6a47329728f5eMike Lockwood    }
9262f38b699713dc2587a771bd5d4c6a47329728f5eMike Lockwood
927dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // returns our value for ADB_SERVER_VERSION
928dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!strcmp(service, "version")) {
929e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        return SendOkay(reply_fd, android::base::StringPrintf("%04x", ADB_SERVER_VERSION));
930dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
931dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
932e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    // These always report "unknown" rather than the actual error, for scripts.
933e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    if (!strcmp(service, "get-serialno")) {
934e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        std::string ignored;
935e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        atransport* t = acquire_one_transport(kCsAny, type, serial, &ignored);
936e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        return SendOkay(reply_fd, (t && t->serial) ? t->serial : "unknown");
937dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
938e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    if (!strcmp(service, "get-devpath")) {
939e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        std::string ignored;
940e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        atransport* t = acquire_one_transport(kCsAny, type, serial, &ignored);
941e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        return SendOkay(reply_fd, (t && t->devpath) ? t->devpath : "unknown");
942e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson    }
943e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    if (!strcmp(service, "get-state")) {
944e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        std::string ignored;
945e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        atransport* t = acquire_one_transport(kCsAny, type, serial, &ignored);
946e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        return SendOkay(reply_fd, t ? t->connection_state_name() : "unknown");
947e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    }
948e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes
949dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // indicates a new emulator instance has started
950e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    if (!strncmp(service, "emulator:", 9)) {
951dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int  port = atoi(service+9);
952dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        local_connect(port);
953dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        /* we don't even need to send a reply */
954dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return 0;
955dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
956dc22c3c7a82ad1919c52dc4d00c601ad7210c61bSimon Ye#endif // ADB_HOST
957dc22c3c7a82ad1919c52dc4d00c601ad7210c61bSimon Ye
9583bd73c12c07dcefc965abeef535ac53c4754d682Elliott Hughes    int ret = handle_forward_request(service, type, serial, reply_fd);
959dc22c3c7a82ad1919c52dc4d00c601ad7210c61bSimon Ye    if (ret >= 0)
960dc22c3c7a82ad1919c52dc4d00c601ad7210c61bSimon Ye      return ret - 1;
961dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return -1;
962dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
963