diff options
author | Mark Michelson <mmichelson@digium.com> | 2013-07-30 18:14:50 +0000 |
---|---|---|
committer | Mark Michelson <mmichelson@digium.com> | 2013-07-30 18:14:50 +0000 |
commit | 735b30ad71110c2a51404cb8686bbe3cf14b630c (patch) | |
tree | 76b1f10135c1b7f210e576be1359539de7e3476c /res/res_pjsip/pjsip_distributor.c | |
parent | 895c8e0d2c97cd04299f3f179e99d8a3873c06c6 (diff) |
The large GULP->PJSIP renaming effort.
The general gist is to have a clear boundary between old SIP stuff
and new SIP stuff by having the word "SIP" for old stuff and "PJSIP"
for new stuff. Here's a brief rundown of the changes:
* The word "Gulp" in dialstrings, functions, and CLI commands is now
"PJSIP"
* chan_gulp.c is now chan_pjsip.c
* Function names in chan_gulp.c that were "gulp_*" are now "chan_pjsip_*"
* All files that were "res_sip*" are now "res_pjsip*"
* The "res_sip" directory is now "res_pjsip"
* Files in the "res_pjsip" directory that began with "sip_*" are now "pjsip_*"
* The configuration file is now "pjsip.conf" instead of "res_sip.conf"
* The module info for all PJSIP-related files now uses "PJSIP" instead of "SIP"
* CLI and AMI commands created by Asterisk's PJSIP modules now have "pjsip" as
the starting word instead of "sip"
git-svn-id: https://origsvn.digium.com/svn/asterisk/trunk@395764 65c4cc65-6c06-0410-ace0-fbb531ad65f3
Diffstat (limited to 'res/res_pjsip/pjsip_distributor.c')
-rw-r--r-- | res/res_pjsip/pjsip_distributor.c | 322 |
1 files changed, 322 insertions, 0 deletions
diff --git a/res/res_pjsip/pjsip_distributor.c b/res/res_pjsip/pjsip_distributor.c new file mode 100644 index 000000000..7597ae79e --- /dev/null +++ b/res/res_pjsip/pjsip_distributor.c @@ -0,0 +1,322 @@ +/* + * Asterisk -- An open source telephony toolkit. + * + * Copyright (C) 2013, Digium, Inc. + * + * Mark Michelson <mmichelson@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. + */ + +#include "asterisk.h" + +#include <pjsip.h> + +#include "asterisk/res_pjsip.h" + +static int distribute(void *data); +static pj_bool_t distributor(pjsip_rx_data *rdata); + +static pjsip_module distributor_mod = { + .name = {"Request Distributor", 19}, + .priority = PJSIP_MOD_PRIORITY_TSX_LAYER - 6, + .on_rx_request = distributor, + .on_rx_response = distributor, +}; + +/*! Dialog-specific information the distributor uses */ +struct distributor_dialog_data { + /* Serializer to distribute tasks to for this dialog */ + struct ast_taskprocessor *serializer; + /* Endpoint associated with this dialog */ + struct ast_sip_endpoint *endpoint; +}; + +/*! + * \internal + * + * \note Call this with the dialog locked + */ +static struct distributor_dialog_data *distributor_dialog_data_alloc(pjsip_dialog *dlg) +{ + struct distributor_dialog_data *dist; + + dist = PJ_POOL_ZALLOC_T(dlg->pool, struct distributor_dialog_data); + pjsip_dlg_set_mod_data(dlg, distributor_mod.id, dist); + + return dist; +} + +void ast_sip_dialog_set_serializer(pjsip_dialog *dlg, struct ast_taskprocessor *serializer) +{ + struct distributor_dialog_data *dist; + SCOPED_LOCK(lock, dlg, pjsip_dlg_inc_lock, pjsip_dlg_dec_lock); + + dist = pjsip_dlg_get_mod_data(dlg, distributor_mod.id); + if (!dist) { + dist = distributor_dialog_data_alloc(dlg); + } + dist->serializer = serializer; +} + +void ast_sip_dialog_set_endpoint(pjsip_dialog *dlg, struct ast_sip_endpoint *endpoint) +{ + struct distributor_dialog_data *dist; + SCOPED_LOCK(lock, dlg, pjsip_dlg_inc_lock, pjsip_dlg_dec_lock); + + dist = pjsip_dlg_get_mod_data(dlg, distributor_mod.id); + if (!dist) { + dist = distributor_dialog_data_alloc(dlg); + } + dist->endpoint = endpoint; +} + +struct ast_sip_endpoint *ast_sip_dialog_get_endpoint(pjsip_dialog *dlg) +{ + struct distributor_dialog_data *dist; + SCOPED_LOCK(lock, dlg, pjsip_dlg_inc_lock, pjsip_dlg_dec_lock); + + dist = pjsip_dlg_get_mod_data(dlg, distributor_mod.id); + if (!dist || !dist->endpoint) { + return NULL; + } + ao2_ref(dist->endpoint, +1); + return dist->endpoint; +} + +static pj_bool_t distributor(pjsip_rx_data *rdata) +{ + pjsip_dialog *dlg = pjsip_ua_find_dialog(&rdata->msg_info.cid->id, &rdata->msg_info.to->tag, &rdata->msg_info.from->tag, PJ_TRUE); + struct distributor_dialog_data *dist = NULL; + struct ast_taskprocessor *serializer = NULL; + pjsip_rx_data *clone; + + if (dlg) { + dist = pjsip_dlg_get_mod_data(dlg, distributor_mod.id); + if (dist) { + serializer = dist->serializer; + } + } + + if (rdata->msg_info.msg->type == PJSIP_REQUEST_MSG && ( + !pjsip_method_cmp(&rdata->msg_info.msg->line.req.method, &pjsip_cancel_method) || + !pjsip_method_cmp(&rdata->msg_info.msg->line.req.method, &pjsip_bye_method)) && + !serializer) { + pjsip_endpt_respond_stateless(ast_sip_get_pjsip_endpoint(), rdata, 481, NULL, NULL, NULL); + goto end; + } + + pjsip_rx_data_clone(rdata, 0, &clone); + + if (dist) { + clone->endpt_info.mod_data[distributor_mod.id] = dist->endpoint; + } + + ast_sip_push_task(serializer, distribute, clone); + +end: + if (dlg) { + pjsip_dlg_dec_lock(dlg); + } + + return PJ_TRUE; +} + +static pj_bool_t endpoint_lookup(pjsip_rx_data *rdata); + +static pjsip_module endpoint_mod = { + .name = {"Endpoint Identifier", 19}, + .priority = PJSIP_MOD_PRIORITY_TSX_LAYER - 3, + .on_rx_request = endpoint_lookup, +}; + +static struct ast_sip_auth *artificial_auth; + +static int create_artificial_auth(void) +{ + if (!(artificial_auth = ast_sorcery_alloc( + ast_sip_get_sorcery(), SIP_SORCERY_AUTH_TYPE, "artificial"))) { + ast_log(LOG_ERROR, "Unable to create artificial auth\n"); + return -1; + } + + ast_string_field_set(artificial_auth, realm, "asterisk"); + ast_string_field_set(artificial_auth, auth_user, ""); + ast_string_field_set(artificial_auth, auth_pass, ""); + artificial_auth->type = AST_SIP_AUTH_TYPE_ARTIFICIAL; + return 0; +} + +struct ast_sip_auth *ast_sip_get_artificial_auth(void) +{ + ao2_ref(artificial_auth, +1); + return artificial_auth; +} + +static struct ast_sip_endpoint *artificial_endpoint; + +static int create_artificial_endpoint(void) +{ + if (!(artificial_endpoint = ast_sorcery_alloc( + ast_sip_get_sorcery(), "endpoint", NULL))) { + return -1; + } + + artificial_endpoint->inbound_auths.num = 1; + return 0; +} + +struct ast_sip_endpoint *ast_sip_get_artificial_endpoint(void) +{ + ao2_ref(artificial_endpoint, +1); + return artificial_endpoint; +} + +static pj_bool_t endpoint_lookup(pjsip_rx_data *rdata) +{ + struct ast_sip_endpoint *endpoint; + int is_ack = rdata->msg_info.msg->line.req.method.id == PJSIP_ACK_METHOD; + + endpoint = rdata->endpt_info.mod_data[distributor_mod.id]; + if (endpoint) { + /* Bumping the refcount makes refcounting consistent whether an endpoint + * is looked up or not */ + ao2_ref(endpoint, +1); + } else { + endpoint = ast_sip_identify_endpoint(rdata); + } + + if (!endpoint && !is_ack) { + char name[AST_UUID_STR_LEN] = ""; + pjsip_uri *from = rdata->msg_info.from->uri; + + /* always use an artificial endpoint - per discussion no reason + to have "alwaysauthreject" as an option. It is felt using it + was a bug fix and it is not needed since we are not worried about + breaking old stuff and we really don't want to enable the discovery + of SIP accounts */ + endpoint = ast_sip_get_artificial_endpoint(); + + if (PJSIP_URI_SCHEME_IS_SIP(from) || PJSIP_URI_SCHEME_IS_SIPS(from)) { + pjsip_sip_uri *sip_from = pjsip_uri_get_uri(from); + ast_copy_pj_str(name, &sip_from->user, sizeof(name)); + } + + ast_sip_report_invalid_endpoint(name, rdata); + } + rdata->endpt_info.mod_data[endpoint_mod.id] = endpoint; + return PJ_FALSE; +} + +static pj_bool_t authenticate(pjsip_rx_data *rdata) +{ + RAII_VAR(struct ast_sip_endpoint *, endpoint, ast_pjsip_rdata_get_endpoint(rdata), ao2_cleanup); + int is_ack = rdata->msg_info.msg->line.req.method.id == PJSIP_ACK_METHOD; + + ast_assert(endpoint != NULL); + + if (!is_ack && ast_sip_requires_authentication(endpoint, rdata)) { + pjsip_tx_data *tdata; + pjsip_endpt_create_response(ast_sip_get_pjsip_endpoint(), rdata, 401, NULL, &tdata); + switch (ast_sip_check_authentication(endpoint, rdata, tdata)) { + case AST_SIP_AUTHENTICATION_CHALLENGE: + /* Send the 401 we created for them */ + ast_sip_report_auth_challenge_sent(endpoint, rdata, tdata); + pjsip_endpt_send_response2(ast_sip_get_pjsip_endpoint(), rdata, tdata, NULL, NULL); + return PJ_TRUE; + case AST_SIP_AUTHENTICATION_SUCCESS: + ast_sip_report_auth_success(endpoint, rdata); + pjsip_tx_data_dec_ref(tdata); + return PJ_FALSE; + case AST_SIP_AUTHENTICATION_FAILED: + ast_sip_report_auth_failed_challenge_response(endpoint, rdata); + pjsip_endpt_send_response2(ast_sip_get_pjsip_endpoint(), rdata, tdata, NULL, NULL); + return PJ_TRUE; + case AST_SIP_AUTHENTICATION_ERROR: + ast_sip_report_auth_failed_challenge_response(endpoint, rdata); + pjsip_tx_data_dec_ref(tdata); + pjsip_endpt_respond_stateless(ast_sip_get_pjsip_endpoint(), rdata, 500, NULL, NULL, NULL); + return PJ_TRUE; + } + } + + return PJ_FALSE; +} + +static pjsip_module auth_mod = { + .name = {"Request Authenticator", 21}, + .priority = PJSIP_MOD_PRIORITY_APPLICATION - 1, + .on_rx_request = authenticate, +}; + +static int distribute(void *data) +{ + static pjsip_process_rdata_param param = { + .start_mod = &distributor_mod, + .idx_after_start = 1, + }; + pj_bool_t handled; + pjsip_rx_data *rdata = data; + int is_request = rdata->msg_info.msg->type == PJSIP_REQUEST_MSG; + int is_ack = is_request ? rdata->msg_info.msg->line.req.method.id == PJSIP_ACK_METHOD : 0; + struct ast_sip_endpoint *endpoint; + + pjsip_endpt_process_rx_data(ast_sip_get_pjsip_endpoint(), rdata, ¶m, &handled); + if (!handled && is_request && !is_ack) { + pjsip_endpt_respond_stateless(ast_sip_get_pjsip_endpoint(), rdata, 501, NULL, NULL, NULL); + } + + /* The endpoint_mod stores an endpoint reference in the mod_data of rdata. This + * is the only appropriate spot to actually decrement the reference. + */ + endpoint = rdata->endpt_info.mod_data[endpoint_mod.id]; + ao2_cleanup(endpoint); + pjsip_rx_data_free_cloned(rdata); + return 0; +} + +struct ast_sip_endpoint *ast_pjsip_rdata_get_endpoint(pjsip_rx_data *rdata) +{ + struct ast_sip_endpoint *endpoint = rdata->endpt_info.mod_data[endpoint_mod.id]; + if (endpoint) { + ao2_ref(endpoint, +1); + } + return endpoint; +} + +int ast_sip_initialize_distributor(void) +{ + if (create_artificial_endpoint() || create_artificial_auth()) { + return -1; + } + + if (ast_sip_register_service(&distributor_mod)) { + return -1; + } + if (ast_sip_register_service(&endpoint_mod)) { + return -1; + } + if (ast_sip_register_service(&auth_mod)) { + return -1; + } + + return 0; +} + +void ast_sip_destroy_distributor(void) +{ + ast_sip_unregister_service(&distributor_mod); + ast_sip_unregister_service(&endpoint_mod); + ast_sip_unregister_service(&auth_mod); + + ao2_cleanup(artificial_auth); + ao2_cleanup(artificial_endpoint); +} |