1 /*
   2  * Copyright (c) 2000, Boris Popov
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *    This product includes software developed by Boris Popov.
  16  * 4. Neither the name of the author nor the names of any co-contributors
  17  *    may be used to endorse or promote products derived from this software
  18  *    without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30  * SUCH DAMAGE.
  31  *
  32  * $Id: view.c,v 1.9 2004/12/13 00:25:39 lindak Exp $
  33  */
  34 
  35 /*
  36  * Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
  37  */
  38 
  39 #include <sys/types.h>
  40 #include <errno.h>
  41 #include <stdio.h>
  42 #include <err.h>
  43 #include <unistd.h>
  44 #include <strings.h>
  45 #include <stdlib.h>
  46 #include <sysexits.h>
  47 #include <libintl.h>
  48 
  49 #include <netsmb/smb.h>
  50 #include <netsmb/smb_lib.h>
  51 #include <netsmb/smb_netshareenum.h>
  52 
  53 #include "common.h"
  54 
  55 int enum_shares(smb_ctx_t *);
  56 void print_shares(int, int, struct share_info *);
  57 
  58 void
  59 view_usage(void)
  60 {
  61         printf(gettext("usage: smbutil view [connection options] //"
  62             "[workgroup;][user[:password]@]server\n"));
  63         exit(1);
  64 }
  65 
  66 int
  67 cmd_view(int argc, char *argv[])
  68 {
  69         struct smb_ctx *ctx;
  70         int error, err2, opt;
  71 
  72         if (argc < 2)
  73                 view_usage();
  74 
  75         error = smb_ctx_alloc(&ctx);
  76         if (error)
  77                 return (error);
  78 
  79         error = smb_ctx_scan_argv(ctx, argc, argv,
  80             SMBL_SERVER, SMBL_SERVER, USE_WILDCARD);
  81         if (error)
  82                 return (error);
  83 
  84         error = smb_ctx_readrc(ctx);
  85         if (error)
  86                 return (error);
  87 
  88         while ((opt = getopt(argc, argv, STDPARAM_OPT)) != EOF) {
  89                 if (opt == '?')
  90                         view_usage();
  91                 error = smb_ctx_opt(ctx, opt, optarg);
  92                 if (error)
  93                         return (error);
  94         }
  95 
  96         smb_ctx_setshare(ctx, "IPC$", USE_IPC);
  97 
  98         /*
  99          * Resolve the server address,
 100          * setup derived defaults.
 101          */
 102         error = smb_ctx_resolve(ctx);
 103         if (error)
 104                 return (error);
 105 
 106         /*
 107          * Have server, share, etc. from above:
 108          * smb_ctx_scan_argv, option settings.
 109          * Get the session and tree.
 110          */
 111 again:
 112         error = smb_ctx_get_ssn(ctx);
 113         if (error == EAUTH) {
 114                 err2 = smb_get_authentication(ctx);
 115                 if (err2 == 0)
 116                         goto again;
 117         }
 118         if (error) {
 119                 smb_error(gettext("//%s: login failed"),
 120                     error, ctx->ct_fullserver);
 121                 return (error);
 122         }
 123 
 124         error = smb_ctx_get_tree(ctx);
 125         if (error) {
 126                 smb_error(gettext("//%s/%s: tree connect failed"),
 127                     error, ctx->ct_fullserver, ctx->ct_origshare);
 128                 return (error);
 129         }
 130 
 131         /*
 132          * Have IPC$ tcon, now list shares.
 133          * This prints its own errors.
 134          */
 135         error = enum_shares(ctx);
 136         if (error)
 137                 return (error);
 138 
 139         smb_ctx_free(ctx);
 140         return (0);
 141 }
 142 
 143 #ifdef I18N     /* not defined, put here so xgettext(1) can find strings */
 144 static char *shtype[] = {
 145         gettext("disk"),
 146         gettext("printer"),
 147         gettext("device"),      /* Communications device */
 148         gettext("IPC"),         /* Inter process communication */
 149         gettext("unknown")
 150 };
 151 #else
 152 static char *shtype[] = {
 153         "disk",
 154         "printer",
 155         "device",               /* Communications device */
 156         "IPC",                  /* IPC Inter process communication */
 157         "unknown"
 158 };
 159 #endif
 160 
 161 int
 162 enum_shares(smb_ctx_t *ctx)
 163 {
 164         struct share_info *share_info;
 165         int error, entries, total;
 166 
 167         /*
 168          * XXX: Later, try RPC first,
 169          * then fall back to RAP...
 170          */
 171         error = smb_netshareenum(ctx, &entries, &total, &share_info);
 172         if (error) {
 173                 smb_error(gettext("//%s failed to list shares"),
 174                     error, ctx->ct_fullserver);
 175                 return (error);
 176         }
 177         print_shares(entries, total, share_info);
 178         return (0);
 179 }
 180 
 181 void
 182 print_shares(int entries, int total,
 183         struct share_info *share_info)
 184 {
 185         struct share_info *ep;
 186         int i;
 187 
 188         printf(gettext("Share        Type       Comment\n"));
 189         printf("-------------------------------\n");
 190 
 191         for (ep = share_info, i = 0; i < entries; i++, ep++) {
 192                 int sti = ep->type & STYPE_MASK;
 193                 if (sti > STYPE_UNKNOWN)
 194                         sti = STYPE_UNKNOWN;
 195                 printf("%-12s %-10s %s\n", ep->netname,
 196                     gettext(shtype[sti]),
 197                     ep->remark ? ep->remark : "");
 198                 free(ep->netname);
 199                 free(ep->remark);
 200         }
 201         printf(gettext("\n%d shares listed from %d available\n"),
 202             entries, total);
 203 
 204         free(share_info);
 205 }