11de7b4b8SPedro F. Giffuni /*- 24d846d26SWarner Losh * SPDX-License-Identifier: BSD-2-Clause 31de7b4b8SPedro F. Giffuni * 4fb2ad9d3SUlrich Spörlein * Copyright © 2002, Jörg Wunsch 5b78dcb87SJohan Karlsson * 6b78dcb87SJohan Karlsson * Redistribution and use in source and binary forms, with or without 7b78dcb87SJohan Karlsson * modification, are permitted provided that the following conditions 8b78dcb87SJohan Karlsson * are met: 9b78dcb87SJohan Karlsson * 1. Redistributions of source code must retain the above copyright 10b78dcb87SJohan Karlsson * notice, this list of conditions and the following disclaimer. 11b78dcb87SJohan Karlsson * 2. Redistributions in binary form must reproduce the above copyright 12b78dcb87SJohan Karlsson * notice, this list of conditions and the following disclaimer in the 13b78dcb87SJohan Karlsson * documentation and/or other materials provided with the distribution. 14b78dcb87SJohan Karlsson * 15e97f67f5SJoerg Wunsch * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 16e97f67f5SJoerg Wunsch * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 17e97f67f5SJoerg Wunsch * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 18e97f67f5SJoerg Wunsch * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, 19e97f67f5SJoerg Wunsch * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 20e97f67f5SJoerg Wunsch * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 21e97f67f5SJoerg Wunsch * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22e97f67f5SJoerg Wunsch * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, 23e97f67f5SJoerg Wunsch * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING 24e97f67f5SJoerg Wunsch * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 25e97f67f5SJoerg Wunsch * POSSIBILITY OF SUCH DAMAGE. 26b78dcb87SJohan Karlsson */ 27b78dcb87SJohan Karlsson 28e97f67f5SJoerg Wunsch /* 29e97f67f5SJoerg Wunsch * 4.3BSD UI-compatible whereis(1) utility. Rewritten from scratch 30e97f67f5SJoerg Wunsch * since the original 4.3BSD version suffers legal problems that 31e97f67f5SJoerg Wunsch * prevent it from being redistributed, and since the 4.4BSD version 32e97f67f5SJoerg Wunsch * was pretty inferior in functionality. 33e97f67f5SJoerg Wunsch */ 349d1b456eSJohan Karlsson 35e97f67f5SJoerg Wunsch #include <sys/types.h> 36b78dcb87SJohan Karlsson 37b68725e3SJohan Karlsson __FBSDID("$FreeBSD$"); 38b68725e3SJohan Karlsson 39b78dcb87SJohan Karlsson #include <sys/stat.h> 40b78dcb87SJohan Karlsson #include <sys/sysctl.h> 41b78dcb87SJohan Karlsson 42e97f67f5SJoerg Wunsch #include <dirent.h> 43b78dcb87SJohan Karlsson #include <err.h> 44b78dcb87SJohan Karlsson #include <errno.h> 454ff23965STim J. Robbins #include <locale.h> 46e97f67f5SJoerg Wunsch #include <regex.h> 47b78dcb87SJohan Karlsson #include <stdio.h> 48b78dcb87SJohan Karlsson #include <stdlib.h> 49b78dcb87SJohan Karlsson #include <string.h> 50e97f67f5SJoerg Wunsch #include <sysexits.h> 51b78dcb87SJohan Karlsson #include <unistd.h> 52b78dcb87SJohan Karlsson 53e97f67f5SJoerg Wunsch #include "pathnames.h" 54b78dcb87SJohan Karlsson 551f05bc6cSJohan Karlsson #define NO_BIN_FOUND 1 561f05bc6cSJohan Karlsson #define NO_MAN_FOUND 2 571f05bc6cSJohan Karlsson #define NO_SRC_FOUND 4 581f05bc6cSJohan Karlsson 59e97f67f5SJoerg Wunsch typedef const char *ccharp; 60e97f67f5SJoerg Wunsch 61558b00f5SEd Schouten static int opt_a, opt_b, opt_m, opt_q, opt_s, opt_u, opt_x; 62558b00f5SEd Schouten static ccharp *bindirs, *mandirs, *sourcedirs; 63558b00f5SEd Schouten static char **query; 64e97f67f5SJoerg Wunsch 65558b00f5SEd Schouten static const char *sourcepath = PATH_SOURCES; 66e97f67f5SJoerg Wunsch 67558b00f5SEd Schouten static char *colonify(ccharp *); 68558b00f5SEd Schouten static int contains(ccharp *, const char *); 69558b00f5SEd Schouten static void decolonify(char *, ccharp **, int *); 70558b00f5SEd Schouten static void defaults(void); 71558b00f5SEd Schouten static void scanopts(int, char **); 72558b00f5SEd Schouten static void usage(void); 73e97f67f5SJoerg Wunsch 74e97f67f5SJoerg Wunsch /* 75e97f67f5SJoerg Wunsch * Throughout this program, a number of strings are dynamically 76e97f67f5SJoerg Wunsch * allocated but never freed. Their memory is written to when 77e97f67f5SJoerg Wunsch * splitting the strings into string lists which will later be 78e97f67f5SJoerg Wunsch * processed. Since it's important that those string lists remain 79e97f67f5SJoerg Wunsch * valid even after the functions allocating the memory returned, 80e97f67f5SJoerg Wunsch * those functions cannot free them. They could be freed only at end 81e97f67f5SJoerg Wunsch * of main(), which is pretty pointless anyway. 82e97f67f5SJoerg Wunsch * 83e97f67f5SJoerg Wunsch * The overall amount of memory to be allocated for processing the 84e97f67f5SJoerg Wunsch * strings is not expected to exceed a few kilobytes. For that 85e97f67f5SJoerg Wunsch * reason, allocation can usually always be assumed to succeed (within 86e97f67f5SJoerg Wunsch * a virtual memory environment), thus we simply bail out using 87e97f67f5SJoerg Wunsch * abort(3) in case of an allocation failure. 88e97f67f5SJoerg Wunsch */ 89e97f67f5SJoerg Wunsch 90558b00f5SEd Schouten static void 91e97f67f5SJoerg Wunsch usage(void) 92b78dcb87SJohan Karlsson { 9348baa772SRuslan Ermilov (void)fprintf(stderr, 9448baa772SRuslan Ermilov "usage: whereis [-abmqsux] [-BMS dir ... -f] program ...\n"); 9548baa772SRuslan Ermilov exit(EX_USAGE); 96e97f67f5SJoerg Wunsch } 97b78dcb87SJohan Karlsson 98e97f67f5SJoerg Wunsch /* 99e97f67f5SJoerg Wunsch * Scan options passed to program. 100e97f67f5SJoerg Wunsch * 101e97f67f5SJoerg Wunsch * Note that the -B/-M/-S options expect a list of directory 102e97f67f5SJoerg Wunsch * names that must be terminated with -f. 103e97f67f5SJoerg Wunsch */ 104558b00f5SEd Schouten static void 105e97f67f5SJoerg Wunsch scanopts(int argc, char **argv) 106e97f67f5SJoerg Wunsch { 1076c97c3d1SStefan Farfeleder int c, i; 108e97f67f5SJoerg Wunsch ccharp **dirlist; 109e97f67f5SJoerg Wunsch 1101f05bc6cSJohan Karlsson while ((c = getopt(argc, argv, "BMSabfmqsux")) != -1) 111e97f67f5SJoerg Wunsch switch (c) { 112e97f67f5SJoerg Wunsch case 'B': 113e97f67f5SJoerg Wunsch dirlist = &bindirs; 114e97f67f5SJoerg Wunsch goto dolist; 115e97f67f5SJoerg Wunsch 116e97f67f5SJoerg Wunsch case 'M': 117e97f67f5SJoerg Wunsch dirlist = &mandirs; 118e97f67f5SJoerg Wunsch goto dolist; 119e97f67f5SJoerg Wunsch 120e97f67f5SJoerg Wunsch case 'S': 121e97f67f5SJoerg Wunsch dirlist = &sourcedirs; 122e97f67f5SJoerg Wunsch dolist: 123e97f67f5SJoerg Wunsch i = 0; 124ad23c583SJohan Karlsson *dirlist = realloc(*dirlist, (i + 1) * sizeof(char *)); 125ad23c583SJohan Karlsson (*dirlist)[i] = NULL; 126e97f67f5SJoerg Wunsch while (optind < argc && 127e97f67f5SJoerg Wunsch strcmp(argv[optind], "-f") != 0 && 128e97f67f5SJoerg Wunsch strcmp(argv[optind], "-B") != 0 && 129e97f67f5SJoerg Wunsch strcmp(argv[optind], "-M") != 0 && 130e97f67f5SJoerg Wunsch strcmp(argv[optind], "-S") != 0) { 131ad23c583SJohan Karlsson decolonify(argv[optind], dirlist, &i); 132e97f67f5SJoerg Wunsch optind++; 133e97f67f5SJoerg Wunsch } 134e97f67f5SJoerg Wunsch break; 135e97f67f5SJoerg Wunsch 1361f05bc6cSJohan Karlsson case 'a': 1371f05bc6cSJohan Karlsson opt_a = 1; 1381f05bc6cSJohan Karlsson break; 1391f05bc6cSJohan Karlsson 140e97f67f5SJoerg Wunsch case 'b': 141e97f67f5SJoerg Wunsch opt_b = 1; 142e97f67f5SJoerg Wunsch break; 143e97f67f5SJoerg Wunsch 144e97f67f5SJoerg Wunsch case 'f': 145e97f67f5SJoerg Wunsch goto breakout; 146e97f67f5SJoerg Wunsch 147e97f67f5SJoerg Wunsch case 'm': 148e97f67f5SJoerg Wunsch opt_m = 1; 149e97f67f5SJoerg Wunsch break; 150e97f67f5SJoerg Wunsch 151e97f67f5SJoerg Wunsch case 'q': 152e97f67f5SJoerg Wunsch opt_q = 1; 153e97f67f5SJoerg Wunsch break; 154e97f67f5SJoerg Wunsch 155e97f67f5SJoerg Wunsch case 's': 156e97f67f5SJoerg Wunsch opt_s = 1; 157e97f67f5SJoerg Wunsch break; 158e97f67f5SJoerg Wunsch 159e97f67f5SJoerg Wunsch case 'u': 160e97f67f5SJoerg Wunsch opt_u = 1; 161e97f67f5SJoerg Wunsch break; 162e97f67f5SJoerg Wunsch 163e97f67f5SJoerg Wunsch case 'x': 164e97f67f5SJoerg Wunsch opt_x = 1; 165e97f67f5SJoerg Wunsch break; 166e97f67f5SJoerg Wunsch 167b78dcb87SJohan Karlsson default: 168b78dcb87SJohan Karlsson usage(); 169b78dcb87SJohan Karlsson } 170e97f67f5SJoerg Wunsch breakout: 171e97f67f5SJoerg Wunsch if (optind == argc) 172b78dcb87SJohan Karlsson usage(); 173e97f67f5SJoerg Wunsch query = argv + optind; 174b78dcb87SJohan Karlsson } 175b78dcb87SJohan Karlsson 176e97f67f5SJoerg Wunsch /* 177e97f67f5SJoerg Wunsch * Find out whether string `s' is contained in list `cpp'. 178e97f67f5SJoerg Wunsch */ 179558b00f5SEd Schouten static int 180e97f67f5SJoerg Wunsch contains(ccharp *cpp, const char *s) 181e97f67f5SJoerg Wunsch { 182e97f67f5SJoerg Wunsch ccharp cp; 183b78dcb87SJohan Karlsson 184e97f67f5SJoerg Wunsch if (cpp == NULL) 185e97f67f5SJoerg Wunsch return (0); 186e97f67f5SJoerg Wunsch 187e97f67f5SJoerg Wunsch while ((cp = *cpp) != NULL) { 188e97f67f5SJoerg Wunsch if (strcmp(cp, s) == 0) 189e97f67f5SJoerg Wunsch return (1); 190e97f67f5SJoerg Wunsch cpp++; 191e97f67f5SJoerg Wunsch } 192b78dcb87SJohan Karlsson return (0); 193b78dcb87SJohan Karlsson } 194b78dcb87SJohan Karlsson 195e97f67f5SJoerg Wunsch /* 196e97f67f5SJoerg Wunsch * Split string `s' at colons, and pass it to the string list pointed 197e97f67f5SJoerg Wunsch * to by `cppp' (which has `*ip' elements). Note that the original 198e97f67f5SJoerg Wunsch * string is modified by replacing the colon with a NUL byte. The 199e97f67f5SJoerg Wunsch * partial string is only added if it has a length greater than 0, and 200e97f67f5SJoerg Wunsch * if it's not already contained in the string list. 201e97f67f5SJoerg Wunsch */ 202558b00f5SEd Schouten static void 203e97f67f5SJoerg Wunsch decolonify(char *s, ccharp **cppp, int *ip) 204b78dcb87SJohan Karlsson { 205e97f67f5SJoerg Wunsch char *cp; 206b78dcb87SJohan Karlsson 207e97f67f5SJoerg Wunsch while ((cp = strchr(s, ':')), *s != '\0') { 208e97f67f5SJoerg Wunsch if (cp) 209e97f67f5SJoerg Wunsch *cp = '\0'; 210e97f67f5SJoerg Wunsch if (strlen(s) && !contains(*cppp, s)) { 211e97f67f5SJoerg Wunsch *cppp = realloc(*cppp, (*ip + 2) * sizeof(char *)); 212c1eac826SBaptiste Daroussin if (*cppp == NULL) 213e97f67f5SJoerg Wunsch abort(); 214e97f67f5SJoerg Wunsch (*cppp)[*ip] = s; 215e97f67f5SJoerg Wunsch (*cppp)[*ip + 1] = NULL; 216e97f67f5SJoerg Wunsch (*ip)++; 217e97f67f5SJoerg Wunsch } 218e97f67f5SJoerg Wunsch if (cp) 219e97f67f5SJoerg Wunsch s = cp + 1; 220e97f67f5SJoerg Wunsch else 221e97f67f5SJoerg Wunsch break; 222e97f67f5SJoerg Wunsch } 223e97f67f5SJoerg Wunsch } 224e97f67f5SJoerg Wunsch 225e97f67f5SJoerg Wunsch /* 226e97f67f5SJoerg Wunsch * Join string list `cpp' into a colon-separated string. 227e97f67f5SJoerg Wunsch */ 228558b00f5SEd Schouten static char * 229e97f67f5SJoerg Wunsch colonify(ccharp *cpp) 230e97f67f5SJoerg Wunsch { 231e97f67f5SJoerg Wunsch size_t s; 232e97f67f5SJoerg Wunsch char *cp; 233e97f67f5SJoerg Wunsch int i; 234e97f67f5SJoerg Wunsch 235e97f67f5SJoerg Wunsch if (cpp == NULL) 236e97f67f5SJoerg Wunsch return (0); 237e97f67f5SJoerg Wunsch 238e97f67f5SJoerg Wunsch for (s = 0, i = 0; cpp[i] != NULL; i++) 239e97f67f5SJoerg Wunsch s += strlen(cpp[i]) + 1; 240e97f67f5SJoerg Wunsch if ((cp = malloc(s + 1)) == NULL) 241e97f67f5SJoerg Wunsch abort(); 242e97f67f5SJoerg Wunsch for (i = 0, *cp = '\0'; cpp[i] != NULL; i++) { 243e97f67f5SJoerg Wunsch strcat(cp, cpp[i]); 244e97f67f5SJoerg Wunsch strcat(cp, ":"); 245e97f67f5SJoerg Wunsch } 246e97f67f5SJoerg Wunsch cp[s - 1] = '\0'; /* eliminate last colon */ 247e97f67f5SJoerg Wunsch 248e97f67f5SJoerg Wunsch return (cp); 249e97f67f5SJoerg Wunsch } 250e97f67f5SJoerg Wunsch 251e97f67f5SJoerg Wunsch /* 252e97f67f5SJoerg Wunsch * Provide defaults for all options and directory lists. 253e97f67f5SJoerg Wunsch */ 254558b00f5SEd Schouten static void 255e97f67f5SJoerg Wunsch defaults(void) 256e97f67f5SJoerg Wunsch { 257e97f67f5SJoerg Wunsch size_t s; 258e97f67f5SJoerg Wunsch char *b, buf[BUFSIZ], *cp; 259e97f67f5SJoerg Wunsch int nele; 260e97f67f5SJoerg Wunsch FILE *p; 261e97f67f5SJoerg Wunsch DIR *dir; 262e97f67f5SJoerg Wunsch struct stat sb; 263e97f67f5SJoerg Wunsch struct dirent *dirp; 264c454c571SStefan Eßer const int oid[2] = {CTL_USER, USER_CS_PATH}; 265e97f67f5SJoerg Wunsch 266e97f67f5SJoerg Wunsch /* default to -bms if none has been specified */ 267e97f67f5SJoerg Wunsch if (!opt_b && !opt_m && !opt_s) 268e97f67f5SJoerg Wunsch opt_b = opt_m = opt_s = 1; 269e97f67f5SJoerg Wunsch 2708e4c33e9SJohan Karlsson /* -b defaults to default path + /usr/libexec + 27111d9aa67SColin Percival * user's path */ 272e97f67f5SJoerg Wunsch if (!bindirs) { 273c454c571SStefan Eßer if (sysctl(oid, 2, NULL, &s, NULL, 0) == -1) 274c454c571SStefan Eßer err(EX_OSERR, "sysctl(\"user.cs_path\")"); 275e97f67f5SJoerg Wunsch if ((b = malloc(s + 1)) == NULL) 276e97f67f5SJoerg Wunsch abort(); 277c454c571SStefan Eßer if (sysctl(oid, 2, b, &s, NULL, 0) == -1) 278c454c571SStefan Eßer err(EX_OSERR, "sysctl(\"user.cs_path\")"); 279e97f67f5SJoerg Wunsch nele = 0; 280e97f67f5SJoerg Wunsch decolonify(b, &bindirs, &nele); 28111d9aa67SColin Percival bindirs = realloc(bindirs, (nele + 2) * sizeof(char *)); 282e97f67f5SJoerg Wunsch if (bindirs == NULL) 283e97f67f5SJoerg Wunsch abort(); 284103d6646SJohan Karlsson bindirs[nele++] = PATH_LIBEXEC; 285e97f67f5SJoerg Wunsch bindirs[nele] = NULL; 286e97f67f5SJoerg Wunsch if ((cp = getenv("PATH")) != NULL) { 287e97f67f5SJoerg Wunsch /* don't destroy the original environment... */ 288f5ce1402SStefan Eßer b = strdup(cp); 289f5ce1402SStefan Eßer if (b == NULL) 290e97f67f5SJoerg Wunsch abort(); 291e97f67f5SJoerg Wunsch decolonify(b, &bindirs, &nele); 292e97f67f5SJoerg Wunsch } 293e97f67f5SJoerg Wunsch } 294e97f67f5SJoerg Wunsch 295e97f67f5SJoerg Wunsch /* -m defaults to $(manpath) */ 296e97f67f5SJoerg Wunsch if (!mandirs) { 297e97f67f5SJoerg Wunsch if ((p = popen(MANPATHCMD, "r")) == NULL) 298e97f67f5SJoerg Wunsch err(EX_OSERR, "cannot execute manpath command"); 299e97f67f5SJoerg Wunsch if (fgets(buf, BUFSIZ - 1, p) == NULL || 300e97f67f5SJoerg Wunsch pclose(p)) 301e97f67f5SJoerg Wunsch err(EX_OSERR, "error processing manpath results"); 302e97f67f5SJoerg Wunsch if ((b = strchr(buf, '\n')) != NULL) 303e97f67f5SJoerg Wunsch *b = '\0'; 304f5ce1402SStefan Eßer b = strdup(buf); 305f5ce1402SStefan Eßer if (b == NULL) 306e97f67f5SJoerg Wunsch abort(); 307e97f67f5SJoerg Wunsch nele = 0; 308e97f67f5SJoerg Wunsch decolonify(b, &mandirs, &nele); 309e97f67f5SJoerg Wunsch } 310e97f67f5SJoerg Wunsch 311e97f67f5SJoerg Wunsch /* -s defaults to precompiled list, plus subdirs of /usr/ports */ 312e97f67f5SJoerg Wunsch if (!sourcedirs) { 313f5ce1402SStefan Eßer b = strdup(sourcepath); 314f5ce1402SStefan Eßer if (b == NULL) 315e97f67f5SJoerg Wunsch abort(); 316e97f67f5SJoerg Wunsch nele = 0; 317e97f67f5SJoerg Wunsch decolonify(b, &sourcedirs, &nele); 318e97f67f5SJoerg Wunsch 319e97f67f5SJoerg Wunsch if (stat(PATH_PORTS, &sb) == -1) { 320e97f67f5SJoerg Wunsch if (errno == ENOENT) 321e97f67f5SJoerg Wunsch /* no /usr/ports, we are done */ 322e97f67f5SJoerg Wunsch return; 323e97f67f5SJoerg Wunsch err(EX_OSERR, "stat(" PATH_PORTS ")"); 324e97f67f5SJoerg Wunsch } 325e97f67f5SJoerg Wunsch if ((sb.st_mode & S_IFMT) != S_IFDIR) 326e97f67f5SJoerg Wunsch /* /usr/ports is not a directory, ignore */ 327e97f67f5SJoerg Wunsch return; 328e97f67f5SJoerg Wunsch if (access(PATH_PORTS, R_OK | X_OK) != 0) 329e97f67f5SJoerg Wunsch return; 330e97f67f5SJoerg Wunsch if ((dir = opendir(PATH_PORTS)) == NULL) 331e97f67f5SJoerg Wunsch err(EX_OSERR, "opendir" PATH_PORTS ")"); 332e97f67f5SJoerg Wunsch while ((dirp = readdir(dir)) != NULL) { 3332b7b4962SJoerg Wunsch /* 3342b7b4962SJoerg Wunsch * Not everything below PATH_PORTS is of 3352b7b4962SJoerg Wunsch * interest. First, all dot files and 3362b7b4962SJoerg Wunsch * directories (e. g. .snap) can be ignored. 3372b7b4962SJoerg Wunsch * Also, all subdirectories starting with a 3382b7b4962SJoerg Wunsch * capital letter are not going to be 3392b7b4962SJoerg Wunsch * examined, as they are used for internal 3402b7b4962SJoerg Wunsch * purposes (Mk, Tools, ...). This also 3412b7b4962SJoerg Wunsch * matches a possible CVS subdirectory. 3422b7b4962SJoerg Wunsch * Finally, the distfiles subdirectory is also 3432b7b4962SJoerg Wunsch * special, and should not be considered to 3442b7b4962SJoerg Wunsch * avoid false matches. 3452b7b4962SJoerg Wunsch */ 346e97f67f5SJoerg Wunsch if (dirp->d_name[0] == '.' || 3472b7b4962SJoerg Wunsch /* 3482b7b4962SJoerg Wunsch * isupper() not used on purpose: the 3492b7b4962SJoerg Wunsch * check is supposed to default to the C 3502b7b4962SJoerg Wunsch * locale instead of the current user's 3512b7b4962SJoerg Wunsch * locale. 3522b7b4962SJoerg Wunsch */ 3532b7b4962SJoerg Wunsch (dirp->d_name[0] >= 'A' && dirp->d_name[0] <= 'Z') || 3542b7b4962SJoerg Wunsch strcmp(dirp->d_name, "distfiles") == 0) 355e97f67f5SJoerg Wunsch continue; 356e97f67f5SJoerg Wunsch if ((b = malloc(sizeof PATH_PORTS + 1 + dirp->d_namlen)) 357e97f67f5SJoerg Wunsch == NULL) 358e97f67f5SJoerg Wunsch abort(); 359e97f67f5SJoerg Wunsch strcpy(b, PATH_PORTS); 360e97f67f5SJoerg Wunsch strcat(b, "/"); 361e97f67f5SJoerg Wunsch strcat(b, dirp->d_name); 362e97f67f5SJoerg Wunsch if (stat(b, &sb) == -1 || 363e97f67f5SJoerg Wunsch (sb.st_mode & S_IFMT) != S_IFDIR || 364e97f67f5SJoerg Wunsch access(b, R_OK | X_OK) != 0) { 365e97f67f5SJoerg Wunsch free(b); 366e97f67f5SJoerg Wunsch continue; 367e97f67f5SJoerg Wunsch } 368e97f67f5SJoerg Wunsch sourcedirs = realloc(sourcedirs, 369e97f67f5SJoerg Wunsch (nele + 2) * sizeof(char *)); 370e97f67f5SJoerg Wunsch if (sourcedirs == NULL) 371e97f67f5SJoerg Wunsch abort(); 372e97f67f5SJoerg Wunsch sourcedirs[nele++] = b; 373e97f67f5SJoerg Wunsch sourcedirs[nele] = NULL; 374e97f67f5SJoerg Wunsch } 375e97f67f5SJoerg Wunsch closedir(dir); 376e97f67f5SJoerg Wunsch } 377e97f67f5SJoerg Wunsch } 378e97f67f5SJoerg Wunsch 379e97f67f5SJoerg Wunsch int 380e97f67f5SJoerg Wunsch main(int argc, char **argv) 381e97f67f5SJoerg Wunsch { 382e97f67f5SJoerg Wunsch int unusual, i, printed; 383e97f67f5SJoerg Wunsch char *bin, buf[BUFSIZ], *cp, *cp2, *man, *name, *src; 384e97f67f5SJoerg Wunsch ccharp *dp; 3851f05bc6cSJohan Karlsson size_t nlen, olen, s; 386e97f67f5SJoerg Wunsch struct stat sb; 387e97f67f5SJoerg Wunsch regex_t re, re2; 388e97f67f5SJoerg Wunsch regmatch_t matches[2]; 389e97f67f5SJoerg Wunsch regoff_t rlen; 390e97f67f5SJoerg Wunsch FILE *p; 391e97f67f5SJoerg Wunsch 3924ff23965STim J. Robbins setlocale(LC_ALL, ""); 3934ff23965STim J. Robbins 394e97f67f5SJoerg Wunsch scanopts(argc, argv); 395e97f67f5SJoerg Wunsch defaults(); 396e97f67f5SJoerg Wunsch 397e97f67f5SJoerg Wunsch if (mandirs == NULL) 398e97f67f5SJoerg Wunsch opt_m = 0; 399e97f67f5SJoerg Wunsch if (bindirs == NULL) 400e97f67f5SJoerg Wunsch opt_b = 0; 401e97f67f5SJoerg Wunsch if (sourcedirs == NULL) 402e97f67f5SJoerg Wunsch opt_s = 0; 403e97f67f5SJoerg Wunsch if (opt_m + opt_b + opt_s == 0) 404e97f67f5SJoerg Wunsch errx(EX_DATAERR, "no directories to search"); 405e97f67f5SJoerg Wunsch 406e97f67f5SJoerg Wunsch if (opt_m) { 407e97f67f5SJoerg Wunsch setenv("MANPATH", colonify(mandirs), 1); 408e97f67f5SJoerg Wunsch if ((i = regcomp(&re, MANWHEREISMATCH, REG_EXTENDED)) != 0) { 409e97f67f5SJoerg Wunsch regerror(i, &re, buf, BUFSIZ - 1); 410e97f67f5SJoerg Wunsch errx(EX_UNAVAILABLE, "regcomp(%s) failed: %s", 411e97f67f5SJoerg Wunsch MANWHEREISMATCH, buf); 412e97f67f5SJoerg Wunsch } 413e97f67f5SJoerg Wunsch } 414e97f67f5SJoerg Wunsch 415e97f67f5SJoerg Wunsch for (; (name = *query) != NULL; query++) { 416e97f67f5SJoerg Wunsch /* strip leading path name component */ 417e97f67f5SJoerg Wunsch if ((cp = strrchr(name, '/')) != NULL) 418e97f67f5SJoerg Wunsch name = cp + 1; 419e97f67f5SJoerg Wunsch /* strip SCCS or RCS suffix/prefix */ 420e97f67f5SJoerg Wunsch if (strlen(name) > 2 && strncmp(name, "s.", 2) == 0) 421e97f67f5SJoerg Wunsch name += 2; 422e97f67f5SJoerg Wunsch if ((s = strlen(name)) > 2 && strcmp(name + s - 2, ",v") == 0) 423e97f67f5SJoerg Wunsch name[s - 2] = '\0'; 424e97f67f5SJoerg Wunsch /* compression suffix */ 425e97f67f5SJoerg Wunsch s = strlen(name); 426e97f67f5SJoerg Wunsch if (s > 2 && 427e97f67f5SJoerg Wunsch (strcmp(name + s - 2, ".z") == 0 || 428e97f67f5SJoerg Wunsch strcmp(name + s - 2, ".Z") == 0)) 429e97f67f5SJoerg Wunsch name[s - 2] = '\0'; 430e97f67f5SJoerg Wunsch else if (s > 3 && 431e97f67f5SJoerg Wunsch strcmp(name + s - 3, ".gz") == 0) 432e97f67f5SJoerg Wunsch name[s - 3] = '\0'; 433e97f67f5SJoerg Wunsch else if (s > 4 && 434e97f67f5SJoerg Wunsch strcmp(name + s - 4, ".bz2") == 0) 435e97f67f5SJoerg Wunsch name[s - 4] = '\0'; 436e97f67f5SJoerg Wunsch 437e97f67f5SJoerg Wunsch unusual = 0; 438e97f67f5SJoerg Wunsch bin = man = src = NULL; 439e97f67f5SJoerg Wunsch s = strlen(name); 440e97f67f5SJoerg Wunsch 441e97f67f5SJoerg Wunsch if (opt_b) { 442e97f67f5SJoerg Wunsch /* 443e97f67f5SJoerg Wunsch * Binaries have to match exactly, and must be regular 444e97f67f5SJoerg Wunsch * executable files. 445e97f67f5SJoerg Wunsch */ 4461f05bc6cSJohan Karlsson unusual = unusual | NO_BIN_FOUND; 447e97f67f5SJoerg Wunsch for (dp = bindirs; *dp != NULL; dp++) { 448e97f67f5SJoerg Wunsch cp = malloc(strlen(*dp) + 1 + s + 1); 449e97f67f5SJoerg Wunsch if (cp == NULL) 450e97f67f5SJoerg Wunsch abort(); 451e97f67f5SJoerg Wunsch strcpy(cp, *dp); 452e97f67f5SJoerg Wunsch strcat(cp, "/"); 453e97f67f5SJoerg Wunsch strcat(cp, name); 454e97f67f5SJoerg Wunsch if (stat(cp, &sb) == 0 && 455e97f67f5SJoerg Wunsch (sb.st_mode & S_IFMT) == S_IFREG && 456e97f67f5SJoerg Wunsch (sb.st_mode & (S_IXUSR | S_IXGRP | S_IXOTH)) 457e97f67f5SJoerg Wunsch != 0) { 4581f05bc6cSJohan Karlsson unusual = unusual & ~NO_BIN_FOUND; 4591f05bc6cSJohan Karlsson if (bin == NULL) { 4601f05bc6cSJohan Karlsson bin = strdup(cp); 4611f05bc6cSJohan Karlsson } else { 4621f05bc6cSJohan Karlsson olen = strlen(bin); 4631f05bc6cSJohan Karlsson nlen = strlen(cp); 4641f05bc6cSJohan Karlsson bin = realloc(bin, 4651f05bc6cSJohan Karlsson olen + nlen + 2); 466d919e0d5SMarcelo Araujo if (bin == NULL) 4671f05bc6cSJohan Karlsson abort(); 4681f05bc6cSJohan Karlsson strcat(bin, " "); 4691f05bc6cSJohan Karlsson strcat(bin, cp); 4701f05bc6cSJohan Karlsson } 4711f05bc6cSJohan Karlsson if (!opt_a) { 4721f05bc6cSJohan Karlsson free(cp); 473e97f67f5SJoerg Wunsch break; 474e97f67f5SJoerg Wunsch } 4751f05bc6cSJohan Karlsson } 476e97f67f5SJoerg Wunsch free(cp); 477e97f67f5SJoerg Wunsch } 478e97f67f5SJoerg Wunsch } 479e97f67f5SJoerg Wunsch 480e97f67f5SJoerg Wunsch if (opt_m) { 481e97f67f5SJoerg Wunsch /* 482e97f67f5SJoerg Wunsch * Ask the man command to perform the search for us. 483e97f67f5SJoerg Wunsch */ 4841f05bc6cSJohan Karlsson unusual = unusual | NO_MAN_FOUND; 4851f05bc6cSJohan Karlsson if (opt_a) 4861f05bc6cSJohan Karlsson cp = malloc(sizeof MANWHEREISALLCMD - 2 + s); 4871f05bc6cSJohan Karlsson else 488e97f67f5SJoerg Wunsch cp = malloc(sizeof MANWHEREISCMD - 2 + s); 4891f05bc6cSJohan Karlsson 490e97f67f5SJoerg Wunsch if (cp == NULL) 491e97f67f5SJoerg Wunsch abort(); 4921f05bc6cSJohan Karlsson 4931f05bc6cSJohan Karlsson if (opt_a) 4941f05bc6cSJohan Karlsson sprintf(cp, MANWHEREISALLCMD, name); 4951f05bc6cSJohan Karlsson else 496e97f67f5SJoerg Wunsch sprintf(cp, MANWHEREISCMD, name); 4971f05bc6cSJohan Karlsson 4981f05bc6cSJohan Karlsson if ((p = popen(cp, "r")) != NULL) { 4991f05bc6cSJohan Karlsson 5001f05bc6cSJohan Karlsson while (fgets(buf, BUFSIZ - 1, p) != NULL) { 5011f05bc6cSJohan Karlsson unusual = unusual & ~NO_MAN_FOUND; 5021f05bc6cSJohan Karlsson 503e97f67f5SJoerg Wunsch if ((cp2 = strchr(buf, '\n')) != NULL) 504e97f67f5SJoerg Wunsch *cp2 = '\0'; 5051f05bc6cSJohan Karlsson if (regexec(&re, buf, 2, 5061f05bc6cSJohan Karlsson matches, 0) == 0 && 5071f05bc6cSJohan Karlsson (rlen = matches[1].rm_eo - 5081f05bc6cSJohan Karlsson matches[1].rm_so) > 0) { 509e97f67f5SJoerg Wunsch /* 510af7ca7c8SPedro F. Giffuni * man -w found formatted 511e97f67f5SJoerg Wunsch * page, need to pick up 512e97f67f5SJoerg Wunsch * source page name. 513e97f67f5SJoerg Wunsch */ 514e97f67f5SJoerg Wunsch cp2 = malloc(rlen + 1); 515e97f67f5SJoerg Wunsch if (cp2 == NULL) 516e97f67f5SJoerg Wunsch abort(); 5171f05bc6cSJohan Karlsson memcpy(cp2, 5181f05bc6cSJohan Karlsson buf + matches[1].rm_so, 519e97f67f5SJoerg Wunsch rlen); 520e97f67f5SJoerg Wunsch cp2[rlen] = '\0'; 521e97f67f5SJoerg Wunsch } else { 522e97f67f5SJoerg Wunsch /* 523e97f67f5SJoerg Wunsch * man -w found plain source 524e97f67f5SJoerg Wunsch * page, use it. 525e97f67f5SJoerg Wunsch */ 526f5ce1402SStefan Eßer cp2 = strdup(buf); 527e97f67f5SJoerg Wunsch if (cp2 == NULL) 528e97f67f5SJoerg Wunsch abort(); 529e97f67f5SJoerg Wunsch } 5301f05bc6cSJohan Karlsson 5311f05bc6cSJohan Karlsson if (man == NULL) { 5321f05bc6cSJohan Karlsson man = strdup(cp2); 5331f05bc6cSJohan Karlsson } else { 5341f05bc6cSJohan Karlsson olen = strlen(man); 5351f05bc6cSJohan Karlsson nlen = strlen(cp2); 5361f05bc6cSJohan Karlsson man = realloc(man, 5371f05bc6cSJohan Karlsson olen + nlen + 2); 538d919e0d5SMarcelo Araujo if (man == NULL) 5391f05bc6cSJohan Karlsson abort(); 5401f05bc6cSJohan Karlsson strcat(man, " "); 5411f05bc6cSJohan Karlsson strcat(man, cp2); 542e97f67f5SJoerg Wunsch } 5431f05bc6cSJohan Karlsson 5441f05bc6cSJohan Karlsson free(cp2); 5451f05bc6cSJohan Karlsson 5461f05bc6cSJohan Karlsson if (!opt_a) 5471f05bc6cSJohan Karlsson break; 5481f05bc6cSJohan Karlsson } 5491f05bc6cSJohan Karlsson pclose(p); 550e97f67f5SJoerg Wunsch free(cp); 551e97f67f5SJoerg Wunsch } 5521f05bc6cSJohan Karlsson } 553e97f67f5SJoerg Wunsch 554e97f67f5SJoerg Wunsch if (opt_s) { 555e97f67f5SJoerg Wunsch /* 556e97f67f5SJoerg Wunsch * Sources match if a subdir with the exact 557e97f67f5SJoerg Wunsch * name is found. 558e97f67f5SJoerg Wunsch */ 5591f05bc6cSJohan Karlsson unusual = unusual | NO_SRC_FOUND; 560e97f67f5SJoerg Wunsch for (dp = sourcedirs; *dp != NULL; dp++) { 561e97f67f5SJoerg Wunsch cp = malloc(strlen(*dp) + 1 + s + 1); 562e97f67f5SJoerg Wunsch if (cp == NULL) 563e97f67f5SJoerg Wunsch abort(); 564e97f67f5SJoerg Wunsch strcpy(cp, *dp); 565e97f67f5SJoerg Wunsch strcat(cp, "/"); 566e97f67f5SJoerg Wunsch strcat(cp, name); 567e97f67f5SJoerg Wunsch if (stat(cp, &sb) == 0 && 568e97f67f5SJoerg Wunsch (sb.st_mode & S_IFMT) == S_IFDIR) { 5691f05bc6cSJohan Karlsson unusual = unusual & ~NO_SRC_FOUND; 5701f05bc6cSJohan Karlsson if (src == NULL) { 5711f05bc6cSJohan Karlsson src = strdup(cp); 5721f05bc6cSJohan Karlsson } else { 5731f05bc6cSJohan Karlsson olen = strlen(src); 5741f05bc6cSJohan Karlsson nlen = strlen(cp); 5751f05bc6cSJohan Karlsson src = realloc(src, 5761f05bc6cSJohan Karlsson olen + nlen + 2); 577d919e0d5SMarcelo Araujo if (src == NULL) 5781f05bc6cSJohan Karlsson abort(); 5791f05bc6cSJohan Karlsson strcat(src, " "); 5801f05bc6cSJohan Karlsson strcat(src, cp); 5811f05bc6cSJohan Karlsson } 5821f05bc6cSJohan Karlsson if (!opt_a) { 5831f05bc6cSJohan Karlsson free(cp); 584e97f67f5SJoerg Wunsch break; 585e97f67f5SJoerg Wunsch } 5861f05bc6cSJohan Karlsson } 587e97f67f5SJoerg Wunsch free(cp); 588e97f67f5SJoerg Wunsch } 589e97f67f5SJoerg Wunsch /* 590e97f67f5SJoerg Wunsch * If still not found, ask locate to search it 591e97f67f5SJoerg Wunsch * for us. This will find sources for things 592e97f67f5SJoerg Wunsch * like lpr that are well hidden in the 593e97f67f5SJoerg Wunsch * /usr/src tree, but takes a lot longer. 594e97f67f5SJoerg Wunsch * Thus, option -x (`expensive') prevents this 595e97f67f5SJoerg Wunsch * search. 596e97f67f5SJoerg Wunsch * 597e97f67f5SJoerg Wunsch * Do only match locate output that starts 598e97f67f5SJoerg Wunsch * with one of our source directories, and at 599e97f67f5SJoerg Wunsch * least one further level of subdirectories. 600e97f67f5SJoerg Wunsch */ 6011f05bc6cSJohan Karlsson if (opt_x || (src && !opt_a)) 602e97f67f5SJoerg Wunsch goto done_sources; 603e97f67f5SJoerg Wunsch 604e97f67f5SJoerg Wunsch cp = malloc(sizeof LOCATECMD - 2 + s); 605e97f67f5SJoerg Wunsch if (cp == NULL) 606e97f67f5SJoerg Wunsch abort(); 607e97f67f5SJoerg Wunsch sprintf(cp, LOCATECMD, name); 608e97f67f5SJoerg Wunsch if ((p = popen(cp, "r")) == NULL) 609e97f67f5SJoerg Wunsch goto done_sources; 6101f05bc6cSJohan Karlsson while ((src == NULL || opt_a) && 611e97f67f5SJoerg Wunsch (fgets(buf, BUFSIZ - 1, p)) != NULL) { 612e97f67f5SJoerg Wunsch if ((cp2 = strchr(buf, '\n')) != NULL) 613e97f67f5SJoerg Wunsch *cp2 = '\0'; 614e97f67f5SJoerg Wunsch for (dp = sourcedirs; 6151f05bc6cSJohan Karlsson (src == NULL || opt_a) && *dp != NULL; 616e97f67f5SJoerg Wunsch dp++) { 617e97f67f5SJoerg Wunsch cp2 = malloc(strlen(*dp) + 9); 618e97f67f5SJoerg Wunsch if (cp2 == NULL) 619e97f67f5SJoerg Wunsch abort(); 620e97f67f5SJoerg Wunsch strcpy(cp2, "^"); 621e97f67f5SJoerg Wunsch strcat(cp2, *dp); 622e97f67f5SJoerg Wunsch strcat(cp2, "/[^/]+/"); 623e97f67f5SJoerg Wunsch if ((i = regcomp(&re2, cp2, 624e97f67f5SJoerg Wunsch REG_EXTENDED|REG_NOSUB)) 625e97f67f5SJoerg Wunsch != 0) { 626e97f67f5SJoerg Wunsch regerror(i, &re, buf, 627e97f67f5SJoerg Wunsch BUFSIZ - 1); 628e97f67f5SJoerg Wunsch errx(EX_UNAVAILABLE, 629e97f67f5SJoerg Wunsch "regcomp(%s) failed: %s", 630e97f67f5SJoerg Wunsch cp2, buf); 631e97f67f5SJoerg Wunsch } 632e97f67f5SJoerg Wunsch free(cp2); 633e97f67f5SJoerg Wunsch if (regexec(&re2, buf, 0, 634e97f67f5SJoerg Wunsch (regmatch_t *)NULL, 0) 635e97f67f5SJoerg Wunsch == 0) { 6361f05bc6cSJohan Karlsson unusual = unusual & 6371f05bc6cSJohan Karlsson ~NO_SRC_FOUND; 6381f05bc6cSJohan Karlsson if (src == NULL) { 6391f05bc6cSJohan Karlsson src = strdup(buf); 6401f05bc6cSJohan Karlsson } else { 6411f05bc6cSJohan Karlsson olen = strlen(src); 6421f05bc6cSJohan Karlsson nlen = strlen(buf); 6431f05bc6cSJohan Karlsson src = realloc(src, 6441f05bc6cSJohan Karlsson olen + 6451f05bc6cSJohan Karlsson nlen + 2); 646d919e0d5SMarcelo Araujo if (src == NULL) 6471f05bc6cSJohan Karlsson abort(); 6481f05bc6cSJohan Karlsson strcat(src, " "); 6491f05bc6cSJohan Karlsson strcat(src, buf); 6501f05bc6cSJohan Karlsson } 651e97f67f5SJoerg Wunsch } 652e97f67f5SJoerg Wunsch regfree(&re2); 653e97f67f5SJoerg Wunsch } 654e97f67f5SJoerg Wunsch } 655e97f67f5SJoerg Wunsch pclose(p); 656e97f67f5SJoerg Wunsch free(cp); 657e97f67f5SJoerg Wunsch } 658e97f67f5SJoerg Wunsch done_sources: 659e97f67f5SJoerg Wunsch 660e97f67f5SJoerg Wunsch if (opt_u && !unusual) 661e97f67f5SJoerg Wunsch continue; 662e97f67f5SJoerg Wunsch 663e97f67f5SJoerg Wunsch printed = 0; 664e97f67f5SJoerg Wunsch if (!opt_q) { 665e97f67f5SJoerg Wunsch printf("%s:", name); 666e97f67f5SJoerg Wunsch printed++; 667e97f67f5SJoerg Wunsch } 668e97f67f5SJoerg Wunsch if (bin) { 669e97f67f5SJoerg Wunsch if (printed++) 670e97f67f5SJoerg Wunsch putchar(' '); 671e97f67f5SJoerg Wunsch fputs(bin, stdout); 672e97f67f5SJoerg Wunsch } 673e97f67f5SJoerg Wunsch if (man) { 674e97f67f5SJoerg Wunsch if (printed++) 675e97f67f5SJoerg Wunsch putchar(' '); 676e97f67f5SJoerg Wunsch fputs(man, stdout); 677e97f67f5SJoerg Wunsch } 678e97f67f5SJoerg Wunsch if (src) { 679e97f67f5SJoerg Wunsch if (printed++) 680e97f67f5SJoerg Wunsch putchar(' '); 681e97f67f5SJoerg Wunsch fputs(src, stdout); 682e97f67f5SJoerg Wunsch } 683e97f67f5SJoerg Wunsch if (printed) 684e97f67f5SJoerg Wunsch putchar('\n'); 685e97f67f5SJoerg Wunsch } 686e97f67f5SJoerg Wunsch 687e97f67f5SJoerg Wunsch if (opt_m) 688e97f67f5SJoerg Wunsch regfree(&re); 689e97f67f5SJoerg Wunsch 690e97f67f5SJoerg Wunsch return (0); 691b78dcb87SJohan Karlsson } 692