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 */ 377