[FFmpeg-devel] [PATCH] avfilter/vf_dnn_classify: add result check for av_frame_get_side_data
Steven Liu
liuqi05 at kuaishou.com
Fri May 7 09:42:48 EEST 2021
CID: 1482090
there can return null from av_frame_get_side_data, and will use sd->data
after av_frame_get_side_data, so should check null return value.
Signed-off-by: Steven Liu <liuqi05 at kuaishou.com>
---
libavfilter/vf_dnn_classify.c | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/libavfilter/vf_dnn_classify.c b/libavfilter/vf_dnn_classify.c
index 18fcd452d0..7900255cfe 100644
--- a/libavfilter/vf_dnn_classify.c
+++ b/libavfilter/vf_dnn_classify.c
@@ -77,6 +77,10 @@ static int dnn_classify_post_proc(AVFrame *frame, DNNData *output, uint32_t bbox
}
sd = av_frame_get_side_data(frame, AV_FRAME_DATA_DETECTION_BBOXES);
+ if (!sd) {
+ av_log(filter_ctx, AV_LOG_ERROR, "Cannot get side data in dnn_classify_post_proc\n");
+ return -1;
+ }
header = (AVDetectionBBoxHeader *)sd->data;
if (bbox_index == 0) {
--
2.25.0
More information about the ffmpeg-devel
mailing list