michael/Spoofy
1
0
Fork 0
mirror of https://github.com/MattKeeley/Spoofy.git synced 2026-02-03 13:33:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix redirect bug

Browse Source
This commit is contained in:
Matt Keeley
2024-08-10 23:08:05 -07:00
parent 5e93f7f9d1
commit aa13836d0e
2 changed files with 24 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
list.txt Normal file
View File

@@ -0,0 +1 @@
reddit.com

29
modules/spf.py
View File

@@ -17,12 +17,14 @@ class SPF:
self.num_includes = self.get_spf_includes()
self.too_many_includes = self.num_includes > 10
def get_spf_record(self):
"""Returns the SPF record for a given domain."""
def get_spf_record(self, domain=None):
"""Fetches the SPF record for the specified domain."""
try:
if not domain:
domain = self.domain
resolver = dns.resolver.Resolver()
resolver.nameservers = [self.dns_server, '1.1.1.1', '8.8.8.8']
query_result = resolver.resolve(self.domain, 'TXT')
query_result = resolver.resolve(domain, 'TXT')
for record in query_result:
if 'spf1' in str(record):
spf_record = str(record).replace('"', '')
@@ -33,12 +35,27 @@ class SPF:
def get_spf_all_string(self):
"""Returns the string value of the 'all' mechanism in the SPF record."""
if self.spf_record:
all_matches = re.findall(r'[-~?+]all', self.spf_record)
spf_record = self.spf_record
visited_domains = set()
while spf_record:
all_matches = re.findall(r'[-~?+]all', spf_record)
if len(all_matches) == 1:
return all_matches[0]
elif len(all_matches) > 1:
return '2many'
redirect_match = re.search(r'redirect=([\w.-]+)', spf_record)
if redirect_match:
redirect_domain = redirect_match.group(1)
if redirect_domain in visited_domains:
break # Prevent infinite loops in case of circular redirects
visited_domains.add(redirect_domain)
spf_record = self.get_spf_record(redirect_domain)
else:
break
return None
def get_spf_includes(self):
@@ -57,7 +74,7 @@ class SPF:
txt_record = txt_string.decode('utf-8')
if txt_record.startswith('v=spf1'):
count += count_includes(txt_record)
except Exception as e:
except Exception:
pass
# Count occurrences of 'a', 'mx', 'ptr', and 'exists' mechanisms