1 /*- 2 * Copyright (c) 1997 Doug Rabson 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 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 */ 26 27 #include <sys/cdefs.h> 28 __FBSDID("$FreeBSD$"); 29 30 #include <sys/types.h> 31 #include <sys/param.h> 32 #include <sys/linker.h> 33 #include <sys/sysctl.h> 34 #include <sys/stat.h> 35 #include <err.h> 36 #include <stdio.h> 37 #include <stdlib.h> 38 #include <string.h> 39 #include <unistd.h> 40 #include <errno.h> 41 42 #define PATHCTL "kern.module_path" 43 44 static int path_check(const char *, int); 45 static void usage(void); 46 47 /* 48 * Check to see if the requested module is specified as a filename with no 49 * path. If so and if a file by the same name exists in the module path, 50 * warn the user that the module in the path will be used in preference. 51 */ 52 static int 53 path_check(const char *kldname, int quiet) 54 { 55 int mib[5], found; 56 size_t miblen, pathlen; 57 char kldpath[MAXPATHLEN]; 58 char *path, *tmppath, *element; 59 struct stat sb; 60 dev_t dev; 61 ino_t ino; 62 63 if (strchr(kldname, '/') != NULL) { 64 return (0); 65 } 66 if (strstr(kldname, ".ko") == NULL) { 67 return (0); 68 } 69 if (stat(kldname, &sb) != 0) { 70 return (0); 71 } 72 73 found = 0; 74 dev = sb.st_dev; 75 ino = sb.st_ino; 76 77 miblen = sizeof(mib) / sizeof(mib[0]); 78 if (sysctlnametomib(PATHCTL, mib, &miblen) != 0) { 79 err(1, "sysctlnametomib(%s)", PATHCTL); 80 } 81 if (sysctl(mib, miblen, NULL, &pathlen, NULL, 0) == -1) { 82 err(1, "getting path: sysctl(%s) - size only", PATHCTL); 83 } 84 path = malloc(pathlen + 1); 85 if (path == NULL) { 86 err(1, "allocating %lu bytes for the path", 87 (unsigned long)pathlen + 1); 88 } 89 if (sysctl(mib, miblen, path, &pathlen, NULL, 0) == -1) { 90 err(1, "getting path: sysctl(%s)", PATHCTL); 91 } 92 tmppath = path; 93 94 while ((element = strsep(&tmppath, ";")) != NULL) { 95 strlcpy(kldpath, element, MAXPATHLEN); 96 if (kldpath[strlen(kldpath) - 1] != '/') { 97 strlcat(kldpath, "/", MAXPATHLEN); 98 } 99 strlcat(kldpath, kldname, MAXPATHLEN); 100 101 if (stat(kldpath, &sb) == -1) { 102 continue; 103 } 104 105 found = 1; 106 107 if (sb.st_dev != dev || sb.st_ino != ino) { 108 if (!quiet) { 109 warnx("%s will be loaded from %s, not the " 110 "current directory", kldname, element); 111 } 112 break; 113 } else if (sb.st_dev == dev && sb.st_ino == ino) { 114 break; 115 } 116 } 117 118 free(path); 119 120 if (!found) { 121 if (!quiet) { 122 warnx("%s is not in the module path", kldname); 123 } 124 return (-1); 125 } 126 127 return (0); 128 } 129 130 static void 131 usage(void) 132 { 133 fprintf(stderr, "usage: kldload [-nqv] file ...\n"); 134 exit(1); 135 } 136 137 int 138 main(int argc, char** argv) 139 { 140 int c; 141 int errors; 142 int fileid; 143 int verbose; 144 int quiet; 145 int check_loaded; 146 147 errors = 0; 148 verbose = 0; 149 quiet = 0; 150 check_loaded = 0; 151 152 while ((c = getopt(argc, argv, "nqv")) != -1) { 153 switch (c) { 154 case 'q': 155 quiet = 1; 156 verbose = 0; 157 break; 158 case 'v': 159 verbose = 1; 160 quiet = 0; 161 break; 162 case 'n': 163 check_loaded = 1; 164 break; 165 default: 166 usage(); 167 } 168 } 169 argc -= optind; 170 argv += optind; 171 172 if (argc == 0) 173 usage(); 174 175 while (argc-- != 0) { 176 if (path_check(argv[0], quiet) == 0) { 177 fileid = kldload(argv[0]); 178 if (fileid < 0) { 179 if (check_loaded != 0 && errno == EEXIST) { 180 if (verbose) 181 printf("%s is already " 182 "loaded\n", argv[0]); 183 } else { 184 warn("can't load %s", argv[0]); 185 errors++; 186 } 187 } else { 188 if (verbose) 189 printf("Loaded %s, id=%d\n", argv[0], 190 fileid); 191 } 192 } else { 193 errors++; 194 } 195 argv++; 196 } 197 198 return (errors ? 1 : 0); 199 } 200