diff options
Diffstat (limited to 'source4/scripting/python')
| -rw-r--r-- | source4/scripting/python/samba/provision/sambadns.py | 549 | 
1 files changed, 435 insertions, 114 deletions
diff --git a/source4/scripting/python/samba/provision/sambadns.py b/source4/scripting/python/samba/provision/sambadns.py index 6b8561e1b5..8e7661b662 100644 --- a/source4/scripting/python/samba/provision/sambadns.py +++ b/source4/scripting/python/samba/provision/sambadns.py @@ -2,6 +2,7 @@  # backend code for provisioning DNS for a Samba4 server  #  # Copyright (C) Kai Blin <kai@samba.org> 2011 +# Copyright (C) Amitay Isaacs <amitay@gmail.com> 2011  #  # This program is free software; you can redistribute it and/or modify  # it under the terms of the GNU General Public License as published by @@ -20,11 +21,48 @@  """DNS-related provisioning"""  import os +import uuid  import ldb  import samba -from samba.ndr import ndr_pack +from samba.ndr import ndr_pack, ndr_unpack  from samba import read_and_sub_file -from samba.dcerpc import dnsp +from samba.dcerpc import dnsp, misc +from samba.dsdb import ( +    DS_DOMAIN_FUNCTION_2000, +    DS_DOMAIN_FUNCTION_2003, +    DS_DOMAIN_FUNCTION_2008, +    DS_DOMAIN_FUNCTION_2008_R2 +    ) + + +def add_ldif(ldb, ldif_file, subst_vars, controls=["relax:0"]): +    ldif_file_path = os.path.join(samba.param.setup_dir(), ldif_file) +    data = read_and_sub_file(ldif_file_path, subst_vars) +    ldb.add_ldif(data, controls) + +def modify_ldif(ldb, ldif_file, subst_vars, controls=["relax:0"]): +    ldif_file_path = os.path.join(samba.param.setup_dir(), ldif_file) +    data = read_and_sub_file(ldif_file_path, subst_vars) +    ldb.modify_ldif(data, controls) + +def get_domainguid(samdb, domaindn): +    res = samdb.search(base=domaindn, scope=ldb.SCOPE_BASE, attrs=["objectGUID"]) +    domainguid =  str(ndr_unpack(misc.GUID, res[0]["objectGUID"][0])) +    return domainguid + +def get_ntdsguid(samdb, domaindn): +    configdn = "CN=Configuration,%s" % domaindn + +    res1 = samdb.search(base="OU=Domain Controllers,%s" % domaindn, scope=ldb.SCOPE_ONELEVEL, +                        attrs=["dNSHostName"]) + +    res2 = samdb.search(expression="serverReference=%s" % res1[0].dn, base=configdn) + +    res3 = samdb.search(base="CN=NTDS Settings,%s" % res2[0].dn, scope=ldb.SCOPE_BASE, +                        attrs=["objectGUID"]) +    ntdsguid = str(ndr_unpack(misc.GUID, res3[0]["objectGUID"][0])) +    return ntdsguid +  class ARecord(dnsp.DnssrvRpcRecord):      def __init__(self, ip_addr, serial=1, ttl=3600): @@ -42,6 +80,14 @@ class AAAARecord(dnsp.DnssrvRpcRecord):          self.dwTtlSeconds = ttl          self.data = ip6_addr +class CNameRecord(dnsp.DnssrvRpcRecord): +    def __init__(self, cname, serial=1, ttl=900): +        super(CNameRecord, self).__init__() +        self.wType = dnsp.DNS_TYPE_CNAME +        self.dwSerial = serial +        self.dwTtlSeconds = ttl +        self.data = cname +  class NSRecord(dnsp.DnssrvRpcRecord):      def __init__(self, dns_server, serial=1, ttl=3600):          super(NSRecord, self).__init__() @@ -50,6 +96,15 @@ class NSRecord(dnsp.DnssrvRpcRecord):          self.dwTtlSeconds = ttl          self.data = dns_server +class RootNSRecord(dnsp.DnssrvRpcRecord): +    def __init__(self, dns_server, serial=1, ttl=3600): +        super(RootNSRecord, self).__init__() +        self.wType = dnsp.DNS_TYPE_NS +        self.dwSerial = serial +        self.dwTtlSeconds = ttl +        self.data = dns_server +        self.rank = dnsp.DNS_RANK_ROOT_HINT +  class SOARecord(dnsp.DnssrvRpcRecord):      def __init__(self, mname, rname, serial=1, refresh=900, retry=600,                   expire=86400, minimum=3600, ttl=3600): @@ -79,153 +134,419 @@ class SRVRecord(dnsp.DnssrvRpcRecord):          srv.wWeight = weight          self.data = srv -def setup_ad_dns(samdb, names, hostip=None, hostip6=None): -    domaindn = names.domaindn -    dnsdomain = names.dnsdomain.lower() -    hostname = names.netbiosname.lower() -    dnsname = "%s.%s" % (hostname, dnsdomain) -    site = names.sitename -    dns_ldif = os.path.join(samba.param.setup_dir(), "provision_dns_add.ldif") +def setup_dns_partitions(samdb, domaindn, forestdn, configdn, serverdn): + +    # FIXME: Default security descriptor for Domain-DNS objectCategory is different in +    #        our documentation from windows + +    domainzone_dn = "DC=DomainDnsZones,%s" % domaindn +    forestzone_dn = "DC=ForestDnsZones,%s" % forestdn + +    add_ldif(samdb, "provision_dnszones_partitions.ldif", { +        "DOMAINZONE_DN": domainzone_dn, +        "FORESTZONE_DN": forestzone_dn, +        }) -    dns_data = read_and_sub_file(dns_ldif, { -              "DOMAINDN": domaindn, -              "DNSDOMAIN" : dnsdomain -              }) -    samdb.add_ldif(dns_data, ["relax:0"]) +    domainzone_guid = get_domainguid(samdb, domainzone_dn) +    forestzone_guid = get_domainguid(samdb, forestzone_dn) + +    domainzone_guid = str(uuid.uuid4()) +    forestzone_guid = str(uuid.uuid4()) + +    domainzone_dns = ldb.Dn(samdb, domainzone_dn).canonical_ex_str().strip() +    forestzone_dns = ldb.Dn(samdb, forestzone_dn).canonical_ex_str().strip() + +    add_ldif(samdb, "provision_dnszones_add.ldif", { +        "DOMAINZONE_DN": domainzone_dn, +        "FORESTZONE_DN": forestzone_dn, +        "DOMAINZONE_GUID": domainzone_guid, +        "FORESTZONE_GUID": forestzone_guid, +        "DOMAINZONE_DNS": domainzone_dns, +        "FORESTZONE_DNS": forestzone_dns, +        "CONFIGDN": configdn, +        }) + +    modify_ldif(samdb, "provision_dnszones_modify.ldif", { +        "CONFIGDN": configdn, +        "SERVERDN": serverdn, +        "DOMAINZONE_DN": domainzone_dn, +        "FORESTZONE_DN": forestzone_dn, +    }) + + +def add_dns_accounts(samdb, domaindn): +    add_ldif(samdb, "provision_dns_accounts_add.ldif", { +        "DOMAINDN": domaindn, +        }) + +def add_dns_container(samdb, domaindn, prefix): +    # CN=MicrosoftDNS,<PREFIX>,<DOMAINDN> +    msg = ldb.Message(ldb.Dn(samdb, "CN=MicrosoftDNS,%s,%s" % (prefix, domaindn))) +    msg["objectClass"] = ["top", "container"] +    msg["displayName"] = ldb.MessageElement("DNS Servers", ldb.FLAG_MOD_ADD, "displayName") +    samdb.add(msg) -    soa_subrecords = [] -    dns_records = [] -    # @ entry for the domain -    at_soa_record = SOARecord(dnsname, "hostmaster.%s" % dnsdomain) -    soa_subrecords.append(ndr_pack(at_soa_record)) +def add_rootservers(samdb, domaindn, prefix): +    rootservers = {} +    rootservers["a.root-servers.net"] = "198.41.0.4" +    rootservers["b.root-servers.net"] = "192.228.79.201" +    rootservers["c.root-servers.net"] = "192.33.4.12" +    rootservers["d.root-servers.net"] = "128.8.10.90" +    rootservers["e.root-servers.net"] = "192.203.230.10" +    rootservers["f.root-servers.net"] = "192.5.5.241" +    rootservers["g.root-servers.net"] = "192.112.36.4" +    rootservers["h.root-servers.net"] = "128.63.2.53" +    rootservers["i.root-servers.net"] = "192.36.148.17" +    rootservers["j.root-servers.net"] = "192.58.128.30" +    rootservers["k.root-servers.net"] = "193.0.14.129" +    rootservers["l.root-servers.net"] = "199.7.83.42" +    rootservers["m.root-servers.net"] = "202.12.27.33" -    at_ns_record = NSRecord(dnsname) -    soa_subrecords.append(ndr_pack(at_ns_record)) +    rootservers_v6 = {} +    rootservers_v6["a.root-servers.net"] = "2001:503:ba3e::2:30" +    rootservers_v6["f.root-servers.net"] = "2001:500:2f::f" +    rootservers_v6["h.root-servers.net"] = "2001:500:1::803f:235" +    rootservers_v6["j.root-servers.net"] = "2001:503:c27::2:30" +    rootservers_v6["k.root-servers.net"] = "2001:7fd::1" +    rootservers_v6["m.root-servers.net"] = "2001:dc3::35" + +    container_dn = "DC=RootDNSServers,CN=MicrosoftDNS,%s,%s" % (prefix, domaindn) + +    # Add DC=RootDNSServers,CN=MicrosoftDNS,<PREFIX>,<DOMAINDN> +    msg = ldb.Message(ldb.Dn(samdb, container_dn)) +    msg["objectClass"] = ["top", "dnsZone"] +    samdb.add(msg) + +    # Add DC=@,DC=RootDNSServers,CN=MicrosoftDNS,<PREFIX>,<DOMAINDN> +    record = [] +    for rserver in rootservers: +        record.append(ndr_pack(RootNSRecord(rserver, serial=0, ttl=0))) + +    msg = ldb.Message(ldb.Dn(samdb, "DC=@,%s" % container_dn)) +    msg["objectClass"] = ["top", "dnsNode"] +    msg["dnsRecord"] = ldb.MessageElement(record, ldb.FLAG_MOD_ADD, "dnsRecord") +    samdb.add(msg) + +    # Add DC=<rootserver>,DC=RootDNSServers,CN=MicrosoftDNS,<PREFIX>,<DOMAINDN> +    for rserver in rootservers: +        record = [ndr_pack(ARecord(rootservers[rserver], serial=0, ttl=0))] +        # Add AAAA record as well (How does W2K* add IPv6 records?) +        #if rserver in rootservers_v6: +        #    record.append(ndr_pack(AAAARecord(rootservers_v6[rserver], serial=0, ttl=0))) +        msg = ldb.Message(ldb.Dn(samdb, "DC=%s,%s" % (rserver, container_dn))) +        msg["objectClass"] = ["top", "dnsNode"] +        msg["dnsRecord"] = ldb.MessageElement(record, ldb.FLAG_MOD_ADD, "dnsRecord") +        samdb.add(msg) + +def add_at_record(samdb, container_dn, prefix, hostname, dnsdomain, hostip, hostip6): + +    fqdn_hostname = "%s.%s" % (hostname, dnsdomain) + +    at_records = [] + +    # SOA record +    at_soa_record = SOARecord(fqdn_hostname, "hostmaster.%s" % dnsdomain) +    at_records.append(ndr_pack(at_soa_record)) + +    # NS record +    at_ns_record = NSRecord(fqdn_hostname) +    at_records.append(ndr_pack(at_ns_record))      if hostip is not None:          # A record          at_a_record = ARecord(hostip) -        dns_records.append(ndr_pack(at_a_record)) +        at_records.append(ndr_pack(at_a_record))      if hostip6 is not None: +        # AAAA record          at_aaaa_record = AAAARecord(hostip6) -        dns_records.append(ndr_pack(at_aaaa_record)) +        at_records.append(ndr_pack(at_aaaa_record)) -    msg = ldb.Message(ldb.Dn(samdb, "DC=@,DC=%s,CN=MicrosoftDNS,CN=System,%s" %\ -                                    (dnsdomain, domaindn ))) +    msg = ldb.Message(ldb.Dn(samdb, "DC=@,%s" % container_dn))      msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = ldb.MessageElement(soa_subrecords + dns_records, -                                          ldb.FLAG_MOD_ADD, "dnsRecord") +    msg["dnsRecord"] = ldb.MessageElement(at_records, ldb.FLAG_MOD_ADD, "dnsRecord")      samdb.add(msg) -    # _gc._tcp record -    gc_tcp_record = SRVRecord(dnsname, 3268) -    msg = ldb.Message(ldb.Dn(samdb, -            "DC=_gc._tcp,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                    (dnsdomain, domaindn))) +def add_srv_record(samdb, container_dn, prefix, host, port): +    srv_record = SRVRecord(host, port) +    msg = ldb.Message(ldb.Dn(samdb, "%s,%s" % (prefix, container_dn)))      msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = [ndr_pack(gc_tcp_record)] +    msg["dnsRecord"] = ldb.MessageElement(ndr_pack(srv_record), ldb.FLAG_MOD_ADD, "dnsRecord")      samdb.add(msg) -    # _gc._tcp.sitename._site record -    msg = ldb.Message(ldb.Dn(samdb, -            "DC=_gc._tcp.%s._sites,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                    (names.sitename, dnsdomain, domaindn))) +def add_ns_record(samdb, container_dn, prefix, host): +    ns_record = NSRecord(host) +    msg = ldb.Message(ldb.Dn(samdb, "%s,%s" % (prefix, container_dn)))      msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = [ndr_pack(gc_tcp_record)] +    msg["dnsRecord"] = ldb.MessageElement(ndr_pack(ns_record), ldb.FLAG_MOD_ADD, "dnsRecord")      samdb.add(msg) -    # _kerberos._tcp record -    kerberos_record = SRVRecord(dnsname, 88) -    msg = ldb.Message(ldb.Dn(samdb, -            "DC=_kerberos._tcp,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                    (dnsdomain, domaindn))) +def add_cname_record(samdb, container_dn, prefix, host): +    cname_record = CNameRecord(host) +    msg = ldb.Message(ldb.Dn(samdb, "%s,%s" % (prefix, container_dn)))      msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = [ndr_pack(kerberos_record)] +    msg["dnsRecord"] = ldb.MessageElement(ndr_pack(cname_record), ldb.FLAG_MOD_ADD, "dnsRecord")      samdb.add(msg) -    # _kerberos._tcp.sitename._site record -    msg = ldb.Message(ldb.Dn(samdb, -            "DC=_kerberos._tcp.%s._sites,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                    (site, dnsdomain, domaindn))) -    msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = [ndr_pack(kerberos_record)] -    samdb.add(msg) +def add_host_record(samdb, container_dn, prefix, hostip, hostip6): +    host_records = [] +    if hostip: +        a_record = ARecord(hostip) +        host_records.append(ndr_pack(a_record)) +    if hostip6: +        aaaa_record = AAAARecord(hostip6) +        host_records.append(ndr_pack(aaaa_record)) +    if host_records: +        msg = ldb.Message(ldb.Dn(samdb, "%s,%s" % (prefix, container_dn))) +        msg["objectClass"] = ["top", "dnsNode"] +        msg["dnsRecord"] = ldb.MessageElement(host_records, ldb.FLAG_MOD_ADD, "dnsRecord") +        samdb.add(msg) -    # _kerberos._udp record -    msg = ldb.Message(ldb.Dn(samdb, -            "DC=_kerberos._udp,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                    (dnsdomain, domaindn))) -    msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = [ndr_pack(kerberos_record)] +def add_domain_record(samdb, domaindn, prefix, dnsdomain): +    # DC=<DNSDOMAIN>,CN=MicrosoftDNS,<PREFIX>,<DOMAINDN> +    msg = ldb.Message(ldb.Dn(samdb, "DC=%s,CN=MicrosoftDNS,%s,%s" % (dnsdomain, prefix, domaindn))) +    msg["objectClass"] = ["top", "dnsZone"]      samdb.add(msg) -    # _kpasswd._tcp record -    kpasswd_record = SRVRecord(dnsname, 464) -    msg = ldb.Message(ldb.Dn(samdb, -            "DC=_kpasswd._tcp,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                    (dnsdomain, domaindn))) -    msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = [ndr_pack(kpasswd_record)] +def add_msdcs_record(samdb, forestdn, prefix, dnsforest): +    # DC=_msdcs.<DNSFOREST>,CN=MicrosoftDNS,<PREFIX>,<FORESTDN> +    msg = ldb.Message(ldb.Dn(samdb, "DC=_msdcs.%s,CN=MicrosoftDNS,%s,%s" % +                                    (dnsforest, prefix, forestdn))) +    msg["objectClass"] = ["top", "dnsZone"]      samdb.add(msg) -    # _kpasswd._udp record -    msg = ldb.Message(ldb.Dn(samdb, -            "DC=_kpasswd._udp,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                    (dnsdomain, domaindn))) -    msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = [ndr_pack(kpasswd_record)] -    samdb.add(msg) -    # _ldap._tcp record -    ldap_record = SRVRecord(dnsname, 389) -    msg = ldb.Message(ldb.Dn(samdb, -            "DC=_ldap._tcp,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                    (dnsdomain, domaindn))) -    msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = [ndr_pack(ldap_record)] -    samdb.add(msg) +def add_dc_domain_records(samdb, domaindn, prefix, site, dnsdomain, hostname, hostip, hostip6): -    # _ldap._tcp.sitename._site record -    msg = ldb.Message(ldb.Dn(samdb, -            "DC=_ldap._tcp.%s._site,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                    (site, dnsdomain, domaindn))) -    msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = [ndr_pack(ldap_record)] -    samdb.add(msg) +    fqdn_hostname = "%s.%s" % (hostname, dnsdomain) -    # _msdcs record -    msdcs_record = NSRecord(dnsname) -    msg = ldb.Message(ldb.Dn(samdb, -            "DC=_msdcs,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                    (dnsdomain, domaindn))) -    msg["objectClass"] = ["top", "dnsNode"] -    msg["dnsRecord"] = [ndr_pack(msdcs_record)] -    samdb.add(msg) +    # Set up domain container - DC=<DNSDOMAIN>,CN=MicrosoftDNS,<PREFIX>,<DOMAINDN> +    domain_container_dn = ldb.Dn(samdb, "DC=%s,CN=MicrosoftDNS,%s,%s" % +                                    (dnsdomain, prefix, domaindn)) -    # the host's own record -    # Only do this if there's IP addresses to set up. -    # This is a bit weird, but the samba4.blackbox.provision.py test apparently -    # doesn't set up any IPs -    if len(dns_records) > 0: -        msg = ldb.Message(ldb.Dn(samdb, -                "DC=%s,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                        (hostname, dnsdomain, domaindn))) -        msg["objectClass"] = ["top", "dnsNode"] -        msg["dnsRecord"] = ldb.MessageElement(dns_records, -                                              ldb.FLAG_MOD_ADD, "dnsRecord") -        samdb.add(msg) +    # DC=@ record +    add_at_record(samdb, domain_container_dn, "DC=@", hostname, dnsdomain, hostip, hostip6) -        # DomainDnsZones record -        msg = ldb.Message(ldb.Dn(samdb, -                "DC=DomainDnsZones,DC=%s,CN=MicrosoftDNS,CN=System,%s" % \ -                        (dnsdomain, domaindn))) -        msg["objectClass"] = ["top", "dnsNode"] -        msg["dnsRecord"] = ldb.MessageElement(dns_records, -                                              ldb.FLAG_MOD_ADD, "dnsRecord") +    # DC=<HOSTNAME> record +    add_host_record(samdb, domain_container_dn, "DC=%s" % hostname, hostip, hostip6) -        samdb.add(msg) +    # DC=_kerberos._tcp record +    add_srv_record(samdb, domain_container_dn, "DC=_kerberos._tcp", fqdn_hostname, 88) + +    # DC=_kerberos._tcp.<SITENAME>._sites record +    add_srv_record(samdb, domain_container_dn, "DC=_kerberos._tcp.%s._sites" % site, +                    fqdn_hostname, 88) + +    # DC=_kerberos._udp record +    add_srv_record(samdb, domain_container_dn, "DC=_kerberos._udp", fqdn_hostname, 88) + +    # DC=_kpasswd._tcp record +    add_srv_record(samdb, domain_container_dn, "DC=_kpasswd._tcp", fqdn_hostname, 464) + +    # DC=_kpasswd._udp record +    add_srv_record(samdb, domain_container_dn, "DC=_kpasswd._udp", fqdn_hostname, 464) + +    # DC=_ldap._tcp record +    add_srv_record(samdb, domain_container_dn, "DC=_ldap._tcp", fqdn_hostname, 389) + +    # DC=_ldap._tcp.<SITENAME>._sites record +    add_srv_record(samdb, domain_container_dn, "DC=_ldap._tcp.%s._sites" % site, +                    fqdn_hostname, 389) + +    # FIXME: The number of SRV records depend on the various roles this DC has. +    #        _gc and _msdcs records are added if the we are the forest dc and not subdomain dc +    # +    # Assumption: current DC is GC and add all the entries + +    # DC=_gc._tcp record +    add_srv_record(samdb, domain_container_dn, "DC=_gc._tcp", fqdn_hostname, 3268) + +    # DC=_gc._tcp.<SITENAME>,_sites record +    add_srv_record(samdb, domain_container_dn, "DC=_gc._tcp.%s._sites" % site, fqdn_hostname, 3268) + +    # DC=_msdcs record +    add_ns_record(samdb, domain_container_dn, "DC=_msdcs", fqdn_hostname) + +    # FIXME: Following entries are added only if DomainDnsZones and ForestDnsZones partitions +    #        are created +    # +    # Assumption: Additional entries won't hurt on os_level = 2000 + +    # DC=_ldap._tcp.<SITENAME>._sites.DomainDnsZones +    add_srv_record(samdb, domain_container_dn, "DC=_ldap._tcp.%s._sites.DomainDnsZones" % site, +                    fqdn_hostname, 389) + +    # DC=_ldap._tcp.<SITENAME>._sites.ForestDnsZones +    add_srv_record(samdb, domain_container_dn, "DC=_ldap._tcp.%s._sites.ForestDnsZones" % site, +                    fqdn_hostname, 389) + +    # DC=_ldap._tcp.DomainDnsZones +    add_srv_record(samdb, domain_container_dn, "DC=_ldap._tcp.DomainDnsZones", +                    fqdn_hostname, 389) + +    # DC=_ldap._tcp.ForestDnsZones +    add_srv_record(samdb, domain_container_dn, "DC=_ldap._tcp.ForestDnsZones", +                    fqdn_hostname, 389) + +    # DC=DomainDnsZones +    add_host_record(samdb, domain_container_dn, "DC=DomainDnsZones", hostip, hostip6) + +    # DC=ForestDnsZones +    add_host_record(samdb, domain_container_dn, "DC=ForestDnsZones", hostip, hostip6) + + +def add_dc_msdcs_records(samdb, forestdn, prefix, site, dnsforest, hostname, +                            hostip, hostip6, domainguid, ntdsguid): + +    fqdn_hostname = "%s.%s" % (hostname, dnsforest) + +    # Set up forest container - DC=<DNSDOMAIN>,CN=MicrosoftDNS,<PREFIX>,<DOMAINDN> +    forest_container_dn = ldb.Dn(samdb, "DC=_msdcs.%s,CN=MicrosoftDNS,%s,%s" % +                                    (dnsforest, prefix, forestdn)) + +    # DC=@ record +    add_at_record(samdb, forest_container_dn, "DC=@", hostname, dnsforest, None, None) + +    # DC=_kerberos._tcp.dc record +    add_srv_record(samdb, forest_container_dn, "DC=_kerberos._tcp.dc", fqdn_hostname, 88) + +    # DC=_kerberos._tcp.<SITENAME>._sites.dc record +    add_srv_record(samdb, forest_container_dn, "DC=_kerberos._tcp.%s._sites.dc" % site, +                    fqdn_hostname, 88) + +    # DC=_ldap._tcp.dc record +    add_srv_record(samdb, forest_container_dn, "DC=_ldap._tcp.dc", fqdn_hostname, 389) + +    # DC=_ldap._tcp.<SITENAME>._sites.dc record +    add_srv_record(samdb, forest_container_dn, "DC=_ldap._tcp.%s._sites.dc" % site, +                    fqdn_hostname, 389) + +    # DC=_ldap._tcp.<SITENAME>._sites.gc record +    add_srv_record(samdb, forest_container_dn, "DC=_ldap._tcp.%s._sites.gc" % site, +                    fqdn_hostname, 3268) + +    # DC=_ldap._tcp.gc record +    add_srv_record(samdb, forest_container_dn, "DC=_ldap._tcp.gc", fqdn_hostname, 3268) + +    # DC=_ldap._tcp.pdc record +    add_srv_record(samdb, forest_container_dn, "DC=_ldap._tcp.pdc", fqdn_hostname, 389) + +    # DC=gc record +    add_host_record(samdb, forest_container_dn, "DC=gc", hostip, hostip6) + +    # DC=_ldap._tcp.<DOMAINGUID>.domains record +    add_srv_record(samdb, forest_container_dn, "DC=_ldap._tcp.%s.domains" % domainguid, +                    fqdn_hostname, 389) + +    # DC=<NTDSGUID> +    add_cname_record(samdb, forest_container_dn, "DC=%s" % ntdsguid, fqdn_hostname) + + +def setup_ad_dns(samdb, names, logger, hostip=None, hostip6=None, dns_backend=None, +                os_level=None): +    """Provision DNS information (assuming GC role) + +    :param samdb: LDB object connected to sam.ldb file +    :param names: Names shortcut +    :param logger: Logger object +    :param hostip: IPv4 address +    :param hostip6: IPv6 address +    :param dns_backend: Type of DNS backend +    :param os_level: Functional level (treated as os level) +    """ + +    if dns_backend is None: +        dns_backend = "BIND9" +        logger.info("Assuming bind9 DNS server backend") + +    # If dns_backend is BIND9 +    #   Populate only CN=MicrosoftDNS,CN=System,<DOMAINDN> +    # +    # If dns_backend is SAMBA or BIND9_DLZ  +    #   Populate DNS partitions + +    if os_level is None: +        os_level = DS_DOMAIN_FUNCTION_2003 + +    # If os_level < 2003 (DS_DOMAIN_FUNCTION_2000) +    #   All dns records are in CN=MicrosoftDNS,CN=System,<DOMAINDN> +    # +    # If os_level >= 2003 (DS_DOMAIN_FUNCTION_2003, DS_DOMAIN_FUNCTION_2008, +    #                        DS_DOMAIN_FUNCTION_2008_R2) +    #   Root server records are in CN=MicrosoftDNS,CN=System,<DOMAINDN> +    #   Domain records are in CN=MicrosoftDNS,CN=System,<DOMAINDN> +    #   Domain records are in CN=MicrosoftDNS,DC=DomainDnsZones,<DOMAINDN> +    #   Forest records are in CN=MicrosoftDNS,DC=ForestDnsZones,<DOMAINDN> + +    domaindn = names.domaindn +    forestdn = samdb.get_root_basedn().get_linearized() + +    dnsdomain = names.dnsdomain.lower() +    dnsforest = dnsdomain + +    hostname = names.netbiosname.lower() +    site = names.sitename + +    domainguid = get_domainguid(samdb, domaindn) +    ntdsguid = get_ntdsguid(samdb, domaindn) + +    # Add dns accounts (DnsAdmins, DnsUpdateProxy) in domain +    logger.info("Adding DNS accounts") +    add_dns_accounts(samdb, domaindn) + +    logger.info("Populating CN=System,%s" % domaindn) + +    # Set up MicrosoftDNS container +    add_dns_container(samdb, domaindn, "CN=System") + +    # Add root servers +    add_rootservers(samdb, domaindn, "CN=System") + +    if os_level == DS_DOMAIN_FUNCTION_2000: + +        # Add domain record +        add_domain_record(samdb, domaindn, "CN=System", dnsdomain) + +        # Add DNS records for a DC in domain +        add_dc_domain_records(samdb, domaindn, "CN=System", site, dnsdomain, +                                hostname, hostip, hostip6) + +    elif (dns_backend == "SAMBA" or dns_backend == "BIND9_DLZ") and ( +            os_level == DS_DOMAIN_FUNCTION_2003 or +            os_level == DS_DOMAIN_FUNCTION_2008 or +            os_level == DS_DOMAIN_FUNCTION_2008_R2): + +        # Set up additional partitions (DomainDnsZones, ForstDnsZones) +        logger.info("Creating DomainDnsZones and ForestDnsZones partitions") +        setup_dns_partitions(samdb, domaindn, forestdn, names.configdn, names.serverdn) + +        ##### Set up DC=DomainDnsZones,<DOMAINDN> +        logger.info("Populating DomainDnsZones partition") + +        # Set up MicrosoftDNS container +        add_dns_container(samdb, domaindn, "DC=DomainDnsZones") + +        # Add rootserver records +        add_rootservers(samdb, domaindn, "DC=DomainDnsZones") + +        # Add domain record +        add_domain_record(samdb, domaindn, "DC=DomainDnsZones", dnsdomain) + +        # Add DNS records for a DC in domain +        add_dc_domain_records(samdb, domaindn, "DC=DomainDnsZones", site, dnsdomain, +                                hostname, hostip, hostip6) + +        ##### Set up DC=ForestDnsZones,<DOMAINDN> +        logger.info("Populating ForestDnsZones partition") + +        # Set up MicrosoftDNS container +        add_dns_container(samdb, forestdn, "DC=ForestDnsZones") +        # Add _msdcs record +        add_msdcs_record(samdb, forestdn, "DC=ForestDnsZones", dnsforest) +        # Add DNS records for a DC in forest +        add_dc_msdcs_records(samdb, forestdn, "DC=ForestDnsZones", site, dnsforest, +                                hostname, hostip, hostip6, domainguid, ntdsguid)  | 
