1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#[macro_use]
extern crate log;
mod lookup;
use self::lookup::*;
use samotop_core::common::*;
use samotop_core::mail::*;
use samotop_core::smtp::*;
pub use viaspf::Config;
use viaspf::{evaluate_spf, SpfResult};
pub fn provide_viaspf() -> Provider<Config> {
Provider(Config::default())
}
#[derive(Clone, Debug, Default)]
pub struct Provider<T>(pub T);
#[derive(Clone, Debug)]
pub struct SpfService {
config: Arc<Config>,
}
impl SpfService {
pub fn new(config: Config) -> Self {
Self {
config: Arc::new(config),
}
}
}
impl MailSetup for Provider<Config> {
fn setup(self, builder: &mut Builder) {
builder
.dispatch
.insert(0, Box::new(SpfService::new(self.0)));
}
}
impl MailDispatch for SpfService {
fn send_mail<'a, 's, 'f>(
&'a self,
session: &'s SessionInfo,
transaction: Transaction,
) -> S2Fut<'f, DispatchResult>
where
'a: 'f,
's: 'f,
{
let peer_addr = match session.connection.peer_addr.as_str().parse() {
Err(_) => std::net::IpAddr::V4(std::net::Ipv4Addr::UNSPECIFIED),
Ok(ip) => ip,
};
let peer_name = session.peer_name.clone().unwrap_or_default();
let sender = match transaction.mail.as_ref().map(|m| m.sender()) {
None | Some(SmtpPath::Null) | Some(SmtpPath::Postmaster) => String::new(),
Some(SmtpPath::Mailbox { host, .. }) => host.domain(),
};
let fut = async move {
let resolver = match new_resolver().await {
Err(e) => {
error!("Could not crerate resolver! {:?}", e);
return Err(DispatchError::FailedTemporarily);
}
Ok(resolver) => resolver,
};
let evaluation = evaluate_spf(
&resolver,
&self.config,
peer_addr,
sender.as_str(),
peer_name.as_str(),
);
match evaluation.result {
SpfResult::Fail(explanation) => {
info!("mail rejected due to SPF fail: {}", explanation);
Err(DispatchError::Refused)
}
result => {
debug!("mail OK with SPF result: {}", result);
Ok(transaction)
}
}
};
Box::pin(fut)
}
}
#[cfg(test)]
mod tests {
use samotop_core::io::ConnectionInfo;
use super::*;
#[test]
fn default_mail_fut_is_sync() {
let sess = SessionInfo::new(ConnectionInfo::default(), "test".to_owned());
let tran = Transaction {
id: "sessionid".to_owned(),
..Default::default()
};
let cfg = Config::default();
let sut = SpfService::new(cfg);
let fut = sut.send_mail(&sess, tran);
is_sync(fut);
}
#[test]
fn config_is_sync() {
let cfg = Config::default();
is_sync(cfg);
}
fn is_sync<T: Sync>(_subject: T) {}
}