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