1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
3 #include <sys/resource.h>
5 #include "alloc-util.h"
6 #include "capability-util.h"
8 #include "missing_resource.h"
9 #include "rlimit-util.h"
10 #include "string-util.h"
12 #include "time-util.h"
14 static void test_rlimit_parse_format_one(int resource
, const char *string
, rlim_t soft
, rlim_t hard
, int ret
, const char *formatted
) {
15 _cleanup_free_
char *f
= NULL
;
24 assert_se(rlimit_parse(resource
, string
, &rl
) == ret
);
28 assert_se(rl
.rlim_cur
== soft
);
29 assert_se(rl
.rlim_max
== hard
);
31 assert_se(rlimit_format(&rl
, &f
) >= 0);
32 assert_se(streq(formatted
, f
));
34 assert_se(rlimit_parse(resource
, formatted
, &rl2
) >= 0);
35 assert_se(memcmp(&rl
, &rl2
, sizeof(struct rlimit
)) == 0);
38 TEST(rlimit_parse_format
) {
39 test_rlimit_parse_format_one(RLIMIT_NOFILE
, "4:5", 4, 5, 0, "4:5");
40 test_rlimit_parse_format_one(RLIMIT_NOFILE
, "6", 6, 6, 0, "6");
41 test_rlimit_parse_format_one(RLIMIT_NOFILE
, "infinity", RLIM_INFINITY
, RLIM_INFINITY
, 0, "infinity");
42 test_rlimit_parse_format_one(RLIMIT_NOFILE
, "infinity:infinity", RLIM_INFINITY
, RLIM_INFINITY
, 0, "infinity");
43 test_rlimit_parse_format_one(RLIMIT_NOFILE
, "8:infinity", 8, RLIM_INFINITY
, 0, "8:infinity");
44 test_rlimit_parse_format_one(RLIMIT_CPU
, "25min:13h", (25*USEC_PER_MINUTE
) / USEC_PER_SEC
, (13*USEC_PER_HOUR
) / USEC_PER_SEC
, 0, "1500:46800");
45 test_rlimit_parse_format_one(RLIMIT_NOFILE
, "", 0, 0, -EINVAL
, NULL
);
46 test_rlimit_parse_format_one(RLIMIT_NOFILE
, "5:4", 0, 0, -EILSEQ
, NULL
);
47 test_rlimit_parse_format_one(RLIMIT_NOFILE
, "5:4:3", 0, 0, -EINVAL
, NULL
);
48 test_rlimit_parse_format_one(RLIMIT_NICE
, "20", 20, 20, 0, "20");
49 test_rlimit_parse_format_one(RLIMIT_NICE
, "40", 40, 40, 0, "40");
50 test_rlimit_parse_format_one(RLIMIT_NICE
, "41", 41, 41, -ERANGE
, "41");
51 test_rlimit_parse_format_one(RLIMIT_NICE
, "0", 0, 0, 0, "0");
52 test_rlimit_parse_format_one(RLIMIT_NICE
, "-7", 27, 27, 0, "27");
53 test_rlimit_parse_format_one(RLIMIT_NICE
, "-20", 40, 40, 0, "40");
54 test_rlimit_parse_format_one(RLIMIT_NICE
, "-21", 41, 41, -ERANGE
, "41");
55 test_rlimit_parse_format_one(RLIMIT_NICE
, "-0", 20, 20, 0, "20");
56 test_rlimit_parse_format_one(RLIMIT_NICE
, "+7", 13, 13, 0, "13");
57 test_rlimit_parse_format_one(RLIMIT_NICE
, "+19", 1, 1, 0, "1");
58 test_rlimit_parse_format_one(RLIMIT_NICE
, "+20", 0, 0, -ERANGE
, "0");
59 test_rlimit_parse_format_one(RLIMIT_NICE
, "+0", 20, 20, 0, "20");
62 TEST(rlimit_from_string
) {
63 assert_se(rlimit_from_string("NOFILE") == RLIMIT_NOFILE
);
64 assert_se(rlimit_from_string("LimitNOFILE") == -EINVAL
);
65 assert_se(rlimit_from_string("RLIMIT_NOFILE") == -EINVAL
);
66 assert_se(rlimit_from_string("xxxNOFILE") == -EINVAL
);
67 assert_se(rlimit_from_string("DefaultLimitNOFILE") == -EINVAL
);
70 TEST(rlimit_from_string_harder
) {
71 assert_se(rlimit_from_string_harder("NOFILE") == RLIMIT_NOFILE
);
72 assert_se(rlimit_from_string_harder("LimitNOFILE") == RLIMIT_NOFILE
);
73 assert_se(rlimit_from_string_harder("RLIMIT_NOFILE") == RLIMIT_NOFILE
);
74 assert_se(rlimit_from_string_harder("xxxNOFILE") == -EINVAL
);
75 assert_se(rlimit_from_string_harder("DefaultLimitNOFILE") == -EINVAL
);
78 TEST(rlimit_to_string_all
) {
79 for (int i
= 0; i
< _RLIMIT_MAX
; i
++) {
80 _cleanup_free_
char *prefixed
= NULL
;
83 assert_se(p
= rlimit_to_string(i
));
84 log_info("%i = %s", i
, p
);
86 assert_se(rlimit_from_string(p
) == i
);
87 assert_se(rlimit_from_string_harder(p
) == i
);
89 assert_se(prefixed
= strjoin("Limit", p
));
91 assert_se(rlimit_from_string(prefixed
) < 0);
92 assert_se(rlimit_from_string_harder(prefixed
) == i
);
94 prefixed
= mfree(prefixed
);
95 assert_se(prefixed
= strjoin("RLIMIT_", p
));
97 assert_se(rlimit_from_string(prefixed
) < 0);
98 assert_se(rlimit_from_string_harder(prefixed
) == i
);
103 struct rlimit old
, new, high
;
104 struct rlimit err
= {
109 assert_se(drop_capability(CAP_SYS_RESOURCE
) == 0);
111 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
112 new.rlim_cur
= MIN(5U, old
.rlim_max
);
113 new.rlim_max
= old
.rlim_max
;
114 assert_se(setrlimit(RLIMIT_NOFILE
, &new) >= 0);
116 assert_se(streq_ptr(rlimit_to_string(RLIMIT_NOFILE
), "NOFILE"));
117 assert_se(rlimit_to_string(-1) == NULL
);
119 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
120 assert_se(setrlimit_closest(RLIMIT_NOFILE
, &old
) == 0);
121 assert_se(getrlimit(RLIMIT_NOFILE
, &new) == 0);
122 assert_se(old
.rlim_cur
== new.rlim_cur
);
123 assert_se(old
.rlim_max
== new.rlim_max
);
125 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
126 high
= RLIMIT_MAKE_CONST(old
.rlim_max
== RLIM_INFINITY
? old
.rlim_max
: old
.rlim_max
+ 1);
127 assert_se(setrlimit_closest(RLIMIT_NOFILE
, &high
) == 0);
128 assert_se(getrlimit(RLIMIT_NOFILE
, &new) == 0);
129 assert_se(new.rlim_max
== old
.rlim_max
);
130 assert_se(new.rlim_cur
== new.rlim_max
);
132 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
133 assert_se(setrlimit_closest(RLIMIT_NOFILE
, &err
) == -EINVAL
);
134 assert_se(getrlimit(RLIMIT_NOFILE
, &new) == 0);
135 assert_se(old
.rlim_cur
== new.rlim_cur
);
136 assert_se(old
.rlim_max
== new.rlim_max
);
139 DEFINE_TEST_MAIN(LOG_INFO
);