diff options
Diffstat (limited to 'pigeonhole/src/lib-sieve/tst-exists.c')
| -rw-r--r-- | pigeonhole/src/lib-sieve/tst-exists.c | 179 |
1 files changed, 179 insertions, 0 deletions
diff --git a/pigeonhole/src/lib-sieve/tst-exists.c b/pigeonhole/src/lib-sieve/tst-exists.c new file mode 100644 index 0000000..0668c30 --- /dev/null +++ b/pigeonhole/src/lib-sieve/tst-exists.c @@ -0,0 +1,179 @@ +/* Copyright (c) 2002-2018 Pigeonhole authors, see the included COPYING file + */ + +#include "lib.h" +#include "str-sanitize.h" + +#include "sieve-common.h" +#include "sieve-commands.h" +#include "sieve-stringlist.h" +#include "sieve-code.h" +#include "sieve-message.h" +#include "sieve-validator.h" +#include "sieve-generator.h" +#include "sieve-interpreter.h" +#include "sieve-code-dumper.h" + +/* + * Exists test + * + * Syntax: + * exists <header-names: string-list> + */ + +static bool tst_exists_validate + (struct sieve_validator *valdtr, struct sieve_command *tst); +static bool tst_exists_generate + (const struct sieve_codegen_env *cgenv, struct sieve_command *tst); + +const struct sieve_command_def tst_exists = { + .identifier = "exists", + .type = SCT_TEST, + .positional_args = 1, + .subtests = 0, + .block_allowed = FALSE, + .block_required = FALSE, + .validate = tst_exists_validate, + .generate = tst_exists_generate +}; + +/* + * Exists operation + */ + +static bool tst_exists_operation_dump + (const struct sieve_dumptime_env *denv, sieve_size_t *address); +static int tst_exists_operation_execute + (const struct sieve_runtime_env *renv, sieve_size_t *address); + +const struct sieve_operation_def tst_exists_operation = { + .mnemonic = "EXISTS", + .code = SIEVE_OPERATION_EXISTS, + .dump = tst_exists_operation_dump, + .execute = tst_exists_operation_execute +}; + +/* + * Validation + */ + +static bool tst_exists_validate + (struct sieve_validator *valdtr, struct sieve_command *tst) +{ + struct sieve_ast_argument *arg = tst->first_positional; + + if ( !sieve_validate_positional_argument + (valdtr, tst, arg, "header names", 1, SAAT_STRING_LIST) ) { + return FALSE; + } + + if ( !sieve_validator_argument_activate(valdtr, tst, arg, FALSE) ) + return FALSE; + + return sieve_command_verify_headers_argument(valdtr, arg); +} + +/* + * Code generation + */ + +static bool tst_exists_generate +(const struct sieve_codegen_env *cgenv, struct sieve_command *tst) +{ + sieve_operation_emit(cgenv->sblock, NULL, &tst_exists_operation); + + /* Generate arguments */ + return sieve_generate_arguments(cgenv, tst, NULL); +} + +/* + * Code dump + */ + +static bool tst_exists_operation_dump +(const struct sieve_dumptime_env *denv, sieve_size_t *address) +{ + sieve_code_dumpf(denv, "EXISTS"); + sieve_code_descend(denv); + + /* Optional operands */ + if ( sieve_message_opr_optional_dump(denv, address, NULL) != 0 ) + return FALSE; + + return sieve_opr_stringlist_dump(denv, address, "header names"); +} + +/* + * Code execution + */ + +static int tst_exists_operation_execute +(const struct sieve_runtime_env *renv, sieve_size_t *address) +{ + struct sieve_stringlist *hdr_list; + ARRAY_TYPE(sieve_message_override) svmos; + string_t *hdr_item; + bool matched; + int ret; + + /* + * Read operands + */ + + /* Optional operands */ + i_zero(&svmos); + if ( sieve_message_opr_optional_read + (renv, address, NULL, &ret, NULL, NULL, NULL, &svmos) < 0 ) + return ret; + + /* Read header-list */ + if ( (ret=sieve_opr_stringlist_read(renv, address, "header-list", &hdr_list)) + <= 0 ) + return ret; + + /* + * Perfrom test + */ + + sieve_runtime_trace(renv, SIEVE_TRLVL_TESTS, "exists test"); + sieve_runtime_trace_descend(renv); + + /* Iterate through all requested headers to match (must find all specified) */ + hdr_item = NULL; + matched = TRUE; + while ( matched && + (ret=sieve_stringlist_next_item(hdr_list, &hdr_item)) > 0 ) { + struct sieve_stringlist *value_list; + string_t *dummy; + + /* Get header */ + if ( (ret=sieve_message_get_header_fields + (renv, sieve_single_stringlist_create(renv, hdr_item, FALSE), + &svmos, FALSE, &value_list)) <= 0 ) + return ret; + + if ( (ret=sieve_stringlist_next_item(value_list, &dummy)) < 0) + return value_list->exec_status; + + if ( ret == 0 ) + matched = FALSE; + + sieve_runtime_trace(renv, SIEVE_TRLVL_MATCHING, + "header `%s' %s", str_sanitize(str_c(hdr_item), 80), + ( matched ? "exists" : "is missing" )); + } + + if ( matched ) + sieve_runtime_trace(renv, SIEVE_TRLVL_MATCHING, "all headers exist"); + else + sieve_runtime_trace(renv, SIEVE_TRLVL_MATCHING, "headers are missing"); + + /* Set test result for subsequent conditional jump */ + if ( ret >= 0 ) { + sieve_interpreter_set_test_result(renv->interp, matched); + return SIEVE_EXEC_OK; + } + + sieve_runtime_trace_error(renv, "invalid header-list item"); + return SIEVE_EXEC_BIN_CORRUPT; +} |