GuanM/hysteria-dev
1
0
Fork 0
mirror of https://github.com/cmz0228/hysteria-dev.git synced 2025-06-14 16:29:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: dynamic client IP for server logs (connection migration)

Browse Source
This commit is contained in:
Toby 2022-10-31 17:03:14 -07:00
parent 72b9ddcfa4
commit 21b2830289
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
pkg/core/server_client.go
View File

@ -23,7 +23,6 @@ type serverClient struct {
CC quic.Connection
Transport *transport.ServerTransport
Auth []byte
ClientAddr net.Addr
DisableUDP bool
ACLEngine *acl.Engine
CTCPRequestFunc TCPRequestFunc
@ -50,7 +49,6 @@ func newServerClient(cc quic.Connection, tr *transport.ServerTransport, auth []b
CC: cc,
Transport: tr,
Auth: auth,
ClientAddr: cc.RemoteAddr(),
DisableUDP: disableUDP,
ACLEngine: ACLEngine,
CTCPRequestFunc: CTCPRequestFunc,
@ -68,6 +66,12 @@ func newServerClient(cc quic.Connection, tr *transport.ServerTransport, auth []b
return sc
}
func (c *serverClient) ClientAddr() net.Addr {
// quic.Connection's remote address may change since we have connection migration now,
// so logs need to dynamically get the remote address every time.
return c.CC.RemoteAddr()
}
func (c *serverClient) Run() error {
if !c.DisableUDP {
go func() {
@ -200,10 +204,10 @@ func (c *serverClient) handleTCP(stream quic.Stream, host string, port uint16) {
OK: false,
Message: "host resolution failure",
})
c.CTCPErrorFunc(c.ClientAddr, c.Auth, addrStr, err)
c.CTCPErrorFunc(c.ClientAddr(), c.Auth, addrStr, err)
return
}
c.CTCPRequestFunc(c.ClientAddr, c.Auth, addrStr, action, arg)
c.CTCPRequestFunc(c.ClientAddr(), c.Auth, addrStr, action, arg)
var conn net.Conn // Connection to be piped
switch action {
@ -221,7 +225,7 @@ func (c *serverClient) handleTCP(stream quic.Stream, host string, port uint16) {
OK: false,
Message: err.Error(),
})
c.CTCPErrorFunc(c.ClientAddr, c.Auth, addrStr, err)
c.CTCPErrorFunc(c.ClientAddr(), c.Auth, addrStr, err)
return
}
case acl.ActionBlock:
@ -237,7 +241,7 @@ func (c *serverClient) handleTCP(stream quic.Stream, host string, port uint16) {
OK: false,
Message: err.Error(),
})
c.CTCPErrorFunc(c.ClientAddr, c.Auth, addrStr, err)
c.CTCPErrorFunc(c.ClientAddr(), c.Auth, addrStr, err)
return
}
addrEx := &transport.AddrEx{
@ -253,7 +257,7 @@ func (c *serverClient) handleTCP(stream quic.Stream, host string, port uint16) {
OK: false,
Message: err.Error(),
})
c.CTCPErrorFunc(c.ClientAddr, c.Auth, addrStr, err)
c.CTCPErrorFunc(c.ClientAddr(), c.Auth, addrStr, err)
return
}
default:
@ -282,7 +286,7 @@ func (c *serverClient) handleTCP(stream quic.Stream, host string, port uint16) {
} else {
err = utils.Pipe2Way(stream, conn, nil)
}
c.CTCPErrorFunc(c.ClientAddr, c.Auth, addrStr, err)
c.CTCPErrorFunc(c.ClientAddr(), c.Auth, addrStr, err)
}
func (c *serverClient) handleUDP(stream quic.Stream) {
@ -293,7 +297,7 @@ func (c *serverClient) handleUDP(stream quic.Stream) {
OK: false,
Message: "UDP initialization failed",
})
c.CUDPErrorFunc(c.ClientAddr, c.Auth, 0, err)
c.CUDPErrorFunc(c.ClientAddr(), c.Auth, 0, err)
return
}
defer conn.Close()
@ -312,7 +316,7 @@ func (c *serverClient) handleUDP(stream quic.Stream) {
if err != nil {
return
}
c.CUDPRequestFunc(c.ClientAddr, c.Auth, id)
c.CUDPRequestFunc(c.ClientAddr(), c.Auth, id)
// Receive UDP packets, send them to the client
go func() {
@ -362,7 +366,7 @@ func (c *serverClient) handleUDP(stream quic.Stream) {
break
}
}
c.CUDPErrorFunc(c.ClientAddr, c.Auth, id, err)
c.CUDPErrorFunc(c.ClientAddr(), c.Auth, id, err)
// Remove the session
c.udpSessionMutex.Lock()