/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 * Copyright 2023 Bill Sommerfeld <sommerfeld@alum.mit.edu>
 */

#pragma weak _closefrom = closefrom
#pragma weak _fdwalk = fdwalk

#include "lint.h"
#include <ctype.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <dirent.h>
#include <limits.h>
#include <errno.h>
#include <alloca.h>
#include <sys/resource.h>

/* Initial size of the open file descriptor array */
#define FDS_SZ  (1024 * sizeof (int))

/*
 * Iterate over all open file descriptors, calling 'func' on each one.
 * Terminate the iteration when 'func' returns non-zero or when all
 * open file descriptors have been processed.  Return the value of
 * the last non-zero return from 'func' or zero.
 */
int
fdwalk(int (*func)(void *, int), void *cd)
{
        int err = errno;
        int rv = 0;
        int max_fds = INT_MAX;
        struct rlimit rl;
        DIR *dirp;
        struct dirent64 *dp;
        int *fds;
        size_t fds_sz;
        int nfds;
        int i;

        nfds = 0;
        fds = alloca(FDS_SZ);
        fds_sz = FDS_SZ;
        if ((dirp = opendir("/proc/self/fd")) != NULL) {
                /*
                 * Collect all of the open file descriptors and close
                 * the directory before calling 'func' on any of them.
                 */
                while ((dp = readdir64(dirp)) != NULL) {
                        /* skip '.', '..' and the opendir() fd */
                        if (!isdigit(dp->d_name[0]) ||
                            (i = atoi(dp->d_name)) == dirp->dd_fd)
                                continue;
                        if (fds_sz <= nfds * sizeof (int)) {
                                fds = memcpy(alloca(fds_sz * 2), fds, fds_sz);
                                fds_sz *= 2;
                        }
                        fds[nfds++] = i;
                }
                (void) closedir(dirp);
        } else {
                /*
                 * We could not open the /proc file descriptor directory.
                 * We have to do it the hard way.
                 */
                if (getrlimit(RLIMIT_NOFILE, &rl) == 0)
                        max_fds = (rl.rlim_max == RLIM_INFINITY)?
                            INT_MAX : rl.rlim_max;
                for (i = 0; i < max_fds; i++) {
                        if (fcntl(i, F_GETFD) < 0)
                                continue;
                        if (fds_sz <= nfds * sizeof (int)) {
                                fds = memcpy(alloca(fds_sz * 2), fds, fds_sz);
                                fds_sz *= 2;
                        }
                        fds[nfds++] = i;
                }
        }

        /*
         * Restore the original value of errno so that
         * the caller sees only the value of errno set
         * by the callback function.
         */
        errno = err;

        /*
         * Perform the callbacks on all of the open files.
         */
        for (i = 0; i < nfds; i++)
                if ((rv = func(cd, fds[i])) != 0)
                        break;

        return (rv);
}

/*
 * Call-back function for closefrom(), below.
 */
static int
void_close(void *lowp, int fd)
{
        if (fd >= *(int *)lowp)
                (void) close(fd);
        return (0);
}

/*
 * Close all open file descriptors greater than or equal to lowfd.
 */
void
closefrom(int lowfd)
{
        int low = (lowfd < 0)? 0 : lowfd;

        /*
         * Close lowfd right away as a hedge against failing
         * to open the /proc file descriptor directory due
         * all file descriptors being currently used up.
         */
        (void) close(low);
        (void) fdwalk(void_close, &low);
}