From 98431cde070a33fe346f1316abd8437cd873dbfa Mon Sep 17 00:00:00 2001 From: Yaowu Xu Date: Tue, 12 Jul 2016 11:11:26 -0700 Subject: [PATCH] Fix encoder crashes for odd size input Change-Id: Id5c30c419282369cc8c3280d9a70b34a859a71d8 --- vp9/vp9_cx_iface.c | 5 +++-- vpx_dsp/psnrhvs.c | 2 ++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/vp9/vp9_cx_iface.c b/vp9/vp9_cx_iface.c index 51c6fbb..f4e989f 100644 --- a/vp9/vp9_cx_iface.c +++ b/vp9/vp9_cx_iface.c @@ -992,6 +992,7 @@ static vpx_codec_frame_flags_t get_frame_pkt_flags(const VP9_COMP *cpi, return flags; } +const size_t kMinCompressedSize = 8192; static vpx_codec_err_t encoder_encode(vpx_codec_alg_priv_t *ctx, const vpx_image_t *img, vpx_codec_pts_t pts, @@ -1013,8 +1014,8 @@ static vpx_codec_err_t encoder_encode(vpx_codec_alg_priv_t *ctx, // instance for its status to determine the compressed data size. data_sz = ctx->cfg.g_w * ctx->cfg.g_h * get_image_bps(img) / 8 * (cpi->multi_arf_allowed ? 8 : 2); - if (data_sz < 4096) - data_sz = 4096; + if (data_sz < kMinCompressedSize) + data_sz = kMinCompressedSize; if (ctx->cx_data == NULL || ctx->cx_data_sz < data_sz) { ctx->cx_data_sz = data_sz; free(ctx->cx_data); diff --git a/vpx_dsp/psnrhvs.c b/vpx_dsp/psnrhvs.c index 095ba5d..3708cc3 100644 --- a/vpx_dsp/psnrhvs.c +++ b/vpx_dsp/psnrhvs.c @@ -245,6 +245,8 @@ static double calc_psnrhvs(const unsigned char *src, int _systride, } } } + if (pixels <=0) + return 0; ret /= pixels; return ret; } -- 2.7.4