Author: file Date: Sun Aug 24 14:34:50 2014 New Revision: 421955 URL: http://svnview.digium.com/svn/asterisk?view=rev&rev=421955 Log: res_pjsip_transport_websocket: Attach the Websocket module on outgoing INVITEs.
In order to alter the Contact header on in-dialog requests and responses the Websocket module must be attached on outgoing INVITEs. The Contact header is modified so that the PJSIP transport layer can find and use the existing Websocket connection based on the source IP address, port, and transport. ASTERISK-24143 #close Reported by: Aleksei Kulakov Modified: branches/12/res/res_pjsip_transport_websocket.c Modified: branches/12/res/res_pjsip_transport_websocket.c URL: http://svnview.digium.com/svn/asterisk/branches/12/res/res_pjsip_transport_websocket.c?view=diff&rev=421955&r1=421954&r2=421955 ============================================================================== --- branches/12/res/res_pjsip_transport_websocket.c (original) +++ branches/12/res/res_pjsip_transport_websocket.c Sun Aug 24 14:34:50 2014 @@ -341,6 +341,22 @@ .id = -1, .priority = PJSIP_MOD_PRIORITY_TRANSPORT_LAYER, .on_rx_request = websocket_on_rx_msg, + .on_rx_response = websocket_on_rx_msg, +}; + +/*! \brief Function called when an INVITE goes out */ +static void websocket_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, &websocket_module, NULL); + } +} + +/*! \brief Supplement for adding Websocket functionality to dialog */ +static struct ast_sip_session_supplement websocket_supplement = { + .method = "INVITE", + .priority = AST_SIP_SUPPLEMENT_PRIORITY_FIRST + 1, + .outgoing_request = websocket_outgoing_invite_request, }; static int load_module(void) @@ -352,17 +368,24 @@ return AST_MODULE_LOAD_DECLINE; } - if (ast_websocket_add_protocol("sip", websocket_cb)) { + if (ast_sip_session_register_supplement(&websocket_supplement)) { ast_sip_unregister_service(&websocket_module); return AST_MODULE_LOAD_DECLINE; } + if (ast_websocket_add_protocol("sip", websocket_cb)) { + ast_sip_session_unregister_supplement(&websocket_supplement); + ast_sip_unregister_service(&websocket_module); + return AST_MODULE_LOAD_DECLINE; + } + return AST_MODULE_LOAD_SUCCESS; } static int unload_module(void) { ast_sip_unregister_service(&websocket_module); + ast_sip_session_unregister_supplement(&websocket_supplement); ast_websocket_remove_protocol("sip", websocket_cb); return 0; -- _____________________________________________________________________ -- Bandwidth and Colocation Provided by http://www.api-digital.com -- svn-commits mailing list To UNSUBSCRIBE or update options visit: http://lists.digium.com/mailman/listinfo/svn-commits