Blame view
kernel/res_counter.c
4.5 KB
e552b6617 Memory controller... |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
/* * resource cgroups * * Copyright 2007 OpenVZ SWsoft Inc * * Author: Pavel Emelianov <xemul@openvz.org> * */ #include <linux/types.h> #include <linux/parser.h> #include <linux/fs.h> #include <linux/res_counter.h> #include <linux/uaccess.h> |
856c13aa1 cgroup files: con... |
15 |
#include <linux/mm.h> |
e552b6617 Memory controller... |
16 |
|
28dbc4b6a memcg: memory cgr... |
17 |
void res_counter_init(struct res_counter *counter, struct res_counter *parent) |
e552b6617 Memory controller... |
18 19 |
{ spin_lock_init(&counter->lock); |
c5b947b28 memcg: add interf... |
20 |
counter->limit = RESOURCE_MAX; |
296c81d89 memory controller... |
21 |
counter->soft_limit = RESOURCE_MAX; |
28dbc4b6a memcg: memory cgr... |
22 |
counter->parent = parent; |
e552b6617 Memory controller... |
23 |
} |
4d8438f04 res_counter: Merg... |
24 25 |
int res_counter_charge_locked(struct res_counter *counter, unsigned long val, bool force) |
e552b6617 Memory controller... |
26 |
{ |
4d8438f04 res_counter: Merg... |
27 |
int ret = 0; |
e552b6617 Memory controller... |
28 29 |
if (counter->usage + val > counter->limit) { counter->failcnt++; |
4d8438f04 res_counter: Merg... |
30 31 32 |
ret = -ENOMEM; if (!force) return ret; |
e552b6617 Memory controller... |
33 34 35 |
} counter->usage += val; |
0d4dde1ac res_counter: Acco... |
36 |
if (counter->usage > counter->max_usage) |
c84872e16 memcgroup: add th... |
37 |
counter->max_usage = counter->usage; |
4d8438f04 res_counter: Merg... |
38 |
return ret; |
e552b6617 Memory controller... |
39 |
} |
4d8438f04 res_counter: Merg... |
40 41 |
static int __res_counter_charge(struct res_counter *counter, unsigned long val, struct res_counter **limit_fail_at, bool force) |
e552b6617 Memory controller... |
42 |
{ |
4d8438f04 res_counter: Merg... |
43 |
int ret, r; |
e552b6617 Memory controller... |
44 |
unsigned long flags; |
28dbc4b6a memcg: memory cgr... |
45 |
struct res_counter *c, *u; |
e552b6617 Memory controller... |
46 |
|
4d8438f04 res_counter: Merg... |
47 |
r = ret = 0; |
28dbc4b6a memcg: memory cgr... |
48 49 50 51 |
*limit_fail_at = NULL; local_irq_save(flags); for (c = counter; c != NULL; c = c->parent) { spin_lock(&c->lock); |
4d8438f04 res_counter: Merg... |
52 |
r = res_counter_charge_locked(c, val, force); |
28dbc4b6a memcg: memory cgr... |
53 |
spin_unlock(&c->lock); |
4d8438f04 res_counter: Merg... |
54 55 |
if (r < 0 && !ret) { ret = r; |
28dbc4b6a memcg: memory cgr... |
56 |
*limit_fail_at = c; |
4d8438f04 res_counter: Merg... |
57 58 |
if (!force) break; |
28dbc4b6a memcg: memory cgr... |
59 60 |
} } |
4d8438f04 res_counter: Merg... |
61 62 63 64 65 66 67 |
if (ret < 0 && !force) { for (u = counter; u != c; u = u->parent) { spin_lock(&u->lock); res_counter_uncharge_locked(u, val); spin_unlock(&u->lock); } |
28dbc4b6a memcg: memory cgr... |
68 |
} |
28dbc4b6a memcg: memory cgr... |
69 |
local_irq_restore(flags); |
4d8438f04 res_counter: Merg... |
70 |
|
e552b6617 Memory controller... |
71 72 |
return ret; } |
4d8438f04 res_counter: Merg... |
73 74 75 76 77 |
int res_counter_charge(struct res_counter *counter, unsigned long val, struct res_counter **limit_fail_at) { return __res_counter_charge(counter, val, limit_fail_at, false); } |
0e90b31f4 net: introduce re... |
78 79 80 |
int res_counter_charge_nofail(struct res_counter *counter, unsigned long val, struct res_counter **limit_fail_at) { |
4d8438f04 res_counter: Merg... |
81 |
return __res_counter_charge(counter, val, limit_fail_at, true); |
0e90b31f4 net: introduce re... |
82 |
} |
4d8438f04 res_counter: Merg... |
83 |
|
e552b6617 Memory controller... |
84 85 86 87 88 89 90 |
void res_counter_uncharge_locked(struct res_counter *counter, unsigned long val) { if (WARN_ON(counter->usage < val)) val = counter->usage; counter->usage -= val; } |
4e649152c memcg: some modif... |
91 |
void res_counter_uncharge(struct res_counter *counter, unsigned long val) |
e552b6617 Memory controller... |
92 93 |
{ unsigned long flags; |
28dbc4b6a memcg: memory cgr... |
94 |
struct res_counter *c; |
e552b6617 Memory controller... |
95 |
|
28dbc4b6a memcg: memory cgr... |
96 97 98 99 100 101 102 |
local_irq_save(flags); for (c = counter; c != NULL; c = c->parent) { spin_lock(&c->lock); res_counter_uncharge_locked(c, val); spin_unlock(&c->lock); } local_irq_restore(flags); |
e552b6617 Memory controller... |
103 |
} |
0eea10301 Memory controller... |
104 105 |
static inline unsigned long long * res_counter_member(struct res_counter *counter, int member) |
e552b6617 Memory controller... |
106 107 108 109 |
{ switch (member) { case RES_USAGE: return &counter->usage; |
c84872e16 memcgroup: add th... |
110 111 |
case RES_MAX_USAGE: return &counter->max_usage; |
e552b6617 Memory controller... |
112 113 114 115 |
case RES_LIMIT: return &counter->limit; case RES_FAILCNT: return &counter->failcnt; |
296c81d89 memory controller... |
116 117 |
case RES_SOFT_LIMIT: return &counter->soft_limit; |
e552b6617 Memory controller... |
118 119 120 121 122 123 124 |
}; BUG(); return NULL; } ssize_t res_counter_read(struct res_counter *counter, int member, |
0eea10301 Memory controller... |
125 126 |
const char __user *userbuf, size_t nbytes, loff_t *pos, int (*read_strategy)(unsigned long long val, char *st_buf)) |
e552b6617 Memory controller... |
127 |
{ |
0eea10301 Memory controller... |
128 |
unsigned long long *val; |
e552b6617 Memory controller... |
129 130 131 132 |
char buf[64], *s; s = buf; val = res_counter_member(counter, member); |
0eea10301 Memory controller... |
133 134 135 136 137 |
if (read_strategy) s += read_strategy(*val, s); else s += sprintf(s, "%llu ", *val); |
e552b6617 Memory controller... |
138 139 140 |
return simple_read_from_buffer((void __user *)userbuf, nbytes, pos, buf, s - buf); } |
6c191cd01 memcg: res_counte... |
141 142 143 144 145 146 147 148 149 150 151 152 153 |
#if BITS_PER_LONG == 32 u64 res_counter_read_u64(struct res_counter *counter, int member) { unsigned long flags; u64 ret; spin_lock_irqsave(&counter->lock, flags); ret = *res_counter_member(counter, member); spin_unlock_irqrestore(&counter->lock, flags); return ret; } #else |
2c7eabf37 CGroup API files:... |
154 155 156 157 |
u64 res_counter_read_u64(struct res_counter *counter, int member) { return *res_counter_member(counter, member); } |
6c191cd01 memcg: res_counte... |
158 |
#endif |
2c7eabf37 CGroup API files:... |
159 |
|
856c13aa1 cgroup files: con... |
160 161 |
int res_counter_memparse_write_strategy(const char *buf, unsigned long long *res) |
e552b6617 Memory controller... |
162 |
{ |
856c13aa1 cgroup files: con... |
163 |
char *end; |
c5b947b28 memcg: add interf... |
164 165 166 167 168 169 170 171 172 |
/* return RESOURCE_MAX(unlimited) if "-1" is specified */ if (*buf == '-') { *res = simple_strtoull(buf + 1, &end, 10); if (*res != 1 || *end != '\0') return -EINVAL; *res = RESOURCE_MAX; return 0; } |
52dcf8a1f resource cgroups:... |
173 |
*res = memparse(buf, &end); |
856c13aa1 cgroup files: con... |
174 175 |
if (*end != '\0') return -EINVAL; |
e552b6617 Memory controller... |
176 |
|
856c13aa1 cgroup files: con... |
177 178 179 |
*res = PAGE_ALIGN(*res); return 0; } |
e552b6617 Memory controller... |
180 |
|
856c13aa1 cgroup files: con... |
181 182 183 184 185 186 |
int res_counter_write(struct res_counter *counter, int member, const char *buf, write_strategy_fn write_strategy) { char *end; unsigned long flags; unsigned long long tmp, *val; |
e552b6617 Memory controller... |
187 |
|
0eea10301 Memory controller... |
188 |
if (write_strategy) { |
856c13aa1 cgroup files: con... |
189 190 |
if (write_strategy(buf, &tmp)) return -EINVAL; |
0eea10301 Memory controller... |
191 192 193 |
} else { tmp = simple_strtoull(buf, &end, 10); if (*end != '\0') |
856c13aa1 cgroup files: con... |
194 |
return -EINVAL; |
0eea10301 Memory controller... |
195 196 |
} spin_lock_irqsave(&counter->lock, flags); |
e552b6617 Memory controller... |
197 198 |
val = res_counter_member(counter, member); *val = tmp; |
0eea10301 Memory controller... |
199 |
spin_unlock_irqrestore(&counter->lock, flags); |
856c13aa1 cgroup files: con... |
200 |
return 0; |
e552b6617 Memory controller... |
201 |
} |