2 files changed, 10 insertions, 2 deletions

diff --git a/drivers/infiniband/hw/cxgb4/mem.c b/drivers/infiniband/hw/cxgb4/mem.c
index f9ca072a99ed..ec7a2988a703 100644
--- a/drivers/infiniband/hw/cxgb4/mem.c
+++ b/drivers/infiniband/hw/cxgb4/mem.c
@@ -259,8 +259,12 @@ static int write_tpt_entry(struct c4iw_rdev *rdev, u32 reset_tpt_entry,
 
         if ((!reset_tpt_entry) && (*stag == T4_STAG_UNSET)) {
                 stag_idx = c4iw_get_resource(&rdev->resource.tpt_table);
-                if (!stag_idx)
+                if (!stag_idx) {
+                        mutex_lock(&rdev->stats.lock);
+                        rdev->stats.stag.fail++;
+                        mutex_unlock(&rdev->stats.lock);
                         return -ENOMEM;
+                }
                 mutex_lock(&rdev->stats.lock);
                 rdev->stats.stag.cur += 32;
                 if (rdev->stats.stag.cur > rdev->stats.stag.max)
diff --git a/drivers/infiniband/hw/cxgb4/resource.c b/drivers/infiniband/hw/cxgb4/resource.c
index 94b5fd9b9379..67df71a7012e 100644
--- a/drivers/infiniband/hw/cxgb4/resource.c
+++ b/drivers/infiniband/hw/cxgb4/resource.c
@@ -179,8 +179,12 @@ u32 c4iw_get_qpid(struct c4iw_rdev *rdev, struct c4iw_dev_ucontext *uctx)
                 kfree(entry);
         } else {
                 qid = c4iw_get_resource(&rdev->resource.qid_table);
-                if (!qid)
+                if (!qid) {
+                        mutex_lock(&rdev->stats.lock);
+                        rdev->stats.qid.fail++;
+                        mutex_unlock(&rdev->stats.lock);
                         goto out;
+                }
                 mutex_lock(&rdev->stats.lock);
                 rdev->stats.qid.cur += rdev->qpmask + 1;
                 mutex_unlock(&rdev->stats.lock);