1 files changed, 8 insertions, 7 deletions

diff --git a/drivers/target/iscsi/iscsi_target.c b/drivers/target/iscsi/iscsi_target.c
index 23a98e658306..9435a3d369a7 100644
--- a/drivers/target/iscsi/iscsi_target.c
+++ b/drivers/target/iscsi/iscsi_target.c
@@ -144,23 +144,24 @@ struct iscsi_tiqn *iscsit_add_tiqn(unsigned char *buf)
         spin_lock_init(&tiqn->login_stats.lock);
         spin_lock_init(&tiqn->logout_stats.lock);
 
-        if (!idr_pre_get(&tiqn_idr, GFP_KERNEL)) {
-                pr_err("idr_pre_get() for tiqn_idr failed\n");
-                kfree(tiqn);
-                return ERR_PTR(-ENOMEM);
-        }
         tiqn->tiqn_state = TIQN_STATE_ACTIVE;
 
+        idr_preload(GFP_KERNEL);
         spin_lock(&tiqn_lock);
-        ret = idr_get_new(&tiqn_idr, NULL, &tiqn->tiqn_index);
+
+        ret = idr_alloc(&tiqn_idr, NULL, 0, 0, GFP_NOWAIT);
         if (ret < 0) {
-                pr_err("idr_get_new() failed for tiqn->tiqn_index\n");
+                pr_err("idr_alloc() failed for tiqn->tiqn_index\n");
                 spin_unlock(&tiqn_lock);
+                idr_preload_end();
                 kfree(tiqn);
                 return ERR_PTR(ret);
         }
+        tiqn->tiqn_index = ret;
         list_add_tail(&tiqn->tiqn_list, &g_tiqn_list);
+
         spin_unlock(&tiqn_lock);
+        idr_preload_end();
 
         pr_debug("CORE[0] - Added iSCSI Target IQN: %s\n", tiqn->tiqn);