oadmin/emailsystemv2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
huangzhenpc
2025-02-27 09:27:44 +08:00
parent 064b44c3b3
commit c158e42ded
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/utils.py
View File

@@ -30,6 +30,9 @@ class CustomSMTPServer(smtpd.SMTPServer):
logger.info(f"Initializing SMTP server on {localaddr}") logger.info(f"Initializing SMTP server on {localaddr}")
super().__init__(localaddr, remoteaddr) super().__init__(localaddr, remoteaddr)
# 定义允许的域名列表
ALLOWED_DOMAINS = ['nosqli.com', 'email.nosqli.com']
def process_message(self, peer, mailfrom, rcpttos, data, **kwargs): def process_message(self, peer, mailfrom, rcpttos, data, **kwargs):
try: try:
logger.debug(f"Connection from peer: {peer}") logger.debug(f"Connection from peer: {peer}")
@@ -44,7 +47,8 @@ class CustomSMTPServer(smtpd.SMTPServer):
# 验证收件人域名 # 验证收件人域名
valid_recipients = [] valid_recipients = []
for rcpt in rcpttos: for rcpt in rcpttos:
if not rcpt.endswith('@nosqli.com'): is_valid = any(rcpt.endswith(f'@{domain}') for domain in self.ALLOWED_DOMAINS)
if not is_valid:
logger.warning(f"Rejected mail to {rcpt}: invalid domain") logger.warning(f"Rejected mail to {rcpt}: invalid domain")
else: else:
valid_recipients.append(rcpt) valid_recipients.append(rcpt)