[PATCH] net: marvell: prestera: use kzalloc_flex
From: Rosen Penev <hidden>
Date: 2026-03-18 19:21:31
Also in:
linux-hardening, lkml
Subsystem:
marvell prestera ethernet switch driver, networking drivers, the rest · Maintainers:
Elad Nachman, Andrew Lunn, "David S. Miller", Eric Dumazet, Jakub Kicinski, Paolo Abeni, Linus Torvalds
Simplify allocation by using a flexible array member in this struct. No need to free separately. Add __counted_by to get extra runtime analysis. Move counting variable assignment to right after allocation as required by __counted_by. Signed-off-by: Rosen Penev <redacted> --- .../ethernet/marvell/prestera/prestera_devlink.c | 16 +++------------- 1 file changed, 3 insertions(+), 13 deletions(-)
diff --git a/drivers/net/ethernet/marvell/prestera/prestera_devlink.c b/drivers/net/ethernet/marvell/prestera/prestera_devlink.c
index 06b876f6b27f..9c043cddb0f9 100644
--- a/drivers/net/ethernet/marvell/prestera/prestera_devlink.c
+++ b/drivers/net/ethernet/marvell/prestera/prestera_devlink.c@@ -125,8 +125,8 @@ struct prestera_trap_item { struct prestera_trap_data { struct prestera_switch *sw; - struct prestera_trap_item *trap_items_arr; u32 traps_count; + struct prestera_trap_item trap_items_arr[] __counted_by(traps_count); }; #define PRESTERA_TRAP_METADATA DEVLINK_TRAP_METADATA_TYPE_F_IN_PORT
@@ -451,19 +451,12 @@ int prestera_devlink_traps_register(struct prestera_switch *sw) struct prestera_trap *prestera_trap; int err, i; - trap_data = kzalloc_obj(*trap_data); + trap_data = kzalloc_flex(*trap_data, trap_items_arr, traps_count); if (!trap_data) return -ENOMEM; - trap_data->trap_items_arr = kzalloc_objs(struct prestera_trap_item, - traps_count); - if (!trap_data->trap_items_arr) { - err = -ENOMEM; - goto err_trap_items_alloc; - } - - trap_data->sw = sw; trap_data->traps_count = traps_count; + trap_data->sw = sw; sw->trap_data = trap_data; err = devlink_trap_groups_register(devlink, prestera_trap_groups_arr,
@@ -489,8 +482,6 @@ int prestera_devlink_traps_register(struct prestera_switch *sw) devlink_trap_groups_unregister(devlink, prestera_trap_groups_arr, groups_count); err_groups_register: - kfree(trap_data->trap_items_arr); -err_trap_items_alloc: kfree(trap_data); return err; }
@@ -594,6 +585,5 @@ void prestera_devlink_traps_unregister(struct prestera_switch *sw) devlink_trap_groups_unregister(dl, prestera_trap_groups_arr, ARRAY_SIZE(prestera_trap_groups_arr)); - kfree(trap_data->trap_items_arr); kfree(trap_data); }
--
2.53.0