an, awi, cue and kue don't use the miibus interface.
[dragonfly.git] / contrib / smbfs / lib / smb / subr.c
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: subr.c,v 1.12 2001/08/22 03:31:37 bp Exp $
33  */
34
35 #include <sys/param.h>
36 #include <sys/types.h>
37 #include <sys/errno.h>
38 #include <sys/sysctl.h>
39 #include <sys/syscall.h>
40 #include <unistd.h>
41 #include <ctype.h>
42 #include <string.h>
43 #include <stdio.h>
44 #include <stdlib.h>
45 #include <stdarg.h>
46 #include <err.h>
47
48 #include <netsmb/netbios.h>
49 #include <netsmb/smb_lib.h>
50 #include <netsmb/nb_lib.h>
51 #include <cflib.h>
52
53 #ifdef APPLE
54 #include <sysexits.h>
55 #include <sys/wait.h>
56 #include <mach/mach.h>
57 #include <mach/mach_error.h>
58
59 uid_t real_uid, eff_uid;
60 #endif
61
62 extern char *__progname;
63
64 static int smblib_initialized;
65
66 struct rcfile *smb_rc;
67
68 int
69 smb_lib_init(void)
70 {
71         int error;
72         int kv;
73         size_t kvlen = sizeof(kv);
74
75         if (smblib_initialized)
76                 return 0;
77 #if __FreeBSD_version > 400000
78         error = sysctlbyname("net.smb.version", &kv, &kvlen, NULL, 0);
79         if (error) {
80                 warnx("%s: can't find kernel module\n", __FUNCTION__);
81                 return error;
82         }
83         if (NSMB_VERSION != kv) {
84                 warnx("%s: kernel module version(%d) don't match library(%d).\n", __FUNCTION__, kv, NSMB_VERSION);
85                 return EINVAL;
86         }
87 #endif
88         if ((error = nls_setlocale("")) != 0) {
89                 warnx("%s: can't initialise locale\n", __FUNCTION__);
90                 return error;
91         }
92         smblib_initialized++;
93         return 0;
94 }
95
96 /*
97  * Print a (descriptive) error message
98  * error values:
99  *         0 - no specific error code available;
100  *  1..32767 - system error
101  */
102 void
103 smb_error(const char *fmt, int error,...) {
104         va_list ap;
105         const char *cp;
106         int errtype = error & SMB_ERRTYPE_MASK;
107
108         fprintf(stderr, "%s: ", __progname);
109         va_start(ap, error);
110         vfprintf(stderr, fmt, ap);
111         va_end(ap);
112         if (error == -1)
113                 error = errno;
114         else
115                 error &= ~SMB_ERRTYPE_MASK;
116         switch (errtype) {
117             case SMB_SYS_ERROR:
118                 if (error)
119                         fprintf(stderr, ": syserr = %s\n", strerror(error));
120                 else
121                         fprintf(stderr, "\n");
122                 break;
123             case SMB_RAP_ERROR:
124                 fprintf(stderr, ": raperr = %d (0x%04x)\n", error, error);
125                 break;
126             case SMB_NB_ERROR:
127                 cp = nb_strerror(error);
128                 if (cp == NULL)
129                         fprintf(stderr, ": nberr = unknown (0x%04x)\n", error);
130                 else
131                         fprintf(stderr, ": nberr = %s\n", cp);
132                 break;
133             default:
134                 fprintf(stderr, "\n");
135         }
136 }
137
138 char *
139 smb_printb(char *dest, int flags, const struct smb_bitname *bnp) {
140         int first = 1;
141
142         strcpy(dest, "<");
143         for(; bnp->bn_bit; bnp++) {
144                 if (flags & bnp->bn_bit) {
145                         strcat(dest, bnp->bn_name);
146                         first = 0;
147                 }
148                 if (!first && (flags & bnp[1].bn_bit))
149                         strcat(dest, "|");
150         }
151         strcat(dest, ">");
152         return dest;
153 }
154
155 /*
156  * first read ~/.smbrc, next try to merge SMB_CFG_FILE
157  */
158 int
159 smb_open_rcfile(void)
160 {
161         char *home, *fn;
162         int error;
163
164         home = getenv("HOME");
165         if (home) {
166                 fn = malloc(strlen(home) + 20);
167                 sprintf(fn, "%s/.nsmbrc", home);
168                 error = rc_open(fn, "r", &smb_rc);
169                 free(fn);
170         }
171         error = rc_merge(SMB_CFG_FILE, &smb_rc);
172         if (smb_rc == NULL) {
173                 printf("Warning: no cfg file(s) found.\n");
174                 return ENOENT;
175         }
176         return 0;
177 }
178
179 void *
180 smb_dumptree(void)
181 {
182         size_t len;
183         void *p;
184         int error;
185         
186 #ifdef APPLE
187         seteuid(eff_uid); /* restore setuid root briefly */
188 #endif
189         error = sysctlbyname("net.smb.treedump", NULL, &len, NULL, 0);
190 #ifdef APPLE
191         seteuid(real_uid); /* and back to real user */
192 #endif
193         if (error)
194                 return NULL;
195         p = malloc(len);
196         if (p == NULL)
197                 return NULL;
198 #ifdef APPLE
199         seteuid(eff_uid); /* restore setuid root briefly */
200 #endif
201         error = sysctlbyname("net.smb.treedump", p, &len, NULL, 0);
202 #ifdef APPLE
203         seteuid(real_uid); /* and back to real user */
204 #endif
205         if (error) {
206                 free(p);
207                 return NULL;
208         }
209         return p;
210 }
211
212 char *
213 smb_simplecrypt(char *dst, const char *src)
214 {
215         int ch, pos;
216         char *dp;
217
218         if (dst == NULL) {
219                 dst = malloc(4 + 2 * strlen(src));
220                 if (dst == NULL)
221                         return NULL;
222         }
223         dp = dst;
224         *dst++ = '$';
225         *dst++ = '$';
226         *dst++ = '1';
227         pos = 27;
228         while (*src) {
229                 ch = *src++;
230                 if (isascii(ch))
231                     ch = (isupper(ch) ? ('A' + (ch - 'A' + 13) % 26) :
232                           islower(ch) ? ('a' + (ch - 'a' + 13) % 26) : ch);
233                 ch ^= pos;
234                 pos += 13;
235                 sprintf(dst, "%02x", ch);
236                 dst += 2;
237         }
238         *dst = 0;
239         return dp;
240 }
241
242 int
243 smb_simpledecrypt(char *dst, const char *src)
244 {
245         char *ep, hexval[3];
246         int len, ch, pos;
247
248         if (strncmp(src, "$$1", 3) != 0)
249                 return EINVAL;
250         src += 3;
251         len = strlen(src);
252         if (len & 1)
253                 return EINVAL;
254         len /= 2;
255         hexval[2] = 0;
256         pos = 27;
257         while (len--) {
258                 hexval[0] = *src++;
259                 hexval[1] = *src++;
260                 ch = strtoul(hexval, &ep, 16);
261                 if (*ep != 0)
262                         return EINVAL;
263                 ch ^= pos;
264                 pos += 13;
265                 if (isascii(ch))
266                     ch = (isupper(ch) ? ('A' + (ch - 'A' + 13) % 26) :
267                           islower(ch) ? ('a' + (ch - 'a' + 13) % 26) : ch);
268                 *dst++ = ch;
269         }
270         *dst = 0;
271         return 0;
272 }
273
274
275 #ifdef APPLE
276 static int
277 safe_execv(char *args[])
278 {       
279         int          pid;   
280         union wait      status;
281         
282         pid = fork();  
283         if (pid == 0) {
284                 (void)execv(args[0], args);
285                 errx(EX_OSERR, "%s: execv %s failed, %s\n", __progname,
286                      args[0], strerror(errno));
287         }
288         if (pid == -1) {
289                 fprintf(stderr, "%s: fork failed, %s\n", __progname,
290                         strerror(errno));
291                 return (1);
292         }
293         if (wait4(pid, (int *)&status, 0, NULL) != pid) {
294                 fprintf(stderr, "%s: BUG executing %s command\n", __progname,
295                         args[0]);  
296                 return (1);
297         } else if (!WIFEXITED(status)) {
298                 fprintf(stderr, "%s: %s command aborted by signal %d\n",
299                         __progname, args[0], WTERMSIG(status));
300                 return (1);
301         } else if (WEXITSTATUS(status)) {
302                 fprintf(stderr, "%s: %s command failed, exit status %d: %s\n",
303                         __progname, args[0], WEXITSTATUS(status),
304                         strerror(WEXITSTATUS(status)));
305                 return (1);
306         }       
307         return (0);
308 }       
309
310
311 void
312 dropsuid()
313 {
314         /* drop setuid root privs asap */
315         eff_uid = geteuid();
316         real_uid = getuid();
317         seteuid(real_uid);
318         return;
319 }
320
321
322 static int
323 kextisloaded(char * kextname)
324 {
325         mach_port_t kernel_port;
326         kmod_info_t *k, *loaded_modules = 0;
327         int err, loaded_count = 0;
328
329         /* on error return not loaded - to make loadsmbvfs fail */
330
331         err = task_for_pid(mach_task_self(), 0, &kernel_port);
332         if (err) {
333                 fprintf(stderr, "%s: %s: %s\n", __progname,
334                         "unable to get kernel task port",
335                         mach_error_string(err));
336                 return (0);
337         }
338         err = kmod_get_info(kernel_port, (void *)&loaded_modules,
339                             &loaded_count); /* never freed */
340         if (err) {
341                 fprintf(stderr, "%s: %s: %s\n", __progname,
342                         "kmod_get_info() failed",
343                         mach_error_string(err));
344                 return (0);
345         }
346         for (k = loaded_modules; k; k = k->next ? k+1 : 0)
347                 if (!strcmp(k->name, kextname))
348                         return (1);
349         return (0);
350 }
351
352
353 #define KEXTLOAD_COMMAND        "/sbin/kextload"
354 #define FS_KEXT_DIR             "/System/Library/Extensions/smbfs.kext"
355 #define FULL_KEXTNAME           "com.apple.filesystems.smbfs"
356
357
358 int
359 loadsmbvfs()
360 {       
361         const char *kextargs[] = {KEXTLOAD_COMMAND, FS_KEXT_DIR, NULL};
362         int error = 0;
363
364         /*
365          * temporarily revert to root (required for kextload)
366          */
367         seteuid(eff_uid);
368         if (!kextisloaded(FULL_KEXTNAME)) {
369                 error = safe_execv(kextargs);
370                 if (!error)
371                         error = !kextisloaded(FULL_KEXTNAME);
372         }
373         seteuid(real_uid); /* and back to real user */
374         return (error);
375 }       
376 #endif /* APPLE */