none/tests/rlimit64_nofile.c

   1 #define _LARGEFILE_SOURCE
   2 #define _LARGEFILE64_SOURCE
   3
   4 #include <errno.h>
   5 #include <fcntl.h>
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <sys/resource.h>
   9 #include <unistd.h>
  10 #include "fdleak.h"
  11
  12 int main(int argc, char **argv)
  13 {
  14    struct rlimit64 oldrlim;
  15    struct rlimit64 newrlim;
  16    int fd;
  17
  18    CLOSE_INHERITED_FDS;
  19
  20    if (getrlimit64(RLIMIT_NOFILE, &oldrlim) < 0)
  21    {
  22       perror("getrlimit");
  23       exit(1);
  24    }
  25
  26    newrlim.rlim_cur = oldrlim.rlim_max+1;
  27    newrlim.rlim_max = oldrlim.rlim_max;
  28    if (setrlimit64(RLIMIT_NOFILE, &newrlim) == -1)
  29    {
  30       if (errno != EINVAL) {
  31          fprintf(stderr, "setrlimit64 exceeding hardlimit must set errno=EINVAL\n");
  32          exit(1);
  33       }
  34    }
  35    else
  36    {
  37         fprintf(stderr, "setrlimit64 exceeding hardlimit must return -1\n");
  38         exit(1);
  39    }
  40
  41    newrlim.rlim_cur = oldrlim.rlim_max;
  42    newrlim.rlim_max = oldrlim.rlim_max+1;
  43    if (setrlimit64(RLIMIT_NOFILE, &newrlim) == -1)
  44    {
  45       if (errno != EPERM) {
  46          fprintf(stderr, "setrlimit64 changing hardlimit must set errno=EPERM\n");
  47          exit(1);
  48       }
  49    }
  50    else
  51    {
  52         fprintf(stderr, "setrlimit64 changing hardlimit must return -1\n");
  53         exit(1);
  54    }
  55
  56    newrlim.rlim_cur = oldrlim.rlim_cur / 2;
  57    newrlim.rlim_max = oldrlim.rlim_max;
  58
  59    if (setrlimit64(RLIMIT_NOFILE, &newrlim) < 0)
  60    {
  61       perror("setrlimit64");
  62       exit(1);
  63    }
  64
  65    if (getrlimit64(RLIMIT_NOFILE, &newrlim) < 0)
  66    {
  67       perror("getrlimit");
  68       exit(1);
  69    }
  70
  71    if (newrlim.rlim_cur != oldrlim.rlim_cur / 2)
  72    {
  73       fprintf(stderr, "rlim_cur is %llu (should be %llu)\n",
  74               (unsigned long long)newrlim.rlim_cur,
  75               (unsigned long long)oldrlim.rlim_cur / 2);
  76    }
  77
  78    if (newrlim.rlim_max != oldrlim.rlim_max)
  79    {
  80       fprintf(stderr, "rlim_max is %llu (should be %llu)\n",
  81               (unsigned long long)newrlim.rlim_max,
  82               (unsigned long long)oldrlim.rlim_max);
  83    }
  84
  85    newrlim.rlim_cur -= 3; /* allow for stdin, stdout and stderr */
  86
  87    while (newrlim.rlim_cur-- > 0)
  88    {
  89       if (open("/dev/null", O_RDONLY) < 0)
  90       {
  91          perror("open");
  92       }
  93    }
  94
  95    if ((fd = open("/dev/null", O_RDONLY)) >= 0)
  96    {
  97       fprintf(stderr, "open succeeded with fd %d - it should have failed!\n", fd);
  98    }
  99    else if (errno != EMFILE)
 100    {
 101       perror("open");
 102    }
 103
 104    exit(0);
 105 }