dnshandler.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/dnshandler.h"
#include "daemon/engine.h"
#include "status.h"
#include "wire/buffer.h"
#include <errno.h>
#include <string.h>
static const char* dnsh_str = "dnshandler";
static void dnshandler_handle_xfr(netio_type* netio,
netio_handler_type* handler, netio_events_type event_types);
dnshandler_type*
dnshandler_create(listener_type* interfaces)
{
dnshandler_type* dnsh = NULL;
if (!interfaces || interfaces->count <= 0) {
return NULL;
}
CHECKALLOC(dnsh = (dnshandler_type*) malloc(sizeof(dnshandler_type)));
if (!dnsh) {
ods_log_error("[%s] unable to create dnshandler: "
"allocator_alloc() failed", dnsh_str);
return NULL;
}
dnsh->need_to_exit = 0;
dnsh->engine = NULL;
dnsh->interfaces = interfaces;
dnsh->socklist = NULL;
dnsh->netio = NULL;
dnsh->query = NULL;
dnsh->tcp_accept_handlers = NULL;
/* setup */
CHECKALLOC(dnsh->socklist = (socklist_type*) malloc(sizeof(socklist_type)));
if (!dnsh->socklist) {
ods_log_error("[%s] unable to create socklist: "
"allocator_alloc() failed", dnsh_str);
dnshandler_cleanup(dnsh);
return NULL;
}
dnsh->netio = netio_create();
if (!dnsh->netio) {
ods_log_error("[%s] unable to create dnshandler: "
"netio_create() failed", dnsh_str);
dnshandler_cleanup(dnsh);
return NULL;
}
dnsh->query = query_create();
if (!dnsh->query) {
ods_log_error("[%s] unable to create dnshandler: "
"query_create() failed", dnsh_str);
dnshandler_cleanup(dnsh);
return NULL;
}
dnsh->xfrhandler.fd = -1;
dnsh->xfrhandler.user_data = (void*) dnsh;
dnsh->xfrhandler.timeout = 0;
dnsh->xfrhandler.event_types = NETIO_EVENT_READ;
dnsh->xfrhandler.event_handler = dnshandler_handle_xfr;
return dnsh;
}
ods_status
dnshandler_listen(dnshandler_type* dnshandler)
{
ods_status status = ODS_STATUS_OK;
ods_log_assert(dnshandler);
status = sock_listen(dnshandler->socklist, dnshandler->interfaces);
if (status != ODS_STATUS_OK) {
ods_log_error("[%s] unable to start: sock_listen() "
"failed (%s)", dnsh_str, ods_status2str(status));
dnshandler->thread_id = 0;
}
return status;
}
void
dnshandler_start(dnshandler_type* dnshandler)
{
size_t i = 0;
engine_type* engine = NULL;
ods_log_assert(dnshandler);
ods_log_assert(dnshandler->engine);
ods_log_debug("[%s] start", dnsh_str);
engine = dnshandler->engine;
/* udp */
for (i=0; i < dnshandler->interfaces->count; i++) {
struct udp_data* data = NULL;
netio_handler_type* handler = NULL;
CHECKALLOC(data = (struct udp_data*) malloc(sizeof(struct udp_data)));
if (!data) {
ods_log_error("[%s] unable to start: allocator_alloc() "
"failed", dnsh_str);
dnshandler->thread_id = 0;
engine->need_to_exit = 1;
break;
}
data->query = dnshandler->query;
data->engine = dnshandler->engine;
data->socket = &dnshandler->socklist->udp[i];
CHECKALLOC(handler = (netio_handler_type*) malloc(sizeof(netio_handler_type)));
if (!handler) {
ods_log_error("[%s] unable to start: allocator_alloc() "
"failed", dnsh_str);
free(data);
dnshandler->thread_id = 0;
engine->need_to_exit = 1;
break;
}
handler->fd = dnshandler->socklist->udp[i].s;
handler->timeout = NULL;
handler->user_data = data;
handler->event_types = NETIO_EVENT_READ;
handler->event_handler = sock_handle_udp;
handler->free_handler = 1;
ods_log_debug("[%s] add udp network handler fd %u", dnsh_str,
(unsigned) handler->fd);
netio_add_handler(dnshandler->netio, handler);
}
/* tcp */
CHECKALLOC(dnshandler->tcp_accept_handlers = (netio_handler_type*) malloc(dnshandler->interfaces->count * sizeof(netio_handler_type)));
for (i=0; i < dnshandler->interfaces->count; i++) {
struct tcp_accept_data* data = NULL;
netio_handler_type* handler = NULL;
CHECKALLOC(data = (struct tcp_accept_data*) malloc(sizeof(struct tcp_accept_data)));
if (!data) {
ods_log_error("[%s] unable to start: allocator_alloc() "
"failed", dnsh_str);
dnshandler->thread_id = 0;
engine->need_to_exit = 1;
return;
}
data->engine = dnshandler->engine;
data->socket = &dnshandler->socklist->udp[i];
data->tcp_accept_handler_count = dnshandler->interfaces->count;
data->tcp_accept_handlers = dnshandler->tcp_accept_handlers;
handler = &dnshandler->tcp_accept_handlers[i];
handler->fd = dnshandler->socklist->tcp[i].s;
handler->timeout = NULL;
handler->user_data = data;
handler->event_types = NETIO_EVENT_READ;
handler->event_handler = sock_handle_tcp_accept;
handler->free_handler = 0;
ods_log_debug("[%s] add tcp network handler fd %u", dnsh_str,
(unsigned) handler->fd);
netio_add_handler(dnshandler->netio, handler);
}
/* service */
while (dnshandler->need_to_exit == 0) {
ods_log_deeebug("[%s] netio dispatch", dnsh_str);
if (netio_dispatch(dnshandler->netio, NULL, NULL) == -1) {
if (errno != EINTR) {
ods_log_error("[%s] unable to dispatch netio: %s", dnsh_str,
strerror(errno));
break;
}
}
}
/* shutdown */
ods_log_debug("[%s] shutdown", dnsh_str);
}
void
dnshandler_signal(dnshandler_type* dnshandler)
{
if (dnshandler && dnshandler->thread_id) {
janitor_thread_signal(dnshandler->thread_id);
}
}
void
dnshandler_fwd_notify(dnshandler_type* dnshandler, uint8_t* pkt, size_t len)
{
ssize_t nb = 0;
ods_log_assert(dnshandler);
ods_log_assert(pkt);
nb = send(dnshandler->xfrhandler.fd, (const void*) pkt, len, 0);
if (nb < 0) {
ods_log_error("[%s] unable to forward notify: send() failed (%s)",
dnsh_str, strerror(errno));
} else {
ods_log_debug("[%s] forwarded notify: %ld bytes sent", dnsh_str, (long)nb);
}
}
static void
dnshandler_handle_xfr(netio_type* ATTR_UNUSED(netio),
netio_handler_type* handler, netio_events_type event_types)
{
dnshandler_type* dnshandler = NULL;
uint8_t buf[MAX_PACKET_SIZE];
ssize_t received = 0;
if (!handler) {
return;
}
dnshandler = (dnshandler_type*) handler->user_data;
ods_log_assert(event_types & NETIO_EVENT_READ);
received = read(dnshandler->xfrhandler.fd, &buf, MAX_PACKET_SIZE);
ods_log_debug("[%s] read forwarded xfr packet: %d bytes received",
dnsh_str, (int) received);
if (received == -1) {
ods_log_error("[%s] unable to forward xfr packet: %s", dnsh_str,
strerror(errno));
}
}
void
dnshandler_cleanup(dnshandler_type* dnshandler)
{
size_t i = 0;
if (!dnshandler) {
return;
}
netio_cleanup(dnshandler->netio);
query_cleanup(dnshandler->query);
for (i = 0; i < dnshandler->interfaces->count; i++) {
if (dnshandler->tcp_accept_handlers)
free(dnshandler->tcp_accept_handlers[i].user_data);
if (dnshandler->socklist->udp[i].s != -1) {
close(dnshandler->socklist->udp[i].s);
freeaddrinfo((void*)dnshandler->socklist->udp[i].addr);
}
if (dnshandler->socklist->tcp[i].s != -1) {
close(dnshandler->socklist->tcp[i].s);
freeaddrinfo((void*)dnshandler->socklist->tcp[i].addr);
}
}
free(dnshandler->tcp_accept_handlers);
free(dnshandler->socklist);
free(dnshandler);
}
buffer.h
MAX_PACKET_SIZE
#define MAX_PACKET_SIZE
Definition buffer.h:49
dnshandler_signal
void dnshandler_signal(dnshandler_type *dnshandler)
Definition dnshandler.c:218
dnshandler_create
dnshandler_type * dnshandler_create(listener_type *interfaces)
Definition dnshandler.c:51
dnshandler_listen
ods_status dnshandler_listen(dnshandler_type *dnshandler)
Definition dnshandler.c:106
dnshandler_start
void dnshandler_start(dnshandler_type *dnshandler)
Definition dnshandler.c:125
dnshandler_cleanup
void dnshandler_cleanup(dnshandler_type *dnshandler)
Definition dnshandler.c:277
dnshandler_fwd_notify
void dnshandler_fwd_notify(dnshandler_type *dnshandler, uint8_t *pkt, size_t len)
Definition dnshandler.c:231
dnshandler.h
dnshandler_type
struct dnshandler_struct dnshandler_type
Definition dnshandler.h:38
engine.h
engine_type
struct engine_struct engine_type
Definition engine.h:38
listener_type
struct listener_struct listener_type
Definition listener.h:80
netio_cleanup
void netio_cleanup(netio_type *netio)
Definition netio.c:336
netio_dispatch
int netio_dispatch(netio_type *netio, const struct timespec *timeout, const sigset_t *sigmask)
Definition netio.c:187
netio_add_handler
void netio_add_handler(netio_type *netio, netio_handler_type *handler)
Definition netio.c:53
netio_create
netio_type * netio_create()
Definition netio.c:39
netio_handler_type
struct netio_handler_struct netio_handler_type
Definition netio.h:79
netio_events_type
enum netio_events_enum netio_events_type
Definition netio.h:76
NETIO_EVENT_READ
@ NETIO_EVENT_READ
Definition netio.h:71
netio_type
struct netio_struct netio_type
Definition netio.h:78
query_create
query_type * query_create(void)
Definition query.c:48
query_cleanup
void query_cleanup(query_type *q)
Definition query.c:1098
sock_handle_udp
void sock_handle_udp(netio_type *ATTR_UNUSED(netio), netio_handler_type *handler, netio_events_type event_types)
Definition sock.c:388
sock_handle_tcp_accept
void sock_handle_tcp_accept(netio_type *netio, netio_handler_type *handler, netio_events_type event_types)
Definition sock.c:445
sock_listen
ods_status sock_listen(socklist_type *sockets, listener_type *listener)
Definition sock.c:292
socklist_type
struct socklist_struct socklist_type
Definition sock.h:55
dnshandler_struct::xfrhandler
netio_handler_type xfrhandler
Definition dnshandler.h:58
dnshandler_struct::tcp_accept_handlers
netio_handler_type * tcp_accept_handlers
Definition dnshandler.h:60
dnshandler_struct::thread_id
janitor_thread_t thread_id
Definition dnshandler.h:52
dnshandler_struct::engine
engine_type * engine
Definition dnshandler.h:53
dnshandler_struct::socklist
socklist_type * socklist
Definition dnshandler.h:55
dnshandler_struct::need_to_exit
unsigned need_to_exit
Definition dnshandler.h:59
dnshandler_struct::interfaces
listener_type * interfaces
Definition dnshandler.h:54
dnshandler_struct::netio
netio_type * netio
Definition dnshandler.h:56
dnshandler_struct::query
query_type * query
Definition dnshandler.h:57
listener_struct::count
size_t count
Definition listener.h:83
netio_handler_struct::timeout
struct timespec * timeout
Definition netio.h:115
netio_handler_struct::free_handler
int free_handler
Definition netio.h:132
netio_handler_struct::fd
int fd
Definition netio.h:108
netio_handler_struct::event_types
netio_events_type event_types
Definition netio.h:124
netio_handler_struct::event_handler
netio_event_handler_type event_handler
Definition netio.h:131
netio_handler_struct::user_data
void * user_data
Definition netio.h:119
sock_struct::addr
struct addrinfo * addr
Definition sock.h:47
sock_struct::s
int s
Definition sock.h:48
socklist_struct::udp
sock_type udp[MAX_INTERFACES]
Definition sock.h:58
socklist_struct::tcp
sock_type tcp[MAX_INTERFACES]
Definition sock.h:57
tcp_accept_data
Definition sock.h:75
tcp_accept_data::tcp_accept_handlers
netio_handler_type * tcp_accept_handlers
Definition sock.h:79
tcp_accept_data::socket
sock_type * socket
Definition sock.h:77
tcp_accept_data::tcp_accept_handler_count
size_t tcp_accept_handler_count
Definition sock.h:78
tcp_accept_data::engine
engine_type * engine
Definition sock.h:76
udp_data
Definition sock.h:65
udp_data::query
query_type * query
Definition sock.h:68
udp_data::engine
engine_type * engine
Definition sock.h:66
udp_data::socket
sock_type * socket
Definition sock.h:67