res/res_stasis: Fix crash on module unload while performing operation
[asterisk/asterisk.git] / res / res_stasis.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2012 - 2013, Digium, Inc.
5  *
6  * David M. Lee, II <dlee@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 /*! \file
20  *
21  * \brief Stasis application support.
22  *
23  * \author David M. Lee, II <dlee@digium.com>
24  *
25  * <code>res_stasis.so</code> brings together the various components of the
26  * Stasis application infrastructure.
27  *
28  * First, there's the Stasis application handler, stasis_app_exec(). This is
29  * called by <code>app_stasis.so</code> to give control of a channel to the
30  * Stasis application code from the dialplan.
31  *
32  * While a channel is in stasis_app_exec(), it has a \ref stasis_app_control
33  * object, which may be used to control the channel.
34  *
35  * To control the channel, commands may be sent to channel using
36  * stasis_app_send_command() and stasis_app_send_async_command().
37  *
38  * Alongside this, applications may be registered/unregistered using
39  * stasis_app_register()/stasis_app_unregister(). While a channel is in Stasis,
40  * events received on the channel's topic are converted to JSON and forwarded to
41  * the \ref stasis_app_cb. The application may also subscribe to the channel to
42  * continue to receive messages even after the channel has left Stasis, but it
43  * will not be able to control it.
44  *
45  * Given all the stuff that comes together in this module, it's been broken up
46  * into several pieces that are in <code>res/stasis/</code> and compiled into
47  * <code>res_stasis.so</code>.
48  */
49
50 /*** MODULEINFO
51         <support_level>core</support_level>
52  ***/
53
54 #include "asterisk.h"
55
56 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
57
58 #include "asterisk/astobj2.h"
59 #include "asterisk/callerid.h"
60 #include "asterisk/module.h"
61 #include "asterisk/stasis_app_impl.h"
62 #include "asterisk/stasis_channels.h"
63 #include "asterisk/stasis_bridges.h"
64 #include "asterisk/stasis_endpoints.h"
65 #include "asterisk/stasis_message_router.h"
66 #include "asterisk/strings.h"
67 #include "stasis/app.h"
68 #include "stasis/control.h"
69 #include "stasis/messaging.h"
70 #include "stasis/stasis_bridge.h"
71 #include "asterisk/core_unreal.h"
72 #include "asterisk/musiconhold.h"
73 #include "asterisk/causes.h"
74 #include "asterisk/stringfields.h"
75 #include "asterisk/bridge_after.h"
76 #include "asterisk/format_cache.h"
77
78 /*! Time to wait for a frame in the application */
79 #define MAX_WAIT_MS 200
80
81 /*!
82  * \brief Number of buckets for the Stasis application hash table.  Remember to
83  * keep it a prime number!
84  */
85 #define APPS_NUM_BUCKETS 127
86
87 /*!
88  * \brief Number of buckets for the Stasis application hash table.  Remember to
89  * keep it a prime number!
90  */
91 #define CONTROLS_NUM_BUCKETS 127
92
93 /*!
94  * \brief Number of buckets for the Stasis bridges hash table.  Remember to
95  * keep it a prime number!
96  */
97 #define BRIDGES_NUM_BUCKETS 127
98
99 /*!
100  * \brief Stasis application container.
101  */
102 struct ao2_container *apps_registry;
103
104 struct ao2_container *app_controls;
105
106 struct ao2_container *app_bridges;
107
108 struct ao2_container *app_bridges_moh;
109
110 struct ao2_container *app_bridges_playback;
111
112 static struct ast_json *stasis_end_json_payload(struct ast_channel_snapshot *snapshot,
113                 const struct stasis_message_sanitizer *sanitize)
114 {
115         return ast_json_pack("{s: s, s: o, s: o}",
116                 "type", "StasisEnd",
117                 "timestamp", ast_json_timeval(ast_tvnow(), NULL),
118                 "channel", ast_channel_snapshot_to_json(snapshot, sanitize));
119 }
120
121 static struct ast_json *stasis_end_to_json(struct stasis_message *message,
122                 const struct stasis_message_sanitizer *sanitize)
123 {
124         struct ast_channel_blob *payload = stasis_message_data(message);
125
126         if (sanitize && sanitize->channel_snapshot &&
127                         sanitize->channel_snapshot(payload->snapshot)) {
128                 return NULL;
129         }
130
131         return stasis_end_json_payload(payload->snapshot, sanitize);
132 }
133
134 STASIS_MESSAGE_TYPE_DEFN(ast_stasis_end_message_type,
135         .to_json = stasis_end_to_json);
136
137 const char *stasis_app_name(const struct stasis_app *app)
138 {
139         return app_name(app);
140 }
141
142 /*! AO2 hash function for \ref app */
143 static int app_hash(const void *obj, const int flags)
144 {
145         const struct stasis_app *app;
146         const char *key;
147
148         switch (flags & OBJ_SEARCH_MASK) {
149         case OBJ_SEARCH_KEY:
150                 key = obj;
151                 break;
152         case OBJ_SEARCH_OBJECT:
153                 app = obj;
154                 key = stasis_app_name(app);
155                 break;
156         default:
157                 /* Hash can only work on something with a full key. */
158                 ast_assert(0);
159                 return 0;
160         }
161         return ast_str_hash(key);
162 }
163
164 /*! AO2 comparison function for \ref app */
165 static int app_compare(void *obj, void *arg, int flags)
166 {
167         const struct stasis_app *object_left = obj;
168         const struct stasis_app *object_right = arg;
169         const char *right_key = arg;
170         int cmp;
171
172         switch (flags & OBJ_SEARCH_MASK) {
173         case OBJ_SEARCH_OBJECT:
174                 right_key = stasis_app_name(object_right);
175                 /* Fall through */
176         case OBJ_SEARCH_KEY:
177                 cmp = strcmp(stasis_app_name(object_left), right_key);
178                 break;
179         case OBJ_SEARCH_PARTIAL_KEY:
180                 /*
181                  * We could also use a partial key struct containing a length
182                  * so strlen() does not get called for every comparison instead.
183                  */
184                 cmp = strncmp(stasis_app_name(object_left), right_key, strlen(right_key));
185                 break;
186         default:
187                 /*
188                  * What arg points to is specific to this traversal callback
189                  * and has no special meaning to astobj2.
190                  */
191                 cmp = 0;
192                 break;
193         }
194         if (cmp) {
195                 return 0;
196         }
197         /*
198          * At this point the traversal callback is identical to a sorted
199          * container.
200          */
201         return CMP_MATCH;
202 }
203
204 /*! AO2 hash function for \ref stasis_app_control */
205 static int control_hash(const void *obj, const int flags)
206 {
207         const struct stasis_app_control *control;
208         const char *key;
209
210         switch (flags & OBJ_SEARCH_MASK) {
211         case OBJ_SEARCH_KEY:
212                 key = obj;
213                 break;
214         case OBJ_SEARCH_OBJECT:
215                 control = obj;
216                 key = stasis_app_control_get_channel_id(control);
217                 break;
218         default:
219                 /* Hash can only work on something with a full key. */
220                 ast_assert(0);
221                 return 0;
222         }
223         return ast_str_hash(key);
224 }
225
226 /*! AO2 comparison function for \ref stasis_app_control */
227 static int control_compare(void *obj, void *arg, int flags)
228 {
229         const struct stasis_app_control *object_left = obj;
230         const struct stasis_app_control *object_right = arg;
231         const char *right_key = arg;
232         int cmp;
233
234         switch (flags & OBJ_SEARCH_MASK) {
235         case OBJ_SEARCH_OBJECT:
236                 right_key = stasis_app_control_get_channel_id(object_right);
237                 /* Fall through */
238         case OBJ_SEARCH_KEY:
239                 cmp = strcmp(stasis_app_control_get_channel_id(object_left), right_key);
240                 break;
241         case OBJ_SEARCH_PARTIAL_KEY:
242                 /*
243                  * We could also use a partial key struct containing a length
244                  * so strlen() does not get called for every comparison instead.
245                  */
246                 cmp = strncmp(stasis_app_control_get_channel_id(object_left), right_key, strlen(right_key));
247                 break;
248         default:
249                 /*
250                  * What arg points to is specific to this traversal callback
251                  * and has no special meaning to astobj2.
252                  */
253                 cmp = 0;
254                 break;
255         }
256         if (cmp) {
257                 return 0;
258         }
259         /*
260          * At this point the traversal callback is identical to a sorted
261          * container.
262          */
263         return CMP_MATCH;
264 }
265
266 static int cleanup_cb(void *obj, void *arg, int flags)
267 {
268         struct stasis_app *app = obj;
269
270         if (!app_is_finished(app)) {
271                 return 0;
272         }
273
274         ast_verb(1, "Shutting down application '%s'\n", stasis_app_name(app));
275         app_shutdown(app);
276
277         return CMP_MATCH;
278
279 }
280
281 /*!
282  * \brief Clean up any old apps that we don't need any more.
283  */
284 static void cleanup(void)
285 {
286         ao2_callback(apps_registry, OBJ_MULTIPLE | OBJ_NODATA | OBJ_UNLINK,
287                 cleanup_cb, NULL);
288 }
289
290 struct stasis_app_control *stasis_app_control_create(struct ast_channel *chan)
291 {
292         return control_create(chan, NULL);
293 }
294
295 struct stasis_app_control *stasis_app_control_find_by_channel(
296         const struct ast_channel *chan)
297 {
298         if (chan == NULL) {
299                 return NULL;
300         }
301
302         return stasis_app_control_find_by_channel_id(
303                 ast_channel_uniqueid(chan));
304 }
305
306 struct stasis_app_control *stasis_app_control_find_by_channel_id(
307         const char *channel_id)
308 {
309         return ao2_find(app_controls, channel_id, OBJ_SEARCH_KEY);
310 }
311
312 /*! AO2 hash function for bridges container  */
313 static int bridges_hash(const void *obj, const int flags)
314 {
315         const struct ast_bridge *bridge;
316         const char *key;
317
318         switch (flags & OBJ_SEARCH_MASK) {
319         case OBJ_SEARCH_KEY:
320                 key = obj;
321                 break;
322         case OBJ_SEARCH_OBJECT:
323                 bridge = obj;
324                 key = bridge->uniqueid;
325                 break;
326         default:
327                 /* Hash can only work on something with a full key. */
328                 ast_assert(0);
329                 return 0;
330         }
331         return ast_str_hash(key);
332 }
333
334 /*! AO2 comparison function for bridges container */
335 static int bridges_compare(void *obj, void *arg, int flags)
336 {
337         const struct ast_bridge *object_left = obj;
338         const struct ast_bridge *object_right = arg;
339         const char *right_key = arg;
340         int cmp;
341
342         switch (flags & OBJ_SEARCH_MASK) {
343         case OBJ_SEARCH_OBJECT:
344                 right_key = object_right->uniqueid;
345                 /* Fall through */
346         case OBJ_SEARCH_KEY:
347                 cmp = strcmp(object_left->uniqueid, right_key);
348                 break;
349         case OBJ_SEARCH_PARTIAL_KEY:
350                 /*
351                  * We could also use a partial key struct containing a length
352                  * so strlen() does not get called for every comparison instead.
353                  */
354                 cmp = strncmp(object_left->uniqueid, right_key, strlen(right_key));
355                 break;
356         default:
357                 /*
358                  * What arg points to is specific to this traversal callback
359                  * and has no special meaning to astobj2.
360                  */
361                 cmp = 0;
362                 break;
363         }
364         if (cmp) {
365                 return 0;
366         }
367         /*
368          * At this point the traversal callback is identical to a sorted
369          * container.
370          */
371         return CMP_MATCH;
372 }
373
374 /*!
375  *  Used with app_bridges_moh and app_bridge_control, they provide links
376  *  between bridges and channels used for ARI application purposes
377  */
378 struct stasis_app_bridge_channel_wrapper {
379         AST_DECLARE_STRING_FIELDS(
380                 AST_STRING_FIELD(channel_id);
381                 AST_STRING_FIELD(bridge_id);
382         );
383 };
384
385 static void stasis_app_bridge_channel_wrapper_destructor(void *obj)
386 {
387         struct stasis_app_bridge_channel_wrapper *wrapper = obj;
388         ast_string_field_free_memory(wrapper);
389 }
390
391 /*! AO2 hash function for the bridges moh container */
392 static int bridges_channel_hash_fn(const void *obj, const int flags)
393 {
394         const struct stasis_app_bridge_channel_wrapper *wrapper;
395         const char *key;
396
397         switch (flags & OBJ_SEARCH_MASK) {
398         case OBJ_SEARCH_KEY:
399                 key = obj;
400                 break;
401         case OBJ_SEARCH_OBJECT:
402                 wrapper = obj;
403                 key = wrapper->bridge_id;
404                 break;
405         default:
406                 /* Hash can only work on something with a full key. */
407                 ast_assert(0);
408                 return 0;
409         }
410         return ast_str_hash(key);
411 }
412
413 static int bridges_channel_sort_fn(const void *obj_left, const void *obj_right, const int flags)
414 {
415         const struct stasis_app_bridge_channel_wrapper *left = obj_left;
416         const struct stasis_app_bridge_channel_wrapper *right = obj_right;
417         const char *right_key = obj_right;
418         int cmp;
419
420         switch (flags & OBJ_SEARCH_MASK) {
421         case OBJ_SEARCH_OBJECT:
422                 right_key = right->bridge_id;
423                 /* Fall through */
424         case OBJ_SEARCH_KEY:
425                 cmp = strcmp(left->bridge_id, right_key);
426                 break;
427         case OBJ_SEARCH_PARTIAL_KEY:
428                 cmp = strncmp(left->bridge_id, right_key, strlen(right_key));
429                 break;
430         default:
431                 /* Sort can only work on something with a full or partial key. */
432                 ast_assert(0);
433                 cmp = 0;
434                 break;
435         }
436         return cmp;
437 }
438
439 /*! Removes the bridge to music on hold channel link */
440 static void remove_bridge_moh(char *bridge_id)
441 {
442         ao2_find(app_bridges_moh, bridge_id, OBJ_SEARCH_KEY | OBJ_UNLINK | OBJ_NODATA);
443         ast_free(bridge_id);
444 }
445
446 /*! After bridge failure callback for moh channels */
447 static void moh_after_bridge_cb_failed(enum ast_bridge_after_cb_reason reason, void *data)
448 {
449         char *bridge_id = data;
450
451         remove_bridge_moh(bridge_id);
452 }
453
454 /*! After bridge callback for moh channels */
455 static void moh_after_bridge_cb(struct ast_channel *chan, void *data)
456 {
457         char *bridge_id = data;
458
459         remove_bridge_moh(bridge_id);
460 }
461
462 /*! Request a bridge MOH channel */
463 static struct ast_channel *prepare_bridge_moh_channel(void)
464 {
465         RAII_VAR(struct ast_format_cap *, cap, NULL, ao2_cleanup);
466
467         cap = ast_format_cap_alloc(AST_FORMAT_CAP_FLAG_DEFAULT);
468         if (!cap) {
469                 return NULL;
470         }
471
472         ast_format_cap_append(cap, ast_format_slin, 0);
473
474         return ast_request("Announcer", cap, NULL, NULL, "ARI_MOH", NULL);
475 }
476
477 /*! Provides the moh channel with a thread so it can actually play its music */
478 static void *moh_channel_thread(void *data)
479 {
480         struct ast_channel *moh_channel = data;
481
482         while (!ast_safe_sleep(moh_channel, 1000)) {
483         }
484
485         ast_moh_stop(moh_channel);
486         ast_hangup(moh_channel);
487
488         return NULL;
489 }
490
491 /*!
492  * \internal
493  * \brief Creates, pushes, and links a channel for playing music on hold to bridge
494  *
495  * \param bridge Which bridge this moh channel exists for
496  *
497  * \retval NULL if the channel could not be created, pushed, or linked
498  * \retval Reference to the channel on success
499  */
500 static struct ast_channel *bridge_moh_create(struct ast_bridge *bridge)
501 {
502         RAII_VAR(struct stasis_app_bridge_channel_wrapper *, new_wrapper, NULL, ao2_cleanup);
503         RAII_VAR(char *, bridge_id, ast_strdup(bridge->uniqueid), ast_free);
504         struct ast_channel *chan;
505         pthread_t threadid;
506
507         if (!bridge_id) {
508                 return NULL;
509         }
510
511         chan = prepare_bridge_moh_channel();
512         if (!chan) {
513                 return NULL;
514         }
515
516         if (stasis_app_channel_unreal_set_internal(chan)) {
517                 ast_hangup(chan);
518                 return NULL;
519         }
520
521         /* The after bridge callback assumes responsibility of the bridge_id. */
522         if (ast_bridge_set_after_callback(chan,
523                 moh_after_bridge_cb, moh_after_bridge_cb_failed, bridge_id)) {
524                 ast_hangup(chan);
525                 return NULL;
526         }
527         bridge_id = NULL;
528
529         if (ast_unreal_channel_push_to_bridge(chan, bridge,
530                 AST_BRIDGE_CHANNEL_FLAG_IMMOVABLE | AST_BRIDGE_CHANNEL_FLAG_LONELY)) {
531                 ast_hangup(chan);
532                 return NULL;
533         }
534
535         new_wrapper = ao2_alloc_options(sizeof(*new_wrapper),
536                 stasis_app_bridge_channel_wrapper_destructor, AO2_ALLOC_OPT_LOCK_NOLOCK);
537         if (!new_wrapper) {
538                 ast_hangup(chan);
539                 return NULL;
540         }
541
542         if (ast_string_field_init(new_wrapper, 32)) {
543                 ast_hangup(chan);
544                 return NULL;
545         }
546         ast_string_field_set(new_wrapper, bridge_id, bridge->uniqueid);
547         ast_string_field_set(new_wrapper, channel_id, ast_channel_uniqueid(chan));
548
549         if (!ao2_link_flags(app_bridges_moh, new_wrapper, OBJ_NOLOCK)) {
550                 ast_hangup(chan);
551                 return NULL;
552         }
553
554         if (ast_pthread_create_detached(&threadid, NULL, moh_channel_thread, chan)) {
555                 ast_log(LOG_ERROR, "Failed to create channel thread. Abandoning MOH channel creation.\n");
556                 ao2_unlink_flags(app_bridges_moh, new_wrapper, OBJ_NOLOCK);
557                 ast_hangup(chan);
558                 return NULL;
559         }
560
561         return chan;
562 }
563
564 struct ast_channel *stasis_app_bridge_moh_channel(struct ast_bridge *bridge)
565 {
566         RAII_VAR(struct stasis_app_bridge_channel_wrapper *, moh_wrapper, NULL, ao2_cleanup);
567
568         {
569                 SCOPED_AO2LOCK(lock, app_bridges_moh);
570
571                 moh_wrapper = ao2_find(app_bridges_moh, bridge->uniqueid, OBJ_SEARCH_KEY | OBJ_NOLOCK);
572                 if (!moh_wrapper) {
573                         return bridge_moh_create(bridge);
574                 }
575         }
576
577         return ast_channel_get_by_name(moh_wrapper->channel_id);
578 }
579
580 int stasis_app_bridge_moh_stop(struct ast_bridge *bridge)
581 {
582         RAII_VAR(struct stasis_app_bridge_channel_wrapper *, moh_wrapper, NULL, ao2_cleanup);
583         struct ast_channel *chan;
584
585         moh_wrapper = ao2_find(app_bridges_moh, bridge->uniqueid, OBJ_SEARCH_KEY | OBJ_UNLINK);
586         if (!moh_wrapper) {
587                 return -1;
588         }
589
590         chan = ast_channel_get_by_name(moh_wrapper->channel_id);
591         if (!chan) {
592                 return -1;
593         }
594
595         ast_moh_stop(chan);
596         ast_softhangup(chan, AST_CAUSE_NORMAL_CLEARING);
597         ao2_cleanup(chan);
598
599         return 0;
600 }
601
602 /*! Removes the bridge to playback channel link */
603 static void remove_bridge_playback(char *bridge_id)
604 {
605         struct stasis_app_bridge_channel_wrapper *wrapper;
606         struct stasis_app_control *control;
607
608         wrapper = ao2_find(app_bridges_playback, bridge_id, OBJ_SEARCH_KEY | OBJ_UNLINK);
609
610         if (wrapper) {
611                 control = stasis_app_control_find_by_channel_id(wrapper->channel_id);
612                 if (control) {
613                         ao2_unlink(app_controls, control);
614                         ao2_ref(control, -1);
615                 }
616                 ao2_ref(wrapper, -1);
617         }
618         ast_free(bridge_id);
619 }
620
621 static void playback_after_bridge_cb_failed(enum ast_bridge_after_cb_reason reason, void *data)
622 {
623         char *bridge_id = data;
624
625         remove_bridge_playback(bridge_id);
626 }
627
628 static void playback_after_bridge_cb(struct ast_channel *chan, void *data)
629 {
630         char *bridge_id = data;
631
632         remove_bridge_playback(bridge_id);
633 }
634
635 int stasis_app_bridge_playback_channel_add(struct ast_bridge *bridge,
636         struct ast_channel *chan,
637         struct stasis_app_control *control)
638 {
639         RAII_VAR(struct stasis_app_bridge_channel_wrapper *, new_wrapper, NULL, ao2_cleanup);
640         char *bridge_id = ast_strdup(bridge->uniqueid);
641
642         if (!bridge_id) {
643                 return -1;
644         }
645
646         if (ast_bridge_set_after_callback(chan,
647                 playback_after_bridge_cb, playback_after_bridge_cb_failed, bridge_id)) {
648                 ast_free(bridge_id);
649                 return -1;
650         }
651
652         new_wrapper = ao2_alloc_options(sizeof(*new_wrapper),
653                 stasis_app_bridge_channel_wrapper_destructor, AO2_ALLOC_OPT_LOCK_NOLOCK);
654         if (!new_wrapper) {
655                 return -1;
656         }
657
658         if (ast_string_field_init(new_wrapper, 32)) {
659                 return -1;
660         }
661
662         ast_string_field_set(new_wrapper, bridge_id, bridge->uniqueid);
663         ast_string_field_set(new_wrapper, channel_id, ast_channel_uniqueid(chan));
664
665         if (!ao2_link(app_bridges_playback, new_wrapper)) {
666                 return -1;
667         }
668
669         ao2_link(app_controls, control);
670         return 0;
671 }
672
673 struct ast_channel *stasis_app_bridge_playback_channel_find(struct ast_bridge *bridge)
674 {
675         struct stasis_app_bridge_channel_wrapper *playback_wrapper;
676         struct ast_channel *chan;
677
678         playback_wrapper = ao2_find(app_bridges_playback, bridge->uniqueid, OBJ_SEARCH_KEY);
679         if (!playback_wrapper) {
680                 return NULL;
681         }
682
683         chan = ast_channel_get_by_name(playback_wrapper->channel_id);
684         ao2_ref(playback_wrapper, -1);
685         return chan;
686 }
687
688 struct ast_bridge *stasis_app_bridge_find_by_id(
689         const char *bridge_id)
690 {
691         return ao2_find(app_bridges, bridge_id, OBJ_SEARCH_KEY);
692 }
693
694
695 /*!
696  * \brief In addition to running ao2_cleanup(), this function also removes the
697  * object from the app_controls container.
698  */
699 static void control_unlink(struct stasis_app_control *control)
700 {
701         if (!control) {
702                 return;
703         }
704
705         ao2_unlink(app_controls, control);
706         ao2_cleanup(control);
707 }
708
709 struct ast_bridge *stasis_app_bridge_create(const char *type, const char *name, const char *id)
710 {
711         struct ast_bridge *bridge;
712         char *requested_type, *requested_types = ast_strdupa(S_OR(type, "mixing"));
713         int capabilities = 0;
714         int flags = AST_BRIDGE_FLAG_MERGE_INHIBIT_FROM | AST_BRIDGE_FLAG_MERGE_INHIBIT_TO
715                 | AST_BRIDGE_FLAG_SWAP_INHIBIT_FROM | AST_BRIDGE_FLAG_SWAP_INHIBIT_TO
716                 | AST_BRIDGE_FLAG_TRANSFER_BRIDGE_ONLY;
717
718         while ((requested_type = strsep(&requested_types, ","))) {
719                 requested_type = ast_strip(requested_type);
720
721                 if (!strcmp(requested_type, "mixing")) {
722                         capabilities |= STASIS_BRIDGE_MIXING_CAPABILITIES;
723                         flags |= AST_BRIDGE_FLAG_SMART;
724                 } else if (!strcmp(requested_type, "holding")) {
725                         capabilities |= AST_BRIDGE_CAPABILITY_HOLDING;
726                 } else if (!strcmp(requested_type, "dtmf_events") ||
727                         !strcmp(requested_type, "proxy_media")) {
728                         capabilities &= ~AST_BRIDGE_CAPABILITY_NATIVE;
729                 }
730         }
731
732         if (!capabilities
733                 /* Holding and mixing capabilities don't mix. */
734                 || ((capabilities & AST_BRIDGE_CAPABILITY_HOLDING)
735                         && (capabilities & (STASIS_BRIDGE_MIXING_CAPABILITIES)))) {
736                 return NULL;
737         }
738
739         bridge = bridge_stasis_new(capabilities, flags, name, id);
740         if (bridge) {
741                 if (!ao2_link(app_bridges, bridge)) {
742                         ast_bridge_destroy(bridge, 0);
743                         bridge = NULL;
744                 }
745         }
746         return bridge;
747 }
748
749 void stasis_app_bridge_destroy(const char *bridge_id)
750 {
751         struct ast_bridge *bridge = stasis_app_bridge_find_by_id(bridge_id);
752         if (!bridge) {
753                 return;
754         }
755         ao2_unlink(app_bridges, bridge);
756         ast_bridge_destroy(bridge, 0);
757 }
758
759 struct replace_channel_store {
760         struct ast_channel_snapshot *snapshot;
761         char *app;
762 };
763
764 static void replace_channel_destroy(void *obj)
765 {
766         struct replace_channel_store *replace = obj;
767
768         ao2_cleanup(replace->snapshot);
769         ast_free(replace->app);
770         ast_free(replace);
771 }
772
773 static const struct ast_datastore_info replace_channel_store_info = {
774         .type = "replace-channel-store",
775         .destroy = replace_channel_destroy,
776 };
777
778 static struct replace_channel_store *get_replace_channel_store(struct ast_channel *chan, int no_create)
779 {
780         struct ast_datastore *datastore;
781
782         SCOPED_CHANNELLOCK(lock, chan);
783         datastore = ast_channel_datastore_find(chan, &replace_channel_store_info, NULL);
784         if (!datastore) {
785                 if (no_create) {
786                         return NULL;
787                 }
788
789                 datastore = ast_datastore_alloc(&replace_channel_store_info, NULL);
790                 if (!datastore) {
791                         return NULL;
792                 }
793                 ast_channel_datastore_add(chan, datastore);
794         }
795
796         if (!datastore->data) {
797                 datastore->data = ast_calloc(1, sizeof(struct replace_channel_store));
798         }
799         return datastore->data;
800 }
801
802 int app_set_replace_channel_snapshot(struct ast_channel *chan, struct ast_channel_snapshot *replace_snapshot)
803 {
804         struct replace_channel_store *replace = get_replace_channel_store(chan, 0);
805
806         if (!replace) {
807                 return -1;
808         }
809
810         ao2_replace(replace->snapshot, replace_snapshot);
811         return 0;
812 }
813
814 int app_set_replace_channel_app(struct ast_channel *chan, const char *replace_app)
815 {
816         struct replace_channel_store *replace = get_replace_channel_store(chan, 0);
817
818         if (!replace) {
819                 return -1;
820         }
821
822         ast_free(replace->app);
823         replace->app = NULL;
824
825         if (replace_app) {
826                 replace->app = ast_strdup(replace_app);
827                 if (!replace->app) {
828                         return -1;
829                 }
830         }
831
832         return 0;
833 }
834
835 static struct ast_channel_snapshot *get_replace_channel_snapshot(struct ast_channel *chan)
836 {
837         struct replace_channel_store *replace = get_replace_channel_store(chan, 1);
838         struct ast_channel_snapshot *replace_channel_snapshot;
839
840         if (!replace) {
841                 return NULL;
842         }
843
844         replace_channel_snapshot = replace->snapshot;
845         replace->snapshot = NULL;
846
847         return replace_channel_snapshot;
848 }
849
850 char *app_get_replace_channel_app(struct ast_channel *chan)
851 {
852         struct replace_channel_store *replace = get_replace_channel_store(chan, 1);
853         char *replace_channel_app;
854
855         if (!replace) {
856                 return NULL;
857         }
858
859         replace_channel_app = replace->app;
860         replace->app = NULL;
861
862         return replace_channel_app;
863 }
864
865 static int send_start_msg_snapshots(struct stasis_app *app,
866         int argc, char *argv[], struct ast_channel_snapshot *snapshot,
867         struct ast_channel_snapshot *replace_channel_snapshot)
868 {
869         RAII_VAR(struct ast_json *, msg, NULL, ast_json_unref);
870         struct ast_json *json_args;
871         struct stasis_message_sanitizer *sanitize = stasis_app_get_sanitizer();
872         int i;
873
874         if (sanitize && sanitize->channel_snapshot
875                 && sanitize->channel_snapshot(snapshot)) {
876                 return 0;
877         }
878
879         msg = ast_json_pack("{s: s, s: o, s: [], s: o}",
880                 "type", "StasisStart",
881                 "timestamp", ast_json_timeval(ast_tvnow(), NULL),
882                 "args",
883                 "channel", ast_channel_snapshot_to_json(snapshot, NULL));
884         if (!msg) {
885                 return -1;
886         }
887
888         if (replace_channel_snapshot) {
889                 int res = ast_json_object_set(msg, "replace_channel",
890                         ast_channel_snapshot_to_json(replace_channel_snapshot, NULL));
891
892                 if (res) {
893                         return -1;
894                 }
895         }
896
897         /* Append arguments to args array */
898         json_args = ast_json_object_get(msg, "args");
899         ast_assert(json_args != NULL);
900         for (i = 0; i < argc; ++i) {
901                 int r = ast_json_array_append(json_args,
902                                               ast_json_string_create(argv[i]));
903                 if (r != 0) {
904                         ast_log(LOG_ERROR, "Error appending start message\n");
905                         return -1;
906                 }
907         }
908
909         app_send(app, msg);
910         return 0;
911 }
912
913 static int send_start_msg(struct stasis_app *app, struct ast_channel *chan,
914         int argc, char *argv[])
915 {
916         RAII_VAR(struct ast_channel_snapshot *, snapshot, NULL, ao2_cleanup);
917         RAII_VAR(struct ast_channel_snapshot *, replace_channel_snapshot,
918                 NULL, ao2_cleanup);
919
920         ast_assert(chan != NULL);
921
922         replace_channel_snapshot = get_replace_channel_snapshot(chan);
923
924         /* Set channel info */
925         ast_channel_lock(chan);
926         snapshot = ast_channel_snapshot_create(chan);
927         ast_channel_unlock(chan);
928         if (!snapshot) {
929                 return -1;
930         }
931         return send_start_msg_snapshots(app, argc, argv, snapshot, replace_channel_snapshot);
932 }
933
934 static int send_end_msg_snapshot(struct stasis_app *app, struct ast_channel_snapshot *snapshot)
935 {
936         struct stasis_message_sanitizer *sanitize = stasis_app_get_sanitizer();
937         struct ast_json *msg;
938
939         if (sanitize && sanitize->channel_snapshot
940                 && sanitize->channel_snapshot(snapshot)) {
941                 return 0;
942         }
943
944         msg = stasis_end_json_payload(snapshot, sanitize);
945         if (!msg) {
946                 return -1;
947         }
948
949         app_send(app, msg);
950         ast_json_unref(msg);
951         return 0;
952 }
953
954 static void remove_masquerade_store(struct ast_channel *chan);
955
956 static int masq_match_cb(void *obj, void *data, int flags)
957 {
958         struct stasis_app_control *control = obj;
959         struct ast_channel *chan = data;
960
961         if (!strcmp(ast_channel_uniqueid(chan),
962                 stasis_app_control_get_channel_id(control))) {
963                 return CMP_MATCH;
964         }
965
966         return 0;
967 }
968
969 static void channel_stolen_cb(void *data, struct ast_channel *old_chan, struct ast_channel *new_chan)
970 {
971         struct ast_channel_snapshot *snapshot;
972         struct stasis_app_control *control;
973
974         /* grab a snapshot */
975         snapshot = ast_channel_snapshot_get_latest(ast_channel_uniqueid(new_chan));
976         if (!snapshot) {
977                 ast_log(LOG_ERROR, "Could not get snapshot for masqueraded channel\n");
978                 return;
979         }
980
981         /* find control */
982         control = ao2_callback(app_controls, 0, masq_match_cb, old_chan);
983         if (!control) {
984                 ast_log(LOG_ERROR, "Could not find control for masqueraded channel\n");
985                 ao2_cleanup(snapshot);
986                 return;
987         }
988
989         /* send the StasisEnd message to the app */
990         send_end_msg_snapshot(control_app(control), snapshot);
991
992         /* remove the datastore */
993         remove_masquerade_store(old_chan);
994
995         ao2_cleanup(control);
996         ao2_cleanup(snapshot);
997 }
998
999 static void channel_replaced_cb(void *data, struct ast_channel *old_chan, struct ast_channel *new_chan)
1000 {
1001         RAII_VAR(struct ast_channel_snapshot *, new_snapshot, NULL, ao2_cleanup);
1002         RAII_VAR(struct ast_channel_snapshot *, old_snapshot, NULL, ao2_cleanup);
1003         struct stasis_app_control *control;
1004
1005         /* At this point, new_chan is the channel pointer that is in Stasis() and
1006          * has the unknown channel's name in it while old_chan is the channel pointer
1007          * that is not in Stasis(), but has the guts of the channel that Stasis() knows
1008          * about */
1009
1010         /* grab a snapshot for the channel that is jumping into Stasis() */
1011         new_snapshot = ast_channel_snapshot_get_latest(ast_channel_uniqueid(new_chan));
1012         if (!new_snapshot) {
1013                 ast_log(LOG_ERROR, "Could not get snapshot for masquerading channel\n");
1014                 return;
1015         }
1016
1017         /* grab a snapshot for the channel that has been kicked out of Stasis() */
1018         old_snapshot = ast_channel_snapshot_get_latest(ast_channel_uniqueid(old_chan));
1019         if (!old_snapshot) {
1020                 ast_log(LOG_ERROR, "Could not get snapshot for masqueraded channel\n");
1021                 return;
1022         }
1023
1024         /* find, unlink, and relink control since the channel has a new name and
1025          * its hash has likely changed */
1026         control = ao2_callback(app_controls, OBJ_UNLINK, masq_match_cb, new_chan);
1027         if (!control) {
1028                 ast_log(LOG_ERROR, "Could not find control for masquerading channel\n");
1029                 return;
1030         }
1031         ao2_link(app_controls, control);
1032
1033
1034         /* send the StasisStart with replace_channel to the app */
1035         send_start_msg_snapshots(control_app(control), 0, NULL, new_snapshot,
1036                 old_snapshot);
1037         /* send the StasisEnd message to the app */
1038         send_end_msg_snapshot(control_app(control), old_snapshot);
1039
1040         /* fixup channel topic forwards */
1041         if (app_replace_channel_forwards(control_app(control), old_snapshot->uniqueid, new_chan)) {
1042                 ast_log(LOG_ERROR, "Failed to fixup channel topic forwards for %s(%s) owned by %s\n",
1043                         old_snapshot->name, old_snapshot->uniqueid, app_name(control_app(control)));
1044         }
1045         ao2_cleanup(control);
1046 }
1047
1048 static const struct ast_datastore_info masquerade_store_info = {
1049         .type = "stasis-masqerade",
1050         .chan_fixup = channel_stolen_cb,
1051         .chan_breakdown = channel_replaced_cb,
1052 };
1053
1054 static int has_masquerade_store(struct ast_channel *chan)
1055 {
1056         SCOPED_CHANNELLOCK(lock, chan);
1057         return !!ast_channel_datastore_find(chan, &masquerade_store_info, NULL);
1058 }
1059
1060 static int add_masquerade_store(struct ast_channel *chan)
1061 {
1062         struct ast_datastore *datastore;
1063
1064         SCOPED_CHANNELLOCK(lock, chan);
1065         if (ast_channel_datastore_find(chan, &masquerade_store_info, NULL)) {
1066                 return 0;
1067         }
1068
1069         datastore = ast_datastore_alloc(&masquerade_store_info, NULL);
1070         if (!datastore) {
1071                 return -1;
1072         }
1073
1074         ast_channel_datastore_add(chan, datastore);
1075
1076         return 0;
1077 }
1078
1079 static void remove_masquerade_store(struct ast_channel *chan)
1080 {
1081         struct ast_datastore *datastore;
1082
1083         SCOPED_CHANNELLOCK(lock, chan);
1084         datastore = ast_channel_datastore_find(chan, &masquerade_store_info, NULL);
1085         if (!datastore) {
1086                 return;
1087         }
1088
1089         ast_channel_datastore_remove(chan, datastore);
1090         ast_datastore_free(datastore);
1091 }
1092
1093 static int send_end_msg(struct stasis_app *app, struct ast_channel *chan)
1094 {
1095         struct ast_channel_snapshot *snapshot;
1096         int res = 0;
1097
1098         ast_assert(chan != NULL);
1099
1100         /* A masquerade has occurred and this message will be wrong so it
1101          * has already been sent elsewhere. */
1102         if (!has_masquerade_store(chan)) {
1103                 return 0;
1104         }
1105
1106         /* Set channel info */
1107         snapshot = ast_channel_snapshot_get_latest(ast_channel_uniqueid(chan));
1108         if (!snapshot) {
1109                 return -1;
1110         }
1111
1112         if (send_end_msg_snapshot(app, snapshot)) {
1113                 res = -1;
1114         }
1115
1116         ao2_cleanup(snapshot);
1117         return res;
1118 }
1119
1120 void stasis_app_control_execute_until_exhausted(struct ast_channel *chan, struct stasis_app_control *control)
1121 {
1122         while (!control_is_done(control)) {
1123                 int command_count;
1124                 command_count = control_dispatch_all(control, chan);
1125
1126                 ao2_lock(control);
1127
1128                 if (control_command_count(control)) {
1129                         /* If the command queue isn't empty, something added to the queue before it was locked. */
1130                         ao2_unlock(control);
1131                         continue;
1132                 }
1133
1134                 if (command_count == 0 || ast_channel_fdno(chan) == -1) {
1135                         control_mark_done(control);
1136                         ao2_unlock(control);
1137                         break;
1138                 }
1139                 ao2_unlock(control);
1140         }
1141 }
1142
1143 int stasis_app_control_is_done(struct stasis_app_control *control)
1144 {
1145         return control_is_done(control);
1146 }
1147
1148 struct ast_datastore_info set_end_published_info = {
1149         .type = "stasis_end_published",
1150 };
1151
1152 void stasis_app_channel_set_stasis_end_published(struct ast_channel *chan)
1153 {
1154         struct ast_datastore *datastore;
1155
1156         datastore = ast_datastore_alloc(&set_end_published_info, NULL);
1157
1158         ast_channel_lock(chan);
1159         ast_channel_datastore_add(chan, datastore);
1160         ast_channel_unlock(chan);
1161 }
1162
1163 int stasis_app_channel_is_stasis_end_published(struct ast_channel *chan)
1164 {
1165         struct ast_datastore *datastore;
1166
1167         ast_channel_lock(chan);
1168         datastore = ast_channel_datastore_find(chan, &set_end_published_info, NULL);
1169         ast_channel_unlock(chan);
1170
1171         return datastore ? 1 : 0;
1172 }
1173
1174 static void remove_stasis_end_published(struct ast_channel *chan)
1175 {
1176         struct ast_datastore *datastore;
1177
1178         ast_channel_lock(chan);
1179         datastore = ast_channel_datastore_find(chan, &set_end_published_info, NULL);
1180         ast_channel_unlock(chan);
1181
1182         if (datastore) {
1183                 ast_channel_datastore_remove(chan, datastore);
1184                 ast_datastore_free(datastore);
1185         }
1186 }
1187
1188 /*! /brief Stasis dialplan application callback */
1189 int stasis_app_exec(struct ast_channel *chan, const char *app_name, int argc,
1190                     char *argv[])
1191 {
1192         SCOPED_MODULE_USE(ast_module_info->self);
1193
1194         RAII_VAR(struct stasis_app *, app, NULL, ao2_cleanup);
1195         RAII_VAR(struct stasis_app_control *, control, NULL, control_unlink);
1196         struct ast_bridge *bridge = NULL;
1197         int res = 0;
1198         int needs_depart;
1199
1200         ast_assert(chan != NULL);
1201
1202         /* Just in case there's a lingering indication that the channel has had a stasis
1203          * end published on it, remove that now.
1204          */
1205         remove_stasis_end_published(chan);
1206
1207         if (!apps_registry) {
1208                 return -1;
1209         }
1210
1211         app = ao2_find(apps_registry, app_name, OBJ_SEARCH_KEY);
1212         if (!app) {
1213                 ast_log(LOG_ERROR,
1214                         "Stasis app '%s' not registered\n", app_name);
1215                 return -1;
1216         }
1217         if (!app_is_active(app)) {
1218                 ast_log(LOG_ERROR,
1219                         "Stasis app '%s' not active\n", app_name);
1220                 return -1;
1221         }
1222
1223         control = control_create(chan, app);
1224         if (!control) {
1225                 ast_log(LOG_ERROR, "Allocated failed\n");
1226                 return -1;
1227         }
1228         ao2_link(app_controls, control);
1229
1230         if (add_masquerade_store(chan)) {
1231                 ast_log(LOG_ERROR, "Failed to attach masquerade detector\n");
1232                 return -1;
1233         }
1234
1235         res = send_start_msg(app, chan, argc, argv);
1236         if (res != 0) {
1237                 ast_log(LOG_ERROR,
1238                         "Error sending start message to '%s'\n", app_name);
1239                 remove_masquerade_store(chan);
1240                 return -1;
1241         }
1242
1243         res = app_subscribe_channel(app, chan);
1244         if (res != 0) {
1245                 ast_log(LOG_ERROR, "Error subscribing app '%s' to channel '%s'\n",
1246                         app_name, ast_channel_name(chan));
1247                 remove_masquerade_store(chan);
1248                 return -1;
1249         }
1250
1251         /* Pull queued prestart commands and execute */
1252         control_prestart_dispatch_all(control, chan);
1253
1254         while (!control_is_done(control)) {
1255                 RAII_VAR(struct ast_frame *, f, NULL, ast_frame_dtor);
1256                 int r;
1257                 int command_count;
1258                 RAII_VAR(struct ast_bridge *, last_bridge, NULL, ao2_cleanup);
1259
1260                 /* Check to see if a bridge absorbed our hangup frame */
1261                 if (ast_check_hangup_locked(chan)) {
1262                         break;
1263                 }
1264
1265                 last_bridge = bridge;
1266                 bridge = ao2_bump(stasis_app_get_bridge(control));
1267
1268                 if (bridge != last_bridge) {
1269                         app_unsubscribe_bridge(app, last_bridge);
1270                         app_subscribe_bridge(app, bridge);
1271                 }
1272
1273                 if (bridge) {
1274                         /* Bridge is handling channel frames */
1275                         control_wait(control);
1276                         control_dispatch_all(control, chan);
1277                         continue;
1278                 }
1279
1280                 r = ast_waitfor(chan, MAX_WAIT_MS);
1281
1282                 if (r < 0) {
1283                         ast_debug(3, "%s: Poll error\n",
1284                                   ast_channel_uniqueid(chan));
1285                         break;
1286                 }
1287
1288                 command_count = control_dispatch_all(control, chan);
1289
1290                 if (command_count > 0 && ast_channel_fdno(chan) == -1) {
1291                         /* Command drained the channel; wait for next frame */
1292                         continue;
1293                 }
1294
1295                 if (r == 0) {
1296                         /* Timeout */
1297                         continue;
1298                 }
1299
1300                 f = ast_read(chan);
1301                 if (!f) {
1302                         /* Continue on in the dialplan */
1303                         ast_debug(3, "%s: Hangup (no more frames)\n",
1304                                 ast_channel_uniqueid(chan));
1305                         break;
1306                 }
1307
1308                 if (f->frametype == AST_FRAME_CONTROL) {
1309                         if (f->subclass.integer == AST_CONTROL_HANGUP) {
1310                                 /* Continue on in the dialplan */
1311                                 ast_debug(3, "%s: Hangup\n",
1312                                         ast_channel_uniqueid(chan));
1313                                 break;
1314                         }
1315                 }
1316         }
1317
1318         ast_channel_lock(chan);
1319         needs_depart = ast_channel_is_bridged(chan);
1320         ast_channel_unlock(chan);
1321         if (needs_depart) {
1322                 ast_bridge_depart(chan);
1323         }
1324
1325         app_unsubscribe_bridge(app, stasis_app_get_bridge(control));
1326         ao2_cleanup(bridge);
1327
1328         /* Only publish a stasis_end event if it hasn't already been published */
1329         if (!stasis_app_channel_is_stasis_end_published(chan)) {
1330                 app_unsubscribe_channel(app, chan);
1331                 res = send_end_msg(app, chan);
1332                 remove_masquerade_store(chan);
1333                 if (res != 0) {
1334                         ast_log(LOG_ERROR,
1335                                 "Error sending end message to %s\n", app_name);
1336                         return res;
1337                 }
1338         } else {
1339                 remove_stasis_end_published(chan);
1340         }
1341
1342         /* There's an off chance that app is ready for cleanup. Go ahead
1343          * and clean up, just in case
1344          */
1345         cleanup();
1346
1347         /* The control needs to be removed from the controls container in
1348          * case a new PBX is started and ends up coming back into Stasis.
1349          */
1350         ao2_cleanup(app);
1351         app = NULL;
1352         control_unlink(control);
1353         control = NULL;
1354
1355         if (!ast_check_hangup_locked(chan) && !ast_channel_pbx(chan)) {
1356                 struct ast_pbx_args pbx_args;
1357
1358                 memset(&pbx_args, 0, sizeof(pbx_args));
1359                 pbx_args.no_hangup_chan = 1;
1360
1361                 res = ast_pbx_run_args(chan, &pbx_args);
1362         }
1363
1364         return res;
1365 }
1366
1367 int stasis_app_send(const char *app_name, struct ast_json *message)
1368 {
1369         RAII_VAR(struct stasis_app *, app, NULL, ao2_cleanup);
1370
1371         if (!apps_registry) {
1372                 return -1;
1373         }
1374
1375         app = ao2_find(apps_registry, app_name, OBJ_SEARCH_KEY);
1376         if (!app) {
1377                 /* XXX We can do a better job handling late binding, queueing up
1378                  * the call for a few seconds to wait for the app to register.
1379                  */
1380                 ast_log(LOG_WARNING,
1381                         "Stasis app '%s' not registered\n", app_name);
1382                 return -1;
1383         }
1384         app_send(app, message);
1385         return 0;
1386 }
1387
1388 static struct stasis_app *find_app_by_name(const char *app_name)
1389 {
1390         struct stasis_app *res = NULL;
1391
1392         if (!apps_registry) {
1393                 return NULL;
1394         }
1395
1396         if (!ast_strlen_zero(app_name)) {
1397                 res = ao2_find(apps_registry, app_name, OBJ_SEARCH_KEY);
1398         }
1399
1400         if (!res) {
1401                 ast_log(LOG_WARNING, "Could not find app '%s'\n",
1402                         app_name ? : "(null)");
1403         }
1404         return res;
1405 }
1406
1407 static int append_name(void *obj, void *arg, int flags)
1408 {
1409         struct stasis_app *app = obj;
1410         struct ao2_container *apps = arg;
1411
1412         ast_str_container_add(apps, stasis_app_name(app));
1413         return 0;
1414 }
1415
1416 struct ao2_container *stasis_app_get_all(void)
1417 {
1418         RAII_VAR(struct ao2_container *, apps, NULL, ao2_cleanup);
1419
1420         if (!apps_registry) {
1421                 return NULL;
1422         }
1423
1424         apps = ast_str_container_alloc(1);
1425         if (!apps) {
1426                 return NULL;
1427         }
1428
1429         ao2_callback(apps_registry, OBJ_NODATA, append_name, apps);
1430
1431         return ao2_bump(apps);
1432 }
1433
1434 int stasis_app_register(const char *app_name, stasis_app_cb handler, void *data)
1435 {
1436         RAII_VAR(struct stasis_app *, app, NULL, ao2_cleanup);
1437
1438         if (!apps_registry) {
1439                 return -1;
1440         }
1441
1442         ao2_lock(apps_registry);
1443         app = ao2_find(apps_registry, app_name, OBJ_SEARCH_KEY | OBJ_NOLOCK);
1444         if (app) {
1445                 app_update(app, handler, data);
1446         } else {
1447                 app = app_create(app_name, handler, data);
1448                 if (app) {
1449                         ao2_link_flags(apps_registry, app, OBJ_NOLOCK);
1450                 } else {
1451                         ao2_unlock(apps_registry);
1452                         return -1;
1453                 }
1454         }
1455
1456         /* We lazily clean up the apps_registry, because it's good enough to
1457          * prevent memory leaks, and we're lazy.
1458          */
1459         cleanup();
1460         ao2_unlock(apps_registry);
1461         return 0;
1462 }
1463
1464 void stasis_app_unregister(const char *app_name)
1465 {
1466         RAII_VAR(struct stasis_app *, app, NULL, ao2_cleanup);
1467
1468         if (!app_name) {
1469                 return;
1470         }
1471
1472         if (!apps_registry) {
1473                 return;
1474         }
1475
1476         app = ao2_find(apps_registry, app_name, OBJ_SEARCH_KEY);
1477         if (!app) {
1478                 ast_log(LOG_ERROR,
1479                         "Stasis app '%s' not registered\n", app_name);
1480                 return;
1481         }
1482
1483         app_deactivate(app);
1484
1485         /* There's a decent chance that app is ready for cleanup. Go ahead
1486          * and clean up, just in case
1487          */
1488         cleanup();
1489 }
1490
1491 /*!
1492  * \internal \brief List of registered event sources.
1493  */
1494 AST_RWLIST_HEAD_STATIC(event_sources, stasis_app_event_source);
1495
1496 void stasis_app_register_event_source(struct stasis_app_event_source *obj)
1497 {
1498         SCOPED_LOCK(lock, &event_sources, AST_RWLIST_WRLOCK, AST_RWLIST_UNLOCK);
1499         AST_LIST_INSERT_TAIL(&event_sources, obj, next);
1500         /* only need to bump the module ref on non-core sources because the
1501            core ones are [un]registered by this module. */
1502         if (!stasis_app_is_core_event_source(obj)) {
1503                 ast_module_ref(ast_module_info->self);
1504         }
1505 }
1506
1507 void stasis_app_unregister_event_source(struct stasis_app_event_source *obj)
1508 {
1509         struct stasis_app_event_source *source;
1510         SCOPED_LOCK(lock, &event_sources, AST_RWLIST_WRLOCK, AST_RWLIST_UNLOCK);
1511         AST_RWLIST_TRAVERSE_SAFE_BEGIN(&event_sources, source, next) {
1512                 if (source == obj) {
1513                         AST_RWLIST_REMOVE_CURRENT(next);
1514                         if (!stasis_app_is_core_event_source(obj)) {
1515                                 ast_module_unref(ast_module_info->self);
1516                         }
1517                         break;
1518                 }
1519         }
1520         AST_RWLIST_TRAVERSE_SAFE_END;
1521 }
1522
1523 /*!
1524  * \internal
1525  * \brief Convert event source data to JSON.
1526  *
1527  * Calls each event source that has a "to_json" handler allowing each
1528  * source to add data to the given JSON object.
1529  *
1530  * \param app application associated with the event source
1531  * \param json a json object to "fill"
1532  *
1533  * \retval The given json object.
1534  */
1535 static struct ast_json *app_event_sources_to_json(
1536         const struct stasis_app *app, struct ast_json *json)
1537 {
1538         struct stasis_app_event_source *source;
1539         SCOPED_LOCK(lock, &event_sources, AST_RWLIST_RDLOCK, AST_RWLIST_UNLOCK);
1540         AST_LIST_TRAVERSE(&event_sources, source, next) {
1541                 if (source->to_json) {
1542                         source->to_json(app, json);
1543                 }
1544         }
1545         return json;
1546 }
1547
1548 static struct ast_json *stasis_app_object_to_json(struct stasis_app *app)
1549 {
1550         if (!app) {
1551                 return NULL;
1552         }
1553
1554         return app_event_sources_to_json(app, app_to_json(app));
1555 }
1556
1557 struct ast_json *stasis_app_to_json(const char *app_name)
1558 {
1559         RAII_VAR(struct stasis_app *, app, find_app_by_name(app_name), ao2_cleanup);
1560
1561         return stasis_app_object_to_json(app);
1562 }
1563
1564 /*!
1565  * \internal
1566  * \brief Finds an event source that matches a uri scheme.
1567  *
1568  * Uri(s) should begin with a particular scheme that can be matched
1569  * against an event source.
1570  *
1571  * \param uri uri containing a scheme to match
1572  *
1573  * \retval an event source if found, NULL otherwise.
1574  */
1575 static struct stasis_app_event_source *app_event_source_find(const char *uri)
1576 {
1577         struct stasis_app_event_source *source;
1578         SCOPED_LOCK(lock, &event_sources, AST_RWLIST_RDLOCK, AST_RWLIST_UNLOCK);
1579         AST_LIST_TRAVERSE(&event_sources, source, next) {
1580                 if (ast_begins_with(uri, source->scheme)) {
1581                         return source;
1582                 }
1583         }
1584         return NULL;
1585 }
1586
1587 /*!
1588  * \internal
1589  * \brief Callback for subscription handling
1590  *
1591  * \param app [un]subscribing application
1592  * \param uri scheme:id of an event source
1593  * \param event_source being [un]subscribed [from]to
1594  *
1595  * \retval stasis_app_subscribe_res return code.
1596  */
1597 typedef enum stasis_app_subscribe_res (*app_subscription_handler)(
1598         struct stasis_app *app, const char *uri,
1599         struct stasis_app_event_source *event_source);
1600
1601 /*!
1602  * \internal
1603  * \brief Subscriptions handler for application [un]subscribing.
1604  *
1605  * \param app_name Name of the application to subscribe.
1606  * \param event_source_uris URIs for the event sources to subscribe to.
1607  * \param event_sources_count Array size of event_source_uris.
1608  * \param json Optional output pointer for JSON representation of the app
1609  *             after adding the subscription.
1610  * \param handler [un]subscribe handler
1611  *
1612  * \retval stasis_app_subscribe_res return code.
1613  */
1614 static enum stasis_app_subscribe_res app_handle_subscriptions(
1615         const char *app_name, const char **event_source_uris,
1616         int event_sources_count, struct ast_json **json,
1617         app_subscription_handler handler)
1618 {
1619         RAII_VAR(struct stasis_app *, app, find_app_by_name(app_name), ao2_cleanup);
1620         int i;
1621
1622         if (!app) {
1623                 return STASIS_ASR_APP_NOT_FOUND;
1624         }
1625
1626         for (i = 0; i < event_sources_count; ++i) {
1627                 const char *uri = event_source_uris[i];
1628                 enum stasis_app_subscribe_res res = STASIS_ASR_INTERNAL_ERROR;
1629                 struct stasis_app_event_source *event_source;
1630
1631                 if (!(event_source = app_event_source_find(uri))) {
1632                         ast_log(LOG_WARNING, "Invalid scheme: %s\n", uri);
1633                         return STASIS_ASR_EVENT_SOURCE_BAD_SCHEME;
1634                 }
1635
1636                 if (handler &&
1637                     ((res = handler(app, uri, event_source)))) {
1638                         return res;
1639                 }
1640         }
1641
1642         if (json) {
1643                 ast_debug(3, "%s: Successful; setting results\n", app_name);
1644                 *json = stasis_app_object_to_json(app);
1645         }
1646
1647         return STASIS_ASR_OK;
1648 }
1649
1650 enum stasis_app_subscribe_res stasis_app_subscribe_channel(const char *app_name,
1651         struct ast_channel *chan)
1652 {
1653         RAII_VAR(struct stasis_app *, app, find_app_by_name(app_name), ao2_cleanup);
1654         int res;
1655
1656         if (!app) {
1657                 return STASIS_ASR_APP_NOT_FOUND;
1658         }
1659
1660         ast_debug(3, "%s: Subscribing to %s\n", app_name, ast_channel_uniqueid(chan));
1661
1662         res = app_subscribe_channel(app, chan);
1663         if (res != 0) {
1664                 ast_log(LOG_ERROR, "Error subscribing app '%s' to channel '%s'\n",
1665                         app_name, ast_channel_uniqueid(chan));
1666                 return STASIS_ASR_INTERNAL_ERROR;
1667         }
1668
1669         return STASIS_ASR_OK;
1670 }
1671
1672
1673 /*!
1674  * \internal
1675  * \brief Subscribe an app to an event source.
1676  *
1677  * \param app subscribing application
1678  * \param uri scheme:id of an event source
1679  * \param event_source being subscribed to
1680  *
1681  * \retval stasis_app_subscribe_res return code.
1682  */
1683 static enum stasis_app_subscribe_res app_subscribe(
1684         struct stasis_app *app, const char *uri,
1685         struct stasis_app_event_source *event_source)
1686 {
1687         const char *app_name = stasis_app_name(app);
1688         RAII_VAR(void *, obj, NULL, ao2_cleanup);
1689
1690         ast_debug(3, "%s: Checking %s\n", app_name, uri);
1691
1692         if (!event_source->find ||
1693             (!(obj = event_source->find(app, uri + strlen(event_source->scheme))))) {
1694                 ast_log(LOG_WARNING, "Event source not found: %s\n", uri);
1695                 return STASIS_ASR_EVENT_SOURCE_NOT_FOUND;
1696         }
1697
1698         ast_debug(3, "%s: Subscribing to %s\n", app_name, uri);
1699
1700         if (!event_source->subscribe || (event_source->subscribe(app, obj))) {
1701                 ast_log(LOG_WARNING, "Error subscribing app '%s' to '%s'\n",
1702                         app_name, uri);
1703                 return STASIS_ASR_INTERNAL_ERROR;
1704         }
1705
1706         return STASIS_ASR_OK;
1707 }
1708
1709 enum stasis_app_subscribe_res stasis_app_subscribe(const char *app_name,
1710         const char **event_source_uris, int event_sources_count,
1711         struct ast_json **json)
1712 {
1713         return app_handle_subscriptions(
1714                 app_name, event_source_uris, event_sources_count,
1715                 json, app_subscribe);
1716 }
1717
1718 /*!
1719  * \internal
1720  * \brief Unsubscribe an app from an event source.
1721  *
1722  * \param app application to unsubscribe
1723  * \param uri scheme:id of an event source
1724  * \param event_source being unsubscribed from
1725  *
1726  * \retval stasis_app_subscribe_res return code.
1727  */
1728 static enum stasis_app_subscribe_res app_unsubscribe(
1729         struct stasis_app *app, const char *uri,
1730         struct stasis_app_event_source *event_source)
1731 {
1732         const char *app_name = stasis_app_name(app);
1733         const char *id = uri + strlen(event_source->scheme);
1734
1735         if (!event_source->is_subscribed ||
1736             (!event_source->is_subscribed(app, id))) {
1737                 return STASIS_ASR_EVENT_SOURCE_NOT_FOUND;
1738         }
1739
1740         ast_debug(3, "%s: Unsubscribing from %s\n", app_name, uri);
1741
1742         if (!event_source->unsubscribe || (event_source->unsubscribe(app, id))) {
1743                 ast_log(LOG_WARNING, "Error unsubscribing app '%s' to '%s'\n",
1744                         app_name, uri);
1745                 return -1;
1746         }
1747         return 0;
1748 }
1749
1750 enum stasis_app_subscribe_res stasis_app_unsubscribe(const char *app_name,
1751         const char **event_source_uris, int event_sources_count,
1752         struct ast_json **json)
1753 {
1754         return app_handle_subscriptions(
1755                 app_name, event_source_uris, event_sources_count,
1756                 json, app_unsubscribe);
1757 }
1758
1759 enum stasis_app_user_event_res stasis_app_user_event(const char *app_name,
1760         const char *event_name,
1761         const char **source_uris, int sources_count,
1762         struct ast_json *json_variables)
1763 {
1764         RAII_VAR(struct stasis_app *, app, find_app_by_name(app_name), ao2_cleanup);
1765         RAII_VAR(struct ast_json *, blob, NULL, ast_json_unref);
1766         RAII_VAR(struct ast_multi_object_blob *, multi, NULL, ao2_cleanup);
1767         RAII_VAR(void *, obj, NULL, ao2_cleanup);
1768         RAII_VAR(struct stasis_message *, message, NULL, ao2_cleanup);
1769         enum stasis_app_subscribe_res res = STASIS_APP_USER_INTERNAL_ERROR;
1770         struct ast_json *json_value;
1771         int have_channel = 0;
1772         int i;
1773
1774         if (!app) {
1775                 ast_log(LOG_WARNING, "App %s not found\n", app_name);
1776                 return STASIS_APP_USER_APP_NOT_FOUND;
1777         }
1778
1779         if (!ast_multi_user_event_type()) {
1780                 return res;
1781         }
1782
1783         blob = json_variables;
1784         if (!blob) {
1785                 blob = ast_json_pack("{}");
1786         }
1787         json_value = ast_json_string_create(event_name);
1788         if (!json_value) {
1789                 ast_log(LOG_ERROR, "unable to create json string\n");
1790                 return res;
1791         }
1792         if (ast_json_object_set(blob, "eventname", json_value)) {
1793                 ast_log(LOG_ERROR, "unable to set eventname to blob\n");
1794                 return res;
1795         }
1796
1797         multi = ast_multi_object_blob_create(blob);
1798
1799         for (i = 0; i < sources_count; ++i) {
1800                 const char *uri = source_uris[i];
1801                 void *snapshot=NULL;
1802                 enum stasis_user_multi_object_snapshot_type type;
1803
1804                 if (ast_begins_with(uri, "channel:")) {
1805                         type = STASIS_UMOS_CHANNEL;
1806                         snapshot = ast_channel_snapshot_get_latest(uri + 8);
1807                         have_channel = 1;
1808                 } else if (ast_begins_with(uri, "bridge:")) {
1809                         type = STASIS_UMOS_BRIDGE;
1810                         snapshot = ast_bridge_snapshot_get_latest(uri + 7);
1811                 } else if (ast_begins_with(uri, "endpoint:")) {
1812                         type = STASIS_UMOS_ENDPOINT;
1813                         snapshot = ast_endpoint_latest_snapshot(uri + 9, NULL);
1814                 } else {
1815                         ast_log(LOG_WARNING, "Invalid scheme: %s\n", uri);
1816                         return STASIS_APP_USER_EVENT_SOURCE_BAD_SCHEME;
1817                 }
1818                 if (!snapshot) {
1819                         ast_log(LOG_ERROR, "Unable to get snapshot for %s\n", uri);
1820                         return STASIS_APP_USER_EVENT_SOURCE_NOT_FOUND;
1821                 }
1822                 ast_multi_object_blob_add(multi, type, snapshot);
1823         }
1824
1825         message = stasis_message_create(ast_multi_user_event_type(), multi);
1826         if (!message) {
1827                 ast_log(LOG_ERROR, "Unable to create stasis user event message\n");
1828                 return res;
1829         }
1830
1831         /*
1832          * Publishing to two different topics is normally to be avoided -- except
1833          * in this case both are final destinations with no forwards (only listeners).
1834          * The message has to be delivered to the application topic for ARI, but a
1835          * copy is also delivered directly to the manager for AMI if there is a channel.
1836          */
1837         stasis_publish(ast_app_get_topic(app), message);
1838
1839         if (have_channel) {
1840                 stasis_publish(ast_manager_get_topic(), message);
1841         }
1842
1843         return STASIS_APP_USER_OK;
1844 }
1845
1846 void stasis_app_ref(void)
1847 {
1848         ast_module_ref(ast_module_info->self);
1849 }
1850
1851 void stasis_app_unref(void)
1852 {
1853         ast_module_unref(ast_module_info->self);
1854 }
1855
1856 /*!
1857  * \brief Subscription to StasisEnd events
1858  */
1859 struct stasis_subscription *stasis_end_sub;
1860
1861 static int unload_module(void)
1862 {
1863         stasis_end_sub = stasis_unsubscribe(stasis_end_sub);
1864
1865         stasis_app_unregister_event_sources();
1866
1867         messaging_cleanup();
1868
1869         cleanup();
1870         ao2_cleanup(apps_registry);
1871         apps_registry = NULL;
1872
1873         ao2_cleanup(app_controls);
1874         app_controls = NULL;
1875
1876         ao2_cleanup(app_bridges);
1877         app_bridges = NULL;
1878
1879         ao2_cleanup(app_bridges_moh);
1880         app_bridges_moh = NULL;
1881
1882         ao2_cleanup(app_bridges_playback);
1883         app_bridges_playback = NULL;
1884
1885         STASIS_MESSAGE_TYPE_CLEANUP(ast_stasis_end_message_type);
1886
1887         return 0;
1888 }
1889
1890 /* \brief Sanitization callback for channel snapshots */
1891 static int channel_snapshot_sanitizer(const struct ast_channel_snapshot *snapshot)
1892 {
1893         if (!snapshot || !(snapshot->tech_properties & AST_CHAN_TP_INTERNAL)) {
1894                 return 0;
1895         }
1896         return 1;
1897 }
1898
1899 /* \brief Sanitization callback for channel unique IDs */
1900 static int channel_id_sanitizer(const char *id)
1901 {
1902         RAII_VAR(struct ast_channel_snapshot *, snapshot, ast_channel_snapshot_get_latest(id), ao2_cleanup);
1903
1904         return channel_snapshot_sanitizer(snapshot);
1905 }
1906
1907 /* \brief Sanitization callbacks for communication to Stasis applications */
1908 struct stasis_message_sanitizer app_sanitizer = {
1909         .channel_id = channel_id_sanitizer,
1910         .channel_snapshot = channel_snapshot_sanitizer,
1911 };
1912
1913 struct stasis_message_sanitizer *stasis_app_get_sanitizer(void)
1914 {
1915         return &app_sanitizer;
1916 }
1917
1918 static void remove_masquerade_store_by_name(const char *channel_name)
1919 {
1920         struct ast_channel *chan;
1921
1922         chan = ast_channel_get_by_name(channel_name);
1923         if (!chan) {
1924                 return;
1925         }
1926
1927         remove_masquerade_store(chan);
1928         ast_channel_unref(chan);
1929 }
1930
1931 static void check_for_stasis_end(void *data, struct stasis_subscription *sub,
1932                 struct stasis_message *message)
1933 {
1934         struct ast_channel_blob *payload;
1935         struct ast_channel_snapshot *snapshot;
1936         const char *app_name;
1937         char *channel_uri;
1938         size_t alloc_size;
1939         const char *channels[1];
1940
1941         if (stasis_message_type(message) != ast_stasis_end_message_type()) {
1942                 return;
1943         }
1944
1945         payload = stasis_message_data(message);
1946         snapshot = payload->snapshot;
1947         app_name = ast_json_string_get(ast_json_object_get(payload->blob, "app"));
1948
1949         /* +8 is for the length of "channel:" */
1950         alloc_size = AST_MAX_UNIQUEID + 8;
1951         channel_uri = ast_alloca(alloc_size);
1952         snprintf(channel_uri, alloc_size, "channel:%s", snapshot->uniqueid);
1953
1954         channels[0] = channel_uri;
1955         stasis_app_unsubscribe(app_name, channels, ARRAY_LEN(channels), NULL);
1956
1957         remove_masquerade_store_by_name(snapshot->name);
1958 }
1959
1960 static const struct ast_datastore_info stasis_internal_channel_info = {
1961         .type = "stasis-internal-channel",
1962 };
1963
1964 static int set_internal_datastore(struct ast_channel *chan)
1965 {
1966         struct ast_datastore *datastore;
1967
1968         datastore = ast_channel_datastore_find(chan, &stasis_internal_channel_info, NULL);
1969         if (!datastore) {
1970                 datastore = ast_datastore_alloc(&stasis_internal_channel_info, NULL);
1971                 if (!datastore) {
1972                         return -1;
1973                 }
1974                 ast_channel_datastore_add(chan, datastore);
1975         }
1976         return 0;
1977 }
1978
1979 int stasis_app_channel_unreal_set_internal(struct ast_channel *chan)
1980 {
1981         struct ast_channel *outchan = NULL, *outowner = NULL;
1982         int res = 0;
1983         struct ast_unreal_pvt *unreal_pvt = ast_channel_tech_pvt(chan);
1984
1985         ao2_ref(unreal_pvt, +1);
1986         ast_unreal_lock_all(unreal_pvt, &outowner, &outchan);
1987         if (outowner) {
1988                 res |= set_internal_datastore(outowner);
1989                 ast_channel_unlock(outowner);
1990                 ast_channel_unref(outowner);
1991         }
1992         if (outchan) {
1993                 res |= set_internal_datastore(outchan);
1994                 ast_channel_unlock(outchan);
1995                 ast_channel_unref(outchan);
1996         }
1997         ao2_unlock(unreal_pvt);
1998         ao2_ref(unreal_pvt, -1);
1999         return res;
2000 }
2001
2002 int stasis_app_channel_set_internal(struct ast_channel *chan)
2003 {
2004         int res;
2005
2006         ast_channel_lock(chan);
2007         res = set_internal_datastore(chan);
2008         ast_channel_unlock(chan);
2009
2010         return res;
2011 }
2012
2013 int stasis_app_channel_is_internal(struct ast_channel *chan)
2014 {
2015         struct ast_datastore *datastore;
2016         int res = 0;
2017
2018         ast_channel_lock(chan);
2019         datastore = ast_channel_datastore_find(chan, &stasis_internal_channel_info, NULL);
2020         if (datastore) {
2021                 res = 1;
2022         }
2023         ast_channel_unlock(chan);
2024
2025         return res;
2026 }
2027
2028 static int load_module(void)
2029 {
2030         if (STASIS_MESSAGE_TYPE_INIT(ast_stasis_end_message_type) != 0) {
2031                 return AST_MODULE_LOAD_DECLINE;
2032         }
2033         apps_registry = ao2_container_alloc(APPS_NUM_BUCKETS, app_hash, app_compare);
2034         app_controls = ao2_container_alloc(CONTROLS_NUM_BUCKETS, control_hash, control_compare);
2035         app_bridges = ao2_container_alloc(BRIDGES_NUM_BUCKETS, bridges_hash, bridges_compare);
2036         app_bridges_moh = ao2_container_alloc_hash(
2037                 AO2_ALLOC_OPT_LOCK_MUTEX, AO2_CONTAINER_ALLOC_OPT_DUPS_REJECT,
2038                 37, bridges_channel_hash_fn, bridges_channel_sort_fn, NULL);
2039         app_bridges_playback = ao2_container_alloc_hash(
2040                 AO2_ALLOC_OPT_LOCK_MUTEX, AO2_CONTAINER_ALLOC_OPT_DUPS_REJECT,
2041                 37, bridges_channel_hash_fn, bridges_channel_sort_fn, NULL);
2042         if (!apps_registry || !app_controls || !app_bridges || !app_bridges_moh || !app_bridges_playback) {
2043                 unload_module();
2044                 return AST_MODULE_LOAD_FAILURE;
2045         }
2046
2047         if (messaging_init()) {
2048                 unload_module();
2049                 return AST_MODULE_LOAD_FAILURE;
2050         }
2051
2052         bridge_stasis_init();
2053
2054         stasis_app_register_event_sources();
2055
2056         stasis_end_sub = stasis_subscribe(ast_channel_topic_all(), check_for_stasis_end, NULL);
2057         if (!stasis_end_sub) {
2058                 unload_module();
2059                 return AST_MODULE_LOAD_DECLINE;
2060         }
2061
2062         return AST_MODULE_LOAD_SUCCESS;
2063 }
2064
2065 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_GLOBAL_SYMBOLS, "Stasis application support",
2066         .support_level = AST_MODULE_SUPPORT_CORE,
2067         .load = load_module,
2068         .unload = unload_module,
2069         );