/* GStreamer * Copyright (C) 2012 Wim Taymans * * 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 #include GST_START_TEST (test_pool) { GstRTSPAddressPool *pool; GstRTSPAddress *addr, *addr2, *addr3; GstRTSPAddressPoolResult res; pool = gst_rtsp_address_pool_new (); fail_if (gst_rtsp_address_pool_add_range (pool, "233.252.0.1", "233.252.0.0", 5000, 5010, 1)); fail_if (gst_rtsp_address_pool_add_range (pool, "233.252.0.1", "::1", 5000, 5010, 1)); fail_if (gst_rtsp_address_pool_add_range (pool, "233.252.0.1", "ff02::1", 5000, 5010, 1)); fail_if (gst_rtsp_address_pool_add_range (pool, "233.252.0.1.1", "233.252.0.1", 5000, 5010, 1)); fail_if (gst_rtsp_address_pool_add_range (pool, "233.252.0.1", "233.252.0.1.1", 5000, 5010, 1)); ASSERT_CRITICAL (gst_rtsp_address_pool_add_range (pool, "233.252.0.0", "233.252.0.1", 5010, 5000, 1)); fail_unless (gst_rtsp_address_pool_add_range (pool, "233.252.0.0", "233.252.0.255", 5000, 5010, 1)); fail_unless (gst_rtsp_address_pool_add_range (pool, "233.255.0.0", "233.255.0.0", 5000, 5010, 1)); fail_unless (gst_rtsp_address_pool_add_range (pool, "233.255.0.0", "233.255.0.0", 5020, 5020, 1)); /* should fail, we can't allocate a block of 256 ports */ addr = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_MULTICAST, 256); fail_unless (addr == NULL); addr = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_MULTICAST, 2); fail_unless (addr != NULL); addr2 = gst_rtsp_address_copy (addr); gst_rtsp_address_free (addr2); gst_rtsp_address_free (addr); addr = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_MULTICAST, 4); fail_unless (addr != NULL); /* Will fail because pool is NULL */ ASSERT_CRITICAL (gst_rtsp_address_pool_clear (NULL)); /* will fail because an address is allocated */ ASSERT_CRITICAL (gst_rtsp_address_pool_clear (pool)); gst_rtsp_address_free (addr); gst_rtsp_address_pool_clear (pool); /* start with odd port to make sure we are allocated address * starting with even port */ fail_unless (gst_rtsp_address_pool_add_range (pool, "FF11:DB8::1", "FF11:DB8::1", 5001, 5003, 1)); addr = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_IPV6 | GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_MULTICAST, 2); fail_unless (addr != NULL); fail_unless (addr->port == 5002); fail_unless (!g_ascii_strcasecmp (addr->address, "FF11:DB8::1")); /* Will fail becuse there is only one IPv6 address left */ addr2 = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_IPV6 | GST_RTSP_ADDRESS_FLAG_MULTICAST, 2); fail_unless (addr2 == NULL); /* Will fail because the only IPv6 address left has an odd port */ addr2 = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_IPV6 | GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_MULTICAST, 1); fail_unless (addr2 == NULL); addr2 = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_IPV4 | GST_RTSP_ADDRESS_FLAG_MULTICAST, 1); fail_unless (addr2 == NULL); gst_rtsp_address_free (addr); gst_rtsp_address_pool_clear (pool); fail_unless (gst_rtsp_address_pool_add_range (pool, "233.252.0.0", "233.252.0.255", 5000, 5002, 1)); addr = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_MULTICAST, 2); fail_unless (addr != NULL); fail_unless (addr->port == 5000); fail_unless (!strcmp (addr->address, "233.252.0.0")); addr2 = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_MULTICAST, 2); fail_unless (addr2 != NULL); fail_unless (addr2->port == 5000); fail_unless (!strcmp (addr2->address, "233.252.0.1")); gst_rtsp_address_free (addr); gst_rtsp_address_free (addr2); addr = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_IPV6 | GST_RTSP_ADDRESS_FLAG_MULTICAST, 1); fail_unless (addr == NULL); gst_rtsp_address_pool_clear (pool); fail_unless (gst_rtsp_address_pool_add_range (pool, "233.252.1.1", "233.252.1.1", 5000, 5001, 1)); res = gst_rtsp_address_pool_reserve_address (pool, "233.252.1.1", 5000, 3, 1, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_ERANGE); fail_unless (addr == NULL); res = gst_rtsp_address_pool_reserve_address (pool, "233.252.1.2", 5000, 2, 1, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_ERANGE); fail_unless (addr == NULL); res = gst_rtsp_address_pool_reserve_address (pool, "233.252.1.1", 500, 2, 1, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_ERANGE); fail_unless (addr == NULL); res = gst_rtsp_address_pool_reserve_address (pool, "233.252.1.1", 5000, 2, 2, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_ERANGE); fail_unless (addr == NULL); res = gst_rtsp_address_pool_reserve_address (pool, "2000::1", 5000, 2, 2, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_EINVAL); fail_unless (addr == NULL); res = gst_rtsp_address_pool_reserve_address (pool, "ff02::1", 5000, 2, 2, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_ERANGE); fail_unless (addr == NULL); res = gst_rtsp_address_pool_reserve_address (pool, "1.1", 5000, 2, 2, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_EINVAL); fail_unless (addr == NULL); res = gst_rtsp_address_pool_reserve_address (pool, "233.252.1.1", 5000, 2, 1, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_OK); fail_unless (addr != NULL); fail_unless (addr->port == 5000); fail_unless (!strcmp (addr->address, "233.252.1.1")); res = gst_rtsp_address_pool_reserve_address (pool, "233.252.1.1", 5000, 2, 1, &addr2); fail_unless (res == GST_RTSP_ADDRESS_POOL_ERESERVED); fail_unless (addr2 == NULL); gst_rtsp_address_free (addr); gst_rtsp_address_pool_clear (pool); fail_unless (gst_rtsp_address_pool_add_range (pool, "233.252.1.1", "233.252.1.3", 5000, 5001, 1)); res = gst_rtsp_address_pool_reserve_address (pool, "233.252.1.1", 5000, 2, 1, &addr); fail_unless (addr != NULL); fail_unless (addr->port == 5000); fail_unless (!strcmp (addr->address, "233.252.1.1")); res = gst_rtsp_address_pool_reserve_address (pool, "233.252.1.3", 5000, 2, 1, &addr2); fail_unless (addr2 != NULL); fail_unless (addr2->port == 5000); fail_unless (!strcmp (addr2->address, "233.252.1.3")); addr3 = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_MULTICAST, 2); fail_unless (addr3 != NULL); fail_unless (addr3->port == 5000); fail_unless (!strcmp (addr3->address, "233.252.1.2")); fail_unless (gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_MULTICAST, 2) == NULL); gst_rtsp_address_free (addr); gst_rtsp_address_free (addr2); gst_rtsp_address_free (addr3); gst_rtsp_address_pool_clear (pool); fail_unless (gst_rtsp_address_pool_add_range (pool, "233.252.1.1", "233.252.1.1", 5000, 5001, 1)); fail_if (gst_rtsp_address_pool_has_unicast_addresses (pool)); fail_unless (gst_rtsp_address_pool_add_range (pool, "192.168.1.1", "192.168.1.1", 6000, 6001, 0)); fail_unless (gst_rtsp_address_pool_has_unicast_addresses (pool)); addr = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_MULTICAST, 2); fail_unless (addr != NULL); fail_unless (addr->port == 5000); fail_unless (!strcmp (addr->address, "233.252.1.1")); gst_rtsp_address_free (addr); addr = gst_rtsp_address_pool_acquire_address (pool, GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_UNICAST, 2); fail_unless (addr != NULL); fail_unless (addr->port == 6000); fail_unless (!strcmp (addr->address, "192.168.1.1")); gst_rtsp_address_free (addr); fail_unless (gst_rtsp_address_pool_add_range (pool, GST_RTSP_ADDRESS_POOL_ANY_IPV4, GST_RTSP_ADDRESS_POOL_ANY_IPV4, 5000, 5001, 0)); res = gst_rtsp_address_pool_reserve_address (pool, "192.168.0.1", 5000, 1, 0, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_ERANGE); res = gst_rtsp_address_pool_reserve_address (pool, "0.0.0.0", 5000, 1, 0, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_OK); gst_rtsp_address_free (addr); gst_rtsp_address_pool_clear (pool); /* Error case 2. Using ANY as min address makes it possible to allocate the * same address twice */ fail_unless (gst_rtsp_address_pool_add_range (pool, GST_RTSP_ADDRESS_POOL_ANY_IPV4, "255.255.255.255", 5000, 5001, 0)); res = gst_rtsp_address_pool_reserve_address (pool, "192.168.0.1", 5000, 1, 0, &addr); fail_unless (res == GST_RTSP_ADDRESS_POOL_OK); res = gst_rtsp_address_pool_reserve_address (pool, "192.168.0.1", 5000, 1, 0, &addr2); fail_unless (res == GST_RTSP_ADDRESS_POOL_ERESERVED); gst_rtsp_address_free (addr); gst_rtsp_address_pool_clear (pool); g_object_unref (pool); } GST_END_TEST; static Suite * rtspaddresspool_suite (void) { Suite *s = suite_create ("rtspaddresspool"); TCase *tc = tcase_create ("general"); suite_add_tcase (s, tc); tcase_set_timeout (tc, 20); tcase_add_test (tc, test_pool); return s; } GST_CHECK_MAIN (rtspaddresspool);