• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 
run_socket_client()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