mirror of
https://github.com/linux-msm/diag.git
synced 2026-02-25 13:11:48 -08:00
365aa41212
Move common helpers related to "diag clients" into a "Diagnostics Monitor" implementation. This becomes necessary in order to support non-HDLC encoding between the DM and device transparently to the rest of the implementation. Signed-off-by: Bjorn Andersson <bjorn.andersson@linaro.org>
132 lines
3.5 KiB
C
132 lines
3.5 KiB
C
/*
|
|
* Copyright (c) 2016-2017, The Linux Foundation. All rights reserved.
|
|
* Copyright (c) 2016, Linaro Ltd.
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright notice,
|
|
* this list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
* and/or other materials provided with the distribution.
|
|
*
|
|
* 3. Neither the name of the copyright holder nor the names of its contributors
|
|
* may be used to endorse or promote products derived from this software without
|
|
* specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
|
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
|
|
#include <err.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <netdb.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
|
|
#include "diag.h"
|
|
#include "dm.h"
|
|
#include "hdlc.h"
|
|
#include "watch.h"
|
|
|
|
#define APPS_BUF_SIZE 16384
|
|
|
|
static int diag_sock_recv(int fd, void* data);
|
|
|
|
int diag_sock_connect(const char *hostname, unsigned short port)
|
|
{
|
|
struct diag_client *client;
|
|
struct sockaddr_in addr;
|
|
struct hostent *host;
|
|
int ret;
|
|
int fd;
|
|
|
|
fd = socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, 0);
|
|
if (fd < 0)
|
|
return -errno;
|
|
|
|
host = gethostbyname(hostname);
|
|
if (!host)
|
|
return -errno;
|
|
|
|
bzero(&addr, sizeof(addr));
|
|
addr.sin_family = AF_INET;
|
|
bcopy(host->h_addr, &addr.sin_addr.s_addr, host->h_length);
|
|
addr.sin_port = htons(port);
|
|
|
|
ret = connect(fd, (const struct sockaddr *)&addr, sizeof(addr));
|
|
if (ret < 0)
|
|
return -errno;
|
|
|
|
ret = fcntl(fd, F_SETFL, O_NONBLOCK);
|
|
if (ret < 0)
|
|
return -errno;
|
|
|
|
printf("Connected to %s:%d\n", hostname, port);
|
|
|
|
client = calloc(1, sizeof(*client));
|
|
if (!client)
|
|
err(1, "failed to allocate client context\n");
|
|
|
|
client->fd = fd;
|
|
client->name = "DIAG CLIENT";
|
|
|
|
watch_add_readfd(client->fd, diag_sock_recv, client);
|
|
watch_add_writeq(client->fd, &client->outq);
|
|
|
|
dm_add(client);
|
|
|
|
return fd;
|
|
}
|
|
|
|
static int diag_sock_recv(int fd, void* data)
|
|
{
|
|
struct diag_client *client = (struct diag_client *)data;
|
|
size_t msglen;
|
|
ssize_t n;
|
|
void *msg;
|
|
int ret = 0;
|
|
|
|
for (;;) {
|
|
n = circ_read(fd, &client->recv_buf);
|
|
if (n < 0 && errno == EAGAIN) {
|
|
break;
|
|
} else if (n < 0) {
|
|
ret = -errno;
|
|
warn("Failed to read from %s\n", client->name);
|
|
break;
|
|
}
|
|
|
|
for (;;) {
|
|
msg = hdlc_decode_one(&client->recv_decoder,
|
|
&client->recv_buf,
|
|
&msglen);
|
|
if (!msg)
|
|
break;
|
|
|
|
diag_client_handle_command(client, msg, msglen);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|