xmengnet/mihomo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: code logic error

Browse source
This commit is contained in:
gVisor bot 2022-04-10 20:01:35 +08:00
parent b6ea3eac16
commit d7d6ed5c70
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
component/sniffer/dispatcher.go
View file

@ -46,9 +46,13 @@ func (sd *SnifferDispatcher) Enable() bool {
func (sd *SnifferDispatcher) cover(conn *CN.BufferedConn, metadata *C.Metadata) {
for _, sniffer := range sd.sniffers {
if sniffer.SupportNetwork() == C.TCP {
conn.Peek(1)
len := conn.Buffered()
bytes, err := conn.Peek(len)
_, err := conn.Peek(1)
if err != nil {
return
}
bufferedLen := conn.Buffered()
bytes, err := conn.Peek(bufferedLen)
if err != nil {
log.Warnln("the data lenght not enough")
continue
@ -61,7 +65,6 @@ func (sd *SnifferDispatcher) cover(conn *CN.BufferedConn, metadata *C.Metadata)
}
metadata.Host = host
metadata.DstIP = nil
metadata.AddrType = C.AtypDomainName
if resolver.FakeIPEnabled() {
metadata.DNSMode = C.DNSFakeIP
@ -70,6 +73,8 @@ func (sd *SnifferDispatcher) cover(conn *CN.BufferedConn, metadata *C.Metadata)
}
resolver.InsertHostByIP(metadata.DstIP, host)
metadata.DstIP = nil
break
}
}