sergiotarxz/nuttx-apps
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
2f7480d11b
nuttx-apps/system/usrsock_rpmsg/usrsock_rpmsg_client.c

302 lines
8.3 KiB
C
Raw Blame History

/****************************************************************************
* apps/system/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;
size_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;
}
Reference in New Issue View Git Blame Copy Permalink