Fix formatting.
[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) Jelmer Vernooij <jelmer@samba.org> 2007-2008
6 # Copyright (C) Andrew Bartlett <abartlet@samba.org> 2008
7 #
8 # Based on the original in EJS:
9 # Copyright (C) Andrew Tridgell 2005
10 #   
11 # This program is free software; you can redistribute it and/or modify
12 # it under the terms of the GNU General Public License as published by
13 # the Free Software Foundation; either version 3 of the License, or
14 # (at your option) any later version.
15 #   
16 # This program is distributed in the hope that it will be useful,
17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19 # GNU General Public License for more details.
20 #   
21 # You should have received a copy of the GNU General Public License
22 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
23 #
24
25 import getopt
26 import optparse
27 import os, sys
28
29 # Add path to the library for in-tree use
30 sys.path.append("scripting/python")
31
32 import samba
33
34 from auth import system_session
35 import samba.getopt as options
36 import param
37 from samba.provision import (provision, 
38                              provision_paths_from_lp,
39                              FILL_FULL, FILL_NT4SYNC,
40                              FILL_DRS)
41
42 parser = optparse.OptionParser("provision [options]")
43 sambaopts = options.SambaOptions(parser)
44 parser.add_option_group(sambaopts)
45 parser.add_option_group(options.VersionOptions(parser))
46 credopts = options.CredentialsOptions(parser)
47 parser.add_option_group(credopts)
48 parser.add_option("--setupdir", type="string", metavar="DIR", 
49                 help="directory with setup files")
50 parser.add_option("--realm", type="string", metavar="REALM", help="set realm")
51 parser.add_option("--domain", type="string", metavar="DOMAIN",
52                                   help="set domain")
53 parser.add_option("--domain-guid", type="string", metavar="GUID", 
54                 help="set domainguid (otherwise random)")
55 parser.add_option("--domain-sid", type="string", metavar="SID", 
56                 help="set domainsid (otherwise random)")
57 parser.add_option("--policy-guid", type="string", metavar="GUID",
58                                   help="set policy guid")
59 parser.add_option("--host-name", type="string", metavar="HOSTNAME", 
60                 help="set hostname")
61 parser.add_option("--host-ip", type="string", metavar="IPADDRESS", 
62                 help="set ipaddress")
63 parser.add_option("--host-guid", type="string", metavar="GUID", 
64                 help="set hostguid (otherwise random)")
65 parser.add_option("--invocationid", type="string", metavar="GUID", 
66                 help="set invocationid (otherwise random)")
67 parser.add_option("--adminpass", type="string", metavar="PASSWORD", 
68                 help="choose admin password (otherwise random)")
69 parser.add_option("--krbtgtpass", type="string", metavar="PASSWORD", 
70                 help="choose krbtgt password (otherwise random)")
71 parser.add_option("--machinepass", type="string", metavar="PASSWORD", 
72                 help="choose machine password (otherwise random)")
73 parser.add_option("--dnspass", type="string", metavar="PASSWORD", 
74                 help="choose dns password (otherwise random)")
75 parser.add_option("--root", type="string", metavar="USERNAME", 
76                 help="choose 'root' unix username")
77 parser.add_option("--nobody", type="string", metavar="USERNAME", 
78                 help="choose 'nobody' user")
79 parser.add_option("--nogroup", type="string", metavar="GROUPNAME", 
80                 help="choose 'nogroup' group")
81 parser.add_option("--wheel", type="string", metavar="GROUPNAME", 
82                 help="choose 'wheel' privileged group")
83 parser.add_option("--users", type="string", metavar="GROUPNAME", 
84                 help="choose 'users' group")
85 parser.add_option("--quiet", help="Be quiet", action="store_true")
86 parser.add_option("--blank", action="store_true",
87                 help="do not add users or groups, just the structure")
88 parser.add_option("--ldap-backend", type="string", metavar="LDAPSERVER", 
89                 help="LDAP server to use for this provision")
90 parser.add_option("--ldap-backend-type", type="choice", metavar="LDAP-BACKEND-TYPE", 
91                 help="LDB mapping module to use for the LDAP backend",
92                 choices=["fedora-ds", "openldap"])
93 parser.add_option("--aci", type="string", metavar="ACI", 
94                 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")
95 parser.add_option("--server-role", type="choice", metavar="ROLE",
96                           choices=["domain controller", "member server"],
97                 help="Set server role to provision for (default standalone)")
98 parser.add_option("--partitions-only", 
99                 help="Configure Samba's partitions, but do not modify them (ie, join a BDC)", action="store_true")
100 parser.add_option("--targetdir", type="string", metavar="DIR", 
101                           help="Set target directory")
102
103 opts = parser.parse_args()[0]
104
105 def message(text):
106         """print a message if quiet is not set."""
107         if not opts.quiet:
108                 print text
109
110 if opts.realm is None or opts.domain is None:
111         if opts.realm is None:
112                 print >>sys.stderr, "No realm set"
113         if opts.domain is None:
114                 print >>sys.stderr, "No domain set"
115         parser.print_usage()
116         sys.exit(1)
117
118 # cope with an initially blank smb.conf 
119 private_dir = None
120 lp = sambaopts.get_loadparm()
121 if opts.targetdir is not None:
122     if not os.path.exists(opts.targetdir):
123         os.mkdir(opts.targetdir)
124     private_dir = os.path.join(opts.targetdir, "private")
125     if not os.path.exists(private_dir):
126         os.mkdir(private_dir)
127     lp.set("private dir", os.path.abspath(private_dir))
128     lp.set("lock dir", os.path.abspath(opts.targetdir))
129 lp.set("realm", opts.realm)
130 lp.set("workgroup", opts.domain)
131 lp.set("server role", opts.server_role or "domain controller")
132
133
134 if opts.aci is not None:
135         print "set ACI: %s" % opts.aci
136
137 paths = provision_paths_from_lp(lp, opts.realm.lower(), private_dir)
138 paths.smbconf = sambaopts.get_loadparm_path()
139
140 creds = credopts.get_credentials()
141
142 setup_dir = opts.setupdir
143 if setup_dir is None:
144         setup_dir = "setup"
145
146 samdb_fill = FILL_FULL
147 if opts.blank:
148     samdb_fill = FILL_NT4SYNC
149 elif opts.partitions_only:
150     samdb_fill = FILL_DRS
151
152 provision(lp, setup_dir, message, paths, 
153           system_session(), creds, opts.ldap_backend, 
154           samdb_fill=samdb_fill, realm=opts.realm,
155           domainguid=opts.domain_guid, domainsid=opts.domain_sid,
156           policyguid=opts.policy_guid, hostname=opts.host_name,
157           hostip=opts.host_ip, hostguid=opts.host_guid, 
158           invocationid=opts.invocationid, adminpass=opts.adminpass,
159           krbtgtpass=opts.krbtgtpass, machinepass=opts.machinepass,
160           dnspass=opts.dnspass, root=opts.root, nobody=opts.nobody,
161           nogroup=opts.nogroup, wheel=opts.wheel, users=opts.users,
162           aci=opts.aci, serverrole=opts.server_role, 
163           ldap_backend=opts.ldap_backend, 
164           ldap_backend_type=opts.ldap_backend_type)
165
166 message("To reproduce this provision, run with:")
167 def shell_escape(arg):
168     if " " in arg:
169         return '"%s"' % arg
170     return arg
171 message(" ".join([shell_escape(arg) for arg in sys.argv]))
172
173 message("All OK")