msg_helpers.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812
  1. package dns
  2. import (
  3. "encoding/base32"
  4. "encoding/base64"
  5. "encoding/binary"
  6. "encoding/hex"
  7. "net"
  8. "sort"
  9. "strings"
  10. )
  11. // helper functions called from the generated zmsg.go
  12. // These function are named after the tag to help pack/unpack, if there is no tag it is the name
  13. // of the type they pack/unpack (string, int, etc). We prefix all with unpackData or packData, so packDataA or
  14. // packDataDomainName.
  15. func unpackDataA(msg []byte, off int) (net.IP, int, error) {
  16. if off+net.IPv4len > len(msg) {
  17. return nil, len(msg), &Error{err: "overflow unpacking a"}
  18. }
  19. a := append(make(net.IP, 0, net.IPv4len), msg[off:off+net.IPv4len]...)
  20. off += net.IPv4len
  21. return a, off, nil
  22. }
  23. func packDataA(a net.IP, msg []byte, off int) (int, error) {
  24. switch len(a) {
  25. case net.IPv4len, net.IPv6len:
  26. // It must be a slice of 4, even if it is 16, we encode only the first 4
  27. if off+net.IPv4len > len(msg) {
  28. return len(msg), &Error{err: "overflow packing a"}
  29. }
  30. copy(msg[off:], a.To4())
  31. off += net.IPv4len
  32. case 0:
  33. // Allowed, for dynamic updates.
  34. default:
  35. return len(msg), &Error{err: "overflow packing a"}
  36. }
  37. return off, nil
  38. }
  39. func unpackDataAAAA(msg []byte, off int) (net.IP, int, error) {
  40. if off+net.IPv6len > len(msg) {
  41. return nil, len(msg), &Error{err: "overflow unpacking aaaa"}
  42. }
  43. aaaa := append(make(net.IP, 0, net.IPv6len), msg[off:off+net.IPv6len]...)
  44. off += net.IPv6len
  45. return aaaa, off, nil
  46. }
  47. func packDataAAAA(aaaa net.IP, msg []byte, off int) (int, error) {
  48. switch len(aaaa) {
  49. case net.IPv6len:
  50. if off+net.IPv6len > len(msg) {
  51. return len(msg), &Error{err: "overflow packing aaaa"}
  52. }
  53. copy(msg[off:], aaaa)
  54. off += net.IPv6len
  55. case 0:
  56. // Allowed, dynamic updates.
  57. default:
  58. return len(msg), &Error{err: "overflow packing aaaa"}
  59. }
  60. return off, nil
  61. }
  62. // unpackHeader unpacks an RR header, returning the offset to the end of the header and a
  63. // re-sliced msg according to the expected length of the RR.
  64. func unpackHeader(msg []byte, off int) (rr RR_Header, off1 int, truncmsg []byte, err error) {
  65. hdr := RR_Header{}
  66. if off == len(msg) {
  67. return hdr, off, msg, nil
  68. }
  69. hdr.Name, off, err = UnpackDomainName(msg, off)
  70. if err != nil {
  71. return hdr, len(msg), msg, err
  72. }
  73. hdr.Rrtype, off, err = unpackUint16(msg, off)
  74. if err != nil {
  75. return hdr, len(msg), msg, err
  76. }
  77. hdr.Class, off, err = unpackUint16(msg, off)
  78. if err != nil {
  79. return hdr, len(msg), msg, err
  80. }
  81. hdr.Ttl, off, err = unpackUint32(msg, off)
  82. if err != nil {
  83. return hdr, len(msg), msg, err
  84. }
  85. hdr.Rdlength, off, err = unpackUint16(msg, off)
  86. if err != nil {
  87. return hdr, len(msg), msg, err
  88. }
  89. msg, err = truncateMsgFromRdlength(msg, off, hdr.Rdlength)
  90. return hdr, off, msg, err
  91. }
  92. // packHeader packs an RR header, returning the offset to the end of the header.
  93. // See PackDomainName for documentation about the compression.
  94. func (hdr RR_Header) packHeader(msg []byte, off int, compression compressionMap, compress bool) (int, error) {
  95. if off == len(msg) {
  96. return off, nil
  97. }
  98. off, err := packDomainName(hdr.Name, msg, off, compression, compress)
  99. if err != nil {
  100. return len(msg), err
  101. }
  102. off, err = packUint16(hdr.Rrtype, msg, off)
  103. if err != nil {
  104. return len(msg), err
  105. }
  106. off, err = packUint16(hdr.Class, msg, off)
  107. if err != nil {
  108. return len(msg), err
  109. }
  110. off, err = packUint32(hdr.Ttl, msg, off)
  111. if err != nil {
  112. return len(msg), err
  113. }
  114. off, err = packUint16(0, msg, off) // The RDLENGTH field will be set later in packRR.
  115. if err != nil {
  116. return len(msg), err
  117. }
  118. return off, nil
  119. }
  120. // helper helper functions.
  121. // truncateMsgFromRdLength truncates msg to match the expected length of the RR.
  122. // Returns an error if msg is smaller than the expected size.
  123. func truncateMsgFromRdlength(msg []byte, off int, rdlength uint16) (truncmsg []byte, err error) {
  124. lenrd := off + int(rdlength)
  125. if lenrd > len(msg) {
  126. return msg, &Error{err: "overflowing header size"}
  127. }
  128. return msg[:lenrd], nil
  129. }
  130. var base32HexNoPadEncoding = base32.HexEncoding.WithPadding(base32.NoPadding)
  131. func fromBase32(s []byte) (buf []byte, err error) {
  132. for i, b := range s {
  133. if b >= 'a' && b <= 'z' {
  134. s[i] = b - 32
  135. }
  136. }
  137. buflen := base32HexNoPadEncoding.DecodedLen(len(s))
  138. buf = make([]byte, buflen)
  139. n, err := base32HexNoPadEncoding.Decode(buf, s)
  140. buf = buf[:n]
  141. return
  142. }
  143. func toBase32(b []byte) string {
  144. return base32HexNoPadEncoding.EncodeToString(b)
  145. }
  146. func fromBase64(s []byte) (buf []byte, err error) {
  147. buflen := base64.StdEncoding.DecodedLen(len(s))
  148. buf = make([]byte, buflen)
  149. n, err := base64.StdEncoding.Decode(buf, s)
  150. buf = buf[:n]
  151. return
  152. }
  153. func toBase64(b []byte) string { return base64.StdEncoding.EncodeToString(b) }
  154. // dynamicUpdate returns true if the Rdlength is zero.
  155. func noRdata(h RR_Header) bool { return h.Rdlength == 0 }
  156. func unpackUint8(msg []byte, off int) (i uint8, off1 int, err error) {
  157. if off+1 > len(msg) {
  158. return 0, len(msg), &Error{err: "overflow unpacking uint8"}
  159. }
  160. return msg[off], off + 1, nil
  161. }
  162. func packUint8(i uint8, msg []byte, off int) (off1 int, err error) {
  163. if off+1 > len(msg) {
  164. return len(msg), &Error{err: "overflow packing uint8"}
  165. }
  166. msg[off] = i
  167. return off + 1, nil
  168. }
  169. func unpackUint16(msg []byte, off int) (i uint16, off1 int, err error) {
  170. if off+2 > len(msg) {
  171. return 0, len(msg), &Error{err: "overflow unpacking uint16"}
  172. }
  173. return binary.BigEndian.Uint16(msg[off:]), off + 2, nil
  174. }
  175. func packUint16(i uint16, msg []byte, off int) (off1 int, err error) {
  176. if off+2 > len(msg) {
  177. return len(msg), &Error{err: "overflow packing uint16"}
  178. }
  179. binary.BigEndian.PutUint16(msg[off:], i)
  180. return off + 2, nil
  181. }
  182. func unpackUint32(msg []byte, off int) (i uint32, off1 int, err error) {
  183. if off+4 > len(msg) {
  184. return 0, len(msg), &Error{err: "overflow unpacking uint32"}
  185. }
  186. return binary.BigEndian.Uint32(msg[off:]), off + 4, nil
  187. }
  188. func packUint32(i uint32, msg []byte, off int) (off1 int, err error) {
  189. if off+4 > len(msg) {
  190. return len(msg), &Error{err: "overflow packing uint32"}
  191. }
  192. binary.BigEndian.PutUint32(msg[off:], i)
  193. return off + 4, nil
  194. }
  195. func unpackUint48(msg []byte, off int) (i uint64, off1 int, err error) {
  196. if off+6 > len(msg) {
  197. return 0, len(msg), &Error{err: "overflow unpacking uint64 as uint48"}
  198. }
  199. // Used in TSIG where the last 48 bits are occupied, so for now, assume a uint48 (6 bytes)
  200. i = uint64(msg[off])<<40 | uint64(msg[off+1])<<32 | uint64(msg[off+2])<<24 | uint64(msg[off+3])<<16 |
  201. uint64(msg[off+4])<<8 | uint64(msg[off+5])
  202. off += 6
  203. return i, off, nil
  204. }
  205. func packUint48(i uint64, msg []byte, off int) (off1 int, err error) {
  206. if off+6 > len(msg) {
  207. return len(msg), &Error{err: "overflow packing uint64 as uint48"}
  208. }
  209. msg[off] = byte(i >> 40)
  210. msg[off+1] = byte(i >> 32)
  211. msg[off+2] = byte(i >> 24)
  212. msg[off+3] = byte(i >> 16)
  213. msg[off+4] = byte(i >> 8)
  214. msg[off+5] = byte(i)
  215. off += 6
  216. return off, nil
  217. }
  218. func unpackUint64(msg []byte, off int) (i uint64, off1 int, err error) {
  219. if off+8 > len(msg) {
  220. return 0, len(msg), &Error{err: "overflow unpacking uint64"}
  221. }
  222. return binary.BigEndian.Uint64(msg[off:]), off + 8, nil
  223. }
  224. func packUint64(i uint64, msg []byte, off int) (off1 int, err error) {
  225. if off+8 > len(msg) {
  226. return len(msg), &Error{err: "overflow packing uint64"}
  227. }
  228. binary.BigEndian.PutUint64(msg[off:], i)
  229. off += 8
  230. return off, nil
  231. }
  232. func unpackString(msg []byte, off int) (string, int, error) {
  233. if off+1 > len(msg) {
  234. return "", off, &Error{err: "overflow unpacking txt"}
  235. }
  236. l := int(msg[off])
  237. off++
  238. if off+l > len(msg) {
  239. return "", off, &Error{err: "overflow unpacking txt"}
  240. }
  241. var s strings.Builder
  242. consumed := 0
  243. for i, b := range msg[off : off+l] {
  244. switch {
  245. case b == '"' || b == '\\':
  246. if consumed == 0 {
  247. s.Grow(l * 2)
  248. }
  249. s.Write(msg[off+consumed : off+i])
  250. s.WriteByte('\\')
  251. s.WriteByte(b)
  252. consumed = i + 1
  253. case b < ' ' || b > '~': // unprintable
  254. if consumed == 0 {
  255. s.Grow(l * 2)
  256. }
  257. s.Write(msg[off+consumed : off+i])
  258. s.WriteString(escapeByte(b))
  259. consumed = i + 1
  260. }
  261. }
  262. if consumed == 0 { // no escaping needed
  263. return string(msg[off : off+l]), off + l, nil
  264. }
  265. s.Write(msg[off+consumed : off+l])
  266. return s.String(), off + l, nil
  267. }
  268. func packString(s string, msg []byte, off int) (int, error) {
  269. txtTmp := make([]byte, 256*4+1)
  270. off, err := packTxtString(s, msg, off, txtTmp)
  271. if err != nil {
  272. return len(msg), err
  273. }
  274. return off, nil
  275. }
  276. func unpackStringBase32(msg []byte, off, end int) (string, int, error) {
  277. if end > len(msg) {
  278. return "", len(msg), &Error{err: "overflow unpacking base32"}
  279. }
  280. s := toBase32(msg[off:end])
  281. return s, end, nil
  282. }
  283. func packStringBase32(s string, msg []byte, off int) (int, error) {
  284. b32, err := fromBase32([]byte(s))
  285. if err != nil {
  286. return len(msg), err
  287. }
  288. if off+len(b32) > len(msg) {
  289. return len(msg), &Error{err: "overflow packing base32"}
  290. }
  291. copy(msg[off:off+len(b32)], b32)
  292. off += len(b32)
  293. return off, nil
  294. }
  295. func unpackStringBase64(msg []byte, off, end int) (string, int, error) {
  296. // Rest of the RR is base64 encoded value, so we don't need an explicit length
  297. // to be set. Thus far all RR's that have base64 encoded fields have those as their
  298. // last one. What we do need is the end of the RR!
  299. if end > len(msg) {
  300. return "", len(msg), &Error{err: "overflow unpacking base64"}
  301. }
  302. s := toBase64(msg[off:end])
  303. return s, end, nil
  304. }
  305. func packStringBase64(s string, msg []byte, off int) (int, error) {
  306. b64, err := fromBase64([]byte(s))
  307. if err != nil {
  308. return len(msg), err
  309. }
  310. if off+len(b64) > len(msg) {
  311. return len(msg), &Error{err: "overflow packing base64"}
  312. }
  313. copy(msg[off:off+len(b64)], b64)
  314. off += len(b64)
  315. return off, nil
  316. }
  317. func unpackStringHex(msg []byte, off, end int) (string, int, error) {
  318. // Rest of the RR is hex encoded value, so we don't need an explicit length
  319. // to be set. NSEC and TSIG have hex fields with a length field.
  320. // What we do need is the end of the RR!
  321. if end > len(msg) {
  322. return "", len(msg), &Error{err: "overflow unpacking hex"}
  323. }
  324. s := hex.EncodeToString(msg[off:end])
  325. return s, end, nil
  326. }
  327. func packStringHex(s string, msg []byte, off int) (int, error) {
  328. h, err := hex.DecodeString(s)
  329. if err != nil {
  330. return len(msg), err
  331. }
  332. if off+len(h) > len(msg) {
  333. return len(msg), &Error{err: "overflow packing hex"}
  334. }
  335. copy(msg[off:off+len(h)], h)
  336. off += len(h)
  337. return off, nil
  338. }
  339. func unpackStringAny(msg []byte, off, end int) (string, int, error) {
  340. if end > len(msg) {
  341. return "", len(msg), &Error{err: "overflow unpacking anything"}
  342. }
  343. return string(msg[off:end]), end, nil
  344. }
  345. func packStringAny(s string, msg []byte, off int) (int, error) {
  346. if off+len(s) > len(msg) {
  347. return len(msg), &Error{err: "overflow packing anything"}
  348. }
  349. copy(msg[off:off+len(s)], s)
  350. off += len(s)
  351. return off, nil
  352. }
  353. func unpackStringTxt(msg []byte, off int) ([]string, int, error) {
  354. txt, off, err := unpackTxt(msg, off)
  355. if err != nil {
  356. return nil, len(msg), err
  357. }
  358. return txt, off, nil
  359. }
  360. func packStringTxt(s []string, msg []byte, off int) (int, error) {
  361. txtTmp := make([]byte, 256*4+1) // If the whole string consists out of \DDD we need this many.
  362. off, err := packTxt(s, msg, off, txtTmp)
  363. if err != nil {
  364. return len(msg), err
  365. }
  366. return off, nil
  367. }
  368. func unpackDataOpt(msg []byte, off int) ([]EDNS0, int, error) {
  369. var edns []EDNS0
  370. Option:
  371. var code uint16
  372. if off+4 > len(msg) {
  373. return nil, len(msg), &Error{err: "overflow unpacking opt"}
  374. }
  375. code = binary.BigEndian.Uint16(msg[off:])
  376. off += 2
  377. optlen := binary.BigEndian.Uint16(msg[off:])
  378. off += 2
  379. if off+int(optlen) > len(msg) {
  380. return nil, len(msg), &Error{err: "overflow unpacking opt"}
  381. }
  382. e := makeDataOpt(code)
  383. if err := e.unpack(msg[off : off+int(optlen)]); err != nil {
  384. return nil, len(msg), err
  385. }
  386. edns = append(edns, e)
  387. off += int(optlen)
  388. if off < len(msg) {
  389. goto Option
  390. }
  391. return edns, off, nil
  392. }
  393. func packDataOpt(options []EDNS0, msg []byte, off int) (int, error) {
  394. for _, el := range options {
  395. b, err := el.pack()
  396. if err != nil || off+4 > len(msg) {
  397. return len(msg), &Error{err: "overflow packing opt"}
  398. }
  399. binary.BigEndian.PutUint16(msg[off:], el.Option()) // Option code
  400. binary.BigEndian.PutUint16(msg[off+2:], uint16(len(b))) // Length
  401. off += 4
  402. if off+len(b) > len(msg) {
  403. return len(msg), &Error{err: "overflow packing opt"}
  404. }
  405. // Actual data
  406. copy(msg[off:off+len(b)], b)
  407. off += len(b)
  408. }
  409. return off, nil
  410. }
  411. func unpackStringOctet(msg []byte, off int) (string, int, error) {
  412. s := string(msg[off:])
  413. return s, len(msg), nil
  414. }
  415. func packStringOctet(s string, msg []byte, off int) (int, error) {
  416. txtTmp := make([]byte, 256*4+1)
  417. off, err := packOctetString(s, msg, off, txtTmp)
  418. if err != nil {
  419. return len(msg), err
  420. }
  421. return off, nil
  422. }
  423. func unpackDataNsec(msg []byte, off int) ([]uint16, int, error) {
  424. var nsec []uint16
  425. length, window, lastwindow := 0, 0, -1
  426. for off < len(msg) {
  427. if off+2 > len(msg) {
  428. return nsec, len(msg), &Error{err: "overflow unpacking NSEC(3)"}
  429. }
  430. window = int(msg[off])
  431. length = int(msg[off+1])
  432. off += 2
  433. if window <= lastwindow {
  434. // RFC 4034: Blocks are present in the NSEC RR RDATA in
  435. // increasing numerical order.
  436. return nsec, len(msg), &Error{err: "out of order NSEC(3) block in type bitmap"}
  437. }
  438. if length == 0 {
  439. // RFC 4034: Blocks with no types present MUST NOT be included.
  440. return nsec, len(msg), &Error{err: "empty NSEC(3) block in type bitmap"}
  441. }
  442. if length > 32 {
  443. return nsec, len(msg), &Error{err: "NSEC(3) block too long in type bitmap"}
  444. }
  445. if off+length > len(msg) {
  446. return nsec, len(msg), &Error{err: "overflowing NSEC(3) block in type bitmap"}
  447. }
  448. // Walk the bytes in the window and extract the type bits
  449. for j, b := range msg[off : off+length] {
  450. // Check the bits one by one, and set the type
  451. if b&0x80 == 0x80 {
  452. nsec = append(nsec, uint16(window*256+j*8+0))
  453. }
  454. if b&0x40 == 0x40 {
  455. nsec = append(nsec, uint16(window*256+j*8+1))
  456. }
  457. if b&0x20 == 0x20 {
  458. nsec = append(nsec, uint16(window*256+j*8+2))
  459. }
  460. if b&0x10 == 0x10 {
  461. nsec = append(nsec, uint16(window*256+j*8+3))
  462. }
  463. if b&0x8 == 0x8 {
  464. nsec = append(nsec, uint16(window*256+j*8+4))
  465. }
  466. if b&0x4 == 0x4 {
  467. nsec = append(nsec, uint16(window*256+j*8+5))
  468. }
  469. if b&0x2 == 0x2 {
  470. nsec = append(nsec, uint16(window*256+j*8+6))
  471. }
  472. if b&0x1 == 0x1 {
  473. nsec = append(nsec, uint16(window*256+j*8+7))
  474. }
  475. }
  476. off += length
  477. lastwindow = window
  478. }
  479. return nsec, off, nil
  480. }
  481. // typeBitMapLen is a helper function which computes the "maximum" length of
  482. // a the NSEC Type BitMap field.
  483. func typeBitMapLen(bitmap []uint16) int {
  484. var l int
  485. var lastwindow, lastlength uint16
  486. for _, t := range bitmap {
  487. window := t / 256
  488. length := (t-window*256)/8 + 1
  489. if window > lastwindow && lastlength != 0 { // New window, jump to the new offset
  490. l += int(lastlength) + 2
  491. lastlength = 0
  492. }
  493. if window < lastwindow || length < lastlength {
  494. // packDataNsec would return Error{err: "nsec bits out of order"} here, but
  495. // when computing the length, we want do be liberal.
  496. continue
  497. }
  498. lastwindow, lastlength = window, length
  499. }
  500. l += int(lastlength) + 2
  501. return l
  502. }
  503. func packDataNsec(bitmap []uint16, msg []byte, off int) (int, error) {
  504. if len(bitmap) == 0 {
  505. return off, nil
  506. }
  507. if off > len(msg) {
  508. return off, &Error{err: "overflow packing nsec"}
  509. }
  510. toZero := msg[off:]
  511. if maxLen := typeBitMapLen(bitmap); maxLen < len(toZero) {
  512. toZero = toZero[:maxLen]
  513. }
  514. for i := range toZero {
  515. toZero[i] = 0
  516. }
  517. var lastwindow, lastlength uint16
  518. for _, t := range bitmap {
  519. window := t / 256
  520. length := (t-window*256)/8 + 1
  521. if window > lastwindow && lastlength != 0 { // New window, jump to the new offset
  522. off += int(lastlength) + 2
  523. lastlength = 0
  524. }
  525. if window < lastwindow || length < lastlength {
  526. return len(msg), &Error{err: "nsec bits out of order"}
  527. }
  528. if off+2+int(length) > len(msg) {
  529. return len(msg), &Error{err: "overflow packing nsec"}
  530. }
  531. // Setting the window #
  532. msg[off] = byte(window)
  533. // Setting the octets length
  534. msg[off+1] = byte(length)
  535. // Setting the bit value for the type in the right octet
  536. msg[off+1+int(length)] |= byte(1 << (7 - t%8))
  537. lastwindow, lastlength = window, length
  538. }
  539. off += int(lastlength) + 2
  540. return off, nil
  541. }
  542. func unpackDataSVCB(msg []byte, off int) ([]SVCBKeyValue, int, error) {
  543. var xs []SVCBKeyValue
  544. var code uint16
  545. var length uint16
  546. var err error
  547. for off < len(msg) {
  548. code, off, err = unpackUint16(msg, off)
  549. if err != nil {
  550. return nil, len(msg), &Error{err: "overflow unpacking SVCB"}
  551. }
  552. length, off, err = unpackUint16(msg, off)
  553. if err != nil || off+int(length) > len(msg) {
  554. return nil, len(msg), &Error{err: "overflow unpacking SVCB"}
  555. }
  556. e := makeSVCBKeyValue(SVCBKey(code))
  557. if e == nil {
  558. return nil, len(msg), &Error{err: "bad SVCB key"}
  559. }
  560. if err := e.unpack(msg[off : off+int(length)]); err != nil {
  561. return nil, len(msg), err
  562. }
  563. if len(xs) > 0 && e.Key() <= xs[len(xs)-1].Key() {
  564. return nil, len(msg), &Error{err: "SVCB keys not in strictly increasing order"}
  565. }
  566. xs = append(xs, e)
  567. off += int(length)
  568. }
  569. return xs, off, nil
  570. }
  571. func packDataSVCB(pairs []SVCBKeyValue, msg []byte, off int) (int, error) {
  572. pairs = append([]SVCBKeyValue(nil), pairs...)
  573. sort.Slice(pairs, func(i, j int) bool {
  574. return pairs[i].Key() < pairs[j].Key()
  575. })
  576. prev := svcb_RESERVED
  577. for _, el := range pairs {
  578. if el.Key() == prev {
  579. return len(msg), &Error{err: "repeated SVCB keys are not allowed"}
  580. }
  581. prev = el.Key()
  582. packed, err := el.pack()
  583. if err != nil {
  584. return len(msg), err
  585. }
  586. off, err = packUint16(uint16(el.Key()), msg, off)
  587. if err != nil {
  588. return len(msg), &Error{err: "overflow packing SVCB"}
  589. }
  590. off, err = packUint16(uint16(len(packed)), msg, off)
  591. if err != nil || off+len(packed) > len(msg) {
  592. return len(msg), &Error{err: "overflow packing SVCB"}
  593. }
  594. copy(msg[off:off+len(packed)], packed)
  595. off += len(packed)
  596. }
  597. return off, nil
  598. }
  599. func unpackDataDomainNames(msg []byte, off, end int) ([]string, int, error) {
  600. var (
  601. servers []string
  602. s string
  603. err error
  604. )
  605. if end > len(msg) {
  606. return nil, len(msg), &Error{err: "overflow unpacking domain names"}
  607. }
  608. for off < end {
  609. s, off, err = UnpackDomainName(msg, off)
  610. if err != nil {
  611. return servers, len(msg), err
  612. }
  613. servers = append(servers, s)
  614. }
  615. return servers, off, nil
  616. }
  617. func packDataDomainNames(names []string, msg []byte, off int, compression compressionMap, compress bool) (int, error) {
  618. var err error
  619. for _, name := range names {
  620. off, err = packDomainName(name, msg, off, compression, compress)
  621. if err != nil {
  622. return len(msg), err
  623. }
  624. }
  625. return off, nil
  626. }
  627. func packDataApl(data []APLPrefix, msg []byte, off int) (int, error) {
  628. var err error
  629. for i := range data {
  630. off, err = packDataAplPrefix(&data[i], msg, off)
  631. if err != nil {
  632. return len(msg), err
  633. }
  634. }
  635. return off, nil
  636. }
  637. func packDataAplPrefix(p *APLPrefix, msg []byte, off int) (int, error) {
  638. if len(p.Network.IP) != len(p.Network.Mask) {
  639. return len(msg), &Error{err: "address and mask lengths don't match"}
  640. }
  641. var err error
  642. prefix, _ := p.Network.Mask.Size()
  643. addr := p.Network.IP.Mask(p.Network.Mask)[:(prefix+7)/8]
  644. switch len(p.Network.IP) {
  645. case net.IPv4len:
  646. off, err = packUint16(1, msg, off)
  647. case net.IPv6len:
  648. off, err = packUint16(2, msg, off)
  649. default:
  650. err = &Error{err: "unrecognized address family"}
  651. }
  652. if err != nil {
  653. return len(msg), err
  654. }
  655. off, err = packUint8(uint8(prefix), msg, off)
  656. if err != nil {
  657. return len(msg), err
  658. }
  659. var n uint8
  660. if p.Negation {
  661. n = 0x80
  662. }
  663. // trim trailing zero bytes as specified in RFC3123 Sections 4.1 and 4.2.
  664. i := len(addr) - 1
  665. for ; i >= 0 && addr[i] == 0; i-- {
  666. }
  667. addr = addr[:i+1]
  668. adflen := uint8(len(addr)) & 0x7f
  669. off, err = packUint8(n|adflen, msg, off)
  670. if err != nil {
  671. return len(msg), err
  672. }
  673. if off+len(addr) > len(msg) {
  674. return len(msg), &Error{err: "overflow packing APL prefix"}
  675. }
  676. off += copy(msg[off:], addr)
  677. return off, nil
  678. }
  679. func unpackDataApl(msg []byte, off int) ([]APLPrefix, int, error) {
  680. var result []APLPrefix
  681. for off < len(msg) {
  682. prefix, end, err := unpackDataAplPrefix(msg, off)
  683. if err != nil {
  684. return nil, len(msg), err
  685. }
  686. off = end
  687. result = append(result, prefix)
  688. }
  689. return result, off, nil
  690. }
  691. func unpackDataAplPrefix(msg []byte, off int) (APLPrefix, int, error) {
  692. family, off, err := unpackUint16(msg, off)
  693. if err != nil {
  694. return APLPrefix{}, len(msg), &Error{err: "overflow unpacking APL prefix"}
  695. }
  696. prefix, off, err := unpackUint8(msg, off)
  697. if err != nil {
  698. return APLPrefix{}, len(msg), &Error{err: "overflow unpacking APL prefix"}
  699. }
  700. nlen, off, err := unpackUint8(msg, off)
  701. if err != nil {
  702. return APLPrefix{}, len(msg), &Error{err: "overflow unpacking APL prefix"}
  703. }
  704. var ip []byte
  705. switch family {
  706. case 1:
  707. ip = make([]byte, net.IPv4len)
  708. case 2:
  709. ip = make([]byte, net.IPv6len)
  710. default:
  711. return APLPrefix{}, len(msg), &Error{err: "unrecognized APL address family"}
  712. }
  713. if int(prefix) > 8*len(ip) {
  714. return APLPrefix{}, len(msg), &Error{err: "APL prefix too long"}
  715. }
  716. afdlen := int(nlen & 0x7f)
  717. if afdlen > len(ip) {
  718. return APLPrefix{}, len(msg), &Error{err: "APL length too long"}
  719. }
  720. if off+afdlen > len(msg) {
  721. return APLPrefix{}, len(msg), &Error{err: "overflow unpacking APL address"}
  722. }
  723. // Address MUST NOT contain trailing zero bytes per RFC3123 Sections 4.1 and 4.2.
  724. off += copy(ip, msg[off:off+afdlen])
  725. if afdlen > 0 {
  726. last := ip[afdlen-1]
  727. if last == 0 {
  728. return APLPrefix{}, len(msg), &Error{err: "extra APL address bits"}
  729. }
  730. }
  731. ipnet := net.IPNet{
  732. IP: ip,
  733. Mask: net.CIDRMask(int(prefix), 8*len(ip)),
  734. }
  735. return APLPrefix{
  736. Negation: (nlen & 0x80) != 0,
  737. Network: ipnet,
  738. }, off, nil
  739. }