4 * Copyright 2007 OpenVZ SWsoft Inc
6 * Author: Pavel Emelianov <xemul@openvz.org>
10 #include <linux/types.h>
11 #include <linux/parser.h>
13 #include <linux/res_counter.h>
14 #include <linux/uaccess.h>
16 void res_counter_init(struct res_counter *counter)
18 spin_lock_init(&counter->lock);
19 counter->limit = (unsigned long)LONG_MAX;
22 int res_counter_charge_locked(struct res_counter *counter, unsigned long val)
24 if (counter->usage + val > counter->limit) {
29 counter->usage += val;
33 int res_counter_charge(struct res_counter *counter, unsigned long val)
38 spin_lock_irqsave(&counter->lock, flags);
39 ret = res_counter_charge_locked(counter, val);
40 spin_unlock_irqrestore(&counter->lock, flags);
44 void res_counter_uncharge_locked(struct res_counter *counter, unsigned long val)
46 if (WARN_ON(counter->usage < val))
49 counter->usage -= val;
52 void res_counter_uncharge(struct res_counter *counter, unsigned long val)
56 spin_lock_irqsave(&counter->lock, flags);
57 res_counter_uncharge_locked(counter, val);
58 spin_unlock_irqrestore(&counter->lock, flags);
62 static inline unsigned long *res_counter_member(struct res_counter *counter,
67 return &counter->usage;
69 return &counter->limit;
71 return &counter->failcnt;
78 ssize_t res_counter_read(struct res_counter *counter, int member,
79 const char __user *userbuf, size_t nbytes, loff_t *pos)
85 val = res_counter_member(counter, member);
86 s += sprintf(s, "%lu\n", *val);
87 return simple_read_from_buffer((void __user *)userbuf, nbytes,
91 ssize_t res_counter_write(struct res_counter *counter, int member,
92 const char __user *userbuf, size_t nbytes, loff_t *pos)
96 unsigned long tmp, *val;
98 buf = kmalloc(nbytes + 1, GFP_KERNEL);
105 if (copy_from_user(buf, userbuf, nbytes))
109 tmp = simple_strtoul(buf, &end, 10);
113 val = res_counter_member(counter, member);