2005-04-09 17:32:58 +02:00
|
|
|
/*
|
|
|
|
* Ogg bitstream support
|
|
|
|
* Luca Barbato <lu_zero@gentoo.org>
|
|
|
|
* Based on tcvp implementation
|
|
|
|
*/
|
|
|
|
|
2011-10-30 19:10:50 +01:00
|
|
|
/*
|
2005-04-09 17:32:58 +02:00
|
|
|
Copyright (C) 2005 Michael Ahlberg, Måns Rullgård
|
|
|
|
|
|
|
|
Permission is hereby granted, free of charge, to any person
|
|
|
|
obtaining a copy of this software and associated documentation
|
|
|
|
files (the "Software"), to deal in the Software without
|
|
|
|
restriction, including without limitation the rights to use, copy,
|
|
|
|
modify, merge, publish, distribute, sublicense, and/or sell copies
|
|
|
|
of the Software, and to permit persons to whom the Software is
|
|
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
|
|
|
|
The above copyright notice and this permission notice shall be
|
|
|
|
included in all copies or substantial portions of the Software.
|
|
|
|
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
|
|
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
|
|
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
|
|
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
|
|
|
HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
|
|
WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
|
|
DEALINGS IN THE SOFTWARE.
|
2011-10-30 19:10:50 +01:00
|
|
|
*/
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
#include <stdio.h>
|
2012-04-17 14:44:06 +02:00
|
|
|
#include "libavutil/avassert.h"
|
2007-11-07 21:22:32 +01:00
|
|
|
#include "oggdec.h"
|
2005-04-09 17:32:58 +02:00
|
|
|
#include "avformat.h"
|
2011-10-16 15:03:30 +02:00
|
|
|
#include "internal.h"
|
2010-03-20 14:36:43 +01:00
|
|
|
#include "vorbiscomment.h"
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
#define MAX_PAGE_SIZE 65307
|
|
|
|
#define DECODER_BUFFER_SIZE MAX_PAGE_SIZE
|
|
|
|
|
2008-11-06 02:50:56 +01:00
|
|
|
static const struct ogg_codec * const ogg_codecs[] = {
|
2010-03-11 08:17:40 +01:00
|
|
|
&ff_skeleton_codec,
|
2010-01-11 01:31:52 +01:00
|
|
|
&ff_dirac_codec,
|
2008-08-24 19:37:43 +02:00
|
|
|
&ff_speex_codec,
|
|
|
|
&ff_vorbis_codec,
|
|
|
|
&ff_theora_codec,
|
|
|
|
&ff_flac_codec,
|
2011-04-15 19:18:04 +02:00
|
|
|
&ff_celt_codec,
|
2012-06-24 11:38:18 +02:00
|
|
|
&ff_opus_codec,
|
2010-01-11 01:31:52 +01:00
|
|
|
&ff_old_dirac_codec,
|
2008-08-24 19:37:43 +02:00
|
|
|
&ff_old_flac_codec,
|
|
|
|
&ff_ogm_video_codec,
|
|
|
|
&ff_ogm_audio_codec,
|
|
|
|
&ff_ogm_text_codec,
|
|
|
|
&ff_ogm_old_codec,
|
2005-04-09 17:32:58 +02:00
|
|
|
NULL
|
|
|
|
};
|
|
|
|
|
2012-05-14 14:07:09 +02:00
|
|
|
static int64_t ogg_calc_pts(AVFormatContext *s, int idx, int64_t *dts);
|
2012-11-20 15:12:37 +01:00
|
|
|
static int ogg_read_close(AVFormatContext *s);
|
2012-05-14 14:07:09 +02:00
|
|
|
|
2005-04-09 17:32:58 +02:00
|
|
|
//FIXME We could avoid some structure duplication
|
2011-05-11 20:43:29 +02:00
|
|
|
static int ogg_save(AVFormatContext *s)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg *ogg = s->priv_data;
|
|
|
|
struct ogg_state *ost =
|
2005-04-10 01:27:48 +02:00
|
|
|
av_malloc(sizeof (*ost) + (ogg->nstreams-1) * sizeof (*ogg->streams));
|
2005-04-09 17:32:58 +02:00
|
|
|
int i;
|
2011-03-03 20:11:45 +01:00
|
|
|
ost->pos = avio_tell (s->pb);
|
2005-04-09 17:32:58 +02:00
|
|
|
ost->curidx = ogg->curidx;
|
|
|
|
ost->next = ogg->state;
|
2007-01-15 23:05:22 +01:00
|
|
|
ost->nstreams = ogg->nstreams;
|
2005-04-09 17:32:58 +02:00
|
|
|
memcpy(ost->streams, ogg->streams, ogg->nstreams * sizeof(*ogg->streams));
|
|
|
|
|
|
|
|
for (i = 0; i < ogg->nstreams; i++){
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg_stream *os = ogg->streams + i;
|
2012-03-07 23:26:58 +01:00
|
|
|
os->buf = av_mallocz (os->bufsize + FF_INPUT_BUFFER_PADDING_SIZE);
|
2005-04-09 17:32:58 +02:00
|
|
|
memcpy (os->buf, ost->streams[i].buf, os->bufpos);
|
|
|
|
}
|
|
|
|
|
|
|
|
ogg->state = ost;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2011-05-11 20:43:29 +02:00
|
|
|
static int ogg_restore(AVFormatContext *s, int discard)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg *ogg = s->priv_data;
|
2011-02-20 11:04:12 +01:00
|
|
|
AVIOContext *bc = s->pb;
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg_state *ost = ogg->state;
|
2005-04-09 17:32:58 +02:00
|
|
|
int i;
|
|
|
|
|
|
|
|
if (!ost)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
ogg->state = ost->next;
|
|
|
|
|
|
|
|
if (!discard){
|
2011-09-11 23:37:44 +02:00
|
|
|
struct ogg_stream *old_streams = ogg->streams;
|
|
|
|
|
2005-04-09 17:32:58 +02:00
|
|
|
for (i = 0; i < ogg->nstreams; i++)
|
|
|
|
av_free (ogg->streams[i].buf);
|
|
|
|
|
2011-02-28 14:57:54 +01:00
|
|
|
avio_seek (bc, ost->pos, SEEK_SET);
|
2005-04-09 17:32:58 +02:00
|
|
|
ogg->curidx = ost->curidx;
|
2007-01-15 23:05:22 +01:00
|
|
|
ogg->nstreams = ost->nstreams;
|
2011-09-11 23:37:44 +02:00
|
|
|
ogg->streams = av_realloc (ogg->streams,
|
|
|
|
ogg->nstreams * sizeof (*ogg->streams));
|
|
|
|
|
|
|
|
if (ogg->streams) {
|
|
|
|
memcpy(ogg->streams, ost->streams,
|
|
|
|
ost->nstreams * sizeof(*ogg->streams));
|
|
|
|
} else {
|
|
|
|
av_free(old_streams);
|
|
|
|
ogg->nstreams = 0;
|
|
|
|
}
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
av_free (ost);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2012-05-04 13:44:28 +02:00
|
|
|
static int ogg_reset(AVFormatContext *s)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2012-05-04 13:44:28 +02:00
|
|
|
struct ogg *ogg = s->priv_data;
|
2005-04-09 17:32:58 +02:00
|
|
|
int i;
|
2012-05-04 13:45:51 +02:00
|
|
|
int64_t start_pos = avio_tell(s->pb);
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
for (i = 0; i < ogg->nstreams; i++){
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg_stream *os = ogg->streams + i;
|
2005-04-09 17:32:58 +02:00
|
|
|
os->bufpos = 0;
|
|
|
|
os->pstart = 0;
|
|
|
|
os->psize = 0;
|
|
|
|
os->granule = -1;
|
2009-12-12 21:18:43 +01:00
|
|
|
os->lastpts = AV_NOPTS_VALUE;
|
2010-01-11 06:51:09 +01:00
|
|
|
os->lastdts = AV_NOPTS_VALUE;
|
2010-03-11 08:17:24 +01:00
|
|
|
os->sync_pos = -1;
|
|
|
|
os->page_pos = 0;
|
2005-04-09 17:32:58 +02:00
|
|
|
os->nsegs = 0;
|
|
|
|
os->segp = 0;
|
2010-02-08 11:13:03 +01:00
|
|
|
os->incomplete = 0;
|
2012-09-16 22:24:11 +02:00
|
|
|
os->got_data = 0;
|
2012-05-04 13:45:51 +02:00
|
|
|
if (start_pos <= s->data_offset) {
|
|
|
|
os->lastpts = 0;
|
|
|
|
}
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
ogg->curidx = -1;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2011-05-11 20:43:29 +02:00
|
|
|
static const struct ogg_codec *ogg_find_codec(uint8_t *buf, int size)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i = 0; ogg_codecs[i]; i++)
|
|
|
|
if (size >= ogg_codecs[i]->magicsize &&
|
|
|
|
!memcmp (buf, ogg_codecs[i]->magic, ogg_codecs[i]->magicsize))
|
|
|
|
return ogg_codecs[i];
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2012-09-15 00:41:08 +02:00
|
|
|
/**
|
|
|
|
* Replace the current stream with a new one. This is a typical webradio
|
|
|
|
* situation where a new audio stream spawn (identified with a new serial) and
|
|
|
|
* must replace the previous one (track switch).
|
|
|
|
*/
|
|
|
|
static int ogg_replace_stream(AVFormatContext *s, uint32_t serial)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2012-09-15 00:41:08 +02:00
|
|
|
struct ogg *ogg = s->priv_data;
|
|
|
|
struct ogg_stream *os;
|
|
|
|
unsigned bufsize;
|
|
|
|
uint8_t *buf;
|
2012-12-05 04:38:57 +01:00
|
|
|
struct ogg_codec *codec;
|
2012-09-15 00:41:08 +02:00
|
|
|
|
|
|
|
if (ogg->nstreams != 1) {
|
2012-10-21 23:13:53 +02:00
|
|
|
av_log_missing_feature(s, "Changing stream parameters in multistream ogg", 0);
|
2012-09-15 00:41:08 +02:00
|
|
|
return AVERROR_PATCHWELCOME;
|
|
|
|
}
|
|
|
|
|
|
|
|
os = &ogg->streams[0];
|
|
|
|
|
|
|
|
buf = os->buf;
|
|
|
|
bufsize = os->bufsize;
|
2012-12-05 04:38:57 +01:00
|
|
|
codec = os->codec;
|
2012-09-15 00:41:08 +02:00
|
|
|
|
|
|
|
if (!ogg->state || ogg->state->streams[0].private != os->private)
|
|
|
|
av_freep(&ogg->streams[0].private);
|
|
|
|
|
|
|
|
/* Set Ogg stream settings similar to what is done in ogg_new_stream(). We
|
|
|
|
* also re-use the ogg_stream allocated buffer */
|
|
|
|
memset(os, 0, sizeof(*os));
|
|
|
|
os->serial = serial;
|
|
|
|
os->bufsize = bufsize;
|
|
|
|
os->buf = buf;
|
|
|
|
os->header = -1;
|
2012-12-05 04:38:57 +01:00
|
|
|
os->codec = codec;
|
2012-09-15 00:41:08 +02:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
2005-04-09 17:32:58 +02:00
|
|
|
|
2012-09-15 00:41:08 +02:00
|
|
|
static int ogg_new_stream(AVFormatContext *s, uint32_t serial)
|
|
|
|
{
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg *ogg = s->priv_data;
|
2012-09-14 23:51:30 +02:00
|
|
|
int idx = ogg->nstreams;
|
2005-04-09 17:32:58 +02:00
|
|
|
AVStream *st;
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg_stream *os;
|
2012-09-14 23:51:30 +02:00
|
|
|
size_t size;
|
2005-04-09 17:32:58 +02:00
|
|
|
|
2012-09-16 22:24:11 +02:00
|
|
|
if (ogg->state) {
|
|
|
|
av_log(s, AV_LOG_ERROR, "New streams are not supposed to be added "
|
|
|
|
"in between Ogg context save/restore operations.\n");
|
|
|
|
return AVERROR_BUG;
|
|
|
|
}
|
|
|
|
|
2012-09-15 00:42:25 +02:00
|
|
|
/* Allocate and init a new Ogg Stream */
|
2012-09-14 23:51:30 +02:00
|
|
|
if (av_size_mult(ogg->nstreams + 1, sizeof(*ogg->streams), &size) < 0 ||
|
|
|
|
!(os = av_realloc(ogg->streams, size)))
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
ogg->streams = os;
|
2005-04-09 17:32:58 +02:00
|
|
|
os = ogg->streams + idx;
|
2012-09-14 23:54:19 +02:00
|
|
|
memset(os, 0, sizeof(*os));
|
2005-04-09 17:32:58 +02:00
|
|
|
os->serial = serial;
|
|
|
|
os->bufsize = DECODER_BUFFER_SIZE;
|
2012-03-07 23:26:58 +01:00
|
|
|
os->buf = av_malloc(os->bufsize + FF_INPUT_BUFFER_PADDING_SIZE);
|
2005-04-09 17:32:58 +02:00
|
|
|
os->header = -1;
|
2012-09-20 01:07:09 +02:00
|
|
|
os->start_granule = OGG_NOGRANULE_VALUE;
|
2012-09-14 23:51:30 +02:00
|
|
|
if (!os->buf)
|
|
|
|
return AVERROR(ENOMEM);
|
2005-04-09 17:32:58 +02:00
|
|
|
|
2012-09-15 00:42:25 +02:00
|
|
|
/* Create the associated AVStream */
|
|
|
|
st = avformat_new_stream(s, NULL);
|
|
|
|
if (!st) {
|
|
|
|
av_freep(&os->buf);
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
}
|
|
|
|
st->id = idx;
|
|
|
|
avpriv_set_pts_info(st, 64, 1, 1000000);
|
2005-04-09 17:32:58 +02:00
|
|
|
|
2012-09-14 23:51:30 +02:00
|
|
|
ogg->nstreams++;
|
2005-04-09 17:32:58 +02:00
|
|
|
return idx;
|
|
|
|
}
|
|
|
|
|
2011-05-11 20:43:29 +02:00
|
|
|
static int ogg_new_buf(struct ogg *ogg, int idx)
|
2006-06-25 02:41:13 +02:00
|
|
|
{
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg_stream *os = ogg->streams + idx;
|
2012-03-07 23:26:58 +01:00
|
|
|
uint8_t *nb = av_malloc(os->bufsize + FF_INPUT_BUFFER_PADDING_SIZE);
|
2006-06-25 02:41:13 +02:00
|
|
|
int size = os->bufpos - os->pstart;
|
|
|
|
if(os->buf){
|
|
|
|
memcpy(nb, os->buf + os->pstart, size);
|
|
|
|
av_free(os->buf);
|
|
|
|
}
|
|
|
|
os->buf = nb;
|
|
|
|
os->bufpos = size;
|
|
|
|
os->pstart = 0;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2012-09-16 22:24:11 +02:00
|
|
|
static int data_packets_seen(const struct ogg *ogg)
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i = 0; i < ogg->nstreams; i++)
|
|
|
|
if (ogg->streams[i].got_data)
|
|
|
|
return 1;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2012-09-14 23:10:38 +02:00
|
|
|
static int ogg_read_page(AVFormatContext *s, int *sid)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2011-02-20 11:04:12 +01:00
|
|
|
AVIOContext *bc = s->pb;
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg *ogg = s->priv_data;
|
|
|
|
struct ogg_stream *os;
|
2011-04-15 17:08:12 +02:00
|
|
|
int ret, i = 0;
|
2005-04-09 17:32:58 +02:00
|
|
|
int flags, nsegs;
|
|
|
|
uint64_t gp;
|
|
|
|
uint32_t serial;
|
|
|
|
int size, idx;
|
2006-09-27 21:47:39 +02:00
|
|
|
uint8_t sync[4];
|
2005-04-09 17:32:58 +02:00
|
|
|
int sp = 0;
|
|
|
|
|
2011-06-24 12:02:41 +02:00
|
|
|
ret = avio_read(bc, sync, 4);
|
2011-04-15 17:08:12 +02:00
|
|
|
if (ret < 4)
|
|
|
|
return ret < 0 ? ret : AVERROR_EOF;
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
do{
|
|
|
|
int c;
|
|
|
|
|
|
|
|
if (sync[sp & 3] == 'O' &&
|
|
|
|
sync[(sp + 1) & 3] == 'g' &&
|
|
|
|
sync[(sp + 2) & 3] == 'g' && sync[(sp + 3) & 3] == 'S')
|
|
|
|
break;
|
|
|
|
|
2011-03-06 19:59:29 +01:00
|
|
|
c = avio_r8(bc);
|
|
|
|
if (url_feof(bc))
|
2011-04-15 17:08:12 +02:00
|
|
|
return AVERROR_EOF;
|
2005-04-09 17:32:58 +02:00
|
|
|
sync[sp++ & 3] = c;
|
|
|
|
}while (i++ < MAX_PAGE_SIZE);
|
|
|
|
|
|
|
|
if (i >= MAX_PAGE_SIZE){
|
|
|
|
av_log (s, AV_LOG_INFO, "ogg, can't find sync word\n");
|
2011-04-15 17:08:12 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
|
2012-05-04 13:48:58 +02:00
|
|
|
if (avio_r8(bc) != 0){ /* version */
|
|
|
|
av_log (s, AV_LOG_ERROR, "ogg page, unsupported version\n");
|
2011-04-15 17:08:12 +02:00
|
|
|
return AVERROR_INVALIDDATA;
|
2012-05-04 13:48:58 +02:00
|
|
|
}
|
2005-04-09 17:32:58 +02:00
|
|
|
|
2011-03-06 19:59:29 +01:00
|
|
|
flags = avio_r8(bc);
|
2011-02-21 16:43:01 +01:00
|
|
|
gp = avio_rl64 (bc);
|
|
|
|
serial = avio_rl32 (bc);
|
2011-06-01 18:26:27 +02:00
|
|
|
avio_skip(bc, 8); /* seq, crc */
|
2011-03-06 19:59:29 +01:00
|
|
|
nsegs = avio_r8(bc);
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
idx = ogg_find_stream (ogg, serial);
|
|
|
|
if (idx < 0){
|
2012-09-16 22:24:11 +02:00
|
|
|
if (data_packets_seen(ogg))
|
2012-09-15 00:41:08 +02:00
|
|
|
idx = ogg_replace_stream(s, serial);
|
|
|
|
else
|
|
|
|
idx = ogg_new_stream(s, serial);
|
2011-04-24 15:40:57 +02:00
|
|
|
|
2012-05-04 13:53:57 +02:00
|
|
|
if (idx < 0) {
|
2012-09-15 02:59:23 +02:00
|
|
|
av_log(s, AV_LOG_ERROR, "failed to create or replace stream\n");
|
2011-04-15 17:08:12 +02:00
|
|
|
return idx;
|
2012-05-04 13:53:57 +02:00
|
|
|
}
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
os = ogg->streams + idx;
|
2011-03-03 20:11:45 +01:00
|
|
|
os->page_pos = avio_tell(bc) - 27;
|
2005-04-09 17:32:58 +02:00
|
|
|
|
2006-06-25 14:23:54 +02:00
|
|
|
if(os->psize > 0)
|
2006-06-25 02:41:13 +02:00
|
|
|
ogg_new_buf(ogg, idx);
|
|
|
|
|
2011-06-24 12:02:41 +02:00
|
|
|
ret = avio_read(bc, os->segments, nsegs);
|
2011-04-15 17:08:12 +02:00
|
|
|
if (ret < nsegs)
|
|
|
|
return ret < 0 ? ret : AVERROR_EOF;
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
os->nsegs = nsegs;
|
|
|
|
os->segp = 0;
|
|
|
|
|
|
|
|
size = 0;
|
|
|
|
for (i = 0; i < nsegs; i++)
|
|
|
|
size += os->segments[i];
|
|
|
|
|
2012-09-16 22:24:11 +02:00
|
|
|
if (!(flags & OGG_FLAG_BOS))
|
|
|
|
os->got_data = 1;
|
|
|
|
|
2010-02-08 11:13:03 +01:00
|
|
|
if (flags & OGG_FLAG_CONT || os->incomplete){
|
2005-04-09 17:32:58 +02:00
|
|
|
if (!os->psize){
|
2012-02-04 15:27:34 +01:00
|
|
|
// If this is the very first segment we started
|
|
|
|
// playback in the middle of a continuation packet.
|
|
|
|
// Discard it since we missed the start of it.
|
2005-04-09 17:32:58 +02:00
|
|
|
while (os->segp < os->nsegs){
|
|
|
|
int seg = os->segments[os->segp++];
|
|
|
|
os->pstart += seg;
|
|
|
|
if (seg < 255)
|
2009-02-14 14:44:21 +01:00
|
|
|
break;
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
2010-03-11 08:17:24 +01:00
|
|
|
os->sync_pos = os->page_pos;
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
}else{
|
2009-02-14 14:44:21 +01:00
|
|
|
os->psize = 0;
|
2010-03-11 08:17:24 +01:00
|
|
|
os->sync_pos = os->page_pos;
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (os->bufsize - os->bufpos < size){
|
2012-03-07 23:26:58 +01:00
|
|
|
uint8_t *nb = av_malloc ((os->bufsize *= 2) + FF_INPUT_BUFFER_PADDING_SIZE);
|
2005-04-09 17:32:58 +02:00
|
|
|
memcpy (nb, os->buf, os->bufpos);
|
|
|
|
av_free (os->buf);
|
|
|
|
os->buf = nb;
|
|
|
|
}
|
|
|
|
|
2011-06-24 12:02:41 +02:00
|
|
|
ret = avio_read(bc, os->buf + os->bufpos, size);
|
2011-04-15 17:08:12 +02:00
|
|
|
if (ret < size)
|
|
|
|
return ret < 0 ? ret : AVERROR_EOF;
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
os->bufpos += size;
|
|
|
|
os->granule = gp;
|
|
|
|
os->flags = flags;
|
|
|
|
|
2012-03-07 23:26:58 +01:00
|
|
|
memset(os->buf + os->bufpos, 0, FF_INPUT_BUFFER_PADDING_SIZE);
|
2012-09-14 23:10:38 +02:00
|
|
|
if (sid)
|
|
|
|
*sid = idx;
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2012-08-05 09:45:04 +02:00
|
|
|
/**
|
|
|
|
* @brief find the next Ogg packet
|
2012-09-14 23:10:38 +02:00
|
|
|
* @param *sid is set to the stream for the packet or -1 if there is
|
2012-08-05 09:45:04 +02:00
|
|
|
* no matching stream, in that case assume all other return
|
|
|
|
* values to be uninitialized.
|
|
|
|
* @return negative value on error or EOF.
|
|
|
|
*/
|
2012-09-14 23:10:38 +02:00
|
|
|
static int ogg_packet(AVFormatContext *s, int *sid, int *dstart, int *dsize,
|
2011-05-11 20:43:29 +02:00
|
|
|
int64_t *fpos)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg *ogg = s->priv_data;
|
2011-04-15 17:08:12 +02:00
|
|
|
int idx, i, ret;
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg_stream *os;
|
2005-04-09 17:32:58 +02:00
|
|
|
int complete = 0;
|
|
|
|
int segp = 0, psize = 0;
|
|
|
|
|
2011-05-23 23:51:39 +02:00
|
|
|
av_dlog(s, "ogg_packet: curidx=%i\n", ogg->curidx);
|
2012-09-14 23:10:38 +02:00
|
|
|
if (sid)
|
|
|
|
*sid = -1;
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
do{
|
|
|
|
idx = ogg->curidx;
|
|
|
|
|
|
|
|
while (idx < 0){
|
2011-06-24 12:02:41 +02:00
|
|
|
ret = ogg_read_page(s, &idx);
|
2011-04-15 17:08:12 +02:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
os = ogg->streams + idx;
|
|
|
|
|
2011-05-23 23:51:39 +02:00
|
|
|
av_dlog(s, "ogg_packet: idx=%d pstart=%d psize=%d segp=%d nsegs=%d\n",
|
2005-04-09 17:32:58 +02:00
|
|
|
idx, os->pstart, os->psize, os->segp, os->nsegs);
|
|
|
|
|
|
|
|
if (!os->codec){
|
|
|
|
if (os->header < 0){
|
|
|
|
os->codec = ogg_find_codec (os->buf, os->bufpos);
|
|
|
|
if (!os->codec){
|
2011-04-15 17:08:12 +02:00
|
|
|
av_log(s, AV_LOG_WARNING, "Codec not found\n");
|
2005-04-09 17:32:58 +02:00
|
|
|
os->header = 0;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}else{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
segp = os->segp;
|
|
|
|
psize = os->psize;
|
|
|
|
|
|
|
|
while (os->segp < os->nsegs){
|
|
|
|
int ss = os->segments[os->segp++];
|
|
|
|
os->psize += ss;
|
|
|
|
if (ss < 255){
|
|
|
|
complete = 1;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!complete && os->segp == os->nsegs){
|
|
|
|
ogg->curidx = -1;
|
2012-02-04 15:27:34 +01:00
|
|
|
// Do not set incomplete for empty packets.
|
|
|
|
// Together with the code in ogg_read_page
|
|
|
|
// that discards all continuation of empty packets
|
|
|
|
// we would get an infinite loop.
|
|
|
|
os->incomplete = !!os->psize;
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
}while (!complete);
|
|
|
|
|
|
|
|
|
2010-04-16 14:21:38 +02:00
|
|
|
if (os->granule == -1)
|
2010-06-09 10:59:41 +02:00
|
|
|
av_log(s, AV_LOG_WARNING, "Page at %"PRId64" is missing granule\n", os->page_pos);
|
2010-04-16 14:21:38 +02:00
|
|
|
|
2005-04-09 17:32:58 +02:00
|
|
|
ogg->curidx = idx;
|
2010-02-08 11:13:03 +01:00
|
|
|
os->incomplete = 0;
|
2005-04-09 17:32:58 +02:00
|
|
|
|
2010-03-12 06:17:01 +01:00
|
|
|
if (os->header) {
|
|
|
|
os->header = os->codec->header (s, idx);
|
|
|
|
if (!os->header){
|
2009-02-14 14:44:21 +01:00
|
|
|
os->segp = segp;
|
|
|
|
os->psize = psize;
|
2010-11-29 20:02:28 +01:00
|
|
|
|
2011-02-05 17:32:35 +01:00
|
|
|
// We have reached the first non-header packet in this stream.
|
|
|
|
// Unfortunately more header packets may still follow for others,
|
2011-04-08 01:17:21 +02:00
|
|
|
// but if we continue with header parsing we may lose data packets.
|
2009-02-14 14:44:21 +01:00
|
|
|
ogg->headers = 1;
|
2010-11-29 20:02:28 +01:00
|
|
|
|
|
|
|
// Update the header state for all streams and
|
|
|
|
// compute the data_offset.
|
2011-02-05 17:32:35 +01:00
|
|
|
if (!s->data_offset)
|
|
|
|
s->data_offset = os->sync_pos;
|
2010-11-29 20:02:28 +01:00
|
|
|
for (i = 0; i < ogg->nstreams; i++) {
|
|
|
|
struct ogg_stream *cur_os = ogg->streams + i;
|
|
|
|
|
|
|
|
// if we have a partial non-header packet, its start is
|
|
|
|
// obviously at or after the data start
|
|
|
|
if (cur_os->incomplete)
|
|
|
|
s->data_offset = FFMIN(s->data_offset, cur_os->sync_pos);
|
|
|
|
}
|
2005-04-09 17:32:58 +02:00
|
|
|
}else{
|
2012-09-20 01:01:43 +02:00
|
|
|
os->nb_header++;
|
2009-02-14 14:44:21 +01:00
|
|
|
os->pstart += os->psize;
|
|
|
|
os->psize = 0;
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
2010-03-12 06:17:01 +01:00
|
|
|
} else {
|
2007-11-11 22:56:18 +01:00
|
|
|
os->pflags = 0;
|
2009-10-12 23:30:03 +02:00
|
|
|
os->pduration = 0;
|
2005-04-09 17:32:58 +02:00
|
|
|
if (os->codec && os->codec->packet)
|
|
|
|
os->codec->packet (s, idx);
|
2012-09-14 23:10:38 +02:00
|
|
|
if (sid)
|
|
|
|
*sid = idx;
|
2006-06-25 02:41:13 +02:00
|
|
|
if (dstart)
|
|
|
|
*dstart = os->pstart;
|
|
|
|
if (dsize)
|
|
|
|
*dsize = os->psize;
|
2010-03-11 08:17:24 +01:00
|
|
|
if (fpos)
|
|
|
|
*fpos = os->sync_pos;
|
2006-06-25 02:41:13 +02:00
|
|
|
os->pstart += os->psize;
|
|
|
|
os->psize = 0;
|
2011-06-03 21:14:30 +02:00
|
|
|
if(os->pstart == os->bufpos)
|
|
|
|
os->bufpos = os->pstart = 0;
|
2010-03-11 08:17:24 +01:00
|
|
|
os->sync_pos = os->page_pos;
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
|
2009-12-12 21:18:43 +01:00
|
|
|
// determine whether there are more complete packets in this page
|
|
|
|
// if not, the page's granule will apply to this packet
|
|
|
|
os->page_end = 1;
|
|
|
|
for (i = os->segp; i < os->nsegs; i++)
|
|
|
|
if (os->segments[i] < 255) {
|
|
|
|
os->page_end = 0;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2005-04-09 17:32:58 +02:00
|
|
|
if (os->segp == os->nsegs)
|
|
|
|
ogg->curidx = -1;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2011-05-11 20:43:29 +02:00
|
|
|
static int ogg_get_length(AVFormatContext *s)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg *ogg = s->priv_data;
|
2010-04-16 14:21:35 +02:00
|
|
|
int i;
|
2008-10-03 12:16:29 +02:00
|
|
|
int64_t size, end;
|
2011-08-24 23:13:57 +02:00
|
|
|
int streams_left=0;
|
2005-05-19 23:03:03 +02:00
|
|
|
|
2011-03-05 21:06:46 +01:00
|
|
|
if(!s->pb->seekable)
|
2005-05-19 23:03:03 +02:00
|
|
|
return 0;
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
// already set
|
|
|
|
if (s->duration != AV_NOPTS_VALUE)
|
|
|
|
return 0;
|
|
|
|
|
2011-03-04 19:57:36 +01:00
|
|
|
size = avio_size(s->pb);
|
2006-06-04 17:06:18 +02:00
|
|
|
if(size < 0)
|
|
|
|
return 0;
|
2009-04-15 08:41:08 +02:00
|
|
|
end = size > MAX_PAGE_SIZE? size - MAX_PAGE_SIZE: 0;
|
2006-06-04 17:06:18 +02:00
|
|
|
|
2005-04-09 17:32:58 +02:00
|
|
|
ogg_save (s);
|
2011-02-28 14:57:54 +01:00
|
|
|
avio_seek (s->pb, end, SEEK_SET);
|
2005-04-09 17:32:58 +02:00
|
|
|
|
|
|
|
while (!ogg_read_page (s, &i)){
|
2007-01-15 23:45:50 +01:00
|
|
|
if (ogg->streams[i].granule != -1 && ogg->streams[i].granule != 0 &&
|
2010-04-16 14:21:35 +02:00
|
|
|
ogg->streams[i].codec) {
|
|
|
|
s->streams[i]->duration =
|
|
|
|
ogg_gptopts (s, i, ogg->streams[i].granule, NULL);
|
2011-08-24 23:13:57 +02:00
|
|
|
if (s->streams[i]->start_time != AV_NOPTS_VALUE){
|
2010-04-16 14:21:35 +02:00
|
|
|
s->streams[i]->duration -= s->streams[i]->start_time;
|
2011-08-24 23:13:57 +02:00
|
|
|
streams_left-= (ogg->streams[i].got_start==-1);
|
|
|
|
ogg->streams[i].got_start= 1;
|
|
|
|
}else if(!ogg->streams[i].got_start){
|
|
|
|
ogg->streams[i].got_start= -1;
|
|
|
|
streams_left++;
|
|
|
|
}
|
2010-04-16 14:21:35 +02:00
|
|
|
}
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
ogg_restore (s, 0);
|
|
|
|
|
2011-05-07 15:31:04 +02:00
|
|
|
ogg_save (s);
|
2012-05-14 14:03:07 +02:00
|
|
|
avio_seek (s->pb, s->data_offset, SEEK_SET);
|
2012-05-14 14:07:09 +02:00
|
|
|
ogg_reset(s);
|
2012-08-05 09:41:20 +02:00
|
|
|
while (streams_left > 0 && !ogg_packet(s, &i, NULL, NULL, NULL)) {
|
|
|
|
int64_t pts;
|
|
|
|
if (i < 0) continue;
|
|
|
|
pts = ogg_calc_pts(s, i, NULL);
|
2012-05-14 14:07:09 +02:00
|
|
|
if (pts != AV_NOPTS_VALUE && s->streams[i]->start_time == AV_NOPTS_VALUE && !ogg->streams[i].got_start){
|
|
|
|
s->streams[i]->duration -= pts;
|
|
|
|
ogg->streams[i].got_start= 1;
|
|
|
|
streams_left--;
|
2012-06-02 22:50:40 +02:00
|
|
|
}else if(s->streams[i]->start_time != AV_NOPTS_VALUE && !ogg->streams[i].got_start){
|
|
|
|
ogg->streams[i].got_start= 1;
|
|
|
|
streams_left--;
|
2012-08-05 04:41:34 +02:00
|
|
|
}
|
2011-05-07 15:31:04 +02:00
|
|
|
}
|
|
|
|
ogg_restore (s, 0);
|
|
|
|
|
2005-04-09 17:32:58 +02:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2012-01-12 13:20:36 +01:00
|
|
|
static int ogg_read_header(AVFormatContext *s)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg *ogg = s->priv_data;
|
2011-04-15 17:08:12 +02:00
|
|
|
int ret, i;
|
2012-09-14 23:05:57 +02:00
|
|
|
|
2005-04-09 17:32:58 +02:00
|
|
|
ogg->curidx = -1;
|
2012-09-14 23:05:57 +02:00
|
|
|
|
2005-04-09 17:32:58 +02:00
|
|
|
//linear headers seek from start
|
2012-09-14 23:05:57 +02:00
|
|
|
do {
|
|
|
|
ret = ogg_packet(s, NULL, NULL, NULL, NULL);
|
2012-11-20 15:12:37 +01:00
|
|
|
if (ret < 0) {
|
|
|
|
ogg_read_close(s);
|
2012-09-14 23:05:57 +02:00
|
|
|
return ret;
|
2012-11-20 15:12:37 +01:00
|
|
|
}
|
2012-09-14 23:05:57 +02:00
|
|
|
} while (!ogg->headers);
|
|
|
|
av_dlog(s, "found headers\n");
|
2005-04-09 17:32:58 +02:00
|
|
|
|
2012-09-25 15:34:58 +02:00
|
|
|
for (i = 0; i < ogg->nstreams; i++) {
|
|
|
|
struct ogg_stream *os = ogg->streams + i;
|
|
|
|
|
2012-09-25 15:22:05 +02:00
|
|
|
if (ogg->streams[i].header < 0) {
|
|
|
|
av_log(s, AV_LOG_ERROR, "Header parsing failed for stream %d\n", i);
|
2009-09-30 11:46:48 +02:00
|
|
|
ogg->streams[i].codec = NULL;
|
2012-09-25 15:34:58 +02:00
|
|
|
} else if (os->codec && os->nb_header < os->codec->nb_header) {
|
|
|
|
av_log(s, AV_LOG_WARNING, "Number of headers (%d) mismatch for stream %d\n", os->nb_header, i);
|
2012-09-25 15:22:05 +02:00
|
|
|
}
|
2012-09-25 16:00:09 +02:00
|
|
|
if (os->start_granule != OGG_NOGRANULE_VALUE)
|
|
|
|
os->lastpts = s->streams[i]->start_time =
|
|
|
|
ogg_gptopts(s, i, os->start_granule, NULL);
|
2012-09-25 15:34:58 +02:00
|
|
|
}
|
2009-09-30 11:46:48 +02:00
|
|
|
|
2005-04-09 17:32:58 +02:00
|
|
|
//linear granulepos seek from end
|
|
|
|
ogg_get_length (s);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2010-03-11 08:17:53 +01:00
|
|
|
static int64_t ogg_calc_pts(AVFormatContext *s, int idx, int64_t *dts)
|
|
|
|
{
|
|
|
|
struct ogg *ogg = s->priv_data;
|
|
|
|
struct ogg_stream *os = ogg->streams + idx;
|
|
|
|
int64_t pts = AV_NOPTS_VALUE;
|
|
|
|
|
|
|
|
if (dts)
|
|
|
|
*dts = AV_NOPTS_VALUE;
|
|
|
|
|
|
|
|
if (os->lastpts != AV_NOPTS_VALUE) {
|
|
|
|
pts = os->lastpts;
|
|
|
|
os->lastpts = AV_NOPTS_VALUE;
|
|
|
|
}
|
|
|
|
if (os->lastdts != AV_NOPTS_VALUE) {
|
|
|
|
if (dts)
|
|
|
|
*dts = os->lastdts;
|
|
|
|
os->lastdts = AV_NOPTS_VALUE;
|
|
|
|
}
|
|
|
|
if (os->page_end) {
|
|
|
|
if (os->granule != -1LL) {
|
|
|
|
if (os->codec && os->codec->granule_is_start)
|
|
|
|
pts = ogg_gptopts(s, idx, os->granule, dts);
|
|
|
|
else
|
|
|
|
os->lastpts = ogg_gptopts(s, idx, os->granule, &os->lastdts);
|
|
|
|
os->granule = -1LL;
|
2010-04-16 14:21:38 +02:00
|
|
|
}
|
2010-03-11 08:17:53 +01:00
|
|
|
}
|
|
|
|
return pts;
|
|
|
|
}
|
2005-04-09 17:32:58 +02:00
|
|
|
|
2012-02-06 22:03:25 +01:00
|
|
|
static void ogg_validate_keyframe(AVFormatContext *s, int idx, int pstart, int psize)
|
|
|
|
{
|
|
|
|
struct ogg *ogg = s->priv_data;
|
|
|
|
struct ogg_stream *os = ogg->streams + idx;
|
2012-08-07 22:45:46 +02:00
|
|
|
if (psize && s->streams[idx]->codec->codec_id == AV_CODEC_ID_THEORA) {
|
2012-02-06 22:03:25 +01:00
|
|
|
if (!!(os->pflags & AV_PKT_FLAG_KEY) != !(os->buf[pstart] & 0x40)) {
|
|
|
|
os->pflags ^= AV_PKT_FLAG_KEY;
|
2012-02-18 21:19:58 +01:00
|
|
|
av_log(s, AV_LOG_WARNING, "Broken file, %skeyframe not correctly marked.\n",
|
|
|
|
(os->pflags & AV_PKT_FLAG_KEY) ? "" : "non-");
|
2012-02-06 22:03:25 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-05-11 20:43:29 +02:00
|
|
|
static int ogg_read_packet(AVFormatContext *s, AVPacket *pkt)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg *ogg;
|
|
|
|
struct ogg_stream *os;
|
2012-08-05 09:45:04 +02:00
|
|
|
int idx, ret;
|
2006-06-25 02:41:13 +02:00
|
|
|
int pstart, psize;
|
2010-03-11 08:18:00 +01:00
|
|
|
int64_t fpos, pts, dts;
|
2005-04-09 17:32:58 +02:00
|
|
|
|
2005-12-17 19:14:38 +01:00
|
|
|
//Get an ogg packet
|
2010-03-11 08:18:00 +01:00
|
|
|
retry:
|
2005-04-09 17:32:58 +02:00
|
|
|
do{
|
2011-06-24 12:02:41 +02:00
|
|
|
ret = ogg_packet(s, &idx, &pstart, &psize, &fpos);
|
2011-06-24 12:02:41 +02:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2005-04-09 17:32:58 +02:00
|
|
|
}while (idx < 0 || !s->streams[idx]);
|
|
|
|
|
|
|
|
ogg = s->priv_data;
|
|
|
|
os = ogg->streams + idx;
|
|
|
|
|
2010-03-11 08:18:00 +01:00
|
|
|
// pflags might not be set until after this
|
|
|
|
pts = ogg_calc_pts(s, idx, &dts);
|
2012-02-06 22:03:25 +01:00
|
|
|
ogg_validate_keyframe(s, idx, pstart, psize);
|
2010-03-11 08:18:00 +01:00
|
|
|
|
2010-03-31 14:29:58 +02:00
|
|
|
if (os->keyframe_seek && !(os->pflags & AV_PKT_FLAG_KEY))
|
2010-03-11 08:18:00 +01:00
|
|
|
goto retry;
|
|
|
|
os->keyframe_seek = 0;
|
|
|
|
|
2005-04-09 17:32:58 +02:00
|
|
|
//Alloc a pkt
|
2011-06-24 12:02:41 +02:00
|
|
|
ret = av_new_packet(pkt, psize);
|
2011-06-24 12:02:41 +02:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
2005-04-09 17:32:58 +02:00
|
|
|
pkt->stream_index = idx;
|
2006-06-25 02:41:13 +02:00
|
|
|
memcpy (pkt->data, os->buf + pstart, psize);
|
2009-12-12 21:18:43 +01:00
|
|
|
|
2010-03-11 08:18:00 +01:00
|
|
|
pkt->pts = pts;
|
|
|
|
pkt->dts = dts;
|
2007-11-11 22:56:18 +01:00
|
|
|
pkt->flags = os->pflags;
|
2009-10-12 23:30:03 +02:00
|
|
|
pkt->duration = os->pduration;
|
2010-03-11 08:17:24 +01:00
|
|
|
pkt->pos = fpos;
|
2007-11-11 22:56:18 +01:00
|
|
|
|
2006-06-25 02:41:13 +02:00
|
|
|
return psize;
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
|
2011-05-11 20:43:29 +02:00
|
|
|
static int ogg_read_close(AVFormatContext *s)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg *ogg = s->priv_data;
|
2005-04-09 17:32:58 +02:00
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i = 0; i < ogg->nstreams; i++){
|
|
|
|
av_free (ogg->streams[i].buf);
|
2005-05-11 18:38:34 +02:00
|
|
|
av_free (ogg->streams[i].private);
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
av_free (ogg->streams);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2011-05-11 20:43:29 +02:00
|
|
|
static int64_t ogg_read_timestamp(AVFormatContext *s, int stream_index,
|
|
|
|
int64_t *pos_arg, int64_t pos_limit)
|
2005-04-09 17:32:58 +02:00
|
|
|
{
|
2008-11-06 02:50:56 +01:00
|
|
|
struct ogg *ogg = s->priv_data;
|
2011-02-20 11:04:12 +01:00
|
|
|
AVIOContext *bc = s->pb;
|
2007-10-10 12:33:07 +02:00
|
|
|
int64_t pts = AV_NOPTS_VALUE;
|
2012-02-04 20:31:21 +01:00
|
|
|
int64_t keypos = -1;
|
2012-08-05 09:45:04 +02:00
|
|
|
int i;
|
2012-02-06 22:03:25 +01:00
|
|
|
int pstart, psize;
|
2011-02-28 14:57:54 +01:00
|
|
|
avio_seek(bc, *pos_arg, SEEK_SET);
|
2012-05-04 13:44:28 +02:00
|
|
|
ogg_reset(s);
|
2010-03-11 08:17:56 +01:00
|
|
|
|
2012-05-04 13:59:14 +02:00
|
|
|
while (avio_tell(bc) <= pos_limit && !ogg_packet(s, &i, &pstart, &psize, pos_arg)) {
|
2010-03-11 08:17:56 +01:00
|
|
|
if (i == stream_index) {
|
2011-06-30 00:44:40 +02:00
|
|
|
struct ogg_stream *os = ogg->streams + stream_index;
|
2010-03-11 08:17:56 +01:00
|
|
|
pts = ogg_calc_pts(s, i, NULL);
|
2012-02-06 22:03:25 +01:00
|
|
|
ogg_validate_keyframe(s, i, pstart, psize);
|
2012-02-04 20:31:21 +01:00
|
|
|
if (os->pflags & AV_PKT_FLAG_KEY) {
|
|
|
|
keypos = *pos_arg;
|
|
|
|
} else if (os->keyframe_seek) {
|
|
|
|
// if we had a previous keyframe but no pts for it,
|
|
|
|
// return that keyframe with this pts value.
|
|
|
|
if (keypos >= 0)
|
|
|
|
*pos_arg = keypos;
|
|
|
|
else
|
|
|
|
pts = AV_NOPTS_VALUE;
|
|
|
|
}
|
2007-10-10 12:33:07 +02:00
|
|
|
}
|
2010-03-11 08:17:56 +01:00
|
|
|
if (pts != AV_NOPTS_VALUE)
|
|
|
|
break;
|
2007-10-10 12:33:07 +02:00
|
|
|
}
|
2012-05-04 13:44:28 +02:00
|
|
|
ogg_reset(s);
|
2007-10-10 12:33:07 +02:00
|
|
|
return pts;
|
2005-04-09 17:32:58 +02:00
|
|
|
}
|
|
|
|
|
2011-05-11 20:43:29 +02:00
|
|
|
static int ogg_read_seek(AVFormatContext *s, int stream_index,
|
|
|
|
int64_t timestamp, int flags)
|
2010-03-11 08:18:00 +01:00
|
|
|
{
|
|
|
|
struct ogg *ogg = s->priv_data;
|
|
|
|
struct ogg_stream *os = ogg->streams + stream_index;
|
|
|
|
int ret;
|
|
|
|
|
2012-04-17 14:44:06 +02:00
|
|
|
av_assert0(stream_index < ogg->nstreams);
|
2012-02-04 21:04:19 +01:00
|
|
|
// Ensure everything is reset even when seeking via
|
|
|
|
// the generated index.
|
2012-05-04 13:44:28 +02:00
|
|
|
ogg_reset(s);
|
2012-02-04 21:04:19 +01:00
|
|
|
|
2010-03-11 08:18:00 +01:00
|
|
|
// Try seeking to a keyframe first. If this fails (very possible),
|
|
|
|
// av_seek_frame will fall back to ignoring keyframes
|
2010-03-31 01:30:55 +02:00
|
|
|
if (s->streams[stream_index]->codec->codec_type == AVMEDIA_TYPE_VIDEO
|
2010-03-11 08:18:00 +01:00
|
|
|
&& !(flags & AVSEEK_FLAG_ANY))
|
|
|
|
os->keyframe_seek = 1;
|
|
|
|
|
2011-10-16 15:03:30 +02:00
|
|
|
ret = ff_seek_frame_binary(s, stream_index, timestamp, flags);
|
2011-06-30 00:44:40 +02:00
|
|
|
os = ogg->streams + stream_index;
|
2010-03-11 08:18:00 +01:00
|
|
|
if (ret < 0)
|
|
|
|
os->keyframe_seek = 0;
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2005-12-05 01:15:51 +01:00
|
|
|
static int ogg_probe(AVProbeData *p)
|
|
|
|
{
|
2011-05-11 20:43:30 +02:00
|
|
|
if (!memcmp("OggS", p->buf, 5) && p->buf[5] <= 0x7)
|
2005-12-05 01:15:51 +01:00
|
|
|
return AVPROBE_SCORE_MAX;
|
2011-05-11 20:43:30 +02:00
|
|
|
return 0;
|
2005-12-05 01:15:51 +01:00
|
|
|
}
|
|
|
|
|
2011-01-25 23:03:28 +01:00
|
|
|
AVInputFormat ff_ogg_demuxer = {
|
2011-05-11 20:43:31 +02:00
|
|
|
.name = "ogg",
|
|
|
|
.long_name = NULL_IF_CONFIG_SMALL("Ogg"),
|
|
|
|
.priv_data_size = sizeof(struct ogg),
|
|
|
|
.read_probe = ogg_probe,
|
|
|
|
.read_header = ogg_read_header,
|
|
|
|
.read_packet = ogg_read_packet,
|
|
|
|
.read_close = ogg_read_close,
|
|
|
|
.read_seek = ogg_read_seek,
|
|
|
|
.read_timestamp = ogg_read_timestamp,
|
|
|
|
.extensions = "ogg",
|
|
|
|
.flags = AVFMT_GENERIC_INDEX,
|
2005-04-09 17:32:58 +02:00
|
|
|
};
|