e8dbdd432f6d80ebafff5637962a5e5a7161938d
[samba.git] / 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()