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