]>
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
5 systemd is free software; you can redistribute it and/or modify it
6 under the terms of the GNU Lesser General Public License as published by
7 the Free Software Foundation; either version 2.1 of the License, or
8 (at your option) any later version.
10 systemd is distributed in the hope that it will be useful, but
11 WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 Lesser General Public License for more details.
15 You should have received a copy of the GNU Lesser General Public License
16 along with systemd; If not, see <http://www.gnu.org/licenses/>.
19 #include <sys/resource.h>
21 #include "alloc-util.h"
22 #include "capability-util.h"
24 #include "rlimit-util.h"
25 #include "string-util.h"
28 static void test_rlimit_parse_format(int resource
, const char *string
, rlim_t soft
, rlim_t hard
, int ret
, const char *formatted
) {
29 _cleanup_free_
char *f
= NULL
;
38 assert_se(rlimit_parse(resource
, string
, &rl
) == ret
);
42 assert_se(rl
.rlim_cur
== soft
);
43 assert_se(rl
.rlim_max
== hard
);
45 assert_se(rlimit_format(&rl
, &f
) >= 0);
46 assert_se(streq(formatted
, f
));
48 assert_se(rlimit_parse(resource
, formatted
, &rl2
) >= 0);
49 assert_se(memcmp(&rl
, &rl2
, sizeof(struct rlimit
)) == 0);
52 int main(int argc
, char *argv
[]) {
53 struct rlimit old
, new, high
;
59 log_parse_environment();
62 assert_se(drop_capability(CAP_SYS_RESOURCE
) == 0);
64 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
65 new.rlim_cur
= MIN(5U, old
.rlim_max
);
66 new.rlim_max
= old
.rlim_max
;
67 assert_se(setrlimit(RLIMIT_NOFILE
, &new) >= 0);
69 assert_se(rlimit_from_string("LimitNOFILE") == RLIMIT_NOFILE
);
70 assert_se(rlimit_from_string("DefaultLimitNOFILE") == -1);
72 assert_se(streq_ptr(rlimit_to_string(RLIMIT_NOFILE
), "LimitNOFILE"));
73 assert_se(rlimit_to_string(-1) == NULL
);
75 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
76 assert_se(setrlimit_closest(RLIMIT_NOFILE
, &old
) == 0);
77 assert_se(getrlimit(RLIMIT_NOFILE
, &new) == 0);
78 assert_se(old
.rlim_cur
== new.rlim_cur
);
79 assert_se(old
.rlim_max
== new.rlim_max
);
81 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
82 high
= RLIMIT_MAKE_CONST(old
.rlim_max
== RLIM_INFINITY
? old
.rlim_max
: old
.rlim_max
+ 1);
83 assert_se(setrlimit_closest(RLIMIT_NOFILE
, &high
) == 0);
84 assert_se(getrlimit(RLIMIT_NOFILE
, &new) == 0);
85 assert_se(new.rlim_max
== old
.rlim_max
);
86 assert_se(new.rlim_cur
== new.rlim_max
);
88 assert_se(getrlimit(RLIMIT_NOFILE
, &old
) == 0);
89 assert_se(setrlimit_closest(RLIMIT_NOFILE
, &err
) == -EINVAL
);
90 assert_se(getrlimit(RLIMIT_NOFILE
, &new) == 0);
91 assert_se(old
.rlim_cur
== new.rlim_cur
);
92 assert_se(old
.rlim_max
== new.rlim_max
);
94 test_rlimit_parse_format(RLIMIT_NOFILE
, "4:5", 4, 5, 0, "4:5");
95 test_rlimit_parse_format(RLIMIT_NOFILE
, "6", 6, 6, 0, "6");
96 test_rlimit_parse_format(RLIMIT_NOFILE
, "infinity", RLIM_INFINITY
, RLIM_INFINITY
, 0, "infinity");
97 test_rlimit_parse_format(RLIMIT_NOFILE
, "infinity:infinity", RLIM_INFINITY
, RLIM_INFINITY
, 0, "infinity");
98 test_rlimit_parse_format(RLIMIT_NOFILE
, "8:infinity", 8, RLIM_INFINITY
, 0, "8:infinity");
99 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");
100 test_rlimit_parse_format(RLIMIT_NOFILE
, "", 0, 0, -EINVAL
, NULL
);
101 test_rlimit_parse_format(RLIMIT_NOFILE
, "5:4", 0, 0, -EILSEQ
, NULL
);
102 test_rlimit_parse_format(RLIMIT_NOFILE
, "5:4:3", 0, 0, -EINVAL
, NULL
);
103 test_rlimit_parse_format(RLIMIT_NICE
, "20", 20, 20, 0, "20");
104 test_rlimit_parse_format(RLIMIT_NICE
, "40", 40, 40, 0, "40");
105 test_rlimit_parse_format(RLIMIT_NICE
, "41", 41, 41, -ERANGE
, "41");
106 test_rlimit_parse_format(RLIMIT_NICE
, "0", 0, 0, 0, "0");
107 test_rlimit_parse_format(RLIMIT_NICE
, "-7", 27, 27, 0, "27");
108 test_rlimit_parse_format(RLIMIT_NICE
, "-20", 40, 40, 0, "40");
109 test_rlimit_parse_format(RLIMIT_NICE
, "-21", 41, 41, -ERANGE
, "41");
110 test_rlimit_parse_format(RLIMIT_NICE
, "-0", 20, 20, 0, "20");
111 test_rlimit_parse_format(RLIMIT_NICE
, "+7", 13, 13, 0, "13");
112 test_rlimit_parse_format(RLIMIT_NICE
, "+19", 1, 1, 0, "1");
113 test_rlimit_parse_format(RLIMIT_NICE
, "+20", 0, 0, -ERANGE
, "0");
114 test_rlimit_parse_format(RLIMIT_NICE
, "+0", 20, 20, 0, "20");