1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
/**
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
* SPDX-License-Identifier: Apache-2.0.
*/
#include <aws/common/process.h>
#include <sys/resource.h>
#include <sys/time.h>
#include <unistd.h>
int aws_get_pid(void) {
return (int)getpid();
}
size_t aws_get_soft_limit_io_handles(void) {
struct rlimit rlimit;
AWS_ZERO_STRUCT(rlimit);
AWS_FATAL_ASSERT(
!getrlimit(RLIMIT_NOFILE, &rlimit) &&
"getrlimit() should never fail for RLIMIT_NOFILE regardless of user permissions");
return rlimit.rlim_cur;
}
size_t aws_get_hard_limit_io_handles(void) {
struct rlimit rlimit;
AWS_ZERO_STRUCT(rlimit);
AWS_FATAL_ASSERT(
!getrlimit(RLIMIT_NOFILE, &rlimit) &&
"getrlimit() should never fail for RLIMIT_NOFILE regardless of user permissions");
return rlimit.rlim_max;
}
int aws_set_soft_limit_io_handles(size_t max_handles) {
size_t hard_limit = aws_get_hard_limit_io_handles();
if (max_handles > hard_limit) {
return aws_raise_error(AWS_ERROR_INVALID_ARGUMENT);
}
struct rlimit rlimit = {
.rlim_cur = max_handles,
.rlim_max = hard_limit,
};
if (setrlimit(RLIMIT_NOFILE, &rlimit)) {
return aws_raise_error(AWS_ERROR_INVALID_ARGUMENT);
}
return AWS_OP_SUCCESS;
}
|
