doubango/tinyNET/src/ice/tnet_ice_utils.c
c732d49e
 #if HAVE_CRT
 #define _CRTDBG_MAP_ALLOC 
 #include <stdlib.h> 
 #include <crtdbg.h>
 #endif //HAVE_CRT
 /*
 * 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.
 *
 */
 
 
 #include "tnet_ice_utils.h"
 #include "tnet_ice_candidate.h"
 #include "tnet_socket.h"
 
 #include "tsk_time.h"
 #include "tsk_string.h"
 #include "tsk_debug.h"
 
 #include <stdlib.h>
 
 static const char ice_chars[] = {'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'k', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', 
 									'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'K', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
 									'0','1', '2', '3', '4', '5', '6', '7', '8', '9'}; // /!\do not add '/' and '+' because of WebRTC password
 static const tsk_size_t ice_chars_count = sizeof(ice_chars);
 
 uint32_t tnet_ice_utils_get_priority(tnet_ice_cand_type_t type, uint16_t local_pref, tsk_bool_t is_rtp)
 {
 	uint32_t pref;
 	switch(type){
 		case tnet_ice_cand_type_host: pref = TNET_ICE_CANDIDATE_PREF_HOST; break;
 		case tnet_ice_cand_type_srflx: pref = TNET_ICE_CANDIDATE_PREF_SRFLX; break;
 		case tnet_ice_cand_type_prflx: pref = TNET_ICE_CANDIDATE_PREF_PRFLX; break;
 		case tnet_ice_cand_type_relay: default: pref = TNET_ICE_CANDIDATE_PREF_RELAY; break;
 	}
 	return (pref << 24) +
 		(local_pref << 8) +
 		((256 - (is_rtp ? TNET_ICE_CANDIDATE_COMPID_RTP : TNET_ICE_CANDIDATE_COMPID_RTCP)) << 0);
 }
 
 
 int tnet_ice_utils_compute_foundation(char* foundation, tsk_size_t size)
 {
 	
 	tsk_size_t i;
 
 	if(!foundation || !size){
 		TSK_DEBUG_ERROR("Invalid argument");
 		return -1;
 	}
 	for(
 		i = 0; i < size; ++i){
 		foundation[i] = ice_chars[(rand() ^ rand()) % ice_chars_count];
 	}
 
 	return 0;
 }
 
 int tnet_ice_utils_create_sockets(tnet_socket_type_t socket_type, const char* local_ip, tnet_socket_t** socket_rtp, tnet_socket_t** socket_rtcp)
 {
 	tsk_bool_t look4_rtp = (socket_rtp != tsk_null);
 	tsk_bool_t look4_rtcp = (socket_rtcp != tsk_null);
 	uint8_t retry_count = 10;
 	tnet_port_t local_port;
 	static const uint64_t port_range_start = 1024;
 	static const uint64_t port_range_stop = (65535 - 1/* to be sure rtcp port will be valid */);
 	static uint64_t counter = 0;
 
 	/* Creates local rtp and rtcp sockets */
 	while(retry_count--){
 		if(look4_rtp && look4_rtcp){
 			tnet_socket_t* socket_fake = tnet_socket_create(local_ip, TNET_SOCKET_PORT_ANY, socket_type);
 			if(!socket_fake){
 				continue;
 			}
 			if(!(socket_fake->port & 0x01)){ // even number ?
 				*socket_rtp = socket_fake;
 			}
 			else{
 				*socket_rtcp = socket_fake;
 			}
 			local_port = (socket_fake->port & ~1);
 		}
 		else{
 			local_port = (tnet_port_t)((((tsk_time_epoch() + rand() ) ^ ++counter) % (port_range_stop - port_range_start)) + port_range_start);
 			local_port = (local_port & 0xFFFE); /* turn to even number */
 		}
 		
 		/* beacuse failure will cause errors in the log, print a message to alert that there is
 		* nothing to worry about */
 		TSK_DEBUG_INFO("RTP/RTCP manager[Begin]: Trying to bind to random ports [%s:%d]", local_ip, local_port);
 		
 		if(look4_rtp){
 			if(!*socket_rtp && !(*socket_rtp = tnet_socket_create(local_ip, local_port, socket_type))){
 				TSK_DEBUG_INFO("Failed to bind to %d", local_port);
 				continue;
 			}
 		}
 
 		if(look4_rtcp){
 			if(!*socket_rtcp && !(*socket_rtcp = tnet_socket_create(local_ip, (local_port + 1), socket_type))){
 				TSK_DEBUG_INFO("Failed to bind to %d", (local_port + 1));
 				if(look4_rtp){
 					TSK_OBJECT_SAFE_FREE((*socket_rtp));
 				}
 				continue;
 			}
 		}
 
 		TSK_DEBUG_INFO("RTP/RTCP manager[End]: Trying to bind to random ports");
 		return 0;
 	}
 
 	TSK_DEBUG_ERROR("Failed to bind sockets");
 	return -1;
 }
 
 int tnet_ice_utils_set_ufrag(char** ufrag)
 {
 	if(ufrag){
 		char tmp[16]; int i;
 		for(i = 0; i < (sizeof(tmp)/sizeof(tmp[0])) - 1; ++i){
 			tmp[i] = ice_chars[(rand() ^ rand()) % ice_chars_count];
 		}
 		tmp[i] = '\0';
 		tsk_strupdate(ufrag, tmp);
 		return 0;
 	}
 	else{
 		TSK_DEBUG_ERROR("Invalid parameter");
 		return -1;
 	}
 }
 
 int tnet_ice_utils_set_pwd(char** pwd)
 {
 	if(pwd){
 		char tmp[23]; int i;
 		for(i = 0; i < (sizeof(tmp)/sizeof(tmp[0])) - 1; ++i){
 			tmp[i] = ice_chars[(rand() ^ rand()) % ice_chars_count];
 		}
 		tmp[i] = '\0';
 		tsk_strupdate(pwd, tmp);
 		return 0;
 	}
 	else{
 		TSK_DEBUG_ERROR("Invalid parameter");
 		return -1;
 	}
 }