303 lines
8.3 KiB
C
303 lines
8.3 KiB
C
/****************************************************************************
|
|
* apps/netutils/usrsock_rpmsg/usrsock_rpmsg_client.c
|
|
* usrsock rpmsg client
|
|
*
|
|
* Copyright (C) 2018 Pinecone Inc. All rights reserved.
|
|
* Author: Jianli Dong <dongjianli@pinecone.net>
|
|
*
|
|
* 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 NuttX 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 OWNER 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.
|
|
*
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* Included Files
|
|
****************************************************************************/
|
|
|
|
#include <nuttx/config.h>
|
|
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
#include <poll.h>
|
|
#include <semaphore.h>
|
|
#include <signal.h>
|
|
|
|
#include <nuttx/fs/fs.h>
|
|
#include <nuttx/net/dns.h>
|
|
#include <nuttx/rptun/openamp.h>
|
|
|
|
#include "usrsock_rpmsg.h"
|
|
|
|
/****************************************************************************
|
|
* Private Types
|
|
****************************************************************************/
|
|
|
|
struct usrsock_rpmsg_s
|
|
{
|
|
struct rpmsg_endpoint ept;
|
|
const char *cpuname;
|
|
pid_t pid;
|
|
sem_t sem;
|
|
struct file file;
|
|
};
|
|
|
|
/****************************************************************************
|
|
* Private Function Prototypes
|
|
****************************************************************************/
|
|
|
|
static int usrsock_rpmsg_dns_handler(struct rpmsg_endpoint *ept, void *data,
|
|
size_t len, uint32_t src, void *priv);
|
|
static int usrsock_rpmsg_default_handler(struct rpmsg_endpoint *ept,
|
|
void *data, size_t len,
|
|
uint32_t src, void *priv_);
|
|
|
|
static void usrsock_rpmsg_device_created(struct rpmsg_device *rdev,
|
|
void *priv_);
|
|
static void usrsock_rpmsg_device_destroy(struct rpmsg_device *rdev,
|
|
void *priv_);
|
|
static int usrsock_rpmsg_ept_cb(struct rpmsg_endpoint *ept, void *data,
|
|
size_t len, uint32_t src, void *priv);
|
|
|
|
/****************************************************************************
|
|
* Private Functions
|
|
****************************************************************************/
|
|
|
|
static void usrsock_rpmsg_device_created(struct rpmsg_device *rdev,
|
|
void *priv_)
|
|
{
|
|
struct usrsock_rpmsg_s *priv = priv_;
|
|
int ret;
|
|
|
|
if (!strcmp(priv->cpuname, rpmsg_get_cpuname(rdev)))
|
|
{
|
|
priv->ept.priv = priv;
|
|
|
|
ret = rpmsg_create_ept(&priv->ept, rdev, USRSOCK_RPMSG_EPT_NAME,
|
|
RPMSG_ADDR_ANY, RPMSG_ADDR_ANY,
|
|
usrsock_rpmsg_ept_cb, NULL);
|
|
if (ret == 0)
|
|
{
|
|
sem_post(&priv->sem);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void usrsock_rpmsg_device_destroy(struct rpmsg_device *rdev,
|
|
void *priv_)
|
|
{
|
|
struct usrsock_rpmsg_s *priv = priv_;
|
|
|
|
if (!strcmp(priv->cpuname, rpmsg_get_cpuname(rdev)))
|
|
{
|
|
rpmsg_destroy_ept(&priv->ept);
|
|
kill(priv->pid, SIGUSR1);
|
|
}
|
|
}
|
|
|
|
static int usrsock_rpmsg_dns_handler(struct rpmsg_endpoint *ept, void *data,
|
|
size_t len, uint32_t src, void *priv)
|
|
{
|
|
int ret = OK;
|
|
#ifdef CONFIG_NETDB_DNSCLIENT
|
|
struct usrsock_rpmsg_dns_event_s *dns = data;
|
|
|
|
ret = dns_add_nameserver((struct sockaddr *)(dns + 1), dns->addrlen);
|
|
#endif
|
|
|
|
return ret;
|
|
}
|
|
|
|
static int usrsock_rpmsg_default_handler(struct rpmsg_endpoint *ept,
|
|
void *data, size_t len,
|
|
uint32_t src, void *priv_)
|
|
{
|
|
struct usrsock_rpmsg_s *priv = priv_;
|
|
|
|
while (len > 0)
|
|
{
|
|
ssize_t ret = file_write(&priv->file, data, len);
|
|
if (ret < 0)
|
|
{
|
|
return ret;
|
|
}
|
|
|
|
data += ret;
|
|
len -= ret;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int usrsock_rpmsg_ept_cb(struct rpmsg_endpoint *ept, void *data,
|
|
size_t len, uint32_t src, void *priv)
|
|
{
|
|
struct usrsock_message_common_s *common = data;
|
|
int ret;
|
|
|
|
switch (common->msgid)
|
|
{
|
|
case USRSOCK_RPMSG_DNS_EVENT:
|
|
ret = usrsock_rpmsg_dns_handler(ept, data, len, src, priv);
|
|
break;
|
|
default:
|
|
ret = usrsock_rpmsg_default_handler(ept, data, len, src, priv);
|
|
break;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/****************************************************************************
|
|
* Public Functions
|
|
****************************************************************************/
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
struct usrsock_rpmsg_s priv =
|
|
{
|
|
};
|
|
|
|
int ret;
|
|
|
|
if (argv[1] == NULL)
|
|
{
|
|
return -EINVAL;
|
|
}
|
|
|
|
priv.cpuname = argv[1];
|
|
priv.pid = getpid();
|
|
|
|
sem_init(&priv.sem, 0, 0);
|
|
sem_setprotocol(&priv.sem, SEM_PRIO_NONE);
|
|
|
|
sigrelse(SIGUSR1);
|
|
|
|
ret = rpmsg_register_callback(&priv,
|
|
usrsock_rpmsg_device_created,
|
|
usrsock_rpmsg_device_destroy,
|
|
NULL);
|
|
if (ret < 0)
|
|
{
|
|
goto destroy_sem;
|
|
}
|
|
|
|
while (1)
|
|
{
|
|
/* Wait until the rpmsg channel is ready */
|
|
|
|
do
|
|
{
|
|
ret = sem_wait(&priv.sem);
|
|
if (ret < 0)
|
|
{
|
|
ret = -errno;
|
|
}
|
|
}
|
|
while (ret == -EINTR);
|
|
|
|
if (ret < 0)
|
|
{
|
|
goto unregister_callback;
|
|
}
|
|
|
|
/* Open the kernel channel */
|
|
|
|
ret = file_open(&priv.file, "/dev/usrsock", O_RDWR);
|
|
if (ret < 0)
|
|
{
|
|
ret = -errno;
|
|
goto destroy_ept;
|
|
}
|
|
|
|
/* Forward the packet from kernel to remote */
|
|
|
|
while (1)
|
|
{
|
|
struct pollfd pfd;
|
|
void *buf;
|
|
uint32_t len;
|
|
|
|
/* Wait the packet ready */
|
|
|
|
pfd.ptr = &priv.file;
|
|
pfd.events = POLLIN | POLLFILE;
|
|
ret = poll(&pfd, 1, -1);
|
|
if (ret < 0)
|
|
{
|
|
ret = -errno;
|
|
break;
|
|
}
|
|
|
|
/* Read the packet from kernel */
|
|
|
|
buf = rpmsg_get_tx_payload_buffer(&priv.ept, &len, true);
|
|
if (!buf)
|
|
{
|
|
ret = -ENOMEM;
|
|
break;
|
|
}
|
|
|
|
ret = file_read(&priv.file, buf, len);
|
|
if (ret < 0)
|
|
{
|
|
break;
|
|
}
|
|
|
|
/* Send the packet to remote */
|
|
|
|
ret = rpmsg_send_nocopy(&priv.ept, buf, ret);
|
|
if (ret < 0)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* Reclaim the resource */
|
|
|
|
file_close(&priv.file);
|
|
|
|
if (is_rpmsg_ept_ready(&priv.ept))
|
|
{
|
|
goto destroy_ept;
|
|
}
|
|
|
|
/* The remote side crash, loop to wait it restore */
|
|
}
|
|
|
|
destroy_ept:
|
|
rpmsg_destroy_ept(&priv.ept);
|
|
|
|
unregister_callback:
|
|
rpmsg_unregister_callback(&priv,
|
|
usrsock_rpmsg_device_created,
|
|
usrsock_rpmsg_device_destroy,
|
|
NULL);
|
|
destroy_sem:
|
|
sem_destroy(&priv.sem);
|
|
return ret;
|
|
}
|