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 }