1  /* Copyright (C) 2005-2023 Free Software Foundation, Inc.
       2     This file is part of the GNU C Library.
       3  
       4     The GNU C Library is free software; you can redistribute it and/or
       5     modify it under the terms of the GNU Lesser General Public
       6     License as published by the Free Software Foundation; either
       7     version 2.1 of the License, or (at your option) any later version.
       8  
       9     The GNU C Library is distributed in the hope that it will be useful,
      10     but WITHOUT ANY WARRANTY; without even the implied warranty of
      11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      12     Lesser General Public License for more details.
      13  
      14     You should have received a copy of the GNU Lesser General Public
      15     License along with the GNU C Library; if not, see
      16     <https://www.gnu.org/licenses/>.  */
      17  
      18  #include <errno.h>
      19  #include <stdbool.h>
      20  #include <stdio.h>
      21  #include <sys/resource.h>
      22  
      23  
      24  static struct
      25  {
      26    const char *name;
      27    int resource;
      28    bool required;
      29  } tests[] =
      30    {
      31      /* The following 7 limits are part of POSIX and must exist.  */
      32      { "RLIMIT_CORE", RLIMIT_CORE, true },
      33      { "RLIMIT_CPU", RLIMIT_CPU, true },
      34      { "RLIMIT_DATA", RLIMIT_DATA, true },
      35      { "RLIMIT_FSIZE", RLIMIT_FSIZE, true },
      36      { "RLIMIT_NOFILE", RLIMIT_NOFILE, true },
      37      { "RLIMIT_STACK", RLIMIT_STACK, true },
      38      { "RLIMIT_AS", RLIMIT_AS, true },
      39      /* The following are traditional Unix limits which are also
      40         expected (by us).  */
      41      { "RLIMIT_RSS", RLIMIT_RSS, true },
      42      { "RLIMIT_NPROC", RLIMIT_NPROC, true },
      43      /* The following are extensions.  */
      44  #ifdef RLIMIT_MEMLOCK
      45      { "RLIMIT_MEMLOCK", RLIMIT_MEMLOCK, false },
      46  #endif
      47  #ifdef RLIMIT_LOCKS
      48      { "RLIMIT_LOCKS", RLIMIT_LOCKS, false },
      49  #endif
      50  #ifdef RLIMIT_SIGPENDING
      51      { "RLIMIT_SIGPENDING", RLIMIT_SIGPENDING, false },
      52  #endif
      53  #ifdef RLIMIT_MSGQUEUE
      54      { "RLIMIT_MSGQUEUE", RLIMIT_MSGQUEUE, false },
      55  #endif
      56  #ifdef RLIMIT_NICE
      57      { "RLIMIT_NICE", RLIMIT_NICE, false },
      58  #endif
      59  #ifdef RLIMIT_RTPRIO
      60      { "RLIMIT_RTPRIO", RLIMIT_RTPRIO, false },
      61  #endif
      62    };
      63  #define ntests (sizeof (tests) / sizeof (tests[0]))
      64  
      65  
      66  static int
      67  do_test (void)
      68  {
      69    int status = 0;
      70  
      71    for (int i = 0; i < ntests; ++i)
      72      {
      73        bool this_ok = true;
      74  
      75        struct rlimit r;
      76        int res = getrlimit (tests[i].resource, &r);
      77        if (res == -1)
      78  	{
      79  	  if (errno == EINVAL)
      80  	    {
      81  	      if (tests[i].required)
      82  		{
      83  		  printf ("limit %s expectedly not available for getrlimit\n",
      84  			  tests[i].name);
      85  		  status = 1;
      86  		  this_ok = false;
      87  		}
      88  	    }
      89  	  else
      90  	    {
      91  	      printf ("getrlimit for %s returned unexpected error: %m\n",
      92  		      tests[i].name);
      93  	      status = 1;
      94  	      this_ok = false;
      95  	    }
      96  	}
      97  
      98        struct rlimit64 r64;
      99        res = getrlimit64 (tests[i].resource, &r64);
     100        if (res == -1)
     101  	{
     102  	  if (errno == EINVAL)
     103  	    {
     104  	      if (tests[i].required)
     105  		{
     106  		  printf ("limit %s expectedly not available for getrlimit64"
     107  			  "\n", tests[i].name);
     108  		  status = 1;
     109  		  this_ok = false;
     110  		}
     111  	    }
     112  	  else
     113  	    {
     114  	      printf ("getrlimit64 for %s returned unexpected error: %m\n",
     115  		      tests[i].name);
     116  	      status = 1;
     117  	      this_ok = false;
     118  	    }
     119  	}
     120  
     121        if (this_ok)
     122  	printf ("limit %s OK\n", tests[i].name);
     123      }
     124  
     125    return status;
     126  }
     127  
     128  #define TEST_FUNCTION do_test ()
     129  #include "../test-skeleton.c"