FFmpeg
rtpdec_qcelp.c
Go to the documentation of this file.
1 /*
2  * RTP Depacketization of QCELP/PureVoice, RFC 2658
3  * Copyright (c) 2010 Martin Storsjo
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include "rtpdec_formats.h"
23 
24 static const uint8_t frame_sizes[] = {
25  1, 4, 8, 17, 35
26 };
27 
28 typedef struct InterleavePacket {
29  int pos;
30  int size;
31  /* The largest frame is 35 bytes, only 10 frames are allowed per
32  * packet, and we return the first one immediately, so allocate
33  * space for 9 frames */
34  uint8_t data[35*9];
36 
37 struct PayloadContext {
42 
43  /* The maximum packet size, 10 frames of 35 bytes each, and one
44  * packet header byte. */
45  uint8_t next_data[1 + 35*10];
46  int next_size;
47  uint32_t next_timestamp;
48 };
49 
51  AVStream *st, AVPacket *pkt, uint32_t *timestamp,
52  const uint8_t *buf, int len);
53 
55  AVStream *st, AVPacket *pkt, uint32_t *timestamp,
56  const uint8_t *buf, int len)
57 {
58  int interleave_size, interleave_index;
59  int frame_size, ret;
60  InterleavePacket* ip;
61 
62  if (len < 2)
63  return AVERROR_INVALIDDATA;
64 
65  interleave_size = buf[0] >> 3 & 7;
66  interleave_index = buf[0] & 7;
67 
68  if (interleave_size > 5) {
69  av_log(ctx, AV_LOG_ERROR, "Invalid interleave size %d\n",
70  interleave_size);
71  return AVERROR_INVALIDDATA;
72  }
73  if (interleave_index > interleave_size) {
74  av_log(ctx, AV_LOG_ERROR, "Invalid interleave index %d/%d\n",
75  interleave_index, interleave_size);
76  return AVERROR_INVALIDDATA;
77  }
78  if (interleave_size != data->interleave_size) {
79  int i;
80  /* First packet, or changed interleave size */
81  data->interleave_size = interleave_size;
82  data->interleave_index = 0;
83  for (i = 0; i < 6; i++)
84  data->group[i].size = 0;
85  }
86 
87  if (interleave_index < data->interleave_index) {
88  /* Wrapped around - missed the last packet of the previous group. */
89  if (data->group_finished) {
90  /* No more data in the packets in this interleaving group, just
91  * start processing the next one */
92  data->interleave_index = 0;
93  } else {
94  /* Stash away the current packet, emit everything we have of the
95  * previous group. */
96  for (; data->interleave_index <= interleave_size;
97  data->interleave_index++)
98  data->group[data->interleave_index].size = 0;
99 
100  if (len > sizeof(data->next_data))
101  return AVERROR_INVALIDDATA;
102  memcpy(data->next_data, buf, len);
103  data->next_size = len;
104  data->next_timestamp = *timestamp;
105  *timestamp = RTP_NOTS_VALUE;
106 
107  data->interleave_index = 0;
108  return return_stored_frame(ctx, data, st, pkt, timestamp, buf, len);
109  }
110  }
111  if (interleave_index > data->interleave_index) {
112  /* We missed a packet */
113  for (; data->interleave_index < interleave_index;
114  data->interleave_index++)
115  data->group[data->interleave_index].size = 0;
116  }
117  data->interleave_index = interleave_index;
118 
119  if (buf[1] >= FF_ARRAY_ELEMS(frame_sizes))
120  return AVERROR_INVALIDDATA;
121  frame_size = frame_sizes[buf[1]];
122  if (1 + frame_size > len)
123  return AVERROR_INVALIDDATA;
124 
125  if (len - 1 - frame_size > sizeof(data->group[0].data))
126  return AVERROR_INVALIDDATA;
127 
128  if ((ret = av_new_packet(pkt, frame_size)) < 0)
129  return ret;
130  memcpy(pkt->data, &buf[1], frame_size);
131  pkt->stream_index = st->index;
132 
133  ip = &data->group[data->interleave_index];
134  ip->size = len - 1 - frame_size;
135  ip->pos = 0;
136  memcpy(ip->data, &buf[1 + frame_size], ip->size);
137  /* Each packet must contain the same number of frames according to the
138  * RFC. If there's no data left in this packet, there shouldn't be any
139  * in any of the other frames in the interleaving group either. */
140  data->group_finished = ip->size == 0;
141 
142  if (interleave_index == interleave_size) {
143  data->interleave_index = 0;
144  return !data->group_finished;
145  } else {
146  data->interleave_index++;
147  return 0;
148  }
149 }
150 
152  AVStream *st, AVPacket *pkt, uint32_t *timestamp,
153  const uint8_t *buf, int len)
154 {
155  InterleavePacket* ip = &data->group[data->interleave_index];
156  int frame_size, ret;
157 
158  if (data->group_finished && data->interleave_index == 0) {
159  *timestamp = data->next_timestamp;
160  ret = store_packet(ctx, data, st, pkt, timestamp, data->next_data,
161  data->next_size);
162  data->next_size = 0;
163  return ret;
164  }
165 
166  if (ip->size == 0) {
167  /* No stored data for this interleave block, output an empty packet */
168  if ((ret = av_new_packet(pkt, 1)) < 0)
169  return ret;
170  pkt->data[0] = 0; // Blank - could also be 14, Erasure
171  } else {
172  if (ip->pos >= ip->size)
173  return AVERROR_INVALIDDATA;
174  if (ip->data[ip->pos] >= FF_ARRAY_ELEMS(frame_sizes))
175  return AVERROR_INVALIDDATA;
176  frame_size = frame_sizes[ip->data[ip->pos]];
177  if (ip->pos + frame_size > ip->size)
178  return AVERROR_INVALIDDATA;
179 
180  if ((ret = av_new_packet(pkt, frame_size)) < 0)
181  return ret;
182  memcpy(pkt->data, &ip->data[ip->pos], frame_size);
183 
184  ip->pos += frame_size;
185  data->group_finished = ip->pos >= ip->size;
186  }
187  pkt->stream_index = st->index;
188 
189  if (data->interleave_index == data->interleave_size) {
190  data->interleave_index = 0;
191  if (!data->group_finished)
192  return 1;
193  else
194  return data->next_size > 0;
195  } else {
196  data->interleave_index++;
197  return 1;
198  }
199 }
200 
202  AVStream *st, AVPacket *pkt, uint32_t *timestamp,
203  const uint8_t *buf, int len, uint16_t seq,
204  int flags)
205 {
206  if (buf)
207  return store_packet(ctx, data, st, pkt, timestamp, buf, len);
208  else
209  return return_stored_frame(ctx, data, st, pkt, timestamp, buf, len);
210 }
211 
213  .enc_name = "x-Purevoice",
214  .codec_type = AVMEDIA_TYPE_AUDIO,
215  .codec_id = AV_CODEC_ID_QCELP,
216  .priv_data_size = sizeof(PayloadContext),
217  .static_payload_id = 12,
219 };
InterleavePacket::pos
int pos
Definition: rtpdec_qcelp.c:29
rtpdec_formats.h
PayloadContext::interleave_index
int interleave_index
Definition: rtpdec_qcelp.c:39
AVPacket::data
uint8_t * data
Definition: packet.h:522
data
const char data[16]
Definition: mxf.c:148
RTPDynamicProtocolHandler::enc_name
const char * enc_name
Definition: rtpdec.h:117
PayloadContext::group
InterleavePacket group[6]
Definition: rtpdec_qcelp.c:40
pkt
AVPacket * pkt
Definition: movenc.c:59
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
FF_ARRAY_ELEMS
#define FF_ARRAY_ELEMS(a)
Definition: sinewin_tablegen.c:29
return_stored_frame
static int return_stored_frame(AVFormatContext *ctx, PayloadContext *data, AVStream *st, AVPacket *pkt, uint32_t *timestamp, const uint8_t *buf, int len)
Definition: rtpdec_qcelp.c:151
av_new_packet
int av_new_packet(AVPacket *pkt, int size)
Allocate the payload of a packet and initialize its fields with default values.
Definition: avpacket.c:98
frame_size
int frame_size
Definition: mxfenc.c:2422
AVMEDIA_TYPE_AUDIO
@ AVMEDIA_TYPE_AUDIO
Definition: avutil.h:202
RTP_NOTS_VALUE
#define RTP_NOTS_VALUE
Definition: rtpdec.h:41
InterleavePacket
Definition: rtpdec_qcelp.c:28
ctx
AVFormatContext * ctx
Definition: movenc.c:48
if
if(ret)
Definition: filter_design.txt:179
AVFormatContext
Format I/O context.
Definition: avformat.h:1255
qcelp_parse_packet
static int qcelp_parse_packet(AVFormatContext *ctx, PayloadContext *data, AVStream *st, AVPacket *pkt, uint32_t *timestamp, const uint8_t *buf, int len, uint16_t seq, int flags)
Definition: rtpdec_qcelp.c:201
PayloadContext::next_data
uint8_t next_data[1+35 *10]
Definition: rtpdec_qcelp.c:45
AV_CODEC_ID_QCELP
@ AV_CODEC_ID_QCELP
Definition: codec_id.h:464
frame_sizes
static const uint8_t frame_sizes[]
Definition: rtpdec_qcelp.c:24
InterleavePacket::size
int size
Definition: rtpdec_qcelp.c:30
PayloadContext::interleave_size
int interleave_size
Definition: rtpdec_qcelp.c:38
store_packet
static int store_packet(AVFormatContext *ctx, PayloadContext *data, AVStream *st, AVPacket *pkt, uint32_t *timestamp, const uint8_t *buf, int len)
Definition: rtpdec_qcelp.c:54
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:255
ff_qcelp_dynamic_handler
const RTPDynamicProtocolHandler ff_qcelp_dynamic_handler
Definition: rtpdec_qcelp.c:212
len
int len
Definition: vorbis_enc_data.h:426
PayloadContext::next_timestamp
uint32_t next_timestamp
Definition: rtpdec_qcelp.c:47
ret
ret
Definition: filter_design.txt:187
AVStream
Stream structure.
Definition: avformat.h:743
AVStream::index
int index
stream index in AVFormatContext
Definition: avformat.h:749
AVPacket::stream_index
int stream_index
Definition: packet.h:524
parse_packet
static int parse_packet(AVFormatContext *s, AVPacket *pkt, int stream_index, int flush)
Parse a packet, add all split parts to parse_queue.
Definition: demux.c:1154
InterleavePacket::data
uint8_t data[35 *9]
Definition: rtpdec_qcelp.c:34
PayloadContext::group_finished
int group_finished
Definition: rtpdec_qcelp.c:41
AVPacket
This structure stores compressed data.
Definition: packet.h:499
flags
#define flags(name, subs,...)
Definition: cbs_av1.c:482
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
PayloadContext
RTP/JPEG specific private data.
Definition: rdt.c:84
PayloadContext::next_size
int next_size
Definition: rtpdec_qcelp.c:46
RTPDynamicProtocolHandler
Definition: rtpdec.h:116