rubydns 0.6.0 → 2.0.0

data/lib/rubydns/transaction.rb

@@ -1,250 +0,0 @@
-# Copyright, 2009, 2012, by Samuel G. D. Williams. <http://www.codeotaku.com>
-# 
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-# 
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-# 
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-require 'eventmachine'
-
-module RubyDNS
-
-	# This class provides all details of a single DNS question and answer. This
-	# is used by the DSL to provide DNS related functionality.
-	class Transaction
-		include EventMachine::Deferrable
-		
-		def initialize(server, query, question, resource_class, answer, options = {})
-			@server = server
-			@query = query
-			@question = question
-			@resource_class = resource_class
-			@answer = answer
-
-			@options = options
-
-			@deferred = false
-			@question_appended = false
-		end
-
-		# The resource_class that was requested. This is typically used to generate a
-		# response.
-		attr :resource_class
-		
-		# The incoming query which is a set of questions.
-		attr :query
-		
-		# The question that this transaction represents.
-		attr :question
-		
-		# The current full answer to the incoming query.
-		attr :answer
-		
-		# Any options or configuration associated with the given transaction.
-		attr :options
-		
-		# Return the name of the question, which is typically the requested hostname.
-		def name
-			@question.to_s
-		end
-
-		# Suitable for debugging purposes
-		def to_s
-			"#{name} #{@resource_class.name}"
-		end
-
-		# Run a new query through the rules with the given name and resource type. The
-		# results of this query are appended to the current transactions <tt>answer</tt>.
-		def append_query!(name, resource_class = nil, options = {})
-			Transaction.new(@server, @query, name, resource_class || @resource_class, @answer, options).process
-		end
-
-		def process(&finished)
-			@server.process(name, @resource_class, self)
-
-			unless @deferred
-				succeed(self)
-			end
-		end
-
-		def defer!
-			@deferred = true
-		end
-
-		# Use the given resolver to respond to the question. The default functionality is
-		# implemented by passthrough, and if a reply is received, it will be merged with the
-		# answer for this transaction.
-		#
-		# If a block is supplied, this function yields with the reply and reply_name if
-		# successful. This could be used, for example, to update a cache or modify the
-		# reply.
-		def passthrough!(resolver, options = {}, &block)
-			passthrough(resolver, options) do |response|
-				if block_given?
-					yield response
-				end
-				
-				@answer.merge!(response)
-				
-				succeed if @deferred
-			end
-			
-			true
-		end
-		
-		# Use the given resolver to respond to the question. If recursion is
-		# not requested, the result is <tt>failure!(:Refused)</tt>. If the resolver does
-		# not respond, the result is <tt>failure!(:NXDomain)</tt>
-		#
-		# If a block is supplied, this function yields with the reply and reply_name if
-		# successful. This block is responsible for doing something useful with the reply,
-		# such as merging it or conditionally discarding it.
-		#
-		# A second argument, options, provides some control over the passthrough process.
-		# :force => true, ensures that the query will occur even if recursion is not requested.
-		def passthrough(resolver, options = {}, &block)
-			if @query.rd || options[:force]
-				# Resolver is asynchronous, so we are now deferred:
-				defer!
-
-				resolver.query(name, resource_class) do |response|
-					case response
-					when RubyDNS::Message
-						yield response
-					when RubyDNS::ResolutionFailure
-						failure!(:ServFail)
-					else
-						# This shouldn't ever happen, but if it does for some reason we shouldn't hang.
-						fail(response)
-					end
-				end
-			else
-				failure!(:Refused)
-			end
-			
-			true
-		end
-
-		# Respond to the given query with a resource record. The arguments to this
-		# function depend on the <tt>resource_class</tt> requested. The last argument
-		# can optionally be a hash of options.
-		#
-		# <tt>options[:resource_class]</tt>:: Override the default <tt>resource_class</tt>
-		# <tt>options[:ttl]</tt>:: Specify the TTL for the resource
-		# <tt>options[:name]</tt>:: Override the name (question) of the response.
-		#
-		# for A records:: <tt>respond!("1.2.3.4")</tt>
-		# for MX records::  <tt>respond!("mail.blah.com", 10)</tt>
-		#
-		# This function instantiates the resource class with the supplied arguments, and
-		# then passes it to <tt>append!</tt>.
-		#
-		# See <tt>Resolv::DNS::Resource</tt> for more information about the various 
-		# <tt>resource_class</tt>s available. 
-		# http://www.ruby-doc.org/stdlib/libdoc/resolv/rdoc/index.html
-		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.info "Resource class: #{resource_class.inspect}"
-			resource = resource_class.new(*data)
-			@server.logger.info "Resource: #{resource.inspect}"
-			
-			append!(resource, options)
-		end
-
-		# Append a given set of resources to the answer. The last argument can 
-		# optionally be a hash of options.
-		# 
-		# <tt>options[:ttl]</tt>:: Specify the TTL for the resource
-		# <tt>options[:name]</tt>:: Override the name (question) of the response.
-		# <tt>options[:section]</tt>:: Specify whether the response should go in the `:answer`
-		#                             `:authority` or `:additional` section.
-		# 
-		# This function can be used to supply multiple responses to a given question.
-		# For example, each argument is expected to be an instantiated resource from
-		# <tt>Resolv::DNS::Resource</tt> module.
-		def append! (*resources)
-			append_question!
-
-			if resources.last.kind_of?(Hash)
-				options = resources.pop
-			else
-				options = {}
-			end
-
-			# Use the default options if provided:
-			options = options.merge(@options)
-
-			options[:ttl] ||= 16000
-			options[:name] ||= @question.to_s + "."
-			
-			method = ("add_" + (options[:section] || 'answer').to_s).to_sym
-
-			resources.each do |resource|
-				@server.logger.debug "#{method}: #{resource.inspect} #{resource.class::TypeValue} #{resource.class::ClassValue}"
-				
-				@answer.send(method, options[:name], options[:ttl], resource)
-			end
-
-			succeed if @deferred
-
-			true
-		end
-
-		# This function indicates that there was a failure to resolve the given
-		# question. The single argument must be an integer error code, typically
-		# given by the constants in <tt>Resolv::DNS::RCode</tt>.
-		#
-		# The easiest way to use this function it to simply supply a symbol. Here is
-		# a list of the most commonly used ones:
-		#
-		# <tt>:NoError</tt>:: No error occurred.
-		# <tt>:FormErr</tt>::	The incoming data was not formatted correctly.
-		# <tt>:ServFail</tt>:: The operation caused a server failure (internal error, etc).
-		# <tt>:NXDomain</tt>:: Non-eXistant Domain (domain record does not exist).
-		# <tt>:NotImp</tt>:: The operation requested is not implemented.
-		# <tt>:Refused</tt>:: The operation was refused by the server.
-		# <tt>:NotAuth</tt>:: The server is not authoritive for the zone.
-		#
-		# See http://www.rfc-editor.org/rfc/rfc2929.txt for more information
-		# about DNS error codes (specifically, page 3).
-		def failure! (rcode)
-			append_question!
-
-			if rcode.kind_of? Symbol
-				@answer.rcode = Resolv::DNS::RCode.const_get(rcode)
-			else
-				@answer.rcode = rcode.to_i
-			end
-
-			# The transaction itself has completed, but contains a failure:
-			succeed(rcode) if @deferred
-
-			true
-		end
-
-		def append_question!
-			if @answer.question.size == 0
-				@answer.add_question(@question, @resource_class) unless @question_appended
-			end
-		end
-	end
-end

data/test/examples/geoip-dns.rb

@@ -1,86 +0,0 @@
-#!/usr/bin/env ruby
-
-# Copyright, 2009, 2012, by Samuel G. D. Williams. <http://www.codeotaku.com>
-# 
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-# 
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-# 
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-require 'rubygems'
-
-require 'geoip'
-
-require 'rexec'
-require 'rexec/daemon'
-
-require 'rubygems'
-require 'rubydns'
-
-require 'rubydns/resolver'
-require 'rubydns/system'
-
-INTERFACES = [
-	[:udp, "0.0.0.0", 5300]
-]
-
-# This daemon requires the file downloaded from http://geolite.maxmind.com/download/geoip/database/GeoLiteCountry/GeoIP.dat.gz
-# For more information, please see http://www.maxmind.com/en/geolite and http://geoip.rubyforge.org
-
-class GeoIPDNSDaemon < RExec::Daemon::Base
-	# You can specify a specific directory to use for run-time information (pid, logs, etc):
-	# @@base_directory = File.expand_path("../", __FILE__)
-	# @@base_directory = "/var"
-
-	Name = Resolv::DNS::Name
-	IN = Resolv::DNS::Resource::IN
-	R = RubyDNS::Resolver.new(RubyDNS::System::nameservers)
-	GEO = GeoIP.new(File.expand_path('../GeoLiteCountry.dat', __FILE__))
-	
-	def self.run
-		RubyDNS::run_server(:listen => INTERFACES) do
-			match(//, IN::A) do |transaction|
-				location = nil
-				peer = transaction.options[:peer]
-				
-				if peer
-					logger.debug "Looking up geographic information for peer #{peer}"
-					location = GEO.country(peer[0])
-				end
-				
-				if location
-					logger.debug "Found location #{location}"
-				end
-				
-				case location.continent_code
-				when "EU"
-					transaction.respond!("1.1.1.1")
-				when "CN", "JP"
-					transaction.respond!("1.1.2.1")
-				else
-					transaction.respond!("1.1.3.1")
-				end
-			end
-
-			# Default DNS handler
-			otherwise do |transaction|
-				transaction.passthrough!(R)
-			end
-		end
-  end
-end
-
-GeoIPDNSDaemon.daemonize

data/test/helper.rb

@@ -1,9 +0,0 @@
-
-$LOAD_PATH.unshift File.expand_path("../../lib/", __FILE__)
-
-require 'rubygems'
-require 'test/unit'
-require 'resolv'
-
-Name = Resolv::DNS::Name
-IN = Resolv::DNS::Resource::IN

data/test/test_daemon.rb

@@ -1,100 +0,0 @@
-#!/usr/bin/env ruby
-
-# Copyright, 2012, by Samuel G. D. Williams. <http://www.codeotaku.com>
-# 
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-# 
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-# 
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-require 'helper'
-require 'pathname'
-
-require 'rubydns'
-require 'rubydns/resolver'
-
-require 'rexec'
-require 'rexec/daemon'
-
-class BasicTestServer < RExec::Daemon::Base
-	SERVER_PORTS = [[:udp, '127.0.0.1', 5350], [:tcp, '127.0.0.1', 5350]]
-
-	@@base_directory = File.dirname(__FILE__)
-
-	def self.run
-		# Start the RubyDNS server
-		RubyDNS::run_server(:listen => SERVER_PORTS) do
-			match("test.local", IN::A) do |transaction|
-				transaction.respond!("192.168.1.1")
-			end
-
-			match(/foo.*/, IN::A) do |transaction|
-				transaction.respond!("192.168.1.2")
-			end
-
-			# Default DNS handler
-			otherwise do |transaction|
-				transaction.failure!(:NXDomain)
-			end
-		end
-	end
-end
-
-class DaemonTest < Test::Unit::TestCase
-	def setup
-		$stderr.puts "Starting test server..."
-		BasicTestServer.start
-	end
-	
-	def teardown
-		$stderr.puts "Stoping test server..."
-		BasicTestServer.stop
-	end
-	
-	def test_basic_dns
-		assert_equal :running, RExec::Daemon::ProcessFile.status(BasicTestServer)
-		
-		EventMachine.run do
-			resolver = resolver = RubyDNS::Resolver.new(BasicTestServer::SERVER_PORTS)
-		
-			resolver.query("test.local") do |response|
-				answer = response.answer.first
-				
-				assert_equal "test.local", answer[0].to_s
-				assert_equal "192.168.1.1", answer[2].address.to_s
-				
-				EventMachine.stop
-			end
-		end
-	end
-	
-	def test_pattern_matching
-		assert_equal :running, RExec::Daemon::ProcessFile.status(BasicTestServer)
-
-		EventMachine.run do
-			resolver = resolver = RubyDNS::Resolver.new(BasicTestServer::SERVER_PORTS)
-
-			resolver.query("foobar") do |response|
-				answer = response.answer.first
-				
-				assert_equal "foobar", answer[0].to_s
-				assert_equal "192.168.1.2", answer[2].address.to_s
-				
-				EventMachine.stop
-			end
-		end
-	end
-end

data/test/test_domains.txt

@@ -1,185 +0,0 @@
-0.channel18.facebook.com
-2leep.com
-EVIntl-ocsp.verisign.com
-EVSecure-ocsp.verisign.com
-a.tribalfusion.com
-a1059.g.akamai.net
-a126.g.akamai.net
-a1338.g.akamai.net
-a199.gi3.akamai.net
-a248.e.akamai.net
-a696.g.akamai.net
-a803.g.akamai.net
-ad.doubleclick.net
-ad.yieldmanager.com
-ad1.netshelter.net
-ad2.netshelter.net
-address.yahoo.com
-ads.adbrite.com
-ads.trademe.co.nz
-ads.tw.adsonar.com
-ads1.perfadbrite.com.akadns.net
-adsfac.us
-ajax.googleapis.com
-aleccolocco.blogspot.com
-anycast-asia.quantserve.com.akadns.net
-apps.facebook.com
-appspot.l.google.com
-ayako.oriontransfer.org
-b.scorecardresearch.com
-b1.adbrite.com
-backend.deviantart.com
-blog.nominet.org.uk
-c.fsdn.com
-cache-01.cleanprint.net
-clients.l.google.com
-clients1.google.com
-cms.myspacecdn.com
-configuration.apple.com
-d.yimg.com
-d1.openx.org
-data.coremetrics.com
-developer.apple.com
-devimages.apple.com
-dg.specificclick.net
-digg.analytics.live.com
-digg.com
-e.vistaprint.com
-englishrussia.com
-feeds.digg.com
-feeds.dzone.com
-feeds.feedburner.com
-feeds2.feedburner.com
-filetransfer.msg.yahoo.com
-filterforge.com
-financier.oriontransfer.co.nz
-gems.oriontransfer.org
-ghs.l.google.com
-googleads.g.doubleclick.net
-googlemail-pop.l.google.com
-groups.google.com
-hat.bmanpn.com
-hogbaysoftware.appspot.com
-i.creativecommons.org
-id.google.com
-images.apple.com
-images.macrumors.com
-images.trademe.co.nz
-img.slate.com
-js.adsonar.com
-js.revsci.net
-kotaku.com
-l.sharethis.com
-l.yimg.com
-lambda-the-ultimate.org
-login.facebook.com
-login.live.com
-login.messaging.aol.com
-login.oscar.aol.com
-login.yahoo.com
-lus.sharethis.com.akadns.net
-m1.2mdn.net
-mac.com
-mail.google.com
-mail.mac.com
-mail.me.com
-managingosx.wordpress.com
-maps.google.com
-marketing.vistaprint.com
-media.digg.com
-media.washingtonpost.com
-messenger.hotmail.com
-metrics.washingtonpost.com
-metro-amz-lb1-1649407802.us-east-1.elb.amazonaws.com
-mm.chitika.net
-news.ycombinator.com
-o.aolcdn.com
-o.sa.aol.com
-ocsp.verisign.net
-odb.outbrain.com
-omega.contacts.msn.com
-omega.contacts.msn.com.nsatc.net
-pagead2.googlesyndication.com
-partner.googleadservices.com
-partnerad.l.google.com
-pixel.quantserve.com
-pop.googlemail.com
-ps.friendconnect.gmodules.com
-publish.iwork.com
-r1rk9np7bpcsfoeekl0khkd2juj27q3o.friendconnect.gmodules.com
-rad.msn.com
-regretfulmorning.com
-resources.infolinks.com
-rss.lists.apple.com
-rss.slashdot.org
-s24.sitemeter.com
-s26.sitemeter.com
-s7.addthis.com
-safebrowsing-cache.google.com
-safebrowsing.clients.google.com
-scripts.chitika.net
-scsa.msg.yahoo.com
-seal.verisign.com
-sec.westpac.co.nz
-secure-nz.imrworldwide.com
-shared-hosted-1561031087.us-east-1.elb.amazonaws.com
-slashdot.org
-slate.cleanprint.net
-spacemusic.libsyn.com
-ssl-google-analytics.l.google.com
-ssl-hints.netflame.cc
-ssl.google-analytics.com
-static.ak.connect.facebook.com
-static.ak.facebook.com
-static.getclicky.com
-stats.wordpress.com
-syndication.thedailywtf.com
-tags.expo9.exponential.com
-talk.google.com
-time.asia.apple.com
-tools.google.com
-votes.buzz.yahoo.com
-widgets.outbrain.com
-wiki.oriontransfer.org
-www-google-analytics.l.google.com
-www.acquire.co.nz
-www.apple.com
-www.apple.com.akadns.net
-www.applejoe.com
-www.apples.com
-www.ascent.co.nz
-www.bing.com
-www.blogcdn.com
-www.blogger.com
-www.blogsmithmedia.com
-www.co2stats.com
-www.demonoid.com
-www.digg.com
-www.drobostore.com
-www.dzone.com
-www.em6live.co.nz
-www.engadget.com
-www.facebook.com
-www.filterforge.com
-www.friendconnect.gmodules.com
-www.geekologie.com
-www.google-analytics.com
-www.google.com
-www.idevgames.com
-www.iwatchstuff.com
-www.l.google.com
-www.linuxquestions.org
-www.lucidsystems.org
-www.macrumors.com
-www.michael.net.nz
-www.potionfactory.com
-www.recipepuppy.com
-www.slashdot.org
-www.slate.com
-www.sqlite.org
-www.trademe.co.nz
-www.vistaprint.com
-www.washingtonpost.com
-www.westpac.co.nz
-www4.l.google.com
-z.digg.com

data/test/test_passthrough.rb

@@ -1,80 +0,0 @@
-#!/usr/bin/env ruby
-
-# Copyright, 2012, by Samuel G. D. Williams. <http://www.codeotaku.com>
-# 
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-# 
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-# 
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-require 'helper'
-require 'pathname'
-
-require 'rubydns'
-require 'rubydns/resolver'
-
-require 'rexec'
-require 'rexec/daemon'
-
-class TestPassthroughServer < RExec::Daemon::Base
-	SERVER_PORTS = [[:udp, '127.0.0.1', 5340], [:tcp, '127.0.0.1', 5340]]
-	
-	@@base_directory = File.dirname(__FILE__)
-
-	def self.run
-		resolver = RubyDNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]])
-		
-		# Start the RubyDNS server
-		RubyDNS::run_server(:listen => SERVER_PORTS) do
-			match(/.*\.com/, IN::A) do |transaction|
-				transaction.passthrough!(resolver)
-			end
-
-			# Default DNS handler
-			otherwise do |transaction|
-				transaction.failure!(:NXDomain)
-			end
-		end
-	end
-end
-
-class PassthroughTest < Test::Unit::TestCase
-	def setup
-		TestPassthroughServer.start
-	end
-	
-	def teardown
-		TestPassthroughServer.stop
-	end
-	
-	def test_basic_dns
-		answer = nil
-		
-		assert_equal :running, RExec::Daemon::ProcessFile.status(TestPassthroughServer)
-		
-		EventMachine.run do
-			resolver = RubyDNS::Resolver.new(TestPassthroughServer::SERVER_PORTS)
-		
-			resolver.query("google.com") do |response|
-				answer = response.answer.first
-				
-				EventMachine.stop
-			end
-		end
-		
-		assert answer.count > 0
-	end
-end