File:  [ELWIX - Embedded LightWeight unIX -] / embedaddon / sudo / common / sudo_dso.c
Revision 1.1.1.1 (vendor branch): download - view: text, annotated - select for diffs - revision graph
Sun Jun 15 16:12:54 2014 UTC (10 years ago) by misho
Branches: sudo, MAIN
CVS tags: v1_8_10p3_0, v1_8_10p3, HEAD
sudo v 1.8.10p3

    1: /*
    2:  * Copyright (c) 2010, 2012-2014 Todd C. Miller <Todd.Miller@courtesan.com>
    3:  *
    4:  * Permission to use, copy, modify, and distribute this software for any
    5:  * purpose with or without fee is hereby granted, provided that the above
    6:  * copyright notice and this permission notice appear in all copies.
    7:  *
    8:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
    9:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
   10:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
   11:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
   12:  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
   13:  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
   14:  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
   15:  */
   16: 
   17: #include <config.h>
   18: 
   19: #include <sys/types.h>
   20: 
   21: #include <stdio.h>
   22: #ifdef STDC_HEADERS
   23: # include <stdlib.h>
   24: # include <stddef.h>
   25: #else
   26: # ifdef HAVE_STDLIB_H
   27: #  include <stdlib.h>
   28: # endif
   29: #endif /* STDC_HEADERS */
   30: #ifdef HAVE_STRING_H
   31: # include <string.h>
   32: #endif /* HAVE_STRING_H */
   33: #ifdef HAVE_STRINGS_H
   34: # include <strings.h>
   35: #endif /* HAVE_STRINGS_H */
   36: #if defined(HAVE_SHL_LOAD)
   37: # include <dl.h>
   38: #elif defined(HAVE_DLOPEN)
   39: # include <dlfcn.h>
   40: #endif
   41: #include <errno.h>
   42: 
   43: #include "sudo_dso.h"
   44: #include "missing.h"
   45: 
   46: /*
   47:  * Pointer for statically compiled symbols.
   48:  */
   49: static struct sudo_preload_table *preload_table;
   50: 
   51: void
   52: sudo_dso_preload_table(struct sudo_preload_table *table)
   53: {
   54:     preload_table = table;
   55: }
   56: 
   57: #if defined(HAVE_SHL_LOAD)
   58: 
   59: # ifndef DYNAMIC_PATH
   60: #  define DYNAMIC_PATH	0
   61: # endif
   62: 
   63: void *
   64: sudo_dso_load(const char *path, int mode)
   65: {
   66:     struct sudo_preload_table *pt;
   67:     int flags = DYNAMIC_PATH | BIND_VERBOSE;
   68: 
   69:     if (mode == 0)
   70: 	mode = SUDO_DSO_LAZY;	/* default behavior */
   71: 
   72:     /* Check prelinked symbols first. */
   73:     if (preload_table != NULL) {
   74: 	for (pt = preload_table; pt->handle != NULL; pt++) {
   75: 	    if (pt->path != NULL && strcmp(path, pt->path) == 0)
   76: 		return pt->handle;
   77: 	}
   78:     }
   79: 
   80:     /* We don't support SUDO_DSO_GLOBAL or SUDO_DSO_LOCAL yet. */
   81:     if (ISSET(mode, SUDO_DSO_LAZY))
   82: 	flags |= BIND_DEFERRED;
   83:     if (ISSET(mode, SUDO_DSO_NOW))
   84: 	flags |= BIND_IMMEDIATE;
   85: 
   86:     return (void *)shl_load(path, flags, 0L);
   87: }
   88: 
   89: int
   90: sudo_dso_unload(void *handle)
   91: {
   92:     struct sudo_preload_table *pt;
   93: 
   94:     /* Check prelinked symbols first. */
   95:     if (preload_table != NULL) {
   96: 	for (pt = preload_table; pt->handle != NULL; pt++) {
   97: 	    if (pt->handle == handle)
   98: 		return 0;
   99: 	}
  100:     }
  101: 
  102:     return shl_unload((shl_t)handle);
  103: }
  104: 
  105: void *
  106: sudo_dso_findsym(void *vhandle, const char *symbol)
  107: {
  108:     struct sudo_preload_table *pt;
  109:     shl_t handle = vhandle;
  110:     void *value = NULL;
  111: 
  112:     /* Check prelinked symbols first. */
  113:     if (preload_table != NULL) {
  114: 	for (pt = preload_table; pt->handle != NULL; pt++) {
  115: 	    if (pt->handle == handle) {
  116: 		struct sudo_preload_symbol *sym;
  117: 		for (sym = pt->symbols; sym->name != NULL; sym++) {
  118: 		    if (strcmp(sym->name, symbol) == 0)
  119: 			return sym->addr;
  120: 		}
  121: 		errno = ENOENT;
  122: 		return NULL;
  123: 	    }
  124: 	}
  125:     }
  126: 
  127:     /*
  128:      * Note that the behavior of of SUDO_DSO_NEXT and SUDO_DSO_SELF 
  129:      * differs from most implementations when called from
  130:      * a shared library.
  131:      */
  132:     if (vhandle == SUDO_DSO_NEXT) {
  133: 	/* Iterate over all shared libs looking for symbol. */
  134: 	shl_t myhandle = PROG_HANDLE;
  135: 	struct shl_descriptor *desc;
  136: 	int idx = 0;
  137: 
  138: 	/* Find program's real handle. */
  139: 	if (shl_gethandle(PROG_HANDLE, &desc) == 0)
  140: 	    myhandle = desc->handle;
  141: 	while (shl_get(idx++, &desc) == 0) {
  142: 	    if (desc->handle == myhandle)
  143: 		continue;
  144: 	    if (shl_findsym(&desc->handle, symbol, TYPE_UNDEFINED, &value) == 0)
  145: 		break;
  146: 	}
  147:     } else {
  148: 	if (vhandle == SUDO_DSO_DEFAULT)
  149: 	    handle = NULL;
  150: 	else if (vhandle == SUDO_DSO_SELF)
  151: 	    handle = PROG_HANDLE;
  152: 	(void)shl_findsym(&handle, symbol, TYPE_UNDEFINED, &value);
  153:     }
  154: 
  155:     return value;
  156: }
  157: 
  158: char *
  159: sudo_dso_strerror(void)
  160: {
  161:     return strerror(errno);
  162: }
  163: 
  164: #elif defined(HAVE_DLOPEN)
  165: 
  166: # ifndef RTLD_GLOBAL
  167: #  define RTLD_GLOBAL	0
  168: # endif
  169: 
  170: void *
  171: sudo_dso_load(const char *path, int mode)
  172: {
  173:     struct sudo_preload_table *pt;
  174:     int flags = 0;
  175: 
  176:     /* Check prelinked symbols first. */
  177:     if (preload_table != NULL) {
  178: 	for (pt = preload_table; pt->handle != NULL; pt++) {
  179: 	    if (pt->path != NULL && strcmp(path, pt->path) == 0)
  180: 		return pt->handle;
  181: 	}
  182:     }
  183: 
  184:     /* Map SUDO_DSO_* -> RTLD_* */
  185:     if (ISSET(mode, SUDO_DSO_LAZY))
  186: 	flags |= RTLD_LAZY;
  187:     if (ISSET(mode, SUDO_DSO_NOW))
  188: 	flags |= RTLD_NOW;
  189:     if (ISSET(mode, SUDO_DSO_GLOBAL))
  190: 	flags |= RTLD_GLOBAL;
  191:     if (ISSET(mode, SUDO_DSO_LOCAL))
  192: 	flags |= RTLD_LOCAL;
  193: 
  194:     return dlopen(path, flags);
  195: }
  196: 
  197: int
  198: sudo_dso_unload(void *handle)
  199: {
  200:     struct sudo_preload_table *pt;
  201: 
  202:     /* Check prelinked symbols first. */
  203:     if (preload_table != NULL) {
  204: 	for (pt = preload_table; pt->handle != NULL; pt++) {
  205: 	    if (pt->handle == handle)
  206: 		return 0;
  207: 	}
  208:     }
  209: 
  210:     return dlclose(handle);
  211: }
  212: 
  213: void *
  214: sudo_dso_findsym(void *handle, const char *symbol)
  215: {
  216:     struct sudo_preload_table *pt;
  217: 
  218:     /* Check prelinked symbols first. */
  219:     if (preload_table != NULL) {
  220: 	for (pt = preload_table; pt->handle != NULL; pt++) {
  221: 	    if (pt->handle == handle) {
  222: 		struct sudo_preload_symbol *sym;
  223: 		for (sym = pt->symbols; sym->name != NULL; sym++) {
  224: 		    if (strcmp(sym->name, symbol) == 0)
  225: 			return sym->addr;
  226: 		}
  227: 		errno = ENOENT;
  228: 		return NULL;
  229: 	    }
  230: 	}
  231:     }
  232: 
  233:     /*
  234:      * Not all implementations support the special handles.
  235:      */
  236:     if (handle == SUDO_DSO_NEXT) {
  237: # ifdef RTLD_NEXT
  238: 	handle = RTLD_NEXT;
  239: # else
  240: 	errno = ENOENT;
  241: 	return NULL;
  242: # endif
  243:     } else if (handle == SUDO_DSO_DEFAULT) {
  244: # ifdef RTLD_DEFAULT
  245: 	handle = RTLD_DEFAULT;
  246: # else
  247: 	errno = ENOENT;
  248: 	return NULL;
  249: # endif
  250:     } else if (handle == SUDO_DSO_SELF) {
  251: # ifdef RTLD_SELF
  252: 	handle = RTLD_SELF;
  253: # else
  254: 	errno = ENOENT;
  255: 	return NULL;
  256: # endif
  257:     }
  258: 
  259:     return dlsym(handle, symbol);
  260: }
  261: 
  262: char *
  263: sudo_dso_strerror(void)
  264: {
  265:     return dlerror();
  266: }
  267: 
  268: #else /* !HAVE_SHL_LOAD && !HAVE_DLOPEN */
  269: 
  270: /*
  271:  * Emulate dlopen() using a static list of symbols compiled into sudo.
  272:  */
  273: void *
  274: sudo_dso_load(const char *path, int mode)
  275: {
  276:     struct sudo_preload_table *pt;
  277: 
  278:     /* Check prelinked symbols first. */
  279:     if (preload_table != NULL) {
  280: 	for (pt = preload_table; pt->handle != NULL; pt++) {
  281: 	    if (pt->path != NULL && strcmp(path, pt->path) == 0)
  282: 		return pt->handle;
  283: 	}
  284:     }
  285:     return NULL;
  286: }
  287: 
  288: int
  289: sudo_dso_unload(void *handle)
  290: {
  291:     struct sudo_preload_table *pt;
  292: 
  293:     if (preload_table != NULL) {
  294: 	for (pt = preload_table; pt->handle != NULL; pt++) {
  295: 	    if (pt->handle == handle)
  296: 		return 0;
  297: 	}
  298:     }
  299:     return -1;
  300: }
  301: 
  302: void *
  303: sudo_dso_findsym(void *handle, const char *symbol)
  304: {
  305:     struct sudo_preload_table *pt;
  306: 
  307:     if (preload_table != NULL) {
  308: 	for (pt = preload_table; pt->handle != NULL; pt++) {
  309: 	    if (pt->handle == handle) {
  310: 		struct sudo_preload_symbol *sym;
  311: 		for (sym = pt->symbols; sym->name != NULL; sym++) {
  312: 		    if (strcmp(sym->name, symbol) == 0)
  313: 			return sym->addr;
  314: 		}
  315: 	    }
  316: 	}
  317:     }
  318:     errno = ENOENT;
  319:     return NULL;
  320: }
  321: 
  322: char *
  323: sudo_dso_strerror(void)
  324: {
  325:     return strerror(errno);
  326: }
  327: #endif /* !HAVE_SHL_LOAD && !HAVE_DLOPEN */

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>