Return to closefrom.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / sudo / compat

Annotation of embedaddon/sudo/compat/closefrom.c, revision 1.1.1.4

1.1       misho       1: /*
1.1.1.2   misho       2:  * Copyright (c) 2004-2005, 2007, 2010, 2012-2013
1.1       misho       3:  *     Todd C. Miller <Todd.Miller@courtesan.com>
                      4:  *
                      5:  * Permission to use, copy, modify, and distribute this software for any
                      6:  * purpose with or without fee is hereby granted, provided that the above
                      7:  * copyright notice and this permission notice appear in all copies.
                      8:  *
                      9:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                     10:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     11:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     12:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     13:  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
                     14:  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
                     15:  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
                     16:  */
                     17: 
                     18: #include <config.h>
                     19: 
1.1.1.2   misho      20: #ifndef HAVE_CLOSEFROM
                     21: 
1.1       misho      22: #include <sys/types.h>
                     23: #include <unistd.h>
                     24: #include <stdio.h>
                     25: #ifdef STDC_HEADERS
                     26: # include <stdlib.h>
                     27: # include <stddef.h>
                     28: #else
                     29: # ifdef HAVE_STDLIB_H
                     30: #  include <stdlib.h>
                     31: # endif
                     32: #endif /* STDC_HEADERS */
                     33: #include <fcntl.h>
1.1.1.4 ! misho      34: #include <limits.h>
1.1.1.2   misho      35: #ifdef HAVE_PSTAT_GETPROC
                     36: # include <sys/param.h>
                     37: # include <sys/pstat.h>
1.1       misho      38: #else
1.1.1.2   misho      39: # ifdef HAVE_DIRENT_H
                     40: #  include <dirent.h>
                     41: #  define NAMLEN(dirent) strlen((dirent)->d_name)
                     42: # else
                     43: #  define dirent direct
                     44: #  define NAMLEN(dirent) (dirent)->d_namlen
                     45: #  ifdef HAVE_SYS_NDIR_H
                     46: #   include <sys/ndir.h>
                     47: #  endif
                     48: #  ifdef HAVE_SYS_DIR_H
                     49: #   include <sys/dir.h>
                     50: #  endif
                     51: #  ifdef HAVE_NDIR_H
                     52: #   include <ndir.h>
                     53: #  endif
1.1       misho      54: # endif
                     55: #endif
                     56: 
                     57: #include "missing.h"
                     58: 
1.1.1.4 ! misho      59: #if defined(HAVE_FCNTL_CLOSEM) && !defined(HAVE_DIRFD)
        !            60: # define closefrom     closefrom_fallback
1.1       misho      61: #endif
                     62: 
                     63: /*
                     64:  * Close all file descriptors greater than or equal to lowfd.
1.1.1.4 ! misho      65:  * This is the expensive (fallback) method.
1.1       misho      66:  */
                     67: void
                     68: closefrom_fallback(int lowfd)
                     69: {
                     70:     long fd, maxfd;
                     71: 
                     72:     /*
                     73:      * Fall back on sysconf() or getdtablesize().  We avoid checking
                     74:      * resource limits since it is possible to open a file descriptor
                     75:      * and then drop the rlimit such that it is below the open fd.
                     76:      */
                     77: #ifdef HAVE_SYSCONF
                     78:     maxfd = sysconf(_SC_OPEN_MAX);
                     79: #else
                     80:     maxfd = getdtablesize();
                     81: #endif /* HAVE_SYSCONF */
                     82:     if (maxfd < 0)
                     83:        maxfd = OPEN_MAX;
                     84: 
1.1.1.3   misho      85:     for (fd = lowfd; fd < maxfd; fd++) {
                     86: #ifdef __APPLE__
1.1.1.4 ! misho      87:        /* Avoid potential libdispatch crash when we close its fds. */
        !            88:        (void) fcntl((int) fd, F_SETFD, FD_CLOEXEC);
1.1.1.3   misho      89: #else
1.1       misho      90:        (void) close((int) fd);
1.1.1.3   misho      91: #endif
                     92:     }
1.1       misho      93: }
                     94: 
                     95: /*
                     96:  * Close all file descriptors greater than or equal to lowfd.
                     97:  * We try the fast way first, falling back on the slow method.
                     98:  */
1.1.1.2   misho      99: #if defined(HAVE_FCNTL_CLOSEM)
1.1       misho     100: void
                    101: closefrom(int lowfd)
                    102: {
                    103:     if (fcntl(lowfd, F_CLOSEM, 0) == -1)
                    104:        closefrom_fallback(lowfd);
                    105: }
1.1.1.2   misho     106: #elif defined(HAVE_PSTAT_GETPROC)
                    107: void
                    108: closefrom(int lowfd)
                    109: {
                    110:     struct pst_status pstat;
                    111:     int fd;
                    112: 
                    113:     if (pstat_getproc(&pstat, sizeof(pstat), 0, getpid()) != -1) {
                    114:        for (fd = lowfd; fd <= pstat.pst_highestfd; fd++)
                    115:            (void) close(fd);
                    116:     } else {
                    117:        closefrom_fallback(lowfd);
                    118:     }
                    119: }
                    120: #elif defined(HAVE_DIRFD)
1.1       misho     121: void
                    122: closefrom(int lowfd)
                    123: {
1.1.1.4 ! misho     124:     const char *path;
1.1       misho     125:     DIR *dirp;
                    126: 
1.1.1.4 ! misho     127:     /* Use /proc/self/fd (or /dev/fd on FreeBSD) if it exists. */
        !           128: # if defined(__FreeBSD__) || defined(__APPLE__)
        !           129:     path = "/dev/fd";
        !           130: # else
        !           131:     path = "/proc/self/fd";
        !           132: # endif
        !           133:     if ((dirp = opendir(path)) != NULL) {
        !           134:        struct dirent *dent;
1.1       misho     135:        while ((dent = readdir(dirp)) != NULL) {
1.1.1.4 ! misho     136:            const char *errstr;
        !           137:            int fd = strtonum(dent->d_name, lowfd, INT_MAX, &errstr);
        !           138:            if (errstr == NULL && fd != dirfd(dirp)) {
        !           139: # ifdef __APPLE__
        !           140:                /* Avoid potential libdispatch crash when we close its fds. */
        !           141:                (void) fcntl(fd, F_SETFD, FD_CLOEXEC);
        !           142: # else
        !           143:                (void) close(fd);
        !           144: # endif
        !           145:            }
1.1       misho     146:        }
                    147:        (void) closedir(dirp);
                    148:     } else
                    149:        closefrom_fallback(lowfd);
                    150: }
                    151: #endif /* HAVE_FCNTL_CLOSEM */
1.1.1.2   misho     152: #endif /* HAVE_CLOSEFROM */