Re: [PATCH] i2c: atr: use kzalloc_flex

[Romain Gantois]

Hello Rosen,

On Friday, 13 March 2026 01:23:08 CEST Rosen Penev wrote:
> Convert kzalloc_obj + kcalloc to kzalloc_flex to save an allocation.
> 
> Add __counted_by to get extra runtime analysis. Move counting variable
> assignment immediately after allocation as required by __counted_by.
> 
> Signed-off-by: Rosen Penev <rosenp@xxxxxxxxx>
> ---
> 
>  drivers/i2c/i2c-atr.c | 15 +++------------
>  1 file changed, 3 insertions(+), 12 deletions(-)
> 
> diff --git a/drivers/i2c/i2c-atr.c b/drivers/i2c/i2c-atr.c
> index f9fcb4793aaf..b0aedcf0bf71 100644
> --- a/drivers/i2c/i2c-atr.c
> +++ b/drivers/i2c/i2c-atr.c
> @@ -58,8 +58,8 @@ struct i2c_atr_alias_pool {
> 
>  	/* Protects aliases and use_mask */
>  	spinlock_t lock;
> 
> -	u16 *aliases;
> 
>  	unsigned long *use_mask;
> 
> +	u16 aliases[] __counted_by(size);
> 
>  };
>  
>  /**
> 
> @@ -137,22 +137,16 @@ static struct i2c_atr_alias_pool
> *i2c_atr_alloc_alias_pool(size_t num_aliases, b struct i2c_atr_alias_pool
> *alias_pool;
> 
>  	int ret;
> 
> -	alias_pool = kzalloc_obj(*alias_pool);
> +	alias_pool = kzalloc_flex(*alias_pool, aliases, num_aliases);
> 
>  	if (!alias_pool)
>  	
>  		return ERR_PTR(-ENOMEM);
>  	
>  	alias_pool->size = num_aliases;
> 
> -	alias_pool->aliases = kcalloc(num_aliases, sizeof(*alias_pool->aliases),
> GFP_KERNEL); -	if (!alias_pool->aliases) {
> -		ret = -ENOMEM;
> -		goto err_free_alias_pool;
> -	}
> -
> 
>  	alias_pool->use_mask = bitmap_zalloc(num_aliases, GFP_KERNEL);
>  	if (!alias_pool->use_mask) {
>  	
>  		ret = -ENOMEM;
> 
> -		goto err_free_aliases;
> +		goto err_free_alias_pool;
> 
>  	}
>  	
>  	alias_pool->shared = shared;
> 
> @@ -161,8 +155,6 @@ static struct i2c_atr_alias_pool
> *i2c_atr_alloc_alias_pool(size_t num_aliases, b
> 
>  	return alias_pool;
> 
> -err_free_aliases:
> -	kfree(alias_pool->aliases);
> 
>  err_free_alias_pool:
>  	kfree(alias_pool);
>  	return ERR_PTR(ret);
> 
> @@ -171,7 +163,6 @@ static struct i2c_atr_alias_pool
> *i2c_atr_alloc_alias_pool(size_t num_aliases, b static void
> i2c_atr_free_alias_pool(struct i2c_atr_alias_pool *alias_pool) {
> 
>  	bitmap_free(alias_pool->use_mask);
> 
> -	kfree(alias_pool->aliases);
> 
>  	kfree(alias_pool);
>  
>  }

Reviewed-by: Romain Gantois <romain.gantois@xxxxxxxxxxx>

Attachment: signature.asc
Description: This is a digitally signed message part.