res_sorcery_realtime: Fix leak of sorcery object type.
[asterisk/asterisk.git] / res / res_sorcery_realtime.c
index 656ee0d..b2df298 100644 (file)
@@ -30,7 +30,7 @@
 
 #include "asterisk.h"
 
-ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
+ASTERISK_REGISTER_FILE()
 
 #include <regex.h>
 
@@ -142,6 +142,8 @@ static struct ast_variable *sorcery_realtime_filter_objectset(struct ast_variabl
                }
        }
 
+       ao2_ref(object_type, -1);
+
        return objectset;
 }
 
@@ -158,7 +160,9 @@ static void *sorcery_realtime_retrieve_fields(const struct ast_sorcery *sorcery,
 
        objectset = sorcery_realtime_filter_objectset(objectset, &id, sorcery, type);
 
-       if (!id || !(object = ast_sorcery_alloc(sorcery, type, id->value)) || ast_sorcery_objectset_apply(sorcery, object, objectset)) {
+       if (!id
+               || !(object = ast_sorcery_alloc(sorcery, type, id->value))
+               || ast_sorcery_objectset_apply(sorcery, object, objectset)) {
                return NULL;
        }
 
@@ -177,7 +181,7 @@ static void sorcery_realtime_retrieve_multiple(const struct ast_sorcery *sorcery
        const char *family = data;
        RAII_VAR(struct ast_config *, rows, NULL, ast_config_destroy);
        RAII_VAR(struct ast_variable *, all, NULL, ast_variables_destroy);
-       char *row = NULL;
+       struct ast_category *row = NULL;
 
        if (!fields) {
                char field[strlen(UUID_FIELD) + 6], value[2];
@@ -197,9 +201,8 @@ static void sorcery_realtime_retrieve_multiple(const struct ast_sorcery *sorcery
                return;
        }
 
-       while ((row = ast_category_browse(rows, row))) {
-               struct ast_category *cat = ast_category_get(rows, row);
-               struct ast_variable *objectset = ast_category_detach_variables(cat);
+       while ((row = ast_category_browse_filtered(rows, NULL, row, NULL))) {
+               struct ast_variable *objectset = ast_category_detach_variables(row);
                RAII_VAR(struct ast_variable *, id, NULL, ast_variables_destroy);
                RAII_VAR(void *, object, NULL, ao2_cleanup);
 
@@ -283,6 +286,7 @@ static int unload_module(void)
 }
 
 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_GLOBAL_SYMBOLS | AST_MODFLAG_LOAD_ORDER, "Sorcery Realtime Object Wizard",
+       .support_level = AST_MODULE_SUPPORT_CORE,
        .load = load_module,
        .unload = unload_module,
        .load_pri = AST_MODPRI_REALTIME_DRIVER,