tests/check/libs/gdp.c

   1 /* GStreamer
   2  *
   3  * unit test for data protocol
   4  *
   5  * Copyright (C) <2004> Thomas Vander Stichele <thomas at apestaart dot org>
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Library General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2 of the License, or (at your option) any later version.
  11  *
  12  * This library 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  * Library General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Library General Public
  18  * License along with this library; if not, write to the
  19  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  20  * Boston, MA 02111-1307, USA.
  21  */
  22
  23 #include "config.h"
  24
  25 #include <gst/check/gstcheck.h>
  26
  27 #include <gst/dataprotocol/dataprotocol.h>
  28 #include "libs/gst/dataprotocol/dp-private.h"   /* private header */
  29
  30 /* test our method of reading and writing headers using TO/FROM_BE */
  31 GST_START_TEST (test_conversion)
  32 {
  33   guint8 array[9];
  34   guint8 write_array[9];
  35   guint16 read_two, expect_two;
  36   guint32 read_four, expect_four;
  37   guint64 read_eight, expect_eight;
  38   int i;
  39
  40   for (i = 0; i < 9; ++i) {
  41     array[i] = i * 0x10;
  42   }
  43
  44   /* read 8 16 bits */
  45   for (i = 0; i < 8; ++i) {
  46     read_two = GST_READ_UINT16_BE (array + i);
  47     expect_two = array[i] * (1 << 8) + array[i + 1];
  48     fail_unless (read_two == expect_two,
  49         "GST_READ_UINT16_BE %d: read %d != %d\n", i, read_two, expect_two);
  50   }
  51
  52   /* write 8 16 bits */
  53   for (i = 0; i < 8; ++i) {
  54     GST_WRITE_UINT16_BE (&write_array[i], read_two);
  55     fail_unless (memcmp (array + 7, write_array + i, 2) == 0,
  56         "GST_WRITE_UINT16_BE %d: memcmp failed", i);
  57   }
  58
  59   /* read 5 32 bits */
  60   for (i = 0; i < 5; ++i) {
  61     read_four = GST_READ_UINT32_BE (array + i);
  62     expect_four = array[i] * (1 << 24) + array[i + 1] * (1 << 16)
  63         + array[i + 2] * (1 << 8) + array[i + 3];
  64     fail_unless (read_four == expect_four,
  65         "GST_READ_UINT32_BE %d: read %d != %d\n", i, read_four, expect_four);
  66   }
  67
  68   /* read 2 64 bits */
  69   for (i = 0; i < 2; ++i) {
  70     read_eight = GST_READ_UINT64_BE (array + i);
  71     expect_eight = array[i] * (1LL << 56) + array[i + 1] * (1LL << 48)
  72         + array[i + 2] * (1LL << 40) + array[i + 3] * (1LL << 32)
  73         + array[i + 4] * (1 << 24) + array[i + 5] * (1 << 16)
  74         + array[i + 6] * (1 << 8) + array[i + 7];
  75     fail_unless (read_eight == expect_eight,
  76         "GST_READ_UINT64_BE %d: read %" G_GUINT64_FORMAT
  77         " != %" G_GUINT64_FORMAT "\n", i, read_eight, expect_eight);
  78   }
  79
  80   /* write 1 64 bit */
  81   GST_WRITE_UINT64_BE (&write_array[0], read_eight);
  82   fail_unless (memcmp (array + 1, write_array, 8) == 0,
  83       "GST_WRITE_UINT64_BE: memcmp failed");
  84 }
  85
  86 GST_END_TEST;
  87
  88 /* test creation of header from buffer and back again */
  89 GST_START_TEST (test_buffer)
  90 {
  91   GstBuffer *buffer;
  92   GstBuffer *newbuffer;
  93
  94   guint header_length;
  95   guint8 *header;
  96
  97   /* create buffer */
  98   g_message ("Creating a new 8-byte buffer with ts 0.5 sec, dur 1 sec\n");
  99   buffer = gst_buffer_new_and_alloc (8);
 100   GST_BUFFER_TIMESTAMP (buffer) = (GstClockTime) (GST_SECOND * 0.5);
 101   GST_BUFFER_DURATION (buffer) = (GstClockTime) GST_SECOND;
 102   GST_BUFFER_OFFSET (buffer) = (guint64) 10;
 103   GST_BUFFER_OFFSET_END (buffer) = (guint64) 19;
 104   GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_IN_CAPS);
 105   memmove (GST_BUFFER_DATA (buffer), "a buffer", 8);
 106
 107   /* create a buffer with CRC checking */
 108   fail_unless (gst_dp_header_from_buffer (buffer, GST_DP_HEADER_FLAG_CRC,
 109           &header_length, &header), "Could not create header from buffer.");
 110
 111   /* validate the header */
 112   fail_unless (gst_dp_validate_header (header_length, header),
 113       "Could not validate header");
 114   /* create a new, empty buffer with the right size */
 115   newbuffer = gst_dp_buffer_from_header (header_length, header);
 116   fail_unless (newbuffer != NULL, "Could not create a new buffer from header");
 117   fail_unless (GST_IS_BUFFER (newbuffer), "Created buffer is not a GstBuffer");
 118   /* read/copy the data */
 119   memmove (GST_BUFFER_DATA (newbuffer), GST_BUFFER_DATA (buffer),
 120       GST_BUFFER_SIZE (buffer));
 121   /* validate the buffer */
 122   fail_unless (gst_dp_validate_payload (header_length, header,
 123           GST_BUFFER_DATA (newbuffer)), "Could not validate payload");
 124
 125   g_message ("new buffer timestamp: %" GST_TIME_FORMAT "\n",
 126       GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (newbuffer)));
 127   g_message ("new buffer duration: %" GST_TIME_FORMAT "\n",
 128       GST_TIME_ARGS (GST_BUFFER_DURATION (newbuffer)));
 129   g_message ("new buffer offset: %" G_GUINT64_FORMAT "\n",
 130       GST_BUFFER_OFFSET (newbuffer));
 131   g_message ("new buffer offset_end: %" G_GUINT64_FORMAT "\n",
 132       GST_BUFFER_OFFSET_END (newbuffer));
 133   fail_unless (GST_BUFFER_TIMESTAMP (newbuffer) ==
 134       GST_BUFFER_TIMESTAMP (buffer), "Timestamps don't match !");
 135   fail_unless (GST_BUFFER_DURATION (newbuffer) == GST_BUFFER_DURATION (buffer),
 136       "Durations don't match !");
 137   fail_unless (GST_BUFFER_OFFSET (newbuffer) == GST_BUFFER_OFFSET (buffer),
 138       "Offsets don't match !");
 139   fail_unless (GST_BUFFER_OFFSET_END (newbuffer) ==
 140       GST_BUFFER_OFFSET_END (buffer), "Offset ends don't match !");
 141   fail_unless (GST_BUFFER_FLAG_IS_SET (newbuffer, GST_BUFFER_FLAG_IN_CAPS),
 142       "GST_BUFFER_IN_CAPS flag should have been copied !");
 143
 144   /* clean up */
 145   gst_buffer_unref (buffer);
 146   gst_buffer_unref (newbuffer);
 147   g_free (header);
 148 }
 149
 150 GST_END_TEST;
 151
 152 GST_START_TEST (test_caps)
 153 {
 154   gchar *string, *newstring;
 155   GstCaps *caps, *newcaps;
 156
 157   guint header_length;
 158   guint8 *header, *payload;
 159
 160   caps = gst_caps_from_string ("audio/x-raw-float, "
 161       "rate = (int) [ 11025, 48000 ], "
 162       "channels = (int) [ 1, 2 ], " "endianness = (int) BYTE_ORDER, "
 163       "width = (int) 32, " "buffer-frames = (int) 0");
 164   string = gst_caps_to_string (caps);
 165   g_message ("Created caps: %s\n", string);
 166   fail_unless (gst_dp_packet_from_caps (caps, 0, &header_length, &header,
 167           &payload), "Could not create packet from caps.");
 168
 169   /* validate the packet */
 170   fail_unless (gst_dp_validate_packet (header_length, header, payload),
 171       "Could not validate packet");
 172   newcaps = gst_dp_caps_from_packet (header_length, header, payload);
 173   fail_unless (newcaps != NULL, "Could not create caps from packet");
 174   fail_unless (GST_IS_CAPS (newcaps));
 175   newstring = gst_caps_to_string (newcaps);
 176   g_message ("Received caps: %s\n", newstring);
 177   fail_unless (strcmp (string, newstring) == 0,
 178       "Created caps do not match original caps");
 179
 180   /* cleanup */
 181   gst_caps_unref (caps);
 182   gst_caps_unref (newcaps);
 183   g_free (header);
 184   g_free (payload);
 185   g_free (string);
 186   g_free (newstring);
 187 }
 188
 189 GST_END_TEST;
 190
 191 GST_START_TEST (test_event)
 192 {
 193   GstEvent *send;
 194   GstEvent *receive;
 195   guint header_length;
 196   guint8 *header, *payload;
 197
 198   g_message ("Testing EOS event at 1s\n");
 199   send = gst_event_new_eos ();
 200   GST_EVENT_TIMESTAMP (send) = GST_SECOND;
 201   fail_unless (gst_dp_packet_from_event (send, GST_DP_HEADER_FLAG_CRC,
 202           &header_length, &header, &payload),
 203       "Could not create packet from eos event");
 204
 205   receive = gst_dp_event_from_packet (header_length, header, payload);
 206
 207   g_message ("EOS, timestamp %" GST_TIME_FORMAT "\n",
 208       GST_TIME_ARGS (GST_EVENT_TIMESTAMP (receive)));
 209   fail_unless (GST_EVENT_TYPE (receive) == GST_EVENT_EOS,
 210       "Received event is not EOS");
 211   fail_unless (GST_EVENT_TIMESTAMP (receive) == GST_SECOND,
 212       "EOS timestamp is not 1.0 sec");
 213
 214   /* clean up */
 215   g_free (header);
 216   g_free (payload);
 217   gst_event_unref (send);
 218   gst_event_unref (receive);
 219
 220   g_message ("Testing FLUSH event at 2s\n");
 221   send = gst_event_new_flush_start ();
 222   GST_EVENT_TIMESTAMP (send) = GST_SECOND * 2;
 223   fail_unless (gst_dp_packet_from_event (send, GST_DP_HEADER_FLAG_CRC,
 224           &header_length, &header, &payload),
 225       "Could not create packet from flush event");
 226
 227   receive = gst_dp_event_from_packet (header_length, header, payload);
 228
 229   g_message ("Flush, timestamp %" GST_TIME_FORMAT "\n",
 230       GST_TIME_ARGS (GST_EVENT_TIMESTAMP (receive)));
 231   fail_unless (GST_EVENT_TYPE (receive) == GST_EVENT_FLUSH_START,
 232       "Received event is not flush");
 233   fail_unless (GST_EVENT_TIMESTAMP (receive) == GST_SECOND * 2,
 234       "Flush timestamp is not 2.0 sec");
 235
 236   /* clean up */
 237   g_free (header);
 238   g_free (payload);
 239   gst_event_unref (send);
 240   gst_event_unref (receive);
 241
 242   g_message ("Testing SEEK event with 1 second at 3 seconds\n");
 243   send =
 244       gst_event_new_seek (1.0, GST_FORMAT_TIME, 0, GST_SEEK_TYPE_SET,
 245       GST_SECOND, GST_SEEK_TYPE_NONE, 0);
 246   GST_EVENT_TIMESTAMP (send) = GST_SECOND * 3;
 247   fail_unless (gst_dp_packet_from_event (send, GST_DP_HEADER_FLAG_CRC,
 248           &header_length, &header, &payload),
 249       "Could not create packet from seek event");
 250
 251   receive = gst_dp_event_from_packet (header_length, header, payload);
 252
 253   {
 254     gdouble rate;
 255     GstFormat format;
 256     GstSeekFlags flags;
 257     GstSeekType cur_type, stop_type;
 258     gint64 cur, stop;
 259
 260     gst_event_parse_seek (receive, &rate, &format, &flags,
 261         &cur_type, &cur, &stop_type, &stop);
 262
 263     g_message ("Seek, timestamp %" GST_TIME_FORMAT ", to %" GST_TIME_FORMAT
 264         "\n", GST_TIME_ARGS (GST_EVENT_TIMESTAMP (receive)),
 265         GST_TIME_ARGS (cur));
 266     fail_unless (GST_EVENT_TYPE (receive) == GST_EVENT_SEEK,
 267         "Returned event is not seek");
 268     fail_unless (GST_EVENT_TIMESTAMP (receive) == GST_SECOND * 3,
 269         "Seek timestamp is not 3.0 sec");
 270     fail_unless (format == GST_FORMAT_TIME, "Seek format is not time");
 271     fail_unless (cur == GST_SECOND, "Seek cur is not 1.0 sec");
 272   }
 273
 274   /* clean up */
 275   g_free (header);
 276   g_free (payload);
 277   gst_event_unref (send);
 278   gst_event_unref (receive);
 279 }
 280
 281 GST_END_TEST;
 282
 283 /* try to segfault the thing by passing NULLs, short headers, etc.. */
 284 GST_START_TEST (test_memory)
 285 {
 286   guint8 foo[5];
 287   GstBuffer *buffer;
 288   GstCaps *caps;
 289   GstEvent *event;
 290   guint length;
 291   guint8 *header;
 292   guint8 *payload;
 293
 294   /* check 0 sized input, data pointer can be NULL or anything. CRC is always 0,
 295    * though. */
 296   fail_if (gst_dp_crc (NULL, 0) != 0);
 297   fail_if (gst_dp_crc (foo, 0) != 0);
 298
 299   /* this is very invalid input and gives a warning. */
 300   ASSERT_CRITICAL (gst_dp_crc (NULL, 1));
 301   ASSERT_CRITICAL (gst_dp_header_payload_length (NULL));
 302   ASSERT_CRITICAL (gst_dp_header_payload_type (NULL));
 303
 304   /* wrong */
 305   ASSERT_CRITICAL (gst_dp_header_from_buffer (NULL, 0, NULL, NULL));
 306
 307   /* empty buffer has NULL as data pointer */
 308   buffer = gst_buffer_new_and_alloc (0);
 309
 310   /* no place to store the length and/or header data */
 311   ASSERT_CRITICAL (gst_dp_header_from_buffer (buffer, 0, NULL, NULL));
 312   ASSERT_CRITICAL (gst_dp_header_from_buffer (buffer, 0, &length, NULL));
 313
 314   /* this should work fine */
 315   fail_if (gst_dp_header_from_buffer (buffer, 0, &length, &header) != TRUE);
 316   fail_unless (length != 0);
 317   fail_unless (header != NULL);
 318
 319   /* this should validate */
 320   fail_if (gst_dp_validate_header (length, header) == FALSE);
 321
 322   /* NULL header pointer */
 323   ASSERT_CRITICAL (gst_dp_validate_header (length, NULL));
 324   /* short header */
 325   ASSERT_CRITICAL (gst_dp_validate_header (5, header));
 326
 327   g_free (header);
 328
 329   /* this should work and not crash trying to calc a CRC on a 0 sized buffer */
 330   fail_if (gst_dp_header_from_buffer (buffer,
 331           GST_DP_HEADER_FLAG_CRC_HEADER | GST_DP_HEADER_FLAG_CRC_PAYLOAD,
 332           &length, &header) != TRUE);
 333
 334   /* this should validate */
 335   fail_if (gst_dp_validate_header (length, header) == FALSE);
 336
 337   /* there was no payload, NULL as payload data should validate the CRC
 338    * checks and all. */
 339   fail_if (gst_dp_validate_payload (length, header, NULL) == FALSE);
 340
 341   /* and the whole packet as well */
 342   fail_if (gst_dp_validate_packet (length, header, NULL) == FALSE);
 343
 344   /* some bogus length */
 345   ASSERT_CRITICAL (gst_dp_validate_packet (5, header, NULL));
 346   gst_buffer_unref (buffer);
 347
 348   /* create buffer from header data, integrity tested elsewhere */
 349   buffer = gst_dp_buffer_from_header (length, header);
 350   fail_if (buffer == NULL);
 351   gst_buffer_unref (buffer);
 352   g_free (header);
 353
 354   ASSERT_CRITICAL (gst_dp_packet_from_caps (NULL, 0, NULL, NULL, NULL));
 355
 356   /* some caps stuff */
 357   caps = gst_caps_new_empty ();
 358   ASSERT_CRITICAL (gst_dp_packet_from_caps (caps, 0, NULL, NULL, NULL));
 359   ASSERT_CRITICAL (gst_dp_packet_from_caps (caps, 0, &length, NULL, NULL));
 360   ASSERT_CRITICAL (gst_dp_packet_from_caps (caps, 0, &length, &header, NULL));
 361
 362   fail_if (gst_dp_packet_from_caps (caps, 0, &length, &header,
 363           &payload) != TRUE);
 364   fail_if (strcmp ((const gchar *) payload, "EMPTY") != 0);
 365   gst_caps_unref (caps);
 366
 367   caps = gst_dp_caps_from_packet (length, header, payload);
 368   fail_if (caps == NULL);
 369   gst_caps_unref (caps);
 370
 371   g_free (header);
 372   g_free (payload);
 373
 374   /* some event stuff */
 375   event = gst_event_new_eos ();
 376   ASSERT_CRITICAL (gst_dp_packet_from_event (event, 0, NULL, NULL, NULL));
 377   ASSERT_CRITICAL (gst_dp_packet_from_event (event, 0, &length, NULL, NULL));
 378   ASSERT_CRITICAL (gst_dp_packet_from_event (event, 0, &length, &header, NULL));
 379
 380   /* payload is not NULL from previous test and points to freed memory, very
 381    * invalid. */
 382   fail_if (payload == NULL);
 383   fail_if (gst_dp_packet_from_event (event, 0, &length, &header,
 384           &payload) != TRUE);
 385
 386   /* the EOS event has no payload */
 387   fail_if (payload != NULL);
 388   gst_event_unref (event);
 389
 390   event = gst_dp_event_from_packet (length, header, payload);
 391   fail_if (event == NULL);
 392   fail_if (GST_EVENT_TYPE (event) != GST_EVENT_EOS);
 393   gst_event_unref (event);
 394
 395   g_free (header);
 396   g_free (payload);
 397 }
 398
 399 GST_END_TEST;
 400
 401 Suite *
 402 gst_dp_suite (void)
 403 {
 404   Suite *s = suite_create ("data protocol");
 405   TCase *tc_chain = tcase_create ("general");
 406
 407   suite_add_tcase (s, tc_chain);
 408   tcase_add_test (tc_chain, test_conversion);
 409 #ifndef HAVE_CPU_PPC64
 410   tcase_add_test (tc_chain, test_buffer);
 411 #endif
 412   tcase_add_test (tc_chain, test_caps);
 413   tcase_add_test (tc_chain, test_event);
 414   tcase_add_test (tc_chain, test_memory);
 415
 416   return s;
 417 }
 418
 419 GST_CHECK_MAIN (gst_dp);