FFmpeg  4.4.4
oggparsespeex.c
Go to the documentation of this file.
1 /*
2  Copyright (C) 2008 Reimar Döffinger
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/bswap.h"
28 #include "libavutil/avstring.h"
30 
31 #include "libavcodec/bytestream.h"
32 
33 #include "avformat.h"
34 #include "internal.h"
35 #include "oggdec.h"
36 
37 struct speex_params {
40  int seq;
41 };
42 
43 static int speex_header(AVFormatContext *s, int idx) {
44  struct ogg *ogg = s->priv_data;
45  struct ogg_stream *os = ogg->streams + idx;
46  struct speex_params *spxp = os->private;
47  AVStream *st = s->streams[idx];
48  uint8_t *p = os->buf + os->pstart;
49  int ret;
50 
51  if (!spxp) {
52  spxp = av_mallocz(sizeof(*spxp));
53  if (!spxp)
54  return AVERROR(ENOMEM);
55  os->private = spxp;
56  }
57 
58  if (spxp->seq > 1)
59  return 0;
60 
61  if (spxp->seq == 0) {
62  int frames_per_packet;
65 
66  if (os->psize < 68) {
67  av_log(s, AV_LOG_ERROR, "speex packet too small\n");
68  return AVERROR_INVALIDDATA;
69  }
70 
71  st->codecpar->sample_rate = AV_RL32(p + 36);
72  if (st->codecpar->sample_rate <= 0) {
73  av_log(s, AV_LOG_ERROR, "Invalid sample rate %d\n", st->codecpar->sample_rate);
74  return AVERROR_INVALIDDATA;
75  }
76  st->codecpar->channels = AV_RL32(p + 48);
77  if (st->codecpar->channels < 1 || st->codecpar->channels > 2) {
78  av_log(s, AV_LOG_ERROR, "invalid channel count. Speex must be mono or stereo.\n");
79  return AVERROR_INVALIDDATA;
80  }
83 
84  spxp->packet_size = AV_RL32(p + 56);
85  frames_per_packet = AV_RL32(p + 64);
86  if (spxp->packet_size < 0 ||
87  frames_per_packet < 0 ||
88  spxp->packet_size * (int64_t)frames_per_packet > INT32_MAX / 256) {
89  av_log(s, AV_LOG_ERROR, "invalid packet_size, frames_per_packet %d %d\n", spxp->packet_size, frames_per_packet);
90  spxp->packet_size = 0;
91  return AVERROR_INVALIDDATA;
92  }
93  if (frames_per_packet)
94  spxp->packet_size *= frames_per_packet;
95 
96  if ((ret = ff_alloc_extradata(st->codecpar, os->psize)) < 0)
97  return ret;
98  memcpy(st->codecpar->extradata, p, st->codecpar->extradata_size);
99 
100  avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate);
101  } else
102  ff_vorbis_stream_comment(s, st, p, os->psize);
103 
104  spxp->seq++;
105  return 1;
106 }
107 
108 static int ogg_page_packets(struct ogg_stream *os)
109 {
110  int i;
111  int packets = 0;
112  for (i = 0; i < os->nsegs; i++)
113  if (os->segments[i] < 255)
114  packets++;
115  return packets;
116 }
117 
118 static int speex_packet(AVFormatContext *s, int idx)
119 {
120  struct ogg *ogg = s->priv_data;
121  struct ogg_stream *os = ogg->streams + idx;
122  struct speex_params *spxp = os->private;
123  int packet_size = spxp->packet_size;
124 
125  if (os->flags & OGG_FLAG_EOS && os->lastpts != AV_NOPTS_VALUE &&
126  os->granule > 0) {
127  /* first packet of final page. we have to calculate the final packet
128  duration here because it is the only place we know the next-to-last
129  granule position. */
130  spxp->final_packet_duration = os->granule - os->lastpts -
131  packet_size * (ogg_page_packets(os) - 1);
132  }
133 
134  if (!os->lastpts && os->granule > 0)
135  /* first packet */
136  os->lastpts = os->lastdts = os->granule - packet_size *
137  ogg_page_packets(os);
138  if (os->flags & OGG_FLAG_EOS && os->segp == os->nsegs &&
139  spxp->final_packet_duration)
140  /* final packet */
141  os->pduration = spxp->final_packet_duration;
142  else
143  os->pduration = packet_size;
144 
145  return 0;
146 }
147 
148 const struct ogg_codec ff_speex_codec = {
149  .magic = "Speex ",
150  .magicsize = 8,
151  .header = speex_header,
152  .packet = speex_packet,
153  .nb_header = 2,
154 };
uint8_t
Main libavformat public API header.
#define AV_RL32
Definition: intreadwrite.h:146
byte swapping routines
#define s(width, name)
Definition: cbs_vp9.c:257
audio channel layout utility functions
#define AV_CH_LAYOUT_MONO
#define AV_CH_LAYOUT_STEREO
@ AV_CODEC_ID_SPEEX
Definition: codec_id.h:459
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
#define AVERROR(e)
Definition: error.h:43
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
Definition: mem.c:237
@ AVMEDIA_TYPE_AUDIO
Definition: avutil.h:202
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:248
int i
Definition: input.c:407
void avpriv_set_pts_info(AVStream *s, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: utils.c:4945
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:3314
common internal API header
int ff_vorbis_stream_comment(AVFormatContext *as, AVStream *st, const uint8_t *buf, int size)
#define OGG_FLAG_EOS
Definition: oggdec.h:113
const struct ogg_codec ff_speex_codec
static int ogg_page_packets(struct ogg_stream *os)
static int speex_header(AVFormatContext *s, int idx)
Definition: oggparsespeex.c:43
static int speex_packet(AVFormatContext *s, int idx)
int extradata_size
Size of the extradata content in bytes.
Definition: codec_par.h:78
uint64_t channel_layout
Audio only.
Definition: codec_par.h:162
int channels
Audio only.
Definition: codec_par.h:166
enum AVMediaType codec_type
General type of the encoded data.
Definition: codec_par.h:56
uint8_t * extradata
Extra binary data needed for initializing the decoder, codec-dependent.
Definition: codec_par.h:74
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:60
int sample_rate
Audio only.
Definition: codec_par.h:170
Format I/O context.
Definition: avformat.h:1232
Stream structure.
Definition: avformat.h:873
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1038
Copyright (C) 2005 Michael Ahlberg, Måns Rullgård.
Definition: oggdec.h:31
const int8_t * magic
Definition: oggdec.h:32
unsigned int pduration
Definition: oggdec.h:68
int nsegs
Definition: oggdec.h:79
uint8_t segments[255]
Definition: oggdec.h:80
uint64_t granule
Definition: oggdec.h:70
int64_t lastdts
Definition: oggdec.h:73
unsigned int pstart
Definition: oggdec.h:65
unsigned int psize
Definition: oggdec.h:66
void * private
Definition: oggdec.h:91
uint8_t * buf
Definition: oggdec.h:62
int64_t lastpts
Definition: oggdec.h:72
int segp
Definition: oggdec.h:79
int flags
Definition: oggdec.h:76
Definition: oggdec.h:102
struct ogg_stream * streams
Definition: oggdec.h:103
int final_packet_duration
Definition: oggparsespeex.c:39
#define av_log(a,...)