r24735: Use torture API in more places.
[kai/samba.git] / source4 / torture / rpc / dssetup.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    test suite for dssetup rpc operations
5
6    Copyright (C) Andrew Tridgell 2004
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 3 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, see <http://www.gnu.org/licenses/>.
20 */
21
22 #include "includes.h"
23 #include "torture/torture.h"
24 #include "librpc/gen_ndr/ndr_dssetup_c.h"
25 #include "torture/rpc/rpc.h"
26
27
28 BOOL test_DsRoleGetPrimaryDomainInformation(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
29 {
30         struct dssetup_DsRoleGetPrimaryDomainInformation r;
31         NTSTATUS status;
32         BOOL ret = True;
33         int i;
34
35         printf("\ntesting DsRoleGetPrimaryDomainInformation\n");
36
37         for (i=DS_ROLE_BASIC_INFORMATION; i <= DS_ROLE_OP_STATUS; i++) {
38                 r.in.level = i;
39
40                 status = dcerpc_dssetup_DsRoleGetPrimaryDomainInformation(p, mem_ctx, &r);
41                 if (!NT_STATUS_IS_OK(status)) {
42                         const char *errstr = nt_errstr(status);
43                         if (NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
44                                 errstr = dcerpc_errstr(mem_ctx, p->last_fault_code);
45                         }
46                         printf("dcerpc_dssetup_DsRoleGetPrimaryDomainInformation level %d failed - %s\n",
47                                 i, errstr);
48                         ret = False;
49                 } else if (!W_ERROR_IS_OK(r.out.result)) {
50                         printf("DsRoleGetPrimaryDomainInformation level %d failed - %s\n",
51                                 i, win_errstr(r.out.result));
52                         ret = False;
53                 }
54         }
55
56         return ret;
57 }
58
59 BOOL torture_rpc_dssetup(struct torture_context *torture)
60 {
61         NTSTATUS status;
62         struct dcerpc_pipe *p;
63         TALLOC_CTX *mem_ctx;
64         BOOL ret = True;
65
66         mem_ctx = talloc_init("torture_rpc_dssetup");
67
68         status = torture_rpc_connection(torture, &p, &ndr_table_dssetup);
69         if (!NT_STATUS_IS_OK(status)) {
70                 talloc_free(mem_ctx);
71
72                 return False;
73         }
74
75         ret &= test_DsRoleGetPrimaryDomainInformation(p, mem_ctx);
76
77         talloc_free(mem_ctx);
78
79         return ret;
80 }