chore: adjust code
This commit is contained in:
parent
7166db2ac9
commit
83e0abaa8c
2 changed files with 6 additions and 5 deletions
|
@ -15,6 +15,9 @@ func LoaderName() string {
|
|||
}
|
||||
|
||||
func SetLoader(newLoader string) {
|
||||
if newLoader == "memc" {
|
||||
newLoader = "memconservative"
|
||||
}
|
||||
geoLoaderName = newLoader
|
||||
}
|
||||
|
||||
|
|
|
@ -34,7 +34,6 @@ func (sd *SnifferDispatcher) Tcp(conn net.Conn, metadata *C.Metadata) {
|
|||
if metadata.Host != "" {
|
||||
return
|
||||
}
|
||||
|
||||
sd.cover(bufConn, metadata)
|
||||
}
|
||||
}
|
||||
|
@ -54,24 +53,23 @@ func (sd *SnifferDispatcher) cover(conn *CN.BufferedConn, metadata *C.Metadata)
|
|||
bufferedLen := conn.Buffered()
|
||||
bytes, err := conn.Peek(bufferedLen)
|
||||
if err != nil {
|
||||
log.Warnln("the data lenght not enough")
|
||||
log.Debugln("[Sniffer] the data lenght not enough")
|
||||
continue
|
||||
}
|
||||
|
||||
host, err := sniffer.SniffTCP(bytes)
|
||||
if err != nil {
|
||||
log.Warnln("Sniff data failed on Sniffer[%s]", sniffer.Protocol())
|
||||
log.Debugln("[Sniffer][%s] Sniff data failed", sniffer.Protocol())
|
||||
continue
|
||||
}
|
||||
|
||||
metadata.Host = host
|
||||
metadata.AddrType = C.AtypDomainName
|
||||
log.Debugln("[Sniffer][%s] %s --> %s", sniffer.Protocol(), metadata.DstIP, metadata.Host)
|
||||
if resolver.FakeIPEnabled() {
|
||||
metadata.DNSMode = C.DNSFakeIP
|
||||
} else {
|
||||
metadata.DNSMode = C.DNSMapping
|
||||
}
|
||||
|
||||
resolver.InsertHostByIP(metadata.DstIP, host)
|
||||
metadata.DstIP = nil
|
||||
|
||||
|
|
Loading…
Reference in a new issue