chan_pjsip: Fix attended transfer connected line name update.
[asterisk/asterisk.git] / res / res_pjsip_caller_id.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 /*** 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/channel.h"
34 #include "asterisk/module.h"
35 #include "asterisk/callerid.h"
36
37 /*!
38  * \internal
39  * \brief Set an ast_party_id name and number based on an identity header.
40  * \param hdr From, P-Asserted-Identity, or Remote-Party-ID header on incoming message
41  * \param[out] id The ID to set data on
42  */
43 static void set_id_from_hdr(pjsip_fromto_hdr *hdr, struct ast_party_id *id)
44 {
45         char cid_name[AST_CHANNEL_NAME];
46         char cid_num[AST_CHANNEL_NAME];
47         pjsip_sip_uri *uri;
48         pjsip_name_addr *id_name_addr = (pjsip_name_addr *) hdr->uri;
49
50         uri = pjsip_uri_get_uri(id_name_addr);
51         ast_copy_pj_str(cid_name, &id_name_addr->display, sizeof(cid_name));
52         ast_copy_pj_str(cid_num, &uri->user, sizeof(cid_num));
53
54         ast_free(id->name.str);
55         id->name.str = ast_strdup(cid_name);
56         if (!ast_strlen_zero(cid_name)) {
57                 id->name.valid = 1;
58         }
59         ast_free(id->number.str);
60         id->number.str = ast_strdup(cid_num);
61         if (!ast_strlen_zero(cid_num)) {
62                 id->number.valid = 1;
63         }
64 }
65
66 /*!
67  * \internal
68  * \brief Get a P-Asserted-Identity or Remote-Party-ID header from an incoming message
69  *
70  * This function will parse the header as if it were a From header. This allows for us
71  * to easily manipulate the URI, as well as add, modify, or remove parameters from the
72  * header
73  *
74  * \param rdata The incoming message
75  * \param header_name The name of the ID header to find
76  * \retval NULL No ID header present or unable to parse ID header
77  * \retval non-NULL The parsed ID header
78  */
79 static pjsip_fromto_hdr *get_id_header(pjsip_rx_data *rdata, const pj_str_t *header_name)
80 {
81         static const pj_str_t from = { "From", 4 };
82         pj_str_t header_content;
83         pjsip_fromto_hdr *parsed_hdr;
84         pjsip_generic_string_hdr *ident = pjsip_msg_find_hdr_by_name(rdata->msg_info.msg,
85                         header_name, NULL);
86         int parsed_len;
87
88         if (!ident) {
89                 return NULL;
90         }
91
92         pj_strdup_with_null(rdata->tp_info.pool, &header_content, &ident->hvalue);
93
94         parsed_hdr = pjsip_parse_hdr(rdata->tp_info.pool, &from, header_content.ptr,
95                         pj_strlen(&header_content), &parsed_len);
96
97         if (!parsed_hdr) {
98                 return NULL;
99         }
100
101         return parsed_hdr;
102 }
103
104 /*!
105  * \internal
106  * \brief Set an ast_party_id structure based on data in a P-Asserted-Identity header
107  *
108  * This makes use of \ref set_id_from_hdr for setting name and number. It uses
109  * the contents of a Privacy header in order to set presentation information.
110  *
111  * \param rdata The incoming message
112  * \param[out] id The ID to set
113  * \retval 0 Successfully set the party ID
114  * \retval non-zero Could not set the party ID
115  */
116 static int set_id_from_pai(pjsip_rx_data *rdata, struct ast_party_id *id)
117 {
118         static const pj_str_t pai_str = { "P-Asserted-Identity", 19 };
119         static const pj_str_t privacy_str = { "Privacy", 7 };
120         pjsip_fromto_hdr *pai_hdr = get_id_header(rdata, &pai_str);
121         pjsip_generic_string_hdr *privacy;
122
123         if (!pai_hdr) {
124                 return -1;
125         }
126
127         set_id_from_hdr(pai_hdr, id);
128
129         if (!id->number.valid) {
130                 return -1;
131         }
132
133         privacy = pjsip_msg_find_hdr_by_name(rdata->msg_info.msg, &privacy_str, NULL);
134         if (!privacy) {
135                 return 0;
136         }
137         if (!pj_stricmp2(&privacy->hvalue, "id")) {
138                 id->number.presentation = AST_PRES_PROHIB_USER_NUMBER_NOT_SCREENED;
139                 id->name.presentation = AST_PRES_PROHIB_USER_NUMBER_NOT_SCREENED;
140         }
141
142         return 0;
143 }
144
145 /*!
146  * \internal
147  * \brief Set an ast_party_id structure based on data in a Remote-Party-ID header
148  *
149  * This makes use of \ref set_id_from_hdr for setting name and number. It uses
150  * the privacy and screen parameters in order to set presentation information.
151  *
152  * \param rdata The incoming message
153  * \param[out] id The ID to set
154  * \retval 0 Succesfully set the party ID
155  * \retval non-zero Could not set the party ID
156  */
157 static int set_id_from_rpid(pjsip_rx_data *rdata, struct ast_party_id *id)
158 {
159         static const pj_str_t rpid_str = { "Remote-Party-ID", 15 };
160         static const pj_str_t privacy_str = { "privacy", 7 };
161         static const pj_str_t screen_str = { "screen", 6 };
162         pjsip_fromto_hdr *rpid_hdr = get_id_header(rdata, &rpid_str);
163         pjsip_param *screen;
164         pjsip_param *privacy;
165
166         if (!rpid_hdr) {
167                 return -1;
168         }
169
170         set_id_from_hdr(rpid_hdr, id);
171
172         if (!id->number.valid) {
173                 return -1;
174         }
175
176         privacy = pjsip_param_find(&rpid_hdr->other_param, &privacy_str);
177         screen = pjsip_param_find(&rpid_hdr->other_param, &screen_str);
178         if (privacy && !pj_stricmp2(&privacy->value, "full")) {
179                 id->number.presentation |= AST_PRES_RESTRICTED;
180                 id->name.presentation |= AST_PRES_RESTRICTED;
181         }
182         if (screen && !pj_stricmp2(&screen->value, "yes")) {
183                 id->number.presentation |= AST_PRES_USER_NUMBER_PASSED_SCREEN;
184                 id->name.presentation |= AST_PRES_USER_NUMBER_PASSED_SCREEN;
185         }
186
187         return 0;
188 }
189
190 /*!
191  * \internal
192  * \brief Set an ast_party_id structure based on data in a From
193  *
194  * This makes use of \ref set_id_from_hdr for setting name and number. It uses
195  * no information from the message in order to set privacy. It relies on endpoint
196  * configuration for privacy information.
197  *
198  * \param rdata The incoming message
199  * \param[out] id The ID to set
200  * \retval 0 Succesfully set the party ID
201  * \retval non-zero Could not set the party ID
202  */
203 static int set_id_from_from(struct pjsip_rx_data *rdata, struct ast_party_id *id)
204 {
205         pjsip_fromto_hdr *from = pjsip_msg_find_hdr(rdata->msg_info.msg,
206                         PJSIP_H_FROM, rdata->msg_info.msg->hdr.next);
207
208         if (!from) {
209                 /* This had better not happen */
210                 return -1;
211         }
212
213         set_id_from_hdr(from, id);
214
215         if (!id->number.valid) {
216                 return -1;
217         }
218
219         return 0;
220 }
221
222 /*!
223  * \internal
224  * \brief Determine if a connected line update should be queued
225  *
226  * This uses information about the session and the ID that would be queued
227  * in the connected line update in order to determine if we should queue
228  * a connected line update.
229  *
230  * \param session The session whose channel we wish to queue the connected line update on
231  * \param id The identification information that would be queued on the connected line update
232  * \retval 0 We should not queue a connected line update
233  * \retval non-zero We should queue a connected line update
234  */
235 static int should_queue_connected_line_update(const struct ast_sip_session *session, const struct ast_party_id *id)
236 {
237         /* Invalid number means no update */
238         if (!id->number.valid) {
239                 return 0;
240         }
241
242         /* If the session has never communicated an update or if the
243          * new ID has a different number than the session, then we
244          * should queue an update
245          */
246         if (ast_strlen_zero(session->id.number.str) ||
247                         strcmp(session->id.number.str, id->number.str)) {
248                 return 1;
249         }
250
251         /* By making it to this point, it means the number is not enough
252          * to determine if an update should be sent. Now we look at
253          * the name
254          */
255
256         /* If the number couldn't warrant an update and the name is
257          * invalid, then no update
258          */
259         if (!id->name.valid) {
260                 return 0;
261         }
262
263         /* If the name has changed or we don't have a name set for the
264          * session, then we should send an update
265          */
266         if (ast_strlen_zero(session->id.name.str) ||
267                         strcmp(session->id.name.str, id->name.str)) {
268                 return 1;
269         }
270
271         /* Neither the name nor the number have changed. No update */
272         return 0;
273 }
274
275 /*!
276  * \internal
277  * \brief Queue a connected line update on a session's channel.
278  * \param session The session whose channel should have the connected line update queued upon.
279  * \param id The identification information to place in the connected line update
280  */
281 static void queue_connected_line_update(struct ast_sip_session *session, const struct ast_party_id *id)
282 {
283         struct ast_party_connected_line connected;
284         struct ast_party_caller caller;
285
286         /* Fill connected line information */
287         ast_party_connected_line_init(&connected);
288         connected.id = *id;
289         connected.id.tag = session->endpoint->id.self.tag;
290         connected.source = AST_CONNECTED_LINE_UPDATE_SOURCE_ANSWER;
291
292         /* Save to channel driver copy */
293         ast_party_id_copy(&session->id, &connected.id);
294
295         /* Update our channel CALLERID() */
296         ast_party_caller_init(&caller);
297         caller.id = connected.id;
298         caller.ani = connected.id;
299         caller.ani2 = ast_channel_caller(session->channel)->ani2;
300         ast_channel_set_caller_event(session->channel, &caller, NULL);
301
302         /* Tell peer about the new connected line information. */
303         ast_channel_queue_connected_line_update(session->channel, &connected, NULL);
304 }
305
306 /*!
307  * \internal
308  * \brief Make updates to connected line information based on an incoming request.
309  *
310  * This will get identity information from an incoming request. Once the identification is
311  * retrieved, we will check if the new information warrants a connected line update and queue
312  * a connected line update if so.
313  *
314  * \param session The session on which we received an incoming request
315  * \param rdata The incoming request
316  */
317 static void update_incoming_connected_line(struct ast_sip_session *session, pjsip_rx_data *rdata)
318 {
319         struct ast_party_id id;
320
321         if (!session->endpoint->id.trust_inbound) {
322                 return;
323         }
324
325         ast_party_id_init(&id);
326         if (!set_id_from_pai(rdata, &id) || !set_id_from_rpid(rdata, &id)) {
327                 if (should_queue_connected_line_update(session, &id)) {
328                         queue_connected_line_update(session, &id);
329                 }
330         }
331         ast_party_id_free(&id);
332 }
333
334 /*!
335  * \internal
336  * \brief Session supplement callback on an incoming INVITE request
337  *
338  * If we are receiving an initial INVITE, then we will set the session's identity
339  * based on the INVITE or configured endpoint values. If we are receiving a reinvite,
340  * then we will potentially queue a connected line update via the \ref update_incoming_connected_line
341  * function
342  *
343  * \param session The session that has received an INVITE
344  * \param rdata The incoming INVITE
345  */
346 static int caller_id_incoming_request(struct ast_sip_session *session, pjsip_rx_data *rdata)
347 {
348         if (session->inv_session->state < PJSIP_INV_STATE_CONFIRMED) {
349                 /*
350                  * Initial inbound INVITE.  Set the session ID directly
351                  * because the channel has not been created yet.
352                  */
353                 if (session->endpoint->id.trust_inbound
354                         && (!set_id_from_pai(rdata, &session->id)
355                                 || !set_id_from_rpid(rdata, &session->id))) {
356                         ast_free(session->id.tag);
357                         session->id.tag = ast_strdup(session->endpoint->id.self.tag);
358                         return 0;
359                 }
360                 ast_party_id_copy(&session->id, &session->endpoint->id.self);
361                 if (!session->endpoint->id.self.number.valid) {
362                         set_id_from_from(rdata, &session->id);
363                 }
364         } else {
365                 /* Reinvite. Check for changes to the ID and queue a connected line
366                  * update if necessary
367                  */
368                 update_incoming_connected_line(session, rdata);
369         }
370         return 0;
371 }
372
373 /*!
374  * \internal
375  * \brief Session supplement callback on INVITE response
376  *
377  * INVITE responses could result in queuing connected line updates.
378  *
379  * \param session The session on which communication is happening
380  * \param rdata The incoming INVITE response
381  */
382 static void caller_id_incoming_response(struct ast_sip_session *session, pjsip_rx_data *rdata)
383 {
384         if (!session->channel) {
385                 return;
386         }
387
388         update_incoming_connected_line(session, rdata);
389 }
390
391 /*!
392  * \internal
393  * \brief Set name and number information on an identity header.
394  * \param pool Memory pool to use for string duplication
395  * \param id_hdr A From, P-Asserted-Identity, or Remote-Party-ID header to modify
396  * \param id The identity information to apply to the header
397  */
398 static void modify_id_header(pj_pool_t *pool, pjsip_fromto_hdr *id_hdr, const struct ast_party_id *id)
399 {
400         pjsip_name_addr *id_name_addr;
401         pjsip_sip_uri *id_uri;
402
403         id_name_addr = (pjsip_name_addr *) id_hdr->uri;
404         id_uri = pjsip_uri_get_uri(id_name_addr->uri);
405
406         if (id->name.valid) {
407                 pj_strdup2(pool, &id_name_addr->display, id->name.str);
408         }
409
410         if (id->number.valid) {
411                 pj_strdup2(pool, &id_uri->user, id->number.str);
412         }
413 }
414
415 /*!
416  * \internal
417  * \brief Create an identity header for an outgoing message
418  * \param hdr_name The name of the header to create
419  * \param tdata The message to place the header on
420  * \param id The identification information for the new header
421  * \return newly-created header
422  */
423 static pjsip_fromto_hdr *create_new_id_hdr(const pj_str_t *hdr_name, pjsip_tx_data *tdata, const struct ast_party_id *id)
424 {
425         pjsip_fromto_hdr *id_hdr;
426         pjsip_fromto_hdr *base;
427         pjsip_name_addr *id_name_addr;
428         pjsip_sip_uri *id_uri;
429
430         base = tdata->msg->type == PJSIP_REQUEST_MSG ? PJSIP_MSG_FROM_HDR(tdata->msg) :
431                 PJSIP_MSG_TO_HDR(tdata->msg);
432         id_hdr = pjsip_from_hdr_create(tdata->pool);
433         id_hdr->type = PJSIP_H_OTHER;
434         pj_strdup(tdata->pool, &id_hdr->name, hdr_name);
435         id_hdr->sname.slen = 0;
436
437         id_name_addr = pjsip_uri_clone(tdata->pool, base->uri);
438         id_uri = pjsip_uri_get_uri(id_name_addr->uri);
439
440         if (id->name.valid) {
441                 pj_strdup2(tdata->pool, &id_name_addr->display, id->name.str);
442         }
443
444         pj_strdup2(tdata->pool, &id_uri->user, id->number.str);
445
446         id_hdr->uri = (pjsip_uri *) id_name_addr;
447         return id_hdr;
448 }
449
450 /*!
451  * \internal
452  * \brief Add a Privacy header to an outbound message
453  *
454  * When sending a P-Asserted-Identity header, if privacy is requested, then we
455  * will need to indicate such by adding a Privacy header. Similarly, if no
456  * privacy is requested, and a Privacy header already exists on the message,
457  * then the old Privacy header should be removed.
458  *
459  * \param tdata The outbound message to add the Privacy header to
460  * \param id The id information used to determine privacy
461  */
462 static void add_privacy_header(pjsip_tx_data *tdata, const struct ast_party_id *id)
463 {
464         static const pj_str_t pj_privacy_name = { "Privacy", 7 };
465         static const pj_str_t pj_privacy_value = { "id", 2 };
466         pjsip_hdr *old_privacy;
467
468         old_privacy = pjsip_msg_find_hdr_by_name(tdata->msg, &pj_privacy_name, NULL);
469
470         if ((id->name.presentation & AST_PRES_RESTRICTION) == AST_PRES_RESTRICTED ||
471                         (id->name.presentation & AST_PRES_RESTRICTION) == AST_PRES_RESTRICTED) {
472                 if (!old_privacy) {
473                         pjsip_generic_string_hdr *privacy_hdr = pjsip_generic_string_hdr_create(
474                                         tdata->pool, &pj_privacy_name, &pj_privacy_value);
475                         pjsip_msg_add_hdr(tdata->msg, (pjsip_hdr *)privacy_hdr);
476                 }
477         } else {
478                 if (old_privacy) {
479                         pj_list_erase(old_privacy);
480                 }
481         }
482 }
483
484 /*!
485  * \internal
486  * \brief Add a P-Asserted-Identity header to an outbound message
487  * \param tdata The message to add the header to
488  * \param id The identification information used to populate the header
489  */
490 static void add_pai_header(pjsip_tx_data *tdata, const struct ast_party_id *id)
491 {
492         static const pj_str_t pj_pai_name = { "P-Asserted-Identity", 19 };
493         pjsip_fromto_hdr *pai_hdr;
494         pjsip_fromto_hdr *old_pai;
495
496         if (!id->number.valid) {
497                 return;
498         }
499
500         /* Since inv_session reuses responses, we have to make sure there's not already
501          * a P-Asserted-Identity present. If there is, we just modify the old one.
502          */
503         old_pai = pjsip_msg_find_hdr_by_name(tdata->msg, &pj_pai_name, NULL);
504         if (old_pai) {
505                 modify_id_header(tdata->pool, old_pai, id);
506                 add_privacy_header(tdata, id);
507                 return;
508         }
509
510         pai_hdr = create_new_id_hdr(&pj_pai_name, tdata, id);
511         if (!pai_hdr) {
512                 return;
513         }
514         add_privacy_header(tdata, id);
515
516         pjsip_msg_add_hdr(tdata->msg, (pjsip_hdr *)pai_hdr);
517 }
518
519 /*!
520  * \internal
521  * \brief Add privacy and screen parameters to a Remote-Party-ID header.
522  *
523  * If privacy is requested, then the privacy and screen parameters need to
524  * reflect this. Similarly, if no privacy or screening is to be communicated,
525  * we need to make sure that any previously set values are updated.
526  *
527  * \param tdata The message where the Remote-Party-ID header is
528  * \param hdr The header on which the parameters are being added
529  * \param id The identification information used to determine privacy
530  */
531 static void add_privacy_params(pjsip_tx_data *tdata, pjsip_fromto_hdr *hdr, const struct ast_party_id *id)
532 {
533         static const pj_str_t privacy_str = { "privacy", 7 };
534         static const pj_str_t screen_str = { "screen", 6 };
535         static const pj_str_t privacy_full_str = { "full", 4 };
536         static const pj_str_t privacy_off_str = { "off", 3 };
537         static const pj_str_t screen_yes_str = { "yes", 3 };
538         static const pj_str_t screen_no_str = { "no", 2 };
539         pjsip_param *old_privacy;
540         pjsip_param *old_screen;
541         pjsip_param *privacy;
542         pjsip_param *screen;
543
544         old_privacy = pjsip_param_find(&hdr->other_param, &privacy_str);
545         old_screen = pjsip_param_find(&hdr->other_param, &screen_str);
546
547         if (!old_privacy) {
548                 privacy = PJ_POOL_ALLOC_T(tdata->pool, pjsip_param);
549                 privacy->name = privacy_str;
550                 pj_list_insert_before(&hdr->other_param, privacy);
551         } else {
552                 privacy = old_privacy;
553         }
554
555         if (!old_screen) {
556                 screen = PJ_POOL_ALLOC_T(tdata->pool, pjsip_param);
557                 screen->name = screen_str;
558                 pj_list_insert_before(&hdr->other_param, screen);
559         } else {
560                 screen = old_screen;
561         }
562
563         if ((id->name.presentation & AST_PRES_RESTRICTION) == AST_PRES_ALLOWED &&
564                         (id->name.presentation & AST_PRES_RESTRICTION) == AST_PRES_ALLOWED) {
565                 privacy->value = privacy_off_str;
566         } else {
567                 privacy->value = privacy_full_str;
568         }
569
570         if ((id->name.presentation & AST_PRES_NUMBER_TYPE) == AST_PRES_USER_NUMBER_PASSED_SCREEN &&
571                         (id->number.presentation & AST_PRES_NUMBER_TYPE) == AST_PRES_USER_NUMBER_PASSED_SCREEN) {
572                 screen->value = screen_yes_str;
573         } else {
574                 screen->value = screen_no_str;
575         }
576 }
577
578 /*!
579  * \internal
580  * \brief Add a Remote-Party-ID header to an outbound message
581  * \param tdata The message to add the header to
582  * \param id The identification information used to populate the header
583  */
584 static void add_rpid_header(pjsip_tx_data *tdata, const struct ast_party_id *id)
585 {
586         static const pj_str_t pj_rpid_name = { "Remote-Party-ID", 15 };
587         pjsip_fromto_hdr *rpid_hdr;
588         pjsip_fromto_hdr *old_rpid;
589
590         if (!id->number.valid) {
591                 return;
592         }
593
594         /* Since inv_session reuses responses, we have to make sure there's not already
595          * a P-Asserted-Identity present. If there is, we just modify the old one.
596          */
597         old_rpid = pjsip_msg_find_hdr_by_name(tdata->msg, &pj_rpid_name, NULL);
598         if (old_rpid) {
599                 modify_id_header(tdata->pool, old_rpid, id);
600                 add_privacy_params(tdata, old_rpid, id);
601                 return;
602         }
603
604         rpid_hdr = create_new_id_hdr(&pj_rpid_name, tdata, id);
605         if (!rpid_hdr) {
606                 return;
607         }
608         add_privacy_params(tdata, rpid_hdr, id);
609         pjsip_msg_add_hdr(tdata->msg, (pjsip_hdr *)rpid_hdr);
610 }
611
612 /*!
613  * \internal
614  * \brief Add any appropriate identification headers to an outbound SIP message
615  *
616  * This will determine if an outbound message should have identification headers and
617  * will add the appropriately configured headers
618  *
619  * \param session The session on which we will be sending the message
620  * \param tdata The outbound message
621  * \param The identity information to place on the message
622  */
623 static void add_id_headers(const struct ast_sip_session *session, pjsip_tx_data *tdata, const struct ast_party_id *id)
624 {
625         if (((id->name.presentation & AST_PRES_RESTRICTION) == AST_PRES_RESTRICTED ||
626                         (id->number.presentation & AST_PRES_RESTRICTION) == AST_PRES_RESTRICTED) &&
627                         !session->endpoint->id.trust_outbound) {
628                 return;
629         }
630         if (session->endpoint->id.send_pai) {
631                 add_pai_header(tdata, id);
632         }
633         if (session->endpoint->id.send_rpid) {
634                 add_rpid_header(tdata, id);
635         }
636 }
637
638 /*!
639  * \internal
640  * \brief Session supplement callback for outgoing INVITE requests
641  *
642  * For an initial INVITE request, we may change the From header to appropriately
643  * reflect the identity information. On all INVITEs (initial and reinvite) we may
644  * add other identity headers such as P-Asserted-Identity and Remote-Party-ID based
645  * on configuration and privacy settings
646  *
647  * \param session The session on which the INVITE will be sent
648  * \param tdata The outbound INVITE request
649  */
650 static void caller_id_outgoing_request(struct ast_sip_session *session, pjsip_tx_data *tdata)
651 {
652         struct ast_party_id effective_id;
653         struct ast_party_id connected_id;
654
655         if (!session->channel) {
656                 return;
657         }
658
659         /* Must do a deep copy unless we hold the channel lock the entire time. */
660         ast_party_id_init(&connected_id);
661         ast_channel_lock(session->channel);
662         effective_id = ast_channel_connected_effective_id(session->channel);
663         ast_party_id_copy(&connected_id, &effective_id);
664         ast_channel_unlock(session->channel);
665
666         if (session->inv_session->state < PJSIP_INV_STATE_CONFIRMED &&
667                         ast_strlen_zero(session->endpoint->fromuser) &&
668                         (session->endpoint->id.trust_outbound ||
669                         ((connected_id.name.presentation & AST_PRES_RESTRICTION) == AST_PRES_ALLOWED &&
670                         (connected_id.number.presentation & AST_PRES_RESTRICTION) == AST_PRES_ALLOWED))) {
671                 /* Only change the From header on the initial outbound INVITE. Switching it
672                  * mid-call might confuse some UAs.
673                  */
674                 pjsip_fromto_hdr *from;
675                 pjsip_dialog *dlg;
676
677                 from = pjsip_msg_find_hdr(tdata->msg, PJSIP_H_FROM, tdata->msg->hdr.next);
678                 dlg = session->inv_session->dlg;
679
680                 modify_id_header(tdata->pool, from, &connected_id);
681                 modify_id_header(dlg->pool, dlg->local.info, &connected_id);
682         }
683         add_id_headers(session, tdata, &connected_id);
684         ast_party_id_free(&connected_id);
685 }
686
687 /*!
688  * \internal
689  * \brief Session supplement for outgoing INVITE response
690  *
691  * This will add P-Asserted-Identity and Remote-Party-ID headers if necessary
692  *
693  * \param session The session on which the INVITE response is to be sent
694  * \param tdata The outbound INVITE response
695  */
696 static void caller_id_outgoing_response(struct ast_sip_session *session, pjsip_tx_data *tdata)
697 {
698         struct ast_party_id effective_id;
699         struct ast_party_id connected_id;
700
701         if (!session->channel) {
702                 return;
703         }
704
705         /* Must do a deep copy unless we hold the channel lock the entire time. */
706         ast_party_id_init(&connected_id);
707         ast_channel_lock(session->channel);
708         effective_id = ast_channel_connected_effective_id(session->channel);
709         ast_party_id_copy(&connected_id, &effective_id);
710         ast_channel_unlock(session->channel);
711
712         add_id_headers(session, tdata, &connected_id);
713         ast_party_id_free(&connected_id);
714 }
715
716 static struct ast_sip_session_supplement caller_id_supplement = {
717         .method = "INVITE,UPDATE",
718         .priority = AST_SIP_SUPPLEMENT_PRIORITY_CHANNEL - 1000,
719         .incoming_request = caller_id_incoming_request,
720         .incoming_response = caller_id_incoming_response,
721         .outgoing_request = caller_id_outgoing_request,
722         .outgoing_response = caller_id_outgoing_response,
723 };
724
725 static int load_module(void)
726 {
727         ast_sip_session_register_supplement(&caller_id_supplement);
728         return AST_MODULE_LOAD_SUCCESS;
729 }
730
731 static int unload_module(void)
732 {
733         ast_sip_session_unregister_supplement(&caller_id_supplement);
734         return 0;
735 }
736
737 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "PJSIP Caller ID Support",
738                 .support_level = AST_MODULE_SUPPORT_CORE,
739                 .load = load_module,
740                 .unload = unload_module,
741                 .load_pri = AST_MODPRI_APP_DEPEND,
742                );