mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-07-10 08:35:58 +00:00
tests/check: add rtprtx::test_drop_one_sender unit test
Test for one sender / one receiver Build the pipeline videotestsrc ! rtpvrawpay ! rtprtxsend ! rtprtxreceive ! fakesink and drop some buffers between rtprtxsend and rtprtxreceive Then it checks that every dropped packet has been re-sent. It also checks that not too much requests has been sent.
This commit is contained in:
Julien Isorce
Wim Taymans
parent
2e4ce28443
commit
2a2fa7ebc0
|
|
@ -32,6 +32,8 @@ static GstPad *srcpad, *sinkpad;
|
|||
/* we also have a list of src buffers */
|
||||
static GList *inbuffers = NULL;
|
||||
|
||||
static GMainLoop *main_loop;
|
||||
|
||||
#define RTP_CAPS_STRING \
|
||||
"application/x-rtp, " \
|
||||
"media = (string)audio, " \
|
||||
|
|
@ -250,15 +252,329 @@ GST_START_TEST (test_push_forward_seq)
|
|||
|
||||
GST_END_TEST;
|
||||
|
||||
static void
|
||||
message_received (GstBus * bus, GstMessage * message, GstPipeline * bin)
|
||||
{
|
||||
GST_INFO ("bus message from \"%" GST_PTR_FORMAT "\": %" GST_PTR_FORMAT,
|
||||
GST_MESSAGE_SRC (message), message);
|
||||
|
||||
switch (message->type) {
|
||||
case GST_MESSAGE_EOS:
|
||||
g_main_loop_quit (main_loop);
|
||||
break;
|
||||
case GST_MESSAGE_WARNING:{
|
||||
GError *gerror;
|
||||
gchar *debug;
|
||||
|
||||
gst_message_parse_warning (message, &gerror, &debug);
|
||||
gst_object_default_error (GST_MESSAGE_SRC (message), gerror, debug);
|
||||
g_error_free (gerror);
|
||||
g_free (debug);
|
||||
break;
|
||||
}
|
||||
case GST_MESSAGE_ERROR:{
|
||||
GError *gerror;
|
||||
gchar *debug;
|
||||
|
||||
gst_message_parse_error (message, &gerror, &debug);
|
||||
gst_object_default_error (GST_MESSAGE_SRC (message), gerror, debug);
|
||||
g_error_free (gerror);
|
||||
g_free (debug);
|
||||
g_main_loop_quit (main_loop);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
typedef struct
|
||||
{
|
||||
guint count;
|
||||
guint nb_packets;
|
||||
guint drop_every_n_packets;
|
||||
} RTXSendData;
|
||||
|
||||
typedef struct
|
||||
{
|
||||
guint nb_packets;
|
||||
guint seqnum_offset;
|
||||
guint seqnum_prev;
|
||||
} RTXReceiveData;
|
||||
|
||||
static GstPadProbeReturn
|
||||
rtprtxsend_srcpad_probe (GstPad * pad, GstPadProbeInfo * info,
|
||||
gpointer user_data)
|
||||
{
|
||||
GstPadProbeReturn ret = GST_PAD_PROBE_OK;
|
||||
|
||||
if (info->type == (GST_PAD_PROBE_TYPE_BUFFER | GST_PAD_PROBE_TYPE_PUSH)) {
|
||||
GstBuffer *buffer = GST_BUFFER (info->data);
|
||||
RTXSendData *rtxdata = (RTXSendData *) user_data;
|
||||
GstRTPBuffer rtp = GST_RTP_BUFFER_INIT;
|
||||
guint payload_type = 0;
|
||||
|
||||
gst_rtp_buffer_map (buffer, GST_MAP_READ, &rtp);
|
||||
payload_type = gst_rtp_buffer_get_payload_type (&rtp);
|
||||
|
||||
/* main stream packets */
|
||||
if (payload_type == 96) {
|
||||
/* count packets of the main stream */
|
||||
++rtxdata->nb_packets;
|
||||
/* drop some packets */
|
||||
if (rtxdata->count < rtxdata->drop_every_n_packets) {
|
||||
++rtxdata->count;
|
||||
} else {
|
||||
/* drop a packet every 'rtxdata->count' packets */
|
||||
rtxdata->count = 1;
|
||||
ret = GST_PAD_PROBE_DROP;
|
||||
}
|
||||
} else {
|
||||
/* retransmission packets */
|
||||
}
|
||||
|
||||
gst_rtp_buffer_unmap (&rtp);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
static GstPadProbeReturn
|
||||
rtprtxreceive_srcpad_probe (GstPad * pad, GstPadProbeInfo * info,
|
||||
gpointer user_data)
|
||||
{
|
||||
if (info->type == (GST_PAD_PROBE_TYPE_BUFFER | GST_PAD_PROBE_TYPE_PUSH)) {
|
||||
GstBuffer *buffer = GST_BUFFER (info->data);
|
||||
RTXReceiveData *rtxdata = (RTXReceiveData *) user_data;
|
||||
GstRTPBuffer rtp = GST_RTP_BUFFER_INIT;
|
||||
guint seqnum = 0;
|
||||
guint i = 0;
|
||||
|
||||
gst_rtp_buffer_map (buffer, GST_MAP_READ, &rtp);
|
||||
seqnum = gst_rtp_buffer_get_seq (&rtp);
|
||||
|
||||
/* check if there is a dropped packet */
|
||||
if (seqnum > rtxdata->seqnum_prev + rtxdata->seqnum_offset) {
|
||||
GstPad *peerpad = gst_pad_get_peer (pad);
|
||||
|
||||
/* ask retransmission of missing packet */
|
||||
for (i = rtxdata->seqnum_prev + rtxdata->seqnum_offset; i < seqnum;
|
||||
i += rtxdata->seqnum_offset) {
|
||||
GstEvent *event = gst_event_new_custom (GST_EVENT_CUSTOM_UPSTREAM,
|
||||
gst_structure_new ("GstRTPRetransmissionRequest",
|
||||
"seqnum", G_TYPE_UINT, i,
|
||||
"ssrc", G_TYPE_UINT, gst_rtp_buffer_get_ssrc (&rtp),
|
||||
"payload-type", G_TYPE_UINT,
|
||||
gst_rtp_buffer_get_payload_type (&rtp),
|
||||
NULL));
|
||||
gst_pad_push_event (peerpad, event);
|
||||
}
|
||||
gst_object_unref (peerpad);
|
||||
|
||||
rtxdata->seqnum_prev = seqnum;
|
||||
} else if (seqnum == rtxdata->seqnum_prev + rtxdata->seqnum_offset) {
|
||||
/* also update previous seqnum in this case */
|
||||
rtxdata->seqnum_prev = seqnum;
|
||||
}
|
||||
|
||||
gst_rtp_buffer_unmap (&rtp);
|
||||
|
||||
++rtxdata->nb_packets;
|
||||
}
|
||||
|
||||
return GST_PAD_PROBE_OK;
|
||||
}
|
||||
|
||||
static void
|
||||
start_test_drop_and_check_results (GstElement * bin, GstElement * rtppayloader,
|
||||
GstElement * rtprtxsend, GstElement * rtprtxreceive,
|
||||
RTXSendData * send_rtxdata, RTXReceiveData * receive_rtxdata,
|
||||
guint drop_every_n_packets)
|
||||
{
|
||||
GstStateChangeReturn state_res = GST_STATE_CHANGE_FAILURE;
|
||||
guint nbrtxrequests = 0;
|
||||
guint nbrtxpackets = 0;
|
||||
guint nb_expected_requests = 0;
|
||||
|
||||
GST_INFO ("starting test");
|
||||
|
||||
g_object_set (rtppayloader, "pt", 96, NULL);
|
||||
g_object_set (rtppayloader, "seqnum-offset", 1, NULL);
|
||||
g_object_set (rtprtxsend, "rtx-payload-type", 99, NULL);
|
||||
g_object_set (rtprtxreceive, "rtx-payload-types", "99:111:125", NULL);
|
||||
|
||||
send_rtxdata->count = 1;
|
||||
send_rtxdata->nb_packets = 0;
|
||||
send_rtxdata->drop_every_n_packets = drop_every_n_packets;
|
||||
|
||||
receive_rtxdata->nb_packets = 0;
|
||||
receive_rtxdata->seqnum_offset = 0;
|
||||
receive_rtxdata->seqnum_prev = 0;
|
||||
|
||||
/* retrieve offset before going to paused */
|
||||
g_object_get (G_OBJECT (rtppayloader), "seqnum-offset",
|
||||
&receive_rtxdata->seqnum_offset, NULL);
|
||||
|
||||
/* prepare playing */
|
||||
state_res = gst_element_set_state (bin, GST_STATE_PAUSED);
|
||||
ck_assert_int_ne (state_res, GST_STATE_CHANGE_FAILURE);
|
||||
|
||||
/* wait for completion */
|
||||
state_res = gst_element_get_state (bin, NULL, NULL, GST_CLOCK_TIME_NONE);
|
||||
ck_assert_int_ne (state_res, GST_STATE_CHANGE_FAILURE);
|
||||
|
||||
/* retrieve seqnum_prev here to make sure it has been reseted */
|
||||
g_object_get (G_OBJECT (rtppayloader), "seqnum",
|
||||
&receive_rtxdata->seqnum_prev, NULL);
|
||||
|
||||
/* run pipeline */
|
||||
state_res = gst_element_set_state (bin, GST_STATE_PLAYING);
|
||||
ck_assert_int_ne (state_res, GST_STATE_CHANGE_FAILURE);
|
||||
|
||||
GST_INFO ("running main loop");
|
||||
g_main_loop_run (main_loop);
|
||||
|
||||
/* check results */
|
||||
|
||||
if (send_rtxdata->nb_packets % drop_every_n_packets == 0) {
|
||||
/* special case because the last buffer will be dropped
|
||||
* so the receiver cannot know if it has been dropped (no next packet)
|
||||
*/
|
||||
nb_expected_requests = send_rtxdata->nb_packets / drop_every_n_packets - 1;
|
||||
fail_unless_equals_int (send_rtxdata->nb_packets,
|
||||
receive_rtxdata->nb_packets + 1);
|
||||
} else {
|
||||
nb_expected_requests = send_rtxdata->nb_packets / drop_every_n_packets;
|
||||
fail_unless_equals_int (send_rtxdata->nb_packets,
|
||||
receive_rtxdata->nb_packets);
|
||||
}
|
||||
|
||||
g_object_get (G_OBJECT (rtprtxsend), "num-rtx-requests", &nbrtxrequests,
|
||||
NULL);
|
||||
fail_unless_equals_int (nbrtxrequests, nb_expected_requests);
|
||||
|
||||
g_object_get (G_OBJECT (rtprtxsend), "num-rtx-packets", &nbrtxpackets, NULL);
|
||||
fail_unless_equals_int (nbrtxpackets, nb_expected_requests);
|
||||
|
||||
g_object_get (G_OBJECT (rtprtxreceive), "num-rtx-requests", &nbrtxrequests,
|
||||
NULL);
|
||||
fail_unless_equals_int (nbrtxrequests, nb_expected_requests);
|
||||
|
||||
g_object_get (G_OBJECT (rtprtxreceive), "num-rtx-packets", &nbrtxpackets,
|
||||
NULL);
|
||||
fail_unless_equals_int (nbrtxpackets, nb_expected_requests);
|
||||
|
||||
g_object_get (G_OBJECT (rtprtxreceive), "num-rtx-assoc-packets",
|
||||
&nbrtxpackets, NULL);
|
||||
fail_unless_equals_int (nbrtxpackets, nb_expected_requests);
|
||||
|
||||
state_res = gst_element_set_state (bin, GST_STATE_NULL);
|
||||
ck_assert_int_ne (state_res, GST_STATE_CHANGE_FAILURE);
|
||||
}
|
||||
|
||||
/* This test build the pipeline videotestsrc ! rtpvrawpay ! rtprtxsend ! rtprtxreceive ! fakesink
|
||||
* and drop some buffer between rtprtxsend and rtprtxreceive
|
||||
* Then it checks that every dropped packet has been re-sent and it checks that
|
||||
* not too much requests has been sent.
|
||||
*/
|
||||
GST_START_TEST (test_drop_one_sender)
|
||||
{
|
||||
GstElement *bin, *src, *rtppayloader, *rtprtxsend, *rtprtxreceive, *sink;
|
||||
GstBus *bus;
|
||||
gboolean res;
|
||||
GstPad *srcpad, *sinkpad;
|
||||
GstStreamConsistency *chk_1, *chk_2, *chk_3;
|
||||
gint num_buffers = 20;
|
||||
guint drop_every_n_packets = 0;
|
||||
RTXSendData send_rtxdata;
|
||||
RTXReceiveData receive_rtxdata;
|
||||
|
||||
GST_INFO ("preparing test");
|
||||
|
||||
/* build pipeline */
|
||||
bin = gst_pipeline_new ("pipeline");
|
||||
bus = gst_element_get_bus (bin);
|
||||
gst_bus_add_signal_watch_full (bus, G_PRIORITY_HIGH);
|
||||
|
||||
src = gst_element_factory_make ("videotestsrc", "src");
|
||||
g_object_set (src, "num-buffers", num_buffers, NULL);
|
||||
rtppayloader = gst_element_factory_make ("rtpvrawpay", "rtppayloader");
|
||||
rtprtxsend = gst_element_factory_make ("rtprtxsend", "rtprtxsend");
|
||||
rtprtxreceive = gst_element_factory_make ("rtprtxreceive", "rtprtxreceive");
|
||||
sink = gst_element_factory_make ("fakesink", "sink");
|
||||
gst_bin_add_many (GST_BIN (bin), src, rtppayloader, rtprtxsend, rtprtxreceive,
|
||||
sink, NULL);
|
||||
|
||||
res = gst_element_link (src, rtppayloader);
|
||||
fail_unless (res == TRUE, NULL);
|
||||
res = gst_element_link (rtppayloader, rtprtxsend);
|
||||
fail_unless (res == TRUE, NULL);
|
||||
res = gst_element_link (rtprtxsend, rtprtxreceive);
|
||||
fail_unless (res == TRUE, NULL);
|
||||
res = gst_element_link (rtprtxreceive, sink);
|
||||
fail_unless (res == TRUE, NULL);
|
||||
|
||||
/* create consistency checkers for the pads */
|
||||
|
||||
srcpad = gst_element_get_static_pad (rtppayloader, "src");
|
||||
chk_1 = gst_consistency_checker_new (srcpad);
|
||||
gst_object_unref (srcpad);
|
||||
|
||||
srcpad = gst_element_get_static_pad (rtprtxsend, "src");
|
||||
gst_pad_add_probe (srcpad,
|
||||
(GST_PAD_PROBE_TYPE_BUFFER | GST_PAD_PROBE_TYPE_PUSH),
|
||||
(GstPadProbeCallback) rtprtxsend_srcpad_probe, &send_rtxdata, NULL);
|
||||
sinkpad = gst_pad_get_peer (srcpad);
|
||||
fail_if (sinkpad == NULL);
|
||||
chk_2 = gst_consistency_checker_new (sinkpad);
|
||||
gst_object_unref (sinkpad);
|
||||
gst_object_unref (srcpad);
|
||||
|
||||
srcpad = gst_element_get_static_pad (rtprtxreceive, "src");
|
||||
gst_pad_add_probe (srcpad,
|
||||
(GST_PAD_PROBE_TYPE_BUFFER | GST_PAD_PROBE_TYPE_PUSH),
|
||||
(GstPadProbeCallback) rtprtxreceive_srcpad_probe, &receive_rtxdata, NULL);
|
||||
sinkpad = gst_pad_get_peer (srcpad);
|
||||
fail_if (sinkpad == NULL);
|
||||
chk_3 = gst_consistency_checker_new (sinkpad);
|
||||
gst_object_unref (sinkpad);
|
||||
gst_object_unref (srcpad);
|
||||
|
||||
main_loop = g_main_loop_new (NULL, FALSE);
|
||||
g_signal_connect (bus, "message::error", (GCallback) message_received, bin);
|
||||
g_signal_connect (bus, "message::warning", (GCallback) message_received, bin);
|
||||
g_signal_connect (bus, "message::eos", (GCallback) message_received, bin);
|
||||
|
||||
for (drop_every_n_packets = 2; drop_every_n_packets < 10;
|
||||
drop_every_n_packets++) {
|
||||
start_test_drop_and_check_results (bin, rtppayloader, rtprtxsend,
|
||||
rtprtxreceive, &send_rtxdata, &receive_rtxdata, drop_every_n_packets);
|
||||
}
|
||||
|
||||
/* cleanup */
|
||||
g_main_loop_unref (main_loop);
|
||||
gst_consistency_checker_free (chk_1);
|
||||
gst_consistency_checker_free (chk_2);
|
||||
gst_consistency_checker_free (chk_3);
|
||||
gst_bus_remove_signal_watch (bus);
|
||||
gst_object_unref (bus);
|
||||
gst_object_unref (bin);
|
||||
}
|
||||
|
||||
GST_END_TEST;
|
||||
|
||||
static Suite *
|
||||
rtprtx_suite (void)
|
||||
{
|
||||
Suite *s = suite_create ("rtprtx");
|
||||
TCase *tc_chain = tcase_create ("general");
|
||||
|
||||
tcase_set_timeout (tc_chain, 10000);
|
||||
|
||||
suite_add_tcase (s, tc_chain);
|
||||
|
||||
tcase_add_test (tc_chain, test_push_forward_seq);
|
||||
tcase_add_test (tc_chain, test_drop_one_sender);
|
||||
|
||||
return s;
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue