1 files changed, 7 insertions, 4 deletions

diff --git a/drivers/block/loop.c b/drivers/block/loop.c
index a076a14ca72d..c59a672a3de0 100644
--- a/drivers/block/loop.c
+++ b/drivers/block/loop.c
@@ -1658,7 +1658,7 @@ static struct kobject *loop_probe(dev_t dev, int *part, void *data)
         struct kobject *kobj;
 
         mutex_lock(&loop_devices_mutex);
-        lo = loop_init_one(dev & MINORMASK);
+        lo = loop_init_one(MINOR(dev) >> part_shift);
         kobj = lo ? get_disk(lo->lo_disk) : ERR_PTR(-ENOMEM);
         mutex_unlock(&loop_devices_mutex);
 
@@ -1691,15 +1691,18 @@ static int __init loop_init(void)
         if (max_part > 0)
                 part_shift = fls(max_part);
 
+        if ((1UL << part_shift) > DISK_MAX_PARTS)
+                return -EINVAL;
+
         if (max_loop > 1UL << (MINORBITS - part_shift))
                 return -EINVAL;
 
         if (max_loop) {
                 nr = max_loop;
-                range = max_loop;
+                range = max_loop << part_shift;
         } else {
                 nr = 8;
-                range = 1UL << (MINORBITS - part_shift);
+                range = 1UL << MINORBITS;
         }
 
         if (register_blkdev(LOOP_MAJOR, "loop"))
@@ -1738,7 +1741,7 @@ static void __exit loop_exit(void)
         unsigned long range;
         struct loop_device *lo, *next;
 
-        range = max_loop ? max_loop :  1UL << (MINORBITS - part_shift);
+        range = max_loop ? max_loop << part_shift : 1UL << MINORBITS;
 
         list_for_each_entry_safe(lo, next, &loop_devices, lo_list)
                 loop_del_one(lo);