socket_client.c revision b015b16c977d2cdf3fe1028c08270833ff428d07 
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
45void 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
92error:
93    fprintf(stderr, "Transport error\n");
94}
95