Commit e767c9e8 authored by Luca Barbato's avatar Luca Barbato

flv: Validate and reject unsupported codecs

And provide a more informative message in case of failure.

CC: libav-stable@libav.org
parent 87e85a13
...@@ -297,6 +297,17 @@ static void write_metadata(AVFormatContext *s, unsigned int ts) ...@@ -297,6 +297,17 @@ static void write_metadata(AVFormatContext *s, unsigned int ts)
avio_wb32(pb, data_size + 11); avio_wb32(pb, data_size + 11);
} }
static int unsupported_codec(AVFormatContext *s,
const char* type, int codec_id)
{
const AVCodecDescriptor *desc = avcodec_descriptor_get(codec_id);
av_log(s, AV_LOG_ERROR,
"%s codec %s not compatible with flv\n",
type,
desc ? desc->name : "unknown");
return AVERROR(ENOSYS);
}
static int flv_write_header(AVFormatContext *s) static int flv_write_header(AVFormatContext *s)
{ {
int i; int i;
...@@ -319,10 +330,8 @@ static int flv_write_header(AVFormatContext *s) ...@@ -319,10 +330,8 @@ static int flv_write_header(AVFormatContext *s)
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
flv->video_enc = enc; flv->video_enc = enc;
if (enc->codec_tag == 0) { if (!ff_codec_get_tag(flv_video_codec_ids, enc->codec_id))
av_log(s, AV_LOG_ERROR, "video codec not compatible with flv\n"); return unsupported_codec(s, "Video", enc->codec_id);
return -1;
}
break; break;
case AVMEDIA_TYPE_AUDIO: case AVMEDIA_TYPE_AUDIO:
if (flv->audio_enc) { if (flv->audio_enc) {
...@@ -332,13 +341,11 @@ static int flv_write_header(AVFormatContext *s) ...@@ -332,13 +341,11 @@ static int flv_write_header(AVFormatContext *s)
} }
flv->audio_enc = enc; flv->audio_enc = enc;
if (get_audio_flags(s, enc) < 0) if (get_audio_flags(s, enc) < 0)
return AVERROR_INVALIDDATA; return unsupported_codec(s, "Audio", enc->codec_id);
break; break;
case AVMEDIA_TYPE_DATA: case AVMEDIA_TYPE_DATA:
if (enc->codec_id != AV_CODEC_ID_TEXT) { if (enc->codec_id != AV_CODEC_ID_TEXT)
av_log(s, AV_LOG_ERROR, "codec not compatible with flv\n"); return unsupported_codec(s, "Data", enc->codec_id);
return AVERROR_INVALIDDATA;
}
flv->data_enc = enc; flv->data_enc = enc;
break; break;
default: default:
...@@ -479,13 +486,7 @@ static int flv_write_packet(AVFormatContext *s, AVPacket *pkt) ...@@ -479,13 +486,7 @@ static int flv_write_packet(AVFormatContext *s, AVPacket *pkt)
case AVMEDIA_TYPE_VIDEO: case AVMEDIA_TYPE_VIDEO:
avio_w8(pb, FLV_TAG_TYPE_VIDEO); avio_w8(pb, FLV_TAG_TYPE_VIDEO);
flags = enc->codec_tag; flags = ff_codec_get_tag(flv_video_codec_ids, enc->codec_id);
if (flags == 0) {
av_log(s, AV_LOG_ERROR,
"video codec %X not compatible with flv\n",
enc->codec_id);
return -1;
}
flags |= pkt->flags & AV_PKT_FLAG_KEY ? FLV_FRAME_KEY : FLV_FRAME_INTER; flags |= pkt->flags & AV_PKT_FLAG_KEY ? FLV_FRAME_KEY : FLV_FRAME_INTER;
break; break;
......
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