2 #include <net/tcp_memcontrol.h>
5 #include <linux/nsproxy.h>
6 #include <linux/memcontrol.h>
7 #include <linux/module.h>
9 static inline struct tcp_memcontrol *tcp_from_cgproto(struct cg_proto *cg_proto)
11 return container_of(cg_proto, struct tcp_memcontrol, cg_proto);
14 static void memcg_tcp_enter_memory_pressure(struct sock *sk)
16 if (!sk->sk_cgrp->memory_pressure)
17 *sk->sk_cgrp->memory_pressure = 1;
19 EXPORT_SYMBOL(memcg_tcp_enter_memory_pressure);
21 int tcp_init_cgroup(struct cgroup *cgrp, struct cgroup_subsys *ss)
24 * The root cgroup does not use res_counters, but rather,
25 * rely on the data already collected by the network
28 struct res_counter *res_parent = NULL;
29 struct cg_proto *cg_proto, *parent_cg;
30 struct tcp_memcontrol *tcp;
31 struct mem_cgroup *memcg = mem_cgroup_from_cont(cgrp);
32 struct mem_cgroup *parent = parent_mem_cgroup(memcg);
33 struct net *net = current->nsproxy->net_ns;
35 cg_proto = tcp_prot.proto_cgroup(memcg);
39 tcp = tcp_from_cgproto(cg_proto);
41 tcp->tcp_prot_mem[0] = net->ipv4.sysctl_tcp_mem[0];
42 tcp->tcp_prot_mem[1] = net->ipv4.sysctl_tcp_mem[1];
43 tcp->tcp_prot_mem[2] = net->ipv4.sysctl_tcp_mem[2];
44 tcp->tcp_memory_pressure = 0;
46 parent_cg = tcp_prot.proto_cgroup(parent);
48 res_parent = parent_cg->memory_allocated;
50 res_counter_init(&tcp->tcp_memory_allocated, res_parent);
51 percpu_counter_init(&tcp->tcp_sockets_allocated, 0);
53 cg_proto->enter_memory_pressure = memcg_tcp_enter_memory_pressure;
54 cg_proto->memory_pressure = &tcp->tcp_memory_pressure;
55 cg_proto->sysctl_mem = tcp->tcp_prot_mem;
56 cg_proto->memory_allocated = &tcp->tcp_memory_allocated;
57 cg_proto->sockets_allocated = &tcp->tcp_sockets_allocated;
58 cg_proto->memcg = memcg;
62 EXPORT_SYMBOL(tcp_init_cgroup);
64 void tcp_destroy_cgroup(struct cgroup *cgrp, struct cgroup_subsys *ss)
66 struct mem_cgroup *memcg = mem_cgroup_from_cont(cgrp);
67 struct cg_proto *cg_proto;
68 struct tcp_memcontrol *tcp;
70 cg_proto = tcp_prot.proto_cgroup(memcg);
74 tcp = tcp_from_cgproto(cg_proto);
75 percpu_counter_destroy(&tcp->tcp_sockets_allocated);
77 EXPORT_SYMBOL(tcp_destroy_cgroup);