control_rfc3542_unix.go 2.7 KB

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