Commit e71149a7 authored by Vittorio Giovara's avatar Vittorio Giovara

nuv: validate image size

Avoid a division by zero.

CC: libav-stable@libav.org
Bug-Id: CID 717750
parent 7c51d79c
...@@ -20,6 +20,7 @@ ...@@ -20,6 +20,7 @@
*/ */
#include "libavutil/channel_layout.h" #include "libavutil/channel_layout.h"
#include "libavutil/imgutils.h"
#include "libavutil/intreadwrite.h" #include "libavutil/intreadwrite.h"
#include "libavutil/intfloat.h" #include "libavutil/intfloat.h"
#include "avformat.h" #include "avformat.h"
...@@ -188,6 +189,10 @@ static int nuv_header(AVFormatContext *s) ...@@ -188,6 +189,10 @@ static int nuv_header(AVFormatContext *s)
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
ctx->v_id = vst->index; ctx->v_id = vst->index;
ret = av_image_check_size(width, height, 0, ctx);
if (ret < 0)
return ret;
vst->codec->codec_type = AVMEDIA_TYPE_VIDEO; vst->codec->codec_type = AVMEDIA_TYPE_VIDEO;
vst->codec->codec_id = AV_CODEC_ID_NUV; vst->codec->codec_id = AV_CODEC_ID_NUV;
vst->codec->width = width; vst->codec->width = width;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment