82e4b1eac7
Even though it is unlikely that a user would remove these options from the DNS config file, it is still good practice to have these safeguards in place.
84 lines
2.6 KiB
Ruby
84 lines
2.6 KiB
Ruby
#
|
|
# Copyright (c) 2006-2014 Wade Alcorn - wade@bindshell.net
|
|
# Browser Exploitation Framework (BeEF) - http://beefproject.com
|
|
# See the file 'doc/COPYING' for copying permission
|
|
#
|
|
module BeEF
|
|
module Extension
|
|
module Dns
|
|
module API
|
|
|
|
module NameserverHandler
|
|
|
|
BeEF::API::Registrar.instance.register(
|
|
BeEF::Extension::Dns::API::NameserverHandler,
|
|
BeEF::API::Server,
|
|
'pre_http_start'
|
|
)
|
|
|
|
BeEF::API::Registrar.instance.register(
|
|
BeEF::Extension::Dns::API::NameserverHandler,
|
|
BeEF::API::Server,
|
|
'mount_handler'
|
|
)
|
|
|
|
# Starts the DNS nameserver at BeEF startup.
|
|
#
|
|
# @param http_hook_server [BeEF::Core::Server] HTTP server instance
|
|
def self.pre_http_start(http_hook_server)
|
|
dns_config = BeEF::Core::Configuration.instance.get('beef.extension.dns')
|
|
dns = BeEF::Extension::Dns::Server.instance
|
|
|
|
protocol = dns_config['protocol'].to_sym rescue :udp
|
|
address = dns_config['address'] || '127.0.0.1'
|
|
port = dns_config['port'] || 5300
|
|
interfaces = [[protocol, address, port]]
|
|
|
|
Thread.new { EventMachine.next_tick { dns.run(:listen => interfaces) } }
|
|
|
|
print_info "DNS Server: #{address}:#{port} (#{protocol})"
|
|
|
|
# @todo Upstream servers are not yet supported. Uncomment this section when they are.
|
|
=begin
|
|
servers = []
|
|
|
|
unless dns_config['upstream'].nil?
|
|
dns_config['upstream'].each do |server|
|
|
next if server[1].nil? or server[2].nil?
|
|
|
|
if server[0] == 'tcp'
|
|
servers << ['tcp', server[1], server[2]]
|
|
elsif server[0] == 'udp'
|
|
servers << ['udp', server[1], server[2]]
|
|
end
|
|
end
|
|
end
|
|
|
|
if servers.empty?
|
|
servers << ['tcp', '8.8.8.8', 53]
|
|
servers << ['udp', '8.8.8.8', 53]
|
|
end
|
|
|
|
upstream_servers = ''
|
|
servers.each do |server|
|
|
upstream_servers << "Upstream Server: #{server[1]}:#{server[2]} (#{server[0]})\n"
|
|
end
|
|
|
|
print_more upstream_servers
|
|
=end
|
|
end
|
|
|
|
# Mounts the handler for processing DNS RESTful API requests.
|
|
#
|
|
# @param beef_server [BeEF::Core::Server] HTTP server instance
|
|
def self.mount_handler(beef_server)
|
|
beef_server.mount('/api/dns', BeEF::Extension::Dns::DnsRest.new)
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
end
|