diff options
Diffstat (limited to 'tinyNET/test/test_nat.h')
| -rw-r--r-- | tinyNET/test/test_nat.h | 219 |
1 files changed, 219 insertions, 0 deletions
diff --git a/tinyNET/test/test_nat.h b/tinyNET/test/test_nat.h new file mode 100644 index 0000000..ac1e4cb --- /dev/null +++ b/tinyNET/test/test_nat.h @@ -0,0 +1,219 @@ +/*
+* Copyright (C) 2009 Mamadou Diop.
+*
+* Contact: Mamadou Diop <diopmamadou(at)doubango.org>
+*
+* This file is part of Open Source Doubango Framework.
+*
+* DOUBANGO is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* DOUBANGO 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 Lesser General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with DOUBANGO.
+*
+*/
+#ifndef TNET_TEST_NAT_H
+#define TNET_TEST_NAT_H
+
+//stun.ekiga.net
+//#define STUN_SERVER_IP "numb.viagenie.ca"
+#define STUN_SERVER_IP "numb.viagenie.ca"
+#define STUN_USERNAME "xxxxx"
+#define STUN_PASSWORD "xxxxx"
+#define STUN_SERVER_PORT TNET_STUN_TCP_UDP_DEFAULT_PORT
+#define STUN_SERVER_PROTO tnet_socket_type_udp_ipv4
+
+void test_nat_stun()
+{
+ tnet_socket_t *socket1 = 0, *socket2 = 0;
+ tnet_nat_context_handle_t *context = 0;
+
+ tnet_stun_binding_id_t bind_id1, bind_id2;
+
+ char* public_ip1 = 0, *public_ip2 = 0;
+ tnet_port_t public_port1 = 0, public_port2 = 0;
+
+ if(!(socket1 = tnet_socket_create(TNET_SOCKET_HOST_ANY, TNET_SOCKET_PORT_ANY, STUN_SERVER_PROTO))
+ || !(socket2 = tnet_socket_create(TNET_SOCKET_HOST_ANY, TNET_SOCKET_PORT_ANY, STUN_SERVER_PROTO)))
+ {
+ goto bail;
+ }
+
+ context = tnet_nat_context_create(STUN_SERVER_PROTO, STUN_USERNAME, STUN_PASSWORD);
+
+ if(tnet_nat_set_server_address(context, STUN_SERVER_IP))
+ {
+ TSK_DEBUG_ERROR("Failed to set STUN/TURN address.");
+ goto bail;
+ }
+
+ /* == BIND
+ */
+ bind_id1 = tnet_nat_stun_bind(context, socket1->fd);
+ bind_id2 = tnet_nat_stun_bind(context, socket2->fd);
+
+ if(!TNET_STUN_IS_VALID_BINDING_ID(bind_id1) ||!TNET_STUN_IS_VALID_BINDING_ID(bind_id2))
+ {
+ TSK_DEBUG_ERROR("Failed to get public IP/port using stun");
+ goto bail;
+ }
+
+ if(!tnet_nat_stun_get_reflexive_address(context, bind_id1, &public_ip1, &public_port1))
+ {
+ TSK_DEBUG_INFO("Public IP1/Port1 ==> %s:%u", public_ip1, public_port1);
+ }
+
+ if(!tnet_nat_stun_get_reflexive_address(context, bind_id2, &public_ip2, &public_port2))
+ {
+ TSK_DEBUG_INFO("Public IP2/Port2 ==> %s:%u", public_ip2, public_port2);
+ }
+
+ /* == UNBIND
+ */
+ tnet_nat_stun_unbind(context, bind_id1);
+ tnet_nat_stun_unbind(context, bind_id2);
+
+bail:
+ TSK_OBJECT_SAFE_FREE(socket1);
+ TSK_OBJECT_SAFE_FREE(socket2);
+
+ TSK_FREE(public_ip1);
+ TSK_FREE(public_ip1);
+
+ TSK_OBJECT_SAFE_FREE(context);
+}
+
+void test_nat_turn()
+{
+ tnet_socket_t *socket1 = 0, *socket2 = 0;
+ tnet_nat_context_handle_t *context = 0;
+ tnet_turn_allocation_id_t alloc_id1, alloc_id2;
+
+ char* public_ip1 = 0, *public_ip2 = 0;
+ tnet_port_t public_port1 = 0, public_port2 = 0;
+
+ tnet_turn_channel_binding_id_t channel_id;
+
+ int ret;
+
+ if(!(socket1 = tnet_socket_create(TNET_SOCKET_HOST_ANY, TNET_SOCKET_PORT_ANY, STUN_SERVER_PROTO))
+ || !(socket2 = tnet_socket_create(TNET_SOCKET_HOST_ANY, TNET_SOCKET_PORT_ANY, STUN_SERVER_PROTO)))
+ {
+ goto bail;
+ }
+
+ context = tnet_nat_context_create(STUN_SERVER_PROTO, STUN_USERNAME, STUN_PASSWORD);
+
+ if(tnet_nat_set_server_address(context, STUN_SERVER_IP))
+ {
+ TSK_DEBUG_ERROR("Failed to set STUN/TURN address.");
+ goto bail;
+ }
+
+ /* == ALLOC
+ */
+ alloc_id1 = tnet_nat_turn_allocate(context, socket1->fd);
+ alloc_id2 = tnet_nat_turn_allocate(context, socket2->fd);
+ if(!TNET_TURN_IS_VALID_ALLOCATION_ID(alloc_id1) || !TNET_TURN_IS_VALID_ALLOCATION_ID(alloc_id2))
+ {
+ TSK_DEBUG_ERROR("TURN allocation failed.");
+ goto bail;
+ }
+ else
+ {
+ TSK_DEBUG_INFO("TURN allocation succeeded and id1=%llu and id2=%llu", alloc_id1, alloc_id2);
+ }
+
+ tsk_thread_sleep(2000);
+
+ /* == RETRIEVE STUN SERVER REFLEXIVE ADDRESSES
+ */
+ if(!tnet_nat_turn_get_reflexive_address(context, alloc_id1, &public_ip1, &public_port1))
+ {
+ TSK_DEBUG_INFO("Public IP1/Port1 ==> %s:%u", public_ip1, public_port1);
+ }
+
+ if(!tnet_nat_turn_get_reflexive_address(context, alloc_id2, &public_ip2, &public_port2))
+ {
+ TSK_DEBUG_INFO("Public IP2/Port2 ==> %s:%u", public_ip2, public_port2);
+ }
+
+ /* == CREATE PERMISSION
+ */
+ //tnet_nat_turn_add_permission(context, alloc_id1, "192.168.0.11", 300);
+
+ /* == CHANNEL BINDING
+ */
+ {
+ /* Try to bind (channel binding) to the socket1 to socket2 */
+ struct sockaddr_storage peer;
+ if((ret = tnet_sockaddr_init(public_ip2, public_port2, STUN_SERVER_PROTO, &peer)))
+ {
+ TSK_DEBUG_ERROR("Failed to init peer with error code %d.", ret);
+ }
+ else
+ {
+ channel_id = tnet_nat_turn_channel_bind(context, alloc_id1,&peer);
+ if(TNET_TURN_IS_VALID_CHANNEL_BINDING_ID(channel_id))
+ {
+ TSK_DEBUG_INFO("TURN channel binding succeeded.");
+
+ /* Try to send data using the newly create channel */
+ if(tnet_nat_turn_channel_senddata(context, channel_id, "Doubango", strlen("Doubango")))
+ {
+ TSK_DEBUG_ERROR("Failed to send data using channel id [%u].", channel_id);
+ }
+ else
+ {
+ TSK_DEBUG_INFO("Data successfuly sent using channel if[%u].", channel_id);
+ }
+ }
+ else
+ {
+ TSK_DEBUG_ERROR("TURN channel binding failed.");
+ }
+ }
+ }
+
+ tsk_thread_sleep(2000);
+
+ /* == UNALLOC
+ */
+ if((ret = tnet_nat_turn_unallocate(context, alloc_id1)) || (ret = tnet_nat_turn_unallocate(context, alloc_id2)))
+ {
+ TSK_DEBUG_ERROR("TURN unallocation failed with error code: %d.", ret);
+ goto bail;
+ }
+ else
+ {
+ TSK_DEBUG_INFO("TURN unallocation succeeded.");
+ }
+
+bail:
+ TSK_OBJECT_SAFE_FREE(socket1);
+ TSK_OBJECT_SAFE_FREE(socket2);
+
+ TSK_FREE(public_ip1);
+ TSK_FREE(public_ip1);
+
+ TSK_OBJECT_SAFE_FREE(context);
+}
+
+
+void test_nat()
+{
+ test_nat_stun();
+ //test_nat_turn();
+ //tsk_thread_sleep(1000);
+}
+
+
+
+#endif /* TNET_TEST_NAT_H */
|
