summary refs log tree commit diff
path: root/resource/tst-getrlimit.c
blob: 0b6a4a69f4bbdb426844c3a2795d8b3fd7cf8170 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/* Copyright (C) 2005-2023 Free Software Foundation, Inc.
   This file is part of the GNU C Library.

   The GNU C Library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 2.1 of the License, or (at your option) any later version.

   The GNU C Library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with the GNU C Library; if not, see
   <https://www.gnu.org/licenses/>.  */

#include <errno.h>
#include <stdbool.h>
#include <stdio.h>
#include <sys/resource.h>


static struct
{
  const char *name;
  int resource;
  bool required;
} tests[] =
  {
    /* The following 7 limits are part of POSIX and must exist.  */
    { "RLIMIT_CORE", RLIMIT_CORE, true },
    { "RLIMIT_CPU", RLIMIT_CPU, true },
    { "RLIMIT_DATA", RLIMIT_DATA, true },
    { "RLIMIT_FSIZE", RLIMIT_FSIZE, true },
    { "RLIMIT_NOFILE", RLIMIT_NOFILE, true },
    { "RLIMIT_STACK", RLIMIT_STACK, true },
    { "RLIMIT_AS", RLIMIT_AS, true },
    /* The following are traditional Unix limits which are also
       expected (by us).  */
    { "RLIMIT_RSS", RLIMIT_RSS, true },
    { "RLIMIT_NPROC", RLIMIT_NPROC, true },
    /* The following are extensions.  */
#ifdef RLIMIT_MEMLOCK
    { "RLIMIT_MEMLOCK", RLIMIT_MEMLOCK, false },
#endif
#ifdef RLIMIT_LOCKS
    { "RLIMIT_LOCKS", RLIMIT_LOCKS, false },
#endif
#ifdef RLIMIT_SIGPENDING
    { "RLIMIT_SIGPENDING", RLIMIT_SIGPENDING, false },
#endif
#ifdef RLIMIT_MSGQUEUE
    { "RLIMIT_MSGQUEUE", RLIMIT_MSGQUEUE, false },
#endif
#ifdef RLIMIT_NICE
    { "RLIMIT_NICE", RLIMIT_NICE, false },
#endif
#ifdef RLIMIT_RTPRIO
    { "RLIMIT_RTPRIO", RLIMIT_RTPRIO, false },
#endif
  };
#define ntests (sizeof (tests) / sizeof (tests[0]))


static int
do_test (void)
{
  int status = 0;

  for (int i = 0; i < ntests; ++i)
    {
      bool this_ok = true;

      struct rlimit r;
      int res = getrlimit (tests[i].resource, &r);
      if (res == -1)
	{
	  if (errno == EINVAL)
	    {
	      if (tests[i].required)
		{
		  printf ("limit %s expectedly not available for getrlimit\n",
			  tests[i].name);
		  status = 1;
		  this_ok = false;
		}
	    }
	  else
	    {
	      printf ("getrlimit for %s returned unexpected error: %m\n",
		      tests[i].name);
	      status = 1;
	      this_ok = false;
	    }
	}

      struct rlimit64 r64;
      res = getrlimit64 (tests[i].resource, &r64);
      if (res == -1)
	{
	  if (errno == EINVAL)
	    {
	      if (tests[i].required)
		{
		  printf ("limit %s expectedly not available for getrlimit64"
			  "\n", tests[i].name);
		  status = 1;
		  this_ok = false;
		}
	    }
	  else
	    {
	      printf ("getrlimit64 for %s returned unexpected error: %m\n",
		      tests[i].name);
	      status = 1;
	      this_ok = false;
	    }
	}

      if (this_ok)
	printf ("limit %s OK\n", tests[i].name);
    }

  return status;
}

#define TEST_FUNCTION do_test ()
#include "../test-skeleton.c"