metasploit-framework/lib/net/dns/rr/soa.rb

86 lines
2.6 KiB
Ruby

# -*- coding: binary -*-
##
#
# Net::DNS::RR::SOA
#
# $Id: SOA.rb,v 1.4 2006/07/28 07:33:36 bluemonk Exp $
#
##
module Net
module DNS
class RR
#------------------------------------------------------------
# RR type SOA
#------------------------------------------------------------
class SOA < RR
attr_reader :mname, :rname, :serial, :refresh, :retry, :expire, :minimum
private
def build_pack
@soa_pack = pack_name(@mname)
@soa_pack += pack_name(@rname)
@soa_pack += [@serial,@refresh,@retry,@expire,@minimum].pack("N5")
end
def set_type
@type = Net::DNS::RR::Types.new("SOA")
end
def get_data
@soa_pack
end
def get_inspect
"#@mname #@rname #@serial #@refresh #@retry #@expire #@minimum"
end
def subclass_new_from_hash(args)
if args.has_key? :rdata
subclass_new_from_string(args[:rdata])
else
[:mname,:rname,:serial,:refresh,:retry,:expire,:minimum].each do |key|
raise RRArgumentError, "Missing field :#{key}" unless args.has_key? key
end
@mname = args[:mname] if valid? args[:mname]
@rname = args[:rname] if valid? args[:rname]
@serial = args[:serial] if number? args[:serial]
@refresh = args[:refresh] if number? args[:refresh]
@retry = args[:retry] if number? args[:retry]
@expire = args[:expire] if number? args[:expire]
@minimum = args[:minimum] if number? args[:minimum]
end
end
def number?(num)
if num.kind_of? Integer and num > 0
true
else
raise RRArgumentError, "Wrong format field: #{num} not a number or less than zero"
end
end
def subclass_new_from_string(str)
mname,rname,serial,refresh,ret,expire,minimum = str.strip.split(" ")
@mname = mname if valid? mname
@rname = rname if valid? rname
@serial,@refresh,@retry,@expire,@minimum = [serial,refresh,ret,expire,minimum].collect do |i|
i.to_i if valid? i.to_i
end
end
def subclass_new_from_binary(data,offset)
@mname,offset = dn_expand(data,offset)
@rname,offset = dn_expand(data,offset)
@serial,@refresh,@retry,@expire,@minimum = data.unpack("@#{offset} N5")
return offset + 5*Net::DNS::INT32SZ
end
end # class SOA
end # class RR
end # module DNS
end # module Net