62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
import collections
|
|
|
|
class DNSBL(object):
|
|
def __init__(self, hostname=None):
|
|
if not hostname == None:
|
|
self.hostname = hostname
|
|
|
|
def process(self, a_record, txt_record):
|
|
out = a_record
|
|
if txt_record is not None:
|
|
out += f" - {txt_record}"
|
|
return out
|
|
|
|
class ZenSpamhaus(DNSBL):
|
|
hostname = "zen.spamhaus.org"
|
|
def process(self, a_record, txt_record):
|
|
result = a_record.rsplit(".", 1)[1]
|
|
if result in ["2", "3", "9"]:
|
|
desc = "spam"
|
|
elif result in ["4", "5", "6", "7"]:
|
|
desc = "exploits"
|
|
else:
|
|
desc = "unknown"
|
|
return f"{result} - {desc}"
|
|
|
|
class EFNetRBL(DNSBL):
|
|
hostname = "rbl.efnetrbl.org"
|
|
def process(self, a_record, txt_record):
|
|
result = a_record.rsplit(".", 1)[1]
|
|
if result == "1":
|
|
desc = "proxy"
|
|
elif result in ["2", "3"]:
|
|
desc = "spamtap"
|
|
elif result == "4":
|
|
desc = "tor"
|
|
elif result == "5":
|
|
desc = "flooding"
|
|
return f"{result} - {desc}"
|
|
|
|
class DroneBL(DNSBL):
|
|
hostname = "dnsbl.dronebl.org"
|
|
|
|
class AbuseAtCBL(DNSBL):
|
|
hostname = "cbl.abuseat.org"
|
|
def process(self, a_record, txt_record):
|
|
result = a_record.rsplit(".", 1)[1]
|
|
if result == "2":
|
|
desc = "abuse"
|
|
else:
|
|
desc = "unknown"
|
|
return f"{result} - {desc}"
|
|
|
|
DEFAULT_LISTS = [
|
|
ZenSpamhaus(),
|
|
EFNetRBL(),
|
|
DroneBL(),
|
|
AbuseAtCBL()
|
|
]
|
|
|
|
def default_lists():
|
|
return collections.OrderedDict(
|
|
(dnsbl.hostname, dnsbl) for dnsbl in DEFAULT_LISTS)
|