Home | History | Annotate | Download | only in netd
      1 /*
      2  * Copyright (C) 2008 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 <stdio.h>
     18 #include <stdlib.h>
     19 #include <unistd.h>
     20 #include <string.h>
     21 #include <signal.h>
     22 #include <errno.h>
     23 #include <fcntl.h>
     24 
     25 #include <sys/socket.h>
     26 #include <sys/select.h>
     27 #include <sys/time.h>
     28 #include <sys/types.h>
     29 #include <sys/un.h>
     30 
     31 #include <cutils/sockets.h>
     32 #include <private/android_filesystem_config.h>
     33 
     34 static void usage(char *progname);
     35 static int do_monitor(int sock, int stop_after_cmd);
     36 static int do_cmd(int sock, int argc, char **argv);
     37 
     38 int main(int argc, char **argv) {
     39     int sock;
     40     int cmdOffset = 0;
     41 
     42     if (argc < 2)
     43         usage(argv[0]);
     44 
     45     // try interpreting the first arg as the socket name - if it fails go back to netd
     46 
     47     if ((sock = socket_local_client(argv[1],
     48                                      ANDROID_SOCKET_NAMESPACE_RESERVED,
     49                                      SOCK_STREAM)) < 0) {
     50         if ((sock = socket_local_client("netd",
     51                                          ANDROID_SOCKET_NAMESPACE_RESERVED,
     52                                          SOCK_STREAM)) < 0) {
     53             fprintf(stderr, "Error connecting (%s)\n", strerror(errno));
     54             exit(4);
     55         }
     56     } else {
     57         if (argc < 3) usage(argv[0]);
     58         printf("Using alt socket %s\n", argv[1]);
     59         cmdOffset = 1;
     60     }
     61 
     62     if (!strcmp(argv[1+cmdOffset], "monitor"))
     63         exit(do_monitor(sock, 0));
     64     exit(do_cmd(sock, argc-cmdOffset, &(argv[cmdOffset])));
     65 }
     66 
     67 static int do_cmd(int sock, int argc, char **argv) {
     68     char *final_cmd = strdup("0 ");
     69     if (final_cmd == NULL) {
     70         perror("strdup");
     71         return errno;
     72     }
     73 
     74     int i;
     75 
     76     for (i = 1; i < argc; i++) {
     77         if (index(argv[i], '"')) {
     78             perror("argument with embedded quotes not allowed");
     79             free(final_cmd);
     80             return 1;
     81         }
     82         bool needs_quoting = index(argv[i], ' ');
     83         const char *format = needs_quoting ? "%s\"%s\"%s" : "%s%s%s";
     84         char *cmp;
     85 
     86         if (asprintf(&cmp, format, final_cmd, argv[i],
     87                      (i == (argc -1)) ? "" : " ") < 0) {
     88             perror("malloc");
     89             free(final_cmd);
     90             return errno;
     91         }
     92         free(final_cmd);
     93         final_cmd = cmp;
     94     }
     95 
     96     if (write(sock, final_cmd, strlen(final_cmd) + 1) < 0) {
     97         perror("write");
     98         free(final_cmd);
     99         return errno;
    100     }
    101     free(final_cmd);
    102 
    103     return do_monitor(sock, 1);
    104 }
    105 
    106 static int do_monitor(int sock, int stop_after_cmd) {
    107     char *buffer = malloc(4096);
    108 
    109     if (!stop_after_cmd)
    110         printf("[Connected to Netd]\n");
    111 
    112     while(1) {
    113         fd_set read_fds;
    114         struct timeval to;
    115         int rc = 0;
    116 
    117         to.tv_sec = 10;
    118         to.tv_usec = 0;
    119 
    120         FD_ZERO(&read_fds);
    121         FD_SET(sock, &read_fds);
    122 
    123         if ((rc = select(sock +1, &read_fds, NULL, NULL, &to)) < 0) {
    124             fprintf(stderr, "Error in select (%s)\n", strerror(errno));
    125             free(buffer);
    126             return errno;
    127         } else if (!rc) {
    128             continue;
    129             fprintf(stderr, "[TIMEOUT]\n");
    130             return ETIMEDOUT;
    131         } else if (FD_ISSET(sock, &read_fds)) {
    132             memset(buffer, 0, 4096);
    133             if ((rc = read(sock, buffer, 4096)) <= 0) {
    134                 if (rc == 0)
    135                     fprintf(stderr, "Lost connection to Netd - did it crash?\n");
    136                 else
    137                     fprintf(stderr, "Error reading data (%s)\n", strerror(errno));
    138                 free(buffer);
    139                 if (rc == 0)
    140                     return ECONNRESET;
    141                 return errno;
    142             }
    143 
    144             int offset = 0;
    145             int i = 0;
    146 
    147             for (i = 0; i < rc; i++) {
    148                 if (buffer[i] == '\0') {
    149                     int code;
    150                     char tmp[4];
    151 
    152                     strncpy(tmp, buffer + offset, 3);
    153                     tmp[3] = '\0';
    154                     code = atoi(tmp);
    155 
    156                     printf("%s\n", buffer + offset);
    157                     if (stop_after_cmd) {
    158                         if (code >= 200 && code < 600)
    159                             return 0;
    160                     }
    161                     offset = i + 1;
    162                 }
    163             }
    164         }
    165     }
    166     free(buffer);
    167     return 0;
    168 }
    169 
    170 static void usage(char *progname) {
    171     fprintf(stderr, "Usage: %s [sockname] <monitor>|<cmd> [arg1] [arg2...]\n", progname);
    172     exit(1);
    173 }
    174