r26523: Refactor provisioning code.
[ira/wip.git] / source / setup / provision.py
1 #!/usr/bin/python
2 #
3 # Unix SMB/CIFS implementation.
4 # provision a Samba4 server
5 # Copyright (C) Andrew Tridgell 2005
6 # Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007
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 getopt
23 import optparse
24 import sys
25
26 # Add path to the library for in-tree use
27 sys.path.append("scripting/python")
28
29 import samba
30
31 from auth import system_session
32 import samba.getopt as options
33 import param
34 from samba.provision import (provision, provision_guess, 
35                              provision_default_paths, provision_ldapbase, 
36                              provision_dns)
37
38 parser = optparse.OptionParser("provision [options]")
39 parser.add_option_group(options.SambaOptions(parser))
40 parser.add_option_group(options.VersionOptions(parser))
41 credopts = options.CredentialsOptions(parser)
42 parser.add_option_group(credopts)
43 parser.add_option("--setupdir", type="string", metavar="DIR", 
44                 help="directory with setup files")
45 parser.add_option("--realm", type="string", metavar="REALM", help="set realm")
46 parser.add_option("--domain", type="string", metavar="DOMAIN",
47                                   help="set domain")
48 parser.add_option("--domain-guid", type="string", metavar="GUID", 
49                 help="set domainguid (otherwise random)")
50 parser.add_option("--domain-sid", type="string", metavar="SID", 
51                 help="set domainsid (otherwise random)")
52 parser.add_option("--policy-guid", type="string", metavar="GUID",
53                                   help="set policy guid")
54 parser.add_option("--host-name", type="string", metavar="HOSTNAME", 
55                 help="set hostname")
56 parser.add_option("--host-ip", type="string", metavar="IPADDRESS", 
57                 help="set ipaddress")
58 parser.add_option("--host-guid", type="string", metavar="GUID", 
59                 help="set hostguid (otherwise random)")
60 parser.add_option("--invocationid", type="string", metavar="GUID", 
61                 help="set invocationid (otherwise random)")
62 parser.add_option("--adminpass", type="string", metavar="PASSWORD", 
63                 help="choose admin password (otherwise random)")
64 parser.add_option("--krbtgtpass", type="string", metavar="PASSWORD", 
65                 help="choose krbtgt password (otherwise random)")
66 parser.add_option("--machinepass", type="string", metavar="PASSWORD", 
67                 help="choose machine password (otherwise random)")
68 parser.add_option("--dnspass", type="string", metavar="PASSWORD", 
69                 help="choose dns password (otherwise random)")
70 parser.add_option("--root", type="string", metavar="USERNAME", 
71                 help="choose 'root' unix username")
72 parser.add_option("--nobody", type="string", metavar="USERNAME", 
73                 help="choose 'nobody' user")
74 parser.add_option("--nogroup", type="string", metavar="GROUPNAME", 
75                 help="choose 'nogroup' group")
76 parser.add_option("--wheel", type="string", metavar="GROUPNAME", 
77                 help="choose 'wheel' privileged group")
78 parser.add_option("--users", type="string", metavar="GROUPNAME", 
79                 help="choose 'users' group")
80 parser.add_option("--quiet", help="Be quiet", action="store_true")
81 parser.add_option("--blank", 
82                 help="do not add users or groups, just the structure")
83 parser.add_option("--ldap-base", 
84                 help="output only an LDIF file, suitable for creating an LDAP baseDN",
85         action="store_true")
86 parser.add_option("--ldap-backend", type="string", metavar="LDAPSERVER", 
87                 help="LDAP server to use for this provision")
88 parser.add_option("--ldap-module=", type="string", metavar="MODULE", 
89                 help="LDB mapping module to use for the LDAP backend")
90 parser.add_option("--aci", type="string", metavar="ACI", 
91                 help="An arbitary LDIF fragment, particularly useful to loading a backend ACI value into a target LDAP server. You must provide at least a realm and domain")
92 parser.add_option("--server-role", type="choice", metavar="ROLE",
93                           choices=["domain controller", "domain server"],
94                 help="Set server role to provision for (default standalone)")
95 parser.add_option("--partitions-only", 
96                 help="Configure Samba's partitions, but do not modify them (ie, join a BDC)", action="store_true")
97
98 opts = parser.parse_args()[0]
99
100 def message(text):
101         """print a message if quiet is not set."""
102         if not opts.quiet:
103                 print text
104
105 hostname = opts.host_name
106
107 if opts.realm is None or opts.domain is None or opts.host_name is None:
108         if opts.realm is None:
109                 print >>sys.stderr, "No realm set"
110         if opts.domain is None:
111                 print >>sys.stderr, "No domain set"
112         if opts.host_name is None:
113                 print >>sys.stderr, "No host name set"
114         parser.print_usage()
115         sys.exit(1)
116
117 # cope with an initially blank smb.conf 
118 lp = param.LoadParm()
119 lp.load(opts.configfile)
120 lp.set("realm", opts.realm)
121 lp.set("workgroup", opts.domain)
122 lp.set("server role", opts.server_role)
123
124 subobj = provision_guess(lp)
125 subobj.domain_guid = opts.domain_guid
126 subobj.host_guid = opts.host_guid
127
128 if opts.aci is not None:
129         print "set ACI: %s" % subobj.aci
130
131 print "set domain sid: %s" % subobj.domainsid
132 paths = provision_default_paths(lp, subobj)
133 paths.smbconf = opts.configfile
134 subobj.fix(paths);
135
136 if opts.ldap_backend:
137         if opts.ldap_backend == "ldapi":
138                 subobj.ldap_backend = subobj.ldapi_uri
139
140         if not opts.ldap_module:
141                 subobj.ldapmodule = "entryuuid"
142
143         subobj.domaindn_ldb = subobj.ldap_backend
144         subobj.domaindn_mod2 = ",%s,paged_searches" % subobj.ldapmodule
145         subobj.configdn_ldb = subobj.ldap_backend
146         subobj.configdn_mod2 = ",%s,paged_searches" % subobj.ldapmodule
147         subobj.schemadn_ldb = subobj.ldap_backend
148         subobj.schemadn_mod2 = ",%s,paged_searches" % subobj.ldapmodule
149         message("LDAP module: %s on backend: %s" % (subobj.ldapmodule, subobj.ldap_backend))
150
151 subobj.validate(lp)
152
153 creds = credopts.get_credentials()
154 message("Provisioning for %s in realm %s" % (subobj.domain, subobj.realm))
155 message("Using administrator password: %s" % subobj.adminpass)
156
157 setup_dir = opts.setupdir
158 if setup_dir is None:
159         setup_dir = "setup"
160 if opts.ldap_base:
161         provision_ldapbase(setup_dir, subobj, message, paths)
162         message("Please install the LDIF located in %s, %s and  into your LDAP server, and re-run with --ldap-backend=ldap://my.ldap.server" % (paths.ldap_basedn_ldif, paths.ldap_config_basedn_ldif, paths.ldap_schema_basedn_ldif))
163 elif opts.partitions_only:
164     provision_become_dc(setup_dir, subobj, message, False, 
165                         paths, lp, system_session(), creds)
166 else:
167     provision(lp, setup_dir, subobj, message, opts.blank, paths, 
168               system_session(), creds, opts.ldap_backend)
169     provision_dns(setup_dir, subobj, message, paths, system_session(), creds, lp)
170     message("To reproduce this provision, run with:")
171     def shell_escape(arg):
172         if " " in arg:
173             return '"%s"' % arg
174         return arg
175     message(" ".join([shell_escape(arg) for arg in sys.argv]))
176
177 message("All OK")