async-dns 0.10.0

data/lib/async/dns/transport.rb

@@ -0,0 +1,78 @@
+# 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 Async::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.read(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.read(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)
+			size_data = [output_data.bytesize].pack('n')
+			
+			# TODO: Validate/check for data written correctly
+			count = socket.write(size_data)
+			count = socket.write(output_data)
+			
+			return output_data.bytesize
+		end
+	end
+end

data/lib/async/dns/version.rb

@@ -0,0 +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 Async
+	module DNS
+		VERSION = '0.10.0'
+	end
+end

data/spec/async/dns/handler_spec.rb

@@ -0,0 +1,58 @@
+# Copyright, 2017, 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'
+
+describe Async::DNS::TCPServerHandler do
+	let(:server) {Async::DNS::Server.new}
+	let(:host) {'127.0.0.1'}
+	let(:port) {6665}
+	
+	subject {described_class.new(server, host, port)}
+	
+	it "can rebind port" do
+		2.times do
+			socket = subject.send(:make_socket)
+			expect(socket).to_not be_closed
+			
+			socket.close
+			expect(socket).to be_closed
+		end
+	end
+end
+
+describe Async::DNS::UDPServerHandler do
+	let(:server) {Async::DNS::Server.new}
+	let(:host) {'127.0.0.1'}
+	let(:port) {6665}
+	
+	subject {described_class.new(server, host, port)}
+	
+	it "can rebind port" do
+		2.times do
+			socket = subject.send(:make_socket)
+			expect(socket).to_not be_closed
+			
+			socket.close
+			expect(socket).to be_closed
+		end
+	end
+end

data/spec/async/dns/hosts.txt

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

data/spec/async/dns/ipv6_spec.rb

@@ -0,0 +1,78 @@
+#!/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::IPv6Spec
+	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
+		let(:server_interfaces) {[[:tcp, '::', 2004]]}
+		let(:server) {TestServer.new(listen: server_interfaces)}
+		
+		include_context "reactor"
+		
+		it "should connect to the server using TCP via IPv6" do
+			task = server.run
+			
+			resolver = Async::DNS::Resolver.new([[:tcp, '::1', 2004]])
+			
+			response = resolver.query('google.com')
+			
+			expect(response.class).to be == Async::DNS::Message
+			expect(response.rcode).to be == 0
+			expect(response.answer).to_not be_empty
+			
+			task.stop
+		end
+	end
+	
+	describe Async::DNS::UDPSocketHandler do
+		let(:server_interfaces) {[[:udp, '::', 2006]]}
+		let(:server) {TestServer.new(listen: server_interfaces)}
+		
+		include_context "reactor"
+		
+		it "should connect to the server using UDP via IPv6" do
+			task = server.run
+			
+			resolver = Async::DNS::Resolver.new([[:udp, '::1', 2006]])
+			
+			response = resolver.query('google.com')
+			
+			expect(response.class).to be == Async::DNS::Message
+			expect(response.rcode).to be == 0
+			expect(response.answer).to_not be_empty
+			
+			task.stop
+		end
+	end
+end

data/spec/async/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 'async/dns'
+require 'base64'
+
+module Async::DNS::MessageSpec
+	describe Async::DNS::Message do
+		it "should be decoded correctly" do
+			data = Base64.decode64(<<-EOF)
+			HQCBgAABAAgAAAABA3d3dwV5YWhvbwNjb20AAAEAAcAMAAUAAQAAASwADwZm
+			ZC1mcDMDd2cxAWLAEMArAAUAAQAAASwACQZkcy1mcDPAMsBGAAUAAQAAADwA
+			FQ5kcy1hbnktZnAzLWxmYgN3YTHANsBbAAUAAQAAASwAEg9kcy1hbnktZnAz
+			LXJlYWzAasB8AAEAAQAAADwABGKK/B7AfAABAAEAAAA8AARii7SVwHwAAQAB
+			AAAAPAAEYou3GMB8AAEAAQAAADwABGKK/W0AACkQAAAAAAAAAA==
+			EOF
+	
+			message = Async::DNS::decode_message(data)
+			expect(message.class).to be == Async::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{Async::DNS::decode_message(data)}.to raise_error(Async::DNS::DecodeError)
+		end
+	end
+end

data/spec/async/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 'async/dns'
+
+module Async::DNS::OriginSpec
+	describe Async::DNS::Resolver do
+		it "should generate fully qualified domain name with specified origin" do
+			resolver = Async::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/async/dns/replace_spec.rb

@@ -0,0 +1,44 @@
+#!/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 'async/dns'
+require 'async/dns/replace'
+
+module Async::DNS::ReplaceSpec
+	describe Async::DNS::Replace do
+		include_context "reactor"
+		
+		let(:default_resolver) {Async::DNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]])}
+		
+		after(:all) do
+			Async::DNS::Replace.resolver = nil
+		end
+		
+		it "should replace TCPSocket hostname lookup" do
+			Async::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