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