mirrors/dae
1
0
Fork 0
mirror of https://github.com/daeuniverse/dae.git synced 2025-07-16 10:50:01 +07:00
Code Issues Packages Projects Releases Wiki Activity

refactor(sniffer): refine code

Browse Source
This commit is contained in:
mzz2017
2023-04-10 00:37:48 +08:00
parent 02423fabec
commit 8465229e35
2 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
control/tcp.go
View File

@ -29,13 +29,11 @@ func (c *ControlPlane) handleConn(lConn net.Conn) (err error) {
defer lConn.Close()
// Sniff target domain.
lConn.SetReadDeadline(time.Now().Add(100 * time.Millisecond))
sniffer := sniffing.NewConnSniffer(lConn, TcpSniffBufSize)
domain, err := sniffer.SniffTcp()
if err != nil && !sniffing.IsSniffingError(err) {
return err
}
lConn.SetReadDeadline(time.Time{})
// ConnSniffer should be used later, so we cannot close it now.
defer sniffer.Close()