dns_server: clobber MNAME in the SOA
[sfrench/samba-autobuild/.git] / source4 / dns_server / dlz_bind9.c
1 /*
2    Unix SMB/CIFS implementation.
3
4    bind9 dlz driver for Samba
5
6    Copyright (C) 2010 Andrew Tridgell
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 "talloc.h"
24 #include "param/param.h"
25 #include "lib/events/events.h"
26 #include "dsdb/samdb/samdb.h"
27 #include "dsdb/common/util.h"
28 #include "auth/auth.h"
29 #include "auth/session.h"
30 #include "auth/gensec/gensec.h"
31 #include "librpc/gen_ndr/security.h"
32 #include "auth/credentials/credentials.h"
33 #include "system/kerberos.h"
34 #include "auth/kerberos/kerberos.h"
35 #include "gen_ndr/ndr_dnsp.h"
36 #include "gen_ndr/server_id.h"
37 #include "messaging/messaging.h"
38 #include <popt.h>
39 #include "lib/util/dlinklist.h"
40 #include "dlz_minimal.h"
41 #include "dns_server/dnsserver_common.h"
42
43 struct b9_options {
44         const char *url;
45         const char *debug;
46 };
47
48 struct b9_zone {
49         char *name;
50         struct b9_zone *prev, *next;
51 };
52
53 struct dlz_bind9_data {
54         struct b9_options options;
55         struct ldb_context *samdb;
56         struct tevent_context *ev_ctx;
57         struct loadparm_context *lp;
58         int *transaction_token;
59         uint32_t soa_serial;
60         struct b9_zone *zonelist;
61
62         /* Used for dynamic update */
63         struct smb_krb5_context *smb_krb5_ctx;
64         struct auth4_context *auth_context;
65         struct auth_session_info *session_info;
66         char *update_name;
67
68         /* helper functions from the dlz_dlopen driver */
69         log_t *log;
70         dns_sdlz_putrr_t *putrr;
71         dns_sdlz_putnamedrr_t *putnamedrr;
72         dns_dlz_writeablezone_t *writeable_zone;
73 };
74
75 static struct dlz_bind9_data *dlz_bind9_state = NULL;
76 static int dlz_bind9_state_ref_count = 0;
77
78 static const char *zone_prefixes[] = {
79         "CN=MicrosoftDNS,DC=DomainDnsZones",
80         "CN=MicrosoftDNS,DC=ForestDnsZones",
81         "CN=MicrosoftDNS,CN=System",
82         NULL
83 };
84
85 /*
86   return the version of the API
87  */
88 _PUBLIC_ int dlz_version(unsigned int *flags)
89 {
90         return DLZ_DLOPEN_VERSION;
91 }
92
93 /*
94    remember a helper function from the bind9 dlz_dlopen driver
95  */
96 static void b9_add_helper(struct dlz_bind9_data *state, const char *helper_name, void *ptr)
97 {
98         if (strcmp(helper_name, "log") == 0) {
99                 state->log = ptr;
100         }
101         if (strcmp(helper_name, "putrr") == 0) {
102                 state->putrr = ptr;
103         }
104         if (strcmp(helper_name, "putnamedrr") == 0) {
105                 state->putnamedrr = ptr;
106         }
107         if (strcmp(helper_name, "writeable_zone") == 0) {
108                 state->writeable_zone = ptr;
109         }
110 }
111
112 /*
113  * Add a trailing '.' if it's missing
114  */
115 static const char *b9_format_fqdn(TALLOC_CTX *mem_ctx, const char *str)
116 {
117         size_t len;
118         const char *tmp;
119
120         if (str == NULL || str[0] == '\0') {
121                 return str;
122         }
123
124         len = strlen(str);
125         if (str[len-1] != '.') {
126                 tmp = talloc_asprintf(mem_ctx, "%s.", str);
127         } else {
128                 tmp = str;
129         }
130         return tmp;
131 }
132
133 /*
134   format a record for bind9
135  */
136 static bool b9_format(struct dlz_bind9_data *state,
137                       TALLOC_CTX *mem_ctx,
138                       struct dnsp_DnssrvRpcRecord *rec,
139                       const char **type, const char **data)
140 {
141         uint32_t i;
142         char *tmp;
143         const char *fqdn;
144
145         switch (rec->wType) {
146         case DNS_TYPE_A:
147                 *type = "a";
148                 *data = rec->data.ipv4;
149                 break;
150
151         case DNS_TYPE_AAAA:
152                 *type = "aaaa";
153                 *data = rec->data.ipv6;
154                 break;
155
156         case DNS_TYPE_CNAME:
157                 *type = "cname";
158                 *data = b9_format_fqdn(mem_ctx, rec->data.cname);
159                 break;
160
161         case DNS_TYPE_TXT:
162                 *type = "txt";
163                 tmp = talloc_asprintf(mem_ctx, "\"%s\"", rec->data.txt.str[0]);
164                 for (i=1; i<rec->data.txt.count; i++) {
165                         tmp = talloc_asprintf_append(tmp, " \"%s\"", rec->data.txt.str[i]);
166                 }
167                 *data = tmp;
168                 break;
169
170         case DNS_TYPE_PTR:
171                 *type = "ptr";
172                 *data = b9_format_fqdn(mem_ctx, rec->data.ptr);
173                 break;
174
175         case DNS_TYPE_SRV:
176                 *type = "srv";
177                 fqdn = b9_format_fqdn(mem_ctx, rec->data.srv.nameTarget);
178                 if (fqdn == NULL) {
179                         return false;
180                 }
181                 *data = talloc_asprintf(mem_ctx, "%u %u %u %s",
182                                         rec->data.srv.wPriority,
183                                         rec->data.srv.wWeight,
184                                         rec->data.srv.wPort,
185                                         fqdn);
186                 break;
187
188         case DNS_TYPE_MX:
189                 *type = "mx";
190                 fqdn = b9_format_fqdn(mem_ctx, rec->data.mx.nameTarget);
191                 if (fqdn == NULL) {
192                         return false;
193                 }
194                 *data = talloc_asprintf(mem_ctx, "%u %s",
195                                         rec->data.mx.wPriority, fqdn);
196                 break;
197
198         case DNS_TYPE_HINFO:
199                 *type = "hinfo";
200                 *data = talloc_asprintf(mem_ctx, "%s %s",
201                                         rec->data.hinfo.cpu,
202                                         rec->data.hinfo.os);
203                 break;
204
205         case DNS_TYPE_NS:
206                 *type = "ns";
207                 *data = b9_format_fqdn(mem_ctx, rec->data.ns);
208                 break;
209
210         case DNS_TYPE_SOA: {
211                 const char *mname;
212                 *type = "soa";
213
214                 /* we need to fake the authoritative nameserver to
215                  * point at ourselves. This is how AD DNS servers
216                  * force clients to send updates to the right local DC
217                  */
218                 mname = talloc_asprintf(mem_ctx, "%s.%s.",
219                                         lpcfg_netbios_name(state->lp),
220                                         lpcfg_dnsdomain(state->lp));
221                 if (mname == NULL) {
222                         return false;
223                 }
224                 mname = strlower_talloc(mem_ctx, mname);
225                 if (mname == NULL) {
226                         return false;
227                 }
228
229                 fqdn = b9_format_fqdn(mem_ctx, rec->data.soa.rname);
230                 if (fqdn == NULL) {
231                         return false;
232                 }
233
234                 state->soa_serial = rec->data.soa.serial;
235
236                 *data = talloc_asprintf(mem_ctx, "%s %s %u %u %u %u %u",
237                                         mname, fqdn,
238                                         rec->data.soa.serial,
239                                         rec->data.soa.refresh,
240                                         rec->data.soa.retry,
241                                         rec->data.soa.expire,
242                                         rec->data.soa.minimum);
243                 break;
244         }
245
246         default:
247                 state->log(ISC_LOG_ERROR, "samba_dlz b9_format: unhandled record type %u",
248                            rec->wType);
249                 return false;
250         }
251
252         return true;
253 }
254
255 static const struct {
256         enum dns_record_type dns_type;
257         const char *typestr;
258         bool single_valued;
259 } dns_typemap[] = {
260         { DNS_TYPE_A,     "A"     , false},
261         { DNS_TYPE_AAAA,  "AAAA"  , false},
262         { DNS_TYPE_CNAME, "CNAME" , true},
263         { DNS_TYPE_TXT,   "TXT"   , false},
264         { DNS_TYPE_PTR,   "PTR"   , false},
265         { DNS_TYPE_SRV,   "SRV"   , false},
266         { DNS_TYPE_MX,    "MX"    , false},
267         { DNS_TYPE_HINFO, "HINFO" , false},
268         { DNS_TYPE_NS,    "NS"    , false},
269         { DNS_TYPE_SOA,   "SOA"   , true},
270 };
271
272
273 /*
274   see if a DNS type is single valued
275  */
276 static bool b9_single_valued(enum dns_record_type dns_type)
277 {
278         int i;
279         for (i=0; i<ARRAY_SIZE(dns_typemap); i++) {
280                 if (dns_typemap[i].dns_type == dns_type) {
281                         return dns_typemap[i].single_valued;
282                 }
283         }
284         return false;
285 }
286
287 /*
288   see if a DNS type is single valued
289  */
290 static bool b9_dns_type(const char *type, enum dns_record_type *dtype)
291 {
292         int i;
293         for (i=0; i<ARRAY_SIZE(dns_typemap); i++) {
294                 if (strcasecmp(dns_typemap[i].typestr, type) == 0) {
295                         *dtype = dns_typemap[i].dns_type;
296                         return true;
297                 }
298         }
299         return false;
300 }
301
302
303 #define DNS_PARSE_STR(ret, str, sep, saveptr) do {      \
304         (ret) = strtok_r(str, sep, &saveptr); \
305         if ((ret) == NULL) return false; \
306         } while (0)
307
308 #define DNS_PARSE_UINT(ret, str, sep, saveptr) do {  \
309         char *istr = strtok_r(str, sep, &saveptr); \
310         if ((istr) == NULL) return false; \
311         (ret) = strtoul(istr, NULL, 10); \
312         } while (0)
313
314 /*
315   parse a record from bind9
316  */
317 static bool b9_parse(struct dlz_bind9_data *state,
318                      const char *rdatastr,
319                      struct dnsp_DnssrvRpcRecord *rec)
320 {
321         char *full_name, *dclass, *type;
322         char *str, *tmp, *saveptr=NULL;
323         int i;
324
325         str = talloc_strdup(rec, rdatastr);
326         if (str == NULL) {
327                 return false;
328         }
329
330         /* parse the SDLZ string form */
331         DNS_PARSE_STR(full_name, str, "\t", saveptr);
332         DNS_PARSE_UINT(rec->dwTtlSeconds, NULL, "\t", saveptr);
333         DNS_PARSE_STR(dclass, NULL, "\t", saveptr);
334         DNS_PARSE_STR(type, NULL, "\t", saveptr);
335
336         /* construct the record */
337         for (i=0; i<ARRAY_SIZE(dns_typemap); i++) {
338                 if (strcasecmp(type, dns_typemap[i].typestr) == 0) {
339                         rec->wType = dns_typemap[i].dns_type;
340                         break;
341                 }
342         }
343         if (i == ARRAY_SIZE(dns_typemap)) {
344                 state->log(ISC_LOG_ERROR, "samba_dlz: unsupported record type '%s' for '%s'",
345                            type, full_name);
346                 return false;
347         }
348
349         switch (rec->wType) {
350         case DNS_TYPE_A:
351                 DNS_PARSE_STR(rec->data.ipv4, NULL, " ", saveptr);
352                 break;
353
354         case DNS_TYPE_AAAA:
355                 DNS_PARSE_STR(rec->data.ipv6, NULL, " ", saveptr);
356                 break;
357
358         case DNS_TYPE_CNAME:
359                 DNS_PARSE_STR(rec->data.cname, NULL, " ", saveptr);
360                 break;
361
362         case DNS_TYPE_TXT:
363                 rec->data.txt.count = 0;
364                 rec->data.txt.str = talloc_array(rec, const char *, rec->data.txt.count);
365                 tmp = strtok_r(NULL, "\t", &saveptr);
366                 while (tmp) {
367                         rec->data.txt.str = talloc_realloc(rec, rec->data.txt.str, const char *,
368                                                         rec->data.txt.count+1);
369                         if (tmp[0] == '"') {
370                                 /* Strip quotes */
371                                 rec->data.txt.str[rec->data.txt.count] = talloc_strndup(rec, &tmp[1], strlen(tmp)-2);
372                         } else {
373                                 rec->data.txt.str[rec->data.txt.count] = talloc_strdup(rec, tmp);
374                         }
375                         rec->data.txt.count++;
376                         tmp = strtok_r(NULL, " ", &saveptr);
377                 }
378                 break;
379
380         case DNS_TYPE_PTR:
381                 DNS_PARSE_STR(rec->data.ptr, NULL, " ", saveptr);
382                 break;
383
384         case DNS_TYPE_SRV:
385                 DNS_PARSE_UINT(rec->data.srv.wPriority, NULL, " ", saveptr);
386                 DNS_PARSE_UINT(rec->data.srv.wWeight, NULL, " ", saveptr);
387                 DNS_PARSE_UINT(rec->data.srv.wPort, NULL, " ", saveptr);
388                 DNS_PARSE_STR(rec->data.srv.nameTarget, NULL, " ", saveptr);
389                 break;
390
391         case DNS_TYPE_MX:
392                 DNS_PARSE_UINT(rec->data.mx.wPriority, NULL, " ", saveptr);
393                 DNS_PARSE_STR(rec->data.mx.nameTarget, NULL, " ", saveptr);
394                 break;
395
396         case DNS_TYPE_HINFO:
397                 DNS_PARSE_STR(rec->data.hinfo.cpu, NULL, " ", saveptr);
398                 DNS_PARSE_STR(rec->data.hinfo.os, NULL, " ", saveptr);
399                 break;
400
401         case DNS_TYPE_NS:
402                 DNS_PARSE_STR(rec->data.ns, NULL, " ", saveptr);
403                 break;
404
405         case DNS_TYPE_SOA:
406                 DNS_PARSE_STR(rec->data.soa.mname, NULL, " ", saveptr);
407                 DNS_PARSE_STR(rec->data.soa.rname, NULL, " ", saveptr);
408                 DNS_PARSE_UINT(rec->data.soa.serial, NULL, " ", saveptr);
409                 DNS_PARSE_UINT(rec->data.soa.refresh, NULL, " ", saveptr);
410                 DNS_PARSE_UINT(rec->data.soa.retry, NULL, " ", saveptr);
411                 DNS_PARSE_UINT(rec->data.soa.expire, NULL, " ", saveptr);
412                 DNS_PARSE_UINT(rec->data.soa.minimum, NULL, " ", saveptr);
413                 break;
414
415         default:
416                 state->log(ISC_LOG_ERROR, "samba_dlz b9_parse: unhandled record type %u",
417                            rec->wType);
418                 return false;
419         }
420
421         /* we should be at the end of the buffer now */
422         if (strtok_r(NULL, "\t ", &saveptr) != NULL) {
423                 state->log(ISC_LOG_ERROR, "samba_dlz b9_parse: unexpected data at end of string for '%s'",
424                            rdatastr);
425                 return false;
426         }
427
428         return true;
429 }
430
431 /*
432   send a resource record to bind9
433  */
434 static isc_result_t b9_putrr(struct dlz_bind9_data *state,
435                              void *handle, struct dnsp_DnssrvRpcRecord *rec,
436                              const char **types)
437 {
438         isc_result_t result;
439         const char *type, *data;
440         TALLOC_CTX *tmp_ctx = talloc_new(state);
441
442         if (!b9_format(state, tmp_ctx, rec, &type, &data)) {
443                 return ISC_R_FAILURE;
444         }
445
446         if (data == NULL) {
447                 talloc_free(tmp_ctx);
448                 return ISC_R_NOMEMORY;
449         }
450
451         if (types) {
452                 int i;
453                 for (i=0; types[i]; i++) {
454                         if (strcmp(types[i], type) == 0) break;
455                 }
456                 if (types[i] == NULL) {
457                         /* skip it */
458                         return ISC_R_SUCCESS;
459                 }
460         }
461
462         result = state->putrr(handle, type, rec->dwTtlSeconds, data);
463         if (result != ISC_R_SUCCESS) {
464                 state->log(ISC_LOG_ERROR, "Failed to put rr");
465         }
466         talloc_free(tmp_ctx);
467         return result;
468 }
469
470
471 /*
472   send a named resource record to bind9
473  */
474 static isc_result_t b9_putnamedrr(struct dlz_bind9_data *state,
475                                   void *handle, const char *name,
476                                   struct dnsp_DnssrvRpcRecord *rec)
477 {
478         isc_result_t result;
479         const char *type, *data;
480         TALLOC_CTX *tmp_ctx = talloc_new(state);
481
482         if (!b9_format(state, tmp_ctx, rec, &type, &data)) {
483                 return ISC_R_FAILURE;
484         }
485
486         if (data == NULL) {
487                 talloc_free(tmp_ctx);
488                 return ISC_R_NOMEMORY;
489         }
490
491         result = state->putnamedrr(handle, name, type, rec->dwTtlSeconds, data);
492         if (result != ISC_R_SUCCESS) {
493                 state->log(ISC_LOG_ERROR, "Failed to put named rr '%s'", name);
494         }
495         talloc_free(tmp_ctx);
496         return result;
497 }
498
499 /*
500    parse options
501  */
502 static isc_result_t parse_options(struct dlz_bind9_data *state,
503                                   unsigned int argc, const char **argv,
504                                   struct b9_options *options)
505 {
506         int opt;
507         poptContext pc;
508         struct poptOption long_options[] = {
509                 { "url", 'H', POPT_ARG_STRING, &options->url, 0, "database URL", "URL" },
510                 { "debug", 'd', POPT_ARG_STRING, &options->debug, 0, "debug level", "DEBUG" },
511                 { NULL }
512         };
513
514         pc = poptGetContext("dlz_bind9", argc, argv, long_options,
515                         POPT_CONTEXT_KEEP_FIRST);
516         while ((opt = poptGetNextOpt(pc)) != -1) {
517                 switch (opt) {
518                 default:
519                         state->log(ISC_LOG_ERROR, "dlz_bind9: Invalid option %s: %s",
520                                    poptBadOption(pc, 0), poptStrerror(opt));
521                         return ISC_R_FAILURE;
522                 }
523         }
524
525         return ISC_R_SUCCESS;
526 }
527
528
529 /*
530  * Create session info from PAC
531  * This is called as auth_context->generate_session_info_pac()
532  */
533 static NTSTATUS b9_generate_session_info_pac(struct auth4_context *auth_context,
534                                              TALLOC_CTX *mem_ctx,
535                                              struct smb_krb5_context *smb_krb5_context,
536                                              DATA_BLOB *pac_blob,
537                                              const char *principal_name,
538                                              const struct tsocket_address *remote_addr,
539                                              uint32_t session_info_flags,
540                                              struct auth_session_info **session_info)
541 {
542         NTSTATUS status;
543         struct auth_user_info_dc *user_info_dc;
544         TALLOC_CTX *tmp_ctx;
545
546         tmp_ctx = talloc_new(mem_ctx);
547         NT_STATUS_HAVE_NO_MEMORY(tmp_ctx);
548
549         status = kerberos_pac_blob_to_user_info_dc(tmp_ctx,
550                                                    *pac_blob,
551                                                    smb_krb5_context->krb5_context,
552                                                    &user_info_dc,
553                                                    NULL,
554                                                    NULL);
555         if (!NT_STATUS_IS_OK(status)) {
556                 talloc_free(tmp_ctx);
557                 return status;
558         }
559
560         if (user_info_dc->info->authenticated) {
561                 session_info_flags |= AUTH_SESSION_INFO_AUTHENTICATED;
562         }
563
564         session_info_flags |= AUTH_SESSION_INFO_SIMPLE_PRIVILEGES;
565
566         status = auth_generate_session_info(mem_ctx, NULL, NULL, user_info_dc,
567                                             session_info_flags, session_info);
568         if (!NT_STATUS_IS_OK(status)) {
569                 talloc_free(tmp_ctx);
570                 return status;
571         }
572
573         talloc_free(tmp_ctx);
574         return status;
575 }
576
577 /* Callback for the DEBUG() system, to catch the remaining messages */
578 static void b9_debug(void *private_ptr, int msg_level, const char *msg)
579 {
580         static const int isc_log_map[] = {
581                 ISC_LOG_CRITICAL, /* 0 */
582                 ISC_LOG_ERROR,    /* 1 */
583                 ISC_LOG_WARNING,   /* 2 */
584                 ISC_LOG_NOTICE    /* 3 */
585         };
586         struct dlz_bind9_data *state = private_ptr;
587         int     isc_log_level;
588         
589         if (msg_level >= ARRAY_SIZE(isc_log_map) || msg_level < 0) {
590                 isc_log_level = ISC_LOG_INFO;
591         } else {
592                 isc_log_level = isc_log_map[msg_level];
593         }
594         state->log(isc_log_level, "samba_dlz: %s", msg);
595 }
596
597 static int dlz_state_debug_unregister(struct dlz_bind9_data *state)
598 {
599         /* Stop logging (to the bind9 logs) */
600         debug_set_callback(NULL, NULL);
601         return 0;
602 }
603
604 /*
605   called to initialise the driver
606  */
607 _PUBLIC_ isc_result_t dlz_create(const char *dlzname,
608                                  unsigned int argc, const char **argv,
609                                  void **dbdata, ...)
610 {
611         struct dlz_bind9_data *state;
612         const char *helper_name;
613         va_list ap;
614         isc_result_t result;
615         struct ldb_dn *dn;
616         NTSTATUS nt_status;
617
618         if (dlz_bind9_state != NULL) {
619                 *dbdata = dlz_bind9_state;
620                 dlz_bind9_state_ref_count++;
621                 return ISC_R_SUCCESS;
622         }
623
624         state = talloc_zero(NULL, struct dlz_bind9_data);
625         if (state == NULL) {
626                 return ISC_R_NOMEMORY;
627         }
628
629         talloc_set_destructor(state, dlz_state_debug_unregister);
630
631         /* fill in the helper functions */
632         va_start(ap, dbdata);
633         while ((helper_name = va_arg(ap, const char *)) != NULL) {
634                 b9_add_helper(state, helper_name, va_arg(ap, void*));
635         }
636         va_end(ap);
637
638         /* Do not install samba signal handlers */
639         fault_setup_disable();
640
641         /* Start logging (to the bind9 logs) */
642         debug_set_callback(state, b9_debug);
643
644         state->ev_ctx = s4_event_context_init(state);
645         if (state->ev_ctx == NULL) {
646                 result = ISC_R_NOMEMORY;
647                 goto failed;
648         }
649
650         result = parse_options(state, argc, argv, &state->options);
651         if (result != ISC_R_SUCCESS) {
652                 goto failed;
653         }
654
655         state->lp = loadparm_init_global(true);
656         if (state->lp == NULL) {
657                 result = ISC_R_NOMEMORY;
658                 goto failed;
659         }
660
661         if (state->options.debug) {
662                 lpcfg_do_global_parameter(state->lp, "log level", state->options.debug);
663         } else {
664                 lpcfg_do_global_parameter(state->lp, "log level", "0");
665         }
666
667         if (smb_krb5_init_context(state, state->lp, &state->smb_krb5_ctx) != 0) {
668                 result = ISC_R_NOMEMORY;
669                 goto failed;
670         }
671
672         nt_status = gensec_init();
673         if (!NT_STATUS_IS_OK(nt_status)) {
674                 result = ISC_R_NOMEMORY;
675                 goto failed;
676         }
677
678         state->auth_context = talloc_zero(state, struct auth4_context);
679         if (state->auth_context == NULL) {
680                 result = ISC_R_NOMEMORY;
681                 goto failed;
682         }
683
684         if (state->options.url == NULL) {
685                 state->options.url = lpcfg_private_path(state, state->lp, "dns/sam.ldb");
686                 if (state->options.url == NULL) {
687                         result = ISC_R_NOMEMORY;
688                         goto failed;
689                 }
690         }
691
692         state->samdb = samdb_connect_url(state, state->ev_ctx, state->lp,
693                                         system_session(state->lp), 0, state->options.url);
694         if (state->samdb == NULL) {
695                 state->log(ISC_LOG_ERROR, "samba_dlz: Failed to connect to %s",
696                         state->options.url);
697                 result = ISC_R_FAILURE;
698                 goto failed;
699         }
700
701         dn = ldb_get_default_basedn(state->samdb);
702         if (dn == NULL) {
703                 state->log(ISC_LOG_ERROR, "samba_dlz: Unable to get basedn for %s - %s",
704                            state->options.url, ldb_errstring(state->samdb));
705                 result = ISC_R_FAILURE;
706                 goto failed;
707         }
708
709         state->log(ISC_LOG_INFO, "samba_dlz: started for DN %s",
710                    ldb_dn_get_linearized(dn));
711
712         state->auth_context->event_ctx = state->ev_ctx;
713         state->auth_context->lp_ctx = state->lp;
714         state->auth_context->sam_ctx = state->samdb;
715         state->auth_context->generate_session_info_pac = b9_generate_session_info_pac;
716
717         *dbdata = state;
718         dlz_bind9_state = state;
719         dlz_bind9_state_ref_count++;
720
721         return ISC_R_SUCCESS;
722
723 failed:
724         talloc_free(state);
725         return result;
726 }
727
728 /*
729   shutdown the backend
730  */
731 _PUBLIC_ void dlz_destroy(void *dbdata)
732 {
733         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
734         state->log(ISC_LOG_INFO, "samba_dlz: shutting down");
735
736         dlz_bind9_state_ref_count--;
737         if (dlz_bind9_state_ref_count == 0) {
738                 talloc_unlink(state, state->samdb);
739                 talloc_free(state);
740                 dlz_bind9_state = NULL;
741         }
742 }
743
744
745 /*
746   return the base DN for a zone
747  */
748 static isc_result_t b9_find_zone_dn(struct dlz_bind9_data *state, const char *zone_name,
749                                     TALLOC_CTX *mem_ctx, struct ldb_dn **zone_dn)
750 {
751         int ret;
752         TALLOC_CTX *tmp_ctx = talloc_new(state);
753         const char *attrs[] = { NULL };
754         int i;
755
756         for (i=0; zone_prefixes[i]; i++) {
757                 struct ldb_dn *dn;
758                 struct ldb_result *res;
759
760                 dn = ldb_dn_copy(tmp_ctx, ldb_get_default_basedn(state->samdb));
761                 if (dn == NULL) {
762                         talloc_free(tmp_ctx);
763                         return ISC_R_NOMEMORY;
764                 }
765
766                 if (!ldb_dn_add_child_fmt(dn, "DC=%s,%s", zone_name, zone_prefixes[i])) {
767                         talloc_free(tmp_ctx);
768                         return ISC_R_NOMEMORY;
769                 }
770
771                 ret = ldb_search(state->samdb, tmp_ctx, &res, dn, LDB_SCOPE_BASE, attrs, "objectClass=dnsZone");
772                 if (ret == LDB_SUCCESS) {
773                         if (zone_dn != NULL) {
774                                 *zone_dn = talloc_steal(mem_ctx, dn);
775                         }
776                         talloc_free(tmp_ctx);
777                         return ISC_R_SUCCESS;
778                 }
779                 talloc_free(dn);
780         }
781
782         talloc_free(tmp_ctx);
783         return ISC_R_NOTFOUND;
784 }
785
786
787 /*
788   return the DN for a name. The record does not need to exist, but the
789   zone must exist
790  */
791 static isc_result_t b9_find_name_dn(struct dlz_bind9_data *state, const char *name,
792                                     TALLOC_CTX *mem_ctx, struct ldb_dn **dn)
793 {
794         const char *p;
795
796         /* work through the name piece by piece, until we find a zone */
797         for (p=name; p; ) {
798                 isc_result_t result;
799                 result = b9_find_zone_dn(state, p, mem_ctx, dn);
800                 if (result == ISC_R_SUCCESS) {
801                         /* we found a zone, now extend the DN to get
802                          * the full DN
803                          */
804                         bool ret;
805                         if (p == name) {
806                                 ret = ldb_dn_add_child_fmt(*dn, "DC=@");
807                         } else {
808                                 ret = ldb_dn_add_child_fmt(*dn, "DC=%.*s", (int)(p-name)-1, name);
809                         }
810                         if (!ret) {
811                                 talloc_free(*dn);
812                                 return ISC_R_NOMEMORY;
813                         }
814                         return ISC_R_SUCCESS;
815                 }
816                 p = strchr(p, '.');
817                 if (p == NULL) {
818                         break;
819                 }
820                 p++;
821         }
822         return ISC_R_NOTFOUND;
823 }
824
825
826 /*
827   see if we handle a given zone
828  */
829 #if DLZ_DLOPEN_VERSION < 3
830 _PUBLIC_ isc_result_t dlz_findzonedb(void *dbdata, const char *name)
831 #else
832 _PUBLIC_ isc_result_t dlz_findzonedb(void *dbdata, const char *name,
833                                      dns_clientinfomethods_t *methods,
834                                      dns_clientinfo_t *clientinfo)
835 #endif
836 {
837         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
838         return b9_find_zone_dn(state, name, NULL, NULL);
839 }
840
841
842 /*
843   lookup one record
844  */
845 static isc_result_t dlz_lookup_types(struct dlz_bind9_data *state,
846                                      const char *zone, const char *name,
847                                      dns_sdlzlookup_t *lookup,
848                                      const char **types)
849 {
850         TALLOC_CTX *tmp_ctx = talloc_new(state);
851         struct ldb_dn *dn;
852         WERROR werr = WERR_DNS_ERROR_NAME_DOES_NOT_EXIST;
853         struct dnsp_DnssrvRpcRecord *records = NULL;
854         uint16_t num_records = 0, i;
855
856         for (i=0; zone_prefixes[i]; i++) {
857                 dn = ldb_dn_copy(tmp_ctx, ldb_get_default_basedn(state->samdb));
858                 if (dn == NULL) {
859                         talloc_free(tmp_ctx);
860                         return ISC_R_NOMEMORY;
861                 }
862
863                 if (!ldb_dn_add_child_fmt(dn, "DC=%s,DC=%s,%s", name, zone, zone_prefixes[i])) {
864                         talloc_free(tmp_ctx);
865                         return ISC_R_NOMEMORY;
866                 }
867
868                 werr = dns_common_lookup(state->samdb, tmp_ctx, dn,
869                                          &records, &num_records, NULL);
870                 if (W_ERROR_IS_OK(werr)) {
871                         break;
872                 }
873         }
874         if (!W_ERROR_IS_OK(werr)) {
875                 talloc_free(tmp_ctx);
876                 return ISC_R_NOTFOUND;
877         }
878
879         for (i=0; i < num_records; i++) {
880                 isc_result_t result;
881
882                 result = b9_putrr(state, lookup, &records[i], types);
883                 if (result != ISC_R_SUCCESS) {
884                         talloc_free(tmp_ctx);
885                         return result;
886                 }
887         }
888
889         talloc_free(tmp_ctx);
890         return ISC_R_SUCCESS;
891 }
892
893 /*
894   lookup one record
895  */
896 #if DLZ_DLOPEN_VERSION == 1
897 _PUBLIC_ isc_result_t dlz_lookup(const char *zone, const char *name,
898                                  void *dbdata, dns_sdlzlookup_t *lookup)
899 #else
900 _PUBLIC_ isc_result_t dlz_lookup(const char *zone, const char *name,
901                                  void *dbdata, dns_sdlzlookup_t *lookup,
902                                  dns_clientinfomethods_t *methods,
903                                  dns_clientinfo_t *clientinfo)
904 #endif
905 {
906         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
907         return dlz_lookup_types(state, zone, name, lookup, NULL);
908 }
909
910
911 /*
912   see if a zone transfer is allowed
913  */
914 _PUBLIC_ isc_result_t dlz_allowzonexfr(void *dbdata, const char *name, const char *client)
915 {
916         /* just say yes for all our zones for now */
917         struct dlz_bind9_data *state = talloc_get_type(
918                 dbdata, struct dlz_bind9_data);
919         return b9_find_zone_dn(state, name, NULL, NULL);
920 }
921
922 /*
923   perform a zone transfer
924  */
925 _PUBLIC_ isc_result_t dlz_allnodes(const char *zone, void *dbdata,
926                                    dns_sdlzallnodes_t *allnodes)
927 {
928         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
929         const char *attrs[] = { "dnsRecord", NULL };
930         int ret = LDB_SUCCESS, i, j;
931         struct ldb_dn *dn;
932         struct ldb_result *res;
933         TALLOC_CTX *tmp_ctx = talloc_new(state);
934
935         for (i=0; zone_prefixes[i]; i++) {
936                 dn = ldb_dn_copy(tmp_ctx, ldb_get_default_basedn(state->samdb));
937                 if (dn == NULL) {
938                         talloc_free(tmp_ctx);
939                         return ISC_R_NOMEMORY;
940                 }
941
942                 if (!ldb_dn_add_child_fmt(dn, "DC=%s,%s", zone, zone_prefixes[i])) {
943                         talloc_free(tmp_ctx);
944                         return ISC_R_NOMEMORY;
945                 }
946
947                 ret = ldb_search(state->samdb, tmp_ctx, &res, dn, LDB_SCOPE_SUBTREE,
948                                  attrs, "objectClass=dnsNode");
949                 if (ret == LDB_SUCCESS) {
950                         break;
951                 }
952         }
953         if (ret != LDB_SUCCESS) {
954                 talloc_free(tmp_ctx);
955                 return ISC_R_NOTFOUND;
956         }
957
958         for (i=0; i<res->count; i++) {
959                 struct ldb_message_element *el;
960                 TALLOC_CTX *el_ctx = talloc_new(tmp_ctx);
961                 const char *rdn, *name;
962                 const struct ldb_val *v;
963                 WERROR werr;
964                 struct dnsp_DnssrvRpcRecord *recs = NULL;
965                 uint16_t num_recs = 0;
966
967                 el = ldb_msg_find_element(res->msgs[i], "dnsRecord");
968                 if (el == NULL || el->num_values == 0) {
969                         state->log(ISC_LOG_INFO, "failed to find dnsRecord for %s",
970                                    ldb_dn_get_linearized(dn));
971                         talloc_free(el_ctx);
972                         continue;
973                 }
974
975                 v = ldb_dn_get_rdn_val(res->msgs[i]->dn);
976                 if (v == NULL) {
977                         state->log(ISC_LOG_INFO, "failed to find RDN for %s",
978                                    ldb_dn_get_linearized(dn));
979                         talloc_free(el_ctx);
980                         continue;
981                 }
982
983                 rdn = talloc_strndup(el_ctx, (char *)v->data, v->length);
984                 if (rdn == NULL) {
985                         talloc_free(tmp_ctx);
986                         return ISC_R_NOMEMORY;
987                 }
988
989                 if (strcmp(rdn, "@") == 0) {
990                         name = zone;
991                 } else {
992                         name = talloc_asprintf(el_ctx, "%s.%s", rdn, zone);
993                 }
994                 name = b9_format_fqdn(el_ctx, name);
995                 if (name == NULL) {
996                         talloc_free(tmp_ctx);
997                         return ISC_R_NOMEMORY;
998                 }
999
1000                 werr = dns_common_extract(state->samdb, el, el_ctx, &recs, &num_recs);
1001                 if (!W_ERROR_IS_OK(werr)) {
1002                         state->log(ISC_LOG_ERROR, "samba_dlz: failed to parse dnsRecord for %s, %s",
1003                                    ldb_dn_get_linearized(dn), win_errstr(werr));
1004                         talloc_free(el_ctx);
1005                         continue;
1006                 }
1007
1008                 for (j=0; j < num_recs; j++) {
1009                         isc_result_t result;
1010
1011                         result = b9_putnamedrr(state, allnodes, name, &recs[j]);
1012                         if (result != ISC_R_SUCCESS) {
1013                                 continue;
1014                         }
1015                 }
1016
1017                 talloc_free(el_ctx);
1018         }
1019
1020         talloc_free(tmp_ctx);
1021
1022         return ISC_R_SUCCESS;
1023 }
1024
1025
1026 /*
1027   start a transaction
1028  */
1029 _PUBLIC_ isc_result_t dlz_newversion(const char *zone, void *dbdata, void **versionp)
1030 {
1031         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
1032
1033         state->log(ISC_LOG_INFO, "samba_dlz: starting transaction on zone %s", zone);
1034
1035         if (state->transaction_token != NULL) {
1036                 state->log(ISC_LOG_INFO, "samba_dlz: transaction already started for zone %s", zone);
1037                 return ISC_R_FAILURE;
1038         }
1039
1040         state->transaction_token = talloc_zero(state, int);
1041         if (state->transaction_token == NULL) {
1042                 return ISC_R_NOMEMORY;
1043         }
1044
1045         if (ldb_transaction_start(state->samdb) != LDB_SUCCESS) {
1046                 state->log(ISC_LOG_INFO, "samba_dlz: failed to start a transaction for zone %s", zone);
1047                 talloc_free(state->transaction_token);
1048                 state->transaction_token = NULL;
1049                 return ISC_R_FAILURE;
1050         }
1051
1052         *versionp = (void *)state->transaction_token;
1053
1054         return ISC_R_SUCCESS;
1055 }
1056
1057 /*
1058   end a transaction
1059  */
1060 _PUBLIC_ void dlz_closeversion(const char *zone, isc_boolean_t commit,
1061                                void *dbdata, void **versionp)
1062 {
1063         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
1064
1065         if (state->transaction_token != (int *)*versionp) {
1066                 state->log(ISC_LOG_INFO, "samba_dlz: transaction not started for zone %s", zone);
1067                 return;
1068         }
1069
1070         if (commit) {
1071                 if (ldb_transaction_commit(state->samdb) != LDB_SUCCESS) {
1072                         state->log(ISC_LOG_INFO, "samba_dlz: failed to commit a transaction for zone %s", zone);
1073                         return;
1074                 }
1075                 state->log(ISC_LOG_INFO, "samba_dlz: committed transaction on zone %s", zone);
1076         } else {
1077                 if (ldb_transaction_cancel(state->samdb) != LDB_SUCCESS) {
1078                         state->log(ISC_LOG_INFO, "samba_dlz: failed to cancel a transaction for zone %s", zone);
1079                         return;
1080                 }
1081                 state->log(ISC_LOG_INFO, "samba_dlz: cancelling transaction on zone %s", zone);
1082         }
1083
1084         talloc_free(state->transaction_token);
1085         state->transaction_token = NULL;
1086         *versionp = NULL;
1087 }
1088
1089
1090 /*
1091   see if there is a SOA record for a zone
1092  */
1093 static bool b9_has_soa(struct dlz_bind9_data *state, struct ldb_dn *dn, const char *zone)
1094 {
1095         TALLOC_CTX *tmp_ctx = talloc_new(state);
1096         WERROR werr;
1097         struct dnsp_DnssrvRpcRecord *records = NULL;
1098         uint16_t num_records = 0, i;
1099
1100         if (!ldb_dn_add_child_fmt(dn, "DC=@,DC=%s", zone)) {
1101                 talloc_free(tmp_ctx);
1102                 return false;
1103         }
1104
1105         werr = dns_common_lookup(state->samdb, tmp_ctx, dn,
1106                                  &records, &num_records, NULL);
1107         if (!W_ERROR_IS_OK(werr)) {
1108                 talloc_free(tmp_ctx);
1109                 return false;
1110         }
1111
1112         for (i=0; i < num_records; i++) {
1113                 if (records[i].wType == DNS_TYPE_SOA) {
1114                         talloc_free(tmp_ctx);
1115                         return true;
1116                 }
1117         }
1118
1119         talloc_free(tmp_ctx);
1120         return false;
1121 }
1122
1123 static bool b9_zone_add(struct dlz_bind9_data *state, const char *name)
1124 {
1125         struct b9_zone *zone;
1126
1127         zone = talloc_zero(state, struct b9_zone);
1128         if (zone == NULL) {
1129                 return false;
1130         }
1131
1132         zone->name = talloc_strdup(zone, name);
1133         if (zone->name == NULL) {
1134                 talloc_free(zone);
1135                 return false;
1136         }
1137
1138         DLIST_ADD(state->zonelist, zone);
1139         return true;
1140 }
1141
1142 static bool b9_zone_exists(struct dlz_bind9_data *state, const char *name)
1143 {
1144         struct b9_zone *zone = state->zonelist;
1145         bool found = false;
1146
1147         while (zone != NULL) {
1148                 if (strcasecmp(name, zone->name) == 0) {
1149                         found = true;
1150                         break;
1151                 }
1152                 zone = zone->next;
1153         }
1154
1155         return found;
1156 }
1157
1158
1159 /*
1160   configure a writeable zone
1161  */
1162 #if DLZ_DLOPEN_VERSION < 3
1163 _PUBLIC_ isc_result_t dlz_configure(dns_view_t *view, void *dbdata)
1164 #else
1165 _PUBLIC_ isc_result_t dlz_configure(dns_view_t *view, dns_dlzdb_t *dlzdb,
1166                                     void *dbdata)
1167 #endif
1168 {
1169         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
1170         TALLOC_CTX *tmp_ctx;
1171         struct ldb_dn *dn;
1172         int i;
1173
1174         state->log(ISC_LOG_INFO, "samba_dlz: starting configure");
1175         if (state->writeable_zone == NULL) {
1176                 state->log(ISC_LOG_INFO, "samba_dlz: no writeable_zone method available");
1177                 return ISC_R_FAILURE;
1178         }
1179
1180         tmp_ctx = talloc_new(state);
1181
1182         for (i=0; zone_prefixes[i]; i++) {
1183                 const char *attrs[] = { "name", NULL };
1184                 int j, ret;
1185                 struct ldb_result *res;
1186
1187                 dn = ldb_dn_copy(tmp_ctx, ldb_get_default_basedn(state->samdb));
1188                 if (dn == NULL) {
1189                         talloc_free(tmp_ctx);
1190                         return ISC_R_NOMEMORY;
1191                 }
1192
1193                 if (!ldb_dn_add_child_fmt(dn, "%s", zone_prefixes[i])) {
1194                         talloc_free(tmp_ctx);
1195                         return ISC_R_NOMEMORY;
1196                 }
1197
1198                 ret = ldb_search(state->samdb, tmp_ctx, &res, dn, LDB_SCOPE_SUBTREE,
1199                                  attrs, "objectClass=dnsZone");
1200                 if (ret != LDB_SUCCESS) {
1201                         continue;
1202                 }
1203
1204                 for (j=0; j<res->count; j++) {
1205                         isc_result_t result;
1206                         const char *zone = ldb_msg_find_attr_as_string(res->msgs[j], "name", NULL);
1207                         struct ldb_dn *zone_dn;
1208
1209                         if (zone == NULL) {
1210                                 continue;
1211                         }
1212                         /* Ignore zones that are not handled in BIND */
1213                         if ((strcmp(zone, "RootDNSServers") == 0) ||
1214                             (strcmp(zone, "..TrustAnchors") == 0)) {
1215                                 continue;
1216                         }
1217                         zone_dn = ldb_dn_copy(tmp_ctx, dn);
1218                         if (zone_dn == NULL) {
1219                                 talloc_free(tmp_ctx);
1220                                 return ISC_R_NOMEMORY;
1221                         }
1222
1223                         if (!b9_has_soa(state, zone_dn, zone)) {
1224                                 continue;
1225                         }
1226
1227                         if (b9_zone_exists(state, zone)) {
1228                                 state->log(ISC_LOG_WARNING, "samba_dlz: Ignoring duplicate zone '%s' from '%s'",
1229                                            zone, ldb_dn_get_linearized(zone_dn));
1230                                 continue;
1231                         }
1232
1233                         if (!b9_zone_add(state, zone)) {
1234                                 talloc_free(tmp_ctx);
1235                                 return ISC_R_NOMEMORY;
1236                         }
1237
1238 #if DLZ_DLOPEN_VERSION < 3
1239                         result = state->writeable_zone(view, zone);
1240 #else
1241                         result = state->writeable_zone(view, dlzdb, zone);
1242 #endif
1243                         if (result != ISC_R_SUCCESS) {
1244                                 state->log(ISC_LOG_ERROR, "samba_dlz: Failed to configure zone '%s'",
1245                                            zone);
1246                                 talloc_free(tmp_ctx);
1247                                 return result;
1248                         }
1249                         state->log(ISC_LOG_INFO, "samba_dlz: configured writeable zone '%s'", zone);
1250                 }
1251         }
1252
1253         talloc_free(tmp_ctx);
1254         return ISC_R_SUCCESS;
1255 }
1256
1257 /*
1258   authorize a zone update
1259  */
1260 _PUBLIC_ isc_boolean_t dlz_ssumatch(const char *signer, const char *name, const char *tcpaddr,
1261                                     const char *type, const char *key, uint32_t keydatalen, uint8_t *keydata,
1262                                     void *dbdata)
1263 {
1264         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
1265         TALLOC_CTX *tmp_ctx;
1266         DATA_BLOB ap_req;
1267         struct cli_credentials *server_credentials;
1268         char *keytab_name;
1269         int ret;
1270         int ldb_ret;
1271         NTSTATUS nt_status;
1272         struct gensec_security *gensec_ctx;
1273         struct auth_session_info *session_info;
1274         struct ldb_dn *dn;
1275         isc_result_t result;
1276         struct ldb_result *res;
1277         const char * attrs[] = { NULL };
1278         uint32_t access_mask;
1279         struct gensec_settings *settings = NULL;
1280         const struct gensec_security_ops **backends = NULL;
1281         size_t idx = 0;
1282
1283         /* Remove cached credentials, if any */
1284         if (state->session_info) {
1285                 talloc_free(state->session_info);
1286                 state->session_info = NULL;
1287         }
1288         if (state->update_name) {
1289                 talloc_free(state->update_name);
1290                 state->update_name = NULL;
1291         }
1292
1293         tmp_ctx = talloc_new(NULL);
1294         if (tmp_ctx == NULL) {
1295                 state->log(ISC_LOG_ERROR, "samba_dlz: no memory");
1296                 return ISC_FALSE;
1297         }
1298
1299         ap_req = data_blob_const(keydata, keydatalen);
1300         server_credentials = cli_credentials_init(tmp_ctx);
1301         if (!server_credentials) {
1302                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to init server credentials");
1303                 talloc_free(tmp_ctx);
1304                 return ISC_FALSE;
1305         }
1306
1307         cli_credentials_set_krb5_context(server_credentials, state->smb_krb5_ctx);
1308         cli_credentials_set_conf(server_credentials, state->lp);
1309
1310         keytab_name = talloc_asprintf(tmp_ctx, "FILE:%s/dns.keytab",
1311                                         lpcfg_private_dir(state->lp));
1312         ret = cli_credentials_set_keytab_name(server_credentials, state->lp, keytab_name,
1313                                                 CRED_SPECIFIED);
1314         if (ret != 0) {
1315                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to obtain server credentials from %s",
1316                            keytab_name);
1317                 talloc_free(tmp_ctx);
1318                 return ISC_FALSE;
1319         }
1320         talloc_free(keytab_name);
1321
1322         settings = lpcfg_gensec_settings(tmp_ctx, state->lp);
1323         if (settings == NULL) {
1324                 state->log(ISC_LOG_ERROR, "samba_dlz: lpcfg_gensec_settings failed");
1325                 talloc_free(tmp_ctx);
1326                 return ISC_FALSE;
1327         }
1328         backends = talloc_zero_array(settings,
1329                                      const struct gensec_security_ops *, 3);
1330         if (backends == NULL) {
1331                 state->log(ISC_LOG_ERROR, "samba_dlz: talloc_zero_array gensec_security_ops failed");
1332                 talloc_free(tmp_ctx);
1333                 return ISC_FALSE;
1334         }
1335         settings->backends = backends;
1336
1337         gensec_init();
1338
1339         backends[idx++] = gensec_security_by_oid(NULL, GENSEC_OID_KERBEROS5);
1340         backends[idx++] = gensec_security_by_oid(NULL, GENSEC_OID_SPNEGO);
1341
1342         nt_status = gensec_server_start(tmp_ctx, settings,
1343                                         state->auth_context, &gensec_ctx);
1344         if (!NT_STATUS_IS_OK(nt_status)) {
1345                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to start gensec server");
1346                 talloc_free(tmp_ctx);
1347                 return ISC_FALSE;
1348         }
1349
1350         gensec_set_credentials(gensec_ctx, server_credentials);
1351
1352         nt_status = gensec_start_mech_by_oid(gensec_ctx, GENSEC_OID_SPNEGO);
1353         if (!NT_STATUS_IS_OK(nt_status)) {
1354                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to start spnego");
1355                 talloc_free(tmp_ctx);
1356                 return ISC_FALSE;
1357         }
1358
1359         /*
1360          * We only allow SPNEGO/KRB5 and make sure the backend
1361          * to is RPC/IPC free.
1362          *
1363          * See gensec_gssapi_update_internal() as
1364          * GENSEC_SERVER.
1365          *
1366          * It allows gensec_update() not to block.
1367          *
1368          * If that changes in future we need to use
1369          * gensec_update_send/recv here!
1370          */
1371         nt_status = gensec_update(gensec_ctx, tmp_ctx, ap_req, &ap_req);
1372         if (!NT_STATUS_IS_OK(nt_status)) {
1373                 state->log(ISC_LOG_ERROR, "samba_dlz: spnego update failed");
1374                 talloc_free(tmp_ctx);
1375                 return ISC_FALSE;
1376         }
1377
1378         nt_status = gensec_session_info(gensec_ctx, tmp_ctx, &session_info);
1379         if (!NT_STATUS_IS_OK(nt_status)) {
1380                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to create session info");
1381                 talloc_free(tmp_ctx);
1382                 return ISC_FALSE;
1383         }
1384
1385         /* Get the DN from name */
1386         result = b9_find_name_dn(state, name, tmp_ctx, &dn);
1387         if (result != ISC_R_SUCCESS) {
1388                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to find name %s", name);
1389                 talloc_free(tmp_ctx);
1390                 return ISC_FALSE;
1391         }
1392
1393         /* make sure the dn exists, or find parent dn in case new object is being added */
1394         ldb_ret = ldb_search(state->samdb, tmp_ctx, &res, dn, LDB_SCOPE_BASE,
1395                                 attrs, "objectClass=dnsNode");
1396         if (ldb_ret == LDB_ERR_NO_SUCH_OBJECT) {
1397                 ldb_dn_remove_child_components(dn, 1);
1398                 access_mask = SEC_ADS_CREATE_CHILD;
1399                 talloc_free(res);
1400         } else if (ldb_ret == LDB_SUCCESS) {
1401                 access_mask = SEC_STD_REQUIRED | SEC_ADS_SELF_WRITE;
1402                 talloc_free(res);
1403         } else {
1404                 talloc_free(tmp_ctx);
1405                 return ISC_FALSE;
1406         }
1407
1408         /* Do ACL check */
1409         ldb_ret = dsdb_check_access_on_dn(state->samdb, tmp_ctx, dn,
1410                                                 session_info->security_token,
1411                                                 access_mask, NULL);
1412         if (ldb_ret != LDB_SUCCESS) {
1413                 state->log(ISC_LOG_INFO,
1414                         "samba_dlz: disallowing update of signer=%s name=%s type=%s error=%s",
1415                         signer, name, type, ldb_strerror(ldb_ret));
1416                 talloc_free(tmp_ctx);
1417                 return ISC_FALSE;
1418         }
1419
1420         /* Cache session_info, so it can be used in the actual add/delete operation */
1421         state->update_name = talloc_strdup(state, name);
1422         if (state->update_name == NULL) {
1423                 state->log(ISC_LOG_ERROR, "samba_dlz: memory allocation error");
1424                 talloc_free(tmp_ctx);
1425                 return ISC_FALSE;
1426         }
1427         state->session_info = talloc_steal(state, session_info);
1428
1429         state->log(ISC_LOG_INFO, "samba_dlz: allowing update of signer=%s name=%s tcpaddr=%s type=%s key=%s",
1430                    signer, name, tcpaddr, type, key);
1431
1432         talloc_free(tmp_ctx);
1433         return ISC_TRUE;
1434 }
1435
1436 /*
1437   see if two DNS names are the same
1438  */
1439 static bool dns_name_equal(const char *name1, const char *name2)
1440 {
1441         size_t len1 = strlen(name1);
1442         size_t len2 = strlen(name2);
1443         if (name1[len1-1] == '.') len1--;
1444         if (name2[len2-1] == '.') len2--;
1445         if (len1 != len2) {
1446                 return false;
1447         }
1448         return strncasecmp_m(name1, name2, len1) == 0;
1449 }
1450
1451
1452 /*
1453   see if two dns records match
1454  */
1455 static bool b9_record_match(struct dlz_bind9_data *state,
1456                             struct dnsp_DnssrvRpcRecord *rec1, struct dnsp_DnssrvRpcRecord *rec2)
1457 {
1458         bool status;
1459         int i;
1460         struct in6_addr rec1_in_addr6;
1461         struct in6_addr rec2_in_addr6;
1462
1463         if (rec1->wType != rec2->wType) {
1464                 return false;
1465         }
1466         /* see if this type is single valued */
1467         if (b9_single_valued(rec1->wType)) {
1468                 return true;
1469         }
1470
1471         /* see if the data matches */
1472         switch (rec1->wType) {
1473         case DNS_TYPE_A:
1474                 return strcmp(rec1->data.ipv4, rec2->data.ipv4) == 0;
1475         case DNS_TYPE_AAAA: {
1476                 int ret;
1477
1478                 ret = inet_pton(AF_INET6, rec1->data.ipv6, &rec1_in_addr6);
1479                 if (ret != 1) {
1480                         return false;
1481                 }
1482                 ret = inet_pton(AF_INET6, rec2->data.ipv6, &rec2_in_addr6);
1483                 if (ret != 1) {
1484                         return false;
1485                 }
1486
1487                 return memcmp(&rec1_in_addr6, &rec2_in_addr6, sizeof(rec1_in_addr6)) == 0;
1488         }
1489         case DNS_TYPE_CNAME:
1490                 return dns_name_equal(rec1->data.cname, rec2->data.cname);
1491         case DNS_TYPE_TXT:
1492                 status = (rec1->data.txt.count == rec2->data.txt.count);
1493                 if (!status) return status;
1494                 for (i=0; i<rec1->data.txt.count; i++) {
1495                         status &= (strcmp(rec1->data.txt.str[i], rec2->data.txt.str[i]) == 0);
1496                 }
1497                 return status;
1498         case DNS_TYPE_PTR:
1499                 return dns_name_equal(rec1->data.ptr, rec2->data.ptr);
1500         case DNS_TYPE_NS:
1501                 return dns_name_equal(rec1->data.ns, rec2->data.ns);
1502
1503         case DNS_TYPE_SRV:
1504                 return rec1->data.srv.wPriority == rec2->data.srv.wPriority &&
1505                         rec1->data.srv.wWeight  == rec2->data.srv.wWeight &&
1506                         rec1->data.srv.wPort    == rec2->data.srv.wPort &&
1507                         dns_name_equal(rec1->data.srv.nameTarget, rec2->data.srv.nameTarget);
1508
1509         case DNS_TYPE_MX:
1510                 return rec1->data.mx.wPriority == rec2->data.mx.wPriority &&
1511                         dns_name_equal(rec1->data.mx.nameTarget, rec2->data.mx.nameTarget);
1512
1513         case DNS_TYPE_HINFO:
1514                 return strcmp(rec1->data.hinfo.cpu, rec2->data.hinfo.cpu) == 0 &&
1515                         strcmp(rec1->data.hinfo.os, rec2->data.hinfo.os) == 0;
1516
1517         case DNS_TYPE_SOA:
1518                 return dns_name_equal(rec1->data.soa.mname, rec2->data.soa.mname) &&
1519                         dns_name_equal(rec1->data.soa.rname, rec2->data.soa.rname) &&
1520                         rec1->data.soa.serial == rec2->data.soa.serial &&
1521                         rec1->data.soa.refresh == rec2->data.soa.refresh &&
1522                         rec1->data.soa.retry == rec2->data.soa.retry &&
1523                         rec1->data.soa.expire == rec2->data.soa.expire &&
1524                         rec1->data.soa.minimum == rec2->data.soa.minimum;
1525         default:
1526                 state->log(ISC_LOG_ERROR, "samba_dlz b9_record_match: unhandled record type %u",
1527                            rec1->wType);
1528                 break;
1529         }
1530
1531         return false;
1532 }
1533
1534 /*
1535  * Update session_info on samdb using the cached credentials
1536  */
1537 static bool b9_set_session_info(struct dlz_bind9_data *state, const char *name)
1538 {
1539         int ret;
1540
1541         if (state->update_name == NULL || state->session_info == NULL) {
1542                 state->log(ISC_LOG_ERROR, "samba_dlz: invalid credentials");
1543                 return false;
1544         }
1545
1546         /* Do not use client credentials, if we're not updating the client specified name */
1547         if (strcmp(state->update_name, name) != 0) {
1548                 return true;
1549         }
1550
1551         ret = ldb_set_opaque(state->samdb, "sessionInfo", state->session_info);
1552         if (ret != LDB_SUCCESS) {
1553                 state->log(ISC_LOG_ERROR, "samba_dlz: unable to set session info");
1554                 return false;
1555         }
1556
1557         return true;
1558 }
1559
1560 /*
1561  * Reset session_info on samdb as system session
1562  */
1563 static void b9_reset_session_info(struct dlz_bind9_data *state)
1564 {
1565         ldb_set_opaque(state->samdb, "sessionInfo", system_session(state->lp));
1566 }
1567
1568 /*
1569   add or modify a rdataset
1570  */
1571 _PUBLIC_ isc_result_t dlz_addrdataset(const char *name, const char *rdatastr, void *dbdata, void *version)
1572 {
1573         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
1574         struct dnsp_DnssrvRpcRecord *rec;
1575         struct ldb_dn *dn;
1576         isc_result_t result;
1577         bool tombstoned = false;
1578         bool needs_add = false;
1579         struct dnsp_DnssrvRpcRecord *recs = NULL;
1580         uint16_t num_recs = 0;
1581         uint16_t first = 0;
1582         uint16_t i;
1583         NTTIME t;
1584         WERROR werr;
1585
1586         if (state->transaction_token != (void*)version) {
1587                 state->log(ISC_LOG_INFO, "samba_dlz: bad transaction version");
1588                 return ISC_R_FAILURE;
1589         }
1590
1591         rec = talloc_zero(state, struct dnsp_DnssrvRpcRecord);
1592         if (rec == NULL) {
1593                 return ISC_R_NOMEMORY;
1594         }
1595
1596         unix_to_nt_time(&t, time(NULL));
1597         t /= 10*1000*1000; /* convert to seconds (NT time is in 100ns units) */
1598         t /= 3600;         /* convert to hours */
1599
1600         rec->rank        = DNS_RANK_ZONE;
1601         rec->dwTimeStamp = (uint32_t)t;
1602
1603         if (!b9_parse(state, rdatastr, rec)) {
1604                 state->log(ISC_LOG_INFO, "samba_dlz: failed to parse rdataset '%s'", rdatastr);
1605                 talloc_free(rec);
1606                 return ISC_R_FAILURE;
1607         }
1608
1609         /* find the DN of the record */
1610         result = b9_find_name_dn(state, name, rec, &dn);
1611         if (result != ISC_R_SUCCESS) {
1612                 talloc_free(rec);
1613                 return result;
1614         }
1615
1616         /* get any existing records */
1617         werr = dns_common_lookup(state->samdb, rec, dn,
1618                                  &recs, &num_recs, &tombstoned);
1619         if (W_ERROR_EQUAL(werr, WERR_DNS_ERROR_NAME_DOES_NOT_EXIST)) {
1620                 needs_add = true;
1621                 werr = WERR_OK;
1622         }
1623         if (!W_ERROR_IS_OK(werr)) {
1624                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to parse dnsRecord for %s, %s",
1625                            ldb_dn_get_linearized(dn), win_errstr(werr));
1626                 talloc_free(rec);
1627                 return ISC_R_FAILURE;
1628         }
1629
1630         if (tombstoned) {
1631                 /*
1632                  * we need to keep the existing tombstone record
1633                  * and ignore it
1634                  */
1635                 first = num_recs;
1636         }
1637
1638         /* there are existing records. We need to see if this will
1639          * replace a record or add to it
1640          */
1641         for (i=first; i < num_recs; i++) {
1642                 if (b9_record_match(state, rec, &recs[i])) {
1643                         break;
1644                 }
1645         }
1646         if (i == UINT16_MAX) {
1647                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to already %u dnsRecord values for %s",
1648                            i, ldb_dn_get_linearized(dn));
1649                 talloc_free(rec);
1650                 return ISC_R_FAILURE;
1651         }
1652
1653         if (i == num_recs) {
1654                 /* adding a new value */
1655                 recs = talloc_realloc(rec, recs,
1656                                       struct dnsp_DnssrvRpcRecord,
1657                                       num_recs + 1);
1658                 if (recs == NULL) {
1659                         talloc_free(rec);
1660                         return ISC_R_NOMEMORY;
1661                 }
1662                 num_recs++;
1663         }
1664
1665         recs[i] = *rec;
1666
1667         if (!b9_set_session_info(state, name)) {
1668                 talloc_free(rec);
1669                 return ISC_R_FAILURE;
1670         }
1671
1672         /* modify the record */
1673         werr = dns_common_replace(state->samdb, rec, dn,
1674                                   needs_add,
1675                                   state->soa_serial,
1676                                   recs, num_recs);
1677         b9_reset_session_info(state);
1678         if (!W_ERROR_IS_OK(werr)) {
1679                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to %s %s - %s",
1680                            needs_add ? "add" : "modify",
1681                            ldb_dn_get_linearized(dn), win_errstr(werr));
1682                 talloc_free(rec);
1683                 return ISC_R_FAILURE;
1684         }
1685
1686         state->log(ISC_LOG_INFO, "samba_dlz: added rdataset %s '%s'", name, rdatastr);
1687
1688         talloc_free(rec);
1689         return ISC_R_SUCCESS;
1690 }
1691
1692 /*
1693   remove a rdataset
1694  */
1695 _PUBLIC_ isc_result_t dlz_subrdataset(const char *name, const char *rdatastr, void *dbdata, void *version)
1696 {
1697         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
1698         struct dnsp_DnssrvRpcRecord *rec;
1699         struct ldb_dn *dn;
1700         isc_result_t result;
1701         struct dnsp_DnssrvRpcRecord *recs = NULL;
1702         uint16_t num_recs = 0;
1703         uint16_t i;
1704         WERROR werr;
1705
1706         if (state->transaction_token != (void*)version) {
1707                 state->log(ISC_LOG_ERROR, "samba_dlz: bad transaction version");
1708                 return ISC_R_FAILURE;
1709         }
1710
1711         rec = talloc_zero(state, struct dnsp_DnssrvRpcRecord);
1712         if (rec == NULL) {
1713                 return ISC_R_NOMEMORY;
1714         }
1715
1716         if (!b9_parse(state, rdatastr, rec)) {
1717                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to parse rdataset '%s'", rdatastr);
1718                 talloc_free(rec);
1719                 return ISC_R_FAILURE;
1720         }
1721
1722         /* find the DN of the record */
1723         result = b9_find_name_dn(state, name, rec, &dn);
1724         if (result != ISC_R_SUCCESS) {
1725                 talloc_free(rec);
1726                 return result;
1727         }
1728
1729         /* get the existing records */
1730         werr = dns_common_lookup(state->samdb, rec, dn,
1731                                  &recs, &num_recs, NULL);
1732         if (!W_ERROR_IS_OK(werr)) {
1733                 talloc_free(rec);
1734                 return ISC_R_NOTFOUND;
1735         }
1736
1737         for (i=0; i < num_recs; i++) {
1738                 if (b9_record_match(state, rec, &recs[i])) {
1739                         recs[i] = (struct dnsp_DnssrvRpcRecord) {
1740                                 .wType = DNS_TYPE_TOMBSTONE,
1741                         };
1742                         break;
1743                 }
1744         }
1745         if (i == num_recs) {
1746                 talloc_free(rec);
1747                 return ISC_R_NOTFOUND;
1748         }
1749
1750         if (!b9_set_session_info(state, name)) {
1751                 talloc_free(rec);
1752                 return ISC_R_FAILURE;
1753         }
1754
1755         /* modify the record */
1756         werr = dns_common_replace(state->samdb, rec, dn,
1757                                   false,/* needs_add */
1758                                   state->soa_serial,
1759                                   recs, num_recs);
1760         b9_reset_session_info(state);
1761         if (!W_ERROR_IS_OK(werr)) {
1762                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to modify %s - %s",
1763                            ldb_dn_get_linearized(dn), win_errstr(werr));
1764                 talloc_free(rec);
1765                 return ISC_R_FAILURE;
1766         }
1767
1768         state->log(ISC_LOG_INFO, "samba_dlz: subtracted rdataset %s '%s'", name, rdatastr);
1769
1770         talloc_free(rec);
1771         return ISC_R_SUCCESS;
1772 }
1773
1774
1775 /*
1776   delete all records of the given type
1777  */
1778 _PUBLIC_ isc_result_t dlz_delrdataset(const char *name, const char *type, void *dbdata, void *version)
1779 {
1780         struct dlz_bind9_data *state = talloc_get_type_abort(dbdata, struct dlz_bind9_data);
1781         TALLOC_CTX *tmp_ctx;
1782         struct ldb_dn *dn;
1783         isc_result_t result;
1784         enum dns_record_type dns_type;
1785         bool found = false;
1786         struct dnsp_DnssrvRpcRecord *recs = NULL;
1787         uint16_t num_recs = 0;
1788         uint16_t ri = 0;
1789         WERROR werr;
1790
1791         if (state->transaction_token != (void*)version) {
1792                 state->log(ISC_LOG_ERROR, "samba_dlz: bad transaction version");
1793                 return ISC_R_FAILURE;
1794         }
1795
1796         if (!b9_dns_type(type, &dns_type)) {
1797                 state->log(ISC_LOG_ERROR, "samba_dlz: bad dns type %s in delete", type);
1798                 return ISC_R_FAILURE;
1799         }
1800
1801         tmp_ctx = talloc_new(state);
1802
1803         /* find the DN of the record */
1804         result = b9_find_name_dn(state, name, tmp_ctx, &dn);
1805         if (result != ISC_R_SUCCESS) {
1806                 talloc_free(tmp_ctx);
1807                 return result;
1808         }
1809
1810         /* get the existing records */
1811         werr = dns_common_lookup(state->samdb, tmp_ctx, dn,
1812                                  &recs, &num_recs, NULL);
1813         if (!W_ERROR_IS_OK(werr)) {
1814                 talloc_free(tmp_ctx);
1815                 return ISC_R_NOTFOUND;
1816         }
1817
1818         for (ri=0; ri < num_recs; ri++) {
1819                 if (dns_type != recs[ri].wType) {
1820                         continue;
1821                 }
1822
1823                 found = true;
1824                 recs[ri] = (struct dnsp_DnssrvRpcRecord) {
1825                         .wType = DNS_TYPE_TOMBSTONE,
1826                 };
1827         }
1828
1829         if (!found) {
1830                 talloc_free(tmp_ctx);
1831                 return ISC_R_FAILURE;
1832         }
1833
1834         if (!b9_set_session_info(state, name)) {
1835                 talloc_free(tmp_ctx);
1836                 return ISC_R_FAILURE;
1837         }
1838
1839         /* modify the record */
1840         werr = dns_common_replace(state->samdb, tmp_ctx, dn,
1841                                   false,/* needs_add */
1842                                   state->soa_serial,
1843                                   recs, num_recs);
1844         b9_reset_session_info(state);
1845         if (!W_ERROR_IS_OK(werr)) {
1846                 state->log(ISC_LOG_ERROR, "samba_dlz: failed to modify %s - %s",
1847                            ldb_dn_get_linearized(dn), win_errstr(werr));
1848                 talloc_free(tmp_ctx);
1849                 return ISC_R_FAILURE;
1850         }
1851
1852         state->log(ISC_LOG_INFO, "samba_dlz: deleted rdataset %s of type %s", name, type);
1853
1854         talloc_free(tmp_ctx);
1855         return ISC_R_SUCCESS;
1856 }