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 2018 Nexenta Systems, Inc.
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) __NORETURN;
60 
61 typedef int cmd_fn_t (int argc, char *argv[]);
62 typedef void cmd_usage_t (void);
63 
64 #define	CMDFL_NO_KMOD	0x0001
65 
66 static struct commands {
67 	const char	*name;
68 	cmd_fn_t	*fn;
69 	cmd_usage_t	*usage;
70 	int		flags;
71 } commands[] = {
72 	{"crypt",	cmd_crypt,	NULL, CMDFL_NO_KMOD},
73 	{"discon",	cmd_discon,	discon_usage, 0},
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 *
lookupcmd(const char * name)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
cmd_crypt(int argc,char * argv[])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
cmd_help(int argc,char * argv[])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
main(int argc,char * argv[])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
help(void)188 help(void)
189 {
190 	printf("\n");
191 	printf(gettext("usage: %s [-hv] subcommand [args]\n"), __progname);
192 	printf(gettext("where subcommands are:\n"
193 	" crypt          slightly obscure password\n"
194 	" help           display help on specified subcommand\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
help_usage(void)208 help_usage(void)
209 {
210 	printf(gettext("usage: smbutil help command\n"));
211 	exit(1);
212 }
213