From b3a11c86e1ca199de9daa8c5f8d7967746db3c77 Mon Sep 17 00:00:00 2001 From: Tim Potter Date: Wed, 22 Sep 2004 04:17:25 +0000 Subject: r2492: Some cleanups. Add tests for samr_CreateDomainGroup, samr_DeleteDomainGroup, samr_QueryDomainInfo, samr_SetDomainInfo, samr_EnumDomainUsers, samr_EnumDomainAliases, samr_EnumDomainGroups. (This used to be commit 991d783c1ade19443c92b85f2b5763b5f51c6d61) --- source4/scripting/swig/torture/samr.py | 174 +++++++++++++++++++++++++++------ 1 file changed, 145 insertions(+), 29 deletions(-) (limited to 'source4/scripting/swig') diff --git a/source4/scripting/swig/torture/samr.py b/source4/scripting/swig/torture/samr.py index 8ee7dfdc34..ebb7e07d9e 100755 --- a/source4/scripting/swig/torture/samr.py +++ b/source4/scripting/swig/torture/samr.py @@ -89,9 +89,9 @@ def test_QuerySecurity(pipe, handle): s['sec_info'] = 7 s['sdbuf'] = result['sdbuf'] - result = dcerpc.samr_SetSecurity(pipe, s) + dcerpc.samr_SetSecurity(pipe, s) - result = dcerpc.samr_QuerySecurity(pipe, r) + dcerpc.samr_QuerySecurity(pipe, r) def test_GetDomPwInfo(pipe, domain): @@ -104,22 +104,24 @@ def test_GetDomPwInfo(pipe, domain): r['name']['name_size'] = 0 r['name']['name'] = domain - result = dcerpc.samr_GetDomPwInfo(pipe, r) + dcerpc.samr_GetDomPwInfo(pipe, r) r['name']['name'] = '\\\\%s' % domain - result = dcerpc.samr_GetDomPwInfo(pipe, r) + dcerpc.samr_GetDomPwInfo(pipe, r) r['name']['name'] = '\\\\__NONAME__' - result = dcerpc.samr_GetDomPwInfo(pipe, r) + dcerpc.samr_GetDomPwInfo(pipe, r) r['name']['name'] = '\\\\Builtin' - result = dcerpc.samr_GetDomPwInfo(pipe, r) + dcerpc.samr_GetDomPwInfo(pipe, r) def test_RemoveMemberFromForeignDomain(pipe, domain_handle): + print 'test samr_RemoveMemberFromForeignDomain' + r = {} r['domain_handle'] = domain_handle r['sid'] = {} @@ -128,13 +130,15 @@ def test_RemoveMemberFromForeignDomain(pipe, domain_handle): r['sid']['num_auths'] = 4 r['sid']['sub_auths'] = [7, 8, 9, 10] - result = dcerpc.samr_RemoveMemberFromForeignDomain(pipe, r) + dcerpc.samr_RemoveMemberFromForeignDomain(pipe, r) def test_CreateUser2(pipe, domain_handle): pass def test_LookupName(pipe, domain_handle, name): + print 'test samr_LookupNames' + r = {} r['domain_handle'] = domain_handle r['num_names'] = 1 @@ -150,14 +154,14 @@ def test_LookupName(pipe, domain_handle, name): try: - result = dcerpc.samr_LookupNames(pipe, r) + 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) + dcerpc.samr_LookupNames(pipe, r) return rid @@ -185,6 +189,8 @@ def test_DeleteUser_byname(pipe, domain_handle, user_name): def test_QueryUserInfo(pipe, user_handle): + print 'test samr_QueryUserInfo' + levels = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 16, 17, 20, 21] for level in levels: @@ -192,10 +198,12 @@ def test_QueryUserInfo(pipe, user_handle): r['user_handle'] = user_handle r['level'] = level - result = dcerpc.samr_QueryUserInfo(pipe, r) + dcerpc.samr_QueryUserInfo(pipe, r) def test_QueryUserInfo2(pipe, user_handle): + print 'test samr_QueryUserInfo2' + levels = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 16, 17, 20, 21] for level in levels: @@ -203,25 +211,29 @@ def test_QueryUserInfo2(pipe, user_handle): r['user_handle'] = user_handle r['level'] = level - result = dcerpc.samr_QueryUserInfo2(pipe, r) + dcerpc.samr_QueryUserInfo2(pipe, r) def test_SetUserInfo(pipe, user_handle): pass def test_GetUserPwInfo(pipe, user_handle): + print 'test samr_GetUserpwInfo' + r = {} r['user_handle'] = user_handle - result = dcerpc.samr_GetUserPwInfo(pipe, r) + dcerpc.samr_GetUserPwInfo(pipe, r) def test_TestPrivateFunctionsUser(pipe, user_handle): + print 'test samr.TestPrivateFunctionsUser' + r = {} r['user_handle'] = user_handle try: - result = dcerpc.samr_TestPrivateFunctionsUser(pipe, r) + dcerpc.samr_TestPrivateFunctionsUser(pipe, r) except dcerpc.NTSTATUS, arg: if arg[0] != 0xc0000002: raise dcerpc.NTSTATUS(arg) @@ -242,6 +254,8 @@ def test_user_ops(pipe, user_handle): def test_CreateUser(pipe, domain_handle): + print 'test samr_CreateUser' + r = {} r['domain_handle'] = domain_handle r['account_name'] = {} @@ -267,7 +281,7 @@ def test_CreateUser(pipe, domain_handle): q['user_handle'] = user_handle q['level'] = 16 - result = dcerpc.samr_QueryUserInfo(pipe, q) + dcerpc.samr_QueryUserInfo(pipe, q) test_user_ops(pipe, user_handle) @@ -294,6 +308,8 @@ def test_alias_ops(pipe, alias_handle, domain_handle, domain_sid): def test_CreateAlias(pipe, domain_handle, domain_sid): + print 'test samr_CreateAlias' + r = {} r['domain_handle'] = domain_handle r['aliasname'] = {} @@ -314,10 +330,28 @@ def test_CreateAlias(pipe, domain_handle, domain_sid): test_alias_ops(pipe, alias_handle, domain_handle, domain_sid) - return result['alias_handle'] + return alias_handle + +def test_DeleteGroup_byname(pipe, domain_handle, group_name): + + rid = test_LookupName(pipe, domain_handle, group_name) + + r = {} + r['domain_handle'] = domain_handle + r['access_mask'] = 0x02000000 + r['rid'] = rid + + result = dcerpc.samr_OpenGroup(pipe, r) + + s = {} + s['group_handle'] = result['group_handle'] + + dcerpc.samr_DeleteDomainGroup(pipe, s) def test_CreateDomainGroup(pipe, domain_handle): + print 'testing samr_CreateDomainGroup' + r = {} r['domain_handle'] = domain_handle r['name'] = {} @@ -326,27 +360,110 @@ def test_CreateDomainGroup(pipe, domain_handle): r['name']['name'] = 'samrtorturetestgroup' r['access_mask'] = 0x02000000 - result = dcerpc.samr_CreateDomGroup(pipe, r) + try: + result = dcerpc.samr_CreateDomainGroup(pipe, r) + except dcerpc.NTSTATUS, arg: + if arg[0] != 0xc0000065: + raise dcerpc.NTSTATUS(arg) - return result['group_handle'] + test_DeleteGroup_byname(pipe, domain_handle, 'samrtorturetestgroup') -def test_CreateDomainGroup(pipe, domain_handle): - pass + result = dcerpc.samr_CreateDomainGroup(pipe, r) + + return result['group_handle'] def test_QueryDomainInfo(pipe, domain_handle): - pass + + print 'testing samr_QueryDomainInfo' + + levels = [1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 12, 13] + set_ok = [1, 0, 1, 1, 0, 1, 1, 0, 1, 0, 1, 0] + for i in range(0, len(levels)): + + r = {} + r['domain_handle'] = domain_handle + r['level'] = levels[i] + + result = dcerpc.samr_QueryDomainInfo(pipe, r) + + s = {} + s['domain_handle'] = domain_handle + s['level'] = levels[i] + s['info'] = result['info'] + + try: + dcerpc.samr_SetDomainInfo(pipe, s) + except dcerpc.NTSTATUS, arg: + if set_ok[i]: + raise dcerpc.NTSTATUS(arg) + if arg[0] != 0xc0000003: + raise dcerpc.NTSTATUS(arg) + def test_QueryDomainInfo2(pipe, domain_handle): - pass + + print 'testing samr_QueryDomainInfo' + + levels = [1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 12, 13] + + for i in range(0, len(levels)): + + r = {} + r['domain_handle'] = domain_handle + r['level'] = levels[i] + + dcerpc.samr_QueryDomainInfo2(pipe, r) def test_EnumDomainUsers(pipe, domain_handle): - pass + + print 'testing samr_EnumDomainUsers' + + r = {} + r['domain_handle'] = domain_handle + r['resume_handle'] = 0 + r['acct_flags'] = 0 + r['max_size'] = -1 + + while 1: + result = dcerpc.samr_EnumDomainUsers(pipe, r) + if result['result'] == 0x00000105: + r['resume_handle'] = result['resume_handle'] + continue + break def test_EnumDomainGroups(pipe, domain_handle): - pass + + print 'testing samr_EnumDomainGroups' + + r = {} + r['domain_handle'] = domain_handle + r['resume_handle'] = 0 + r['acct_flags'] = 0 + r['max_size'] = -1 + + while 1: + result = dcerpc.samr_EnumDomainGroups(pipe, r) + if result['result'] == 0x00000105: + r['resume_handle'] = result['resume_handle'] + continue + break def test_EnumDomainAliases(pipe, domain_handle): - pass + + print 'testing samr_EnumDomainAliases' + + r = {} + r['domain_handle'] = domain_handle + r['resume_handle'] = 0 + r['acct_flags'] = 0 + r['max_size'] = -1 + + while 1: + result = dcerpc.samr_EnumDomainAliases(pipe, r) + if result['result'] == 0x00000105: + r['resume_handle'] = result['resume_handle'] + continue + break def test_QueryDisplayInfo(pipe, domain_handle): pass @@ -410,9 +527,6 @@ def test_OpenDomain(pipe, connect_handle, domain_sid): alias_handle = test_CreateAlias(pipe, domain_handle, domain_sid) - print alias_handle - sys.exit(1) - group_handle = test_CreateDomainGroup(pipe, domain_handle) test_QueryDomainInfo(pipe, domain_handle) @@ -425,6 +539,8 @@ def test_OpenDomain(pipe, connect_handle, domain_sid): test_EnumDomainAliases(pipe, domain_handle) + sys.exit(1) + test_QueryDisplayInfo(pipe, domain_handle) test_QueryDisplayInfo2(pipe, domain_handle) @@ -463,7 +579,7 @@ def test_LookupDomain(pipe, connect_handle, domain): r['domain']['name'] = None try: - result = dcerpc.samr_LookupDomain(pipe, r) + dcerpc.samr_LookupDomain(pipe, r) except dcerpc.NTSTATUS, arg: if arg[0] != 0xc000000d: raise dcerpc.NTSTATUS(arg) @@ -471,7 +587,7 @@ def test_LookupDomain(pipe, connect_handle, domain): r['domain']['name'] = 'xxNODOMAINxx' try: - result = dcerpc.samr_LookupDomain(pipe, r) + dcerpc.samr_LookupDomain(pipe, r) except dcerpc.NTSTATUS, arg: if arg[0] != 0xc00000df: raise dcerpc.NTSTATUS(arg) -- cgit