1 files changed, 5 insertions, 8 deletions

diff --git a/block/genhd.c b/block/genhd.c
index cd65b97e84d4..841b3037d866 100644
--- a/block/genhd.c
+++ b/block/genhd.c
@@ -26,7 +26,7 @@ static DEFINE_MUTEX(block_class_lock);
 struct kobject *block_depr;
 
 /* for extended dynamic devt allocation, currently only one major is used */
-#define MAX_EXT_DEVT            (1 << MINORBITS)
+#define NR_EXT_DEVT             (1 << MINORBITS)
 
 /* For extended devt allocation.  ext_devt_mutex prevents look up
  * results from going away underneath its user.
@@ -425,19 +425,16 @@ int blk_alloc_devt(struct hd_struct *part, dev_t *devt)
                         return -ENOMEM;
                 mutex_lock(&ext_devt_mutex);
                 rc = idr_get_new(&ext_devt_idr, part, &idx);
+                if (!rc && idx >= NR_EXT_DEVT) {
+                        idr_remove(&ext_devt_idr, idx);
+                        rc = -EBUSY;
+                }
                 mutex_unlock(&ext_devt_mutex);
         } while (rc == -EAGAIN);
 
         if (rc)
                 return rc;
 
-        if (idx > MAX_EXT_DEVT) {
-                mutex_lock(&ext_devt_mutex);
-                idr_remove(&ext_devt_idr, idx);
-                mutex_unlock(&ext_devt_mutex);
-                return -EBUSY;
-        }
-
         *devt = MKDEV(BLOCK_EXT_MAJOR, blk_mangle_minor(idx));
         return 0;
 }