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>

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"