Просмотр исходного кода

yuv4mpeg: complain and exit if a non-rawvideo stream is selected

The yuv4mpeg muxer will crash otherwise.
Stefano Sabatini 13 лет назад
Родитель
Сommit
1f95fb5813
1 измененных файлов с 6 добавлено и 0 удалено
  1. 6 0
      libavformat/yuv4mpeg.c

+ 6 - 0
libavformat/yuv4mpeg.c

@@ -154,6 +154,12 @@ static int yuv4_write_header(AVFormatContext *s)
     if (s->nb_streams != 1)
         return AVERROR(EIO);
 
+    if (s->streams[0]->codec->codec_id != CODEC_ID_RAWVIDEO) {
+        av_log(s, AV_LOG_ERROR,
+               "A non-rawvideo stream was selected, but yuv4mpeg only handles rawvideo streams\n");
+        return AVERROR(EINVAL);
+    }
+
     if (s->streams[0]->codec->pix_fmt == PIX_FMT_YUV411P) {
         av_log(s, AV_LOG_ERROR, "Warning: generating rarely used 4:1:1 YUV stream, some mjpegtools might not work.\n");
     }