/*
 * Copyright 1990 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*
 * Copyright (c) 1980 Regents of the University of California.
 * All rights reserved.  The Berkeley software License Agreement
 * specifies the terms and conditions for redistribution.
 */

#include <sys/param.h>
#include <dirent.h>
#include <stdio.h>
#include <ctype.h>

static char *bindirs[] = {
        "/etc",
        "/sbin",
        "/usr/bin",
        "/usr/ccs/bin",
        "/usr/ccs/lib",
        "/usr/lang",
        "/usr/lbin",
        "/usr/lib",
        "/usr/sbin",
        "/usr/ucb",
        "/usr/ucblib",
        "/usr/ucbinclude",
        "/usr/games",
        "/usr/local",
        "/usr/local/bin",
        "/usr/new",
        "/usr/old",
        "/usr/hosts",
        "/usr/include",
        "/usr/etc",
        0
};
static char *mandirs[] = {
        "/usr/man/man1",
        "/usr/man/man1b",
        "/usr/man/man1c",
        "/usr/man/man1f",
        "/usr/man/man1m",
        "/usr/man/man1s",
        "/usr/man/man2",
        "/usr/man/man3",
        "/usr/man/man3b",
        "/usr/man/man3c",
        "/usr/man/man3e",
        "/usr/man/man3g",
        "/usr/man/man3j",
        "/usr/man/man3k",
        "/usr/man/man3l",
        "/usr/man/man3m",
        "/usr/man/man3n",
        "/usr/man/man3s",
        "/usr/man/man3w",
        "/usr/man/man3x",
        "/usr/man/man3x11",
        "/usr/man/man3xt",
        "/usr/man/man4",
        "/usr/man/man4b",
        "/usr/man/man5",
        "/usr/man/man6",
        "/usr/man/man7",
        "/usr/man/man7b",
        "/usr/man/man8",
        "/usr/man/man9e",
        "/usr/man/man9f",
        "/usr/man/man9s",
        "/usr/man/manl",
        "/usr/man/mann",
        "/usr/man/mano",
        0
};
static char *srcdirs[]  = {
        "/usr/src/cmd",
        "/usr/src/head",
        "/usr/src/lib",
        "/usr/src/lib/libc",
        "/usr/src/lib/libc/port",
        "/usr/src/lib/libc/port/gen",
        "/usr/src/lib/libc/port/print",
        "/usr/src/lib/libc/port/stdio",
        "/usr/src/lib/libc/port/sys",
        "/usr/src/lib/libc/sparc",
        "/usr/src/lib/libc/sparc/gen",
        "/usr/src/lib/libc/sparc/sys",
        "/usr/src/ucbcmd",
        "/usr/src/ucblib",
        "/usr/src/ucbinclude",
        "/usr/src/uts",
        "/usr/src/uts/common",
        "/usr/src/uts/sun",
        "/usr/src/uts/sun4",
        "/usr/src/uts/sun4c",
        "/usr/src/uts/sparc",
        "/usr/src/local",
        "/usr/src/new",
        "/usr/src/old",
        0
};

char    sflag = 1;
char    bflag = 1;
char    mflag = 1;
char    **Sflag;
int     Scnt;
char    **Bflag;
int     Bcnt;
char    **Mflag;
int     Mcnt;
char    uflag;

void getlist(int *, char ***, char ***, int *);
void zerof(void);
void lookup(char *);
void looksrc(char *);
void lookbin(char *);
void lookman(char *);
void findv(char **, int, char *);
void find(char **, char *);
void findin(char *, char *);

/*
 * whereis name
 * look for source, documentation and binaries
 */
int
main(int argc, char *argv[])
{

        argc--, argv++;
        if (argc == 0) {
usage:
                fprintf(stderr, "whereis [ -sbmu ] [ -SBM dir ... -f ] "
                    "name...\n");
                exit(1);
        }
        do
                if (argv[0][0] == '-') {
                        char *cp = argv[0] + 1;
                        while (*cp) {

                                switch (*cp++) {

                        case 'f':
                                break;

                        case 'S':
                                getlist(&argc, &argv, &Sflag, &Scnt);
                                break;

                        case 'B':
                                getlist(&argc, &argv, &Bflag, &Bcnt);
                                break;

                        case 'M':
                                getlist(&argc, &argv, &Mflag, &Mcnt);
                                break;

                        case 's':
                                zerof();
                                sflag++;
                                continue;

                        case 'u':
                                uflag++;
                                continue;

                        case 'b':
                                zerof();
                                bflag++;
                                continue;

                        case 'm':
                                zerof();
                                mflag++;
                                continue;

                        default:
                                goto usage;
                                }
                        }
                        argv++;
                } else
                        lookup(*argv++);
        while (--argc > 0);
        return (0);
}

void
getlist(int *argcp, char ***argvp, char ***flagp, int *cntp)
{

        (*argvp)++;
        *flagp = *argvp;
        *cntp = 0;
        for ((*argcp)--; *argcp > 0 && (*argvp)[0][0] != '-'; (*argcp)--)
                (*cntp)++, (*argvp)++;
        (*argcp)++;
        (*argvp)--;
}

void
zerof(void)
{

        if (sflag && bflag && mflag)
                sflag = bflag = mflag = 0;
}
int     count;
int     print;

void
lookup(char *cp)
{
        char *dp;

        for (dp = cp; *dp; dp++)
                continue;
        for (; dp > cp; dp--) {
                if (*dp == '.') {
                        *dp = 0;
                        break;
                }
        }
        for (dp = cp; *dp; dp++)
                if (*dp == '/')
                        cp = dp + 1;
        if (uflag) {
                print = 0;
                count = 0;
        } else
                print = 1;
again:
        if (print)
                printf("%s:", cp);
        if (sflag) {
                looksrc(cp);
                if (uflag && print == 0 && count != 1) {
                        print = 1;
                        goto again;
                }
        }
        count = 0;
        if (bflag) {
                lookbin(cp);
                if (uflag && print == 0 && count != 1) {
                        print = 1;
                        goto again;
                }
        }
        count = 0;
        if (mflag) {
                lookman(cp);
                if (uflag && print == 0 && count != 1) {
                        print = 1;
                        goto again;
                }
        }
        if (print)
                printf("\n");
}

void
looksrc(char *cp)
{
        if (Sflag == 0) {
                find(srcdirs, cp);
        } else
                findv(Sflag, Scnt, cp);
}

void
lookbin(char *cp)
{
        if (Bflag == 0)
                find(bindirs, cp);
        else
                findv(Bflag, Bcnt, cp);
}

void
lookman(char *cp)
{
        if (Mflag == 0) {
                find(mandirs, cp);
        } else
                findv(Mflag, Mcnt, cp);
}

void
findv(char **dirv, int dirc, char *cp)
{

        while (dirc > 0)
                findin(*dirv++, cp), dirc--;
}

void
find(char **dirs, char *cp)
{

        while (*dirs)
                findin(*dirs++, cp);
}

void
findin(char *dir, char *cp)
{
        DIR *dirp;
        struct dirent *dp;

        dirp = opendir(dir);
        if (dirp == NULL)
                return;
        while ((dp = readdir(dirp)) != NULL) {
                if (itsit(cp, dp->d_name)) {
                        count++;
                        if (print)
                                printf(" %s/%s", dir, dp->d_name);
                }
        }
        closedir(dirp);
}

int
itsit(char *cp, char *dp)
{
        int i = strlen(dp);

        if (dp[0] == 's' && dp[1] == '.' && itsit(cp, dp+2))
                return (1);
        while (*cp && *dp && *cp == *dp)
                cp++, dp++, i--;
        if (*cp == 0 && *dp == 0)
                return (1);
        while (isdigit(*dp))
                dp++;
        if (*cp == 0 && *dp++ == '.') {
                --i;
                while (i > 0 && *dp)
                        if (--i, *dp++ == '.')
                                return (*dp++ == 'C' && *dp++ == 0);
                return (1);
        }
        return (0);
}