rubydns 1.0.3 → 2.0.0.pre.rc1

data/lib/rubydns/chunked.rb

@@ -1,34 +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.
-
-module RubyDNS
-	# Produces an array of arrays of binary data with each sub-array a maximum of chunk_size bytes.
-	def self.chunked(string, chunk_size = 255)
-		chunks = []
-		
-		offset = 0
-		while offset < string.bytesize
-			chunks << string.byteslice(offset, chunk_size)
-			offset += chunk_size
-		end
-		
-		return chunks
-	end
-end

data/lib/rubydns/extensions/resolv.rb

@@ -1,85 +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 'resolv'
-
-class Resolv
-	class DNS
-		class Message
-			# Merge the given message with this message. A number of heuristics are applied in order to ensure that the result makes sense. For example, If the current message is not recursive but is being merged with a message that was recursive, this bit is maintained. If either message is authoritive, then the result is also authoritive.
-			#
-			# Modifies the current message in place.
-			def merge! (other)
-				# Authoritive Answer
-				@aa = @aa && other.aa
-
-				@question += other.question
-				@answer += other.answer
-				@authority += other.authority
-				@additional += other.additional
-
-				# Recursion Available
-				@ra = @ra || other.ra
-
-				# Result Code (Error Code)
-				@rcode = other.rcode unless other.rcode == 0
-
-				# Recursion Desired
-				@rd = @rd || other.rd
-			end
-		end
-		
-		class OriginError < ArgumentError
-		end
-		
-		class Name
-			def to_s
-				"#{@labels.join('.')}#{@absolute ? '.' : ''}"
-			end
-			
-			def inspect
-				"#<#{self.class}: #{self.to_s}>"
-			end
-			
-			# Return the name, typically absolute, with the specified origin as a suffix. If the origin is nil, don't change the name, but change it to absolute (as specified).
-			def with_origin(origin, absolute = true)
-				return self.class.new(@labels, absolute) if origin == nil
-				
-				origin = Label.split(origin) if String === origin
-				
-				return self.class.new(@labels + origin, absolute)
-			end
-			
-			
-			# Return the name, typically relative, without the specified origin suffix. If the origin is nil, don't change the name, but change it to absolute (as specified).
-			def without_origin(origin, absolute = false)
-				return self.class.new(@labels, absolute) if origin == nil
-				
-				origin = Label.split(origin) if String === origin
-				
-				if @labels.last(origin.length) == origin
-					return self.class.new(@labels.first(@labels.length - origin.length), absolute)
-				else
-					raise OriginError.new("#{self} does not end with #{origin.join('.')}")
-				end
-			end
-		end
-	end
-end

data/lib/rubydns/extensions/string.rb

@@ -1,28 +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_relative '../chunked'
-
-class String
-	# Chunk a string which is required for the TEXT `resource_class`.
-	def chunked(chunk_size = 255)
-		RubyDNS::chunked(self)
-	end
-end

data/lib/rubydns/handler.rb

@@ -1,188 +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_relative 'transport'
-
-module RubyDNS
-	class GenericHandler
-		include Celluloid::IO
-		
-		def initialize(server)
-			@server = server
-			@logger = @server.logger || Celluloid.logger
-		end
-		
-		def error_response(query = nil, code = Resolv::DNS::RCode::ServFail)
-			# Encoding may fail, so we need to handle this particular case:
-			server_failure = Resolv::DNS::Message::new(query ? query.id : 0)
-			
-			server_failure.qr = 1
-			server_failure.opcode = query ? query.opcode : 0
-			server_failure.aa = 1
-			server_failure.rd = 0
-			server_failure.ra = 0
-
-			server_failure.rcode = code
-
-			# We can't do anything at this point...
-			return server_failure
-		end
-		
-		def process_query(data, options)
-			@logger.debug "<> Receiving incoming query (#{data.bytesize} bytes) to #{self.class.name}..."
-			query = nil
-
-			begin
-				query = RubyDNS::decode_message(data)
-				
-				return @server.process_query(query, options)
-			rescue Celluloid::ResumableError
-				# Celluloid terminates tasks, we may be stuck in a task when the server is terminated. We don't want to reply to the client in this case, because the server is being terminated. It might be an option to return a server failure
-				raise
-			rescue StandardError => error
-				@logger.error "<> Error processing request: #{error.inspect}!"
-				RubyDNS::log_exception(@logger, error)
-				
-				return error_response(query)
-			end
-		end
-	end
-	
-	# Handling incoming UDP requests, which are single data packets, and pass them on to the given server.
-	class UDPSocketHandler < GenericHandler
-		include Celluloid::IO
-		
-		def initialize(server, socket)
-			super(server)
-			
-			@socket = socket
-			
-			async.run
-		end
-		
-		def run
-			loop { handle_connection }
-		end
-		
-		def respond(input_data, remote_host, remote_port)
-			options = {peer: remote_host, port: remote_port, proto: :udp}
-			
-			response = process_query(input_data, options)
-			
-			output_data = response.encode
-			
-			@logger.debug "<#{response.id}> Writing #{output_data.bytesize} bytes response to client via UDP..."
-			
-			if output_data.bytesize > UDP_TRUNCATION_SIZE
-				@logger.warn "<#{response.id}>Response via UDP was larger than #{UDP_TRUNCATION_SIZE}!"
-				
-				# Reencode data with truncation flag marked as true:
-				truncation_error = Resolv::DNS::Message.new(response.id)
-				truncation_error.tc = 1
-				
-				output_data = truncation_error.encode
-			end
-			
-			@socket.send(output_data, 0, remote_host, remote_port)
-		rescue IOError => error
-			@logger.warn "<> UDP response failed: #{error.inspect}!"
-		rescue EOFError => error
-			@logger.warn "<> UDP session ended prematurely: #{error.inspect}!"
-		rescue DecodeError
-			@logger.warn "<> Could not decode incoming UDP data!"
-		end
-		
-		def handle_connection
-			# @logger.debug "Waiting for incoming UDP packet #{@socket.inspect}..."
-			
-			input_data, (_, remote_port, remote_host) = @socket.recvfrom(UDP_TRUNCATION_SIZE, 0)
-			
-			async.respond(input_data, remote_host, remote_port)
-		rescue IOError => error
-			@logger.warn "<> UDP connection failed: #{error.inspect}!"
-		rescue EOFError => error
-			@logger.warn "<> UDP session ended prematurely!"
-		end
-	end
-	
-	class UDPHandler < UDPSocketHandler
-		def initialize(server, host, port)
-			family = RubyDNS::address_family(host)
-			socket = UDPSocket.new(family)
-			
-			socket.bind(host, port)
-			
-			super(server, socket)
-		end
-		
-		finalizer :finalize
-		
-		def finalize
-			@socket.close if @socket
-		end
-	end
-	
-	class TCPSocketHandler < GenericHandler
-		def initialize(server, socket)
-			super(server)
-			
-			@socket = socket
-			
-			async.run
-		end
-		
-		def run
-			loop { async.handle_connection @socket.accept }
-		end
-		
-		def handle_connection(socket)
-			_, remote_port, remote_host = socket.peeraddr
-			options = {peer: remote_host, port: remote_port, proto: :tcp}
-			
-			input_data = StreamTransport.read_chunk(socket)
-			
-			response = process_query(input_data, options)
-			
-			length = StreamTransport.write_message(socket, response)
-			
-			@logger.debug "<#{response.id}> Wrote #{length} bytes via TCP..."
-		rescue EOFError => error
-			@logger.warn "<> TCP session ended prematurely!"
-		rescue DecodeError
-			@logger.warn "<> Could not decode incoming TCP data!"
-		ensure
-			socket.close
-		end
-	end
-	
-	class TCPHandler < TCPSocketHandler
-		def initialize(server, host, port)
-			socket = TCPServer.new(host, port)
-			
-			super(server, socket)
-		end
-		
-		finalizer :finalize
-		
-		def finalize
-			@socket.close if @socket
-		end
-	end
-end

data/lib/rubydns/logger.rb

@@ -1,31 +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 'logger'
-
-module RubyDNS
-	# Logs an exception nicely to a standard `Logger`.
-	def self.log_exception(logger, exception)
-		logger.error "#{exception.class}: #{exception.message}"
-		if exception.backtrace
-			Array(exception.backtrace).each { |at| logger.error at }
-		end
-	end
-end

data/lib/rubydns/message.rb

@@ -1,76 +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 'stringio'
-require 'resolv'
-
-require 'base64'
-
-require_relative 'logger'
-require_relative 'extensions/resolv'
-
-module RubyDNS
-	UDP_TRUNCATION_SIZE = 512
-	
-	# The DNS message container.
-	Message = Resolv::DNS::Message
-
-	DecodeError = Resolv::DNS::DecodeError
-
-	@@dump_bad_message = nil
-	
-	# Call this function with a path where bad messages will be saved. Any message that causes an exception to be thrown while decoding the binary will be saved in base64 for later inspection. The log file could grow quickly so be careful - not designed for long term use.
-	def self.log_bad_messages!(log_path)
-		bad_messages_log = Logger.new(log_path, 10, 1024*100)
-		bad_messages_log.level = Logger::DEBUG
-		
-		@dump_bad_message = lambda do |error, data|
-			bad_messages_log.debug("Bad message: #{Base64.encode64(data)}")
-			RubyDNS.log_exception(bad_messages_log, error)
-		end
-	end
-	
-	# Decodes binary data into a {Message}.
-	def self.decode_message(data)
-		# Otherwise the decode process might fail with non-binary data.
-		if data.respond_to? :force_encoding
-			data.force_encoding("BINARY")
-		end
-		
-		begin
-			return Message.decode(data)
-		rescue DecodeError
-			raise
-		rescue StandardError => error
-			new_error = DecodeError.new(error.message)
-			new_error.set_backtrace(error.backtrace)
-			
-			raise new_error
-		end
-		
-	rescue => error
-		# Log the bad messsage if required:
-		if @dump_bad_message
-			@dump_bad_message.call(error, data)
-		end
-		
-		raise
-	end
-end