mirror of
https://github.com/SagerNet/sing-box.git
synced 2026-04-11 17:47:20 +10:00
61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
//go:build linux
|
|
|
|
package process
|
|
|
|
import (
|
|
"net"
|
|
"net/netip"
|
|
"os"
|
|
"syscall"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestQuerySocketDiagUDPExact(t *testing.T) {
|
|
t.Parallel()
|
|
server, err := net.ListenUDP("udp4", &net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 0})
|
|
require.NoError(t, err)
|
|
defer server.Close()
|
|
|
|
client, err := net.DialUDP("udp4", nil, server.LocalAddr().(*net.UDPAddr))
|
|
require.NoError(t, err)
|
|
defer client.Close()
|
|
|
|
err = client.SetDeadline(time.Now().Add(time.Second))
|
|
require.NoError(t, err)
|
|
_, err = client.Write([]byte{0})
|
|
require.NoError(t, err)
|
|
|
|
err = server.SetReadDeadline(time.Now().Add(time.Second))
|
|
require.NoError(t, err)
|
|
buffer := make([]byte, 1)
|
|
_, _, err = server.ReadFromUDP(buffer)
|
|
require.NoError(t, err)
|
|
|
|
source := addrPortFromUDPAddr(t, client.LocalAddr())
|
|
destination := addrPortFromUDPAddr(t, client.RemoteAddr())
|
|
|
|
fd, err := openSocketDiag()
|
|
require.NoError(t, err)
|
|
defer syscall.Close(fd)
|
|
|
|
inode, uid, err := querySocketDiag(fd, packSocketDiagRequest(syscall.AF_INET, syscall.IPPROTO_UDP, source, destination, false))
|
|
require.NoError(t, err)
|
|
require.NotZero(t, inode)
|
|
require.EqualValues(t, os.Getuid(), uid)
|
|
}
|
|
|
|
func addrPortFromUDPAddr(t *testing.T, addr net.Addr) netip.AddrPort {
|
|
t.Helper()
|
|
|
|
udpAddr, ok := addr.(*net.UDPAddr)
|
|
require.True(t, ok)
|
|
|
|
ip, ok := netip.AddrFromSlice(udpAddr.IP)
|
|
require.True(t, ok)
|
|
|
|
return netip.AddrPortFrom(ip.Unmap(), uint16(udpAddr.Port))
|
|
}
|