adb.cpp revision 1f4ec19e499ba981e4117f647d191603c2713e79
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
17aed3c61c4437ebb05eadfb3bf85d6962c30b9935Yabin Cui#define TRACE_TAG 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>
35dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
365f787ed2b3b9f6cc02aa5923b95d77e2a5865438David Pursell#include <android-base/errors.h>
374f71319df011d796a60a43fc1bc68e16fbf7d321Elliott Hughes#include <android-base/logging.h>
384f71319df011d796a60a43fc1bc68e16fbf7d321Elliott Hughes#include <android-base/macros.h>
39706955ff0d159d28bb6eb06dc10178624c505f4eDavid Pursell#include <android-base/parsenetaddress.h>
404f71319df011d796a60a43fc1bc68e16fbf7d321Elliott Hughes#include <android-base/stringprintf.h>
414f71319df011d796a60a43fc1bc68e16fbf7d321Elliott Hughes#include <android-base/strings.h>
427b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes
43d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby#include "adb_auth.h"
44cc731cc76786b6bdc58764aad9924c0d0c8d645fDan Albert#include "adb_io.h"
45e9fca14c9e555a9d02e1d94ba6d3b290673ea26bDan Albert#include "adb_listeners.h"
463d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes#include "adb_utils.h"
477664901a355b959f312e9acff5a0fd31b7139623Dan Albert#include "transport.h"
48dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
49e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))
50e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
51dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if !ADB_HOST
52893a4a47e8c9290128b9254af0246e36f821c260Nick Kralevich#include <cutils/properties.h>
53e2864bf727d6a404bd871f063c03e36cc8eedeecNick Kralevich#include <sys/capability.h>
54885342a0f2c834a6b680284047c47c9d04b32565Jeff Sharkey#include <sys/mount.h>
55dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
56dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
5742ae2604f43ef88be748f125e8733f2c08b83faeElliott Hughesstd::string adb_version() {
5842ae2604f43ef88be748f125e8733f2c08b83faeElliott Hughes    // Don't change the format of this --- it's parsed by ddmlib.
5942ae2604f43ef88be748f125e8733f2c08b83faeElliott Hughes    return android::base::StringPrintf("Android Debug Bridge version %d.%d.%d\n"
6042ae2604f43ef88be748f125e8733f2c08b83faeElliott Hughes                                       "Revision %s\n",
6142ae2604f43ef88be748f125e8733f2c08b83faeElliott Hughes                                       ADB_VERSION_MAJOR, ADB_VERSION_MINOR, ADB_SERVER_VERSION,
6242ae2604f43ef88be748f125e8733f2c08b83faeElliott Hughes                                       ADB_REVISION);
6342ae2604f43ef88be748f125e8733f2c08b83faeElliott Hughes}
6442ae2604f43ef88be748f125e8733f2c08b83faeElliott Hughes
659313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertvoid fatal(const char *fmt, ...) {
66dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_list ap;
67dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_start(ap, fmt);
68dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "error: ");
69dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    vfprintf(stderr, fmt, ap);
70dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "\n");
71dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_end(ap);
72dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    exit(-1);
73dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
74dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
759313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albertvoid fatal_errno(const char* fmt, ...) {
76dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_list ap;
77dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_start(ap, fmt);
78dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "error: %s: ", strerror(errno));
79dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    vfprintf(stderr, fmt, ap);
80dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "\n");
81dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_end(ap);
82dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    exit(-1);
83dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
84dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
85bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albertapacket* get_apacket(void)
86dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
87bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albert    apacket* p = reinterpret_cast<apacket*>(malloc(sizeof(apacket)));
88bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albert    if (p == nullptr) {
89bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albert      fatal("failed to allocate an apacket");
90bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albert    }
91bac3474a8256cb32a29e8d46f78cad95a5502692Dan Albert
92dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    memset(p, 0, sizeof(apacket) - MAX_PAYLOAD);
93dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return p;
94dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
95dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
96dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid put_apacket(apacket *p)
97dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
98dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    free(p);
99dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
100dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
101d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Gobyvoid handle_online(atransport *t)
102dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
1037a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui    D("adb: online");
104d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    t->online = 1;
105dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
106dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
107dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid handle_offline(atransport *t)
108dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
1097a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui    D("adb: offline");
110dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    //Close the associated usb
111d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    t->online = 0;
11234f45c566365c966c78264e8a4dc0cf69957b6faYabin Cui
113b329824e6c5373ae303269dca285d835ce57e514Yabin Cui    // This is necessary to avoid a race condition that occurred when a transport closes
11434f45c566365c966c78264e8a4dc0cf69957b6faYabin Cui    // while a client socket is still active.
11534f45c566365c966c78264e8a4dc0cf69957b6faYabin Cui    close_all_sockets(t);
11634f45c566365c966c78264e8a4dc0cf69957b6faYabin Cui
117b329824e6c5373ae303269dca285d835ce57e514Yabin Cui    t->RunDisconnects();
118dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
119dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
120d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby#if DEBUG_PACKETS
121dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#define DUMPMAX 32
122dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid print_packet(const char *label, apacket *p)
123dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
124dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char *tag;
125dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char *x;
126dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    unsigned count;
127dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
128dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    switch(p->msg.command){
129dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_SYNC: tag = "SYNC"; break;
130dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_CNXN: tag = "CNXN" ; break;
131dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OPEN: tag = "OPEN"; break;
132dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OKAY: tag = "OKAY"; break;
133dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_CLSE: tag = "CLSE"; break;
134dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_WRTE: tag = "WRTE"; break;
135d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    case A_AUTH: tag = "AUTH"; break;
136dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    default: tag = "????"; break;
137dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
138dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
139dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "%s: %s %08x %08x %04x \"",
140dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            label, tag, p->msg.arg0, p->msg.arg1, p->msg.data_length);
141dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    count = p->msg.data_length;
142dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    x = (char*) p->data;
143dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(count > DUMPMAX) {
144dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        count = DUMPMAX;
145dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        tag = "\n";
146dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } else {
147dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        tag = "\"\n";
148dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
149dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    while(count-- > 0){
150dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if((*x >= ' ') && (*x < 127)) {
151dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fputc(*x, stderr);
152dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else {
153dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fputc('.', stderr);
154dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
155dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        x++;
156dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
157d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    fputs(tag, stderr);
158dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
159dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
160dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
161dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void send_ready(unsigned local, unsigned remote, atransport *t)
162dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
1637a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui    D("Calling send_ready");
164dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    apacket *p = get_apacket();
165dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.command = A_OKAY;
166dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg0 = local;
167dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg1 = remote;
168dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    send_packet(p, t);
169dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
170dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
171dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void send_close(unsigned local, unsigned remote, atransport *t)
172dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
1737a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui    D("Calling send_close");
174dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    apacket *p = get_apacket();
175dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.command = A_CLSE;
176dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg0 = local;
177dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg1 = remote;
178dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    send_packet(p, t);
179dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
180dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1811792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albertstd::string get_connection_string() {
1821792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    std::vector<std::string> connection_properties;
1831792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert
1841792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert#if !ADB_HOST
1851792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    static const char* cnxn_props[] = {
186e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        "ro.product.name",
187e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        "ro.product.model",
188e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        "ro.product.device",
189e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    };
1901792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert
1911792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    for (const auto& prop_name : cnxn_props) {
192e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        char value[PROPERTY_VALUE_MAX];
1931792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        property_get(prop_name, value, "");
1941792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        connection_properties.push_back(
1951792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert            android::base::StringPrintf("%s=%s", prop_name, value));
196e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    }
197e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#endif
1981792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert
1991792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    connection_properties.push_back(android::base::StringPrintf(
2004e2fd36bc8c16147cab323b0418a7666812d3bc7David Pursell        "features=%s", FeatureSetToString(supported_features()).c_str()));
2011792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert
2021792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    return android::base::StringPrintf(
2031792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        "%s::%s", adb_device_banner,
2041792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        android::base::Join(connection_properties, ';').c_str());
205e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson}
206e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
2071792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albertvoid send_connect(atransport* t) {
2087a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui    D("Calling send_connect");
2091792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    apacket* cp = get_apacket();
210dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    cp->msg.command = A_CNXN;
2113d2904cdf2371e26c0465184436bd063979a5d97Tamas Berghammer    cp->msg.arg0 = t->get_protocol_version();
2123d2904cdf2371e26c0465184436bd063979a5d97Tamas Berghammer    cp->msg.arg1 = t->get_max_payload();
2131792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert
2141792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    std::string connection_str = get_connection_string();
2151792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    // Connect and auth packets are limited to MAX_PAYLOAD_V1 because we don't
2161792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    // yet know how much data the other size is willing to accept.
2171792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    if (connection_str.length() > MAX_PAYLOAD_V1) {
2181792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        LOG(FATAL) << "Connection banner is too long (length = "
2191792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert                   << connection_str.length() << ")";
2201792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    }
2211792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert
2221792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    memcpy(cp->data, connection_str.c_str(), connection_str.length());
2231792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    cp->msg.data_length = connection_str.length();
2241792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert
225dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    send_packet(cp, t);
226d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby}
227d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
2288d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes// qual_overwrite is used to overwrite a qualifier string.  dst is a
2298d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes// pointer to a char pointer.  It is assumed that if *dst is non-NULL, it
2308d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes// was malloc'ed and needs to freed.  *dst will be set to a dup of src.
2318d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes// TODO: switch to std::string for these atransport fields instead.
2328d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughesstatic void qual_overwrite(char** dst, const std::string& src) {
233e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    free(*dst);
2348d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    *dst = strdup(src.c_str());
235e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson}
236e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
2371792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albertvoid parse_banner(const std::string& banner, atransport* t) {
2387a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui    D("parse_banner: %s", banner.c_str());
2398d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes
2408d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    // The format is something like:
2418d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    // "device::ro.product.name=x;ro.product.model=y;ro.product.device=z;".
2428d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    std::vector<std::string> pieces = android::base::Split(banner, ":");
2438d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes
244d2b588e23901538f4b459a71fefdac6fc2748f7eDavid Pursell    // Reset the features list or else if the server sends no features we may
245d2b588e23901538f4b459a71fefdac6fc2748f7eDavid Pursell    // keep the existing feature set (http://b/24405971).
246d2b588e23901538f4b459a71fefdac6fc2748f7eDavid Pursell    t->SetFeatures("");
247d2b588e23901538f4b459a71fefdac6fc2748f7eDavid Pursell
2488d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    if (pieces.size() > 2) {
2498d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes        const std::string& props = pieces[2];
25065fe2516b402ed8903f2ce39a86fa0bdc2b263a6Elliott Hughes        for (const auto& prop : android::base::Split(props, ";")) {
2518d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            // The list of properties was traditionally ;-terminated rather than ;-separated.
2528d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            if (prop.empty()) continue;
2538d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes
2548d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            std::vector<std::string> key_value = android::base::Split(prop, "=");
2558d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            if (key_value.size() != 2) continue;
2568d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes
2578d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            const std::string& key = key_value[0];
2588d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            const std::string& value = key_value[1];
2598d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            if (key == "ro.product.name") {
2608d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes                qual_overwrite(&t->product, value);
2618d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            } else if (key == "ro.product.model") {
2628d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes                qual_overwrite(&t->model, value);
2638d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes            } else if (key == "ro.product.device") {
2648d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes                qual_overwrite(&t->device, value);
2651792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert            } else if (key == "features") {
2664e2fd36bc8c16147cab323b0418a7666812d3bc7David Pursell                t->SetFeatures(value);
267e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson            }
268e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        }
269dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
270dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
2718d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    const std::string& type = pieces[0];
2728d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    if (type == "bootloader") {
2737a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui        D("setting connection_state to kCsBootloader");
274dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        t->connection_state = kCsBootloader;
275dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        update_transports();
2768d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    } else if (type == "device") {
2777a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui        D("setting connection_state to kCsDevice");
278dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        t->connection_state = kCsDevice;
279dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        update_transports();
2808d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    } else if (type == "recovery") {
2817a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui        D("setting connection_state to kCsRecovery");
282dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        t->connection_state = kCsRecovery;
283dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        update_transports();
2848d5fa6da44d56511b3e173bc463cbc65ff221b4aElliott Hughes    } else if (type == "sideload") {
2857a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui        D("setting connection_state to kCsSideload");
286dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        t->connection_state = kCsSideload;
287447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker        update_transports();
2883ce9575af76fa6dc110506080434303b9459abf4Elliott Hughes    } else {
2897a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui        D("setting connection_state to kCsHost");
290dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert        t->connection_state = kCsHost;
291447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker    }
292dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
293dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
2941792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albertstatic void handle_new_connection(atransport* t, apacket* p) {
2951792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    if (t->connection_state != kCsOffline) {
2961792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        t->connection_state = kCsOffline;
2971792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        handle_offline(t);
2981792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    }
2991792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert
3001792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    t->update_version(p->msg.arg0, p->msg.arg1);
3011792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    std::string banner(reinterpret_cast<const char*>(p->data),
3021792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert                       p->msg.data_length);
3031792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    parse_banner(banner, t);
3041792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert
3051792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert#if ADB_HOST
3061792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    handle_online(t);
3071792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert#else
3081792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    if (!auth_required) {
3091792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        handle_online(t);
3101792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        send_connect(t);
3111792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    } else {
3121792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        send_auth_request(t);
3131792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    }
3141792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert#endif
3151792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert}
3161792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert
317dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid handle_packet(apacket *p, atransport *t)
318dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
319dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    asocket *s;
320dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
3217a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui    D("handle_packet() %c%c%c%c", ((char*) (&(p->msg.command)))[0],
322899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta            ((char*) (&(p->msg.command)))[1],
323899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta            ((char*) (&(p->msg.command)))[2],
324899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta            ((char*) (&(p->msg.command)))[3]);
325dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    print_packet("recv", p);
326dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
327dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    switch(p->msg.command){
328dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_SYNC:
329dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if(p->msg.arg0){
330dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            send_packet(p, t);
331661327e8e40f95076e0e8abe86193da16b061bd8Yabin Cui#if ADB_HOST
332661327e8e40f95076e0e8abe86193da16b061bd8Yabin Cui            send_connect(t);
333661327e8e40f95076e0e8abe86193da16b061bd8Yabin Cui#endif
334dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else {
335dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert            t->connection_state = kCsOffline;
336dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            handle_offline(t);
337dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            send_packet(p, t);
338dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
339dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return;
340dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
3411792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert    case A_CNXN:  // CONNECT(version, maxdata, "system-id-string")
3421792c23cb892ab58590b2cdfce0d0ece30c21787Dan Albert        handle_new_connection(t, p);
343d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        break;
344d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
345d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    case A_AUTH:
346d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        if (p->msg.arg0 == ADB_AUTH_TOKEN) {
347dcd78a15d0be143d48fc93af6a9fa5748dbf9790Dan Albert            t->connection_state = kCsUnauthorized;
348d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            t->key = adb_auth_nextkey(t->key);
349d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            if (t->key) {
350d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                send_auth_response(p->data, p->msg.data_length, t);
351d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            } else {
352d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                /* No more private keys to try, send the public key */
353d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                send_auth_publickey(t);
354d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            }
355d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        } else if (p->msg.arg0 == ADB_AUTH_SIGNATURE) {
356d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            if (adb_auth_verify(t->token, p->data, p->msg.data_length)) {
357d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                adb_auth_verified(t);
358d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                t->failed_auth_attempts = 0;
359d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            } else {
360d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                if (t->failed_auth_attempts++ > 10)
361d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                    adb_sleep_ms(1000);
362d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                send_auth_request(t);
363d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            }
364d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        } else if (p->msg.arg0 == ADB_AUTH_RSAPUBLICKEY) {
365d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            adb_auth_confirm_key(p->data, p->msg.data_length, t);
366d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        }
367dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
368dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
369dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OPEN: /* OPEN(local-id, 0, "destination") */
370818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg0 != 0 && p->msg.arg1 == 0) {
371dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            char *name = (char*) p->data;
372dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
3730955c66b226db7a7f34613f834f7b0a145fd407dDavid Pursell            s = create_local_service_socket(name, t);
374dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            if(s == 0) {
375dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                send_close(0, p->msg.arg0, t);
376dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            } else {
377dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                s->peer = create_remote_socket(p->msg.arg0, t);
378dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                s->peer->peer = s;
379dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                send_ready(s->id, s->peer->id, t);
380dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                s->ready(s);
381dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
382dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
383dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
384dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
385dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OKAY: /* READY(local-id, remote-id, "") */
386818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
387818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner            if((s = find_local_socket(p->msg.arg1, 0))) {
388dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                if(s->peer == 0) {
389818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    /* On first READY message, create the connection. */
390dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    s->peer = create_remote_socket(p->msg.arg0, t);
391dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    s->peer->peer = s;
392818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    s->ready(s);
393818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                } else if (s->peer->id == p->msg.arg0) {
394818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    /* Other READY messages must use the same local-id */
395818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    s->ready(s);
396818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                } else {
3977a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui                    D("Invalid A_OKAY(%d,%d), expected A_OKAY(%d,%d) on transport %s",
398818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                      p->msg.arg0, p->msg.arg1, s->peer->id, p->msg.arg1, t->serial);
399dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                }
400aa77e22d7384a97757e2d2194f4e265f7ad3b71aYabin Cui            } else {
401aa77e22d7384a97757e2d2194f4e265f7ad3b71aYabin Cui                // When receiving A_OKAY from device for A_OPEN request, the host server may
402aa77e22d7384a97757e2d2194f4e265f7ad3b71aYabin Cui                // have closed the local socket because of client disconnection. Then we need
403aa77e22d7384a97757e2d2194f4e265f7ad3b71aYabin Cui                // to send A_CLSE back to device to close the service on device.
404aa77e22d7384a97757e2d2194f4e265f7ad3b71aYabin Cui                send_close(p->msg.arg1, p->msg.arg0, t);
405dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
406dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
407dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
408dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
409818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner    case A_CLSE: /* CLOSE(local-id, remote-id, "") or CLOSE(0, remote-id, "") */
410818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg1 != 0) {
411818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner            if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
412818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                /* According to protocol.txt, p->msg.arg0 might be 0 to indicate
413818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * a failed OPEN only. However, due to a bug in previous ADB
414818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * versions, CLOSE(0, remote-id, "") was also used for normal
415818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * CLOSE() operations.
416818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 *
417818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * This is bad because it means a compromised adbd could
418818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * send packets to close connections between the host and
419818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * other devices. To avoid this, only allow this if the local
420818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * socket has a peer on the same transport.
421818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 */
422818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                if (p->msg.arg0 == 0 && s->peer && s->peer->transport != t) {
4237a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui                    D("Invalid A_CLSE(0, %u) from transport %s, expected transport %s",
424818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                      p->msg.arg1, t->serial, s->peer->transport->serial);
425818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                } else {
426818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    s->close(s);
427818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                }
428dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
429dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
430dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
431dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
432818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner    case A_WRTE: /* WRITE(local-id, remote-id, <data>) */
433818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
434818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner            if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
435dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                unsigned rid = p->msg.arg0;
436dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                p->len = p->msg.data_length;
437dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
438dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                if(s->enqueue(s, p) == 0) {
4397a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui                    D("Enqueue the socket");
440dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    send_ready(s->id, rid, t);
441dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                }
442dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                return;
443dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
444dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
445dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
446dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
447dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    default:
448dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        printf("handle_packet: what is %08x?!\n", p->msg.command);
449dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
450dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
451dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    put_apacket(p);
452dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
453dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
454dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if ADB_HOST
455571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall
4562122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low#ifdef _WIN32
4572122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
458bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low// Try to make a handle non-inheritable and if there is an error, don't output
459bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low// any error info, but leave GetLastError() for the caller to read. This is
460bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low// convenient if the caller is expecting that this may fail and they'd like to
461bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low// ignore such a failure.
462bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Lowstatic bool _try_make_handle_noninheritable(HANDLE h) {
4632122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (h != INVALID_HANDLE_VALUE && h != NULL) {
464bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low        return SetHandleInformation(h, HANDLE_FLAG_INHERIT, 0) ? true : false;
465bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low    }
466bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low
467bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low    return true;
468bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low}
469bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low
470bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low// Try to make a handle non-inheritable with the expectation that this should
471bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low// succeed, so if this fails, output error info.
472bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Lowstatic bool _make_handle_noninheritable(HANDLE h) {
473bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low    if (!_try_make_handle_noninheritable(h)) {
474bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low        // Show the handle value to give us a clue in case we have problems
475bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low        // with pseudo-handle values.
476bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low        fprintf(stderr, "Cannot make handle 0x%p non-inheritable: %s\n",
4775f787ed2b3b9f6cc02aa5923b95d77e2a5865438David Pursell                h, android::base::SystemErrorCodeToString(GetLastError()).c_str());
478bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low        return false;
4792122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
4802122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
4812122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    return true;
4822122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low}
4832122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
4842122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low// Create anonymous pipe, preventing inheritance of the read pipe and setting
4852122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low// security of the write pipe to sa.
4862122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Lowstatic bool _create_anonymous_pipe(unique_handle* pipe_read_out,
4872122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                                   unique_handle* pipe_write_out,
4882122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                                   SECURITY_ATTRIBUTES* sa) {
4892122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    HANDLE pipe_read_raw = NULL;
4902122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    HANDLE pipe_write_raw = NULL;
4912122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (!CreatePipe(&pipe_read_raw, &pipe_write_raw, sa, 0)) {
4922122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        fprintf(stderr, "Cannot create pipe: %s\n",
4935f787ed2b3b9f6cc02aa5923b95d77e2a5865438David Pursell                android::base::SystemErrorCodeToString(GetLastError()).c_str());
4942122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        return false;
4952122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
4962122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
4972122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    unique_handle pipe_read(pipe_read_raw);
4982122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    pipe_read_raw = NULL;
4992122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    unique_handle pipe_write(pipe_write_raw);
5002122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    pipe_write_raw = NULL;
5012122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
5022122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (!_make_handle_noninheritable(pipe_read.get())) {
5032122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        return false;
5042122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
5052122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
5062122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    *pipe_read_out = std::move(pipe_read);
5072122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    *pipe_write_out = std::move(pipe_write);
5082122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
5092122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    return true;
5102122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low}
5112122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
51253a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low// Read from a pipe (that we take ownership of) and write the result to stdout/stderr. Return on
51353a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low// error or when the pipe is closed. Internally makes inheritable handles, so this should not be
51453a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low// called if subprocesses may be started concurrently.
5152122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Lowstatic unsigned _redirect_pipe_thread(HANDLE h, DWORD nStdHandle) {
5162122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // Take ownership of the HANDLE and close when we're done.
5172122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    unique_handle   read_pipe(h);
51853a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    const char*     output_name = nStdHandle == STD_OUTPUT_HANDLE ? "stdout" : "stderr";
51953a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    const int       original_fd = fileno(nStdHandle == STD_OUTPUT_HANDLE ? stdout : stderr);
52053a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    std::unique_ptr<FILE, decltype(&fclose)> stream(nullptr, fclose);
52153a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low
52253a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    if (original_fd == -1) {
52353a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        fprintf(stderr, "Failed to get file descriptor for %s: %s\n", output_name, strerror(errno));
52453a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        return EXIT_FAILURE;
52553a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    }
52653a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low
52753a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    // If fileno() is -2, stdout/stderr is not associated with an output stream, so we should read,
52853a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    // but don't write. Otherwise, make a FILE* identical to stdout/stderr except that it is in
52953a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    // binary mode with no CR/LR translation since we're reading raw.
53053a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    if (original_fd >= 0) {
53153a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        // This internally makes a duplicate file handle that is inheritable, so callers should not
53253a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        // call this function if subprocesses may be started concurrently.
53353a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        const int fd = dup(original_fd);
53453a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        if (fd == -1) {
53553a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            fprintf(stderr, "Failed to duplicate file descriptor for %s: %s\n", output_name,
53653a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low                    strerror(errno));
53753a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            return EXIT_FAILURE;
53853a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        }
53953a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low
54053a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        // Note that although we call fdopen() below with a binary flag, it may not adhere to that
54153a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        // flag, so we have to set the mode manually.
54253a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        if (_setmode(fd, _O_BINARY) == -1) {
54353a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            fprintf(stderr, "Failed to set binary mode for duplicate of %s: %s\n", output_name,
54453a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low                    strerror(errno));
54553a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            unix_close(fd);
54653a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            return EXIT_FAILURE;
54753a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        }
54853a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low
54953a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        stream.reset(fdopen(fd, "wb"));
55053a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        if (stream.get() == nullptr) {
55153a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            fprintf(stderr, "Failed to open duplicate stream for %s: %s\n", output_name,
55253a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low                    strerror(errno));
55353a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            unix_close(fd);
55453a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            return EXIT_FAILURE;
55553a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        }
55653a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low
55753a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        // Unbuffer the stream because it will be buffered by default and we want subprocess output
55853a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        // to be shown immediately.
55953a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        if (setvbuf(stream.get(), NULL, _IONBF, 0) == -1) {
56053a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            fprintf(stderr, "Failed to unbuffer %s: %s\n", output_name, strerror(errno));
56153a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            return EXIT_FAILURE;
56253a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        }
56353a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low
56453a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        // fd will be closed when stream is closed.
56553a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    }
5662122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
5672122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    while (true) {
5682122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        char    buf[64 * 1024];
5692122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        DWORD   bytes_read = 0;
5702122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        if (!ReadFile(read_pipe.get(), buf, sizeof(buf), &bytes_read, NULL)) {
5712122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            const DWORD err = GetLastError();
5722122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            // ERROR_BROKEN_PIPE is expected when the subprocess closes
5732122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            // the other end of the pipe.
5742122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            if (err == ERROR_BROKEN_PIPE) {
5752122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                return EXIT_SUCCESS;
5762122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            } else {
5772122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                fprintf(stderr, "Failed to read from %s: %s\n", output_name,
5785f787ed2b3b9f6cc02aa5923b95d77e2a5865438David Pursell                        android::base::SystemErrorCodeToString(err).c_str());
5792122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                return EXIT_FAILURE;
5802122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            }
5812122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        }
5822122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
58353a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        // Don't try to write if our stdout/stderr was not setup by the parent process.
58453a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low        if (stream) {
58553a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            // fwrite() actually calls adb_fwrite() which can write UTF-8 to the console.
58653a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low            const size_t bytes_written = fwrite(buf, 1, bytes_read, stream.get());
5872122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            if (bytes_written != bytes_read) {
58853a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low                fprintf(stderr, "Only wrote %zu of %lu bytes to %s\n", bytes_written, bytes_read,
58953a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low                        output_name);
5902122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                return EXIT_FAILURE;
5912122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            }
5922122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        }
5932122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
5942122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low}
5952122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
5962122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Lowstatic unsigned __stdcall _redirect_stdout_thread(HANDLE h) {
59749ee7cf9a10f66ffa9a3490c69db5fa46e0a966bSiva Velusamy    adb_thread_setname("stdout redirect");
5982122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    return _redirect_pipe_thread(h, STD_OUTPUT_HANDLE);
5992122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low}
6002122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
6012122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Lowstatic unsigned __stdcall _redirect_stderr_thread(HANDLE h) {
60249ee7cf9a10f66ffa9a3490c69db5fa46e0a966bSiva Velusamy    adb_thread_setname("stderr redirect");
6032122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    return _redirect_pipe_thread(h, STD_ERROR_HANDLE);
6042122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low}
6052122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
6062122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low#endif
6072122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
608a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzingerint launch_server(int server_port)
609dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
610e77b6a08623bba383ce55cd6653bec76cdf57792Yabin Cui#if defined(_WIN32)
611dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* we need to start the server in the background                    */
612dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* we create a PIPE that will be used to wait for the server's "OK" */
613dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* message since the pipe handles must be inheritable, we use a     */
614dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* security attribute                                               */
615dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    SECURITY_ATTRIBUTES   sa;
616dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    sa.nLength = sizeof(sa);
617dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    sa.lpSecurityDescriptor = NULL;
618dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    sa.bInheritHandle = TRUE;
619dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
6202122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // Redirect stdin to Windows /dev/null. If we instead pass an original
6212122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // stdin/stdout/stderr handle and it is a console handle, when the adb
6222122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // server starts up, the C Runtime will see a console handle for a process
6232122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // that isn't connected to a console and it will configure
6242122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // stdin/stdout/stderr to be closed. At that point, freopen() could be used
6252122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // to reopen stderr/out, but it would take more massaging to fixup the file
6262122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // descriptor number that freopen() uses. It's simplest to avoid all of this
6272122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // complexity by just redirecting stdin to `nul' and then the C Runtime acts
6282122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // as expected.
6292122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    unique_handle   nul_read(CreateFileW(L"nul", GENERIC_READ,
6302122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            FILE_SHARE_READ | FILE_SHARE_WRITE, &sa, OPEN_EXISTING,
6312122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            FILE_ATTRIBUTE_NORMAL, NULL));
6322122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (nul_read.get() == INVALID_HANDLE_VALUE) {
6332122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        fprintf(stderr, "Cannot open 'nul': %s\n",
6345f787ed2b3b9f6cc02aa5923b95d77e2a5865438David Pursell                android::base::SystemErrorCodeToString(GetLastError()).c_str());
635d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low        return -1;
636d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    }
637d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low
63853a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    // Create pipes with non-inheritable read handle, inheritable write handle. We need to connect
63953a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    // the subprocess to pipes instead of just letting the subprocess inherit our existing
64053a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    // stdout/stderr handles because a DETACHED_PROCESS cannot write to a console that it is not
64153a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    // attached to.
6422122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    unique_handle   ack_read, ack_write;
6432122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (!_create_anonymous_pipe(&ack_read, &ack_write, &sa)) {
644d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low        return -1;
645d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low    }
6462122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    unique_handle   stdout_read, stdout_write;
6472122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (!_create_anonymous_pipe(&stdout_read, &stdout_write, &sa)) {
6482122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        return -1;
6492122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
6502122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    unique_handle   stderr_read, stderr_write;
6512122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (!_create_anonymous_pipe(&stderr_read, &stderr_write, &sa)) {
652dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return -1;
653dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
654dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
655267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    /* Some programs want to launch an adb command and collect its output by
656267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * calling CreateProcess with inheritable stdout/stderr handles, then
657267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * using read() to get its output. When this happens, the stdout/stderr
658267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * handles passed to the adb client process will also be inheritable.
659267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * When starting the adb server here, care must be taken to reset them
660267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * to non-inheritable.
661267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * Otherwise, something bad happens: even if the adb command completes,
662267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * the calling process is stuck while read()-ing from the stdout/stderr
663267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * descriptors, because they're connected to corresponding handles in the
664267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * adb server process (even if the latter never uses/writes to them).
6652122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low     * Note that even if we don't pass these handles in the STARTUPINFO struct,
6662122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low     * if they're marked inheritable, they're still inherited, requiring us to
6672122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low     * deal with this.
6682122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low     *
6692122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low     * If we're still having problems with inheriting random handles in the
6702122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low     * future, consider using PROC_THREAD_ATTRIBUTE_HANDLE_LIST to explicitly
6712122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low     * specify which handles should be inherited: http://blogs.msdn.com/b/oldnewthing/archive/2011/12/16/10248328.aspx
672bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low     *
673bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low     * Older versions of Windows return console pseudo-handles that cannot be
674bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low     * made non-inheritable, so ignore those failures.
675267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     */
676bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low    _try_make_handle_noninheritable(GetStdHandle(STD_INPUT_HANDLE));
677bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low    _try_make_handle_noninheritable(GetStdHandle(STD_OUTPUT_HANDLE));
678bc36bdbdc8bb64011bc573a14d54576946fa1006Spencer Low    _try_make_handle_noninheritable(GetStdHandle(STD_ERROR_HANDLE));
679267aa8b00eadf273321e492026af74b7baad890eRay Donnelly
6802122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    STARTUPINFOW    startup;
681dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    ZeroMemory( &startup, sizeof(startup) );
682dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    startup.cb = sizeof(startup);
6832122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    startup.hStdInput  = nul_read.get();
6842122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    startup.hStdOutput = stdout_write.get();
6852122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    startup.hStdError  = stderr_write.get();
686dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    startup.dwFlags    = STARTF_USESTDHANDLES;
687dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
6885c398d2ce96150e94e596e959bfe2246b94de660Spencer Low    // Verify that the pipe_write handle value can be passed on the command line
6895c398d2ce96150e94e596e959bfe2246b94de660Spencer Low    // as %d and that the rest of adb code can pass it around in an int.
6902122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    const int ack_write_as_int = cast_handle_to_int(ack_write.get());
6912122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (cast_int_to_handle(ack_write_as_int) != ack_write.get()) {
6925c398d2ce96150e94e596e959bfe2246b94de660Spencer Low        // If this fires, either handle values are larger than 32-bits or else
6935c398d2ce96150e94e596e959bfe2246b94de660Spencer Low        // there is a bug in our casting.
6945c398d2ce96150e94e596e959bfe2246b94de660Spencer Low        // https://msdn.microsoft.com/en-us/library/windows/desktop/aa384203%28v=vs.85%29.aspx
6952122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        fprintf(stderr, "Cannot fit pipe handle value into 32-bits: 0x%p\n",
6962122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                ack_write.get());
6975c398d2ce96150e94e596e959bfe2246b94de660Spencer Low        return -1;
6985c398d2ce96150e94e596e959bfe2246b94de660Spencer Low    }
6995c398d2ce96150e94e596e959bfe2246b94de660Spencer Low
7002122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // get path of current program
7012122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    WCHAR       program_path[MAX_PATH];
7022122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    const DWORD module_result = GetModuleFileNameW(NULL, program_path,
7032122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                                                   arraysize(program_path));
7042122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if ((module_result >= arraysize(program_path)) || (module_result == 0)) {
7052122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        // String truncation or some other error.
7062122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        fprintf(stderr, "Cannot get executable path: %s\n",
7075f787ed2b3b9f6cc02aa5923b95d77e2a5865438David Pursell                android::base::SystemErrorCodeToString(GetLastError()).c_str());
7082122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        return -1;
7092122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
7102122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
7112122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    WCHAR   args[64];
712cf4ff64f2d3cdc4ffa2f06e594d18a965f591a82Spencer Low    snwprintf(args, arraysize(args),
7135c398d2ce96150e94e596e959bfe2246b94de660Spencer Low              L"adb -P %d fork-server server --reply-fd %d", server_port,
7142122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low              ack_write_as_int);
7152122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
7162122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    PROCESS_INFORMATION   pinfo;
7172122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    ZeroMemory(&pinfo, sizeof(pinfo));
7182122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
7192122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (!CreateProcessW(
720dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            program_path,                              /* program path  */
721a09558c2e205771e6830eeb6241c5c0e3b1cf4ceWenhao Li            args,
722dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                                    /* the fork-server argument will set the
723dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                                       debug = 2 in the child           */
724dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                   /* process handle is not inheritable */
725dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                    /* thread handle is not inheritable */
726dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            TRUE,                          /* yes, inherit some handles */
727dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            DETACHED_PROCESS, /* the new process doesn't have a console */
728dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                     /* use parent's environment block */
729dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                    /* use parent's starting directory */
730dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            &startup,                 /* startup info, i.e. std handles */
7312122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            &pinfo )) {
7322122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        fprintf(stderr, "Cannot create process: %s\n",
7335f787ed2b3b9f6cc02aa5923b95d77e2a5865438David Pursell                android::base::SystemErrorCodeToString(GetLastError()).c_str());
7342122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        return -1;
7352122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
736dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
7372122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    unique_handle   process_handle(pinfo.hProcess);
7382122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    pinfo.hProcess = NULL;
7392122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
7402122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // Close handles that we no longer need to complete the rest.
7412122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    CloseHandle(pinfo.hThread);
7422122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    pinfo.hThread = NULL;
7432122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
7442122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    nul_read.reset();
7452122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    ack_write.reset();
7462122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    stdout_write.reset();
7472122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    stderr_write.reset();
7482122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
74953a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    // Start threads to read from subprocess stdout/stderr and write to ours to make subprocess
75053a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    // errors easier to diagnose. Note that the threads internally create inheritable handles, but
75153a0a99fe8197984acd5ed6324e945c09772a73eSpencer Low    // that is ok because we've already spawned the subprocess.
7522122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
7532122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // In the past, reading from a pipe before the child process's C Runtime
7542122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // started up and called GetFileType() caused a hang: http://blogs.msdn.com/b/oldnewthing/archive/2011/12/02/10243553.aspx#10244216
7552122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // This is reportedly fixed in Windows Vista: https://support.microsoft.com/en-us/kb/2009703
7562122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // I was unable to reproduce the problem on Windows XP. It sounds like a
7572122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // Windows Update may have fixed this: https://www.duckware.com/tech/peeknamedpipe.html
7582122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    unique_handle   stdout_thread(reinterpret_cast<HANDLE>(
7592122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            _beginthreadex(NULL, 0, _redirect_stdout_thread, stdout_read.get(),
7602122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                           0, NULL)));
7612122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (stdout_thread.get() == nullptr) {
7622122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        fprintf(stderr, "Cannot create thread: %s\n", strerror(errno));
7632122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        return -1;
7642122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
7652122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    stdout_read.release();  // Transfer ownership to new thread
766dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
7672122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    unique_handle   stderr_thread(reinterpret_cast<HANDLE>(
7682122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            _beginthreadex(NULL, 0, _redirect_stderr_thread, stderr_read.get(),
7692122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                           0, NULL)));
7702122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (stderr_thread.get() == nullptr) {
7712122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        fprintf(stderr, "Cannot create thread: %s\n", strerror(errno));
772dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return -1;
773dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
7742122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    stderr_read.release();  // Transfer ownership to new thread
775dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
7762122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    bool    got_ack = false;
777dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
7782122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // Wait for the "OK\n" message, for the pipe to be closed, or other error.
779dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    {
7802122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        char    temp[3];
7812122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        DWORD   count = 0;
7822122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
7832122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        if (ReadFile(ack_read.get(), temp, sizeof(temp), &count, NULL)) {
7842122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            const CHAR  expected[] = "OK\n";
7852122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            const DWORD expected_length = arraysize(expected) - 1;
7862122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            if (count == expected_length &&
7872122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                memcmp(temp, expected, expected_length) == 0) {
7882122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                got_ack = true;
7892122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            } else {
7902122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                fprintf(stderr, "ADB server didn't ACK\n");
7912122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            }
7922122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        } else {
7932122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            const DWORD err = GetLastError();
7942122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            // If the ACK was not written and the process exited, GetLastError()
7952122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            // is probably ERROR_BROKEN_PIPE, in which case that info is not
7962122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            // useful to the user.
7972122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            fprintf(stderr, "could not read ok from ADB Server%s\n",
7982122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                    err == ERROR_BROKEN_PIPE ? "" :
7992122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                    android::base::StringPrintf(": %s",
8005f787ed2b3b9f6cc02aa5923b95d77e2a5865438David Pursell                            android::base::SystemErrorCodeToString(err).c_str()).c_str());
801dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
8022122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
8032122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
8042122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // Always try to wait a bit for threads reading stdout/stderr to finish.
8052122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // If the process started ok, it should close the pipes causing the threads
8062122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // to finish. If the process had an error, it should exit, also causing
8072122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // the pipes to be closed. In that case we want to read all of the output
8082122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // and write it out so that the user can diagnose failures.
8092122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    const DWORD     thread_timeout_ms = 15 * 1000;
8102122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    const HANDLE    threads[] = { stdout_thread.get(), stderr_thread.get() };
8112122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    const DWORD     wait_result = WaitForMultipleObjects(arraysize(threads),
8122122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            threads, TRUE, thread_timeout_ms);
8132122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (wait_result == WAIT_TIMEOUT) {
8142122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        // Threads did not finish after waiting a little while. Perhaps the
8152122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        // server didn't close pipes, or it is hung.
8162122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        fprintf(stderr, "Timed-out waiting for threads to finish reading from "
8172122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                "ADB Server\n");
8182122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        // Process handles are signaled when the process exits, so if we wait
8192122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        // on the handle for 0 seconds and it returns 'timeout', that means that
8202122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        // the process is still running.
8212122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        if (WaitForSingleObject(process_handle.get(), 0) == WAIT_TIMEOUT) {
8222122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            // We could TerminateProcess(), but that seems somewhat presumptive.
8232122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low            fprintf(stderr, "ADB Server is running: process id %lu\n",
8242122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low                    pinfo.dwProcessId);
825dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
8262122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        return -1;
8272122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
8282122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
8292122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (wait_result != WAIT_OBJECT_0) {
8302122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        fprintf(stderr, "Unexpected result waiting for threads: %lu: %s\n",
8315f787ed2b3b9f6cc02aa5923b95d77e2a5865438David Pursell                wait_result, android::base::SystemErrorCodeToString(GetLastError()).c_str());
8322122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        return -1;
8332122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    }
8342122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
8352122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    // For now ignore the thread exit codes and assume they worked properly.
8362122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low
8372122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low    if (!got_ack) {
8382122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low        return -1;
839dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
840e77b6a08623bba383ce55cd6653bec76cdf57792Yabin Cui#else /* !defined(_WIN32) */
841dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char    path[PATH_MAX];
842dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    int     fd[2];
843dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
844dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // set up a pipe so the child can tell us when it is ready.
8459f2d1a9cfc04e1d5970823da1878097288a9a9cdSiva Velusamy    // fd[0] will be parent's end, and the child will write on fd[1]
846dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (pipe(fd)) {
847dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "pipe failed in launch_server, errno: %d\n", errno);
848dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return -1;
849dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
8503166410a82f43d39201be98a8d35c51baa86cb53Alexey Tarasov    get_my_path(path, PATH_MAX);
851dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    pid_t pid = fork();
852dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(pid < 0) return -1;
853dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
854dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (pid == 0) {
855dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // child side of the fork
856dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
857dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[0]);
858dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
859d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel        char str_port[30];
8609f2d1a9cfc04e1d5970823da1878097288a9a9cdSiva Velusamy        snprintf(str_port, sizeof(str_port), "%d", server_port);
8619f2d1a9cfc04e1d5970823da1878097288a9a9cdSiva Velusamy        char reply_fd[30];
8629f2d1a9cfc04e1d5970823da1878097288a9a9cdSiva Velusamy        snprintf(reply_fd, sizeof(reply_fd), "%d", fd[1]);
863dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // child process
8649f2d1a9cfc04e1d5970823da1878097288a9a9cdSiva Velusamy        int result = execl(path, "adb", "-P", str_port, "fork-server", "server", "--reply-fd", reply_fd, NULL);
865dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // this should not return
866dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "OOPS! execl returned %d, errno: %d\n", result, errno);
867dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } else  {
868dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // parent side of the fork
869dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
870dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        char  temp[3];
871dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
872dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        temp[0] = 'A'; temp[1] = 'B'; temp[2] = 'C';
873dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // wait for the "OK\n" message
874dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[1]);
875dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int ret = adb_read(fd[0], temp, 3);
876408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall        int saved_errno = errno;
877dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[0]);
878dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (ret < 0) {
879408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall            fprintf(stderr, "could not read ok from ADB Server, errno = %d\n", saved_errno);
880dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return -1;
881dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
882dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (ret != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
883dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fprintf(stderr, "ADB server didn't ACK\n" );
884dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return -1;
885dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
886dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
887e77b6a08623bba383ce55cd6653bec76cdf57792Yabin Cui#endif /* !defined(_WIN32) */
888dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return 0;
889dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
890e77b6a08623bba383ce55cd6653bec76cdf57792Yabin Cui#endif /* ADB_HOST */
891dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
892252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner// Try to handle a network forwarding request.
893252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner// This returns 1 on success, 0 on failure, and -1 to indicate this is not
894252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner// a forwarding-related request.
895aee80fb67bcc3fc5209050c40bd8efc1bbcc0610Elliott Hughesint handle_forward_request(const char* service, TransportType type, const char* serial, int reply_fd)
896252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner{
897252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    if (!strcmp(service, "list-forward")) {
898252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        // Create the list of forward redirections.
899e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        std::string listeners = format_listeners();
900252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#if ADB_HOST
901e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendOkay(reply_fd);
902252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#endif
9037e6683ce40ba79c94ccc1e3b984cd39e63651dd5Josh Gao        return SendProtocolString(reply_fd, listeners);
904252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    }
905252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
906252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    if (!strcmp(service, "killforward-all")) {
907252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        remove_all_listeners();
908252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#if ADB_HOST
909252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        /* On the host: 1st OKAY is connect, 2nd OKAY is status */
910e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendOkay(reply_fd);
911252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#endif
912e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendOkay(reply_fd);
913252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        return 1;
914252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    }
915252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
916424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes    if (!strncmp(service, "forward:", 8) || !strncmp(service, "killforward:", 12)) {
917424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        // killforward:local
918424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        // forward:(norebind:)?local;remote
919424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        bool kill_forward = false;
920424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        bool no_rebind = false;
921424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        if (android::base::StartsWith(service, "killforward:")) {
922424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            kill_forward = true;
923424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            service += 12;
924a6b340af123b82e3aaa33cb2a0d45f637eeb4759Spencer Low        } else {
925a6b340af123b82e3aaa33cb2a0d45f637eeb4759Spencer Low            service += 8;   // skip past "forward:"
926424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            if (android::base::StartsWith(service, "norebind:")) {
927424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes                no_rebind = true;
928424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes                service += 9;
929424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            }
930252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
931252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
932424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        std::vector<std::string> pieces = android::base::Split(service, ";");
933252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
934424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        if (kill_forward) {
935424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            // Check killforward: parameter format: '<local>'
936424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            if (pieces.size() != 1 || pieces[0].empty()) {
937424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes                SendFail(reply_fd, android::base::StringPrintf("bad killforward: %s", service));
938252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                return 1;
939252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            }
940252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        } else {
941424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            // Check forward: parameter format: '<local>;<remote>'
942424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            if (pieces.size() != 2 || pieces[0].empty() || pieces[1].empty() || pieces[1][0] == '*') {
943424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes                SendFail(reply_fd, android::base::StringPrintf("bad forward: %s", service));
944252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                return 1;
945252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            }
946252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
947252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
9487be29c819b4fb8bf9b1d4b69a4333f8765de0281Elliott Hughes        std::string error_msg;
9498d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        atransport* transport = acquire_one_transport(type, serial, nullptr, &error_msg);
950252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        if (!transport) {
951e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            SendFail(reply_fd, error_msg);
952252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            return 1;
953252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
954252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
9555200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low        std::string error;
9563bd73c12c07dcefc965abeef535ac53c4754d682Elliott Hughes        InstallStatus r;
957424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes        if (kill_forward) {
958424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            r = remove_listener(pieces[0].c_str(), transport);
959252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        } else {
9605200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low            r = install_listener(pieces[0], pieces[1].c_str(), transport,
9615200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low                                 no_rebind, &error);
962252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
9637b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes        if (r == INSTALL_STATUS_OK) {
964252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#if ADB_HOST
965252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            /* On the host: 1st OKAY is connect, 2nd OKAY is status */
966e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            SendOkay(reply_fd);
967252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#endif
968e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            SendOkay(reply_fd);
969252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            return 1;
970252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
971252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
9727b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes        std::string message;
9737b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes        switch (r) {
974424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes          case INSTALL_STATUS_OK: message = "success (!)"; break;
9757b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes          case INSTALL_STATUS_INTERNAL_ERROR: message = "internal error"; break;
9767b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes          case INSTALL_STATUS_CANNOT_BIND:
977bf7c605d87f87c03066c384cecb0f0c91aa31403Spencer Low            message = android::base::StringPrintf("cannot bind listener: %s",
9785200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low                                                  error.c_str());
9797b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes            break;
9807b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes          case INSTALL_STATUS_CANNOT_REBIND:
981a6b340af123b82e3aaa33cb2a0d45f637eeb4759Spencer Low            message = android::base::StringPrintf("cannot rebind existing socket");
9827b506090e1ac51f4990f17621c6e33847b0632a2Elliott Hughes            break;
983424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes          case INSTALL_STATUS_LISTENER_NOT_FOUND:
984424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            message = android::base::StringPrintf("listener '%s' not found", service);
985424af02f363a305a349ff99e1cc253ac5bc642c9Elliott Hughes            break;
986252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
987e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendFail(reply_fd, message);
988252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        return 1;
989252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    }
990252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    return 0;
991252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner}
992252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
993e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes#if ADB_HOST
994e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughesstatic int SendOkay(int fd, const std::string& s) {
995e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    SendOkay(fd);
996e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    SendProtocolString(fd, s);
997e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    return 0;
998e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes}
999e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes#endif
1000e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes
1001c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertint handle_host_request(const char* service, TransportType type,
1002c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert                        const char* serial, int reply_fd, asocket* s) {
1003c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert    if (strcmp(service, "kill") == 0) {
1004c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert        fprintf(stderr, "adb server killed by remote request\n");
1005dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fflush(stdout);
1006e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        SendOkay(reply_fd);
1007f18fc0879be9cd506be6b1e051feb543b0746bdbSpencer Low
1008351ecd15b29f0ce528ffac119640b9c01874562bSpencer Low        // On Windows, if the process exits with open sockets that
1009351ecd15b29f0ce528ffac119640b9c01874562bSpencer Low        // shutdown(SD_SEND) has not been called on, TCP RST segments will be
1010351ecd15b29f0ce528ffac119640b9c01874562bSpencer Low        // sent to the peers which will cause their next recv() to error-out
1011351ecd15b29f0ce528ffac119640b9c01874562bSpencer Low        // with WSAECONNRESET. In the case of this code, that means the client
1012351ecd15b29f0ce528ffac119640b9c01874562bSpencer Low        // may not read the OKAY sent above.
1013f18fc0879be9cd506be6b1e051feb543b0746bdbSpencer Low        adb_shutdown(reply_fd);
1014f18fc0879be9cd506be6b1e051feb543b0746bdbSpencer Low
1015dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        exit(0);
1016dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1017dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1018dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if ADB_HOST
1019dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport:" is used for switching transport with a specified serial number
1020dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport-usb:" is used for switching transport to the only USB transport
1021dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport-local:" is used for switching transport to the only local transport
1022dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport-any:" is used for switching transport to the only transport
1023dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!strncmp(service, "transport", strlen("transport"))) {
10243bd73c12c07dcefc965abeef535ac53c4754d682Elliott Hughes        TransportType type = kTransportAny;
1025dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1026dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (!strncmp(service, "transport-usb", strlen("transport-usb"))) {
1027dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            type = kTransportUsb;
1028dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else if (!strncmp(service, "transport-local", strlen("transport-local"))) {
1029dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            type = kTransportLocal;
1030dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else if (!strncmp(service, "transport-any", strlen("transport-any"))) {
1031dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            type = kTransportAny;
1032dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else if (!strncmp(service, "transport:", strlen("transport:"))) {
1033dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            service += strlen("transport:");
10343175c8e9bfcb01f181d7300755da15a999e5461cTom Marlin            serial = service;
1035dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
1036dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
10378d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        std::string error;
10388d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        atransport* t = acquire_one_transport(type, serial, nullptr, &error);
1039e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        if (t != nullptr) {
1040e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes            s->transport = t;
1041e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            SendOkay(reply_fd);
1042dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else {
10438d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes            SendFail(reply_fd, error);
1044dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
1045dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return 1;
1046dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1047dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1048dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // return a list of all connected devices
1049e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson    if (!strncmp(service, "devices", 7)) {
1050e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        bool long_listing = (strcmp(service+7, "-l") == 0);
1051e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        if (long_listing || service[7] == 0) {
10527a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui            D("Getting device list...");
1053e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes            std::string device_list = list_transports(long_listing);
10547a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui            D("Sending device list...");
1055e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes            return SendOkay(reply_fd, device_list);
1056e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson        }
1057e67f1f87d9b1188ec8617035db7006c37ee7b21eElliott Hughes        return 1;
1058dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1059dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
106090d4b739ab94fded63086de1261e70a3d39ceb1dDan Albert    if (!strcmp(service, "features")) {
10618d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        std::string error;
10628d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        atransport* t = acquire_one_transport(type, serial, nullptr, &error);
1063880be434a9ec197100fe5c6f0b299353d13b916bDavid Pursell        if (t != nullptr) {
10644e2fd36bc8c16147cab323b0418a7666812d3bc7David Pursell            SendOkay(reply_fd, FeatureSetToString(t->features()));
1065880be434a9ec197100fe5c6f0b299353d13b916bDavid Pursell        } else {
10668d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes            SendFail(reply_fd, error);
1067880be434a9ec197100fe5c6f0b299353d13b916bDavid Pursell        }
1068880be434a9ec197100fe5c6f0b299353d13b916bDavid Pursell        return 0;
1069880be434a9ec197100fe5c6f0b299353d13b916bDavid Pursell    }
1070880be434a9ec197100fe5c6f0b299353d13b916bDavid Pursell
107174d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood    // remove TCP transport
107274d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood    if (!strncmp(service, "disconnect:", 11)) {
10733d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        const std::string address(service + 11);
10743d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        if (address.empty()) {
1075f4b9928563e97620fc1d9bd5c2efdaa0ded96488Yabin Cui            kick_all_tcp_devices();
10763d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes            return SendOkay(reply_fd, "disconnected everything");
107774d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood        }
107874d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood
10793d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        std::string serial;
10803d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        std::string host;
10813d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        int port = DEFAULT_ADB_LOCAL_TRANSPORT_PORT;
10823d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        std::string error;
1083706955ff0d159d28bb6eb06dc10178624c505f4eDavid Pursell        if (!android::base::ParseNetAddress(address, &host, &port, &serial, &error)) {
10843d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes            return SendFail(reply_fd, android::base::StringPrintf("couldn't parse '%s': %s",
10853d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes                                                                  address.c_str(), error.c_str()));
10863d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        }
10873d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        atransport* t = find_transport(serial.c_str());
10883d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        if (t == nullptr) {
10893d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes            return SendFail(reply_fd, android::base::StringPrintf("no such device '%s'",
10903d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes                                                                  serial.c_str()));
10913d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        }
1092f4b9928563e97620fc1d9bd5c2efdaa0ded96488Yabin Cui        kick_transport(t);
10933d5f60dbba3bc0feabc05457d181547b295bb3b2Elliott Hughes        return SendOkay(reply_fd, android::base::StringPrintf("disconnected %s", address.c_str()));
10942f38b699713dc2587a771bd5d4c6a47329728f5eMike Lockwood    }
10952f38b699713dc2587a771bd5d4c6a47329728f5eMike Lockwood
10968d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes    // Returns our value for ADB_SERVER_VERSION.
1097dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!strcmp(service, "version")) {
1098e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes        return SendOkay(reply_fd, android::base::StringPrintf("%04x", ADB_SERVER_VERSION));
1099dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1100dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1101e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    // These always report "unknown" rather than the actual error, for scripts.
1102e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    if (!strcmp(service, "get-serialno")) {
11038d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        std::string error;
11048d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        atransport* t = acquire_one_transport(type, serial, nullptr, &error);
11058d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        if (t) {
11068d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes            return SendOkay(reply_fd, t->serial ? t->serial : "unknown");
11078d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        } else {
11088d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes            return SendFail(reply_fd, error);
11098d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        }
1110dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1111e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    if (!strcmp(service, "get-devpath")) {
11128d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        std::string error;
11138d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        atransport* t = acquire_one_transport(type, serial, nullptr, &error);
11148d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        if (t) {
11158d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes            return SendOkay(reply_fd, t->devpath ? t->devpath : "unknown");
11168d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        } else {
11178d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes            return SendFail(reply_fd, error);
11188d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        }
1119e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson    }
1120e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    if (!strcmp(service, "get-state")) {
11218d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        std::string error;
11228d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        atransport* t = acquire_one_transport(type, serial, nullptr, &error);
11238d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        if (t) {
11248d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes            return SendOkay(reply_fd, t->connection_state_name());
11258d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        } else {
11268d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes            return SendFail(reply_fd, error);
11278d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes        }
1128e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    }
1129e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes
11308d28e191c5ba81b82e51e3fd120d03851e1d905fElliott Hughes    // Indicates a new emulator instance has started.
1131e2d3677cc2239a8530d33e3c5e24e65fff096801Elliott Hughes    if (!strncmp(service, "emulator:", 9)) {
1132dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int  port = atoi(service+9);
1133dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        local_connect(port);
1134dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        /* we don't even need to send a reply */
1135dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return 0;
1136dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
11371f4ec19e499ba981e4117f647d191603c2713e79Yabin Cui
11381f4ec19e499ba981e4117f647d191603c2713e79Yabin Cui    if (!strcmp(service, "reconnect")) {
11391f4ec19e499ba981e4117f647d191603c2713e79Yabin Cui        if (s->transport != nullptr) {
11401f4ec19e499ba981e4117f647d191603c2713e79Yabin Cui            kick_transport(s->transport);
11411f4ec19e499ba981e4117f647d191603c2713e79Yabin Cui        }
11421f4ec19e499ba981e4117f647d191603c2713e79Yabin Cui        return SendOkay(reply_fd, "done");
11431f4ec19e499ba981e4117f647d191603c2713e79Yabin Cui    }
1144dc22c3c7a82ad1919c52dc4d00c601ad7210c61bSimon Ye#endif // ADB_HOST
1145dc22c3c7a82ad1919c52dc4d00c601ad7210c61bSimon Ye
11463bd73c12c07dcefc965abeef535ac53c4754d682Elliott Hughes    int ret = handle_forward_request(service, type, serial, reply_fd);
1147dc22c3c7a82ad1919c52dc4d00c601ad7210c61bSimon Ye    if (ret >= 0)
1148dc22c3c7a82ad1919c52dc4d00c601ad7210c61bSimon Ye      return ret - 1;
1149dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return -1;
1150dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
1151