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,135 @@
+/*
+ * SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
+ */
+
+use crate::utils::{dns::DnsCache, server::TestServerBuilder};
+use common::expr::{tokenizer::TokenMap, *};
+use mail_auth::{DnssecStatus, MX};
+use registry::schema::{
+    enums::ExpressionVariable,
+    prelude::{ObjectType, Property},
+    structs::{LookupStore, SqliteStore, StoreLookup},
+};
+use smtp::queue::RecipientDomain;
+use std::time::{Duration, Instant};
+
+const TESTS: &[(&str, &str)] = &[
+    ("dns_query(rcpt_domain, 'mx')[0]", "mx.foobar.org"),
+    (
+        "key_get('sql', 'hello') + '-' + key_exists('sql', 'hello') + '-' + key_set('sql', 'hello', 'world') + '-' + key_get('sql', 'hello') + '-' + key_exists('sql', 'hello')",
+        "-0-1-world-1",
+    ),
+    (
+        "counter_get('sql', 'county') + '-' + counter_incr('sql', 'county', 1) + '-' + counter_incr('sql', 'county', 1) + '-' + counter_get('sql', 'county')",
+        "0-1-2-2",
+    ),
+    (
+        "sql_query('sql', 'SELECT description FROM domains WHERE name = ?', 'foobar.org')",
+        "Main domain",
+    ),
+    (
+        "is_local_domain('foobar.org') + '-' + is_local_domain('unknown.org')  + '-' + is_local_address('[email protected]') + '-' + is_local_address('[email protected]')",
+        "1-0-1-0",
+    ),
+    (
+        "is_local_domain('FooBar.org') + '-' + is_local_address('[email protected]') + '-' + is_local_address('[email protected]')",
+        "1-1-1",
+    ),
+];
+
+#[tokio::test]
+async fn expressions() {
+    let mut test = TestServerBuilder::new("smtp_lookup_test")
+        .await
+        .with_http_listener(19017)
+        .await
+        .disable_services()
+        .capture_queue()
+        .build()
+        .await;
+
+    // Create test data
+    let admin = test.account("admin");
+    for (name, secret, description, aliases) in [
+        ("[email protected]", "12345 + extra safety", "John Doe", &[]),
+        ("[email protected]", "abcde + extra safety", "Jane Smith", &[]),
+    ] {
+        admin
+            .create_user_account(name, secret, description, aliases, vec![])
+            .await;
+    }
+    admin
+        .registry_create_object(StoreLookup {
+            namespace: "sql".into(),
+            store: LookupStore::Sqlite(SqliteStore {
+                path: format!("{}/smtp_sql.db", test.tmp_dir()),
+                pool_max_connections: 10,
+                pool_workers: None,
+            }),
+        })
+        .await;
+    admin.reload_lookup_stores().await;
+    test.reload_core();
+
+    test.server.mx_add(
+        "test.org",
+        vec![MX {
+            exchanges: vec!["mx.foobar.org".into()].into_boxed_slice(),
+            preference: 10,
+        }],
+        DnssecStatus::Secure,
+        Instant::now() + Duration::from_secs(10),
+    );
+
+    let sql = test
+        .server
+        .get_lookup_store("sql")
+        .unwrap()
+        .into_store()
+        .unwrap();
+    sql.create_tables().await.unwrap();
+    for query in [
+        "CREATE TABLE domains (name TEXT PRIMARY KEY, description TEXT);",
+        "INSERT INTO domains (name, description) VALUES ('foobar.org', 'Main domain');",
+        "INSERT INTO domains (name, description) VALUES ('foobar.net', 'Secondary domain');",
+        "CREATE TABLE allowed_ips (addr TEXT PRIMARY KEY);",
+        "INSERT INTO allowed_ips (addr) VALUES ('10.0.0.50');",
+    ] {
+        sql.sql_query::<usize>(query, Vec::new()).await.unwrap();
+    }
+
+    // Test expression functions
+    let token_map = TokenMap::default().with_variables(&[
+        ExpressionVariable::Rcpt,
+        ExpressionVariable::RcptDomain,
+        ExpressionVariable::Sender,
+        ExpressionVariable::SenderDomain,
+        ExpressionVariable::Mx,
+        ExpressionVariable::HeloDomain,
+        ExpressionVariable::AuthenticatedAs,
+        ExpressionVariable::Listener,
+        ExpressionVariable::RemoteIp,
+        ExpressionVariable::LocalIp,
+        ExpressionVariable::Priority,
+    ]);
+    for (expr, expected) in TESTS {
+        let e = Expression::parse(&token_map, expr);
+        assert_eq!(
+            test.server
+                .eval_expr::<String, _>(
+                    &e,
+                    &RecipientDomain::new("test.org"),
+                    ObjectType::Account.singleton(),
+                    Property::AccountName,
+                    0
+                )
+                .await
+                .unwrap(),
+            *expected,
+            "failed for '{}'",
+            expr
+        );
+    }
+}