diff mbox series

[FFmpeg-devel] fftools/ffmpeg_mux_init: fix memory leak in ffmpeg_mux_init.c

Message ID 20240412164441.1727089-1-lumingyindetect@163.com
State New
Headers show
Series [FFmpeg-devel] fftools/ffmpeg_mux_init: fix memory leak in ffmpeg_mux_init.c | expand

Checks

Context Check Description
yinshiyou/make_loongarch64 success Make finished
yinshiyou/make_fate_loongarch64 success Make fate finished
andriy/make_x86 success Make finished
andriy/make_fate_x86 success Make fate finished

Commit Message

LuMingYin April 12, 2024, 4:44 p.m. UTC
Signed-off-by: LuMingYin <lumingyindetect@163.com>
---
 fftools/ffmpeg_mux_init.c | 7 ++++---
 libavutil/mem.c           | 5 ++++-
 2 files changed, 8 insertions(+), 4 deletions(-)

Comments

James Almer April 12, 2024, 7:11 p.m. UTC | #1
On 4/12/2024 1:44 PM, LuMingYin wrote:
> Signed-off-by: LuMingYin <lumingyindetect@163.com>
> ---
>   fftools/ffmpeg_mux_init.c | 7 ++++---
>   libavutil/mem.c           | 5 ++++-
>   2 files changed, 8 insertions(+), 4 deletions(-)
> 
> diff --git a/fftools/ffmpeg_mux_init.c b/fftools/ffmpeg_mux_init.c
> index 6d8bd5bcdf..e7e2281bd0 100644
> --- a/fftools/ffmpeg_mux_init.c
> +++ b/fftools/ffmpeg_mux_init.c
> @@ -2851,9 +2851,10 @@ static int parse_forced_key_frames(void *log, KeyframeForceCtx *kf,
>   
>               if (nb_ch > INT_MAX - size ||
>                   !(pts = av_realloc_f(pts, size += nb_ch - 1,
> -                                     sizeof(*pts))))
> -                return AVERROR(ENOMEM);
> -
> +                                     sizeof(*pts)))) {
> +                ret = AVERROR(ENOMEM);
> +                goto fail;
> +            }
>               if (p[8]) {
>                   ret = av_parse_time(&t, p + 8, 1);
>                   if (ret < 0) {
> diff --git a/libavutil/mem.c b/libavutil/mem.c
> index b205d3fb25..7f34765fe7 100644
> --- a/libavutil/mem.c
> +++ b/libavutil/mem.c
> @@ -177,11 +177,14 @@ void *av_realloc_f(void *ptr, size_t nelem, size_t elsize)
>   
>       if (size_mult(elsize, nelem, &size)) {
>           av_free(ptr);
> +        ptr = NULL;
>           return NULL;
>       }
>       r = av_realloc(ptr, size);
> -    if (!r)
> +    if (!r) {
>           av_free(ptr);
> +        ptr = NULL;
> +    }

This doesn't do what you think it does. ptr is a local variable in this 
scope. Setting it to NULL will not be reflected in ffmpeg_mux_init.c
And even if it did, it would be an unexpected behavior change in a 
public API function.

>       return r;
>   }
>
LuMingYin April 13, 2024, 2:42 a.m. UTC | #2
Thank&nbsp;you&nbsp;for&nbsp;your&nbsp;patient&nbsp;explanation!&nbsp;As&nbsp;you&nbsp;pointed&nbsp;out,&nbsp;I&nbsp;made&nbsp;unnecessary&nbsp;modifications&nbsp;in&nbsp;the&nbsp;public&nbsp;functions&nbsp;in&nbsp;the&nbsp;previous&nbsp;patch.&nbsp;Now,&nbsp;I&nbsp;have&nbsp;rectified&nbsp;them&nbsp;and&nbsp;created&nbsp;a&nbsp;new&nbsp;patch.&nbsp;The&nbsp;patch&nbsp;link&nbsp;is:https://patchwork.ffmpeg.org/project/ffmpeg/patch/20240413023726.1843852-1-lumingyindetect@163.com/
在 2024-04-13 03:11:22,"James Almer" <jamrial@gmail.com> 写道:
On 4/12/2024 1:44 PM, LuMingYin wrote:
> Signed-off-by: LuMingYin <lumingyindetect@163.com>
> ---
>   fftools/ffmpeg_mux_init.c | 7 ++++---
>   libavutil/mem.c           | 5 ++++-
>   2 files changed, 8 insertions(+), 4 deletions(-)
> 
> diff --git a/fftools/ffmpeg_mux_init.c b/fftools/ffmpeg_mux_init.c
> index 6d8bd5bcdf..e7e2281bd0 100644
> --- a/fftools/ffmpeg_mux_init.c
> +++ b/fftools/ffmpeg_mux_init.c
> @@ -2851,9 +2851,10 @@ static int parse_forced_key_frames(void *log, KeyframeForceCtx *kf,
>   
>               if (nb_ch > INT_MAX - size ||
>                   !(pts = av_realloc_f(pts, size += nb_ch - 1,
> -                                     sizeof(*pts))))
> -                return AVERROR(ENOMEM);
> -
> +                                     sizeof(*pts)))) {
> +                ret = AVERROR(ENOMEM);
> +                goto fail;
> +            }
>               if (p[8]) {
>                   ret = av_parse_time(&t, p + 8, 1);
>                   if (ret < 0) {
> diff --git a/libavutil/mem.c b/libavutil/mem.c
> index b205d3fb25..7f34765fe7 100644
> --- a/libavutil/mem.c
> +++ b/libavutil/mem.c
> @@ -177,11 +177,14 @@ void *av_realloc_f(void *ptr, size_t nelem, size_t elsize)
>   
>       if (size_mult(elsize, nelem, &size)) {
>           av_free(ptr);
> +        ptr = NULL;
>           return NULL;
>       }
>       r = av_realloc(ptr, size);
> -    if (!r)
> +    if (!r) {
>           av_free(ptr);
> +        ptr = NULL;
> +    }

This doesn't do what you think it does. ptr is a local variable in this 
scope. Setting it to NULL will not be reflected in ffmpeg_mux_init.c
And even if it did, it would be an unexpected behavior change in a 
public API function.

>       return r;
>   }
>   
_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".
LuMingYin April 13, 2024, 2:44 a.m. UTC | #3
Thank you for your patient explanation! As you pointed out, I made unnecessary modifications in the public functions in the previous patch. Now, I have rectified them and created a new patch. The patch link is:https://patchwork.ffmpeg.org/project/ffmpeg/patch/20240413023726.1843852-1-lumingyindetect@163.com/
At 2024-04-13 03:11:22, "James Almer" <jamrial@gmail.com> wrote:
>On 4/12/2024 1:44 PM, LuMingYin wrote:
>> Signed-off-by: LuMingYin <lumingyindetect@163.com>
>> ---
>>   fftools/ffmpeg_mux_init.c | 7 ++++---
>>   libavutil/mem.c           | 5 ++++-
>>   2 files changed, 8 insertions(+), 4 deletions(-)
>> 
>> diff --git a/fftools/ffmpeg_mux_init.c b/fftools/ffmpeg_mux_init.c
>> index 6d8bd5bcdf..e7e2281bd0 100644
>> --- a/fftools/ffmpeg_mux_init.c
>> +++ b/fftools/ffmpeg_mux_init.c
>> @@ -2851,9 +2851,10 @@ static int parse_forced_key_frames(void *log, KeyframeForceCtx *kf,
>>   
>>               if (nb_ch > INT_MAX - size ||
>>                   !(pts = av_realloc_f(pts, size += nb_ch - 1,
>> -                                     sizeof(*pts))))
>> -                return AVERROR(ENOMEM);
>> -
>> +                                     sizeof(*pts)))) {
>> +                ret = AVERROR(ENOMEM);
>> +                goto fail;
>> +            }
>>               if (p[8]) {
>>                   ret = av_parse_time(&t, p + 8, 1);
>>                   if (ret < 0) {
>> diff --git a/libavutil/mem.c b/libavutil/mem.c
>> index b205d3fb25..7f34765fe7 100644
>> --- a/libavutil/mem.c
>> +++ b/libavutil/mem.c
>> @@ -177,11 +177,14 @@ void *av_realloc_f(void *ptr, size_t nelem, size_t elsize)
>>   
>>       if (size_mult(elsize, nelem, &size)) {
>>           av_free(ptr);
>> +        ptr = NULL;
>>           return NULL;
>>       }
>>       r = av_realloc(ptr, size);
>> -    if (!r)
>> +    if (!r) {
>>           av_free(ptr);
>> +        ptr = NULL;
>> +    }
>
>This doesn't do what you think it does. ptr is a local variable in this 
>scope. Setting it to NULL will not be reflected in ffmpeg_mux_init.c
>And even if it did, it would be an unexpected behavior change in a 
>public API function.
>
>>       return r;
>>   }
>>   
>_______________________________________________
>ffmpeg-devel mailing list
>ffmpeg-devel@ffmpeg.org
>https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
>
>To unsubscribe, visit link above, or email
>ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".
diff mbox series

Patch

diff --git a/fftools/ffmpeg_mux_init.c b/fftools/ffmpeg_mux_init.c
index 6d8bd5bcdf..e7e2281bd0 100644
--- a/fftools/ffmpeg_mux_init.c
+++ b/fftools/ffmpeg_mux_init.c
@@ -2851,9 +2851,10 @@  static int parse_forced_key_frames(void *log, KeyframeForceCtx *kf,
 
             if (nb_ch > INT_MAX - size ||
                 !(pts = av_realloc_f(pts, size += nb_ch - 1,
-                                     sizeof(*pts))))
-                return AVERROR(ENOMEM);
-
+                                     sizeof(*pts)))) {
+                ret = AVERROR(ENOMEM);
+                goto fail;
+            }
             if (p[8]) {
                 ret = av_parse_time(&t, p + 8, 1);
                 if (ret < 0) {
diff --git a/libavutil/mem.c b/libavutil/mem.c
index b205d3fb25..7f34765fe7 100644
--- a/libavutil/mem.c
+++ b/libavutil/mem.c
@@ -177,11 +177,14 @@  void *av_realloc_f(void *ptr, size_t nelem, size_t elsize)
 
     if (size_mult(elsize, nelem, &size)) {
         av_free(ptr);
+        ptr = NULL;
         return NULL;
     }
     r = av_realloc(ptr, size);
-    if (!r)
+    if (!r) {
         av_free(ptr);
+        ptr = NULL;
+    }
     return r;
 }