clamav plugin: fix signature count
[claws.git] / src / plugins / clamav / clamav_plugin.c
index 0d729b8dca785479193f1e9a6347e9476c90a6e0..c63ae0491937f18a783f702878f2719467514516 100644 (file)
 #include "defs.h"
 
 #include <glib.h>
-
-#if HAVE_LOCALE_H
-#  include <locale.h>
-#endif
-
 #include <clamav.h>
 
-#include "common/plugin.h"
-#include "common/utils.h"
+#include "common/sylpheed.h"
+#include "common/version.h"
+#include "intl.h"
+#include "plugin.h"
+#include "utils.h"
 #include "hooks.h"
 #include "inc.h"
 #include "mimeview.h"
 
 #include "clamav_plugin.h"
 
-static gint hook_id;
+static guint hook_id;
 
 static ClamAvConfig config;
 
 static PrefParam param[] = {
        {"clamav_enable", "FALSE", &config.clamav_enable, P_BOOL,
         NULL, NULL, NULL},
-       {"clamav_enable_archive", "FALSE", &config.clamav_archive_enable, P_BOOL,
+       {"clamav_enable_arc", "FALSE", &config.clamav_enable_arc, P_BOOL,
         NULL, NULL, NULL},
        {"clamav_max_size", "1", &config.clamav_max_size, P_USHORT,
         NULL, NULL, NULL},
-       {"clamav_receive_infected", "TRUE", &config.clamav_receive_infected, P_BOOL,
+       {"clamav_recv_infected", "TRUE", &config.clamav_recv_infected, P_BOOL,
         NULL, NULL, NULL},
        {"clamav_save_folder", NULL, &config.clamav_save_folder, P_STRING,
         NULL, NULL, NULL},
@@ -61,142 +59,109 @@ static PrefParam param[] = {
        {NULL, NULL, NULL, P_OTHER, NULL, NULL, NULL}
 };
 
-#define IS_FIRST_PART_TEXT(info) \
-       ((info->mime_type == MIME_TEXT || info->mime_type == MIME_TEXT_HTML || \
-         info->mime_type == MIME_TEXT_ENRICHED) || \
-        (info->mime_type == MIME_MULTIPART && info->content_type && \
-         !strcasecmp(info->content_type, "multipart/alternative") && \
-         (info->children && \
-          (info->children->mime_type == MIME_TEXT || \
-           info->children->mime_type == MIME_TEXT_HTML || \
-           info->children->mime_type == MIME_TEXT_ENRICHED))))
+struct scan_parameters {
+       gboolean is_infected;
+
+       struct cl_node *root;
+       struct cl_limits limits;
+       gboolean scan_archive;
+};
+
+static gboolean scan_func(GNode *node, gpointer data)
+{
+       struct scan_parameters *params = (struct scan_parameters *) data;
+       MimeInfo *mimeinfo = (MimeInfo *) node->data;
+       gchar *outfile;
+       int ret;
+       unsigned long int size;
+       char *virname;
+
+       outfile = procmime_get_tmp_file_name(mimeinfo);
+       if (procmime_get_part(outfile, mimeinfo) < 0)
+               g_warning("Can't get the part of multipart message.");
+       else {
+               debug_print("Scanning %s\n", outfile);
+               if ((ret = cl_scanfile(outfile, &virname, &size, params->root, 
+                                     &params->limits, params->scan_archive)) == CL_VIRUS) {
+                       params->is_infected = TRUE;
+                       debug_print("Detected %s virus.\n", virname); 
+               } else {
+                       debug_print("No virus detected.\n");
+                       if (ret != CL_CLEAN)
+                               debug_print("Error: %s\n", cl_perror(ret));
+               }
+
+               unlink(outfile);
+       }
+
+       return params->is_infected;
+}
 
 static gboolean mail_filtering_hook(gpointer source, gpointer data)
 {
        MailFilteringData *mail_filtering_data = (MailFilteringData *) source;
        MsgInfo *msginfo = mail_filtering_data->msginfo;
-       gboolean is_infected = FALSE;
-       FILE  *fp;
        MimeInfo *mimeinfo;
-       MimeInfo *child;
-       gchar *infile;
-       gchar *outfile;
-       gint scan_archive = 0;
 
-       int ret, no;
-       unsigned long int size;
-       long double kb;
-       char *virname;
-       struct cl_node *root = NULL;
-       struct cl_limits limits;
+       int ret, no = 0;
+       struct scan_parameters params;
 
        if (!config.clamav_enable)
                return FALSE;
 
-       debug_print("Scanning message %d for viruses\n", msginfo->msgnum);
-
-       fp = procmsg_open_message(msginfo);
-       if (fp == NULL) {
-               debug_print("failed to open message file");
-       } 
-       
        mimeinfo = procmime_scan_message(msginfo);
        if (!mimeinfo) return FALSE;
 
-       child = mimeinfo->children;
-       if (!child || IS_FIRST_PART_TEXT(mimeinfo)) {
-               procmime_mimeinfo_free_all(mimeinfo);
-               return FALSE;
-       }
+       debug_print("Scanning message %d for viruses\n", msginfo->msgnum);
 
-       if (IS_FIRST_PART_TEXT(child))
-               child = child->next;
+       params.is_infected = FALSE;
+       params.root = NULL;
 
-       infile = procmsg_get_message_file_path(msginfo);
+       params.limits.maxfiles = 1000; /* max files */
+       params.limits.maxfilesize = config.clamav_max_size * 1048576; /* maximum archived file size */
+       params.limits.maxreclevel = 8; /* maximum recursion level */
 
+       if (config.clamav_enable_arc)
+               params.scan_archive = TRUE;
 
-       if((ret = cl_loaddbdir(cl_retdbdir(), &root, &no))) {
+       if((ret = cl_loaddbdir(cl_retdbdir(), &params.root, &no))) {
                debug_print("cl_loaddbdir: %s\n", cl_perror(ret));
                exit(2);
        }
+       debug_print("Database loaded (containing in total %d signatures)\n", no / 2);
 
-       debug_print("Loaded %d viruses.\n", no);
+       cl_buildtrie(params.root);
 
-       cl_buildtrie(root);
+       g_node_traverse(mimeinfo->node, G_PRE_ORDER, G_TRAVERSE_ALL, -1, scan_func, &params);
 
-       limits.maxfiles = 1000; /* max files */
-       limits.maxfilesize = config.clamav_max_size * 1048576; /* maximal archived file size == 10 Mb */
-       limits.maxreclevel = 8; /* maximal recursion level */
+       if (params.is_infected) {
+               if (config.clamav_recv_infected) {
+                       FolderItem *clamav_save_folder;
 
-       if (config.clamav_archive_enable)
-               scan_archive = TRUE;
+                       if ((!config.clamav_save_folder) ||
+                           (config.clamav_save_folder[0] == '\0') ||
+                           ((clamav_save_folder = folder_find_item_from_identifier(config.clamav_save_folder)) == NULL))
+                                   clamav_save_folder = folder_get_default_trash();
 
-       while (child != NULL) {
-               if (child->children || child->mime_type == MIME_MULTIPART) {
-                       child = procmime_mimeinfo_next(child);
-                       continue;
-               }
-               if(child->parent && child->parent->parent
-               && !strcasecmp(child->parent->parent->content_type, "multipart/signed")
-               && child->mime_type == MIME_TEXT) {
-                       /* this is the main text part of a signed message */
-                       child = procmime_mimeinfo_next(child);
-                       continue;
-               }
-               outfile = procmime_get_tmp_file_name(child);
-               if (procmime_get_part(outfile, infile, child) < 0)
-                       g_warning("Can't get the part of multipart message.");
-               else {
-                       debug_print("Scanning %s\n", outfile);
-                       if((ret = cl_scanfile(outfile, &virname, &size, root, 
-                                             &limits, scan_archive)) == CL_VIRUS) {
-                               is_infected = TRUE;
-                               debug_print("Detected %s virus.\n", virname); 
-                       } else {
-                               debug_print("No virus detected.\n");
-                               if(ret != CL_CLEAN)
-                                       debug_print("Error: %s\n", cl_perror(ret));
-                       }
-
-                       kb = size * (CL_COUNT_PRECISION / 1024);
-                       debug_print("Data scanned: %2.2Lf Kb\n", kb);
-    
-                       if (is_infected) {
-                               debug_print("message part(s) infected with %s\n", virname);
-                               if (config.clamav_receive_infected) {
-                                       FolderItem *clamav_save_folder;
-
-                                       if ((!config.clamav_save_folder) ||
-                                       (config.clamav_save_folder[0] == '\0') ||
-                                       ((clamav_save_folder = folder_find_item_from_identifier(config.clamav_save_folder)) == NULL))
-                                               clamav_save_folder = folder_get_default_trash();
-
-                                       procmsg_msginfo_unset_flags(msginfo, ~0, 0);
-                                       folder_item_move_msg(clamav_save_folder, msginfo);
-                               } else {
-                                       folder_item_remove_msg(msginfo->folder, msginfo->msgnum);
-                               }
-
-                               return TRUE;
-                       }
-                               child = child->next;
+                       procmsg_msginfo_unset_flags(msginfo, ~0, 0);
+                       folder_item_move_msg(clamav_save_folder, msginfo);
+               } else {
+                       folder_item_remove_msg(msginfo->folder, msginfo->msgnum);
                }
        }
-       cl_freetrie(root);
-       g_free(infile);
+       
+       cl_freetrie(params.root);
        procmime_mimeinfo_free_all(mimeinfo);
-
-       return FALSE;
+       
+       return params.is_infected;
 }
 
-#undef IS_FIRST_PART_TEXT
-
-ClamAvConfig *clamav_get_config()
+ClamAvConfig *clamav_get_config(void)
 {
        return &config;
 }
 
-void clamav_save_config()
+void clamav_save_config(void)
 {
        PrefFile *pfile;
        gchar *rcpath;
@@ -221,6 +186,16 @@ void clamav_save_config()
 
 gint plugin_init(gchar **error)
 {
+       if ((sylpheed_get_version() > VERSION_NUMERIC)) {
+               *error = g_strdup("Your sylpheed version is newer than the version the plugin was built with");
+               return -1;
+       }
+
+       if ((sylpheed_get_version() < MAKE_NUMERIC_VERSION(0, 9, 3, 86))) {
+               *error = g_strdup("Your sylpheed version is too old");
+               return -1;
+       }
+
        hook_id = hooks_register_hook(MAIL_FILTERING_HOOKLIST, mail_filtering_hook, NULL);
        if (hook_id == -1) {
                *error = g_strdup("Failed to register mail filtering hook");
@@ -236,7 +211,7 @@ gint plugin_init(gchar **error)
        
 }
 
-void plugin_done()
+void plugin_done(void)
 {
        hooks_unregister_hook(MAIL_FILTERING_HOOKLIST, hook_id);
        g_free(config.clamav_save_folder);
@@ -244,27 +219,26 @@ void plugin_done()
        debug_print("ClamAV plugin unloaded\n");
 }
 
-const gchar *plugin_name()
+const gchar *plugin_name(void)
 {
-       return "Clam AntiVirus";
+       return _("Clam AntiVirus");
 }
 
-const gchar *plugin_desc()
+const gchar *plugin_desc(void)
 {
-       return "This plugin checks all message attachments that are received "
-              "from a POP account for viruses using Clam AntiVirus. You will "
-              "need to have ClamAv installed.\n"
+       return _("This plugin uses Clam AntiVirus to scan all message attachments "
+              "that are received from a POP account.\n"
               "\n"
               "When a message attachment is found to contain a virus it can be "
-              "deleted or saved in a special folder.\n"
+              "deleted or saved in a specially designated folder.\n"
               "\n"
-              "This plugin only contains the actual function for filtering "
+              "This plugin only contains the actual function for scanning "
               "and deleting or moving the message. You probably want to load "
-              "a User Interface plugin too, otherwise you will have to "
-              "manually write the plugin configuration.\n";
+              "the Gtk+ User Interface plugin too, otherwise you will have to "
+              "manually write the plugin configuration.\n");
 }
 
-const gchar *plugin_type()
+const gchar *plugin_type(void)
 {
        return "Common";
 }