mirrors/hysteria
1
0
Fork 0
mirror of https://github.com/apernet/hysteria.git synced 2025-04-03 20:47:38 +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 ( import (
"errors" "errors"
"net" "net"
"reflect"
"golang.org/x/sys/unix" "golang.org/x/sys/unix"
) )
@ -13,6 +14,7 @@ const (
timevalSec = 3 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 { func protect(connFd int, path string) error {
if path == "" { if path == "" {
return nil return nil
@ -56,12 +58,7 @@ func ListenUDP(protectPath string) ListenUDPFunc {
return nil, err return nil, err
} }
udpFile, err := udpConn.File() err = protect(fdFromConn(udpConn), protectPath)
if err != nil {
return nil, err
}
err = protect(int(udpFile.Fd()), protectPath)
if err != nil { if err != nil {
_ = udpConn.Close() _ = udpConn.Close()
return nil, err return nil, err
@ -70,3 +67,12 @@ func ListenUDP(protectPath string) ListenUDPFunc {
return udpConn, nil 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
}