1 files changed, 4 insertions, 1 deletions

diff --git a/mm/swap_state.c b/mm/swap_state.c
index 67daecb6031a..b52635601dfe 100644
--- a/mm/swap_state.c
+++ b/mm/swap_state.c
@@ -74,6 +74,7 @@ static int __add_to_swap_cache(struct page *page, swp_entry_t entry,
 {
         int error;
 
+        BUG_ON(!PageLocked(page));
         BUG_ON(PageSwapCache(page));
         BUG_ON(PagePrivate(page));
         error = radix_tree_preload(gfp_mask);
@@ -83,7 +84,6 @@ static int __add_to_swap_cache(struct page *page, swp_entry_t entry,
                                                 entry.val, page);
                 if (!error) {
                         page_cache_get(page);
-                        SetPageLocked(page);
                         SetPageSwapCache(page);
                         set_page_private(page, entry.val);
                         total_swapcache_pages++;
@@ -99,15 +99,18 @@ static int add_to_swap_cache(struct page *page, swp_entry_t entry)
 {
         int error;
 
+        BUG_ON(PageLocked(page));
         if (!swap_duplicate(entry)) {
                 INC_CACHE_INFO(noent_race);
                 return -ENOENT;
         }
+        SetPageLocked(page);
         error = __add_to_swap_cache(page, entry, GFP_KERNEL);
         /*
          * Anon pages are already on the LRU, we don't run lru_cache_add here.
          */
         if (error) {
+                ClearPageLocked(page);
                 swap_free(entry);
                 if (error == -EEXIST)
                         INC_CACHE_INFO(exist_race);