1 //===-- Unittests for getrlimit and setrlimit -----------------------------===//
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7 //===----------------------------------------------------------------------===//
9 #include "src/__support/CPP/string_view.h"
10 #include "src/errno/libc_errno.h"
11 #include "src/fcntl/open.h"
12 #include "src/sys/resource/getrlimit.h"
13 #include "src/sys/resource/setrlimit.h"
14 #include "src/unistd/close.h"
15 #include "src/unistd/unlink.h"
16 #include "test/UnitTest/ErrnoSetterMatcher.h"
17 #include "test/UnitTest/Test.h"
19 #include <sys/resource.h>
22 TEST(LlvmLibcResourceLimitsTest
, SetNoFileLimit
) {
23 using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Fails
;
24 using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds
;
26 // The test strategy is to first create initialize two file descriptors
27 // successfully. Next, close the files and set the file descriptor limit
28 // to 4. This will allow us to open one of those file but not the other.
30 constexpr const char *TEST_FILE1
= "testdata/resource_limits1.test";
31 constexpr const char *TEST_FILE2
= "testdata/resource_limits2.test";
32 LIBC_NAMESPACE::libc_errno
= 0;
34 int fd1
= LIBC_NAMESPACE::open(TEST_FILE1
, O_CREAT
| O_WRONLY
, S_IRWXU
);
36 ASSERT_ERRNO_SUCCESS();
37 int fd2
= LIBC_NAMESPACE::open(TEST_FILE2
, O_CREAT
| O_WRONLY
, S_IRWXU
);
39 ASSERT_ERRNO_SUCCESS();
41 ASSERT_THAT(LIBC_NAMESPACE::close(fd1
), Succeeds(0));
42 ASSERT_THAT(LIBC_NAMESPACE::close(fd2
), Succeeds(0));
44 struct rlimit limits
{
47 ASSERT_THAT(LIBC_NAMESPACE::setrlimit(RLIMIT_NOFILE
, &limits
), Succeeds(0));
49 // One can now only open one of the files successfully.
50 fd1
= LIBC_NAMESPACE::open(TEST_FILE1
, O_RDONLY
);
52 ASSERT_ERRNO_SUCCESS();
53 fd2
= LIBC_NAMESPACE::open(TEST_FILE2
, O_RDONLY
);
55 ASSERT_ERRNO_FAILURE();
57 LIBC_NAMESPACE::libc_errno
= 0;
58 ASSERT_THAT(LIBC_NAMESPACE::close(fd1
), Succeeds(0));
60 fd2
= LIBC_NAMESPACE::open(TEST_FILE2
, O_RDONLY
);
62 ASSERT_ERRNO_SUCCESS();
63 fd1
= LIBC_NAMESPACE::open(TEST_FILE1
, O_RDONLY
);
65 ASSERT_ERRNO_FAILURE();
67 LIBC_NAMESPACE::libc_errno
= 0;
68 ASSERT_THAT(LIBC_NAMESPACE::close(fd2
), Succeeds(0));
70 ASSERT_THAT(LIBC_NAMESPACE::unlink(TEST_FILE1
), Succeeds(0));
71 ASSERT_THAT(LIBC_NAMESPACE::unlink(TEST_FILE2
), Succeeds(0));
73 struct rlimit current_limits
;
74 ASSERT_THAT(LIBC_NAMESPACE::getrlimit(RLIMIT_NOFILE
, ¤t_limits
),
76 ASSERT_EQ(current_limits
.rlim_cur
, rlim_t(4));
77 ASSERT_EQ(current_limits
.rlim_max
, rlim_t(4));