2 * Ogg bitstream support
3 * Luca Barbato <lu_zero@gentoo.org>
4 * Based on tcvp implementation
9 Copyright (C) 2005 Michael Ahlberg, Måns Rullgård
11 Permission is hereby granted, free of charge, to any person
12 obtaining a copy of this software and associated documentation
13 files (the "Software"), to deal in the Software without
14 restriction, including without limitation the rights to use, copy,
15 modify, merge, publish, distribute, sublicense, and/or sell copies
16 of the Software, and to permit persons to whom the Software is
17 furnished to do so, subject to the following conditions:
19 The above copyright notice and this permission notice shall be
20 included in all copies or substantial portions of the Software.
22 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
23 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
24 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
25 NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
26 HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
27 WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
28 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
29 DEALINGS IN THE SOFTWARE.
36 #include "vorbiscomment.h"
38 #define MAX_PAGE_SIZE 65307
39 #define DECODER_BUFFER_SIZE MAX_PAGE_SIZE
41 static const struct ogg_codec * const ogg_codecs[] = {
57 //FIXME We could avoid some structure duplication
59 ogg_save (AVFormatContext * s)
61 struct ogg *ogg = s->priv_data;
62 struct ogg_state *ost =
63 av_malloc(sizeof (*ost) + (ogg->nstreams-1) * sizeof (*ogg->streams));
65 ost->pos = url_ftell (s->pb);
66 ost->curidx = ogg->curidx;
67 ost->next = ogg->state;
68 ost->nstreams = ogg->nstreams;
69 memcpy(ost->streams, ogg->streams, ogg->nstreams * sizeof(*ogg->streams));
71 for (i = 0; i < ogg->nstreams; i++){
72 struct ogg_stream *os = ogg->streams + i;
73 os->buf = av_malloc (os->bufsize);
74 memset (os->buf, 0, os->bufsize);
75 memcpy (os->buf, ost->streams[i].buf, os->bufpos);
84 ogg_restore (AVFormatContext * s, int discard)
86 struct ogg *ogg = s->priv_data;
87 AVIOContext *bc = s->pb;
88 struct ogg_state *ost = ogg->state;
94 ogg->state = ost->next;
97 for (i = 0; i < ogg->nstreams; i++)
98 av_free (ogg->streams[i].buf);
100 url_fseek (bc, ost->pos, SEEK_SET);
101 ogg->curidx = ost->curidx;
102 ogg->nstreams = ost->nstreams;
103 memcpy(ogg->streams, ost->streams,
104 ost->nstreams * sizeof(*ogg->streams));
113 ogg_reset (struct ogg * ogg)
117 for (i = 0; i < ogg->nstreams; i++){
118 struct ogg_stream *os = ogg->streams + i;
123 os->lastpts = AV_NOPTS_VALUE;
124 os->lastdts = AV_NOPTS_VALUE;
137 static const struct ogg_codec *
138 ogg_find_codec (uint8_t * buf, int size)
142 for (i = 0; ogg_codecs[i]; i++)
143 if (size >= ogg_codecs[i]->magicsize &&
144 !memcmp (buf, ogg_codecs[i]->magic, ogg_codecs[i]->magicsize))
145 return ogg_codecs[i];
151 ogg_new_stream (AVFormatContext * s, uint32_t serial)
154 struct ogg *ogg = s->priv_data;
155 int idx = ogg->nstreams++;
157 struct ogg_stream *os;
159 ogg->streams = av_realloc (ogg->streams,
160 ogg->nstreams * sizeof (*ogg->streams));
161 memset (ogg->streams + idx, 0, sizeof (*ogg->streams));
162 os = ogg->streams + idx;
164 os->bufsize = DECODER_BUFFER_SIZE;
165 os->buf = av_malloc(os->bufsize);
168 st = av_new_stream (s, idx);
170 return AVERROR(ENOMEM);
172 av_set_pts_info(st, 64, 1, 1000000);
178 ogg_new_buf(struct ogg *ogg, int idx)
180 struct ogg_stream *os = ogg->streams + idx;
181 uint8_t *nb = av_malloc(os->bufsize);
182 int size = os->bufpos - os->pstart;
184 memcpy(nb, os->buf + os->pstart, size);
195 ogg_read_page (AVFormatContext * s, int *str)
197 AVIOContext *bc = s->pb;
198 struct ogg *ogg = s->priv_data;
199 struct ogg_stream *os;
210 if (avio_read (bc, sync, 4) < 4)
216 if (sync[sp & 3] == 'O' &&
217 sync[(sp + 1) & 3] == 'g' &&
218 sync[(sp + 2) & 3] == 'g' && sync[(sp + 3) & 3] == 'S')
225 }while (i++ < MAX_PAGE_SIZE);
227 if (i >= MAX_PAGE_SIZE){
228 av_log (s, AV_LOG_INFO, "ogg, can't find sync word\n");
232 if (url_fgetc (bc) != 0) /* version */
235 flags = url_fgetc (bc);
237 serial = avio_rl32 (bc);
238 seq = avio_rl32 (bc);
239 crc = avio_rl32 (bc);
240 nsegs = url_fgetc (bc);
242 idx = ogg_find_stream (ogg, serial);
244 idx = ogg_new_stream (s, serial);
249 os = ogg->streams + idx;
250 os->page_pos = url_ftell(bc) - 27;
253 ogg_new_buf(ogg, idx);
255 if (avio_read (bc, os->segments, nsegs) < nsegs)
262 for (i = 0; i < nsegs; i++)
263 size += os->segments[i];
265 if (flags & OGG_FLAG_CONT || os->incomplete){
267 while (os->segp < os->nsegs){
268 int seg = os->segments[os->segp++];
273 os->sync_pos = os->page_pos;
277 os->sync_pos = os->page_pos;
280 if (os->bufsize - os->bufpos < size){
281 uint8_t *nb = av_malloc (os->bufsize *= 2);
282 memcpy (nb, os->buf, os->bufpos);
287 if (avio_read (bc, os->buf + os->bufpos, size) < size)
301 ogg_packet (AVFormatContext * s, int *str, int *dstart, int *dsize, int64_t *fpos)
303 struct ogg *ogg = s->priv_data;
305 struct ogg_stream *os;
307 int segp = 0, psize = 0;
310 av_log (s, AV_LOG_DEBUG, "ogg_packet: curidx=%i\n", ogg->curidx);
317 if (ogg_read_page (s, &idx) < 0)
321 os = ogg->streams + idx;
324 av_log (s, AV_LOG_DEBUG,
325 "ogg_packet: idx=%d pstart=%d psize=%d segp=%d nsegs=%d\n",
326 idx, os->pstart, os->psize, os->segp, os->nsegs);
331 os->codec = ogg_find_codec (os->buf, os->bufpos);
344 while (os->segp < os->nsegs){
345 int ss = os->segments[os->segp++];
353 if (!complete && os->segp == os->nsegs){
360 av_log (s, AV_LOG_DEBUG,
361 "ogg_packet: idx %i, frame size %i, start %i\n",
362 idx, os->psize, os->pstart);
365 if (os->granule == -1)
366 av_log(s, AV_LOG_WARNING, "Page at %"PRId64" is missing granule\n", os->page_pos);
372 os->header = os->codec->header (s, idx);
377 // We have reached the first non-header packet in this stream.
378 // Unfortunately more header packets may still follow for others,
379 // so we reset this later unless we are done with the headers
383 // Update the header state for all streams and
384 // compute the data_offset.
386 s->data_offset = os->sync_pos;
387 for (i = 0; i < ogg->nstreams; i++) {
388 struct ogg_stream *cur_os = ogg->streams + i;
389 if (cur_os->header > 0)
392 // if we have a partial non-header packet, its start is
393 // obviously at or after the data start
394 if (cur_os->incomplete)
395 s->data_offset = FFMIN(s->data_offset, cur_os->sync_pos);
398 os->pstart += os->psize;
404 if (os->codec && os->codec->packet)
405 os->codec->packet (s, idx);
409 *dstart = os->pstart;
413 *fpos = os->sync_pos;
414 os->pstart += os->psize;
416 os->sync_pos = os->page_pos;
419 // determine whether there are more complete packets in this page
420 // if not, the page's granule will apply to this packet
422 for (i = os->segp; i < os->nsegs; i++)
423 if (os->segments[i] < 255) {
428 if (os->segp == os->nsegs)
435 ogg_get_headers (AVFormatContext * s)
437 struct ogg *ogg = s->priv_data;
440 if (ogg_packet (s, NULL, NULL, NULL, NULL) < 0)
442 }while (!ogg->headers);
445 av_log (s, AV_LOG_DEBUG, "found headers\n");
452 ogg_get_length (AVFormatContext * s)
454 struct ogg *ogg = s->priv_data;
458 if(url_is_streamed(s->pb))
462 if (s->duration != AV_NOPTS_VALUE)
465 size = url_fsize(s->pb);
468 end = size > MAX_PAGE_SIZE? size - MAX_PAGE_SIZE: 0;
471 url_fseek (s->pb, end, SEEK_SET);
473 while (!ogg_read_page (s, &i)){
474 if (ogg->streams[i].granule != -1 && ogg->streams[i].granule != 0 &&
475 ogg->streams[i].codec) {
476 s->streams[i]->duration =
477 ogg_gptopts (s, i, ogg->streams[i].granule, NULL);
478 if (s->streams[i]->start_time != AV_NOPTS_VALUE)
479 s->streams[i]->duration -= s->streams[i]->start_time;
490 ogg_read_header (AVFormatContext * s, AVFormatParameters * ap)
492 struct ogg *ogg = s->priv_data;
495 //linear headers seek from start
496 if (ogg_get_headers (s) < 0){
500 for (i = 0; i < ogg->nstreams; i++)
501 if (ogg->streams[i].header < 0)
502 ogg->streams[i].codec = NULL;
504 //linear granulepos seek from end
507 //fill the extradata in the per codec callbacks
511 static int64_t ogg_calc_pts(AVFormatContext *s, int idx, int64_t *dts)
513 struct ogg *ogg = s->priv_data;
514 struct ogg_stream *os = ogg->streams + idx;
515 int64_t pts = AV_NOPTS_VALUE;
518 *dts = AV_NOPTS_VALUE;
520 if (os->lastpts != AV_NOPTS_VALUE) {
522 os->lastpts = AV_NOPTS_VALUE;
524 if (os->lastdts != AV_NOPTS_VALUE) {
527 os->lastdts = AV_NOPTS_VALUE;
530 if (os->granule != -1LL) {
531 if (os->codec && os->codec->granule_is_start)
532 pts = ogg_gptopts(s, idx, os->granule, dts);
534 os->lastpts = ogg_gptopts(s, idx, os->granule, &os->lastdts);
542 ogg_read_packet (AVFormatContext * s, AVPacket * pkt)
545 struct ogg_stream *os;
548 int64_t fpos, pts, dts;
553 if (ogg_packet (s, &idx, &pstart, &psize, &fpos) < 0)
555 }while (idx < 0 || !s->streams[idx]);
558 os = ogg->streams + idx;
560 // pflags might not be set until after this
561 pts = ogg_calc_pts(s, idx, &dts);
563 if (os->keyframe_seek && !(os->pflags & AV_PKT_FLAG_KEY))
565 os->keyframe_seek = 0;
568 if (av_new_packet (pkt, psize) < 0)
570 pkt->stream_index = idx;
571 memcpy (pkt->data, os->buf + pstart, psize);
575 pkt->flags = os->pflags;
576 pkt->duration = os->pduration;
584 ogg_read_close (AVFormatContext * s)
586 struct ogg *ogg = s->priv_data;
589 for (i = 0; i < ogg->nstreams; i++){
590 av_free (ogg->streams[i].buf);
591 av_free (ogg->streams[i].private);
593 av_free (ogg->streams);
599 ogg_read_timestamp (AVFormatContext * s, int stream_index, int64_t * pos_arg,
602 struct ogg *ogg = s->priv_data;
603 struct ogg_stream *os = ogg->streams + stream_index;
604 AVIOContext *bc = s->pb;
605 int64_t pts = AV_NOPTS_VALUE;
607 url_fseek(bc, *pos_arg, SEEK_SET);
610 while (url_ftell(bc) < pos_limit && !ogg_packet(s, &i, NULL, NULL, pos_arg)) {
611 if (i == stream_index) {
612 pts = ogg_calc_pts(s, i, NULL);
613 if (os->keyframe_seek && !(os->pflags & AV_PKT_FLAG_KEY))
614 pts = AV_NOPTS_VALUE;
616 if (pts != AV_NOPTS_VALUE)
623 static int ogg_read_seek(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
625 struct ogg *ogg = s->priv_data;
626 struct ogg_stream *os = ogg->streams + stream_index;
629 // Try seeking to a keyframe first. If this fails (very possible),
630 // av_seek_frame will fall back to ignoring keyframes
631 if (s->streams[stream_index]->codec->codec_type == AVMEDIA_TYPE_VIDEO
632 && !(flags & AVSEEK_FLAG_ANY))
633 os->keyframe_seek = 1;
635 ret = av_seek_frame_binary(s, stream_index, timestamp, flags);
637 os->keyframe_seek = 0;
641 static int ogg_probe(AVProbeData *p)
643 if (p->buf[0] == 'O' && p->buf[1] == 'g' &&
644 p->buf[2] == 'g' && p->buf[3] == 'S' &&
645 p->buf[4] == 0x0 && p->buf[5] <= 0x7 )
646 return AVPROBE_SCORE_MAX;
651 AVInputFormat ff_ogg_demuxer = {
653 NULL_IF_CONFIG_SMALL("Ogg"),
662 .flags = AVFMT_GENERIC_INDEX,