In all non-pbx modules, AST_MODULE_LOAD_FAILURE has been changed to AST_MODULE_LOAD_DECLINE. This prevents asterisk from exiting if a module can't be loaded. If the user wishes to retain the FAILURE behavior for a specific module, they can use the "require" or "preload-require" keyword in modules.conf. A new API was added to logger: ast_is_logger_initialized(). This allows asterisk.c/check_init() to print to the error log once the logger subsystem is ready instead of just to stdout. If something does fail before the logger is initialized, we now print to stderr instead of stdout. Change-Id: I5f4b50623d9b5a6cb7c5624a8c5c1274c13b2b25
382 lines
12 KiB
C
382 lines
12 KiB
C
/*
|
|
* Asterisk -- An open source telephony toolkit.
|
|
*
|
|
* Copyright (C) 2013, Digium, Inc.
|
|
*
|
|
* Joshua Colp <jcolp@digium.com>
|
|
*
|
|
* See http://www.asterisk.org for more information about
|
|
* the Asterisk project. Please do not directly contact
|
|
* any of the maintainers of this project for assistance;
|
|
* the project provides a web site, mailing lists and IRC
|
|
* channels for your use.
|
|
*
|
|
* This program is free software, distributed under the terms of
|
|
* the GNU General Public License Version 2. See the LICENSE file
|
|
* 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/res_pjsip_session.h"
|
|
#include "asterisk/module.h"
|
|
#include "asterisk/acl.h"
|
|
|
|
static void rewrite_uri(pjsip_rx_data *rdata, pjsip_sip_uri *uri)
|
|
{
|
|
pj_cstr(&uri->host, rdata->pkt_info.src_name);
|
|
if (!strcasecmp("WSS", rdata->tp_info.transport->type_name)) {
|
|
/* WSS is special, we don't want to overwrite the URI at all as it needs to be ws */
|
|
} else if (strcasecmp("udp", rdata->tp_info.transport->type_name)) {
|
|
uri->transport_param = pj_str(rdata->tp_info.transport->type_name);
|
|
} else {
|
|
uri->transport_param.slen = 0;
|
|
}
|
|
uri->port = rdata->pkt_info.src_port;
|
|
}
|
|
|
|
static int rewrite_route_set(pjsip_rx_data *rdata, pjsip_dialog *dlg)
|
|
{
|
|
pjsip_rr_hdr *rr = NULL;
|
|
pjsip_sip_uri *uri;
|
|
|
|
if (rdata->msg_info.msg->type == PJSIP_RESPONSE_MSG) {
|
|
pjsip_hdr *iter;
|
|
for (iter = rdata->msg_info.msg->hdr.prev; iter != &rdata->msg_info.msg->hdr; iter = iter->prev) {
|
|
if (iter->type == PJSIP_H_RECORD_ROUTE) {
|
|
rr = (pjsip_rr_hdr *)iter;
|
|
break;
|
|
}
|
|
}
|
|
} else if (pjsip_method_cmp(&rdata->msg_info.msg->line.req.method, &pjsip_register_method)) {
|
|
rr = pjsip_msg_find_hdr(rdata->msg_info.msg, PJSIP_H_RECORD_ROUTE, NULL);
|
|
}
|
|
|
|
if (rr) {
|
|
uri = pjsip_uri_get_uri(&rr->name_addr);
|
|
rewrite_uri(rdata, uri);
|
|
if (dlg && !pj_list_empty(&dlg->route_set) && !dlg->route_set_frozen) {
|
|
pjsip_routing_hdr *route = dlg->route_set.next;
|
|
uri = pjsip_uri_get_uri(&route->name_addr);
|
|
rewrite_uri(rdata, uri);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
static int rewrite_contact(pjsip_rx_data *rdata, pjsip_dialog *dlg)
|
|
{
|
|
pjsip_contact_hdr *contact;
|
|
|
|
contact = pjsip_msg_find_hdr(rdata->msg_info.msg, PJSIP_H_CONTACT, NULL);
|
|
if (contact && !contact->star && (PJSIP_URI_SCHEME_IS_SIP(contact->uri) || PJSIP_URI_SCHEME_IS_SIPS(contact->uri))) {
|
|
pjsip_sip_uri *uri = pjsip_uri_get_uri(contact->uri);
|
|
|
|
rewrite_uri(rdata, uri);
|
|
|
|
if (dlg && pj_list_empty(&dlg->route_set) && (!dlg->remote.contact
|
|
|| pjsip_uri_cmp(PJSIP_URI_IN_REQ_URI, dlg->remote.contact->uri, contact->uri))) {
|
|
dlg->remote.contact = (pjsip_contact_hdr*)pjsip_hdr_clone(dlg->pool, contact);
|
|
dlg->target = dlg->remote.contact->uri;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
static pj_bool_t handle_rx_message(struct ast_sip_endpoint *endpoint, pjsip_rx_data *rdata)
|
|
{
|
|
pjsip_dialog *dlg = pjsip_rdata_get_dlg(rdata);
|
|
|
|
if (!endpoint) {
|
|
return PJ_FALSE;
|
|
}
|
|
|
|
if (endpoint->nat.rewrite_contact) {
|
|
/* rewrite_contact is intended to ensure we send requests/responses to
|
|
* a routeable address when NAT is involved. The URI that dictates where
|
|
* we send requests/responses can be determined either by Record-Route
|
|
* headers or by the Contact header if no Record-Route headers are present.
|
|
* We therefore will attempt to rewrite a Record-Route header first, and if
|
|
* none are present, we fall back to rewriting the Contact header instead.
|
|
*/
|
|
if (rewrite_route_set(rdata, dlg)) {
|
|
rewrite_contact(rdata, dlg);
|
|
}
|
|
}
|
|
|
|
if (endpoint->nat.force_rport) {
|
|
rdata->msg_info.via->rport_param = rdata->pkt_info.src_port;
|
|
}
|
|
|
|
return PJ_FALSE;
|
|
}
|
|
|
|
static pj_bool_t nat_on_rx_message(pjsip_rx_data *rdata)
|
|
{
|
|
pj_bool_t res;
|
|
struct ast_sip_endpoint *endpoint;
|
|
|
|
endpoint = ast_pjsip_rdata_get_endpoint(rdata);
|
|
res = handle_rx_message(endpoint, rdata);
|
|
ao2_cleanup(endpoint);
|
|
return res;
|
|
}
|
|
|
|
/*! \brief Structure which contains information about a transport */
|
|
struct request_transport_details {
|
|
/*! \brief Type of transport */
|
|
enum ast_transport type;
|
|
/*! \brief Potential pointer to the transport itself, if UDP */
|
|
pjsip_transport *transport;
|
|
/*! \brief Potential pointer to the transport factory itself, if TCP/TLS */
|
|
pjsip_tpfactory *factory;
|
|
/*! \brief Local address for transport */
|
|
pj_str_t local_address;
|
|
/*! \brief Local port for transport */
|
|
int local_port;
|
|
};
|
|
|
|
/*! \brief Callback function for finding the transport the request is going out on */
|
|
static int find_transport_state_in_use(void *obj, void *arg, int flags)
|
|
{
|
|
struct ast_sip_transport_state *transport_state = obj;
|
|
struct request_transport_details *details = arg;
|
|
|
|
/* If an explicit transport or factory matches then this is what is in use, if we are unavailable
|
|
* to compare based on that we make sure that the type is the same and the source IP address/port are the same
|
|
*/
|
|
if (transport_state && ((details->transport && details->transport == transport_state->transport) ||
|
|
(details->factory && details->factory == transport_state->factory) ||
|
|
((details->type == transport_state->type) && (transport_state->factory) &&
|
|
!pj_strcmp(&transport_state->factory->addr_name.host, &details->local_address) &&
|
|
transport_state->factory->addr_name.port == details->local_port))) {
|
|
return CMP_MATCH | CMP_STOP;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*! \brief Helper function which returns the SIP URI of a Contact header */
|
|
static pjsip_sip_uri *nat_get_contact_sip_uri(pjsip_tx_data *tdata)
|
|
{
|
|
pjsip_contact_hdr *contact = pjsip_msg_find_hdr(tdata->msg, PJSIP_H_CONTACT, NULL);
|
|
|
|
if (!contact || (!PJSIP_URI_SCHEME_IS_SIP(contact->uri) && !PJSIP_URI_SCHEME_IS_SIPS(contact->uri))) {
|
|
return NULL;
|
|
}
|
|
|
|
return pjsip_uri_get_uri(contact->uri);
|
|
}
|
|
|
|
/*! \brief Structure which contains hook details */
|
|
struct nat_hook_details {
|
|
/*! \brief Outgoing message itself */
|
|
pjsip_tx_data *tdata;
|
|
/*! \brief Chosen transport */
|
|
struct ast_sip_transport *transport;
|
|
};
|
|
|
|
/*! \brief Callback function for invoking hooks */
|
|
static int nat_invoke_hook(void *obj, void *arg, int flags)
|
|
{
|
|
struct ast_sip_nat_hook *hook = obj;
|
|
struct nat_hook_details *details = arg;
|
|
|
|
if (hook->outgoing_external_message) {
|
|
hook->outgoing_external_message(details->tdata, details->transport);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static pj_status_t nat_on_tx_message(pjsip_tx_data *tdata)
|
|
{
|
|
RAII_VAR(struct ao2_container *, transport_states, NULL, ao2_cleanup);
|
|
RAII_VAR(struct ast_sip_transport *, transport, NULL, ao2_cleanup);
|
|
RAII_VAR(struct ast_sip_transport_state *, transport_state, NULL, ao2_cleanup);
|
|
struct request_transport_details details = { 0, };
|
|
pjsip_via_hdr *via = NULL;
|
|
struct ast_sockaddr addr = { { 0, } };
|
|
pjsip_sip_uri *uri = NULL;
|
|
RAII_VAR(struct ao2_container *, hooks, NULL, ao2_cleanup);
|
|
|
|
/* If a transport selector is in use we know the transport or factory, so explicitly find it */
|
|
if (tdata->tp_sel.type == PJSIP_TPSELECTOR_TRANSPORT) {
|
|
details.transport = tdata->tp_sel.u.transport;
|
|
} else if (tdata->tp_sel.type == PJSIP_TPSELECTOR_LISTENER) {
|
|
details.factory = tdata->tp_sel.u.listener;
|
|
} else if (tdata->tp_info.transport->key.type == PJSIP_TRANSPORT_UDP || tdata->tp_info.transport->key.type == PJSIP_TRANSPORT_UDP6) {
|
|
/* Connectionless uses the same transport for all requests */
|
|
details.type = AST_TRANSPORT_UDP;
|
|
details.transport = tdata->tp_info.transport;
|
|
} else {
|
|
if (tdata->tp_info.transport->key.type == PJSIP_TRANSPORT_TCP) {
|
|
details.type = AST_TRANSPORT_TCP;
|
|
} else if (tdata->tp_info.transport->key.type == PJSIP_TRANSPORT_TLS) {
|
|
details.type = AST_TRANSPORT_TLS;
|
|
} else {
|
|
/* Unknown transport type, we can't map and thus can't apply NAT changes */
|
|
return PJ_SUCCESS;
|
|
}
|
|
|
|
if ((uri = nat_get_contact_sip_uri(tdata))) {
|
|
details.local_address = uri->host;
|
|
details.local_port = uri->port;
|
|
} else if ((tdata->msg->type == PJSIP_REQUEST_MSG) &&
|
|
(via = pjsip_msg_find_hdr(tdata->msg, PJSIP_H_VIA, NULL))) {
|
|
details.local_address = via->sent_by.host;
|
|
details.local_port = via->sent_by.port;
|
|
} else {
|
|
return PJ_SUCCESS;
|
|
}
|
|
|
|
if (!details.local_port) {
|
|
details.local_port = (details.type == AST_TRANSPORT_TLS) ? 5061 : 5060;
|
|
}
|
|
}
|
|
|
|
if (!(transport_states = ast_sip_get_transport_states())) {
|
|
return PJ_SUCCESS;
|
|
}
|
|
|
|
if (!(transport_state = ao2_callback(transport_states, 0, find_transport_state_in_use, &details))) {
|
|
return PJ_SUCCESS;
|
|
}
|
|
|
|
if (!(transport = ast_sorcery_retrieve_by_id(ast_sip_get_sorcery(), "transport", transport_state->id))) {
|
|
return PJ_SUCCESS;
|
|
}
|
|
|
|
if (transport_state->localnet) {
|
|
ast_sockaddr_parse(&addr, tdata->tp_info.dst_name, PARSE_PORT_FORBID);
|
|
ast_sockaddr_set_port(&addr, tdata->tp_info.dst_port);
|
|
|
|
/* See if where we are sending this request is local or not, and if not that we can get a Contact URI to modify */
|
|
if (ast_apply_ha(transport_state->localnet, &addr) != AST_SENSE_ALLOW) {
|
|
ast_debug(5, "Request is being sent to local address, skipping NAT manipulation\n");
|
|
return PJ_SUCCESS;
|
|
}
|
|
}
|
|
|
|
if (!ast_sockaddr_isnull(&transport_state->external_address)) {
|
|
/* Update the contact header with the external address */
|
|
if (uri || (uri = nat_get_contact_sip_uri(tdata))) {
|
|
pj_strdup2(tdata->pool, &uri->host, ast_sockaddr_stringify_host(&transport_state->external_address));
|
|
if (transport->external_signaling_port) {
|
|
uri->port = transport->external_signaling_port;
|
|
ast_debug(4, "Re-wrote Contact URI port to %d\n", uri->port);
|
|
}
|
|
}
|
|
|
|
/* Update the via header if relevant */
|
|
if ((tdata->msg->type == PJSIP_REQUEST_MSG) && (via || (via = pjsip_msg_find_hdr(tdata->msg, PJSIP_H_VIA, NULL)))) {
|
|
pj_strdup2(tdata->pool, &via->sent_by.host, ast_sockaddr_stringify_host(&transport_state->external_address));
|
|
if (transport->external_signaling_port) {
|
|
via->sent_by.port = transport->external_signaling_port;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Invoke any additional hooks that may be registered */
|
|
if ((hooks = ast_sorcery_retrieve_by_fields(ast_sip_get_sorcery(), "nat_hook", AST_RETRIEVE_FLAG_MULTIPLE | AST_RETRIEVE_FLAG_ALL, NULL))) {
|
|
struct nat_hook_details hook_details = {
|
|
.tdata = tdata,
|
|
.transport = transport,
|
|
};
|
|
ao2_callback(hooks, 0, nat_invoke_hook, &hook_details);
|
|
}
|
|
|
|
return PJ_SUCCESS;
|
|
}
|
|
|
|
static pjsip_module nat_module = {
|
|
.name = { "NAT", 3 },
|
|
.id = -1,
|
|
.priority = PJSIP_MOD_PRIORITY_TSX_LAYER - 2,
|
|
.on_rx_request = nat_on_rx_message,
|
|
.on_rx_response = nat_on_rx_message,
|
|
.on_tx_request = nat_on_tx_message,
|
|
.on_tx_response = nat_on_tx_message,
|
|
};
|
|
|
|
/*! \brief Function called when an INVITE goes out */
|
|
static int nat_incoming_invite_request(struct ast_sip_session *session, struct pjsip_rx_data *rdata)
|
|
{
|
|
if (session->inv_session->state == PJSIP_INV_STATE_INCOMING) {
|
|
pjsip_dlg_add_usage(session->inv_session->dlg, &nat_module, NULL);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*! \brief Function called when an INVITE response comes in */
|
|
static void nat_incoming_invite_response(struct ast_sip_session *session, struct pjsip_rx_data *rdata)
|
|
{
|
|
handle_rx_message(session->endpoint, rdata);
|
|
}
|
|
|
|
/*! \brief Function called when an INVITE comes in */
|
|
static void nat_outgoing_invite_request(struct ast_sip_session *session, struct pjsip_tx_data *tdata)
|
|
{
|
|
if (session->inv_session->state == PJSIP_INV_STATE_NULL) {
|
|
pjsip_dlg_add_usage(session->inv_session->dlg, &nat_module, NULL);
|
|
}
|
|
}
|
|
|
|
/*! \brief Supplement for adding NAT functionality to dialog */
|
|
static struct ast_sip_session_supplement nat_supplement = {
|
|
.method = "INVITE",
|
|
.priority = AST_SIP_SUPPLEMENT_PRIORITY_FIRST + 1,
|
|
.incoming_request = nat_incoming_invite_request,
|
|
.outgoing_request = nat_outgoing_invite_request,
|
|
.incoming_response = nat_incoming_invite_response,
|
|
};
|
|
|
|
|
|
static int unload_module(void)
|
|
{
|
|
ast_sip_session_unregister_supplement(&nat_supplement);
|
|
ast_sip_unregister_service(&nat_module);
|
|
return 0;
|
|
}
|
|
|
|
static int load_module(void)
|
|
{
|
|
CHECK_PJSIP_SESSION_MODULE_LOADED();
|
|
|
|
if (ast_sip_register_service(&nat_module)) {
|
|
ast_log(LOG_ERROR, "Could not register NAT module for incoming and outgoing requests\n");
|
|
return AST_MODULE_LOAD_DECLINE;
|
|
}
|
|
|
|
if (ast_sip_session_register_supplement(&nat_supplement)) {
|
|
ast_log(LOG_ERROR, "Could not register NAT session supplement for incoming and outgoing INVITE requests\n");
|
|
unload_module();
|
|
return AST_MODULE_LOAD_DECLINE;
|
|
}
|
|
|
|
return AST_MODULE_LOAD_SUCCESS;
|
|
}
|
|
|
|
AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "PJSIP NAT Support",
|
|
.support_level = AST_MODULE_SUPPORT_CORE,
|
|
.load = load_module,
|
|
.unload = unload_module,
|
|
.load_pri = AST_MODPRI_APP_DEPEND,
|
|
);
|