r19392: Use torture_setting_* rather than lp_parm_* where possible.
[sfrench/samba-autobuild/.git] / source4 / torture / libnet / libnet_share.c
1 /* 
2    Unix SMB/CIFS implementation.
3    Test suite for libnet calls.
4
5    Copyright (C) Gregory LEOCADIE <gleocadie@idealx.com> 2005
6    Copyright (C) Rafal Szczesniak  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 "torture/rpc/rpc.h"
25 #include "libnet/libnet.h"
26 #include "lib/cmdline/popt_common.h"
27 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
28
29
30 #define TEST_SHARENAME "libnetsharetest"
31
32
33 static void test_displayshares(struct libnet_ListShares s)
34 {
35         int i, j;
36
37         struct share_type {
38                 enum srvsvc_ShareType type;
39                 const char *desc;
40         } share_types[] = {
41                 { STYPE_DISKTREE, "STYPE_DISKTREE" },
42                 { STYPE_DISKTREE_TEMPORARY, "STYPE_DISKTREE_TEMPORARY" },
43                 { STYPE_DISKTREE_HIDDEN, "STYPE_DISKTREE_HIDDEN" },
44                 { STYPE_PRINTQ, "STYPE_PRINTQ" },
45                 { STYPE_PRINTQ_TEMPORARY, "STYPE_PRINTQ_TEMPORARY" },
46                 { STYPE_PRINTQ_HIDDEN, "STYPE_PRINTQ_HIDDEN" },
47                 { STYPE_DEVICE, "STYPE_DEVICE" },
48                 { STYPE_DEVICE_TEMPORARY, "STYPE_DEVICE_TEMPORARY" },
49                 { STYPE_DEVICE_HIDDEN, "STYPE_DEVICE_HIDDEN" },
50                 { STYPE_IPC, "STYPE_IPC" },
51                 { STYPE_IPC_TEMPORARY, "STYPE_IPC_TEMPORARY" },
52                 { STYPE_IPC_HIDDEN, "STYPE_IPC_HIDDEN" }
53         };
54
55         switch (s.in.level) {
56         case 0:
57                 for (i = 0; i < s.out.ctr.ctr0->count; i++) {
58                         struct srvsvc_NetShareInfo0 *info = &s.out.ctr.ctr0->array[i];
59                         d_printf("\t[%d] %s\n", i, info->name);
60                 }
61                 break;
62
63         case 1:
64                 for (i = 0; i < s.out.ctr.ctr1->count; i++) {
65                         struct srvsvc_NetShareInfo1 *info = &s.out.ctr.ctr1->array[i];
66                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
67                                 if (share_types[j].type == info->type) break;
68                         }
69                         d_printf("\t[%d] %s (%s)\t%s\n", i, info->name,
70                                info->comment, share_types[j].desc);
71                 }
72                 break;
73
74         case 2:
75                 for (i = 0; i < s.out.ctr.ctr2->count; i++) {
76                         struct srvsvc_NetShareInfo2 *info = &s.out.ctr.ctr2->array[i];
77                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
78                                 if (share_types[j].type == info->type) break;
79                         }
80                         d_printf("\t[%d] %s\t%s\n\t    %s\n\t    [perms=0x%08x, max_usr=%d, cur_usr=%d, path=%s, pass=%s]\n",
81                                  i, info->name, share_types[j].desc, info->comment,
82                                  info->permissions, info->max_users,
83                                  info->current_users, info->path,
84                                  info->password);
85                 }
86                 break;
87
88         case 501:
89                 for (i = 0; i < s.out.ctr.ctr501->count; i++) {
90                         struct srvsvc_NetShareInfo501 *info = &s.out.ctr.ctr501->array[i];
91                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
92                                 if (share_types[j].type == info->type) break;
93                         }
94                         d_printf("\t[%d] %s\t%s [csc_policy=0x%08x]\n\t    %s\n", i, info->name,
95                                  share_types[j].desc, info->csc_policy,
96                                  info->comment);
97                 }
98                 break;
99
100         case 502:
101                 for (i = 0; i < s.out.ctr.ctr502->count; i++) {
102                         struct srvsvc_NetShareInfo502 *info = &s.out.ctr.ctr502->array[i];
103                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
104                                 if (share_types[j].type == info->type) break;
105                         }
106                         d_printf("\t[%d] %s\t%s\n\t    %s\n\t    [perms=0x%08x, max_usr=%d, cur_usr=%d, path=%s, pass=%s, unknown=0x%08x]\n",
107                                  i, info->name, share_types[j].desc, info->comment,
108                                  info->permissions, info->max_users,
109                                  info->current_users, info->path,
110                                  info->password, info->unknown);
111                 }
112                 break;
113         }
114 }
115
116
117 BOOL torture_listshares(struct torture_context *torture)
118 {
119         struct libnet_ListShares share;
120         NTSTATUS  status;
121         uint32_t levels[] = { 0, 1, 2, 501, 502 };
122         int i;
123         BOOL ret = True;
124         struct libnet_context* libnetctx;
125         const char *binding;
126         struct dcerpc_binding *bind;
127         TALLOC_CTX *mem_ctx;
128
129         mem_ctx = talloc_init("test_listshares");
130         binding = torture_setting_string(torture, "binding", NULL);
131         status = dcerpc_parse_binding(mem_ctx, binding, &bind);
132         if (!NT_STATUS_IS_OK(status)) {
133                 printf("Error while parsing the binding string\n");
134                 ret = False;
135                 goto done;
136         }
137
138         libnetctx = libnet_context_init(NULL);
139         if (!libnetctx) {
140                 printf("Couldn't allocate libnet context\n");
141                 ret = False;
142                 goto done;
143         }
144
145         libnetctx->cred = cmdline_credentials;
146         
147         printf("Testing libnet_ListShare\n");
148         
149         share.in.server_name = talloc_asprintf(mem_ctx, "%s", bind->host);
150
151         for (i = 0; i < ARRAY_SIZE(levels); i++) {
152                 share.in.level = levels[i];
153                 printf("testing libnet_ListShare level %u\n", share.in.level);
154
155                 status = libnet_ListShares(libnetctx, mem_ctx, &share);
156                 if (!NT_STATUS_IS_OK(status)) {
157                         printf("libnet_ListShare level %u failed - %s\n", share.in.level, share.out.error_string);
158                         ret = False;
159                         goto done;
160                 }
161
162                 printf("listing shares:\n");
163                 test_displayshares(share);
164         }
165
166 done:
167         talloc_free(mem_ctx);
168         return ret;
169 }
170
171
172 static BOOL test_addshare(struct dcerpc_pipe *pipe, TALLOC_CTX *mem_ctx, const char *host,
173                           const char* share)
174 {
175         NTSTATUS status;
176         struct srvsvc_NetShareAdd add;
177         struct srvsvc_NetShareInfo2 i;
178         
179         i.name         = share;
180         i.type         = STYPE_DISKTREE;
181         i.path         = "C:\\WINDOWS\\TEMP";
182         i.max_users    = 5;
183         i.comment      = "Comment to the test share";
184         i.password     = NULL;
185         i.permissions  = 0x0;
186
187         add.in.server_unc = host;
188         add.in.level      = 2;
189         add.in.info.info2 = &i;
190
191         status = dcerpc_srvsvc_NetShareAdd(pipe, mem_ctx, &add);
192         if (!NT_STATUS_IS_OK(status)) {
193                 printf("Failed to add a new share\n");
194                 return False;
195         }
196
197         printf("share added\n");
198         return True;
199 }
200
201
202 BOOL torture_delshare(struct torture_context *torture)
203 {
204         struct dcerpc_pipe *p;
205         struct dcerpc_binding *bind;
206         struct libnet_context* libnetctx;
207         const char *host, *binding;
208         TALLOC_CTX *mem_ctx;
209         NTSTATUS  status;
210         BOOL ret = True;
211         struct libnet_DelShare share;
212         
213         mem_ctx = talloc_init("test_listshares");
214         host = torture_setting_string(torture, "host", NULL);
215         binding = torture_setting_string(torture, "binding", NULL);
216         status = dcerpc_parse_binding(mem_ctx, binding, &bind);
217         if (!NT_STATUS_IS_OK(status)) {
218                 printf("Error while parsing the binding string\n");
219                 ret = False;
220                 goto done;
221         }
222
223         libnetctx = libnet_context_init(NULL);
224         libnetctx->cred = cmdline_credentials;
225
226         status = torture_rpc_connection(mem_ctx,
227                                         &p,
228                                         &dcerpc_table_srvsvc);
229
230         if (!test_addshare(p, mem_ctx, host, TEST_SHARENAME)) {
231                 ret = False;
232                 goto done;
233         }
234
235         share.in.server_name    = bind->host;
236         share.in.share_name     = TEST_SHARENAME;
237
238         status = libnet_DelShare(libnetctx, mem_ctx, &share);
239         if (!NT_STATUS_IS_OK(status)) {
240                 ret = False;
241                 goto done;
242         }
243
244
245 done:
246         talloc_free(mem_ctx);
247         return ret;
248 }