route/vendor/golang.org/x/net/ipv4/control_pktinfo.go

40 lines
945 B
Go
Raw Normal View History

2017-03-26 19:50:20 +00:00
// Copyright 2014 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// +build darwin linux solaris
package ipv4
import (
2017-09-30 16:47:47 +00:00
"net"
2017-03-26 19:50:20 +00:00
"unsafe"
"golang.org/x/net/internal/iana"
2017-09-30 16:47:47 +00:00
"golang.org/x/net/internal/socket"
2017-03-26 19:50:20 +00:00
)
func marshalPacketInfo(b []byte, cm *ControlMessage) []byte {
2017-09-30 16:47:47 +00:00
m := socket.ControlMessage(b)
m.MarshalHeader(iana.ProtocolIP, sysIP_PKTINFO, sizeofInetPktinfo)
2017-03-26 19:50:20 +00:00
if cm != nil {
2017-09-30 16:47:47 +00:00
pi := (*inetPktinfo)(unsafe.Pointer(&m.Data(sizeofInetPktinfo)[0]))
2017-03-26 19:50:20 +00:00
if ip := cm.Src.To4(); ip != nil {
copy(pi.Spec_dst[:], ip)
}
if cm.IfIndex > 0 {
pi.setIfindex(cm.IfIndex)
}
}
2017-09-30 16:47:47 +00:00
return m.Next(sizeofInetPktinfo)
2017-03-26 19:50:20 +00:00
}
func parsePacketInfo(cm *ControlMessage, b []byte) {
pi := (*inetPktinfo)(unsafe.Pointer(&b[0]))
cm.IfIndex = int(pi.Ifindex)
2017-09-30 16:47:47 +00:00
if len(cm.Dst) < net.IPv4len {
cm.Dst = make(net.IP, net.IPv4len)
}
copy(cm.Dst, pi.Addr[:])
2017-03-26 19:50:20 +00:00
}