408 lines
10 KiB
C
408 lines
10 KiB
C
/****************************************************************************
|
|
* net/local/local_netpoll.c
|
|
*
|
|
* Copyright (C) 2015 Gregory Nutt. All rights reserved.
|
|
* Author: Gregory Nutt <gnutt@nuttx.org>
|
|
*
|
|
* 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 <unistd.h>
|
|
#include <assert.h>
|
|
#include <errno.h>
|
|
#include <debug.h>
|
|
|
|
#include <nuttx/kmalloc.h>
|
|
#include <nuttx/net/net.h>
|
|
#include <nuttx/fs/fs.h>
|
|
|
|
#include "socket/socket.h"
|
|
#include "local/local.h"
|
|
|
|
#ifdef HAVE_LOCAL_POLL
|
|
|
|
/****************************************************************************
|
|
* Function: local_accept_pollsetup
|
|
****************************************************************************/
|
|
|
|
#ifdef CONFIG_NET_LOCAL_STREAM
|
|
static int local_accept_pollsetup(FAR struct local_conn_s *conn,
|
|
FAR struct pollfd *fds,
|
|
bool setup)
|
|
{
|
|
net_lock_t state;
|
|
pollevent_t eventset;
|
|
int ret = OK;
|
|
int i;
|
|
|
|
state = net_lock();
|
|
if (setup)
|
|
{
|
|
/* This is a request to set up the poll. Find an available
|
|
* slot for the poll structure reference
|
|
*/
|
|
|
|
for (i = 0; i < LOCAL_ACCEPT_NPOLLWAITERS; i++)
|
|
{
|
|
/* Find an available slot */
|
|
|
|
if (!conn->lc_accept_fds[i])
|
|
{
|
|
/* Bind the poll structure and this slot */
|
|
|
|
conn->lc_accept_fds[i] = fds;
|
|
fds->priv = &conn->lc_accept_fds[i];
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (i >= LOCAL_ACCEPT_NPOLLWAITERS)
|
|
{
|
|
fds->priv = NULL;
|
|
ret = -EBUSY;
|
|
goto errout;
|
|
}
|
|
|
|
eventset = 0;
|
|
if (dq_peek(&conn->u.server.lc_waiters) != NULL)
|
|
{
|
|
eventset |= POLLIN;
|
|
}
|
|
|
|
if (eventset)
|
|
{
|
|
local_accept_pollnotify(conn, eventset);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* This is a request to tear down the poll. */
|
|
|
|
struct pollfd **slot = (struct pollfd **)fds->priv;
|
|
|
|
if (!slot)
|
|
{
|
|
ret = -EIO;
|
|
goto errout;
|
|
}
|
|
|
|
/* Remove all memory of the poll setup */
|
|
|
|
*slot = NULL;
|
|
fds->priv = NULL;
|
|
}
|
|
|
|
errout:
|
|
net_unlock(state);
|
|
return ret;
|
|
}
|
|
#endif
|
|
|
|
/****************************************************************************
|
|
* Public Functions
|
|
****************************************************************************/
|
|
|
|
/****************************************************************************
|
|
* Name: local_accept_pollnotify
|
|
****************************************************************************/
|
|
|
|
void local_accept_pollnotify(FAR struct local_conn_s *conn,
|
|
pollevent_t eventset)
|
|
{
|
|
#ifdef CONFIG_NET_LOCAL_STREAM
|
|
int i;
|
|
|
|
for (i = 0; i < LOCAL_ACCEPT_NPOLLWAITERS; i++)
|
|
{
|
|
struct pollfd *fds = conn->lc_accept_fds[i];
|
|
if (fds)
|
|
{
|
|
fds->revents |= (fds->events & eventset);
|
|
if (fds->revents != 0)
|
|
{
|
|
ndbg("Report events: %02x\n", fds->revents);
|
|
sem_post(fds->sem);
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
/****************************************************************************
|
|
* Function: local_pollsetup
|
|
*
|
|
* Description:
|
|
* Setup to monitor events on one Unix domain socket
|
|
*
|
|
* Input Parameters:
|
|
* psock - The Unix domain socket of interest
|
|
* fds - The structure describing the events to be monitored, OR NULL if
|
|
* this is a request to stop monitoring events.
|
|
*
|
|
* Returned Value:
|
|
* 0: Success; Negated errno on failure
|
|
*
|
|
****************************************************************************/
|
|
|
|
int local_pollsetup(FAR struct socket *psock, FAR struct pollfd *fds)
|
|
{
|
|
FAR struct local_conn_s *conn;
|
|
int ret = -ENOSYS;
|
|
|
|
conn = (FAR struct local_conn_s *)psock->s_conn;
|
|
|
|
if (conn->lc_proto == SOCK_DGRAM)
|
|
{
|
|
return ret;
|
|
}
|
|
|
|
#ifdef CONFIG_NET_LOCAL_STREAM
|
|
if (conn->lc_state == LOCAL_STATE_LISTENING &&
|
|
conn->lc_type == LOCAL_TYPE_PATHNAME)
|
|
{
|
|
return local_accept_pollsetup(conn, fds, true);
|
|
}
|
|
|
|
if (conn->lc_state == LOCAL_STATE_DISCONNECTED)
|
|
{
|
|
fds->priv = NULL;
|
|
goto pollerr;
|
|
}
|
|
|
|
switch (fds->events & (POLLIN | POLLOUT))
|
|
{
|
|
case (POLLIN | POLLOUT):
|
|
{
|
|
FAR struct pollfd *shadowfds;
|
|
|
|
/* Poll wants to check state for both input and output. */
|
|
|
|
if (conn->lc_infd < 0 || conn->lc_outfd < 0)
|
|
{
|
|
fds->priv = NULL;
|
|
goto pollerr;
|
|
}
|
|
|
|
/* Allocate shadow pollfds. */
|
|
|
|
shadowfds = kmm_zalloc(2 * sizeof(struct pollfd));
|
|
if (!shadowfds)
|
|
{
|
|
return -ENOMEM;
|
|
}
|
|
|
|
shadowfds[0].fd = conn->lc_infd;
|
|
shadowfds[0].sem = fds->sem;
|
|
shadowfds[0].events = fds->events & ~POLLOUT;
|
|
|
|
shadowfds[1].fd = conn->lc_outfd;
|
|
shadowfds[1].sem = fds->sem;
|
|
shadowfds[1].events = fds->events & ~POLLIN;
|
|
|
|
/* Setup poll for both shadow pollfds. */
|
|
|
|
ret = file_poll(conn->lc_infd, &shadowfds[0], true);
|
|
if (ret >= 0)
|
|
{
|
|
ret = file_poll(conn->lc_outfd, &shadowfds[1], true);
|
|
if (ret < 0)
|
|
{
|
|
(void)file_poll(conn->lc_infd, &shadowfds[0], false);
|
|
}
|
|
}
|
|
|
|
if (ret < 0)
|
|
{
|
|
kmm_free(shadowfds);
|
|
fds->priv = NULL;
|
|
goto pollerr;
|
|
}
|
|
else
|
|
{
|
|
fds->priv = shadowfds;
|
|
ret = OK;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case POLLIN:
|
|
{
|
|
/* Poll wants to check state for input only. */
|
|
|
|
if (conn->lc_infd < 0)
|
|
{
|
|
fds->priv = NULL;
|
|
goto pollerr;
|
|
}
|
|
|
|
ret = file_poll(conn->lc_infd, fds, true);
|
|
}
|
|
break;
|
|
|
|
case POLLOUT:
|
|
{
|
|
/* Poll wants to check state for output only. */
|
|
|
|
if (conn->lc_outfd < 0)
|
|
{
|
|
fds->priv = NULL;
|
|
goto pollerr;
|
|
}
|
|
|
|
ret = file_poll(conn->lc_outfd, fds, true);
|
|
}
|
|
break;
|
|
|
|
default:
|
|
ret = OK;
|
|
break;
|
|
}
|
|
#endif
|
|
|
|
return ret;
|
|
|
|
pollerr:
|
|
fds->revents |= POLLERR;
|
|
sem_post(fds->sem);
|
|
return OK;
|
|
}
|
|
|
|
/****************************************************************************
|
|
* Function: local_pollteardown
|
|
*
|
|
* Description:
|
|
* Teardown monitoring of events on a Unix domain socket
|
|
*
|
|
* Input Parameters:
|
|
* psock - The Unix domain socket of interest
|
|
* fds - The structure describing the events to be monitored, OR NULL if
|
|
* this is a request to stop monitoring events.
|
|
*
|
|
* Returned Value:
|
|
* 0: Success; Negated errno on failure
|
|
*
|
|
****************************************************************************/
|
|
|
|
int local_pollteardown(FAR struct socket *psock, FAR struct pollfd *fds)
|
|
{
|
|
FAR struct local_conn_s *conn;
|
|
int status = OK;
|
|
int ret = -ENOSYS;
|
|
|
|
conn = (FAR struct local_conn_s *)psock->s_conn;
|
|
|
|
if (conn->lc_proto == SOCK_DGRAM)
|
|
{
|
|
return ret;
|
|
}
|
|
|
|
#ifdef CONFIG_NET_LOCAL_STREAM
|
|
if (conn->lc_state == LOCAL_STATE_LISTENING &&
|
|
conn->lc_type == LOCAL_TYPE_PATHNAME)
|
|
{
|
|
return local_accept_pollsetup(conn, fds, false);
|
|
}
|
|
|
|
if (conn->lc_state == LOCAL_STATE_DISCONNECTED)
|
|
{
|
|
return OK;
|
|
}
|
|
|
|
switch (fds->events & (POLLIN | POLLOUT))
|
|
{
|
|
case (POLLIN | POLLOUT):
|
|
{
|
|
FAR struct pollfd *shadowfds = fds->priv;
|
|
|
|
if (shadowfds == NULL)
|
|
{
|
|
return OK;
|
|
}
|
|
|
|
DEBUGASSERT(shadowfds[0].fd == conn->lc_infd);
|
|
DEBUGASSERT(shadowfds[1].fd == conn->lc_outfd);
|
|
|
|
/* Teardown for both shadow pollfds. */
|
|
|
|
ret = file_poll(conn->lc_infd, &shadowfds[0], false);
|
|
if (ret < 0)
|
|
{
|
|
status = ret;
|
|
}
|
|
|
|
ret = file_poll(conn->lc_outfd, &shadowfds[1], false);
|
|
if (ret < 0)
|
|
{
|
|
status = ret;
|
|
}
|
|
|
|
fds->revents |= shadowfds[0].revents | shadowfds[1].revents;
|
|
fds->priv = NULL;
|
|
kmm_free(shadowfds);
|
|
}
|
|
break;
|
|
|
|
case POLLIN:
|
|
{
|
|
if (fds->priv == NULL)
|
|
{
|
|
return OK;
|
|
}
|
|
|
|
status = file_poll(conn->lc_infd, fds, false);
|
|
}
|
|
break;
|
|
|
|
case POLLOUT:
|
|
{
|
|
if (fds->priv == NULL)
|
|
{
|
|
return OK;
|
|
}
|
|
|
|
status = file_poll(conn->lc_outfd, fds, false);
|
|
}
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
#endif
|
|
|
|
return status;
|
|
}
|
|
|
|
#endif /* HAVE_LOCAL_POLL */
|