summaryrefslogtreecommitdiffstats
path: root/pigeonhole/src/lib-sieve/tst-anyof.c
blob: 84bfedd58734ac6c2cfb563fb3f78d8b592a2e1b (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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/* Copyright (c) 2002-2018 Pigeonhole authors, see the included COPYING file
 */

#include "sieve-common.h"
#include "sieve-commands.h"
#include "sieve-generator.h"
#include "sieve-validator.h"
#include "sieve-binary.h"
#include "sieve-code.h"
#include "sieve-binary.h"

/*
 * Anyof test
 *
 * Syntax
 *   anyof <tests: test-list>
 */

static bool tst_anyof_generate
	(const struct sieve_codegen_env *cgenv, struct sieve_command *ctx,
		struct sieve_jumplist *jumps, bool jump_true);
static bool tst_anyof_validate_const
	(struct sieve_validator *valdtr, struct sieve_command *tst,
		int *const_current, int const_next);

const struct sieve_command_def tst_anyof = {
	.identifier = "anyof",
	.type = SCT_TEST,
	.positional_args = 0,
	.subtests = 2,
	.block_allowed = FALSE,
	.block_required = FALSE,
	.validate_const = tst_anyof_validate_const,
	.control_generate = tst_anyof_generate
};

/*
 * Code validation
 */

static bool tst_anyof_validate_const
(struct sieve_validator *valdtr ATTR_UNUSED,
	struct sieve_command *tst ATTR_UNUSED, int *const_current, int const_next)
{
	if ( const_next > 0 ) {
		*const_current = 1;
		return FALSE;
	}

	if ( *const_current != -1 )
		*const_current = const_next;
	return TRUE;
}

/*
 * Code generation
 */

static bool tst_anyof_generate
	(const struct sieve_codegen_env *cgenv, struct sieve_command *ctx,
		struct sieve_jumplist *jumps, bool jump_true)
{
	struct sieve_binary_block *sblock = cgenv->sblock;
	struct sieve_ast_node *test;
	struct sieve_jumplist true_jumps;

	if ( sieve_ast_test_count(ctx->ast_node) > 1 ) {
		if ( !jump_true ) {
			/* Prepare jumplist */
			sieve_jumplist_init_temp(&true_jumps, sblock);
		}

		test = sieve_ast_test_first(ctx->ast_node);
		while ( test != NULL ) {
			bool result;

			/* If this test list must jump on true, all sub-tests can simply add their jumps
			 * to the caller's jump list, otherwise this test redirects all true jumps to the
			 * end of the currently generated code. This is just after a final jump to the false
			 * case
			 */
			if ( !jump_true )
				result = sieve_generate_test(cgenv, test, &true_jumps, TRUE);
			else
				result = sieve_generate_test(cgenv, test, jumps, TRUE);

			if ( !result ) return FALSE;

			test = sieve_ast_test_next(test);
		}

		if ( !jump_true ) {
			/* All tests failed, jump to case FALSE */
			sieve_operation_emit(sblock, NULL, &sieve_jmp_operation);
			sieve_jumplist_add(jumps, sieve_binary_emit_offset(sblock, 0));

			/* All true exits jump here */
			sieve_jumplist_resolve(&true_jumps);
		}
	} else {
		/* Script author is being inefficient; we can optimize the allof test away */
		test = sieve_ast_test_first(ctx->ast_node);
		sieve_generate_test(cgenv, test, jumps, jump_true);
	}

	return TRUE;
}