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 sprintf(dst, "%02x", ch); 234 dst += 2; 235 } 236 *dst = 0; 237 return dp; 238 } 239 240 int 241 smb_simpledecrypt(char *dst, const char *src) 242 { 243 char *ep, hexval[3]; 244 int len, ch, pos; 245 246 if (strncmp(src, "$$1", 3) != 0) 247 return EINVAL; 248 src += 3; 249 len = strlen(src); 250 if (len & 1) 251 return EINVAL; 252 len /= 2; 253 hexval[2] = 0; 254 pos = 27; 255 while (len--) { 256 hexval[0] = *src++; 257 hexval[1] = *src++; 258 ch = strtoul(hexval, &ep, 16); 259 if (*ep != 0) 260 return EINVAL; 261 ch ^= pos; 262 pos += 13; 263 if (isascii(ch)) 264 ch = (isupper(ch) ? ('A' + (ch - 'A' + 13) % 26) : 265 islower(ch) ? ('a' + (ch - 'a' + 13) % 26) : ch); 266 *dst++ = ch; 267 } 268 *dst = 0; 269 return 0; 270 } 271 272 273 #ifdef APPLE 274 static int 275 safe_execv(char *args[]) 276 { 277 int pid; 278 union wait status; 279 280 pid = fork(); 281 if (pid == 0) { 282 (void)execv(args[0], args); 283 errx(EX_OSERR, "%s: execv %s failed, %s\n", __progname, 284 args[0], strerror(errno)); 285 } 286 if (pid == -1) { 287 fprintf(stderr, "%s: fork failed, %s\n", __progname, 288 strerror(errno)); 289 return (1); 290 } 291 if (wait4(pid, (int *)&status, 0, NULL) != pid) { 292 fprintf(stderr, "%s: BUG executing %s command\n", __progname, 293 args[0]); 294 return (1); 295 } else if (!WIFEXITED(status)) { 296 fprintf(stderr, "%s: %s command aborted by signal %d\n", 297 __progname, args[0], WTERMSIG(status)); 298 return (1); 299 } else if (WEXITSTATUS(status)) { 300 fprintf(stderr, "%s: %s command failed, exit status %d: %s\n", 301 __progname, args[0], WEXITSTATUS(status), 302 strerror(WEXITSTATUS(status))); 303 return (1); 304 } 305 return (0); 306 } 307 308 309 void 310 dropsuid() 311 { 312 /* drop setuid root privs asap */ 313 eff_uid = geteuid(); 314 real_uid = getuid(); 315 seteuid(real_uid); 316 return; 317 } 318 319 320 static int 321 kextisloaded(char * kextname) 322 { 323 mach_port_t kernel_port; 324 kmod_info_t *k, *loaded_modules = 0; 325 int err, loaded_count = 0; 326 327 /* on error return not loaded - to make loadsmbvfs fail */ 328 329 err = task_for_pid(mach_task_self(), 0, &kernel_port); 330 if (err) { 331 fprintf(stderr, "%s: %s: %s\n", __progname, 332 "unable to get kernel task port", 333 mach_error_string(err)); 334 return (0); 335 } 336 err = kmod_get_info(kernel_port, (void *)&loaded_modules, 337 &loaded_count); /* never freed */ 338 if (err) { 339 fprintf(stderr, "%s: %s: %s\n", __progname, 340 "kmod_get_info() failed", 341 mach_error_string(err)); 342 return (0); 343 } 344 for (k = loaded_modules; k; k = k->next ? k+1 : 0) 345 if (!strcmp(k->name, kextname)) 346 return (1); 347 return (0); 348 } 349 350 351 #define KEXTLOAD_COMMAND "/sbin/kextload" 352 #define FS_KEXT_DIR "/System/Library/Extensions/smbfs.kext" 353 #define FULL_KEXTNAME "com.apple.filesystems.smbfs" 354 355 356 int 357 loadsmbvfs() 358 { 359 const char *kextargs[] = {KEXTLOAD_COMMAND, FS_KEXT_DIR, NULL}; 360 int error = 0; 361 362 /* 363 * temporarily revert to root (required for kextload) 364 */ 365 seteuid(eff_uid); 366 if (!kextisloaded(FULL_KEXTNAME)) { 367 error = safe_execv(kextargs); 368 if (!error) 369 error = !kextisloaded(FULL_KEXTNAME); 370 } 371 seteuid(real_uid); /* and back to real user */ 372 return (error); 373 } 374 #endif /* APPLE */ 375