adb.c revision c57a22c8561cc571d23cd7f321103b2b4d2c7cde
1b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber/*
2b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * Copyright (C) 2007 The Android Open Source Project
3b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber *
4b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * Licensed under the Apache License, Version 2.0 (the "License");
5b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * you may not use this file except in compliance with the License.
6b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * You may obtain a copy of the License at
7b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber *
8b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber *      http://www.apache.org/licenses/LICENSE-2.0
9b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber *
10b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * Unless required by applicable law or agreed to in writing, software
11b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * distributed under the License is distributed on an "AS IS" BASIS,
12b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * See the License for the specific language governing permissions and
14b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * limitations under the License.
15b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber */
16b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
17b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#define  TRACE_TAG   TRACE_ADB
18b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
19b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <stdio.h>
20b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <stdlib.h>
21b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <ctype.h>
22b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <stdarg.h>
23b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <errno.h>
24b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <string.h>
25b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <time.h>
26b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <sys/time.h>
27b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
28b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include "sysdeps.h"
29b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include "adb.h"
30b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
31b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#if !ADB_HOST
32b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <private/android_filesystem_config.h>
33b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <linux/capability.h>
34b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include <linux/prctl.h>
35b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#else
36b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#include "usb_vendors.h"
37b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber#endif
38b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
39b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
40b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huberint HOST = 0;
41b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
42b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huberstatic const char *adb_device_banner = "device";
43b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
44b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Hubervoid fatal(const char *fmt, ...)
45b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber{
46b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    va_list ap;
47b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    va_start(ap, fmt);
48b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    fprintf(stderr, "error: ");
49b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    vfprintf(stderr, fmt, ap);
50b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    fprintf(stderr, "\n");
51b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    va_end(ap);
52b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    exit(-1);
53b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber}
54b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
55b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Hubervoid fatal_errno(const char *fmt, ...)
56b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber{
57b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    va_list ap;
58b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    va_start(ap, fmt);
59b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    fprintf(stderr, "error: %s: ", strerror(errno));
60b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    vfprintf(stderr, fmt, ap);
61b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    fprintf(stderr, "\n");
62b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    va_end(ap);
63b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    exit(-1);
64b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber}
65b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
66b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huberint   adb_trace_mask;
67b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
68b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber/* read a comma/space/colum/semi-column separated list of tags
69b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * from the ADB_TRACE environment variable and build the trace
70b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * mask from it. note that '1' and 'all' are special cases to
71b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber * enable all tracing
72b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber */
73b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Hubervoid  adb_trace_init(void)
74b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber{
75b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    const char*  p = getenv("ADB_TRACE");
76b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    const char*  q;
77b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
78b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    static const struct {
79b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        const char*  tag;
80b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        int           flag;
81b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    } tags[] = {
82b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "1", 0 },
83b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "all", 0 },
84b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "adb", TRACE_ADB },
85b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "sockets", TRACE_SOCKETS },
86b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "packets", TRACE_PACKETS },
87b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "rwx", TRACE_RWX },
88b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "usb", TRACE_USB },
89b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "sync", TRACE_SYNC },
90b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "sysdeps", TRACE_SYSDEPS },
91b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "transport", TRACE_TRANSPORT },
92b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { "jdwp", TRACE_JDWP },
93b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        { NULL, 0 }
94b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    };
95b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
96b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    if (p == NULL)
97b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber            return;
98b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
99b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    /* use a comma/column/semi-colum/space separated list */
100b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber    while (*p) {
101b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        int  len, tagn;
102b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
103b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        q = strpbrk(p, " ,:;");
104b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        if (q == NULL) {
105b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber            q = p + strlen(p);
106b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        }
107b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        len = q - p;
108b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
109b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        for (tagn = 0; tags[tagn].tag != NULL; tagn++)
110b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        {
111b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber            int  taglen = strlen(tags[tagn].tag);
112b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber
113b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber            if (len == taglen && !memcmp(tags[tagn].tag, p, len) )
114b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber            {
115b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber                int  flag = tags[tagn].flag;
116b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber                if (flag == 0) {
117b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber                    adb_trace_mask = ~0;
118b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber                    return;
119b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber                }
120b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber                adb_trace_mask |= (1 << flag);
121b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber                break;
122b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber            }
123b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        }
124b841f14f8e51f2365945281fbfa54ef6a1b1b5a6Andreas Huber        p = q;
125        if (*p)
126            p++;
127    }
128}
129
130
131apacket *get_apacket(void)
132{
133    apacket *p = malloc(sizeof(apacket));
134    if(p == 0) fatal("failed to allocate an apacket");
135    memset(p, 0, sizeof(apacket) - MAX_PAYLOAD);
136    return p;
137}
138
139void put_apacket(apacket *p)
140{
141    free(p);
142}
143
144void handle_online(void)
145{
146    D("adb: online\n");
147}
148
149void handle_offline(atransport *t)
150{
151    D("adb: offline\n");
152    //Close the associated usb
153    run_transport_disconnects(t);
154}
155
156#if TRACE_PACKETS
157#define DUMPMAX 32
158void print_packet(const char *label, apacket *p)
159{
160    char *tag;
161    char *x;
162    unsigned count;
163
164    switch(p->msg.command){
165    case A_SYNC: tag = "SYNC"; break;
166    case A_CNXN: tag = "CNXN" ; break;
167    case A_OPEN: tag = "OPEN"; break;
168    case A_OKAY: tag = "OKAY"; break;
169    case A_CLSE: tag = "CLSE"; break;
170    case A_WRTE: tag = "WRTE"; break;
171    default: tag = "????"; break;
172    }
173
174    fprintf(stderr, "%s: %s %08x %08x %04x \"",
175            label, tag, p->msg.arg0, p->msg.arg1, p->msg.data_length);
176    count = p->msg.data_length;
177    x = (char*) p->data;
178    if(count > DUMPMAX) {
179        count = DUMPMAX;
180        tag = "\n";
181    } else {
182        tag = "\"\n";
183    }
184    while(count-- > 0){
185        if((*x >= ' ') && (*x < 127)) {
186            fputc(*x, stderr);
187        } else {
188            fputc('.', stderr);
189        }
190        x++;
191    }
192    fprintf(stderr, tag);
193}
194#endif
195
196static void send_ready(unsigned local, unsigned remote, atransport *t)
197{
198    D("Calling send_ready \n");
199    apacket *p = get_apacket();
200    p->msg.command = A_OKAY;
201    p->msg.arg0 = local;
202    p->msg.arg1 = remote;
203    send_packet(p, t);
204}
205
206static void send_close(unsigned local, unsigned remote, atransport *t)
207{
208    D("Calling send_close \n");
209    apacket *p = get_apacket();
210    p->msg.command = A_CLSE;
211    p->msg.arg0 = local;
212    p->msg.arg1 = remote;
213    send_packet(p, t);
214}
215
216static void send_connect(atransport *t)
217{
218    D("Calling send_connect \n");
219    apacket *cp = get_apacket();
220    cp->msg.command = A_CNXN;
221    cp->msg.arg0 = A_VERSION;
222    cp->msg.arg1 = MAX_PAYLOAD;
223    snprintf((char*) cp->data, sizeof cp->data, "%s::",
224            HOST ? "host" : adb_device_banner);
225    cp->msg.data_length = strlen((char*) cp->data) + 1;
226    send_packet(cp, t);
227#if ADB_HOST
228        /* XXX why sleep here? */
229    // allow the device some time to respond to the connect message
230    adb_sleep_ms(1000);
231#endif
232}
233
234static char *connection_state_name(atransport *t)
235{
236    if (t == NULL) {
237        return "unknown";
238    }
239
240    switch(t->connection_state) {
241    case CS_BOOTLOADER:
242        return "bootloader";
243    case CS_DEVICE:
244        return "device";
245    case CS_OFFLINE:
246        return "offline";
247    default:
248        return "unknown";
249    }
250}
251
252void parse_banner(char *banner, atransport *t)
253{
254    char *type, *product, *end;
255
256    D("parse_banner: %s\n", banner);
257    type = banner;
258    product = strchr(type, ':');
259    if(product) {
260        *product++ = 0;
261    } else {
262        product = "";
263    }
264
265        /* remove trailing ':' */
266    end = strchr(product, ':');
267    if(end) *end = 0;
268
269        /* save product name in device structure */
270    if (t->product == NULL) {
271        t->product = strdup(product);
272    } else if (strcmp(product, t->product) != 0) {
273        free(t->product);
274        t->product = strdup(product);
275    }
276
277    if(!strcmp(type, "bootloader")){
278        D("setting connection_state to CS_BOOTLOADER\n");
279        t->connection_state = CS_BOOTLOADER;
280        update_transports();
281        return;
282    }
283
284    if(!strcmp(type, "device")) {
285        D("setting connection_state to CS_DEVICE\n");
286        t->connection_state = CS_DEVICE;
287        update_transports();
288        return;
289    }
290
291    if(!strcmp(type, "recovery")) {
292        D("setting connection_state to CS_RECOVERY\n");
293        t->connection_state = CS_RECOVERY;
294        update_transports();
295        return;
296    }
297
298    t->connection_state = CS_HOST;
299}
300
301void handle_packet(apacket *p, atransport *t)
302{
303    asocket *s;
304
305    D("handle_packet() %d\n", p->msg.command);
306
307    print_packet("recv", p);
308
309    switch(p->msg.command){
310    case A_SYNC:
311        if(p->msg.arg0){
312            send_packet(p, t);
313            if(HOST) send_connect(t);
314        } else {
315            t->connection_state = CS_OFFLINE;
316            handle_offline(t);
317            send_packet(p, t);
318        }
319        return;
320
321    case A_CNXN: /* CONNECT(version, maxdata, "system-id-string") */
322            /* XXX verify version, etc */
323        if(t->connection_state != CS_OFFLINE) {
324            t->connection_state = CS_OFFLINE;
325            handle_offline(t);
326        }
327        parse_banner((char*) p->data, t);
328        handle_online();
329        if(!HOST) send_connect(t);
330        break;
331
332    case A_OPEN: /* OPEN(local-id, 0, "destination") */
333        if(t->connection_state != CS_OFFLINE) {
334            char *name = (char*) p->data;
335            name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
336            s = create_local_service_socket(name);
337            if(s == 0) {
338                send_close(0, p->msg.arg0, t);
339            } else {
340                s->peer = create_remote_socket(p->msg.arg0, t);
341                s->peer->peer = s;
342                send_ready(s->id, s->peer->id, t);
343                s->ready(s);
344            }
345        }
346        break;
347
348    case A_OKAY: /* READY(local-id, remote-id, "") */
349        if(t->connection_state != CS_OFFLINE) {
350            if((s = find_local_socket(p->msg.arg1))) {
351                if(s->peer == 0) {
352                    s->peer = create_remote_socket(p->msg.arg0, t);
353                    s->peer->peer = s;
354                }
355                s->ready(s);
356            }
357        }
358        break;
359
360    case A_CLSE: /* CLOSE(local-id, remote-id, "") */
361        if(t->connection_state != CS_OFFLINE) {
362            if((s = find_local_socket(p->msg.arg1))) {
363                s->close(s);
364            }
365        }
366        break;
367
368    case A_WRTE:
369        if(t->connection_state != CS_OFFLINE) {
370            if((s = find_local_socket(p->msg.arg1))) {
371                unsigned rid = p->msg.arg0;
372                p->len = p->msg.data_length;
373
374                if(s->enqueue(s, p) == 0) {
375                    D("Enqueue the socket\n");
376                    send_ready(s->id, rid, t);
377                }
378                return;
379            }
380        }
381        break;
382
383    default:
384        printf("handle_packet: what is %08x?!\n", p->msg.command);
385    }
386
387    put_apacket(p);
388}
389
390alistener listener_list = {
391    .next = &listener_list,
392    .prev = &listener_list,
393};
394
395static void ss_listener_event_func(int _fd, unsigned ev, void *_l)
396{
397    asocket *s;
398
399    if(ev & FDE_READ) {
400        struct sockaddr addr;
401        socklen_t alen;
402        int fd;
403
404        alen = sizeof(addr);
405        fd = adb_socket_accept(_fd, &addr, &alen);
406        if(fd < 0) return;
407
408        adb_socket_setbufsize(fd, CHUNK_SIZE);
409
410        s = create_local_socket(fd);
411        if(s) {
412            connect_to_smartsocket(s);
413            return;
414        }
415
416        adb_close(fd);
417    }
418}
419
420static void listener_event_func(int _fd, unsigned ev, void *_l)
421{
422    alistener *l = _l;
423    asocket *s;
424
425    if(ev & FDE_READ) {
426        struct sockaddr addr;
427        socklen_t alen;
428        int fd;
429
430        alen = sizeof(addr);
431        fd = adb_socket_accept(_fd, &addr, &alen);
432        if(fd < 0) return;
433
434        s = create_local_socket(fd);
435        if(s) {
436            s->transport = l->transport;
437            connect_to_remote(s, l->connect_to);
438            return;
439        }
440
441        adb_close(fd);
442    }
443}
444
445static void  free_listener(alistener*  l)
446{
447    if (l->next) {
448        l->next->prev = l->prev;
449        l->prev->next = l->next;
450        l->next = l->prev = l;
451    }
452
453    // closes the corresponding fd
454    fdevent_remove(&l->fde);
455
456    if (l->local_name)
457        free((char*)l->local_name);
458
459    if (l->connect_to)
460        free((char*)l->connect_to);
461
462    if (l->transport) {
463        remove_transport_disconnect(l->transport, &l->disconnect);
464    }
465    free(l);
466}
467
468static void listener_disconnect(void*  _l, atransport*  t)
469{
470    alistener*  l = _l;
471
472    free_listener(l);
473}
474
475int local_name_to_fd(const char *name)
476{
477    int port;
478
479    if(!strncmp("tcp:", name, 4)){
480        int  ret;
481        port = atoi(name + 4);
482        ret = socket_loopback_server(port, SOCK_STREAM);
483        return ret;
484    }
485#ifndef HAVE_WIN32_IPC  /* no Unix-domain sockets on Win32 */
486    // It's non-sensical to support the "reserved" space on the adb host side
487    if(!strncmp(name, "local:", 6)) {
488        return socket_local_server(name + 6,
489                ANDROID_SOCKET_NAMESPACE_ABSTRACT, SOCK_STREAM);
490    } else if(!strncmp(name, "localabstract:", 14)) {
491        return socket_local_server(name + 14,
492                ANDROID_SOCKET_NAMESPACE_ABSTRACT, SOCK_STREAM);
493    } else if(!strncmp(name, "localfilesystem:", 16)) {
494        return socket_local_server(name + 16,
495                ANDROID_SOCKET_NAMESPACE_FILESYSTEM, SOCK_STREAM);
496    }
497
498#endif
499    printf("unknown local portname '%s'\n", name);
500    return -1;
501}
502
503static int remove_listener(const char *local_name, const char *connect_to, atransport* transport)
504{
505    alistener *l;
506
507    for (l = listener_list.next; l != &listener_list; l = l->next) {
508        if (!strcmp(local_name, l->local_name) &&
509            !strcmp(connect_to, l->connect_to) &&
510            l->transport && l->transport == transport) {
511
512            listener_disconnect(l, transport);
513            return 0;
514        }
515    }
516
517    return -1;
518}
519
520static int install_listener(const char *local_name, const char *connect_to, atransport* transport)
521{
522    alistener *l;
523
524    //printf("install_listener('%s','%s')\n", local_name, connect_to);
525
526    for(l = listener_list.next; l != &listener_list; l = l->next){
527        if(strcmp(local_name, l->local_name) == 0) {
528            char *cto;
529
530                /* can't repurpose a smartsocket */
531            if(l->connect_to[0] == '*') {
532                return -1;
533            }
534
535            cto = strdup(connect_to);
536            if(cto == 0) {
537                return -1;
538            }
539
540            //printf("rebinding '%s' to '%s'\n", local_name, connect_to);
541            free((void*) l->connect_to);
542            l->connect_to = cto;
543            if (l->transport != transport) {
544                remove_transport_disconnect(l->transport, &l->disconnect);
545                l->transport = transport;
546                add_transport_disconnect(l->transport, &l->disconnect);
547            }
548            return 0;
549        }
550    }
551
552    if((l = calloc(1, sizeof(alistener))) == 0) goto nomem;
553    if((l->local_name = strdup(local_name)) == 0) goto nomem;
554    if((l->connect_to = strdup(connect_to)) == 0) goto nomem;
555
556
557    l->fd = local_name_to_fd(local_name);
558    if(l->fd < 0) {
559        free((void*) l->local_name);
560        free((void*) l->connect_to);
561        free(l);
562        printf("cannot bind '%s'\n", local_name);
563        return -2;
564    }
565
566    close_on_exec(l->fd);
567    if(!strcmp(l->connect_to, "*smartsocket*")) {
568        fdevent_install(&l->fde, l->fd, ss_listener_event_func, l);
569    } else {
570        fdevent_install(&l->fde, l->fd, listener_event_func, l);
571    }
572    fdevent_set(&l->fde, FDE_READ);
573
574    l->next = &listener_list;
575    l->prev = listener_list.prev;
576    l->next->prev = l;
577    l->prev->next = l;
578    l->transport = transport;
579
580    if (transport) {
581        l->disconnect.opaque = l;
582        l->disconnect.func   = listener_disconnect;
583        add_transport_disconnect(transport, &l->disconnect);
584    }
585    return 0;
586
587nomem:
588    fatal("cannot allocate listener");
589    return 0;
590}
591
592#ifdef HAVE_FORKEXEC
593static void sigchld_handler(int n)
594{
595    int status;
596    while(waitpid(-1, &status, WNOHANG) > 0) ;
597}
598#endif
599
600#ifdef HAVE_WIN32_PROC
601static BOOL WINAPI ctrlc_handler(DWORD type)
602{
603    exit(STATUS_CONTROL_C_EXIT);
604    return TRUE;
605}
606#endif
607
608static void adb_cleanup(void)
609{
610    usb_cleanup();
611}
612
613void start_logging(void)
614{
615#ifdef HAVE_WIN32_PROC
616    char    temp[ MAX_PATH ];
617    FILE*   fnul;
618    FILE*   flog;
619
620    GetTempPath( sizeof(temp) - 8, temp );
621    strcat( temp, "adb.log" );
622
623    /* Win32 specific redirections */
624    fnul = fopen( "NUL", "rt" );
625    if (fnul != NULL)
626        stdin[0] = fnul[0];
627
628    flog = fopen( temp, "at" );
629    if (flog == NULL)
630        flog = fnul;
631
632    setvbuf( flog, NULL, _IONBF, 0 );
633
634    stdout[0] = flog[0];
635    stderr[0] = flog[0];
636    fprintf(stderr,"--- adb starting (pid %d) ---\n", getpid());
637#else
638    int fd;
639
640    fd = unix_open("/dev/null", O_RDONLY);
641    dup2(fd, 0);
642
643    fd = unix_open("/tmp/adb.log", O_WRONLY | O_CREAT | O_APPEND, 0640);
644    if(fd < 0) {
645        fd = unix_open("/dev/null", O_WRONLY);
646    }
647    dup2(fd, 1);
648    dup2(fd, 2);
649    fprintf(stderr,"--- adb starting (pid %d) ---\n", getpid());
650#endif
651}
652
653#if !ADB_HOST
654void start_device_log(void)
655{
656    int fd;
657    char    path[PATH_MAX];
658    struct tm now;
659    time_t t;
660    char value[PROPERTY_VALUE_MAX];
661
662    // read the trace mask from persistent property persist.adb.trace_mask
663    // give up if the property is not set or cannot be parsed
664    property_get("persist.adb.trace_mask", value, "");
665    if (sscanf(value, "%x", &adb_trace_mask) != 1)
666        return;
667
668    adb_mkdir("/data/adb", 0775);
669    tzset();
670    time(&t);
671    localtime_r(&t, &now);
672    strftime(path, sizeof(path),
673                "/data/adb/adb-%Y-%m-%d-%H-%M-%S.txt",
674                &now);
675    fd = unix_open(path, O_WRONLY | O_CREAT | O_TRUNC, 0640);
676    if (fd < 0)
677        return;
678
679    // redirect stdout and stderr to the log file
680    dup2(fd, 1);
681    dup2(fd, 2);
682    fprintf(stderr,"--- adb starting (pid %d) ---\n", getpid());
683
684    fd = unix_open("/dev/null", O_RDONLY);
685    dup2(fd, 0);
686}
687#endif
688
689#if ADB_HOST
690int launch_server(int server_port)
691{
692#ifdef HAVE_WIN32_PROC
693    /* we need to start the server in the background                    */
694    /* we create a PIPE that will be used to wait for the server's "OK" */
695    /* message since the pipe handles must be inheritable, we use a     */
696    /* security attribute                                               */
697    HANDLE                pipe_read, pipe_write;
698    SECURITY_ATTRIBUTES   sa;
699    STARTUPINFO           startup;
700    PROCESS_INFORMATION   pinfo;
701    char                  program_path[ MAX_PATH ];
702    int                   ret;
703
704    sa.nLength = sizeof(sa);
705    sa.lpSecurityDescriptor = NULL;
706    sa.bInheritHandle = TRUE;
707
708    /* create pipe, and ensure its read handle isn't inheritable */
709    ret = CreatePipe( &pipe_read, &pipe_write, &sa, 0 );
710    if (!ret) {
711        fprintf(stderr, "CreatePipe() failure, error %ld\n", GetLastError() );
712        return -1;
713    }
714
715    SetHandleInformation( pipe_read, HANDLE_FLAG_INHERIT, 0 );
716
717    ZeroMemory( &startup, sizeof(startup) );
718    startup.cb = sizeof(startup);
719    startup.hStdInput  = GetStdHandle( STD_INPUT_HANDLE );
720    startup.hStdOutput = pipe_write;
721    startup.hStdError  = GetStdHandle( STD_ERROR_HANDLE );
722    startup.dwFlags    = STARTF_USESTDHANDLES;
723
724    ZeroMemory( &pinfo, sizeof(pinfo) );
725
726    /* get path of current program */
727    GetModuleFileName( NULL, program_path, sizeof(program_path) );
728
729    ret = CreateProcess(
730            program_path,                              /* program path  */
731            "adb fork-server server",
732                                    /* the fork-server argument will set the
733                                       debug = 2 in the child           */
734            NULL,                   /* process handle is not inheritable */
735            NULL,                    /* thread handle is not inheritable */
736            TRUE,                          /* yes, inherit some handles */
737            DETACHED_PROCESS, /* the new process doesn't have a console */
738            NULL,                     /* use parent's environment block */
739            NULL,                    /* use parent's starting directory */
740            &startup,                 /* startup info, i.e. std handles */
741            &pinfo );
742
743    CloseHandle( pipe_write );
744
745    if (!ret) {
746        fprintf(stderr, "CreateProcess failure, error %ld\n", GetLastError() );
747        CloseHandle( pipe_read );
748        return -1;
749    }
750
751    CloseHandle( pinfo.hProcess );
752    CloseHandle( pinfo.hThread );
753
754    /* wait for the "OK\n" message */
755    {
756        char  temp[3];
757        DWORD  count;
758
759        ret = ReadFile( pipe_read, temp, 3, &count, NULL );
760        CloseHandle( pipe_read );
761        if ( !ret ) {
762            fprintf(stderr, "could not read ok from ADB Server, error = %ld\n", GetLastError() );
763            return -1;
764        }
765        if (count != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
766            fprintf(stderr, "ADB server didn't ACK\n" );
767            return -1;
768        }
769    }
770#elif defined(HAVE_FORKEXEC)
771    char    path[PATH_MAX];
772    int     fd[2];
773
774    // set up a pipe so the child can tell us when it is ready.
775    // fd[0] will be parent's end, and fd[1] will get mapped to stderr in the child.
776    if (pipe(fd)) {
777        fprintf(stderr, "pipe failed in launch_server, errno: %d\n", errno);
778        return -1;
779    }
780    get_my_path(path, PATH_MAX);
781    pid_t pid = fork();
782    if(pid < 0) return -1;
783
784    if (pid == 0) {
785        // child side of the fork
786
787        // redirect stderr to the pipe
788        // we use stderr instead of stdout due to stdout's buffering behavior.
789        adb_close(fd[0]);
790        dup2(fd[1], STDERR_FILENO);
791        adb_close(fd[1]);
792
793        // child process
794        int result = execl(path, "adb", "fork-server", "server", NULL);
795        // this should not return
796        fprintf(stderr, "OOPS! execl returned %d, errno: %d\n", result, errno);
797    } else  {
798        // parent side of the fork
799
800        char  temp[3];
801
802        temp[0] = 'A'; temp[1] = 'B'; temp[2] = 'C';
803        // wait for the "OK\n" message
804        adb_close(fd[1]);
805        int ret = adb_read(fd[0], temp, 3);
806        adb_close(fd[0]);
807        if (ret < 0) {
808            fprintf(stderr, "could not read ok from ADB Server, errno = %d\n", errno);
809            return -1;
810        }
811        if (ret != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
812            fprintf(stderr, "ADB server didn't ACK\n" );
813            return -1;
814        }
815
816        setsid();
817    }
818#else
819#error "cannot implement background server start on this platform"
820#endif
821    return 0;
822}
823#endif
824
825/* Constructs a local name of form tcp:port.
826 * target_str points to the target string, it's content will be overwritten.
827 * target_size is the capacity of the target string.
828 * server_port is the port number to use for the local name.
829 */
830void build_local_name(char* target_str, size_t target_size, int server_port)
831{
832  snprintf(target_str, target_size, "tcp:%d", server_port);
833}
834
835int adb_main(int is_daemon, int server_port)
836{
837#if !ADB_HOST
838    int secure = 0;
839    int port;
840    char value[PROPERTY_VALUE_MAX];
841#endif
842
843    atexit(adb_cleanup);
844#ifdef HAVE_WIN32_PROC
845    SetConsoleCtrlHandler( ctrlc_handler, TRUE );
846#elif defined(HAVE_FORKEXEC)
847    signal(SIGCHLD, sigchld_handler);
848    signal(SIGPIPE, SIG_IGN);
849#endif
850
851    init_transport_registration();
852
853
854#if ADB_HOST
855    HOST = 1;
856    usb_vendors_init();
857    usb_init();
858    local_init(DEFAULT_ADB_LOCAL_TRANSPORT_PORT);
859
860    char local_name[30];
861    build_local_name(local_name, sizeof(local_name), server_port);
862    if(install_listener(local_name, "*smartsocket*", NULL)) {
863        exit(1);
864    }
865#else
866    /* run adbd in secure mode if ro.secure is set and
867    ** we are not in the emulator
868    */
869    property_get("ro.kernel.qemu", value, "");
870    if (strcmp(value, "1") != 0) {
871        property_get("ro.secure", value, "");
872        if (strcmp(value, "1") == 0) {
873            // don't run as root if ro.secure is set...
874            secure = 1;
875
876            // ... except we allow running as root in userdebug builds if the
877            // service.adb.root property has been set by the "adb root" command
878            property_get("ro.debuggable", value, "");
879            if (strcmp(value, "1") == 0) {
880                property_get("service.adb.root", value, "");
881                if (strcmp(value, "1") == 0) {
882                    secure = 0;
883                }
884            }
885        }
886    }
887
888    /* don't listen on a port (default 5037) if running in secure mode */
889    /* don't run as root if we are running in secure mode */
890    if (secure) {
891        struct __user_cap_header_struct header;
892        struct __user_cap_data_struct cap;
893
894        prctl(PR_SET_KEEPCAPS, 1, 0, 0, 0);
895
896        /* add extra groups:
897        ** AID_ADB to access the USB driver
898        ** AID_LOG to read system logs (adb logcat)
899        ** AID_INPUT to diagnose input issues (getevent)
900        ** AID_INET to diagnose network issues (netcfg, ping)
901        ** AID_GRAPHICS to access the frame buffer
902        ** AID_NET_BT and AID_NET_BT_ADMIN to diagnose bluetooth (hcidump)
903        ** AID_SDCARD_RW to allow writing to the SD card
904        ** AID_MOUNT to allow unmounting the SD card before rebooting
905        */
906        gid_t groups[] = { AID_ADB, AID_LOG, AID_INPUT, AID_INET, AID_GRAPHICS,
907                           AID_NET_BT, AID_NET_BT_ADMIN, AID_SDCARD_RW, AID_MOUNT };
908        setgroups(sizeof(groups)/sizeof(groups[0]), groups);
909
910        /* then switch user and group to "shell" */
911        setgid(AID_SHELL);
912        setuid(AID_SHELL);
913
914        /* set CAP_SYS_BOOT capability, so "adb reboot" will succeed */
915        header.version = _LINUX_CAPABILITY_VERSION;
916        header.pid = 0;
917        cap.effective = cap.permitted = (1 << CAP_SYS_BOOT);
918        cap.inheritable = 0;
919        capset(&header, &cap);
920
921        D("Local port disabled\n");
922    } else {
923        char local_name[30];
924        build_local_name(local_name, sizeof(local_name), server_port);
925        if(install_listener(local_name, "*smartsocket*", NULL)) {
926            exit(1);
927        }
928    }
929
930        /* for the device, start the usb transport if the
931        ** android usb device exists and the "service.adb.tcp.port" and
932        ** "persist.adb.tcp.port" properties are not set.
933        ** Otherwise start the network transport.
934        */
935    property_get("service.adb.tcp.port", value, "");
936    if (!value[0])
937        property_get("persist.adb.tcp.port", value, "");
938    if (sscanf(value, "%d", &port) == 1 && port > 0) {
939        // listen on TCP port specified by service.adb.tcp.port property
940        local_init(port);
941    } else if (access("/dev/android_adb", F_OK) == 0) {
942        // listen on USB
943        usb_init();
944    } else {
945        // listen on default port
946        local_init(DEFAULT_ADB_LOCAL_TRANSPORT_PORT);
947    }
948    init_jdwp();
949#endif
950
951    if (is_daemon)
952    {
953        // inform our parent that we are up and running.
954#ifdef HAVE_WIN32_PROC
955        DWORD  count;
956        WriteFile( GetStdHandle( STD_OUTPUT_HANDLE ), "OK\n", 3, &count, NULL );
957#elif defined(HAVE_FORKEXEC)
958        fprintf(stderr, "OK\n");
959#endif
960        start_logging();
961    }
962
963    fdevent_loop();
964
965    usb_cleanup();
966
967    return 0;
968}
969
970#if ADB_HOST
971void connect_device(char* host, char* buffer, int buffer_size)
972{
973    int port, fd;
974    char* portstr = strchr(host, ':');
975    char buf[4096];
976
977    if (!portstr) {
978        snprintf(buffer, buffer_size, "unable to parse %s as <host>:<port>", host);
979        return;
980    }
981    if (find_transport(host)) {
982        snprintf(buffer, buffer_size, "already connected to %s", host);
983        return;
984    }
985
986    // zero terminate host by overwriting the ':'
987    *portstr++ = 0;
988    if (sscanf(portstr, "%d", &port) == 0) {
989        snprintf(buffer, buffer_size, "bad port number %s", portstr);
990        return;
991    }
992
993    fd = socket_network_client(host, port, SOCK_STREAM);
994    if (fd < 0) {
995        snprintf(buffer, buffer_size, "unable to connect to %s:%d", host, port);
996        return;
997    }
998
999    D("client: connected on remote on fd %d\n", fd);
1000    close_on_exec(fd);
1001    disable_tcp_nagle(fd);
1002    snprintf(buf, sizeof buf, "%s:%d", host, port);
1003    register_socket_transport(fd, buf, port, 0);
1004    snprintf(buffer, buffer_size, "connected to %s:%d", host, port);
1005}
1006
1007void connect_emulator(char* port_spec, char* buffer, int buffer_size)
1008{
1009    char* port_separator = strchr(port_spec, ',');
1010    if (!port_separator) {
1011        snprintf(buffer, buffer_size,
1012                "unable to parse '%s' as <console port>,<adb port>",
1013                port_spec);
1014        return;
1015    }
1016
1017    // Zero-terminate console port and make port_separator point to 2nd port.
1018    *port_separator++ = 0;
1019    int console_port = strtol(port_spec, NULL, 0);
1020    int adb_port = strtol(port_separator, NULL, 0);
1021    if (!(console_port > 0 && adb_port > 0)) {
1022        *(port_separator - 1) = ',';
1023        snprintf(buffer, buffer_size,
1024                "Invalid port numbers: Expected positive numbers, got '%s'",
1025                port_spec);
1026        return;
1027    }
1028
1029    /* Check if the emulator is already known.
1030     * Note: There's a small but harmless race condition here: An emulator not
1031     * present just yet could be registered by another invocation right
1032     * after doing this check here. However, local_connect protects
1033     * against double-registration too. From here, a better error message
1034     * can be produced. In the case of the race condition, the very specific
1035     * error message won't be shown, but the data doesn't get corrupted. */
1036    atransport* known_emulator = find_emulator_transport_by_adb_port(adb_port);
1037    if (known_emulator != NULL) {
1038        snprintf(buffer, buffer_size,
1039                "Emulator on port %d already registered.", adb_port);
1040        return;
1041    }
1042
1043    /* Check if more emulators can be registered. Similar unproblematic
1044     * race condition as above. */
1045    int candidate_slot = get_available_local_transport_index();
1046    if (candidate_slot < 0) {
1047        snprintf(buffer, buffer_size, "Cannot accept more emulators.");
1048        return;
1049    }
1050
1051    /* Preconditions met, try to connect to the emulator. */
1052    if (!local_connect_arbitrary_ports(console_port, adb_port)) {
1053        snprintf(buffer, buffer_size,
1054                "Connected to emulator on ports %d,%d", console_port, adb_port);
1055    } else {
1056        snprintf(buffer, buffer_size,
1057                "Could not connect to emulator on ports %d,%d",
1058                console_port, adb_port);
1059    }
1060}
1061#endif
1062
1063int handle_host_request(char *service, transport_type ttype, char* serial, int reply_fd, asocket *s)
1064{
1065    atransport *transport = NULL;
1066    char buf[4096];
1067
1068    if(!strcmp(service, "kill")) {
1069        fprintf(stderr,"adb server killed by remote request\n");
1070        fflush(stdout);
1071        adb_write(reply_fd, "OKAY", 4);
1072        usb_cleanup();
1073        exit(0);
1074    }
1075
1076#if ADB_HOST
1077    // "transport:" is used for switching transport with a specified serial number
1078    // "transport-usb:" is used for switching transport to the only USB transport
1079    // "transport-local:" is used for switching transport to the only local transport
1080    // "transport-any:" is used for switching transport to the only transport
1081    if (!strncmp(service, "transport", strlen("transport"))) {
1082        char* error_string = "unknown failure";
1083        transport_type type = kTransportAny;
1084
1085        if (!strncmp(service, "transport-usb", strlen("transport-usb"))) {
1086            type = kTransportUsb;
1087        } else if (!strncmp(service, "transport-local", strlen("transport-local"))) {
1088            type = kTransportLocal;
1089        } else if (!strncmp(service, "transport-any", strlen("transport-any"))) {
1090            type = kTransportAny;
1091        } else if (!strncmp(service, "transport:", strlen("transport:"))) {
1092            service += strlen("transport:");
1093            serial = strdup(service);
1094        }
1095
1096        transport = acquire_one_transport(CS_ANY, type, serial, &error_string);
1097
1098        if (transport) {
1099            s->transport = transport;
1100            adb_write(reply_fd, "OKAY", 4);
1101        } else {
1102            sendfailmsg(reply_fd, error_string);
1103        }
1104        return 1;
1105    }
1106
1107    // return a list of all connected devices
1108    if (!strcmp(service, "devices")) {
1109        char buffer[4096];
1110        memset(buf, 0, sizeof(buf));
1111        memset(buffer, 0, sizeof(buffer));
1112        D("Getting device list \n");
1113        list_transports(buffer, sizeof(buffer));
1114        snprintf(buf, sizeof(buf), "OKAY%04x%s",(unsigned)strlen(buffer),buffer);
1115        D("Wrote device list \n");
1116        writex(reply_fd, buf, strlen(buf));
1117        return 0;
1118    }
1119
1120    // add a new TCP transport, device or emulator
1121    if (!strncmp(service, "connect:", 8)) {
1122        char buffer[4096];
1123        char* host = service + 8;
1124        if (!strncmp(host, "emu:", 4)) {
1125            connect_emulator(host + 4, buffer, sizeof(buffer));
1126        } else {
1127            connect_device(host, buffer, sizeof(buffer));
1128        }
1129        // Send response for emulator and device
1130        snprintf(buf, sizeof(buf), "OKAY%04x%s",(unsigned)strlen(buffer), buffer);
1131        writex(reply_fd, buf, strlen(buf));
1132        return 0;
1133    }
1134
1135    // remove TCP transport
1136    if (!strncmp(service, "disconnect:", 11)) {
1137        char buffer[4096];
1138        memset(buffer, 0, sizeof(buffer));
1139        char* serial = service + 11;
1140        atransport *t = find_transport(serial);
1141
1142        if (t) {
1143            unregister_transport(t);
1144        } else {
1145            snprintf(buffer, sizeof(buffer), "No such device %s", serial);
1146        }
1147
1148        snprintf(buf, sizeof(buf), "OKAY%04x%s",(unsigned)strlen(buffer), buffer);
1149        writex(reply_fd, buf, strlen(buf));
1150        return 0;
1151    }
1152
1153    // returns our value for ADB_SERVER_VERSION
1154    if (!strcmp(service, "version")) {
1155        char version[12];
1156        snprintf(version, sizeof version, "%04x", ADB_SERVER_VERSION);
1157        snprintf(buf, sizeof buf, "OKAY%04x%s", (unsigned)strlen(version), version);
1158        writex(reply_fd, buf, strlen(buf));
1159        return 0;
1160    }
1161
1162    if(!strncmp(service,"get-serialno",strlen("get-serialno"))) {
1163        char *out = "unknown";
1164         transport = acquire_one_transport(CS_ANY, ttype, serial, NULL);
1165       if (transport && transport->serial) {
1166            out = transport->serial;
1167        }
1168        snprintf(buf, sizeof buf, "OKAY%04x%s",(unsigned)strlen(out),out);
1169        writex(reply_fd, buf, strlen(buf));
1170        return 0;
1171    }
1172    // indicates a new emulator instance has started
1173    if (!strncmp(service,"emulator:",9)) {
1174        int  port = atoi(service+9);
1175        local_connect(port);
1176        /* we don't even need to send a reply */
1177        return 0;
1178    }
1179#endif // ADB_HOST
1180
1181    if(!strncmp(service,"forward:",8) || !strncmp(service,"killforward:",12)) {
1182        char *local, *remote, *err;
1183        int r;
1184        atransport *transport;
1185
1186        int createForward = strncmp(service,"kill",4);
1187
1188        local = service + (createForward ? 8 : 12);
1189        remote = strchr(local,';');
1190        if(remote == 0) {
1191            sendfailmsg(reply_fd, "malformed forward spec");
1192            return 0;
1193        }
1194
1195        *remote++ = 0;
1196        if((local[0] == 0) || (remote[0] == 0) || (remote[0] == '*')){
1197            sendfailmsg(reply_fd, "malformed forward spec");
1198            return 0;
1199        }
1200
1201        transport = acquire_one_transport(CS_ANY, ttype, serial, &err);
1202        if (!transport) {
1203            sendfailmsg(reply_fd, err);
1204            return 0;
1205        }
1206
1207        if (createForward) {
1208            r = install_listener(local, remote, transport);
1209        } else {
1210            r = remove_listener(local, remote, transport);
1211        }
1212        if(r == 0) {
1213                /* 1st OKAY is connect, 2nd OKAY is status */
1214            writex(reply_fd, "OKAYOKAY", 8);
1215            return 0;
1216        }
1217
1218        if (createForward) {
1219            sendfailmsg(reply_fd, (r == -1) ? "cannot rebind smartsocket" : "cannot bind socket");
1220        } else {
1221            sendfailmsg(reply_fd, "cannot remove listener");
1222        }
1223        return 0;
1224    }
1225
1226    if(!strncmp(service,"get-state",strlen("get-state"))) {
1227        transport = acquire_one_transport(CS_ANY, ttype, serial, NULL);
1228        char *state = connection_state_name(transport);
1229        snprintf(buf, sizeof buf, "OKAY%04x%s",(unsigned)strlen(state),state);
1230        writex(reply_fd, buf, strlen(buf));
1231        return 0;
1232    }
1233    return -1;
1234}
1235
1236#if !ADB_HOST
1237int recovery_mode = 0;
1238#endif
1239
1240int main(int argc, char **argv)
1241{
1242    adb_trace_init();
1243#if ADB_HOST
1244    adb_sysdeps_init();
1245    return adb_commandline(argc - 1, argv + 1);
1246#else
1247    if((argc > 1) && (!strcmp(argv[1],"recovery"))) {
1248        adb_device_banner = "recovery";
1249        recovery_mode = 1;
1250    }
1251
1252    start_device_log();
1253    return adb_main(0, DEFAULT_ADB_PORT);
1254#endif
1255}
1256