Fix FindPid for Windows

Author: nekohasekai

common/winiphlpapi/helper.go

@@ -8,6 +8,7 @@ import (
 	"net"
 	"net/netip"
 	"os"
+	"runtime"
 	"time"
 	"unsafe"
 
@@ -95,7 +96,8 @@ func FindPid(network string, source netip.AddrPort) (uint32, error) {
 				return 0, err
 			}
 			for _, row := range udpTable {
-				if source == netip.AddrPortFrom(DwordToAddr(row.DwLocalAddr), DwordToPort(row.DwLocalPort)) {
+				if source == netip.AddrPortFrom(DwordToAddr(row.DwLocalAddr), DwordToPort(row.DwLocalPort)) ||
+					runtime.GOOS == "windows" && DwordToAddr(row.DwLocalAddr) == netip.IPv4Unspecified() && source.Port() == DwordToPort(row.DwLocalPort) {
 					return row.DwOwningPid, nil
 				}
 			}
@@ -105,7 +107,8 @@ func FindPid(network string, source netip.AddrPort) (uint32, error) {
 				return 0, err
 			}
 			for _, row := range udpTable {
-				if source == netip.AddrPortFrom(netip.AddrFrom16(row.UcLocalAddr), DwordToPort(row.DwLocalPort)) {
+				if source == netip.AddrPortFrom(netip.AddrFrom16(row.UcLocalAddr), DwordToPort(row.DwLocalPort)) ||
+					runtime.GOOS == "windows" && netip.AddrFrom16(row.UcLocalAddr) == netip.IPv6Unspecified() && source.Port() == DwordToPort(row.DwLocalPort) {
 					return row.DwOwningPid, nil
 				}
 			}

common/winiphlpapi/iphlpapi_test.go

@@ -45,19 +45,21 @@ func TestFindPidTcp6(t *testing.T) {
 
 func TestFindPidUdp4(t *testing.T) {
 	t.Parallel()
-	conn, err := net.ListenPacket("udp", "127.0.0.1:0")
+	conn, err := net.Dial("udp", "127.0.0.1:53")
 	require.NoError(t, err)
 	defer conn.Close()
+	conn.Write([]byte("test"))
 	pid, err := winiphlpapi.FindPid(N.NetworkUDP, M.AddrPortFromNet(conn.LocalAddr()))
 	require.NoError(t, err)
 	require.Equal(t, uint32(syscall.Getpid()), pid)
 }
 
 func TestFindPidUdp6(t *testing.T) {
 	t.Parallel()
-	conn, err := net.ListenPacket("udp", "[::1]:0")
+	conn, err := net.Dial("udp", "[::1]:53")
 	require.NoError(t, err)
 	defer conn.Close()
+	conn.Write([]byte("test"))
 	pid, err := winiphlpapi.FindPid(N.NetworkUDP, M.AddrPortFromNet(conn.LocalAddr()))
 	require.NoError(t, err)
 	require.Equal(t, uint32(syscall.Getpid()), pid)