diff options
| -rwxr-xr-x | source4/scripting/swig/torture/samr.py | 152 |
1 files changed, 150 insertions, 2 deletions
diff --git a/source4/scripting/swig/torture/samr.py b/source4/scripting/swig/torture/samr.py index 1f3b60eed4..abf505e919 100755 --- a/source4/scripting/swig/torture/samr.py +++ b/source4/scripting/swig/torture/samr.py @@ -77,6 +77,142 @@ def test_QuerySecurity(pipe, handle): result = dcerpc.samr_QuerySecurity(pipe, r) +def test_GetDomPwInfo(pipe, domain): + + print 'testing samr_GetDomPwInfo' + + r = {} + r['handle'] = handle + r['name'] = {} + r['name']['name_len'] = 0 + r['name']['name_size'] = 0 + r['name']['name'] = domain + + result = dcerpc.samr_GetDomPwInfo(pipe, r) + + r['name']['name'] = '\\\\%s' % domain + + result = dcerpc.samr_GetDomPwInfo(pipe, r) + + r['name']['name'] = '\\\\__NONAME__' + + result = dcerpc.samr_GetDomPwInfo(pipe, r) + + r['name']['name'] = '\\\\Builtin' + + result = dcerpc.samr_GetDomPwInfo(pipe, r) + +def test_RemoveMemberFromForeignDomain(pipe, domain_handle): + + r = {} + r['handle'] = domain_handle + r['sid'] = {} + r['sid']['sid_rev_num'] = 1 + r['sid']['id_auth'] = [1, 2, 3, 4, 5, 6] + r['sid']['num_auths'] = 4 + r['sid']['sub_auths'] = [7, 8, 9, 10] + + result = dcerpc.samr_RemoveMemberFromForeignDomain(pipe, r) + +def test_CreateUser2(pipe, domain_handle): + pass + +def test_LookupName(pipe, domain_handle, name): + + r = {} + r['handle'] = domain_handle + r['num_names'] = 1 + r['names'] = [] + r['names'].append({'name_len': 0, 'name_size': 0, 'name': name}) + + result = dcerpc.samr_LookupNames(pipe, r) + + rid = result['rids']['ids'][0] + + r['num_names'] = 2 + r['names'].append({'name_len': 0, 'name_size': 0, 'name': 'xxNONAMExx'}) + + + try: + result = dcerpc.samr_LookupNames(pipe, r) + except dcerpc.NTSTATUS, arg: + if arg[0] != 0x00000107: + raise dcerpc.NTSTATUS(arg) + + r['num_names'] = 0 + + result = dcerpc.samr_LookupNames(pipe, r) + + return rid + +def test_OpenUser_byname(pipe, domain_handle, name): + + rid = test_LookupName(pipe, domain_handle, name) + + r = {} + r['handle'] = domain_handle + r['access_mask'] = 0x02000000 + r['rid'] = rid + + result = dcerpc.samr_OpenUser(pipe, r) + + return result['acct_handle'] + +def test_DeleteUser_byname(pipe, domain_handle, name): + + user_handle = test_OpenUser_byname(pipe, domain_handle, name) + + r = {} + r['handle'] = user_handle + + dcerpc.samr_DeleteUser(pipe, r) + +def test_CreateUser(pipe, domain_handle): + + r = {} + r['handle'] = domain_handle + r['account_name'] = {} + r['account_name']['name_len'] = 0 + r['account_name']['name_size'] = 0 + r['account_name']['name'] = 'samrtorturetest' + r['access_mask'] = 0x02000000 + + try: + result = dcerpc.samr_CreateUser(pipe, r) + except dcerpc.NTSTATUS, arg: + if arg[0] == 0xc0000022: + return + elif arg[0] == 0xc0000063: + test_DeleteUser_byname(pipe, domain_handle, 'samrtorturetest') + result = dcerpc.samr_CreateUser(pipe, r) + else: + raise dcerpc.NTSTATUS(arg) + + user_handle = result['acct_handle'] + + # samr_QueryUserInfo(), etc + +def test_OpenDomain(pipe, handle, domain_sid): + + print 'testing samr_OpenDomain' + + r = {} + r['handle'] = handle + r['access_mask'] = 0x02000000 + r['sid'] = domain_sid + + result = dcerpc.samr_OpenDomain(pipe, r) + + domain_handle = result['domain_handle'] + + test_QuerySecurity(pipe, domain_handle) + + test_RemoveMemberFromForeignDomain(pipe, domain_handle) + + test_CreateUser2(pipe, domain_handle) + + test_CreateUser(pipe, domain_handle) + def test_LookupDomain(pipe, handle, domain): print 'testing samr_LookupDomain' @@ -86,9 +222,13 @@ def test_LookupDomain(pipe, handle, domain): r['domain'] = {} r['domain']['name_len'] = 0 r['domain']['name_size'] = 0 - r['domain']['name'] = domain + r['domain']['name'] = None - result = dcerpc.samr_LookupDomain(pipe, r) + try: + result = dcerpc.samr_LookupDomain(pipe, r) + except dcerpc.NTSTATUS, arg: + if arg[0] != 0xc000000d: + raise dcerpc.NTSTATUS(arg) r['domain']['name'] = 'xxNODOMAINxx' @@ -98,6 +238,14 @@ def test_LookupDomain(pipe, handle, domain): if arg[0] != 0xc00000df: raise dcerpc.NTSTATUS(arg) + r['domain']['name'] = domain + + result = dcerpc.samr_LookupDomain(pipe, r) + + test_GetDomPwInfo(pipe, domain) + + test_OpenDomain(pipe, handle, result['sid']) + def test_EnumDomains(pipe, handle): print 'testing samr_EnumDomains' |