FFmpeg
oggparseogm.c
Go to the documentation of this file.
1 /**
2  Copyright (C) 2005 Michael Ahlberg, Måns Rullgård
3 
4  Permission is hereby granted, free of charge, to any person
5  obtaining a copy of this software and associated documentation
6  files (the "Software"), to deal in the Software without
7  restriction, including without limitation the rights to use, copy,
8  modify, merge, publish, distribute, sublicense, and/or sell copies
9  of the Software, and to permit persons to whom the Software is
10  furnished to do so, subject to the following conditions:
11 
12  The above copyright notice and this permission notice shall be
13  included in all copies or substantial portions of the Software.
14 
15  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
16  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
17  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
18  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
19  HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
20  WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22  DEALINGS IN THE SOFTWARE.
23 **/
24 
25 #include <stdlib.h>
26 
27 #include "libavutil/intreadwrite.h"
28 
29 #include "libavcodec/bytestream.h"
30 
31 #include "avformat.h"
32 #include "internal.h"
33 #include "oggdec.h"
34 #include "riff.h"
35 
36 static int
38 {
39  struct ogg *ogg = s->priv_data;
40  struct ogg_stream *os = ogg->streams + idx;
41  AVStream *st = s->streams[idx];
42  FFStream *const sti = ffstream(st);
44  uint64_t time_unit;
45  uint64_t spu;
46  uint32_t size;
47  int ret;
48 
49  bytestream2_init(&p, os->buf + os->pstart, os->psize);
50  if (!(bytestream2_peek_byte(&p) & 1))
51  return 0;
52 
53  if (bytestream2_peek_byte(&p) == 1) {
54  bytestream2_skip(&p, 1);
55 
56  if (bytestream2_peek_byte(&p) == 'v'){
57  int tag;
59  bytestream2_skip(&p, 8);
60  tag = bytestream2_get_le32(&p);
62  st->codecpar->codec_tag = tag;
65  } else if (bytestream2_peek_byte(&p) == 't') {
68  bytestream2_skip(&p, 12);
69  } else {
70  uint8_t acid[5] = { 0 };
71  int cid;
73  bytestream2_skip(&p, 8);
74  bytestream2_get_buffer(&p, acid, 4);
75  acid[4] = 0;
76  cid = strtol(acid, NULL, 16);
78  // our parser completely breaks AAC in Ogg
79  if (st->codecpar->codec_id != AV_CODEC_ID_AAC)
81  }
82 
83  size = bytestream2_get_le32(&p);
84  size = FFMIN(size, os->psize);
85  time_unit = bytestream2_get_le64(&p);
86  spu = bytestream2_get_le64(&p);
87  if (!time_unit || !spu) {
88  av_log(s, AV_LOG_ERROR, "Invalid timing values.\n");
89  return AVERROR_INVALIDDATA;
90  }
91 
92  bytestream2_skip(&p, 4); /* default_len */
93  bytestream2_skip(&p, 8); /* buffersize + bits_per_sample */
94 
96  st->codecpar->width = bytestream2_get_le32(&p);
97  st->codecpar->height = bytestream2_get_le32(&p);
98  avpriv_set_pts_info(st, 64, time_unit, spu * 10000000);
99  } else {
100  st->codecpar->ch_layout.nb_channels = bytestream2_get_le16(&p);
101  bytestream2_skip(&p, 2); /* block_align */
102  st->codecpar->bit_rate = bytestream2_get_le32(&p) * 8;
103  st->codecpar->sample_rate = spu * 10000000 / time_unit;
104  avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate);
105  if (size >= 56 && st->codecpar->codec_id == AV_CODEC_ID_AAC) {
106  bytestream2_skip(&p, 4);
107  size -= 4;
108  }
109  if (size > 52) {
110  size -= 52;
112  return AVERROR_INVALIDDATA;
113  if ((ret = ff_alloc_extradata(st->codecpar, size)) < 0)
114  return ret;
116  }
117  }
118 
119  // Update internal avctx with changes to codecpar above.
120  sti->need_context_update = 1;
121  } else if (bytestream2_peek_byte(&p) == 3) {
122  bytestream2_skip(&p, 7);
123  if (bytestream2_get_bytes_left(&p) > 1)
125  }
126 
127  return 1;
128 }
129 
130 static int
132 {
133  struct ogg *ogg = s->priv_data;
134  struct ogg_stream *os = ogg->streams + idx;
135  AVStream *st = s->streams[idx];
136  uint8_t *p = os->buf + os->pstart;
137  uint32_t t;
138 
139  if(!(*p & 1))
140  return 0;
141  if(*p != 1)
142  return 1;
143 
144  if (os->psize < 100)
145  return AVERROR_INVALIDDATA;
146  t = AV_RL32(p + 96);
147 
148  if(t == 0x05589f80){
149  if (os->psize < 184)
150  return AVERROR_INVALIDDATA;
151 
154  avpriv_set_pts_info(st, 64, AV_RL64(p + 164), 10000000);
155  st->codecpar->width = AV_RL32(p + 176);
156  st->codecpar->height = AV_RL32(p + 180);
157  } else if(t == 0x05589f81){
158  if (os->psize < 136)
159  return AVERROR_INVALIDDATA;
160 
163  st->codecpar->ch_layout.nb_channels = AV_RL16(p + 126);
164  st->codecpar->sample_rate = AV_RL32(p + 128);
165  st->codecpar->bit_rate = AV_RL32(p + 132) * 8;
166  }
167 
168  return 1;
169 }
170 
171 static int
173 {
174  struct ogg *ogg = s->priv_data;
175  struct ogg_stream *os = ogg->streams + idx;
176  uint8_t *p = os->buf + os->pstart;
177  int lb;
178 
179  if(*p & 8)
180  os->pflags |= AV_PKT_FLAG_KEY;
181 
182  lb = ((*p & 2) << 1) | ((*p >> 6) & 3);
183  if (os->psize < lb + 1)
184  return AVERROR_INVALIDDATA;
185 
186  os->pstart += lb + 1;
187  os->psize -= lb + 1;
188 
189  while (lb--)
190  os->pduration += (uint64_t)p[lb+1] << (lb*8);
191 
192  return 0;
193 }
194 
196  .magic = "\001video",
197  .magicsize = 6,
198  .header = ogm_header,
199  .packet = ogm_packet,
200  .granule_is_start = 1,
201  .nb_header = 2,
202 };
203 
205  .magic = "\001audio",
206  .magicsize = 6,
207  .header = ogm_header,
208  .packet = ogm_packet,
209  .granule_is_start = 1,
210  .nb_header = 2,
211 };
212 
213 const struct ogg_codec ff_ogm_text_codec = {
214  .magic = "\001text",
215  .magicsize = 5,
216  .header = ogm_header,
217  .packet = ogm_packet,
218  .granule_is_start = 1,
219  .nb_header = 2,
220 };
221 
222 const struct ogg_codec ff_ogm_old_codec = {
223  .magic = "\001Direct Show Samples embedded in Ogg",
224  .magicsize = 35,
225  .header = ogm_dshow_header,
226  .packet = ogm_packet,
227  .granule_is_start = 1,
228  .nb_header = 1,
229 };
AVMEDIA_TYPE_SUBTITLE
@ AVMEDIA_TYPE_SUBTITLE
Definition: avutil.h:204
AVCodecParameters::extradata
uint8_t * extradata
Extra binary data needed for initializing the decoder, codec-dependent.
Definition: codec_par.h:69
AVCodecParameters::codec_type
enum AVMediaType codec_type
General type of the encoded data.
Definition: codec_par.h:51
GetByteContext
Definition: bytestream.h:33
AV_RL64
uint64_t_TMPL AV_RL64
Definition: bytestream.h:91
ff_ogm_old_codec
const struct ogg_codec ff_ogm_old_codec
Definition: oggparseogm.c:222
AV_CODEC_ID_MPEG4
@ AV_CODEC_ID_MPEG4
Definition: codec_id.h:64
ff_codec_wav_tags
const AVCodecTag ff_codec_wav_tags[]
Definition: riff.c:518
AVCodecParameters::codec_tag
uint32_t codec_tag
Additional information about the codec (corresponds to the AVI FOURCC).
Definition: codec_par.h:59
ogg_stream::buf
uint8_t * buf
Definition: oggdec.h:61
AVChannelLayout::nb_channels
int nb_channels
Number of channels in this layout.
Definition: channel_layout.h:313
ogg
Definition: oggdec.h:101
AV_PKT_FLAG_KEY
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: packet.h:577
bytestream2_skip
static av_always_inline void bytestream2_skip(GetByteContext *g, unsigned int size)
Definition: bytestream.h:168
avpriv_set_pts_info
void avpriv_set_pts_info(AVStream *st, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: avformat.c:853
ffstream
static av_always_inline FFStream * ffstream(AVStream *st)
Definition: internal.h:417
ogg_stream::pstart
unsigned int pstart
Definition: oggdec.h:64
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
intreadwrite.h
s
#define s(width, name)
Definition: cbs_vp9.c:198
AVMEDIA_TYPE_AUDIO
@ AVMEDIA_TYPE_AUDIO
Definition: avutil.h:202
AVCodecParameters::width
int width
Video only.
Definition: codec_par.h:134
GetByteContext::buffer
const uint8_t * buffer
Definition: bytestream.h:34
AV_RL16
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_RL16
Definition: bytestream.h:94
FFStream::need_parsing
enum AVStreamParseType need_parsing
Definition: internal.h:386
AVFormatContext
Format I/O context.
Definition: avformat.h:1255
internal.h
AVStream::codecpar
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:766
ff_ogm_text_codec
const struct ogg_codec ff_ogm_text_codec
Definition: oggparseogm.c:213
NULL
#define NULL
Definition: coverity.c:32
bytestream2_get_buffer
static av_always_inline unsigned int bytestream2_get_buffer(GetByteContext *g, uint8_t *dst, unsigned int size)
Definition: bytestream.h:267
AVCodecParameters::ch_layout
AVChannelLayout ch_layout
Audio only.
Definition: codec_par.h:180
ogg::streams
struct ogg_stream * streams
Definition: oggdec.h:102
AVCodecParameters::sample_rate
int sample_rate
Audio only.
Definition: codec_par.h:184
cid
uint16_t cid
Definition: mxfenc.c:2262
bytestream2_get_bytes_left
static av_always_inline int bytestream2_get_bytes_left(GetByteContext *g)
Definition: bytestream.h:158
AVCodecParameters::extradata_size
int extradata_size
Size of the extradata content in bytes.
Definition: codec_par.h:73
AV_CODEC_ID_AAC
@ AV_CODEC_ID_AAC
Definition: codec_id.h:442
ogm_header
static int ogm_header(AVFormatContext *s, int idx)
Copyright (C) 2005 Michael Ahlberg, Måns Rullgård.
Definition: oggparseogm.c:37
ff_codec_get_id
enum AVCodecID ff_codec_get_id(const AVCodecTag *tags, unsigned int tag)
Definition: utils.c:145
FFStream
Definition: internal.h:193
size
int size
Definition: twinvq_data.h:10344
ff_ogm_video_codec
const struct ogg_codec ff_ogm_video_codec
Definition: oggparseogm.c:195
ogg_stream::pflags
unsigned int pflags
Definition: oggdec.h:66
ogg_stream
Definition: oggdec.h:60
AVCodecParameters::height
int height
Definition: codec_par.h:135
FFMIN
#define FFMIN(a, b)
Definition: macros.h:49
ogm_dshow_header
static int ogm_dshow_header(AVFormatContext *s, int idx)
Definition: oggparseogm.c:131
tag
uint32_t tag
Definition: movenc.c:1791
ret
ret
Definition: filter_design.txt:187
AVStream
Stream structure.
Definition: avformat.h:743
AVSTREAM_PARSE_HEADERS
@ AVSTREAM_PARSE_HEADERS
Only parse headers, do not repack.
Definition: avformat.h:594
avformat.h
AV_CODEC_ID_TEXT
@ AV_CODEC_ID_TEXT
raw UTF-8 text
Definition: codec_id.h:551
AV_RL32
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:92
oggdec.h
ff_codec_bmp_tags
const AVCodecTag ff_codec_bmp_tags[]
Definition: riff.c:36
ff_ogm_audio_codec
const struct ogg_codec ff_ogm_audio_codec
Definition: oggparseogm.c:204
ogm_packet
static int ogm_packet(AVFormatContext *s, int idx)
Definition: oggparseogm.c:172
ff_vorbis_stream_comment
int ff_vorbis_stream_comment(AVFormatContext *as, AVStream *st, const uint8_t *buf, int size)
Parse Vorbis comments and add metadata to an AVStream.
Definition: oggparsevorbis.c:73
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
AVCodecParameters::codec_id
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:55
riff.h
ogg_stream::psize
unsigned int psize
Definition: oggdec.h:65
ogg_codec
Copyright (C) 2005 Michael Ahlberg, Måns Rullgård.
Definition: oggdec.h:30
bytestream.h
AVSTREAM_PARSE_FULL
@ AVSTREAM_PARSE_FULL
full parsing and repack
Definition: avformat.h:593
bytestream2_init
static av_always_inline void bytestream2_init(GetByteContext *g, const uint8_t *buf, int buf_size)
Definition: bytestream.h:137
FFStream::need_context_update
int need_context_update
Whether the internal avctx needs to be updated from codecpar (after a late change to codecpar)
Definition: internal.h:238
AVCodecParameters::bit_rate
int64_t bit_rate
The average bitrate of the encoded data (in bits per second).
Definition: codec_par.h:97
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
AVERROR_INVALIDDATA
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:61
ogg_codec::magic
const int8_t * magic
Definition: oggdec.h:31
ogg_stream::pduration
unsigned int pduration
Definition: oggdec.h:67
ff_alloc_extradata
int ff_alloc_extradata(AVCodecParameters *par, int size)
Allocate extradata with additional AV_INPUT_BUFFER_PADDING_SIZE at end which is always set to 0.
Definition: utils.c:239