Différences

Ci-dessous, les différences entre deux révisions de la page.

--- ateliers:serveurmail:rspamd [2018/01/28 13:30] – Plus de détail sur l'installation okhin
+++ ateliers:serveurmail:rspamd [2018/01/28 13:52] – [Configuration] Update the script to python version okhin
@@ Ligne 31: / Ligne 31: @@
-==== Configuration ====
+#!/usr/bin/env python3
+# Python 3.5+ required
+# Script by https://sitedethib.com/
+# Source adapted to 3.5 from dashie version https://dev.sigpipe.me/dashie/setup_mail/src/master/rspamd.py
-FIXME: A écrire
+import sys
+import smtplib
+import subprocess
+import syslog
+import time
-Script qui permet de faire le pont vers du lmtp. On émule en environnement mail lmtp.
+t_start = time.time()
-Actuellement fonctionnel mais probablement amené à être réécris en GOland
+SMTP_HOST   = '127.0.0.1'
+SMTP_PORT   = 10030
+RSPAMC_BIN  = '/usr/bin/rspamc'
+RSPAMC_ADDR = '127.0.0.1:11333'
+mail_from     = sys.argv[1]
+rcpt_to       = sys.argv[2]
-<code|h /usr/local/bin/rspamd.sh>
+def size(size, decimal_places):
-#!/bin/sh
+    for unit in ['','KB','MB','GB','TB']:
+        if size < 1024.0:
+            break
+        size /= 1024.0
+    return "{size:.{decimal_places}f}{unit}".format({size: size, decimal_places: decimal_places, unir: unit})
-mail_file=$(/bin/mktemp)
+syslog.openlog(logoption=syslog.LOG_PID, facility=syslog.LOG_MAIL)
-/bin/echo 'HELO localhost' >> $mail_file
+# Pipe stdin to rspamc, and get the processed mail back
-/bin/echo "MAIL FROM: <$1>" >> $mail_file
+#XXX: this means the processed mail stays in RAM, which may not be what we want?
-/bin/echo "RCPT TO: <$2>" >> $mail_file
+completed_process = subprocess.run([RSPAMC_BIN, '--mime', '-h', RSPAMC_ADDR, '-P', 'saucisse'],
-/bin/echo 'DATA' >> $mail_file
+                                   stdin=sys.stdin, stdout=subprocess.PIPE,
+                                   check=True)
-/usr/bin/rspamc --mime -h 127.0.0.1:11333 -P q1 >> $mail_file
+# Remove 'Delivered-To' header from rspamc's output
+mail_contents = completed_process.stdout
+mail_contents = b'\n'.join(l for l in mail_contents.split(b'\n') if not l.startswith(b'Delivered-To'))
-/bin/echo '' >> $mail_file
+with smtplib.SMTP(SMTP_HOST, SMTP_PORT) as smtp:
-/bin/echo '.' >> $mail_file
+    smtp.sendmail(mail_from, rcpt_to, mail_contents)
-/bin/echo 'QUIT' >> $mail_file
-cut_file=$(/bin/mktemp)
+t_end = time.time()
-/bin/sed '/Delivered-To/d' $mail_file > $cut_file
-/bin/rm "$mail_file"
-count=0; IFS=''; while read -r line ; do if [ "$count" -gt "5" ]; then sleep 0.05; else sleep 0.1; fi; echo "$line"; count=$((count+1)); done < "$cut_file" | /bin/netcat 127.0.0.1 10025
+syslog.syslog("rspamd-bridge processed {} mail in {}".format(size(len(mail_contents), 1), t_end - t_start))
-/bin/rm "$cut_file"
-</code>