Backups/mihomo
1
0
Fork 0
mirror of https://github.com/MetaCubeX/mihomo.git synced 2025-01-09 10:49:35 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: tuic-server support heartbeat command

Browse Source
This commit is contained in:
gVisor bot 2022-11-28 18:28:22 +08:00
parent 81e03ec904
commit df306be1ec
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
transport/tuic/protocol.go
View File

@ -349,7 +349,7 @@ func NewHeartbeat() Heartbeat {
}
}
func ReadHeartbeatWithHead(head CommandHead, reader BufferedReader) (c Response, err error) {
func ReadHeartbeatWithHead(head CommandHead, reader BufferedReader) (c Heartbeat, err error) {
c.CommandHead = head
if c.CommandHead.TYPE != HeartbeatType {
err = fmt.Errorf("error command type: %s", c.CommandHead.TYPE)
@ -357,7 +357,7 @@ func ReadHeartbeatWithHead(head CommandHead, reader BufferedReader) (c Response,
return
}
func ReadHeartbeat(reader BufferedReader) (c Response, err error) {
func ReadHeartbeat(reader BufferedReader) (c Heartbeat, err error) {
head, err := ReadCommandHead(reader)
if err != nil {
return

7
transport/tuic/server.go
View File

@ -261,6 +261,13 @@ func (s *serverHandler) handleUniStream() (err error) {
writeClosed := v.(*atomic.Bool)
writeClosed.Store(true)
}
case HeartbeatType:
var heartbeat Heartbeat
heartbeat, err = ReadHeartbeatWithHead(commandHead, reader)
if err != nil {
return
}
heartbeat.BytesLen()
}
return
}()