mirror of https://git.meli.delivery/meli/meli
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
215 lines
7.7 KiB
Rust
215 lines
7.7 KiB
Rust
/*
|
|
* Copyright (c) 2020-2023, Stalwart Labs Ltd.
|
|
*
|
|
* This file is part of the Stalwart Sieve Interpreter.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of
|
|
* the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
* in the LICENSE file at the top-level directory of this distribution.
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
* You can be released from the requirements of the AGPLv3 license by
|
|
* purchasing a commercial license. Please contact licensing@stalw.art
|
|
* for more details.
|
|
*/
|
|
|
|
use mail_parser::HeaderName;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
use crate::sieve::compiler::{
|
|
grammar::{
|
|
instruction::{CompilerState, Instruction},
|
|
Capability, Comparator,
|
|
},
|
|
lexer::{word::Word, Token},
|
|
CompileError, ErrorType, Value,
|
|
};
|
|
|
|
use crate::sieve::compiler::grammar::MatchType;
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
|
|
pub(crate) struct AddHeader {
|
|
pub last: bool,
|
|
pub field_name: Value,
|
|
pub value: Value,
|
|
}
|
|
|
|
/*
|
|
Usage: "deleteheader" [":index" <fieldno: number> [":last"]]
|
|
[COMPARATOR] [MATCH-TYPE]
|
|
<field-name: string>
|
|
[<value-patterns: string-list>]
|
|
|
|
*/
|
|
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
|
|
pub(crate) struct DeleteHeader {
|
|
pub index: Option<i32>,
|
|
pub comparator: Comparator,
|
|
pub match_type: MatchType,
|
|
pub field_name: Value,
|
|
pub value_patterns: Vec<Value>,
|
|
pub mime_anychild: bool,
|
|
}
|
|
|
|
impl<'x> CompilerState<'x> {
|
|
pub(crate) fn parse_addheader(&mut self) -> Result<(), CompileError> {
|
|
let mut field_name = None;
|
|
let value;
|
|
let mut last = false;
|
|
|
|
loop {
|
|
let token_info = self.tokens.unwrap_next()?;
|
|
match token_info.token {
|
|
Token::Tag(Word::Last) => {
|
|
self.validate_argument(1, None, token_info.line_num, token_info.line_pos)?;
|
|
last = true;
|
|
}
|
|
_ => {
|
|
let string = self.parse_string_token(token_info)?;
|
|
if field_name.is_none() {
|
|
if let Value::Text(header_name) = &string {
|
|
if HeaderName::parse(header_name).is_none() {
|
|
return Err(self
|
|
.tokens
|
|
.unwrap_next()?
|
|
.custom(ErrorType::InvalidHeaderName));
|
|
}
|
|
}
|
|
|
|
field_name = string.into();
|
|
} else {
|
|
if matches!(
|
|
&string,
|
|
Value::Text(value) if value.len() > self.compiler.max_header_size
|
|
) {
|
|
return Err(self
|
|
.tokens
|
|
.unwrap_next()?
|
|
.custom(ErrorType::HeaderTooLong));
|
|
}
|
|
value = string;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
self.instructions.push(Instruction::AddHeader(AddHeader {
|
|
last,
|
|
field_name: field_name.unwrap(),
|
|
value,
|
|
}));
|
|
Ok(())
|
|
}
|
|
|
|
pub(crate) fn parse_deleteheader(&mut self) -> Result<(), CompileError> {
|
|
let field_name: Value;
|
|
let mut match_type = MatchType::Is;
|
|
let mut comparator = Comparator::AsciiCaseMap;
|
|
let mut index = None;
|
|
let mut index_last = false;
|
|
let mut mime = false;
|
|
let mut mime_anychild = false;
|
|
|
|
loop {
|
|
let token_info = self.tokens.unwrap_next()?;
|
|
match token_info.token {
|
|
Token::Tag(
|
|
word @ (Word::Is
|
|
| Word::Contains
|
|
| Word::Matches
|
|
| Word::Value
|
|
| Word::Count
|
|
| Word::Regex),
|
|
) => {
|
|
self.validate_argument(
|
|
1,
|
|
match word {
|
|
Word::Value | Word::Count => Capability::Relational.into(),
|
|
Word::Regex => Capability::Regex.into(),
|
|
Word::List => Capability::ExtLists.into(),
|
|
_ => None,
|
|
},
|
|
token_info.line_num,
|
|
token_info.line_pos,
|
|
)?;
|
|
match_type = self.parse_match_type(word)?;
|
|
}
|
|
Token::Tag(Word::Comparator) => {
|
|
self.validate_argument(2, None, token_info.line_num, token_info.line_pos)?;
|
|
comparator = self.parse_comparator()?;
|
|
}
|
|
Token::Tag(Word::Index) => {
|
|
self.validate_argument(3, None, token_info.line_num, token_info.line_pos)?;
|
|
index = (self.tokens.expect_number(u16::MAX as usize)? as i32).into();
|
|
}
|
|
Token::Tag(Word::Last) => {
|
|
self.validate_argument(4, None, token_info.line_num, token_info.line_pos)?;
|
|
index_last = true;
|
|
}
|
|
Token::Tag(Word::Mime) => {
|
|
self.validate_argument(
|
|
5,
|
|
Capability::Mime.into(),
|
|
token_info.line_num,
|
|
token_info.line_pos,
|
|
)?;
|
|
mime = true;
|
|
}
|
|
Token::Tag(Word::AnyChild) => {
|
|
self.validate_argument(
|
|
6,
|
|
Capability::Mime.into(),
|
|
token_info.line_num,
|
|
token_info.line_pos,
|
|
)?;
|
|
mime_anychild = true;
|
|
}
|
|
_ => {
|
|
field_name = self.parse_string_token(token_info)?;
|
|
if let Value::Text(header_name) = &field_name {
|
|
if HeaderName::parse(header_name).is_none() {
|
|
return Err(self
|
|
.tokens
|
|
.unwrap_next()?
|
|
.custom(ErrorType::InvalidHeaderName));
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if !mime && mime_anychild {
|
|
return Err(self.tokens.unwrap_next()?.missing_tag(":mime"));
|
|
}
|
|
|
|
let cmd = Instruction::DeleteHeader(DeleteHeader {
|
|
index: if index_last { index.map(|i| -i) } else { index },
|
|
comparator,
|
|
match_type,
|
|
field_name,
|
|
value_patterns: if let Some(Ok(
|
|
Token::StringConstant(_) | Token::StringVariable(_) | Token::BracketOpen,
|
|
)) = self.tokens.peek().map(|r| r.map(|t| &t.token))
|
|
{
|
|
let mut key_list = self.parse_strings(false)?;
|
|
self.validate_match(&match_type, &mut key_list)?;
|
|
key_list
|
|
} else {
|
|
Vec::new()
|
|
},
|
|
mime_anychild,
|
|
});
|
|
self.instructions.push(cmd);
|
|
Ok(())
|
|
}
|
|
}
|