[irc/rqf/shadowircd.git] / libratbox / src / kqueue.c

/*
 *  ircd-ratbox: A slightly useful ircd.
 *  kqueue.c: FreeBSD kqueue compatible network routines.
 *
 *  Copyright (C) 1990 Jarkko Oikarinen and University of Oulu, Co Center
 *  Copyright (C) 1996-2002 Hybrid Development Team
 *  Copyright (C) 2001 Adrian Chadd <adrian@creative.net.au>
 *  Copyright (C) 2002-2005 ircd-ratbox development team
 *
 *  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 2 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, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301
 *  USA
 *
 *  $Id: kqueue.c 25038 2008-01-23 16:03:08Z androsyn $
 */

#include <libratbox_config.h>
#include <ratbox_lib.h>
#include <commio-int.h>
#include <event-int.h>

#if defined(HAVE_SYS_EVENT_H) && (HAVE_KEVENT)

#include <sys/event.h>

/* jlemon goofed up and didn't add EV_SET until fbsd 4.3 */

#ifndef EV_SET
#define EV_SET(kevp, a, b, c, d, e, f) do {     \
        (kevp)->ident = (a);                    \
        (kevp)->filter = (b);                   \
        (kevp)->flags = (c);                    \
        (kevp)->fflags = (d);                   \
        (kevp)->data = (e);                     \
        (kevp)->udata = (f);                    \
} while(0)
#endif

#ifdef EVFILT_TIMER
#define KQUEUE_SCHED_EVENT
#endif


static void kq_update_events(rb_fde_t *, short, PF *);
static int kq;
static struct timespec zero_timespec;

static struct kevent *kqlst;	/* kevent buffer */
static struct kevent *kqout;	/* kevent output buffer */
static int kqmax;		/* max structs to buffer */
static int kqoff;		/* offset into the buffer */


int
rb_setup_fd_kqueue(rb_fde_t * F)
{
	return 0;
}

static void
kq_update_events(rb_fde_t * F, short filter, PF * handler)
{
	PF *cur_handler;
	int kep_flags;

	switch (filter)
	{
	case EVFILT_READ:
		cur_handler = F->read_handler;
		break;
	case EVFILT_WRITE:
		cur_handler = F->write_handler;
		break;
	default:
		/* XXX bad! -- adrian */
		return;
		break;
	}

	if((cur_handler == NULL && handler != NULL) || (cur_handler != NULL && handler == NULL))
	{
		struct kevent *kep;

		kep = kqlst + kqoff;

		if(handler != NULL)
		{
			kep_flags = EV_ADD | EV_ONESHOT;
		}
		else
		{
			kep_flags = EV_DELETE;
		}

		EV_SET(kep, (uintptr_t) F->fd, filter, kep_flags, 0, 0, (void *) F);

		if(++kqoff == kqmax)
		{
			int ret, i;

			/* Add them one at a time, because there may be
			 * already closed fds in it. The kernel will try
			 * to report invalid fds in the output; if there
			 * is no space, it silently stops processing the
			 * array at that point. We cannot give output space
			 * because that would also return events we cannot
			 * process at this point.
			 */
			for (i = 0; i < kqoff; i++)
			{
				ret = kevent(kq, kqlst + i, 1, NULL, 0, &zero_timespec);
				/* jdc -- someone needs to do error checking... */
				/* EBADF is normal here -- jilles */
				if(ret == -1 && errno != EBADF)
					rb_lib_log("kq_update_events(): kevent(): %s", strerror(errno));
			}
			kqoff = 0;
		}
	}
}


/* XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX */
/* Public functions */


/*
 * rb_init_netio
 *
 * This is a needed exported function which will be called to initialise
 * the network loop code.
 */
int
rb_init_netio_kqueue(void)
{
	kq = kqueue();
	if(kq < 0)
	{
		return errno;
	}
	kqmax = getdtablesize();
	kqlst = rb_malloc(sizeof(struct kevent) * kqmax);
	kqout = rb_malloc(sizeof(struct kevent) * kqmax);
	rb_open(kq, RB_FD_UNKNOWN, "kqueue fd");
	zero_timespec.tv_sec = 0;
	zero_timespec.tv_nsec = 0;

	return 0;
}

/*
 * rb_setselect
 *
 * This is a needed exported function which will be called to register
 * and deregister interest in a pending IO state for a given FD.
 */
void
rb_setselect_kqueue(rb_fde_t * F, unsigned int type, PF * handler, void *client_data)
{
	lrb_assert(IsFDOpen(F));

	if(type & RB_SELECT_READ)
	{
		kq_update_events(F, EVFILT_READ, handler);
		F->read_handler = handler;
		F->read_data = client_data;
	}
	if(type & RB_SELECT_WRITE)
	{
		kq_update_events(F, EVFILT_WRITE, handler);
		F->write_handler = handler;
		F->write_data = client_data;
	}
}

/*
 * Check all connections for new connections and input data that is to be
 * processed. Also check for connections with data queued and whether we can
 * write it out.
 */

/*
 * rb_select
 *
 * Called to do the new-style IO, courtesy of squid (like most of this
 * new IO code). This routine handles the stuff we've hidden in
 * rb_setselect and fd_table[] and calls callbacks for IO ready
 * events.
 */

int
rb_select_kqueue(long delay)
{
	int num, i;
	struct timespec poll_time;
	struct timespec *pt;
	rb_fde_t *F;


	if(delay < 0) {
		pt = NULL;
	}
	else {
		pt = &poll_time;
		poll_time.tv_sec = delay / 1000;
		poll_time.tv_nsec = (delay % 1000) * 1000000;
	}

	for (;;)
	{
		num = kevent(kq, kqlst, kqoff, kqout, kqmax, pt);
		kqoff = 0;

		if(num >= 0)
			break;

		if(rb_ignore_errno(errno))
			break;

		rb_set_time();

		return RB_ERROR;

		/* NOTREACHED */
	}

	rb_set_time();

	if(num == 0)
		return RB_OK;	/* No error.. */

	for (i = 0; i < num; i++)
	{
		PF *hdl = NULL;

		if(kqout[i].flags & EV_ERROR)
		{
			errno = kqout[i].data;
			/* XXX error == bad! -- adrian */
			continue;	/* XXX! */
		}

		switch (kqout[i].filter)
		{

		case EVFILT_READ:
			F = kqout[i].udata;
			if((hdl = F->read_handler) != NULL)
			{
				F->read_handler = NULL;
				hdl(F, F->read_data);
			}

			break;

		case EVFILT_WRITE:
			F = kqout[i].udata;
			if((hdl = F->write_handler) != NULL)
			{
				F->write_handler = NULL;
				hdl(F, F->write_data);
			}
			break;
#if defined(EVFILT_TIMER)
		case EVFILT_TIMER:
			rb_run_event(kqout[i].udata);
			break;
#endif
		default:
			/* Bad! -- adrian */
			break;
		}
	}
	return RB_OK;
}
static int can_do_event = 0;
int
rb_kqueue_supports_event(void)
{
	struct kevent kv;
	struct timespec ts;
	int xkq;

	if(can_do_event == 1)
		return 1;
	if(can_do_event == -1)
		return 0;

	xkq = kqueue();
	ts.tv_sec = 0;
	ts.tv_nsec = 1000;


	EV_SET(&kv, (uintptr_t) 0x0, EVFILT_TIMER, EV_ADD | EV_ONESHOT, 0, 1, 0);
	if(kevent(xkq, &kv, 1, NULL, 0, NULL) < 0)
	{
		can_do_event = -1;
		close(xkq);
		return 0;
	}
	close(xkq);
	can_do_event = 1;
	return 1;
}

int
rb_kqueue_sched_event(struct ev_entry *event, int when)
{
	struct kevent kev;
	int kep_flags;

	kep_flags = EV_ADD;
	if(event->frequency == 0)
		kep_flags |= EV_ONESHOT;
	EV_SET(&kev, (uintptr_t) event, EVFILT_TIMER, kep_flags, 0, when * 1000, event);
	if(kevent(kq, &kev, 1, NULL, 0, NULL) < 0)
		return 0; 
	return 1;
}

void
rb_kqueue_unsched_event(struct ev_entry *event)
{
	struct kevent kev;
	EV_SET(&kev, (uintptr_t) event, EVFILT_TIMER, EV_DELETE, 0, 0, event);
	kevent(kq, &kev, 1, NULL, 0, NULL);
}

void
rb_kqueue_init_event(void)
{
	return;
}

#else /* kqueue not supported */
int
rb_init_netio_kqueue(void)
{
	errno = ENOSYS;
	return -1;
}

void
rb_setselect_kqueue(rb_fde_t * F, unsigned int type, PF * handler, void *client_data)
{
	errno = ENOSYS;
	return;
}

int
rb_select_kqueue(long delay)
{
	errno = ENOSYS;
	return -1;
}

int
rb_setup_fd_kqueue(rb_fde_t * F)
{
	errno = ENOSYS;
	return -1;
}

#endif

#if !defined(HAVE_KEVENT) || !defined(KQUEUE_SCHED_EVENT)
void
rb_kqueue_init_event(void)
{
	return;
}

int
rb_kqueue_sched_event(struct ev_entry *event, int when)
{
	errno = ENOSYS;
	return -1;
}

void
rb_kqueue_unsched_event(struct ev_entry *event)
{
	return;
}

int
rb_kqueue_supports_event(void)
{
	errno = ENOSYS;
	return 0;
}
#endif /* !HAVE_KEVENT || !KQUEUE_SCHED_EVENT */
Commit	Line	Data
b57f37fb WP	1	/*
	2	* ircd-ratbox: A slightly useful ircd.
	3	* kqueue.c: FreeBSD kqueue compatible network routines.
	4	*
	5	* Copyright (C) 1990 Jarkko Oikarinen and University of Oulu, Co Center
	6	* Copyright (C) 1996-2002 Hybrid Development Team
	7	* Copyright (C) 2001 Adrian Chadd <adrian@creative.net.au>
	8	* Copyright (C) 2002-2005 ircd-ratbox development team
	9	*
	10	* This program is free software; you can redistribute it and/or modify
	11	* it under the terms of the GNU General Public License as published by
	12	* the Free Software Foundation; either version 2 of the License, or
	13	* (at your option) any later version.
	14	*
	15	* This program is distributed in the hope that it will be useful,
	16	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	17	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	18	* GNU General Public License for more details.
	19	*
	20	* You should have received a copy of the GNU General Public License
	21	* along with this program; if not, write to the Free Software
	22	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301
	23	* USA
	24	*
	25	* $Id: kqueue.c 25038 2008-01-23 16:03:08Z androsyn $
	26	*/
	27
	28	#include <libratbox_config.h>
	29	#include <ratbox_lib.h>
	30	#include <commio-int.h>
	31	#include <event-int.h>
	32
	33	#if defined(HAVE_SYS_EVENT_H) && (HAVE_KEVENT)
	34
	35	#include <sys/event.h>
	36
b57f37fb WP	37	/* jlemon goofed up and didn't add EV_SET until fbsd 4.3 */
	38
	39	#ifndef EV_SET
	40	#define EV_SET(kevp, a, b, c, d, e, f) do { \
	41	(kevp)->ident = (a); \
	42	(kevp)->filter = (b); \
	43	(kevp)->flags = (c); \
	44	(kevp)->fflags = (d); \
	45	(kevp)->data = (e); \
	46	(kevp)->udata = (f); \
	47	} while(0)
	48	#endif
	49
	50	#ifdef EVFILT_TIMER
	51	#define KQUEUE_SCHED_EVENT
	52	#endif
	53
	54
	55	static void kq_update_events(rb_fde_t , short, PF );
	56	static int kq;
	57	static struct timespec zero_timespec;
	58
	59	static struct kevent kqlst; / kevent buffer */
98686f18	60	static struct kevent kqout; / kevent output buffer */
b57f37fb WP	61	static int kqmax; /* max structs to buffer */
	62	static int kqoff; /* offset into the buffer */
	63
	64
	65	int
	66	rb_setup_fd_kqueue(rb_fde_t * F)
	67	{
	68	return 0;
	69	}
	70
	71	static void
	72	kq_update_events(rb_fde_t * F, short filter, PF * handler)
	73	{
	74	PF *cur_handler;
	75	int kep_flags;
	76
	77	switch (filter)
	78	{
	79	case EVFILT_READ:
	80	cur_handler = F->read_handler;
	81	break;
	82	case EVFILT_WRITE:
	83	cur_handler = F->write_handler;
	84	break;
	85	default:
	86	/* XXX bad! -- adrian */
	87	return;
	88	break;
	89	}
	90
	91	if((cur_handler == NULL && handler != NULL) \|\| (cur_handler != NULL && handler == NULL))
	92	{
	93	struct kevent *kep;
	94
	95	kep = kqlst + kqoff;
	96
	97	if(handler != NULL)
	98	{
c50bb62c	99	kep_flags = EV_ADD \| EV_ONESHOT;
b57f37fb WP	100	}
	101	else
	102	{
	103	kep_flags = EV_DELETE;
	104	}
	105
	106	EV_SET(kep, (uintptr_t) F->fd, filter, kep_flags, 0, 0, (void *) F);
	107
	108	if(++kqoff == kqmax)
	109	{
98686f18 JT	110	int ret, i;
	111
	112	/* Add them one at a time, because there may be
	113	* already closed fds in it. The kernel will try
	114	* to report invalid fds in the output; if there
	115	* is no space, it silently stops processing the
	116	* array at that point. We cannot give output space
	117	* because that would also return events we cannot
	118	* process at this point.
	119	*/
	120	for (i = 0; i < kqoff; i++)
b57f37fb	121	{
98686f18 JT	122	ret = kevent(kq, kqlst + i, 1, NULL, 0, &zero_timespec);
98686f18 JT	123	/* jdc -- someone needs to do error checking... */
e0bd36a5 JT	124	/* EBADF is normal here -- jilles */
e0bd36a5 JT	125	if(ret == -1 && errno != EBADF)
98686f18	126	rb_lib_log("kq_update_events(): kevent(): %s", strerror(errno));
b57f37fb WP	127	}
	128	kqoff = 0;
	129	}
	130	}
	131	}
	132
	133
	134
	135	/* XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX */
	136	/* Public functions */
	137
	138
	139	/*
	140	* rb_init_netio
	141	*
	142	* This is a needed exported function which will be called to initialise
	143	* the network loop code.
	144	*/
	145	int
	146	rb_init_netio_kqueue(void)
	147	{
	148	kq = kqueue();
	149	if(kq < 0)
	150	{
	151	return errno;
	152	}
	153	kqmax = getdtablesize();
	154	kqlst = rb_malloc(sizeof(struct kevent) * kqmax);
98686f18	155	kqout = rb_malloc(sizeof(struct kevent) * kqmax);
b57f37fb WP	156	rb_open(kq, RB_FD_UNKNOWN, "kqueue fd");
	157	zero_timespec.tv_sec = 0;
	158	zero_timespec.tv_nsec = 0;
	159
	160	return 0;
	161	}
	162
	163	/*
	164	* rb_setselect
	165	*
	166	* This is a needed exported function which will be called to register
	167	* and deregister interest in a pending IO state for a given FD.
	168	*/
	169	void
	170	rb_setselect_kqueue(rb_fde_t * F, unsigned int type, PF * handler, void *client_data)
	171	{
	172	lrb_assert(IsFDOpen(F));
	173
	174	if(type & RB_SELECT_READ)
	175	{
	176	kq_update_events(F, EVFILT_READ, handler);
	177	F->read_handler = handler;
	178	F->read_data = client_data;
	179	}
	180	if(type & RB_SELECT_WRITE)
	181	{
	182	kq_update_events(F, EVFILT_WRITE, handler);
	183	F->write_handler = handler;
	184	F->write_data = client_data;
	185	}
	186	}
	187
	188	/*
	189	* Check all connections for new connections and input data that is to be
	190	* processed. Also check for connections with data queued and whether we can
	191	* write it out.
	192	*/
	193
	194	/*
	195	* rb_select
	196	*
	197	* Called to do the new-style IO, courtesy of squid (like most of this
	198	* new IO code). This routine handles the stuff we've hidden in
	199	* rb_setselect and fd_table[] and calls callbacks for IO ready
	200	* events.
	201	*/
	202
	203	int
	204	rb_select_kqueue(long delay)
	205	{
	206	int num, i;
b57f37fb WP	207	struct timespec poll_time;
	208	struct timespec *pt;
	209	rb_fde_t *F;
	210
	211
	212	if(delay < 0) {
	213	pt = NULL;
	214	}
	215	else {
	216	pt = &poll_time;
	217	poll_time.tv_sec = delay / 1000;
	218	poll_time.tv_nsec = (delay % 1000) * 1000000;
	219	}
	220
	221	for (;;)
	222	{
98686f18	223	num = kevent(kq, kqlst, kqoff, kqout, kqmax, pt);
b57f37fb WP	224	kqoff = 0;
	225
	226	if(num >= 0)
	227	break;
	228
	229	if(rb_ignore_errno(errno))
	230	break;
	231
	232	rb_set_time();
	233
	234	return RB_ERROR;
	235
	236	/* NOTREACHED */
	237	}
	238
	239	rb_set_time();
	240
	241	if(num == 0)
	242	return RB_OK; /* No error.. */
	243
	244	for (i = 0; i < num; i++)
	245	{
	246	PF *hdl = NULL;
	247
98686f18	248	if(kqout[i].flags & EV_ERROR)
b57f37fb	249	{
98686f18	250	errno = kqout[i].data;
b57f37fb WP	251	/* XXX error == bad! -- adrian */
	252	continue; /* XXX! */
	253	}
	254
98686f18	255	switch (kqout[i].filter)
b57f37fb WP	256	{
	257
	258	case EVFILT_READ:
98686f18	259	F = kqout[i].udata;
b57f37fb WP	260	if((hdl = F->read_handler) != NULL)
	261	{
	262	F->read_handler = NULL;
	263	hdl(F, F->read_data);
	264	}
	265
	266	break;
	267
	268	case EVFILT_WRITE:
98686f18	269	F = kqout[i].udata;
b57f37fb WP	270	if((hdl = F->write_handler) != NULL)
	271	{
	272	F->write_handler = NULL;
	273	hdl(F, F->write_data);
	274	}
	275	break;
	276	#if defined(EVFILT_TIMER)
	277	case EVFILT_TIMER:
98686f18	278	rb_run_event(kqout[i].udata);
b57f37fb WP	279	break;
	280	#endif
	281	default:
	282	/* Bad! -- adrian */
	283	break;
	284	}
	285	}
	286	return RB_OK;
	287	}
	288	static int can_do_event = 0;
	289	int
	290	rb_kqueue_supports_event(void)
	291	{
	292	struct kevent kv;
	293	struct timespec ts;
	294	int xkq;
	295
	296	if(can_do_event == 1)
	297	return 1;
	298	if(can_do_event == -1)
	299	return 0;
	300
	301	xkq = kqueue();
	302	ts.tv_sec = 0;
	303	ts.tv_nsec = 1000;
	304
	305
	306	EV_SET(&kv, (uintptr_t) 0x0, EVFILT_TIMER, EV_ADD \| EV_ONESHOT, 0, 1, 0);
	307	if(kevent(xkq, &kv, 1, NULL, 0, NULL) < 0)
	308	{
	309	can_do_event = -1;
	310	close(xkq);
	311	return 0;
	312	}
	313	close(xkq);
	314	can_do_event = 1;
	315	return 1;
	316	}
	317
	318	int
	319	rb_kqueue_sched_event(struct ev_entry *event, int when)
	320	{
	321	struct kevent kev;
	322	int kep_flags;
	323
	324	kep_flags = EV_ADD;
	325	if(event->frequency == 0)
	326	kep_flags \|= EV_ONESHOT;
	327	EV_SET(&kev, (uintptr_t) event, EVFILT_TIMER, kep_flags, 0, when * 1000, event);
	328	if(kevent(kq, &kev, 1, NULL, 0, NULL) < 0)
	329	return 0;
	330	return 1;
	331	}
	332
	333	void
	334	rb_kqueue_unsched_event(struct ev_entry *event)
	335	{
	336	struct kevent kev;
	337	EV_SET(&kev, (uintptr_t) event, EVFILT_TIMER, EV_DELETE, 0, 0, event);
	338	kevent(kq, &kev, 1, NULL, 0, NULL);
	339	}
	340
	341	void
	342	rb_kqueue_init_event(void)
343	{
344	return;
345	}
346
347	#else /* kqueue not supported */
348	int
349	rb_init_netio_kqueue(void)
350	{
351	errno = ENOSYS;
352	return -1;
353	}
354
355	void
356	rb_setselect_kqueue(rb_fde_t * F, unsigned int type, PF * handler, void *client_data)
357	{
358	errno = ENOSYS;
359	return;
360	}
361
362	int
363	rb_select_kqueue(long delay)
364	{
365	errno = ENOSYS;
366	return -1;
367	}
368
369	int
370	rb_setup_fd_kqueue(rb_fde_t * F)
371	{
372	errno = ENOSYS;
373	return -1;
374	}
375
376	#endif
377
378	#if !defined(HAVE_KEVENT) \|\| !defined(KQUEUE_SCHED_EVENT)
379	void
380	rb_kqueue_init_event(void)
381	{
382	return;
383	}
384
385	int
386	rb_kqueue_sched_event(struct ev_entry *event, int when)
387	{
388	errno = ENOSYS;
389	return -1;
390	}
391
392	void
393	rb_kqueue_unsched_event(struct ev_entry *event)
394	{
395	return;
396	}
397
398	int
399	rb_kqueue_supports_event(void)
400	{
401	errno = ENOSYS;
402	return 0;
403	}
404	#endif /* !HAVE_KEVENT \|\| !KQUEUE_SCHED_EVENT */