diff --git a/src/config.h b/src/config.h index 778b5681..cf86155c 100644 --- a/src/config.h +++ b/src/config.h @@ -14,7 +14,7 @@ #define VERSION "1.2.0-master" -#define BUILD 918 +#define BUILD 919 #define HOSTID "mailarchiver" @@ -84,6 +84,7 @@ #define SQL_RECIPIENT_TABLE "rcpt" #define SQL_ARCHIVING_RULE_TABLE "archiving_rule" #define SQL_RETENTION_RULE_TABLE "retention_rule" +#define SQL_FOLDER_RULE_TABLE "folder_rule" #define SQL_COUNTER_TABLE "counter" #define SQL_OPTION_TABLE "option" #define SQL_DOMAIN_TABLE "domain" diff --git a/src/defs.h b/src/defs.h index e91ae951..336603aa 100644 --- a/src/defs.h +++ b/src/defs.h @@ -120,6 +120,7 @@ struct rule { int domainlen; int days; + int folder_id; char emptyfrom, emptyto, emptysubject, emptybody, emptyaname, emptyatype; @@ -130,6 +131,20 @@ struct rule { }; +struct rule_cond { + char domain[SMALLBUFSIZE]; + char from[SMALLBUFSIZE]; + char to[SMALLBUFSIZE]; + char subject[SMALLBUFSIZE]; + char body[SMALLBUFSIZE]; + char _size[SMALLBUFSIZE]; + char attachment_name[SMALLBUFSIZE]; + char attachment_type[SMALLBUFSIZE]; + char _attachment_size[SMALLBUFSIZE]; + int size, attachment_size, spam, days, folder_id; +}; + + struct _state { int line_num; int message_state; @@ -321,6 +336,7 @@ struct __data { #ifdef HAVE_TRE struct node *archiving_rules[1]; struct node *retention_rules[1]; + struct node *folder_rules[1]; #endif #ifdef HAVE_MEMCACHED diff --git a/src/message.c b/src/message.c index b6eb523e..c3ae6672 100644 --- a/src/message.c +++ b/src/message.c @@ -18,9 +18,16 @@ int store_index_data(struct session_data *sdata, struct _state *state, struct __data *data, uint64 id, struct __config *cfg){ - int rc=ERR; + int rc=ERR, folder_id=0; char *subj; + if(data->folder > 0){ + folder_id = data->folder; + } + else { + folder_id = get_folder_id_by_rule(data, state, sdata->tot_len, sdata->spam_message, cfg); + } + subj = state->b_subject; if(*subj == ' ') subj++; @@ -47,7 +54,7 @@ int store_index_data(struct session_data *sdata, struct _state *state, struct __ data->sql[data->pos] = (char *)&sdata->sent; data->type[data->pos] = TYPE_LONG; data->pos++; data->sql[data->pos] = (char *)&sdata->tot_len; data->type[data->pos] = TYPE_LONG; data->pos++; data->sql[data->pos] = (char *)&sdata->direction; data->type[data->pos] = TYPE_LONG; data->pos++; - data->sql[data->pos] = (char *)&data->folder; data->type[data->pos] = TYPE_LONG; data->pos++; + data->sql[data->pos] = (char *)&folder_id; data->type[data->pos] = TYPE_LONG; data->pos++; data->sql[data->pos] = (char *)&state->n_attachments; data->type[data->pos] = TYPE_LONG; data->pos++; data->sql[data->pos] = sdata->attachments; data->type[data->pos] = TYPE_STRING; data->pos++; diff --git a/src/piler.c b/src/piler.c index 0711cf99..438199c4 100644 --- a/src/piler.c +++ b/src/piler.c @@ -235,6 +235,7 @@ void p_clean_exit(){ clearrules(data.archiving_rules); clearrules(data.retention_rules); + clearrules(data.folder_rules); clearhash(data.mydomains); @@ -312,6 +313,7 @@ void initialise_configuration(){ clearrules(data.archiving_rules); clearrules(data.retention_rules); + clearrules(data.folder_rules); clearhash(data.mydomains); @@ -321,6 +323,7 @@ void initialise_configuration(){ inithash(data.mydomains); initrules(data.archiving_rules); initrules(data.retention_rules); + initrules(data.folder_rules); #ifdef HAVE_STARTTLS if(cfg.tls_enable > 0 && data.ctx == NULL && init_ssl() == OK){ @@ -335,6 +338,7 @@ void initialise_configuration(){ load_rules(&sdata, &data, data.archiving_rules, SQL_ARCHIVING_RULE_TABLE, &cfg); load_rules(&sdata, &data, data.retention_rules, SQL_RETENTION_RULE_TABLE, &cfg); + load_rules(&sdata, &data, data.folder_rules, SQL_FOLDER_RULE_TABLE, &cfg); load_mydomains(&sdata, &data, &cfg); @@ -390,6 +394,7 @@ int main(int argc, char **argv){ inithash(data.mydomains); initrules(data.archiving_rules); initrules(data.retention_rules); + initrules(data.folder_rules); data.ctx = NULL; data.ssl = NULL; data.dedup = MAP_FAILED; diff --git a/src/pilerimport.c b/src/pilerimport.c index 6554a61b..090e9a72 100644 --- a/src/pilerimport.c +++ b/src/pilerimport.c @@ -507,6 +507,7 @@ int main(int argc, char **argv){ inithash(data.mydomains); initrules(data.archiving_rules); initrules(data.retention_rules); + initrules(data.folder_rules); while(1){ @@ -711,6 +712,7 @@ int main(int argc, char **argv){ load_rules(&sdata, &data, data.archiving_rules, SQL_ARCHIVING_RULE_TABLE, &cfg); load_rules(&sdata, &data, data.retention_rules, SQL_RETENTION_RULE_TABLE, &cfg); + load_rules(&sdata, &data, data.folder_rules, SQL_FOLDER_RULE_TABLE, &cfg); load_mydomains(&sdata, &data, &cfg); @@ -729,6 +731,7 @@ int main(int argc, char **argv){ clearrules(data.archiving_rules); clearrules(data.retention_rules); + clearrules(data.folder_rules); clearhash(data.mydomains); diff --git a/src/reindex.c b/src/reindex.c index 3af3cf68..9d3240ee 100644 --- a/src/reindex.c +++ b/src/reindex.c @@ -224,6 +224,7 @@ int main(int argc, char **argv){ inithash(data.mydomains); initrules(data.archiving_rules); initrules(data.retention_rules); + initrules(data.folder_rules); init_session_data(&sdata, &cfg); @@ -232,6 +233,8 @@ int main(int argc, char **argv){ p_clean_exit("cannot connect to mysql server", 1); } + load_rules(&sdata, &data, data.folder_rules, SQL_FOLDER_RULE_TABLE, &cfg); + if(folder){ data.folder = get_folder_id(&sdata, &data, folder, 0, &cfg); if(data.folder == 0){ @@ -252,6 +255,8 @@ int main(int argc, char **argv){ printf("put %llu messages to %s table for reindexing\n", n, SQL_SPHINX_TABLE); + clearrules(data.folder_rules); + clearhash(data.mydomains); close_database(&sdata); diff --git a/src/rules.c b/src/rules.c index b7e48f1d..e6607e31 100644 --- a/src/rules.c +++ b/src/rules.c @@ -12,21 +12,21 @@ void load_rules(struct session_data *sdata, struct __data *data, struct node *xhash[], char *table, struct __config *cfg){ char s[SMALLBUFSIZE]; - char domain[SMALLBUFSIZE], from[SMALLBUFSIZE], to[SMALLBUFSIZE], subject[SMALLBUFSIZE], body[SMALLBUFSIZE], _size[SMALLBUFSIZE], attachment_name[SMALLBUFSIZE], attachment_type[SMALLBUFSIZE], _attachment_size[SMALLBUFSIZE]; - int size=0, attachment_size=0, spam=0, days=0; + struct rule_cond rule_cond; - memset(domain, 0, sizeof(domain)); - memset(from, 0, sizeof(from)); - memset(to, 0, sizeof(to)); - memset(subject, 0, sizeof(subject)); - memset(body, 0, sizeof(body)); - memset(_size, 0, sizeof(_size)); - memset(attachment_name, 0, sizeof(attachment_name)); - memset(attachment_type, 0, sizeof(attachment_type)); - memset(_attachment_size, 0, sizeof(_attachment_size)); + memset(rule_cond.domain, 0, SMALLBUFSIZE); + memset(rule_cond.from, 0, SMALLBUFSIZE); + memset(rule_cond.to, 0, SMALLBUFSIZE); + memset(rule_cond.subject, 0, SMALLBUFSIZE); + memset(rule_cond.body, 0, SMALLBUFSIZE); + memset(rule_cond._size, 0, SMALLBUFSIZE); + memset(rule_cond.attachment_name, 0, SMALLBUFSIZE); + memset(rule_cond.attachment_type, 0, SMALLBUFSIZE); + memset(rule_cond._attachment_size, 0, SMALLBUFSIZE); + rule_cond.size = rule_cond.attachment_size = rule_cond.spam = rule_cond.days = rule_cond.folder_id = 0; - snprintf(s, sizeof(s)-1, "SELECT `domain`, `from`, `to`, `subject`, `body`, `_size`, `size`, `attachment_name`, `attachment_type`, `_attachment_size`, `attachment_size`, `spam`, `days` FROM `%s`", table); + snprintf(s, sizeof(s)-1, "SELECT `domain`, `from`, `to`, `subject`, `body`, `_size`, `size`, `attachment_name`, `attachment_type`, `_attachment_size`, `attachment_size`, `spam`, `days`, `folder_id` FROM `%s`", table); if(prepare_sql_statement(sdata, &(data->stmt_generic), s, cfg) == ERR) return; @@ -39,49 +39,49 @@ void load_rules(struct session_data *sdata, struct __data *data, struct node *xh p_bind_init(data); - data->sql[data->pos] = &domain[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = sizeof(domain)-2; data->pos++; - data->sql[data->pos] = &from[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = sizeof(from)-2; data->pos++; - data->sql[data->pos] = &to[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = sizeof(to)-2; data->pos++; - data->sql[data->pos] = &subject[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = sizeof(subject)-2; data->pos++; - data->sql[data->pos] = &body[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = sizeof(body)-2; data->pos++; - data->sql[data->pos] = &_size[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = sizeof(_size)-2; data->pos++; - data->sql[data->pos] = (char *)&size; data->type[data->pos] = TYPE_LONG; data->len[data->pos] = sizeof(size); data->pos++; - data->sql[data->pos] = &attachment_name[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = sizeof(attachment_name)-2; data->pos++; - data->sql[data->pos] = &attachment_type[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = sizeof(attachment_type)-2; data->pos++; - data->sql[data->pos] = &_attachment_size[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = sizeof(_attachment_size)-2; data->pos++; - data->sql[data->pos] = (char *)&attachment_size; data->type[data->pos] = TYPE_LONG; data->len[data->pos] = sizeof(attachment_size); data->pos++; - data->sql[data->pos] = (char *)&spam; data->type[data->pos] = TYPE_LONG; data->len[data->pos] = sizeof(spam); data->pos++; - data->sql[data->pos] = (char *)&days; data->type[data->pos] = TYPE_LONG; data->len[data->pos] = sizeof(days); data->pos++; + data->sql[data->pos] = &rule_cond.domain[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = SMALLBUFSIZE-2; data->pos++; + data->sql[data->pos] = &rule_cond.from[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = SMALLBUFSIZE-2; data->pos++; + data->sql[data->pos] = &rule_cond.to[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = SMALLBUFSIZE-2; data->pos++; + data->sql[data->pos] = &rule_cond.subject[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = SMALLBUFSIZE-2; data->pos++; + data->sql[data->pos] = &rule_cond.body[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = SMALLBUFSIZE-2; data->pos++; + data->sql[data->pos] = &rule_cond._size[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = SMALLBUFSIZE-2; data->pos++; + data->sql[data->pos] = (char *)&rule_cond.size; data->type[data->pos] = TYPE_LONG; data->len[data->pos] = sizeof(int); data->pos++; + data->sql[data->pos] = &rule_cond.attachment_name[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = SMALLBUFSIZE-2; data->pos++; + data->sql[data->pos] = &rule_cond.attachment_type[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = SMALLBUFSIZE-2; data->pos++; + data->sql[data->pos] = &rule_cond._attachment_size[0]; data->type[data->pos] = TYPE_STRING; data->len[data->pos] = SMALLBUFSIZE-2; data->pos++; + data->sql[data->pos] = (char *)&rule_cond.attachment_size; data->type[data->pos] = TYPE_LONG; data->len[data->pos] = sizeof(int); data->pos++; + data->sql[data->pos] = (char *)&rule_cond.spam; data->type[data->pos] = TYPE_LONG; data->len[data->pos] = sizeof(int); data->pos++; + data->sql[data->pos] = (char *)&rule_cond.days; data->type[data->pos] = TYPE_LONG; data->len[data->pos] = sizeof(int); data->pos++; + data->sql[data->pos] = (char *)&rule_cond.folder_id; data->type[data->pos] = TYPE_LONG; data->len[data->pos] = sizeof(int); data->pos++; p_store_results(sdata, data->stmt_generic, data); while(p_fetch_results(data->stmt_generic) == OK){ - append_rule(xhash, domain, from, to, subject, body, _size, size, attachment_name, attachment_type, _attachment_size, attachment_size, spam, days, data); + append_rule(xhash, &rule_cond, data); - memset(domain, 0, sizeof(domain)); - memset(from, 0, sizeof(from)); - memset(to, 0, sizeof(to)); - memset(subject, 0, sizeof(subject)); - memset(body, 0, sizeof(body)); - memset(_size, 0, sizeof(_size)); - memset(attachment_name, 0, sizeof(attachment_name)); - memset(attachment_type, 0, sizeof(attachment_type)); - memset(_attachment_size, 0, sizeof(_attachment_size)); + memset(rule_cond.domain, 0, SMALLBUFSIZE); + memset(rule_cond.from, 0, SMALLBUFSIZE); + memset(rule_cond.to, 0, SMALLBUFSIZE); + memset(rule_cond.subject, 0, SMALLBUFSIZE); + memset(rule_cond.body, 0, SMALLBUFSIZE); + memset(rule_cond._size, 0, SMALLBUFSIZE); + memset(rule_cond.attachment_name, 0, SMALLBUFSIZE); + memset(rule_cond.attachment_type, 0, SMALLBUFSIZE); + memset(rule_cond._attachment_size, 0, SMALLBUFSIZE); - size=0; attachment_size=0; spam=0; days=0; + rule_cond.size = rule_cond.attachment_size = rule_cond.spam = rule_cond.days = rule_cond.folder_id = 0; } p_free_results(data->stmt_generic); ENDE: close_prepared_statement(data->stmt_generic); - } -int append_rule(struct node *xhash[], char *domain, char *from, char *to, char *subject, char *body, char *_size, int size, char *attachment_name, char *attachment_type, char *_attachment_size, int attachment_size, int spam, int days, struct __data *data){ +int append_rule(struct node *xhash[], struct rule_cond *rule_cond, struct __data *data){ struct node *q, *Q=NULL, *node; struct rule *rule; int rc=0; @@ -91,11 +91,11 @@ int append_rule(struct node *xhash[], char *domain, char *from, char *to, char * memset(node, 0, sizeof(struct node)); node->r = NULL; - rule = create_rule_item(domain, from, to, subject, body, _size, size, attachment_name, attachment_type, _attachment_size, attachment_size, spam, days, data); + rule = create_rule_item(rule_cond, data); if(rule == NULL){ free(node); - syslog(LOG_INFO, "could not load rule=%s/%s/%s/%s/%s/%s,%d", domain, from, to, subject, body, _size, size); + syslog(LOG_INFO, "could not load rule=%s/%s/%s/%s/%s/%s,%d", rule_cond->domain, rule_cond->from, rule_cond->to, rule_cond->subject, rule_cond->body, rule_cond->_size, rule_cond->size); return rc; } @@ -119,7 +119,7 @@ int append_rule(struct node *xhash[], char *domain, char *from, char *to, char * } -struct rule *create_rule_item(char *domain, char *from, char *to, char *subject, char *body, char *_size, int size, char *attachment_name, char *attachment_type, char *_attachment_size, int attachment_size, int spam, int days, struct __data *data){ +struct rule *create_rule_item(struct rule_cond *rule_cond, struct __data *data){ struct rule *h=NULL; char empty = '\0'; int len; @@ -135,54 +135,55 @@ struct rule *create_rule_item(char *domain, char *from, char *to, char *subject, if(h->domainlen > 2){ h->domain = malloc(h->domainlen+2); - if(h->domain) snprintf(h->domain, h->domainlen, "%s", domain); + if(h->domain) snprintf(h->domain, h->domainlen, "%s", rule_cond->domain); else { h->compiled = 0; - syslog(LOG_INFO, "malloc error in create_rule_item() for '%s'", domain); + syslog(LOG_INFO, "malloc error in create_rule_item() for '%s'", rule_cond->domain); } } h->emptyfrom = h->emptyto = h->emptysubject = h->emptyaname = h->emptyatype = 0; - if(!from || strlen(from) < 1){ from = ∅ h->emptyfrom = 1; } - if(regcomp(&(h->from), from, REG_ICASE | REG_EXTENDED)) h->compiled = 0; + if(rule_cond->from == NULL || strlen(rule_cond->from) < 1){ rule_cond->from[0] = empty; h->emptyfrom = 1; } + if(regcomp(&(h->from), rule_cond->from, REG_ICASE | REG_EXTENDED)) h->compiled = 0; - if(!to || strlen(to) < 1){ to = ∅ h->emptyto = 1; } - if(regcomp(&(h->to), to, REG_ICASE | REG_EXTENDED)) h->compiled = 0; + if(rule_cond->to == NULL || strlen(rule_cond->to) < 1){ rule_cond->to[0] = empty; h->emptyto = 1; } + if(regcomp(&(h->to), rule_cond->to, REG_ICASE | REG_EXTENDED)) h->compiled = 0; - if(!subject || strlen(subject) < 1){ subject = ∅ h->emptysubject = 1; } - if(regcomp(&(h->subject), subject, REG_ICASE | REG_EXTENDED)) h->compiled = 0; + if(rule_cond->subject == NULL || strlen(rule_cond->subject) < 1){ rule_cond->subject[0] = empty; h->emptysubject = 1; } + if(regcomp(&(h->subject), rule_cond->subject, REG_ICASE | REG_EXTENDED)) h->compiled = 0; - if(!body || strlen(body) < 1){ body = ∅ h->emptybody = 1; } - if(regcomp(&(h->body), body, REG_ICASE | REG_EXTENDED)) h->compiled = 0; + if(rule_cond->body == NULL || strlen(rule_cond->body) < 1){ rule_cond->body[0] = empty; h->emptybody = 1; } + if(regcomp(&(h->body), rule_cond->body, REG_ICASE | REG_EXTENDED)) h->compiled = 0; - h->spam = spam; - h->days = days; + h->spam = rule_cond->spam; + h->days = rule_cond->days; + h->folder_id = rule_cond->folder_id; - h->size = size; + h->size = rule_cond->size; - if(!_size) _size = ∅ - snprintf(h->_size, 3, "%s", _size); + if(rule_cond->_size == NULL) rule_cond->_size[0] = empty; + snprintf(h->_size, 3, "%s", rule_cond->_size); - if(!attachment_name || strlen(attachment_name) < 1){ attachment_name = ∅ h->emptyaname = 1; } - if(regcomp(&(h->attachment_name), attachment_name, REG_ICASE | REG_EXTENDED)) h->compiled = 0; + if(rule_cond->attachment_name == NULL || strlen(rule_cond->attachment_name) < 1){ rule_cond->attachment_name[0] = empty; h->emptyaname = 1; } + if(regcomp(&(h->attachment_name), rule_cond->attachment_name, REG_ICASE | REG_EXTENDED)) h->compiled = 0; - if(!attachment_type || strlen(attachment_type) < 1){ attachment_type = ∅ h->emptyatype = 1; } - if(regcomp(&(h->attachment_type), attachment_type, REG_ICASE | REG_EXTENDED)) h->compiled = 0; + if(rule_cond->attachment_type == NULL || strlen(rule_cond->attachment_type) < 1){ rule_cond->attachment_type[0] = empty; h->emptyatype = 1; } + if(regcomp(&(h->attachment_type), rule_cond->attachment_type, REG_ICASE | REG_EXTENDED)) h->compiled = 0; - h->attachment_size = attachment_size; + h->attachment_size = rule_cond->attachment_size; - if(!_attachment_size) _attachment_size = ∅ - snprintf(h->_attachment_size, 3, "%s", _attachment_size); + if(rule_cond->_attachment_size == NULL) rule_cond->_attachment_size[0] = empty; + snprintf(h->_attachment_size, 3, "%s", rule_cond->_attachment_size); - len = strlen(domain)+8 + strlen(from)+6 + strlen(to)+4 + strlen(subject)+9 + strlen(body)+6 + strlen(_size)+6 + strlen(attachment_name)+10 + strlen(attachment_type)+10 + strlen(_attachment_size)+10 + 9 + 15 + 15; + len = strlen(rule_cond->domain)+8 + strlen(rule_cond->from)+6 + strlen(rule_cond->to)+4 + strlen(rule_cond->subject)+9 + strlen(rule_cond->body)+6 + strlen(rule_cond->_size)+6 + strlen(rule_cond->attachment_name)+10 + strlen(rule_cond->attachment_type)+10 + strlen(rule_cond->_attachment_size)+10 + 9 + 15 + 15; h->rulestr = malloc(len); - if(h->rulestr) snprintf(h->rulestr, len-1, "domain=%s,from=%s,to=%s,subject=%s,body=%s,size%s%d,att.name=%s,att.type=%s,att.size%s%d,spam=%d", domain, from, to, subject, body, _size, size, attachment_name, attachment_type, _attachment_size, attachment_size, spam); + if(h->rulestr) snprintf(h->rulestr, len-1, "domain=%s,from=%s,to=%s,subject=%s,body=%s,size%s%d,att.name=%s,att.type=%s,att.size%s%d,spam=%d", rule_cond->domain, rule_cond->from, rule_cond->to, rule_cond->subject, rule_cond->body, rule_cond->_size, rule_cond->size, rule_cond->attachment_name, rule_cond->attachment_type, rule_cond->_attachment_size, rule_cond->attachment_size, rule_cond->spam); else h->compiled = 0; h->r = NULL; @@ -316,6 +317,69 @@ unsigned long query_retain_period(struct __data *data, struct _state *state, int } +int get_folder_id_by_rule(struct __data *data, struct _state *state, int size, int spam, struct __config *cfg){ + size_t nmatch=0; + struct rule *p; + struct node *q; + int ismatch; + + q = data->folder_rules[0]; + + while(q != NULL){ + + if(q->str){ + p = q->str; + + ismatch = 0; + + if(p->domainlen > 2){ + if(strcasestr(state->b_to_domain, p->domain) || strcasestr(state->b_from_domain, p->domain)){ + return p->folder_id; + } + } + else { + + ismatch += check_spam_rule(spam, p->spam); + ismatch += check_size_rule(size, p->size, p->_size); + ismatch += check_attachment_rule(state, p); + + if(p->compiled == 1){ + if(p->emptyfrom == 1){ + ismatch += RULE_UNDEF; + } + else if(regexec(&(p->from), state->b_from, nmatch, NULL, 0) == 0) ismatch += RULE_MATCH; else ismatch += RULE_NO_MATCH; + + if(p->emptyto == 1){ + ismatch += RULE_UNDEF; + } + else if(regexec(&(p->to), state->b_to, nmatch, NULL, 0) == 0) ismatch += RULE_MATCH; else ismatch += RULE_NO_MATCH; + + if(p->emptysubject == 1){ + ismatch += RULE_UNDEF; + } + else if(regexec(&(p->subject), state->b_subject, nmatch, NULL, 0) == 0) ismatch += RULE_MATCH; else ismatch += RULE_NO_MATCH; + + if(p->emptybody == 1){ + ismatch += RULE_UNDEF; + } + else if(regexec(&(p->body), state->b_body, nmatch, NULL, 0) == 0) ismatch += RULE_MATCH; else ismatch += RULE_NO_MATCH; + + } + + if(ismatch > 0){ + return p->folder_id; + } + } + + } + + q = q->r; + } + + return 0; // default folder_id +} + + int check_size_rule(int message_size, int size, char *_size){ if(size <= 0) return RULE_UNDEF; diff --git a/src/rules.h b/src/rules.h index 799d5ef5..03783354 100644 --- a/src/rules.h +++ b/src/rules.h @@ -8,10 +8,11 @@ #include "defs.h" void load_rules(struct session_data *sdata, struct __data *data, struct node *xhash[], char *table, struct __config *cfg); -int append_rule(struct node *xhash[], char *domain, char *from, char *to, char *subject, char *body, char *_size, int size, char *attachment_name, char *attachment_type, char *_attachment_size, int attachment_size, int spam, int days, struct __data *data); -struct rule *create_rule_item(char *domain, char *from, char *to, char *subject, char *body, char *_size, int size, char *attachment_name, char *attachment_type, char *_attachment_size, int attachment_size, int spam, int days, struct __data *data); +int append_rule(struct node *xhash[], struct rule_cond *rule_cond, struct __data *data); +struct rule *create_rule_item(struct rule_cond *rule_cond, struct __data *data); char *check_againt_ruleset(struct node *xhash[], struct _state *state, int size, int spam); unsigned long query_retain_period(struct __data *data, struct _state *state, int size, int spam, struct __config *cfg); +int get_folder_id_by_rule(struct __data *data, struct _state *state, int size, int spam, struct __config *cfg); int check_size_rule(int message_size, int size, char *_size); int check_spam_rule(int is_spam, int spam); int check_attachment_rule(struct _state *state, struct rule *rule); diff --git a/src/session.c b/src/session.c index e13c4169..30a9c141 100644 --- a/src/session.c +++ b/src/session.c @@ -281,7 +281,7 @@ int handle_smtp_session(int new_sd, struct __data *data, struct __config *cfg){ (sdata.__acquire+sdata.__parsed+sdata.__av+sdata.__compress+sdata.__encrypt+sdata.__store)/1000000.0, sdata.__acquire/1000000.0, sdata.__parsed/1000000.0, sdata.__av/1000000.0, sdata.__compress/1000000.0, sdata.__encrypt/1000000.0, sdata.__store/1000000.0); - syslog(LOG_PRIORITY, "%s: from=%s, size=%d/%d, attachments=%d, reference=%s, message-id=%s, retention=%d, %s, status=%s", sdata.ttmpfile, sdata.fromemail, sdata.tot_len, sdata.stored_len, sstate.n_attachments, sstate.reference, sstate.message_id, sstate.retention, delay, status); + syslog(LOG_PRIORITY, "%s: from=%s, size=%d/%d, attachments=%d, reference=%s, message-id=%s, retention=%d, folder=%d, %s, status=%s", sdata.ttmpfile, sdata.fromemail, sdata.tot_len, sdata.stored_len, sstate.n_attachments, sstate.reference, sstate.message_id, sstate.retention, sdata.folder_id, delay, status); diff --git a/src/test.c b/src/test.c index 756f0e19..e9774f25 100644 --- a/src/test.c +++ b/src/test.c @@ -63,11 +63,13 @@ int main(int argc, char **argv){ initrules(data.archiving_rules); initrules(data.retention_rules); + initrules(data.folder_rules); load_mydomains(&sdata, &data, &cfg); load_rules(&sdata, &data, data.archiving_rules, SQL_ARCHIVING_RULE_TABLE, &cfg); load_rules(&sdata, &data, data.retention_rules, SQL_RETENTION_RULE_TABLE, &cfg); + load_rules(&sdata, &data, data.folder_rules, SQL_FOLDER_RULE_TABLE, &cfg); init_session_data(&sdata, &cfg); @@ -108,10 +110,13 @@ int main(int argc, char **argv){ sdata.retained = sdata.now + query_retain_period(&data, &state, st.st_size, sdata.spam_message, &cfg); + printf("folder: %d\n", get_folder_id_by_rule(&data, &state, st.st_size, sdata.spam_message, &cfg)); + printf("retention period: %ld\n", sdata.retained); clearrules(data.archiving_rules); clearrules(data.retention_rules); + clearrules(data.folder_rules); clearhash(data.mydomains); diff --git a/util/db-mysql.sql b/util/db-mysql.sql index 2b578300..7e81030f 100644 --- a/util/db-mysql.sql +++ b/util/db-mysql.sql @@ -122,6 +122,7 @@ create table if not exists `archiving_rule` ( `attachment_size` int default 0, `spam` tinyint(1) default -1, `days` int default 0, + `folder_id` int default 0, primary key (`id`), unique(`from`,`to`,`subject`,`body`,`_size`,`size`,`attachment_name`,`attachment_type`,`_attachment_size`,`attachment_size`,`spam`) ) ENGINE=InnoDB; @@ -142,6 +143,28 @@ create table if not exists `retention_rule` ( `attachment_size` int default 0, `spam` tinyint(1) default -1, `days` int default 0, + `folder_id` int default 0, + primary key (`id`), + unique (`domain`,`from`,`to`,`subject`,`body`,`_size`,`size`,`attachment_name`,`attachment_type`,`_attachment_size`,`attachment_size`,`spam`) +) ENGINE=InnoDB; + + +create table if not exists `folder_rule` ( + `id` bigint unsigned not null auto_increment, + `domain` varchar(128) default null, + `from` varchar(128) default null, + `to` varchar(128) default null, + `subject` varchar(128) default null, + `body` varchar(128) default null, + `_size` char(2) default null, + `size` int default 0, + `attachment_name` varchar(128) default null, + `attachment_type` varchar(128) default null, + `_attachment_size` char(2) default null, + `attachment_size` int default 0, + `spam` tinyint(1) default -1, + `days` int default 0, + `folder_id` int default 0, primary key (`id`), unique (`domain`,`from`,`to`,`subject`,`body`,`_size`,`size`,`attachment_name`,`attachment_type`,`_attachment_size`,`attachment_size`,`spam`) ) ENGINE=InnoDB; diff --git a/util/db-upgrade-1.1.0-vs-1.2.0.sql b/util/db-upgrade-1.1.0-vs-1.2.0.sql index 9a77435d..61818582 100644 --- a/util/db-upgrade-1.1.0-vs-1.2.0.sql +++ b/util/db-upgrade-1.1.0-vs-1.2.0.sql @@ -26,3 +26,10 @@ create table if not exists `timestamp` ( ) Engine=InnoDB; + +-- 2015.08.28 + +alter table archiving_rule add column `folder_id` int default 0; +alter table retention_rule add column `folder_id` int default 0; + +