commit cacd9adade86f998163c5694b93082d0cbfd180c
author jianxing cai <jianxing.cai@intel.com> Thu Mar 25 12:21:17 2021 +0800
committer Carl Zhang <carl.zhang@intel.com> Tue Mar 30 10:20:52 2021 -0400
tree eb9259331a91342e29dd0b2400002e0766836d72
parent b70ba2f4303fb11f7071328cf608c72fbf2297d3

add check of malloc result for jpegenc

encode/jpegenc.c

diff --git a/encode/jpegenc.c b/encode/jpegenc.c
index 5879892..b00ad8d 100644
--- a/encode/jpegenc.c
+++ b/encode/jpegenc.c
@@ -501,6 +501,11 @@
     u_size = ((picture_width >> 1) * (picture_height >> 1));
 
     newImageBuffer = malloc(frame_size);
+    if(newImageBuffer == NULL)
+    {
+        printf("ERROR......upload_yuv_to_surface malloc failed");
+        exit(1);
+    }
     memset(newImageBuffer,0,frame_size);
     do {
         n_items = fread(newImageBuffer, frame_size, 1, yuv_fp);