diff options
author | Jakob Schlyter <jakob@cvs.openbsd.org> | 2010-01-15 19:24:59 +0000 |
---|---|---|
committer | Jakob Schlyter <jakob@cvs.openbsd.org> | 2010-01-15 19:24:59 +0000 |
commit | 54a1a10c8035ff9ec7b252904e0675a1ac7f2d15 (patch) | |
tree | 10b07cafeb640e396c3d65c26ff7652109393eab | |
parent | 81db87de733ad83093fe047ea7411bc8afaa1db0 (diff) |
NSD v3.2.4
-rw-r--r-- | usr.sbin/nsd/ipc.c | 635 | ||||
-rw-r--r-- | usr.sbin/nsd/xfrd-disk.h | 11 |
2 files changed, 342 insertions, 304 deletions
diff --git a/usr.sbin/nsd/ipc.c b/usr.sbin/nsd/ipc.c index 1f188e54951..449c0514fc1 100644 --- a/usr.sbin/nsd/ipc.c +++ b/usr.sbin/nsd/ipc.c @@ -7,11 +7,10 @@ * */ -#include "config.h" +#include <config.h> #include <errno.h> #include <unistd.h> #include <stdlib.h> -#include <fcntl.h> #include "ipc.h" #include "buffer.h" #include "xfrd-tcp.h" @@ -19,46 +18,92 @@ #include "namedb.h" #include "xfrd.h" #include "xfrd-notify.h" -#include "difffile.h" +/* set is_ok for the zone according to the zone message */ +static zone_type* handle_xfrd_zone_state(struct nsd* nsd, buffer_type* packet); +/* write ipc ZONE_STATE message into the buffer */ +static void write_zone_state_packet(buffer_type* packet, zone_type* zone); /* attempt to send NSD_STATS command to child fd */ static void send_stat_to_child(struct main_ipc_handler_data* data, int fd); +/* write IPC expire notification msg to a buffer */ +static void xfrd_write_expire_notification(buffer_type* buffer, xfrd_zone_t* zone); /* send reload request over the IPC channel */ static void xfrd_send_reload_req(xfrd_state_t* xfrd); /* send quit request over the IPC channel */ static void xfrd_send_quit_req(xfrd_state_t* xfrd); +/* get SOA INFO out of IPC packet buffer */ +static void xfrd_handle_ipc_SOAINFO(xfrd_state_t* xfrd, buffer_type* packet); /* perform read part of handle ipc for xfrd */ -static void xfrd_handle_ipc_read(struct event* handler, xfrd_state_t* xfrd); +static void xfrd_handle_ipc_read(netio_handler_type *handler, xfrd_state_t* xfrd); -static void -ipc_child_quit(struct nsd* nsd) +static zone_type* +handle_xfrd_zone_state(struct nsd* nsd, buffer_type* packet) { - /* call shutdown and quit routines */ - nsd->mode = NSD_QUIT; -#ifdef BIND8_STATS - bind8_stats(nsd); -#endif /* BIND8_STATS */ - -#if 0 /* OS collects memory pages */ - event_base_free(event_base); - region_destroy(server_region); -#endif - server_shutdown(nsd); - exit(0); + uint8_t ok; + const dname_type *dname; + domain_type *domain; + zone_type *zone; + + ok = buffer_read_u8(packet); + dname = (dname_type*)buffer_current(packet); + DEBUG(DEBUG_IPC,1, (LOG_INFO, "handler zone state %s is %s", + dname_to_string(dname, NULL), ok?"ok":"expired")); + /* find in zone_types, if does not exist, we cannot serve anyway */ + /* find zone in config, since that one always exists */ + domain = domain_table_find(nsd->db->domains, dname); + if(!domain) { + DEBUG(DEBUG_IPC,1, (LOG_INFO, "zone state msg, empty zone (domain %s)", + dname_to_string(dname, NULL))); + return NULL; + } + zone = domain_find_zone(domain); + if(!zone || dname_compare(domain_dname(zone->apex), dname) != 0) { + DEBUG(DEBUG_IPC,1, (LOG_INFO, "zone state msg, empty zone (zone %s)", + dname_to_string(dname, NULL))); + return NULL; + } + assert(zone); + /* only update zone->is_ok if needed to minimize copy-on-write + of memory pages shared after fork() */ + if(ok && !zone->is_ok) + zone->is_ok = 1; + if(!ok && zone->is_ok) + zone->is_ok = 0; + return zone; } void -child_handle_parent_command(int fd, short event, void* arg) +child_handle_parent_command(netio_type *ATTR_UNUSED(netio), + netio_handler_type *handler, + netio_event_types_type event_types) { sig_atomic_t mode; int len; struct ipc_handler_conn_data *data = - (struct ipc_handler_conn_data *) arg; - if (!(event & EV_READ)) { + (struct ipc_handler_conn_data *) handler->user_data; + if (!(event_types & NETIO_EVENT_READ)) { return; } - if ((len = read(fd, &mode, sizeof(mode))) == -1) { + if(data->conn->is_reading) { + int ret = conn_read(data->conn); + if(ret == -1) { + log_msg(LOG_ERR, "handle_parent_command: error in conn_read: %s", + strerror(errno)); + data->conn->is_reading = 0; + return; + } + if(ret == 0) { + return; /* continue later */ + } + /* completed */ + data->conn->is_reading = 0; + buffer_flip(data->conn->packet); + (void)handle_xfrd_zone_state(data->nsd, data->conn->packet); + return; + } + + if ((len = read(handler->fd, &mode, sizeof(mode))) == -1) { log_msg(LOG_ERR, "handle_parent_command: read: %s", strerror(errno)); return; @@ -66,38 +111,21 @@ child_handle_parent_command(int fd, short event, void* arg) if (len == 0) { /* parent closed the connection. Quit */ - ipc_child_quit(data->nsd); + data->nsd->mode = NSD_QUIT; return; } switch (mode) { case NSD_STATS: - data->nsd->mode = mode; - break; case NSD_QUIT: - ipc_child_quit(data->nsd); + data->nsd->mode = mode; break; - case NSD_QUIT_CHILD: - /* close our listening sockets and ack */ - server_close_all_sockets(data->nsd->udp, data->nsd->ifs); - server_close_all_sockets(data->nsd->tcp, data->nsd->ifs); - /* mode == NSD_QUIT_CHILD */ - (void)write(fd, &mode, sizeof(mode)); - ipc_child_quit(data->nsd); - break; - case NSD_QUIT_WITH_STATS: -#ifdef BIND8_STATS - DEBUG(DEBUG_IPC, 2, (LOG_INFO, "quit QUIT_WITH_STATS")); - /* reply with ack and stats and then quit */ - if(!write_socket(fd, &mode, sizeof(mode))) { - log_msg(LOG_ERR, "cannot write quitwst to parent"); - } - if(!write_socket(fd, &data->nsd->st, sizeof(data->nsd->st))) { - log_msg(LOG_ERR, "cannot write stats to parent"); - } - fsync(fd); -#endif /* BIND8_STATS */ - ipc_child_quit(data->nsd); + case NSD_ZONE_STATE: + data->conn->is_reading = 1; + data->conn->total_bytes = 0; + data->conn->msglen = 0; + data->conn->fd = handler->fd; + buffer_clear(data->conn->packet); break; default: log_msg(LOG_ERR, "handle_parent_command: bad mode %d", @@ -119,6 +147,38 @@ parent_handle_xfrd_command(netio_type *ATTR_UNUSED(netio), return; } + if(data->conn->is_reading) { + /* handle ZONE_STATE forward to children */ + int ret = conn_read(data->conn); + size_t i; + zone_type* zone; + if(ret == -1) { + log_msg(LOG_ERR, "main xfrd listener: error in conn_read: %s", + strerror(errno)); + data->conn->is_reading = 0; + return; + } + if(ret == 0) { + return; /* continue later */ + } + /* completed */ + data->conn->is_reading = 0; + buffer_flip(data->conn->packet); + zone = handle_xfrd_zone_state(data->nsd, data->conn->packet); + if(!zone) + return; + /* forward to all children */ + for (i = 0; i < data->nsd->child_count; ++i) { + if(!zone->dirty[i]) { + zone->dirty[i] = 1; + stack_push(data->nsd->children[i].dirty_zones, zone); + data->nsd->children[i].handler->event_types |= + NETIO_EVENT_WRITE; + } + } + return; + } + if ((len = read(handler->fd, &mode, sizeof(mode))) == -1) { log_msg(LOG_ERR, "handle_xfrd_command: read: %s", strerror(errno)); @@ -126,29 +186,29 @@ parent_handle_xfrd_command(netio_type *ATTR_UNUSED(netio), } if (len == 0) { - /* xfrd closed, we must quit */ - DEBUG(DEBUG_IPC,1, (LOG_INFO, "handle_xfrd_command: xfrd closed channel.")); + DEBUG(DEBUG_IPC,1, (LOG_ERR, "handle_xfrd_command: xfrd closed channel.")); close(handler->fd); handler->fd = -1; - data->nsd->mode = NSD_SHUTDOWN; return; } switch (mode) { case NSD_RELOAD: - DEBUG(DEBUG_IPC,1, (LOG_INFO, "parent handle xfrd command RELOAD")); data->nsd->signal_hint_reload = 1; break; case NSD_QUIT: - case NSD_SHUTDOWN: data->nsd->mode = mode; break; - case NSD_STATS: - data->nsd->signal_hint_stats = 1; - break; case NSD_REAP_CHILDREN: data->nsd->signal_hint_child = 1; break; + case NSD_ZONE_STATE: + data->conn->is_reading = 1; + data->conn->total_bytes = 0; + data->conn->msglen = 0; + data->conn->fd = handler->fd; + buffer_clear(data->conn->packet); + break; default: log_msg(LOG_ERR, "handle_xfrd_command: bad mode %d", (int) mode); @@ -157,6 +217,27 @@ parent_handle_xfrd_command(netio_type *ATTR_UNUSED(netio), } static void +write_zone_state_packet(buffer_type* packet, zone_type* zone) +{ + sig_atomic_t cmd = NSD_ZONE_STATE; + uint8_t ok = zone->is_ok; + uint16_t sz; + if(!zone->apex) { + return; + } + sz = dname_total_size(domain_dname(zone->apex)) + 1; + sz = htons(sz); + + buffer_clear(packet); + buffer_write(packet, &cmd, sizeof(cmd)); + buffer_write(packet, &sz, sizeof(sz)); + buffer_write(packet, &ok, sizeof(ok)); + buffer_write(packet, domain_dname(zone->apex), + dname_total_size(domain_dname(zone->apex))); + buffer_flip(packet); +} + +static void send_stat_to_child(struct main_ipc_handler_data* data, int fd) { sig_atomic_t cmd = NSD_STATS; @@ -170,7 +251,6 @@ send_stat_to_child(struct main_ipc_handler_data* data, int fd) data->child->need_to_send_STATS = 0; } -#ifndef NDEBUG int packet_read_query_section(buffer_type *packet, uint8_t* dest, uint16_t* qtype, uint16_t* qclass); static void debug_print_fwd_name(int ATTR_UNUSED(len), buffer_type* packet, int acl_num) @@ -193,16 +273,11 @@ debug_print_fwd_name(int ATTR_UNUSED(len), buffer_type* packet, int acl_num) buffer_set_position(packet, bufpos); region_destroy(tempregion); } -#endif static void send_quit_to_child(struct main_ipc_handler_data* data, int fd) { -#ifdef BIND8_STATS - sig_atomic_t cmd = NSD_QUIT_WITH_STATS; -#else sig_atomic_t cmd = NSD_QUIT; -#endif if(write(fd, &cmd, sizeof(cmd)) == -1) { if(errno == EAGAIN || errno == EINTR) return; /* try again later */ @@ -215,116 +290,6 @@ send_quit_to_child(struct main_ipc_handler_data* data, int fd) (int)data->child->pid)); } -/** the child is done, mark it as exited */ -static void -child_is_done(struct nsd* nsd, int fd) -{ - size_t i; - if(fd != -1) close(fd); - for(i=0; i<nsd->child_count; ++i) - if(nsd->children[i].child_fd == fd) { - nsd->children[i].child_fd = -1; - nsd->children[i].handler->fd = -1; - if(nsd->children[i].need_to_exit) { - DEBUG(DEBUG_IPC,1, (LOG_INFO, "server %d is done", - (int)nsd->children[i].pid)); - nsd->children[i].has_exited = 1; - } else { - log_msg(LOG_WARNING, - "server %d died unexpectedly, restarting", - (int)nsd->children[i].pid); - /* this child is now going to be re-forked as - * a subprocess of this server-main, and if a - * reload is in progress the other children - * are subprocesses of reload. Until the - * reload is done and they are all reforked. */ - nsd->children[i].pid = -1; - nsd->restart_children = 1; - } - } - parent_check_all_children_exited(nsd); -} - -#ifdef BIND8_STATS -/** add stats to total */ -void -stats_add(struct nsdst* total, struct nsdst* s) -{ - unsigned i; - for(i=0; i<sizeof(total->qtype)/sizeof(stc_t); i++) - total->qtype[i] += s->qtype[i]; - for(i=0; i<sizeof(total->qclass)/sizeof(stc_t); i++) - total->qclass[i] += s->qclass[i]; - total->qudp += s->qudp; - total->qudp6 += s->qudp6; - total->ctcp += s->ctcp; - total->ctcp6 += s->ctcp6; - for(i=0; i<sizeof(total->rcode)/sizeof(stc_t); i++) - total->rcode[i] += s->rcode[i]; - for(i=0; i<sizeof(total->opcode)/sizeof(stc_t); i++) - total->opcode[i] += s->opcode[i]; - total->dropped += s->dropped; - total->truncated += s->truncated; - total->wrongzone += s->wrongzone; - total->txerr += s->txerr; - total->rxerr += s->rxerr; - total->edns += s->edns; - total->ednserr += s->ednserr; - total->raxfr += s->raxfr; - total->nona += s->nona; - - total->db_disk = s->db_disk; - total->db_mem = s->db_mem; -} - -/** subtract stats from total */ -void -stats_subtract(struct nsdst* total, struct nsdst* s) -{ - unsigned i; - for(i=0; i<sizeof(total->qtype)/sizeof(stc_t); i++) - total->qtype[i] -= s->qtype[i]; - for(i=0; i<sizeof(total->qclass)/sizeof(stc_t); i++) - total->qclass[i] -= s->qclass[i]; - total->qudp -= s->qudp; - total->qudp6 -= s->qudp6; - total->ctcp -= s->ctcp; - total->ctcp6 -= s->ctcp6; - for(i=0; i<sizeof(total->rcode)/sizeof(stc_t); i++) - total->rcode[i] -= s->rcode[i]; - for(i=0; i<sizeof(total->opcode)/sizeof(stc_t); i++) - total->opcode[i] -= s->opcode[i]; - total->dropped -= s->dropped; - total->truncated -= s->truncated; - total->wrongzone -= s->wrongzone; - total->txerr -= s->txerr; - total->rxerr -= s->rxerr; - total->edns -= s->edns; - total->ednserr -= s->ednserr; - total->raxfr -= s->raxfr; - total->nona -= s->nona; -} - -#define FINAL_STATS_TIMEOUT 10 /* seconds */ -static void -read_child_stats(struct nsd* nsd, struct nsd_child* child, int fd) -{ - struct nsdst s; - errno=0; - if(block_read(nsd, fd, &s, sizeof(s), FINAL_STATS_TIMEOUT)!=sizeof(s)) { - log_msg(LOG_ERR, "problems reading finalstats from server " - "%d: %s", (int)child->pid, strerror(errno)); - } else { - stats_add(&nsd->st, &s); - child->query_count = s.qudp + s.qudp6 + s.ctcp + s.ctcp6; - /* we know that the child is going to close the connection - * now (this is an ACK of the QUIT_W_STATS so we know the - * child is done, no longer sending e.g. NOTIFY contents) */ - child_is_done(nsd, fd); - } -} -#endif /* BIND8_STATS */ - void parent_handle_child_command(netio_type *ATTR_UNUSED(netio), netio_handler_type *handler, @@ -337,14 +302,43 @@ parent_handle_child_command(netio_type *ATTR_UNUSED(netio), /* do a nonblocking write to the child if it is ready. */ if (event_types & NETIO_EVENT_WRITE) { - if(data->child->need_to_send_STATS && - !data->child->need_to_exit) { + if(!data->busy_writing_zone_state && + !data->child->need_to_send_STATS && + !data->child->need_to_send_QUIT && + !data->child->need_to_exit && + data->child->dirty_zones->num > 0) { + /* create packet from next dirty zone */ + zone_type* zone = (zone_type*)stack_pop(data->child->dirty_zones); + assert(zone); + zone->dirty[data->child_num] = 0; + data->busy_writing_zone_state = 1; + write_zone_state_packet(data->write_conn->packet, zone); + data->write_conn->msglen = buffer_limit(data->write_conn->packet); + data->write_conn->total_bytes = sizeof(uint16_t); /* len bytes already in packet */ + data->write_conn->fd = handler->fd; + } + if(data->busy_writing_zone_state) { + /* write more of packet */ + int ret = conn_write(data->write_conn); + if(ret == -1) { + log_msg(LOG_ERR, "handle_child_cmd %d: could not write: %s", + (int)data->child->pid, strerror(errno)); + data->busy_writing_zone_state = 0; + } else if(ret == 1) { + data->busy_writing_zone_state = 0; /* completed */ + } + } else if(data->child->need_to_send_STATS && + !data->child->need_to_exit) { send_stat_to_child(data, handler->fd); } else if(data->child->need_to_send_QUIT) { send_quit_to_child(data, handler->fd); if(!data->child->need_to_send_QUIT) handler->event_types = NETIO_EVENT_READ; - } else { + } + if(!data->busy_writing_zone_state && + !data->child->need_to_send_STATS && + !data->child->need_to_send_QUIT && + data->child->dirty_zones->num == 0) { handler->event_types = NETIO_EVENT_READ; } } @@ -402,10 +396,10 @@ parent_handle_child_command(netio_type *ATTR_UNUSED(netio), /* read rest later */ return; } - /* read the acl numbers */ + /* read the acl number */ got_acl = data->got_bytes - sizeof(data->total_bytes) - data->total_bytes; if((len = read(handler->fd, (char*)&data->acl_num+got_acl, - sizeof(data->acl_num)+sizeof(data->acl_xfr)-got_acl)) == -1 ) { + sizeof(data->acl_num)-got_acl)) == -1 ) { log_msg(LOG_ERR, "handle_child_command: read: %s", strerror(errno)); return; @@ -417,7 +411,7 @@ parent_handle_child_command(netio_type *ATTR_UNUSED(netio), } got_acl += len; data->got_bytes += len; - if(got_acl >= (int)(sizeof(data->acl_num)+sizeof(data->acl_xfr))) { + if(got_acl >= (int)sizeof(data->acl_num)) { uint16_t len = htons(data->total_bytes); DEBUG(DEBUG_IPC,2, (LOG_INFO, "main fwd passed packet write %d", (int)data->got_bytes)); @@ -432,9 +426,7 @@ parent_handle_child_command(netio_type *ATTR_UNUSED(netio), !write_socket(*data->xfrd_sock, buffer_begin(data->packet), data->total_bytes) || !write_socket(*data->xfrd_sock, &data->acl_num, - sizeof(data->acl_num)) || - !write_socket(*data->xfrd_sock, &data->acl_xfr, - sizeof(data->acl_xfr))) { + sizeof(data->acl_num))) { log_msg(LOG_ERR, "error in ipc fwd main2xfrd: %s", strerror(errno)); } @@ -450,7 +442,18 @@ parent_handle_child_command(netio_type *ATTR_UNUSED(netio), } if (len == 0) { - child_is_done(data->nsd, handler->fd); + size_t i; + if(handler->fd > 0) close(handler->fd); + for(i=0; i<data->nsd->child_count; ++i) + if(data->nsd->children[i].child_fd == handler->fd) { + data->nsd->children[i].child_fd = -1; + data->nsd->children[i].has_exited = 1; + DEBUG(DEBUG_IPC,1, (LOG_INFO, + "server %d closed cmd channel", + (int) data->nsd->children[i].pid)); + } + handler->fd = -1; + parent_check_all_children_exited(data->nsd); return; } @@ -458,11 +461,6 @@ parent_handle_child_command(netio_type *ATTR_UNUSED(netio), case NSD_QUIT: data->nsd->mode = mode; break; -#ifdef BIND8_STATS - case NSD_QUIT_WITH_STATS: - read_child_stats(data->nsd, data->child, handler->fd); - break; -#endif /* BIND8_STATS */ case NSD_STATS: data->nsd->signal_hint_stats = 1; break; @@ -516,12 +514,11 @@ parent_handle_reload_command(netio_type *ATTR_UNUSED(netio), } if (len == 0) { - if(handler->fd != -1) { + if(handler->fd > 0) { close(handler->fd); handler->fd = -1; } log_msg(LOG_ERR, "handle_reload_cmd: reload closed cmd channel"); - nsd->reload_failed = 1; return; } switch (mode) { @@ -531,7 +528,7 @@ parent_handle_reload_command(netio_type *ATTR_UNUSED(netio), for(i=0; i < nsd->child_count; i++) { nsd->children[i].need_to_exit = 1; if(nsd->children[i].pid > 0 && - nsd->children[i].child_fd != -1) { + nsd->children[i].child_fd > 0) { nsd->children[i].need_to_send_QUIT = 1; nsd->children[i].handler->event_types |= NETIO_EVENT_WRITE; @@ -550,16 +547,33 @@ parent_handle_reload_command(netio_type *ATTR_UNUSED(netio), } static void +xfrd_write_expire_notification(buffer_type* buffer, xfrd_zone_t* zone) +{ + sig_atomic_t cmd = NSD_ZONE_STATE; + uint8_t ok = 1; + uint16_t sz = dname_total_size(zone->apex) + 1; + sz = htons(sz); + if(zone->state == xfrd_zone_expired) + ok = 0; + + DEBUG(DEBUG_IPC,1, (LOG_INFO, + "xfrd encoding ipc zone state msg for zone %s state %d.", + zone->apex_str, (int)zone->state)); + + buffer_clear(buffer); + buffer_write(buffer, &cmd, sizeof(cmd)); + buffer_write(buffer, &sz, sizeof(sz)); + buffer_write(buffer, &ok, sizeof(ok)); + buffer_write(buffer, zone->apex, dname_total_size(zone->apex)); + buffer_flip(buffer); +} + +static void xfrd_send_reload_req(xfrd_state_t* xfrd) { sig_atomic_t req = NSD_RELOAD; - uint64_t p = xfrd->last_task->data; - udb_ptr_unlink(xfrd->last_task, xfrd->nsd->task[xfrd->nsd->mytask]); - task_process_sync(xfrd->nsd->task[xfrd->nsd->mytask]); /* ask server_main for a reload */ - if(write(xfrd->ipc_handler.ev_fd, &req, sizeof(req)) == -1) { - udb_ptr_init(xfrd->last_task, xfrd->nsd->task[xfrd->nsd->mytask]); - udb_ptr_set(xfrd->last_task, xfrd->nsd->task[xfrd->nsd->mytask], p); + if(write(xfrd->ipc_handler.fd, &req, sizeof(req)) == -1) { if(errno == EAGAIN || errno == EINTR) return; /* try again later */ log_msg(LOG_ERR, "xfrd: problems sending reload command: %s", @@ -567,55 +581,21 @@ xfrd_send_reload_req(xfrd_state_t* xfrd) return; } DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: asked nsd to reload new updates")); - /* swapped task to other side, start to use other task udb. */ - xfrd->nsd->mytask = 1 - xfrd->nsd->mytask; - task_remap(xfrd->nsd->task[xfrd->nsd->mytask]); - udb_ptr_init(xfrd->last_task, xfrd->nsd->task[xfrd->nsd->mytask]); - assert(udb_base_get_userdata(xfrd->nsd->task[xfrd->nsd->mytask])->data == 0); - xfrd_prepare_zones_for_reload(); xfrd->reload_cmd_last_sent = xfrd_time(); xfrd->need_to_send_reload = 0; xfrd->can_send_reload = 0; } -void -ipc_xfrd_set_listening(struct xfrd_state* xfrd, short mode) -{ - int fd = xfrd->ipc_handler.ev_fd; - struct event_base* base = xfrd->event_base; - event_del(&xfrd->ipc_handler); - event_set(&xfrd->ipc_handler, fd, mode, xfrd_handle_ipc, xfrd); - if(event_base_set(base, &xfrd->ipc_handler) != 0) - log_msg(LOG_ERR, "ipc: cannot set event_base"); - /* no timeout for IPC events */ - if(event_add(&xfrd->ipc_handler, NULL) != 0) - log_msg(LOG_ERR, "ipc: cannot add event"); - xfrd->ipc_handler_flags = mode; -} - -static void -xfrd_send_shutdown_req(xfrd_state_t* xfrd) -{ - sig_atomic_t cmd = NSD_SHUTDOWN; - xfrd->ipc_send_blocked = 1; - ipc_xfrd_set_listening(xfrd, EV_PERSIST|EV_READ); - DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: ipc send shutdown")); - if(!write_socket(xfrd->ipc_handler.ev_fd, &cmd, sizeof(cmd))) { - log_msg(LOG_ERR, "xfrd: error writing shutdown to main: %s", - strerror(errno)); - } - xfrd->need_to_send_shutdown = 0; -} - static void xfrd_send_quit_req(xfrd_state_t* xfrd) { sig_atomic_t cmd = NSD_QUIT; xfrd->ipc_send_blocked = 1; - ipc_xfrd_set_listening(xfrd, EV_PERSIST|EV_READ); + xfrd->ipc_handler.event_types &= (~NETIO_EVENT_WRITE); + xfrd->sending_zone_state = 0; DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: ipc send ackreload(quit)")); - if(!write_socket(xfrd->ipc_handler.ev_fd, &cmd, sizeof(cmd))) { + if(write_socket(xfrd->ipc_handler.fd, &cmd, sizeof(cmd)) == -1) { log_msg(LOG_ERR, "xfrd: error writing ack to main: %s", strerror(errno)); } @@ -623,55 +603,119 @@ xfrd_send_quit_req(xfrd_state_t* xfrd) } static void -xfrd_send_stats(xfrd_state_t* xfrd) +xfrd_handle_ipc_SOAINFO(xfrd_state_t* xfrd, buffer_type* packet) { - sig_atomic_t cmd = NSD_STATS; - DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: ipc send stats")); - if(!write_socket(xfrd->ipc_handler.ev_fd, &cmd, sizeof(cmd))) { - log_msg(LOG_ERR, "xfrd: error writing stats to main: %s", - strerror(errno)); + xfrd_soa_t soa; + xfrd_soa_t* soa_ptr = &soa; + xfrd_zone_t* zone; + /* dname is sent in memory format */ + const dname_type* dname = (const dname_type*)buffer_begin(packet); + + /* find zone and decode SOA */ + zone = (xfrd_zone_t*)rbtree_search(xfrd->zones, dname); + buffer_skip(packet, dname_total_size(dname)); + + if(!buffer_available(packet, sizeof(uint32_t)*6 + sizeof(uint8_t)*2)) { + /* NSD has zone without any info */ + DEBUG(DEBUG_IPC,1, (LOG_INFO, "SOAINFO for %s lost zone", + dname_to_string(dname,0))); + soa_ptr = NULL; + } else { + /* read soa info */ + memset(&soa, 0, sizeof(soa)); + /* left out type, klass, count for speed */ + soa.type = htons(TYPE_SOA); + soa.klass = htons(CLASS_IN); + soa.ttl = htonl(buffer_read_u32(packet)); + soa.rdata_count = htons(7); + soa.prim_ns[0] = buffer_read_u8(packet); + if(!buffer_available(packet, soa.prim_ns[0])) + return; + buffer_read(packet, soa.prim_ns+1, soa.prim_ns[0]); + soa.email[0] = buffer_read_u8(packet); + if(!buffer_available(packet, soa.email[0])) + return; + buffer_read(packet, soa.email+1, soa.email[0]); + + soa.serial = htonl(buffer_read_u32(packet)); + soa.refresh = htonl(buffer_read_u32(packet)); + soa.retry = htonl(buffer_read_u32(packet)); + soa.expire = htonl(buffer_read_u32(packet)); + soa.minimum = htonl(buffer_read_u32(packet)); + DEBUG(DEBUG_IPC,1, (LOG_INFO, "SOAINFO for %s %u", + dname_to_string(dname,0), ntohl(soa.serial))); + } + + if(!zone) { + DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: zone %s master zone updated", + dname_to_string(dname,0))); + notify_handle_master_zone_soainfo(xfrd->notify_zones, + dname, soa_ptr); + return; } - xfrd->need_to_send_stats = 0; + xfrd_handle_incoming_soa(zone, soa_ptr, xfrd_time()); } void -xfrd_handle_ipc(int ATTR_UNUSED(fd), short event, void* arg) +xfrd_handle_ipc(netio_type* ATTR_UNUSED(netio), + netio_handler_type *handler, + netio_event_types_type event_types) { - xfrd_state_t* xfrd = (xfrd_state_t*)arg; - if ((event & EV_READ)) + xfrd_state_t* xfrd = (xfrd_state_t*)handler->user_data; + if ((event_types & NETIO_EVENT_READ)) { /* first attempt to read as a signal from main * could block further send operations */ - xfrd_handle_ipc_read(&xfrd->ipc_handler, xfrd); + xfrd_handle_ipc_read(handler, xfrd); } - if ((event & EV_WRITE)) + if ((event_types & NETIO_EVENT_WRITE)) { - if(xfrd->ipc_send_blocked) { /* wait for RELOAD_DONE */ - ipc_xfrd_set_listening(xfrd, EV_PERSIST|EV_READ); + if(xfrd->ipc_send_blocked) { /* wait for SOA_END */ + handler->event_types = NETIO_EVENT_READ; return; } - if(xfrd->need_to_send_shutdown) { - xfrd_send_shutdown_req(xfrd); + /* if necessary prepare a packet */ + if(!(xfrd->can_send_reload && xfrd->need_to_send_reload) && + !xfrd->need_to_send_quit && + !xfrd->sending_zone_state && + xfrd->dirty_zones->num > 0) { + xfrd_zone_t* zone = (xfrd_zone_t*)stack_pop(xfrd->dirty_zones); + assert(zone); + zone->dirty = 0; + xfrd->sending_zone_state = 1; + xfrd_write_expire_notification(xfrd->ipc_conn_write->packet, zone); + xfrd->ipc_conn_write->msglen = buffer_limit(xfrd->ipc_conn_write->packet); + /* skip length bytes; they are encoded in the packet, after cmd */ + xfrd->ipc_conn_write->total_bytes = sizeof(uint16_t); + } + /* write a bit */ + if(xfrd->sending_zone_state) { + /* call conn_write */ + int ret = conn_write(xfrd->ipc_conn_write); + if(ret == -1) { + log_msg(LOG_ERR, "xfrd: error in write ipc: %s", strerror(errno)); + xfrd->sending_zone_state = 0; + } + else if(ret == 1) { /* done */ + xfrd->sending_zone_state = 0; + } } else if(xfrd->need_to_send_quit) { xfrd_send_quit_req(xfrd); } else if(xfrd->can_send_reload && xfrd->need_to_send_reload) { xfrd_send_reload_req(xfrd); - } else if(xfrd->need_to_send_stats) { - xfrd_send_stats(xfrd); } if(!(xfrd->can_send_reload && xfrd->need_to_send_reload) && - !xfrd->need_to_send_shutdown && !xfrd->need_to_send_quit && - !xfrd->need_to_send_stats) { - /* disable writing for now */ - ipc_xfrd_set_listening(xfrd, EV_PERSIST|EV_READ); + !xfrd->sending_zone_state && + xfrd->dirty_zones->num == 0) { + handler->event_types = NETIO_EVENT_READ; /* disable writing for now */ } } } static void -xfrd_handle_ipc_read(struct event* handler, xfrd_state_t* xfrd) +xfrd_handle_ipc_read(netio_handler_type *handler, xfrd_state_t* xfrd) { sig_atomic_t cmd; int len; @@ -679,7 +723,6 @@ xfrd_handle_ipc_read(struct event* handler, xfrd_state_t* xfrd) if(xfrd->ipc_conn->is_reading==2) { buffer_type* tmp = xfrd->ipc_pass; uint32_t acl_num; - int32_t acl_xfr; /* read acl_num */ int ret = conn_read(xfrd->ipc_conn); if(ret == -1) { @@ -694,13 +737,11 @@ xfrd_handle_ipc_read(struct event* handler, xfrd_state_t* xfrd) xfrd->ipc_conn->packet = tmp; xfrd->ipc_conn->is_reading = 0; acl_num = buffer_read_u32(xfrd->ipc_pass); - acl_xfr = (int32_t)buffer_read_u32(xfrd->ipc_pass); - xfrd_handle_passed_packet(xfrd->ipc_conn->packet, acl_num, acl_xfr); + xfrd_handle_passed_packet(xfrd->ipc_conn->packet, acl_num); return; } if(xfrd->ipc_conn->is_reading) { /* reading an IPC message */ - buffer_type* tmp; int ret = conn_read(xfrd->ipc_conn); if(ret == -1) { log_msg(LOG_ERR, "xfrd: error in read ipc: %s", strerror(errno)); @@ -710,22 +751,26 @@ xfrd_handle_ipc_read(struct event* handler, xfrd_state_t* xfrd) if(ret == 0) return; buffer_flip(xfrd->ipc_conn->packet); - /* use ipc_conn to read remaining data as well */ - tmp = xfrd->ipc_pass; - xfrd->ipc_conn->is_reading=2; - xfrd->ipc_pass = xfrd->ipc_conn->packet; - xfrd->ipc_conn->packet = tmp; - xfrd->ipc_conn->total_bytes = sizeof(xfrd->ipc_conn->msglen); - xfrd->ipc_conn->msglen = 2*sizeof(uint32_t); - buffer_clear(xfrd->ipc_conn->packet); - buffer_set_limit(xfrd->ipc_conn->packet, xfrd->ipc_conn->msglen); + if(xfrd->ipc_is_soa) { + xfrd->ipc_conn->is_reading = 0; + xfrd_handle_ipc_SOAINFO(xfrd, xfrd->ipc_conn->packet); + } else { + /* use ipc_conn to read remaining data as well */ + buffer_type* tmp = xfrd->ipc_pass; + xfrd->ipc_conn->is_reading=2; + xfrd->ipc_pass = xfrd->ipc_conn->packet; + xfrd->ipc_conn->packet = tmp; + xfrd->ipc_conn->total_bytes = sizeof(xfrd->ipc_conn->msglen); + xfrd->ipc_conn->msglen = sizeof(uint32_t); + buffer_clear(xfrd->ipc_conn->packet); + buffer_set_limit(xfrd->ipc_conn->packet, xfrd->ipc_conn->msglen); + } return; } - if((len = read(handler->ev_fd, &cmd, sizeof(cmd))) == -1) { - if(errno != EINTR && errno != EAGAIN) - log_msg(LOG_ERR, "xfrd_handle_ipc: read: %s", - strerror(errno)); + if((len = read(handler->fd, &cmd, sizeof(cmd))) == -1) { + log_msg(LOG_ERR, "xfrd_handle_ipc: read: %s", + strerror(errno)); return; } if(len == 0) @@ -739,47 +784,48 @@ xfrd_handle_ipc_read(struct event* handler, xfrd_state_t* xfrd) switch(cmd) { case NSD_QUIT: case NSD_SHUTDOWN: - DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: main sent shutdown cmd.")); + DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: main send shutdown cmd.")); xfrd->shutdown = 1; break; - case NSD_RELOAD_DONE: + case NSD_SOA_BEGIN: + DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: ipc recv SOA_BEGIN")); + /* reload starts sending SOA INFOs; don't block */ + xfrd->parent_soa_info_pass = 1; + /* reset the nonblocking ipc write; + the new parent does not want half a packet */ + xfrd->sending_zone_state = 0; + break; + case NSD_SOA_INFO: + DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: ipc recv SOA_INFO")); + assert(xfrd->parent_soa_info_pass); + xfrd->ipc_is_soa = 1; + xfrd->ipc_conn->is_reading = 1; + break; + case NSD_SOA_END: /* reload has finished */ - DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: ipc recv RELOAD_DONE")); - if(block_read(NULL, handler->ev_fd, &xfrd->reload_pid, - sizeof(pid_t), -1) != sizeof(pid_t)) { - log_msg(LOG_ERR, "xfrd cannot get reload_pid"); - } - /* read the not-mytask for the results and soainfo */ - xfrd_process_task_result(xfrd, - xfrd->nsd->task[1-xfrd->nsd->mytask]); - /* reset the IPC, (and the nonblocking ipc write; - the new parent does not want half a packet) */ + DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: ipc recv SOA_END")); xfrd->can_send_reload = 1; + xfrd->parent_soa_info_pass = 0; xfrd->ipc_send_blocked = 0; - ipc_xfrd_set_listening(xfrd, EV_PERSIST|EV_READ|EV_WRITE); + handler->event_types |= NETIO_EVENT_WRITE; xfrd_reopen_logfile(); xfrd_check_failed_updates(); + xfrd_send_expy_all_zones(); break; case NSD_PASS_TO_XFRD: DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: ipc recv PASS_TO_XFRD")); + xfrd->ipc_is_soa = 0; xfrd->ipc_conn->is_reading = 1; break; - case NSD_RELOAD_REQ: - DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: ipc recv RELOAD_REQ")); - /* make reload happen, right away, and schedule file check */ - task_new_check_zonefiles(xfrd->nsd->task[xfrd->nsd->mytask], - xfrd->last_task, NULL); - xfrd_set_reload_now(xfrd); - break; case NSD_RELOAD: /* main tells us that reload is done, stop ipc send to main */ DEBUG(DEBUG_IPC,1, (LOG_INFO, "xfrd: ipc recv RELOAD")); - ipc_xfrd_set_listening(xfrd, EV_PERSIST|EV_READ|EV_WRITE); + handler->event_types |= NETIO_EVENT_WRITE; xfrd->need_to_send_quit = 1; break; default: log_msg(LOG_ERR, "xfrd_handle_ipc: bad mode %d (%d)", (int)cmd, - (int)ntohl(cmd)); + ntohl(cmd)); break; } @@ -790,3 +836,4 @@ xfrd_handle_ipc_read(struct event* handler, xfrd_state_t* xfrd) buffer_clear(xfrd->ipc_conn->packet); } } + diff --git a/usr.sbin/nsd/xfrd-disk.h b/usr.sbin/nsd/xfrd-disk.h index 2c8e23fc752..217ecc122b9 100644 --- a/usr.sbin/nsd/xfrd-disk.h +++ b/usr.sbin/nsd/xfrd-disk.h @@ -10,8 +10,8 @@ #ifndef XFRD_DISK_H #define XFRD_DISK_H +#include <config.h> struct xfrd_state; -struct nsd; /* magic string to identify xfrd state file */ #define XFRD_FILE_MAGIC "NSDXFRD1" @@ -21,13 +21,4 @@ void xfrd_read_state(struct xfrd_state* xfrd); /* write xfrd zone state if possible */ void xfrd_write_state(struct xfrd_state* xfrd); -/* create temp directory */ -void xfrd_make_tempdir(struct nsd* nsd); -/* rmdir temp directory */ -void xfrd_del_tempdir(struct nsd* nsd); -/* open temp file, makes directory if needed */ -FILE* xfrd_open_xfrfile(struct nsd* nsd, uint64_t number, char* mode); -/* unlink temp file */ -void xfrd_unlink_xfrfile(struct nsd* nsd, uint64_t number); - #endif /* XFRD_DISK_H */ |