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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
#include <cctk.h>
#include <cctk_Parameters.h>
#include <algorithm>
#include <cassert>
#include <iostream>
#include <sys/resource.h>
#include "defs.hh"
#include "limits.hh"
namespace CarpetLib {
using namespace std;
static
void
set_limit (int resource, char const * name, CCTK_INT value);
static
ostream &
operator<< (ostream & s, struct rlimit const & limit);
static
void
output (ostream & s, rlim_t const & value);
void
set_system_limits ()
{
DECLARE_CCTK_PARAMETERS;
set_limit (RLIMIT_CORE, "core file size", max_core_size_MB);
set_limit (RLIMIT_AS, "memory size", max_memory_size_MB);
}
void
set_limit (int const resource, char const * const name, CCTK_INT const value)
{
struct rlimit limit;
check (not getrlimit (resource, & limit));
if (value == -2 ) {
// Only show limit
cout << "Current " << name << " limit: " << limit << endl;
return;
}
cout << "Old " << name << " limit: " << limit << endl;
if (value == -1) {
limit.rlim_cur = limit.rlim_max;
} else {
limit.rlim_cur = min ((rlim_t) value * 1024 * 1024, limit.rlim_max);
}
check (not setrlimit (resource, & limit));
check (not getrlimit (resource, & limit));
cout << "New " << name << " limit: " << limit << endl;
}
static
ostream &
operator<< (ostream & s, struct rlimit const & limit)
{
s << "hard=";
output (s, limit.rlim_max);
s << ", soft=";
output (s, limit.rlim_cur);
return s;
}
static
void
output (ostream & s, rlim_t const & value)
{
if (value == RLIM_INFINITY) {
s << "[unlimited]";
} else {
s << (value / CCTK_REAL (1024*1024)) << " MB";
}
}
} // namespace Carpet
|
