Re: [FFmpeg-devel] [PATCH 22/23] dnn/dnn_backend_native_layer_mathbinary: Fix leak upon error

[Guo, Yejun]

> -----Original Message-----
> From: ffmpeg-devel <ffmpeg-devel-boun...@ffmpeg.org> On Behalf Of
> Andreas Rheinhardt
> Sent: 2021年3月11日 5:55
> To: ffmpeg-devel@ffmpeg.org
> Cc: Andreas Rheinhardt <andreas.rheinha...@gmail.com>
> Subject: [FFmpeg-devel] [PATCH 22/23]
> dnn/dnn_backend_native_layer_mathbinary: Fix leak upon error
> 
> Fixes Coverity issue #1473568.
> 
> Signed-off-by: Andreas Rheinhardt <andreas.rheinha...@gmail.com>
> ---
>  .../dnn/dnn_backend_native_layer_mathbinary.c | 23 +++++++++----------
>  1 file changed, 11 insertions(+), 12 deletions(-)
> 
> diff --git a/libavfilter/dnn/dnn_backend_native_layer_mathbinary.c
> b/libavfilter/dnn/dnn_backend_native_layer_mathbinary.c
> index c116188bab..1c33b4633d 100644
> --- a/libavfilter/dnn/dnn_backend_native_layer_mathbinary.c
> +++ b/libavfilter/dnn/dnn_backend_native_layer_mathbinary.c
> @@ -100,20 +100,17 @@ static void math_binary_not_commutative(FunType
> pfun, const DnnLayerMathBinaryPa  }  int
> ff_dnn_load_layer_math_binary(Layer *layer, AVIOContext
> *model_file_context, int file_size, int operands_num)  {
> -    DnnLayerMathBinaryParams *params;
> +    DnnLayerMathBinaryParams params = { 0 };
>      int dnn_size = 0;
>      int input_index = 0;
> -    params = av_malloc(sizeof(*params));
> -    if (!params)
> -        return 0;
> 
> -    params->bin_op = (int32_t)avio_rl32(model_file_context);
> +    params.bin_op = (int32_t)avio_rl32(model_file_context);
>      dnn_size += 4;
> 
> -    params->input0_broadcast = (int32_t)avio_rl32(model_file_context);
> +    params.input0_broadcast = (int32_t)avio_rl32(model_file_context);
>      dnn_size += 4;
> -    if (params->input0_broadcast) {
> -        params->v = av_int2float(avio_rl32(model_file_context));
> +    if (params.input0_broadcast) {
> +        params.v = av_int2float(avio_rl32(model_file_context));
>      } else {
>          layer->input_operand_indexes[input_index] =
> (int32_t)avio_rl32(model_file_context);
>          if (layer->input_operand_indexes[input_index] >= operands_num)
> { @@ -123,10 +120,10 @@ int ff_dnn_load_layer_math_binary(Layer *layer,
> AVIOContext *model_file_context,
>      }
>      dnn_size += 4;
> 
> -    params->input1_broadcast = (int32_t)avio_rl32(model_file_context);
> +    params.input1_broadcast = (int32_t)avio_rl32(model_file_context);
>      dnn_size += 4;
> -    if (params->input1_broadcast) {
> -        params->v = av_int2float(avio_rl32(model_file_context));
> +    if (params.input1_broadcast) {
> +        params.v = av_int2float(avio_rl32(model_file_context));
>      } else {
>          layer->input_operand_indexes[input_index] =
> (int32_t)avio_rl32(model_file_context);
>          if (layer->input_operand_indexes[input_index] >= operands_num)
> { @@ -138,11 +135,13 @@ int ff_dnn_load_layer_math_binary(Layer *layer,
> AVIOContext *model_file_context,
> 
>      layer->output_operand_index = (int32_t)avio_rl32(model_file_context);
>      dnn_size += 4;
> -    layer->params = params;
> 
>      if (layer->output_operand_index >= operands_num) {
>          return 0;
>      }
> +    layer->params = av_memdup(&params, sizeof(params));
> +    if (!layer->params)
> +        return 0;
> 
>      return dnn_size;
>  }

LGTM, thanks.

_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".