adb.cpp revision 252586941934d23073a8d167ec240b221062505f
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
17dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#define  TRACE_TAG   TRACE_ADB
18dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
19dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <stdio.h>
20dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <stdlib.h>
21dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <ctype.h>
22dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <stdarg.h>
23dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <errno.h>
24c7993af64baec271a238646bc20aaa846866c4a9Scott Anderson#include <stddef.h>
25dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <string.h>
26dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <time.h>
271f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood#include <sys/time.h>
28cbb9891049628a2a89d566cade88c80ba1ba4a7bRay Donnelly#include <stdint.h>
29dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
30dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include "sysdeps.h"
31dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include "adb.h"
32d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby#include "adb_auth.h"
33dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
34e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))
35e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
36dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if !ADB_HOST
37893a4a47e8c9290128b9254af0246e36f821c260Nick Kralevich#include <cutils/properties.h>
38dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#include <private/android_filesystem_config.h>
39e2864bf727d6a404bd871f063c03e36cc8eedeecNick Kralevich#include <sys/capability.h>
405f4b051235c3e9e9e5b34b4af885e42a1c711fc4Mike Lockwood#include <linux/prctl.h>
41885342a0f2c834a6b680284047c47c9d04b32565Jeff Sharkey#include <sys/mount.h>
42d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich#include <getopt.h>
43d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich#include <selinux/selinux.h>
44a09fbd164d2e088bc5433d310e25640ae048d47dXavier Ducrohet#else
45a09fbd164d2e088bc5433d310e25640ae048d47dXavier Ducrohet#include "usb_vendors.h"
46dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
47dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
48408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall#if ADB_TRACE
49408fa57864c01113deaa213e5c1848a9c594ae92JP AbgrallADB_MUTEX_DEFINE( D_lock );
50408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall#endif
51dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
52dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectint HOST = 0;
53d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbelint gListenAll = 0;
54dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
55d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Gobystatic int auth_enabled = 0;
56d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
57e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#if !ADB_HOST
58dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic const char *adb_device_banner = "device";
59d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevichstatic const char *root_seclabel = NULL;
60e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#endif
61dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
62dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid fatal(const char *fmt, ...)
63dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
64dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_list ap;
65dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_start(ap, fmt);
66dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "error: ");
67dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    vfprintf(stderr, fmt, ap);
68dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "\n");
69dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_end(ap);
70dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    exit(-1);
71dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
72dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
73dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid fatal_errno(const char *fmt, ...)
74dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
75dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_list ap;
76dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_start(ap, fmt);
77dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "error: %s: ", strerror(errno));
78dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    vfprintf(stderr, fmt, ap);
79dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "\n");
80dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    va_end(ap);
81dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    exit(-1);
82dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
83dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
84dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectint   adb_trace_mask;
85dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
86dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project/* read a comma/space/colum/semi-column separated list of tags
87dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * from the ADB_TRACE environment variable and build the trace
88dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * mask from it. note that '1' and 'all' are special cases to
89dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project * enable all tracing
90dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project */
91dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid  adb_trace_init(void)
92dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
93dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    const char*  p = getenv("ADB_TRACE");
94dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    const char*  q;
95dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
96dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    static const struct {
97dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        const char*  tag;
98dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int           flag;
99dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } tags[] = {
100dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "1", 0 },
101dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "all", 0 },
102dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "adb", TRACE_ADB },
103dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "sockets", TRACE_SOCKETS },
104dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "packets", TRACE_PACKETS },
105dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "rwx", TRACE_RWX },
106dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "usb", TRACE_USB },
107dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "sync", TRACE_SYNC },
108dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "sysdeps", TRACE_SYSDEPS },
109dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "transport", TRACE_TRANSPORT },
110dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { "jdwp", TRACE_JDWP },
111408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall        { "services", TRACE_SERVICES },
112d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        { "auth", TRACE_AUTH },
113dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        { NULL, 0 }
114dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    };
115dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
116dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (p == NULL)
117dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return;
118dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
119dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* use a comma/column/semi-colum/space separated list */
120dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    while (*p) {
121dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int  len, tagn;
122dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
123dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        q = strpbrk(p, " ,:;");
124dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (q == NULL) {
125dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            q = p + strlen(p);
126dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
127dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        len = q - p;
128dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
129dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        for (tagn = 0; tags[tagn].tag != NULL; tagn++)
130dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        {
131dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            int  taglen = strlen(tags[tagn].tag);
132dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
133dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            if (len == taglen && !memcmp(tags[tagn].tag, p, len) )
134dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            {
135dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                int  flag = tags[tagn].flag;
136dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                if (flag == 0) {
137dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    adb_trace_mask = ~0;
138dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    return;
139dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                }
140dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                adb_trace_mask |= (1 << flag);
141dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                break;
142dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
143dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
144dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        p = q;
145dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (*p)
146dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            p++;
147dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
148dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
149dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
15028781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#if !ADB_HOST
15128781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine/*
15228781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine * Implements ADB tracing inside the emulator.
15328781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine */
15428781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine
15528781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#include <stdarg.h>
15628781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine
15728781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine/*
15828781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine * Redefine open and write for qemu_pipe.h that contains inlined references
15928781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine * to those routines. We will redifine them back after qemu_pipe.h inclusion.
16028781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine */
16128781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine
16228781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#undef open
16328781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#undef write
16428781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#define open    adb_open
16528781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#define write   adb_write
16628781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#include <hardware/qemu_pipe.h>
16728781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#undef open
16828781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#undef write
16928781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#define open    ___xxx_open
17028781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#define write   ___xxx_write
17128781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine
17228781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine/* A handle to adb-debug qemud service in the emulator. */
17328781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkineint   adb_debug_qemu = -1;
17428781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine
17528781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine/* Initializes connection with the adb-debug qemud service in the emulator. */
17628781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkinestatic int adb_qemu_trace_init(void)
17728781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine{
17828781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    char con_name[32];
17928781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine
18028781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    if (adb_debug_qemu >= 0) {
18128781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine        return 0;
18228781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    }
18328781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine
18428781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    /* adb debugging QEMUD service connection request. */
18528781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    snprintf(con_name, sizeof(con_name), "qemud:adb-debug");
18628781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    adb_debug_qemu = qemu_pipe_open(con_name);
18728781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    return (adb_debug_qemu >= 0) ? 0 : -1;
18828781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine}
18928781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine
19028781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkinevoid adb_qemu_trace(const char* fmt, ...)
19128781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine{
19228781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    va_list args;
19328781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    va_start(args, fmt);
19428781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    char msg[1024];
19528781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine
19628781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    if (adb_debug_qemu >= 0) {
19728781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine        vsnprintf(msg, sizeof(msg), fmt, args);
19828781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine        adb_write(adb_debug_qemu, msg, strlen(msg));
19928781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    }
20028781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine}
20128781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine#endif  /* !ADB_HOST */
202dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
203dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectapacket *get_apacket(void)
204dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
205dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    apacket *p = malloc(sizeof(apacket));
206dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(p == 0) fatal("failed to allocate an apacket");
207dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    memset(p, 0, sizeof(apacket) - MAX_PAYLOAD);
208dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return p;
209dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
210dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
211dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid put_apacket(apacket *p)
212dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
213dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    free(p);
214dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
215dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
216d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Gobyvoid handle_online(atransport *t)
217dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
218dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("adb: online\n");
219d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    t->online = 1;
220dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
221dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
222dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid handle_offline(atransport *t)
223dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
224dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("adb: offline\n");
225dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    //Close the associated usb
226d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    t->online = 0;
227dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    run_transport_disconnects(t);
228dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
229dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
230d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby#if DEBUG_PACKETS
231dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#define DUMPMAX 32
232dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid print_packet(const char *label, apacket *p)
233dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
234dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char *tag;
235dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char *x;
236dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    unsigned count;
237dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
238dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    switch(p->msg.command){
239dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_SYNC: tag = "SYNC"; break;
240dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_CNXN: tag = "CNXN" ; break;
241dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OPEN: tag = "OPEN"; break;
242dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OKAY: tag = "OKAY"; break;
243dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_CLSE: tag = "CLSE"; break;
244dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_WRTE: tag = "WRTE"; break;
245d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    case A_AUTH: tag = "AUTH"; break;
246dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    default: tag = "????"; break;
247dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
248dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
249dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr, "%s: %s %08x %08x %04x \"",
250dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            label, tag, p->msg.arg0, p->msg.arg1, p->msg.data_length);
251dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    count = p->msg.data_length;
252dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    x = (char*) p->data;
253dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(count > DUMPMAX) {
254dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        count = DUMPMAX;
255dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        tag = "\n";
256dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } else {
257dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        tag = "\"\n";
258dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
259dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    while(count-- > 0){
260dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if((*x >= ' ') && (*x < 127)) {
261dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fputc(*x, stderr);
262dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else {
263dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fputc('.', stderr);
264dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
265dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        x++;
266dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
267d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    fputs(tag, stderr);
268dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
269dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
270dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
271dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void send_ready(unsigned local, unsigned remote, atransport *t)
272dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
273dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("Calling send_ready \n");
274dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    apacket *p = get_apacket();
275dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.command = A_OKAY;
276dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg0 = local;
277dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg1 = remote;
278dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    send_packet(p, t);
279dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
280dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
281dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void send_close(unsigned local, unsigned remote, atransport *t)
282dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
283dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("Calling send_close \n");
284dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    apacket *p = get_apacket();
285dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.command = A_CLSE;
286dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg0 = local;
287dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    p->msg.arg1 = remote;
288dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    send_packet(p, t);
289dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
290dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
291e82c2db05cae70a0490a1f84b7211ef42c329671Scott Andersonstatic size_t fill_connect_data(char *buf, size_t bufsize)
292e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson{
293e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#if ADB_HOST
294e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    return snprintf(buf, bufsize, "host::") + 1;
295e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#else
296e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    static const char *cnxn_props[] = {
297e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        "ro.product.name",
298e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        "ro.product.model",
299e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        "ro.product.device",
300e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    };
301e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    static const int num_cnxn_props = ARRAY_SIZE(cnxn_props);
302e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    int i;
303e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    size_t remaining = bufsize;
304e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    size_t len;
305e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
306e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    len = snprintf(buf, remaining, "%s::", adb_device_banner);
307e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    remaining -= len;
308e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    buf += len;
309e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    for (i = 0; i < num_cnxn_props; i++) {
310e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        char value[PROPERTY_VALUE_MAX];
311e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        property_get(cnxn_props[i], value, "");
312e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        len = snprintf(buf, remaining, "%s=%s;", cnxn_props[i], value);
313e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        remaining -= len;
314e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        buf += len;
315e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    }
316e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
317e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    return bufsize - remaining + 1;
318e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson#endif
319e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson}
320e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
321252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#if !ADB_HOST
322252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turnerstatic void send_msg_with_header(int fd, const char* msg, size_t msglen) {
323252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    char header[5];
324252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    if (msglen > 0xffff)
325252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        msglen = 0xffff;
326252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    snprintf(header, sizeof(header), "%04x", (unsigned)msglen);
327252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    writex(fd, header, 4);
328252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    writex(fd, msg, msglen);
329252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner}
330252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#endif
331252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
332252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turnerstatic void send_msg_with_okay(int fd, const char* msg, size_t msglen) {
3332264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow    char header[9];
3342264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow    if (msglen > 0xffff)
3352264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow        msglen = 0xffff;
3362264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow    snprintf(header, sizeof(header), "OKAY%04x", (unsigned)msglen);
3372264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow    writex(fd, header, 8);
3382264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow    writex(fd, msg, msglen);
3392264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow}
3402264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow
341dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void send_connect(atransport *t)
342dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
343dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("Calling send_connect \n");
344dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    apacket *cp = get_apacket();
345dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    cp->msg.command = A_CNXN;
346dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    cp->msg.arg0 = A_VERSION;
347dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    cp->msg.arg1 = MAX_PAYLOAD;
348e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    cp->msg.data_length = fill_connect_data((char *)cp->data,
349e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                                            sizeof(cp->data));
350dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    send_packet(cp, t);
351d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby}
352d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
353045a4a9c3a317e60ab1078f5a689bf302ee2ab19Benoit Gobyvoid send_auth_request(atransport *t)
354d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby{
355d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    D("Calling send_auth_request\n");
356d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    apacket *p;
357d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    int ret;
358d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
359d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    ret = adb_auth_generate_token(t->token, sizeof(t->token));
360d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    if (ret != sizeof(t->token)) {
361d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        D("Error generating token ret=%d\n", ret);
362d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        return;
363d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    }
364d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
365d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    p = get_apacket();
366d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    memcpy(p->data, t->token, ret);
367d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    p->msg.command = A_AUTH;
368d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    p->msg.arg0 = ADB_AUTH_TOKEN;
369d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    p->msg.data_length = ret;
370d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    send_packet(p, t);
371d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby}
372d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
373d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Gobystatic void send_auth_response(uint8_t *token, size_t token_size, atransport *t)
374d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby{
375d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    D("Calling send_auth_response\n");
376d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    apacket *p = get_apacket();
377d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    int ret;
378d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
379d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    ret = adb_auth_sign(t->key, token, token_size, p->data);
380d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    if (!ret) {
381d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        D("Error signing the token\n");
382d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        put_apacket(p);
383d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        return;
384d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    }
385d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
386d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    p->msg.command = A_AUTH;
387d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    p->msg.arg0 = ADB_AUTH_SIGNATURE;
388d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    p->msg.data_length = ret;
389d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    send_packet(p, t);
390d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby}
391d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
392d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Gobystatic void send_auth_publickey(atransport *t)
393d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby{
394d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    D("Calling send_auth_publickey\n");
395d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    apacket *p = get_apacket();
396d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    int ret;
397d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
398d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    ret = adb_auth_get_userkey(p->data, sizeof(p->data));
399d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    if (!ret) {
400d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        D("Failed to get user public key\n");
401d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        put_apacket(p);
402d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        return;
403d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    }
404d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
405d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    p->msg.command = A_AUTH;
406d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    p->msg.arg0 = ADB_AUTH_RSAPUBLICKEY;
407d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    p->msg.data_length = ret;
408d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    send_packet(p, t);
409d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby}
410d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
411d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Gobyvoid adb_auth_verified(atransport *t)
412d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby{
413d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    handle_online(t);
414d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    send_connect(t);
415dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
416dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
417dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic char *connection_state_name(atransport *t)
418dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
419dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (t == NULL) {
420dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return "unknown";
421dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
422dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
423dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    switch(t->connection_state) {
424dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case CS_BOOTLOADER:
425dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return "bootloader";
426dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case CS_DEVICE:
427dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return "device";
428a5ad539cff9dba92650425b8bec79c071ef5993btrevd    case CS_RECOVERY:
429a5ad539cff9dba92650425b8bec79c071ef5993btrevd        return "recovery";
430a5ad539cff9dba92650425b8bec79c071ef5993btrevd    case CS_SIDELOAD:
431a5ad539cff9dba92650425b8bec79c071ef5993btrevd        return "sideload";
432dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case CS_OFFLINE:
433dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return "offline";
43477e8e5851d3b951b5382663bd50f1f83cb531eb2Benoit Goby    case CS_UNAUTHORIZED:
43577e8e5851d3b951b5382663bd50f1f83cb531eb2Benoit Goby        return "unauthorized";
436dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    default:
437dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return "unknown";
438dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
439dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
440dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
441e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson/* qual_overwrite is used to overwrite a qualifier string.  dst is a
442e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson * pointer to a char pointer.  It is assumed that if *dst is non-NULL, it
4432ca3e6b35f79136418ebc32fef57580698dbd045Scott Anderson * was malloc'ed and needs to freed.  *dst will be set to a dup of src.
444e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson */
445e82c2db05cae70a0490a1f84b7211ef42c329671Scott Andersonstatic void qual_overwrite(char **dst, const char *src)
446e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson{
447e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    if (!dst)
448e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        return;
449e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
450e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    free(*dst);
451e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    *dst = NULL;
452e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
453e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    if (!src || !*src)
454e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        return;
455e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
456e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    *dst = strdup(src);
457e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson}
458e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson
459dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid parse_banner(char *banner, atransport *t)
460dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
461e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    static const char *prop_seps = ";";
462e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    static const char key_val_sep = '=';
4632ca3e6b35f79136418ebc32fef57580698dbd045Scott Anderson    char *cp;
4642ca3e6b35f79136418ebc32fef57580698dbd045Scott Anderson    char *type;
465dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
466dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    D("parse_banner: %s\n", banner);
467dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    type = banner;
468e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    cp = strchr(type, ':');
469e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson    if (cp) {
470e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        *cp++ = 0;
471e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        /* Nothing is done with second field. */
472e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        cp = strchr(cp, ':');
473e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        if (cp) {
474e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson            char *save;
475e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson            char *key;
4761b7a7e81195ff06a7482f81cb92b094bb3481cb1Scott Anderson            key = adb_strtok_r(cp + 1, prop_seps, &save);
477e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson            while (key) {
478e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                cp = strchr(key, key_val_sep);
479e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                if (cp) {
480e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                    *cp++ = '\0';
481e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                    if (!strcmp(key, "ro.product.name"))
482e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                        qual_overwrite(&t->product, cp);
483e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                    else if (!strcmp(key, "ro.product.model"))
484e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                        qual_overwrite(&t->model, cp);
485e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                    else if (!strcmp(key, "ro.product.device"))
486e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                        qual_overwrite(&t->device, cp);
487e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson                }
4881b7a7e81195ff06a7482f81cb92b094bb3481cb1Scott Anderson                key = adb_strtok_r(NULL, prop_seps, &save);
489e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson            }
490e82c2db05cae70a0490a1f84b7211ef42c329671Scott Anderson        }
491dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
492dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
493dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(!strcmp(type, "bootloader")){
494dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        D("setting connection_state to CS_BOOTLOADER\n");
495dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        t->connection_state = CS_BOOTLOADER;
496dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        update_transports();
497dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return;
498dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
499dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
500dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(!strcmp(type, "device")) {
501dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        D("setting connection_state to CS_DEVICE\n");
502dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        t->connection_state = CS_DEVICE;
503dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        update_transports();
504dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return;
505dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
506dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
507dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(!strcmp(type, "recovery")) {
508dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        D("setting connection_state to CS_RECOVERY\n");
509dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        t->connection_state = CS_RECOVERY;
510dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        update_transports();
511dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return;
512dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
513dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
514447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker    if(!strcmp(type, "sideload")) {
515447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker        D("setting connection_state to CS_SIDELOAD\n");
516447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker        t->connection_state = CS_SIDELOAD;
517447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker        update_transports();
518447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker        return;
519447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker    }
520447f061da19fe46bae35f1cdd93eeb16bc225463Doug Zongker
521dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    t->connection_state = CS_HOST;
522dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
523dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
524dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid handle_packet(apacket *p, atransport *t)
525dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
526dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    asocket *s;
527dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
528899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta    D("handle_packet() %c%c%c%c\n", ((char*) (&(p->msg.command)))[0],
529899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta            ((char*) (&(p->msg.command)))[1],
530899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta            ((char*) (&(p->msg.command)))[2],
531899913f8168b54e00971c0e8d4ae16d06a4651feViral Mehta            ((char*) (&(p->msg.command)))[3]);
532dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    print_packet("recv", p);
533dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
534dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    switch(p->msg.command){
535dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_SYNC:
536dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if(p->msg.arg0){
537dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            send_packet(p, t);
538dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            if(HOST) send_connect(t);
539dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else {
540dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            t->connection_state = CS_OFFLINE;
541dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            handle_offline(t);
542dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            send_packet(p, t);
543dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
544dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return;
545dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
546dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_CNXN: /* CONNECT(version, maxdata, "system-id-string") */
547dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            /* XXX verify version, etc */
548dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if(t->connection_state != CS_OFFLINE) {
549dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            t->connection_state = CS_OFFLINE;
550dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            handle_offline(t);
551dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
552d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
553dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        parse_banner((char*) p->data, t);
554d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
555d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        if (HOST || !auth_enabled) {
556d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            handle_online(t);
557d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            if(!HOST) send_connect(t);
558d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        } else {
559d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            send_auth_request(t);
560d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        }
561d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        break;
562d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby
563d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    case A_AUTH:
564d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        if (p->msg.arg0 == ADB_AUTH_TOKEN) {
56577e8e5851d3b951b5382663bd50f1f83cb531eb2Benoit Goby            t->connection_state = CS_UNAUTHORIZED;
566d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            t->key = adb_auth_nextkey(t->key);
567d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            if (t->key) {
568d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                send_auth_response(p->data, p->msg.data_length, t);
569d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            } else {
570d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                /* No more private keys to try, send the public key */
571d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                send_auth_publickey(t);
572d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            }
573d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        } else if (p->msg.arg0 == ADB_AUTH_SIGNATURE) {
574d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            if (adb_auth_verify(t->token, p->data, p->msg.data_length)) {
575d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                adb_auth_verified(t);
576d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                t->failed_auth_attempts = 0;
577d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            } else {
578d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                if (t->failed_auth_attempts++ > 10)
579d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                    adb_sleep_ms(1000);
580d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby                send_auth_request(t);
581d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            }
582d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        } else if (p->msg.arg0 == ADB_AUTH_RSAPUBLICKEY) {
583d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby            adb_auth_confirm_key(p->data, p->msg.data_length, t);
584d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        }
585dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
586dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
587dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OPEN: /* OPEN(local-id, 0, "destination") */
588818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg0 != 0 && p->msg.arg1 == 0) {
589dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            char *name = (char*) p->data;
590dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
591dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            s = create_local_service_socket(name);
592dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            if(s == 0) {
593dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                send_close(0, p->msg.arg0, t);
594dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            } else {
595dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                s->peer = create_remote_socket(p->msg.arg0, t);
596dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                s->peer->peer = s;
597dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                send_ready(s->id, s->peer->id, t);
598dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                s->ready(s);
599dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
600dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
601dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
602dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
603dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    case A_OKAY: /* READY(local-id, remote-id, "") */
604818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
605818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner            if((s = find_local_socket(p->msg.arg1, 0))) {
606dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                if(s->peer == 0) {
607818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    /* On first READY message, create the connection. */
608dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    s->peer = create_remote_socket(p->msg.arg0, t);
609dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    s->peer->peer = s;
610818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    s->ready(s);
611818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                } else if (s->peer->id == p->msg.arg0) {
612818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    /* Other READY messages must use the same local-id */
613818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    s->ready(s);
614818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                } else {
615818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    D("Invalid A_OKAY(%d,%d), expected A_OKAY(%d,%d) on transport %s\n",
616818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                      p->msg.arg0, p->msg.arg1, s->peer->id, p->msg.arg1, t->serial);
617dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                }
618dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
619dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
620dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
621dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
622818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner    case A_CLSE: /* CLOSE(local-id, remote-id, "") or CLOSE(0, remote-id, "") */
623818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg1 != 0) {
624818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner            if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
625818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                /* According to protocol.txt, p->msg.arg0 might be 0 to indicate
626818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * a failed OPEN only. However, due to a bug in previous ADB
627818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * versions, CLOSE(0, remote-id, "") was also used for normal
628818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * CLOSE() operations.
629818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 *
630818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * This is bad because it means a compromised adbd could
631818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * send packets to close connections between the host and
632818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * other devices. To avoid this, only allow this if the local
633818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 * socket has a peer on the same transport.
634818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                 */
635818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                if (p->msg.arg0 == 0 && s->peer && s->peer->transport != t) {
636818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    D("Invalid A_CLSE(0, %u) from transport %s, expected transport %s\n",
637818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                      p->msg.arg1, t->serial, s->peer->transport->serial);
638818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                } else {
639818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                    s->close(s);
640818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner                }
641dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
642dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
643dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
644dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
645818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner    case A_WRTE: /* WRITE(local-id, remote-id, <data>) */
646818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner        if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
647818d641c420f5d4f35f4e8706937c17b974146d1David 'Digit' Turner            if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
648dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                unsigned rid = p->msg.arg0;
649dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                p->len = p->msg.data_length;
650dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
651dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                if(s->enqueue(s, p) == 0) {
652dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    D("Enqueue the socket\n");
653dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                    send_ready(s->id, rid, t);
654dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                }
655dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                return;
656dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
657dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
658dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        break;
659dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
660dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    default:
661dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        printf("handle_packet: what is %08x?!\n", p->msg.command);
662dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
663dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
664dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    put_apacket(p);
665dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
666dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
667dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectalistener listener_list = {
668dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    .next = &listener_list,
669dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    .prev = &listener_list,
670dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project};
671dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
672dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void ss_listener_event_func(int _fd, unsigned ev, void *_l)
673dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
674dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    asocket *s;
675dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
676dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(ev & FDE_READ) {
677dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        struct sockaddr addr;
678dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        socklen_t alen;
679dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int fd;
680dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
681dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        alen = sizeof(addr);
682dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fd = adb_socket_accept(_fd, &addr, &alen);
683dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if(fd < 0) return;
684dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
685dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_socket_setbufsize(fd, CHUNK_SIZE);
686dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
687dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        s = create_local_socket(fd);
688dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if(s) {
689dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            connect_to_smartsocket(s);
690dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return;
691dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
692dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
693dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd);
694dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
695dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
696dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
697dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void listener_event_func(int _fd, unsigned ev, void *_l)
698dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
699dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    alistener *l = _l;
700dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    asocket *s;
701dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
702dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(ev & FDE_READ) {
703dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        struct sockaddr addr;
704dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        socklen_t alen;
705dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int fd;
706dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
707dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        alen = sizeof(addr);
708dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fd = adb_socket_accept(_fd, &addr, &alen);
709dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if(fd < 0) return;
710dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
711dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        s = create_local_socket(fd);
712dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if(s) {
713dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            s->transport = l->transport;
714dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            connect_to_remote(s, l->connect_to);
715dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return;
716dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
717dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
718dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd);
719dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
720dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
721dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
722dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void  free_listener(alistener*  l)
723dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
724dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (l->next) {
725dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        l->next->prev = l->prev;
726dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        l->prev->next = l->next;
727dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        l->next = l->prev = l;
728dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
729dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
730dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // closes the corresponding fd
731dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fdevent_remove(&l->fde);
732dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
733dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (l->local_name)
734dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        free((char*)l->local_name);
735dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
736dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (l->connect_to)
737dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        free((char*)l->connect_to);
738dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
739dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (l->transport) {
740dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        remove_transport_disconnect(l->transport, &l->disconnect);
741dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
742dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    free(l);
743dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
744dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
745dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void listener_disconnect(void*  _l, atransport*  t)
746dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
747dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    alistener*  l = _l;
748dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
749dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    free_listener(l);
750dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
751dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
752dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectint local_name_to_fd(const char *name)
753dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
754dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    int port;
755dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
756dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(!strncmp("tcp:", name, 4)){
757dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int  ret;
758dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        port = atoi(name + 4);
759d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel
760d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel        if (gListenAll > 0) {
761d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel            ret = socket_inaddr_any_server(port, SOCK_STREAM);
762d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel        } else {
763d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel            ret = socket_loopback_server(port, SOCK_STREAM);
764d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel        }
765d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel
766dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return ret;
767dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
768dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#ifndef HAVE_WIN32_IPC  /* no Unix-domain sockets on Win32 */
769dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // It's non-sensical to support the "reserved" space on the adb host side
770dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(!strncmp(name, "local:", 6)) {
771dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return socket_local_server(name + 6,
772dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                ANDROID_SOCKET_NAMESPACE_ABSTRACT, SOCK_STREAM);
773dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } else if(!strncmp(name, "localabstract:", 14)) {
774dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return socket_local_server(name + 14,
775dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                ANDROID_SOCKET_NAMESPACE_ABSTRACT, SOCK_STREAM);
776dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } else if(!strncmp(name, "localfilesystem:", 16)) {
777dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return socket_local_server(name + 16,
778dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                ANDROID_SOCKET_NAMESPACE_FILESYSTEM, SOCK_STREAM);
779dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
780dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
781dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
782dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    printf("unknown local portname '%s'\n", name);
783dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return -1;
784dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
785dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
7860d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner// Write a single line describing a listener to a user-provided buffer.
7870d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner// Appends a trailing zero, even in case of truncation, but the function
7880d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner// returns the full line length.
7890d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner// If |buffer| is NULL, does not write but returns required size.
7900d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turnerstatic int format_listener(alistener* l, char* buffer, size_t buffer_len) {
7910d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    // Format is simply:
7920d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    //
7930d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    //  <device-serial> " " <local-name> " " <remote-name> "\n"
7940d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    //
7950d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    int local_len = strlen(l->local_name);
7960d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    int connect_len = strlen(l->connect_to);
7970d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    int serial_len = strlen(l->transport->serial);
7980d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner
7990d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    if (buffer != NULL) {
8000d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        snprintf(buffer, buffer_len, "%s %s %s\n",
8010d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner                l->transport->serial, l->local_name, l->connect_to);
8020d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    }
8030d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    // NOTE: snprintf() on Windows returns -1 in case of truncation, so
8040d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    // return the computed line length instead.
8050d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    return local_len + connect_len + serial_len + 3;
8060d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner}
8070d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner
8080d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner// Write the list of current listeners (network redirections) into a
8090d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner// user-provided buffer. Appends a trailing zero, even in case of
8100d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner// trunctaion, but return the full size in bytes.
8110d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner// If |buffer| is NULL, does not write but returns required size.
8120d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turnerstatic int format_listeners(char* buf, size_t buflen)
8130d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner{
8140d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    alistener* l;
8150d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    int result = 0;
8160d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    for (l = listener_list.next; l != &listener_list; l = l->next) {
8170d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        // Ignore special listeners like those for *smartsocket*
8180d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        if (l->connect_to[0] == '*')
8190d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner          continue;
8200d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        int len = format_listener(l, buf, buflen);
8210d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        // Ensure there is space for the trailing zero.
8220d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        result += len;
8230d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        if (buf != NULL) {
8240d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner          buf += len;
8250d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner          buflen -= len;
8260d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner          if (buflen <= 0)
8270d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner              break;
8280d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        }
8290d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    }
8300d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    return result;
8310d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner}
8320d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner
8330d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turnerstatic int remove_listener(const char *local_name, atransport* transport)
834dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
835dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    alistener *l;
836dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
837dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    for (l = listener_list.next; l != &listener_list; l = l->next) {
8380d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        if (!strcmp(local_name, l->local_name)) {
8390d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner            listener_disconnect(l, l->transport);
840dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return 0;
841dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
842dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
843dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return -1;
844dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
845dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
8460d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turnerstatic void remove_all_listeners(void)
8470d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner{
8480d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    alistener *l, *l_next;
8490d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    for (l = listener_list.next; l != &listener_list; l = l_next) {
8500d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        l_next = l->next;
8510d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        // Never remove smart sockets.
8520d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        if (l->connect_to[0] == '*')
8530d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner            continue;
8540d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        listener_disconnect(l, l->transport);
8550d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    }
8560d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner}
8570d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner
8580d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner// error/status codes for install_listener.
8590d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turnertypedef enum {
8600d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner  INSTALL_STATUS_OK = 0,
8610d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner  INSTALL_STATUS_INTERNAL_ERROR = -1,
8620d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner  INSTALL_STATUS_CANNOT_BIND = -2,
8630d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner  INSTALL_STATUS_CANNOT_REBIND = -3,
8640d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner} install_status_t;
8650d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner
8660d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turnerstatic install_status_t install_listener(const char *local_name,
8670d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner                                         const char *connect_to,
8680d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner                                         atransport* transport,
8690d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner                                         int no_rebind)
870dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
871dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    alistener *l;
872dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
873dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    //printf("install_listener('%s','%s')\n", local_name, connect_to);
874dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
875dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    for(l = listener_list.next; l != &listener_list; l = l->next){
876dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if(strcmp(local_name, l->local_name) == 0) {
877dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            char *cto;
878dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
879dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                /* can't repurpose a smartsocket */
880dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            if(l->connect_to[0] == '*') {
8810d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner                return INSTALL_STATUS_INTERNAL_ERROR;
8820d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner            }
8830d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner
8840d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner                /* can't repurpose a listener if 'no_rebind' is true */
8850d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner            if (no_rebind) {
8860d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner                return INSTALL_STATUS_CANNOT_REBIND;
887dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
888dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
889dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            cto = strdup(connect_to);
890dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            if(cto == 0) {
8910d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner                return INSTALL_STATUS_INTERNAL_ERROR;
892dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
893dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
894dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            //printf("rebinding '%s' to '%s'\n", local_name, connect_to);
895dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            free((void*) l->connect_to);
896dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            l->connect_to = cto;
897dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            if (l->transport != transport) {
898dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                remove_transport_disconnect(l->transport, &l->disconnect);
899dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                l->transport = transport;
900dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                add_transport_disconnect(l->transport, &l->disconnect);
901dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            }
9020d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner            return INSTALL_STATUS_OK;
903dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
904dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
905dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
906dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if((l = calloc(1, sizeof(alistener))) == 0) goto nomem;
907dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if((l->local_name = strdup(local_name)) == 0) goto nomem;
908dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if((l->connect_to = strdup(connect_to)) == 0) goto nomem;
909dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
910dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
911dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    l->fd = local_name_to_fd(local_name);
912dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(l->fd < 0) {
913dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        free((void*) l->local_name);
914dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        free((void*) l->connect_to);
915dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        free(l);
916dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        printf("cannot bind '%s'\n", local_name);
917dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return -2;
918dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
919dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
920dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    close_on_exec(l->fd);
921dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(!strcmp(l->connect_to, "*smartsocket*")) {
922dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fdevent_install(&l->fde, l->fd, ss_listener_event_func, l);
923dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } else {
924dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fdevent_install(&l->fde, l->fd, listener_event_func, l);
925dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
926dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fdevent_set(&l->fde, FDE_READ);
927dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
928dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    l->next = &listener_list;
929dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    l->prev = listener_list.prev;
930dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    l->next->prev = l;
931dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    l->prev->next = l;
932dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    l->transport = transport;
933dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
934dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (transport) {
935dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        l->disconnect.opaque = l;
936dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        l->disconnect.func   = listener_disconnect;
937dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        add_transport_disconnect(transport, &l->disconnect);
938dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
9390d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    return INSTALL_STATUS_OK;
940dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
941dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectnomem:
942dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fatal("cannot allocate listener");
9430d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    return INSTALL_STATUS_INTERNAL_ERROR;
944dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
945dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
946dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#ifdef HAVE_WIN32_PROC
947dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic BOOL WINAPI ctrlc_handler(DWORD type)
948dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
949dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    exit(STATUS_CONTROL_C_EXIT);
950dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return TRUE;
951dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
952dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
953dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
954dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectstatic void adb_cleanup(void)
955dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
956dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    usb_cleanup();
957dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
958dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
959dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid start_logging(void)
960dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
961dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#ifdef HAVE_WIN32_PROC
962dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char    temp[ MAX_PATH ];
963dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    FILE*   fnul;
964dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    FILE*   flog;
965dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
966dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    GetTempPath( sizeof(temp) - 8, temp );
967dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    strcat( temp, "adb.log" );
968dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
969dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* Win32 specific redirections */
970dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fnul = fopen( "NUL", "rt" );
971dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (fnul != NULL)
972dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        stdin[0] = fnul[0];
973dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
974dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    flog = fopen( temp, "at" );
975dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (flog == NULL)
976dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        flog = fnul;
977dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
978dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    setvbuf( flog, NULL, _IONBF, 0 );
979dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
980dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    stdout[0] = flog[0];
981dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    stderr[0] = flog[0];
982dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr,"--- adb starting (pid %d) ---\n", getpid());
983dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#else
984dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    int fd;
985dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
986dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fd = unix_open("/dev/null", O_RDONLY);
987dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    dup2(fd, 0);
988408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall    adb_close(fd);
989dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
990dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fd = unix_open("/tmp/adb.log", O_WRONLY | O_CREAT | O_APPEND, 0640);
991dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(fd < 0) {
992dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fd = unix_open("/dev/null", O_WRONLY);
993dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
994dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    dup2(fd, 1);
995dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    dup2(fd, 2);
996408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall    adb_close(fd);
997dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr,"--- adb starting (pid %d) ---\n", getpid());
998dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
999dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
1000dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1001dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if !ADB_HOST
1002dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectvoid start_device_log(void)
1003dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
1004dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    int fd;
10051f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    char    path[PATH_MAX];
10061f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    struct tm now;
10071f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    time_t t;
10081f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    char value[PROPERTY_VALUE_MAX];
10091f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood
10101f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    // read the trace mask from persistent property persist.adb.trace_mask
10111f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    // give up if the property is not set or cannot be parsed
10121f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    property_get("persist.adb.trace_mask", value, "");
10131f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    if (sscanf(value, "%x", &adb_trace_mask) != 1)
10141f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood        return;
1015dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
10161f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    adb_mkdir("/data/adb", 0775);
10171f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    tzset();
10181f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    time(&t);
10191f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    localtime_r(&t, &now);
10201f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    strftime(path, sizeof(path),
10211f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood                "/data/adb/adb-%Y-%m-%d-%H-%M-%S.txt",
10221f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood                &now);
10231f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood    fd = unix_open(path, O_WRONLY | O_CREAT | O_TRUNC, 0640);
1024dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (fd < 0)
1025dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return;
1026dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1027dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // redirect stdout and stderr to the log file
1028dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    dup2(fd, 1);
1029dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    dup2(fd, 2);
1030dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fprintf(stderr,"--- adb starting (pid %d) ---\n", getpid());
103195ef82866c7a922bf588027aa38c58a45eb84d9cBenoit Goby    adb_close(fd);
1032dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1033dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fd = unix_open("/dev/null", O_RDONLY);
1034dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    dup2(fd, 0);
103595ef82866c7a922bf588027aa38c58a45eb84d9cBenoit Goby    adb_close(fd);
1036dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
1037dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
1038dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1039dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if ADB_HOST
1040571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall
1041571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall#ifdef WORKAROUND_BUG6558362
1042571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall#include <sched.h>
1043571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall#define AFFINITY_ENVVAR "ADB_CPU_AFFINITY_BUG6558362"
1044571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrallvoid adb_set_affinity(void)
1045571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall{
1046571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   cpu_set_t cpu_set;
1047571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   const char* cpunum_str = getenv(AFFINITY_ENVVAR);
1048571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   char* strtol_res;
1049571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   int cpu_num;
1050571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall
1051571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   if (!cpunum_str || !*cpunum_str)
1052571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall       return;
1053571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   cpu_num = strtol(cpunum_str, &strtol_res, 0);
1054571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   if (*strtol_res != '\0')
1055571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall     fatal("bad number (%s) in env var %s. Expecting 0..n.\n", cpunum_str, AFFINITY_ENVVAR);
1056571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall
1057571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   sched_getaffinity(0, sizeof(cpu_set), &cpu_set);
1058571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   D("orig cpu_set[0]=0x%08lx\n", cpu_set.__bits[0]);
1059571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   CPU_ZERO(&cpu_set);
1060571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   CPU_SET(cpu_num, &cpu_set);
1061571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   sched_setaffinity(0, sizeof(cpu_set), &cpu_set);
1062571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   sched_getaffinity(0, sizeof(cpu_set), &cpu_set);
1063571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall   D("new cpu_set[0]=0x%08lx\n", cpu_set.__bits[0]);
1064571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall}
1065571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall#endif
1066571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall
1067a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzingerint launch_server(int server_port)
1068dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
1069dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#ifdef HAVE_WIN32_PROC
1070dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* we need to start the server in the background                    */
1071dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* we create a PIPE that will be used to wait for the server's "OK" */
1072dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* message since the pipe handles must be inheritable, we use a     */
1073dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* security attribute                                               */
1074dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    HANDLE                pipe_read, pipe_write;
1075267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    HANDLE                stdout_handle, stderr_handle;
1076dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    SECURITY_ATTRIBUTES   sa;
1077dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    STARTUPINFO           startup;
1078dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    PROCESS_INFORMATION   pinfo;
1079dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char                  program_path[ MAX_PATH ];
1080dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    int                   ret;
1081dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1082dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    sa.nLength = sizeof(sa);
1083dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    sa.lpSecurityDescriptor = NULL;
1084dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    sa.bInheritHandle = TRUE;
1085dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1086dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* create pipe, and ensure its read handle isn't inheritable */
1087dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    ret = CreatePipe( &pipe_read, &pipe_write, &sa, 0 );
1088dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!ret) {
1089dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "CreatePipe() failure, error %ld\n", GetLastError() );
1090dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return -1;
1091dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1092dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1093dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    SetHandleInformation( pipe_read, HANDLE_FLAG_INHERIT, 0 );
1094dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1095267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    /* Some programs want to launch an adb command and collect its output by
1096267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * calling CreateProcess with inheritable stdout/stderr handles, then
1097267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * using read() to get its output. When this happens, the stdout/stderr
1098267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * handles passed to the adb client process will also be inheritable.
1099267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * When starting the adb server here, care must be taken to reset them
1100267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * to non-inheritable.
1101267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * Otherwise, something bad happens: even if the adb command completes,
1102267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * the calling process is stuck while read()-ing from the stdout/stderr
1103267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * descriptors, because they're connected to corresponding handles in the
1104267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     * adb server process (even if the latter never uses/writes to them).
1105267aa8b00eadf273321e492026af74b7baad890eRay Donnelly     */
1106267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    stdout_handle = GetStdHandle( STD_OUTPUT_HANDLE );
1107267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    stderr_handle = GetStdHandle( STD_ERROR_HANDLE );
1108267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    if (stdout_handle != INVALID_HANDLE_VALUE) {
1109267aa8b00eadf273321e492026af74b7baad890eRay Donnelly        SetHandleInformation( stdout_handle, HANDLE_FLAG_INHERIT, 0 );
1110267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    }
1111267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    if (stderr_handle != INVALID_HANDLE_VALUE) {
1112267aa8b00eadf273321e492026af74b7baad890eRay Donnelly        SetHandleInformation( stderr_handle, HANDLE_FLAG_INHERIT, 0 );
1113267aa8b00eadf273321e492026af74b7baad890eRay Donnelly    }
1114267aa8b00eadf273321e492026af74b7baad890eRay Donnelly
1115dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    ZeroMemory( &startup, sizeof(startup) );
1116dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    startup.cb = sizeof(startup);
1117dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    startup.hStdInput  = GetStdHandle( STD_INPUT_HANDLE );
1118dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    startup.hStdOutput = pipe_write;
1119dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    startup.hStdError  = GetStdHandle( STD_ERROR_HANDLE );
1120dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    startup.dwFlags    = STARTF_USESTDHANDLES;
1121dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1122dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    ZeroMemory( &pinfo, sizeof(pinfo) );
1123dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1124dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* get path of current program */
1125dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    GetModuleFileName( NULL, program_path, sizeof(program_path) );
1126dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1127dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    ret = CreateProcess(
1128dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            program_path,                              /* program path  */
1129dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            "adb fork-server server",
1130dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                                    /* the fork-server argument will set the
1131dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project                                       debug = 2 in the child           */
1132dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                   /* process handle is not inheritable */
1133dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                    /* thread handle is not inheritable */
1134dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            TRUE,                          /* yes, inherit some handles */
1135dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            DETACHED_PROCESS, /* the new process doesn't have a console */
1136dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                     /* use parent's environment block */
1137dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            NULL,                    /* use parent's starting directory */
1138dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            &startup,                 /* startup info, i.e. std handles */
1139dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            &pinfo );
1140dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1141dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    CloseHandle( pipe_write );
1142dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1143dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!ret) {
1144dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "CreateProcess failure, error %ld\n", GetLastError() );
1145dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        CloseHandle( pipe_read );
1146dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return -1;
1147dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1148dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1149dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    CloseHandle( pinfo.hProcess );
1150dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    CloseHandle( pinfo.hThread );
1151dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1152dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* wait for the "OK\n" message */
1153dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    {
1154dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        char  temp[3];
1155dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        DWORD  count;
1156dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1157dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        ret = ReadFile( pipe_read, temp, 3, &count, NULL );
1158dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        CloseHandle( pipe_read );
1159dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if ( !ret ) {
1160dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fprintf(stderr, "could not read ok from ADB Server, error = %ld\n", GetLastError() );
1161dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return -1;
1162dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
1163dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (count != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
1164dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fprintf(stderr, "ADB server didn't ACK\n" );
1165dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return -1;
1166dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
1167dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1168dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#elif defined(HAVE_FORKEXEC)
1169dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char    path[PATH_MAX];
1170dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    int     fd[2];
1171dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1172dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // set up a pipe so the child can tell us when it is ready.
1173dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // fd[0] will be parent's end, and fd[1] will get mapped to stderr in the child.
1174dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (pipe(fd)) {
1175dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "pipe failed in launch_server, errno: %d\n", errno);
1176dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return -1;
1177dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
11783166410a82f43d39201be98a8d35c51baa86cb53Alexey Tarasov    get_my_path(path, PATH_MAX);
1179dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    pid_t pid = fork();
1180dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(pid < 0) return -1;
1181dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1182dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (pid == 0) {
1183dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // child side of the fork
1184dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1185dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // redirect stderr to the pipe
1186dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // we use stderr instead of stdout due to stdout's buffering behavior.
1187dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[0]);
1188dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        dup2(fd[1], STDERR_FILENO);
1189dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[1]);
1190dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1191d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel        char str_port[30];
1192d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel        snprintf(str_port, sizeof(str_port), "%d",  server_port);
1193dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // child process
1194d7b33085117ccbb908a883f624cb1fe5495ee92aMatt Gumbel        int result = execl(path, "adb", "-P", str_port, "fork-server", "server", NULL);
1195dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // this should not return
1196dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "OOPS! execl returned %d, errno: %d\n", result, errno);
1197dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } else  {
1198dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // parent side of the fork
1199dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1200dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        char  temp[3];
1201dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1202dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        temp[0] = 'A'; temp[1] = 'B'; temp[2] = 'C';
1203dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // wait for the "OK\n" message
1204dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[1]);
1205dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int ret = adb_read(fd[0], temp, 3);
1206408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall        int saved_errno = errno;
1207dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_close(fd[0]);
1208dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (ret < 0) {
1209408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall            fprintf(stderr, "could not read ok from ADB Server, errno = %d\n", saved_errno);
1210dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return -1;
1211dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
1212dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (ret != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
1213dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            fprintf(stderr, "ADB server didn't ACK\n" );
1214dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            return -1;
1215dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
1216dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1217dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        setsid();
1218dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1219dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#else
1220dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#error "cannot implement background server start on this platform"
1221dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
1222dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return 0;
1223dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
1224dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
1225dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1226a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger/* Constructs a local name of form tcp:port.
1227a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger * target_str points to the target string, it's content will be overwritten.
1228a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger * target_size is the capacity of the target string.
1229a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger * server_port is the port number to use for the local name.
1230a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger */
1231a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzingervoid build_local_name(char* target_str, size_t target_size, int server_port)
1232a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger{
1233a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger  snprintf(target_str, target_size, "tcp:%d", server_port);
1234a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger}
1235a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger
1236bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich#if !ADB_HOST
1237080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich
1238080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevichstatic void drop_capabilities_bounding_set_if_needed() {
1239080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich#ifdef ALLOW_ADBD_ROOT
1240080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich    char value[PROPERTY_VALUE_MAX];
1241080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich    property_get("ro.debuggable", value, "");
1242080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich    if (strcmp(value, "1") == 0) {
1243080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich        return;
1244080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich    }
1245080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich#endif
1246080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich    int i;
1247080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich    for (i = 0; prctl(PR_CAPBSET_READ, i, 0, 0, 0) >= 0; i++) {
1248ca8e66a8b0f843812014a28d49208f9f6f64ecbcNick Kralevich        if (i == CAP_SETUID || i == CAP_SETGID) {
1249080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich            // CAP_SETUID CAP_SETGID needed by /system/bin/run-as
1250080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich            continue;
1251080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich        }
1252080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich        int err = prctl(PR_CAPBSET_DROP, i, 0, 0, 0);
1253080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich
1254080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich        // Some kernels don't have file capabilities compiled in, and
1255080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich        // prctl(PR_CAPBSET_DROP) returns EINVAL. Don't automatically
1256080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich        // die when we see such misconfigured kernels.
1257080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich        if ((err < 0) && (errno != EINVAL)) {
1258080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich            exit(1);
1259080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich        }
1260080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich    }
1261080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich}
1262080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich
1263bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevichstatic int should_drop_privileges() {
12645890fe33141a9efd124c86c40a8c1ff6170ecf20Nick Kralevich#ifndef ALLOW_ADBD_ROOT
12655890fe33141a9efd124c86c40a8c1ff6170ecf20Nick Kralevich    return 1;
12665890fe33141a9efd124c86c40a8c1ff6170ecf20Nick Kralevich#else /* ALLOW_ADBD_ROOT */
1267bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich    int secure = 0;
1268bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich    char value[PROPERTY_VALUE_MAX];
1269bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich
1270bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich   /* run adbd in secure mode if ro.secure is set and
1271bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich    ** we are not in the emulator
1272bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich    */
1273bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich    property_get("ro.kernel.qemu", value, "");
1274bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich    if (strcmp(value, "1") != 0) {
1275bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich        property_get("ro.secure", value, "1");
1276bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich        if (strcmp(value, "1") == 0) {
1277bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich            // don't run as root if ro.secure is set...
1278bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich            secure = 1;
1279bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich
1280bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich            // ... except we allow running as root in userdebug builds if the
1281bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich            // service.adb.root property has been set by the "adb root" command
1282bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich            property_get("ro.debuggable", value, "");
1283bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich            if (strcmp(value, "1") == 0) {
1284bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich                property_get("service.adb.root", value, "");
1285bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich                if (strcmp(value, "1") == 0) {
1286bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich                    secure = 0;
1287bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich                }
1288bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich            }
1289bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich        }
1290bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich    }
1291bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich    return secure;
12925890fe33141a9efd124c86c40a8c1ff6170ecf20Nick Kralevich#endif /* ALLOW_ADBD_ROOT */
1293bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich}
1294bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich#endif /* !ADB_HOST */
1295bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich
1296a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzingerint adb_main(int is_daemon, int server_port)
1297dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
1298dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if !ADB_HOST
12992f38b699713dc2587a771bd5d4c6a47329728f5eMike Lockwood    int port;
1300dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    char value[PROPERTY_VALUE_MAX];
1301eb68fa8153d97f5f8b6d9062fcf91fe393e3bff3Nick Kralevich
1302eb68fa8153d97f5f8b6d9062fcf91fe393e3bff3Nick Kralevich    umask(000);
1303dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
1304dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1305dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    atexit(adb_cleanup);
1306dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#ifdef HAVE_WIN32_PROC
1307dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    SetConsoleCtrlHandler( ctrlc_handler, TRUE );
1308dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#elif defined(HAVE_FORKEXEC)
1309408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall    // No SIGCHLD. Let the service subproc handle its children.
1310dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    signal(SIGPIPE, SIG_IGN);
1311dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
1312dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1313dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    init_transport_registration();
1314dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1315dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if ADB_HOST
1316dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    HOST = 1;
1317571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall
1318571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall#ifdef WORKAROUND_BUG6558362
1319571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall    if(is_daemon) adb_set_affinity();
1320571c1367682c83cb23481cc9c1028c8588a1fed0JP Abgrall#endif
1321a09fbd164d2e088bc5433d310e25640ae048d47dXavier Ducrohet    usb_vendors_init();
1322dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    usb_init();
1323a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger    local_init(DEFAULT_ADB_LOCAL_TRANSPORT_PORT);
1324d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    adb_auth_init();
1325dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1326a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger    char local_name[30];
1327a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger    build_local_name(local_name, sizeof(local_name), server_port);
13280d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner    if(install_listener(local_name, "*smartsocket*", NULL, 0)) {
1329dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        exit(1);
1330dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1331dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#else
1332d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    property_get("ro.adb.secure", value, "0");
1333d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    auth_enabled = !strcmp(value, "1");
1334d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby    if (auth_enabled)
1335d5fcafaf41f8ec90986c813f75ec78402096af2dBenoit Goby        adb_auth_init();
1336dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1337d6d4286a28b4a9aef902585625ea6656ab06cf52Jeff Sharkey    // Our external storage path may be different than apps, since
1338d6d4286a28b4a9aef902585625ea6656ab06cf52Jeff Sharkey    // we aren't able to bind mount after dropping root.
1339d6d4286a28b4a9aef902585625ea6656ab06cf52Jeff Sharkey    const char* adb_external_storage = getenv("ADB_EXTERNAL_STORAGE");
1340d6d4286a28b4a9aef902585625ea6656ab06cf52Jeff Sharkey    if (NULL != adb_external_storage) {
1341d6d4286a28b4a9aef902585625ea6656ab06cf52Jeff Sharkey        setenv("EXTERNAL_STORAGE", adb_external_storage, 1);
1342d6d4286a28b4a9aef902585625ea6656ab06cf52Jeff Sharkey    } else {
1343d6d4286a28b4a9aef902585625ea6656ab06cf52Jeff Sharkey        D("Warning: ADB_EXTERNAL_STORAGE is not set.  Leaving EXTERNAL_STORAGE"
1344d6d4286a28b4a9aef902585625ea6656ab06cf52Jeff Sharkey          " unchanged.\n");
1345d6d4286a28b4a9aef902585625ea6656ab06cf52Jeff Sharkey    }
1346d6d4286a28b4a9aef902585625ea6656ab06cf52Jeff Sharkey
1347a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger    /* don't listen on a port (default 5037) if running in secure mode */
1348dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    /* don't run as root if we are running in secure mode */
1349bd9206bc7399b22d3917c737bb75f8534fa77e35Nick Kralevich    if (should_drop_privileges()) {
1350080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich        drop_capabilities_bounding_set_if_needed();
1351080427e4e2b1b72718b660e16b6cf38b3a3c4e3fNick Kralevich
1352dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        /* add extra groups:
1353dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        ** AID_ADB to access the USB driver
1354dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        ** AID_LOG to read system logs (adb logcat)
1355dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        ** AID_INPUT to diagnose input issues (getevent)
1356dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        ** AID_INET to diagnose network issues (netcfg, ping)
1357dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        ** AID_GRAPHICS to access the frame buffer
13582015549667fb77706a9879e974a3875ebccd8198The Android Open Source Project        ** AID_NET_BT and AID_NET_BT_ADMIN to diagnose bluetooth (hcidump)
135950458cf76ea7b0b03598c785acb1481ed0ae5b1dDianne Hackborn        ** AID_SDCARD_R to allow reading from the SD card
13606a3075c78287ea5a7ee7a3c85cde58e894350822Mike Lockwood        ** AID_SDCARD_RW to allow writing to the SD card
1361d969faa161310d0a3792766320daa3200b84bd74Mike Lockwood        ** AID_MOUNT to allow unmounting the SD card before rebooting
136261b90bd211021886049f72cb666a5b49b83b3c64JP Abgrall        ** AID_NET_BW_STATS to read out qtaguid statistics
1363dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        */
13642015549667fb77706a9879e974a3875ebccd8198The Android Open Source Project        gid_t groups[] = { AID_ADB, AID_LOG, AID_INPUT, AID_INET, AID_GRAPHICS,
136550458cf76ea7b0b03598c785acb1481ed0ae5b1dDianne Hackborn                           AID_NET_BT, AID_NET_BT_ADMIN, AID_SDCARD_R, AID_SDCARD_RW,
136650458cf76ea7b0b03598c785acb1481ed0ae5b1dDianne Hackborn                           AID_MOUNT, AID_NET_BW_STATS };
136744db990d3a4ce0edbdd16fa7ac20693ef601b723Nick Kralevich        if (setgroups(sizeof(groups)/sizeof(groups[0]), groups) != 0) {
136844db990d3a4ce0edbdd16fa7ac20693ef601b723Nick Kralevich            exit(1);
136944db990d3a4ce0edbdd16fa7ac20693ef601b723Nick Kralevich        }
1370dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1371dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        /* then switch user and group to "shell" */
137244db990d3a4ce0edbdd16fa7ac20693ef601b723Nick Kralevich        if (setgid(AID_SHELL) != 0) {
137344db990d3a4ce0edbdd16fa7ac20693ef601b723Nick Kralevich            exit(1);
137444db990d3a4ce0edbdd16fa7ac20693ef601b723Nick Kralevich        }
137544db990d3a4ce0edbdd16fa7ac20693ef601b723Nick Kralevich        if (setuid(AID_SHELL) != 0) {
137644db990d3a4ce0edbdd16fa7ac20693ef601b723Nick Kralevich            exit(1);
137744db990d3a4ce0edbdd16fa7ac20693ef601b723Nick Kralevich        }
1378dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1379a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger        D("Local port disabled\n");
1380dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    } else {
1381a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger        char local_name[30];
1382d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        if ((root_seclabel != NULL) && (is_selinux_enabled() > 0)) {
1383d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            // b/12587913: fix setcon to allow const pointers
1384d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            if (setcon((char *)root_seclabel) < 0) {
1385d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich                exit(1);
1386d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            }
1387d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        }
1388a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger        build_local_name(local_name, sizeof(local_name), server_port);
13890d82fbf04d6db847cf598f370aa6986af794bd72David 'Digit' Turner        if(install_listener(local_name, "*smartsocket*", NULL, 0)) {
1390dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            exit(1);
1391dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
1392dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1393dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
13941dd55c53af264b24319d721281fe32c67c053120Mike J. Chen    int usb = 0;
13951dd55c53af264b24319d721281fe32c67c053120Mike J. Chen    if (access(USB_ADB_PATH, F_OK) == 0 || access(USB_FFS_ADB_EP0, F_OK) == 0) {
13961dd55c53af264b24319d721281fe32c67c053120Mike J. Chen        // listen on USB
13971dd55c53af264b24319d721281fe32c67c053120Mike J. Chen        usb_init();
13981dd55c53af264b24319d721281fe32c67c053120Mike J. Chen        usb = 1;
13991dd55c53af264b24319d721281fe32c67c053120Mike J. Chen    }
14001dd55c53af264b24319d721281fe32c67c053120Mike J. Chen
14011dd55c53af264b24319d721281fe32c67c053120Mike J. Chen    // If one of these properties is set, also listen on that port
14021dd55c53af264b24319d721281fe32c67c053120Mike J. Chen    // If one of the properties isn't set and we couldn't listen on usb,
14031dd55c53af264b24319d721281fe32c67c053120Mike J. Chen    // listen on the default port.
14048e2ceaeacf9fa1eb5ac41e4d3498836a6e2c7c97Mike Lockwood    property_get("service.adb.tcp.port", value, "");
14051dd55c53af264b24319d721281fe32c67c053120Mike J. Chen    if (!value[0]) {
14068e2ceaeacf9fa1eb5ac41e4d3498836a6e2c7c97Mike Lockwood        property_get("persist.adb.tcp.port", value, "");
14071dd55c53af264b24319d721281fe32c67c053120Mike J. Chen    }
1408cef31a02544d91b85481a5fa032b6126dbb32365Mike Lockwood    if (sscanf(value, "%d", &port) == 1 && port > 0) {
14091dd55c53af264b24319d721281fe32c67c053120Mike J. Chen        printf("using port=%d\n", port);
1410cef31a02544d91b85481a5fa032b6126dbb32365Mike Lockwood        // listen on TCP port specified by service.adb.tcp.port property
1411cef31a02544d91b85481a5fa032b6126dbb32365Mike Lockwood        local_init(port);
14121dd55c53af264b24319d721281fe32c67c053120Mike J. Chen    } else if (!usb) {
1413cef31a02544d91b85481a5fa032b6126dbb32365Mike Lockwood        // listen on default port
1414a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger        local_init(DEFAULT_ADB_LOCAL_TRANSPORT_PORT);
1415dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
14161dd55c53af264b24319d721281fe32c67c053120Mike J. Chen
1417408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall    D("adb_main(): pre init_jdwp()\n");
1418dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    init_jdwp();
1419408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall    D("adb_main(): post init_jdwp()\n");
1420dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
1421dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1422dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (is_daemon)
1423dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    {
1424dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        // inform our parent that we are up and running.
1425dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#ifdef HAVE_WIN32_PROC
1426dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        DWORD  count;
1427dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        WriteFile( GetStdHandle( STD_OUTPUT_HANDLE ), "OK\n", 3, &count, NULL );
1428dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#elif defined(HAVE_FORKEXEC)
1429dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr, "OK\n");
1430dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
1431dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        start_logging();
1432dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1433408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall    D("Event loop starting\n");
1434dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1435dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    fdevent_loop();
1436dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1437dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    usb_cleanup();
1438dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1439dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return 0;
1440dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
1441dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1442252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner// Try to handle a network forwarding request.
1443252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner// This returns 1 on success, 0 on failure, and -1 to indicate this is not
1444252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner// a forwarding-related request.
1445252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turnerint handle_forward_request(const char* service, transport_type ttype, char* serial, int reply_fd)
1446252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner{
1447252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    if (!strcmp(service, "list-forward")) {
1448252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        // Create the list of forward redirections.
1449252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        int buffer_size = format_listeners(NULL, 0);
1450252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        // Add one byte for the trailing zero.
1451252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        char* buffer = malloc(buffer_size + 1);
1452252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        if (buffer == NULL) {
1453252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            sendfailmsg(reply_fd, "not enough memory");
1454252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            return 1;
1455252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
1456252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        (void) format_listeners(buffer, buffer_size + 1);
1457252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#if ADB_HOST
1458252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        send_msg_with_okay(reply_fd, buffer, buffer_size);
1459252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#else
1460252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        send_msg_with_header(reply_fd, buffer, buffer_size);
1461252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#endif
1462252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        free(buffer);
1463252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        return 1;
1464252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    }
1465252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1466252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    if (!strcmp(service, "killforward-all")) {
1467252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        remove_all_listeners();
1468252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#if ADB_HOST
1469252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        /* On the host: 1st OKAY is connect, 2nd OKAY is status */
1470252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        adb_write(reply_fd, "OKAY", 4);
1471252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#endif
1472252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        adb_write(reply_fd, "OKAY", 4);
1473252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        return 1;
1474252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    }
1475252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1476252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    if (!strncmp(service, "forward:",8) ||
1477252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        !strncmp(service, "killforward:",12)) {
1478252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        char *local, *remote, *err;
1479252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        int r;
1480252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        atransport *transport;
1481252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1482252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        int createForward = strncmp(service, "kill", 4);
1483252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        int no_rebind = 0;
1484252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1485252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        local = strchr(service, ':') + 1;
1486252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1487252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        // Handle forward:norebind:<local>... here
1488252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        if (createForward && !strncmp(local, "norebind:", 9)) {
1489252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            no_rebind = 1;
1490252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            local = strchr(local, ':') + 1;
1491252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
1492252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1493252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        remote = strchr(local,';');
1494252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1495252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        if (createForward) {
1496252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            // Check forward: parameter format: '<local>;<remote>'
1497252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            if(remote == 0) {
1498252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                sendfailmsg(reply_fd, "malformed forward spec");
1499252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                return 1;
1500252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            }
1501252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1502252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            *remote++ = 0;
1503252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            if((local[0] == 0) || (remote[0] == 0) || (remote[0] == '*')) {
1504252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                sendfailmsg(reply_fd, "malformed forward spec");
1505252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                return 1;
1506252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            }
1507252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        } else {
1508252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            // Check killforward: parameter format: '<local>'
1509252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            if (local[0] == 0) {
1510252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                sendfailmsg(reply_fd, "malformed forward spec");
1511252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                return 1;
1512252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            }
1513252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
1514252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1515252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        transport = acquire_one_transport(CS_ANY, ttype, serial, &err);
1516252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        if (!transport) {
1517252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            sendfailmsg(reply_fd, err);
1518252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            return 1;
1519252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
1520252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1521252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        if (createForward) {
1522252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            r = install_listener(local, remote, transport, no_rebind);
1523252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        } else {
1524252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            r = remove_listener(local, transport);
1525252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
1526252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        if(r == 0) {
1527252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#if ADB_HOST
1528252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            /* On the host: 1st OKAY is connect, 2nd OKAY is status */
1529252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            writex(reply_fd, "OKAY", 4);
1530252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner#endif
1531252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            writex(reply_fd, "OKAY", 4);
1532252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            return 1;
1533252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
1534252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1535252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        if (createForward) {
1536252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            const char* message;
1537252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            switch (r) {
1538252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner              case INSTALL_STATUS_CANNOT_BIND:
1539252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                message = "cannot bind to socket";
1540252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                break;
1541252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner              case INSTALL_STATUS_CANNOT_REBIND:
1542252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                message = "cannot rebind existing socket";
1543252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                break;
1544252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner              default:
1545252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner                message = "internal error";
1546252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            }
1547252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            sendfailmsg(reply_fd, message);
1548252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        } else {
1549252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner            sendfailmsg(reply_fd, "cannot remove listener");
1550252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        }
1551252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner        return 1;
1552252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    }
1553252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    return 0;
1554252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner}
1555252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner
1556dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectint handle_host_request(char *service, transport_type ttype, char* serial, int reply_fd, asocket *s)
1557dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
1558dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    atransport *transport = NULL;
1559dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1560dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(!strcmp(service, "kill")) {
1561dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fprintf(stderr,"adb server killed by remote request\n");
1562dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        fflush(stdout);
1563dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        adb_write(reply_fd, "OKAY", 4);
1564dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        usb_cleanup();
1565dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        exit(0);
1566dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1567dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1568dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if ADB_HOST
1569dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport:" is used for switching transport with a specified serial number
1570dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport-usb:" is used for switching transport to the only USB transport
1571dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport-local:" is used for switching transport to the only local transport
1572dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // "transport-any:" is used for switching transport to the only transport
1573dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!strncmp(service, "transport", strlen("transport"))) {
1574dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        char* error_string = "unknown failure";
1575dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        transport_type type = kTransportAny;
1576dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1577dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (!strncmp(service, "transport-usb", strlen("transport-usb"))) {
1578dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            type = kTransportUsb;
1579dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else if (!strncmp(service, "transport-local", strlen("transport-local"))) {
1580dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            type = kTransportLocal;
1581dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else if (!strncmp(service, "transport-any", strlen("transport-any"))) {
1582dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            type = kTransportAny;
1583dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else if (!strncmp(service, "transport:", strlen("transport:"))) {
1584dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            service += strlen("transport:");
15853175c8e9bfcb01f181d7300755da15a999e5461cTom Marlin            serial = service;
1586dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
1587dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1588dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        transport = acquire_one_transport(CS_ANY, type, serial, &error_string);
1589dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1590dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        if (transport) {
1591dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            s->transport = transport;
1592dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            adb_write(reply_fd, "OKAY", 4);
1593dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        } else {
1594dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            sendfailmsg(reply_fd, error_string);
1595dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
1596dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return 1;
1597dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1598dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1599dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // return a list of all connected devices
1600e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson    if (!strncmp(service, "devices", 7)) {
1601dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        char buffer[4096];
1602e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson        int use_long = !strcmp(service+7, "-l");
1603e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson        if (use_long || service[7] == 0) {
1604e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson            memset(buffer, 0, sizeof(buffer));
1605e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson            D("Getting device list \n");
1606e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson            list_transports(buffer, sizeof(buffer), use_long);
1607e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson            D("Wrote device list \n");
16082264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow            send_msg_with_okay(reply_fd, buffer, strlen(buffer));
1609e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson            return 0;
1610e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson        }
1611dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1612dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
161374d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood    // remove TCP transport
161474d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood    if (!strncmp(service, "disconnect:", 11)) {
161574d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood        char buffer[4096];
161674d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood        memset(buffer, 0, sizeof(buffer));
161774d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood        char* serial = service + 11;
1618cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood        if (serial[0] == 0) {
1619cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            // disconnect from all TCP devices
1620cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            unregister_all_tcp_transports();
162174d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood        } else {
1622cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            char hostbuf[100];
1623cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            // assume port 5555 if no port is specified
1624cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            if (!strchr(serial, ':')) {
1625cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood                snprintf(hostbuf, sizeof(hostbuf) - 1, "%s:5555", serial);
1626cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood                serial = hostbuf;
1627cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            }
1628cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            atransport *t = find_transport(serial);
1629cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood
1630cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            if (t) {
1631cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood                unregister_transport(t);
1632cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            } else {
1633cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood                snprintf(buffer, sizeof(buffer), "No such device %s", serial);
1634cbbe79add1410b53ec65936cfb1b74cac68467f0Mike Lockwood            }
163574d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood        }
163674d7ff8cfd490852d3df1c4b9d8a21beed619caaMike Lockwood
16372264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow        send_msg_with_okay(reply_fd, buffer, strlen(buffer));
16382f38b699713dc2587a771bd5d4c6a47329728f5eMike Lockwood        return 0;
16392f38b699713dc2587a771bd5d4c6a47329728f5eMike Lockwood    }
16402f38b699713dc2587a771bd5d4c6a47329728f5eMike Lockwood
1641dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // returns our value for ADB_SERVER_VERSION
1642dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!strcmp(service, "version")) {
1643dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        char version[12];
1644dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        snprintf(version, sizeof version, "%04x", ADB_SERVER_VERSION);
16452264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow        send_msg_with_okay(reply_fd, version, strlen(version));
1646dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return 0;
1647dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1648dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1649dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(!strncmp(service,"get-serialno",strlen("get-serialno"))) {
1650dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        char *out = "unknown";
1651dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project         transport = acquire_one_transport(CS_ANY, ttype, serial, NULL);
1652dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project       if (transport && transport->serial) {
1653dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project            out = transport->serial;
1654dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        }
16552264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow        send_msg_with_okay(reply_fd, out, strlen(out));
1656dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return 0;
1657dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1658e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson    if(!strncmp(service,"get-devpath",strlen("get-devpath"))) {
1659e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson        char *out = "unknown";
1660e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson         transport = acquire_one_transport(CS_ANY, ttype, serial, NULL);
1661e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson       if (transport && transport->devpath) {
1662e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson            out = transport->devpath;
1663e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson        }
16642264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow        send_msg_with_okay(reply_fd, out, strlen(out));
1665e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson        return 0;
1666e109d266c12c5f537d429ca4b892f2719e02c2daScott Anderson    }
1667dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    // indicates a new emulator instance has started
1668dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if (!strncmp(service,"emulator:",9)) {
1669dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        int  port = atoi(service+9);
1670dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        local_connect(port);
1671dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        /* we don't even need to send a reply */
1672dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return 0;
1673dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1674dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif // ADB_HOST
1675dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1676252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    int ret = handle_forward_request(service, ttype, serial, reply_fd);
1677252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner    if (ret >= 0)
1678252586941934d23073a8d167ec240b221062505fDavid 'Digit' Turner      return ret - 1;
1679dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1680dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    if(!strncmp(service,"get-state",strlen("get-state"))) {
1681dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        transport = acquire_one_transport(CS_ANY, ttype, serial, NULL);
1682dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        char *state = connection_state_name(transport);
16832264e7cfef6b1236a90a13b1d99abb4aadcb0b93Snild Dolkow        send_msg_with_okay(reply_fd, state, strlen(state));
1684dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project        return 0;
1685dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
1686dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return -1;
1687dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
1688dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project
1689dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Projectint main(int argc, char **argv)
1690dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project{
1691dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#if ADB_HOST
1692dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    adb_sysdeps_init();
1693408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall    adb_trace_init();
1694408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall    D("Handling commandline()\n");
1695dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    return adb_commandline(argc - 1, argv + 1);
1696dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#else
169728781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    /* If adbd runs inside the emulator this will enable adb tracing via
169828781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine     * adb-debug qemud service in the emulator. */
169928781b0a52dfb5ad19121afcc6aef56918992b30Vladimir Chtchetkine    adb_qemu_trace_init();
1700d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich    while(1) {
1701d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        int c;
1702d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        int option_index = 0;
1703d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        static struct option opts[] = {
1704d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            {"root_seclabel", required_argument, 0, 's' },
1705d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            {"device_banner", required_argument, 0, 'b' }
1706d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        };
1707d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        c = getopt_long(argc, argv, "", opts, &option_index);
1708d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        if (c == -1)
1709d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            break;
1710d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        switch (c) {
1711d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        case 's':
1712d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            root_seclabel = optarg;
1713d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            break;
1714d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        case 'b':
1715d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            adb_device_banner = optarg;
1716d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            break;
1717d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        default:
1718d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich            break;
1719d49aa2537cf0b454dfaa4b0312a403ebff202d70Nick Kralevich        }
1720dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    }
17211f546e6d1f6ccd1964336ddf0d8e8b3b11b1e945Mike Lockwood
1722dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project    start_device_log();
1723408fa57864c01113deaa213e5c1848a9c594ae92JP Abgrall    D("Handling main()\n");
1724a84a42eb20d43ffa2695a69d583a6e09532b49d9Stefan Hilzinger    return adb_main(0, DEFAULT_ADB_PORT);
1725dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project#endif
1726dd7bc3319deb2b77c5d07a51b7d6cd7e11b5beb0The Android Open Source Project}
1727