|
@@ -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;
|
|
|
+}
|