services.c revision be0045aafdbc2ec2ee448be86952082361f5d672 
1/*
2 * Copyright (C) 2007 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <stdlib.h>
18#include <stdio.h>
19#include <unistd.h>
20#include <string.h>
21#include <errno.h>
22
23#include "sysdeps.h"
24
25#define  TRACE_TAG  TRACE_ADB
26#include "adb.h"
27#include "file_sync_service.h"
28
29#if ADB_HOST
30#  ifndef HAVE_WINSOCK
31#    include <netinet/in.h>
32#    include <netdb.h>
33#  endif
34#else
35#include <sys/poll.h>
36#include <sys/reboot.h>
37#endif
38
39typedef struct stinfo stinfo;
40
41struct stinfo {
42    void (*func)(int fd, void *cookie);
43    int fd;
44    void *cookie;
45};
46
47
48void *service_bootstrap_func(void *x)
49{
50    stinfo *sti = x;
51    sti->func(sti->fd, sti->cookie);
52    free(sti);
53    return 0;
54}
55
56#if ADB_HOST
57ADB_MUTEX_DEFINE( dns_lock );
58
59static void dns_service(int fd, void *cookie)
60{
61    char *hostname = cookie;
62    struct hostent *hp;
63    unsigned zero = 0;
64
65    adb_mutex_lock(&dns_lock);
66    hp = gethostbyname(hostname);
67    if(hp == 0) {
68        writex(fd, &zero, 4);
69    } else {
70        writex(fd, hp->h_addr, 4);
71    }
72    adb_mutex_unlock(&dns_lock);
73    adb_close(fd);
74}
75#else
76extern int recovery_mode;
77
78static void recover_service(int s, void *cookie)
79{
80    unsigned char buf[4096];
81    unsigned count = (unsigned) cookie;
82    int fd;
83
84    fd = adb_creat("/tmp/update", 0644);
85    if(fd < 0) {
86        adb_close(s);
87        return;
88    }
89
90    while(count > 0) {
91        unsigned xfer = (count > 4096) ? 4096 : count;
92        if(readx(s, buf, xfer)) break;
93        if(writex(fd, buf, xfer)) break;
94        count -= xfer;
95    }
96
97    if(count == 0) {
98        writex(s, "OKAY", 4);
99    } else {
100        writex(s, "FAIL", 4);
101    }
102    adb_close(fd);
103    adb_close(s);
104
105    fd = adb_creat("/tmp/update.begin", 0644);
106    adb_close(fd);
107}
108
109void restart_root_service(int fd, void *cookie)
110{
111    char buf[100];
112    char value[PROPERTY_VALUE_MAX];
113
114    if (getuid() == 0) {
115        snprintf(buf, sizeof(buf), "adbd is already running as root\n");
116        writex(fd, buf, strlen(buf));
117        adb_close(fd);
118    } else {
119        property_get("ro.debuggable", value, "");
120        if (strcmp(value, "1") != 0) {
121            snprintf(buf, sizeof(buf), "adbd cannot run as root in production builds\n");
122            writex(fd, buf, strlen(buf));
123            adb_close(fd);
124            return;
125        }
126
127        property_set("service.adb.root", "1");
128        snprintf(buf, sizeof(buf), "restarting adbd as root\n");
129        writex(fd, buf, strlen(buf));
130        adb_close(fd);
131
132        // quit, and init will restart us as root
133        sleep(1);
134        exit(1);
135    }
136}
137
138void restart_tcp_service(int fd, void *cookie)
139{
140    char buf[100];
141    char value[PROPERTY_VALUE_MAX];
142    int port = (int)cookie;
143
144    if (port <= 0) {
145        snprintf(buf, sizeof(buf), "invalid port\n");
146        writex(fd, buf, strlen(buf));
147        adb_close(fd);
148        return;
149    }
150
151    snprintf(value, sizeof(value), "%d", port);
152    property_set("service.adb.tcp.port", value);
153    snprintf(buf, sizeof(buf), "restarting in TCP mode port: %d\n", port);
154    writex(fd, buf, strlen(buf));
155    adb_close(fd);
156
157    // quit, and init will restart us in TCP mode
158    sleep(1);
159    exit(1);
160}
161
162void restart_usb_service(int fd, void *cookie)
163{
164    char buf[100];
165
166    property_set("service.adb.tcp.port", "0");
167    snprintf(buf, sizeof(buf), "restarting in USB mode\n");
168    writex(fd, buf, strlen(buf));
169    adb_close(fd);
170
171    // quit, and init will restart us in USB mode
172    sleep(1);
173    exit(1);
174}
175
176void reboot_service(int fd, void *arg)
177{
178    char buf[100];
179    int ret;
180
181    sync();
182    ret = __reboot(LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2,
183                    LINUX_REBOOT_CMD_RESTART2, (char *)arg);
184    if (ret < 0) {
185        snprintf(buf, sizeof(buf), "reboot failed: %s\n", strerror(errno));
186        writex(fd, buf, strlen(buf));
187    }
188    adb_close(fd);
189}
190
191void bugreport_service(int fd, void *cookie)
192{
193    char    buffer[MAX_PAYLOAD];
194    int i, s;
195
196    /* start the dumpstate service */
197    property_set("ctl.start", "dumpstate");
198
199    /* socket will not be available until service starts */
200    for (i = 0; i < 10; i++) {
201        s = socket_local_client("dumpstate",
202                             ANDROID_SOCKET_NAMESPACE_RESERVED,
203                             SOCK_STREAM);
204        if (s >= 0)
205            break;
206        /* try again in 1 second */
207        sleep(1);
208    }
209
210    if (s < 0) {
211        const char* failed = "Failed to connect to dumpstate service\n";
212        writex(fd, failed, strlen(failed));
213        adb_close(fd);
214        return;
215    }
216
217    while (1) {
218        int length = adb_read(s, buffer, sizeof(buffer));
219        if (length <= 0)
220            break;
221        if (adb_write(fd, buffer, length) <= 0)
222            break;
223    }
224    adb_close(s);
225    adb_close(fd);
226}
227
228#endif
229
230#if 0
231static void echo_service(int fd, void *cookie)
232{
233    char buf[4096];
234    int r;
235    char *p;
236    int c;
237
238    for(;;) {
239        r = read(fd, buf, 4096);
240        if(r == 0) goto done;
241        if(r < 0) {
242            if(errno == EINTR) continue;
243            else goto done;
244        }
245
246        c = r;
247        p = buf;
248        while(c > 0) {
249            r = write(fd, p, c);
250            if(r > 0) {
251                c -= r;
252                p += r;
253                continue;
254            }
255            if((r < 0) && (errno == EINTR)) continue;
256            goto done;
257        }
258    }
259done:
260    close(fd);
261}
262#endif
263
264static int create_service_thread(void (*func)(int, void *), void *cookie)
265{
266    stinfo *sti;
267    adb_thread_t t;
268    int s[2];
269
270    if(adb_socketpair(s)) {
271        printf("cannot create service socket pair\n");
272        return -1;
273    }
274
275    sti = malloc(sizeof(stinfo));
276    if(sti == 0) fatal("cannot allocate stinfo");
277    sti->func = func;
278    sti->cookie = cookie;
279    sti->fd = s[1];
280
281    if(adb_thread_create( &t, service_bootstrap_func, sti)){
282        free(sti);
283        adb_close(s[0]);
284        adb_close(s[1]);
285        printf("cannot create service thread\n");
286        return -1;
287    }
288
289    D("service thread started, %d:%d\n",s[0], s[1]);
290    return s[0];
291}
292
293#if !ADB_HOST
294static int create_subprocess(const char *cmd, const char *arg0, const char *arg1)
295{
296    char *devname;
297    int ptm;
298    pid_t pid;
299
300    ptm = unix_open("/dev/ptmx", O_RDWR); // | O_NOCTTY);
301    if(ptm < 0){
302        printf("[ cannot open /dev/ptmx - %s ]\n",strerror(errno));
303        return -1;
304    }
305    fcntl(ptm, F_SETFD, FD_CLOEXEC);
306
307    if(grantpt(ptm) || unlockpt(ptm) ||
308       ((devname = (char*) ptsname(ptm)) == 0)){
309        printf("[ trouble with /dev/ptmx - %s ]\n", strerror(errno));
310        return -1;
311    }
312
313    pid = fork();
314    if(pid < 0) {
315        printf("- fork failed: %s -\n", strerror(errno));
316        return -1;
317    }
318
319    if(pid == 0){
320        int pts;
321
322        setsid();
323
324        pts = unix_open(devname, O_RDWR);
325        if(pts < 0) exit(-1);
326
327        dup2(pts, 0);
328        dup2(pts, 1);
329        dup2(pts, 2);
330
331        adb_close(ptm);
332
333        execl(cmd, cmd, arg0, arg1, NULL);
334        fprintf(stderr, "- exec '%s' failed: %s (%d) -\n",
335                cmd, strerror(errno), errno);
336        exit(-1);
337    } else {
338        // set child's OOM adjustment to zero
339        char text[64];
340        snprintf(text, sizeof text, "/proc/%d/oom_adj", pid);
341        int fd = adb_open(text, O_WRONLY);
342        if (fd >= 0) {
343            adb_write(fd, "0", 1);
344            adb_close(fd);
345        } else {
346           D("adb: unable to open %s\n", text);
347        }
348
349        return ptm;
350    }
351}
352#endif /* !ADB_HOST */
353
354#if ADB_HOST
355#define SHELL_COMMAND "/bin/sh"
356#else
357#define SHELL_COMMAND "/system/bin/sh"
358#endif
359
360#if !ADB_HOST
361static void shell_service(int s, void *command)
362{
363    char    buffer[MAX_PAYLOAD];
364    char    buffer2[MAX_PAYLOAD];
365    struct pollfd ufds[2];
366    int     fd, ret = 0;
367    unsigned count = 0;
368    char** args = (char **)command;
369    fd = create_subprocess(SHELL_COMMAND, args[0], args[1]);
370
371    while (1) {
372        while (count < sizeof(buffer)) {
373            ufds[0].fd = fd;
374            ufds[0].events = POLLIN | POLLHUP;
375            ufds[0].revents = 0;
376            ufds[1].fd = s;
377            ufds[1].events = POLLIN | POLLHUP;
378            ufds[1].revents = 0;
379            // use a 100ms timeout so we don't block indefinitely with our
380            // buffer partially filled.
381            ret = poll(ufds, 2, 100);
382            if (ret <= 0) {
383                D("poll returned %d\n", ret);
384                // file has closed or we timed out
385                // set ret to 1 so we don't exit the outer loop
386                ret = 1;
387                break;
388            }
389
390            if (ufds[0].revents & POLLIN) {
391                ret = adb_read(fd, buffer + count, sizeof(buffer) - count);
392                D("read fd ret: %d, count: %d\n", ret, count);
393                if (ret > 0)
394                    count += ret;
395                else
396                    break;
397            }
398            if (ufds[1].revents & POLLIN) {
399                ret = adb_read(s, buffer2, sizeof(buffer2));
400                D("read s ret: %d\n", ret);
401                if (ret > 0)
402                    adb_write(fd, buffer2, ret);
403                else
404                    break;
405            }
406
407            if ((ufds[0].revents & POLLHUP) || (ufds[1].revents & POLLHUP)) {
408                // set flag to exit after flushing the buffer
409                ret = -1;
410                break;
411            }
412        }
413
414        D("writing: %d\n", count);
415        if (count > 0) {
416            adb_write(s, buffer, count);
417            count = 0;
418        }
419        if (ret <= 0)
420            break;
421    }
422
423    D("shell_service done\n");
424
425    adb_close(fd);
426    adb_close(s);
427}
428#endif // !ADB_HOST
429
430int service_to_fd(const char *name)
431{
432    int ret = -1;
433
434    if(!strncmp(name, "tcp:", 4)) {
435        int port = atoi(name + 4);
436        name = strchr(name + 4, ':');
437        if(name == 0) {
438            ret = socket_loopback_client(port, SOCK_STREAM);
439            if (ret >= 0)
440                disable_tcp_nagle(ret);
441        } else {
442#if ADB_HOST
443            adb_mutex_lock(&dns_lock);
444            ret = socket_network_client(name + 1, port, SOCK_STREAM);
445            adb_mutex_unlock(&dns_lock);
446#else
447            return -1;
448#endif
449        }
450#ifndef HAVE_WINSOCK   /* winsock doesn't implement unix domain sockets */
451    } else if(!strncmp(name, "local:", 6)) {
452        ret = socket_local_client(name + 6,
453                ANDROID_SOCKET_NAMESPACE_RESERVED, SOCK_STREAM);
454    } else if(!strncmp(name, "localreserved:", 14)) {
455        ret = socket_local_client(name + 14,
456                ANDROID_SOCKET_NAMESPACE_RESERVED, SOCK_STREAM);
457    } else if(!strncmp(name, "localabstract:", 14)) {
458        ret = socket_local_client(name + 14,
459                ANDROID_SOCKET_NAMESPACE_ABSTRACT, SOCK_STREAM);
460    } else if(!strncmp(name, "localfilesystem:", 16)) {
461        ret = socket_local_client(name + 16,
462                ANDROID_SOCKET_NAMESPACE_FILESYSTEM, SOCK_STREAM);
463#endif
464#if ADB_HOST
465    } else if(!strncmp("dns:", name, 4)){
466        char *n = strdup(name + 4);
467        if(n == 0) return -1;
468        ret = create_service_thread(dns_service, n);
469#else /* !ADB_HOST */
470    } else if(!strncmp("dev:", name, 4)) {
471        ret = unix_open(name + 4, O_RDWR);
472    } else if(!strncmp(name, "framebuffer:", 12)) {
473        ret = create_service_thread(framebuffer_service, 0);
474    } else if(recovery_mode && !strncmp(name, "recover:", 8)) {
475        ret = create_service_thread(recover_service, (void*) atoi(name + 8));
476    } else if (!strncmp(name, "jdwp:", 5)) {
477        ret = create_jdwp_connection_fd(atoi(name+5));
478    } else if (!strncmp(name, "log:", 4)) {
479        ret = create_service_thread(log_service, get_log_file_path(name + 4));
480    } else if(!HOST && !strncmp(name, "shell:", 6)) {
481        const char* args[2];
482        if(name[6]) {
483            args[0] = "-c";
484            args[1] = name + 6;
485        } else {
486            args[0] = "-";
487            args[1] = 0;
488        }
489        ret = create_service_thread(shell_service, (void *)args);
490    } else if(!strncmp(name, "sync:", 5)) {
491        ret = create_service_thread(file_sync_service, NULL);
492    } else if(!strncmp(name, "remount:", 8)) {
493        ret = create_service_thread(remount_service, NULL);
494    } else if(!strncmp(name, "reboot:", 7)) {
495        const char* arg = name + 7;
496        if (*name == 0)
497            arg = NULL;
498        ret = create_service_thread(reboot_service, (void *)arg);
499    } else if(!strncmp(name, "root:", 5)) {
500        ret = create_service_thread(restart_root_service, NULL);
501    } else if(!strncmp(name, "tcpip:", 6)) {
502        int port;
503        if (sscanf(name + 6, "%d", &port) == 0) {
504            port = 0;
505        }
506        ret = create_service_thread(restart_tcp_service, (void *)port);
507    } else if(!strncmp(name, "usb:", 4)) {
508        ret = create_service_thread(restart_usb_service, NULL);
509    } else if(!strncmp(name, "bugreport:", 10)) {
510        ret = create_service_thread(bugreport_service, NULL);
511#endif
512#if 0
513    } else if(!strncmp(name, "echo:", 5)){
514        ret = create_service_thread(echo_service, 0);
515#endif
516    }
517    if (ret >= 0) {
518        close_on_exec(ret);
519    }
520    return ret;
521}
522
523#if ADB_HOST
524struct state_info {
525    transport_type transport;
526    char* serial;
527    int state;
528};
529
530static void wait_for_state(int fd, void* cookie)
531{
532    struct state_info* sinfo = cookie;
533    char* err = "unknown error";
534
535    D("wait_for_state %d\n", sinfo->state);
536
537    atransport *t = acquire_one_transport(sinfo->state, sinfo->transport, sinfo->serial, &err);
538    if(t != 0) {
539        writex(fd, "OKAY", 4);
540    } else {
541        sendfailmsg(fd, err);
542    }
543
544    if (sinfo->serial)
545        free(sinfo->serial);
546    free(sinfo);
547    adb_close(fd);
548    D("wait_for_state is done\n");
549}
550#endif
551
552#if ADB_HOST
553asocket*  host_service_to_socket(const char*  name, const char *serial)
554{
555    if (!strcmp(name,"track-devices")) {
556        return create_device_tracker();
557    } else if (!strncmp(name, "wait-for-", strlen("wait-for-"))) {
558        struct state_info* sinfo = malloc(sizeof(struct state_info));
559
560        if (serial)
561            sinfo->serial = strdup(serial);
562        else
563            sinfo->serial = NULL;
564
565        name += strlen("wait-for-");
566
567        if (!strncmp(name, "local", strlen("local"))) {
568            sinfo->transport = kTransportLocal;
569            sinfo->state = CS_DEVICE;
570        } else if (!strncmp(name, "usb", strlen("usb"))) {
571            sinfo->transport = kTransportUsb;
572            sinfo->state = CS_DEVICE;
573        } else if (!strncmp(name, "any", strlen("any"))) {
574            sinfo->transport = kTransportAny;
575            sinfo->state = CS_DEVICE;
576        } else {
577            free(sinfo);
578            return NULL;
579        }
580
581        int fd = create_service_thread(wait_for_state, sinfo);
582        return create_local_socket(fd);
583    }
584    return NULL;
585}
586#endif /* ADB_HOST */
587