Commit 6cca95fb authored by Denis Charmet's avatar Denis Charmet

Avoid double free in case of corrupted files

parent b7979366
......@@ -121,6 +121,11 @@ void EbmlParser::Keep( void )
mb_keep = true;
}
void EbmlParser::Unkeep()
{
mb_keep = false;
}
int EbmlParser::GetLevel( void ) const
{
return mi_user_level;
......
......@@ -41,6 +41,7 @@ class EbmlParser
void Reset( demux_t *p_demux );
EbmlElement *Get( int n_call = 0 );
void Keep( void );
void Unkeep( void );
EbmlElement *UnGet( uint64 i_block_pos, uint64 i_cluster_pos );
int GetLevel( void ) const;
......
......@@ -1215,7 +1215,7 @@ int matroska_segment_c::BlockGet( KaxBlock * & pp_block, KaxSimpleBlock * & pp_s
/* Check blocks validity to protect againts broken files */
if( BlockFindTrackIndex( &i_tk, pp_block , pp_simpleblock ) )
{
delete pp_block;
ep->Unkeep();
pp_simpleblock = NULL;
pp_block = NULL;
continue;
......@@ -1361,6 +1361,9 @@ int matroska_segment_c::BlockGet( KaxBlock * & pp_block, KaxSimpleBlock * & pp_s
{
msg_Err( &sys.demuxer, "Error while reading %s... upping level", typeid(*el).name());
ep->Up();
ep->Unkeep();
pp_simpleblock = NULL;
pp_block = NULL;
break;
}
if( MKV_IS_ID( el, KaxBlock ) )
......@@ -1414,6 +1417,9 @@ int matroska_segment_c::BlockGet( KaxBlock * & pp_block, KaxSimpleBlock * & pp_s
{
msg_Err( &sys.demuxer, "Error while reading %s... upping level", typeid(*el).name());
ep->Up();
ep->Unkeep();
pp_simpleblock = NULL;
pp_block = NULL;
}
}
}
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment