celluloid-dns 0.0.1 → 0.17.3

data/lib/celluloid/dns/transport.rb

@@ -0,0 +1,75 @@
+# 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 'stringio'
+require 'ipaddr'
+
+require_relative 'message'
+
+module Celluloid::DNS
+	def self.address_family(host)
+		return IPAddr.new(host).family
+	end
+	
+	# A helper class for processing incoming network data.
+	class BinaryStringIO < StringIO
+		def initialize
+			super
+		
+			set_encoding("BINARY")
+		end
+	end
+	
+	module StreamTransport
+		def self.read_chunk(socket)
+			# The data buffer:
+			buffer = BinaryStringIO.new
+			
+			# First we need to read in the length of the packet
+			while buffer.size < 2
+				buffer.write socket.readpartial(1)
+			end
+			
+			# Read in the length, the first two bytes:
+			length = buffer.string.byteslice(0, 2).unpack('n')[0]
+			
+			# Read data until we have the amount specified:
+			while (buffer.size - 2) < length
+				required = (2 + length) - buffer.size
+				
+				# Read precisely the required amount:
+				buffer.write socket.readpartial(required)
+			end
+			
+			return buffer.string.byteslice(2, length)
+		end
+		
+		def self.write_message(socket, message)
+			write_chunk(socket, message.encode)
+		end
+		
+		def self.write_chunk(socket, output_data)
+			socket.write([output_data.bytesize].pack('n'))
+			socket.write(output_data)
+			
+			return output_data.bytesize
+		end
+	end
+end

data/lib/celluloid/dns/version.rb

@@ -1,5 +1,25 @@
+# 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.
+
 module Celluloid
-  module DNS
-    VERSION = "0.0.1"
-  end
+	module DNS
+		VERSION = '0.17.3'
+	end
 end

data/spec/celluloid/dns/celluloid_bug_spec.rb

@@ -0,0 +1,92 @@
+#!/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::CelluloidBugSpec
+	describe Celluloid::DNS::Resolver do
+		context 'benchmark' do
+			domains = %W(
+				Facebook.com
+				Twitter.com
+				Google.com
+				Youtube.com
+				Wordpress.org
+				Adobe.com
+				Blogspot.com
+				Wikipedia.org
+				Linkedin.com
+				Wordpress.com
+				Yahoo.com
+				Amazon.com
+				Flickr.com
+				Pinterest.com
+				Tumblr.com
+				W3.org
+				Apple.com
+				Myspace.com
+				Vimeo.com
+				Microsoft.com
+				Youtu.be
+				Qq.com
+				Digg.com
+				Baidu.com
+				Stumbleupon.com
+				Addthis.com
+				Statcounter.com
+				Feedburner.com
+				TradeMe.co.nz
+				Delicious.com
+				Nytimes.com
+				Reddit.com
+				Weebly.com
+				Bbc.co.uk
+				Blogger.com
+				Msn.com
+				Macromedia.com
+				Goo.gl
+				Instagram.com
+				Gov.uk
+				Icio.us
+				Yandex.ru
+				Cnn.com
+				Webs.com
+				Google.de
+				T.co
+				Livejournal.com
+				Imdb.com
+				Mail.ru
+				Jimdo.com
+			)
+			
+			it 'should resolve domains using futures' do
+				# Celluloid.logger.level = Logger::ERROR
+				
+				resolver = Celluloid::DNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]], timeout: 1.0)
+				
+				futures = domains.map { |domain| resolver.future.addresses_for(domain) }
+				
+				futures.map { |future| future.value }
+			end
+		end
+	end
+end

data/spec/celluloid/dns/hosts.txt

@@ -0,0 +1,2 @@
+# A testing host:
+1.2.3.4 testing apples

data/spec/celluloid/dns/ipv6_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 'celluloid/dns'
+require 'celluloid/dns/system'
+
+module Celluloid::DNS::IPv6Spec
+	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
+		before(:all) do
+			@server = TestServer.new(listen: [[:tcp, '::', 2004]])
+			@server.run
+		end
+		
+		after(:all) do
+			@server.terminate
+		end
+		
+		it "should connect to the server using TCP via IPv6" do
+			resolver = Celluloid::DNS::Resolver.new([[:tcp, '::1', 2004]])
+			response = resolver.query('google.com')
+			expect(response.class).to be == Celluloid::DNS::Message
+		end
+	end
+	
+	describe Celluloid::DNS::UDPSocketHandler do
+		before(:all) do
+			@server = TestServer.new(listen: [[:udp, '::', 2006]])
+			@server.run
+		end
+		
+		after(:all) do
+			@server.terminate
+		end
+		
+		it "should connect to the server using UDP via IPv6" do
+			resolver = Celluloid::DNS::Resolver.new([[:udp, '::1', 2006]])
+			response = resolver.query('google.com')
+			expect(response.class).to be == Celluloid::DNS::Message
+		end
+	end
+end

data/spec/celluloid/dns/message_spec.rb

@@ -0,0 +1,56 @@
+#!/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 'base64'
+
+module Celluloid::DNS::MessageSpec
+	describe Celluloid::DNS::Message do
+		it "should be decoded correctly" do
+			data = Base64.decode64(<<-EOF)
+			HQCBgAABAAgAAAABA3d3dwV5YWhvbwNjb20AAAEAAcAMAAUAAQAAASwADwZm
+			ZC1mcDMDd2cxAWLAEMArAAUAAQAAASwACQZkcy1mcDPAMsBGAAUAAQAAADwA
+			FQ5kcy1hbnktZnAzLWxmYgN3YTHANsBbAAUAAQAAASwAEg9kcy1hbnktZnAz
+			LXJlYWzAasB8AAEAAQAAADwABGKK/B7AfAABAAEAAAA8AARii7SVwHwAAQAB
+			AAAAPAAEYou3GMB8AAEAAQAAADwABGKK/W0AACkQAAAAAAAAAA==
+			EOF
+	
+			message = Celluloid::DNS::decode_message(data)
+			expect(message.class).to be == Celluloid::DNS::Message
+			expect(message.id).to be == 0x1d00
+	
+			expect(message.question.count).to be == 1
+			expect(message.answer.count).to be == 8
+			expect(message.authority.count).to be == 0
+			expect(message.additional.count).to be == 1
+		end
+
+		it "should fail to decode due to bad AAAA length" do
+			data = Base64.decode64(<<-EOF)
+			6p6BgAABAAEAAAABCGJhaWNhaWNuA2NvbQAAHAABwAwAHAABAAABHgAEMhd7
+			dwAAKRAAAAAAAAAA
+			EOF
+	
+			expect{Celluloid::DNS::decode_message(data)}.to raise_error(Celluloid::DNS::DecodeError)
+		end
+	end
+end

data/spec/celluloid/dns/origin_spec.rb

@@ -0,0 +1,106 @@
+#!/usr/bin/env rspec
+
+# 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'
+
+module Celluloid::DNS::OriginSpec
+	describe Celluloid::DNS::Resolver do
+		it "should generate fully qualified domain name with specified origin" do
+			resolver = Celluloid::DNS::Resolver.new([], origin: "foo.bar.")
+			
+			fully_qualified_name = resolver.fully_qualified_name("baz")
+			
+			expect(fully_qualified_name).to be_absolute
+			expect(fully_qualified_name.to_s).to be == "baz.foo.bar."
+		end
+	end
+	
+	describe Resolv::DNS::Name do
+		let(:name) {Resolv::DNS::Name.create("foo.bar")}
+		
+		it "should be relative" do
+			expect(name).to_not be_absolute
+			expect(name.to_s).to be == "foo.bar"
+		end
+		
+		it "should add the specified origin" do
+			fully_qualified_name = name.with_origin("org")
+			
+			expect(fully_qualified_name.to_a.size).to be 3
+			expect(fully_qualified_name).to be_absolute
+			expect(fully_qualified_name.to_s).to be == "foo.bar.org."
+		end
+		
+		it "should handle nil origin as absolute" do
+			fully_qualified_name = name.with_origin(nil)
+			
+			expect(fully_qualified_name.to_a.size).to be 2
+			expect(fully_qualified_name).to be_absolute
+			expect(fully_qualified_name.to_s).to be == "foo.bar."
+		end
+		
+		it "should handle empty origin as absolute" do
+			fully_qualified_name = name.with_origin('')
+			
+			expect(fully_qualified_name.to_a.size).to be 2
+			expect(fully_qualified_name).to be_absolute
+			expect(fully_qualified_name.to_s).to be == "foo.bar."
+		end
+	end
+	
+	describe Resolv::DNS::Name do
+		let(:name) {Resolv::DNS::Name.create("foo.bar.")}
+		
+		it "should be absolute" do
+			expect(name).to be_absolute
+			expect(name.to_s).to be == "foo.bar."
+		end
+		
+		it "should remove the specified origin" do
+			relative_name = name.without_origin("bar")
+			
+			expect(relative_name.to_a.size).to be 1
+			expect(relative_name).to_not be_absolute
+			expect(relative_name.to_s).to be == "foo"
+		end
+		
+		it "should not remove nil origin but become relative" do
+			relative_name = name.without_origin(nil)
+			
+			expect(relative_name.to_a.size).to be 2
+			expect(relative_name).to_not be_absolute
+			expect(relative_name.to_s).to be == "foo.bar"
+		end
+		
+		it "should not remove empty string origin but become relative" do
+			relative_name = name.without_origin('')
+			
+			expect(relative_name.to_a.size).to be 2
+			expect(relative_name).to_not be_absolute
+			expect(relative_name.to_s).to be == "foo.bar"
+		end
+		
+		it "should not raise an exception when origin isn't valid" do
+			expect{name.without_origin('bob')}.to raise_exception(Resolv::DNS::OriginError)
+		end
+	end
+end

data/spec/celluloid/dns/replace_spec.rb

@@ -0,0 +1,42 @@
+#!/usr/bin/env rspec
+
+# Copyright, 2015, 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/replace'
+
+module Celluloid::DNS::ReplaceSpec
+	describe Celluloid::DNS::Replace do
+		let(:default_resolver) {Celluloid::DNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]])}
+		
+		after(:all) do
+			Celluloid::DNS::Replace.resolver = nil
+		end
+		
+		it "should replace TCPSocket hostname lookup" do
+			Celluloid::DNS::Replace.resolver = default_resolver
+			
+			expect(default_resolver).to receive(:addresses_for).with('www.google.com').and_call_original
+			
+			TCPSocket.new('www.google.com', 80)
+		end
+	end
+end