r4624: Several crash fixes for DCOM
[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
36         mem_ctx = talloc_init("torture_rpc_alter_context");
37
38         printf("opening LSA connection\n");
39         status = torture_rpc_connection(&p, 
40                                         DCERPC_LSARPC_NAME, 
41                                         DCERPC_LSARPC_UUID, 
42                                         DCERPC_LSARPC_VERSION);
43         if (!NT_STATUS_IS_OK(status)) {
44                 return False;
45         }
46
47         if (!test_lsa_OpenPolicy2(p, mem_ctx, &handle)) {
48                 ret = False;
49         }
50
51         printf("Opening secondary DSSETUP context\n");
52         status = dcerpc_secondary_context(p, &p2, DCERPC_DSSETUP_UUID, DCERPC_DSSETUP_VERSION);
53         if (!NT_STATUS_IS_OK(status)) {
54                 printf("dcerpc_alter_context failed - %s\n", nt_errstr(status));
55                 return False;
56         }
57
58         printf("Opening bad secondary connection\n");
59         status = dcerpc_secondary_context(p, &p2, DCERPC_DSSETUP_UUID, DCERPC_DSSETUP_VERSION+100);
60         if (NT_STATUS_IS_OK(status)) {
61                 printf("dcerpc_alter_context with wrong version should fail\n");
62                 return False;
63         }
64
65         printf("testing DSSETUP pipe operations\n");
66         ret &= test_DsRoleGetPrimaryDomainInformation(p2, mem_ctx);
67
68         if (!test_lsa_Close(p, mem_ctx, &handle)) {
69                 ret = False;
70         }
71
72         talloc_destroy(mem_ctx);
73
74         torture_rpc_close(p);
75
76         return ret;
77 }