clash/component/process/process_darwin.go

110 lines
2.2 KiB
Go
Raw Normal View History

package process
2020-07-19 05:17:05 +00:00
import (
"encoding/binary"
2022-04-19 17:52:51 +00:00
"net/netip"
2020-07-19 05:17:05 +00:00
"syscall"
"unsafe"
2022-04-19 17:52:51 +00:00
"github.com/Dreamacro/clash/common/nnip"
"golang.org/x/sys/unix"
2020-07-19 05:17:05 +00:00
)
const (
procpidpathinfo = 0xb
procpidpathinfosize = 1024
proccallnumpidinfo = 0x2
)
func resolveSocketByNetlink(network string, ip netip.Addr, srcPort int) (int32, int32, error) {
return 0, 0, ErrPlatformNotSupport
}
2022-04-19 17:52:51 +00:00
func findProcessName(network string, ip netip.Addr, port int) (string, error) {
var spath string
switch network {
case TCP:
spath = "net.inet.tcp.pcblist_n"
case UDP:
2020-07-19 05:17:05 +00:00
spath = "net.inet.udp.pcblist_n"
default:
return "", ErrInvalidNetwork
2020-07-19 05:17:05 +00:00
}
2022-04-19 17:52:51 +00:00
isIPv4 := ip.Is4()
2020-07-19 05:17:05 +00:00
value, err := syscall.Sysctl(spath)
if err != nil {
return "", err
}
buf := []byte(value)
// from darwin-xnu/bsd/netinet/in_pcblist.c:get_pcblist_n
// size/offset are round up (aligned) to 8 bytes in darwin
// rup8(sizeof(xinpcb_n)) + rup8(sizeof(xsocket_n)) +
// 2 * rup8(sizeof(xsockbuf_n)) + rup8(sizeof(xsockstat_n))
itemSize := 384
if network == TCP {
// rup8(sizeof(xtcpcb_n))
itemSize += 208
}
// skip the first xinpgen(24 bytes) block
for i := 24; i+itemSize <= len(buf); i += itemSize {
// offset of xinpcb_n and xsocket_n
inp, so := i, i+104
2020-07-19 05:17:05 +00:00
srcPort := binary.BigEndian.Uint16(buf[inp+18 : inp+20])
if uint16(port) != srcPort {
continue
}
// xinpcb_n.inp_vflag
flag := buf[inp+44]
2022-04-19 17:52:51 +00:00
var srcIP netip.Addr
switch {
case flag&0x1 > 0 && isIPv4:
2020-07-19 05:17:05 +00:00
// ipv4
2022-04-19 17:52:51 +00:00
srcIP = nnip.IpToAddr(buf[inp+76 : inp+80])
case flag&0x2 > 0 && !isIPv4:
2020-07-19 05:17:05 +00:00
// ipv6
2022-04-19 17:52:51 +00:00
srcIP = nnip.IpToAddr(buf[inp+64 : inp+80])
default:
2020-07-19 05:17:05 +00:00
continue
}
2022-04-19 17:52:51 +00:00
if ip != srcIP && (network == TCP || !srcIP.IsUnspecified()) {
2020-07-19 05:17:05 +00:00
continue
}
// xsocket_n.so_last_pid
pid := readNativeUint32(buf[so+68 : so+72])
2020-07-19 05:17:05 +00:00
return getExecPathFromPID(pid)
}
return "", ErrNotFound
}
func getExecPathFromPID(pid uint32) (string, error) {
buf := make([]byte, procpidpathinfosize)
_, _, errno := syscall.Syscall6(
syscall.SYS_PROC_INFO,
proccallnumpidinfo,
uintptr(pid),
procpidpathinfo,
0,
uintptr(unsafe.Pointer(&buf[0])),
procpidpathinfosize)
if errno != 0 {
return "", errno
}
return unix.ByteSliceToString(buf), nil
2020-07-19 05:17:05 +00:00
}
func readNativeUint32(b []byte) uint32 {
return *(*uint32)(unsafe.Pointer(&b[0]))
2020-07-19 05:17:05 +00:00
}