control_rfc3542_unix.go 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596
  1. // Copyright 2013 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. //go:build aix || darwin || dragonfly || freebsd || linux || netbsd || openbsd || solaris || zos
  5. package ipv6
  6. import (
  7. "net"
  8. "unsafe"
  9. "golang.org/x/net/internal/iana"
  10. "golang.org/x/net/internal/socket"
  11. "golang.org/x/sys/unix"
  12. )
  13. func marshalTrafficClass(b []byte, cm *ControlMessage) []byte {
  14. m := socket.ControlMessage(b)
  15. m.MarshalHeader(iana.ProtocolIPv6, unix.IPV6_TCLASS, 4)
  16. if cm != nil {
  17. socket.NativeEndian.PutUint32(m.Data(4), uint32(cm.TrafficClass))
  18. }
  19. return m.Next(4)
  20. }
  21. func parseTrafficClass(cm *ControlMessage, b []byte) {
  22. cm.TrafficClass = int(socket.NativeEndian.Uint32(b[:4]))
  23. }
  24. func marshalHopLimit(b []byte, cm *ControlMessage) []byte {
  25. m := socket.ControlMessage(b)
  26. m.MarshalHeader(iana.ProtocolIPv6, unix.IPV6_HOPLIMIT, 4)
  27. if cm != nil {
  28. socket.NativeEndian.PutUint32(m.Data(4), uint32(cm.HopLimit))
  29. }
  30. return m.Next(4)
  31. }
  32. func parseHopLimit(cm *ControlMessage, b []byte) {
  33. cm.HopLimit = int(socket.NativeEndian.Uint32(b[:4]))
  34. }
  35. func marshalPacketInfo(b []byte, cm *ControlMessage) []byte {
  36. m := socket.ControlMessage(b)
  37. m.MarshalHeader(iana.ProtocolIPv6, unix.IPV6_PKTINFO, sizeofInet6Pktinfo)
  38. if cm != nil {
  39. pi := (*inet6Pktinfo)(unsafe.Pointer(&m.Data(sizeofInet6Pktinfo)[0]))
  40. if ip := cm.Src.To16(); ip != nil && ip.To4() == nil {
  41. copy(pi.Addr[:], ip)
  42. }
  43. if cm.IfIndex > 0 {
  44. pi.setIfindex(cm.IfIndex)
  45. }
  46. }
  47. return m.Next(sizeofInet6Pktinfo)
  48. }
  49. func parsePacketInfo(cm *ControlMessage, b []byte) {
  50. pi := (*inet6Pktinfo)(unsafe.Pointer(&b[0]))
  51. if len(cm.Dst) < net.IPv6len {
  52. cm.Dst = make(net.IP, net.IPv6len)
  53. }
  54. copy(cm.Dst, pi.Addr[:])
  55. cm.IfIndex = int(pi.Ifindex)
  56. }
  57. func marshalNextHop(b []byte, cm *ControlMessage) []byte {
  58. m := socket.ControlMessage(b)
  59. m.MarshalHeader(iana.ProtocolIPv6, unix.IPV6_NEXTHOP, sizeofSockaddrInet6)
  60. if cm != nil {
  61. sa := (*sockaddrInet6)(unsafe.Pointer(&m.Data(sizeofSockaddrInet6)[0]))
  62. sa.setSockaddr(cm.NextHop, cm.IfIndex)
  63. }
  64. return m.Next(sizeofSockaddrInet6)
  65. }
  66. func parseNextHop(cm *ControlMessage, b []byte) {
  67. }
  68. func marshalPathMTU(b []byte, cm *ControlMessage) []byte {
  69. m := socket.ControlMessage(b)
  70. m.MarshalHeader(iana.ProtocolIPv6, unix.IPV6_PATHMTU, sizeofIPv6Mtuinfo)
  71. return m.Next(sizeofIPv6Mtuinfo)
  72. }
  73. func parsePathMTU(cm *ControlMessage, b []byte) {
  74. mi := (*ipv6Mtuinfo)(unsafe.Pointer(&b[0]))
  75. if len(cm.Dst) < net.IPv6len {
  76. cm.Dst = make(net.IP, net.IPv6len)
  77. }
  78. copy(cm.Dst, mi.Addr.Addr[:])
  79. cm.IfIndex = int(mi.Addr.Scope_id)
  80. cm.MTU = int(mi.Mtu)
  81. }