async-dns 0.10.0

data/spec/async/dns/server/bind9/named.conf

@@ -0,0 +1,14 @@
+
+options {
+	directory "./";
+
+	listen-on-v6 { none; };
+	listen-on { 127.0.0.1; };
+
+	pid-file "./named.pid";
+};
+
+zone "local" {
+	type master;
+	file "./local.zone";
+};

data/spec/async/dns/server/million.rb

@@ -0,0 +1,85 @@
+#!/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 'async/dns'
+require 'benchmark'
+
+require 'stackprof'
+
+Name = Resolv::DNS::Name
+IN = Resolv::DNS::Resource::IN
+
+# Generate a million A record "domains":
+
+million = {}
+
+Benchmark.bm do |x|
+	x.report("Generate names") do
+		(1..1_000_000).each do |i|
+			domain = "domain#{i}.local"
+	
+			million[domain] = "#{69}.#{(i >> 16)%256}.#{(i >> 8)%256}.#{i%256}"
+		end
+	end
+end
+
+# Run the server:
+
+StackProf.run(mode: :cpu, out: 'async/dns.stackprof') do
+	Async::DNS::run_server(:listen => [[:udp, '0.0.0.0', 5300]]) do
+		@logger.level = Logger::WARN
+	
+		match(//, IN::A) do |transaction|
+			transaction.respond!(million[transaction.name])
+		end
+		
+		# Default DNS handler
+		otherwise do |transaction|
+			logger.info "Passing DNS request upstream..."
+			transaction.fail!(:NXDomain)
+		end
+	end
+end
+
+# Expected output:
+#
+# > dig @localhost -p 5300 domain1000000
+# 
+# ; <<>> DiG 9.8.3-P1 <<>> @localhost -p 5300 domain1000000
+# ; (3 servers found)
+# ;; global options: +cmd
+# ;; Got answer:
+# ;; ->>HEADER<<- opcode: QUERY, status: NOERROR, id: 50336
+# ;; flags: qr aa rd; QUERY: 1, ANSWER: 1, AUTHORITY: 0, ADDITIONAL: 0
+# ;; WARNING: recursion requested but not available
+# 
+# ;; QUESTION SECTION:
+# ;domain1000000.			IN	A
+# 
+# ;; ANSWER SECTION:
+# domain1000000.		86400	IN	A	69.15.66.64
+# 
+# ;; Query time: 1 msec
+# ;; SERVER: 127.0.0.1#5300(127.0.0.1)
+# ;; WHEN: Fri May 16 19:17:48 2014
+# ;; MSG SIZE  rcvd: 47
+# 

data/spec/async/dns/server_performance_spec.rb

@@ -0,0 +1,138 @@
+#!/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 'async/dns'
+require 'benchmark'
+require 'process/daemon'
+
+module Async::DNS::ServerPerformanceSpec
+	describe Async::DNS::Server do
+		include_context "reactor"
+		
+		context 'benchmark' do
+			class MillionServer < Async::DNS::Server
+				def initialize(*)
+					super
+					
+					@million = {}
+					
+					(1..5_000).each do |i|
+						domain = "domain#{i}.local"
+	
+						@million[domain] = "#{69}.#{(i >> 16)%256}.#{(i >> 8)%256}.#{i%256}"
+					end
+				end
+				
+				def process(name, resource_class, transaction)
+					transaction.respond!(@million[name])
+				end
+			end
+			
+			class AsyncServerDaemon < Process::Daemon
+				def working_directory
+					File.expand_path("../tmp", __FILE__)
+				end
+				
+				def reactor
+					@reactor ||= Async::Reactor.new
+				end
+				
+				def startup
+					puts "Starting DNS server..."
+					@server = MillionServer.new(listen: [[:udp, '0.0.0.0', 5300]])
+					
+					reactor.async do
+						@task = @server.run
+					end
+				end
+				
+				def run
+					reactor.run
+				end
+				
+				def shutdown
+					@task.stop if @task
+				end
+			end
+
+			class Bind9ServerDaemon < Process::Daemon
+				def working_directory
+					File.expand_path("../server/bind9", __FILE__)
+				end
+	
+				def run
+					exec(self.class.named_executable, "-c", "named.conf", "-f", "-p", "5400", "-g")
+				end
+	
+				def self.named_executable
+					# Check if named executable is available:
+					@named ||= `which named`.chomp
+				end
+			end
+		
+			before(:all) do
+				@servers = []
+				@servers << ["Async::DNS::Server", 5300]
+			
+				@domains = (1..1000).collect do |i|
+					"domain#{i}.local"
+				end
+			
+				AsyncServerDaemon.start
+			
+				unless Bind9ServerDaemon.named_executable.empty?
+					Bind9ServerDaemon.start
+					@servers << ["Bind9", 5400]
+				end
+			end
+		
+			after(:all) do
+				AsyncServerDaemon.stop
+			
+				unless Bind9ServerDaemon.named_executable.empty?
+					Bind9ServerDaemon.stop
+				end
+			end
+		
+			it 'takes time' do
+				Async.logger.level = Logger::ERROR
+				
+				Benchmark.bm(30) do |x|
+					@servers.each do |name, port|
+						resolver = Async::DNS::Resolver.new([[:udp, '127.0.0.1', port]])
+					
+						x.report(name) do
+							# Number of requests remaining since this is an asynchronous event loop:
+							5.times do
+								pending = @domains.size
+							
+								resolved = @domains.collect{|domain| resolver.addresses_for(domain)}
+								
+								expect(resolved).to_not include(nil)
+							end
+						end
+					end
+				end
+			end
+		end
+	end
+end

data/spec/async/dns/slow_server_spec.rb

@@ -0,0 +1,97 @@
+#!/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 'async/dns'
+
+module Async::DNS::SlowServerSpec
+	IN = Resolv::DNS::Resource::IN
+	
+	class SlowServer < Async::DNS::Server
+		def process(name, resource_class, transaction)
+			@resolver ||= Async::DNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]])
+			
+			Async::Task.current.sleep(2) if name.end_with?('.com')
+			
+			transaction.fail!(:NXDomain)
+		end
+	end
+	
+	describe "Async::DNS Slow Server" do
+		include_context "reactor"
+		
+		let(:server_interfaces) {[[:udp, '0.0.0.0', 5330], [:tcp, '0.0.0.0', 5330]]}
+		let(:server) {SlowServer.new(listen: server_interfaces)}
+		
+		around(:each) do |example|
+			begin
+				task = server.run
+				
+				example.run
+			ensure
+				task.stop
+			end
+		end
+		
+		it "get no answer after 2 seconds" do
+			start_time = Time.now
+			
+			resolver = Async::DNS::Resolver.new(server_interfaces, :timeout => 10)
+			
+			response = resolver.query("apple.com", IN::A)
+			
+			expect(response.answer.length).to be == 0
+			
+			end_time = Time.now
+			
+			expect(end_time - start_time).to be_within(0.1).of(2.0)
+		end
+	
+		it "times out after 1 second" do
+			start_time = Time.now
+			
+			# Two server interfaces, timeout of 0.5s each:
+			resolver = Async::DNS::Resolver.new(server_interfaces, :timeout => 0.5)
+			
+			response = resolver.query("apple.com", IN::A)
+			
+			expect(response).to be nil
+			
+			end_time = Time.now
+			
+			expect(end_time - start_time).to be_within(0.1).of(1.0)
+		end
+	
+		it "gets no answer immediately" do
+			start_time = Time.now
+			
+			resolver = Async::DNS::Resolver.new(server_interfaces, :timeout => 0.5)
+			
+			response = resolver.query("oriontransfer.org", IN::A)
+			
+			expect(response.answer.length).to be 0
+			
+			end_time = Time.now
+			
+			expect(end_time - start_time).to be_within(0.1).of(0.0)
+		end
+	end
+end

data/spec/async/dns/socket_spec.rb

@@ -0,0 +1,70 @@
+#!/usr/bin/env ruby
+
+# Copyright, 2014, 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 'async/dns'
+require 'async/dns/system'
+
+module Async::DNS::SocketSpec
+	IN = Resolv::DNS::Resource::IN
+	
+	class TestServer < Async::DNS::Server
+		def process(name, resource_class, transaction)
+			@resolver ||= Async::DNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]])
+			
+			transaction.passthrough!(@resolver)
+		end
+	end
+	
+	describe Async::DNS::TCPSocketHandler do
+		include_context "reactor"
+		
+		let(:server_interfaces) {[TCPServer.new('127.0.0.1', 2002)]}
+		let(:server) {TestServer.new(listen: server_interfaces)}
+		
+		it "should create server with existing TCP socket" do
+			task = server.run
+			
+			resolver = Async::DNS::Resolver.new([[:tcp, '127.0.0.1', 2002]])
+			response = resolver.query('google.com')
+			expect(response.class).to be == Async::DNS::Message
+			
+			task.stop
+		end
+	end
+	
+	describe Async::DNS::UDPSocketHandler do
+		include_context "reactor"
+		
+		let(:server_interfaces) {[UDPSocket.new.tap{|socket| socket.bind('127.0.0.1', 2002)}]}
+		let(:server) {TestServer.new(listen: server_interfaces)}
+		
+		it "should create server with existing UDP socket" do
+			task = server.run
+			
+			resolver = Async::DNS::Resolver.new([[:udp, '127.0.0.1', 2002]])
+			response = resolver.query('google.com')
+			expect(response.class).to be == Async::DNS::Message
+			
+			task.stop
+		end
+	end
+end

data/spec/async/dns/system_spec.rb

@@ -0,0 +1,57 @@
+#!/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 'async/dns'
+require 'async/dns/system'
+
+module Async::DNS::SystemSpec
+	describe Async::DNS::System do
+		include_context "reactor"
+		
+		it "should have at least one namesever" do
+			expect(Async::DNS::System::nameservers.length).to be > 0
+		end
+	
+		it "should respond to query for google.com" do
+			resolver = Async::DNS::Resolver.new(Async::DNS::System::nameservers)
+		
+			response = resolver.query('google.com')
+		
+			expect(response.class).to be == Async::DNS::Message
+			expect(response.rcode).to be == Resolv::DNS::RCode::NoError
+		end
+	end
+
+	describe Async::DNS::System::Hosts do
+		it "should parse the hosts file" do
+			hosts = Async::DNS::System::Hosts.new
+		
+			# Load the test hosts data:
+			File.open(File.expand_path("../hosts.txt", __FILE__)) do |file|
+				hosts.parse_hosts(file)
+			end
+		
+			expect(hosts.call('testing')).to be == true
+			expect(hosts['testing']).to be == '1.2.3.4'
+		end
+	end
+end