diff options
Diffstat (limited to 'sbin/kldconfig/kldconfig.c')
-rw-r--r-- | sbin/kldconfig/kldconfig.c | 444 |
1 files changed, 444 insertions, 0 deletions
diff --git a/sbin/kldconfig/kldconfig.c b/sbin/kldconfig/kldconfig.c new file mode 100644 index 0000000..979ff2e --- /dev/null +++ b/sbin/kldconfig/kldconfig.c @@ -0,0 +1,444 @@ +/* + * Copyright (c) 2001 Peter Pentchev + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + */ + +#ifndef lint +static const char rcsid[] = + "$FreeBSD$"; +#endif /* not lint */ + +#include <sys/param.h> +#include <sys/types.h> +#include <sys/queue.h> +#include <sys/sysctl.h> + +#include <err.h> +#include <errno.h> +#include <limits.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> + +#if defined(__FreeBSD_version) +#if __FreeBSD_version < 500000 +#define NEED_SLASHTERM +#endif /* < 500000 */ +#else /* defined(__FreeBSD_version) */ +/* just in case.. */ +#define NEED_SLASHTERM +#endif /* defined(__FreeBSD_version) */ + +/* the default sysctl name */ +#define PATHCTL "kern.module_path" + +/* queue structure for the module path broken down into components */ +TAILQ_HEAD(pathhead, pathentry); +struct pathentry { + char *path; + TAILQ_ENTRY(pathentry) next; +}; + +/* the Management Information Base entries for the search path sysctl */ +static int mib[5]; +static size_t miblen; +/* the sysctl name, defaults to PATHCTL */ +static char *pathctl; +/* the sysctl value - the current module search path */ +static char *modpath; +/* flag whether user actions require changing the sysctl value */ +static int changed; + +/* Top-level path management functions */ +static void addpath(struct pathhead *, char *, int, int); +static void rempath(struct pathhead *, char *, int, int); +static void showpath(struct pathhead *); + +/* Low-level path management functions */ +static char *qstring(struct pathhead *); + +/* sysctl-related functions */ +static void getmib(void); +static void getpath(void); +static void parsepath(struct pathhead *, char *, int); +static void setpath(struct pathhead *); + +static void usage(void); + +/* Get the MIB entry for our sysctl */ +static void +getmib(void) +{ + + /* have we already fetched it? */ + if (miblen != 0) + return; + + miblen = sizeof(mib) / sizeof(mib[0]); + if (sysctlnametomib(pathctl, mib, &miblen) != 0) + err(1, "sysctlnametomib(%s)", pathctl); +} + +/* Get the current module search path */ +static void +getpath(void) +{ + char *path; + size_t sz; + + if (modpath != NULL) { + free(modpath); + modpath = NULL; + } + + if (miblen == 0) + getmib(); + if (sysctl(mib, miblen, NULL, &sz, NULL, NULL) == -1) + err(1, "getting path: sysctl(%s) - size only", pathctl); + if ((path = malloc(sz + 1)) == NULL) { + errno = ENOMEM; + err(1, "allocating %u bytes for the path", sz+1); + } + if (sysctl(mib, miblen, path, &sz, NULL, NULL) == -1) + err(1, "getting path: sysctl(%s)", pathctl); + modpath = path; +} + +/* Set the module search path after changing it */ +static void +setpath(struct pathhead *pathq) +{ + char *newpath; + + if (miblen == 0) + getmib(); + if ((newpath = qstring(pathq)) == NULL) { + errno = ENOMEM; + err(1, "building path string"); + } + if (sysctl(mib, miblen, NULL, NULL, newpath, strlen(newpath)+1) == -1) + err(1, "setting path: sysctl(%s)", pathctl); + + if (modpath) + free(modpath); + modpath = newpath; +} + +/* Add/insert a new component to the module search path */ +static void +addpath(struct pathhead *pathq, char *path, int force, int insert) +{ + struct pathentry *pe, *pskip; + char pathbuf[MAXPATHLEN+1]; + size_t len; + static unsigned added = 0; + unsigned i; + + /* + * If the path exists, use it; otherwise, take the user-specified + * path at face value - may be a removed directory. + */ + if (realpath(path, pathbuf) == NULL) + strlcpy(pathbuf, path, sizeof(pathbuf)); + + len = strlen(pathbuf); +#ifdef NEED_SLASHTERM + /* slash-terminate, because the kernel linker said so. */ + if ((len == 0) || (pathbuf[len-1] != '/')) { + if (len == sizeof(pathbuf) - 1) + errx(1, "path too long: %s", pathbuf); + pathbuf[len] = '/'; + } +#else /* NEED_SLASHTERM */ + /* remove a terminating slash if present */ + if ((len > 0) && (pathbuf[len-1] == '/')) + pathbuf[--len] = '\0'; +#endif /* NEED_SLASHTERM */ + + /* is it already in there? */ + TAILQ_FOREACH(pe, pathq, next) + if (!strcmp(pe->path, pathbuf)) + break; + if (pe != NULL) { + if (force) + return; + errx(1, "already in the module search path: %s", pathbuf); + } + + /* OK, allocate and add it. */ + if (((pe = malloc(sizeof(*pe))) == NULL) || + ((pe->path = strdup(pathbuf)) == NULL)) { + errno = ENOMEM; + err(1, "allocating path component"); + } + if (!insert) { + TAILQ_INSERT_TAIL(pathq, pe, next); + } else { + for (i = 0, pskip = TAILQ_FIRST(pathq); i < added; i++) + pskip = TAILQ_NEXT(pskip, next); + if (pskip != NULL) + TAILQ_INSERT_BEFORE(pskip, pe, next); + else + TAILQ_INSERT_TAIL(pathq, pe, next); + added++; + } + changed = 1; +} + +/* Remove a path component from the module search path */ +static void +rempath(struct pathhead *pathq, char *path, int force, int insert __unused) +{ + char pathbuf[MAXPATHLEN+1]; + struct pathentry *pe; + size_t len; + + /* same logic as in addpath() */ + if (realpath(path, pathbuf) == NULL) + strlcpy(pathbuf, path, sizeof(pathbuf)); + + len = strlen(pathbuf); +#ifdef NEED_SLASHTERM + /* slash-terminate, because the kernel linker said so. */ + if ((len == 0) || (pathbuf[len-1] != '/')) { + if (len == sizeof(pathbuf) - 1) + errx(1, "path too long: %s", pathbuf); + pathbuf[len] = '/'; + } +#else /* NEED_SLASHTERM */ + /* remove a terminating slash if present */ + if ((len > 0) && (pathbuf[len-1] == '/')) + pathbuf[--len] = '\0'; +#endif /* NEED_SLASHTERM */ + + /* Is it in there? */ + TAILQ_FOREACH(pe, pathq, next) + if (!strcmp(pe->path, pathbuf)) + break; + if (pe == NULL) { + if (force) + return; + errx(1, "not in module search path: %s", pathbuf); + } + + /* OK, remove it now.. */ + TAILQ_REMOVE(pathq, pe, next); + changed = 1; +} + +/* Display the retrieved module search path */ +static void +showpath(struct pathhead *pathq) +{ + char *s; + + if ((s = qstring(pathq)) == NULL) { + errno = ENOMEM; + err(1, "building path string"); + } + printf("%s\n", s); + free(s); +} + +/* Break a string down into path components, store them into a queue */ +static void +parsepath(struct pathhead *pathq, char *path, int uniq) +{ + char *p; + struct pathentry *pe; + + while ((p = strsep(&path, ";")) != NULL) + if (!uniq) { + if (((pe = malloc(sizeof(pe))) == NULL) || + ((pe->path = strdup(p)) == NULL)) { + errno = ENOMEM; + err(1, "allocating path element"); + } + TAILQ_INSERT_TAIL(pathq, pe, next); + } else { + addpath(pathq, p, 1, 0); + } +} + +/* Recreate a path string from a components queue */ +static char * +qstring(struct pathhead *pathq) +{ + char *s, *p; + struct pathentry *pe; + + s = strdup(""); + TAILQ_FOREACH(pe, pathq, next) { + asprintf(&p, "%s%s%s", + s, pe->path, (TAILQ_NEXT(pe, next) != NULL? ";": "")); + free(s); + if (p == NULL) + return (NULL); + s = p; + } + + return (s); +} + +/* Usage message */ +static void +usage(void) +{ + + fprintf(stderr, "%s\n%s\n", + "usage:\tkldconfig [-dfimnUv] [-S sysctlname] [path..]", + "\tkldconfig -r"); + exit(1); +} + +/* Main function */ +int +main(int argc, char *argv[]) +{ + /* getopt() iterator */ + int c; + /* iterator over argv[] path components */ + int i; + /* Command-line flags: */ + /* "-f" - no diagnostic messages */ + int fflag; + /* "-i" - insert before the first element */ + int iflag; + /* "-m" - merge into the existing path, do not replace it */ + int mflag; + /* "-n" - do not actually set the new module path */ + int nflag; + /* "-r" - print out the current search path */ + int rflag; + /* "-U" - remove duplicate values from the path */ + int uniqflag; + /* "-v" - verbose operation (currently a no-op) */ + int vflag; + /* The higher-level function to call - add/remove */ + void (*act)(struct pathhead *, char *, int, int); + /* The original path */ + char *origpath; + /* The module search path broken down into components */ + struct pathhead pathq; + + fflag = iflag = mflag = nflag = rflag = uniqflag = vflag = 0; + act = addpath; + origpath = NULL; + if ((pathctl = strdup(PATHCTL)) == NULL) { + /* this is just too paranoid ;) */ + errno = ENOMEM; + err(1, "initializing sysctl name %s", PATHCTL); + } + + /* If no arguments and no options are specified, force '-m' */ + if (argc == 1) + mflag = 1; + + while ((c = getopt(argc, argv, "dfimnrS:Uv")) != -1) + switch (c) { + case 'd': + if (iflag || mflag) + usage(); + act = rempath; + break; + case 'f': + fflag = 1; + break; + case 'i': + if (act != addpath) + usage(); + iflag = 1; + break; + case 'm': + if (act != addpath) + usage(); + mflag = 1; + break; + case 'n': + nflag = 1; + break; + case 'r': + rflag = 1; + break; + case 'S': + free(pathctl); + if ((pathctl = strdup(optarg)) == NULL) { + errno = ENOMEM; + err(1, "sysctl name %s", optarg); + } + break; + case 'U': + uniqflag = 1; + break; + case 'v': + vflag++; + break; + default: + usage(); + } + + argc -= optind; + argv += optind; + + /* The '-r' flag cannot be used when paths are also specified */ + if (rflag && (argc > 0)) + usage(); + + TAILQ_INIT(&pathq); + + /* Retrieve and store the path from the sysctl value */ + getpath(); + if ((origpath = strdup(modpath)) == NULL) { + errno = ENOMEM; + err(1, "saving the original search path"); + } + + /* + * Break down the path into the components queue if: + * - we are NOT adding paths, OR + * - the 'merge' flag is specified, OR + * - the 'print only' flag is specified, OR + * - the 'unique' flag is specified. + */ + if ((act != addpath) || mflag || rflag || uniqflag) + parsepath(&pathq, modpath, uniqflag); + else if (modpath[0] != '\0') + changed = 1; + + /* Process the path arguments */ + for (i = 0; i < argc; i++) + act(&pathq, argv[i], fflag, iflag); + + if (changed && !nflag) + setpath(&pathq); + + if (rflag || (changed && vflag)) { + if (changed && (vflag > 1)) + printf("%s -> ", origpath); + showpath(&pathq); + } + + return (0); +} |