summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorPavel Kozlov <pavel.kozlov@synopsys.com>2023-11-07 14:59:39 +0400
committerWaldemar Brodkorb <wbx@openadk.org>2023-11-07 13:43:24 +0100
commitd29d9e9257891f6ed86bc1e48632eade495915b5 (patch)
tree29b3b0466ab4a2dab87178358fd621241e718a93 /test
parent24d21cfbcdbc738074b13236cd296599d7df6a7b (diff)
misc: add tests for getrlimit/setrlimit/prlimit
Add tst-rlimit test to check setrlimit/getrlimit/prlimit and setrlimit64/getrlimit64/prlimit64 functions. Add tst-rlimit64 to check the 64-bit offset variant. Signed-off-by: Pavel Kozlov <pavel.kozlov@synopsys.com>
Diffstat (limited to 'test')
-rw-r--r--test/misc/Makefile.in1
-rw-r--r--test/misc/tst-rlimit.c157
-rw-r--r--test/misc/tst-rlimit64.c1
3 files changed, 159 insertions, 0 deletions
diff --git a/test/misc/Makefile.in b/test/misc/Makefile.in
index a97b124..96c1fd9 100644
--- a/test/misc/Makefile.in
+++ b/test/misc/Makefile.in
@@ -8,6 +8,7 @@ TESTS_DISABLED += tst-inotify
endif
CFLAGS_dirent64 := -D_LARGEFILE_SOURCE -D_LARGEFILE64_SOURCE -D_FILE_OFFSET_BITS=64
+CFLAGS_tst-rlimit64 := -D_LARGEFILE_SOURCE -D_LARGEFILE64_SOURCE -D_FILE_OFFSET_BITS=64
DODIFF_dirent := 1
DODIFF_dirent64 := 1
diff --git a/test/misc/tst-rlimit.c b/test/misc/tst-rlimit.c
new file mode 100644
index 0000000..84aa465
--- /dev/null
+++ b/test/misc/tst-rlimit.c
@@ -0,0 +1,157 @@
+/*
+ * setrlimit/getrlimit/prlimit and setrlimit64/getrlimit64/prlimit64 functions
+ * test for uClibc.
+ *
+ * The prlimit64 function is not called directly in this test, because it is a
+ * new function for uclibc and can cause build problems with uclibc-ng <= 1.0.44.
+ * With _FILE_OFFSET_BITS == 64 the prlimit call is redirected to the prlimit64
+ * call.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <errno.h>
+#include <sys/resource.h>
+
+#define __ASSERT(x, f, l) \
+ if (!(x)) { \
+ fprintf(stderr, "%s: LINE %d: ASSERT: " #x "\n", f, l); \
+ exit(1); \
+ }
+
+#define ASSERT(x) __ASSERT(x, __FILE__, __LINE__)
+
+static int resources[] = {
+ RLIMIT_CORE,
+ RLIMIT_CPU,
+ RLIMIT_DATA,
+ RLIMIT_FSIZE,
+ RLIMIT_NOFILE,
+ RLIMIT_STACK,
+ RLIMIT_AS
+};
+
+#define nresources (sizeof (resources) / sizeof (resources[0]))
+#define TEST_VAL 0x11223344
+
+
+static void test_getrlimit(int rnum, rlim_t exp_cur, rlim_t exp_max)
+{
+ struct rlimit r;
+
+ ASSERT(getrlimit(rnum, &r) == 0);
+ ASSERT(r.rlim_cur == exp_cur);
+ ASSERT(r.rlim_max == exp_max);
+}
+
+static void test_setrlimit(int rnum, rlim_t new_cur, rlim_t new_max)
+{
+ struct rlimit r = {.rlim_cur = new_cur, .rlim_max = new_max};
+
+ ASSERT(setrlimit(rnum, &r) == 0);
+}
+
+static void test_prlimit_get(int rnum, rlim_t exp_cur, rlim_t exp_max)
+{
+ struct rlimit r;
+
+ ASSERT(prlimit(0, rnum, NULL, &r) == 0);
+ ASSERT(r.rlim_cur == exp_cur);
+ ASSERT(r.rlim_max == exp_max);
+}
+
+static void test_prlimit_set(int rnum, rlim_t new_cur, rlim_t new_max)
+{
+ struct rlimit r = {.rlim_cur = new_cur, .rlim_max = new_max};
+ ASSERT(prlimit(0, rnum, &r, NULL) == 0);
+}
+
+#if defined(__USE_LARGEFILE64)
+static void test_getrlimit64(int rnum, rlim64_t exp_cur, rlim64_t exp_max)
+{
+ struct rlimit64 r;
+
+ ASSERT(getrlimit64(rnum, &r) == 0);
+ ASSERT(r.rlim_cur == exp_cur);
+ ASSERT(r.rlim_max == exp_max);
+}
+
+static void test_setrlimit64(int rnum, rlim64_t new_cur, rlim64_t new_max)
+{
+ struct rlimit64 r = {.rlim_cur = new_cur, .rlim_max = new_max};
+
+ ASSERT(setrlimit64(rnum, &r) == 0);
+}
+#endif
+
+int main(void)
+{
+ int rnum = -1;
+ struct rlimit rlim;
+ int i, ret;
+
+ /* Find a resource with hard limit set to infinity */
+ for (i = 0; i < nresources; ++i) {
+ ret = getrlimit(resources[i], &rlim);
+ if ((!ret) && (rlim.rlim_max == RLIM_INFINITY)) {
+ rnum = resources[i];
+ break;
+ }
+ }
+
+ /* Can't continue, return unsupported */
+ if (rnum == -1)
+ return 23;
+
+ /* Test cases */
+ test_setrlimit(rnum, TEST_VAL, RLIM_INFINITY);
+ test_getrlimit(rnum, TEST_VAL, RLIM_INFINITY);
+ test_setrlimit(rnum, RLIM_INFINITY, RLIM_INFINITY);
+ test_getrlimit(rnum, RLIM_INFINITY, RLIM_INFINITY);
+ test_prlimit_get(rnum, RLIM_INFINITY, RLIM_INFINITY);
+ test_prlimit_set(rnum, TEST_VAL, RLIM_INFINITY);
+ test_prlimit_get(rnum, TEST_VAL, RLIM_INFINITY);
+ test_getrlimit(rnum, TEST_VAL, RLIM_INFINITY);
+ test_prlimit_set(rnum, RLIM_INFINITY, RLIM_INFINITY);
+ test_getrlimit(rnum, RLIM_INFINITY, RLIM_INFINITY);
+ test_prlimit_get(rnum, RLIM_INFINITY, RLIM_INFINITY);
+
+#if defined(__USE_LARGEFILE64)
+ /* Check setrlim64 and getrlim64 in 32-bit offset LFS environment */
+ test_setrlimit64(rnum, TEST_VAL, RLIM64_INFINITY);
+ test_getrlimit(rnum, TEST_VAL, RLIM_INFINITY);
+ test_getrlimit64(rnum, TEST_VAL, RLIM64_INFINITY);
+ test_setrlimit64(rnum, RLIM64_INFINITY, RLIM64_INFINITY);
+ test_prlimit_get(rnum, RLIM_INFINITY, RLIM_INFINITY);
+ test_getrlimit64(rnum, RLIM64_INFINITY, RLIM64_INFINITY);
+
+ test_setrlimit(rnum, TEST_VAL, RLIM_INFINITY);
+ test_getrlimit64(rnum, TEST_VAL, RLIM64_INFINITY);
+ test_prlimit_set(rnum, RLIM_INFINITY, RLIM_INFINITY);
+ test_getrlimit64(rnum, RLIM64_INFINITY, RLIM64_INFINITY);
+#endif
+
+
+#if defined(__USE_LARGEFILE64) && _FILE_OFFSET_BITS == 64
+ /* Check setrlim64/getrlim64/prlimit64 in 64-bit offset environment */
+ test_setrlimit64(rnum, TEST_VAL, RLIM64_INFINITY);
+ test_getrlimit64(rnum, TEST_VAL, RLIM64_INFINITY);
+ test_getrlimit(rnum, TEST_VAL, RLIM_INFINITY);
+ test_setrlimit64(rnum, RLIM64_INFINITY, RLIM64_INFINITY);
+ test_prlimit_get(rnum, RLIM64_INFINITY, RLIM64_INFINITY);
+ test_getrlimit(rnum, RLIM_INFINITY, RLIM_INFINITY);
+ test_prlimit_set(rnum, TEST_VAL, RLIM64_INFINITY);
+ test_prlimit_get(rnum, TEST_VAL, RLIM64_INFINITY);
+ test_prlimit_set(rnum, RLIM64_INFINITY, RLIM64_INFINITY);
+ test_getrlimit64(rnum, RLIM64_INFINITY, RLIM64_INFINITY);
+ test_getrlimit(rnum, RLIM_INFINITY, RLIM_INFINITY);
+
+ test_setrlimit(rnum, TEST_VAL, RLIM_INFINITY);
+ test_getrlimit64(rnum, TEST_VAL, RLIM64_INFINITY);
+ test_prlimit_set(rnum, RLIM_INFINITY, RLIM_INFINITY);
+ test_getrlimit64(rnum, RLIM64_INFINITY, RLIM64_INFINITY);
+#endif
+
+ return 0;
+}
diff --git a/test/misc/tst-rlimit64.c b/test/misc/tst-rlimit64.c
new file mode 100644
index 0000000..490c6c5
--- /dev/null
+++ b/test/misc/tst-rlimit64.c
@@ -0,0 +1 @@
+#include "tst-rlimit.c"