def _process_message(self, peer: Tuple[str, int], mailfrom: str, rcpttos: Sequence[str], data: bytes, *,
channel: PatchedSMTPChannel,
**kwargs: Any) -> Union[str, None]:
# we can't import the Domain model before Django has been initialized
from mailauth.models import Domain
data = self.add_received_header(peer, data, channel)
mfdomain = mailfrom.split("@", 1)[1]
dom = None # type: Domain
try:
dom = Domain.objects.get(name=mfdomain)
except Domain.DoesNotExist:
_log.debug("Unknown domain: %s (%s)", mfdomain, mailfrom)
except OperationalError:
# this is a hacky hack, but psycopg2 falls over when haproxy closes the connection on us
_log.info("Database connection closed, Operational Error, retrying")
from django.db import connection
connection.close()
if "retry" in kwargs:
_log.exception("Database unavailable.")
return "421 Processing problem. Please try again later."
else:
return self.process_message(peer, mailfrom, rcpttos, data, retry=True, **kwargs)
signed = False
if dom is not None and dom.dkimkey:
sig = dkim.sign(data, dom.dkimselector.encode("utf-8"), dom.name.encode("utf-8"),
dom.dkimkey.replace("\r\n", "\n").encode("utf-8"))
data = b"%s%s" % (sig, data)
try:
logstr = data.decode('utf-8')
enc = "utf-8"
except UnicodeDecodeError:
logstr = data.decode('latin1')
enc = "latin1"
_log.debug("Signed output (%s):\n%s", enc, logstr)
signed = True
# now send the mail back to be processed
_log.info("Relaying %semail from <%s> to <%s>",
"DKIM signed " if signed else "",
mailfrom, rcpttos)
self.smtp.sendmail(mailfrom, rcpttos, data)
return None
评论列表
文章目录
