usr.bin/which/which.c

   1 /**
   2  * Copyright (c) 2000 Dan Papasian.  All rights reserved.
   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  * 3. The name of the author may not be used to endorse or promote products
  13  *    derived from this software without specific prior written permission.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  * $FreeBSD: src/usr.bin/which/which.c,v 1.5 2002/06/30 06:02:39 tjr Exp $");
  26  * $DragonFly: src/usr.bin/which/which.c,v 1.2 2005/03/03 17:09:58 liamfoy Exp $
  27  */
  28
  29 #include <sys/stat.h>
  30 #include <sys/param.h>
  31
  32 #include <err.h>
  33 #include <stdio.h>
  34 #include <stdlib.h>
  35 #include <string.h>
  36 #include <unistd.h>
  37
  38 static void     usage(void);
  39 static int      print_matches(char *, char *);
  40
  41 int    silent;
  42 int    allpaths;
  43
  44 int
  45 main(int argc, char **argv)
  46 {
  47        char *p, *path;
  48        ssize_t pathlen;
  49        int opt, status;
  50
  51        status = EXIT_SUCCESS;
  52
  53        /* If called without args, die silently to conform */
  54        if (argc < 2)
  55                exit(EXIT_FAILURE);
  56
  57        while ((opt = getopt(argc, argv, "as")) != -1) {
  58                switch (opt) {
  59                case 'a':
  60                        allpaths = 1;
  61                        break;
  62                case 's':
  63                        silent = 1;
  64                        break;
  65                default:
  66                        usage();
  67                        break;
  68                }
  69        }
  70
  71        argv += optind;
  72        argc -= optind;
  73
  74        if ((p = getenv("PATH")) == NULL)
  75                exit(EXIT_FAILURE);
  76        pathlen = strlen(p) + 1;
  77        path = malloc(pathlen);
  78        if (path == NULL)
  79                err(EXIT_FAILURE, "malloc failed");
  80
  81        if (argc == 0)
  82                status = EXIT_FAILURE;
  83
  84        while (argc > 0) {
  85                memcpy(path, p, pathlen);
  86
  87                if (strlen(*argv) >= FILENAME_MAX ||
  88                    print_matches(path, *argv) == -1)
  89                        status = EXIT_FAILURE;
  90
  91                argv++;
  92                argc--;
  93        }
  94
  95        exit(status);
  96 }
  97
  98 static void
  99 usage(void)
 100 {
 101        errx(EXIT_FAILURE, "usage: which [-as] program ...");
 102 }
 103
 104 static int
 105 is_there(const char *candidate)
 106 {
 107        struct stat fin;
 108
 109        /* XXX work around access(2) false positives for superuser */
 110        if (access(candidate, X_OK) == 0 &&
 111            stat(candidate, &fin) == 0 &&
 112            S_ISREG(fin.st_mode) &&
 113            (getuid() != 0 ||
 114            (fin.st_mode & (S_IXUSR | S_IXGRP | S_IXOTH)) != 0)) {
 115                if (!silent)
 116                        printf("%s\n", candidate);
 117                return (1);
 118        }
 119        return (0);
 120 }
 121
 122 static int
 123 print_matches(char *path, char *filename)
 124 {
 125        char candidate[PATH_MAX];
 126        const char *d;
 127        int found;
 128
 129        if (strchr(filename, '/') != NULL)
 130                return (is_there(filename) ? 0 : -1);
 131        found = 0;
 132        while ((d = strsep(&path, ":")) != NULL) {
 133                if (*d == '\0')
 134                        d = ".";
 135                if (snprintf(candidate, sizeof(candidate), "%s/%s", d,
 136                    filename) >= (int)sizeof(candidate))
 137                        continue;
 138                if (is_there(candidate)) {
 139                        found = 1;
 140                        if (!allpaths)
 141                                break;
 142                }
 143        }
 144        return (found ? 0 : -1);
 145 }
 146