71 lines
2.1 KiB
Ruby
71 lines
2.1 KiB
Ruby
#
|
|
# Copyright (c) 2006-2013 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'
|
|
)
|
|
|
|
# Begins main DNS server run-loop at BeEF startup
|
|
def self.pre_http_start(http_hook_server)
|
|
dns_config = BeEF::Core::Configuration.instance.get('beef.extension.dns')
|
|
|
|
address = dns_config['address']
|
|
port = dns_config['port']
|
|
|
|
dns = BeEF::Extension::Dns::Server.instance
|
|
dns.run_server(address, port)
|
|
|
|
print_info "DNS Server: #{address}:#{port}"
|
|
servers = []
|
|
unless dns_config['upstream'].nil?
|
|
dns_config['upstream'].each do |server|
|
|
if server[1].nil? or server[2].nil?
|
|
next
|
|
end
|
|
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
|
|
|
|
# Mounts handler for processing RESTful API calls
|
|
def self.mount_handler(beef_server)
|
|
beef_server.mount('/api/dns', BeEF::Extension::Dns::DnsRest.new)
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
end
|