- Fixes on MPEG demuxer to handle VOBs without System Header.

Originally committed as revision 335 to svn://svn.ffmpeg.org/ffmpeg/trunk
pull/126/head
Juanjo 23 years ago
parent 9cdd6a24ad
commit 001e3f555d
  1. 184
      libav/mpeg.c
  2. 1
      libavcodec/mp3lameaudio.c

@ -59,6 +59,8 @@ typedef struct {
#define AUDIO_ID 0xc0
#define VIDEO_ID 0xe0
static int mpeg_mux_check_packet(AVFormatContext *s, int *size);
static int put_pack_header(AVFormatContext *ctx,
UINT8 *buf, INT64 timestamp)
{
@ -393,6 +395,19 @@ static int find_start_code(ByteIOContext *pb, int *size_ptr,
return val;
}
static int check_stream_id(AVFormatContext *s, int c_id)
{
AVStream *st;
int i;
for(i = 0;i < s->nb_streams;i++) {
st = s->streams[i];
if (st && st->id == c_id)
return 1;
}
return 0;
}
static int mpeg_mux_read_header(AVFormatContext *s,
AVFormatParameters *ap)
{
@ -400,6 +415,7 @@ static int mpeg_mux_read_header(AVFormatContext *s,
int size, startcode, c, rate_bound, audio_bound, video_bound, mux_rate, val;
int codec_id, n, i, type;
AVStream *st;
offset_t start_pos;
m = av_mallocz(sizeof(MpegDemuxContext));
if (!m)
@ -409,12 +425,69 @@ static int mpeg_mux_read_header(AVFormatContext *s,
/* search first pack header */
m->header_state = 0xff;
size = MAX_SYNC_SIZE;
start_pos = url_ftell(&s->pb); /* remember this pos */
for(;;) {
while (size > 0) {
startcode = find_start_code(&s->pb, &size, &m->header_state);
if (startcode == PACK_START_CODE)
goto found;
}
/* System Header not found find streams searching through file */
fprintf(stderr,"libav: MPEG-PS System Header not found!\n");
url_fseek(&s->pb, start_pos, SEEK_SET);
video_bound = 0;
audio_bound = 0;
c = 0;
s->nb_streams = 0;
size = 15*MAX_SYNC_SIZE;
while (size > 0) {
type = 0;
codec_id = 0;
n = 0;
startcode = find_start_code(&s->pb, &size, &m->header_state);
//fprintf(stderr,"\nstartcode: %x pos=0x%Lx\n", startcode, url_ftell(&s->pb));
if (startcode == 0x1bd) {
url_fseek(&s->pb, -4, SEEK_CUR);
size += 4;
startcode = mpeg_mux_check_packet(s, &size);
//fprintf(stderr,"\nstartcode: %x pos=0x%Lx\n", startcode, url_ftell(&s->pb));
if (startcode >= 0x80 && startcode <= 0x9f && !check_stream_id(s, startcode)) {
//fprintf(stderr,"Found AC3 stream ID: 0x%x\n", startcode);
type = CODEC_TYPE_AUDIO;
codec_id = CODEC_ID_AC3;
audio_bound++;
n = 1;
c = startcode;
}
} else if (startcode == 0x1e0 && !check_stream_id(s, startcode)) {
//fprintf(stderr,"Found MPEGVIDEO stream ID: 0x%x\n", startcode);
type = CODEC_TYPE_VIDEO;
codec_id = CODEC_ID_MPEG1VIDEO;
n = 1;
c = startcode;
video_bound++;
} /*else if (startcode >= 0x1c0 && startcode <= 0x1df && !check_stream_id(s, startcode)) {
fprintf(stderr,"Found MPEGAUDIO stream ID: 0x%x\n", startcode);
type = CODEC_TYPE_AUDIO;
codec_id = CODEC_ID_MP2;
n = 1;
c = startcode;
audio_bound++;
} */
for(i=0;i<n;i++) {
st = av_mallocz(sizeof(AVStream));
if (!st)
return -ENOMEM;
s->streams[s->nb_streams++] = st;
st->id = c;
st->codec.codec_type = type;
st->codec.codec_id = codec_id;
}
}
if (video_bound || audio_bound) {
url_fseek(&s->pb, start_pos, SEEK_SET);
return 0;
} else
return -ENODATA;
found:
/* search system header just after pack header */
@ -635,6 +708,117 @@ static int mpeg_mux_read_packet(AVFormatContext *s,
return 0;
}
static int mpeg_mux_check_packet(AVFormatContext *s, int *size)
{
MpegDemuxContext *m = s->priv_data;
int len, startcode, c, n, flags, header_len;
INT64 pts, dts;
/* next start code (should be immediately after */
redo:
m->header_state = 0xff;
startcode = find_start_code(&s->pb, size, &m->header_state);
if (startcode < 0)
return -EIO;
if (startcode == PACK_START_CODE)
goto redo;
if (startcode == SYSTEM_HEADER_START_CODE)
goto redo;
if (startcode == PADDING_STREAM ||
startcode == PRIVATE_STREAM_2) {
/* skip them */
len = get_be16(&s->pb);
url_fskip(&s->pb, len);
goto redo;
}
/* find matching stream */
if (!((startcode >= 0x1c0 && startcode <= 0x1df) ||
(startcode >= 0x1e0 && startcode <= 0x1ef) ||
(startcode == 0x1bd)))
goto redo;
n = *size;
len = get_be16(&s->pb);
n -= 2;
pts = 0;
dts = 0;
/* stuffing */
for(;;) {
c = get_byte(&s->pb);
len--;
n--;
/* XXX: for mpeg1, should test only bit 7 */
if (c != 0xff)
break;
}
if ((c & 0xc0) == 0x40) {
/* buffer scale & size */
get_byte(&s->pb);
c = get_byte(&s->pb);
len -= 2;
n -= 2;
}
if ((c & 0xf0) == 0x20) {
pts = get_pts(&s->pb, c);
len -= 4;
n -= 4;
dts = pts;
} else if ((c & 0xf0) == 0x30) {
pts = get_pts(&s->pb, c);
dts = get_pts(&s->pb, -1);
len -= 9;
n -= 9;
} else if ((c & 0xc0) == 0x80) {
/* mpeg 2 PES */
if ((c & 0x30) != 0) {
fprintf(stderr, "Encrypted multiplex not handled\n");
return -EIO;
}
flags = get_byte(&s->pb);
header_len = get_byte(&s->pb);
len -= 2;
n -= 2;
if (header_len > len)
goto redo;
if ((flags & 0xc0) == 0x40) {
pts = get_pts(&s->pb, -1);
dts = pts;
header_len -= 5;
len -= 5;
n -= 5;
} if ((flags & 0xc0) == 0xc0) {
pts = get_pts(&s->pb, -1);
dts = get_pts(&s->pb, -1);
header_len -= 10;
len -= 10;
n -= 10;
}
len -= header_len;
n -= header_len;
while (header_len > 0) {
get_byte(&s->pb);
header_len--;
}
}
if (startcode == 0x1bd) {
startcode = get_byte(&s->pb);
len--;
n--;
if (startcode >= 0x80 && startcode <= 0xbf) {
/* audio: skip header */
get_byte(&s->pb);
get_byte(&s->pb);
get_byte(&s->pb);
len -= 3;
n -= 3;
}
}
*size = n;
return startcode;
}
static int mpeg_mux_read_close(AVFormatContext *s)
{
MpegDemuxContext *m = s->priv_data;

@ -81,7 +81,6 @@ int MP3lame_encode_frame(AVCodecContext *avctx,
/* lame 3.91 outputs the first frame as garbage */
if (s->first_frame)
s->first_frame = num = 0;
return num;
}

Loading…
Cancel
Save