Avoid including libds/common/roles.h in public loadparm.h header.
[sfrench/samba-autobuild/.git] / source4 / dns_server / dns_server.c
index cd121f9d8b2418471053db35383a3d5ec0ac74c1..a2dc15195bf7e8ddd6d6703b7bd34900e789289b 100644 (file)
 #include "lib/util/tevent_werror.h"
 #include "auth/auth.h"
 #include "auth/credentials/credentials.h"
+#include "librpc/gen_ndr/ndr_irpc.h"
+#include "lib/messaging/irpc.h"
+#include "libds/common/roles.h"
+
+#undef DBGC_CLASS
+#define DBGC_CLASS DBGC_DNS
 
 NTSTATUS server_service_dns_init(void);
 
@@ -117,11 +123,12 @@ static struct tevent_req *dns_process_send(TALLOC_CTX *mem_ctx,
        struct dns_process_state *state;
        enum ndr_err_code ndr_err;
        WERROR ret;
-
+       const char *forwarder = lpcfg_dns_forwarder(dns->task->lp_ctx);
        req = tevent_req_create(mem_ctx, &state, struct dns_process_state);
        if (req == NULL) {
                return NULL;
        }
+       state->state.mem_ctx = state;
        state->in = in;
 
        state->dns = dns;
@@ -130,7 +137,7 @@ static struct tevent_req *dns_process_send(TALLOC_CTX *mem_ctx,
                tevent_req_werror(req, WERR_INVALID_PARAM);
                return tevent_req_post(req, ev);
        }
-       dump_data(8, in->data, in->length);
+       dump_data_dbgc(DBGC_DNS, 8, in->data, in->length);
 
        ndr_err = ndr_pull_struct_blob(
                in, state, &state->in_packet,
@@ -141,16 +148,29 @@ static struct tevent_req *dns_process_send(TALLOC_CTX *mem_ctx,
                tevent_req_done(req);
                return tevent_req_post(req, ev);
        }
-       if (DEBUGLVL(8)) {
-               NDR_PRINT_DEBUG(dns_name_packet, &state->in_packet);
+       if (DEBUGLVLC(DBGC_DNS, 8)) {
+               NDR_PRINT_DEBUGC(DBGC_DNS, dns_name_packet, &state->in_packet);
+       }
+
+       ret = dns_verify_tsig(dns, state, &state->state, &state->in_packet, in);
+       if (!W_ERROR_IS_OK(ret)) {
+               DEBUG(1, ("Failed to verify TSIG!\n"));
+               state->dns_err = werr_to_dns_err(ret);
+               tevent_req_done(req);
+               return tevent_req_post(req, ev);
        }
 
-       ret = dns_verify_tsig(dns, &state->state, &state->in_packet);
+       if (state->in_packet.operation & DNS_FLAG_REPLY) {
+               DEBUG(1, ("Won't reply to replies.\n"));
+               tevent_req_werror(req, WERR_INVALID_PARAM);
+               return tevent_req_post(req, ev);
+       }
 
        state->state.flags = state->in_packet.operation;
        state->state.flags |= DNS_FLAG_REPLY;
 
-       if (lpcfg_dns_recursive_queries(dns->task->lp_ctx)) {
+       
+       if (forwarder && *forwarder) {
                state->state.flags |= DNS_FLAG_RECURSION_AVAIL;
        }
 
@@ -215,9 +235,13 @@ static WERROR dns_process_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
        if (tevent_req_is_werror(req, &ret)) {
                return ret;
        }
-       if (state->dns_err != DNS_RCODE_OK) {
+       if ((state->dns_err != DNS_RCODE_OK) &&
+           (state->dns_err != DNS_RCODE_NXDOMAIN)) {
                goto drop;
        }
+       if (state->dns_err != DNS_RCODE_OK) {
+               state->out_packet.operation |= state->dns_err;
+       }
        state->out_packet.operation |= state->state.flags;
 
        if (state->state.sign) {
@@ -229,6 +253,10 @@ static WERROR dns_process_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
                }
        }
 
+       if (DEBUGLVLC(DBGC_DNS, 8)) {
+               NDR_PRINT_DEBUGC(DBGC_DNS, dns_name_packet, &state->out_packet);
+       }
+
        ndr_err = ndr_push_struct_blob(
                out, mem_ctx, &state->out_packet,
                (ndr_push_flags_fn_t)ndr_push_dns_name_packet);
@@ -559,10 +587,9 @@ static void dns_udp_call_sendto_done(struct tevent_req *subreq)
 {
        struct dns_udp_call *call = tevent_req_callback_data(subreq,
                                       struct dns_udp_call);
-       ssize_t ret;
        int sys_errno;
 
-       ret = tdgram_sendto_queue_recv(subreq, &sys_errno);
+       tdgram_sendto_queue_recv(subreq, &sys_errno);
 
        /* We don't care about errors */
 
@@ -644,7 +671,7 @@ static NTSTATUS dns_add_socket(struct dns_server *dns,
 /*
   setup our listening sockets on the configured network interfaces
 */
-static NTSTATUS dns_startup_interfaces(struct dns_server *dns, struct loadparm_context *lp_ctx,
+static NTSTATUS dns_startup_interfaces(struct dns_server *dns,
                                       struct interface *ifaces)
 {
        const struct model_ops *model_ops;
@@ -662,13 +689,36 @@ static NTSTATUS dns_startup_interfaces(struct dns_server *dns, struct loadparm_c
                return NT_STATUS_INTERNAL_ERROR;
        }
 
-       num_interfaces = iface_list_count(ifaces);
+       if (ifaces != NULL) {
+               num_interfaces = iface_list_count(ifaces);
 
-       for (i=0; i<num_interfaces; i++) {
-               const char *address = talloc_strdup(tmp_ctx, iface_list_n_ip(ifaces, i));
+               for (i=0; i<num_interfaces; i++) {
+                       const char *address = talloc_strdup(tmp_ctx,
+                                                           iface_list_n_ip(ifaces, i));
 
-               status = dns_add_socket(dns, model_ops, "dns", address, DNS_SERVICE_PORT);
-               NT_STATUS_NOT_OK_RETURN(status);
+                       status = dns_add_socket(dns, model_ops, "dns", address,
+                                               DNS_SERVICE_PORT);
+                       NT_STATUS_NOT_OK_RETURN(status);
+               }
+       } else {
+               int num_binds = 0;
+               char **wcard;
+               wcard = iface_list_wildcard(tmp_ctx);
+               if (wcard == NULL) {
+                       DEBUG(0, ("No wildcard address available\n"));
+                       return NT_STATUS_INTERNAL_ERROR;
+               }
+               for (i = 0; wcard[i] != NULL; i++) {
+                       status = dns_add_socket(dns, model_ops, "dns", wcard[i],
+                                               DNS_SERVICE_PORT);
+                       if (NT_STATUS_IS_OK(status)) {
+                               num_binds++;
+                       }
+               }
+               if (num_binds == 0) {
+                       talloc_free(tmp_ctx);
+                       return NT_STATUS_INVALID_PARAMETER_MIX;
+               }
        }
 
        talloc_free(tmp_ctx);
@@ -676,27 +726,6 @@ static NTSTATUS dns_startup_interfaces(struct dns_server *dns, struct loadparm_c
        return NT_STATUS_OK;
 }
 
-static int dns_server_sort_zones(struct ldb_message **m1, struct ldb_message **m2)
-{
-       const char *n1, *n2;
-       size_t l1, l2;
-
-       n1 = ldb_msg_find_attr_as_string(*m1, "name", NULL);
-       n2 = ldb_msg_find_attr_as_string(*m2, "name", NULL);
-
-       l1 = strlen(n1);
-       l2 = strlen(n2);
-
-       /* If the string lengths are not equal just sort by length */
-       if (l1 != l2) {
-               /* If m1 is the larger zone name, return it first */
-               return l2 - l1;
-       }
-
-       /*TODO: We need to compare DNs here, we want the DomainDNSZones first */
-       return 0;
-}
-
 static struct dns_server_tkey_store *tkey_store_init(TALLOC_CTX *mem_ctx,
                                                     uint16_t size)
 {
@@ -718,15 +747,56 @@ static struct dns_server_tkey_store *tkey_store_init(TALLOC_CTX *mem_ctx,
        return buffer;
 }
 
+static NTSTATUS dns_server_reload_zones(struct dns_server *dns)
+{
+       NTSTATUS status;
+       struct dns_server_zone *new_list = NULL;
+       struct dns_server_zone *old_list = NULL;
+       struct dns_server_zone *old_zone;
+       status = dns_common_zones(dns->samdb, dns, &new_list);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
+       }
+       dns->zones = new_list;
+       while ((old_zone = DLIST_TAIL(old_list)) != NULL) {
+               DLIST_REMOVE(old_list, old_zone);
+               talloc_free(old_zone);
+       }
+
+       return NT_STATUS_OK;
+}
+
+/**
+ * Called when the internal DNS server should reload the zones from DB, for
+ * example, when zones are added or deleted through RPC or replicated by
+ * inbound DRS.
+ */
+static NTSTATUS dns_reload_zones(struct irpc_message *msg,
+                                struct dnssrv_reload_dns_zones *r)
+{
+       struct dns_server *dns;
+
+       dns = talloc_get_type(msg->private_data, struct dns_server);
+       if (dns == NULL) {
+               r->out.result = NT_STATUS_INTERNAL_ERROR;
+               return NT_STATUS_INTERNAL_ERROR;
+       }
+
+       r->out.result = dns_server_reload_zones(dns);
+
+       return NT_STATUS_OK;
+}
+
 static void dns_task_init(struct task_server *task)
 {
        struct dns_server *dns;
        NTSTATUS status;
-       struct interface *ifaces;
+       struct interface *ifaces = NULL;
        int ret;
-       struct ldb_result *res;
-       static const char * const attrs[] = { "name", NULL};
-       unsigned int i;
+       static const char * const attrs_none[] = { NULL};
+       struct ldb_message *dns_acc;
+       char *hostname_lower;
+       char *dns_spn;
 
        switch (lpcfg_server_role(task->lp_ctx)) {
        case ROLE_STANDALONE:
@@ -740,11 +810,13 @@ static void dns_task_init(struct task_server *task)
                break;
        }
 
-       load_interface_list(task, task->lp_ctx, &ifaces);
+       if (lpcfg_interfaces(task->lp_ctx) && lpcfg_bind_interfaces_only(task->lp_ctx)) {
+               load_interface_list(task, task->lp_ctx, &ifaces);
 
-       if (iface_list_count(ifaces) == 0) {
-               task_server_terminate(task, "dns: no network interfaces configured", false);
-               return;
+               if (iface_list_count(ifaces) == 0) {
+                       task_server_terminate(task, "dns: no network interfaces configured", false);
+                       return;
+               }
        }
 
        task_server_set_title(task, "task[dns]");
@@ -756,6 +828,8 @@ static void dns_task_init(struct task_server *task)
        }
 
        dns->task = task;
+       /*FIXME: Make this a configurable option */
+       dns->max_payload = 4096;
 
        dns->server_credentials = cli_credentials_init(dns);
        if (!dns->server_credentials) {
@@ -763,57 +837,81 @@ static void dns_task_init(struct task_server *task)
                return;
        }
 
-       cli_credentials_set_conf(dns->server_credentials, task->lp_ctx);
-       status = cli_credentials_set_machine_account(dns->server_credentials, task->lp_ctx);
-       if (!NT_STATUS_IS_OK(status)) {
-               task_server_terminate(task,
-                       talloc_asprintf(task, "Failed to obtain server credentials, perhaps a standalone server?: %s\n",
-                                       nt_errstr(status)),
-                       true);
+       dns->samdb = samdb_connect(dns, dns->task->event_ctx, dns->task->lp_ctx,
+                             system_session(dns->task->lp_ctx), 0);
+       if (!dns->samdb) {
+               task_server_terminate(task, "dns: samdb_connect failed", true);
                return;
        }
 
+       cli_credentials_set_conf(dns->server_credentials, task->lp_ctx);
+
+       hostname_lower = strlower_talloc(dns, lpcfg_netbios_name(task->lp_ctx));
+       dns_spn = talloc_asprintf(dns, "DNS/%s.%s",
+                                 hostname_lower,
+                                 lpcfg_dnsdomain(task->lp_ctx));
+       TALLOC_FREE(hostname_lower);
+
+       ret = dsdb_search_one(dns->samdb, dns, &dns_acc,
+                             ldb_get_default_basedn(dns->samdb), LDB_SCOPE_SUBTREE,
+                             attrs_none, 0, "(servicePrincipalName=%s)",
+                             dns_spn);
+       if (ret == LDB_SUCCESS) {
+               TALLOC_FREE(dns_acc);
+               if (!dns_spn) {
+                       task_server_terminate(task, "dns: talloc_asprintf failed", true);
+                       return;
+               }
+               status = cli_credentials_set_stored_principal(dns->server_credentials, task->lp_ctx, dns_spn);
+               if (!NT_STATUS_IS_OK(status)) {
+                       task_server_terminate(task,
+                                             talloc_asprintf(task, "Failed to obtain server credentials for DNS, "
+                                                             "despite finding it in the samdb! %s\n",
+                                                             nt_errstr(status)),
+                                             true);
+                       return;
+               }
+       } else {
+               TALLOC_FREE(dns_spn);
+               status = cli_credentials_set_machine_account(dns->server_credentials, task->lp_ctx);
+               if (!NT_STATUS_IS_OK(status)) {
+                       task_server_terminate(task,
+                                             talloc_asprintf(task, "Failed to obtain server credentials, perhaps a standalone server?: %s\n",
+                                                             nt_errstr(status)),
+                                             true);
+                       return;
+               }
+       }
+
        dns->tkeys = tkey_store_init(dns, TKEY_BUFFER_SIZE);
        if (!dns->tkeys) {
                task_server_terminate(task, "Failed to allocate tkey storage\n", true);
                return;
        }
 
-       dns->samdb = samdb_connect(dns, dns->task->event_ctx, dns->task->lp_ctx,
-                             system_session(dns->task->lp_ctx), 0);
-       if (!dns->samdb) {
-               task_server_terminate(task, "dns: samdb_connect failed", true);
+       status = dns_server_reload_zones(dns);
+       if (!NT_STATUS_IS_OK(status)) {
+               task_server_terminate(task, "dns: failed to load DNS zones", true);
                return;
        }
 
-       // TODO: this search does not work against windows
-       ret = dsdb_search(dns->samdb, dns, &res, NULL, LDB_SCOPE_SUBTREE,
-                         attrs, DSDB_SEARCH_SEARCH_ALL_PARTITIONS, "(objectClass=dnsZone)");
-       if (ret != LDB_SUCCESS) {
-               task_server_terminate(task,
-                                     "dns: failed to look up root DNS zones",
-                                     true);
+       status = dns_startup_interfaces(dns, ifaces);
+       if (!NT_STATUS_IS_OK(status)) {
+               task_server_terminate(task, "dns failed to setup interfaces", true);
                return;
        }
 
-       TYPESAFE_QSORT(res->msgs, res->count, dns_server_sort_zones);
-
-       for (i=0; i < res->count; i++) {
-               struct dns_server_zone *z;
-
-               z = talloc_zero(dns, struct dns_server_zone);
-               if (z == NULL) {
-               }
-
-               z->name = ldb_msg_find_attr_as_string(res->msgs[i], "name", NULL);
-               z->dn = talloc_move(z, &res->msgs[i]->dn);
-
-               DLIST_ADD_END(dns->zones, z, NULL);
+       /* Setup the IRPC interface and register handlers */
+       status = irpc_add_name(task->msg_ctx, "dnssrv");
+       if (!NT_STATUS_IS_OK(status)) {
+               task_server_terminate(task, "dns: failed to register IRPC name", true);
+               return;
        }
 
-       status = dns_startup_interfaces(dns, task->lp_ctx, ifaces);
+       status = IRPC_REGISTER(task->msg_ctx, irpc, DNSSRV_RELOAD_DNS_ZONES,
+                              dns_reload_zones, dns);
        if (!NT_STATUS_IS_OK(status)) {
-               task_server_terminate(task, "dns failed to setup interfaces", true);
+               task_server_terminate(task, "dns: failed to setup reload handler", true);
                return;
        }
 }