res_hep_{pjsip|rtcp}: Decline module loads if res_hep had not loaded
authorMatt Jordan <mjordan@digium.com>
Wed, 8 Jun 2016 17:26:29 +0000 (12:26 -0500)
committerMatt Jordan <mjordan@digium.com>
Wed, 8 Jun 2016 17:32:02 +0000 (12:32 -0500)
A crash can occur in res_hep_pjsip or res_hep_rtcp if res_hep has not
loaded and does not have a configuration file. Previously when this
occurred, checks were put in to see if the configuration was loaded
successfully. While this is a good idea - and has been added to the
offending function in res_hep - the reality is res_hep_pjsip and
res_hep_rtcp have no business running if res_hep isn't also running.

As such, this patch also adds a function to res_hep that returns whether
or not it successfully loaded. Oddly enough, ast_module_check returns
"everything is peachy" even if a module declined its load - so it cannot
be solely relied on. res_hep_pjsip and res_hep_rtcp now also check this
function to see if they should continue to load; if it fails, they
decline their load as well.

ASTERISK-26096 #close

Change-Id: I007e535fcc2e51c2ca48534f48c5fc2ac38935ea

include/asterisk/res_hep.h
res/res_hep.c
res/res_hep.exports.in
res/res_hep_pjsip.c
res/res_hep_rtcp.c

index bd0129e..cfd213a 100644 (file)
@@ -118,6 +118,14 @@ int hepv3_send_packet(struct hepv3_capture_info *capture_info);
  */
 enum hep_uuid_type hepv3_get_uuid_type(void);
 
+/*!
+ * \brief Return whether or not we're currently loaded and active
+ *
+ * \retval 0 The module is not loaded
+ * \retval 1 The module is loaded
+ */
+int hepv3_is_loaded(void);
+
 #if defined(__cplusplus) || defined(c_plusplus)
 }
 #endif
index 4520135..e79f2b6 100644 (file)
@@ -409,9 +409,21 @@ enum hep_uuid_type hepv3_get_uuid_type(void)
 {
        RAII_VAR(struct module_config *, config, ao2_global_obj_ref(global_config), ao2_cleanup);
 
+       if (!config) {
+               /* Well, that's unfortunate. Return something. */
+               return HEP_UUID_TYPE_CALL_ID;
+       }
+
        return config->general->uuid_type;
 }
 
+int hepv3_is_loaded(void)
+{
+       RAII_VAR(struct module_config *, config, ao2_global_obj_ref(global_config), ao2_cleanup);
+
+       return (config != NULL) ? 1 : 0;
+}
+
 struct hepv3_capture_info *hepv3_create_capture_info(const void *payload, size_t len)
 {
        struct hepv3_capture_info *info;
index df0f2b4..e318ac9 100644 (file)
@@ -3,6 +3,7 @@
                LINKER_SYMBOL_PREFIX*hepv3_send_packet;
                LINKER_SYMBOL_PREFIX*hepv3_create_capture_info;
                LINKER_SYMBOL_PREFIX*hepv3_get_uuid_type;
+               LINKER_SYMBOL_PREFIX*hepv3_is_loaded;
        local:
                *;
 };
index 0cc54c2..a3a93e9 100644 (file)
@@ -210,6 +210,11 @@ static int load_module(void)
 {
        CHECK_PJSIP_MODULE_LOADED();
 
+       if (!ast_module_check("res_hep.so") || !hepv3_is_loaded()) {
+               ast_log(AST_LOG_WARNING, "res_hep is not loaded or running; declining module load\n");
+               return AST_MODULE_LOAD_DECLINE;
+       }
+
        ast_sip_register_service(&logging_module);
        return AST_MODULE_LOAD_SUCCESS;
 }
index 8643d4d..03db181 100644 (file)
@@ -149,6 +149,10 @@ static void rtp_topic_handler(void *data, struct stasis_subscription *sub, struc
 
 static int load_module(void)
 {
+       if (!ast_module_check("res_hep.so") || !hepv3_is_loaded()) {
+               ast_log(AST_LOG_WARNING, "res_hep is not loaded or running; declining module load\n");
+               return AST_MODULE_LOAD_DECLINE;
+       }
 
        stasis_rtp_subscription = stasis_subscribe(ast_rtp_topic(),
                rtp_topic_handler, NULL);