real: fix potential memory leak.

d2a4fbf9 · Rémi Duraffort · a80d58fe · d2a4fbf9
Commit d2a4fbf9 authored Apr 07, 2012 by Rémi Duraffort
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

modules/demux/real.c modules/demux/real.c +6 -0

No files found.
--- a/modules/demux/real.c
+++ b/modules/demux/real.c
@@ -1567,7 +1567,10 @@ static int CodecAudioParse( demux_t *p_demux, int i_tk_id, const uint8_t *p_data
        {
            fmt.p_extra = malloc( i_extra_codec );
            if( !fmt.p_extra || i_extra_codec > i_data )
+            {
+                free( fmt.p_extra );
                return VLC_ENOMEM;
+            }

            fmt.i_extra = i_extra_codec;
            memcpy( fmt.p_extra, p_data, fmt.i_extra );
@@ -1606,7 +1609,10 @@ static int CodecAudioParse( demux_t *p_demux, int i_tk_id, const uint8_t *p_data
        {
            fmt.p_extra = malloc( i_extra_codec );
            if( !fmt.p_extra || i_extra_codec > i_data )
+            {
+                free( fmt.p_extra );
                return VLC_ENOMEM;
+            }

            fmt.i_extra = i_extra_codec;
            memcpy( fmt.p_extra, p_data, fmt.i_extra );