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 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/rpc/torture_rpc.h"
  24 #include "libnet/libnet.h"
  25 #include "lib/cmdline/cmdline.h"
  26 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
  27 #include "torture/libnet/proto.h"
  28
  29
  30 #define TEST_SHARENAME "libnetsharetest"
  31
  32
  33 static void test_displayshares(struct torture_context *tctx,
  34                                struct libnet_ListShares s)
  35 {
  36         int i, j;
  37
  38         struct share_type {
  39                 enum srvsvc_ShareType type;
  40                 const char *desc;
  41         } share_types[] = {
  42                 { STYPE_DISKTREE, "STYPE_DISKTREE" },
  43                 { STYPE_DISKTREE_TEMPORARY, "STYPE_DISKTREE_TEMPORARY" },
  44                 { STYPE_DISKTREE_HIDDEN, "STYPE_DISKTREE_HIDDEN" },
  45                 { STYPE_PRINTQ, "STYPE_PRINTQ" },
  46                 { STYPE_PRINTQ_TEMPORARY, "STYPE_PRINTQ_TEMPORARY" },
  47                 { STYPE_PRINTQ_HIDDEN, "STYPE_PRINTQ_HIDDEN" },
  48                 { STYPE_DEVICE, "STYPE_DEVICE" },
  49                 { STYPE_DEVICE_TEMPORARY, "STYPE_DEVICE_TEMPORARY" },
  50                 { STYPE_DEVICE_HIDDEN, "STYPE_DEVICE_HIDDEN" },
  51                 { STYPE_IPC, "STYPE_IPC" },
  52                 { STYPE_IPC_TEMPORARY, "STYPE_IPC_TEMPORARY" },
  53                 { STYPE_IPC_HIDDEN, "STYPE_IPC_HIDDEN" }
  54         };
  55
  56         switch (s.in.level) {
  57         case 0:
  58                 for (i = 0; i < s.out.ctr.ctr0->count; i++) {
  59                         struct srvsvc_NetShareInfo0 *info = &s.out.ctr.ctr0->array[i];
  60                         torture_comment(tctx, "\t[%d] %s\n", i, info->name);
  61                 }
  62                 break;
  63
  64         case 1:
  65                 for (i = 0; i < s.out.ctr.ctr1->count; i++) {
  66                         struct srvsvc_NetShareInfo1 *info = &s.out.ctr.ctr1->array[i];
  67                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
  68                                 if (share_types[j].type == info->type) {
  69                                         torture_comment(tctx,
  70                                                         "\t[%d] %s (%s)\t%s\n",
  71                                                         i,
  72                                                         info->name,
  73                                                         info->comment,
  74                                                         share_types[j].desc);
  75                                         break;
  76                                 }
  77                         }
  78                 }
  79                 break;
  80
  81         case 2:
  82                 for (i = 0; i < s.out.ctr.ctr2->count; i++) {
  83                         struct srvsvc_NetShareInfo2 *info = &s.out.ctr.ctr2->array[i];
  84                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
  85                                 if (share_types[j].type == info->type) {
  86                                         torture_comment(tctx,
  87                                                         "\t[%d] %s\t%s\n"
  88                                                         "\t    %s\n"
  89                                                         "\t    [perms=0x%08x, "
  90                                                         "max_usr=%d, "
  91                                                         "cur_usr=%d, "
  92                                                         "path=%s, "
  93                                                         "pass=%s]\n",
  94                                                         i,
  95                                                         info->name,
  96                                                         share_types[j].desc,
  97                                                         info->comment,
  98                                                         info->permissions,
  99                                                         info->max_users,
 100                                                         info->current_users,
 101                                                         info->path,
 102                                                         info->password);
 103                                         break;
 104                                 }
 105                         }
 106                 }
 107                 break;
 108
 109         case 501:
 110                 for (i = 0; i < s.out.ctr.ctr501->count; i++) {
 111                         struct srvsvc_NetShareInfo501 *info = &s.out.ctr.ctr501->array[i];
 112                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
 113                                 if (share_types[j].type == info->type) {
 114                                         torture_comment(tctx,
 115                                                         "\t[%d] %s"
 116                                                         "\t%s "
 117                                                         "[csc_policy=0x%08x]\n"
 118                                                         "\t    %s\n",
 119                                                         i,
 120                                                         info->name,
 121                                                         share_types[j].desc,
 122                                                         info->csc_policy,
 123                                                         info->comment);
 124                                         break;
 125                                 }
 126                         }
 127                 }
 128                 break;
 129
 130         case 502:
 131                 for (i = 0; i < s.out.ctr.ctr502->count; i++) {
 132                         struct srvsvc_NetShareInfo502 *info = &s.out.ctr.ctr502->array[i];
 133                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
 134                                 if (share_types[j].type == info->type) {
 135                                         torture_comment(tctx,
 136                                                         "\t[%d] %s\t%s\n"
 137                                                         "\t    %s\n"
 138                                                         "\t    [perms=0x%08x, "
 139                                                         "max_usr=%d, "
 140                                                         "cur_usr=%d, "
 141                                                         "path=%s, pass=%s]\n",
 142                                                         i,
 143                                                         info->name,
 144                                                         share_types[j].desc,
 145                                                         info->comment,
 146                                                         info->permissions,
 147                                                         info->max_users,
 148                                                         info->current_users,
 149                                                         info->path,
 150                                                         info->password);
 151                                         break;
 152                                 }
 153                         }
 154                 }
 155                 break;
 156         }
 157 }
 158
 159
 160 bool torture_listshares(struct torture_context *torture)
 161 {
 162         struct libnet_ListShares share;
 163         NTSTATUS  status;
 164         uint32_t levels[] = { 0, 1, 2, 501, 502 };
 165         int i;
 166         bool ret = true;
 167         struct libnet_context* libnetctx;
 168         struct dcerpc_binding *binding;
 169         TALLOC_CTX *mem_ctx;
 170
 171         mem_ctx = talloc_init("test_listshares");
 172         status = torture_rpc_binding(torture, &binding);
 173         if (!NT_STATUS_IS_OK(status)) {
 174                 ret = false;
 175                 goto done;
 176         }
 177
 178         libnetctx = libnet_context_init(torture->ev, torture->lp_ctx);
 179         if (!libnetctx) {
 180                 torture_comment(torture, "Couldn't allocate libnet context\n");
 181                 ret = false;
 182                 goto done;
 183         }
 184
 185         libnetctx->cred = samba_cmdline_get_creds();
 186
 187         torture_comment(torture, "Testing libnet_ListShare\n");
 188
 189         share.in.server_name = dcerpc_binding_get_string_option(binding, "host");
 190
 191         for (i = 0; i < ARRAY_SIZE(levels); i++) {
 192                 share.in.level = levels[i];
 193                 torture_comment(torture, "Testing libnet_ListShare level %u\n", share.in.level);
 194
 195                 status = libnet_ListShares(libnetctx, mem_ctx, &share);
 196                 if (!NT_STATUS_IS_OK(status)) {
 197                         torture_comment(torture, "libnet_ListShare level %u failed - %s\n", share.in.level, share.out.error_string);
 198                         ret = false;
 199                         goto done;
 200                 }
 201
 202                 torture_comment(torture, "listing shares:\n");
 203                 test_displayshares(torture, share);
 204         }
 205
 206 done:
 207         talloc_free(mem_ctx);
 208         return ret;
 209 }
 210
 211
 212 static bool test_addshare(struct torture_context *tctx,
 213                           struct dcerpc_binding_handle *b, TALLOC_CTX *mem_ctx, const char *host,
 214                           const char* share)
 215 {
 216         NTSTATUS status;
 217         struct srvsvc_NetShareAdd add;
 218         union srvsvc_NetShareInfo info;
 219         struct srvsvc_NetShareInfo2 i;
 220
 221         ZERO_STRUCT(i);
 222         ZERO_STRUCT(info);
 223         ZERO_STRUCT(add);
 224
 225         i.name         = share;
 226         i.type         = STYPE_DISKTREE;
 227         i.path         = "C:\\WINDOWS\\TEMP";
 228         i.max_users    = 5;
 229         i.comment      = "Comment to the test share";
 230         i.password     = NULL;
 231         i.permissions  = 0x0;
 232
 233         info.info2 = &i;
 234
 235         add.in.server_unc = host;
 236         add.in.level      = 2;
 237         add.in.info       = &info;
 238         add.in.parm_error = NULL;
 239
 240         status = dcerpc_srvsvc_NetShareAdd_r(b, mem_ctx, &add);
 241         if (!NT_STATUS_IS_OK(status)) {
 242                 torture_comment(tctx, "Failed to add a new share\n");
 243                 return false;
 244         }
 245
 246         torture_comment(tctx, "share added\n");
 247         return true;
 248 }
 249
 250
 251 bool torture_delshare(struct torture_context *torture)
 252 {
 253         struct dcerpc_pipe *p;
 254         struct dcerpc_binding *binding;
 255         struct libnet_context* libnetctx;
 256         const char *host;
 257         NTSTATUS  status;
 258         bool ret = true;
 259         struct libnet_DelShare share;
 260
 261         host = torture_setting_string(torture, "host", NULL);
 262         status = torture_rpc_binding(torture, &binding);
 263         torture_assert_ntstatus_ok(torture, status, "Failed to get binding");
 264
 265         libnetctx = libnet_context_init(torture->ev, torture->lp_ctx);
 266         libnetctx->cred = samba_cmdline_get_creds();
 267
 268         status = torture_rpc_connection(torture,
 269                                         &p,
 270                                         &ndr_table_srvsvc);
 271
 272         torture_assert_ntstatus_ok(torture, status, "Failed to get rpc connection");
 273
 274         if (!test_addshare(torture, p->binding_handle, torture, host, TEST_SHARENAME)) {
 275                 return false;
 276         }
 277
 278         share.in.server_name    = dcerpc_binding_get_string_option(binding, "host");
 279         share.in.share_name     = TEST_SHARENAME;
 280
 281         status = libnet_DelShare(libnetctx, torture, &share);
 282         torture_assert_ntstatus_ok(torture, status, "Failed to delete share");
 283
 284         return ret;
 285 }