Extended the Python API for update databases.

1 files changed, 13 insertions, 6 deletions

diff --git a/src/analysis/binary.c b/src/analysis/binary.c
index ac7556c..822510c 100644
--- a/src/analysis/binary.c
+++ b/src/analysis/binary.c
@@ -995,7 +995,7 @@ GHubClient *g_loaded_binary_get_client(const GLoadedBinary *binary, bool interna
 *                                                                             *
 ******************************************************************************/
 
-GDbCollection **g_loaded_binary_get_all_collections(const GLoadedBinary *binary, size_t *count)
+GDbCollection **g_loaded_binary_get_collections(const GLoadedBinary *binary, size_t *count)
 {
     GDbCollection **result;                 /* Liste à retourner           */
     GList *c;                               /* Boucle de parcours #1       */
@@ -1003,14 +1003,21 @@ GDbCollection **g_loaded_binary_get_all_collections(const GLoadedBinary *binary,
 
     *count = g_list_length(binary->collections);
 
-    result = malloc(*count * sizeof(GDbCollection *));
+    if (*count == 0)
+        result = NULL;
 
-    for (c = g_list_first(binary->collections), i = 0; c != NULL; c = g_list_next(c), i++)
+    else
     {
-        assert(i < *count);
+        result = malloc(*count * sizeof(GDbCollection *));
 
-        result[i] = G_DB_COLLECTION(c->data);
-        g_object_ref(G_OBJECT(result[i]));
+        for (c = g_list_first(binary->collections), i = 0; c != NULL; c = g_list_next(c), i++)
+        {
+            assert(i < *count);
+
+            result[i] = G_DB_COLLECTION(c->data);
+            g_object_ref(G_OBJECT(result[i]));
+
+        }
 
     }