/*
 * ddev.c - AIX device support functions for lsof
 */

/*
 * Copyright 1994 Purdue Research Foundation, West Lafayette, Indiana
 * 47907.  All rights reserved.
 *
 * Written by Victor A. Abell
 *
 * This software is not subject to any license of the American Telephone
 * and Telegraph Company or the Regents of the University of California.
 *
 * Permission is granted to anyone to use this software for any purpose on
 * any computer system, and to alter it and redistribute it freely, subject
 * to the following restrictions:
 *
 * 1. Neither the authors nor Purdue University are responsible for any
 *    consequences of the use of this software.
 *
 * 2. The origin of this software must not be misrepresented, either by
 *    explicit claim or by omission.  Credit to the authors and Purdue
 *    University must appear in documentation and sources.
 *
 * 3. Altered versions must be plainly marked as such, and must not be
 *    misrepresented as being the original software.
 *
 * 4. This notice may not be removed or altered.
 */

#ifndef lint
static char copyright[] =
    "@(#) Copyright 1994 Purdue Research Foundation.\nAll rights reserved.\n";
#endif

#include "common.h"

/*
 * Local definitions
 */

#define LIKE_BLK_SPEC "like block special"
#define LIKE_CHR_SPEC "like character special"

/*
 * Local function prototypes
 */

static int rmdupdev(struct lsof_context *ctx, struct l_dev ***dp, int n,
                    char *nm);

#if defined(HASDCACHE) && AIXV >= 4140

/*
 * clr_sect() - clear cached clone and pseudo sections
 */

void clr_sect(struct lsof_context *ctx) {
    struct clone *c, *c1;

    if (Clone) {
        for (c = Clone; c; c = c1) {
            c1 = c->next;
            if (c->cd.name)
                (void)free((FREE_P *)c->cd.name);
            (void)free((FREE_P *)c);
        }
        Clone = (struct clone *)NULL;
    }
}
#endif /* defined(HASDCACHE) && AIXV>=4140 */

/*
 * getchan() - get channel from file path name
 */

int getchan(char *p) /* file path name */
{
    int ch;
    char *s;

    if (!(s = strrchr(p, '/')))
        return (-1);
    if (*(++s) == '\0')
        return (-1);
    for (ch = 0; *s; s++) {

#if defined(__STDC__)
        if (!isdigit(*s))
#else
        if (!isascii(*s) || !isdigit(*s))
#endif /* __STDC__ */

            return (-1);
        ch = (ch * 10) + *s - '0';
    }
    return (ch);
}

/*
 * printdevname() - print device name
 */

int printdevname(struct lsof_context *ctx, /* context */
                 dev_t *dev,               /* device */
                 dev_t *rdev,              /* raw device */
                 int f,                    /* 1 = follow with '\n' */
                 int nty)                  /* node type: N_BLK or N_CHR */
{
    struct l_dev *dp;
    /*
     * Search device table for a full match.
     */
    if ((dp = lkupdev(ctx, dev, rdev, 1, 1))) {
        if (Lf->ch < 0)
            safestrprt(dp->name, stdout, f);
        else {
            safestrprt(dp->name, stdout, 0);
            (void)printf("/%d%s", Lf->ch, f ? "\n" : "");
        }
        return (1);
    }
    /*
     * Search device table for a match without inode number and dev.
     */
    if ((dp = lkupdev(ctx, &DevDev, rdev, 0, 1))) {

        /*
         * A raw device match was found.  Record it as a name column addition.
         */
        char *cp, *ttl;
        int len;

        ttl = (nty == N_BLK) ? LIKE_BLK_SPEC : LIKE_CHR_SPEC;
        len = (int)(1 + strlen(ttl) + 1 + strlen(dp->name) + 1);
        if (!(cp = (char *)malloc((MALLOC_S)(len + 1)))) {
            (void)fprintf(stderr, "%s: no nma space for: (%s %s)\n", Pn, ttl,
                          dp->name);
            Error(ctx);
        }
        (void)snpf(cp, len + 1, "(%s %s)", ttl, dp->name);
        (void)add_nma(ctx, cp, len);
        (void)free((MALLOC_P *)cp);
        return (0);
    }
    return (0);
}

/*
 * readdev() - read device names, modes and types
 */

void readdev(struct lsof_context *ctx, /* context */
             int skip)                 /* skip device cache read if 1 */
{

#if defined(HASDCACHE)
    int dcrd;
#endif /* defined(HASDCACHE) */

    DIR *dfp;
    struct dirent *dp;
    char *fp = (char *)NULL;
    int i = 0;

#if defined(HASBLKDEV)
    int j = 0;
#endif /* defined(HASBLKDEV) */

    char *path = (char *)NULL;
    MALLOC_S pl;
    struct stat sb;

#if AIXV >= 4140
    struct clone *c;
    dev_t cd;
#endif /* AIXV >=4140 */

    if (Sdev)
        return;

#if defined(HASDCACHE)
    /*
     * Read device cache, as directed.
     */
    if (!skip) {
        if (DCstate == 2 || DCstate == 3) {
            if ((dcrd = read_dcache(ctx)) == 0)
                return;
        }
    } else
        dcrd = 1;
#endif /* defined(HASDCACHE) */

#if AIXV >= 4140
    /*
     * Establish the clone major device for AIX 4.1.4 and above.
     */
    if (stat("/dev/clone", &sb) == 0) {
        cd = sb.st_rdev;
        CloneMaj = GET_MAJ_DEV(cd);
    }
#endif /* AIXV >=4140 */

    Dstk = (char **)NULL;
    Dstkn = Dstkx = 0;
    (void)stkdir(ctx, "/dev");
    /*
     * Unstack the next /dev or /dev/<subdirectory> directory.
     */
    while (--Dstkx >= 0) {
        if (!(dfp = opendir(Dstk[Dstkx]))) {

#if defined(WARNDEVACCESS)
            if (!Fwarn) {
                (void)fprintf(stderr, "%s: WARNING: can't open: ", Pn);
                safestrprt(Dstk[Dstkx], stderr, 1);
            }
#endif /* defined(WARNDEVACCESS) */

            (void)free((FREE_P *)Dstk[Dstkx]);
            Dstk[Dstkx] = (char *)NULL;
            continue;
        }
        if (path) {
            (void)free((FREE_P *)path);
            path = (char *)NULL;
        }
        if (!(path =
                  mkstrcat(Dstk[Dstkx], -1, "/", 1, (char *)NULL, -1, &pl))) {
            (void)fprintf(stderr, "%s: no space for: ", Pn);
            safestrprt(Dstk[Dstkx], stderr, 1);
            Error(ctx);
        }
        (void)free((FREE_P *)Dstk[Dstkx]);
        Dstk[Dstkx] = (char *)NULL;
        /*
         * Scan the directory.
         */
        for (dp = readdir(dfp); dp; dp = readdir(dfp)) {
            if (!dp->d_ino || (dp->d_name[0] == '.'))
                continue;
            /*
             * Form the full path name and get its status.
             */
            if (fp) {
                (void)free((FREE_P *)fp);
                fp = (char *)NULL;
            }
            if (!(fp = mkstrcat(path, (int)pl, dp->d_name, dp->d_namlen,
                                (char *)NULL, -1, (MALLOC_S *)NULL))) {
                (void)fprintf(stderr, "%s: no space for: ", Pn);
                safestrprt(path, stderr, 0);
                safestrprt(dp->d_name, stderr, 1);
                Error(ctx);
            }

#if defined(USE_STAT)
            if (stat(fp, &sb) != 0)
#else  /* !defined(USE_STAT) */
            if (lstat(fp, &sb) != 0)
#endif /* defined(USE_STAT) */

            {
                if (errno == ENOENT) /* symbolic link to nowhere? */
                    continue;

#if defined(WARNDEVACCESS)
                if (!Fwarn) {
                    int errno_save = errno;

                    (void)fprintf(stderr, "%s: can't stat: ", Pn);
                    safestrprt(fp, stderr, 0);
                    (void)fprintf(stderr, ": %s\n", strerror(errno_save));
                }
#endif /* defined(WARNDEVACCESS) */

                continue;
            }
            /*
             * If it's a subdirectory, stack its name for later processing.
             */
            if ((sb.st_mode & S_IFMT) == S_IFDIR) {
                (void)stkdir(ctx, fp);
                continue;
            }
            if ((sb.st_mode & S_IFMT) == S_IFCHR) {

                /*
                 * Save character device information.
                 */
                if (i >= Ndev) {
                    Ndev += DEVINCR;
                    if (!Devtp)
                        Devtp = (struct l_dev *)malloc(
                            (MALLOC_S)(sizeof(struct l_dev) * Ndev));
                    else
                        Devtp = (struct l_dev *)realloc(
                            (MALLOC_P *)Devtp,
                            (MALLOC_S)(sizeof(struct l_dev) * Ndev));
                    if (!Devtp) {
                        (void)fprintf(
                            stderr, "%s: no space for character device\n", Pn);
                        Error(ctx);
                    }
                }
                Devtp[i].rdev = sb.st_rdev;
                Devtp[i].inode = (INODETYPE)sb.st_ino;
                if (!(Devtp[i].name = mkstrcpy(fp, (MALLOC_S *)NULL))) {
                    (void)fprintf(stderr, "%s: no space for: ", Pn);
                    safestrprt(fp, stderr, 1);
                    Error(ctx);
                }
                Devtp[i].v = 0;
                i++;

#if AIXV >= 4140
                /*
                 * Save information on AIX 4.1.4 and above clone devices.
                 */
                if (CloneMaj >= 0 && CloneMaj == GET_MAJ_DEV(sb.st_rdev)) {
                    if (!(c = (struct clone *)malloc(
                              (MALLOC_S)sizeof(struct clone)))) {
                        (void)fprintf(stderr,
                                      "%s: no space for clone device: ", Pn);
                        safestrprt(fp, stderr, 1);
                        Error(ctx);
                    }
                    if (!(c->cd.name = mkstrcpy(fp, (MALLOC_S)NULL))) {
                        (void)fprintf(stderr,
                                      "%s: no space for clone name: ", Pn);
                        safestrprt(fp, stderr, 1);
                        Error(ctx);
                    }
                    c->cd.inode = (INODETYPE)sb.st_ino;
                    c->cd.rdev = sb.st_rdev;
                    c->cd.v = 0;
                    c->next = Clone;
                    Clone = c;
                    if (ClonePtc < 0 && strcmp(path, "/dev/ptc") == 0)
                        ClonePtc = GET_MIN_DEV(sb.st_rdev);
                }
#endif /* AIXV >=4140 */
            }

#if defined(HASBLKDEV)
            if ((sb.st_mode & S_IFMT) == S_IFBLK) {

                /*
                 * Save block device information in BDevtp[].
                 */
                if (j >= BNdev) {
                    BNdev += DEVINCR;
                    if (!BDevtp)
                        BDevtp = (struct l_dev *)malloc(
                            (MALLOC_S)(sizeof(struct l_dev) * BNdev));
                    else
                        BDevtp = (struct l_dev *)realloc(
                            (MALLOC_P *)BDevtp,
                            (MALLOC_S)(sizeof(struct l_dev) * BNdev));
                    if (!BDevtp) {
                        (void)fprintf(stderr, "%s: no space for block device\n",
                                      Pn);
                        Error(ctx);
                    }
                }
                BDevtp[j].rdev = sb.st_rdev;
                BDevtp[j].inode = (INODETYPE)sb.st_ino;
                BDevtp[j].name = fp;
                fp = (char *)NULL;
                BDevtp[j].v = 0;
                j++;
            }
#endif /* defined(HASBLKDEV) */
        }
        (void)closedir(dfp);
    }
    /*
     * Free any allocated space.
     */
    if (Dstk) {
        (void)free((FREE_P *)Dstk);
        Dstk = (char **)NULL;
        Dstkn = Dstkx = 0;
    }
    if (fp)
        (void)free((FREE_P *)fp);
    if (path)
        (void)free((FREE_P *)path);
        /*
         * Reduce the BDevtp[] (optional) and Devtp[] tables to their minimum
         * sizes; allocate and build sort pointer lists; and sort the tables by
         * device number.
         */

#if defined(HASBLKDEV)
    if (BNdev) {
        if (BNdev > j) {
            BNdev = j;
            BDevtp = (struct l_dev *)realloc(
                (MALLOC_P *)BDevtp, (MALLOC_S)(sizeof(struct l_dev) * BNdev));
        }
        if (!(BSdev = (struct l_dev **)malloc(
                  (MALLOC_S)(sizeof(struct l_dev *) * BNdev)))) {
            (void)fprintf(stderr,
                          "%s: no space for block device sort pointers\n", Pn);
            Error(ctx);
        }
        for (j = 0; j < BNdev; j++) {
            BSdev[j] = &BDevtp[j];
        }
        (void)qsort((QSORT_P *)BSdev, (size_t)BNdev,
                    (size_t)sizeof(struct l_dev *), compdev);
        BNdev = rmdupdev(ctx, &BSdev, BNdev, "block");
    } else {
        if (!Fwarn)
            (void)fprintf(stderr, "%s: WARNING: no block devices found\n", Pn);
    }
#endif /* defined(HASBLKDEV) */

    if (Ndev) {
        if (Ndev > i) {
            Ndev = i;
            Devtp = (struct l_dev *)realloc(
                (MALLOC_P *)Devtp, (MALLOC_S)(sizeof(struct l_dev) * Ndev));
        }
        if (!(Sdev = (struct l_dev **)malloc(
                  (MALLOC_S)(sizeof(struct l_dev *) * Ndev)))) {
            (void)fprintf(stderr,
                          "%s: no space for character device sort pointers\n",
                          Pn);
            Error(ctx);
        }
        for (i = 0; i < Ndev; i++) {
            Sdev[i] = &Devtp[i];
        }
        (void)qsort((QSORT_P *)Sdev, (size_t)Ndev,
                    (size_t)sizeof(struct l_dev *), compdev);
        Ndev = rmdupdev(ctx, &Sdev, Ndev, "char");
    } else {
        (void)fprintf(stderr, "%s: no character devices found\n", Pn);
        Error(ctx);
    }

#if defined(HASDCACHE)
    /*
     * Write device cache file, as required.
     */
    if (DCstate == 1 || (DCstate == 3 && dcrd))
        write_dcache(ctx);
#endif /* defined(HASDCACHE) */
}

#if defined(HASDCACHE)
/*
 * rereaddev() - reread device names, modes and types
 */

void rereaddev(struct lsof_context *ctx) {
    (void)clr_devtab(ctx);

#    if defined(DCACHE_CLR)
    (void)DCACHE_CLR(ctx);
#    endif /* defined(DCACHE_CLR) */

    readdev(ctx, 1);
    DCunsafe = 0;
}
#endif /* defined(HASDCACHE) */

/*
 * rmdupdev() - remove duplicate (major/minor/inode) devices
 */

static int rmdupdev(struct lsof_context *ctx, /* context */
                    struct l_dev ***dp, /* device table pointers address */
                    int n,              /* number of pointers */
                    char *nm) /* device table name for error message */
{

#if AIXV >= 4140
    struct clone *c, *cp;
#endif /* AIXV>=4140 */

    int i, j, k;
    struct l_dev **p;

    for (i = j = 0, p = *dp; i < n;) {
        for (k = i + 1; k < n; k++) {
            if (p[i]->rdev != p[k]->rdev || p[i]->inode != p[k]->inode)
                break;

#if AIXV >= 4140
            /*
             * See if we're deleting a duplicate clone device.  If so,
             * delete its clone table entry.
             */
            for (c = Clone, cp = (struct clone *)NULL; c; cp = c, c = c->next) {
                if (c->cd.rdev != p[k]->rdev || c->cd.inode != p[k]->inode ||
                    strcmp(c->cd.name, p[k]->name))
                    continue;
                if (!cp)
                    Clone = c->next;
                else
                    cp->next = c->next;
                if (c->cd.name)
                    (void)free((FREE_P *)c->cd.name);
                (void)free((FREE_P *)c);
                break;
            }
#endif /* AIXV>=4140 */
        }
        if (i != j)
            p[j] = p[i];
        j++;
        i = k;
    }
    if (n == j)
        return (n);
    if (!(*dp = (struct l_dev **)realloc(
              (MALLOC_P *)*dp, (MALLOC_S)(j * sizeof(struct l_dev *))))) {
        (void)fprintf(stderr, "%s: can't realloc %s device pointers\n", Pn, nm);
        Error(ctx);
    }
    return (j);
}

#if defined(HASDCACHE) && AIXV >= 4140
/*
 * rw_clone_sect() - read/write the device cache file clone section
 */

int rw_clone_sect(struct lsof_context *ctx, /* context */
                  int m)                    /* mode: 1 = read; 2 = write */
{
    char buf[MAXPATHLEN * 2], *cp;
    struct clone *c;
    int i, len, n;

    if (m == 1) {

        /*
         * Read the clone section header and validate it.
         */
        if (!fgets(buf, sizeof(buf), DCfs)) {

        bad_clone_sect:

            if (!Fwarn) {
                (void)fprintf(stderr,
                              "%s: bad clone section header in %s: ", Pn,
                              DCpath[DCpathX]);
                safestrprt(buf, stderr, 1);
            }
            return (1);
        }
        (void)crc(buf, strlen(buf), &DCcksum);
        len = strlen("clone section: ");
        if (strncmp(buf, "clone section: ", len) != 0)
            goto bad_clone_sect;
        if ((n = atoi(&buf[len])) < 0)
            goto bad_clone_sect;
        /*
         * Read the clone section lines and create the Clone list.
         */
        for (i = 0; i < n; i++) {
            if (!fgets(buf, sizeof(buf), DCfs)) {
                if (!Fwarn) {
                    (void)fprintf(stderr, "%s: bad clone line in %s: ", Pn,
                                  DCpath[DCpathX]);
                    safestrprt(buf, stderr, 1);
                }
                return (1);
            }
            (void)crc(buf, strlen(buf), &DCcksum);
            /*
             * Allocate a clone structure.
             */
            if (!(c = (struct clone *)calloc(1, sizeof(struct clone)))) {
                (void)fprintf(stderr, "%s: no space for cached clone: ", Pn);
                safestrprt(buf, stderr, 1);
                Error(ctx);
            }
            /*
             * Enter the clone device number.
             */
            if (!(cp = x2dev(buf, &c->cd.rdev)) || *cp++ != ' ') {

            bad_cached_clone:
                if (!Fwarn) {
                    (void)fprintf(stderr, "%s: bad cached clone device: ", Pn);
                    safestrprt(buf, stderr, 1);
                }
                return (1);
            }
            CloneMaj = GET_MAJ_DEV(c->cd.rdev);
            /*
             * Enter the clone inode number.
             */
            for (c->cd.inode = (INODETYPE)0; *cp != ' '; cp++) {
                if (*cp < '0' || *cp > '9')
                    goto bad_cached_clone;
                c->cd.inode = (INODETYPE)((c->cd.inode * 10) + (*cp - '0'));
            }
            /*
             * Enter the clone path name.
             */
            if ((len = strlen(++cp)) < 2 || *(cp + len - 1) != '\n') {
                if (!Fwarn) {
                    (void)fprintf(stderr, "%s: bad cached clone path: ", Pn);
                    safestrprt(buf, stderr, 1);
                }
                return (1);
            }
            *(cp + len - 1) = '\0';
            if (!(c->cd.name = mkstrcpy(cp, (MALLOC_S *)NULL))) {
                (void)fprintf(stderr,
                              "%s: no space for cached clone path: ", Pn);
                safestrprt(buf, stderr, 1);
                Error(ctx);
            }
            c->cd.v = 0;
            c->next = Clone;
            Clone = c;
            if (ClonePtc < 0 && strcmp(c->cd.name, "/dev/ptc") == 0)
                ClonePtc = GET_MIN_DEV(c->cd.rdev);
        }
        return (0);
    } else if (m == 2) {

        /*
         * Write the clone section header.
         */
        for (c = Clone, n = 0; c; c = c->next, n++)
            ;
        (void)snpf(buf, sizeof(buf), "clone section: %d\n", n);
        if (wr2DCfd(ctx, buf, &DCcksum))
            return (1);
        /*
         * Write the clone section lines.
         */
        for (c = Clone; c; c = c->next) {
            (void)snpf(buf, sizeof(buf), "%x %ld %s\n", c->cd.rdev,
                       (long)c->cd.inode, c->cd.name);
            if (wr2DCfd(ctx, buf, &DCcksum))
                return (1);
        }
        return (0);
    }
    /*
     * A shouldn't-happen case: mode neither 1 nor 2.
     */
    (void)fprintf(stderr, "%s: internal rw_clone_sect error: %d\n", Pn, m);
    Error(ctx);
}
#endif /* defined(HASDCACHE) && AIXV>=4140 */

#if defined(HASDCACHE)
/*
 * vfy_dev() - verify a device table entry (usually when DCunsafe == 1)
 *
 * Note: rereads entire device table when an entry can't be verified.
 */

int vfy_dev(struct lsof_context *ctx, /* context */
            struct l_dev *dp)         /* device table pointer */
{
    struct stat sb;

    if (!DCunsafe || dp->v)
        return (1);

#    if defined(USE_STAT)
    if (stat(dp->name, &sb) != 0
#    else  /* !defined(USE_STAT) */
    if (lstat(dp->name, &sb) != 0
#    endif /* defined(USE_STAT) */

        || dp->rdev != sb.st_rdev || dp->inode != (INODETYPE)sb.st_ino) {
        (void)rereaddev(ctx);
        return (0);
    }
    dp->v = 1;
    return (1);
}
#endif /* defined(HASDCACHE) */
