Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Allow specifying of nameserver as '!dns <host> @1.2.3.4'

Browse source
This commit is contained in:
jesopo 2019-06-25 14:50:30 +01:00
parent e8ebaadd11
commit 6f5c692606
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
modules/ip_addresses.py
View file

@ -25,12 +25,19 @@ class Module(ModuleManager.BaseModule):
:prefix: DNS :prefix: DNS
""" """
hostname = event["args_split"][0] hostname = event["args_split"][0]
nameserver = event["server"].get_setting("dns-nameserver", None)
has_nameserver = not nameserver == None
record_types = ["A?", "AAAA?"] nameserver = event["server"].get_setting("dns-nameserver", None)
record_types = []
if len(event["args_split"]) > 1: if len(event["args_split"]) > 1:
record_types = [t.upper() for t in event["args_split"][1:]] for arg in event["args_split"][1:]:
if arg.startswith("@"):
nameserver = arg[1:]
else:
record_types.append(arg.upper())
if not record_types:
record_types = ["A?", "AAAA?"]
if not nameserver == None: if not nameserver == None:
resolver = dns.resolver.Resolver(configure=False) resolver = dns.resolver.Resolver(configure=False)