diff options
Diffstat (limited to 'sysdeps/unix/sysv/linux/tst-rlimit-infinity.c')
| -rw-r--r-- | sysdeps/unix/sysv/linux/tst-rlimit-infinity.c | 173 |
1 files changed, 173 insertions, 0 deletions
diff --git a/sysdeps/unix/sysv/linux/tst-rlimit-infinity.c b/sysdeps/unix/sysv/linux/tst-rlimit-infinity.c new file mode 100644 index 0000000000..9460f6ca83 --- /dev/null +++ b/sysdeps/unix/sysv/linux/tst-rlimit-infinity.c @@ -0,0 +1,173 @@ +/* Copyright (C) 2018 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 + <http://www.gnu.org/licenses/>. */ + +#include <errno.h> +#include <stdio.h> +#include <sys/resource.h> +#include <support/check.h> + +static int resources[] = { + /* The following 7 limits are part of POSIX and must exist. */ + RLIMIT_CORE, + RLIMIT_CPU, + RLIMIT_DATA, + RLIMIT_FSIZE, + RLIMIT_NOFILE, + RLIMIT_STACK, + RLIMIT_AS +}; + +#define nresources (sizeof (resources) / sizeof (resources[0])) + +/* Assume that the prlimit64 function calls the prlimit64 syscall without + mangling the arguments. */ +#define PRLIMIT64_INFINITY ((rlim64_t) -1) + +/* As we don't know which limit will be modified, use a sufficiently high + value to not shoot ourself in the foot. Use a 32-bit value to test + both the 32- and 64-bit versions, and keep the highest bit clear to + avoid sign extension. */ +#define PRLIMIT64_TESTVAL ((rlim64_t) 0x42420000) + +static void +test_getrlimit (int resource, rlim_t exp_cur, rlim_t exp_max) +{ + struct rlimit r; + TEST_VERIFY_EXIT (getrlimit (resource, &r) == 0); + TEST_COMPARE (r.rlim_cur, exp_cur); + TEST_COMPARE (r.rlim_max, exp_max); +} + +static void +test_getrlimit64 (int resource, rlim64_t exp_cur, rlim64_t exp_max) +{ + struct rlimit64 r; + TEST_VERIFY_EXIT (getrlimit64 (resource, &r) == 0); + TEST_COMPARE (r.rlim_cur, exp_cur); + TEST_COMPARE (r.rlim_max, exp_max); +} + +static void +test_prlimit_get (int resource, rlim_t exp_cur, rlim_t exp_max) +{ + struct rlimit r; + TEST_VERIFY_EXIT (prlimit (0, resource, NULL, &r) == 0); + TEST_COMPARE (r.rlim_cur, exp_cur); + TEST_COMPARE (r.rlim_max, exp_max); +} + +static void +test_prlimit64_get (int resource, rlim64_t exp_cur, rlim64_t exp_max) +{ + struct rlimit64 r; + TEST_COMPARE (prlimit64 (0, resource, NULL, &r), 0); + TEST_COMPARE (r.rlim_cur, exp_cur); + TEST_COMPARE (r.rlim_max, exp_max); +} + +static void +test_setrlimit (int resource, rlim_t new_cur, rlim_t new_max) +{ + struct rlimit r = { new_cur, new_max }; + TEST_COMPARE (setrlimit (resource, &r), 0); +} + +static void +test_setrlimit64 (int resource, rlim64_t new_cur, rlim64_t new_max) +{ + struct rlimit64 r = { new_cur, new_max }; + TEST_COMPARE (setrlimit64 (resource, &r), 0); +} + +static void +test_prlimit_set (int resource, rlim_t new_cur, rlim_t new_max) +{ + struct rlimit r = { new_cur, new_max }; + TEST_COMPARE (prlimit (0, resource, &r, NULL), 0); +} + +static void +test_prlimit64_set (int resource, rlim64_t new_cur, rlim64_t new_max) +{ + struct rlimit64 r = { new_cur, new_max }; + TEST_COMPARE (prlimit64 (0, resource, &r, NULL), 0); +} + +static int +do_test (void) +{ + int resource = -1; + + /* Find a resource with hard limit set to infinity, so that the soft limit + can be manipulated to any value. */ + for (int i = 0; i < nresources; ++i) + { + struct rlimit64 r64; + int res = prlimit64 (0, resources[i], NULL, &r64); + if ((res == 0) && (r64.rlim_max == PRLIMIT64_INFINITY)) + { + resource = resources[i]; + break; + } + } + + if (resource == -1) + FAIL_UNSUPPORTED + ("Could not find and limit with hard limit set to infinity."); + + /* First check that the get functions work correctly with the test value. */ + test_prlimit64_set (resource, PRLIMIT64_TESTVAL, PRLIMIT64_INFINITY); + test_getrlimit (resource, PRLIMIT64_TESTVAL, RLIM_INFINITY); + test_getrlimit64 (resource, PRLIMIT64_TESTVAL, RLIM64_INFINITY); + test_prlimit_get (resource, PRLIMIT64_TESTVAL, RLIM_INFINITY); + test_prlimit64_get (resource, PRLIMIT64_TESTVAL, RLIM64_INFINITY); + + /* Then check that the get functions work correctly with infinity. */ + test_prlimit64_set (resource, PRLIMIT64_INFINITY, PRLIMIT64_INFINITY); + test_getrlimit (resource, RLIM_INFINITY, RLIM_INFINITY); + test_getrlimit64 (resource, RLIM64_INFINITY, RLIM64_INFINITY); + test_prlimit_get (resource, RLIM_INFINITY, RLIM_INFINITY); + test_prlimit64_get (resource, RLIM64_INFINITY, RLIM64_INFINITY); + + /* Then check that setrlimit works correctly with the test value. */ + test_setrlimit (resource, PRLIMIT64_TESTVAL, RLIM_INFINITY); + test_prlimit64_get (resource, PRLIMIT64_TESTVAL, PRLIMIT64_INFINITY); + + /* Then check that setrlimit works correctly with infinity. */ + test_setrlimit (resource, RLIM_INFINITY, RLIM_INFINITY); + test_prlimit64_get (resource, PRLIMIT64_INFINITY, PRLIMIT64_INFINITY); + + /* Then check that setrlimit64 works correctly with the test value. */ + test_setrlimit64 (resource, PRLIMIT64_TESTVAL, RLIM64_INFINITY); + test_prlimit64_get (resource, PRLIMIT64_TESTVAL, PRLIMIT64_INFINITY); + + /* Then check that setrlimit64 works correctly with infinity. */ + test_setrlimit64 (resource, RLIM64_INFINITY, RLIM64_INFINITY); + test_prlimit64_get (resource, PRLIMIT64_INFINITY, PRLIMIT64_INFINITY); + + /* Then check that prlimit works correctly with the test value. */ + test_prlimit_set (resource, RLIM_INFINITY, RLIM_INFINITY); + test_prlimit64_get (resource, PRLIMIT64_INFINITY, PRLIMIT64_INFINITY); + + /* Finally check that prlimit works correctly with infinity. */ + test_prlimit_set (resource, PRLIMIT64_TESTVAL, RLIM_INFINITY); + test_prlimit64_get (resource, PRLIMIT64_TESTVAL, PRLIMIT64_INFINITY); + + return 0; +} + +#include <support/test-driver.c> |