111 lines
2.7 KiB
Ruby
111 lines
2.7 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 RubyDNS
|
|
|
|
# Behaves exactly the same, except without any logger output
|
|
def self.run_server(options = {}, &block)
|
|
server = RubyDNS::Server.new(&block)
|
|
|
|
options[:listen] ||= [[:udp, "0.0.0.0", 53], [:tcp, "0.0.0.0", 53]]
|
|
|
|
EventMachine.run do
|
|
server.fire(:setup)
|
|
|
|
options[:listen].each do |spec|
|
|
if spec[0] == :udp
|
|
EventMachine.open_datagram_socket(spec[1], spec[2], UDPHandler, server)
|
|
elsif spec[0] == :tcp
|
|
EventMachine.start_server(spec[1], spec[2], TCPHandler, server)
|
|
end
|
|
end
|
|
|
|
server.load_rules
|
|
server.fire(:start)
|
|
end
|
|
|
|
server.fire(:stop)
|
|
end
|
|
|
|
class Server
|
|
|
|
attr_accessor :rules
|
|
|
|
class Rule
|
|
|
|
attr_accessor :id
|
|
|
|
# Now uses an 'id' parameter to uniquely identify rules
|
|
def initialize(id, pattern, callback)
|
|
@id = id
|
|
@pattern = pattern
|
|
@callback = callback
|
|
end
|
|
|
|
end
|
|
|
|
# Now uses an 'id' parameter to uniquely identify rules
|
|
def match(id, *pattern, block)
|
|
catch :match do
|
|
# Check if rule is already present
|
|
BeEF::Core::Models::DNS.each { |rule| throw :match if rule.id == id }
|
|
|
|
@rules << Rule.new(id, pattern, block)
|
|
|
|
# Add new rule to database
|
|
BeEF::Core::Models::DNS.create(
|
|
:id => id,
|
|
:pattern => pattern,
|
|
:block => block.to_source
|
|
)
|
|
end
|
|
end
|
|
|
|
# New method that removes a rule given its id
|
|
def remove_rule(id)
|
|
@rules.delete_if { |rule| rule.id == id }
|
|
|
|
begin
|
|
BeEF::Core::Models::DNS.get!(id).destroy
|
|
rescue DataMapper::ObjectNotFoundError => e
|
|
@logger.error(e.message)
|
|
end
|
|
end
|
|
|
|
# New method that loads all rules from the database at server startup
|
|
def load_rules
|
|
BeEF::Core::Models::DNS.each do |rule|
|
|
id = rule.id
|
|
pattern = rule.pattern
|
|
block = eval rule.block
|
|
|
|
@rules << Rule.new(id, pattern, block)
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
class Transaction
|
|
|
|
# Behaves exactly the same, except using debug logger instead of info
|
|
def respond!(*data)
|
|
options = data.last.kind_of?(Hash) ? data.pop : {}
|
|
resource_class = options[:resource_class] || @resource_class
|
|
|
|
if resource_class == nil
|
|
raise ArgumentError, "Could not instantiate resource #{resource_class}!"
|
|
end
|
|
|
|
@server.logger.debug("Resource class: #{resource_class.inspect}")
|
|
resource = resource_class.new(*data)
|
|
@server.logger.debug("Resource: #{resource.inspect}")
|
|
|
|
append!(resource, options)
|
|
end
|
|
|
|
end
|
|
|
|
end
|