1 /* GStreamer
2 * Copyright (C) <2005> Wim Taymans <wim.taymans@gmail.com>
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Library General Public
6 * License as published by the Free Software Foundation; either
7 * version 2 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Library General Public License for more details.
13 *
14 * You should have received a copy of the GNU Library General Public
15 * License along with this library; if not, write to the
16 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
17 * Boston, MA 02110-1301, USA.
18 */
19
20 #ifdef HAVE_CONFIG_H
21 # include "config.h"
22 #endif
23
24 #include <string.h>
25 #include <stdlib.h>
26 #include <gst/rtp/gstrtpbuffer.h>
27
28 #include "gstrtpelements.h"
29 #include "gstrtpmp4gdepay.h"
30 #include "gstrtputils.h"
31
32 GST_DEBUG_CATEGORY_STATIC (rtpmp4gdepay_debug);
33 #define GST_CAT_DEFAULT (rtpmp4gdepay_debug)
34
35 static GstStaticPadTemplate gst_rtp_mp4g_depay_src_template =
36 GST_STATIC_PAD_TEMPLATE ("src",
37 GST_PAD_SRC,
38 GST_PAD_ALWAYS,
39 GST_STATIC_CAPS ("video/mpeg,"
40 "mpegversion=(int) 4,"
41 "systemstream=(boolean)false;"
42 "audio/mpeg," "mpegversion=(int) 4, " "stream-format=(string)raw")
43 );
44
45 static GstStaticPadTemplate gst_rtp_mp4g_depay_sink_template =
46 GST_STATIC_PAD_TEMPLATE ("sink",
47 GST_PAD_SINK,
48 GST_PAD_ALWAYS,
49 GST_STATIC_CAPS ("application/x-rtp, "
50 "media = (string) { \"video\", \"audio\", \"application\" }, "
51 "clock-rate = (int) [1, MAX ], "
52 "encoding-name = (string) \"MPEG4-GENERIC\", "
53 /* required string params */
54 /* "streamtype = (string) { \"4\", \"5\" }, " Not set by Wowza 4 = video, 5 = audio */
55 /* "profile-level-id = (string) [1,MAX], " */
56 /* "config = (string) [1,MAX]" */
57 "mode = (string) { \"generic\", \"CELP-cbr\", \"CELP-vbr\", \"AAC-lbr\", \"AAC-hbr\", \"aac-hbr\" } "
58 /* Optional general parameters */
59 /* "objecttype = (string) [1,MAX], " */
60 /* "constantsize = (string) [1,MAX], " *//* constant size of each AU */
61 /* "constantduration = (string) [1,MAX], " *//* constant duration of each AU */
62 /* "maxdisplacement = (string) [1,MAX], " */
63 /* "de-interleavebuffersize = (string) [1,MAX], " */
64 /* Optional configuration parameters */
65 /* "sizelength = (string) [1, 32], " */
66 /* "indexlength = (string) [1, 32], " */
67 /* "indexdeltalength = (string) [1, 32], " */
68 /* "ctsdeltalength = (string) [1, 32], " */
69 /* "dtsdeltalength = (string) [1, 32], " */
70 /* "randomaccessindication = (string) {0, 1}, " */
71 /* "streamstateindication = (string) [0, 32], " */
72 /* "auxiliarydatasizelength = (string) [0, 32]" */ )
73 );
74
75 /* simple bitstream parser */
76 typedef struct
77 {
78 const guint8 *data;
79 const guint8 *end;
80 gint head; /* bitpos in the cache of next bit */
81 guint64 cache; /* cached bytes */
82 } GstBsParse;
83
84 static void
gst_bs_parse_init(GstBsParse * bs,const guint8 * data,guint size)85 gst_bs_parse_init (GstBsParse * bs, const guint8 * data, guint size)
86 {
87 bs->data = data;
88 bs->end = data + size;
89 bs->head = 0;
90 bs->cache = 0xffffffff;
91 }
92
93 static guint32
gst_bs_parse_read(GstBsParse * bs,guint n)94 gst_bs_parse_read (GstBsParse * bs, guint n)
95 {
96 guint32 res = 0;
97 gint shift;
98
99 if (n == 0)
100 return res;
101
102 /* fill up the cache if we need to */
103 while (bs->head < n) {
104 if (bs->data >= bs->end) {
105 /* we're at the end, can't produce more than head number of bits */
106 n = bs->head;
107 break;
108 }
109 /* shift bytes in cache, moving the head bits of the cache left */
110 bs->cache = (bs->cache << 8) | *bs->data++;
111 bs->head += 8;
112 }
113
114 /* bring the required bits down and truncate */
115 if ((shift = bs->head - n) > 0)
116 res = bs->cache >> shift;
117 else
118 res = bs->cache;
119
120 /* mask out required bits */
121 if (n < 32)
122 res &= (1 << n) - 1;
123
124 bs->head = shift;
125
126 return res;
127 }
128
129
130 #define gst_rtp_mp4g_depay_parent_class parent_class
131 G_DEFINE_TYPE (GstRtpMP4GDepay, gst_rtp_mp4g_depay,
132 GST_TYPE_RTP_BASE_DEPAYLOAD);
133 GST_ELEMENT_REGISTER_DEFINE_WITH_CODE (rtpmp4gdepay, "rtpmp4gdepay",
134 GST_RANK_SECONDARY, GST_TYPE_RTP_MP4G_DEPAY, rtp_element_init (plugin));
135
136 static void gst_rtp_mp4g_depay_finalize (GObject * object);
137
138 static gboolean gst_rtp_mp4g_depay_setcaps (GstRTPBaseDepayload * depayload,
139 GstCaps * caps);
140 static GstBuffer *gst_rtp_mp4g_depay_process (GstRTPBaseDepayload * depayload,
141 GstRTPBuffer * rtp);
142 static gboolean gst_rtp_mp4g_depay_handle_event (GstRTPBaseDepayload * filter,
143 GstEvent * event);
144
145 static GstStateChangeReturn gst_rtp_mp4g_depay_change_state (GstElement *
146 element, GstStateChange transition);
147
148
149 static void
gst_rtp_mp4g_depay_class_init(GstRtpMP4GDepayClass * klass)150 gst_rtp_mp4g_depay_class_init (GstRtpMP4GDepayClass * klass)
151 {
152 GObjectClass *gobject_class;
153 GstElementClass *gstelement_class;
154 GstRTPBaseDepayloadClass *gstrtpbasedepayload_class;
155
156 gobject_class = (GObjectClass *) klass;
157 gstelement_class = (GstElementClass *) klass;
158 gstrtpbasedepayload_class = (GstRTPBaseDepayloadClass *) klass;
159
160 gobject_class->finalize = gst_rtp_mp4g_depay_finalize;
161
162 gstelement_class->change_state = gst_rtp_mp4g_depay_change_state;
163
164 gstrtpbasedepayload_class->process_rtp_packet = gst_rtp_mp4g_depay_process;
165 gstrtpbasedepayload_class->set_caps = gst_rtp_mp4g_depay_setcaps;
166 gstrtpbasedepayload_class->handle_event = gst_rtp_mp4g_depay_handle_event;
167
168 gst_element_class_add_static_pad_template (gstelement_class,
169 &gst_rtp_mp4g_depay_src_template);
170 gst_element_class_add_static_pad_template (gstelement_class,
171 &gst_rtp_mp4g_depay_sink_template);
172
173 gst_element_class_set_static_metadata (gstelement_class,
174 "RTP MPEG4 ES depayloader", "Codec/Depayloader/Network/RTP",
175 "Extracts MPEG4 elementary streams from RTP packets (RFC 3640)",
176 "Wim Taymans <wim.taymans@gmail.com>");
177
178 GST_DEBUG_CATEGORY_INIT (rtpmp4gdepay_debug, "rtpmp4gdepay", 0,
179 "MP4-generic RTP Depayloader");
180 }
181
182 static void
gst_rtp_mp4g_depay_init(GstRtpMP4GDepay * rtpmp4gdepay)183 gst_rtp_mp4g_depay_init (GstRtpMP4GDepay * rtpmp4gdepay)
184 {
185 rtpmp4gdepay->adapter = gst_adapter_new ();
186 rtpmp4gdepay->packets = g_queue_new ();
187 }
188
189 static void
gst_rtp_mp4g_depay_finalize(GObject * object)190 gst_rtp_mp4g_depay_finalize (GObject * object)
191 {
192 GstRtpMP4GDepay *rtpmp4gdepay;
193
194 rtpmp4gdepay = GST_RTP_MP4G_DEPAY (object);
195
196 g_object_unref (rtpmp4gdepay->adapter);
197 rtpmp4gdepay->adapter = NULL;
198 g_queue_free (rtpmp4gdepay->packets);
199 rtpmp4gdepay->packets = NULL;
200
201 G_OBJECT_CLASS (parent_class)->finalize (object);
202 }
203
204 static gint
gst_rtp_mp4g_depay_parse_int(GstStructure * structure,const gchar * field,gint def)205 gst_rtp_mp4g_depay_parse_int (GstStructure * structure, const gchar * field,
206 gint def)
207 {
208 const gchar *str;
209 gint res;
210
211 if ((str = gst_structure_get_string (structure, field)))
212 return atoi (str);
213
214 if (gst_structure_get_int (structure, field, &res))
215 return res;
216
217 return def;
218 }
219
220 static gboolean
gst_rtp_mp4g_depay_setcaps(GstRTPBaseDepayload * depayload,GstCaps * caps)221 gst_rtp_mp4g_depay_setcaps (GstRTPBaseDepayload * depayload, GstCaps * caps)
222 {
223 GstStructure *structure;
224 GstRtpMP4GDepay *rtpmp4gdepay;
225 GstCaps *srccaps = NULL;
226 const gchar *str;
227 gint clock_rate;
228 gint someint;
229 gboolean res;
230
231 rtpmp4gdepay = GST_RTP_MP4G_DEPAY (depayload);
232
233 structure = gst_caps_get_structure (caps, 0);
234
235 if (!gst_structure_get_int (structure, "clock-rate", &clock_rate))
236 clock_rate = 90000; /* default */
237 depayload->clock_rate = clock_rate;
238
239 rtpmp4gdepay->check_adts = FALSE;
240
241 if ((str = gst_structure_get_string (structure, "media"))) {
242 if (strcmp (str, "audio") == 0) {
243 srccaps = gst_caps_new_simple ("audio/mpeg",
244 "mpegversion", G_TYPE_INT, 4, "stream-format", G_TYPE_STRING, "raw",
245 NULL);
246 rtpmp4gdepay->check_adts = TRUE;
247 rtpmp4gdepay->warn_adts = TRUE;
248 } else if (strcmp (str, "video") == 0) {
249 srccaps = gst_caps_new_simple ("video/mpeg",
250 "mpegversion", G_TYPE_INT, 4,
251 "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
252 }
253 }
254 if (srccaps == NULL)
255 goto unknown_media;
256
257 /* these values are optional and have a default value of 0 (no header) */
258 rtpmp4gdepay->sizelength =
259 gst_rtp_mp4g_depay_parse_int (structure, "sizelength", 0);
260 rtpmp4gdepay->indexlength =
261 gst_rtp_mp4g_depay_parse_int (structure, "indexlength", 0);
262 rtpmp4gdepay->indexdeltalength =
263 gst_rtp_mp4g_depay_parse_int (structure, "indexdeltalength", 0);
264 rtpmp4gdepay->ctsdeltalength =
265 gst_rtp_mp4g_depay_parse_int (structure, "ctsdeltalength", 0);
266 rtpmp4gdepay->dtsdeltalength =
267 gst_rtp_mp4g_depay_parse_int (structure, "dtsdeltalength", 0);
268 someint =
269 gst_rtp_mp4g_depay_parse_int (structure, "randomaccessindication", 0);
270 rtpmp4gdepay->randomaccessindication = someint > 0 ? 1 : 0;
271 rtpmp4gdepay->streamstateindication =
272 gst_rtp_mp4g_depay_parse_int (structure, "streamstateindication", 0);
273 rtpmp4gdepay->auxiliarydatasizelength =
274 gst_rtp_mp4g_depay_parse_int (structure, "auxiliarydatasizelength", 0);
275 rtpmp4gdepay->constantSize =
276 gst_rtp_mp4g_depay_parse_int (structure, "constantsize", 0);
277 rtpmp4gdepay->constantDuration =
278 gst_rtp_mp4g_depay_parse_int (structure, "constantduration", 0);
279 rtpmp4gdepay->maxDisplacement =
280 gst_rtp_mp4g_depay_parse_int (structure, "maxdisplacement", 0);
281
282
283 /* get config string */
284 if ((str = gst_structure_get_string (structure, "config"))) {
285 GValue v = { 0 };
286
287 g_value_init (&v, GST_TYPE_BUFFER);
288 if (gst_value_deserialize (&v, str)) {
289 GstBuffer *buffer;
290
291 buffer = gst_value_get_buffer (&v);
292 gst_caps_set_simple (srccaps,
293 "codec_data", GST_TYPE_BUFFER, buffer, NULL);
294 g_value_unset (&v);
295 } else {
296 g_warning ("cannot convert config to buffer");
297 }
298 }
299
300 res = gst_pad_set_caps (depayload->srcpad, srccaps);
301 gst_caps_unref (srccaps);
302
303 return res;
304
305 /* ERRORS */
306 unknown_media:
307 {
308 GST_DEBUG_OBJECT (rtpmp4gdepay, "Unknown media type");
309 return FALSE;
310 }
311 }
312
313 static void
gst_rtp_mp4g_depay_clear_queue(GstRtpMP4GDepay * rtpmp4gdepay)314 gst_rtp_mp4g_depay_clear_queue (GstRtpMP4GDepay * rtpmp4gdepay)
315 {
316 GstBuffer *outbuf;
317
318 while ((outbuf = g_queue_pop_head (rtpmp4gdepay->packets)))
319 gst_buffer_unref (outbuf);
320 }
321
322 static void
gst_rtp_mp4g_depay_reset(GstRtpMP4GDepay * rtpmp4gdepay)323 gst_rtp_mp4g_depay_reset (GstRtpMP4GDepay * rtpmp4gdepay)
324 {
325 gst_adapter_clear (rtpmp4gdepay->adapter);
326 rtpmp4gdepay->max_AU_index = -1;
327 rtpmp4gdepay->next_AU_index = -1;
328 rtpmp4gdepay->prev_AU_index = -1;
329 rtpmp4gdepay->prev_rtptime = -1;
330 rtpmp4gdepay->last_AU_index = -1;
331 gst_rtp_mp4g_depay_clear_queue (rtpmp4gdepay);
332 }
333
334 static void
gst_rtp_mp4g_depay_push_outbuf(GstRtpMP4GDepay * rtpmp4gdepay,GstBuffer * outbuf,guint AU_index)335 gst_rtp_mp4g_depay_push_outbuf (GstRtpMP4GDepay * rtpmp4gdepay,
336 GstBuffer * outbuf, guint AU_index)
337 {
338 gboolean discont = FALSE;
339
340 if (AU_index != rtpmp4gdepay->next_AU_index) {
341 GST_DEBUG_OBJECT (rtpmp4gdepay, "discont, expected AU_index %u",
342 rtpmp4gdepay->next_AU_index);
343 GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_FLAG_DISCONT);
344 discont = TRUE;
345 }
346
347 GST_DEBUG_OBJECT (rtpmp4gdepay, "pushing %sAU_index %u",
348 discont ? "" : "expected ", AU_index);
349
350 gst_rtp_drop_meta (GST_ELEMENT_CAST (rtpmp4gdepay), outbuf, 0);
351 gst_rtp_base_depayload_push (GST_RTP_BASE_DEPAYLOAD (rtpmp4gdepay), outbuf);
352 rtpmp4gdepay->next_AU_index = AU_index + 1;
353 }
354
355 static void
gst_rtp_mp4g_depay_flush_queue(GstRtpMP4GDepay * rtpmp4gdepay)356 gst_rtp_mp4g_depay_flush_queue (GstRtpMP4GDepay * rtpmp4gdepay)
357 {
358 GstBuffer *outbuf;
359 guint AU_index;
360
361 while ((outbuf = g_queue_pop_head (rtpmp4gdepay->packets))) {
362 AU_index = GST_BUFFER_OFFSET (outbuf);
363
364 GST_DEBUG_OBJECT (rtpmp4gdepay, "next available AU_index %u", AU_index);
365
366 gst_rtp_mp4g_depay_push_outbuf (rtpmp4gdepay, outbuf, AU_index);
367 }
368 }
369
370 static void
gst_rtp_mp4g_depay_queue(GstRtpMP4GDepay * rtpmp4gdepay,GstBuffer * outbuf)371 gst_rtp_mp4g_depay_queue (GstRtpMP4GDepay * rtpmp4gdepay, GstBuffer * outbuf)
372 {
373 guint AU_index = GST_BUFFER_OFFSET (outbuf);
374
375 if (rtpmp4gdepay->next_AU_index == -1) {
376 GST_DEBUG_OBJECT (rtpmp4gdepay, "Init AU counter %u", AU_index);
377 rtpmp4gdepay->next_AU_index = AU_index;
378 }
379
380 if (rtpmp4gdepay->next_AU_index == AU_index) {
381 GST_DEBUG_OBJECT (rtpmp4gdepay, "pushing expected AU_index %u", AU_index);
382
383 /* we received the expected packet, push it and flush as much as we can from
384 * the queue */
385 gst_rtp_mp4g_depay_push_outbuf (rtpmp4gdepay, outbuf, AU_index);
386
387 while ((outbuf = g_queue_peek_head (rtpmp4gdepay->packets))) {
388 AU_index = GST_BUFFER_OFFSET (outbuf);
389
390 GST_DEBUG_OBJECT (rtpmp4gdepay, "next available AU_index %u", AU_index);
391
392 if (rtpmp4gdepay->next_AU_index == AU_index) {
393 outbuf = g_queue_pop_head (rtpmp4gdepay->packets);
394 gst_rtp_mp4g_depay_push_outbuf (rtpmp4gdepay, outbuf, AU_index);
395 } else {
396 GST_DEBUG_OBJECT (rtpmp4gdepay, "waiting for next AU_index %u",
397 rtpmp4gdepay->next_AU_index);
398 break;
399 }
400 }
401 } else {
402 GList *list;
403
404 GST_DEBUG_OBJECT (rtpmp4gdepay, "queueing AU_index %u", AU_index);
405
406 /* loop the list to skip strictly smaller AU_index buffers */
407 for (list = rtpmp4gdepay->packets->head; list; list = g_list_next (list)) {
408 guint idx;
409 gint gap;
410
411 idx = GST_BUFFER_OFFSET (GST_BUFFER_CAST (list->data));
412
413 /* compare the new seqnum to the one in the buffer */
414 gap = (gint) (idx - AU_index);
415
416 GST_DEBUG_OBJECT (rtpmp4gdepay, "compare with AU_index %u, gap %d", idx,
417 gap);
418
419 /* AU_index <= idx, we can stop looking */
420 if (G_LIKELY (gap > 0))
421 break;
422 }
423 if (G_LIKELY (list))
424 g_queue_insert_before (rtpmp4gdepay->packets, list, outbuf);
425 else
426 g_queue_push_tail (rtpmp4gdepay->packets, outbuf);
427 }
428 }
429
430 static GstBuffer *
gst_rtp_mp4g_depay_process(GstRTPBaseDepayload * depayload,GstRTPBuffer * rtp)431 gst_rtp_mp4g_depay_process (GstRTPBaseDepayload * depayload, GstRTPBuffer * rtp)
432 {
433 GstRtpMP4GDepay *rtpmp4gdepay;
434 GstBuffer *outbuf = NULL;
435 GstClockTime timestamp;
436
437 rtpmp4gdepay = GST_RTP_MP4G_DEPAY (depayload);
438
439 /* flush remaining data on discont */
440 if (GST_BUFFER_IS_DISCONT (rtp->buffer)) {
441 GST_DEBUG_OBJECT (rtpmp4gdepay, "received DISCONT");
442 gst_adapter_clear (rtpmp4gdepay->adapter);
443 }
444
445 timestamp = GST_BUFFER_PTS (rtp->buffer);
446
447 {
448 gint payload_len, payload_AU;
449 guint8 *payload;
450 guint32 rtptime;
451 guint AU_headers_len;
452 guint AU_size, AU_index, AU_index_delta, payload_AU_size;
453 gboolean M;
454
455 payload_len = gst_rtp_buffer_get_payload_len (rtp);
456 payload = gst_rtp_buffer_get_payload (rtp);
457
458 GST_DEBUG_OBJECT (rtpmp4gdepay, "received payload of %d", payload_len);
459
460 rtptime = gst_rtp_buffer_get_timestamp (rtp);
461 M = gst_rtp_buffer_get_marker (rtp);
462
463 if (rtpmp4gdepay->sizelength > 0) {
464 gint num_AU_headers, AU_headers_bytes, i;
465 GstBsParse bs;
466
467 if (payload_len < 2)
468 goto short_payload;
469
470 /* +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+- .. -+-+-+-+-+-+-+-+-+-+
471 * |AU-headers-length|AU-header|AU-header| |AU-header|padding|
472 * | | (1) | (2) | | (n) * | bits |
473 * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+- .. -+-+-+-+-+-+-+-+-+-+
474 *
475 * The length is 2 bytes and contains the length of the following
476 * AU-headers in bits.
477 */
478 AU_headers_len = (payload[0] << 8) | payload[1];
479 AU_headers_bytes = (AU_headers_len + 7) / 8;
480 num_AU_headers = AU_headers_len / 16;
481
482 GST_DEBUG_OBJECT (rtpmp4gdepay, "AU headers len %d, bytes %d, num %d",
483 AU_headers_len, AU_headers_bytes, num_AU_headers);
484
485 /* skip header */
486 payload += 2;
487 payload_len -= 2;
488
489 if (payload_len < AU_headers_bytes)
490 goto short_payload;
491
492 /* skip special headers, point to first payload AU */
493 payload_AU = 2 + AU_headers_bytes;
494 payload_AU_size = payload_len - AU_headers_bytes;
495
496 if (G_UNLIKELY (rtpmp4gdepay->auxiliarydatasizelength)) {
497 gint aux_size;
498
499 /* point the bitstream parser to the first auxiliary data bit */
500 gst_bs_parse_init (&bs, payload + AU_headers_bytes,
501 payload_len - AU_headers_bytes);
502 aux_size =
503 gst_bs_parse_read (&bs, rtpmp4gdepay->auxiliarydatasizelength);
504 /* convert to bytes */
505 aux_size = (aux_size + 7) / 8;
506 /* AU data then follows auxiliary data */
507 if (payload_AU_size < aux_size)
508 goto short_payload;
509 payload_AU += aux_size;
510 payload_AU_size -= aux_size;
511 }
512
513 /* point the bitstream parser to the first AU header bit */
514 gst_bs_parse_init (&bs, payload, payload_len);
515 AU_index = AU_index_delta = 0;
516
517 for (i = 0; i < num_AU_headers && payload_AU_size > 0; i++) {
518 /* parse AU header
519 * +---------------------------------------+
520 * | AU-size |
521 * +---------------------------------------+
522 * | AU-Index / AU-Index-delta |
523 * +---------------------------------------+
524 * | CTS-flag |
525 * +---------------------------------------+
526 * | CTS-delta |
527 * +---------------------------------------+
528 * | DTS-flag |
529 * +---------------------------------------+
530 * | DTS-delta |
531 * +---------------------------------------+
532 * | RAP-flag |
533 * +---------------------------------------+
534 * | Stream-state |
535 * +---------------------------------------+
536 */
537 AU_size = gst_bs_parse_read (&bs, rtpmp4gdepay->sizelength);
538
539 /* calculate the AU_index, which is only on the first AU of the packet
540 * and the AU_index_delta on the other AUs. This will be used to
541 * reconstruct the AU ordering when interleaving. */
542 if (i == 0) {
543 AU_index = gst_bs_parse_read (&bs, rtpmp4gdepay->indexlength);
544
545 GST_DEBUG_OBJECT (rtpmp4gdepay, "AU index %u", AU_index);
546
547 if (AU_index == 0 && rtpmp4gdepay->prev_AU_index == 0) {
548 gint diff;
549 gint cd;
550
551 /* if we see two consecutive packets with AU_index of 0, we can
552 * assume we have constantDuration packets. Since we don't have
553 * the index we must use the AU duration to calculate the
554 * index. Get the diff between the timestamps first, this can be
555 * positive or negative. */
556 if (rtpmp4gdepay->prev_rtptime <= rtptime)
557 diff = rtptime - rtpmp4gdepay->prev_rtptime;
558 else
559 diff = -(rtpmp4gdepay->prev_rtptime - rtptime);
560
561 /* if no constantDuration was given, make one */
562 if (rtpmp4gdepay->constantDuration != 0) {
563 cd = rtpmp4gdepay->constantDuration;
564 GST_DEBUG_OBJECT (depayload, "using constantDuration %d", cd);
565 } else if (rtpmp4gdepay->prev_AU_num > 0) {
566 /* use number of packets and of previous frame */
567 cd = diff / rtpmp4gdepay->prev_AU_num;
568 GST_DEBUG_OBJECT (depayload, "guessing constantDuration %d", cd);
569 if (!GST_BUFFER_IS_DISCONT (rtp->buffer)) {
570 /* rfc3640 - 3.2.3.2
571 * if we see two consecutive packets with AU_index of 0 and
572 * there has been no discontinuity, we must conclude that this
573 * value of constantDuration is correct from now on. */
574 GST_DEBUG_OBJECT (depayload,
575 "constantDuration of %d detected", cd);
576 rtpmp4gdepay->constantDuration = cd;
577 }
578 } else {
579 /* assume this frame has the same number of packets as the
580 * previous one */
581 cd = diff / num_AU_headers;
582 GST_DEBUG_OBJECT (depayload, "guessing constantDuration %d", cd);
583 }
584
585 if (cd > 0) {
586 /* get the number of packets by dividing with the duration */
587 diff /= cd;
588 } else {
589 diff = 0;
590 }
591
592 rtpmp4gdepay->last_AU_index += diff;
593 rtpmp4gdepay->prev_AU_index = AU_index;
594
595 AU_index = rtpmp4gdepay->last_AU_index;
596
597 GST_DEBUG_OBJECT (rtpmp4gdepay, "diff %d, AU index %u", diff,
598 AU_index);
599 } else {
600 rtpmp4gdepay->prev_AU_index = AU_index;
601 rtpmp4gdepay->last_AU_index = AU_index;
602 }
603
604 /* keep track of the highest AU_index */
605 if (rtpmp4gdepay->max_AU_index != -1
606 && rtpmp4gdepay->max_AU_index <= AU_index) {
607 GST_DEBUG_OBJECT (rtpmp4gdepay, "new interleave group, flushing");
608 /* a new interleave group started, flush */
609 gst_rtp_mp4g_depay_flush_queue (rtpmp4gdepay);
610 }
611 if (G_UNLIKELY (!rtpmp4gdepay->maxDisplacement &&
612 rtpmp4gdepay->max_AU_index != -1
613 && rtpmp4gdepay->max_AU_index >= AU_index)) {
614 GstBuffer *outbuf;
615
616 /* some broken non-interleaved streams have AU-index jumping around
617 * all over the place, apparently assuming receiver disregards */
618 GST_DEBUG_OBJECT (rtpmp4gdepay, "non-interleaved broken AU indices;"
619 " forcing continuous flush");
620 /* reset AU to avoid repeated DISCONT in such case */
621 outbuf = g_queue_peek_head (rtpmp4gdepay->packets);
622 if (G_LIKELY (outbuf)) {
623 rtpmp4gdepay->next_AU_index = GST_BUFFER_OFFSET (outbuf);
624 gst_rtp_mp4g_depay_flush_queue (rtpmp4gdepay);
625 }
626 /* rebase next_AU_index to current rtp's first AU_index */
627 rtpmp4gdepay->next_AU_index = AU_index;
628 }
629 rtpmp4gdepay->prev_rtptime = rtptime;
630 rtpmp4gdepay->prev_AU_num = num_AU_headers;
631 } else {
632 AU_index_delta =
633 gst_bs_parse_read (&bs, rtpmp4gdepay->indexdeltalength);
634 AU_index += AU_index_delta + 1;
635 }
636 /* keep track of highest AU_index */
637 if (rtpmp4gdepay->max_AU_index == -1
638 || AU_index > rtpmp4gdepay->max_AU_index)
639 rtpmp4gdepay->max_AU_index = AU_index;
640
641 /* the presentation time offset, a 2s-complement value, we need this to
642 * calculate the timestamp on the output packet. */
643 if (rtpmp4gdepay->ctsdeltalength > 0) {
644 if (gst_bs_parse_read (&bs, 1))
645 gst_bs_parse_read (&bs, rtpmp4gdepay->ctsdeltalength);
646 }
647 /* the decoding time offset, a 2s-complement value */
648 if (rtpmp4gdepay->dtsdeltalength > 0) {
649 if (gst_bs_parse_read (&bs, 1))
650 gst_bs_parse_read (&bs, rtpmp4gdepay->dtsdeltalength);
651 }
652 /* RAP-flag to indicate that the AU contains a keyframe */
653 if (rtpmp4gdepay->randomaccessindication)
654 gst_bs_parse_read (&bs, 1);
655 /* stream-state */
656 if (rtpmp4gdepay->streamstateindication > 0)
657 gst_bs_parse_read (&bs, rtpmp4gdepay->streamstateindication);
658
659 GST_DEBUG_OBJECT (rtpmp4gdepay, "size %d, index %d, delta %d", AU_size,
660 AU_index, AU_index_delta);
661
662 /* fragmented pakets have the AU_size set to the size of the
663 * unfragmented AU. */
664 if (AU_size > payload_AU_size)
665 AU_size = payload_AU_size;
666
667 /* collect stuff in the adapter, strip header from payload and push in
668 * the adapter */
669 outbuf =
670 gst_rtp_buffer_get_payload_subbuffer (rtp, payload_AU, AU_size);
671 gst_adapter_push (rtpmp4gdepay->adapter, outbuf);
672
673 if (M) {
674 guint32 v = 0;
675 guint avail;
676
677 /* packet is complete, flush */
678 avail = gst_adapter_available (rtpmp4gdepay->adapter);
679
680 /* Some broken senders send ADTS headers (e.g. some Sony cameras).
681 * Try to detect those and skip them (still needs config set), but
682 * don't check every frame, only the first (unless we detect ADTS) */
683 if (rtpmp4gdepay->check_adts && avail >= 7) {
684 if (gst_adapter_masked_scan_uint32_peek (rtpmp4gdepay->adapter,
685 0xfffe0000, 0xfff00000, 0, 4, &v) == 0) {
686 guint adts_hdr_len = (((v >> 16) & 0x1) == 0) ? 9 : 7;
687 if (avail > adts_hdr_len) {
688 if (rtpmp4gdepay->warn_adts) {
689 GST_WARNING_OBJECT (rtpmp4gdepay, "Detected ADTS header of "
690 "%u bytes, skipping", adts_hdr_len);
691 rtpmp4gdepay->warn_adts = FALSE;
692 }
693 gst_adapter_flush (rtpmp4gdepay->adapter, adts_hdr_len);
694 avail -= adts_hdr_len;
695 }
696 } else {
697 rtpmp4gdepay->check_adts = FALSE;
698 rtpmp4gdepay->warn_adts = TRUE;
699 }
700 }
701
702 outbuf = gst_adapter_take_buffer (rtpmp4gdepay->adapter, avail);
703
704 /* copy some of the fields we calculated above on the buffer. We also
705 * copy the AU_index so that we can sort the packets in our queue. */
706 GST_BUFFER_PTS (outbuf) = timestamp;
707 GST_BUFFER_OFFSET (outbuf) = AU_index;
708
709 if (rtpmp4gdepay->constantDuration != 0) {
710 /* if we have constantDuration, calculate timestamp for next AU
711 * in this RTP packet. */
712 timestamp += (rtpmp4gdepay->constantDuration * GST_SECOND) /
713 depayload->clock_rate;
714 } else {
715 /* otherwise, make sure we don't use the timestamp again for other
716 * AUs. */
717 timestamp = GST_CLOCK_TIME_NONE;
718 }
719
720 GST_DEBUG_OBJECT (depayload,
721 "pushing buffer of size %" G_GSIZE_FORMAT,
722 gst_buffer_get_size (outbuf));
723
724 gst_rtp_mp4g_depay_queue (rtpmp4gdepay, outbuf);
725
726 }
727 payload_AU += AU_size;
728 payload_AU_size -= AU_size;
729 }
730 } else {
731 /* push complete buffer in adapter */
732 outbuf = gst_rtp_buffer_get_payload_subbuffer (rtp, 0, payload_len);
733 gst_adapter_push (rtpmp4gdepay->adapter, outbuf);
734
735 /* if this was the last packet of the VOP, create and push a buffer */
736 if (M) {
737 guint avail;
738
739 avail = gst_adapter_available (rtpmp4gdepay->adapter);
740
741 outbuf = gst_adapter_take_buffer (rtpmp4gdepay->adapter, avail);
742
743 GST_DEBUG ("gst_rtp_mp4g_depay_chain: pushing buffer of size %"
744 G_GSIZE_FORMAT, gst_buffer_get_size (outbuf));
745
746 return outbuf;
747 }
748 }
749 }
750
751 return NULL;
752
753 /* ERRORS */
754 short_payload:
755 {
756 GST_ELEMENT_WARNING (rtpmp4gdepay, STREAM, DECODE,
757 ("Packet payload was too short."), (NULL));
758 return NULL;
759 }
760 }
761
762 static gboolean
gst_rtp_mp4g_depay_handle_event(GstRTPBaseDepayload * filter,GstEvent * event)763 gst_rtp_mp4g_depay_handle_event (GstRTPBaseDepayload * filter, GstEvent * event)
764 {
765 gboolean ret;
766 GstRtpMP4GDepay *rtpmp4gdepay;
767
768 rtpmp4gdepay = GST_RTP_MP4G_DEPAY (filter);
769
770 switch (GST_EVENT_TYPE (event)) {
771 case GST_EVENT_FLUSH_STOP:
772 gst_rtp_mp4g_depay_reset (rtpmp4gdepay);
773 break;
774 default:
775 break;
776 }
777
778 ret =
779 GST_RTP_BASE_DEPAYLOAD_CLASS (parent_class)->handle_event (filter, event);
780
781 return ret;
782 }
783
784 static GstStateChangeReturn
gst_rtp_mp4g_depay_change_state(GstElement * element,GstStateChange transition)785 gst_rtp_mp4g_depay_change_state (GstElement * element,
786 GstStateChange transition)
787 {
788 GstRtpMP4GDepay *rtpmp4gdepay;
789 GstStateChangeReturn ret;
790
791 rtpmp4gdepay = GST_RTP_MP4G_DEPAY (element);
792
793 switch (transition) {
794 case GST_STATE_CHANGE_READY_TO_PAUSED:
795 gst_rtp_mp4g_depay_reset (rtpmp4gdepay);
796 break;
797 default:
798 break;
799 }
800
801 ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
802
803 switch (transition) {
804 case GST_STATE_CHANGE_PAUSED_TO_READY:
805 gst_rtp_mp4g_depay_reset (rtpmp4gdepay);
806 break;
807 default:
808 break;
809 }
810 return ret;
811 }
812