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 
  33 /*
  34  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
  35  * Use is subject to license terms.
  36  *
  37  * Copyright 2013 Nexenta Systems, Inc.  All rights reserved.
  38  */
  39 
  40 #include <sys/param.h>
  41 #include <sys/time.h>
  42 #include <stdio.h>
  43 #include <string.h>
  44 #include <unistd.h>
  45 #include <stdlib.h>
  46 #include <err.h>
  47 #include <sysexits.h>
  48 #include <locale.h>
  49 #include <libintl.h>
  50 
  51 #include <netsmb/smb_lib.h>
  52 
  53 #include "common.h"
  54 
  55 #ifndef EX_DATAERR
  56 #define EX_DATAERR 1
  57 #endif
  58 
  59 static void help(void);
  60 
  61 
  62 typedef int cmd_fn_t (int argc, char *argv[]);
  63 typedef void cmd_usage_t (void);
  64 
  65 #define CMDFL_NO_KMOD   0x0001
  66 
  67 static struct commands {
  68         const char      *name;
  69         cmd_fn_t        *fn;
  70         cmd_usage_t     *usage;
  71         int             flags;
  72 } commands[] = {
  73         {"crypt",       cmd_crypt,      NULL, CMDFL_NO_KMOD},
  74         {"help",        cmd_help,       help_usage, CMDFL_NO_KMOD},
  75         {"info",        cmd_info,       info_usage, 0},
  76         {"login",       cmd_login,      login_usage, 0},
  77         {"logout",      cmd_logout,     logout_usage, 0},
  78         {"logoutall",   cmd_logoutall,  logoutall_usage, 0},
  79         {"lookup",      cmd_lookup,     lookup_usage, CMDFL_NO_KMOD},
  80         {"print",       cmd_print,      print_usage, 0},
  81         {"status",      cmd_status,     status_usage, CMDFL_NO_KMOD},
  82         {"view",        cmd_view,       view_usage, 0},
  83         {NULL, NULL, NULL, 0}
  84 };
  85 
  86 static struct commands *
  87 lookupcmd(const char *name)
  88 {
  89         struct commands *cmd;
  90 
  91         for (cmd = commands; cmd->name; cmd++) {
  92                 if (strcmp(cmd->name, name) == 0)
  93                         return (cmd);
  94         }
  95         return (NULL);
  96 }
  97 
  98 int
  99 cmd_crypt(int argc, char *argv[])
 100 {
 101         char *cp, *psw;
 102 
 103         if (argc < 2)
 104                 psw = getpassphrase(gettext("Password:"));
 105         else
 106                 psw = argv[1];
 107         /* XXX Better to embed malloc/free in smb_simplecrypt? */
 108         cp = malloc(4 + 2 * strlen(psw));
 109         if (cp == NULL)
 110                 errx(EX_DATAERR, gettext("out of memory"));
 111         smb_simplecrypt(cp, psw);
 112         printf("%s\n", cp);
 113         free(cp);
 114         return (0);
 115 }
 116 
 117 int
 118 cmd_help(int argc, char *argv[])
 119 {
 120         struct commands *cmd;
 121         char *cp;
 122 
 123         if (argc < 2)
 124                 help_usage();
 125         cp = argv[1];
 126         cmd = lookupcmd(cp);
 127         if (cmd == NULL)
 128                 errx(EX_DATAERR, gettext("unknown command %s"), cp);
 129         if (cmd->usage == NULL)
 130                 errx(EX_DATAERR,
 131                     gettext("no specific help for command %s"), cp);
 132         cmd->usage();
 133         return (0);
 134 }
 135 
 136 int
 137 main(int argc, char *argv[])
 138 {
 139         struct commands *cmd;
 140         char *cp;
 141         int err, opt;
 142 
 143         (void) setlocale(LC_ALL, "");
 144         (void) textdomain(TEXT_DOMAIN);
 145 
 146 #ifdef APPLE
 147         dropsuid(); /* see libsmbfs */
 148 #endif
 149 
 150         if (argc < 2)
 151                 help();
 152 
 153         while ((opt = getopt(argc, argv, "dhv")) != EOF) {
 154                 switch (opt) {
 155                 case 'd':
 156                         smb_debug++;
 157                         break;
 158                 case 'h':
 159                         help();
 160                         /* NOTREACHED */
 161                 case 'v':
 162                         smb_verbose++;
 163                         break;
 164                 default:
 165                         help();
 166                         /* NOTREACHED */
 167                 }
 168         }
 169         if (optind >= argc)
 170                 help();
 171 
 172         cp = argv[optind];
 173         cmd = lookupcmd(cp);
 174         if (cmd == NULL)
 175                 errx(EX_DATAERR, gettext("unknown command %s"), cp);
 176 
 177         if ((cmd->flags & CMDFL_NO_KMOD) == 0 && smb_lib_init() != 0)
 178                 exit(1);
 179 
 180         argc -= optind;
 181         argv += optind;
 182         optind = 1;
 183         err = cmd->fn(argc, argv);
 184         return ((err) ? 1 : 0);
 185 }
 186 
 187 static void
 188 help(void) {
 189         printf("\n");
 190         printf(gettext("usage: %s [-hv] subcommand [args]\n"), __progname);
 191         printf(gettext("where subcommands are:\n"
 192         " crypt         slightly obscure password\n"
 193         " help          display help on specified subcommand\n"
 194         /* " lc                 display active connections\n" */
 195         " info          display server type and version\n"
 196         " login         login to specified host\n"
 197         " logout        logout from specified host\n"
 198         " logoutall     logout all users (requires privilege)\n"
 199         " lookup        resolve NetBIOS name to IP address\n"
 200         " print         print file to the specified remote printer\n"
 201         " status        resolve IP address or DNS name to NetBIOS names\n"
 202         " view          list resources on specified host\n"
 203         "\n"));
 204         exit(1);
 205 }
 206 
 207 void
 208 help_usage(void) {
 209         printf(gettext("usage: smbutil help command\n"));
 210         exit(1);
 211 }