github.com/bepass-org/wireguard-go@v1.0.4-rc2.0.20240304192354-ebce6572bc24/conn/controlfns_linux.go (about)

     1  /* SPDX-License-Identifier: MIT
     2   *
     3   * Copyright (C) 2017-2023 WireGuard LLC. All Rights Reserved.
     4   */
     5  
     6  package conn
     7  
     8  import (
     9  	"fmt"
    10  	"runtime"
    11  	"syscall"
    12  
    13  	"golang.org/x/sys/unix"
    14  )
    15  
    16  func init() {
    17  	controlFns = append(controlFns,
    18  
    19  		// Attempt to set the socket buffer size beyond net.core.{r,w}mem_max by
    20  		// using SO_*BUFFORCE. This requires CAP_NET_ADMIN, and is allowed here to
    21  		// fail silently - the result of failure is lower performance on very fast
    22  		// links or high latency links.
    23  		func(network, address string, c syscall.RawConn) error {
    24  			return c.Control(func(fd uintptr) {
    25  				// Set up to *mem_max
    26  				_ = unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_RCVBUF, socketBufferSize)
    27  				_ = unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_SNDBUF, socketBufferSize)
    28  				// Set beyond *mem_max if CAP_NET_ADMIN
    29  				_ = unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_RCVBUFFORCE, socketBufferSize)
    30  				_ = unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_SNDBUFFORCE, socketBufferSize)
    31  			})
    32  		},
    33  
    34  		// Enable receiving of the packet information (IP_PKTINFO for IPv4,
    35  		// IPV6_PKTINFO for IPv6) that is used to implement sticky socket support.
    36  		func(network, address string, c syscall.RawConn) error {
    37  			var err error
    38  			switch network {
    39  			case "udp4":
    40  				if runtime.GOOS != "android" {
    41  					c.Control(func(fd uintptr) {
    42  						err = unix.SetsockoptInt(int(fd), unix.IPPROTO_IP, unix.IP_PKTINFO, 1)
    43  					})
    44  				}
    45  			case "udp6":
    46  				c.Control(func(fd uintptr) {
    47  					if runtime.GOOS != "android" {
    48  						err = unix.SetsockoptInt(int(fd), unix.IPPROTO_IPV6, unix.IPV6_RECVPKTINFO, 1)
    49  						if err != nil {
    50  							return
    51  						}
    52  					}
    53  					err = unix.SetsockoptInt(int(fd), unix.IPPROTO_IPV6, unix.IPV6_V6ONLY, 1)
    54  				})
    55  			default:
    56  				err = fmt.Errorf("unhandled network: %s: %w", network, unix.EINVAL)
    57  			}
    58  			return err
    59  		},
    60  
    61  		// Attempt to enable UDP_GRO
    62  		func(network, address string, c syscall.RawConn) error {
    63  			c.Control(func(fd uintptr) {
    64  				_ = unix.SetsockoptInt(int(fd), unix.IPPROTO_UDP, unix.UDP_GRO, 1)
    65  			})
    66  			return nil
    67  		},
    68  	)
    69  }