mirror of
https://github.com/rspamd/rspamd.git
synced 2024-08-08 14:10:24 +02:00
233 lines
6.3 KiB
C
233 lines
6.3 KiB
C
/*
|
|
* Copyright (c) 2014, Vsevolod Stakhov
|
|
*
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
* * Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* * 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.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY AUTHOR ''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 AUTHOR 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.
|
|
*/
|
|
#ifndef RDNS_EV_H_
|
|
#define RDNS_EV_H_
|
|
|
|
#include <ev.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include "rdns.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
static void* rdns_libev_add_read (void *priv_data, int fd, void *user_data);
|
|
static void rdns_libev_del_read(void *priv_data, void *ev_data);
|
|
static void* rdns_libev_add_write (void *priv_data, int fd, void *user_data);
|
|
static void rdns_libev_del_write (void *priv_data, void *ev_data);
|
|
static void* rdns_libev_add_timer (void *priv_data, double after, void *user_data);
|
|
static void* rdns_libev_add_periodic (void *priv_data, double after,
|
|
rdns_periodic_callback cb, void *user_data);
|
|
static void rdns_libev_del_periodic (void *priv_data, void *ev_data);
|
|
static void rdns_libev_repeat_timer (void *priv_data, void *ev_data);
|
|
static void rdns_libev_del_timer (void *priv_data, void *ev_data);
|
|
|
|
struct rdns_ev_periodic_cbdata {
|
|
ev_timer *ev;
|
|
rdns_periodic_callback cb;
|
|
void *cbdata;
|
|
};
|
|
|
|
static void
|
|
rdns_bind_libev (struct rdns_resolver *resolver, struct ev_loop *loop)
|
|
{
|
|
static struct rdns_async_context ev_ctx = {
|
|
.data = NULL,
|
|
.add_read = rdns_libev_add_read,
|
|
.del_read = rdns_libev_del_read,
|
|
.add_write = rdns_libev_add_write,
|
|
.del_write = rdns_libev_del_write,
|
|
.add_timer = rdns_libev_add_timer,
|
|
.repeat_timer = rdns_libev_repeat_timer,
|
|
.del_timer = rdns_libev_del_timer,
|
|
.add_periodic = rdns_libev_add_periodic,
|
|
.del_periodic = rdns_libev_del_periodic,
|
|
.cleanup = NULL
|
|
}, *nctx;
|
|
void *ptr;
|
|
|
|
/* XXX: never got freed */
|
|
ptr = malloc (sizeof (struct rdns_async_context));
|
|
if (ptr != NULL) {
|
|
nctx = (struct rdns_async_context *)ptr;
|
|
memcpy (ptr, (void *)&ev_ctx, sizeof (struct rdns_async_context));
|
|
nctx->data = (void*)loop;
|
|
rdns_resolver_async_bind (resolver, nctx);
|
|
}
|
|
}
|
|
|
|
static void
|
|
rdns_libev_read_event (struct ev_loop *loop, ev_io *ev, int revents)
|
|
{
|
|
rdns_process_read (ev->fd, ev->data);
|
|
}
|
|
|
|
static void
|
|
rdns_libev_write_event (struct ev_loop *loop, ev_io *ev, int revents)
|
|
{
|
|
rdns_process_retransmit (ev->fd, ev->data);
|
|
}
|
|
|
|
static void
|
|
rdns_libev_timer_event (struct ev_loop *loop, ev_timer *ev, int revents)
|
|
{
|
|
rdns_process_timer (ev->data);
|
|
}
|
|
|
|
static void
|
|
rdns_libev_periodic_event (struct ev_loop *loop, ev_timer *ev, int revents)
|
|
{
|
|
struct rdns_ev_periodic_cbdata *cbdata = (struct rdns_ev_periodic_cbdata *)
|
|
ev->data;
|
|
cbdata->cb (cbdata->cbdata);
|
|
}
|
|
|
|
static void*
|
|
rdns_libev_add_read (void *priv_data, int fd, void *user_data)
|
|
{
|
|
ev_io *ev;
|
|
void *ptr;
|
|
|
|
ptr = malloc (sizeof (ev_io));
|
|
if (ptr != NULL) {
|
|
ev = (ev_io *)ptr;
|
|
ev_io_init (ev, rdns_libev_read_event, fd, EV_READ);
|
|
ev->data = user_data;
|
|
ev_io_start ((struct ev_loop *)priv_data, ev);
|
|
}
|
|
return ptr;
|
|
}
|
|
|
|
static void
|
|
rdns_libev_del_read (void *priv_data, void *ev_data)
|
|
{
|
|
ev_io *ev = (ev_io*)ev_data;
|
|
if (ev != NULL) {
|
|
ev_io_stop ((struct ev_loop *)priv_data, ev);
|
|
free ((void *)ev);
|
|
}
|
|
}
|
|
static void*
|
|
rdns_libev_add_write (void *priv_data, int fd, void *user_data)
|
|
{
|
|
ev_io *ev;
|
|
|
|
ev = (ev_io *)malloc (sizeof (ev_io));
|
|
if (ev != NULL) {
|
|
ev_io_init (ev, rdns_libev_write_event, fd, EV_WRITE);
|
|
ev->data = user_data;
|
|
ev_io_start ((struct ev_loop *)priv_data, ev);
|
|
}
|
|
return (void *)ev;
|
|
}
|
|
|
|
static void
|
|
rdns_libev_del_write (void *priv_data, void *ev_data)
|
|
{
|
|
ev_io *ev = (ev_io *)ev_data;
|
|
if (ev != NULL) {
|
|
ev_io_stop ((struct ev_loop *)priv_data, ev);
|
|
free ((void *)ev);
|
|
}
|
|
}
|
|
|
|
static void*
|
|
rdns_libev_add_timer (void *priv_data, double after, void *user_data)
|
|
{
|
|
ev_timer *ev;
|
|
ev = (ev_timer *)malloc (sizeof (ev_timer));
|
|
if (ev != NULL) {
|
|
ev_timer_init (ev, rdns_libev_timer_event, after, after);
|
|
ev->data = user_data;
|
|
ev_timer_start ((struct ev_loop *)priv_data, ev);
|
|
}
|
|
return (void *)ev;
|
|
}
|
|
|
|
static void*
|
|
rdns_libev_add_periodic (void *priv_data, double after,
|
|
rdns_periodic_callback cb, void *user_data)
|
|
{
|
|
ev_timer *ev;
|
|
struct rdns_ev_periodic_cbdata *cbdata = NULL;
|
|
|
|
ev = (ev_timer *)malloc (sizeof (ev_timer));
|
|
if (ev != NULL) {
|
|
cbdata = (struct rdns_ev_periodic_cbdata *)
|
|
malloc (sizeof (struct rdns_ev_periodic_cbdata));
|
|
if (cbdata != NULL) {
|
|
cbdata->cb = cb;
|
|
cbdata->cbdata = user_data;
|
|
cbdata->ev = ev;
|
|
ev_timer_init (ev, rdns_libev_periodic_event, after, after);
|
|
ev->data = cbdata;
|
|
ev_timer_start ((struct ev_loop *)priv_data, ev);
|
|
}
|
|
else {
|
|
free ((void *)ev);
|
|
return NULL;
|
|
}
|
|
}
|
|
return (void *)cbdata;
|
|
}
|
|
|
|
static void
|
|
rdns_libev_del_periodic (void *priv_data, void *ev_data)
|
|
{
|
|
struct rdns_ev_periodic_cbdata *cbdata = (struct rdns_ev_periodic_cbdata *)
|
|
ev_data;
|
|
if (cbdata != NULL) {
|
|
ev_timer_stop ((struct ev_loop *)priv_data, (ev_timer *)cbdata->ev);
|
|
free ((void *)cbdata->ev);
|
|
free ((void *)cbdata);
|
|
}
|
|
}
|
|
|
|
static void
|
|
rdns_libev_repeat_timer (void *priv_data, void *ev_data)
|
|
{
|
|
ev_timer *ev = (ev_timer *)ev_data;
|
|
if (ev != NULL) {
|
|
ev_timer_again ((struct ev_loop *)priv_data, ev);
|
|
}
|
|
}
|
|
|
|
static void
|
|
rdns_libev_del_timer (void *priv_data, void *ev_data)
|
|
{
|
|
ev_timer *ev = (ev_timer *)ev_data;
|
|
if (ev != NULL) {
|
|
ev_timer_stop ((struct ev_loop *)priv_data, ev);
|
|
free ((void *)ev);
|
|
}
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /* RDNS_EV_H_ */
|