1 files changed, 16 insertions, 23 deletions

diff --git a/kernel/power/user.c b/kernel/power/user.c
index 7cf6713b2325..040560d9c312 100644
--- a/kernel/power/user.c
+++ b/kernel/power/user.c
@@ -33,25 +33,29 @@
 static struct snapshot_data {
         struct snapshot_handle handle;
         int swap;
-        struct bitmap_page *bitmap;
         int mode;
         char frozen;
         char ready;
         char platform_suspend;
 } snapshot_state;
 
-static atomic_t device_available = ATOMIC_INIT(1);
+atomic_t snapshot_device_available = ATOMIC_INIT(1);
 
 static int snapshot_open(struct inode *inode, struct file *filp)
 {
         struct snapshot_data *data;
 
-        if (!atomic_add_unless(&device_available, -1, 0))
+        if (!atomic_add_unless(&snapshot_device_available, -1, 0))
                 return -EBUSY;
 
-        if ((filp->f_flags & O_ACCMODE) == O_RDWR)
+        if ((filp->f_flags & O_ACCMODE) == O_RDWR) {
+                atomic_inc(&snapshot_device_available);
                 return -ENOSYS;
-
+        }
+        if(create_basic_memory_bitmaps()) {
+                atomic_inc(&snapshot_device_available);
+                return -ENOMEM;
+        }
         nonseekable_open(inode, filp);
         data = &snapshot_state;
         filp->private_data = data;
@@ -64,7 +68,6 @@ static int snapshot_open(struct inode *inode, struct file *filp)
                 data->swap = -1;
                 data->mode = O_WRONLY;
         }
-        data->bitmap = NULL;
         data->frozen = 0;
         data->ready = 0;
         data->platform_suspend = 0;
@@ -77,16 +80,15 @@ static int snapshot_release(struct inode *inode, struct file *filp)
         struct snapshot_data *data;
 
         swsusp_free();
+        free_basic_memory_bitmaps();
         data = filp->private_data;
-        free_all_swap_pages(data->swap, data->bitmap);
-        free_bitmap(data->bitmap);
+        free_all_swap_pages(data->swap);
         if (data->frozen) {
                 mutex_lock(&pm_mutex);
                 thaw_processes();
-                enable_nonboot_cpus();
                 mutex_unlock(&pm_mutex);
         }
-        atomic_inc(&device_available);
+        atomic_inc(&snapshot_device_available);
         return 0;
 }
 
@@ -294,14 +296,7 @@ static int snapshot_ioctl(struct inode *inode, struct file *filp,
                         error = -ENODEV;
                         break;
                 }
-                if (!data->bitmap) {
-                        data->bitmap = alloc_bitmap(count_swap_pages(data->swap, 0));
-                        if (!data->bitmap) {
-                                error = -ENOMEM;
-                                break;
-                        }
-                }
-                offset = alloc_swapdev_block(data->swap, data->bitmap);
+                offset = alloc_swapdev_block(data->swap);
                 if (offset) {
                         offset <<= PAGE_SHIFT;
                         error = put_user(offset, (sector_t __user *)arg);
@@ -315,13 +310,11 @@ static int snapshot_ioctl(struct inode *inode, struct file *filp,
                         error = -ENODEV;
                         break;
                 }
-                free_all_swap_pages(data->swap, data->bitmap);
-                free_bitmap(data->bitmap);
-                data->bitmap = NULL;
+                free_all_swap_pages(data->swap);
                 break;
 
         case SNAPSHOT_SET_SWAP_FILE:
-                if (!data->bitmap) {
+                if (!swsusp_swap_in_use()) {
                         /*
                          * User space encodes device types as two-byte values,
                          * so we need to recode them
@@ -420,7 +413,7 @@ static int snapshot_ioctl(struct inode *inode, struct file *filp,
                 break;
 
         case SNAPSHOT_SET_SWAP_AREA:
-                if (data->bitmap) {
+                if (swsusp_swap_in_use()) {
                         error = -EPERM;
                 } else {
                         struct resume_swap_area swap_area;