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