From dd384713c824b36d230733adcf88de9225886e9f Mon Sep 17 00:00:00 2001
From: greyhatsec-au <111026372+greyhatsec-au@users.noreply.github.com>
Date: Fri, 30 Sep 2022 19:32:28 +1000
Subject: [PATCH] Fix where not using wildcard domains and discarding unknown
 domains

---
 python/mailserver.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/mailserver.py b/python/mailserver.py
index 1df9d9d..deab5ef 100644
--- a/python/mailserver.py
+++ b/python/mailserver.py
@@ -121,6 +121,8 @@ class CustomSMTPServer(smtpd.SMTPServer):
                 for x in DOMAINS:
                     if  "*" in x and domain.endswith(x.replace('*', '')):
                         found = True
+                    elif domain == x:
+                        found = True
                 if(DISCARD_UNKNOWN and found==False):
                     logger.info('Discarding email for unknown domain: %s' % domain)
                     continue