celluloid-dns 0.0.1 → 0.17.3

data/spec/celluloid/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/celluloid/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 'celluloid/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: 'celluloid/dns.stackprof') do
+	Celluloid::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/celluloid/dns/server_performance_spec.rb

@@ -0,0 +1,139 @@
+#!/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 'celluloid/dns'
+require 'benchmark'
+require 'process/daemon'
+
+module Celluloid::DNS::ServerPerformanceSpec
+	describe Celluloid::DNS::Server do
+		context 'benchmark' do
+			class MillionServer < Celluloid::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 CelluloidServerDaemon < Process::Daemon
+				IN = Resolv::DNS::Resource::IN
+	
+				def working_directory
+					File.expand_path("../tmp", __FILE__)
+				end
+	
+				def startup
+					puts "Booting celluloid..."
+					Celluloid.boot
+					
+					puts "Starting DNS server..."
+					@server = MillionServer.new(listen: [[:udp, '0.0.0.0', 5300]])
+					
+					@server.async.run
+				end
+				
+				def shutdown
+					@server.terminate
+				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
+				Celluloid.shutdown
+			
+				@servers = []
+				@servers << ["Celluloid::DNS::Server", 5300]
+			
+				@domains = (1..1000).collect do |i|
+					"domain#{i}.local"
+				end
+			
+				CelluloidServerDaemon.start
+			
+				unless Bind9ServerDaemon.named_executable.empty?
+					Bind9ServerDaemon.start
+					@servers << ["Bind9", 5400]
+				end
+				
+				sleep 2
+				
+				Celluloid.boot
+			end
+		
+			after(:all) do
+				CelluloidServerDaemon.stop
+			
+				unless Bind9ServerDaemon.named_executable.empty?
+					Bind9ServerDaemon.stop
+				end
+			end
+		
+			it 'takes time' do
+				# Celluloid.logger.level = Logger::ERROR
+			
+				Benchmark.bm(30) do |x|
+					@servers.each do |name, port|
+						resolver = Celluloid::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
+							
+								futures = @domains.collect{|domain| resolver.future.addresses_for(domain)}
+							
+								futures.collect do |future|
+									expect(future.value).to_not be nil
+								end
+							end
+						end
+					end
+				end
+			end
+		end
+	end
+end

data/spec/celluloid/dns/slow_server_spec.rb

@@ -0,0 +1,91 @@
+#!/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 'celluloid/dns'
+
+module Celluloid::DNS::SlowServerSpec
+	SERVER_PORTS = [[:udp, '127.0.0.1', 5330], [:tcp, '127.0.0.1', 5330]]
+	IN = Resolv::DNS::Resource::IN
+	
+	class TestServer < Celluloid::DNS::Server
+		def process(name, resource_class, transaction)
+			@resolver ||= Celluloid::DNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]])
+			
+			sleep(2) if name.end_with?('.com')
+			
+			transaction.fail!(:NXDomain)
+		end
+	end
+	
+	describe "Celluloid::DNS Slow Server" do
+		before(:all) do
+			@server = TestServer.new(listen: SERVER_PORTS)
+			@server.run
+		end
+		
+		after(:all) do
+			@server.terminate
+		end
+	
+		it "get no answer after 2 seconds" do
+			start_time = Time.now
+		
+			resolver = Celluloid::DNS::Resolver.new(SERVER_PORTS, :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
+		
+			resolver = Celluloid::DNS::Resolver.new(SERVER_PORTS, :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 = Celluloid::DNS::Resolver.new(SERVER_PORTS, :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/celluloid/dns/socket_spec.rb

@@ -0,0 +1,71 @@
+#!/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 'celluloid/dns'
+require 'celluloid/dns/system'
+
+module Celluloid::DNS::SocketSpec
+	IN = Resolv::DNS::Resource::IN
+	
+	class TestServer < Celluloid::DNS::Server
+		def process(name, resource_class, transaction)
+			@resolver ||= Celluloid::DNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]])
+			
+			transaction.passthrough!(@resolver)
+		end
+	end
+	
+	describe Celluloid::DNS::TCPSocketHandler do
+		after(:each) do
+			@server.terminate
+		end
+		
+		it "should create server with existing TCP socket" do
+			socket = TCPServer.new('127.0.0.1', 2002)
+			
+			@server = TestServer.new(listen: [socket])
+			@server.run
+			
+			resolver = Celluloid::DNS::Resolver.new([[:tcp, '127.0.0.1', 2002]])
+			response = resolver.query('google.com')
+			expect(response.class).to be == Celluloid::DNS::Message
+		end
+	end
+	
+	describe Celluloid::DNS::UDPSocketHandler do
+		after(:each) do
+			@server.terminate
+		end
+		
+		it "should create server with existing UDP socket" do
+			socket = UDPSocket.new
+			socket.bind('127.0.0.1', 2002)
+			
+			@server = TestServer.new(listen: [socket])
+			@server.run
+			
+			resolver = Celluloid::DNS::Resolver.new([[:udp, '127.0.0.1', 2002]])
+			response = resolver.query('google.com')
+			expect(response.class).to be == Celluloid::DNS::Message
+		end
+	end
+end

data/spec/celluloid/dns/system_spec.rb

@@ -0,0 +1,60 @@
+#!/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 'celluloid/dns'
+require 'celluloid/dns/system'
+
+module Celluloid::DNS::SystemSpec
+	describe Celluloid::DNS::System do
+		before(:all) do
+			Celluloid.shutdown
+			Celluloid.boot
+		end
+	
+		it "should have at least one namesever" do
+			expect(Celluloid::DNS::System::nameservers.length).to be > 0
+		end
+	
+		it "should respond to query for google.com" do
+			resolver = Celluloid::DNS::Resolver.new(Celluloid::DNS::System::nameservers)
+		
+			response = resolver.query('google.com')
+		
+			expect(response.class).to be == Celluloid::DNS::Message
+			expect(response.rcode).to be == Resolv::DNS::RCode::NoError
+		end
+	end
+
+	describe Celluloid::DNS::System::Hosts do
+		it "should parse the hosts file" do
+			hosts = Celluloid::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