Use a flexible array member to combine allocations Add __counted_by for extra runtime analysis.
Signed-off-by: Rosen Penev <[email protected]> --- drivers/edac/ecs.c | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/drivers/edac/ecs.c b/drivers/edac/ecs.c index 51c451c7f0f0..f0833080c5cc 100644 --- a/drivers/edac/ecs.c +++ b/drivers/edac/ecs.c @@ -33,7 +33,7 @@ struct edac_ecs_fru_context { struct edac_ecs_context { u16 num_media_frus; - struct edac_ecs_fru_context *fru_ctxs; + struct edac_ecs_fru_context fru_ctxs[] __counted_by(num_media_frus); }; #define TO_ECS_DEV_ATTR(_dev_attr) \ @@ -149,17 +149,11 @@ static int ecs_create_desc(struct device *ecs_dev, const struct attribute_group struct edac_ecs_context *ecs_ctx; u32 fru; - ecs_ctx = devm_kzalloc(ecs_dev, sizeof(*ecs_ctx), GFP_KERNEL); + ecs_ctx = devm_kzalloc(ecs_dev, struct_size(ecs_ctx, fru_ctxs, num_media_frus), GFP_KERNEL); if (!ecs_ctx) return -ENOMEM; ecs_ctx->num_media_frus = num_media_frus; - ecs_ctx->fru_ctxs = devm_kcalloc(ecs_dev, num_media_frus, - sizeof(*ecs_ctx->fru_ctxs), - GFP_KERNEL); - if (!ecs_ctx->fru_ctxs) - return -ENOMEM; - for (fru = 0; fru < num_media_frus; fru++) { struct edac_ecs_fru_context *fru_ctx = &ecs_ctx->fru_ctxs[fru]; struct attribute_group *group = &fru_ctx->group; -- 2.54.0

