samotop-core 0.13.1+smtp

The domain model of Samotop and core functionality - SMTP server and library built on async-std. A base crate for samotop extensions.
Documentation 
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
121
122
123
124

use super::Esmtp;
use crate::common::S1Fut;
use crate::mail::{DispatchError, MailDispatch};
use crate::smtp::{command::SmtpData, Action, SmtpContext};

impl Action<SmtpData> for Esmtp {
    fn apply<'a, 's, 'f>(&'a self, _cmd: SmtpData, state: &'s mut SmtpContext) -> S1Fut<'f, ()>
    where
        'a: 'f,
        's: 'f,
    {
        Box::pin(async move {
            if state.session.transaction.id.is_empty()
                || state.session.peer_name.is_none()
                || state.session.transaction.mail.is_none()
                || state.session.transaction.rcpts.is_empty()
            {
                state.session.reset();
                state.session.say_command_sequence_fail();
                return;
            }

            match state.service().open_mail_body(&mut state.session).await {
                Ok(()) if state.session.transaction.sink.is_none() => {
                    warn!(
                        "Send_mail returned OK message without sink for transaction {}",
                        state.session.transaction.id
                    );
                    state.session.reset();
                    state.session.say_mail_queue_failed_temporarily();
                }
                Ok(()) => {
                    state.session.say_start_data_challenge();
                }
                Err(DispatchError::Permanent) => {
                    state.session.reset();
                    state.session.say_mail_queue_refused();
                }
                Err(DispatchError::Temporary) => {
                    state.session.reset();
                    state.session.say_mail_queue_failed_temporarily();
                }
            };
        })
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::{
        mail::Recipient,
        smtp::{command::SmtpMail, DriverControl, SmtpPath},
    };

    #[test]
    fn sink_gets_set() {
        async_std::task::block_on(async move {
            let mut set = SmtpContext::default();
            set.session.peer_name = Some("xx.io".to_owned());
            set.session.transaction.id = "someid".to_owned();
            set.session.transaction.mail = Some(SmtpMail::Mail(SmtpPath::Null, vec![]));
            set.session.transaction.rcpts.push(Recipient::null());
            set.session
                .transaction
                .extra_headers
                .insert_str(0, "feeeha");
            let sink: Vec<u8> = vec![];
            set.session.transaction.sink = Some(Box::pin(sink));

            Esmtp.apply(SmtpData, &mut set).await;
            match set.session.pop_control() {
                Some(DriverControl::Response(bytes)) if bytes.starts_with(b"354 ") => {}
                otherwise => panic!("Expected mail data input challenge, got {:?}", otherwise),
            }

            assert!(set.session.transaction.sink.is_some());
        })
    }

    #[test]
    fn command_sequence_is_assured_missing_helo() {
        async_std::task::block_on(async move {
            let mut set = SmtpContext::default();

            Esmtp.apply(SmtpData, &mut set).await;
            match set.session.pop_control() {
                Some(DriverControl::Response(bytes)) if bytes.starts_with(b"503 ") => {}
                otherwise => panic!("Expected command sequence failure, got {:?}", otherwise),
            }
            assert!(set.session.transaction.sink.is_none());
        })
    }

    #[test]
    fn command_sequence_is_assured_missing_mail() {
        async_std::task::block_on(async move {
            let mut set = SmtpContext::default();
            set.session.peer_name = Some("xx.iu".to_owned());

            Esmtp.apply(SmtpData, &mut set).await;
            match set.session.pop_control() {
                Some(DriverControl::Response(bytes)) if bytes.starts_with(b"503 ") => {}
                otherwise => panic!("Expected command sequence failure, got {:?}", otherwise),
            }
            assert!(set.session.transaction.sink.is_none());
        })
    }
    #[test]
    fn command_sequence_is_assured_missing_rcpt() {
        async_std::task::block_on(async move {
            let mut set = SmtpContext::default();
            set.session.peer_name = Some("xx.iu".to_owned());
            set.session.transaction.mail = Some(SmtpMail::Mail(SmtpPath::Null, vec![]));

            Esmtp.apply(SmtpData, &mut set).await;
            match set.session.pop_control() {
                Some(DriverControl::Response(bytes)) if bytes.starts_with(b"503 ") => {}
                otherwise => panic!("Expected command sequence failure, got {:?}", otherwise),
            }
            assert!(set.session.transaction.sink.is_none());
        })
    }
}