| c732d49e |
/*
* Copyright (C) 2017, University of the Basque Country (UPV/EHU)
* Contact for licensing options: <licensing-mcpttclient(at)mcopenplatform(dot)com>
*
* The original file was part of Open Source Doubango Framework
* Copyright (C) 2010-2011 Mamadou Diop.
* Copyright (C) 2012 Doubango Telecom <http://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 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
#define STUN_SERVER_IP "ns313841.ovh.net" // "numb.viagenie.ca" /* stun.ekiga.net, */
#define STUN_USERNAME "bossiel@yahoo.fr"
#define STUN_PASSWORD "tinynet"
#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;
struct tnet_nat_ctx_s *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(!bind_id1 || !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) == 0) {
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) == 0) {
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;
// struct tnet_nat_ctx_s *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);
// ((tnet_nat_context_t*)context)->enable_evenport = 0;
// ((tnet_nat_context_t*)context)->enable_fingerprint = 0;
// ((tnet_nat_context_t*)context)->enable_dontfrag = 0;
// ((tnet_nat_context_t*)context)->enable_integrity = 0;
//
// 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 */ |