From 5efe29baed70b0cbe732350a02b24fc23016e552 Mon Sep 17 00:00:00 2001 From: Stefan Metzmacher Date: Tue, 10 Jan 2012 15:15:19 +0100 Subject: [PATCH] s4:repl_cleartext_pwd.py: add 'attmode' parameter to convert the attname to utf8 metze --- source4/scripting/devel/repl_cleartext_pwd.py | 27 ++++++++++++++++++++++----- 1 file changed, 22 insertions(+), 5 deletions(-) diff --git a/source4/scripting/devel/repl_cleartext_pwd.py b/source4/scripting/devel/repl_cleartext_pwd.py index 576f4b85f1a..d4ad6f0846d 100755 --- a/source4/scripting/devel/repl_cleartext_pwd.py +++ b/source4/scripting/devel/repl_cleartext_pwd.py @@ -76,15 +76,19 @@ def attid_equal(a1,a2): ########### main code ########### if __name__ == "__main__": - parser = OptionParser("repl_cleartext_pwd.py [options] server dn cookie_file cleartext_name [attid attname]") + parser = OptionParser("repl_cleartext_pwd.py [options] server dn cookie_file cleartext_name [attid attname attmode]") sambaopts = options.SambaOptions(parser) credopts = options.CredentialsOptions(parser) parser.add_option_group(credopts) (opts, args) = parser.parse_args() - if len(args) < 4 or len(args) == 5: - parser.error("more arguments required") + if len(args) == 4: + pass + elif len(args) >= 7: + pass + else: + parser.error("more arguments required - given=%d" % (len(args))) server = args[0] dn = args[1] @@ -92,15 +96,19 @@ if __name__ == "__main__": if len(cookie_file) == 0: cookie_file = None cleartext_name = args[3] - if len(args) >= 5: + if len(args) >= 7: try: attid = int(args[4], 16) except: attid = int(args[4]) attname = args[5] + attmode = args[6] + if attmode not in ["raw", "utf8"]: + parser.error("attmode should be 'raw' or 'utf8'") else: attid = -1 attname = None + attmode = "raw" lp = sambaopts.get_loadparm() creds = credopts.get_credentials(lp) @@ -252,7 +260,16 @@ if __name__ == "__main__": attvals = [] for j in range(0, attr.value_ctr.num_values): assert attr.value_ctr.values[j].blob is not None - attvals.append(attr.value_ctr.values[j].blob) + val_raw = attr.value_ctr.values[j].blob + val = None + if attmode == "utf8": + val_unicode = unicode(val_raw, 'utf-16-le') + val = val_unicode.encode('utf-8') + elif attmode == "raw": + val = val_raw + else: + assert False, "attmode[%s]" % attmode + attvals.append(val) if not attid_equal(attr.attid, drsuapi.DRSUAPI_ATTID_supplementalCredentials): continue assert attr.value_ctr.num_values <= 1 -- 2.11.4.GIT