package/tun2socks
1
0
Fork 0
mirror of https://github.com/xjasonlyu/tun2socks.git synced 2025-11-02 21:04:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix process name empty

Browse Source
This commit is contained in:
Jason
2019-08-10 22:38:30 +08:00
parent bb59dfed62
commit 0025491b5e
2 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
proxy/socks/tcp.go
View File

@@ -93,13 +93,16 @@ func (h *tcpHandler) Handle(localConn net.Conn, target *net.TCPAddr) error {
return err return err
} }
var process = "N/A" var process string
var sess *stats.Session var sess *stats.Session
if h.sessionStater != nil { if h.sessionStater != nil {
// Get name of the process. // Get name of the process.
localHost, localPortStr, _ := net.SplitHostPort(localConn.LocalAddr().String()) localHost, localPortStr, _ := net.SplitHostPort(localConn.LocalAddr().String())
localPortInt, _ := strconv.Atoi(localPortStr) localPortInt, _ := strconv.Atoi(localPortStr)
process, _ = lsof.GetCommandNameBySocket(target.Network(), localHost, uint16(localPortInt)) process, err = lsof.GetCommandNameBySocket(target.Network(), localHost, uint16(localPortInt))
if err != nil {
process = "N/A"
}
sess = &stats.Session{ sess = &stats.Session{
ProcessName: process, ProcessName: process,

7
proxy/socks/udp.go
View File

@@ -164,12 +164,15 @@ func (h *udpHandler) connectInternal(conn core.UDPConn, targetAddr string) error
return err return err
} }
var process = "N/A" var process string
if h.sessionStater != nil { if h.sessionStater != nil {
// Get name of the process. // Get name of the process.
localHost, localPortStr, _ := net.SplitHostPort(conn.LocalAddr().String()) localHost, localPortStr, _ := net.SplitHostPort(conn.LocalAddr().String())
localPortInt, _ := strconv.Atoi(localPortStr) localPortInt, _ := strconv.Atoi(localPortStr)
process, _ = lsof.GetCommandNameBySocket(conn.LocalAddr().Network(), localHost, uint16(localPortInt)) process, err = lsof.GetCommandNameBySocket(conn.LocalAddr().Network(), localHost, uint16(localPortInt))
if err != nil {
process = "N/A"
}
sess := &stats.Session{ sess := &stats.Session{
ProcessName: process, ProcessName: process,