python/samba/netcmd/drs.py

   1 # implement samba_tool drs commands
   2 #
   3 # Copyright Andrew Tridgell 2010
   4 # Copyright Andrew Bartlett 2017
   5 #
   6 # based on C implementation by Kamen Mazdrashki <kamen.mazdrashki@postpath.com>
   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 import samba.getopt as options
  22 import ldb
  23 import logging
  24 from . import common
  25 import json
  26
  27 from samba.auth import system_session
  28 from samba.netcmd import (
  29     Command,
  30     CommandError,
  31     Option,
  32     SuperCommand,
  33 )
  34 from samba.samdb import SamDB
  35 from samba import drs_utils, nttime2string, dsdb
  36 from samba.dcerpc import drsuapi, misc
  37 from samba.join import join_clone
  38 from samba.ndr import ndr_unpack
  39 from samba.dcerpc import drsblobs
  40 from samba import colour
  41 import logging
  42
  43
  44 def drsuapi_connect(ctx):
  45     '''make a DRSUAPI connection to the server'''
  46     try:
  47         (ctx.drsuapi, ctx.drsuapi_handle, ctx.bind_supported_extensions) = drs_utils.drsuapi_connect(ctx.server, ctx.lp, ctx.creds)
  48     except Exception as e:
  49         raise CommandError("DRS connection to %s failed" % ctx.server, e)
  50
  51 def samdb_connect(ctx):
  52     '''make a ldap connection to the server'''
  53     try:
  54         ctx.samdb = SamDB(url="ldap://%s" % ctx.server,
  55                           session_info=system_session(),
  56                           credentials=ctx.creds, lp=ctx.lp)
  57     except Exception as e:
  58         raise CommandError("LDAP connection to %s failed" % ctx.server, e)
  59
  60 def drs_errmsg(werr):
  61     '''return "was successful" or an error string'''
  62     (ecode, estring) = werr
  63     if ecode == 0:
  64         return "was successful"
  65     return "failed, result %u (%s)" % (ecode, estring)
  66
  67
  68
  69 def attr_default(msg, attrname, default):
  70     '''get an attribute from a ldap msg with a default'''
  71     if attrname in msg:
  72         return msg[attrname][0]
  73     return default
  74
  75
  76
  77 def drs_parse_ntds_dn(ntds_dn):
  78     '''parse a NTDS DN returning a site and server'''
  79     a = ntds_dn.split(',')
  80     if a[0] != "CN=NTDS Settings" or a[2] != "CN=Servers" or a[4] != 'CN=Sites':
  81         raise RuntimeError("bad NTDS DN %s" % ntds_dn)
  82     server = a[1].split('=')[1]
  83     site   = a[3].split('=')[1]
  84     return (site, server)
  85
  86
  87 DEFAULT_SHOWREPL_FORMAT = 'classic'
  88
  89 class cmd_drs_showrepl(Command):
  90     """Show replication status."""
  91
  92     synopsis = "%prog [<DC>] [options]"
  93
  94     takes_optiongroups = {
  95         "sambaopts": options.SambaOptions,
  96         "versionopts": options.VersionOptions,
  97         "credopts": options.CredentialsOptions,
  98     }
  99
 100     takes_options = [
 101         Option("--json", help="replication details in JSON format",
 102                dest='format', action='store_const', const='json'),
 103         Option("--summary", help=("summarize overall DRS health as seen "
 104                                   "from this server"),
 105                dest='format', action='store_const', const='summary'),
 106         Option("--pull-summary", help=("Have we successfully replicated "
 107                                        "from all relevent servers?"),
 108                dest='format', action='store_const', const='pull_summary'),
 109         Option("--notify-summary", action='store_const',
 110                const='notify_summary', dest='format',
 111                help=("Have we successfully notified all relevent servers of "
 112                      "local changes, and did they say they successfully "
 113                      "replicated?")),
 114         Option("--classic", help="print local replication details",
 115                dest='format', action='store_const', const='classic',
 116                default=DEFAULT_SHOWREPL_FORMAT),
 117         Option("-v", "--verbose", help="Be verbose", action="store_true"),
 118         Option("--color", help="Use colour output (yes|no|auto)",
 119                default='no'),
 120     ]
 121
 122     takes_args = ["DC?"]
 123
 124     def parse_neighbour(self, n):
 125         """Convert an ldb neighbour object into a python dictionary"""
 126         dsa_objectguid = str(n.source_dsa_obj_guid)
 127         d = {
 128             'NC dn': n.naming_context_dn,
 129             "DSA objectGUID": dsa_objectguid,
 130             "last attempt time": nttime2string(n.last_attempt),
 131             "last attempt message": drs_errmsg(n.result_last_attempt),
 132             "consecutive failures": n.consecutive_sync_failures,
 133             "last success": nttime2string(n.last_success),
 134             "NTDS DN": str(n.source_dsa_obj_dn),
 135             'is deleted': False
 136         }
 137
 138         try:
 139             self.samdb.search(base="<GUID=%s>" % dsa_objectguid,
 140                               scope=ldb.SCOPE_BASE,
 141                               attrs=[])
 142         except ldb.LdbError as e:
 143             (errno, _) = e.args
 144             if errno == ldb.ERR_NO_SUCH_OBJECT:
 145                 d['is deleted'] = True
 146             else:
 147                 raise
 148         try:
 149             (site, server) = drs_parse_ntds_dn(n.source_dsa_obj_dn)
 150             d["DSA"] = "%s\%s" % (site, server)
 151         except RuntimeError:
 152             pass
 153         return d
 154
 155     def print_neighbour(self, d):
 156         '''print one set of neighbour information'''
 157         self.message("%s" % d['NC dn'])
 158         if 'DSA' in d:
 159             self.message("\t%s via RPC" % d['DSA'])
 160         else:
 161             self.message("\tNTDS DN: %s" % d['NTDS DN'])
 162         self.message("\t\tDSA object GUID: %s" % d['DSA objectGUID'])
 163         self.message("\t\tLast attempt @ %s %s" % (d['last attempt time'],
 164                                                    d['last attempt message']))
 165         self.message("\t\t%u consecutive failure(s)." %
 166                      d['consecutive failures'])
 167         self.message("\t\tLast success @ %s" % d['last success'])
 168         self.message("")
 169
 170     def get_neighbours(self, info_type):
 171         req1 = drsuapi.DsReplicaGetInfoRequest1()
 172         req1.info_type = info_type
 173         try:
 174             (info_type, info) = self.drsuapi.DsReplicaGetInfo(
 175                 self.drsuapi_handle, 1, req1)
 176         except Exception as e:
 177             raise CommandError("DsReplicaGetInfo of type %u failed" % info_type, e)
 178
 179         reps = [self.parse_neighbour(n) for n in info.array]
 180         return reps
 181
 182     def run(self, DC=None, sambaopts=None,
 183             credopts=None, versionopts=None,
 184             format=DEFAULT_SHOWREPL_FORMAT,
 185             verbose=False, color='no'):
 186         self.apply_colour_choice(color)
 187         self.lp = sambaopts.get_loadparm()
 188         if DC is None:
 189             DC = common.netcmd_dnsname(self.lp)
 190         self.server = DC
 191         self.creds = credopts.get_credentials(self.lp, fallback_machine=True)
 192         self.verbose = verbose
 193
 194         output_function = {
 195             'summary': self.summary_output,
 196             'notify_summary': self.notify_summary_output,
 197             'pull_summary': self.pull_summary_output,
 198             'json': self.json_output,
 199             'classic': self.classic_output,
 200         }.get(format)
 201         if output_function is None:
 202             raise CommandError("unknown showrepl format %s" % format)
 203
 204         return output_function()
 205
 206     def json_output(self):
 207         data = self.get_local_repl_data()
 208         del data['site']
 209         del data['server']
 210         json.dump(data, self.outf, indent=2)
 211
 212     def summary_output_handler(self, typeof_output):
 213         """Print a short message if every seems fine, but print details of any
 214         links that seem broken."""
 215         failing_repsto = []
 216         failing_repsfrom = []
 217
 218         local_data = self.get_local_repl_data()
 219
 220         if typeof_output != "pull_summary":
 221             for rep in local_data['repsTo']:
 222                 if rep['is deleted']:
 223                     continue
 224                 if rep["consecutive failures"] != 0 or rep["last success"] == 0:
 225                     failing_repsto.append(rep)
 226
 227         if typeof_output != "notify_summary":
 228             for rep in local_data['repsFrom']:
 229                 if rep['is deleted']:
 230                     continue
 231                 if rep["consecutive failures"] != 0 or rep["last success"] == 0:
 232                     failing_repsfrom.append(rep)
 233
 234         if failing_repsto or failing_repsfrom:
 235             self.message(colour.c_RED("There are failing connections"))
 236             if failing_repsto:
 237                 self.message(colour.c_RED("Failing outbound connections:"))
 238                 for rep in failing_repsto:
 239                     self.print_neighbour(rep)
 240             if failing_repsfrom:
 241                 self.message(colour.c_RED("Failing inbound connection:"))
 242                 for rep in failing_repsfrom:
 243                     self.print_neighbour(rep)
 244
 245             return 1
 246
 247         self.message(colour.c_GREEN("[ALL GOOD]"))
 248
 249
 250     def summary_output(self):
 251         return self.summary_output_handler("summary")
 252
 253     def notify_summary_output(self):
 254         return self.summary_output_handler("notify_summary")
 255
 256     def pull_summary_output(self):
 257         return self.summary_output_handler("pull_summary")
 258
 259     def get_local_repl_data(self):
 260         drsuapi_connect(self)
 261         samdb_connect(self)
 262
 263         # show domain information
 264         ntds_dn = self.samdb.get_dsServiceName()
 265
 266         (site, server) = drs_parse_ntds_dn(ntds_dn)
 267         try:
 268             ntds = self.samdb.search(base=ntds_dn, scope=ldb.SCOPE_BASE, attrs=['options', 'objectGUID', 'invocationId'])
 269         except Exception as e:
 270             raise CommandError("Failed to search NTDS DN %s" % ntds_dn)
 271
 272         dsa_details = {
 273             "options": int(attr_default(ntds[0], "options", 0)),
 274             "objectGUID": self.samdb.schema_format_value(
 275                 "objectGUID", ntds[0]["objectGUID"][0]),
 276             "invocationId": self.samdb.schema_format_value(
 277                 "objectGUID", ntds[0]["invocationId"][0])
 278         }
 279
 280         conn = self.samdb.search(base=ntds_dn, expression="(objectClass=nTDSConnection)")
 281         repsfrom = self.get_neighbours(drsuapi.DRSUAPI_DS_REPLICA_INFO_NEIGHBORS)
 282         repsto = self.get_neighbours(drsuapi.DRSUAPI_DS_REPLICA_INFO_REPSTO)
 283
 284         conn_details = []
 285         for c in conn:
 286             c_rdn, sep, c_server_dn = c['fromServer'][0].partition(',')
 287             d = {
 288                 'name': str(c['name']),
 289                 'remote DN': c['fromServer'][0],
 290                 'options': int(attr_default(c, 'options', 0)),
 291                 'enabled': (attr_default(c, 'enabledConnection',
 292                                          'TRUE').upper() == 'TRUE')
 293             }
 294
 295             conn_details.append(d)
 296             try:
 297                 c_server_res = self.samdb.search(base=c_server_dn,
 298                                                  scope=ldb.SCOPE_BASE,
 299                                                  attrs=["dnsHostName"])
 300                 d['dns name'] = c_server_res[0]["dnsHostName"][0]
 301             except ldb.LdbError as e:
 302                 (errno, _) = e.args
 303                 if errno == ldb.ERR_NO_SUCH_OBJECT:
 304                     d['is deleted'] = True
 305             except KeyError:
 306                 pass
 307
 308             d['replicates NC'] = []
 309             for r in c.get('mS-DS-ReplicatesNCReason', []):
 310                 a = str(r).split(':')
 311                 d['replicates NC'].append((a[3], int(a[2])))
 312
 313         return {
 314             'dsa': dsa_details,
 315             'repsFrom': repsfrom,
 316             'repsTo': repsto,
 317             'NTDSConnections': conn_details,
 318             'site': site,
 319             'server': server
 320         }
 321
 322     def classic_output(self):
 323         data = self.get_local_repl_data()
 324         dsa_details = data['dsa']
 325         repsfrom = data['repsFrom']
 326         repsto = data['repsTo']
 327         conn_details = data['NTDSConnections']
 328         site = data['site']
 329         server = data['server']
 330
 331         self.message("%s\\%s" % (site, server))
 332         self.message("DSA Options: 0x%08x" % dsa_details["options"])
 333         self.message("DSA object GUID: %s" % dsa_details["objectGUID"])
 334         self.message("DSA invocationId: %s\n" % dsa_details["invocationId"])
 335
 336         self.message("==== INBOUND NEIGHBORS ====\n")
 337         for n in repsfrom:
 338             self.print_neighbour(n)
 339
 340         self.message("==== OUTBOUND NEIGHBORS ====\n")
 341         for n in repsto:
 342             self.print_neighbour(n)
 343
 344         reasons = ['NTDSCONN_KCC_GC_TOPOLOGY',
 345                    'NTDSCONN_KCC_RING_TOPOLOGY',
 346                    'NTDSCONN_KCC_MINIMIZE_HOPS_TOPOLOGY',
 347                    'NTDSCONN_KCC_STALE_SERVERS_TOPOLOGY',
 348                    'NTDSCONN_KCC_OSCILLATING_CONNECTION_TOPOLOGY',
 349                    'NTDSCONN_KCC_INTERSITE_GC_TOPOLOGY',
 350                    'NTDSCONN_KCC_INTERSITE_TOPOLOGY',
 351                    'NTDSCONN_KCC_SERVER_FAILOVER_TOPOLOGY',
 352                    'NTDSCONN_KCC_SITE_FAILOVER_TOPOLOGY',
 353                    'NTDSCONN_KCC_REDUNDANT_SERVER_TOPOLOGY']
 354
 355         self.message("==== KCC CONNECTION OBJECTS ====\n")
 356         for d in conn_details:
 357             self.message("Connection --")
 358             if d.get('is deleted'):
 359                 self.message("\tWARNING: Connection to DELETED server!")
 360
 361             self.message("\tConnection name: %s" % d['name'])
 362             self.message("\tEnabled        : %s" % str(d['enabled']).upper())
 363             self.message("\tServer DNS name : %s" % d['dns name'])
 364             self.message("\tServer DN name  : %s" % d['remote DN'])
 365             self.message("\t\tTransportType: RPC")
 366             self.message("\t\toptions: 0x%08X" % d['options'])
 367
 368             if d['replicates NC']:
 369                 for nc, reason in d['replicates NC']:
 370                     self.message("\t\tReplicatesNC: %s" % nc)
 371                     self.message("\t\tReason: 0x%08x" % reason)
 372                     for s in reasons:
 373                         if getattr(dsdb, s, 0) & reason:
 374                             self.message("\t\t\t%s" % s)
 375             else:
 376                 self.message("Warning: No NC replicated for Connection!")
 377
 378
 379 class cmd_drs_kcc(Command):
 380     """Trigger knowledge consistency center run."""
 381
 382     synopsis = "%prog [<DC>] [options]"
 383
 384     takes_optiongroups = {
 385         "sambaopts": options.SambaOptions,
 386         "versionopts": options.VersionOptions,
 387         "credopts": options.CredentialsOptions,
 388     }
 389
 390     takes_args = ["DC?"]
 391
 392     def run(self, DC=None, sambaopts=None,
 393             credopts=None, versionopts=None):
 394
 395         self.lp = sambaopts.get_loadparm()
 396         if DC is None:
 397             DC = common.netcmd_dnsname(self.lp)
 398         self.server = DC
 399
 400         self.creds = credopts.get_credentials(self.lp, fallback_machine=True)
 401
 402         drsuapi_connect(self)
 403
 404         req1 = drsuapi.DsExecuteKCC1()
 405         try:
 406             self.drsuapi.DsExecuteKCC(self.drsuapi_handle, 1, req1)
 407         except Exception as e:
 408             raise CommandError("DsExecuteKCC failed", e)
 409         self.message("Consistency check on %s successful." % DC)
 410
 411
 412 class cmd_drs_replicate(Command):
 413     """Replicate a naming context between two DCs."""
 414
 415     synopsis = "%prog <destinationDC> <sourceDC> <NC> [options]"
 416
 417     takes_optiongroups = {
 418         "sambaopts": options.SambaOptions,
 419         "versionopts": options.VersionOptions,
 420         "credopts": options.CredentialsOptions,
 421     }
 422
 423     takes_args = ["DEST_DC", "SOURCE_DC", "NC"]
 424
 425     takes_options = [
 426         Option("--add-ref", help="use ADD_REF to add to repsTo on source", action="store_true"),
 427         Option("--sync-forced", help="use SYNC_FORCED to force inbound replication", action="store_true"),
 428         Option("--sync-all", help="use SYNC_ALL to replicate from all DCs", action="store_true"),
 429         Option("--full-sync", help="resync all objects", action="store_true"),
 430         Option("--local", help="pull changes directly into the local database (destination DC is ignored)", action="store_true"),
 431         Option("--local-online", help="pull changes into the local database (destination DC is ignored) as a normal online replication", action="store_true"),
 432         Option("--async-op", help="use ASYNC_OP for the replication", action="store_true"),
 433         Option("--single-object", help="Replicate only the object specified, instead of the whole Naming Context (only with --local)", action="store_true"),
 434     ]
 435
 436     def drs_local_replicate(self, SOURCE_DC, NC, full_sync=False,
 437                             single_object=False,
 438                             sync_forced=False):
 439         '''replicate from a source DC to the local SAM'''
 440
 441         self.server = SOURCE_DC
 442         drsuapi_connect(self)
 443
 444         self.local_samdb = SamDB(session_info=system_session(), url=None,
 445                                  credentials=self.creds, lp=self.lp)
 446
 447         self.samdb = SamDB(url="ldap://%s" % self.server,
 448                            session_info=system_session(),
 449                            credentials=self.creds, lp=self.lp)
 450
 451         # work out the source and destination GUIDs
 452         res = self.local_samdb.search(base="", scope=ldb.SCOPE_BASE,
 453                                       attrs=["dsServiceName"])
 454         self.ntds_dn = res[0]["dsServiceName"][0]
 455
 456         res = self.local_samdb.search(base=self.ntds_dn, scope=ldb.SCOPE_BASE,
 457                                       attrs=["objectGUID"])
 458         self.ntds_guid = misc.GUID(
 459             self.samdb.schema_format_value("objectGUID",
 460                                            res[0]["objectGUID"][0]))
 461
 462         source_dsa_invocation_id = misc.GUID(self.samdb.get_invocation_id())
 463         dest_dsa_invocation_id = misc.GUID(self.local_samdb.get_invocation_id())
 464         destination_dsa_guid = self.ntds_guid
 465
 466         exop = drsuapi.DRSUAPI_EXOP_NONE
 467
 468         if single_object:
 469             exop = drsuapi.DRSUAPI_EXOP_REPL_OBJ
 470             full_sync = True
 471
 472         self.samdb.transaction_start()
 473         repl = drs_utils.drs_Replicate("ncacn_ip_tcp:%s[seal]" % self.server,
 474                                        self.lp,
 475                                        self.creds, self.local_samdb,
 476                                        dest_dsa_invocation_id)
 477
 478         # Work out if we are an RODC, so that a forced local replicate
 479         # with the admin pw does not sync passwords
 480         rodc = self.local_samdb.am_rodc()
 481         try:
 482             (num_objects, num_links) = repl.replicate(NC,
 483                                                       source_dsa_invocation_id,
 484                                                       destination_dsa_guid,
 485                                                       rodc=rodc,
 486                                                       full_sync=full_sync,
 487                                                       exop=exop,
 488                                                       sync_forced=sync_forced)
 489         except Exception as e:
 490             raise CommandError("Error replicating DN %s" % NC, e)
 491         self.samdb.transaction_commit()
 492
 493         if full_sync:
 494             self.message("Full Replication of all %d objects and %d links "
 495                          "from %s to %s was successful." %
 496                          (num_objects, num_links, SOURCE_DC,
 497                           self.local_samdb.url))
 498         else:
 499             self.message("Incremental replication of %d objects and %d links "
 500                          "from %s to %s was successful." %
 501                          (num_objects, num_links, SOURCE_DC,
 502                           self.local_samdb.url))
 503
 504     def run(self, DEST_DC, SOURCE_DC, NC,
 505             add_ref=False, sync_forced=False, sync_all=False, full_sync=False,
 506             local=False, local_online=False, async_op=False, single_object=False,
 507             sambaopts=None, credopts=None, versionopts=None):
 508
 509         self.server = DEST_DC
 510         self.lp = sambaopts.get_loadparm()
 511
 512         self.creds = credopts.get_credentials(self.lp, fallback_machine=True)
 513
 514         if local:
 515             self.drs_local_replicate(SOURCE_DC, NC, full_sync=full_sync,
 516                                      single_object=single_object,
 517                                      sync_forced=sync_forced)
 518             return
 519
 520         if local_online:
 521             server_bind = drsuapi.drsuapi("irpc:dreplsrv", lp_ctx=self.lp)
 522             server_bind_handle = misc.policy_handle()
 523         else:
 524             drsuapi_connect(self)
 525             server_bind = self.drsuapi
 526             server_bind_handle = self.drsuapi_handle
 527
 528         if not async_op:
 529             # Give the sync replication 5 minutes time
 530             server_bind.request_timeout = 5 * 60
 531
 532         samdb_connect(self)
 533
 534         # we need to find the NTDS GUID of the source DC
 535         msg = self.samdb.search(base=self.samdb.get_config_basedn(),
 536                                 expression="(&(objectCategory=server)(|(name=%s)(dNSHostName=%s)))" % (
 537             ldb.binary_encode(SOURCE_DC),
 538             ldb.binary_encode(SOURCE_DC)),
 539                                 attrs=[])
 540         if len(msg) == 0:
 541             raise CommandError("Failed to find source DC %s" % SOURCE_DC)
 542         server_dn = msg[0]['dn']
 543
 544         msg = self.samdb.search(base=server_dn, scope=ldb.SCOPE_ONELEVEL,
 545                                 expression="(|(objectCategory=nTDSDSA)(objectCategory=nTDSDSARO))",
 546                                 attrs=['objectGUID', 'options'])
 547         if len(msg) == 0:
 548             raise CommandError("Failed to find source NTDS DN %s" % SOURCE_DC)
 549         source_dsa_guid = msg[0]['objectGUID'][0]
 550         dsa_options = int(attr_default(msg, 'options', 0))
 551
 552
 553         req_options = 0
 554         if not (dsa_options & dsdb.DS_NTDSDSA_OPT_DISABLE_OUTBOUND_REPL):
 555             req_options |= drsuapi.DRSUAPI_DRS_WRIT_REP
 556         if add_ref:
 557             req_options |= drsuapi.DRSUAPI_DRS_ADD_REF
 558         if sync_forced:
 559             req_options |= drsuapi.DRSUAPI_DRS_SYNC_FORCED
 560         if sync_all:
 561             req_options |= drsuapi.DRSUAPI_DRS_SYNC_ALL
 562         if full_sync:
 563             req_options |= drsuapi.DRSUAPI_DRS_FULL_SYNC_NOW
 564         if async_op:
 565             req_options |= drsuapi.DRSUAPI_DRS_ASYNC_OP
 566
 567         try:
 568             drs_utils.sendDsReplicaSync(server_bind, server_bind_handle, source_dsa_guid, NC, req_options)
 569         except drs_utils.drsException as estr:
 570             raise CommandError("DsReplicaSync failed", estr)
 571         if async_op:
 572             self.message("Replicate from %s to %s was started." % (SOURCE_DC, DEST_DC))
 573         else:
 574             self.message("Replicate from %s to %s was successful." % (SOURCE_DC, DEST_DC))
 575
 576
 577
 578 class cmd_drs_bind(Command):
 579     """Show DRS capabilities of a server."""
 580
 581     synopsis = "%prog [<DC>] [options]"
 582
 583     takes_optiongroups = {
 584         "sambaopts": options.SambaOptions,
 585         "versionopts": options.VersionOptions,
 586         "credopts": options.CredentialsOptions,
 587     }
 588
 589     takes_args = ["DC?"]
 590
 591     def run(self, DC=None, sambaopts=None,
 592             credopts=None, versionopts=None):
 593
 594         self.lp = sambaopts.get_loadparm()
 595         if DC is None:
 596             DC = common.netcmd_dnsname(self.lp)
 597         self.server = DC
 598         self.creds = credopts.get_credentials(self.lp, fallback_machine=True)
 599
 600         drsuapi_connect(self)
 601
 602         bind_info = drsuapi.DsBindInfoCtr()
 603         bind_info.length = 28
 604         bind_info.info = drsuapi.DsBindInfo28()
 605         (info, handle) = self.drsuapi.DsBind(misc.GUID(drsuapi.DRSUAPI_DS_BIND_GUID), bind_info)
 606
 607         optmap = [
 608             ("DRSUAPI_SUPPORTED_EXTENSION_BASE",     "DRS_EXT_BASE"),
 609             ("DRSUAPI_SUPPORTED_EXTENSION_ASYNC_REPLICATION",   "DRS_EXT_ASYNCREPL"),
 610             ("DRSUAPI_SUPPORTED_EXTENSION_REMOVEAPI",    "DRS_EXT_REMOVEAPI"),
 611             ("DRSUAPI_SUPPORTED_EXTENSION_MOVEREQ_V2",   "DRS_EXT_MOVEREQ_V2"),
 612             ("DRSUAPI_SUPPORTED_EXTENSION_GETCHG_COMPRESS",   "DRS_EXT_GETCHG_DEFLATE"),
 613             ("DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V1",    "DRS_EXT_DCINFO_V1"),
 614             ("DRSUAPI_SUPPORTED_EXTENSION_RESTORE_USN_OPTIMIZATION",   "DRS_EXT_RESTORE_USN_OPTIMIZATION"),
 615             ("DRSUAPI_SUPPORTED_EXTENSION_ADDENTRY",    "DRS_EXT_ADDENTRY"),
 616             ("DRSUAPI_SUPPORTED_EXTENSION_KCC_EXECUTE",   "DRS_EXT_KCC_EXECUTE"),
 617             ("DRSUAPI_SUPPORTED_EXTENSION_ADDENTRY_V2",   "DRS_EXT_ADDENTRY_V2"),
 618             ("DRSUAPI_SUPPORTED_EXTENSION_LINKED_VALUE_REPLICATION",   "DRS_EXT_LINKED_VALUE_REPLICATION"),
 619             ("DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V2",    "DRS_EXT_DCINFO_V2"),
 620             ("DRSUAPI_SUPPORTED_EXTENSION_INSTANCE_TYPE_NOT_REQ_ON_MOD","DRS_EXT_INSTANCE_TYPE_NOT_REQ_ON_MOD"),
 621             ("DRSUAPI_SUPPORTED_EXTENSION_CRYPTO_BIND",   "DRS_EXT_CRYPTO_BIND"),
 622             ("DRSUAPI_SUPPORTED_EXTENSION_GET_REPL_INFO",   "DRS_EXT_GET_REPL_INFO"),
 623             ("DRSUAPI_SUPPORTED_EXTENSION_STRONG_ENCRYPTION",   "DRS_EXT_STRONG_ENCRYPTION"),
 624             ("DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V01",   "DRS_EXT_DCINFO_VFFFFFFFF"),
 625             ("DRSUAPI_SUPPORTED_EXTENSION_TRANSITIVE_MEMBERSHIP",  "DRS_EXT_TRANSITIVE_MEMBERSHIP"),
 626             ("DRSUAPI_SUPPORTED_EXTENSION_ADD_SID_HISTORY",   "DRS_EXT_ADD_SID_HISTORY"),
 627             ("DRSUAPI_SUPPORTED_EXTENSION_POST_BETA3",   "DRS_EXT_POST_BETA3"),
 628             ("DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V5",   "DRS_EXT_GETCHGREQ_V5"),
 629             ("DRSUAPI_SUPPORTED_EXTENSION_GET_MEMBERSHIPS2",   "DRS_EXT_GETMEMBERSHIPS2"),
 630             ("DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V6",   "DRS_EXT_GETCHGREQ_V6"),
 631             ("DRSUAPI_SUPPORTED_EXTENSION_NONDOMAIN_NCS",   "DRS_EXT_NONDOMAIN_NCS"),
 632             ("DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V8",   "DRS_EXT_GETCHGREQ_V8"),
 633             ("DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V5",   "DRS_EXT_GETCHGREPLY_V5"),
 634             ("DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V6",   "DRS_EXT_GETCHGREPLY_V6"),
 635             ("DRSUAPI_SUPPORTED_EXTENSION_ADDENTRYREPLY_V3",   "DRS_EXT_WHISTLER_BETA3"),
 636             ("DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V7",   "DRS_EXT_WHISTLER_BETA3"),
 637             ("DRSUAPI_SUPPORTED_EXTENSION_VERIFY_OBJECT",   "DRS_EXT_WHISTLER_BETA3"),
 638             ("DRSUAPI_SUPPORTED_EXTENSION_XPRESS_COMPRESS",   "DRS_EXT_W2K3_DEFLATE"),
 639             ("DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V10",   "DRS_EXT_GETCHGREQ_V10"),
 640             ("DRSUAPI_SUPPORTED_EXTENSION_RESERVED_PART2",   "DRS_EXT_RESERVED_FOR_WIN2K_OR_DOTNET_PART2"),
 641             ("DRSUAPI_SUPPORTED_EXTENSION_RESERVED_PART3", "DRS_EXT_RESERVED_FOR_WIN2K_OR_DOTNET_PART3")
 642         ]
 643
 644         optmap_ext = [
 645             ("DRSUAPI_SUPPORTED_EXTENSION_ADAM", "DRS_EXT_ADAM"),
 646             ("DRSUAPI_SUPPORTED_EXTENSION_LH_BETA2", "DRS_EXT_LH_BETA2"),
 647             ("DRSUAPI_SUPPORTED_EXTENSION_RECYCLE_BIN", "DRS_EXT_RECYCLE_BIN")]
 648
 649         self.message("Bind to %s succeeded." % DC)
 650         self.message("Extensions supported:")
 651         for (opt, str) in optmap:
 652             optval = getattr(drsuapi, opt, 0)
 653             if info.info.supported_extensions & optval:
 654                 yesno = "Yes"
 655             else:
 656                 yesno = "No "
 657             self.message("  %-60s: %s (%s)" % (opt, yesno, str))
 658
 659         if isinstance(info.info, drsuapi.DsBindInfo48):
 660             self.message("\nExtended Extensions supported:")
 661             for (opt, str) in optmap_ext:
 662                 optval = getattr(drsuapi, opt, 0)
 663                 if info.info.supported_extensions_ext & optval:
 664                     yesno = "Yes"
 665                 else:
 666                     yesno = "No "
 667                 self.message("  %-60s: %s (%s)" % (opt, yesno, str))
 668
 669         self.message("\nSite GUID: %s" % info.info.site_guid)
 670         self.message("Repl epoch: %u" % info.info.repl_epoch)
 671         if isinstance(info.info, drsuapi.DsBindInfo48):
 672             self.message("Forest GUID: %s" % info.info.config_dn_guid)
 673
 674
 675
 676 class cmd_drs_options(Command):
 677     """Query or change 'options' for NTDS Settings object of a Domain Controller."""
 678
 679     synopsis = "%prog [<DC>] [options]"
 680
 681     takes_optiongroups = {
 682         "sambaopts": options.SambaOptions,
 683         "versionopts": options.VersionOptions,
 684         "credopts": options.CredentialsOptions,
 685     }
 686
 687     takes_args = ["DC?"]
 688
 689     takes_options = [
 690         Option("--dsa-option", help="DSA option to enable/disable", type="str",
 691                metavar="{+|-}IS_GC | {+|-}DISABLE_INBOUND_REPL | {+|-}DISABLE_OUTBOUND_REPL | {+|-}DISABLE_NTDSCONN_XLATE"),
 692     ]
 693
 694     option_map = {"IS_GC": 0x00000001,
 695                   "DISABLE_INBOUND_REPL": 0x00000002,
 696                   "DISABLE_OUTBOUND_REPL": 0x00000004,
 697                   "DISABLE_NTDSCONN_XLATE": 0x00000008}
 698
 699     def run(self, DC=None, dsa_option=None,
 700             sambaopts=None, credopts=None, versionopts=None):
 701
 702         self.lp = sambaopts.get_loadparm()
 703         if DC is None:
 704             DC = common.netcmd_dnsname(self.lp)
 705         self.server = DC
 706         self.creds = credopts.get_credentials(self.lp, fallback_machine=True)
 707
 708         samdb_connect(self)
 709
 710         ntds_dn = self.samdb.get_dsServiceName()
 711         res = self.samdb.search(base=ntds_dn, scope=ldb.SCOPE_BASE, attrs=["options"])
 712         dsa_opts = int(res[0]["options"][0])
 713
 714         # print out current DSA options
 715         cur_opts = [x for x in self.option_map if self.option_map[x] & dsa_opts]
 716         self.message("Current DSA options: " + ", ".join(cur_opts))
 717
 718         # modify options
 719         if dsa_option:
 720             if dsa_option[:1] not in ("+", "-"):
 721                 raise CommandError("Unknown option %s" % dsa_option)
 722             flag = dsa_option[1:]
 723             if flag not in self.option_map.keys():
 724                 raise CommandError("Unknown option %s" % dsa_option)
 725             if dsa_option[:1] == "+":
 726                 dsa_opts |= self.option_map[flag]
 727             else:
 728                 dsa_opts &= ~self.option_map[flag]
 729             #save new options
 730             m = ldb.Message()
 731             m.dn = ldb.Dn(self.samdb, ntds_dn)
 732             m["options"] = ldb.MessageElement(str(dsa_opts), ldb.FLAG_MOD_REPLACE, "options")
 733             self.samdb.modify(m)
 734             # print out new DSA options
 735             cur_opts = [x for x in self.option_map if self.option_map[x] & dsa_opts]
 736             self.message("New DSA options: " + ", ".join(cur_opts))
 737
 738
 739 class cmd_drs_clone_dc_database(Command):
 740     """Replicate an initial clone of domain, but DO NOT JOIN it."""
 741
 742     synopsis = "%prog <dnsdomain> [options]"
 743
 744     takes_optiongroups = {
 745         "sambaopts": options.SambaOptions,
 746         "versionopts": options.VersionOptions,
 747         "credopts": options.CredentialsOptions,
 748     }
 749
 750     takes_options = [
 751         Option("--server", help="DC to join", type=str),
 752         Option("--targetdir", help="where to store provision (required)", type=str),
 753         Option("-q", "--quiet", help="Be quiet", action="store_true"),
 754         Option("--include-secrets", help="Also replicate secret values", action="store_true"),
 755         Option("-v", "--verbose", help="Be verbose", action="store_true")
 756     ]
 757
 758     takes_args = ["domain"]
 759
 760     def run(self, domain, sambaopts=None, credopts=None,
 761             versionopts=None, server=None, targetdir=None,
 762             quiet=False, verbose=False, include_secrets=False):
 763         lp = sambaopts.get_loadparm()
 764         creds = credopts.get_credentials(lp)
 765
 766         logger = self.get_logger()
 767         if verbose:
 768             logger.setLevel(logging.DEBUG)
 769         elif quiet:
 770             logger.setLevel(logging.WARNING)
 771         else:
 772             logger.setLevel(logging.INFO)
 773
 774         if targetdir is None:
 775             raise CommandError("--targetdir option must be specified")
 776
 777         join_clone(logger=logger, server=server, creds=creds, lp=lp,
 778                    domain=domain, dns_backend='SAMBA_INTERNAL',
 779                    targetdir=targetdir, include_secrets=include_secrets)
 780
 781
 782 class cmd_drs(SuperCommand):
 783     """Directory Replication Services (DRS) management."""
 784
 785     subcommands = {}
 786     subcommands["bind"] = cmd_drs_bind()
 787     subcommands["kcc"] = cmd_drs_kcc()
 788     subcommands["replicate"] = cmd_drs_replicate()
 789     subcommands["showrepl"] = cmd_drs_showrepl()
 790     subcommands["options"] = cmd_drs_options()
 791     subcommands["clone-dc-database"] = cmd_drs_clone_dc_database()