Shikong/Clash.Meta
1
0
Fork 0
mirror of https://gitclone.com/github.com/MetaCubeX/Clash.Meta synced 2024-11-15 13:41:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: tuic server set authentication timeout after quic handshake complete

Browse Source
This commit is contained in:
wwqgtxx 2023-03-10 12:26:17 +08:00
parent 9ae0bd9c2b
commit 6fe7f4641e
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
transport/tuic/server.go
View File

@ -76,7 +76,7 @@ func (s *Server) Close() error {
type serverHandler struct { type serverHandler struct {
*Server *Server
quicConn quic.Connection quicConn quic.EarlyConnection
uuid uuid.UUID uuid uuid.UUID
authCh chan struct{} authCh chan struct{}
@ -87,13 +87,6 @@ type serverHandler struct {
} }
func (s *serverHandler) handle() { func (s *serverHandler) handle() {
time.AfterFunc(s.AuthenticationTimeout, func() {
s.authOnce.Do(func() {
_ = s.quicConn.CloseWithError(AuthenticationTimeout, "AuthenticationTimeout")
s.authOk = false
close(s.authCh)
})
})
go func() { go func() {
_ = s.handleUniStream() _ = s.handleUniStream()
}() }()
@ -103,6 +96,15 @@ func (s *serverHandler) handle() {
go func() { go func() {
_ = s.handleMessage() _ = s.handleMessage()
}() }()
<-s.quicConn.HandshakeComplete().Done()
time.AfterFunc(s.AuthenticationTimeout, func() {
s.authOnce.Do(func() {
_ = s.quicConn.CloseWithError(AuthenticationTimeout, "AuthenticationTimeout")
s.authOk = false
close(s.authCh)
})
})
} }
func (s *serverHandler) handleMessage() (err error) { func (s *serverHandler) handleMessage() (err error) {