On Wed 03-04-13 17:12:54, Li Zefan wrote: > Use css_get/put instead of mem_cgroup_get/put. > > Signed-off-by: Li Zefan <lize...@huawei.com>
Looks good to me. Swapped in pages still use css_tryget so they would fallback to recharge in __mem_cgroup_try_charge_swapin if the group was removed. Acked-by: Michal Hocko <mho...@suse.cz> Thanks! > --- > mm/memcontrol.c | 26 ++++++++++++++++---------- > 1 file changed, 16 insertions(+), 10 deletions(-) > > diff --git a/mm/memcontrol.c b/mm/memcontrol.c > index 877551d..ad576e8 100644 > --- a/mm/memcontrol.c > +++ b/mm/memcontrol.c > @@ -4137,12 +4137,12 @@ __mem_cgroup_uncharge_common(struct page *page, enum > charge_type ctype, > unlock_page_cgroup(pc); > /* > * even after unlock, we have memcg->res.usage here and this memcg > - * will never be freed. > + * will never be freed, so it's safe to call css_get(). > */ > memcg_check_events(memcg, page); > if (do_swap_account && ctype == MEM_CGROUP_CHARGE_TYPE_SWAPOUT) { > mem_cgroup_swap_statistics(memcg, true); > - mem_cgroup_get(memcg); > + css_get(&memcg->css); > } > /* > * Migration does not charge the res_counter for the > @@ -4242,7 +4242,7 @@ mem_cgroup_uncharge_swapcache(struct page *page, > swp_entry_t ent, bool swapout) > > /* > * record memcg information, if swapout && memcg != NULL, > - * mem_cgroup_get() was called in uncharge(). > + * css_get() was called in uncharge(). > */ > if (do_swap_account && swapout && memcg) > swap_cgroup_record(ent, css_id(&memcg->css)); > @@ -4273,7 +4273,7 @@ void mem_cgroup_uncharge_swap(swp_entry_t ent) > if (!mem_cgroup_is_root(memcg)) > res_counter_uncharge(&memcg->memsw, PAGE_SIZE); > mem_cgroup_swap_statistics(memcg, false); > - mem_cgroup_put(memcg); > + css_put(&memcg->css); > } > rcu_read_unlock(); > } > @@ -4307,11 +4307,14 @@ static int mem_cgroup_move_swap_account(swp_entry_t > entry, > * This function is only called from task migration context now. > * It postpones res_counter and refcount handling till the end > * of task migration(mem_cgroup_clear_mc()) for performance > - * improvement. But we cannot postpone mem_cgroup_get(to) > - * because if the process that has been moved to @to does > - * swap-in, the refcount of @to might be decreased to 0. > + * improvement. But we cannot postpone css_get(to) because if > + * the process that has been moved to @to does swap-in, the > + * refcount of @to might be decreased to 0. > + * > + * We are in attach() phase, so the cgroup is guaranteed to be > + * alive, so we can just call css_get(). > */ > - mem_cgroup_get(to); > + css_get(&to->css); > return 0; > } > return -EINVAL; > @@ -6597,6 +6600,7 @@ static void __mem_cgroup_clear_mc(void) > { > struct mem_cgroup *from = mc.from; > struct mem_cgroup *to = mc.to; > + int i; > > /* we must uncharge all the leftover precharges from mc.to */ > if (mc.precharge) { > @@ -6617,7 +6621,9 @@ static void __mem_cgroup_clear_mc(void) > if (!mem_cgroup_is_root(mc.from)) > res_counter_uncharge(&mc.from->memsw, > PAGE_SIZE * mc.moved_swap); > - __mem_cgroup_put(mc.from, mc.moved_swap); > + > + for (i = 0; i < mc.moved_swap; i++) > + css_put(&mc.from->css); > > if (!mem_cgroup_is_root(mc.to)) { > /* > @@ -6627,7 +6633,7 @@ static void __mem_cgroup_clear_mc(void) > res_counter_uncharge(&mc.to->res, > PAGE_SIZE * mc.moved_swap); > } > - /* we've already done mem_cgroup_get(mc.to) */ > + /* we've already done css_get(mc.to) */ > mc.moved_swap = 0; > } > memcg_oom_recover(from); > -- > 1.8.0.2 -- Michal Hocko SUSE Labs -- To unsubscribe from this list: send the line "unsubscribe linux-kernel" in the body of a message to majord...@vger.kernel.org More majordomo info at http://vger.kernel.org/majordomo-info.html Please read the FAQ at http://www.tux.org/lkml/