memcg: do not allow to disable tcp accounting after limit is set

There are two bits defined for cg_proto->flags - MEMCG_SOCK_ACTIVATED
and MEMCG_SOCK_ACTIVE - both are set in tcp_update_limit, but the former
is never cleared while the latter can be cleared by unsetting the limit.
This allows to disable tcp socket accounting for new sockets after it
was enabled by writing -1 to memory.kmem.tcp.limit_in_bytes while still
guaranteeing that memcg_socket_limit_enabled static key will be
decremented on memcg destruction.

This functionality looks dubious, because it is not clear what a use
case would be.  By enabling tcp accounting a user accepts the price.  If
they then find the performance degradation unacceptable, they can always
restart their workload with tcp accounting disabled.  It does not seem
there is any need to flip it while the workload is running.

Besides, it contradicts to how kmem accounting API works: writing
whatever to memory.kmem.limit_in_bytes enables kmem accounting for the
cgroup in question, after which it cannot be disabled.  Therefore one
might expect that writing -1 to memory.kmem.tcp.limit_in_bytes just
enables socket accounting w/o limiting it, which might be useful by
itself, but it isn't true.

Since this API peculiarity is not documented anywhere, I propose to drop
it.  This will allow to simplify the code by dropping cg_proto->flags.

Signed-off-by: Vladimir Davydov <vdavydov@virtuozzo.com>
Cc: Johannes Weiner <hannes@cmpxchg.org>
Cc: Michal Hocko <mhocko@kernel.org>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
This commit is contained in:
Vladimir Davydov 2016-01-14 15:19:41 -08:00 committed by Linus Torvalds
parent 316bda0e6c
commit 9ee11ba425
3 changed files with 7 additions and 24 deletions

View File

@ -85,22 +85,12 @@ enum mem_cgroup_events_target {
MEM_CGROUP_NTARGETS, MEM_CGROUP_NTARGETS,
}; };
/*
* Bits in struct cg_proto.flags
*/
enum cg_proto_flags {
/* Currently active and new sockets should be assigned to cgroups */
MEMCG_SOCK_ACTIVE,
/* It was ever activated; we must disarm static keys on destruction */
MEMCG_SOCK_ACTIVATED,
};
struct cg_proto { struct cg_proto {
struct page_counter memory_allocated; /* Current allocated memory. */ struct page_counter memory_allocated; /* Current allocated memory. */
struct percpu_counter sockets_allocated; /* Current number of sockets. */ struct percpu_counter sockets_allocated; /* Current number of sockets. */
int memory_pressure; int memory_pressure;
bool active;
long sysctl_mem[3]; long sysctl_mem[3];
unsigned long flags;
/* /*
* memcg field is used to find which memcg we belong directly * memcg field is used to find which memcg we belong directly
* Each memcg struct can hold more than one cg_proto, so container_of * Each memcg struct can hold more than one cg_proto, so container_of

View File

@ -316,7 +316,7 @@ void sock_update_memcg(struct sock *sk)
rcu_read_lock(); rcu_read_lock();
memcg = mem_cgroup_from_task(current); memcg = mem_cgroup_from_task(current);
cg_proto = sk->sk_prot->proto_cgroup(memcg); cg_proto = sk->sk_prot->proto_cgroup(memcg);
if (cg_proto && test_bit(MEMCG_SOCK_ACTIVE, &cg_proto->flags) && if (cg_proto && cg_proto->active &&
css_tryget_online(&memcg->css)) { css_tryget_online(&memcg->css)) {
sk->sk_cgrp = cg_proto; sk->sk_cgrp = cg_proto;
} }

View File

@ -48,7 +48,7 @@ void tcp_destroy_cgroup(struct mem_cgroup *memcg)
percpu_counter_destroy(&cg_proto->sockets_allocated); percpu_counter_destroy(&cg_proto->sockets_allocated);
if (test_bit(MEMCG_SOCK_ACTIVATED, &cg_proto->flags)) if (cg_proto->active)
static_key_slow_dec(&memcg_socket_limit_enabled); static_key_slow_dec(&memcg_socket_limit_enabled);
} }
@ -72,11 +72,9 @@ static int tcp_update_limit(struct mem_cgroup *memcg, unsigned long nr_pages)
cg_proto->sysctl_mem[i] = min_t(long, nr_pages, cg_proto->sysctl_mem[i] = min_t(long, nr_pages,
sysctl_tcp_mem[i]); sysctl_tcp_mem[i]);
if (nr_pages == PAGE_COUNTER_MAX) if (!cg_proto->active) {
clear_bit(MEMCG_SOCK_ACTIVE, &cg_proto->flags);
else {
/* /*
* The active bit needs to be written after the static_key * The active flag needs to be written after the static_key
* update. This is what guarantees that the socket activation * update. This is what guarantees that the socket activation
* function is the last one to run. See sock_update_memcg() for * function is the last one to run. See sock_update_memcg() for
* details, and note that we don't mark any socket as belonging * details, and note that we don't mark any socket as belonging
@ -90,14 +88,9 @@ static int tcp_update_limit(struct mem_cgroup *memcg, unsigned long nr_pages)
* We never race with the readers in sock_update_memcg(), * We never race with the readers in sock_update_memcg(),
* because when this value change, the code to process it is not * because when this value change, the code to process it is not
* patched in yet. * patched in yet.
*
* The activated bit is used to guarantee that no two writers
* will do the update in the same memcg. Without that, we can't
* properly shutdown the static key.
*/ */
if (!test_and_set_bit(MEMCG_SOCK_ACTIVATED, &cg_proto->flags)) static_key_slow_inc(&memcg_socket_limit_enabled);
static_key_slow_inc(&memcg_socket_limit_enabled); cg_proto->active = true;
set_bit(MEMCG_SOCK_ACTIVE, &cg_proto->flags);
} }
return 0; return 0;