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 error = sysctlbyname("net.smb.version", &kv, &kvlen, NULL, 0); 78 if (error) { 79 warnx("%s: can't find kernel module\n", __FUNCTION__); 80 return error; 81 } 82 if (NSMB_VERSION != kv) { 83 warnx("%s: kernel module version(%d) don't match library(%d).\n", __FUNCTION__, kv, NSMB_VERSION); 84 return EINVAL; 85 } 86 if ((error = nls_setlocale("")) != 0) { 87 warnx("%s: can't initialise locale\n", __FUNCTION__); 88 return error; 89 } 90 smblib_initialized++; 91 return 0; 92 } 93 94 /* 95 * Print a (descriptive) error message 96 * error values: 97 * 0 - no specific error code available; 98 * 1..32767 - system error 99 */ 100 void 101 smb_error(const char *fmt, int error,...) { 102 va_list ap; 103 const char *cp; 104 int errtype = error & SMB_ERRTYPE_MASK; 105 106 fprintf(stderr, "%s: ", __progname); 107 va_start(ap, error); 108 vfprintf(stderr, fmt, ap); 109 va_end(ap); 110 if (error == -1) 111 error = errno; 112 else 113 error &= ~SMB_ERRTYPE_MASK; 114 switch (errtype) { 115 case SMB_SYS_ERROR: 116 if (error) 117 fprintf(stderr, ": syserr = %s\n", strerror(error)); 118 else 119 fprintf(stderr, "\n"); 120 break; 121 case SMB_RAP_ERROR: 122 fprintf(stderr, ": raperr = %d (0x%04x)\n", error, error); 123 break; 124 case SMB_NB_ERROR: 125 cp = nb_strerror(error); 126 if (cp == NULL) 127 fprintf(stderr, ": nberr = unknown (0x%04x)\n", error); 128 else 129 fprintf(stderr, ": nberr = %s\n", cp); 130 break; 131 default: 132 fprintf(stderr, "\n"); 133 } 134 } 135 136 char * 137 smb_printb(char *dest, int flags, const struct smb_bitname *bnp) { 138 int first = 1; 139 140 strcpy(dest, "<"); 141 for(; bnp->bn_bit; bnp++) { 142 if (flags & bnp->bn_bit) { 143 strcat(dest, bnp->bn_name); 144 first = 0; 145 } 146 if (!first && (flags & bnp[1].bn_bit)) 147 strcat(dest, "|"); 148 } 149 strcat(dest, ">"); 150 return dest; 151 } 152 153 /* 154 * first read ~/.smbrc, next try to merge SMB_CFG_FILE 155 */ 156 int 157 smb_open_rcfile(void) 158 { 159 char *home, *fn; 160 int error; 161 162 home = getenv("HOME"); 163 if (home) { 164 fn = malloc(strlen(home) + 20); 165 sprintf(fn, "%s/.nsmbrc", home); 166 error = rc_open(fn, "r", &smb_rc); 167 free(fn); 168 } 169 error = rc_merge(SMB_CFG_FILE, &smb_rc); 170 if (smb_rc == NULL) { 171 printf("Warning: no cfg file(s) found.\n"); 172 return ENOENT; 173 } 174 return 0; 175 } 176 177 void * 178 smb_dumptree(void) 179 { 180 size_t len; 181 void *p; 182 int error; 183 184 #ifdef APPLE 185 seteuid(eff_uid); /* restore setuid root briefly */ 186 #endif 187 error = sysctlbyname("net.smb.treedump", NULL, &len, NULL, 0); 188 #ifdef APPLE 189 seteuid(real_uid); /* and back to real user */ 190 #endif 191 if (error) 192 return NULL; 193 p = malloc(len); 194 if (p == NULL) 195 return NULL; 196 #ifdef APPLE 197 seteuid(eff_uid); /* restore setuid root briefly */ 198 #endif 199 error = sysctlbyname("net.smb.treedump", p, &len, NULL, 0); 200 #ifdef APPLE 201 seteuid(real_uid); /* and back to real user */ 202 #endif 203 if (error) { 204 free(p); 205 return NULL; 206 } 207 return p; 208 } 209 210 char * 211 smb_simplecrypt(char *dst, const char *src) 212 { 213 int ch, pos; 214 char *dp; 215 216 if (dst == NULL) { 217 dst = malloc(4 + 2 * strlen(src)); 218 if (dst == NULL) 219 return NULL; 220 } 221 dp = dst; 222 *dst++ = '$'; 223 *dst++ = '$'; 224 *dst++ = '1'; 225 pos = 27; 226 while (*src) { 227 ch = *src++; 228 if (isascii(ch)) 229 ch = (isupper(ch) ? ('A' + (ch - 'A' + 13) % 26) : 230 islower(ch) ? ('a' + (ch - 'a' + 13) % 26) : ch); 231 ch ^= pos; 232 pos += 13; 233 if (pos > 256) 234 pos -= 256; 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 (pos > 256) 266 pos -= 256; 267 if (isascii(ch)) 268 ch = (isupper(ch) ? ('A' + (ch - 'A' + 13) % 26) : 269 islower(ch) ? ('a' + (ch - 'a' + 13) % 26) : ch); 270 *dst++ = ch; 271 } 272 *dst = 0; 273 return 0; 274 } 275 276 277 #ifdef APPLE 278 static int 279 safe_execv(char *args[]) 280 { 281 int pid; 282 union wait status; 283 284 pid = fork(); 285 if (pid == 0) { 286 (void)execv(args[0], args); 287 errx(EX_OSERR, "%s: execv %s failed, %s\n", __progname, 288 args[0], strerror(errno)); 289 } 290 if (pid == -1) { 291 fprintf(stderr, "%s: fork failed, %s\n", __progname, 292 strerror(errno)); 293 return (1); 294 } 295 if (wait4(pid, (int *)&status, 0, NULL) != pid) { 296 fprintf(stderr, "%s: BUG executing %s command\n", __progname, 297 args[0]); 298 return (1); 299 } else if (!WIFEXITED(status)) { 300 fprintf(stderr, "%s: %s command aborted by signal %d\n", 301 __progname, args[0], WTERMSIG(status)); 302 return (1); 303 } else if (WEXITSTATUS(status)) { 304 fprintf(stderr, "%s: %s command failed, exit status %d: %s\n", 305 __progname, args[0], WEXITSTATUS(status), 306 strerror(WEXITSTATUS(status))); 307 return (1); 308 } 309 return (0); 310 } 311 312 313 void 314 dropsuid() 315 { 316 /* drop setuid root privs asap */ 317 eff_uid = geteuid(); 318 real_uid = getuid(); 319 seteuid(real_uid); 320 return; 321 } 322 323 324 static int 325 kextisloaded(char * kextname) 326 { 327 mach_port_t kernel_port; 328 kmod_info_t *k, *loaded_modules = 0; 329 int err, loaded_count = 0; 330 331 /* on error return not loaded - to make loadsmbvfs fail */ 332 333 err = task_for_pid(mach_task_self(), 0, &kernel_port); 334 if (err) { 335 fprintf(stderr, "%s: %s: %s\n", __progname, 336 "unable to get kernel task port", 337 mach_error_string(err)); 338 return (0); 339 } 340 err = kmod_get_info(kernel_port, (void *)&loaded_modules, 341 &loaded_count); /* never freed */ 342 if (err) { 343 fprintf(stderr, "%s: %s: %s\n", __progname, 344 "kmod_get_info() failed", 345 mach_error_string(err)); 346 return (0); 347 } 348 for (k = loaded_modules; k; k = k->next ? k+1 : 0) 349 if (!strcmp(k->name, kextname)) 350 return (1); 351 return (0); 352 } 353 354 355 #define KEXTLOAD_COMMAND "/sbin/kextload" 356 #define FS_KEXT_DIR "/System/Library/Extensions/smbfs.kext" 357 #define FULL_KEXTNAME "com.apple.filesystems.smbfs" 358 359 360 int 361 loadsmbvfs() 362 { 363 const char *kextargs[] = {KEXTLOAD_COMMAND, FS_KEXT_DIR, NULL}; 364 int error = 0; 365 366 /* 367 * temporarily revert to root (required for kextload) 368 */ 369 seteuid(eff_uid); 370 if (!kextisloaded(FULL_KEXTNAME)) { 371 error = safe_execv(kextargs); 372 if (!error) 373 error = !kextisloaded(FULL_KEXTNAME); 374 } 375 seteuid(real_uid); /* and back to real user */ 376 return (error); 377 } 378 #endif /* APPLE */ 379