Use a flexible array member to combine allocations.

Add __counted_by for extra runtime analysis.

Remove goto paths as they are not really needed anymore.

Signed-off-by: Rosen Penev <[email protected]>
---
 arch/arm/mach-omap2/omap_device.c | 21 ++++++---------------
 arch/arm/mach-omap2/omap_device.h |  2 +-
 2 files changed, 7 insertions(+), 16 deletions(-)

diff --git a/arch/arm/mach-omap2/omap_device.c 
b/arch/arm/mach-omap2/omap_device.c
index 79db4c49ffc9..55f2a9d5f4df 100644
--- a/arch/arm/mach-omap2/omap_device.c
+++ b/arch/arm/mach-omap2/omap_device.c
@@ -309,17 +309,15 @@ static struct omap_device *omap_device_alloc(struct 
platform_device *pdev,
        int i;
        struct omap_hwmod **hwmods;
 
-       od = kzalloc_obj(struct omap_device);
-       if (!od)
-               goto oda_exit1;
+       od = kzalloc_flex(*od, hwmods, oh_cnt);
+       if (!od) {
+               dev_err(&pdev->dev, "omap_device: build failed (%d)\n", ret);
+               return ERR_PTR(ret);
+       }
 
        od->hwmods_cnt = oh_cnt;
+       memcpy(od->hwmods, ohs, oh_cnt * sizeof(*od->hwmods));
 
-       hwmods = kmemdup_array(ohs, oh_cnt, sizeof(*hwmods), GFP_KERNEL);
-       if (!hwmods)
-               goto oda_exit2;
-
-       od->hwmods = hwmods;
        od->pdev = pdev;
        pdev->archdata.od = od;
 
@@ -329,13 +327,6 @@ static struct omap_device *omap_device_alloc(struct 
platform_device *pdev,
        }
 
        return od;
-
-oda_exit2:
-       kfree(od);
-oda_exit1:
-       dev_err(&pdev->dev, "omap_device: build failed (%d)\n", ret);
-
-       return ERR_PTR(ret);
 }
 
 static void omap_device_delete(struct omap_device *od)
diff --git a/arch/arm/mach-omap2/omap_device.h 
b/arch/arm/mach-omap2/omap_device.h
index aa8096ecb23c..9283c15c2403 100644
--- a/arch/arm/mach-omap2/omap_device.h
+++ b/arch/arm/mach-omap2/omap_device.h
@@ -51,11 +51,11 @@
  */
 struct omap_device {
        struct platform_device          *pdev;
-       struct omap_hwmod               **hwmods;
        unsigned long                   _driver_status;
        u8                              hwmods_cnt;
        u8                              _state;
        u8                              flags;
+       struct omap_hwmod               *hwmods[] __counted_by(hwmods_cnt);
 };
 
 /* Device driver interface (call via platform_data fn ptrs) */
-- 
2.53.0


Reply via email to