0000: 2f 2f 21 20 54 68 65 20 65 6e 74 72 79 20 70 6f //! The entry po
0010: 69 6e 74 20 69 73 20 70 72 69 76 61 74 65 20 74 int is private t
0020: 6f 20 74 68 65 20 62 69 6e 61 72 79 2c 20 73 6f o the binary, so
0030: 20 74 65 73 74 20 69 74 73 20 70 75 62 6c 69 63 test its public
0040: 20 43 4c 49 20 61 6e 64 20 53 4d 54 50 0a 2f 2f CLI and SMTP.//
0050: 21 20 62 65 68 61 76 69 6f 72 20 77 69 74 68 20 ! behavior with
0060: 69 73 6f 6c 61 74 65 64 20 63 6f 6e 66 69 67 75 isolated configu
0070: 72 61 74 69 6f 6e 20 64 69 72 65 63 74 6f 72 69 ration directori
0080: 65 73 20 61 6e 64 20 62 6f 75 6e 64 65 64 20 73 es and bounded s
0090: 75 62 70 72 6f 63 65 73 73 65 73 2e 0a 75 73 65 ubprocesses..use
00a0: 20 73 74 64 3a 3a 7b 0a 09 66 73 3a 3a 7b 73 65 std::{..fs::{se
00b0: 6c 66 2c 20 4f 70 65 6e 4f 70 74 69 6f 6e 73 7d lf, OpenOptions}
00c0: 2c 0a 09 69 6f 3a 3a 7b 42 75 66 52 65 61 64 2c ,..io::{BufRead,
00d0: 20 42 75 66 52 65 61 64 65 72 2c 20 57 72 69 74 BufReader, Writ
00e0: 65 7d 2c 0a 09 6e 65 74 3a 3a 7b 53 6f 63 6b 65 e},..net::{Socke
00f0: 74 41 64 64 72 2c 20 54 63 70 4c 69 73 74 65 6e tAddr, TcpListen
0100: 65 72 2c 20 54 63 70 53 74 72 65 61 6d 7d 2c 0a er, TcpStream},.
0110: 09 6f 73 3a 3a 75 6e 69 78 3a 3a 66 73 3a 3a 7b .os::unix::fs::{
0120: 4f 70 65 6e 4f 70 74 69 6f 6e 73 45 78 74 2c 20 OpenOptionsExt,
0130: 50 65 72 6d 69 73 73 69 6f 6e 73 45 78 74 7d 2c PermissionsExt},
0140: 0a 09 70 61 74 68 3a 3a 50 61 74 68 42 75 66 2c ..path::PathBuf,
0150: 0a 09 70 72 6f 63 65 73 73 3a 3a 7b 43 68 69 6c ..process::{Chil
0160: 64 2c 20 43 6f 6d 6d 61 6e 64 2c 20 4f 75 74 70 d, Command, Outp
0170: 75 74 2c 20 53 74 64 69 6f 7d 2c 0a 09 73 79 6e ut, Stdio},..syn
0180: 63 3a 3a 61 74 6f 6d 69 63 3a 3a 7b 41 74 6f 6d c::atomic::{Atom
0190: 69 63 55 73 69 7a 65 2c 20 4f 72 64 65 72 69 6e icUsize, Orderin
01a0: 67 7d 2c 0a 09 74 68 72 65 61 64 2c 0a 09 74 69 g},..thread,..ti
01b0: 6d 65 3a 3a 7b 44 75 72 61 74 69 6f 6e 2c 20 49 me::{Duration, I
01c0: 6e 73 74 61 6e 74 7d 2c 0a 7d 3b 0a 63 6f 6e 73 nstant},.};.cons
01d0: 74 20 44 45 41 44 4c 49 4e 45 3a 20 44 75 72 61 t DEADLINE: Dura
01e0: 74 69 6f 6e 20 3d 20 44 75 72 61 74 69 6f 6e 3a tion = Duration:
01f0: 3a 66 72 6f 6d 5f 73 65 63 73 28 35 29 3b 0a 73 :from_secs(5);.s
0200: 74 72 75 63 74 20 54 65 73 74 44 69 72 28 50 61 truct TestDir(Pa
0210: 74 68 42 75 66 29 3b 0a 69 6d 70 6c 20 54 65 73 thBuf);.impl Tes
0220: 74 44 69 72 20 7b 0a 09 66 6e 20 6e 65 77 20 28 tDir {..fn new (
0230: 29 20 2d 3e 20 53 65 6c 66 20 7b 0a 09 09 73 74 ) -> Self {...st
0240: 61 74 69 63 20 4e 45 58 54 3a 20 41 74 6f 6d 69 atic NEXT: Atomi
0250: 63 55 73 69 7a 65 20 3d 20 41 74 6f 6d 69 63 55 cUsize = AtomicU
0260: 73 69 7a 65 3a 3a 6e 65 77 28 30 29 3b 0a 09 09 size::new(0);...
0270: 6c 6f 6f 70 20 7b 0a 09 09 09 6c 65 74 20 70 61 loop {....let pa
0280: 74 68 20 3d 20 73 74 64 3a 3a 65 6e 76 3a 3a 74 th = std::env::t
0290: 65 6d 70 5f 64 69 72 28 29 2e 6a 6f 69 6e 28 66 emp_dir().join(f
02a0: 6f 72 6d 61 74 21 28 22 73 6d 74 70 32 74 67 2d ormat!("smtp2tg-
02b0: 63 6c 69 2d 7b 7d 2d 7b 7d 22 2c 0a 09 09 09 09 cli-{}-{}",.....
02c0: 73 74 64 3a 3a 70 72 6f 63 65 73 73 3a 3a 69 64 std::process::id
02d0: 28 29 2c 20 4e 45 58 54 2e 66 65 74 63 68 5f 61 (), NEXT.fetch_a
02e0: 64 64 28 31 2c 20 4f 72 64 65 72 69 6e 67 3a 3a dd(1, Ordering::
02f0: 52 65 6c 61 78 65 64 29 29 29 3b 0a 09 09 09 6d Relaxed)));....m
0300: 61 74 63 68 20 66 73 3a 3a 63 72 65 61 74 65 5f atch fs::create_
0310: 64 69 72 28 26 70 61 74 68 29 20 7b 0a 09 09 09 dir(&path) {....
0320: 09 4f 6b 28 28 29 29 20 3d 3e 20 72 65 74 75 72 .Ok(()) => retur
0330: 6e 20 53 65 6c 66 28 70 61 74 68 29 2c 0a 09 09 n Self(path),...
0340: 09 09 45 72 72 28 65 72 72 6f 72 29 20 69 66 20 ..Err(error) if
0350: 65 72 72 6f 72 2e 6b 69 6e 64 28 29 20 3d 3d 20 error.kind() ==
0360: 73 74 64 3a 3a 69 6f 3a 3a 45 72 72 6f 72 4b 69 std::io::ErrorKi
0370: 6e 64 3a 3a 41 6c 72 65 61 64 79 45 78 69 73 74 nd::AlreadyExist
0380: 73 20 3d 3e 20 63 6f 6e 74 69 6e 75 65 2c 0a 09 s => continue,..
0390: 09 09 09 45 72 72 28 65 72 72 6f 72 29 20 3d 3e ...Err(error) =>
03a0: 20 70 61 6e 69 63 21 28 22 43 61 6e 6e 6f 74 20 panic!("Cannot
03b0: 63 72 65 61 74 65 20 74 65 73 74 20 64 69 72 65 create test dire
03c0: 63 74 6f 72 79 3a 20 7b 65 72 72 6f 72 7d 22 29 ctory: {error}")
03d0: 2c 0a 09 09 09 7d 0a 09 09 7d 0a 09 7d 0a 09 66 ,....}...}..}..f
03e0: 6e 20 63 6f 6e 66 69 67 20 28 26 73 65 6c 66 2c n config (&self,
03f0: 20 6e 61 6d 65 3a 20 26 73 74 72 2c 20 74 65 78 name: &str, tex
0400: 74 3a 20 26 73 74 72 2c 20 6d 6f 64 65 3a 20 75 t: &str, mode: u
0410: 33 32 29 20 7b 0a 09 09 6c 65 74 20 70 61 74 68 32) {...let path
0420: 20 3d 20 73 65 6c 66 2e 30 2e 6a 6f 69 6e 28 6e = self.0.join(n
0430: 61 6d 65 29 3b 0a 09 09 6c 65 74 20 6d 75 74 20 ame);...let mut
0440: 66 69 6c 65 20 3d 20 4f 70 65 6e 4f 70 74 69 6f file = OpenOptio
0450: 6e 73 3a 3a 6e 65 77 28 29 2e 63 72 65 61 74 65 ns::new().create
0460: 5f 6e 65 77 28 74 72 75 65 29 2e 77 72 69 74 65 _new(true).write
0470: 28 74 72 75 65 29 2e 6d 6f 64 65 28 30 6f 36 30 (true).mode(0o60
0480: 30 29 0a 09 09 09 2e 6f 70 65 6e 28 26 70 61 74 0).....open(&pat
0490: 68 29 2e 75 6e 77 72 61 70 28 29 3b 0a 09 09 66 h).unwrap();...f
04a0: 69 6c 65 2e 77 72 69 74 65 5f 61 6c 6c 28 74 65 ile.write_all(te
04b0: 78 74 2e 61 73 5f 62 79 74 65 73 28 29 29 2e 75 xt.as_bytes()).u
04c0: 6e 77 72 61 70 28 29 3b 0a 09 09 66 73 3a 3a 73 nwrap();...fs::s
04d0: 65 74 5f 70 65 72 6d 69 73 73 69 6f 6e 73 28 70 et_permissions(p
04e0: 61 74 68 2c 20 66 73 3a 3a 50 65 72 6d 69 73 73 ath, fs::Permiss
04f0: 69 6f 6e 73 3a 3a 66 72 6f 6d 5f 6d 6f 64 65 28 ions::from_mode(
0500: 6d 6f 64 65 29 29 2e 75 6e 77 72 61 70 28 29 3b mode)).unwrap();
0510: 0a 09 7d 0a 09 66 6e 20 73 70 61 77 6e 20 28 26 ..}..fn spawn (&
0520: 73 65 6c 66 2c 20 61 72 67 73 3a 20 26 5b 26 73 self, args: &[&s
0530: 74 72 5d 29 20 2d 3e 20 50 72 6f 63 65 73 73 20 tr]) -> Process
0540: 7b 0a 09 09 50 72 6f 63 65 73 73 28 53 6f 6d 65 {...Process(Some
0550: 28 43 6f 6d 6d 61 6e 64 3a 3a 6e 65 77 28 65 6e (Command::new(en
0560: 76 21 28 22 43 41 52 47 4f 5f 42 49 4e 5f 45 58 v!("CARGO_BIN_EX
0570: 45 5f 73 6d 74 70 32 74 67 22 29 29 0a 09 09 09 E_smtp2tg"))....
0580: 2e 63 75 72 72 65 6e 74 5f 64 69 72 28 26 73 65 .current_dir(&se
0590: 6c 66 2e 30 29 2e 61 72 67 73 28 61 72 67 73 29 lf.0).args(args)
05a0: 0a 09 09 09 2e 73 74 64 69 6e 28 53 74 64 69 6f .....stdin(Stdio
05b0: 3a 3a 6e 75 6c 6c 28 29 29 2e 73 74 64 6f 75 74 ::null()).stdout
05c0: 28 53 74 64 69 6f 3a 3a 70 69 70 65 64 28 29 29 (Stdio::piped())
05d0: 2e 73 74 64 65 72 72 28 53 74 64 69 6f 3a 3a 70 .stderr(Stdio::p
05e0: 69 70 65 64 28 29 29 0a 09 09 09 2e 73 70 61 77 iped()).....spaw
05f0: 6e 28 29 2e 75 6e 77 72 61 70 28 29 29 29 0a 09 n().unwrap()))..
0600: 7d 0a 09 66 6e 20 72 75 6e 20 28 26 73 65 6c 66 }..fn run (&self
0610: 2c 20 61 72 67 73 3a 20 26 5b 26 73 74 72 5d 29 , args: &[&str])
0620: 20 2d 3e 20 4f 75 74 70 75 74 20 7b 0a 09 09 73 -> Output {...s
0630: 65 6c 66 2e 73 70 61 77 6e 28 61 72 67 73 29 2e elf.spawn(args).
0640: 6f 75 74 70 75 74 28 29 0a 09 7d 0a 7d 0a 69 6d output()..}.}.im
0650: 70 6c 20 44 72 6f 70 20 66 6f 72 20 54 65 73 74 pl Drop for Test
0660: 44 69 72 20 7b 0a 09 66 6e 20 64 72 6f 70 20 28 Dir {..fn drop (
0670: 26 6d 75 74 20 73 65 6c 66 29 20 7b 0a 09 09 6c &mut self) {...l
0680: 65 74 20 5f 20 3d 20 66 73 3a 3a 72 65 6d 6f 76 et _ = fs::remov
0690: 65 5f 64 69 72 5f 61 6c 6c 28 26 73 65 6c 66 2e e_dir_all(&self.
06a0: 30 29 3b 0a 09 7d 0a 7d 0a 73 74 72 75 63 74 20 0);..}.}.struct
06b0: 50 72 6f 63 65 73 73 28 4f 70 74 69 6f 6e 3c 43 Process(Option<C
06c0: 68 69 6c 64 3e 29 3b 0a 69 6d 70 6c 20 50 72 6f hild>);.impl Pro
06d0: 63 65 73 73 20 7b 0a 09 66 6e 20 6f 75 74 70 75 cess {..fn outpu
06e0: 74 20 28 6d 75 74 20 73 65 6c 66 29 20 2d 3e 20 t (mut self) ->
06f0: 4f 75 74 70 75 74 20 7b 0a 09 09 6c 65 74 20 64 Output {...let d
0700: 65 61 64 6c 69 6e 65 20 3d 20 49 6e 73 74 61 6e eadline = Instan
0710: 74 3a 3a 6e 6f 77 28 29 20 2b 20 44 45 41 44 4c t::now() + DEADL
0720: 49 4e 45 3b 0a 09 09 77 68 69 6c 65 20 73 65 6c INE;...while sel
0730: 66 2e 30 2e 61 73 5f 6d 75 74 28 29 2e 75 6e 77 f.0.as_mut().unw
0740: 72 61 70 28 29 2e 74 72 79 5f 77 61 69 74 28 29 rap().try_wait()
0750: 2e 75 6e 77 72 61 70 28 29 2e 69 73 5f 6e 6f 6e .unwrap().is_non
0760: 65 28 29 20 7b 0a 09 09 09 61 73 73 65 72 74 21 e() {....assert!
0770: 28 49 6e 73 74 61 6e 74 3a 3a 6e 6f 77 28 29 20 (Instant::now()
0780: 3c 20 64 65 61 64 6c 69 6e 65 2c 20 22 43 4c 49 < deadline, "CLI
0790: 20 64 69 64 20 6e 6f 74 20 65 78 69 74 20 77 69 did not exit wi
07a0: 74 68 69 6e 20 7b 44 45 41 44 4c 49 4e 45 3a 3f thin {DEADLINE:?
07b0: 7d 22 29 3b 0a 09 09 09 74 68 72 65 61 64 3a 3a }");....thread::
07c0: 73 6c 65 65 70 28 44 75 72 61 74 69 6f 6e 3a 3a sleep(Duration::
07d0: 66 72 6f 6d 5f 6d 69 6c 6c 69 73 28 31 30 29 29 from_millis(10))
07e0: 3b 0a 09 09 7d 0a 09 09 73 65 6c 66 2e 30 2e 74 ;...}...self.0.t
07f0: 61 6b 65 28 29 2e 75 6e 77 72 61 70 28 29 2e 77 ake().unwrap().w
0800: 61 69 74 5f 77 69 74 68 5f 6f 75 74 70 75 74 28 ait_with_output(
0810: 29 2e 75 6e 77 72 61 70 28 29 0a 09 7d 0a 09 66 ).unwrap()..}..f
0820: 6e 20 63 6f 6e 6e 65 63 74 20 28 26 6d 75 74 20 n connect (&mut
0830: 73 65 6c 66 2c 20 61 64 64 72 65 73 73 3a 20 53 self, address: S
0840: 6f 63 6b 65 74 41 64 64 72 29 20 2d 3e 20 54 63 ocketAddr) -> Tc
0850: 70 53 74 72 65 61 6d 20 7b 0a 09 09 6c 65 74 20 pStream {...let
0860: 64 65 61 64 6c 69 6e 65 20 3d 20 49 6e 73 74 61 deadline = Insta
0870: 6e 74 3a 3a 6e 6f 77 28 29 20 2b 20 44 45 41 44 nt::now() + DEAD
0880: 4c 49 4e 45 3b 0a 09 09 6c 6f 6f 70 20 7b 0a 09 LINE;...loop {..
0890: 09 09 61 73 73 65 72 74 21 28 73 65 6c 66 2e 30 ..assert!(self.0
08a0: 2e 61 73 5f 6d 75 74 28 29 2e 75 6e 77 72 61 70 .as_mut().unwrap
08b0: 28 29 2e 74 72 79 5f 77 61 69 74 28 29 2e 75 6e ().try_wait().un
08c0: 77 72 61 70 28 29 2e 69 73 5f 6e 6f 6e 65 28 29 wrap().is_none()
08d0: 2c 0a 09 09 09 09 22 53 4d 54 50 20 73 65 72 76 ,....."SMTP serv
08e0: 65 72 20 65 78 69 74 65 64 20 62 65 66 6f 72 65 er exited before
08f0: 20 61 63 63 65 70 74 69 6e 67 20 63 6f 6e 6e 65 accepting conne
0900: 63 74 69 6f 6e 73 22 29 3b 0a 09 09 09 69 66 20 ctions");....if
0910: 6c 65 74 20 4f 6b 28 73 74 72 65 61 6d 29 20 3d let Ok(stream) =
0920: 20 54 63 70 53 74 72 65 61 6d 3a 3a 63 6f 6e 6e TcpStream::conn
0930: 65 63 74 5f 74 69 6d 65 6f 75 74 28 26 61 64 64 ect_timeout(&add
0940: 72 65 73 73 2c 20 44 75 72 61 74 69 6f 6e 3a 3a ress, Duration::
0950: 66 72 6f 6d 5f 6d 69 6c 6c 69 73 28 31 30 30 29 from_millis(100)
0960: 29 20 7b 0a 09 09 09 09 73 74 72 65 61 6d 2e 73 ) {.....stream.s
0970: 65 74 5f 72 65 61 64 5f 74 69 6d 65 6f 75 74 28 et_read_timeout(
0980: 53 6f 6d 65 28 44 45 41 44 4c 49 4e 45 29 29 2e Some(DEADLINE)).
0990: 75 6e 77 72 61 70 28 29 3b 0a 09 09 09 09 73 74 unwrap();.....st
09a0: 72 65 61 6d 2e 73 65 74 5f 77 72 69 74 65 5f 74 ream.set_write_t
09b0: 69 6d 65 6f 75 74 28 53 6f 6d 65 28 44 45 41 44 imeout(Some(DEAD
09c0: 4c 49 4e 45 29 29 2e 75 6e 77 72 61 70 28 29 3b LINE)).unwrap();
09d0: 0a 09 09 09 09 72 65 74 75 72 6e 20 73 74 72 65 .....return stre
09e0: 61 6d 3b 0a 09 09 09 7d 0a 09 09 09 61 73 73 65 am;....}....asse
09f0: 72 74 21 28 49 6e 73 74 61 6e 74 3a 3a 6e 6f 77 rt!(Instant::now
0a00: 28 29 20 3c 20 64 65 61 64 6c 69 6e 65 2c 20 22 () < deadline, "
0a10: 53 4d 54 50 20 73 65 72 76 65 72 20 64 69 64 20 SMTP server did
0a20: 6e 6f 74 20 73 74 61 72 74 20 77 69 74 68 69 6e not start within
0a30: 20 7b 44 45 41 44 4c 49 4e 45 3a 3f 7d 22 29 3b {DEADLINE:?}");
0a40: 0a 09 09 09 74 68 72 65 61 64 3a 3a 73 6c 65 65 ....thread::slee
0a50: 70 28 44 75 72 61 74 69 6f 6e 3a 3a 66 72 6f 6d p(Duration::from
0a60: 5f 6d 69 6c 6c 69 73 28 31 30 29 29 3b 0a 09 09 _millis(10));...
0a70: 7d 0a 09 7d 0a 7d 0a 69 6d 70 6c 20 44 72 6f 70 }..}.}.impl Drop
0a80: 20 66 6f 72 20 50 72 6f 63 65 73 73 20 7b 0a 09 for Process {..
0a90: 66 6e 20 64 72 6f 70 20 28 26 6d 75 74 20 73 65 fn drop (&mut se
0aa0: 6c 66 29 20 7b 0a 09 09 69 66 20 6c 65 74 20 53 lf) {...if let S
0ab0: 6f 6d 65 28 63 68 69 6c 64 29 20 3d 20 26 6d 75 ome(child) = &mu
0ac0: 74 20 73 65 6c 66 2e 30 20 7b 0a 09 09 09 6c 65 t self.0 {....le
0ad0: 74 20 5f 20 3d 20 63 68 69 6c 64 2e 6b 69 6c 6c t _ = child.kill
0ae0: 28 29 3b 0a 09 09 09 6c 65 74 20 5f 20 3d 20 63 ();....let _ = c
0af0: 68 69 6c 64 2e 77 61 69 74 28 29 3b 0a 09 09 7d hild.wait();...}
0b00: 0a 09 7d 0a 7d 0a 66 6e 20 61 73 73 65 72 74 5f ..}.}.fn assert_
0b10: 65 72 72 6f 72 20 28 6f 75 74 70 75 74 3a 20 4f error (output: O
0b20: 75 74 70 75 74 2c 20 65 78 70 65 63 74 65 64 3a utput, expected:
0b30: 20 26 73 74 72 29 20 7b 0a 09 6c 65 74 20 73 74 &str) {..let st
0b40: 64 65 72 72 20 3d 20 53 74 72 69 6e 67 3a 3a 66 derr = String::f
0b50: 72 6f 6d 5f 75 74 66 38 5f 6c 6f 73 73 79 28 26 rom_utf8_lossy(&
0b60: 6f 75 74 70 75 74 2e 73 74 64 65 72 72 29 3b 0a output.stderr);.
0b70: 09 61 73 73 65 72 74 21 28 21 6f 75 74 70 75 74 .assert!(!output
0b80: 2e 73 74 61 74 75 73 2e 73 75 63 63 65 73 73 28 .status.success(
0b90: 29 2c 20 22 45 78 70 65 63 74 65 64 20 66 61 69 ), "Expected fai
0ba0: 6c 75 72 65 2c 20 73 74 64 65 72 72 3a 20 7b 73 lure, stderr: {s
0bb0: 74 64 65 72 72 7d 22 29 3b 0a 09 61 73 73 65 72 tderr}");..asser
0bc0: 74 21 28 73 74 64 65 72 72 2e 63 6f 6e 74 61 69 t!(stderr.contai
0bd0: 6e 73 28 65 78 70 65 63 74 65 64 29 2c 20 22 4d ns(expected), "M
0be0: 69 73 73 69 6e 67 20 7b 65 78 70 65 63 74 65 64 issing {expected
0bf0: 3a 3f 7d 20 69 6e 20 73 74 64 65 72 72 3a 20 7b :?} in stderr: {
0c00: 73 74 64 65 72 72 7d 22 29 3b 0a 7d 0a 23 5b 74 stderr}");.}.#[t
0c10: 65 73 74 5d 0a 66 6e 20 68 65 6c 70 5f 73 75 63 est].fn help_suc
0c20: 63 65 65 64 73 5f 77 69 74 68 6f 75 74 5f 61 5f ceeds_without_a_
0c30: 63 6f 6e 66 69 67 75 72 61 74 69 6f 6e 5f 66 69 configuration_fi
0c40: 6c 65 20 28 29 20 7b 0a 09 6c 65 74 20 64 69 72 le () {..let dir
0c50: 20 3d 20 54 65 73 74 44 69 72 3a 3a 6e 65 77 28 = TestDir::new(
0c60: 29 3b 0a 09 66 6f 72 20 66 6c 61 67 20 69 6e 20 );..for flag in
0c70: 5b 22 2d 68 22 2c 20 22 2d 2d 68 65 6c 70 22 5d ["-h", "--help"]
0c80: 20 7b 0a 09 09 6c 65 74 20 6f 75 74 70 75 74 20 {...let output
0c90: 3d 20 64 69 72 2e 72 75 6e 28 26 5b 66 6c 61 67 = dir.run(&[flag
0ca0: 5d 29 3b 0a 09 09 61 73 73 65 72 74 21 28 6f 75 ]);...assert!(ou
0cb0: 74 70 75 74 2e 73 74 61 74 75 73 2e 73 75 63 63 tput.status.succ
0cc0: 65 73 73 28 29 29 3b 0a 09 09 6c 65 74 20 68 65 ess());...let he
0cd0: 6c 70 20 3d 20 53 74 72 69 6e 67 3a 3a 66 72 6f lp = String::fro
0ce0: 6d 5f 75 74 66 38 28 6f 75 74 70 75 74 2e 73 74 m_utf8(output.st
0cf0: 64 6f 75 74 29 2e 75 6e 77 72 61 70 28 29 3b 0a dout).unwrap();.
0d00: 09 09 66 6f 72 20 65 78 70 65 63 74 65 64 20 69 ..for expected i
0d10: 6e 20 5b 22 53 4d 54 50 2d 74 6f 2d 54 65 6c 65 n ["SMTP-to-Tele
0d20: 67 72 61 6d 20 67 61 74 65 77 61 79 22 2c 20 65 gram gateway", e
0d30: 6e 76 21 28 22 43 41 52 47 4f 5f 50 4b 47 5f 56 nv!("CARGO_PKG_V
0d40: 45 52 53 49 4f 4e 22 29 2c 0a 09 09 09 22 2d 63 ERSION"),...."-c
0d50: 2c 20 2d 2d 63 6f 6e 66 69 67 22 2c 20 22 5b 64 , --config", "[d
0d60: 65 66 61 75 6c 74 3a 20 73 6d 74 70 32 74 67 2e efault: smtp2tg.
0d70: 74 6f 6d 6c 5d 22 5d 20 7b 0a 09 09 09 61 73 73 toml]"] {....ass
0d80: 65 72 74 21 28 68 65 6c 70 2e 63 6f 6e 74 61 69 ert!(help.contai
0d90: 6e 73 28 65 78 70 65 63 74 65 64 29 2c 20 22 4d ns(expected), "M
0da0: 69 73 73 69 6e 67 20 7b 65 78 70 65 63 74 65 64 issing {expected
0db0: 3a 3f 7d 20 69 6e 20 68 65 6c 70 3a 20 7b 68 65 :?} in help: {he
0dc0: 6c 70 7d 22 29 3b 0a 09 09 7d 0a 09 7d 0a 7d 0a lp}");...}..}.}.
0dd0: 23 5b 74 65 73 74 5d 0a 66 6e 20 6d 69 73 73 69 #[test].fn missi
0de0: 6e 67 5f 64 65 66 61 75 6c 74 5f 61 6e 64 5f 65 ng_default_and_e
0df0: 78 70 6c 69 63 69 74 5f 63 6f 6e 66 69 67 75 72 xplicit_configur
0e00: 61 74 69 6f 6e 5f 70 61 74 68 73 5f 61 72 65 5f ation_paths_are_
0e10: 72 65 70 6f 72 74 65 64 20 28 29 20 7b 0a 09 6c reported () {..l
0e20: 65 74 20 64 69 72 20 3d 20 54 65 73 74 44 69 72 et dir = TestDir
0e30: 3a 3a 6e 65 77 28 29 3b 0a 09 66 6f 72 20 28 61 ::new();..for (a
0e40: 72 67 73 2c 20 70 61 74 68 29 20 69 6e 20 5b 0a rgs, path) in [.
0e50: 09 09 28 76 65 63 21 5b 5d 2c 20 22 73 6d 74 70 ..(vec![], "smtp
0e60: 32 74 67 2e 74 6f 6d 6c 22 29 2c 0a 09 09 28 76 2tg.toml"),...(v
0e70: 65 63 21 5b 22 2d 63 22 2c 20 22 6d 69 73 73 69 ec!["-c", "missi
0e80: 6e 67 20 66 69 6c 65 2e 74 6f 6d 6c 22 5d 2c 20 ng file.toml"],
0e90: 22 6d 69 73 73 69 6e 67 20 66 69 6c 65 2e 74 6f "missing file.to
0ea0: 6d 6c 22 29 2c 0a 09 09 28 76 65 63 21 5b 22 2d ml"),...(vec!["-
0eb0: 2d 63 6f 6e 66 69 67 22 2c 20 22 6d 69 73 73 69 -config", "missi
0ec0: 6e 67 20 66 69 6c 65 2e 74 6f 6d 6c 22 5d 2c 20 ng file.toml"],
0ed0: 22 6d 69 73 73 69 6e 67 20 66 69 6c 65 2e 74 6f "missing file.to
0ee0: 6d 6c 22 29 2c 0a 09 5d 20 7b 0a 09 09 6c 65 74 ml"),..] {...let
0ef0: 20 6f 75 74 70 75 74 20 3d 20 64 69 72 2e 72 75 output = dir.ru
0f00: 6e 28 26 61 72 67 73 29 3b 0a 09 09 6c 65 74 20 n(&args);...let
0f10: 73 74 64 65 72 72 20 3d 20 53 74 72 69 6e 67 3a stderr = String:
0f20: 3a 66 72 6f 6d 5f 75 74 66 38 5f 6c 6f 73 73 79 :from_utf8_lossy
0f30: 28 26 6f 75 74 70 75 74 2e 73 74 64 65 72 72 29 (&output.stderr)
0f40: 3b 0a 09 09 61 73 73 65 72 74 21 28 73 74 64 65 ;...assert!(stde
0f50: 72 72 2e 63 6f 6e 74 61 69 6e 73 28 70 61 74 68 rr.contains(path
0f60: 29 2c 20 22 4d 69 73 73 69 6e 67 20 70 61 74 68 ), "Missing path
0f70: 20 69 6e 20 64 69 61 67 6e 6f 73 74 69 63 3a 20 in diagnostic:
0f80: 7b 73 74 64 65 72 72 7d 22 29 3b 0a 09 09 61 73 {stderr}");...as
0f90: 73 65 72 74 5f 65 72 72 6f 72 28 6f 75 74 70 75 sert_error(outpu
0fa0: 74 2c 20 22 43 6f 6e 66 69 67 75 72 61 74 69 6f t, "Configuratio
0fb0: 6e 20 66 69 6c 65 20 6e 6f 74 20 66 6f 75 6e 64 n file not found
0fc0: 22 29 3b 0a 09 7d 0a 7d 0a 23 5b 74 65 73 74 5d ");..}.}.#[test]
0fd0: 0a 66 6e 20 69 6e 76 61 6c 69 64 5f 61 72 67 75 .fn invalid_argu
0fe0: 6d 65 6e 74 73 5f 66 61 69 6c 5f 62 65 66 6f 72 ments_fail_befor
0ff0: 65 5f 63 6f 6e 66 69 67 75 72 61 74 69 6f 6e 5f e_configuration_
1000: 6c 6f 61 64 69 6e 67 20 28 29 20 7b 0a 09 6c 65 loading () {..le
1010: 74 20 64 69 72 20 3d 20 54 65 73 74 44 69 72 3a t dir = TestDir:
1020: 3a 6e 65 77 28 29 3b 0a 09 66 6f 72 20 61 72 67 :new();..for arg
1030: 73 20 69 6e 20 5b 76 65 63 21 5b 22 2d 63 22 5d s in [vec!["-c"]
1040: 2c 20 76 65 63 21 5b 22 2d 2d 63 6f 6e 66 69 67 , vec!["--config
1050: 22 5d 2c 20 76 65 63 21 5b 22 2d 2d 75 6e 6b 6e "], vec!["--unkn
1060: 6f 77 6e 2d 6f 70 74 69 6f 6e 22 5d 5d 20 7b 0a own-option"]] {.
1070: 09 09 6c 65 74 20 6f 75 74 70 75 74 20 3d 20 64 ..let output = d
1080: 69 72 2e 72 75 6e 28 26 61 72 67 73 29 3b 0a 09 ir.run(&args);..
1090: 09 61 73 73 65 72 74 5f 65 71 21 28 6f 75 74 70 .assert_eq!(outp
10a0: 75 74 2e 73 74 61 74 75 73 2e 63 6f 64 65 28 29 ut.status.code()
10b0: 2c 20 53 6f 6d 65 28 32 29 2c 20 22 61 72 67 75 , Some(2), "argu
10c0: 6d 65 6e 74 73 3a 20 7b 61 72 67 73 3a 3f 7d 22 ments: {args:?}"
10d0: 29 3b 0a 09 09 61 73 73 65 72 74 21 28 21 53 74 );...assert!(!St
10e0: 72 69 6e 67 3a 3a 66 72 6f 6d 5f 75 74 66 38 5f ring::from_utf8_
10f0: 6c 6f 73 73 79 28 26 6f 75 74 70 75 74 2e 73 74 lossy(&output.st
1100: 64 65 72 72 29 2e 63 6f 6e 74 61 69 6e 73 28 22 derr).contains("
1110: 43 6f 6e 66 69 67 75 72 61 74 69 6f 6e 20 66 69 Configuration fi
1120: 6c 65 20 6e 6f 74 20 66 6f 75 6e 64 22 29 29 3b le not found"));
1130: 0a 09 7d 0a 7d 0a 23 5b 74 65 73 74 5d 0a 66 6e ..}.}.#[test].fn
1140: 20 69 6e 73 65 63 75 72 65 5f 70 65 72 6d 69 73 insecure_permis
1150: 73 69 6f 6e 73 5f 61 72 65 5f 72 65 6a 65 63 74 sions_are_reject
1160: 65 64 5f 62 65 66 6f 72 65 5f 70 61 72 73 69 6e ed_before_parsin
1170: 67 20 28 29 20 7b 0a 09 2f 2f 20 45 61 63 68 20 g () {..// Each
1180: 66 6f 72 62 69 64 64 65 6e 20 70 65 72 6d 69 73 forbidden permis
1190: 73 69 6f 6e 20 69 73 20 69 6e 64 65 70 65 6e 64 sion is independ
11a0: 65 6e 74 3a 20 67 72 6f 75 70 20 72 65 61 64 2f ent: group read/
11b0: 77 72 69 74 65 2f 65 78 65 63 75 74 65 2c 0a 09 write/execute,..
11c0: 2f 2f 20 6f 74 68 65 72 20 72 65 61 64 2f 77 72 // other read/wr
11d0: 69 74 65 2f 65 78 65 63 75 74 65 2c 20 61 6e 64 ite/execute, and
11e0: 20 6f 77 6e 65 72 20 65 78 65 63 75 74 65 2e 0a owner execute..
11f0: 09 66 6f 72 20 6d 6f 64 65 20 69 6e 20 5b 30 6f .for mode in [0o
1200: 36 34 30 2c 20 30 6f 36 32 30 2c 20 30 6f 36 31 640, 0o620, 0o61
1210: 30 2c 20 30 6f 36 30 34 2c 20 30 6f 36 30 32 2c 0, 0o604, 0o602,
1220: 20 30 6f 36 30 31 2c 20 30 6f 37 30 30 5d 20 7b 0o601, 0o700] {
1230: 0a 09 09 6c 65 74 20 64 69 72 20 3d 20 54 65 73 ...let dir = Tes
1240: 74 44 69 72 3a 3a 6e 65 77 28 29 3b 0a 09 09 64 tDir::new();...d
1250: 69 72 2e 63 6f 6e 66 69 67 28 22 73 6d 74 70 32 ir.config("smtp2
1260: 74 67 2e 74 6f 6d 6c 22 2c 20 22 74 68 69 73 20 tg.toml", "this
1270: 69 73 20 6e 6f 74 20 76 61 6c 69 64 20 54 4f 4d is not valid TOM
1280: 4c 22 2c 20 6d 6f 64 65 29 3b 0a 09 09 6c 65 74 L", mode);...let
1290: 20 6f 75 74 70 75 74 20 3d 20 64 69 72 2e 72 75 output = dir.ru
12a0: 6e 28 26 5b 5d 29 3b 0a 09 09 6c 65 74 20 73 74 n(&[]);...let st
12b0: 64 65 72 72 20 3d 20 53 74 72 69 6e 67 3a 3a 66 derr = String::f
12c0: 72 6f 6d 5f 75 74 66 38 5f 6c 6f 73 73 79 28 26 rom_utf8_lossy(&
12d0: 6f 75 74 70 75 74 2e 73 74 64 65 72 72 29 3b 0a output.stderr);.
12e0: 09 09 61 73 73 65 72 74 21 28 73 74 64 65 72 72 ..assert!(stderr
12f0: 2e 63 6f 6e 74 61 69 6e 73 28 22 52 65 71 75 69 .contains("Requi
1300: 72 65 64 3a 20 30 36 30 30 22 29 2c 20 22 6d 6f red: 0600"), "mo
1310: 64 65 20 7b 6d 6f 64 65 3a 6f 7d 3a 20 7b 73 74 de {mode:o}: {st
1320: 64 65 72 72 7d 22 29 3b 0a 09 09 61 73 73 65 72 derr}");...asser
1330: 74 21 28 73 74 64 65 72 72 2e 63 6f 6e 74 61 69 t!(stderr.contai
1340: 6e 73 28 22 63 68 6d 6f 64 20 36 30 30 22 29 2c ns("chmod 600"),
1350: 20 22 6d 6f 64 65 20 7b 6d 6f 64 65 3a 6f 7d 3a "mode {mode:o}:
1360: 20 7b 73 74 64 65 72 72 7d 22 29 3b 0a 09 09 61 {stderr}");...a
1370: 73 73 65 72 74 5f 65 72 72 6f 72 28 6f 75 74 70 ssert_error(outp
1380: 75 74 2c 20 22 43 6f 6e 66 69 67 75 72 61 74 69 ut, "Configurati
1390: 6f 6e 20 66 69 6c 65 20 70 65 72 6d 69 73 73 69 on file permissi
13a0: 6f 6e 73 20 61 72 65 20 69 6e 73 65 63 75 72 65 ons are insecure
13b0: 22 29 3b 0a 09 7d 0a 7d 0a 23 5b 74 65 73 74 5d ");..}.}.#[test]
13c0: 0a 66 6e 20 73 65 63 75 72 65 5f 6d 61 6c 66 6f .fn secure_malfo
13d0: 72 6d 65 64 5f 63 6f 6e 66 69 67 5f 72 65 70 6f rmed_config_repo
13e0: 72 74 73 5f 70 61 72 73 65 5f 63 6f 6e 74 65 78 rts_parse_contex
13f0: 74 20 28 29 20 7b 0a 09 6c 65 74 20 64 69 72 20 t () {..let dir
1400: 3d 20 54 65 73 74 44 69 72 3a 3a 6e 65 77 28 29 = TestDir::new()
1410: 3b 0a 09 2f 2f 20 41 20 73 70 61 63 65 73 2d 63 ;..// A spaces-c
1420: 6f 6e 74 61 69 6e 69 6e 67 20 70 61 74 68 20 6d ontaining path m
1430: 75 73 74 20 62 65 20 70 61 73 73 65 64 20 69 6e ust be passed in
1440: 74 61 63 74 2c 20 61 6e 64 20 74 68 65 20 73 65 tact, and the se
1450: 6c 65 63 74 65 64 20 66 69 6c 65 0a 09 2f 2f 20 lected file..//
1460: 6d 75 73 74 20 74 61 6b 65 20 70 72 65 63 65 64 must take preced
1470: 65 6e 63 65 20 6f 76 65 72 20 61 6e 20 69 6e 73 ence over an ins
1480: 65 63 75 72 65 20 64 65 66 61 75 6c 74 20 63 6f ecure default co
1490: 6e 66 69 67 75 72 61 74 69 6f 6e 2e 0a 09 64 69 nfiguration...di
14a0: 72 2e 63 6f 6e 66 69 67 28 22 73 6d 74 70 32 74 r.config("smtp2t
14b0: 67 2e 74 6f 6d 6c 22 2c 20 22 69 67 6e 6f 72 65 g.toml", "ignore
14c0: 64 22 2c 20 30 6f 36 34 34 29 3b 0a 09 64 69 72 d", 0o644);..dir
14d0: 2e 63 6f 6e 66 69 67 28 22 73 65 6c 65 63 74 65 .config("selecte
14e0: 64 20 63 6f 6e 66 69 67 2e 74 6f 6d 6c 22 2c 20 d config.toml",
14f0: 22 61 70 69 5f 6b 65 79 20 3d 20 5b 22 2c 20 30 "api_key = [", 0
1500: 6f 36 30 30 29 3b 0a 09 66 6f 72 20 66 6c 61 67 o600);..for flag
1510: 20 69 6e 20 5b 22 2d 63 22 2c 20 22 2d 2d 63 6f in ["-c", "--co
1520: 6e 66 69 67 22 5d 20 7b 0a 09 09 6c 65 74 20 6f nfig"] {...let o
1530: 75 74 70 75 74 20 3d 20 64 69 72 2e 72 75 6e 28 utput = dir.run(
1540: 26 5b 66 6c 61 67 2c 20 22 73 65 6c 65 63 74 65 &[flag, "selecte
1550: 64 20 63 6f 6e 66 69 67 2e 74 6f 6d 6c 22 5d 29 d config.toml"])
1560: 3b 0a 09 09 6c 65 74 20 73 74 64 65 72 72 20 3d ;...let stderr =
1570: 20 53 74 72 69 6e 67 3a 3a 66 72 6f 6d 5f 75 74 String::from_ut
1580: 66 38 5f 6c 6f 73 73 79 28 26 6f 75 74 70 75 74 f8_lossy(&output
1590: 2e 73 74 64 65 72 72 29 3b 0a 09 09 61 73 73 65 .stderr);...asse
15a0: 72 74 21 28 73 74 64 65 72 72 2e 63 6f 6e 74 61 rt!(stderr.conta
15b0: 69 6e 73 28 22 73 65 6c 65 63 74 65 64 20 63 6f ins("selected co
15c0: 6e 66 69 67 2e 74 6f 6d 6c 22 29 29 3b 0a 09 09 nfig.toml"));...
15d0: 61 73 73 65 72 74 21 28 73 74 64 65 72 72 2e 63 assert!(stderr.c
15e0: 6f 6e 74 61 69 6e 73 28 22 73 6d 74 70 32 74 67 ontains("smtp2tg
15f0: 2e 74 6f 6d 6c 2e 65 78 61 6d 70 6c 65 22 29 29 .toml.example"))
1600: 3b 0a 09 09 61 73 73 65 72 74 5f 65 72 72 6f 72 ;...assert_error
1610: 28 6f 75 74 70 75 74 2c 20 22 46 61 69 6c 65 64 (output, "Failed
1620: 20 74 6f 20 70 61 72 73 65 20 63 6f 6e 66 69 67 to parse config
1630: 75 72 61 74 69 6f 6e 20 66 69 6c 65 22 29 3b 0a uration file");.
1640: 09 7d 0a 7d 0a 23 5b 74 65 73 74 5d 0a 66 6e 20 .}.}.#[test].fn
1650: 72 65 71 75 69 72 65 64 5f 73 65 74 74 69 6e 67 required_setting
1660: 73 5f 72 65 6d 61 69 6e 5f 72 65 71 75 69 72 65 s_remain_require
1670: 64 5f 61 66 74 65 72 5f 61 70 70 6c 79 69 6e 67 d_after_applying
1680: 5f 64 65 66 61 75 6c 74 73 20 28 29 20 7b 0a 09 _defaults () {..
1690: 66 6f 72 20 28 63 6f 6e 66 69 67 2c 20 65 78 70 for (config, exp
16a0: 65 63 74 65 64 29 20 69 6e 20 5b 0a 09 09 28 22 ected) in [...("
16b0: 64 65 66 61 75 6c 74 20 3d 20 30 5c 6e 5b 72 65 default = 0\n[re
16c0: 63 69 70 69 65 6e 74 73 5d 5c 6e 22 2c 20 22 61 cipients]\n", "a
16d0: 70 69 5f 6b 65 79 22 29 2c 0a 09 09 28 22 61 70 pi_key"),...("ap
16e0: 69 5f 6b 65 79 20 3d 20 27 74 65 73 74 2d 61 70 i_key = 'test-ap
16f0: 69 2d 6b 65 79 27 5c 6e 5b 72 65 63 69 70 69 65 i-key'\n[recipie
1700: 6e 74 73 5d 5c 6e 22 2c 20 22 64 65 66 61 75 6c nts]\n", "defaul
1710: 74 22 29 2c 0a 09 09 28 22 61 70 69 5f 6b 65 79 t"),...("api_key
1720: 20 3d 20 27 74 65 73 74 2d 61 70 69 2d 6b 65 79 = 'test-api-key
1730: 27 5c 6e 64 65 66 61 75 6c 74 20 3d 20 30 5c 6e '\ndefault = 0\n
1740: 22 2c 20 22 72 65 63 69 70 69 65 6e 74 73 22 29 ", "recipients")
1750: 2c 0a 09 5d 20 7b 0a 09 09 6c 65 74 20 64 69 72 ,..] {...let dir
1760: 20 3d 20 54 65 73 74 44 69 72 3a 3a 6e 65 77 28 = TestDir::new(
1770: 29 3b 0a 09 09 64 69 72 2e 63 6f 6e 66 69 67 28 );...dir.config(
1780: 22 73 6d 74 70 32 74 67 2e 74 6f 6d 6c 22 2c 20 "smtp2tg.toml",
1790: 63 6f 6e 66 69 67 2c 20 30 6f 36 30 30 29 3b 0a config, 0o600);.
17a0: 09 09 6c 65 74 20 6f 75 74 70 75 74 20 3d 20 64 ..let output = d
17b0: 69 72 2e 72 75 6e 28 26 5b 5d 29 3b 0a 09 09 61 ir.run(&[]);...a
17c0: 73 73 65 72 74 5f 65 72 72 6f 72 28 6f 75 74 70 ssert_error(outp
17d0: 75 74 2c 20 65 78 70 65 63 74 65 64 29 3b 0a 09 ut, expected);..
17e0: 7d 0a 7d 0a 66 6e 20 73 6d 74 70 5f 72 65 70 6c }.}.fn smtp_repl
17f0: 79 20 28 73 74 72 65 61 6d 3a 20 26 6d 75 74 20 y (stream: &mut
1800: 42 75 66 52 65 61 64 65 72 3c 54 63 70 53 74 72 BufReader<TcpStr
1810: 65 61 6d 3e 29 20 2d 3e 20 53 74 72 69 6e 67 20 eam>) -> String
1820: 7b 0a 09 6c 65 74 20 6d 75 74 20 72 65 73 70 6f {..let mut respo
1830: 6e 73 65 20 3d 20 53 74 72 69 6e 67 3a 3a 6e 65 nse = String::ne
1840: 77 28 29 3b 0a 09 6c 6f 6f 70 20 7b 0a 09 09 6c w();..loop {...l
1850: 65 74 20 6d 75 74 20 6c 69 6e 65 20 3d 20 53 74 et mut line = St
1860: 72 69 6e 67 3a 3a 6e 65 77 28 29 3b 0a 09 09 61 ring::new();...a
1870: 73 73 65 72 74 21 28 73 74 72 65 61 6d 2e 72 65 ssert!(stream.re
1880: 61 64 5f 6c 69 6e 65 28 26 6d 75 74 20 6c 69 6e ad_line(&mut lin
1890: 65 29 2e 75 6e 77 72 61 70 28 29 20 3e 20 30 2c e).unwrap() > 0,
18a0: 20 22 53 4d 54 50 20 63 6f 6e 6e 65 63 74 69 6f "SMTP connectio
18b0: 6e 20 63 6c 6f 73 65 64 3a 20 7b 72 65 73 70 6f n closed: {respo
18c0: 6e 73 65 7d 22 29 3b 0a 09 09 6c 65 74 20 6d 6f nse}");...let mo
18d0: 72 65 20 3d 20 6c 69 6e 65 2e 61 73 5f 62 79 74 re = line.as_byt
18e0: 65 73 28 29 2e 67 65 74 28 33 29 20 3d 3d 20 53 es().get(3) == S
18f0: 6f 6d 65 28 26 62 27 2d 27 29 3b 0a 09 09 72 65 ome(&b'-');...re
1900: 73 70 6f 6e 73 65 2e 70 75 73 68 5f 73 74 72 28 sponse.push_str(
1910: 26 6c 69 6e 65 29 3b 0a 09 09 69 66 20 21 6d 6f &line);...if !mo
1920: 72 65 20 7b 0a 09 09 09 72 65 74 75 72 6e 20 72 re {....return r
1930: 65 73 70 6f 6e 73 65 3b 0a 09 09 7d 0a 09 7d 0a esponse;...}..}.
1940: 7d 0a 23 5b 74 65 73 74 5d 0a 66 6e 20 65 6e 74 }.#[test].fn ent
1950: 72 79 5f 70 6f 69 6e 74 5f 73 65 72 76 65 73 5f ry_point_serves_
1960: 73 6d 74 70 5f 77 69 74 68 5f 64 65 66 61 75 6c smtp_with_defaul
1970: 74 73 5f 61 6e 64 5f 63 6f 6e 66 69 67 75 72 61 ts_and_configura
1980: 74 69 6f 6e 5f 6f 76 65 72 72 69 64 65 73 20 28 tion_overrides (
1990: 29 20 7b 0a 09 66 6f 72 20 28 6f 76 65 72 72 69 ) {..for (overri
19a0: 64 65 73 2c 20 68 6f 73 74 6e 61 6d 65 2c 20 61 des, hostname, a
19b0: 6c 6c 6f 77 65 64 2c 20 64 65 6e 69 65 64 29 20 llowed, denied)
19c0: 69 6e 20 5b 0a 09 09 28 22 22 2c 20 22 73 6d 74 in [...("", "smt
19d0: 70 2e 32 2e 74 67 22 2c 20 22 6c 6f 63 61 6c 68 p.2.tg", "localh
19e0: 6f 73 74 22 2c 20 22 6f 75 74 73 69 64 65 2e 69 ost", "outside.i
19f0: 6e 76 61 6c 69 64 22 29 2c 0a 09 09 28 22 68 6f nvalid"),...("ho
1a00: 73 74 6e 61 6d 65 20 3d 20 27 63 75 73 74 6f 6d stname = 'custom
1a10: 2e 73 6d 74 70 27 5c 6e 64 6f 6d 61 69 6e 73 20 .smtp'\ndomains
1a20: 3d 20 5b 27 65 78 61 6d 70 6c 65 2e 63 6f 6d 27 = ['example.com'
1a30: 5d 5c 6e 66 69 65 6c 64 73 20 3d 20 5b 5d 5c 6e ]\nfields = []\n
1a40: 22 2c 0a 09 09 09 22 63 75 73 74 6f 6d 2e 73 6d ",...."custom.sm
1a50: 74 70 22 2c 20 22 65 78 61 6d 70 6c 65 2e 63 6f tp", "example.co
1a60: 6d 22 2c 20 22 6c 6f 63 61 6c 68 6f 73 74 22 29 m", "localhost")
1a70: 2c 0a 09 5d 20 7b 0a 09 09 6c 65 74 20 64 69 72 ,..] {...let dir
1a80: 20 3d 20 54 65 73 74 44 69 72 3a 3a 6e 65 77 28 = TestDir::new(
1a90: 29 3b 0a 09 09 6c 65 74 20 72 65 73 65 72 76 61 );...let reserva
1aa0: 74 69 6f 6e 20 3d 20 54 63 70 4c 69 73 74 65 6e tion = TcpListen
1ab0: 65 72 3a 3a 62 69 6e 64 28 22 31 32 37 2e 30 2e er::bind("127.0.
1ac0: 30 2e 31 3a 30 22 29 2e 75 6e 77 72 61 70 28 29 0.1:0").unwrap()
1ad0: 3b 0a 09 09 6c 65 74 20 61 64 64 72 65 73 73 20 ;...let address
1ae0: 3d 20 72 65 73 65 72 76 61 74 69 6f 6e 2e 6c 6f = reservation.lo
1af0: 63 61 6c 5f 61 64 64 72 28 29 2e 75 6e 77 72 61 cal_addr().unwra
1b00: 70 28 29 3b 0a 09 09 64 69 72 2e 63 6f 6e 66 69 p();...dir.confi
1b10: 67 28 22 73 6d 74 70 32 74 67 2e 74 6f 6d 6c 22 g("smtp2tg.toml"
1b20: 2c 20 26 66 6f 72 6d 61 74 21 28 0a 09 09 09 22 , &format!(...."
1b30: 61 70 69 5f 6b 65 79 20 3d 20 27 74 65 73 74 2d api_key = 'test-
1b40: 61 70 69 2d 6b 65 79 27 5c 6e 64 65 66 61 75 6c api-key'\ndefaul
1b50: 74 20 3d 20 30 5c 6e 6c 69 73 74 65 6e 5f 6f 6e t = 0\nlisten_on
1b60: 20 3d 20 27 7b 61 64 64 72 65 73 73 7d 27 5c 6e = '{address}'\n
1b70: 7b 6f 76 65 72 72 69 64 65 73 7d 5b 72 65 63 69 {overrides}[reci
1b80: 70 69 65 6e 74 73 5d 5c 6e 22 29 2c 20 30 6f 36 pients]\n"), 0o6
1b90: 30 30 29 3b 0a 09 09 64 72 6f 70 28 72 65 73 65 00);...drop(rese
1ba0: 72 76 61 74 69 6f 6e 29 3b 0a 09 09 6c 65 74 20 rvation);...let
1bb0: 6d 75 74 20 70 72 6f 63 65 73 73 20 3d 20 64 69 mut process = di
1bc0: 72 2e 73 70 61 77 6e 28 26 5b 5d 29 3b 0a 09 09 r.spawn(&[]);...
1bd0: 6c 65 74 20 6d 75 74 20 73 74 72 65 61 6d 20 3d let mut stream =
1be0: 20 42 75 66 52 65 61 64 65 72 3a 3a 6e 65 77 28 BufReader::new(
1bf0: 70 72 6f 63 65 73 73 2e 63 6f 6e 6e 65 63 74 28 process.connect(
1c00: 61 64 64 72 65 73 73 29 29 3b 0a 09 09 6c 65 74 address));...let
1c10: 20 62 61 6e 6e 65 72 20 3d 20 73 6d 74 70 5f 72 banner = smtp_r
1c20: 65 70 6c 79 28 26 6d 75 74 20 73 74 72 65 61 6d eply(&mut stream
1c30: 29 3b 0a 09 09 61 73 73 65 72 74 21 28 62 61 6e );...assert!(ban
1c40: 6e 65 72 2e 73 74 61 72 74 73 5f 77 69 74 68 28 ner.starts_with(
1c50: 22 32 32 30 20 22 29 2c 20 22 7b 62 61 6e 6e 65 "220 "), "{banne
1c60: 72 7d 22 29 3b 0a 09 09 61 73 73 65 72 74 21 28 r}");...assert!(
1c70: 62 61 6e 6e 65 72 2e 63 6f 6e 74 61 69 6e 73 28 banner.contains(
1c80: 68 6f 73 74 6e 61 6d 65 29 2c 20 22 7b 62 61 6e hostname), "{ban
1c90: 6e 65 72 7d 22 29 3b 0a 09 09 66 6f 72 20 28 63 ner}");...for (c
1ca0: 6f 6d 6d 61 6e 64 2c 20 63 6f 64 65 29 20 69 6e ommand, code) in
1cb0: 20 5b 0a 09 09 09 28 22 45 48 4c 4f 20 74 65 73 [....("EHLO tes
1cc0: 74 2e 6c 6f 63 61 6c 5c 72 5c 6e 22 2e 69 6e 74 t.local\r\n".int
1cd0: 6f 28 29 2c 20 22 32 35 30 22 29 2c 0a 09 09 09 o(), "250"),....
1ce0: 28 22 4d 41 49 4c 20 46 52 4f 4d 3a 3c 73 65 6e ("MAIL FROM:<sen
1cf0: 64 65 72 40 65 78 61 6d 70 6c 65 2e 6f 72 67 3e der@example.org>
1d00: 5c 72 5c 6e 22 2e 69 6e 74 6f 28 29 2c 20 22 32 \r\n".into(), "2
1d10: 35 30 22 29 2c 0a 09 09 09 28 66 6f 72 6d 61 74 50"),....(format
1d20: 21 28 22 52 43 50 54 20 54 4f 3a 3c 75 73 65 72 !("RCPT TO:<user
1d30: 40 7b 61 6c 6c 6f 77 65 64 7d 3e 5c 72 5c 6e 22 @{allowed}>\r\n"
1d40: 29 2c 20 22 32 35 30 22 29 2c 0a 09 09 09 28 66 ), "250"),....(f
1d50: 6f 72 6d 61 74 21 28 22 52 43 50 54 20 54 4f 3a ormat!("RCPT TO:
1d60: 3c 75 73 65 72 40 7b 64 65 6e 69 65 64 7d 3e 5c <user@{denied}>\
1d70: 72 5c 6e 22 29 2c 20 22 35 35 30 22 29 2c 0a 09 r\n"), "550"),..
1d80: 09 09 28 22 51 55 49 54 5c 72 5c 6e 22 2e 69 6e ..("QUIT\r\n".in
1d90: 74 6f 28 29 2c 20 22 32 32 31 22 29 2c 0a 09 09 to(), "221"),...
1da0: 5d 20 7b 0a 09 09 09 73 74 72 65 61 6d 2e 67 65 ] {....stream.ge
1db0: 74 5f 6d 75 74 28 29 2e 77 72 69 74 65 5f 61 6c t_mut().write_al
1dc0: 6c 28 63 6f 6d 6d 61 6e 64 2e 61 73 5f 62 79 74 l(command.as_byt
1dd0: 65 73 28 29 29 2e 75 6e 77 72 61 70 28 29 3b 0a es()).unwrap();.
1de0: 09 09 09 6c 65 74 20 72 65 70 6c 79 20 3d 20 73 ...let reply = s
1df0: 6d 74 70 5f 72 65 70 6c 79 28 26 6d 75 74 20 73 mtp_reply(&mut s
1e00: 74 72 65 61 6d 29 3b 0a 09 09 09 61 73 73 65 72 tream);....asser
1e10: 74 21 28 72 65 70 6c 79 2e 73 74 61 72 74 73 5f t!(reply.starts_
1e20: 77 69 74 68 28 63 6f 64 65 29 2c 20 22 63 6f 6d with(code), "com
1e30: 6d 61 6e 64 20 7b 63 6f 6d 6d 61 6e 64 3a 3f 7d mand {command:?}
1e40: 3a 20 7b 72 65 70 6c 79 7d 22 29 3b 0a 09 09 7d : {reply}");...}
1e50: 0a 09 7d 0a 7d 0a ..}.}.