Add SRS implementation - scripts and configuration files.
This commit is contained in:
Executable
+258
@@ -0,0 +1,258 @@
|
||||
#!/usr/bin/python3
|
||||
|
||||
import subprocess
|
||||
from collections import deque
|
||||
import sys
|
||||
|
||||
ALIAS_MAPS = [
|
||||
"pgsql:/etc/postfix/pgsql_virtual_alias_maps.cf",
|
||||
"pgsql:/etc/postfix/pgsql_virtual_alias_domain_maps.cf",
|
||||
"pgsql:/etc/postfix/pgsql_virtual_alias_address_rewriting.cf",
|
||||
"pgsql:/etc/postfix/pgsql_virtual_alias_domain_catchall_maps.cf",
|
||||
"btree:/etc/postfix/virtual_alias_maps",
|
||||
]
|
||||
|
||||
MAILBOX_MAPS = [
|
||||
"pgsql:/etc/postfix/pgsql_virtual_mailbox_maps.cf",
|
||||
"pgsql:/etc/postfix/pgsql_virtual_alias_domain_mailbox_maps.cf",
|
||||
]
|
||||
|
||||
DOMAIN_MAPS = [
|
||||
"pgsql:/etc/postfix/pgsql_virtual_domains_maps.cf",
|
||||
"pgsql:/etc/postfix/pgsql_relay-domain-maps.cf",
|
||||
"btree:/etc/postfix/virtual_alias_domains",
|
||||
]
|
||||
|
||||
TRANSPORT_MAPS = [
|
||||
"pgsql:/etc/postfix/pgsql_transport_maps.cf",
|
||||
"btree:/etc/postfix/transport",
|
||||
"btree:/etc/postfix/relay_domains",
|
||||
]
|
||||
|
||||
|
||||
def lookup(key, table):
|
||||
result = subprocess.run(
|
||||
["/usr/sbin/postmap", "-q", key, table],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=10,
|
||||
)
|
||||
|
||||
if result.returncode == 0:
|
||||
return result.stdout.strip() or None
|
||||
|
||||
if result.returncode == 1 and not result.stderr.strip():
|
||||
return None
|
||||
|
||||
raise RuntimeError(
|
||||
f"{table}: {result.stderr.strip() or 'Lookup fehlgeschlagen'}"
|
||||
)
|
||||
|
||||
|
||||
def lookup_any(key, tables):
|
||||
for table in tables:
|
||||
value = lookup(key, table)
|
||||
if value is not None:
|
||||
return value
|
||||
return None
|
||||
|
||||
|
||||
def postconf(name):
|
||||
result = subprocess.run(
|
||||
["/usr/sbin/postconf", "-c", "/etc/postfix", "-x", "-h", name],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=10,
|
||||
)
|
||||
|
||||
if result.returncode:
|
||||
raise RuntimeError(result.stderr.strip())
|
||||
|
||||
return result.stdout.strip()
|
||||
|
||||
|
||||
def transport_keys(address):
|
||||
localpart, domain = address.rsplit("@", 1)
|
||||
|
||||
keys = [address, f"{localpart}@"]
|
||||
|
||||
parts = domain.split(".")
|
||||
|
||||
for i in range(len(parts)):
|
||||
suffix = ".".join(parts[i:])
|
||||
keys.extend([suffix, "." + suffix])
|
||||
|
||||
return list(dict.fromkeys(keys))
|
||||
|
||||
|
||||
def classify_destination(address):
|
||||
domain = address.rsplit("@", 1)[1]
|
||||
|
||||
if lookup_any(address, MAILBOX_MAPS):
|
||||
return "LOCAL"
|
||||
|
||||
if domain in postconf("mydestination").lower().split():
|
||||
return "LOCAL"
|
||||
|
||||
if lookup_any(domain, DOMAIN_MAPS):
|
||||
return "UNKNOWN"
|
||||
|
||||
for key in transport_keys(address):
|
||||
transport = lookup_any(key, TRANSPORT_MAPS)
|
||||
|
||||
if transport:
|
||||
transport_type = transport.split(":", 1)[0].lower()
|
||||
|
||||
if transport_type in ("smtp", "srs-forward"):
|
||||
return "EXTERNAL"
|
||||
|
||||
return "UNKNOWN"
|
||||
|
||||
default_transport = postconf("default_transport").strip().lower()
|
||||
if (
|
||||
default_transport in ("smtp", "srs-forward")
|
||||
or default_transport.startswith(("smtp:", "srs-forward:"))
|
||||
):
|
||||
return "EXTERNAL"
|
||||
|
||||
return "UNKNOWN"
|
||||
|
||||
|
||||
MAX_ALIAS_DEPTH = 10
|
||||
MAX_ALIAS_NODES = 200
|
||||
|
||||
|
||||
def alias_reaches(original, recipient):
|
||||
"""Find alias paths with BFS and detect cycles in the explored graph."""
|
||||
pending = deque([(original, 0)])
|
||||
visited = {original}
|
||||
edges = {}
|
||||
nodes = 0
|
||||
depth_exceeded = False
|
||||
|
||||
while pending:
|
||||
current, depth = pending.popleft()
|
||||
nodes += 1
|
||||
|
||||
if nodes > MAX_ALIAS_NODES:
|
||||
return None, "Zu viele Alias-Knoten"
|
||||
|
||||
aliases = lookup_any(current, ALIAS_MAPS)
|
||||
if not aliases:
|
||||
edges[current] = []
|
||||
continue
|
||||
|
||||
targets = [
|
||||
x.strip().lower()
|
||||
for x in aliases.split(",")
|
||||
if (
|
||||
x.strip()
|
||||
and x.strip().count("@") == 1
|
||||
and x.strip().lower() != current
|
||||
)
|
||||
]
|
||||
|
||||
edges[current] = targets
|
||||
|
||||
for target in targets:
|
||||
if target == recipient:
|
||||
if depth + 1 <= MAX_ALIAS_DEPTH:
|
||||
return True, "Alias-Kette erkannt"
|
||||
depth_exceeded = True
|
||||
continue
|
||||
|
||||
if target in visited:
|
||||
continue
|
||||
|
||||
if depth + 1 >= MAX_ALIAS_DEPTH:
|
||||
depth_exceeded = True
|
||||
continue
|
||||
|
||||
visited.add(target)
|
||||
pending.append((target, depth + 1))
|
||||
|
||||
if depth_exceeded:
|
||||
return None, "Maximale Alias-Tiefe erreicht"
|
||||
|
||||
# Detect cycles in the explored directed graph.
|
||||
# 0 = unseen, 1 = active, 2 = completed
|
||||
state = {}
|
||||
|
||||
def has_cycle(node):
|
||||
state[node] = 1
|
||||
|
||||
for target in edges.get(node, []):
|
||||
if target not in edges:
|
||||
continue
|
||||
|
||||
status = state.get(target, 0)
|
||||
|
||||
if status == 1:
|
||||
return True
|
||||
|
||||
if status == 0 and has_cycle(target):
|
||||
return True
|
||||
|
||||
state[node] = 2
|
||||
return False
|
||||
|
||||
for node in edges:
|
||||
if state.get(node, 0) == 0 and has_cycle(node):
|
||||
return None, "Alias-Schleife erkannt"
|
||||
|
||||
return False, "Empfaenger nicht in direkter Alias-Kette"
|
||||
|
||||
|
||||
def decide(sender, recipient, original):
|
||||
sender = sender.strip().lower()
|
||||
recipient = recipient.strip().lower()
|
||||
original = original.strip().lower()
|
||||
|
||||
if not sender or sender == "<>":
|
||||
return "NO_SRS", "Null-Absender"
|
||||
|
||||
if sender.count("@") != 1 or recipient.count("@") != 1:
|
||||
return "UNKNOWN", "Ungueltige Empfaengeradresse"
|
||||
|
||||
if original.count("@") == 0:
|
||||
return "NO_SRS", "Lokaler Alias"
|
||||
|
||||
if original.count("@") != 1:
|
||||
return "UNKNOWN", "Ungueltige Empfaengeradresse"
|
||||
|
||||
if recipient == original:
|
||||
return "NO_SRS", "Keine Empfaengerumschreibung"
|
||||
|
||||
reaches, reason = alias_reaches(original, recipient)
|
||||
if reaches is not True:
|
||||
return "UNKNOWN", reason
|
||||
|
||||
destination = classify_destination(recipient)
|
||||
if destination == "LOCAL":
|
||||
return "NO_SRS", "Lokales Postfach"
|
||||
if destination == "EXTERNAL":
|
||||
return "SRS", "PostfixAdmin-Alias-Kette mit SMTP-Zustellung"
|
||||
return "UNKNOWN", "Zustellweg nicht eindeutig"
|
||||
|
||||
|
||||
def main():
|
||||
if len(sys.argv) != 4:
|
||||
print(
|
||||
"Aufruf: srs-alias-decision.py "
|
||||
"<sender> <recipient> <original_recipient>",
|
||||
file=sys.stderr,
|
||||
)
|
||||
return 2
|
||||
|
||||
try:
|
||||
decision, reason = decide(*sys.argv[1:])
|
||||
except (RuntimeError, subprocess.TimeoutExpired) as exc:
|
||||
print(f"UNKNOWN: Lookup-Fehler: {exc}")
|
||||
return 1
|
||||
|
||||
print(f"{decision}: {reason}")
|
||||
return 1 if decision == "UNKNOWN" else 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(main())
|
||||
Executable
+177
@@ -0,0 +1,177 @@
|
||||
#!/usr/bin/python3
|
||||
|
||||
import smtplib
|
||||
import subprocess
|
||||
import sys
|
||||
|
||||
SMTP_HOST = "127.0.0.1"
|
||||
SMTP_PORT = 10028
|
||||
|
||||
DECISION_SCRIPT = "/usr/local/sbin/srs-alias-decision.py"
|
||||
SRS_MAP = "socketmap:inet:127.0.0.1:10003:forward"
|
||||
|
||||
EX_TEMPFAIL = 75
|
||||
EX_UNAVAILABLE = 69
|
||||
|
||||
|
||||
def run_command(args):
|
||||
result = subprocess.run(
|
||||
args,
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=30
|
||||
)
|
||||
|
||||
if result.returncode != 0:
|
||||
raise RuntimeError(
|
||||
result.stderr.strip()
|
||||
or result.stdout.strip()
|
||||
or f"Exit-Code {result.returncode}"
|
||||
)
|
||||
|
||||
return result.stdout.strip()
|
||||
|
||||
|
||||
def determine_sender(sender, recipient, original_recipient):
|
||||
decision = run_command([
|
||||
DECISION_SCRIPT,
|
||||
sender,
|
||||
recipient,
|
||||
original_recipient
|
||||
])
|
||||
|
||||
if decision.startswith("NO_SRS:"):
|
||||
return sender
|
||||
|
||||
if not decision.startswith("SRS:"):
|
||||
raise RuntimeError(
|
||||
f"Keine eindeutige SRS-Entscheidung: {decision}"
|
||||
)
|
||||
|
||||
rewritten_sender = run_command([
|
||||
"/usr/sbin/postmap",
|
||||
"-q",
|
||||
sender,
|
||||
SRS_MAP
|
||||
])
|
||||
|
||||
if not rewritten_sender:
|
||||
raise RuntimeError("PostSRSd lieferte keinen Absender")
|
||||
|
||||
if not rewritten_sender.lower().endswith("@srs.warenform.de"):
|
||||
raise RuntimeError(
|
||||
f"Unerwartete SRS-Domain: {rewritten_sender}"
|
||||
)
|
||||
|
||||
if not rewritten_sender.upper().startswith(("SRS0=", "SRS1=")):
|
||||
raise RuntimeError(
|
||||
f"Ungueltige SRS-Adresse: {rewritten_sender}"
|
||||
)
|
||||
|
||||
return rewritten_sender
|
||||
|
||||
|
||||
def main():
|
||||
if len(sys.argv) not in (5, 6):
|
||||
print(
|
||||
"Aufruf: srs-forward.py "
|
||||
"sender recipient original_recipient queue_id [envid]",
|
||||
file=sys.stderr
|
||||
)
|
||||
return EX_TEMPFAIL
|
||||
|
||||
sender, recipient, original_recipient, queue_id = sys.argv[1:5]
|
||||
envid = sys.argv[5] if len(sys.argv) == 6 else ""
|
||||
|
||||
if envid.startswith("envid="):
|
||||
envid = envid[len("envid="):]
|
||||
|
||||
try:
|
||||
envelope_sender = determine_sender(
|
||||
sender,
|
||||
recipient,
|
||||
original_recipient
|
||||
)
|
||||
|
||||
message = sys.stdin.buffer.read()
|
||||
|
||||
with smtplib.SMTP(
|
||||
SMTP_HOST,
|
||||
SMTP_PORT,
|
||||
timeout=30
|
||||
) as smtp:
|
||||
smtp.ehlo()
|
||||
|
||||
mail_options = []
|
||||
rcpt_options = []
|
||||
|
||||
if envid:
|
||||
if not smtp.has_extn("dsn"):
|
||||
raise RuntimeError(
|
||||
"SMTP-Ziel unterstuetzt kein DSN/ENVID"
|
||||
)
|
||||
mail_options.append(f"ENVID={envid}")
|
||||
|
||||
if original_recipient and smtp.has_extn("dsn"):
|
||||
rcpt_options.append(
|
||||
f"ORCPT=rfc822;{original_recipient}"
|
||||
)
|
||||
|
||||
smtp.sendmail(
|
||||
envelope_sender,
|
||||
[recipient],
|
||||
message,
|
||||
mail_options=mail_options,
|
||||
rcpt_options=rcpt_options
|
||||
)
|
||||
|
||||
except smtplib.SMTPRecipientsRefused as exc:
|
||||
codes = [
|
||||
code
|
||||
for code, response in exc.recipients.values()
|
||||
]
|
||||
|
||||
permanent = bool(codes) and all(
|
||||
500 <= code < 600 for code in codes
|
||||
)
|
||||
|
||||
result = EX_UNAVAILABLE if permanent else EX_TEMPFAIL
|
||||
enhanced_status = "5.0.0" if permanent else "4.0.0"
|
||||
|
||||
print(
|
||||
f"{enhanced_status} SRS queue_id={queue_id}: "
|
||||
f"SMTP-Empfaengerablehnung: {exc.recipients!r}",
|
||||
file=sys.stderr
|
||||
)
|
||||
return result
|
||||
|
||||
except smtplib.SMTPResponseException as exc:
|
||||
code = exc.smtp_code
|
||||
result = EX_UNAVAILABLE if 500 <= code < 600 else EX_TEMPFAIL
|
||||
|
||||
enhanced_status = "5.0.0" if result == EX_UNAVAILABLE else "4.0.0"
|
||||
|
||||
print(
|
||||
f"{enhanced_status} SRS queue_id={queue_id}: "
|
||||
f"SMTP {code}: {exc.smtp_error!r}",
|
||||
file=sys.stderr
|
||||
)
|
||||
return result
|
||||
|
||||
except (
|
||||
subprocess.TimeoutExpired,
|
||||
RuntimeError,
|
||||
smtplib.SMTPException,
|
||||
OSError
|
||||
) as exc:
|
||||
print(
|
||||
f"SRS queue_id={queue_id}: temporaerer Fehler: {exc}",
|
||||
file=sys.stderr
|
||||
)
|
||||
return EX_TEMPFAIL
|
||||
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(main())
|
||||
Executable
+112
@@ -0,0 +1,112 @@
|
||||
#!/usr/bin/python3
|
||||
|
||||
import socket
|
||||
import socketserver
|
||||
|
||||
LISTEN_HOST = "127.0.0.1"
|
||||
LISTEN_PORT = 10004
|
||||
SRS_HOST = "127.0.0.1"
|
||||
SRS_PORT = 10003
|
||||
SRS_DOMAIN = "srs.warenform.de"
|
||||
|
||||
MAX_REQUEST = 8192
|
||||
TIMEOUT = 3
|
||||
|
||||
|
||||
def read_netstring(sock):
|
||||
length = bytearray()
|
||||
|
||||
while True:
|
||||
c = sock.recv(1)
|
||||
if not c:
|
||||
raise EOFError()
|
||||
if c == b":":
|
||||
break
|
||||
if not c.isdigit() or len(length) >= 8:
|
||||
raise ValueError("Invalid netstring length")
|
||||
length.extend(c)
|
||||
|
||||
if not length:
|
||||
raise ValueError("Empty netstring length")
|
||||
|
||||
size = int(length)
|
||||
if size > MAX_REQUEST:
|
||||
raise ValueError("Netstring too large")
|
||||
|
||||
data = bytearray()
|
||||
while len(data) < size:
|
||||
chunk = sock.recv(size - len(data))
|
||||
if not chunk:
|
||||
raise EOFError()
|
||||
data.extend(chunk)
|
||||
|
||||
if sock.recv(1) != b",":
|
||||
raise ValueError("Invalid netstring terminator")
|
||||
|
||||
return bytes(data)
|
||||
|
||||
|
||||
def netstring(data):
|
||||
return str(len(data)).encode() + b":" + data + b","
|
||||
|
||||
|
||||
class Handler(socketserver.BaseRequestHandler):
|
||||
def handle(self):
|
||||
self.request.settimeout(TIMEOUT)
|
||||
|
||||
while True:
|
||||
try:
|
||||
request = read_netstring(self.request)
|
||||
except (EOFError, OSError):
|
||||
return
|
||||
except ValueError:
|
||||
self.request.sendall(netstring(b"PERM Invalid request"))
|
||||
return
|
||||
|
||||
try:
|
||||
operation, address = request.decode("utf-8").split(" ", 1)
|
||||
except (UnicodeError, ValueError):
|
||||
self.request.sendall(netstring(b"PERM Invalid request"))
|
||||
continue
|
||||
|
||||
if operation != "reverse":
|
||||
self.request.sendall(netstring(b"NOTFOUND "))
|
||||
continue
|
||||
|
||||
if "@" not in address:
|
||||
self.request.sendall(netstring(b"NOTFOUND "))
|
||||
continue
|
||||
|
||||
domain = address.rsplit("@", 1)[1].lower()
|
||||
|
||||
if domain != SRS_DOMAIN:
|
||||
self.request.sendall(netstring(b"NOTFOUND "))
|
||||
continue
|
||||
|
||||
try:
|
||||
with socket.create_connection(
|
||||
(SRS_HOST, SRS_PORT), timeout=TIMEOUT
|
||||
) as upstream:
|
||||
upstream.settimeout(TIMEOUT)
|
||||
upstream.sendall(netstring(request))
|
||||
response = read_netstring(upstream)
|
||||
|
||||
if response.startswith((b"OK ", b"NOTFOUND ", b"TEMP ", b"PERM ")):
|
||||
result = response
|
||||
else:
|
||||
result = b"TEMP Invalid upstream response"
|
||||
|
||||
except (OSError, EOFError, ValueError):
|
||||
result = b"TEMP SRS backend unavailable"
|
||||
|
||||
self.request.sendall(netstring(result))
|
||||
|
||||
|
||||
class Server(socketserver.ThreadingTCPServer):
|
||||
allow_reuse_address = True
|
||||
daemon_threads = True
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
with Server((LISTEN_HOST, LISTEN_PORT), Handler) as server:
|
||||
server.serve_forever()
|
||||
Executable
+113
@@ -0,0 +1,113 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
import socket
|
||||
import sys
|
||||
|
||||
SRS_DOMAIN = "srs.warenform.de"
|
||||
SOCKET_HOST = "127.0.0.1"
|
||||
SOCKET_PORT = 10003
|
||||
TIMEOUT = 5
|
||||
MAX_RESPONSE = 4096
|
||||
|
||||
|
||||
def read_netstring(sock):
|
||||
length_bytes = bytearray()
|
||||
|
||||
while True:
|
||||
char = sock.recv(1)
|
||||
if not char:
|
||||
raise ValueError("Incomplete netstring length")
|
||||
if char == b":":
|
||||
break
|
||||
if char < b"0" or char > b"9" or len(length_bytes) >= 8:
|
||||
raise ValueError("Invalid netstring length")
|
||||
length_bytes.extend(char)
|
||||
|
||||
if not length_bytes:
|
||||
raise ValueError("Empty netstring length")
|
||||
|
||||
length = int(length_bytes)
|
||||
|
||||
if length > MAX_RESPONSE:
|
||||
raise ValueError("Netstring response too large")
|
||||
|
||||
data = bytearray()
|
||||
|
||||
while len(data) < length + 1:
|
||||
chunk = sock.recv(length + 1 - len(data))
|
||||
if not chunk:
|
||||
raise ValueError("Incomplete netstring response")
|
||||
data.extend(chunk)
|
||||
|
||||
if data[-1:] != b",":
|
||||
raise ValueError("Invalid netstring terminator")
|
||||
|
||||
return bytes(data[:-1]).decode("utf-8")
|
||||
|
||||
|
||||
def verify_srs(address):
|
||||
request = f"reverse {address}".encode("utf-8")
|
||||
packet = str(len(request)).encode("ascii") + b":" + request + b","
|
||||
|
||||
with socket.create_connection(
|
||||
(SOCKET_HOST, SOCKET_PORT), timeout=TIMEOUT
|
||||
) as sock:
|
||||
sock.settimeout(TIMEOUT)
|
||||
sock.sendall(packet)
|
||||
response = read_netstring(sock)
|
||||
|
||||
if response.startswith("OK "):
|
||||
original = response[3:]
|
||||
if not original or "@" not in original:
|
||||
raise ValueError("Invalid reverse result")
|
||||
return "OK"
|
||||
|
||||
if response.startswith("NOTFOUND "):
|
||||
return "REJECT 550 5.1.1 Invalid SRS address"
|
||||
|
||||
raise ValueError(f"Unexpected socketmap response: {response}")
|
||||
|
||||
|
||||
def check_recipient(recipient):
|
||||
if "@" not in recipient:
|
||||
return "DUNNO"
|
||||
|
||||
localpart, domain = recipient.rsplit("@", 1)
|
||||
|
||||
if domain.lower() != SRS_DOMAIN:
|
||||
return "DUNNO"
|
||||
|
||||
if not localpart.upper().startswith(("SRS0=", "SRS1=")):
|
||||
return "REJECT 550 5.1.1 Invalid SRS address"
|
||||
|
||||
try:
|
||||
return verify_srs(recipient)
|
||||
except (OSError, UnicodeError, ValueError) as exc:
|
||||
print(f"SRS verification error: {exc}", file=sys.stderr)
|
||||
return "DEFER 451 4.3.0 SRS verification unavailable"
|
||||
|
||||
|
||||
def main():
|
||||
while True:
|
||||
attributes = {}
|
||||
|
||||
for line in sys.stdin:
|
||||
line = line.rstrip("\r\n")
|
||||
|
||||
if not line:
|
||||
break
|
||||
|
||||
if "=" in line:
|
||||
key, value = line.split("=", 1)
|
||||
attributes[key] = value
|
||||
else:
|
||||
return
|
||||
|
||||
recipient = attributes.get("recipient", "")
|
||||
action = check_recipient(recipient)
|
||||
|
||||
print(f"action={action}\n", flush=True)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user