Merge branch 'master' of git://git.samba.org/samba
[ira/wip.git] / source4 / scripting / python / samba / netcmd / domainlevel.py
1 #!/usr/bin/python
2 #
3 # Raises domain and forest function levels
4 #
5 # Copyright Matthias Dieter Wallnoefer 2009
6 # Copyright Jelmer Vernooij 2009
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 # Notice: At the moment we have some more checks to do here on the special
23 # attributes (consider attribute "msDS-Behavior-Version). This is due to the
24 # fact that we on s4 LDB don't implement their change policy (only certain
25 # values, only increments possible...) yet.
26
27 import samba.getopt as options
28 import ldb
29
30 from samba.auth import system_session
31 from samba.netcmd import (
32     Command,
33     CommandError,
34     Option,
35     )
36 from samba.samdb import SamDB
37 from samba import (
38     DS_DOMAIN_FUNCTION_2000,
39     DS_DOMAIN_FUNCTION_2003,
40     DS_DOMAIN_FUNCTION_2003_MIXED,
41     DS_DOMAIN_FUNCTION_2008,
42     DS_DOMAIN_FUNCTION_2008_R2,
43     DS_DC_FUNCTION_2000,
44     DS_DC_FUNCTION_2003,
45     DS_DC_FUNCTION_2008,
46     DS_DC_FUNCTION_2008_R2,
47     )
48
49 class cmd_domainlevel(Command):
50     """Raises domain and forest function levels."""
51
52     synopsis = "(show | raise <options>)"
53
54     takes_optiongroups = {
55         "sambaopts": options.SambaOptions,
56         "credopts": options.CredentialsOptions,
57         "versionopts": options.VersionOptions,
58         }
59
60     takes_options = [
61         Option("-H", help="LDB URL for database or target server", type=str),
62         Option("--quiet", help="Be quiet", action="store_true"),
63         Option("--forest", type="choice", choices=["2003", "2008", "2008_R2"],
64             help="The forest function level (2003 | 2008 | 2008_R2)"),
65         Option("--domain", type="choice", choices=["2003", "2008", "2008_R2"],
66             help="The domain function level (2003 | 2008 | 2008_R2)"),
67         ]
68
69     takes_args = ["subcommand"]
70
71     def run(self, subcommand, H=None, forest=None, domain=None, quiet=False,
72             credopts=None, sambaopts=None, versionopts=None):
73         lp = sambaopts.get_loadparm()
74         creds = credopts.get_credentials(lp)
75
76         if H is not None:
77             url = H
78         else:
79             url = lp.get("sam database")
80
81         samdb = SamDB(url=url, session_info=system_session(),
82             credentials=creds, lp=lp)
83
84         domain_dn = SamDB.domain_dn(samdb)
85
86         res_forest = samdb.search("CN=Partitions,CN=Configuration," + domain_dn,
87           scope=ldb.SCOPE_BASE, attrs=["msDS-Behavior-Version"])
88         assert(len(res_forest) == 1)
89
90         res_domain = samdb.search(domain_dn, scope=ldb.SCOPE_BASE,
91           attrs=["msDS-Behavior-Version", "nTMixedDomain"])
92         assert(len(res_domain) == 1)
93
94         res_dc_s = samdb.search("CN=Sites,CN=Configuration," + domain_dn,
95           scope=ldb.SCOPE_SUBTREE, expression="(objectClass=nTDSDSA)",
96           attrs=["msDS-Behavior-Version"])
97         assert(len(res_dc_s) >= 1)
98
99         try:
100             level_forest = int(res_forest[0]["msDS-Behavior-Version"][0])
101             level_domain = int(res_domain[0]["msDS-Behavior-Version"][0])
102             level_domain_mixed = int(res_domain[0]["nTMixedDomain"][0])
103
104             min_level_dc = int(res_dc_s[0]["msDS-Behavior-Version"][0]) # Init value
105             for msg in res_dc_s:
106                 if int(msg["msDS-Behavior-Version"][0]) < min_level_dc:
107                     min_level_dc = int(msg["msDS-Behavior-Version"][0])
108
109             if level_forest < 0 or level_domain < 0:
110                 raise CommandError("Domain and/or forest function level(s) is/are invalid. Correct them or reprovision!")
111             if min_level_dc < 0:
112                 raise CommandError("Lowest function level of a DC is invalid. Correct this or reprovision!")
113             if level_forest > level_domain:
114                 raise CommandError("Forest function level is higher than the domain level(s). Correct this or reprovision!")
115             if level_domain > min_level_dc:
116                 raise CommandError("Domain function level is higher than the lowest function level of a DC. Correct this or reprovision!")
117
118         except KeyError:
119             raise CommandError("Could not retrieve the actual domain, forest level and/or lowest DC function level!")
120
121         if subcommand == "show":
122             self.message("Domain and forest function level for domain '" + domain_dn + "'")
123             if level_forest < DS_DOMAIN_FUNCTION_2003:
124                 self.message("\nATTENTION: You run SAMBA 4 on a forest function level lower than Windows 2003 (Native). This isn't supported! Please raise!")
125             if level_domain < DS_DOMAIN_FUNCTION_2003:
126                 self.message("\nATTENTION: You run SAMBA 4 on a domain function level lower than Windows 2003 (Native). This isn't supported! Please raise!")
127             if min_level_dc < DS_DC_FUNCTION_2003:
128                 self.message("\nATTENTION: You run SAMBA 4 on a lowest function level of a DC lower than Windows 2003. This isn't supported! Please step-up or upgrade the concerning DC(s)!")
129
130             self.message("")
131
132             if level_forest == DS_DOMAIN_FUNCTION_2000:
133                 outstr = "2000"
134             elif level_forest == DS_DOMAIN_FUNCTION_2003_MIXED:
135                 outstr = "2003 with mixed domains/interim (NT4 DC support)"
136             elif level_forest == DS_DOMAIN_FUNCTION_2003:
137                 outstr = "2003"
138             elif level_forest == DS_DOMAIN_FUNCTION_2008:
139                 outstr = "2008"
140             elif level_forest == DS_DOMAIN_FUNCTION_2008_R2:
141                 outstr = "2008 R2"
142             else:
143                 outstr = "higher than 2008 R2"
144             self.message("Forest function level: (Windows) " + outstr)
145
146             if level_domain == DS_DOMAIN_FUNCTION_2000 and level_domain_mixed != 0:
147                 outstr = "2000 mixed (NT4 DC support)"
148             elif level_domain == DS_DOMAIN_FUNCTION_2000 and level_domain_mixed == 0:
149                 outstr = "2000"
150             elif level_domain == DS_DOMAIN_FUNCTION_2003_MIXED:
151                 outstr = "2003 with mixed domains/interim (NT4 DC support)"
152             elif level_domain == DS_DOMAIN_FUNCTION_2003:
153                 outstr = "2003"
154             elif level_domain == DS_DOMAIN_FUNCTION_2008:
155                 outstr = "2008"
156             elif level_domain == DS_DOMAIN_FUNCTION_2008_R2:
157                 outstr = "2008 R2"
158             else:
159                 outstr = "higher than 2008 R2"
160             self.message("Domain function level: (Windows) " + outstr)
161
162             if min_level_dc == DS_DC_FUNCTION_2000:
163                 outstr = "2000"
164             elif min_level_dc == DS_DC_FUNCTION_2003:
165                 outstr = "2003"
166             elif min_level_dc == DS_DC_FUNCTION_2008:
167                 outstr = "2008"
168             elif min_level_dc == DS_DC_FUNCTION_2008_R2:
169                 outstr = "2008 R2"
170             else:
171                 outstr = "higher than 2008 R2"
172             self.message("Lowest function level of a DC: (Windows) " + outstr)
173
174         elif subcommand == "raise":
175             msgs = []
176
177             if domain is not None:
178                 if domain == "2003":
179                     new_level_domain = DS_DOMAIN_FUNCTION_2003
180                 elif domain == "2008":
181                     new_level_domain = DS_DOMAIN_FUNCTION_2008
182                 elif domain == "2008_R2":
183                     new_level_domain = DS_DOMAIN_FUNCTION_2008_R2
184
185                 if new_level_domain <= level_domain and level_domain_mixed == 0:
186                     raise CommandError("Domain function level can't be smaller equal to the actual one!")
187
188                 if new_level_domain > min_level_dc:
189                     raise CommandError("Domain function level can't be higher than the lowest function level of a DC!")
190
191                 # Deactivate mixed/interim domain support
192                 if level_domain_mixed != 0:
193                     m = ldb.Message()
194                     m.dn = ldb.Dn(samdb, domain_dn)
195                     m["nTMixedDomain"] = ldb.MessageElement("0",
196                       ldb.FLAG_MOD_REPLACE, "nTMixedDomain")
197                     samdb.modify(m)
198                 m = ldb.Message()
199                 m.dn = ldb.Dn(samdb, domain_dn)
200                 m["msDS-Behavior-Version"]= ldb.MessageElement(
201                   str(new_level_domain), ldb.FLAG_MOD_REPLACE,
202                           "msDS-Behavior-Version")
203                 samdb.modify(m)
204                 level_domain = new_level_domain
205                 msgs.append("Domain function level changed!")
206
207             if forest is not None:
208                 if forest == "2003":
209                     new_level_forest = DS_DOMAIN_FUNCTION_2003
210                 elif forest == "2008":
211                     new_level_forest = DS_DOMAIN_FUNCTION_2008
212                 elif forest == "2008_R2":
213                     new_level_forest = DS_DOMAIN_FUNCTION_2008_R2
214                 if new_level_forest <= level_forest:
215                     raise CommandError("Forest function level can't be smaller equal to the actual one!")
216                 if new_level_forest > level_domain:
217                     raise CommandError("Forest function level can't be higher than the domain function level(s). Please raise it/them first!")
218                 m = ldb.Message()
219                 m.dn = ldb.Dn(samdb, "CN=Partitions,CN=Configuration,"
220                   + domain_dn)
221                 m["msDS-Behavior-Version"]= ldb.MessageElement(
222                   str(new_level_forest), ldb.FLAG_MOD_REPLACE,
223                           "msDS-Behavior-Version")
224                 samdb.modify(m)
225                 msgs.append("Forest function level changed!")
226             msgs.append("All changes applied successfully!")
227             self.message("\n".join(msgs))
228         else:
229             raise CommandError("Wrong argument '%s'!" % subcommand)