/*
 * GStreamer AVTP Plugin
 * Copyright (C) 2019 Intel Corporation
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301 USA
 */

#include <gst/check/gstcheck.h>
#include <gst/check/gstharness.h>

#include <avtp.h>
#include <avtp_cvf.h>

#define AVTP_CVF_H264_HEADER_SIZE (sizeof(struct avtp_stream_pdu) + sizeof(guint32))
#define STREAM_ID 0xAABBCCDDEEFF0001

/* Simple codec data, with only the NAL size len, no SPS/PPS. */
static GstCaps *
generate_caps (guint8 nal_size_len)
{
  GstBuffer *codec_data;
  GstMapInfo map;
  GstCaps *caps;

  /* 7 is the minimal codec_data size, when no SPS/PPS is sent */
  codec_data = gst_buffer_new_allocate (NULL, 7, NULL);
  gst_buffer_map (codec_data, &map, GST_MAP_READWRITE);

  memset (map.data, 0, map.size);
  map.data[0] = 1;              /* version */
  map.data[4] = (nal_size_len - 1) | 0xfc;      /* Other 6 bits are 1 */
  map.data[5] = 0xe0;           /* first 3 bits are 1 */

  gst_buffer_unmap (codec_data, &map);

  caps = gst_caps_new_simple ("video/x-h264",
      "stream-format", G_TYPE_STRING, "avc",
      "alignment", G_TYPE_STRING, "au",
      "codec_data", GST_TYPE_BUFFER, codec_data, NULL);
  gst_buffer_unref (codec_data);

  return caps;
}

static void
fill (guint8 * buf, gsize size)
{
  guint8 i = 0;

  while (size--)
    *buf++ = i++;
}

static gboolean
check_nal_filling (GstBuffer * buffer, guint8 first)
{
  GstMapInfo map;
  gint i;
  gsize offset = AVTP_CVF_H264_HEADER_SIZE + 1;
  gboolean result = TRUE;

  gst_buffer_map (buffer, &map, GST_MAP_READ);
  if ((map.data[AVTP_CVF_H264_HEADER_SIZE] & 0x1f) == 28)
    offset++;                   /* Fragmented NALs have 2 bytes header */

  for (i = offset; i < map.size; i++) {
    if (map.data[i] != first++) {
      result = FALSE;
      break;
    }
  }

  return result;
}

static void
add_nal (GstBuffer * buffer, gsize size, guint type, gsize offset)
{
  GstMapInfo map;

  gst_buffer_map (buffer, &map, GST_MAP_READWRITE);

  map.data[offset] = map.data[offset + 1] = 0;
  map.data[offset + 2] = size >> 8;
  map.data[offset + 3] = size & 0xff;
  map.data[offset + 4] = type & 0x1f;
  fill (&map.data[offset + 5], size - 1);

  gst_buffer_unmap (buffer, &map);
}

/* This function assumes that NAL size len is 2 */
static void
add_nal_2 (GstBuffer * buffer, gsize size, guint type, gsize offset)
{
  GstMapInfo map;

  gst_buffer_map (buffer, &map, GST_MAP_READWRITE);

  map.data[offset] = size >> 8;
  map.data[offset + 1] = size & 0xff;
  map.data[offset + 2] = type & 0x1f;
  fill (&map.data[offset + 3], size - 1);

  gst_buffer_unmap (buffer, &map);
}

static gboolean
compare_h264_avtpdu (struct avtp_stream_pdu *pdu, GstBuffer * buffer)
{
  GstMapInfo map;
  gboolean result;

  gst_buffer_map (buffer, &map, GST_MAP_READ);
  /* buffer must have at least the header size */
  if (map.size < AVTP_CVF_H264_HEADER_SIZE)
    return FALSE;

  result = memcmp (map.data, pdu, AVTP_CVF_H264_HEADER_SIZE) == 0;

  gst_buffer_unmap (buffer, &map);

  return result;
}

GST_START_TEST (test_payloader_zero_sized_nal)
{
  GstHarness *h;
  GstBuffer *in;
  GstMapInfo map;

  /* Create the harness for the avtpcvfpay */
  h = gst_harness_new_parse
      ("avtpcvfpay streamid=0xAABBCCDDEEFF0001 mtt=1000000 tu=1000000 processing-deadline=0");
  gst_harness_set_src_caps (h, generate_caps (4));

  /* We have the buffer with the nal size (4 bytes) and the nal (4 bytes), but
   * nal size will be zero */
  in = gst_harness_create_buffer (h, 8);
  GST_BUFFER_DTS (in) = 1000000;
  GST_BUFFER_PTS (in) = 2000000;

  gst_buffer_map (in, &map, GST_MAP_READWRITE);
  map.data[0] = map.data[1] = map.data[2] = map.data[3] = 0;    /* Set NAL size to 0 */
  map.data[4] = 1;              /* Some dummy vcl NAL type */
  gst_buffer_unmap (in, &map);

  gst_harness_push (h, in);

  /* No buffer shuld come out */
  fail_unless_equals_int (gst_harness_buffers_received (h), 0);

  gst_harness_teardown (h);
}

GST_END_TEST;

GST_START_TEST (test_payloader_no_codec_data)
{
  GstHarness *h;
  GstCaps *caps;
  GstBuffer *in;

  /* Caps without codec_data */
  caps = gst_caps_new_simple ("video/x-h264",
      "stream-format", G_TYPE_STRING, "avc",
      "alignment", G_TYPE_STRING, "au", NULL);

  /* Create the harness for the avtpcvfpay */
  h = gst_harness_new_parse
      ("avtpcvfpay streamid=0xAABBCCDDEEFF0001 mtt=1000000 tu=1000000 processing-deadline=0");
  gst_harness_set_src_caps (h, caps);

  /* No buffer should come out when we send input */
  in = gst_harness_create_buffer (h, 8);
  add_nal (in, 4, 1, 0);
  GST_BUFFER_DTS (in) = 1000000;
  GST_BUFFER_PTS (in) = 2000000;

  gst_harness_push (h, in);
  fail_unless_equals_int (gst_harness_buffers_received (h), 0);

  gst_harness_teardown (h);
}

GST_END_TEST;

GST_START_TEST (test_payloader_invalid_caps)
{
  GstBuffer *codec_data;
  GstElement *element;
  GstPad *sinkpad;
  GstMapInfo map;
  GstCaps *caps;
  GstHarness *h;

  /* 7 is the minimal codec_data size, when no SPS/PPS is sent */
  codec_data = gst_buffer_new_allocate (NULL, 7, NULL);
  gst_buffer_map (codec_data, &map, GST_MAP_READWRITE);

  memset (map.data, 0, map.size);
  map.data[0] = 0;              /* version */
  map.data[4] = 0x03 | 0xfc;    /* Other 6 bits are 1 */
  map.data[5] = 0xe0;           /* first 3 bits are 1 */

  gst_buffer_unmap (codec_data, &map);

  caps = gst_caps_new_simple ("video/x-h264",
      "stream-format", G_TYPE_STRING, "avc",
      "alignment", G_TYPE_STRING, "au",
      "codec_data", GST_TYPE_BUFFER, codec_data, NULL);
  gst_buffer_unref (codec_data);

  /* Create the harness for the avtpcvfpay */
  h = gst_harness_new_parse
      ("avtpcvfpay streamid=0xAABBCCDDEEFF0001 mtt=1000000 tu=1000000");
  element = gst_harness_find_element (h, "avtpcvfpay");
  sinkpad = gst_element_get_static_pad (element, "sink");

  /* 'codec_data' caps has invalid version */
  gst_harness_push_event (h, gst_event_new_caps (caps));
  fail_unless (gst_pad_get_current_caps (sinkpad) == NULL);
  gst_caps_unref (caps);

  /* Send a 'codec_data' too small */
  codec_data = gst_buffer_new_allocate (NULL, 6, NULL);
  caps = gst_caps_new_simple ("video/x-h264",
      "stream-format", G_TYPE_STRING, "avc",
      "alignment", G_TYPE_STRING, "au",
      "codec_data", GST_TYPE_BUFFER, codec_data, NULL);
  gst_buffer_unref (codec_data);

  gst_harness_push_event (h, gst_event_new_caps (caps));
  fail_unless (gst_pad_get_current_caps (sinkpad) == NULL);
  gst_caps_unref (caps);

  gst_harness_teardown (h);
}

GST_END_TEST;

GST_START_TEST (test_payloader_incomplete_nal)
{
  GstHarness *h;
  GstBuffer *in, *out;
  GstMapInfo map;
  struct avtp_stream_pdu *pdu = alloca (AVTP_CVF_H264_HEADER_SIZE);
  const gint DATA_LEN = sizeof (guint32) + 3;

  /* Create the 'expected' header */
  avtp_cvf_pdu_init (pdu, AVTP_CVF_FORMAT_SUBTYPE_H264);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_ID, STREAM_ID);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_M, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 3000000);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_PTV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_TIMESTAMP, 4000000);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN);

  /* Create the harness for the avtpcvfpay */
  h = gst_harness_new_parse
      ("avtpcvfpay streamid=0xAABBCCDDEEFF0001 mtt=1000000 tu=1000000 processing-deadline=0");
  gst_harness_set_src_caps (h, generate_caps (4));

  /* Buffer must have the nal len (4 bytes) and the nal (3 bytes) */
  in = gst_harness_create_buffer (h, 7);
  GST_BUFFER_DTS (in) = 1000000;
  GST_BUFFER_PTS (in) = 2000000;

  gst_buffer_map (in, &map, GST_MAP_READWRITE);
  map.data[0] = map.data[1] = map.data[2] = 0;
  map.data[3] = 8;              /* Lie that NAL size is 8, when buffer is only 7 (so NAL is 3) */
  map.data[4] = 1;              /* Some dummy vcl NAL type */
  map.data[5] = 0x0;
  map.data[6] = 0x1;

  out = gst_harness_push_and_pull (h, in);

  /* avtpcvfpay will happily payload the three byte nal. Now, we check it */
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 0) == TRUE);

  gst_buffer_unref (out);
  gst_harness_teardown (h);
}

GST_END_TEST;

GST_START_TEST (test_payloader_properties)
{
  GstHarness *h;
  GstElement *element;
  guint mtu, mtt, tu;
  guint64 streamid, processing_deadline;

  /* Create the harness for the avtpcvfpay */
  h = gst_harness_new_parse
      ("avtpcvfpay streamid=0xAABBCCDDEEFF0001 mtt=1000000 tu=2000000 mtu=100 processing-deadline=5000");

  /* Check if all properties were properly set up */
  element = gst_harness_find_element (h, "avtpcvfpay");
  g_object_get (G_OBJECT (element), "mtt", &mtt, NULL);
  fail_unless_equals_uint64 (mtt, 1000000);

  g_object_get (G_OBJECT (element), "mtu", &mtu, NULL);
  fail_unless_equals_uint64 (mtu, 100);

  g_object_get (G_OBJECT (element), "tu", &tu, NULL);
  fail_unless_equals_uint64 (tu, 2000000);

  g_object_get (G_OBJECT (element), "streamid", &streamid, NULL);
  fail_unless_equals_uint64 (streamid, 0xAABBCCDDEEFF0001);

  g_object_get (G_OBJECT (element), "processing-deadline", &processing_deadline,
      NULL);
  fail_unless_equals_uint64 (processing_deadline, 5000);

  gst_harness_teardown (h);
}

GST_END_TEST;
GST_START_TEST (test_payloader_single_and_fragment_edge)
{
  GstHarness *h;
  GstBuffer *in, *out;
  struct avtp_stream_pdu *pdu = alloca (AVTP_CVF_H264_HEADER_SIZE);
  const gint DATA_LEN_1 = sizeof (guint32) + 100;
  const gint DATA_LEN_2 = sizeof (guint32) + 100;
  const gint DATA_LEN_3 = sizeof (guint32) + 4;

  /* Create the 'expected' header */
  avtp_cvf_pdu_init (pdu, AVTP_CVF_FORMAT_SUBTYPE_H264);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_ID, STREAM_ID);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 3000000);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_PTV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_TIMESTAMP, 4000000);

  /* Create the harness for the avtpcvfpay. Setting mtu=128 ensures that
   * NAL units will be broken roughly at 100 bytes. More details below. */
  h = gst_harness_new_parse
      ("avtpcvfpay streamid=0xAABBCCDDEEFF0001 mtt=1000000 tu=1000000 mtu=128 processing-deadline=0");
  gst_harness_set_src_caps (h, generate_caps (4));

  /* Create a buffer to contain the multiple NAL units. This buffer
   * will hold two NAL units, with 100 and 101 bytes, each preceded
   * by a 4 bytes header */
  in = gst_harness_create_buffer (h, 100 + 101 + 2 * 4);
  add_nal (in, 100, 7, 0);
  add_nal (in, 101, 1, 104);
  GST_BUFFER_DTS (in) = 1000000;
  GST_BUFFER_PTS (in) = 2000000;

  /* We now push the buffer, and check if we get three from the avtpcvfpay */
  gst_harness_push (h, in);
  fail_unless (gst_harness_buffers_received (h) == 3);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_1);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  gst_buffer_unref (out);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_2);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_SEQ_NUM, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 0);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 0);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  gst_buffer_unref (out);

  /* DATA_LEN_3 is 4 because only 98 bytes from the original NAL unit are
   * sent on the first buffer (due 2 bytes header), and the two remaining
   * bytes are preceded by the 2 bytes header. Note that the first byte of
   * the NAL is stripped before the fragmentation (see comment on
   * test_payloader_single_and_fragment below for more details). */
  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_3);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_SEQ_NUM, 2);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_M, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 3000000);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  gst_buffer_unref (out);
  gst_harness_teardown (h);
}

GST_END_TEST;

GST_START_TEST (test_payloader_single_and_fragment)
{
  GstHarness *h;
  GstBuffer *in, *out;
  struct avtp_stream_pdu *pdu = alloca (AVTP_CVF_H264_HEADER_SIZE);
  const gint DATA_LEN_1 = sizeof (guint32) + 4;
  const gint DATA_LEN_2 = sizeof (guint32) + 100;
  const gint DATA_LEN_3 = sizeof (guint32) + 100;
  const gint DATA_LEN_4 = sizeof (guint32) + 55;

  /* Create the 'expected' header */
  avtp_cvf_pdu_init (pdu, AVTP_CVF_FORMAT_SUBTYPE_H264);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_ID, STREAM_ID);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 4000000);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_PTV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_TIMESTAMP, 5000000);

  /* Create the harness for the avtpcvfpay. Setting mtu=128 ensures that
   * NAL units will be broken roughly at 100 bytes. More details below. */
  h = gst_harness_new_parse
      ("avtpcvfpay streamid=0xAABBCCDDEEFF0001 mtt=1000000 tu=1000000 processing-deadline=1000000 mtu=128");
  gst_harness_set_src_caps (h, generate_caps (4));

  /* Create a buffer to contain the multiple NAL units. This buffer
   * will hold two NAL units, with 4 and 250 bytes, each preceded
   * by a 4 bytes header */
  in = gst_harness_create_buffer (h, 4 + 250 + 2 * 4);
  add_nal (in, 4, 7, 0);
  add_nal (in, 250, 1, 8);
  GST_BUFFER_DTS (in) = 1000000;
  GST_BUFFER_PTS (in) = 2000000;

  /* We now push the buffer, and check if we get four from the avtpcvfpay */
  gst_harness_push (h, in);
  fail_unless (gst_harness_buffers_received (h) == 4);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_1);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 0) == TRUE);
  gst_buffer_unref (out);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_2);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_SEQ_NUM, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 0);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 0);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 0) == TRUE);
  gst_buffer_unref (out);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_3);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_SEQ_NUM, 2);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 0);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 0);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 98) == TRUE);
  gst_buffer_unref (out);

  /* For those wondering why DATA_LEN_4 is 55 and not 50 - or why
   * comment above states that NAL units are broken "rougly" at 100 bytes:
   * With mtu=128, there are only 100 bytes left for NAL units, so anything
   * bigger will be broken. But AVTP NAL units fragments have a header with
   * two bytes, so NAL units will use only 98 bytes. This leaves the last
   * fragment with 54 bytes. However, instead of being 56 (54 bytes plus
   * 2 bytes header), it is 55 (53 bytes plus 2 bytes header) due to the
   * fact that the first byte of the NAL unit (the NAL unit header) is
   * in fact stripped from the NAL unit before the fragmentation. */
  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_4);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_SEQ_NUM, 3);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_M, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 4000000);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 196) == TRUE);
  gst_buffer_unref (out);
  gst_harness_teardown (h);
}

GST_END_TEST;

GST_START_TEST (test_payloader_multiple_single_2)
{
  GstHarness *h;
  GstBuffer *in, *out;
  struct avtp_stream_pdu *pdu = alloca (AVTP_CVF_H264_HEADER_SIZE);
  const gint DATA_LEN_1 = sizeof (guint32) + 32;
  const gint DATA_LEN_2 = sizeof (guint32) + 16;
  const gint DATA_LEN_3 = sizeof (guint32) + 8;

  /* Create the 'expected' header */
  avtp_cvf_pdu_init (pdu, AVTP_CVF_FORMAT_SUBTYPE_H264);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_ID, STREAM_ID);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 3000000);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_PTV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_TIMESTAMP, 4000000);

  /* Create the harness for the avtpcvfpay */
  h = gst_harness_new_parse
      ("avtpcvfpay streamid=0xAABBCCDDEEFF0001 mtt=1000000 tu=1000000 processing-deadline=0");
  gst_harness_set_src_caps (h, generate_caps (2));

  /* Create a buffer to contain the multiple NAL units. This buffer
   * will hold three NAL units, with 32, 16 and 8 bytes, each preceded
   * by a 2 bytes header */
  in = gst_harness_create_buffer (h, 32 + 16 + 8 + 4 * 2);
  add_nal_2 (in, 32, 7, 0);
  add_nal_2 (in, 16, 7, 34);
  add_nal_2 (in, 8, 1, 52);
  GST_BUFFER_DTS (in) = 1000000;
  GST_BUFFER_PTS (in) = 2000000;

  /* We now push the buffer, and check if we get three from the avtpcvfpay */
  gst_harness_push (h, in);
  fail_unless (gst_harness_buffers_received (h) == 3);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_1);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 0) == TRUE);
  gst_buffer_unref (out);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_2);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_SEQ_NUM, 1);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 0) == TRUE);
  gst_buffer_unref (out);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_3);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_SEQ_NUM, 2);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_M, 1);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 0) == TRUE);
  gst_buffer_unref (out);
  gst_harness_teardown (h);
}

GST_END_TEST;

GST_START_TEST (test_payloader_multiple_single)
{
  GstHarness *h;
  GstBuffer *in, *out;
  struct avtp_stream_pdu *pdu = alloca (AVTP_CVF_H264_HEADER_SIZE);
  const gint DATA_LEN_1 = sizeof (guint32) + 32;
  const gint DATA_LEN_2 = sizeof (guint32) + 16;
  const gint DATA_LEN_3 = sizeof (guint32) + 8;

  /* Create the 'expected' header */
  avtp_cvf_pdu_init (pdu, AVTP_CVF_FORMAT_SUBTYPE_H264);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_ID, STREAM_ID);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 3000000);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_PTV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_TIMESTAMP, 4000000);

  /* Create the harness for the avtpcvfpay */
  h = gst_harness_new_parse
      ("avtpcvfpay streamid=0xAABBCCDDEEFF0001 mtt=1000000 tu=1000000 processing-deadline=0");
  gst_harness_set_src_caps (h, generate_caps (4));

  /* Create a buffer to contain the multiple NAL units. This buffer
   * will hold three NAL units, with 32, 16 and 8 bytes, each preceded
   * by a 4 bytes header */
  in = gst_harness_create_buffer (h, 32 + 16 + 8 + 4 * 4);
  add_nal (in, 32, 7, 0);
  add_nal (in, 16, 7, 36);
  add_nal (in, 8, 1, 56);
  GST_BUFFER_DTS (in) = 1000000;
  GST_BUFFER_PTS (in) = 2000000;

  /* We now push the buffer, and check if we get three from the avtpcvfpay */
  gst_harness_push (h, in);
  fail_unless (gst_harness_buffers_received (h) == 3);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_1);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 0) == TRUE);
  gst_buffer_unref (out);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_2);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_SEQ_NUM, 1);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 0) == TRUE);
  gst_buffer_unref (out);

  out = gst_harness_pull (h);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN_3);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_SEQ_NUM, 2);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_M, 1);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 0) == TRUE);
  gst_buffer_unref (out);
  gst_harness_teardown (h);
}

GST_END_TEST;

GST_START_TEST (test_payloader_single)
{
  GstHarness *h;
  GstBuffer *in, *out;
  struct avtp_stream_pdu *pdu = alloca (AVTP_CVF_H264_HEADER_SIZE);
  const gint DATA_LEN = sizeof (guint32) + 4;

  /* Create the 'expected' header */
  avtp_cvf_pdu_init (pdu, AVTP_CVF_FORMAT_SUBTYPE_H264);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_ID, STREAM_ID);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_M, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_TIMESTAMP, 3000000);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_PTV, 1);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_H264_TIMESTAMP, 4000000);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_STREAM_DATA_LEN, DATA_LEN);

  /* Create the harness for the avtpcvfpay */
  h = gst_harness_new_parse
      ("avtpcvfpay streamid=0xAABBCCDDEEFF0001 mtt=1000000 tu=1000000 processing-deadline=0");
  gst_harness_set_src_caps (h, generate_caps (4));

  /* Buffer must have the nal len (4 bytes) and the nal (4 bytes) */
  in = gst_harness_create_buffer (h, 8);
  add_nal (in, 4, 1, 0);
  GST_BUFFER_DTS (in) = 1000000;
  GST_BUFFER_PTS (in) = 2000000;

  out = gst_harness_push_and_pull (h, in);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  gst_buffer_unref (out);

  /* Now test if, when nal_type is not vcl (not between 1 and 5), M is not set.
   * Also, as we're using the same element, seqnum should increase by one */
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_M, 0);
  avtp_cvf_pdu_set (pdu, AVTP_CVF_FIELD_SEQ_NUM, 1);

  in = gst_harness_create_buffer (h, 8);
  add_nal (in, 4, 6, 0);
  GST_BUFFER_DTS (in) = 1000000;
  GST_BUFFER_PTS (in) = 2000000;

  out = gst_harness_push_and_pull (h, in);
  fail_unless (compare_h264_avtpdu (pdu, out) == TRUE);
  fail_unless (check_nal_filling (out, 0) == TRUE);
  gst_buffer_unref (out);
  gst_harness_teardown (h);
}

GST_END_TEST;

static Suite *
avtpcvfpay_suite (void)
{
  Suite *s = suite_create ("avtpcvfpay");
  TCase *tc_chain = tcase_create ("general");

  suite_add_tcase (s, tc_chain);
  tcase_add_test (tc_chain, test_payloader_single);
  tcase_add_test (tc_chain, test_payloader_multiple_single);
  tcase_add_test (tc_chain, test_payloader_multiple_single_2);
  tcase_add_test (tc_chain, test_payloader_single_and_fragment);
  tcase_add_test (tc_chain, test_payloader_single_and_fragment_edge);
  tcase_add_test (tc_chain, test_payloader_incomplete_nal);
  tcase_add_test (tc_chain, test_payloader_invalid_caps);
  tcase_add_test (tc_chain, test_payloader_properties);
  tcase_add_test (tc_chain, test_payloader_no_codec_data);
  tcase_add_test (tc_chain, test_payloader_zero_sized_nal);

  return s;
}

GST_CHECK_MAIN (avtpcvfpay);