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 * $Id: login.c,v 1.8 2004/03/19 01:49:48 lindak Exp $
33 */
34
35 #pragma ident "%Z%%M% %I% %E% SMI"
36
37 #include <sys/param.h>
38 #include <sys/errno.h>
39 #include <sys/stat.h>
40
41 #include <stdio.h>
42 #include <stdlib.h>
43 #include <strings.h>
44 #include <unistd.h>
45 #include <err.h>
46 #include <libintl.h>
47
48 #include <netsmb/smb_lib.h>
49 #include <netsmb/smb_keychain.h>
50
51 #include "common.h"
52
53 /* defaults */
54 static char def_dom[256];
55 static char def_usr[256];
56 static char tmp_arg[256];
57
58
59 /*
60 * Parse the string: domuser, which may be any of:
61 * "user@domain" or "domain/user" or "domain\\user"
62 * and return pointers to the domain and user parts.
63 * Modifies the string domuser in-place. Returned
64 * string pointers are within the string domusr.
65 */
66 int
smbfs_parse_domuser(char * domuser,char ** dom,char ** usr)67 smbfs_parse_domuser(char *domuser, char **dom, char **usr)
68 {
69 const char sep[] = "@/\\";
70 char sc, *p, *s1, *s2;
71
72 p = strpbrk(domuser, sep);
73 if (p == NULL) {
74 /* No separators - whole string is the user. */
75 *dom = NULL;
76 *usr = domuser;
77 return (0);
78 }
79
80 /* Have two strings. */
81 s1 = domuser;
82 sc = *p; /* Save the sep. char */
83 *p++ = '\0'; /* zap it */
84 s2 = p;
85
86 /* Enforce just one separator */
87 p = strpbrk(s2, sep);
88 if (p)
89 return (-1);
90
91 /*
92 * Now, which order are they?
93 * "user@domain" or "domain/user"
94 */
95 if (sc == '@') {
96 *usr = s1;
97 *dom = s2;
98 } else {
99 *dom = s1;
100 *usr = s2;
101 }
102
103 return (0);
104 }
105
106 void
login_usage(void)107 login_usage(void)
108 {
109 printf(gettext("usage: smbutil login [-c] [[domain/]user]\n"));
110 printf(gettext(" smbutil login [-c] [user[@domain]]\n"));
111 exit(1);
112 }
113
114 int
cmd_login(int argc,char * argv[])115 cmd_login(int argc, char *argv[])
116 {
117 static char prompt[64];
118 char *dom, *usr, *pass;
119 int err, opt;
120 int check = 0;
121
122 while ((opt = getopt(argc, argv, "c")) != EOF) {
123 switch (opt) {
124
125 case 'c': /* smbutil login -c ... */
126 check = 1;
127 break;
128
129 default:
130 login_usage();
131 break;
132 }
133 }
134
135 dom = usr = NULL;
136 if (optind < argc) {
137 strcpy(tmp_arg, argv[optind]);
138 err = smbfs_parse_domuser(tmp_arg, &dom, &usr);
139 if (err)
140 errx(1, gettext("failed to parse %s"), argv[optind]);
141 optind++;
142 }
143 if (optind != argc)
144 login_usage();
145
146 if (dom == NULL || usr == NULL) {
147 err = smbfs_default_dom_usr(NULL, NULL,
148 def_dom, sizeof (def_dom),
149 def_usr, sizeof (def_usr));
150 if (err)
151 errx(1, gettext("failed to get defaults"));
152 }
153 if (dom == NULL)
154 dom = def_dom;
155 else
156 nls_str_upper(dom, dom);
157 if (usr == NULL)
158 usr = def_usr;
159
160 if (check) {
161 err = smbfs_keychain_chk(dom, usr);
162 if (!err)
163 printf(gettext("Keychain entry exists.\n"));
164 else
165 printf(gettext("Keychain entry not found.\n"));
166 return (0);
167 }
168
169 snprintf(prompt, sizeof (prompt),
170 gettext("Password for %s/%s:"), dom, usr);
171 pass = getpassphrase(prompt);
172
173 err = smbfs_keychain_add((uid_t)-1, dom, usr, pass);
174 if (err)
175 errx(1, gettext("failed to add keychain entry"));
176
177 return (0);
178 }
179
180
181 void
logout_usage(void)182 logout_usage(void)
183 {
184 printf(gettext("usage: smbutil logout [[domain/]user]\n"));
185 printf(gettext(" smbutil logout [user[@domain]]\n"));
186 printf(gettext(" smbutil logout -a\n"));
187 exit(1);
188 }
189
190 int
cmd_logout(int argc,char * argv[])191 cmd_logout(int argc, char *argv[])
192 {
193 char *dom, *usr;
194 int err, opt;
195
196 while ((opt = getopt(argc, argv, "a")) != EOF) {
197 switch (opt) {
198
199 case 'a': /* smbutil logout -a */
200 if (optind != argc)
201 logout_usage();
202 err = smbfs_keychain_del_owner();
203 if (err)
204 errx(1,
205 gettext("failed to delete keychain entries"));
206 return (0);
207
208 default:
209 logout_usage();
210 break;
211 }
212 }
213
214 /* This part is like login. */
215 dom = usr = NULL;
216 if (optind < argc) {
217 strcpy(tmp_arg, argv[optind]);
218 err = smbfs_parse_domuser(tmp_arg, &dom, &usr);
219 if (err)
220 errx(1, gettext("failed to parse %s"), argv[optind]);
221 optind++;
222 }
223 if (optind != argc)
224 logout_usage();
225
226 if (dom == NULL || usr == NULL) {
227 err = smbfs_default_dom_usr(NULL, NULL,
228 def_dom, sizeof (def_dom),
229 def_usr, sizeof (def_usr));
230 if (err)
231 errx(1, gettext("failed to get defaults"));
232 }
233 if (dom == NULL)
234 dom = def_dom;
235 else
236 nls_str_upper(dom, dom);
237 if (usr == NULL)
238 usr = def_usr;
239
240 err = smbfs_keychain_del((uid_t)-1, dom, usr);
241 if (err)
242 errx(1, gettext("failed to delete keychain entry"));
243
244 return (0);
245 }
246
247
248 void
logoutall_usage(void)249 logoutall_usage(void)
250 {
251 printf(gettext("usage: smbutil logoutall\n"));
252 exit(1);
253 }
254
255 int
cmd_logoutall(int argc,char * argv[])256 cmd_logoutall(int argc, char *argv[])
257 {
258 int err;
259
260 if (optind != argc)
261 logoutall_usage();
262
263 err = smbfs_keychain_del_everyone();
264 if (err == EPERM) {
265 errx(1,
266 gettext("You must have super-user privileges to use this sub-command\n"));
267 }
268 if (err) {
269 errx(1, gettext("Failed to delete all keychain entries: %s\n"),
270 smb_strerror(err));
271 }
272
273 return (0);
274 }
275