diff options
Diffstat (limited to 'lib/dns/message.go')
| -rw-r--r-- | lib/dns/message.go | 34 |
1 files changed, 6 insertions, 28 deletions
diff --git a/lib/dns/message.go b/lib/dns/message.go index 7a43a7c8..a35473b0 100644 --- a/lib/dns/message.go +++ b/lib/dns/message.go @@ -730,37 +730,15 @@ func (msg *Message) packAAAA(rr *ResourceRecord) { } } +// packOPT pack the RDLEN and RDATA of OPT record. func (msg *Message) packOPT(rr *ResourceRecord) { - var ( - rrOPT, _ = rr.Value.(*RDataOPT) - off = uint(len(msg.packet)) - - n uint16 - ) - - // Reserve two octets for rdlength. - msg.packet = libbytes.AppendUint16(msg.packet, 0) + var rrOPT *RDataOPT - if rrOPT.Length == 0 { - return - } - - // Pack OPT rdata - msg.packet = libbytes.AppendUint16(msg.packet, rrOPT.Code) - - // Values of less than 512 bytes MUST be treated as equal to 512 - // bytes (RFC6891 P11). - if rrOPT.Length < 512 { - msg.packet = libbytes.AppendUint16(msg.packet, 512) - } else { - msg.packet = libbytes.AppendUint16(msg.packet, rrOPT.Length) - } + rrOPT, _ = rr.Value.(*RDataOPT) - msg.packet = append(msg.packet, rrOPT.Data[:rrOPT.Length]...) - - // Write rdlength. - n = 4 + rrOPT.Length - libbytes.WriteUint16(msg.packet, off, n) + var rdata = rrOPT.pack() + msg.packet = libbytes.AppendUint16(msg.packet, uint16(len(rdata))) + msg.packet = append(msg.packet, rdata...) } func (msg *Message) packSVCB(rr *ResourceRecord) { |