github.com/Andyfoo/golang/x/net@v0.0.0-20190901054642-57c1bf301704/dns/dnsmessage/message.go (about) 1 // Copyright 2009 The Go Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 // Package dnsmessage provides a mostly RFC 1035 compliant implementation of 6 // DNS message packing and unpacking. 7 // 8 // The package also supports messages with Extension Mechanisms for DNS 9 // (EDNS(0)) as defined in RFC 6891. 10 // 11 // This implementation is designed to minimize heap allocations and avoid 12 // unnecessary packing and unpacking as much as possible. 13 package dnsmessage 14 15 import ( 16 "errors" 17 ) 18 19 // Message formats 20 21 // A Type is a type of DNS request and response. 22 type Type uint16 23 24 const ( 25 // ResourceHeader.Type and Question.Type 26 TypeA Type = 1 27 TypeNS Type = 2 28 TypeCNAME Type = 5 29 TypeSOA Type = 6 30 TypePTR Type = 12 31 TypeMX Type = 15 32 TypeTXT Type = 16 33 TypeAAAA Type = 28 34 TypeSRV Type = 33 35 TypeOPT Type = 41 36 37 // Question.Type 38 TypeWKS Type = 11 39 TypeHINFO Type = 13 40 TypeMINFO Type = 14 41 TypeAXFR Type = 252 42 TypeALL Type = 255 43 ) 44 45 var typeNames = map[Type]string{ 46 TypeA: "TypeA", 47 TypeNS: "TypeNS", 48 TypeCNAME: "TypeCNAME", 49 TypeSOA: "TypeSOA", 50 TypePTR: "TypePTR", 51 TypeMX: "TypeMX", 52 TypeTXT: "TypeTXT", 53 TypeAAAA: "TypeAAAA", 54 TypeSRV: "TypeSRV", 55 TypeOPT: "TypeOPT", 56 TypeWKS: "TypeWKS", 57 TypeHINFO: "TypeHINFO", 58 TypeMINFO: "TypeMINFO", 59 TypeAXFR: "TypeAXFR", 60 TypeALL: "TypeALL", 61 } 62 63 // String implements fmt.Stringer.String. 64 func (t Type) String() string { 65 if n, ok := typeNames[t]; ok { 66 return n 67 } 68 return printUint16(uint16(t)) 69 } 70 71 // GoString implements fmt.GoStringer.GoString. 72 func (t Type) GoString() string { 73 if n, ok := typeNames[t]; ok { 74 return "dnsmessage." + n 75 } 76 return printUint16(uint16(t)) 77 } 78 79 // A Class is a type of network. 80 type Class uint16 81 82 const ( 83 // ResourceHeader.Class and Question.Class 84 ClassINET Class = 1 85 ClassCSNET Class = 2 86 ClassCHAOS Class = 3 87 ClassHESIOD Class = 4 88 89 // Question.Class 90 ClassANY Class = 255 91 ) 92 93 var classNames = map[Class]string{ 94 ClassINET: "ClassINET", 95 ClassCSNET: "ClassCSNET", 96 ClassCHAOS: "ClassCHAOS", 97 ClassHESIOD: "ClassHESIOD", 98 ClassANY: "ClassANY", 99 } 100 101 // String implements fmt.Stringer.String. 102 func (c Class) String() string { 103 if n, ok := classNames[c]; ok { 104 return n 105 } 106 return printUint16(uint16(c)) 107 } 108 109 // GoString implements fmt.GoStringer.GoString. 110 func (c Class) GoString() string { 111 if n, ok := classNames[c]; ok { 112 return "dnsmessage." + n 113 } 114 return printUint16(uint16(c)) 115 } 116 117 // An OpCode is a DNS operation code. 118 type OpCode uint16 119 120 // GoString implements fmt.GoStringer.GoString. 121 func (o OpCode) GoString() string { 122 return printUint16(uint16(o)) 123 } 124 125 // An RCode is a DNS response status code. 126 type RCode uint16 127 128 const ( 129 // Message.Rcode 130 RCodeSuccess RCode = 0 131 RCodeFormatError RCode = 1 132 RCodeServerFailure RCode = 2 133 RCodeNameError RCode = 3 134 RCodeNotImplemented RCode = 4 135 RCodeRefused RCode = 5 136 ) 137 138 var rCodeNames = map[RCode]string{ 139 RCodeSuccess: "RCodeSuccess", 140 RCodeFormatError: "RCodeFormatError", 141 RCodeServerFailure: "RCodeServerFailure", 142 RCodeNameError: "RCodeNameError", 143 RCodeNotImplemented: "RCodeNotImplemented", 144 RCodeRefused: "RCodeRefused", 145 } 146 147 // String implements fmt.Stringer.String. 148 func (r RCode) String() string { 149 if n, ok := rCodeNames[r]; ok { 150 return n 151 } 152 return printUint16(uint16(r)) 153 } 154 155 // GoString implements fmt.GoStringer.GoString. 156 func (r RCode) GoString() string { 157 if n, ok := rCodeNames[r]; ok { 158 return "dnsmessage." + n 159 } 160 return printUint16(uint16(r)) 161 } 162 163 func printPaddedUint8(i uint8) string { 164 b := byte(i) 165 return string([]byte{ 166 b/100 + '0', 167 b/10%10 + '0', 168 b%10 + '0', 169 }) 170 } 171 172 func printUint8Bytes(buf []byte, i uint8) []byte { 173 b := byte(i) 174 if i >= 100 { 175 buf = append(buf, b/100+'0') 176 } 177 if i >= 10 { 178 buf = append(buf, b/10%10+'0') 179 } 180 return append(buf, b%10+'0') 181 } 182 183 func printByteSlice(b []byte) string { 184 if len(b) == 0 { 185 return "" 186 } 187 buf := make([]byte, 0, 5*len(b)) 188 buf = printUint8Bytes(buf, uint8(b[0])) 189 for _, n := range b[1:] { 190 buf = append(buf, ',', ' ') 191 buf = printUint8Bytes(buf, uint8(n)) 192 } 193 return string(buf) 194 } 195 196 const hexDigits = "0123456789abcdef" 197 198 func printString(str []byte) string { 199 buf := make([]byte, 0, len(str)) 200 for i := 0; i < len(str); i++ { 201 c := str[i] 202 if c == '.' || c == '-' || c == ' ' || 203 'A' <= c && c <= 'Z' || 204 'a' <= c && c <= 'z' || 205 '0' <= c && c <= '9' { 206 buf = append(buf, c) 207 continue 208 } 209 210 upper := c >> 4 211 lower := (c << 4) >> 4 212 buf = append( 213 buf, 214 '\\', 215 'x', 216 hexDigits[upper], 217 hexDigits[lower], 218 ) 219 } 220 return string(buf) 221 } 222 223 func printUint16(i uint16) string { 224 return printUint32(uint32(i)) 225 } 226 227 func printUint32(i uint32) string { 228 // Max value is 4294967295. 229 buf := make([]byte, 10) 230 for b, d := buf, uint32(1000000000); d > 0; d /= 10 { 231 b[0] = byte(i/d%10 + '0') 232 if b[0] == '0' && len(b) == len(buf) && len(buf) > 1 { 233 buf = buf[1:] 234 } 235 b = b[1:] 236 i %= d 237 } 238 return string(buf) 239 } 240 241 func printBool(b bool) string { 242 if b { 243 return "true" 244 } 245 return "false" 246 } 247 248 var ( 249 // ErrNotStarted indicates that the prerequisite information isn't 250 // available yet because the previous records haven't been appropriately 251 // parsed, skipped or finished. 252 ErrNotStarted = errors.New("parsing/packing of this type isn't available yet") 253 254 // ErrSectionDone indicated that all records in the section have been 255 // parsed or finished. 256 ErrSectionDone = errors.New("parsing/packing of this section has completed") 257 258 errBaseLen = errors.New("insufficient data for base length type") 259 errCalcLen = errors.New("insufficient data for calculated length type") 260 errReserved = errors.New("segment prefix is reserved") 261 errTooManyPtr = errors.New("too many pointers (>10)") 262 errInvalidPtr = errors.New("invalid pointer") 263 errNilResouceBody = errors.New("nil resource body") 264 errResourceLen = errors.New("insufficient data for resource body length") 265 errSegTooLong = errors.New("segment length too long") 266 errZeroSegLen = errors.New("zero length segment") 267 errResTooLong = errors.New("resource length too long") 268 errTooManyQuestions = errors.New("too many Questions to pack (>65535)") 269 errTooManyAnswers = errors.New("too many Answers to pack (>65535)") 270 errTooManyAuthorities = errors.New("too many Authorities to pack (>65535)") 271 errTooManyAdditionals = errors.New("too many Additionals to pack (>65535)") 272 errNonCanonicalName = errors.New("name is not in canonical format (it must end with a .)") 273 errStringTooLong = errors.New("character string exceeds maximum length (255)") 274 errCompressedSRV = errors.New("compressed name in SRV resource data") 275 ) 276 277 // Internal constants. 278 const ( 279 // packStartingCap is the default initial buffer size allocated during 280 // packing. 281 // 282 // The starting capacity doesn't matter too much, but most DNS responses 283 // Will be <= 512 bytes as it is the limit for DNS over UDP. 284 packStartingCap = 512 285 286 // uint16Len is the length (in bytes) of a uint16. 287 uint16Len = 2 288 289 // uint32Len is the length (in bytes) of a uint32. 290 uint32Len = 4 291 292 // headerLen is the length (in bytes) of a DNS header. 293 // 294 // A header is comprised of 6 uint16s and no padding. 295 headerLen = 6 * uint16Len 296 ) 297 298 type nestedError struct { 299 // s is the current level's error message. 300 s string 301 302 // err is the nested error. 303 err error 304 } 305 306 // nestedError implements error.Error. 307 func (e *nestedError) Error() string { 308 return e.s + ": " + e.err.Error() 309 } 310 311 // Header is a representation of a DNS message header. 312 type Header struct { 313 ID uint16 314 Response bool 315 OpCode OpCode 316 Authoritative bool 317 Truncated bool 318 RecursionDesired bool 319 RecursionAvailable bool 320 RCode RCode 321 } 322 323 func (m *Header) pack() (id uint16, bits uint16) { 324 id = m.ID 325 bits = uint16(m.OpCode)<<11 | uint16(m.RCode) 326 if m.RecursionAvailable { 327 bits |= headerBitRA 328 } 329 if m.RecursionDesired { 330 bits |= headerBitRD 331 } 332 if m.Truncated { 333 bits |= headerBitTC 334 } 335 if m.Authoritative { 336 bits |= headerBitAA 337 } 338 if m.Response { 339 bits |= headerBitQR 340 } 341 return 342 } 343 344 // GoString implements fmt.GoStringer.GoString. 345 func (m *Header) GoString() string { 346 return "dnsmessage.Header{" + 347 "ID: " + printUint16(m.ID) + ", " + 348 "Response: " + printBool(m.Response) + ", " + 349 "OpCode: " + m.OpCode.GoString() + ", " + 350 "Authoritative: " + printBool(m.Authoritative) + ", " + 351 "Truncated: " + printBool(m.Truncated) + ", " + 352 "RecursionDesired: " + printBool(m.RecursionDesired) + ", " + 353 "RecursionAvailable: " + printBool(m.RecursionAvailable) + ", " + 354 "RCode: " + m.RCode.GoString() + "}" 355 } 356 357 // Message is a representation of a DNS message. 358 type Message struct { 359 Header 360 Questions []Question 361 Answers []Resource 362 Authorities []Resource 363 Additionals []Resource 364 } 365 366 type section uint8 367 368 const ( 369 sectionNotStarted section = iota 370 sectionHeader 371 sectionQuestions 372 sectionAnswers 373 sectionAuthorities 374 sectionAdditionals 375 sectionDone 376 377 headerBitQR = 1 << 15 // query/response (response=1) 378 headerBitAA = 1 << 10 // authoritative 379 headerBitTC = 1 << 9 // truncated 380 headerBitRD = 1 << 8 // recursion desired 381 headerBitRA = 1 << 7 // recursion available 382 ) 383 384 var sectionNames = map[section]string{ 385 sectionHeader: "header", 386 sectionQuestions: "Question", 387 sectionAnswers: "Answer", 388 sectionAuthorities: "Authority", 389 sectionAdditionals: "Additional", 390 } 391 392 // header is the wire format for a DNS message header. 393 type header struct { 394 id uint16 395 bits uint16 396 questions uint16 397 answers uint16 398 authorities uint16 399 additionals uint16 400 } 401 402 func (h *header) count(sec section) uint16 { 403 switch sec { 404 case sectionQuestions: 405 return h.questions 406 case sectionAnswers: 407 return h.answers 408 case sectionAuthorities: 409 return h.authorities 410 case sectionAdditionals: 411 return h.additionals 412 } 413 return 0 414 } 415 416 // pack appends the wire format of the header to msg. 417 func (h *header) pack(msg []byte) []byte { 418 msg = packUint16(msg, h.id) 419 msg = packUint16(msg, h.bits) 420 msg = packUint16(msg, h.questions) 421 msg = packUint16(msg, h.answers) 422 msg = packUint16(msg, h.authorities) 423 return packUint16(msg, h.additionals) 424 } 425 426 func (h *header) unpack(msg []byte, off int) (int, error) { 427 newOff := off 428 var err error 429 if h.id, newOff, err = unpackUint16(msg, newOff); err != nil { 430 return off, &nestedError{"id", err} 431 } 432 if h.bits, newOff, err = unpackUint16(msg, newOff); err != nil { 433 return off, &nestedError{"bits", err} 434 } 435 if h.questions, newOff, err = unpackUint16(msg, newOff); err != nil { 436 return off, &nestedError{"questions", err} 437 } 438 if h.answers, newOff, err = unpackUint16(msg, newOff); err != nil { 439 return off, &nestedError{"answers", err} 440 } 441 if h.authorities, newOff, err = unpackUint16(msg, newOff); err != nil { 442 return off, &nestedError{"authorities", err} 443 } 444 if h.additionals, newOff, err = unpackUint16(msg, newOff); err != nil { 445 return off, &nestedError{"additionals", err} 446 } 447 return newOff, nil 448 } 449 450 func (h *header) header() Header { 451 return Header{ 452 ID: h.id, 453 Response: (h.bits & headerBitQR) != 0, 454 OpCode: OpCode(h.bits>>11) & 0xF, 455 Authoritative: (h.bits & headerBitAA) != 0, 456 Truncated: (h.bits & headerBitTC) != 0, 457 RecursionDesired: (h.bits & headerBitRD) != 0, 458 RecursionAvailable: (h.bits & headerBitRA) != 0, 459 RCode: RCode(h.bits & 0xF), 460 } 461 } 462 463 // A Resource is a DNS resource record. 464 type Resource struct { 465 Header ResourceHeader 466 Body ResourceBody 467 } 468 469 func (r *Resource) GoString() string { 470 return "dnsmessage.Resource{" + 471 "Header: " + r.Header.GoString() + 472 ", Body: &" + r.Body.GoString() + 473 "}" 474 } 475 476 // A ResourceBody is a DNS resource record minus the header. 477 type ResourceBody interface { 478 // pack packs a Resource except for its header. 479 pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) 480 481 // realType returns the actual type of the Resource. This is used to 482 // fill in the header Type field. 483 realType() Type 484 485 // GoString implements fmt.GoStringer.GoString. 486 GoString() string 487 } 488 489 // pack appends the wire format of the Resource to msg. 490 func (r *Resource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 491 if r.Body == nil { 492 return msg, errNilResouceBody 493 } 494 oldMsg := msg 495 r.Header.Type = r.Body.realType() 496 msg, lenOff, err := r.Header.pack(msg, compression, compressionOff) 497 if err != nil { 498 return msg, &nestedError{"ResourceHeader", err} 499 } 500 preLen := len(msg) 501 msg, err = r.Body.pack(msg, compression, compressionOff) 502 if err != nil { 503 return msg, &nestedError{"content", err} 504 } 505 if err := r.Header.fixLen(msg, lenOff, preLen); err != nil { 506 return oldMsg, err 507 } 508 return msg, nil 509 } 510 511 // A Parser allows incrementally parsing a DNS message. 512 // 513 // When parsing is started, the Header is parsed. Next, each Question can be 514 // either parsed or skipped. Alternatively, all Questions can be skipped at 515 // once. When all Questions have been parsed, attempting to parse Questions 516 // will return (nil, nil) and attempting to skip Questions will return 517 // (true, nil). After all Questions have been either parsed or skipped, all 518 // Answers, Authorities and Additionals can be either parsed or skipped in the 519 // same way, and each type of Resource must be fully parsed or skipped before 520 // proceeding to the next type of Resource. 521 // 522 // Note that there is no requirement to fully skip or parse the message. 523 type Parser struct { 524 msg []byte 525 header header 526 527 section section 528 off int 529 index int 530 resHeaderValid bool 531 resHeader ResourceHeader 532 } 533 534 // Start parses the header and enables the parsing of Questions. 535 func (p *Parser) Start(msg []byte) (Header, error) { 536 if p.msg != nil { 537 *p = Parser{} 538 } 539 p.msg = msg 540 var err error 541 if p.off, err = p.header.unpack(msg, 0); err != nil { 542 return Header{}, &nestedError{"unpacking header", err} 543 } 544 p.section = sectionQuestions 545 return p.header.header(), nil 546 } 547 548 func (p *Parser) checkAdvance(sec section) error { 549 if p.section < sec { 550 return ErrNotStarted 551 } 552 if p.section > sec { 553 return ErrSectionDone 554 } 555 p.resHeaderValid = false 556 if p.index == int(p.header.count(sec)) { 557 p.index = 0 558 p.section++ 559 return ErrSectionDone 560 } 561 return nil 562 } 563 564 func (p *Parser) resource(sec section) (Resource, error) { 565 var r Resource 566 var err error 567 r.Header, err = p.resourceHeader(sec) 568 if err != nil { 569 return r, err 570 } 571 p.resHeaderValid = false 572 r.Body, p.off, err = unpackResourceBody(p.msg, p.off, r.Header) 573 if err != nil { 574 return Resource{}, &nestedError{"unpacking " + sectionNames[sec], err} 575 } 576 p.index++ 577 return r, nil 578 } 579 580 func (p *Parser) resourceHeader(sec section) (ResourceHeader, error) { 581 if p.resHeaderValid { 582 return p.resHeader, nil 583 } 584 if err := p.checkAdvance(sec); err != nil { 585 return ResourceHeader{}, err 586 } 587 var hdr ResourceHeader 588 off, err := hdr.unpack(p.msg, p.off) 589 if err != nil { 590 return ResourceHeader{}, err 591 } 592 p.resHeaderValid = true 593 p.resHeader = hdr 594 p.off = off 595 return hdr, nil 596 } 597 598 func (p *Parser) skipResource(sec section) error { 599 if p.resHeaderValid { 600 newOff := p.off + int(p.resHeader.Length) 601 if newOff > len(p.msg) { 602 return errResourceLen 603 } 604 p.off = newOff 605 p.resHeaderValid = false 606 p.index++ 607 return nil 608 } 609 if err := p.checkAdvance(sec); err != nil { 610 return err 611 } 612 var err error 613 p.off, err = skipResource(p.msg, p.off) 614 if err != nil { 615 return &nestedError{"skipping: " + sectionNames[sec], err} 616 } 617 p.index++ 618 return nil 619 } 620 621 // Question parses a single Question. 622 func (p *Parser) Question() (Question, error) { 623 if err := p.checkAdvance(sectionQuestions); err != nil { 624 return Question{}, err 625 } 626 var name Name 627 off, err := name.unpack(p.msg, p.off) 628 if err != nil { 629 return Question{}, &nestedError{"unpacking Question.Name", err} 630 } 631 typ, off, err := unpackType(p.msg, off) 632 if err != nil { 633 return Question{}, &nestedError{"unpacking Question.Type", err} 634 } 635 class, off, err := unpackClass(p.msg, off) 636 if err != nil { 637 return Question{}, &nestedError{"unpacking Question.Class", err} 638 } 639 p.off = off 640 p.index++ 641 return Question{name, typ, class}, nil 642 } 643 644 // AllQuestions parses all Questions. 645 func (p *Parser) AllQuestions() ([]Question, error) { 646 // Multiple questions are valid according to the spec, 647 // but servers don't actually support them. There will 648 // be at most one question here. 649 // 650 // Do not pre-allocate based on info in p.header, since 651 // the data is untrusted. 652 qs := []Question{} 653 for { 654 q, err := p.Question() 655 if err == ErrSectionDone { 656 return qs, nil 657 } 658 if err != nil { 659 return nil, err 660 } 661 qs = append(qs, q) 662 } 663 } 664 665 // SkipQuestion skips a single Question. 666 func (p *Parser) SkipQuestion() error { 667 if err := p.checkAdvance(sectionQuestions); err != nil { 668 return err 669 } 670 off, err := skipName(p.msg, p.off) 671 if err != nil { 672 return &nestedError{"skipping Question Name", err} 673 } 674 if off, err = skipType(p.msg, off); err != nil { 675 return &nestedError{"skipping Question Type", err} 676 } 677 if off, err = skipClass(p.msg, off); err != nil { 678 return &nestedError{"skipping Question Class", err} 679 } 680 p.off = off 681 p.index++ 682 return nil 683 } 684 685 // SkipAllQuestions skips all Questions. 686 func (p *Parser) SkipAllQuestions() error { 687 for { 688 if err := p.SkipQuestion(); err == ErrSectionDone { 689 return nil 690 } else if err != nil { 691 return err 692 } 693 } 694 } 695 696 // AnswerHeader parses a single Answer ResourceHeader. 697 func (p *Parser) AnswerHeader() (ResourceHeader, error) { 698 return p.resourceHeader(sectionAnswers) 699 } 700 701 // Answer parses a single Answer Resource. 702 func (p *Parser) Answer() (Resource, error) { 703 return p.resource(sectionAnswers) 704 } 705 706 // AllAnswers parses all Answer Resources. 707 func (p *Parser) AllAnswers() ([]Resource, error) { 708 // The most common query is for A/AAAA, which usually returns 709 // a handful of IPs. 710 // 711 // Pre-allocate up to a certain limit, since p.header is 712 // untrusted data. 713 n := int(p.header.answers) 714 if n > 20 { 715 n = 20 716 } 717 as := make([]Resource, 0, n) 718 for { 719 a, err := p.Answer() 720 if err == ErrSectionDone { 721 return as, nil 722 } 723 if err != nil { 724 return nil, err 725 } 726 as = append(as, a) 727 } 728 } 729 730 // SkipAnswer skips a single Answer Resource. 731 func (p *Parser) SkipAnswer() error { 732 return p.skipResource(sectionAnswers) 733 } 734 735 // SkipAllAnswers skips all Answer Resources. 736 func (p *Parser) SkipAllAnswers() error { 737 for { 738 if err := p.SkipAnswer(); err == ErrSectionDone { 739 return nil 740 } else if err != nil { 741 return err 742 } 743 } 744 } 745 746 // AuthorityHeader parses a single Authority ResourceHeader. 747 func (p *Parser) AuthorityHeader() (ResourceHeader, error) { 748 return p.resourceHeader(sectionAuthorities) 749 } 750 751 // Authority parses a single Authority Resource. 752 func (p *Parser) Authority() (Resource, error) { 753 return p.resource(sectionAuthorities) 754 } 755 756 // AllAuthorities parses all Authority Resources. 757 func (p *Parser) AllAuthorities() ([]Resource, error) { 758 // Authorities contains SOA in case of NXDOMAIN and friends, 759 // otherwise it is empty. 760 // 761 // Pre-allocate up to a certain limit, since p.header is 762 // untrusted data. 763 n := int(p.header.authorities) 764 if n > 10 { 765 n = 10 766 } 767 as := make([]Resource, 0, n) 768 for { 769 a, err := p.Authority() 770 if err == ErrSectionDone { 771 return as, nil 772 } 773 if err != nil { 774 return nil, err 775 } 776 as = append(as, a) 777 } 778 } 779 780 // SkipAuthority skips a single Authority Resource. 781 func (p *Parser) SkipAuthority() error { 782 return p.skipResource(sectionAuthorities) 783 } 784 785 // SkipAllAuthorities skips all Authority Resources. 786 func (p *Parser) SkipAllAuthorities() error { 787 for { 788 if err := p.SkipAuthority(); err == ErrSectionDone { 789 return nil 790 } else if err != nil { 791 return err 792 } 793 } 794 } 795 796 // AdditionalHeader parses a single Additional ResourceHeader. 797 func (p *Parser) AdditionalHeader() (ResourceHeader, error) { 798 return p.resourceHeader(sectionAdditionals) 799 } 800 801 // Additional parses a single Additional Resource. 802 func (p *Parser) Additional() (Resource, error) { 803 return p.resource(sectionAdditionals) 804 } 805 806 // AllAdditionals parses all Additional Resources. 807 func (p *Parser) AllAdditionals() ([]Resource, error) { 808 // Additionals usually contain OPT, and sometimes A/AAAA 809 // glue records. 810 // 811 // Pre-allocate up to a certain limit, since p.header is 812 // untrusted data. 813 n := int(p.header.additionals) 814 if n > 10 { 815 n = 10 816 } 817 as := make([]Resource, 0, n) 818 for { 819 a, err := p.Additional() 820 if err == ErrSectionDone { 821 return as, nil 822 } 823 if err != nil { 824 return nil, err 825 } 826 as = append(as, a) 827 } 828 } 829 830 // SkipAdditional skips a single Additional Resource. 831 func (p *Parser) SkipAdditional() error { 832 return p.skipResource(sectionAdditionals) 833 } 834 835 // SkipAllAdditionals skips all Additional Resources. 836 func (p *Parser) SkipAllAdditionals() error { 837 for { 838 if err := p.SkipAdditional(); err == ErrSectionDone { 839 return nil 840 } else if err != nil { 841 return err 842 } 843 } 844 } 845 846 // CNAMEResource parses a single CNAMEResource. 847 // 848 // One of the XXXHeader methods must have been called before calling this 849 // method. 850 func (p *Parser) CNAMEResource() (CNAMEResource, error) { 851 if !p.resHeaderValid || p.resHeader.Type != TypeCNAME { 852 return CNAMEResource{}, ErrNotStarted 853 } 854 r, err := unpackCNAMEResource(p.msg, p.off) 855 if err != nil { 856 return CNAMEResource{}, err 857 } 858 p.off += int(p.resHeader.Length) 859 p.resHeaderValid = false 860 p.index++ 861 return r, nil 862 } 863 864 // MXResource parses a single MXResource. 865 // 866 // One of the XXXHeader methods must have been called before calling this 867 // method. 868 func (p *Parser) MXResource() (MXResource, error) { 869 if !p.resHeaderValid || p.resHeader.Type != TypeMX { 870 return MXResource{}, ErrNotStarted 871 } 872 r, err := unpackMXResource(p.msg, p.off) 873 if err != nil { 874 return MXResource{}, err 875 } 876 p.off += int(p.resHeader.Length) 877 p.resHeaderValid = false 878 p.index++ 879 return r, nil 880 } 881 882 // NSResource parses a single NSResource. 883 // 884 // One of the XXXHeader methods must have been called before calling this 885 // method. 886 func (p *Parser) NSResource() (NSResource, error) { 887 if !p.resHeaderValid || p.resHeader.Type != TypeNS { 888 return NSResource{}, ErrNotStarted 889 } 890 r, err := unpackNSResource(p.msg, p.off) 891 if err != nil { 892 return NSResource{}, err 893 } 894 p.off += int(p.resHeader.Length) 895 p.resHeaderValid = false 896 p.index++ 897 return r, nil 898 } 899 900 // PTRResource parses a single PTRResource. 901 // 902 // One of the XXXHeader methods must have been called before calling this 903 // method. 904 func (p *Parser) PTRResource() (PTRResource, error) { 905 if !p.resHeaderValid || p.resHeader.Type != TypePTR { 906 return PTRResource{}, ErrNotStarted 907 } 908 r, err := unpackPTRResource(p.msg, p.off) 909 if err != nil { 910 return PTRResource{}, err 911 } 912 p.off += int(p.resHeader.Length) 913 p.resHeaderValid = false 914 p.index++ 915 return r, nil 916 } 917 918 // SOAResource parses a single SOAResource. 919 // 920 // One of the XXXHeader methods must have been called before calling this 921 // method. 922 func (p *Parser) SOAResource() (SOAResource, error) { 923 if !p.resHeaderValid || p.resHeader.Type != TypeSOA { 924 return SOAResource{}, ErrNotStarted 925 } 926 r, err := unpackSOAResource(p.msg, p.off) 927 if err != nil { 928 return SOAResource{}, err 929 } 930 p.off += int(p.resHeader.Length) 931 p.resHeaderValid = false 932 p.index++ 933 return r, nil 934 } 935 936 // TXTResource parses a single TXTResource. 937 // 938 // One of the XXXHeader methods must have been called before calling this 939 // method. 940 func (p *Parser) TXTResource() (TXTResource, error) { 941 if !p.resHeaderValid || p.resHeader.Type != TypeTXT { 942 return TXTResource{}, ErrNotStarted 943 } 944 r, err := unpackTXTResource(p.msg, p.off, p.resHeader.Length) 945 if err != nil { 946 return TXTResource{}, err 947 } 948 p.off += int(p.resHeader.Length) 949 p.resHeaderValid = false 950 p.index++ 951 return r, nil 952 } 953 954 // SRVResource parses a single SRVResource. 955 // 956 // One of the XXXHeader methods must have been called before calling this 957 // method. 958 func (p *Parser) SRVResource() (SRVResource, error) { 959 if !p.resHeaderValid || p.resHeader.Type != TypeSRV { 960 return SRVResource{}, ErrNotStarted 961 } 962 r, err := unpackSRVResource(p.msg, p.off) 963 if err != nil { 964 return SRVResource{}, err 965 } 966 p.off += int(p.resHeader.Length) 967 p.resHeaderValid = false 968 p.index++ 969 return r, nil 970 } 971 972 // AResource parses a single AResource. 973 // 974 // One of the XXXHeader methods must have been called before calling this 975 // method. 976 func (p *Parser) AResource() (AResource, error) { 977 if !p.resHeaderValid || p.resHeader.Type != TypeA { 978 return AResource{}, ErrNotStarted 979 } 980 r, err := unpackAResource(p.msg, p.off) 981 if err != nil { 982 return AResource{}, err 983 } 984 p.off += int(p.resHeader.Length) 985 p.resHeaderValid = false 986 p.index++ 987 return r, nil 988 } 989 990 // AAAAResource parses a single AAAAResource. 991 // 992 // One of the XXXHeader methods must have been called before calling this 993 // method. 994 func (p *Parser) AAAAResource() (AAAAResource, error) { 995 if !p.resHeaderValid || p.resHeader.Type != TypeAAAA { 996 return AAAAResource{}, ErrNotStarted 997 } 998 r, err := unpackAAAAResource(p.msg, p.off) 999 if err != nil { 1000 return AAAAResource{}, err 1001 } 1002 p.off += int(p.resHeader.Length) 1003 p.resHeaderValid = false 1004 p.index++ 1005 return r, nil 1006 } 1007 1008 // OPTResource parses a single OPTResource. 1009 // 1010 // One of the XXXHeader methods must have been called before calling this 1011 // method. 1012 func (p *Parser) OPTResource() (OPTResource, error) { 1013 if !p.resHeaderValid || p.resHeader.Type != TypeOPT { 1014 return OPTResource{}, ErrNotStarted 1015 } 1016 r, err := unpackOPTResource(p.msg, p.off, p.resHeader.Length) 1017 if err != nil { 1018 return OPTResource{}, err 1019 } 1020 p.off += int(p.resHeader.Length) 1021 p.resHeaderValid = false 1022 p.index++ 1023 return r, nil 1024 } 1025 1026 // Unpack parses a full Message. 1027 func (m *Message) Unpack(msg []byte) error { 1028 var p Parser 1029 var err error 1030 if m.Header, err = p.Start(msg); err != nil { 1031 return err 1032 } 1033 if m.Questions, err = p.AllQuestions(); err != nil { 1034 return err 1035 } 1036 if m.Answers, err = p.AllAnswers(); err != nil { 1037 return err 1038 } 1039 if m.Authorities, err = p.AllAuthorities(); err != nil { 1040 return err 1041 } 1042 if m.Additionals, err = p.AllAdditionals(); err != nil { 1043 return err 1044 } 1045 return nil 1046 } 1047 1048 // Pack packs a full Message. 1049 func (m *Message) Pack() ([]byte, error) { 1050 return m.AppendPack(make([]byte, 0, packStartingCap)) 1051 } 1052 1053 // AppendPack is like Pack but appends the full Message to b and returns the 1054 // extended buffer. 1055 func (m *Message) AppendPack(b []byte) ([]byte, error) { 1056 // Validate the lengths. It is very unlikely that anyone will try to 1057 // pack more than 65535 of any particular type, but it is possible and 1058 // we should fail gracefully. 1059 if len(m.Questions) > int(^uint16(0)) { 1060 return nil, errTooManyQuestions 1061 } 1062 if len(m.Answers) > int(^uint16(0)) { 1063 return nil, errTooManyAnswers 1064 } 1065 if len(m.Authorities) > int(^uint16(0)) { 1066 return nil, errTooManyAuthorities 1067 } 1068 if len(m.Additionals) > int(^uint16(0)) { 1069 return nil, errTooManyAdditionals 1070 } 1071 1072 var h header 1073 h.id, h.bits = m.Header.pack() 1074 1075 h.questions = uint16(len(m.Questions)) 1076 h.answers = uint16(len(m.Answers)) 1077 h.authorities = uint16(len(m.Authorities)) 1078 h.additionals = uint16(len(m.Additionals)) 1079 1080 compressionOff := len(b) 1081 msg := h.pack(b) 1082 1083 // RFC 1035 allows (but does not require) compression for packing. RFC 1084 // 1035 requires unpacking implementations to support compression, so 1085 // unconditionally enabling it is fine. 1086 // 1087 // DNS lookups are typically done over UDP, and RFC 1035 states that UDP 1088 // DNS messages can be a maximum of 512 bytes long. Without compression, 1089 // many DNS response messages are over this limit, so enabling 1090 // compression will help ensure compliance. 1091 compression := map[string]int{} 1092 1093 for i := range m.Questions { 1094 var err error 1095 if msg, err = m.Questions[i].pack(msg, compression, compressionOff); err != nil { 1096 return nil, &nestedError{"packing Question", err} 1097 } 1098 } 1099 for i := range m.Answers { 1100 var err error 1101 if msg, err = m.Answers[i].pack(msg, compression, compressionOff); err != nil { 1102 return nil, &nestedError{"packing Answer", err} 1103 } 1104 } 1105 for i := range m.Authorities { 1106 var err error 1107 if msg, err = m.Authorities[i].pack(msg, compression, compressionOff); err != nil { 1108 return nil, &nestedError{"packing Authority", err} 1109 } 1110 } 1111 for i := range m.Additionals { 1112 var err error 1113 if msg, err = m.Additionals[i].pack(msg, compression, compressionOff); err != nil { 1114 return nil, &nestedError{"packing Additional", err} 1115 } 1116 } 1117 1118 return msg, nil 1119 } 1120 1121 // GoString implements fmt.GoStringer.GoString. 1122 func (m *Message) GoString() string { 1123 s := "dnsmessage.Message{Header: " + m.Header.GoString() + ", " + 1124 "Questions: []dnsmessage.Question{" 1125 if len(m.Questions) > 0 { 1126 s += m.Questions[0].GoString() 1127 for _, q := range m.Questions[1:] { 1128 s += ", " + q.GoString() 1129 } 1130 } 1131 s += "}, Answers: []dnsmessage.Resource{" 1132 if len(m.Answers) > 0 { 1133 s += m.Answers[0].GoString() 1134 for _, a := range m.Answers[1:] { 1135 s += ", " + a.GoString() 1136 } 1137 } 1138 s += "}, Authorities: []dnsmessage.Resource{" 1139 if len(m.Authorities) > 0 { 1140 s += m.Authorities[0].GoString() 1141 for _, a := range m.Authorities[1:] { 1142 s += ", " + a.GoString() 1143 } 1144 } 1145 s += "}, Additionals: []dnsmessage.Resource{" 1146 if len(m.Additionals) > 0 { 1147 s += m.Additionals[0].GoString() 1148 for _, a := range m.Additionals[1:] { 1149 s += ", " + a.GoString() 1150 } 1151 } 1152 return s + "}}" 1153 } 1154 1155 // A Builder allows incrementally packing a DNS message. 1156 // 1157 // Example usage: 1158 // buf := make([]byte, 2, 514) 1159 // b := NewBuilder(buf, Header{...}) 1160 // b.EnableCompression() 1161 // // Optionally start a section and add things to that section. 1162 // // Repeat adding sections as necessary. 1163 // buf, err := b.Finish() 1164 // // If err is nil, buf[2:] will contain the built bytes. 1165 type Builder struct { 1166 // msg is the storage for the message being built. 1167 msg []byte 1168 1169 // section keeps track of the current section being built. 1170 section section 1171 1172 // header keeps track of what should go in the header when Finish is 1173 // called. 1174 header header 1175 1176 // start is the starting index of the bytes allocated in msg for header. 1177 start int 1178 1179 // compression is a mapping from name suffixes to their starting index 1180 // in msg. 1181 compression map[string]int 1182 } 1183 1184 // NewBuilder creates a new builder with compression disabled. 1185 // 1186 // Note: Most users will want to immediately enable compression with the 1187 // EnableCompression method. See that method's comment for why you may or may 1188 // not want to enable compression. 1189 // 1190 // The DNS message is appended to the provided initial buffer buf (which may be 1191 // nil) as it is built. The final message is returned by the (*Builder).Finish 1192 // method, which may return the same underlying array if there was sufficient 1193 // capacity in the slice. 1194 func NewBuilder(buf []byte, h Header) Builder { 1195 if buf == nil { 1196 buf = make([]byte, 0, packStartingCap) 1197 } 1198 b := Builder{msg: buf, start: len(buf)} 1199 b.header.id, b.header.bits = h.pack() 1200 var hb [headerLen]byte 1201 b.msg = append(b.msg, hb[:]...) 1202 b.section = sectionHeader 1203 return b 1204 } 1205 1206 // EnableCompression enables compression in the Builder. 1207 // 1208 // Leaving compression disabled avoids compression related allocations, but can 1209 // result in larger message sizes. Be careful with this mode as it can cause 1210 // messages to exceed the UDP size limit. 1211 // 1212 // According to RFC 1035, section 4.1.4, the use of compression is optional, but 1213 // all implementations must accept both compressed and uncompressed DNS 1214 // messages. 1215 // 1216 // Compression should be enabled before any sections are added for best results. 1217 func (b *Builder) EnableCompression() { 1218 b.compression = map[string]int{} 1219 } 1220 1221 func (b *Builder) startCheck(s section) error { 1222 if b.section <= sectionNotStarted { 1223 return ErrNotStarted 1224 } 1225 if b.section > s { 1226 return ErrSectionDone 1227 } 1228 return nil 1229 } 1230 1231 // StartQuestions prepares the builder for packing Questions. 1232 func (b *Builder) StartQuestions() error { 1233 if err := b.startCheck(sectionQuestions); err != nil { 1234 return err 1235 } 1236 b.section = sectionQuestions 1237 return nil 1238 } 1239 1240 // StartAnswers prepares the builder for packing Answers. 1241 func (b *Builder) StartAnswers() error { 1242 if err := b.startCheck(sectionAnswers); err != nil { 1243 return err 1244 } 1245 b.section = sectionAnswers 1246 return nil 1247 } 1248 1249 // StartAuthorities prepares the builder for packing Authorities. 1250 func (b *Builder) StartAuthorities() error { 1251 if err := b.startCheck(sectionAuthorities); err != nil { 1252 return err 1253 } 1254 b.section = sectionAuthorities 1255 return nil 1256 } 1257 1258 // StartAdditionals prepares the builder for packing Additionals. 1259 func (b *Builder) StartAdditionals() error { 1260 if err := b.startCheck(sectionAdditionals); err != nil { 1261 return err 1262 } 1263 b.section = sectionAdditionals 1264 return nil 1265 } 1266 1267 func (b *Builder) incrementSectionCount() error { 1268 var count *uint16 1269 var err error 1270 switch b.section { 1271 case sectionQuestions: 1272 count = &b.header.questions 1273 err = errTooManyQuestions 1274 case sectionAnswers: 1275 count = &b.header.answers 1276 err = errTooManyAnswers 1277 case sectionAuthorities: 1278 count = &b.header.authorities 1279 err = errTooManyAuthorities 1280 case sectionAdditionals: 1281 count = &b.header.additionals 1282 err = errTooManyAdditionals 1283 } 1284 if *count == ^uint16(0) { 1285 return err 1286 } 1287 *count++ 1288 return nil 1289 } 1290 1291 // Question adds a single Question. 1292 func (b *Builder) Question(q Question) error { 1293 if b.section < sectionQuestions { 1294 return ErrNotStarted 1295 } 1296 if b.section > sectionQuestions { 1297 return ErrSectionDone 1298 } 1299 msg, err := q.pack(b.msg, b.compression, b.start) 1300 if err != nil { 1301 return err 1302 } 1303 if err := b.incrementSectionCount(); err != nil { 1304 return err 1305 } 1306 b.msg = msg 1307 return nil 1308 } 1309 1310 func (b *Builder) checkResourceSection() error { 1311 if b.section < sectionAnswers { 1312 return ErrNotStarted 1313 } 1314 if b.section > sectionAdditionals { 1315 return ErrSectionDone 1316 } 1317 return nil 1318 } 1319 1320 // CNAMEResource adds a single CNAMEResource. 1321 func (b *Builder) CNAMEResource(h ResourceHeader, r CNAMEResource) error { 1322 if err := b.checkResourceSection(); err != nil { 1323 return err 1324 } 1325 h.Type = r.realType() 1326 msg, lenOff, err := h.pack(b.msg, b.compression, b.start) 1327 if err != nil { 1328 return &nestedError{"ResourceHeader", err} 1329 } 1330 preLen := len(msg) 1331 if msg, err = r.pack(msg, b.compression, b.start); err != nil { 1332 return &nestedError{"CNAMEResource body", err} 1333 } 1334 if err := h.fixLen(msg, lenOff, preLen); err != nil { 1335 return err 1336 } 1337 if err := b.incrementSectionCount(); err != nil { 1338 return err 1339 } 1340 b.msg = msg 1341 return nil 1342 } 1343 1344 // MXResource adds a single MXResource. 1345 func (b *Builder) MXResource(h ResourceHeader, r MXResource) error { 1346 if err := b.checkResourceSection(); err != nil { 1347 return err 1348 } 1349 h.Type = r.realType() 1350 msg, lenOff, err := h.pack(b.msg, b.compression, b.start) 1351 if err != nil { 1352 return &nestedError{"ResourceHeader", err} 1353 } 1354 preLen := len(msg) 1355 if msg, err = r.pack(msg, b.compression, b.start); err != nil { 1356 return &nestedError{"MXResource body", err} 1357 } 1358 if err := h.fixLen(msg, lenOff, preLen); err != nil { 1359 return err 1360 } 1361 if err := b.incrementSectionCount(); err != nil { 1362 return err 1363 } 1364 b.msg = msg 1365 return nil 1366 } 1367 1368 // NSResource adds a single NSResource. 1369 func (b *Builder) NSResource(h ResourceHeader, r NSResource) error { 1370 if err := b.checkResourceSection(); err != nil { 1371 return err 1372 } 1373 h.Type = r.realType() 1374 msg, lenOff, err := h.pack(b.msg, b.compression, b.start) 1375 if err != nil { 1376 return &nestedError{"ResourceHeader", err} 1377 } 1378 preLen := len(msg) 1379 if msg, err = r.pack(msg, b.compression, b.start); err != nil { 1380 return &nestedError{"NSResource body", err} 1381 } 1382 if err := h.fixLen(msg, lenOff, preLen); err != nil { 1383 return err 1384 } 1385 if err := b.incrementSectionCount(); err != nil { 1386 return err 1387 } 1388 b.msg = msg 1389 return nil 1390 } 1391 1392 // PTRResource adds a single PTRResource. 1393 func (b *Builder) PTRResource(h ResourceHeader, r PTRResource) error { 1394 if err := b.checkResourceSection(); err != nil { 1395 return err 1396 } 1397 h.Type = r.realType() 1398 msg, lenOff, err := h.pack(b.msg, b.compression, b.start) 1399 if err != nil { 1400 return &nestedError{"ResourceHeader", err} 1401 } 1402 preLen := len(msg) 1403 if msg, err = r.pack(msg, b.compression, b.start); err != nil { 1404 return &nestedError{"PTRResource body", err} 1405 } 1406 if err := h.fixLen(msg, lenOff, preLen); err != nil { 1407 return err 1408 } 1409 if err := b.incrementSectionCount(); err != nil { 1410 return err 1411 } 1412 b.msg = msg 1413 return nil 1414 } 1415 1416 // SOAResource adds a single SOAResource. 1417 func (b *Builder) SOAResource(h ResourceHeader, r SOAResource) error { 1418 if err := b.checkResourceSection(); err != nil { 1419 return err 1420 } 1421 h.Type = r.realType() 1422 msg, lenOff, err := h.pack(b.msg, b.compression, b.start) 1423 if err != nil { 1424 return &nestedError{"ResourceHeader", err} 1425 } 1426 preLen := len(msg) 1427 if msg, err = r.pack(msg, b.compression, b.start); err != nil { 1428 return &nestedError{"SOAResource body", err} 1429 } 1430 if err := h.fixLen(msg, lenOff, preLen); err != nil { 1431 return err 1432 } 1433 if err := b.incrementSectionCount(); err != nil { 1434 return err 1435 } 1436 b.msg = msg 1437 return nil 1438 } 1439 1440 // TXTResource adds a single TXTResource. 1441 func (b *Builder) TXTResource(h ResourceHeader, r TXTResource) error { 1442 if err := b.checkResourceSection(); err != nil { 1443 return err 1444 } 1445 h.Type = r.realType() 1446 msg, lenOff, err := h.pack(b.msg, b.compression, b.start) 1447 if err != nil { 1448 return &nestedError{"ResourceHeader", err} 1449 } 1450 preLen := len(msg) 1451 if msg, err = r.pack(msg, b.compression, b.start); err != nil { 1452 return &nestedError{"TXTResource body", err} 1453 } 1454 if err := h.fixLen(msg, lenOff, preLen); err != nil { 1455 return err 1456 } 1457 if err := b.incrementSectionCount(); err != nil { 1458 return err 1459 } 1460 b.msg = msg 1461 return nil 1462 } 1463 1464 // SRVResource adds a single SRVResource. 1465 func (b *Builder) SRVResource(h ResourceHeader, r SRVResource) error { 1466 if err := b.checkResourceSection(); err != nil { 1467 return err 1468 } 1469 h.Type = r.realType() 1470 msg, lenOff, err := h.pack(b.msg, b.compression, b.start) 1471 if err != nil { 1472 return &nestedError{"ResourceHeader", err} 1473 } 1474 preLen := len(msg) 1475 if msg, err = r.pack(msg, b.compression, b.start); err != nil { 1476 return &nestedError{"SRVResource body", err} 1477 } 1478 if err := h.fixLen(msg, lenOff, preLen); err != nil { 1479 return err 1480 } 1481 if err := b.incrementSectionCount(); err != nil { 1482 return err 1483 } 1484 b.msg = msg 1485 return nil 1486 } 1487 1488 // AResource adds a single AResource. 1489 func (b *Builder) AResource(h ResourceHeader, r AResource) error { 1490 if err := b.checkResourceSection(); err != nil { 1491 return err 1492 } 1493 h.Type = r.realType() 1494 msg, lenOff, err := h.pack(b.msg, b.compression, b.start) 1495 if err != nil { 1496 return &nestedError{"ResourceHeader", err} 1497 } 1498 preLen := len(msg) 1499 if msg, err = r.pack(msg, b.compression, b.start); err != nil { 1500 return &nestedError{"AResource body", err} 1501 } 1502 if err := h.fixLen(msg, lenOff, preLen); err != nil { 1503 return err 1504 } 1505 if err := b.incrementSectionCount(); err != nil { 1506 return err 1507 } 1508 b.msg = msg 1509 return nil 1510 } 1511 1512 // AAAAResource adds a single AAAAResource. 1513 func (b *Builder) AAAAResource(h ResourceHeader, r AAAAResource) error { 1514 if err := b.checkResourceSection(); err != nil { 1515 return err 1516 } 1517 h.Type = r.realType() 1518 msg, lenOff, err := h.pack(b.msg, b.compression, b.start) 1519 if err != nil { 1520 return &nestedError{"ResourceHeader", err} 1521 } 1522 preLen := len(msg) 1523 if msg, err = r.pack(msg, b.compression, b.start); err != nil { 1524 return &nestedError{"AAAAResource body", err} 1525 } 1526 if err := h.fixLen(msg, lenOff, preLen); err != nil { 1527 return err 1528 } 1529 if err := b.incrementSectionCount(); err != nil { 1530 return err 1531 } 1532 b.msg = msg 1533 return nil 1534 } 1535 1536 // OPTResource adds a single OPTResource. 1537 func (b *Builder) OPTResource(h ResourceHeader, r OPTResource) error { 1538 if err := b.checkResourceSection(); err != nil { 1539 return err 1540 } 1541 h.Type = r.realType() 1542 msg, lenOff, err := h.pack(b.msg, b.compression, b.start) 1543 if err != nil { 1544 return &nestedError{"ResourceHeader", err} 1545 } 1546 preLen := len(msg) 1547 if msg, err = r.pack(msg, b.compression, b.start); err != nil { 1548 return &nestedError{"OPTResource body", err} 1549 } 1550 if err := h.fixLen(msg, lenOff, preLen); err != nil { 1551 return err 1552 } 1553 if err := b.incrementSectionCount(); err != nil { 1554 return err 1555 } 1556 b.msg = msg 1557 return nil 1558 } 1559 1560 // Finish ends message building and generates a binary message. 1561 func (b *Builder) Finish() ([]byte, error) { 1562 if b.section < sectionHeader { 1563 return nil, ErrNotStarted 1564 } 1565 b.section = sectionDone 1566 // Space for the header was allocated in NewBuilder. 1567 b.header.pack(b.msg[b.start:b.start]) 1568 return b.msg, nil 1569 } 1570 1571 // A ResourceHeader is the header of a DNS resource record. There are 1572 // many types of DNS resource records, but they all share the same header. 1573 type ResourceHeader struct { 1574 // Name is the domain name for which this resource record pertains. 1575 Name Name 1576 1577 // Type is the type of DNS resource record. 1578 // 1579 // This field will be set automatically during packing. 1580 Type Type 1581 1582 // Class is the class of network to which this DNS resource record 1583 // pertains. 1584 Class Class 1585 1586 // TTL is the length of time (measured in seconds) which this resource 1587 // record is valid for (time to live). All Resources in a set should 1588 // have the same TTL (RFC 2181 Section 5.2). 1589 TTL uint32 1590 1591 // Length is the length of data in the resource record after the header. 1592 // 1593 // This field will be set automatically during packing. 1594 Length uint16 1595 } 1596 1597 // GoString implements fmt.GoStringer.GoString. 1598 func (h *ResourceHeader) GoString() string { 1599 return "dnsmessage.ResourceHeader{" + 1600 "Name: " + h.Name.GoString() + ", " + 1601 "Type: " + h.Type.GoString() + ", " + 1602 "Class: " + h.Class.GoString() + ", " + 1603 "TTL: " + printUint32(h.TTL) + ", " + 1604 "Length: " + printUint16(h.Length) + "}" 1605 } 1606 1607 // pack appends the wire format of the ResourceHeader to oldMsg. 1608 // 1609 // lenOff is the offset in msg where the Length field was packed. 1610 func (h *ResourceHeader) pack(oldMsg []byte, compression map[string]int, compressionOff int) (msg []byte, lenOff int, err error) { 1611 msg = oldMsg 1612 if msg, err = h.Name.pack(msg, compression, compressionOff); err != nil { 1613 return oldMsg, 0, &nestedError{"Name", err} 1614 } 1615 msg = packType(msg, h.Type) 1616 msg = packClass(msg, h.Class) 1617 msg = packUint32(msg, h.TTL) 1618 lenOff = len(msg) 1619 msg = packUint16(msg, h.Length) 1620 return msg, lenOff, nil 1621 } 1622 1623 func (h *ResourceHeader) unpack(msg []byte, off int) (int, error) { 1624 newOff := off 1625 var err error 1626 if newOff, err = h.Name.unpack(msg, newOff); err != nil { 1627 return off, &nestedError{"Name", err} 1628 } 1629 if h.Type, newOff, err = unpackType(msg, newOff); err != nil { 1630 return off, &nestedError{"Type", err} 1631 } 1632 if h.Class, newOff, err = unpackClass(msg, newOff); err != nil { 1633 return off, &nestedError{"Class", err} 1634 } 1635 if h.TTL, newOff, err = unpackUint32(msg, newOff); err != nil { 1636 return off, &nestedError{"TTL", err} 1637 } 1638 if h.Length, newOff, err = unpackUint16(msg, newOff); err != nil { 1639 return off, &nestedError{"Length", err} 1640 } 1641 return newOff, nil 1642 } 1643 1644 // fixLen updates a packed ResourceHeader to include the length of the 1645 // ResourceBody. 1646 // 1647 // lenOff is the offset of the ResourceHeader.Length field in msg. 1648 // 1649 // preLen is the length that msg was before the ResourceBody was packed. 1650 func (h *ResourceHeader) fixLen(msg []byte, lenOff int, preLen int) error { 1651 conLen := len(msg) - preLen 1652 if conLen > int(^uint16(0)) { 1653 return errResTooLong 1654 } 1655 1656 // Fill in the length now that we know how long the content is. 1657 packUint16(msg[lenOff:lenOff], uint16(conLen)) 1658 h.Length = uint16(conLen) 1659 1660 return nil 1661 } 1662 1663 // EDNS(0) wire costants. 1664 const ( 1665 edns0Version = 0 1666 1667 edns0DNSSECOK = 0x00008000 1668 ednsVersionMask = 0x00ff0000 1669 edns0DNSSECOKMask = 0x00ff8000 1670 ) 1671 1672 // SetEDNS0 configures h for EDNS(0). 1673 // 1674 // The provided extRCode must be an extedned RCode. 1675 func (h *ResourceHeader) SetEDNS0(udpPayloadLen int, extRCode RCode, dnssecOK bool) error { 1676 h.Name = Name{Data: [nameLen]byte{'.'}, Length: 1} // RFC 6891 section 6.1.2 1677 h.Type = TypeOPT 1678 h.Class = Class(udpPayloadLen) 1679 h.TTL = uint32(extRCode) >> 4 << 24 1680 if dnssecOK { 1681 h.TTL |= edns0DNSSECOK 1682 } 1683 return nil 1684 } 1685 1686 // DNSSECAllowed reports whether the DNSSEC OK bit is set. 1687 func (h *ResourceHeader) DNSSECAllowed() bool { 1688 return h.TTL&edns0DNSSECOKMask == edns0DNSSECOK // RFC 6891 section 6.1.3 1689 } 1690 1691 // ExtendedRCode returns an extended RCode. 1692 // 1693 // The provided rcode must be the RCode in DNS message header. 1694 func (h *ResourceHeader) ExtendedRCode(rcode RCode) RCode { 1695 if h.TTL&ednsVersionMask == edns0Version { // RFC 6891 section 6.1.3 1696 return RCode(h.TTL>>24<<4) | rcode 1697 } 1698 return rcode 1699 } 1700 1701 func skipResource(msg []byte, off int) (int, error) { 1702 newOff, err := skipName(msg, off) 1703 if err != nil { 1704 return off, &nestedError{"Name", err} 1705 } 1706 if newOff, err = skipType(msg, newOff); err != nil { 1707 return off, &nestedError{"Type", err} 1708 } 1709 if newOff, err = skipClass(msg, newOff); err != nil { 1710 return off, &nestedError{"Class", err} 1711 } 1712 if newOff, err = skipUint32(msg, newOff); err != nil { 1713 return off, &nestedError{"TTL", err} 1714 } 1715 length, newOff, err := unpackUint16(msg, newOff) 1716 if err != nil { 1717 return off, &nestedError{"Length", err} 1718 } 1719 if newOff += int(length); newOff > len(msg) { 1720 return off, errResourceLen 1721 } 1722 return newOff, nil 1723 } 1724 1725 // packUint16 appends the wire format of field to msg. 1726 func packUint16(msg []byte, field uint16) []byte { 1727 return append(msg, byte(field>>8), byte(field)) 1728 } 1729 1730 func unpackUint16(msg []byte, off int) (uint16, int, error) { 1731 if off+uint16Len > len(msg) { 1732 return 0, off, errBaseLen 1733 } 1734 return uint16(msg[off])<<8 | uint16(msg[off+1]), off + uint16Len, nil 1735 } 1736 1737 func skipUint16(msg []byte, off int) (int, error) { 1738 if off+uint16Len > len(msg) { 1739 return off, errBaseLen 1740 } 1741 return off + uint16Len, nil 1742 } 1743 1744 // packType appends the wire format of field to msg. 1745 func packType(msg []byte, field Type) []byte { 1746 return packUint16(msg, uint16(field)) 1747 } 1748 1749 func unpackType(msg []byte, off int) (Type, int, error) { 1750 t, o, err := unpackUint16(msg, off) 1751 return Type(t), o, err 1752 } 1753 1754 func skipType(msg []byte, off int) (int, error) { 1755 return skipUint16(msg, off) 1756 } 1757 1758 // packClass appends the wire format of field to msg. 1759 func packClass(msg []byte, field Class) []byte { 1760 return packUint16(msg, uint16(field)) 1761 } 1762 1763 func unpackClass(msg []byte, off int) (Class, int, error) { 1764 c, o, err := unpackUint16(msg, off) 1765 return Class(c), o, err 1766 } 1767 1768 func skipClass(msg []byte, off int) (int, error) { 1769 return skipUint16(msg, off) 1770 } 1771 1772 // packUint32 appends the wire format of field to msg. 1773 func packUint32(msg []byte, field uint32) []byte { 1774 return append( 1775 msg, 1776 byte(field>>24), 1777 byte(field>>16), 1778 byte(field>>8), 1779 byte(field), 1780 ) 1781 } 1782 1783 func unpackUint32(msg []byte, off int) (uint32, int, error) { 1784 if off+uint32Len > len(msg) { 1785 return 0, off, errBaseLen 1786 } 1787 v := uint32(msg[off])<<24 | uint32(msg[off+1])<<16 | uint32(msg[off+2])<<8 | uint32(msg[off+3]) 1788 return v, off + uint32Len, nil 1789 } 1790 1791 func skipUint32(msg []byte, off int) (int, error) { 1792 if off+uint32Len > len(msg) { 1793 return off, errBaseLen 1794 } 1795 return off + uint32Len, nil 1796 } 1797 1798 // packText appends the wire format of field to msg. 1799 func packText(msg []byte, field string) ([]byte, error) { 1800 l := len(field) 1801 if l > 255 { 1802 return nil, errStringTooLong 1803 } 1804 msg = append(msg, byte(l)) 1805 msg = append(msg, field...) 1806 1807 return msg, nil 1808 } 1809 1810 func unpackText(msg []byte, off int) (string, int, error) { 1811 if off >= len(msg) { 1812 return "", off, errBaseLen 1813 } 1814 beginOff := off + 1 1815 endOff := beginOff + int(msg[off]) 1816 if endOff > len(msg) { 1817 return "", off, errCalcLen 1818 } 1819 return string(msg[beginOff:endOff]), endOff, nil 1820 } 1821 1822 func skipText(msg []byte, off int) (int, error) { 1823 if off >= len(msg) { 1824 return off, errBaseLen 1825 } 1826 endOff := off + 1 + int(msg[off]) 1827 if endOff > len(msg) { 1828 return off, errCalcLen 1829 } 1830 return endOff, nil 1831 } 1832 1833 // packBytes appends the wire format of field to msg. 1834 func packBytes(msg []byte, field []byte) []byte { 1835 return append(msg, field...) 1836 } 1837 1838 func unpackBytes(msg []byte, off int, field []byte) (int, error) { 1839 newOff := off + len(field) 1840 if newOff > len(msg) { 1841 return off, errBaseLen 1842 } 1843 copy(field, msg[off:newOff]) 1844 return newOff, nil 1845 } 1846 1847 func skipBytes(msg []byte, off int, field []byte) (int, error) { 1848 newOff := off + len(field) 1849 if newOff > len(msg) { 1850 return off, errBaseLen 1851 } 1852 return newOff, nil 1853 } 1854 1855 const nameLen = 255 1856 1857 // A Name is a non-encoded domain name. It is used instead of strings to avoid 1858 // allocations. 1859 type Name struct { 1860 Data [nameLen]byte 1861 Length uint8 1862 } 1863 1864 // NewName creates a new Name from a string. 1865 func NewName(name string) (Name, error) { 1866 if len([]byte(name)) > nameLen { 1867 return Name{}, errCalcLen 1868 } 1869 n := Name{Length: uint8(len(name))} 1870 copy(n.Data[:], []byte(name)) 1871 return n, nil 1872 } 1873 1874 // MustNewName creates a new Name from a string and panics on error. 1875 func MustNewName(name string) Name { 1876 n, err := NewName(name) 1877 if err != nil { 1878 panic("creating name: " + err.Error()) 1879 } 1880 return n 1881 } 1882 1883 // String implements fmt.Stringer.String. 1884 func (n Name) String() string { 1885 return string(n.Data[:n.Length]) 1886 } 1887 1888 // GoString implements fmt.GoStringer.GoString. 1889 func (n *Name) GoString() string { 1890 return `dnsmessage.MustNewName("` + printString(n.Data[:n.Length]) + `")` 1891 } 1892 1893 // pack appends the wire format of the Name to msg. 1894 // 1895 // Domain names are a sequence of counted strings split at the dots. They end 1896 // with a zero-length string. Compression can be used to reuse domain suffixes. 1897 // 1898 // The compression map will be updated with new domain suffixes. If compression 1899 // is nil, compression will not be used. 1900 func (n *Name) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 1901 oldMsg := msg 1902 1903 // Add a trailing dot to canonicalize name. 1904 if n.Length == 0 || n.Data[n.Length-1] != '.' { 1905 return oldMsg, errNonCanonicalName 1906 } 1907 1908 // Allow root domain. 1909 if n.Data[0] == '.' && n.Length == 1 { 1910 return append(msg, 0), nil 1911 } 1912 1913 // Emit sequence of counted strings, chopping at dots. 1914 for i, begin := 0, 0; i < int(n.Length); i++ { 1915 // Check for the end of the segment. 1916 if n.Data[i] == '.' { 1917 // The two most significant bits have special meaning. 1918 // It isn't allowed for segments to be long enough to 1919 // need them. 1920 if i-begin >= 1<<6 { 1921 return oldMsg, errSegTooLong 1922 } 1923 1924 // Segments must have a non-zero length. 1925 if i-begin == 0 { 1926 return oldMsg, errZeroSegLen 1927 } 1928 1929 msg = append(msg, byte(i-begin)) 1930 1931 for j := begin; j < i; j++ { 1932 msg = append(msg, n.Data[j]) 1933 } 1934 1935 begin = i + 1 1936 continue 1937 } 1938 1939 // We can only compress domain suffixes starting with a new 1940 // segment. A pointer is two bytes with the two most significant 1941 // bits set to 1 to indicate that it is a pointer. 1942 if (i == 0 || n.Data[i-1] == '.') && compression != nil { 1943 if ptr, ok := compression[string(n.Data[i:])]; ok { 1944 // Hit. Emit a pointer instead of the rest of 1945 // the domain. 1946 return append(msg, byte(ptr>>8|0xC0), byte(ptr)), nil 1947 } 1948 1949 // Miss. Add the suffix to the compression table if the 1950 // offset can be stored in the available 14 bytes. 1951 if len(msg) <= int(^uint16(0)>>2) { 1952 compression[string(n.Data[i:])] = len(msg) - compressionOff 1953 } 1954 } 1955 } 1956 return append(msg, 0), nil 1957 } 1958 1959 // unpack unpacks a domain name. 1960 func (n *Name) unpack(msg []byte, off int) (int, error) { 1961 return n.unpackCompressed(msg, off, true /* allowCompression */) 1962 } 1963 1964 func (n *Name) unpackCompressed(msg []byte, off int, allowCompression bool) (int, error) { 1965 // currOff is the current working offset. 1966 currOff := off 1967 1968 // newOff is the offset where the next record will start. Pointers lead 1969 // to data that belongs to other names and thus doesn't count towards to 1970 // the usage of this name. 1971 newOff := off 1972 1973 // ptr is the number of pointers followed. 1974 var ptr int 1975 1976 // Name is a slice representation of the name data. 1977 name := n.Data[:0] 1978 1979 Loop: 1980 for { 1981 if currOff >= len(msg) { 1982 return off, errBaseLen 1983 } 1984 c := int(msg[currOff]) 1985 currOff++ 1986 switch c & 0xC0 { 1987 case 0x00: // String segment 1988 if c == 0x00 { 1989 // A zero length signals the end of the name. 1990 break Loop 1991 } 1992 endOff := currOff + c 1993 if endOff > len(msg) { 1994 return off, errCalcLen 1995 } 1996 name = append(name, msg[currOff:endOff]...) 1997 name = append(name, '.') 1998 currOff = endOff 1999 case 0xC0: // Pointer 2000 if !allowCompression { 2001 return off, errCompressedSRV 2002 } 2003 if currOff >= len(msg) { 2004 return off, errInvalidPtr 2005 } 2006 c1 := msg[currOff] 2007 currOff++ 2008 if ptr == 0 { 2009 newOff = currOff 2010 } 2011 // Don't follow too many pointers, maybe there's a loop. 2012 if ptr++; ptr > 10 { 2013 return off, errTooManyPtr 2014 } 2015 currOff = (c^0xC0)<<8 | int(c1) 2016 default: 2017 // Prefixes 0x80 and 0x40 are reserved. 2018 return off, errReserved 2019 } 2020 } 2021 if len(name) == 0 { 2022 name = append(name, '.') 2023 } 2024 if len(name) > len(n.Data) { 2025 return off, errCalcLen 2026 } 2027 n.Length = uint8(len(name)) 2028 if ptr == 0 { 2029 newOff = currOff 2030 } 2031 return newOff, nil 2032 } 2033 2034 func skipName(msg []byte, off int) (int, error) { 2035 // newOff is the offset where the next record will start. Pointers lead 2036 // to data that belongs to other names and thus doesn't count towards to 2037 // the usage of this name. 2038 newOff := off 2039 2040 Loop: 2041 for { 2042 if newOff >= len(msg) { 2043 return off, errBaseLen 2044 } 2045 c := int(msg[newOff]) 2046 newOff++ 2047 switch c & 0xC0 { 2048 case 0x00: 2049 if c == 0x00 { 2050 // A zero length signals the end of the name. 2051 break Loop 2052 } 2053 // literal string 2054 newOff += c 2055 if newOff > len(msg) { 2056 return off, errCalcLen 2057 } 2058 case 0xC0: 2059 // Pointer to somewhere else in msg. 2060 2061 // Pointers are two bytes. 2062 newOff++ 2063 2064 // Don't follow the pointer as the data here has ended. 2065 break Loop 2066 default: 2067 // Prefixes 0x80 and 0x40 are reserved. 2068 return off, errReserved 2069 } 2070 } 2071 2072 return newOff, nil 2073 } 2074 2075 // A Question is a DNS query. 2076 type Question struct { 2077 Name Name 2078 Type Type 2079 Class Class 2080 } 2081 2082 // pack appends the wire format of the Question to msg. 2083 func (q *Question) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2084 msg, err := q.Name.pack(msg, compression, compressionOff) 2085 if err != nil { 2086 return msg, &nestedError{"Name", err} 2087 } 2088 msg = packType(msg, q.Type) 2089 return packClass(msg, q.Class), nil 2090 } 2091 2092 // GoString implements fmt.GoStringer.GoString. 2093 func (q *Question) GoString() string { 2094 return "dnsmessage.Question{" + 2095 "Name: " + q.Name.GoString() + ", " + 2096 "Type: " + q.Type.GoString() + ", " + 2097 "Class: " + q.Class.GoString() + "}" 2098 } 2099 2100 func unpackResourceBody(msg []byte, off int, hdr ResourceHeader) (ResourceBody, int, error) { 2101 var ( 2102 r ResourceBody 2103 err error 2104 name string 2105 ) 2106 switch hdr.Type { 2107 case TypeA: 2108 var rb AResource 2109 rb, err = unpackAResource(msg, off) 2110 r = &rb 2111 name = "A" 2112 case TypeNS: 2113 var rb NSResource 2114 rb, err = unpackNSResource(msg, off) 2115 r = &rb 2116 name = "NS" 2117 case TypeCNAME: 2118 var rb CNAMEResource 2119 rb, err = unpackCNAMEResource(msg, off) 2120 r = &rb 2121 name = "CNAME" 2122 case TypeSOA: 2123 var rb SOAResource 2124 rb, err = unpackSOAResource(msg, off) 2125 r = &rb 2126 name = "SOA" 2127 case TypePTR: 2128 var rb PTRResource 2129 rb, err = unpackPTRResource(msg, off) 2130 r = &rb 2131 name = "PTR" 2132 case TypeMX: 2133 var rb MXResource 2134 rb, err = unpackMXResource(msg, off) 2135 r = &rb 2136 name = "MX" 2137 case TypeTXT: 2138 var rb TXTResource 2139 rb, err = unpackTXTResource(msg, off, hdr.Length) 2140 r = &rb 2141 name = "TXT" 2142 case TypeAAAA: 2143 var rb AAAAResource 2144 rb, err = unpackAAAAResource(msg, off) 2145 r = &rb 2146 name = "AAAA" 2147 case TypeSRV: 2148 var rb SRVResource 2149 rb, err = unpackSRVResource(msg, off) 2150 r = &rb 2151 name = "SRV" 2152 case TypeOPT: 2153 var rb OPTResource 2154 rb, err = unpackOPTResource(msg, off, hdr.Length) 2155 r = &rb 2156 name = "OPT" 2157 } 2158 if err != nil { 2159 return nil, off, &nestedError{name + " record", err} 2160 } 2161 if r == nil { 2162 return nil, off, errors.New("invalid resource type: " + string(hdr.Type+'0')) 2163 } 2164 return r, off + int(hdr.Length), nil 2165 } 2166 2167 // A CNAMEResource is a CNAME Resource record. 2168 type CNAMEResource struct { 2169 CNAME Name 2170 } 2171 2172 func (r *CNAMEResource) realType() Type { 2173 return TypeCNAME 2174 } 2175 2176 // pack appends the wire format of the CNAMEResource to msg. 2177 func (r *CNAMEResource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2178 return r.CNAME.pack(msg, compression, compressionOff) 2179 } 2180 2181 // GoString implements fmt.GoStringer.GoString. 2182 func (r *CNAMEResource) GoString() string { 2183 return "dnsmessage.CNAMEResource{CNAME: " + r.CNAME.GoString() + "}" 2184 } 2185 2186 func unpackCNAMEResource(msg []byte, off int) (CNAMEResource, error) { 2187 var cname Name 2188 if _, err := cname.unpack(msg, off); err != nil { 2189 return CNAMEResource{}, err 2190 } 2191 return CNAMEResource{cname}, nil 2192 } 2193 2194 // An MXResource is an MX Resource record. 2195 type MXResource struct { 2196 Pref uint16 2197 MX Name 2198 } 2199 2200 func (r *MXResource) realType() Type { 2201 return TypeMX 2202 } 2203 2204 // pack appends the wire format of the MXResource to msg. 2205 func (r *MXResource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2206 oldMsg := msg 2207 msg = packUint16(msg, r.Pref) 2208 msg, err := r.MX.pack(msg, compression, compressionOff) 2209 if err != nil { 2210 return oldMsg, &nestedError{"MXResource.MX", err} 2211 } 2212 return msg, nil 2213 } 2214 2215 // GoString implements fmt.GoStringer.GoString. 2216 func (r *MXResource) GoString() string { 2217 return "dnsmessage.MXResource{" + 2218 "Pref: " + printUint16(r.Pref) + ", " + 2219 "MX: " + r.MX.GoString() + "}" 2220 } 2221 2222 func unpackMXResource(msg []byte, off int) (MXResource, error) { 2223 pref, off, err := unpackUint16(msg, off) 2224 if err != nil { 2225 return MXResource{}, &nestedError{"Pref", err} 2226 } 2227 var mx Name 2228 if _, err := mx.unpack(msg, off); err != nil { 2229 return MXResource{}, &nestedError{"MX", err} 2230 } 2231 return MXResource{pref, mx}, nil 2232 } 2233 2234 // An NSResource is an NS Resource record. 2235 type NSResource struct { 2236 NS Name 2237 } 2238 2239 func (r *NSResource) realType() Type { 2240 return TypeNS 2241 } 2242 2243 // pack appends the wire format of the NSResource to msg. 2244 func (r *NSResource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2245 return r.NS.pack(msg, compression, compressionOff) 2246 } 2247 2248 // GoString implements fmt.GoStringer.GoString. 2249 func (r *NSResource) GoString() string { 2250 return "dnsmessage.NSResource{NS: " + r.NS.GoString() + "}" 2251 } 2252 2253 func unpackNSResource(msg []byte, off int) (NSResource, error) { 2254 var ns Name 2255 if _, err := ns.unpack(msg, off); err != nil { 2256 return NSResource{}, err 2257 } 2258 return NSResource{ns}, nil 2259 } 2260 2261 // A PTRResource is a PTR Resource record. 2262 type PTRResource struct { 2263 PTR Name 2264 } 2265 2266 func (r *PTRResource) realType() Type { 2267 return TypePTR 2268 } 2269 2270 // pack appends the wire format of the PTRResource to msg. 2271 func (r *PTRResource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2272 return r.PTR.pack(msg, compression, compressionOff) 2273 } 2274 2275 // GoString implements fmt.GoStringer.GoString. 2276 func (r *PTRResource) GoString() string { 2277 return "dnsmessage.PTRResource{PTR: " + r.PTR.GoString() + "}" 2278 } 2279 2280 func unpackPTRResource(msg []byte, off int) (PTRResource, error) { 2281 var ptr Name 2282 if _, err := ptr.unpack(msg, off); err != nil { 2283 return PTRResource{}, err 2284 } 2285 return PTRResource{ptr}, nil 2286 } 2287 2288 // An SOAResource is an SOA Resource record. 2289 type SOAResource struct { 2290 NS Name 2291 MBox Name 2292 Serial uint32 2293 Refresh uint32 2294 Retry uint32 2295 Expire uint32 2296 2297 // MinTTL the is the default TTL of Resources records which did not 2298 // contain a TTL value and the TTL of negative responses. (RFC 2308 2299 // Section 4) 2300 MinTTL uint32 2301 } 2302 2303 func (r *SOAResource) realType() Type { 2304 return TypeSOA 2305 } 2306 2307 // pack appends the wire format of the SOAResource to msg. 2308 func (r *SOAResource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2309 oldMsg := msg 2310 msg, err := r.NS.pack(msg, compression, compressionOff) 2311 if err != nil { 2312 return oldMsg, &nestedError{"SOAResource.NS", err} 2313 } 2314 msg, err = r.MBox.pack(msg, compression, compressionOff) 2315 if err != nil { 2316 return oldMsg, &nestedError{"SOAResource.MBox", err} 2317 } 2318 msg = packUint32(msg, r.Serial) 2319 msg = packUint32(msg, r.Refresh) 2320 msg = packUint32(msg, r.Retry) 2321 msg = packUint32(msg, r.Expire) 2322 return packUint32(msg, r.MinTTL), nil 2323 } 2324 2325 // GoString implements fmt.GoStringer.GoString. 2326 func (r *SOAResource) GoString() string { 2327 return "dnsmessage.SOAResource{" + 2328 "NS: " + r.NS.GoString() + ", " + 2329 "MBox: " + r.MBox.GoString() + ", " + 2330 "Serial: " + printUint32(r.Serial) + ", " + 2331 "Refresh: " + printUint32(r.Refresh) + ", " + 2332 "Retry: " + printUint32(r.Retry) + ", " + 2333 "Expire: " + printUint32(r.Expire) + ", " + 2334 "MinTTL: " + printUint32(r.MinTTL) + "}" 2335 } 2336 2337 func unpackSOAResource(msg []byte, off int) (SOAResource, error) { 2338 var ns Name 2339 off, err := ns.unpack(msg, off) 2340 if err != nil { 2341 return SOAResource{}, &nestedError{"NS", err} 2342 } 2343 var mbox Name 2344 if off, err = mbox.unpack(msg, off); err != nil { 2345 return SOAResource{}, &nestedError{"MBox", err} 2346 } 2347 serial, off, err := unpackUint32(msg, off) 2348 if err != nil { 2349 return SOAResource{}, &nestedError{"Serial", err} 2350 } 2351 refresh, off, err := unpackUint32(msg, off) 2352 if err != nil { 2353 return SOAResource{}, &nestedError{"Refresh", err} 2354 } 2355 retry, off, err := unpackUint32(msg, off) 2356 if err != nil { 2357 return SOAResource{}, &nestedError{"Retry", err} 2358 } 2359 expire, off, err := unpackUint32(msg, off) 2360 if err != nil { 2361 return SOAResource{}, &nestedError{"Expire", err} 2362 } 2363 minTTL, _, err := unpackUint32(msg, off) 2364 if err != nil { 2365 return SOAResource{}, &nestedError{"MinTTL", err} 2366 } 2367 return SOAResource{ns, mbox, serial, refresh, retry, expire, minTTL}, nil 2368 } 2369 2370 // A TXTResource is a TXT Resource record. 2371 type TXTResource struct { 2372 TXT []string 2373 } 2374 2375 func (r *TXTResource) realType() Type { 2376 return TypeTXT 2377 } 2378 2379 // pack appends the wire format of the TXTResource to msg. 2380 func (r *TXTResource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2381 oldMsg := msg 2382 for _, s := range r.TXT { 2383 var err error 2384 msg, err = packText(msg, s) 2385 if err != nil { 2386 return oldMsg, err 2387 } 2388 } 2389 return msg, nil 2390 } 2391 2392 // GoString implements fmt.GoStringer.GoString. 2393 func (r *TXTResource) GoString() string { 2394 s := "dnsmessage.TXTResource{TXT: []string{" 2395 if len(r.TXT) == 0 { 2396 return s + "}}" 2397 } 2398 s += `"` + printString([]byte(r.TXT[0])) 2399 for _, t := range r.TXT[1:] { 2400 s += `", "` + printString([]byte(t)) 2401 } 2402 return s + `"}}` 2403 } 2404 2405 func unpackTXTResource(msg []byte, off int, length uint16) (TXTResource, error) { 2406 txts := make([]string, 0, 1) 2407 for n := uint16(0); n < length; { 2408 var t string 2409 var err error 2410 if t, off, err = unpackText(msg, off); err != nil { 2411 return TXTResource{}, &nestedError{"text", err} 2412 } 2413 // Check if we got too many bytes. 2414 if length-n < uint16(len(t))+1 { 2415 return TXTResource{}, errCalcLen 2416 } 2417 n += uint16(len(t)) + 1 2418 txts = append(txts, t) 2419 } 2420 return TXTResource{txts}, nil 2421 } 2422 2423 // An SRVResource is an SRV Resource record. 2424 type SRVResource struct { 2425 Priority uint16 2426 Weight uint16 2427 Port uint16 2428 Target Name // Not compressed as per RFC 2782. 2429 } 2430 2431 func (r *SRVResource) realType() Type { 2432 return TypeSRV 2433 } 2434 2435 // pack appends the wire format of the SRVResource to msg. 2436 func (r *SRVResource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2437 oldMsg := msg 2438 msg = packUint16(msg, r.Priority) 2439 msg = packUint16(msg, r.Weight) 2440 msg = packUint16(msg, r.Port) 2441 msg, err := r.Target.pack(msg, nil, compressionOff) 2442 if err != nil { 2443 return oldMsg, &nestedError{"SRVResource.Target", err} 2444 } 2445 return msg, nil 2446 } 2447 2448 // GoString implements fmt.GoStringer.GoString. 2449 func (r *SRVResource) GoString() string { 2450 return "dnsmessage.SRVResource{" + 2451 "Priority: " + printUint16(r.Priority) + ", " + 2452 "Weight: " + printUint16(r.Weight) + ", " + 2453 "Port: " + printUint16(r.Port) + ", " + 2454 "Target: " + r.Target.GoString() + "}" 2455 } 2456 2457 func unpackSRVResource(msg []byte, off int) (SRVResource, error) { 2458 priority, off, err := unpackUint16(msg, off) 2459 if err != nil { 2460 return SRVResource{}, &nestedError{"Priority", err} 2461 } 2462 weight, off, err := unpackUint16(msg, off) 2463 if err != nil { 2464 return SRVResource{}, &nestedError{"Weight", err} 2465 } 2466 port, off, err := unpackUint16(msg, off) 2467 if err != nil { 2468 return SRVResource{}, &nestedError{"Port", err} 2469 } 2470 var target Name 2471 if _, err := target.unpackCompressed(msg, off, false /* allowCompression */); err != nil { 2472 return SRVResource{}, &nestedError{"Target", err} 2473 } 2474 return SRVResource{priority, weight, port, target}, nil 2475 } 2476 2477 // An AResource is an A Resource record. 2478 type AResource struct { 2479 A [4]byte 2480 } 2481 2482 func (r *AResource) realType() Type { 2483 return TypeA 2484 } 2485 2486 // pack appends the wire format of the AResource to msg. 2487 func (r *AResource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2488 return packBytes(msg, r.A[:]), nil 2489 } 2490 2491 // GoString implements fmt.GoStringer.GoString. 2492 func (r *AResource) GoString() string { 2493 return "dnsmessage.AResource{" + 2494 "A: [4]byte{" + printByteSlice(r.A[:]) + "}}" 2495 } 2496 2497 func unpackAResource(msg []byte, off int) (AResource, error) { 2498 var a [4]byte 2499 if _, err := unpackBytes(msg, off, a[:]); err != nil { 2500 return AResource{}, err 2501 } 2502 return AResource{a}, nil 2503 } 2504 2505 // An AAAAResource is an AAAA Resource record. 2506 type AAAAResource struct { 2507 AAAA [16]byte 2508 } 2509 2510 func (r *AAAAResource) realType() Type { 2511 return TypeAAAA 2512 } 2513 2514 // GoString implements fmt.GoStringer.GoString. 2515 func (r *AAAAResource) GoString() string { 2516 return "dnsmessage.AAAAResource{" + 2517 "AAAA: [16]byte{" + printByteSlice(r.AAAA[:]) + "}}" 2518 } 2519 2520 // pack appends the wire format of the AAAAResource to msg. 2521 func (r *AAAAResource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2522 return packBytes(msg, r.AAAA[:]), nil 2523 } 2524 2525 func unpackAAAAResource(msg []byte, off int) (AAAAResource, error) { 2526 var aaaa [16]byte 2527 if _, err := unpackBytes(msg, off, aaaa[:]); err != nil { 2528 return AAAAResource{}, err 2529 } 2530 return AAAAResource{aaaa}, nil 2531 } 2532 2533 // An OPTResource is an OPT pseudo Resource record. 2534 // 2535 // The pseudo resource record is part of the extension mechanisms for DNS 2536 // as defined in RFC 6891. 2537 type OPTResource struct { 2538 Options []Option 2539 } 2540 2541 // An Option represents a DNS message option within OPTResource. 2542 // 2543 // The message option is part of the extension mechanisms for DNS as 2544 // defined in RFC 6891. 2545 type Option struct { 2546 Code uint16 // option code 2547 Data []byte 2548 } 2549 2550 // GoString implements fmt.GoStringer.GoString. 2551 func (o *Option) GoString() string { 2552 return "dnsmessage.Option{" + 2553 "Code: " + printUint16(o.Code) + ", " + 2554 "Data: []byte{" + printByteSlice(o.Data) + "}}" 2555 } 2556 2557 func (r *OPTResource) realType() Type { 2558 return TypeOPT 2559 } 2560 2561 func (r *OPTResource) pack(msg []byte, compression map[string]int, compressionOff int) ([]byte, error) { 2562 for _, opt := range r.Options { 2563 msg = packUint16(msg, opt.Code) 2564 l := uint16(len(opt.Data)) 2565 msg = packUint16(msg, l) 2566 msg = packBytes(msg, opt.Data) 2567 } 2568 return msg, nil 2569 } 2570 2571 // GoString implements fmt.GoStringer.GoString. 2572 func (r *OPTResource) GoString() string { 2573 s := "dnsmessage.OPTResource{Options: []dnsmessage.Option{" 2574 if len(r.Options) == 0 { 2575 return s + "}}" 2576 } 2577 s += r.Options[0].GoString() 2578 for _, o := range r.Options[1:] { 2579 s += ", " + o.GoString() 2580 } 2581 return s + "}}" 2582 } 2583 2584 func unpackOPTResource(msg []byte, off int, length uint16) (OPTResource, error) { 2585 var opts []Option 2586 for oldOff := off; off < oldOff+int(length); { 2587 var err error 2588 var o Option 2589 o.Code, off, err = unpackUint16(msg, off) 2590 if err != nil { 2591 return OPTResource{}, &nestedError{"Code", err} 2592 } 2593 var l uint16 2594 l, off, err = unpackUint16(msg, off) 2595 if err != nil { 2596 return OPTResource{}, &nestedError{"Data", err} 2597 } 2598 o.Data = make([]byte, l) 2599 if copy(o.Data, msg[off:]) != int(l) { 2600 return OPTResource{}, &nestedError{"Data", errCalcLen} 2601 } 2602 off += int(l) 2603 opts = append(opts, o) 2604 } 2605 return OPTResource{opts}, nil 2606 }