gstrtph264depay.c 29.3 KB
Newer Older
1
/* GStreamer
2
 * Copyright (C) <2006> Wim Taymans <wim.taymans@gmail.com>
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#  include "config.h"
#endif

24
#include <stdio.h>
25
26
27
28
29
#include <string.h>

#include <gst/rtp/gstrtpbuffer.h>
#include "gstrtph264depay.h"

30
31
32
GST_DEBUG_CATEGORY_STATIC (rtph264depay_debug);
#define GST_CAT_DEFAULT (rtph264depay_debug)

33
#define DEFAULT_BYTE_STREAM	TRUE
34
#define DEFAULT_ACCESS_UNIT	FALSE
35
36
37
38
39

enum
{
  PROP_0,
  PROP_BYTE_STREAM,
40
  PROP_ACCESS_UNIT,
41
42
43
44
  PROP_LAST
};


45
46
/* 3 zero bytes syncword */
static const guint8 sync_bytes[] = { 0, 0, 0, 1 };
47

48
static GstStaticPadTemplate gst_rtp_h264_depay_src_template =
49
    GST_STATIC_PAD_TEMPLATE ("src",
50
51
    GST_PAD_SRC,
    GST_PAD_ALWAYS,
52
53
54
55
    GST_STATIC_CAPS ("video/x-h264, "
        "stream-format = (string) avc, alignment = (string) au; "
        "video/x-h264, "
        "stream-format = (string) byte-stream, alignment = (string) { nal, au }")
56
57
58
59
60
61
62
63
    );

static GstStaticPadTemplate gst_rtp_h264_depay_sink_template =
GST_STATIC_PAD_TEMPLATE ("sink",
    GST_PAD_SINK,
    GST_PAD_ALWAYS,
    GST_STATIC_CAPS ("application/x-rtp, "
        "media = (string) \"video\", "
64
        "payload = (int) " GST_RTP_PAYLOAD_DYNAMIC_STRING ", "
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
        "clock-rate = (int) 90000, " "encoding-name = (string) \"H264\"")
        /** optional parameters **/
    /* "profile-level-id = (string) ANY, " */
    /* "max-mbps = (string) ANY, " */
    /* "max-fs = (string) ANY, " */
    /* "max-cpb = (string) ANY, " */
    /* "max-dpb = (string) ANY, " */
    /* "max-br = (string) ANY, " */
    /* "redundant-pic-cap = (string) { \"0\", \"1\" }, " */
    /* "sprop-parameter-sets = (string) ANY, " */
    /* "parameter-add = (string) { \"0\", \"1\" }, " */
    /* "packetization-mode = (string) { \"0\", \"1\", \"2\" }, " */
    /* "sprop-interleaving-depth = (string) ANY, " */
    /* "sprop-deint-buf-req = (string) ANY, " */
    /* "deint-buf-cap = (string) ANY, " */
    /* "sprop-init-buf-time = (string) ANY, " */
    /* "sprop-max-don-diff = (string) ANY, " */
    /* "max-rcmd-nalu-size = (string) ANY " */
    );

Wim Taymans's avatar
Wim Taymans committed
85
86
#define gst_rtp_h264_depay_parent_class parent_class
G_DEFINE_TYPE (GstRtpH264Depay, gst_rtp_h264_depay,
Wim Taymans's avatar
Wim Taymans committed
87
    GST_TYPE_RTP_BASE_DEPAYLOAD);
88
89

static void gst_rtp_h264_depay_finalize (GObject * object);
90
91
92
93
static void gst_rtp_h264_depay_set_property (GObject * object, guint prop_id,
    const GValue * value, GParamSpec * pspec);
static void gst_rtp_h264_depay_get_property (GObject * object, guint prop_id,
    GValue * value, GParamSpec * pspec);
94
95
96
97

static GstStateChangeReturn gst_rtp_h264_depay_change_state (GstElement *
    element, GstStateChange transition);

Wim Taymans's avatar
Wim Taymans committed
98
static GstBuffer *gst_rtp_h264_depay_process (GstRTPBaseDepayload * depayload,
99
    GstBuffer * buf);
Wim Taymans's avatar
Wim Taymans committed
100
static gboolean gst_rtp_h264_depay_setcaps (GstRTPBaseDepayload * filter,
101
    GstCaps * caps);
Wim Taymans's avatar
Wim Taymans committed
102
static gboolean gst_rtp_h264_depay_handle_event (GstRTPBaseDepayload * depay,
103
    GstEvent * event);
104
105
106
107
108
109

static void
gst_rtp_h264_depay_class_init (GstRtpH264DepayClass * klass)
{
  GObjectClass *gobject_class;
  GstElementClass *gstelement_class;
Wim Taymans's avatar
Wim Taymans committed
110
  GstRTPBaseDepayloadClass *gstrtpbasedepayload_class;
111
112
113

  gobject_class = (GObjectClass *) klass;
  gstelement_class = (GstElementClass *) klass;
Wim Taymans's avatar
Wim Taymans committed
114
  gstrtpbasedepayload_class = (GstRTPBaseDepayloadClass *) klass;
115
116
117

  gobject_class->finalize = gst_rtp_h264_depay_finalize;

118
119
120
121
122
  gobject_class->set_property = gst_rtp_h264_depay_set_property;
  gobject_class->get_property = gst_rtp_h264_depay_get_property;

  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_BYTE_STREAM,
      g_param_spec_boolean ("byte-stream", "Byte Stream",
123
124
          "Generate byte stream format of NALU (deprecated; use caps)",
          DEFAULT_BYTE_STREAM, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
125
126
  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_ACCESS_UNIT,
      g_param_spec_boolean ("access-unit", "Access Unit",
127
128
          "Merge NALU into AU (picture) (deprecated; use caps)",
          DEFAULT_ACCESS_UNIT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
129

Wim Taymans's avatar
Wim Taymans committed
130
131
132
133
134
135
136
137
138
  gst_element_class_add_pad_template (gstelement_class,
      gst_static_pad_template_get (&gst_rtp_h264_depay_src_template));
  gst_element_class_add_pad_template (gstelement_class,
      gst_static_pad_template_get (&gst_rtp_h264_depay_sink_template));

  gst_element_class_set_details_simple (gstelement_class,
      "RTP H264 depayloader", "Codec/Depayloader/Network/RTP",
      "Extracts H264 video from RTP packets (RFC 3984)",
      "Wim Taymans <wim.taymans@gmail.com>");
139
  gstelement_class->change_state = gst_rtp_h264_depay_change_state;
140

Wim Taymans's avatar
Wim Taymans committed
141
142
143
  gstrtpbasedepayload_class->process = gst_rtp_h264_depay_process;
  gstrtpbasedepayload_class->set_caps = gst_rtp_h264_depay_setcaps;
  gstrtpbasedepayload_class->handle_event = gst_rtp_h264_depay_handle_event;
144

145
146
  GST_DEBUG_CATEGORY_INIT (rtph264depay_debug, "rtph264depay", 0,
      "H264 Video RTP Depayloader");
147
148
149
}

static void
Wim Taymans's avatar
Wim Taymans committed
150
gst_rtp_h264_depay_init (GstRtpH264Depay * rtph264depay)
151
152
{
  rtph264depay->adapter = gst_adapter_new ();
153
  rtph264depay->picture_adapter = gst_adapter_new ();
154
  rtph264depay->byte_stream = DEFAULT_BYTE_STREAM;
155
  rtph264depay->merge = DEFAULT_ACCESS_UNIT;
156
157
}

158
159
160
161
162
163
164
165
166
static void
gst_rtp_h264_depay_reset (GstRtpH264Depay * rtph264depay)
{
  gst_adapter_clear (rtph264depay->adapter);
  rtph264depay->wait_start = TRUE;
  gst_adapter_clear (rtph264depay->picture_adapter);
  rtph264depay->picture_start = FALSE;
  rtph264depay->last_keyframe = FALSE;
  rtph264depay->last_ts = 0;
167
  rtph264depay->current_fu_type = 0;
168
169
}

170
171
172
173
174
175
176
static void
gst_rtp_h264_depay_finalize (GObject * object)
{
  GstRtpH264Depay *rtph264depay;

  rtph264depay = GST_RTP_H264_DEPAY (object);

177
178
179
  if (rtph264depay->codec_data)
    gst_buffer_unref (rtph264depay->codec_data);

180
  g_object_unref (rtph264depay->adapter);
181
  g_object_unref (rtph264depay->picture_adapter);
182
183
184
185

  G_OBJECT_CLASS (parent_class)->finalize (object);
}

186
187
188
189
190
191
192
193
194
195
196
197
static void
gst_rtp_h264_depay_set_property (GObject * object, guint prop_id,
    const GValue * value, GParamSpec * pspec)
{
  GstRtpH264Depay *rtph264depay;

  rtph264depay = GST_RTP_H264_DEPAY (object);

  switch (prop_id) {
    case PROP_BYTE_STREAM:
      rtph264depay->byte_stream = g_value_get_boolean (value);
      break;
198
199
200
    case PROP_ACCESS_UNIT:
      rtph264depay->merge = g_value_get_boolean (value);
      break;
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}

static void
gst_rtp_h264_depay_get_property (GObject * object, guint prop_id,
    GValue * value, GParamSpec * pspec)
{
  GstRtpH264Depay *rtph264depay;

  rtph264depay = GST_RTP_H264_DEPAY (object);

  switch (prop_id) {
    case PROP_BYTE_STREAM:
      g_value_set_boolean (value, rtph264depay->byte_stream);
      break;
219
220
221
    case PROP_ACCESS_UNIT:
      g_value_set_boolean (value, rtph264depay->merge);
      break;
222
223
224
225
226
227
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}

228
229
230
231
232
233
234
235
static void
gst_rtp_h264_depay_negotiate (GstRtpH264Depay * rtph264depay)
{
  GstCaps *caps;
  gint byte_stream = -1;
  gint merge = -1;

  caps =
Wim Taymans's avatar
Wim Taymans committed
236
      gst_pad_get_allowed_caps (GST_RTP_BASE_DEPAYLOAD_SRCPAD (rtph264depay));
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286

  GST_DEBUG_OBJECT (rtph264depay, "allowed caps: %" GST_PTR_FORMAT, caps);

  if (caps) {
    if (gst_caps_get_size (caps) > 0) {
      GstStructure *s = gst_caps_get_structure (caps, 0);
      const gchar *str = NULL;

      if ((str = gst_structure_get_string (s, "stream-format"))) {
        if (strcmp (str, "avc") == 0) {
          byte_stream = FALSE;
        } else if (strcmp (str, "byte-stream") == 0) {
          byte_stream = TRUE;
        } else {
          GST_DEBUG_OBJECT (rtph264depay, "unknown stream-format: %s", str);
        }
      }

      if ((str = gst_structure_get_string (s, "alignment"))) {
        if (strcmp (str, "au") == 0) {
          merge = TRUE;
        } else if (strcmp (str, "nal") == 0) {
          merge = FALSE;
        } else {
          GST_DEBUG_OBJECT (rtph264depay, "unknown alignment: %s", str);
        }
      }
    }
    gst_caps_unref (caps);
  }

  if (byte_stream >= 0) {
    GST_DEBUG_OBJECT (rtph264depay, "downstream requires byte-stream %d",
        byte_stream);
    if (rtph264depay->byte_stream != byte_stream) {
      GST_WARNING_OBJECT (rtph264depay,
          "overriding property setting based on caps");
      rtph264depay->byte_stream = byte_stream;
    }
  }
  if (merge >= 0) {
    GST_DEBUG_OBJECT (rtph264depay, "downstream requires merge %d", merge);
    if (rtph264depay->merge != merge) {
      GST_WARNING_OBJECT (rtph264depay,
          "overriding property setting based on caps");
      rtph264depay->merge = merge;
    }
  }
}

287
static gboolean
Wim Taymans's avatar
Wim Taymans committed
288
gst_rtp_h264_depay_setcaps (GstRTPBaseDepayload * depayload, GstCaps * caps)
289
{
290
291
  GstCaps *srccaps;
  gint clock_rate;
292
  GstStructure *structure = gst_caps_get_structure (caps, 0);
293
  GstRtpH264Depay *rtph264depay;
294
295
  const gchar *ps, *profile;
  GstBuffer *codec_data;
Wim Taymans's avatar
Wim Taymans committed
296
297
  GstMapInfo map;
  guint8 *ptr;
298
  gboolean res;
299
300

  rtph264depay = GST_RTP_H264_DEPAY (depayload);
301

302
303
  if (!gst_structure_get_int (structure, "clock-rate", &clock_rate))
    clock_rate = 90000;
304
  depayload->clock_rate = clock_rate;
305

306
  srccaps = gst_caps_new_empty_simple ("video/x-h264");
307

308
309
310
311
312
  /* Base64 encoded, comma separated config NALs */
  ps = gst_structure_get_string (structure, "sprop-parameter-sets");
  /* hex: AVCProfileIndication:8 | profile_compat:8 | AVCLevelIndication:8 */
  profile = gst_structure_get_string (structure, "profile-level-id");

313
314
315
  /* negotiate with downstream w.r.t. output format and alignment */
  gst_rtp_h264_depay_negotiate (rtph264depay);

316
317
318
  if (rtph264depay->byte_stream && ps != NULL) {
    /* for bytestream we only need the parameter sets but we don't error out
     * when they are not there, we assume they are in the stream. */
319
320
321
322
323
324
    gchar **params;
    guint len, total;
    gint i;

    params = g_strsplit (ps, ",", 0);

325
326
    /* count total number of bytes in base64. Also include the sync bytes in
     * front of the params. */
327
328
329
    len = 0;
    for (i = 0; params[i]; i++) {
      len += strlen (params[i]);
330
      len += sizeof (sync_bytes);
331
332
333
    }
    /* we seriously overshoot the length, but it's fine. */
    codec_data = gst_buffer_new_and_alloc (len);
Wim Taymans's avatar
Wim Taymans committed
334

Wim Taymans's avatar
Wim Taymans committed
335
336
    gst_buffer_map (codec_data, &map, GST_MAP_WRITE);
    ptr = map.data;
337
338
    total = 0;
    for (i = 0; params[i]; i++) {
339
340
341
      guint save = 0;
      gint state = 0;

342
      GST_DEBUG_OBJECT (depayload, "decoding param %d (%s)", i, params[i]);
Wim Taymans's avatar
Wim Taymans committed
343
344
      memcpy (ptr, sync_bytes, sizeof (sync_bytes));
      ptr += sizeof (sync_bytes);
345
      len =
Wim Taymans's avatar
Wim Taymans committed
346
          g_base64_decode_step (params[i], strlen (params[i]), ptr, &state,
347
348
          &save);
      GST_DEBUG_OBJECT (depayload, "decoded %d bytes", len);
349
      total += len + sizeof (sync_bytes);
Wim Taymans's avatar
Wim Taymans committed
350
      ptr += len;
351
    }
Wim Taymans's avatar
Wim Taymans committed
352
353
    gst_buffer_unmap (codec_data, &map);
    gst_buffer_resize (codec_data, 0, total);
354
    g_strfreev (params);
355

356
357
358
359
360
361
    /* keep the codec_data, we need to send it as the first buffer. We cannot
     * push it in the adapter because the adapter might be flushed on discont.
     */
    if (rtph264depay->codec_data)
      gst_buffer_unref (rtph264depay->codec_data);
    rtph264depay->codec_data = codec_data;
362
363
364
365
366
367
  } else if (!rtph264depay->byte_stream) {
    gchar **params;
    guint8 **sps, **pps;
    guint len, num_sps, num_pps;
    gint i;

368
    if (ps == NULL)
369
370
371
372
373
374
375
376
377
378
379
380
381
382
      goto incomplete_caps;

    params = g_strsplit (ps, ",", 0);
    len = g_strv_length (params);

    GST_DEBUG_OBJECT (depayload, "we have %d params", len);

    sps = g_new0 (guint8 *, len + 1);
    pps = g_new0 (guint8 *, len + 1);
    num_sps = num_pps = 0;

    /* start with 7 bytes header */
    len = 7;
    for (i = 0; params[i]; i++) {
383
      gsize nal_len;
384
      guint8 *nalp;
385
386
      guint save = 0;
      gint state = 0;
387
388
389

      nal_len = strlen (params[i]);
      nalp = g_malloc (nal_len + 2);
390
391
392

      nal_len =
          g_base64_decode_step (params[i], nal_len, nalp + 2, &state, &save);
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
      nalp[0] = (nal_len >> 8) & 0xff;
      nalp[1] = nal_len & 0xff;
      len += nal_len + 2;

      /* copy to the right list */
      if ((nalp[2] & 0x1f) == 7) {
        GST_DEBUG_OBJECT (depayload, "adding param %d as SPS %d", i, num_sps);
        sps[num_sps++] = nalp;
      } else {
        GST_DEBUG_OBJECT (depayload, "adding param %d as PPS %d", i, num_pps);
        pps[num_pps++] = nalp;
      }
    }
    g_strfreev (params);

408
409
410
411
412
413
    if (num_sps == 0 || (GST_READ_UINT16_BE (sps[0]) < 3) || num_pps == 0) {
      g_strfreev ((gchar **) pps);
      g_strfreev ((gchar **) sps);
      goto incomplete_caps;
    }

414
    codec_data = gst_buffer_new_and_alloc (len);
Wim Taymans's avatar
Wim Taymans committed
415

Wim Taymans's avatar
Wim Taymans committed
416
417
    gst_buffer_map (codec_data, &map, GST_MAP_WRITE);
    ptr = map.data;
418
419

    /* 8 bits version == 1 */
Wim Taymans's avatar
Wim Taymans committed
420
    *ptr++ = 1;
421
422
423
424
425
    if (profile) {
      guint32 profile_id;

      /* hex: AVCProfileIndication:8 | profile_compat:8 | AVCLevelIndication:8 */
      sscanf (profile, "%6x", &profile_id);
Wim Taymans's avatar
Wim Taymans committed
426
427
428
      *ptr++ = (profile_id >> 16) & 0xff;
      *ptr++ = (profile_id >> 8) & 0xff;
      *ptr++ = profile_id & 0xff;
429
430
    } else {
      /* extract from SPS */
Wim Taymans's avatar
Wim Taymans committed
431
432
433
      *ptr++ = sps[0][3];
      *ptr++ = sps[0][4];
      *ptr++ = sps[0][5];
434
    }
435
    /* 6 bits reserved | 2 bits lengthSizeMinusOn */
Wim Taymans's avatar
Wim Taymans committed
436
    *ptr++ = 0xff;
437
    /* 3 bits reserved | 5 bits numOfSequenceParameterSets */
Wim Taymans's avatar
Wim Taymans committed
438
    *ptr++ = 0xe0 | (num_sps & 0x1f);
439
440
441
442
443

    /* copy all SPS */
    for (i = 0; sps[i]; i++) {
      len = ((sps[i][0] << 8) | sps[i][1]) + 2;
      GST_DEBUG_OBJECT (depayload, "copy SPS %d of length %d", i, len);
Wim Taymans's avatar
Wim Taymans committed
444
      memcpy (ptr, sps[i], len);
445
      g_free (sps[i]);
Wim Taymans's avatar
Wim Taymans committed
446
      ptr += len;
447
448
449
    }
    g_free (sps);
    /* 8 bits numOfPictureParameterSets */
Wim Taymans's avatar
Wim Taymans committed
450
    *ptr++ = num_pps;
451
452
453
454
    /* copy all PPS */
    for (i = 0; pps[i]; i++) {
      len = ((pps[i][0] << 8) | pps[i][1]) + 2;
      GST_DEBUG_OBJECT (depayload, "copy PPS %d of length %d", i, len);
Wim Taymans's avatar
Wim Taymans committed
455
      memcpy (ptr, pps[i], len);
456
      g_free (pps[i]);
Wim Taymans's avatar
Wim Taymans committed
457
      ptr += len;
458
459
    }
    g_free (pps);
Wim Taymans's avatar
Wim Taymans committed
460
461
    gst_buffer_resize (codec_data, 0, ptr - map.data);
    gst_buffer_unmap (codec_data, &map);
462
463
464

    gst_caps_set_simple (srccaps,
        "codec_data", GST_TYPE_BUFFER, codec_data, NULL);
465
    gst_buffer_unref (codec_data);
466
467
  }

468
469
470
471
  gst_caps_set_simple (srccaps, "stream-format", G_TYPE_STRING,
      rtph264depay->byte_stream ? "byte-stream" : "avc",
      "alignment", G_TYPE_STRING, rtph264depay->merge ? "au" : "nal", NULL);

472
  res = gst_pad_set_caps (depayload->srcpad, srccaps);
473
474
  gst_caps_unref (srccaps);

475
  return res;
476
477
478
479
480

  /* ERRORS */
incomplete_caps:
  {
    GST_DEBUG_OBJECT (depayload, "we have incomplete caps");
481
    gst_caps_unref (srccaps);
482
483
    return FALSE;
  }
484
485
}

486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
static GstBuffer *
gst_rtp_h264_complete_au (GstRtpH264Depay * rtph264depay,
    GstClockTime * out_timestamp, gboolean * out_keyframe)
{
  guint outsize;
  GstBuffer *outbuf;

  /* we had a picture in the adapter and we completed it */
  GST_DEBUG_OBJECT (rtph264depay, "taking completed AU");
  outsize = gst_adapter_available (rtph264depay->picture_adapter);
  outbuf = gst_adapter_take_buffer (rtph264depay->picture_adapter, outsize);

  *out_timestamp = rtph264depay->last_ts;
  *out_keyframe = rtph264depay->last_keyframe;

  rtph264depay->last_keyframe = FALSE;
  rtph264depay->picture_start = FALSE;

  return outbuf;
}

507
508
509
/* SPS/PPS/IDR considered key, all others DELTA;
 * so downstream waiting for keyframe can pick up at SPS/PPS/IDR */
#define NAL_TYPE_IS_KEY(nt) (((nt) == 5) || ((nt) == 7) || ((nt) == 8))
510

511
static GstBuffer *
512
gst_rtp_h264_depay_handle_nal (GstRtpH264Depay * rtph264depay, GstBuffer * nal,
513
    GstClockTime in_timestamp, gboolean marker)
514
{
Wim Taymans's avatar
Wim Taymans committed
515
  GstRTPBaseDepayload *depayload = GST_RTP_BASE_DEPAYLOAD (rtph264depay);
516
  gint nal_type;
Wim Taymans's avatar
Wim Taymans committed
517
  GstMapInfo map;
518
  GstBuffer *outbuf = NULL;
519
  GstClockTime out_timestamp;
520
  gboolean keyframe, out_keyframe;
521

Wim Taymans's avatar
Wim Taymans committed
522
523
  gst_buffer_map (nal, &map, GST_MAP_READ);
  if (G_UNLIKELY (map.size < 5))
524
    goto short_nal;
525

Wim Taymans's avatar
Wim Taymans committed
526
  nal_type = map.data[4] & 0x1f;
527
528
  GST_DEBUG_OBJECT (rtph264depay, "handle NAL type %d", nal_type);

529
  keyframe = NAL_TYPE_IS_KEY (nal_type);
530

531
  out_keyframe = keyframe;
532
533
534
535
536
537
538
539
540
541
542
543
544
545
  out_timestamp = in_timestamp;

  if (rtph264depay->merge) {
    gboolean start = FALSE, complete = FALSE;

    /* consider a coded slices (IDR or not) to start a picture,
     * (so ending the previous one) if first_mb_in_slice == 0
     * (non-0 is part of previous one) */
    /* NOTE this is not entirely according to Access Unit specs in 7.4.1.2.4,
     * but in practice it works in sane cases, needs not much parsing,
     * and also works with broken frame_num in NAL (where spec-wise would fail) */
    if (nal_type == 1 || nal_type == 2 || nal_type == 5) {
      /* we have a picture start */
      start = TRUE;
Wim Taymans's avatar
Wim Taymans committed
546
      if (map.data[5] & 0x80) {
547
548
        /* first_mb_in_slice == 0 completes a picture */
        complete = TRUE;
549
      }
550
551
552
    } else if (nal_type >= 6 && nal_type <= 9) {
      /* SEI, SPS, PPS, AU terminate picture */
      complete = TRUE;
553
    }
554
    GST_DEBUG_OBJECT (depayload, "start %d, complete %d", start, complete);
555

556
557
558
    if (complete && rtph264depay->picture_start)
      outbuf = gst_rtp_h264_complete_au (rtph264depay, &out_timestamp,
          &out_keyframe);
559
560

    /* add to adapter */
Wim Taymans's avatar
Wim Taymans committed
561
562
    gst_buffer_unmap (nal, &map);

563
564
565
    GST_DEBUG_OBJECT (depayload, "adding NAL to picture adapter");
    gst_adapter_push (rtph264depay->picture_adapter, nal);
    rtph264depay->last_ts = in_timestamp;
566
    rtph264depay->last_keyframe |= keyframe;
567
    rtph264depay->picture_start |= start;
568
569
570
571

    if (marker)
      outbuf = gst_rtp_h264_complete_au (rtph264depay, &out_timestamp,
          &out_keyframe);
572
573
574
575
  } else {
    /* no merge, output is input nal */
    GST_DEBUG_OBJECT (depayload, "using NAL as output");
    outbuf = nal;
Wim Taymans's avatar
Wim Taymans committed
576
    gst_buffer_unmap (nal, &map);
577
  }
578

579
580
581
582
  if (outbuf) {
    /* prepend codec_data */
    if (rtph264depay->codec_data) {
      GST_DEBUG_OBJECT (depayload, "prepending codec_data");
583
584
      outbuf = gst_buffer_join (rtph264depay->codec_data, outbuf);
      rtph264depay->codec_data = NULL;
585
      out_keyframe = TRUE;
586
    }
Wim Taymans's avatar
Wim Taymans committed
587
    outbuf = gst_buffer_make_writable (outbuf);
588

589
    GST_BUFFER_TIMESTAMP (outbuf) = out_timestamp;
590

591
    if (out_keyframe)
592
      GST_BUFFER_FLAG_UNSET (outbuf, GST_BUFFER_FLAG_DELTA_UNIT);
593
594
    else
      GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_FLAG_DELTA_UNIT);
595
596
  }

597
  return outbuf;
598

599
600
601
602
  /* ERRORS */
short_nal:
  {
    GST_WARNING_OBJECT (depayload, "dropping short NAL");
Wim Taymans's avatar
Wim Taymans committed
603
    gst_buffer_unmap (nal, &map);
604
    gst_buffer_unref (nal);
605
    return NULL;
606
  }
607
608
}

609
610
611
static GstBuffer *
gst_rtp_h264_push_fragmentation_unit (GstRtpH264Depay * rtph264depay,
    gboolean send)
612
613
{
  guint outsize;
Wim Taymans's avatar
Wim Taymans committed
614
  GstMapInfo map;
615
616
617
618
619
  GstBuffer *outbuf;

  outsize = gst_adapter_available (rtph264depay->adapter);
  outbuf = gst_adapter_take_buffer (rtph264depay->adapter, outsize);

Wim Taymans's avatar
Wim Taymans committed
620
  gst_buffer_map (outbuf, &map, GST_MAP_WRITE);
621
622
623
  GST_DEBUG_OBJECT (rtph264depay, "output %d bytes", outsize);

  if (rtph264depay->byte_stream) {
Wim Taymans's avatar
Wim Taymans committed
624
    memcpy (map.data, sync_bytes, sizeof (sync_bytes));
625
626
  } else {
    outsize -= 4;
Wim Taymans's avatar
Wim Taymans committed
627
628
629
630
    map.data[0] = (outsize >> 24);
    map.data[1] = (outsize >> 16);
    map.data[2] = (outsize >> 8);
    map.data[3] = (outsize);
631
  }
Wim Taymans's avatar
Wim Taymans committed
632
  gst_buffer_unmap (outbuf, &map);
633
634

  rtph264depay->current_fu_type = 0;
635
636
637
638
639

  if (send) {
    outbuf = gst_rtp_h264_depay_handle_nal (rtph264depay, outbuf,
        rtph264depay->fu_timestamp, rtph264depay->fu_marker);
    if (outbuf)
Wim Taymans's avatar
Wim Taymans committed
640
      gst_rtp_base_depayload_push (GST_RTP_BASE_DEPAYLOAD (rtph264depay),
641
642
643
644
645
646
          outbuf);
    return NULL;
  } else {
    return gst_rtp_h264_depay_handle_nal (rtph264depay, outbuf,
        rtph264depay->fu_timestamp, rtph264depay->fu_marker);
  }
647
648
}

649
static GstBuffer *
Wim Taymans's avatar
Wim Taymans committed
650
gst_rtp_h264_depay_process (GstRTPBaseDepayload * depayload, GstBuffer * buf)
651
652
{
  GstRtpH264Depay *rtph264depay;
653
  GstBuffer *outbuf = NULL;
654
  guint8 nal_unit_type;
Wim Taymans's avatar
Wim Taymans committed
655
  GstRTPBuffer rtp = { NULL };
656
657
658

  rtph264depay = GST_RTP_H264_DEPAY (depayload);

659
660
661
662
  /* flush remaining data on discont */
  if (GST_BUFFER_IS_DISCONT (buf)) {
    gst_adapter_clear (rtph264depay->adapter);
    rtph264depay->wait_start = TRUE;
663
    rtph264depay->current_fu_type = 0;
664
665
  }

666
667
668
669
  {
    gint payload_len;
    guint8 *payload;
    guint header_len;
670
    guint8 nal_ref_idc;
Wim Taymans's avatar
Wim Taymans committed
671
    GstMapInfo map;
672
    guint outsize, nalu_size;
673
    GstClockTime timestamp;
674
    gboolean marker;
675

676
    timestamp = GST_BUFFER_TIMESTAMP (buf);
677

Wim Taymans's avatar
Wim Taymans committed
678
679
680
681
    gst_rtp_buffer_map (buf, GST_MAP_READ, &rtp);

    payload_len = gst_rtp_buffer_get_payload_len (&rtp);
    payload = gst_rtp_buffer_get_payload (&rtp);
Wim Taymans's avatar
Wim Taymans committed
682
    marker = gst_rtp_buffer_get_marker (&rtp);
683

684
685
    GST_DEBUG_OBJECT (rtph264depay, "receiving %d bytes", payload_len);

686
687
688
    if (payload_len == 0)
      return NULL;

689
690
691
692
693
694
695
696
697
698
699
    /* +---------------+
     * |0|1|2|3|4|5|6|7|
     * +-+-+-+-+-+-+-+-+
     * |F|NRI|  Type   |
     * +---------------+
     *
     * F must be 0.
     */
    nal_ref_idc = (payload[0] & 0x60) >> 5;
    nal_unit_type = payload[0] & 0x1f;

700
701
702
    /* at least one byte header with type */
    header_len = 1;

703
704
705
    GST_DEBUG_OBJECT (rtph264depay, "NRI %d, Type %d", nal_ref_idc,
        nal_unit_type);

706
707
708
709
710
    /* If FU unit was being processed, but the current nal is of a different
     * type.  Assume that the remote payloader is buggy (didn't set the end bit
     * when the FU ended) and send out what we gathered thusfar */
    if (G_UNLIKELY (rtph264depay->current_fu_type != 0 &&
            nal_unit_type != rtph264depay->current_fu_type))
711
      gst_rtp_h264_push_fragmentation_unit (rtph264depay, TRUE);
712

713
714
715
716
717
718
719
720
    switch (nal_unit_type) {
      case 0:
      case 30:
      case 31:
        /* undefined */
        goto undefined_type;
      case 25:
        /* STAP-B    Single-time aggregation packet     5.7.1 */
721
722
723
724
725
726
727
728
729
        /* 2 byte extra header for DON */
        header_len += 2;
        /* fallthrough */
      case 24:
      {
        /* strip headers */
        payload += header_len;
        payload_len -= header_len;

730
731
        rtph264depay->wait_start = FALSE;

732

733
734
735
736
737
738
739
740
741
742
        /* STAP-A    Single-time aggregation packet     5.7.1 */
        while (payload_len > 2) {
          /*                      1          
           *  0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 
           * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
           * |         NALU Size             |
           * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
           */
          nalu_size = (payload[0] << 8) | payload[1];

743
744
745
          /* dont include nalu_size */
          if (nalu_size > (payload_len - 2))
            nalu_size = payload_len - 2;
746
747
748

          outsize = nalu_size + sizeof (sync_bytes);
          outbuf = gst_buffer_new_and_alloc (outsize);
Wim Taymans's avatar
Wim Taymans committed
749

Wim Taymans's avatar
Wim Taymans committed
750
          gst_buffer_map (outbuf, &map, GST_MAP_WRITE);
751
          if (rtph264depay->byte_stream) {
Wim Taymans's avatar
Wim Taymans committed
752
            memcpy (map.data, sync_bytes, sizeof (sync_bytes));
753
          } else {
Wim Taymans's avatar
Wim Taymans committed
754
755
756
            map.data[0] = map.data[1] = 0;
            map.data[2] = payload[0];
            map.data[3] = payload[1];
757
758
759
760
761
762
          }

          /* strip NALU size */
          payload += 2;
          payload_len -= 2;

Wim Taymans's avatar
Wim Taymans committed
763
764
          memcpy (map.data + sizeof (sync_bytes), payload, nalu_size);
          gst_buffer_unmap (outbuf, &map);
765

766
          gst_adapter_push (rtph264depay->adapter, outbuf);
767
768
769
770
771

          payload += nalu_size;
          payload_len -= nalu_size;
        }

772
773
        outsize = gst_adapter_available (rtph264depay->adapter);
        outbuf = gst_adapter_take_buffer (rtph264depay->adapter, outsize);
774

775
776
        outbuf = gst_rtp_h264_depay_handle_nal (rtph264depay, outbuf, timestamp,
            marker);
777
        break;
778
      }
779
780
      case 26:
        /* MTAP16    Multi-time aggregation packet      5.7.2 */
781
        header_len = 5;
782
        /* fallthrough, not implemented */
783
784
      case 27:
        /* MTAP24    Multi-time aggregation packet      5.7.2 */
785
        header_len = 6;
786
        goto not_implemented;
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
        break;
      case 28:
      case 29:
      {
        /* FU-A      Fragmentation unit                 5.8 */
        /* FU-B      Fragmentation unit                 5.8 */
        gboolean S, E;

        /* +---------------+
         * |0|1|2|3|4|5|6|7|
         * +-+-+-+-+-+-+-+-+
         * |S|E|R|  Type   |
         * +---------------+
         *
         * R is reserved and always 0
         */
        S = (payload[1] & 0x80) == 0x80;
        E = (payload[1] & 0x40) == 0x40;

        GST_DEBUG_OBJECT (rtph264depay, "S %d, E %d", S, E);

808
809
810
        if (rtph264depay->wait_start && !S)
          goto waiting_start;

811
812
        if (S) {
          /* NAL unit starts here */
813
          guint8 nal_header;
814

815
816
817
818
          /* If a new FU unit started, while still processing an older one.
           * Assume that the remote payloader is buggy (doesn't set the end
           * bit) and send out what we've gathered thusfar */
          if (G_UNLIKELY (rtph264depay->current_fu_type != 0))
819
            gst_rtp_h264_push_fragmentation_unit (rtph264depay, TRUE);
820
821
822
823

          rtph264depay->current_fu_type = nal_unit_type;
          rtph264depay->fu_timestamp = timestamp;

824
825
          rtph264depay->wait_start = FALSE;

826
827
828
829
830
831
832
          /* reconstruct NAL header */
          nal_header = (payload[0] & 0xe0) | (payload[1] & 0x1f);

          /* strip type header, keep FU header, we'll reuse it to reconstruct
           * the NAL header. */
          payload += 1;
          payload_len -= 1;
833

834
835
836
          nalu_size = payload_len;
          outsize = nalu_size + sizeof (sync_bytes);
          outbuf = gst_buffer_new_and_alloc (outsize);
Wim Taymans's avatar
Wim Taymans committed
837

Wim Taymans's avatar
Wim Taymans committed
838
839
840
841
          gst_buffer_map (outbuf, &map, GST_MAP_WRITE);
          memcpy (map.data + sizeof (sync_bytes), payload, nalu_size);
          map.data[sizeof (sync_bytes)] = nal_header;
          gst_buffer_unmap (outbuf, &map);
842
843

          GST_DEBUG_OBJECT (rtph264depay, "queueing %d bytes", outsize);
844
845
846
847

          /* and assemble in the adapter */
          gst_adapter_push (rtph264depay->adapter, outbuf);
        } else {
848
849
850
          /* strip off FU indicator and FU header bytes */
          payload += 2;
          payload_len -= 2;
851

852
          outsize = payload_len;
853
          outbuf = gst_buffer_new_and_alloc (outsize);
Wim Taymans's avatar
Wim Taymans committed
854
          gst_buffer_fill (outbuf, 0, payload, outsize);
855
856

          GST_DEBUG_OBJECT (rtph264depay, "queueing %d bytes", outsize);
857
858
859
860
861

          /* and assemble in the adapter */
          gst_adapter_push (rtph264depay->adapter, outbuf);
        }

862
        outbuf = NULL;
863
        rtph264depay->fu_marker = marker;
Wim Taymans's avatar
Wim Taymans committed
864

865
866
        /* if NAL unit ends, flush the adapter */
        if (E)
867
          outbuf = gst_rtp_h264_push_fragmentation_unit (rtph264depay, FALSE);
868
869
870
871
        break;
      }
      default:
      {
872
873
        rtph264depay->wait_start = FALSE;

874
875
        /* 1-23   NAL unit  Single NAL unit packet per H.264   5.6 */
        /* the entire payload is the output buffer */
876
877
        nalu_size = payload_len;
        outsize = nalu_size + sizeof (sync_bytes);
878
        outbuf = gst_buffer_new_and_alloc (outsize);
Wim Taymans's avatar
Wim Taymans committed
879

Wim Taymans's avatar
Wim Taymans committed
880
        gst_buffer_map (outbuf, &map, GST_MAP_WRITE);
881
        if (rtph264depay->byte_stream) {
Wim Taymans's avatar
Wim Taymans committed
882
          memcpy (map.data, sync_bytes, sizeof (sync_bytes));
883
        } else {
Wim Taymans's avatar
Wim Taymans committed
884
885
886
          map.data[0] = map.data[1] = 0;
          map.data[2] = nalu_size >> 8;
          map.data[3] = nalu_size & 0xff;
887
        }
Wim Taymans's avatar
Wim Taymans committed
888
889
        memcpy (map.data + sizeof (sync_bytes), payload, nalu_size);
        gst_buffer_unmap (outbuf, &map);
890

891
892
        outbuf = gst_rtp_h264_depay_handle_nal (rtph264depay, outbuf, timestamp,
            marker);
893
        break;
894
895
      }
    }
Wim Taymans's avatar
Wim Taymans committed
896
    gst_rtp_buffer_unmap (&rtp);
897
898
  }

899
  return outbuf;
900
901
902
903
904

  /* ERRORS */
undefined_type:
  {
    GST_ELEMENT_WARNING (rtph264depay, STREAM, DECODE,
905
        (NULL), ("Undefined packet type"));
Wim Taymans's avatar
Wim Taymans committed
906
    gst_rtp_buffer_unmap (&rtp);
907
908
    return NULL;
  }
909
910
911
waiting_start:
  {
    GST_DEBUG_OBJECT (rtph264depay, "waiting for start");
Wim Taymans's avatar
Wim Taymans committed
912
    gst_rtp_buffer_unmap (&rtp);
913
914
    return NULL;
  }
915
916
917
not_implemented:
  {
    GST_ELEMENT_ERROR (rtph264depay, STREAM, FORMAT,
918
        (NULL), ("NAL unit type %d not supported yet", nal_unit_type));
Wim Taymans's avatar
Wim Taymans committed
919
    gst_rtp_buffer_unmap (&rtp);
920
921
    return NULL;
  }
922
923
}

924
static gboolean
Wim Taymans's avatar
Wim Taymans committed
925
gst_rtp_h264_depay_handle_event (GstRTPBaseDepayload * depay, GstEvent * event)
926
927
928
929
930
931
932
933
934
935
936
937
938
939
{
  GstRtpH264Depay *rtph264depay;

  rtph264depay = GST_RTP_H264_DEPAY (depay);

  switch (GST_EVENT_TYPE (event)) {
    case GST_EVENT_FLUSH_STOP:
      gst_rtp_h264_depay_reset (rtph264depay);
      break;
    default:
      break;
  }

  return
Wim Taymans's avatar
Wim Taymans committed
940
      GST_RTP_BASE_DEPAYLOAD_CLASS (parent_class)->handle_event (depay, event);
941
942
}

943
944
945
946
947
948
949
950
951
952
953
954
955
static GstStateChangeReturn
gst_rtp_h264_depay_change_state (GstElement * element,
    GstStateChange transition)
{
  GstRtpH264Depay *rtph264depay;
  GstStateChangeReturn ret;

  rtph264depay = GST_RTP_H264_DEPAY (element);

  switch (transition) {
    case GST_STATE_CHANGE_NULL_TO_READY:
      break;
    case GST_STATE_CHANGE_READY_TO_PAUSED:
956
      gst_rtp_h264_depay_reset (rtph264depay);
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
      break;
    default:
      break;
  }

  ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);

  switch (transition) {
    case GST_STATE_CHANGE_READY_TO_NULL:
      break;
    default:
      break;
  }
  return ret;
}

gboolean
gst_rtp_h264_depay_plugin_init (GstPlugin * plugin)
{
  return gst_element_register (plugin, "rtph264depay",
977
      GST_RANK_SECONDARY, GST_TYPE_RTP_H264_DEPAY);
978
}