jcoffey-dev
3a272096c0
trivy / Check (pull_request) Waiting to run
Upstream commit: 9d1c75ab68435e4417337f768291e5f947686203 Enterprise-only files removed or emptied: 63 Enterprise-only snippets removed: 118 in 50 files Dangling module declarations removed: 5 Edits turning enterprise off: 25 Third-party code: 14 files, 0 not in THIRD-PARTY.md Verification: clean One snippet more than v0.16.22, in crates/common/src/auth/authentication.rs (3, was 2).
98 lines
3.2 KiB
Rust
98 lines
3.2 KiB
Rust
/*
|
|
* SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
|
|
*
|
|
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
|
|
*/
|
|
|
|
use crate::core::Session;
|
|
use common::{config::mailstore::spamfilter::SpamFilterAction, network::SessionStream};
|
|
use mail_auth::{ArcOutput, DkimOutput, DmarcResult, dkim2::Dkim2Output, dmarc::Policy};
|
|
use mail_parser::Message;
|
|
use spam_filter::{
|
|
SpamFilterInput,
|
|
analysis::{
|
|
init::SpamFilterInit,
|
|
score::{SpamFilterAnalyzeScore, SpamFilterScore},
|
|
},
|
|
};
|
|
|
|
impl<T: SessionStream> Session<T> {
|
|
pub async fn spam_classify<'x>(
|
|
&'x self,
|
|
message: &'x Message<'x>,
|
|
dkim_result: &'x [DkimOutput<'x>],
|
|
dkim2_result: Option<&'x Dkim2Output<'x>>,
|
|
arc_result: Option<&'x ArcOutput<'x>>,
|
|
dmarc_result: Option<&'x DmarcResult>,
|
|
dmarc_policy: Option<&'x Policy>,
|
|
) -> SpamFilterAction<SpamFilterScore> {
|
|
let server = &self.server;
|
|
let mut ctx = server.spam_filter_init(self.build_spam_input(
|
|
message,
|
|
dkim_result,
|
|
dkim2_result,
|
|
arc_result,
|
|
dmarc_result,
|
|
dmarc_policy,
|
|
));
|
|
|
|
if !self.is_authenticated() {
|
|
// Spam classification
|
|
server.spam_filter_classify(&mut ctx).await
|
|
} else {
|
|
// Do not classify authenticated sessions
|
|
SpamFilterAction::Disabled
|
|
}
|
|
}
|
|
|
|
pub fn build_spam_input<'x>(
|
|
&'x self,
|
|
message: &'x Message<'x>,
|
|
dkim_result: &'x [DkimOutput<'x>],
|
|
dkim2_result: Option<&'x Dkim2Output<'x>>,
|
|
arc_result: Option<&'x ArcOutput>,
|
|
dmarc_result: Option<&'x DmarcResult>,
|
|
dmarc_policy: Option<&'x Policy>,
|
|
) -> SpamFilterInput<'x> {
|
|
SpamFilterInput {
|
|
message,
|
|
span_id: self.data.session_id,
|
|
arc_result,
|
|
spf_ehlo_result: self.data.spf_ehlo.as_ref(),
|
|
spf_mail_from_result: self.data.spf_mail_from.as_ref(),
|
|
dkim_result,
|
|
dkim2_result,
|
|
dmarc_result,
|
|
dmarc_policy,
|
|
iprev_result: self.data.iprev.as_ref(),
|
|
remote_ip: self.data.remote_ip,
|
|
ehlo_domain: self.data.helo_domain.as_str().into(),
|
|
authenticated_as: self.data.authenticated_as.as_ref().map(|a| a.name()),
|
|
asn: self.data.asn_geo_data.asn.as_ref().map(|a| a.id),
|
|
country: self.data.asn_geo_data.country.as_ref().map(|c| c.as_str()),
|
|
is_tls: self.stream.is_tls(),
|
|
env_from: self
|
|
.data
|
|
.mail_from
|
|
.as_ref()
|
|
.map(|m| m.address_lcase.as_str())
|
|
.unwrap_or_default(),
|
|
env_from_flags: self
|
|
.data
|
|
.mail_from
|
|
.as_ref()
|
|
.map(|m| m.flags)
|
|
.unwrap_or_default(),
|
|
env_rcpt_rewritten_to: self
|
|
.data
|
|
.rcpt_to
|
|
.iter()
|
|
.map(|r| r.address_lcase.as_str())
|
|
.collect(),
|
|
env_rcpt_orig_to: self.data.rcpt_to.iter().map(|r| r.orig_address()).collect(),
|
|
is_test: false,
|
|
is_train: false,
|
|
}
|
|
}
|
|
}
|