You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

msg_helpers.go 21KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804
  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 nsecx"}
  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 block"}
  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 block"}
  441. }
  442. if length > 32 {
  443. return nsec, len(msg), &Error{err: "NSEC block too long"}
  444. }
  445. if off+length > len(msg) {
  446. return nsec, len(msg), &Error{err: "overflowing NSEC block"}
  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. var lastwindow, lastlength uint16
  508. for _, t := range bitmap {
  509. window := t / 256
  510. length := (t-window*256)/8 + 1
  511. if window > lastwindow && lastlength != 0 { // New window, jump to the new offset
  512. off += int(lastlength) + 2
  513. lastlength = 0
  514. }
  515. if window < lastwindow || length < lastlength {
  516. return len(msg), &Error{err: "nsec bits out of order"}
  517. }
  518. if off+2+int(length) > len(msg) {
  519. return len(msg), &Error{err: "overflow packing nsec"}
  520. }
  521. // Setting the window #
  522. msg[off] = byte(window)
  523. // Setting the octets length
  524. msg[off+1] = byte(length)
  525. // Setting the bit value for the type in the right octet
  526. msg[off+1+int(length)] |= byte(1 << (7 - t%8))
  527. lastwindow, lastlength = window, length
  528. }
  529. off += int(lastlength) + 2
  530. return off, nil
  531. }
  532. func unpackDataSVCB(msg []byte, off int) ([]SVCBKeyValue, int, error) {
  533. var xs []SVCBKeyValue
  534. var code uint16
  535. var length uint16
  536. var err error
  537. for off < len(msg) {
  538. code, off, err = unpackUint16(msg, off)
  539. if err != nil {
  540. return nil, len(msg), &Error{err: "overflow unpacking SVCB"}
  541. }
  542. length, off, err = unpackUint16(msg, off)
  543. if err != nil || off+int(length) > len(msg) {
  544. return nil, len(msg), &Error{err: "overflow unpacking SVCB"}
  545. }
  546. e := makeSVCBKeyValue(SVCBKey(code))
  547. if e == nil {
  548. return nil, len(msg), &Error{err: "bad SVCB key"}
  549. }
  550. if err := e.unpack(msg[off : off+int(length)]); err != nil {
  551. return nil, len(msg), err
  552. }
  553. if len(xs) > 0 && e.Key() <= xs[len(xs)-1].Key() {
  554. return nil, len(msg), &Error{err: "SVCB keys not in strictly increasing order"}
  555. }
  556. xs = append(xs, e)
  557. off += int(length)
  558. }
  559. return xs, off, nil
  560. }
  561. func packDataSVCB(pairs []SVCBKeyValue, msg []byte, off int) (int, error) {
  562. pairs = append([]SVCBKeyValue(nil), pairs...)
  563. sort.Slice(pairs, func(i, j int) bool {
  564. return pairs[i].Key() < pairs[j].Key()
  565. })
  566. prev := svcb_RESERVED
  567. for _, el := range pairs {
  568. if el.Key() == prev {
  569. return len(msg), &Error{err: "repeated SVCB keys are not allowed"}
  570. }
  571. prev = el.Key()
  572. packed, err := el.pack()
  573. if err != nil {
  574. return len(msg), err
  575. }
  576. off, err = packUint16(uint16(el.Key()), msg, off)
  577. if err != nil {
  578. return len(msg), &Error{err: "overflow packing SVCB"}
  579. }
  580. off, err = packUint16(uint16(len(packed)), msg, off)
  581. if err != nil || off+len(packed) > len(msg) {
  582. return len(msg), &Error{err: "overflow packing SVCB"}
  583. }
  584. copy(msg[off:off+len(packed)], packed)
  585. off += len(packed)
  586. }
  587. return off, nil
  588. }
  589. func unpackDataDomainNames(msg []byte, off, end int) ([]string, int, error) {
  590. var (
  591. servers []string
  592. s string
  593. err error
  594. )
  595. if end > len(msg) {
  596. return nil, len(msg), &Error{err: "overflow unpacking domain names"}
  597. }
  598. for off < end {
  599. s, off, err = UnpackDomainName(msg, off)
  600. if err != nil {
  601. return servers, len(msg), err
  602. }
  603. servers = append(servers, s)
  604. }
  605. return servers, off, nil
  606. }
  607. func packDataDomainNames(names []string, msg []byte, off int, compression compressionMap, compress bool) (int, error) {
  608. var err error
  609. for _, name := range names {
  610. off, err = packDomainName(name, msg, off, compression, compress)
  611. if err != nil {
  612. return len(msg), err
  613. }
  614. }
  615. return off, nil
  616. }
  617. func packDataApl(data []APLPrefix, msg []byte, off int) (int, error) {
  618. var err error
  619. for i := range data {
  620. off, err = packDataAplPrefix(&data[i], msg, off)
  621. if err != nil {
  622. return len(msg), err
  623. }
  624. }
  625. return off, nil
  626. }
  627. func packDataAplPrefix(p *APLPrefix, msg []byte, off int) (int, error) {
  628. if len(p.Network.IP) != len(p.Network.Mask) {
  629. return len(msg), &Error{err: "address and mask lengths don't match"}
  630. }
  631. var err error
  632. prefix, _ := p.Network.Mask.Size()
  633. addr := p.Network.IP.Mask(p.Network.Mask)[:(prefix+7)/8]
  634. switch len(p.Network.IP) {
  635. case net.IPv4len:
  636. off, err = packUint16(1, msg, off)
  637. case net.IPv6len:
  638. off, err = packUint16(2, msg, off)
  639. default:
  640. err = &Error{err: "unrecognized address family"}
  641. }
  642. if err != nil {
  643. return len(msg), err
  644. }
  645. off, err = packUint8(uint8(prefix), msg, off)
  646. if err != nil {
  647. return len(msg), err
  648. }
  649. var n uint8
  650. if p.Negation {
  651. n = 0x80
  652. }
  653. // trim trailing zero bytes as specified in RFC3123 Sections 4.1 and 4.2.
  654. i := len(addr) - 1
  655. for ; i >= 0 && addr[i] == 0; i-- {
  656. }
  657. addr = addr[:i+1]
  658. adflen := uint8(len(addr)) & 0x7f
  659. off, err = packUint8(n|adflen, msg, off)
  660. if err != nil {
  661. return len(msg), err
  662. }
  663. if off+len(addr) > len(msg) {
  664. return len(msg), &Error{err: "overflow packing APL prefix"}
  665. }
  666. off += copy(msg[off:], addr)
  667. return off, nil
  668. }
  669. func unpackDataApl(msg []byte, off int) ([]APLPrefix, int, error) {
  670. var result []APLPrefix
  671. for off < len(msg) {
  672. prefix, end, err := unpackDataAplPrefix(msg, off)
  673. if err != nil {
  674. return nil, len(msg), err
  675. }
  676. off = end
  677. result = append(result, prefix)
  678. }
  679. return result, off, nil
  680. }
  681. func unpackDataAplPrefix(msg []byte, off int) (APLPrefix, int, error) {
  682. family, off, err := unpackUint16(msg, off)
  683. if err != nil {
  684. return APLPrefix{}, len(msg), &Error{err: "overflow unpacking APL prefix"}
  685. }
  686. prefix, off, err := unpackUint8(msg, off)
  687. if err != nil {
  688. return APLPrefix{}, len(msg), &Error{err: "overflow unpacking APL prefix"}
  689. }
  690. nlen, off, err := unpackUint8(msg, off)
  691. if err != nil {
  692. return APLPrefix{}, len(msg), &Error{err: "overflow unpacking APL prefix"}
  693. }
  694. var ip []byte
  695. switch family {
  696. case 1:
  697. ip = make([]byte, net.IPv4len)
  698. case 2:
  699. ip = make([]byte, net.IPv6len)
  700. default:
  701. return APLPrefix{}, len(msg), &Error{err: "unrecognized APL address family"}
  702. }
  703. if int(prefix) > 8*len(ip) {
  704. return APLPrefix{}, len(msg), &Error{err: "APL prefix too long"}
  705. }
  706. afdlen := int(nlen & 0x7f)
  707. if afdlen > len(ip) {
  708. return APLPrefix{}, len(msg), &Error{err: "APL length too long"}
  709. }
  710. if off+afdlen > len(msg) {
  711. return APLPrefix{}, len(msg), &Error{err: "overflow unpacking APL address"}
  712. }
  713. off += copy(ip, msg[off:off+afdlen])
  714. if afdlen > 0 {
  715. last := ip[afdlen-1]
  716. if last == 0 {
  717. return APLPrefix{}, len(msg), &Error{err: "extra APL address bits"}
  718. }
  719. }
  720. ipnet := net.IPNet{
  721. IP: ip,
  722. Mask: net.CIDRMask(int(prefix), 8*len(ip)),
  723. }
  724. network := ipnet.IP.Mask(ipnet.Mask)
  725. if !network.Equal(ipnet.IP) {
  726. return APLPrefix{}, len(msg), &Error{err: "invalid APL address length"}
  727. }
  728. return APLPrefix{
  729. Negation: (nlen & 0x80) != 0,
  730. Network: ipnet,
  731. }, off, nil
  732. }