Merge the pimp_my_sip branch into trunk.
[asterisk/asterisk.git] / res / res_sip / sip_outbound_auth.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2013, Digium, Inc.
5  *
6  * Mark Michelson <mmichelson@digium.com>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15  * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 #include "asterisk.h"
20 #undef bzero
21 #define bzero bzero
22 #include "pjsip.h"
23
24 #include "asterisk/res_sip.h"
25 #include "asterisk/module.h"
26 #include "include/res_sip_private.h"
27
28 static pj_bool_t outbound_auth(pjsip_rx_data *rdata);
29
30 static pjsip_module outbound_auth_mod = {
31         .name = {"Outbound Authentication", 19},
32         .priority = PJSIP_MOD_PRIORITY_DIALOG_USAGE,
33         .on_rx_response = outbound_auth,
34 };
35
36 struct outbound_auth_cb_data {
37         ast_sip_dialog_outbound_auth_cb cb;
38         void *user_data;
39 };
40
41 static pj_bool_t outbound_auth(pjsip_rx_data *rdata)
42 {
43         RAII_VAR(struct ast_sip_endpoint *, endpoint, NULL, ao2_cleanup);
44         pjsip_transaction *tsx;
45         pjsip_dialog *dlg;
46         struct outbound_auth_cb_data *cb_data;
47         pjsip_tx_data *tdata;
48
49         if (rdata->msg_info.msg->line.status.code != 401 &&
50                         rdata->msg_info.msg->line.status.code != 407) {
51                 /* Doesn't pertain to us. Move on */
52                 return PJ_FALSE;
53         }
54
55         tsx = pjsip_rdata_get_tsx(rdata);
56         dlg = pjsip_rdata_get_dlg(rdata);
57         ast_assert(dlg != NULL && tsx != NULL);
58         endpoint = ast_sip_dialog_get_endpoint(dlg);
59
60         if (!endpoint) {
61                 return PJ_FALSE;
62         }
63
64         if (ast_sip_create_request_with_auth(endpoint->sip_outbound_auths, endpoint->num_outbound_auths, rdata, tsx, &tdata)) {
65                 return PJ_FALSE;
66         }
67
68         cb_data = dlg->mod_data[outbound_auth_mod.id];
69         if (cb_data) {
70                 cb_data->cb(dlg, tdata, cb_data->user_data);
71                 return PJ_TRUE;
72         }
73
74         pjsip_dlg_send_request(dlg, tdata, -1, NULL);
75         return PJ_TRUE;
76 }
77
78 int ast_sip_dialog_setup_outbound_authentication(pjsip_dialog *dlg, const struct ast_sip_endpoint *endpoint,
79                 ast_sip_dialog_outbound_auth_cb cb, void *user_data)
80 {
81         struct outbound_auth_cb_data *cb_data = PJ_POOL_ZALLOC_T(dlg->pool, struct outbound_auth_cb_data);
82         cb_data->cb = cb;
83         cb_data->user_data = user_data;
84
85         dlg->sess_count++;
86         pjsip_dlg_add_usage(dlg, &outbound_auth_mod, cb_data);
87         dlg->sess_count--;
88
89         return 0;
90 }
91
92 int ast_sip_initialize_outbound_authentication(void) {
93         return ast_sip_register_service(&outbound_auth_mod);
94 }