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