Import upstream v0.16.22, stripped

Upstream commit: 474dd0229cb20cf513036619781ed97bd8073c3f Enterprise-only files removed or emptied: 63 Enterprise-only snippets removed: 117 in 50 files Dangling module declarations removed: 5 Cargo edits turning enterprise off: 14 Verification: clean Enterprise feature gates left for rebuilt features: 19 in 18 files Produced by tools/fork/strip.py. The full report is in docs/fork/strip-reports/ on main.
2026-09-18 10:21:56 -07:00
commit 7dae9b29fd
1650 changed files with 485521 additions and 0 deletions
@@ -0,0 +1,107 @@
+/*
+ * 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.dsn_info
+                        .as_deref()
+                        .and_then(|info| info.strip_prefix("rfc822;"))
+                        .unwrap_or(r.address_lcase.as_str())
+                })
+                .collect(),
+            is_test: false,
+            is_train: false,
+        }
+    }
+}