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 }