Fix bug 4313 "Recursion stack overflow with rebuilding folder tree" by

[claws.git] / src / imap.c

diff --git a/src/imap.c b/src/imap.c
index 8911175b5753b9f0e1bf4dd43aa1773c7c9ffec4..2b6bf360567e419db090e2282401e8616e3bf857 100644 (file)
--- a/src/imap.c
+++ b/src/imap.c
@@ -123,6 +123,8 @@ struct _IMAPSession
        gboolean cancelled;
        gboolean sens_update_block;
        gboolean do_destroy;
+
+    gint scan_tree_recurs_depth;
 };
 
 struct _IMAPNameSpace
@@ -278,6 +280,9 @@ static gint imap_auth                       (IMAPSession    *session,
 static gint imap_scan_tree_recursive   (IMAPSession    *session,
                                         FolderItem     *item,
                                         gboolean        subs_only);
+static gint imap_scan_tree_recursive_dive      (IMAPSession    *session,
+                                        FolderItem     *item,
+                                        gboolean        subs_only);
 
 static void imap_create_missing_folders        (Folder         *folder);
 static FolderItem *imap_create_special_folder
@@ -2751,6 +2756,14 @@ static gint imap_scan_tree(Folder *folder)
 }
 
 static gint imap_scan_tree_recursive(IMAPSession *session, FolderItem *item, gboolean subs_only)
+{
+    /* reset recursion depth counter */
+    session->scan_tree_recurs_depth = 0;
+
+    return imap_scan_tree_recursive_dive(session, item, subs_only);
+}
+
+static gint imap_scan_tree_recursive_dive(IMAPSession *session, FolderItem *item, gboolean subs_only)
 {
        Folder *folder;
        IMAPFolder *imapfolder;
@@ -2768,6 +2781,15 @@ static gint imap_scan_tree_recursive(IMAPSession *session, FolderItem *item, gbo
        g_return_val_if_fail(item->folder != NULL, -1);
        g_return_val_if_fail(item->no_sub == FALSE, -1);
 
+    /* recursion depth limiter */
+    if(session->scan_tree_recurs_depth >= prefs_common.imap_scan_tree_recurs_limit) {
+        g_warning("IMAP scan tree recursion limit reached (%d, folder '%s')",
+                prefs_common.imap_scan_tree_recurs_limit, item->folder->name);
+        return -1;
+    }
+    /* entering recursion func: increase depth counter */
+    session->scan_tree_recurs_depth++;
+
        folder = item->folder;
        imapfolder = IMAP_FOLDER(folder);
 
@@ -2895,8 +2917,15 @@ static gint imap_scan_tree_recursive(IMAPSession *session, FolderItem *item, gbo
                        g_free(base);
                }
 
-               if (new_item->no_sub == FALSE)
-                       imap_scan_tree_recursive(session, new_item, subs_only);
+               if (new_item->no_sub == FALSE) {
+                       imap_scan_tree_recursive_dive(session, new_item, subs_only);
+
+            /* entering recursion func: increase depth counter */
+            session->scan_tree_recurs_depth--;
+            if (session->scan_tree_recurs_depth < 0)
+                g_error("IMAP scan tree recursion underflow (%d)",
+                        session->scan_tree_recurs_depth);
+        }
        }
 
        g_slist_free(item_list);