rssViedos mostly

config/scripts/smtpdummy

@@ -1,43 +1,74 @@
 #!/usr/bin/env python3
 
-import argparse
+import base64
+import colorama
+import configargparse
 import datetime
-import time
 import email.utils
-import subprocess
+import io
 import pprint
-
-
-def command(command: str) -> None:
-    cmd = command.encode() + b"\n"
-    subprocess.run(["xdotool", "type", "--file", "-"], input=cmd)
-    time.sleep(2)
+import subprocess
+import sys
 
 
 if __name__ == "__main__":
-    parser = argparse.ArgumentParser(
+    parser = configargparse.ArgParser(
         description="Generate SMTP messages to send a mail"
     )
 
     now = datetime.datetime.now()
     now_email = email.utils.formatdate(now.timestamp(), True)
 
-    parser.add_argument("-s", "--sender", default="geoffrey@frogeye.fr")
-    parser.add_argument("-r", "--receiver", default="geoffrey@frogeye.fr")
-    parser.add_argument("-l", "--helo", default="frogeye.fr")
+    parser.add_argument("-o", "--origin", env_var="ORIGIN", default="localhost")
     parser.add_argument(
-        "-o", "--subject", default=f"Test message {now.strftime('%H:%M:%S')}"
+        "-d", "--destination", env_var="DESTINATION", default="localhost"
+    )
+    parser.add_argument("-p", "--port", env_var="PORT", default=25)
+    parser.add_argument(
+        "-S", "--security", env_var="SECURITY", choices=["plain", "ssl", "starttls"], default='plain'
     )
-    parser.add_argument("-m", "--me", default="Geoffrey")
-    parser.add_argument("-g", "--gtube", action="store_true")
-    parser.add_argument("-d", "--debug", action="store_true")
-    parser.add_argument("-b", "--body", default="")
 
+    parser.add_argument("-l", "--helo", env_var="HELO")
+    parser.add_argument(
+        "-s", "--sender", env_var="SENDER", default="geoffrey@frogeye.fr"
+    )
+    parser.add_argument(
+        "-r", "--receiver", env_var="RECEIVER", default="geoffrey@frogeye.fr"
+    )
+    # parser.add_argument("-a", "--auth", env_var="AUTH", default="PLAIN")
+    parser.add_argument("-u", "--user", env_var="MUSER")
+    parser.add_argument("-w", "--password", env_var="PASSWORD")
+
+    parser.add_argument("-f", "--from", env_var="FROM")
+    parser.add_argument("-t", "--to", env_var="TO")
+    parser.add_argument(
+        "-j",
+        "--subject",
+        env_var="SUBJECT",
+        default=f"Test message {now.strftime('%H:%M:%S')}",
+    )
+
+    parser.add_argument("-b", "--body", env_var="BODY", default="")
+    parser.add_argument("-g", "--gtube", env_var="GTUBE", action="store_true")
+    parser.add_argument("-m", "--me", env_var="ME", default="Geoffrey")
+
+    parser.add_argument("-y", "--dryrun", env_var="DRYRUN", action="store_true")
+    parser.add_argument("-q", "--quiet", env_var="QUIET", action="store_true")
 
     args = parser.parse_args()
 
-    if args.debug:
-        command = print
+    # Default values
+    if args.helo is None:
+        args.helo = args.origin
+    if getattr(args, "from") is None:
+        setattr(args, "from", args.sender)
+    if args.to is None:
+        args.to = args.receiver
+    if args.password:
+        password = args.password
+        args.password = '********'
+
+    # Transmission content
 
     gtube = ""
     if args.gtube:
@@ -49,11 +80,10 @@ XJS*C4JDBQADN1.NSBN3*2IDNEN*GTUBE-STANDARD-ANTI-UBE-TEST-EMAIL*C.34X"""
     if args.body:
         body = f"\n\n{args.body}"
 
-
     text = f"""Date: {now_email}
-From: {args.sender}
+From: {getattr(args, 'from')}
 Subject: {args.subject}
-To: {args.receiver}
+To: {args.to}
 
 Hello there,
 
@@ -63,28 +93,96 @@ If you didn't expect to see this message, please contact {args.me}.{gtube}{body}
 Greetings,
 
 Input arguments:
-{pprint.pformat(args, indent=4)}
+{pprint.pformat(args.__dict__, indent=4)}
 
 -- 
 {args.me}
 ."""
 
-    if not args.debug:
-        for i in range(3, 0, -1):
-            print(f"Typing mail in {i}…")
-            time.sleep(1)
+    # Transmission setup
+    cmd = ["ssh", args.origin]
+    if args.security == "plain":
+        cmd += ["socat", "-", f"tcp:{args.destination}:{args.port}"]
+    elif args.security == "ssl":
+        cmd += ["socat", "-", f"openssl:{args.destination}:{args.port}"]
+    elif args.security == "starttls":
+        cmd += [
+            "openssl",
+            "s_client",
+            "-starttls",
+            "smtp",
+            "-crlf",
+            "-connect",
+            f"{args.destination}:{args.port}",
+            "-quiet",
+        ]
 
-    command(f"EHLO {args.helo}")
-    command(f"MAIL FROM: <{args.sender}>")
-    command(f"RCPT TO: <{args.receiver}>")
-    command("DATA")
-    command(text)
-    command("QUIT")
+    if not args.quiet:
+        print(colorama.Fore.MAGENTA + f"# {' '.join(cmd)}" + colorama.Fore.RESET)
 
-    print("Done")
+    if not args.dryrun:
+        p = subprocess.Popen(
+            cmd,
+            stdin=subprocess.PIPE,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.DEVNULL,
+        )
+
+    def recv() -> None:
+        if args.dryrun:
+            return
+
+        assert isinstance(p.stdout, io.BufferedReader)
+        next = True
+        while next:
+            line = p.stdout.readline()
+            code = int(line[:3])
+            success = code < 400
+            color = colorama.Fore.GREEN if success else colorama.Fore.RED
+            if not args.quiet:
+                print(color + f"< {line[:-1].decode()}" + colorama.Fore.RESET)
+            next = line[3] == b"-"[0]
+            if not next and not success:
+                send("QUIT")  # TODO Can loop if QUIT fails
+                sys.exit(1)
+
+    def send(command: str) -> None:
+        if not args.quiet:
+            print(colorama.Fore.BLUE + f"> {command}" + colorama.Fore.RESET)
+
+        if args.dryrun:
+            return
+
+        assert isinstance(p.stdin, io.BufferedWriter)
+        cmd = command.encode() + b"\n"
+        p.stdin.write(cmd)
+        p.stdin.flush()
+
+        recv()
+
+    # Transmission
+
+    if args.security != 'starttls':
+        recv()
+    send(f"EHLO {args.helo}")
+    if args.user:
+        encoded = base64.b64encode(
+            args.user.encode()
+            + b"\x00"
+            + args.user.encode()
+            + b"\x00"
+            + password.encode()
+        ).decode()
+        send(f"AUTH PLAIN {encoded}")
+    send(f"MAIL FROM: <{args.sender}>")
+    send(f"RCPT TO: <{args.receiver}>")
+    send("DATA")
+    send(text)
+    send("QUIT")
+    sys.exit(0)
 
     # For reference:
-    # command("RSET")
-    # command("VRFY")
-    # command("NOOP")
-    # command("QUIT")
+    # send("RSET")
+    # send("VRFY")
+    # send("NOOP")
+    # send("QUIT")