mirrors/hysteria
1
0
Fork 0
mirror of https://github.com/apernet/hysteria.git synced 2025-04-03 04:27:39 +03:00
Code Issues Projects Releases Packages Wiki Activity

fix: use reflect to get fd

Browse source 
conn.File() not returns real file.

Signed-off-by: HystericalDragon <HystericalDragons@proton.me>
This commit is contained in:
HystericalDragon 2024-04-03 18:40:25 +08:00
parent 03c8b5e6b9
commit a05383c2a1
No known key found for this signature in database
GPG key ID: E205724F8931D99B
1 changed files with 12 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

18
extras/protect/protect_linux.go
View file

@ -5,6 +5,7 @@ package protect
import (
"errors"
"net"
"reflect"
"golang.org/x/sys/unix"
)
@ -13,6 +14,7 @@ const (
timevalSec = 3
)
// protect try to connect with path by unix socket, then send the conn's fd to it.
func protect(connFd int, path string) error {
if path == "" {
return nil
@ -56,12 +58,7 @@ func ListenUDP(protectPath string) ListenUDPFunc {
return nil, err
}
udpFile, err := udpConn.File()
if err != nil {
return nil, err
}
err = protect(int(udpFile.Fd()), protectPath)
err = protect(fdFromConn(udpConn), protectPath)
if err != nil {
_ = udpConn.Close()
return nil, err
@ -70,3 +67,12 @@ func ListenUDP(protectPath string) ListenUDPFunc {
return udpConn, nil
}
}
// fdFromConn get net.Conn's file descriptor.
func fdFromConn(conn net.Conn) int {
v := reflect.ValueOf(conn)
netFD := reflect.Indirect(reflect.Indirect(v).FieldByName("fd"))
pfd := reflect.Indirect(netFD.FieldByName("pfd"))
fd := int(pfd.FieldByName("Sysfd").Int())
return fd
}