xfrhandler.c
Go to the documentation of this file.
/*
* Copyright (c) 2009 NLNet Labs. All rights reserved.
*
* 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.
*
* THIS SOFTWARE IS PROVIDED BY THE 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 THE 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.
*
*/
#include "config.h"
#include "daemon/engine.h"
#include "daemon/xfrhandler.h"
#include "duration.h"
#include "status.h"
#include <errno.h>
#include <string.h>
static const char* xfrh_str = "xfrhandler";
static void xfrhandler_handle_dns(netio_type* netio,
netio_handler_type* handler, netio_events_type event_types);
xfrhandler_type*
xfrhandler_create()
{
xfrhandler_type* xfrh = NULL;
CHECKALLOC(xfrh = (xfrhandler_type*) malloc(sizeof(xfrhandler_type)));
xfrh->engine = NULL;
xfrh->packet = NULL;
xfrh->netio = NULL;
xfrh->tcp_set = NULL;
xfrh->tcp_waiting_first = NULL;
xfrh->udp_waiting_first = NULL;
xfrh->udp_waiting_last = NULL;
xfrh->udp_use_num = 0;
xfrh->start_time = 0;
xfrh->current_time = 0;
xfrh->got_time = 0;
xfrh->need_to_exit = 0;
xfrh->started = 0;
/* notify */
xfrh->notify_waiting_first = NULL;
xfrh->notify_waiting_last = NULL;
xfrh->notify_udp_num = 0;
/* setup */
xfrh->netio = netio_create();
if (!xfrh->netio) {
ods_log_error("[%s] unable to create xfrhandler: "
"netio_create() failed", xfrh_str);
xfrhandler_cleanup(xfrh);
return NULL;
}
xfrh->packet = buffer_create(PACKET_BUFFER_SIZE);
if (!xfrh->packet) {
ods_log_error("[%s] unable to create xfrhandler: "
"buffer_create() failed", xfrh_str);
xfrhandler_cleanup(xfrh);
return NULL;
}
xfrh->tcp_set = tcp_set_create();
if (!xfrh->tcp_set) {
ods_log_error("[%s] unable to create xfrhandler: "
"tcp_set_create() failed", xfrh_str);
xfrhandler_cleanup(xfrh);
return NULL;
}
xfrh->dnshandler.fd = -1;
xfrh->dnshandler.user_data = (void*) xfrh;
xfrh->dnshandler.timeout = 0;
xfrh->dnshandler.event_types = NETIO_EVENT_READ;
xfrh->dnshandler.event_handler = xfrhandler_handle_dns;
xfrh->dnshandler.free_handler = 0;
return xfrh;
}
void
xfrhandler_start(xfrhandler_type* xfrhandler)
{
ods_log_assert(xfrhandler);
ods_log_assert(xfrhandler->engine);
ods_log_debug("[%s] start", xfrh_str);
/* setup */
xfrhandler->start_time = time_now();
/* handlers */
netio_add_handler(xfrhandler->netio, &xfrhandler->dnshandler);
/* service */
while (xfrhandler->need_to_exit == 0) {
/* dispatch may block for a longer period, so current is gone */
xfrhandler->got_time = 0;
ods_log_deeebug("[%s] netio dispatch", xfrh_str);
if (netio_dispatch(xfrhandler->netio, NULL, NULL) == -1) {
if (errno != EINTR) {
ods_log_error("[%s] unable to dispatch netio: %s", xfrh_str,
strerror(errno));
}
}
}
/* shutdown */
ods_log_debug("[%s] shutdown", xfrh_str);
}
time_t
xfrhandler_time(xfrhandler_type* xfrhandler)
{
if (!xfrhandler) {
return 0;
}
if (!xfrhandler->got_time) {
xfrhandler->current_time = time_now();
xfrhandler->got_time = 1;
}
return xfrhandler->current_time;
}
void
xfrhandler_signal(xfrhandler_type* xfrhandler)
{
if (xfrhandler && xfrhandler->started) {
janitor_thread_signal(xfrhandler->thread_id);
}
}
static void
xfrhandler_handle_dns(netio_type* ATTR_UNUSED(netio),
netio_handler_type* handler, netio_events_type event_types)
{
xfrhandler_type* xfrhandler = NULL;
uint8_t buf[MAX_PACKET_SIZE];
ssize_t received = 0;
if (!handler) {
return;
}
xfrhandler = (xfrhandler_type*) handler->user_data;
ods_log_assert(event_types & NETIO_EVENT_READ);
received = read(xfrhandler->dnshandler.fd, &buf, MAX_PACKET_SIZE);
ods_log_debug("[%s] read forwarded dns packet: %d bytes received",
xfrh_str, (int) received);
if (received == -1) {
ods_log_error("[%s] unable to forward dns packet: %s", xfrh_str,
strerror(errno));
}
}
void
xfrhandler_cleanup(xfrhandler_type* xfrhandler)
{
if (!xfrhandler) {
return;
}
netio_cleanup_shallow(xfrhandler->netio);
buffer_cleanup(xfrhandler->packet);
tcp_set_cleanup(xfrhandler->tcp_set);
free(xfrhandler);
}
netio_cleanup_shallow
void netio_cleanup_shallow(netio_type *netio)
Definition: netio.c:355
xfrhandler_struct::udp_waiting_first
xfrd_type * udp_waiting_first
Definition: xfrhandler.h:64
xfrhandler_struct::udp_waiting_last
xfrd_type * udp_waiting_last
Definition: xfrhandler.h:65
engine.h
xfrhandler_struct::current_time
time_t current_time
Definition: xfrhandler.h:58
xfrhandler_struct::tcp_waiting_first
xfrd_type * tcp_waiting_first
Definition: xfrhandler.h:63
xfrhandler_cleanup
void xfrhandler_cleanup(xfrhandler_type *xfrhandler)
Definition: xfrhandler.c:198
xfrhandler_struct::packet
buffer_type * packet
Definition: xfrhandler.h:62
xfrhandler_struct::netio
netio_type * netio
Definition: xfrhandler.h:60
xfrhandler_struct::thread_id
janitor_thread_t thread_id
Definition: xfrhandler.h:54
xfrhandler_struct::need_to_exit
unsigned need_to_exit
Definition: xfrhandler.h:72
xfrhandler.h
xfrhandler_struct::engine
engine_type * engine
Definition: xfrhandler.h:55
xfrhandler_struct
Definition: xfrhandler.h:52
netio_handler_struct::free_handler
int free_handler
Definition: netio.h:132
netio_events_type
enum netio_events_enum netio_events_type
Definition: netio.h:76
xfrhandler_struct::dnshandler
netio_handler_type dnshandler
Definition: xfrhandler.h:70
netio_handler_struct::user_data
void * user_data
Definition: netio.h:119
netio_handler_struct::fd
int fd
Definition: netio.h:108
netio_add_handler
void netio_add_handler(netio_type *netio, netio_handler_type *handler)
Definition: netio.c:53
xfrhandler_struct::got_time
unsigned got_time
Definition: xfrhandler.h:71
xfrhandler_time
time_t xfrhandler_time(xfrhandler_type *xfrhandler)
Definition: xfrhandler.c:141
xfrhandler_struct::udp_use_num
size_t udp_use_num
Definition: xfrhandler.h:66
buffer_cleanup
void buffer_cleanup(buffer_type *buffer)
Definition: buffer.c:1145
xfrhandler_create
xfrhandler_type * xfrhandler_create()
Definition: xfrhandler.c:52
xfrhandler_start
void xfrhandler_start(xfrhandler_type *xfrhandler)
Definition: xfrhandler.c:110
xfrhandler_struct::notify_waiting_first
notify_type * notify_waiting_first
Definition: xfrhandler.h:67
netio_handler_struct::event_handler
netio_event_handler_type event_handler
Definition: netio.h:131
NETIO_EVENT_READ
Definition: netio.h:71
xfrhandler_struct::tcp_set
tcp_set_type * tcp_set
Definition: xfrhandler.h:61
netio_handler_struct
Definition: netio.h:102
xfrhandler_struct::notify_waiting_last
notify_type * notify_waiting_last
Definition: xfrhandler.h:68
netio_struct
Definition: netio.h:139
xfrhandler_struct::start_time
time_t start_time
Definition: xfrhandler.h:57
tcp_set_cleanup
void tcp_set_cleanup(tcp_set_type *set)
Definition: tcpset.c:242
xfrhandler_signal
void xfrhandler_signal(xfrhandler_type *xfrhandler)
Definition: xfrhandler.c:159
MAX_PACKET_SIZE
#define MAX_PACKET_SIZE
Definition: buffer.h:49
PACKET_BUFFER_SIZE
#define PACKET_BUFFER_SIZE
Definition: buffer.h:50
xfrhandler_struct::notify_udp_num
int notify_udp_num
Definition: xfrhandler.h:69
netio_handler_struct::event_types
netio_events_type event_types
Definition: netio.h:124
netio_create
netio_type * netio_create()
Definition: netio.c:39
xfrhandler_struct::started
unsigned started
Definition: xfrhandler.h:73
netio_handler_struct::timeout
struct timespec * timeout
Definition: netio.h:115
tcp_set_create
tcp_set_type * tcp_set_create()
Definition: tcpset.c:67
buffer_create
buffer_type * buffer_create(size_t capacity)
Definition: buffer.c:78
netio_dispatch
int netio_dispatch(netio_type *netio, const struct timespec *timeout, const sigset_t *sigmask)
Definition: netio.c:187