source4/scripting/bin/samba_spnupdate

   1 #!/usr/bin/env python
   2 #
   3 # update our servicePrincipalName names from spn_update_list
   4 #
   5 # Copyright (C) Andrew Tridgell 2010
   6 # Copyright (C) Matthieu Patou <mat@matws.net> 2012
   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 import os, sys, re
  23
  24 # ensure we get messages out immediately, so they get in the samba logs,
  25 # and don't get swallowed by a timeout
  26 os.environ['PYTHONUNBUFFERED'] = '1'
  27
  28 # forcing GMT avoids a problem in some timezones with kerberos. Both MIT
  29 # heimdal can get mutual authentication errors due to the 24 second difference
  30 # between UTC and GMT when using some zone files (eg. the PDT zone from
  31 # the US)
  32 os.environ["TZ"] = "GMT"
  33
  34 # Find right directory when running from source tree
  35 sys.path.insert(0, "bin/python")
  36
  37 import samba, ldb
  38 import optparse
  39 from samba import Ldb
  40 from samba import getopt as options
  41 from samba.auth import system_session
  42 from samba.samdb import SamDB
  43 from samba.credentials import Credentials, DONT_USE_KERBEROS
  44
  45 parser = optparse.OptionParser("samba_spnupdate")
  46 sambaopts = options.SambaOptions(parser)
  47 parser.add_option_group(sambaopts)
  48 parser.add_option_group(options.VersionOptions(parser))
  49 parser.add_option("--verbose", action="store_true")
  50
  51 credopts = options.CredentialsOptions(parser)
  52 parser.add_option_group(credopts)
  53
  54 ccachename = None
  55
  56 opts, args = parser.parse_args()
  57
  58 if len(args) != 0:
  59     parser.print_usage()
  60     sys.exit(1)
  61
  62 lp = sambaopts.get_loadparm()
  63 creds = credopts.get_credentials(lp)
  64
  65 domain = lp.get("realm")
  66 host = lp.get("netbios name")
  67
  68
  69 # get the list of substitution vars
  70 def get_subst_vars(samdb):
  71     global lp
  72     vars = {}
  73
  74     vars['DNSDOMAIN'] = samdb.domain_dns_name()
  75     vars['DNSFOREST'] = samdb.forest_dns_name()
  76     vars['HOSTNAME']  = samdb.host_dns_name()
  77     vars['NETBIOSNAME'] = lp.get('netbios name').upper()
  78     vars['WORKGROUP'] = lp.get('workgroup')
  79     vars['NTDSGUID']  = samdb.get_ntds_GUID()
  80     res = samdb.search(base=samdb.get_default_basedn(), scope=ldb.SCOPE_BASE, attrs=["objectGUID"])
  81     guid = samdb.schema_format_value("objectGUID", res[0]['objectGUID'][0])
  82     vars['DOMAINGUID'] = guid
  83     return vars
  84
  85 try:
  86     private_dir = lp.get("private dir")
  87     secrets_path = os.path.join(private_dir, "secrets.ldb")
  88
  89     secrets_db = Ldb(url=secrets_path, session_info=system_session(),
  90                      credentials=creds, lp=lp)
  91     res = secrets_db.search(base=None,
  92                             expression="(&(objectclass=ldapSecret)(cn=SAMDB Credentials))",
  93                             attrs=["samAccountName", "secret"])
  94
  95     if len(res) == 1:
  96         credentials = Credentials()
  97         credentials.set_kerberos_state(DONT_USE_KERBEROS)
  98
  99         if "samAccountName" in res[0]:
 100             credentials.set_username(res[0]["samAccountName"][0])
 101
 102         if "secret" in res[0]:
 103             credentials.set_password(res[0]["secret"][0])
 104
 105     else:
 106         credentials = None
 107
 108     samdb = SamDB(url=lp.samdb_url(), session_info=system_session(), credentials=credentials, lp=lp)
 109 except ldb.LdbError, (num, msg):
 110     print("Unable to open sam database %s : %s" % (lp.samdb_url(), msg))
 111     sys.exit(1)
 112
 113
 114 # get the substitution dictionary
 115 sub_vars = get_subst_vars(samdb)
 116
 117 # get the list of SPN entries we should have
 118 spn_update_list = lp.private_path('spn_update_list')
 119
 120 file = open(spn_update_list, "r")
 121
 122 spn_list = []
 123
 124 has_forest_dns = False
 125 has_domain_dns = False
 126 # check if we "are DNS server"
 127 res = samdb.search(base=samdb.get_config_basedn(),
 128                    expression='(objectguid=%s)' % sub_vars['NTDSGUID'],
 129                    attrs=["msDS-hasMasterNCs"])
 130
 131 basedn = str(samdb.get_default_basedn())
 132 if len(res) == 1:
 133     for e in res[0]["msDS-hasMasterNCs"]:
 134         if str(e) == "DC=DomainDnsZones,%s" % basedn:
 135             has_domain_dns = True
 136         if str(e) == "DC=ForestDnsZones,%s" % basedn:
 137             has_forest_dns = True
 138
 139
 140 # build the spn list
 141 for line in file:
 142     line = line.strip()
 143     if line == '' or line[0] == "#":
 144         continue
 145     if re.match(r".*/DomainDnsZones\..*", line) and not has_domain_dns:
 146         continue
 147     if re.match(r".*/ForestDnsZones\..*", line) and not has_forest_dns:
 148         continue
 149     line = samba.substitute_var(line, sub_vars)
 150     spn_list.append(line)
 151
 152 # get the current list of SPNs in our sam
 153 res = samdb.search(base=samdb.get_default_basedn(),
 154                    expression='(&(objectClass=computer)(samaccountname=%s$))' % sub_vars['NETBIOSNAME'],
 155                    attrs=["servicePrincipalName"])
 156 if not res or len(res) != 1:
 157     print("Failed to find computer object for %s$" % sub_vars['NETBIOSNAME'])
 158     sys.exit(1)
 159
 160 machine_dn = res[0]["dn"]
 161
 162 old_spns = []
 163 if "servicePrincipalName" in res[0]:
 164     for s in res[0]["servicePrincipalName"]:
 165         old_spns.append(s)
 166
 167 if opts.verbose:
 168     print("Existing SPNs: %s" % old_spns)
 169
 170 add_list = []
 171
 172 # work out what needs to be added
 173 for s in spn_list:
 174     in_list = False
 175     for s2 in old_spns:
 176         if s2.upper() == s.upper():
 177             in_list = True
 178             break
 179     if not in_list:
 180         add_list.append(s)
 181
 182 if opts.verbose:
 183     print("New SPNs: %s" % add_list)
 184
 185 if add_list == []:
 186     if opts.verbose:
 187         print("Nothing to add")
 188     sys.exit(0)
 189
 190 def local_update(add_list):
 191     '''store locally'''
 192     global res
 193     msg = ldb.Message()
 194     msg.dn = res[0]['dn']
 195     msg[""] = ldb.MessageElement(add_list,
 196                                  ldb.FLAG_MOD_ADD, "servicePrincipalName")
 197     res = samdb.modify(msg)
 198
 199 def call_rodc_update(d):
 200     '''RODCs need to use the writeSPN DRS call'''
 201     global lp, sub_vars
 202     from samba import drs_utils
 203     from samba.dcerpc import drsuapi, nbt
 204     from samba.net import Net
 205
 206     if opts.verbose:
 207         print("Using RODC SPN update")
 208
 209     creds = credopts.get_credentials(lp)
 210     creds.set_machine_account(lp)
 211
 212     net = Net(creds=creds, lp=lp)
 213     try:
 214         cldap_ret = net.finddc(domain=domain, flags=nbt.NBT_SERVER_DS | nbt.NBT_SERVER_WRITABLE)
 215     except Exception, reason:
 216         print("Unable to find writeable DC for domain '%s' to send DRS writeSPN to : %s" % (domain, reason))
 217         sys.exit(1)
 218     server = cldap_ret.pdc_dns_name
 219     try:
 220         binding_options = "seal"
 221         if int(lp.get("log level")) >= 5:
 222             binding_options += ",print"
 223         drs = drsuapi.drsuapi('ncacn_ip_tcp:%s[%s]' % (server, binding_options), lp, creds)
 224         (drs_handle, supported_extensions) = drs_utils.drs_DsBind(drs)
 225     except Exception, reason:
 226         print("Unable to connect to DC '%s' for domain '%s' : %s" % (server, domain, reason))
 227         sys.exit(1)
 228     req1 = drsuapi.DsWriteAccountSpnRequest1()
 229     req1.operation = drsuapi.DRSUAPI_DS_SPN_OPERATION_ADD
 230     req1.object_dn = str(machine_dn)
 231     req1.count = 0
 232     spn_names = []
 233     for n in add_list:
 234         if n.find('E3514235-4B06-11D1-AB04-00C04FC2DCD2') != -1:
 235             # this one isn't allowed for RODCs, but we don't know why yet
 236             continue
 237         ns = drsuapi.DsNameString()
 238         ns.str = n
 239         spn_names.append(ns)
 240         req1.count = req1.count + 1
 241     if spn_names == []:
 242         return
 243     req1.spn_names = spn_names
 244     (level, res) = drs.DsWriteAccountSpn(drs_handle, 1, req1)
 245     if (res.status != (0, 'WERR_OK')):
 246         print "WriteAccountSpn has failed with error %s" % str(res.status)
 247
 248 if samdb.am_rodc():
 249     call_rodc_update(add_list)
 250 else:
 251     local_update(add_list)