2010-01-26 22:30:32 +00:00
|
|
|
/*
|
|
|
|
* uhub - A tiny ADC p2p connection hub
|
|
|
|
* Copyright (C) 2007-2010, Jan Vidar Krey
|
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation; either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "uhub.h"
|
|
|
|
|
|
|
|
#ifdef USE_KQUEUE
|
|
|
|
|
|
|
|
#include "network/connection.h"
|
|
|
|
#include "network/common.h"
|
|
|
|
#include "network/backend.h"
|
|
|
|
|
|
|
|
#define KQUEUE_EVBUFFER 512
|
|
|
|
|
|
|
|
struct net_connection_kqueue
|
|
|
|
{
|
|
|
|
NET_CON_STRUCT_COMMON
|
2010-04-22 14:10:17 +00:00
|
|
|
struct kevent ev_r;
|
|
|
|
struct kevent ev_w;
|
2010-01-26 22:30:32 +00:00
|
|
|
};
|
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
struct net_backend_kqueue
|
2010-01-26 22:30:32 +00:00
|
|
|
{
|
|
|
|
int kqfd;
|
2010-01-27 16:50:02 +00:00
|
|
|
struct net_connection_kqueue** conns;
|
2010-01-26 22:30:32 +00:00
|
|
|
struct kevent** changes;
|
|
|
|
size_t nchanges;
|
|
|
|
struct kevent events[KQUEUE_EVBUFFER];
|
2010-02-10 20:57:27 +00:00
|
|
|
struct net_backend_common* common;
|
2010-01-26 22:30:32 +00:00
|
|
|
};
|
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
static void net_backend_set_handlers(struct net_backend_handler* handler);
|
2010-01-26 22:30:32 +00:00
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
const char* net_backend_name_kqueue()
|
|
|
|
{
|
|
|
|
return "kqueue";
|
|
|
|
}
|
2010-01-26 22:30:32 +00:00
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
int net_backend_poll_kqueue(struct net_backend* data, int ms)
|
2010-01-26 22:30:32 +00:00
|
|
|
{
|
2010-02-10 20:57:27 +00:00
|
|
|
int res;
|
|
|
|
struct timespec tspec = { 0, };
|
|
|
|
struct net_backend_kqueue* backend = (struct net_backend_kqueue*) data;
|
2010-01-26 22:30:32 +00:00
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
tspec.tv_sec = (ms / 1000);
|
2010-04-22 14:10:17 +00:00
|
|
|
tspec.tv_nsec = ((ms % 1000) * 1000000);
|
2010-01-26 22:30:32 +00:00
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
res = kevent(backend->kqfd, *backend->changes, backend->nchanges, backend->events, KQUEUE_EVBUFFER, &tspec);
|
|
|
|
backend->nchanges = 0;
|
2010-01-26 22:30:32 +00:00
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
if (res == -1 && errno == EINTR)
|
|
|
|
return 0;
|
|
|
|
return res;
|
2010-01-26 22:30:32 +00:00
|
|
|
}
|
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
void net_backend_process_kqueue(struct net_backend* data, int res)
|
2010-01-26 22:30:32 +00:00
|
|
|
{
|
2010-01-27 16:50:02 +00:00
|
|
|
int n;
|
2010-02-10 20:57:27 +00:00
|
|
|
struct net_backend_kqueue* backend = (struct net_backend_kqueue*) data;
|
2010-01-27 16:50:02 +00:00
|
|
|
|
|
|
|
for (n = 0; n < res; n++)
|
|
|
|
{
|
2010-02-10 20:57:27 +00:00
|
|
|
struct net_connection_kqueue* con = (struct net_connection_kqueue*) backend->events[n].udata;
|
2010-04-22 14:10:17 +00:00
|
|
|
int ev = -1;
|
|
|
|
if (backend->events[n].filter == EVFILT_READ) ev = NET_EVENT_READ;
|
|
|
|
else if (backend->events[n].filter == EVFILT_WRITE) ev = NET_EVENT_WRITE;
|
|
|
|
if (con)
|
|
|
|
net_con_callback((struct net_connection*) con, ev);
|
2010-01-27 16:50:02 +00:00
|
|
|
}
|
2010-01-26 22:30:32 +00:00
|
|
|
}
|
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
struct net_connection* net_con_create_kqueue(struct net_backend* data)
|
2010-01-26 22:30:32 +00:00
|
|
|
{
|
|
|
|
struct net_connection* con = (struct net_connection*) hub_malloc_zero(sizeof(struct net_connection_kqueue));
|
|
|
|
con->sd = -1;
|
|
|
|
return con;
|
|
|
|
}
|
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
void net_con_initialize_kqueue(struct net_backend* data, struct net_connection* con_, int sd, net_connection_cb callback, const void* ptr)
|
2010-01-26 22:30:32 +00:00
|
|
|
{
|
|
|
|
struct net_connection_kqueue* con = (struct net_connection_kqueue*) con_;
|
|
|
|
con->sd = sd;
|
|
|
|
con->flags = 0;
|
|
|
|
con->callback = callback;
|
|
|
|
con->ptr = (void*) ptr;
|
|
|
|
}
|
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
void net_con_backend_add_kqueue(struct net_backend* data, struct net_connection* con_, int events)
|
2010-01-26 22:30:32 +00:00
|
|
|
{
|
2010-04-22 14:10:17 +00:00
|
|
|
unsigned short flags_r = EV_ADD;
|
|
|
|
unsigned short flags_w = EV_ADD;
|
2010-02-10 20:57:27 +00:00
|
|
|
struct net_backend_kqueue* backend = (struct net_backend_kqueue*) data;
|
|
|
|
struct net_connection_kqueue* con = (struct net_connection_kqueue*) con_;
|
2010-04-22 14:10:17 +00:00
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
backend->conns[con->sd] = con;
|
2010-04-22 14:10:17 +00:00
|
|
|
|
|
|
|
if (events & NET_EVENT_READ)
|
|
|
|
flags_r |= EV_ENABLE;
|
|
|
|
else
|
|
|
|
flags_r |= EV_DISABLE;
|
|
|
|
|
|
|
|
EV_SET(&con->ev_r, con->sd, EVFILT_READ, flags_r, 0, 0, con);
|
|
|
|
backend->changes[backend->nchanges++] = &con->ev_r;
|
|
|
|
|
|
|
|
if (events & NET_EVENT_WRITE)
|
|
|
|
flags_w |= EV_ENABLE;
|
|
|
|
else
|
|
|
|
flags_w |= EV_DISABLE;
|
|
|
|
|
|
|
|
EV_SET(&con->ev_w, con->sd, EVFILT_WRITE, flags_w, 0, 0, con);
|
|
|
|
backend->changes[backend->nchanges++] = &con->ev_w;
|
2010-01-26 22:30:32 +00:00
|
|
|
}
|
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
void net_con_backend_mod_kqueue(struct net_backend* data, struct net_connection* con_, int events)
|
2010-01-26 22:30:32 +00:00
|
|
|
{
|
2010-04-22 14:10:17 +00:00
|
|
|
unsigned short flags_r = 0;
|
|
|
|
unsigned short flags_w = 0;
|
2010-02-10 20:57:27 +00:00
|
|
|
struct net_backend_kqueue* backend = (struct net_backend_kqueue*) data;
|
2010-01-26 22:30:32 +00:00
|
|
|
struct net_connection_kqueue* con = (struct net_connection_kqueue*) con_;
|
2010-01-26 23:43:10 +00:00
|
|
|
|
2010-04-22 14:10:17 +00:00
|
|
|
if (events & NET_EVENT_READ)
|
|
|
|
flags_r |= EV_ENABLE;
|
|
|
|
else
|
|
|
|
flags_r |= EV_DISABLE;
|
|
|
|
|
|
|
|
if (!(con->ev_r.flags & flags_r))
|
|
|
|
{
|
|
|
|
EV_SET(&con->ev_r, con->sd, EVFILT_READ, flags_r, 0, 0, con);
|
|
|
|
backend->changes[backend->nchanges++] = &con->ev_r;
|
|
|
|
}
|
2010-01-26 23:43:10 +00:00
|
|
|
|
2010-04-22 14:10:17 +00:00
|
|
|
if (events & NET_EVENT_WRITE)
|
|
|
|
flags_r |= EV_ENABLE;
|
|
|
|
else
|
|
|
|
flags_r |= EV_DISABLE;
|
2010-01-26 23:43:10 +00:00
|
|
|
|
2010-04-22 14:10:17 +00:00
|
|
|
if (!(con->ev_w.flags & flags_w))
|
|
|
|
{
|
|
|
|
EV_SET(&con->ev_w, con->sd, EVFILT_WRITE, flags_w, 0, 0, con);
|
|
|
|
backend->changes[backend->nchanges++] = &con->ev_w;
|
|
|
|
}
|
2010-01-26 22:30:32 +00:00
|
|
|
}
|
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
void net_con_backend_del_kqueue(struct net_backend* data, struct net_connection* con_)
|
2010-01-26 22:30:32 +00:00
|
|
|
{
|
2010-02-10 20:57:27 +00:00
|
|
|
struct net_backend_kqueue* backend = (struct net_backend_kqueue*) data;
|
2010-01-27 00:04:32 +00:00
|
|
|
struct net_connection_kqueue* con = (struct net_connection_kqueue*) con_;
|
2010-01-26 22:30:32 +00:00
|
|
|
|
2010-01-27 16:50:02 +00:00
|
|
|
/* No need to remove it from the kqueue filter, the kqueue man page says
|
|
|
|
it is automatically removed when the descriptor is closed. */
|
2010-04-22 14:10:17 +00:00
|
|
|
EV_SET(&con->ev_r, con->sd, EVFILT_READ, EV_DELETE, 0, 0, 0);
|
|
|
|
backend->changes[backend->nchanges++] = &con->ev_r;
|
|
|
|
EV_SET(&con->ev_w, con->sd, EVFILT_WRITE, EV_DELETE, 0, 0, 0);
|
|
|
|
backend->changes[backend->nchanges++] = &con->ev_w;
|
|
|
|
|
|
|
|
// Unmap the socket descriptor.
|
|
|
|
backend->conns[con->sd] = 0;
|
2010-02-10 20:57:27 +00:00
|
|
|
}
|
2010-01-26 22:30:32 +00:00
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
void net_backend_shutdown_kqueue(struct net_backend* data)
|
|
|
|
{
|
|
|
|
struct net_backend_kqueue* backend = (struct net_backend_kqueue*) data;
|
|
|
|
close(backend->kqfd);
|
|
|
|
hub_free(backend->conns);
|
|
|
|
hub_free(backend->changes);
|
|
|
|
hub_free(backend);
|
|
|
|
}
|
|
|
|
|
|
|
|
struct net_backend* net_backend_init_kqueue(struct net_backend_handler* handler, struct net_backend_common* common)
|
|
|
|
{
|
|
|
|
struct net_backend_kqueue* backend;
|
|
|
|
|
|
|
|
if (getenv("EVENT_NOKQUEUE"))
|
|
|
|
return 0;
|
2010-01-26 22:30:32 +00:00
|
|
|
|
2010-02-10 20:57:27 +00:00
|
|
|
backend = hub_malloc_zero(sizeof(struct net_backend_kqueue));
|
2010-02-11 06:54:42 +00:00
|
|
|
backend->kqfd = kqueue();
|
2010-02-10 20:57:27 +00:00
|
|
|
if (backend->kqfd == -1)
|
|
|
|
{
|
|
|
|
LOG_WARN("Unable to create kqueue socket.");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
backend->conns = hub_malloc_zero(sizeof(struct net_connection_kqueue*) * common->max);
|
|
|
|
backend->conns = hub_malloc_zero(sizeof(struct net_connection_kqueue*) * common->max);
|
2010-04-22 14:10:17 +00:00
|
|
|
backend->changes = hub_malloc_zero(sizeof(struct kevent*) * common->max * 2);
|
2010-02-10 20:57:27 +00:00
|
|
|
backend->common = common;
|
|
|
|
|
|
|
|
net_backend_set_handlers(handler);
|
|
|
|
return (struct net_backend*) backend;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void net_backend_set_handlers(struct net_backend_handler* handler)
|
|
|
|
{
|
|
|
|
handler->backend_name = net_backend_name_kqueue;
|
|
|
|
handler->backend_poll = net_backend_poll_kqueue;
|
|
|
|
handler->backend_process = net_backend_process_kqueue;
|
|
|
|
handler->backend_shutdown = net_backend_shutdown_kqueue;
|
|
|
|
handler->con_create = net_con_create_kqueue;
|
|
|
|
handler->con_init = net_con_initialize_kqueue;
|
|
|
|
handler->con_add = net_con_backend_add_kqueue;
|
|
|
|
handler->con_mod = net_con_backend_mod_kqueue;
|
|
|
|
handler->con_del = net_con_backend_del_kqueue;
|
2010-01-26 22:30:32 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#endif /* USE_KQUEUE */
|