mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-12 09:50:47 +01:00
9203690033
Code cleaning around the util/net package. The goal was to write a more understandable source code but modify nothing on the logic. Protect the WireGuard UDP listeners with marks. The implementation can support the VPN permission revocation events in thread safe way. It will be important if we start to support the running time route and DNS update features. - uniformize the file name convention: [struct_name] _ [functions] _ [os].go - code cleaning in net_linux.go - move env variables to env.go file
38 lines
947 B
Go
38 lines
947 B
Go
//go:build !ios
|
|
|
|
package net
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"sync"
|
|
|
|
"github.com/pion/transport/v3"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// ListenUDP listens on the network address and returns a transport.UDPConn
|
|
// which includes support for write and close hooks.
|
|
func ListenUDP(network string, laddr *net.UDPAddr) (transport.UDPConn, error) {
|
|
if CustomRoutingDisabled() {
|
|
return net.ListenUDP(network, laddr)
|
|
}
|
|
|
|
conn, err := NewListener().ListenPacket(context.Background(), network, laddr.String())
|
|
if err != nil {
|
|
return nil, fmt.Errorf("listen UDP: %w", err)
|
|
}
|
|
|
|
packetConn := conn.(*PacketConn)
|
|
udpConn, ok := packetConn.PacketConn.(*net.UDPConn)
|
|
if !ok {
|
|
if err := packetConn.Close(); err != nil {
|
|
log.Errorf("Failed to close connection: %v", err)
|
|
}
|
|
return nil, fmt.Errorf("expected UDPConn, got different type: %T", udpConn)
|
|
}
|
|
|
|
return &UDPConn{UDPConn: udpConn, ID: packetConn.ID, seenAddrs: &sync.Map{}}, nil
|
|
}
|