Home | History | Annotate | Download | only in fastbootd 
      1 /*
      2  * Copyright (c) 2009-2013, Google Inc.
      3  * All rights reserved.
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions
      7  * are met:
      8  *  * Redistributions of source code must retain the above copyright
      9  *    notice, this list of conditions and the following disclaimer.
     10  *  * Redistributions in binary form must reproduce the above copyright
     11  *    notice, this list of conditions and the following disclaimer in
     12  *    the documentation and/or other materials provided with the
     13  *    distribution.
     14  *  * Neither the name of Google, Inc. nor the names of its contributors
     15  *    may be used to endorse or promote products derived from this
     16  *    software without specific prior written permission.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
     21  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
     22  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
     23  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
     24  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
     25  * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
     26  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
     27  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
     28  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29  * SUCH DAMAGE.
     30  */
     31 
     32 #include <stdio.h>
     33 #include <cutils/sockets.h>
     34 #include <poll.h>
     35 #include <unistd.h>
     36 
     37 #include "utils.h"
     38 
     39 #define BUFFER_SIZE 256
     40 
     41 #define STDIN_FD 0
     42 #define STDOUT_FD 1
     43 #define STDERR_FD 2
     44 
     45 void run_socket_client() {
     46     int fd;
     47     char buffer[BUFFER_SIZE];
     48     int n;
     49     struct pollfd fds[2];
     50 
     51     fd = socket_local_client("fastbootd",
     52                          ANDROID_SOCKET_NAMESPACE_RESERVED,
     53                          SOCK_STREAM);
     54 
     55     if (fd < 0) {
     56         fprintf(stderr, "ERROR: Unable to open fastbootd socket\n");
     57         return;
     58     }
     59 
     60     fds[0].fd = STDIN_FD;
     61     fds[0].events = POLLIN;
     62     fds[1].fd = fd;
     63     fds[1].events = POLLIN;
     64 
     65     while(true) {
     66         if (poll(fds, 2, -1) <= 0) {
     67             fprintf(stderr, "ERROR: socket error");
     68             return;
     69         }
     70 
     71         if (fds[0].revents & POLLIN) {
     72             if ((n = read(STDIN_FD, buffer, BUFFER_SIZE)) < 0) {
     73                 goto error;
     74             }
     75 
     76             if (bulk_write(fd, buffer, n) < 0) {
     77                 goto error;
     78             }
     79         }
     80 
     81         if (fds[1].revents & POLLIN) {
     82             if ((n = read(fd, buffer, BUFFER_SIZE)) < 0) {
     83                 goto error;
     84             }
     85 
     86             if (bulk_write(STDOUT_FD, buffer, n) < 0) {
     87                 goto error;
     88             }
     89         }
     90     }
     91 
     92 error:
     93     fprintf(stderr, "Transport error\n");
     94 }
     95