diff options
| -rw-r--r-- | tests/Makefile.am | 5 | ||||
| -rw-r--r-- | tests/test-drop-invalid.c | 517 |
2 files changed, 521 insertions, 1 deletions
diff --git a/tests/Makefile.am b/tests/Makefile.am index d24a2aa..62d5d64 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -54,7 +54,8 @@ check_PROGRAMS = \ test-tcp \ test-icetcp \ test-credentials \ - test-turn + test-turn \ + test-drop-invalid dist_check_SCRIPTS = \ check-test-fullmode-with-stun.sh \ @@ -128,6 +129,8 @@ test_credentials_LDADD = $(COMMON_LDADD) test_turn_LDADD = $(COMMON_LDADD) +test_drop_invalid_LDADD = $(COMMON_LDADD) + test_gstreamer_CFLAGS = $(AM_CFLAGS) $(GST_CHECK_CFLAGS) test_gstreamer_LDADD = -lnice -L$(top_builddir)/nice/.libs $(GLIB_LIBS) $(GUPNP_LIBS) $(GST_CHECK_LIBS) $(GST_LIBS) diff --git a/tests/test-drop-invalid.c b/tests/test-drop-invalid.c new file mode 100644 index 0000000..97e3586 --- /dev/null +++ b/tests/test-drop-invalid.c @@ -0,0 +1,517 @@ +/* + * This file is part of the Nice GLib ICE library. + * + * Unit test for ICE full-mode related features. + * + * (C) 2007 Nokia Corporation. All rights reserved. + * Contact: Kai Vehmanen + * (C) 2017 Collabora Ltd + * Contact: Olivier Crete <olivier.crete@collabora.com> + * + * The contents of this file are subject to the Mozilla Public License Version + * 1.1 (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * http://www.mozilla.org/MPL/ + * + * Software distributed under the License is distributed on an "AS IS" basis, + * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License + * for the specific language governing rights and limitations under the + * License. + * + * The Original Code is the Nice GLib ICE library. + * + * The Initial Developers of the Original Code are Collabora Ltd and Nokia + * Corporation. All Rights Reserved. + * + * Contributors: + * Kai Vehmanen, Nokia + * + * Alternatively, the contents of this file may be used under the terms of the + * the GNU Lesser General Public License Version 2.1 (the "LGPL"), in which + * case the provisions of LGPL are applicable instead of those above. If you + * wish to allow use of your version of this file only under the terms of the + * LGPL and not to allow others to use your version of this file under the + * MPL, indicate your decision by deleting the provisions above and replace + * them with the notice and other provisions required by the LGPL. If you do + * not delete the provisions above, a recipient may use your version of this + * file under either the MPL or the LGPL. + */ +#ifdef HAVE_CONFIG_H +# include <config.h> +#endif + +#include "agent.h" + +#include "socket/socket.h" + +#include <stdlib.h> +#include <string.h> + + + +static NiceComponentState global_lagent_state[2] = { NICE_COMPONENT_STATE_LAST, NICE_COMPONENT_STATE_LAST }; +static NiceComponentState global_ragent_state[2] = { NICE_COMPONENT_STATE_LAST, NICE_COMPONENT_STATE_LAST }; +static guint global_components_ready = 0; +static guint global_components_ready_exit = 0; +static guint global_components_failed = 0; +static guint global_components_failed_exit = 0; +static GMainLoop *global_mainloop = NULL; +static gboolean global_lagent_gathering_done = FALSE; +static gboolean global_ragent_gathering_done = FALSE; +static gboolean global_lagent_ibr_received = FALSE; +static gboolean global_ragent_ibr_received = FALSE; +static int global_lagent_cands = 0; +static int global_ragent_cands = 0; +static gint global_ragent_read = 0; +static guint global_exit_when_ibr_received = 0; + +static void priv_print_global_status (void) +{ + g_debug ("\tgathering_done=%d", global_lagent_gathering_done && global_ragent_gathering_done); + g_debug ("\tlstate[rtp]=%d [rtcp]=%d", global_lagent_state[0], global_lagent_state[1]); + g_debug ("\trstate[rtp]=%d [rtcp]=%d", global_ragent_state[0], global_ragent_state[1]); + g_debug ("\tL cands=%d R cands=%d", global_lagent_cands, global_ragent_cands); +} + +static gboolean timer_cb (gpointer pointer) +{ + g_debug ("test-drop-invalid:%s: %p", G_STRFUNC, pointer); + + /* signal status via a global variable */ + + /* note: should not be reached, abort */ + g_error ("ERROR: test has got stuck, aborting..."); + + return FALSE; +} + +static void cb_nice_recv (NiceAgent *agent, guint stream_id, guint component_id, guint len, gchar *buf, gpointer user_data) +{ + g_debug ("test-drop-invalid:%s: %p", G_STRFUNC, user_data); + + /* XXX: dear compiler, these are for you: */ + (void)agent; (void)stream_id; (void)component_id; (void)buf; + + /* Core of the test + * Assert on any unreleated packet received. This would include anything + * send before the negotiation is over. + */ + g_assert (len == 16); + g_assert (strncmp ("1234567812345678", buf, 16) == 0); + + if (component_id == 2) + return; + + if (GPOINTER_TO_UINT (user_data) == 2) { + g_debug ("right agent received %d bytes, stopping mainloop", len); + global_ragent_read = len; + g_main_loop_quit (global_mainloop); + } +} + +static void cb_candidate_gathering_done(NiceAgent *agent, guint stream_id, gpointer data) +{ + g_debug ("test-drop-invalid:%s: %p", G_STRFUNC, data); + + if (GPOINTER_TO_UINT (data) == 1) + global_lagent_gathering_done = TRUE; + else if (GPOINTER_TO_UINT (data) == 2) + global_ragent_gathering_done = TRUE; + + if (global_lagent_gathering_done && + global_ragent_gathering_done) + g_main_loop_quit (global_mainloop); + + /* XXX: dear compiler, these are for you: */ + (void)agent; +} + +static void cb_component_state_changed (NiceAgent *agent, guint stream_id, guint component_id, guint state, gpointer data) +{ + gboolean ready_to_connected = FALSE; + g_debug ("test-drop-invalid:%s: %p", G_STRFUNC, data); + + if (GPOINTER_TO_UINT (data) == 1) { + if (global_lagent_state[component_id - 1] == NICE_COMPONENT_STATE_READY && + state == NICE_COMPONENT_STATE_CONNECTED) + ready_to_connected = TRUE; + global_lagent_state[component_id - 1] = state; + } else if (GPOINTER_TO_UINT (data) == 2) { + if (global_ragent_state[component_id - 1] == NICE_COMPONENT_STATE_READY && + state == NICE_COMPONENT_STATE_CONNECTED) + ready_to_connected = TRUE; + global_ragent_state[component_id - 1] = state; + } + + if (state == NICE_COMPONENT_STATE_READY) + global_components_ready++; + else if (state == NICE_COMPONENT_STATE_CONNECTED && ready_to_connected) + global_components_ready--; + if (state == NICE_COMPONENT_STATE_FAILED) + global_components_failed++; + + g_debug ("test-drop-invalid: checks READY/EXIT-AT %u/%u.", global_components_ready, global_components_ready_exit); + g_debug ("test-drop-invalid: checks FAILED/EXIT-AT %u/%u.", global_components_failed, global_components_failed_exit); + + /* signal status via a global variable */ + if (global_components_ready == global_components_ready_exit && + global_components_failed == global_components_failed_exit) { + g_debug ("Components ready/failed achieved. Stopping mailoop"); + g_main_loop_quit (global_mainloop); + return; + } + + /* XXX: dear compiler, these are for you: */ + (void)agent; (void)stream_id; (void)data; (void)component_id; +} + +static void cb_new_selected_pair(NiceAgent *agent, guint stream_id, guint component_id, + gchar *lfoundation, gchar* rfoundation, gpointer data) +{ + g_debug ("test-drop-invalid:%s: %p", G_STRFUNC, data); + + if (GPOINTER_TO_UINT (data) == 1) + ++global_lagent_cands; + else if (GPOINTER_TO_UINT (data) == 2) + ++global_ragent_cands; + + /* XXX: dear compiler, these are for you: */ + (void)agent; (void)stream_id; (void)component_id; (void)lfoundation; (void)rfoundation; +} + +static void cb_new_candidate(NiceAgent *agent, guint stream_id, guint component_id, + gchar *foundation, gpointer data) +{ + g_debug ("test-drop-invalid:%s: %p", G_STRFUNC, data); + + /* XXX: dear compiler, these are for you: */ + (void)agent; (void)stream_id; (void)data; (void)component_id; (void)foundation; +} + +static void cb_initial_binding_request_received(NiceAgent *agent, guint stream_id, gpointer data) +{ + g_debug ("test-drop-invalid:%s: %p", G_STRFUNC, data); + + if (GPOINTER_TO_UINT (data) == 1) + global_lagent_ibr_received = TRUE; + else if (GPOINTER_TO_UINT (data) == 2) + global_ragent_ibr_received = TRUE; + + if (global_exit_when_ibr_received) { + g_debug ("Received initial binding request. Stopping mailoop"); + g_main_loop_quit (global_mainloop); + } + + /* XXX: dear compiler, these are for you: */ + (void)agent; (void)stream_id; (void)data; +} + +static void set_candidates (NiceAgent *from, guint from_stream, + NiceAgent *to, guint to_stream, guint component) +{ + GSList *cands = NULL; + GSList *peer_cands = NULL; + GSList *item1, *item2; + + cands = nice_agent_get_local_candidates (from, from_stream, component); + peer_cands = nice_agent_get_local_candidates (to, to_stream, component); + + /* + * Core of the test: + * + * Send packets that shoudl be dropped. + */ + + for (item1 = cands; item1; item1 = item1->next) { + NiceCandidate *cand = item1->data; + NiceSocket *nicesock = cand->sockptr; + + g_assert (nicesock); + + for (item2 = peer_cands; item2; item2 = item2->next) { + NiceCandidate *target_cand = item2->data; + + nice_socket_send (nicesock, &target_cand->addr, 12, "123456789AB"); + } + + } + + nice_agent_set_remote_candidates (to, to_stream, component, cands); + + g_slist_free_full (cands, (GDestroyNotify) nice_candidate_free); + g_slist_free_full (peer_cands, (GDestroyNotify) nice_candidate_free); +} + +static void set_credentials (NiceAgent *lagent, guint lstream, + NiceAgent *ragent, guint rstream) +{ + gchar *ufrag = NULL, *password = NULL; + + nice_agent_get_local_credentials(lagent, lstream, &ufrag, &password); + nice_agent_set_remote_credentials (ragent, rstream, ufrag, password); + g_free (ufrag); + g_free (password); + nice_agent_get_local_credentials(ragent, rstream, &ufrag, &password); + nice_agent_set_remote_credentials (lagent, lstream, ufrag, password); + g_free (ufrag); + g_free (password); +} + +static int run_full_test (NiceAgent *lagent, NiceAgent *ragent, NiceAddress *baseaddr, guint ready, guint failed) +{ + guint ls_id, rs_id; + gint ret; + + /* XXX: dear compiler, this is for you */ + (void)baseaddr; + + /* step: initialize variables modified by the callbacks */ + global_components_ready = 0; + global_components_ready_exit = ready; + global_components_failed = 0; + global_components_failed_exit = failed; + global_lagent_gathering_done = FALSE; + global_ragent_gathering_done = FALSE; + global_lagent_ibr_received = + global_ragent_ibr_received = FALSE; + global_lagent_cands = + global_ragent_cands = 0; + + g_object_set (G_OBJECT (lagent), "controlling-mode", TRUE, NULL); + g_object_set (G_OBJECT (ragent), "controlling-mode", FALSE, NULL); + + /* step: add one stream, with RTP+RTCP components, to each agent */ + ls_id = nice_agent_add_stream (lagent, 2); + + rs_id = nice_agent_add_stream (ragent, 2); + g_assert (ls_id > 0); + g_assert (rs_id > 0); + + /* Gather candidates and test nice_agent_set_port_range */ + nice_agent_set_port_range (lagent, ls_id, 1, 10000, 10000); + nice_agent_set_port_range (lagent, ls_id, 2, 10001, 10001); + nice_agent_set_port_range (ragent, rs_id, 1, 12345, 12345); + nice_agent_set_port_range (ragent, rs_id, 2, 10000, 10001); + g_assert (nice_agent_gather_candidates (lagent, ls_id) == TRUE); + g_assert (nice_agent_gather_candidates (ragent, rs_id) == FALSE); + g_assert (nice_agent_get_local_candidates (ragent, rs_id, 1) == NULL); + g_assert (nice_agent_get_local_candidates (ragent, rs_id, 2) == NULL); + nice_agent_set_port_range (ragent, rs_id, 2, 10000, 10002); + g_assert (nice_agent_gather_candidates (ragent, rs_id) == TRUE); + + { + GSList *cands = NULL, *i; + NiceCandidate *cand = NULL; + + cands = nice_agent_get_local_candidates (lagent, ls_id, 1); + g_assert (g_slist_length (cands) == 1); + cand = cands->data; + g_assert (cand->type == NICE_CANDIDATE_TYPE_HOST); + g_assert (nice_address_get_port (&cand->addr) == 10000); + for (i = cands; i; i = i->next) + nice_candidate_free ((NiceCandidate *) i->data); + g_slist_free (cands); + + cands = nice_agent_get_local_candidates (lagent, ls_id, 2); + g_assert (g_slist_length (cands) == 1); + cand = cands->data; + g_assert (cand->type == NICE_CANDIDATE_TYPE_HOST); + g_assert (nice_address_get_port (&cand->addr) == 10001); + for (i = cands; i; i = i->next) + nice_candidate_free ((NiceCandidate *) i->data); + g_slist_free (cands); + + cands = nice_agent_get_local_candidates (ragent, rs_id, 1); + g_assert (g_slist_length (cands) == 1); + cand = cands->data; + g_assert (cand->type == NICE_CANDIDATE_TYPE_HOST); + g_assert (nice_address_get_port (&cand->addr) == 12345); + for (i = cands; i; i = i->next) + nice_candidate_free ((NiceCandidate *) i->data); + g_slist_free (cands); + + cands = nice_agent_get_local_candidates (ragent, rs_id, 2); + g_assert (g_slist_length (cands) == 1); + cand = cands->data; + g_assert (cand->type == NICE_CANDIDATE_TYPE_HOST); + g_assert (nice_address_get_port (&cand->addr) == 10002); + for (i = cands; i; i = i->next) + nice_candidate_free ((NiceCandidate *) i->data); + g_slist_free (cands); + + } + + /* step: attach to mainloop (needed to register the fds) */ + nice_agent_attach_recv (lagent, ls_id, NICE_COMPONENT_TYPE_RTP, + g_main_loop_get_context (global_mainloop), cb_nice_recv, + GUINT_TO_POINTER (1)); + nice_agent_attach_recv (lagent, ls_id, NICE_COMPONENT_TYPE_RTCP, + g_main_loop_get_context (global_mainloop), cb_nice_recv, + GUINT_TO_POINTER (1)); + nice_agent_attach_recv (ragent, rs_id, NICE_COMPONENT_TYPE_RTP, + g_main_loop_get_context (global_mainloop), cb_nice_recv, + GUINT_TO_POINTER (2)); + nice_agent_attach_recv (ragent, rs_id, NICE_COMPONENT_TYPE_RTCP, + g_main_loop_get_context (global_mainloop), cb_nice_recv, + GUINT_TO_POINTER (2)); + + /* step: run mainloop until local candidates are ready + * (see timer_cb() above) */ + if (global_lagent_gathering_done != TRUE || + global_ragent_gathering_done != TRUE) { + g_debug ("test-drop-invalid: Added streams, running mainloop until 'candidate-gathering-done'..."); + g_main_loop_run (global_mainloop); + g_assert (global_lagent_gathering_done == TRUE); + g_assert (global_ragent_gathering_done == TRUE); + } + + set_credentials (lagent, ls_id, ragent, rs_id); + + /* step: pass the remote candidates to agents */ + set_candidates (ragent, rs_id, lagent, ls_id, NICE_COMPONENT_TYPE_RTP); + set_candidates (ragent, rs_id, lagent, ls_id, NICE_COMPONENT_TYPE_RTCP); + set_candidates (lagent, ls_id, ragent, rs_id, NICE_COMPONENT_TYPE_RTP); + set_candidates (lagent, ls_id, ragent, rs_id, NICE_COMPONENT_TYPE_RTCP); + + g_debug ("test-drop-invalid: Set properties, next running mainloop until connectivity checks succeed..."); + + /* step: run the mainloop until connectivity checks succeed + * (see timer_cb() above) */ + g_main_loop_run (global_mainloop); + + /* note: verify that STUN binding requests were sent */ + g_assert (global_lagent_ibr_received == TRUE); + g_assert (global_ragent_ibr_received == TRUE); + + /* note: Send a packet from another address */ + /* These should also be ignored */ + { + NiceCandidate *local_cand = NULL; + NiceCandidate *remote_cand = NULL; + NiceSocket *tmpsock; + + g_assert (nice_agent_get_selected_pair (lagent, ls_id, 1, &local_cand, + &remote_cand)); + g_assert (local_cand); + g_assert (remote_cand); + + tmpsock = nice_udp_bsd_socket_new (NULL); + nice_socket_send (tmpsock, &remote_cand->addr, 4, "ABCD"); + nice_socket_send (tmpsock, &local_cand->addr, 5, "ABCDE"); + nice_socket_free (tmpsock); + } + + /* note: test payload send and receive */ + global_ragent_read = 0; + ret = nice_agent_send (lagent, ls_id, 1, 16, "1234567812345678"); + g_assert (ret != -1); + g_debug ("Sent %d bytes", ret); + g_assert (ret == 16); + while (global_ragent_read != 16) + g_main_context_iteration (NULL, TRUE); + g_assert (global_ragent_read == 16); + + g_debug ("test-drop-invalid: Ran mainloop, removing streams..."); + + /* step: clean up resources and exit */ + + nice_agent_remove_stream (lagent, ls_id); + nice_agent_remove_stream (ragent, rs_id); + + return 0; +} + +int main (void) +{ + NiceAgent *lagent, *ragent; /* agent's L and R */ + NiceAddress baseaddr; + int result; + guint timer_id; + +#ifdef G_OS_WIN32 + WSADATA w; + + WSAStartup(0x0202, &w); +#endif + + global_mainloop = g_main_loop_new (NULL, FALSE); + + /* step: create the agents L and R */ + lagent = nice_agent_new (g_main_loop_get_context (global_mainloop), + NICE_COMPATIBILITY_RFC5245); + ragent = nice_agent_new (g_main_loop_get_context (global_mainloop), + NICE_COMPATIBILITY_RFC5245); + + g_object_set (G_OBJECT (lagent), "ice-tcp", FALSE, NULL); + g_object_set (G_OBJECT (ragent), "ice-tcp", FALSE, NULL); + + + nice_agent_set_software (lagent, "test-drop-invalid, Left Agent"); + nice_agent_set_software (ragent, "test-drop-invalid, Right Agent"); + + /* step: add a timer to catch state changes triggered by signals */ + timer_id = g_timeout_add (30000, timer_cb, NULL); + + /* step: specify which local interface to use */ + if (!nice_address_set_from_string (&baseaddr, "127.0.0.1")) + g_assert_not_reached (); + nice_agent_add_local_address (lagent, &baseaddr); + nice_agent_add_local_address (ragent, &baseaddr); + + g_signal_connect (G_OBJECT (lagent), "candidate-gathering-done", + G_CALLBACK (cb_candidate_gathering_done), GUINT_TO_POINTER(1)); + g_signal_connect (G_OBJECT (ragent), "candidate-gathering-done", + G_CALLBACK (cb_candidate_gathering_done), GUINT_TO_POINTER (2)); + g_signal_connect (G_OBJECT (lagent), "component-state-changed", + G_CALLBACK (cb_component_state_changed), GUINT_TO_POINTER (1)); + g_signal_connect (G_OBJECT (ragent), "component-state-changed", + G_CALLBACK (cb_component_state_changed), GUINT_TO_POINTER (2)); + g_signal_connect (G_OBJECT (lagent), "new-selected-pair", + G_CALLBACK (cb_new_selected_pair), GUINT_TO_POINTER(1)); + g_signal_connect (G_OBJECT (ragent), "new-selected-pair", + G_CALLBACK (cb_new_selected_pair), GUINT_TO_POINTER (2)); + g_signal_connect (G_OBJECT (lagent), "new-candidate", + G_CALLBACK (cb_new_candidate), GUINT_TO_POINTER (1)); + g_signal_connect (G_OBJECT (ragent), "new-candidate", + G_CALLBACK (cb_new_candidate), GUINT_TO_POINTER (2)); + g_signal_connect (G_OBJECT (lagent), "initial-binding-request-received", + G_CALLBACK (cb_initial_binding_request_received), + GUINT_TO_POINTER (1)); + g_signal_connect (G_OBJECT (ragent), "initial-binding-request-received", + G_CALLBACK (cb_initial_binding_request_received), + GUINT_TO_POINTER (2)); + + g_object_set (G_OBJECT (lagent), "upnp", FALSE, NULL); + g_object_set (G_OBJECT (ragent), "upnp", FALSE, NULL); + + + /* step: run test the first time */ + g_debug ("test-drop-invalid: TEST STARTS / running test for the 1st time"); + result = run_full_test (lagent, ragent, &baseaddr, 4 ,0); + priv_print_global_status (); + g_assert (result == 0); + g_assert (global_lagent_state[0] == NICE_COMPONENT_STATE_READY); + g_assert (global_lagent_state[1] == NICE_COMPONENT_STATE_READY); + g_assert (global_ragent_state[0] == NICE_COMPONENT_STATE_READY); + g_assert (global_ragent_state[1] == NICE_COMPONENT_STATE_READY); + /* When using TURN, we get peer reflexive candidates for the host cands + that we removed so we can get another new_selected_pair signal later + depending on timing/racing, we could double (or not) the amount we expected + */ + + /* note: verify that correct number of local candidates were reported */ + g_assert (global_lagent_cands == 2); + g_assert (global_ragent_cands == 2); + + g_object_unref (lagent); + g_object_unref (ragent); + + g_main_loop_unref (global_mainloop); + global_mainloop = NULL; + + g_source_remove (timer_id); +#ifdef G_OS_WIN32 + WSACleanup(); +#endif + return result; +} |