2 files changed, 21 insertions, 15 deletions

diff --git a/drivers/net/cnic.c b/drivers/net/cnic.c
index 04e299f46455..70a53cc7df32 100644
--- a/drivers/net/cnic.c
+++ b/drivers/net/cnic.c
@@ -257,7 +257,7 @@ static int cnic_get_l5_cid(struct cnic_local *cp, u32 cid, u32 *l5_cid)
 {
         u32 i;
 
-        for (i = 0; i < MAX_ISCSI_TBL_SZ; i++) {
+        for (i = 0; i < cp->max_cid_space; i++) {
                 if (cp->ctx_tbl[i].cid == cid) {
                         *l5_cid = i;
                         return 0;
@@ -981,17 +981,10 @@ error:
 static int cnic_alloc_bnx2x_context(struct cnic_dev *dev)
 {
         struct cnic_local *cp = dev->cnic_priv;
-        struct cnic_eth_dev *ethdev = cp->ethdev;
         int ctx_blk_size = cp->ethdev->ctx_blk_size;
-        int total_mem, blks, i, cid_space;
-
-        if (BNX2X_ISCSI_START_CID < ethdev->starting_cid)
-                return -EINVAL;
-
-        cid_space = MAX_ISCSI_TBL_SZ +
-                    (BNX2X_ISCSI_START_CID - ethdev->starting_cid);
+        int total_mem, blks, i;
 
-        total_mem = BNX2X_CONTEXT_MEM_SIZE * cid_space;
+        total_mem = BNX2X_CONTEXT_MEM_SIZE * cp->max_cid_space;
         blks = total_mem / ctx_blk_size;
         if (total_mem % ctx_blk_size)
                 blks++;
@@ -1035,16 +1028,27 @@ static int cnic_alloc_bnx2x_context(struct cnic_dev *dev)
 static int cnic_alloc_bnx2x_resc(struct cnic_dev *dev)
 {
         struct cnic_local *cp = dev->cnic_priv;
+        struct cnic_eth_dev *ethdev = cp->ethdev;
+        u32 start_cid = ethdev->starting_cid;
         int i, j, n, ret, pages;
         struct cnic_dma *kwq_16_dma = &cp->kwq_16_data_info;
 
+        cp->max_cid_space = MAX_ISCSI_TBL_SZ;
+        cp->iscsi_start_cid = start_cid;
+        if (start_cid < BNX2X_ISCSI_START_CID) {
+                u32 delta = BNX2X_ISCSI_START_CID - start_cid;
+
+                cp->iscsi_start_cid = BNX2X_ISCSI_START_CID;
+                cp->max_cid_space += delta;
+        }
+
         cp->iscsi_tbl = kzalloc(sizeof(struct cnic_iscsi) * MAX_ISCSI_TBL_SZ,
                                 GFP_KERNEL);
         if (!cp->iscsi_tbl)
                 goto error;
 
         cp->ctx_tbl = kzalloc(sizeof(struct cnic_context) *
-                                  MAX_CNIC_L5_CONTEXT, GFP_KERNEL);
+                                cp->max_cid_space, GFP_KERNEL);
         if (!cp->ctx_tbl)
                 goto error;
 
@@ -1053,7 +1057,7 @@ static int cnic_alloc_bnx2x_resc(struct cnic_dev *dev)
                 cp->ctx_tbl[i].ulp_proto_id = CNIC_ULP_ISCSI;
         }
 
-        pages = PAGE_ALIGN(MAX_CNIC_L5_CONTEXT * CNIC_KWQ16_DATA_SIZE) /
+        pages = PAGE_ALIGN(cp->max_cid_space * CNIC_KWQ16_DATA_SIZE) /
                 PAGE_SIZE;
 
         ret = cnic_alloc_dma(dev, kwq_16_dma, pages, 0);
@@ -1061,7 +1065,7 @@ static int cnic_alloc_bnx2x_resc(struct cnic_dev *dev)
                 return -ENOMEM;
 
         n = PAGE_SIZE / CNIC_KWQ16_DATA_SIZE;
-        for (i = 0, j = 0; i < MAX_ISCSI_TBL_SZ; i++) {
+        for (i = 0, j = 0; i < cp->max_cid_space; i++) {
                 long off = CNIC_KWQ16_DATA_SIZE * (i % n);
 
                 cp->ctx_tbl[i].kwqe_data = kwq_16_dma->pg_arr[j] + off;
@@ -4129,7 +4133,7 @@ static int cnic_start_bnx2x_hw(struct cnic_dev *dev)
         u8 sb_id = cp->status_blk_num;
 
         ret = cnic_init_id_tbl(&cp->cid_tbl, MAX_ISCSI_TBL_SZ,
-                               BNX2X_ISCSI_START_CID);
+                               cp->iscsi_start_cid);
 
         if (ret)
                 return -ENOMEM;
diff --git a/drivers/net/cnic.h b/drivers/net/cnic.h
index 08b1235d987d..b7e2f7fcfb1c 100644
--- a/drivers/net/cnic.h
+++ b/drivers/net/cnic.h
@@ -248,8 +248,10 @@ struct cnic_local {
         struct cnic_iscsi       *iscsi_tbl;
         struct cnic_context     *ctx_tbl;
         struct cnic_id_tbl      cid_tbl;
-        int                     max_iscsi_conn;
         atomic_t                iscsi_conn;
+        u32                     iscsi_start_cid;
+
+        u32                     max_cid_space;
 
         /* per connection parameters */
         int                     num_iscsi_tasks;