239 lines
7.8 KiB
C
Executable File
239 lines
7.8 KiB
C
Executable File
/* Copyright (C) 2014 Mamadou DIOP.
|
|
*
|
|
* 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_ICE_H
|
|
#define TNET_TEST_ICE_H
|
|
|
|
#undef kStunUsrName
|
|
#undef kStunPwd
|
|
|
|
#define kStunUsrName "bossiel@yahoo.fr"
|
|
#define kStunPwd "tinynet"
|
|
#define kStunServerIP "ns313841.ovh.net" /*"numb.viagenie.ca"*/ /*stun.ekiga.net*/
|
|
|
|
#define kSkipHosts 1
|
|
#define kSkipReflexives 1 // server reflexive: STUN
|
|
#define kSkipPeers 1 // peer reflexive
|
|
#define kSkipRelays 0 // relays: TURN
|
|
|
|
#define kTurnTrue 1
|
|
#define kStunTrue 1
|
|
#define kTurnFalse 0
|
|
#define kStunFalse 0
|
|
|
|
static const tsk_bool_t use_rtcp = tsk_false;
|
|
|
|
static struct tnet_ice_ctx_s *p_ice_ctx1 = tsk_null;
|
|
static struct tnet_ice_ctx_s *p_ice_ctx2 = tsk_null;
|
|
|
|
static void test_ice_print_local_candidates(const struct tnet_ice_ctx_s *pc_ctx)
|
|
{
|
|
tsk_size_t index = 0;
|
|
const tnet_ice_candidate_t* candidate;
|
|
char* p_str = tsk_null;
|
|
|
|
while ((candidate = tnet_ice_ctx_get_local_candidate_at(pc_ctx, index++))) {
|
|
tsk_strcat_2(&p_str, "%s\r\n", tnet_ice_candidate_tostring((tnet_ice_candidate_t*)candidate));
|
|
}
|
|
TSK_DEBUG_INFO("ICE LOCAL CANDIDATES:\n%s", p_str);
|
|
TSK_FREE(p_str);
|
|
}
|
|
|
|
static char* test_ice_get_local_candidates(const struct tnet_ice_ctx_s *pc_ctx)
|
|
{
|
|
tsk_size_t index = 0;
|
|
const tnet_ice_candidate_t* candidate;
|
|
char* p_str = tsk_null;
|
|
|
|
while ((candidate = tnet_ice_ctx_get_local_candidate_at(pc_ctx, index++))) {
|
|
if (kSkipHosts && candidate->type_e == tnet_ice_cand_type_host) {
|
|
continue;
|
|
}
|
|
if (kSkipReflexives && candidate->type_e == tnet_ice_cand_type_srflx) {
|
|
continue;
|
|
}
|
|
if (kSkipRelays && candidate->type_e == tnet_ice_cand_type_relay) {
|
|
continue;
|
|
}
|
|
if (kSkipPeers && candidate->type_e == tnet_ice_cand_type_prflx) {
|
|
continue;
|
|
}
|
|
tsk_strcat_2(&p_str, "%s\r\n", tnet_ice_candidate_tostring((tnet_ice_candidate_t*)candidate));
|
|
}
|
|
return p_str;
|
|
}
|
|
|
|
static int test_ice_rtp_callback(const void* callback_data, const uint8_t* data_ptr, tsk_size_t data_size, tnet_fd_t local_fd, const struct sockaddr_storage* remote_addr)
|
|
{
|
|
struct tnet_ice_ctx_s *p_ice_ctx = (struct tnet_ice_ctx_s *)callback_data;
|
|
|
|
TSK_DEBUG_INFO("\n\nICE rtp callback (incoming data): %.*s\n\n", data_size, data_ptr);
|
|
|
|
#if 0
|
|
tnet_ice_ctx_send_turn_rtp(p_ice_ctx, data_ptr, data_size);
|
|
#endif
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int test_ice_state_callback(const tnet_ice_event_t *e)
|
|
{
|
|
struct tnet_ice_ctx_s *p_ice_ctx = (struct tnet_ice_ctx_s *)e->ctx;
|
|
int ret = 0;
|
|
|
|
TSK_DEBUG_INFO("ICE state callback: %s", e->phrase);
|
|
|
|
switch(e->type)
|
|
{
|
|
case tnet_ice_event_type_gathering_completed:
|
|
{
|
|
test_ice_print_local_candidates(p_ice_ctx);
|
|
if (p_ice_ctx == p_ice_ctx1) {
|
|
if ((ret = tnet_ice_ctx_start(p_ice_ctx2))) {
|
|
goto bail;
|
|
}
|
|
}
|
|
else {
|
|
const tnet_ice_candidate_t* candidate;
|
|
char* p_cand;
|
|
|
|
p_cand = test_ice_get_local_candidates(p_ice_ctx2);
|
|
candidate = tnet_ice_ctx_get_local_candidate_first(p_ice_ctx2);
|
|
ret = tnet_ice_ctx_set_remote_candidates(p_ice_ctx1, p_cand, candidate->ufrag, candidate->pwd, tsk_true, tsk_false);
|
|
if (ret == 0) {
|
|
TSK_FREE(p_cand);
|
|
p_cand = test_ice_get_local_candidates(p_ice_ctx1);
|
|
candidate = tnet_ice_ctx_get_local_candidate_first(p_ice_ctx1);
|
|
ret = tnet_ice_ctx_set_remote_candidates(p_ice_ctx2, p_cand, candidate->ufrag, candidate->pwd, tsk_false, tsk_false);
|
|
TSK_FREE(p_cand);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
case tnet_ice_event_type_conncheck_succeed:
|
|
{
|
|
const char kTurnData[] = "TURN data to send for testing";
|
|
const tnet_ice_candidate_t *candidate_offer, *candidate_answer_src, *candidate_answer_dest;
|
|
|
|
// === RTP === //
|
|
ret = tnet_ice_ctx_get_nominated_symetric_candidates(p_ice_ctx, TNET_ICE_CANDIDATE_COMPID_RTP, &candidate_offer, &candidate_answer_src, &candidate_answer_dest);
|
|
if (ret == 0) {
|
|
TSK_DEBUG_INFO("Nominated candidate(RTP): Offer=[[%s]], AnswerSrc=[[%s]], AnswerDest=[[%s]]",
|
|
tnet_ice_candidate_tostring((tnet_ice_candidate_t*)candidate_offer),
|
|
tnet_ice_candidate_tostring((tnet_ice_candidate_t*)candidate_answer_src),
|
|
tnet_ice_candidate_tostring((tnet_ice_candidate_t*)candidate_answer_dest));
|
|
|
|
if (tnet_ice_ctx_is_turn_rtp_active(p_ice_ctx)) {
|
|
tnet_ice_ctx_send_turn_rtp(p_ice_ctx, kTurnData, sizeof(kTurnData));
|
|
}
|
|
}
|
|
// === RTCP === //
|
|
if (use_rtcp) {
|
|
ret = tnet_ice_ctx_get_nominated_symetric_candidates(p_ice_ctx, TNET_ICE_CANDIDATE_COMPID_RTCP, &candidate_offer, &candidate_answer_src, &candidate_answer_dest);
|
|
if (ret == 0) {
|
|
TSK_DEBUG_INFO("Nominated candidate(RTCP): Offer=[[%s]], AnswerSrc=[[%s]], AnswerDest=[[%s]]",
|
|
tnet_ice_candidate_tostring((tnet_ice_candidate_t*)candidate_offer),
|
|
tnet_ice_candidate_tostring((tnet_ice_candidate_t*)candidate_answer_src),
|
|
tnet_ice_candidate_tostring((tnet_ice_candidate_t*)candidate_answer_dest));
|
|
if (tnet_ice_ctx_is_turn_rtcp_active(p_ice_ctx)) {
|
|
tnet_ice_ctx_send_turn_rtcp(p_ice_ctx, kTurnData, sizeof(kTurnData));
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
bail:
|
|
return ret;
|
|
}
|
|
|
|
void test_ice()
|
|
{
|
|
int ret;
|
|
static const tsk_bool_t use_ipv6 = tsk_false;
|
|
static const tsk_bool_t use_ice_jingle = tsk_false;
|
|
static const tsk_bool_t use_video = tsk_false;
|
|
|
|
if (!(p_ice_ctx1 = tnet_ice_ctx_create(use_ice_jingle, use_ipv6, use_rtcp, use_video, test_ice_state_callback, tsk_null))) {
|
|
goto bail;
|
|
}
|
|
if (!(p_ice_ctx2 = tnet_ice_ctx_create(use_ice_jingle, use_ipv6, use_rtcp, use_video, test_ice_state_callback, tsk_null))) {
|
|
goto bail;
|
|
}
|
|
if ((ret = tnet_ice_ctx_set_turn_enabled(p_ice_ctx1, 1))) {
|
|
goto bail;
|
|
}
|
|
if ((ret = tnet_ice_ctx_set_turn_enabled(p_ice_ctx2, 1))) {
|
|
goto bail;
|
|
}
|
|
if ((ret = tnet_ice_ctx_set_stun_enabled(p_ice_ctx1, 1))) {
|
|
goto bail;
|
|
}
|
|
if ((ret = tnet_ice_ctx_set_stun_enabled(p_ice_ctx2, 1))) {
|
|
goto bail;
|
|
}
|
|
if ((ret = tnet_ice_ctx_set_userdata(p_ice_ctx1, p_ice_ctx1))) {
|
|
goto bail;
|
|
}
|
|
if ((ret = tnet_ice_ctx_set_userdata(p_ice_ctx2, p_ice_ctx2))) {
|
|
goto bail;
|
|
}
|
|
if ((ret = tnet_ice_ctx_rtp_callback(p_ice_ctx1, test_ice_rtp_callback, p_ice_ctx1))) {
|
|
goto bail;
|
|
}
|
|
if ((ret = tnet_ice_ctx_rtp_callback(p_ice_ctx2, test_ice_rtp_callback, p_ice_ctx2))) {
|
|
goto bail;
|
|
}
|
|
#if 0 //@deprecated
|
|
if ((ret = tnet_ice_ctx_set_stun(p_ice_ctx1, kStunServerIP, 3478, kStunSoftware, kStunUsrName, kStunPwd))) {
|
|
goto bail;
|
|
}
|
|
if ((ret = tnet_ice_ctx_set_stun(p_ice_ctx2, kStunServerIP, 3478, kStunSoftware, kStunUsrName, kStunPwd))) {
|
|
goto bail;
|
|
}
|
|
#else
|
|
tnet_ice_ctx_add_server(p_ice_ctx1, "udp", kStunServerIP, 3478, kTurnFalse, kStunTrue, kStunUsrName, kStunPwd); // STUN-UDP
|
|
tnet_ice_ctx_add_server(p_ice_ctx1, "tcp", kStunServerIP, 3478, kTurnTrue, kStunFalse, kStunUsrName, kStunPwd); // TURN-TCP
|
|
tnet_ice_ctx_add_server(p_ice_ctx2, "udp", kStunServerIP, 3478, kTurnFalse, kStunTrue, kStunUsrName, kStunPwd); // STUN-UDP
|
|
tnet_ice_ctx_add_server(p_ice_ctx2, "tcp", kStunServerIP, 3478, kTurnTrue, kStunFalse, kStunUsrName, kStunPwd); // TURN-TCP
|
|
#endif
|
|
if ((ret = tnet_ice_ctx_start(p_ice_ctx1))) {
|
|
goto bail;
|
|
}
|
|
// start ctx2 when we finish gathering ctx1's candidates
|
|
//if ((ret = tnet_ice_ctx_start(p_ice_ctx2))) {
|
|
// goto bail;
|
|
//}
|
|
|
|
getchar();
|
|
|
|
// ret = tnet_ice_ctx_stop(p_ice_ctx1);
|
|
// ret = tnet_ice_ctx_stop(p_ice_ctx2);
|
|
|
|
bail:
|
|
TSK_OBJECT_SAFE_FREE(p_ice_ctx1);
|
|
TSK_OBJECT_SAFE_FREE(p_ice_ctx2);
|
|
}
|
|
|
|
|
|
#endif /* TNET_TEST_ICE_H */
|
|
|