mirror of
https://git.zx2c4.com/wireguard-go
synced 2024-11-15 01:05:15 +01:00
25eb973e00
Replace the src storage inside StdNetEndpoint with a copy of the raw control message buffer, to reduce allocation and perform less work on a per-packet basis. Signed-off-by: James Tucker <james@tailscale.com> Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
//go:build linux && !android
|
|
|
|
/* SPDX-License-Identifier: MIT
|
|
*
|
|
* Copyright (C) 2017-2023 WireGuard LLC. All Rights Reserved.
|
|
*/
|
|
|
|
package conn
|
|
|
|
import (
|
|
"net/netip"
|
|
"unsafe"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
func (e *StdNetEndpoint) SrcIP() netip.Addr {
|
|
switch len(e.src) {
|
|
case unix.CmsgSpace(unix.SizeofInet4Pktinfo):
|
|
info := (*unix.Inet4Pktinfo)(unsafe.Pointer(&e.src[unix.CmsgLen(0)]))
|
|
return netip.AddrFrom4(info.Spec_dst)
|
|
case unix.CmsgSpace(unix.SizeofInet6Pktinfo):
|
|
info := (*unix.Inet6Pktinfo)(unsafe.Pointer(&e.src[unix.CmsgLen(0)]))
|
|
// TODO: set zone. in order to do so we need to check if the address is
|
|
// link local, and if it is perform a syscall to turn the ifindex into a
|
|
// zone string because netip uses string zones.
|
|
return netip.AddrFrom16(info.Addr)
|
|
}
|
|
return netip.Addr{}
|
|
}
|
|
|
|
func (e *StdNetEndpoint) SrcIfidx() int32 {
|
|
switch len(e.src) {
|
|
case unix.CmsgSpace(unix.SizeofInet4Pktinfo):
|
|
info := (*unix.Inet4Pktinfo)(unsafe.Pointer(&e.src[unix.CmsgLen(0)]))
|
|
return info.Ifindex
|
|
case unix.CmsgSpace(unix.SizeofInet6Pktinfo):
|
|
info := (*unix.Inet6Pktinfo)(unsafe.Pointer(&e.src[unix.CmsgLen(0)]))
|
|
return int32(info.Ifindex)
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func (e *StdNetEndpoint) SrcToString() string {
|
|
return e.SrcIP().String()
|
|
}
|
|
|
|
// getSrcFromControl parses the control for PKTINFO and if found updates ep with
|
|
// the source information found.
|
|
func getSrcFromControl(control []byte, ep *StdNetEndpoint) {
|
|
ep.ClearSrc()
|
|
|
|
var (
|
|
hdr unix.Cmsghdr
|
|
data []byte
|
|
rem []byte = control
|
|
err error
|
|
)
|
|
|
|
for len(rem) > unix.SizeofCmsghdr {
|
|
hdr, data, rem, err = unix.ParseOneSocketControlMessage(rem)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if hdr.Level == unix.IPPROTO_IP &&
|
|
hdr.Type == unix.IP_PKTINFO {
|
|
|
|
if ep.src == nil || cap(ep.src) < unix.CmsgSpace(unix.SizeofInet4Pktinfo) {
|
|
ep.src = make([]byte, 0, unix.CmsgSpace(unix.SizeofInet4Pktinfo))
|
|
}
|
|
ep.src = ep.src[:unix.CmsgSpace(unix.SizeofInet4Pktinfo)]
|
|
|
|
hdrBuf := unsafe.Slice((*byte)(unsafe.Pointer(&hdr)), unix.SizeofCmsghdr)
|
|
copy(ep.src, hdrBuf)
|
|
copy(ep.src[unix.CmsgLen(0):], data)
|
|
return
|
|
}
|
|
|
|
if hdr.Level == unix.IPPROTO_IPV6 &&
|
|
hdr.Type == unix.IPV6_PKTINFO {
|
|
|
|
if ep.src == nil || cap(ep.src) < unix.CmsgSpace(unix.SizeofInet6Pktinfo) {
|
|
ep.src = make([]byte, 0, unix.CmsgSpace(unix.SizeofInet6Pktinfo))
|
|
}
|
|
|
|
ep.src = ep.src[:unix.CmsgSpace(unix.SizeofInet6Pktinfo)]
|
|
|
|
hdrBuf := unsafe.Slice((*byte)(unsafe.Pointer(&hdr)), unix.SizeofCmsghdr)
|
|
copy(ep.src, hdrBuf)
|
|
copy(ep.src[unix.CmsgLen(0):], data)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// setSrcControl sets an IP{V6}_PKTINFO in control based on the source address
|
|
// and source ifindex found in ep. control's len will be set to 0 in the event
|
|
// that ep is a default value.
|
|
func setSrcControl(control *[]byte, ep *StdNetEndpoint) {
|
|
if cap(*control) < len(ep.src) {
|
|
return
|
|
}
|
|
*control = (*control)[:0]
|
|
*control = append(*control, ep.src...)
|
|
}
|
|
|
|
var srcControlSize = unix.CmsgSpace(unix.SizeofInet6Pktinfo)
|
|
|
|
const StdNetSupportsStickySockets = true
|