mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2025-02-17 11:45:25 +00:00
tests: Add test for mpegts library
Tests for PAT, PMT, and NIT Creates a new table, and populates it with descriptors. Parses the newly created tables, and checks the data. Creates a GstMpegTsSection from the tables, and packetize the sections. The packetized section data is byte-wise compared to a static byte array https://bugzilla.gnome.org/show_bug.cgi?id=723953
This commit is contained in:
parent
2662f0f72a
commit
bb22c6e7c4
3 changed files with 399 additions and 0 deletions
|
@ -233,6 +233,7 @@ check_PROGRAMS = \
|
|||
pipelines/mxf \
|
||||
$(check_mimic) \
|
||||
libs/mpegvideoparser \
|
||||
libs/mpegts \
|
||||
libs/h264parser \
|
||||
$(check_uvch264) \
|
||||
libs/vc1parser \
|
||||
|
@ -280,6 +281,16 @@ libs_mpegvideoparser_LDADD = \
|
|||
$(GST_PLUGINS_BAD_LIBS) -lgstcodecparsers-@GST_API_VERSION@ \
|
||||
$(GST_BASE_LIBS) $(GST_LIBS) $(LDADD)
|
||||
|
||||
libs_mpegts_CFLAGS = \
|
||||
$(GST_PLUGINS_BAD_CFLAGS) \
|
||||
-DGST_USE_UNSTABLE_API \
|
||||
$(GST_CFLAGS)
|
||||
|
||||
libs_mpegts_LDADD = \
|
||||
$(top_builddir)/gst-libs/gst/mpegts/libgstmpegts-@GST_API_VERSION@.la \
|
||||
$(GST_BASE_LIBS) \
|
||||
$(GST_LIBS) $(LDADD)
|
||||
|
||||
libs_h264parser_CFLAGS = \
|
||||
$(GST_PLUGINS_BAD_CFLAGS) $(GST_PLUGINS_BASE_CFLAGS) \
|
||||
-DGST_USE_UNSTABLE_API \
|
||||
|
|
1
tests/check/libs/.gitignore
vendored
1
tests/check/libs/.gitignore
vendored
|
@ -1,5 +1,6 @@
|
|||
.dirstamp
|
||||
h264parser
|
||||
mpegvideoparser
|
||||
mpegts
|
||||
vc1parser
|
||||
insertbin
|
||||
|
|
387
tests/check/libs/mpegts.c
Normal file
387
tests/check/libs/mpegts.c
Normal file
|
@ -0,0 +1,387 @@
|
|||
/* Gstreamer
|
||||
* Copyright (C) <2014> Jesper Larsen <knorr.jesper@gmail.com>
|
||||
*
|
||||
* 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., 51 Franklin St, Fifth Floor,
|
||||
* Boston, MA 02110-1301, USA.
|
||||
*/
|
||||
|
||||
#include <gst/check/gstcheck.h>
|
||||
#include <gst/mpegts/mpegts.h>
|
||||
|
||||
static const guint8 pat_data_check[] = {
|
||||
0x00, 0xB0, 0x11, 0x00, 0x00, 0xc1, 0x00,
|
||||
0x00, 0x00, 0x00, 0xe0, 0x30, 0x00, 0x01,
|
||||
0xe0, 0x31, 0x98, 0xdf, 0x37, 0xc4
|
||||
};
|
||||
|
||||
static const guint8 pmt_data_check[] = {
|
||||
0x02, 0xb0, 0x29, 0x00, 0x01, 0xc1, 0x00,
|
||||
0x00, 0xff, 0xff, 0xf0, 0x06, 0x05, 0x04,
|
||||
0x48, 0x44, 0x4d, 0x56, 0x1b, 0xe0, 0x40,
|
||||
0xF0, 0x06, 0x05, 0x04, 0x48, 0x44, 0x4d,
|
||||
0x56, 0x1b, 0xe0, 0x41, 0xF0, 0x06, 0x05,
|
||||
0x04, 0x48, 0x44, 0x4d, 0x56, 0x15, 0x41,
|
||||
0x5f, 0x5b
|
||||
};
|
||||
|
||||
static const guint8 nit_data_check[] = {
|
||||
0x40, 0xf0, 0x49, 0x1f, 0xff, 0xc1, 0x00,
|
||||
0x00, 0xf0, 0x0e, 0x40, 0x0c, 0x4e, 0x65,
|
||||
0x74, 0x77, 0x6f, 0x72, 0x6b, 0x20, 0x6e,
|
||||
0x61, 0x6d, 0x65, 0xf0, 0x2e, 0x1f, 0xff,
|
||||
0x1f, 0xfe, 0xf0, 0x11, 0x40, 0x0f, 0x41,
|
||||
0x6e, 0x6f, 0x74, 0x68, 0x65, 0x72, 0x20,
|
||||
0x6e, 0x65, 0x74, 0x77, 0x6f, 0x72, 0x6b,
|
||||
0x1f, 0xff, 0x1f, 0xfe, 0xf0, 0x11, 0x40,
|
||||
0x0f, 0x41, 0x6e, 0x6f, 0x74, 0x68, 0x65,
|
||||
0x72, 0x20, 0x6e, 0x65, 0x74, 0x77, 0x6f,
|
||||
0x72, 0x6b, 0xce, 0x03, 0xf5, 0x94
|
||||
};
|
||||
|
||||
GST_START_TEST (test_mpegts_pat)
|
||||
{
|
||||
GstMpegTsPatProgram *program;
|
||||
GPtrArray *pat;
|
||||
GstMpegTsSection *pat_section;
|
||||
gint i;
|
||||
guint8 *data;
|
||||
gsize data_size;
|
||||
|
||||
/* Check creation of PAT */
|
||||
pat = g_ptr_array_new ();
|
||||
|
||||
for (i = 0; i < 2; i++) {
|
||||
program = gst_mpegts_pat_program_new ();
|
||||
|
||||
program->program_number = i;
|
||||
program->network_or_program_map_PID = 0x30 + i;
|
||||
|
||||
g_ptr_array_add (pat, program);
|
||||
}
|
||||
|
||||
pat_section = gst_mpegts_section_from_pat (pat, 0);
|
||||
fail_if (pat_section == NULL);
|
||||
|
||||
/* Re-parse the PAT from section */
|
||||
pat = gst_mpegts_section_get_pat (pat_section);
|
||||
|
||||
fail_unless (pat->len == 2);
|
||||
|
||||
for (i = 0; i < 2; i++) {
|
||||
program = g_ptr_array_index (pat, i);
|
||||
|
||||
assert_equals_int (program->program_number, i);
|
||||
assert_equals_int (program->network_or_program_map_PID, 0x30 + i);
|
||||
}
|
||||
|
||||
/* Packetize the section, and check the data integrity */
|
||||
data = gst_mpegts_section_packetize (pat_section, &data_size);
|
||||
fail_if (data == NULL);
|
||||
|
||||
for (i = 0; i < data_size; i++) {
|
||||
if (data[i] != pat_data_check[i])
|
||||
fail ("0x%X != 0x%X in byte %d of PAT section", data[i],
|
||||
pat_data_check[i], i);
|
||||
}
|
||||
|
||||
/* Check assertion on bad CRC. Reset parsed data, and make the CRC corrupt */
|
||||
pat_section->data[pat_section->section_length - 1]++;
|
||||
pat_section->cached_parsed = NULL;
|
||||
pat = gst_mpegts_section_get_pat (pat_section);
|
||||
|
||||
fail_unless (pat == NULL);
|
||||
|
||||
gst_mpegts_section_unref (pat_section);
|
||||
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
GST_START_TEST (test_mpegts_pmt)
|
||||
{
|
||||
GstMpegTsPMT *pmt;
|
||||
GstMpegTsPMTStream *stream;
|
||||
GstMpegTsDescriptor *desc;
|
||||
GstMpegTsSection *pmt_section;
|
||||
guint8 *data;
|
||||
gsize data_size;
|
||||
gint i;
|
||||
|
||||
/* Check creation of PMT */
|
||||
pmt = gst_mpegts_pmt_new ();
|
||||
|
||||
pmt->pcr_pid = 0x1FFF;
|
||||
pmt->program_number = 1;
|
||||
|
||||
desc = gst_mpegts_descriptor_from_registration ("HDMV", NULL, 0);
|
||||
g_ptr_array_add (pmt->descriptors, desc);
|
||||
|
||||
for (i = 0; i < 2; i++) {
|
||||
stream = gst_mpegts_pmt_stream_new ();
|
||||
|
||||
stream->stream_type = GST_MPEG_TS_STREAM_TYPE_VIDEO_H264;
|
||||
stream->pid = 0x40 + i;
|
||||
|
||||
desc = gst_mpegts_descriptor_from_registration ("HDMV", NULL, 0);
|
||||
|
||||
g_ptr_array_add (stream->descriptors, desc);
|
||||
g_ptr_array_add (pmt->streams, stream);
|
||||
}
|
||||
|
||||
pmt_section = gst_mpegts_section_from_pmt (pmt, 0x30);
|
||||
fail_if (pmt_section == NULL);
|
||||
|
||||
/* Re-parse the PMT from section */
|
||||
pmt = (GstMpegTsPMT *) gst_mpegts_section_get_pmt (pmt_section);
|
||||
|
||||
fail_unless (pmt->pcr_pid == 0x1FFF);
|
||||
fail_unless (pmt->program_number == 1);
|
||||
fail_unless (pmt->descriptors->len == 1);
|
||||
fail_unless (pmt->streams->len == 2);
|
||||
|
||||
desc = (GstMpegTsDescriptor *) gst_mpegts_find_descriptor (pmt->descriptors,
|
||||
GST_MTS_DESC_REGISTRATION);
|
||||
fail_if (desc == NULL);
|
||||
|
||||
for (i = 0; i < 2; i++) {
|
||||
stream = g_ptr_array_index (pmt->streams, i);
|
||||
|
||||
fail_unless (stream->stream_type == GST_MPEG_TS_STREAM_TYPE_VIDEO_H264);
|
||||
fail_unless (stream->pid == 0x40 + i);
|
||||
fail_unless (stream->descriptors->len == 1);
|
||||
|
||||
desc =
|
||||
(GstMpegTsDescriptor *) gst_mpegts_find_descriptor (stream->descriptors,
|
||||
GST_MTS_DESC_REGISTRATION);
|
||||
fail_if (desc == NULL);
|
||||
}
|
||||
|
||||
/* Packetize the section, and check data integrity */
|
||||
data = gst_mpegts_section_packetize (pmt_section, &data_size);
|
||||
|
||||
for (i = 0; i < data_size; i++) {
|
||||
if (data[i] != pmt_data_check[i])
|
||||
fail ("0x%X != 0x%X in byte %d of PMT section", data[i],
|
||||
pmt_data_check[i], i);
|
||||
}
|
||||
|
||||
/* Check assertion on bad CRC. Reset parsed data, and make the CRC corrupt */
|
||||
pmt_section->data[pmt_section->section_length - 1]++;
|
||||
pmt_section->cached_parsed = NULL;
|
||||
pmt = (GstMpegTsPMT *) gst_mpegts_section_get_pmt (pmt_section);
|
||||
|
||||
fail_unless (pmt == NULL);
|
||||
|
||||
gst_mpegts_section_unref (pmt_section);
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
|
||||
GST_START_TEST (test_mpegts_nit)
|
||||
{
|
||||
GstMpegTsNITStream *stream;
|
||||
GstMpegTsNIT *nit;
|
||||
GstMpegTsDescriptor *desc;
|
||||
GstMpegTsSection *nit_section;
|
||||
gchar *name;
|
||||
guint8 *data;
|
||||
gsize data_size;
|
||||
gint i;
|
||||
|
||||
/* Check creation of NIT */
|
||||
nit = gst_mpegts_nit_new ();
|
||||
|
||||
nit->actual_network = TRUE;
|
||||
nit->network_id = 0x1FFF;
|
||||
|
||||
desc = gst_mpegts_descriptor_from_dvb_network_name ("Network name");
|
||||
|
||||
fail_if (desc == NULL);
|
||||
|
||||
g_ptr_array_add (nit->descriptors, desc);
|
||||
|
||||
for (i = 0; i < 2; i++) {
|
||||
stream = gst_mpegts_nit_stream_new ();
|
||||
stream->transport_stream_id = 0x1FFF;
|
||||
stream->original_network_id = 0x1FFE;
|
||||
|
||||
desc = gst_mpegts_descriptor_from_dvb_network_name ("Another network");
|
||||
|
||||
g_ptr_array_add (stream->descriptors, desc);
|
||||
g_ptr_array_add (nit->streams, stream);
|
||||
}
|
||||
|
||||
nit_section = gst_mpegts_section_from_nit (nit);
|
||||
fail_if (nit_section == NULL);
|
||||
|
||||
/* Re-parse the NIT from section */
|
||||
nit = (GstMpegTsNIT *) gst_mpegts_section_get_nit (nit_section);
|
||||
|
||||
fail_unless (nit->descriptors->len == 1);
|
||||
fail_unless (nit->streams->len == 2);
|
||||
fail_unless (nit->actual_network == TRUE);
|
||||
fail_unless (nit->network_id == 0x1FFF);
|
||||
|
||||
desc = (GstMpegTsDescriptor *) gst_mpegts_find_descriptor (nit->descriptors,
|
||||
GST_MTS_DESC_DVB_NETWORK_NAME);
|
||||
|
||||
fail_if (desc == NULL);
|
||||
|
||||
fail_unless (gst_mpegts_descriptor_parse_dvb_network_name (desc,
|
||||
&name) == TRUE);
|
||||
|
||||
for (i = 0; i < 2; i++) {
|
||||
stream = g_ptr_array_index (nit->streams, i);
|
||||
|
||||
fail_unless (stream->transport_stream_id == 0x1FFF);
|
||||
fail_unless (stream->original_network_id == 0x1FFE);
|
||||
fail_unless (stream->descriptors->len == 1);
|
||||
|
||||
desc =
|
||||
(GstMpegTsDescriptor *) gst_mpegts_find_descriptor (stream->descriptors,
|
||||
GST_MTS_DESC_DVB_NETWORK_NAME);
|
||||
|
||||
fail_unless (gst_mpegts_descriptor_parse_dvb_network_name (desc,
|
||||
&name) == TRUE);
|
||||
}
|
||||
|
||||
/* Packetize the section, and check data integrity */
|
||||
data = gst_mpegts_section_packetize (nit_section, &data_size);
|
||||
|
||||
fail_if (data == NULL);
|
||||
|
||||
for (i = 0; i < data_size; i++) {
|
||||
if (data[i] != nit_data_check[i])
|
||||
fail ("0x%X != 0x%X in byte %d of NIT section", data[i],
|
||||
nit_data_check[i], i);
|
||||
}
|
||||
|
||||
/* Check assertion on bad CRC. Reset parsed data, and make the CRC corrupt */
|
||||
nit_section->data[nit_section->section_length - 1]++;
|
||||
nit_section->cached_parsed = NULL;
|
||||
nit = (GstMpegTsNIT *) gst_mpegts_section_get_nit (nit_section);
|
||||
|
||||
fail_unless (nit == NULL);
|
||||
|
||||
gst_mpegts_section_unref (nit_section);
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
static const guint8 registration_descriptor[] = {
|
||||
0x05, 0x04, 0x48, 0x44, 0x4d, 0x56
|
||||
};
|
||||
|
||||
GST_START_TEST (test_mpegts_descriptors)
|
||||
{
|
||||
GstMpegTsDescriptor *desc;
|
||||
guint i;
|
||||
|
||||
/*
|
||||
* Registration descriptor (0x05)
|
||||
*/
|
||||
|
||||
/* Check creation of descriptor */
|
||||
desc = gst_mpegts_descriptor_from_registration ("HDMV", NULL, 0);
|
||||
fail_if (desc == NULL);
|
||||
fail_unless (desc->length == 4);
|
||||
fail_unless (desc->tag == 0x05);
|
||||
for (i = 0; i < 6; i++) {
|
||||
if (registration_descriptor[i] != desc->data[i])
|
||||
fail ("0x%X != 0x%X in byte %d of registration descriptor",
|
||||
desc->data[i], registration_descriptor[i], i);
|
||||
}
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
static const guint8 network_name_descriptor[] = {
|
||||
0x40, 0x04, 0x4e, 0x61, 0x6d, 0x65
|
||||
};
|
||||
|
||||
GST_START_TEST (test_mpegts_dvb_descriptors)
|
||||
{
|
||||
GstMpegTsDescriptor *desc;
|
||||
gchar *string;
|
||||
gchar long_string[257];
|
||||
gboolean ret;
|
||||
guint i;
|
||||
|
||||
/*
|
||||
* Network name descriptor (0x40)
|
||||
*/
|
||||
|
||||
/* Check creation of descriptor */
|
||||
desc = gst_mpegts_descriptor_from_dvb_network_name ("Name");
|
||||
fail_if (desc == NULL);
|
||||
fail_unless (desc->length == 4);
|
||||
fail_unless (desc->tag == 0x40);
|
||||
|
||||
for (i = 0; i < 6; i++) {
|
||||
if (desc->data[i] != network_name_descriptor[i])
|
||||
fail ("0x%X != 0x%X in byte %d of network name descriptor",
|
||||
desc->data[i], network_name_descriptor[i], i);
|
||||
}
|
||||
|
||||
/* Check parsing of descriptor */
|
||||
ret = gst_mpegts_descriptor_parse_dvb_network_name (desc, &string);
|
||||
fail_unless (ret == TRUE);
|
||||
fail_unless (strcmp (string, "Name") == 0);
|
||||
|
||||
/* Descriptor should fail if string is more than 255 bytes */
|
||||
memset (long_string, 0x41, 256);
|
||||
long_string[256] = 0x00;
|
||||
ASSERT_CRITICAL (gst_mpegts_descriptor_from_dvb_network_name (long_string));
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
static Suite *
|
||||
mpegts_suite (void)
|
||||
{
|
||||
Suite *s = suite_create ("MPEG Transport Stream helper library");
|
||||
|
||||
TCase *tc_chain = tcase_create ("general");
|
||||
|
||||
gst_mpegts_initialize ();
|
||||
|
||||
suite_add_tcase (s, tc_chain);
|
||||
tcase_add_test (tc_chain, test_mpegts_pat);
|
||||
tcase_add_test (tc_chain, test_mpegts_pmt);
|
||||
tcase_add_test (tc_chain, test_mpegts_nit);
|
||||
tcase_add_test (tc_chain, test_mpegts_descriptors);
|
||||
tcase_add_test (tc_chain, test_mpegts_dvb_descriptors);
|
||||
|
||||
return s;
|
||||
}
|
||||
|
||||
int
|
||||
main (int argc, char **argv)
|
||||
{
|
||||
int nf;
|
||||
|
||||
Suite *s = mpegts_suite ();
|
||||
|
||||
SRunner *sr = srunner_create (s);
|
||||
|
||||
gst_check_init (&argc, &argv);
|
||||
|
||||
srunner_run_all (sr, CK_NORMAL);
|
||||
nf = srunner_ntests_failed (sr);
|
||||
srunner_free (sr);
|
||||
|
||||
return nf;
|
||||
}
|
Loading…
Reference in a new issue