/*
* Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
* Copyright (c) 1995-1999 by Internet Software Consortium
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
* OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
/* ev_connects.c - implement asynch connect/accept for the eventlib
* vix 16sep96 [initial]
*/
#if !defined(LINT) && !defined(CODECENTER)
static const char rcsid[] = "$Id: ev_connects.c,v 1.8 2006/03/09 23:57:56 marka Exp $";
#endif
/* Import. */
#include "port_before.h"
#include "fd_setsize.h"
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <unistd.h>
#include <isc/eventlib.h>
#include <isc/assertions.h>
#include "eventlib_p.h"
#include "port_after.h"
/* Macros. */
#define GETXXXNAME(f, s, sa, len) ( \
(f((s), (&sa), (&len)) >= 0) ? 0 : \
(errno != EAFNOSUPPORT && errno != EOPNOTSUPP) ? -1 : ( \
memset(&(sa), 0, sizeof (sa)), \
(len) = sizeof (sa), \
(sa).sa_family = AF_UNIX, \
0 \
) \
)
/* Forward. */
static void listener(evContext ctx, void *uap, int fd, int evmask);
static void connector(evContext ctx, void *uap, int fd, int evmask);
/* Public. */
int
evListen(evContext opaqueCtx, int fd, int maxconn,
evConnFunc func, void *uap, evConnID *id)
{
evContext_p *ctx = opaqueCtx.opaque;
evConn *new;
int mode;
OKNEW(new);
new->flags = EV_CONN_LISTEN;
OKFREE(mode = fcntl(fd, F_GETFL, NULL), new); /*%< side effect: validate fd. */
/*
* Remember the nonblocking status. We assume that either evSelectFD
* has not been done to this fd, or that if it has then the caller
* will evCancelConn before they evDeselectFD. If our assumptions
* are not met, then we might restore the old nonblocking status
* incorrectly.
*/
if ((mode & PORT_NONBLOCK) == 0) {
#ifdef USE_FIONBIO_IOCTL
int on = 1;
OKFREE(ioctl(fd, FIONBIO, (char *)&on), new);
#else
OKFREE(fcntl(fd, F_SETFL, mode | PORT_NONBLOCK), new);
#endif
new->flags |= EV_CONN_BLOCK;
}
OKFREE(listen(fd, maxconn), new);
if (evSelectFD(opaqueCtx, fd, EV_READ, listener, new, &new->file) < 0){
int save = errno;
FREE(new);
errno = save;
return (-1);
}
new->flags |= EV_CONN_SELECTED;
new->func = func;
new->uap = uap;
new->fd = fd;
if (ctx->conns != NULL)
ctx->conns->prev = new;
new->prev = NULL;
new->next = ctx->conns;
ctx->conns = new;
if (id)
id->opaque = new;
return (0);
}
int
evConnect(evContext opaqueCtx, int fd, const void *ra, int ralen,
evConnFunc func, void *uap, evConnID *id)
{
evContext_p *ctx = opaqueCtx.opaque;
evConn *new;
OKNEW(new);
new->flags = 0;
/* Do the select() first to get the socket into nonblocking mode. */
if (evSelectFD(opaqueCtx, fd, EV_MASK_ALL,
connector, new, &new->file) < 0) {
int save = errno;
FREE(new);
errno = save;
return (-1);
}
new->flags |= EV_CONN_SELECTED;
if (connect(fd, ra, ralen) < 0 &&
errno != EWOULDBLOCK &&
errno != EAGAIN &&
errno != EINPROGRESS) {
int save = errno;
(void) evDeselectFD(opaqueCtx, new->file);
FREE(new);
errno = save;
return (-1);
}
/* No error, or EWOULDBLOCK. select() tells when it's ready. */
new->func = func;
new->uap = uap;
new->fd = fd;
if (ctx->conns != NULL)
ctx->conns->prev = new;
new->prev = NULL;
new->next = ctx->conns;
ctx->conns = new;
if (id)
id->opaque = new;
return (0);
}
int
evCancelConn(evContext opaqueCtx, evConnID id) {
evContext_p *ctx = opaqueCtx.opaque;
evConn *this = id.opaque;
evAccept *acc, *nxtacc;
int mode;
if ((this->flags & EV_CONN_SELECTED) != 0)
(void) evDeselectFD(opaqueCtx, this->file);
if ((this->flags & EV_CONN_BLOCK) != 0) {
mode = fcntl(this->fd, F_GETFL, NULL);
if (mode == -1) {
if (errno != EBADF)
return (-1);
} else {
#ifdef USE_FIONBIO_IOCTL
int off = 0;
OK(ioctl(this->fd, FIONBIO, (char *)&off));
#else
OK(fcntl(this->fd, F_SETFL, mode & ~PORT_NONBLOCK));
#endif
}
}
/* Unlink from ctx->conns. */
if (this->prev != NULL)
this->prev->next = this->next;
else
ctx->conns = this->next;
if (this->next != NULL)
this->next->prev = this->prev;
/*
* Remove `this' from the ctx->accepts list (zero or more times).
*/
for (acc = HEAD(ctx->accepts), nxtacc = NULL;
acc != NULL;
acc = nxtacc)
{
nxtacc = NEXT(acc, link);
if (acc->conn == this) {
UNLINK(ctx->accepts, acc, link);
close(acc->fd);
FREE(acc);
}
}
/* Wrap up and get out. */
FREE(this);
return (0);
}
int evHold(evContext opaqueCtx, evConnID id) {
evConn *this = id.opaque;
if ((this->flags & EV_CONN_LISTEN) == 0) {
errno = EINVAL;
return (-1);
}
if ((this->flags & EV_CONN_SELECTED) == 0)
return (0);
this->flags &= ~EV_CONN_SELECTED;
return (evDeselectFD(opaqueCtx, this->file));
}
int evUnhold(evContext opaqueCtx, evConnID id) {
evConn *this = id.opaque;
int ret;
if ((this->flags & EV_CONN_LISTEN) == 0) {
errno = EINVAL;
return (-1);
}
if ((this->flags & EV_CONN_SELECTED) != 0)
return (0);
ret = evSelectFD(opaqueCtx, this->fd, EV_READ, listener, this,
&this->file);
if (ret == 0)
this->flags |= EV_CONN_SELECTED;
return (ret);
}
int
evTryAccept(evContext opaqueCtx, evConnID id, int *sys_errno) {
evContext_p *ctx = opaqueCtx.opaque;
evConn *conn = id.opaque;
evAccept *new;
if ((conn->flags & EV_CONN_LISTEN) == 0) {
errno = EINVAL;
return (-1);
}
OKNEW(new);
new->conn = conn;
new->ralen = sizeof new->ra;
new->fd = accept(conn->fd, &new->ra.sa, &new->ralen);
if (new->fd > ctx->highestFD) {
close(new->fd);
new->fd = -1;
new->ioErrno = ENOTSOCK;
}
if (new->fd >= 0) {
new->lalen = sizeof new->la;
if (GETXXXNAME(getsockname, new->fd, new->la.sa, new->lalen) < 0) {
new->ioErrno = errno;
(void) close(new->fd);
new->fd = -1;
} else
new->ioErrno = 0;
} else {
new->ioErrno = errno;
if (errno == EAGAIN || errno == EWOULDBLOCK) {
FREE(new);
return (-1);
}
}
INIT_LINK(new, link);
APPEND(ctx->accepts, new, link);
*sys_errno = new->ioErrno;
return (0);
}
/* Private. */
static void
listener(evContext opaqueCtx, void *uap, int fd, int evmask) {
evContext_p *ctx = opaqueCtx.opaque;
evConn *conn = uap;
union {
struct sockaddr sa;
struct sockaddr_in in;
#ifndef NO_SOCKADDR_UN
struct sockaddr_un un;
#endif
} la, ra;
int new;
ISC_SOCKLEN_T lalen = 0, ralen;
REQUIRE((evmask & EV_READ) != 0);
ralen = sizeof ra;
new = accept(fd, &ra.sa, &ralen);
if (new > ctx->highestFD) {
close(new);
new = -1;
errno = ENOTSOCK;
}
if (new >= 0) {
lalen = sizeof la;
if (GETXXXNAME(getsockname, new, la.sa, lalen) < 0) {
int save = errno;
(void) close(new);
errno = save;
new = -1;
}
} else if (errno == EAGAIN || errno == EWOULDBLOCK)
return;
(*conn->func)(opaqueCtx, conn->uap, new, &la.sa, lalen, &ra.sa, ralen);
}
static void
connector(evContext opaqueCtx, void *uap, int fd, int evmask) {
evConn *conn = uap;
union {
struct sockaddr sa;
struct sockaddr_in in;
#ifndef NO_SOCKADDR_UN
struct sockaddr_un un;
#endif
} la, ra;
ISC_SOCKLEN_T lalen, ralen;
#ifndef NETREAD_BROKEN
char buf[1];
#endif
void *conn_uap;
evConnFunc conn_func;
evConnID id;
int socket_errno = 0;
ISC_SOCKLEN_T optlen;
UNUSED(evmask);
lalen = sizeof la;
ralen = sizeof ra;
conn_uap = conn->uap;
conn_func = conn->func;
id.opaque = conn;
#ifdef SO_ERROR
optlen = sizeof socket_errno;
if (fd < 0 &&
getsockopt(conn->fd, SOL_SOCKET, SO_ERROR, (char *)&socket_errno,
&optlen) < 0)
socket_errno = errno;
else
errno = socket_errno;
#endif
if (evCancelConn(opaqueCtx, id) < 0 ||
socket_errno ||
#ifdef NETREAD_BROKEN
0 ||
#else
read(fd, buf, 0) < 0 ||
#endif
GETXXXNAME(getsockname, fd, la.sa, lalen) < 0 ||
GETXXXNAME(getpeername, fd, ra.sa, ralen) < 0) {
int save = errno;
(void) close(fd); /*%< XXX closing caller's fd */
errno = save;
fd = -1;
}
(*conn_func)(opaqueCtx, conn_uap, fd, &la.sa, lalen, &ra.sa, ralen);
}
/*! \file */