1e97f67f5SJoerg Wunsch /* 2e97f67f5SJoerg Wunsch * Copyright � 2002, J�rg Wunsch 3b78dcb87SJohan Karlsson * 4b78dcb87SJohan Karlsson * Redistribution and use in source and binary forms, with or without 5b78dcb87SJohan Karlsson * modification, are permitted provided that the following conditions 6b78dcb87SJohan Karlsson * are met: 7b78dcb87SJohan Karlsson * 1. Redistributions of source code must retain the above copyright 8b78dcb87SJohan Karlsson * notice, this list of conditions and the following disclaimer. 9b78dcb87SJohan Karlsson * 2. Redistributions in binary form must reproduce the above copyright 10b78dcb87SJohan Karlsson * notice, this list of conditions and the following disclaimer in the 11b78dcb87SJohan Karlsson * documentation and/or other materials provided with the distribution. 12b78dcb87SJohan Karlsson * 13e97f67f5SJoerg Wunsch * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 14e97f67f5SJoerg Wunsch * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 15e97f67f5SJoerg Wunsch * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 16e97f67f5SJoerg Wunsch * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, 17e97f67f5SJoerg Wunsch * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 18e97f67f5SJoerg Wunsch * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 19e97f67f5SJoerg Wunsch * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 20e97f67f5SJoerg Wunsch * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, 21e97f67f5SJoerg Wunsch * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING 22e97f67f5SJoerg Wunsch * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 23e97f67f5SJoerg Wunsch * POSSIBILITY OF SUCH DAMAGE. 24b78dcb87SJohan Karlsson */ 25b78dcb87SJohan Karlsson 26e97f67f5SJoerg Wunsch /* 27e97f67f5SJoerg Wunsch * 4.3BSD UI-compatible whereis(1) utility. Rewritten from scratch 28e97f67f5SJoerg Wunsch * since the original 4.3BSD version suffers legal problems that 29e97f67f5SJoerg Wunsch * prevent it from being redistributed, and since the 4.4BSD version 30e97f67f5SJoerg Wunsch * was pretty inferior in functionality. 31e97f67f5SJoerg Wunsch */ 329d1b456eSJohan Karlsson 33e97f67f5SJoerg Wunsch #include <sys/types.h> 34b78dcb87SJohan Karlsson 35b68725e3SJohan Karlsson __FBSDID("$FreeBSD$"); 36b68725e3SJohan Karlsson 37b78dcb87SJohan Karlsson #include <sys/stat.h> 38b78dcb87SJohan Karlsson #include <sys/sysctl.h> 39b78dcb87SJohan Karlsson 40e97f67f5SJoerg Wunsch #include <dirent.h> 41b78dcb87SJohan Karlsson #include <err.h> 42b78dcb87SJohan Karlsson #include <errno.h> 43e97f67f5SJoerg Wunsch #include <regex.h> 44b78dcb87SJohan Karlsson #include <stdio.h> 45b78dcb87SJohan Karlsson #include <stdlib.h> 46b78dcb87SJohan Karlsson #include <string.h> 47e97f67f5SJoerg Wunsch #include <sysexits.h> 48b78dcb87SJohan Karlsson #include <unistd.h> 49b78dcb87SJohan Karlsson 50e97f67f5SJoerg Wunsch #include "pathnames.h" 51b78dcb87SJohan Karlsson 52e97f67f5SJoerg Wunsch typedef const char *ccharp; 53e97f67f5SJoerg Wunsch 54e97f67f5SJoerg Wunsch int opt_b, opt_m, opt_q, opt_s, opt_u, opt_x; 55e97f67f5SJoerg Wunsch ccharp *bindirs, *mandirs, *sourcedirs; 56e97f67f5SJoerg Wunsch char **query; 57e97f67f5SJoerg Wunsch 58e97f67f5SJoerg Wunsch const char *sourcepath = PATH_SOURCES; 59e97f67f5SJoerg Wunsch 60e97f67f5SJoerg Wunsch char *colonify(ccharp *); 61e97f67f5SJoerg Wunsch int contains(ccharp *, const char *); 62e97f67f5SJoerg Wunsch void decolonify(char *, ccharp **, int *); 63e97f67f5SJoerg Wunsch void defaults(void); 64e97f67f5SJoerg Wunsch void scanopts(int, char **); 65e97f67f5SJoerg Wunsch void usage(void); 66e97f67f5SJoerg Wunsch 67e97f67f5SJoerg Wunsch /* 68e97f67f5SJoerg Wunsch * Throughout this program, a number of strings are dynamically 69e97f67f5SJoerg Wunsch * allocated but never freed. Their memory is written to when 70e97f67f5SJoerg Wunsch * splitting the strings into string lists which will later be 71e97f67f5SJoerg Wunsch * processed. Since it's important that those string lists remain 72e97f67f5SJoerg Wunsch * valid even after the functions allocating the memory returned, 73e97f67f5SJoerg Wunsch * those functions cannot free them. They could be freed only at end 74e97f67f5SJoerg Wunsch * of main(), which is pretty pointless anyway. 75e97f67f5SJoerg Wunsch * 76e97f67f5SJoerg Wunsch * The overall amount of memory to be allocated for processing the 77e97f67f5SJoerg Wunsch * strings is not expected to exceed a few kilobytes. For that 78e97f67f5SJoerg Wunsch * reason, allocation can usually always be assumed to succeed (within 79e97f67f5SJoerg Wunsch * a virtual memory environment), thus we simply bail out using 80e97f67f5SJoerg Wunsch * abort(3) in case of an allocation failure. 81e97f67f5SJoerg Wunsch */ 82e97f67f5SJoerg Wunsch 83e97f67f5SJoerg Wunsch void 84e97f67f5SJoerg Wunsch usage(void) 85b78dcb87SJohan Karlsson { 86e97f67f5SJoerg Wunsch errx(EX_USAGE, 87e97f67f5SJoerg Wunsch "usage: whereis [-bmqsux] [-BMS dir... -f] name ..."); 88e97f67f5SJoerg Wunsch } 89b78dcb87SJohan Karlsson 90e97f67f5SJoerg Wunsch /* 91e97f67f5SJoerg Wunsch * Scan options passed to program. 92e97f67f5SJoerg Wunsch * 93e97f67f5SJoerg Wunsch * Note that the -B/-M/-S options expect a list of directory 94e97f67f5SJoerg Wunsch * names that must be terminated with -f. 95e97f67f5SJoerg Wunsch */ 96e97f67f5SJoerg Wunsch void 97e97f67f5SJoerg Wunsch scanopts(int argc, char **argv) 98e97f67f5SJoerg Wunsch { 99e97f67f5SJoerg Wunsch int c, i, opt_f; 100e97f67f5SJoerg Wunsch ccharp **dirlist; 101e97f67f5SJoerg Wunsch 102e97f67f5SJoerg Wunsch opt_f = 0; 103e97f67f5SJoerg Wunsch while ((c = getopt(argc, argv, "BMSbfmqsux")) != -1) 104e97f67f5SJoerg Wunsch switch (c) { 105e97f67f5SJoerg Wunsch case 'B': 106e97f67f5SJoerg Wunsch dirlist = &bindirs; 107e97f67f5SJoerg Wunsch goto dolist; 108e97f67f5SJoerg Wunsch 109e97f67f5SJoerg Wunsch case 'M': 110e97f67f5SJoerg Wunsch dirlist = &mandirs; 111e97f67f5SJoerg Wunsch goto dolist; 112e97f67f5SJoerg Wunsch 113e97f67f5SJoerg Wunsch case 'S': 114e97f67f5SJoerg Wunsch dirlist = &sourcedirs; 115e97f67f5SJoerg Wunsch dolist: 116e97f67f5SJoerg Wunsch i = 0; 117e97f67f5SJoerg Wunsch while (optind < argc && 118e97f67f5SJoerg Wunsch strcmp(argv[optind], "-f") != 0 && 119e97f67f5SJoerg Wunsch strcmp(argv[optind], "-B") != 0 && 120e97f67f5SJoerg Wunsch strcmp(argv[optind], "-M") != 0 && 121e97f67f5SJoerg Wunsch strcmp(argv[optind], "-S") != 0) { 122e97f67f5SJoerg Wunsch *dirlist = realloc(*dirlist, 123e97f67f5SJoerg Wunsch (i + 2) * sizeof(char *)); 124e97f67f5SJoerg Wunsch if (*dirlist == NULL) 125e97f67f5SJoerg Wunsch abort(); 126e97f67f5SJoerg Wunsch (*dirlist)[i] = argv[optind]; 127e97f67f5SJoerg Wunsch i++; 128e97f67f5SJoerg Wunsch optind++; 129e97f67f5SJoerg Wunsch } 130e97f67f5SJoerg Wunsch (*dirlist)[i] = NULL; 131e97f67f5SJoerg Wunsch break; 132e97f67f5SJoerg Wunsch 133e97f67f5SJoerg Wunsch case 'b': 134e97f67f5SJoerg Wunsch opt_b = 1; 135e97f67f5SJoerg Wunsch break; 136e97f67f5SJoerg Wunsch 137e97f67f5SJoerg Wunsch case 'f': 138e97f67f5SJoerg Wunsch goto breakout; 139e97f67f5SJoerg Wunsch 140e97f67f5SJoerg Wunsch case 'm': 141e97f67f5SJoerg Wunsch opt_m = 1; 142e97f67f5SJoerg Wunsch break; 143e97f67f5SJoerg Wunsch 144e97f67f5SJoerg Wunsch case 'q': 145e97f67f5SJoerg Wunsch opt_q = 1; 146e97f67f5SJoerg Wunsch break; 147e97f67f5SJoerg Wunsch 148e97f67f5SJoerg Wunsch case 's': 149e97f67f5SJoerg Wunsch opt_s = 1; 150e97f67f5SJoerg Wunsch break; 151e97f67f5SJoerg Wunsch 152e97f67f5SJoerg Wunsch case 'u': 153e97f67f5SJoerg Wunsch opt_u = 1; 154e97f67f5SJoerg Wunsch break; 155e97f67f5SJoerg Wunsch 156e97f67f5SJoerg Wunsch case 'x': 157e97f67f5SJoerg Wunsch opt_x = 1; 158e97f67f5SJoerg Wunsch break; 159e97f67f5SJoerg Wunsch 160b78dcb87SJohan Karlsson default: 161b78dcb87SJohan Karlsson usage(); 162b78dcb87SJohan Karlsson } 163e97f67f5SJoerg Wunsch breakout: 164e97f67f5SJoerg Wunsch if (optind == argc) 165b78dcb87SJohan Karlsson usage(); 166e97f67f5SJoerg Wunsch query = argv + optind; 167b78dcb87SJohan Karlsson } 168b78dcb87SJohan Karlsson 169e97f67f5SJoerg Wunsch /* 170e97f67f5SJoerg Wunsch * Find out whether string `s' is contained in list `cpp'. 171e97f67f5SJoerg Wunsch */ 172e97f67f5SJoerg Wunsch int 173e97f67f5SJoerg Wunsch contains(ccharp *cpp, const char *s) 174e97f67f5SJoerg Wunsch { 175e97f67f5SJoerg Wunsch ccharp cp; 176b78dcb87SJohan Karlsson 177e97f67f5SJoerg Wunsch if (cpp == NULL) 178e97f67f5SJoerg Wunsch return (0); 179e97f67f5SJoerg Wunsch 180e97f67f5SJoerg Wunsch while ((cp = *cpp) != NULL) { 181e97f67f5SJoerg Wunsch if (strcmp(cp, s) == 0) 182e97f67f5SJoerg Wunsch return (1); 183e97f67f5SJoerg Wunsch cpp++; 184e97f67f5SJoerg Wunsch } 185b78dcb87SJohan Karlsson return (0); 186b78dcb87SJohan Karlsson } 187b78dcb87SJohan Karlsson 188e97f67f5SJoerg Wunsch /* 189e97f67f5SJoerg Wunsch * Split string `s' at colons, and pass it to the string list pointed 190e97f67f5SJoerg Wunsch * to by `cppp' (which has `*ip' elements). Note that the original 191e97f67f5SJoerg Wunsch * string is modified by replacing the colon with a NUL byte. The 192e97f67f5SJoerg Wunsch * partial string is only added if it has a length greater than 0, and 193e97f67f5SJoerg Wunsch * if it's not already contained in the string list. 194e97f67f5SJoerg Wunsch */ 195b78dcb87SJohan Karlsson void 196e97f67f5SJoerg Wunsch decolonify(char *s, ccharp **cppp, int *ip) 197b78dcb87SJohan Karlsson { 198e97f67f5SJoerg Wunsch char *cp; 199b78dcb87SJohan Karlsson 200e97f67f5SJoerg Wunsch while ((cp = strchr(s, ':')), *s != '\0') { 201e97f67f5SJoerg Wunsch if (cp) 202e97f67f5SJoerg Wunsch *cp = '\0'; 203e97f67f5SJoerg Wunsch if (strlen(s) && !contains(*cppp, s)) { 204e97f67f5SJoerg Wunsch *cppp = realloc(*cppp, (*ip + 2) * sizeof(char *)); 205e97f67f5SJoerg Wunsch if (cppp == NULL) 206e97f67f5SJoerg Wunsch abort(); 207e97f67f5SJoerg Wunsch (*cppp)[*ip] = s; 208e97f67f5SJoerg Wunsch (*cppp)[*ip + 1] = NULL; 209e97f67f5SJoerg Wunsch (*ip)++; 210e97f67f5SJoerg Wunsch } 211e97f67f5SJoerg Wunsch if (cp) 212e97f67f5SJoerg Wunsch s = cp + 1; 213e97f67f5SJoerg Wunsch else 214e97f67f5SJoerg Wunsch break; 215e97f67f5SJoerg Wunsch } 216e97f67f5SJoerg Wunsch } 217e97f67f5SJoerg Wunsch 218e97f67f5SJoerg Wunsch /* 219e97f67f5SJoerg Wunsch * Join string list `cpp' into a colon-separated string. 220e97f67f5SJoerg Wunsch */ 221e97f67f5SJoerg Wunsch char * 222e97f67f5SJoerg Wunsch colonify(ccharp *cpp) 223e97f67f5SJoerg Wunsch { 224e97f67f5SJoerg Wunsch size_t s; 225e97f67f5SJoerg Wunsch char *cp; 226e97f67f5SJoerg Wunsch int i; 227e97f67f5SJoerg Wunsch 228e97f67f5SJoerg Wunsch if (cpp == NULL) 229e97f67f5SJoerg Wunsch return (0); 230e97f67f5SJoerg Wunsch 231e97f67f5SJoerg Wunsch for (s = 0, i = 0; cpp[i] != NULL; i++) 232e97f67f5SJoerg Wunsch s += strlen(cpp[i]) + 1; 233e97f67f5SJoerg Wunsch if ((cp = malloc(s + 1)) == NULL) 234e97f67f5SJoerg Wunsch abort(); 235e97f67f5SJoerg Wunsch for (i = 0, *cp = '\0'; cpp[i] != NULL; i++) { 236e97f67f5SJoerg Wunsch strcat(cp, cpp[i]); 237e97f67f5SJoerg Wunsch strcat(cp, ":"); 238e97f67f5SJoerg Wunsch } 239e97f67f5SJoerg Wunsch cp[s - 1] = '\0'; /* eliminate last colon */ 240e97f67f5SJoerg Wunsch 241e97f67f5SJoerg Wunsch return (cp); 242e97f67f5SJoerg Wunsch } 243e97f67f5SJoerg Wunsch 244e97f67f5SJoerg Wunsch /* 245e97f67f5SJoerg Wunsch * Provide defaults for all options and directory lists. 246e97f67f5SJoerg Wunsch */ 247e97f67f5SJoerg Wunsch void 248e97f67f5SJoerg Wunsch defaults(void) 249e97f67f5SJoerg Wunsch { 250e97f67f5SJoerg Wunsch size_t s; 251e97f67f5SJoerg Wunsch char *b, buf[BUFSIZ], *cp; 252e97f67f5SJoerg Wunsch int nele; 253e97f67f5SJoerg Wunsch FILE *p; 254e97f67f5SJoerg Wunsch DIR *dir; 255e97f67f5SJoerg Wunsch struct stat sb; 256e97f67f5SJoerg Wunsch struct dirent *dirp; 257e97f67f5SJoerg Wunsch 258e97f67f5SJoerg Wunsch /* default to -bms if none has been specified */ 259e97f67f5SJoerg Wunsch if (!opt_b && !opt_m && !opt_s) 260e97f67f5SJoerg Wunsch opt_b = opt_m = opt_s = 1; 261e97f67f5SJoerg Wunsch 2628e4c33e9SJohan Karlsson /* -b defaults to default path + /usr/libexec + 2638e4c33e9SJohan Karlsson * /usr/games + user's path */ 264e97f67f5SJoerg Wunsch if (!bindirs) { 265e97f67f5SJoerg Wunsch if (sysctlbyname("user.cs_path", (void *)NULL, &s, 266e97f67f5SJoerg Wunsch (void *)NULL, 0) == -1) 267e97f67f5SJoerg Wunsch err(EX_OSERR, "sysctlbyname(\"user.cs_path\")"); 268e97f67f5SJoerg Wunsch if ((b = malloc(s + 1)) == NULL) 269e97f67f5SJoerg Wunsch abort(); 270e97f67f5SJoerg Wunsch if (sysctlbyname("user.cs_path", b, &s, (void *)NULL, 0) == -1) 271e97f67f5SJoerg Wunsch err(EX_OSERR, "sysctlbyname(\"user.cs_path\")"); 272e97f67f5SJoerg Wunsch nele = 0; 273e97f67f5SJoerg Wunsch decolonify(b, &bindirs, &nele); 2748e4c33e9SJohan Karlsson bindirs = realloc(bindirs, (nele + 3) * sizeof(char *)); 275e97f67f5SJoerg Wunsch if (bindirs == NULL) 276e97f67f5SJoerg Wunsch abort(); 277e97f67f5SJoerg Wunsch bindirs[nele++] = "/usr/libexec"; 2788e4c33e9SJohan Karlsson bindirs[nele++] = "/usr/games"; 279e97f67f5SJoerg Wunsch bindirs[nele] = NULL; 280e97f67f5SJoerg Wunsch if ((cp = getenv("PATH")) != NULL) { 281e97f67f5SJoerg Wunsch /* don't destroy the original environment... */ 282e97f67f5SJoerg Wunsch if ((b = malloc(strlen(cp) + 1)) == NULL) 283e97f67f5SJoerg Wunsch abort(); 284e97f67f5SJoerg Wunsch strcpy(b, cp); 285e97f67f5SJoerg Wunsch decolonify(b, &bindirs, &nele); 286e97f67f5SJoerg Wunsch } 287e97f67f5SJoerg Wunsch } 288e97f67f5SJoerg Wunsch 289e97f67f5SJoerg Wunsch /* -m defaults to $(manpath) */ 290e97f67f5SJoerg Wunsch if (!mandirs) { 291e97f67f5SJoerg Wunsch if ((p = popen(MANPATHCMD, "r")) == NULL) 292e97f67f5SJoerg Wunsch err(EX_OSERR, "cannot execute manpath command"); 293e97f67f5SJoerg Wunsch if (fgets(buf, BUFSIZ - 1, p) == NULL || 294e97f67f5SJoerg Wunsch pclose(p)) 295e97f67f5SJoerg Wunsch err(EX_OSERR, "error processing manpath results"); 296e97f67f5SJoerg Wunsch if ((b = strchr(buf, '\n')) != NULL) 297e97f67f5SJoerg Wunsch *b = '\0'; 298e97f67f5SJoerg Wunsch if ((b = malloc(strlen(buf) + 1)) == NULL) 299e97f67f5SJoerg Wunsch abort(); 300e97f67f5SJoerg Wunsch strcpy(b, buf); 301e97f67f5SJoerg Wunsch nele = 0; 302e97f67f5SJoerg Wunsch decolonify(b, &mandirs, &nele); 303e97f67f5SJoerg Wunsch } 304e97f67f5SJoerg Wunsch 305e97f67f5SJoerg Wunsch /* -s defaults to precompiled list, plus subdirs of /usr/ports */ 306e97f67f5SJoerg Wunsch if (!sourcedirs) { 307e97f67f5SJoerg Wunsch if ((b = malloc(strlen(sourcepath) + 1)) == NULL) 308e97f67f5SJoerg Wunsch abort(); 309e97f67f5SJoerg Wunsch strcpy(b, sourcepath); 310e97f67f5SJoerg Wunsch nele = 0; 311e97f67f5SJoerg Wunsch decolonify(b, &sourcedirs, &nele); 312e97f67f5SJoerg Wunsch 313e97f67f5SJoerg Wunsch if (stat(PATH_PORTS, &sb) == -1) { 314e97f67f5SJoerg Wunsch if (errno == ENOENT) 315e97f67f5SJoerg Wunsch /* no /usr/ports, we are done */ 316e97f67f5SJoerg Wunsch return; 317e97f67f5SJoerg Wunsch err(EX_OSERR, "stat(" PATH_PORTS ")"); 318e97f67f5SJoerg Wunsch } 319e97f67f5SJoerg Wunsch if ((sb.st_mode & S_IFMT) != S_IFDIR) 320e97f67f5SJoerg Wunsch /* /usr/ports is not a directory, ignore */ 321e97f67f5SJoerg Wunsch return; 322e97f67f5SJoerg Wunsch if (access(PATH_PORTS, R_OK | X_OK) != 0) 323e97f67f5SJoerg Wunsch return; 324e97f67f5SJoerg Wunsch if ((dir = opendir(PATH_PORTS)) == NULL) 325e97f67f5SJoerg Wunsch err(EX_OSERR, "opendir" PATH_PORTS ")"); 326e97f67f5SJoerg Wunsch while ((dirp = readdir(dir)) != NULL) { 327e97f67f5SJoerg Wunsch if (dirp->d_name[0] == '.' || 328e97f67f5SJoerg Wunsch strcmp(dirp->d_name, "CVS") == 0) 329e97f67f5SJoerg Wunsch /* ignore dot entries and CVS subdir */ 330e97f67f5SJoerg Wunsch continue; 331e97f67f5SJoerg Wunsch if ((b = malloc(sizeof PATH_PORTS + 1 + dirp->d_namlen)) 332e97f67f5SJoerg Wunsch == NULL) 333e97f67f5SJoerg Wunsch abort(); 334e97f67f5SJoerg Wunsch strcpy(b, PATH_PORTS); 335e97f67f5SJoerg Wunsch strcat(b, "/"); 336e97f67f5SJoerg Wunsch strcat(b, dirp->d_name); 337e97f67f5SJoerg Wunsch if (stat(b, &sb) == -1 || 338e97f67f5SJoerg Wunsch (sb.st_mode & S_IFMT) != S_IFDIR || 339e97f67f5SJoerg Wunsch access(b, R_OK | X_OK) != 0) { 340e97f67f5SJoerg Wunsch free(b); 341e97f67f5SJoerg Wunsch continue; 342e97f67f5SJoerg Wunsch } 343e97f67f5SJoerg Wunsch sourcedirs = realloc(sourcedirs, 344e97f67f5SJoerg Wunsch (nele + 2) * sizeof(char *)); 345e97f67f5SJoerg Wunsch if (sourcedirs == NULL) 346e97f67f5SJoerg Wunsch abort(); 347e97f67f5SJoerg Wunsch sourcedirs[nele++] = b; 348e97f67f5SJoerg Wunsch sourcedirs[nele] = NULL; 349e97f67f5SJoerg Wunsch } 350e97f67f5SJoerg Wunsch closedir(dir); 351e97f67f5SJoerg Wunsch } 352e97f67f5SJoerg Wunsch } 353e97f67f5SJoerg Wunsch 354e97f67f5SJoerg Wunsch int 355e97f67f5SJoerg Wunsch main(int argc, char **argv) 356e97f67f5SJoerg Wunsch { 357e97f67f5SJoerg Wunsch int unusual, i, printed; 358e97f67f5SJoerg Wunsch char *bin, buf[BUFSIZ], *cp, *cp2, *man, *name, *src; 359e97f67f5SJoerg Wunsch ccharp *dp; 360e97f67f5SJoerg Wunsch size_t s; 361e97f67f5SJoerg Wunsch struct stat sb; 362e97f67f5SJoerg Wunsch regex_t re, re2; 363e97f67f5SJoerg Wunsch regmatch_t matches[2]; 364e97f67f5SJoerg Wunsch regoff_t rlen; 365e97f67f5SJoerg Wunsch FILE *p; 366e97f67f5SJoerg Wunsch 367e97f67f5SJoerg Wunsch scanopts(argc, argv); 368e97f67f5SJoerg Wunsch defaults(); 369e97f67f5SJoerg Wunsch 370e97f67f5SJoerg Wunsch if (mandirs == NULL) 371e97f67f5SJoerg Wunsch opt_m = 0; 372e97f67f5SJoerg Wunsch if (bindirs == NULL) 373e97f67f5SJoerg Wunsch opt_b = 0; 374e97f67f5SJoerg Wunsch if (sourcedirs == NULL) 375e97f67f5SJoerg Wunsch opt_s = 0; 376e97f67f5SJoerg Wunsch if (opt_m + opt_b + opt_s == 0) 377e97f67f5SJoerg Wunsch errx(EX_DATAERR, "no directories to search"); 378e97f67f5SJoerg Wunsch 379e97f67f5SJoerg Wunsch if (opt_m) { 380e97f67f5SJoerg Wunsch setenv("MANPATH", colonify(mandirs), 1); 381e97f67f5SJoerg Wunsch if ((i = regcomp(&re, MANWHEREISMATCH, REG_EXTENDED)) != 0) { 382e97f67f5SJoerg Wunsch regerror(i, &re, buf, BUFSIZ - 1); 383e97f67f5SJoerg Wunsch errx(EX_UNAVAILABLE, "regcomp(%s) failed: %s", 384e97f67f5SJoerg Wunsch MANWHEREISMATCH, buf); 385e97f67f5SJoerg Wunsch } 386e97f67f5SJoerg Wunsch } 387e97f67f5SJoerg Wunsch 388e97f67f5SJoerg Wunsch for (; (name = *query) != NULL; query++) { 389e97f67f5SJoerg Wunsch /* strip leading path name component */ 390e97f67f5SJoerg Wunsch if ((cp = strrchr(name, '/')) != NULL) 391e97f67f5SJoerg Wunsch name = cp + 1; 392e97f67f5SJoerg Wunsch /* strip SCCS or RCS suffix/prefix */ 393e97f67f5SJoerg Wunsch if (strlen(name) > 2 && strncmp(name, "s.", 2) == 0) 394e97f67f5SJoerg Wunsch name += 2; 395e97f67f5SJoerg Wunsch if ((s = strlen(name)) > 2 && strcmp(name + s - 2, ",v") == 0) 396e97f67f5SJoerg Wunsch name[s - 2] = '\0'; 397e97f67f5SJoerg Wunsch /* compression suffix */ 398e97f67f5SJoerg Wunsch s = strlen(name); 399e97f67f5SJoerg Wunsch if (s > 2 && 400e97f67f5SJoerg Wunsch (strcmp(name + s - 2, ".z") == 0 || 401e97f67f5SJoerg Wunsch strcmp(name + s - 2, ".Z") == 0)) 402e97f67f5SJoerg Wunsch name[s - 2] = '\0'; 403e97f67f5SJoerg Wunsch else if (s > 3 && 404e97f67f5SJoerg Wunsch strcmp(name + s - 3, ".gz") == 0) 405e97f67f5SJoerg Wunsch name[s - 3] = '\0'; 406e97f67f5SJoerg Wunsch else if (s > 4 && 407e97f67f5SJoerg Wunsch strcmp(name + s - 4, ".bz2") == 0) 408e97f67f5SJoerg Wunsch name[s - 4] = '\0'; 409e97f67f5SJoerg Wunsch 410e97f67f5SJoerg Wunsch unusual = 0; 411e97f67f5SJoerg Wunsch bin = man = src = NULL; 412e97f67f5SJoerg Wunsch s = strlen(name); 413e97f67f5SJoerg Wunsch 414e97f67f5SJoerg Wunsch if (opt_b) { 415e97f67f5SJoerg Wunsch /* 416e97f67f5SJoerg Wunsch * Binaries have to match exactly, and must be regular 417e97f67f5SJoerg Wunsch * executable files. 418e97f67f5SJoerg Wunsch */ 419e97f67f5SJoerg Wunsch unusual++; 420e97f67f5SJoerg Wunsch for (dp = bindirs; *dp != NULL; dp++) { 421e97f67f5SJoerg Wunsch cp = malloc(strlen(*dp) + 1 + s + 1); 422e97f67f5SJoerg Wunsch if (cp == NULL) 423e97f67f5SJoerg Wunsch abort(); 424e97f67f5SJoerg Wunsch strcpy(cp, *dp); 425e97f67f5SJoerg Wunsch strcat(cp, "/"); 426e97f67f5SJoerg Wunsch strcat(cp, name); 427e97f67f5SJoerg Wunsch if (stat(cp, &sb) == 0 && 428e97f67f5SJoerg Wunsch (sb.st_mode & S_IFMT) == S_IFREG && 429e97f67f5SJoerg Wunsch (sb.st_mode & (S_IXUSR | S_IXGRP | S_IXOTH)) 430e97f67f5SJoerg Wunsch != 0) { 431e97f67f5SJoerg Wunsch unusual--; 432e97f67f5SJoerg Wunsch bin = cp; 433e97f67f5SJoerg Wunsch break; 434e97f67f5SJoerg Wunsch } 435e97f67f5SJoerg Wunsch free(cp); 436e97f67f5SJoerg Wunsch } 437e97f67f5SJoerg Wunsch } 438e97f67f5SJoerg Wunsch 439e97f67f5SJoerg Wunsch if (opt_m) { 440e97f67f5SJoerg Wunsch /* 441e97f67f5SJoerg Wunsch * Ask the man command to perform the search for us. 442e97f67f5SJoerg Wunsch */ 443e97f67f5SJoerg Wunsch unusual++; 444e97f67f5SJoerg Wunsch cp = malloc(sizeof MANWHEREISCMD - 2 + s); 445e97f67f5SJoerg Wunsch if (cp == NULL) 446e97f67f5SJoerg Wunsch abort(); 447e97f67f5SJoerg Wunsch sprintf(cp, MANWHEREISCMD, name); 448e97f67f5SJoerg Wunsch if ((p = popen(cp, "r")) != NULL && 449e97f67f5SJoerg Wunsch fgets(buf, BUFSIZ - 1, p) != NULL && 450e97f67f5SJoerg Wunsch pclose(p) == 0) { 451e97f67f5SJoerg Wunsch unusual--; 452e97f67f5SJoerg Wunsch if ((cp2 = strchr(buf, '\n')) != NULL) 453e97f67f5SJoerg Wunsch *cp2 = '\0'; 454e97f67f5SJoerg Wunsch if (regexec(&re, buf, 2, matches, 0) == 0 && 455e97f67f5SJoerg Wunsch (rlen = matches[1].rm_eo - matches[1].rm_so) 456e97f67f5SJoerg Wunsch > 0) { 457e97f67f5SJoerg Wunsch /* 458e97f67f5SJoerg Wunsch * man -w found compressed 459e97f67f5SJoerg Wunsch * page, need to pick up 460e97f67f5SJoerg Wunsch * source page name. 461e97f67f5SJoerg Wunsch */ 462e97f67f5SJoerg Wunsch cp2 = malloc(rlen + 1); 463e97f67f5SJoerg Wunsch if (cp2 == NULL) 464e97f67f5SJoerg Wunsch abort(); 465e97f67f5SJoerg Wunsch memcpy(cp2, buf + matches[1].rm_so, 466e97f67f5SJoerg Wunsch rlen); 467e97f67f5SJoerg Wunsch cp2[rlen] = '\0'; 468e97f67f5SJoerg Wunsch man = cp2; 469e97f67f5SJoerg Wunsch } else { 470e97f67f5SJoerg Wunsch /* 471e97f67f5SJoerg Wunsch * man -w found plain source 472e97f67f5SJoerg Wunsch * page, use it. 473e97f67f5SJoerg Wunsch */ 474e97f67f5SJoerg Wunsch s = strlen(buf); 475e97f67f5SJoerg Wunsch cp2 = malloc(s + 1); 476e97f67f5SJoerg Wunsch if (cp2 == NULL) 477e97f67f5SJoerg Wunsch abort(); 478e97f67f5SJoerg Wunsch strcpy(cp2, buf); 479e97f67f5SJoerg Wunsch man = cp2; 480e97f67f5SJoerg Wunsch } 481e97f67f5SJoerg Wunsch } 482e97f67f5SJoerg Wunsch free(cp); 483e97f67f5SJoerg Wunsch } 484e97f67f5SJoerg Wunsch 485e97f67f5SJoerg Wunsch if (opt_s) { 486e97f67f5SJoerg Wunsch /* 487e97f67f5SJoerg Wunsch * Sources match if a subdir with the exact 488e97f67f5SJoerg Wunsch * name is found. 489e97f67f5SJoerg Wunsch */ 490e97f67f5SJoerg Wunsch unusual++; 491e97f67f5SJoerg Wunsch for (dp = sourcedirs; *dp != NULL; dp++) { 492e97f67f5SJoerg Wunsch cp = malloc(strlen(*dp) + 1 + s + 1); 493e97f67f5SJoerg Wunsch if (cp == NULL) 494e97f67f5SJoerg Wunsch abort(); 495e97f67f5SJoerg Wunsch strcpy(cp, *dp); 496e97f67f5SJoerg Wunsch strcat(cp, "/"); 497e97f67f5SJoerg Wunsch strcat(cp, name); 498e97f67f5SJoerg Wunsch if (stat(cp, &sb) == 0 && 499e97f67f5SJoerg Wunsch (sb.st_mode & S_IFMT) == S_IFDIR) { 500e97f67f5SJoerg Wunsch unusual--; 501e97f67f5SJoerg Wunsch src = cp; 502e97f67f5SJoerg Wunsch break; 503e97f67f5SJoerg Wunsch } 504e97f67f5SJoerg Wunsch free(cp); 505e97f67f5SJoerg Wunsch } 506e97f67f5SJoerg Wunsch /* 507e97f67f5SJoerg Wunsch * If still not found, ask locate to search it 508e97f67f5SJoerg Wunsch * for us. This will find sources for things 509e97f67f5SJoerg Wunsch * like lpr that are well hidden in the 510e97f67f5SJoerg Wunsch * /usr/src tree, but takes a lot longer. 511e97f67f5SJoerg Wunsch * Thus, option -x (`expensive') prevents this 512e97f67f5SJoerg Wunsch * search. 513e97f67f5SJoerg Wunsch * 514e97f67f5SJoerg Wunsch * Do only match locate output that starts 515e97f67f5SJoerg Wunsch * with one of our source directories, and at 516e97f67f5SJoerg Wunsch * least one further level of subdirectories. 517e97f67f5SJoerg Wunsch */ 518e97f67f5SJoerg Wunsch if (opt_x || src) 519e97f67f5SJoerg Wunsch goto done_sources; 520e97f67f5SJoerg Wunsch 521e97f67f5SJoerg Wunsch cp = malloc(sizeof LOCATECMD - 2 + s); 522e97f67f5SJoerg Wunsch if (cp == NULL) 523e97f67f5SJoerg Wunsch abort(); 524e97f67f5SJoerg Wunsch sprintf(cp, LOCATECMD, name); 525e97f67f5SJoerg Wunsch if ((p = popen(cp, "r")) == NULL) 526e97f67f5SJoerg Wunsch goto done_sources; 527e97f67f5SJoerg Wunsch while (src == NULL && 528e97f67f5SJoerg Wunsch (fgets(buf, BUFSIZ - 1, p)) != NULL) { 529e97f67f5SJoerg Wunsch if ((cp2 = strchr(buf, '\n')) != NULL) 530e97f67f5SJoerg Wunsch *cp2 = '\0'; 531e97f67f5SJoerg Wunsch for (dp = sourcedirs; 532e97f67f5SJoerg Wunsch src == NULL && *dp != NULL; 533e97f67f5SJoerg Wunsch dp++) { 534e97f67f5SJoerg Wunsch cp2 = malloc(strlen(*dp) + 9); 535e97f67f5SJoerg Wunsch if (cp2 == NULL) 536e97f67f5SJoerg Wunsch abort(); 537e97f67f5SJoerg Wunsch strcpy(cp2, "^"); 538e97f67f5SJoerg Wunsch strcat(cp2, *dp); 539e97f67f5SJoerg Wunsch strcat(cp2, "/[^/]+/"); 540e97f67f5SJoerg Wunsch if ((i = regcomp(&re2, cp2, 541e97f67f5SJoerg Wunsch REG_EXTENDED|REG_NOSUB)) 542e97f67f5SJoerg Wunsch != 0) { 543e97f67f5SJoerg Wunsch regerror(i, &re, buf, 544e97f67f5SJoerg Wunsch BUFSIZ - 1); 545e97f67f5SJoerg Wunsch errx(EX_UNAVAILABLE, 546e97f67f5SJoerg Wunsch "regcomp(%s) failed: %s", 547e97f67f5SJoerg Wunsch cp2, buf); 548e97f67f5SJoerg Wunsch } 549e97f67f5SJoerg Wunsch free(cp2); 550e97f67f5SJoerg Wunsch if (regexec(&re2, buf, 0, 551e97f67f5SJoerg Wunsch (regmatch_t *)NULL, 0) 552e97f67f5SJoerg Wunsch == 0) { 553e97f67f5SJoerg Wunsch unusual--; 554e97f67f5SJoerg Wunsch src = buf; 555e97f67f5SJoerg Wunsch } 556e97f67f5SJoerg Wunsch regfree(&re2); 557e97f67f5SJoerg Wunsch } 558e97f67f5SJoerg Wunsch } 559e97f67f5SJoerg Wunsch pclose(p); 560e97f67f5SJoerg Wunsch free(cp); 561e97f67f5SJoerg Wunsch } 562e97f67f5SJoerg Wunsch done_sources: 563e97f67f5SJoerg Wunsch 564e97f67f5SJoerg Wunsch if (opt_u && !unusual) 565e97f67f5SJoerg Wunsch continue; 566e97f67f5SJoerg Wunsch 567e97f67f5SJoerg Wunsch printed = 0; 568e97f67f5SJoerg Wunsch if (!opt_q) { 569e97f67f5SJoerg Wunsch printf("%s:", name); 570e97f67f5SJoerg Wunsch printed++; 571e97f67f5SJoerg Wunsch } 572e97f67f5SJoerg Wunsch if (bin) { 573e97f67f5SJoerg Wunsch if (printed++) 574e97f67f5SJoerg Wunsch putchar(' '); 575e97f67f5SJoerg Wunsch fputs(bin, stdout); 576e97f67f5SJoerg Wunsch } 577e97f67f5SJoerg Wunsch if (man) { 578e97f67f5SJoerg Wunsch if (printed++) 579e97f67f5SJoerg Wunsch putchar(' '); 580e97f67f5SJoerg Wunsch fputs(man, stdout); 581e97f67f5SJoerg Wunsch } 582e97f67f5SJoerg Wunsch if (src) { 583e97f67f5SJoerg Wunsch if (printed++) 584e97f67f5SJoerg Wunsch putchar(' '); 585e97f67f5SJoerg Wunsch fputs(src, stdout); 586e97f67f5SJoerg Wunsch } 587e97f67f5SJoerg Wunsch if (printed) 588e97f67f5SJoerg Wunsch putchar('\n'); 589e97f67f5SJoerg Wunsch } 590e97f67f5SJoerg Wunsch 591e97f67f5SJoerg Wunsch if (opt_m) 592e97f67f5SJoerg Wunsch regfree(&re); 593e97f67f5SJoerg Wunsch 594e97f67f5SJoerg Wunsch return (0); 595b78dcb87SJohan Karlsson } 596