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
|
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "sandbox/linux/services/resource_limits.h"
#include "base/numerics/safe_math.h"
#include <sys/resource.h>
#include <sys/time.h>
#include <errno.h>
#include <algorithm>
namespace sandbox {
// static
int ResourceLimits::Lower(int resource, rlim_t limit) {
return LowerSoftAndHardLimits(resource, limit, limit);
}
// static
int ResourceLimits::LowerSoftAndHardLimits(int resource,
rlim_t soft_limit,
rlim_t hard_limit) {
struct rlimit old_rlimit;
if (getrlimit(resource, &old_rlimit))
return errno;
// Make sure we don't raise the existing limit.
const struct rlimit new_rlimit = {std::min(old_rlimit.rlim_cur, soft_limit),
std::min(old_rlimit.rlim_max, hard_limit)};
return setrlimit(resource, &new_rlimit) ? errno : 0;
}
// static
int ResourceLimits::AdjustCurrent(int resource, long long int change) {
struct rlimit old_rlimit;
if (getrlimit(resource, &old_rlimit))
return errno;
base::CheckedNumeric<rlim_t> checked_limit = old_rlimit.rlim_cur;
checked_limit += change;
const rlim_t new_limit = checked_limit.ValueOrDefault(old_rlimit.rlim_max);
const struct rlimit new_rlimit = {std::min(new_limit, old_rlimit.rlim_max),
old_rlimit.rlim_max};
// setrlimit will fail if limit > old_rlimit.rlim_max.
return setrlimit(resource, &new_rlimit) ? errno : 0;
}
} // namespace sandbox
|