Module: RubyDNS::TCPHandler

Defined in:
lib/rubydns/handler.rb

Instance Method Summary collapse

Instance Method Details

#initialize(server) ⇒ Object 



44
45
46
47
48
49
 
# File 'lib/rubydns/handler.rb', line 44

def initialize(server)
	@server = server
	@buffer = nil
	@length = nil
	@processed = 0
end

#receive_data(data) ⇒ Object 



51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/rubydns/handler.rb', line 51

def receive_data(data)
	@buffer ||= StringIO.new
	@buffer.write(data)
	
	# Message includes a 16-bit length field
	if @length == nil
		if (@buffer.size - @processed) < 2
			raise LengthError.new("Malformed message smaller than two bytes received")
		end
		
		@length = @buffer.string[@processed, 2].unpack('n')[0]
		@processed += 2
	end
	
	
	if (@buffer.size - @processed) >= @length
		data = @buffer.string[@processed, @length]
		
		UDPHandler.process(@server, data) do |result|
			self.send_data([result.size].pack('n'))
			self.send_data(result)
		end
		
		@processed += @length
		@length = nil
	end
end

#unbind ⇒ Object 



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

def unbind
	if @processed != @buffer.size
		raise LengthError.new("Unprocessed data remaining (#{@buffer.size - @processed} bytes unprocessed)")
	end
end