s4-finddc: use NBT lookup for a 1C name if joining a short domain name
[kai/samba.git] / source4 / scripting / python / samba / netcmd / join.py
1 #!/usr/bin/env python
2 #
3 # joins
4
5 # Copyright Jelmer Vernooij 2010
6 #
7 # This program is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 3 of the License, or
10 # (at your option) any later version.
11 #
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 # GNU General Public License for more details.
16 #
17 # You should have received a copy of the GNU General Public License
18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
19 #
20
21 import samba.getopt as options
22
23 from samba.net import Net, LIBNET_JOIN_AUTOMATIC
24 from samba.netcmd import Command, CommandError, Option
25 from samba.dcerpc.misc import SEC_CHAN_WKSTA, SEC_CHAN_BDC
26 from samba.join import join_rodc
27
28 class cmd_join(Command):
29     """Joins domain as either member or backup domain controller [server connection needed]"""
30
31     synopsis = "%prog join <dnsdomain> [BDC | MEMBER | RODC] [options]"
32
33     takes_optiongroups = {
34         "sambaopts": options.SambaOptions,
35         "versionopts": options.VersionOptions,
36         "credopts": options.CredentialsOptions,
37     }
38
39     takes_options = [
40         Option("--server", help="DC to join", type=str),
41         Option("--site", help="site to join", type=str),
42         ]
43
44     takes_args = ["domain", "role?"]
45
46     def run(self, domain, role=None, sambaopts=None, credopts=None,
47             versionopts=None, server=None, site=None):
48         lp = sambaopts.get_loadparm()
49         creds = credopts.get_credentials(lp)
50         net = Net(creds, lp)
51
52         if site is None:
53             site = "Default-First-Site-Name"
54
55         netbios_name = lp.get("netbios name")
56
57         if not role is None:
58             role = role.upper()
59
60         if role is None:
61             secure_channel_type = SEC_CHAN_WKSTA
62         elif role == "BDC":
63             secure_channel_type = SEC_CHAN_BDC
64         elif role == "MEMBER":
65             secure_channel_type = SEC_CHAN_WKSTA
66         elif role == "RODC":
67             join_rodc(server=server, creds=creds, lp=lp, domain=domain,
68                       site=site, netbios_name=netbios_name)
69             return
70         else:
71             raise CommandError("Invalid role %s (possible values: MEMBER, BDC, RODC)" % role)
72
73         (join_password, sid, domain_name) = net.join(domain,
74                                                      netbios_name,
75                                                      secure_channel_type,
76                                                      LIBNET_JOIN_AUTOMATIC)
77
78         self.outf.write("Joined domain %s (%s)\n" % (domain_name, sid))