Fix memory issues with tokens normalization.

diff --git a/src/libmime/message.c b/src/libmime/message.c
index e6f0d0a2b12522efb10c14ee2210cd8118b485e5..786510822f92028dd4637718cbfa94593d1b0e2b 100644 (file)
--- a/src/libmime/message.c
+++ b/src/libmime/message.c
@@ -1184,6 +1184,7 @@ rspamd_normalize_text_part (struct rspamd_task *task,
        struct sb_stemmer *stem = NULL;
        rspamd_fstring_t *w;
        const guchar *r;
+       gchar *temp_word;
        guint i, nlen;
        GArray *tmp;
 
@@ -1210,10 +1211,15 @@ rspamd_normalize_text_part (struct rspamd_task *task,
                        if (stem != NULL && r != NULL) {
                                nlen = strlen (r);
                                nlen = MIN (nlen, w->len);
+                               w->begin = rspamd_mempool_alloc (task->task_pool, nlen);
                                memcpy (w->begin, r, nlen);
                                w->len = nlen;
                        }
                        else {
+                               temp_word = w->begin;
+                               w->begin = rspamd_mempool_alloc (task->task_pool, w->len);
+                               memcpy (w->begin, temp_word, w->len);
+
                                if (IS_PART_UTF (part)) {
                                        rspamd_str_lc_utf8 (w->begin, w->len);
                                }