summaryrefslogtreecommitdiffstats
path: root/test/rspamd_expression_test.c
blob: 7cf4bb1232b641c1961e1305e6acd44808e99398 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#include "../src/config.h"
#include "../src/main.h"
#include "../src/cfg_file.h"
#include "../src/expressions.h"
#include "tests.h"

/* Vector of test expressions */
char *test_expressions[] = {
	"(A&B|!C)&!(D|E)",
	"/test&!/&!/\\/|/",
	"header_exists(f(b(aaa)))|header=/bbb/",
	"!(header_exists(X-Mailer, /aaa,/) | header_exists(User-Agent)) & Received=/cp-out\\d+\\.libero\\.it/H & Message-Id=/<[\\da-f]{12}\\.[\\da-f]{16}@/H",
	NULL
}; 

void 
rspamd_expression_test_func ()
{
	memory_pool_t *pool;
	struct expression *cur;
	struct expression_argument *arg;
	char **line, *outstr;
	int r, s;
	GList *cur_arg;

	pool = memory_pool_new (1024);
	
	line = test_expressions;
	while (*line) {
		r = 0;
		cur = parse_expression (pool, *line);
		s = strlen (*line) * 4;
		outstr = memory_pool_alloc (pool, s);
		while (cur) {
			if (cur->type == EXPR_REGEXP) {
				r += rspamd_snprintf (outstr + r, s - r, "OP:%s ", (char *)cur->content.operand);
			} else if (cur->type == EXPR_STR) {
				r += rspamd_snprintf (outstr + r, s - r, "S:%s ", (char *)cur->content.operand);

			} else if (cur->type == EXPR_FUNCTION) {
				r += rspamd_snprintf (outstr + r, s - r, "F:%s ", ((struct expression_function *)cur->content.operand)->name);
				cur_arg = ((struct expression_function *)cur->content.operand)->args;
				while (cur_arg) {
					arg = cur_arg->data;
					if (arg->type == EXPRESSION_ARGUMENT_NORMAL) {
						r += rspamd_snprintf (outstr + r, s - r, "A:%s ", (char *)arg->data);
					}
					else {
						r += rspamd_snprintf (outstr + r, s - r, "AF:%p ", arg->data);
					}
					cur_arg = g_list_next (cur_arg);
				}
			}
			else {
				r += rspamd_snprintf (outstr + r, s - r, "O:%c ", cur->content.operation);
			}
			cur = cur->next;
		}
		msg_debug ("Parsed expression: '%s' -> '%s'", *line, outstr);
		line ++;
	}

	memory_pool_delete (pool);
}