gboolean
rspamd_task_add_recipient (struct rspamd_task *task, const gchar *rcpt)
{
InternetAddressList *tmp_addr;
if (task->rcpt_envelope == NULL) {
task->rcpt_envelope = internet_address_list_new ();
#ifdef GMIME24
rspamd_mempool_add_destructor (task->task_pool,
(rspamd_mempool_destruct_t) g_object_unref,
task->rcpt_envelope);
#else
rspamd_mempool_add_destructor (task->task_pool,
(rspamd_mempool_destruct_t) internet_address_list_destroy,
task->rcpt_envelope);
#endif
}
tmp_addr = internet_address_list_parse_string (rcpt);
if (tmp_addr) {
internet_address_list_append (task->rcpt_envelope, tmp_addr);
#ifdef GMIME24
g_object_unref (tmp_addr);
#else
internet_address_list_destroy (tmp_addr);
#endif
return TRUE;
}
return FALSE;
}
gboolean
rspamd_task_add_sender (struct rspamd_task *task, const gchar *sender)
{
InternetAddressList *tmp_addr;
if (task->from_envelope == NULL) {
task->from_envelope = internet_address_list_new ();
#ifdef GMIME24
rspamd_mempool_add_destructor (task->task_pool,
(rspamd_mempool_destruct_t) g_object_unref,
task->from_envelope);
#else
rspamd_mempool_add_destructor (task->task_pool,
(rspamd_mempool_destruct_t) internet_address_list_destroy,
task->from_envelope);
#endif
}
if (strcmp (sender, "<>") == 0) {
/* Workaround for gmime */
internet_address_list_add (task->from_envelope,
internet_address_mailbox_new ("", ""));
return TRUE;
}
else {
tmp_addr = internet_address_list_parse_string (sender);
if (tmp_addr) {
internet_address_list_append (task->from_envelope, tmp_addr);
#ifdef GMIME24
g_object_unref (tmp_addr);
#else
internet_address_list_destroy (tmp_addr);
#endif
return TRUE;
}
}
return FALSE;
}
document* parse_file(const char *file_name) {
int tallied_length = 0;
GMimeStream *stream;
GMimeMessage *msg = 0;
// struct stat stat_buf;
const char *author, *subject, *xref, *xref_end;
time_t date;
int offset;
int num_words = 0;
int file;
InternetAddress *iaddr;
InternetAddressList *iaddr_list;
char *address;
#if DEBUG
printf("%s\n", file_name);
#endif
/*
if ((file = stat(file_name, &stat_buf)) == -1) {
perror("tokenizer");
return NULL;
}
*/
// |O_STREAMING
if ((file = open(file_name, O_RDONLY|O_STREAMING)) == -1) {
perror("tokenizer");
return NULL;
}
#ifdef POSIX_FADV_NOREUSE
no_reuse(file);
#endif
stream = g_mime_stream_fs_new(file);
msg = g_mime_parser_construct_message(stream);
g_mime_stream_unref(stream);
if (msg != 0) {
table = g_hash_table_new(g_str_hash, g_str_equal);
bufp = buffer;
dword_table[0].word = NULL;
bzero(saved_body, MAX_SAVED_BODY_LENGTH);
saved_body_length = 0;
author = g_mime_message_get_sender(msg);
subject = g_mime_message_get_subject(msg);
xref = g_mime_message_get_header(msg, "Xref");
g_mime_message_get_date(msg, &date, &offset);
if (author != NULL && subject != NULL && xref != NULL) {
tallied_length = tally_string(author, tallied_length);
strncpy(doc.author, author, MAX_HEADER_LENGTH-1);
/* Get the address from the From header. */
if ((iaddr_list = internet_address_parse_string(author)) != NULL) {
iaddr = iaddr_list->address;
internet_address_set_name(iaddr, NULL);
address = internet_address_to_string(iaddr, FALSE);
strncpy(doc.address, address, MAX_HEADER_LENGTH-1);
downcase_string(doc.address);
free(address);
internet_address_list_destroy(iaddr_list);
} else {
*doc.address = 0;
}
tallied_length = tally_string(subject, tallied_length);
strncpy(doc.subject, subject, MAX_HEADER_LENGTH-1);
doc.time = date;
if ((xref = strchr(xref, ' ')) != NULL) {
xref++;
xref_end = strchr(xref, ':');
*doc.group = 0;
strncat(doc.group, xref, min(xref_end-xref, MAX_HEADER_LENGTH-1));
xref_end++;
sscanf(xref_end, "%d", &doc.article);
}
g_mime_message_foreach_part(msg, partFound, (gpointer) &tallied_length);
strncpy(doc.body, saved_body, MAX_SAVED_BODY_LENGTH);
g_hash_table_foreach(table, add_word_to_table, (gpointer) &num_words);
dword_table[num_words].word = NULL;
g_hash_table_destroy(table);
g_mime_object_unref(GMIME_OBJECT(msg));
} else {
close(file);
return NULL;
}
}
close(file);
doc.words = dword_table;
doc.num_words = num_words;
return &doc;
}
