|
|
|
|
@ -1,7 +1,7 @@
|
|
|
|
|
/*
|
|
|
|
|
* Asterisk -- An open source telephony toolkit.
|
|
|
|
|
*
|
|
|
|
|
* Copyright (C) 2014, Digium, Inc.
|
|
|
|
|
* Copyright (C) 2014-2016, Digium, Inc.
|
|
|
|
|
*
|
|
|
|
|
* Joshua Colp <jcolp@digium.com>
|
|
|
|
|
*
|
|
|
|
|
@ -16,19 +16,78 @@
|
|
|
|
|
* at the top of the source tree.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
/*** MODULEINFO
|
|
|
|
|
<depend>pjproject</depend>
|
|
|
|
|
<depend>res_pjsip</depend>
|
|
|
|
|
<support_level>core</support_level>
|
|
|
|
|
***/
|
|
|
|
|
|
|
|
|
|
#include "asterisk.h"
|
|
|
|
|
|
|
|
|
|
#include <pjsip.h>
|
|
|
|
|
#include <pjsip_ua.h>
|
|
|
|
|
|
|
|
|
|
#include "asterisk/res_pjsip.h"
|
|
|
|
|
#include "asterisk/module.h"
|
|
|
|
|
#include "asterisk/res_pjsip_session.h"
|
|
|
|
|
#include "include/res_pjsip_private.h"
|
|
|
|
|
|
|
|
|
|
#define MOD_DATA_RESTRICTIONS "restrictions"
|
|
|
|
|
|
|
|
|
|
static pj_status_t multihomed_on_tx_message(pjsip_tx_data *tdata);
|
|
|
|
|
|
|
|
|
|
/*! \brief Outgoing message modification restrictions */
|
|
|
|
|
struct multihomed_message_restrictions {
|
|
|
|
|
/*! \brief Disallow modification of the From domain */
|
|
|
|
|
unsigned int disallow_from_domain_modification;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
static pjsip_module multihomed_module = {
|
|
|
|
|
.name = { "Multihomed Routing", 18 },
|
|
|
|
|
.id = -1,
|
|
|
|
|
.priority = PJSIP_MOD_PRIORITY_TSX_LAYER - 1,
|
|
|
|
|
.on_tx_request = multihomed_on_tx_message,
|
|
|
|
|
.on_tx_response = multihomed_on_tx_message,
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
/*! \brief Helper function to get (or allocate if not already present) restrictions on a message */
|
|
|
|
|
static struct multihomed_message_restrictions *multihomed_get_restrictions(pjsip_tx_data *tdata)
|
|
|
|
|
{
|
|
|
|
|
struct multihomed_message_restrictions *restrictions;
|
|
|
|
|
|
|
|
|
|
restrictions = ast_sip_mod_data_get(tdata->mod_data, multihomed_module.id, MOD_DATA_RESTRICTIONS);
|
|
|
|
|
if (restrictions) {
|
|
|
|
|
return restrictions;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
restrictions = PJ_POOL_ALLOC_T(tdata->pool, struct multihomed_message_restrictions);
|
|
|
|
|
ast_sip_mod_data_set(tdata->pool, tdata->mod_data, multihomed_module.id, MOD_DATA_RESTRICTIONS, restrictions);
|
|
|
|
|
|
|
|
|
|
return restrictions;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*! \brief Callback invoked on non-session outgoing messages */
|
|
|
|
|
static void multihomed_outgoing_message(struct ast_sip_endpoint *endpoint, struct ast_sip_contact *contact, struct pjsip_tx_data *tdata)
|
|
|
|
|
{
|
|
|
|
|
struct multihomed_message_restrictions *restrictions = multihomed_get_restrictions(tdata);
|
|
|
|
|
|
|
|
|
|
restrictions->disallow_from_domain_modification = !ast_strlen_zero(endpoint->fromdomain);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*! \brief PJSIP Supplement for tagging messages with restrictions */
|
|
|
|
|
static struct ast_sip_supplement multihomed_supplement = {
|
|
|
|
|
.priority = AST_SIP_SUPPLEMENT_PRIORITY_FIRST,
|
|
|
|
|
.outgoing_request = multihomed_outgoing_message,
|
|
|
|
|
.outgoing_response = multihomed_outgoing_message,
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
/*! \brief Callback invoked on session outgoing messages */
|
|
|
|
|
static void multihomed_session_outgoing_message(struct ast_sip_session *session, struct pjsip_tx_data *tdata)
|
|
|
|
|
{
|
|
|
|
|
struct multihomed_message_restrictions *restrictions = multihomed_get_restrictions(tdata);
|
|
|
|
|
|
|
|
|
|
restrictions->disallow_from_domain_modification = !ast_strlen_zero(session->endpoint->fromdomain);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*! \brief PJSIP Session Supplement for tagging messages with restrictions */
|
|
|
|
|
static struct ast_sip_session_supplement multihomed_session_supplement = {
|
|
|
|
|
.priority = 1,
|
|
|
|
|
.outgoing_request = multihomed_session_outgoing_message,
|
|
|
|
|
.outgoing_response = multihomed_session_outgoing_message,
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
/*! \brief Helper function which returns a UDP transport bound to the given address and port */
|
|
|
|
|
static pjsip_transport *multihomed_get_udp_transport(pj_str_t *address, int port)
|
|
|
|
|
@ -59,6 +118,21 @@ static pjsip_transport *multihomed_get_udp_transport(pj_str_t *address, int port
|
|
|
|
|
return sip_transport;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*! \brief Helper function which determines if a transport is bound to any */
|
|
|
|
|
static int multihomed_bound_any(pjsip_transport *transport)
|
|
|
|
|
{
|
|
|
|
|
pj_uint32_t loop6[4] = {0, 0, 0, 0};
|
|
|
|
|
|
|
|
|
|
if ((transport->local_addr.addr.sa_family == pj_AF_INET() &&
|
|
|
|
|
transport->local_addr.ipv4.sin_addr.s_addr == PJ_INADDR_ANY) ||
|
|
|
|
|
(transport->local_addr.addr.sa_family == pj_AF_INET6() &&
|
|
|
|
|
!pj_memcmp(&transport->local_addr.ipv6.sin6_addr, loop6, sizeof(loop6)))) {
|
|
|
|
|
return 1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*! \brief Helper function which determines if the address within SDP should be rewritten */
|
|
|
|
|
static int multihomed_rewrite_sdp(struct pjmedia_sdp_session *sdp)
|
|
|
|
|
{
|
|
|
|
|
@ -77,26 +151,13 @@ static int multihomed_rewrite_sdp(struct pjmedia_sdp_session *sdp)
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*! \brief Helper function which determines if a transport is bound to any */
|
|
|
|
|
static int multihomed_bound_any(pjsip_transport *transport)
|
|
|
|
|
{
|
|
|
|
|
pj_uint32_t loop6[4] = {0, 0, 0, 0};
|
|
|
|
|
|
|
|
|
|
if ((transport->local_addr.addr.sa_family == pj_AF_INET() &&
|
|
|
|
|
transport->local_addr.ipv4.sin_addr.s_addr == PJ_INADDR_ANY) ||
|
|
|
|
|
(transport->local_addr.addr.sa_family == pj_AF_INET6() &&
|
|
|
|
|
!pj_memcmp(&transport->local_addr.ipv6.sin6_addr, loop6, sizeof(loop6)))) {
|
|
|
|
|
return 1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static pj_status_t multihomed_on_tx_message(pjsip_tx_data *tdata)
|
|
|
|
|
{
|
|
|
|
|
struct multihomed_message_restrictions *restrictions = ast_sip_mod_data_get(tdata->mod_data, multihomed_module.id, MOD_DATA_RESTRICTIONS);
|
|
|
|
|
pjsip_tpmgr_fla2_param prm;
|
|
|
|
|
pjsip_cseq_hdr *cseq;
|
|
|
|
|
pjsip_via_hdr *via;
|
|
|
|
|
pjsip_fromto_hdr *from;
|
|
|
|
|
|
|
|
|
|
/* Use the destination information to determine what local interface this message will go out on */
|
|
|
|
|
pjsip_tpmgr_fla2_param_default(&prm);
|
|
|
|
|
@ -153,6 +214,13 @@ static pj_status_t multihomed_on_tx_message(pjsip_tx_data *tdata)
|
|
|
|
|
ast_debug(4, "Re-wrote Contact URI host/port to %.*s:%d\n",
|
|
|
|
|
(int)pj_strlen(&uri->host), pj_strbuf(&uri->host), uri->port);
|
|
|
|
|
|
|
|
|
|
if (tdata->tp_info.transport->key.type == PJSIP_TRANSPORT_UDP ||
|
|
|
|
|
tdata->tp_info.transport->key.type == PJSIP_TRANSPORT_UDP6) {
|
|
|
|
|
uri->transport_param.slen = 0;
|
|
|
|
|
} else {
|
|
|
|
|
pj_strdup2(tdata->pool, &uri->transport_param, pjsip_transport_get_type_name(tdata->tp_info.transport->key.type));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pjsip_tx_data_invalidate_msg(tdata);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
@ -164,17 +232,38 @@ static pj_status_t multihomed_on_tx_message(pjsip_tx_data *tdata)
|
|
|
|
|
pjsip_tx_data_invalidate_msg(tdata);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (tdata->msg->type == PJSIP_REQUEST_MSG && (from = pjsip_msg_find_hdr(tdata->msg, PJSIP_H_FROM, NULL)) &&
|
|
|
|
|
(restrictions && !restrictions->disallow_from_domain_modification)) {
|
|
|
|
|
pjsip_name_addr *id_name_addr = (pjsip_name_addr *)from->uri;
|
|
|
|
|
pjsip_sip_uri *uri = pjsip_uri_get_uri(id_name_addr);
|
|
|
|
|
pj_sockaddr ip;
|
|
|
|
|
|
|
|
|
|
if (pj_strcmp2(&uri->host, "localhost") && pj_sockaddr_parse(pj_AF_UNSPEC(), 0, &uri->host, &ip) == PJ_SUCCESS) {
|
|
|
|
|
pj_strassign(&uri->host, &prm.ret_addr);
|
|
|
|
|
pjsip_tx_data_invalidate_msg(tdata);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Update the SDP if it is present */
|
|
|
|
|
if (tdata->msg->body && ast_sip_is_content_type(&tdata->msg->body->content_type, "application", "sdp") &&
|
|
|
|
|
multihomed_rewrite_sdp(tdata->msg->body->data)) {
|
|
|
|
|
struct pjmedia_sdp_session *sdp = tdata->msg->body->data;
|
|
|
|
|
static const pj_str_t STR_IP4 = { "IP4", 3 };
|
|
|
|
|
static const pj_str_t STR_IP6 = { "IP6", 3 };
|
|
|
|
|
pj_str_t STR_IP;
|
|
|
|
|
int stream;
|
|
|
|
|
|
|
|
|
|
STR_IP = tdata->tp_info.transport->key.type & PJSIP_TRANSPORT_IPV6 ? STR_IP6 : STR_IP4;
|
|
|
|
|
|
|
|
|
|
pj_strassign(&sdp->origin.addr, &prm.ret_addr);
|
|
|
|
|
sdp->origin.addr_type = STR_IP;
|
|
|
|
|
pj_strassign(&sdp->conn->addr, &prm.ret_addr);
|
|
|
|
|
sdp->conn->addr_type = STR_IP;
|
|
|
|
|
|
|
|
|
|
for (stream = 0; stream < sdp->media_count; ++stream) {
|
|
|
|
|
if (sdp->media[stream]->conn) {
|
|
|
|
|
pj_strassign(&sdp->media[stream]->conn->addr, &prm.ret_addr);
|
|
|
|
|
sdp->media[stream]->conn->addr_type = STR_IP;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@ -184,42 +273,31 @@ static pj_status_t multihomed_on_tx_message(pjsip_tx_data *tdata)
|
|
|
|
|
return PJ_SUCCESS;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static pjsip_module multihomed_module = {
|
|
|
|
|
.name = { "Multihomed Routing", 18 },
|
|
|
|
|
.id = -1,
|
|
|
|
|
.priority = PJSIP_MOD_PRIORITY_TSX_LAYER - 1,
|
|
|
|
|
.on_tx_request = multihomed_on_tx_message,
|
|
|
|
|
.on_tx_response = multihomed_on_tx_message,
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
static int unload_module(void)
|
|
|
|
|
void ast_res_pjsip_cleanup_message_ip_updater(void)
|
|
|
|
|
{
|
|
|
|
|
ast_sip_unregister_service(&multihomed_module);
|
|
|
|
|
return 0;
|
|
|
|
|
ast_sip_unregister_supplement(&multihomed_supplement);
|
|
|
|
|
ast_sip_session_unregister_supplement(&multihomed_session_supplement);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int load_module(void)
|
|
|
|
|
int ast_res_pjsip_init_message_ip_updater(void)
|
|
|
|
|
{
|
|
|
|
|
char hostname[MAXHOSTNAMELEN] = "";
|
|
|
|
|
|
|
|
|
|
CHECK_PJSIP_MODULE_LOADED();
|
|
|
|
|
if (ast_sip_session_register_supplement(&multihomed_session_supplement)) {
|
|
|
|
|
ast_log(LOG_ERROR, "Could not register multihomed session supplement for outgoing requests\n");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (!gethostname(hostname, sizeof(hostname) - 1)) {
|
|
|
|
|
ast_verb(2, "Performing DNS resolution of local hostname '%s' to get local IPv4 and IPv6 address\n",
|
|
|
|
|
hostname);
|
|
|
|
|
if (ast_sip_register_supplement(&multihomed_supplement)) {
|
|
|
|
|
ast_log(LOG_ERROR, "Could not register multihomed supplement for outgoing requests\n");
|
|
|
|
|
ast_res_pjsip_cleanup_message_ip_updater();
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (ast_sip_register_service(&multihomed_module)) {
|
|
|
|
|
ast_log(LOG_ERROR, "Could not register multihomed module for incoming and outgoing requests\n");
|
|
|
|
|
return AST_MODULE_LOAD_FAILURE;
|
|
|
|
|
ast_res_pjsip_cleanup_message_ip_updater();
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return AST_MODULE_LOAD_SUCCESS;
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "PJSIP Multihomed Routing Support",
|
|
|
|
|
.support_level = AST_MODULE_SUPPORT_CORE,
|
|
|
|
|
.load = load_module,
|
|
|
|
|
.unload = unload_module,
|
|
|
|
|
.load_pri = AST_MODPRI_APP_DEPEND,
|
|
|
|
|
);
|
Joshua Colp
9 years ago
committed by
Gerrit Code Review