[vlc-commits] vpx: encoder: check for block allocation failure

[Tristan Matthews]

vlc | branch: master | Tristan Matthews <tmatth at videolan.org> | Mon Apr 16 10:10:49 2018 -0400| [1d22b9a39dc707570f2fd75f5cb13463143c4966] | committer: Tristan Matthews

vpx: encoder: check for block allocation failure

> http://git.videolan.org/gitweb.cgi/vlc.git/?a=commit;h=1d22b9a39dc707570f2fd75f5cb13463143c4966
---

 modules/codec/vpx.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/modules/codec/vpx.c b/modules/codec/vpx.c
index f92dea2c56..c69b5ff768 100644
--- a/modules/codec/vpx.c
+++ b/modules/codec/vpx.c
@@ -494,6 +494,11 @@ static block_t *Encode(encoder_t *p_enc, picture_t *p_pict)
         {
             int keyframe = pkt->data.frame.flags & VPX_FRAME_IS_KEY;
             block_t *p_block = block_Alloc(pkt->data.frame.sz);
+            if (unlikely(p_block == NULL))
+            {
+                vpx_img_free(&img);
+                return NULL;
+            }
 
             memcpy(p_block->p_buffer, pkt->data.frame.buf, pkt->data.frame.sz);
             p_block->i_dts = p_block->i_pts = pkt->data.frame.pts;