Class: RubyDNS::UDPSocketHandler

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

Overview

Handling incoming UDP requests, which are single data packets, and pass them on to the given server.

Direct Known Subclasses

UDPHandler

Instance Method Summary collapse

Methods inherited from GenericHandler

#error_response, #process_query

Constructor Details

#initialize(server, socket) ⇒ UDPSocketHandler

Returns a new instance of UDPSocketHandler.



72
73
74
75
76
77
78
 
# File 'lib/rubydns/handler.rb', line 72

def initialize(server, socket)
  super(server)
  
  @socket = socket
  
  async.run
end

Instance Method Details

#handle_connection ⇒ Object 



112
113
114
115
116
117
118
119
120
121
122
 
# File 'lib/rubydns/handler.rb', line 112

def handle_connection
  # @logger.debug "Waiting for incoming UDP packet #{@socket.inspect}..."
  
  input_data, (_, remote_port, remote_host) = @socket.recvfrom(UDP_TRUNCATION_SIZE)
  
  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

#respond(input_data, remote_host, remote_port) ⇒ Object 



84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
 
# File 'lib/rubydns/handler.rb', line 84

def respond(input_data, remote_host, remote_port)
  options = {peer: remote_host}
  
  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

#run ⇒ Object 



80
81
82
 
# File 'lib/rubydns/handler.rb', line 80

def run
  loop { handle_connection }
end