1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/infiniband/hw/mthca/mthca_memfree.c b/drivers/infiniband/hw/mthca/mthca_memfree.c
index 2fe503e86c1d..7a31be3c3e73 100644
--- a/drivers/infiniband/hw/mthca/mthca_memfree.c
+++ b/drivers/infiniband/hw/mthca/mthca_memfree.c
@@ -367,7 +367,7 @@ struct mthca_icm_table *mthca_alloc_icm_table(struct mthca_dev *dev,
         obj_per_chunk = MTHCA_TABLE_CHUNK_SIZE / obj_size;
         num_icm = DIV_ROUND_UP(nobj, obj_per_chunk);
 
-        table = kmalloc(sizeof *table + num_icm * sizeof *table->icm, GFP_KERNEL);
+        table = kmalloc(struct_size(table, icm, num_icm), GFP_KERNEL);
         if (!table)
                 return NULL;
 
@@ -529,7 +529,7 @@ struct mthca_user_db_table *mthca_init_user_db_tab(struct mthca_dev *dev)
                 return NULL;
 
         npages = dev->uar_table.uarc_size / MTHCA_ICM_PAGE_SIZE;
-        db_tab = kmalloc(sizeof *db_tab + npages * sizeof *db_tab->page, GFP_KERNEL);
+        db_tab = kmalloc(struct_size(db_tab, page, npages), GFP_KERNEL);
         if (!db_tab)
                 return ERR_PTR(-ENOMEM);