packetizer_helper: add missing alloc check

2 years ago · 07d50a2b39
1 changed files with 5 additions and 0 deletions
--- a/modules/packetizer/packetizer_helper.h
+++ b/modules/packetizer/packetizer_helper.h
@ -181,6 +181,11 @@ static block_t *packetizer_PacketizeBlock( packetizer_t *p_pack, block_t **pp_bl
            block_t *p_block_bytestream = p_pack->bytestream.p_block;

            p_pic = block_Alloc( p_pack->i_offset + p_pack->i_au_prepend );
+            if( p_pic == NULL )
+            {
+                p_pack->i_state = STATE_NOSYNC;
+                return NULL;
+            }
            p_pic->i_pts = p_block_bytestream->i_pts;
            p_pic->i_dts = p_block_bytestream->i_dts;