s4-provision: Provision DNS records with correct rank
[rusty/samba.git] / source4 / scripting / python / samba / provision / sambadns.py
index 6b8561e1b53ecb25ee1ad848fdc916c4c53a045c..6e58f07e18db73b664fed423cc5a3e234bcb9c00 100644 (file)
@@ -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
 """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 = samdb.get_config_basedn()
+
+    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):
+    def __init__(self, ip_addr, serial=1, ttl=900, rank=dnsp.DNS_RANK_ZONE):
         super(ARecord, self).__init__()
         self.wType = dnsp.DNS_TYPE_A
+        self.rank = rank
         self.dwSerial = serial
         self.dwTtlSeconds = ttl
         self.data = ip_addr
 
 class AAAARecord(dnsp.DnssrvRpcRecord):
-    def __init__(self, ip6_addr, serial=1, ttl=3600):
+    def __init__(self, ip6_addr, serial=1, ttl=900, rank=dnsp.DNS_RANK_ZONE):
         super(AAAARecord, self).__init__()
         self.wType = dnsp.DNS_TYPE_AAAA
+        self.rank = rank
         self.dwSerial = serial
         self.dwTtlSeconds = ttl
         self.data = ip6_addr
 
+class CNameRecord(dnsp.DnssrvRpcRecord):
+    def __init__(self, cname, serial=1, ttl=900, rank=dnsp.DNS_RANK_ZONE):
+        super(CNameRecord, self).__init__()
+        self.wType = dnsp.DNS_TYPE_CNAME
+        self.rank = rank
+        self.dwSerial = serial
+        self.dwTtlSeconds = ttl
+        self.data = cname
+
 class NSRecord(dnsp.DnssrvRpcRecord):
-    def __init__(self, dns_server, serial=1, ttl=3600):
+    def __init__(self, dns_server, serial=1, ttl=900, rank=dnsp.DNS_RANK_ZONE):
         super(NSRecord, self).__init__()
         self.wType = dnsp.DNS_TYPE_NS
+        self.rank = rank
         self.dwSerial = serial
         self.dwTtlSeconds = ttl
         self.data = dns_server
 
 class SOARecord(dnsp.DnssrvRpcRecord):
     def __init__(self, mname, rname, serial=1, refresh=900, retry=600,
-                 expire=86400, minimum=3600, ttl=3600):
+                 expire=86400, minimum=3600, ttl=3600, rank=dnsp.DNS_RANK_ZONE):
         super(SOARecord, self).__init__()
         self.wType = dnsp.DNS_TYPE_SOA
+        self.rank = rank
         self.dwSerial = serial
         self.dwTtlSeconds = ttl
         soa = dnsp.soa()
@@ -67,9 +118,11 @@ class SOARecord(dnsp.DnssrvRpcRecord):
         self.data = soa
 
 class SRVRecord(dnsp.DnssrvRpcRecord):
-    def __init__(self, target, port, priority=0, weight=0, serial=1, ttl=3600):
+    def __init__(self, target, port, priority=0, weight=100, serial=1, ttl=900,
+                rank=dnsp.DNS_RANK_ZONE):
         super(SRVRecord, self).__init__()
         self.wType = dnsp.DNS_TYPE_SRV
+        self.rank = rank
         self.dwSerial = serial
         self.dwTtlSeconds = ttl
         srv = dnsp.srv()
@@ -79,153 +132,436 @@ 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,
+        })
+
+    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,
+        "SERVERDN": serverdn,
+        })
+
+    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)
+
+
+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"
+
+    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)
 
-    dns_data = read_and_sub_file(dns_ldif, {
-              "DOMAINDN": domaindn,
-              "DNSDOMAIN" : dnsdomain
-              })
-    samdb.add_ldif(dns_data, ["relax:0"])
+    # 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(NSRecord(rserver, serial=0, ttl=0, rank=dnsp.DNS_RANK_ROOT_HINT)))
+
+    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, rank=dnsp.DNS_RANK_ROOT_HINT))]
+        # 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)
 
-    soa_subrecords = []
-    dns_records = []
+def add_at_record(samdb, container_dn, prefix, hostname, dnsdomain, hostip, hostip6):
 
-    # @ entry for the domain
-    at_soa_record = SOARecord(dnsname, "hostmaster.%s" % dnsdomain)
-    soa_subrecords.append(ndr_pack(at_soa_record))
+    fqdn_hostname = "%s.%s" % (hostname, dnsdomain)
 
-    at_ns_record = NSRecord(dnsname)
-    soa_subrecords.append(ndr_pack(at_ns_record))
+    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_ns_glue_record(samdb, container_dn, prefix, host):
+    ns_record = NSRecord(host, rank=dnsp.DNS_RANK_NS_GLUE)
+    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(ns_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)))
+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._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)]
-    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)
 
-    # _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_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._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)]
+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)
 
-    # _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)
 
-    # _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)
+def add_dc_domain_records(samdb, domaindn, prefix, site, dnsdomain, hostname, hostip, hostip6):
 
-    # _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)
+    fqdn_hostname = "%s.%s" % (hostname, dnsdomain)
 
-    # 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)
+    # 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))
 
-        # 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=@ record
+    add_at_record(samdb, domain_container_dn, "DC=@", hostname, dnsdomain, hostip, hostip6)
 
-        samdb.add(msg)
+    # DC=<HOSTNAME> record
+    add_host_record(samdb, domain_container_dn, "DC=%s" % hostname, hostip, hostip6)
+
+    # 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_glue_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 is_valid_dns_backend(dns_backend):
+        return dns_backend in ("BIND9_FLATFILE", "BIND9_DLZ", "SAMBA_INTERNAL", "NONE")
+
+
+def is_valid_os_level(os_level):
+    return DS_DOMAIN_FUNCTION_2000 <= os_level <= DS_DOMAIN_FUNCTION_2008_R2
+
+
+def setup_ad_dns(samdb, names, logger, dns_backend, os_level, hostip=None,
+                 hostip6=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 dns_backend: Type of DNS backend
+    :param os_level: Functional level (treated as os level)
+    :param hostip: IPv4 address
+    :param hostip6: IPv6 address
+    """
+
+    if not is_valid_dns_backend(dns_backend):
+        raise Exception("Invalid dns backend: %r" % dns_backend)
+
+    if not is_valid_os_level(os_level):
+        raise Exception("Invalid os level: %r" % os_level)
+
+    if dns_backend is "NONE":
+        logger.info("No DNS backend set, not configuring DNS")
+        return
+
+    # If dns_backend is BIND9_FLATFILE
+    #   Populate only CN=MicrosoftDNS,CN=System,<DOMAINDN>
+    #
+    # If dns_backend is SAMBA_INTERNAL or BIND9_DLZ
+    #   Populate DNS partitions
+
+    # 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=MicrosoftDNS,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 in ("SAMBA_INTERNAL", "BIND9_DLZ") and \
+            os_level >= DS_DOMAIN_FUNCTION_2003:
+
+        # 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)