| 1 | #!/usr/bin/python
 | 
|---|
| 2 | #
 | 
|---|
| 3 | # Raises domain and forest function levels
 | 
|---|
| 4 | #
 | 
|---|
| 5 | # Copyright Matthias Dieter Wallnoefer 2009
 | 
|---|
| 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 sys
 | 
|---|
| 22 | 
 | 
|---|
| 23 | # Find right directory when running from source tree
 | 
|---|
| 24 | sys.path.insert(0, "bin/python")
 | 
|---|
| 25 | 
 | 
|---|
| 26 | import samba.getopt as options
 | 
|---|
| 27 | import optparse
 | 
|---|
| 28 | import ldb
 | 
|---|
| 29 | 
 | 
|---|
| 30 | from samba.auth import system_session
 | 
|---|
| 31 | from samba.samdb import SamDB
 | 
|---|
| 32 | from samba import DS_DOMAIN_FUNCTION_2000, DS_DOMAIN_FUNCTION_2003
 | 
|---|
| 33 | from samba import DS_DOMAIN_FUNCTION_2003_MIXED, DS_DOMAIN_FUNCTION_2008
 | 
|---|
| 34 | from samba import DS_DOMAIN_FUNCTION_2008_R2
 | 
|---|
| 35 | 
 | 
|---|
| 36 | parser = optparse.OptionParser("domainlevel (show | raise <options>)")
 | 
|---|
| 37 | sambaopts = options.SambaOptions(parser)
 | 
|---|
| 38 | parser.add_option_group(sambaopts)
 | 
|---|
| 39 | parser.add_option_group(options.VersionOptions(parser))
 | 
|---|
| 40 | credopts = options.CredentialsOptions(parser)
 | 
|---|
| 41 | parser.add_option_group(credopts)
 | 
|---|
| 42 | parser.add_option("-H", help="LDB URL for database or target server", type=str)
 | 
|---|
| 43 | parser.add_option("--quiet", help="Be quiet", action="store_true")
 | 
|---|
| 44 | parser.add_option("--forest",
 | 
|---|
| 45 |   help="The forest function level (2000 | 2003 | 2008 | 2008_R2). We don't support the 2003 with mixed domains (NT4 DC support) level.", type=str)
 | 
|---|
| 46 | parser.add_option("--domain",
 | 
|---|
| 47 |   help="The domain function level (2000 | 2003 | 2008 | 2008_R2). We don't support mixed/interim (NT4 DC support) levels.", type=str)
 | 
|---|
| 48 | opts, args = parser.parse_args()
 | 
|---|
| 49 | 
 | 
|---|
| 50 | #
 | 
|---|
| 51 | #  print a message if quiet is not set
 | 
|---|
| 52 | #
 | 
|---|
| 53 | def message(text):
 | 
|---|
| 54 |         if not opts.quiet:
 | 
|---|
| 55 |                 print text
 | 
|---|
| 56 | 
 | 
|---|
| 57 | if len(args) == 0:
 | 
|---|
| 58 |         parser.print_usage()
 | 
|---|
| 59 |         sys.exit(1)
 | 
|---|
| 60 | 
 | 
|---|
| 61 | lp = sambaopts.get_loadparm()
 | 
|---|
| 62 | creds = credopts.get_credentials(lp)
 | 
|---|
| 63 | 
 | 
|---|
| 64 | if opts.H is not None:
 | 
|---|
| 65 |         url = opts.H
 | 
|---|
| 66 | else:
 | 
|---|
| 67 |         url = lp.get("sam database")
 | 
|---|
| 68 | 
 | 
|---|
| 69 | samdb = SamDB(url=url, session_info=system_session(), credentials=creds, lp=lp)
 | 
|---|
| 70 | 
 | 
|---|
| 71 | domain_dn = SamDB.domain_dn(samdb)
 | 
|---|
| 72 | 
 | 
|---|
| 73 | res_forest = samdb.search("CN=Partitions,CN=Configuration," + domain_dn,
 | 
|---|
| 74 |   scope=ldb.SCOPE_BASE, attrs=["msDS-Behavior-Version"])
 | 
|---|
| 75 | assert(len(res_forest) == 1)
 | 
|---|
| 76 | 
 | 
|---|
| 77 | res_domain = samdb.search(domain_dn, scope=ldb.SCOPE_BASE,
 | 
|---|
| 78 |   attrs=["msDS-Behavior-Version", "nTMixedDomain"])
 | 
|---|
| 79 | assert(len(res_domain) == 1)
 | 
|---|
| 80 | 
 | 
|---|
| 81 | try:
 | 
|---|
| 82 |         level_forest = int(res_forest[0]["msDS-Behavior-Version"][0])
 | 
|---|
| 83 |         level_domain = int(res_domain[0]["msDS-Behavior-Version"][0])
 | 
|---|
| 84 |         level_domain_mixed = int(res_domain[0]["nTMixedDomain"][0])
 | 
|---|
| 85 | 
 | 
|---|
| 86 |         if level_forest < 0 or level_domain < 0:
 | 
|---|
| 87 |                 print "ERROR: Domain and/or forest functional level(s) is/are invalid. Correct them or reprovision!"
 | 
|---|
| 88 |                 sys.exit(1)
 | 
|---|
| 89 |         if level_forest > level_domain:
 | 
|---|
| 90 |                 print "ERROR: Forest function level is higher than the domain level(s). That can't be. Correct this or reprovision!"
 | 
|---|
| 91 |                 sys.exit(1)
 | 
|---|
| 92 | except:
 | 
|---|
| 93 |         print "ERROR: Could not retrieve the actual domain and/or forest level!"
 | 
|---|
| 94 |         if args[0] == "show":
 | 
|---|
| 95 |                 print "So the levels can't be displayed!"
 | 
|---|
| 96 |         sys.exit(1)
 | 
|---|
| 97 | 
 | 
|---|
| 98 | if args[0] == "show":
 | 
|---|
| 99 |         message("Domain and forest function level for domain '" + domain_dn + "'")
 | 
|---|
| 100 |         if level_forest == DS_DOMAIN_FUNCTION_2003_MIXED:
 | 
|---|
| 101 |                 message("\nATTENTION: You run SAMBA 4 on the 2003 with mixed domains (NT4 DC support) forest level. This isn't supported! Please raise!")
 | 
|---|
| 102 |         if (level_domain == DS_DOMAIN_FUNCTION_2000 and level_domain_mixed != 0) or level_domain == DS_DOMAIN_FUNCTION_2003_MIXED:
 | 
|---|
| 103 |                 message("\nATTENTION: You run SAMBA 4 on a mixed/interim (NT4 DC support) domain level. This isn't supported! Please raise!")
 | 
|---|
| 104 | 
 | 
|---|
| 105 |         message("")
 | 
|---|
| 106 | 
 | 
|---|
| 107 |         if level_forest == DS_DOMAIN_FUNCTION_2000:
 | 
|---|
| 108 |                 outstr = "2000"
 | 
|---|
| 109 |         elif level_forest == DS_DOMAIN_FUNCTION_2003_MIXED:
 | 
|---|
| 110 |                 outstr = "2003 with mixed domains/interim (NT4 DC support)"
 | 
|---|
| 111 |         elif level_forest == DS_DOMAIN_FUNCTION_2003:
 | 
|---|
| 112 |                 outstr = "2003"
 | 
|---|
| 113 |         elif level_forest == DS_DOMAIN_FUNCTION_2008:
 | 
|---|
| 114 |                 outstr = "2008"
 | 
|---|
| 115 |         elif level_forest == DS_DOMAIN_FUNCTION_2008_R2:
 | 
|---|
| 116 |                 outstr = "2008 R2"
 | 
|---|
| 117 |         else:
 | 
|---|
| 118 |                 outstr = "higher than 2008 R2"
 | 
|---|
| 119 |         message("Forest function level: (Windows) " + outstr)
 | 
|---|
| 120 | 
 | 
|---|
| 121 |         if level_domain == DS_DOMAIN_FUNCTION_2000 and level_domain_mixed != 0:
 | 
|---|
| 122 |                 outstr = "2000 mixed (NT4 DC support)"
 | 
|---|
| 123 |         elif level_domain == DS_DOMAIN_FUNCTION_2000 and level_domain_mixed == 0:
 | 
|---|
| 124 |                 outstr = "2000"
 | 
|---|
| 125 |         elif level_domain == DS_DOMAIN_FUNCTION_2003_MIXED:
 | 
|---|
| 126 |                 outstr = "2003 with mixed domains/interim (NT4 DC support)"
 | 
|---|
| 127 |         elif level_domain == DS_DOMAIN_FUNCTION_2003:
 | 
|---|
| 128 |                 outstr = "2003"
 | 
|---|
| 129 |         elif level_domain == DS_DOMAIN_FUNCTION_2008:
 | 
|---|
| 130 |                 outstr = "2008"
 | 
|---|
| 131 |         elif level_domain == DS_DOMAIN_FUNCTION_2008_R2:
 | 
|---|
| 132 |                 outstr = "2008 R2"
 | 
|---|
| 133 |         else:
 | 
|---|
| 134 |                 outstr = "higher than 2008 R2"
 | 
|---|
| 135 |         message("Domain function level: (Windows) " + outstr)
 | 
|---|
| 136 | 
 | 
|---|
| 137 | elif args[0] == "raise":
 | 
|---|
| 138 |         msgs = []
 | 
|---|
| 139 | 
 | 
|---|
| 140 |         if opts.domain is not None:
 | 
|---|
| 141 |                 arg = opts.domain
 | 
|---|
| 142 | 
 | 
|---|
| 143 |                 if arg == "2000":
 | 
|---|
| 144 |                         new_level_domain = DS_DOMAIN_FUNCTION_2000      
 | 
|---|
| 145 |                 elif arg == "2003":
 | 
|---|
| 146 |                         new_level_domain = DS_DOMAIN_FUNCTION_2003
 | 
|---|
| 147 |                 elif arg == "2008":
 | 
|---|
| 148 |                         new_level_domain = DS_DOMAIN_FUNCTION_2008
 | 
|---|
| 149 |                 elif arg == "2008_R2":
 | 
|---|
| 150 |                         new_level_domain = DS_DOMAIN_FUNCTION_2008_R2
 | 
|---|
| 151 |                 else:
 | 
|---|
| 152 |                         print "ERROR: Wrong argument '" + arg + "'!"
 | 
|---|
| 153 |                         sys.exit(1)
 | 
|---|
| 154 | 
 | 
|---|
| 155 |                 if new_level_domain <= level_domain and level_domain_mixed == 0:
 | 
|---|
| 156 |                         print "ERROR: Domain function level can't be smaller equal to the actual one!"
 | 
|---|
| 157 |                         sys.exit(1)
 | 
|---|
| 158 | 
 | 
|---|
| 159 |                 # Deactivate mixed/interim domain support
 | 
|---|
| 160 |                 if level_domain_mixed != 0:
 | 
|---|
| 161 |                         m = ldb.Message()
 | 
|---|
| 162 |                         m.dn = ldb.Dn(samdb, domain_dn)
 | 
|---|
| 163 |                         m["nTMixedDomain"] = ldb.MessageElement("0",
 | 
|---|
| 164 |                           ldb.FLAG_MOD_REPLACE, "nTMixedDomain")
 | 
|---|
| 165 |                         samdb.modify(m)
 | 
|---|
| 166 | 
 | 
|---|
| 167 |                 m = ldb.Message()
 | 
|---|
| 168 |                 m.dn = ldb.Dn(samdb, domain_dn)
 | 
|---|
| 169 |                 m["msDS-Behavior-Version"]= ldb.MessageElement(
 | 
|---|
| 170 |                   str(new_level_domain), ldb.FLAG_MOD_REPLACE,
 | 
|---|
| 171 |                   "msDS-Behavior-Version")
 | 
|---|
| 172 |                 samdb.modify(m)
 | 
|---|
| 173 | 
 | 
|---|
| 174 |                 level_domain = new_level_domain
 | 
|---|
| 175 | 
 | 
|---|
| 176 |                 msgs.append("Domain function level changed!")
 | 
|---|
| 177 | 
 | 
|---|
| 178 |         if opts.forest is not None:
 | 
|---|
| 179 |                 arg = opts.forest
 | 
|---|
| 180 | 
 | 
|---|
| 181 |                 if arg == "2000":
 | 
|---|
| 182 |                         new_level_forest = DS_DOMAIN_FUNCTION_2000      
 | 
|---|
| 183 |                 elif arg == "2003":
 | 
|---|
| 184 |                         new_level_forest = DS_DOMAIN_FUNCTION_2003
 | 
|---|
| 185 |                 elif arg == "2008":
 | 
|---|
| 186 |                         new_level_forest = DS_DOMAIN_FUNCTION_2008
 | 
|---|
| 187 |                 elif arg == "2008_R2":
 | 
|---|
| 188 |                         new_level_forest = DS_DOMAIN_FUNCTION_2008_R2
 | 
|---|
| 189 |                 else:
 | 
|---|
| 190 |                         print "ERROR: Wrong argument '" + arg + "'!"
 | 
|---|
| 191 |                         sys.exit(1)
 | 
|---|
| 192 | 
 | 
|---|
| 193 |                 if new_level_forest <= level_forest:
 | 
|---|
| 194 |                         print "ERROR: Forest function level can't be smaller equal to the actual one!"
 | 
|---|
| 195 |                         sys.exit(1)
 | 
|---|
| 196 | 
 | 
|---|
| 197 |                 if new_level_forest > level_domain:
 | 
|---|
| 198 |                         print "ERROR: Forest function level can't be higher than the domain function level(s). Please raise it/them first!"
 | 
|---|
| 199 |                         sys.exit(1)
 | 
|---|
| 200 | 
 | 
|---|
| 201 |                 m = ldb.Message()
 | 
|---|
| 202 |                 m.dn = ldb.Dn(samdb, "CN=Partitions,CN=Configuration,"
 | 
|---|
| 203 |                   + domain_dn)
 | 
|---|
| 204 |                 m["msDS-Behavior-Version"]= ldb.MessageElement(
 | 
|---|
| 205 |                   str(new_level_forest), ldb.FLAG_MOD_REPLACE,
 | 
|---|
| 206 |                   "msDS-Behavior-Version")
 | 
|---|
| 207 |                 samdb.modify(m)
 | 
|---|
| 208 | 
 | 
|---|
| 209 |                 msgs.append("Forest function level changed!")
 | 
|---|
| 210 | 
 | 
|---|
| 211 |         msgs.append("All changes applied successfully!")
 | 
|---|
| 212 | 
 | 
|---|
| 213 |         message("\n".join(msgs))
 | 
|---|
| 214 | else:
 | 
|---|
| 215 |         print "ERROR: Wrong argument '" + args[0] + "'!"
 | 
|---|
| 216 |         sys.exit(1)
 | 
|---|