82c3caaed6ea3a40532ee3b99737982a82a46bbe
[asterisk/asterisk.git] / res / res_pjsip_diversion.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2013, Digium, Inc.
5  *
6  * Kevin Harwell <kharwell@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 /*** MODULEINFO
20         <depend>pjproject</depend>
21         <depend>res_pjsip</depend>
22         <depend>res_pjsip_session</depend>
23         <support_level>core</support_level>
24  ***/
25
26 #include "asterisk.h"
27
28 #include <pjsip.h>
29 #include <pjsip_ua.h>
30
31 #include "asterisk/res_pjsip.h"
32 #include "asterisk/res_pjsip_session.h"
33 #include "asterisk/callerid.h"
34 #include "asterisk/channel.h"
35 #include "asterisk/module.h"
36 #include "asterisk/strings.h"
37
38 static const pj_str_t diversion_name = { "Diversion", 9 };
39
40 /*!
41  * \internal
42  * \brief Determine if the given string is a SIP token.
43  * \since 13.8.0
44  *
45  * \param str String to determine if is a SIP token.
46  *
47  * \note A token is defined by RFC3261 Section 25.1
48  *
49  * \return Non-zero if the string is a SIP token.
50  */
51 static int sip_is_token(const char *str)
52 {
53         int is_token;
54
55         if (ast_strlen_zero(str)) {
56                 /* An empty string is not a token. */
57                 return 0;
58         }
59
60         is_token = 1;
61         do {
62                 if (!isalnum(*str)
63                         && !strchr("-.!%*_+`'~", *str)) {
64                         /* The character is not allowed in a token. */
65                         is_token = 0;
66                         break;
67                 }
68         } while (*++str);
69
70         return is_token;
71 }
72
73 /*! \brief Diversion header reasons
74  *
75  * The core defines a bunch of constants used to define
76  * redirecting reasons. This provides a translation table
77  * between those and the strings which may be present in
78  * a SIP Diversion header
79  */
80 static const struct reasons {
81         enum AST_REDIRECTING_REASON code;
82         const char *text;
83 } reason_table[] = {
84         { AST_REDIRECTING_REASON_UNKNOWN, "unknown" },
85         { AST_REDIRECTING_REASON_USER_BUSY, "user-busy" },
86         { AST_REDIRECTING_REASON_NO_ANSWER, "no-answer" },
87         { AST_REDIRECTING_REASON_UNAVAILABLE, "unavailable" },
88         { AST_REDIRECTING_REASON_UNCONDITIONAL, "unconditional" },
89         { AST_REDIRECTING_REASON_TIME_OF_DAY, "time-of-day" },
90         { AST_REDIRECTING_REASON_DO_NOT_DISTURB, "do-not-disturb" },
91         { AST_REDIRECTING_REASON_DEFLECTION, "deflection" },
92         { AST_REDIRECTING_REASON_FOLLOW_ME, "follow-me" },
93         { AST_REDIRECTING_REASON_OUT_OF_ORDER, "out-of-service" },
94         { AST_REDIRECTING_REASON_AWAY, "away" },
95         { AST_REDIRECTING_REASON_CALL_FWD_DTE, "cf_dte" },              /* Non-standard */
96         { AST_REDIRECTING_REASON_SEND_TO_VM, "send_to_vm" },    /* Non-standard */
97 };
98
99 static const char *reason_code_to_str(const struct ast_party_redirecting_reason *reason)
100 {
101         int idx;
102         int code;
103
104         /* use specific string if given */
105         if (!ast_strlen_zero(reason->str)) {
106                 return reason->str;
107         }
108
109         code = reason->code;
110         for (idx = 0; idx < ARRAY_LEN(reason_table); ++idx) {
111                 if (code == reason_table[idx].code) {
112                         return reason_table[idx].text;
113                 }
114         }
115
116         return "unknown";
117 }
118
119 static pjsip_fromto_hdr *get_diversion_header(pjsip_rx_data *rdata)
120 {
121         static const pj_str_t from_name = { "From", 4 };
122
123         pjsip_generic_string_hdr *hdr;
124         pj_str_t value;
125         int size;
126
127         if (!(hdr = pjsip_msg_find_hdr_by_name(rdata->msg_info.msg, &diversion_name, NULL))) {
128                 return NULL;
129         }
130
131         pj_strdup_with_null(rdata->tp_info.pool, &value, &hdr->hvalue);
132
133         /* parse as a fromto header */
134         return pjsip_parse_hdr(rdata->tp_info.pool, &from_name, value.ptr,
135                                pj_strlen(&value), &size);
136 }
137
138 static void set_redirecting_value(char **dst, const pj_str_t *src)
139 {
140         ast_free(*dst);
141         *dst = ast_malloc(pj_strlen(src) + 1);
142         if (*dst) {
143                 ast_copy_pj_str(*dst, src, pj_strlen(src) + 1);
144         }
145 }
146
147 static void set_redirecting_id(pjsip_name_addr *name_addr, struct ast_party_id *data,
148                                struct ast_set_party_id *update)
149 {
150         pjsip_sip_uri *uri = pjsip_uri_get_uri(name_addr->uri);
151
152         if (pj_strlen(&uri->user)) {
153                 update->number = 1;
154                 data->number.valid = 1;
155                 set_redirecting_value(&data->number.str, &uri->user);
156         }
157
158         if (pj_strlen(&name_addr->display)) {
159                 update->name = 1;
160                 data->name.valid = 1;
161                 set_redirecting_value(&data->name.str, &name_addr->display);
162         }
163 }
164
165 static void copy_redirecting_id(struct ast_party_id *dst, const struct ast_party_id *src,
166                                 struct ast_set_party_id *update)
167 {
168         ast_party_id_copy(dst, src);
169
170         if (dst->number.valid) {
171                 update->number = 1;
172         }
173
174         if (dst->name.valid) {
175                 update->name = 1;
176         }
177 }
178
179 static void set_redirecting_reason(pjsip_fromto_hdr *hdr,
180                                    struct ast_party_redirecting_reason *data)
181 {
182         static const pj_str_t reason_name = { "reason", 6 };
183         pjsip_param *reason = pjsip_param_find(&hdr->other_param, &reason_name);
184         char *reason_str;
185
186         if (!reason) {
187                 return;
188         }
189
190         set_redirecting_value(&data->str, &reason->value);
191         if (!data->str) {
192                 /* Oops, allocation failure */
193                 return;
194         }
195         reason_str = ast_strdupa(data->str);
196
197         /* Remove any enclosing double-quotes */
198         if (*reason_str == '"') {
199                 reason_str = ast_strip_quoted(reason_str, "\"", "\"");
200         }
201
202         data->code = ast_redirecting_reason_parse(reason_str);
203         if (data->code < 0) {
204                 data->code = AST_REDIRECTING_REASON_UNKNOWN;
205         } else {
206                 ast_free(data->str);
207                 data->str = ast_strdup("");
208         }
209 }
210
211 static void set_redirecting(struct ast_sip_session *session,
212                             pjsip_fromto_hdr *from_info,
213                             pjsip_name_addr *to_info)
214 {
215         struct ast_party_redirecting data;
216         struct ast_set_party_redirecting update;
217
218         if (!session->channel) {
219                 return;
220         }
221
222         ast_party_redirecting_init(&data);
223         memset(&update, 0, sizeof(update));
224
225         if (from_info) {
226                 set_redirecting_id((pjsip_name_addr*)from_info->uri,
227                         &data.from, &update.from);
228                 set_redirecting_reason(from_info, &data.reason);
229         } else {
230                 copy_redirecting_id(&data.from, &session->id, &update.from);
231         }
232
233         set_redirecting_id(to_info, &data.to, &update.to);
234
235         ast_set_party_id_all(&update.priv_orig);
236         ast_set_party_id_all(&update.priv_from);
237         ast_set_party_id_all(&update.priv_to);
238         ++data.count;
239
240         ast_channel_set_redirecting(session->channel, &data, &update);
241         ast_party_redirecting_free(&data);
242 }
243
244 static int diversion_incoming_request(struct ast_sip_session *session, pjsip_rx_data *rdata)
245 {
246         pjsip_fromto_hdr *hdr = get_diversion_header(rdata);
247
248         if (hdr) {
249                 set_redirecting(session, hdr, (pjsip_name_addr*)
250                                 PJSIP_MSG_TO_HDR(rdata->msg_info.msg)->uri);
251         }
252
253         return 0;
254 }
255
256 static void diversion_incoming_response(struct ast_sip_session *session, pjsip_rx_data *rdata)
257 {
258         static const pj_str_t contact_name = { "Contact", 7 };
259
260         pjsip_status_line status = rdata->msg_info.msg->line.status;
261         pjsip_fromto_hdr *div_hdr;
262         pjsip_contact_hdr *contact_hdr;
263
264         if ((status.code != 302) && (status.code != 181)) {
265                 return;
266         }
267
268         /* use the diversion header info if there is one. if not one then use the
269            session caller id info. if that doesn't exist use info from the To hdr*/
270         if (!(div_hdr = get_diversion_header(rdata)) && !session->id.number.valid) {
271                 div_hdr = PJSIP_MSG_TO_HDR(rdata->msg_info.msg);
272         }
273
274         contact_hdr = pjsip_msg_find_hdr_by_name(rdata->msg_info.msg, &contact_name, NULL);
275
276         set_redirecting(session, div_hdr, contact_hdr ? (pjsip_name_addr*)contact_hdr->uri :
277                         (pjsip_name_addr*)PJSIP_MSG_FROM_HDR(rdata->msg_info.msg)->uri);
278 }
279
280 /*!
281  * \internal
282  * \brief Adds diversion header information to an outbound SIP message
283  *
284  * \param tdata The outbound message
285  * \param data The redirecting data used to fill parts of the diversion header
286  */
287 static void add_diversion_header(pjsip_tx_data *tdata, struct ast_party_redirecting *data)
288 {
289         pjsip_fromto_hdr *hdr;
290         pjsip_name_addr *name_addr;
291         pjsip_sip_uri *uri;
292         pjsip_param *param;
293         pjsip_fromto_hdr *old_hdr;
294         const char *reason_str;
295         const char *quote_str;
296         char *reason_buf;
297
298         struct ast_party_id *id = &data->from;
299         pjsip_uri *base = PJSIP_MSG_FROM_HDR(tdata->msg)->uri;
300
301         if (!id->number.valid || ast_strlen_zero(id->number.str)) {
302                 return;
303         }
304
305         hdr = pjsip_from_hdr_create(tdata->pool);
306         hdr->type = PJSIP_H_OTHER;
307         pj_strdup(tdata->pool, &hdr->name, &diversion_name);
308         hdr->sname = hdr->name;
309
310         name_addr = pjsip_uri_clone(tdata->pool, base);
311         uri = pjsip_uri_get_uri(name_addr->uri);
312
313         pj_strdup2(tdata->pool, &name_addr->display, id->name.str);
314         pj_strdup2(tdata->pool, &uri->user, id->number.str);
315
316         param = PJ_POOL_ALLOC_T(tdata->pool, pjsip_param);
317         param->name = pj_str("reason");
318
319         reason_str = reason_code_to_str(&data->reason);
320
321         /* Reason is either already quoted or it is a token to not need quotes added. */
322         quote_str = *reason_str == '\"' || sip_is_token(reason_str) ? "" : "\"";
323
324         reason_buf = pj_pool_alloc(tdata->pool, strlen(reason_str) + 3);
325         sprintf(reason_buf, "%s%s%s", quote_str, reason_str, quote_str);/* Safe */
326
327         param->value = pj_str(reason_buf);
328
329         pj_list_insert_before(&hdr->other_param, param);
330
331         hdr->uri = (pjsip_uri *) name_addr;
332         old_hdr = pjsip_msg_find_hdr_by_name(tdata->msg, &diversion_name, NULL);
333         if (old_hdr) {
334                 pj_list_erase(old_hdr);
335         }
336         pjsip_msg_add_hdr(tdata->msg, (pjsip_hdr *)hdr);
337 }
338
339 static void get_redirecting_add_diversion(struct ast_sip_session *session, pjsip_tx_data *tdata)
340 {
341         struct ast_party_redirecting *data;
342
343         if (session->channel && session->endpoint->id.send_diversion &&
344             (data = ast_channel_redirecting(session->channel))->count) {
345                 add_diversion_header(tdata, data);
346         }
347 }
348
349 /*!
350  * \internal
351  * \brief Adds a diversion header to an outgoing INVITE request if
352  *  redirecting information is available.
353  *
354  * \param session The session on which the INVITE request is to be sent
355  * \param tdata The outbound INVITE request
356  */
357 static void diversion_outgoing_request(struct ast_sip_session *session, pjsip_tx_data *tdata)
358 {
359         get_redirecting_add_diversion(session, tdata);
360 }
361
362 /*!
363  * \internal
364  * \brief Adds a diversion header to an outgoing 3XX response
365  *
366  * \param session The session on which the INVITE response is to be sent
367  * \param tdata The outbound INVITE response
368  */
369 static void diversion_outgoing_response(struct ast_sip_session *session, pjsip_tx_data *tdata)
370 {
371         struct pjsip_status_line status = tdata->msg->line.status;
372
373         /* add to 302 and 181 */
374         if (PJSIP_IS_STATUS_IN_CLASS(status.code, 300) || (status.code == 181)) {
375                 get_redirecting_add_diversion(session, tdata);
376         }
377 }
378
379 static struct ast_sip_session_supplement diversion_supplement = {
380         .method = "INVITE",
381         /* this supplement needs to be called after caller id
382            and after the channel has been created */
383         .priority = AST_SIP_SUPPLEMENT_PRIORITY_CHANNEL + 100,
384         .incoming_request = diversion_incoming_request,
385         .incoming_response = diversion_incoming_response,
386         .outgoing_request = diversion_outgoing_request,
387         .outgoing_response = diversion_outgoing_response,
388         .response_priority = AST_SIP_SESSION_BEFORE_REDIRECTING,
389 };
390
391 static int load_module(void)
392 {
393         CHECK_PJSIP_SESSION_MODULE_LOADED();
394
395         ast_sip_session_register_supplement(&diversion_supplement);
396         return AST_MODULE_LOAD_SUCCESS;
397 }
398
399 static int unload_module(void)
400 {
401         ast_sip_session_unregister_supplement(&diversion_supplement);
402         return 0;
403 }
404
405 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "PJSIP Add Diversion Header Support",
406         .support_level = AST_MODULE_SUPPORT_CORE,
407         .load = load_module,
408         .unload = unload_module,
409         .load_pri = AST_MODPRI_APP_DEPEND,
410 );