1/* Copyright (C) 2005-2022 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
24static 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
66static int
67do_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 (resource: tests[i].resource, rlimits: &r);
77 if (res == -1)
78 {
79 if (errno == EINVAL)
80 {
81 if (tests[i].required)
82 {
83 printf (format: "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 (format: "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 (format: "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 (format: "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 (format: "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"
130

source code of glibc/resource/tst-getrlimit.c