]>
git.ipfire.org Git - thirdparty/systemd.git/blob - src/test/test-rlimit-util.c
1 /* SPDX-License-Identifier: LGPL-2.1+ */
3 This file is part of systemd
6 #include <sys/resource.h>
8 #include "alloc-util.h"
9 #include "capability-util.h"
11 #include "rlimit-util.h"
12 #include "string-util.h"
15 static void test_rlimit_parse_format(int resource
, const char *string
, rlim_t soft
, rlim_t hard
, int ret
, const char *formatted
) {
16 _cleanup_free_
char *f
= NULL
;
25 assert_se(rlimit_parse(resource
, string
, &rl
) == ret
);
29 assert_se(rl
.rlim_cur
== soft
);
30 assert_se(rl
.rlim_max
== hard
);
32 assert_se(rlimit_format(&rl
, &f
) >= 0);
33 assert_se(streq(formatted
, f
));
35 assert_se(rlimit_parse(resource
, formatted
, &rl2
) >= 0);
36 assert_se(memcmp(&rl
, &rl2
, sizeof(struct rlimit
)) == 0);
39 int main(int argc
, char *argv
[]) {
40 struct rlimit old
, new, high
;
47 log_parse_environment();
50 assert_se(drop_capability(CAP_SYS_RESOURCE
) == 0);
52 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
53 new.rlim_cur
= MIN(5U, old
.rlim_max
);
54 new.rlim_max
= old
.rlim_max
;
55 assert_se(setrlimit(RLIMIT_NOFILE
, &new) >= 0);
57 assert_se(rlimit_from_string("NOFILE") == RLIMIT_NOFILE
);
58 assert_se(rlimit_from_string("LimitNOFILE") == -1);
59 assert_se(rlimit_from_string("RLIMIT_NOFILE") == -1);
60 assert_se(rlimit_from_string("xxxNOFILE") == -1);
61 assert_se(rlimit_from_string("DefaultLimitNOFILE") == -1);
63 assert_se(rlimit_from_string_harder("NOFILE") == RLIMIT_NOFILE
);
64 assert_se(rlimit_from_string_harder("LimitNOFILE") == RLIMIT_NOFILE
);
65 assert_se(rlimit_from_string_harder("RLIMIT_NOFILE") == RLIMIT_NOFILE
);
66 assert_se(rlimit_from_string_harder("xxxNOFILE") == -1);
67 assert_se(rlimit_from_string_harder("DefaultLimitNOFILE") == -1);
69 for (i
= 0; i
< _RLIMIT_MAX
; i
++) {
70 _cleanup_free_
char *prefixed
= NULL
;
73 assert_se(p
= rlimit_to_string(i
));
74 log_info("%i = %s", i
, p
);
76 assert_se(rlimit_from_string(p
) == i
);
77 assert_se(rlimit_from_string_harder(p
) == i
);
79 assert_se(prefixed
= strjoin("Limit", p
));
81 assert_se(rlimit_from_string(prefixed
) < 0);
82 assert_se(rlimit_from_string_harder(prefixed
) == i
);
84 prefixed
= mfree(prefixed
);
85 assert_se(prefixed
= strjoin("RLIMIT_", p
));
87 assert_se(rlimit_from_string(prefixed
) < 0);
88 assert_se(rlimit_from_string_harder(prefixed
) == i
);
91 assert_se(streq_ptr(rlimit_to_string(RLIMIT_NOFILE
), "NOFILE"));
92 assert_se(rlimit_to_string(-1) == NULL
);
94 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
95 assert_se(setrlimit_closest(RLIMIT_NOFILE
, &old
) == 0);
96 assert_se(getrlimit(RLIMIT_NOFILE
, &new) == 0);
97 assert_se(old
.rlim_cur
== new.rlim_cur
);
98 assert_se(old
.rlim_max
== new.rlim_max
);
100 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
101 high
= RLIMIT_MAKE_CONST(old
.rlim_max
== RLIM_INFINITY
? old
.rlim_max
: old
.rlim_max
+ 1);
102 assert_se(setrlimit_closest(RLIMIT_NOFILE
, &high
) == 0);
103 assert_se(getrlimit(RLIMIT_NOFILE
, &new) == 0);
104 assert_se(new.rlim_max
== old
.rlim_max
);
105 assert_se(new.rlim_cur
== new.rlim_max
);
107 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
108 assert_se(setrlimit_closest(RLIMIT_NOFILE
, &err
) == -EINVAL
);
109 assert_se(getrlimit(RLIMIT_NOFILE
, &new) == 0);
110 assert_se(old
.rlim_cur
== new.rlim_cur
);
111 assert_se(old
.rlim_max
== new.rlim_max
);
113 test_rlimit_parse_format(RLIMIT_NOFILE
, "4:5", 4, 5, 0, "4:5");
114 test_rlimit_parse_format(RLIMIT_NOFILE
, "6", 6, 6, 0, "6");
115 test_rlimit_parse_format(RLIMIT_NOFILE
, "infinity", RLIM_INFINITY
, RLIM_INFINITY
, 0, "infinity");
116 test_rlimit_parse_format(RLIMIT_NOFILE
, "infinity:infinity", RLIM_INFINITY
, RLIM_INFINITY
, 0, "infinity");
117 test_rlimit_parse_format(RLIMIT_NOFILE
, "8:infinity", 8, RLIM_INFINITY
, 0, "8:infinity");
118 test_rlimit_parse_format(RLIMIT_CPU
, "25min:13h", (25*USEC_PER_MINUTE
) / USEC_PER_SEC
, (13*USEC_PER_HOUR
) / USEC_PER_SEC
, 0, "1500:46800");
119 test_rlimit_parse_format(RLIMIT_NOFILE
, "", 0, 0, -EINVAL
, NULL
);
120 test_rlimit_parse_format(RLIMIT_NOFILE
, "5:4", 0, 0, -EILSEQ
, NULL
);
121 test_rlimit_parse_format(RLIMIT_NOFILE
, "5:4:3", 0, 0, -EINVAL
, NULL
);
122 test_rlimit_parse_format(RLIMIT_NICE
, "20", 20, 20, 0, "20");
123 test_rlimit_parse_format(RLIMIT_NICE
, "40", 40, 40, 0, "40");
124 test_rlimit_parse_format(RLIMIT_NICE
, "41", 41, 41, -ERANGE
, "41");
125 test_rlimit_parse_format(RLIMIT_NICE
, "0", 0, 0, 0, "0");
126 test_rlimit_parse_format(RLIMIT_NICE
, "-7", 27, 27, 0, "27");
127 test_rlimit_parse_format(RLIMIT_NICE
, "-20", 40, 40, 0, "40");
128 test_rlimit_parse_format(RLIMIT_NICE
, "-21", 41, 41, -ERANGE
, "41");
129 test_rlimit_parse_format(RLIMIT_NICE
, "-0", 20, 20, 0, "20");
130 test_rlimit_parse_format(RLIMIT_NICE
, "+7", 13, 13, 0, "13");
131 test_rlimit_parse_format(RLIMIT_NICE
, "+19", 1, 1, 0, "1");
132 test_rlimit_parse_format(RLIMIT_NICE
, "+20", 0, 0, -ERANGE
, "0");
133 test_rlimit_parse_format(RLIMIT_NICE
, "+0", 20, 20, 0, "20");