Asterisk - The Open Source Telephony Project GIT-master-b023714
Loading...
Searching...
No Matches
Data Structures | Macros | Functions | Variables
res_pjsip_transport_websocket.c File Reference
#include "asterisk.h"
#include <pjsip.h>
#include <pjsip_ua.h>
#include "asterisk/module.h"
#include "asterisk/http_websocket.h"
#include "asterisk/res_pjsip.h"
#include "asterisk/res_pjsip_session.h"
#include "asterisk/taskprocessor.h"
Include dependency graph for res_pjsip_transport_websocket.c:

Go to the source code of this file.

Data Structures

struct  transport_create_data
 
struct  transport_read_data
 
struct  ws_transport
 Wrapper for pjsip_transport, for storing the WebSocket session. More...
 

Macros

#define COLON_LEN   1
 
#define MAX_PORT_LEN   5
 

Functions

static void __reg_module (void)
 
static void __unreg_module (void)
 
struct ast_moduleAST_MODULE_SELF_SYM (void)
 
static struct ast_taskprocessorcreate_websocket_serializer (void)
 
static int get_write_timeout (void)
 
static int load_module (void)
 
static void save_orig_contact_host (pjsip_rx_data *rdata, pjsip_sip_uri *uri)
 
static int transport_create (void *data)
 Create a pjsip transport.
 
static void transport_dtor (void *arg)
 
static int transport_read (void *data)
 Pass WebSocket data into pjsip transport manager.
 
static int transport_shutdown (void *data)
 
static int unload_module (void)
 
static void websocket_cb (struct ast_websocket *session, struct ast_variable *parameters, struct ast_variable *headers)
 WebSocket connection handler.
 
static pj_bool_t websocket_on_rx_msg (pjsip_rx_data *rdata)
 Store the transport a message came in on, so it can be used for outbound messages to that contact.
 
static void websocket_outgoing_invite_request (struct ast_sip_session *session, struct pjsip_tx_data *tdata)
 Function called when an INVITE goes out.
 
static pj_status_t ws_destroy (pjsip_transport *transport)
 Destroy the pjsip transport.
 
static pj_status_t ws_send_msg (pjsip_transport *transport, pjsip_tx_data *tdata, const pj_sockaddr_t *rem_addr, int addr_len, void *token, pjsip_transport_callback callback)
 Send a message over the WebSocket connection.
 
static pj_status_t ws_shutdown (pjsip_transport *transport)
 Shut down the pjsip transport.
 

Variables

static struct ast_module_info __mod_info = { .name = AST_MODULE, .flags = AST_MODFLAG_LOAD_ORDER , .description = "PJSIP WebSocket Transport Support" , .key = ASTERISK_GPL_KEY , .buildopt_sum = AST_BUILDOPT_SUM, .support_level = AST_MODULE_SUPPORT_CORE, .load = load_module, .unload = unload_module, .load_pri = AST_MODPRI_APP_DEPEND, .requires = "res_pjsip,res_http_websocket", }
 
static const struct ast_module_infoast_module_info = &__mod_info
 
static int transport_type_wss
 WebSocket transport module.
 
static int transport_type_wss_ipv6
 
static pjsip_module websocket_module
 
static struct ast_sip_session_supplement websocket_supplement
 Supplement for adding Websocket functionality to dialog.
 
static int ws_obj_name_serial
 

Macro Definition Documentation

◆ COLON_LEN

#define COLON_LEN   1

◆ MAX_PORT_LEN

#define MAX_PORT_LEN   5

Function Documentation

◆ __reg_module()

static void __reg_module ( void  )
static

Definition at line 582 of file res_pjsip_transport_websocket.c.

◆ __unreg_module()

static void __unreg_module ( void  )
static

Definition at line 582 of file res_pjsip_transport_websocket.c.

◆ AST_MODULE_SELF_SYM()

struct ast_module * AST_MODULE_SELF_SYM ( void  )

Definition at line 582 of file res_pjsip_transport_websocket.c.

◆ create_websocket_serializer()

static struct ast_taskprocessor * create_websocket_serializer ( void  )
static

Definition at line 364 of file res_pjsip_transport_websocket.c.

365{
366 char tps_name[AST_TASKPROCESSOR_MAX_NAME + 1];
367
368 /* Create name with seq number appended. */
369 ast_taskprocessor_build_name(tps_name, sizeof(tps_name), "pjsip/websocket");
370
371 return ast_sip_create_serializer(tps_name);
372}
struct ast_taskprocessor * ast_sip_create_serializer(const char *name)
Create a new serializer for SIP tasks.
Definition res_pjsip.c:2094
void ast_taskprocessor_build_name(char *buf, unsigned int size, const char *format,...)
Build a taskprocessor name with a sequence number on the end.
#define AST_TASKPROCESSOR_MAX_NAME
Suggested maximum taskprocessor name length (less null terminator).

References ast_sip_create_serializer(), ast_taskprocessor_build_name(), and AST_TASKPROCESSOR_MAX_NAME.

Referenced by websocket_cb().

◆ get_write_timeout()

static int get_write_timeout ( void  )
static

Definition at line 326 of file res_pjsip_transport_websocket.c.

327{
328 int write_timeout = -1;
330
332
333 if (transport_states) {
334 struct ao2_iterator it_transport_states = ao2_iterator_init(transport_states, 0);
335 struct ast_sip_transport_state *transport_state;
336
337 for (; (transport_state = ao2_iterator_next(&it_transport_states)); ao2_cleanup(transport_state)) {
338 struct ast_sip_transport *transport;
339
340 if (transport_state->type != AST_TRANSPORT_WS && transport_state->type != AST_TRANSPORT_WSS) {
341 continue;
342 }
343 transport = ast_sorcery_retrieve_by_id(ast_sip_get_sorcery(), "transport", transport_state->id);
344 if (!transport) {
345 continue;
346 }
347 ast_debug(5, "Found %s transport with write timeout: %d\n",
348 transport->type == AST_TRANSPORT_WS ? "WS" : "WSS",
349 transport->write_timeout);
351 }
352 ao2_iterator_destroy(&it_transport_states);
354 }
355
356 if (write_timeout < 0) {
358 }
359
360 ast_debug(1, "Write timeout for WS/WSS transports: %d\n", write_timeout);
361 return write_timeout;
362}
#define ao2_iterator_next(iter)
Definition astobj2.h:1911
#define ao2_cleanup(obj)
Definition astobj2.h:1934
struct ao2_iterator ao2_iterator_init(struct ao2_container *c, int flags) attribute_warn_unused_result
Create an iterator for a container.
void ao2_iterator_destroy(struct ao2_iterator *iter)
Destroy a container iterator.
static struct ao2_container * transport_states
#define AST_DEFAULT_WEBSOCKET_WRITE_TIMEOUT
Default websocket write timeout, in ms.
#define ast_debug(level,...)
Log a DEBUG message.
@ AST_TRANSPORT_WSS
Definition netsock2.h:64
@ AST_TRANSPORT_WS
Definition netsock2.h:63
struct ao2_container * ast_sip_get_transport_states(void)
Retrieves all transport states.
struct ast_sorcery * ast_sip_get_sorcery(void)
Get a pointer to the SIP sorcery structure.
void * ast_sorcery_retrieve_by_id(const struct ast_sorcery *sorcery, const char *type, const char *id)
Retrieve an object using its unique identifier.
Definition sorcery.c:1917
Generic container type.
When we need to walk through a container, we use an ao2_iterator to keep track of the current positio...
Definition astobj2.h:1821
Structure for SIP transport information.
Definition res_pjsip.h:117
enum ast_transport type
Definition res_pjsip.h:131
Transport to bind to.
Definition res_pjsip.h:219
enum ast_transport type
Definition res_pjsip.h:241
#define MAX(a, b)
Definition utils.h:251

References ao2_cleanup, ao2_iterator_destroy(), ao2_iterator_init(), ao2_iterator_next, ast_debug, AST_DEFAULT_WEBSOCKET_WRITE_TIMEOUT, ast_sip_get_sorcery(), ast_sip_get_transport_states(), ast_sorcery_retrieve_by_id(), AST_TRANSPORT_WS, AST_TRANSPORT_WSS, ast_sip_transport_state::id, MAX, transport_states, ast_sip_transport_state::type, ast_sip_transport::type, and ast_sip_transport::write_timeout.

Referenced by websocket_cb().

◆ load_module()

static int load_module ( void  )
static

Definition at line 537 of file res_pjsip_transport_websocket.c.

538{
539 /*
540 * We only need one transport type name (ws) defined. Firefox
541 * and Chrome do not support anything other than secure websockets
542 * anymore.
543 *
544 * Also we really cannot have two transports with the same name
545 * and address family because it would be ambiguous. Outgoing
546 * requests may try to find the transport by name and pjproject
547 * only finds the first one registered.
548 */
549 pjsip_transport_register_type(PJSIP_TRANSPORT_RELIABLE | PJSIP_TRANSPORT_SECURE, "ws", 5060, &transport_type_wss);
550 pjsip_transport_register_type(PJSIP_TRANSPORT_RELIABLE | PJSIP_TRANSPORT_SECURE | PJSIP_TRANSPORT_IPV6, "ws", 5060, &transport_type_wss_ipv6);
551
552 if (ast_sip_register_service(&websocket_module) != PJ_SUCCESS) {
554 }
555
557
562 }
563
565}
int AST_OPTIONAL_API_NAME() ast_websocket_add_protocol(const char *name, ast_websocket_callback callback)
Add a sub-protocol handler to the default /ws server.
@ AST_MODULE_LOAD_SUCCESS
Definition module.h:70
@ AST_MODULE_LOAD_DECLINE
Module has failed to load, may be in an inconsistent state.
Definition module.h:78
void ast_sip_unregister_service(pjsip_module *module)
Definition res_pjsip.c:133
int ast_sip_register_service(pjsip_module *module)
Register a SIP service in Asterisk.
Definition res_pjsip.c:117
#define ast_sip_session_register_supplement(supplement)
void ast_sip_session_unregister_supplement(struct ast_sip_session_supplement *supplement)
Unregister a an supplement to SIP session processing.
static pjsip_module websocket_module
static int transport_type_wss_ipv6
static struct ast_sip_session_supplement websocket_supplement
Supplement for adding Websocket functionality to dialog.
static void websocket_cb(struct ast_websocket *session, struct ast_variable *parameters, struct ast_variable *headers)
WebSocket connection handler.
static int transport_type_wss
WebSocket transport module.

References AST_MODULE_LOAD_DECLINE, AST_MODULE_LOAD_SUCCESS, ast_sip_register_service(), ast_sip_session_register_supplement, ast_sip_session_unregister_supplement(), ast_sip_unregister_service(), ast_websocket_add_protocol(), transport_type_wss, transport_type_wss_ipv6, websocket_cb(), websocket_module, and websocket_supplement.

◆ save_orig_contact_host()

static void save_orig_contact_host ( pjsip_rx_data *  rdata,
pjsip_sip_uri *  uri 
)
static

Definition at line 435 of file res_pjsip_transport_websocket.c.

436{
437 pjsip_param *x_orig_host;
438 pj_str_t p_value;
439#define COLON_LEN 1
440#define MAX_PORT_LEN 5
441
442 if (rdata->msg_info.msg->type != PJSIP_REQUEST_MSG ||
443 rdata->msg_info.msg->line.req.method.id != PJSIP_REGISTER_METHOD) {
444 return;
445 }
446
447 ast_debug(1, "Saving contact '%.*s:%d'\n",
448 (int)uri->host.slen, uri->host.ptr, uri->port);
449
450 x_orig_host = PJ_POOL_ALLOC_T(rdata->tp_info.pool, pjsip_param);
451 x_orig_host->name = pj_strdup3(rdata->tp_info.pool, "x-ast-orig-host");
452 p_value.slen = pj_strlen(&uri->host) + COLON_LEN + MAX_PORT_LEN;
453 p_value.ptr = (char*)pj_pool_alloc(rdata->tp_info.pool, p_value.slen + 1);
454 p_value.slen = snprintf(p_value.ptr, p_value.slen + 1, "%.*s:%d", (int)uri->host.slen, uri->host.ptr, uri->port);
455 pj_strassign(&x_orig_host->value, &p_value);
456 pj_list_insert_before(&uri->other_param, x_orig_host);
457
458 return;
459}
#define MAX_PORT_LEN
#define COLON_LEN

References ast_debug, COLON_LEN, and MAX_PORT_LEN.

Referenced by websocket_on_rx_msg().

◆ transport_create()

static int transport_create ( void *  data)
static

Create a pjsip transport.

Definition at line 161 of file res_pjsip_transport_websocket.c.

162{
163 struct transport_create_data *create_data = data;
164 struct ws_transport *newtransport = NULL;
165 pjsip_tp_state_callback state_cb;
166
167 pjsip_endpoint *endpt = ast_sip_get_pjsip_endpoint();
168 struct pjsip_tpmgr *tpmgr = pjsip_endpt_get_tpmgr(endpt);
169
170 char *ws_addr_str;
171 pj_pool_t *pool;
172 pj_str_t buf;
173 pj_status_t status;
174
175 newtransport = ao2_t_alloc_options(sizeof(*newtransport), transport_dtor,
176 AO2_ALLOC_OPT_LOCK_NOLOCK, "pjsip websocket transport");
177 if (!newtransport) {
178 ast_log(LOG_ERROR, "Failed to allocate WebSocket transport.\n");
179 goto on_error;
180 }
181
182 /* Give websocket transport a unique name for its lifetime */
183 snprintf(newtransport->transport.obj_name, PJ_MAX_OBJ_NAME, "ws%p-%d",
185
186 newtransport->transport.endpt = endpt;
187
188 if (!(pool = pjsip_endpt_create_pool(endpt, "ws", 512, 512))) {
189 ast_log(LOG_ERROR, "Failed to allocate WebSocket endpoint pool.\n");
190 goto on_error;
191 }
192
193 newtransport->transport.pool = pool;
194 newtransport->ws_session = create_data->ws_session;
195
196 /* Keep the session until transport dies */
197 ast_websocket_ref(newtransport->ws_session);
198
199 status = pj_atomic_create(pool, 0, &newtransport->transport.ref_cnt);
200 if (status != PJ_SUCCESS) {
201 goto on_error;
202 }
203
204 status = pj_lock_create_recursive_mutex(pool, pool->obj_name, &newtransport->transport.lock);
205 if (status != PJ_SUCCESS) {
206 goto on_error;
207 }
208
209 /*
210 * The type_name here is mostly used by log messages eihter in
211 * pjproject or Asterisk. Other places are reconstituting subscriptions
212 * after a restart (which could never work for a websocket connection anyway),
213 * received MESSAGE requests to set PJSIP_TRANSPORT, and most importantly
214 * by pjproject when generating the Via header.
215 */
216 newtransport->transport.type_name = ast_websocket_is_secure(newtransport->ws_session)
217 ? "WSS" : "WS";
218
220 ast_debug(4, "Creating websocket transport for %s:%s\n",
221 newtransport->transport.type_name, ws_addr_str);
222
223 newtransport->transport.info = (char *) pj_pool_alloc(newtransport->transport.pool,
224 strlen(newtransport->transport.type_name) + strlen(ws_addr_str) + sizeof(" to "));
225 sprintf(newtransport->transport.info, "%s to %s", newtransport->transport.type_name, ws_addr_str);
226
227 pj_sockaddr_parse(pj_AF_UNSPEC(), 0, pj_cstr(&buf, ws_addr_str), &newtransport->transport.key.rem_addr);
228 if (newtransport->transport.key.rem_addr.addr.sa_family == pj_AF_INET6()) {
229 newtransport->transport.key.type = transport_type_wss_ipv6;
230 } else {
231 newtransport->transport.key.type = transport_type_wss;
232 }
233
234 newtransport->transport.addr_len = pj_sockaddr_get_len(&newtransport->transport.key.rem_addr);
235
237 pj_sockaddr_parse(pj_AF_UNSPEC(), 0, pj_cstr(&buf, ws_addr_str), &newtransport->transport.local_addr);
238 pj_strdup2(pool, &newtransport->transport.local_name.host, ast_sockaddr_stringify_addr(ast_websocket_local_address(newtransport->ws_session)));
239 newtransport->transport.local_name.port = ast_sockaddr_port(ast_websocket_local_address(newtransport->ws_session));
240 pj_strdup2(pool, &newtransport->transport.remote_name.host, ast_sockaddr_stringify_addr(ast_websocket_remote_address(newtransport->ws_session)));
241 newtransport->transport.remote_name.port = ast_sockaddr_port(ast_websocket_remote_address(newtransport->ws_session));
242
243 newtransport->transport.flag = pjsip_transport_get_flag_from_type((pjsip_transport_type_e)newtransport->transport.key.type);
244 newtransport->transport.dir = PJSIP_TP_DIR_INCOMING;
245 newtransport->transport.tpmgr = tpmgr;
246 newtransport->transport.send_msg = &ws_send_msg;
247 newtransport->transport.do_shutdown = &ws_shutdown;
248 newtransport->transport.destroy = &ws_destroy;
249
250 status = pjsip_transport_register(newtransport->transport.tpmgr,
251 (pjsip_transport *)newtransport);
252 if (status != PJ_SUCCESS) {
253 goto on_error;
254 }
255
256 /* Add a reference for pjsip transport manager */
257 ao2_ref(newtransport, +1);
258
259 newtransport->rdata.tp_info.transport = &newtransport->transport;
260 newtransport->rdata.tp_info.pool = pjsip_endpt_create_pool(endpt, "rtd%p",
261 PJSIP_POOL_RDATA_LEN, PJSIP_POOL_RDATA_INC);
262 if (!newtransport->rdata.tp_info.pool) {
263 ast_log(LOG_ERROR, "Failed to allocate WebSocket rdata.\n");
264 pjsip_transport_destroy((pjsip_transport *)newtransport);
265 goto on_error;
266 }
267
268 create_data->transport = newtransport;
269
270 /* Notify application of transport state */
271 state_cb = pjsip_tpmgr_get_state_cb(newtransport->transport.tpmgr);
272 if (state_cb) {
273 pjsip_transport_state_info state_info;
274
275 memset(&state_info, 0, sizeof(state_info));
276 state_cb(&newtransport->transport, PJSIP_TP_STATE_CONNECTED, &state_info);
277 }
278
279 return 0;
280
281on_error:
282 ao2_cleanup(newtransport);
283 return -1;
284}
jack_status_t status
Definition app_jack.c:149
#define ast_log
Definition astobj2.c:42
@ AO2_ALLOC_OPT_LOCK_NOLOCK
Definition astobj2.h:367
#define ao2_t_alloc_options(data_size, destructor_fn, options, debug_msg)
Allocate and initialize an object.
Definition astobj2.h:402
#define ao2_ref(o, delta)
Reference/unreference an object and return the old refcount.
Definition astobj2.h:459
char buf[BUFSIZE]
Definition eagi_proxy.c:66
void AST_OPTIONAL_API_NAME() ast_websocket_ref(struct ast_websocket *session)
Increase the reference count for a WebSocket session.
struct ast_sockaddr *AST_OPTIONAL_API_NAME() ast_websocket_local_address(struct ast_websocket *session)
Get the local address for a WebSocket connection session.
struct ast_sockaddr *AST_OPTIONAL_API_NAME() ast_websocket_remote_address(struct ast_websocket *session)
Get the remote address for a WebSocket connected session.
int AST_OPTIONAL_API_NAME() ast_websocket_is_secure(struct ast_websocket *session)
Get whether the WebSocket session is using a secure transport or not.
#define LOG_ERROR
int ast_atomic_fetchadd_int(volatile int *p, int v)
Atomically add v to *p and return the previous value of *p.
Definition lock.h:764
static char * ast_sockaddr_stringify(const struct ast_sockaddr *addr)
Wrapper around ast_sockaddr_stringify_fmt() with default format.
Definition netsock2.h:256
#define ast_sockaddr_port(addr)
Get the port number of a socket address.
Definition netsock2.h:517
static char * ast_sockaddr_stringify_addr(const struct ast_sockaddr *addr)
Wrapper around ast_sockaddr_stringify_fmt() to return an address only.
Definition netsock2.h:286
pjsip_endpoint * ast_sip_get_pjsip_endpoint(void)
Get a pointer to the PJSIP endpoint.
Definition res_pjsip.c:520
static void transport_dtor(void *arg)
static int ws_obj_name_serial
static pj_status_t ws_send_msg(pjsip_transport *transport, pjsip_tx_data *tdata, const pj_sockaddr_t *rem_addr, int addr_len, void *token, pjsip_transport_callback callback)
Send a message over the WebSocket connection.
static pj_status_t ws_shutdown(pjsip_transport *transport)
Shut down the pjsip transport.
static pj_status_t ws_destroy(pjsip_transport *transport)
Destroy the pjsip transport.
#define NULL
Definition resample.c:96
Wrapper for pjsip_transport, for storing the WebSocket session.
struct ast_websocket * ws_session

References AO2_ALLOC_OPT_LOCK_NOLOCK, ao2_cleanup, ao2_ref, ao2_t_alloc_options, ast_atomic_fetchadd_int(), ast_debug, ast_log, ast_sip_get_pjsip_endpoint(), ast_sockaddr_port, ast_sockaddr_stringify(), ast_sockaddr_stringify_addr(), ast_websocket_is_secure(), ast_websocket_local_address(), ast_websocket_ref(), ast_websocket_remote_address(), buf, LOG_ERROR, NULL, ws_transport::rdata, status, ws_transport::transport, transport_create_data::transport, transport_dtor(), transport_type_wss, transport_type_wss_ipv6, ws_destroy(), ws_obj_name_serial, ws_send_msg(), ws_transport::ws_session, transport_create_data::ws_session, and ws_shutdown().

Referenced by load_module(), unload_module(), and websocket_cb().

◆ transport_dtor()

static void transport_dtor ( void *  arg)
static

Definition at line 112 of file res_pjsip_transport_websocket.c.

113{
114 struct ws_transport *wstransport = arg;
115
116 if (wstransport->ws_session) {
117 ast_websocket_unref(wstransport->ws_session);
118 }
119
120 if (wstransport->transport.ref_cnt) {
121 pj_atomic_destroy(wstransport->transport.ref_cnt);
122 }
123
124 if (wstransport->transport.lock) {
125 pj_lock_destroy(wstransport->transport.lock);
126 }
127
128 if (wstransport->transport.endpt && wstransport->transport.pool) {
129 pjsip_endpt_release_pool(wstransport->transport.endpt, wstransport->transport.pool);
130 }
131
132 if (wstransport->rdata.tp_info.pool) {
133 pjsip_endpt_release_pool(wstransport->transport.endpt, wstransport->rdata.tp_info.pool);
134 }
135}
void AST_OPTIONAL_API_NAME() ast_websocket_unref(struct ast_websocket *session)
Decrease the reference count for a WebSocket session.

References ast_websocket_unref(), ws_transport::rdata, ws_transport::transport, and ws_transport::ws_session.

Referenced by transport_create().

◆ transport_read()

static int transport_read ( void *  data)
static

Pass WebSocket data into pjsip transport manager.

Definition at line 295 of file res_pjsip_transport_websocket.c.

296{
297 struct transport_read_data *read_data = data;
298 struct ws_transport *newtransport = read_data->transport;
299 struct ast_websocket *session = newtransport->ws_session;
300
301 pjsip_rx_data *rdata = &newtransport->rdata;
302 int recvd;
303 pj_str_t buf;
304 int pjsip_pkt_len;
305
306 pj_gettimeofday(&rdata->pkt_info.timestamp);
307
308 pjsip_pkt_len = PJSIP_MAX_PKT_LEN < read_data->payload_len ? PJSIP_MAX_PKT_LEN : read_data->payload_len;
309 pj_memcpy(rdata->pkt_info.packet, read_data->payload, pjsip_pkt_len);
310 rdata->pkt_info.len = pjsip_pkt_len;
311 rdata->pkt_info.zero = 0;
312
313 pj_sockaddr_parse(pj_AF_UNSPEC(), 0, pj_cstr(&buf, ast_sockaddr_stringify(ast_websocket_remote_address(session))), &rdata->pkt_info.src_addr);
314 rdata->pkt_info.src_addr_len = sizeof(rdata->pkt_info.src_addr);
315
316 pj_ansi_strcpy(rdata->pkt_info.src_name, ast_sockaddr_stringify_addr(ast_websocket_remote_address(session)));
317 rdata->pkt_info.src_port = ast_sockaddr_port(ast_websocket_remote_address(session));
318
319 recvd = pjsip_tpmgr_receive_packet(rdata->tp_info.transport->tpmgr, rdata);
320
321 pj_pool_reset(rdata->tp_info.pool);
322
323 return (read_data->payload_len == recvd) ? 0 : -1;
324}
static struct ast_mansession session
Structure definition for session.

References ast_sockaddr_port, ast_sockaddr_stringify(), ast_sockaddr_stringify_addr(), ast_websocket_remote_address(), buf, transport_read_data::payload, transport_read_data::payload_len, ws_transport::rdata, session, transport_read_data::transport, and ws_transport::ws_session.

Referenced by websocket_cb().

◆ transport_shutdown()

static int transport_shutdown ( void *  data)
static

Definition at line 137 of file res_pjsip_transport_websocket.c.

138{
139 struct ws_transport *wstransport = data;
140
141 if (!wstransport->transport.is_shutdown && !wstransport->transport.is_destroying) {
142 pjsip_transport_shutdown(&wstransport->transport);
143 }
144
145 /* Note that the destructor calls PJSIP functions,
146 * therefore it must be called in a PJSIP thread.
147 */
148 ao2_ref(wstransport, -1);
149
150 return 0;
151}

References ao2_ref, and ws_transport::transport.

Referenced by websocket_cb().

◆ unload_module()

static int unload_module ( void  )
static

Definition at line 567 of file res_pjsip_transport_websocket.c.

568{
572
573 return 0;
574}
int AST_OPTIONAL_API_NAME() ast_websocket_remove_protocol(const char *name, ast_websocket_callback callback)
Remove a sub-protocol handler from the default /ws server.

References ast_sip_session_unregister_supplement(), ast_sip_unregister_service(), ast_websocket_remove_protocol(), websocket_cb(), websocket_module, and websocket_supplement.

◆ websocket_cb()

static void websocket_cb ( struct ast_websocket session,
struct ast_variable parameters,
struct ast_variable headers 
)
static

WebSocket connection handler.

Definition at line 375 of file res_pjsip_transport_websocket.c.

376{
378 struct transport_create_data create_data;
379 struct ws_transport *transport;
380 struct transport_read_data read_data;
381
384 return;
385 }
386
389 return;
390 }
391
393 if (!serializer) {
395 return;
396 }
397
398 create_data.ws_session = session;
399
401 ast_log(LOG_ERROR, "Could not create WebSocket transport.\n");
404 return;
405 }
406
407 transport = create_data.transport;
408 read_data.transport = transport;
409
410 pjsip_transport_add_ref(&transport->transport);
411 while (ast_websocket_wait_for_input(session, -1) > 0) {
412 enum ast_websocket_opcode opcode;
413 int fragmented;
414
415 if (ast_websocket_read(session, &read_data.payload, &read_data.payload_len, &opcode, &fragmented)) {
416 break;
417 }
418
419 if (opcode == AST_WEBSOCKET_OPCODE_TEXT || opcode == AST_WEBSOCKET_OPCODE_BINARY) {
420 if (read_data.payload_len) {
422 }
423 } else if (opcode == AST_WEBSOCKET_OPCODE_CLOSE) {
424 break;
425 }
426 }
427 pjsip_transport_dec_ref(&transport->transport);
428
430
433}
int ast_sip_push_task_wait_serializer(struct ast_taskprocessor *serializer, int(*sip_task)(void *), void *task_data)
Push a task to the serializer and wait for it to complete.
Definition res_pjsip.c:2179
int AST_OPTIONAL_API_NAME() ast_websocket_wait_for_input(struct ast_websocket *session, int timeout)
Wait for the WebSocket session to be ready to be read.
ast_websocket_opcode
WebSocket operation codes.
@ AST_WEBSOCKET_OPCODE_BINARY
@ AST_WEBSOCKET_OPCODE_CLOSE
@ AST_WEBSOCKET_OPCODE_TEXT
int AST_OPTIONAL_API_NAME() ast_websocket_set_nonblock(struct ast_websocket *session)
Set the socket of a WebSocket session to be non-blocking.
int AST_OPTIONAL_API_NAME() ast_websocket_set_timeout(struct ast_websocket *session, int timeout)
Set the timeout on a non-blocking WebSocket session.
int AST_OPTIONAL_API_NAME() ast_websocket_read(struct ast_websocket *session, char **payload, uint64_t *payload_len, enum ast_websocket_opcode *opcode, int *fragmented)
Read a WebSocket frame and handle it.
static int get_write_timeout(void)
static int transport_create(void *data)
Create a pjsip transport.
static struct ast_taskprocessor * create_websocket_serializer(void)
static int transport_shutdown(void *data)
static int transport_read(void *data)
Pass WebSocket data into pjsip transport manager.
A ast_taskprocessor structure is a singleton by name.
void * ast_taskprocessor_unreference(struct ast_taskprocessor *tps)
Unreference the specified taskprocessor and its reference count will decrement.

References ast_log, ast_sip_push_task_wait_serializer(), ast_taskprocessor_unreference(), AST_WEBSOCKET_OPCODE_BINARY, AST_WEBSOCKET_OPCODE_CLOSE, AST_WEBSOCKET_OPCODE_TEXT, ast_websocket_read(), ast_websocket_set_nonblock(), ast_websocket_set_timeout(), ast_websocket_unref(), ast_websocket_wait_for_input(), create_websocket_serializer(), get_write_timeout(), LOG_ERROR, transport_read_data::payload, transport_read_data::payload_len, session, ws_transport::transport, transport_create_data::transport, transport_read_data::transport, transport_create(), transport_read(), transport_shutdown(), and transport_create_data::ws_session.

Referenced by load_module(), and unload_module().

◆ websocket_on_rx_msg()

static pj_bool_t websocket_on_rx_msg ( pjsip_rx_data *  rdata)
static

Store the transport a message came in on, so it can be used for outbound messages to that contact.

Definition at line 464 of file res_pjsip_transport_websocket.c.

465{
466 static const pj_str_t STR_WS = { "ws", 2 };
467 pjsip_contact_hdr *contact;
468
469 long type = rdata->tp_info.transport->key.type;
470
471 if (type != (long) transport_type_wss && type != (long) transport_type_wss_ipv6) {
472 return PJ_FALSE;
473 }
474
475 contact = pjsip_msg_find_hdr(rdata->msg_info.msg, PJSIP_H_CONTACT, NULL);
476 if (contact
477 && !contact->star
478 && (PJSIP_URI_SCHEME_IS_SIP(contact->uri) || PJSIP_URI_SCHEME_IS_SIPS(contact->uri))) {
479 pjsip_sip_uri *uri = pjsip_uri_get_uri(contact->uri);
480 const pj_str_t *txp_str = &STR_WS;
481
482 /* Saving the contact on REGISTER so it can be restored on outbound response
483 * This will actually be done by restore_orig_contact_host in res_pjsip_nat, via nat_on_tx_message */
484 save_orig_contact_host(rdata, uri);
485
486 if (DEBUG_ATLEAST(4)) {
487 char src_addr_buffer[AST_SOCKADDR_BUFLEN];
488 const char *ipv6_s = "", *ipv6_e = "";
489
490 if (pj_strchr(&uri->host, ':')) {
491 ipv6_s = "[";
492 ipv6_e = "]";
493 }
494
495 ast_log(LOG_DEBUG, "%s re-writing Contact URI from %s%.*s%s:%d%s%.*s to %s;transport=%s\n",
496 pjsip_rx_data_get_info(rdata),
497 ipv6_s, (int) pj_strlen(&uri->host), pj_strbuf(&uri->host), ipv6_e, uri->port,
498 pj_strlen(&uri->transport_param) ? ";transport=" : "",
499 (int) pj_strlen(&uri->transport_param), pj_strbuf(&uri->transport_param),
500 pj_sockaddr_print(&rdata->pkt_info.src_addr, src_addr_buffer, sizeof(src_addr_buffer), 3),
501 pj_strbuf(txp_str));
502 }
503
504 pj_strdup2(rdata->tp_info.pool, &uri->host, rdata->pkt_info.src_name);
505 uri->port = rdata->pkt_info.src_port;
506 pj_strdup(rdata->tp_info.pool, &uri->transport_param, txp_str);
507 }
508
509 rdata->msg_info.via->rport_param = 0;
510
511 return PJ_FALSE;
512}
static const char type[]
#define DEBUG_ATLEAST(level)
#define LOG_DEBUG
#define AST_SOCKADDR_BUFLEN
Definition netsock2.h:46
static void save_orig_contact_host(pjsip_rx_data *rdata, pjsip_sip_uri *uri)

References ast_log, AST_SOCKADDR_BUFLEN, DEBUG_ATLEAST, LOG_DEBUG, NULL, save_orig_contact_host(), transport_type_wss, transport_type_wss_ipv6, and type.

◆ websocket_outgoing_invite_request()

static void websocket_outgoing_invite_request ( struct ast_sip_session session,
struct pjsip_tx_data *  tdata 
)
static

Function called when an INVITE goes out.

Definition at line 523 of file res_pjsip_transport_websocket.c.

524{
525 if (session->inv_session->state == PJSIP_INV_STATE_NULL) {
526 pjsip_dlg_add_usage(session->inv_session->dlg, &websocket_module, NULL);
527 }
528}

References NULL, session, and websocket_module.

◆ ws_destroy()

static pj_status_t ws_destroy ( pjsip_transport *  transport)
static

Destroy the pjsip transport.

Called by pjsip transport manager.

Definition at line 85 of file res_pjsip_transport_websocket.c.

86{
87 struct ws_transport *wstransport = (struct ws_transport *)transport;
88
89 ao2_ref(wstransport, -1);
90
91 return PJ_SUCCESS;
92}

References ao2_ref, and ws_transport::transport.

Referenced by transport_create().

◆ ws_send_msg()

static pj_status_t ws_send_msg ( pjsip_transport *  transport,
pjsip_tx_data *  tdata,
const pj_sockaddr_t *  rem_addr,
int  addr_len,
void *  token,
pjsip_transport_callback  callback 
)
static

Send a message over the WebSocket connection.

Called by pjsip transport manager.

Definition at line 63 of file res_pjsip_transport_websocket.c.

69{
70 struct ws_transport *wstransport = (struct ws_transport *)transport;
71 uint64_t len = tdata->buf.cur - tdata->buf.start;
72
73 if (ast_websocket_write(wstransport->ws_session, AST_WEBSOCKET_OPCODE_TEXT, tdata->buf.start, len)) {
74 return PJ_EUNKNOWN;
75 }
76
77 return PJ_SUCCESS;
78}
static int len(struct ast_channel *chan, const char *cmd, char *data, char *buf, size_t buflen)
int AST_OPTIONAL_API_NAME() ast_websocket_write(struct ast_websocket *session, enum ast_websocket_opcode opcode, char *payload, uint64_t payload_size)
Construct and transmit a WebSocket frame.
char buf[AST_WEBSOCKET_MAX_RX_PAYLOAD_SIZE]

References AST_WEBSOCKET_OPCODE_TEXT, ast_websocket_write(), ast_websocket::buf, len(), ws_transport::transport, and ws_transport::ws_session.

Referenced by transport_create().

◆ ws_shutdown()

static pj_status_t ws_shutdown ( pjsip_transport *  transport)
static

Shut down the pjsip transport.

Called by pjsip transport manager.

Definition at line 99 of file res_pjsip_transport_websocket.c.

100{
101 struct ws_transport *wstransport = (struct ws_transport *)transport;
102 int fd = ast_websocket_fd(wstransport->ws_session);
103
104 if (fd > 0) {
105 ast_websocket_close(wstransport->ws_session, 1000);
106 shutdown(fd, SHUT_RDWR);
107 }
108
109 return PJ_SUCCESS;
110}
int AST_OPTIONAL_API_NAME() ast_websocket_fd(struct ast_websocket *session)
Get the file descriptor for a WebSocket session.
int AST_OPTIONAL_API_NAME() ast_websocket_close(struct ast_websocket *session, uint16_t reason)
Close a WebSocket session by sending a message with the CLOSE opcode and an optional code.

References ast_websocket_close(), ast_websocket_fd(), ws_transport::transport, and ws_transport::ws_session.

Referenced by transport_create().

Variable Documentation

◆ __mod_info

struct ast_module_info __mod_info = { .name = AST_MODULE, .flags = AST_MODFLAG_LOAD_ORDER , .description = "PJSIP WebSocket Transport Support" , .key = ASTERISK_GPL_KEY , .buildopt_sum = AST_BUILDOPT_SUM, .support_level = AST_MODULE_SUPPORT_CORE, .load = load_module, .unload = unload_module, .load_pri = AST_MODPRI_APP_DEPEND, .requires = "res_pjsip,res_http_websocket", }
static

Definition at line 582 of file res_pjsip_transport_websocket.c.

◆ ast_module_info

const struct ast_module_info* ast_module_info = &__mod_info
static

Definition at line 582 of file res_pjsip_transport_websocket.c.

◆ transport_type_wss

int transport_type_wss
static

WebSocket transport module.

Definition at line 41 of file res_pjsip_transport_websocket.c.

Referenced by load_module(), transport_create(), and websocket_on_rx_msg().

◆ transport_type_wss_ipv6

int transport_type_wss_ipv6
static

◆ websocket_module

pjsip_module websocket_module
static

Definition at line 514 of file res_pjsip_transport_websocket.c.

514 {
515 .name = { "WebSocket Transport Module", 26 },
516 .id = -1,
517 .priority = PJSIP_MOD_PRIORITY_TRANSPORT_LAYER,
518 .on_rx_request = websocket_on_rx_msg,
519 .on_rx_response = websocket_on_rx_msg,
520};
static pj_bool_t websocket_on_rx_msg(pjsip_rx_data *rdata)
Store the transport a message came in on, so it can be used for outbound messages to that contact.

Referenced by load_module(), unload_module(), and websocket_outgoing_invite_request().

◆ websocket_supplement

struct ast_sip_session_supplement websocket_supplement
static
Initial value:
= {
.method = "INVITE",
.outgoing_request = websocket_outgoing_invite_request,
}
@ AST_SIP_SUPPLEMENT_PRIORITY_FIRST
Definition res_pjsip.h:3334
static void websocket_outgoing_invite_request(struct ast_sip_session *session, struct pjsip_tx_data *tdata)
Function called when an INVITE goes out.

Supplement for adding Websocket functionality to dialog.

Definition at line 531 of file res_pjsip_transport_websocket.c.

531 {
532 .method = "INVITE",
533 .priority = AST_SIP_SUPPLEMENT_PRIORITY_FIRST + 1,
534 .outgoing_request = websocket_outgoing_invite_request,
535};

Referenced by load_module(), and unload_module().

◆ ws_obj_name_serial

int ws_obj_name_serial
static

Used to ensure uniqueness among WS transport names

Definition at line 47 of file res_pjsip_transport_websocket.c.

Referenced by transport_create().