source4/dns_server/dns_utils.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    DNS server utils
   5
   6    Copyright (C) 2010 Kai Blin  <kai@samba.org>
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 3 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include "includes.h"
  23 #include "libcli/util/ntstatus.h"
  24 #include "libcli/util/werror.h"
  25 #include "librpc/ndr/libndr.h"
  26 #include "librpc/gen_ndr/ndr_dns.h"
  27 #include "librpc/gen_ndr/ndr_dnsp.h"
  28 #include <ldb.h>
  29 #include "dsdb/samdb/samdb.h"
  30 #include "dsdb/common/util.h"
  31 #include "dns_server/dns_server.h"
  32
  33 bool dns_name_match(const char *zone, const char *name, size_t *host_part_len)
  34 {
  35         size_t zl = strlen(zone);
  36         size_t nl = strlen(name);
  37         ssize_t zi, ni;
  38         static const size_t fixup = 'a' - 'A';
  39
  40         if (zl > nl) {
  41                 return false;
  42         }
  43
  44         for (zi = zl, ni = nl; zi >= 0; zi--, ni--) {
  45                 char zc = zone[zi];
  46                 char nc = name[ni];
  47
  48                 /* convert to lower case */
  49                 if (zc >= 'A' && zc <= 'Z') {
  50                         zc += fixup;
  51                 }
  52                 if (nc >= 'A' && nc <= 'Z') {
  53                         nc += fixup;
  54                 }
  55
  56                 if (zc != nc) {
  57                         return false;
  58                 }
  59         }
  60
  61         if (ni >= 0) {
  62                 if (name[ni] != '.') {
  63                         return false;
  64                 }
  65
  66                 ni--;
  67         }
  68
  69         *host_part_len = ni+1;
  70
  71         return true;
  72 }
  73
  74 /* Names are equal if they match and there's nothing left over */
  75 bool dns_name_equal(const char *name1, const char *name2)
  76 {
  77         size_t host_part_len;
  78         bool ret = dns_name_match(name1, name2, &host_part_len);
  79
  80         return ret && (host_part_len == 0);
  81 }
  82
  83 /*
  84   see if two dns records match
  85  */
  86 bool dns_records_match(struct dnsp_DnssrvRpcRecord *rec1,
  87                        struct dnsp_DnssrvRpcRecord *rec2)
  88 {
  89         bool status;
  90         int i;
  91
  92         if (rec1->wType != rec2->wType) {
  93                 return false;
  94         }
  95
  96         /* see if the data matches */
  97         switch (rec1->wType) {
  98         case DNS_TYPE_A:
  99                 return strcmp(rec1->data.ipv4, rec2->data.ipv4) == 0;
 100         case DNS_TYPE_AAAA:
 101                 return strcmp(rec1->data.ipv6, rec2->data.ipv6) == 0;
 102         case DNS_TYPE_CNAME:
 103                 return dns_name_equal(rec1->data.cname, rec2->data.cname);
 104         case DNS_TYPE_TXT:
 105                 if (rec1->data.txt.count != rec2->data.txt.count) {
 106                         return false;
 107                 }
 108                 status = true;
 109                 for (i=0; i<rec1->data.txt.count; i++) {
 110                         status = status && (strcmp(rec1->data.txt.str[i],
 111                                                 rec2->data.txt.str[i]) == 0);
 112                 }
 113                 return status;
 114         case DNS_TYPE_PTR:
 115                 return strcmp(rec1->data.ptr, rec2->data.ptr) == 0;
 116         case DNS_TYPE_NS:
 117                 return dns_name_equal(rec1->data.ns, rec2->data.ns);
 118
 119         case DNS_TYPE_SRV:
 120                 return rec1->data.srv.wPriority == rec2->data.srv.wPriority &&
 121                         rec1->data.srv.wWeight  == rec2->data.srv.wWeight &&
 122                         rec1->data.srv.wPort    == rec2->data.srv.wPort &&
 123                         dns_name_equal(rec1->data.srv.nameTarget, rec2->data.srv.nameTarget);
 124
 125         case DNS_TYPE_MX:
 126                 return rec1->data.mx.wPriority == rec2->data.mx.wPriority &&
 127                         dns_name_equal(rec1->data.mx.nameTarget, rec2->data.mx.nameTarget);
 128
 129         case DNS_TYPE_HINFO:
 130                 return strcmp(rec1->data.hinfo.cpu, rec2->data.hinfo.cpu) == 0 &&
 131                         strcmp(rec1->data.hinfo.os, rec2->data.hinfo.os) == 0;
 132
 133         case DNS_TYPE_SOA:
 134                 return dns_name_equal(rec1->data.soa.mname, rec2->data.soa.mname) &&
 135                         dns_name_equal(rec1->data.soa.rname, rec2->data.soa.rname) &&
 136                         rec1->data.soa.serial == rec2->data.soa.serial &&
 137                         rec1->data.soa.refresh == rec2->data.soa.refresh &&
 138                         rec1->data.soa.retry == rec2->data.soa.retry &&
 139                         rec1->data.soa.expire == rec2->data.soa.expire &&
 140                         rec1->data.soa.minimum == rec2->data.soa.minimum;
 141         default:
 142                 break;
 143         }
 144
 145         return false;
 146 }
 147
 148 WERROR dns_lookup_records(struct dns_server *dns,
 149                           TALLOC_CTX *mem_ctx,
 150                           struct ldb_dn *dn,
 151                           struct dnsp_DnssrvRpcRecord **records,
 152                           uint16_t *rec_count)
 153 {
 154         return dns_common_lookup(dns->samdb, mem_ctx, dn, records, rec_count);
 155 }
 156
 157 WERROR dns_replace_records(struct dns_server *dns,
 158                            TALLOC_CTX *mem_ctx,
 159                            struct ldb_dn *dn,
 160                            bool needs_add,
 161                            struct dnsp_DnssrvRpcRecord *records,
 162                            uint16_t rec_count)
 163 {
 164         struct ldb_message_element *el;
 165         uint16_t i;
 166         int ret;
 167         struct ldb_message *msg = NULL;
 168
 169         msg = ldb_msg_new(mem_ctx);
 170         W_ERROR_HAVE_NO_MEMORY(msg);
 171
 172         msg->dn = dn;
 173
 174         ret = ldb_msg_add_empty(msg, "dnsRecord", LDB_FLAG_MOD_REPLACE, &el);
 175         if (ret != LDB_SUCCESS) {
 176                 return DNS_ERR(SERVER_FAILURE);
 177         }
 178
 179         el->values = talloc_zero_array(el, struct ldb_val, rec_count);
 180         if (rec_count > 0) {
 181                 W_ERROR_HAVE_NO_MEMORY(el->values);
 182         }
 183
 184         for (i = 0; i < rec_count; i++) {
 185                 static const struct dnsp_DnssrvRpcRecord zero;
 186                 struct ldb_val *v = &el->values[el->num_values];
 187                 enum ndr_err_code ndr_err;
 188
 189                 if (memcmp(&records[i], &zero, sizeof(zero)) == 0) {
 190                         continue;
 191                 }
 192                 ndr_err = ndr_push_struct_blob(v, el->values, &records[i],
 193                                 (ndr_push_flags_fn_t)ndr_push_dnsp_DnssrvRpcRecord);
 194                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 195                         DEBUG(0, ("Failed to grab dnsp_DnssrvRpcRecord\n"));
 196                         return DNS_ERR(SERVER_FAILURE);
 197                 }
 198                 el->num_values++;
 199         }
 200
 201
 202         if (el->num_values == 0) {
 203                 if (needs_add) {
 204                         return WERR_OK;
 205                 }
 206                 /* TODO: Delete object? */
 207                 el->flags = LDB_FLAG_MOD_DELETE;
 208         }
 209
 210         if (needs_add) {
 211                 ret = ldb_msg_add_string(msg, "objectClass", "dnsNode");
 212                 if (ret != LDB_SUCCESS) {
 213                         return DNS_ERR(SERVER_FAILURE);
 214                 }
 215
 216                 ret = ldb_add(dns->samdb, msg);
 217                 if (ret != LDB_SUCCESS) {
 218                         return DNS_ERR(SERVER_FAILURE);
 219                 }
 220
 221                 return WERR_OK;
 222         }
 223
 224         ret = ldb_modify(dns->samdb, msg);
 225         if (ret != LDB_SUCCESS) {
 226                 return DNS_ERR(SERVER_FAILURE);
 227         }
 228
 229         return WERR_OK;
 230 }
 231
 232 bool dns_authorative_for_zone(struct dns_server *dns,
 233                               const char *name)
 234 {
 235         const struct dns_server_zone *z;
 236         size_t host_part_len = 0;
 237
 238         if (name == NULL) {
 239                 return false;
 240         }
 241
 242         if (strcmp(name, "") == 0) {
 243                 return true;
 244         }
 245         for (z = dns->zones; z != NULL; z = z->next) {
 246                 bool match;
 247
 248                 match = dns_name_match(z->name, name, &host_part_len);
 249                 if (match) {
 250                         break;
 251                 }
 252         }
 253         if (z == NULL) {
 254                 return false;
 255         }
 256
 257         return true;
 258 }
 259
 260 WERROR dns_name2dn(struct dns_server *dns,
 261                    TALLOC_CTX *mem_ctx,
 262                    const char *name,
 263                    struct ldb_dn **_dn)
 264 {
 265         struct ldb_dn *base;
 266         struct ldb_dn *dn;
 267         const struct dns_server_zone *z;
 268         size_t host_part_len = 0;
 269
 270         if (name == NULL) {
 271                 return DNS_ERR(FORMAT_ERROR);
 272         }
 273
 274         /*TODO: Check if 'name' is a valid DNS name */
 275
 276         if (strcmp(name, "") == 0) {
 277                 base = ldb_get_default_basedn(dns->samdb);
 278                 dn = ldb_dn_copy(mem_ctx, base);
 279                 ldb_dn_add_child_fmt(dn, "DC=@,DC=RootDNSServers,CN=MicrosoftDNS,CN=System");
 280                 *_dn = dn;
 281                 return WERR_OK;
 282         }
 283
 284         for (z = dns->zones; z != NULL; z = z->next) {
 285                 bool match;
 286
 287                 match = dns_name_match(z->name, name, &host_part_len);
 288                 if (match) {
 289                         break;
 290                 }
 291         }
 292
 293         if (z == NULL) {
 294                 return DNS_ERR(NAME_ERROR);
 295         }
 296
 297         if (host_part_len == 0) {
 298                 dn = ldb_dn_copy(mem_ctx, z->dn);
 299                 ldb_dn_add_child_fmt(dn, "DC=@");
 300                 *_dn = dn;
 301                 return WERR_OK;
 302         }
 303
 304         dn = ldb_dn_copy(mem_ctx, z->dn);
 305         ldb_dn_add_child_fmt(dn, "DC=%*.*s", (int)host_part_len, (int)host_part_len, name);
 306         *_dn = dn;
 307         return WERR_OK;
 308 }
 309
 310 WERROR dns_generate_options(struct dns_server *dns,
 311                             TALLOC_CTX *mem_ctx,
 312                             struct dns_res_rec **options)
 313 {
 314         struct dns_res_rec *o;
 315
 316         o = talloc_zero(mem_ctx, struct dns_res_rec);
 317         if (o == NULL) {
 318                 return WERR_NOMEM;
 319         }
 320         o->name = '\0';
 321         o->rr_type = DNS_QTYPE_OPT;
 322         /* This is ugly, but RFC2671 wants the payload size in this field */
 323         o->rr_class = (enum dns_qclass) dns->max_payload;
 324         o->ttl = 0;
 325         o->length = 0;
 326
 327         *options = o;
 328         return WERR_OK;
 329 }