1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 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 78 79 80 81 82 83 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 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213
|
#--
#Copyright 2007 Nominet UK
#
#Licensed under the Apache License, Version 2.0 (the "License");
#you may not use this file except in compliance with the License.
#You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
#Unless required by applicable law or agreed to in writing, software
#distributed under the License is distributed on an "AS IS" BASIS,
#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#See the License for the specific language governing permissions and
#limitations under the License.
#++
module Dnsruby
class RR
#Class for EDNS pseudo resource record OPT.
#This class is effectively internal to Dnsruby
#See RFC 2671, RFC 2435 Section 3
# @TODO@ Extended labels RFC2671 section 3
class OPT < RR #:nodoc: all
ClassValue = nil #:nodoc: all
TypeValue = Types::OPT #:nodoc: all
DO_BIT = 0x8000
# @TODO@ Add BADVERS to an XRCode CodeMapper object
#Can be called with up to 3 arguments, none of which must be present
#* OPT.new()
#* OPT.new(size)
#* OPT.new(size,flags)
#* OPT.new(size,flags,options)
def initialize(*args)
@type = Types.new('OPT')
@ttl = nil
@options=nil
if (args.length > 0)
self.payloadsize=(args[0])
if (args.length > 1)
self.flags=(args[1])
if (args.length > 2)
self.options=(args[2])
else
self.options=nil
end
else
self.flags=0
end
else
self.payloadsize=0
end
end
# From RFC 2671 :
# 4.3. The fixed part of an OPT RR is structured as follows:
#
# Field Name Field Type Description
# ------------------------------------------------------
# NAME domain name empty (root domain)
# TYPE u_int16_t OPT
# CLASS u_int16_t sender's UDP payload size
# TTL u_int32_t extended RCODE and flags
# RDLEN u_int16_t describes RDATA
# RDATA octet stream {attribute,value} pairs
#4.6. The extended RCODE and flags (which OPT stores in the RR TTL field)
#are structured as follows:
#
# +0 (MSB) +1 (LSB)
# +---+---+---+---+---+---+---+---+---+---+---+---+---+---+---+---+
# 0: | EXTENDED-RCODE | VERSION |
# +---+---+---+---+---+---+---+---+---+---+---+---+---+---+---+---+
# 2: | Z |
# +---+---+---+---+---+---+---+---+---+---+---+---+---+---+---+---+
#
# EXTENDED-RCODE Forms upper 8 bits of extended 12-bit RCODE. Note
# that EXTENDED-RCODE value "0" indicates that an
# unextended RCODE is in use (values "0" through "15").
#
# VERSION Indicates the implementation level of whoever sets
# it. Full conformance with this specification is
# indicated by version "0."
def flags_from_ttl
if (@ttl)
return [@ttl].pack("N")
else
return [0].pack("N")
end
end
def xrcode
return ExtendedRCode.new(flags_from_ttl[0, 1].unpack("C")[0])
end
def xrcode=(c)
code = ExtendedRCode.new(c)
@ttl = (code.code << 24) + (version() << 16) + flags()
end
def version
return flags_from_ttl[1, 1].unpack("C")[0]
end
def version=(code)
@ttl = (xrcode().code << 24) + (code << 16) + flags()
end
def flags
return flags_from_ttl[2, 2].unpack("n")[0]
end
def flags=(code)
set_flags(code)
end
def set_flags(code) # Should always be zero
@ttl = (xrcode().code << 24) + (version() << 16) + code
end
def dnssec_ok
return ((flags() & DO_BIT) == DO_BIT)
end
def dnssec_ok=(on)
if (on)
set_flags(flags() | DO_BIT)
else
set_flags(flags() & (~DO_BIT))
end
end
def payloadsize
return @klass.code
end
def payloadsize=(size)
self.klass=size
end
def options(args)
if (args==nil)
return @options
elsif args.kind_of?Fixnum
# return list of options with that code
ret = []
@options.each do |option|
if (option.code == args)
ret.push(option)
end
end
return ret
end
end
def options=(options)
@options = options
end
def from_data(data)
@options = data
end
def from_string(input)
raise NotImplementedError
end
def to_s
ret = "OPT pseudo-record : payloadsize #{payloadsize}, xrcode #{xrcode.code}, version #{version}, flags #{flags}"
if @options
@options.each do |opt|
ret = ret + " " + opt.to_s
end
end
ret = ret + "\n"
return ret
end
def encode_rdata(msg, canonical=false)
if (@options)
@options.each do |opt|
msg.put_pack('n', opt.code)
msg.put_pack('n', opt.data.length)
msg.put_bytes(opt.data)
end
end
end
def self.decode_rdata(msg)#:nodoc: all
if (msg.has_remaining)
options = []
while (msg.has_remaining) do
code = msg.get_unpack('n')[0]
len = msg.get_unpack('n')[0]
data = msg.get_bytes(len)
options.push(Option.new(code, data))
end
end
return self.new([options])
end
class Option
attr_accessor :code, :data
def initialize(code, data)
@code = code
@data = data
end
end
end
end
end
|