1: #include <errno.h>
    2: #include <stdbool.h>
    3: #include <stdio.h>
    4: #include <sys/resource.h>
    5: 
    6: 
    7: static struct
    8: {
    9:   const char *name;
   10:   int resource;
   11:   bool required;
   12: } tests[] =
   13:   {
   14:     /* The following 7 limits are part of POSIX and must exist.  */
   15:     { "RLIMIT_CORE", RLIMIT_CORE, true },
   16:     { "RLIMIT_CPU", RLIMIT_CPU, true },
   17:     { "RLIMIT_DATA", RLIMIT_DATA, true },
   18:     { "RLIMIT_FSIZE", RLIMIT_FSIZE, true },
   19:     { "RLIMIT_NOFILE", RLIMIT_NOFILE, true },
   20:     { "RLIMIT_STACK", RLIMIT_STACK, true },
   21:     { "RLIMIT_AS", RLIMIT_AS, true },
   22:     /* The following are traditional Unix limits which are also
   23:        expected (by us).  */
   24:     { "RLIMIT_RSS", RLIMIT_RSS, true },
   25:     { "RLIMIT_NPROC", RLIMIT_NPROC, true },
   26:     /* The following are extensions.  */
   27: #ifdef RLIMIT_MEMLOCK
   28:     { "RLIMIT_MEMLOCK", RLIMIT_MEMLOCK, false },
   29: #endif
   30: #ifdef RLIMIT_LOCKS
   31:     { "RLIMIT_LOCKS", RLIMIT_LOCKS, false },
   32: #endif
   33: #ifdef RLIMIT_SIGPENDING
   34:     { "RLIMIT_SIGPENDING", RLIMIT_SIGPENDING, false },
   35: #endif
   36: #ifdef RLIMIT_MSGQUEUE
   37:     { "RLIMIT_MSGQUEUE", RLIMIT_MSGQUEUE, false },
   38: #endif
   39: #ifdef RLIMIT_NICE
   40:     { "RLIMIT_NICE", RLIMIT_NICE, false },
   41: #endif
   42: #ifdef RLIMIT_RTPRIO
   43:     { "RLIMIT_RTPRIO", RLIMIT_RTPRIO, false },
   44: #endif
   45:   };
   46: #define ntests (sizeof (tests) / sizeof (tests[0]))
   47: 
   48: 
   49: static int
   50: do_test (void)
   51: {
   52:   int status = 0;
   53: 
   54:   for (int i = 0; i < ntests; ++i)
   55:     {
   56:       bool this_ok = true;
   57: 
   58:       struct rlimit r;
   59:       int res = getrlimit (tests[i].resource, &r);
   60:       if (res == -1)
   61:         {
   62:           if (errno == EINVAL)
   63:             {
   64:               if (tests[i].required)
   65:                 {
   66:                   printf ("limit %s expectedly not available for getrlimit\n",
   67:                           tests[i].name);
   68:                   status = 1;
   69:                   this_ok = false;
   70:                 }
   71:             }
   72:           else
   73:             {
   74:               printf ("getrlimit for %s returned unexpected error: %m\n",
   75:                       tests[i].name);
   76:               status = 1;
   77:               this_ok = false;
   78:             }
   79:         }
   80: 
   81:       struct rlimit64 r64;
   82:       res = getrlimit64 (tests[i].resource, &r64);
   83:       if (res == -1)
   84:         {
   85:           if (errno == EINVAL)
   86:             {
   87:               if (tests[i].required)
   88:                 {
   89:                   printf ("limit %s expectedly not available for getrlimit64"
   90:                           "\n", tests[i].name);
   91:                   status = 1;
   92:                   this_ok = false;
   93:                 }
   94:             }
   95:           else
   96:             {
   97:               printf ("getrlimit64 for %s returned unexpected error: %m\n",
   98:                       tests[i].name);
   99:               status = 1;
  100:               this_ok = false;
  101:             }
  102:         }
  103: 
  104:       if (this_ok)
  105:         printf ("limit %s OK\n", tests[i].name);
  106:     }
  107: 
  108:   return status;
  109: }
  110: 
  111: #define TEST_FUNCTION do_test ()
  112: #include "../test-skeleton.c"