Class: RubyDNS::GenericHandler

Inherits:
Object
  • Object
show all
Includes:
Celluloid::IO
Defined in:
lib/rubydns/handler.rb

Direct Known Subclasses

TCPSocketHandler, UDPSocketHandler

Instance Method Summary collapse

Constructor Details

#initialize(server) ⇒ GenericHandler

Returns a new instance of GenericHandler.



27
28
29
30
# File 'lib/rubydns/handler.rb', line 27

def initialize(server)
	@server = server
	@logger = @server.logger || Celluloid.logger
end

Instance Method Details

#error_response(query = nil, code = Resolv::DNS::RCode::ServFail) ⇒ Object



32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
# File 'lib/rubydns/handler.rb', line 32

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

#process_query(data, options) ⇒ Object



48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# File 'lib/rubydns/handler.rb', line 48

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