r4625: added a test that demonstrates that once a context_id is established,
[samba.git] / source4 / torture / rpc / alter_context.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    test suite for dcerpc alter_context operations
5
6    Copyright (C) Andrew Tridgell 2005
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 2 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, write to the Free Software
20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22
23 #include "includes.h"
24 #include "librpc/gen_ndr/ndr_lsa.h"
25 #include "librpc/gen_ndr/ndr_dssetup.h"
26
27
28 BOOL torture_rpc_alter_context(void)
29 {
30         NTSTATUS status;
31         struct dcerpc_pipe *p, *p2;
32         TALLOC_CTX *mem_ctx;
33         BOOL ret = True;
34         struct policy_handle handle;
35         struct dcerpc_syntax_id syntax;
36         struct dcerpc_syntax_id transfer_syntax;
37
38         mem_ctx = talloc_init("torture_rpc_alter_context");
39
40         printf("opening LSA connection\n");
41         status = torture_rpc_connection(&p, 
42                                         DCERPC_LSARPC_NAME, 
43                                         DCERPC_LSARPC_UUID, 
44                                         DCERPC_LSARPC_VERSION);
45         if (!NT_STATUS_IS_OK(status)) {
46                 return False;
47         }
48
49         if (!test_lsa_OpenPolicy2(p, mem_ctx, &handle)) {
50                 ret = False;
51         }
52
53         printf("Opening secondary DSSETUP context\n");
54         status = dcerpc_secondary_context(p, &p2, DCERPC_DSSETUP_UUID, DCERPC_DSSETUP_VERSION);
55         if (!NT_STATUS_IS_OK(status)) {
56                 printf("dcerpc_alter_context failed - %s\n", nt_errstr(status));
57                 return False;
58         }
59
60         printf("Opening bad secondary connection\n");
61         status = dcerpc_secondary_context(p, &p2, DCERPC_DSSETUP_UUID, DCERPC_DSSETUP_VERSION+100);
62         if (NT_STATUS_IS_OK(status)) {
63                 printf("dcerpc_alter_context with wrong version should fail\n");
64                 return False;
65         }
66
67         printf("testing DSSETUP pipe operations\n");
68         ret &= test_DsRoleGetPrimaryDomainInformation(p2, mem_ctx);
69
70         if (!test_lsa_Close(p, mem_ctx, &handle)) {
71                 ret = False;
72         }
73
74         syntax = p->syntax;
75         transfer_syntax = p->transfer_syntax;
76
77         printf("Testing change of primary context\n");
78         status = dcerpc_alter_context(p, mem_ctx, &p2->syntax, &p2->transfer_syntax);
79         if (!NT_STATUS_IS_OK(status)) {
80                 printf("dcerpc_alter_context failed - %s\n", nt_errstr(status));
81                 return False;
82         }
83
84         printf("testing DSSETUP pipe operations - should fault\n");
85         if (test_DsRoleGetPrimaryDomainInformation(p, mem_ctx)) {
86                 ret = False;
87         }
88
89         if (!test_lsa_OpenPolicy2(p, mem_ctx, &handle)) {
90                 ret = False;
91         }
92
93         if (!test_lsa_Close(p, mem_ctx, &handle)) {
94                 ret = False;
95         }
96
97         printf("testing DSSETUP pipe operations\n");
98         ret &= test_DsRoleGetPrimaryDomainInformation(p2, mem_ctx);
99
100         talloc_destroy(mem_ctx);
101
102         torture_rpc_close(p);
103
104         return ret;
105 }