From 9116d8773665e937ca3f4e357cfcbe3abba93b27 Mon Sep 17 00:00:00 2001
From: Namhyung Kim <namhyung@gmail.com>
Date: Thu, 26 May 2011 16:25:31 -0700
Subject: [PATCH] --- yaml --- r: 252039 b: refs/heads/master c:
 6a5b18d2bd79795407bf65451a2013bf4dfb588b h: refs/heads/master i:   252037:
 3547977bc2965859cb0f39ae780eb27c7bf79745   252035:
 c55bf6183b01ce1357585a68be942132a9f9d2ed   252031:
 375fe4123fded2bed42e810f4d4dbff0e2ae2d59 v: v3

---
 [refs]                 |  2 +-
 trunk/mm/page_cgroup.c | 22 +++++++++++++---------
 2 files changed, 14 insertions(+), 10 deletions(-)

diff --git a/[refs] b/[refs]
index f47515aeed63..9aa34dc9f0ea 100644
--- a/[refs]
+++ b/[refs]
@@ -1,2 +1,2 @@
 ---
-refs/heads/master: 33278f7f0a9e5a9b29f59e07b7e3182d27769ac2
+refs/heads/master: 6a5b18d2bd79795407bf65451a2013bf4dfb588b
diff --git a/trunk/mm/page_cgroup.c b/trunk/mm/page_cgroup.c
index 5524e8bd7437..74ccff61d1be 100644
--- a/trunk/mm/page_cgroup.c
+++ b/trunk/mm/page_cgroup.c
@@ -492,8 +492,8 @@ int swap_cgroup_swapon(int type, unsigned long max_pages)
 		/* memory shortage */
 		ctrl->map = NULL;
 		ctrl->length = 0;
-		vfree(array);
 		mutex_unlock(&swap_cgroup_mutex);
+		vfree(array);
 		goto nomem;
 	}
 	mutex_unlock(&swap_cgroup_mutex);
@@ -508,7 +508,8 @@ int swap_cgroup_swapon(int type, unsigned long max_pages)
 
 void swap_cgroup_swapoff(int type)
 {
-	int i;
+	struct page **map;
+	unsigned long i, length;
 	struct swap_cgroup_ctrl *ctrl;
 
 	if (!do_swap_account)
@@ -516,17 +517,20 @@ void swap_cgroup_swapoff(int type)
 
 	mutex_lock(&swap_cgroup_mutex);
 	ctrl = &swap_cgroup_ctrl[type];
-	if (ctrl->map) {
-		for (i = 0; i < ctrl->length; i++) {
-			struct page *page = ctrl->map[i];
+	map = ctrl->map;
+	length = ctrl->length;
+	ctrl->map = NULL;
+	ctrl->length = 0;
+	mutex_unlock(&swap_cgroup_mutex);
+
+	if (map) {
+		for (i = 0; i < length; i++) {
+			struct page *page = map[i];
 			if (page)
 				__free_page(page);
 		}
-		vfree(ctrl->map);
-		ctrl->map = NULL;
-		ctrl->length = 0;
+		vfree(map);
 	}
-	mutex_unlock(&swap_cgroup_mutex);
 }
 
 #endif