diff options
Diffstat (limited to 'main/strcompat.c')
-rw-r--r-- | main/strcompat.c | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/main/strcompat.c b/main/strcompat.c index ff123713e..b1d8b0d1d 100644 --- a/main/strcompat.c +++ b/main/strcompat.c @@ -22,6 +22,10 @@ #include "asterisk.h" #include <ctype.h> +#include <sys/time.h> /* for getrlimit(2) */ +#include <sys/resource.h> /* for getrlimit(2) */ +#include <sys/types.h> /* for opendir(3) */ +#include <dirent.h> /* for opendir(3) */ #ifndef HAVE_STRSEP char *strsep(char **str, const char *delims) @@ -399,3 +403,33 @@ int ffsll(long long n) } #endif +#ifndef HAVE_CLOSEFROM +void closefrom(int n) +{ + long x; + struct rlimit rl; + DIR *dir; + char path[16], *result; + struct dirent *entry; + + snprintf(path, sizeof(path), "/proc/%d/fd", (int) getpid()); + if ((dir = opendir(path))) { + while ((entry = readdir(dir))) { + /* Skip . and .. */ + if (entry->d_name[0] == '.') { + continue; + } + if ((x = strtol(entry->d_name, &result, 10)) && x >= n) { + close(x); + } + } + closedir(dir); + } else { + getrlimit(RLIMIT_NOFILE, &rl); + for (x = n; x < rl.rlim_cur; x++) { + close(x); + } + } +} +#endif + |