memcontrol: move oom task exclusion to tasklist scan

[safe/jmp/linux-2.6] / mm / oom_kill.c

diff --git a/mm/oom_kill.c b/mm/oom_kill.c
index 64751dc..ef5084d 100644 (file)
--- a/mm/oom_kill.c
+++ b/mm/oom_kill.c
@@ -65,13 +65,6 @@ unsigned long badness(struct task_struct *p, unsigned long uptime,
                return 0;
        }
 
-#ifdef CONFIG_CGROUP_MEM_CONT
-       if (mem != NULL && mm->mem_cgroup != mem) {
-               task_unlock(p);
-               return 0;
-       }
-#endif
-
        /*
         * The memory size of the process is the basis for the badness.
         */
@@ -223,6 +216,8 @@ static struct task_struct *select_bad_process(unsigned long *ppoints,
                /* skip the init task */
                if (is_global_init(p))
                        continue;
+               if (mem && !task_in_mem_cgroup(p, mem))
+                       continue;
 
                /*
                 * This task already has access to memory reserves and is