root/source4/torture/libnet/libnet_share.c

/* [<][>][^][v][top][bottom][index][help] */

DEFINITIONS

This source file includes following definitions.
  1. test_displayshares
  2. torture_listshares
  3. test_addshare
  4. torture_delshare

   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/rpc.h"
  24 #include "libnet/libnet.h"
  25 #include "lib/cmdline/popt_common.h"
  26 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
  27 
  28 
  29 #define TEST_SHARENAME "libnetsharetest"
  30 
  31 
  32 static void test_displayshares(struct libnet_ListShares s)
     /* [<][>][^][v][top][bottom][index][help] */
  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                         struct srvsvc_NetShareInfo0 *info = &s.out.ctr.ctr0->array[i];
  58                         d_printf("\t[%d] %s\n", i, info->name);
  59                 }
  60                 break;
  61 
  62         case 1:
  63                 for (i = 0; i < s.out.ctr.ctr1->count; i++) {
  64                         struct srvsvc_NetShareInfo1 *info = &s.out.ctr.ctr1->array[i];
  65                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
  66                                 if (share_types[j].type == info->type) break;
  67                         }
  68                         d_printf("\t[%d] %s (%s)\t%s\n", i, info->name,
  69                                info->comment, share_types[j].desc);
  70                 }
  71                 break;
  72 
  73         case 2:
  74                 for (i = 0; i < s.out.ctr.ctr2->count; i++) {
  75                         struct srvsvc_NetShareInfo2 *info = &s.out.ctr.ctr2->array[i];
  76                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
  77                                 if (share_types[j].type == info->type) break;
  78                         }
  79                         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",
  80                                  i, info->name, share_types[j].desc, info->comment,
  81                                  info->permissions, info->max_users,
  82                                  info->current_users, info->path,
  83                                  info->password);
  84                 }
  85                 break;
  86 
  87         case 501:
  88                 for (i = 0; i < s.out.ctr.ctr501->count; i++) {
  89                         struct srvsvc_NetShareInfo501 *info = &s.out.ctr.ctr501->array[i];
  90                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
  91                                 if (share_types[j].type == info->type) break;
  92                         }
  93                         d_printf("\t[%d] %s\t%s [csc_policy=0x%08x]\n\t    %s\n", i, info->name,
  94                                  share_types[j].desc, info->csc_policy,
  95                                  info->comment);
  96                 }
  97                 break;
  98 
  99         case 502:
 100                 for (i = 0; i < s.out.ctr.ctr502->count; i++) {
 101                         struct srvsvc_NetShareInfo502 *info = &s.out.ctr.ctr502->array[i];
 102                         for (j = 0; j < ARRAY_SIZE(share_types); j++) {
 103                                 if (share_types[j].type == info->type) break;
 104                         }
 105                         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",
 106                                  i, info->name, share_types[j].desc, info->comment,
 107                                  info->permissions, info->max_users,
 108                                  info->current_users, info->path,
 109                                  info->password);
 110                 }
 111                 break;
 112         }
 113 }
 114 
 115 
 116 bool torture_listshares(struct torture_context *torture)
     /* [<][>][^][v][top][bottom][index][help] */
 117 {
 118         struct libnet_ListShares share;
 119         NTSTATUS  status;
 120         uint32_t levels[] = { 0, 1, 2, 501, 502 };
 121         int i;
 122         bool ret = true;
 123         struct libnet_context* libnetctx;
 124         struct dcerpc_binding *binding;
 125         TALLOC_CTX *mem_ctx;
 126 
 127         mem_ctx = talloc_init("test_listshares");
 128         status = torture_rpc_binding(torture, &binding);
 129         if (!NT_STATUS_IS_OK(status)) {
 130                 ret = false;
 131                 goto done;
 132         }
 133 
 134         libnetctx = libnet_context_init(torture->ev, torture->lp_ctx);
 135         if (!libnetctx) {
 136                 printf("Couldn't allocate libnet context\n");
 137                 ret = false;
 138                 goto done;
 139         }
 140 
 141         libnetctx->cred = cmdline_credentials;
 142         
 143         printf("Testing libnet_ListShare\n");
 144         
 145         share.in.server_name = talloc_asprintf(mem_ctx, "%s", binding->host);
 146 
 147         for (i = 0; i < ARRAY_SIZE(levels); i++) {
 148                 share.in.level = levels[i];
 149                 printf("testing libnet_ListShare level %u\n", share.in.level);
 150 
 151                 status = libnet_ListShares(libnetctx, mem_ctx, &share);
 152                 if (!NT_STATUS_IS_OK(status)) {
 153                         printf("libnet_ListShare level %u failed - %s\n", share.in.level, share.out.error_string);
 154                         ret = false;
 155                         goto done;
 156                 }
 157 
 158                 printf("listing shares:\n");
 159                 test_displayshares(share);
 160         }
 161 
 162 done:
 163         talloc_free(mem_ctx);
 164         return ret;
 165 }
 166 
 167 
 168 static bool test_addshare(struct dcerpc_pipe *svc_pipe, TALLOC_CTX *mem_ctx, const char *host,
     /* [<][>][^][v][top][bottom][index][help] */
 169                           const char* share)
 170 {
 171         NTSTATUS status;
 172         struct srvsvc_NetShareAdd add;
 173         union srvsvc_NetShareInfo info;
 174         struct srvsvc_NetShareInfo2 i;
 175         
 176         i.name         = share;
 177         i.type         = STYPE_DISKTREE;
 178         i.path         = "C:\\WINDOWS\\TEMP";
 179         i.max_users    = 5;
 180         i.comment      = "Comment to the test share";
 181         i.password     = NULL;
 182         i.permissions  = 0x0;
 183 
 184         info.info2 = &i;
 185 
 186         add.in.server_unc = host;
 187         add.in.level      = 2;
 188         add.in.info       = &info;
 189         add.in.parm_error = NULL;
 190 
 191         status = dcerpc_srvsvc_NetShareAdd(svc_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)
     /* [<][>][^][v][top][bottom][index][help] */
 203 {
 204         struct dcerpc_pipe *p;
 205         struct dcerpc_binding *binding;
 206         struct libnet_context* libnetctx;
 207         const char *host;
 208         NTSTATUS  status;
 209         bool ret = true;
 210         struct libnet_DelShare share;
 211         
 212         host = torture_setting_string(torture, "host", NULL);
 213         status = torture_rpc_binding(torture, &binding);
 214         torture_assert_ntstatus_ok(torture, status, "Failed to get binding");
 215 
 216         libnetctx = libnet_context_init(torture->ev, torture->lp_ctx);
 217         libnetctx->cred = cmdline_credentials;
 218 
 219         status = torture_rpc_connection(torture,
 220                                         &p,
 221                                         &ndr_table_srvsvc);
 222 
 223         torture_assert_ntstatus_ok(torture, status, "Failed to get rpc connection");
 224 
 225         if (!test_addshare(p, torture, host, TEST_SHARENAME)) {
 226                 return false;
 227         }
 228 
 229         share.in.server_name    = binding->host;
 230         share.in.share_name     = TEST_SHARENAME;
 231 
 232         status = libnet_DelShare(libnetctx, torture, &share);
 233         torture_assert_ntstatus_ok(torture, status, "Failed to delete share");
 234 
 235         return ret;
 236 }

/* [<][>][^][v][top][bottom][index][help] */